Tests: Fix possible segfault in health check thread exit test

[lttng-tools.git] / tests / regression / tools / health / health_exit.c

diff --git a/tests/regression/tools/health/health_exit.c b/tests/regression/tools/health/health_exit.c
index 8e414053e9d961867acc2b075ec8f1df6342af9f..8d1d05c5311dfa069bc66557a0448bb31ef10d72 100644 (file)
--- a/tests/regression/tools/health/health_exit.c
+++ b/tests/regression/tools/health/health_exit.c
@@ -18,6 +18,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <pthread.h>
+#include <urcu.h>
 
 /*
  * Check if the specified environment variable is set.
@@ -40,6 +41,7 @@ int __testpoint_thread_manage_clients(void)
        const char *var = "LTTNG_THREAD_MANAGE_CLIENTS_EXIT";
 
        if (check_env_var(var)) {
+               rcu_unregister_thread();
                pthread_exit(NULL);
        }
 
@@ -62,6 +64,7 @@ int __testpoint_thread_manage_apps(void)
        const char *var = "LTTNG_THREAD_MANAGE_APPS_EXIT";
 
        if (check_env_var(var)) {
+               rcu_unregister_thread();
                pthread_exit(NULL);
        }