X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fcmd.cpp;h=659e389b29df286c1d3c7dab2049ce7b5909a48d;hb=e68d8bdb16c60aca0acdf4c4887d138031199af3;hp=f4ce9a8829f70b77ece1e958b1ac42767aa41c8d;hpb=26e1c61ff394774de86a3ebad6a7caa02db69367;p=lttng-tools.git

diff --git a/src/bin/lttng-sessiond/cmd.cpp b/src/bin/lttng-sessiond/cmd.cpp
index f4ce9a882..659e389b2 100644
--- a/src/bin/lttng-sessiond/cmd.cpp
+++ b/src/bin/lttng-sessiond/cmd.cpp
@@ -387,7 +387,7 @@ static enum lttng_error_code list_lttng_ust_global_events(char *channel_name,
 	node = lttng_ht_iter_get_node_str(&iter);
 	if (node == NULL) {
 		ret_code = LTTNG_ERR_UST_CHAN_NOT_FOUND;
-		goto end;
+		goto error;
 	}
 
 	uchan = caa_container_of(&node->node, struct ltt_ust_channel, node.node);
@@ -420,14 +420,14 @@ static enum lttng_error_code list_lttng_ust_global_events(char *channel_name,
 		tmp_event = lttng_event_create();
 		if (!tmp_event) {
 			ret_code = LTTNG_ERR_NOMEM;
-			goto end;
+			goto error;
 		}
 
 		if (lttng_strncpy(tmp_event->name, uevent->attr.name,
 				LTTNG_SYMBOL_NAME_LEN)) {
 			ret_code = LTTNG_ERR_FATAL;
 			lttng_event_destroy(tmp_event);
-			goto end;
+			goto error;
 		}
 
 		tmp_event->name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
@@ -2559,31 +2559,70 @@ error:
 /*
  * Command LTTNG_LIST_TRACEPOINT_FIELDS processed by the client thread.
  */
-ssize_t cmd_list_tracepoint_fields(enum lttng_domain_type domain,
-		struct lttng_event_field **fields)
+enum lttng_error_code cmd_list_tracepoint_fields(enum lttng_domain_type domain,
+		struct lttng_payload *reply)
 {
+	enum lttng_error_code ret_code;
 	int ret;
-	ssize_t nb_fields = 0;
+	unsigned int i, nb_fields;
+	struct lttng_event_field *fields = NULL;
+	struct lttcomm_list_command_header reply_command_header = {};
+	size_t reply_command_header_offset;
+
+	assert(reply);
+
+	/* Reserve space for command reply header. */
+	reply_command_header_offset = reply->buffer.size;
+	ret = lttng_dynamic_buffer_set_size(&reply->buffer,
+			reply_command_header_offset +
+				sizeof(struct lttcomm_list_command_header));
+	if (ret) {
+		ret_code = LTTNG_ERR_NOMEM;
+		goto error;
+	}
 
 	switch (domain) {
 	case LTTNG_DOMAIN_UST:
-		nb_fields = ust_app_list_event_fields(fields);
-		if (nb_fields < 0) {
-			ret = LTTNG_ERR_UST_LIST_FAIL;
+		ret = ust_app_list_event_fields(&fields);
+		if (ret < 0) {
+			ret_code = LTTNG_ERR_UST_LIST_FAIL;
 			goto error;
 		}
+
 		break;
 	case LTTNG_DOMAIN_KERNEL:
 	default:	/* fall-through */
-		ret = LTTNG_ERR_UND;
+		ret_code = LTTNG_ERR_UND;
+		goto error;
+	}
+
+	nb_fields = ret;
+
+	for (i = 0; i < nb_fields; i++) {
+		ret = lttng_event_field_serialize(&fields[i], reply);
+		if (ret) {
+			ret_code = LTTNG_ERR_NOMEM;
+			goto error;
+		}
+	}
+
+	if (nb_fields > UINT32_MAX) {
+		ERR("Tracepoint field count would overflow the tracepoint field listing command's reply");
+		ret_code = LTTNG_ERR_OVERFLOW;
 		goto error;
 	}
 
-	return nb_fields;
+	/* Update command reply header. */
+	reply_command_header.count = (uint32_t) nb_fields;
+
+	memcpy(reply->buffer.data + reply_command_header_offset, &reply_command_header,
+			sizeof(reply_command_header));
+
+	ret_code = LTTNG_OK;
 
 error:
-	/* Return negative value to differentiate return code */
-	return -ret;
+	free(fields);
+	return ret_code;
 }
 
 enum lttng_error_code cmd_list_syscalls(
@@ -3791,7 +3830,7 @@ enum lttng_error_code cmd_list_events(enum lttng_domain_type domain,
 	enum lttng_error_code ret_code = LTTNG_OK;
 	struct lttcomm_list_command_header reply_command_header = {};
 	size_t reply_command_header_offset;
-	unsigned int nb_events;
+	unsigned int nb_events = 0;
 
 	assert(reply_payload);