X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fust-thread.c;h=0248d7a0aa305a17cfba12cb8ceb60deb2266472;hp=552b7ddd8176a6a246fd90168189dce3dd691aba;hb=6cd525e813795a1d5e38feac8dedf2c73ffb1274;hpb=5d2e1e66a968d9e555f9b8b00d0589ebfaf3de32

diff --git a/src/bin/lttng-sessiond/ust-thread.c b/src/bin/lttng-sessiond/ust-thread.c
index 552b7ddd8..0248d7a0a 100644
--- a/src/bin/lttng-sessiond/ust-thread.c
+++ b/src/bin/lttng-sessiond/ust-thread.c
@@ -23,13 +23,15 @@
 #include "fd-limit.h"
 #include "lttng-sessiond.h"
 #include "ust-thread.h"
+#include "health-sessiond.h"
 
 /*
  * This thread manage application notify communication.
  */
 void *ust_thread_manage_notify(void *data)
 {
-	int i, ret, pollfd;
+	int i, ret, pollfd, err = -1;
+	ssize_t size_ret;
 	uint32_t revents, nb_fd;
 	struct lttng_poll_event events;
 
@@ -38,6 +40,11 @@ void *ust_thread_manage_notify(void *data)
 	rcu_register_thread();
 	rcu_thread_online();
 
+	health_register(health_sessiond,
+		HEALTH_SESSIOND_TYPE_APP_MANAGE_NOTIFY);
+
+	health_code_update();
+
 	ret = sessiond_set_thread_pollset(&events, 2);
 	if (ret < 0) {
 		goto error_poll_create;
@@ -49,13 +56,17 @@ void *ust_thread_manage_notify(void *data)
 		goto error;
 	}
 
+	health_code_update();
+
 	while (1) {
 		DBG3("[ust-thread] Manage notify polling on %d fds",
 				LTTNG_POLL_GETNB(&events));
 
 		/* Inifinite blocking call, waiting for transmission */
 restart:
+		health_poll_entry();
 		ret = lttng_poll_wait(&events, -1);
+		health_poll_exit();
 		if (ret < 0) {
 			/*
 			 * Restart interrupted system call.
@@ -69,6 +80,8 @@ restart:
 		nb_fd = ret;
 
 		for (i = 0; i < nb_fd; i++) {
+			health_code_update();
+
 			/* Fetch once the poll data */
 			revents = LTTNG_POLL_GETEV(&events, i);
 			pollfd = LTTNG_POLL_GETFD(&events, i);
@@ -76,6 +89,7 @@ restart:
 			/* Thread quit pipe has been closed. Killing thread. */
 			ret = sessiond_check_thread_quit_pipe(pollfd, revents);
 			if (ret) {
+				err = 0;
 				goto exit;
 			}
 
@@ -92,14 +106,14 @@ restart:
 					goto error;
 				}
 
-				do {
-					/* Get socket from dispatch thread. */
-					ret = read(apps_cmd_notify_pipe[0], &sock, sizeof(sock));
-				} while (ret < 0 && errno == EINTR);
-				if (ret < 0 || ret < sizeof(sock)) {
+				/* Get socket from dispatch thread. */
+				size_ret = lttng_read(apps_cmd_notify_pipe[0],
+						&sock, sizeof(sock));
+				if (size_ret < sizeof(sock)) {
 					PERROR("read apps notify pipe");
 					goto error;
 				}
+				health_code_update();
 
 				ret = lttng_poll_add(&events, sock,
 						LPOLLIN | LPOLLERR | LPOLLHUP | LPOLLRDHUP);
@@ -147,6 +161,7 @@ restart:
 					ERR("Unknown poll events %u for sock %d", revents, pollfd);
 					continue;
 				}
+				health_code_update();
 			}
 		}
 	}
@@ -158,6 +173,11 @@ error_poll_create:
 	utils_close_pipe(apps_cmd_notify_pipe);
 	apps_cmd_notify_pipe[0] = apps_cmd_notify_pipe[1] = -1;
 	DBG("Application notify communication apps thread cleanup complete");
+	if (err) {
+		health_error();
+		ERR("Health error occurred in %s", __func__);
+	}
+	health_unregister(health_sessiond);
 	rcu_thread_offline();
 	rcu_unregister_thread();
 	return NULL;