Add wildcard support

[lttng-ust.git] / liblttng-ust / ltt-events.c

diff --git a/liblttng-ust/ltt-events.c b/liblttng-ust/ltt-events.c
index 06d2f89714feeeaa3771361efd15bff5a003f2ad..9650e8b0703d608492d61ac273e1ef6f1768296b 100644 (file)
--- a/liblttng-ust/ltt-events.c
+++ b/liblttng-ust/ltt-events.c
@@ -31,6 +31,7 @@
 
 #include <usterr-signal-safe.h>
 #include <helper.h>
+#include "error.h"
 
 #include "ltt-tracer.h"
 #include "ltt-tracer-core.h"
@@ -73,6 +74,8 @@ struct ust_pending_probe {
 };
 
 static void _ltt_event_destroy(struct ltt_event *event);
+static void _ltt_loglevel_destroy(struct session_loglevel *sl);
+static void _ltt_wildcard_destroy(struct session_wildcard *sw);
 static void _ltt_channel_destroy(struct ltt_channel *chan);
 static int _ltt_event_unregister(struct ltt_event *event);
 static
@@ -133,6 +136,78 @@ int pending_probe_fix_events(const struct lttng_event_desc *desc)
        size_t name_len = strlen(name) + 1;
        uint32_t hash = jhash(name, name_len - 1, 0);
        int ret = 0;
+       struct lttng_ust_event event_param;
+
+       /*
+        * For this event, we need to lookup the loglevel. If active (in
+        * the active loglevels hash table), we must create the event.
+        */
+       if (desc->loglevel) {
+               const struct tracepoint_loglevel_entry *ev_ll;
+               struct loglevel_entry *loglevel;
+
+               ev_ll = *desc->loglevel;
+               loglevel = get_loglevel(ev_ll->identifier);
+               if (!loglevel)
+                       loglevel = get_loglevel_value(ev_ll->value);
+               if (loglevel) {
+                       struct session_loglevel *sl;
+
+                       cds_list_for_each_entry(sl, &loglevel->session_list,
+                                       session_list) {
+                               struct ltt_event *ev;
+                               int ret;
+
+                               memcpy(&event_param, &sl->event_param,
+                                               sizeof(event_param));
+                               memcpy(event_param.name,
+                                       desc->name,
+                                       sizeof(event_param.name));
+                               /* create event */
+                               ret = ltt_event_create(sl->chan,
+                                       &event_param, NULL,
+                                       &ev);
+                               if (ret) {
+                                       DBG("Error creating event");
+                                       continue;
+                               }
+                               cds_list_add(&ev->loglevel_list,
+                                       &sl->events);
+                       }
+               }
+       }
+
+       /* Wildcard */
+       {
+               struct wildcard_entry *wildcard;
+
+               wildcard = match_wildcard(desc->name);
+               if (strcmp(desc->name, "lttng_ust:metadata") && wildcard) {
+                       struct session_wildcard *sw;
+
+                       cds_list_for_each_entry(sw, &wildcard->session_list,
+                                       session_list) {
+                               struct ltt_event *ev;
+                               int ret;
+
+                               memcpy(&event_param, &sw->event_param,
+                                               sizeof(event_param));
+                               memcpy(event_param.name,
+                                       desc->name,
+                                       sizeof(event_param.name));
+                               /* create event */
+                               ret = ltt_event_create(sw->chan,
+                                       &event_param, NULL,
+                                       &ev);
+                               if (ret) {
+                                       DBG("Error creating event");
+                                       continue;
+                               }
+                               cds_list_add(&ev->wildcard_list,
+                                       &sw->events);
+                       }
+               }
+       }
 
        head = &pending_probe_table[hash & (PENDING_PROBE_HASH_SIZE - 1)];
        cds_hlist_for_each_entry_safe(e, node, p, head, node) {
@@ -173,6 +248,8 @@ struct ltt_session *ltt_session_create(void)
                return NULL;
        CDS_INIT_LIST_HEAD(&session->chan);
        CDS_INIT_LIST_HEAD(&session->events);
+       CDS_INIT_LIST_HEAD(&session->loglevels);
+       CDS_INIT_LIST_HEAD(&session->wildcards);
        uuid_generate(session->uuid);
        cds_list_add(&session->list, &sessions);
        return session;
@@ -182,6 +259,8 @@ void ltt_session_destroy(struct ltt_session *session)
 {
        struct ltt_channel *chan, *tmpchan;
        struct ltt_event *event, *tmpevent;
+       struct session_loglevel *loglevel, *tmploglevel;
+       struct session_wildcard *wildcard, *tmpwildcard;
        int ret;
 
        CMM_ACCESS_ONCE(session->active) = 0;
@@ -190,6 +269,10 @@ void ltt_session_destroy(struct ltt_session *session)
                WARN_ON(ret);
        }
        synchronize_trace();    /* Wait for in-flight events to complete */
+       cds_list_for_each_entry_safe(loglevel, tmploglevel, &session->loglevels, list)
+               _ltt_loglevel_destroy(loglevel);
+       cds_list_for_each_entry_safe(wildcard, tmpwildcard, &session->wildcards, list)
+               _ltt_wildcard_destroy(wildcard);
        cds_list_for_each_entry_safe(event, tmpevent, &session->events, list)
                _ltt_event_destroy(event);
        cds_list_for_each_entry_safe(chan, tmpchan, &session->chan, list)
@@ -347,6 +430,46 @@ void _ltt_channel_destroy(struct ltt_channel *chan)
        chan->ops->channel_destroy(chan);
 }
 
