Backport: Fix: tracker: ensure consistency of tracker states
[lttng-tools.git] / src / bin / lttng-sessiond / trace-ust.c
index b70d63673fd8ed9d275d39066a39be3dc2c04380..29263e7c8a3195e539b04e21d95bf1fa99a35671 100644 (file)
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
+ * Copyright (C) 2016 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License, version 2 only,
@@ -15,7 +16,6 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#define _GNU_SOURCE
 #define _LGPL_SOURCE
 #include <stdio.h>
 #include <stdlib.h>
@@ -31,6 +31,7 @@
 #include "utils.h"
 #include "ust-app.h"
 #include "agent.h"
+#include "tracker.h"
 
 /*
  * Match function for the events hash table lookup.
@@ -71,12 +72,15 @@ int trace_ust_ht_match_event(struct cds_lfht_node *node, const void *_key)
 {
        struct ltt_ust_event *event;
        const struct ltt_ust_ht_key *key;
+       int ev_loglevel_value;
+       int ll_match;
 
        assert(node);
        assert(_key);
 
        event = caa_container_of(node, struct ltt_ust_event, node.node);
        key = _key;
+       ev_loglevel_value = event->attr.loglevel;
 
        /* Match the 4 elements of the key: name, filter, loglevel, exclusions. */
 
@@ -85,19 +89,13 @@ int trace_ust_ht_match_event(struct cds_lfht_node *node, const void *_key)
                goto no_match;
        }
 
