X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=ltt-sessiond%2Fmain.c;h=c6913fa0a5fd41888507f576e6f8387b824e0f92;hb=8d5bb86ea2e1a863a5d0dd91249f32e1fcd29fd6;hp=c28358563ebc7da0a882888d6fad9d34b5af2309;hpb=ccca3feae44902bab18bafcd6a774b4fb28f22c1;p=lttng-tools.git

diff --git a/ltt-sessiond/main.c b/ltt-sessiond/main.c
index c28358563..c6913fa0a 100644
--- a/ltt-sessiond/main.c
+++ b/ltt-sessiond/main.c
@@ -875,6 +875,8 @@ static void *thread_manage_kconsumerd(void *data)
 		goto error;
 	}
 
+	DBG2("Receiving code from kconsumerd_err_sock");
+
 	/* Getting status code from kconsumerd */
 	ret = lttcomm_recv_unix_sock(sock, &code,
 			sizeof(enum lttcomm_return_code));
@@ -894,7 +896,7 @@ static void *thread_manage_kconsumerd(void *data)
 		sem_post(&kconsumerd_sem);
 		DBG("Kconsumerd command socket ready");
 	} else {
-		DBG("Kconsumerd error when waiting for SOCK_READY : %s",
+		ERR("Kconsumerd error when waiting for SOCK_READY : %s",
 				lttcomm_get_readable_code(-code));
 		goto error;
 	}
@@ -1323,29 +1325,66 @@ error:
 static int spawn_kconsumerd_thread(void)
 {
 	int ret;
+	struct timespec timeout;
+
+	timeout.tv_sec = DEFAULT_SEM_WAIT_TIMEOUT;
+	timeout.tv_nsec = 0;
 
 	/* Setup semaphore */
-	sem_init(&kconsumerd_sem, 0, 0);
+	ret = sem_init(&kconsumerd_sem, 0, 0);
+	if (ret < 0) {
+		PERROR("sem_init kconsumerd_sem");
+		goto error;
+	}
 
 	ret = pthread_create(&kconsumerd_thread, NULL,
 			thread_manage_kconsumerd, (void *) NULL);
 	if (ret != 0) {
-		perror("pthread_create kconsumerd");
+		PERROR("pthread_create kconsumerd");
+		ret = -1;
 		goto error;
 	}
 
-	/* Wait for the kconsumerd thread to be ready */
-	sem_wait(&kconsumerd_sem);
+	/* Get time for sem_timedwait absolute timeout */
+	ret = clock_gettime(CLOCK_REALTIME, &timeout);
+	if (ret < 0) {
+		PERROR("clock_gettime spawn kconsumerd");
+		/* Infinite wait for the kconsumerd thread to be ready */
+		ret = sem_wait(&kconsumerd_sem);
+	} else {
+		/* Normal timeout if the gettime was successful */
+		timeout.tv_sec += DEFAULT_SEM_WAIT_TIMEOUT;
+		ret = sem_timedwait(&kconsumerd_sem, &timeout);
+	}
 
+	if (ret < 0) {
+		if (errno == ETIMEDOUT) {
+			/*
+			 * Call has timed out so we kill the kconsumerd_thread and return
+			 * an error.
+			 */
+			ERR("The kconsumerd thread was never ready. Killing it");
+			ret = pthread_cancel(kconsumerd_thread);
+			if (ret < 0) {
+				PERROR("pthread_cancel kconsumerd_thread");
+			}
+		} else {
+			PERROR("semaphore wait failed kconsumerd thread");
+		}
+		goto error;
+	}
+
+	pthread_mutex_lock(&kconsumerd_pid_mutex);
 	if (kconsumerd_pid == 0) {
 		ERR("Kconsumerd did not start");
+		pthread_mutex_unlock(&kconsumerd_pid_mutex);
 		goto error;
 	}
+	pthread_mutex_unlock(&kconsumerd_pid_mutex);
 
 	return 0;
 
 error:
-	ret = LTTCOMM_KERN_CONSUMER_FAIL;
 	return ret;
 }
 
@@ -1427,18 +1466,16 @@ static int start_kconsumerd(void)
 	ret = spawn_kconsumerd();
 	if (ret < 0) {
 		ERR("Spawning kconsumerd failed");
-		ret = LTTCOMM_KERN_CONSUMER_FAIL;
 		pthread_mutex_unlock(&kconsumerd_pid_mutex);
 		goto error;
 	}
 
 	/* Setting up the global kconsumerd_pid */
 	kconsumerd_pid = ret;
+	DBG2("Kconsumerd pid %d", kconsumerd_pid);
 	pthread_mutex_unlock(&kconsumerd_pid_mutex);
 
-	DBG("Kconsumerd pid %d", ret);
-
-	DBG("Spawning kconsumerd thread");
+	DBG2("Spawning kconsumerd thread");
 	ret = spawn_kconsumerd_thread();
 	if (ret < 0) {
 		ERR("Fatal error spawning kconsumerd thread");
@@ -2585,7 +2622,9 @@ static int process_client_msg(struct command_ctx *cmd_ctx)
 		break;
 	default:
 		DBG("Getting session %s by name", cmd_ctx->lsm->session.name);
+		session_lock_list();
 		cmd_ctx->session = session_find_by_name(cmd_ctx->lsm->session.name);
+		session_unlock_list();
 		if (cmd_ctx->session == NULL) {
 			if (cmd_ctx->lsm->session.name != NULL) {
 				ret = LTTCOMM_SESS_NOT_FOUND;
@@ -2627,14 +2666,17 @@ static int process_client_msg(struct command_ctx *cmd_ctx)
 			}
 
 			/* Start the kernel consumer daemon */
+			pthread_mutex_lock(&kconsumerd_pid_mutex);
 			if (kconsumerd_pid == 0 &&
 					cmd_ctx->lsm->cmd_type != LTTNG_REGISTER_CONSUMER) {
+				pthread_mutex_unlock(&kconsumerd_pid_mutex);
 				ret = start_kconsumerd();
 				if (ret < 0) {
 					ret = LTTCOMM_KERN_CONSUMER_FAIL;
 					goto error;
 				}
 			}
+			pthread_mutex_unlock(&kconsumerd_pid_mutex);
 		}
 		break;
 	case LTTNG_DOMAIN_UST_PID: