Fix: miscellaneous memory handling fixes

[lttng-tools.git] / src / bin / lttng-relayd / health-relayd.c

diff --git a/src/bin/lttng-relayd/health-relayd.c b/src/bin/lttng-relayd/health-relayd.c
index 6b4a742f8e5fa95066e29b3e922ff13bd449eb31..8a5cbdfcb550f86a2010586e6eed924d19232861 100644 (file)
--- a/src/bin/lttng-relayd/health-relayd.c
+++ b/src/bin/lttng-relayd/health-relayd.c
@@ -53,7 +53,8 @@
 #include "health-relayd.h"
 
 /* Global health check unix path */
-static char health_unix_sock_path[PATH_MAX];
+static
+char health_unix_sock_path[PATH_MAX];
 
 int health_quit_pipe[2];
 
@@ -129,12 +130,32 @@ error:
        return ret;
 }
 
+static
+int parse_health_env(void)
+{
+       const char *health_path;
+
+       health_path = getenv(LTTNG_RELAYD_HEALTH_ENV);
+       if (health_path) {
+               strncpy(health_unix_sock_path, health_path,
+                       PATH_MAX);
+               health_unix_sock_path[PATH_MAX - 1] = '\0';
+       }
+
+       return 0;
+}
+
 static
 int setup_health_path(void)
 {
        int is_root, ret = 0;
        char *home_path = NULL, *rundir = NULL, *relayd_path;
 
+       ret = parse_health_env();
+       if (ret) {
+               return ret;
+       }
+
        is_root = !getuid();
 
        if (is_root) {
@@ -279,6 +300,8 @@ void *thread_manage_health(void *data)
                goto error;
        }
 
+       lttng_relay_notify_ready();
+
        while (1) {
                DBG("Health check ready");
 
@@ -345,7 +368,7 @@ restart:
 
                assert(msg.cmd == HEALTH_CMD_CHECK);
 
-               reply.ret_code = 0;
+               memset(&reply, 0, sizeof(reply));
                for (i = 0; i < NR_HEALTH_RELAYD_TYPES; i++) {
                        /*
                         * health_check_state return 0 if thread is in