X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=src%2Fcommon%2Fust-consumer%2Fust-consumer.c;h=0d61b866f46d426c120694d9138f57a83be26a5d;hb=3b16476a9b2015e59bb869cdeb07975b8d1ec6c5;hp=ff9d31d5287a92b29bf6ac16bff0e6a17876d837;hpb=0d88e04674ead21c741c6f4ed7fadf666c5e7bce;p=lttng-tools.git

diff --git a/src/common/ust-consumer/ust-consumer.c b/src/common/ust-consumer/ust-consumer.c
index ff9d31d52..0d61b866f 100644
--- a/src/common/ust-consumer/ust-consumer.c
+++ b/src/common/ust-consumer/ust-consumer.c
@@ -2008,8 +2008,8 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 	case LTTNG_CONSUMER_CREATE_TRACE_CHUNK:
 	{
 		const struct lttng_credentials credentials = {
-			.uid = msg.u.create_trace_chunk.credentials.uid,
-			.gid = msg.u.create_trace_chunk.credentials.gid,
+			.uid = msg.u.create_trace_chunk.credentials.value.uid,
+			.gid = msg.u.create_trace_chunk.credentials.value.gid,
 		};
 		const bool is_local_trace =
 				!msg.u.create_trace_chunk.relayd_id.is_set;
@@ -2062,9 +2062,12 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 				!is_local_trace ? &relayd_id : NULL,
 				msg.u.create_trace_chunk.session_id,
 				msg.u.create_trace_chunk.chunk_id,
-				(time_t) msg.u.create_trace_chunk.creation_timestamp,
+				(time_t) msg.u.create_trace_chunk
+						.creation_timestamp,
 				chunk_override_name,
-				&credentials,
+				msg.u.create_trace_chunk.credentials.is_set ?
+						&credentials :
+						NULL,
 				chunk_directory_handle.is_set ?
 						&chunk_directory_handle.value :
 						NULL);
@@ -2077,15 +2080,21 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 	}
 	case LTTNG_CONSUMER_CLOSE_TRACE_CHUNK:
 	{
+		enum lttng_trace_chunk_command_type close_command =
+				msg.u.close_trace_chunk.close_command.value;
 		const uint64_t relayd_id =
 				msg.u.close_trace_chunk.relayd_id.value;
 
 		ret_code = lttng_consumer_close_trace_chunk(
 				msg.u.close_trace_chunk.relayd_id.is_set ?
-						&relayd_id : NULL,
+						&relayd_id :
+						NULL,
 				msg.u.close_trace_chunk.session_id,
 				msg.u.close_trace_chunk.chunk_id,
-				(time_t) msg.u.close_trace_chunk.close_timestamp);
+				(time_t) msg.u.close_trace_chunk.close_timestamp,
+				msg.u.close_trace_chunk.close_command.is_set ?
+						&close_command :
+						NULL);
 		goto end_msg_sessiond;
 	}
 	case LTTNG_CONSUMER_TRACE_CHUNK_EXISTS: