Fix: liblttng-ust-fork Makefile flags mismatch
[lttng-ust.git] / liblttng-ust / lttng-ust-abi.c
index e03024f49d2bade05e0b5472ddfb50f373e20ae4..ddadfc1d7f33c9a4fb9310976af296f33bdd66db 100644 (file)
@@ -1,10 +1,25 @@
 /*
  * lttng-ust-abi.c
  *
- * Copyright 2010-2011 (c) - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
- *
  * LTTng UST ABI
  *
+ * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; only
+ * version 2.1 of the License.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ *
+ *
  * Mimic system calls for:
  * - session creation, returns an object descriptor or failure.
  *   - channel creation, returns an object descriptor or failure.
  *     - Takes an instrumentation source as parameter
  *       - e.g. tracepoints, dynamic_probes...
  *     - Takes instrumentation source specific arguments.
- *
- * Dual LGPL v2.1/GPL v2 license.
  */
 
 #include <lttng/ust-abi.h>
+#include <lttng/ust-error.h>
 #include <urcu/compiler.h>
 #include <urcu/list.h>
 #include <lttng/ust-events.h>
-#include <lttng/usterr-signal-safe.h>
-#include "lttng/core.h"
-#include "ltt-tracer.h"
+#include <lttng/ust-version.h>
+#include <lttng/tracepoint.h>
 #include "tracepoint-internal.h"
+#include <usterr-signal-safe.h>
+#include <helper.h>
+#include "lttng-tracer.h"
 
-struct ltt_tracepoint_list {
-       struct tracepoint_iter iter;
-       int got_first;
-};
+#define OBJ_NAME_LEN   16
 
 static int lttng_ust_abi_close_in_progress;
 
 static
-int lttng_abi_tracepoint_list(void);
+int lttng_abi_tracepoint_list(void *owner);
+static
+int lttng_abi_tracepoint_field_list(void *owner);
 
 /*
  * Object descriptor table. Should be protected from concurrent access
@@ -54,6 +69,9 @@ struct lttng_ust_obj {
                        void *private_data;
                        const struct lttng_ust_objd_ops *ops;
                        int f_count;
+                       int owner_ref;  /* has ref from owner */
+                       void *owner;
+                       char name[OBJ_NAME_LEN];
                } s;
                int freelist_next;      /* offset freelist. end is -1. */
        } u;
@@ -70,7 +88,8 @@ static struct lttng_ust_objd_table objd_table = {
 };
 
 static
-int objd_alloc(void *private_data, const struct lttng_ust_objd_ops *ops)
+int objd_alloc(void *private_data, const struct lttng_ust_objd_ops *ops,
+               void *owner, const char *name)
 {
        struct lttng_ust_obj *obj;
 
@@ -106,6 +125,10 @@ end:
        obj->u.s.ops = ops;
        obj->u.s.f_count = 2;   /* count == 1 : object is allocated */
                                /* count == 2 : allocated + hold ref */
+       obj->u.s.owner_ref = 1; /* One owner reference */
+       obj->u.s.owner = owner;
+       strncpy(obj->u.s.name, name, OBJ_NAME_LEN);
+       obj->u.s.name[OBJ_NAME_LEN - 1] = '\0';
        return obj - objd_table.array;
 }
 
@@ -163,7 +186,7 @@ void objd_ref(int id)
        obj->u.s.f_count++;
 }
 
-int lttng_ust_objd_unref(int id)
+int lttng_ust_objd_unref(int id, int is_owner)
 {
        struct lttng_ust_obj *obj = _objd_get(id);
 
@@ -173,6 +196,13 @@ int lttng_ust_objd_unref(int id)
                ERR("Reference counting error\n");
                return -EINVAL;
        }
