X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fbin%2Flttng-relayd%2Fviewer-stream.c;h=acd488b61846202813e8a7efe07898c1f67cfef7;hb=c51b2177a083e94884c11cb3c2a6bbfe9dc463f7;hp=70cd1cea15c13c27823c7555383d37f26f07cae9;hpb=dc3bfa8d4cbb95c7420673b677ef1b2447673642;p=lttng-tools.git

diff --git a/src/bin/lttng-relayd/viewer-stream.c b/src/bin/lttng-relayd/viewer-stream.c
index 70cd1cea1..acd488b61 100644
--- a/src/bin/lttng-relayd/viewer-stream.c
+++ b/src/bin/lttng-relayd/viewer-stream.c
@@ -27,7 +27,6 @@
 
 static void viewer_stream_destroy(struct relay_viewer_stream *vstream)
 {
-	lttng_trace_chunk_put(vstream->stream_file.trace_chunk);
 	free(vstream->path_name);
 	free(vstream->channel_name);
 	free(vstream);
@@ -42,11 +41,12 @@ static void viewer_stream_destroy_rcu(struct rcu_head *head)
 }
 
 struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream,
+		struct lttng_trace_chunk *viewer_trace_chunk,
 		enum lttng_viewer_seek seek_t)
 {
 	struct relay_viewer_stream *vstream = NULL;
 	const bool acquired_reference = lttng_trace_chunk_get(
-			stream->trace_chunk);
+			viewer_trace_chunk);
 
 	if (!acquired_reference) {
 		goto error;
@@ -58,7 +58,8 @@ struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream,
 		goto error;
 	}
 
-	vstream->stream_file.trace_chunk = stream->trace_chunk;
+	vstream->stream_file.trace_chunk = viewer_trace_chunk;
+	viewer_trace_chunk = NULL;
 	vstream->path_name = lttng_strndup(stream->path_name, LTTNG_VIEWER_PATH_MAX);
 	if (vstream->path_name == NULL) {
 		PERROR("relay viewer path_name alloc");
@@ -104,7 +105,7 @@ struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream,
 	}
 	case LTTNG_VIEWER_SEEK_LAST:
 		vstream->current_tracefile_id =
-			tracefile_array_get_file_index_head(stream->tfa);
+			tracefile_array_get_read_file_index_head(stream->tfa);
 		/*
 		 * We seek at the very end of each stream, awaiting for
 		 * a future packet to eventually come in.
@@ -130,7 +131,8 @@ struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream,
 		const uint32_t connection_minor = stream->trace->session->minor;
 
 		vstream->index_file = lttng_index_file_create_from_trace_chunk_read_only(
-				stream->trace_chunk, stream->path_name,
+				vstream->stream_file.trace_chunk,
+				stream->path_name,
 				stream->channel_name, stream->tracefile_size,
 				vstream->current_tracefile_id,
 				lttng_to_index_major(connection_major,
@@ -158,9 +160,8 @@ struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream,
 
 	/* Globally visible after the add unique. */
 	lttng_ht_node_init_u64(&vstream->stream_n, stream->stream_handle);
-	lttng_ht_add_unique_u64(viewer_streams_ht, &vstream->stream_n);
-
 	urcu_ref_init(&vstream->ref);
+	lttng_ht_add_unique_u64(viewer_streams_ht, &vstream->stream_n);
 
 	return vstream;
 
@@ -170,6 +171,9 @@ error:
 	if (vstream) {
 		viewer_stream_destroy(vstream);
 	}
+	if (viewer_trace_chunk && acquired_reference) {
+		lttng_trace_chunk_put(viewer_trace_chunk);
+	}
 	return NULL;
 }
 
@@ -206,7 +210,8 @@ static void viewer_stream_release(struct urcu_ref *ref)
 		stream_put(vstream->stream);
 		vstream->stream = NULL;
 	}
-
+	lttng_trace_chunk_put(vstream->stream_file.trace_chunk);
+	vstream->stream_file.trace_chunk = NULL;
 	call_rcu(&vstream->rcu_node, viewer_stream_destroy_rcu);
 }
 
@@ -313,7 +318,8 @@ int viewer_stream_rotate(struct relay_viewer_stream *vstream)
 	}
 	vstream->index_file =
 			lttng_index_file_create_from_trace_chunk_read_only(
-					stream->trace_chunk, stream->path_name,
+					vstream->stream_file.trace_chunk,
+					stream->path_name,
 					stream->channel_name,
 					stream->tracefile_size,
 					vstream->current_tracefile_id,