Fix: typo 'retrive' -> 'retrieve'
[lttng-tools.git] / src / bin / lttng / commands / add_context.c
index 405b7828f043dea695beff3bde4206cec59ee4ff..7aef4d50f7dd575e34c704f5b2a82896af8f5db1 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,7 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#define _GNU_SOURCE
+#define _LGPL_SOURCE
 #include <ctype.h>
 #include <popt.h>
 #include <stdio.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <unistd.h>
+#include <assert.h>
 
 #include <urcu/list.h>
 
-#include "../command.h"
+#include <common/mi-lttng.h>
 
-#define PRINT_LINE_LEN 80
+#include "../command.h"
 
 static char *opt_channel_name;
 static char *opt_session_name;
 static int opt_kernel;
 static int opt_userspace;
+static int opt_jul;
+static int opt_log4j;
 static char *opt_type;
-#if 0
-/* Not implemented yet */
-static char *opt_cmd_name;
-static pid_t opt_pid;
+
+#ifdef LTTNG_EMBED_HELP
+static const char help_msg[] =
+#include <lttng-add-context.1.h>
+;
 #endif
 
 enum {
        OPT_HELP = 1,
        OPT_TYPE,
        OPT_USERSPACE,
+       OPT_JUL,
+       OPT_LOG4J,
        OPT_LIST_OPTIONS,
+       OPT_LIST,
 };
 
 static struct lttng_handle *handle;
+static struct mi_writer *writer;
 
 /*
  * Taken from the LTTng ABI
@@ -70,6 +79,13 @@ enum context_type {
        CONTEXT_IP           = 12,
        CONTEXT_PERF_CPU_COUNTER = 13,
        CONTEXT_PERF_THREAD_COUNTER = 14,
+       CONTEXT_APP_CONTEXT  = 15,
+       CONTEXT_INTERRUPTIBLE = 16,
+       CONTEXT_PREEMPTIBLE  = 17,
+       CONTEXT_NEED_RESCHEDULE = 18,
+       CONTEXT_MIGRATABLE   = 19,
+       CONTEXT_CALLSTACK_KERNEL = 20,
+       CONTEXT_CALLSTACK_USER = 21,
 };
 
 /*
@@ -79,6 +95,7 @@ enum perf_type {
        PERF_TYPE_HARDWARE = 0,
        PERF_TYPE_SOFTWARE = 1,
        PERF_TYPE_HW_CACHE = 3,
+       PERF_TYPE_RAW = 4,
 };
 
 enum perf_count_hard {
@@ -145,7 +162,10 @@ static struct poptOption long_options[] = {
        {"channel",        'c', POPT_ARG_STRING, &opt_channel_name, 0, 0, 0},
        {"kernel",         'k', POPT_ARG_VAL, &opt_kernel, 1, 0, 0},
        {"userspace",      'u', POPT_ARG_NONE, 0, OPT_USERSPACE, 0, 0},
+       {"jul",            'j', POPT_ARG_NONE, 0, OPT_JUL, 0, 0},
+       {"log4j",          'l', POPT_ARG_NONE, 0, OPT_LOG4J, 0, 0},
        {"type",           't', POPT_ARG_STRING, &opt_type, OPT_TYPE, 0, 0},
+       {"list",           0, POPT_ARG_NONE, NULL, OPT_LIST, NULL, NULL},
        {"list-options",   0, POPT_ARG_NONE, NULL, OPT_LIST_OPTIONS, NULL, NULL},
        {0, 0, 0, 0, 0, 0, 0}
 };
@@ -200,6 +220,10 @@ const struct ctx_opts {
                        uint32_t type;
                        uint64_t config;
                } perf;
+               struct {
+                       char *provider_name;
+                       char *ctx_name;
+               } app_ctx;
        } u;
 } ctx_opts[] = {
        { "pid", CONTEXT_PID },
@@ -214,6 +238,14 @@ const struct ctx_opts {
        { "vppid", CONTEXT_VPPID },
        { "hostname", CONTEXT_HOSTNAME },
        { "ip", CONTEXT_IP },
+       { "interruptible", CONTEXT_INTERRUPTIBLE },
+       { "preemptible", CONTEXT_PREEMPTIBLE },
+       { "need_reschedule", CONTEXT_NEED_RESCHEDULE },
+       { "migratable", CONTEXT_MIGRATABLE },
+       { "callstack-kernel", CONTEXT_CALLSTACK_KERNEL },
+#if HAVE_MODULES_USERSPACE_CALLSTACK_CONTEXT
+       { "callstack-user", CONTEXT_CALLSTACK_USER },
+#endif
 
        /* Perf options */
 
