X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Frelayd%2Frelayd.c;h=b065f364eb67f3e06410b9c3043082abdd8f85ad;hp=90c6649acc3f7b26cac59b2c89eb706eb74b0d72;hb=e5add6d004793894ef4c7e047bc0f8885763b205;hpb=3aa937ab446c323bec53fc3581c424a3cc5c020e

diff --git a/src/common/relayd/relayd.c b/src/common/relayd/relayd.c
index 90c6649ac..b065f364e 100644
--- a/src/common/relayd/relayd.c
+++ b/src/common/relayd/relayd.c
@@ -1195,3 +1195,91 @@ error:
 	free(msg);
 	return ret;
 }
+
+int relayd_create_trace_chunk(struct lttcomm_relayd_sock *sock,
+		struct lttng_trace_chunk *chunk)
+{
+	int ret = 0;
+	enum lttng_trace_chunk_status status;
+	struct lttcomm_relayd_create_trace_chunk msg = {};
+	struct lttcomm_relayd_generic_reply reply = {};
+	struct lttng_dynamic_buffer payload;
+	uint64_t chunk_id;
+	time_t creation_timestamp;
+	const char *chunk_name;
+	size_t chunk_name_length;
+	bool overriden_name;
+
+	lttng_dynamic_buffer_init(&payload);
+
+	status = lttng_trace_chunk_get_id(chunk, &chunk_id);
+	if (status != LTTNG_TRACE_CHUNK_STATUS_OK) {
+		ret = -1;
+		goto end;
+	}
+
+	status = lttng_trace_chunk_get_creation_timestamp(
+			chunk, &creation_timestamp);
+	if (status != LTTNG_TRACE_CHUNK_STATUS_OK) {
+		ret = -1;
+		goto end;
+	}
+
+	status = lttng_trace_chunk_get_name(
+			chunk, &chunk_name, &overriden_name);
+	if (status != LTTNG_TRACE_CHUNK_STATUS_OK &&
+			status != LTTNG_TRACE_CHUNK_STATUS_NONE) {
+		ret = -1;
+		goto end;
+	}
+
+	chunk_name_length = overriden_name ? (strlen(chunk_name) + 1) : 0;
+	msg = (typeof(msg)){
+		.chunk_id = htobe64(chunk_id),
+		.creation_timestamp = htobe64((uint64_t) creation_timestamp),
+		.override_name_length = htobe32((uint32_t) chunk_name_length),
+	};
+
+	ret = lttng_dynamic_buffer_append(&payload, &msg, sizeof(msg));
+	if (ret) {
+		goto end;
+	}
+	if (chunk_name_length) {
+		ret = lttng_dynamic_buffer_append(
+				&payload, chunk_name, chunk_name_length);
+		if (ret) {
+			goto end;
+		}
+	}
+
+	ret = send_command(sock,
+			RELAYD_CREATE_TRACE_CHUNK,
+			payload.data,
+			payload.size,
+			0);
+	if (ret < 0) {
+		ERR("Failed to send trace chunk creation command to relay daemon");
+		goto end;
+	}
+
+	ret = recv_reply(sock, &reply, sizeof(reply));
+	if (ret < 0) {
+		ERR("Failed to receive relay daemon trace chunk creation command reply");
+		goto end;
+	}
+
+	reply.ret_code = be32toh(reply.ret_code);
+	if (reply.ret_code != LTTNG_OK) {
+		ret = -1;
+		ERR("Relayd trace chunk create replied error %d",
+				reply.ret_code);
+	} else {
+		ret = 0;
+		DBG("Relayd successfully created trace chunk: chunk_id = %" PRIu64,
+				chunk_id);
+	}
+
+end:
+	lttng_dynamic_buffer_reset(&payload);
+	return ret;
+}