-       /* Event loglevel. */
-       if (event->attr.loglevel != key->loglevel) {
-               if (event->attr.loglevel_type == LTTNG_UST_LOGLEVEL_ALL
-                               && key->loglevel == 0 && event->attr.loglevel == -1) {
-                       /*
-                        * Match is accepted. This is because on event creation, the
-                        * loglevel is set to -1 if the event loglevel type is ALL so 0 and
-                        * -1 are accepted for this loglevel type since 0 is the one set by
-                        * the API when receiving an enable event.
-                        */
-               } else {
-                       goto no_match;
-               }
+       /* Event loglevel value and type. */
+       ll_match = loglevels_match(event->attr.loglevel_type,
+               ev_loglevel_value, key->loglevel_type,
+               key->loglevel_value, LTTNG_UST_LOGLEVEL_ALL);
+
+       if (!ll_match) {
+               goto no_match;
        }
 
        /* Only one of the filters is NULL, fail. */
@@ -120,12 +118,46 @@ int trace_ust_ht_match_event(struct cds_lfht_node *node, const void *_key)
        }
 
        if (key->exclusion && event->exclusion) {
-               /* Both exclusions exist; check count followed by names. */
-               if (event->exclusion->count != key->exclusion->count ||
-                               memcmp(event->exclusion->names, key->exclusion->names,
-                                       event->exclusion->count * LTTNG_SYMBOL_NAME_LEN) != 0) {
+               size_t i;
+
+               /* Check exclusion counts first. */
+               if (event->exclusion->count != key->exclusion->count) {
                        goto no_match;
                }
+
+               /* Compare names individually. */
+               for (i = 0; i < event->exclusion->count; ++i) {
+                       size_t j;
+                       bool found = false;
+                       const char *name_ev =
+                               LTTNG_EVENT_EXCLUSION_NAME_AT(
+                                       event->exclusion, i);
+
+                       /*
+                        * Compare this exclusion name to all the key's
+                        * exclusion names.
+                        */
+                       for (j = 0; j < key->exclusion->count; ++j) {
+                               const char *name_key =
+                                       LTTNG_EVENT_EXCLUSION_NAME_AT(
+                                               key->exclusion, j);
+
+                               if (!strncmp(name_ev, name_key,
+                                               LTTNG_SYMBOL_NAME_LEN)) {
+                                       /* Names match! */
+                                       found = true;
+                                       break;
+                               }
+                       }
+
+                       /*
+                        * If the current exclusion name was not found amongst
+                        * the key's exclusion names, then there's no match.
+                        */
+                       if (!found) {
+                               goto no_match;
+                       }
+               }
        }
        /* Match. */
        return 1;
@@ -171,7 +203,8 @@ error:
  * MUST be acquired before calling this.
  */
 struct ltt_ust_event *trace_ust_find_event(struct lttng_ht *ht,
-               char *name, struct lttng_filter_bytecode *filter, int loglevel,
+               char *name, struct lttng_filter_bytecode *filter,
+               enum lttng_ust_loglevel_type loglevel_type, int loglevel_value,
                struct lttng_event_exclusion *exclusion)
 {
        struct lttng_ht_node_str *node;
@@ -183,7 +216,8 @@ struct ltt_ust_event *trace_ust_find_event(struct lttng_ht *ht,
 
        key.name = name;
        key.filter = filter;
-       key.loglevel = loglevel;
+       key.loglevel_type = loglevel_type;
+       key.loglevel_value = loglevel_value;
        key.exclusion = exclusion;
 
        cds_lfht_lookup(ht->ht, ht->hash_fct((void *) name, lttng_ht_seed),
@@ -247,7 +281,7 @@ struct ltt_ust_session *trace_ust_create_session(uint64_t session_id)
        lus = zmalloc(sizeof(struct ltt_ust_session));
        if (lus == NULL) {
                PERROR("create ust session zmalloc");
-               goto error;
+               goto error_alloc;
        }
 
        /* Init data structure */
@@ -279,20 +313,35 @@ struct ltt_ust_session *trace_ust_create_session(uint64_t session_id)
        /* Alloc agent hash table. */
        lus->agents = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
 
+       lus->tracker_list_vpid = lttng_tracker_list_create();
+       if (!lus->tracker_list_vpid) {
+               goto error;
+       }
+       lus->tracker_list_vuid = lttng_tracker_list_create();
+       if (!lus->tracker_list_vuid) {
+               goto error;
+       }
+       lus->tracker_list_vgid = lttng_tracker_list_create();
+       if (!lus->tracker_list_vgid) {
+               goto error;
+       }
        lus->consumer = consumer_create_output(CONSUMER_DST_LOCAL);
        if (lus->consumer == NULL) {
-               goto error_consumer;
+               goto error;
        }
 
        DBG2("UST trace session create successful");
 
        return lus;
 
-error_consumer:
+error:
+       lttng_tracker_list_destroy(lus->tracker_list_vpid);
+       lttng_tracker_list_destroy(lus->tracker_list_vuid);
+       lttng_tracker_list_destroy(lus->tracker_list_vgid);
        ht_cleanup_push(lus->domain_global.channels);
        ht_cleanup_push(lus->agents);
        free(lus);
-error:
+error_alloc:
        return NULL;
 }
 
@@ -361,6 +410,39 @@ error:
        return luc;
 }
 
+/*
+ * Validates an exclusion list.
+ *
+ * Returns 0 if valid, negative value if invalid.
+ */
+static int validate_exclusion(struct lttng_event_exclusion *exclusion)
+{
+       size_t i;
+       int ret = 0;
+
+       assert(exclusion);
+
+       for (i = 0; i < exclusion->count; ++i) {
+               size_t j;
+               const char *name_a =
+                       LTTNG_EVENT_EXCLUSION_NAME_AT(exclusion, i);
+
+               for (j = 0; j < i; ++j) {
+                       const char *name_b =
+                               LTTNG_EVENT_EXCLUSION_NAME_AT(exclusion, j);
+
+                       if (!strncmp(name_a, name_b, LTTNG_SYMBOL_NAME_LEN)) {
+                               /* Match! */
+                               ret = -1;
+                               goto end;
+                       }
+               }
+       }
+
+end:
+       return ret;
+}
+
 /*
  * Allocate and initialize a ust event. Set name and event type.
  * We own filter_expression, filter, and exclusion.
@@ -377,6 +459,10 @@ struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev,
 
        assert(ev);
 
+       if (exclusion && validate_exclusion(exclusion)) {
+               goto error;
+       }
+
        lue = zmalloc(sizeof(struct ltt_ust_event));
        if (lue == NULL) {
                PERROR("ust event zmalloc");
@@ -449,7 +535,8 @@ error:
 }
 
 static
-int trace_ust_context_type_event_to_ust(enum lttng_event_context_type type)
+int trace_ust_context_type_event_to_ust(
+               enum lttng_event_context_type type)
 {
        int utype;
 
@@ -477,8 +564,10 @@ int trace_ust_context_type_event_to_ust(enum lttng_event_context_type type)
                        utype = LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER;
                }
                break;
+       case LTTNG_EVENT_CONTEXT_APP_CONTEXT:
+               utype = LTTNG_UST_CONTEXT_APP_CONTEXT;
+               break;
        default:
-               ERR("Invalid UST context");
                utype = -1;
                break;
        }
@@ -516,6 +605,15 @@ int trace_ust_match_context(struct ltt_ust_context *uctx,
                        return 0;
                }
                break;
+       case LTTNG_UST_CONTEXT_APP_CONTEXT:
+               assert(uctx->ctx.u.app_ctx.provider_name);
+               assert(uctx->ctx.u.app_ctx.ctx_name);
+               if (strcmp(uctx->ctx.u.app_ctx.provider_name,
+                               ctx->u.app_ctx.provider_name) ||
+                               strcmp(uctx->ctx.u.app_ctx.ctx_name,
+                               ctx->u.app_ctx.ctx_name)) {
+                       return 0;
+               }
        default:
                break;
 
@@ -531,20 +629,21 @@ int trace_ust_match_context(struct ltt_ust_context *uctx,
 struct ltt_ust_context *trace_ust_create_context(
                struct lttng_event_context *ctx)
 {
-       struct ltt_ust_context *uctx;
+       struct ltt_ust_context *uctx = NULL;
        int utype;
 
        assert(ctx);
 
        utype = trace_ust_context_type_event_to_ust(ctx->ctx);
        if (utype < 0) {
-               return NULL;
+               ERR("Invalid UST context");
+               goto end;
        }
 
        uctx = zmalloc(sizeof(struct ltt_ust_context));
-       if (uctx == NULL) {
+       if (!uctx) {
                PERROR("zmalloc ltt_ust_context");
-               goto error;
+               goto end;
        }
 
        uctx->ctx.ctx = (enum lttng_ust_context_type) utype;
@@ -556,40 +655,57 @@ struct ltt_ust_context *trace_ust_create_context(
                                LTTNG_UST_SYM_NAME_LEN);
                uctx->ctx.u.perf_counter.name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
                break;
+       case LTTNG_UST_CONTEXT_APP_CONTEXT:
+       {
+               char *provider_name = NULL, *ctx_name = NULL;
+
+               provider_name = strdup(ctx->u.app_ctx.provider_name);
+               if (!provider_name) {
+                       goto error;
+               }
+               uctx->ctx.u.app_ctx.provider_name = provider_name;
+
+               ctx_name = strdup(ctx->u.app_ctx.ctx_name);
+               if (!ctx_name) {
+                       goto error;
+               }
+               uctx->ctx.u.app_ctx.ctx_name = ctx_name;
+               break;
+       }
        default:
                break;
        }
        lttng_ht_node_init_ulong(&uctx->node, (unsigned long) uctx->ctx.ctx);
-
+end:
        return uctx;
-
 error:
+       trace_ust_destroy_context(uctx);
        return NULL;
 }
 
 static
-void destroy_pid_tracker_node_rcu(struct rcu_head *head)
+void destroy_id_tracker_node_rcu(struct rcu_head *head)
 {
-       struct ust_pid_tracker_node *tracker_node =
-               caa_container_of(head, struct ust_pid_tracker_node, node.head);
+       struct ust_id_tracker_node *tracker_node =
+               caa_container_of(head, struct ust_id_tracker_node, node.head);
        free(tracker_node);
 }
 
 static
-void destroy_pid_tracker_node(struct ust_pid_tracker_node *tracker_node)
+void destroy_id_tracker_node(struct ust_id_tracker_node *tracker_node)
 {
 
-       call_rcu(&tracker_node->node.head, destroy_pid_tracker_node_rcu);
+       call_rcu(&tracker_node->node.head, destroy_id_tracker_node_rcu);
 }
 
 static
-int init_pid_tracker(struct ust_pid_tracker *pid_tracker)
+int init_id_tracker(struct ust_id_tracker *id_tracker)
 {
-       int ret = 0;
+       int ret = LTTNG_OK;
 
-       pid_tracker->ht = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
-       if (!pid_tracker->ht) {
-               ret = -1;
+       id_tracker->ht = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
+       if (!id_tracker->ht) {
+               ret = LTTNG_ERR_NOMEM;
                goto end;
        }
 
@@ -598,42 +714,42 @@ end:
 }
 
 /*
- * Teardown pid tracker content, but don't free pid_tracker object.
+ * Teardown id tracker content, but don't free id_tracker object.
  */
 static
-void fini_pid_tracker(struct ust_pid_tracker *pid_tracker)
+void fini_id_tracker(struct ust_id_tracker *id_tracker)
 {
-       struct ust_pid_tracker_node *tracker_node;
+       struct ust_id_tracker_node *tracker_node;
        struct lttng_ht_iter iter;
 
-       if (!pid_tracker->ht) {
+       if (!id_tracker->ht) {
                return;
        }
        rcu_read_lock();
-       cds_lfht_for_each_entry(pid_tracker->ht->ht,
+       cds_lfht_for_each_entry(id_tracker->ht->ht,
                        &iter.iter, tracker_node, node.node) {
-               int ret = lttng_ht_del(pid_tracker->ht, &iter);
+               int ret = lttng_ht_del(id_tracker->ht, &iter);
 
                assert(!ret);
-               destroy_pid_tracker_node(tracker_node);
+               destroy_id_tracker_node(tracker_node);
        }
        rcu_read_unlock();
-       ht_cleanup_push(pid_tracker->ht);
-       pid_tracker->ht = NULL;
+       ht_cleanup_push(id_tracker->ht);
+       id_tracker->ht = NULL;
 }
 
 static
-struct ust_pid_tracker_node *pid_tracker_lookup(
-               struct ust_pid_tracker *pid_tracker, int pid,
+struct ust_id_tracker_node *id_tracker_lookup(
+               struct ust_id_tracker *id_tracker, int id,
                struct lttng_ht_iter *iter)
 {
-       unsigned long _pid = (unsigned long) pid;
+       unsigned long _id = (unsigned long) id;
        struct lttng_ht_node_ulong *node;
 
-       lttng_ht_lookup(pid_tracker->ht, (void *) _pid, iter);
+       lttng_ht_lookup(id_tracker->ht, (void *) _id, iter);
        node = lttng_ht_iter_get_node_ulong(iter);
        if (node) {
-               return caa_container_of(node, struct ust_pid_tracker_node,
+               return caa_container_of(node, struct ust_id_tracker_node,
                        node);
        } else {
                return NULL;
@@ -641,20 +757,20 @@ struct ust_pid_tracker_node *pid_tracker_lookup(
 }
 
 static
-int pid_tracker_add_pid(struct ust_pid_tracker *pid_tracker, int pid)
+int id_tracker_add_id(struct ust_id_tracker *id_tracker, int id)
 {
        int retval = LTTNG_OK;
-       struct ust_pid_tracker_node *tracker_node;
+       struct ust_id_tracker_node *tracker_node;
        struct lttng_ht_iter iter;
 
-       if (pid < 0) {
+       if (id < 0) {
                retval = LTTNG_ERR_INVALID;
                goto end;
        }
-       tracker_node = pid_tracker_lookup(pid_tracker, pid, &iter);
+       tracker_node = id_tracker_lookup(id_tracker, id, &iter);
        if (tracker_node) {
                /* Already exists. */
-               retval = LTTNG_ERR_PID_TRACKED;
+               retval = LTTNG_ERR_ID_TRACKED;
                goto end;
        }
        tracker_node = zmalloc(sizeof(*tracker_node));
@@ -662,48 +778,86 @@ int pid_tracker_add_pid(struct ust_pid_tracker *pid_tracker, int pid)
                retval = LTTNG_ERR_NOMEM;
                goto end;
        }
-       lttng_ht_node_init_ulong(&tracker_node->node, (unsigned long) pid);
-       lttng_ht_add_unique_ulong(pid_tracker->ht, &tracker_node->node);
+       lttng_ht_node_init_ulong(&tracker_node->node, (unsigned long) id);
+       lttng_ht_add_unique_ulong(id_tracker->ht, &tracker_node->node);
 end:
        return retval;
 }
 
 static
-int pid_tracker_del_pid(struct ust_pid_tracker *pid_tracker, int pid)
+int id_tracker_del_id(struct ust_id_tracker *id_tracker, int id)
 {
        int retval = LTTNG_OK, ret;
-       struct ust_pid_tracker_node *tracker_node;
+       struct ust_id_tracker_node *tracker_node;
        struct lttng_ht_iter iter;
 
-       if (pid < 0) {
+       if (id < 0) {
                retval = LTTNG_ERR_INVALID;
                goto end;
        }
-       tracker_node = pid_tracker_lookup(pid_tracker, pid, &iter);
+       tracker_node = id_tracker_lookup(id_tracker, id, &iter);
        if (!tracker_node) {
                /* Not found */
-               retval = LTTNG_ERR_PID_NOT_TRACKED;
+               retval = LTTNG_ERR_ID_NOT_TRACKED;
                goto end;
        }
-       ret = lttng_ht_del(pid_tracker->ht, &iter);
+       ret = lttng_ht_del(id_tracker->ht, &iter);
        assert(!ret);
 
-       destroy_pid_tracker_node(tracker_node);
+       destroy_id_tracker_node(tracker_node);
 end:
        return retval;
 }
 
+static
+struct ust_id_tracker *get_id_tracker(struct ltt_ust_session *session,
+               enum lttng_tracker_type tracker_type)
+{
+       switch (tracker_type) {
+       case LTTNG_TRACKER_VPID:
+               return &session->vpid_tracker;
+       case LTTNG_TRACKER_VUID:
+               return &session->vuid_tracker;
+       case LTTNG_TRACKER_VGID:
+               return &session->vgid_tracker;
+       default:
+               return NULL;
+       }
+}
+
+static
+struct lttng_tracker_list *get_id_tracker_list(struct ltt_ust_session *session,
+               enum lttng_tracker_type tracker_type)
+{
+       switch (tracker_type) {
+       case LTTNG_TRACKER_VPID:
+               return session->tracker_list_vpid;
+       case LTTNG_TRACKER_VUID:
+               return session->tracker_list_vuid;
+       case LTTNG_TRACKER_VGID:
+               return session->tracker_list_vgid;
+       default:
+               return NULL;
+       }
+}
+
 /*
  * The session lock is held when calling this function.
  */
-int trace_ust_pid_tracker_lookup(struct ltt_ust_session *session, int pid)
+int trace_ust_id_tracker_lookup(enum lttng_tracker_type tracker_type,
+               struct ltt_ust_session *session, int id)
 {
        struct lttng_ht_iter iter;
+       struct ust_id_tracker *id_tracker;
 
-       if (!session->pid_tracker.ht) {
+       id_tracker = get_id_tracker(session, tracker_type);
+       if (!id_tracker) {
+               abort();
+       }
+       if (!id_tracker->ht) {
                return 1;
        }
-       if (pid_tracker_lookup(&session->pid_tracker, pid, &iter)) {
+       if (id_tracker_lookup(id_tracker, id, &iter)) {
                return 1;
        }
        return 0;
@@ -712,143 +866,222 @@ int trace_ust_pid_tracker_lookup(struct ltt_ust_session *session, int pid)
 /*
  * Called with the session lock held.
  */
-int trace_ust_track_pid(struct ltt_ust_session *session, int pid)
+int trace_ust_track_id(enum lttng_tracker_type tracker_type,
+               struct ltt_ust_session *session,
+               struct lttng_tracker_id *id)
 {
        int retval = LTTNG_OK;
+       struct ust_id_tracker *id_tracker;
+       struct lttng_tracker_list *tracker_list;
+       int value;
+       struct lttng_tracker_id *saved_ids;
+       ssize_t saved_ids_count, i;
+
+       if (tracker_type == LTTNG_TRACKER_PID) {
+               DBG("Backward compatible behavior: translate PID tracker to VPID tracker for UST domain.");
+               tracker_type = LTTNG_TRACKER_VPID;
+       }
 
-       if (pid == -1) {
-               /* Track all pids: destroy tracker if exists. */
-               if (session->pid_tracker.ht) {
-                       fini_pid_tracker(&session->pid_tracker);
+       retval = lttng_tracker_id_lookup_string(tracker_type,
+                       id, &value);
+       if (retval != LTTNG_OK) {
+               return retval;
+       }
+       tracker_list = get_id_tracker_list(session, tracker_type);
+       if (!tracker_list) {
+               return LTTNG_ERR_INVALID;
+       }
+       /* Save list for restore on error. */
+       saved_ids_count = lttng_tracker_id_get_list(tracker_list, &saved_ids);
+       if (saved_ids_count < 0) {
+               return LTTNG_ERR_INVALID;
+       }
+       /* Add to list. */
+       retval = lttng_tracker_list_add(tracker_list, id);
+       if (retval != LTTNG_OK) {
+               goto end;
+       }
+
+       id_tracker = get_id_tracker(session, tracker_type);
+       if (!id_tracker) {
+               abort();
+       }
+       if (value == -1) {
+               /* Track all ids: destroy tracker if exists. */
+               if (id_tracker->ht) {
+                       fini_id_tracker(id_tracker);
                        /* Ensure all apps have session. */
                        ust_app_global_update_all(session);
                }
        } else {
-               int ret;
-
-               if (!session->pid_tracker.ht) {
+               if (!id_tracker->ht) {
                        /* Create tracker. */
-                       if (init_pid_tracker(&session->pid_tracker)) {
-                               ERR("Error initializing PID tracker");
-                               retval = LTTNG_ERR_NOMEM;
-                               goto end;
+                       retval = init_id_tracker(id_tracker);
+                       if (retval != LTTNG_OK) {
+                               ERR("Error initializing ID tracker");
+                               goto end_restore;
                        }
-                       ret = pid_tracker_add_pid(&session->pid_tracker, pid);
-                       if (ret != LTTNG_OK) {
-                               retval = ret;
-                               fini_pid_tracker(&session->pid_tracker);
-                               goto end;
+                       retval = id_tracker_add_id(id_tracker, value);
+                       if (retval != LTTNG_OK) {
+                               fini_id_tracker(id_tracker);
+                               goto end_restore;
                        }
-                       /* Remove all apps from session except pid. */
+                       /* Keep only apps matching ID. */
                        ust_app_global_update_all(session);
                } else {
                        struct ust_app *app;
 
-                       ret = pid_tracker_add_pid(&session->pid_tracker, pid);
-                       if (ret != LTTNG_OK) {
-                               retval = ret;
-                               goto end;
+                       retval = id_tracker_add_id(id_tracker, value);
+                       if (retval != LTTNG_OK) {
+                               goto end_restore;
                        }
                        /* Add session to application */
-                       app = ust_app_find_by_pid(pid);
-                       if (app) {
-                               ust_app_global_update(session, app);
+                       switch (tracker_type) {
+                       case LTTNG_TRACKER_VPID:
+                               app = ust_app_find_by_pid(value);
+                               if (app) {
+                                       ust_app_global_update(session, app);
+                               }
+                               break;
+                       default:
+                               /* Keep only apps matching ID. */
+                               ust_app_global_update_all(session);
                        }
                }
        }
+       goto end;
+
+end_restore:
+       if (lttng_tracker_id_set_list(tracker_list, saved_ids, saved_ids_count) != LTTNG_OK) {
+               ERR("Error on tracker add error handling.\n");
+       }
 end:
+       for (i = 0; i < saved_ids_count; i++) {
+               free(saved_ids[i].string);
+       }
+       free(saved_ids);
        return retval;
 }
 
 /*
  * Called with the session lock held.
  */
-int trace_ust_untrack_pid(struct ltt_ust_session *session, int pid)
+int trace_ust_untrack_id(enum lttng_tracker_type tracker_type,
+               struct ltt_ust_session *session, struct lttng_tracker_id *id)
 {
        int retval = LTTNG_OK;
+       struct ust_id_tracker *id_tracker;
+       struct lttng_tracker_list *tracker_list;
+       int value;
+       struct lttng_tracker_id *saved_ids;
+       ssize_t saved_ids_count, i;
+
+       if (tracker_type == LTTNG_TRACKER_PID) {
+               DBG("Backward compatible behavior: translate PID tracker to VPID tracker for UST domain.");
+               tracker_type = LTTNG_TRACKER_VPID;
+       }
 
-       if (pid == -1) {
+       retval = lttng_tracker_id_lookup_string(tracker_type,
+                       id, &value);
+       if (retval != LTTNG_OK) {
+               return retval;
+       }
+
+       tracker_list = get_id_tracker_list(session, tracker_type);
+       if (!tracker_list) {
+               return LTTNG_ERR_INVALID;
+       }
+       /* Save list for restore on error. */
+       saved_ids_count = lttng_tracker_id_get_list(tracker_list, &saved_ids);
+       if (saved_ids_count < 0) {
+               return LTTNG_ERR_INVALID;
+       }
+       /* Remove from list. */
+       retval = lttng_tracker_list_remove(tracker_list, id);
+       if (retval != LTTNG_OK) {
+               goto end;
+       }
+
+       id_tracker = get_id_tracker(session, tracker_type);
+       if (!id_tracker) {
+               abort();
+       }
+
+       if (value == -1) {
                /* Create empty tracker, replace old tracker. */
-               struct ust_pid_tracker tmp_tracker;
+               struct ust_id_tracker tmp_tracker;
 
-               tmp_tracker = session->pid_tracker;
-               if (init_pid_tracker(&session->pid_tracker)) {
-                       ERR("Error initializing PID tracker");
-                       retval = LTTNG_ERR_NOMEM;
+               tmp_tracker = *id_tracker;
+               retval = init_id_tracker(id_tracker);
+               if (retval != LTTNG_OK) {
+                       ERR("Error initializing ID tracker");
                        /* Rollback operation. */
-                       session->pid_tracker = tmp_tracker;
-                       goto end;
+                       *id_tracker = tmp_tracker;
+                       goto end_restore;
                }
-               fini_pid_tracker(&tmp_tracker);
+               fini_id_tracker(&tmp_tracker);
 
-               /* Remove session from all applications */
+               /* Keep only apps matching ID. */
                ust_app_global_update_all(session);
        } else {
-               int ret;
                struct ust_app *app;
 
-               if (!session->pid_tracker.ht) {
-                       retval = LTTNG_ERR_INVALID;
-                       goto end;
+               if (!id_tracker->ht) {
+                       /* No ID being tracked. */
+                       retval = LTTNG_ERR_ID_NOT_TRACKED;
+                       goto end_restore;
                }
-               /* Remove PID from tracker */
-               ret = pid_tracker_del_pid(&session->pid_tracker, pid);
-               if (ret != LTTNG_OK) {
-                       retval = ret;
-                       goto end;
+               /* Remove ID from tracker */
+               retval = id_tracker_del_id(id_tracker, value);
+               if (retval != LTTNG_OK) {
+                       goto end_restore;
                }
-               /* Remove session from application. */
-               app = ust_app_find_by_pid(pid);
-               if (app) {
-                       ust_app_global_update(session, app);
+               switch (tracker_type) {
+               case LTTNG_TRACKER_VPID:
+                       /* Remove session from application. */
+                       app = ust_app_find_by_pid(value);
+                       if (app) {
+                               ust_app_global_update(session, app);
+                       }
+                       break;
+               default:
+                       /* Keep only apps matching ID. */
+                       ust_app_global_update_all(session);
                }
        }
+       goto end;
+
+end_restore:
+       if (lttng_tracker_id_set_list(tracker_list, saved_ids, saved_ids_count) != LTTNG_OK) {
+               ERR("Error on tracker remove error handling.\n");
+       }
 end:
+       for (i = 0; i < saved_ids_count; i++) {
+               free(saved_ids[i].string);
+       }
+       free(saved_ids);
        return retval;
 }
 
 /*
  * Called with session lock held.
  */
-ssize_t trace_ust_list_tracker_pids(struct ltt_ust_session *session,
-               int32_t **_pids)
+ssize_t trace_ust_list_tracker_ids(enum lttng_tracker_type tracker_type,
+               struct ltt_ust_session *session,
+               struct lttng_tracker_id **_ids)
 {
-       struct ust_pid_tracker_node *tracker_node;
-       struct lttng_ht_iter iter;
-       unsigned long count, i = 0;
-       long approx[2];
-       int32_t *pids;
-       int ret = 0;
+       struct lttng_tracker_list *tracker_list;
 
-       if (!session->pid_tracker.ht) {
-               /* Tracker disabled. Set first entry to -1. */
-               pids = zmalloc(sizeof(*pids));
-               if (!pids) {
-                       ret = -1;
-                       goto end;
-               }
-               pids[0] = -1;
-               *_pids = pids;
-               return 1;
+       if (tracker_type == LTTNG_TRACKER_PID) {
+               DBG("Backward compatible behavior: translate PID tracker to VPID tracker for UST domain.");
+               tracker_type = LTTNG_TRACKER_VPID;
        }
 
-       rcu_read_lock();
-       cds_lfht_count_nodes(session->pid_tracker.ht->ht,
-               &approx[0], &count, &approx[1]);
-       pids = zmalloc(sizeof(*pids) * count);
-       if (!pids) {
-               ret = -1;
-               goto end;
-       }
-       cds_lfht_for_each_entry(session->pid_tracker.ht->ht,
-                       &iter.iter, tracker_node, node.node) {
-               pids[i++] = tracker_node->node.key;
+       tracker_list = get_id_tracker_list(session, tracker_type);
+       if (!tracker_list) {
+               return -LTTNG_ERR_INVALID;
        }
-       *_pids = pids;
-       ret = count;
-end:
-       rcu_read_unlock();
-       return ret;
+       return lttng_tracker_id_get_list(tracker_list, _ids);
 }
 
 /*
@@ -861,7 +1094,7 @@ static void destroy_context_rcu(struct rcu_head *head)
        struct ltt_ust_context *ctx =
                caa_container_of(node, struct ltt_ust_context, node);
 
-       free(ctx);
+       trace_ust_destroy_context(ctx);
 }
 
 /*
@@ -906,6 +1139,20 @@ void trace_ust_destroy_event(struct ltt_ust_event *event)
        free(event);
 }
 
+/*
+ * Cleanup ust context structure.
+ */
+void trace_ust_destroy_context(struct ltt_ust_context *ctx)
+{
+       assert(ctx);
+
+       if (ctx->ctx.ctx == LTTNG_UST_CONTEXT_APP_CONTEXT) {
+               free(ctx->ctx.u.app_ctx.provider_name);
+               free(ctx->ctx.u.app_ctx.ctx_name);
+       }
+       free(ctx);
+}
+
 /*
  * URCU intermediate call to complete destroy event.
  */
@@ -1067,7 +1314,13 @@ void trace_ust_destroy_session(struct ltt_ust_session *session)
 
        consumer_output_put(session->consumer);
 
-       fini_pid_tracker(&session->pid_tracker);
+       lttng_tracker_list_destroy(session->tracker_list_vpid);
+       lttng_tracker_list_destroy(session->tracker_list_vuid);
+       lttng_tracker_list_destroy(session->tracker_list_vgid);
+
+       fini_id_tracker(&session->vpid_tracker);
+       fini_id_tracker(&session->vuid_tracker);
+       fini_id_tracker(&session->vgid_tracker);
 
        free(session);
 }
This page took 0.0362749999999999 seconds and 5 git commands to generate.