Fix trace UST destroy channel
[lttng-tools.git] / lttng-sessiond / trace-ust.c
index a7c853530d0ab7db9906c588d7da72c9a4e3ced5..c85e79ce14fba897c280d1efaefa3d3602664cc0 100644 (file)
 #include <unistd.h>
 
 #include <lttngerr.h>
+#include <lttng-ht.h>
 #include <lttng-share.h>
 
-#include "hashtable.h"
 #include "trace-ust.h"
 
 /*
  * Find the channel in the hashtable.
  */
-struct ltt_ust_channel *trace_ust_find_channel_by_name(struct cds_lfht *ht,
+struct ltt_ust_channel *trace_ust_find_channel_by_name(struct lttng_ht *ht,
                char *name)
 {
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
+       struct lttng_ht_node_str *node;
+       struct lttng_ht_iter iter;
 
        rcu_read_lock();
-       node = hashtable_lookup(ht, (void *) name, strlen(name), &iter);
+       lttng_ht_lookup(ht, (void *)name, &iter);
+       node = lttng_ht_iter_get_node_str(&iter);
        if (node == NULL) {
                rcu_read_unlock();
                goto error;
@@ -56,25 +57,27 @@ error:
 /*
  * Find the event in the hashtable.
  */
-struct ltt_ust_event *trace_ust_find_event_by_name(struct cds_lfht *ht,
+struct ltt_ust_event *trace_ust_find_event_by_name(struct lttng_ht *ht,
                char *name)
 {
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
+       struct lttng_ht_node_str *node;
+       struct lttng_ht_iter iter;
 
        rcu_read_lock();
-       node = hashtable_lookup(ht, (void *) name, strlen(name), &iter);
+       lttng_ht_lookup(ht, (void *) name, &iter);
+       node = lttng_ht_iter_get_node_str(&iter);
        if (node == NULL) {
                rcu_read_unlock();
                goto error;
        }
        rcu_read_unlock();
 
-       DBG2("Found UST event by name %s", name);
+       DBG2("Trace UST event found by name %s", name);
 
        return caa_container_of(node, struct ltt_ust_event, node);
 
 error:
+       DBG2("Trace UST event NOT found by name %s", name);
        return NULL;
 }
 
@@ -83,30 +86,29 @@ error:
  *
  * Return pointer to structure or NULL.
  */
-struct ltt_ust_session *trace_ust_create_session(char *path, unsigned int uid,
+struct ltt_ust_session *trace_ust_create_session(char *path, int session_id,
                struct lttng_domain *domain)
 {
        int ret;
        struct ltt_ust_session *lus;
 
        /* Allocate a new ltt ust session */
-       lus = malloc(sizeof(struct ltt_ust_session));
+       lus = zmalloc(sizeof(struct ltt_ust_session));
        if (lus == NULL) {
-               PERROR("create ust session malloc");
+               PERROR("create ust session zmalloc");
                goto error;
        }
 
        /* Init data structure */
-       lus->consumer_fds_sent = 0;
-       lus->uid = uid;
+       lus->id = session_id;
        lus->start_trace = 0;
 
        /* Alloc UST domain hash tables */
-       lus->domain_pid = hashtable_new(0);
-       lus->domain_exec = hashtable_new_str(0);
+       lus->domain_pid = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
+       lus->domain_exec = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
 
        /* Alloc UST global domain channels' HT */
-       lus->domain_global.channels = hashtable_new_str(0);
+       lus->domain_global.channels = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
 
        /* Set session path */
        ret = snprintf(lus->pathname, PATH_MAX, "%s/ust", path);
@@ -136,9 +138,9 @@ struct ltt_ust_channel *trace_ust_create_channel(struct lttng_channel *chan,
        int ret;
        struct ltt_ust_channel *luc;
 
-       luc = malloc(sizeof(struct ltt_ust_channel));
+       luc = zmalloc(sizeof(struct ltt_ust_channel));
        if (luc == NULL) {
-               perror("ltt_ust_channel malloc");
+               perror("ltt_ust_channel zmalloc");
                goto error;
        }
 
@@ -158,14 +160,14 @@ struct ltt_ust_channel *trace_ust_create_channel(struct lttng_channel *chan,
        }
 
        /* Copy channel name */
-       strncpy(luc->name, chan->name, sizeof(&luc->name));
+       strncpy(luc->name, chan->name, sizeof(luc->name));
        luc->name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
 
        /* Init node */
-       hashtable_node_init(&luc->node, (void *) luc->name, strlen(luc->name));
+       lttng_ht_node_init_str(&luc->node, luc->name);
        /* Alloc hash tables */
-       luc->events = hashtable_new_str(0);
-       luc->ctx = hashtable_new_str(0);
+       luc->events = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
+       luc->ctx = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
 
        /* Set trace output path */
        ret = snprintf(luc->pathname, PATH_MAX, "%s", path);
@@ -193,9 +195,9 @@ struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev)
 {
        struct ltt_ust_event *lue;
 
-       lue = malloc(sizeof(struct ltt_ust_event));
+       lue = zmalloc(sizeof(struct ltt_ust_event));
        if (lue == NULL) {
-               PERROR("ust event malloc");
+               PERROR("ust event zmalloc");
                goto error;
        }
 
@@ -212,6 +214,9 @@ struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev)
        case LTTNG_EVENT_TRACEPOINT:
                lue->attr.instrumentation = LTTNG_UST_TRACEPOINT;
                break;
+       case LTTNG_EVENT_TRACEPOINT_LOGLEVEL:
+               lue->attr.instrumentation = LTTNG_UST_TRACEPOINT_LOGLEVEL;
+               break;
        default:
                ERR("Unknown ust instrumentation type (%d)", ev->type);
                goto error_free_event;
@@ -222,10 +227,11 @@ struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev)
        lue->attr.name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
 
        /* Init node */
-       hashtable_node_init(&lue->node, (void *) lue->attr.name,
-                       strlen(lue->attr.name));
+       lttng_ht_node_init_str(&lue->node, lue->attr.name);
        /* Alloc context hash tables */
-       lue->ctx = hashtable_new_str(0);
+       lue->ctx = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
+
+       DBG2("Trace UST event %s created", lue->attr.name);
 
        return lue;
 
@@ -245,9 +251,9 @@ struct ltt_ust_metadata *trace_ust_create_metadata(char *path)
        int ret;
        struct ltt_ust_metadata *lum;
 
-       lum = malloc(sizeof(struct ltt_ust_metadata));
+       lum = zmalloc(sizeof(struct ltt_ust_metadata));
        if (lum == NULL) {
-               perror("ust metadata malloc");
+               perror("ust metadata zmalloc");
                goto error;
        }
 
@@ -275,13 +281,38 @@ error:
        return NULL;
 }
 
+/*
+ * Allocate and initialize an UST context.
+ *
+ * Return pointer to structure or NULL.
+ */
+struct ltt_ust_context *trace_ust_create_context(
+               struct lttng_event_context *ctx)
+{
+       struct ltt_ust_context *uctx;
+
+       uctx = zmalloc(sizeof(struct ltt_ust_context));
+       if (uctx == NULL) {
+               PERROR("zmalloc ltt_ust_context");
+               goto error;
+       }
+
+       uctx->ctx.ctx = ctx->ctx;
+       lttng_ht_node_init_ulong(&uctx->node, (unsigned long) uctx->ctx.ctx);
+
+       return uctx;
+
+error:
+       return NULL;
+}
+
 /*
  * RCU safe free context structure.
  */
 static void destroy_context_rcu(struct rcu_head *head)
 {
-       struct cds_lfht_node *node =
-               caa_container_of(head, struct cds_lfht_node, head);
+       struct lttng_ht_node_ulong *node =
+               caa_container_of(head, struct lttng_ht_node_ulong, head);
        struct ltt_ust_context *ctx =
                caa_container_of(node, struct ltt_ust_context, node);
 
@@ -291,20 +322,20 @@ static void destroy_context_rcu(struct rcu_head *head)
 /*
  * Cleanup UST context hash table.
  */
-static void destroy_context(struct cds_lfht *ht)
+static void destroy_contexts(struct lttng_ht *ht)
 {
        int ret;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
+       struct lttng_ht_node_ulong *node;
+       struct lttng_ht_iter iter;
 
-       hashtable_get_first(ht, &iter);
-       while ((node = hashtable_iter_get_node(&iter)) != NULL) {
-               ret = hashtable_del(ht, &iter);
+       cds_lfht_for_each_entry(ht->ht, &iter.iter, node, node) {
+               ret = lttng_ht_del(ht, &iter);
                if (!ret) {
                        call_rcu(&node->head, destroy_context_rcu);
                }
-               hashtable_get_next(ht, &iter);
        }
+
+       lttng_ht_destroy(ht);
 }
 
 /*
@@ -313,7 +344,8 @@ static void destroy_context(struct cds_lfht *ht)
 void trace_ust_destroy_event(struct ltt_ust_event *event)
 {
        DBG2("Trace destroy UST event %s", event->attr.name);
-       destroy_context(event->ctx);
+       destroy_contexts(event->ctx);
+
        free(event);
 }
 
@@ -322,8 +354,8 @@ void trace_ust_destroy_event(struct ltt_ust_event *event)
  */
 static void destroy_event_rcu(struct rcu_head *head)
 {
-       struct cds_lfht_node *node =
-               caa_container_of(head, struct cds_lfht_node, head);
+       struct lttng_ht_node_str *node =
+               caa_container_of(head, struct lttng_ht_node_str, head);
        struct ltt_ust_event *event =
                caa_container_of(node, struct ltt_ust_event, node);
 
@@ -331,29 +363,37 @@ static void destroy_event_rcu(struct rcu_head *head)
 }
 
 /*
- * Cleanup ust channel structure.
+ * Cleanup UST events hashtable.
  */
-void trace_ust_destroy_channel(struct ltt_ust_channel *channel)
+static void destroy_events(struct lttng_ht *events)
 {
        int ret;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
+       struct lttng_ht_node_str *node;
+       struct lttng_ht_iter iter;
 
+       cds_lfht_for_each_entry(events->ht, &iter.iter, node, node) {
+               ret = lttng_ht_del(events, &iter);
+               assert(!ret);
+               call_rcu(&node->head, destroy_event_rcu);
+       }
+
+       lttng_ht_destroy(events);
+}
+
+/*
+ * Cleanup ust channel structure.
+ */
+void trace_ust_destroy_channel(struct ltt_ust_channel *channel)
+{
        DBG2("Trace destroy UST channel %s", channel->name);
 
        rcu_read_lock();
 
-       hashtable_get_first(channel->events, &iter);
-       while ((node = hashtable_iter_get_node(&iter)) != NULL) {
-               ret = hashtable_del(channel->events, &iter);
-               if (!ret) {
-                       call_rcu(&node->head, destroy_event_rcu);
-               }
-               hashtable_get_next(channel->events, &iter);
-       }
+       /* Destroying all events of the channel */
+       destroy_events(channel->events);
+       /* Destroying all context of the channel */
+       destroy_contexts(channel->ctx);
 
-       free(channel->events);
-       destroy_context(channel->ctx);
        free(channel);
 
        rcu_read_unlock();
@@ -364,8 +404,8 @@ void trace_ust_destroy_channel(struct ltt_ust_channel *channel)
  */
 static void destroy_channel_rcu(struct rcu_head *head)
 {
-       struct cds_lfht_node *node =
-               caa_container_of(head, struct cds_lfht_node, head);
+       struct lttng_ht_node_str *node =
+               caa_container_of(head, struct lttng_ht_node_str, head);
        struct ltt_ust_channel *channel =
                caa_container_of(node, struct ltt_ust_channel, node);
 
@@ -385,62 +425,59 @@ void trace_ust_destroy_metadata(struct ltt_ust_metadata *metadata)
 /*
  * Iterate over a hash table containing channels and cleanup safely.
  */
-static void destroy_channels(struct cds_lfht *channels)
+static void destroy_channels(struct lttng_ht *channels)
 {
        int ret;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
+       struct lttng_ht_node_str *node;
+       struct lttng_ht_iter iter;
 
-       hashtable_get_first(channels, &iter);
-       while ((node = hashtable_iter_get_node(&iter)) != NULL) {
-               ret = hashtable_del(channels, &iter);
-               if (!ret) {
-                       call_rcu(&node->head, destroy_channel_rcu);
-               }
-               hashtable_get_next(channels, &iter);
+       rcu_read_lock();
+
+       cds_lfht_for_each_entry(channels->ht, &iter.iter, node, node) {
+               ret = lttng_ht_del(channels, &iter);
+               assert(!ret);
+               call_rcu(&node->head, destroy_channel_rcu);
        }
+
+       lttng_ht_destroy(channels);
+
+       rcu_read_unlock();
 }
 
 /*
  * Cleanup UST pid domain.
  */
-static void destroy_domain_pid(struct cds_lfht *ht)
+static void destroy_domain_pid(struct lttng_ht *ht)
 {
        int ret;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
-       struct ltt_ust_domain_pid *d;
+       struct lttng_ht_iter iter;
+       struct ltt_ust_domain_pid *dpid;
 
-       hashtable_get_first(ht, &iter);
-       while ((node = hashtable_iter_get_node(&iter)) != NULL) {
-               ret = hashtable_del(ht , &iter);
-               if (!ret) {
-                       d = caa_container_of(node, struct ltt_ust_domain_pid, node);
-                       destroy_channels(d->channels);
-               }
-               hashtable_get_next(ht, &iter);
+       cds_lfht_for_each_entry(ht->ht, &iter.iter, dpid, node.node) {
+               ret = lttng_ht_del(ht , &iter);
+               assert(!ret);
+               destroy_channels(dpid->channels);
        }
+
+       lttng_ht_destroy(ht);
 }
 
 /*
  * Cleanup UST exec name domain.
  */
-static void destroy_domain_exec(struct cds_lfht *ht)
+static void destroy_domain_exec(struct lttng_ht *ht)
 {
        int ret;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
-       struct ltt_ust_domain_exec *d;
+       struct lttng_ht_iter iter;
+       struct ltt_ust_domain_exec *dexec;
 
-       hashtable_get_first(ht, &iter);
-       while ((node = hashtable_iter_get_node(&iter)) != NULL) {
-               ret = hashtable_del(ht , &iter);
-               if (!ret) {
-                       d = caa_container_of(node, struct ltt_ust_domain_exec, node);
-                       destroy_channels(d->channels);
-               }
-               hashtable_get_next(ht, &iter);
+       cds_lfht_for_each_entry(ht->ht, &iter.iter, dexec, node.node) {
+               ret = lttng_ht_del(ht , &iter);
+               assert(!ret);
+               destroy_channels(dexec->channels);
        }
+
+       lttng_ht_destroy(ht);
 }
 
 /*
@@ -456,18 +493,14 @@ static void destroy_domain_global(struct ltt_ust_domain_global *dom)
  */
 void trace_ust_destroy_session(struct ltt_ust_session *session)
 {
-       /* Extra safety */
+       /* Extra protection */
        if (session == NULL) {
                return;
        }
 
        rcu_read_lock();
 
-       DBG2("Trace UST destroy session %d", session->uid);
-
-       //if (session->metadata != NULL) {
-       //      trace_ust_destroy_metadata(session->metadata);
-       //}
+       DBG2("Trace UST destroy session %d", session->id);
 
        /* Cleaning up UST domain */
        destroy_domain_global(&session->domain_global);
This page took 0.029609 seconds and 4 git commands to generate.