X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fmain.c;h=1bdef652eb64d7ad60ffb3b582ca540f24a62361;hp=f943488be5b94309efd332196bd52e876d2eea1a;hb=87b576ecd158f43c99a16b9c7c2092f7291aee00;hpb=43c5442ea57aca20701fd49fba1e3456d48cc7b2

diff --git a/src/bin/lttng-relayd/main.c b/src/bin/lttng-relayd/main.c
index f943488be..1bdef652e 100644
--- a/src/bin/lttng-relayd/main.c
+++ b/src/bin/lttng-relayd/main.c
@@ -898,6 +898,8 @@ void relay_delete_session(struct relay_command *cmd,
 		stream = caa_container_of(node, struct relay_stream, stream_n);
 		if (stream->session == cmd->session) {
 			destroy_stream(stream);
+			cmd->session->stream_count--;
+			assert(cmd->session->stream_count >= 0);
 		}
 	}
 
@@ -1094,6 +1096,7 @@ int relay_add_stream(struct lttcomm_relayd_hdr *recv_hdr,
 
 	lttng_ht_node_init_str(&stream->ctf_trace_node, stream->path_name);
 	lttng_ht_add_str(cmd->ctf_traces_ht, &stream->ctf_trace_node);
+	session->stream_count++;
 
 	DBG("Relay new stream added %s with ID %" PRIu64, stream->channel_name,
 			stream->stream_handle);
@@ -1170,6 +1173,8 @@ int relay_close_stream(struct lttcomm_relayd_hdr *recv_hdr,
 
 	stream->last_net_seq_num = be64toh(stream_info.last_net_seq_num);
 	stream->close_flag = 1;
+	session->stream_count--;
+	assert(session->stream_count >= 0);
 
 	if (close_stream_check(stream)) {
 		destroy_stream(stream);