Fix: params arg parsing: apply to current component params
[babeltrace.git] / converter / babeltrace-cfg.c
index 2d66600ac6553b8ee9e93d4840b918748a586a84..c9744475fc7b88fc18168135952333954f566371 100644 (file)
@@ -35,6 +35,8 @@
 #include <glib.h>
 #include "babeltrace-cfg.h"
 
+#define DEFAULT_SOURCE_COMPONENT_NAME  "ctf.fs"
+
 /*
  * Error printf() macro which prepends "Error: " the first time it's
  * called. This gives a nicer feel than having a bunch of error prefixes
@@ -777,6 +779,9 @@ void print_usage(FILE *fp)
        fprintf(fp, "  -i, --source=PLUGIN.COMPCLS       Instantiate a source component from plugin\n");
        fprintf(fp, "                                    PLUGIN and component class COMPCLS (may be\n");
        fprintf(fp, "                                    repeated)\n");
+       fprintf(fp, "      --begin                       Start time: [YYYY-MM-DD [hh:mm:]]ss[.nnnnnnnnn]\n");
+       fprintf(fp, "      --end                         End time: [YYYY-MM-DD [hh:mm:]]ss[.nnnnnnnnn]\n");
+       fprintf(fp, "      --timerange                   Time range: begin,end or [begin,end] (where [] are actual brackets)\n");
        fprintf(fp, "  -h  --help                        Show this help\n");
        fprintf(fp, "      --help-legacy                 Show Babeltrace 1.x legacy options\n");
        fprintf(fp, "  -v, --verbose                     Enable verbose output\n");
@@ -2108,6 +2113,7 @@ int parse_int64(const char *arg, int64_t *val)
 enum {
        OPT_NONE = 0,
        OPT_BASE_PARAMS,
+       OPT_BEGIN,
        OPT_CLOCK_CYCLES,
        OPT_CLOCK_DATE,
        OPT_CLOCK_FORCE_CORRELATE,
@@ -2119,6 +2125,7 @@ enum {
        OPT_DEBUG_INFO_DIR,
        OPT_DEBUG_INFO_FULL_PATH,
        OPT_DEBUG_INFO_TARGET_PREFIX,
+       OPT_END,
        OPT_FIELDS,
        OPT_HELP,
        OPT_HELP_LEGACY,
@@ -2135,6 +2142,7 @@ enum {
        OPT_SINK,
        OPT_SOURCE,
        OPT_STREAM_INTERSECTION,
+       OPT_TIMERANGE,
        OPT_VERBOSE,
        OPT_VERSION,
 };
@@ -2143,6 +2151,7 @@ enum {
 static struct poptOption long_options[] = {
        /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
        { "base-params", 'b', POPT_ARG_STRING, NULL, OPT_BASE_PARAMS, NULL, NULL },
+       { "begin", '\0', POPT_ARG_STRING, NULL, OPT_BEGIN, NULL, NULL },
        { "clock-cycles", '\0', POPT_ARG_NONE, NULL, OPT_CLOCK_CYCLES, NULL, NULL },
        { "clock-date", '\0', POPT_ARG_NONE, NULL, OPT_CLOCK_DATE, NULL, NULL },
        { "clock-force-correlate", '\0', POPT_ARG_NONE, NULL, OPT_CLOCK_FORCE_CORRELATE, NULL, NULL },
@@ -2154,6 +2163,7 @@ static struct poptOption long_options[] = {
        { "debug-info-dir", 0, POPT_ARG_STRING, NULL, OPT_DEBUG_INFO_DIR, NULL, NULL },
        { "debug-info-full-path", 0, POPT_ARG_NONE, NULL, OPT_DEBUG_INFO_FULL_PATH, NULL, NULL },
        { "debug-info-target-prefix", 0, POPT_ARG_STRING, NULL, OPT_DEBUG_INFO_TARGET_PREFIX, NULL, NULL },
+       { "end", '\0', POPT_ARG_STRING, NULL, OPT_END, NULL, NULL },
        { "fields", 'f', POPT_ARG_STRING, NULL, OPT_FIELDS, NULL, NULL },
        { "help", 'h', POPT_ARG_NONE, NULL, OPT_HELP, NULL, NULL },
        { "help-legacy", '\0', POPT_ARG_NONE, NULL, OPT_HELP_LEGACY, NULL, NULL },
@@ -2170,6 +2180,7 @@ static struct poptOption long_options[] = {
        { "sink", '\0', POPT_ARG_STRING, NULL, OPT_SINK, NULL, NULL },
        { "source", '\0', POPT_ARG_STRING, NULL, OPT_SOURCE, NULL, NULL },
        { "stream-intersection", '\0', POPT_ARG_NONE, NULL, OPT_STREAM_INTERSECTION, NULL, NULL },
+       { "timerange", '\0', POPT_ARG_STRING, NULL, OPT_TIMERANGE, NULL, NULL },
        { "verbose", 'v', POPT_ARG_NONE, NULL, OPT_VERBOSE, NULL, NULL },
        { "version", 'V', POPT_ARG_NONE, NULL, OPT_VERSION, NULL, NULL },
        { NULL, 0, 0, NULL, 0, NULL, NULL },
@@ -2204,6 +2215,39 @@ static void add_cfg_comp(struct bt_config *cfg,
        }
 }
 
+static int split_timerange(const char *arg, const char **begin, const char **end)
+{
+       const char *c;
+
+       /* Try to match [begin,end] */
+       c = strchr(arg, '[');
+       if (!c)
+               goto skip;
+       *begin = ++c;
+       c = strchr(c, ',');
+       if (!c)
+               goto skip;
+       *end = ++c;
+       c = strchr(c, ']');
+       if (!c)
+               goto skip;
+       goto found;
+
+skip:
+       /* Try to match begin,end */
+       c = arg;
+       *begin = c;
+       c = strchr(c, ',');
+       if (!c)
+               goto not_found;
+       *end = ++c;
+       /* fall-through */
+found:
+       return 0;
+not_found:
+       return -1;
+}
+
 /*
  * Returns a Babeltrace configuration, out of command-line arguments,
  * containing everything that is needed to instanciate specific
@@ -2225,12 +2269,14 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
        enum legacy_output_format legacy_output_format =
                LEGACY_OUTPUT_FORMAT_NONE;
        struct bt_value *legacy_input_paths = NULL;
+       struct bt_config_component *implicit_source_comp = NULL;
        struct bt_config_component *cur_cfg_comp = NULL;
+       bool cur_is_implicit_source = false;
+       bool use_implicit_source = false;
        enum bt_config_component_dest cur_cfg_comp_dest =
                BT_CONFIG_COMPONENT_DEST_SOURCE;
        struct bt_value *cur_base_params = NULL;
        int opt;
-       bool cur_cfg_comp_params_set = false;
 
        memset(&ctf_legacy_opts, 0, sizeof(ctf_legacy_opts));
        memset(&text_legacy_opts, 0, sizeof(text_legacy_opts));
@@ -2291,6 +2337,17 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                goto error;
        }
 
+       /* Note: implicit source never gets positional base params. */
+       implicit_source_comp = bt_config_component_from_arg(DEFAULT_SOURCE_COMPONENT_NAME);
+       if (implicit_source_comp) {
+               cur_cfg_comp = implicit_source_comp;
+               cur_is_implicit_source = true;
+               use_implicit_source = true;
+       } else {
+               printf_debug("Cannot find implicit source plugin \"%s\"",
+                       DEFAULT_SOURCE_COMPONENT_NAME);
+       }
+
        /* Parse options */
        pc = poptGetContext(NULL, argc, (const char **) argv, long_options, 0);
        if (!pc) {
@@ -2367,8 +2424,10 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                                }
                        }
 
