Backport: trackers: update list/track/untrack commands
authorMathieu Desnoyers <mathieu.desnoyers@efficios.com>
Wed, 4 Jul 2018 20:17:48 +0000 (16:17 -0400)
committerJonathan Rajotte <jonathan.rajotte-julien@efficios.com>
Fri, 21 Sep 2018 04:00:52 +0000 (00:00 -0400)
Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
src/bin/lttng/commands/list.c
src/bin/lttng/commands/track-untrack.c

index 8c6d1ef4afe5a7dc016957026b2258efb589e535..e6f52f847868da74e7fcf2993cc530434730dc25 100644 (file)
@@ -718,7 +718,7 @@ static int mi_list_ust_event_fields(struct lttng_event_field *fields, int count,
                }
        }
 
-       /* Close pid, domain, domains */
+       /* Close pids, domain, domains */
        ret = mi_lttng_close_multi_element(writer, 3);
 end:
        return ret;
@@ -1346,43 +1346,80 @@ error_channels:
        return ret;
 }
 
+static
+const char *get_tracker_str(enum lttng_tracker_type tracker_type)
+{
+       switch (tracker_type) {
+       case LTTNG_TRACKER_PID:
+               return "PID";
+       case LTTNG_TRACKER_VPID:
+               return "VPID";
+       case LTTNG_TRACKER_UID:
+               return "UID";
+       case LTTNG_TRACKER_VUID:
+               return "VUID";
+       case LTTNG_TRACKER_GID:
+               return "GID";
+       case LTTNG_TRACKER_VGID:
+               return "VGID";
+       }
+       return NULL;
+}
+
 /*
- * List tracker PID(s) of session and domain.
+ * List tracker ID(s) of session and domain.
  */
