b9dc1efb095de163ff3d4ca50bfba28ed6596378
[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 /* Validate strings used as literals in the expression */
829 ret = filter_visitor_ir_validate_string(ctx);
830 if (ret) {
831 ret = -LTTNG_ERR_FILTER_INVAL;
832 goto parse_error;
833 }
834 dbg_printf("done\n");
835
836 dbg_printf("Generating bytecode... ");
837 fflush(stdout);
838 ret = filter_visitor_bytecode_generate(ctx);
839 if (ret) {
840 fprintf(stderr, "Generate bytecode error\n");
841 ret = -LTTNG_ERR_FILTER_INVAL;
842 goto parse_error;
843 }
844 dbg_printf("done\n");
845 dbg_printf("Size of bytecode generated: %u bytes.\n",
846 bytecode_get_len(&ctx->bytecode->b));
847
848 lsm->u.enable.bytecode_len = sizeof(ctx->bytecode->b)
849 + bytecode_get_len(&ctx->bytecode->b);
850 lsm->u.enable.expression_len = strlen(filter_expression) + 1;
851
852 /* No need to keep the memory stream. */
853 if (fclose(fmem) != 0) {
854 perror("fclose");
855 }
856
857 *ctxp = ctx;
858 return 0;
859
860 parse_error:
861 filter_ir_free(ctx);
862 filter_parser_ctx_free(ctx);
863 filter_alloc_error:
864 if (fclose(fmem) != 0) {
865 perror("fclose");
866 }
867 error:
868 return ret;
869 }
870
871 /*
872 * Enable event(s) for a channel, possibly with exclusions and a filter.
873 * If no event name is specified, all events are enabled.
874 * If no channel name is specified, the default name is used.
875 * If filter expression is not NULL, the filter is set for the event.
876 * If exclusion count is not zero, the exclusions are set for the event.
877 * Returns size of returned session payload data or a negative error code.
878 */
879 int lttng_enable_event_with_exclusions(struct lttng_handle *handle,
880 struct lttng_event *ev, const char *channel_name,
881 const char *original_filter_expression,
882 int exclusion_count, char **exclusion_list)
883 {
884 struct lttcomm_session_msg lsm;
885 char *varlen_data;
886 int ret = 0;
887 unsigned int free_filter_expression = 0;
888 struct filter_parser_ctx *ctx = NULL;
889 /*
890 * Cast as non-const since we may replace the filter expression
891 * by a dynamically allocated string. Otherwise, the original
892 * string is not modified.
893 */
894 char *filter_expression = (char *) original_filter_expression;
895
896 if (handle == NULL || ev == NULL) {
897 ret = -LTTNG_ERR_INVALID;
898 goto error;
899 }
900
901 /* Empty filter string will always be rejected by the parser
902 * anyway, so treat this corner-case early to eliminate
903 * lttng_fmemopen error for 0-byte allocation.
904 */
905 if (filter_expression && filter_expression[0] == '\0') {
906 ret = -LTTNG_ERR_INVALID;
907 goto error;
908 }
909
910 memset(&lsm, 0, sizeof(lsm));
911
912 /* If no channel name, send empty string. */
913 if (channel_name == NULL) {
914 lttng_ctl_copy_string(lsm.u.enable.channel_name, "",
915 sizeof(lsm.u.enable.channel_name));
916 } else {
917 lttng_ctl_copy_string(lsm.u.enable.channel_name, channel_name,
918 sizeof(lsm.u.enable.channel_name));
919 }
920
921 if (ev->name[0] != '\0') {
922 lsm.cmd_type = LTTNG_ENABLE_EVENT;
923 } else {
924 lsm.cmd_type = LTTNG_ENABLE_ALL_EVENT;
925 }
926
927 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
928 memcpy(&lsm.u.enable.event, ev, sizeof(lsm.u.enable.event));
929
930 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
931 sizeof(lsm.session.name));
932 lsm.u.enable.exclusion_count = exclusion_count;
933 lsm.u.enable.bytecode_len = 0;
934
935 /*
936 * For the JUL domain, a filter is enforced except for the enable all
937 * event. This is done to avoid having the event in all sessions thus
938 * filtering by logger name.
939 */
940 if (exclusion_count == 0 && filter_expression == NULL &&
941 handle->domain.type != LTTNG_DOMAIN_JUL) {
942 goto ask_sessiond;
943 }
944
945 /*
946 * We have either a filter or some exclusions, so we need to set up
947 * a variable-length memory block from where to send the data
948 */
949
950 /* Parse filter expression */
951 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL) {
952 if (handle->domain.type == LTTNG_DOMAIN_JUL) {
953 char *jul_filter;
954
955 /* Setup JUL filter if needed. */
956 jul_filter = set_jul_filter(filter_expression, ev);
957 if (!jul_filter) {
958 if (!filter_expression) {
959 /* No JUL and no filter, just skip everything below. */
960 goto ask_sessiond;
961 }
962 } else {
963 /*
964 * With a JUL filter, the original filter has been added to it
965 * thus replace the filter expression.
966 */
967 filter_expression = jul_filter;
968 free_filter_expression = 1;
969 }
970 }
971
972 ret = generate_filter(filter_expression, &lsm, &ctx);
973 if (ret) {
974 goto filter_error;
975 }
976 }
977
978 varlen_data = zmalloc(lsm.u.enable.bytecode_len
979 + lsm.u.enable.expression_len
980 + LTTNG_SYMBOL_NAME_LEN * exclusion_count);
981 if (!varlen_data) {
982 ret = -LTTNG_ERR_EXCLUSION_NOMEM;
983 goto mem_error;
984 }
985
986 /* Put exclusion names first in the data */
987 while (exclusion_count--) {
988 strncpy(varlen_data + LTTNG_SYMBOL_NAME_LEN * exclusion_count,
989 *(exclusion_list + exclusion_count), LTTNG_SYMBOL_NAME_LEN);
990 }
991 /* Add filter expression next */
992 if (lsm.u.enable.expression_len != 0) {
993 memcpy(varlen_data
994 + LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count,
995 filter_expression,
996 lsm.u.enable.expression_len);
997 }
998 /* Add filter bytecode next */
999 if (ctx && lsm.u.enable.bytecode_len != 0) {
1000 memcpy(varlen_data
1001 + LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count
1002 + lsm.u.enable.expression_len,
1003 &ctx->bytecode->b,
1004 lsm.u.enable.bytecode_len);
1005 }
1006
1007 ret = lttng_ctl_ask_sessiond_varlen(&lsm, varlen_data,
1008 (LTTNG_SYMBOL_NAME_LEN * lsm.u.enable.exclusion_count) +
1009 lsm.u.enable.bytecode_len + lsm.u.enable.expression_len, NULL);
1010 free(varlen_data);
1011
1012 mem_error:
1013 if (filter_expression && ctx) {
1014 filter_bytecode_free(ctx);
1015 filter_ir_free(ctx);
1016 filter_parser_ctx_free(ctx);
1017 }
1018 filter_error:
1019 if (free_filter_expression) {
1020 /*
1021 * The filter expression has been replaced and must be freed as it is
1022 * not the original filter expression received as a parameter.
1023 */
1024 free(filter_expression);
1025 }
1026 error:
1027 /*
1028 * Return directly to the caller and don't ask the sessiond since something
1029 * went wrong in the parsing of data above.
1030 */
1031 return ret;
1032
1033 ask_sessiond:
1034 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1035 return ret;
1036 }
1037
1038 /*
1039 * Disable event(s) of a channel and domain.
1040 * If no event name is specified, all events are disabled.
1041 * If no channel name is specified, the default 'channel0' is used.
1042 * Returns size of returned session payload data or a negative error code.
1043 */
1044 int lttng_disable_event(struct lttng_handle *handle, const char *name,
1045 const char *channel_name)
1046 {
1047 struct lttcomm_session_msg lsm;
1048
1049 if (handle == NULL) {
1050 return -LTTNG_ERR_INVALID;
1051 }
1052
1053 memset(&lsm, 0, sizeof(lsm));
1054
1055 /* If no channel name, send empty string. */
1056 if (channel_name == NULL) {
1057 lttng_ctl_copy_string(lsm.u.disable.channel_name, "",
1058 sizeof(lsm.u.disable.channel_name));
1059 } else {
1060 lttng_ctl_copy_string(lsm.u.disable.channel_name, channel_name,
1061 sizeof(lsm.u.disable.channel_name));
1062 }
1063
1064 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1065
1066 if (name != NULL) {
1067 lttng_ctl_copy_string(lsm.u.disable.name, name,
1068 sizeof(lsm.u.disable.name));
1069 lsm.cmd_type = LTTNG_DISABLE_EVENT;
1070 } else {
1071 lsm.cmd_type = LTTNG_DISABLE_ALL_EVENT;
1072 }
1073
1074 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1075 sizeof(lsm.session.name));
1076
1077 return lttng_ctl_ask_sessiond(&lsm, NULL);
1078 }
1079
1080 /*
1081 * Enable channel per domain
1082 * Returns size of returned session payload data or a negative error code.
1083 */
1084 int lttng_enable_channel(struct lttng_handle *handle,
1085 struct lttng_channel *chan)
1086 {
1087 struct lttcomm_session_msg lsm;
1088
1089 /*
1090 * NULL arguments are forbidden. No default values.
1091 */
1092 if (handle == NULL || chan == NULL) {
1093 return -LTTNG_ERR_INVALID;
1094 }
1095
1096 memset(&lsm, 0, sizeof(lsm));
1097
1098 memcpy(&lsm.u.channel.chan, chan, sizeof(lsm.u.channel.chan));
1099
1100 lsm.cmd_type = LTTNG_ENABLE_CHANNEL;
1101
1102 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1103
1104 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1105 sizeof(lsm.session.name));
1106
1107 return lttng_ctl_ask_sessiond(&lsm, NULL);
1108 }
1109
1110 /*
1111 * All tracing will be stopped for registered events of the channel.
1112 * Returns size of returned session payload data or a negative error code.
1113 */
1114 int lttng_disable_channel(struct lttng_handle *handle, const char *name)
1115 {
1116 struct lttcomm_session_msg lsm;
1117
1118 /* Safety check. Both are mandatory */
1119 if (handle == NULL || name == NULL) {
1120 return -LTTNG_ERR_INVALID;
1121 }
1122
1123 memset(&lsm, 0, sizeof(lsm));
1124
1125 lsm.cmd_type = LTTNG_DISABLE_CHANNEL;
1126
1127 lttng_ctl_copy_string(lsm.u.disable.channel_name, name,
1128 sizeof(lsm.u.disable.channel_name));
1129
1130 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1131
1132 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1133 sizeof(lsm.session.name));
1134
1135 return lttng_ctl_ask_sessiond(&lsm, NULL);
1136 }
1137
1138 /*
1139 * Lists all available tracepoints of domain.
1140 * Sets the contents of the events array.
1141 * Returns the number of lttng_event entries in events;
1142 * on error, returns a negative value.
1143 */
1144 int lttng_list_tracepoints(struct lttng_handle *handle,
1145 struct lttng_event **events)
1146 {
1147 int ret;
1148 struct lttcomm_session_msg lsm;
1149
1150 if (handle == NULL) {
1151 return -LTTNG_ERR_INVALID;
1152 }
1153
1154 memset(&lsm, 0, sizeof(lsm));
1155 lsm.cmd_type = LTTNG_LIST_TRACEPOINTS;
1156 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1157
1158 ret = lttng_ctl_ask_sessiond(&lsm, (void **) events);
1159 if (ret < 0) {
1160 return ret;
1161 }
1162
1163 return ret / sizeof(struct lttng_event);
1164 }
1165
1166 /*
1167 * Lists all available tracepoint fields of domain.
1168 * Sets the contents of the event field array.
1169 * Returns the number of lttng_event_field entries in events;
1170 * on error, returns a negative value.
1171 */
1172 int lttng_list_tracepoint_fields(struct lttng_handle *handle,
1173 struct lttng_event_field **fields)
1174 {
1175 int ret;
1176 struct lttcomm_session_msg lsm;
1177
1178 if (handle == NULL) {
1179 return -LTTNG_ERR_INVALID;
1180 }
1181
1182 memset(&lsm, 0, sizeof(lsm));
1183 lsm.cmd_type = LTTNG_LIST_TRACEPOINT_FIELDS;
1184 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1185
1186 ret = lttng_ctl_ask_sessiond(&lsm, (void **) fields);
1187 if (ret < 0) {
1188 return ret;
1189 }
1190
1191 return ret / sizeof(struct lttng_event_field);
1192 }
1193
1194 /*
1195 * Returns a human readable string describing
1196 * the error code (a negative value).
1197 */
1198 const char *lttng_strerror(int code)
1199 {
1200 return error_get_str(code);
1201 }
1202
1203 /*
1204 * Create a brand new session using name and url for destination.
1205 *
1206 * Returns LTTNG_OK on success or a negative error code.
1207 */
1208 int lttng_create_session(const char *name, const char *url)
1209 {
1210 int ret;
1211 ssize_t size;
1212 struct lttcomm_session_msg lsm;
1213 struct lttng_uri *uris = NULL;
1214
1215 if (name == NULL) {
1216 return -LTTNG_ERR_INVALID;
1217 }
1218
1219 memset(&lsm, 0, sizeof(lsm));
1220
1221 lsm.cmd_type = LTTNG_CREATE_SESSION;
1222 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1223
1224 /* There should never be a data URL */
1225 size = uri_parse_str_urls(url, NULL, &uris);
1226 if (size < 0) {
1227 return -LTTNG_ERR_INVALID;
1228 }
1229
1230 lsm.u.uri.size = size;
1231
1232 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1233 sizeof(struct lttng_uri) * size, NULL);
1234
1235 free(uris);
1236 return ret;
1237 }
1238
1239 /*
1240 * Destroy session using name.
1241 * Returns size of returned session payload data or a negative error code.
1242 */
1243 int lttng_destroy_session(const char *session_name)
1244 {
1245 struct lttcomm_session_msg lsm;
1246
1247 if (session_name == NULL) {
1248 return -LTTNG_ERR_INVALID;
1249 }
1250
1251 memset(&lsm, 0, sizeof(lsm));
1252 lsm.cmd_type = LTTNG_DESTROY_SESSION;
1253
1254 lttng_ctl_copy_string(lsm.session.name, session_name,
1255 sizeof(lsm.session.name));
1256
1257 return lttng_ctl_ask_sessiond(&lsm, NULL);
1258 }
1259
1260 /*
1261 * Ask the session daemon for all available sessions.
1262 * Sets the contents of the sessions array.
1263 * Returns the number of lttng_session entries in sessions;
1264 * on error, returns a negative value.
1265 */
1266 int lttng_list_sessions(struct lttng_session **sessions)
1267 {
1268 int ret;
1269 struct lttcomm_session_msg lsm;
1270
1271 memset(&lsm, 0, sizeof(lsm));
1272 lsm.cmd_type = LTTNG_LIST_SESSIONS;
1273 ret = lttng_ctl_ask_sessiond(&lsm, (void**) sessions);
1274 if (ret < 0) {
1275 return ret;
1276 }
1277
1278 return ret / sizeof(struct lttng_session);
1279 }
1280
1281 /*
1282 * Ask the session daemon for all available domains of a session.
1283 * Sets the contents of the domains array.
1284 * Returns the number of lttng_domain entries in domains;
1285 * on error, returns a negative value.
1286 */
1287 int lttng_list_domains(const char *session_name,
1288 struct lttng_domain **domains)
1289 {
1290 int ret;
1291 struct lttcomm_session_msg lsm;
1292
1293 if (session_name == NULL) {
1294 return -LTTNG_ERR_INVALID;
1295 }
1296
1297 memset(&lsm, 0, sizeof(lsm));
1298 lsm.cmd_type = LTTNG_LIST_DOMAINS;
1299
1300 lttng_ctl_copy_string(lsm.session.name, session_name,
1301 sizeof(lsm.session.name));
1302
1303 ret = lttng_ctl_ask_sessiond(&lsm, (void**) domains);
1304 if (ret < 0) {
1305 return ret;
1306 }
1307
1308 return ret / sizeof(struct lttng_domain);
1309 }
1310
1311 /*
1312 * Ask the session daemon for all available channels of a session.
1313 * Sets the contents of the channels array.
1314 * Returns the number of lttng_channel entries in channels;
1315 * on error, returns a negative value.
1316 */
1317 int lttng_list_channels(struct lttng_handle *handle,
1318 struct lttng_channel **channels)
1319 {
1320 int ret;
1321 struct lttcomm_session_msg lsm;
1322
1323 if (handle == NULL) {
1324 return -LTTNG_ERR_INVALID;
1325 }
1326
1327 memset(&lsm, 0, sizeof(lsm));
1328 lsm.cmd_type = LTTNG_LIST_CHANNELS;
1329 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1330 sizeof(lsm.session.name));
1331
1332 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1333
1334 ret = lttng_ctl_ask_sessiond(&lsm, (void**) channels);
1335 if (ret < 0) {
1336 return ret;
1337 }
1338
1339 return ret / sizeof(struct lttng_channel);
1340 }
1341
1342 /*
1343 * Ask the session daemon for all available events of a session channel.
1344 * Sets the contents of the events array.
1345 * Returns the number of lttng_event entries in events;
1346 * on error, returns a negative value.
1347 */
1348 int lttng_list_events(struct lttng_handle *handle,
1349 const char *channel_name, struct lttng_event **events)
1350 {
1351 int ret;
1352 struct lttcomm_session_msg lsm;
1353
1354 /* Safety check. An handle and channel name are mandatory */
1355 if (handle == NULL || channel_name == NULL) {
1356 return -LTTNG_ERR_INVALID;
1357 }
1358
1359 memset(&lsm, 0, sizeof(lsm));
1360 lsm.cmd_type = LTTNG_LIST_EVENTS;
1361 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1362 sizeof(lsm.session.name));
1363 lttng_ctl_copy_string(lsm.u.list.channel_name, channel_name,
1364 sizeof(lsm.u.list.channel_name));
1365
1366 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1367
1368 ret = lttng_ctl_ask_sessiond(&lsm, (void**) events);
1369 if (ret < 0) {
1370 return ret;
1371 }
1372
1373 return ret / sizeof(struct lttng_event);
1374 }
1375
1376 /*
1377 * Sets the tracing_group variable with name.
1378 * This function allocates memory pointed to by tracing_group.
1379 * On success, returns 0, on error, returns -1 (null name) or -ENOMEM.
1380 */
1381 int lttng_set_tracing_group(const char *name)
1382 {
1383 if (name == NULL) {
1384 return -LTTNG_ERR_INVALID;
1385 }
1386
1387 if (asprintf(&tracing_group, "%s", name) < 0) {
1388 return -LTTNG_ERR_FATAL;
1389 }
1390
1391 return 0;
1392 }
1393
1394 /*
1395 * Returns size of returned session payload data or a negative error code.
1396 */
1397 int lttng_calibrate(struct lttng_handle *handle,
1398 struct lttng_calibrate *calibrate)
1399 {
1400 struct lttcomm_session_msg lsm;
1401
1402 /* Safety check. NULL pointer are forbidden */
1403 if (handle == NULL || calibrate == NULL) {
1404 return -LTTNG_ERR_INVALID;
1405 }
1406
1407 memset(&lsm, 0, sizeof(lsm));
1408 lsm.cmd_type = LTTNG_CALIBRATE;
1409 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1410
1411 memcpy(&lsm.u.calibrate, calibrate, sizeof(lsm.u.calibrate));
1412
1413 return lttng_ctl_ask_sessiond(&lsm, NULL);
1414 }
1415
1416 /*
1417 * Set default channel attributes.
1418 * If either or both of the arguments are null, attr content is zeroe'd.
1419 */
1420 void lttng_channel_set_default_attr(struct lttng_domain *domain,
1421 struct lttng_channel_attr *attr)
1422 {
1423 /* Safety check */
1424 if (attr == NULL || domain == NULL) {
1425 return;
1426 }
1427
1428 memset(attr, 0, sizeof(struct lttng_channel_attr));
1429
1430 /* Same for all domains. */
1431 attr->overwrite = DEFAULT_CHANNEL_OVERWRITE;
1432 attr->tracefile_size = DEFAULT_CHANNEL_TRACEFILE_SIZE;
1433 attr->tracefile_count = DEFAULT_CHANNEL_TRACEFILE_COUNT;
1434
1435 switch (domain->type) {
1436 case LTTNG_DOMAIN_KERNEL:
1437 attr->switch_timer_interval = DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER;
1438 attr->read_timer_interval = DEFAULT_KERNEL_CHANNEL_READ_TIMER;
1439 attr->subbuf_size = default_get_kernel_channel_subbuf_size();
1440 attr->num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
1441 attr->output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
1442 break;
1443 case LTTNG_DOMAIN_UST:
1444 switch (domain->buf_type) {
1445 case LTTNG_BUFFER_PER_UID:
1446 attr->subbuf_size = default_get_ust_uid_channel_subbuf_size();
1447 attr->num_subbuf = DEFAULT_UST_UID_CHANNEL_SUBBUF_NUM;
1448 attr->output = DEFAULT_UST_UID_CHANNEL_OUTPUT;
1449 attr->switch_timer_interval = DEFAULT_UST_UID_CHANNEL_SWITCH_TIMER;
1450 attr->read_timer_interval = DEFAULT_UST_UID_CHANNEL_READ_TIMER;
1451 break;
1452 case LTTNG_BUFFER_PER_PID:
1453 default:
1454 attr->subbuf_size = default_get_ust_pid_channel_subbuf_size();
1455 attr->num_subbuf = DEFAULT_UST_PID_CHANNEL_SUBBUF_NUM;
1456 attr->output = DEFAULT_UST_PID_CHANNEL_OUTPUT;
1457 attr->switch_timer_interval = DEFAULT_UST_PID_CHANNEL_SWITCH_TIMER;
1458 attr->read_timer_interval = DEFAULT_UST_PID_CHANNEL_READ_TIMER;
1459 break;
1460 }
1461 default:
1462 /* Default behavior: leave set to 0. */
1463 break;
1464 }
1465 }
1466
1467 /*
1468 * Check if session daemon is alive.
1469 *
1470 * Return 1 if alive or 0 if not.
1471 * On error returns a negative value.
1472 */
1473 int lttng_session_daemon_alive(void)
1474 {
1475 int ret;
1476
1477 ret = set_session_daemon_path();
1478 if (ret < 0) {
1479 /* Error */
1480 return ret;
1481 }
1482
1483 if (*sessiond_sock_path == '\0') {
1484 /*
1485 * No socket path set. Weird error which means the constructor was not
1486 * called.
1487 */
1488 assert(0);
1489 }
1490
1491 ret = try_connect_sessiond(sessiond_sock_path);
1492 if (ret < 0) {
1493 /* Not alive */
1494 return 0;
1495 }
1496
1497 /* Is alive */
1498 return 1;
1499 }
1500
1501 /*
1502 * Set URL for a consumer for a session and domain.
1503 *
1504 * Return 0 on success, else a negative value.
1505 */
1506 int lttng_set_consumer_url(struct lttng_handle *handle,
1507 const char *control_url, const char *data_url)
1508 {
1509 int ret;
1510 ssize_t size;
1511 struct lttcomm_session_msg lsm;
1512 struct lttng_uri *uris = NULL;
1513
1514 if (handle == NULL || (control_url == NULL && data_url == NULL)) {
1515 return -LTTNG_ERR_INVALID;
1516 }
1517
1518 memset(&lsm, 0, sizeof(lsm));
1519
1520 lsm.cmd_type = LTTNG_SET_CONSUMER_URI;
1521
1522 lttng_ctl_copy_string(lsm.session.name, handle->session_name,
1523 sizeof(lsm.session.name));
1524 lttng_ctl_copy_lttng_domain(&lsm.domain, &handle->domain);
1525
1526 size = uri_parse_str_urls(control_url, data_url, &uris);
1527 if (size < 0) {
1528 return -LTTNG_ERR_INVALID;
1529 }
1530
1531 lsm.u.uri.size = size;
1532
1533 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1534 sizeof(struct lttng_uri) * size, NULL);
1535
1536 free(uris);
1537 return ret;
1538 }
1539
1540 /*
1541 * [OBSOLETE]
1542 */
1543 int lttng_enable_consumer(struct lttng_handle *handle)
1544 {
1545 return -ENOSYS;
1546 }
1547
1548 /*
1549 * [OBSOLETE]
1550 */
1551 int lttng_disable_consumer(struct lttng_handle *handle)
1552 {
1553 return -ENOSYS;
1554 }
1555
1556 /*
1557 * This is an extension of create session that is ONLY and SHOULD only be used
1558 * by the lttng command line program. It exists to avoid using URI parsing in
1559 * the lttng client.
1560 *
1561 * We need the date and time for the trace path subdirectory for the case where
1562 * the user does NOT define one using either -o or -U. Using the normal
1563 * lttng_create_session API call, we have no clue on the session daemon side if
1564 * the URL was generated automatically by the client or define by the user.
1565 *
1566 * So this function "wrapper" is hidden from the public API, takes the datetime
1567 * string and appends it if necessary to the URI subdirectory before sending it
1568 * to the session daemon.
1569 *
1570 * With this extra function, the lttng_create_session call behavior is not
1571 * changed and the timestamp is appended to the URI on the session daemon side
1572 * if necessary.
1573 */
1574 int _lttng_create_session_ext(const char *name, const char *url,
1575 const char *datetime)
1576 {
1577 int ret;
1578 ssize_t size;
1579 struct lttcomm_session_msg lsm;
1580 struct lttng_uri *uris = NULL;
1581
1582 if (name == NULL || datetime == NULL) {
1583 return -LTTNG_ERR_INVALID;
1584 }
1585
1586 memset(&lsm, 0, sizeof(lsm));
1587
1588 lsm.cmd_type = LTTNG_CREATE_SESSION;
1589 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1590
1591 /* There should never be a data URL */
1592 size = uri_parse_str_urls(url, NULL, &uris);
1593 if (size < 0) {
1594 ret = -LTTNG_ERR_INVALID;
1595 goto error;
1596 }
1597
1598 lsm.u.uri.size = size;
1599
1600 if (size > 0 && uris[0].dtype != LTTNG_DST_PATH && strlen(uris[0].subdir) == 0) {
1601 /* Don't append datetime if the name was automatically created. */
1602 if (strncmp(name, DEFAULT_SESSION_NAME "-",
1603 strlen(DEFAULT_SESSION_NAME) + 1)) {
1604 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s-%s",
1605 name, datetime);
1606 } else {
1607 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s", name);
1608 }
1609 if (ret < 0) {
1610 PERROR("snprintf uri subdir");
1611 ret = -LTTNG_ERR_FATAL;
1612 goto error;
1613 }
1614 }
1615
1616 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1617 sizeof(struct lttng_uri) * size, NULL);
1618
1619 error:
1620 free(uris);
1621 return ret;
1622 }
1623
1624 /*
1625 * For a given session name, this call checks if the data is ready to be read
1626 * or is still being extracted by the consumer(s) hence not ready to be used by
1627 * any readers.
1628 */
1629 int lttng_data_pending(const char *session_name)
1630 {
1631 int ret;
1632 struct lttcomm_session_msg lsm;
1633
1634 if (session_name == NULL) {
1635 return -LTTNG_ERR_INVALID;
1636 }
1637
1638 memset(&lsm, 0, sizeof(lsm));
1639 lsm.cmd_type = LTTNG_DATA_PENDING;
1640
1641 lttng_ctl_copy_string(lsm.session.name, session_name,
1642 sizeof(lsm.session.name));
1643
1644 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1645
1646 /*
1647 * The lttng_ctl_ask_sessiond function negate the return code if it's not
1648 * LTTNG_OK so getting -1 means that the reply ret_code was 1 thus meaning
1649 * that the data is available. Yes it is hackish but for now this is the
1650 * only way.
1651 */
1652 if (ret == -1) {
1653 ret = 1;
1654 }
1655
1656 return ret;
1657 }
1658
1659 /*
1660 * Create a session exclusively used for snapshot.
1661 *
1662 * Returns LTTNG_OK on success or a negative error code.
1663 */
1664 int lttng_create_session_snapshot(const char *name, const char *snapshot_url)
1665 {
1666 int ret;
1667 ssize_t size;
1668 struct lttcomm_session_msg lsm;
1669 struct lttng_uri *uris = NULL;
1670
1671 if (name == NULL) {
1672 return -LTTNG_ERR_INVALID;
1673 }
1674
1675 memset(&lsm, 0, sizeof(lsm));
1676
1677 lsm.cmd_type = LTTNG_CREATE_SESSION_SNAPSHOT;
1678 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1679
1680 size = uri_parse_str_urls(snapshot_url, NULL, &uris);
1681 if (size < 0) {
1682 return -LTTNG_ERR_INVALID;
1683 }
1684
1685 lsm.u.uri.size = size;
1686
1687 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1688 sizeof(struct lttng_uri) * size, NULL);
1689
1690 free(uris);
1691 return ret;
1692 }
1693
1694 /*
1695 * Create a session exclusively used for live.
1696 *
1697 * Returns LTTNG_OK on success or a negative error code.
1698 */
1699 int lttng_create_session_live(const char *name, const char *url,
1700 unsigned int timer_interval)
1701 {
1702 int ret;
1703 ssize_t size;
1704 struct lttcomm_session_msg lsm;
1705 struct lttng_uri *uris = NULL;
1706
1707 if (name == NULL) {
1708 return -LTTNG_ERR_INVALID;
1709 }
1710
1711 memset(&lsm, 0, sizeof(lsm));
1712
1713 lsm.cmd_type = LTTNG_CREATE_SESSION_LIVE;
1714 lttng_ctl_copy_string(lsm.session.name, name, sizeof(lsm.session.name));
1715
1716 if (url) {
1717 size = uri_parse_str_urls(url, NULL, &uris);
1718 if (size <= 0) {
1719 ret = -LTTNG_ERR_INVALID;
1720 goto end;
1721 }
1722
1723 /* file:// is not accepted for live session. */
1724 if (uris[0].dtype == LTTNG_DST_PATH) {
1725 ret = -LTTNG_ERR_INVALID;
1726 goto end;
1727 }
1728 } else {
1729 size = 0;
1730 }
1731
1732 lsm.u.session_live.nb_uri = size;
1733 lsm.u.session_live.timer_interval = timer_interval;
1734
1735 ret = lttng_ctl_ask_sessiond_varlen(&lsm, uris,
1736 sizeof(struct lttng_uri) * size, NULL);
1737
1738 end:
1739 free(uris);
1740 return ret;
1741 }
1742
1743 /*
1744 * lib constructor
1745 */
1746 static void __attribute__((constructor)) init()
1747 {
1748 /* Set default session group */
1749 lttng_set_tracing_group(DEFAULT_TRACING_GROUP);
1750 }
1751
1752 /*
1753 * lib destructor
1754 */
1755 static void __attribute__((destructor)) lttng_ctl_exit()
1756 {
1757 free(tracing_group);
1758 }
This page took 0.106284 seconds and 4 git commands to generate.