X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fcommon%2Fconsumer-stream.c;h=717e0a7351a49461e4702c872fc91914b8efaffe;hb=f82d9449a304561c5914b2c5f9471a640ca99248;hp=03bac86870031eb2b63661f178f5cee0601748c6;hpb=4891ece8d4eeb2645efdf1467680b037c63ab425;p=lttng-tools.git

diff --git a/src/common/consumer-stream.c b/src/common/consumer-stream.c
index 03bac8687..717e0a735 100644
--- a/src/common/consumer-stream.c
+++ b/src/common/consumer-stream.c
@@ -58,8 +58,10 @@ void consumer_stream_relayd_close(struct lttng_consumer_stream *stream,
 	assert(stream);
 	assert(relayd);
 
-	uatomic_dec(&relayd->refcount);
-	assert(uatomic_read(&relayd->refcount) >= 0);
+	if (stream->sent_to_relayd) {
+		uatomic_dec(&relayd->refcount);
+		assert(uatomic_read(&relayd->refcount) >= 0);
+	}
 
 	/* Closing streams requires to lock the control socket. */
 	pthread_mutex_lock(&relayd->ctrl_sock_mutex);
@@ -82,6 +84,7 @@ void consumer_stream_relayd_close(struct lttng_consumer_stream *stream,
 		consumer_destroy_relayd(relayd);
 	}
 	stream->net_seq_idx = (uint64_t) -1ULL;
+	stream->sent_to_relayd = 0;
 }
 
 /*
@@ -277,6 +280,7 @@ void consumer_stream_destroy(struct lttng_consumer_stream *stream,
 		 */
 		if (stream->globally_visible) {
 			pthread_mutex_lock(&consumer_data.lock);
+			pthread_mutex_lock(&stream->chan->lock);
 			pthread_mutex_lock(&stream->lock);
 			/* Remove every reference of the stream in the consumer. */
 			consumer_stream_delete(stream, ht);
@@ -290,6 +294,7 @@ void consumer_stream_destroy(struct lttng_consumer_stream *stream,
 			consumer_data.need_update = 1;
 
 			pthread_mutex_unlock(&stream->lock);
+			pthread_mutex_unlock(&stream->chan->lock);
 			pthread_mutex_unlock(&consumer_data.lock);
 		} else {
 			/*