X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=ltt-sessiond%2Fmain.c;h=f5431bc0ae0299102b1d52b7c42bbc4f4b68de30;hp=4e1fb1f6b5b441aa73a05f9024e4cdbe039e154d;hb=50dc1fa67a38503ba9426e3d78edb17f48446796;hpb=62bd06d80051a13a2fb6cf72e82e9bdc1683eea5

diff --git a/ltt-sessiond/main.c b/ltt-sessiond/main.c
index 4e1fb1f6b..f5431bc0a 100644
--- a/ltt-sessiond/main.c
+++ b/ltt-sessiond/main.c
@@ -35,12 +35,13 @@
 
 #include <urcu/list.h>		/* URCU list library (-lurcu) */
 #include <ust/ustctl.h>		/* UST control lib (-lust) */
-#include <lttng/liblttngctl.h>
+#include <lttng/lttng.h>
 
 #include "liblttsessiondcomm.h"
 #include "ltt-sessiond.h"
 #include "lttngerr.h"
 #include "session.h"
+#include "trace.h"
 #include "traceable-app.h"
 
 /* Const values */
@@ -51,20 +52,23 @@ const char default_global_apps_pipe[] = DEFAULT_GLOBAL_APPS_PIPE;
 
 /* Static functions */
 static int check_existing_daemon(void);
-static int connect_app(pid_t pid);
+static int ust_connect_app(pid_t pid);
 static int init_daemon_socket(void);
 static int notify_apps(const char* name);
 static int process_client_msg(int sock, struct lttcomm_session_msg*);
 static int send_unix_sock(int sock, void *buf, size_t len);
 static int set_signal_handler(void);
-static int set_socket_perms(void);
-static int setup_data_buffer(char **buf, size_t size, struct lttcomm_lttng_msg *llm);
+static int set_permissions(void);
+static int setup_data_buffer(char **buf, size_t size, struct lttcomm_lttng_header *llh);
+static int create_lttng_rundir(void);
+static int set_kconsumerd_sockets(void);
 static void cleanup(void);
-static void copy_common_data(struct lttcomm_lttng_msg *llm, struct lttcomm_session_msg *lsm);
+static void copy_common_data(struct lttcomm_lttng_header *llh, struct lttcomm_session_msg *lsm);
 static void sighandler(int sig);
 
 static void *thread_manage_clients(void *data);
 static void *thread_manage_apps(void *data);
+static void *thread_manage_kconsumerd(void *data);
 
 /* Variables */
 int opt_verbose;
@@ -76,13 +80,51 @@ static int opt_daemon;
 static int is_root;			/* Set to 1 if the daemon is running as root */
 static pid_t ppid;
 
-static char apps_unix_sock_path[PATH_MAX];			/* Global application Unix socket path */
-static char client_unix_sock_path[PATH_MAX];		/* Global client Unix socket path */
+static char apps_unix_sock_path[PATH_MAX];				/* Global application Unix socket path */
+static char client_unix_sock_path[PATH_MAX];			/* Global client Unix socket path */
+static char kconsumerd_err_unix_sock_path[PATH_MAX];	/* kconsumerd error Unix socket path */
+static char kconsumerd_cmd_unix_sock_path[PATH_MAX];	/* kconsumerd command Unix socket path */
 
-static int client_socket;
-static int apps_socket;
+static int client_sock;
+static int apps_sock;
+static int kconsumerd_err_sock;
 
-static struct ltt_session *current_session;
+/* Extern in session.h */
+struct ltt_session *current_session;
+
+/*
+ *  thread_manage_kconsumerd
+ *
+ *  This thread manage the kconsumerd error sent
+ *  back to the session daemon.
+ */
+static void *thread_manage_kconsumerd(void *data)
+{
+	int sock, ret;
+
+	DBG("[thread] Manage kconsumerd started");
+
+	ret = lttcomm_listen_unix_sock(kconsumerd_err_sock);
+	if (ret < 0) {
+		goto error;
+	}
+
+	sock = lttcomm_accept_unix_sock(kconsumerd_err_sock);
+	if (sock < 0) {
+		goto error;
+	}
+
+	while (1) {
+		//ret = lttcomm_recv_unix_sock(sock, &lsm, sizeof(lsm));
+		if (ret <= 0) {
+			/* TODO: Consumerd died? */
+			continue;
+		}
+	}
+
+error:
+	return NULL;
+}
 
 /*
  * 	thread_manage_apps
@@ -100,17 +142,19 @@ static void *thread_manage_apps(void *data)
 		uid_t uid;
 	} reg_msg;
 
+	DBG("[thread] Manage apps started");
+
 	/* Notify all applications to register */
 	notify_apps(default_global_apps_pipe);
 
