MI: implement all objects related to trigger machine interface

[lttng-tools.git] / src / common / actions / action.c

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index 498e41f79405fb2fe71488f91c29b4cdec1ef907..4dabfff303e547bc973c389e5d7baae38954a792 100644 (file)
--- 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;
+}