X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;h=86628542bc50186c32b20a0bc026530a235672d6;hp=dc72d37b4a88f75e6bf0575e16bf0c8c911435d3;hb=17182cfd13b6d35cf8c80d4f9ccf8d2bdd1a05f2;hpb=03bb2358875cfca96394e78357c96baaa1e91efa diff --git a/src/common/actions/action.c b/src/common/actions/action.c index dc72d37b4..86628542b 100644 --- a/src/common/actions/action.c +++ b/src/common/actions/action.c @@ -5,31 +5,90 @@ * */ +#include +#include #include +#include #include -#include -#include +#include +#include +#include +#include + +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: + return "STOP_SESSION"; + default: + return "???"; + } +} -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) +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) { - return action->type; + urcu_ref_init(&action->ref); + action->type = type; + action->validate = validate; + action->serialize = serialize; + action->equal = equal; + action->destroy = destroy; } -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 @@ -55,20 +114,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; } @@ -77,35 +136,101 @@ end: } LTTNG_HIDDEN -ssize_t lttng_action_create_from_buffer(const struct lttng_buffer_view *view, - struct lttng_action **_action) +ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view, + struct lttng_action **action) { - ssize_t ret, action_size = sizeof(struct lttng_action_comm); - 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; - if (!view || !_action) { - ret = -1; + if (!view || !action) { + consumed_len = -1; goto end; } - action_comm = (const struct lttng_action_comm *) view->data; - DBG("Deserializing action from buffer"); + action_comm = (const struct lttng_action_comm *) view->buffer.data; + + DBG("Create action from payload: action-type=%s", + lttng_action_type_string(action_comm->action_type)); + switch (action_comm->action_type) { case LTTNG_ACTION_TYPE_NOTIFY: - action = lttng_action_notify_create(); + 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: - ret = -1; + 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; } - if (!action) { - ret = -1; + { + /* 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_payload_cb( + &specific_action_view, action); + } + if (specific_action_consumed_len < 0) { + ERR("Failed to create specific action from buffer."); + consumed_len = -1; goto end; } - ret = action_size; - *_action = action; + + assert(*action); + + consumed_len = sizeof(struct lttng_action_comm) + + specific_action_consumed_len; + end: - return ret; + 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; }