+                       use_implicit_source = false;
+
                        /* Non-legacy: try to create a component config */
-                       if (cur_cfg_comp) {
+                       if (cur_cfg_comp && !cur_is_implicit_source) {
                                add_cfg_comp(cfg, cur_cfg_comp,
                                        cur_cfg_comp_dest);
                        }
@@ -2379,6 +2438,7 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                                        arg);
                                goto error;
                        }
+                       cur_is_implicit_source = false;
 
                        assert(cur_base_params);
                        bt_put(cur_cfg_comp->params);
@@ -2389,7 +2449,6 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                        }
 
                        cur_cfg_comp_dest = BT_CONFIG_COMPONENT_DEST_SOURCE;
-                       cur_cfg_comp_params_set = false;
                        break;
                }
                case OPT_OUTPUT_FORMAT:
@@ -2430,7 +2489,7 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                        }
 
                        /* Non-legacy: try to create a component config */
-                       if (cur_cfg_comp) {
+                       if (cur_cfg_comp && !cur_is_implicit_source) {
                                add_cfg_comp(cfg, cur_cfg_comp,
                                        cur_cfg_comp_dest);
                        }
@@ -2441,6 +2500,7 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                                        arg);
                                goto error;
                        }
+                       cur_is_implicit_source = false;
 
                        assert(cur_base_params);
                        bt_put(cur_cfg_comp->params);
@@ -2451,7 +2511,6 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                        }
 
                        cur_cfg_comp_dest = BT_CONFIG_COMPONENT_DEST_SINK;
-                       cur_cfg_comp_params_set = false;
                        break;
                }
                case OPT_PARAMS:
@@ -2460,14 +2519,8 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                        struct bt_value *params_to_set;
 
                        if (!cur_cfg_comp) {
-                               printf_err("--params option must follow a --source or --sink option\n");
-                               goto error;
-                       }
-
-                       if (cur_cfg_comp_params_set) {
-                               printf_err("Duplicate --params option for the same current component\ninstance (class %s.%s)\n",
-                                       cur_cfg_comp->plugin_name->str,
-                                       cur_cfg_comp->component_name->str);
+                               printf_err("Can not apply parameter to unavailable default source component \"%s\".\n",
+                                       DEFAULT_SOURCE_COMPONENT_NAME);
                                goto error;
                        }
 
