relayd: add remote trace chunk creation command

[lttng-tools.git] / src / common / relayd / relayd.c

diff --git a/src/common/relayd/relayd.c b/src/common/relayd/relayd.c
index 90c6649acc3f7b26cac59b2c89eb706eb74b0d72..b065f364eb67f3e06410b9c3043082abdd8f85ad 100644 (file)
--- 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;
+}