X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=src%2Fcommon%2Frelayd%2Frelayd.c;h=242abbedff15c5c2f079460cb8805dbc567268cd;hb=d73bf3d793ee0b0c5b56cb47cb50c27d1789d3bd;hp=7e6520d29f51d3c9f8dbd4781d9383fa4fbbb829;hpb=06586bbe5daca94845ab53f177883eb97e56bf50;p=lttng-tools.git

diff --git a/src/common/relayd/relayd.c b/src/common/relayd/relayd.c
index 7e6520d29..242abbedf 100644
--- a/src/common/relayd/relayd.c
+++ b/src/common/relayd/relayd.c
@@ -942,6 +942,150 @@ 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)
+{
+	int ret;
+	struct lttcomm_relayd_rotate_rename *msg = NULL;
+	struct lttcomm_relayd_generic_reply reply;
+	size_t old_path_length, new_path_length;
+	size_t msg_length;
+
+	/* Code flow error. Safety net. */
+	assert(rsock);
+
+	DBG("Relayd rename chunk %s to %s", old_path, new_path);
+
+	/* The two paths are sent with a '\0' delimiter between them. */
+	old_path_length = strlen(old_path) + 1;
+	new_path_length = strlen(new_path) + 1;
+
+	msg_length = sizeof(*msg) + old_path_length + new_path_length;
+	msg = zmalloc(msg_length);
+	if (!msg) {
+		PERROR("zmalloc rotate-rename command message");
+		ret = -1;
+		goto error;
+	}
+
+	assert(old_path_length <= UINT32_MAX);
+	msg->old_path_length = htobe32(old_path_length);
+
+	assert(new_path_length <= UINT32_MAX);
+	msg->new_path_length = htobe32(new_path_length);
+
+	strcpy(msg->paths, old_path);
+	strcpy(msg->paths + old_path_length, new_path);
+
+	/* Send command */
+	ret = send_command(rsock, RELAYD_ROTATE_RENAME, (const void *) msg,
+			msg_length, 0);
+	if (ret < 0) {
+		goto error;
+	}
+
+	/* Receive response */
+	ret = recv_reply(rsock, (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 != LTTNG_OK) {
+		ret = -1;
+		ERR("Relayd rotate rename replied error %d", reply.ret_code);
+	} else {
+		/* Success */
+		ret = 0;
+	}
+
+	DBG("Relayd rotate rename completed successfully");
+
+error:
+	free(msg);
+	return ret;
+}
+
 int relayd_mkdir(struct lttcomm_relayd_sock *rsock, const char *path)
 {
 	int ret;
@@ -997,5 +1141,4 @@ int relayd_mkdir(struct lttcomm_relayd_sock *rsock, const char *path)
 error:
 	free(msg);
 	return ret;
-
 }