X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;h=07eefeeb050ab14877abcf2d55ac4b93e78d91b1;hb=2d57482cbac710612a36b7beae7b59c277006185;hp=c6b57b9cff5c7a50a42325b68b5d8e3175f27794;hpb=c0a66c84b5b2484b75798aec7543b680b4d4ab6c;p=lttng-tools.git

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index c6b57b9cf..07eefeeb0 100644
--- a/src/common/actions/action.c
+++ b/src/common/actions/action.c
@@ -8,20 +8,28 @@
 #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>
+#include <lttng/action/snapshot-session-internal.h>
 #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:
 		return "UNKNOWN";
+	case LTTNG_ACTION_TYPE_GROUP:
+		return "GROUP";
 	case LTTNG_ACTION_TYPE_NOTIFY:
 		return "NOTIFY";
 	case LTTNG_ACTION_TYPE_ROTATE_SESSION:
 		return "ROTATE_SESSION";
+	case LTTNG_ACTION_TYPE_SNAPSHOT_SESSION:
+		return "SNAPSHOT_SESSION";
 	case LTTNG_ACTION_TYPE_START_SESSION:
 		return "START_SESSION";
 	case LTTNG_ACTION_TYPE_STOP_SESSION:
@@ -31,42 +39,62 @@ 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;
 	action->validate = validate;
 	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;
 }
 
-void lttng_action_destroy(struct lttng_action *action)
+static
+void action_destroy_ref(struct urcu_ref *ref)
+{
+	struct lttng_action *action =
+			container_of(ref, struct lttng_action, ref);
+
+	action->destroy(action);
+}
+
+LTTNG_HIDDEN
+void lttng_action_get(struct lttng_action *action)
+{
+	urcu_ref_get(&action->ref);
+}
+
+LTTNG_HIDDEN
+void lttng_action_put(struct lttng_action *action)
 {
 	if (!action) {
 		return;
 	}
 
 	assert(action->destroy);
-	action->destroy(action);
+	urcu_ref_put(&action->ref, action_destroy_ref);
+}
+
+void lttng_action_destroy(struct lttng_action *action)
+{
+	lttng_action_put(action);
 }
 
 LTTNG_HIDDEN
@@ -118,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));
@@ -139,6 +176,10 @@ ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view,
 		create_from_payload_cb =
 				lttng_action_rotate_session_create_from_payload;
 		break;
+	case LTTNG_ACTION_TYPE_SNAPSHOT_SESSION:
+		create_from_payload_cb =
+				lttng_action_snapshot_session_create_from_payload;
+		break;
 	case LTTNG_ACTION_TYPE_START_SESSION:
 		create_from_payload_cb =
 				lttng_action_start_session_create_from_payload;
@@ -147,6 +188,9 @@ ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view,
 		create_from_payload_cb =
 				lttng_action_stop_session_create_from_payload;
 		break;
+	case LTTNG_ACTION_TYPE_GROUP:
+		create_from_payload_cb = lttng_action_group_create_from_payload;
+		break;
 	default:
 		ERR("Failed to create action from payload, unhandled action type: action-type=%u (%s)",
 				action_comm->action_type,
@@ -205,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;
+}