X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fsessiond-trace-chunks.c;h=7005ef4965b5b99ce40e7ea931222edf3ccbf4b1;hb=2c93dc32c897a1bf6535a84817a6e5430b51189b;hp=1065234739a8d6e529c1dc58ba3dde24140c2666;hpb=e8a09c095b320c14cf6791c8709c6744d57d9416;p=lttng-tools.git

diff --git a/src/bin/lttng-relayd/sessiond-trace-chunks.c b/src/bin/lttng-relayd/sessiond-trace-chunks.c
index 106523473..7005ef496 100644
--- a/src/bin/lttng-relayd/sessiond-trace-chunks.c
+++ b/src/bin/lttng-relayd/sessiond-trace-chunks.c
@@ -360,6 +360,7 @@ struct lttng_trace_chunk *sessiond_trace_chunk_registry_publish_chunk(
 	char uuid_str[LTTNG_UUID_STR_LEN];
 	char chunk_id_str[MAX_INT_DEC_LEN(typeof(chunk_id))] = "-1";
 	struct lttng_trace_chunk *published_chunk = NULL;
+	bool trace_chunk_already_published;
 
 	lttng_uuid_to_str(sessiond_uuid, uuid_str);
 	lttng_uuid_copy(key.sessiond_uuid, sessiond_uuid);
@@ -393,21 +394,30 @@ struct lttng_trace_chunk *sessiond_trace_chunk_registry_publish_chunk(
 		goto end;
 	}
 
-	published_chunk = lttng_trace_chunk_registry_publish_chunk(
-			element->trace_chunk_registry, session_id, new_chunk);
+	published_chunk = lttng_trace_chunk_registry_publish_chunk_published(
+			element->trace_chunk_registry, session_id, new_chunk,
+			&trace_chunk_already_published);
 	/*
-	 * At this point, two references to the published chunks exist. One
-	 * is taken by the registry while the other is being returned to the
-	 * caller. In the use case of the relay daemon, the reference held
-	 * by the registry itself is undesirable.
+	 * When the trace chunk is first published, two references to the
+	 * published chunks exist. One is taken by the registry while the other
+	 * is being returned to the caller. In the use case of the relay daemon,
+	 * the reference held by the registry itself is undesirable.
 	 *
 	 * We want the trace chunk to be removed from the registry as soon
 	 * as it is not being used by the relay daemon (through a session
 	 * or a stream). This differs from the behaviour of the consumer
 	 * daemon which relies on an explicit command from the session
 	 * daemon to release the registry's reference.
+	 *
+	 * In cases where the trace chunk had already been published,
+	 * the reference belonging to the sessiond trace chunk
+	 * registry instance has already been 'put'. We simply return
+	 * the published trace chunk with a reference taken on behalf of the
+	 * caller.
 	 */
-	lttng_trace_chunk_put(published_chunk);
+	if (!trace_chunk_already_published) {
+		lttng_trace_chunk_put(published_chunk);
+	}
 end:
 	trace_chunk_registry_ht_element_put(element);
 	return published_chunk;