X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;fp=src%2Fcommon%2Factions%2Faction.c;h=4dabfff303e547bc973c389e5d7baae38954a792;hp=498e41f79405fb2fe71488f91c29b4cdec1ef907;hb=6a751b953a43c566b74818ec6325db0978e16c66;hpb=1aa9c49c60b7d7370ceafc3ac076de423de22b7f

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index 498e41f79..4dabfff30 100644
--- a/src/common/actions/action.c
+++ b/src/common/actions/action.c
@@ -7,6 +7,7 @@
 
 #include <assert.h>
 #include <common/error.h>
+#include <common/mi-lttng.h>
 #include <lttng/action/action-internal.h>
 #include <lttng/action/list-internal.h>
 #include <lttng/action/notify-internal.h>
@@ -53,7 +54,8 @@ void lttng_action_init(struct lttng_action *action,
 		action_equal_cb equal,
 		action_destroy_cb destroy,
 		action_get_rate_policy_cb get_rate_policy,
-		action_add_error_query_results_cb add_error_query_results)
+		action_add_error_query_results_cb add_error_query_results,
+		action_mi_serialize_cb mi)
 {
 	urcu_ref_init(&action->ref);
 	action->type = type;
@@ -63,6 +65,7 @@ void lttng_action_init(struct lttng_action *action,
 	action->destroy = destroy;
 	action->get_rate_policy = get_rate_policy;
 	action->add_error_query_results = add_error_query_results;
+	action->mi_serialize = mi;
 
 	action->execution_request_counter = 0;
 	action->execution_counter = 0;
@@ -334,3 +337,96 @@ end:
 	lttng_error_query_result_destroy(error_counter);
 	return action_status;
 }
+
+LTTNG_HIDDEN
+enum lttng_error_code lttng_action_mi_serialize(const struct lttng_trigger *trigger,
+		const struct lttng_action *action,
+		struct mi_writer *writer,
+		const struct mi_lttng_error_query_callbacks
+				*error_query_callbacks,
+		struct lttng_dynamic_array *action_path_indexes)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	struct lttng_action_path *action_path = NULL;
+	struct lttng_error_query_results *error_query_results = NULL;
+
+	assert(action);
+	assert(writer);
+
+	/* Open action. */
+	ret = mi_lttng_writer_open_element(writer, mi_lttng_element_action);
+	if (ret) {
+		goto mi_error;
+	}
+
+	if (action->type == LTTNG_ACTION_TYPE_LIST) {
+		/*
+		 * Recursion is safe since action lists can't be nested for
+		 * the moment.
+		 */
+		ret_code = lttng_action_list_mi_serialize(trigger, action, writer,
+				error_query_callbacks, action_path_indexes);
+		if (ret_code != LTTNG_OK) {
+			goto end;
+		}
+
+		/* Nothing else to do. */
+		goto close_action_element;
+	}
+
+	assert(action->mi_serialize);
+	ret_code = action->mi_serialize(action, writer);
+	if (ret_code != LTTNG_OK) {
+		goto end;
+	}
+
+	/* Error query for the action. */
+	if (error_query_callbacks && error_query_callbacks->action_cb) {
+		const uint64_t *action_path_indexes_raw_pointer = NULL;
+		const size_t action_path_indexes_size =
+				lttng_dynamic_array_get_count(
+						action_path_indexes);
+
+		if (action_path_indexes_size != 0) {
+			action_path_indexes_raw_pointer =
+					(const uint64_t *) action_path_indexes
+							->buffer.data;
+		}
+
+		action_path = lttng_action_path_create(
+				action_path_indexes_raw_pointer,
+				action_path_indexes_size);
+		assert(action_path);
+
+		ret_code = error_query_callbacks->action_cb(
+				trigger, action_path, &error_query_results);
+		if (ret_code != LTTNG_OK) {
+			goto end;
+		}
+
+		/* Serialize the error query results. */
+		ret_code = lttng_error_query_results_mi_serialize(
+				error_query_results, writer);
+		if (ret_code != LTTNG_OK) {
+			goto end;
+		}
+	}
+
+close_action_element:
+	/* Close action. */
+	ret = mi_lttng_writer_close_element(writer);
+	if (ret) {
+		goto mi_error;
+	}
+
+	ret_code = LTTNG_OK;
+	goto end;
+
+mi_error:
+	ret_code = LTTNG_ERR_MI_IO_FAIL;
+end:
+	lttng_action_path_destroy(action_path);
+	lttng_error_query_results_destroy(error_query_results);
+	return ret_code;
+}