+       if (is_owner) {
+               if (!obj->u.s.owner_ref) {
+                       ERR("Error decrementing owner reference");
+                       return -EINVAL;
+               }
+               obj->u.s.owner_ref--;
+       }
        if ((--obj->u.s.f_count) == 1) {
                const struct lttng_ust_objd_ops *ops = objd_ops(id);
                
@@ -188,8 +218,16 @@ void objd_table_destroy(void)
 {
        int i;
 
-       for (i = 0; i < objd_table.allocated_len; i++)
-               (void) lttng_ust_objd_unref(i);
+       for (i = 0; i < objd_table.allocated_len; i++) {
+               struct lttng_ust_obj *obj;
+
+               obj = _objd_get(i);
+               if (!obj)
+                       continue;
+               if (!obj->u.s.owner_ref)
+                       continue;       /* only unref owner ref. */
+               (void) lttng_ust_objd_unref(i, 1);
+       }
        free(objd_table.array);
        objd_table.array = NULL;
        objd_table.len = 0;
@@ -197,6 +235,25 @@ void objd_table_destroy(void)
        objd_table.freelist_head = -1;
 }
 
+void lttng_ust_objd_table_owner_cleanup(void *owner)
+{
+       int i;
+
+       for (i = 0; i < objd_table.allocated_len; i++) {
+               struct lttng_ust_obj *obj;
+
+               obj = _objd_get(i);
+               if (!obj)
+                       continue;
+               if (!obj->u.s.owner)
+                       continue;       /* skip root handles */
+               if (!obj->u.s.owner_ref)
+                       continue;       /* only unref owner ref. */
+               if (obj->u.s.owner == owner)
+                       (void) lttng_ust_objd_unref(i, 1);
+       }
+}
+
 /*
  * This is LTTng's own personal way to create an ABI for sessiond.
  * We send commands over a socket.
@@ -206,9 +263,10 @@ static const struct lttng_ust_objd_ops lttng_ops;
 static const struct lttng_ust_objd_ops lttng_session_ops;
 static const struct lttng_ust_objd_ops lttng_channel_ops;
 static const struct lttng_ust_objd_ops lttng_metadata_ops;
-static const struct lttng_ust_objd_ops lttng_event_ops;
+static const struct lttng_ust_objd_ops lttng_enabler_ops;
 static const struct lttng_ust_objd_ops lib_ring_buffer_objd_ops;
 static const struct lttng_ust_objd_ops lttng_tracepoint_list_ops;
+static const struct lttng_ust_objd_ops lttng_tracepoint_field_list_ops;
 
 enum channel_type {
        PER_CPU_CHANNEL,
@@ -219,20 +277,21 @@ int lttng_abi_create_root_handle(void)
 {
        int root_handle;
 
-       root_handle = objd_alloc(NULL, &lttng_ops);
+       /* root handles have NULL owners */
+       root_handle = objd_alloc(NULL, &lttng_ops, NULL, "root");
        return root_handle;
 }
 
 static
