X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Frelayd%2Frelayd.c;h=6f877c96813d8a78b6d5f2f82235b43058d8ed43;hp=8945afea3645948d07a590b37ce81a916ab372ff;hb=173af62f4804133d4a7f45e34b6f72126f3eca5f;hpb=f64161251bd649abe5b6a473531adfa3af9bd6b6

diff --git a/src/common/relayd/relayd.c b/src/common/relayd/relayd.c
index 8945afea3..6f877c968 100644
--- a/src/common/relayd/relayd.c
+++ b/src/common/relayd/relayd.c
@@ -385,3 +385,50 @@ int relayd_send_data_hdr(struct lttcomm_sock *sock,
 error:
 	return ret;
 }
+
+/*
+ * Send close stream command to the relayd.
+ */
+int relayd_send_close_stream(struct lttcomm_sock *sock, uint64_t stream_id,
+		uint64_t last_net_seq_num)
+{
+	int ret;
+	struct lttcomm_relayd_close_stream msg;
+	struct lttcomm_relayd_generic_reply reply;
+
+	/* Code flow error. Safety net. */
+	assert(sock);
+
+	DBG("Relayd closing stream id %zu", stream_id);
+
+	msg.stream_id = htobe64(stream_id);
+	msg.last_net_seq_num = htobe64(last_net_seq_num);
+
+	/* Send command */
+	ret = send_command(sock, RELAYD_CLOSE_STREAM, (void *) &msg, sizeof(msg), 0);
+	if (ret < 0) {
+		goto error;
+	}
+
+	/* Recevie response */
+	ret = recv_reply(sock, (void *) &reply, sizeof(reply));
+	if (ret < 0) {
+		goto error;
+	}
+
+	reply.ret_code = be32toh(reply.ret_code);
+
+	/* Return session id or negative ret code. */
+	if (reply.ret_code != LTTCOMM_OK) {
+		ret = -reply.ret_code;
+		ERR("Relayd close stream replied error %d", ret);
+	} else {
+		/* Success */
+		ret = 0;
+	}
+
+	DBG("Relayd close stream id %zu successfully", stream_id);
+
+error:
+	return ret;
+}