From: Jonathan Rajotte <jonathan.rajotte-julien@efficios.com>
Date: Mon, 5 Feb 2018 22:58:19 +0000 (-0500)
Subject: Use ust_lock during sock_info operations for atomicity against fork
X-Git-Tag: v2.11.0-rc1~27
X-Git-Url: https://git.lttng.org./?a=commitdiff_plain;h=101dace0dc164352c81cb347a7a0022aa84eabc6;p=lttng-ust.git

Use ust_lock during sock_info operations for atomicity against fork

Signed-off-by: Jonathan Rajotte <jonathan.rajotte-julien@efficios.com>
Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
---

diff --git a/liblttng-ust/lttng-ust-comm.c b/liblttng-ust/lttng-ust-comm.c
index 32ba3c1d..0cf5ecd9 100644
--- a/liblttng-ust/lttng-ust-comm.c
+++ b/liblttng-ust/lttng-ust-comm.c
@@ -1396,6 +1396,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);
@@ -1415,9 +1419,6 @@ restart:
 		sock_info->notify_socket = -1;
 	}
 
-	if (ust_lock()) {
-		goto quit;
-	}
 
 	/*
 	 * Register. We need to perform both connect and sending