Save/load: support session trace format
[deliverable/lttng-tools.git] / src / lib / lttng-ctl / lttng-ctl.cpp
1 /*
2 * lttng-ctl.c
3 *
4 * Linux Trace Toolkit Control Library
5 *
6 * Copyright (C) 2011 EfficiOS Inc.
7 * Copyright (C) 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
8 *
9 * SPDX-License-Identifier: LGPL-2.1-only
10 *
11 */
12
13 #define _LGPL_SOURCE
14 #include <grp.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <stdint.h>
18 #include <string.h>
19 #include <unistd.h>
20
21 #include <common/bytecode/bytecode.hpp>
22 #include <common/align.hpp>
23 #include <common/common.hpp>
24 #include <common/compat/errno.hpp>
25 #include <common/compat/string.hpp>
26 #include <common/defaults.hpp>
27 #include <common/dynamic-array.hpp>
28 #include <common/dynamic-buffer.hpp>
29 #include <common/payload-view.hpp>
30 #include <common/payload.hpp>
31 #include <common/sessiond-comm/sessiond-comm.hpp>
32 #include <common/tracker.hpp>
33 #include <common/unix.hpp>
34 #include <common/uri.hpp>
35 #include <common/utils.hpp>
36 #include <lttng/channel-internal.hpp>
37 #include <lttng/destruction-handle.h>
38 #include <lttng/endpoint.h>
39 #include <lttng/error-query-internal.hpp>
40 #include <lttng/event-internal.hpp>
41 #include <lttng/health-internal.hpp>
42 #include <lttng/lttng-error.h>
43 #include <lttng/lttng.h>
44 #include <lttng/session-descriptor-internal.hpp>
45 #include <lttng/session-internal.hpp>
46 #include <lttng/trigger/trigger-internal.hpp>
47 #include <lttng/userspace-probe-internal.hpp>
48
49 #include "lttng-ctl-helper.hpp"
50 #include <common/filter/filter-ast.hpp>
51 #include <common/filter/filter-parser.hpp>
52 #include <common/filter/memstream.hpp>
53
54 #define COPY_DOMAIN_PACKED(dst, src) \
55 do { \
56 struct lttng_domain _tmp_domain; \
57 \
58 lttng_ctl_copy_lttng_domain(&_tmp_domain, &src); \
59 dst = _tmp_domain; \
60 } while (0)
61
62 /* Socket to session daemon for communication */
63 static int sessiond_socket = -1;
64 static char sessiond_sock_path[PATH_MAX];
65
66 /* Variables */
67 static char *tracing_group;
68 static int connected;
69
70 /* Global */
71
72 /*
73 * Those two variables are used by error.h to silent or control the verbosity of
74 * error message. They are global to the library so application linking with it
75 * are able to compile correctly and also control verbosity of the library.
76 */
77 LTTNG_EXPORT int lttng_opt_quiet;
78 LTTNG_EXPORT int lttng_opt_verbose;
79 LTTNG_EXPORT int lttng_opt_mi;
80
81 /*
82 * Copy domain to lttcomm_session_msg domain.
83 *
84 * If domain is unknown, default domain will be the kernel.
85 */
86 void lttng_ctl_copy_lttng_domain(struct lttng_domain *dst,
87 struct lttng_domain *src)
88 {
89 if (src && dst) {
90 switch (src->type) {
91 case LTTNG_DOMAIN_KERNEL:
92 case LTTNG_DOMAIN_UST:
93 case LTTNG_DOMAIN_JUL:
94 case LTTNG_DOMAIN_LOG4J:
95 case LTTNG_DOMAIN_PYTHON:
96 memcpy(dst, src, sizeof(struct lttng_domain));
97 break;
98 default:
99 memset(dst, 0, sizeof(struct lttng_domain));
100 break;
101 }
102 }
103 }
104
105 /*
106 * Send lttcomm_session_msg to the session daemon.
107 *
108 * On success, returns the number of bytes sent (>=0)
109 * On error, returns -1
110 */
111 static int send_session_msg(struct lttcomm_session_msg *lsm)
112 {
113 int ret;
114
115 if (!connected) {
116 ret = -LTTNG_ERR_NO_SESSIOND;
117 goto end;
118 }
119
120 DBG("LSM cmd type: '%s' (%d)", lttcomm_sessiond_command_str((lttcomm_sessiond_command) lsm->cmd_type),
121 lsm->cmd_type);
122
123 ret = lttcomm_send_creds_unix_sock(sessiond_socket, lsm,
124 sizeof(struct lttcomm_session_msg));
125 if (ret < 0) {
126 ret = -LTTNG_ERR_FATAL;
127 }
128
129 end:
130 return ret;
131 }
132
133 /*
134 * Send var len data to the session daemon.
135 *
136 * On success, returns the number of bytes sent (>=0)
137 * On error, returns -1
138 */
139 static int send_session_varlen(const void *data, size_t len)
140 {
141 int ret;
142
143 if (!connected) {
144 ret = -LTTNG_ERR_NO_SESSIOND;
145 goto end;
146 }
147
148 if (!data || !len) {
149 ret = 0;
150 goto end;
151 }
152
153 ret = lttcomm_send_unix_sock(sessiond_socket, data, len);
154 if (ret < 0) {
155 ret = -LTTNG_ERR_FATAL;
156 }
157
158 end:
159 return ret;
160 }
161
162 /*
163 * Send file descriptors to the session daemon.
164 *
165 * On success, returns the number of bytes sent (>=0)
166 * On error, returns -1
167 */
168 static int send_session_fds(const int *fds, size_t nb_fd)
169 {
170 int ret;
171
172 if (!connected) {
173 ret = -LTTNG_ERR_NO_SESSIOND;
174 goto end;
175 }
176
177 if (!fds || !nb_fd) {
178 ret = 0;
179 goto end;
180 }
181
182 ret = lttcomm_send_fds_unix_sock(sessiond_socket, fds, nb_fd);
183 if (ret < 0) {
184 ret = -LTTNG_ERR_FATAL;
185 }
186
187 end:
188 return ret;
189 }
190
191 /*
192 * Receive data from the sessiond socket.
193 *
194 * On success, returns the number of bytes received (>=0)
195 * On error, returns a negative lttng_error_code.
196 */
197 static int recv_data_sessiond(void *buf, size_t len)
198 {
199 int ret;
200
201 LTTNG_ASSERT(len > 0);
202
203 if (!connected) {
204 ret = -LTTNG_ERR_NO_SESSIOND;
205 goto end;
206 }
207
208 ret = lttcomm_recv_unix_sock(sessiond_socket, buf, len);
209 if (ret < 0) {
210 ret = -LTTNG_ERR_FATAL;
211 } else if (ret == 0) {
212 ret = -LTTNG_ERR_NO_SESSIOND;
213 }
214
215 end:
216 return ret;
217 }
218
219 /*
220 * Receive a payload from the session daemon by appending to an existing
221 * payload.
222 * On success, returns the number of bytes received (>=0)
223 * On error, returns a negative lttng_error_code.
224 */
225 static int recv_payload_sessiond(struct lttng_payload *payload, size_t len)
226 {
227 int ret;
228 const size_t original_payload_size = payload->buffer.size;
229
230 ret = lttng_dynamic_buffer_set_size(
231 &payload->buffer, payload->buffer.size + len);
232 if (ret) {
233 ret = -LTTNG_ERR_NOMEM;
234 goto end;
235 }
236
237 ret = recv_data_sessiond(
238 payload->buffer.data + original_payload_size, len);
239 end:
240 return ret;
241 }
242
243 /*
244 * Check if we are in the specified group.
245 *
246 * If yes return 1, else return -1.
247 */
248 int lttng_check_tracing_group(void)
249 {
250 gid_t *grp_list, tracing_gid;
251 int grp_list_size, grp_id, i;
252 int ret = -1;
253 const char *grp_name = tracing_group;
254
255 /* Get GID of group 'tracing' */
256 if (utils_get_group_id(grp_name, false, &tracing_gid)) {
257 /* If grp_tracing is NULL, the group does not exist. */
258 goto end;
259 }
260
261 /* Get number of supplementary group IDs */
262 grp_list_size = getgroups(0, NULL);
263 if (grp_list_size < 0) {
264 PERROR("getgroups");
265 goto end;
266 }
267
268 /* Alloc group list of the right size */
269 grp_list = calloc<gid_t>(grp_list_size);
270 if (!grp_list) {
271 PERROR("malloc");
272 goto end;
273 }
274 grp_id = getgroups(grp_list_size, grp_list);
275 if (grp_id < 0) {
276 PERROR("getgroups");
277 goto free_list;
278 }
279
280 for (i = 0; i < grp_list_size; i++) {
281 if (grp_list[i] == tracing_gid) {
282 ret = 1;
283 break;
284 }
285 }
286
287 free_list:
288 free(grp_list);
289
290 end:
291 return ret;
292 }
293
294 static enum lttng_error_code check_enough_available_memory(
295 uint64_t num_bytes_requested_per_cpu)
296 {
297 int ret;
298 enum lttng_error_code ret_code;
299 long num_cpu;
300 uint64_t best_mem_info;
301 uint64_t num_bytes_requested_total;
302
303 /*
304 * Get the number of CPU currently online to compute the amount of
305 * memory needed to create a buffer for every CPU.
306 */
307 num_cpu = sysconf(_SC_NPROCESSORS_ONLN);
308 if (num_cpu == -1) {
309 ret_code = LTTNG_ERR_FATAL;
310 goto end;
311 }
312
313 if (num_bytes_requested_per_cpu > UINT64_MAX / (uint64_t) num_cpu) {
314 /* Overflow */
315 ret_code = LTTNG_ERR_OVERFLOW;
316 goto end;
317 }
318
319 num_bytes_requested_total =
320 num_bytes_requested_per_cpu * (uint64_t) num_cpu;
321
322 /*
323 * Try to get the `MemAvail` field of `/proc/meminfo`. This is the most
324 * reliable estimate we can get but it is only exposed by the kernel
325 * since 3.14. (See Linux kernel commit:
326 * 34e431b0ae398fc54ea69ff85ec700722c9da773)
327 */
328 ret = utils_get_memory_available(&best_mem_info);
329 if (ret >= 0) {
330 goto success;
331 }
332
333 /*
334 * As a backup plan, use `MemTotal` field of `/proc/meminfo`. This
335 * is a sanity check for obvious user error.
336 */
337 ret = utils_get_memory_total(&best_mem_info);
338 if (ret >= 0) {
339 goto success;
340 }
341
342 /* No valid source of information. */
343 ret_code = LTTNG_ERR_NOMEM;
344 goto end;
345
346 success:
347 if (best_mem_info >= num_bytes_requested_total) {
348 ret_code = LTTNG_OK;
349 } else {
350 ret_code = LTTNG_ERR_NOMEM;
351 }
352 end:
353 return ret_code;
354 }
355
356 /*
357 * Try connect to session daemon with sock_path.
358 *
359 * Return 0 on success, else -1
360 */
361 static int try_connect_sessiond(const char *sock_path)
362 {
363 int ret;
364
365 /* If socket exist, we check if the daemon listens for connect. */
366 ret = access(sock_path, F_OK);
367 if (ret < 0) {
368 /* Not alive */
369 goto error;
370 }
371
372 ret = lttcomm_connect_unix_sock(sock_path);
373 if (ret < 0) {
374 /* Not alive. */
375 goto error;
376 }
377
378 ret = lttcomm_close_unix_sock(ret);
379 if (ret < 0) {
380 PERROR("lttcomm_close_unix_sock");
381 }
382
383 return 0;
384
385 error:
386 return -1;
387 }
388
389 /*
390 * Set sessiond socket path by putting it in the global sessiond_sock_path
391 * variable.
392 *
393 * Returns 0 on success, negative value on failure (the sessiond socket path
394 * is somehow too long or ENOMEM).
395 */
396 static int set_session_daemon_path(void)
397 {
398 int in_tgroup = 0; /* In tracing group. */
399 uid_t uid;
400
401 uid = getuid();
402
403 if (uid != 0) {
404 /* Are we in the tracing group ? */
405 in_tgroup = lttng_check_tracing_group();
406 }
407
408 if ((uid == 0) || in_tgroup == 1) {
409 const int ret = lttng_strncpy(sessiond_sock_path,
410 DEFAULT_GLOBAL_CLIENT_UNIX_SOCK,
411 sizeof(sessiond_sock_path));
412
413 if (ret) {
414 goto error;
415 }
416 }
417
418 if (uid != 0) {
419 int ret;
420
421 if (in_tgroup) {
422 /* Tracing group. */
423 ret = try_connect_sessiond(sessiond_sock_path);
424 if (ret >= 0) {
425 goto end;
426 }
427 /* Global session daemon not available... */
428 }
429 /* ...or not in tracing group (and not root), default */
430
431 /*
432 * With GNU C < 2.1, snprintf returns -1 if the target buffer
433 * is too small;
434 * With GNU C >= 2.1, snprintf returns the required size
435 * (excluding closing null)
436 */
437 ret = snprintf(sessiond_sock_path, sizeof(sessiond_sock_path),
438 DEFAULT_HOME_CLIENT_UNIX_SOCK, utils_get_home_dir());
439 if ((ret < 0) || (ret >= sizeof(sessiond_sock_path))) {
440 goto error;
441 }
442 }
443 end:
444 return 0;
445
446 error:
447 return -1;
448 }
449
450 /*
451 * Connect to the LTTng session daemon.
452 *
453 * On success, return the socket's file descriptor. On error, return -1.
454 */
455 int connect_sessiond(void)
456 {
457 int ret;
458
459 ret = set_session_daemon_path();
460 if (ret < 0) {
461 goto error;
462 }
463
464 /* Connect to the sesssion daemon. */
465 ret = lttcomm_connect_unix_sock(sessiond_sock_path);
466 if (ret < 0) {
467 goto error;
468 }
469
470 return ret;
471
472 error:
473 return -1;
474 }
475
476 static void reset_global_sessiond_connection_state(void)
477 {
478 sessiond_socket = -1;
479 connected = 0;
480 }
481
482 /*
483 * Clean disconnect from the session daemon.
484 *
485 * On success, return 0. On error, return -1.
486 */
487 static int disconnect_sessiond(void)
488 {
489 int ret = 0;
490
491 if (connected) {
492 ret = lttcomm_close_unix_sock(sessiond_socket);
493 reset_global_sessiond_connection_state();
494 }
495
496 return ret;
497 }
498
499 static int recv_sessiond_optional_data(size_t len, void **user_buf,
500 size_t *user_len)
501 {
502 int ret = 0;
503 char *buf = NULL;
504
505 if (len) {
506 if (!user_len) {
507 ret = -LTTNG_ERR_INVALID;
508 goto end;
509 }
510
511 buf = zmalloc<char>(len);
512 if (!buf) {
513 ret = -ENOMEM;
514 goto end;
515 }
516
517 ret = recv_data_sessiond(buf, len);
518 if (ret < 0) {
519 goto end;
520 }
521
522 if (!user_buf) {
523 ret = -LTTNG_ERR_INVALID;
524 goto end;
525 }
526
527 /* Move ownership of command header buffer to user. */
528 *user_buf = buf;
529 buf = NULL;
530 *user_len = len;
531 } else {
532 /* No command header. */
533 if (user_len) {
534 *user_len = 0;
535 }
536
537 if (user_buf) {
538 *user_buf = NULL;
539 }
540 }
541
542 end:
543 free(buf);
544 return ret;
545 }
546
547 /*
548 * Ask the session daemon a specific command and put the data into buf.
549 * Takes extra var. len. data and file descriptors as input to send to the
550 * session daemon.
551 *
552 * Return size of data (only payload, not header) or a negative error code.
553 */
554 int lttng_ctl_ask_sessiond_fds_varlen(struct lttcomm_session_msg *lsm,
555 const int *fds, size_t nb_fd, const void *vardata,
556 size_t vardata_len, void **user_payload_buf,
557 void **user_cmd_header_buf, size_t *user_cmd_header_len)
558 {
559 int ret;
560 size_t payload_len;
561 struct lttcomm_lttng_msg llm;
562
563 ret = connect_sessiond();
564 if (ret < 0) {
565 ret = -LTTNG_ERR_NO_SESSIOND;
566 goto end;
567 } else {
568 sessiond_socket = ret;
569 connected = 1;
570 }
571
572 ret = send_session_msg(lsm);
573 if (ret < 0) {
574 /* Ret value is a valid lttng error code. */
575 goto end;
576 }
577 /* Send var len data */
578 ret = send_session_varlen(vardata, vardata_len);
579 if (ret < 0) {
580 /* Ret value is a valid lttng error code. */
581 goto end;
582 }
583
584 /* Send fds */
585 ret = send_session_fds(fds, nb_fd);
586 if (ret < 0) {
587 /* Ret value is a valid lttng error code. */
588 goto end;
589 }
590
591 /* Get header from data transmission */
592 ret = recv_data_sessiond(&llm, sizeof(llm));
593 if (ret < 0) {
594 /* Ret value is a valid lttng error code. */
595 goto end;
596 }
597
598 /* Check error code if OK */
599 if (llm.ret_code != LTTNG_OK) {
600 ret = -llm.ret_code;
601 goto end;
602 }
603
604 /* Get command header from data transmission */
605 ret = recv_sessiond_optional_data(llm.cmd_header_size,
606 user_cmd_header_buf, user_cmd_header_len);
607 if (ret < 0) {
608 goto end;
609 }
610
611 /* Get payload from data transmission */
612 ret = recv_sessiond_optional_data(llm.data_size, user_payload_buf,
613 &payload_len);
614 if (ret < 0) {
615 goto end;
616 }
617
618 ret = llm.data_size;
619
620 end:
621 disconnect_sessiond();
622 return ret;
623 }
624
625 int lttng_ctl_ask_sessiond_payload(struct lttng_payload_view *message,
626 struct lttng_payload *reply)
627 {
628 int ret;
629 struct lttcomm_lttng_msg llm;
630 const int fd_count = lttng_payload_view_get_fd_handle_count(message);
631
632 LTTNG_ASSERT(reply->buffer.size == 0);
633 LTTNG_ASSERT(lttng_dynamic_pointer_array_get_count(&reply->_fd_handles) == 0);
634
635 ret = connect_sessiond();
636 if (ret < 0) {
637 ret = -LTTNG_ERR_NO_SESSIOND;
638 goto end;
639 } else {
640 sessiond_socket = ret;
641 connected = 1;
642 }
643
644 /* Send command to session daemon */
645 ret = lttcomm_send_creds_unix_sock(sessiond_socket, message->buffer.data,
646 message->buffer.size);
647 if (ret < 0) {
648 ret = -LTTNG_ERR_FATAL;
649 goto end;
650 }
651
652 if (fd_count > 0) {
653 ret = lttcomm_send_payload_view_fds_unix_sock(sessiond_socket,
654 message);
655 if (ret < 0) {
656 ret = -LTTNG_ERR_FATAL;
657 goto end;
658 }
659 }
660
661 /* Get header from data transmission */
662 ret = recv_payload_sessiond(reply, sizeof(llm));
663 if (ret < 0) {
664 /* Ret value is a valid lttng error code. */
665 goto end;
666 }
667
668 llm = *((typeof(llm) *) reply->buffer.data);
669
670 /* Check error code if OK */
671 if (llm.ret_code != LTTNG_OK) {
672 if (llm.ret_code < LTTNG_OK || llm.ret_code >= LTTNG_ERR_NR) {
673 /* Invalid error code received. */
674 ret = -LTTNG_ERR_UNK;
675 } else {
676 ret = -llm.ret_code;
677 }
678 goto end;
679 }
680
681 if (llm.cmd_header_size > 0) {
682 ret = recv_payload_sessiond(reply, llm.cmd_header_size);
683 if (ret < 0) {
684 goto end;
685 }
686 }
687
688 /* Get command header from data transmission */
689 if (llm.data_size > 0) {
690 ret = recv_payload_sessiond(reply, llm.data_size);
691 if (ret < 0) {
692 goto end;
693 }
694 }
695
696 if (llm.fd_count > 0) {
697 ret = lttcomm_recv_payload_fds_unix_sock(
698 sessiond_socket, llm.fd_count, reply);
699 if (ret < 0) {
700 goto end;
701 }
702 }
703
704 /* Don't return the llm header to the caller. */
705 memmove(reply->buffer.data, reply->buffer.data + sizeof(llm),
706 reply->buffer.size - sizeof(llm));
707 ret = lttng_dynamic_buffer_set_size(
708 &reply->buffer, reply->buffer.size - sizeof(llm));
709 if (ret) {
710 /* Can't happen as size is reduced. */
711 abort();
712 }
713
714 ret = reply->buffer.size;
715
716 end:
717 disconnect_sessiond();
718 return ret;
719 }
720
721 /*
722 * Create lttng handle and return pointer.
723 *
724 * The returned pointer will be NULL in case of malloc() error.
725 */
726 struct lttng_handle *lttng_create_handle(const char *session_name,
727 struct lttng_domain *domain)
728 {
729 int ret;
730 struct lttng_handle *handle = NULL;
731
732 handle = zmalloc<lttng_handle>();
733 if (handle == NULL) {
734 PERROR("malloc handle");
735 goto end;
736 }
737
738 /* Copy session name */
739 ret = lttng_strncpy(handle->session_name, session_name ? : "",
740 sizeof(handle->session_name));
741 if (ret) {
742 goto error;
743 }
744
745 /* Copy lttng domain or leave initialized to 0. */
746 if (domain) {
747 lttng_ctl_copy_lttng_domain(&handle->domain, domain);
748 }
749
750 end:
751 return handle;
752 error:
753 free(handle);
754 return NULL;
755 }
756
757 /*
758 * Destroy handle by free(3) the pointer.
759 */
760 void lttng_destroy_handle(struct lttng_handle *handle)
761 {
762 free(handle);
763 }
764
765 /*
766 * Register an outside consumer.
767 *
768 * Returns size of returned session payload data or a negative error code.
769 */
770 int lttng_register_consumer(struct lttng_handle *handle,
771 const char *socket_path)
772 {
773 int ret;
774 struct lttcomm_session_msg lsm;
775
776 if (handle == NULL || socket_path == NULL) {
777 ret = -LTTNG_ERR_INVALID;
778 goto end;
779 }
780
781 memset(&lsm, 0, sizeof(lsm));
782 lsm.cmd_type = LTTNG_REGISTER_CONSUMER;
783 ret = lttng_strncpy(lsm.session.name, handle->session_name,
784 sizeof(lsm.session.name));
785 if (ret) {
786 ret = -LTTNG_ERR_INVALID;
787 goto end;
788 }
789
790 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
791
792 ret = lttng_strncpy(lsm.u.reg.path, socket_path,
793 sizeof(lsm.u.reg.path));
794 if (ret) {
795 ret = -LTTNG_ERR_INVALID;
796 goto end;
797 }
798
799 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
800 end:
801 return ret;
802 }
803
804 /*
805 * Start tracing for all traces of the session.
806 *
807 * Returns size of returned session payload data or a negative error code.
808 */
809 int lttng_start_tracing(const char *session_name)
810 {
811 int ret;
812 struct lttcomm_session_msg lsm;
813
814 if (session_name == NULL) {
815 ret = -LTTNG_ERR_INVALID;
816 goto end;
817 }
818
819 memset(&lsm, 0, sizeof(lsm));
820 lsm.cmd_type = LTTNG_START_TRACE;
821
822 ret = lttng_strncpy(lsm.session.name, session_name,
823 sizeof(lsm.session.name));
824 if (ret) {
825 ret = -LTTNG_ERR_INVALID;
826 goto end;
827 }
828
829 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
830 end:
831 return ret;
832 }
833
834 /*
835 * Stop tracing for all traces of the session.
836 */
837 static int _lttng_stop_tracing(const char *session_name, int wait)
838 {
839 int ret, data_ret;
840 struct lttcomm_session_msg lsm;
841
842 if (session_name == NULL) {
843 ret = -LTTNG_ERR_INVALID;
844 goto error;
845 }
846
847 memset(&lsm, 0, sizeof(lsm));
848 lsm.cmd_type = LTTNG_STOP_TRACE;
849
850 ret = lttng_strncpy(lsm.session.name, session_name,
851 sizeof(lsm.session.name));
852 if (ret) {
853 ret = -LTTNG_ERR_INVALID;
854 goto error;
855 }
856
857 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
858 if (ret < 0 && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
859 goto error;
860 }
861
862 if (!wait) {
863 goto end;
864 }
865
866 /* Check for data availability */
867 do {
868 data_ret = lttng_data_pending(session_name);
869 if (data_ret < 0) {
870 /* Return the data available call error. */
871 ret = data_ret;
872 goto error;
873 }
874
875 /*
876 * Data sleep time before retrying (in usec). Don't sleep if the
877 * call returned value indicates availability.
878 */
879 if (data_ret) {
880 usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME_US);
881 }
882 } while (data_ret != 0);
883
884 end:
885 error:
886 return ret;
887 }
888
889 /*
890 * Stop tracing and wait for data availability.
891 */
892 int lttng_stop_tracing(const char *session_name)
893 {
894 return _lttng_stop_tracing(session_name, 1);
895 }
896
897 /*
898 * Stop tracing but _don't_ wait for data availability.
899 */
900 int lttng_stop_tracing_no_wait(const char *session_name)
901 {
902 return _lttng_stop_tracing(session_name, 0);
903 }
904
905 /*
906 * Add context to a channel.
907 *
908 * If the given channel is NULL, add the contexts to all channels.
909 * The event_name param is ignored.
910 *
911 * Returns the size of the returned payload data or a negative error code.
912 */
913 int lttng_add_context(struct lttng_handle *handle,
914 struct lttng_event_context *ctx,
915 const char *event_name __attribute__((unused)),
916 const char *channel_name)
917 {
918 int ret;
919 struct lttcomm_session_msg lsm = {
920 .cmd_type = LTTNG_ADD_CONTEXT,
921 .session = {},
922 .domain = {},
923 .u = {},
924 .fd_count = 0,
925 };
926 struct lttng_payload payload;
927
928 lttng_payload_init(&payload);
929
930 /* Safety check. Both are mandatory. */
931 if (handle == NULL || ctx == NULL) {
932 ret = -LTTNG_ERR_INVALID;
933 goto end;
934 }
935
936 ret = lttng_dynamic_buffer_set_size(&payload.buffer, sizeof(lsm));
937 if (ret) {
938 ret = -LTTNG_ERR_NOMEM;
939 goto end;
940 }
941
942 /* If no channel name, send empty string. */
943 ret = lttng_strncpy(lsm.u.context.channel_name, channel_name ?: "",
944 sizeof(lsm.u.context.channel_name));
945 if (ret) {
946 ret = -LTTNG_ERR_INVALID;
947 goto end;
948 }
949
950 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
951 ret = lttng_strncpy(lsm.session.name, handle->session_name,
952 sizeof(lsm.session.name));
953 if (ret) {
954 ret = -LTTNG_ERR_INVALID;
955 goto end;
956 }
957
958 ret = lttng_event_context_serialize(ctx, &payload);
959 if (ret) {
960 ret = -LTTNG_ERR_INVALID;
961 goto end;
962 }
963
964 lsm.u.context.length = payload.buffer.size - sizeof(lsm);
965
966 /* Update message header. */
967 memcpy(payload.buffer.data, &lsm, sizeof(lsm));
968
969 {
970 struct lttng_payload reply;
971 struct lttng_payload_view payload_view =
972 lttng_payload_view_from_payload(&payload, 0,
973 -1);
974
975 lttng_payload_init(&reply);
976 ret = lttng_ctl_ask_sessiond_payload(&payload_view, &reply);
977 lttng_payload_reset(&reply);
978 if (ret) {
979 goto end;
980 }
981 }
982
983 end:
984 lttng_payload_reset(&payload);
985 return ret;
986 }
987
988 /*
989 * Enable event(s) for a channel.
990 *
991 * If no event name is specified, all events are enabled.
992 * If no channel name is specified, the default 'channel0' is used.
993 *
994 * Returns size of returned session payload data or a negative error code.
995 */
996 int lttng_enable_event(struct lttng_handle *handle,
997 struct lttng_event *ev, const char *channel_name)
998 {
999 return lttng_enable_event_with_exclusions(handle, ev, channel_name,
1000 NULL, 0, NULL);
1001 }
1002
1003 /*
1004 * Create or enable an event with a filter expression.
1005 *
1006 * Return negative error value on error.
1007 * Return size of returned session payload data if OK.
1008 */
1009 int lttng_enable_event_with_filter(struct lttng_handle *handle,
1010 struct lttng_event *event, const char *channel_name,
1011 const char *filter_expression)
1012 {
1013 return lttng_enable_event_with_exclusions(handle, event, channel_name,
1014 filter_expression, 0, NULL);
1015 }
1016
1017 /*
1018 * Depending on the event, return a newly allocated agent filter expression or
1019 * NULL if not applicable.
1020 *
1021 * An event with NO loglevel and the name is * will return NULL.
1022 */
1023 static char *set_agent_filter(const char *filter, struct lttng_event *ev)
1024 {
1025 int err;
1026 char *agent_filter = NULL;
1027
1028 LTTNG_ASSERT(ev);
1029
1030 /* Don't add filter for the '*' event. */
1031 if (strcmp(ev->name, "*") != 0) {
1032 if (filter) {
1033 err = asprintf(&agent_filter, "(%s) && (logger_name == \"%s\")", filter,
1034 ev->name);
1035 } else {
1036 err = asprintf(&agent_filter, "logger_name == \"%s\"", ev->name);
1037 }
1038 if (err < 0) {
1039 PERROR("asprintf");
1040 goto error;
1041 }
1042 }
1043
1044 /* Add loglevel filtering if any for the JUL domain. */
1045 if (ev->loglevel_type != LTTNG_EVENT_LOGLEVEL_ALL) {
1046 const char *op;
1047
1048 if (ev->loglevel_type == LTTNG_EVENT_LOGLEVEL_RANGE) {
1049 op = ">=";
1050 } else {
1051 op = "==";
1052 }
1053
1054 if (filter || agent_filter) {
1055 char *new_filter;
1056
1057 err = asprintf(&new_filter, "(%s) && (int_loglevel %s %d)",
1058 agent_filter ? agent_filter : filter, op,
1059 ev->loglevel);
1060 if (agent_filter) {
1061 free(agent_filter);
1062 }
1063 agent_filter = new_filter;
1064 } else {
1065 err = asprintf(&agent_filter, "int_loglevel %s %d", op,
1066 ev->loglevel);
1067 }
1068 if (err < 0) {
1069 PERROR("asprintf");
1070 goto error;
1071 }
1072 }
1073
1074 return agent_filter;
1075 error:
1076 free(agent_filter);
1077 return NULL;
1078 }
1079
1080 /*
1081 * Enable event(s) for a channel, possibly with exclusions and a filter.
1082 * If no event name is specified, all events are enabled.
1083 * If no channel name is specified, the default name is used.
1084 * If filter expression is not NULL, the filter is set for the event.
1085 * If exclusion count is not zero, the exclusions are set for the event.
1086 * Returns size of returned session payload data or a negative error code.
1087 */
1088 int lttng_enable_event_with_exclusions(struct lttng_handle *handle,
1089 struct lttng_event *ev, const char *channel_name,
1090 const char *original_filter_expression,
1091 int exclusion_count, char **exclusion_list)
1092 {
1093 struct lttcomm_session_msg lsm = {
1094 .cmd_type = LTTNG_ENABLE_EVENT,
1095 .session = {},
1096 .domain = {},
1097 .u = {},
1098 .fd_count = 0,
1099 };
1100 struct lttng_payload payload;
1101 int ret = 0;
1102 unsigned int free_filter_expression = 0;
1103 struct filter_parser_ctx *ctx = NULL;
1104 size_t bytecode_len = 0;
1105
1106 /*
1107 * We have either a filter or some exclusions, so we need to set up
1108 * a variable-length payload from where to send the data.
1109 */
1110 lttng_payload_init(&payload);
1111
1112 /*
1113 * Cast as non-const since we may replace the filter expression
1114 * by a dynamically allocated string. Otherwise, the original
1115 * string is not modified.
1116 */
1117 char *filter_expression = (char *) original_filter_expression;
1118
1119 if (handle == NULL || ev == NULL) {
1120 ret = -LTTNG_ERR_INVALID;
1121 goto error;
1122 }
1123
1124 /*
1125 * Empty filter string will always be rejected by the parser
1126 * anyway, so treat this corner-case early to eliminate
1127 * lttng_fmemopen error for 0-byte allocation.
1128 */
1129 if (filter_expression && filter_expression[0] == '\0') {
1130 ret = -LTTNG_ERR_INVALID;
1131 goto error;
1132 }
1133
1134 if (ev->name[0] == '\0') {
1135 /* Enable all events. */
1136 ret = lttng_strncpy(ev->name, "*", sizeof(ev->name));
1137 LTTNG_ASSERT(ret == 0);
1138 }
1139
1140 /* Parse filter expression. */
1141 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
1142 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1143 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1144 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
1145 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1146 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1147 char *agent_filter;
1148
1149 /* Setup agent filter if needed. */
1150 agent_filter = set_agent_filter(filter_expression, ev);
1151 if (!agent_filter) {
1152 if (!filter_expression) {
1153 /*
1154 * No JUL and no filter, just skip
1155 * everything below.
1156 */
1157 goto serialize;
1158 }
1159 } else {
1160 /*
1161 * With an agent filter, the original filter has
1162 * been added to it thus replace the filter
1163 * expression.
1164 */
1165 filter_expression = agent_filter;
1166 free_filter_expression = 1;
1167 }
1168 }
1169
1170 if (strnlen(filter_expression, LTTNG_FILTER_MAX_LEN) ==
1171 LTTNG_FILTER_MAX_LEN) {
1172 ret = -LTTNG_ERR_FILTER_INVAL;
1173 goto error;
1174 }
1175
1176 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
1177 if (ret) {
1178 goto error;
1179 }
1180
1181 bytecode_len = bytecode_get_len(&ctx->bytecode->b) +
1182 sizeof(ctx->bytecode->b);
1183 if (bytecode_len > LTTNG_FILTER_MAX_LEN) {
1184 ret = -LTTNG_ERR_FILTER_INVAL;
1185 goto error;
1186 }
1187 }
1188
1189 serialize:
1190 ret = lttng_event_serialize(ev, exclusion_count, exclusion_list,
1191 filter_expression, bytecode_len,
1192 (ctx && bytecode_len) ? &ctx->bytecode->b : NULL,
1193 &payload);
1194 if (ret) {
1195 ret = -LTTNG_ERR_INVALID;
1196 goto error;
1197 }
1198
1199 /* If no channel name, send empty string. */
1200 ret = lttng_strncpy(lsm.u.enable.channel_name, channel_name ?: "",
1201 sizeof(lsm.u.enable.channel_name));
1202 if (ret) {
1203 ret = -LTTNG_ERR_INVALID;
1204 goto error;
1205 }
1206
1207 /* Domain */
1208 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1209
1210 /* Session name */
1211 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1212 sizeof(lsm.session.name));
1213 if (ret) {
1214 ret = -LTTNG_ERR_INVALID;
1215 goto error;
1216 }
1217
1218 /* Length of the serialized event. */
1219 lsm.u.enable.length = (uint32_t) payload.buffer.size;
1220
1221 {
1222 struct lttng_payload_view view = lttng_payload_view_from_payload(
1223 &payload, 0, -1);
1224 int fd_count = lttng_payload_view_get_fd_handle_count(&view);
1225 int fd_to_send;
1226
1227 if (fd_count < 0) {
1228 goto error;
1229 }
1230
1231 LTTNG_ASSERT(fd_count == 0 || fd_count == 1);
1232 if (fd_count == 1) {
1233 struct fd_handle *h =
1234 lttng_payload_view_pop_fd_handle(&view);
1235
1236 if (!h) {
1237 goto error;
1238 }
1239
1240 fd_to_send = fd_handle_get_fd(h);
1241 fd_handle_put(h);
1242 }
1243
1244 lsm.fd_count = fd_count;
1245
1246 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm,
1247 fd_count ? &fd_to_send : NULL, fd_count,
1248 view.buffer.size ? view.buffer.data : NULL,
1249 view.buffer.size, NULL, NULL, 0);
1250 }
1251
1252 error:
1253 if (filter_expression && ctx) {
1254 filter_bytecode_free(ctx);
1255 filter_ir_free(ctx);
1256 filter_parser_ctx_free(ctx);
1257 }
1258 if (free_filter_expression) {
1259 /*
1260 * The filter expression has been replaced and must be freed as
1261 * it is not the original filter expression received as a
1262 * parameter.
1263 */
1264 free(filter_expression);
1265 }
1266 /*
1267 * Return directly to the caller and don't ask the sessiond since
1268 * something went wrong in the parsing of data above.
1269 */
1270 lttng_payload_reset(&payload);
1271 return ret;
1272 }
1273
1274 int lttng_disable_event_ext(struct lttng_handle *handle,
1275 struct lttng_event *ev, const char *channel_name,
1276 const char *original_filter_expression)
1277 {
1278 struct lttcomm_session_msg lsm = {
1279 .cmd_type = LTTNG_DISABLE_EVENT,
1280 .session = {},
1281 .domain = {},
1282 .u = {},
1283 .fd_count = 0,
1284 };
1285 struct lttng_payload payload;
1286 int ret = 0;
1287 unsigned int free_filter_expression = 0;
1288 struct filter_parser_ctx *ctx = NULL;
1289 size_t bytecode_len = 0;
1290
1291 /*
1292 * We have either a filter or some exclusions, so we need to set up
1293 * a variable-length payload from where to send the data.
1294 */
1295 lttng_payload_init(&payload);
1296
1297 /*
1298 * Cast as non-const since we may replace the filter expression
1299 * by a dynamically allocated string. Otherwise, the original
1300 * string is not modified.
1301 */
1302 char *filter_expression = (char *) original_filter_expression;
1303
1304 if (handle == NULL || ev == NULL) {
1305 ret = -LTTNG_ERR_INVALID;
1306 goto error;
1307 }
1308
1309 /*
1310 * Empty filter string will always be rejected by the parser
1311 * anyway, so treat this corner-case early to eliminate
1312 * lttng_fmemopen error for 0-byte allocation.
1313 */
1314 if (filter_expression && filter_expression[0] == '\0') {
1315 ret = -LTTNG_ERR_INVALID;
1316 goto error;
1317 }
1318
1319 /* Parse filter expression. */
1320 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
1321 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1322 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1323 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
1324 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1325 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1326 char *agent_filter;
1327
1328 /* Setup agent filter if needed. */
1329 agent_filter = set_agent_filter(filter_expression, ev);
1330 if (!agent_filter) {
1331 if (!filter_expression) {
1332 /*
1333 * No JUL and no filter, just skip
1334 * everything below.
1335 */
1336 goto serialize;
1337 }
1338 } else {
1339 /*
1340 * With an agent filter, the original filter has
1341 * been added to it thus replace the filter
1342 * expression.
1343 */
1344 filter_expression = agent_filter;
1345 free_filter_expression = 1;
1346 }
1347 }
1348
1349 if (strnlen(filter_expression, LTTNG_FILTER_MAX_LEN) ==
1350 LTTNG_FILTER_MAX_LEN) {
1351 ret = -LTTNG_ERR_FILTER_INVAL;
1352 goto error;
1353 }
1354
1355 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
1356 if (ret) {
1357 goto error;
1358 }
1359
1360 bytecode_len = bytecode_get_len(&ctx->bytecode->b) +
1361 sizeof(ctx->bytecode->b);
1362 if (bytecode_len > LTTNG_FILTER_MAX_LEN) {
1363 ret = -LTTNG_ERR_FILTER_INVAL;
1364 goto error;
1365 }
1366 }
1367
1368 serialize:
1369 ret = lttng_event_serialize(ev, 0, NULL,
1370 filter_expression, bytecode_len,
1371 (ctx && bytecode_len) ? &ctx->bytecode->b : NULL,
1372 &payload);
1373 if (ret) {
1374 ret = -LTTNG_ERR_INVALID;
1375 goto error;
1376 }
1377
1378 /* If no channel name, send empty string. */
1379 ret = lttng_strncpy(lsm.u.disable.channel_name, channel_name ?: "",
1380 sizeof(lsm.u.disable.channel_name));
1381 if (ret) {
1382 ret = -LTTNG_ERR_INVALID;
1383 goto error;
1384 }
1385
1386 /* Domain */
1387 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1388
1389 /* Session name */
1390 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1391 sizeof(lsm.session.name));
1392 if (ret) {
1393 ret = -LTTNG_ERR_INVALID;
1394 goto error;
1395 }
1396
1397 /* Length of the serialized event. */
1398 lsm.u.disable.length = (uint32_t) payload.buffer.size;
1399
1400 {
1401 struct lttng_payload_view view = lttng_payload_view_from_payload(
1402 &payload, 0, -1);
1403 int fd_count = lttng_payload_view_get_fd_handle_count(&view);
1404 int fd_to_send;
1405
1406 if (fd_count < 0) {
1407 goto error;
1408 }
1409
1410 LTTNG_ASSERT(fd_count == 0 || fd_count == 1);
1411 if (fd_count == 1) {
1412 struct fd_handle *h =
1413 lttng_payload_view_pop_fd_handle(&view);
1414
1415 if (!h) {
1416 goto error;
1417 }
1418
1419 fd_to_send = fd_handle_get_fd(h);
1420 fd_handle_put(h);
1421 }
1422
1423 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm,
1424 fd_count ? &fd_to_send : NULL, fd_count,
1425 view.buffer.size ? view.buffer.data : NULL,
1426 view.buffer.size, NULL, NULL, 0);
1427 }
1428
1429 error:
1430 if (filter_expression && ctx) {
1431 filter_bytecode_free(ctx);
1432 filter_ir_free(ctx);
1433 filter_parser_ctx_free(ctx);
1434 }
1435 if (free_filter_expression) {
1436 /*
1437 * The filter expression has been replaced and must be freed as
1438 * it is not the original filter expression received as a
1439 * parameter.
1440 */
1441 free(filter_expression);
1442 }
1443 /*
1444 * Return directly to the caller and don't ask the sessiond since
1445 * something went wrong in the parsing of data above.
1446 */
1447 lttng_payload_reset(&payload);
1448 return ret;
1449 }
1450
1451 /*
1452 * Disable event(s) of a channel and domain.
1453 * If no event name is specified, all events are disabled.
1454 * If no channel name is specified, the default 'channel0' is used.
1455 * Returns size of returned session payload data or a negative error code.
1456 */
1457 int lttng_disable_event(struct lttng_handle *handle, const char *name,
1458 const char *channel_name)
1459 {
1460 int ret;
1461 struct lttng_event ev;
1462
1463 memset(&ev, 0, sizeof(ev));
1464 ev.loglevel = -1;
1465 ev.type = LTTNG_EVENT_ALL;
1466 ret = lttng_strncpy(ev.name, name ?: "", sizeof(ev.name));
1467 if (ret) {
1468 ret = -LTTNG_ERR_INVALID;
1469 goto end;
1470 }
1471
1472 ret = lttng_disable_event_ext(handle, &ev, channel_name, NULL);
1473 end:
1474 return ret;
1475 }
1476
1477 struct lttng_channel *lttng_channel_create(struct lttng_domain *domain)
1478 {
1479 struct lttng_channel *channel = NULL;
1480
1481 if (!domain) {
1482 goto end;
1483 }
1484
1485 /* Validate domain. */
1486 switch (domain->type) {
1487 case LTTNG_DOMAIN_UST:
1488 switch (domain->buf_type) {
1489 case LTTNG_BUFFER_PER_UID:
1490 case LTTNG_BUFFER_PER_PID:
1491 break;
1492 default:
1493 goto end;
1494 }
1495 break;
1496 case LTTNG_DOMAIN_KERNEL:
1497 if (domain->buf_type != LTTNG_BUFFER_GLOBAL) {
1498 goto end;
1499 }
1500 break;
1501 default:
1502 goto end;
1503 }
1504
1505 channel = lttng_channel_create_internal();
1506 if (!channel) {
1507 goto end;
1508 }
1509
1510 lttng_channel_set_default_attr(domain, &channel->attr);
1511 end:
1512 return channel;
1513 }
1514
1515 void lttng_channel_destroy(struct lttng_channel *channel)
1516 {
1517 if (!channel) {
1518 return;
1519 }
1520
1521 if (channel->attr.extended.ptr) {
1522 free(channel->attr.extended.ptr);
1523 }
1524 free(channel);
1525 }
1526
1527 /*
1528 * Enable channel per domain
1529 * Returns size of returned session payload data or a negative error code.
1530 */
1531 int lttng_enable_channel(struct lttng_handle *handle,
1532 struct lttng_channel *in_chan)
1533 {
1534 enum lttng_error_code ret_code;
1535 int ret;
1536 struct lttng_dynamic_buffer buffer;
1537 struct lttcomm_session_msg lsm;
1538 uint64_t total_buffer_size_needed_per_cpu = 0;
1539 struct lttng_channel *channel = NULL;
1540
1541 lttng_dynamic_buffer_init(&buffer);
1542
1543 /* NULL arguments are forbidden. No default values. */
1544 if (handle == NULL || in_chan == NULL) {
1545 ret = -LTTNG_ERR_INVALID;
1546 goto end;
1547 }
1548
1549 /*
1550 * Verify that the amount of memory required to create the requested
1551 * buffer is available on the system at the moment.
1552 */
1553 if (in_chan->attr.num_subbuf >
1554 UINT64_MAX / in_chan->attr.subbuf_size) {
1555 /* Overflow */
1556 ret = -LTTNG_ERR_OVERFLOW;
1557 goto end;
1558 }
1559
1560 total_buffer_size_needed_per_cpu =
1561 in_chan->attr.num_subbuf * in_chan->attr.subbuf_size;
1562 ret_code = check_enough_available_memory(
1563 total_buffer_size_needed_per_cpu);
1564 if (ret_code != LTTNG_OK) {
1565 ret = -ret_code;
1566 goto end;
1567 }
1568
1569 /* Copy the channel for easier manipulation. */
1570 channel = lttng_channel_copy(in_chan);
1571 if (!channel) {
1572 ret = -LTTNG_ERR_NOMEM;
1573 goto end;
1574 }
1575
1576 /* Populate the channel extended attribute if necessary. */
1577 if (!channel->attr.extended.ptr) {
1578 struct lttng_channel_extended *extended =
1579 zmalloc<lttng_channel_extended>();
1580
1581 if (!extended) {
1582 ret = -LTTNG_ERR_NOMEM;
1583 goto end;
1584 }
1585
1586 lttng_channel_set_default_extended_attr(
1587 &handle->domain, extended);
1588 channel->attr.extended.ptr = extended;
1589 }
1590
1591 /* Prepare the payload */
1592 memset(&lsm, 0, sizeof(lsm));
1593
1594 lsm.cmd_type = LTTNG_ENABLE_CHANNEL;
1595 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1596
1597 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1598 sizeof(lsm.session.name));
1599 if (ret) {
1600 ret = -LTTNG_ERR_INVALID;
1601 goto end;
1602 }
1603
1604 ret = lttng_channel_serialize(channel, &buffer);
1605 if (ret) {
1606 ret = -LTTNG_ERR_FATAL;
1607 goto end;
1608 }
1609
1610 lsm.u.channel.length = buffer.size;
1611
1612 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(
1613 &lsm, buffer.data, buffer.size, NULL);
1614 end:
1615 lttng_channel_destroy(channel);
1616 lttng_dynamic_buffer_reset(&buffer);
1617 return ret;
1618 }
1619
1620 /*
1621 * All tracing will be stopped for registered events of the channel.
1622 * Returns size of returned session payload data or a negative error code.
1623 */
1624 int lttng_disable_channel(struct lttng_handle *handle, const char *name)
1625 {
1626 int ret;
1627 struct lttcomm_session_msg lsm;
1628
1629 /* Safety check. Both are mandatory. */
1630 if (handle == NULL || name == NULL) {
1631 return -LTTNG_ERR_INVALID;
1632 }
1633
1634 memset(&lsm, 0, sizeof(lsm));
1635
1636 lsm.cmd_type = LTTNG_DISABLE_CHANNEL;
1637
1638 ret = lttng_strncpy(lsm.u.disable.channel_name, name,
1639 sizeof(lsm.u.disable.channel_name));
1640 if (ret) {
1641 ret = -LTTNG_ERR_INVALID;
1642 goto end;
1643 }
1644
1645 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1646
1647 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1648 sizeof(lsm.session.name));
1649 if (ret) {
1650 ret = -LTTNG_ERR_INVALID;
1651 goto end;
1652 }
1653
1654 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1655 end:
1656 return ret;
1657 }
1658
1659 /*
1660 * Lists all available tracepoints of domain.
1661 * Sets the contents of the events array.
1662 * Returns the number of lttng_event entries in events;
1663 * on error, returns a negative value.
1664 */
1665 int lttng_list_tracepoints(struct lttng_handle *handle,
1666 struct lttng_event **events)
1667 {
1668 enum lttng_error_code ret_code;
1669 int ret, total_payload_received;
1670 char *reception_buffer = NULL;
1671 struct lttcomm_session_msg lsm = {
1672 .cmd_type = LTTNG_LIST_TRACEPOINTS,
1673 .session = {},
1674 .domain = {},
1675 .u = {},
1676 .fd_count = 0,
1677 };
1678 struct lttcomm_list_command_header *cmd_header = NULL;
1679 size_t cmd_header_len;
1680 unsigned int nb_events = 0;
1681
1682 if (handle == NULL) {
1683 ret = -LTTNG_ERR_INVALID;
1684 goto end;
1685 }
1686
1687 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1688
1689 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm, NULL, 0, NULL, 0,
1690 (void **) &reception_buffer, (void **) &cmd_header,
1691 &cmd_header_len);
1692 if (ret < 0) {
1693 goto end;
1694 }
1695
1696 total_payload_received = ret;
1697
1698 if (!cmd_header) {
1699 ret = -LTTNG_ERR_UNK;
1700 goto end;
1701 }
1702
1703 if (cmd_header->count > INT_MAX) {
1704 ret = -LTTNG_ERR_OVERFLOW;
1705 goto end;
1706 }
1707
1708 nb_events = (unsigned int) cmd_header->count;
1709
1710 {
1711 struct lttng_buffer_view events_view =
1712 lttng_buffer_view_init(reception_buffer, 0,
1713 total_payload_received);
1714 struct lttng_payload_view events_payload_view =
1715 lttng_payload_view_from_buffer_view(
1716 &events_view, 0, -1);
1717
1718 ret_code = lttng_events_create_and_flatten_from_payload(
1719 &events_payload_view, nb_events, events);
1720 if (ret_code != LTTNG_OK) {
1721 ret = -ret_code;
1722 goto end;
1723 }
1724 }
1725
1726 ret = (int) nb_events;
1727
1728 end:
1729 free(cmd_header);
1730 free(reception_buffer);
1731 return ret;
1732 }
1733
1734 /*
1735 * Lists all available tracepoint fields of domain.
1736 * Sets the contents of the event field array.
1737 * Returns the number of lttng_event_field entries in events;
1738 * on error, returns a negative value.
1739 */
1740 int lttng_list_tracepoint_fields(struct lttng_handle *handle,
1741 struct lttng_event_field **fields)
1742 {
1743 enum lttng_error_code ret_code;
1744 int ret;
1745 struct lttcomm_session_msg lsm;
1746 const struct lttcomm_list_command_header *cmd_header = NULL;
1747 unsigned int nb_event_fields = 0;
1748 struct lttng_payload reply;
1749
1750 lttng_payload_init(&reply);
1751
1752 if (handle == NULL) {
1753 ret = -LTTNG_ERR_INVALID;
1754 goto end;
1755 }
1756
1757 memset(&lsm, 0, sizeof(lsm));
1758 lsm.cmd_type = LTTNG_LIST_TRACEPOINT_FIELDS;
1759 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
1760
1761 {
1762 lttng_payload_view message_view =
1763 lttng_payload_view_init_from_buffer(
1764 (const char *) &lsm, 0,
1765 sizeof(lsm));
1766
1767 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
1768 if (ret < 0) {
1769 goto end;
1770 }
1771 }
1772
1773 {
1774 const lttng_buffer_view cmd_header_view =
1775 lttng_buffer_view_from_dynamic_buffer(
1776 &reply.buffer, 0, sizeof(*cmd_header));
1777
1778 if (!lttng_buffer_view_is_valid(&cmd_header_view)) {
1779 ret = -LTTNG_ERR_INVALID_PROTOCOL;
1780 goto end;
1781 }
1782
1783 cmd_header = (struct lttcomm_list_command_header *)
1784 cmd_header_view.data;
1785 }
1786
1787 if (cmd_header->count > INT_MAX) {
1788 ret = -LTTNG_ERR_OVERFLOW;
1789 goto end;
1790 }
1791
1792 nb_event_fields = cmd_header->count;
1793
1794 {
1795 lttng_payload_view reply_view =
1796 lttng_payload_view_from_payload(&reply,
1797 sizeof(*cmd_header), -1);
1798
1799 ret_code = lttng_event_fields_create_and_flatten_from_payload(
1800 &reply_view, nb_event_fields, fields);
1801 if (ret_code != LTTNG_OK) {
1802 ret = -ret_code;
1803 goto end;
1804 }
1805 }
1806
1807 ret = nb_event_fields;
1808
1809 end:
1810 lttng_payload_reset(&reply);
1811 return ret;
1812 }
1813
1814 /*
1815 * Lists all available kernel system calls. Allocates and sets the contents of
1816 * the events array.
1817 *
1818 * Returns the number of lttng_event entries in events; on error, returns a
1819 * negative value.
1820 */
1821 int lttng_list_syscalls(struct lttng_event **events)
1822 {
1823 enum lttng_error_code ret_code;
1824 int ret, total_payload_received;
1825 char *reception_buffer = NULL;
1826 struct lttcomm_session_msg lsm = {};
1827 struct lttcomm_list_command_header *cmd_header = NULL;
1828 size_t cmd_header_len;
1829 uint32_t nb_events = 0;
1830
1831 if (!events) {
1832 ret = -LTTNG_ERR_INVALID;
1833 goto end;
1834 }
1835
1836 lsm.cmd_type = LTTNG_LIST_SYSCALLS;
1837 /* Force kernel domain for system calls. */
1838 lsm.domain.type = LTTNG_DOMAIN_KERNEL;
1839
1840 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm, NULL, 0, NULL, 0,
1841 (void **) &reception_buffer, (void **) &cmd_header,
1842 &cmd_header_len);
1843 if (ret < 0) {
1844 goto end;
1845 }
1846 total_payload_received = ret;
1847
1848 if (!cmd_header) {
1849 ret = -LTTNG_ERR_UNK;
1850 goto end;
1851 }
1852
1853 if (cmd_header->count > INT_MAX) {
1854 ret = -LTTNG_ERR_OVERFLOW;
1855 goto end;
1856 }
1857
1858 nb_events = (unsigned int) cmd_header->count;
1859
1860 {
1861 const struct lttng_buffer_view events_view =
1862 lttng_buffer_view_init(reception_buffer, 0,
1863 total_payload_received);
1864 struct lttng_payload_view events_payload_view =
1865 lttng_payload_view_from_buffer_view(
1866 &events_view, 0, -1);
1867
1868 ret_code = lttng_events_create_and_flatten_from_payload(
1869 &events_payload_view, nb_events, events);
1870 if (ret_code != LTTNG_OK) {
1871 ret = -ret_code;
1872 goto end;
1873 }
1874 }
1875
1876 ret = (int) nb_events;
1877
1878 end:
1879 free(reception_buffer);
1880 free(cmd_header);
1881 return ret;
1882 }
1883
1884 /*
1885 * Returns a human readable string describing
1886 * the error code (positive or negative value).
1887 */
1888 const char *lttng_strerror(int code)
1889 {
1890 if (code > 0) {
1891 code = -code;
1892 }
1893
1894 return error_get_str(code);
1895 }
1896
1897 enum lttng_error_code lttng_create_session_ext(
1898 struct lttng_session_descriptor *session_descriptor)
1899 {
1900 enum lttng_error_code ret_code;
1901 struct lttcomm_session_msg lsm = {
1902 .cmd_type = LTTNG_CREATE_SESSION_EXT,
1903 .session = {},
1904 .domain = {},
1905 .u = {},
1906 .fd_count = 0,
1907 };
1908 void *reply = NULL;
1909 struct lttng_buffer_view reply_view;
1910 int reply_ret;
1911 bool sessiond_must_generate_ouput;
1912 struct lttng_payload payload;
1913 int ret;
1914 size_t descriptor_size;
1915 struct lttng_session_descriptor *descriptor_reply = NULL;
1916
1917 lttng_payload_init(&payload);
1918 if (!session_descriptor) {
1919 ret_code = LTTNG_ERR_INVALID;
1920 goto end;
1921 }
1922
1923 sessiond_must_generate_ouput =
1924 !lttng_session_descriptor_is_output_destination_initialized(
1925 session_descriptor);
1926 if (sessiond_must_generate_ouput) {
1927 const char *home_dir = utils_get_home_dir();
1928 size_t home_dir_len = home_dir ? strlen(home_dir) + 1 : 0;
1929
1930 if (!home_dir || home_dir_len > LTTNG_PATH_MAX) {
1931 ret_code = LTTNG_ERR_FATAL;
1932 goto end;
1933 }
1934
1935 lsm.u.create_session.home_dir_size = (uint16_t) home_dir_len;
1936 ret = lttng_dynamic_buffer_append(&payload.buffer, home_dir, home_dir_len);
1937 if (ret) {
1938 ret_code = LTTNG_ERR_NOMEM;
1939 goto end;
1940 }
1941 }
1942
1943 descriptor_size = payload.buffer.size;
1944 ret = lttng_session_descriptor_serialize(session_descriptor,
1945 &payload);
1946 if (ret) {
1947 ret_code = LTTNG_ERR_INVALID;
1948 goto end;
1949 }
1950 descriptor_size = payload.buffer.size - descriptor_size;
1951 lsm.u.create_session.session_descriptor_size = descriptor_size;
1952
1953 /* Command returns a session descriptor on success. */
1954 reply_ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(
1955 &lsm, payload.buffer.data, payload.buffer.size, &reply);
1956 if (reply_ret < 0) {
1957 ret_code = (lttng_error_code) -reply_ret;
1958 goto end;
1959 } else if (reply_ret == 0) {
1960 /* Socket unexpectedly closed by the session daemon. */
1961 ret_code = LTTNG_ERR_FATAL;
1962 goto end;
1963 }
1964
1965 reply_view = lttng_buffer_view_init((const char *) reply, 0, reply_ret);
1966 ret = lttng_session_descriptor_create_from_buffer(&reply_view,
1967 &descriptor_reply);
1968 if (ret < 0) {
1969 ret_code = LTTNG_ERR_FATAL;
1970 goto end;
1971 }
1972 ret_code = LTTNG_OK;
1973 lttng_session_descriptor_assign(session_descriptor, descriptor_reply);
1974 end:
1975 free(reply);
1976 lttng_payload_reset(&payload);
1977 lttng_session_descriptor_destroy(descriptor_reply);
1978 return ret_code;
1979 }
1980
1981 /*
1982 * Create a new session using name and url for destination.
1983 *
1984 * Return 0 on success else a negative LTTng error code.
1985 */
1986 int lttng_create_session(const char *name, const char *url)
1987 {
1988 int ret;
1989 ssize_t size;
1990 struct lttng_uri *uris = NULL;
1991 struct lttng_session_descriptor *descriptor = NULL;
1992 enum lttng_error_code ret_code;
1993
1994 if (!name) {
1995 ret = -LTTNG_ERR_INVALID;
1996 goto end;
1997 }
1998
1999 size = uri_parse_str_urls(url, NULL, &uris);
2000 if (size < 0) {
2001 ret = -LTTNG_ERR_INVALID;
2002 goto end;
2003 }
2004 switch (size) {
2005 case 0:
2006 descriptor = lttng_session_descriptor_create(name);
2007 break;
2008 case 1:
2009 if (uris[0].dtype != LTTNG_DST_PATH) {
2010 ret = -LTTNG_ERR_INVALID;
2011 goto end;
2012 }
2013 descriptor = lttng_session_descriptor_local_create(name,
2014 uris[0].dst.path);
2015 break;
2016 case 2:
2017 descriptor = lttng_session_descriptor_network_create(name, url,
2018 NULL);
2019 break;
2020 default:
2021 ret = -LTTNG_ERR_INVALID;
2022 goto end;
2023 }
2024 if (!descriptor) {
2025 ret = -LTTNG_ERR_INVALID;
2026 goto end;
2027 }
2028 ret_code = lttng_create_session_ext(descriptor);
2029 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2030 end:
2031 lttng_session_descriptor_destroy(descriptor);
2032 free(uris);
2033 return ret;
2034 }
2035
2036 /*
2037 * Create a session exclusively used for snapshot.
2038 *
2039 * Return 0 on success else a negative LTTng error code.
2040 */
2041 int lttng_create_session_snapshot(const char *name, const char *snapshot_url)
2042 {
2043 int ret;
2044 enum lttng_error_code ret_code;
2045 ssize_t size;
2046 struct lttng_uri *uris = NULL;
2047 struct lttng_session_descriptor *descriptor = NULL;
2048
2049 if (!name) {
2050 ret = -LTTNG_ERR_INVALID;
2051 goto end;
2052 }
2053
2054 size = uri_parse_str_urls(snapshot_url, NULL, &uris);
2055 if (size < 0) {
2056 ret = -LTTNG_ERR_INVALID;
2057 goto end;
2058 }
2059 /*
2060 * If the user does not specify a custom subdir, use the session name.
2061 */
2062 if (size > 0 && uris[0].dtype != LTTNG_DST_PATH &&
2063 strlen(uris[0].subdir) == 0) {
2064 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s",
2065 name);
2066 if (ret < 0) {
2067 PERROR("Failed to set session name as network destination sub-directory");
2068 ret = -LTTNG_ERR_FATAL;
2069 goto end;
2070 } else if (ret >= sizeof(uris[0].subdir)) {
2071 /* Truncated output. */
2072 ret = -LTTNG_ERR_INVALID;
2073 goto end;
2074 }
2075 }
2076
2077 switch (size) {
2078 case 0:
2079 descriptor = lttng_session_descriptor_snapshot_create(name);
2080 break;
2081 case 1:
2082 if (uris[0].dtype != LTTNG_DST_PATH) {
2083 ret = -LTTNG_ERR_INVALID;
2084 goto end;
2085 }
2086 descriptor = lttng_session_descriptor_snapshot_local_create(
2087 name,
2088 uris[0].dst.path);
2089 break;
2090 case 2:
2091 descriptor = lttng_session_descriptor_snapshot_network_create(
2092 name,
2093 snapshot_url,
2094 NULL);
2095 break;
2096 default:
2097 ret = -LTTNG_ERR_INVALID;
2098 goto end;
2099 }
2100 if (!descriptor) {
2101 ret = -LTTNG_ERR_INVALID;
2102 goto end;
2103 }
2104 ret_code = lttng_create_session_ext(descriptor);
2105 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2106 end:
2107 lttng_session_descriptor_destroy(descriptor);
2108 free(uris);
2109 return ret;
2110 }
2111
2112 /*
2113 * Create a session exclusively used for live.
2114 *
2115 * Return 0 on success else a negative LTTng error code.
2116 */
2117 int lttng_create_session_live(const char *name, const char *url,
2118 unsigned int timer_interval)
2119 {
2120 int ret;
2121 enum lttng_error_code ret_code;
2122 struct lttng_session_descriptor *descriptor = NULL;
2123
2124 if (!name) {
2125 ret = -LTTNG_ERR_INVALID;
2126 goto end;
2127 }
2128
2129 if (url) {
2130 descriptor = lttng_session_descriptor_live_network_create(
2131 name, url, NULL, timer_interval);
2132 } else {
2133 descriptor = lttng_session_descriptor_live_create(
2134 name, timer_interval);
2135 }
2136 if (!descriptor) {
2137 ret = -LTTNG_ERR_INVALID;
2138 goto end;
2139 }
2140 ret_code = lttng_create_session_ext(descriptor);
2141 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2142 end:
2143 lttng_session_descriptor_destroy(descriptor);
2144 return ret;
2145 }
2146
2147 /*
2148 * Stop the session and wait for the data before destroying it
2149 *
2150 * Return 0 on success else a negative LTTng error code.
2151 */
2152 int lttng_destroy_session(const char *session_name)
2153 {
2154 int ret;
2155 enum lttng_error_code ret_code;
2156 enum lttng_destruction_handle_status status;
2157 struct lttng_destruction_handle *handle = NULL;
2158
2159 /*
2160 * Stop the tracing and wait for the data to be
2161 * consumed.
2162 */
2163 ret = _lttng_stop_tracing(session_name, 1);
2164 if (ret && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
2165 goto end;
2166 }
2167
2168 ret_code = lttng_destroy_session_ext(session_name, &handle);
2169 if (ret_code != LTTNG_OK) {
2170 ret = (int) -ret_code;
2171 goto end;
2172 }
2173 LTTNG_ASSERT(handle);
2174
2175 /* Block until the completion of the destruction of the session. */
2176 status = lttng_destruction_handle_wait_for_completion(handle, -1);
2177 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_COMPLETED) {
2178 ret = -LTTNG_ERR_UNK;
2179 goto end;
2180 }
2181
2182 status = lttng_destruction_handle_get_result(handle, &ret_code);
2183 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_OK) {
2184 ret = -LTTNG_ERR_UNK;
2185 goto end;
2186 }
2187 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2188 end:
2189 lttng_destruction_handle_destroy(handle);
2190 return ret;
2191 }
2192
2193 /*
2194 * Destroy the session without waiting for the data.
2195 */
2196 int lttng_destroy_session_no_wait(const char *session_name)
2197 {
2198 enum lttng_error_code ret_code;
2199
2200 ret_code = lttng_destroy_session_ext(session_name, NULL);
2201 return ret_code == LTTNG_OK ? 0 : -ret_code;
2202 }
2203
2204 /*
2205 * Ask the session daemon for all available sessions.
2206 * Sets the contents of the sessions array.
2207 * Returns the number of lttng_session entries in sessions;
2208 * on error, returns a negative value.
2209 */
2210 int lttng_list_sessions(struct lttng_session **out_sessions)
2211 {
2212 int ret;
2213 struct lttcomm_session_msg lsm = {};
2214 struct lttng_payload reply;
2215 struct lttng_payload_view lsm_view =
2216 lttng_payload_view_init_from_buffer((const char *) &lsm, 0, sizeof(lsm));
2217 unsigned int nb_sessions = 0;
2218
2219 lttng_payload_init(&reply);
2220
2221 /* Initialize command parameters. */
2222 lsm.cmd_type = LTTNG_LIST_SESSIONS;
2223
2224 /* Execute command against the session daemon. */
2225 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
2226 if (ret < 0) {
2227 goto end;
2228 }
2229
2230 {
2231 const struct lttcomm_list_command_header *cmd_reply_header = NULL;
2232 const lttng_payload_view cmd_reply_header_view = lttng_payload_view_from_payload(
2233 &reply, 0, sizeof(*cmd_reply_header));
2234
2235 if (!lttng_payload_view_is_valid(&cmd_reply_header_view)) {
2236 ret = -LTTNG_ERR_INVALID_PROTOCOL;
2237 goto end;
2238 }
2239
2240 cmd_reply_header = (const struct lttcomm_list_command_header *)
2241 cmd_reply_header_view.buffer.data;
2242 if (cmd_reply_header->count > INT_MAX) {
2243 ret = -LTTNG_ERR_OVERFLOW;
2244 goto end;
2245 }
2246
2247 nb_sessions = (unsigned int) cmd_reply_header->count;
2248 }
2249
2250 {
2251 enum lttng_error_code ret_code;
2252 lttng_payload_view cmd_reply_payload = lttng_payload_view_from_payload(
2253 &reply, sizeof(struct lttcomm_list_command_header), -1);
2254
2255 ret_code = lttng_sessions_create_and_flatten_from_payload(
2256 &cmd_reply_payload, nb_sessions, out_sessions);
2257 if (ret_code != LTTNG_OK) {
2258 ret = -((int) ret_code);
2259 goto end;
2260 }
2261 }
2262
2263 ret = (int) nb_sessions;
2264 end:
2265 lttng_payload_reset(&reply);
2266 return ret;
2267 }
2268
2269 enum lttng_error_code lttng_session_get_creation_time(
2270 const struct lttng_session *session, uint64_t *creation_time)
2271 {
2272 enum lttng_error_code ret = LTTNG_OK;
2273 struct lttng_session_extended *extended;
2274
2275 if (!session || !creation_time || !session->extended.ptr) {
2276 ret = LTTNG_ERR_INVALID;
2277 goto end;
2278 }
2279
2280 extended = (lttng_session_extended *) session->extended.ptr;
2281 if (!extended->creation_time.is_set) {
2282 /* Not created on the session daemon yet. */
2283 ret = LTTNG_ERR_SESSION_NOT_EXIST;
2284 goto end;
2285 }
2286 *creation_time = extended->creation_time.value;
2287 end:
2288 return ret;
2289 }
2290
2291 int lttng_set_session_shm_path(const char *session_name,
2292 const char *shm_path)
2293 {
2294 int ret;
2295 struct lttcomm_session_msg lsm;
2296
2297 if (session_name == NULL) {
2298 return -LTTNG_ERR_INVALID;
2299 }
2300
2301 memset(&lsm, 0, sizeof(lsm));
2302 lsm.cmd_type = LTTNG_SET_SESSION_SHM_PATH;
2303
2304 ret = lttng_strncpy(lsm.session.name, session_name,
2305 sizeof(lsm.session.name));
2306 if (ret) {
2307 ret = -LTTNG_ERR_INVALID;
2308 goto end;
2309 }
2310
2311 ret = lttng_strncpy(lsm.u.set_shm_path.shm_path, shm_path ?: "",
2312 sizeof(lsm.u.set_shm_path.shm_path));
2313 if (ret) {
2314 ret = -LTTNG_ERR_INVALID;
2315 goto end;
2316 }
2317
2318 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
2319 end:
2320 return ret;
2321 }
2322
2323 /*
2324 * Ask the session daemon for all available domains of a session.
2325 * Sets the contents of the domains array.
2326 * Returns the number of lttng_domain entries in domains;
2327 * on error, returns a negative value.
2328 */
2329 int lttng_list_domains(const char *session_name,
2330 struct lttng_domain **domains)
2331 {
2332 int ret;
2333 struct lttcomm_session_msg lsm;
2334
2335 if (session_name == NULL) {
2336 ret = -LTTNG_ERR_INVALID;
2337 goto error;
2338 }
2339
2340 memset(&lsm, 0, sizeof(lsm));
2341 lsm.cmd_type = LTTNG_LIST_DOMAINS;
2342
2343 ret = lttng_strncpy(lsm.session.name, session_name,
2344 sizeof(lsm.session.name));
2345 if (ret) {
2346 ret = -LTTNG_ERR_INVALID;
2347 goto error;
2348 }
2349
2350 ret = lttng_ctl_ask_sessiond(&lsm, (void**) domains);
2351 if (ret < 0) {
2352 goto error;
2353 }
2354
2355 return ret / sizeof(struct lttng_domain);
2356 error:
2357 return ret;
2358 }
2359
2360 /*
2361 * Ask the session daemon for all available channels of a session.
2362 * Sets the contents of the channels array.
2363 * Returns the number of lttng_channel entries in channels;
2364 * on error, returns a negative value.
2365 */
2366 int lttng_list_channels(struct lttng_handle *handle,
2367 struct lttng_channel **channels)
2368 {
2369 int ret, total_payload_received;
2370 struct lttcomm_session_msg lsm;
2371 char *reception_buffer = NULL;
2372 size_t cmd_header_len = 0;
2373 struct lttcomm_list_command_header *cmd_header = NULL;
2374 struct lttng_dynamic_buffer tmp_buffer;
2375
2376 lttng_dynamic_buffer_init(&tmp_buffer);
2377
2378 if (handle == NULL) {
2379 ret = -LTTNG_ERR_INVALID;
2380 goto end;
2381 }
2382
2383 memset(&lsm, 0, sizeof(lsm));
2384 lsm.cmd_type = LTTNG_LIST_CHANNELS;
2385 ret = lttng_strncpy(lsm.session.name, handle->session_name,
2386 sizeof(lsm.session.name));
2387 if (ret) {
2388 ret = -LTTNG_ERR_INVALID;
2389 goto end;
2390 }
2391
2392 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
2393
2394 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm, NULL, 0, NULL, 0,
2395 (void **) &reception_buffer, (void **) &cmd_header,
2396 &cmd_header_len);
2397 if (ret < 0) {
2398 goto end;
2399 }
2400
2401 total_payload_received = ret;
2402
2403 if (cmd_header_len != sizeof(*cmd_header)) {
2404 ret = -LTTNG_ERR_FATAL;
2405 goto end;
2406 }
2407
2408 if (!cmd_header) {
2409 ret = LTTNG_ERR_UNK;
2410 goto end;
2411 }
2412
2413 if (cmd_header->count > INT_MAX) {
2414 ret = -LTTNG_ERR_OVERFLOW;
2415 goto end;
2416 }
2417
2418 {
2419 enum lttng_error_code ret_code;
2420 const struct lttng_buffer_view events_view =
2421 lttng_buffer_view_init(reception_buffer, 0,
2422 total_payload_received);
2423
2424 ret_code = lttng_channels_create_and_flatten_from_buffer(
2425 &events_view, cmd_header->count, channels);
2426 if (ret_code != LTTNG_OK) {
2427 ret = -ret_code;
2428 goto end;
2429 }
2430 }
2431
2432 ret = (int) cmd_header->count;
2433 end:
2434 free(cmd_header);
2435 free(reception_buffer);
2436 return ret;
2437 }
2438
2439 /*
2440 * Ask the session daemon for all available events of a session channel.
2441 * Sets the contents of the events array.
2442 * Returns the number of lttng_event entries in events;
2443 * on error, returns a negative value.
2444 */
2445 int lttng_list_events(struct lttng_handle *handle,
2446 const char *channel_name, struct lttng_event **events)
2447 {
2448 int ret;
2449 struct lttcomm_session_msg lsm = {};
2450 struct lttng_payload reply;
2451 struct lttng_payload_view lsm_view =
2452 lttng_payload_view_init_from_buffer(
2453 (const char *) &lsm, 0, sizeof(lsm));
2454 unsigned int nb_events = 0;
2455
2456 lttng_payload_init(&reply);
2457
2458 /* Safety check. An handle and channel name are mandatory. */
2459 if (handle == NULL || channel_name == NULL) {
2460 ret = -LTTNG_ERR_INVALID;
2461 goto end;
2462 }
2463
2464 /* Initialize command parameters. */
2465 lsm.cmd_type = LTTNG_LIST_EVENTS;
2466 ret = lttng_strncpy(lsm.session.name, handle->session_name,
2467 sizeof(lsm.session.name));
2468 if (ret) {
2469 ret = -LTTNG_ERR_INVALID;
2470 goto end;
2471 }
2472
2473 ret = lttng_strncpy(lsm.u.list.channel_name, channel_name,
2474 sizeof(lsm.u.list.channel_name));
2475 if (ret) {
2476 ret = -LTTNG_ERR_INVALID;
2477 goto end;
2478 }
2479
2480 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
2481
2482 /* Execute command against the session daemon. */
2483 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
2484 if (ret < 0) {
2485 goto end;
2486 }
2487
2488 {
2489 const struct lttcomm_list_command_header *cmd_reply_header =
2490 NULL;
2491 const lttng_payload_view cmd_reply_header_view =
2492 lttng_payload_view_from_payload(&reply, 0,
2493 sizeof(*cmd_reply_header));
2494
2495 if (!lttng_payload_view_is_valid(&cmd_reply_header_view)) {
2496 ret = -LTTNG_ERR_INVALID_PROTOCOL;
2497 goto end;
2498 }
2499
2500 cmd_reply_header = (const struct lttcomm_list_command_header *)
2501 cmd_reply_header_view.buffer
2502 .data;
2503 if (cmd_reply_header->count > INT_MAX) {
2504 ret = -LTTNG_ERR_OVERFLOW;
2505 goto end;
2506 }
2507
2508 nb_events = (unsigned int) cmd_reply_header->count;
2509 }
2510
2511 {
2512 enum lttng_error_code ret_code;
2513 lttng_payload_view cmd_reply_payload = lttng_payload_view_from_payload(
2514 &reply,
2515 sizeof(struct lttcomm_list_command_header), -1);
2516
2517 ret_code = lttng_events_create_and_flatten_from_payload(
2518 &cmd_reply_payload, nb_events, events);
2519 if (ret_code != LTTNG_OK) {
2520 ret = -((int) ret_code);
2521 goto end;
2522 }
2523 }
2524
2525 ret = (int) nb_events;
2526 end:
2527 lttng_payload_reset(&reply);
2528 return ret;
2529 }
2530
2531 /*
2532 * Sets the tracing_group variable with name.
2533 * This function allocates memory pointed to by tracing_group.
2534 * On success, returns 0, on error, returns -1 (null name) or -ENOMEM.
2535 */
2536 int lttng_set_tracing_group(const char *name)
2537 {
2538 int ret = 0;
2539 char *new_group;
2540
2541 if (name == NULL) {
2542 ret = -LTTNG_ERR_INVALID;
2543 goto end;
2544 }
2545
2546 new_group = strdup(name);
2547 if (!new_group) {
2548 ret = -LTTNG_ERR_FATAL;
2549 goto end;
2550 }
2551
2552 free(tracing_group);
2553 tracing_group = new_group;
2554 new_group = NULL;
2555
2556 end:
2557 return ret;
2558 }
2559
2560 int lttng_calibrate(struct lttng_handle *handle __attribute__((unused)),
2561 struct lttng_calibrate *calibrate __attribute__((unused)))
2562 {
2563 /*
2564 * This command was removed in LTTng 2.9.
2565 */
2566 return -LTTNG_ERR_UND;
2567 }
2568
2569 /*
2570 * Set default channel attributes.
2571 * If either or both of the arguments are null, attr content is zeroe'd.
2572 */
2573 void lttng_channel_set_default_attr(struct lttng_domain *domain,
2574 struct lttng_channel_attr *attr)
2575 {
2576 struct lttng_channel_extended *extended;
2577
2578 /* Safety check */
2579 if (attr == NULL || domain == NULL) {
2580 return;
2581 }
2582
2583 /* Save the pointer for later use */
2584 extended = (struct lttng_channel_extended *) attr->extended.ptr;
2585 memset(attr, 0, sizeof(struct lttng_channel_attr));
2586
2587 /* Same for all domains. */
2588 attr->overwrite = DEFAULT_CHANNEL_OVERWRITE;
2589 attr->tracefile_size = DEFAULT_CHANNEL_TRACEFILE_SIZE;
2590 attr->tracefile_count = DEFAULT_CHANNEL_TRACEFILE_COUNT;
2591
2592 switch (domain->type) {
2593 case LTTNG_DOMAIN_KERNEL:
2594 attr->switch_timer_interval =
2595 DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER;
2596 attr->read_timer_interval = DEFAULT_KERNEL_CHANNEL_READ_TIMER;
2597 attr->subbuf_size = default_get_kernel_channel_subbuf_size();
2598 attr->num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
2599 attr->output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
2600 break;
2601 case LTTNG_DOMAIN_UST:
2602 switch (domain->buf_type) {
2603 case LTTNG_BUFFER_PER_UID:
2604 attr->subbuf_size = default_get_ust_uid_channel_subbuf_size();
2605 attr->num_subbuf = DEFAULT_UST_UID_CHANNEL_SUBBUF_NUM;
2606 attr->output = DEFAULT_UST_UID_CHANNEL_OUTPUT;
2607 attr->switch_timer_interval =
2608 DEFAULT_UST_UID_CHANNEL_SWITCH_TIMER;
2609 attr->read_timer_interval =
2610 DEFAULT_UST_UID_CHANNEL_READ_TIMER;
2611 break;
2612 case LTTNG_BUFFER_PER_PID:
2613 default:
2614 attr->subbuf_size = default_get_ust_pid_channel_subbuf_size();
2615 attr->num_subbuf = DEFAULT_UST_PID_CHANNEL_SUBBUF_NUM;
2616 attr->output = DEFAULT_UST_PID_CHANNEL_OUTPUT;
2617 attr->switch_timer_interval =
2618 DEFAULT_UST_PID_CHANNEL_SWITCH_TIMER;
2619 attr->read_timer_interval =
2620 DEFAULT_UST_PID_CHANNEL_READ_TIMER;
2621 break;
2622 }
2623 default:
2624 /* Default behavior: leave set to 0. */
2625 break;
2626 }
2627
2628 if (extended) {
2629 lttng_channel_set_default_extended_attr(domain, extended);
2630 }
2631
2632 /* Reassign the extended pointer. */
2633 attr->extended.ptr = extended;
2634 }
2635
2636 int lttng_channel_get_discarded_event_count(struct lttng_channel *channel,
2637 uint64_t *discarded_events)
2638 {
2639 int ret = 0;
2640 struct lttng_channel_extended *chan_ext;
2641
2642 if (!channel || !discarded_events) {
2643 ret = -LTTNG_ERR_INVALID;
2644 goto end;
2645 }
2646
2647 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
2648 if (!chan_ext) {
2649 /*
2650 * This can happen since the lttng_channel structure is
2651 * used for other tasks where this pointer is never set.
2652 */
2653 *discarded_events = 0;
2654 goto end;
2655 }
2656
2657 *discarded_events = chan_ext->discarded_events;
2658 end:
2659 return ret;
2660 }
2661
2662 int lttng_channel_get_lost_packet_count(struct lttng_channel *channel,
2663 uint64_t *lost_packets)
2664 {
2665 int ret = 0;
2666 struct lttng_channel_extended *chan_ext;
2667
2668 if (!channel || !lost_packets) {
2669 ret = -LTTNG_ERR_INVALID;
2670 goto end;
2671 }
2672
2673 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
2674 if (!chan_ext) {
2675 /*
2676 * This can happen since the lttng_channel structure is
2677 * used for other tasks where this pointer is never set.
2678 */
2679 *lost_packets = 0;
2680 goto end;
2681 }
2682
2683 *lost_packets = chan_ext->lost_packets;
2684 end:
2685 return ret;
2686 }
2687
2688 int lttng_channel_get_monitor_timer_interval(struct lttng_channel *chan,
2689 uint64_t *monitor_timer_interval)
2690 {
2691 int ret = 0;
2692
2693 if (!chan || !monitor_timer_interval) {
2694 ret = -LTTNG_ERR_INVALID;
2695 goto end;
2696 }
2697
2698 if (!chan->attr.extended.ptr) {
2699 ret = -LTTNG_ERR_INVALID;
2700 goto end;
2701 }
2702
2703 *monitor_timer_interval = ((struct lttng_channel_extended *)
2704 chan->attr.extended.ptr)->monitor_timer_interval;
2705 end:
2706 return ret;
2707 }
2708
2709 int lttng_channel_set_monitor_timer_interval(struct lttng_channel *chan,
2710 uint64_t monitor_timer_interval)
2711 {
2712 int ret = 0;
2713
2714 if (!chan || !chan->attr.extended.ptr) {
2715 ret = -LTTNG_ERR_INVALID;
2716 goto end;
2717 }
2718
2719 ((struct lttng_channel_extended *)
2720 chan->attr.extended.ptr)->monitor_timer_interval =
2721 monitor_timer_interval;
2722 end:
2723 return ret;
2724 }
2725
2726 int lttng_channel_get_blocking_timeout(struct lttng_channel *chan,
2727 int64_t *blocking_timeout)
2728 {
2729 int ret = 0;
2730
2731 if (!chan || !blocking_timeout) {
2732 ret = -LTTNG_ERR_INVALID;
2733 goto end;
2734 }
2735
2736 if (!chan->attr.extended.ptr) {
2737 ret = -LTTNG_ERR_INVALID;
2738 goto end;
2739 }
2740
2741 *blocking_timeout = ((struct lttng_channel_extended *)
2742 chan->attr.extended.ptr)->blocking_timeout;
2743 end:
2744 return ret;
2745 }
2746
2747 int lttng_channel_set_blocking_timeout(struct lttng_channel *chan,
2748 int64_t blocking_timeout)
2749 {
2750 int ret = 0;
2751 int64_t msec_timeout;
2752
2753 if (!chan || !chan->attr.extended.ptr) {
2754 ret = -LTTNG_ERR_INVALID;
2755 goto end;
2756 }
2757
2758 if (blocking_timeout < 0 && blocking_timeout != -1) {
2759 ret = -LTTNG_ERR_INVALID;
2760 goto end;
2761 }
2762
2763 /*
2764 * LTTng-ust's use of poll() to implement this timeout mechanism forces
2765 * us to accept a narrower range of values (msecs expressed as a signed
2766 * 32-bit integer).
2767 */
2768 msec_timeout = blocking_timeout / 1000;
2769 if (msec_timeout != (int32_t) msec_timeout) {
2770 ret = -LTTNG_ERR_INVALID;
2771 goto end;
2772 }
2773
2774 ((struct lttng_channel_extended *)
2775 chan->attr.extended.ptr)->blocking_timeout =
2776 blocking_timeout;
2777 end:
2778 return ret;
2779 }
2780
2781 /*
2782 * Check if session daemon is alive.
2783 *
2784 * Return 1 if alive or 0 if not.
2785 * On error returns a negative value.
2786 */
2787 int lttng_session_daemon_alive(void)
2788 {
2789 int ret;
2790
2791 ret = set_session_daemon_path();
2792 if (ret < 0) {
2793 /* Error. */
2794 return ret;
2795 }
2796
2797 if (*sessiond_sock_path == '\0') {
2798 /*
2799 * No socket path set. Weird error which means the constructor
2800 * was not called.
2801 */
2802 abort();
2803 }
2804
2805 ret = try_connect_sessiond(sessiond_sock_path);
2806 if (ret < 0) {
2807 /* Not alive. */
2808 return 0;
2809 }
2810
2811 /* Is alive. */
2812 return 1;
2813 }
2814
2815 /*
2816 * Set URL for a consumer for a session and domain.
2817 *
2818 * Return 0 on success, else a negative value.
2819 */
2820 int lttng_set_consumer_url(struct lttng_handle *handle,
2821 const char *control_url, const char *data_url)
2822 {
2823 int ret;
2824 ssize_t size;
2825 struct lttcomm_session_msg lsm;
2826 struct lttng_uri *uris = NULL;
2827
2828 if (handle == NULL || (control_url == NULL && data_url == NULL)) {
2829 ret = -LTTNG_ERR_INVALID;
2830 goto error;
2831 }
2832
2833 memset(&lsm, 0, sizeof(lsm));
2834
2835 lsm.cmd_type = LTTNG_SET_CONSUMER_URI;
2836
2837 ret = lttng_strncpy(lsm.session.name, handle->session_name,
2838 sizeof(lsm.session.name));
2839 if (ret) {
2840 ret = -LTTNG_ERR_INVALID;
2841 goto error;
2842 }
2843
2844 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
2845
2846 size = uri_parse_str_urls(control_url, data_url, &uris);
2847 if (size < 0) {
2848 ret = -LTTNG_ERR_INVALID;
2849 goto error;
2850 }
2851
2852 lsm.u.uri.size = size;
2853
2854 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, uris,
2855 sizeof(struct lttng_uri) * size, NULL);
2856
2857 free(uris);
2858 error:
2859 return ret;
2860 }
2861
2862 /*
2863 * [OBSOLETE]
2864 */
2865 extern "C"
2866 LTTNG_EXPORT int lttng_enable_consumer(struct lttng_handle *handle);
2867 int lttng_enable_consumer(struct lttng_handle *handle __attribute__((unused)))
2868 {
2869 return -ENOSYS;
2870 }
2871
2872 /*
2873 * [OBSOLETE]
2874 */
2875 extern "C"
2876 LTTNG_EXPORT int lttng_disable_consumer(struct lttng_handle *handle);
2877 int lttng_disable_consumer(struct lttng_handle *handle __attribute__((unused)))
2878 {
2879 return -ENOSYS;
2880 }
2881
2882 /*
2883 * [OBSOLETE]
2884 */
2885 extern "C"
2886 LTTNG_EXPORT int _lttng_create_session_ext(const char *name, const char *url,
2887 const char *datetime);
2888 int _lttng_create_session_ext(const char *name __attribute__((unused)),
2889 const char *url __attribute__((unused)),
2890 const char *datetime __attribute__((unused)))
2891 {
2892 return -ENOSYS;
2893 }
2894
2895 /*
2896 * For a given session name, this call checks if the data is ready to be read
2897 * or is still being extracted by the consumer(s) hence not ready to be used by
2898 * any readers.
2899 */
2900 int lttng_data_pending(const char *session_name)
2901 {
2902 int ret;
2903 struct lttcomm_session_msg lsm;
2904 uint8_t *pending = NULL;
2905
2906 if (session_name == NULL) {
2907 return -LTTNG_ERR_INVALID;
2908 }
2909
2910 memset(&lsm, 0, sizeof(lsm));
2911 lsm.cmd_type = LTTNG_DATA_PENDING;
2912
2913 ret = lttng_strncpy(lsm.session.name, session_name,
2914 sizeof(lsm.session.name));
2915 if (ret) {
2916 ret = -LTTNG_ERR_INVALID;
2917 goto end;
2918 }
2919
2920 ret = lttng_ctl_ask_sessiond(&lsm, (void **) &pending);
2921 if (ret < 0) {
2922 goto end;
2923 } else if (ret != 1) {
2924 /* Unexpected payload size */
2925 ret = -LTTNG_ERR_INVALID;
2926 goto end;
2927 } else if (!pending) {
2928 /* Internal error. */
2929 ret = -LTTNG_ERR_UNK;
2930 goto end;
2931 }
2932
2933 ret = (int) *pending;
2934 end:
2935 free(pending);
2936 return ret;
2937 }
2938
2939 /*
2940 * Regenerate the metadata for a session.
2941 * Return 0 on success, a negative error code on error.
2942 */
2943 int lttng_regenerate_metadata(const char *session_name)
2944 {
2945 int ret;
2946 struct lttcomm_session_msg lsm;
2947
2948 if (!session_name) {
2949 ret = -LTTNG_ERR_INVALID;
2950 goto end;
2951 }
2952
2953 memset(&lsm, 0, sizeof(lsm));
2954 lsm.cmd_type = LTTNG_REGENERATE_METADATA;
2955
2956 ret = lttng_strncpy(lsm.session.name, session_name,
2957 sizeof(lsm.session.name));
2958 if (ret) {
2959 ret = -LTTNG_ERR_INVALID;
2960 goto end;
2961 }
2962
2963 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
2964 if (ret < 0) {
2965 goto end;
2966 }
2967
2968 ret = 0;
2969 end:
2970 return ret;
2971 }
2972
2973 /*
2974 * Deprecated, replaced by lttng_regenerate_metadata.
2975 */
2976 int lttng_metadata_regenerate(const char *session_name)
2977 {
2978 return lttng_regenerate_metadata(session_name);
2979 }
2980
2981 /*
2982 * Regenerate the statedump of a session.
2983 * Return 0 on success, a negative error code on error.
2984 */
2985 int lttng_regenerate_statedump(const char *session_name)
2986 {
2987 int ret;
2988 struct lttcomm_session_msg lsm;
2989
2990 if (!session_name) {
2991 ret = -LTTNG_ERR_INVALID;
2992 goto end;
2993 }
2994
2995 memset(&lsm, 0, sizeof(lsm));
2996 lsm.cmd_type = LTTNG_REGENERATE_STATEDUMP;
2997
2998 ret = lttng_strncpy(lsm.session.name, session_name,
2999 sizeof(lsm.session.name));
3000 if (ret) {
3001 ret = -LTTNG_ERR_INVALID;
3002 goto end;
3003 }
3004
3005 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
3006 if (ret < 0) {
3007 goto end;
3008 }
3009
3010 ret = 0;
3011 end:
3012 return ret;
3013 }
3014
3015 static
3016 int _lttng_register_trigger(struct lttng_trigger *trigger, const char *name,
3017 bool generate_name)
3018 {
3019 int ret;
3020 struct lttcomm_session_msg lsm = {
3021 .cmd_type = LTTNG_REGISTER_TRIGGER,
3022 .session = {},
3023 .domain = {},
3024 .u = {},
3025 .fd_count = 0,
3026 };
3027 lsm.u.trigger.is_trigger_anonymous = !name && !generate_name;
3028 struct lttcomm_session_msg *message_lsm;
3029 struct lttng_payload message;
3030 struct lttng_payload reply;
3031 struct lttng_trigger *reply_trigger = NULL;
3032 enum lttng_domain_type domain_type;
3033 const struct lttng_credentials user_creds = {
3034 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3035 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3036 };
3037 const char *unused_trigger_name = NULL;
3038 enum lttng_trigger_status trigger_status;
3039
3040 lttng_payload_init(&message);
3041 lttng_payload_init(&reply);
3042
3043 if (!trigger) {
3044 ret = -LTTNG_ERR_INVALID;
3045 goto end;
3046 }
3047
3048 trigger_status = lttng_trigger_get_name(trigger, &unused_trigger_name);
3049 if (trigger_status != LTTNG_TRIGGER_STATUS_UNSET) {
3050 /* Re-using already registered trigger. */
3051 ret = -LTTNG_ERR_INVALID;
3052 goto end;
3053 }
3054
3055 if (name) {
3056 trigger_status = lttng_trigger_set_name(trigger, name);
3057 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3058 ret = -LTTNG_ERR_NOMEM;
3059 goto end;
3060 }
3061 }
3062
3063 if (!trigger->creds.uid.is_set) {
3064 /* Use the client's credentials as the trigger credentials. */
3065 lttng_trigger_set_credentials(trigger, &user_creds);
3066 } else {
3067 /*
3068 * Validate that either the current trigger credentials and the
3069 * client credentials are identical or that the current user is
3070 * root. The root user can register, unregister triggers for
3071 * himself and other users.
3072 *
3073 * This check is also present on the sessiond side, using the
3074 * credentials passed on the socket. These check are all
3075 * "safety" checks.
3076 */
3077 const struct lttng_credentials *trigger_creds =
3078 lttng_trigger_get_credentials(trigger);
3079
3080 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3081 if (lttng_credentials_get_uid(&user_creds) != 0) {
3082 ret = -LTTNG_ERR_EPERM;
3083 goto end_unset_name;
3084 }
3085 }
3086 }
3087
3088 if (!lttng_trigger_validate(trigger)) {
3089 ret = -LTTNG_ERR_INVALID_TRIGGER;
3090 goto end_unset_name;
3091 }
3092
3093 domain_type = lttng_trigger_get_underlying_domain_type_restriction(
3094 trigger);
3095
3096 lsm.domain.type = domain_type;
3097
3098 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3099 if (ret) {
3100 ret = -LTTNG_ERR_NOMEM;
3101 goto end_unset_name;
3102 }
3103
3104 ret = lttng_trigger_serialize(trigger, &message);
3105 if (ret < 0) {
3106 ret = -LTTNG_ERR_UNK;
3107 goto end_unset_name;
3108 }
3109
3110 /*
3111 * This is needed to populate the trigger object size for the command
3112 * header.
3113 */
3114 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3115
3116 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3117
3118 {
3119 struct lttng_payload_view message_view =
3120 lttng_payload_view_from_payload(
3121 &message, 0, -1);
3122
3123 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3124 &message_view);
3125 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3126 if (ret < 0) {
3127 goto end_unset_name;
3128 }
3129 }
3130
3131 {
3132 struct lttng_payload_view reply_view =
3133 lttng_payload_view_from_payload(
3134 &reply, 0, reply.buffer.size);
3135
3136 ret = lttng_trigger_create_from_payload(
3137 &reply_view, &reply_trigger);
3138 if (ret < 0) {
3139 ret = -LTTNG_ERR_INVALID_PROTOCOL;
3140 goto end_unset_name;
3141 }
3142 }
3143
3144 if (name || generate_name) {
3145 ret = lttng_trigger_assign_name(trigger, reply_trigger);
3146 if (ret < 0) {
3147 ret = -LTTNG_ERR_NOMEM;
3148 goto end;
3149 }
3150 }
3151
3152 ret = 0;
3153 goto end;
3154
3155 end_unset_name:
3156 trigger_status = lttng_trigger_set_name(trigger, NULL);
3157 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3158 ret = -LTTNG_ERR_UNK;
3159 }
3160 end:
3161 lttng_payload_reset(&message);
3162 lttng_payload_reset(&reply);
3163 lttng_trigger_destroy(reply_trigger);
3164 return ret;
3165 }
3166
3167 int lttng_register_trigger(struct lttng_trigger *trigger)
3168 {
3169 /* Register an anonymous trigger. */
3170 return _lttng_register_trigger(trigger, NULL, false);
3171 }
3172
3173 enum lttng_error_code lttng_register_trigger_with_name(
3174 struct lttng_trigger *trigger, const char *name)
3175 {
3176 const int ret = _lttng_register_trigger(trigger, name, false);
3177
3178 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3179 }
3180
3181 enum lttng_error_code lttng_register_trigger_with_automatic_name(
3182 struct lttng_trigger *trigger)
3183 {
3184 const int ret = _lttng_register_trigger(trigger, nullptr, true);
3185
3186 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3187 }
3188
3189 enum lttng_error_code lttng_error_query_execute(
3190 const struct lttng_error_query *query,
3191 const struct lttng_endpoint *endpoint,
3192 struct lttng_error_query_results **results)
3193 {
3194 int ret;
3195 enum lttng_error_code ret_code;
3196 struct lttcomm_session_msg lsm = {
3197 .cmd_type = LTTNG_EXECUTE_ERROR_QUERY,
3198 .session = {},
3199 .domain = {},
3200 .u = {},
3201 .fd_count = 0,
3202 };
3203 struct lttng_payload message;
3204 struct lttng_payload reply;
3205 struct lttcomm_session_msg *message_lsm;
3206
3207 lttng_payload_init(&message);
3208 lttng_payload_init(&reply);
3209
3210 if (!query || !results) {
3211 ret_code = LTTNG_ERR_INVALID;
3212 goto end;
3213 }
3214
3215 if (endpoint != lttng_session_daemon_command_endpoint) {
3216 ret_code = LTTNG_ERR_INVALID_ERROR_QUERY_TARGET;
3217 goto end;
3218 }
3219
3220 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3221 if (ret) {
3222 ret_code = LTTNG_ERR_NOMEM;
3223 goto end;
3224 }
3225
3226 ret = lttng_error_query_serialize(query, &message);
3227 if (ret) {
3228 ret_code = LTTNG_ERR_UNK;
3229 goto end;
3230 }
3231
3232 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3233 message_lsm->u.error_query.length =
3234 (uint32_t) message.buffer.size - sizeof(lsm);
3235
3236 {
3237 struct lttng_payload_view message_view =
3238 lttng_payload_view_from_payload(
3239 &message, 0, -1);
3240
3241 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3242 &message_view);
3243 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3244 if (ret < 0) {
3245 ret_code =(lttng_error_code) -ret;
3246 goto end;
3247 }
3248 }
3249
3250 {
3251 ssize_t reply_create_ret;
3252 struct lttng_payload_view reply_view =
3253 lttng_payload_view_from_payload(
3254 &reply, 0, reply.buffer.size);
3255
3256 reply_create_ret = lttng_error_query_results_create_from_payload(
3257 &reply_view, results);
3258 if (reply_create_ret < 0) {
3259 ret_code = LTTNG_ERR_INVALID_PROTOCOL;
3260 goto end;
3261 }
3262 }
3263
3264 ret_code = LTTNG_OK;
3265 end:
3266 lttng_payload_reset(&message);
3267 lttng_payload_reset(&reply);
3268 return ret_code;
3269 }
3270
3271 int lttng_unregister_trigger(const struct lttng_trigger *trigger)
3272 {
3273 int ret;
3274 struct lttcomm_session_msg lsm;
3275 struct lttcomm_session_msg *message_lsm;
3276 struct lttng_payload message;
3277 struct lttng_payload reply;
3278 struct lttng_trigger *copy = NULL;
3279 const struct lttng_credentials user_creds = {
3280 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3281 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3282 };
3283
3284 lttng_payload_init(&message);
3285 lttng_payload_init(&reply);
3286
3287 if (!trigger) {
3288 ret = -LTTNG_ERR_INVALID;
3289 goto end;
3290 }
3291
3292 copy = lttng_trigger_copy(trigger);
3293 if (!copy) {
3294 ret = -LTTNG_ERR_UNK;
3295 goto end;
3296 }
3297
3298 if (!copy->creds.uid.is_set) {
3299 /* Use the client credentials as the trigger credentials */
3300 lttng_trigger_set_credentials(copy, &user_creds);
3301 } else {
3302 /*
3303 * Validate that either the current trigger credentials and the
3304 * client credentials are identical or that the current user is
3305 * root. The root user can register, unregister triggers for
3306 * himself and other users.
3307 *
3308 * This check is also present on the sessiond side, using the
3309 * credentials passed on the socket. These check are all
3310 * "safety" checks.
3311 */
3312 const struct lttng_credentials *trigger_creds =
3313 lttng_trigger_get_credentials(copy);
3314 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3315 if (lttng_credentials_get_uid(&user_creds) != 0) {
3316 ret = -LTTNG_ERR_EPERM;
3317 goto end;
3318 }
3319 }
3320 }
3321
3322 if (!lttng_trigger_validate(copy)) {
3323 ret = -LTTNG_ERR_INVALID_TRIGGER;
3324 goto end;
3325 }
3326
3327 memset(&lsm, 0, sizeof(lsm));
3328 lsm.cmd_type = LTTNG_UNREGISTER_TRIGGER;
3329
3330 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3331 if (ret) {
3332 ret = -LTTNG_ERR_NOMEM;
3333 goto end;
3334 }
3335
3336 ret = lttng_trigger_serialize(copy, &message);
3337 if (ret < 0) {
3338 ret = -LTTNG_ERR_UNK;
3339 goto end;
3340 }
3341
3342 /*
3343 * This is needed to populate the trigger object size for the command
3344 * header and number of fds sent.
3345 */
3346 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3347
3348 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3349
3350 {
3351 struct lttng_payload_view message_view =
3352 lttng_payload_view_from_payload(
3353 &message, 0, -1);
3354
3355 /*
3356 * Update the message header with the number of fd that will be
3357 * sent.
3358 */
3359 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3360 &message_view);
3361
3362 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3363 if (ret < 0) {
3364 goto end;
3365 }
3366 }
3367
3368 ret = 0;
3369 end:
3370 lttng_trigger_destroy(copy);
3371 lttng_payload_reset(&message);
3372 lttng_payload_reset(&reply);
3373 return ret;
3374 }
3375
3376 /*
3377 * Ask the session daemon for all registered triggers for the current user.
3378 *
3379 * Allocates and return an lttng_triggers set.
3380 * On error, returns a suitable lttng_error_code.
3381 */
3382 enum lttng_error_code lttng_list_triggers(struct lttng_triggers **triggers)
3383 {
3384 int ret;
3385 enum lttng_error_code ret_code = LTTNG_OK;
3386 struct lttcomm_session_msg lsm = {
3387 .cmd_type = LTTNG_LIST_TRIGGERS,
3388 .session = {},
3389 .domain = {},
3390 .u = {},
3391 .fd_count = 0,
3392 };
3393 struct lttng_triggers *local_triggers = NULL;
3394 struct lttng_payload reply;
3395 struct lttng_payload_view lsm_view =
3396 lttng_payload_view_init_from_buffer(
3397 (const char *) &lsm, 0, sizeof(lsm));
3398
3399 lttng_payload_init(&reply);
3400
3401 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
3402 if (ret < 0) {
3403 ret_code = (enum lttng_error_code) -ret;
3404 goto end;
3405 }
3406
3407 {
3408 struct lttng_payload_view reply_view =
3409 lttng_payload_view_from_payload(
3410 &reply, 0, reply.buffer.size);
3411
3412 ret = lttng_triggers_create_from_payload(
3413 &reply_view, &local_triggers);
3414 if (ret < 0) {
3415 ret_code = LTTNG_ERR_FATAL;
3416 goto end;
3417 }
3418 }
3419
3420 *triggers = local_triggers;
3421 local_triggers = NULL;
3422 end:
3423 lttng_payload_reset(&reply);
3424 lttng_triggers_destroy(local_triggers);
3425 return ret_code;
3426 }
3427
3428 /*
3429 * lib constructor.
3430 */
3431 static void __attribute__((constructor)) init(void)
3432 {
3433 /* Set default session group */
3434 lttng_set_tracing_group(DEFAULT_TRACING_GROUP);
3435 }
3436
3437 /*
3438 * lib destructor.
3439 */
3440 static void __attribute__((destructor)) lttng_ctl_exit(void)
3441 {
3442 free(tracing_group);
3443 }
3444
3445 LTTNG_EXPORT enum lttng_error_code lttng_domain_supports_trace_format(
3446 const struct lttng_endpoint *session_daemon_endpoint,
3447 enum lttng_domain_type domain_type,
3448 const struct lttng_trace_format_descriptor *trace_format_descriptor)
3449 {
3450 int ret;
3451 enum lttng_error_code ret_code;
3452 struct lttcomm_session_msg lsm = {
3453 .cmd_type = LTTNG_EXECUTE_TRACE_FORMAT_SUPPORT_QUERY,
3454 .session = {},
3455 .domain = {},
3456 .u = {},
3457 .fd_count = 0,
3458 };
3459 struct lttng_payload message;
3460 struct lttng_payload reply;
3461 struct lttcomm_session_msg *message_lsm;
3462
3463 lttng_payload_init(&message);
3464 lttng_payload_init(&reply);
3465
3466 if (!session_daemon_endpoint || !trace_format_descriptor) {
3467 ret_code = LTTNG_ERR_INVALID;
3468 goto end;
3469 }
3470
3471 lsm.domain.type = domain_type;
3472 lsm.u.trace_format_support_query.query_type = 0;
3473 lsm.u.trace_format_support_query.type_payload_length = 0;
3474
3475 if (session_daemon_endpoint != lttng_session_daemon_command_endpoint) {
3476 ret_code = LTTNG_ERR_INVALID_ERROR_QUERY_TARGET;
3477 goto end;
3478 }
3479
3480 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3481 if (ret) {
3482 ret_code = LTTNG_ERR_NOMEM;
3483 goto end;
3484 }
3485
3486 ret = reinterpret_cast<const lttng::trace_format_descriptor *>(trace_format_descriptor)
3487 ->serialize(&message);
3488 if (ret) {
3489 ret_code = LTTNG_ERR_UNK;
3490 goto end;
3491 }
3492
3493 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3494 message_lsm->u.trace_format_support_query.trace_format_length =
3495 (uint32_t) message.buffer.size - sizeof(lsm);
3496
3497 {
3498 struct lttng_payload_view message_view =
3499 lttng_payload_view_from_payload(&message, 0, -1);
3500
3501 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(&message_view);
3502 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3503 if (ret < 0) {
3504 ret_code = (lttng_error_code) -ret;
3505 goto end;
3506 }
3507 if (ret != 0) {
3508 /* No payload expected */
3509 ret = LTTNG_ERR_INVALID_PROTOCOL;
3510 }
3511 }
3512
3513 ret_code = LTTNG_OK;
3514 end:
3515 lttng_payload_reset(&message);
3516 lttng_payload_reset(&reply);
3517 return ret_code;
3518 }
3519
3520 LTTNG_EXPORT extern enum lttng_error_code lttng_destination_supports_trace_format(
3521 const struct lttng_endpoint *session_daemon_endpoint,
3522 const char *destination_url,
3523 const struct lttng_trace_format_descriptor *trace_format_descriptor)
3524 {
3525 int ret;
3526 ssize_t size;
3527 enum lttng_error_code ret_code;
3528 struct lttcomm_session_msg lsm = {
3529 .cmd_type = LTTNG_EXECUTE_TRACE_FORMAT_SUPPORT_QUERY,
3530 .session = {},
3531 .domain = {},
3532 .u = {},
3533 .fd_count = 0,
3534 };
3535 struct lttng_payload message;
3536 struct lttng_payload reply;
3537 struct lttcomm_session_msg *message_lsm;
3538 struct lttng_uri *uris = NULL;
3539 size_t size_before_payload = 0;
3540
3541 lttng_payload_init(&message);
3542 lttng_payload_init(&reply);
3543
3544 if (!session_daemon_endpoint || !trace_format_descriptor) {
3545 ret_code = LTTNG_ERR_INVALID;
3546 goto end;
3547 }
3548
3549 size = uri_parse_str_urls(destination_url, NULL, &uris);
3550 if (size < 1) {
3551 ret_code = LTTNG_ERR_INVALID;
3552 goto end;
3553 }
3554
3555 if (uris[0].dtype == LTTNG_DST_PATH) {
3556 /* Local output */
3557 ret_code = LTTNG_OK;
3558 goto end;
3559 }
3560
3561 lsm.u.trace_format_support_query.query_type = 1;
3562
3563 if (session_daemon_endpoint != lttng_session_daemon_command_endpoint) {
3564 ret_code = LTTNG_ERR_INVALID_ERROR_QUERY_TARGET;
3565 goto end;
3566 }
3567
3568 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3569 if (ret) {
3570 ret_code = LTTNG_ERR_NOMEM;
3571 goto end;
3572 }
3573
3574 size_before_payload = message.buffer.size;
3575
3576 ret = reinterpret_cast<const lttng::trace_format_descriptor *>(trace_format_descriptor)
3577 ->serialize(&message);
3578 if (ret) {
3579 ret_code = LTTNG_ERR_UNK;
3580 goto end;
3581 }
3582
3583 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3584 message_lsm->u.trace_format_support_query.trace_format_length =
3585 (uint32_t) message.buffer.size - size_before_payload;
3586
3587 /* FIXME replace all this with good old serialization of sub query type
3588 * object. (internal) */
3589 size_before_payload = message.buffer.size;
3590 ret = lttng_dynamic_buffer_append(&message.buffer, &uris[0], sizeof(*uris));
3591 if (ret) {
3592 ret_code = LTTNG_ERR_NOMEM;
3593 goto end;
3594 }
3595
3596 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3597 message_lsm->u.trace_format_support_query.type_payload_length =
3598 (uint32_t) message.buffer.size - size_before_payload;
3599
3600 {
3601 struct lttng_payload_view message_view =
3602 lttng_payload_view_from_payload(&message, 0, -1);
3603
3604 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(&message_view);
3605 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3606 if (ret < 0) {
3607 ret_code = (lttng_error_code) -ret;
3608 goto end;
3609 }
3610 if (ret != 0) {
3611 /* No payload expected */
3612 ret_code = LTTNG_ERR_INVALID_PROTOCOL;
3613 goto end;
3614 }
3615 }
3616
3617 ret_code = LTTNG_OK;
3618 end:
3619 lttng_payload_reset(&message);
3620 lttng_payload_reset(&reply);
3621 return ret_code;
3622 }
This page took 0.147626 seconds and 5 git commands to generate.