Fix pointer dereference
[lttng-tools.git] / ltt-sessiond / main.c
CommitLineData
826d496d
MD
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
fac6795d
DG
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
91d76f53 8 *
fac6795d
DG
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
91d76f53 13 *
fac6795d
DG
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
fac6795d
DG
17 */
18
19#define _GNU_SOURCE
20#include <fcntl.h>
21#include <getopt.h>
22#include <grp.h>
23#include <limits.h>
24#include <pthread.h>
25#include <signal.h>
26#include <stdio.h>
27#include <stdlib.h>
28#include <string.h>
29#include <sys/ipc.h>
30#include <sys/shm.h>
31#include <sys/socket.h>
32#include <sys/stat.h>
33#include <sys/types.h>
34#include <unistd.h>
35
36#include <urcu/list.h> /* URCU list library (-lurcu) */
37#include <ust/ustctl.h> /* UST control lib (-lust) */
5b97ec60 38#include <lttng/lttng.h>
fac6795d
DG
39
40#include "liblttsessiondcomm.h"
41#include "ltt-sessiond.h"
75462a81 42#include "lttngerr.h"
20fe2104 43#include "kernel-ctl.h"
5b74c7b1 44#include "session.h"
fda89c9b 45#include "trace.h"
91d76f53 46#include "traceable-app.h"
fac6795d 47
5e16da05
MD
48/*
49 * TODO:
50 * teardown: signal SIGTERM handler -> write into pipe. Threads waits
51 * with epoll on pipe and on other pipes/sockets for commands. Main
52 * simply waits on pthread join.
53 */
54
75462a81 55/* Const values */
686204ab
MD
56const char default_home_dir[] = DEFAULT_HOME_DIR;
57const char default_tracing_group[] = DEFAULT_TRACING_GROUP;
58const char default_ust_sock_dir[] = DEFAULT_UST_SOCK_DIR;
59const char default_global_apps_pipe[] = DEFAULT_GLOBAL_APPS_PIPE;
60
fac6795d 61/* Static functions */
fac6795d 62static int check_existing_daemon(void);
471d1693 63static int ust_connect_app(pid_t pid);
fac6795d 64static int init_daemon_socket(void);
62bd06d8 65static int notify_apps(const char* name);
5461b305 66static int process_client_msg(struct command_ctx *cmd_ctx);
e065084a 67static int send_unix_sock(int sock, void *buf, size_t len);
62bd06d8 68static int set_signal_handler(void);
d6f42150 69static int set_permissions(void);
5461b305 70static int setup_lttng_msg(struct command_ctx *cmd_ctx, size_t size);
d6f42150
DG
71static int create_lttng_rundir(void);
72static int set_kconsumerd_sockets(void);
62bd06d8 73static void cleanup(void);
62bd06d8 74static void sighandler(int sig);
5461b305 75static void clean_command_ctx(struct command_ctx *cmd_ctx);
fac6795d 76
1fd70b72
DG
77static void *thread_manage_clients(void *data);
78static void *thread_manage_apps(void *data);
d6f42150 79static void *thread_manage_kconsumerd(void *data);
fac6795d 80
fac6795d 81/* Variables */
62bd06d8
DG
82int opt_verbose;
83int opt_quiet;
fac6795d
DG
84const char *progname;
85const char *opt_tracing_group;
5b8719f5 86static int opt_sig_parent;
fac6795d
DG
87static int opt_daemon;
88static int is_root; /* Set to 1 if the daemon is running as root */
5b8719f5 89static pid_t ppid;
fac6795d 90
d6f42150
DG
91static char apps_unix_sock_path[PATH_MAX]; /* Global application Unix socket path */
92static char client_unix_sock_path[PATH_MAX]; /* Global client Unix socket path */
93static char kconsumerd_err_unix_sock_path[PATH_MAX]; /* kconsumerd error Unix socket path */
94static char kconsumerd_cmd_unix_sock_path[PATH_MAX]; /* kconsumerd command Unix socket path */
fac6795d 95
d6f42150
DG
96static int client_sock;
97static int apps_sock;
98static int kconsumerd_err_sock;
20fe2104 99static int kernel_tracer_fd;
fac6795d 100
d6f42150
DG
101/*
102 * thread_manage_kconsumerd
103 *
104 * This thread manage the kconsumerd error sent
105 * back to the session daemon.
106 */
107static void *thread_manage_kconsumerd(void *data)
108{
109 int sock, ret;
110
111 DBG("[thread] Manage kconsumerd started");
112
113 ret = lttcomm_listen_unix_sock(kconsumerd_err_sock);
114 if (ret < 0) {
115 goto error;
116 }
117
118 sock = lttcomm_accept_unix_sock(kconsumerd_err_sock);
119 if (sock < 0) {
120 goto error;
121 }
122
123 while (1) {
124 //ret = lttcomm_recv_unix_sock(sock, &lsm, sizeof(lsm));
125 if (ret <= 0) {
126 /* TODO: Consumerd died? */
127 continue;
128 }
129 }
130
131error:
132 return NULL;
133}
134
fac6795d
DG
135/*
136 * thread_manage_apps
137 *
138 * This thread manage the application socket communication
139 */
140static void *thread_manage_apps(void *data)
141{
142 int sock, ret;
fac6795d
DG
143
144 /* TODO: Something more elegant is needed but fine for now */
5e16da05
MD
145 /* FIXME: change all types to either uint8_t, uint32_t, uint64_t
146 * for 32-bit vs 64-bit compat processes. */
147 /* replicate in ust with version number */
686204ab 148 struct {
fac6795d 149 int reg; /* 1:register, 0:unregister */
686204ab
MD
150 pid_t pid;
151 uid_t uid;
152 } reg_msg;
fac6795d 153
e07ae692
DG
154 DBG("[thread] Manage apps started");
155
d6f42150 156 ret = lttcomm_listen_unix_sock(apps_sock);
fac6795d
DG
157 if (ret < 0) {
158 goto error;
159 }
160
5e16da05
MD
161 /* Notify all applications to register */
162 notify_apps(default_global_apps_pipe);
163
fac6795d 164 while (1) {
894be886 165 DBG("Accepting application registration");
fac6795d 166 /* Blocking call, waiting for transmission */
d6f42150 167 sock = lttcomm_accept_unix_sock(apps_sock);
fac6795d
DG
168 if (sock < 0) {
169 goto error;
170 }
171
172 /* Basic recv here to handle the very simple data
173 * that the libust send to register (reg_msg).
174 */
175 ret = recv(sock, &reg_msg, sizeof(reg_msg), 0);
176 if (ret < 0) {
177 perror("recv");
178 continue;
179 }
180
181 /* Add application to the global traceable list */
182 if (reg_msg.reg == 1) {
183 /* Registering */
91d76f53
DG
184 ret = register_traceable_app(reg_msg.pid, reg_msg.uid);
185 if (ret < 0) {
186 /* register_traceable_app only return an error with
187 * ENOMEM. At this point, we better stop everything.
188 */
189 goto error;
190 }
fac6795d
DG
191 } else {
192 /* Unregistering */
91d76f53 193 unregister_traceable_app(reg_msg.pid);
fac6795d
DG
194 }
195 }
196
197error:
198
199 return NULL;
200}
201
202/*
203 * thread_manage_clients
204 *
205 * This thread manage all clients request using the unix
206 * client socket for communication.
207 */
208static void *thread_manage_clients(void *data)
209{
210 int sock, ret;
5461b305 211 struct command_ctx *cmd_ctx;
fac6795d 212
e07ae692
DG
213 DBG("[thread] Manage client started");
214
d6f42150 215 ret = lttcomm_listen_unix_sock(client_sock);
fac6795d
DG
216 if (ret < 0) {
217 goto error;
218 }
219
5b8719f5
DG
220 /* Notify parent pid that we are ready
221 * to accept command for client side.
222 */
223 if (opt_sig_parent) {
224 kill(ppid, SIGCHLD);
225 }
226
fac6795d
DG
227 while (1) {
228 /* Blocking call, waiting for transmission */
5461b305 229 DBG("Accepting client command ...");
d6f42150 230 sock = lttcomm_accept_unix_sock(client_sock);
fac6795d
DG
231 if (sock < 0) {
232 goto error;
233 }
234
5461b305
DG
235 /* Allocate context command to process the client request */
236 cmd_ctx = malloc(sizeof(struct command_ctx));
237
238 /* Allocate data buffer for reception */
239 cmd_ctx->lsm = malloc(sizeof(struct lttcomm_session_msg));
240 cmd_ctx->llm = NULL;
241 cmd_ctx->session = NULL;
242
fac6795d
DG
243 /*
244 * Data is received from the lttng client. The struct
5461b305
DG
245 * lttcomm_session_msg (lsm) contains the command and data request of
246 * the client.
fac6795d 247 */
5461b305
DG
248 DBG("Receiving data from client ...");
249 ret = lttcomm_recv_unix_sock(sock, cmd_ctx->lsm, sizeof(struct lttcomm_session_msg));
87378cf5 250 if (ret <= 0) {
fac6795d
DG
251 continue;
252 }
253
5461b305
DG
254 /*
255 * This function dispatch the work to the kernel or userspace tracer
256 * libs and fill the lttcomm_lttng_msg data structure of all the needed
257 * informations for the client. The command context struct contains
258 * everything this function may needs.
fac6795d 259 */
5461b305 260 ret = process_client_msg(cmd_ctx);
e065084a 261 if (ret < 0) {
5461b305
DG
262 /* TODO: Inform client somehow of the fatal error. At this point,
263 * ret < 0 means that a malloc failed (ENOMEM). */
e065084a 264 /* Error detected but still accept command */
5461b305 265 clean_command_ctx(cmd_ctx);
e065084a 266 continue;
fac6795d 267 }
5461b305 268
894be886
DG
269 DBG("Sending response (size: %d, retcode: %d)",
270 cmd_ctx->lttng_msg_size, cmd_ctx->llm->ret_code);
5461b305
DG
271 ret = send_unix_sock(sock, cmd_ctx->llm, cmd_ctx->lttng_msg_size);
272 if (ret < 0) {
273 ERR("Failed to send data back to client");
274 }
275
276 clean_command_ctx(cmd_ctx);
fac6795d
DG
277 }
278
279error:
280 return NULL;
281}
282
e065084a
DG
283/*
284 * send_unix_sock
285 *
286 * Send data on a unix socket using the liblttsessiondcomm API.
287 *
288 * Return lttcomm error code.
289 */
290static int send_unix_sock(int sock, void *buf, size_t len)
291{
292 /* Check valid length */
293 if (len <= 0) {
294 return -1;
295 }
296
297 return lttcomm_send_unix_sock(sock, buf, len);
298}
299
5461b305
DG
300/*
301 * clean_command_ctx
302 *
303 * Free memory of a command context structure.
304 */
305static void clean_command_ctx(struct command_ctx *cmd_ctx)
306{
307 DBG("Clean command context structure %p", cmd_ctx);
308 if (cmd_ctx) {
309 if (cmd_ctx->llm) {
310 free(cmd_ctx->llm);
311 }
312 if (cmd_ctx->lsm) {
313 free(cmd_ctx->lsm);
314 }
315 free(cmd_ctx);
316 cmd_ctx = NULL;
317 }
318}
319
fac6795d 320/*
471d1693 321 * ust_connect_app
fac6795d
DG
322 *
323 * Return a socket connected to the libust communication socket
324 * of the application identified by the pid.
379473d2
DG
325 *
326 * If the pid is not found in the traceable list,
327 * return -1 to indicate error.
fac6795d 328 */
471d1693 329static int ust_connect_app(pid_t pid)
fac6795d 330{
91d76f53
DG
331 int sock;
332 struct ltt_traceable_app *lta;
379473d2 333
e07ae692
DG
334 DBG("Connect to application pid %d", pid);
335
91d76f53
DG
336 lta = find_app_by_pid(pid);
337 if (lta == NULL) {
338 /* App not found */
7442b2ba 339 DBG("Application pid %d not found", pid);
379473d2
DG
340 return -1;
341 }
fac6795d 342
91d76f53 343 sock = ustctl_connect_pid(lta->pid);
fac6795d 344 if (sock < 0) {
75462a81 345 ERR("Fail connecting to the PID %d\n", pid);
fac6795d
DG
346 }
347
348 return sock;
349}
350
351/*
352 * notify_apps
353 *
354 * Notify apps by writing 42 to a named pipe using name.
355 * Every applications waiting for a ltt-sessiond will be notified
356 * and re-register automatically to the session daemon.
357 *
358 * Return open or write error value.
359 */
360static int notify_apps(const char *name)
361{
362 int fd;
363 int ret = -1;
364
e07ae692
DG
365 DBG("Notify the global application pipe");
366
fac6795d
DG
367 /* Try opening the global pipe */
368 fd = open(name, O_WRONLY);
369 if (fd < 0) {
370 goto error;
371 }
372
373 /* Notify by writing on the pipe */
374 ret = write(fd, "42", 2);
375 if (ret < 0) {
376 perror("write");
377 }
378
379error:
380 return ret;
381}
382
e065084a 383/*
5461b305 384 * setup_lttng_msg
ca95a216 385 *
5461b305
DG
386 * Setup the outgoing data buffer for the response (llm) by allocating the
387 * right amount of memory and copying the original information from the lsm
388 * structure.
ca95a216
DG
389 *
390 * Return total size of the buffer pointed by buf.
391 */
5461b305 392static int setup_lttng_msg(struct command_ctx *cmd_ctx, size_t size)
ca95a216 393{
5461b305 394 int ret, buf_size, trace_name_size;
ca95a216 395
5461b305
DG
396 /*
397 * Check for the trace_name. If defined, it's part of the payload data of
398 * the llm structure.
399 */
400 trace_name_size = strlen(cmd_ctx->lsm->trace_name);
401 buf_size = trace_name_size + size;
402
403 cmd_ctx->llm = malloc(sizeof(struct lttcomm_lttng_msg) + buf_size);
404 if (cmd_ctx->llm == NULL) {
ca95a216 405 perror("malloc");
5461b305 406 ret = -ENOMEM;
ca95a216
DG
407 goto error;
408 }
409
5461b305
DG
410 /* Copy common data */
411 cmd_ctx->llm->cmd_type = cmd_ctx->lsm->cmd_type;
412 cmd_ctx->llm->pid = cmd_ctx->lsm->pid;
413 if (!uuid_is_null(cmd_ctx->lsm->session_uuid)) {
414 uuid_copy(cmd_ctx->llm->session_uuid, cmd_ctx->lsm->session_uuid);
415 }
416
417 cmd_ctx->llm->trace_name_offset = trace_name_size;
418 cmd_ctx->llm->data_size = size;
419 cmd_ctx->lttng_msg_size = sizeof(struct lttcomm_lttng_msg) + buf_size;
420
421 /* Copy trace name to the llm structure. Begining of the payload. */
422 memcpy(cmd_ctx->llm->payload, cmd_ctx->lsm->trace_name, trace_name_size);
ca95a216
DG
423
424 return buf_size;
425
426error:
427 return ret;
428}
429
fac6795d
DG
430/*
431 * process_client_msg
432 *
5461b305
DG
433 * Process the command requested by the lttng client within the command
434 * context structure. This function make sure that the return structure (llm)
435 * is set and ready for transmission before returning.
fac6795d 436 *
e065084a 437 * Return any error encountered or 0 for success.
fac6795d 438 */
5461b305 439static int process_client_msg(struct command_ctx *cmd_ctx)
fac6795d 440{
5461b305 441 int ret;
fac6795d 442
5461b305 443 DBG("Processing client command %d", cmd_ctx->lsm->cmd_type);
fac6795d 444
379473d2 445 /* Check command that needs a session */
5461b305 446 switch (cmd_ctx->lsm->cmd_type) {
5e16da05
MD
447 case LTTNG_CREATE_SESSION:
448 case LTTNG_LIST_SESSIONS:
449 case UST_LIST_APPS:
450 break;
451 default:
5461b305
DG
452 cmd_ctx->session = find_session_by_uuid(cmd_ctx->lsm->session_uuid);
453 if (cmd_ctx->session == NULL) {
379473d2 454 ret = LTTCOMM_SELECT_SESS;
5461b305 455 goto error;
379473d2 456 }
5e16da05 457 break;
379473d2
DG
458 }
459
20fe2104
DG
460 /* Check command for kernel tracing */
461 switch (cmd_ctx->lsm->cmd_type) {
462 case KERNEL_CREATE_SESSION:
463 case KERNEL_CREATE_CHANNEL:
464 if (kernel_tracer_fd == 0) {
465 ret = LTTCOMM_KERN_NA;
466 goto error;
467 }
468 break;
469 }
470
471d1693 471 /* Connect to ust apps if available pid */
5461b305 472 if (cmd_ctx->lsm->pid > 0) {
471d1693 473 /* Connect to app using ustctl API */
5461b305
DG
474 cmd_ctx->ust_sock = ust_connect_app(cmd_ctx->lsm->pid);
475 if (cmd_ctx->ust_sock < 0) {
471d1693 476 ret = LTTCOMM_NO_TRACEABLE;
5461b305 477 goto error;
471d1693
DG
478 }
479 }
480
fac6795d 481 /* Process by command type */
5461b305 482 switch (cmd_ctx->lsm->cmd_type) {
20fe2104
DG
483 case KERNEL_CREATE_SESSION:
484 {
485 ret = setup_lttng_msg(cmd_ctx, 0);
486 if (ret < 0) {
487 goto setup_error;
488 }
489
490 DBG("Creating kernel session");
491
492 ret = kernel_create_session(cmd_ctx, kernel_tracer_fd);
493 if (ret < 0) {
494 ret = LTTCOMM_KERN_SESS_FAIL;
495 goto error;
496 }
497
498 ret = LTTCOMM_OK;
499 break;
500 }
501 case KERNEL_CREATE_CHANNEL:
502 {
503 ret = setup_lttng_msg(cmd_ctx, 0);
504 if (ret < 0) {
505 goto setup_error;
506 }
507
508 DBG("Creating kernel session");
509
510 ret = kernel_create_channel(cmd_ctx);
511 if (ret < 0) {
512 ret = LTTCOMM_KERN_CHAN_FAIL;
513 goto error;
514 }
515
516 ret = LTTCOMM_OK;
517 break;
518 }
894be886
DG
519 case KERNEL_ENABLE_EVENT:
520 {
521 /* Setup lttng message with no payload */
522 ret = setup_lttng_msg(cmd_ctx, 0);
523 if (ret < 0) {
524 goto setup_error;
525 }
526
527 DBG("Enabling kernel event %s", cmd_ctx->lsm->u.event.event_name);
528
529 ret = LTTCOMM_OK;
530 break;
531 }
5e16da05
MD
532 case LTTNG_CREATE_SESSION:
533 {
5461b305
DG
534 /* Setup lttng message with no payload */
535 ret = setup_lttng_msg(cmd_ctx, 0);
536 if (ret < 0) {
537 goto setup_error;
538 }
539
540 ret = create_session(cmd_ctx->lsm->session_name, &cmd_ctx->llm->session_uuid);
5e16da05
MD
541 if (ret < 0) {
542 if (ret == -1) {
543 ret = LTTCOMM_EXIST_SESS;
544 } else {
aaf97519 545 ret = LTTCOMM_FATAL;
aaf97519 546 }
5461b305 547 goto error;
8028d920 548 }
1657e9bb 549
5461b305 550 ret = LTTCOMM_OK;
5e16da05
MD
551 break;
552 }
553 case LTTNG_DESTROY_SESSION:
554 {
5461b305
DG
555 /* Setup lttng message with no payload */
556 ret = setup_lttng_msg(cmd_ctx, 0);
557 if (ret < 0) {
558 goto setup_error;
559 }
560
561 ret = destroy_session(&cmd_ctx->lsm->session_uuid);
5e16da05
MD
562 if (ret < 0) {
563 ret = LTTCOMM_NO_SESS;
5461b305 564 goto error;
5e16da05 565 }
1657e9bb 566
5461b305
DG
567 ret = LTTCOMM_OK;
568 break;
5e16da05
MD
569 }
570 case LTTNG_LIST_TRACES:
571 {
5461b305 572 unsigned int trace_count;
1657e9bb 573
5461b305 574 trace_count = get_trace_count_per_session(cmd_ctx->session);
5e16da05
MD
575 if (trace_count == 0) {
576 ret = LTTCOMM_NO_TRACE;
5461b305 577 goto error;
1657e9bb 578 }
df0da139 579
5461b305
DG
580 ret = setup_lttng_msg(cmd_ctx, sizeof(struct lttng_trace) * trace_count);
581 if (ret < 0) {
582 goto setup_error;
df0da139 583 }
ca95a216 584
5461b305
DG
585 get_traces_per_session(cmd_ctx->session,
586 (struct lttng_trace *)(cmd_ctx->llm->payload));
587
588 ret = LTTCOMM_OK;
5e16da05
MD
589 break;
590 }
591 case UST_CREATE_TRACE:
592 {
5461b305
DG
593 /* Setup lttng message with no payload */
594 ret = setup_lttng_msg(cmd_ctx, 0);
5e16da05 595 if (ret < 0) {
5461b305 596 goto setup_error;
fac6795d 597 }
ce3d728c 598
5461b305
DG
599 ret = ust_create_trace(cmd_ctx);
600 if (ret < 0) {
601 goto setup_error;
602 }
603 break;
5e16da05
MD
604 }
605 case UST_LIST_APPS:
606 {
5461b305
DG
607 unsigned int app_count;
608
609 app_count = get_app_count();
610 DBG("Traceable application count : %d", app_count);
5e16da05
MD
611 if (app_count == 0) {
612 ret = LTTCOMM_NO_APPS;
5461b305 613 goto error;
ce3d728c 614 }
520ff687 615
5461b305
DG
616 ret = setup_lttng_msg(cmd_ctx, sizeof(pid_t) * app_count);
617 if (ret < 0) {
618 goto setup_error;
520ff687 619 }
57167058 620
5461b305 621 get_app_list_pids((pid_t *)(cmd_ctx->llm->payload));
5e16da05 622
5461b305 623 ret = LTTCOMM_OK;
5e16da05
MD
624 break;
625 }
626 case UST_START_TRACE:
627 {
5461b305
DG
628 /* Setup lttng message with no payload */
629 ret = setup_lttng_msg(cmd_ctx, 0);
630 if (ret < 0) {
631 goto setup_error;
632 }
ca95a216 633
5461b305
DG
634 ret = ust_start_trace(cmd_ctx);
635 if (ret < 0) {
636 goto setup_error;
637 }
638 break;
5e16da05
MD
639 }
640 case UST_STOP_TRACE:
641 {
5461b305
DG
642 /* Setup lttng message with no payload */
643 ret = setup_lttng_msg(cmd_ctx, 0);
644 if (ret < 0) {
645 goto setup_error;
646 }
ca95a216 647
5461b305
DG
648 ret = ust_stop_trace(cmd_ctx);
649 if (ret < 0) {
650 goto setup_error;
651 }
652 break;
5e16da05
MD
653 }
654 case LTTNG_LIST_SESSIONS:
655 {
5461b305
DG
656 unsigned int session_count;
657
658 session_count = get_session_count();
5e16da05
MD
659 if (session_count == 0) {
660 ret = LTTCOMM_NO_SESS;
5461b305 661 goto error;
57167058 662 }
5e16da05 663
5461b305
DG
664 ret = setup_lttng_msg(cmd_ctx, sizeof(struct lttng_session) * session_count);
665 if (ret < 0) {
666 goto setup_error;
e065084a 667 }
5e16da05 668
5461b305 669 get_lttng_session((struct lttng_session *)(cmd_ctx->llm->payload));
5e16da05 670
5461b305 671 ret = LTTCOMM_OK;
5e16da05
MD
672 break;
673 }
674 default:
675 /* Undefined command */
5461b305
DG
676 ret = setup_lttng_msg(cmd_ctx, 0);
677 if (ret < 0) {
678 goto setup_error;
679 }
680
5e16da05 681 ret = LTTCOMM_UND;
5461b305 682 break;
fac6795d
DG
683 }
684
5461b305
DG
685 /* Set return code */
686 cmd_ctx->llm->ret_code = ret;
ca95a216 687
e065084a
DG
688 return ret;
689
5461b305 690error:
5461b305
DG
691 if (cmd_ctx->llm == NULL) {
692 DBG("Missing llm structure. Allocating one.");
894be886 693 if (setup_lttng_msg(cmd_ctx, 0) < 0) {
5461b305
DG
694 goto setup_error;
695 }
696 }
e065084a 697 /* Notify client of error */
5461b305 698 cmd_ctx->llm->ret_code = ret;
e065084a 699
5461b305 700setup_error:
8028d920 701 return ret;
fac6795d
DG
702}
703
704/*
705 * usage function on stderr
706 */
707static void usage(void)
708{
b716ce68 709 fprintf(stderr, "Usage: %s OPTIONS\n\nOptions:\n", progname);
d6f42150
DG
710 fprintf(stderr, " -h, --help Display this usage.\n");
711 fprintf(stderr, " -c, --client-sock PATH Specify path for the client unix socket\n");
712 fprintf(stderr, " -a, --apps-sock PATH Specify path for apps unix socket\n");
713 fprintf(stderr, " --kconsumerd-err-sock PATH Specify path for the kernel consumer error socket\n");
714 fprintf(stderr, " --kconsumerd-cmd-sock PATH Specify path for the kernel consumer command socket\n");
715 fprintf(stderr, " -d, --daemonize Start as a daemon.\n");
716 fprintf(stderr, " -g, --group NAME Specify the tracing group name. (default: tracing)\n");
717 fprintf(stderr, " -V, --version Show version number.\n");
718 fprintf(stderr, " -S, --sig-parent Send SIGCHLD to parent pid to notify readiness.\n");
719 fprintf(stderr, " -q, --quiet No output at all.\n");
720 fprintf(stderr, " -v, --verbose Verbose mode. Activate DBG() macro.\n");
fac6795d
DG
721}
722
723/*
724 * daemon argument parsing
725 */
726static int parse_args(int argc, char **argv)
727{
728 int c;
729
730 static struct option long_options[] = {
731 { "client-sock", 1, 0, 'c' },
732 { "apps-sock", 1, 0, 'a' },
d6f42150
DG
733 { "kconsumerd-cmd-sock", 1, 0, 0 },
734 { "kconsumerd-err-sock", 1, 0, 0 },
fac6795d 735 { "daemonize", 0, 0, 'd' },
5b8719f5 736 { "sig-parent", 0, 0, 'S' },
fac6795d
DG
737 { "help", 0, 0, 'h' },
738 { "group", 1, 0, 'g' },
739 { "version", 0, 0, 'V' },
75462a81 740 { "quiet", 0, 0, 'q' },
3f9947db 741 { "verbose", 0, 0, 'v' },
fac6795d
DG
742 { NULL, 0, 0, 0 }
743 };
744
745 while (1) {
746 int option_index = 0;
d6f42150 747 c = getopt_long(argc, argv, "dhqvVS" "a:c:g:s:E:C:", long_options, &option_index);
fac6795d
DG
748 if (c == -1) {
749 break;
750 }
751
752 switch (c) {
753 case 0:
754 fprintf(stderr, "option %s", long_options[option_index].name);
755 if (optarg) {
756 fprintf(stderr, " with arg %s\n", optarg);
757 }
758 break;
b716ce68 759 case 'c':
fac6795d
DG
760 snprintf(client_unix_sock_path, PATH_MAX, "%s", optarg);
761 break;
762 case 'a':
763 snprintf(apps_unix_sock_path, PATH_MAX, "%s", optarg);
764 break;
765 case 'd':
766 opt_daemon = 1;
767 break;
768 case 'g':
769 opt_tracing_group = strdup(optarg);
770 break;
771 case 'h':
772 usage();
773 exit(EXIT_FAILURE);
774 case 'V':
775 fprintf(stdout, "%s\n", VERSION);
776 exit(EXIT_SUCCESS);
5b8719f5
DG
777 case 'S':
778 opt_sig_parent = 1;
779 break;
d6f42150
DG
780 case 'E':
781 snprintf(kconsumerd_err_unix_sock_path, PATH_MAX, "%s", optarg);
782 break;
783 case 'C':
784 snprintf(kconsumerd_cmd_unix_sock_path, PATH_MAX, "%s", optarg);
785 break;
75462a81
DG
786 case 'q':
787 opt_quiet = 1;
788 break;
3f9947db
DG
789 case 'v':
790 opt_verbose = 1;
791 break;
fac6795d
DG
792 default:
793 /* Unknown option or other error.
794 * Error is printed by getopt, just return */
795 return -1;
796 }
797 }
798
799 return 0;
800}
801
802/*
803 * init_daemon_socket
804 *
805 * Creates the two needed socket by the daemon.
d6f42150
DG
806 * apps_sock - The communication socket for all UST apps.
807 * client_sock - The communication of the cli tool (lttng).
fac6795d
DG
808 */
809static int init_daemon_socket()
810{
811 int ret = 0;
812 mode_t old_umask;
813
814 old_umask = umask(0);
815
816 /* Create client tool unix socket */
d6f42150
DG
817 client_sock = lttcomm_create_unix_sock(client_unix_sock_path);
818 if (client_sock < 0) {
819 ERR("Create unix sock failed: %s", client_unix_sock_path);
fac6795d
DG
820 ret = -1;
821 goto end;
822 }
823
824 /* File permission MUST be 660 */
825 ret = chmod(client_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
826 if (ret < 0) {
d6f42150 827 ERR("Set file permissions failed: %s", client_unix_sock_path);
fac6795d
DG
828 perror("chmod");
829 goto end;
830 }
831
832 /* Create the application unix socket */
d6f42150
DG
833 apps_sock = lttcomm_create_unix_sock(apps_unix_sock_path);
834 if (apps_sock < 0) {
835 ERR("Create unix sock failed: %s", apps_unix_sock_path);
fac6795d
DG
836 ret = -1;
837 goto end;
838 }
839
d6f42150 840 /* File permission MUST be 666 */
fac6795d
DG
841 ret = chmod(apps_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
842 if (ret < 0) {
d6f42150 843 ERR("Set file permissions failed: %s", apps_unix_sock_path);
fac6795d
DG
844 perror("chmod");
845 goto end;
846 }
847
848end:
849 umask(old_umask);
850 return ret;
851}
852
853/*
854 * check_existing_daemon
855 *
856 * Check if the global socket is available.
62bd06d8 857 * If yes, error is returned.
fac6795d
DG
858 */
859static int check_existing_daemon()
860{
861 int ret;
862
863 ret = access(client_unix_sock_path, F_OK);
864 if (ret == 0) {
865 ret = access(apps_unix_sock_path, F_OK);
866 }
867
868 return ret;
869}
870
871/*
62bd06d8 872 * get_home_dir
fac6795d 873 *
62bd06d8
DG
874 * Return pointer to home directory path using
875 * the env variable HOME.
fac6795d 876 *
62bd06d8 877 * Default : /tmp
fac6795d
DG
878 */
879static const char *get_home_dir(void)
880{
881 const char *home_path;
882
686204ab 883 if ((home_path = (const char *) getenv("HOME")) == NULL) {
fac6795d
DG
884 home_path = default_home_dir;
885 }
886
887 return home_path;
888}
889
890/*
d6f42150 891 * set_permissions
fac6795d 892 *
5e16da05
MD
893 * Set the tracing group gid onto the client socket.
894 *
895 * Race window between mkdir and chown is OK because we are going from
896 * less permissive (root.root) to more permissive (root.tracing).
fac6795d 897 */
d6f42150 898static int set_permissions(void)
fac6795d
DG
899{
900 int ret;
901 struct group *grp;
902
903 /* Decide which group name to use */
904 (opt_tracing_group != NULL) ?
905 (grp = getgrnam(opt_tracing_group)) :
906 (grp = getgrnam(default_tracing_group));
907
908 if (grp == NULL) {
75462a81 909 ERR("Missing tracing group. Aborting execution.\n");
fac6795d
DG
910 ret = -1;
911 goto end;
912 }
913
d6f42150
DG
914 /* Set lttng run dir */
915 ret = chown(LTTNG_RUNDIR, 0, grp->gr_gid);
916 if (ret < 0) {
917 ERR("Unable to set group on " LTTNG_RUNDIR);
918 perror("chown");
919 }
920
921 /* lttng client socket path */
fac6795d
DG
922 ret = chown(client_unix_sock_path, 0, grp->gr_gid);
923 if (ret < 0) {
d6f42150
DG
924 ERR("Unable to set group on %s", client_unix_sock_path);
925 perror("chown");
926 }
927
928 /* kconsumerd error socket path */
929 ret = chown(kconsumerd_err_unix_sock_path, 0, grp->gr_gid);
930 if (ret < 0) {
931 ERR("Unable to set group on %s", kconsumerd_err_unix_sock_path);
fac6795d
DG
932 perror("chown");
933 }
934
d6f42150 935 DBG("All permissions are set");
e07ae692 936
fac6795d
DG
937end:
938 return ret;
939}
940
d6f42150
DG
941/*
942 * create_lttng_rundir
943 *
944 * Create the lttng run directory needed for all
945 * global sockets and pipe.
946 */
947static int create_lttng_rundir(void)
948{
949 int ret;
950
951 ret = mkdir(LTTNG_RUNDIR, S_IRWXU | S_IRWXG );
952 if (ret < 0) {
953 ERR("Unable to create " LTTNG_RUNDIR);
954 goto error;
955 }
956
957error:
958 return ret;
959}
960
20fe2104
DG
961/*
962 * init_kernel_tracer
963 *
964 * Setup necessary data for kernel tracer action.
965 */
966static void init_kernel_tracer(void)
967{
968 /* Set the global kernel tracer fd */
969 kernel_tracer_fd = open(DEFAULT_KERNEL_TRACER_PATH, O_RDWR);
970 if (kernel_tracer_fd < 0) {
971 WARN("No kernel tracer available");
972 kernel_tracer_fd = 0;
973 }
974}
975
d6f42150
DG
976/*
977 * set_kconsumerd_sockets
978 *
979 * Setup sockets and directory needed by the kconsumerd
980 * communication with the session daemon.
981 */
982static int set_kconsumerd_sockets(void)
983{
984 int ret;
985
986 if (strlen(kconsumerd_err_unix_sock_path) == 0) {
987 snprintf(kconsumerd_err_unix_sock_path, PATH_MAX, KCONSUMERD_ERR_SOCK_PATH);
988 }
989
990 if (strlen(kconsumerd_cmd_unix_sock_path) == 0) {
991 snprintf(kconsumerd_cmd_unix_sock_path, PATH_MAX, KCONSUMERD_CMD_SOCK_PATH);
992 }
993
994 ret = mkdir(KCONSUMERD_PATH, S_IRWXU | S_IRWXG);
995 if (ret < 0) {
996 ERR("Failed to create " KCONSUMERD_PATH);
997 goto error;
998 }
999
1000 /* Create the kconsumerd error unix socket */
1001 kconsumerd_err_sock = lttcomm_create_unix_sock(kconsumerd_err_unix_sock_path);
1002 if (kconsumerd_err_sock < 0) {
1003 ERR("Create unix sock failed: %s", kconsumerd_err_unix_sock_path);
1004 ret = -1;
1005 goto error;
1006 }
1007
1008 /* File permission MUST be 660 */
1009 ret = chmod(kconsumerd_err_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
1010 if (ret < 0) {
1011 ERR("Set file permissions failed: %s", kconsumerd_err_unix_sock_path);
1012 perror("chmod");
1013 goto error;
1014 }
1015
1016error:
1017 return ret;
1018}
1019
fac6795d 1020/*
62bd06d8 1021 * set_signal_handler
fac6795d 1022 *
62bd06d8 1023 * Setup signal handler for :
fac6795d
DG
1024 * SIGINT, SIGTERM, SIGPIPE
1025 */
1026static int set_signal_handler(void)
1027{
1028 int ret = 0;
1029 struct sigaction sa;
1030 sigset_t sigset;
1031
1032 if ((ret = sigemptyset(&sigset)) < 0) {
1033 perror("sigemptyset");
1034 return ret;
1035 }
1036
1037 sa.sa_handler = sighandler;
1038 sa.sa_mask = sigset;
1039 sa.sa_flags = 0;
1040 if ((ret = sigaction(SIGTERM, &sa, NULL)) < 0) {
1041 perror("sigaction");
1042 return ret;
1043 }
1044
1045 if ((ret = sigaction(SIGINT, &sa, NULL)) < 0) {
1046 perror("sigaction");
1047 return ret;
1048 }
1049
1050 if ((ret = sigaction(SIGPIPE, &sa, NULL)) < 0) {
1051 perror("sigaction");
1052 return ret;
1053 }
1054
e07ae692
DG
1055 DBG("Signal handler set for SIGTERM, SIGPIPE and SIGINT");
1056
fac6795d
DG
1057 return ret;
1058}
1059
1060/**
62bd06d8 1061 * sighandler
fac6795d 1062 *
62bd06d8 1063 * Signal handler for the daemon
fac6795d
DG
1064 */
1065static void sighandler(int sig)
1066{
1067 switch (sig) {
1068 case SIGPIPE:
e07ae692 1069 DBG("SIGPIPE catched");
87378cf5 1070 return;
fac6795d 1071 case SIGINT:
e07ae692
DG
1072 DBG("SIGINT catched");
1073 cleanup();
1074 break;
fac6795d 1075 case SIGTERM:
e07ae692 1076 DBG("SIGTERM catched");
fac6795d 1077 cleanup();
686204ab 1078 break;
fac6795d
DG
1079 default:
1080 break;
1081 }
1082
1083 exit(EXIT_SUCCESS);
1084}
1085
1086/*
62bd06d8 1087 * cleanup
fac6795d 1088 *
62bd06d8 1089 * Cleanup the daemon on exit
fac6795d
DG
1090 */
1091static void cleanup()
1092{
d6f42150
DG
1093 int ret;
1094 char *cmd;
1095
e07ae692
DG
1096 DBG("Cleaning up");
1097
fac6795d 1098 /* <fun> */
b716ce68
DG
1099 MSG("\n%c[%d;%dm*** assert failed *** ==> %c[%dm", 27,1,31,27,0);
1100 MSG("%c[%d;%dmMatthew, BEET driven development works!%c[%dm",27,1,33,27,0);
fac6795d
DG
1101 /* </fun> */
1102
1103 unlink(client_unix_sock_path);
1104 unlink(apps_unix_sock_path);
d6f42150
DG
1105 unlink(kconsumerd_err_unix_sock_path);
1106
1107 ret = asprintf(&cmd, "rm -rf " LTTNG_RUNDIR);
1108 if (ret < 0) {
1109 ERR("asprintf failed. Something is really wrong!");
1110 }
1111
1112 /* Remove lttng run directory */
1113 ret = system(cmd);
1114 if (ret < 0) {
1115 ERR("Unable to clean " LTTNG_RUNDIR);
1116 }
fac6795d
DG
1117}
1118
1119/*
1120 * main
1121 */
1122int main(int argc, char **argv)
1123{
1124 int i;
1125 int ret = 0;
1126 void *status;
1127 pthread_t threads[2];
1128
1129 /* Parse arguments */
1130 progname = argv[0];
1131 if ((ret = parse_args(argc, argv) < 0)) {
1132 goto error;
1133 }
1134
1135 /* Daemonize */
1136 if (opt_daemon) {
53094c05
DG
1137 ret = daemon(0, 0);
1138 if (ret < 0) {
1139 perror("daemon");
1140 goto error;
1141 }
fac6795d
DG
1142 }
1143
1144 /* Check if daemon is UID = 0 */
1145 is_root = !getuid();
1146
1147 /* Set all sockets path */
1148 if (is_root) {
d6f42150
DG
1149 ret = create_lttng_rundir();
1150 if (ret < 0) {
1151 goto error;
1152 }
1153
fac6795d 1154 if (strlen(apps_unix_sock_path) == 0) {
d6f42150
DG
1155 snprintf(apps_unix_sock_path, PATH_MAX,
1156 DEFAULT_GLOBAL_APPS_UNIX_SOCK);
fac6795d
DG
1157 }
1158
1159 if (strlen(client_unix_sock_path) == 0) {
d6f42150
DG
1160 snprintf(client_unix_sock_path, PATH_MAX,
1161 DEFAULT_GLOBAL_CLIENT_UNIX_SOCK);
1162 }
1163
1164 ret = set_kconsumerd_sockets();
1165 if (ret < 0) {
1166 goto error;
fac6795d 1167 }
20fe2104
DG
1168
1169 /* Setup kernel tracer */
1170 init_kernel_tracer();
fac6795d
DG
1171 } else {
1172 if (strlen(apps_unix_sock_path) == 0) {
d6f42150
DG
1173 snprintf(apps_unix_sock_path, PATH_MAX,
1174 DEFAULT_HOME_APPS_UNIX_SOCK, get_home_dir());
fac6795d
DG
1175 }
1176
1177 /* Set the cli tool unix socket path */
1178 if (strlen(client_unix_sock_path) == 0) {
d6f42150
DG
1179 snprintf(client_unix_sock_path, PATH_MAX,
1180 DEFAULT_HOME_CLIENT_UNIX_SOCK, get_home_dir());
fac6795d
DG
1181 }
1182 }
1183
847177cd
DG
1184 DBG("Client socket path %s", client_unix_sock_path);
1185 DBG("Application socket path %s", apps_unix_sock_path);
1186
fac6795d
DG
1187 /* See if daemon already exist. If any of the two
1188 * socket needed by the daemon are present, this test fails
1189 */
1190 if ((ret = check_existing_daemon()) == 0) {
75462a81 1191 ERR("Already running daemon.\n");
ab118b20 1192 /* We do not goto error because we must not
d6f42150 1193 * cleanup() because a daemon is already running.
ab118b20 1194 */
5e16da05 1195 exit(EXIT_FAILURE);
fac6795d
DG
1196 }
1197
1198 if (set_signal_handler() < 0) {
1199 goto error;
1200 }
1201
d6f42150 1202 /* Setup the needed unix socket */
fac6795d
DG
1203 if (init_daemon_socket() < 0) {
1204 goto error;
1205 }
1206
1207 /* Set credentials to socket */
d6f42150 1208 if (is_root && (set_permissions() < 0)) {
fac6795d
DG
1209 goto error;
1210 }
1211
5b8719f5
DG
1212 /* Get parent pid if -S, --sig-parent is specified. */
1213 if (opt_sig_parent) {
1214 ppid = getppid();
1215 }
1216
fac6795d
DG
1217 while (1) {
1218 /* Create thread to manage the client socket */
1219 ret = pthread_create(&threads[0], NULL, thread_manage_clients, (void *) NULL);
1220 if (ret != 0) {
1221 perror("pthread_create");
1222 goto error;
1223 }
1224
1225 /* Create thread to manage application socket */
1226 ret = pthread_create(&threads[1], NULL, thread_manage_apps, (void *) NULL);
1227 if (ret != 0) {
1228 perror("pthread_create");
1229 goto error;
1230 }
1231
1232 for (i = 0; i < 2; i++) {
1233 ret = pthread_join(threads[i], &status);
1234 if (ret != 0) {
1235 perror("pthread_join");
1236 goto error;
1237 }
1238 }
1239 }
1240
1241 cleanup();
5e16da05 1242 exit(EXIT_SUCCESS);
fac6795d
DG
1243
1244error:
1245 cleanup();
5e16da05 1246 exit(EXIT_FAILURE);
fac6795d 1247}
This page took 0.13496 seconds and 4 git commands to generate.