-	ret = lttcomm_listen_unix_sock(apps_socket);
+	ret = lttcomm_listen_unix_sock(apps_sock);
 	if (ret < 0) {
 		goto error;
 	}
 
 	while (1) {
 		/* Blocking call, waiting for transmission */
-		sock = lttcomm_accept_unix_sock(apps_socket);
+		sock = lttcomm_accept_unix_sock(apps_sock);
 		if (sock < 0) {
 			goto error;
 		}
@@ -156,7 +200,9 @@ static void *thread_manage_clients(void *data)
 	int sock, ret;
 	struct lttcomm_session_msg lsm;
 
-	ret = lttcomm_listen_unix_sock(client_socket);
+	DBG("[thread] Manage client started");
+
+	ret = lttcomm_listen_unix_sock(client_sock);
 	if (ret < 0) {
 		goto error;
 	}
@@ -170,7 +216,7 @@ static void *thread_manage_clients(void *data)
 
 	while (1) {
 		/* Blocking call, waiting for transmission */
-		sock = lttcomm_accept_unix_sock(client_socket);
+		sock = lttcomm_accept_unix_sock(client_sock);
 		if (sock < 0) {
 			goto error;
 		}
@@ -187,7 +233,7 @@ static void *thread_manage_clients(void *data)
 
 		/* This function dispatch the work to the LTTng or UST libs
 		 * and then sends back the response to the client. This is needed
-		 * because there might be more then one lttcomm_lttng_msg to
+		 * because there might be more then one lttcomm_lttng_header to
 		 * send out so process_client_msg do both jobs.
 		 */
 		ret = process_client_msg(sock, &lsm);
@@ -219,7 +265,7 @@ static int send_unix_sock(int sock, void *buf, size_t len)
 }
 
 /*
- * 	connect_app
+ * 	ust_connect_app
  *
  * 	Return a socket connected to the libust communication socket
  * 	of the application identified by the pid.
@@ -227,14 +273,17 @@ static int send_unix_sock(int sock, void *buf, size_t len)
  * 	If the pid is not found in the traceable list,
  * 	return -1 to indicate error.
  */
-static int connect_app(pid_t pid)
+static int ust_connect_app(pid_t pid)
 {
 	int sock;
 	struct ltt_traceable_app *lta;
 
+	DBG("Connect to application pid %d", pid);
+
 	lta = find_app_by_pid(pid);
 	if (lta == NULL) {
 		/* App not found */
+		DBG("Application pid %d not found", pid);
 		return -1;
 	}
 
@@ -260,6 +309,8 @@ static int notify_apps(const char *name)
 	int fd;
 	int ret = -1;
 
+	DBG("Notify the global application pipe");
+
 	/* Try opening the global pipe */
 	fd = open(name, O_WRONLY);
 	if (fd < 0) {
@@ -276,68 +327,23 @@ error:
 	return ret;
 }
 
-/*
- *  ust_create_trace
- *
- *  Create an userspace trace using pid.
- *  This trace is then appended to the current session
- *  ust trace list.
- */
-static int ust_create_trace(pid_t pid)
-{
-	int sock, ret;
-	struct ltt_ust_trace *trace;
-
-	trace = malloc(sizeof(struct ltt_ust_trace));
-	if (trace == NULL) {
-		perror("malloc");
-		ret = -1;
-		goto error;
-	}
-
-	/* Init */
-	trace->pid = pid;
-	trace->shmid = 0;
-
-	/* Connect to app using ustctl API */
-	sock = connect_app(pid);
-	if (sock < 0) {
-		ret = LTTCOMM_NO_TRACEABLE;
-		goto error;
-	}
-
-	ret = ustctl_create_trace(sock, "auto");
-	if (ret < 0) {
-		ret = LTTCOMM_CREATE_FAIL;
-		goto error;
-	}
-
-	/* Check if current session is valid */
-	if (current_session) {
-		cds_list_add(&trace->list, &current_session->ust_traces);
-	}
-
-error:
-	return ret;
-}
-
 /*
  *  copy_common_data
  *
- *  Copy common data between lttcomm_lttng_msg and lttcomm_session_msg
+ *  Copy common data between lttcomm_lttng_header and lttcomm_session_msg
  */
-static void copy_common_data(struct lttcomm_lttng_msg *llm, struct lttcomm_session_msg *lsm)
+static void copy_common_data(struct lttcomm_lttng_header *llh, struct lttcomm_session_msg *lsm)
 {
-	llm->cmd_type = lsm->cmd_type;
-	llm->pid = lsm->pid;
+	llh->cmd_type = lsm->cmd_type;
+	llh->pid = lsm->pid;
 
 	/* Manage uuid */
 	if (!uuid_is_null(lsm->session_id)) {
-		uuid_copy(llm->session_id, lsm->session_id);
+		uuid_copy(llh->session_id, lsm->session_id);
 	}
 
-	strncpy(llm->trace_name, lsm->trace_name, strlen(llm->trace_name));
-	llm->trace_name[strlen(llm->trace_name) - 1] = '\0';
+	strncpy(llh->trace_name, lsm->trace_name, strlen(llh->trace_name));
+	llh->trace_name[strlen(llh->trace_name) - 1] = '\0';
 }
 
 /*
@@ -348,12 +354,12 @@ static void copy_common_data(struct lttcomm_lttng_msg *llm, struct lttcomm_sessi
  *
  *  Return total size of the buffer pointed by buf.
  */
-static int setup_data_buffer(char **buf, size_t s_data, struct lttcomm_lttng_msg *llm)
+static int setup_data_buffer(char **buf, size_t s_data, struct lttcomm_lttng_header *llh)
 {
 	int ret = 0;
 	size_t buf_size;
 
-	buf_size = sizeof(struct lttcomm_lttng_msg) + s_data;
+	buf_size = sizeof(struct lttcomm_lttng_header) + s_data;
 	*buf = malloc(buf_size);
 	if (*buf == NULL) {
 		perror("malloc");
@@ -361,11 +367,11 @@ static int setup_data_buffer(char **buf, size_t s_data, struct lttcomm_lttng_msg
 		goto error;
 	}
 
-	/* Setup lttcomm_lttng_msg data and copy
+	/* Setup lttcomm_lttng_header data and copy
 	 * it to the newly allocated buffer.
 	 */
-	llm->size_payload = s_data;
-	memcpy(*buf, llm, sizeof(struct lttcomm_lttng_msg));
+	llh->payload_size = s_data;
+	memcpy(*buf, llh, sizeof(struct lttcomm_lttng_header));
 
 	return buf_size;
 
@@ -384,16 +390,17 @@ error:
  */
 static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 {
-	int ret;
-	int buf_size;
+	int ust_sock, ret, buf_size;
 	size_t header_size;
 	char *send_buf = NULL;
-	struct lttcomm_lttng_msg llm;
+	struct lttcomm_lttng_header llh;
+
+	DBG("Processing client message");
 
 	/* Copy common data to identify the response
 	 * on the lttng client side.
 	 */
-	copy_common_data(&llm, lsm);
+	copy_common_data(&llh, lsm);
 
 	/* Check command that needs a session */
 	if (lsm->cmd_type != LTTNG_CREATE_SESSION &&
@@ -410,15 +417,25 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 	/* Default return code.
 	 * In our world, everything is OK... right? ;)
 	 */
-	llm.ret_code = LTTCOMM_OK;
+	llh.ret_code = LTTCOMM_OK;
+
+	header_size = sizeof(struct lttcomm_lttng_header);
 
-	header_size = sizeof(struct lttcomm_lttng_msg);
+	/* Connect to ust apps if available pid */
+	if (lsm->pid != 0) {
+		/* Connect to app using ustctl API */
+		ust_sock = ust_connect_app(lsm->pid);
+		if (ust_sock < 0) {
+			ret = LTTCOMM_NO_TRACEABLE;
+			goto end;
+		}
+	}
 
 	/* Process by command type */
 	switch (lsm->cmd_type) {
 		case LTTNG_CREATE_SESSION:
 		{
-			ret = create_session(lsm->session_name, &llm.session_id);
+			ret = create_session(lsm->session_name, &llh.session_id);
 			if (ret < 0) {
 				if (ret == -1) {
 					ret = LTTCOMM_EXIST_SESS;
@@ -428,7 +445,7 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 				goto end;
 			}
 
-			buf_size = setup_data_buffer(&send_buf, 0, &llm);
+			buf_size = setup_data_buffer(&send_buf, 0, &llh);
 			if (buf_size < 0) {
 				ret = LTTCOMM_FATAL;
 				goto end;
@@ -445,18 +462,40 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 				ret = LTTCOMM_OK;
 			}
 
-			/* No auxiliary data so only send the llm struct. */
+			/* No auxiliary data so only send the llh struct. */
 			goto end;
 		}
+		case LTTNG_LIST_TRACES:
+		{
+			unsigned int trace_count = get_trace_count_per_session(current_session);
+
+			if (trace_count == 0) {
+				ret = LTTCOMM_NO_TRACE;
+				goto end;
+			}
+
+			buf_size = setup_data_buffer(&send_buf,
+					sizeof(struct lttng_trace) * trace_count, &llh);
+			if (buf_size < 0) {
+				ret = LTTCOMM_FATAL;
+				goto end;
+			}
+
+			get_traces_per_session(current_session, (struct lttng_trace *)(send_buf + header_size));
+			break;
+		}
 		case UST_CREATE_TRACE:
 		{
-			ret = ust_create_trace(lsm->pid);
+			ret = ust_create_trace(ust_sock, lsm->pid);
 			if (ret < 0) {
-				ret = LTTCOMM_CREATE_FAIL;
+				/* If -1 is returned from ust_create_trace, malloc
+				 * failed so it's pretty much a fatal error.
+				 */
+				ret = LTTCOMM_FATAL;
 				goto end;
 			}
 
-			/* No auxiliary data so only send the llm struct. */
+			/* No auxiliary data so only send the llh struct. */
 			goto end;
 		}
 		case UST_LIST_APPS:
@@ -470,7 +509,7 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 
 			/* Setup data buffer and details for transmission */
 			buf_size = setup_data_buffer(&send_buf,
-					sizeof(pid_t) * app_count, &llm);
+					sizeof(pid_t) * app_count, &llh);
 			if (buf_size < 0) {
 				ret = LTTCOMM_FATAL;
 				goto end;
@@ -480,6 +519,20 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 
 			break;
 		}
+		case UST_START_TRACE:
+		{
+			ret = ust_start_trace(ust_sock, lsm->pid);
+
+			/* No auxiliary data so only send the llh struct. */
+			goto end;
+		}
+		case UST_STOP_TRACE:
+		{
+			ret = ust_stop_trace(ust_sock, lsm->pid);
+
+			/* No auxiliary data so only send the llh struct. */
+			goto end;
+		}
 		case LTTNG_LIST_SESSIONS:
 		{
 			unsigned int session_count = get_session_count();
@@ -491,7 +544,7 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 
 			/* Setup data buffer and details for transmission */
 			buf_size = setup_data_buffer(&send_buf,
-					(sizeof(struct lttng_session) * session_count), &llm);
+					(sizeof(struct lttng_session) * session_count), &llh);
 			if (buf_size < 0) {
 				ret = LTTCOMM_FATAL;
 				goto end;
@@ -518,10 +571,11 @@ static int process_client_msg(int sock, struct lttcomm_session_msg *lsm)
 	return ret;
 
 end:
+	DBG("Return code to client %d", ret);
 	/* Notify client of error */
-	llm.ret_code = ret;
-	llm.size_payload = 0;
-	send_unix_sock(sock, (void*) &llm, sizeof(llm));
+	llh.ret_code = ret;
+	llh.payload_size = 0;
+	send_unix_sock(sock, (void*) &llh, sizeof(llh));
 
 	return ret;
 }
@@ -532,14 +586,17 @@ end:
 static void usage(void)
 {
 	fprintf(stderr, "Usage: %s OPTIONS\n\nOptions:\n", progname);
-	fprintf(stderr, "  -h, --help                Display this usage.\n");
-	fprintf(stderr, "  -c, --client-sock PATH    Specify path for the client unix socket\n");
-	fprintf(stderr, "  -a, --apps-sock PATH      Specify path for apps unix socket.\n");
-	fprintf(stderr, "  -d, --daemonize           Start as a daemon.\n");
-	fprintf(stderr, "  -g, --group NAME          Specify the tracing group name. (default: tracing)\n");
-	fprintf(stderr, "  -V, --version             Show version number.\n");
-	fprintf(stderr, "  -S, --sig-parent          Send SIGCHLD to parent pid to notify readiness.\n");
-	fprintf(stderr, "  -q, --quiet               No output at all.\n");
+	fprintf(stderr, "  -h, --help                         Display this usage.\n");
+	fprintf(stderr, "  -c, --client-sock PATH             Specify path for the client unix socket\n");
+	fprintf(stderr, "  -a, --apps-sock PATH               Specify path for apps unix socket\n");
+	fprintf(stderr, "      --kconsumerd-err-sock PATH     Specify path for the kernel consumer error socket\n");
+	fprintf(stderr, "      --kconsumerd-cmd-sock PATH     Specify path for the kernel consumer command socket\n");
+	fprintf(stderr, "  -d, --daemonize                    Start as a daemon.\n");
+	fprintf(stderr, "  -g, --group NAME                   Specify the tracing group name. (default: tracing)\n");
+	fprintf(stderr, "  -V, --version                      Show version number.\n");
+	fprintf(stderr, "  -S, --sig-parent                   Send SIGCHLD to parent pid to notify readiness.\n");
+	fprintf(stderr, "  -q, --quiet                        No output at all.\n");
+	fprintf(stderr, "  -v, --verbose                      Verbose mode. Activate DBG() macro.\n");
 }
 
 /*
@@ -552,18 +609,21 @@ static int parse_args(int argc, char **argv)
 	static struct option long_options[] = {
 		{ "client-sock", 1, 0, 'c' },
 		{ "apps-sock", 1, 0, 'a' },
+		{ "kconsumerd-cmd-sock", 1, 0, 0 },
+		{ "kconsumerd-err-sock", 1, 0, 0 },
 		{ "daemonize", 0, 0, 'd' },
 		{ "sig-parent", 0, 0, 'S' },
 		{ "help", 0, 0, 'h' },
 		{ "group", 1, 0, 'g' },
 		{ "version", 0, 0, 'V' },
 		{ "quiet", 0, 0, 'q' },
+		{ "verbose", 0, 0, 'v' },
 		{ NULL, 0, 0, 0 }
 	};
 
 	while (1) {
 		int option_index = 0;
-		c = getopt_long(argc, argv, "dhqVS" "a:c:g:s:", long_options, &option_index);
+		c = getopt_long(argc, argv, "dhqvVS" "a:c:g:s:E:C:", long_options, &option_index);
 		if (c == -1) {
 			break;
 		}
@@ -596,9 +656,18 @@ static int parse_args(int argc, char **argv)
 		case 'S':
 			opt_sig_parent = 1;
 			break;
+		case 'E':
+			snprintf(kconsumerd_err_unix_sock_path, PATH_MAX, "%s", optarg);
+			break;
+		case 'C':
+			snprintf(kconsumerd_cmd_unix_sock_path, PATH_MAX, "%s", optarg);
+			break;
 		case 'q':
 			opt_quiet = 1;
 			break;
+		case 'v':
+			opt_verbose = 1;
+			break;
 		default:
 			/* Unknown option or other error.
 			 * Error is printed by getopt, just return */
@@ -613,8 +682,8 @@ static int parse_args(int argc, char **argv)
  * 	init_daemon_socket
  *
  * 	Creates the two needed socket by the daemon.
- * 	    apps_socket - The communication socket for all UST apps.
- * 	    client_socket - The communication of the cli tool (lttng).
+ * 	    apps_sock - The communication socket for all UST apps.
+ * 	    client_sock - The communication of the cli tool (lttng).
  */
 static int init_daemon_socket()
 {
@@ -624,8 +693,9 @@ static int init_daemon_socket()
 	old_umask = umask(0);
 
 	/* Create client tool unix socket */
-	client_socket = lttcomm_create_unix_sock(client_unix_sock_path);
-	if (client_socket < 0) {
+	client_sock = lttcomm_create_unix_sock(client_unix_sock_path);
+	if (client_sock < 0) {
+		ERR("Create unix sock failed: %s", client_unix_sock_path);
 		ret = -1;
 		goto end;
 	}
@@ -633,20 +703,23 @@ static int init_daemon_socket()
 	/* File permission MUST be 660 */
 	ret = chmod(client_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
 	if (ret < 0) {
+		ERR("Set file permissions failed: %s", client_unix_sock_path);
 		perror("chmod");
 		goto end;
 	}
 
 	/* Create the application unix socket */
-	apps_socket = lttcomm_create_unix_sock(apps_unix_sock_path);
-	if (apps_socket < 0) {
+	apps_sock = lttcomm_create_unix_sock(apps_unix_sock_path);
+	if (apps_sock < 0) {
+		ERR("Create unix sock failed: %s", apps_unix_sock_path);
 		ret = -1;
 		goto end;
 	}
 
-	/* File permission MUST be 660 */
+	/* File permission MUST be 666 */
 	ret = chmod(apps_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
 	if (ret < 0) {
+		ERR("Set file permissions failed: %s", apps_unix_sock_path);
 		perror("chmod");
 		goto end;
 	}
@@ -694,11 +767,11 @@ static const char *get_home_dir(void)
 }
 
 /*
- *  set_socket_perms
+ *  set_permissions
  *
  *	Set the tracing group gid onto the client socket.
  */
-static int set_socket_perms(void)
+static int set_permissions(void)
 {
 	int ret;
 	struct group *grp;
@@ -714,15 +787,97 @@ static int set_socket_perms(void)
 		goto end;
 	}
 
+	/* Set lttng run dir */
+	ret = chown(LTTNG_RUNDIR, 0, grp->gr_gid);
+	if (ret < 0) {
+		ERR("Unable to set group on " LTTNG_RUNDIR);
+		perror("chown");
+	}
+
+	/* lttng client socket path */
 	ret = chown(client_unix_sock_path, 0, grp->gr_gid);
 	if (ret < 0) {
+		ERR("Unable to set group on %s", client_unix_sock_path);
+		perror("chown");
+	}
+
+	/* kconsumerd error socket path */
+	ret = chown(kconsumerd_err_unix_sock_path, 0, grp->gr_gid);
+	if (ret < 0) {
+		ERR("Unable to set group on %s", kconsumerd_err_unix_sock_path);
 		perror("chown");
 	}
 
+	DBG("All permissions are set");
+
 end:
 	return ret;
 }
 
+/*
+ *  create_lttng_rundir
+ *
+ *  Create the lttng run directory needed for all
+ *  global sockets and pipe.
+ */
+static int create_lttng_rundir(void)
+{
+	int ret;
+
+	ret = mkdir(LTTNG_RUNDIR, S_IRWXU | S_IRWXG );
+	if (ret < 0) {
+		ERR("Unable to create " LTTNG_RUNDIR);
+		goto error;
+	}
+
+error:
+	return ret;
+}
+
+/*
+ *  set_kconsumerd_sockets
+ *
+ *  Setup sockets and directory needed by the kconsumerd
+ *  communication with the session daemon.
+ */
+static int set_kconsumerd_sockets(void)
+{
+	int ret;
+
+	if (strlen(kconsumerd_err_unix_sock_path) == 0) {
+		snprintf(kconsumerd_err_unix_sock_path, PATH_MAX, KCONSUMERD_ERR_SOCK_PATH);
+	}
+
+	if (strlen(kconsumerd_cmd_unix_sock_path) == 0) {
+		snprintf(kconsumerd_cmd_unix_sock_path, PATH_MAX, KCONSUMERD_CMD_SOCK_PATH);
+	}
+
+	ret = mkdir(KCONSUMERD_PATH, S_IRWXU | S_IRWXG);
+	if (ret < 0) {
+		ERR("Failed to create " KCONSUMERD_PATH);
+		goto error;
+	}
+
+	/* Create the kconsumerd error unix socket */
+	kconsumerd_err_sock = lttcomm_create_unix_sock(kconsumerd_err_unix_sock_path);
+	if (kconsumerd_err_sock < 0) {
+		ERR("Create unix sock failed: %s", kconsumerd_err_unix_sock_path);
+		ret = -1;
+		goto error;
+	}
+
+	/* File permission MUST be 660 */
+	ret = chmod(kconsumerd_err_unix_sock_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
+	if (ret < 0) {
+		ERR("Set file permissions failed: %s", kconsumerd_err_unix_sock_path);
+		perror("chmod");
+		goto error;
+	}
+
+error:
+	return ret;
+}
+
 /*
  *  set_signal_handler
  *
@@ -758,6 +913,8 @@ static int set_signal_handler(void)
 		return ret;
 	}
 
+	DBG("Signal handler set for SIGTERM, SIGPIPE and SIGINT");
+
 	return ret;
 }
 
@@ -770,9 +927,14 @@ static void sighandler(int sig)
 {
 	switch (sig) {
 		case SIGPIPE:
+			DBG("SIGPIPE catched");
 			return;
 		case SIGINT:
+			DBG("SIGINT catched");
+			cleanup();
+			break;
 		case SIGTERM:
+			DBG("SIGTERM catched");
 			cleanup();
 			break;
 		default:
@@ -789,6 +951,11 @@ static void sighandler(int sig)
  */
 static void cleanup()
 {
+	int ret;
+	char *cmd;
+
+	DBG("Cleaning up");
+
 	/* <fun> */
 	MSG("\n%c[%d;%dm*** assert failed *** ==> %c[%dm", 27,1,31,27,0);
 	MSG("%c[%d;%dmMatthew, BEET driven development works!%c[%dm",27,1,33,27,0);
@@ -796,6 +963,18 @@ static void cleanup()
 
 	unlink(client_unix_sock_path);
 	unlink(apps_unix_sock_path);
+	unlink(kconsumerd_err_unix_sock_path);
+
+	ret = asprintf(&cmd, "rm -rf " LTTNG_RUNDIR);
+	if (ret < 0) {
+		ERR("asprintf failed. Something is really wrong!");
+	}
+
+	/* Remove lttng run directory */
+	ret = system(cmd);
+	if (ret < 0) {
+		ERR("Unable to clean " LTTNG_RUNDIR);
+	}
 }
 
 /*
@@ -828,25 +1007,35 @@ int main(int argc, char **argv)
 
 	/* Set all sockets path */
 	if (is_root) {
+		ret = create_lttng_rundir();
+		if (ret < 0) {
+			goto error;
+		}
+
 		if (strlen(apps_unix_sock_path) == 0) {
-			(snprintf(apps_unix_sock_path, PATH_MAX,
-					DEFAULT_GLOBAL_APPS_UNIX_SOCK));
+			snprintf(apps_unix_sock_path, PATH_MAX,
+					DEFAULT_GLOBAL_APPS_UNIX_SOCK);
 		}
 
 		if (strlen(client_unix_sock_path) == 0) {
-			(snprintf(client_unix_sock_path, PATH_MAX,
-					DEFAULT_GLOBAL_CLIENT_UNIX_SOCK));
+			snprintf(client_unix_sock_path, PATH_MAX,
+					DEFAULT_GLOBAL_CLIENT_UNIX_SOCK);
+		}
+
+		ret = set_kconsumerd_sockets();
+		if (ret < 0) {
+			goto error;
 		}
 	} else {
 		if (strlen(apps_unix_sock_path) == 0) {
-			(snprintf(apps_unix_sock_path, PATH_MAX,
-					DEFAULT_HOME_APPS_UNIX_SOCK, get_home_dir()));
+			snprintf(apps_unix_sock_path, PATH_MAX,
+					DEFAULT_HOME_APPS_UNIX_SOCK, get_home_dir());
 		}
 
 		/* Set the cli tool unix socket path */
 		if (strlen(client_unix_sock_path) == 0) {
-			(snprintf(client_unix_sock_path, PATH_MAX,
-					DEFAULT_HOME_CLIENT_UNIX_SOCK, get_home_dir()));
+			snprintf(client_unix_sock_path, PATH_MAX,
+					DEFAULT_HOME_CLIENT_UNIX_SOCK, get_home_dir());
 		}
 	}
 
@@ -856,7 +1045,7 @@ int main(int argc, char **argv)
 	if ((ret = check_existing_daemon()) == 0) {
 		ERR("Already running daemon.\n");
 		/* We do not goto error because we must not
-		 * cleanup() because a daemon is already working.
+		 * cleanup() because a daemon is already running.
 		 */
 		return EXIT_FAILURE;
 	}
@@ -865,13 +1054,13 @@ int main(int argc, char **argv)
 		goto error;
 	}
 
-	/* Setup the two needed unix socket */
+	/* Setup the needed unix socket */
 	if (init_daemon_socket() < 0) {
 		goto error;
 	}
 
 	/* Set credentials to socket */
-	if (is_root && (set_socket_perms() < 0)) {
+	if (is_root && (set_permissions() < 0)) {
 		goto error;
 	}