X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;h=95a0c0f4d20a19eb743b60d392c65934a82d656f;hp=1a541ce469efc22381f198261a484b9df7ee4f26;hb=3e6e0df2f8f9f23d252c2508b6d741916dfcc4b3;hpb=c852ce4e1d4ab0028ecee333a3ffc83de32f44bf diff --git a/src/common/actions/action.c b/src/common/actions/action.c index 1a541ce46..95a0c0f4d 100644 --- a/src/common/actions/action.c +++ b/src/common/actions/action.c @@ -8,17 +8,21 @@ #include #include #include +#include #include #include #include #include #include -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: @@ -34,18 +38,11 @@ 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, @@ -143,15 +140,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)); @@ -176,6 +182,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,