libcommon: move event.c to libcommon-lgpl
[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
0ae3cfc6 21#include <common/bytecode/bytecode.h>
1cbd136b 22#include <common/align.h>
990570ed 23#include <common/common.h>
edf4b93e 24#include <common/compat/errno.h>
15e37663 25#include <common/compat/string.h>
990570ed 26#include <common/defaults.h>
fe489250 27#include <common/dynamic-array.h>
b99a0cb3 28#include <common/dynamic-buffer.h>
9e620ea7 29#include <common/payload-view.h>
b99a0cb3 30#include <common/payload.h>
db758600 31#include <common/sessiond-comm/sessiond-comm.h>
159b042f 32#include <common/tracker.h>
fe489250 33#include <common/unix.h>
a4b92340 34#include <common/uri.h>
feb0f3e5 35#include <common/utils.h>
a58c490f 36#include <lttng/channel-internal.h>
159b042f
JG
37#include <lttng/destruction-handle.h>
38#include <lttng/endpoint.h>
b99a0cb3 39#include <lttng/error-query-internal.h>
de453daa 40#include <lttng/event-internal.h>
159b042f 41#include <lttng/health-internal.h>
b99a0cb3 42#include <lttng/lttng-error.h>
159b042f 43#include <lttng/lttng.h>
b178f53e 44#include <lttng/session-descriptor-internal.h>
159b042f
JG
45#include <lttng/session-internal.h>
46#include <lttng/trigger/trigger-internal.h>
47#include <lttng/userspace-probe-internal.h>
fac6795d 48
b99a0cb3 49#include "lttng-ctl-helper.h"
e4d2f27a 50#include <common/filter/filter-ast.h>
348ddc5c 51#include <common/filter/filter-parser.hpp>
e4d2f27a 52#include <common/filter/memstream.h>
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 */
4bd69c5f 269 grp_list = (gid_t *) zmalloc(grp_list_size * sizeof(gid_t));
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;
503 void *buf = NULL;
504
505 if (len) {
506 if (!user_len) {
507 ret = -LTTNG_ERR_INVALID;
508 goto end;
509 }
510
511 buf = zmalloc(len);
512 if (!buf) {
513 ret = -ENOMEM;
514 goto end;
515 }
516
517 ret = recv_data_sessiond(buf, len);
518 if (ret < 0) {
519 goto end;
520 }
521
522 if (!user_buf) {
523 ret = -LTTNG_ERR_INVALID;
524 goto end;
525 }
526
527 /* Move ownership of command header buffer to user. */
528 *user_buf = buf;
529 buf = NULL;
530 *user_len = len;
531 } else {
532 /* No command header. */
533 if (user_len) {
534 *user_len = 0;
535 }
536
537 if (user_buf) {
538 *user_buf = NULL;
539 }
540 }
541
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
4bd69c5f 732 handle = (lttng_handle *) zmalloc(sizeof(struct 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,
38057ed1
DG
914 struct lttng_event_context *ctx, const char *event_name,
915 const char *channel_name)
d65106b1 916{
2001793c
JG
917 int ret;
918 size_t len = 0;
919 char *buf = NULL;
cd80958d
DG
920 struct lttcomm_session_msg lsm;
921
9ae110e2 922 /* Safety check. Both are mandatory. */
9d697d3d 923 if (handle == NULL || ctx == NULL) {
2001793c
JG
924 ret = -LTTNG_ERR_INVALID;
925 goto end;
cd80958d
DG
926 }
927
441c16a7 928 memset(&lsm, 0, sizeof(lsm));
cd80958d
DG
929 lsm.cmd_type = LTTNG_ADD_CONTEXT;
930
85076754 931 /* If no channel name, send empty string. */
e1b624d0
JG
932 ret = lttng_strncpy(lsm.u.context.channel_name, channel_name ?: "",
933 sizeof(lsm.u.context.channel_name));
934 if (ret) {
935 ret = -LTTNG_ERR_INVALID;
936 goto end;
85076754 937 }
cd80958d 938
60160d2a 939 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
e1b624d0 940 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 941 sizeof(lsm.session.name));
e1b624d0
JG
942 if (ret) {
943 ret = -LTTNG_ERR_INVALID;
944 goto end;
945 }
cd80958d 946
2001793c
JG
947 if (ctx->ctx == LTTNG_EVENT_CONTEXT_APP_CONTEXT) {
948 size_t provider_len, ctx_len;
949 const char *provider_name = ctx->u.app_ctx.provider_name;
950 const char *ctx_name = ctx->u.app_ctx.ctx_name;
951
952 if (!provider_name || !ctx_name) {
953 ret = -LTTNG_ERR_INVALID;
954 goto end;
955 }
956
957 provider_len = strlen(provider_name);
958 if (provider_len == 0) {
959 ret = -LTTNG_ERR_INVALID;
960 goto end;
961 }
962 lsm.u.context.provider_name_len = provider_len;
963
964 ctx_len = strlen(ctx_name);
965 if (ctx_len == 0) {
966 ret = -LTTNG_ERR_INVALID;
967 goto end;
968 }
969 lsm.u.context.context_name_len = ctx_len;
970
971 len = provider_len + ctx_len;
4bd69c5f 972 buf = (char *) zmalloc(len);
2001793c
JG
973 if (!buf) {
974 ret = -LTTNG_ERR_NOMEM;
975 goto end;
976 }
977
978 memcpy(buf, provider_name, provider_len);
979 memcpy(buf + provider_len, ctx_name, ctx_len);
980 }
981 memcpy(&lsm.u.context.ctx, ctx, sizeof(struct lttng_event_context));
46f44e2a
JR
982
983 if (ctx->ctx == LTTNG_EVENT_CONTEXT_APP_CONTEXT) {
984 /*
985 * Don't leak application addresses to the sessiond.
986 * This is only necessary when ctx is for an app ctx otherwise
987 * the values inside the union (type & config) are overwritten.
988 */
989 lsm.u.context.ctx.u.app_ctx.provider_name = NULL;
990 lsm.u.context.ctx.u.app_ctx.ctx_name = NULL;
991 }
2001793c 992
795a978d 993 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, buf, len, NULL);
2001793c
JG
994end:
995 free(buf);
996 return ret;
d65106b1
DG
997}
998
f3ed775e 999/*
9ae110e2
JG
1000 * Enable event(s) for a channel.
1001 *
1002 * If no event name is specified, all events are enabled.
1003 * If no channel name is specified, the default 'channel0' is used.
1004 *
1005 * Returns size of returned session payload data or a negative error code.
f3ed775e 1006 */
cd80958d 1007int lttng_enable_event(struct lttng_handle *handle,
38057ed1 1008 struct lttng_event *ev, const char *channel_name)
1df4dedd 1009{
f8a96544
JI
1010 return lttng_enable_event_with_exclusions(handle, ev, channel_name,
1011 NULL, 0, NULL);
1df4dedd
DG
1012}
1013
53a80697 1014/*
025faf73 1015 * Create or enable an event with a filter expression.
178191b3 1016 *
53a80697
MD
1017 * Return negative error value on error.
1018 * Return size of returned session payload data if OK.
1019 */
025faf73 1020int lttng_enable_event_with_filter(struct lttng_handle *handle,
178191b3 1021 struct lttng_event *event, const char *channel_name,
53a80697
MD
1022 const char *filter_expression)
1023{
f8a96544
JI
1024 return lttng_enable_event_with_exclusions(handle, event, channel_name,
1025 filter_expression, 0, NULL);
53a80697
MD
1026}
1027
347c5ab5 1028/*
0e115563 1029 * Depending on the event, return a newly allocated agent filter expression or
347c5ab5
DG
1030 * NULL if not applicable.
1031 *
1032 * An event with NO loglevel and the name is * will return NULL.
1033 */
0e115563 1034static char *set_agent_filter(const char *filter, struct lttng_event *ev)
347c5ab5
DG
1035{
1036 int err;
0e115563 1037 char *agent_filter = NULL;
347c5ab5 1038
a0377dfe 1039 LTTNG_ASSERT(ev);
347c5ab5
DG
1040
1041 /* Don't add filter for the '*' event. */
9f449915 1042 if (strcmp(ev->name, "*") != 0) {
347c5ab5 1043 if (filter) {
0e115563 1044 err = asprintf(&agent_filter, "(%s) && (logger_name == \"%s\")", filter,
347c5ab5
DG
1045 ev->name);
1046 } else {
0e115563 1047 err = asprintf(&agent_filter, "logger_name == \"%s\"", ev->name);
347c5ab5
DG
1048 }
1049 if (err < 0) {
1050 PERROR("asprintf");
6a556f7b 1051 goto error;
347c5ab5
DG
1052 }
1053 }
1054
1055 /* Add loglevel filtering if any for the JUL domain. */
1056 if (ev->loglevel_type != LTTNG_EVENT_LOGLEVEL_ALL) {
b53d4e59 1057 const char *op;
347c5ab5
DG
1058
1059 if (ev->loglevel_type == LTTNG_EVENT_LOGLEVEL_RANGE) {
1060 op = ">=";
1061 } else {
1062 op = "==";
1063 }
1064
0e115563 1065 if (filter || agent_filter) {
6a556f7b
JG
1066 char *new_filter;
1067
fb0edb23 1068 err = asprintf(&new_filter, "(%s) && (int_loglevel %s %d)",
0e115563 1069 agent_filter ? agent_filter : filter, op,
347c5ab5 1070 ev->loglevel);
0e115563
DG
1071 if (agent_filter) {
1072 free(agent_filter);
6a556f7b 1073 }
0e115563 1074 agent_filter = new_filter;
347c5ab5 1075 } else {
0e115563 1076 err = asprintf(&agent_filter, "int_loglevel %s %d", op,
347c5ab5
DG
1077 ev->loglevel);
1078 }
1079 if (err < 0) {
1080 PERROR("asprintf");
6a556f7b 1081 goto error;
347c5ab5
DG
1082 }
1083 }
1084
0e115563 1085 return agent_filter;
6a556f7b 1086error:
0e115563 1087 free(agent_filter);
6a556f7b 1088 return NULL;
347c5ab5
DG
1089}
1090
93deb080
JI
1091/*
1092 * Enable event(s) for a channel, possibly with exclusions and a filter.
1093 * If no event name is specified, all events are enabled.
1094 * If no channel name is specified, the default name is used.
1095 * If filter expression is not NULL, the filter is set for the event.
1096 * If exclusion count is not zero, the exclusions are set for the event.
1097 * Returns size of returned session payload data or a negative error code.
1098 */
1099int lttng_enable_event_with_exclusions(struct lttng_handle *handle,
1100 struct lttng_event *ev, const char *channel_name,
e9efbcd3 1101 const char *original_filter_expression,
93deb080
JI
1102 int exclusion_count, char **exclusion_list)
1103{
1104 struct lttcomm_session_msg lsm;
e368fb43
JG
1105 struct lttng_payload payload;
1106 int ret = 0, i;
137b9942 1107 unsigned int free_filter_expression = 0;
93deb080 1108 struct filter_parser_ctx *ctx = NULL;
ec005ec6 1109
6a0838b7
YL
1110 /*
1111 * We have either a filter or some exclusions, so we need to set up
e368fb43 1112 * a variable-length payload from where to send the data.
6a0838b7 1113 */
e368fb43 1114 lttng_payload_init(&payload);
ec005ec6 1115
e9efbcd3
JG
1116 /*
1117 * Cast as non-const since we may replace the filter expression
1118 * by a dynamically allocated string. Otherwise, the original
1119 * string is not modified.
1120 */
1121 char *filter_expression = (char *) original_filter_expression;
93deb080
JI
1122
1123 if (handle == NULL || ev == NULL) {
137b9942
DG
1124 ret = -LTTNG_ERR_INVALID;
1125 goto error;
93deb080
JI
1126 }
1127
9ae110e2
JG
1128 /*
1129 * Empty filter string will always be rejected by the parser
93deb080
JI
1130 * anyway, so treat this corner-case early to eliminate
1131 * lttng_fmemopen error for 0-byte allocation.
1132 */
1133 if (filter_expression && filter_expression[0] == '\0') {
137b9942
DG
1134 ret = -LTTNG_ERR_INVALID;
1135 goto error;
93deb080
JI
1136 }
1137
1138 memset(&lsm, 0, sizeof(lsm));
1139
1140 /* If no channel name, send empty string. */
e1b624d0
JG
1141 ret = lttng_strncpy(lsm.u.enable.channel_name, channel_name ?: "",
1142 sizeof(lsm.u.enable.channel_name));
1143 if (ret) {
1144 ret = -LTTNG_ERR_INVALID;
1145 goto error;
93deb080
JI
1146 }
1147
18a720cd
MD
1148 lsm.cmd_type = LTTNG_ENABLE_EVENT;
1149 if (ev->name[0] == '\0') {
e1b624d0
JG
1150 /* Enable all events. */
1151 ret = lttng_strncpy(ev->name, "*", sizeof(ev->name));
a0377dfe 1152 LTTNG_ASSERT(ret == 0);
6565421f 1153 }
93deb080 1154
60160d2a 1155 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
93deb080
JI
1156 memcpy(&lsm.u.enable.event, ev, sizeof(lsm.u.enable.event));
1157
e1b624d0 1158 ret = lttng_strncpy(lsm.session.name, handle->session_name,
93deb080 1159 sizeof(lsm.session.name));
e1b624d0
JG
1160 if (ret) {
1161 ret = -LTTNG_ERR_INVALID;
1162 goto error;
1163 }
1164
93deb080
JI
1165 lsm.u.enable.exclusion_count = exclusion_count;
1166 lsm.u.enable.bytecode_len = 0;
1167
9ae110e2 1168 /* Parse filter expression. */
5cdb6027 1169 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
0e115563
DG
1170 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1171 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
5cdb6027 1172 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
0e115563
DG
1173 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1174 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1175 char *agent_filter;
64226865 1176
347c5ab5 1177 /* Setup JUL filter if needed. */
0e115563
DG
1178 agent_filter = set_agent_filter(filter_expression, ev);
1179 if (!agent_filter) {
137b9942 1180 if (!filter_expression) {
9ae110e2
JG
1181 /*
1182 * No JUL and no filter, just skip
1183 * everything below.
1184 */
137b9942
DG
1185 goto ask_sessiond;
1186 }
64226865
DG
1187 } else {
1188 /*
9ae110e2
JG
1189 * With an agent filter, the original filter has
1190 * been added to it thus replace the filter
1191 * expression.
64226865 1192 */
0e115563 1193 filter_expression = agent_filter;
e9efbcd3 1194 free_filter_expression = 1;
9b21e6d5 1195 }
9b21e6d5 1196 }
93deb080 1197
e4d2f27a 1198 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
93deb080 1199 if (ret) {
137b9942 1200 goto filter_error;
93deb080 1201 }
e4d2f27a
JR
1202
1203 lsm.u.enable.bytecode_len = sizeof(ctx->bytecode->b)
1204 + bytecode_get_len(&ctx->bytecode->b);
1205 lsm.u.enable.expression_len = strlen(filter_expression) + 1;
6b453b5e
JG
1206 }
1207
e368fb43
JG
1208 ret = lttng_dynamic_buffer_set_capacity(&payload.buffer,
1209 lsm.u.enable.bytecode_len +
1210 lsm.u.enable.expression_len +
1211 LTTNG_SYMBOL_NAME_LEN *
1212 exclusion_count);
15e37663 1213 if (ret) {
6b453b5e 1214 ret = -LTTNG_ERR_EXCLUSION_NOMEM;
7ca1dc6f 1215 goto mem_error;
6b453b5e 1216 }
137b9942 1217
9ae110e2 1218 /* Put exclusion names first in the data. */
15e37663
JG
1219 for (i = 0; i < exclusion_count; i++) {
1220 size_t exclusion_len;
1221
53a5c209 1222 exclusion_len = lttng_strnlen(exclusion_list[i],
15e37663
JG
1223 LTTNG_SYMBOL_NAME_LEN);
1224 if (exclusion_len == LTTNG_SYMBOL_NAME_LEN) {
1225 /* Exclusion is not NULL-terminated. */
1226 ret = -LTTNG_ERR_INVALID;
1227 goto mem_error;
1228 }
1229
e368fb43 1230 ret = lttng_dynamic_buffer_append(&payload.buffer,
53a5c209
FD
1231 exclusion_list[i], exclusion_len);
1232 if (ret) {
1233 goto mem_error;
1234 }
1235
1236 /*
1237 * Padding the rest of the entry with zeros. Every exclusion
1238 * entries take LTTNG_SYMBOL_NAME_LEN bytes in the buffer.
1239 */
1240 ret = lttng_dynamic_buffer_set_size(&payload.buffer,
1241 LTTNG_SYMBOL_NAME_LEN * (i + 1));
15e37663
JG
1242 if (ret) {
1243 goto mem_error;
1244 }
6b453b5e 1245 }
15e37663 1246
9ae110e2 1247 /* Add filter expression next. */
15e37663 1248 if (filter_expression) {
e368fb43 1249 ret = lttng_dynamic_buffer_append(&payload.buffer,
15e37663
JG
1250 filter_expression, lsm.u.enable.expression_len);
1251 if (ret) {
1252 goto mem_error;
1253 }
6b453b5e 1254 }
9ae110e2 1255 /* Add filter bytecode next. */
137b9942 1256 if (ctx && lsm.u.enable.bytecode_len != 0) {
e368fb43 1257 ret = lttng_dynamic_buffer_append(&payload.buffer,
15e37663
JG
1258 &ctx->bytecode->b, lsm.u.enable.bytecode_len);
1259 if (ret) {
1260 goto mem_error;
1261 }
1262 }
1263 if (ev->extended.ptr) {
1264 struct lttng_event_extended *ev_ext =
1265 (struct lttng_event_extended *) ev->extended.ptr;
1266
1267 if (ev_ext->probe_location) {
1268 /*
1269 * lttng_userspace_probe_location_serialize returns the
1270 * number of bytes that was appended to the buffer.
1271 */
1272 ret = lttng_userspace_probe_location_serialize(
e368fb43 1273 ev_ext->probe_location, &payload);
56f0bc67 1274 if (ret < 0) {
15e37663
JG
1275 goto mem_error;
1276 }
1277
15e37663
JG
1278 /*
1279 * Set the size of the userspace probe location element
1280 * of the buffer so that the receiving side knows where
1281 * to split it.
1282 */
1283 lsm.u.enable.userspace_probe_location_len = ret;
1284 }
93deb080
JI
1285 }
1286
e368fb43
JG
1287 {
1288 struct lttng_payload_view view = lttng_payload_view_from_payload(
1289 &payload, 0, -1);
fe489250 1290 int fd_count = lttng_payload_view_get_fd_handle_count(&view);
e368fb43
JG
1291 int fd_to_send;
1292
1293 if (fd_count < 0) {
1294 goto mem_error;
1295 }
1296
a0377dfe 1297 LTTNG_ASSERT(fd_count == 0 || fd_count == 1);
e368fb43 1298 if (fd_count == 1) {
e2fc8e9d 1299 struct fd_handle *h =
fe489250
JG
1300 lttng_payload_view_pop_fd_handle(&view);
1301
e2fc8e9d 1302 if (!h) {
e368fb43
JG
1303 goto mem_error;
1304 }
1305
e2fc8e9d
SM
1306 fd_to_send = fd_handle_get_fd(h);
1307 fd_handle_put(h);
e368fb43
JG
1308 }
1309
1310 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm,
1311 fd_count ? &fd_to_send : NULL, fd_count,
1312 view.buffer.size ? view.buffer.data : NULL,
1313 view.buffer.size, NULL, NULL, 0);
1314 }
93deb080 1315
7ca1dc6f
DG
1316mem_error:
1317 if (filter_expression && ctx) {
93deb080
JI
1318 filter_bytecode_free(ctx);
1319 filter_ir_free(ctx);
1320 filter_parser_ctx_free(ctx);
7ca1dc6f
DG
1321 }
1322filter_error:
1323 if (free_filter_expression) {
1324 /*
9ae110e2
JG
1325 * The filter expression has been replaced and must be freed as
1326 * it is not the original filter expression received as a
1327 * parameter.
7ca1dc6f
DG
1328 */
1329 free(filter_expression);
93deb080 1330 }
137b9942
DG
1331error:
1332 /*
9ae110e2
JG
1333 * Return directly to the caller and don't ask the sessiond since
1334 * something went wrong in the parsing of data above.
137b9942 1335 */
e368fb43 1336 lttng_payload_reset(&payload);
93deb080 1337 return ret;
3e1c9ff7
DG
1338
1339ask_sessiond:
1340 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1341 return ret;
93deb080
JI
1342}
1343
6e911cad
MD
1344int lttng_disable_event_ext(struct lttng_handle *handle,
1345 struct lttng_event *ev, const char *channel_name,
1346 const char *original_filter_expression)
1df4dedd 1347{
cd80958d 1348 struct lttcomm_session_msg lsm;
6e911cad
MD
1349 char *varlen_data;
1350 int ret = 0;
1351 unsigned int free_filter_expression = 0;
1352 struct filter_parser_ctx *ctx = NULL;
1353 /*
1354 * Cast as non-const since we may replace the filter expression
1355 * by a dynamically allocated string. Otherwise, the original
1356 * string is not modified.
1357 */
1358 char *filter_expression = (char *) original_filter_expression;
1df4dedd 1359
6e911cad
MD
1360 if (handle == NULL || ev == NULL) {
1361 ret = -LTTNG_ERR_INVALID;
1362 goto error;
1363 }
1364
9ae110e2
JG
1365 /*
1366 * Empty filter string will always be rejected by the parser
6e911cad
MD
1367 * anyway, so treat this corner-case early to eliminate
1368 * lttng_fmemopen error for 0-byte allocation.
1369 */
1370 if (filter_expression && filter_expression[0] == '\0') {
1371 ret = -LTTNG_ERR_INVALID;
1372 goto error;
cd80958d
DG
1373 }
1374
441c16a7
MD
1375 memset(&lsm, 0, sizeof(lsm));
1376
85076754 1377 /* If no channel name, send empty string. */
e1b624d0
JG
1378 ret = lttng_strncpy(lsm.u.disable.channel_name, channel_name ?: "",
1379 sizeof(lsm.u.disable.channel_name));
1380 if (ret) {
1381 ret = -LTTNG_ERR_INVALID;
1382 goto error;
eb354453
DG
1383 }
1384
18a720cd 1385 lsm.cmd_type = LTTNG_DISABLE_EVENT;
f3ed775e 1386
60160d2a 1387 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
6e911cad
MD
1388 memcpy(&lsm.u.disable.event, ev, sizeof(lsm.u.disable.event));
1389
e1b624d0 1390 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 1391 sizeof(lsm.session.name));
e1b624d0
JG
1392 if (ret) {
1393 ret = -LTTNG_ERR_INVALID;
1394 goto error;
1395 }
1396
6e911cad 1397 lsm.u.disable.bytecode_len = 0;
cd80958d 1398
6e911cad
MD
1399 /*
1400 * For the JUL domain, a filter is enforced except for the
1401 * disable all event. This is done to avoid having the event in
1402 * all sessions thus filtering by logger name.
1403 */
1404 if (filter_expression == NULL &&
1405 (handle->domain.type != LTTNG_DOMAIN_JUL &&
0e115563
DG
1406 handle->domain.type != LTTNG_DOMAIN_LOG4J &&
1407 handle->domain.type != LTTNG_DOMAIN_PYTHON)) {
6e911cad
MD
1408 goto ask_sessiond;
1409 }
1410
1411 /*
1412 * We have a filter, so we need to set up a variable-length
1413 * memory block from where to send the data.
1414 */
1415
1416 /* Parse filter expression */
1417 if (filter_expression != NULL || handle->domain.type == LTTNG_DOMAIN_JUL
0e115563
DG
1418 || handle->domain.type == LTTNG_DOMAIN_LOG4J
1419 || handle->domain.type == LTTNG_DOMAIN_PYTHON) {
6e911cad 1420 if (handle->domain.type == LTTNG_DOMAIN_JUL ||
0e115563
DG
1421 handle->domain.type == LTTNG_DOMAIN_LOG4J ||
1422 handle->domain.type == LTTNG_DOMAIN_PYTHON) {
1423 char *agent_filter;
6e911cad
MD
1424
1425 /* Setup JUL filter if needed. */
0e115563
DG
1426 agent_filter = set_agent_filter(filter_expression, ev);
1427 if (!agent_filter) {
6e911cad 1428 if (!filter_expression) {
9ae110e2
JG
1429 /*
1430 * No JUL and no filter, just skip
1431 * everything below.
1432 */
6e911cad
MD
1433 goto ask_sessiond;
1434 }
1435 } else {
1436 /*
9ae110e2
JG
1437 * With a JUL filter, the original filter has
1438 * been added to it thus replace the filter
1439 * expression.
6e911cad 1440 */
0e115563 1441 filter_expression = agent_filter;
6e911cad
MD
1442 free_filter_expression = 1;
1443 }
1444 }
1445
e4d2f27a 1446 ret = filter_parser_ctx_create_from_filter_expression(filter_expression, &ctx);
6e911cad
MD
1447 if (ret) {
1448 goto filter_error;
1449 }
e4d2f27a
JR
1450
1451 lsm.u.enable.bytecode_len = sizeof(ctx->bytecode->b)
1452 + bytecode_get_len(&ctx->bytecode->b);
1453 lsm.u.enable.expression_len = strlen(filter_expression) + 1;
6e911cad
MD
1454 }
1455
4bd69c5f 1456 varlen_data = (char *) zmalloc(lsm.u.disable.bytecode_len
6e911cad
MD
1457 + lsm.u.disable.expression_len);
1458 if (!varlen_data) {
1459 ret = -LTTNG_ERR_EXCLUSION_NOMEM;
1460 goto mem_error;
1461 }
1462
9ae110e2 1463 /* Add filter expression. */
6e911cad
MD
1464 if (lsm.u.disable.expression_len != 0) {
1465 memcpy(varlen_data,
1466 filter_expression,
1467 lsm.u.disable.expression_len);
1468 }
9ae110e2 1469 /* Add filter bytecode next. */
6e911cad
MD
1470 if (ctx && lsm.u.disable.bytecode_len != 0) {
1471 memcpy(varlen_data
1472 + lsm.u.disable.expression_len,
1473 &ctx->bytecode->b,
1474 lsm.u.disable.bytecode_len);
1475 }
1476
795a978d 1477 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, varlen_data,
6e911cad
MD
1478 lsm.u.disable.bytecode_len + lsm.u.disable.expression_len, NULL);
1479 free(varlen_data);
1480
1481mem_error:
1482 if (filter_expression && ctx) {
1483 filter_bytecode_free(ctx);
1484 filter_ir_free(ctx);
1485 filter_parser_ctx_free(ctx);
1486 }
1487filter_error:
1488 if (free_filter_expression) {
1489 /*
9ae110e2
JG
1490 * The filter expression has been replaced and must be freed as
1491 * it is not the original filter expression received as a
1492 * parameter.
6e911cad
MD
1493 */
1494 free(filter_expression);
1495 }
1496error:
1497 /*
9ae110e2
JG
1498 * Return directly to the caller and don't ask the sessiond since
1499 * something went wrong in the parsing of data above.
6e911cad
MD
1500 */
1501 return ret;
1502
1503ask_sessiond:
1504 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1505 return ret;
1506}
1507
1508/*
9ae110e2
JG
1509 * Disable event(s) of a channel and domain.
1510 * If no event name is specified, all events are disabled.
1511 * If no channel name is specified, the default 'channel0' is used.
1512 * Returns size of returned session payload data or a negative error code.
6e911cad
MD
1513 */
1514int lttng_disable_event(struct lttng_handle *handle, const char *name,
1515 const char *channel_name)
1516{
e1b624d0 1517 int ret;
6e911cad
MD
1518 struct lttng_event ev;
1519
1520 memset(&ev, 0, sizeof(ev));
9b7431cf 1521 ev.loglevel = -1;
6e911cad 1522 ev.type = LTTNG_EVENT_ALL;
e1b624d0
JG
1523 ret = lttng_strncpy(ev.name, name ?: "", sizeof(ev.name));
1524 if (ret) {
1525 ret = -LTTNG_ERR_INVALID;
1526 goto end;
1527 }
1528
1529 ret = lttng_disable_event_ext(handle, &ev, channel_name, NULL);
1530end:
1531 return ret;
1df4dedd
DG
1532}
1533
cf0bcb51
JG
1534struct lttng_channel *lttng_channel_create(struct lttng_domain *domain)
1535{
1536 struct lttng_channel *channel = NULL;
cf0bcb51
JG
1537
1538 if (!domain) {
999af9c1 1539 goto end;
cf0bcb51
JG
1540 }
1541
1542 /* Validate domain. */
1543 switch (domain->type) {
1544 case LTTNG_DOMAIN_UST:
1545 switch (domain->buf_type) {
1546 case LTTNG_BUFFER_PER_UID:
1547 case LTTNG_BUFFER_PER_PID:
1548 break;
1549 default:
999af9c1 1550 goto end;
cf0bcb51
JG
1551 }
1552 break;
1553 case LTTNG_DOMAIN_KERNEL:
1554 if (domain->buf_type != LTTNG_BUFFER_GLOBAL) {
999af9c1 1555 goto end;
cf0bcb51
JG
1556 }
1557 break;
1558 default:
999af9c1 1559 goto end;
cf0bcb51
JG
1560 }
1561
999af9c1 1562 channel = lttng_channel_create_internal();
cf0bcb51 1563 if (!channel) {
999af9c1 1564 goto end;
cf0bcb51
JG
1565 }
1566
cf0bcb51 1567 lttng_channel_set_default_attr(domain, &channel->attr);
999af9c1 1568end:
cf0bcb51 1569 return channel;
cf0bcb51
JG
1570}
1571
1572void lttng_channel_destroy(struct lttng_channel *channel)
1573{
1574 if (!channel) {
1575 return;
1576 }
1577
1578 if (channel->attr.extended.ptr) {
1579 free(channel->attr.extended.ptr);
1580 }
1581 free(channel);
1582}
1583
1df4dedd 1584/*
9ae110e2
JG
1585 * Enable channel per domain
1586 * Returns size of returned session payload data or a negative error code.
a5c5a2bd 1587 */
cd80958d 1588int lttng_enable_channel(struct lttng_handle *handle,
cf0bcb51 1589 struct lttng_channel *in_chan)
a5c5a2bd 1590{
13dd7782 1591 enum lttng_error_code ret_code;
e1b624d0 1592 int ret;
999af9c1 1593 struct lttng_dynamic_buffer buffer;
cd80958d 1594 struct lttcomm_session_msg lsm;
13dd7782 1595 uint64_t total_buffer_size_needed_per_cpu = 0;
999af9c1
JR
1596 struct lttng_channel *channel = NULL;
1597
1598 lttng_dynamic_buffer_init(&buffer);
cd80958d 1599
9ae110e2 1600 /* NULL arguments are forbidden. No default values. */
cf0bcb51 1601 if (handle == NULL || in_chan == NULL) {
999af9c1
JR
1602 ret = -LTTNG_ERR_INVALID;
1603 goto end;
cf0bcb51 1604 }
cd80958d 1605
09b72f7a
FD
1606 /*
1607 * Verify that the amount of memory required to create the requested
1608 * buffer is available on the system at the moment.
1609 */
999af9c1
JR
1610 if (in_chan->attr.num_subbuf >
1611 UINT64_MAX / in_chan->attr.subbuf_size) {
13dd7782
JR
1612 /* Overflow */
1613 ret = -LTTNG_ERR_OVERFLOW;
1614 goto end;
1615 }
1616
999af9c1
JR
1617 total_buffer_size_needed_per_cpu =
1618 in_chan->attr.num_subbuf * in_chan->attr.subbuf_size;
13dd7782
JR
1619 ret_code = check_enough_available_memory(
1620 total_buffer_size_needed_per_cpu);
1621 if (ret_code != LTTNG_OK) {
1622 ret = -ret_code;
1623 goto end;
09b72f7a
FD
1624 }
1625
999af9c1
JR
1626 /* Copy the channel for easier manipulation. */
1627 channel = lttng_channel_copy(in_chan);
1628 if (!channel) {
1629 ret = -LTTNG_ERR_NOMEM;
1630 goto end;
1631 }
1632
1633 /* Populate the channel extended attribute if necessary. */
1634 if (!channel->attr.extended.ptr) {
1635 struct lttng_channel_extended *extended =
1636 (struct lttng_channel_extended *) zmalloc(
1637 sizeof(*extended));
1638
1639 if (!extended) {
1640 ret = -LTTNG_ERR_NOMEM;
1641 goto end;
1642 }
1643 lttng_channel_set_default_extended_attr(
1644 &handle->domain, extended);
1645 channel->attr.extended.ptr = extended;
1646 }
1647
1648 /* Prepare the payload */
1649 memset(&lsm, 0, sizeof(lsm));
1650
cf0bcb51 1651 lsm.cmd_type = LTTNG_ENABLE_CHANNEL;
60160d2a 1652 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
7d29a247 1653
e1b624d0
JG
1654 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1655 sizeof(lsm.session.name));
1656 if (ret) {
1657 ret = -LTTNG_ERR_INVALID;
1658 goto end;
1659 }
cd80958d 1660
999af9c1
JR
1661 ret = lttng_channel_serialize(channel, &buffer);
1662 if (ret) {
1663 ret = -LTTNG_ERR_FATAL;
1664 goto end;
1665 }
1666
1667 lsm.u.channel.length = buffer.size;
1668
1669 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(
1670 &lsm, buffer.data, buffer.size, NULL);
e1b624d0 1671end:
999af9c1
JR
1672 lttng_channel_destroy(channel);
1673 lttng_dynamic_buffer_reset(&buffer);
e1b624d0 1674 return ret;
8c0faa1d 1675}
1df4dedd 1676
2ef84c95 1677/*
9ae110e2
JG
1678 * All tracing will be stopped for registered events of the channel.
1679 * Returns size of returned session payload data or a negative error code.
2ef84c95 1680 */
cd80958d 1681int lttng_disable_channel(struct lttng_handle *handle, const char *name)
2ef84c95 1682{
e1b624d0 1683 int ret;
cd80958d
DG
1684 struct lttcomm_session_msg lsm;
1685
9ae110e2 1686 /* Safety check. Both are mandatory. */
9d697d3d 1687 if (handle == NULL || name == NULL) {
2f70b271 1688 return -LTTNG_ERR_INVALID;
cd80958d
DG
1689 }
1690
441c16a7
MD
1691 memset(&lsm, 0, sizeof(lsm));
1692
cd80958d 1693 lsm.cmd_type = LTTNG_DISABLE_CHANNEL;
1df4dedd 1694
e1b624d0 1695 ret = lttng_strncpy(lsm.u.disable.channel_name, name,
9d697d3d 1696 sizeof(lsm.u.disable.channel_name));
e1b624d0
JG
1697 if (ret) {
1698 ret = -LTTNG_ERR_INVALID;
1699 goto end;
1700 }
9d697d3d 1701
60160d2a 1702 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
cd80958d 1703
e1b624d0
JG
1704 ret = lttng_strncpy(lsm.session.name, handle->session_name,
1705 sizeof(lsm.session.name));
1706 if (ret) {
1707 ret = -LTTNG_ERR_INVALID;
1708 goto end;
1709 }
cd80958d 1710
e1b624d0
JG
1711 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
1712end:
1713 return ret;
ca95a216
DG
1714}
1715
fac6795d 1716/*
9ae110e2
JG
1717 * Lists all available tracepoints of domain.
1718 * Sets the contents of the events array.
1719 * Returns the number of lttng_event entries in events;
1720 * on error, returns a negative value.
fac6795d 1721 */
cd80958d 1722int lttng_list_tracepoints(struct lttng_handle *handle,
2a71efd5 1723 struct lttng_event **events)
fac6795d 1724{
052da939 1725 int ret;
cd80958d
DG
1726 struct lttcomm_session_msg lsm;
1727
9d697d3d 1728 if (handle == NULL) {
2f70b271 1729 return -LTTNG_ERR_INVALID;
cd80958d 1730 }
fac6795d 1731
53efb85a 1732 memset(&lsm, 0, sizeof(lsm));
cd80958d 1733 lsm.cmd_type = LTTNG_LIST_TRACEPOINTS;
60160d2a 1734 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
2a71efd5 1735
cac3069d 1736 ret = lttng_ctl_ask_sessiond(&lsm, (void **) events);
052da939
DG
1737 if (ret < 0) {
1738 return ret;
eb354453 1739 }
fac6795d 1740
9f19cc17 1741 return ret / sizeof(struct lttng_event);
fac6795d
DG
1742}
1743
f37d259d 1744/*
9ae110e2
JG
1745 * Lists all available tracepoint fields of domain.
1746 * Sets the contents of the event field array.
1747 * Returns the number of lttng_event_field entries in events;
1748 * on error, returns a negative value.
f37d259d
MD
1749 */
1750int lttng_list_tracepoint_fields(struct lttng_handle *handle,
1751 struct lttng_event_field **fields)
1752{
1753 int ret;
1754 struct lttcomm_session_msg lsm;
1755
1756 if (handle == NULL) {
2f70b271 1757 return -LTTNG_ERR_INVALID;
f37d259d
MD
1758 }
1759
53efb85a 1760 memset(&lsm, 0, sizeof(lsm));
f37d259d 1761 lsm.cmd_type = LTTNG_LIST_TRACEPOINT_FIELDS;
60160d2a 1762 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
f37d259d 1763
cac3069d 1764 ret = lttng_ctl_ask_sessiond(&lsm, (void **) fields);
f37d259d
MD
1765 if (ret < 0) {
1766 return ret;
1767 }
1768
1769 return ret / sizeof(struct lttng_event_field);
1770}
1771
834978fd 1772/*
9ae110e2
JG
1773 * Lists all available kernel system calls. Allocates and sets the contents of
1774 * the events array.
834978fd 1775 *
9ae110e2
JG
1776 * Returns the number of lttng_event entries in events; on error, returns a
1777 * negative value.
834978fd
DG
1778 */
1779int lttng_list_syscalls(struct lttng_event **events)
1780{
1781 int ret;
1782 struct lttcomm_session_msg lsm;
1783
1784 if (!events) {
1785 return -LTTNG_ERR_INVALID;
1786 }
1787
1788 memset(&lsm, 0, sizeof(lsm));
1789 lsm.cmd_type = LTTNG_LIST_SYSCALLS;
1790 /* Force kernel domain for system calls. */
1791 lsm.domain.type = LTTNG_DOMAIN_KERNEL;
1792
1793 ret = lttng_ctl_ask_sessiond(&lsm, (void **) events);
1794 if (ret < 0) {
1795 return ret;
1796 }
1797
1798 return ret / sizeof(struct lttng_event);
1799}
1800
1657e9bb 1801/*
9ae110e2 1802 * Returns a human readable string describing
8346beb6 1803 * the error code (positive or negative value).
1657e9bb 1804 */
9a745bc7 1805const char *lttng_strerror(int code)
1657e9bb 1806{
8346beb6
PP
1807 if (code > 0) {
1808 code = -code;
1809 }
1810
f73fabfd 1811 return error_get_str(code);
1657e9bb
DG
1812}
1813
b178f53e
JG
1814enum lttng_error_code lttng_create_session_ext(
1815 struct lttng_session_descriptor *session_descriptor)
1816{
1817 enum lttng_error_code ret_code;
1818 struct lttcomm_session_msg lsm = {
1819 .cmd_type = LTTNG_CREATE_SESSION_EXT,
1820 };
1821 void *reply = NULL;
1822 struct lttng_buffer_view reply_view;
1823 int reply_ret;
1824 bool sessiond_must_generate_ouput;
1825 struct lttng_dynamic_buffer payload;
1826 int ret;
1827 size_t descriptor_size;
1828 struct lttng_session_descriptor *descriptor_reply = NULL;
1829
1830 lttng_dynamic_buffer_init(&payload);
1831 if (!session_descriptor) {
1832 ret_code = LTTNG_ERR_INVALID;
1833 goto end;
1834 }
1835
1836 sessiond_must_generate_ouput =
1837 !lttng_session_descriptor_is_output_destination_initialized(
1838 session_descriptor);
1839 if (sessiond_must_generate_ouput) {
1840 const char *home_dir = utils_get_home_dir();
1841 size_t home_dir_len = home_dir ? strlen(home_dir) + 1 : 0;
1842
1843 if (!home_dir || home_dir_len > LTTNG_PATH_MAX) {
1844 ret_code = LTTNG_ERR_FATAL;
1845 goto end;
1846 }
1847
1848 lsm.u.create_session.home_dir_size = (uint16_t) home_dir_len;
1849 ret = lttng_dynamic_buffer_append(&payload, home_dir,
1850 home_dir_len);
1851 if (ret) {
1852 ret_code = LTTNG_ERR_NOMEM;
1853 goto end;
1854 }
1855 }
1856
1857 descriptor_size = payload.size;
1858 ret = lttng_session_descriptor_serialize(session_descriptor,
1859 &payload);
1860 if (ret) {
1861 ret_code = LTTNG_ERR_INVALID;
1862 goto end;
1863 }
1864 descriptor_size = payload.size - descriptor_size;
1865 lsm.u.create_session.session_descriptor_size = descriptor_size;
1866
1867 /* Command returns a session descriptor on success. */
1868 reply_ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, payload.data,
1869 payload.size, &reply);
1870 if (reply_ret < 0) {
4bd69c5f 1871 ret_code = (lttng_error_code) -reply_ret;
b178f53e
JG
1872 goto end;
1873 } else if (reply_ret == 0) {
1874 /* Socket unexpectedly closed by the session daemon. */
1875 ret_code = LTTNG_ERR_FATAL;
1876 goto end;
1877 }
1878
4bd69c5f 1879 reply_view = lttng_buffer_view_init((const char *) reply, 0, reply_ret);
b178f53e
JG
1880 ret = lttng_session_descriptor_create_from_buffer(&reply_view,
1881 &descriptor_reply);
1882 if (ret < 0) {
1883 ret_code = LTTNG_ERR_FATAL;
1884 goto end;
1885 }
1886 ret_code = LTTNG_OK;
1887 lttng_session_descriptor_assign(session_descriptor, descriptor_reply);
1888end:
1889 free(reply);
1890 lttng_dynamic_buffer_reset(&payload);
1891 lttng_session_descriptor_destroy(descriptor_reply);
1892 return ret_code;
1893}
1894
aaf97519 1895/*
b178f53e 1896 * Create a new session using name and url for destination.
a4b92340 1897 *
780d4bb8 1898 * Return 0 on success else a negative LTTng error code.
00e2e675 1899 */
a4b92340 1900int lttng_create_session(const char *name, const char *url)
00e2e675 1901{
3dd05a85 1902 int ret;
a4b92340 1903 ssize_t size;
a4b92340 1904 struct lttng_uri *uris = NULL;
b178f53e
JG
1905 struct lttng_session_descriptor *descriptor = NULL;
1906 enum lttng_error_code ret_code;
00e2e675 1907
b178f53e
JG
1908 if (!name) {
1909 ret = -LTTNG_ERR_INVALID;
1910 goto end;
00e2e675
DG
1911 }
1912
bc894455 1913 size = uri_parse_str_urls(url, NULL, &uris);
a4b92340 1914 if (size < 0) {
b178f53e
JG
1915 ret = -LTTNG_ERR_INVALID;
1916 goto end;
1917 }
1918 switch (size) {
1919 case 0:
1920 descriptor = lttng_session_descriptor_create(name);
1921 break;
1922 case 1:
1923 if (uris[0].dtype != LTTNG_DST_PATH) {
1924 ret = -LTTNG_ERR_INVALID;
1925 goto end;
1926 }
1927 descriptor = lttng_session_descriptor_local_create(name,
1928 uris[0].dst.path);
1929 break;
1930 case 2:
1931 descriptor = lttng_session_descriptor_network_create(name, url,
1932 NULL);
1933 break;
1934 default:
1935 ret = -LTTNG_ERR_INVALID;
1936 goto end;
1937 }
1938 if (!descriptor) {
1939 ret = -LTTNG_ERR_INVALID;
1940 goto end;
00e2e675 1941 }
b178f53e
JG
1942 ret_code = lttng_create_session_ext(descriptor);
1943 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
1944end:
1945 lttng_session_descriptor_destroy(descriptor);
1946 free(uris);
1947 return ret;
1948}
00e2e675 1949
b178f53e
JG
1950/*
1951 * Create a session exclusively used for snapshot.
1952 *
780d4bb8 1953 * Return 0 on success else a negative LTTng error code.
b178f53e
JG
1954 */
1955int lttng_create_session_snapshot(const char *name, const char *snapshot_url)
1956{
1957 int ret;
1958 enum lttng_error_code ret_code;
1959 ssize_t size;
1960 struct lttng_uri *uris = NULL;
1961 struct lttng_session_descriptor *descriptor = NULL;
a4b92340 1962
b178f53e
JG
1963 if (!name) {
1964 ret = -LTTNG_ERR_INVALID;
1965 goto end;
1966 }
1967
1968 size = uri_parse_str_urls(snapshot_url, NULL, &uris);
1969 if (size < 0) {
1970 ret = -LTTNG_ERR_INVALID;
1971 goto end;
1972 }
1973 /*
1974 * If the user does not specify a custom subdir, use the session name.
1975 */
1976 if (size > 0 && uris[0].dtype != LTTNG_DST_PATH &&
1977 strlen(uris[0].subdir) == 0) {
1978 ret = snprintf(uris[0].subdir, sizeof(uris[0].subdir), "%s",
1979 name);
1980 if (ret < 0) {
1981 PERROR("Failed to set session name as network destination sub-directory");
1982 ret = -LTTNG_ERR_FATAL;
1983 goto end;
1984 } else if (ret >= sizeof(uris[0].subdir)) {
1985 /* Truncated output. */
1986 ret = -LTTNG_ERR_INVALID;
1987 goto end;
1988 }
1989 }
3dd05a85 1990
b178f53e
JG
1991 switch (size) {
1992 case 0:
1993 descriptor = lttng_session_descriptor_snapshot_create(name);
1994 break;
1995 case 1:
1996 if (uris[0].dtype != LTTNG_DST_PATH) {
1997 ret = -LTTNG_ERR_INVALID;
1998 goto end;
1999 }
2000 descriptor = lttng_session_descriptor_snapshot_local_create(
2001 name,
2002 uris[0].dst.path);
2003 break;
2004 case 2:
2005 descriptor = lttng_session_descriptor_snapshot_network_create(
2006 name,
2007 snapshot_url,
2008 NULL);
2009 break;
2010 default:
2011 ret = -LTTNG_ERR_INVALID;
2012 goto end;
2013 }
2014 if (!descriptor) {
2015 ret = -LTTNG_ERR_INVALID;
2016 goto end;
2017 }
2018 ret_code = lttng_create_session_ext(descriptor);
2019 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2020end:
2021 lttng_session_descriptor_destroy(descriptor);
3dd05a85
DG
2022 free(uris);
2023 return ret;
00e2e675
DG
2024}
2025
b178f53e
JG
2026/*
2027 * Create a session exclusively used for live.
2028 *
780d4bb8 2029 * Return 0 on success else a negative LTTng error code.
b178f53e
JG
2030 */
2031int lttng_create_session_live(const char *name, const char *url,
2032 unsigned int timer_interval)
2033{
2034 int ret;
2035 enum lttng_error_code ret_code;
2036 struct lttng_session_descriptor *descriptor = NULL;
2037
2038 if (!name) {
2039 ret = -LTTNG_ERR_INVALID;
2040 goto end;
2041 }
2042
2043 if (url) {
2044 descriptor = lttng_session_descriptor_live_network_create(
2045 name, url, NULL, timer_interval);
2046 } else {
2047 descriptor = lttng_session_descriptor_live_create(
2048 name, timer_interval);
2049 }
2050 if (!descriptor) {
2051 ret = -LTTNG_ERR_INVALID;
2052 goto end;
2053 }
2054 ret_code = lttng_create_session_ext(descriptor);
2055 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
2056end:
2057 lttng_session_descriptor_destroy(descriptor);
2058 return ret;
2059}
2060
e20ee7c2
JD
2061/*
2062 * Stop the session and wait for the data before destroying it
780d4bb8
MD
2063 *
2064 * Return 0 on success else a negative LTTng error code.
e20ee7c2
JD
2065 */
2066int lttng_destroy_session(const char *session_name)
2067{
2068 int ret;
3e3665b8
JG
2069 enum lttng_error_code ret_code;
2070 enum lttng_destruction_handle_status status;
2071 struct lttng_destruction_handle *handle = NULL;
e20ee7c2
JD
2072
2073 /*
3e3665b8
JG
2074 * Stop the tracing and wait for the data to be
2075 * consumed.
e20ee7c2
JD
2076 */
2077 ret = _lttng_stop_tracing(session_name, 1);
2078 if (ret && ret != -LTTNG_ERR_TRACE_ALREADY_STOPPED) {
2079 goto end;
2080 }
2081
3e3665b8
JG
2082 ret_code = lttng_destroy_session_ext(session_name, &handle);
2083 if (ret_code != LTTNG_OK) {
2084 ret = (int) -ret_code;
2085 goto end;
2086 }
a0377dfe 2087 LTTNG_ASSERT(handle);
3e3665b8
JG
2088
2089 /* Block until the completion of the destruction of the session. */
2090 status = lttng_destruction_handle_wait_for_completion(handle, -1);
2091 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_COMPLETED) {
2092 ret = -LTTNG_ERR_UNK;
2093 goto end;
2094 }
2095
2096 status = lttng_destruction_handle_get_result(handle, &ret_code);
2097 if (status != LTTNG_DESTRUCTION_HANDLE_STATUS_OK) {
2098 ret = -LTTNG_ERR_UNK;
2099 goto end;
2100 }
780d4bb8 2101 ret = ret_code == LTTNG_OK ? 0 : -ret_code;
e20ee7c2 2102end:
3e3665b8 2103 lttng_destruction_handle_destroy(handle);
e20ee7c2
JD
2104 return ret;
2105}
2106
2107/*
2108 * Destroy the session without waiting for the data.
2109 */
2110int lttng_destroy_session_no_wait(const char *session_name)
2111{
3e3665b8 2112 enum lttng_error_code ret_code;
e20ee7c2 2113
3e3665b8 2114 ret_code = lttng_destroy_session_ext(session_name, NULL);
3204017e 2115 return ret_code == LTTNG_OK ? 0 : -ret_code;
e20ee7c2
JD
2116}
2117
57167058 2118/*
9ae110e2
JG
2119 * Ask the session daemon for all available sessions.
2120 * Sets the contents of the sessions array.
2121 * Returns the number of lttng_session entries in sessions;
2122 * on error, returns a negative value.
57167058 2123 */
b178f53e 2124int lttng_list_sessions(struct lttng_session **out_sessions)
57167058 2125{
ca95a216 2126 int ret;
cd80958d 2127 struct lttcomm_session_msg lsm;
b178f53e
JG
2128 const size_t session_size = sizeof(struct lttng_session) +
2129 sizeof(struct lttng_session_extended);
2130 size_t session_count, i;
2131 struct lttng_session_extended *sessions_extended_begin;
2132 struct lttng_session *sessions = NULL;
57167058 2133
53efb85a 2134 memset(&lsm, 0, sizeof(lsm));
cd80958d 2135 lsm.cmd_type = LTTNG_LIST_SESSIONS;
985aea18
MD
2136 /*
2137 * Initialize out_sessions to NULL so it is initialized when
2138 * lttng_list_sessions returns 0, thus allowing *out_sessions to
2139 * be subsequently freed.
2140 */
2141 *out_sessions = NULL;
b178f53e
JG
2142 ret = lttng_ctl_ask_sessiond(&lsm, (void**) &sessions);
2143 if (ret <= 0) {
b178f53e
JG
2144 goto end;
2145 }
2146 if (!sessions) {
2147 ret = -LTTNG_ERR_FATAL;
2148 goto end;
2149 }
2150
2151 if (ret % session_size) {
2152 ret = -LTTNG_ERR_UNK;
2153 free(sessions);
b178f53e 2154 goto end;
57167058 2155 }
b178f53e
JG
2156 session_count = (size_t) ret / session_size;
2157 sessions_extended_begin = (struct lttng_session_extended *)
2158 (&sessions[session_count]);
57167058 2159
b178f53e
JG
2160 /* Set extended session info pointers. */
2161 for (i = 0; i < session_count; i++) {
2162 struct lttng_session *session = &sessions[i];
2163 struct lttng_session_extended *extended =
2164 &(sessions_extended_begin[i]);
2165
2166 session->extended.ptr = extended;
2167 }
2168
2169 ret = (int) session_count;
2170 *out_sessions = sessions;
2171end:
2172 return ret;
2173}
2174
2175enum lttng_error_code lttng_session_get_creation_time(
2176 const struct lttng_session *session, uint64_t *creation_time)
2177{
2178 enum lttng_error_code ret = LTTNG_OK;
2179 struct lttng_session_extended *extended;
2180
2181 if (!session || !creation_time || !session->extended.ptr) {
2182 ret = LTTNG_ERR_INVALID;
2183 goto end;
2184 }
2185
4bd69c5f 2186 extended = (lttng_session_extended *) session->extended.ptr;
b178f53e
JG
2187 if (!extended->creation_time.is_set) {
2188 /* Not created on the session daemon yet. */
2189 ret = LTTNG_ERR_SESSION_NOT_EXIST;
2190 goto end;
2191 }
2192 *creation_time = extended->creation_time.value;
2193end:
2194 return ret;
57167058
DG
2195}
2196
d7ba1388
MD
2197int lttng_set_session_shm_path(const char *session_name,
2198 const char *shm_path)
2199{
e1b624d0 2200 int ret;
d7ba1388
MD
2201 struct lttcomm_session_msg lsm;
2202
2203 if (session_name == NULL) {
2204 return -LTTNG_ERR_INVALID;
2205 }
2206
2207 memset(&lsm, 0, sizeof(lsm));
2208 lsm.cmd_type = LTTNG_SET_SESSION_SHM_PATH;
2209
e1b624d0 2210 ret = lttng_strncpy(lsm.session.name, session_name,
d7ba1388 2211 sizeof(lsm.session.name));
e1b624d0
JG
2212 if (ret) {
2213 ret = -LTTNG_ERR_INVALID;
2214 goto end;
2215 }
2216
2217 ret = lttng_strncpy(lsm.u.set_shm_path.shm_path, shm_path ?: "",
d7ba1388 2218 sizeof(lsm.u.set_shm_path.shm_path));
e1b624d0
JG
2219 if (ret) {
2220 ret = -LTTNG_ERR_INVALID;
2221 goto end;
2222 }
d7ba1388 2223
e1b624d0
JG
2224 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
2225end:
2226 return ret;
d7ba1388
MD
2227}
2228
9f19cc17 2229/*
9ae110e2
JG
2230 * Ask the session daemon for all available domains of a session.
2231 * Sets the contents of the domains array.
2232 * Returns the number of lttng_domain entries in domains;
2233 * on error, returns a negative value.
9f19cc17 2234 */
330be774 2235int lttng_list_domains(const char *session_name,
cd80958d 2236 struct lttng_domain **domains)
9f19cc17
DG
2237{
2238 int ret;
cd80958d
DG
2239 struct lttcomm_session_msg lsm;
2240
330be774 2241 if (session_name == NULL) {
e1b624d0
JG
2242 ret = -LTTNG_ERR_INVALID;
2243 goto error;
cd80958d 2244 }
9f19cc17 2245
53efb85a 2246 memset(&lsm, 0, sizeof(lsm));
cd80958d
DG
2247 lsm.cmd_type = LTTNG_LIST_DOMAINS;
2248
e1b624d0 2249 ret = lttng_strncpy(lsm.session.name, session_name,
cac3069d 2250 sizeof(lsm.session.name));
e1b624d0
JG
2251 if (ret) {
2252 ret = -LTTNG_ERR_INVALID;
2253 goto error;
2254 }
cd80958d 2255
cac3069d 2256 ret = lttng_ctl_ask_sessiond(&lsm, (void**) domains);
9f19cc17 2257 if (ret < 0) {
e1b624d0 2258 goto error;
9f19cc17
DG
2259 }
2260
2261 return ret / sizeof(struct lttng_domain);
e1b624d0
JG
2262error:
2263 return ret;
9f19cc17
DG
2264}
2265
2266/*
9ae110e2
JG
2267 * Ask the session daemon for all available channels of a session.
2268 * Sets the contents of the channels array.
2269 * Returns the number of lttng_channel entries in channels;
2270 * on error, returns a negative value.
9f19cc17 2271 */
cd80958d
DG
2272int lttng_list_channels(struct lttng_handle *handle,
2273 struct lttng_channel **channels)
9f19cc17 2274{
999af9c1 2275 int ret, total_payload_received;
cd80958d 2276 struct lttcomm_session_msg lsm;
999af9c1
JR
2277 char *reception_buffer = NULL;
2278 size_t cmd_header_len = 0;
2279 struct lttcomm_list_command_header *cmd_header = NULL;
2280 struct lttng_dynamic_buffer tmp_buffer;
2281
2282 lttng_dynamic_buffer_init(&tmp_buffer);
cd80958d 2283
9d697d3d 2284 if (handle == NULL) {
53e367f9
JG
2285 ret = -LTTNG_ERR_INVALID;
2286 goto end;
cd80958d
DG
2287 }
2288
53efb85a 2289 memset(&lsm, 0, sizeof(lsm));
cd80958d 2290 lsm.cmd_type = LTTNG_LIST_CHANNELS;
e1b624d0 2291 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 2292 sizeof(lsm.session.name));
e1b624d0
JG
2293 if (ret) {
2294 ret = -LTTNG_ERR_INVALID;
2295 goto end;
2296 }
9f19cc17 2297
60160d2a 2298 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
9f19cc17 2299
999af9c1
JR
2300 ret = lttng_ctl_ask_sessiond_fds_varlen(&lsm, NULL, 0, NULL, 0,
2301 (void **) &reception_buffer, (void **) &cmd_header,
2302 &cmd_header_len);
9f19cc17 2303 if (ret < 0) {
53e367f9
JG
2304 goto end;
2305 }
2306
999af9c1
JR
2307 total_payload_received = ret;
2308
2309 if (cmd_header_len != sizeof(*cmd_header)) {
2310 ret = -LTTNG_ERR_FATAL;
53e367f9 2311 goto end;
9f19cc17
DG
2312 }
2313
999af9c1
JR
2314 if (!cmd_header) {
2315 ret = LTTNG_ERR_UNK;
2316 goto end;
2317 }
2318
2319 if (cmd_header->count > INT_MAX) {
2320 ret = -LTTNG_ERR_OVERFLOW;
2321 goto end;
2322 }
53e367f9 2323
999af9c1
JR
2324 {
2325 enum lttng_error_code ret_code;
2326 const struct lttng_buffer_view events_view =
2327 lttng_buffer_view_init(reception_buffer, 0,
2328 total_payload_received);
2329
2330 ret_code = lttng_channels_create_and_flatten_from_buffer(
2331 &events_view, cmd_header->count, channels);
2332 if (ret_code != LTTNG_OK) {
2333 ret = -ret_code;
2334 goto end;
2335 }
53e367f9
JG
2336 }
2337
999af9c1 2338 ret = (int) cmd_header->count;
53e367f9 2339end:
999af9c1
JR
2340 free(cmd_header);
2341 free(reception_buffer);
53e367f9 2342 return ret;
9f19cc17
DG
2343}
2344
2345/*
9ae110e2
JG
2346 * Ask the session daemon for all available events of a session channel.
2347 * Sets the contents of the events array.
2348 * Returns the number of lttng_event entries in events;
2349 * on error, returns a negative value.
9f19cc17 2350 */
cd80958d
DG
2351int lttng_list_events(struct lttng_handle *handle,
2352 const char *channel_name, struct lttng_event **events)
9f19cc17
DG
2353{
2354 int ret;
e368fb43
JG
2355 struct lttcomm_session_msg lsm = {};
2356 const struct lttcomm_event_command_header *cmd_header = NULL;
b4e3ceb9 2357 uint32_t nb_events, i;
4bd69c5f 2358 const char *comm_ext_at;
56f0bc67 2359 struct lttng_dynamic_buffer listing;
de453daa 2360 size_t storage_req;
e368fb43
JG
2361 struct lttng_payload payload;
2362 struct lttng_payload payload_copy;
2363 struct lttng_payload_view lsm_view =
2364 lttng_payload_view_init_from_buffer(
2365 (const char *) &lsm, 0, sizeof(lsm));
2366 struct lttng_buffer_view cmd_header_view;
2367 struct lttng_buffer_view cmd_payload_view;
2368 struct lttng_buffer_view flat_events_view;
2369 struct lttng_buffer_view ext_view;
9f19cc17 2370
9d697d3d
DG
2371 /* Safety check. An handle and channel name are mandatory */
2372 if (handle == NULL || channel_name == NULL) {
e368fb43
JG
2373 ret = -LTTNG_ERR_INVALID;
2374 goto end;
cd80958d
DG
2375 }
2376
e368fb43
JG
2377 lttng_payload_init(&payload);
2378 lttng_payload_init(&payload_copy);
2379
cd80958d 2380 lsm.cmd_type = LTTNG_LIST_EVENTS;
e1b624d0 2381 ret = lttng_strncpy(lsm.session.name, handle->session_name,
cd80958d 2382 sizeof(lsm.session.name));
e1b624d0
JG
2383 if (ret) {
2384 ret = -LTTNG_ERR_INVALID;
2385 goto end;
2386 }
2387
2388 ret = lttng_strncpy(lsm.u.list.channel_name, channel_name,
cd80958d 2389 sizeof(lsm.u.list.channel_name));
e1b624d0
JG
2390 if (ret) {
2391 ret = -LTTNG_ERR_INVALID;
2392 goto end;
2393 }
2394
60160d2a 2395 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
9f19cc17 2396
e368fb43 2397 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &payload);
9f19cc17 2398 if (ret < 0) {
de453daa 2399 goto end;
9f19cc17
DG
2400 }
2401
e368fb43
JG
2402 /*
2403 * A copy of the payload is performed since it will be
2404 * consumed twice. Consuming the same payload twice is invalid since
2405 * it will cause any received file descriptor to become "shared"
2406 * between different instances of the resulting objects.
2407 */
2408 ret = lttng_payload_copy(&payload, &payload_copy);
2409 if (ret) {
2410 ret = -LTTNG_ERR_NOMEM;
2411 goto end;
2412 }
2413
2414 cmd_header_view = lttng_buffer_view_from_dynamic_buffer(
2415 &payload.buffer, 0, sizeof(*cmd_header));
3e6e0df2 2416 if (!lttng_buffer_view_is_valid(&cmd_header_view)) {
e368fb43 2417 ret = -LTTNG_ERR_INVALID_PROTOCOL;
d9f484bc
JG
2418 goto end;
2419 }
2420
e368fb43
JG
2421 cmd_header = (typeof(cmd_header)) cmd_header_view.data;
2422
b4e3ceb9
PP
2423 /* Set number of events and free command header */
2424 nb_events = cmd_header->nb_events;
2425 if (nb_events > INT_MAX) {
55c9e7ca 2426 ret = -LTTNG_ERR_OVERFLOW;
de453daa 2427 goto end;
b4e3ceb9 2428 }
e368fb43
JG
2429
2430 cmd_payload_view = lttng_buffer_view_from_dynamic_buffer(
2431 &payload.buffer, sizeof(*cmd_header), -1);
b4e3ceb9 2432
de453daa
JG
2433 /*
2434 * The buffer that is returned must contain a "flat" version of
2435 * the events that are returned. In other words, all pointers
2436 * within an lttng_event must point to a location within the returned
56f0bc67
JG
2437 * buffer so that the user may free everything by simply calling free()
2438 * on the returned buffer. This is needed in order to maintain API
de453daa
JG
2439 * compatibility.
2440 *
56f0bc67
JG
2441 * A first pass is performed to compute the size of the buffer that
2442 * must be allocated. A second pass is then performed to setup
de453daa
JG
2443 * the returned events so that their members always point within the
2444 * buffer.
2445 *
2446 * The layout of the returned buffer is as follows:
2447 * - struct lttng_event[nb_events],
2448 * - nb_events times the following:
2449 * - struct lttng_event_extended,
2450 * - flattened version of userspace_probe_location
2451 * - filter_expression
2452 * - exclusions
2453 * - padding to align to 64-bits
2454 */
e368fb43
JG
2455 ext_view = lttng_buffer_view_from_view(&cmd_payload_view,
2456 nb_events * sizeof(struct lttng_event), -1);
2457 comm_ext_at = ext_view.data;
de453daa 2458 storage_req = nb_events * sizeof(struct lttng_event);
e368fb43
JG
2459 {
2460 struct lttng_payload_view payload_view =
2461 lttng_payload_view_from_payload(&payload, 0, -1);
b4e3ceb9 2462
e368fb43
JG
2463 for (i = 0; i < nb_events; i++) {
2464 const struct lttcomm_event_extended_header *ext_comm =
2465 (struct lttcomm_event_extended_header *)
2466 comm_ext_at;
2467 int probe_storage_req = 0;
b4e3ceb9 2468
e368fb43
JG
2469 comm_ext_at += sizeof(*ext_comm);
2470 comm_ext_at += ext_comm->filter_len;
2471 comm_ext_at += ext_comm->nb_exclusions *
2472 LTTNG_SYMBOL_NAME_LEN;
2473
2474 if (ext_comm->userspace_probe_location_len) {
2475 struct lttng_userspace_probe_location
2476 *probe_location = NULL;
2477 struct lttng_payload_view probe_location_view = lttng_payload_view_from_view(
2478 &payload_view,
2479 (const char *) comm_ext_at -
2480 payload_view.buffer.data,
2481 ext_comm->userspace_probe_location_len);
56f0bc67 2482
3e6e0df2
JG
2483 if (!lttng_payload_view_is_valid(&probe_location_view)) {
2484 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2485 goto end;
2486 }
2487
e368fb43
JG
2488 /*
2489 * Create a temporary userspace probe location
2490 * to determine the size needed by a "flattened"
2491 * version of that same probe location.
2492 */
2493 ret = lttng_userspace_probe_location_create_from_payload(
2494 &probe_location_view,
2495 &probe_location);
2496 if (ret < 0) {
2497 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2498 goto end;
2499 }
56f0bc67 2500
e368fb43
JG
2501 ret = lttng_userspace_probe_location_flatten(
2502 probe_location, NULL);
2503 lttng_userspace_probe_location_destroy(
2504 probe_location);
2505 if (ret < 0) {
2506 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2507 goto end;
2508 }
56f0bc67 2509
e368fb43
JG
2510 probe_storage_req = ret;
2511 comm_ext_at += ext_comm->userspace_probe_location_len;
56f0bc67
JG
2512 }
2513
e368fb43
JG
2514 storage_req += sizeof(struct lttng_event_extended);
2515 storage_req += ext_comm->filter_len;
2516 storage_req += ext_comm->nb_exclusions *
2517 LTTNG_SYMBOL_NAME_LEN;
2518 /* Padding to ensure the flat probe is aligned. */
1cbd136b 2519 storage_req = lttng_align_ceil(storage_req, sizeof(uint64_t));
e368fb43 2520 storage_req += probe_storage_req;
56f0bc67 2521 }
b4e3ceb9
PP
2522 }
2523
56f0bc67
JG
2524 lttng_dynamic_buffer_init(&listing);
2525 /*
2526 * We must ensure that "listing" is never resized so as to preserve
2527 * the validity of the flattened objects.
2528 */
2529 ret = lttng_dynamic_buffer_set_capacity(&listing, storage_req);
2530 if (ret) {
2531 ret = -LTTNG_ERR_NOMEM;
de453daa
JG
2532 goto end;
2533 }
56f0bc67 2534
e368fb43
JG
2535 cmd_payload_view = lttng_buffer_view_from_dynamic_buffer(
2536 &payload_copy.buffer, sizeof(*cmd_header), -1);
2537 flat_events_view = lttng_buffer_view_from_view(&cmd_payload_view, 0,
56f0bc67 2538 nb_events * sizeof(struct lttng_event));
e368fb43 2539 ret = lttng_dynamic_buffer_append_view(&listing, &flat_events_view);
56f0bc67
JG
2540 if (ret) {
2541 ret = -LTTNG_ERR_NOMEM;
2542 goto free_dynamic_buffer;
2543 }
de453daa 2544
e368fb43
JG
2545 ext_view = lttng_buffer_view_from_view(&cmd_payload_view,
2546 nb_events * sizeof(struct lttng_event), -1);
2547 comm_ext_at = ext_view.data;
2548
2549 {
2550 struct lttng_payload_view payload_copy_view =
2551 lttng_payload_view_from_payload(
2552 &payload_copy, 0, -1);
2553
2554 for (i = 0; i < nb_events; i++) {
2555 struct lttng_event *event = (typeof(event))(
2556 listing.data +
2557 (sizeof(struct lttng_event) * i));
2558 const struct lttcomm_event_extended_header *ext_comm =
2559 (typeof(ext_comm)) comm_ext_at;
2560 struct lttng_event_extended *event_extended =
2561 (typeof(event_extended))(listing.data +
2562 listing.size);
2563
2564 /* Insert struct lttng_event_extended. */
2565 ret = lttng_dynamic_buffer_set_size(&listing,
2566 listing.size + sizeof(*event_extended));
56f0bc67
JG
2567 if (ret) {
2568 ret = -LTTNG_ERR_NOMEM;
2569 goto free_dynamic_buffer;
2570 }
e368fb43
JG
2571 event->extended.ptr = event_extended;
2572
2573 comm_ext_at += sizeof(*ext_comm);
2574
2575 /* Insert filter expression. */
2576 if (ext_comm->filter_len) {
2577 event_extended->filter_expression =
2578 listing.data + listing.size;
2579 ret = lttng_dynamic_buffer_append(&listing,
2580 comm_ext_at,
2581 ext_comm->filter_len);
2582 if (ret) {
2583 ret = -LTTNG_ERR_NOMEM;
2584 goto free_dynamic_buffer;
2585 }
2586 comm_ext_at += ext_comm->filter_len;
2587 }
56f0bc67 2588
e368fb43
JG
2589 /* Insert exclusions. */
2590 if (ext_comm->nb_exclusions) {
2591 event_extended->exclusions.count =
2592 ext_comm->nb_exclusions;
2593 event_extended->exclusions.strings =
2594 listing.data + listing.size;
2595
2596 ret = lttng_dynamic_buffer_append(&listing,
2597 comm_ext_at,
2598 ext_comm->nb_exclusions *
2599 LTTNG_SYMBOL_NAME_LEN);
2600 if (ret) {
2601 ret = -LTTNG_ERR_NOMEM;
2602 goto free_dynamic_buffer;
2603 }
2604 comm_ext_at += ext_comm->nb_exclusions *
2605 LTTNG_SYMBOL_NAME_LEN;
2606 }
56f0bc67 2607
e368fb43
JG
2608 /* Insert padding to align to 64-bits. */
2609 ret = lttng_dynamic_buffer_set_size(&listing,
1cbd136b 2610 lttng_align_ceil(listing.size,
e368fb43 2611 sizeof(uint64_t)));
56f0bc67
JG
2612 if (ret) {
2613 ret = -LTTNG_ERR_NOMEM;
5e4f4898 2614 goto free_dynamic_buffer;
56f0bc67 2615 }
56f0bc67 2616
e368fb43
JG
2617 /* Insert flattened userspace probe location. */
2618 if (ext_comm->userspace_probe_location_len) {
2619 struct lttng_userspace_probe_location
2620 *probe_location = NULL;
2621 struct lttng_payload_view probe_location_view = lttng_payload_view_from_view(
2622 &payload_copy_view,
2623 (const char *) comm_ext_at -
2624 payload_copy_view.buffer.data,
2625 ext_comm->userspace_probe_location_len);
2626
3e6e0df2
JG
2627 if (!lttng_payload_view_is_valid(&probe_location_view)) {
2628 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2629 goto free_dynamic_buffer;
2630 }
2631
e368fb43
JG
2632 ret = lttng_userspace_probe_location_create_from_payload(
2633 &probe_location_view,
2634 &probe_location);
2635 if (ret < 0) {
2636 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2637 goto free_dynamic_buffer;
2638 }
de453daa 2639
e368fb43
JG
2640 event_extended->probe_location = (struct lttng_userspace_probe_location
2641 *) (listing.data +
2642 listing.size);
2643 ret = lttng_userspace_probe_location_flatten(
2644 probe_location, &listing);
2645 lttng_userspace_probe_location_destroy(
2646 probe_location);
2647 if (ret < 0) {
2648 ret = -LTTNG_ERR_PROBE_LOCATION_INVAL;
2649 goto free_dynamic_buffer;
2650 }
56f0bc67 2651
e368fb43 2652 comm_ext_at += ext_comm->userspace_probe_location_len;
56f0bc67 2653 }
56f0bc67 2654 }
de453daa
JG
2655 }
2656
56f0bc67
JG
2657 /* Don't reset listing buffer as we return its content. */
2658 *events = (struct lttng_event *) listing.data;
2659 lttng_dynamic_buffer_init(&listing);
de453daa 2660 ret = (int) nb_events;
56f0bc67
JG
2661free_dynamic_buffer:
2662 lttng_dynamic_buffer_reset(&listing);
de453daa 2663end:
e368fb43
JG
2664 lttng_payload_reset(&payload);
2665 lttng_payload_reset(&payload_copy);
b4e3ceb9 2666 return ret;
9f19cc17
DG
2667}
2668
fac6795d 2669/*
1c8d13c8
TD
2670 * Sets the tracing_group variable with name.
2671 * This function allocates memory pointed to by tracing_group.
2672 * On success, returns 0, on error, returns -1 (null name) or -ENOMEM.
fac6795d
DG
2673 */
2674int lttng_set_tracing_group(const char *name)
2675{
e32a3d0f 2676 int ret = 0;
76da29b6 2677 char *new_group;
e32a3d0f 2678
9d697d3d 2679 if (name == NULL) {
e32a3d0f
JG
2680 ret = -LTTNG_ERR_INVALID;
2681 goto end;
9d697d3d
DG
2682 }
2683
e32a3d0f
JG
2684 new_group = strdup(name);
2685 if (!new_group) {
2686 ret = -LTTNG_ERR_FATAL;
2687 goto end;
fac6795d
DG
2688 }
2689
76da29b6
JR
2690 free(tracing_group);
2691 tracing_group = new_group;
2692 new_group = NULL;
2693
e32a3d0f
JG
2694end:
2695 return ret;
fac6795d
DG
2696}
2697
cd80958d 2698int lttng_calibrate(struct lttng_handle *handle,
d0254c7c
MD
2699 struct lttng_calibrate *calibrate)
2700{
b812e5ca
PP
2701 /*
2702 * This command was removed in LTTng 2.9.
2703 */
2704 return -LTTNG_ERR_UND;
d0254c7c
MD
2705}
2706
5edd7e09
DG
2707/*
2708 * Set default channel attributes.
441c16a7 2709 * If either or both of the arguments are null, attr content is zeroe'd.
5edd7e09
DG
2710 */
2711void lttng_channel_set_default_attr(struct lttng_domain *domain,
2712 struct lttng_channel_attr *attr)
2713{
294851b0 2714 struct lttng_channel_extended *extended;
cf0bcb51 2715
5edd7e09
DG
2716 /* Safety check */
2717 if (attr == NULL || domain == NULL) {
2718 return;
2719 }
2720
999af9c1 2721 /* Save the pointer for later use */
294851b0 2722 extended = (struct lttng_channel_extended *) attr->extended.ptr;
eacaa7a6
DS
2723 memset(attr, 0, sizeof(struct lttng_channel_attr));
2724
0a9c6494
DG
2725 /* Same for all domains. */
2726 attr->overwrite = DEFAULT_CHANNEL_OVERWRITE;
2727 attr->tracefile_size = DEFAULT_CHANNEL_TRACEFILE_SIZE;
2728 attr->tracefile_count = DEFAULT_CHANNEL_TRACEFILE_COUNT;
2729
5edd7e09
DG
2730 switch (domain->type) {
2731 case LTTNG_DOMAIN_KERNEL:
9ae110e2
JG
2732 attr->switch_timer_interval =
2733 DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER;
d92ff3ef 2734 attr->read_timer_interval = DEFAULT_KERNEL_CHANNEL_READ_TIMER;
3e230f92 2735 attr->subbuf_size = default_get_kernel_channel_subbuf_size();
5edd7e09
DG
2736 attr->num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
2737 attr->output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
2738 break;
2739 case LTTNG_DOMAIN_UST:
0a9c6494
DG
2740 switch (domain->buf_type) {
2741 case LTTNG_BUFFER_PER_UID:
2742 attr->subbuf_size = default_get_ust_uid_channel_subbuf_size();
2743 attr->num_subbuf = DEFAULT_UST_UID_CHANNEL_SUBBUF_NUM;
2744 attr->output = DEFAULT_UST_UID_CHANNEL_OUTPUT;
9ae110e2
JG
2745 attr->switch_timer_interval =
2746 DEFAULT_UST_UID_CHANNEL_SWITCH_TIMER;
2747 attr->read_timer_interval =
2748 DEFAULT_UST_UID_CHANNEL_READ_TIMER;
0a9c6494
DG
2749 break;
2750 case LTTNG_BUFFER_PER_PID:
2751 default:
2752 attr->subbuf_size = default_get_ust_pid_channel_subbuf_size();
2753 attr->num_subbuf = DEFAULT_UST_PID_CHANNEL_SUBBUF_NUM;
2754 attr->output = DEFAULT_UST_PID_CHANNEL_OUTPUT;
9ae110e2
JG
2755 attr->switch_timer_interval =
2756 DEFAULT_UST_PID_CHANNEL_SWITCH_TIMER;
2757 attr->read_timer_interval =
2758 DEFAULT_UST_PID_CHANNEL_READ_TIMER;
0a9c6494
DG
2759 break;
2760 }
5edd7e09 2761 default:
441c16a7 2762 /* Default behavior: leave set to 0. */
5edd7e09
DG
2763 break;
2764 }
cf0bcb51 2765
999af9c1
JR
2766 if (extended) {
2767 lttng_channel_set_default_extended_attr(domain, extended);
2768 }
2769
2770 /* Reassign the extended pointer. */
cf0bcb51 2771 attr->extended.ptr = extended;
5edd7e09
DG
2772}
2773
5ba3702f
JG
2774int lttng_channel_get_discarded_event_count(struct lttng_channel *channel,
2775 uint64_t *discarded_events)
2776{
2777 int ret = 0;
cf0bcb51 2778 struct lttng_channel_extended *chan_ext;
5ba3702f
JG
2779
2780 if (!channel || !discarded_events) {
2781 ret = -LTTNG_ERR_INVALID;
2782 goto end;
2783 }
2784
4bd69c5f 2785 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
5ba3702f
JG
2786 if (!chan_ext) {
2787 /*
2788 * This can happen since the lttng_channel structure is
2789 * used for other tasks where this pointer is never set.
2790 */
2791 *discarded_events = 0;
2792 goto end;
2793 }
2794
2795 *discarded_events = chan_ext->discarded_events;
2796end:
2797 return ret;
2798}
2799
2800int lttng_channel_get_lost_packet_count(struct lttng_channel *channel,
2801 uint64_t *lost_packets)
2802{
2803 int ret = 0;
cf0bcb51 2804 struct lttng_channel_extended *chan_ext;
5ba3702f
JG
2805
2806 if (!channel || !lost_packets) {
2807 ret = -LTTNG_ERR_INVALID;
2808 goto end;
2809 }
2810
4bd69c5f 2811 chan_ext = (lttng_channel_extended *) channel->attr.extended.ptr;
5ba3702f
JG
2812 if (!chan_ext) {
2813 /*
2814 * This can happen since the lttng_channel structure is
2815 * used for other tasks where this pointer is never set.
2816 */
2817 *lost_packets = 0;
2818 goto end;
2819 }
2820
2821 *lost_packets = chan_ext->lost_packets;
2822end:
2823 return ret;
2824}
2825
cf0bcb51
JG
2826int lttng_channel_get_monitor_timer_interval(struct lttng_channel *chan,
2827 uint64_t *monitor_timer_interval)
2828{
2829 int ret = 0;
2830
2831 if (!chan || !monitor_timer_interval) {
2832 ret = -LTTNG_ERR_INVALID;
2833 goto end;
2834 }
2835
2836 if (!chan->attr.extended.ptr) {
2837 ret = -LTTNG_ERR_INVALID;
2838 goto end;
2839 }
2840
2841 *monitor_timer_interval = ((struct lttng_channel_extended *)
2842 chan->attr.extended.ptr)->monitor_timer_interval;
2843end:
2844 return ret;
2845}
2846
2847int lttng_channel_set_monitor_timer_interval(struct lttng_channel *chan,
2848 uint64_t monitor_timer_interval)
2849{
2850 int ret = 0;
2851
2852 if (!chan || !chan->attr.extended.ptr) {
2853 ret = -LTTNG_ERR_INVALID;
2854 goto end;
2855 }
2856
2857 ((struct lttng_channel_extended *)
2858 chan->attr.extended.ptr)->monitor_timer_interval =
2859 monitor_timer_interval;
2860end:
2861 return ret;
2862}
2863
491d1539
MD
2864int lttng_channel_get_blocking_timeout(struct lttng_channel *chan,
2865 int64_t *blocking_timeout)
2866{
2867 int ret = 0;
2868
2869 if (!chan || !blocking_timeout) {
2870 ret = -LTTNG_ERR_INVALID;
2871 goto end;
2872 }
2873
2874 if (!chan->attr.extended.ptr) {
2875 ret = -LTTNG_ERR_INVALID;
2876 goto end;
2877 }
2878
2879 *blocking_timeout = ((struct lttng_channel_extended *)
2880 chan->attr.extended.ptr)->blocking_timeout;
2881end:
2882 return ret;
2883}
2884
2885int lttng_channel_set_blocking_timeout(struct lttng_channel *chan,
2886 int64_t blocking_timeout)
2887{
2888 int ret = 0;
2889 int64_t msec_timeout;
2890
2891 if (!chan || !chan->attr.extended.ptr) {
2892 ret = -LTTNG_ERR_INVALID;
2893 goto end;
2894 }
2895
2896 if (blocking_timeout < 0 && blocking_timeout != -1) {
2897 ret = -LTTNG_ERR_INVALID;
2898 goto end;
2899 }
2900
2901 /*
2902 * LTTng-ust's use of poll() to implement this timeout mechanism forces
2903 * us to accept a narrower range of values (msecs expressed as a signed
2904 * 32-bit integer).
2905 */
2906 msec_timeout = blocking_timeout / 1000;
2907 if (msec_timeout != (int32_t) msec_timeout) {
2908 ret = -LTTNG_ERR_INVALID;
2909 goto end;
2910 }
2911
2912 ((struct lttng_channel_extended *)
2913 chan->attr.extended.ptr)->blocking_timeout =
2914 blocking_timeout;
2915end:
2916 return ret;
2917}
2918
fac6795d 2919/*
2269e89e 2920 * Check if session daemon is alive.
fac6795d 2921 *
2269e89e 2922 * Return 1 if alive or 0 if not.
1c8d13c8 2923 * On error returns a negative value.
fac6795d 2924 */
947308c4 2925int lttng_session_daemon_alive(void)
fac6795d
DG
2926{
2927 int ret;
2928
2929 ret = set_session_daemon_path();
2930 if (ret < 0) {
9ae110e2 2931 /* Error. */
fac6795d
DG
2932 return ret;
2933 }
2934
9d035200 2935 if (*sessiond_sock_path == '\0') {
2f70b271 2936 /*
9ae110e2
JG
2937 * No socket path set. Weird error which means the constructor
2938 * was not called.
2f70b271 2939 */
a0377dfe 2940 abort();
fac6795d
DG
2941 }
2942
2269e89e 2943 ret = try_connect_sessiond(sessiond_sock_path);
7d8234d9 2944 if (ret < 0) {
9ae110e2 2945 /* Not alive. */
7d8234d9
MD
2946 return 0;
2947 }
7d8234d9 2948
9ae110e2 2949 /* Is alive. */
947308c4 2950 return 1;
fac6795d
DG
2951}
2952
00e2e675 2953/*
a4b92340 2954 * Set URL for a consumer for a session and domain.
00e2e675
DG
2955 *
2956 * Return 0 on success, else a negative value.
2957 */
a4b92340
DG
2958int lttng_set_consumer_url(struct lttng_handle *handle,
2959 const char *control_url, const char *data_url)
00e2e675 2960{
3dd05a85 2961 int ret;
a4b92340 2962 ssize_t size;
00e2e675 2963 struct lttcomm_session_msg lsm;
a4b92340 2964 struct lttng_uri *uris = NULL;
00e2e675 2965
a4b92340 2966 if (handle == NULL || (control_url == NULL && data_url == NULL)) {
e1b624d0
JG
2967 ret = -LTTNG_ERR_INVALID;
2968 goto error;
00e2e675
DG
2969 }
2970
a4b92340
DG
2971 memset(&lsm, 0, sizeof(lsm));
2972
00e2e675
DG
2973 lsm.cmd_type = LTTNG_SET_CONSUMER_URI;
2974
e1b624d0 2975 ret = lttng_strncpy(lsm.session.name, handle->session_name,
00e2e675 2976 sizeof(lsm.session.name));
e1b624d0
JG
2977 if (ret) {
2978 ret = -LTTNG_ERR_INVALID;
2979 goto error;
2980 }
2981
60160d2a 2982 COPY_DOMAIN_PACKED(lsm.domain, handle->domain);
00e2e675 2983
bc894455 2984 size = uri_parse_str_urls(control_url, data_url, &uris);
a4b92340 2985 if (size < 0) {
e1b624d0
JG
2986 ret = -LTTNG_ERR_INVALID;
2987 goto error;
a4b92340
DG
2988 }
2989
2990 lsm.u.uri.size = size;
00e2e675 2991
795a978d 2992 ret = lttng_ctl_ask_sessiond_varlen_no_cmd_header(&lsm, uris,
cac3069d 2993 sizeof(struct lttng_uri) * size, NULL);
3dd05a85
DG
2994
2995 free(uris);
e1b624d0 2996error:
3dd05a85 2997 return ret;
00e2e675
DG
2998}
2999
3000/*
9c6bda17 3001 * [OBSOLETE]
00e2e675 3002 */
4bd69c5f
SM
3003extern "C"
3004LTTNG_EXPORT int lttng_enable_consumer(struct lttng_handle *handle);
00e2e675
DG
3005int lttng_enable_consumer(struct lttng_handle *handle)
3006{
785d2d0d 3007 return -ENOSYS;
00e2e675
DG
3008}
3009
3010/*
9c6bda17 3011 * [OBSOLETE]
00e2e675 3012 */
4bd69c5f
SM
3013extern "C"
3014LTTNG_EXPORT int lttng_disable_consumer(struct lttng_handle *handle);
00e2e675
DG
3015int lttng_disable_consumer(struct lttng_handle *handle)
3016{
785d2d0d 3017 return -ENOSYS;
00e2e675
DG
3018}
3019
07424f16 3020/*
b178f53e 3021 * [OBSOLETE]
07424f16 3022 */
4bd69c5f
SM
3023extern "C"
3024LTTNG_EXPORT int _lttng_create_session_ext(const char *name, const char *url,
2ec40111 3025 const char *datetime);
07424f16
DG
3026int _lttng_create_session_ext(const char *name, const char *url,
3027 const char *datetime)
3028{
b178f53e 3029 return -ENOSYS;
07424f16
DG
3030}
3031
806e2684
DG
3032/*
3033 * For a given session name, this call checks if the data is ready to be read
3034 * or is still being extracted by the consumer(s) hence not ready to be used by
3035 * any readers.
3036 */
6d805429 3037int lttng_data_pending(const char *session_name)
806e2684
DG
3038{
3039 int ret;
3040 struct lttcomm_session_msg lsm;
f6151c55 3041 uint8_t *pending = NULL;
806e2684
DG
3042
3043 if (session_name == NULL) {
3044 return -LTTNG_ERR_INVALID;
3045 }
3046
53efb85a 3047 memset(&lsm, 0, sizeof(lsm));
6d805429 3048 lsm.cmd_type = LTTNG_DATA_PENDING;
806e2684 3049
e1b624d0 3050 ret = lttng_strncpy(lsm.session.name, session_name,
cac3069d 3051 sizeof(lsm.session.name));
e1b624d0
JG
3052 if (ret) {
3053 ret = -LTTNG_ERR_INVALID;
3054 goto end;
3055 }
806e2684 3056
f6151c55
JG
3057 ret = lttng_ctl_ask_sessiond(&lsm, (void **) &pending);
3058 if (ret < 0) {
3059 goto end;
3060 } else if (ret != 1) {
3061 /* Unexpected payload size */
3062 ret = -LTTNG_ERR_INVALID;
3063 goto end;
e848d36d
JG
3064 } else if (!pending) {
3065 /* Internal error. */
3066 ret = -LTTNG_ERR_UNK;
3067 goto end;
806e2684
DG
3068 }
3069
f6151c55
JG
3070 ret = (int) *pending;
3071end:
3072 free(pending);
806e2684
DG
3073 return ret;
3074}
3075
93ec662e
JD
3076/*
3077 * Regenerate the metadata for a session.
3078 * Return 0 on success, a negative error code on error.
3079 */
eded6438 3080int lttng_regenerate_metadata(const char *session_name)
93ec662e
JD
3081{
3082 int ret;
3083 struct lttcomm_session_msg lsm;
3084
3085 if (!session_name) {
3086 ret = -LTTNG_ERR_INVALID;
3087 goto end;
3088 }
3089
3090 memset(&lsm, 0, sizeof(lsm));
eded6438 3091 lsm.cmd_type = LTTNG_REGENERATE_METADATA;
93ec662e 3092
e1b624d0 3093 ret = lttng_strncpy(lsm.session.name, session_name,
93ec662e 3094 sizeof(lsm.session.name));
e1b624d0
JG
3095 if (ret) {
3096 ret = -LTTNG_ERR_INVALID;
3097 goto end;
3098 }
93ec662e
JD
3099
3100 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
3101 if (ret < 0) {
3102 goto end;
3103 }
3104
3105 ret = 0;
3106end:
3107 return ret;
3108}
3109
eded6438
JD
3110/*
3111 * Deprecated, replaced by lttng_regenerate_metadata.
3112 */
3113int lttng_metadata_regenerate(const char *session_name)
3114{
3115 return lttng_regenerate_metadata(session_name);
3116}
3117
c2561365
JD
3118/*
3119 * Regenerate the statedump of a session.
3120 * Return 0 on success, a negative error code on error.
3121 */
3122int lttng_regenerate_statedump(const char *session_name)
3123{
3124 int ret;
3125 struct lttcomm_session_msg lsm;
3126
3127 if (!session_name) {
3128 ret = -LTTNG_ERR_INVALID;
3129 goto end;
3130 }
3131
3132 memset(&lsm, 0, sizeof(lsm));
3133 lsm.cmd_type = LTTNG_REGENERATE_STATEDUMP;
3134
e1b624d0 3135 ret = lttng_strncpy(lsm.session.name, session_name,
c2561365 3136 sizeof(lsm.session.name));
e1b624d0
JG
3137 if (ret) {
3138 ret = -LTTNG_ERR_INVALID;
3139 goto end;
3140 }
c2561365
JD
3141
3142 ret = lttng_ctl_ask_sessiond(&lsm, NULL);
3143 if (ret < 0) {
3144 goto end;
3145 }
3146
3147 ret = 0;
3148end:
3149 return ret;
3150}
3151
a5c2d2a7
JG
3152static
3153int _lttng_register_trigger(struct lttng_trigger *trigger, const char *name,
3154 bool generate_name)
a58c490f
JG
3155{
3156 int ret;
99608320
JR
3157 struct lttcomm_session_msg lsm = {
3158 .cmd_type = LTTNG_REGISTER_TRIGGER,
3159 };
4bd69c5f 3160 lsm.u.trigger.is_trigger_anonymous = !name && !generate_name;
99608320
JR
3161 struct lttcomm_session_msg *message_lsm;
3162 struct lttng_payload message;
3163 struct lttng_payload reply;
242388e4 3164 struct lttng_trigger *reply_trigger = NULL;
9124c630 3165 enum lttng_domain_type domain_type;
64eafdf6
JR
3166 const struct lttng_credentials user_creds = {
3167 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3168 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3169 };
a5c2d2a7
JG
3170 const char *unused_trigger_name = NULL;
3171 enum lttng_trigger_status trigger_status;
99608320
JR
3172
3173 lttng_payload_init(&message);
3174 lttng_payload_init(&reply);
a58c490f
JG
3175
3176 if (!trigger) {
3177 ret = -LTTNG_ERR_INVALID;
3178 goto end;
3179 }
3180
a5c2d2a7
JG
3181 trigger_status = lttng_trigger_get_name(trigger, &unused_trigger_name);
3182 if (trigger_status != LTTNG_TRIGGER_STATUS_UNSET) {
3183 /* Re-using already registered trigger. */
3184 ret = -LTTNG_ERR_INVALID;
3185 goto end;
3186 }
3187
3188 if (name) {
3189 trigger_status = lttng_trigger_set_name(trigger, name);
3190 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3191 ret = -LTTNG_ERR_NOMEM;
3192 goto end;
3193 }
3194 }
3195
64eafdf6
JR
3196 if (!trigger->creds.uid.is_set) {
3197 /* Use the client's credentials as the trigger credentials. */
3198 lttng_trigger_set_credentials(trigger, &user_creds);
3199 } else {
3200 /*
3201 * Validate that either the current trigger credentials and the
3202 * client credentials are identical or that the current user is
3203 * root. The root user can register, unregister triggers for
3204 * himself and other users.
3205 *
3206 * This check is also present on the sessiond side, using the
3207 * credentials passed on the socket. These check are all
3208 * "safety" checks.
3209 */
3210 const struct lttng_credentials *trigger_creds =
3211 lttng_trigger_get_credentials(trigger);
3212
3213 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3214 if (lttng_credentials_get_uid(&user_creds) != 0) {
3215 ret = -LTTNG_ERR_EPERM;
a5c2d2a7 3216 goto end_unset_name;
64eafdf6
JR
3217 }
3218 }
3219 }
3220
a58c490f 3221 if (!lttng_trigger_validate(trigger)) {
eac4828d 3222 ret = -LTTNG_ERR_INVALID_TRIGGER;
a5c2d2a7 3223 goto end_unset_name;
a58c490f
JG
3224 }
3225
9124c630
JR
3226 domain_type = lttng_trigger_get_underlying_domain_type_restriction(
3227 trigger);
3228
3229 lsm.domain.type = domain_type;
3230
97285430
JG
3231 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3232 if (ret) {
3233 ret = -LTTNG_ERR_NOMEM;
a5c2d2a7 3234 goto end_unset_name;
97285430 3235 }
99608320 3236
99608320 3237 ret = lttng_trigger_serialize(trigger, &message);
3647288f 3238 if (ret < 0) {
a58c490f 3239 ret = -LTTNG_ERR_UNK;
a5c2d2a7 3240 goto end_unset_name;
a58c490f
JG
3241 }
3242
b22f4f54
JG
3243 /*
3244 * This is needed to populate the trigger object size for the command
3245 * header.
3246 */
3247 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3248
99608320
JR
3249 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3250
3251 {
3252 struct lttng_payload_view message_view =
3253 lttng_payload_view_from_payload(
3254 &message, 0, -1);
3255
3256 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3257 &message_view);
3258 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3259 if (ret < 0) {
a5c2d2a7 3260 goto end_unset_name;
99608320
JR
3261 }
3262 }
3263
242388e4
JR
3264 {
3265 struct lttng_payload_view reply_view =
3266 lttng_payload_view_from_payload(
3267 &reply, 0, reply.buffer.size);
3268
3269 ret = lttng_trigger_create_from_payload(
3270 &reply_view, &reply_trigger);
3271 if (ret < 0) {
a5c2d2a7
JG
3272 ret = -LTTNG_ERR_INVALID_PROTOCOL;
3273 goto end_unset_name;
242388e4
JR
3274 }
3275 }
3276
a5c2d2a7
JG
3277 if (name || generate_name) {
3278 ret = lttng_trigger_assign_name(trigger, reply_trigger);
3279 if (ret < 0) {
3280 ret = -LTTNG_ERR_NOMEM;
3281 goto end;
3282 }
242388e4
JR
3283 }
3284
99608320 3285 ret = 0;
a5c2d2a7
JG
3286 goto end;
3287
3288end_unset_name:
3289 trigger_status = lttng_trigger_set_name(trigger, NULL);
3290 if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
3291 ret = -LTTNG_ERR_UNK;
3292 }
a58c490f 3293end:
99608320
JR
3294 lttng_payload_reset(&message);
3295 lttng_payload_reset(&reply);
242388e4 3296 lttng_trigger_destroy(reply_trigger);
a58c490f
JG
3297 return ret;
3298}
3299
a5c2d2a7
JG
3300int lttng_register_trigger(struct lttng_trigger *trigger)
3301{
3302 /* Register an anonymous trigger. */
3303 return _lttng_register_trigger(trigger, NULL, false);
3304}
3305
3306enum lttng_error_code lttng_register_trigger_with_name(
3307 struct lttng_trigger *trigger, const char *name)
3308{
3309 const int ret = _lttng_register_trigger(trigger, name, false);
3310
3311 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3312}
3313
3314enum lttng_error_code lttng_register_trigger_with_automatic_name(
3315 struct lttng_trigger *trigger)
3316{
4bd69c5f 3317 const int ret = _lttng_register_trigger(trigger, nullptr, true);
a5c2d2a7
JG
3318
3319 return ret == 0 ? LTTNG_OK : (enum lttng_error_code) -ret;
3320}
3321
b99a0cb3
JG
3322enum lttng_error_code lttng_error_query_execute(
3323 const struct lttng_error_query *query,
3324 const struct lttng_endpoint *endpoint,
3325 struct lttng_error_query_results **results)
3326{
3327 int ret;
3328 enum lttng_error_code ret_code;
3329 struct lttcomm_session_msg lsm = {
3330 .cmd_type = LTTNG_EXECUTE_ERROR_QUERY,
3331 };
3332 struct lttng_payload message;
3333 struct lttng_payload reply;
3334 struct lttcomm_session_msg *message_lsm;
3335
3336 lttng_payload_init(&message);
3337 lttng_payload_init(&reply);
3338
3339 if (!query || !results) {
3340 ret_code = LTTNG_ERR_INVALID;
3341 goto end;
3342 }
3343
3344 if (endpoint != lttng_session_daemon_command_endpoint) {
3345 ret_code = LTTNG_ERR_INVALID_ERROR_QUERY_TARGET;
3346 goto end;
3347 }
3348
3349 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3350 if (ret) {
3351 ret_code = LTTNG_ERR_NOMEM;
3352 goto end;
3353 }
3354
3355 ret = lttng_error_query_serialize(query, &message);
3356 if (ret) {
3357 ret_code = LTTNG_ERR_UNK;
3358 goto end;
3359 }
3360
3361 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3362 message_lsm->u.error_query.length =
3363 (uint32_t) message.buffer.size - sizeof(lsm);
3364
3365 {
3366 struct lttng_payload_view message_view =
3367 lttng_payload_view_from_payload(
3368 &message, 0, -1);
3369
3370 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3371 &message_view);
3372 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3373 if (ret < 0) {
4bd69c5f 3374 ret_code =(lttng_error_code) -ret;
b99a0cb3
JG
3375 goto end;
3376 }
3377 }
3378
3379 {
3380 ssize_t reply_create_ret;
3381 struct lttng_payload_view reply_view =
3382 lttng_payload_view_from_payload(
3383 &reply, 0, reply.buffer.size);
3384
3385 reply_create_ret = lttng_error_query_results_create_from_payload(
3386 &reply_view, results);
3387 if (reply_create_ret < 0) {
3388 ret_code = LTTNG_ERR_INVALID_PROTOCOL;
3389 goto end;
3390 }
3391 }
3392
3393 ret_code = LTTNG_OK;
3394end:
3395 lttng_payload_reset(&message);
3396 lttng_payload_reset(&reply);
3397 return ret_code;
3398}
3399
b61776fb 3400int lttng_unregister_trigger(const struct lttng_trigger *trigger)
a58c490f
JG
3401{
3402 int ret;
3403 struct lttcomm_session_msg lsm;
99608320
JR
3404 struct lttcomm_session_msg *message_lsm;
3405 struct lttng_payload message;
3406 struct lttng_payload reply;
b61776fb 3407 struct lttng_trigger *copy = NULL;
64eafdf6
JR
3408 const struct lttng_credentials user_creds = {
3409 .uid = LTTNG_OPTIONAL_INIT_VALUE(geteuid()),
3410 .gid = LTTNG_OPTIONAL_INIT_UNSET,
3411 };
99608320
JR
3412
3413 lttng_payload_init(&message);
3414 lttng_payload_init(&reply);
a58c490f
JG
3415
3416 if (!trigger) {
3417 ret = -LTTNG_ERR_INVALID;
3418 goto end;
3419 }
3420
b61776fb
SM
3421 copy = lttng_trigger_copy(trigger);
3422 if (!copy) {
3423 ret = -LTTNG_ERR_UNK;
3424 goto end;
3425 }
3426
3427 if (!copy->creds.uid.is_set) {
3428 /* Use the client credentials as the trigger credentials */
3429 lttng_trigger_set_credentials(copy, &user_creds);
64eafdf6
JR
3430 } else {
3431 /*
3432 * Validate that either the current trigger credentials and the
3433 * client credentials are identical or that the current user is
3434 * root. The root user can register, unregister triggers for
3435 * himself and other users.
3436 *
3437 * This check is also present on the sessiond side, using the
3438 * credentials passed on the socket. These check are all
3439 * "safety" checks.
3440 */
3441 const struct lttng_credentials *trigger_creds =
b61776fb 3442 lttng_trigger_get_credentials(copy);
64eafdf6
JR
3443 if (!lttng_credentials_is_equal_uid(trigger_creds, &user_creds)) {
3444 if (lttng_credentials_get_uid(&user_creds) != 0) {
3445 ret = -LTTNG_ERR_EPERM;
3446 goto end;
3447 }
3448 }
3449 }
3450
b61776fb 3451 if (!lttng_trigger_validate(copy)) {
3647288f 3452 ret = -LTTNG_ERR_INVALID_TRIGGER;
a58c490f
JG
3453 goto end;
3454 }
3455
99608320
JR
3456 memset(&lsm, 0, sizeof(lsm));
3457 lsm.cmd_type = LTTNG_UNREGISTER_TRIGGER;
3458
97285430
JG
3459 ret = lttng_dynamic_buffer_append(&message.buffer, &lsm, sizeof(lsm));
3460 if (ret) {
3461 ret = -LTTNG_ERR_NOMEM;
3462 goto end;
3463 }
99608320 3464
b61776fb 3465 ret = lttng_trigger_serialize(copy, &message);
3647288f 3466 if (ret < 0) {
a58c490f
JG
3467 ret = -LTTNG_ERR_UNK;
3468 goto end;
3469 }
3470
b0a79813
FD
3471 /*
3472 * This is needed to populate the trigger object size for the command
3473 * header and number of fds sent.
3474 */
3475 message_lsm = (struct lttcomm_session_msg *) message.buffer.data;
3476
99608320
JR
3477 message_lsm->u.trigger.length = (uint32_t) message.buffer.size - sizeof(lsm);
3478
3479 {
3480 struct lttng_payload_view message_view =
3481 lttng_payload_view_from_payload(
3482 &message, 0, -1);
3483
3484 /*
3485 * Update the message header with the number of fd that will be
3486 * sent.
3487 */
3488 message_lsm->fd_count = lttng_payload_view_get_fd_handle_count(
3489 &message_view);
3490
3491 ret = lttng_ctl_ask_sessiond_payload(&message_view, &reply);
3492 if (ret < 0) {
3493 goto end;
3494 }
3495 }
3496
3497 ret = 0;
a58c490f 3498end:
b61776fb 3499 lttng_trigger_destroy(copy);
99608320
JR
3500 lttng_payload_reset(&message);
3501 lttng_payload_reset(&reply);
a58c490f
JG
3502 return ret;
3503}
3504
fbc9f37d
JR
3505/*
3506 * Ask the session daemon for all registered triggers for the current user.
3507 *
3508 * Allocates and return an lttng_triggers set.
3509 * On error, returns a suitable lttng_error_code.
3510 */
3511enum lttng_error_code lttng_list_triggers(struct lttng_triggers **triggers)
3512{
3513 int ret;
3514 enum lttng_error_code ret_code = LTTNG_OK;
3515 struct lttcomm_session_msg lsm = { .cmd_type = LTTNG_LIST_TRIGGERS };
3516 struct lttng_triggers *local_triggers = NULL;
3517 struct lttng_payload reply;
3518 struct lttng_payload_view lsm_view =
3519 lttng_payload_view_init_from_buffer(
3520 (const char *) &lsm, 0, sizeof(lsm));
3521
3522 lttng_payload_init(&reply);
3523
3524 ret = lttng_ctl_ask_sessiond_payload(&lsm_view, &reply);
3525 if (ret < 0) {
3526 ret_code = (enum lttng_error_code) -ret;
3527 goto end;
3528 }
3529
3530 {
3531 struct lttng_payload_view reply_view =
3532 lttng_payload_view_from_payload(
3533 &reply, 0, reply.buffer.size);
3534
3535 ret = lttng_triggers_create_from_payload(
3536 &reply_view, &local_triggers);
3537 if (ret < 0) {
3538 ret_code = LTTNG_ERR_FATAL;
3539 goto end;
3540 }
3541 }
3542
3543 *triggers = local_triggers;
3544 local_triggers = NULL;
3545end:
3546 lttng_payload_reset(&reply);
3547 lttng_triggers_destroy(local_triggers);
3548 return ret_code;
3549}
3550
fac6795d 3551/*
9ae110e2 3552 * lib constructor.
fac6795d 3553 */
b2b89e8a 3554static void __attribute__((constructor)) init(void)
fac6795d
DG
3555{
3556 /* Set default session group */
bbccc3d2 3557 lttng_set_tracing_group(DEFAULT_TRACING_GROUP);
fac6795d 3558}
49cca668
DG
3559
3560/*
9ae110e2 3561 * lib destructor.
49cca668 3562 */
b2b89e8a 3563static void __attribute__((destructor)) lttng_ctl_exit(void)
49cca668
DG
3564{
3565 free(tracing_group);
3566}
This page took 0.26826 seconds and 4 git commands to generate.