X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fcommon%2Factions%2Faction.c;h=64864a6273ae1fabac87cd8e6fd813831335f9d0;hb=7f4d5b07cf7be895b38b69073389a4fcc318ec29;hp=86628542bc50186c32b20a0bc026530a235672d6;hpb=17182cfd13b6d35cf8c80d4f9ccf8d2bdd1a05f2;p=lttng-tools.git

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index 86628542b..64864a627 100644
--- a/src/common/actions/action.c
+++ b/src/common/actions/action.c
@@ -10,6 +10,7 @@
 #include <lttng/action/action-internal.h>
 #include <lttng/action/group-internal.h>
 #include <lttng/action/notify-internal.h>
+#include <lttng/action/rate-policy-internal.h>
 #include <lttng/action/rotate-session-internal.h>
 #include <lttng/action/snapshot-session-internal.h>
 #include <lttng/action/start-session-internal.h>
@@ -44,13 +45,13 @@ enum lttng_action_type lttng_action_get_type(const struct lttng_action *action)
 }
 
 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_rate_policy_cb get_rate_policy)
 {
 	urcu_ref_init(&action->ref);
 	action->type = type;
@@ -58,6 +59,11 @@ void lttng_action_init(
 	action->serialize = serialize;
 	action->equal = equal;
 	action->destroy = destroy;
+	action->get_rate_policy = get_rate_policy;
+
+	action->execution_request_counter = 0;
+	action->execution_counter = 0;
+	action->execution_failure_counter = 0;
 }
 
 static
@@ -140,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));
@@ -234,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_rate_policy *policy = NULL;
+	bool execute = false;
+
+	if (action->get_rate_policy == NULL) {
+		execute = true;
+		goto end;
+	}
+
+	policy = action->get_rate_policy(action);
+	if (policy == NULL) {
+		execute = true;
+		goto end;
+	}
+
+	execute = lttng_rate_policy_should_execute(
+			policy, action->execution_request_counter);
+end:
+	return execute;
+}