X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=libust%2Ftracectl.c;h=2e9052386c7121131d91c0495522752c96a9b57b;hb=fc253ce0fd26aa660b249013f4c802e54eefc0e9;hp=916b3548eee3de1f953713159c44794431c9874f;hpb=d5adede06d9dc0c4675c4c4052eb98f74973f4a5;p=ust.git

diff --git a/libust/tracectl.c b/libust/tracectl.c
index 916b354..2e90523 100644
--- a/libust/tracectl.c
+++ b/libust/tracectl.c
@@ -17,6 +17,7 @@
 
 #define _GNU_SOURCE
 #include <stdio.h>
+#include <stdlib.h>
 #include <stdint.h>
 #include <signal.h>
 #include <sys/types.h>
@@ -26,8 +27,7 @@
 #include <fcntl.h>
 #include <poll.h>
 #include <regex.h>
-
-#include <urcu-bp.h>
+#include <urcu/uatomic_arch.h>
 
 #include <ust/marker.h>
 #include <ust/tracectl.h>
@@ -125,15 +125,6 @@ static void print_markers(FILE *fp)
 
 static int init_socket(void);
 
-/* This needs to be called whenever a new thread is created. It notifies
- * liburcu of the new thread.
- */
-
-void ust_register_thread(void)
-{
-	rcu_register_thread();
-}
-
 int fd_notif = -1;
 void notif_cb(void)
 {
@@ -241,9 +232,13 @@ void process_blocked_consumers(void)
 				continue;
 			}
 			if(result == 0) {
+				int res;
 				DBG("PRODUCER END");
 
-				close(bc->fd_producer);
+				res = close(bc->fd_producer);
+				if(res == -1) {
+					PERROR("close");
+				}
 
 				list_del(&bc->list);
 
@@ -659,7 +654,7 @@ static int do_cmd_get_subbuffer(const char *recvbuf, struct ustcomm_source *src)
 	if(trace == NULL) {
 		int result;
 
-		WARN("Cannot find trace. It was likely destroyed by the user.");
+		DBG("Cannot find trace. It was likely destroyed by the user.");
 		result = ustcomm_send_reply(&ustcomm_app.server, "NOTFOUND", src);
 		if(result) {
 			ERR("ustcomm_send_reply failed");
@@ -695,7 +690,7 @@ static int do_cmd_get_subbuffer(const char *recvbuf, struct ustcomm_source *src)
 			/* Being here is the proof the daemon has mapped the buffer in its
 			 * memory. We may now decrement buffers_to_export.
 			 */
-			if(atomic_long_read(&buf->consumed) == 0) {
+			if(uatomic_read(&buf->consumed) == 0) {
 				DBG("decrementing buffers_to_export");
 				buffers_to_export--;
 			}
@@ -766,7 +761,7 @@ static int do_cmd_put_subbuffer(const char *recvbuf, struct ustcomm_source *src)
 	trace = _ltt_trace_find(trace_name);
 
 	if(trace == NULL) {
-		WARN("Cannot find trace. It was likely destroyed by the user.");
+		DBG("Cannot find trace. It was likely destroyed by the user.");
 		result = ustcomm_send_reply(&ustcomm_app.server, "NOTFOUND", src);
 		if(result) {
 			ERR("ustcomm_send_reply failed");
@@ -824,14 +819,19 @@ static int do_cmd_put_subbuffer(const char *recvbuf, struct ustcomm_source *src)
 	return retval;
 }
 
+static void listener_cleanup(void *ptr)
+{
+	ustcomm_fini_app(&ustcomm_app, 0);
+}
+
 void *listener_main(void *p)
 {
 	int result;
 
-	ust_register_thread();
-
 	DBG("LISTENER");
 
+	pthread_cleanup_push(listener_cleanup, NULL);
+
 	for(;;) {
 		char trace_name[] = "auto";
 		char trace_type[] = "ustrelay";
@@ -1106,17 +1106,20 @@ void *listener_main(void *p)
 	next_cmd:
 		free(recvbuf);
 	}
+
+	pthread_cleanup_pop(1);
 }
 
 volatile sig_atomic_t have_listener = 0;
+#ifndef USE_CLONE
+static pthread_t listener_thread;
+#endif
 
 void create_listener(void)
 {
 #ifdef USE_CLONE
 	static char listener_stack[16384];
 	int result;
-#else
-	pthread_t thread;
 #endif
 
 	if(have_listener) {
@@ -1132,7 +1135,7 @@ void create_listener(void)
 	}
 #else
 
-	pthread_create(&thread, NULL, listener_main, NULL);
+	pthread_create(&listener_thread, NULL, listener_main, NULL);
 #endif
 
 	have_listener = 1;
@@ -1191,16 +1194,8 @@ static void __attribute__((constructor)) init()
 	 */
 	pidunique = make_pidunique();
 
-	/* Initialize RCU in case the constructor order is not good. */
-	rcu_init();
-
-	/* It is important to do this before events start to be generated. */
-	ust_register_thread();
-
 	DBG("Tracectl constructor");
 
-	/* Must create socket before signal handler to prevent races.
-         */
 	result = init_socket();
 	if(result == -1) {
 		ERR("init_socket error");
@@ -1400,6 +1395,20 @@ int restarting_usleep(useconds_t usecs)
 	return result;
 }
 
+static void stop_listener()
+{
+	int result;
+
+	result = pthread_cancel(listener_thread);
+	if(result == -1) {
+		PERROR("pthread_cancel");
+	}
+	result = pthread_join(listener_thread, NULL);
+	if(result == -1) {
+		PERROR("pthread_join");
+	}
+}
+
 /* This destructor keeps the process alive for a few seconds in order
  * to leave time to ustd to connect to its buffers. This is necessary
  * for programs whose execution is very short. It is also useful in all
@@ -1430,7 +1439,8 @@ static void __attribute__((destructor)) keepalive()
 
 	destroy_traces();
 
-	ustcomm_fini_app(&ustcomm_app);
+	/* Ask the listener to stop and clean up. */
+	stop_listener();
 }
 
 void ust_potential_exec(void)
@@ -1478,6 +1488,10 @@ static void ust_fork(void)
 		list_del(&bc->list);
 	}
 
+	/* free app, keeping socket file */
+	ustcomm_fini_app(&ustcomm_app, 1);
+
+	buffers_to_export = 0;
 	have_listener = 0;
 	init_socket();
 	create_listener();