Basic serialization/deserialization for lttng_session
[deliverable/lttng-tools.git] / src / lib / lttng-ctl / lttng-ctl.cpp
CommitLineData
826d496d 1/*
eb5c4f4e 2 * lttng-ctl.c
82a3637f
DG
3 *
4 * Linux Trace Toolkit Control Library
5 *
21cf9b6b 6 * Copyright (C) 2011 EfficiOS Inc.
ab5be9fa 7 * Copyright (C) 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
fac6795d 8 *
ab5be9fa 9 * SPDX-License-Identifier: LGPL-2.1-only
82a3637f 10 *
fac6795d
DG
11 */
12
6c1c0768 13#define _LGPL_SOURCE
fac6795d
DG
14#include <grp.h>
15#include <stdio.h>
16#include <stdlib.h>
13dd7782 17#include <stdint.h>
fac6795d
DG
18#include <string.h>
19#include <unistd.h>
20
c9e313bc
SM
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>
159b042f
JG
37#include <lttng/destruction-handle.h>
38#include <lttng/endpoint.h>
c9e313bc
SM
39#include <lttng/error-query-internal.hpp>
40#include <lttng/event-internal.hpp>
41#include <lttng/health-internal.hpp>
b99a0cb3 42#include <lttng/lttng-error.h>
159b042f 43#include <lttng/lttng.h>
c9e313bc
SM
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>
fac6795d 48
c9e313bc
SM
49#include "lttng-ctl-helper.hpp"
50#include <common/filter/filter-ast.hpp>
348ddc5c 51#include <common/filter/filter-parser.hpp>
c9e313bc 52#include <common/filter/memstream.hpp>
53a80697 53
60160d2a
JG
54#define COPY_DOMAIN_PACKED(dst, src) \
55do { \
56 struct lttng_domain _tmp_domain; \
57 \
58 lttng_ctl_copy_lttng_domain(&_tmp_domain, &src); \
59 dst = _tmp_domain; \
60} while (0)
53a80697 61
fac6795d 62/* Socket to session daemon for communication */
3e3665b8 63static int sessiond_socket = -1;
fac6795d
DG
64static char sessiond_sock_path[PATH_MAX];
65
fac6795d
DG
66/* Variables */
67static char *tracing_group;
68static int connected;
69
97e19046
DG
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.
97e19046 76 */
4bd69c5f
SM
77LTTNG_EXPORT int lttng_opt_quiet;
78LTTNG_EXPORT int lttng_opt_verbose;
79LTTNG_EXPORT int lttng_opt_mi;
97e19046 80
fac6795d 81/*
cd80958d 82 * Copy domain to lttcomm_session_msg domain.
fac6795d 83 *
cd80958d
DG
84 * If domain is unknown, default domain will be the kernel.
85 */
cac3069d
DG
86void lttng_ctl_copy_lttng_domain(struct lttng_domain *dst,
87 struct lttng_domain *src)
cd80958d
DG
88{
89 if (src && dst) {
90 switch (src->type) {
00e2e675
DG
91 case LTTNG_DOMAIN_KERNEL:
92 case LTTNG_DOMAIN_UST:
b9dfb167 93 case LTTNG_DOMAIN_JUL:
5cdb6027 94 case LTTNG_DOMAIN_LOG4J:
0e115563 95 case LTTNG_DOMAIN_PYTHON:
00e2e675
DG
96 memcpy(dst, src, sizeof(struct lttng_domain));
97 break;
98 default:
99 memset(dst, 0, sizeof(struct lttng_domain));
00e2e675 100 break;
cd80958d
DG
101 }
102 }
103}
104
105/*
106 * Send lttcomm_session_msg to the session daemon.
fac6795d 107 *
1c8d13c8
TD
108 * On success, returns the number of bytes sent (>=0)
109 * On error, returns -1
fac6795d 110 */
cd80958d 111static int send_session_msg(struct lttcomm_session_msg *lsm)
fac6795d
DG
112{
113 int ret;
114
115 if (!connected) {
2f70b271 116 ret = -LTTNG_ERR_NO_SESSIOND;
e065084a 117 goto end;
fac6795d
DG
118 }
119
4bd69c5f 120 DBG("LSM cmd type: '%s' (%d)", lttcomm_sessiond_command_str((lttcomm_sessiond_command) lsm->cmd_type),
19f912db 121 lsm->cmd_type);
a4b92340 122
be040666 123 ret = lttcomm_send_creds_unix_sock(sessiond_socket, lsm,
cd80958d 124 sizeof(struct lttcomm_session_msg));
2f70b271
DG
125 if (ret < 0) {
126 ret = -LTTNG_ERR_FATAL;
127 }
e065084a
DG
128
129end:
130 return ret;
131}
132
53a80697
MD
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 */
c2d69327 139static int send_session_varlen(const void *data, size_t len)
53a80697
MD
140{
141 int ret;
142
143 if (!connected) {
2f70b271 144 ret = -LTTNG_ERR_NO_SESSIOND;
53a80697
MD
145 goto end;
146 }
a4b92340 147
53a80697
MD
148 if (!data || !len) {
149 ret = 0;
150 goto end;
151 }
152
153 ret = lttcomm_send_unix_sock(sessiond_socket, data, len);
2f70b271
DG
154 if (ret < 0) {
155 ret = -LTTNG_ERR_FATAL;
156 }
53a80697
MD
157
158end:
159 return ret;
160}
161
a04d53fc
FD
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 */
168static 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
187end:
188 return ret;
189}
190
e065084a 191/*
cd80958d 192 * Receive data from the sessiond socket.
e065084a 193 *
1c8d13c8 194 * On success, returns the number of bytes received (>=0)
e368fb43 195 * On error, returns a negative lttng_error_code.
e065084a 196 */
ca95a216 197static int recv_data_sessiond(void *buf, size_t len)
e065084a
DG
198{
199 int ret;
200
a0377dfe 201 LTTNG_ASSERT(len > 0);
bacc41cc 202
e065084a 203 if (!connected) {
2f70b271 204 ret = -LTTNG_ERR_NO_SESSIOND;
e065084a 205 goto end;
fac6795d
DG
206 }
207
ca95a216 208 ret = lttcomm_recv_unix_sock(sessiond_socket, buf, len);
2f70b271
DG
209 if (ret < 0) {
210 ret = -LTTNG_ERR_FATAL;
bacc41cc
FD
211 } else if (ret == 0) {
212 ret = -LTTNG_ERR_NO_SESSIOND;
2f70b271 213 }
fac6795d 214
e065084a 215end:
fac6795d
DG
216 return ret;
217}
218
e368fb43
JG
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 */
225static 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);
239end:
240 return ret;
241}
242
fac6795d 243/*
9ae110e2 244 * Check if we are in the specified group.
65beb5ff 245 *
9ae110e2 246 * If yes return 1, else return -1.
947308c4 247 */
6c71277b 248int lttng_check_tracing_group(void)
947308c4 249{
28ab59d0 250 gid_t *grp_list, tracing_gid;
947308c4
DG
251 int grp_list_size, grp_id, i;
252 int ret = -1;
6c71277b 253 const char *grp_name = tracing_group;
947308c4
DG
254
255 /* Get GID of group 'tracing' */
28ab59d0 256 if (utils_get_group_id(grp_name, false, &tracing_gid)) {
b4d8603b 257 /* If grp_tracing is NULL, the group does not exist. */
947308c4
DG
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) {
6f04ed72 264 PERROR("getgroups");
947308c4
DG
265 goto end;
266 }
267
268 /* Alloc group list of the right size */
64803277 269 grp_list = calloc<gid_t>(grp_list_size);
00795392 270 if (!grp_list) {
6f04ed72 271 PERROR("malloc");
00795392
MD
272 goto end;
273 }
947308c4 274 grp_id = getgroups(grp_list_size, grp_list);
1c8d13c8 275 if (grp_id < 0) {
6f04ed72 276 PERROR("getgroups");
947308c4
DG
277 goto free_list;
278 }
279
280 for (i = 0; i < grp_list_size; i++) {
28ab59d0 281 if (grp_list[i] == tracing_gid) {
2269e89e 282 ret = 1;
947308c4
DG
283 break;
284 }
285 }
286
287free_list:
288 free(grp_list);
289
290end:
291 return ret;
292}
293
13dd7782
JR
294static enum lttng_error_code check_enough_available_memory(
295 uint64_t num_bytes_requested_per_cpu)
09b72f7a
FD
296{
297 int ret;
13dd7782 298 enum lttng_error_code ret_code;
09b72f7a 299 long num_cpu;
13dd7782
JR
300 uint64_t best_mem_info;
301 uint64_t num_bytes_requested_total;
09b72f7a
FD
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) {
13dd7782
JR
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;
09b72f7a
FD
317 }
318
13dd7782
JR
319 num_bytes_requested_total =
320 num_bytes_requested_per_cpu * (uint64_t) num_cpu;
09b72f7a
FD
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
13dd7782
JR
342 /* No valid source of information. */
343 ret_code = LTTNG_ERR_NOMEM;
344 goto end;
345
09b72f7a 346success:
13dd7782
JR
347 if (best_mem_info >= num_bytes_requested_total) {
348 ret_code = LTTNG_OK;
349 } else {
350 ret_code = LTTNG_ERR_NOMEM;
351 }
352end:
353 return ret_code;
09b72f7a
FD
354}
355
947308c4 356/*
2269e89e
DG
357 * Try connect to session daemon with sock_path.
358 *
359 * Return 0 on success, else -1
360 */
361static 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 */
2f70b271 369 goto error;
2269e89e
DG
370 }
371
372 ret = lttcomm_connect_unix_sock(sock_path);
373 if (ret < 0) {
9ae110e2 374 /* Not alive. */
2f70b271 375 goto error;
2269e89e
DG
376 }
377
378 ret = lttcomm_close_unix_sock(ret);
379 if (ret < 0) {
6f04ed72 380 PERROR("lttcomm_close_unix_sock");
2269e89e
DG
381 }
382
383 return 0;
2f70b271
DG
384
385error:
386 return -1;
2269e89e
DG
387}
388
389/*
2f70b271
DG
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).
947308c4
DG
395 */
396static int set_session_daemon_path(void)
397{
9ae110e2 398 int in_tgroup = 0; /* In tracing group. */
2269e89e
DG
399 uid_t uid;
400
401 uid = getuid();
947308c4 402
2269e89e
DG
403 if (uid != 0) {
404 /* Are we in the tracing group ? */
6c71277b 405 in_tgroup = lttng_check_tracing_group();
2269e89e
DG
406 }
407
c295ec41 408 if ((uid == 0) || in_tgroup == 1) {
e1b624d0
JG
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 }
08a9c49f 416 }
2269e89e 417
08a9c49f 418 if (uid != 0) {
c617c0c6
MD
419 int ret;
420
08a9c49f 421 if (in_tgroup) {
9ae110e2 422 /* Tracing group. */
08a9c49f
TD
423 ret = try_connect_sessiond(sessiond_sock_path);
424 if (ret >= 0) {
425 goto end;
2269e89e 426 }
08a9c49f 427 /* Global session daemon not available... */
2269e89e 428 }
08a9c49f
TD
429 /* ...or not in tracing group (and not root), default */
430
431 /*
9ae110e2
JG
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)
08a9c49f
TD
436 */
437 ret = snprintf(sessiond_sock_path, sizeof(sessiond_sock_path),
feb0f3e5 438 DEFAULT_HOME_CLIENT_UNIX_SOCK, utils_get_home_dir());
08a9c49f 439 if ((ret < 0) || (ret >= sizeof(sessiond_sock_path))) {
2f70b271 440 goto error;
947308c4 441 }
947308c4 442 }
08a9c49f 443end:
947308c4 444 return 0;
2f70b271
DG
445
446error:
447 return -1;
947308c4
DG
448}
449
65beb5ff 450/*
9ae110e2 451 * Connect to the LTTng session daemon.
65beb5ff 452 *
3e3665b8 453 * On success, return the socket's file descriptor. On error, return -1.
65beb5ff 454 */
ca806b0b 455int connect_sessiond(void)
65beb5ff
DG
456{
457 int ret;
458
459 ret = set_session_daemon_path();
460 if (ret < 0) {
2f70b271 461 goto error;
65beb5ff
DG
462 }
463
9ae110e2 464 /* Connect to the sesssion daemon. */
65beb5ff
DG
465 ret = lttcomm_connect_unix_sock(sessiond_sock_path);
466 if (ret < 0) {
2f70b271 467 goto error;
65beb5ff
DG
468 }
469
3e3665b8 470 return ret;
2f70b271
DG
471
472error:
473 return -1;
65beb5ff
DG
474}
475
3e3665b8
JG
476static void reset_global_sessiond_connection_state(void)
477{
478 sessiond_socket = -1;
479 connected = 0;
480}
481
65beb5ff 482/*
1c8d13c8 483 * Clean disconnect from the session daemon.
9ae110e2 484 *
1c8d13c8 485 * On success, return 0. On error, return -1.
65beb5ff
DG
486 */
487static int disconnect_sessiond(void)
488{
489 int ret = 0;
490
491 if (connected) {
492 ret = lttcomm_close_unix_sock(sessiond_socket);
3e3665b8 493 reset_global_sessiond_connection_state();
65beb5ff
DG
494 }
495
496 return ret;
497}
498
795a978d
PP
499static int recv_sessiond_optional_data(size_t len, void **user_buf,
500 size_t *user_len)
501{
502 int ret = 0;
64803277 503 char *buf = NULL;
795a978d
PP
504
505 if (len) {
506 if (!user_len) {
507 ret = -LTTNG_ERR_INVALID;
508 goto end;
509 }
510
64803277 511 buf = zmalloc<char>(len);
795a978d
PP
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
542end:
543 free(buf);
544 return ret;
545}
546
35a6fdb7 547/*
cd80958d 548 * Ask the session daemon a specific command and put the data into buf.
a04d53fc
FD
549 * Takes extra var. len. data and file descriptors as input to send to the
550 * session daemon.
65beb5ff 551 *
af87c45a 552 * Return size of data (only payload, not header) or a negative error code.
65beb5ff 553 */
a04d53fc
FD
554int 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)
65beb5ff
DG
558{
559 int ret;
795a978d 560 size_t payload_len;
cd80958d 561 struct lttcomm_lttng_msg llm;
65beb5ff
DG
562
563 ret = connect_sessiond();
564 if (ret < 0) {
2f70b271 565 ret = -LTTNG_ERR_NO_SESSIOND;
65beb5ff 566 goto end;
3e3665b8
JG
567 } else {
568 sessiond_socket = ret;
569 connected = 1;
65beb5ff
DG
570 }
571
cd80958d 572 ret = send_session_msg(lsm);
65beb5ff 573 if (ret < 0) {
2f70b271 574 /* Ret value is a valid lttng error code. */
65beb5ff
DG
575 goto end;
576 }
53a80697 577 /* Send var len data */
795a978d 578 ret = send_session_varlen(vardata, vardata_len);
53a80697 579 if (ret < 0) {
2f70b271 580 /* Ret value is a valid lttng error code. */
53a80697
MD
581 goto end;
582 }
65beb5ff 583
a04d53fc
FD
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
65beb5ff
DG
591 /* Get header from data transmission */
592 ret = recv_data_sessiond(&llm, sizeof(llm));
593 if (ret < 0) {
2f70b271 594 /* Ret value is a valid lttng error code. */
65beb5ff
DG
595 goto end;
596 }
597
598 /* Check error code if OK */
f73fabfd 599 if (llm.ret_code != LTTNG_OK) {
65beb5ff
DG
600 ret = -llm.ret_code;
601 goto end;
602 }
603
795a978d
PP
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);
65beb5ff 607 if (ret < 0) {
65beb5ff
DG
608 goto end;
609 }
610
795a978d
PP
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) {
83009e5e
DG
615 goto end;
616 }
617
795a978d 618 ret = llm.data_size;
65beb5ff
DG
619
620end:
621 disconnect_sessiond();
622 return ret;
623}
624
e368fb43
JG
625int 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;
fe489250 630 const int fd_count = lttng_payload_view_get_fd_handle_count(message);
e368fb43 631
a0377dfe
FD
632 LTTNG_ASSERT(reply->buffer.size == 0);
633 LTTNG_ASSERT(lttng_dynamic_pointer_array_get_count(&reply->_fd_handles) == 0);
e368fb43
JG
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
fe489250
JG
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 }
e368fb43
JG
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) {
eb441106
JG
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 }
e368fb43
JG
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) {
fe489250
JG
697 ret = lttcomm_recv_payload_fds_unix_sock(
698 sessiond_socket, llm.fd_count, reply);
699 if (ret < 0) {
e368fb43
JG
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
716end:
717 disconnect_sessiond();
718 return ret;
719}
720
9f19cc17 721/*
cd80958d 722 * Create lttng handle and return pointer.
9ae110e2 723 *
1c8d13c8 724 * The returned pointer will be NULL in case of malloc() error.
9f19cc17 725 */
cd80958d
DG
726struct lttng_handle *lttng_create_handle(const char *session_name,
727 struct lttng_domain *domain)
9f19cc17 728{
e1b624d0 729 int ret;
2f70b271
DG
730 struct lttng_handle *handle = NULL;
731
64803277 732 handle = zmalloc<lttng_handle>();
cd80958d 733 if (handle == NULL) {
2f70b271 734 PERROR("malloc handle");
cd80958d
DG
735 goto end;
736 }
737
738 /* Copy session name */
e1b624d0
JG
739 ret = lttng_strncpy(handle->session_name, session_name ? : "",
740 sizeof(handle->session_name));
741 if (ret) {
742 goto error;
743 }
cd80958d 744
95681498
JG
745 /* Copy lttng domain or leave initialized to 0. */
746 if (domain) {
747 lttng_ctl_copy_lttng_domain(&handle->domain, domain);
748 }
cd80958d
DG
749
750end:
751 return handle;
e1b624d0
JG
752error:
753 free(handle);
754 return NULL;
cd80958d
DG
755}
756
757/*
758 * Destroy handle by free(3) the pointer.
759 */
760void lttng_destroy_handle(struct lttng_handle *handle)
761{
0e428499 762 free(handle);
eb354453
DG
763}
764
d9800920
DG
765/*
766 * Register an outside consumer.
9ae110e2 767 *
1c8d13c8 768 * Returns size of returned session payload data or a negative error code.
d9800920
DG
769 */
770int lttng_register_consumer(struct lttng_handle *handle,
771 const char *socket_path)
772{
e1b624d0 773 int ret;
d9800920
DG
774 struct lttcomm_session_msg lsm;
775
2f70b271 776 if (handle == NULL || socket_path == NULL) {
e1b624d0
JG
777 ret = -LTTNG_ERR_INVALID;
778 goto end;
2f70b271
DG
779 }
780
53efb85a 781 memset(&lsm, 0, sizeof(lsm));
d9800920 782 lsm.cmd_type = LTTNG_REGISTER_CONSUMER;
e1b624d0 783 ret = lttng_strncpy(lsm.session.name, handle->session_name,
d9800920 784 sizeof(lsm.session.name));
e1b624d0
JG
785 if (ret) {
786 ret = -LTTNG_ERR_INVALID;
787 goto end;
788 }
789
60160d2a 790 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
d9800920 791
e1b624d0
JG
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 }
d9800920 798
e1b624d0
JG
799 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
800end:
801 return ret;
d9800920
DG
802}
803
1df4dedd 804/*
9ae110e2
JG
805 * Start tracing for all traces of the session.
806 *
807 * Returns size of returned session payload data or a negative error code.
1df4dedd 808 */
6a4f824d 809int lttng_start_tracing(const char *session_name)
f3ed775e 810{
e1b624d0 811 int ret;
cd80958d
DG
812 struct lttcomm_session_msg lsm;
813
6a4f824d 814 if (session_name == NULL) {
e1b624d0
JG
815 ret = -LTTNG_ERR_INVALID;
816 goto end;
cd80958d
DG
817 }
818
53efb85a 819 memset(&lsm, 0, sizeof(lsm));
cd80958d 820 lsm.cmd_type = LTTNG_START_TRACE;
6a4f824d 821
e1b624d0
JG
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 }
cd80958d 828
e1b624d0
JG
829 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
830end:
831 return ret;
f3ed775e 832}
1df4dedd
DG
833
834/*
38ee087f 835 * Stop tracing for all traces of the session.
f3ed775e 836 */
38ee087f 837static int _lttng_stop_tracing(const char *session_name, int wait)
f3ed775e 838{
38ee087f 839 int ret, data_ret;
cd80958d
DG
840 struct lttcomm_session_msg lsm;
841
6a4f824d 842 if (session_name == NULL) {
e1b624d0
JG
843 ret = -LTTNG_ERR_INVALID;
844 goto error;
6a4f824d
DG
845 }
846
53efb85a 847 memset(&lsm, 0, sizeof(lsm));
cd80958d 848 lsm.cmd_type = LTTNG_STOP_TRACE;
6a4f824d 849
e1b624d0
JG
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 }
cd80958d 856
cac3069d 857 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
38ee087f
DG
858 if (ret < 0 && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
859 goto error;
860 }
861
862 if (!wait) {
863 goto end;
864 }
865
38ee087f
DG
866 /* Check for data availability */
867 do {
6d805429 868 data_ret = lttng_data_pending(session_name);
38ee087f
DG
869 if (data_ret < 0) {
870 /* Return the data available call error. */
871 ret = data_ret;
872 goto error;
873 }
874
875 /*
9ae110e2
JG
876 * Data sleep time before retrying (in usec). Don't sleep if the
877 * call returned value indicates availability.
38ee087f 878 */
6d805429 879 if (data_ret) {
c8f61fd4 880 usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME_US);
38ee087f 881 }
6d805429 882 } while (data_ret != 0);
38ee087f 883
38ee087f
DG
884end:
885error:
886 return ret;
887}
888
889/*
890 * Stop tracing and wait for data availability.
891 */
892int 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 */
900int lttng_stop_tracing_no_wait(const char *session_name)
901{
902 return _lttng_stop_tracing(session_name, 0);
f3ed775e
DG
903}
904
905/*
601d5acf
DG
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.
af87c45a
DG
910 *
911 * Returns the size of the returned payload data or a negative error code.
1df4dedd 912 */
cd80958d 913int lttng_add_context(struct lttng_handle *handle,
f46376a1
MJ
914 struct lttng_event_context *ctx,
915 const char *event_name __attribute__((unused)),
38057ed1 916 const char *channel_name)
d65106b1 917{
2001793c 918 int ret;
1c9a0b0e
MJ
919 struct lttcomm_session_msg lsm = {
920 .cmd_type = LTTNG_ADD_CONTEXT,
921 .session = {},
922 .domain = {},
923 .u = {},
924 .fd_count = 0,
925 };
26e1c61f
JR
926 struct lttng_payload payload;
927
928 lttng_payload_init(&payload);
cd80958d 929
9ae110e2 930 /* Safety check. Both are mandatory. */
9d697d3d 931 if (handle == NULL || ctx == NULL) {
2001793c
JG
932 ret = -LTTNG_ERR_INVALID;
933 goto end;
cd80958d
DG
934 }
935
26e1c61f
JR
936 ret = lttng_dynamic_buffer_set_size(&payload.buffer, sizeof(lsm));
937 if (ret) {
938 ret = -LTTNG_ERR_NOMEM;
939 goto end;
940 }
cd80958d 941
85076754 942 /* If no channel name, send empty string. */
e1b624d0
JG
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;
85076754 948 }
cd80958d 949
60160d2a 950 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
e1b624d0 951 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 952 sizeof(lsm.session.name));
e1b624d0
JG
953 if (ret) {
954 ret = -LTTNG_ERR_INVALID;
955 goto end;
956 }
cd80958d 957
26e1c61f
JR
958 ret = lttng_event_context_serialize(ctx, &payload);
959 if (ret) {
960 ret = -LTTNG_ERR_INVALID;
961 goto end;
962 }
2001793c 963
26e1c61f 964 lsm.u.context.length = payload.buffer.size - sizeof(lsm);
2001793c 965
26e1c61f
JR
966 /* Update message header. */
967 memcpy(payload.buffer.data, &lsm, sizeof(lsm));
2001793c 968
26e1c61f
JR
969 {
970 struct lttng_payload reply;
971 struct lttng_payload_view payload_view =
972 lttng_payload_view_from_payload(&payload, 0,
973 -1);
2001793c 974
26e1c61f
JR
975 lttng_payload_init(&reply);
976 ret = lttng_ctl_ask_sessiond_payload(&payload_view, &reply);
977 lttng_payload_reset(&reply);
978 if (ret) {
2001793c
JG
979 goto end;
980 }
2001793c 981 }
46f44e2a 982
2001793c 983end:
26e1c61f 984 lttng_payload_reset(&payload);
2001793c 985 return ret;
d65106b1
DG
986}
987
f3ed775e 988/*
9ae110e2
JG
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.
f3ed775e 995 */
cd80958d 996int lttng_enable_event(struct lttng_handle *handle,
38057ed1 997 struct lttng_event *ev, const char *channel_name)
1df4dedd 998{
f8a96544
JI
999 return lttng_enable_event_with_exclusions(handle, ev, channel_name,
1000 NULL, 0, NULL);
1df4dedd
DG
1001}
1002
53a80697 1003/*
025faf73 1004 * Create or enable an event with a filter expression.
178191b3 1005 *
53a80697
MD
1006 * Return negative error value on error.
1007 * Return size of returned session payload data if OK.
1008 */
025faf73 1009int lttng_enable_event_with_filter(struct lttng_handle *handle,
178191b3 1010 struct lttng_event *event, const char *channel_name,
53a80697
MD
1011 const char *filter_expression)
1012{
f8a96544
JI
1013 return lttng_enable_event_with_exclusions(handle, event, channel_name,
1014 filter_expression, 0, NULL);
53a80697
MD
1015}
1016
347c5ab5 1017/*
0e115563 1018 * Depending on the event, return a newly allocated agent filter expression or
347c5ab5
DG
1019 * NULL if not applicable.
1020 *
1021 * An event with NO loglevel and the name is * will return NULL.
1022 */
0e115563 1023static char *set_agent_filter(const char *filter, struct lttng_event *ev)
347c5ab5
DG
1024{
1025 int err;
0e115563 1026 char *agent_filter = NULL;
347c5ab5 1027
a0377dfe 1028 LTTNG_ASSERT(ev);
347c5ab5
DG
1029
1030 /* Don't add filter for the '*' event. */
9f449915 1031 if (strcmp(ev->name, "*") != 0) {
347c5ab5 1032 if (filter) {
0e115563 1033 err = asprintf(&agent_filter, "(%s) && (logger_name == \"%s\")", filter,
347c5ab5
DG
1034 ev->name);
1035 } else {
0e115563 1036 err = asprintf(&agent_filter, "logger_name == \"%s\"", ev->name);
347c5ab5
DG
1037 }
1038 if (err < 0) {
1039 PERROR("asprintf");
6a556f7b 1040 goto error;
347c5ab5
DG
1041 }
1042 }
1043
1044 /* Add loglevel filtering if any for the JUL domain. */
1045 if (ev->loglevel_type != LTTNG_EVENT_LOGLEVEL_ALL) {
b53d4e59 1046 const char *op;
347c5ab5
DG
1047
1048 if (ev->loglevel_type == LTTNG_EVENT_LOGLEVEL_RANGE) {
1049 op = ">=";
1050 } else {
1051 op = "==";
1052 }
1053
0e115563 1054 if (filter || agent_filter) {
6a556f7b
JG
1055 char *new_filter;
1056
fb0edb23 1057 err = asprintf(&new_filter, "(%s) && (int_loglevel %s %d)",
0e115563 1058 agent_filter ? agent_filter : filter, op,
347c5ab5 1059 ev->loglevel);
0e115563
DG
1060 if (agent_filter) {
1061 free(agent_filter);
6a556f7b 1062 }
0e115563 1063 agent_filter = new_filter;
347c5ab5 1064 } else {
0e115563 1065 err = asprintf(&agent_filter, "int_loglevel %s %d", op,
347c5ab5
DG
1066 ev->loglevel);
1067 }
1068 if (err < 0) {
1069 PERROR("asprintf");
6a556f7b 1070 goto error;
347c5ab5
DG
1071 }
1072 }
1073
0e115563 1074 return agent_filter;
6a556f7b 1075error:
0e115563 1076 free(agent_filter);
6a556f7b 1077 return NULL;
347c5ab5
DG
1078}
1079
93deb080
JI
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 */
1088int lttng_enable_event_with_exclusions(struct lttng_handle *handle,
1089 struct lttng_event *ev, const char *channel_name,
e9efbcd3 1090 const char *original_filter_expression,
93deb080
JI
1091 int exclusion_count, char **exclusion_list)
1092{
1c9a0b0e
MJ
1093 struct lttcomm_session_msg lsm = {
1094 .cmd_type = LTTNG_ENABLE_EVENT,
1095 .session = {},
1096 .domain = {},
1097 .u = {},
1098 .fd_count = 0,
1099 };
e368fb43 1100 struct lttng_payload payload;
8ddd72ef 1101 int ret = 0;
137b9942 1102 unsigned int free_filter_expression = 0;
93deb080 1103 struct filter_parser_ctx *ctx = NULL;
8ddd72ef 1104 size_t bytecode_len = 0;
ec005ec6 1105
6a0838b7
YL
1106 /*
1107 * We have either a filter or some exclusions, so we need to set up
e368fb43 1108 * a variable-length payload from where to send the data.
6a0838b7 1109 */
e368fb43 1110 lttng_payload_init(&payload);
ec005ec6 1111
e9efbcd3
JG
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;
93deb080
JI
1118
1119 if (handle == NULL || ev == NULL) {
137b9942
DG
1120 ret = -LTTNG_ERR_INVALID;
1121 goto error;
93deb080
JI
1122 }
1123
9ae110e2
JG
1124 /*
1125 * Empty filter string will always be rejected by the parser
93deb080
JI
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') {
137b9942
DG
1130 ret = -LTTNG_ERR_INVALID;
1131 goto error;
93deb080
JI
1132 }
1133
18a720cd 1134 if (ev->name[0] == '\0') {
e1b624d0
JG
1135 /* Enable all events. */
1136 ret = lttng_strncpy(ev->name, "*", sizeof(ev->name));
a0377dfe 1137 LTTNG_ASSERT(ret == 0);
6565421f 1138 }
93deb080 1139
9ae110e2 1140 /* Parse filter expression. */
5cdb6027 1141 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
0e115563
DG
1142 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1143 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
5cdb6027 1144 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
0e115563
DG
1145 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1146 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1147 char *agent_filter;
64226865 1148
8ddd72ef 1149 /* Setup agent filter if needed. */
0e115563
DG
1150 agent_filter = set_agent_filter(filter_expression, ev);
1151 if (!agent_filter) {
137b9942 1152 if (!filter_expression) {
9ae110e2
JG
1153 /*
1154 * No JUL and no filter, just skip
1155 * everything below.
1156 */
8ddd72ef 1157 goto serialize;
137b9942 1158 }
64226865
DG
1159 } else {
1160 /*
9ae110e2
JG
1161 * With an agent filter, the original filter has
1162 * been added to it thus replace the filter
1163 * expression.
64226865 1164 */
0e115563 1165 filter_expression = agent_filter;
e9efbcd3 1166 free_filter_expression = 1;
9b21e6d5 1167 }
9b21e6d5 1168 }
93deb080 1169
8ddd72ef
JR
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
e4d2f27a 1176 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
93deb080 1177 if (ret) {
8ddd72ef 1178 goto error;
93deb080 1179 }
e4d2f27a 1180
8ddd72ef
JR
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 }
6b453b5e
JG
1187 }
1188
8ddd72ef
JR
1189serialize:
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);
15e37663 1194 if (ret) {
8ddd72ef
JR
1195 ret = -LTTNG_ERR_INVALID;
1196 goto error;
6b453b5e 1197 }
137b9942 1198
8ddd72ef
JR
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;
6b453b5e 1205 }
15e37663 1206
8ddd72ef
JR
1207 /* Domain */
1208 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
15e37663 1209
8ddd72ef
JR
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;
93deb080
JI
1216 }
1217
8ddd72ef
JR
1218 /* Length of the serialized event. */
1219 lsm.u.enable.length = (uint32_t) payload.buffer.size;
1220
e368fb43
JG
1221 {
1222 struct lttng_payload_view view = lttng_payload_view_from_payload(
1223 &payload, 0, -1);
fe489250 1224 int fd_count = lttng_payload_view_get_fd_handle_count(&view);
e368fb43
JG
1225 int fd_to_send;
1226
1227 if (fd_count < 0) {
8ddd72ef 1228 goto error;
e368fb43
JG
1229 }
1230
a0377dfe 1231 LTTNG_ASSERT(fd_count == 0 || fd_count == 1);
e368fb43 1232 if (fd_count == 1) {
e2fc8e9d 1233 struct fd_handle *h =
fe489250
JG
1234 lttng_payload_view_pop_fd_handle(&view);
1235
e2fc8e9d 1236 if (!h) {
8ddd72ef 1237 goto error;
e368fb43
JG
1238 }
1239
e2fc8e9d
SM
1240 fd_to_send = fd_handle_get_fd(h);
1241 fd_handle_put(h);
e368fb43
JG
1242 }
1243
8ddd72ef
JR
1244 lsm.fd_count = fd_count;
1245
e368fb43
JG
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 }
93deb080 1251
8ddd72ef 1252error:
7ca1dc6f 1253 if (filter_expression && ctx) {
93deb080
JI
1254 filter_bytecode_free(ctx);
1255 filter_ir_free(ctx);
1256 filter_parser_ctx_free(ctx);
7ca1dc6f 1257 }
7ca1dc6f
DG
1258 if (free_filter_expression) {
1259 /*
9ae110e2
JG
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.
7ca1dc6f
DG
1263 */
1264 free(filter_expression);
93deb080 1265 }
137b9942 1266 /*
9ae110e2
JG
1267 * Return directly to the caller and don't ask the sessiond since
1268 * something went wrong in the parsing of data above.
137b9942 1269 */
e368fb43 1270 lttng_payload_reset(&payload);
93deb080
JI
1271 return ret;
1272}
1273
6e911cad
MD
1274int lttng_disable_event_ext(struct lttng_handle *handle,
1275 struct lttng_event *ev, const char *channel_name,
1276 const char *original_filter_expression)
1df4dedd 1277{
1c9a0b0e
MJ
1278 struct lttcomm_session_msg lsm = {
1279 .cmd_type = LTTNG_DISABLE_EVENT,
1280 .session = {},
1281 .domain = {},
1282 .u = {},
1283 .fd_count = 0,
1284 };
8ddd72ef 1285 struct lttng_payload payload;
6e911cad
MD
1286 int ret = 0;
1287 unsigned int free_filter_expression = 0;
1288 struct filter_parser_ctx *ctx = NULL;
8ddd72ef
JR
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
6e911cad
MD
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;
1df4dedd 1303
6e911cad
MD
1304 if (handle == NULL || ev == NULL) {
1305 ret = -LTTNG_ERR_INVALID;
1306 goto error;
1307 }
1308
9ae110e2
JG
1309 /*
1310 * Empty filter string will always be rejected by the parser
6e911cad
MD
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;
cd80958d
DG
1317 }
1318
8ddd72ef 1319 /* Parse filter expression. */
6e911cad 1320 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
0e115563
DG
1321 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1322 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
6e911cad 1323 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
0e115563
DG
1324 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1325 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1326 char *agent_filter;
6e911cad 1327
8ddd72ef 1328 /* Setup agent filter if needed. */
0e115563
DG
1329 agent_filter = set_agent_filter(filter_expression, ev);
1330 if (!agent_filter) {
6e911cad 1331 if (!filter_expression) {
9ae110e2
JG
1332 /*
1333 * No JUL and no filter, just skip
1334 * everything below.
1335 */
8ddd72ef 1336 goto serialize;
6e911cad
MD
1337 }
1338 } else {
1339 /*
8ddd72ef 1340 * With an agent filter, the original filter has
9ae110e2
JG
1341 * been added to it thus replace the filter
1342 * expression.
6e911cad 1343 */
0e115563 1344 filter_expression = agent_filter;
6e911cad
MD
1345 free_filter_expression = 1;
1346 }
1347 }
1348
8ddd72ef
JR
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
e4d2f27a 1355 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
6e911cad 1356 if (ret) {
8ddd72ef 1357 goto error;
6e911cad 1358 }
e4d2f27a 1359
8ddd72ef
JR
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 }
6e911cad
MD
1366 }
1367
8ddd72ef
JR
1368serialize:
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;
6e911cad
MD
1376 }
1377
8ddd72ef
JR
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;
6e911cad 1384 }
8ddd72ef
JR
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;
6e911cad
MD
1395 }
1396
8ddd72ef
JR
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);
6e911cad 1414
8ddd72ef
JR
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
1429error:
6e911cad
MD
1430 if (filter_expression && ctx) {
1431 filter_bytecode_free(ctx);
1432 filter_ir_free(ctx);
1433 filter_parser_ctx_free(ctx);
1434 }
6e911cad
MD
1435 if (free_filter_expression) {
1436 /*
9ae110e2
JG
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.
6e911cad
MD
1440 */
1441 free(filter_expression);
1442 }
6e911cad 1443 /*
9ae110e2
JG
1444 * Return directly to the caller and don't ask the sessiond since
1445 * something went wrong in the parsing of data above.
6e911cad 1446 */
8ddd72ef 1447 lttng_payload_reset(&payload);
6e911cad
MD
1448 return ret;
1449}
1450
1451/*
9ae110e2
JG
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.
6e911cad
MD
1456 */
1457int lttng_disable_event(struct lttng_handle *handle, const char *name,
1458 const char *channel_name)
1459{
e1b624d0 1460 int ret;
6e911cad
MD
1461 struct lttng_event ev;
1462
1463 memset(&ev, 0, sizeof(ev));
9b7431cf 1464 ev.loglevel = -1;
6e911cad 1465 ev.type = LTTNG_EVENT_ALL;
e1b624d0
JG
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);
1473end:
1474 return ret;
1df4dedd
DG
1475}
1476
cf0bcb51
JG
1477struct lttng_channel *lttng_channel_create(struct lttng_domain *domain)
1478{
1479 struct lttng_channel *channel = NULL;
cf0bcb51
JG
1480
1481 if (!domain) {
999af9c1 1482 goto end;
cf0bcb51
JG
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:
999af9c1 1493 goto end;
cf0bcb51
JG
1494 }
1495 break;
1496 case LTTNG_DOMAIN_KERNEL:
1497 if (domain->buf_type != LTTNG_BUFFER_GLOBAL) {
999af9c1 1498 goto end;
cf0bcb51
JG
1499 }
1500 break;
1501 default:
999af9c1 1502 goto end;
cf0bcb51
JG
1503 }
1504
999af9c1 1505 channel = lttng_channel_create_internal();
cf0bcb51 1506 if (!channel) {
999af9c1 1507 goto end;
cf0bcb51
JG
1508 }
1509
cf0bcb51 1510 lttng_channel_set_default_attr(domain, &channel->attr);
999af9c1 1511end:
cf0bcb51 1512 return channel;
cf0bcb51
JG
1513}
1514
1515void 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
1df4dedd 1527/*
9ae110e2
JG
1528 * Enable channel per domain
1529 * Returns size of returned session payload data or a negative error code.
a5c5a2bd 1530 */
cd80958d 1531int lttng_enable_channel(struct lttng_handle *handle,
cf0bcb51 1532 struct lttng_channel *in_chan)
a5c5a2bd 1533{
13dd7782 1534 enum lttng_error_code ret_code;
e1b624d0 1535 int ret;
999af9c1 1536 struct lttng_dynamic_buffer buffer;
cd80958d 1537 struct lttcomm_session_msg lsm;
13dd7782 1538 uint64_t total_buffer_size_needed_per_cpu = 0;
999af9c1
JR
1539 struct lttng_channel *channel = NULL;
1540
1541 lttng_dynamic_buffer_init(&buffer);
cd80958d 1542
9ae110e2 1543 /* NULL arguments are forbidden. No default values. */
cf0bcb51 1544 if (handle == NULL || in_chan == NULL) {
999af9c1
JR
1545 ret = -LTTNG_ERR_INVALID;
1546 goto end;
cf0bcb51 1547 }
cd80958d 1548
09b72f7a
FD
1549 /*
1550 * Verify that the amount of memory required to create the requested
1551 * buffer is available on the system at the moment.
1552 */
999af9c1
JR
1553 if (in_chan->attr.num_subbuf >
1554 UINT64_MAX / in_chan->attr.subbuf_size) {
13dd7782
JR
1555 /* Overflow */
1556 ret = -LTTNG_ERR_OVERFLOW;
1557 goto end;
1558 }
1559
999af9c1
JR
1560 total_buffer_size_needed_per_cpu =
1561 in_chan->attr.num_subbuf * in_chan->attr.subbuf_size;
13dd7782
JR
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;
09b72f7a
FD
1567 }
1568
999af9c1
JR
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 =
64803277 1579 zmalloc<lttng_channel_extended>();
999af9c1
JR
1580
1581 if (!extended) {
1582 ret = -LTTNG_ERR_NOMEM;
1583 goto end;
1584 }
64803277 1585
999af9c1
JR
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
cf0bcb51 1594 lsm.cmd_type = LTTNG_ENABLE_CHANNEL;
60160d2a 1595 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
7d29a247 1596
e1b624d0
JG
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 }
cd80958d 1603
999af9c1
JR
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);
e1b624d0 1614end:
999af9c1
JR
1615 lttng_channel_destroy(channel);
1616 lttng_dynamic_buffer_reset(&buffer);
e1b624d0 1617 return ret;
8c0faa1d 1618}
1df4dedd 1619
2ef84c95 1620/*
9ae110e2
JG
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.
2ef84c95 1623 */
cd80958d 1624int lttng_disable_channel(struct lttng_handle *handle, const char *name)
2ef84c95 1625{
e1b624d0 1626 int ret;
cd80958d
DG
1627 struct lttcomm_session_msg lsm;
1628
9ae110e2 1629 /* Safety check. Both are mandatory. */
9d697d3d 1630 if (handle == NULL || name == NULL) {
2f70b271 1631 return -LTTNG_ERR_INVALID;
cd80958d
DG
1632 }
1633
441c16a7
MD
1634 memset(&lsm, 0, sizeof(lsm));
1635
cd80958d 1636 lsm.cmd_type = LTTNG_DISABLE_CHANNEL;
1df4dedd 1637
e1b624d0 1638 ret = lttng_strncpy(lsm.u.disable.channel_name, name,
9d697d3d 1639 sizeof(lsm.u.disable.channel_name));
e1b624d0
JG
1640 if (ret) {
1641 ret = -LTTNG_ERR_INVALID;
1642 goto end;
1643 }
9d697d3d 1644
60160d2a 1645 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
cd80958d 1646
e1b624d0
JG
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 }
cd80958d 1653
e1b624d0
JG
1654 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1655end:
1656 return ret;
ca95a216
DG
1657}
1658
fac6795d 1659/*
9ae110e2
JG
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.
fac6795d 1664 */
cd80958d 1665int lttng_list_tracepoints(struct lttng_handle *handle,
2a71efd5 1666 struct lttng_event **events)
fac6795d 1667{
8ddd72ef
JR
1668 enum lttng_error_code ret_code;
1669 int ret, total_payload_received;
1670 char *reception_buffer = NULL;
1c9a0b0e
MJ
1671 struct lttcomm_session_msg lsm = {
1672 .cmd_type = LTTNG_LIST_TRACEPOINTS,
1673 .session = {},
1674 .domain = {},
1675 .u = {},
1676 .fd_count = 0,
1677 };
8ddd72ef
JR
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;
fac6795d 1709
8ddd72ef
JR
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);
2a71efd5 1717
8ddd72ef
JR
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 }
eb354453 1724 }
fac6795d 1725
8ddd72ef
JR
1726 ret = (int) nb_events;
1727
1728end:
1729 free(cmd_header);
1730 free(reception_buffer);
1731 return ret;
fac6795d
DG
1732}
1733
f37d259d 1734/*
9ae110e2
JG
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.
f37d259d
MD
1739 */
1740int lttng_list_tracepoint_fields(struct lttng_handle *handle,
1741 struct lttng_event_field **fields)
1742{
b2d68839 1743 enum lttng_error_code ret_code;
f37d259d
MD
1744 int ret;
1745 struct lttcomm_session_msg lsm;
b2d68839
JR
1746 const struct lttcomm_list_command_header *cmd_header = NULL;
1747 unsigned int nb_event_fields = 0;
1748 struct lttng_payload reply;
f37d259d 1749
5c5c6942
JG
1750 lttng_payload_init(&reply);
1751
f37d259d 1752 if (handle == NULL) {
b2d68839
JR
1753 ret = -LTTNG_ERR_INVALID;
1754 goto end;
f37d259d
MD
1755 }
1756
53efb85a 1757 memset(&lsm, 0, sizeof(lsm));
f37d259d 1758 lsm.cmd_type = LTTNG_LIST_TRACEPOINT_FIELDS;
60160d2a 1759 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
f37d259d 1760
b2d68839
JR
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;
f37d259d
MD
1790 }
1791
b2d68839
JR
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
1809end:
5c5c6942 1810 lttng_payload_reset(&reply);
b2d68839 1811 return ret;
f37d259d
MD
1812}
1813
834978fd 1814/*
9ae110e2
JG
1815 * Lists all available kernel system calls. Allocates and sets the contents of
1816 * the events array.
834978fd 1817 *
9ae110e2
JG
1818 * Returns the number of lttng_event entries in events; on error, returns a
1819 * negative value.
834978fd
DG
1820 */
1821int lttng_list_syscalls(struct lttng_event **events)
1822{
8ddd72ef
JR
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;
834978fd 1877
8ddd72ef
JR
1878end:
1879 free(reception_buffer);
1880 free(cmd_header);
1881 return ret;
834978fd
DG
1882}
1883
1657e9bb 1884/*
9ae110e2 1885 * Returns a human readable string describing
8346beb6 1886 * the error code (positive or negative value).
1657e9bb 1887 */
9a745bc7 1888const char *lttng_strerror(int code)
1657e9bb 1889{
8346beb6
PP
1890 if (code > 0) {
1891 code = -code;
1892 }
1893
f73fabfd 1894 return error_get_str(code);
1657e9bb
DG
1895}
1896
b178f53e
JG
1897enum 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,
1c9a0b0e
MJ
1903 .session = {},
1904 .domain = {},
1905 .u = {},
1906 .fd_count = 0,
b178f53e
JG
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_dynamic_buffer payload;
1913 int ret;
1914 size_t descriptor_size;
1915 struct lttng_session_descriptor *descriptor_reply = NULL;
1916
1917 lttng_dynamic_buffer_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, home_dir,
1937 home_dir_len);
1938 if (ret) {
1939 ret_code = LTTNG_ERR_NOMEM;
1940 goto end;
1941 }
1942 }
1943
1944 descriptor_size = payload.size;
1945 ret = lttng_session_descriptor_serialize(session_descriptor,
1946 &payload);
1947 if (ret) {
1948 ret_code = LTTNG_ERR_INVALID;
1949 goto end;
1950 }
1951 descriptor_size = payload.size - descriptor_size;
1952 lsm.u.create_session.session_descriptor_size = descriptor_size;
1953
1954 /* Command returns a session descriptor on success. */
1955 reply_ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, payload.data,
1956 payload.size, &reply);
1957 if (reply_ret < 0) {
4bd69c5f 1958 ret_code = (lttng_error_code) -reply_ret;
b178f53e
JG
1959 goto end;
1960 } else if (reply_ret == 0) {
1961 /* Socket unexpectedly closed by the session daemon. */
1962 ret_code = LTTNG_ERR_FATAL;
1963 goto end;
1964 }
1965
4bd69c5f 1966 reply_view = lttng_buffer_view_init((const char *) reply, 0, reply_ret);
b178f53e
JG
1967 ret = lttng_session_descriptor_create_from_buffer(&reply_view,
1968 &descriptor_reply);
1969 if (ret < 0) {
1970 ret_code = LTTNG_ERR_FATAL;
1971 goto end;
1972 }
1973 ret_code = LTTNG_OK;
1974 lttng_session_descriptor_assign(session_descriptor, descriptor_reply);
1975end:
1976 free(reply);
1977 lttng_dynamic_buffer_reset(&payload);
1978 lttng_session_descriptor_destroy(descriptor_reply);
1979 return ret_code;
1980}
1981
aaf97519 1982/*
b178f53e 1983 * Create a new session using name and url for destination.
a4b92340 1984 *
780d4bb8 1985 * Return 0 on success else a negative LTTng error code.
00e2e675 1986 */
a4b92340 1987int lttng_create_session(const char *name, const char *url)
00e2e675 1988{
3dd05a85 1989 int ret;
a4b92340 1990 ssize_t size;
a4b92340 1991 struct lttng_uri *uris = NULL;
b178f53e
JG
1992 struct lttng_session_descriptor *descriptor = NULL;
1993 enum lttng_error_code ret_code;
00e2e675 1994
b178f53e
JG
1995 if (!name) {
1996 ret = -LTTNG_ERR_INVALID;
1997 goto end;
00e2e675
DG
1998 }
1999
bc894455 2000 size = uri_parse_str_urls(url, NULL, &uris);
a4b92340 2001 if (size < 0) {
b178f53e
JG
2002 ret = -LTTNG_ERR_INVALID;
2003 goto end;
2004 }
2005 switch (size) {
2006 case 0:
2007 descriptor = lttng_session_descriptor_create(name);
2008 break;
2009 case 1:
2010 if (uris[0].dtype != LTTNG_DST_PATH) {
2011 ret = -LTTNG_ERR_INVALID;
2012 goto end;
2013 }
2014 descriptor = lttng_session_descriptor_local_create(name,
2015 uris[0].dst.path);
2016 break;
2017 case 2:
2018 descriptor = lttng_session_descriptor_network_create(name, url,
2019 NULL);
2020 break;
2021 default:
2022 ret = -LTTNG_ERR_INVALID;
2023 goto end;
2024 }
2025 if (!descriptor) {
2026 ret = -LTTNG_ERR_INVALID;
2027 goto end;
00e2e675 2028 }
b178f53e
JG
2029 ret_code = lttng_create_session_ext(descriptor);
2030 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2031end:
2032 lttng_session_descriptor_destroy(descriptor);
2033 free(uris);
2034 return ret;
2035}
00e2e675 2036
b178f53e
JG
2037/*
2038 * Create a session exclusively used for snapshot.
2039 *
780d4bb8 2040 * Return 0 on success else a negative LTTng error code.
b178f53e
JG
2041 */
2042int lttng_create_session_snapshot(const char *name, const char *snapshot_url)
2043{
2044 int ret;
2045 enum lttng_error_code ret_code;
2046 ssize_t size;
2047 struct lttng_uri *uris = NULL;
2048 struct lttng_session_descriptor *descriptor = NULL;
a4b92340 2049
b178f53e
JG
2050 if (!name) {
2051 ret = -LTTNG_ERR_INVALID;
2052 goto end;
2053 }
2054
2055 size = uri_parse_str_urls(snapshot_url, NULL, &uris);
2056 if (size < 0) {
2057 ret = -LTTNG_ERR_INVALID;
2058 goto end;
2059 }
2060 /*
2061 * If the user does not specify a custom subdir, use the session name.
2062 */
2063 if (size > 0 && uris[0].dtype != LTTNG_DST_PATH &&
2064 strlen(uris[0].subdir) == 0) {
2065 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s",
2066 name);
2067 if (ret < 0) {
2068 PERROR("Failed to set session name as network destination sub-directory");
2069 ret = -LTTNG_ERR_FATAL;
2070 goto end;
2071 } else if (ret >= sizeof(uris[0].subdir)) {
2072 /* Truncated output. */
2073 ret = -LTTNG_ERR_INVALID;
2074 goto end;
2075 }
2076 }
3dd05a85 2077
b178f53e
JG
2078 switch (size) {
2079 case 0:
2080 descriptor = lttng_session_descriptor_snapshot_create(name);
2081 break;
2082 case 1:
2083 if (uris[0].dtype != LTTNG_DST_PATH) {
2084 ret = -LTTNG_ERR_INVALID;
2085 goto end;
2086 }
2087 descriptor = lttng_session_descriptor_snapshot_local_create(
2088 name,
2089 uris[0].dst.path);
2090 break;
2091 case 2:
2092 descriptor = lttng_session_descriptor_snapshot_network_create(
2093 name,
2094 snapshot_url,
2095 NULL);
2096 break;
2097 default:
2098 ret = -LTTNG_ERR_INVALID;
2099 goto end;
2100 }
2101 if (!descriptor) {
2102 ret = -LTTNG_ERR_INVALID;
2103 goto end;
2104 }
2105 ret_code = lttng_create_session_ext(descriptor);
2106 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2107end:
2108 lttng_session_descriptor_destroy(descriptor);
3dd05a85
DG
2109 free(uris);
2110 return ret;
00e2e675
DG
2111}
2112
b178f53e
JG
2113/*
2114 * Create a session exclusively used for live.
2115 *
780d4bb8 2116 * Return 0 on success else a negative LTTng error code.
b178f53e
JG
2117 */
2118int lttng_create_session_live(const char *name, const char *url,
2119 unsigned int timer_interval)
2120{
2121 int ret;
2122 enum lttng_error_code ret_code;
2123 struct lttng_session_descriptor *descriptor = NULL;
2124
2125 if (!name) {
2126 ret = -LTTNG_ERR_INVALID;
2127 goto end;
2128 }
2129
2130 if (url) {
2131 descriptor = lttng_session_descriptor_live_network_create(
2132 name, url, NULL, timer_interval);
2133 } else {
2134 descriptor = lttng_session_descriptor_live_create(
2135 name, timer_interval);
2136 }
2137 if (!descriptor) {
2138 ret = -LTTNG_ERR_INVALID;
2139 goto end;
2140 }
2141 ret_code = lttng_create_session_ext(descriptor);
2142 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2143end:
2144 lttng_session_descriptor_destroy(descriptor);
2145 return ret;
2146}
2147
e20ee7c2
JD
2148/*
2149 * Stop the session and wait for the data before destroying it
780d4bb8
MD
2150 *
2151 * Return 0 on success else a negative LTTng error code.
e20ee7c2
JD
2152 */
2153int lttng_destroy_session(const char *session_name)
2154{
2155 int ret;
3e3665b8
JG
2156 enum lttng_error_code ret_code;
2157 enum lttng_destruction_handle_status status;
2158 struct lttng_destruction_handle *handle = NULL;
e20ee7c2
JD
2159
2160 /*
3e3665b8
JG
2161 * Stop the tracing and wait for the data to be
2162 * consumed.
e20ee7c2
JD
2163 */
2164 ret = _lttng_stop_tracing(session_name, 1);
2165 if (ret && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
2166 goto end;
2167 }
2168
3e3665b8
JG
2169 ret_code = lttng_destroy_session_ext(session_name, &handle);
2170 if (ret_code != LTTNG_OK) {
2171 ret = (int) -ret_code;
2172 goto end;
2173 }
a0377dfe 2174 LTTNG_ASSERT(handle);
3e3665b8
JG
2175
2176 /* Block until the completion of the destruction of the session. */
2177 status = lttng_destruction_handle_wait_for_completion(handle, -1);
2178 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_COMPLETED) {
2179 ret = -LTTNG_ERR_UNK;
2180 goto end;
2181 }
2182
2183 status = lttng_destruction_handle_get_result(handle, &ret_code);
2184 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_OK) {
2185 ret = -LTTNG_ERR_UNK;
2186 goto end;
2187 }
780d4bb8 2188 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
e20ee7c2 2189end:
3e3665b8 2190 lttng_destruction_handle_destroy(handle);
e20ee7c2
JD
2191 return ret;
2192}
2193
2194/*
2195 * Destroy the session without waiting for the data.
2196 */
2197int lttng_destroy_session_no_wait(const char *session_name)
2198{
3e3665b8 2199 enum lttng_error_code ret_code;
e20ee7c2 2200
3e3665b8 2201 ret_code = lttng_destroy_session_ext(session_name, NULL);
3204017e 2202 return ret_code == LTTNG_OK ? 0 : -ret_code;
e20ee7c2
JD
2203}
2204
57167058 2205/*
9ae110e2
JG
2206 * Ask the session daemon for all available sessions.
2207 * Sets the contents of the sessions array.
2208 * Returns the number of lttng_session entries in sessions;
2209 * on error, returns a negative value.
57167058 2210 */
b178f53e 2211int lttng_list_sessions(struct lttng_session **out_sessions)
57167058 2212{
ca95a216 2213 int ret;
2d321d3e
JR
2214 struct lttcomm_session_msg lsm = {};
2215 struct lttng_payload reply;
2216 struct lttng_payload_view lsm_view =
2217 lttng_payload_view_init_from_buffer((const char *) &lsm, 0, sizeof(lsm));
2218 unsigned int nb_sessions = 0;
57167058 2219
2d321d3e
JR
2220 lttng_payload_init(&reply);
2221
2222 /* Initialize command parameters. */
cd80958d 2223 lsm.cmd_type = LTTNG_LIST_SESSIONS;
2d321d3e
JR
2224
2225 /* Execute command against the session daemon. */
2226 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
2227 if (ret < 0) {
b178f53e
JG
2228 goto end;
2229 }
2230
2d321d3e
JR
2231 {
2232 const struct lttcomm_list_command_header *cmd_reply_header = NULL;
2233 const lttng_payload_view cmd_reply_header_view = lttng_payload_view_from_payload(
2234 &reply, 0, sizeof(*cmd_reply_header));
2235
2236 if (!lttng_payload_view_is_valid(&cmd_reply_header_view)) {
2237 ret = -LTTNG_ERR_INVALID_PROTOCOL;
2238 goto end;
2239 }
2240
2241 cmd_reply_header = (const struct lttcomm_list_command_header *)
2242 cmd_reply_header_view.buffer.data;
2243 if (cmd_reply_header->count > INT_MAX) {
2244 ret = -LTTNG_ERR_OVERFLOW;
2245 goto end;
2246 }
2247
2248 nb_sessions = (unsigned int) cmd_reply_header->count;
57167058
DG
2249 }
2250
2d321d3e
JR
2251 {
2252 enum lttng_error_code ret_code;
2253 lttng_payload_view cmd_reply_payload = lttng_payload_view_from_payload(
2254 &reply, sizeof(struct lttcomm_list_command_header), -1);
b178f53e 2255
2d321d3e
JR
2256 ret_code = lttng_sessions_create_and_flatten_from_payload(
2257 &cmd_reply_payload, nb_sessions, out_sessions);
2258 if (ret_code != LTTNG_OK) {
2259 ret = -((int) ret_code);
2260 goto end;
2261 }
b178f53e
JG
2262 }
2263
2d321d3e 2264 ret = (int) nb_sessions;
b178f53e 2265end:
2d321d3e 2266 lttng_payload_reset(&reply);
b178f53e
JG
2267 return ret;
2268}
2269
2270enum lttng_error_code lttng_session_get_creation_time(
2271 const struct lttng_session *session, uint64_t *creation_time)
2272{
2273 enum lttng_error_code ret = LTTNG_OK;
2274 struct lttng_session_extended *extended;
2275
2276 if (!session || !creation_time || !session->extended.ptr) {
2277 ret = LTTNG_ERR_INVALID;
2278 goto end;
2279 }
2280
4bd69c5f 2281 extended = (lttng_session_extended *) session->extended.ptr;
b178f53e
JG
2282 if (!extended->creation_time.is_set) {
2283 /* Not created on the session daemon yet. */
2284 ret = LTTNG_ERR_SESSION_NOT_EXIST;
2285 goto end;
2286 }
2287 *creation_time = extended->creation_time.value;
2288end:
2289 return ret;
57167058
DG
2290}
2291
d7ba1388
MD
2292int lttng_set_session_shm_path(const char *session_name,
2293 const char *shm_path)
2294{
e1b624d0 2295 int ret;
d7ba1388
MD
2296 struct lttcomm_session_msg lsm;
2297
2298 if (session_name == NULL) {
2299 return -LTTNG_ERR_INVALID;
2300 }
2301
2302 memset(&lsm, 0, sizeof(lsm));
2303 lsm.cmd_type = LTTNG_SET_SESSION_SHM_PATH;
2304
e1b624d0 2305 ret = lttng_strncpy(lsm.session.name, session_name,
d7ba1388 2306 sizeof(lsm.session.name));
e1b624d0
JG
2307 if (ret) {
2308 ret = -LTTNG_ERR_INVALID;
2309 goto end;
2310 }
2311
2312 ret = lttng_strncpy(lsm.u.set_shm_path.shm_path, shm_path ?: "",
d7ba1388 2313 sizeof(lsm.u.set_shm_path.shm_path));
e1b624d0
JG
2314 if (ret) {
2315 ret = -LTTNG_ERR_INVALID;
2316 goto end;
2317 }
d7ba1388 2318
e1b624d0
JG
2319 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
2320end:
2321 return ret;
d7ba1388
MD
2322}
2323
9f19cc17 2324/*
9ae110e2
JG
2325 * Ask the session daemon for all available domains of a session.
2326 * Sets the contents of the domains array.
2327 * Returns the number of lttng_domain entries in domains;
2328 * on error, returns a negative value.
9f19cc17 2329 */
330be774 2330int lttng_list_domains(const char *session_name,
cd80958d 2331 struct lttng_domain **domains)
9f19cc17
DG
2332{
2333 int ret;
cd80958d
DG
2334 struct lttcomm_session_msg lsm;
2335
330be774 2336 if (session_name == NULL) {
e1b624d0
JG
2337 ret = -LTTNG_ERR_INVALID;
2338 goto error;
cd80958d 2339 }
9f19cc17 2340
53efb85a 2341 memset(&lsm, 0, sizeof(lsm));
cd80958d
DG
2342 lsm.cmd_type = LTTNG_LIST_DOMAINS;
2343
e1b624d0 2344 ret = lttng_strncpy(lsm.session.name, session_name,
cac3069d 2345 sizeof(lsm.session.name));
e1b624d0
JG
2346 if (ret) {
2347 ret = -LTTNG_ERR_INVALID;
2348 goto error;
2349 }
cd80958d 2350
cac3069d 2351 ret = lttng_ctl_ask_sessiond(&lsm, (void**) domains);
9f19cc17 2352 if (ret < 0) {
e1b624d0 2353 goto error;
9f19cc17
DG
2354 }
2355
2356 return ret / sizeof(struct lttng_domain);
e1b624d0
JG
2357error:
2358 return ret;
9f19cc17
DG
2359}
2360
2361/*
9ae110e2
JG
2362 * Ask the session daemon for all available channels of a session.
2363 * Sets the contents of the channels array.
2364 * Returns the number of lttng_channel entries in channels;
2365 * on error, returns a negative value.
9f19cc17 2366 */
cd80958d
DG
2367int lttng_list_channels(struct lttng_handle *handle,
2368 struct lttng_channel **channels)
9f19cc17 2369{
999af9c1 2370 int ret, total_payload_received;
cd80958d 2371 struct lttcomm_session_msg lsm;
999af9c1
JR
2372 char *reception_buffer = NULL;
2373 size_t cmd_header_len = 0;
2374 struct lttcomm_list_command_header *cmd_header = NULL;
2375 struct lttng_dynamic_buffer tmp_buffer;
2376
2377 lttng_dynamic_buffer_init(&tmp_buffer);
cd80958d 2378
9d697d3d 2379 if (handle == NULL) {
53e367f9
JG
2380 ret = -LTTNG_ERR_INVALID;
2381 goto end;
cd80958d
DG
2382 }
2383
53efb85a 2384 memset(&lsm, 0, sizeof(lsm));
cd80958d 2385 lsm.cmd_type = LTTNG_LIST_CHANNELS;
e1b624d0 2386 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 2387 sizeof(lsm.session.name));
e1b624d0
JG
2388 if (ret) {
2389 ret = -LTTNG_ERR_INVALID;
2390 goto end;
2391 }
9f19cc17 2392
60160d2a 2393 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
9f19cc17 2394
999af9c1
JR
2395 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm, NULL, 0, NULL, 0,
2396 (void **) &reception_buffer, (void **) &cmd_header,
2397 &cmd_header_len);
9f19cc17 2398 if (ret < 0) {
53e367f9
JG
2399 goto end;
2400 }
2401
999af9c1
JR
2402 total_payload_received = ret;
2403
2404 if (cmd_header_len != sizeof(*cmd_header)) {
2405 ret = -LTTNG_ERR_FATAL;
53e367f9 2406 goto end;
9f19cc17
DG
2407 }
2408
999af9c1
JR
2409 if (!cmd_header) {
2410 ret = LTTNG_ERR_UNK;
2411 goto end;
2412 }
2413
2414 if (cmd_header->count > INT_MAX) {
2415 ret = -LTTNG_ERR_OVERFLOW;
2416 goto end;
2417 }
53e367f9 2418
999af9c1
JR
2419 {
2420 enum lttng_error_code ret_code;
2421 const struct lttng_buffer_view events_view =
2422 lttng_buffer_view_init(reception_buffer, 0,
2423 total_payload_received);
2424
2425 ret_code = lttng_channels_create_and_flatten_from_buffer(
2426 &events_view, cmd_header->count, channels);
2427 if (ret_code != LTTNG_OK) {
2428 ret = -ret_code;
2429 goto end;
2430 }
53e367f9
JG
2431 }
2432
999af9c1 2433 ret = (int) cmd_header->count;
53e367f9 2434end:
999af9c1
JR
2435 free(cmd_header);
2436 free(reception_buffer);
53e367f9 2437 return ret;
9f19cc17
DG
2438}
2439
2440/*
9ae110e2
JG
2441 * Ask the session daemon for all available events of a session channel.
2442 * Sets the contents of the events array.
2443 * Returns the number of lttng_event entries in events;
2444 * on error, returns a negative value.
9f19cc17 2445 */
cd80958d
DG
2446int lttng_list_events(struct lttng_handle *handle,
2447 const char *channel_name, struct lttng_event **events)
9f19cc17
DG
2448{
2449 int ret;
e368fb43 2450 struct lttcomm_session_msg lsm = {};
8ddd72ef 2451 struct lttng_payload reply;
e368fb43
JG
2452 struct lttng_payload_view lsm_view =
2453 lttng_payload_view_init_from_buffer(
2454 (const char *) &lsm, 0, sizeof(lsm));
8ddd72ef 2455 unsigned int nb_events = 0;
9f19cc17 2456
821a8b53
JR
2457 lttng_payload_init(&reply);
2458
8ddd72ef 2459 /* Safety check. An handle and channel name are mandatory. */
9d697d3d 2460 if (handle == NULL || channel_name == NULL) {
e368fb43
JG
2461 ret = -LTTNG_ERR_INVALID;
2462 goto end;
cd80958d
DG
2463 }
2464
8ddd72ef 2465 /* Initialize command parameters. */
cd80958d 2466 lsm.cmd_type = LTTNG_LIST_EVENTS;
e1b624d0 2467 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 2468 sizeof(lsm.session.name));
e1b624d0
JG
2469 if (ret) {
2470 ret = -LTTNG_ERR_INVALID;
2471 goto end;
2472 }
2473
2474 ret = lttng_strncpy(lsm.u.list.channel_name, channel_name,
cd80958d 2475 sizeof(lsm.u.list.channel_name));
e1b624d0
JG
2476 if (ret) {
2477 ret = -LTTNG_ERR_INVALID;
2478 goto end;
2479 }
2480
60160d2a 2481 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
9f19cc17 2482
8ddd72ef
JR
2483 /* Execute command against the session daemon. */
2484 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
9f19cc17 2485 if (ret < 0) {
de453daa 2486 goto end;
9f19cc17
DG
2487 }
2488
e368fb43 2489 {
8ddd72ef
JR
2490 const struct lttcomm_list_command_header *cmd_reply_header =
2491 NULL;
2492 const lttng_payload_view cmd_reply_header_view =
2493 lttng_payload_view_from_payload(&reply, 0,
2494 sizeof(*cmd_reply_header));
2495
2496 if (!lttng_payload_view_is_valid(&cmd_reply_header_view)) {
2497 ret = -LTTNG_ERR_INVALID_PROTOCOL;
2498 goto end;
56f0bc67 2499 }
b4e3ceb9 2500
8ddd72ef
JR
2501 cmd_reply_header = (const struct lttcomm_list_command_header *)
2502 cmd_reply_header_view.buffer
2503 .data;
2504 if (cmd_reply_header->count > INT_MAX) {
2505 ret = -LTTNG_ERR_OVERFLOW;
2506 goto end;
2507 }
56f0bc67 2508
8ddd72ef 2509 nb_events = (unsigned int) cmd_reply_header->count;
56f0bc67 2510 }
de453daa 2511
e368fb43 2512 {
8ddd72ef
JR
2513 enum lttng_error_code ret_code;
2514 lttng_payload_view cmd_reply_payload = lttng_payload_view_from_payload(
2515 &reply,
2516 sizeof(struct lttcomm_list_command_header), -1);
56f0bc67 2517
8ddd72ef
JR
2518 ret_code = lttng_events_create_and_flatten_from_payload(
2519 &cmd_reply_payload, nb_events, events);
2520 if (ret_code != LTTNG_OK) {
2521 ret = -((int) ret_code);
2522 goto end;
56f0bc67 2523 }
de453daa
JG
2524 }
2525
de453daa
JG
2526 ret = (int) nb_events;
2527end:
8ddd72ef 2528 lttng_payload_reset(&reply);
b4e3ceb9 2529 return ret;
9f19cc17
DG
2530}
2531
fac6795d 2532/*
1c8d13c8
TD
2533 * Sets the tracing_group variable with name.
2534 * This function allocates memory pointed to by tracing_group.
2535 * On success, returns 0, on error, returns -1 (null name) or -ENOMEM.
fac6795d
DG
2536 */
2537int lttng_set_tracing_group(const char *name)
2538{
e32a3d0f 2539 int ret = 0;
76da29b6 2540 char *new_group;
e32a3d0f 2541
9d697d3d 2542 if (name == NULL) {
e32a3d0f
JG
2543 ret = -LTTNG_ERR_INVALID;
2544 goto end;
9d697d3d
DG
2545 }
2546
e32a3d0f
JG
2547 new_group = strdup(name);
2548 if (!new_group) {
2549 ret = -LTTNG_ERR_FATAL;
2550 goto end;
fac6795d
DG
2551 }
2552
76da29b6
JR
2553 free(tracing_group);
2554 tracing_group = new_group;
2555 new_group = NULL;
2556
e32a3d0f
JG
2557end:
2558 return ret;
fac6795d
DG
2559}
2560
f46376a1
MJ
2561int lttng_calibrate(struct lttng_handle *handle __attribute__((unused)),
2562 struct lttng_calibrate *calibrate __attribute__((unused)))
d0254c7c 2563{
b812e5ca
PP
2564 /*
2565 * This command was removed in LTTng 2.9.
2566 */
2567 return -LTTNG_ERR_UND;
d0254c7c
MD
2568}
2569
5edd7e09
DG
2570/*
2571 * Set default channel attributes.
441c16a7 2572 * If either or both of the arguments are null, attr content is zeroe'd.
5edd7e09
DG
2573 */
2574void lttng_channel_set_default_attr(struct lttng_domain *domain,
2575 struct lttng_channel_attr *attr)
2576{
294851b0 2577 struct lttng_channel_extended *extended;
cf0bcb51 2578
5edd7e09
DG
2579 /* Safety check */
2580 if (attr == NULL || domain == NULL) {
2581 return;
2582 }
2583
999af9c1 2584 /* Save the pointer for later use */
294851b0 2585 extended = (struct lttng_channel_extended *) attr->extended.ptr;
eacaa7a6
DS
2586 memset(attr, 0, sizeof(struct lttng_channel_attr));
2587
0a9c6494
DG
2588 /* Same for all domains. */
2589 attr->overwrite = DEFAULT_CHANNEL_OVERWRITE;
2590 attr->tracefile_size = DEFAULT_CHANNEL_TRACEFILE_SIZE;
2591 attr->tracefile_count = DEFAULT_CHANNEL_TRACEFILE_COUNT;
2592
5edd7e09
DG
2593 switch (domain->type) {
2594 case LTTNG_DOMAIN_KERNEL:
9ae110e2
JG
2595 attr->switch_timer_interval =
2596 DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER;
d92ff3ef 2597 attr->read_timer_interval = DEFAULT_KERNEL_CHANNEL_READ_TIMER;
3e230f92 2598 attr->subbuf_size = default_get_kernel_channel_subbuf_size();
5edd7e09
DG
2599 attr->num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
2600 attr->output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
2601 break;
2602 case LTTNG_DOMAIN_UST:
0a9c6494
DG
2603 switch (domain->buf_type) {
2604 case LTTNG_BUFFER_PER_UID:
2605 attr->subbuf_size = default_get_ust_uid_channel_subbuf_size();
2606 attr->num_subbuf = DEFAULT_UST_UID_CHANNEL_SUBBUF_NUM;
2607 attr->output = DEFAULT_UST_UID_CHANNEL_OUTPUT;
9ae110e2
JG
2608 attr->switch_timer_interval =
2609 DEFAULT_UST_UID_CHANNEL_SWITCH_TIMER;
2610 attr->read_timer_interval =
2611 DEFAULT_UST_UID_CHANNEL_READ_TIMER;
0a9c6494
DG
2612 break;
2613 case LTTNG_BUFFER_PER_PID:
2614 default:
2615 attr->subbuf_size = default_get_ust_pid_channel_subbuf_size();
2616 attr->num_subbuf = DEFAULT_UST_PID_CHANNEL_SUBBUF_NUM;
2617 attr->output = DEFAULT_UST_PID_CHANNEL_OUTPUT;
9ae110e2
JG
2618 attr->switch_timer_interval =
2619 DEFAULT_UST_PID_CHANNEL_SWITCH_TIMER;
2620 attr->read_timer_interval =
2621 DEFAULT_UST_PID_CHANNEL_READ_TIMER;
0a9c6494
DG
2622 break;
2623 }
5edd7e09 2624 default:
441c16a7 2625 /* Default behavior: leave set to 0. */
5edd7e09
DG
2626 break;
2627 }
cf0bcb51 2628
999af9c1
JR
2629 if (extended) {
2630 lttng_channel_set_default_extended_attr(domain, extended);
2631 }
2632
2633 /* Reassign the extended pointer. */
cf0bcb51 2634 attr->extended.ptr = extended;
5edd7e09
DG
2635}
2636
5ba3702f
JG
2637int lttng_channel_get_discarded_event_count(struct lttng_channel *channel,
2638 uint64_t *discarded_events)
2639{
2640 int ret = 0;
cf0bcb51 2641 struct lttng_channel_extended *chan_ext;
5ba3702f
JG
2642
2643 if (!channel || !discarded_events) {
2644 ret = -LTTNG_ERR_INVALID;
2645 goto end;
2646 }
2647
4bd69c5f 2648 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
5ba3702f
JG
2649 if (!chan_ext) {
2650 /*
2651 * This can happen since the lttng_channel structure is
2652 * used for other tasks where this pointer is never set.
2653 */
2654 *discarded_events = 0;
2655 goto end;
2656 }
2657
2658 *discarded_events = chan_ext->discarded_events;
2659end:
2660 return ret;
2661}
2662
2663int lttng_channel_get_lost_packet_count(struct lttng_channel *channel,
2664 uint64_t *lost_packets)
2665{
2666 int ret = 0;
cf0bcb51 2667 struct lttng_channel_extended *chan_ext;
5ba3702f
JG
2668
2669 if (!channel || !lost_packets) {
2670 ret = -LTTNG_ERR_INVALID;
2671 goto end;
2672 }
2673
4bd69c5f 2674 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
5ba3702f
JG
2675 if (!chan_ext) {
2676 /*
2677 * This can happen since the lttng_channel structure is
2678 * used for other tasks where this pointer is never set.
2679 */
2680 *lost_packets = 0;
2681 goto end;
2682 }
2683
2684 *lost_packets = chan_ext->lost_packets;
2685end:
2686 return ret;
2687}
2688
cf0bcb51
JG
2689int lttng_channel_get_monitor_timer_interval(struct lttng_channel *chan,
2690 uint64_t *monitor_timer_interval)
2691{
2692 int ret = 0;
2693
2694 if (!chan || !monitor_timer_interval) {
2695 ret = -LTTNG_ERR_INVALID;
2696 goto end;
2697 }
2698
2699 if (!chan->attr.extended.ptr) {
2700 ret = -LTTNG_ERR_INVALID;
2701 goto end;
2702 }
2703
2704 *monitor_timer_interval = ((struct lttng_channel_extended *)
2705 chan->attr.extended.ptr)->monitor_timer_interval;
2706end:
2707 return ret;
2708}
2709
2710int lttng_channel_set_monitor_timer_interval(struct lttng_channel *chan,
2711 uint64_t monitor_timer_interval)
2712{
2713 int ret = 0;
2714
2715 if (!chan || !chan->attr.extended.ptr) {
2716 ret = -LTTNG_ERR_INVALID;
2717 goto end;
2718 }
2719
2720 ((struct lttng_channel_extended *)
2721 chan->attr.extended.ptr)->monitor_timer_interval =
2722 monitor_timer_interval;
2723end:
2724 return ret;
2725}
2726
491d1539
MD
2727int lttng_channel_get_blocking_timeout(struct lttng_channel *chan,
2728 int64_t *blocking_timeout)
2729{
2730 int ret = 0;
2731
2732 if (!chan || !blocking_timeout) {
2733 ret = -LTTNG_ERR_INVALID;
2734 goto end;
2735 }
2736
2737 if (!chan->attr.extended.ptr) {
2738 ret = -LTTNG_ERR_INVALID;
2739 goto end;
2740 }
2741
2742 *blocking_timeout = ((struct lttng_channel_extended *)
2743 chan->attr.extended.ptr)->blocking_timeout;
2744end:
2745 return ret;
2746}
2747
2748int lttng_channel_set_blocking_timeout(struct lttng_channel *chan,
2749 int64_t blocking_timeout)
2750{
2751 int ret = 0;
2752 int64_t msec_timeout;
2753
2754 if (!chan || !chan->attr.extended.ptr) {
2755 ret = -LTTNG_ERR_INVALID;
2756 goto end;
2757 }
2758
2759 if (blocking_timeout < 0 && blocking_timeout != -1) {
2760 ret = -LTTNG_ERR_INVALID;
2761 goto end;
2762 }
2763
2764 /*
2765 * LTTng-ust's use of poll() to implement this timeout mechanism forces
2766 * us to accept a narrower range of values (msecs expressed as a signed
2767 * 32-bit integer).
2768 */
2769 msec_timeout = blocking_timeout / 1000;
2770 if (msec_timeout != (int32_t) msec_timeout) {
2771 ret = -LTTNG_ERR_INVALID;
2772 goto end;
2773 }
2774
2775 ((struct lttng_channel_extended *)
2776 chan->attr.extended.ptr)->blocking_timeout =
2777 blocking_timeout;
2778end:
2779 return ret;
2780}
2781
fac6795d 2782/*
2269e89e 2783 * Check if session daemon is alive.
fac6795d 2784 *
2269e89e 2785 * Return 1 if alive or 0 if not.
1c8d13c8 2786 * On error returns a negative value.
fac6795d 2787 */
947308c4 2788int lttng_session_daemon_alive(void)
fac6795d
DG
2789{
2790 int ret;
2791
2792 ret = set_session_daemon_path();
2793 if (ret < 0) {
9ae110e2 2794 /* Error. */
fac6795d
DG
2795 return ret;
2796 }
2797
9d035200 2798 if (*sessiond_sock_path == '\0') {
2f70b271 2799 /*
9ae110e2
JG
2800 * No socket path set. Weird error which means the constructor
2801 * was not called.
2f70b271 2802 */
a0377dfe 2803 abort();
fac6795d
DG
2804 }
2805
2269e89e 2806 ret = try_connect_sessiond(sessiond_sock_path);
7d8234d9 2807 if (ret < 0) {
9ae110e2 2808 /* Not alive. */
7d8234d9
MD
2809 return 0;
2810 }
7d8234d9 2811
9ae110e2 2812 /* Is alive. */
947308c4 2813 return 1;
fac6795d
DG
2814}
2815
00e2e675 2816/*
a4b92340 2817 * Set URL for a consumer for a session and domain.
00e2e675
DG
2818 *
2819 * Return 0 on success, else a negative value.
2820 */
a4b92340
DG
2821int lttng_set_consumer_url(struct lttng_handle *handle,
2822 const char *control_url, const char *data_url)
00e2e675 2823{
3dd05a85 2824 int ret;
a4b92340 2825 ssize_t size;
00e2e675 2826 struct lttcomm_session_msg lsm;
a4b92340 2827 struct lttng_uri *uris = NULL;
00e2e675 2828
a4b92340 2829 if (handle == NULL || (control_url == NULL && data_url == NULL)) {
e1b624d0
JG
2830 ret = -LTTNG_ERR_INVALID;
2831 goto error;
00e2e675
DG
2832 }
2833
a4b92340
DG
2834 memset(&lsm, 0, sizeof(lsm));
2835
00e2e675
DG
2836 lsm.cmd_type = LTTNG_SET_CONSUMER_URI;
2837
e1b624d0 2838 ret = lttng_strncpy(lsm.session.name, handle->session_name,
00e2e675 2839 sizeof(lsm.session.name));
e1b624d0
JG
2840 if (ret) {
2841 ret = -LTTNG_ERR_INVALID;
2842 goto error;
2843 }
2844
60160d2a 2845 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
00e2e675 2846
bc894455 2847 size = uri_parse_str_urls(control_url, data_url, &uris);
a4b92340 2848 if (size < 0) {
e1b624d0
JG
2849 ret = -LTTNG_ERR_INVALID;
2850 goto error;
a4b92340
DG
2851 }
2852
2853 lsm.u.uri.size = size;
00e2e675 2854
795a978d 2855 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, uris,
cac3069d 2856 sizeof(struct lttng_uri) * size, NULL);
3dd05a85
DG
2857
2858 free(uris);
e1b624d0 2859error:
3dd05a85 2860 return ret;
00e2e675
DG
2861}
2862
2863/*
9c6bda17 2864 * [OBSOLETE]
00e2e675 2865 */
4bd69c5f
SM
2866extern "C"
2867LTTNG_EXPORT int lttng_enable_consumer(struct lttng_handle *handle);
f46376a1 2868int lttng_enable_consumer(struct lttng_handle *handle __attribute__((unused)))
00e2e675 2869{
785d2d0d 2870 return -ENOSYS;
00e2e675
DG
2871}
2872
2873/*
9c6bda17 2874 * [OBSOLETE]
00e2e675 2875 */
4bd69c5f
SM
2876extern "C"
2877LTTNG_EXPORT int lttng_disable_consumer(struct lttng_handle *handle);
f46376a1 2878int lttng_disable_consumer(struct lttng_handle *handle __attribute__((unused)))
00e2e675 2879{
785d2d0d 2880 return -ENOSYS;
00e2e675
DG
2881}
2882
07424f16 2883/*
b178f53e 2884 * [OBSOLETE]
07424f16 2885 */
4bd69c5f
SM
2886extern "C"
2887LTTNG_EXPORT int _lttng_create_session_ext(const char *name, const char *url,
2ec40111 2888 const char *datetime);
f46376a1
MJ
2889int _lttng_create_session_ext(const char *name __attribute__((unused)),
2890 const char *url __attribute__((unused)),
2891 const char *datetime __attribute__((unused)))
07424f16 2892{
b178f53e 2893 return -ENOSYS;
07424f16
DG
2894}
2895
806e2684
DG
2896/*
2897 * For a given session name, this call checks if the data is ready to be read
2898 * or is still being extracted by the consumer(s) hence not ready to be used by
2899 * any readers.
2900 */
6d805429 2901int lttng_data_pending(const char *session_name)
806e2684
DG
2902{
2903 int ret;
2904 struct lttcomm_session_msg lsm;
f6151c55 2905 uint8_t *pending = NULL;
806e2684
DG
2906
2907 if (session_name == NULL) {
2908 return -LTTNG_ERR_INVALID;
2909 }
2910
53efb85a 2911 memset(&lsm, 0, sizeof(lsm));
6d805429 2912 lsm.cmd_type = LTTNG_DATA_PENDING;
806e2684 2913
e1b624d0 2914 ret = lttng_strncpy(lsm.session.name, session_name,
cac3069d 2915 sizeof(lsm.session.name));
e1b624d0
JG
2916 if (ret) {
2917 ret = -LTTNG_ERR_INVALID;
2918 goto end;
2919 }
806e2684 2920
f6151c55
JG
2921 ret = lttng_ctl_ask_sessiond(&lsm, (void **) &pending);
2922 if (ret < 0) {
2923 goto end;
2924 } else if (ret != 1) {
2925 /* Unexpected payload size */
2926 ret = -LTTNG_ERR_INVALID;
2927 goto end;
e848d36d
JG
2928 } else if (!pending) {
2929 /* Internal error. */
2930 ret = -LTTNG_ERR_UNK;
2931 goto end;
806e2684
DG
2932 }
2933
f6151c55
JG
2934 ret = (int) *pending;
2935end:
2936 free(pending);
806e2684
DG
2937 return ret;
2938}
2939
93ec662e
JD
2940/*
2941 * Regenerate the metadata for a session.
2942 * Return 0 on success, a negative error code on error.
2943 */
eded6438 2944int lttng_regenerate_metadata(const char *session_name)
93ec662e
JD
2945{
2946 int ret;
2947 struct lttcomm_session_msg lsm;
2948
2949 if (!session_name) {
2950 ret = -LTTNG_ERR_INVALID;
2951 goto end;
2952 }
2953
2954 memset(&lsm, 0, sizeof(lsm));
eded6438 2955 lsm.cmd_type = LTTNG_REGENERATE_METADATA;
93ec662e 2956
e1b624d0 2957 ret = lttng_strncpy(lsm.session.name, session_name,
93ec662e 2958 sizeof(lsm.session.name));
e1b624d0
JG
2959 if (ret) {
2960 ret = -LTTNG_ERR_INVALID;
2961 goto end;
2962 }
93ec662e
JD
2963
2964 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
2965 if (ret < 0) {
2966 goto end;
2967 }
2968
2969 ret = 0;
2970end:
2971 return ret;
2972}
2973
eded6438
JD
2974/*
2975 * Deprecated, replaced by lttng_regenerate_metadata.
2976 */
2977int lttng_metadata_regenerate(const char *session_name)
2978{
2979 return lttng_regenerate_metadata(session_name);
2980}
2981
c2561365
JD
2982/*
2983 * Regenerate the statedump of a session.
2984 * Return 0 on success, a negative error code on error.
2985 */
2986int lttng_regenerate_statedump(const char *session_name)
2987{
2988 int ret;
2989 struct lttcomm_session_msg lsm;
2990
2991 if (!session_name) {
2992 ret = -LTTNG_ERR_INVALID;
2993 goto end;
2994 }
2995
2996 memset(&lsm, 0, sizeof(lsm));
2997 lsm.cmd_type = LTTNG_REGENERATE_STATEDUMP;
2998
e1b624d0 2999 ret = lttng_strncpy(lsm.session.name, session_name,
c2561365 3000 sizeof(lsm.session.name));
e1b624d0
JG
3001 if (ret) {
3002 ret = -LTTNG_ERR_INVALID;
3003 goto end;
3004 }
c2561365
JD
3005
3006 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
3007 if (ret < 0) {
3008 goto end;
3009 }
3010
3011 ret = 0;
3012end:
3013 return ret;
3014}
3015
a5c2d2a7
JG
3016static
3017int _lttng_register_trigger(struct lttng_trigger *trigger, const char *name,
3018 bool generate_name)
a58c490f
JG
3019{
3020 int ret;
99608320
JR
3021 struct lttcomm_session_msg lsm = {
3022 .cmd_type = LTTNG_REGISTER_TRIGGER,
1c9a0b0e
MJ
3023 .session = {},
3024 .domain = {},
3025 .u = {},
3026 .fd_count = 0,
99608320 3027 };
4bd69c5f 3028 lsm.u.trigger.is_trigger_anonymous = !name && !generate_name;
99608320
JR
3029 struct lttcomm_session_msg *message_lsm;
3030 struct lttng_payload message;
3031 struct lttng_payload reply;
242388e4 3032 struct lttng_trigger *reply_trigger = NULL;
9124c630 3033 enum lttng_domain_type domain_type;
64eafdf6
JR
3034 const struct lttng_credentials user_creds = {
3035 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3036 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3037 };
a5c2d2a7
JG
3038 const char *unused_trigger_name = NULL;
3039 enum lttng_trigger_status trigger_status;
99608320
JR
3040
3041 lttng_payload_init(&message);
3042 lttng_payload_init(&reply);
a58c490f
JG
3043
3044 if (!trigger) {
3045 ret = -LTTNG_ERR_INVALID;
3046 goto end;
3047 }
3048
a5c2d2a7
JG
3049 trigger_status = lttng_trigger_get_name(trigger, &unused_trigger_name);
3050 if (trigger_status != LTTNG_TRIGGER_STATUS_UNSET) {
3051 /* Re-using already registered trigger. */
3052 ret = -LTTNG_ERR_INVALID;
3053 goto end;
3054 }
3055
3056 if (name) {
3057 trigger_status = lttng_trigger_set_name(trigger, name);
3058 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3059 ret = -LTTNG_ERR_NOMEM;
3060 goto end;
3061 }
3062 }
3063
64eafdf6
JR
3064 if (!trigger->creds.uid.is_set) {
3065 /* Use the client's credentials as the trigger credentials. */
3066 lttng_trigger_set_credentials(trigger, &user_creds);
3067 } else {
3068 /*
3069 * Validate that either the current trigger credentials and the
3070 * client credentials are identical or that the current user is
3071 * root. The root user can register, unregister triggers for
3072 * himself and other users.
3073 *
3074 * This check is also present on the sessiond side, using the
3075 * credentials passed on the socket. These check are all
3076 * "safety" checks.
3077 */
3078 const struct lttng_credentials *trigger_creds =
3079 lttng_trigger_get_credentials(trigger);
3080
3081 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3082 if (lttng_credentials_get_uid(&user_creds) != 0) {
3083 ret = -LTTNG_ERR_EPERM;
a5c2d2a7 3084 goto end_unset_name;
64eafdf6
JR
3085 }
3086 }
3087 }
3088
a58c490f 3089 if (!lttng_trigger_validate(trigger)) {
eac4828d 3090 ret = -LTTNG_ERR_INVALID_TRIGGER;
a5c2d2a7 3091 goto end_unset_name;
a58c490f
JG
3092 }
3093
9124c630
JR
3094 domain_type = lttng_trigger_get_underlying_domain_type_restriction(
3095 trigger);
3096
3097 lsm.domain.type = domain_type;
3098
97285430
JG
3099 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3100 if (ret) {
3101 ret = -LTTNG_ERR_NOMEM;
a5c2d2a7 3102 goto end_unset_name;
97285430 3103 }
99608320 3104
99608320 3105 ret = lttng_trigger_serialize(trigger, &message);
3647288f 3106 if (ret < 0) {
a58c490f 3107 ret = -LTTNG_ERR_UNK;
a5c2d2a7 3108 goto end_unset_name;
a58c490f
JG
3109 }
3110
b22f4f54
JG
3111 /*
3112 * This is needed to populate the trigger object size for the command
3113 * header.
3114 */
3115 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3116
99608320
JR
3117 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3118
3119 {
3120 struct lttng_payload_view message_view =
3121 lttng_payload_view_from_payload(
3122 &message, 0, -1);
3123
3124 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3125 &message_view);
3126 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3127 if (ret < 0) {
a5c2d2a7 3128 goto end_unset_name;
99608320
JR
3129 }
3130 }
3131
242388e4
JR
3132 {
3133 struct lttng_payload_view reply_view =
3134 lttng_payload_view_from_payload(
3135 &reply, 0, reply.buffer.size);
3136
3137 ret = lttng_trigger_create_from_payload(
3138 &reply_view, &reply_trigger);
3139 if (ret < 0) {
a5c2d2a7
JG
3140 ret = -LTTNG_ERR_INVALID_PROTOCOL;
3141 goto end_unset_name;
242388e4
JR
3142 }
3143 }
3144
a5c2d2a7
JG
3145 if (name || generate_name) {
3146 ret = lttng_trigger_assign_name(trigger, reply_trigger);
3147 if (ret < 0) {
3148 ret = -LTTNG_ERR_NOMEM;
3149 goto end;
3150 }
242388e4
JR
3151 }
3152
99608320 3153 ret = 0;
a5c2d2a7
JG
3154 goto end;
3155
3156end_unset_name:
3157 trigger_status = lttng_trigger_set_name(trigger, NULL);
3158 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3159 ret = -LTTNG_ERR_UNK;
3160 }
a58c490f 3161end:
99608320
JR
3162 lttng_payload_reset(&message);
3163 lttng_payload_reset(&reply);
242388e4 3164 lttng_trigger_destroy(reply_trigger);
a58c490f
JG
3165 return ret;
3166}
3167
a5c2d2a7
JG
3168int lttng_register_trigger(struct lttng_trigger *trigger)
3169{
3170 /* Register an anonymous trigger. */
3171 return _lttng_register_trigger(trigger, NULL, false);
3172}
3173
3174enum lttng_error_code lttng_register_trigger_with_name(
3175 struct lttng_trigger *trigger, const char *name)
3176{
3177 const int ret = _lttng_register_trigger(trigger, name, false);
3178
3179 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3180}
3181
3182enum lttng_error_code lttng_register_trigger_with_automatic_name(
3183 struct lttng_trigger *trigger)
3184{
4bd69c5f 3185 const int ret = _lttng_register_trigger(trigger, nullptr, true);
a5c2d2a7
JG
3186
3187 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3188}
3189
b99a0cb3
JG
3190enum lttng_error_code lttng_error_query_execute(
3191 const struct lttng_error_query *query,
3192 const struct lttng_endpoint *endpoint,
3193 struct lttng_error_query_results **results)
3194{
3195 int ret;
3196 enum lttng_error_code ret_code;
3197 struct lttcomm_session_msg lsm = {
3198 .cmd_type = LTTNG_EXECUTE_ERROR_QUERY,
1c9a0b0e
MJ
3199 .session = {},
3200 .domain = {},
3201 .u = {},
3202 .fd_count = 0,
b99a0cb3
JG
3203 };
3204 struct lttng_payload message;
3205 struct lttng_payload reply;
3206 struct lttcomm_session_msg *message_lsm;
3207
3208 lttng_payload_init(&message);
3209 lttng_payload_init(&reply);
3210
3211 if (!query || !results) {
3212 ret_code = LTTNG_ERR_INVALID;
3213 goto end;
3214 }
3215
3216 if (endpoint != lttng_session_daemon_command_endpoint) {
3217 ret_code = LTTNG_ERR_INVALID_ERROR_QUERY_TARGET;
3218 goto end;
3219 }
3220
3221 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3222 if (ret) {
3223 ret_code = LTTNG_ERR_NOMEM;
3224 goto end;
3225 }
3226
3227 ret = lttng_error_query_serialize(query, &message);
3228 if (ret) {
3229 ret_code = LTTNG_ERR_UNK;
3230 goto end;
3231 }
3232
3233 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3234 message_lsm->u.error_query.length =
3235 (uint32_t) message.buffer.size - sizeof(lsm);
3236
3237 {
3238 struct lttng_payload_view message_view =
3239 lttng_payload_view_from_payload(
3240 &message, 0, -1);
3241
3242 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3243 &message_view);
3244 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3245 if (ret < 0) {
4bd69c5f 3246 ret_code =(lttng_error_code) -ret;
b99a0cb3
JG
3247 goto end;
3248 }
3249 }
3250
3251 {
3252 ssize_t reply_create_ret;
3253 struct lttng_payload_view reply_view =
3254 lttng_payload_view_from_payload(
3255 &reply, 0, reply.buffer.size);
3256
3257 reply_create_ret = lttng_error_query_results_create_from_payload(
3258 &reply_view, results);
3259 if (reply_create_ret < 0) {
3260 ret_code = LTTNG_ERR_INVALID_PROTOCOL;
3261 goto end;
3262 }
3263 }
3264
3265 ret_code = LTTNG_OK;
3266end:
3267 lttng_payload_reset(&message);
3268 lttng_payload_reset(&reply);
3269 return ret_code;
3270}
3271
b61776fb 3272int lttng_unregister_trigger(const struct lttng_trigger *trigger)
a58c490f
JG
3273{
3274 int ret;
3275 struct lttcomm_session_msg lsm;
99608320
JR
3276 struct lttcomm_session_msg *message_lsm;
3277 struct lttng_payload message;
3278 struct lttng_payload reply;
b61776fb 3279 struct lttng_trigger *copy = NULL;
64eafdf6
JR
3280 const struct lttng_credentials user_creds = {
3281 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3282 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3283 };
99608320
JR
3284
3285 lttng_payload_init(&message);
3286 lttng_payload_init(&reply);
a58c490f
JG
3287
3288 if (!trigger) {
3289 ret = -LTTNG_ERR_INVALID;
3290 goto end;
3291 }
3292
b61776fb
SM
3293 copy = lttng_trigger_copy(trigger);
3294 if (!copy) {
3295 ret = -LTTNG_ERR_UNK;
3296 goto end;
3297 }
3298
3299 if (!copy->creds.uid.is_set) {
3300 /* Use the client credentials as the trigger credentials */
3301 lttng_trigger_set_credentials(copy, &user_creds);
64eafdf6
JR
3302 } else {
3303 /*
3304 * Validate that either the current trigger credentials and the
3305 * client credentials are identical or that the current user is
3306 * root. The root user can register, unregister triggers for
3307 * himself and other users.
3308 *
3309 * This check is also present on the sessiond side, using the
3310 * credentials passed on the socket. These check are all
3311 * "safety" checks.
3312 */
3313 const struct lttng_credentials *trigger_creds =
b61776fb 3314 lttng_trigger_get_credentials(copy);
64eafdf6
JR
3315 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3316 if (lttng_credentials_get_uid(&user_creds) != 0) {
3317 ret = -LTTNG_ERR_EPERM;
3318 goto end;
3319 }
3320 }
3321 }
3322
b61776fb 3323 if (!lttng_trigger_validate(copy)) {
3647288f 3324 ret = -LTTNG_ERR_INVALID_TRIGGER;
a58c490f
JG
3325 goto end;
3326 }
3327
99608320
JR
3328 memset(&lsm, 0, sizeof(lsm));
3329 lsm.cmd_type = LTTNG_UNREGISTER_TRIGGER;
3330
97285430
JG
3331 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3332 if (ret) {
3333 ret = -LTTNG_ERR_NOMEM;
3334 goto end;
3335 }
99608320 3336
b61776fb 3337 ret = lttng_trigger_serialize(copy, &message);
3647288f 3338 if (ret < 0) {
a58c490f
JG
3339 ret = -LTTNG_ERR_UNK;
3340 goto end;
3341 }
3342
b0a79813
FD
3343 /*
3344 * This is needed to populate the trigger object size for the command
3345 * header and number of fds sent.
3346 */
3347 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3348
99608320
JR
3349 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3350
3351 {
3352 struct lttng_payload_view message_view =
3353 lttng_payload_view_from_payload(
3354 &message, 0, -1);
3355
3356 /*
3357 * Update the message header with the number of fd that will be
3358 * sent.
3359 */
3360 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3361 &message_view);
3362
3363 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3364 if (ret < 0) {
3365 goto end;
3366 }
3367 }
3368
3369 ret = 0;
a58c490f 3370end:
b61776fb 3371 lttng_trigger_destroy(copy);
99608320
JR
3372 lttng_payload_reset(&message);
3373 lttng_payload_reset(&reply);
a58c490f
JG
3374 return ret;
3375}
3376
fbc9f37d
JR
3377/*
3378 * Ask the session daemon for all registered triggers for the current user.
3379 *
3380 * Allocates and return an lttng_triggers set.
3381 * On error, returns a suitable lttng_error_code.
3382 */
3383enum lttng_error_code lttng_list_triggers(struct lttng_triggers **triggers)
3384{
3385 int ret;
3386 enum lttng_error_code ret_code = LTTNG_OK;
1c9a0b0e
MJ
3387 struct lttcomm_session_msg lsm = {
3388 .cmd_type = LTTNG_LIST_TRIGGERS,
3389 .session = {},
3390 .domain = {},
3391 .u = {},
3392 .fd_count = 0,
3393 };
fbc9f37d
JR
3394 struct lttng_triggers *local_triggers = NULL;
3395 struct lttng_payload reply;
3396 struct lttng_payload_view lsm_view =
3397 lttng_payload_view_init_from_buffer(
3398 (const char *) &lsm, 0, sizeof(lsm));
3399
3400 lttng_payload_init(&reply);
3401
3402 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
3403 if (ret < 0) {
3404 ret_code = (enum lttng_error_code) -ret;
3405 goto end;
3406 }
3407
3408 {
3409 struct lttng_payload_view reply_view =
3410 lttng_payload_view_from_payload(
3411 &reply, 0, reply.buffer.size);
3412
3413 ret = lttng_triggers_create_from_payload(
3414 &reply_view, &local_triggers);
3415 if (ret < 0) {
3416 ret_code = LTTNG_ERR_FATAL;
3417 goto end;
3418 }
3419 }
3420
3421 *triggers = local_triggers;
3422 local_triggers = NULL;
3423end:
3424 lttng_payload_reset(&reply);
3425 lttng_triggers_destroy(local_triggers);
3426 return ret_code;
3427}
3428
fac6795d 3429/*
9ae110e2 3430 * lib constructor.
fac6795d 3431 */
b2b89e8a 3432static void __attribute__((constructor)) init(void)
fac6795d
DG
3433{
3434 /* Set default session group */
bbccc3d2 3435 lttng_set_tracing_group(DEFAULT_TRACING_GROUP);
fac6795d 3436}
49cca668
DG
3437
3438/*
9ae110e2 3439 * lib destructor.
49cca668 3440 */
b2b89e8a 3441static void __attribute__((destructor)) lttng_ctl_exit(void)
49cca668
DG
3442{
3443 free(tracing_group);
3444}
This page took 0.313772 seconds and 5 git commands to generate.