X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=src%2Fbin%2Flttng%2Flttng.c;h=031b015efba4dd05da900b25a8d0c8ead4dfa7b7;hb=1ab1ea0b77d5fc71765e06b5c84431e403e8bd2e;hp=65cea55a25e73a8698fd01c2846b4a405488916a;hpb=10a8a2237343699e3923d87e24dbf2d7fe225377;p=lttng-tools.git

diff --git a/src/bin/lttng/lttng.c b/src/bin/lttng/lttng.c
index 65cea55a2..031b015ef 100644
--- a/src/bin/lttng/lttng.c
+++ b/src/bin/lttng/lttng.c
@@ -22,14 +22,15 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/types.h>
+#include <sys/wait.h>
 #include <unistd.h>
 #include <config.h>
 
 #include <lttng/lttng.h>
-#include <common/lttngerr.h>
+#include <common/error.h>
 
-#include "cmd.h"
-#include "conf.h"
+#include "command.h"
 
 /* Variables */
 static char *progname;
@@ -38,6 +39,7 @@ int opt_quiet;
 int opt_verbose;
 static int opt_no_sessiond;
 static char *opt_sessiond_path;
+static pid_t sessiond_pid;
 
 enum {
 	OPT_SESSION_PATH,
@@ -169,17 +171,26 @@ static void clean_exit(int code)
  */
 static void sighandler(int sig)
 {
+	int status;
+
 	switch (sig) {
 		case SIGTERM:
-			DBG("SIGTERM catched");
+			DBG("SIGTERM caugth");
 			clean_exit(EXIT_FAILURE);
 			break;
 		case SIGCHLD:
+			DBG("SIGCHLD caugth");
+			waitpid(sessiond_pid, &status, 0);
+			/* Indicate that the session daemon died */
+			sessiond_pid = 0;
+			ERR("Session daemon died (exit status %d)", WEXITSTATUS(status));
+			break;
+		case SIGUSR1:
 			/* Notify is done */
-			DBG("SIGCHLD catched");
+			DBG("SIGUSR1 caugth");
 			break;
 		default:
-			DBG("Unknown signal %d catched", sig);
+			DBG("Unknown signal %d caugth", sig);
 			break;
 	}
 
@@ -205,7 +216,7 @@ static int set_signal_handler(void)
 	sa.sa_handler = sighandler;
 	sa.sa_mask = sigset;
 	sa.sa_flags = 0;
-	if ((ret = sigaction(SIGCHLD, &sa, NULL)) < 0) {
+	if ((ret = sigaction(SIGUSR1, &sa, NULL)) < 0) {
 		perror("sigaction");
 		goto end;
 	}
@@ -215,6 +226,11 @@ static int set_signal_handler(void)
 		goto end;
 	}
 
+	if ((ret = sigaction(SIGCHLD, &sa, NULL)) < 0) {
+		perror("sigaction");
+		goto end;
+	}
+
 end:
 	return ret;
 }
@@ -247,9 +263,6 @@ static int handle_command(int argc, char **argv)
 			case CMD_ERROR:
 				ERR("Command error");
 				break;
-			case CMD_NOT_IMPLEMENTED:
-				ERR("Options not implemented");
-				break;
 			case CMD_UNDEFINED:
 				ERR("Undefined command");
 				break;
@@ -297,8 +310,12 @@ static int spawn_sessiond(char *pathname)
 		kill(getppid(), SIGTERM);	/* unpause parent */
 		exit(EXIT_FAILURE);
 	} else if (pid > 0) {
+		sessiond_pid = pid;
 		/* Wait for lttng-sessiond to start */
 		pause();
+		if (!sessiond_pid) {
+			exit(EXIT_FAILURE);
+		}
 		goto end;
 	} else {
 		perror("fork");