*
*/
+#include <assert.h>
+#include <common/error.h>
+#include <common/macros.h>
#include <lttng/action/action-internal.h>
#include <lttng/action/notify-internal.h>
-#include <common/macros.h>
-#include <assert.h>
+#include <lttng/action/rate-policy-internal.h>
+
+#define IS_NOTIFY_ACTION(action) \
+ (lttng_action_get_type(action) == LTTNG_ACTION_TYPE_NOTIFY)
+
+static struct lttng_action_notify *action_notify_from_action(
+ struct lttng_action *action)
+{
+ assert(action);
+
+ return container_of(action, struct lttng_action_notify, parent);
+}
+
+static const struct lttng_action_notify *action_notify_from_action_const(
+ const struct lttng_action *action)
+{
+ assert(action);
+
+ return container_of(action, struct lttng_action_notify, parent);
+}
static
void lttng_action_notify_destroy(struct lttng_action *action)
{
- free(action);
+ struct lttng_action_notify *notify_action;
+ notify_action = action_notify_from_action(action);
+ lttng_rate_policy_destroy(notify_action->policy);
+ free(notify_action);
}
static
int lttng_action_notify_serialize(struct lttng_action *action,
- struct lttng_dynamic_buffer *buf)
+ struct lttng_payload *payload)
{
- return 0;
+ int ret;
+ struct lttng_action_notify *notify_action;
+
+ if (!action || !IS_NOTIFY_ACTION(action) || !payload) {
+ ret = -1;
+ goto end;
+ }
+
+ DBG("Serializing notify action");
+
+ notify_action = action_notify_from_action(action);
+ DBG("Serializing notify action rate policy");
+ ret = lttng_rate_policy_serialize(notify_action->policy, payload);
+
+end:
+ return ret;
+}
+
+static
+bool lttng_action_notify_is_equal(const struct lttng_action *a,
+ const struct lttng_action *b)
+{
+ const struct lttng_action_notify *_a, *_b;
+
+ _a = action_notify_from_action_const(a);
+ _b = action_notify_from_action_const(b);
+ return lttng_rate_policy_is_equal(_a->policy, _b->policy);
+}
+
+static const struct lttng_rate_policy *
+lttng_action_notify_internal_get_rate_policy(const struct lttng_action *action)
+{
+ const struct lttng_action_notify *_action;
+ _action = action_notify_from_action_const(action);
+
+ return _action->policy;
}
struct lttng_action *lttng_action_notify_create(void)
{
- struct lttng_action_notify *notify;
+ struct lttng_rate_policy *policy = NULL;
+ struct lttng_action_notify *notify = NULL;
+ struct lttng_action *action = NULL;
notify = zmalloc(sizeof(struct lttng_action_notify));
if (!notify) {
goto end;
}
- notify->parent.type = LTTNG_ACTION_TYPE_NOTIFY;
- notify->parent.serialize = lttng_action_notify_serialize;
- notify->parent.destroy = lttng_action_notify_destroy;
+ /* Default policy. */
+ policy = lttng_rate_policy_every_n_create(1);
+ if (!policy) {
+ goto end;
+ }
+
+ lttng_action_init(¬ify->parent, LTTNG_ACTION_TYPE_NOTIFY, NULL,
+ lttng_action_notify_serialize,
+ lttng_action_notify_is_equal,
+ lttng_action_notify_destroy,
+ lttng_action_notify_internal_get_rate_policy);
+
+ notify->policy = policy;
+ policy = NULL;
+
+ action = ¬ify->parent;
+ notify = NULL;
+
+end:
+ free(notify);
+ lttng_rate_policy_destroy(policy);
+ return action;
+}
+
+ssize_t lttng_action_notify_create_from_payload(
+ struct lttng_payload_view *view,
+ struct lttng_action **action)
+{
+ enum lttng_action_status status;
+ ssize_t consumed_length;
+ struct lttng_rate_policy *rate_policy = NULL;
+ struct lttng_action *_action = NULL;
+
+ consumed_length = lttng_rate_policy_create_from_payload(
+ view, &rate_policy);
+ if (!rate_policy) {
+ consumed_length = -1;
+ goto end;
+ }
+
+ _action = lttng_action_notify_create();
+ if (!_action) {
+ consumed_length = -1;
+ goto end;
+ }
+
+ status = lttng_action_notify_set_rate_policy(_action, rate_policy);
+ if (status != LTTNG_ACTION_STATUS_OK) {
+ consumed_length = -1;
+ goto end;
+ }
+
+ *action = _action;
+ _action = NULL;
+
+end:
+ lttng_rate_policy_destroy(rate_policy);
+ lttng_action_destroy(_action);
+ return consumed_length;
+}
+
+enum lttng_action_status lttng_action_notify_set_rate_policy(
+ struct lttng_action *action,
+ const struct lttng_rate_policy *policy)
+{
+ enum lttng_action_status status;
+ struct lttng_action_notify *notify_action;
+ struct lttng_rate_policy *copy = NULL;
+
+ if (!action || !policy || !IS_NOTIFY_ACTION(action)) {
+ status = LTTNG_ACTION_STATUS_INVALID;
+ goto end;
+ }
+
+ copy = lttng_rate_policy_copy(policy);
+ if (!copy) {
+ status = LTTNG_ACTION_STATUS_ERROR;
+ goto end;
+ }
+
+ notify_action = action_notify_from_action(action);
+
+ /* Free the previous rate policy .*/
+ lttng_rate_policy_destroy(notify_action->policy);
+
+ /* Assign the policy. */
+ notify_action->policy = copy;
+ status = LTTNG_ACTION_STATUS_OK;
+ copy = NULL;
+
+end:
+ lttng_rate_policy_destroy(copy);
+ return status;
+}
+
+enum lttng_action_status lttng_action_notify_get_rate_policy(
+ const struct lttng_action *action,
+ const struct lttng_rate_policy **policy)
+{
+ enum lttng_action_status status;
+ const struct lttng_action_notify *notify_action;
+
+ if (!action || !policy || !IS_NOTIFY_ACTION(action)) {
+ status = LTTNG_ACTION_STATUS_INVALID;
+ goto end;
+ }
+
+ notify_action = action_notify_from_action_const(action);
+
+ *policy = notify_action->policy;
+ status = LTTNG_ACTION_STATUS_OK;
end:
- return ¬ify->parent;
+ return status;
}