action-executor: consider action firing policy on action execution

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

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index a0347a34c66ee53abd7b1851bdc362b3fffe2c78..07eefeeb050ab14877abcf2d55ac4b93e78d91b1 100644 (file)
--- a/src/common/actions/action.c
+++ b/src/common/actions/action.c
@@ -8,6 +8,7 @@
 #include <assert.h>
 #include <common/error.h>
 #include <lttng/action/action-internal.h>
+#include <lttng/action/firing-policy-internal.h>
 #include <lttng/action/group-internal.h>
 #include <lttng/action/notify-internal.h>
 #include <lttng/action/rotate-session-internal.h>
@@ -15,7 +16,8 @@
 #include <lttng/action/start-session-internal.h>
 #include <lttng/action/stop-session-internal.h>
 
-static const char *lttng_action_type_string(enum lttng_action_type action_type)
+LTTNG_HIDDEN
+const char *lttng_action_type_string(enum lttng_action_type action_type)
 {
        switch (action_type) {
        case LTTNG_ACTION_TYPE_UNKNOWN:
@@ -37,26 +39,19 @@ static const char *lttng_action_type_string(enum lttng_action_type action_type)
        }
 }
 
-enum lttng_action_type lttng_action_get_type(struct lttng_action *action)
+enum lttng_action_type lttng_action_get_type(const struct lttng_action *action)
 {
        return action ? action->type : LTTNG_ACTION_TYPE_UNKNOWN;
 }
 
 LTTNG_HIDDEN
-enum lttng_action_type lttng_action_get_type_const(
-               const struct lttng_action *action)
-{
-       return action->type;
-}
-
-LTTNG_HIDDEN
-void lttng_action_init(
-               struct lttng_action *action,
+void lttng_action_init(struct lttng_action *action,
                enum lttng_action_type type,
                action_validate_cb validate,
                action_serialize_cb serialize,
                action_equal_cb equal,
-               action_destroy_cb destroy)
+               action_destroy_cb destroy,
+               action_get_firing_policy_cb get_firing_policy)
 {
        urcu_ref_init(&action->ref);
        action->type = type;
@@ -64,6 +59,11 @@ void lttng_action_init(
        action->serialize = serialize;
        action->equal = equal;
        action->destroy = destroy;
+       action->get_firing_policy = get_firing_policy;
+
+       action->execution_request_counter = 0;
+       action->execution_counter = 0;
+       action->execution_failure_counter = 0;
 }
 
 static
@@ -146,15 +146,24 @@ ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view,
                struct lttng_action **action)
 {
        ssize_t consumed_len, specific_action_consumed_len;
-       const struct lttng_action_comm *action_comm;
        action_create_from_payload_cb create_from_payload_cb;
+       const struct lttng_action_comm *action_comm;
+       const struct lttng_payload_view action_comm_view =
+                       lttng_payload_view_from_view(
+                                       view, 0, sizeof(*action_comm));
 
        if (!view || !action) {
                consumed_len = -1;
                goto end;
        }
 
-       action_comm = (const struct lttng_action_comm *) view->buffer.data;
+       if (!lttng_payload_view_is_valid(&action_comm_view)) {
+               /* Payload not large enough to contain the header. */
+               consumed_len = -1;
+               goto end;
+       }
+
+       action_comm = (const struct lttng_action_comm *) action_comm_view.buffer.data;
 
        DBG("Create action from payload: action-type=%s",
                        lttng_action_type_string(action_comm->action_type));
@@ -240,3 +249,44 @@ bool lttng_action_is_equal(const struct lttng_action *a,
 end:
        return is_equal;
 }
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_request_count(struct lttng_action *action)
+{
+       action->execution_request_counter++;
+}
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_count(struct lttng_action *action)
+{
+       action->execution_counter++;
+}
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_failure_count(struct lttng_action *action)
+{
+       action->execution_failure_counter++;
+}
+
+LTTNG_HIDDEN
+bool lttng_action_should_execute(const struct lttng_action *action)
+{
+       const struct lttng_firing_policy *policy = NULL;
+       bool execute = false;
+
+       if (action->get_firing_policy == NULL) {
+               execute = true;
+               goto end;
+       }
+
+       policy = action->get_firing_policy(action);
+       if (policy == NULL) {
+               execute = true;
+               goto end;
+       }
+
+       execute = lttng_firing_policy_should_execute(
+                       policy, action->execution_request_counter);
+end:
+       return execute;
+}