@@ -442,7 +474,7 @@ const struct ctx_opts {
  * Context type for command line option parsing.
  */
 struct ctx_type {
-       const struct ctx_opts *opt;
+       struct ctx_opts *opt;
        struct cds_list_head list;
 };
 
@@ -456,88 +488,232 @@ struct ctx_type_list {
        .head = CDS_LIST_HEAD_INIT(ctx_type_list.head),
 };
 
+
+
 /*
- * Pretty print context type.
+ * Find context numerical value from string.
+ *
+ * Return -1 if not found.
  */
-static void print_ctx_type(FILE *ofp)
+static int find_ctx_type_idx(const char *opt)
 {
-       const char *indent = "                               ";
-       int indent_len = strlen(indent);
-       int len, i = 0;
+       int ret, i = 0;
 
-       fprintf(ofp, "%s", indent);
-       len = indent_len;
        while (ctx_opts[i].symbol != NULL) {
-               if (!ctx_opts[i].hide_help) {
-                       if (len > indent_len) {
-                               if (len + strlen(ctx_opts[i].symbol) + 2
-                                               >= PRINT_LINE_LEN) {
-                                       fprintf(ofp, ",\n");
-                                       fprintf(ofp, "%s", indent);
-                                       len = indent_len;
-                               } else {
-                                       len += fprintf(ofp, ", ");
-                               }
-                       }
-                       len += fprintf(ofp, "%s", ctx_opts[i].symbol);
+               if (strcmp(opt, ctx_opts[i].symbol) == 0) {
+                       ret = i;
+                       goto end;
                }
                i++;
        }
+
+       ret = -1;
+end:
+       return ret;
 }
 
-/*
- * usage
- */
-static void usage(FILE *ofp)
+static
+enum lttng_domain_type get_domain(void)
 {
-       fprintf(ofp, "usage: lttng add-context -t TYPE [-k|-u] [OPTIONS]\n");
-       fprintf(ofp, "\n");
-       fprintf(ofp, "If no channel is given (-c), the context is added to\n");
-       fprintf(ofp, "all channels.\n");
-       fprintf(ofp, "\n");
-       fprintf(ofp, "Otherwise the context is added only to the channel (-c).\n");
-       fprintf(ofp, "\n");
-       fprintf(ofp, "Exactly one domain (-k or -u) must be specified.\n");
-       fprintf(ofp, "\n");
-       fprintf(ofp, "Options:\n");
-       fprintf(ofp, "  -h, --help               Show this help\n");
-       fprintf(ofp, "      --list-options       Simple listing of options\n");
-       fprintf(ofp, "  -s, --session NAME       Apply to session name\n");
-       fprintf(ofp, "  -c, --channel NAME       Apply to channel\n");
-       fprintf(ofp, "  -k, --kernel             Apply to the kernel tracer\n");
-       fprintf(ofp, "  -u, --userspace          Apply to the user-space tracer\n");
-       fprintf(ofp, "\n");
-       fprintf(ofp, "Context:\n");
-       fprintf(ofp, "  -t, --type TYPE          Context type. You can repeat that option on\n");
-       fprintf(ofp, "                           the command line to specify multiple contexts at once.\n");
-       fprintf(ofp, "                           (--kernel preempts --userspace)\n");
-       fprintf(ofp, "                           TYPE can be one of the strings below:\n");
-       print_ctx_type(ofp);
-       fprintf(ofp, "\n");
-       fprintf(ofp, "Example:\n");
-       fprintf(ofp, "This command will add the context information 'prio' and two per-cpu\n"
-                       "perf counters (hardware branch misses and cache misses), to all channels\n"
-                       "in the trace data output:\n");
-       fprintf(ofp, "# lttng add-context -k -t prio -t perf:cpu:branch-misses -t perf:cpu:cache-misses\n");
-       fprintf(ofp, "\n");
+       if (opt_kernel) {
+               return LTTNG_DOMAIN_KERNEL;
+       } else if (opt_userspace) {
+               return LTTNG_DOMAIN_UST;
+       } else if (opt_jul) {
+               return LTTNG_DOMAIN_JUL;
+       } else if (opt_log4j) {
+               return LTTNG_DOMAIN_LOG4J;
+       } else {
+               assert(0);
+       }
+}
+
+static
+int mi_open(void)
+{
+       int ret;
+
+       /* MI check */
+       if (!lttng_opt_mi) {
+               ret = 0;
+               goto end;
+       }
+
+       ret = fileno(stdout);
+       if (ret < 0) {
+               PERROR("Unable to retrieve fileno of stdout");
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       writer = mi_lttng_writer_create(ret, lttng_opt_mi);
+       if (!writer) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       /* Open command element */
+       ret = mi_lttng_writer_command_open(writer,
+                       mi_lttng_element_command_add_context);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       /* Open output element */
+       ret = mi_lttng_writer_open_element(writer,
+                       mi_lttng_element_command_output);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+end:
+       return ret;
+}
+
+static
+int mi_close(enum cmd_error_code success)
+{
+       int ret;
+
+       /* MI closing */
+       if (!lttng_opt_mi) {
+               ret = 0;
+               goto end;
+       }
+       /* Close  output element */
+       ret = mi_lttng_writer_close_element(writer);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       /* Success ? */
+       ret = mi_lttng_writer_write_element_bool(writer,
+                       mi_lttng_element_command_success, !success);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       /* Command element close */
+       ret = mi_lttng_writer_command_close(writer);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+end:
+       return ret;
+}
+
+static
+void populate_context(struct lttng_event_context *context,
+               const struct ctx_opts *opt)
+{
+       char *ptr;
+
+       context->ctx = (enum lttng_event_context_type) opt->ctx_type;
+       switch (context->ctx) {
+       case LTTNG_EVENT_CONTEXT_PERF_COUNTER:
+       case LTTNG_EVENT_CONTEXT_PERF_CPU_COUNTER:
+       case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER:
+               context->u.perf_counter.type = opt->u.perf.type;
+               context->u.perf_counter.config = opt->u.perf.config;
+               strncpy(context->u.perf_counter.name, opt->symbol,
+                               LTTNG_SYMBOL_NAME_LEN);
+               context->u.perf_counter.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
+               /* Replace : and - by _ */
+               while ((ptr = strchr(context->u.perf_counter.name, '-')) != NULL) {
+                       *ptr = '_';
+               }
+               while ((ptr = strchr(context->u.perf_counter.name, ':')) != NULL) {
+                       *ptr = '_';
+               }
+               break;
+       case LTTNG_EVENT_CONTEXT_APP_CONTEXT:
+               context->u.app_ctx.provider_name =
+                       opt->u.app_ctx.provider_name;
+               context->u.app_ctx.ctx_name =
+                       opt->u.app_ctx.ctx_name;
+               break;
+       default:
+               break;
+       }
 }
 
 /*
- * Find context numerical value from string.
+ * Pretty print context type.
  */
-static int find_ctx_type_idx(const char *opt)
+static
+int print_ctx_type(void)
 {
-       int ret = -1, i = 0;
 
-       while (ctx_opts[i].symbol != NULL) {
-               if (strcmp(opt, ctx_opts[i].symbol) == 0) {
-                       ret = i;
+       FILE *ofp = stdout;
+       int i = 0;
+       int ret;
+       struct lttng_event_context context;
+
+       memset(&context, 0, sizeof(context));
+
+       ret = mi_open();
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       if (lttng_opt_mi) {
+               /* Open a contexts element */
+               ret = mi_lttng_writer_open_element(writer, config_element_contexts);
+               if (ret) {
+                       ret = CMD_ERROR;
                        goto end;
                }
+       }
+
+       while (ctx_opts[i].symbol != NULL) {
+               if (!ctx_opts[i].hide_help) {
+                       if (lttng_opt_mi) {
+                               populate_context(&context, &ctx_opts[i]);
+                               ret = mi_lttng_context(writer, &context, 1);
+                               if (ret) {
+                                       ret = CMD_ERROR;
+                                       goto end;
+                               }
+
+                               ret = mi_lttng_writer_write_element_string(
+                                               writer,
+                                               mi_lttng_element_context_symbol,
+                                               ctx_opts[i].symbol);
+                               if (ret) {
+                                       ret = CMD_ERROR;
+                                       goto end;
+                               }
+
+                               ret = mi_lttng_writer_close_element(writer);
+                               if (ret) {
+                                       ret = CMD_ERROR;
+                                       goto end;
+                               }
+                       } else {
+                               fprintf(ofp, "%s\n", ctx_opts[i].symbol);
+                       }
+               }
                i++;
        }
 
+       if (lttng_opt_mi) {
+               /* Close contexts element */
+               ret = mi_lttng_writer_close_element(writer);
+               if (ret) {
+                       goto end;
+               }
+       }
+
 end:
+       ret = mi_close(ret);
+       if (ret) {
+               ret = CMD_ERROR;
+       }
        return ret;
 }
 
@@ -546,70 +722,92 @@ end:
  */
 static int add_context(char *session_name)
 {
-       int ret = CMD_SUCCESS, warn = 0;
+       int ret = CMD_SUCCESS, warn = 0, success = 0;
        struct lttng_event_context context;
        struct lttng_domain dom;
        struct ctx_type *type;
-       char *ptr;
 
        memset(&context, 0, sizeof(context));
        memset(&dom, 0, sizeof(dom));
 
-       if (opt_kernel) {
-               dom.type = LTTNG_DOMAIN_KERNEL;
-       } else if (opt_userspace) {
-               dom.type = LTTNG_DOMAIN_UST;
-       } else {
-               print_missing_domain();
-               ret = CMD_ERROR;
-               goto error;
-       }
-
+       dom.type = get_domain();
        handle = lttng_create_handle(session_name, &dom);
        if (handle == NULL) {
                ret = CMD_ERROR;
                goto error;
        }
 
+       if (lttng_opt_mi) {
+               /* Open a contexts element */
+               ret = mi_lttng_writer_open_element(writer, config_element_contexts);
+               if (ret) {
+                       goto error;
+               }
+       }
+
        /* Iterate over all the context types given */
        cds_list_for_each_entry(type, &ctx_type_list.head, list) {
-               context.ctx = (enum lttng_event_context_type) type->opt->ctx_type;
-               switch (context.ctx) {
-               case LTTNG_EVENT_CONTEXT_PERF_COUNTER:
-               case LTTNG_EVENT_CONTEXT_PERF_CPU_COUNTER:
-               case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER:
-                       context.u.perf_counter.type = type->opt->u.perf.type;
-                       context.u.perf_counter.config = type->opt->u.perf.config;
-                       strncpy(context.u.perf_counter.name, type->opt->symbol,
-                               LTTNG_SYMBOL_NAME_LEN);
-                       context.u.perf_counter.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
-                       /* Replace : and - by _ */
-                       while ((ptr = strchr(context.u.perf_counter.name, '-')) != NULL) {
-                               *ptr = '_';
+               DBG("Adding context...");
+
+               populate_context(&context, type->opt);
+
+               if (lttng_opt_mi) {
+                       /* We leave context open the update the success of the command */
+                       ret = mi_lttng_context(writer, &context, 1);
+                       if (ret) {
+                               ret = CMD_ERROR;
+                               goto error;
                        }
-                       while ((ptr = strchr(context.u.perf_counter.name, ':')) != NULL) {
-                               *ptr = '_';
+
+                       ret = mi_lttng_writer_write_element_string(writer,
+                                       mi_lttng_element_context_symbol,
+                                       type->opt->symbol);
+                       if (ret) {
+                               ret = CMD_ERROR;
+                               goto error;
                        }
-                       break;
-               default:
-                       break;
                }
-               DBG("Adding context...");
 
                ret = lttng_add_context(handle, &context, NULL, opt_channel_name);
                if (ret < 0) {
                        ERR("%s: %s", type->opt->symbol, lttng_strerror(ret));
                        warn = 1;
-                       continue;
+                       success = 0;
                } else {
                        if (opt_channel_name) {
                                MSG("%s context %s added to channel %s",
-                                               opt_kernel ? "kernel" : "UST", type->opt->symbol,
+                                               get_domain_str(dom.type), type->opt->symbol,
                                                opt_channel_name);
                        } else {
                                MSG("%s context %s added to all channels",
-                                               opt_kernel ? "kernel" : "UST", type->opt->symbol)
+                                               get_domain_str(dom.type), type->opt->symbol);
                        }
+                       success = 1;
+               }
+
+               if (lttng_opt_mi) {
+                       /* Is the single operation a success ? */
+                       ret = mi_lttng_writer_write_element_bool(writer,
+                                       mi_lttng_element_success, success);
+                       if (ret) {
+                               ret = CMD_ERROR;
+                               goto error;
+                       }
+
+                       /* Close the context element */
+                       ret = mi_lttng_writer_close_element(writer);
+                       if (ret) {
+                               ret = CMD_ERROR;
+                               goto error;
+                       }
+               }
+       }
+
+       if (lttng_opt_mi) {
+               /* Close contexts element */
+               ret = mi_lttng_writer_close_element(writer);
+               if (ret) {
+                       goto error;
                }
        }
 
@@ -622,24 +820,250 @@ error:
         * This means that at least one add_context failed and tells the user to
         * look on stderr for error(s).
         */
-       if (warn) {
+       if (!ret && warn) {
                ret = CMD_WARNING;
        }
        return ret;
 }
 
+static
+void destroy_ctx_type(struct ctx_type *type)
+{
+       if (!type) {
+               return;
+       }
+       if (type->opt) {
+               free(type->opt->symbol);
+       }
+       free(type->opt);
+       free(type);
+}
+
+static
+struct ctx_type *create_ctx_type(void)
+{
+       struct ctx_type *type = zmalloc(sizeof(*type));
+
+       if (!type) {
+               PERROR("malloc ctx_type");
+               goto end;
+       }
+
+       type->opt = zmalloc(sizeof(*type->opt));
+       if (!type->opt) {
+               PERROR("malloc ctx_type options");
+               destroy_ctx_type(type);
+               type = NULL;
+               goto end;
+       }
+end:
+       return type;
+}
+
+static
+int find_ctx_type_perf_raw(const char *ctx, struct ctx_type *type)
+{
+       int ret;
+       int field_pos = 0;
+       char *tmp_list, *cur_list;
+
+       cur_list = tmp_list = strdup(ctx);
+       if (!tmp_list) {
+               PERROR("strdup temp list");
+               ret = -ENOMEM;
+               goto end;
+       }
+
+       /* Looking for "perf:[cpu|thread]:raw:<mask>:<name>". */
+       for (;;) {
+               char *next;
+
+               next = strtok(cur_list, ":");
+               if (!next) {
+                       break;
+               }
+               cur_list = NULL;
+               switch (field_pos) {
+               case 0:
+                       if (strncmp(next, "perf", 4) != 0) {
+                               ret = -1;
+                               goto end;
+                       }
+                       break;
+               case 1:
+                       if (strncmp(next, "cpu", 3) == 0) {
+                               type->opt->ctx_type = CONTEXT_PERF_CPU_COUNTER;
+                       } else if (strncmp(next, "thread", 4) == 0) {
+                               type->opt->ctx_type = CONTEXT_PERF_THREAD_COUNTER;
+                       } else {
+                               ret = -1;
+                               goto end;
+                       }
+                       break;
+               case 2:
+                       if (strncmp(next, "raw", 3) != 0) {
+                               ret = -1;
+                               goto end;
+                       }
+                       break;
+               case 3:
+               {
+                       char *endptr;
+
+                       if (strlen(next) < 2 || next[0] != 'r') {
+                               ERR("Wrong perf raw mask format: expected rNNN");
+                               ret = -1;
+                               goto end;
+                       }
+                       errno = 0;
+                       type->opt->u.perf.config = strtoll(next +  1, &endptr, 16);
+                       if (errno != 0 || !endptr || *endptr) {
+                               ERR("Wrong perf raw mask format: expected rNNN");
+                               ret = -1;
+                               goto end;
+                       }
+                       break;
+               }
+               case 4:
+                       /* name */
+                       break;
+               case 5:
+                       ERR("Too many ':' in perf raw format");
+                       ret = -1;
+                       goto end;
+               };
+               field_pos++;
+       }
+
+       if (field_pos < 5) {
+               ERR("Invalid perf counter specifier, expected a specifier of "
+                       "the form perf:cpu:raw:rNNN:<name> or "
+                       "perf:thread:raw:rNNN:<name>");
+               ret = -1;
+               goto end;
+       }
+
+       ret = 0;
+       goto end;
+
+end:
+       free(tmp_list);
+       return ret;
+}
+
+static
+struct ctx_type *get_context_type(const char *ctx)
+{
+       int opt_index, ret;
+       struct ctx_type *type = NULL;
+       const char app_ctx_prefix[] = "$app.";
+       char *provider_name = NULL, *ctx_name = NULL;
+       size_t i, len, colon_pos = 0, provider_name_len, ctx_name_len;
+
+       if (!ctx) {
+               goto not_found;
+       }
+
+       type = create_ctx_type();
+       if (!type) {
+               goto not_found;
+       }
+
+       /* Check if ctx matches a known static context. */
+       opt_index = find_ctx_type_idx(ctx);
+       if (opt_index >= 0) {
+               *type->opt = ctx_opts[opt_index];
+               type->opt->symbol = strdup(ctx_opts[opt_index].symbol);
+               goto found;
+       }
+
+       /* Check if ctx is a raw perf context. */
+       ret = find_ctx_type_perf_raw(ctx, type);
+       if (ret == 0) {
+               type->opt->u.perf.type = PERF_TYPE_RAW;
+               type->opt->symbol = strdup(ctx);
+               if (!type->opt->symbol) {
+                       PERROR("Copy perf field name");
+                       goto not_found;
+               }
+               goto found;
+       }
+
+       /*
+        * No match found against static contexts; check if it is an app
+        * context.
+        */
+       len = strlen(ctx);
+       if (len <= sizeof(app_ctx_prefix) - 1) {
+               goto not_found;
+       }
+
+       /* String starts with $app. */
+       if (strncmp(ctx, app_ctx_prefix, sizeof(app_ctx_prefix) - 1)) {
+               goto not_found;
+       }
+
+       /* Validate that the ':' separator is present. */
+       for (i = sizeof(app_ctx_prefix); i < len; i++) {
+               const char c = ctx[i];
+
+               if (c == ':') {
+                       colon_pos = i;
+                       break;
+               }
+       }
+
+       /*
+        * No colon found or no ctx name ("$app.provider:") or no provider name
+        * given ("$app.:..."), which is invalid.
+        */
+       if (!colon_pos || colon_pos == len ||
+                       colon_pos == sizeof(app_ctx_prefix)) {
+               ERR("Invalid application context provided: no provider or context name provided.");
+               goto not_found;
+       }
+
+       provider_name_len = colon_pos - sizeof(app_ctx_prefix) + 2;
+       provider_name = zmalloc(provider_name_len);
+       if (!provider_name) {
+               PERROR("malloc provider_name");
+               goto not_found;
+       }
+       strncpy(provider_name, ctx + sizeof(app_ctx_prefix) - 1,
+                       provider_name_len - 1);
+       type->opt->u.app_ctx.provider_name = provider_name;
+
+       ctx_name_len = len - colon_pos;
+       ctx_name = zmalloc(ctx_name_len);
+       if (!ctx_name) {
+               PERROR("malloc ctx_name");
+               goto not_found;
+       }
+       strncpy(ctx_name, ctx + colon_pos + 1, ctx_name_len - 1);
+       type->opt->u.app_ctx.ctx_name = ctx_name;
+       type->opt->ctx_type = CONTEXT_APP_CONTEXT;
+       type->opt->symbol = strdup(ctx);
+found:
+       return type;
+not_found:
+       free(provider_name);
+       free(ctx_name);
+       destroy_ctx_type(type);
+       return NULL;
+}
+
 /*
  * Add context to channel or event.
  */
 int cmd_add_context(int argc, const char **argv)
 {
-       int index, opt, ret = CMD_SUCCESS;
+       int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS;
        static poptContext pc;
        struct ctx_type *type, *tmptype;
        char *session_name = NULL;
+       const char *leftover = NULL;
 
        if (argc < 2) {
-               usage(stderr);
                ret = CMD_ERROR;
                goto end;
        }
@@ -650,56 +1074,56 @@ int cmd_add_context(int argc, const char **argv)
        while ((opt = poptGetNextOpt(pc)) != -1) {
                switch (opt) {
                case OPT_HELP:
-                       usage(stdout);
+                       SHOW_HELP();
+                       goto end;
+               case OPT_LIST:
+                       ret = print_ctx_type();
                        goto end;
                case OPT_TYPE:
-                       /*
-                        * Look up the index of opt_type in ctx_opts[] first, so we don't
-                        * have to free(type) on failure.
-                        */
-                       index = find_ctx_type_idx(opt_type);
-                       if (index < 0) {
+               {
+                       type = get_context_type(opt_type);
+                       if (!type) {
                                ERR("Unknown context type %s", opt_type);
-                               ret = CMD_ERROR;
-                               goto end;
-                       }
-
-                       type = malloc(sizeof(struct ctx_type));
-                       if (type == NULL) {
-                               perror("malloc ctx_type");
                                ret = CMD_FATAL;
                                goto end;
                        }
-
-                       type->opt = &ctx_opts[index];
-                       if (type->opt->symbol == NULL) {
-                               ERR("Unknown context type %s", opt_type);
-                               free(type);
-                               ret = CMD_ERROR;
-                               goto end;
-                       } else {
-                               cds_list_add_tail(&type->list, &ctx_type_list.head);
-                       }
+                       cds_list_add_tail(&type->list, &ctx_type_list.head);
                        break;
+               }
                case OPT_USERSPACE:
                        opt_userspace = 1;
-#if 0
-                       opt_cmd_name = poptGetOptArg(pc);
-#endif
+                       break;
+               case OPT_JUL:
+                       opt_jul = 1;
+                       break;
+               case OPT_LOG4J:
+                       opt_log4j = 1;
                        break;
                case OPT_LIST_OPTIONS:
                        list_cmd_options(stdout, long_options);
                        goto end;
                default:
-                       usage(stderr);
                        ret = CMD_UNDEFINED;
                        goto end;
                }
        }
 
+       leftover = poptGetArg(pc);
+       if (leftover) {
+               ERR("Unknown argument: %s", leftover);
+               ret = CMD_ERROR;
+               goto end;
+       }
+
+       ret = print_missing_or_multiple_domains(opt_kernel + opt_userspace +
+                       opt_jul + opt_log4j);
+       if (ret) {
+               ret = CMD_ERROR;
+               goto end;
+       }
+
        if (!opt_type) {
                ERR("Missing mandatory -t TYPE");
-               usage(stderr);
                ret = CMD_ERROR;
                goto end;
        }
@@ -714,18 +1138,36 @@ int cmd_add_context(int argc, const char **argv)
                session_name = opt_session_name;
        }
 
-       ret = add_context(session_name);
+       ret = mi_open();
+       if (ret) {
+               goto end;
+       }
+
+       command_ret = add_context(session_name);
+       ret = mi_close(command_ret);
+       if (ret) {
+               goto end;
+       }
 
+end:
        if (!opt_session_name) {
                free(session_name);
        }
 
-end:
+       /* Mi clean-up */
+       if (writer && mi_lttng_writer_destroy(writer)) {
+               /* Preserve original error code */
+               ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL;
+       }
+
        /* Cleanup allocated memory */
        cds_list_for_each_entry_safe(type, tmptype, &ctx_type_list.head, list) {
-               free(type);
+               destroy_ctx_type(type);
        }
 
+       /* Overwrite ret if an error occurred during add_context() */
+       ret = command_ret ? command_ret : ret;
+
        poptFreeContext(pc);
        return ret;
 }
This page took 0.034201 seconds and 5 git commands to generate.