X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Faction.c;h=1c1049ebee3adead84689bcb396fbe0dd27ea65d;hp=98e1849dd41950018b2276e98408afc763895591;hb=702f26c8641ee4554c629d13b9367b1e93f96e31;hpb=bfb2ec6a46481414a99e55556d7b0586e57b074e diff --git a/src/common/actions/action.c b/src/common/actions/action.c index 98e1849dd..1c1049ebe 100644 --- a/src/common/actions/action.c +++ b/src/common/actions/action.c @@ -8,20 +8,29 @@ #include #include #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: 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 +40,64 @@ 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_rate_policy_cb get_rate_policy, + action_add_error_query_results_cb add_error_query_results) { + urcu_ref_init(&action->ref); action->type = type; action->validate = validate; action->serialize = serialize; action->equal = equal; action->destroy = destroy; + action->get_rate_policy = get_rate_policy; + action->add_error_query_results = add_error_query_results; + + 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 @@ -92,20 +123,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; } @@ -114,42 +145,57 @@ 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; + } - DBG("Create action from buffer: action-type=%s", + 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)); 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_buffer_cb = - lttng_action_rotate_session_create_from_buffer; + 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_buffer_cb = - lttng_action_start_session_create_from_buffer; + create_from_payload_cb = + lttng_action_start_session_create_from_payload; break; case LTTNG_ACTION_TYPE_STOP_SESSION: - create_from_buffer_cb = - lttng_action_stop_session_create_from_buffer; + create_from_payload_cb = + lttng_action_stop_session_create_from_payload; + break; + case LTTNG_ACTION_TYPE_GROUP: + create_from_payload_cb = lttng_action_list_create_from_payload; break; default: - ERR("Failed to create action from buffer, unhandled action type: action-type=%u (%s)", + 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)); @@ -157,13 +203,16 @@ ssize_t lttng_action_create_from_buffer(const struct lttng_buffer_view *view, 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; @@ -203,3 +252,85 @@ 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) +{ + uatomic_inc(&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; +} + +LTTNG_HIDDEN +enum lttng_action_status lttng_action_add_error_query_results( + const struct lttng_action *action, + struct lttng_error_query_results *results) +{ + return action->add_error_query_results(action, results); +} + +LTTNG_HIDDEN +enum lttng_action_status lttng_action_generic_add_error_query_results( + const struct lttng_action *action, + struct lttng_error_query_results *results) +{ + enum lttng_action_status action_status; + struct lttng_error_query_result *error_counter = NULL; + const uint64_t execution_failure_counter = + uatomic_read(&action->execution_failure_counter); + + error_counter = lttng_error_query_result_counter_create( + "total execution failures", + "Aggregated count of errors encountered when executing the action", + execution_failure_counter); + if (!error_counter) { + action_status = LTTNG_ACTION_STATUS_ERROR; + goto end; + } + + if (lttng_error_query_results_add_result( + results, error_counter)) { + action_status = LTTNG_ACTION_STATUS_ERROR; + goto end; + } + + /* Ownership transferred to the results. */ + error_counter = NULL; + action_status = LTTNG_ACTION_STATUS_OK; +end: + lttng_error_query_result_destroy(error_counter); + return action_status; +}