Add missing rcu register thread calls

[lttng-tools.git] / liblttng-consumer / lttng-consumer.c

diff --git a/liblttng-consumer/lttng-consumer.c b/liblttng-consumer/lttng-consumer.c
index 617282d9c38b3690f5969f883690c7922c478b2b..5a4cf08c59109a7a5f21d8cc8070eb699debd3e0 100644 (file)
--- 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;
 }