From 005aeac5da3c85a5f57f570e3279ab7436ac81e6 Mon Sep 17 00:00:00 2001 From: Jonathan Rajotte Date: Mon, 5 Feb 2018 17:58:19 -0500 Subject: [PATCH] Use ust_lock during sock_info operations for atomicity against fork Signed-off-by: Jonathan Rajotte Signed-off-by: Mathieu Desnoyers --- liblttng-ust/lttng-ust-comm.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/liblttng-ust/lttng-ust-comm.c b/liblttng-ust/lttng-ust-comm.c index e92bb870..6c11288c 100644 --- a/liblttng-ust/lttng-ust-comm.c +++ b/liblttng-ust/lttng-ust-comm.c @@ -1382,6 +1382,10 @@ restart: prev_connect_failed = 0; } + if (ust_lock()) { + goto quit; + } + if (sock_info->socket != -1) { /* FD tracker is updated by ustcomm_close_unix_sock() */ ret = ustcomm_close_unix_sock(sock_info->socket); @@ -1401,9 +1405,6 @@ restart: sock_info->notify_socket = -1; } - if (ust_lock()) { - goto quit; - } /* * Register. We need to perform both connect and sending -- 2.34.1