From: David Goulet <dgoulet@efficios.com>
Date: Thu, 12 Jan 2012 15:59:38 +0000 (-0500)
Subject: Add missing rcu register thread calls
X-Git-Tag: v2.0-pre17~24
X-Git-Url: https://git.lttng.org./?a=commitdiff_plain;h=e7b994a365c48d2dbe576b64811ec861da7c6375;p=lttng-tools.git

Add missing rcu register thread calls

Signed-off-by: David Goulet <dgoulet@efficios.com>
---

diff --git a/liblttng-consumer/lttng-consumer.c b/liblttng-consumer/lttng-consumer.c
index 617282d9c..5a4cf08c5 100644
--- a/liblttng-consumer/lttng-consumer.c
+++ b/liblttng-consumer/lttng-consumer.c
@@ -845,6 +845,8 @@ void *lttng_consumer_thread_poll_fds(void *data)
 	int tmp2;
 	struct lttng_consumer_local_data *ctx = data;
 
+	rcu_register_thread();
+
 	local_stream = zmalloc(sizeof(struct lttng_consumer_stream));
 
 	while (1) {
@@ -1005,6 +1007,7 @@ end:
 		free(local_stream);
 		local_stream = NULL;
 	}
+	rcu_unregister_thread();
 	return NULL;
 }
 
@@ -1022,6 +1025,8 @@ void *lttng_consumer_thread_receive_fds(void *data)
 	struct pollfd consumer_sockpoll[2];
 	struct lttng_consumer_local_data *ctx = data;
 
+	rcu_register_thread();
+
 	DBG("Creating command socket %s", ctx->consumer_command_sock_path);
 	unlink(ctx->consumer_command_sock_path);
 	client_socket = lttcomm_create_unix_sock(ctx->consumer_command_sock_path);
@@ -1117,6 +1122,7 @@ end:
 	if (ret < 0) {
 		perror("poll pipe write");
 	}
+	rcu_unregister_thread();
 	return NULL;
 }