Add probe provider unregister function
[lttng-ust.git] / liblttng-ust / lttng-events.c
index 64f70f92bd9701ff15936d9fb6ec84176d389187..57eb6f7f3a2c745e77191f7ba9b00d1234383f71 100644 (file)
@@ -21,6 +21,7 @@
  */
 
 #define _GNU_SOURCE
+#define _LGPL_SOURCE
 #include <stdio.h>
 #include <urcu/list.h>
 #include <urcu/hlist.h>
@@ -32,6 +33,7 @@
 #include <stddef.h>
 #include <inttypes.h>
 #include <time.h>
+#include <stdbool.h>
 #include <lttng/ust-endian.h>
 #include "clock.h"
 
@@ -54,6 +56,7 @@
 #include "lttng-ust-uuid.h"
 
 #include "tracepoint-internal.h"
+#include "string-utils.h"
 #include "lttng-tracer.h"
 #include "lttng-tracer-core.h"
 #include "lttng-ust-statedump.h"
@@ -534,7 +537,6 @@ int lttng_event_create(const struct lttng_event_desc *desc,
        struct lttng_event *event;
        struct lttng_session *session = chan->session;
        struct cds_hlist_head *head;
-       struct cds_hlist_node *node;
        int ret = 0;
        size_t name_len = strlen(event_name);
        uint32_t hash;
@@ -543,15 +545,6 @@ int lttng_event_create(const struct lttng_event_desc *desc,
 
        hash = jhash(event_name, name_len, 0);
        head = &chan->session->events_ht.table[hash & (LTTNG_UST_EVENT_HT_SIZE - 1)];
-       cds_hlist_for_each_entry(event, node, head, hlist) {
-               assert(event->desc);
-               if (!strncmp(event->desc->name, desc->name,
-                                       LTTNG_UST_SYM_NAME_LEN - 1)
-                               && chan == event->chan) {
-                       ret = -EEXIST;
-                       goto exist;
-               }
-       }
 
        notify_socket = lttng_get_notify_socket(session->owner);
        if (notify_socket < 0) {
@@ -620,21 +613,19 @@ sessiond_register_error:
 cache_error:
 create_enum_error:
 socket_error:
-exist:
        return ret;
 }
 
 static
-int lttng_desc_match_wildcard_enabler(const struct lttng_event_desc *desc,
+int lttng_desc_match_star_glob_enabler(const struct lttng_event_desc *desc,
                struct lttng_enabler *enabler)
 {
        int loglevel = 0;
        unsigned int has_loglevel = 0;
 
-       assert(enabler->type == LTTNG_ENABLER_WILDCARD);
-       /* Compare excluding final '*' */
-       if (strncmp(desc->name, enabler->event_param.name,
-                       strlen(enabler->event_param.name) - 1))
+       assert(enabler->type == LTTNG_ENABLER_STAR_GLOB);
+       if (!strutils_star_glob_match(enabler->event_param.name, SIZE_MAX,
+                       desc->name, SIZE_MAX))
                return 0;
        if (desc->loglevel) {
                loglevel = *(*desc->loglevel);
@@ -674,34 +665,35 @@ static
 int lttng_desc_match_enabler(const struct lttng_event_desc *desc,
                struct lttng_enabler *enabler)
 {
-       struct lttng_ust_excluder_node *excluder;
-
-       /* If event matches with an excluder, return 'does not match' */
-       cds_list_for_each_entry(excluder, &enabler->excluder_head, node) {
-               int count;
-
-               for (count = 0; count < excluder->excluder.count; count++) {
-                       int found, len;
-                       char *excluder_name;
-
-                       excluder_name = (char *) (excluder->excluder.names)
-                                       + count * LTTNG_UST_SYM_NAME_LEN;
-                       len = strnlen(excluder_name, LTTNG_UST_SYM_NAME_LEN);
-                       if (len > 0 && excluder_name[len - 1] == '*') {
-                               found = !strncmp(desc->name, excluder_name,
-                                               len - 1);
-                       } else {
-                               found = !strncmp(desc->name, excluder_name,
-                                               LTTNG_UST_SYM_NAME_LEN - 1);
-                       }
-                       if (found) {
-                               return 0;
+       switch (enabler->type) {
+       case LTTNG_ENABLER_STAR_GLOB:
+       {
+               struct lttng_ust_excluder_node *excluder;
+
+               if (!lttng_desc_match_star_glob_enabler(desc, enabler)) {
+                       return 0;
+               }
+
+               /*
+                * If the matching event matches with an excluder,
+                * return 'does not match'
+                */
+               cds_list_for_each_entry(excluder, &enabler->excluder_head, node) {
+                       int count;
+
+                       for (count = 0; count < excluder->excluder.count; count++) {
+                               int len;
+                               char *excluder_name;
+
+                               excluder_name = (char *) (excluder->excluder.names)
+                                               + count * LTTNG_UST_SYM_NAME_LEN;
+                               len = strnlen(excluder_name, LTTNG_UST_SYM_NAME_LEN);
+                               if (len > 0 && strutils_star_glob_match(excluder_name, len, desc->name, SIZE_MAX))
+                                       return 0;
                        }
                }
+               return 1;
        }
-       switch (enabler->type) {
-       case LTTNG_ENABLER_WILDCARD:
-               return lttng_desc_match_wildcard_enabler(desc, enabler);
        case LTTNG_ENABLER_EVENT:
                return lttng_desc_match_event_enabler(desc, enabler);
        default:
@@ -796,6 +788,111 @@ void lttng_create_event_if_missing(struct lttng_enabler *enabler)
        }
 }
 
+/*
+ * Iterate over all the UST sessions to unregister and destroy all probes from
+ * the probe provider descriptor received as argument. Must me called with the
+ * ust_lock held.
+ */
+void lttng_probe_provider_unregister_events(struct lttng_probe_desc *provider_desc)
+{
+       struct cds_hlist_node *node, *tmp_node;
+       struct cds_list_head *sessionsp;
+       struct lttng_session *session;
+       struct cds_hlist_head *head;
+       struct lttng_event *event;
+       int i;
+
+       /* Get handle on list of sessions. */
+       sessionsp = _lttng_get_sessions();
+
+       /*
+        * Iterate over all events in the probe provider descriptions and sessions
+        * to queue the unregistration of the events.
+        */
+       for (i = 0; i < provider_desc->nr_events; i++) {
+               const struct lttng_event_desc *event_desc;
+               const char *event_name;
+               size_t name_len;
+               uint32_t hash;
+
+               event_desc = provider_desc->event_desc[i];
+               event_name = event_desc->name;
+               name_len = strlen(event_name);
+               hash = jhash(event_name, name_len, 0);
+
+               /* Iterate over all session to find the current event description. */
+               cds_list_for_each_entry(session, sessionsp, node) {
+                       /*
+                        * Get the list of events in the hashtable bucket and iterate to
+                        * find the event matching this descriptor.
+                        */
+                       head = &session->events_ht.table[hash & (LTTNG_UST_EVENT_HT_SIZE - 1)];
+                       cds_hlist_for_each_entry(event, node, head, hlist) {
+                               if (event_desc == event->desc) {
+                                       /* Queue the unregistration of this event. */
+                                       _lttng_event_unregister(event);
+                                       break;
+                               }
+                       }
+               }
+       }
+
+       /* Wait for grace period. */
+       synchronize_trace();
+       /* Prune the unregistration queue. */
+       __tracepoint_probe_prune_release_queue();
+
+       /*
+        * It is now safe to destroy the events and remove them from the event list
+        * and hashtables.
+        */
+       for (i = 0; i < provider_desc->nr_events; i++) {
+               const struct lttng_event_desc *event_desc;
+               const char *event_name;
+               size_t name_len;
+               uint32_t hash;
+
+               event_desc = provider_desc->event_desc[i];
+               event_name = event_desc->name;
+               name_len = strlen(event_name);
+               hash = jhash(event_name, name_len, 0);
+
+               /* Iterate over all sessions to find the current event description. */
+               cds_list_for_each_entry(session, sessionsp, node) {
+                       /*
+                        * Get the list of events in the hashtable bucket and iterate to
+                        * find the event matching this descriptor.
+                        */
+                       head = &session->events_ht.table[hash & (LTTNG_UST_EVENT_HT_SIZE - 1)];
+                       cds_hlist_for_each_entry_safe(event, node, tmp_node, head, hlist) {
+                               if (event_desc == event->desc) {
+                                       /* Destroy enums of the current event. */
+                                       for (j = 0; j < event->desc->nr_fields; j++) {
+                                               const struct lttng_enum_desc *enum_desc;
+                                               const struct lttng_event_field *field;
+                                               struct lttng_enum *curr_enum;
+
+                                               field = &(event->desc->fields[j]);
+                                               if (field->type.atype != atype_enum) {
+                                                       continue;
+                                               }
+
+                                               enum_desc = field->type.u.basic.enumeration.desc;
+                                               curr_enum = lttng_ust_enum_get_from_desc(session, enum_desc);
+                                               if (curr_enum) {
+                                                       _lttng_enum_destroy(curr_enum);
+                                               }
+                                       }
+
+                                       /* Destroy event. */
+                                       _lttng_event_destroy(event);
+                                       break;
+                               }
+                       }
+               }
+       }
+}
+
 /*
  * Create events associated with an enabler (if not already present),
  * and add backward reference from the event to the enabler.
@@ -891,7 +988,11 @@ void _lttng_event_destroy(struct lttng_event *event)
 {
        struct lttng_enabler_ref *enabler_ref, *tmp_enabler_ref;
 
+       /* Remove from event list. */
        cds_list_del(&event->node);
+       /* Remove from event hash table. */
+       cds_hlist_del(&event->hlist);
+
        lttng_destroy_context(event->ctx);
        lttng_free_event_filter_runtime(event);
        /* Free event enabler refs */
@@ -905,6 +1006,7 @@ static
 void _lttng_enum_destroy(struct lttng_enum *_enum)
 {
        cds_list_del(&_enum->node);
+       cds_hlist_del(&_enum->hlist);
        free(_enum);
 }
 
This page took 0.027937 seconds and 4 git commands to generate.