-int lttng_abi_create_session(void)
+int lttng_abi_create_session(void *owner)
 {
-       struct ltt_session *session;
+       struct lttng_session *session;
        int session_objd, ret;
 
-       session = ltt_session_create();
+       session = lttng_session_create();
        if (!session)
                return -ENOMEM;
-       session_objd = objd_alloc(session, &lttng_session_ops);
+       session_objd = objd_alloc(session, &lttng_session_ops, owner, "session");
        if (session_objd < 0) {
                ret = session_objd;
                goto objd_error;
@@ -241,7 +300,7 @@ int lttng_abi_create_session(void)
        return session_objd;
 
 objd_error:
-       ltt_session_destroy(session);
+       lttng_session_destroy(session);
        return ret;
 }
 
@@ -249,32 +308,18 @@ static
 long lttng_abi_tracer_version(int objd,
        struct lttng_ust_tracer_version *v)
 {
-       v->version = LTTNG_UST_VERSION;
-       v->patchlevel = LTTNG_UST_PATCHLEVEL;
-       v->sublevel = LTTNG_UST_SUBLEVEL;
+       v->major = LTTNG_UST_INTERNAL_MAJOR_VERSION;
+       v->minor = LTTNG_UST_INTERNAL_MINOR_VERSION;
+       v->patchlevel = LTTNG_UST_INTERNAL_PATCHLEVEL_VERSION;
        return 0;
 }
 
 static
 long lttng_abi_add_context(int objd,
        struct lttng_ust_context *context_param,
-       struct lttng_ctx **ctx, struct ltt_session *session)
-{
-       if (session->been_active)
-               return -EPERM;
-
-       switch (context_param->ctx) {
-       case LTTNG_UST_CONTEXT_PTHREAD_ID:
-               return lttng_add_pthread_id_to_ctx(ctx);
-       case LTTNG_UST_CONTEXT_VTID:
-               return lttng_add_vtid_to_ctx(ctx);
-       case LTTNG_UST_CONTEXT_VPID:
-               return lttng_add_vpid_to_ctx(ctx);
-       case LTTNG_UST_CONTEXT_PROCNAME:
-               return lttng_add_procname_to_ctx(ctx);
-       default:
-               return -EINVAL;
-       }
+       struct lttng_ctx **ctx, struct lttng_session *session)
+{
+       return lttng_attach_context(context_param, ctx, session);
 }
 
 /**
@@ -283,6 +328,8 @@ long lttng_abi_add_context(int objd,
  *     @objd: the object descriptor
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This descriptor implements lttng commands:
  *     LTTNG_UST_SESSION
@@ -291,22 +338,27 @@ long lttng_abi_add_context(int objd,
  *             Returns the LTTng kernel tracer version
  *     LTTNG_UST_TRACEPOINT_LIST
  *             Returns a file descriptor listing available tracepoints
+ *     LTTNG_UST_TRACEPOINT_FIELD_LIST
+ *             Returns a file descriptor listing available tracepoint fields
  *     LTTNG_UST_WAIT_QUIESCENT
  *             Returns after all previously running probes have completed
  *
  * The returned session will be deleted when its file descriptor is closed.
  */
 static
-long lttng_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
        switch (cmd) {
        case LTTNG_UST_SESSION:
-               return lttng_abi_create_session();
+               return lttng_abi_create_session(owner);
        case LTTNG_UST_TRACER_VERSION:
                return lttng_abi_tracer_version(objd,
                                (struct lttng_ust_tracer_version *) arg);
        case LTTNG_UST_TRACEPOINT_LIST:
-               return lttng_abi_tracepoint_list();
+               return lttng_abi_tracepoint_list(owner);
+       case LTTNG_UST_TRACEPOINT_FIELD_LIST:
+               return lttng_abi_tracepoint_field_list(owner);
        case LTTNG_UST_WAIT_QUIESCENT:
                synchronize_trace();
                return 0;
@@ -327,19 +379,22 @@ static const struct lttng_ust_objd_ops lttng_ops = {
 static
 void lttng_metadata_create_events(int channel_objd)
 {
-       struct ltt_channel *channel = objd_private(channel_objd);
+       struct lttng_channel *chan = objd_private(channel_objd);
+       struct lttng_enabler *enabler;
        static struct lttng_ust_event metadata_params = {
                .instrumentation = LTTNG_UST_TRACEPOINT,
                .name = "lttng_ust:metadata",
+               .loglevel_type = LTTNG_UST_LOGLEVEL_ALL,
+               .loglevel = TRACE_DEFAULT,
        };
-       struct ltt_event *event;
 
        /*
         * We tolerate no failure path after event creation. It will stay
         * invariant for the rest of the session.
         */
-       event = ltt_event_create(channel, &metadata_params, NULL);
-       if (!event) {
+       enabler = lttng_enabler_create(LTTNG_ENABLER_EVENT,
+                               &metadata_params, chan);
+       if (!enabler) {
                goto create_error;
        }
        return;
@@ -351,15 +406,17 @@ create_error:
 
 int lttng_abi_create_channel(int session_objd,
                             struct lttng_ust_channel *chan_param,
-                            enum channel_type channel_type)
+                            enum channel_type channel_type,
+                            union ust_args *uargs,
+                            void *owner)
 {
-       struct ltt_session *session = objd_private(session_objd);
+       struct lttng_session *session = objd_private(session_objd);
        const struct lttng_ust_objd_ops *ops;
        const char *transport_name;
-       struct ltt_channel *chan;
+       struct lttng_channel *chan;
        int chan_objd;
        int ret = 0;
-       struct ltt_channel chan_priv_init;
+       struct lttng_channel chan_priv_init;
 
        switch (channel_type) {
        case PER_CPU_CHANNEL:
@@ -382,7 +439,7 @@ int lttng_abi_create_channel(int session_objd,
                transport_name = "<unknown>";
                return -EINVAL;
        }
-       chan_objd = objd_alloc(NULL, ops);
+       chan_objd = objd_alloc(NULL, ops, owner, "channel");
        if (chan_objd < 0) {
                ret = chan_objd;
                goto objd_error;
@@ -395,14 +452,14 @@ int lttng_abi_create_channel(int session_objd,
         * We tolerate no failure path after channel creation. It will stay
         * invariant for the rest of the session.
         */
-       chan = ltt_channel_create(session, transport_name, NULL,
+       chan = lttng_channel_create(session, transport_name, NULL,
                                  chan_param->subbuf_size,
                                  chan_param->num_subbuf,
                                  chan_param->switch_timer_interval,
                                  chan_param->read_timer_interval,
-                                 &chan_param->shm_fd,
-                                 &chan_param->wait_fd,
-                                 &chan_param->memory_map_size,
+                                 &uargs->channel.shm_fd,
+                                 &uargs->channel.wait_fd,
+                                 &uargs->channel.memory_map_size,
                                  &chan_priv_init);
        if (!chan) {
                ret = -EINVAL;
@@ -423,7 +480,7 @@ chan_error:
        {
                int err;
 
-               err = lttng_ust_objd_unref(chan_objd);
+               err = lttng_ust_objd_unref(chan_objd, 1);
                assert(!err);
        }
 objd_error:
@@ -436,6 +493,8 @@ objd_error:
  *     @obj: the object
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This descriptor implements lttng commands:
  *     LTTNG_UST_CHANNEL
@@ -450,25 +509,26 @@ objd_error:
  * The returned channel will be deleted when its file descriptor is closed.
  */
 static
-long lttng_session_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_session_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
-       struct ltt_session *session = objd_private(objd);
+       struct lttng_session *session = objd_private(objd);
 
        switch (cmd) {
        case LTTNG_UST_CHANNEL:
                return lttng_abi_create_channel(objd,
                                (struct lttng_ust_channel *) arg,
-                               PER_CPU_CHANNEL);
+                               PER_CPU_CHANNEL, uargs, owner);
        case LTTNG_UST_SESSION_START:
        case LTTNG_UST_ENABLE:
-               return ltt_session_enable(session);
+               return lttng_session_enable(session);
        case LTTNG_UST_SESSION_STOP:
        case LTTNG_UST_DISABLE:
-               return ltt_session_disable(session);
+               return lttng_session_disable(session);
        case LTTNG_UST_METADATA:
                return lttng_abi_create_channel(objd,
                                (struct lttng_ust_channel *) arg,
-                               METADATA_CHANNEL);
+                               METADATA_CHANNEL, uargs, owner);
        default:
                return -EINVAL;
        }
@@ -485,10 +545,10 @@ long lttng_session_cmd(int objd, unsigned int cmd, unsigned long arg)
 static
 int lttng_release_session(int objd)
 {
-       struct ltt_session *session = objd_private(objd);
+       struct lttng_session *session = objd_private(objd);
 
        if (session) {
-               ltt_session_destroy(session);
+               lttng_session_destroy(session);
                return 0;
        } else {
                return -EINVAL;
@@ -500,59 +560,39 @@ static const struct lttng_ust_objd_ops lttng_session_ops = {
        .cmd = lttng_session_cmd,
 };
 
-/*
- * beware: we don't keep the mutex over the send, but we must walk the
- * whole list each time we are called again. So sending one tracepoint
- * at a time means this is O(n^2). TODO: do as in the kernel and send
- * multiple tracepoints for each call to amortize this cost.
- */
-static
-void ltt_tracepoint_list_get(struct ltt_tracepoint_list *list,
-               char *tp_list_entry)
-{
-next:
-       if (!list->got_first) {
-               tracepoint_iter_start(&list->iter);
-               list->got_first = 1;
-               goto copy;
-       }
-       tracepoint_iter_next(&list->iter);
-copy:
-       if (!list->iter.tracepoint) {
-               tp_list_entry[0] = '\0';        /* end of list */
-       } else {
-               if (!strcmp((*list->iter.tracepoint)->name,
-                               "lttng_ust:metadata"))
-                       goto next;
-               memcpy(tp_list_entry, (*list->iter.tracepoint)->name,
-                       LTTNG_UST_SYM_NAME_LEN);
-       }
-}
-
 static
-long lttng_tracepoint_list_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_tracepoint_list_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
-       struct ltt_tracepoint_list *list = objd_private(objd);
-       char *str = (char *) arg;
+       struct lttng_ust_tracepoint_list *list = objd_private(objd);
+       struct lttng_ust_tracepoint_iter *tp =
+               (struct lttng_ust_tracepoint_iter *) arg;
+       struct lttng_ust_tracepoint_iter *iter;
 
        switch (cmd) {
        case LTTNG_UST_TRACEPOINT_LIST_GET:
-               ltt_tracepoint_list_get(list, str);
-               if (str[0] == '\0')
-                       return -ENOENT;
+       {
+       retry:
+               iter = lttng_ust_tracepoint_list_get_iter_next(list);
+               if (!iter)
+                       return -LTTNG_UST_ERR_NOENT;
+               if (!strcmp(iter->name, "lttng_ust:metadata"))
+                       goto retry;
+               memcpy(tp, iter, sizeof(*tp));
                return 0;
+       }
        default:
                return -EINVAL;
        }
 }
 
 static
-int lttng_abi_tracepoint_list(void)
+int lttng_abi_tracepoint_list(void *owner)
 {
        int list_objd, ret;
-       struct ltt_tracepoint_list *list;
+       struct lttng_ust_tracepoint_list *list;
 
-       list_objd = objd_alloc(NULL, &lttng_tracepoint_list_ops);
+       list_objd = objd_alloc(NULL, &lttng_tracepoint_list_ops, owner, "tp_list");
        if (list_objd < 0) {
                ret = list_objd;
                goto objd_error;
@@ -564,13 +604,20 @@ int lttng_abi_tracepoint_list(void)
        }
        objd_set_private(list_objd, list);
 
+       /* populate list by walking on all registered probes. */
+       ret = lttng_probes_get_event_list(list);
+       if (ret) {
+               goto list_error;
+       }
        return list_objd;
 
+list_error:
+       free(list);
 alloc_error:
        {
                int err;
 
-               err = lttng_ust_objd_unref(list_objd);
+               err = lttng_ust_objd_unref(list_objd, 1);
                assert(!err);
        }
 objd_error:
@@ -580,10 +627,10 @@ objd_error:
 static
 int lttng_release_tracepoint_list(int objd)
 {
-       struct ltt_tracepoint_list *list = objd_private(objd);
+       struct lttng_ust_tracepoint_list *list = objd_private(objd);
 
        if (list) {
-               tracepoint_iter_stop(&list->iter);
+               lttng_probes_prune_event_list(list);
                free(list);
                return 0;
        } else {
@@ -596,21 +643,107 @@ static const struct lttng_ust_objd_ops lttng_tracepoint_list_ops = {
        .cmd = lttng_tracepoint_list_cmd,
 };
 
+static
+long lttng_tracepoint_field_list_cmd(int objd, unsigned int cmd,
+       unsigned long arg, union ust_args *uargs, void *owner)
+{
+       struct lttng_ust_field_list *list = objd_private(objd);
+       struct lttng_ust_field_iter *tp = &uargs->field_list.entry;
+       struct lttng_ust_field_iter *iter;
+
+       switch (cmd) {
+       case LTTNG_UST_TRACEPOINT_FIELD_LIST_GET:
+       {
+       retry:
+               iter = lttng_ust_field_list_get_iter_next(list);
+               if (!iter)
+                       return -LTTNG_UST_ERR_NOENT;
+               if (!strcmp(iter->event_name, "lttng_ust:metadata"))
+                       goto retry;
+               memcpy(tp, iter, sizeof(*tp));
+               return 0;
+       }
+       default:
+               return -EINVAL;
+       }
+}
+
+static
+int lttng_abi_tracepoint_field_list(void *owner)
+{
+       int list_objd, ret;
+       struct lttng_ust_field_list *list;
+
+       list_objd = objd_alloc(NULL, &lttng_tracepoint_field_list_ops, owner,
+                       "tp_field_list");
+       if (list_objd < 0) {
+               ret = list_objd;
+               goto objd_error;
+       }
+       list = zmalloc(sizeof(*list));
+       if (!list) {
+               ret = -ENOMEM;
+               goto alloc_error;
+       }
+       objd_set_private(list_objd, list);
+
+       /* populate list by walking on all registered probes. */
+       ret = lttng_probes_get_field_list(list);
+       if (ret) {
+               goto list_error;
+       }
+       return list_objd;
+
+list_error:
+       free(list);
+alloc_error:
+       {
+               int err;
+
+               err = lttng_ust_objd_unref(list_objd, 1);
+               assert(!err);
+       }
+objd_error:
+       return ret;
+}
+
+static
+int lttng_release_tracepoint_field_list(int objd)
+{
+       struct lttng_ust_field_list *list = objd_private(objd);
+
+       if (list) {
+               lttng_probes_prune_field_list(list);
+               free(list);
+               return 0;
+       } else {
+               return -EINVAL;
+       }
+}
+
+static const struct lttng_ust_objd_ops lttng_tracepoint_field_list_ops = {
+       .release = lttng_release_tracepoint_field_list,
+       .cmd = lttng_tracepoint_field_list_cmd,
+};
+
 struct stream_priv_data {
        struct lttng_ust_lib_ring_buffer *buf;
-       struct ltt_channel *ltt_chan;
+       struct lttng_channel *lttng_chan;
 };
 
 static
-int lttng_abi_open_stream(int channel_objd, struct lttng_ust_stream *info)
+int lttng_abi_open_stream(int channel_objd, struct lttng_ust_stream *info,
+               union ust_args *uargs, void *owner)
 {
-       struct ltt_channel *channel = objd_private(channel_objd);
+       struct lttng_channel *channel = objd_private(channel_objd);
        struct lttng_ust_lib_ring_buffer *buf;
        struct stream_priv_data *priv;
        int stream_objd, ret;
 
        buf = channel->ops->buffer_read_open(channel->chan, channel->handle,
-                       &info->shm_fd, &info->wait_fd, &info->memory_map_size);
+                       &uargs->stream.shm_fd,
+                       &uargs->stream.wait_fd,
+                       &uargs->stream.memory_map_size);
        if (!buf)
                return -ENOENT;
 
@@ -620,8 +753,8 @@ int lttng_abi_open_stream(int channel_objd, struct lttng_ust_stream *info)
                goto alloc_error;
        }
        priv->buf = buf;
-       priv->ltt_chan = channel;
-       stream_objd = objd_alloc(priv, &lib_ring_buffer_objd_ops);
+       priv->lttng_chan = channel;
+       stream_objd = objd_alloc(priv, &lib_ring_buffer_objd_ops, owner, "open_stream");
        if (stream_objd < 0) {
                ret = stream_objd;
                goto objd_error;
@@ -638,15 +771,17 @@ alloc_error:
 }
 
 static
-int lttng_abi_create_event(int channel_objd,
-                          struct lttng_ust_event *event_param)
+int lttng_abi_create_enabler(int channel_objd,
+                          struct lttng_ust_event *event_param,
+                          void *owner,
+                          enum lttng_enabler_type type)
 {
-       struct ltt_channel *channel = objd_private(channel_objd);
-       struct ltt_event *event;
+       struct lttng_channel *channel = objd_private(channel_objd);
+       struct lttng_enabler *enabler;
        int event_objd, ret;
 
        event_param->name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
-       event_objd = objd_alloc(NULL, &lttng_event_ops);
+       event_objd = objd_alloc(NULL, &lttng_enabler_ops, owner, "enabler");
        if (event_objd < 0) {
                ret = event_objd;
                goto objd_error;
@@ -655,12 +790,12 @@ int lttng_abi_create_event(int channel_objd,
         * We tolerate no failure path after event creation. It will stay
         * invariant for the rest of the session.
         */
-       event = ltt_event_create(channel, event_param, NULL);
-       if (!event) {
-               ret = -EINVAL;
+       enabler = lttng_enabler_create(type, event_param, channel);
+       if (!enabler) {
+               ret = -ENOMEM;
                goto event_error;
        }
-       objd_set_private(event_objd, event);
+       objd_set_private(event_objd, enabler);
        /* The event holds a reference on the channel */
        objd_ref(channel_objd);
        return event_objd;
@@ -669,7 +804,7 @@ event_error:
        {
                int err;
 
-               err = lttng_ust_objd_unref(event_objd);
+               err = lttng_ust_objd_unref(event_objd, 1);
                assert(!err);
        }
 objd_error:
@@ -682,6 +817,8 @@ objd_error:
  *     @objd: the object descriptor
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This object descriptor implements lttng commands:
  *      LTTNG_UST_STREAM
@@ -699,9 +836,10 @@ objd_error:
  * Channel and event file descriptors also hold a reference on the session.
  */
 static
-long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
-       struct ltt_channel *channel = objd_private(objd);
+       struct lttng_channel *channel = objd_private(objd);
 
        switch (cmd) {
        case LTTNG_UST_STREAM:
@@ -710,18 +848,29 @@ long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
 
                stream = (struct lttng_ust_stream *) arg;
                /* stream used as output */
-               return lttng_abi_open_stream(objd, stream);
+               return lttng_abi_open_stream(objd, stream, uargs, owner);
        }
        case LTTNG_UST_EVENT:
-               return lttng_abi_create_event(objd, (struct lttng_ust_event *) arg);
+       {
+               struct lttng_ust_event *event_param =
+                       (struct lttng_ust_event *) arg;
+               if (event_param->name[strlen(event_param->name) - 1] == '*') {
+                       /* If ends with wildcard, create wildcard. */
+                       return lttng_abi_create_enabler(objd, event_param,
+                                       owner, LTTNG_ENABLER_WILDCARD);
+               } else {
+                       return lttng_abi_create_enabler(objd, event_param,
+                                       owner, LTTNG_ENABLER_EVENT);
+               }
+       }
        case LTTNG_UST_CONTEXT:
                return lttng_abi_add_context(objd,
                                (struct lttng_ust_context *) arg,
                                &channel->ctx, channel->session);
        case LTTNG_UST_ENABLE:
-               return ltt_channel_enable(channel);
+               return lttng_channel_enable(channel);
        case LTTNG_UST_DISABLE:
-               return ltt_channel_disable(channel);
+               return lttng_channel_disable(channel);
        case LTTNG_UST_FLUSH_BUFFER:
                return channel->ops->flush_buffer(channel->chan, channel->handle);
        default:
@@ -735,6 +884,8 @@ long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
  *     @objd: the object descriptor
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This object descriptor implements lttng commands:
  *      LTTNG_UST_STREAM
@@ -743,9 +894,10 @@ long lttng_channel_cmd(int objd, unsigned int cmd, unsigned long arg)
  * Channel and event file descriptors also hold a reference on the session.
  */
 static
-long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
-       struct ltt_channel *channel = objd_private(objd);
+       struct lttng_channel *channel = objd_private(objd);
 
        switch (cmd) {
        case LTTNG_UST_STREAM:
@@ -754,7 +906,7 @@ long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg)
 
                stream = (struct lttng_ust_stream *) arg;
                /* stream used as output */
-               return lttng_abi_open_stream(objd, stream);
+               return lttng_abi_open_stream(objd, stream, uargs, owner);
        }
        case LTTNG_UST_FLUSH_BUFFER:
                return channel->ops->flush_buffer(channel->chan, channel->handle);
@@ -763,49 +915,18 @@ long lttng_metadata_cmd(int objd, unsigned int cmd, unsigned long arg)
        }
 }
 
-#if 0
-/**
- *     lttng_channel_poll - lttng stream addition/removal monitoring
- *
- *     @file: the file
- *     @wait: poll table
- */
-unsigned int lttng_channel_poll(struct file *file, poll_table *wait)
-{
-       struct ltt_channel *channel = file->private_data;
-       unsigned int mask = 0;
-
-       if (file->f_mode & FMODE_READ) {
-               poll_wait_set_exclusive(wait);
-               poll_wait(file, channel->ops->get_hp_wait_queue(channel->chan),
-                         wait);
-
-               if (channel->ops->is_disabled(channel->chan))
-                       return POLLERR;
-               if (channel->ops->is_finalized(channel->chan))
-                       return POLLHUP;
-               if (channel->ops->buffer_has_read_closed_stream(channel->chan))
-                       return POLLIN | POLLRDNORM;
-               return 0;
-       }
-       return mask;
-
-}
-#endif //0
-
 static
 int lttng_channel_release(int objd)
 {
-       struct ltt_channel *channel = objd_private(objd);
+       struct lttng_channel *channel = objd_private(objd);
 
        if (channel)
-               return lttng_ust_objd_unref(channel->session->objd);
+               return lttng_ust_objd_unref(channel->session->objd, 0);
        return 0;
 }
 
 static const struct lttng_ust_objd_ops lttng_channel_ops = {
        .release = lttng_channel_release,
-       //.poll = lttng_channel_poll,
        .cmd = lttng_channel_cmd,
 };
 
@@ -820,12 +941,15 @@ static const struct lttng_ust_objd_ops lttng_metadata_ops = {
  *     @objd: the object descriptor
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This object descriptor implements lttng commands:
  *             (None for now. Access is done directly though shm.)
  */
 static
-long lttng_rb_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_rb_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
        switch (cmd) {
        default:
@@ -838,11 +962,11 @@ int lttng_rb_release(int objd)
 {
        struct stream_priv_data *priv = objd_private(objd);
        struct lttng_ust_lib_ring_buffer *buf;
-       struct ltt_channel *channel;
+       struct lttng_channel *channel;
 
        if (priv) {
                buf = priv->buf;
-               channel = priv->ltt_chan;
+               channel = priv->lttng_chan;
                free(priv);
                /*
                 * If we are at ABI exit, we don't want to close the
@@ -860,7 +984,7 @@ int lttng_rb_release(int objd)
                if (!lttng_ust_abi_close_in_progress)
                        channel->ops->buffer_read_close(buf, channel->handle);
 
-               return lttng_ust_objd_unref(channel->objd);
+               return lttng_ust_objd_unref(channel->objd, 0);
        }
        return 0;
 }
@@ -871,53 +995,67 @@ static const struct lttng_ust_objd_ops lib_ring_buffer_objd_ops = {
 };
 
 /**
- *     lttng_event_cmd - lttng control through object descriptors
+ *     lttng_enabler_cmd - lttng control through object descriptors
  *
  *     @objd: the object descriptor
  *     @cmd: the command
  *     @arg: command arg
+ *     @uargs: UST arguments (internal)
+ *     @owner: objd owner
  *
  *     This object descriptor implements lttng commands:
  *     LTTNG_UST_CONTEXT
- *             Prepend a context field to each record of this event
+ *             Prepend a context field to each record of events of this
+ *             enabler.
  *     LTTNG_UST_ENABLE
- *             Enable recording for this event (weak enable)
+ *             Enable recording for this enabler
  *     LTTNG_UST_DISABLE
- *             Disable recording for this event (strong disable)
+ *             Disable recording for this enabler
+ *     LTTNG_UST_FILTER
+ *             Attach a filter to an enabler.
  */
 static
-long lttng_event_cmd(int objd, unsigned int cmd, unsigned long arg)
+long lttng_enabler_cmd(int objd, unsigned int cmd, unsigned long arg,
+       union ust_args *uargs, void *owner)
 {
-       struct ltt_event *event = objd_private(objd);
+       struct lttng_enabler *enabler = objd_private(objd);
 
        switch (cmd) {
        case LTTNG_UST_CONTEXT:
-               return lttng_abi_add_context(objd,
-                               (struct lttng_ust_context *) arg,
-                               &event->ctx, event->chan->session);
+               return lttng_enabler_attach_context(enabler,
+                               (struct lttng_ust_context *) arg);
        case LTTNG_UST_ENABLE:
-               return ltt_event_enable(event);
+               return lttng_enabler_enable(enabler);
        case LTTNG_UST_DISABLE:
-               return ltt_event_disable(event);
+               return lttng_enabler_disable(enabler);
+       case LTTNG_UST_FILTER:
+       {
+               int ret;
+
+               ret = lttng_enabler_attach_bytecode(enabler,
+                               (struct lttng_ust_filter_bytecode_node *) arg);
+               if (ret)
+                       return ret;
+               return 0;
+       }
        default:
                return -EINVAL;
        }
 }
 
 static
-int lttng_event_release(int objd)
+int lttng_enabler_release(int objd)
 {
-       struct ltt_event *event = objd_private(objd);
+       struct lttng_enabler *enabler = objd_private(objd);
 
-       if (event)
-               return lttng_ust_objd_unref(event->chan->objd);
+       if (enabler)
+               return lttng_ust_objd_unref(enabler->chan->objd, 0);
        return 0;
 }
 
-/* TODO: filter control ioctl */
-static const struct lttng_ust_objd_ops lttng_event_ops = {
-       .release = lttng_event_release,
-       .cmd = lttng_event_cmd,
+static const struct lttng_ust_objd_ops lttng_enabler_ops = {
+       .release = lttng_enabler_release,
+       .cmd = lttng_enabler_cmd,
 };
 
 void lttng_ust_abi_exit(void)
This page took 0.040922 seconds and 4 git commands to generate.