Mi: mi backend + mi for command version
[lttng-tools.git] / src / lib / lttng-ctl / lttng-ctl.c
... / ...
CommitLineData
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
46static const int print_xml = 1;
47#define dbg_printf(fmt, args...) \
48 printf("[debug liblttng-ctl] " fmt, ## args)
49#else
50static const int print_xml = 0;
51#define dbg_printf(fmt, args...) \
52do { \
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 */
61static int sessiond_socket;
62static char sessiond_sock_path[PATH_MAX];
63
64/* Variables */
65static char *tracing_group;
66static 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 */
75int lttng_opt_quiet;
76int lttng_opt_verbose;
77int lttng_opt_mi;
78
79/*
80 * Copy string from src to dst and enforce null terminated byte.
81 */
82LTTNG_HIDDEN
83void 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 */
99LTTNG_HIDDEN
100void 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 */
123static 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
140end:
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 */
150static 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
169end:
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 */
179static 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
193end:
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 */
202LTTNG_HIDDEN
203int 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
244free_list:
245 free(grp_list);
246
247end:
248 return ret;
249}
250
251/*
252 * Try connect to session daemon with sock_path.
253 *
254 * Return 0 on success, else -1
255 */
256static 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
280error:
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 */
291static 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 }
331end:
332 return 0;
333
334error:
335 return -1;
336}
337
338/*
339 * Connect to the LTTng session daemon.
340 *
341 * On success, return 0. On error, return -1.
342 */
343static 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
368error:
369 return -1;
370}
371
372/*
373 * Clean disconnect from the session daemon.
374 * On success, return 0. On error, return -1.
375 */
376static 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 */
395LTTNG_HIDDEN
396int 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
468end:
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 */
477struct 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
499end:
500 return handle;
501}
502
503/*
504 * Destroy handle by free(3) the pointer.
505 */
506void 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 */
515int 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 */
539int 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 */
559static 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
601end:
602error:
603 return ret;
604}
605
606/*
607 * Stop tracing and wait for data availability.
608 */
609int 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 */
617int 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 */
630int 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 */
670int 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 */
683int 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 */
697static 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;
749error:
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 */
761static 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
854parse_error:
855 filter_ir_free(ctx);
856 filter_parser_ctx_free(ctx);
857filter_alloc_error:
858 if (fclose(fmem) != 0) {
859 perror("fclose");
860 }
861error:
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 */
873int 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
1006mem_error:
1007 if (filter_expression && ctx) {
1008 filter_bytecode_free(ctx);
1009 filter_ir_free(ctx);
1010 filter_parser_ctx_free(ctx);
1011 }
1012filter_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 }
1020error:
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
1027ask_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 */
1038int 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 */
1078int 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 */
1108int 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 */
1138int 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 */
1166int 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 */
1192const 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 */
1202int 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 */
1237int 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 */
1260int 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 */
1281int 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 */
1311int 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 */
1342int 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 */
1375int 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 */
1391int 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 */
1414void 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 */
1467int 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 */
1500int 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 */
1537int lttng_enable_consumer(struct lttng_handle *handle)
1538{
1539 return -ENOSYS;
1540}
1541
1542/*
1543 * [OBSOLETE]
1544 */
1545int 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 */
1568int _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
1613error:
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 */
1623int 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 */
1658int 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 */
1693int 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
1732end:
1733 free(uris);
1734 return ret;
1735}
1736
1737/*
1738 * lib constructor
1739 */
1740static 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 */
1749static void __attribute__((destructor)) lttng_ctl_exit()
1750{
1751 free(tracing_group);
1752}
This page took 0.049231 seconds and 4 git commands to generate.