Add ust stop trace feature
[lttng-tools.git] / ltt-sessiond / main.c
1 /*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
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.
8 *
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.
13 *
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.
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) */
38 #include <lttng/liblttngctl.h>
39
40 #include "liblttsessiondcomm.h"
41 #include "ltt-sessiond.h"
42 #include "lttngerr.h"
43 #include "session.h"
44 #include "trace.h"
45 #include "traceable-app.h"
46
47 /* Const values */
48 const char default_home_dir[] = DEFAULT_HOME_DIR;
49 const char default_tracing_group[] = DEFAULT_TRACING_GROUP;
50 const char default_ust_sock_dir[] = DEFAULT_UST_SOCK_DIR;
51 const char default_global_apps_pipe[] = DEFAULT_GLOBAL_APPS_PIPE;
52
53 /* Static functions */
54 static int check_existing_daemon(void);
55 static int connect_app(pid_t pid);
56 static int init_daemon_socket(void);
57 static int notify_apps(const char* name);
58 static int process_client_msg(int sock, struct lttcomm_session_msg*);
59 static int send_unix_sock(int sock, void *buf, size_t len);
60 static int set_signal_handler(void);
61 static int set_socket_perms(void);
62 static int setup_data_buffer(char **buf, size_t size, struct lttcomm_lttng_msg *llm);
63 static void cleanup(void);
64 static void copy_common_data(struct lttcomm_lttng_msg *llm, struct lttcomm_session_msg *lsm);
65 static void sighandler(int sig);
66
67 static void *thread_manage_clients(void *data);
68 static void *thread_manage_apps(void *data);
69
70 /* Variables */
71 int opt_verbose;
72 int opt_quiet;
73 const char *progname;
74 const char *opt_tracing_group;
75 static int opt_sig_parent;
76 static int opt_daemon;
77 static int is_root; /* Set to 1 if the daemon is running as root */
78 static pid_t ppid;
79
80 static char apps_unix_sock_path[PATH_MAX]; /* Global application Unix socket path */
81 static char client_unix_sock_path[PATH_MAX]; /* Global client Unix socket path */
82
83 static int client_socket;
84 static int apps_socket;
85
86 static struct ltt_session *current_session;
87
88 /*
89 * thread_manage_apps
90 *
91 * This thread manage the application socket communication
92 */
93 static void *thread_manage_apps(void *data)
94 {
95 int sock, ret;
96
97 /* TODO: Something more elegant is needed but fine for now */
98 struct {
99 int reg; /* 1:register, 0:unregister */
100 pid_t pid;
101 uid_t uid;
102 } reg_msg;
103
104 DBG("[thread] Manage apps started");
105
106 /* Notify all applications to register */
107 notify_apps(default_global_apps_pipe);
108
109 ret = lttcomm_listen_unix_sock(apps_socket);
110 if (ret < 0) {
111 goto error;
112 }
113
114 while (1) {
115 /* Blocking call, waiting for transmission */
116 sock = lttcomm_accept_unix_sock(apps_socket);
117 if (sock < 0) {
118 goto error;
119 }
120
121 /* Basic recv here to handle the very simple data
122 * that the libust send to register (reg_msg).
123 */
124 ret = recv(sock, &reg_msg, sizeof(reg_msg), 0);
125 if (ret < 0) {
126 perror("recv");
127 continue;
128 }
129
130 /* Add application to the global traceable list */
131 if (reg_msg.reg == 1) {
132 /* Registering */
133 ret = register_traceable_app(reg_msg.pid, reg_msg.uid);
134 if (ret < 0) {
135 /* register_traceable_app only return an error with
136 * ENOMEM. At this point, we better stop everything.
137 */
138 goto error;
139 }
140 } else {
141 /* Unregistering */
142 unregister_traceable_app(reg_msg.pid);
143 }
144 }
145
146 error:
147
148 return NULL;
149 }
150
151 /*
152 * thread_manage_clients
153 *
154 * This thread manage all clients request using the unix
155 * client socket for communication.
156 */
157 static void *thread_manage_clients(void *data)
158 {
159 int sock, ret;
160 struct lttcomm_session_msg lsm;
161
162 DBG("[thread] Manage client started");
163
164 ret = lttcomm_listen_unix_sock(client_socket);
165 if (ret < 0) {
166 goto error;
167 }
168
169 /* Notify parent pid that we are ready
170 * to accept command for client side.
171 */
172 if (opt_sig_parent) {
173 kill(ppid, SIGCHLD);
174 }
175
176 while (1) {
177 /* Blocking call, waiting for transmission */
178 sock = lttcomm_accept_unix_sock(client_socket);
179 if (sock < 0) {
180 goto error;
181 }
182
183 /*
184 * Data is received from the lttng client. The struct
185 * lttcomm_session_msg (lsm) contains the command and data
186 * request of the client.
187 */
188 ret = lttcomm_recv_unix_sock(sock, &lsm, sizeof(lsm));
189 if (ret <= 0) {
190 continue;
191 }
192
193 /* This function dispatch the work to the LTTng or UST libs
194 * and then sends back the response to the client. This is needed
195 * because there might be more then one lttcomm_lttng_msg to
196 * send out so process_client_msg do both jobs.
197 */
198 ret = process_client_msg(sock, &lsm);
199 if (ret < 0) {
200 /* Error detected but still accept command */
201 continue;
202 }
203 }
204
205 error:
206 return NULL;
207 }
208
209 /*
210 * send_unix_sock
211 *
212 * Send data on a unix socket using the liblttsessiondcomm API.
213 *
214 * Return lttcomm error code.
215 */
216 static int send_unix_sock(int sock, void *buf, size_t len)
217 {
218 /* Check valid length */
219 if (len <= 0) {
220 return -1;
221 }
222
223 return lttcomm_send_unix_sock(sock, buf, len);
224 }
225
226 /*
227 * connect_app
228 *
229 * Return a socket connected to the libust communication socket
230 * of the application identified by the pid.
231 *
232 * If the pid is not found in the traceable list,
233 * return -1 to indicate error.
234 */
235 static int connect_app(pid_t pid)
236 {
237 int sock;
238 struct ltt_traceable_app *lta;
239
240 DBG("Connect to application pid %d", pid);
241
242 lta = find_app_by_pid(pid);
243 if (lta == NULL) {
244 /* App not found */
245 return -1;
246 }
247
248 sock = ustctl_connect_pid(lta->pid);
249 if (sock < 0) {
250 ERR("Fail connecting to the PID %d\n", pid);
251 }
252
253 return sock;
254 }
255
256 /*
257 * notify_apps
258 *
259 * Notify apps by writing 42 to a named pipe using name.
260 * Every applications waiting for a ltt-sessiond will be notified
261 * and re-register automatically to the session daemon.
262 *
263 * Return open or write error value.
264 */
265 static int notify_apps(const char *name)
266 {
267 int fd;
268 int ret = -1;
269
270 DBG("Notify the global application pipe");
271
272 /* Try opening the global pipe */
273 fd = open(name, O_WRONLY);
274 if (fd < 0) {
275 goto error;
276 }
277
278 /* Notify by writing on the pipe */
279 ret = write(fd, "42", 2);
280 if (ret < 0) {
281 perror("write");
282 }
283
284 error:
285 return ret;
286 }
287
288 /*
289 * ust_create_trace
290 *
291 * Create an userspace trace using pid.
292 * This trace is then appended to the current session
293 * ust trace list.
294 */
295 static int ust_create_trace(pid_t pid)
296 {
297 int sock, ret;
298 struct ltt_ust_trace *trace;
299
300 DBG("Creating trace for pid %d", pid);
301
302 trace = malloc(sizeof(struct ltt_ust_trace));
303 if (trace == NULL) {
304 perror("malloc");
305 ret = -1;
306 goto error;
307 }
308
309 /* Init */
310 trace->pid = pid;
311 trace->shmid = 0;
312 /* NOTE: to be removed. Trace name will no longer be
313 * required for LTTng userspace tracer. For now, we set it
314 * to 'auto' for API compliance.
315 */
316 snprintf(trace->name, 5, "auto");
317
318 /* Connect to app using ustctl API */
319 sock = connect_app(pid);
320 if (sock < 0) {
321 ret = LTTCOMM_NO_TRACEABLE;
322 goto error;
323 }
324
325 ret = ustctl_create_trace(sock, trace->name);
326 if (ret < 0) {
327 ret = LTTCOMM_CREATE_FAIL;
328 goto error;
329 }
330
331 /* Check if current session is valid */
332 if (current_session) {
333 cds_list_add(&trace->list, &current_session->ust_traces);
334 current_session->ust_trace_count++;
335 }
336
337 error:
338 return ret;
339 }
340
341 /*
342 * ust_start_trace
343 *
344 * Start a trace. This trace, identified by the pid, must be
345 * in the current session ust_traces list.
346 */
347 static int ust_start_trace(pid_t pid)
348 {
349 int sock, ret;
350 struct ltt_ust_trace *trace;
351
352 DBG("Starting trace for pid %d", pid);
353
354 trace = find_session_ust_trace_by_pid(current_session, pid);
355 if (trace == NULL) {
356 ret = LTTCOMM_NO_TRACE;
357 goto error;
358 }
359
360 /* Connect to app using ustctl API */
361 sock = connect_app(pid);
362 if (sock < 0) {
363 ret = LTTCOMM_NO_TRACEABLE;
364 goto error;
365 }
366
367 ret = ustctl_start_trace(sock, "auto");
368 if (ret < 0) {
369 ret = LTTCOMM_START_FAIL;
370 goto error;
371 }
372
373 error:
374 return ret;
375 }
376
377 /*
378 * ust_stop_trace
379 *
380 * Stop a trace. This trace, identified by the pid, must be
381 * in the current session ust_traces list.
382 */
383 static int ust_stop_trace(pid_t pid)
384 {
385 int sock, ret;
386 struct ltt_ust_trace *trace;
387
388 DBG("Stopping trace for pid %d", pid);
389
390 trace = find_session_ust_trace_by_pid(current_session, pid);
391 if (trace == NULL) {
392 ret = LTTCOMM_NO_TRACE;
393 goto error;
394 }
395
396 /* Connect to app using ustctl API */
397 sock = connect_app(pid);
398 if (sock < 0) {
399 ret = LTTCOMM_NO_TRACEABLE;
400 goto error;
401 }
402
403 ret = ustctl_stop_trace(sock, trace->name);
404 if (ret < 0) {
405 ret = LTTCOMM_STOP_FAIL;
406 goto error;
407 }
408
409 error:
410 return ret;
411 }
412
413 /*
414 * copy_common_data
415 *
416 * Copy common data between lttcomm_lttng_msg and lttcomm_session_msg
417 */
418 static void copy_common_data(struct lttcomm_lttng_msg *llm, struct lttcomm_session_msg *lsm)
419 {
420 llm->cmd_type = lsm->cmd_type;
421 llm->pid = lsm->pid;
422
423 /* Manage uuid */
424 if (!uuid_is_null(lsm->session_id)) {
425 uuid_copy(llm->session_id, lsm->session_id);
426 }
427
428 strncpy(llm->trace_name, lsm->trace_name, strlen(llm->trace_name));
429 llm->trace_name[strlen(llm->trace_name) - 1] = '\0';
430 }
431
432 /*
433 * setup_data_buffer
434 *
435 * Setup the outgoing data buffer for the response
436 * data allocating the right amount of memory.
437 *
438 * Return total size of the buffer pointed by buf.
439 */
440 static int setup_data_buffer(char **buf, size_t s_data, struct lttcomm_lttng_msg *llm)
441 {
442 int ret = 0;
443 size_t buf_size;
444
445 buf_size = sizeof(struct lttcomm_lttng_msg) + s_data;
446 *buf = malloc(buf_size);
447 if (*buf == NULL) {
448 perror("malloc");
449 ret = -1;
450 goto error;
451 }
452
453 /* Setup lttcomm_lttng_msg data and copy
454 * it to the newly allocated buffer.
455 */
456 llm->size_payload = s_data;
457 memcpy(*buf, llm, sizeof(struct lttcomm_lttng_msg));
458
459 return buf_size;
460
461 error:
462 return ret;
463 }
464
465 /*
466 * process_client_msg
467 *
468 * This takes the lttcomm_session_msg struct and process the command requested
469 * by the client. It then creates response(s) and send it back to the
470 * given socket (sock).
471 *
472 * Return any error encountered or 0 for success.
473 */
474 static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
475 {
476 int ret;
477 int buf_size;
478 size_t header_size;
479 char *send_buf = NULL;
480 struct lttcomm_lttng_msg llm;
481
482 DBG("Processing client message");
483
484 /* Copy common data to identify the response
485 * on the lttng client side.
486 */
487 copy_common_data(&llm, lsm);
488
489 /* Check command that needs a session */
490 if (lsm->cmd_type != LTTNG_CREATE_SESSION &&
491 lsm->cmd_type != LTTNG_LIST_SESSIONS &&
492 lsm->cmd_type != UST_LIST_APPS)
493 {
494 current_session = find_session_by_uuid(lsm->session_id);
495 if (current_session == NULL) {
496 ret = LTTCOMM_SELECT_SESS;
497 goto end;
498 }
499 }
500
501 /* Default return code.
502 * In our world, everything is OK... right? ;)
503 */
504 llm.ret_code = LTTCOMM_OK;
505
506 header_size = sizeof(struct lttcomm_lttng_msg);
507
508 /* Process by command type */
509 switch (lsm->cmd_type) {
510 case LTTNG_CREATE_SESSION:
511 {
512 ret = create_session(lsm->session_name, &llm.session_id);
513 if (ret < 0) {
514 if (ret == -1) {
515 ret = LTTCOMM_EXIST_SESS;
516 } else {
517 ret = LTTCOMM_FATAL;
518 }
519 goto end;
520 }
521
522 buf_size = setup_data_buffer(&send_buf, 0, &llm);
523 if (buf_size < 0) {
524 ret = LTTCOMM_FATAL;
525 goto end;
526 }
527
528 break;
529 }
530 case LTTNG_DESTROY_SESSION:
531 {
532 ret = destroy_session(&lsm->session_id);
533 if (ret < 0) {
534 ret = LTTCOMM_NO_SESS;
535 } else {
536 ret = LTTCOMM_OK;
537 }
538
539 /* No auxiliary data so only send the llm struct. */
540 goto end;
541 }
542 case LTTNG_LIST_TRACES:
543 {
544 unsigned int trace_count = get_trace_count_per_session(current_session);
545
546 if (trace_count == 0) {
547 ret = LTTCOMM_NO_TRACE;
548 goto end;
549 }
550
551 buf_size = setup_data_buffer(&send_buf,
552 sizeof(struct lttng_trace) * trace_count, &llm);
553 if (buf_size < 0) {
554 ret = LTTCOMM_FATAL;
555 goto end;
556 }
557
558 get_traces_per_session(current_session, (struct lttng_trace *)(send_buf + header_size));
559 break;
560 }
561 case UST_CREATE_TRACE:
562 {
563 ret = ust_create_trace(lsm->pid);
564 if (ret < 0) {
565 /* If -1 is returned from ust_create_trace, malloc
566 * failed so it's pretty much a fatal error.
567 */
568 ret = LTTCOMM_FATAL;
569 goto end;
570 }
571
572 /* No auxiliary data so only send the llm struct. */
573 goto end;
574 }
575 case UST_LIST_APPS:
576 {
577 unsigned int app_count = get_app_count();
578 /* Stop right now if no apps */
579 if (app_count == 0) {
580 ret = LTTCOMM_NO_APPS;
581 goto end;
582 }
583
584 /* Setup data buffer and details for transmission */
585 buf_size = setup_data_buffer(&send_buf,
586 sizeof(pid_t) * app_count, &llm);
587 if (buf_size < 0) {
588 ret = LTTCOMM_FATAL;
589 goto end;
590 }
591
592 get_app_list_pids((pid_t *)(send_buf + header_size));
593
594 break;
595 }
596 case UST_START_TRACE:
597 {
598 ret = ust_start_trace(lsm->pid);
599
600 /* No auxiliary data so only send the llm struct. */
601 goto end;
602 }
603 case UST_STOP_TRACE:
604 {
605 ret = ust_stop_trace(lsm->pid);
606
607 /* No auxiliary data so only send the llm struct. */
608 goto end;
609 }
610 case LTTNG_LIST_SESSIONS:
611 {
612 unsigned int session_count = get_session_count();
613 /* Stop right now if no session */
614 if (session_count == 0) {
615 ret = LTTCOMM_NO_SESS;
616 goto end;
617 }
618
619 /* Setup data buffer and details for transmission */
620 buf_size = setup_data_buffer(&send_buf,
621 (sizeof(struct lttng_session) * session_count), &llm);
622 if (buf_size < 0) {
623 ret = LTTCOMM_FATAL;
624 goto end;
625 }
626
627 get_lttng_session((struct lttng_session *)(send_buf + header_size));
628
629 break;
630 }
631 default:
632 {
633 /* Undefined command */
634 ret = LTTCOMM_UND;
635 goto end;
636 }
637 }
638
639 ret = send_unix_sock(sock, send_buf, buf_size);
640
641 if (send_buf != NULL) {
642 free(send_buf);
643 }
644
645 return ret;
646
647 end:
648 /* Notify client of error */
649 llm.ret_code = ret;
650 llm.size_payload = 0;
651 send_unix_sock(sock, (void*) &llm, sizeof(llm));
652
653 return ret;
654 }
655
656 /*
657 * usage function on stderr
658 */
659 static void usage(void)
660 {
661 fprintf(stderr, "Usage: %s OPTIONS\n\nOptions:\n", progname);
662 fprintf(stderr, " -h, --help Display this usage.\n");
663 fprintf(stderr, " -c, --client-sock PATH Specify path for the client unix socket\n");
664 fprintf(stderr, " -a, --apps-sock PATH Specify path for apps unix socket.\n");
665 fprintf(stderr, " -d, --daemonize Start as a daemon.\n");
666 fprintf(stderr, " -g, --group NAME Specify the tracing group name. (default: tracing)\n");
667 fprintf(stderr, " -V, --version Show version number.\n");
668 fprintf(stderr, " -S, --sig-parent Send SIGCHLD to parent pid to notify readiness.\n");
669 fprintf(stderr, " -q, --quiet No output at all.\n");
670 fprintf(stderr, " -v, --verbose Verbose mode. Activate DBG() macro.\n");
671 }
672
673 /*
674 * daemon argument parsing
675 */
676 static int parse_args(int argc, char **argv)
677 {
678 int c;
679
680 static struct option long_options[] = {
681 { "client-sock", 1, 0, 'c' },
682 { "apps-sock", 1, 0, 'a' },
683 { "daemonize", 0, 0, 'd' },
684 { "sig-parent", 0, 0, 'S' },
685 { "help", 0, 0, 'h' },
686 { "group", 1, 0, 'g' },
687 { "version", 0, 0, 'V' },
688 { "quiet", 0, 0, 'q' },
689 { "verbose", 0, 0, 'v' },
690 { NULL, 0, 0, 0 }
691 };
692
693 while (1) {
694 int option_index = 0;
695 c = getopt_long(argc, argv, "dhqvVS" "a:c:g:s:", long_options, &option_index);
696 if (c == -1) {
697 break;
698 }
699
700 switch (c) {
701 case 0:
702 fprintf(stderr, "option %s", long_options[option_index].name);
703 if (optarg) {
704 fprintf(stderr, " with arg %s\n", optarg);
705 }
706 break;
707 case 'c':
708 snprintf(client_unix_sock_path, PATH_MAX, "%s", optarg);
709 break;
710 case 'a':
711 snprintf(apps_unix_sock_path, PATH_MAX, "%s", optarg);
712 break;
713 case 'd':
714 opt_daemon = 1;
715 break;
716 case 'g':
717 opt_tracing_group = strdup(optarg);
718 break;
719 case 'h':
720 usage();
721 exit(EXIT_FAILURE);
722 case 'V':
723 fprintf(stdout, "%s\n", VERSION);
724 exit(EXIT_SUCCESS);
725 case 'S':
726 opt_sig_parent = 1;
727 break;
728 case 'q':
729 opt_quiet = 1;
730 break;
731 case 'v':
732 opt_verbose = 1;
733 break;
734 default:
735 /* Unknown option or other error.
736 * Error is printed by getopt, just return */
737 return -1;
738 }
739 }
740
741 return 0;
742 }
743
744 /*
745 * init_daemon_socket
746 *
747 * Creates the two needed socket by the daemon.
748 * apps_socket - The communication socket for all UST apps.
749 * client_socket - The communication of the cli tool (lttng).
750 */
751 static int init_daemon_socket()
752 {
753 int ret = 0;
754 mode_t old_umask;
755
756 old_umask = umask(0);
757
758 /* Create client tool unix socket */
759 client_socket = lttcomm_create_unix_sock(client_unix_sock_path);
760 if (client_socket < 0) {
761 ret = -1;
762 goto end;
763 }
764
765 /* File permission MUST be 660 */
766 ret = chmod(client_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
767 if (ret < 0) {
768 perror("chmod");
769 goto end;
770 }
771
772 /* Create the application unix socket */
773 apps_socket = lttcomm_create_unix_sock(apps_unix_sock_path);
774 if (apps_socket < 0) {
775 ret = -1;
776 goto end;
777 }
778
779 /* File permission MUST be 660 */
780 ret = chmod(apps_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
781 if (ret < 0) {
782 perror("chmod");
783 goto end;
784 }
785
786 end:
787 umask(old_umask);
788 return ret;
789 }
790
791 /*
792 * check_existing_daemon
793 *
794 * Check if the global socket is available.
795 * If yes, error is returned.
796 */
797 static int check_existing_daemon()
798 {
799 int ret;
800
801 ret = access(client_unix_sock_path, F_OK);
802 if (ret == 0) {
803 ret = access(apps_unix_sock_path, F_OK);
804 }
805
806 return ret;
807 }
808
809 /*
810 * get_home_dir
811 *
812 * Return pointer to home directory path using
813 * the env variable HOME.
814 *
815 * Default : /tmp
816 */
817 static const char *get_home_dir(void)
818 {
819 const char *home_path;
820
821 if ((home_path = (const char *) getenv("HOME")) == NULL) {
822 home_path = default_home_dir;
823 }
824
825 return home_path;
826 }
827
828 /*
829 * set_socket_perms
830 *
831 * Set the tracing group gid onto the client socket.
832 */
833 static int set_socket_perms(void)
834 {
835 int ret;
836 struct group *grp;
837
838 /* Decide which group name to use */
839 (opt_tracing_group != NULL) ?
840 (grp = getgrnam(opt_tracing_group)) :
841 (grp = getgrnam(default_tracing_group));
842
843 if (grp == NULL) {
844 ERR("Missing tracing group. Aborting execution.\n");
845 ret = -1;
846 goto end;
847 }
848
849 ret = chown(client_unix_sock_path, 0, grp->gr_gid);
850 if (ret < 0) {
851 perror("chown");
852 }
853
854 DBG("Sockets permissions set");
855
856 end:
857 return ret;
858 }
859
860 /*
861 * set_signal_handler
862 *
863 * Setup signal handler for :
864 * SIGINT, SIGTERM, SIGPIPE
865 */
866 static int set_signal_handler(void)
867 {
868 int ret = 0;
869 struct sigaction sa;
870 sigset_t sigset;
871
872 if ((ret = sigemptyset(&sigset)) < 0) {
873 perror("sigemptyset");
874 return ret;
875 }
876
877 sa.sa_handler = sighandler;
878 sa.sa_mask = sigset;
879 sa.sa_flags = 0;
880 if ((ret = sigaction(SIGTERM, &sa, NULL)) < 0) {
881 perror("sigaction");
882 return ret;
883 }
884
885 if ((ret = sigaction(SIGINT, &sa, NULL)) < 0) {
886 perror("sigaction");
887 return ret;
888 }
889
890 if ((ret = sigaction(SIGPIPE, &sa, NULL)) < 0) {
891 perror("sigaction");
892 return ret;
893 }
894
895 DBG("Signal handler set for SIGTERM, SIGPIPE and SIGINT");
896
897 return ret;
898 }
899
900 /**
901 * sighandler
902 *
903 * Signal handler for the daemon
904 */
905 static void sighandler(int sig)
906 {
907 switch (sig) {
908 case SIGPIPE:
909 DBG("SIGPIPE catched");
910 return;
911 case SIGINT:
912 DBG("SIGINT catched");
913 cleanup();
914 break;
915 case SIGTERM:
916 DBG("SIGTERM catched");
917 cleanup();
918 break;
919 default:
920 break;
921 }
922
923 exit(EXIT_SUCCESS);
924 }
925
926 /*
927 * cleanup
928 *
929 * Cleanup the daemon on exit
930 */
931 static void cleanup()
932 {
933 DBG("Cleaning up");
934
935 /* <fun> */
936 MSG("\n%c[%d;%dm*** assert failed *** ==> %c[%dm", 27,1,31,27,0);
937 MSG("%c[%d;%dmMatthew, BEET driven development works!%c[%dm",27,1,33,27,0);
938 /* </fun> */
939
940 unlink(client_unix_sock_path);
941 unlink(apps_unix_sock_path);
942 }
943
944 /*
945 * main
946 */
947 int main(int argc, char **argv)
948 {
949 int i;
950 int ret = 0;
951 void *status;
952 pthread_t threads[2];
953
954 /* Parse arguments */
955 progname = argv[0];
956 if ((ret = parse_args(argc, argv) < 0)) {
957 goto error;
958 }
959
960 /* Daemonize */
961 if (opt_daemon) {
962 ret = daemon(0, 0);
963 if (ret < 0) {
964 perror("daemon");
965 goto error;
966 }
967 }
968
969 /* Check if daemon is UID = 0 */
970 is_root = !getuid();
971
972 /* Set all sockets path */
973 if (is_root) {
974 if (strlen(apps_unix_sock_path) == 0) {
975 (snprintf(apps_unix_sock_path, PATH_MAX,
976 DEFAULT_GLOBAL_APPS_UNIX_SOCK));
977 }
978
979 if (strlen(client_unix_sock_path) == 0) {
980 (snprintf(client_unix_sock_path, PATH_MAX,
981 DEFAULT_GLOBAL_CLIENT_UNIX_SOCK));
982 }
983 } else {
984 if (strlen(apps_unix_sock_path) == 0) {
985 (snprintf(apps_unix_sock_path, PATH_MAX,
986 DEFAULT_HOME_APPS_UNIX_SOCK, get_home_dir()));
987 }
988
989 /* Set the cli tool unix socket path */
990 if (strlen(client_unix_sock_path) == 0) {
991 (snprintf(client_unix_sock_path, PATH_MAX,
992 DEFAULT_HOME_CLIENT_UNIX_SOCK, get_home_dir()));
993 }
994 }
995
996 /* See if daemon already exist. If any of the two
997 * socket needed by the daemon are present, this test fails
998 */
999 if ((ret = check_existing_daemon()) == 0) {
1000 ERR("Already running daemon.\n");
1001 /* We do not goto error because we must not
1002 * cleanup() because a daemon is already working.
1003 */
1004 return EXIT_FAILURE;
1005 }
1006
1007 if (set_signal_handler() < 0) {
1008 goto error;
1009 }
1010
1011 /* Setup the two needed unix socket */
1012 if (init_daemon_socket() < 0) {
1013 goto error;
1014 }
1015
1016 /* Set credentials to socket */
1017 if (is_root && (set_socket_perms() < 0)) {
1018 goto error;
1019 }
1020
1021 /* Get parent pid if -S, --sig-parent is specified. */
1022 if (opt_sig_parent) {
1023 ppid = getppid();
1024 }
1025
1026 while (1) {
1027 /* Create thread to manage the client socket */
1028 ret = pthread_create(&threads[0], NULL, thread_manage_clients, (void *) NULL);
1029 if (ret != 0) {
1030 perror("pthread_create");
1031 goto error;
1032 }
1033
1034 /* Create thread to manage application socket */
1035 ret = pthread_create(&threads[1], NULL, thread_manage_apps, (void *) NULL);
1036 if (ret != 0) {
1037 perror("pthread_create");
1038 goto error;
1039 }
1040
1041 for (i = 0; i < 2; i++) {
1042 ret = pthread_join(threads[i], &status);
1043 if (ret != 0) {
1044 perror("pthread_join");
1045 goto error;
1046 }
1047 }
1048 }
1049
1050 cleanup();
1051 return 0;
1052
1053 error:
1054 cleanup();
1055
1056 return EXIT_FAILURE;
1057 }
This page took 0.050204 seconds and 5 git commands to generate.