X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fconsumer.c;h=0565f54dfad4da97503a6eb6575e4c51110edc32;hp=e3b6cd5bfd3d3673063b12c0d1900808a5258758;hb=07b86b528dc279d59cdf16e6cb946c144fe773f2;hpb=2bba9e532ca1910822005ff7f67400a2e871467c

diff --git a/src/common/consumer.c b/src/common/consumer.c
index e3b6cd5bf..0565f54df 100644
--- a/src/common/consumer.c
+++ b/src/common/consumer.c
@@ -312,6 +312,25 @@ void consumer_del_channel(struct lttng_consumer_channel *channel)
 		goto end;
 	}
 
+	/* Empty no monitor streams list. */
+	if (!channel->monitor) {
+		struct lttng_consumer_stream *stream, *stmp;
+
+		/*
+		 * So, these streams are not visible to any data thread. This is why we
+		 * close and free them because they were never added to any data
+		 * structure apart from this one.
+		 */
+		cds_list_for_each_entry_safe(stream, stmp,
+				&channel->stream_no_monitor_list.head, no_monitor_node) {
+			cds_list_del(&stream->no_monitor_node);
+			/* Close everything in that stream. */
+			consumer_stream_close(stream);
+			/* Free the ressource. */
+			consumer_stream_free(stream);
+		}
+	}
+
 	rcu_read_lock();
 	iter.iter.node = &channel->node.node;
 	ret = lttng_ht_del(consumer_data.channel_ht, &iter);
@@ -509,8 +528,10 @@ struct lttng_consumer_stream *consumer_allocate_stream(uint64_t channel_key,
 	/* Init session id node with the stream session id */
 	lttng_ht_node_init_u64(&stream->node_session_id, stream->session_id);
 
-	DBG3("Allocated stream %s (key %" PRIu64 ", chan_key %" PRIu64 " relayd_id %" PRIu64 ", session_id %" PRIu64,
-			stream->name, stream->key, channel_key, stream->net_seq_idx, stream->session_id);
+	DBG3("Allocated stream %s (key %" PRIu64 ", chan_key %" PRIu64
+			" relayd_id %" PRIu64 ", session_id %" PRIu64,
+			stream->name, stream->key, channel_key,
+			stream->net_seq_idx, stream->session_id);
 
 	rcu_read_unlock();
 	return stream;
@@ -769,6 +790,33 @@ struct lttng_consumer_channel *consumer_allocate_channel(uint64_t key,
 	channel->tracefile_count = tracefile_count;
 	channel->monitor = monitor;
 
+	/*
+	 * In monitor mode, the streams associated with the channel will be put in
+	 * a special list ONLY owned by this channel. So, the refcount is set to 1
+	 * here meaning that the channel itself has streams that are referenced.
+	 *
+	 * On a channel deletion, once the channel is no longer visible, the
+	 * refcount is decremented and checked for a zero value to delete it. With
+	 * streams in no monitor mode, it will now be safe to destroy the channel.
+	 */
+	if (!channel->monitor) {
+		channel->refcount = 1;
+	}
+
+	switch (output) {
+	case LTTNG_EVENT_SPLICE:
+		channel->output = CONSUMER_CHANNEL_SPLICE;
+		break;
+	case LTTNG_EVENT_MMAP:
+		channel->output = CONSUMER_CHANNEL_MMAP;
+		break;
+	default:
+		ERR("Allocate channel output unknown %d", output);
+		free(channel);
+		channel = NULL;
+		goto end;
+	}
+
 	strncpy(channel->pathname, pathname, sizeof(channel->pathname));
 	channel->pathname[sizeof(channel->pathname) - 1] = '\0';
 
@@ -780,6 +828,7 @@ struct lttng_consumer_channel *consumer_allocate_channel(uint64_t key,
 	channel->wait_fd = -1;
 
 	CDS_INIT_LIST_HEAD(&channel->streams.head);
+	CDS_INIT_LIST_HEAD(&channel->stream_no_monitor_list.head);
 
 	DBG("Allocated channel (key %" PRIu64 ")", channel->key)
 
@@ -1785,7 +1834,6 @@ void consumer_del_metadata_stream(struct lttng_consumer_stream *stream,
 				PERROR("munmap metadata stream");
 			}
 		}
-
 		if (stream->wait_fd >= 0) {
 			ret = close(stream->wait_fd);
 			if (ret < 0) {