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