+int ltt_loglevel_create(struct ltt_channel *chan,
+       struct lttng_ust_event *event_param,
+       struct session_loglevel **_sl)
+{
+       struct session_loglevel *sl;
+
+       sl = add_loglevel(event_param->name, chan, event_param);
+       if (!sl || IS_ERR(sl)) {
+               return PTR_ERR(sl);
+       }
+       *_sl = sl;
+       return 0;
+}
+
+static
+void _ltt_loglevel_destroy(struct session_loglevel *sl)
+{
+       _remove_loglevel(sl);
+}
+
+int ltt_wildcard_create(struct ltt_channel *chan,
+       struct lttng_ust_event *event_param,
+       struct session_wildcard **_sw)
+{
+       struct session_wildcard *sw;
+
+       sw = add_wildcard(event_param->name, chan, event_param);
+       if (!sw || IS_ERR(sw)) {
+               return PTR_ERR(sw);
+       }
+       *_sw = sw;
+       return 0;
+}
+
+static
+void _ltt_wildcard_destroy(struct session_wildcard *sw)
+{
+       _remove_wildcard(sw);
+}
+
 /*
  * Supports event creation while tracing session is active.
  */
@@ -410,11 +533,7 @@ int ltt_event_create(struct ltt_channel *chan,
                }
                break;
        case LTTNG_UST_TRACEPOINT_LOGLEVEL:
-               /*
-                * TODO: add tracepoint loglevel to hash table, with
-                * event info. Enable all events corresponding to
-                * loglevel.
-                */
+               assert(0);
                break;
        default:
                WARN_ON_ONCE(1);
@@ -748,6 +867,19 @@ int _ltt_event_metadata_statedump(struct ltt_session *session,
        if (ret)
                goto end;
 
+       if (event->desc->loglevel) {
+               const struct tracepoint_loglevel_entry *ll_entry;
+
+               ll_entry = *event->desc->loglevel;
+               ret = lttng_metadata_printf(session,
+                       "       loglevel.identifier = \"%s\";\n"
+                       "       loglevel.value = %lld;\n",
+                       ll_entry->identifier,
+                       (long long) ll_entry->value);
+               if (ret)
+                       goto end;
+       }
+
        if (event->ctx) {
                ret = lttng_metadata_printf(session,
                        "       context := struct {\n");