SoW-2019-0007-2: Dynamic Snapshot: Triggers send partial event payload with notifications

[lttng-tools.git] / src / common / actions / action.c

diff --git a/src/common/actions/action.c b/src/common/actions/action.c
index 4cad01b743189ff5318fa9e388ed2bd1d82f933a..fd0263c23d32fc731bbf866d8ec1cc72471544de 100644 (file)
--- a/src/common/actions/action.c
+++ b/src/common/actions/action.c
@@ -5,24 +5,40 @@
  *
  */
 
+#include <assert.h>
+#include <common/error.h>
 #include <lttng/action/action-internal.h>
+#include <lttng/action/group-internal.h>
 #include <lttng/action/notify-internal.h>
-#include <common/error.h>
-#include <assert.h>
+#include <lttng/action/rotate-session-internal.h>
+#include <lttng/action/snapshot-session-internal.h>
+#include <lttng/action/start-session-internal.h>
+#include <lttng/action/stop-session-internal.h>
 
-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:
+               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;
 }
@@ -40,22 +56,46 @@ void lttng_action_init(
                enum lttng_action_type type,
                action_validate_cb validate,
                action_serialize_cb serialize,
+               action_equal_cb equal,
                action_destroy_cb destroy)
 {
+       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
@@ -125,6 +165,25 @@ ssize_t lttng_action_create_from_buffer(const struct lttng_buffer_view *view,
        case LTTNG_ACTION_TYPE_NOTIFY:
                create_from_buffer_cb = lttng_action_notify_create_from_buffer;
                break;
+       case LTTNG_ACTION_TYPE_ROTATE_SESSION:
+               create_from_buffer_cb =
+                               lttng_action_rotate_session_create_from_buffer;
+               break;
+       case LTTNG_ACTION_TYPE_SNAPSHOT_SESSION:
+               create_from_buffer_cb =
+                               lttng_action_snapshot_session_create_from_buffer;
+               break;
+       case LTTNG_ACTION_TYPE_START_SESSION:
+               create_from_buffer_cb =
+                               lttng_action_start_session_create_from_buffer;
+               break;
+       case LTTNG_ACTION_TYPE_STOP_SESSION:
+               create_from_buffer_cb =
+                               lttng_action_stop_session_create_from_buffer;
+               break;
+       case LTTNG_ACTION_TYPE_GROUP:
+               create_from_buffer_cb = lttng_action_group_create_from_buffer;
+               break;
        default:
                ERR("Failed to create action from buffer, unhandled action type: action-type=%u (%s)",
                                action_comm->action_type,
@@ -155,3 +214,28 @@ 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;
+       }
+
+       is_equal = a->equal ? a->equal(a, b) : true;
+end:
+       return is_equal;
+}
+