@@ -2478,22 +2531,22 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                                goto error;
                        }
 
-                       params_to_set = bt_value_map_extend(cur_base_params,
+                       params_to_set = bt_value_map_extend(cur_cfg_comp->params,
                                params);
                        BT_PUT(params);
                        if (!params_to_set) {
-                               printf_err("Cannot extend current base parameters with --params option's argument:\n    %s\n",
+                               printf_err("Cannot extend current component parameters with --params option's argument:\n    %s\n",
                                        arg);
                                goto error;
                        }
 
                        BT_MOVE(cur_cfg_comp->params, params_to_set);
-                       cur_cfg_comp_params_set = true;
                        break;
                }
                case OPT_PATH:
                        if (!cur_cfg_comp) {
-                               printf_err("--path option must follow a --source or --sink option\n");
+                               printf_err("Can not apply parameter to unavailable default source component \"%s\".\n",
+                                       DEFAULT_SOURCE_COMPONENT_NAME);
                                goto error;
                        }
 
@@ -2608,6 +2661,67 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
                case OPT_CLOCK_FORCE_CORRELATE:
                        cfg->force_correlate = true;
                        break;
+               case OPT_BEGIN:
+                       if (!cur_cfg_comp) {
+                               printf_err("Can not apply parameter to unavailable default source component \"%s\".\n",
+                                       DEFAULT_SOURCE_COMPONENT_NAME);
+                               goto error;
+                       }
+                       if (cur_cfg_comp_dest != BT_CONFIG_COMPONENT_DEST_SOURCE) {
+                               printf_err("--begin option must follow a --source option\n");
+                               goto error;
+                       }
+                       if (bt_value_map_insert_string(cur_cfg_comp->params,
+                                       "begin", arg)) {
+                               print_err_oom();
+                               goto error;
+                       }
+                       break;
+               case OPT_END:
+                       if (!cur_cfg_comp) {
+                               printf_err("Can not apply parameter to unavailable default source component \"%s\".\n",
+                                       DEFAULT_SOURCE_COMPONENT_NAME);
+                               goto error;
+                       }
+                       if (cur_cfg_comp_dest != BT_CONFIG_COMPONENT_DEST_SOURCE) {
+                               printf_err("--end option must follow a --source option\n");
+                               goto error;
+                       }
+                       if (bt_value_map_insert_string(cur_cfg_comp->params,
+                                       "end", arg)) {
+                               print_err_oom();
+                               goto error;
+                       }
+                       break;
+               case OPT_TIMERANGE:
+               {
+                       const char *begin, *end;
+
+                       if (!cur_cfg_comp) {
+                               printf_err("Can not apply parameter to unavailable default source component \"%s\".\n",
+                                       DEFAULT_SOURCE_COMPONENT_NAME);
+                               goto error;
+                       }
+                       if (cur_cfg_comp_dest != BT_CONFIG_COMPONENT_DEST_SOURCE) {
+                               printf_err("--timerange option must follow a --source option\n");
+                               goto error;
+                       }
+                       if (split_timerange(arg, &begin, &end)) {
+                               printf_err("Invalid --timerange format, expecting: begin,end or [begin,end] (where [] are actual brackets)\n");
+                               goto error;
+                       }
+                       if (bt_value_map_insert_string(cur_cfg_comp->params,
+                                       "begin", begin)) {
+                               print_err_oom();
+                               goto error;
+                       }
+                       if (bt_value_map_insert_string(cur_cfg_comp->params,
+                                       "end", end)) {
+                               print_err_oom();
+                               goto error;
+                       }
+                       break;
+               }
                case OPT_HELP:
                        BT_PUT(cfg);
                        print_usage(stdout);
@@ -2640,9 +2754,21 @@ struct bt_config *bt_config_from_args(int argc, const char *argv[], int *exit_co
        }
 
        /* Append current component configuration, if any */
-       if (cur_cfg_comp) {
+       if (cur_cfg_comp && !cur_is_implicit_source) {
                add_cfg_comp(cfg, cur_cfg_comp, cur_cfg_comp_dest);
-               cur_cfg_comp = NULL;
+       }
+       cur_cfg_comp = NULL;
+
+       if (use_implicit_source) {
+               add_cfg_comp(cfg, implicit_source_comp,
+                       BT_CONFIG_COMPONENT_DEST_SOURCE);
+               implicit_source_comp = NULL;
+       } else {
+               if (implicit_source_comp
+                               && !bt_value_map_is_empty(implicit_source_comp->params)) {
+                       printf_err("Arguments specified for implicit source, but an explicit source has been specified, overriding it\n");
+                       goto error;
+               }
        }
 
        /* Check for option parsing error */
@@ -2725,6 +2851,7 @@ end:
        }
 
        free(arg);
+       BT_PUT(implicit_source_comp);
        BT_PUT(cur_cfg_comp);
        BT_PUT(cur_base_params);
        BT_PUT(text_legacy_opts.names);
This page took 0.02827 seconds and 4 git commands to generate.