536d41baa2bc4848b334405afa598a210dc5422e
[lttng-tools.git] / src / lib / lttng-ctl / lttng-ctl.c
1 /*
2 * liblttngctl.c
3 *
4 * Linux Trace Toolkit Control Library
5 *
6 * Copyright (C) 2011 David Goulet <david.goulet@polymtl.ca>
7 *
8 * This library is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU Lesser General Public License, version 2.1 only,
10 * as published by the Free Software Foundation.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public License
18 * along with this library; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #define _GNU_SOURCE
23 #include <assert.h>
24 #include <grp.h>
25 #include <errno.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <unistd.h>
30
31 #include <common/common.h>
32 #include <common/defaults.h>
33 #include <common/sessiond-comm/sessiond-comm.h>
34 #include <common/uri.h>
35 #include <common/utils.h>
36 #include <lttng/lttng.h>
37 #include <lttng/health-internal.h>
38
39 #include "filter/filter-ast.h"
40 #include "filter/filter-parser.h"
41 #include "filter/filter-bytecode.h"
42 #include "filter/memstream.h"
43 #include "lttng-ctl-helper.h"
44
45 #ifdef DEBUG
46 static const int print_xml = 1;
47 #define dbg_printf(fmt, args...) \
48 printf("[debug liblttng-ctl] " fmt, ## args)
49 #else
50 static const int print_xml = 0;
51 #define dbg_printf(fmt, args...) \
52 do { \
53 /* do nothing but check printf format */ \
54 if (0) \
55 printf("[debug liblttnctl] " fmt, ## args); \
56 } while (0)
57 #endif
58
59
60 /* Socket to session daemon for communication */
61 static int sessiond_socket;
62 static char sessiond_sock_path[PATH_MAX];
63
64 /* Variables */
65 static char *tracing_group;
66 static int connected;
67
68 /* Global */
69
70 /*
71 * Those two variables are used by error.h to silent or control the verbosity of
72 * error message. They are global to the library so application linking with it
73 * are able to compile correctly and also control verbosity of the library.
74 */
75 int lttng_opt_quiet;
76 int lttng_opt_verbose;
77 int lttng_opt_mi;
78
79 /*
80 * Copy string from src to dst and enforce null terminated byte.
81 */
82 LTTNG_HIDDEN
83 void lttng_ctl_copy_string(char *dst, const char *src, size_t len)
84 {
85 if (src && dst) {
86 strncpy(dst, src, len);
87 /* Enforce the NULL terminated byte */
88 dst[len - 1] = '\0';
89 } else if (dst) {
90 dst[0] = '\0';
91 }
92 }
93
94 /*
95 * Copy domain to lttcomm_session_msg domain.
96 *
97 * If domain is unknown, default domain will be the kernel.
98 */
99 LTTNG_HIDDEN
100 void lttng_ctl_copy_lttng_domain(struct lttng_domain *dst,
101 struct lttng_domain *src)
102 {
103 if (src && dst) {
104 switch (src->type) {
105 case LTTNG_DOMAIN_KERNEL:
106 case LTTNG_DOMAIN_UST:
107 case LTTNG_DOMAIN_JUL:
108 memcpy(dst, src, sizeof(struct lttng_domain));
109 break;
110 default:
111 memset(dst, 0, sizeof(struct lttng_domain));
112 break;
113 }
114 }
115 }
116
117 /*
118 * Send lttcomm_session_msg to the session daemon.
119 *
120 * On success, returns the number of bytes sent (>=0)
121 * On error, returns -1
122 */
123 static int send_session_msg(struct lttcomm_session_msg *lsm)
124 {
125 int ret;
126
127 if (!connected) {
128 ret = -LTTNG_ERR_NO_SESSIOND;
129 goto end;
130 }
131
132 DBG("LSM cmd type : %d", lsm->cmd_type);
133
134 ret = lttcomm_send_creds_unix_sock(sessiond_socket, lsm,
135 sizeof(struct lttcomm_session_msg));
136 if (ret < 0) {
137 ret = -LTTNG_ERR_FATAL;
138 }
139
140 end:
141 return ret;
142 }
143
144 /*
145 * Send var len data to the session daemon.
146 *
147 * On success, returns the number of bytes sent (>=0)
148 * On error, returns -1
149 */
150 static int send_session_varlen(void *data, size_t len)
151 {
152 int ret;
153
154 if (!connected) {
155 ret = -LTTNG_ERR_NO_SESSIOND;
156 goto end;
157 }
158
159 if (!data || !len) {
160 ret = 0;
161 goto end;
162 }
163
164 ret = lttcomm_send_unix_sock(sessiond_socket, data, len);
165 if (ret < 0) {
166 ret = -LTTNG_ERR_FATAL;
167 }
168
169 end:
170 return ret;
171 }
172
173 /*
174 * Receive data from the sessiond socket.
175 *
176 * On success, returns the number of bytes received (>=0)
177 * On error, returns -1 (recvmsg() error) or -ENOTCONN
178 */
179 static int recv_data_sessiond(void *buf, size_t len)
180 {
181 int ret;
182
183 if (!connected) {
184 ret = -LTTNG_ERR_NO_SESSIOND;
185 goto end;
186 }
187
188 ret = lttcomm_recv_unix_sock(sessiond_socket, buf, len);
189 if (ret < 0) {
190 ret = -LTTNG_ERR_FATAL;
191 }
192
193 end:
194 return ret;
195 }
196
197 /*
198 * Check if we are in the specified group.
199 *
200 * If yes return 1, else return -1.
201 */
202 LTTNG_HIDDEN
203 int lttng_check_tracing_group(void)
204 {
205 struct group *grp_tracing; /* no free(). See getgrnam(3) */
206 gid_t *grp_list;
207 int grp_list_size, grp_id, i;
208 int ret = -1;
209 const char *grp_name = tracing_group;
210
211 /* Get GID of group 'tracing' */
212 grp_tracing = getgrnam(grp_name);
213 if (!grp_tracing) {
214 /* If grp_tracing is NULL, the group does not exist. */
215 goto end;
216 }
217
218 /* Get number of supplementary group IDs */
219 grp_list_size = getgroups(0, NULL);
220 if (grp_list_size < 0) {
221 perror("getgroups");
222 goto end;
223 }
224
225 /* Alloc group list of the right size */
226 grp_list = malloc(grp_list_size * sizeof(gid_t));
227 if (!grp_list) {
228 perror("malloc");
229 goto end;
230 }
231 grp_id = getgroups(grp_list_size, grp_list);
232 if (grp_id < 0) {
233 perror("getgroups");
234 goto free_list;
235 }
236
237 for (i = 0; i < grp_list_size; i++) {
238 if (grp_list[i] == grp_tracing->gr_gid) {
239 ret = 1;
240 break;
241 }
242 }
243
244 free_list:
245 free(grp_list);
246
247 end:
248 return ret;
249 }
250
251 /*
252 * Try connect to session daemon with sock_path.
253 *
254 * Return 0 on success, else -1
255 */
256 static int try_connect_sessiond(const char *sock_path)
257 {
258 int ret;
259
260 /* If socket exist, we check if the daemon listens for connect. */
261 ret = access(sock_path, F_OK);
262 if (ret < 0) {
263 /* Not alive */
264 goto error;
265 }
266
267 ret = lttcomm_connect_unix_sock(sock_path);
268 if (ret < 0) {
269 /* Not alive */
270 goto error;
271 }
272
273 ret = lttcomm_close_unix_sock(ret);
274 if (ret < 0) {
275 perror("lttcomm_close_unix_sock");
276 }
277
278 return 0;
279
280 error:
281 return -1;
282 }
283
284 /*
285 * Set sessiond socket path by putting it in the global sessiond_sock_path
286 * variable.
287 *
288 * Returns 0 on success, negative value on failure (the sessiond socket path
289 * is somehow too long or ENOMEM).
290 */
291 static int set_session_daemon_path(void)
292 {
293 int in_tgroup = 0; /* In tracing group */
294 uid_t uid;
295
296 uid = getuid();
297
298 if (uid != 0) {
299 /* Are we in the tracing group ? */
300 in_tgroup = lttng_check_tracing_group();
301 }
302
303 if ((uid == 0) || in_tgroup) {
304 lttng_ctl_copy_string(sessiond_sock_path,
305 DEFAULT_GLOBAL_CLIENT_UNIX_SOCK, sizeof(sessiond_sock_path));
306 }
307
308 if (uid != 0) {
309 int ret;
310
311 if (in_tgroup) {
312 /* Tracing group */
313 ret = try_connect_sessiond(sessiond_sock_path);
314 if (ret >= 0) {
315 goto end;
316 }
317 /* Global session daemon not available... */
318 }
319 /* ...or not in tracing group (and not root), default */
320
321 /*
322 * With GNU C < 2.1, snprintf returns -1 if the target buffer is too small;
323 * With GNU C >= 2.1, snprintf returns the required size (excluding closing null)
324 */
325 ret = snprintf(sessiond_sock_path, sizeof(sessiond_sock_path),
326 DEFAULT_HOME_CLIENT_UNIX_SOCK, utils_get_home_dir());
327 if ((ret < 0) || (ret >= sizeof(sessiond_sock_path))) {
328 goto error;
329 }
330 }
331 end:
332 return 0;
333
334 error:
335 return -1;
336 }
337
338 /*
339 * Connect to the LTTng session daemon.
340 *
341 * On success, return 0. On error, return -1.
342 */
343 static int connect_sessiond(void)
344 {
345 int ret;
346
347 /* Don't try to connect if already connected. */
348 if (connected) {
349 return 0;
350 }
351
352 ret = set_session_daemon_path();
353 if (ret < 0) {
354 goto error;
355 }
356
357 /* Connect to the sesssion daemon */
358 ret = lttcomm_connect_unix_sock(sessiond_sock_path);
359 if (ret < 0) {
360 goto error;
361 }
362
363 sessiond_socket = ret;
364 connected = 1;
365
366 return 0;
367
368 error:
369 return -1;
370 }
371
372 /*
373 * Clean disconnect from the session daemon.
374 * On success, return 0. On error, return -1.
375 */
376 static int disconnect_sessiond(void)
377 {
378 int ret = 0;
379
380 if (connected) {
381 ret = lttcomm_close_unix_sock(sessiond_socket);
382 sessiond_socket = 0;
383 connected = 0;
384 }
385
386 return ret;
387 }
388
389 /*
390 * Ask the session daemon a specific command and put the data into buf.
391 * Takes extra var. len. data as input to send to the session daemon.
392 *
393 * Return size of data (only payload, not header) or a negative error code.
394 */
395 LTTNG_HIDDEN
396 int lttng_ctl_ask_sessiond_varlen(struct lttcomm_session_msg *lsm,
397 void *vardata, size_t varlen, void **buf)
398 {
399 int ret;
400 size_t size;
401 void *data = NULL;
402 struct lttcomm_lttng_msg llm;
403
404 ret = connect_sessiond();
405 if (ret < 0) {
406 ret = -LTTNG_ERR_NO_SESSIOND;
407 goto end;
408 }
409
410 /* Send command to session daemon */
411 ret = send_session_msg(lsm);
412 if (ret < 0) {
413 /* Ret value is a valid lttng error code. */
414 goto end;
415 }
416 /* Send var len data */
417 ret = send_session_varlen(vardata, varlen);
418 if (ret < 0) {
419 /* Ret value is a valid lttng error code. */
420 goto end;
421 }
422
423 /* Get header from data transmission */
424 ret = recv_data_sessiond(&llm, sizeof(llm));
425 if (ret < 0) {
426 /* Ret value is a valid lttng error code. */
427 goto end;
428 }
429
430 /* Check error code if OK */
431 if (llm.ret_code != LTTNG_OK) {
432 ret = -llm.ret_code;
433 goto end;
434 }
435
436 size = llm.data_size;
437 if (size == 0) {
438 /* If client free with size 0 */
439 if (buf != NULL) {
440 *buf = NULL;
441 }
442 ret = 0;
443 goto end;
444 }
445
446 data = (void*) malloc(size);
447
448 /* Get payload data */
449 ret = recv_data_sessiond(data, size);
450 if (ret < 0) {
451 free(data);
452 goto end;
453 }
454
455 /*
456 * Extra protection not to dereference a NULL pointer. If buf is NULL at
457 * this point, an error is returned and data is freed.
458 */
459 if (buf == NULL) {
460 ret = -LTTNG_ERR_INVALID;
461 free(data);
462 goto end;
463 }
464
465 *buf = data;
466 ret = size;
467
468 end:
469 disconnect_sessiond();
470 return ret;
471 }
472
473 /*
474 * Create lttng handle and return pointer.
475 * The returned pointer will be NULL in case of malloc() error.
476 */
477 struct lttng_handle *lttng_create_handle(const char *session_name,
478 struct lttng_domain *domain)
479 {
480 struct lttng_handle *handle = NULL;
481
482 if (domain == NULL) {
483 goto end;
484 }
485
486 handle = malloc(sizeof(struct lttng_handle));
487 if (handle == NULL) {
488 PERROR("malloc handle");
489 goto end;
490 }
491
492 /* Copy session name */
493 lttng_ctl_copy_string(handle->session_name, session_name,
494 sizeof(handle->session_name));
495
496 /* Copy lttng domain */
497 lttng_ctl_copy_lttng_domain(&handle->domain, domain);
498
499 end:
500 return handle;
501 }
502
503 /*
504 * Destroy handle by free(3) the pointer.
505 */
506 void lttng_destroy_handle(struct lttng_handle *handle)
507 {
508 free(handle);
509 }
510
511 /*
512 * Register an outside consumer.
513 * Returns size of returned session payload data or a negative error code.
514 */
515 int lttng_register_consumer(struct lttng_handle *handle,
516 const char *socket_path)
517 {
518 struct lttcomm_session_msg lsm;
519
520 if (handle == NULL || socket_path == NULL) {
521 return -LTTNG_ERR_INVALID;
522 }
523
524 memset(&lsm, 0, sizeof(lsm));
525 lsm.cmd_type = LTTNG_REGISTER_CONSUMER;
526 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
527 sizeof(lsm.session.name));
528 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
529
530 lttng_ctl_copy_string(lsm.u.reg.path, socket_path, sizeof(lsm.u.reg.path));
531
532 return lttng_ctl_ask_sessiond(&lsm, NULL);
533 }
534
535 /*
536 * Start tracing for all traces of the session.
537 * Returns size of returned session payload data or a negative error code.
538 */
539 int lttng_start_tracing(const char *session_name)
540 {
541 struct lttcomm_session_msg lsm;
542
543 if (session_name == NULL) {
544 return -LTTNG_ERR_INVALID;
545 }
546
547 memset(&lsm, 0, sizeof(lsm));
548 lsm.cmd_type = LTTNG_START_TRACE;
549
550 lttng_ctl_copy_string(lsm.session.name, session_name,
551 sizeof(lsm.session.name));
552
553 return lttng_ctl_ask_sessiond(&lsm, NULL);
554 }
555
556 /*
557 * Stop tracing for all traces of the session.
558 */
559 static int _lttng_stop_tracing(const char *session_name, int wait)
560 {
561 int ret, data_ret;
562 struct lttcomm_session_msg lsm;
563
564 if (session_name == NULL) {
565 return -LTTNG_ERR_INVALID;
566 }
567
568 memset(&lsm, 0, sizeof(lsm));
569 lsm.cmd_type = LTTNG_STOP_TRACE;
570
571 lttng_ctl_copy_string(lsm.session.name, session_name,
572 sizeof(lsm.session.name));
573
574 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
575 if (ret < 0 && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
576 goto error;
577 }
578
579 if (!wait) {
580 goto end;
581 }
582
583 /* Check for data availability */
584 do {
585 data_ret = lttng_data_pending(session_name);
586 if (data_ret < 0) {
587 /* Return the data available call error. */
588 ret = data_ret;
589 goto error;
590 }
591
592 /*
593 * Data sleep time before retrying (in usec). Don't sleep if the call
594 * returned value indicates availability.
595 */
596 if (data_ret) {
597 usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME);
598 }
599 } while (data_ret != 0);
600
601 end:
602 error:
603 return ret;
604 }
605
606 /*
607 * Stop tracing and wait for data availability.
608 */
609 int lttng_stop_tracing(const char *session_name)
610 {
611 return _lttng_stop_tracing(session_name, 1);
612 }
613
614 /*
615 * Stop tracing but _don't_ wait for data availability.
616 */
617 int lttng_stop_tracing_no_wait(const char *session_name)
618 {
619 return _lttng_stop_tracing(session_name, 0);
620 }
621
622 /*
623 * Add context to a channel.
624 *
625 * If the given channel is NULL, add the contexts to all channels.
626 * The event_name param is ignored.
627 *
628 * Returns the size of the returned payload data or a negative error code.
629 */
630 int lttng_add_context(struct lttng_handle *handle,
631 struct lttng_event_context *ctx, const char *event_name,
632 const char *channel_name)
633 {
634 struct lttcomm_session_msg lsm;
635
636 /* Safety check. Both are mandatory */
637 if (handle == NULL || ctx == NULL) {
638 return -LTTNG_ERR_INVALID;
639 }
640
641 memset(&lsm, 0, sizeof(lsm));
642
643 lsm.cmd_type = LTTNG_ADD_CONTEXT;
644
645 /* If no channel name, send empty string. */
646 if (channel_name == NULL) {
647 lttng_ctl_copy_string(lsm.u.context.channel_name, "",
648 sizeof(lsm.u.context.channel_name));
649 } else {
650 lttng_ctl_copy_string(lsm.u.context.channel_name, channel_name,
651 sizeof(lsm.u.context.channel_name));
652 }
653
654 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
655
656 memcpy(&lsm.u.context.ctx, ctx, sizeof(struct lttng_event_context));
657
658 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
659 sizeof(lsm.session.name));
660
661 return lttng_ctl_ask_sessiond(&lsm, NULL);
662 }
663
664 /*
665 * Enable event(s) for a channel.
666 * If no event name is specified, all events are enabled.
667 * If no channel name is specified, the default 'channel0' is used.
668 * Returns size of returned session payload data or a negative error code.
669 */
670 int lttng_enable_event(struct lttng_handle *handle,
671 struct lttng_event *ev, const char *channel_name)
672 {
673 return lttng_enable_event_with_exclusions(handle, ev, channel_name,
674 NULL, 0, NULL);
675 }
676
677 /*
678 * Create or enable an event with a filter expression.
679 *
680 * Return negative error value on error.
681 * Return size of returned session payload data if OK.
682 */
683 int lttng_enable_event_with_filter(struct lttng_handle *handle,
684 struct lttng_event *event, const char *channel_name,
685 const char *filter_expression)
686 {
687 return lttng_enable_event_with_exclusions(handle, event, channel_name,
688 filter_expression, 0, NULL);
689 }
690
691 /*
692 * Depending on the event, return a newly allocated JUL filter expression or
693 * NULL if not applicable.
694 *
695 * An event with NO loglevel and the name is * will return NULL.
696 */
697 static char *set_jul_filter(const char *filter, struct lttng_event *ev)
698 {
699 int err;
700 char *jul_filter = NULL;
701
702 assert(ev);
703
704 /* Don't add filter for the '*' event. */
705 if (ev->name[0] != '*') {
706 if (filter) {
707 err = asprintf(&jul_filter, "%s && logger_name == \"%s\"", filter,
708 ev->name);
709 } else {
710 err = asprintf(&jul_filter, "logger_name == \"%s\"", ev->name);
711 }
712 if (err < 0) {
713 PERROR("asprintf");
714 goto error;
715 }
716 }
717
718 /* Add loglevel filtering if any for the JUL domain. */
719 if (ev->loglevel_type != LTTNG_EVENT_LOGLEVEL_ALL) {
720 char *op;
721
722 if (ev->loglevel_type == LTTNG_EVENT_LOGLEVEL_RANGE) {
723 op = ">=";
724 } else {
725 op = "==";
726 }
727
728 if (filter || jul_filter) {
729 char *new_filter;
730
731 err = asprintf(&new_filter, "%s && int_loglevel %s %d",
732 jul_filter ? jul_filter : filter, op,
733 ev->loglevel);
734 if (jul_filter) {
735 free(jul_filter);
736 }
737 jul_filter = new_filter;
738 } else {
739 err = asprintf(&jul_filter, "int_loglevel %s %d", op,
740 ev->loglevel);
741 }
742 if (err < 0) {
743 PERROR("asprintf");
744 goto error;
745 }
746 }
747
748 return jul_filter;
749 error:
750 free(jul_filter);
751 return NULL;
752 }
753
754 /*
755 * Generate the filter bytecode from a give filter expression string. Put the
756 * newly allocated parser context in ctxp and populate the lsm object with the
757 * expression len.
758 *
759 * Return 0 on success else a LTTNG_ERR_* code and ctxp is untouched.
760 */
761 static int generate_filter(char *filter_expression,
762 struct lttcomm_session_msg *lsm, struct filter_parser_ctx **ctxp)
763 {
764 int ret;
765 struct filter_parser_ctx *ctx = NULL;
766 FILE *fmem = NULL;
767
768 assert(filter_expression);
769 assert(lsm);
770 assert(ctxp);
771
772 /*
773 * Casting const to non-const, as the underlying function will use it in
774 * read-only mode.
775 */
776 fmem = lttng_fmemopen((void *) filter_expression,
777 strlen(filter_expression), "r");
778 if (!fmem) {
779 fprintf(stderr, "Error opening memory as stream\n");
780 ret = -LTTNG_ERR_FILTER_NOMEM;
781 goto error;
782 }
783 ctx = filter_parser_ctx_alloc(fmem);
784 if (!ctx) {
785 fprintf(stderr, "Error allocating parser\n");
786 ret = -LTTNG_ERR_FILTER_NOMEM;
787 goto filter_alloc_error;
788 }
789 ret = filter_parser_ctx_append_ast(ctx);
790 if (ret) {
791 fprintf(stderr, "Parse error\n");
792 ret = -LTTNG_ERR_FILTER_INVAL;
793 goto parse_error;
794 }
795 ret = filter_visitor_set_parent(ctx);
796 if (ret) {
797 fprintf(stderr, "Set parent error\n");
798 ret = -LTTNG_ERR_FILTER_INVAL;
799 goto parse_error;
800 }
801 if (print_xml) {
802 ret = filter_visitor_print_xml(ctx, stdout, 0);
803 if (ret) {
804 fflush(stdout);
805 fprintf(stderr, "XML print error\n");
806 ret = -LTTNG_ERR_FILTER_INVAL;
807 goto parse_error;
808 }
809 }
810
811 dbg_printf("Generating IR... ");
812 fflush(stdout);
813 ret = filter_visitor_ir_generate(ctx);
814 if (ret) {
815 fprintf(stderr, "Generate IR error\n");
816 ret = -LTTNG_ERR_FILTER_INVAL;
817 goto parse_error;
818 }
819 dbg_printf("done\n");
820
821 dbg_printf("Validating IR... ");
822 fflush(stdout);
823 ret = filter_visitor_ir_check_binary_op_nesting(ctx);
824 if (ret) {
825 ret = -LTTNG_ERR_FILTER_INVAL;
826 goto parse_error;
827 }
828 dbg_printf("done\n");
829
830 dbg_printf("Generating bytecode... ");
831 fflush(stdout);
832 ret = filter_visitor_bytecode_generate(ctx);
833 if (ret) {
834 fprintf(stderr, "Generate bytecode error\n");
835 ret = -LTTNG_ERR_FILTER_INVAL;
836 goto parse_error;
837 }
838 dbg_printf("done\n");
839 dbg_printf("Size of bytecode generated: %u bytes.\n",
840 bytecode_get_len(&ctx->bytecode->b));
841
842 lsm->u.enable.bytecode_len = sizeof(ctx->bytecode->b)
843 + bytecode_get_len(&ctx->bytecode->b);
844 lsm->u.enable.expression_len = strlen(filter_expression) + 1;
845
846 /* No need to keep the memory stream. */
847 if (fclose(fmem) != 0) {
848 perror("fclose");
849 }
850
851 *ctxp = ctx;
852 return 0;
853
854 parse_error:
855 filter_ir_free(ctx);
856 filter_parser_ctx_free(ctx);
857 filter_alloc_error:
858 if (fclose(fmem) != 0) {
859 perror("fclose");
860 }
861 error:
862 return ret;
863 }
864
865 /*
866 * Enable event(s) for a channel, possibly with exclusions and a filter.
867 * If no event name is specified, all events are enabled.
868 * If no channel name is specified, the default name is used.
869 * If filter expression is not NULL, the filter is set for the event.
870 * If exclusion count is not zero, the exclusions are set for the event.
871 * Returns size of returned session payload data or a negative error code.
872 */
873 int lttng_enable_event_with_exclusions(struct lttng_handle *handle,
874 struct lttng_event *ev, const char *channel_name,
875 const char *original_filter_expression,
876 int exclusion_count, char **exclusion_list)
877 {
878 struct lttcomm_session_msg lsm;
879 char *varlen_data;
880 int ret = 0;
881 unsigned int free_filter_expression = 0;
882 struct filter_parser_ctx *ctx = NULL;
883 /*
884 * Cast as non-const since we may replace the filter expression
885 * by a dynamically allocated string. Otherwise, the original
886 * string is not modified.
887 */
888 char *filter_expression = (char *) original_filter_expression;
889
890 if (handle == NULL || ev == NULL) {
891 ret = -LTTNG_ERR_INVALID;
892 goto error;
893 }
894
895 /* Empty filter string will always be rejected by the parser
896 * anyway, so treat this corner-case early to eliminate
897 * lttng_fmemopen error for 0-byte allocation.
898 */
899 if (filter_expression && filter_expression[0] == '\0') {
900 ret = -LTTNG_ERR_INVALID;
901 goto error;
902 }
903
904 memset(&lsm, 0, sizeof(lsm));
905
906 /* If no channel name, send empty string. */
907 if (channel_name == NULL) {
908 lttng_ctl_copy_string(lsm.u.enable.channel_name, "",
909 sizeof(lsm.u.enable.channel_name));
910 } else {
911 lttng_ctl_copy_string(lsm.u.enable.channel_name, channel_name,
912 sizeof(lsm.u.enable.channel_name));
913 }
914
915 if (ev->name[0] != '\0') {
916 lsm.cmd_type = LTTNG_ENABLE_EVENT;
917 } else {
918 lsm.cmd_type = LTTNG_ENABLE_ALL_EVENT;
919 }
920
921 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
922 memcpy(&lsm.u.enable.event, ev, sizeof(lsm.u.enable.event));
923
924 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
925 sizeof(lsm.session.name));
926 lsm.u.enable.exclusion_count = exclusion_count;
927 lsm.u.enable.bytecode_len = 0;
928
929 /*
930 * For the JUL domain, a filter is enforced except for the enable all
931 * event. This is done to avoid having the event in all sessions thus
932 * filtering by logger name.
933 */
934 if (exclusion_count == 0 && filter_expression == NULL &&
935 handle->domain.type != LTTNG_DOMAIN_JUL) {
936 goto ask_sessiond;
937 }
938
939 /*
940 * We have either a filter or some exclusions, so we need to set up
941 * a variable-length memory block from where to send the data
942 */
943
944 /* Parse filter expression */
945 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL) {
946 if (handle->domain.type == LTTNG_DOMAIN_JUL) {
947 char *jul_filter;
948
949 /* Setup JUL filter if needed. */
950 jul_filter = set_jul_filter(filter_expression, ev);
951 if (!jul_filter) {
952 if (!filter_expression) {
953 /* No JUL and no filter, just skip everything below. */
954 goto ask_sessiond;
955 }
956 } else {
957 /*
958 * With a JUL filter, the original filter has been added to it
959 * thus replace the filter expression.
960 */
961 filter_expression = jul_filter;
962 free_filter_expression = 1;
963 }
964 }
965
966 ret = generate_filter(filter_expression, &lsm, &ctx);
967 if (ret) {
968 goto filter_error;
969 }
970 }
971
972 varlen_data = zmalloc(lsm.u.enable.bytecode_len
973 + lsm.u.enable.expression_len
974 + LTTNG_SYMBOL_NAME_LEN * exclusion_count);
975 if (!varlen_data) {
976 ret = -LTTNG_ERR_EXCLUSION_NOMEM;
977 goto mem_error;
978 }
979
980 /* Put exclusion names first in the data */
981 while (exclusion_count--) {
982 strncpy(varlen_data + LTTNG_SYMBOL_NAME_LEN * exclusion_count,
983 *(exclusion_list + exclusion_count), LTTNG_SYMBOL_NAME_LEN);
984 }
985 /* Add filter expression next */
986 if (lsm.u.enable.expression_len != 0) {
987 memcpy(varlen_data
988 + LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count,
989 filter_expression,
990 lsm.u.enable.expression_len);
991 }
992 /* Add filter bytecode next */
993 if (ctx && lsm.u.enable.bytecode_len != 0) {
994 memcpy(varlen_data
995 + LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count
996 + lsm.u.enable.expression_len,
997 &ctx->bytecode->b,
998 lsm.u.enable.bytecode_len);
999 }
1000
1001 ret = lttng_ctl_ask_sessiond_varlen(&lsm, varlen_data,
1002 (LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count) +
1003 lsm.u.enable.bytecode_len + lsm.u.enable.expression_len, NULL);
1004 free(varlen_data);
1005
1006 mem_error:
1007 if (filter_expression && ctx) {
1008 filter_bytecode_free(ctx);
1009 filter_ir_free(ctx);
1010 filter_parser_ctx_free(ctx);
1011 }
1012 filter_error:
1013 if (free_filter_expression) {
1014 /*
1015 * The filter expression has been replaced and must be freed as it is
1016 * not the original filter expression received as a parameter.
1017 */
1018 free(filter_expression);
1019 }
1020 error:
1021 /*
1022 * Return directly to the caller and don't ask the sessiond since something
1023 * went wrong in the parsing of data above.
1024 */
1025 return ret;
1026
1027 ask_sessiond:
1028 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1029 return ret;
1030 }
1031
1032 /*
1033 * Disable event(s) of a channel and domain.
1034 * If no event name is specified, all events are disabled.
1035 * If no channel name is specified, the default 'channel0' is used.
1036 * Returns size of returned session payload data or a negative error code.
1037 */
1038 int lttng_disable_event(struct lttng_handle *handle, const char *name,
1039 const char *channel_name)
1040 {
1041 struct lttcomm_session_msg lsm;
1042
1043 if (handle == NULL) {
1044 return -LTTNG_ERR_INVALID;
1045 }
1046
1047 memset(&lsm, 0, sizeof(lsm));
1048
1049 /* If no channel name, send empty string. */
1050 if (channel_name == NULL) {
1051 lttng_ctl_copy_string(lsm.u.disable.channel_name, "",
1052 sizeof(lsm.u.disable.channel_name));
1053 } else {
1054 lttng_ctl_copy_string(lsm.u.disable.channel_name, channel_name,
1055 sizeof(lsm.u.disable.channel_name));
1056 }
1057
1058 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1059
1060 if (name != NULL) {
1061 lttng_ctl_copy_string(lsm.u.disable.name, name,
1062 sizeof(lsm.u.disable.name));
1063 lsm.cmd_type = LTTNG_DISABLE_EVENT;
1064 } else {
1065 lsm.cmd_type = LTTNG_DISABLE_ALL_EVENT;
1066 }
1067
1068 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1069 sizeof(lsm.session.name));
1070
1071 return lttng_ctl_ask_sessiond(&lsm, NULL);
1072 }
1073
1074 /*
1075 * Enable channel per domain
1076 * Returns size of returned session payload data or a negative error code.
1077 */
1078 int lttng_enable_channel(struct lttng_handle *handle,
1079 struct lttng_channel *chan)
1080 {
1081 struct lttcomm_session_msg lsm;
1082
1083 /*
1084 * NULL arguments are forbidden. No default values.
1085 */
1086 if (handle == NULL || chan == NULL) {
1087 return -LTTNG_ERR_INVALID;
1088 }
1089
1090 memset(&lsm, 0, sizeof(lsm));
1091
1092 memcpy(&lsm.u.channel.chan, chan, sizeof(lsm.u.channel.chan));
1093
1094 lsm.cmd_type = LTTNG_ENABLE_CHANNEL;
1095
1096 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1097
1098 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1099 sizeof(lsm.session.name));
1100
1101 return lttng_ctl_ask_sessiond(&lsm, NULL);
1102 }
1103
1104 /*
1105 * All tracing will be stopped for registered events of the channel.
1106 * Returns size of returned session payload data or a negative error code.
1107 */
1108 int lttng_disable_channel(struct lttng_handle *handle, const char *name)
1109 {
1110 struct lttcomm_session_msg lsm;
1111
1112 /* Safety check. Both are mandatory */
1113 if (handle == NULL || name == NULL) {
1114 return -LTTNG_ERR_INVALID;
1115 }
1116
1117 memset(&lsm, 0, sizeof(lsm));
1118
1119 lsm.cmd_type = LTTNG_DISABLE_CHANNEL;
1120
1121 lttng_ctl_copy_string(lsm.u.disable.channel_name, name,
1122 sizeof(lsm.u.disable.channel_name));
1123
1124 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1125
1126 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1127 sizeof(lsm.session.name));
1128
1129 return lttng_ctl_ask_sessiond(&lsm, NULL);
1130 }
1131
1132 /*
1133 * Lists all available tracepoints of domain.
1134 * Sets the contents of the events array.
1135 * Returns the number of lttng_event entries in events;
1136 * on error, returns a negative value.
1137 */
1138 int lttng_list_tracepoints(struct lttng_handle *handle,
1139 struct lttng_event **events)
1140 {
1141 int ret;
1142 struct lttcomm_session_msg lsm;
1143
1144 if (handle == NULL) {
1145 return -LTTNG_ERR_INVALID;
1146 }
1147
1148 memset(&lsm, 0, sizeof(lsm));
1149 lsm.cmd_type = LTTNG_LIST_TRACEPOINTS;
1150 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1151
1152 ret = lttng_ctl_ask_sessiond(&lsm, (void **) events);
1153 if (ret < 0) {
1154 return ret;
1155 }
1156
1157 return ret / sizeof(struct lttng_event);
1158 }
1159
1160 /*
1161 * Lists all available tracepoint fields of domain.
1162 * Sets the contents of the event field array.
1163 * Returns the number of lttng_event_field entries in events;
1164 * on error, returns a negative value.
1165 */
1166 int lttng_list_tracepoint_fields(struct lttng_handle *handle,
1167 struct lttng_event_field **fields)
1168 {
1169 int ret;
1170 struct lttcomm_session_msg lsm;
1171
1172 if (handle == NULL) {
1173 return -LTTNG_ERR_INVALID;
1174 }
1175
1176 memset(&lsm, 0, sizeof(lsm));
1177 lsm.cmd_type = LTTNG_LIST_TRACEPOINT_FIELDS;
1178 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1179
1180 ret = lttng_ctl_ask_sessiond(&lsm, (void **) fields);
1181 if (ret < 0) {
1182 return ret;
1183 }
1184
1185 return ret / sizeof(struct lttng_event_field);
1186 }
1187
1188 /*
1189 * Returns a human readable string describing
1190 * the error code (a negative value).
1191 */
1192 const char *lttng_strerror(int code)
1193 {
1194 return error_get_str(code);
1195 }
1196
1197 /*
1198 * Create a brand new session using name and url for destination.
1199 *
1200 * Returns LTTNG_OK on success or a negative error code.
1201 */
1202 int lttng_create_session(const char *name, const char *url)
1203 {
1204 int ret;
1205 ssize_t size;
1206 struct lttcomm_session_msg lsm;
1207 struct lttng_uri *uris = NULL;
1208
1209 if (name == NULL) {
1210 return -LTTNG_ERR_INVALID;
1211 }
1212
1213 memset(&lsm, 0, sizeof(lsm));
1214
1215 lsm.cmd_type = LTTNG_CREATE_SESSION;
1216 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1217
1218 /* There should never be a data URL */
1219 size = uri_parse_str_urls(url, NULL, &uris);
1220 if (size < 0) {
1221 return -LTTNG_ERR_INVALID;
1222 }
1223
1224 lsm.u.uri.size = size;
1225
1226 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1227 sizeof(struct lttng_uri) * size, NULL);
1228
1229 free(uris);
1230 return ret;
1231 }
1232
1233 /*
1234 * Destroy session using name.
1235 * Returns size of returned session payload data or a negative error code.
1236 */
1237 int lttng_destroy_session(const char *session_name)
1238 {
1239 struct lttcomm_session_msg lsm;
1240
1241 if (session_name == NULL) {
1242 return -LTTNG_ERR_INVALID;
1243 }
1244
1245 memset(&lsm, 0, sizeof(lsm));
1246 lsm.cmd_type = LTTNG_DESTROY_SESSION;
1247
1248 lttng_ctl_copy_string(lsm.session.name, session_name,
1249 sizeof(lsm.session.name));
1250
1251 return lttng_ctl_ask_sessiond(&lsm, NULL);
1252 }
1253
1254 /*
1255 * Ask the session daemon for all available sessions.
1256 * Sets the contents of the sessions array.
1257 * Returns the number of lttng_session entries in sessions;
1258 * on error, returns a negative value.
1259 */
1260 int lttng_list_sessions(struct lttng_session **sessions)
1261 {
1262 int ret;
1263 struct lttcomm_session_msg lsm;
1264
1265 memset(&lsm, 0, sizeof(lsm));
1266 lsm.cmd_type = LTTNG_LIST_SESSIONS;
1267 ret = lttng_ctl_ask_sessiond(&lsm, (void**) sessions);
1268 if (ret < 0) {
1269 return ret;
1270 }
1271
1272 return ret / sizeof(struct lttng_session);
1273 }
1274
1275 /*
1276 * Ask the session daemon for all available domains of a session.
1277 * Sets the contents of the domains array.
1278 * Returns the number of lttng_domain entries in domains;
1279 * on error, returns a negative value.
1280 */
1281 int lttng_list_domains(const char *session_name,
1282 struct lttng_domain **domains)
1283 {
1284 int ret;
1285 struct lttcomm_session_msg lsm;
1286
1287 if (session_name == NULL) {
1288 return -LTTNG_ERR_INVALID;
1289 }
1290
1291 memset(&lsm, 0, sizeof(lsm));
1292 lsm.cmd_type = LTTNG_LIST_DOMAINS;
1293
1294 lttng_ctl_copy_string(lsm.session.name, session_name,
1295 sizeof(lsm.session.name));
1296
1297 ret = lttng_ctl_ask_sessiond(&lsm, (void**) domains);
1298 if (ret < 0) {
1299 return ret;
1300 }
1301
1302 return ret / sizeof(struct lttng_domain);
1303 }
1304
1305 /*
1306 * Ask the session daemon for all available channels of a session.
1307 * Sets the contents of the channels array.
1308 * Returns the number of lttng_channel entries in channels;
1309 * on error, returns a negative value.
1310 */
1311 int lttng_list_channels(struct lttng_handle *handle,
1312 struct lttng_channel **channels)
1313 {
1314 int ret;
1315 struct lttcomm_session_msg lsm;
1316
1317 if (handle == NULL) {
1318 return -LTTNG_ERR_INVALID;
1319 }
1320
1321 memset(&lsm, 0, sizeof(lsm));
1322 lsm.cmd_type = LTTNG_LIST_CHANNELS;
1323 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1324 sizeof(lsm.session.name));
1325
1326 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1327
1328 ret = lttng_ctl_ask_sessiond(&lsm, (void**) channels);
1329 if (ret < 0) {
1330 return ret;
1331 }
1332
1333 return ret / sizeof(struct lttng_channel);
1334 }
1335
1336 /*
1337 * Ask the session daemon for all available events of a session channel.
1338 * Sets the contents of the events array.
1339 * Returns the number of lttng_event entries in events;
1340 * on error, returns a negative value.
1341 */
1342 int lttng_list_events(struct lttng_handle *handle,
1343 const char *channel_name, struct lttng_event **events)
1344 {
1345 int ret;
1346 struct lttcomm_session_msg lsm;
1347
1348 /* Safety check. An handle and channel name are mandatory */
1349 if (handle == NULL || channel_name == NULL) {
1350 return -LTTNG_ERR_INVALID;
1351 }
1352
1353 memset(&lsm, 0, sizeof(lsm));
1354 lsm.cmd_type = LTTNG_LIST_EVENTS;
1355 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1356 sizeof(lsm.session.name));
1357 lttng_ctl_copy_string(lsm.u.list.channel_name, channel_name,
1358 sizeof(lsm.u.list.channel_name));
1359
1360 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1361
1362 ret = lttng_ctl_ask_sessiond(&lsm, (void**) events);
1363 if (ret < 0) {
1364 return ret;
1365 }
1366
1367 return ret / sizeof(struct lttng_event);
1368 }
1369
1370 /*
1371 * Sets the tracing_group variable with name.
1372 * This function allocates memory pointed to by tracing_group.
1373 * On success, returns 0, on error, returns -1 (null name) or -ENOMEM.
1374 */
1375 int lttng_set_tracing_group(const char *name)
1376 {
1377 if (name == NULL) {
1378 return -LTTNG_ERR_INVALID;
1379 }
1380
1381 if (asprintf(&tracing_group, "%s", name) < 0) {
1382 return -LTTNG_ERR_FATAL;
1383 }
1384
1385 return 0;
1386 }
1387
1388 /*
1389 * Returns size of returned session payload data or a negative error code.
1390 */
1391 int lttng_calibrate(struct lttng_handle *handle,
1392 struct lttng_calibrate *calibrate)
1393 {
1394 struct lttcomm_session_msg lsm;
1395
1396 /* Safety check. NULL pointer are forbidden */
1397 if (handle == NULL || calibrate == NULL) {
1398 return -LTTNG_ERR_INVALID;
1399 }
1400
1401 memset(&lsm, 0, sizeof(lsm));
1402 lsm.cmd_type = LTTNG_CALIBRATE;
1403 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1404
1405 memcpy(&lsm.u.calibrate, calibrate, sizeof(lsm.u.calibrate));
1406
1407 return lttng_ctl_ask_sessiond(&lsm, NULL);
1408 }
1409
1410 /*
1411 * Set default channel attributes.
1412 * If either or both of the arguments are null, attr content is zeroe'd.
1413 */
1414 void lttng_channel_set_default_attr(struct lttng_domain *domain,
1415 struct lttng_channel_attr *attr)
1416 {
1417 /* Safety check */
1418 if (attr == NULL || domain == NULL) {
1419 return;
1420 }
1421
1422 memset(attr, 0, sizeof(struct lttng_channel_attr));
1423
1424 /* Same for all domains. */
1425 attr->overwrite = DEFAULT_CHANNEL_OVERWRITE;
1426 attr->tracefile_size = DEFAULT_CHANNEL_TRACEFILE_SIZE;
1427 attr->tracefile_count = DEFAULT_CHANNEL_TRACEFILE_COUNT;
1428
1429 switch (domain->type) {
1430 case LTTNG_DOMAIN_KERNEL:
1431 attr->switch_timer_interval = DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER;
1432 attr->read_timer_interval = DEFAULT_KERNEL_CHANNEL_READ_TIMER;
1433 attr->subbuf_size = default_get_kernel_channel_subbuf_size();
1434 attr->num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
1435 attr->output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
1436 break;
1437 case LTTNG_DOMAIN_UST:
1438 switch (domain->buf_type) {
1439 case LTTNG_BUFFER_PER_UID:
1440 attr->subbuf_size = default_get_ust_uid_channel_subbuf_size();
1441 attr->num_subbuf = DEFAULT_UST_UID_CHANNEL_SUBBUF_NUM;
1442 attr->output = DEFAULT_UST_UID_CHANNEL_OUTPUT;
1443 attr->switch_timer_interval = DEFAULT_UST_UID_CHANNEL_SWITCH_TIMER;
1444 attr->read_timer_interval = DEFAULT_UST_UID_CHANNEL_READ_TIMER;
1445 break;
1446 case LTTNG_BUFFER_PER_PID:
1447 default:
1448 attr->subbuf_size = default_get_ust_pid_channel_subbuf_size();
1449 attr->num_subbuf = DEFAULT_UST_PID_CHANNEL_SUBBUF_NUM;
1450 attr->output = DEFAULT_UST_PID_CHANNEL_OUTPUT;
1451 attr->switch_timer_interval = DEFAULT_UST_PID_CHANNEL_SWITCH_TIMER;
1452 attr->read_timer_interval = DEFAULT_UST_PID_CHANNEL_READ_TIMER;
1453 break;
1454 }
1455 default:
1456 /* Default behavior: leave set to 0. */
1457 break;
1458 }
1459 }
1460
1461 /*
1462 * Check if session daemon is alive.
1463 *
1464 * Return 1 if alive or 0 if not.
1465 * On error returns a negative value.
1466 */
1467 int lttng_session_daemon_alive(void)
1468 {
1469 int ret;
1470
1471 ret = set_session_daemon_path();
1472 if (ret < 0) {
1473 /* Error */
1474 return ret;
1475 }
1476
1477 if (*sessiond_sock_path == '\0') {
1478 /*
1479 * No socket path set. Weird error which means the constructor was not
1480 * called.
1481 */
1482 assert(0);
1483 }
1484
1485 ret = try_connect_sessiond(sessiond_sock_path);
1486 if (ret < 0) {
1487 /* Not alive */
1488 return 0;
1489 }
1490
1491 /* Is alive */
1492 return 1;
1493 }
1494
1495 /*
1496 * Set URL for a consumer for a session and domain.
1497 *
1498 * Return 0 on success, else a negative value.
1499 */
1500 int lttng_set_consumer_url(struct lttng_handle *handle,
1501 const char *control_url, const char *data_url)
1502 {
1503 int ret;
1504 ssize_t size;
1505 struct lttcomm_session_msg lsm;
1506 struct lttng_uri *uris = NULL;
1507
1508 if (handle == NULL || (control_url == NULL && data_url == NULL)) {
1509 return -LTTNG_ERR_INVALID;
1510 }
1511
1512 memset(&lsm, 0, sizeof(lsm));
1513
1514 lsm.cmd_type = LTTNG_SET_CONSUMER_URI;
1515
1516 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1517 sizeof(lsm.session.name));
1518 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1519
1520 size = uri_parse_str_urls(control_url, data_url, &uris);
1521 if (size < 0) {
1522 return -LTTNG_ERR_INVALID;
1523 }
1524
1525 lsm.u.uri.size = size;
1526
1527 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1528 sizeof(struct lttng_uri) * size, NULL);
1529
1530 free(uris);
1531 return ret;
1532 }
1533
1534 /*
1535 * [OBSOLETE]
1536 */
1537 int lttng_enable_consumer(struct lttng_handle *handle)
1538 {
1539 return -ENOSYS;
1540 }
1541
1542 /*
1543 * [OBSOLETE]
1544 */
1545 int lttng_disable_consumer(struct lttng_handle *handle)
1546 {
1547 return -ENOSYS;
1548 }
1549
1550 /*
1551 * This is an extension of create session that is ONLY and SHOULD only be used
1552 * by the lttng command line program. It exists to avoid using URI parsing in
1553 * the lttng client.
1554 *
1555 * We need the date and time for the trace path subdirectory for the case where
1556 * the user does NOT define one using either -o or -U. Using the normal
1557 * lttng_create_session API call, we have no clue on the session daemon side if
1558 * the URL was generated automatically by the client or define by the user.
1559 *
1560 * So this function "wrapper" is hidden from the public API, takes the datetime
1561 * string and appends it if necessary to the URI subdirectory before sending it
1562 * to the session daemon.
1563 *
1564 * With this extra function, the lttng_create_session call behavior is not
1565 * changed and the timestamp is appended to the URI on the session daemon side
1566 * if necessary.
1567 */
1568 int _lttng_create_session_ext(const char *name, const char *url,
1569 const char *datetime)
1570 {
1571 int ret;
1572 ssize_t size;
1573 struct lttcomm_session_msg lsm;
1574 struct lttng_uri *uris = NULL;
1575
1576 if (name == NULL || datetime == NULL) {
1577 return -LTTNG_ERR_INVALID;
1578 }
1579
1580 memset(&lsm, 0, sizeof(lsm));
1581
1582 lsm.cmd_type = LTTNG_CREATE_SESSION;
1583 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1584
1585 /* There should never be a data URL */
1586 size = uri_parse_str_urls(url, NULL, &uris);
1587 if (size < 0) {
1588 ret = -LTTNG_ERR_INVALID;
1589 goto error;
1590 }
1591
1592 lsm.u.uri.size = size;
1593
1594 if (size > 0 && uris[0].dtype != LTTNG_DST_PATH && strlen(uris[0].subdir) == 0) {
1595 /* Don't append datetime if the name was automatically created. */
1596 if (strncmp(name, DEFAULT_SESSION_NAME "-",
1597 strlen(DEFAULT_SESSION_NAME) + 1)) {
1598 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s-%s",
1599 name, datetime);
1600 } else {
1601 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s", name);
1602 }
1603 if (ret < 0) {
1604 PERROR("snprintf uri subdir");
1605 ret = -LTTNG_ERR_FATAL;
1606 goto error;
1607 }
1608 }
1609
1610 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1611 sizeof(struct lttng_uri) * size, NULL);
1612
1613 error:
1614 free(uris);
1615 return ret;
1616 }
1617
1618 /*
1619 * For a given session name, this call checks if the data is ready to be read
1620 * or is still being extracted by the consumer(s) hence not ready to be used by
1621 * any readers.
1622 */
1623 int lttng_data_pending(const char *session_name)
1624 {
1625 int ret;
1626 struct lttcomm_session_msg lsm;
1627
1628 if (session_name == NULL) {
1629 return -LTTNG_ERR_INVALID;
1630 }
1631
1632 memset(&lsm, 0, sizeof(lsm));
1633 lsm.cmd_type = LTTNG_DATA_PENDING;
1634
1635 lttng_ctl_copy_string(lsm.session.name, session_name,
1636 sizeof(lsm.session.name));
1637
1638 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1639
1640 /*
1641 * The lttng_ctl_ask_sessiond function negate the return code if it's not
1642 * LTTNG_OK so getting -1 means that the reply ret_code was 1 thus meaning
1643 * that the data is available. Yes it is hackish but for now this is the
1644 * only way.
1645 */
1646 if (ret == -1) {
1647 ret = 1;
1648 }
1649
1650 return ret;
1651 }
1652
1653 /*
1654 * Create a session exclusively used for snapshot.
1655 *
1656 * Returns LTTNG_OK on success or a negative error code.
1657 */
1658 int lttng_create_session_snapshot(const char *name, const char *snapshot_url)
1659 {
1660 int ret;
1661 ssize_t size;
1662 struct lttcomm_session_msg lsm;
1663 struct lttng_uri *uris = NULL;
1664
1665 if (name == NULL) {
1666 return -LTTNG_ERR_INVALID;
1667 }
1668
1669 memset(&lsm, 0, sizeof(lsm));
1670
1671 lsm.cmd_type = LTTNG_CREATE_SESSION_SNAPSHOT;
1672 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1673
1674 size = uri_parse_str_urls(snapshot_url, NULL, &uris);
1675 if (size < 0) {
1676 return -LTTNG_ERR_INVALID;
1677 }
1678
1679 lsm.u.uri.size = size;
1680
1681 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1682 sizeof(struct lttng_uri) * size, NULL);
1683
1684 free(uris);
1685 return ret;
1686 }
1687
1688 /*
1689 * Create a session exclusively used for live.
1690 *
1691 * Returns LTTNG_OK on success or a negative error code.
1692 */
1693 int lttng_create_session_live(const char *name, const char *url,
1694 unsigned int timer_interval)
1695 {
1696 int ret;
1697 ssize_t size;
1698 struct lttcomm_session_msg lsm;
1699 struct lttng_uri *uris = NULL;
1700
1701 if (name == NULL) {
1702 return -LTTNG_ERR_INVALID;
1703 }
1704
1705 memset(&lsm, 0, sizeof(lsm));
1706
1707 lsm.cmd_type = LTTNG_CREATE_SESSION_LIVE;
1708 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1709
1710 if (url) {
1711 size = uri_parse_str_urls(url, NULL, &uris);
1712 if (size <= 0) {
1713 ret = -LTTNG_ERR_INVALID;
1714 goto end;
1715 }
1716
1717 /* file:// is not accepted for live session. */
1718 if (uris[0].dtype == LTTNG_DST_PATH) {
1719 ret = -LTTNG_ERR_INVALID;
1720 goto end;
1721 }
1722 } else {
1723 size = 0;
1724 }
1725
1726 lsm.u.session_live.nb_uri = size;
1727 lsm.u.session_live.timer_interval = timer_interval;
1728
1729 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1730 sizeof(struct lttng_uri) * size, NULL);
1731
1732 end:
1733 free(uris);
1734 return ret;
1735 }
1736
1737 /*
1738 * lib constructor
1739 */
1740 static void __attribute__((constructor)) init()
1741 {
1742 /* Set default session group */
1743 lttng_set_tracing_group(DEFAULT_TRACING_GROUP);
1744 }
1745
1746 /*
1747 * lib destructor
1748 */
1749 static void __attribute__((destructor)) lttng_ctl_exit()
1750 {
1751 free(tracing_group);
1752 }
This page took 0.112076 seconds and 4 git commands to generate.