X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Frelayd%2Frelayd.c;h=242abbedff15c5c2f079460cb8805dbc567268cd;hp=458553bfe997069706aa9b43987a99d1cfcf0ece;hb=d73bf3d793ee0b0c5b56cb47cb50c27d1789d3bd;hpb=e2acc8d25bc0073554a2f3e742e070b7034ec1d0

diff --git a/src/common/relayd/relayd.c b/src/common/relayd/relayd.c
index 458553bfe..242abbedf 100644
--- a/src/common/relayd/relayd.c
+++ b/src/common/relayd/relayd.c
@@ -942,6 +942,84 @@ error:
 	return ret;
 }
 
+int relayd_rotate_stream(struct lttcomm_relayd_sock *rsock, uint64_t stream_id,
+		const char *new_pathname, uint64_t new_chunk_id,
+		uint64_t seq_num)
+{
+	int ret;
+	struct lttcomm_relayd_rotate_stream *msg = NULL;
+	struct lttcomm_relayd_generic_reply reply;
+	size_t len;
+	int msg_len;
+
+	/* Code flow error. Safety net. */
+	assert(rsock);
+
+	DBG("Sending rotate stream id %" PRIu64 " command to relayd", stream_id);
+
+	/* Account for the trailing NULL. */
+	len = strnlen(new_pathname, LTTNG_PATH_MAX) + 1;
+	if (len > LTTNG_PATH_MAX) {
+		ERR("Path used in relayd rotate stream command exceeds the maximal allowed length");
+		ret = -1;
+		goto error;
+	}
+
+	msg_len = offsetof(struct lttcomm_relayd_rotate_stream, new_pathname) + len;
+	msg = zmalloc(msg_len);
+	if (!msg) {
+		PERROR("Failed to allocate relayd rotate stream command of %d bytes",
+				msg_len);
+		ret = -1;
+		goto error;
+	}
+
+	if (lttng_strncpy(msg->new_pathname, new_pathname, len)) {
+		ret = -1;
+		ERR("Failed to copy relayd rotate stream command's new path name");
+		goto error;
+	}
+
+	msg->pathname_length = htobe32(len);
+	msg->stream_id = htobe64(stream_id);
+	msg->new_chunk_id = htobe64(new_chunk_id);
+	/*
+	 * The seq_num is invalid for metadata streams, but it is ignored on
+	 * the relay.
+	 */
+	msg->rotate_at_seq_num = htobe64(seq_num);
+
+	/* Send command. */
+	ret = send_command(rsock, RELAYD_ROTATE_STREAM, (void *) msg, msg_len, 0);
+	if (ret < 0) {
+		ERR("Send rotate command");
+		goto error;
+	}
+
+	/* Receive response. */
+	ret = recv_reply(rsock, (void *) &reply, sizeof(reply));
+	if (ret < 0) {
+		ERR("Receive rotate reply");
+		goto error;
+	}
+
+	reply.ret_code = be32toh(reply.ret_code);
+
+	/* Return session id or negative ret code. */
+	if (reply.ret_code != LTTNG_OK) {
+		ret = -1;
+		ERR("Relayd rotate stream replied error %d", reply.ret_code);
+	} else {
+		/* Success. */
+		ret = 0;
+		DBG("Relayd rotated stream id %" PRIu64 " successfully", stream_id);
+	}
+
+error:
+	free(msg);
+	return ret;
+}
+
 int relayd_rotate_rename(struct lttcomm_relayd_sock *rsock,
 		const char *old_path, const char *new_path)
 {