Deliverables 3 and 4
[deliverable/lttng-tools.git] / src / bin / lttng-sessiond / main.c
CommitLineData
826d496d
MD
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
0fdd1e2c 3 * Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
26296c48 4 * 2013 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
fac6795d 5 *
d14d33bf
AM
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License, version 2 only,
8 * as published by the Free Software Foundation.
91d76f53 9 *
d14d33bf
AM
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
91d76f53 14 *
d14d33bf
AM
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
fac6795d
DG
18 */
19
6c1c0768 20#define _LGPL_SOURCE
fac6795d
DG
21#include <getopt.h>
22#include <grp.h>
23#include <limits.h>
0bb7724a 24#include <paths.h>
fac6795d
DG
25#include <pthread.h>
26#include <signal.h>
27#include <stdio.h>
28#include <stdlib.h>
29#include <string.h>
331744e3 30#include <inttypes.h>
0fdd1e2c 31#include <sys/mman.h>
b73401da 32#include <sys/mount.h>
1e307fab 33#include <sys/resource.h>
fac6795d
DG
34#include <sys/socket.h>
35#include <sys/stat.h>
36#include <sys/types.h>
0fdd1e2c 37#include <sys/wait.h>
5c827ce0 38#include <urcu/uatomic.h>
fac6795d 39#include <unistd.h>
4f0b90ee 40#include <ctype.h>
fac6795d 41
990570ed 42#include <common/common.h>
d27c42b8 43#include <common/compat/socket.h>
e8fa9fb0 44#include <common/compat/getenv.h>
db758600
DG
45#include <common/defaults.h>
46#include <common/kernel-consumer/kernel-consumer.h>
50c8f484 47#include <common/futex.h>
00e2e675 48#include <common/relayd/relayd.h>
81b86775 49#include <common/utils.h>
3ccdf997 50#include <common/daemonize.h>
f40ef1d5 51#include <common/config/session-config.h>
fac6795d 52
10a8a223 53#include "lttng-sessiond.h"
7972aab2 54#include "buffer-registry.h"
54d01ffb 55#include "channel.h"
2f77fc4b 56#include "cmd.h"
00e2e675 57#include "consumer.h"
099e26bd 58#include "context.h"
54d01ffb 59#include "event.h"
4771f025 60#include "kernel.h"
f1e16794 61#include "kernel-consumer.h"
096102bd 62#include "modprobe.h"
0fdd1e2c 63#include "shm.h"
1e307fab 64#include "ust-ctl.h"
00e2e675 65#include "ust-consumer.h"
8e68d1c8 66#include "utils.h"
4063050c 67#include "fd-limit.h"
8782cc74 68#include "health-sessiond.h"
8ac94142 69#include "testpoint.h"
d0b96690 70#include "ust-thread.h"
022d91ba 71#include "agent-thread.h"
fb198a11 72#include "save.h"
ef367a93 73#include "load-session-thread.h"
daa81f58 74#include "notification-thread.h"
834978fd 75#include "syscall.h"
7c1d2758 76#include "agent.h"
5e97de00 77#include "ht-cleanup.h"
fac6795d 78
ebaeda94
MD
79#define CONSUMERD_FILE "lttng-consumerd"
80
fac6795d 81const char *progname;
6c71277b 82static const char *tracing_group_name = DEFAULT_TRACING_GROUP;
26296c48
JG
83static int tracing_group_name_override;
84static char *opt_pidfile;
5b8719f5 85static int opt_sig_parent;
97e19046 86static int opt_verbose_consumer;
72dd7491 87static int opt_daemon, opt_background;
4fba7219 88static int opt_no_kernel;
ef367a93 89static char *opt_load_session_path;
1d4b027a 90static pid_t ppid; /* Parent PID for --sig-parent option */
0bb7724a 91static pid_t child_ppid; /* Internal parent PID use with daemonize. */
67e40797 92static char *rundir;
c9cb3e7d 93static int lockfile_fd = -1;
3bd1e081 94
0bb7724a
DG
95/* Set to 1 when a SIGUSR1 signal is received. */
96static int recv_child_signal;
97
a23ec3a7
DG
98/*
99 * Consumer daemon specific control data. Every value not initialized here is
100 * set to 0 by the static definition.
101 */
3bd1e081
MD
102static struct consumer_data kconsumer_data = {
103 .type = LTTNG_CONSUMER_KERNEL,
60922cb0
DG
104 .err_unix_sock_path = DEFAULT_KCONSUMERD_ERR_SOCK_PATH,
105 .cmd_unix_sock_path = DEFAULT_KCONSUMERD_CMD_SOCK_PATH,
03550b58
MD
106 .err_sock = -1,
107 .cmd_sock = -1,
daa81f58 108 .channel_monitor_pipe = -1,
173af62f
DG
109 .pid_mutex = PTHREAD_MUTEX_INITIALIZER,
110 .lock = PTHREAD_MUTEX_INITIALIZER,
a23ec3a7
DG
111 .cond = PTHREAD_COND_INITIALIZER,
112 .cond_mutex = PTHREAD_MUTEX_INITIALIZER,
3bd1e081 113};
7753dea8
MD
114static struct consumer_data ustconsumer64_data = {
115 .type = LTTNG_CONSUMER64_UST,
60922cb0
DG
116 .err_unix_sock_path = DEFAULT_USTCONSUMERD64_ERR_SOCK_PATH,
117 .cmd_unix_sock_path = DEFAULT_USTCONSUMERD64_CMD_SOCK_PATH,
03550b58
MD
118 .err_sock = -1,
119 .cmd_sock = -1,
daa81f58 120 .channel_monitor_pipe = -1,
173af62f
DG
121 .pid_mutex = PTHREAD_MUTEX_INITIALIZER,
122 .lock = PTHREAD_MUTEX_INITIALIZER,
a23ec3a7
DG
123 .cond = PTHREAD_COND_INITIALIZER,
124 .cond_mutex = PTHREAD_MUTEX_INITIALIZER,
7753dea8
MD
125};
126static struct consumer_data ustconsumer32_data = {
127 .type = LTTNG_CONSUMER32_UST,
60922cb0
DG
128 .err_unix_sock_path = DEFAULT_USTCONSUMERD32_ERR_SOCK_PATH,
129 .cmd_unix_sock_path = DEFAULT_USTCONSUMERD32_CMD_SOCK_PATH,
03550b58
MD
130 .err_sock = -1,
131 .cmd_sock = -1,
daa81f58 132 .channel_monitor_pipe = -1,
173af62f
DG
133 .pid_mutex = PTHREAD_MUTEX_INITIALIZER,
134 .lock = PTHREAD_MUTEX_INITIALIZER,
a23ec3a7
DG
135 .cond = PTHREAD_COND_INITIALIZER,
136 .cond_mutex = PTHREAD_MUTEX_INITIALIZER,
3bd1e081
MD
137};
138
26296c48
JG
139/* Command line options */
140static const struct option long_options[] = {
0f5ea17c
JG
141 { "client-sock", required_argument, 0, 'c' },
142 { "apps-sock", required_argument, 0, 'a' },
143 { "kconsumerd-cmd-sock", required_argument, 0, '\0' },
144 { "kconsumerd-err-sock", required_argument, 0, '\0' },
145 { "ustconsumerd32-cmd-sock", required_argument, 0, '\0' },
146 { "ustconsumerd32-err-sock", required_argument, 0, '\0' },
147 { "ustconsumerd64-cmd-sock", required_argument, 0, '\0' },
148 { "ustconsumerd64-err-sock", required_argument, 0, '\0' },
149 { "consumerd32-path", required_argument, 0, '\0' },
150 { "consumerd32-libdir", required_argument, 0, '\0' },
151 { "consumerd64-path", required_argument, 0, '\0' },
152 { "consumerd64-libdir", required_argument, 0, '\0' },
153 { "daemonize", no_argument, 0, 'd' },
154 { "background", no_argument, 0, 'b' },
155 { "sig-parent", no_argument, 0, 'S' },
156 { "help", no_argument, 0, 'h' },
157 { "group", required_argument, 0, 'g' },
158 { "version", no_argument, 0, 'V' },
159 { "quiet", no_argument, 0, 'q' },
160 { "verbose", no_argument, 0, 'v' },
161 { "verbose-consumer", no_argument, 0, '\0' },
162 { "no-kernel", no_argument, 0, '\0' },
163 { "pidfile", required_argument, 0, 'p' },
164 { "agent-tcp-port", required_argument, 0, '\0' },
165 { "config", required_argument, 0, 'f' },
166 { "load", required_argument, 0, 'l' },
167 { "kmod-probes", required_argument, 0, '\0' },
168 { "extra-kmod-probes", required_argument, 0, '\0' },
26296c48
JG
169 { NULL, 0, 0, 0 }
170};
171
172/* Command line options to ignore from configuration file */
173static const char *config_ignore_options[] = { "help", "version", "config" };
174
26c9d55e 175/* Shared between threads */
099e26bd 176static int dispatch_thread_exit;
fac6795d 177
54d01ffb
DG
178/* Global application Unix socket path */
179static char apps_unix_sock_path[PATH_MAX];
180/* Global client Unix socket path */
181static char client_unix_sock_path[PATH_MAX];
54d01ffb
DG
182/* global wait shm path for UST */
183static char wait_shm_path[PATH_MAX];
44a5e5eb
DG
184/* Global health check unix path */
185static char health_unix_sock_path[PATH_MAX];
fac6795d 186
1d4b027a 187/* Sockets and FDs */
a4b35e07
MD
188static int client_sock = -1;
189static int apps_sock = -1;
2f77fc4b 190int kernel_tracer_fd = -1;
76d7553f 191static int kernel_poll_pipe[2] = { -1, -1 };
1d4b027a 192
273ea72c
DG
193/*
194 * Quit pipe for all threads. This permits a single cancellation point
195 * for all threads when receiving an event on the pipe.
196 */
76d7553f 197static int thread_quit_pipe[2] = { -1, -1 };
273ea72c 198
099e26bd
DG
199/*
200 * This pipe is used to inform the thread managing application communication
201 * that a command is queued and ready to be processed.
202 */
76d7553f 203static int apps_cmd_pipe[2] = { -1, -1 };
099e26bd 204
d0b96690
DG
205int apps_cmd_notify_pipe[2] = { -1, -1 };
206
1d4b027a 207/* Pthread, Mutexes and Semaphores */
1d4b027a 208static pthread_t apps_thread;
d0b96690 209static pthread_t apps_notify_thread;
099e26bd 210static pthread_t reg_apps_thread;
1d4b027a 211static pthread_t client_thread;
7a485870 212static pthread_t kernel_thread;
099e26bd 213static pthread_t dispatch_thread;
44a5e5eb 214static pthread_t health_thread;
0b2dc8df 215static pthread_t ht_cleanup_thread;
022d91ba 216static pthread_t agent_reg_thread;
ef367a93 217static pthread_t load_session_thread;
daa81f58 218static pthread_t notification_thread;
5eb91c98 219
099e26bd
DG
220/*
221 * UST registration command queue. This queue is tied with a futex and uses a N
222 * wakers / 1 waiter implemented and detailed in futex.c/.h
223 *
b22c5da8
DG
224 * The thread_registration_apps and thread_dispatch_ust_registration uses this
225 * queue along with the wait/wake scheme. The thread_manage_apps receives down
226 * the line new application socket and monitors it for any I/O error or clean
227 * close that triggers an unregistration of the application.
099e26bd
DG
228 */
229static struct ust_cmd_queue ust_cmd_queue;
230
b5541356
DG
231/*
232 * Pointer initialized before thread creation.
233 *
234 * This points to the tracing session list containing the session count and a
235 * mutex lock. The lock MUST be taken if you iterate over the list. The lock
236 * MUST NOT be taken if you call a public function in session.c.
04ea676f 237 *
d063d709 238 * The lock is nested inside the structure: session_list_ptr->lock. Please use
54d01ffb 239 * session_lock_list and session_unlock_list for lock acquisition.
b5541356
DG
240 */
241static struct ltt_session_list *session_list_ptr;
242
7753dea8
MD
243int ust_consumerd64_fd = -1;
244int ust_consumerd32_fd = -1;
245
fb6f1fa2
YB
246static const char *consumerd32_bin = CONFIG_CONSUMERD32_BIN;
247static const char *consumerd64_bin = CONFIG_CONSUMERD64_BIN;
248static const char *consumerd32_libdir = CONFIG_CONSUMERD32_LIBDIR;
249static const char *consumerd64_libdir = CONFIG_CONSUMERD64_LIBDIR;
26296c48
JG
250static int consumerd32_bin_override;
251static int consumerd64_bin_override;
252static int consumerd32_libdir_override;
253static int consumerd64_libdir_override;
fb09408a 254
2f77fc4b
DG
255static const char *module_proc_lttng = "/proc/lttng";
256
5c827ce0
DG
257/*
258 * Consumer daemon state which is changed when spawning it, killing it or in
259 * case of a fatal error.
260 */
261enum consumerd_state {
262 CONSUMER_STARTED = 1,
263 CONSUMER_STOPPED = 2,
264 CONSUMER_ERROR = 3,
265};
266
267/*
268 * This consumer daemon state is used to validate if a client command will be
269 * able to reach the consumer. If not, the client is informed. For instance,
270 * doing a "lttng start" when the consumer state is set to ERROR will return an
271 * error to the client.
272 *
273 * The following example shows a possible race condition of this scheme:
274 *
275 * consumer thread error happens
276 * client cmd arrives
277 * client cmd checks state -> still OK
278 * consumer thread exit, sets error
279 * client cmd try to talk to consumer
280 * ...
281 *
282 * However, since the consumer is a different daemon, we have no way of making
283 * sure the command will reach it safely even with this state flag. This is why
284 * we consider that up to the state validation during command processing, the
285 * command is safe. After that, we can not guarantee the correctness of the
286 * client request vis-a-vis the consumer.
287 */
288static enum consumerd_state ust_consumerd_state;
289static enum consumerd_state kernel_consumerd_state;
290
ae9e45b3
DG
291/*
292 * Socket timeout for receiving and sending in seconds.
293 */
294static int app_socket_timeout;
295
12744796
DG
296/* Set in main() with the current page size. */
297long page_size;
298
8782cc74
MD
299/* Application health monitoring */
300struct health_app *health_sessiond;
301
022d91ba
DG
302/* Agent TCP port for registration. Used by the agent thread. */
303unsigned int agent_tcp_port = DEFAULT_AGENT_TCP_PORT;
4d076222 304
f43f95a9
DG
305/* Am I root or not. */
306int is_root; /* Set to 1 if the daemon is running as root */
307
26296c48
JG
308const char * const config_section_name = "sessiond";
309
ef367a93
JG
310/* Load session thread information to operate. */
311struct load_session_thread_data *load_info;
312
daa81f58
JG
313/* Notification thread handle. */
314struct notification_thread_handle *notification_thread_handle;
315
7c1d2758
JG
316/* Global hash tables */
317struct lttng_ht *agent_apps_ht_by_sock = NULL;
318
97bc1426 319/*
daa81f58
JG
320 * Whether sessiond is ready for commands/notification channel/health check
321 * requests.
97bc1426 322 * NR_LTTNG_SESSIOND_READY must match the number of calls to
ef367a93 323 * sessiond_notify_ready().
97bc1426 324 */
daa81f58 325#define NR_LTTNG_SESSIOND_READY 4
97bc1426
MD
326int lttng_sessiond_ready = NR_LTTNG_SESSIOND_READY;
327
5e97de00
JG
328int sessiond_check_thread_quit_pipe(int fd, uint32_t events)
329{
330 return (fd == thread_quit_pipe[0] && (events & LPOLLIN)) ? 1 : 0;
331}
332
97bc1426 333/* Notify parents that we are ready for cmd and health check */
ef367a93
JG
334LTTNG_HIDDEN
335void sessiond_notify_ready(void)
97bc1426
MD
336{
337 if (uatomic_sub_return(&lttng_sessiond_ready, 1) == 0) {
338 /*
339 * Notify parent pid that we are ready to accept command
340 * for client side. This ppid is the one from the
341 * external process that spawned us.
342 */
343 if (opt_sig_parent) {
344 kill(ppid, SIGUSR1);
345 }
346
347 /*
348 * Notify the parent of the fork() process that we are
349 * ready.
350 */
72dd7491 351 if (opt_daemon || opt_background) {
97bc1426
MD
352 kill(child_ppid, SIGUSR1);
353 }
354 }
355}
356
fb09408a 357static
7753dea8 358void setup_consumerd_path(void)
fb09408a 359{
fc7a59ce 360 const char *bin, *libdir;
fb09408a 361
7753dea8
MD
362 /*
363 * Allow INSTALL_BIN_PATH to be used as a target path for the
ebaeda94
MD
364 * native architecture size consumer if CONFIG_CONSUMER*_PATH
365 * has not been defined.
7753dea8 366 */
ebaeda94 367#if (CAA_BITS_PER_LONG == 32)
fc7a59ce
AM
368 if (!consumerd32_bin[0]) {
369 consumerd32_bin = INSTALL_BIN_PATH "/" CONSUMERD_FILE;
ebaeda94
MD
370 }
371 if (!consumerd32_libdir[0]) {
372 consumerd32_libdir = INSTALL_LIB_PATH;
373 }
374#elif (CAA_BITS_PER_LONG == 64)
fc7a59ce
AM
375 if (!consumerd64_bin[0]) {
376 consumerd64_bin = INSTALL_BIN_PATH "/" CONSUMERD_FILE;
7753dea8 377 }
ebaeda94
MD
378 if (!consumerd64_libdir[0]) {
379 consumerd64_libdir = INSTALL_LIB_PATH;
7753dea8
MD
380 }
381#else
382#error "Unknown bitness"
383#endif
384
fb09408a
MD
385 /*
386 * runtime env. var. overrides the build default.
387 */
e8fa9fb0 388 bin = lttng_secure_getenv("LTTNG_CONSUMERD32_BIN");
fc7a59ce
AM
389 if (bin) {
390 consumerd32_bin = bin;
7753dea8 391 }
e8fa9fb0 392 bin = lttng_secure_getenv("LTTNG_CONSUMERD64_BIN");
fc7a59ce
AM
393 if (bin) {
394 consumerd64_bin = bin;
ebaeda94 395 }
e8fa9fb0 396 libdir = lttng_secure_getenv("LTTNG_CONSUMERD32_LIBDIR");
ebaeda94
MD
397 if (libdir) {
398 consumerd32_libdir = libdir;
399 }
e8fa9fb0 400 libdir = lttng_secure_getenv("LTTNG_CONSUMERD64_LIBDIR");
ebaeda94
MD
401 if (libdir) {
402 consumerd64_libdir = libdir;
fb09408a
MD
403 }
404}
405
4a15001e
MD
406static
407int __sessiond_set_thread_pollset(struct lttng_poll_event *events, size_t size,
408 int *a_pipe)
5eb91c98
DG
409{
410 int ret;
411
d0b96690 412 assert(events);
5eb91c98
DG
413
414 ret = lttng_poll_create(events, size, LTTNG_CLOEXEC);
415 if (ret < 0) {
416 goto error;
417 }
418
419 /* Add quit pipe */
4a15001e 420 ret = lttng_poll_add(events, a_pipe[0], LPOLLIN | LPOLLERR);
5eb91c98
DG
421 if (ret < 0) {
422 goto error;
423 }
424
425 return 0;
426
427error:
428 return ret;
429}
430
4a15001e
MD
431/*
432 * Create a poll set with O_CLOEXEC and add the thread quit pipe to the set.
433 */
434int sessiond_set_thread_pollset(struct lttng_poll_event *events, size_t size)
435{
436 return __sessiond_set_thread_pollset(events, size, thread_quit_pipe);
437}
438
273ea72c 439/*
5eb91c98 440 * Init thread quit pipe.
273ea72c
DG
441 *
442 * Return -1 on error or 0 if all pipes are created.
443 */
4a15001e 444static int __init_thread_quit_pipe(int *a_pipe)
273ea72c 445{
730389d9 446 int ret, i;
273ea72c 447
4a15001e 448 ret = pipe(a_pipe);
273ea72c 449 if (ret < 0) {
730389d9 450 PERROR("thread quit pipe");
273ea72c
DG
451 goto error;
452 }
453
730389d9 454 for (i = 0; i < 2; i++) {
4a15001e 455 ret = fcntl(a_pipe[i], F_SETFD, FD_CLOEXEC);
730389d9
DG
456 if (ret < 0) {
457 PERROR("fcntl");
458 goto error;
459 }
460 }
461
273ea72c
DG
462error:
463 return ret;
464}
465
4a15001e
MD
466static int init_thread_quit_pipe(void)
467{
468 return __init_thread_quit_pipe(thread_quit_pipe);
469}
470
099e26bd
DG
471/*
472 * Stop all threads by closing the thread quit pipe.
473 */
cf3af59e
MD
474static void stop_threads(void)
475{
5eb91c98
DG
476 int ret;
477
cf3af59e
MD
478 /* Stopping all threads */
479 DBG("Terminating all threads");
54d01ffb 480 ret = notify_thread_pipe(thread_quit_pipe[1]);
5eb91c98
DG
481 if (ret < 0) {
482 ERR("write error on thread quit pipe");
483 }
484
099e26bd 485 /* Dispatch thread */
26c9d55e 486 CMM_STORE_SHARED(dispatch_thread_exit, 1);
099e26bd 487 futex_nto1_wake(&ust_cmd_queue.futex);
cf3af59e
MD
488}
489
e975f9f8
DG
490/*
491 * Close every consumer sockets.
492 */
493static void close_consumer_sockets(void)
494{
495 int ret;
496
497 if (kconsumer_data.err_sock >= 0) {
498 ret = close(kconsumer_data.err_sock);
499 if (ret < 0) {
500 PERROR("kernel consumer err_sock close");
501 }
502 }
503 if (ustconsumer32_data.err_sock >= 0) {
504 ret = close(ustconsumer32_data.err_sock);
505 if (ret < 0) {
a76cbd9f 506 PERROR("UST consumerd32 err_sock close");
e975f9f8
DG
507 }
508 }
509 if (ustconsumer64_data.err_sock >= 0) {
510 ret = close(ustconsumer64_data.err_sock);
511 if (ret < 0) {
a76cbd9f 512 PERROR("UST consumerd64 err_sock close");
e975f9f8
DG
513 }
514 }
515 if (kconsumer_data.cmd_sock >= 0) {
516 ret = close(kconsumer_data.cmd_sock);
517 if (ret < 0) {
518 PERROR("kernel consumer cmd_sock close");
519 }
520 }
521 if (ustconsumer32_data.cmd_sock >= 0) {
522 ret = close(ustconsumer32_data.cmd_sock);
523 if (ret < 0) {
a76cbd9f 524 PERROR("UST consumerd32 cmd_sock close");
e975f9f8
DG
525 }
526 }
527 if (ustconsumer64_data.cmd_sock >= 0) {
528 ret = close(ustconsumer64_data.cmd_sock);
529 if (ret < 0) {
a76cbd9f 530 PERROR("UST consumerd64 cmd_sock close");
e975f9f8
DG
531 }
532 }
daa81f58
JG
533 if (kconsumer_data.channel_monitor_pipe >= 0) {
534 ret = close(kconsumer_data.channel_monitor_pipe);
535 if (ret < 0) {
536 PERROR("kernel consumer channel monitor pipe close");
537 }
538 }
539 if (ustconsumer32_data.channel_monitor_pipe >= 0) {
540 ret = close(ustconsumer32_data.channel_monitor_pipe);
541 if (ret < 0) {
542 PERROR("UST consumerd32 channel monitor pipe close");
543 }
544 }
545 if (ustconsumer64_data.channel_monitor_pipe >= 0) {
546 ret = close(ustconsumer64_data.channel_monitor_pipe);
547 if (ret < 0) {
548 PERROR("UST consumerd64 channel monitor pipe close");
549 }
550 }
e975f9f8
DG
551}
552
c9cb3e7d
JG
553/*
554 * Generate the full lock file path using the rundir.
555 *
556 * Return the snprintf() return value thus a negative value is an error.
557 */
558static int generate_lock_file_path(char *path, size_t len)
559{
560 int ret;
561
562 assert(path);
563 assert(rundir);
564
565 /* Build lockfile path from rundir. */
566 ret = snprintf(path, len, "%s/" DEFAULT_LTTNG_SESSIOND_LOCKFILE, rundir);
567 if (ret < 0) {
568 PERROR("snprintf lockfile path");
569 }
570
571 return ret;
572}
573
4e4714cb
JR
574/*
575 * Wait on consumer process termination.
576 *
577 * Need to be called with the consumer data lock held or from a context
578 * ensuring no concurrent access to data (e.g: cleanup).
579 */
580static void wait_consumer(struct consumer_data *consumer_data)
581{
582 pid_t ret;
583 int status;
584
585 if (consumer_data->pid <= 0) {
586 return;
587 }
588
589 DBG("Waiting for complete teardown of consumerd (PID: %d)",
590 consumer_data->pid);
591 ret = waitpid(consumer_data->pid, &status, 0);
592 if (ret == -1) {
593 PERROR("consumerd waitpid pid: %d", consumer_data->pid)
594 }
595 if (!WIFEXITED(status)) {
596 ERR("consumerd termination with error: %d",
597 WEXITSTATUS(ret));
598 }
599 consumer_data->pid = 0;
600}
601
fac6795d 602/*
4a15001e 603 * Cleanup the session daemon's data structures.
fac6795d 604 */
4a15001e 605static void sessiond_cleanup(void)
fac6795d 606{
ef599319 607 int ret;
af9737e9 608 struct ltt_session *sess, *stmp;
8c6c56c2 609 char path[PATH_MAX];
fac6795d 610
4a15001e 611 DBG("Cleanup sessiond");
e07ae692 612
4e449f3f
MD
613 /*
614 * Close the thread quit pipe. It has already done its job,
615 * since we are now called.
616 */
2f77fc4b
DG
617 utils_close_pipe(thread_quit_pipe);
618
35f90c40
DG
619 /*
620 * If opt_pidfile is undefined, the default file will be wiped when
621 * removing the rundir.
622 */
623 if (opt_pidfile) {
624 ret = remove(opt_pidfile);
625 if (ret < 0) {
626 PERROR("remove pidfile %s", opt_pidfile);
627 }
628 }
629
8c6c56c2
MD
630 DBG("Removing sessiond and consumerd content of directory %s", rundir);
631
632 /* sessiond */
633 snprintf(path, PATH_MAX,
634 "%s/%s",
635 rundir, DEFAULT_LTTNG_SESSIOND_PIDFILE);
636 DBG("Removing %s", path);
637 (void) unlink(path);
638
cd9290dd 639 snprintf(path, PATH_MAX, "%s/%s", rundir,
022d91ba 640 DEFAULT_LTTNG_SESSIOND_AGENTPORT_FILE);
cd9290dd
DG
641 DBG("Removing %s", path);
642 (void) unlink(path);
643
8c6c56c2
MD
644 /* kconsumerd */
645 snprintf(path, PATH_MAX,
646 DEFAULT_KCONSUMERD_ERR_SOCK_PATH,
647 rundir);
648 DBG("Removing %s", path);
649 (void) unlink(path);
650
651 snprintf(path, PATH_MAX,
652 DEFAULT_KCONSUMERD_PATH,
653 rundir);
654 DBG("Removing directory %s", path);
655 (void) rmdir(path);
656
657 /* ust consumerd 32 */
658 snprintf(path, PATH_MAX,
659 DEFAULT_USTCONSUMERD32_ERR_SOCK_PATH,
660 rundir);
661 DBG("Removing %s", path);
662 (void) unlink(path);
663
664 snprintf(path, PATH_MAX,
665 DEFAULT_USTCONSUMERD32_PATH,
666 rundir);
667 DBG("Removing directory %s", path);
668 (void) rmdir(path);
669
670 /* ust consumerd 64 */
671 snprintf(path, PATH_MAX,
672 DEFAULT_USTCONSUMERD64_ERR_SOCK_PATH,
673 rundir);
674 DBG("Removing %s", path);
675 (void) unlink(path);
676
677 snprintf(path, PATH_MAX,
678 DEFAULT_USTCONSUMERD64_PATH,
679 rundir);
680 DBG("Removing directory %s", path);
681 (void) rmdir(path);
5461b305 682
99bab54f 683 DBG("Cleaning up all sessions");
fac6795d 684
b5541356 685 /* Destroy session list mutex */
273ea72c
DG
686 if (session_list_ptr != NULL) {
687 pthread_mutex_destroy(&session_list_ptr->lock);
688
689 /* Cleanup ALL session */
54d01ffb
DG
690 cds_list_for_each_entry_safe(sess, stmp,
691 &session_list_ptr->head, list) {
2f77fc4b 692 cmd_destroy_session(sess, kernel_poll_pipe[1]);
273ea72c
DG
693 }
694 }
695
4e4714cb
JR
696 wait_consumer(&kconsumer_data);
697 wait_consumer(&ustconsumer64_data);
698 wait_consumer(&ustconsumer32_data);
699
6a4e4039
JG
700 DBG("Cleaning up all agent apps");
701 agent_app_ht_clean();
702
099e26bd 703 DBG("Closing all UST sockets");
56fff090 704 ust_app_clean_list();
7972aab2 705 buffer_reg_destroy_registries();
099e26bd 706
4fba7219
DG
707 if (is_root && !opt_no_kernel) {
708 DBG2("Closing kernel fd");
a4b35e07 709 if (kernel_tracer_fd >= 0) {
76d7553f
MD
710 ret = close(kernel_tracer_fd);
711 if (ret) {
712 PERROR("close");
713 }
a4b35e07 714 }
2f50c8a3 715 DBG("Unloading kernel modules");
096102bd 716 modprobe_remove_lttng_all();
834978fd 717 free(syscall_table);
2f50c8a3 718 }
2f77fc4b 719
e975f9f8
DG
720 close_consumer_sockets();
721
ef367a93
JG
722 if (load_info) {
723 load_session_destroy_data(load_info);
724 free(load_info);
725 }
726
daa81f58
JG
727 if (notification_thread_handle) {
728 notification_thread_handle_destroy(notification_thread_handle);
729 }
730
c9cb3e7d
JG
731 /*
732 * Cleanup lock file by deleting it and finaly closing it which will
733 * release the file system lock.
734 */
735 if (lockfile_fd >= 0) {
736 char lockfile_path[PATH_MAX];
737
4a15001e
MD
738 ret = generate_lock_file_path(lockfile_path,
739 sizeof(lockfile_path));
c9cb3e7d
JG
740 if (ret > 0) {
741 ret = remove(lockfile_path);
742 if (ret < 0) {
743 PERROR("remove lock file");
744 }
745 ret = close(lockfile_fd);
746 if (ret < 0) {
747 PERROR("close lock file");
748 }
749 }
750 }
751
752 /*
753 * We do NOT rmdir rundir because there are other processes
754 * using it, for instance lttng-relayd, which can start in
755 * parallel with this teardown.
756 */
757
758 free(rundir);
4a15001e
MD
759}
760
761/*
762 * Cleanup the daemon's option data structures.
763 */
764static void sessiond_cleanup_options(void)
765{
766 DBG("Cleaning up options");
767
768 /*
769 * If the override option is set, the pointer points to a *non* const
770 * thus freeing it even though the variable type is set to const.
771 */
772 if (tracing_group_name_override) {
773 free((void *) tracing_group_name);
774 }
775 if (consumerd32_bin_override) {
776 free((void *) consumerd32_bin);
777 }
778 if (consumerd64_bin_override) {
779 free((void *) consumerd64_bin);
780 }
781 if (consumerd32_libdir_override) {
782 free((void *) consumerd32_libdir);
783 }
784 if (consumerd64_libdir_override) {
785 free((void *) consumerd64_libdir);
786 }
787
788 free(opt_pidfile);
789 free(opt_load_session_path);
790 free(kmod_probes_list);
791 free(kmod_extra_probes_list);
c9cb3e7d 792
7567352f 793 run_as_destroy_worker();
fac6795d
DG
794}
795
e065084a 796/*
d063d709 797 * Send data on a unix socket using the liblttsessiondcomm API.
e065084a 798 *
d063d709 799 * Return lttcomm error code.
e065084a
DG
800 */
801static int send_unix_sock(int sock, void *buf, size_t len)
802{
803 /* Check valid length */
c617c0c6 804 if (len == 0) {
e065084a
DG
805 return -1;
806 }
807
808 return lttcomm_send_unix_sock(sock, buf, len);
809}
810
5461b305 811/*
d063d709 812 * Free memory of a command context structure.
5461b305 813 */
a2fb29a5 814static void clean_command_ctx(struct command_ctx **cmd_ctx)
5461b305 815{
a2fb29a5
DG
816 DBG("Clean command context structure");
817 if (*cmd_ctx) {
818 if ((*cmd_ctx)->llm) {
819 free((*cmd_ctx)->llm);
5461b305 820 }
a2fb29a5
DG
821 if ((*cmd_ctx)->lsm) {
822 free((*cmd_ctx)->lsm);
5461b305 823 }
a2fb29a5
DG
824 free(*cmd_ctx);
825 *cmd_ctx = NULL;
5461b305
DG
826 }
827}
828
fac6795d 829/*
0fdd1e2c 830 * Notify UST applications using the shm mmap futex.
fac6795d 831 */
0fdd1e2c 832static int notify_ust_apps(int active)
fac6795d 833{
0fdd1e2c 834 char *wait_shm_mmap;
fac6795d 835
0fdd1e2c 836 DBG("Notifying applications of session daemon state: %d", active);
e07ae692 837
0fdd1e2c
DG
838 /* See shm.c for this call implying mmap, shm and futex calls */
839 wait_shm_mmap = shm_ust_get_mmap(wait_shm_path, is_root);
840 if (wait_shm_mmap == NULL) {
fac6795d
DG
841 goto error;
842 }
843
0fdd1e2c
DG
844 /* Wake waiting process */
845 futex_wait_update((int32_t *) wait_shm_mmap, active);
846
847 /* Apps notified successfully */
848 return 0;
fac6795d
DG
849
850error:
0fdd1e2c 851 return -1;
fac6795d
DG
852}
853
e065084a 854/*
d063d709
DG
855 * Setup the outgoing data buffer for the response (llm) by allocating the
856 * right amount of memory and copying the original information from the lsm
857 * structure.
ca95a216 858 *
6e10c9b9 859 * Return 0 on success, negative value on error.
ca95a216 860 */
6e10c9b9
PP
861static int setup_lttng_msg(struct command_ctx *cmd_ctx,
862 const void *payload_buf, size_t payload_len,
863 const void *cmd_header_buf, size_t cmd_header_len)
ca95a216 864{
6e10c9b9
PP
865 int ret = 0;
866 const size_t header_len = sizeof(struct lttcomm_lttng_msg);
867 const size_t cmd_header_offset = header_len;
868 const size_t payload_offset = cmd_header_offset + cmd_header_len;
869 const size_t total_msg_size = header_len + cmd_header_len + payload_len;
ca95a216 870
6e10c9b9 871 cmd_ctx->llm = zmalloc(total_msg_size);
5461b305 872
5461b305 873 if (cmd_ctx->llm == NULL) {
76d7553f 874 PERROR("zmalloc");
5461b305 875 ret = -ENOMEM;
6e10c9b9 876 goto end;
ca95a216
DG
877 }
878
5461b305
DG
879 /* Copy common data */
880 cmd_ctx->llm->cmd_type = cmd_ctx->lsm->cmd_type;
9f19cc17 881 cmd_ctx->llm->pid = cmd_ctx->lsm->domain.attr.pid;
6e10c9b9
PP
882 cmd_ctx->llm->cmd_header_size = cmd_header_len;
883 cmd_ctx->llm->data_size = payload_len;
884 cmd_ctx->lttng_msg_size = total_msg_size;
5461b305 885
6e10c9b9 886 /* Copy command header */
b4e3ceb9
PP
887 if (cmd_header_len) {
888 memcpy(((uint8_t *) cmd_ctx->llm) + cmd_header_offset, cmd_header_buf,
889 cmd_header_len);
890 }
5461b305 891
6e10c9b9 892 /* Copy payload */
b4e3ceb9
PP
893 if (payload_len) {
894 memcpy(((uint8_t *) cmd_ctx->llm) + payload_offset, payload_buf,
895 payload_len);
896 }
ca95a216 897
6e10c9b9 898end:
ca95a216
DG
899 return ret;
900}
901
6e10c9b9
PP
902/*
903 * Version of setup_lttng_msg() without command header.
904 */
905static int setup_lttng_msg_no_cmd_header(struct command_ctx *cmd_ctx,
906 void *payload_buf, size_t payload_len)
907{
908 return setup_lttng_msg(cmd_ctx, payload_buf, payload_len, NULL, 0);
909}
7a485870 910/*
5eb91c98 911 * Update the kernel poll set of all channel fd available over all tracing
d063d709 912 * session. Add the wakeup pipe at the end of the set.
7a485870 913 */
5eb91c98 914static int update_kernel_poll(struct lttng_poll_event *events)
7a485870 915{
5eb91c98 916 int ret;
7a485870
DG
917 struct ltt_session *session;
918 struct ltt_kernel_channel *channel;
919
5eb91c98 920 DBG("Updating kernel poll set");
7a485870 921
54d01ffb 922 session_lock_list();
b5541356 923 cds_list_for_each_entry(session, &session_list_ptr->head, list) {
54d01ffb 924 session_lock(session);
7a485870 925 if (session->kernel_session == NULL) {
54d01ffb 926 session_unlock(session);
7a485870
DG
927 continue;
928 }
7a485870 929
54d01ffb
DG
930 cds_list_for_each_entry(channel,
931 &session->kernel_session->channel_list.head, list) {
5eb91c98
DG
932 /* Add channel fd to the kernel poll set */
933 ret = lttng_poll_add(events, channel->fd, LPOLLIN | LPOLLRDNORM);
934 if (ret < 0) {
54d01ffb 935 session_unlock(session);
5eb91c98
DG
936 goto error;
937 }
938 DBG("Channel fd %d added to kernel set", channel->fd);
7a485870 939 }
54d01ffb 940 session_unlock(session);
7a485870 941 }
54d01ffb 942 session_unlock_list();
7a485870 943
5eb91c98 944 return 0;
7a485870
DG
945
946error:
54d01ffb 947 session_unlock_list();
7a485870
DG
948 return -1;
949}
950
951/*
54d01ffb 952 * Find the channel fd from 'fd' over all tracing session. When found, check
d063d709 953 * for new channel stream and send those stream fds to the kernel consumer.
7a485870 954 *
d063d709 955 * Useful for CPU hotplug feature.
7a485870 956 */
2bdd86d4 957static int update_kernel_stream(struct consumer_data *consumer_data, int fd)
7a485870
DG
958{
959 int ret = 0;
960 struct ltt_session *session;
173af62f 961 struct ltt_kernel_session *ksess;
7a485870
DG
962 struct ltt_kernel_channel *channel;
963
964 DBG("Updating kernel streams for channel fd %d", fd);
965
54d01ffb 966 session_lock_list();
b5541356 967 cds_list_for_each_entry(session, &session_list_ptr->head, list) {
54d01ffb 968 session_lock(session);
7a485870 969 if (session->kernel_session == NULL) {
54d01ffb 970 session_unlock(session);
7a485870
DG
971 continue;
972 }
173af62f 973 ksess = session->kernel_session;
d9800920 974
4a15001e
MD
975 cds_list_for_each_entry(channel,
976 &ksess->channel_list.head, list) {
977 struct lttng_ht_iter iter;
978 struct consumer_socket *socket;
d9800920 979
4a15001e
MD
980 if (channel->fd != fd) {
981 continue;
982 }
983 DBG("Channel found, updating kernel streams");
984 ret = kernel_open_channel_stream(channel);
985 if (ret < 0) {
986 goto error;
987 }
988 /* Update the stream global counter */
989 ksess->stream_count_global += ret;
990
991 /*
992 * Have we already sent fds to the consumer? If yes, it
993 * means that tracing is started so it is safe to send
994 * our updated stream fds.
995 */
996 if (ksess->consumer_fds_sent != 1
997 || ksess->consumer == NULL) {
998 ret = -1;
999 goto error;
1000 }
1001
1002 rcu_read_lock();
1003 cds_lfht_for_each_entry(ksess->consumer->socks->ht,
1004 &iter.iter, socket, node.node) {
1005 pthread_mutex_lock(socket->lock);
1006 ret = kernel_consumer_send_channel_stream(socket,
1007 channel, ksess,
1008 session->output_traces ? 1 : 0);
1009 pthread_mutex_unlock(socket->lock);
1010 if (ret < 0) {
e7fe706f 1011 rcu_read_unlock();
4a15001e 1012 goto error;
7a485870 1013 }
7a485870 1014 }
4a15001e 1015 rcu_read_unlock();
7a485870 1016 }
54d01ffb 1017 session_unlock(session);
7a485870 1018 }
54d01ffb 1019 session_unlock_list();
b3c750d2 1020 return ret;
7a485870 1021
b3c750d2 1022error:
54d01ffb
DG
1023 session_unlock(session);
1024 session_unlock_list();
7a485870
DG
1025 return ret;
1026}
1027
487cf67c 1028/*
ffe60014
DG
1029 * For each tracing session, update newly registered apps. The session list
1030 * lock MUST be acquired before calling this.
487cf67c
DG
1031 */
1032static void update_ust_app(int app_sock)
1033{
1034 struct ltt_session *sess, *stmp;
1035
fdadac08
DG
1036 /* Consumer is in an ERROR state. Stop any application update. */
1037 if (uatomic_read(&ust_consumerd_state) == CONSUMER_ERROR) {
1038 /* Stop the update process since the consumer is dead. */
1039 return;
1040 }
1041
487cf67c
DG
1042 /* For all tracing session(s) */
1043 cds_list_for_each_entry_safe(sess, stmp, &session_list_ptr->head, list) {
a9ad0c8f
MD
1044 struct ust_app *app;
1045
4ee14516 1046 session_lock(sess);
a9ad0c8f
MD
1047 if (!sess->ust_session) {
1048 goto unlock_session;
1049 }
1050
1051 rcu_read_lock();
1052 assert(app_sock >= 0);
1053 app = ust_app_find_by_sock(app_sock);
1054 if (app == NULL) {
1055 /*
1056 * Application can be unregistered before so
1057 * this is possible hence simply stopping the
1058 * update.
1059 */
1060 DBG3("UST app update failed to find app sock %d",
1061 app_sock);
1062 goto unlock_rcu;
421cb601 1063 }
a9ad0c8f
MD
1064 ust_app_global_update(sess->ust_session, app);
1065 unlock_rcu:
1066 rcu_read_unlock();
1067 unlock_session:
4ee14516 1068 session_unlock(sess);
487cf67c
DG
1069 }
1070}
1071
7a485870 1072/*
d063d709 1073 * This thread manage event coming from the kernel.
7a485870 1074 *
d063d709
DG
1075 * Features supported in this thread:
1076 * -) CPU Hotplug
7a485870
DG
1077 */
1078static void *thread_manage_kernel(void *data)
1079{
139ac872 1080 int ret, i, pollfd, update_poll_flag = 1, err = -1;
5eb91c98 1081 uint32_t revents, nb_fd;
7a485870 1082 char tmp;
5eb91c98 1083 struct lttng_poll_event events;
7a485870 1084
6993eeb3 1085 DBG("[thread] Thread manage kernel started");
7a485870 1086
6c71277b 1087 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_KERNEL);
927ca06a 1088
d5d63bf1
DG
1089 /*
1090 * This first step of the while is to clean this structure which could free
6d737ce4 1091 * non NULL pointers so initialize it before the loop.
d5d63bf1 1092 */
6d737ce4 1093 lttng_poll_init(&events);
d5d63bf1 1094
e547b070 1095 if (testpoint(sessiond_thread_manage_kernel)) {
6993eeb3
CB
1096 goto error_testpoint;
1097 }
8ac94142 1098
840cb59c 1099 health_code_update();
44a5e5eb 1100
e547b070 1101 if (testpoint(sessiond_thread_manage_kernel_before_loop)) {
d21b0d71 1102 goto error_testpoint;
6993eeb3
CB
1103 }
1104
7a485870 1105 while (1) {
840cb59c 1106 health_code_update();
44a5e5eb 1107
7a485870 1108 if (update_poll_flag == 1) {
d21b0d71
DG
1109 /* Clean events object. We are about to populate it again. */
1110 lttng_poll_clean(&events);
1111
d0b96690 1112 ret = sessiond_set_thread_pollset(&events, 2);
d21b0d71
DG
1113 if (ret < 0) {
1114 goto error_poll_create;
1115 }
1116
1117 ret = lttng_poll_add(&events, kernel_poll_pipe[0], LPOLLIN);
1118 if (ret < 0) {
1119 goto error;
1120 }
5f822d0a 1121
d21b0d71 1122 /* This will add the available kernel channel if any. */
5eb91c98
DG
1123 ret = update_kernel_poll(&events);
1124 if (ret < 0) {
7a485870
DG
1125 goto error;
1126 }
1127 update_poll_flag = 0;
1128 }
1129
7fa2082e 1130 DBG("Thread kernel polling");
7a485870
DG
1131
1132 /* Poll infinite value of time */
88f2b785 1133 restart:
a78af745 1134 health_poll_entry();
5eb91c98 1135 ret = lttng_poll_wait(&events, -1);
7fa2082e
MD
1136 DBG("Thread kernel return from poll on %d fds",
1137 LTTNG_POLL_GETNB(&events));
a78af745 1138 health_poll_exit();
7a485870 1139 if (ret < 0) {
88f2b785
MD
1140 /*
1141 * Restart interrupted system call.
1142 */
1143 if (errno == EINTR) {
1144 goto restart;
1145 }
7a485870
DG
1146 goto error;
1147 } else if (ret == 0) {
1148 /* Should not happen since timeout is infinite */
85611738
DG
1149 ERR("Return value of poll is 0 with an infinite timeout.\n"
1150 "This should not have happened! Continuing...");
7a485870
DG
1151 continue;
1152 }
1153
0d9c5d77
DG
1154 nb_fd = ret;
1155
5eb91c98
DG
1156 for (i = 0; i < nb_fd; i++) {
1157 /* Fetch once the poll data */
1158 revents = LTTNG_POLL_GETEV(&events, i);
1159 pollfd = LTTNG_POLL_GETFD(&events, i);
7a485870 1160
840cb59c 1161 health_code_update();
44a5e5eb 1162
fd20dac9
MD
1163 if (!revents) {
1164 /* No activity for this FD (poll implementation). */
1165 continue;
1166 }
1167
5eb91c98 1168 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 1169 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 1170 if (ret) {
139ac872
MD
1171 err = 0;
1172 goto exit;
5eb91c98 1173 }
7a485870 1174
5eb91c98 1175 /* Check for data on kernel pipe */
03e43155
MD
1176 if (revents & LPOLLIN) {
1177 if (pollfd == kernel_poll_pipe[0]) {
1178 (void) lttng_read(kernel_poll_pipe[0],
1179 &tmp, 1);
1180 /*
1181 * Ret value is useless here, if this pipe gets any actions an
1182 * update is required anyway.
1183 */
1184 update_poll_flag = 1;
1185 continue;
1186 } else {
1187 /*
1188 * New CPU detected by the kernel. Adding kernel stream to
1189 * kernel session and updating the kernel consumer
1190 */
2bdd86d4 1191 ret = update_kernel_stream(&kconsumer_data, pollfd);
5eb91c98
DG
1192 if (ret < 0) {
1193 continue;
1194 }
1195 break;
7a485870 1196 }
03e43155
MD
1197 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
1198 update_poll_flag = 1;
1199 continue;
1200 } else {
1201 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
1202 goto error;
7a485870
DG
1203 }
1204 }
1205 }
1206
139ac872 1207exit:
7a485870 1208error:
5eb91c98 1209 lttng_poll_clean(&events);
76d7553f 1210error_poll_create:
6993eeb3 1211error_testpoint:
6620da75
DG
1212 utils_close_pipe(kernel_poll_pipe);
1213 kernel_poll_pipe[0] = kernel_poll_pipe[1] = -1;
139ac872 1214 if (err) {
840cb59c 1215 health_error();
139ac872 1216 ERR("Health error occurred in %s", __func__);
6620da75
DG
1217 WARN("Kernel thread died unexpectedly. "
1218 "Kernel tracing can continue but CPU hotplug is disabled.");
139ac872 1219 }
8782cc74 1220 health_unregister(health_sessiond);
76d7553f 1221 DBG("Kernel thread dying");
7a485870
DG
1222 return NULL;
1223}
1224
a23ec3a7
DG
1225/*
1226 * Signal pthread condition of the consumer data that the thread.
1227 */
1228static void signal_consumer_condition(struct consumer_data *data, int state)
1229{
1230 pthread_mutex_lock(&data->cond_mutex);
1231
1232 /*
1233 * The state is set before signaling. It can be any value, it's the waiter
1234 * job to correctly interpret this condition variable associated to the
1235 * consumer pthread_cond.
1236 *
1237 * A value of 0 means that the corresponding thread of the consumer data
1238 * was not started. 1 indicates that the thread has started and is ready
1239 * for action. A negative value means that there was an error during the
1240 * thread bootstrap.
1241 */
1242 data->consumer_thread_is_ready = state;
1243 (void) pthread_cond_signal(&data->cond);
1244
1245 pthread_mutex_unlock(&data->cond_mutex);
1246}
1247
1d4b027a 1248/*
3bd1e081 1249 * This thread manage the consumer error sent back to the session daemon.
1d4b027a 1250 */
3bd1e081 1251static void *thread_manage_consumer(void *data)
1d4b027a 1252{
42fc1d0b 1253 int sock = -1, i, ret, pollfd, err = -1, should_quit = 0;
5eb91c98 1254 uint32_t revents, nb_fd;
1d4b027a 1255 enum lttcomm_return_code code;
5eb91c98 1256 struct lttng_poll_event events;
3bd1e081 1257 struct consumer_data *consumer_data = data;
daa81f58 1258 struct consumer_socket *cmd_socket_wrapper = NULL;
1d4b027a 1259
3bd1e081 1260 DBG("[thread] Manage consumer started");
1d4b027a 1261
34c1e15a
MD
1262 rcu_register_thread();
1263 rcu_thread_online();
1264
6c71277b 1265 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_CONSUMER);
927ca06a 1266
855060f8 1267 health_code_update();
9449cc75 1268
5eb91c98 1269 /*
331744e3
JD
1270 * Pass 3 as size here for the thread quit pipe, consumerd_err_sock and the
1271 * metadata_sock. Nothing more will be added to this poll set.
5eb91c98 1272 */
331744e3 1273 ret = sessiond_set_thread_pollset(&events, 3);
5eb91c98 1274 if (ret < 0) {
76d7553f 1275 goto error_poll;
5eb91c98 1276 }
273ea72c 1277
edb8b045
DG
1278 /*
1279 * The error socket here is already in a listening state which was done
1280 * just before spawning this thread to avoid a race between the consumer
1281 * daemon exec trying to connect and the listen() call.
1282 */
3bd1e081 1283 ret = lttng_poll_add(&events, consumer_data->err_sock, LPOLLIN | LPOLLRDHUP);
5eb91c98
DG
1284 if (ret < 0) {
1285 goto error;
1286 }
1287
840cb59c 1288 health_code_update();
44a5e5eb 1289
331744e3 1290 /* Infinite blocking call, waiting for transmission */
88f2b785 1291restart:
a78af745 1292 health_poll_entry();
8ac94142 1293
e547b070 1294 if (testpoint(sessiond_thread_manage_consumer)) {
6993eeb3
CB
1295 goto error;
1296 }
8ac94142 1297
5eb91c98 1298 ret = lttng_poll_wait(&events, -1);
a78af745 1299 health_poll_exit();
273ea72c 1300 if (ret < 0) {
88f2b785
MD
1301 /*
1302 * Restart interrupted system call.
1303 */
1304 if (errno == EINTR) {
1305 goto restart;
1306 }
273ea72c
DG
1307 goto error;
1308 }
1309
0d9c5d77
DG
1310 nb_fd = ret;
1311
5eb91c98
DG
1312 for (i = 0; i < nb_fd; i++) {
1313 /* Fetch once the poll data */
1314 revents = LTTNG_POLL_GETEV(&events, i);
1315 pollfd = LTTNG_POLL_GETFD(&events, i);
1316
840cb59c 1317 health_code_update();
44a5e5eb 1318
fd20dac9
MD
1319 if (!revents) {
1320 /* No activity for this FD (poll implementation). */
1321 continue;
1322 }
1323
5eb91c98 1324 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 1325 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 1326 if (ret) {
139ac872
MD
1327 err = 0;
1328 goto exit;
5eb91c98
DG
1329 }
1330
1331 /* Event on the registration socket */
3bd1e081 1332 if (pollfd == consumer_data->err_sock) {
03e43155
MD
1333 if (revents & LPOLLIN) {
1334 continue;
1335 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
3bd1e081 1336 ERR("consumer err socket poll error");
5eb91c98 1337 goto error;
03e43155
MD
1338 } else {
1339 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
1340 goto error;
5eb91c98
DG
1341 }
1342 }
273ea72c
DG
1343 }
1344
3bd1e081 1345 sock = lttcomm_accept_unix_sock(consumer_data->err_sock);
1d4b027a
DG
1346 if (sock < 0) {
1347 goto error;
1348 }
1349
b662582b
DG
1350 /*
1351 * Set the CLOEXEC flag. Return code is useless because either way, the
1352 * show must go on.
1353 */
1354 (void) utils_set_fd_cloexec(sock);
1355
840cb59c 1356 health_code_update();
44a5e5eb 1357
3bd1e081 1358 DBG2("Receiving code from consumer err_sock");
ee0b0061 1359
712ea556 1360 /* Getting status code from kconsumerd */
54d01ffb
DG
1361 ret = lttcomm_recv_unix_sock(sock, &code,
1362 sizeof(enum lttcomm_return_code));
1d4b027a
DG
1363 if (ret <= 0) {
1364 goto error;
1365 }
1366
840cb59c 1367 health_code_update();
daa81f58 1368 if (code != LTTCOMM_CONSUMERD_COMMAND_SOCK_READY) {
3bd1e081 1369 ERR("consumer error when waiting for SOCK_READY : %s",
1d4b027a
DG
1370 lttcomm_get_readable_code(-code));
1371 goto error;
1372 }
1373
daa81f58
JG
1374 /* Connect both command and metadata sockets. */
1375 consumer_data->cmd_sock =
1376 lttcomm_connect_unix_sock(
1377 consumer_data->cmd_unix_sock_path);
1378 consumer_data->metadata_fd =
1379 lttcomm_connect_unix_sock(
1380 consumer_data->cmd_unix_sock_path);
1381 if (consumer_data->cmd_sock < 0 || consumer_data->metadata_fd < 0) {
1382 PERROR("consumer connect cmd socket");
1383 /* On error, signal condition and quit. */
1384 signal_consumer_condition(consumer_data, -1);
1385 goto error;
1386 }
1387
1388 consumer_data->metadata_sock.fd_ptr = &consumer_data->metadata_fd;
1389
1390 /* Create metadata socket lock. */
1391 consumer_data->metadata_sock.lock = zmalloc(sizeof(pthread_mutex_t));
1392 if (consumer_data->metadata_sock.lock == NULL) {
1393 PERROR("zmalloc pthread mutex");
1394 goto error;
1395 }
1396 pthread_mutex_init(consumer_data->metadata_sock.lock, NULL);
1397
1398 DBG("Consumer command socket ready (fd: %d", consumer_data->cmd_sock);
1399 DBG("Consumer metadata socket ready (fd: %d)",
1400 consumer_data->metadata_fd);
1401
1402 /*
1403 * Remove the consumerd error sock since we've established a connection.
1404 */
3bd1e081 1405 ret = lttng_poll_del(&events, consumer_data->err_sock);
72079cae 1406 if (ret < 0) {
72079cae
DG
1407 goto error;
1408 }
1409
331744e3 1410 /* Add new accepted error socket. */
5eb91c98
DG
1411 ret = lttng_poll_add(&events, sock, LPOLLIN | LPOLLRDHUP);
1412 if (ret < 0) {
72079cae 1413 goto error;
5eb91c98
DG
1414 }
1415
331744e3 1416 /* Add metadata socket that is successfully connected. */
4ce514c4 1417 ret = lttng_poll_add(&events, consumer_data->metadata_fd,
331744e3
JD
1418 LPOLLIN | LPOLLRDHUP);
1419 if (ret < 0) {
1420 goto error;
1421 }
1422
840cb59c 1423 health_code_update();
44a5e5eb 1424
daa81f58
JG
1425 /*
1426 * Transfer the write-end of the channel monitoring pipe to the
1427 * by issuing a SET_CHANNEL_MONITOR_PIPE command.
1428 */
1429 cmd_socket_wrapper = consumer_allocate_socket(&consumer_data->cmd_sock);
1430 if (!cmd_socket_wrapper) {
1431 goto error;
1432 }
1433
1434 ret = consumer_send_channel_monitor_pipe(cmd_socket_wrapper,
1435 consumer_data->channel_monitor_pipe);
1436 if (ret) {
1437 goto error;
1438 }
1439 /* Discard the socket wrapper as it is no longer needed. */
1440 consumer_destroy_socket(cmd_socket_wrapper);
1441 cmd_socket_wrapper = NULL;
1442
1443 /* The thread is completely initialized, signal that it is ready. */
1444 signal_consumer_condition(consumer_data, 1);
1445
331744e3 1446 /* Infinite blocking call, waiting for transmission */
88f2b785 1447restart_poll:
331744e3 1448 while (1) {
42fc1d0b
DG
1449 health_code_update();
1450
1451 /* Exit the thread because the thread quit pipe has been triggered. */
1452 if (should_quit) {
1453 /* Not a health error. */
1454 err = 0;
1455 goto exit;
1456 }
1457
331744e3
JD
1458 health_poll_entry();
1459 ret = lttng_poll_wait(&events, -1);
1460 health_poll_exit();
1461 if (ret < 0) {
1462 /*
1463 * Restart interrupted system call.
1464 */
1465 if (errno == EINTR) {
1466 goto restart_poll;
1467 }
1468 goto error;
88f2b785 1469 }
72079cae 1470
331744e3 1471 nb_fd = ret;
0d9c5d77 1472
331744e3
JD
1473 for (i = 0; i < nb_fd; i++) {
1474 /* Fetch once the poll data */
1475 revents = LTTNG_POLL_GETEV(&events, i);
1476 pollfd = LTTNG_POLL_GETFD(&events, i);
5eb91c98 1477
331744e3 1478 health_code_update();
44a5e5eb 1479
fd20dac9
MD
1480 if (!revents) {
1481 /* No activity for this FD (poll implementation). */
1482 continue;
1483 }
1484
42fc1d0b
DG
1485 /*
1486 * Thread quit pipe has been triggered, flag that we should stop
1487 * but continue the current loop to handle potential data from
1488 * consumer.
1489 */
1490 should_quit = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 1491
331744e3
JD
1492 if (pollfd == sock) {
1493 /* Event on the consumerd socket */
03e43155
MD
1494 if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)
1495 && !(revents & LPOLLIN)) {
331744e3
JD
1496 ERR("consumer err socket second poll error");
1497 goto error;
1498 }
1499 health_code_update();
1500 /* Wait for any kconsumerd error */
1501 ret = lttcomm_recv_unix_sock(sock, &code,
1502 sizeof(enum lttcomm_return_code));
1503 if (ret <= 0) {
1504 ERR("consumer closed the command socket");
1505 goto error;
1506 }
1507
1508 ERR("consumer return code : %s",
1509 lttcomm_get_readable_code(-code));
1510
1511 goto exit;
4ce514c4 1512 } else if (pollfd == consumer_data->metadata_fd) {
03e43155
MD
1513 if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)
1514 && !(revents & LPOLLIN)) {
1515 ERR("consumer err metadata socket second poll error");
1516 goto error;
1517 }
331744e3
JD
1518 /* UST metadata requests */
1519 ret = ust_consumer_metadata_request(
1520 &consumer_data->metadata_sock);
1521 if (ret < 0) {
1522 ERR("Handling metadata request");
1523 goto error;
1524 }
5eb91c98 1525 }
42fc1d0b 1526 /* No need for an else branch all FDs are tested prior. */
5eb91c98 1527 }
331744e3 1528 health_code_update();
5eb91c98
DG
1529 }
1530
139ac872 1531exit:
1d4b027a 1532error:
fdadac08
DG
1533 /*
1534 * We lock here because we are about to close the sockets and some other
92db7cdc
DG
1535 * thread might be using them so get exclusive access which will abort all
1536 * other consumer command by other threads.
fdadac08
DG
1537 */
1538 pthread_mutex_lock(&consumer_data->lock);
1539
5c827ce0
DG
1540 /* Immediately set the consumerd state to stopped */
1541 if (consumer_data->type == LTTNG_CONSUMER_KERNEL) {
1542 uatomic_set(&kernel_consumerd_state, CONSUMER_ERROR);
1543 } else if (consumer_data->type == LTTNG_CONSUMER64_UST ||
1544 consumer_data->type == LTTNG_CONSUMER32_UST) {
1545 uatomic_set(&ust_consumerd_state, CONSUMER_ERROR);
1546 } else {
1547 /* Code flow error... */
1548 assert(0);
1549 }
1550
76d7553f
MD
1551 if (consumer_data->err_sock >= 0) {
1552 ret = close(consumer_data->err_sock);
1553 if (ret) {
1554 PERROR("close");
1555 }
a76cbd9f 1556 consumer_data->err_sock = -1;
76d7553f
MD
1557 }
1558 if (consumer_data->cmd_sock >= 0) {
1559 ret = close(consumer_data->cmd_sock);
1560 if (ret) {
1561 PERROR("close");
1562 }
a76cbd9f 1563 consumer_data->cmd_sock = -1;
76d7553f 1564 }
96544455
SS
1565 if (consumer_data->metadata_sock.fd_ptr &&
1566 *consumer_data->metadata_sock.fd_ptr >= 0) {
9363801e 1567 ret = close(*consumer_data->metadata_sock.fd_ptr);
331744e3
JD
1568 if (ret) {
1569 PERROR("close");
1570 }
1571 }
76d7553f
MD
1572 if (sock >= 0) {
1573 ret = close(sock);
1574 if (ret) {
1575 PERROR("close");
1576 }
1577 }
273ea72c 1578
3bd1e081
MD
1579 unlink(consumer_data->err_unix_sock_path);
1580 unlink(consumer_data->cmd_unix_sock_path);
fdadac08 1581 pthread_mutex_unlock(&consumer_data->lock);
92db7cdc 1582
fdadac08 1583 /* Cleanup metadata socket mutex. */
96544455
SS
1584 if (consumer_data->metadata_sock.lock) {
1585 pthread_mutex_destroy(consumer_data->metadata_sock.lock);
1586 free(consumer_data->metadata_sock.lock);
1587 }
5eb91c98 1588 lttng_poll_clean(&events);
daa81f58
JG
1589
1590 if (cmd_socket_wrapper) {
1591 consumer_destroy_socket(cmd_socket_wrapper);
1592 }
76d7553f 1593error_poll:
139ac872 1594 if (err) {
840cb59c 1595 health_error();
139ac872
MD
1596 ERR("Health error occurred in %s", __func__);
1597 }
8782cc74 1598 health_unregister(health_sessiond);
76d7553f 1599 DBG("consumer thread cleanup completed");
0177d773 1600
34c1e15a
MD
1601 rcu_thread_offline();
1602 rcu_unregister_thread();
1603
5eb91c98 1604 return NULL;
099e26bd
DG
1605}
1606
099e26bd
DG
1607/*
1608 * This thread manage application communication.
1d4b027a
DG
1609 */
1610static void *thread_manage_apps(void *data)
099e26bd 1611{
139ac872 1612 int i, ret, pollfd, err = -1;
6cd525e8 1613 ssize_t size_ret;
5eb91c98 1614 uint32_t revents, nb_fd;
5eb91c98 1615 struct lttng_poll_event events;
099e26bd
DG
1616
1617 DBG("[thread] Manage application started");
1618
f6a9efaa
DG
1619 rcu_register_thread();
1620 rcu_thread_online();
1621
6c71277b 1622 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_APP_MANAGE);
927ca06a 1623
e547b070 1624 if (testpoint(sessiond_thread_manage_apps)) {
6993eeb3
CB
1625 goto error_testpoint;
1626 }
1627
840cb59c 1628 health_code_update();
44a5e5eb 1629
d0b96690 1630 ret = sessiond_set_thread_pollset(&events, 2);
5eb91c98 1631 if (ret < 0) {
76d7553f 1632 goto error_poll_create;
5eb91c98 1633 }
099e26bd 1634
5eb91c98
DG
1635 ret = lttng_poll_add(&events, apps_cmd_pipe[0], LPOLLIN | LPOLLRDHUP);
1636 if (ret < 0) {
1637 goto error;
1638 }
099e26bd 1639
e547b070 1640 if (testpoint(sessiond_thread_manage_apps_before_loop)) {
6993eeb3
CB
1641 goto error;
1642 }
8ac94142 1643
840cb59c 1644 health_code_update();
44a5e5eb 1645
5eb91c98 1646 while (1) {
7fa2082e 1647 DBG("Apps thread polling");
099e26bd
DG
1648
1649 /* Inifinite blocking call, waiting for transmission */
88f2b785 1650 restart:
a78af745 1651 health_poll_entry();
5eb91c98 1652 ret = lttng_poll_wait(&events, -1);
7fa2082e
MD
1653 DBG("Apps thread return from poll on %d fds",
1654 LTTNG_POLL_GETNB(&events));
a78af745 1655 health_poll_exit();
099e26bd 1656 if (ret < 0) {
88f2b785
MD
1657 /*
1658 * Restart interrupted system call.
1659 */
1660 if (errno == EINTR) {
1661 goto restart;
1662 }
099e26bd
DG
1663 goto error;
1664 }
1665
0d9c5d77
DG
1666 nb_fd = ret;
1667
5eb91c98
DG
1668 for (i = 0; i < nb_fd; i++) {
1669 /* Fetch once the poll data */
1670 revents = LTTNG_POLL_GETEV(&events, i);
1671 pollfd = LTTNG_POLL_GETFD(&events, i);
1672
840cb59c 1673 health_code_update();
44a5e5eb 1674
fd20dac9
MD
1675 if (!revents) {
1676 /* No activity for this FD (poll implementation). */
1677 continue;
1678 }
1679
5eb91c98 1680 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 1681 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 1682 if (ret) {
139ac872
MD
1683 err = 0;
1684 goto exit;
5eb91c98 1685 }
099e26bd 1686
5eb91c98
DG
1687 /* Inspect the apps cmd pipe */
1688 if (pollfd == apps_cmd_pipe[0]) {
03e43155 1689 if (revents & LPOLLIN) {
d0b96690
DG
1690 int sock;
1691
5eb91c98 1692 /* Empty pipe */
6cd525e8
MD
1693 size_ret = lttng_read(apps_cmd_pipe[0], &sock, sizeof(sock));
1694 if (size_ret < sizeof(sock)) {
76d7553f 1695 PERROR("read apps cmd pipe");
5eb91c98
DG
1696 goto error;
1697 }
099e26bd 1698
840cb59c 1699 health_code_update();
44a5e5eb 1700
ffe60014 1701 /*
03e43155
MD
1702 * Since this is a command socket (write then read),
1703 * we only monitor the error events of the socket.
ffe60014 1704 */
d0b96690
DG
1705 ret = lttng_poll_add(&events, sock,
1706 LPOLLERR | LPOLLHUP | LPOLLRDHUP);
1707 if (ret < 0) {
5eb91c98 1708 goto error;
e0c7ec2b 1709 }
acc7b41b 1710
d0b96690 1711 DBG("Apps with sock %d added to poll set", sock);
03e43155
MD
1712 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
1713 ERR("Apps command pipe error");
1714 goto error;
1715 } else {
1716 ERR("Unknown poll events %u for sock %d", revents, pollfd);
1717 goto error;
0177d773 1718 }
5eb91c98
DG
1719 } else {
1720 /*
54d01ffb
DG
1721 * At this point, we know that a registered application made
1722 * the event at poll_wait.
5eb91c98
DG
1723 */
1724 if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
1725 /* Removing from the poll set */
1726 ret = lttng_poll_del(&events, pollfd);
1727 if (ret < 0) {
1728 goto error;
1729 }
099e26bd 1730
b9d9b220 1731 /* Socket closed on remote end. */
56fff090 1732 ust_app_unregister(pollfd);
03e43155
MD
1733 } else {
1734 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
1735 goto error;
5eb91c98 1736 }
099e26bd 1737 }
44a5e5eb 1738
840cb59c 1739 health_code_update();
099e26bd 1740 }
099e26bd
DG
1741 }
1742
139ac872 1743exit:
099e26bd 1744error:
5eb91c98 1745 lttng_poll_clean(&events);
76d7553f 1746error_poll_create:
6993eeb3 1747error_testpoint:
6620da75
DG
1748 utils_close_pipe(apps_cmd_pipe);
1749 apps_cmd_pipe[0] = apps_cmd_pipe[1] = -1;
1750
1751 /*
1752 * We don't clean the UST app hash table here since already registered
1753 * applications can still be controlled so let them be until the session
1754 * daemon dies or the applications stop.
1755 */
1756
139ac872 1757 if (err) {
840cb59c 1758 health_error();
139ac872
MD
1759 ERR("Health error occurred in %s", __func__);
1760 }
8782cc74 1761 health_unregister(health_sessiond);
76d7553f 1762 DBG("Application communication apps thread cleanup complete");
f6a9efaa
DG
1763 rcu_thread_offline();
1764 rcu_unregister_thread();
099e26bd
DG
1765 return NULL;
1766}
1767
d0b96690 1768/*
d88aee68
DG
1769 * Send a socket to a thread This is called from the dispatch UST registration
1770 * thread once all sockets are set for the application.
d0b96690 1771 *
b85dc84c
DG
1772 * The sock value can be invalid, we don't really care, the thread will handle
1773 * it and make the necessary cleanup if so.
1774 *
d0b96690
DG
1775 * On success, return 0 else a negative value being the errno message of the
1776 * write().
1777 */
d88aee68 1778static int send_socket_to_thread(int fd, int sock)
d0b96690 1779{
6cd525e8 1780 ssize_t ret;
d0b96690 1781
b85dc84c
DG
1782 /*
1783 * It's possible that the FD is set as invalid with -1 concurrently just
1784 * before calling this function being a shutdown state of the thread.
1785 */
1786 if (fd < 0) {
1787 ret = -EBADF;
1788 goto error;
1789 }
d0b96690 1790
6cd525e8
MD
1791 ret = lttng_write(fd, &sock, sizeof(sock));
1792 if (ret < sizeof(sock)) {
d88aee68 1793 PERROR("write apps pipe %d", fd);
d0b96690
DG
1794 if (ret < 0) {
1795 ret = -errno;
1796 }
1797 goto error;
1798 }
1799
1800 /* All good. Don't send back the write positive ret value. */
1801 ret = 0;
1802error:
6cd525e8 1803 return (int) ret;
d0b96690
DG
1804}
1805
f45e313d
DG
1806/*
1807 * Sanitize the wait queue of the dispatch registration thread meaning removing
1808 * invalid nodes from it. This is to avoid memory leaks for the case the UST
1809 * notify socket is never received.
1810 */
1811static void sanitize_wait_queue(struct ust_reg_wait_queue *wait_queue)
1812{
1813 int ret, nb_fd = 0, i;
1814 unsigned int fd_added = 0;
1815 struct lttng_poll_event events;
1816 struct ust_reg_wait_node *wait_node = NULL, *tmp_wait_node;
1817
1818 assert(wait_queue);
1819
1820 lttng_poll_init(&events);
1821
1822 /* Just skip everything for an empty queue. */
1823 if (!wait_queue->count) {
1824 goto end;
1825 }
1826
1827 ret = lttng_poll_create(&events, wait_queue->count, LTTNG_CLOEXEC);
1828 if (ret < 0) {
1829 goto error_create;
1830 }
1831
1832 cds_list_for_each_entry_safe(wait_node, tmp_wait_node,
1833 &wait_queue->head, head) {
1834 assert(wait_node->app);
1835 ret = lttng_poll_add(&events, wait_node->app->sock,
1836 LPOLLHUP | LPOLLERR);
1837 if (ret < 0) {
1838 goto error;
1839 }
1840
1841 fd_added = 1;
1842 }
1843
1844 if (!fd_added) {
1845 goto end;
1846 }
1847
1848 /*
1849 * Poll but don't block so we can quickly identify the faulty events and
1850 * clean them afterwards from the wait queue.
1851 */
1852 ret = lttng_poll_wait(&events, 0);
1853 if (ret < 0) {
1854 goto error;
1855 }
1856 nb_fd = ret;
1857
1858 for (i = 0; i < nb_fd; i++) {
1859 /* Get faulty FD. */
1860 uint32_t revents = LTTNG_POLL_GETEV(&events, i);
1861 int pollfd = LTTNG_POLL_GETFD(&events, i);
1862
fd20dac9
MD
1863 if (!revents) {
1864 /* No activity for this FD (poll implementation). */
1865 continue;
1866 }
1867
f45e313d
DG
1868 cds_list_for_each_entry_safe(wait_node, tmp_wait_node,
1869 &wait_queue->head, head) {
1870 if (pollfd == wait_node->app->sock &&
1871 (revents & (LPOLLHUP | LPOLLERR))) {
1872 cds_list_del(&wait_node->head);
1873 wait_queue->count--;
1874 ust_app_destroy(wait_node->app);
1875 free(wait_node);
48b40bcf
JG
1876 /*
1877 * Silence warning of use-after-free in
1878 * cds_list_for_each_entry_safe which uses
1879 * __typeof__(*wait_node).
1880 */
1881 wait_node = NULL;
f45e313d 1882 break;
03e43155
MD
1883 } else {
1884 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
1885 goto error;
f45e313d
DG
1886 }
1887 }
1888 }
1889
1890 if (nb_fd > 0) {
1891 DBG("Wait queue sanitized, %d node were cleaned up", nb_fd);
1892 }
1893
1894end:
1895 lttng_poll_clean(&events);
1896 return;
1897
1898error:
1899 lttng_poll_clean(&events);
1900error_create:
1901 ERR("Unable to sanitize wait queue");
1902 return;
1903}
1904
099e26bd
DG
1905/*
1906 * Dispatch request from the registration threads to the application
1907 * communication thread.
1908 */
1909static void *thread_dispatch_ust_registration(void *data)
1910{
12e2b881 1911 int ret, err = -1;
8bdee6e2 1912 struct cds_wfcq_node *node;
099e26bd 1913 struct ust_command *ust_cmd = NULL;
f45e313d
DG
1914 struct ust_reg_wait_node *wait_node = NULL, *tmp_wait_node;
1915 struct ust_reg_wait_queue wait_queue = {
1916 .count = 0,
1917 };
d0b96690 1918
967e3668
MD
1919 rcu_register_thread();
1920
6c71277b 1921 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_APP_REG_DISPATCH);
12e2b881 1922
9ad42ec1
MD
1923 if (testpoint(sessiond_thread_app_reg_dispatch)) {
1924 goto error_testpoint;
1925 }
1926
12e2b881
MD
1927 health_code_update();
1928
f45e313d 1929 CDS_INIT_LIST_HEAD(&wait_queue.head);
099e26bd
DG
1930
1931 DBG("[thread] Dispatch UST command started");
1932
26c9d55e 1933 while (!CMM_LOAD_SHARED(dispatch_thread_exit)) {
12e2b881
MD
1934 health_code_update();
1935
099e26bd
DG
1936 /* Atomically prepare the queue futex */
1937 futex_nto1_prepare(&ust_cmd_queue.futex);
1938
1939 do {
d0b96690 1940 struct ust_app *app = NULL;
7972aab2 1941 ust_cmd = NULL;
d0b96690 1942
f45e313d
DG
1943 /*
1944 * Make sure we don't have node(s) that have hung up before receiving
1945 * the notify socket. This is to clean the list in order to avoid
1946 * memory leaks from notify socket that are never seen.
1947 */
1948 sanitize_wait_queue(&wait_queue);
1949
12e2b881 1950 health_code_update();
099e26bd 1951 /* Dequeue command for registration */
8bdee6e2 1952 node = cds_wfcq_dequeue_blocking(&ust_cmd_queue.head, &ust_cmd_queue.tail);
099e26bd 1953 if (node == NULL) {
00a17c97 1954 DBG("Woken up but nothing in the UST command queue");
099e26bd
DG
1955 /* Continue thread execution */
1956 break;
1957 }
1958
1959 ust_cmd = caa_container_of(node, struct ust_command, node);
1960
2f50c8a3
DG
1961 DBG("Dispatching UST registration pid:%d ppid:%d uid:%d"
1962 " gid:%d sock:%d name:%s (version %d.%d)",
1963 ust_cmd->reg_msg.pid, ust_cmd->reg_msg.ppid,
1964 ust_cmd->reg_msg.uid, ust_cmd->reg_msg.gid,
1965 ust_cmd->sock, ust_cmd->reg_msg.name,
1966 ust_cmd->reg_msg.major, ust_cmd->reg_msg.minor);
d0b96690
DG
1967
1968 if (ust_cmd->reg_msg.type == USTCTL_SOCKET_CMD) {
1969 wait_node = zmalloc(sizeof(*wait_node));
1970 if (!wait_node) {
1971 PERROR("zmalloc wait_node dispatch");
020d7f60
DG
1972 ret = close(ust_cmd->sock);
1973 if (ret < 0) {
1974 PERROR("close ust sock dispatch %d", ust_cmd->sock);
1975 }
51dec90d 1976 lttng_fd_put(LTTNG_FD_APPS, 1);
7972aab2 1977 free(ust_cmd);
d0b96690
DG
1978 goto error;
1979 }
1980 CDS_INIT_LIST_HEAD(&wait_node->head);
1981
1982 /* Create application object if socket is CMD. */
1983 wait_node->app = ust_app_create(&ust_cmd->reg_msg,
1984 ust_cmd->sock);
1985 if (!wait_node->app) {
1986 ret = close(ust_cmd->sock);
1987 if (ret < 0) {
1988 PERROR("close ust sock dispatch %d", ust_cmd->sock);
6620da75 1989 }
51dec90d 1990 lttng_fd_put(LTTNG_FD_APPS, 1);
d88aee68 1991 free(wait_node);
7972aab2 1992 free(ust_cmd);
d0b96690
DG
1993 continue;
1994 }
1995 /*
1996 * Add application to the wait queue so we can set the notify
1997 * socket before putting this object in the global ht.
1998 */
f45e313d
DG
1999 cds_list_add(&wait_node->head, &wait_queue.head);
2000 wait_queue.count++;
d0b96690 2001
7972aab2 2002 free(ust_cmd);
d0b96690
DG
2003 /*
2004 * We have to continue here since we don't have the notify
2005 * socket and the application MUST be added to the hash table
2006 * only at that moment.
2007 */
2008 continue;
2009 } else {
2010 /*
2011 * Look for the application in the local wait queue and set the
2012 * notify socket if found.
2013 */
d88aee68 2014 cds_list_for_each_entry_safe(wait_node, tmp_wait_node,
f45e313d 2015 &wait_queue.head, head) {
12e2b881 2016 health_code_update();
d0b96690
DG
2017 if (wait_node->app->pid == ust_cmd->reg_msg.pid) {
2018 wait_node->app->notify_sock = ust_cmd->sock;
2019 cds_list_del(&wait_node->head);
f45e313d 2020 wait_queue.count--;
d0b96690
DG
2021 app = wait_node->app;
2022 free(wait_node);
2023 DBG3("UST app notify socket %d is set", ust_cmd->sock);
2024 break;
2025 }
2026 }
020d7f60
DG
2027
2028 /*
2029 * With no application at this stage the received socket is
2030 * basically useless so close it before we free the cmd data
2031 * structure for good.
2032 */
2033 if (!app) {
2034 ret = close(ust_cmd->sock);
2035 if (ret < 0) {
2036 PERROR("close ust sock dispatch %d", ust_cmd->sock);
2037 }
51dec90d 2038 lttng_fd_put(LTTNG_FD_APPS, 1);
020d7f60 2039 }
7972aab2 2040 free(ust_cmd);
d0b96690
DG
2041 }
2042
2043 if (app) {
d0b96690
DG
2044 /*
2045 * @session_lock_list
2046 *
2047 * Lock the global session list so from the register up to the
2048 * registration done message, no thread can see the application
2049 * and change its state.
2050 */
2051 session_lock_list();
2052 rcu_read_lock();
d88aee68 2053
d0b96690
DG
2054 /*
2055 * Add application to the global hash table. This needs to be
2056 * done before the update to the UST registry can locate the
2057 * application.
2058 */
2059 ust_app_add(app);
d88aee68
DG
2060
2061 /* Set app version. This call will print an error if needed. */
2062 (void) ust_app_version(app);
2063
2064 /* Send notify socket through the notify pipe. */
2065 ret = send_socket_to_thread(apps_cmd_notify_pipe[1],
2066 app->notify_sock);
2067 if (ret < 0) {
2068 rcu_read_unlock();
2069 session_unlock_list();
b85dc84c
DG
2070 /*
2071 * No notify thread, stop the UST tracing. However, this is
2072 * not an internal error of the this thread thus setting
2073 * the health error code to a normal exit.
2074 */
2075 err = 0;
d88aee68 2076 goto error;
6620da75 2077 }
d88aee68 2078
d0b96690
DG
2079 /*
2080 * Update newly registered application with the tracing
2081 * registry info already enabled information.
2082 */
2083 update_ust_app(app->sock);
d88aee68
DG
2084
2085 /*
2086 * Don't care about return value. Let the manage apps threads
2087 * handle app unregistration upon socket close.
2088 */
fb45065e 2089 (void) ust_app_register_done(app);
d88aee68
DG
2090
2091 /*
2092 * Even if the application socket has been closed, send the app
2093 * to the thread and unregistration will take place at that
2094 * place.
2095 */
2096 ret = send_socket_to_thread(apps_cmd_pipe[1], app->sock);
d0b96690 2097 if (ret < 0) {
d88aee68
DG
2098 rcu_read_unlock();
2099 session_unlock_list();
b85dc84c
DG
2100 /*
2101 * No apps. thread, stop the UST tracing. However, this is
2102 * not an internal error of the this thread thus setting
2103 * the health error code to a normal exit.
2104 */
2105 err = 0;
d88aee68 2106 goto error;
d0b96690 2107 }
d88aee68 2108
d0b96690
DG
2109 rcu_read_unlock();
2110 session_unlock_list();
099e26bd 2111 }
099e26bd
DG
2112 } while (node != NULL);
2113
12e2b881 2114 health_poll_entry();
099e26bd
DG
2115 /* Futex wait on queue. Blocking call on futex() */
2116 futex_nto1_wait(&ust_cmd_queue.futex);
12e2b881 2117 health_poll_exit();
099e26bd 2118 }
12e2b881
MD
2119 /* Normal exit, no error */
2120 err = 0;
099e26bd
DG
2121
2122error:
d88aee68
DG
2123 /* Clean up wait queue. */
2124 cds_list_for_each_entry_safe(wait_node, tmp_wait_node,
f45e313d 2125 &wait_queue.head, head) {
d88aee68 2126 cds_list_del(&wait_node->head);
f45e313d 2127 wait_queue.count--;
d88aee68
DG
2128 free(wait_node);
2129 }
2130
772b8f4d
MD
2131 /* Empty command queue. */
2132 for (;;) {
2133 /* Dequeue command for registration */
2134 node = cds_wfcq_dequeue_blocking(&ust_cmd_queue.head, &ust_cmd_queue.tail);
2135 if (node == NULL) {
2136 break;
2137 }
2138 ust_cmd = caa_container_of(node, struct ust_command, node);
2139 ret = close(ust_cmd->sock);
2140 if (ret < 0) {
2141 PERROR("close ust sock exit dispatch %d", ust_cmd->sock);
2142 }
2143 lttng_fd_put(LTTNG_FD_APPS, 1);
2144 free(ust_cmd);
2145 }
2146
9ad42ec1 2147error_testpoint:
099e26bd 2148 DBG("Dispatch thread dying");
12e2b881
MD
2149 if (err) {
2150 health_error();
2151 ERR("Health error occurred in %s", __func__);
2152 }
8782cc74 2153 health_unregister(health_sessiond);
967e3668 2154 rcu_unregister_thread();
099e26bd
DG
2155 return NULL;
2156}
2157
2158/*
2159 * This thread manage application registration.
2160 */
2161static void *thread_registration_apps(void *data)
1d4b027a 2162{
139ac872 2163 int sock = -1, i, ret, pollfd, err = -1;
5eb91c98
DG
2164 uint32_t revents, nb_fd;
2165 struct lttng_poll_event events;
099e26bd
DG
2166 /*
2167 * Get allocated in this thread, enqueued to a global queue, dequeued and
2168 * freed in the manage apps thread.
2169 */
2170 struct ust_command *ust_cmd = NULL;
1d4b027a 2171
099e26bd 2172 DBG("[thread] Manage application registration started");
1d4b027a 2173
6c71277b 2174 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_APP_REG);
927ca06a 2175
e547b070 2176 if (testpoint(sessiond_thread_registration_apps)) {
6993eeb3
CB
2177 goto error_testpoint;
2178 }
8ac94142 2179
1d4b027a
DG
2180 ret = lttcomm_listen_unix_sock(apps_sock);
2181 if (ret < 0) {
76d7553f 2182 goto error_listen;
1d4b027a
DG
2183 }
2184
5eb91c98
DG
2185 /*
2186 * Pass 2 as size here for the thread quit pipe and apps socket. Nothing
2187 * more will be added to this poll set.
2188 */
d0b96690 2189 ret = sessiond_set_thread_pollset(&events, 2);
5eb91c98 2190 if (ret < 0) {
76d7553f 2191 goto error_create_poll;
5eb91c98 2192 }
273ea72c 2193
5eb91c98
DG
2194 /* Add the application registration socket */
2195 ret = lttng_poll_add(&events, apps_sock, LPOLLIN | LPOLLRDHUP);
2196 if (ret < 0) {
76d7553f 2197 goto error_poll_add;
5eb91c98 2198 }
273ea72c 2199
1d4b027a 2200 /* Notify all applications to register */
0fdd1e2c
DG
2201 ret = notify_ust_apps(1);
2202 if (ret < 0) {
2203 ERR("Failed to notify applications or create the wait shared memory.\n"
54d01ffb
DG
2204 "Execution continues but there might be problem for already\n"
2205 "running applications that wishes to register.");
0fdd1e2c 2206 }
1d4b027a
DG
2207
2208 while (1) {
2209 DBG("Accepting application registration");
273ea72c
DG
2210
2211 /* Inifinite blocking call, waiting for transmission */
88f2b785 2212 restart:
a78af745 2213 health_poll_entry();
5eb91c98 2214 ret = lttng_poll_wait(&events, -1);
a78af745 2215 health_poll_exit();
273ea72c 2216 if (ret < 0) {
88f2b785
MD
2217 /*
2218 * Restart interrupted system call.
2219 */
2220 if (errno == EINTR) {
2221 goto restart;
2222 }
273ea72c
DG
2223 goto error;
2224 }
2225
0d9c5d77
DG
2226 nb_fd = ret;
2227
5eb91c98 2228 for (i = 0; i < nb_fd; i++) {
840cb59c 2229 health_code_update();
139ac872 2230
5eb91c98
DG
2231 /* Fetch once the poll data */
2232 revents = LTTNG_POLL_GETEV(&events, i);
2233 pollfd = LTTNG_POLL_GETFD(&events, i);
273ea72c 2234
fd20dac9
MD
2235 if (!revents) {
2236 /* No activity for this FD (poll implementation). */
2237 continue;
2238 }
2239
5eb91c98 2240 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 2241 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 2242 if (ret) {
139ac872
MD
2243 err = 0;
2244 goto exit;
90014c57 2245 }
1d4b027a 2246
5eb91c98
DG
2247 /* Event on the registration socket */
2248 if (pollfd == apps_sock) {
03e43155 2249 if (revents & LPOLLIN) {
5eb91c98
DG
2250 sock = lttcomm_accept_unix_sock(apps_sock);
2251 if (sock < 0) {
2252 goto error;
2253 }
099e26bd 2254
16c5c8fa
DG
2255 /*
2256 * Set socket timeout for both receiving and ending.
2257 * app_socket_timeout is in seconds, whereas
2258 * lttcomm_setsockopt_rcv_timeout and
2259 * lttcomm_setsockopt_snd_timeout expect msec as
2260 * parameter.
2261 */
28ce0ff2
JG
2262 if (app_socket_timeout >= 0) {
2263 (void) lttcomm_setsockopt_rcv_timeout(sock,
2264 app_socket_timeout * 1000);
2265 (void) lttcomm_setsockopt_snd_timeout(sock,
2266 app_socket_timeout * 1000);
2267 }
16c5c8fa 2268
b662582b
DG
2269 /*
2270 * Set the CLOEXEC flag. Return code is useless because
2271 * either way, the show must go on.
2272 */
2273 (void) utils_set_fd_cloexec(sock);
2274
5eb91c98 2275 /* Create UST registration command for enqueuing */
ba7f0ae5 2276 ust_cmd = zmalloc(sizeof(struct ust_command));
5eb91c98 2277 if (ust_cmd == NULL) {
76d7553f 2278 PERROR("ust command zmalloc");
41ed8e47
MD
2279 ret = close(sock);
2280 if (ret) {
2281 PERROR("close");
2282 }
5eb91c98
DG
2283 goto error;
2284 }
1d4b027a 2285
5eb91c98
DG
2286 /*
2287 * Using message-based transmissions to ensure we don't
2288 * have to deal with partially received messages.
2289 */
4063050c
MD
2290 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2291 if (ret < 0) {
2292 ERR("Exhausted file descriptors allowed for applications.");
2293 free(ust_cmd);
2294 ret = close(sock);
2295 if (ret) {
2296 PERROR("close");
2297 }
2298 sock = -1;
2299 continue;
2300 }
d88aee68 2301
840cb59c 2302 health_code_update();
d0b96690
DG
2303 ret = ust_app_recv_registration(sock, &ust_cmd->reg_msg);
2304 if (ret < 0) {
5eb91c98 2305 free(ust_cmd);
d0b96690 2306 /* Close socket of the application. */
76d7553f
MD
2307 ret = close(sock);
2308 if (ret) {
2309 PERROR("close");
2310 }
4063050c 2311 lttng_fd_put(LTTNG_FD_APPS, 1);
76d7553f 2312 sock = -1;
5eb91c98
DG
2313 continue;
2314 }
840cb59c 2315 health_code_update();
099e26bd 2316
5eb91c98 2317 ust_cmd->sock = sock;
34a2494f 2318 sock = -1;
099e26bd 2319
5eb91c98
DG
2320 DBG("UST registration received with pid:%d ppid:%d uid:%d"
2321 " gid:%d sock:%d name:%s (version %d.%d)",
2322 ust_cmd->reg_msg.pid, ust_cmd->reg_msg.ppid,
2323 ust_cmd->reg_msg.uid, ust_cmd->reg_msg.gid,
2324 ust_cmd->sock, ust_cmd->reg_msg.name,
2325 ust_cmd->reg_msg.major, ust_cmd->reg_msg.minor);
54d01ffb 2326
5eb91c98
DG
2327 /*
2328 * Lock free enqueue the registration request. The red pill
54d01ffb 2329 * has been taken! This apps will be part of the *system*.
5eb91c98 2330 */
8bdee6e2 2331 cds_wfcq_enqueue(&ust_cmd_queue.head, &ust_cmd_queue.tail, &ust_cmd->node);
5eb91c98
DG
2332
2333 /*
2334 * Wake the registration queue futex. Implicit memory
8bdee6e2 2335 * barrier with the exchange in cds_wfcq_enqueue.
5eb91c98
DG
2336 */
2337 futex_nto1_wake(&ust_cmd_queue.futex);
03e43155
MD
2338 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
2339 ERR("Register apps socket poll error");
2340 goto error;
2341 } else {
2342 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
2343 goto error;
5eb91c98
DG
2344 }
2345 }
90014c57 2346 }
1d4b027a
DG
2347 }
2348
139ac872 2349exit:
1d4b027a 2350error:
0fdd1e2c
DG
2351 /* Notify that the registration thread is gone */
2352 notify_ust_apps(0);
2353
a4b35e07 2354 if (apps_sock >= 0) {
76d7553f
MD
2355 ret = close(apps_sock);
2356 if (ret) {
2357 PERROR("close");
2358 }
a4b35e07 2359 }
46c3f085 2360 if (sock >= 0) {
76d7553f
MD
2361 ret = close(sock);
2362 if (ret) {
2363 PERROR("close");
2364 }
4063050c 2365 lttng_fd_put(LTTNG_FD_APPS, 1);
a4b35e07 2366 }
273ea72c 2367 unlink(apps_unix_sock_path);
0fdd1e2c 2368
76d7553f 2369error_poll_add:
5eb91c98 2370 lttng_poll_clean(&events);
76d7553f
MD
2371error_listen:
2372error_create_poll:
6993eeb3 2373error_testpoint:
76d7553f 2374 DBG("UST Registration thread cleanup complete");
9ad42ec1
MD
2375 if (err) {
2376 health_error();
2377 ERR("Health error occurred in %s", __func__);
2378 }
8782cc74 2379 health_unregister(health_sessiond);
5eb91c98 2380
1d4b027a
DG
2381 return NULL;
2382}
2383
8c0faa1d 2384/*
3bd1e081 2385 * Start the thread_manage_consumer. This must be done after a lttng-consumerd
d063d709 2386 * exec or it will fails.
8c0faa1d 2387 */
3bd1e081 2388static int spawn_consumer_thread(struct consumer_data *consumer_data)
8c0faa1d 2389{
a23ec3a7 2390 int ret, clock_ret;
ee0b0061
DG
2391 struct timespec timeout;
2392
13a7bce3
JG
2393 /*
2394 * Make sure we set the readiness flag to 0 because we are NOT ready.
2395 * This access to consumer_thread_is_ready does not need to be
2396 * protected by consumer_data.cond_mutex (yet) since the consumer
2397 * management thread has not been started at this point.
2398 */
a23ec3a7 2399 consumer_data->consumer_thread_is_ready = 0;
8c0faa1d 2400
a23ec3a7
DG
2401 /* Setup pthread condition */
2402 ret = pthread_condattr_init(&consumer_data->condattr);
4a15001e 2403 if (ret) {
a23ec3a7
DG
2404 errno = ret;
2405 PERROR("pthread_condattr_init consumer data");
2406 goto error;
2407 }
2408
2409 /*
2410 * Set the monotonic clock in order to make sure we DO NOT jump in time
2411 * between the clock_gettime() call and the timedwait call. See bug #324
2412 * for a more details and how we noticed it.
2413 */
2414 ret = pthread_condattr_setclock(&consumer_data->condattr, CLOCK_MONOTONIC);
4a15001e 2415 if (ret) {
a23ec3a7
DG
2416 errno = ret;
2417 PERROR("pthread_condattr_setclock consumer data");
ee0b0061
DG
2418 goto error;
2419 }
8c0faa1d 2420
a23ec3a7 2421 ret = pthread_cond_init(&consumer_data->cond, &consumer_data->condattr);
4a15001e 2422 if (ret) {
a23ec3a7
DG
2423 errno = ret;
2424 PERROR("pthread_cond_init consumer data");
2425 goto error;
2426 }
2427
1a1a34b4
MJ
2428 ret = pthread_create(&consumer_data->thread, default_pthread_attr(),
2429 thread_manage_consumer, consumer_data);
4a15001e
MD
2430 if (ret) {
2431 errno = ret;
3bd1e081 2432 PERROR("pthread_create consumer");
ee0b0061 2433 ret = -1;
8c0faa1d
DG
2434 goto error;
2435 }
2436
a23ec3a7
DG
2437 /* We are about to wait on a pthread condition */
2438 pthread_mutex_lock(&consumer_data->cond_mutex);
2439
ee0b0061 2440 /* Get time for sem_timedwait absolute timeout */
389fbf04 2441 clock_ret = lttng_clock_gettime(CLOCK_MONOTONIC, &timeout);
a23ec3a7
DG
2442 /*
2443 * Set the timeout for the condition timed wait even if the clock gettime
2444 * call fails since we might loop on that call and we want to avoid to
2445 * increment the timeout too many times.
2446 */
2447 timeout.tv_sec += DEFAULT_SEM_WAIT_TIMEOUT;
2448
2449 /*
2450 * The following loop COULD be skipped in some conditions so this is why we
2451 * set ret to 0 in order to make sure at least one round of the loop is
2452 * done.
2453 */
2454 ret = 0;
2455
2456 /*
2457 * Loop until the condition is reached or when a timeout is reached. Note
2458 * that the pthread_cond_timedwait(P) man page specifies that EINTR can NOT
2459 * be returned but the pthread_cond(3), from the glibc-doc, says that it is
2460 * possible. This loop does not take any chances and works with both of
2461 * them.
2462 */
2463 while (!consumer_data->consumer_thread_is_ready && ret != ETIMEDOUT) {
2464 if (clock_ret < 0) {
2465 PERROR("clock_gettime spawn consumer");
2466 /* Infinite wait for the consumerd thread to be ready */
2467 ret = pthread_cond_wait(&consumer_data->cond,
2468 &consumer_data->cond_mutex);
2469 } else {
2470 ret = pthread_cond_timedwait(&consumer_data->cond,
2471 &consumer_data->cond_mutex, &timeout);
2472 }
ee0b0061 2473 }
8c0faa1d 2474
a23ec3a7
DG
2475 /* Release the pthread condition */
2476 pthread_mutex_unlock(&consumer_data->cond_mutex);
2477
2478 if (ret != 0) {
2479 errno = ret;
2480 if (ret == ETIMEDOUT) {
4282f9a3
DG
2481 int pth_ret;
2482
ee0b0061
DG
2483 /*
2484 * Call has timed out so we kill the kconsumerd_thread and return
2485 * an error.
2486 */
a23ec3a7
DG
2487 ERR("Condition timed out. The consumer thread was never ready."
2488 " Killing it");
4282f9a3
DG
2489 pth_ret = pthread_cancel(consumer_data->thread);
2490 if (pth_ret < 0) {
3bd1e081 2491 PERROR("pthread_cancel consumer thread");
ee0b0061
DG
2492 }
2493 } else {
a23ec3a7 2494 PERROR("pthread_cond_wait failed consumer thread");
ee0b0061 2495 }
4282f9a3
DG
2496 /* Caller is expecting a negative value on failure. */
2497 ret = -1;
ee0b0061
DG
2498 goto error;
2499 }
2500
3bd1e081
MD
2501 pthread_mutex_lock(&consumer_data->pid_mutex);
2502 if (consumer_data->pid == 0) {
a23ec3a7 2503 ERR("Consumerd did not start");
3bd1e081 2504 pthread_mutex_unlock(&consumer_data->pid_mutex);
712ea556
DG
2505 goto error;
2506 }
3bd1e081 2507 pthread_mutex_unlock(&consumer_data->pid_mutex);
712ea556 2508
8c0faa1d
DG
2509 return 0;
2510
2511error:
2512 return ret;
2513}
2514
d9800920 2515/*
3bd1e081 2516 * Join consumer thread
d9800920 2517 */
3bd1e081 2518static int join_consumer_thread(struct consumer_data *consumer_data)
cf3af59e
MD
2519{
2520 void *status;
cf3af59e 2521
e8209f6b
DG
2522 /* Consumer pid must be a real one. */
2523 if (consumer_data->pid > 0) {
c617c0c6 2524 int ret;
3bd1e081 2525 ret = kill(consumer_data->pid, SIGTERM);
cf3af59e 2526 if (ret) {
4a15001e 2527 PERROR("Error killing consumer daemon");
cf3af59e
MD
2528 return ret;
2529 }
3bd1e081 2530 return pthread_join(consumer_data->thread, &status);
cf3af59e
MD
2531 } else {
2532 return 0;
2533 }
2534}
2535
8c0faa1d 2536/*
3bd1e081 2537 * Fork and exec a consumer daemon (consumerd).
8c0faa1d 2538 *
d063d709 2539 * Return pid if successful else -1.
8c0faa1d 2540 */
3bd1e081 2541static pid_t spawn_consumerd(struct consumer_data *consumer_data)
8c0faa1d
DG
2542{
2543 int ret;
2544 pid_t pid;
94c55f17 2545 const char *consumer_to_use;
53086306 2546 const char *verbosity;
94c55f17 2547 struct stat st;
8c0faa1d 2548
3bd1e081 2549 DBG("Spawning consumerd");
c49dc785 2550
8c0faa1d
DG
2551 pid = fork();
2552 if (pid == 0) {
2553 /*
3bd1e081 2554 * Exec consumerd.
8c0faa1d 2555 */
daee5345 2556 if (opt_verbose_consumer) {
53086306 2557 verbosity = "--verbose";
4421f712 2558 } else if (lttng_opt_quiet) {
53086306 2559 verbosity = "--quiet";
4421f712
DG
2560 } else {
2561 verbosity = "";
53086306 2562 }
4421f712 2563
3bd1e081
MD
2564 switch (consumer_data->type) {
2565 case LTTNG_CONSUMER_KERNEL:
94c55f17 2566 /*
c7704d57
DG
2567 * Find out which consumerd to execute. We will first try the
2568 * 64-bit path, then the sessiond's installation directory, and
2569 * fallback on the 32-bit one,
94c55f17 2570 */
63a799e8
AM
2571 DBG3("Looking for a kernel consumer at these locations:");
2572 DBG3(" 1) %s", consumerd64_bin);
2573 DBG3(" 2) %s/%s", INSTALL_BIN_PATH, CONSUMERD_FILE);
2574 DBG3(" 3) %s", consumerd32_bin);
94c55f17 2575 if (stat(consumerd64_bin, &st) == 0) {
63a799e8 2576 DBG3("Found location #1");
94c55f17 2577 consumer_to_use = consumerd64_bin;
94c55f17 2578 } else if (stat(INSTALL_BIN_PATH "/" CONSUMERD_FILE, &st) == 0) {
63a799e8 2579 DBG3("Found location #2");
94c55f17 2580 consumer_to_use = INSTALL_BIN_PATH "/" CONSUMERD_FILE;
eb1e0bd4 2581 } else if (stat(consumerd32_bin, &st) == 0) {
63a799e8 2582 DBG3("Found location #3");
eb1e0bd4 2583 consumer_to_use = consumerd32_bin;
94c55f17 2584 } else {
63a799e8 2585 DBG("Could not find any valid consumerd executable");
4282f9a3 2586 ret = -EINVAL;
94c55f17
AM
2587 break;
2588 }
2589 DBG("Using kernel consumer at: %s", consumer_to_use);
4282f9a3 2590 ret = execl(consumer_to_use,
94c55f17
AM
2591 "lttng-consumerd", verbosity, "-k",
2592 "--consumerd-cmd-sock", consumer_data->cmd_unix_sock_path,
2593 "--consumerd-err-sock", consumer_data->err_unix_sock_path,
6c71277b 2594 "--group", tracing_group_name,
94c55f17 2595 NULL);
3bd1e081 2596 break;
7753dea8
MD
2597 case LTTNG_CONSUMER64_UST:
2598 {
b1e0b6b6 2599 char *tmpnew = NULL;
8f4905da
MD
2600
2601 if (consumerd64_libdir[0] != '\0') {
2602 char *tmp;
2603 size_t tmplen;
2604
e8fa9fb0 2605 tmp = lttng_secure_getenv("LD_LIBRARY_PATH");
8f4905da
MD
2606 if (!tmp) {
2607 tmp = "";
2608 }
2609 tmplen = strlen("LD_LIBRARY_PATH=")
2610 + strlen(consumerd64_libdir) + 1 /* : */ + strlen(tmp);
2611 tmpnew = zmalloc(tmplen + 1 /* \0 */);
2612 if (!tmpnew) {
2613 ret = -ENOMEM;
2614 goto error;
2615 }
2616 strcpy(tmpnew, "LD_LIBRARY_PATH=");
2617 strcat(tmpnew, consumerd64_libdir);
2618 if (tmp[0] != '\0') {
2619 strcat(tmpnew, ":");
2620 strcat(tmpnew, tmp);
2621 }
2622 ret = putenv(tmpnew);
2623 if (ret) {
2624 ret = -errno;
c6f76da9 2625 free(tmpnew);
8f4905da
MD
2626 goto error;
2627 }
2628 }
94c55f17 2629 DBG("Using 64-bit UST consumer at: %s", consumerd64_bin);
a5a6aff3 2630 ret = execl(consumerd64_bin, "lttng-consumerd", verbosity, "-u",
7753dea8
MD
2631 "--consumerd-cmd-sock", consumer_data->cmd_unix_sock_path,
2632 "--consumerd-err-sock", consumer_data->err_unix_sock_path,
6c71277b 2633 "--group", tracing_group_name,
7753dea8 2634 NULL);
8f4905da
MD
2635 if (consumerd64_libdir[0] != '\0') {
2636 free(tmpnew);
2637 }
3bd1e081 2638 break;
7753dea8
MD
2639 }
2640 case LTTNG_CONSUMER32_UST:
2641 {
937dde8e 2642 char *tmpnew = NULL;
8f4905da
MD
2643
2644 if (consumerd32_libdir[0] != '\0') {
2645 char *tmp;
2646 size_t tmplen;
2647
e8fa9fb0 2648 tmp = lttng_secure_getenv("LD_LIBRARY_PATH");
8f4905da
MD
2649 if (!tmp) {
2650 tmp = "";
2651 }
2652 tmplen = strlen("LD_LIBRARY_PATH=")
2653 + strlen(consumerd32_libdir) + 1 /* : */ + strlen(tmp);
2654 tmpnew = zmalloc(tmplen + 1 /* \0 */);
2655 if (!tmpnew) {
2656 ret = -ENOMEM;
2657 goto error;
2658 }
2659 strcpy(tmpnew, "LD_LIBRARY_PATH=");
2660 strcat(tmpnew, consumerd32_libdir);
2661 if (tmp[0] != '\0') {
2662 strcat(tmpnew, ":");
2663 strcat(tmpnew, tmp);
2664 }
2665 ret = putenv(tmpnew);
2666 if (ret) {
2667 ret = -errno;
c6f76da9 2668 free(tmpnew);
8f4905da
MD
2669 goto error;
2670 }
2671 }
94c55f17 2672 DBG("Using 32-bit UST consumer at: %s", consumerd32_bin);
a5a6aff3 2673 ret = execl(consumerd32_bin, "lttng-consumerd", verbosity, "-u",
7753dea8
MD
2674 "--consumerd-cmd-sock", consumer_data->cmd_unix_sock_path,
2675 "--consumerd-err-sock", consumer_data->err_unix_sock_path,
6c71277b 2676 "--group", tracing_group_name,
7753dea8 2677 NULL);
8f4905da
MD
2678 if (consumerd32_libdir[0] != '\0') {
2679 free(tmpnew);
2680 }
7753dea8
MD
2681 break;
2682 }
3bd1e081 2683 default:
76d7553f 2684 PERROR("unknown consumer type");
3bd1e081
MD
2685 exit(EXIT_FAILURE);
2686 }
8c0faa1d 2687 if (errno != 0) {
4282f9a3 2688 PERROR("Consumer execl()");
8c0faa1d 2689 }
4282f9a3 2690 /* Reaching this point, we got a failure on our execl(). */
8c0faa1d
DG
2691 exit(EXIT_FAILURE);
2692 } else if (pid > 0) {
2693 ret = pid;
8c0faa1d 2694 } else {
76d7553f 2695 PERROR("start consumer fork");
8c0faa1d 2696 ret = -errno;
8c0faa1d 2697 }
8f4905da 2698error:
8c0faa1d
DG
2699 return ret;
2700}
2701
693bd40b 2702/*
3bd1e081 2703 * Spawn the consumerd daemon and session daemon thread.
693bd40b 2704 */
3bd1e081 2705static int start_consumerd(struct consumer_data *consumer_data)
693bd40b 2706{
c617c0c6 2707 int ret;
edb8b045
DG
2708
2709 /*
2710 * Set the listen() state on the socket since there is a possible race
2711 * between the exec() of the consumer daemon and this call if place in the
2712 * consumer thread. See bug #366 for more details.
2713 */
2714 ret = lttcomm_listen_unix_sock(consumer_data->err_sock);
2715 if (ret < 0) {
2716 goto error;
2717 }
693bd40b 2718
3bd1e081
MD
2719 pthread_mutex_lock(&consumer_data->pid_mutex);
2720 if (consumer_data->pid != 0) {
2721 pthread_mutex_unlock(&consumer_data->pid_mutex);
c49dc785
DG
2722 goto end;
2723 }
693bd40b 2724
3bd1e081 2725 ret = spawn_consumerd(consumer_data);
c49dc785 2726 if (ret < 0) {
3bd1e081
MD
2727 ERR("Spawning consumerd failed");
2728 pthread_mutex_unlock(&consumer_data->pid_mutex);
c49dc785 2729 goto error;
693bd40b 2730 }
c49dc785 2731
3bd1e081
MD
2732 /* Setting up the consumer_data pid */
2733 consumer_data->pid = ret;
48842b30 2734 DBG2("Consumer pid %d", consumer_data->pid);
3bd1e081 2735 pthread_mutex_unlock(&consumer_data->pid_mutex);
693bd40b 2736
3bd1e081
MD
2737 DBG2("Spawning consumer control thread");
2738 ret = spawn_consumer_thread(consumer_data);
693bd40b 2739 if (ret < 0) {
3bd1e081 2740 ERR("Fatal error spawning consumer control thread");
693bd40b
DG
2741 goto error;
2742 }
2743
c49dc785 2744end:
693bd40b
DG
2745 return 0;
2746
2747error:
331744e3 2748 /* Cleanup already created sockets on error. */
edb8b045 2749 if (consumer_data->err_sock >= 0) {
c617c0c6
MD
2750 int err;
2751
edb8b045
DG
2752 err = close(consumer_data->err_sock);
2753 if (err < 0) {
2754 PERROR("close consumer data error socket");
2755 }
2756 }
693bd40b
DG
2757 return ret;
2758}
2759
b73401da 2760/*
096102bd 2761 * Setup necessary data for kernel tracer action.
b73401da 2762 */
096102bd 2763static int init_kernel_tracer(void)
b73401da
DG
2764{
2765 int ret;
b73401da 2766
096102bd
DG
2767 /* Modprobe lttng kernel modules */
2768 ret = modprobe_lttng_control();
b73401da 2769 if (ret < 0) {
b73401da
DG
2770 goto error;
2771 }
2772
096102bd
DG
2773 /* Open debugfs lttng */
2774 kernel_tracer_fd = open(module_proc_lttng, O_RDWR);
2775 if (kernel_tracer_fd < 0) {
2776 DBG("Failed to open %s", module_proc_lttng);
2f77fc4b
DG
2777 ret = -1;
2778 goto error_open;
54d01ffb
DG
2779 }
2780
2f77fc4b
DG
2781 /* Validate kernel version */
2782 ret = kernel_validate_version(kernel_tracer_fd);
2783 if (ret < 0) {
2784 goto error_version;
b551a063 2785 }
54d01ffb 2786
2f77fc4b
DG
2787 ret = modprobe_lttng_data();
2788 if (ret < 0) {
2789 goto error_modules;
54d01ffb
DG
2790 }
2791
2f77fc4b
DG
2792 DBG("Kernel tracer fd %d", kernel_tracer_fd);
2793 return 0;
2794
2795error_version:
2796 modprobe_remove_lttng_control();
2797 ret = close(kernel_tracer_fd);
2798 if (ret) {
2799 PERROR("close");
b551a063 2800 }
2f77fc4b 2801 kernel_tracer_fd = -1;
f73fabfd 2802 return LTTNG_ERR_KERN_VERSION;
b551a063 2803
2f77fc4b
DG
2804error_modules:
2805 ret = close(kernel_tracer_fd);
2806 if (ret) {
2807 PERROR("close");
b551a063 2808 }
54d01ffb 2809
2f77fc4b
DG
2810error_open:
2811 modprobe_remove_lttng_control();
54d01ffb
DG
2812
2813error:
2f77fc4b
DG
2814 WARN("No kernel tracer available");
2815 kernel_tracer_fd = -1;
2816 if (!is_root) {
f73fabfd 2817 return LTTNG_ERR_NEED_ROOT_SESSIOND;
2f77fc4b 2818 } else {
f73fabfd 2819 return LTTNG_ERR_KERN_NA;
2f77fc4b 2820 }
54d01ffb
DG
2821}
2822
2f77fc4b 2823
54d01ffb 2824/*
2f77fc4b
DG
2825 * Copy consumer output from the tracing session to the domain session. The
2826 * function also applies the right modification on a per domain basis for the
2827 * trace files destination directory.
36b588ed
MD
2828 *
2829 * Should *NOT* be called with RCU read-side lock held.
54d01ffb 2830 */
2f77fc4b 2831static int copy_session_consumer(int domain, struct ltt_session *session)
54d01ffb
DG
2832{
2833 int ret;
2f77fc4b
DG
2834 const char *dir_name;
2835 struct consumer_output *consumer;
2836
2837 assert(session);
2838 assert(session->consumer);
54d01ffb 2839
b551a063
DG
2840 switch (domain) {
2841 case LTTNG_DOMAIN_KERNEL:
2f77fc4b 2842 DBG3("Copying tracing session consumer output in kernel session");
09a90bcd
DG
2843 /*
2844 * XXX: We should audit the session creation and what this function
2845 * does "extra" in order to avoid a destroy since this function is used
2846 * in the domain session creation (kernel and ust) only. Same for UST
2847 * domain.
2848 */
2849 if (session->kernel_session->consumer) {
6addfa37 2850 consumer_output_put(session->kernel_session->consumer);
09a90bcd 2851 }
2f77fc4b
DG
2852 session->kernel_session->consumer =
2853 consumer_copy_output(session->consumer);
2854 /* Ease our life a bit for the next part */
2855 consumer = session->kernel_session->consumer;
2856 dir_name = DEFAULT_KERNEL_TRACE_DIR;
b551a063 2857 break;
f20baf8e 2858 case LTTNG_DOMAIN_JUL:
5cdb6027 2859 case LTTNG_DOMAIN_LOG4J:
0e115563 2860 case LTTNG_DOMAIN_PYTHON:
b551a063 2861 case LTTNG_DOMAIN_UST:
2f77fc4b 2862 DBG3("Copying tracing session consumer output in UST session");
09a90bcd 2863 if (session->ust_session->consumer) {
6addfa37 2864 consumer_output_put(session->ust_session->consumer);
09a90bcd 2865 }
2f77fc4b
DG
2866 session->ust_session->consumer =
2867 consumer_copy_output(session->consumer);
2868 /* Ease our life a bit for the next part */
2869 consumer = session->ust_session->consumer;
2870 dir_name = DEFAULT_UST_TRACE_DIR;
b551a063
DG
2871 break;
2872 default:
f73fabfd 2873 ret = LTTNG_ERR_UNKNOWN_DOMAIN;
54d01ffb
DG
2874 goto error;
2875 }
2876
2f77fc4b 2877 /* Append correct directory to subdir */
c30ce0b3
CB
2878 strncat(consumer->subdir, dir_name,
2879 sizeof(consumer->subdir) - strlen(consumer->subdir) - 1);
2f77fc4b
DG
2880 DBG3("Copy session consumer subdir %s", consumer->subdir);
2881
f73fabfd 2882 ret = LTTNG_OK;
54d01ffb
DG
2883
2884error:
2885 return ret;
2886}
2887
00e2e675 2888/*
2f77fc4b 2889 * Create an UST session and add it to the session ust list.
36b588ed
MD
2890 *
2891 * Should *NOT* be called with RCU read-side lock held.
00e2e675 2892 */
2f77fc4b
DG
2893static int create_ust_session(struct ltt_session *session,
2894 struct lttng_domain *domain)
00e2e675
DG
2895{
2896 int ret;
2f77fc4b 2897 struct ltt_ust_session *lus = NULL;
00e2e675 2898
a4b92340 2899 assert(session);
2f77fc4b
DG
2900 assert(domain);
2901 assert(session->consumer);
a4b92340 2902
2f77fc4b 2903 switch (domain->type) {
f20baf8e 2904 case LTTNG_DOMAIN_JUL:
5cdb6027 2905 case LTTNG_DOMAIN_LOG4J:
0e115563 2906 case LTTNG_DOMAIN_PYTHON:
2f77fc4b
DG
2907 case LTTNG_DOMAIN_UST:
2908 break;
2909 default:
2910 ERR("Unknown UST domain on create session %d", domain->type);
f73fabfd 2911 ret = LTTNG_ERR_UNKNOWN_DOMAIN;
00e2e675
DG
2912 goto error;
2913 }
2914
2f77fc4b
DG
2915 DBG("Creating UST session");
2916
dec56f6c 2917 lus = trace_ust_create_session(session->id);
2f77fc4b 2918 if (lus == NULL) {
f73fabfd 2919 ret = LTTNG_ERR_UST_SESS_FAIL;
a4b92340
DG
2920 goto error;
2921 }
2922
2f77fc4b
DG
2923 lus->uid = session->uid;
2924 lus->gid = session->gid;
2bba9e53 2925 lus->output_traces = session->output_traces;
27babd3a 2926 lus->snapshot_mode = session->snapshot_mode;
ecc48a90 2927 lus->live_timer_interval = session->live_timer;
2f77fc4b 2928 session->ust_session = lus;
d7ba1388 2929 if (session->shm_path[0]) {
3d071855
MD
2930 strncpy(lus->root_shm_path, session->shm_path,
2931 sizeof(lus->root_shm_path));
2932 lus->root_shm_path[sizeof(lus->root_shm_path) - 1] = '\0';
d7ba1388
MD
2933 strncpy(lus->shm_path, session->shm_path,
2934 sizeof(lus->shm_path));
2935 lus->shm_path[sizeof(lus->shm_path) - 1] = '\0';
2936 strncat(lus->shm_path, "/ust",
2937 sizeof(lus->shm_path) - strlen(lus->shm_path) - 1);
2938 }
2f77fc4b
DG
2939 /* Copy session output to the newly created UST session */
2940 ret = copy_session_consumer(domain->type, session);
f73fabfd 2941 if (ret != LTTNG_OK) {
00e2e675
DG
2942 goto error;
2943 }
2944
f73fabfd 2945 return LTTNG_OK;
00e2e675
DG
2946
2947error:
2f77fc4b
DG
2948 free(lus);
2949 session->ust_session = NULL;
00e2e675
DG
2950 return ret;
2951}
2952
2953/*
2f77fc4b 2954 * Create a kernel tracer session then create the default channel.
00e2e675 2955 */
2f77fc4b 2956static int create_kernel_session(struct ltt_session *session)
00e2e675
DG
2957{
2958 int ret;
a4b92340 2959
2f77fc4b 2960 DBG("Creating kernel session");
00e2e675 2961
2f77fc4b
DG
2962 ret = kernel_create_session(session, kernel_tracer_fd);
2963 if (ret < 0) {
f73fabfd 2964 ret = LTTNG_ERR_KERN_SESS_FAIL;
00e2e675
DG
2965 goto error;
2966 }
2967
2f77fc4b
DG
2968 /* Code flow safety */
2969 assert(session->kernel_session);
2970
2971 /* Copy session output to the newly created Kernel session */
2972 ret = copy_session_consumer(LTTNG_DOMAIN_KERNEL, session);
f73fabfd 2973 if (ret != LTTNG_OK) {
a4b92340
DG
2974 goto error;
2975 }
2976
2f77fc4b
DG
2977 /* Create directory(ies) on local filesystem. */
2978 if (session->kernel_session->consumer->type == CONSUMER_DST_LOCAL &&
2979 strlen(session->kernel_session->consumer->dst.trace_path) > 0) {
2980 ret = run_as_mkdir_recursive(
2981 session->kernel_session->consumer->dst.trace_path,
2982 S_IRWXU | S_IRWXG, session->uid, session->gid);
2983 if (ret < 0) {
df5b86c8 2984 if (errno != EEXIST) {
2f77fc4b 2985 ERR("Trace directory creation error");
00e2e675
DG
2986 goto error;
2987 }
00e2e675 2988 }
2f77fc4b 2989 }
00e2e675 2990
2f77fc4b
DG
2991 session->kernel_session->uid = session->uid;
2992 session->kernel_session->gid = session->gid;
2bba9e53 2993 session->kernel_session->output_traces = session->output_traces;
27babd3a 2994 session->kernel_session->snapshot_mode = session->snapshot_mode;
00e2e675 2995
f73fabfd 2996 return LTTNG_OK;
00e2e675 2997
2f77fc4b
DG
2998error:
2999 trace_kernel_destroy_session(session->kernel_session);
3000 session->kernel_session = NULL;
3001 return ret;
3002}
00e2e675 3003
2f77fc4b
DG
3004/*
3005 * Count number of session permitted by uid/gid.
3006 */
3007static unsigned int lttng_sessions_count(uid_t uid, gid_t gid)
3008{
3009 unsigned int i = 0;
3010 struct ltt_session *session;
07424f16 3011
2f77fc4b
DG
3012 DBG("Counting number of available session for UID %d GID %d",
3013 uid, gid);
3014 cds_list_for_each_entry(session, &session_list_ptr->head, list) {
00e2e675 3015 /*
2f77fc4b 3016 * Only list the sessions the user can control.
00e2e675 3017 */
2f77fc4b
DG
3018 if (!session_access_ok(session, uid, gid)) {
3019 continue;
3020 }
3021 i++;
a4b92340 3022 }
2f77fc4b 3023 return i;
00e2e675
DG
3024}
3025
54d01ffb
DG
3026/*
3027 * Process the command requested by the lttng client within the command
3028 * context structure. This function make sure that the return structure (llm)
3029 * is set and ready for transmission before returning.
3030 *
3031 * Return any error encountered or 0 for success.
53a80697
MD
3032 *
3033 * "sock" is only used for special-case var. len data.
36b588ed
MD
3034 *
3035 * Should *NOT* be called with RCU read-side lock held.
54d01ffb 3036 */
53a80697
MD
3037static int process_client_msg(struct command_ctx *cmd_ctx, int sock,
3038 int *sock_error)
54d01ffb 3039{
f73fabfd 3040 int ret = LTTNG_OK;
44d3bd01 3041 int need_tracing_session = 1;
2e09ba09 3042 int need_domain;
54d01ffb
DG
3043
3044 DBG("Processing client command %d", cmd_ctx->lsm->cmd_type);
3045
3745d315
MD
3046 assert(!rcu_read_ongoing());
3047
53a80697
MD
3048 *sock_error = 0;
3049
2e09ba09
MD
3050 switch (cmd_ctx->lsm->cmd_type) {
3051 case LTTNG_CREATE_SESSION:
27babd3a 3052 case LTTNG_CREATE_SESSION_SNAPSHOT:
ecc48a90 3053 case LTTNG_CREATE_SESSION_LIVE:
2e09ba09
MD
3054 case LTTNG_DESTROY_SESSION:
3055 case LTTNG_LIST_SESSIONS:
3056 case LTTNG_LIST_DOMAINS:
3057 case LTTNG_START_TRACE:
3058 case LTTNG_STOP_TRACE:
6d805429 3059 case LTTNG_DATA_PENDING:
da3c9ec1
DG
3060 case LTTNG_SNAPSHOT_ADD_OUTPUT:
3061 case LTTNG_SNAPSHOT_DEL_OUTPUT:
3062 case LTTNG_SNAPSHOT_LIST_OUTPUT:
3063 case LTTNG_SNAPSHOT_RECORD:
fb198a11 3064 case LTTNG_SAVE_SESSION:
d7ba1388 3065 case LTTNG_SET_SESSION_SHM_PATH:
eded6438 3066 case LTTNG_REGENERATE_METADATA:
c2561365 3067 case LTTNG_REGENERATE_STATEDUMP:
daa81f58
JG
3068 case LTTNG_REGISTER_TRIGGER:
3069 case LTTNG_UNREGISTER_TRIGGER:
2e09ba09 3070 need_domain = 0;
3aace903 3071 break;
2e09ba09
MD
3072 default:
3073 need_domain = 1;
3074 }
3075
3076 if (opt_no_kernel && need_domain
3077 && cmd_ctx->lsm->domain.type == LTTNG_DOMAIN_KERNEL) {
531d29f9 3078 if (!is_root) {
f73fabfd 3079 ret = LTTNG_ERR_NEED_ROOT_SESSIOND;
531d29f9 3080 } else {
f73fabfd 3081 ret = LTTNG_ERR_KERN_NA;
531d29f9 3082 }
4fba7219
DG
3083 goto error;
3084 }
3085
8d3113b2
DG
3086 /* Deny register consumer if we already have a spawned consumer. */
3087 if (cmd_ctx->lsm->cmd_type == LTTNG_REGISTER_CONSUMER) {
3088 pthread_mutex_lock(&kconsumer_data.pid_mutex);
3089 if (kconsumer_data.pid > 0) {
f73fabfd 3090 ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
fa317f24 3091 pthread_mutex_unlock(&kconsumer_data.pid_mutex);
8d3113b2
DG
3092 goto error;
3093 }
3094 pthread_mutex_unlock(&kconsumer_data.pid_mutex);
3095 }
3096
54d01ffb
DG
3097 /*
3098 * Check for command that don't needs to allocate a returned payload. We do
44d3bd01 3099 * this here so we don't have to make the call for no payload at each
54d01ffb
DG
3100 * command.
3101 */
3102 switch(cmd_ctx->lsm->cmd_type) {
3103 case LTTNG_LIST_SESSIONS:
3104 case LTTNG_LIST_TRACEPOINTS:
f37d259d 3105 case LTTNG_LIST_TRACEPOINT_FIELDS:
54d01ffb
DG
3106 case LTTNG_LIST_DOMAINS:
3107 case LTTNG_LIST_CHANNELS:
3108 case LTTNG_LIST_EVENTS:
834978fd 3109 case LTTNG_LIST_SYSCALLS:
a5dfbb9d 3110 case LTTNG_LIST_TRACKER_PIDS:
5cd0780d 3111 case LTTNG_DATA_PENDING:
54d01ffb
DG
3112 break;
3113 default:
3114 /* Setup lttng message with no payload */
6e10c9b9 3115 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, NULL, 0);
54d01ffb
DG
3116 if (ret < 0) {
3117 /* This label does not try to unlock the session */
3118 goto init_setup_error;
3119 }
3120 }
3121
3122 /* Commands that DO NOT need a session. */
3123 switch (cmd_ctx->lsm->cmd_type) {
54d01ffb 3124 case LTTNG_CREATE_SESSION:
27babd3a 3125 case LTTNG_CREATE_SESSION_SNAPSHOT:
ecc48a90 3126 case LTTNG_CREATE_SESSION_LIVE:
54d01ffb
DG
3127 case LTTNG_LIST_SESSIONS:
3128 case LTTNG_LIST_TRACEPOINTS:
834978fd 3129 case LTTNG_LIST_SYSCALLS:
f37d259d 3130 case LTTNG_LIST_TRACEPOINT_FIELDS:
fb198a11 3131 case LTTNG_SAVE_SESSION:
daa81f58
JG
3132 case LTTNG_REGISTER_TRIGGER:
3133 case LTTNG_UNREGISTER_TRIGGER:
44d3bd01 3134 need_tracing_session = 0;
54d01ffb
DG
3135 break;
3136 default:
3137 DBG("Getting session %s by name", cmd_ctx->lsm->session.name);
256a5576
MD
3138 /*
3139 * We keep the session list lock across _all_ commands
3140 * for now, because the per-session lock does not
3141 * handle teardown properly.
3142 */
74babd95 3143 session_lock_list();
54d01ffb
DG
3144 cmd_ctx->session = session_find_by_name(cmd_ctx->lsm->session.name);
3145 if (cmd_ctx->session == NULL) {
bba2d65f 3146 ret = LTTNG_ERR_SESS_NOT_FOUND;
54d01ffb
DG
3147 goto error;
3148 } else {
3149 /* Acquire lock for the session */
3150 session_lock(cmd_ctx->session);
3151 }
3152 break;
3153 }
b389abbe 3154
5f3ecf22
DG
3155 /*
3156 * Commands that need a valid session but should NOT create one if none
3157 * exists. Instead of creating one and destroying it when the command is
3158 * handled, process that right before so we save some round trip in useless
3159 * code path.
3160 */
3161 switch (cmd_ctx->lsm->cmd_type) {
3162 case LTTNG_DISABLE_CHANNEL:
3163 case LTTNG_DISABLE_EVENT:
5f3ecf22
DG
3164 switch (cmd_ctx->lsm->domain.type) {
3165 case LTTNG_DOMAIN_KERNEL:
3166 if (!cmd_ctx->session->kernel_session) {
3167 ret = LTTNG_ERR_NO_CHANNEL;
3168 goto error;
3169 }
3170 break;
3171 case LTTNG_DOMAIN_JUL:
5cdb6027 3172 case LTTNG_DOMAIN_LOG4J:
0e115563 3173 case LTTNG_DOMAIN_PYTHON:
5f3ecf22
DG
3174 case LTTNG_DOMAIN_UST:
3175 if (!cmd_ctx->session->ust_session) {
3176 ret = LTTNG_ERR_NO_CHANNEL;
3177 goto error;
3178 }
3179 break;
3180 default:
3181 ret = LTTNG_ERR_UNKNOWN_DOMAIN;
3182 goto error;
3183 }
3184 default:
3185 break;
3186 }
3187
2e09ba09
MD
3188 if (!need_domain) {
3189 goto skip_domain;
3190 }
a4b92340 3191
54d01ffb
DG
3192 /*
3193 * Check domain type for specific "pre-action".
3194 */
3195 switch (cmd_ctx->lsm->domain.type) {
3196 case LTTNG_DOMAIN_KERNEL:
d1f1c568 3197 if (!is_root) {
f73fabfd 3198 ret = LTTNG_ERR_NEED_ROOT_SESSIOND;
d1f1c568
DG
3199 goto error;
3200 }
3201
54d01ffb 3202 /* Kernel tracer check */
a4b35e07 3203 if (kernel_tracer_fd == -1) {
54d01ffb 3204 /* Basically, load kernel tracer modules */
096102bd
DG
3205 ret = init_kernel_tracer();
3206 if (ret != 0) {
54d01ffb
DG
3207 goto error;
3208 }
3209 }
5eb91c98 3210
5c827ce0
DG
3211 /* Consumer is in an ERROR state. Report back to client */
3212 if (uatomic_read(&kernel_consumerd_state) == CONSUMER_ERROR) {
f73fabfd 3213 ret = LTTNG_ERR_NO_KERNCONSUMERD;
5c827ce0
DG
3214 goto error;
3215 }
3216
54d01ffb 3217 /* Need a session for kernel command */
44d3bd01 3218 if (need_tracing_session) {
54d01ffb 3219 if (cmd_ctx->session->kernel_session == NULL) {
6df2e2c9 3220 ret = create_kernel_session(cmd_ctx->session);
5eb91c98 3221 if (ret < 0) {
f73fabfd 3222 ret = LTTNG_ERR_KERN_SESS_FAIL;
5eb91c98
DG
3223 goto error;
3224 }
b389abbe 3225 }
7d29a247 3226
54d01ffb 3227 /* Start the kernel consumer daemon */
3bd1e081
MD
3228 pthread_mutex_lock(&kconsumer_data.pid_mutex);
3229 if (kconsumer_data.pid == 0 &&
785d2d0d 3230 cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) {
3bd1e081
MD
3231 pthread_mutex_unlock(&kconsumer_data.pid_mutex);
3232 ret = start_consumerd(&kconsumer_data);
7d29a247 3233 if (ret < 0) {
f73fabfd 3234 ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
54d01ffb 3235 goto error;
950131af 3236 }
5c827ce0 3237 uatomic_set(&kernel_consumerd_state, CONSUMER_STARTED);
3ff2ecac
MD
3238 } else {
3239 pthread_mutex_unlock(&kconsumer_data.pid_mutex);
33a2b854 3240 }
173af62f 3241
a4b92340
DG
3242 /*
3243 * The consumer was just spawned so we need to add the socket to
3244 * the consumer output of the session if exist.
3245 */
3246 ret = consumer_create_socket(&kconsumer_data,
3247 cmd_ctx->session->kernel_session->consumer);
3248 if (ret < 0) {
3249 goto error;
173af62f 3250 }
0d0c377a 3251 }
5c827ce0 3252
54d01ffb 3253 break;
b9dfb167 3254 case LTTNG_DOMAIN_JUL:
5cdb6027 3255 case LTTNG_DOMAIN_LOG4J:
0e115563 3256 case LTTNG_DOMAIN_PYTHON:
2bdd86d4 3257 case LTTNG_DOMAIN_UST:
44d3bd01 3258 {
b51ec5b4
MD
3259 if (!ust_app_supported()) {
3260 ret = LTTNG_ERR_NO_UST;
3261 goto error;
3262 }
5c827ce0
DG
3263 /* Consumer is in an ERROR state. Report back to client */
3264 if (uatomic_read(&ust_consumerd_state) == CONSUMER_ERROR) {
f73fabfd 3265 ret = LTTNG_ERR_NO_USTCONSUMERD;
5c827ce0
DG
3266 goto error;
3267 }
3268
44d3bd01 3269 if (need_tracing_session) {
a4b92340 3270 /* Create UST session if none exist. */
f6a9efaa 3271 if (cmd_ctx->session->ust_session == NULL) {
44d3bd01 3272 ret = create_ust_session(cmd_ctx->session,
6df2e2c9 3273 &cmd_ctx->lsm->domain);
f73fabfd 3274 if (ret != LTTNG_OK) {
44d3bd01
DG
3275 goto error;
3276 }
3277 }
00e2e675 3278
7753dea8
MD
3279 /* Start the UST consumer daemons */
3280 /* 64-bit */
3281 pthread_mutex_lock(&ustconsumer64_data.pid_mutex);
fc7a59ce 3282 if (consumerd64_bin[0] != '\0' &&
7753dea8 3283 ustconsumer64_data.pid == 0 &&
785d2d0d 3284 cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) {
7753dea8
MD
3285 pthread_mutex_unlock(&ustconsumer64_data.pid_mutex);
3286 ret = start_consumerd(&ustconsumer64_data);
2bdd86d4 3287 if (ret < 0) {
f73fabfd 3288 ret = LTTNG_ERR_UST_CONSUMER64_FAIL;
173af62f 3289 uatomic_set(&ust_consumerd64_fd, -EINVAL);
2bdd86d4
MD
3290 goto error;
3291 }
48842b30 3292
173af62f 3293 uatomic_set(&ust_consumerd64_fd, ustconsumer64_data.cmd_sock);
5c827ce0 3294 uatomic_set(&ust_consumerd_state, CONSUMER_STARTED);
3ff2ecac 3295 } else {
7753dea8
MD
3296 pthread_mutex_unlock(&ustconsumer64_data.pid_mutex);
3297 }
173af62f
DG
3298
3299 /*
3300 * Setup socket for consumer 64 bit. No need for atomic access
3301 * since it was set above and can ONLY be set in this thread.
3302 */
a4b92340
DG
3303 ret = consumer_create_socket(&ustconsumer64_data,
3304 cmd_ctx->session->ust_session->consumer);
3305 if (ret < 0) {
3306 goto error;
173af62f
DG
3307 }
3308
7753dea8 3309 /* 32-bit */
385b881b 3310 pthread_mutex_lock(&ustconsumer32_data.pid_mutex);
fc7a59ce 3311 if (consumerd32_bin[0] != '\0' &&
7753dea8 3312 ustconsumer32_data.pid == 0 &&
785d2d0d 3313 cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) {
7753dea8
MD
3314 pthread_mutex_unlock(&ustconsumer32_data.pid_mutex);
3315 ret = start_consumerd(&ustconsumer32_data);
3316 if (ret < 0) {
f73fabfd 3317 ret = LTTNG_ERR_UST_CONSUMER32_FAIL;
173af62f 3318 uatomic_set(&ust_consumerd32_fd, -EINVAL);
7753dea8
MD
3319 goto error;
3320 }
5c827ce0 3321
173af62f 3322 uatomic_set(&ust_consumerd32_fd, ustconsumer32_data.cmd_sock);
5c827ce0 3323 uatomic_set(&ust_consumerd_state, CONSUMER_STARTED);
7753dea8
MD
3324 } else {
3325 pthread_mutex_unlock(&ustconsumer32_data.pid_mutex);
2bdd86d4 3326 }
173af62f
DG
3327
3328 /*
3329 * Setup socket for consumer 64 bit. No need for atomic access
3330 * since it was set above and can ONLY be set in this thread.
3331 */
a4b92340
DG
3332 ret = consumer_create_socket(&ustconsumer32_data,
3333 cmd_ctx->session->ust_session->consumer);
3334 if (ret < 0) {
3335 goto error;
173af62f 3336 }
44d3bd01
DG
3337 }
3338 break;
48842b30 3339 }
54d01ffb 3340 default:
54d01ffb
DG
3341 break;
3342 }
2e09ba09 3343skip_domain:
33a2b854 3344
5c827ce0
DG
3345 /* Validate consumer daemon state when start/stop trace command */
3346 if (cmd_ctx->lsm->cmd_type == LTTNG_START_TRACE ||
3347 cmd_ctx->lsm->cmd_type == LTTNG_STOP_TRACE) {
3348 switch (cmd_ctx->lsm->domain.type) {
e7619875
JG
3349 case LTTNG_DOMAIN_NONE:
3350 break;
b9dfb167 3351 case LTTNG_DOMAIN_JUL:
5cdb6027 3352 case LTTNG_DOMAIN_LOG4J:
0e115563 3353 case LTTNG_DOMAIN_PYTHON:
5c827ce0
DG
3354 case LTTNG_DOMAIN_UST:
3355 if (uatomic_read(&ust_consumerd_state) != CONSUMER_STARTED) {
f73fabfd 3356 ret = LTTNG_ERR_NO_USTCONSUMERD;
5c827ce0
DG
3357 goto error;
3358 }
3359 break;
3360 case LTTNG_DOMAIN_KERNEL:
3361 if (uatomic_read(&kernel_consumerd_state) != CONSUMER_STARTED) {
f73fabfd 3362 ret = LTTNG_ERR_NO_KERNCONSUMERD;
5c827ce0
DG
3363 goto error;
3364 }
3365 break;
825535cc
JG
3366 default:
3367 ret = LTTNG_ERR_UNKNOWN_DOMAIN;
3368 goto error;
5c827ce0
DG
3369 }
3370 }
3371
8e0af1b4
MD
3372 /*
3373 * Check that the UID or GID match that of the tracing session.
3374 * The root user can interact with all sessions.
3375 */
3376 if (need_tracing_session) {
3377 if (!session_access_ok(cmd_ctx->session,
730389d9
DG
3378 LTTNG_SOCK_GET_UID_CRED(&cmd_ctx->creds),
3379 LTTNG_SOCK_GET_GID_CRED(&cmd_ctx->creds))) {
f73fabfd 3380 ret = LTTNG_ERR_EPERM;
8e0af1b4
MD
3381 goto error;
3382 }
3383 }
3384
ffe60014
DG
3385 /*
3386 * Send relayd information to consumer as soon as we have a domain and a
3387 * session defined.
3388 */
3389 if (cmd_ctx->session && need_domain) {
3390 /*
3391 * Setup relayd if not done yet. If the relayd information was already
3392 * sent to the consumer, this call will gracefully return.
3393 */
3394 ret = cmd_setup_relayd(cmd_ctx->session);
3395 if (ret != LTTNG_OK) {
3396 goto error;
3397 }
3398 }
3399
54d01ffb
DG
3400 /* Process by command type */
3401 switch (cmd_ctx->lsm->cmd_type) {
3402 case LTTNG_ADD_CONTEXT:
3403 {
2001793c
JG
3404 /*
3405 * An LTTNG_ADD_CONTEXT command might have a supplementary
3406 * payload if the context being added is an application context.
3407 */
3408 if (cmd_ctx->lsm->u.context.ctx.ctx ==
3409 LTTNG_EVENT_CONTEXT_APP_CONTEXT) {
3410 char *provider_name = NULL, *context_name = NULL;
3411 size_t provider_name_len =
3412 cmd_ctx->lsm->u.context.provider_name_len;
3413 size_t context_name_len =
3414 cmd_ctx->lsm->u.context.context_name_len;
3415
3416 if (provider_name_len == 0 || context_name_len == 0) {
3417 /*
3418 * Application provider and context names MUST
3419 * be provided.
3420 */
3421 ret = -LTTNG_ERR_INVALID;
3422 goto error;
3423 }
3424
3425 provider_name = zmalloc(provider_name_len + 1);
3426 if (!provider_name) {
3427 ret = -LTTNG_ERR_NOMEM;
3428 goto error;
3429 }
e0f4d107
JG
3430 cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name =
3431 provider_name;
2001793c
JG
3432
3433 context_name = zmalloc(context_name_len + 1);
3434 if (!context_name) {
3435 ret = -LTTNG_ERR_NOMEM;
3436 goto error_add_context;
3437 }
e0f4d107
JG
3438 cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name =
3439 context_name;
2001793c
JG
3440
3441 ret = lttcomm_recv_unix_sock(sock, provider_name,
3442 provider_name_len);
3443 if (ret < 0) {
3444 goto error_add_context;
3445 }
3446
3447 ret = lttcomm_recv_unix_sock(sock, context_name,
3448 context_name_len);
3449 if (ret < 0) {
3450 goto error_add_context;
3451 }
2001793c
JG
3452 }
3453
3454 /*
3455 * cmd_add_context assumes ownership of the provider and context
3456 * names.
3457 */
3458 ret = cmd_add_context(cmd_ctx->session,
3459 cmd_ctx->lsm->domain.type,
54d01ffb 3460 cmd_ctx->lsm->u.context.channel_name,
2001793c
JG
3461 &cmd_ctx->lsm->u.context.ctx,
3462 kernel_poll_pipe[1]);
3463
3464 cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name = NULL;
3465 cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name = NULL;
3466error_add_context:
3467 free(cmd_ctx->lsm->u.context.ctx.u.app_ctx.provider_name);
3468 free(cmd_ctx->lsm->u.context.ctx.u.app_ctx.ctx_name);
3469 if (ret < 0) {
3470 goto error;
3471 }
54d01ffb
DG
3472 break;
3473 }
3474 case LTTNG_DISABLE_CHANNEL:
3475 {
3476 ret = cmd_disable_channel(cmd_ctx->session, cmd_ctx->lsm->domain.type,
3477 cmd_ctx->lsm->u.disable.channel_name);
3478 break;
3479 }
3480 case LTTNG_DISABLE_EVENT:
3481 {
47b7e2f9
JG
3482
3483 /*
3484 * FIXME: handle filter; for now we just receive the filter's
3485 * bytecode along with the filter expression which are sent by
3486 * liblttng-ctl and discard them.
3487 *
3488 * This fixes an issue where the client may block while sending
3489 * the filter payload and encounter an error because the session
3490 * daemon closes the socket without ever handling this data.
3491 */
3492 size_t count = cmd_ctx->lsm->u.disable.expression_len +
3493 cmd_ctx->lsm->u.disable.bytecode_len;
3494
3495 if (count) {
3496 char data[LTTNG_FILTER_MAX_LEN];
3497
3498 DBG("Discarding disable event command payload of size %zu", count);
3499 while (count) {
3500 ret = lttcomm_recv_unix_sock(sock, data,
3501 count > sizeof(data) ? sizeof(data) : count);
3502 if (ret < 0) {
3503 goto error;
3504 }
3505
3506 count -= (size_t) ret;
3507 }
3508 }
6e911cad 3509 /* FIXME: passing packed structure to non-packed pointer */
54d01ffb
DG
3510 ret = cmd_disable_event(cmd_ctx->session, cmd_ctx->lsm->domain.type,
3511 cmd_ctx->lsm->u.disable.channel_name,
6e911cad 3512 &cmd_ctx->lsm->u.disable.event);
33a2b854
DG
3513 break;
3514 }
54d01ffb
DG
3515 case LTTNG_ENABLE_CHANNEL:
3516 {
7972aab2 3517 ret = cmd_enable_channel(cmd_ctx->session, &cmd_ctx->lsm->domain,
2f77fc4b 3518 &cmd_ctx->lsm->u.channel.chan, kernel_poll_pipe[1]);
54d01ffb
DG
3519 break;
3520 }
ccf10263
MD
3521 case LTTNG_TRACK_PID:
3522 {
3523 ret = cmd_track_pid(cmd_ctx->session,
3524 cmd_ctx->lsm->domain.type,
3525 cmd_ctx->lsm->u.pid_tracker.pid);
3526 break;
3527 }
3528 case LTTNG_UNTRACK_PID:
3529 {
3530 ret = cmd_untrack_pid(cmd_ctx->session,
3531 cmd_ctx->lsm->domain.type,
3532 cmd_ctx->lsm->u.pid_tracker.pid);
3533 break;
3534 }
54d01ffb
DG
3535 case LTTNG_ENABLE_EVENT:
3536 {
882ef835
JI
3537 struct lttng_event_exclusion *exclusion = NULL;
3538 struct lttng_filter_bytecode *bytecode = NULL;
6b453b5e 3539 char *filter_expression = NULL;
882ef835 3540
67676bd8
DG
3541 /* Handle exclusion events and receive it from the client. */
3542 if (cmd_ctx->lsm->u.enable.exclusion_count > 0) {
3543 size_t count = cmd_ctx->lsm->u.enable.exclusion_count;
3544
3545 exclusion = zmalloc(sizeof(struct lttng_event_exclusion) +
3546 (count * LTTNG_SYMBOL_NAME_LEN));
3547 if (!exclusion) {
3548 ret = LTTNG_ERR_EXCLUSION_NOMEM;
3549 goto error;
882ef835 3550 }
882ef835 3551
67676bd8
DG
3552 DBG("Receiving var len exclusion event list from client ...");
3553 exclusion->count = count;
3554 ret = lttcomm_recv_unix_sock(sock, exclusion->names,
3555 count * LTTNG_SYMBOL_NAME_LEN);
3556 if (ret <= 0) {
3557 DBG("Nothing recv() from client var len data... continuing");
3558 *sock_error = 1;
3559 free(exclusion);
3560 ret = LTTNG_ERR_EXCLUSION_INVAL;
3561 goto error;
3562 }
3563 }
3564
6b453b5e
JG
3565 /* Get filter expression from client. */
3566 if (cmd_ctx->lsm->u.enable.expression_len > 0) {
3567 size_t expression_len =
3568 cmd_ctx->lsm->u.enable.expression_len;
3569
3570 if (expression_len > LTTNG_FILTER_MAX_LEN) {
3571 ret = LTTNG_ERR_FILTER_INVAL;
3572 free(exclusion);
3573 goto error;
3574 }
3575
3576 filter_expression = zmalloc(expression_len);
3577 if (!filter_expression) {
3578 free(exclusion);
3579 ret = LTTNG_ERR_FILTER_NOMEM;
3580 goto error;
3581 }
3582
3583 /* Receive var. len. data */
3584 DBG("Receiving var len filter's expression from client ...");
3585 ret = lttcomm_recv_unix_sock(sock, filter_expression,
3586 expression_len);
3587 if (ret <= 0) {
3588 DBG("Nothing recv() from client car len data... continuing");
3589 *sock_error = 1;
3590 free(filter_expression);
3591 free(exclusion);
3592 ret = LTTNG_ERR_FILTER_INVAL;
3593 goto error;
3594 }
3595 }
3596
67676bd8
DG
3597 /* Handle filter and get bytecode from client. */
3598 if (cmd_ctx->lsm->u.enable.bytecode_len > 0) {
3599 size_t bytecode_len = cmd_ctx->lsm->u.enable.bytecode_len;
3600
3601 if (bytecode_len > LTTNG_FILTER_MAX_LEN) {
3602 ret = LTTNG_ERR_FILTER_INVAL;
49d21f93 3603 free(filter_expression);
67676bd8
DG
3604 free(exclusion);
3605 goto error;
3606 }
3607
3608 bytecode = zmalloc(bytecode_len);
3609 if (!bytecode) {
49d21f93 3610 free(filter_expression);
67676bd8
DG
3611 free(exclusion);
3612 ret = LTTNG_ERR_FILTER_NOMEM;
3613 goto error;
882ef835
JI
3614 }
3615
67676bd8
DG
3616 /* Receive var. len. data */
3617 DBG("Receiving var len filter's bytecode from client ...");
3618 ret = lttcomm_recv_unix_sock(sock, bytecode, bytecode_len);
3619 if (ret <= 0) {
3620 DBG("Nothing recv() from client car len data... continuing");
3621 *sock_error = 1;
49d21f93 3622 free(filter_expression);
67676bd8
DG
3623 free(bytecode);
3624 free(exclusion);
3625 ret = LTTNG_ERR_FILTER_INVAL;
3626 goto error;
3627 }
3628
3629 if ((bytecode->len + sizeof(*bytecode)) != bytecode_len) {
49d21f93 3630 free(filter_expression);
67676bd8
DG
3631 free(bytecode);
3632 free(exclusion);
3633 ret = LTTNG_ERR_FILTER_INVAL;
3634 goto error;
3635 }
882ef835 3636 }
67676bd8
DG
3637
3638 ret = cmd_enable_event(cmd_ctx->session, &cmd_ctx->lsm->domain,
3639 cmd_ctx->lsm->u.enable.channel_name,
6b453b5e
JG
3640 &cmd_ctx->lsm->u.enable.event,
3641 filter_expression, bytecode, exclusion,
67676bd8 3642 kernel_poll_pipe[1]);
54d01ffb
DG
3643 break;
3644 }
052da939 3645 case LTTNG_LIST_TRACEPOINTS:
2ef84c95 3646 {
9f19cc17 3647 struct lttng_event *events;
54d01ffb 3648 ssize_t nb_events;
052da939 3649
0845bbfa 3650 session_lock_list();
54d01ffb 3651 nb_events = cmd_list_tracepoints(cmd_ctx->lsm->domain.type, &events);
0845bbfa 3652 session_unlock_list();
54d01ffb 3653 if (nb_events < 0) {
f73fabfd 3654 /* Return value is a negative lttng_error_code. */
54d01ffb
DG
3655 ret = -nb_events;
3656 goto error;
2ef84c95
DG
3657 }
3658
3659 /*
3660 * Setup lttng message with payload size set to the event list size in
3661 * bytes and then copy list into the llm payload.
3662 */
6e10c9b9
PP
3663 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, events,
3664 sizeof(struct lttng_event) * nb_events);
3665 free(events);
3666
2ef84c95
DG
3667 if (ret < 0) {
3668 goto setup_error;
3669 }
3670
f73fabfd 3671 ret = LTTNG_OK;
2ef84c95
DG
3672 break;
3673 }
f37d259d
MD
3674 case LTTNG_LIST_TRACEPOINT_FIELDS:
3675 {
3676 struct lttng_event_field *fields;
3677 ssize_t nb_fields;
3678
0845bbfa 3679 session_lock_list();
a4b92340
DG
3680 nb_fields = cmd_list_tracepoint_fields(cmd_ctx->lsm->domain.type,
3681 &fields);
0845bbfa 3682 session_unlock_list();
f37d259d 3683 if (nb_fields < 0) {
f73fabfd 3684 /* Return value is a negative lttng_error_code. */
f37d259d
MD
3685 ret = -nb_fields;
3686 goto error;
3687 }
3688
3689 /*
3690 * Setup lttng message with payload size set to the event list size in
3691 * bytes and then copy list into the llm payload.
3692 */
6e10c9b9 3693 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, fields,
a4b92340 3694 sizeof(struct lttng_event_field) * nb_fields);
6e10c9b9
PP
3695 free(fields);
3696
f37d259d 3697 if (ret < 0) {
f37d259d
MD
3698 goto setup_error;
3699 }
3700
f73fabfd 3701 ret = LTTNG_OK;
f37d259d
MD
3702 break;
3703 }
834978fd
DG
3704 case LTTNG_LIST_SYSCALLS:
3705 {
3706 struct lttng_event *events;
3707 ssize_t nb_events;
3708
3709 nb_events = cmd_list_syscalls(&events);
3710 if (nb_events < 0) {
3711 /* Return value is a negative lttng_error_code. */
3712 ret = -nb_events;
3713 goto error;
3714 }
3715
3716 /*
3717 * Setup lttng message with payload size set to the event list size in
3718 * bytes and then copy list into the llm payload.
3719 */
6e10c9b9
PP
3720 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, events,
3721 sizeof(struct lttng_event) * nb_events);
3722 free(events);
3723
834978fd 3724 if (ret < 0) {
834978fd
DG
3725 goto setup_error;
3726 }
3727
834978fd
DG
3728 ret = LTTNG_OK;
3729 break;
3730 }
a5dfbb9d
MD
3731 case LTTNG_LIST_TRACKER_PIDS:
3732 {
3733 int32_t *pids = NULL;
3734 ssize_t nr_pids;
3735
3736 nr_pids = cmd_list_tracker_pids(cmd_ctx->session,
3737 cmd_ctx->lsm->domain.type, &pids);
3738 if (nr_pids < 0) {
3739 /* Return value is a negative lttng_error_code. */
3740 ret = -nr_pids;
3741 goto error;
3742 }
3743
3744 /*
3745 * Setup lttng message with payload size set to the event list size in
3746 * bytes and then copy list into the llm payload.
3747 */
6e10c9b9
PP
3748 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, pids,
3749 sizeof(int32_t) * nr_pids);
3750 free(pids);
3751
a5dfbb9d 3752 if (ret < 0) {
a5dfbb9d
MD
3753 goto setup_error;
3754 }
3755
a5dfbb9d
MD
3756 ret = LTTNG_OK;
3757 break;
3758 }
00e2e675
DG
3759 case LTTNG_SET_CONSUMER_URI:
3760 {
a4b92340
DG
3761 size_t nb_uri, len;
3762 struct lttng_uri *uris;
3763
3764 nb_uri = cmd_ctx->lsm->u.uri.size;
3765 len = nb_uri * sizeof(struct lttng_uri);
3766
3767 if (nb_uri == 0) {
f73fabfd 3768 ret = LTTNG_ERR_INVALID;
a4b92340
DG
3769 goto error;
3770 }
3771
3772 uris = zmalloc(len);
3773 if (uris == NULL) {
f73fabfd 3774 ret = LTTNG_ERR_FATAL;
a4b92340
DG
3775 goto error;
3776 }
3777
3778 /* Receive variable len data */
77c7c900 3779 DBG("Receiving %zu URI(s) from client ...", nb_uri);
a4b92340
DG
3780 ret = lttcomm_recv_unix_sock(sock, uris, len);
3781 if (ret <= 0) {
3782 DBG("No URIs received from client... continuing");
3783 *sock_error = 1;
f73fabfd 3784 ret = LTTNG_ERR_SESSION_FAIL;
b1d41407 3785 free(uris);
a4b92340
DG
3786 goto error;
3787 }
3788
bda32d56
JG
3789 ret = cmd_set_consumer_uri(cmd_ctx->session, nb_uri, uris);
3790 free(uris);
f73fabfd 3791 if (ret != LTTNG_OK) {
a4b92340
DG
3792 goto error;
3793 }
3794
b1d41407 3795
00e2e675
DG
3796 break;
3797 }
f3ed775e 3798 case LTTNG_START_TRACE:
8c0faa1d 3799 {
54d01ffb 3800 ret = cmd_start_trace(cmd_ctx->session);
8c0faa1d
DG
3801 break;
3802 }
f3ed775e 3803 case LTTNG_STOP_TRACE:
8c0faa1d 3804 {
54d01ffb 3805 ret = cmd_stop_trace(cmd_ctx->session);
8c0faa1d
DG
3806 break;
3807 }
5e16da05
MD
3808 case LTTNG_CREATE_SESSION:
3809 {
a4b92340
DG
3810 size_t nb_uri, len;
3811 struct lttng_uri *uris = NULL;
3812
3813 nb_uri = cmd_ctx->lsm->u.uri.size;
3814 len = nb_uri * sizeof(struct lttng_uri);
3815
3816 if (nb_uri > 0) {
3817 uris = zmalloc(len);
3818 if (uris == NULL) {
f73fabfd 3819 ret = LTTNG_ERR_FATAL;
a4b92340
DG
3820 goto error;
3821 }
3822
3823 /* Receive variable len data */
77c7c900 3824 DBG("Waiting for %zu URIs from client ...", nb_uri);
a4b92340
DG
3825 ret = lttcomm_recv_unix_sock(sock, uris, len);
3826 if (ret <= 0) {
3827 DBG("No URIs received from client... continuing");
3828 *sock_error = 1;
f73fabfd 3829 ret = LTTNG_ERR_SESSION_FAIL;
b1d41407 3830 free(uris);
a4b92340
DG
3831 goto error;
3832 }
3833
3834 if (nb_uri == 1 && uris[0].dtype != LTTNG_DST_PATH) {
3835 DBG("Creating session with ONE network URI is a bad call");
f73fabfd 3836 ret = LTTNG_ERR_SESSION_FAIL;
b1d41407 3837 free(uris);
a4b92340
DG
3838 goto error;
3839 }
3840 }
3841
3842 ret = cmd_create_session_uri(cmd_ctx->lsm->session.name, uris, nb_uri,
ecc48a90 3843 &cmd_ctx->creds, 0);
a4b92340 3844
b1d41407
DG
3845 free(uris);
3846
00e2e675
DG
3847 break;
3848 }
5e16da05
MD
3849 case LTTNG_DESTROY_SESSION:
3850 {
2f77fc4b 3851 ret = cmd_destroy_session(cmd_ctx->session, kernel_poll_pipe[1]);
a4b92340
DG
3852
3853 /* Set session to NULL so we do not unlock it after free. */
256a5576 3854 cmd_ctx->session = NULL;
5461b305 3855 break;
5e16da05 3856 }
9f19cc17 3857 case LTTNG_LIST_DOMAINS:
5e16da05 3858 {
54d01ffb 3859 ssize_t nb_dom;
fa64dfb4 3860 struct lttng_domain *domains = NULL;
5461b305 3861
54d01ffb
DG
3862 nb_dom = cmd_list_domains(cmd_ctx->session, &domains);
3863 if (nb_dom < 0) {
f73fabfd 3864 /* Return value is a negative lttng_error_code. */
54d01ffb
DG
3865 ret = -nb_dom;
3866 goto error;
ce3d728c 3867 }
520ff687 3868
6e10c9b9
PP
3869 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, domains,
3870 nb_dom * sizeof(struct lttng_domain));
3871 free(domains);
3872
5461b305
DG
3873 if (ret < 0) {
3874 goto setup_error;
520ff687 3875 }
57167058 3876
f73fabfd 3877 ret = LTTNG_OK;
5e16da05
MD
3878 break;
3879 }
9f19cc17 3880 case LTTNG_LIST_CHANNELS:
5e16da05 3881 {
53e367f9 3882 ssize_t payload_size;
ade7ce52 3883 struct lttng_channel *channels = NULL;
54d01ffb 3884
53e367f9 3885 payload_size = cmd_list_channels(cmd_ctx->lsm->domain.type,
b551a063 3886 cmd_ctx->session, &channels);
53e367f9 3887 if (payload_size < 0) {
f73fabfd 3888 /* Return value is a negative lttng_error_code. */
53e367f9 3889 ret = -payload_size;
54d01ffb 3890 goto error;
5461b305 3891 }
ca95a216 3892
6e10c9b9 3893 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, channels,
53e367f9 3894 payload_size);
6e10c9b9
PP
3895 free(channels);
3896
5461b305
DG
3897 if (ret < 0) {
3898 goto setup_error;
3899 }
9f19cc17 3900
f73fabfd 3901 ret = LTTNG_OK;
5461b305 3902 break;
5e16da05 3903 }
9f19cc17 3904 case LTTNG_LIST_EVENTS:
5e16da05 3905 {
b551a063 3906 ssize_t nb_event;
684d34d2 3907 struct lttng_event *events = NULL;
b4e3ceb9
PP
3908 struct lttcomm_event_command_header cmd_header;
3909 size_t total_size;
3910
b2d66287 3911 memset(&cmd_header, 0, sizeof(cmd_header));
b4e3ceb9
PP
3912 /* Extended infos are included at the end of events */
3913 nb_event = cmd_list_events(cmd_ctx->lsm->domain.type,
3914 cmd_ctx->session, cmd_ctx->lsm->u.list.channel_name,
3915 &events, &total_size);
9f19cc17 3916
54d01ffb 3917 if (nb_event < 0) {
f73fabfd 3918 /* Return value is a negative lttng_error_code. */
54d01ffb
DG
3919 ret = -nb_event;
3920 goto error;
5461b305 3921 }
ca95a216 3922
b4e3ceb9
PP
3923 cmd_header.nb_events = nb_event;
3924 ret = setup_lttng_msg(cmd_ctx, events, total_size,
3925 &cmd_header, sizeof(cmd_header));
6e10c9b9
PP
3926 free(events);
3927
5461b305
DG
3928 if (ret < 0) {
3929 goto setup_error;
3930 }
9f19cc17 3931
f73fabfd 3932 ret = LTTNG_OK;
5461b305 3933 break;
5e16da05
MD
3934 }
3935 case LTTNG_LIST_SESSIONS:
3936 {
8e0af1b4 3937 unsigned int nr_sessions;
6e10c9b9
PP
3938 void *sessions_payload;
3939 size_t payload_len;
5461b305 3940
8e0af1b4 3941 session_lock_list();
730389d9
DG
3942 nr_sessions = lttng_sessions_count(
3943 LTTNG_SOCK_GET_UID_CRED(&cmd_ctx->creds),
3944 LTTNG_SOCK_GET_GID_CRED(&cmd_ctx->creds));
6e10c9b9
PP
3945 payload_len = sizeof(struct lttng_session) * nr_sessions;
3946 sessions_payload = zmalloc(payload_len);
d32fb093 3947
6e10c9b9 3948 if (!sessions_payload) {
54d01ffb 3949 session_unlock_list();
6e10c9b9 3950 ret = -ENOMEM;
5461b305 3951 goto setup_error;
e065084a 3952 }
5e16da05 3953
6e10c9b9 3954 cmd_list_lttng_sessions(sessions_payload,
730389d9
DG
3955 LTTNG_SOCK_GET_UID_CRED(&cmd_ctx->creds),
3956 LTTNG_SOCK_GET_GID_CRED(&cmd_ctx->creds));
54d01ffb 3957 session_unlock_list();
5e16da05 3958
6e10c9b9
PP
3959 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, sessions_payload,
3960 payload_len);
3961 free(sessions_payload);
3962
3963 if (ret < 0) {
3964 goto setup_error;
3965 }
3966
f73fabfd 3967 ret = LTTNG_OK;
5e16da05
MD
3968 break;
3969 }
d9800920
DG
3970 case LTTNG_REGISTER_CONSUMER:
3971 {
2f77fc4b
DG
3972 struct consumer_data *cdata;
3973
3974 switch (cmd_ctx->lsm->domain.type) {
3975 case LTTNG_DOMAIN_KERNEL:
3976 cdata = &kconsumer_data;
3977 break;
3978 default:
f73fabfd 3979 ret = LTTNG_ERR_UND;
2f77fc4b
DG
3980 goto error;
3981 }
3982
54d01ffb 3983 ret = cmd_register_consumer(cmd_ctx->session, cmd_ctx->lsm->domain.type,
2f77fc4b 3984 cmd_ctx->lsm->u.reg.path, cdata);
d9800920
DG
3985 break;
3986 }
6d805429 3987 case LTTNG_DATA_PENDING:
806e2684 3988 {
f6151c55 3989 int pending_ret;
6e10c9b9 3990 uint8_t pending_ret_byte;
f6151c55
JG
3991
3992 pending_ret = cmd_data_pending(cmd_ctx->session);
6e10c9b9 3993
f6151c55
JG
3994 /*
3995 * FIXME
3996 *
3997 * This function may returns 0 or 1 to indicate whether or not
3998 * there is data pending. In case of error, it should return an
3999 * LTTNG_ERR code. However, some code paths may still return
4000 * a nondescript error code, which we handle by returning an
4001 * "unknown" error.
4002 */
4003 if (pending_ret == 0 || pending_ret == 1) {
6e10c9b9
PP
4004 /*
4005 * ret will be set to LTTNG_OK at the end of
4006 * this function.
4007 */
f6151c55
JG
4008 } else if (pending_ret < 0) {
4009 ret = LTTNG_ERR_UNK;
4010 goto setup_error;
4011 } else {
4012 ret = pending_ret;
4013 goto setup_error;
4014 }
4015
6e10c9b9
PP
4016 pending_ret_byte = (uint8_t) pending_ret;
4017
4018 /* 1 byte to return whether or not data is pending */
4019 ret = setup_lttng_msg_no_cmd_header(cmd_ctx,
4020 &pending_ret_byte, 1);
4021
4022 if (ret < 0) {
4023 goto setup_error;
4024 }
4025
4026 ret = LTTNG_OK;
806e2684
DG
4027 break;
4028 }
da3c9ec1
DG
4029 case LTTNG_SNAPSHOT_ADD_OUTPUT:
4030 {
6dc3064a
DG
4031 struct lttcomm_lttng_output_id reply;
4032
4033 ret = cmd_snapshot_add_output(cmd_ctx->session,
4034 &cmd_ctx->lsm->u.snapshot_output.output, &reply.id);
4035 if (ret != LTTNG_OK) {
4036 goto error;
4037 }
4038
6e10c9b9
PP
4039 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, &reply,
4040 sizeof(reply));
6dc3064a
DG
4041 if (ret < 0) {
4042 goto setup_error;
4043 }
4044
4045 /* Copy output list into message payload */
6dc3064a 4046 ret = LTTNG_OK;
da3c9ec1
DG
4047 break;
4048 }
4049 case LTTNG_SNAPSHOT_DEL_OUTPUT:
4050 {
6dc3064a
DG
4051 ret = cmd_snapshot_del_output(cmd_ctx->session,
4052 &cmd_ctx->lsm->u.snapshot_output.output);
da3c9ec1
DG
4053 break;
4054 }
4055 case LTTNG_SNAPSHOT_LIST_OUTPUT:
4056 {
6dc3064a
DG
4057 ssize_t nb_output;
4058 struct lttng_snapshot_output *outputs = NULL;
4059
4060 nb_output = cmd_snapshot_list_outputs(cmd_ctx->session, &outputs);
4061 if (nb_output < 0) {
4062 ret = -nb_output;
4063 goto error;
4064 }
4065
6e10c9b9
PP
4066 assert((nb_output > 0 && outputs) || nb_output == 0);
4067 ret = setup_lttng_msg_no_cmd_header(cmd_ctx, outputs,
6dc3064a 4068 nb_output * sizeof(struct lttng_snapshot_output));
6e10c9b9
PP
4069 free(outputs);
4070
6dc3064a 4071 if (ret < 0) {
6dc3064a
DG
4072 goto setup_error;
4073 }
4074
6dc3064a 4075 ret = LTTNG_OK;
da3c9ec1
DG
4076 break;
4077 }
4078 case LTTNG_SNAPSHOT_RECORD:
4079 {
6dc3064a
DG
4080 ret = cmd_snapshot_record(cmd_ctx->session,
4081 &cmd_ctx->lsm->u.snapshot_record.output,
4082 cmd_ctx->lsm->u.snapshot_record.wait);
da3c9ec1
DG
4083 break;
4084 }
27babd3a
DG
4085 case LTTNG_CREATE_SESSION_SNAPSHOT:
4086 {
4087 size_t nb_uri, len;
4088 struct lttng_uri *uris = NULL;
4089
4090 nb_uri = cmd_ctx->lsm->u.uri.size;
4091 len = nb_uri * sizeof(struct lttng_uri);
4092
4093 if (nb_uri > 0) {
4094 uris = zmalloc(len);
4095 if (uris == NULL) {
4096 ret = LTTNG_ERR_FATAL;
4097 goto error;
4098 }
4099
4100 /* Receive variable len data */
4101 DBG("Waiting for %zu URIs from client ...", nb_uri);
4102 ret = lttcomm_recv_unix_sock(sock, uris, len);
4103 if (ret <= 0) {
4104 DBG("No URIs received from client... continuing");
4105 *sock_error = 1;
4106 ret = LTTNG_ERR_SESSION_FAIL;
4107 free(uris);
4108 goto error;
4109 }
4110
4111 if (nb_uri == 1 && uris[0].dtype != LTTNG_DST_PATH) {
4112 DBG("Creating session with ONE network URI is a bad call");
4113 ret = LTTNG_ERR_SESSION_FAIL;
4114 free(uris);
4115 goto error;
4116 }
4117 }
4118
4119 ret = cmd_create_session_snapshot(cmd_ctx->lsm->session.name, uris,
4120 nb_uri, &cmd_ctx->creds);
4121 free(uris);
4122 break;
4123 }
ecc48a90
JD
4124 case LTTNG_CREATE_SESSION_LIVE:
4125 {
4126 size_t nb_uri, len;
4127 struct lttng_uri *uris = NULL;
4128
4129 nb_uri = cmd_ctx->lsm->u.uri.size;
4130 len = nb_uri * sizeof(struct lttng_uri);
4131
4132 if (nb_uri > 0) {
4133 uris = zmalloc(len);
4134 if (uris == NULL) {
4135 ret = LTTNG_ERR_FATAL;
4136 goto error;
4137 }
4138
4139 /* Receive variable len data */
4140 DBG("Waiting for %zu URIs from client ...", nb_uri);
4141 ret = lttcomm_recv_unix_sock(sock, uris, len);
4142 if (ret <= 0) {
4143 DBG("No URIs received from client... continuing");
4144 *sock_error = 1;
4145 ret = LTTNG_ERR_SESSION_FAIL;
4146 free(uris);
4147 goto error;
4148 }
4149
4150 if (nb_uri == 1 && uris[0].dtype != LTTNG_DST_PATH) {
4151 DBG("Creating session with ONE network URI is a bad call");
4152 ret = LTTNG_ERR_SESSION_FAIL;
4153 free(uris);
4154 goto error;
4155 }
4156 }
4157
4158 ret = cmd_create_session_uri(cmd_ctx->lsm->session.name, uris,
4159 nb_uri, &cmd_ctx->creds, cmd_ctx->lsm->u.session_live.timer_interval);
4160 free(uris);
4161 break;
4162 }
fb198a11
JG
4163 case LTTNG_SAVE_SESSION:
4164 {
4165 ret = cmd_save_sessions(&cmd_ctx->lsm->u.save_session.attr,
4166 &cmd_ctx->creds);
4167 break;
4168 }
d7ba1388
MD
4169 case LTTNG_SET_SESSION_SHM_PATH:
4170 {
4171 ret = cmd_set_session_shm_path(cmd_ctx->session,
4172 cmd_ctx->lsm->u.set_shm_path.shm_path);
4173 break;
4174 }
eded6438 4175 case LTTNG_REGENERATE_METADATA:
93ec662e 4176 {
eded6438 4177 ret = cmd_regenerate_metadata(cmd_ctx->session);
93ec662e
JD
4178 break;
4179 }
c2561365
JD
4180 case LTTNG_REGENERATE_STATEDUMP:
4181 {
4182 ret = cmd_regenerate_statedump(cmd_ctx->session);
4183 break;
4184 }
daa81f58
JG
4185 case LTTNG_REGISTER_TRIGGER:
4186 {
4187 ret = cmd_register_trigger(cmd_ctx, sock,
4188 notification_thread_handle);
4189 break;
4190 }
4191 case LTTNG_UNREGISTER_TRIGGER:
4192 {
4193 ret = cmd_unregister_trigger(cmd_ctx, sock,
4194 notification_thread_handle);
4195 break;
4196 }
5e16da05 4197 default:
f73fabfd 4198 ret = LTTNG_ERR_UND;
5461b305 4199 break;
fac6795d
DG
4200 }
4201
5461b305 4202error:
5461b305
DG
4203 if (cmd_ctx->llm == NULL) {
4204 DBG("Missing llm structure. Allocating one.");
6e10c9b9 4205 if (setup_lttng_msg_no_cmd_header(cmd_ctx, NULL, 0) < 0) {
5461b305
DG
4206 goto setup_error;
4207 }
4208 }
54d01ffb 4209 /* Set return code */
5461b305 4210 cmd_ctx->llm->ret_code = ret;
5461b305 4211setup_error:
b5541356 4212 if (cmd_ctx->session) {
54d01ffb 4213 session_unlock(cmd_ctx->session);
b5541356 4214 }
256a5576
MD
4215 if (need_tracing_session) {
4216 session_unlock_list();
4217 }
54d01ffb 4218init_setup_error:
3745d315 4219 assert(!rcu_read_ongoing());
8028d920 4220 return ret;
fac6795d
DG
4221}
4222
44a5e5eb
DG
4223/*
4224 * Thread managing health check socket.
4225 */
4226static void *thread_manage_health(void *data)
4227{
eb4a2943 4228 int sock = -1, new_sock = -1, ret, i, pollfd, err = -1;
44a5e5eb
DG
4229 uint32_t revents, nb_fd;
4230 struct lttng_poll_event events;
0c89d795
MD
4231 struct health_comm_msg msg;
4232 struct health_comm_reply reply;
44a5e5eb
DG
4233
4234 DBG("[thread] Manage health check started");
4235
4236 rcu_register_thread();
4237
6d737ce4
DG
4238 /* We might hit an error path before this is created. */
4239 lttng_poll_init(&events);
3cc04881 4240
44a5e5eb
DG
4241 /* Create unix socket */
4242 sock = lttcomm_create_unix_sock(health_unix_sock_path);
4243 if (sock < 0) {
4244 ERR("Unable to create health check Unix socket");
44a5e5eb
DG
4245 goto error;
4246 }
4247
6c71277b
MD
4248 if (is_root) {
4249 /* lttng health client socket path permissions */
4250 ret = chown(health_unix_sock_path, 0,
4251 utils_get_group_id(tracing_group_name));
4252 if (ret < 0) {
4253 ERR("Unable to set group on %s", health_unix_sock_path);
4254 PERROR("chown");
6c71277b
MD
4255 goto error;
4256 }
4257
4258 ret = chmod(health_unix_sock_path,
4259 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
4260 if (ret < 0) {
4261 ERR("Unable to set permissions on %s", health_unix_sock_path);
4262 PERROR("chmod");
6c71277b
MD
4263 goto error;
4264 }
4265 }
4266
b662582b
DG
4267 /*
4268 * Set the CLOEXEC flag. Return code is useless because either way, the
4269 * show must go on.
4270 */
4271 (void) utils_set_fd_cloexec(sock);
4272
44a5e5eb
DG
4273 ret = lttcomm_listen_unix_sock(sock);
4274 if (ret < 0) {
4275 goto error;
4276 }
4277
4278 /*
4279 * Pass 2 as size here for the thread quit pipe and client_sock. Nothing
4280 * more will be added to this poll set.
4281 */
d0b96690 4282 ret = sessiond_set_thread_pollset(&events, 2);
44a5e5eb
DG
4283 if (ret < 0) {
4284 goto error;
4285 }
4286
4287 /* Add the application registration socket */
4288 ret = lttng_poll_add(&events, sock, LPOLLIN | LPOLLPRI);
4289 if (ret < 0) {
4290 goto error;
4291 }
4292
ef367a93 4293 sessiond_notify_ready();
97bc1426 4294
44a5e5eb
DG
4295 while (1) {
4296 DBG("Health check ready");
4297
44a5e5eb
DG
4298 /* Inifinite blocking call, waiting for transmission */
4299restart:
4300 ret = lttng_poll_wait(&events, -1);
4301 if (ret < 0) {
4302 /*
4303 * Restart interrupted system call.
4304 */
4305 if (errno == EINTR) {
4306 goto restart;
4307 }
4308 goto error;
4309 }
4310
0d9c5d77
DG
4311 nb_fd = ret;
4312
44a5e5eb
DG
4313 for (i = 0; i < nb_fd; i++) {
4314 /* Fetch once the poll data */
4315 revents = LTTNG_POLL_GETEV(&events, i);
4316 pollfd = LTTNG_POLL_GETFD(&events, i);
4317
fd20dac9
MD
4318 if (!revents) {
4319 /* No activity for this FD (poll implementation). */
4320 continue;
4321 }
4322
44a5e5eb 4323 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 4324 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
44a5e5eb 4325 if (ret) {
139ac872
MD
4326 err = 0;
4327 goto exit;
44a5e5eb
DG
4328 }
4329
4330 /* Event on the registration socket */
4331 if (pollfd == sock) {
03e43155
MD
4332 if (revents & LPOLLIN) {
4333 continue;
4334 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
44a5e5eb
DG
4335 ERR("Health socket poll error");
4336 goto error;
03e43155
MD
4337 } else {
4338 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
4339 goto error;
44a5e5eb
DG
4340 }
4341 }
4342 }
4343
4344 new_sock = lttcomm_accept_unix_sock(sock);
4345 if (new_sock < 0) {
4346 goto error;
4347 }
4348
b662582b
DG
4349 /*
4350 * Set the CLOEXEC flag. Return code is useless because either way, the
4351 * show must go on.
4352 */
4353 (void) utils_set_fd_cloexec(new_sock);
4354
44a5e5eb
DG
4355 DBG("Receiving data from client for health...");
4356 ret = lttcomm_recv_unix_sock(new_sock, (void *)&msg, sizeof(msg));
4357 if (ret <= 0) {
4358 DBG("Nothing recv() from client... continuing");
4359 ret = close(new_sock);
4360 if (ret) {
4361 PERROR("close");
4362 }
44a5e5eb
DG
4363 continue;
4364 }
4365
4366 rcu_thread_online();
4367
53efb85a 4368 memset(&reply, 0, sizeof(reply));
6c71277b
MD
4369 for (i = 0; i < NR_HEALTH_SESSIOND_TYPES; i++) {
4370 /*
4371 * health_check_state returns 0 if health is
4372 * bad.
4373 */
4374 if (!health_check_state(health_sessiond, i)) {
4375 reply.ret_code |= 1ULL << i;
4376 }
44a5e5eb
DG
4377 }
4378
6c71277b 4379 DBG2("Health check return value %" PRIx64, reply.ret_code);
44a5e5eb
DG
4380
4381 ret = send_unix_sock(new_sock, (void *) &reply, sizeof(reply));
4382 if (ret < 0) {
4383 ERR("Failed to send health data back to client");
4384 }
4385
4386 /* End of transmission */
4387 ret = close(new_sock);
4388 if (ret) {
4389 PERROR("close");
4390 }
44a5e5eb
DG
4391 }
4392
139ac872 4393exit:
44a5e5eb 4394error:
139ac872
MD
4395 if (err) {
4396 ERR("Health error occurred in %s", __func__);
4397 }
44a5e5eb
DG
4398 DBG("Health check thread dying");
4399 unlink(health_unix_sock_path);
4400 if (sock >= 0) {
4401 ret = close(sock);
4402 if (ret) {
4403 PERROR("close");
4404 }
4405 }
44a5e5eb
DG
4406
4407 lttng_poll_clean(&events);
b11d6a7f 4408 stop_threads();
44a5e5eb
DG
4409 rcu_unregister_thread();
4410 return NULL;
4411}
4412
1d4b027a 4413/*
d063d709
DG
4414 * This thread manage all clients request using the unix client socket for
4415 * communication.
1d4b027a
DG
4416 */
4417static void *thread_manage_clients(void *data)
4418{
139ac872 4419 int sock = -1, ret, i, pollfd, err = -1;
53a80697 4420 int sock_error;
5eb91c98 4421 uint32_t revents, nb_fd;
273ea72c 4422 struct command_ctx *cmd_ctx = NULL;
5eb91c98 4423 struct lttng_poll_event events;
1d4b027a
DG
4424
4425 DBG("[thread] Manage client started");
4426
f6a9efaa
DG
4427 rcu_register_thread();
4428
6c71277b 4429 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_CMD);
927ca06a 4430
840cb59c 4431 health_code_update();
44a5e5eb 4432
1d4b027a
DG
4433 ret = lttcomm_listen_unix_sock(client_sock);
4434 if (ret < 0) {
35df2755 4435 goto error_listen;
1d4b027a
DG
4436 }
4437
5eb91c98
DG
4438 /*
4439 * Pass 2 as size here for the thread quit pipe and client_sock. Nothing
4440 * more will be added to this poll set.
4441 */
d0b96690 4442 ret = sessiond_set_thread_pollset(&events, 2);
5eb91c98 4443 if (ret < 0) {
35df2755 4444 goto error_create_poll;
5eb91c98 4445 }
273ea72c 4446
5eb91c98
DG
4447 /* Add the application registration socket */
4448 ret = lttng_poll_add(&events, client_sock, LPOLLIN | LPOLLPRI);
4449 if (ret < 0) {
4450 goto error;
4451 }
273ea72c 4452
ef367a93
JG
4453 sessiond_notify_ready();
4454 ret = sem_post(&load_info->message_thread_ready);
4455 if (ret) {
4456 PERROR("sem_post message_thread_ready");
4457 goto error;
4458 }
0bb7724a 4459
72453c8c
MD
4460 /* This testpoint is after we signal readiness to the parent. */
4461 if (testpoint(sessiond_thread_manage_clients)) {
4462 goto error;
4463 }
4464
e547b070 4465 if (testpoint(sessiond_thread_manage_clients_before_loop)) {
6993eeb3
CB
4466 goto error;
4467 }
8ac94142 4468
840cb59c 4469 health_code_update();
44a5e5eb 4470
1d4b027a 4471 while (1) {
1d4b027a 4472 DBG("Accepting client command ...");
273ea72c
DG
4473
4474 /* Inifinite blocking call, waiting for transmission */
88f2b785 4475 restart:
a78af745 4476 health_poll_entry();
5eb91c98 4477 ret = lttng_poll_wait(&events, -1);
a78af745 4478 health_poll_exit();
273ea72c 4479 if (ret < 0) {
88f2b785
MD
4480 /*
4481 * Restart interrupted system call.
4482 */
4483 if (errno == EINTR) {
4484 goto restart;
4485 }
273ea72c
DG
4486 goto error;
4487 }
4488
0d9c5d77
DG
4489 nb_fd = ret;
4490
5eb91c98
DG
4491 for (i = 0; i < nb_fd; i++) {
4492 /* Fetch once the poll data */
4493 revents = LTTNG_POLL_GETEV(&events, i);
4494 pollfd = LTTNG_POLL_GETFD(&events, i);
4495
840cb59c 4496 health_code_update();
44a5e5eb 4497
fd20dac9
MD
4498 if (!revents) {
4499 /* No activity for this FD (poll implementation). */
4500 continue;
4501 }
4502
5eb91c98 4503 /* Thread quit pipe has been closed. Killing thread. */
d0b96690 4504 ret = sessiond_check_thread_quit_pipe(pollfd, revents);
5eb91c98 4505 if (ret) {
139ac872
MD
4506 err = 0;
4507 goto exit;
5eb91c98
DG
4508 }
4509
4510 /* Event on the registration socket */
4511 if (pollfd == client_sock) {
03e43155
MD
4512 if (revents & LPOLLIN) {
4513 continue;
4514 } else if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
5eb91c98
DG
4515 ERR("Client socket poll error");
4516 goto error;
03e43155
MD
4517 } else {
4518 ERR("Unexpected poll events %u for sock %d", revents, pollfd);
4519 goto error;
5eb91c98
DG
4520 }
4521 }
4522 }
4523
4524 DBG("Wait for client response");
4525
840cb59c 4526 health_code_update();
44a5e5eb 4527
5eb91c98
DG
4528 sock = lttcomm_accept_unix_sock(client_sock);
4529 if (sock < 0) {
273ea72c 4530 goto error;
273ea72c
DG
4531 }
4532
b662582b
DG
4533 /*
4534 * Set the CLOEXEC flag. Return code is useless because either way, the
4535 * show must go on.
4536 */
4537 (void) utils_set_fd_cloexec(sock);
4538
be040666
DG
4539 /* Set socket option for credentials retrieval */
4540 ret = lttcomm_setsockopt_creds_unix_sock(sock);
4541 if (ret < 0) {
4542 goto error;
4543 }
4544
5eb91c98 4545 /* Allocate context command to process the client request */
ba7f0ae5 4546 cmd_ctx = zmalloc(sizeof(struct command_ctx));
5eb91c98 4547 if (cmd_ctx == NULL) {
76d7553f 4548 PERROR("zmalloc cmd_ctx");
1d4b027a 4549 goto error;
5eb91c98 4550 }
1d4b027a 4551
5eb91c98 4552 /* Allocate data buffer for reception */
ba7f0ae5 4553 cmd_ctx->lsm = zmalloc(sizeof(struct lttcomm_session_msg));
5eb91c98 4554 if (cmd_ctx->lsm == NULL) {
76d7553f 4555 PERROR("zmalloc cmd_ctx->lsm");
5eb91c98
DG
4556 goto error;
4557 }
1d4b027a 4558
5eb91c98
DG
4559 cmd_ctx->llm = NULL;
4560 cmd_ctx->session = NULL;
1d4b027a 4561
840cb59c 4562 health_code_update();
44a5e5eb 4563
5eb91c98
DG
4564 /*
4565 * Data is received from the lttng client. The struct
4566 * lttcomm_session_msg (lsm) contains the command and data request of
4567 * the client.
4568 */
4569 DBG("Receiving data from client ...");
be040666
DG
4570 ret = lttcomm_recv_creds_unix_sock(sock, cmd_ctx->lsm,
4571 sizeof(struct lttcomm_session_msg), &cmd_ctx->creds);
5eb91c98
DG
4572 if (ret <= 0) {
4573 DBG("Nothing recv() from client... continuing");
76d7553f
MD
4574 ret = close(sock);
4575 if (ret) {
4576 PERROR("close");
4577 }
4578 sock = -1;
a2c0da86 4579 clean_command_ctx(&cmd_ctx);
5eb91c98
DG
4580 continue;
4581 }
1d4b027a 4582
840cb59c 4583 health_code_update();
44a5e5eb 4584
5eb91c98
DG
4585 // TODO: Validate cmd_ctx including sanity check for
4586 // security purpose.
f7776ea7 4587
f6a9efaa 4588 rcu_thread_online();
5eb91c98
DG
4589 /*
4590 * This function dispatch the work to the kernel or userspace tracer
4591 * libs and fill the lttcomm_lttng_msg data structure of all the needed
4592 * informations for the client. The command context struct contains
4593 * everything this function may needs.
4594 */
53a80697 4595 ret = process_client_msg(cmd_ctx, sock, &sock_error);
f6a9efaa 4596 rcu_thread_offline();
5eb91c98 4597 if (ret < 0) {
36134aa1
DG
4598 ret = close(sock);
4599 if (ret) {
4600 PERROR("close");
53a80697 4601 }
36134aa1 4602 sock = -1;
bbd973c2 4603 /*
5eb91c98 4604 * TODO: Inform client somehow of the fatal error. At
ba7f0ae5 4605 * this point, ret < 0 means that a zmalloc failed
53a80697
MD
4606 * (ENOMEM). Error detected but still accept
4607 * command, unless a socket error has been
4608 * detected.
bbd973c2 4609 */
bbd973c2 4610 clean_command_ctx(&cmd_ctx);
5eb91c98
DG
4611 continue;
4612 }
d6e4fca4 4613
840cb59c 4614 health_code_update();
44a5e5eb 4615
a1ac8b97 4616 DBG("Sending response (size: %d, retcode: %s (%d))",
54d01ffb 4617 cmd_ctx->lttng_msg_size,
a1ac8b97
JG
4618 lttng_strerror(-cmd_ctx->llm->ret_code),
4619 cmd_ctx->llm->ret_code);
54d01ffb 4620 ret = send_unix_sock(sock, cmd_ctx->llm, cmd_ctx->lttng_msg_size);
5eb91c98
DG
4621 if (ret < 0) {
4622 ERR("Failed to send data back to client");
bbd973c2 4623 }
1d4b027a 4624
5eb91c98 4625 /* End of transmission */
76d7553f
MD
4626 ret = close(sock);
4627 if (ret) {
4628 PERROR("close");
4629 }
4630 sock = -1;
be040666
DG
4631
4632 clean_command_ctx(&cmd_ctx);
44a5e5eb 4633
840cb59c 4634 health_code_update();
273ea72c
DG
4635 }
4636
139ac872 4637exit:
5eb91c98 4638error:
35df2755
CB
4639 if (sock >= 0) {
4640 ret = close(sock);
4641 if (ret) {
4642 PERROR("close");
4643 }
139ac872 4644 }
44a5e5eb 4645
35df2755
CB
4646 lttng_poll_clean(&events);
4647 clean_command_ctx(&cmd_ctx);
4648
4649error_listen:
4650error_create_poll:
273ea72c 4651 unlink(client_unix_sock_path);
76d7553f
MD
4652 if (client_sock >= 0) {
4653 ret = close(client_sock);
4654 if (ret) {
4655 PERROR("close");
4656 }
a4b35e07 4657 }
35df2755
CB
4658
4659 if (err) {
840cb59c 4660 health_error();
35df2755 4661 ERR("Health error occurred in %s", __func__);
a4b35e07 4662 }
273ea72c 4663
8782cc74 4664 health_unregister(health_sessiond);
35df2755
CB
4665
4666 DBG("Client thread dying");
f6a9efaa
DG
4667
4668 rcu_unregister_thread();
4a15001e
MD
4669
4670 /*
4671 * Since we are creating the consumer threads, we own them, so we need
4672 * to join them before our thread exits.
4673 */
4674 ret = join_consumer_thread(&kconsumer_data);
4675 if (ret) {
4676 errno = ret;
4677 PERROR("join_consumer");
4678 }
4679
4680 ret = join_consumer_thread(&ustconsumer32_data);
4681 if (ret) {
4682 errno = ret;
4683 PERROR("join_consumer ust32");
4684 }
4685
4686 ret = join_consumer_thread(&ustconsumer64_data);
4687 if (ret) {
4688 errno = ret;
4689 PERROR("join_consumer ust64");
4690 }
1d4b027a
DG
4691 return NULL;
4692}
4693
c0232ea5
JG
4694static int string_match(const char *str1, const char *str2)
4695{
4696 return (str1 && str2) && !strcmp(str1, str2);
4697}
4698
fac6795d 4699/*
26296c48
JG
4700 * Take an option from the getopt output and set it in the right variable to be
4701 * used later.
4702 *
4703 * Return 0 on success else a negative value.
fac6795d 4704 */
26296c48 4705static int set_option(int opt, const char *arg, const char *optname)
fac6795d 4706{
26296c48 4707 int ret = 0;
fac6795d 4708
c0232ea5 4709 if (string_match(optname, "client-sock") || opt == 'c') {
66b2ce8e
JG
4710 if (!arg || *arg == '\0') {
4711 ret = -EINVAL;
4712 goto end;
4713 }
e8fa9fb0
MD
4714 if (lttng_is_setuid_setgid()) {
4715 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4716 "-c, --client-sock");
4717 } else {
4718 snprintf(client_unix_sock_path, PATH_MAX, "%s", arg);
4719 }
c0232ea5 4720 } else if (string_match(optname, "apps-sock") || opt == 'a') {
66b2ce8e
JG
4721 if (!arg || *arg == '\0') {
4722 ret = -EINVAL;
4723 goto end;
4724 }
e8fa9fb0
MD
4725 if (lttng_is_setuid_setgid()) {
4726 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4727 "-a, --apps-sock");
4728 } else {
4729 snprintf(apps_unix_sock_path, PATH_MAX, "%s", arg);
4730 }
c0232ea5 4731 } else if (string_match(optname, "daemonize") || opt == 'd') {
26296c48 4732 opt_daemon = 1;
c0232ea5 4733 } else if (string_match(optname, "background") || opt == 'b') {
72dd7491 4734 opt_background = 1;
c0232ea5 4735 } else if (string_match(optname, "group") || opt == 'g') {
66b2ce8e
JG
4736 if (!arg || *arg == '\0') {
4737 ret = -EINVAL;
4738 goto end;
4739 }
e8fa9fb0
MD
4740 if (lttng_is_setuid_setgid()) {
4741 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4742 "-g, --group");
4743 } else {
4744 /*
4745 * If the override option is set, the pointer points to a
4746 * *non* const thus freeing it even though the variable type is
4747 * set to const.
4748 */
4749 if (tracing_group_name_override) {
4750 free((void *) tracing_group_name);
4751 }
4752 tracing_group_name = strdup(arg);
4753 if (!tracing_group_name) {
4754 PERROR("strdup");
4755 ret = -ENOMEM;
4756 }
4757 tracing_group_name_override = 1;
db322c4d 4758 }
c0232ea5 4759 } else if (string_match(optname, "help") || opt == 'h') {
8190767e
PP
4760 ret = utils_show_man_page(8, "lttng-sessiond");
4761 if (ret) {
4762 ERR("Cannot view man page lttng-sessiond(8)");
4763 perror("exec");
4764 }
4765 exit(ret ? EXIT_FAILURE : EXIT_SUCCESS);
c0232ea5 4766 } else if (string_match(optname, "version") || opt == 'V') {
26296c48
JG
4767 fprintf(stdout, "%s\n", VERSION);
4768 exit(EXIT_SUCCESS);
c0232ea5 4769 } else if (string_match(optname, "sig-parent") || opt == 'S') {
26296c48 4770 opt_sig_parent = 1;
c0232ea5 4771 } else if (string_match(optname, "kconsumerd-err-sock")) {
66b2ce8e
JG
4772 if (!arg || *arg == '\0') {
4773 ret = -EINVAL;
4774 goto end;
4775 }
e8fa9fb0
MD
4776 if (lttng_is_setuid_setgid()) {
4777 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4778 "--kconsumerd-err-sock");
4779 } else {
4780 snprintf(kconsumer_data.err_unix_sock_path, PATH_MAX, "%s", arg);
4781 }
c0232ea5 4782 } else if (string_match(optname, "kconsumerd-cmd-sock")) {
66b2ce8e
JG
4783 if (!arg || *arg == '\0') {
4784 ret = -EINVAL;
4785 goto end;
4786 }
e8fa9fb0
MD
4787 if (lttng_is_setuid_setgid()) {
4788 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4789 "--kconsumerd-cmd-sock");
4790 } else {
4791 snprintf(kconsumer_data.cmd_unix_sock_path, PATH_MAX, "%s", arg);
4792 }
c0232ea5 4793 } else if (string_match(optname, "ustconsumerd64-err-sock")) {
66b2ce8e
JG
4794 if (!arg || *arg == '\0') {
4795 ret = -EINVAL;
4796 goto end;
4797 }
e8fa9fb0
MD
4798 if (lttng_is_setuid_setgid()) {
4799 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4800 "--ustconsumerd64-err-sock");
4801 } else {
4802 snprintf(ustconsumer64_data.err_unix_sock_path, PATH_MAX, "%s", arg);
4803 }
c0232ea5 4804 } else if (string_match(optname, "ustconsumerd64-cmd-sock")) {
66b2ce8e
JG
4805 if (!arg || *arg == '\0') {
4806 ret = -EINVAL;
4807 goto end;
4808 }
e8fa9fb0
MD
4809 if (lttng_is_setuid_setgid()) {
4810 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4811 "--ustconsumerd64-cmd-sock");
4812 } else {
4813 snprintf(ustconsumer64_data.cmd_unix_sock_path, PATH_MAX, "%s", arg);
4814 }
c0232ea5 4815 } else if (string_match(optname, "ustconsumerd32-err-sock")) {
66b2ce8e
JG
4816 if (!arg || *arg == '\0') {
4817 ret = -EINVAL;
4818 goto end;
4819 }
e8fa9fb0
MD
4820 if (lttng_is_setuid_setgid()) {
4821 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4822 "--ustconsumerd32-err-sock");
4823 } else {
4824 snprintf(ustconsumer32_data.err_unix_sock_path, PATH_MAX, "%s", arg);
4825 }
c0232ea5 4826 } else if (string_match(optname, "ustconsumerd32-cmd-sock")) {
66b2ce8e
JG
4827 if (!arg || *arg == '\0') {
4828 ret = -EINVAL;
4829 goto end;
4830 }
e8fa9fb0
MD
4831 if (lttng_is_setuid_setgid()) {
4832 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4833 "--ustconsumerd32-cmd-sock");
4834 } else {
4835 snprintf(ustconsumer32_data.cmd_unix_sock_path, PATH_MAX, "%s", arg);
4836 }
c0232ea5 4837 } else if (string_match(optname, "no-kernel")) {
26296c48 4838 opt_no_kernel = 1;
c0232ea5 4839 } else if (string_match(optname, "quiet") || opt == 'q') {
26296c48 4840 lttng_opt_quiet = 1;
c0232ea5 4841 } else if (string_match(optname, "verbose") || opt == 'v') {
26296c48
JG
4842 /* Verbose level can increase using multiple -v */
4843 if (arg) {
13755a18 4844 /* Value obtained from config file */
26296c48
JG
4845 lttng_opt_verbose = config_parse_value(arg);
4846 } else {
13755a18
JG
4847 /* -v used on command line */
4848 lttng_opt_verbose++;
26296c48 4849 }
13755a18
JG
4850 /* Clamp value to [0, 3] */
4851 lttng_opt_verbose = lttng_opt_verbose < 0 ? 0 :
4852 (lttng_opt_verbose <= 3 ? lttng_opt_verbose : 3);
c0232ea5 4853 } else if (string_match(optname, "verbose-consumer")) {
26296c48
JG
4854 if (arg) {
4855 opt_verbose_consumer = config_parse_value(arg);
4856 } else {
ca939de8 4857 opt_verbose_consumer++;
26296c48 4858 }
c0232ea5 4859 } else if (string_match(optname, "consumerd32-path")) {
66b2ce8e
JG
4860 if (!arg || *arg == '\0') {
4861 ret = -EINVAL;
4862 goto end;
4863 }
e8fa9fb0
MD
4864 if (lttng_is_setuid_setgid()) {
4865 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4866 "--consumerd32-path");
4867 } else {
4868 if (consumerd32_bin_override) {
4869 free((void *) consumerd32_bin);
4870 }
4871 consumerd32_bin = strdup(arg);
4872 if (!consumerd32_bin) {
4873 PERROR("strdup");
4874 ret = -ENOMEM;
4875 }
4876 consumerd32_bin_override = 1;
db322c4d 4877 }
c0232ea5 4878 } else if (string_match(optname, "consumerd32-libdir")) {
66b2ce8e
JG
4879 if (!arg || *arg == '\0') {
4880 ret = -EINVAL;
4881 goto end;
4882 }
e8fa9fb0
MD
4883 if (lttng_is_setuid_setgid()) {
4884 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4885 "--consumerd32-libdir");
4886 } else {
4887 if (consumerd32_libdir_override) {
4888 free((void *) consumerd32_libdir);
4889 }
4890 consumerd32_libdir = strdup(arg);
4891 if (!consumerd32_libdir) {
4892 PERROR("strdup");
4893 ret = -ENOMEM;
4894 }
4895 consumerd32_libdir_override = 1;
db322c4d 4896 }
c0232ea5 4897 } else if (string_match(optname, "consumerd64-path")) {
66b2ce8e
JG
4898 if (!arg || *arg == '\0') {
4899 ret = -EINVAL;
4900 goto end;
4901 }
e8fa9fb0
MD
4902 if (lttng_is_setuid_setgid()) {
4903 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4904 "--consumerd64-path");
4905 } else {
4906 if (consumerd64_bin_override) {
4907 free((void *) consumerd64_bin);
4908 }
4909 consumerd64_bin = strdup(arg);
4910 if (!consumerd64_bin) {
4911 PERROR("strdup");
4912 ret = -ENOMEM;
4913 }
4914 consumerd64_bin_override = 1;
db322c4d 4915 }
c0232ea5 4916 } else if (string_match(optname, "consumerd64-libdir")) {
66b2ce8e
JG
4917 if (!arg || *arg == '\0') {
4918 ret = -EINVAL;
4919 goto end;
4920 }
e8fa9fb0
MD
4921 if (lttng_is_setuid_setgid()) {
4922 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4923 "--consumerd64-libdir");
4924 } else {
4925 if (consumerd64_libdir_override) {
4926 free((void *) consumerd64_libdir);
4927 }
4928 consumerd64_libdir = strdup(arg);
4929 if (!consumerd64_libdir) {
4930 PERROR("strdup");
4931 ret = -ENOMEM;
4932 }
4933 consumerd64_libdir_override = 1;
db322c4d 4934 }
c0232ea5 4935 } else if (string_match(optname, "pidfile") || opt == 'p') {
66b2ce8e
JG
4936 if (!arg || *arg == '\0') {
4937 ret = -EINVAL;
4938 goto end;
4939 }
e8fa9fb0
MD
4940 if (lttng_is_setuid_setgid()) {
4941 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4942 "-p, --pidfile");
4943 } else {
4944 free(opt_pidfile);
4945 opt_pidfile = strdup(arg);
4946 if (!opt_pidfile) {
4947 PERROR("strdup");
4948 ret = -ENOMEM;
4949 }
db322c4d 4950 }
c0232ea5 4951 } else if (string_match(optname, "agent-tcp-port")) {
66b2ce8e
JG
4952 if (!arg || *arg == '\0') {
4953 ret = -EINVAL;
4954 goto end;
4955 }
e8fa9fb0
MD
4956 if (lttng_is_setuid_setgid()) {
4957 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4958 "--agent-tcp-port");
4959 } else {
4960 unsigned long v;
4d076222 4961
e8fa9fb0
MD
4962 errno = 0;
4963 v = strtoul(arg, NULL, 0);
4964 if (errno != 0 || !isdigit(arg[0])) {
4965 ERR("Wrong value in --agent-tcp-port parameter: %s", arg);
4966 return -1;
4967 }
4968 if (v == 0 || v >= 65535) {
4969 ERR("Port overflow in --agent-tcp-port parameter: %s", arg);
4970 return -1;
4971 }
4972 agent_tcp_port = (uint32_t) v;
4973 DBG3("Agent TCP port set to non default: %u", agent_tcp_port);
26296c48 4974 }
c0232ea5 4975 } else if (string_match(optname, "load") || opt == 'l') {
66b2ce8e
JG
4976 if (!arg || *arg == '\0') {
4977 ret = -EINVAL;
4978 goto end;
4979 }
e8fa9fb0
MD
4980 if (lttng_is_setuid_setgid()) {
4981 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4982 "-l, --load");
4983 } else {
4984 free(opt_load_session_path);
4985 opt_load_session_path = strdup(arg);
4986 if (!opt_load_session_path) {
4987 PERROR("strdup");
4988 ret = -ENOMEM;
4989 }
ef367a93 4990 }
c0232ea5 4991 } else if (string_match(optname, "kmod-probes")) {
66b2ce8e
JG
4992 if (!arg || *arg == '\0') {
4993 ret = -EINVAL;
4994 goto end;
4995 }
e8fa9fb0
MD
4996 if (lttng_is_setuid_setgid()) {
4997 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
4998 "--kmod-probes");
4999 } else {
5000 free(kmod_probes_list);
5001 kmod_probes_list = strdup(arg);
5002 if (!kmod_probes_list) {
5003 PERROR("strdup");
5004 ret = -ENOMEM;
5005 }
c9d42407 5006 }
c0232ea5 5007 } else if (string_match(optname, "extra-kmod-probes")) {
66b2ce8e
JG
5008 if (!arg || *arg == '\0') {
5009 ret = -EINVAL;
5010 goto end;
5011 }
e8fa9fb0
MD
5012 if (lttng_is_setuid_setgid()) {
5013 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
5014 "--extra-kmod-probes");
5015 } else {
5016 free(kmod_extra_probes_list);
5017 kmod_extra_probes_list = strdup(arg);
5018 if (!kmod_extra_probes_list) {
5019 PERROR("strdup");
5020 ret = -ENOMEM;
5021 }
db322c4d 5022 }
c0232ea5
JG
5023 } else if (string_match(optname, "config") || opt == 'f') {
5024 /* This is handled in set_options() thus silent skip. */
5025 goto end;
5026 } else {
26296c48
JG
5027 /* Unknown option or other error.
5028 * Error is printed by getopt, just return */
5029 ret = -1;
5030 }
5031
f3dd7ce7 5032end:
c5d350b2
JG
5033 if (ret == -EINVAL) {
5034 const char *opt_name = "unknown";
5035 int i;
5036
5037 for (i = 0; i < sizeof(long_options) / sizeof(struct option);
5038 i++) {
5039 if (opt == long_options[i].val) {
5040 opt_name = long_options[i].name;
5041 break;
5042 }
5043 }
5044
5045 WARN("Invalid argument provided for option \"%s\", using default value.",
5046 opt_name);
5047 }
f3dd7ce7 5048
26296c48
JG
5049 return ret;
5050}
5051
5052/*
5053 * config_entry_handler_cb used to handle options read from a config file.
f40ef1d5 5054 * See config_entry_handler_cb comment in common/config/session-config.h for the
26296c48
JG
5055 * return value conventions.
5056 */
5057static int config_entry_handler(const struct config_entry *entry, void *unused)
5058{
5059 int ret = 0, i;
5060
5061 if (!entry || !entry->name || !entry->value) {
5062 ret = -EINVAL;
5063 goto end;
5064 }
5065
5066 /* Check if the option is to be ignored */
5067 for (i = 0; i < sizeof(config_ignore_options) / sizeof(char *); i++) {
5068 if (!strcmp(entry->name, config_ignore_options[i])) {
5069 goto end;
5070 }
5071 }
5072
5073 for (i = 0; i < (sizeof(long_options) / sizeof(struct option)) - 1;
5074 i++) {
5075
5076 /* Ignore if not fully matched. */
5077 if (strcmp(entry->name, long_options[i].name)) {
5078 continue;
5079 }
5080
5081 /*
5082 * If the option takes no argument on the command line, we have to
5083 * check if the value is "true". We support non-zero numeric values,
5084 * true, on and yes.
5085 */
5086 if (!long_options[i].has_arg) {
5087 ret = config_parse_value(entry->value);
5088 if (ret <= 0) {
5089 if (ret) {
5090 WARN("Invalid configuration value \"%s\" for option %s",
5091 entry->value, entry->name);
5092 }
5093 /* False, skip boolean config option. */
5094 goto end;
4d076222 5095 }
26296c48
JG
5096 }
5097
5098 ret = set_option(long_options[i].val, entry->value, entry->name);
5099 goto end;
5100 }
5101
5102 WARN("Unrecognized option \"%s\" in daemon configuration file.", entry->name);
5103
5104end:
5105 return ret;
5106}
5107
5108/*
5109 * daemon configuration loading and argument parsing
5110 */
5111static int set_options(int argc, char **argv)
5112{
5113 int ret = 0, c = 0, option_index = 0;
5114 int orig_optopt = optopt, orig_optind = optind;
5115 char *optstring;
5116 const char *config_path = NULL;
5117
5118 optstring = utils_generate_optstring(long_options,
5119 sizeof(long_options) / sizeof(struct option));
5120 if (!optstring) {
5121 ret = -ENOMEM;
5122 goto end;
5123 }
5124
5125 /* Check for the --config option */
5126 while ((c = getopt_long(argc, argv, optstring, long_options,
5127 &option_index)) != -1) {
5128 if (c == '?') {
5129 ret = -EINVAL;
5130 goto end;
5131 } else if (c != 'f') {
5132 /* if not equal to --config option. */
5133 continue;
5134 }
5135
e8fa9fb0
MD
5136 if (lttng_is_setuid_setgid()) {
5137 WARN("Getting '%s' argument from setuid/setgid binary refused for security reasons.",
5138 "-f, --config");
5139 } else {
5140 config_path = utils_expand_path(optarg);
5141 if (!config_path) {
5142 ERR("Failed to resolve path: %s", optarg);
5143 }
26296c48
JG
5144 }
5145 }
5146
5147 ret = config_get_section_entries(config_path, config_section_name,
5148 config_entry_handler, NULL);
5149 if (ret) {
5150 if (ret > 0) {
5151 ERR("Invalid configuration option at line %i", ret);
5152 ret = -1;
5153 }
5154 goto end;
5155 }
5156
5157 /* Reset getopt's global state */
5158 optopt = orig_optopt;
5159 optind = orig_optind;
5160 while (1) {
c0232ea5
JG
5161 option_index = -1;
5162 /*
5163 * getopt_long() will not set option_index if it encounters a
5164 * short option.
5165 */
5166 c = getopt_long(argc, argv, optstring, long_options,
5167 &option_index);
26296c48 5168 if (c == -1) {
4d076222
DG
5169 break;
5170 }
26296c48 5171
c0232ea5
JG
5172 /*
5173 * Pass NULL as the long option name if popt left the index
5174 * unset.
5175 */
5176 ret = set_option(c, optarg,
5177 option_index < 0 ? NULL :
5178 long_options[option_index].name);
26296c48
JG
5179 if (ret < 0) {
5180 break;
fac6795d
DG
5181 }
5182 }
5183
26296c48
JG
5184end:
5185 free(optstring);
5186 return ret;
fac6795d
DG
5187}
5188
5189/*
d063d709 5190 * Creates the two needed socket by the daemon.
d6f42150
DG
5191 * apps_sock - The communication socket for all UST apps.
5192 * client_sock - The communication of the cli tool (lttng).
fac6795d 5193 */
cf3af59e 5194static int init_daemon_socket(void)
fac6795d
DG
5195{
5196 int ret = 0;
5197 mode_t old_umask;
5198
5199 old_umask = umask(0);
5200
5201 /* Create client tool unix socket */
d6f42150
DG
5202 client_sock = lttcomm_create_unix_sock(client_unix_sock_path);
5203 if (client_sock < 0) {
5204 ERR("Create unix sock failed: %s", client_unix_sock_path);
fac6795d
DG
5205 ret = -1;
5206 goto end;
5207 }
5208
b662582b
DG
5209 /* Set the cloexec flag */
5210 ret = utils_set_fd_cloexec(client_sock);
5211 if (ret < 0) {
5212 ERR("Unable to set CLOEXEC flag to the client Unix socket (fd: %d). "
5213 "Continuing but note that the consumer daemon will have a "
5214 "reference to this socket on exec()", client_sock);
5215 }
5216
fac6795d
DG
5217 /* File permission MUST be 660 */
5218 ret = chmod(client_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
5219 if (ret < 0) {
d6f42150 5220 ERR("Set file permissions failed: %s", client_unix_sock_path);
76d7553f 5221 PERROR("chmod");
fac6795d
DG
5222 goto end;
5223 }
5224
5225 /* Create the application unix socket */
d6f42150
DG
5226 apps_sock = lttcomm_create_unix_sock(apps_unix_sock_path);
5227 if (apps_sock < 0) {
5228 ERR("Create unix sock failed: %s", apps_unix_sock_path);
fac6795d
DG
5229 ret = -1;
5230 goto end;
5231 }
5232
b662582b
DG
5233 /* Set the cloexec flag */
5234 ret = utils_set_fd_cloexec(apps_sock);
5235 if (ret < 0) {
5236 ERR("Unable to set CLOEXEC flag to the app Unix socket (fd: %d). "
5237 "Continuing but note that the consumer daemon will have a "
5238 "reference to this socket on exec()", apps_sock);
5239 }
5240
d6f42150 5241 /* File permission MUST be 666 */
54d01ffb
DG
5242 ret = chmod(apps_unix_sock_path,
5243 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
fac6795d 5244 if (ret < 0) {
d6f42150 5245 ERR("Set file permissions failed: %s", apps_unix_sock_path);
76d7553f 5246 PERROR("chmod");
fac6795d
DG
5247 goto end;
5248 }
5249
b662582b
DG
5250 DBG3("Session daemon client socket %d and application socket %d created",
5251 client_sock, apps_sock);
5252
fac6795d
DG
5253end:
5254 umask(old_umask);
5255 return ret;
5256}
5257
5258/*
54d01ffb
DG
5259 * Check if the global socket is available, and if a daemon is answering at the
5260 * other side. If yes, error is returned.
fac6795d 5261 */
cf3af59e 5262static int check_existing_daemon(void)
fac6795d 5263{
7d8234d9 5264 /* Is there anybody out there ? */
099e26bd 5265 if (lttng_session_daemon_alive()) {
7d8234d9 5266 return -EEXIST;
099e26bd 5267 }
b09c7c76
DG
5268
5269 return 0;
fac6795d
DG
5270}
5271
fac6795d 5272/*
d063d709 5273 * Set the tracing group gid onto the client socket.
5e16da05 5274 *
d063d709 5275 * Race window between mkdir and chown is OK because we are going from more
d1613cf5 5276 * permissive (root.root) to less permissive (root.tracing).
fac6795d 5277 */
be040666 5278static int set_permissions(char *rundir)
fac6795d
DG
5279{
5280 int ret;
996b65c8 5281 gid_t gid;
fac6795d 5282
6c71277b 5283 gid = utils_get_group_id(tracing_group_name);
fac6795d 5284
d6f42150 5285 /* Set lttng run dir */
be040666 5286 ret = chown(rundir, 0, gid);
d6f42150 5287 if (ret < 0) {
be040666 5288 ERR("Unable to set group on %s", rundir);
76d7553f 5289 PERROR("chown");
d6f42150
DG
5290 }
5291
6c71277b
MD
5292 /*
5293 * Ensure all applications and tracing group can search the run
5294 * dir. Allow everyone to read the directory, since it does not
5295 * buy us anything to hide its content.
5296 */
5297 ret = chmod(rundir, S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
d1613cf5
JN
5298 if (ret < 0) {
5299 ERR("Unable to set permissions on %s", rundir);
76d7553f 5300 PERROR("chmod");
d1613cf5
JN
5301 }
5302
d6f42150 5303 /* lttng client socket path */
996b65c8 5304 ret = chown(client_unix_sock_path, 0, gid);
fac6795d 5305 if (ret < 0) {
d6f42150 5306 ERR("Unable to set group on %s", client_unix_sock_path);
76d7553f 5307 PERROR("chown");
d6f42150
DG
5308 }
5309
3bd1e081 5310 /* kconsumer error socket path */
6c71277b 5311 ret = chown(kconsumer_data.err_unix_sock_path, 0, 0);
d6f42150 5312 if (ret < 0) {
3bd1e081 5313 ERR("Unable to set group on %s", kconsumer_data.err_unix_sock_path);
76d7553f 5314 PERROR("chown");
3bd1e081
MD
5315 }
5316
7753dea8 5317 /* 64-bit ustconsumer error socket path */
6c71277b 5318 ret = chown(ustconsumer64_data.err_unix_sock_path, 0, 0);
7753dea8
MD
5319 if (ret < 0) {
5320 ERR("Unable to set group on %s", ustconsumer64_data.err_unix_sock_path);
76d7553f 5321 PERROR("chown");
7753dea8
MD
5322 }
5323
5324 /* 32-bit ustconsumer compat32 error socket path */
6c71277b 5325 ret = chown(ustconsumer32_data.err_unix_sock_path, 0, 0);
3bd1e081 5326 if (ret < 0) {
7753dea8 5327 ERR("Unable to set group on %s", ustconsumer32_data.err_unix_sock_path);
76d7553f 5328 PERROR("chown");
fac6795d
DG
5329 }
5330
d6f42150 5331 DBG("All permissions are set");
e07ae692 5332
fac6795d
DG
5333 return ret;
5334}
5335
d6f42150 5336/*
d063d709 5337 * Create the lttng run directory needed for all global sockets and pipe.
d6f42150 5338 */
67e40797 5339static int create_lttng_rundir(const char *rundir)
d6f42150
DG
5340{
5341 int ret;
5342
67e40797
DG
5343 DBG3("Creating LTTng run directory: %s", rundir);
5344
d1613cf5 5345 ret = mkdir(rundir, S_IRWXU);
d6f42150 5346 if (ret < 0) {
b1f11e69 5347 if (errno != EEXIST) {
67e40797 5348 ERR("Unable to create %s", rundir);
b1f11e69
DG
5349 goto error;
5350 } else {
5351 ret = 0;
5352 }
d6f42150
DG
5353 }
5354
5355error:
5356 return ret;
5357}
5358
5359/*
d063d709
DG
5360 * Setup sockets and directory needed by the kconsumerd communication with the
5361 * session daemon.
d6f42150 5362 */
67e40797
DG
5363static int set_consumer_sockets(struct consumer_data *consumer_data,
5364 const char *rundir)
d6f42150
DG
5365{
5366 int ret;
67e40797 5367 char path[PATH_MAX];
d6f42150 5368
6c71277b 5369 switch (consumer_data->type) {
7753dea8 5370 case LTTNG_CONSUMER_KERNEL:
60922cb0 5371 snprintf(path, PATH_MAX, DEFAULT_KCONSUMERD_PATH, rundir);
7753dea8
MD
5372 break;
5373 case LTTNG_CONSUMER64_UST:
60922cb0 5374 snprintf(path, PATH_MAX, DEFAULT_USTCONSUMERD64_PATH, rundir);
7753dea8
MD
5375 break;
5376 case LTTNG_CONSUMER32_UST:
60922cb0 5377 snprintf(path, PATH_MAX, DEFAULT_USTCONSUMERD32_PATH, rundir);
7753dea8
MD
5378 break;
5379 default:
5380 ERR("Consumer type unknown");
5381 ret = -EINVAL;
5382 goto error;
d6f42150
DG
5383 }
5384
67e40797
DG
5385 DBG2("Creating consumer directory: %s", path);
5386
6c71277b 5387 ret = mkdir(path, S_IRWXU | S_IRGRP | S_IXGRP);
d6f42150 5388 if (ret < 0) {
6beb2242 5389 if (errno != EEXIST) {
f11e84c2 5390 PERROR("mkdir");
3bd1e081 5391 ERR("Failed to create %s", path);
6beb2242
DG
5392 goto error;
5393 }
f11e84c2 5394 ret = -1;
d6f42150 5395 }
6c71277b
MD
5396 if (is_root) {
5397 ret = chown(path, 0, utils_get_group_id(tracing_group_name));
5398 if (ret < 0) {
5399 ERR("Unable to set group on %s", path);
5400 PERROR("chown");
5401 goto error;
5402 }
5403 }
d6f42150
DG
5404
5405 /* Create the kconsumerd error unix socket */
3bd1e081
MD
5406 consumer_data->err_sock =
5407 lttcomm_create_unix_sock(consumer_data->err_unix_sock_path);
5408 if (consumer_data->err_sock < 0) {
5409 ERR("Create unix sock failed: %s", consumer_data->err_unix_sock_path);
d6f42150
DG
5410 ret = -1;
5411 goto error;
5412 }
5413
a24f05ab
MD
5414 /*
5415 * Set the CLOEXEC flag. Return code is useless because either way, the
5416 * show must go on.
5417 */
5418 ret = utils_set_fd_cloexec(consumer_data->err_sock);
5419 if (ret < 0) {
5420 PERROR("utils_set_fd_cloexec");
5421 /* continue anyway */
5422 }
5423
d6f42150 5424 /* File permission MUST be 660 */
3bd1e081 5425 ret = chmod(consumer_data->err_unix_sock_path,
54d01ffb 5426 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
d6f42150 5427 if (ret < 0) {
3bd1e081 5428 ERR("Set file permissions failed: %s", consumer_data->err_unix_sock_path);
67e40797 5429 PERROR("chmod");
d6f42150
DG
5430 goto error;
5431 }
5432
5433error:
5434 return ret;
5435}
5436
fac6795d 5437/*
d063d709 5438 * Signal handler for the daemon
cf3af59e 5439 *
54d01ffb
DG
5440 * Simply stop all worker threads, leaving main() return gracefully after
5441 * joining all threads and calling cleanup().
fac6795d
DG
5442 */
5443static void sighandler(int sig)
5444{
5445 switch (sig) {
cf3af59e 5446 case SIGINT:
af87c45a 5447 DBG("SIGINT caught");
cf3af59e
MD
5448 stop_threads();
5449 break;
5450 case SIGTERM:
af87c45a 5451 DBG("SIGTERM caught");
cf3af59e
MD
5452 stop_threads();
5453 break;
0bb7724a
DG
5454 case SIGUSR1:
5455 CMM_STORE_SHARED(recv_child_signal, 1);
5456 break;
cf3af59e
MD
5457 default:
5458 break;
fac6795d 5459 }
fac6795d
DG
5460}
5461
5462/*
d063d709 5463 * Setup signal handler for :
1d4b027a 5464 * SIGINT, SIGTERM, SIGPIPE
fac6795d 5465 */
1d4b027a 5466static int set_signal_handler(void)
fac6795d 5467{
1d4b027a
DG
5468 int ret = 0;
5469 struct sigaction sa;
5470 sigset_t sigset;
fac6795d 5471
1d4b027a 5472 if ((ret = sigemptyset(&sigset)) < 0) {
76d7553f 5473 PERROR("sigemptyset");
1d4b027a
DG
5474 return ret;
5475 }
d6f42150 5476
1d4b027a
DG
5477 sa.sa_mask = sigset;
5478 sa.sa_flags = 0;
0072e5e2
MD
5479
5480 sa.sa_handler = sighandler;
1d4b027a 5481 if ((ret = sigaction(SIGTERM, &sa, NULL)) < 0) {
76d7553f 5482 PERROR("sigaction");
1d4b027a 5483 return ret;
d6f42150
DG
5484 }
5485
1d4b027a 5486 if ((ret = sigaction(SIGINT, &sa, NULL)) < 0) {
76d7553f 5487 PERROR("sigaction");
1d4b027a 5488 return ret;
d6f42150 5489 }
aaf26714 5490
0072e5e2 5491 if ((ret = sigaction(SIGUSR1, &sa, NULL)) < 0) {
76d7553f 5492 PERROR("sigaction");
1d4b027a 5493 return ret;
8c0faa1d
DG
5494 }
5495
0072e5e2
MD
5496 sa.sa_handler = SIG_IGN;
5497 if ((ret = sigaction(SIGPIPE, &sa, NULL)) < 0) {
0bb7724a
DG
5498 PERROR("sigaction");
5499 return ret;
5500 }
5501
5502 DBG("Signal handler set for SIGTERM, SIGUSR1, SIGPIPE and SIGINT");
1d4b027a
DG
5503
5504 return ret;
fac6795d
DG
5505}
5506
f3ed775e 5507/*
d063d709 5508 * Set open files limit to unlimited. This daemon can open a large number of
514bb9f0 5509 * file descriptors in order to consume multiple kernel traces.
f3ed775e
DG
5510 */
5511static void set_ulimit(void)
5512{
5513 int ret;
5514 struct rlimit lim;
5515
514bb9f0 5516 /* The kernel does not allow an infinite limit for open files */
f3ed775e
DG
5517 lim.rlim_cur = 65535;
5518 lim.rlim_max = 65535;
5519
5520 ret = setrlimit(RLIMIT_NOFILE, &lim);
5521 if (ret < 0) {
76d7553f 5522 PERROR("failed to set open files limit");
f3ed775e
DG
5523 }
5524}
5525
35f90c40
DG
5526/*
5527 * Write pidfile using the rundir and opt_pidfile.
5528 */
4a15001e 5529static int write_pidfile(void)
35f90c40
DG
5530{
5531 int ret;
5532 char pidfile_path[PATH_MAX];
5533
5534 assert(rundir);
5535
5536 if (opt_pidfile) {
b8d76705
MD
5537 if (lttng_strncpy(pidfile_path, opt_pidfile, sizeof(pidfile_path))) {
5538 ret = -1;
5539 goto error;
5540 }
35f90c40
DG
5541 } else {
5542 /* Build pidfile path from rundir and opt_pidfile. */
5543 ret = snprintf(pidfile_path, sizeof(pidfile_path), "%s/"
5544 DEFAULT_LTTNG_SESSIOND_PIDFILE, rundir);
5545 if (ret < 0) {
5546 PERROR("snprintf pidfile path");
5547 goto error;
5548 }
5549 }
5550
5551 /*
4a15001e 5552 * Create pid file in rundir.
35f90c40 5553 */
4a15001e 5554 ret = utils_create_pid_file(getpid(), pidfile_path);
35f90c40 5555error:
4a15001e 5556 return ret;
35f90c40
DG
5557}
5558
c9cb3e7d
JG
5559/*
5560 * Create lockfile using the rundir and return its fd.
5561 */
5562static int create_lockfile(void)
5563{
5564 int ret;
5565 char lockfile_path[PATH_MAX];
5566
5567 ret = generate_lock_file_path(lockfile_path, sizeof(lockfile_path));
5568 if (ret < 0) {
5569 goto error;
5570 }
5571
5572 ret = utils_create_lock_file(lockfile_path);
5573error:
5574 return ret;
5575}
5576
cd9290dd 5577/*
022d91ba 5578 * Write agent TCP port using the rundir.
cd9290dd 5579 */
4a15001e 5580static int write_agent_port(void)
cd9290dd
DG
5581{
5582 int ret;
5583 char path[PATH_MAX];
5584
5585 assert(rundir);
5586
5587 ret = snprintf(path, sizeof(path), "%s/"
022d91ba 5588 DEFAULT_LTTNG_SESSIOND_AGENTPORT_FILE, rundir);
cd9290dd 5589 if (ret < 0) {
022d91ba 5590 PERROR("snprintf agent port path");
cd9290dd
DG
5591 goto error;
5592 }
5593
5594 /*
4a15001e 5595 * Create TCP agent port file in rundir.
cd9290dd 5596 */
4a15001e 5597 ret = utils_create_pid_file(agent_tcp_port, path);
cd9290dd
DG
5598
5599error:
4a15001e 5600 return ret;
ef367a93
JG
5601}
5602
fac6795d
DG
5603/*
5604 * main
5605 */
5606int main(int argc, char **argv)
5607{
4a15001e 5608 int ret = 0, retval = 0;
fac6795d 5609 void *status;
ae9e45b3 5610 const char *home_path, *env_app_timeout;
daa81f58
JG
5611 struct lttng_pipe *ust32_channel_monitor_pipe = NULL,
5612 *ust64_channel_monitor_pipe = NULL,
5613 *kernel_channel_monitor_pipe = NULL;
fac6795d 5614
335a95b7
MD
5615 init_kernel_workarounds();
5616
f6a9efaa
DG
5617 rcu_register_thread();
5618
4a15001e
MD
5619 if (set_signal_handler()) {
5620 retval = -1;
5621 goto exit_set_signal_handler;
0bb7724a
DG
5622 }
5623
7753dea8 5624 setup_consumerd_path();
fb09408a 5625
12744796
DG
5626 page_size = sysconf(_SC_PAGESIZE);
5627 if (page_size < 0) {
5628 PERROR("sysconf _SC_PAGESIZE");
5629 page_size = LONG_MAX;
5630 WARN("Fallback page size to %ld", page_size);
5631 }
5632
4a15001e
MD
5633 /*
5634 * Parse arguments and load the daemon configuration file.
5635 *
5636 * We have an exit_options exit path to free memory reserved by
5637 * set_options. This is needed because the rest of sessiond_cleanup()
5638 * depends on ht_cleanup_thread, which depends on lttng_daemonize, which
5639 * depends on set_options.
5640 */
fac6795d 5641 progname = argv[0];
4a15001e
MD
5642 if (set_options(argc, argv)) {
5643 retval = -1;
5644 goto exit_options;
fac6795d
DG
5645 }
5646
5647 /* Daemonize */
72dd7491 5648 if (opt_daemon || opt_background) {
ceed52b5
MD
5649 int i;
5650
72dd7491
MD
5651 ret = lttng_daemonize(&child_ppid, &recv_child_signal,
5652 !opt_background);
53094c05 5653 if (ret < 0) {
4a15001e
MD
5654 retval = -1;
5655 goto exit_options;
53094c05 5656 }
0bb7724a 5657
ceed52b5 5658 /*
0bb7724a 5659 * We are in the child. Make sure all other file descriptors are
4a15001e
MD
5660 * closed, in case we are called with more opened file
5661 * descriptors than the standard ones.
ceed52b5
MD
5662 */
5663 for (i = 3; i < sysconf(_SC_OPEN_MAX); i++) {
5664 (void) close(i);
5665 }
5666 }
5667
7567352f
MD
5668 if (run_as_create_worker(argv[0]) < 0) {
5669 goto exit_create_run_as_worker_cleanup;
5670 }
5671
4a15001e
MD
5672 /*
5673 * Starting from here, we can create threads. This needs to be after
5674 * lttng_daemonize due to RCU.
5675 */
5676
5677 /*
5678 * Initialize the health check subsystem. This call should set the
5679 * appropriate time values.
5680 */
5681 health_sessiond = health_app_create(NR_HEALTH_SESSIOND_TYPES);
5682 if (!health_sessiond) {
5683 PERROR("health_app_create error");
5684 retval = -1;
5685 goto exit_health_sessiond_cleanup;
5686 }
5687
4a15001e 5688 /* Create thread to clean up RCU hash tables */
5e97de00 5689 if (init_ht_cleanup_thread(&ht_cleanup_thread)) {
4a15001e
MD
5690 retval = -1;
5691 goto exit_ht_cleanup;
5692 }
5693
ceed52b5 5694 /* Create thread quit pipe */
4a15001e
MD
5695 if (init_thread_quit_pipe()) {
5696 retval = -1;
5697 goto exit_init_data;
fac6795d
DG
5698 }
5699
5700 /* Check if daemon is UID = 0 */
5701 is_root = !getuid();
5702
fac6795d 5703 if (is_root) {
990570ed 5704 rundir = strdup(DEFAULT_LTTNG_RUNDIR);
4ea184d1 5705 if (!rundir) {
4a15001e
MD
5706 retval = -1;
5707 goto exit_init_data;
4ea184d1 5708 }
67e40797
DG
5709
5710 /* Create global run dir with root access */
4a15001e
MD
5711 if (create_lttng_rundir(rundir)) {
5712 retval = -1;
5713 goto exit_init_data;
d6f42150
DG
5714 }
5715
fac6795d 5716 if (strlen(apps_unix_sock_path) == 0) {
4a15001e 5717 ret = snprintf(apps_unix_sock_path, PATH_MAX,
d6f42150 5718 DEFAULT_GLOBAL_APPS_UNIX_SOCK);
4a15001e
MD
5719 if (ret < 0) {
5720 retval = -1;
5721 goto exit_init_data;
5722 }
fac6795d
DG
5723 }
5724
5725 if (strlen(client_unix_sock_path) == 0) {
4a15001e 5726 ret = snprintf(client_unix_sock_path, PATH_MAX,
d6f42150 5727 DEFAULT_GLOBAL_CLIENT_UNIX_SOCK);
4a15001e
MD
5728 if (ret < 0) {
5729 retval = -1;
5730 goto exit_init_data;
5731 }
d6f42150 5732 }
0fdd1e2c
DG
5733
5734 /* Set global SHM for ust */
5735 if (strlen(wait_shm_path) == 0) {
4a15001e 5736 ret = snprintf(wait_shm_path, PATH_MAX,
0fdd1e2c 5737 DEFAULT_GLOBAL_APPS_WAIT_SHM_PATH);
4a15001e
MD
5738 if (ret < 0) {
5739 retval = -1;
5740 goto exit_init_data;
5741 }
0fdd1e2c 5742 }
67e40797 5743
44a5e5eb 5744 if (strlen(health_unix_sock_path) == 0) {
4a15001e
MD
5745 ret = snprintf(health_unix_sock_path,
5746 sizeof(health_unix_sock_path),
44a5e5eb 5747 DEFAULT_GLOBAL_HEALTH_UNIX_SOCK);
4a15001e
MD
5748 if (ret < 0) {
5749 retval = -1;
5750 goto exit_init_data;
5751 }
44a5e5eb
DG
5752 }
5753
67e40797 5754 /* Setup kernel consumerd path */
4a15001e 5755 ret = snprintf(kconsumer_data.err_unix_sock_path, PATH_MAX,
60922cb0 5756 DEFAULT_KCONSUMERD_ERR_SOCK_PATH, rundir);
4a15001e
MD
5757 if (ret < 0) {
5758 retval = -1;
5759 goto exit_init_data;
5760 }
5761 ret = snprintf(kconsumer_data.cmd_unix_sock_path, PATH_MAX,
60922cb0 5762 DEFAULT_KCONSUMERD_CMD_SOCK_PATH, rundir);
4a15001e
MD
5763 if (ret < 0) {
5764 retval = -1;
5765 goto exit_init_data;
5766 }
67e40797
DG
5767
5768 DBG2("Kernel consumer err path: %s",
5769 kconsumer_data.err_unix_sock_path);
5770 DBG2("Kernel consumer cmd path: %s",
5771 kconsumer_data.cmd_unix_sock_path);
daa81f58
JG
5772 kernel_channel_monitor_pipe = lttng_pipe_open(0);
5773 if (!kernel_channel_monitor_pipe) {
5774 ERR("Failed to create kernel consumer channel monitor pipe");
5775 retval = -1;
5776 goto exit_init_data;
5777 }
5778 kconsumer_data.channel_monitor_pipe =
5779 lttng_pipe_release_writefd(
5780 kernel_channel_monitor_pipe);
5781 if (kconsumer_data.channel_monitor_pipe < 0) {
5782 retval = -1;
5783 goto exit_init_data;
5784 }
fac6795d 5785 } else {
feb0f3e5 5786 home_path = utils_get_home_dir();
b082db07 5787 if (home_path == NULL) {
273ea72c
DG
5788 /* TODO: Add --socket PATH option */
5789 ERR("Can't get HOME directory for sockets creation.");
4a15001e
MD
5790 retval = -1;
5791 goto exit_init_data;
b082db07
DG
5792 }
5793
67e40797
DG
5794 /*
5795 * Create rundir from home path. This will create something like
5796 * $HOME/.lttng
5797 */
990570ed 5798 ret = asprintf(&rundir, DEFAULT_LTTNG_HOME_RUNDIR, home_path);
67e40797 5799 if (ret < 0) {
4a15001e
MD
5800 retval = -1;
5801 goto exit_init_data;
67e40797
DG
5802 }
5803
4a15001e
MD
5804 if (create_lttng_rundir(rundir)) {
5805 retval = -1;
5806 goto exit_init_data;
67e40797
DG
5807 }
5808
fac6795d 5809 if (strlen(apps_unix_sock_path) == 0) {
4a15001e
MD
5810 ret = snprintf(apps_unix_sock_path, PATH_MAX,
5811 DEFAULT_HOME_APPS_UNIX_SOCK,
5812 home_path);
5813 if (ret < 0) {
5814 retval = -1;
5815 goto exit_init_data;
5816 }
fac6795d
DG
5817 }
5818
5819 /* Set the cli tool unix socket path */
5820 if (strlen(client_unix_sock_path) == 0) {
4a15001e
MD
5821 ret = snprintf(client_unix_sock_path, PATH_MAX,
5822 DEFAULT_HOME_CLIENT_UNIX_SOCK,
5823 home_path);
5824 if (ret < 0) {
5825 retval = -1;
5826 goto exit_init_data;
5827 }
fac6795d 5828 }
0fdd1e2c
DG
5829
5830 /* Set global SHM for ust */
5831 if (strlen(wait_shm_path) == 0) {
4a15001e
MD
5832 ret = snprintf(wait_shm_path, PATH_MAX,
5833 DEFAULT_HOME_APPS_WAIT_SHM_PATH,
5834 getuid());
5835 if (ret < 0) {
5836 retval = -1;
5837 goto exit_init_data;
5838 }
0fdd1e2c 5839 }
44a5e5eb
DG
5840
5841 /* Set health check Unix path */
5842 if (strlen(health_unix_sock_path) == 0) {
4a15001e
MD
5843 ret = snprintf(health_unix_sock_path,
5844 sizeof(health_unix_sock_path),
5845 DEFAULT_HOME_HEALTH_UNIX_SOCK,
5846 home_path);
5847 if (ret < 0) {
5848 retval = -1;
5849 goto exit_init_data;
5850 }
44a5e5eb 5851 }
fac6795d
DG
5852 }
5853
c9cb3e7d
JG
5854 lockfile_fd = create_lockfile();
5855 if (lockfile_fd < 0) {
4a15001e
MD
5856 retval = -1;
5857 goto exit_init_data;
c9cb3e7d
JG
5858 }
5859
5c827ce0
DG
5860 /* Set consumer initial state */
5861 kernel_consumerd_state = CONSUMER_STOPPED;
5862 ust_consumerd_state = CONSUMER_STOPPED;
5863
847177cd
DG
5864 DBG("Client socket path %s", client_unix_sock_path);
5865 DBG("Application socket path %s", apps_unix_sock_path);
d0b96690 5866 DBG("Application wait path %s", wait_shm_path);
67e40797
DG
5867 DBG("LTTng run directory path: %s", rundir);
5868
5869 /* 32 bits consumerd path setup */
4a15001e 5870 ret = snprintf(ustconsumer32_data.err_unix_sock_path, PATH_MAX,
60922cb0 5871 DEFAULT_USTCONSUMERD32_ERR_SOCK_PATH, rundir);
4a15001e
MD
5872 if (ret < 0) {
5873 PERROR("snprintf 32-bit consumer error socket path");
5874 retval = -1;
5875 goto exit_init_data;
5876 }
5877 ret = snprintf(ustconsumer32_data.cmd_unix_sock_path, PATH_MAX,
60922cb0 5878 DEFAULT_USTCONSUMERD32_CMD_SOCK_PATH, rundir);
4a15001e
MD
5879 if (ret < 0) {
5880 PERROR("snprintf 32-bit consumer command socket path");
5881 retval = -1;
5882 goto exit_init_data;
5883 }
67e40797
DG
5884
5885 DBG2("UST consumer 32 bits err path: %s",
5886 ustconsumer32_data.err_unix_sock_path);
5887 DBG2("UST consumer 32 bits cmd path: %s",
5888 ustconsumer32_data.cmd_unix_sock_path);
daa81f58
JG
5889 ust32_channel_monitor_pipe = lttng_pipe_open(0);
5890 if (!ust32_channel_monitor_pipe) {
5891 ERR("Failed to create 32-bit user space consumer channel monitor pipe");
5892 retval = -1;
5893 goto exit_init_data;
5894 }
5895 ustconsumer32_data.channel_monitor_pipe = lttng_pipe_release_writefd(
5896 ust32_channel_monitor_pipe);
5897 if (ustconsumer32_data.channel_monitor_pipe < 0) {
5898 retval = -1;
5899 goto exit_init_data;
5900 }
67e40797
DG
5901
5902 /* 64 bits consumerd path setup */
4a15001e 5903 ret = snprintf(ustconsumer64_data.err_unix_sock_path, PATH_MAX,
60922cb0 5904 DEFAULT_USTCONSUMERD64_ERR_SOCK_PATH, rundir);
4a15001e
MD
5905 if (ret < 0) {
5906 PERROR("snprintf 64-bit consumer error socket path");
5907 retval = -1;
5908 goto exit_init_data;
5909 }
5910 ret = snprintf(ustconsumer64_data.cmd_unix_sock_path, PATH_MAX,
60922cb0 5911 DEFAULT_USTCONSUMERD64_CMD_SOCK_PATH, rundir);
4a15001e
MD
5912 if (ret < 0) {
5913 PERROR("snprintf 64-bit consumer command socket path");
5914 retval = -1;
5915 goto exit_init_data;
5916 }
67e40797
DG
5917
5918 DBG2("UST consumer 64 bits err path: %s",
5919 ustconsumer64_data.err_unix_sock_path);
5920 DBG2("UST consumer 64 bits cmd path: %s",
5921 ustconsumer64_data.cmd_unix_sock_path);
daa81f58
JG
5922 ust64_channel_monitor_pipe = lttng_pipe_open(0);
5923 if (!ust64_channel_monitor_pipe) {
5924 ERR("Failed to create 64-bit user space consumer channel monitor pipe");
5925 retval = -1;
5926 goto exit_init_data;
5927 }
5928 ustconsumer64_data.channel_monitor_pipe = lttng_pipe_release_writefd(
5929 ust64_channel_monitor_pipe);
5930 if (ustconsumer64_data.channel_monitor_pipe < 0) {
5931 retval = -1;
5932 goto exit_init_data;
5933 }
847177cd 5934
273ea72c 5935 /*
7d8234d9 5936 * See if daemon already exist.
fac6795d 5937 */
4a15001e 5938 if (check_existing_daemon()) {
75462a81 5939 ERR("Already running daemon.\n");
273ea72c 5940 /*
cf3af59e
MD
5941 * We do not goto exit because we must not cleanup()
5942 * because a daemon is already running.
ab118b20 5943 */
4a15001e
MD
5944 retval = -1;
5945 goto exit_init_data;
a88df331
DG
5946 }
5947
1427f9b2
DG
5948 /*
5949 * Init UST app hash table. Alloc hash table before this point since
5950 * cleanup() can get called after that point.
5951 */
4a15001e 5952 if (ust_app_ht_alloc()) {
ddbeb0f6 5953 ERR("Failed to allocate UST app hash table");
4a15001e
MD
5954 retval = -1;
5955 goto exit_init_data;
5956 }
1427f9b2 5957
6a4e4039
JG
5958 /*
5959 * Initialize agent app hash table. We allocate the hash table here
5960 * since cleanup() can get called after this point.
5961 */
5962 if (agent_app_ht_alloc()) {
5963 ERR("Failed to allocate Agent app hash table");
4a15001e
MD
5964 retval = -1;
5965 goto exit_init_data;
f20baf8e
DG
5966 }
5967
a88df331
DG
5968 /*
5969 * These actions must be executed as root. We do that *after* setting up
5970 * the sockets path because we MUST make the check for another daemon using
5971 * those paths *before* trying to set the kernel consumer sockets and init
5972 * kernel tracer.
5973 */
5974 if (is_root) {
4a15001e
MD
5975 if (set_consumer_sockets(&kconsumer_data, rundir)) {
5976 retval = -1;
5977 goto exit_init_data;
7753dea8
MD
5978 }
5979
a88df331 5980 /* Setup kernel tracer */
4fba7219
DG
5981 if (!opt_no_kernel) {
5982 init_kernel_tracer();
89588270
DG
5983 if (kernel_tracer_fd >= 0) {
5984 ret = syscall_init_table();
5985 if (ret < 0) {
4a15001e
MD
5986 ERR("Unable to populate syscall table. "
5987 "Syscall tracing won't work "
5988 "for this session daemon.");
89588270 5989 }
834978fd 5990 }
4fba7219 5991 }
a88df331
DG
5992
5993 /* Set ulimit for open files */
5994 set_ulimit();
fac6795d 5995 }
4063050c
MD
5996 /* init lttng_fd tracking must be done after set_ulimit. */
5997 lttng_fd_init();
fac6795d 5998
4a15001e
MD
5999 if (set_consumer_sockets(&ustconsumer64_data, rundir)) {
6000 retval = -1;
6001 goto exit_init_data;
67e40797
DG
6002 }
6003
4a15001e
MD
6004 if (set_consumer_sockets(&ustconsumer32_data, rundir)) {
6005 retval = -1;
6006 goto exit_init_data;
67e40797
DG
6007 }
6008
d6f42150 6009 /* Setup the needed unix socket */
4a15001e
MD
6010 if (init_daemon_socket()) {
6011 retval = -1;
6012 goto exit_init_data;
fac6795d
DG
6013 }
6014
6015 /* Set credentials to socket */
4a15001e
MD
6016 if (is_root && set_permissions(rundir)) {
6017 retval = -1;
6018 goto exit_init_data;
fac6795d
DG
6019 }
6020
5b8719f5
DG
6021 /* Get parent pid if -S, --sig-parent is specified. */
6022 if (opt_sig_parent) {
6023 ppid = getppid();
6024 }
6025
7a485870 6026 /* Setup the kernel pipe for waking up the kernel thread */
6620da75 6027 if (is_root && !opt_no_kernel) {
4a15001e
MD
6028 if (utils_create_pipe_cloexec(kernel_poll_pipe)) {
6029 retval = -1;
6030 goto exit_init_data;
6620da75 6031 }
7a485870
DG
6032 }
6033
099e26bd 6034 /* Setup the thread apps communication pipe. */
4a15001e
MD
6035 if (utils_create_pipe_cloexec(apps_cmd_pipe)) {
6036 retval = -1;
6037 goto exit_init_data;
099e26bd
DG
6038 }
6039
d0b96690 6040 /* Setup the thread apps notify communication pipe. */
4a15001e
MD
6041 if (utils_create_pipe_cloexec(apps_cmd_notify_pipe)) {
6042 retval = -1;
6043 goto exit_init_data;
d0b96690
DG
6044 }
6045
7972aab2
DG
6046 /* Initialize global buffer per UID and PID registry. */
6047 buffer_reg_init_uid_registry();
6048 buffer_reg_init_pid_registry();
6049
099e26bd 6050 /* Init UST command queue. */
8bdee6e2 6051 cds_wfcq_init(&ust_cmd_queue.head, &ust_cmd_queue.tail);
099e26bd 6052
273ea72c 6053 /*
4a15001e
MD
6054 * Get session list pointer. This pointer MUST NOT be free'd. This list
6055 * is statically declared in session.c
273ea72c 6056 */
54d01ffb 6057 session_list_ptr = session_get_list();
b5541356 6058
2f77fc4b 6059 cmd_init();
00e2e675 6060
ae9e45b3
DG
6061 /* Check for the application socket timeout env variable. */
6062 env_app_timeout = getenv(DEFAULT_APP_SOCKET_TIMEOUT_ENV);
6063 if (env_app_timeout) {
6064 app_socket_timeout = atoi(env_app_timeout);
6065 } else {
6066 app_socket_timeout = DEFAULT_APP_SOCKET_RW_TIMEOUT;
6067 }
6068
4a15001e
MD
6069 ret = write_pidfile();
6070 if (ret) {
6071 ERR("Error in write_pidfile");
6072 retval = -1;
6073 goto exit_init_data;
6074 }
6075 ret = write_agent_port();
6076 if (ret) {
6077 ERR("Error in write_agent_port");
6078 retval = -1;
6079 goto exit_init_data;
6080 }
35f90c40 6081
554831e7
MD
6082 /* Initialize communication library */
6083 lttcomm_init();
4a15001e 6084 /* Initialize TCP timeout values */
d831c249 6085 lttcomm_inet_init();
554831e7 6086
ef367a93 6087 if (load_session_init_data(&load_info) < 0) {
4a15001e
MD
6088 retval = -1;
6089 goto exit_init_data;
ef367a93
JG
6090 }
6091 load_info->path = opt_load_session_path;
6092
daa81f58 6093 /* Create health-check thread. */
1a1a34b4 6094 ret = pthread_create(&health_thread, default_pthread_attr(),
44a5e5eb 6095 thread_manage_health, (void *) NULL);
4a15001e
MD
6096 if (ret) {
6097 errno = ret;
44a5e5eb 6098 PERROR("pthread_create health");
4a15001e 6099 retval = -1;
44a5e5eb
DG
6100 goto exit_health;
6101 }
6102
daa81f58
JG
6103 /* notification_thread_data acquires the pipes' read side. */
6104 notification_thread_handle = notification_thread_handle_create(
6105 ust32_channel_monitor_pipe,
6106 ust64_channel_monitor_pipe,
6107 kernel_channel_monitor_pipe);
6108 if (!notification_thread_handle) {
6109 retval = -1;
6110 ERR("Failed to create notification thread shared data");
6111 stop_threads();
6112 goto exit_notification;
6113 }
6114
6115 /* Create notification thread. */
6116 ret = pthread_create(&notification_thread, default_pthread_attr(),
6117 thread_notification, notification_thread_handle);
6118 if (ret) {
6119 errno = ret;
6120 PERROR("pthread_create notification");
6121 retval = -1;
6122 stop_threads();
6123 goto exit_notification;
6124 }
6125
cf3af59e 6126 /* Create thread to manage the client socket */
1a1a34b4 6127 ret = pthread_create(&client_thread, default_pthread_attr(),
099e26bd 6128 thread_manage_clients, (void *) NULL);
4a15001e
MD
6129 if (ret) {
6130 errno = ret;
76d7553f 6131 PERROR("pthread_create clients");
4a15001e 6132 retval = -1;
daa81f58 6133 stop_threads();
cf3af59e
MD
6134 goto exit_client;
6135 }
fac6795d 6136
099e26bd 6137 /* Create thread to dispatch registration */
1a1a34b4 6138 ret = pthread_create(&dispatch_thread, default_pthread_attr(),
099e26bd 6139 thread_dispatch_ust_registration, (void *) NULL);
4a15001e
MD
6140 if (ret) {
6141 errno = ret;
76d7553f 6142 PERROR("pthread_create dispatch");
4a15001e 6143 retval = -1;
daa81f58 6144 stop_threads();
099e26bd
DG
6145 goto exit_dispatch;
6146 }
6147
6148 /* Create thread to manage application registration. */
1a1a34b4 6149 ret = pthread_create(&reg_apps_thread, default_pthread_attr(),
099e26bd 6150 thread_registration_apps, (void *) NULL);
4a15001e
MD
6151 if (ret) {
6152 errno = ret;
76d7553f 6153 PERROR("pthread_create registration");
4a15001e 6154 retval = -1;
daa81f58 6155 stop_threads();
099e26bd
DG
6156 goto exit_reg_apps;
6157 }
6158
cf3af59e 6159 /* Create thread to manage application socket */
1a1a34b4 6160 ret = pthread_create(&apps_thread, default_pthread_attr(),
54d01ffb 6161 thread_manage_apps, (void *) NULL);
4a15001e
MD
6162 if (ret) {
6163 errno = ret;
d0b96690 6164 PERROR("pthread_create apps");
4a15001e 6165 retval = -1;
daa81f58 6166 stop_threads();
d0b96690
DG
6167 goto exit_apps;
6168 }
6169
6170 /* Create thread to manage application notify socket */
1a1a34b4 6171 ret = pthread_create(&apps_notify_thread, default_pthread_attr(),
d0b96690 6172 ust_thread_manage_notify, (void *) NULL);
4a15001e
MD
6173 if (ret) {
6174 errno = ret;
f5c32ef1 6175 PERROR("pthread_create notify");
4a15001e 6176 retval = -1;
daa81f58 6177 stop_threads();
9563b0ad 6178 goto exit_apps_notify;
cf3af59e 6179 }
fac6795d 6180
022d91ba 6181 /* Create agent registration thread. */
1a1a34b4 6182 ret = pthread_create(&agent_reg_thread, default_pthread_attr(),
022d91ba 6183 agent_thread_manage_registration, (void *) NULL);
4a15001e
MD
6184 if (ret) {
6185 errno = ret;
022d91ba 6186 PERROR("pthread_create agent");
4a15001e 6187 retval = -1;
daa81f58 6188 stop_threads();
022d91ba 6189 goto exit_agent_reg;
4d076222
DG
6190 }
6191
6620da75
DG
6192 /* Don't start this thread if kernel tracing is not requested nor root */
6193 if (is_root && !opt_no_kernel) {
6194 /* Create kernel thread to manage kernel event */
1a1a34b4 6195 ret = pthread_create(&kernel_thread, default_pthread_attr(),
6620da75 6196 thread_manage_kernel, (void *) NULL);
4a15001e
MD
6197 if (ret) {
6198 errno = ret;
6620da75 6199 PERROR("pthread_create kernel");
4a15001e 6200 retval = -1;
daa81f58 6201 stop_threads();
6620da75
DG
6202 goto exit_kernel;
6203 }
ef367a93 6204 }
7a485870 6205
4a15001e 6206 /* Create session loading thread. */
1a1a34b4
MJ
6207 ret = pthread_create(&load_session_thread, default_pthread_attr(),
6208 thread_load_session, load_info);
4a15001e
MD
6209 if (ret) {
6210 errno = ret;
6211 PERROR("pthread_create load_session_thread");
6212 retval = -1;
daa81f58 6213 stop_threads();
4a15001e
MD
6214 goto exit_load_session;
6215 }
6216
6217 /*
6218 * This is where we start awaiting program completion (e.g. through
6219 * signal that asks threads to teardown).
6220 */
6221
6222 ret = pthread_join(load_session_thread, &status);
6223 if (ret) {
6224 errno = ret;
6225 PERROR("pthread_join load_session_thread");
6226 retval = -1;
6227 }
6228exit_load_session:
ef367a93
JG
6229
6230 if (is_root && !opt_no_kernel) {
6620da75 6231 ret = pthread_join(kernel_thread, &status);
4a15001e
MD
6232 if (ret) {
6233 errno = ret;
6620da75 6234 PERROR("pthread_join");
4a15001e 6235 retval = -1;
6620da75 6236 }
fac6795d 6237 }
cf3af59e 6238exit_kernel:
4a15001e 6239
022d91ba 6240 ret = pthread_join(agent_reg_thread, &status);
4a15001e
MD
6241 if (ret) {
6242 errno = ret;
022d91ba 6243 PERROR("pthread_join agent");
4a15001e 6244 retval = -1;
4d076222 6245 }
022d91ba 6246exit_agent_reg:
4a15001e 6247
9563b0ad 6248 ret = pthread_join(apps_notify_thread, &status);
4a15001e
MD
6249 if (ret) {
6250 errno = ret;
9563b0ad 6251 PERROR("pthread_join apps notify");
4a15001e 6252 retval = -1;
9563b0ad 6253 }
9563b0ad 6254exit_apps_notify:
4a15001e 6255
cf3af59e 6256 ret = pthread_join(apps_thread, &status);
4a15001e
MD
6257 if (ret) {
6258 errno = ret;
9563b0ad 6259 PERROR("pthread_join apps");
4a15001e 6260 retval = -1;
cf3af59e 6261 }
cf3af59e 6262exit_apps:
4a15001e 6263
099e26bd 6264 ret = pthread_join(reg_apps_thread, &status);
4a15001e
MD
6265 if (ret) {
6266 errno = ret;
76d7553f 6267 PERROR("pthread_join");
4a15001e 6268 retval = -1;
099e26bd 6269 }
099e26bd 6270exit_reg_apps:
4a15001e 6271
772b8f4d
MD
6272 /*
6273 * Join dispatch thread after joining reg_apps_thread to ensure
6274 * we don't leak applications in the queue.
6275 */
099e26bd 6276 ret = pthread_join(dispatch_thread, &status);
4a15001e
MD
6277 if (ret) {
6278 errno = ret;
76d7553f 6279 PERROR("pthread_join");
4a15001e 6280 retval = -1;
099e26bd 6281 }
099e26bd 6282exit_dispatch:
4a15001e 6283
cf3af59e 6284 ret = pthread_join(client_thread, &status);
4a15001e
MD
6285 if (ret) {
6286 errno = ret;
76d7553f 6287 PERROR("pthread_join");
4a15001e 6288 retval = -1;
cf3af59e 6289 }
daa81f58 6290
4a15001e 6291exit_client:
daa81f58
JG
6292 ret = pthread_join(notification_thread, &status);
6293 if (ret) {
6294 errno = ret;
6295 PERROR("pthread_join notification thread");
6296 retval = -1;
6297 }
a88df331 6298
daa81f58 6299exit_notification:
4a15001e
MD
6300 ret = pthread_join(health_thread, &status);
6301 if (ret) {
6302 errno = ret;
6303 PERROR("pthread_join health thread");
6304 retval = -1;
06f525de
DG
6305 }
6306
daa81f58 6307exit_health:
4a15001e 6308exit_init_data:
4d62fbf8
MD
6309 /*
6310 * Wait for all pending call_rcu work to complete before tearing
6311 * down data structures. call_rcu worker may be trying to
6312 * perform lookups in those structures.
6313 */
6314 rcu_barrier();
4a15001e
MD
6315 /*
6316 * sessiond_cleanup() is called when no other thread is running, except
6317 * the ht_cleanup thread, which is needed to destroy the hash tables.
6318 */
6319 rcu_thread_online();
6320 sessiond_cleanup();
6321 rcu_thread_offline();
6322 rcu_unregister_thread();
06f525de 6323
c4d5de9d
JG
6324 /*
6325 * Ensure all prior call_rcu are done. call_rcu callbacks may push
6326 * hash tables to the ht_cleanup thread. Therefore, we ensure that
6327 * the queue is empty before shutting down the clean-up thread.
6328 */
6329 rcu_barrier();
6330
5e97de00 6331 ret = fini_ht_cleanup_thread(&ht_cleanup_thread);
4a15001e 6332 if (ret) {
4a15001e 6333 retval = -1;
0b2dc8df 6334 }
daa81f58
JG
6335 lttng_pipe_destroy(ust32_channel_monitor_pipe);
6336 lttng_pipe_destroy(ust64_channel_monitor_pipe);
6337 lttng_pipe_destroy(kernel_channel_monitor_pipe);
0b2dc8df 6338exit_ht_cleanup:
4a15001e 6339
4a15001e
MD
6340 health_app_destroy(health_sessiond);
6341exit_health_sessiond_cleanup:
7567352f 6342exit_create_run_as_worker_cleanup:
4a15001e
MD
6343
6344exit_options:
6345 sessiond_cleanup_options();
6346
6347exit_set_signal_handler:
6348 if (!retval) {
cf3af59e 6349 exit(EXIT_SUCCESS);
4a15001e
MD
6350 } else {
6351 exit(EXIT_FAILURE);
67e40797 6352 }
fac6795d 6353}
This page took 0.56275 seconds and 5 git commands to generate.