-static int list_tracker_pids(void)
+static int list_tracker_ids(enum lttng_tracker_type tracker_type)
 {
        int ret = 0;
-       int enabled;
-       int *pids = NULL;
-       size_t nr_pids;
+       int enabled = 1;
+       struct lttng_tracker_id *ids = NULL;
+       size_t nr_ids, i;
 
-       ret = lttng_list_tracker_pids(handle,
-               &enabled, &pids, &nr_pids);
+       ret = lttng_list_tracker_ids(handle, tracker_type,
+               &ids, &nr_ids);
        if (ret) {
                return ret;
        }
+       if (nr_ids == 1 && ids[0].type == LTTNG_ID_ALL) {
+               enabled = 0;
+       }
        if (enabled) {
-               int i;
-               _MSG("PID tracker: [");
+               _MSG("%s tracker: [", get_tracker_str(tracker_type));
 
-               /* Mi tracker_pid element*/
+               /* Mi tracker_id element */
                if (writer) {
-                       /* Open tracker_pid and targets elements */
-                       ret = mi_lttng_pid_tracker_open(writer);
+                       /* Open tracker_id and targets elements */
+                       ret = mi_lttng_id_tracker_open(writer, tracker_type);
                        if (ret) {
                                goto end;
                        }
                }
 
-               for (i = 0; i < nr_pids; i++) {
+               for (i = 0; i < nr_ids; i++) {
+                       struct lttng_tracker_id *id = &ids[i];
+
                        if (i) {
                                _MSG(",");
                        }
-                       _MSG(" %d", pids[i]);
+                       switch (id->type) {
+                       case LTTNG_ID_ALL:
+                               _MSG(" *");
+                               break;
+                       case LTTNG_ID_VALUE:
+                               _MSG(" %d", ids[i].value);
+                               break;
+                       case LTTNG_ID_STRING:
+                               _MSG(" %s", ids[i].string);
+                               break;
+                       case LTTNG_ID_UNKNOWN:
+                               return CMD_ERROR;
+                       }
 
                        /* Mi */
                        if (writer) {
-                               ret = mi_lttng_pid_target(writer, pids[i], 0);
+                               ret = mi_lttng_id_target(writer,
+                                       tracker_type, id, 0);
                                if (ret) {
                                        goto end;
                                }
@@ -1390,24 +1427,26 @@ static int list_tracker_pids(void)
                }
                _MSG(" ]\n\n");
 
-               /* Mi close tracker_pid and targets */
+               /* Mi close tracker_id and targets */
                if (writer) {
-                       ret = mi_lttng_close_multi_element(writer,2);
+                       ret = mi_lttng_close_multi_element(writer, 2);
                        if (ret) {
                                goto end;
                        }
                }
        }
 end:
-       free(pids);
+       for (i = 0; i < nr_ids; i++) {
+               free(ids[i].string);
+       }
+       free(ids);
        return ret;
-
 }
 
 /*
- * List all tracker of a domain
+ * List all trackers of a domain
  */
-static int list_trackers(void)
+static int list_trackers(struct lttng_domain *domain)
 {
        int ret;
 
@@ -1419,12 +1458,59 @@ static int list_trackers(void)
                }
        }
 
-       /* pid tracker */
-       ret = list_tracker_pids();
-       if (ret) {
-               goto end;
+       switch (domain->type) {
+       case LTTNG_DOMAIN_KERNEL:
+               /* pid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_PID);
+               if (ret) {
+                       goto end;
+               }
+               /* vpid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VPID);
+               if (ret) {
+                       goto end;
+               }
+               /* uid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_UID);
+               if (ret) {
+                       goto end;
+               }
+               /* vuid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VUID);
+               if (ret) {
+                       goto end;
+               }
+               /* gid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_GID);
+               if (ret) {
+                       goto end;
+               }
+               /* vgid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VGID);
+               if (ret) {
+                       goto end;
+               }
+               break;
+       case LTTNG_DOMAIN_UST:
+               /* vpid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VPID);
+               if (ret) {
+                       goto end;
+               }
+               /* vuid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VUID);
+               if (ret) {
+                       goto end;
+               }
+               /* vgid tracker */
+               ret = list_tracker_ids(LTTNG_TRACKER_VGID);
+               if (ret) {
+                       goto end;
+               }
+               break;
+       default:
+               break;
        }
-
        if (lttng_opt_mi) {
                /* Close trackers element */
                ret = mi_lttng_writer_close_element(writer);
@@ -1868,7 +1954,7 @@ int cmd_list(int argc, const char **argv)
 
 
                        /* Trackers */
-                       ret = list_trackers();
+                       ret = list_trackers(&domain);
                        if (ret) {
                                goto end;
                        }
@@ -1965,7 +2051,7 @@ int cmd_list(int argc, const char **argv)
                                switch (domains[i].type) {
                                case LTTNG_DOMAIN_KERNEL:
                                case LTTNG_DOMAIN_UST:
-                                       ret = list_trackers();
+                                       ret = list_trackers(&domains[i]);
                                        if (ret) {
                                                goto end;
                                        }
index f5580544361f85d0f84d9669b228c677f89431a7..b5de1bee0869ebaad1deff5e60ba0f59b16d6fda 100644 (file)
@@ -38,18 +38,46 @@ enum cmd_type {
        CMD_UNTRACK,
 };
 
+enum tracker_type_state {
+       STATE_NONE = 0,
+       STATE_PID,
+       STATE_VPID,
+       STATE_UID,
+       STATE_VUID,
+       STATE_GID,
+       STATE_VGID,
+};
+
+struct opt_type {
+       int used;
+       int all;
+       char *string;
+};
+
+struct id_list {
+       size_t nr;
+       struct lttng_tracker_id *array;
+};
+
 static char *opt_session_name;
 static int opt_kernel;
 static int opt_userspace;
-static int opt_all;
-static char *opt_pid_string;
-static int opt_pid;
+
+static struct opt_type opt_pid, opt_vpid, opt_uid, opt_vuid, opt_gid, opt_vgid;
+
+static enum tracker_type_state type_state;
 
 enum {
        OPT_HELP = 1,
        OPT_LIST_OPTIONS,
        OPT_SESSION,
        OPT_PID,
+       OPT_VPID,
+       OPT_UID,
+       OPT_VUID,
+       OPT_GID,
+       OPT_VGID,
+       OPT_ALL,
 };
 
 static struct poptOption long_options[] = {
@@ -58,162 +86,251 @@ static struct poptOption long_options[] = {
        { "session",            's', POPT_ARG_STRING, &opt_session_name, OPT_SESSION, 0, 0, },
        { "kernel",             'k', POPT_ARG_VAL, &opt_kernel, 1, 0, 0, },
        { "userspace",          'u', POPT_ARG_VAL, &opt_userspace, 1, 0, 0, },
-       { "pid",                'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_pid_string, OPT_PID, 0, 0, },
-       { "all",                'a', POPT_ARG_VAL, &opt_all, 1, 0, 0, },
+       { "pid",                'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_pid.string, OPT_PID, 0, 0, },
+       { "vpid",               'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_vpid.string, OPT_VPID, 0, 0, },
+       { "uid",                'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_uid.string, OPT_UID, 0, 0, },
+       { "vuid",               'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_vuid.string, OPT_VUID, 0, 0, },
+       { "gid",                'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_gid.string, OPT_GID, 0, 0, },
+       { "vgid",               'p', POPT_ARG_STRING | POPT_ARGFLAG_OPTIONAL, &opt_vgid.string, OPT_VGID, 0, 0, },
+       { "all",                'a', POPT_ARG_NONE, 0, OPT_ALL, 0, 0, },
        { "list-options",       0, POPT_ARG_NONE, NULL, OPT_LIST_OPTIONS, 0, 0, },
        { 0, 0, 0, 0, 0, 0, 0, },
 };
 
 static
-int parse_pid_string(const char *_pid_string,
-               int all, int **_pid_list, int *nr_pids)
+struct id_list *alloc_id_list(size_t nr_items)
+{
+       struct id_list *id_list;
+       struct lttng_tracker_id *items;
+
+       id_list = zmalloc(sizeof(*id_list));
+       if (!id_list) {
+               return NULL;
+       }
+       items = zmalloc(nr_items * sizeof(*items));
+       if (!items) {
+               goto error;
+       }
+       id_list->nr = nr_items;
+       id_list->array = items;
+       return id_list;
+error:
+       free(id_list);
+       return NULL;
+}
+
+static
+void free_id_list(struct id_list *list)
 {
-       const char *one_pid_str;
+       size_t nr_items;
+       int i;
+
+       if (!list)
+               return;
+       nr_items = list->nr;
+       for (i = 0; i < nr_items; i++) {
+               struct lttng_tracker_id *item = &list->array[i];
+
+               free(item->string);
+       }
+       free(list);
+}
+
+static
+int parse_id_string(const char *_id_string,
+               int all, struct id_list **_id_list)
+{
+       const char *one_id_str;
        char *iter;
        int retval = CMD_SUCCESS;
        int count = 0;
-       int *pid_list = NULL;
-       char *pid_string = NULL;
+       struct id_list *id_list = NULL;
+       char *id_string = NULL;
        char *endptr;
 
-       if (all && _pid_string) {
-               ERR("An empty PID string is expected with --all");
+       if (all && _id_string) {
+               ERR("An empty ID string is expected with --all");
                retval = CMD_ERROR;
                goto error;
        }
-       if (!all && !_pid_string) {
-               ERR("Please specify --all with an empty PID string");
+       if (!all && !_id_string) {
+               ERR("Please specify --all with an empty ID string");
                retval = CMD_ERROR;
                goto error;
        }
        if (all) {
-               pid_list = zmalloc(sizeof(*pid_list));
-               if (!pid_list) {
+               /* Empty ID string means all IDs */
+               id_list = alloc_id_list(1);
+               if (!id_list) {
                        ERR("Out of memory");
                        retval = CMD_ERROR;
                        goto error;
                }
-               /* Empty PID string means all PIDs */
-               count = 1;
-               pid_list[0] = -1;
+               id_list->array[0].type = LTTNG_ID_ALL;
                goto assign;
        }
 
-       pid_string = strdup(_pid_string);
-       if (!pid_string) {
+       id_string = strdup(_id_string);
+       if (!id_string) {
                ERR("Out of memory");
                retval = CMD_ERROR;
                goto error;
        }
 
        /* Count */
-       one_pid_str = strtok_r(pid_string, ",", &iter);
-       while (one_pid_str != NULL) {
+       one_id_str = strtok_r(id_string, ",", &iter);
+       while (one_id_str != NULL) {
                unsigned long v;
 
-               errno = 0;
-               v = strtoul(one_pid_str, &endptr, 10);
-               if ((v == 0 && errno == EINVAL)
-                               || (v == ULONG_MAX && errno == ERANGE)
-                               || (*one_pid_str != '\0' && *endptr != '\0')){
-                       ERR("Error parsing PID %s", one_pid_str);
-                       retval = CMD_ERROR;
-                       goto error;
-               }
+               if (isdigit(one_id_str[0])) {
+                       errno = 0;
+                       v = strtoul(one_id_str, &endptr, 10);
+                       if ((v == 0 && errno == EINVAL)
+                                       || (v == ULONG_MAX && errno == ERANGE)
+                                       || (*one_id_str != '\0' && *endptr != '\0')){
+                               ERR("Error parsing ID %s", one_id_str);
+                               retval = CMD_ERROR;
+                               goto error;
+                       }
 
-               if ((long) v > INT_MAX || (int) v < 0) {
-                       ERR("Invalid PID value %ld", (long) v);
-                       retval = CMD_ERROR;
-                       goto error;
+                       if ((long) v > INT_MAX || (int) v < 0) {
+                               ERR("Invalid ID value %ld", (long) v);
+                               retval = CMD_ERROR;
+                               goto error;
+                       }
                }
                count++;
 
                /* For next loop */
-               one_pid_str = strtok_r(NULL, ",", &iter);
+               one_id_str = strtok_r(NULL, ",", &iter);
        }
 
-       free(pid_string);
+       free(id_string);
        /* Identity of delimiter has been lost in first pass. */
-       pid_string = strdup(_pid_string);
-       if (!pid_string) {
+       id_string = strdup(_id_string);
+       if (!id_string) {
                ERR("Out of memory");
                retval = CMD_ERROR;
                goto error;
        }
 
        /* Allocate */
-       pid_list = zmalloc(count * sizeof(*pid_list));
-       if (!pid_list) {
+       id_list = alloc_id_list(count);
+       if (!id_list) {
                ERR("Out of memory");
                retval = CMD_ERROR;
                goto error;
        }
 
-       /* Reparse string and populate the pid list. */
+       /* Reparse string and populate the id list. */
        count = 0;
-       one_pid_str = strtok_r(pid_string, ",", &iter);
-       while (one_pid_str != NULL) {
-               unsigned long v;
+       one_id_str = strtok_r(id_string, ",", &iter);
+       while (one_id_str != NULL) {
+               struct lttng_tracker_id *item;
+
+               item = &id_list->array[count++];
+               if (isdigit(one_id_str[0])) {
+                       unsigned long v;
 
-               v = strtoul(one_pid_str, NULL, 10);
-               pid_list[count++] = (int) v;
+                       v = strtoul(one_id_str, NULL, 10);
+                       item->type = LTTNG_ID_VALUE;
+                       item->value = (int) v;
+               } else {
+                       item->type = LTTNG_ID_STRING;
+                       item->string = strdup(one_id_str);
+                       if (!item->string) {
+                               ERR("Out of memory");
+                               retval = CMD_ERROR;
+                               goto error;
+                       }
+               }
 
                /* For next loop */
-               one_pid_str = strtok_r(NULL, ",", &iter);
+               one_id_str = strtok_r(NULL, ",", &iter);
        }
 
 assign:
-       *nr_pids = count;
-       *_pid_list = pid_list;
+       *_id_list = id_list;
        goto end;       /* SUCCESS */
 
        /* ERROR */
 error:
-       free(pid_list);
+       free_id_list(id_list);
 end:
-       free(pid_string);
+       free(id_string);
        return retval;
 }
 
 static
-enum cmd_error_code track_untrack_pid(enum cmd_type cmd_type, const char *cmd_str,
-               const char *session_name, const char *pid_string,
-               int all, struct mi_writer *writer)
+const char *get_tracker_str(enum lttng_tracker_type tracker_type)
+{
+       switch (tracker_type) {
+       case LTTNG_TRACKER_PID:
+               return "PID";
+       case LTTNG_TRACKER_VPID:
+               return "VPID";
+       case LTTNG_TRACKER_UID:
+               return "UID";
+       case LTTNG_TRACKER_VUID:
+               return "VUID";
+       case LTTNG_TRACKER_GID:
+               return "GID";
+       case LTTNG_TRACKER_VGID:
+               return "VGID";
+       }
+       return NULL;
+}
+
+static
+enum cmd_error_code track_untrack_id(enum cmd_type cmd_type, const char *cmd_str,
+               const char *session_name, const char *id_string,
+               int all, struct mi_writer *writer,
+               enum lttng_tracker_type tracker_type)
 {
        int ret, success = 1 , i;
        enum cmd_error_code retval = CMD_SUCCESS;
-       int *pid_list = NULL;
-       int nr_pids;
+       struct id_list *id_list = NULL;
        struct lttng_domain dom;
        struct lttng_handle *handle = NULL;
-       int (*cmd_func)(struct lttng_handle *handle, int pid);
+       int (*cmd_func)(struct lttng_handle *handle,
+                       enum lttng_tracker_type tracker_type,
+                       struct lttng_tracker_id *id);
+       const char *tracker_str;
 
        switch (cmd_type) {
        case CMD_TRACK:
-               cmd_func = lttng_track_pid;
+               cmd_func = lttng_track_id;
                break;
        case CMD_UNTRACK:
-               cmd_func = lttng_untrack_pid;
+               cmd_func = lttng_untrack_id;
                break;
        default:
                ERR("Unknown command");
                retval = CMD_ERROR;
                goto end;
        }
-
        memset(&dom, 0, sizeof(dom));
        if (opt_kernel) {
                dom.type = LTTNG_DOMAIN_KERNEL;
        } else if (opt_userspace) {
                dom.type = LTTNG_DOMAIN_UST;
+               if (tracker_type == LTTNG_TRACKER_PID) {
+                       MSG("For UST domain, the PID tracker is actually tracking VPID.");
+                       tracker_type = LTTNG_TRACKER_VPID;
+               }
        } else {
                /* Checked by the caller. */
                assert(0);
        }
-
-       ret = parse_pid_string(pid_string, all, &pid_list, &nr_pids);
+       tracker_str = get_tracker_str(tracker_type);
+       if (!tracker_str) {
+               ERR("Unknown tracker type");
+               retval = CMD_ERROR;
+               goto end;
+       }
+       ret = parse_id_string(id_string, all, &id_list);
        if (ret != CMD_SUCCESS) {
-               ERR("Error parsing PID string");
+               ERR("Error parsing %s string", tracker_str);
                retval = CMD_ERROR;
                goto end;
        }
@@ -225,28 +342,42 @@ enum cmd_error_code track_untrack_pid(enum cmd_type cmd_type, const char *cmd_st
        }
 
        if (writer) {
-               /* Open process element */
-               ret = mi_lttng_targets_open(writer);
+               /* Open tracker_id and targets elements */
+               ret = mi_lttng_id_tracker_open(writer, tracker_type);
                if (ret) {
-                       retval = CMD_ERROR;
                        goto end;
                }
        }
 
-       for (i = 0; i < nr_pids; i++) {
-               DBG("%s PID %d", cmd_str, pid_list[i]);
-               ret = cmd_func(handle, pid_list[i]);
+       for (i = 0; i < id_list->nr; i++) {
+               struct lttng_tracker_id *item = &id_list->array[i];
+
+               switch (item->type) {
+               case LTTNG_ID_ALL:
+                       DBG("%s all IDs", cmd_str);
+                       break;
+               case LTTNG_ID_VALUE:
+                       DBG("%s ID %d", cmd_str, item->value);
+                       break;
+               case LTTNG_ID_STRING:
+                       DBG("%s ID '%s'", cmd_str, item->string);
+                       break;
+               default:
+                       retval = CMD_ERROR;
+                       goto end;
+               }
+               ret = cmd_func(handle, tracker_type, item);
                if (ret) {
+                       char *msg = NULL;
+
                        switch (-ret) {
-                       case LTTNG_ERR_PID_TRACKED:
-                               WARN("PID %i already tracked in session %s",
-                                               pid_list[i], session_name);
+                       case LTTNG_ERR_ID_TRACKED:
+                               msg = "already tracked";
                                success = 1;
                                retval = CMD_SUCCESS;
                                break;
-                       case LTTNG_ERR_PID_NOT_TRACKED:
-                               WARN("PID %i not tracked in session %s",
-                                               pid_list[i], session_name);
+                       case LTTNG_ERR_ID_NOT_TRACKED:
+                               msg = "not tracked";
                                success = 1;
                                retval = CMD_SUCCESS;
                                break;
@@ -256,15 +387,54 @@ enum cmd_error_code track_untrack_pid(enum cmd_type cmd_type, const char *cmd_st
                                retval = CMD_ERROR;
                                break;
                        }
+                       if (msg) {
+                               switch (item->type) {
+                               case LTTNG_ID_ALL:
+                                       WARN("All %s %s in session %s",
+                                               tracker_str, msg, session_name);
+                                       break;
+                               case LTTNG_ID_VALUE:
+                                       WARN("%s %i %s in session %s",
+                                               tracker_str, item->value,
+                                               msg, session_name);
+                                       break;
+                               case LTTNG_ID_STRING:
+                                       WARN("%s '%s' %s in session %s",
+                                               tracker_str, item->string,
+                                               msg, session_name);
+                                       break;
+                               default:
+                                       retval = CMD_ERROR;
+                                       goto end;
+                               }
+                       }
                } else {
-                       MSG("PID %i %sed in session %s",
-                                       pid_list[i], cmd_str, session_name);
+                       switch (item->type) {
+                       case LTTNG_ID_ALL:
+                               MSG("All %ss %sed in session %s",
+                                       tracker_str, cmd_str, session_name);
+                               break;
+                       case LTTNG_ID_VALUE:
+                               MSG("%s %i %sed in session %s",
+                                       tracker_str, item->value, cmd_str,
+                                       session_name);
+                               break;
+                       case LTTNG_ID_STRING:
+                               MSG("%s '%s' %sed in session %s",
+                                       tracker_str, item->string, cmd_str,
+                                       session_name);
+                               break;
+                       default:
+                               retval = CMD_ERROR;
+                               goto end;
+                       }
                        success = 1;
                }
 
                /* Mi */
                if (writer) {
-                       ret = mi_lttng_pid_target(writer, pid_list[i], 1);
+                       ret = mi_lttng_id_target(writer,
+                               tracker_type, item, 1);
                        if (ret) {
                                retval = CMD_ERROR;
                                goto end;
@@ -286,8 +456,8 @@ enum cmd_error_code track_untrack_pid(enum cmd_type cmd_type, const char *cmd_st
        }
 
        if (writer) {
-               /* Close targets element */
-               ret = mi_lttng_writer_close_element(writer);
+               /* Close targets and tracker_id elements */
+               ret = mi_lttng_close_multi_element(writer, 2);
                if (ret) {
                        retval = CMD_ERROR;
                        goto end;
@@ -298,7 +468,7 @@ end:
        if (handle) {
                lttng_destroy_handle(handle);
        }
-       free(pid_list);
+       free_id_list(id_list);
        return retval;
 }
 
@@ -346,8 +516,55 @@ int cmd_track_untrack(enum cmd_type cmd_type, const char *cmd_str,
                        list_cmd_options(stdout, long_options);
                        goto end;
                case OPT_SESSION:
+                       break;
                case OPT_PID:
-                       opt_pid = 1;
+                       opt_pid.used = 1;
+                       type_state = STATE_PID;
+                       break;
+               case OPT_VPID:
+                       opt_vpid.used = 1;
+                       type_state = STATE_VPID;
+                       break;
+               case OPT_UID:
+                       opt_uid.used = 1;
+                       type_state = STATE_UID;
+                       break;
+               case OPT_VUID:
+                       opt_vuid.used = 1;
+                       type_state = STATE_VUID;
+                       break;
+               case OPT_GID:
+                       opt_gid.used = 1;
+                       type_state = STATE_GID;
+                       break;
+               case OPT_VGID:
+                       opt_vgid.used = 1;
+                       type_state = STATE_VGID;
+                       break;
+               case OPT_ALL:
+                       switch (type_state) {
+                       case STATE_PID:
+                               opt_pid.all = 1;
+                               break;
+                       case STATE_VPID:
+                               opt_vpid.all = 1;
+                               break;
+                       case STATE_UID:
+                               opt_uid.all = 1;
+                               break;
+                       case STATE_VUID:
+                               opt_vuid.all = 1;
+                               break;
+                       case STATE_GID:
+                               opt_gid.all = 1;
+                               break;
+                       case STATE_VGID:
+                               opt_vgid.all = 1;
+                               break;
+                       default:
+                               command_ret = CMD_ERROR;
+                               goto end;
+                       }
                        break;
                default:
                        command_ret = CMD_UNDEFINED;
@@ -371,13 +588,6 @@ int cmd_track_untrack(enum cmd_type cmd_type, const char *cmd_str,
                session_name = opt_session_name;
        }
 
-       /* Currently only PID tracker is supported */
-       if (!opt_pid) {
-               ERR("Please specify at least one tracker with its expected arguments");
-               command_ret = CMD_ERROR;
-               goto end;
-       }
-
        /* Mi check */
        if (lttng_opt_mi) {
                writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
@@ -403,19 +613,66 @@ int cmd_track_untrack(enum cmd_type cmd_type, const char *cmd_str,
                        command_ret = CMD_ERROR;
                        goto end;
                }
+
+               ret = mi_lttng_trackers_open(writer);
+               if (ret) {
+                       goto end;
+               }
        }
 
-       command_ret = track_untrack_pid(cmd_type,
-                       cmd_str, session_name, opt_pid_string,
-                       opt_all, writer);
-       if (command_ret != CMD_SUCCESS) {
-               success = 0;
+       if (opt_pid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_pid.string,
+                               opt_pid.all, writer, LTTNG_TRACKER_PID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
+       }
+       if (opt_vpid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_vpid.string,
+                               opt_vpid.all, writer, LTTNG_TRACKER_VPID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
+       }
+       if (opt_uid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_uid.string,
+                               opt_uid.all, writer, LTTNG_TRACKER_UID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
+       }
+       if (opt_vuid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_vuid.string,
+                               opt_vuid.all, writer, LTTNG_TRACKER_VUID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
+       }
+       if (opt_gid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_gid.string,
+                               opt_gid.all, writer, LTTNG_TRACKER_GID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
+       }
+       if (opt_vgid.used) {
+               command_ret = track_untrack_id(cmd_type,
+                               cmd_str, session_name, opt_vgid.string,
+                               opt_vgid.all, writer, LTTNG_TRACKER_VGID);
+               if (command_ret != CMD_SUCCESS) {
+                       success = 0;
+               }
        }
 
        /* Mi closing */
        if (writer) {
-               /* Close  output element */
-               ret = mi_lttng_writer_close_element(writer);
+               /* Close trackers and output elements */
+               ret = mi_lttng_close_multi_element(writer, 2);
                if (ret) {
                        command_ret = CMD_ERROR;
                        goto end;
This page took 0.037775 seconds and 5 git commands to generate.