X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;h=07eefeeb050ab14877abcf2d55ac4b93e78d91b1;hp=6894036cb779d09ced9c5acee4dacf9a2e99d876;hb=2d57482cbac710612a36b7beae7b59c277006185;hpb=6acb3f46333c25d5a2f3a3ff8158956a4689031e diff --git a/src/common/actions/action.c b/src/common/actions/action.c index 6894036cb..07eefeeb0 100644 --- a/src/common/actions/action.c +++ b/src/common/actions/action.c @@ -5,45 +5,96 @@ * */ +#include +#include #include +#include +#include #include -#include -#include +#include +#include +#include +#include -enum lttng_action_type lttng_action_get_type(struct lttng_action *action) +LTTNG_HIDDEN +const char *lttng_action_type_string(enum lttng_action_type action_type) { - return action ? action->type : LTTNG_ACTION_TYPE_UNKNOWN; + 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: + return "STOP_SESSION"; + default: + return "???"; + } } -LTTNG_HIDDEN -enum lttng_action_type lttng_action_get_type_const( - const struct lttng_action *action) +enum lttng_action_type lttng_action_get_type(const struct lttng_action *action) { - return action->type; + return action ? action->type : LTTNG_ACTION_TYPE_UNKNOWN; } 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_destroy_cb destroy) + action_equal_cb equal, + 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 @@ -69,20 +120,20 @@ end: LTTNG_HIDDEN int lttng_action_serialize(struct lttng_action *action, - struct lttng_dynamic_buffer *buf) + struct lttng_payload *payload) { int ret; struct lttng_action_comm action_comm = { .action_type = (int8_t) action->type, }; - ret = lttng_dynamic_buffer_append(buf, &action_comm, + ret = lttng_dynamic_buffer_append(&payload->buffer, &action_comm, sizeof(action_comm)); if (ret) { goto end; } - ret = action->serialize(action, buf); + ret = action->serialize(action, payload); if (ret) { goto end; } @@ -91,38 +142,74 @@ end: } LTTNG_HIDDEN -ssize_t lttng_action_create_from_buffer(const struct lttng_buffer_view *view, +ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view, struct lttng_action **action) { ssize_t consumed_len, specific_action_consumed_len; + action_create_from_payload_cb create_from_payload_cb; const struct lttng_action_comm *action_comm; - action_create_from_buffer_cb create_from_buffer_cb; - struct lttng_buffer_view specific_action_view; + 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->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)); - DBG("Deserializing action from buffer"); switch (action_comm->action_type) { case LTTNG_ACTION_TYPE_NOTIFY: - create_from_buffer_cb = lttng_action_notify_create_from_buffer; + create_from_payload_cb = lttng_action_notify_create_from_payload; + break; + case LTTNG_ACTION_TYPE_ROTATE_SESSION: + 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; + break; + case LTTNG_ACTION_TYPE_STOP_SESSION: + 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, + lttng_action_type_string( + action_comm->action_type)); consumed_len = -1; goto end; } - /* Create buffer view for the action-type-specific data. */ - specific_action_view = lttng_buffer_view_from_view(view, - sizeof(struct lttng_action_comm), - view->size - sizeof(struct lttng_action_comm)); + { + /* Create buffer view for the action-type-specific data. */ + struct lttng_payload_view specific_action_view = + lttng_payload_view_from_view(view, + sizeof(struct lttng_action_comm), + -1); - specific_action_consumed_len = - create_from_buffer_cb(&specific_action_view, action); + specific_action_consumed_len = create_from_payload_cb( + &specific_action_view, action); + } if (specific_action_consumed_len < 0) { ERR("Failed to create specific action from buffer."); consumed_len = -1; @@ -137,3 +224,69 @@ ssize_t lttng_action_create_from_buffer(const struct lttng_buffer_view *view, end: return consumed_len; } + +LTTNG_HIDDEN +bool lttng_action_is_equal(const struct lttng_action *a, + const struct lttng_action *b) +{ + bool is_equal = false; + + if (!a || !b) { + goto end; + } + + if (a->type != b->type) { + goto end; + } + + if (a == b) { + is_equal = true; + goto end; + } + + assert(a->equal); + is_equal = a->equal(a, b); +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; +}