X-Git-Url: http://git.efficios.com/?p=argpar.git;a=blobdiff_plain;f=argpar%2Fargpar.c;h=45658a5b3e914e538cb47848281a1ffcdc1b178d;hp=7f4e46b3d758332eff529b1396cfb3d4c578e59e;hb=4d6198b505eaeb2e60a14331ccb52f90ea397bb4;hpb=903a5b8ab5ab38d3b200b1d692ba0d29d080c92c diff --git a/argpar/argpar.c b/argpar/argpar.c index 7f4e46b..45658a5 100644 --- a/argpar/argpar.c +++ b/argpar/argpar.c @@ -1,71 +1,275 @@ /* - * Copyright 2019 Philippe Proulx + * SPDX-License-Identifier: MIT * - * Permission is hereby granted, free of charge, to any person obtaining a copy - * of this software and associated documentation files (the "Software"), to deal - * in the Software without restriction, including without limitation the rights - * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell - * copies of the Software, and to permit persons to whom the Software is - * furnished to do so, subject to the following conditions: - * - * The above copyright notice and this permission notice shall be included in - * all copies or substantial portions of the Software. - * - * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR - * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, - * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE - * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER - * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, - * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE - * SOFTWARE. + * Copyright (c) 2019-2021 Philippe Proulx + * Copyright (c) 2020-2021 Simon Marchi */ +#include +#include #include +#include #include #include -#include - -#include "common/assert.h" -#include "common/common.h" #include "argpar.h" -static -void destroy_item(struct bt_argpar_item * const item) +#define ARGPAR_REALLOC(_ptr, _type, _nmemb) \ + ((_type *) realloc(_ptr, (_nmemb) * sizeof(_type))) + +#define ARGPAR_CALLOC(_type, _nmemb) \ + ((_type *) calloc((_nmemb), sizeof(_type))) + +#define ARGPAR_ZALLOC(_type) ARGPAR_CALLOC(_type, 1) + +#define ARGPAR_ASSERT(_cond) assert(_cond) + +#ifdef __MINGW_PRINTF_FORMAT +# define ARGPAR_PRINTF_FORMAT __MINGW_PRINTF_FORMAT +#else +# define ARGPAR_PRINTF_FORMAT printf +#endif + +/* + * An argpar iterator. + * + * Such a structure contains the state of an iterator between calls to + * argpar_iter_next(). + */ +struct argpar_iter { + /* + * Data provided by the user to argpar_iter_create(); immutable + * afterwards. + */ + unsigned int argc; + const char * const *argv; + const struct argpar_opt_descr *descrs; + + /* + * Index of the argument to process in the next + * argpar_iter_next() call. + */ + unsigned int i; + + /* Counter of non-option arguments */ + int non_opt_index; + + /* + * Current character of the current short option group: if it's + * not `NULL`, the parser is in within a short option group, + * therefore it must resume there in the next + * argpar_iter_next() call. + */ + const char *short_opt_ch; + + /* Temporary character buffer which only grows */ + struct { + size_t size; + char *data; + } tmp_buf; +}; + +/* Base parsing item */ +struct argpar_item { + enum argpar_item_type type; +}; + +/* Option parsing item */ +struct argpar_item_opt { + struct argpar_item base; + + /* Corresponding descriptor */ + const struct argpar_opt_descr *descr; + + /* Argument, or `NULL` if none; owned by this */ + char *arg; +}; + +/* Non-option parsing item */ +struct argpar_item_non_opt { + struct argpar_item base; + + /* + * Complete argument, pointing to one of the entries of the + * original arguments (`argv`). + */ + const char *arg; + + /* + * Index of this argument amongst all original arguments + * (`argv`). + */ + unsigned int orig_index; + + /* Index of this argument amongst other non-option arguments */ + unsigned int non_opt_index; +}; + +static __attribute__((format(ARGPAR_PRINTF_FORMAT, 1, 0))) +char *argpar_vasprintf(const char * const fmt, va_list args) +{ + int len1, len2; + char *str; + va_list args2; + + va_copy(args2, args); + len1 = vsnprintf(NULL, 0, fmt, args); + if (len1 < 0) { + str = NULL; + goto end; + } + + str = malloc(len1 + 1); + if (!str) { + goto end; + } + + len2 = vsnprintf(str, len1 + 1, fmt, args2); + ARGPAR_ASSERT(len1 == len2); + +end: + va_end(args2); + return str; +} + + +static __attribute__((format(ARGPAR_PRINTF_FORMAT, 1, 2))) +char *argpar_asprintf(const char * const fmt, ...) +{ + va_list args; + char *str; + + va_start(args, fmt); + str = argpar_vasprintf(fmt, args); + va_end(args); + return str; +} + +static __attribute__((format(ARGPAR_PRINTF_FORMAT, 2, 3))) +bool try_append_string_printf(char ** const str, const char *fmt, ...) +{ + char *new_str = NULL; + char *addendum = NULL; + bool success; + va_list args; + + if (!str) { + success = true; + goto end; + } + + ARGPAR_ASSERT(str); + va_start(args, fmt); + addendum = argpar_vasprintf(fmt, args); + va_end(args); + + if (!addendum) { + success = false; + goto end; + } + + new_str = argpar_asprintf("%s%s", *str ? *str : "", addendum); + if (!new_str) { + success = false; + goto end; + } + + free(*str); + *str = new_str; + success = true; + +end: + free(addendum); + return success; +} + +ARGPAR_HIDDEN +enum argpar_item_type argpar_item_type(const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + return item->type; +} + +ARGPAR_HIDDEN +const struct argpar_opt_descr *argpar_item_opt_descr( + const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + ARGPAR_ASSERT(item->type == ARGPAR_ITEM_TYPE_OPT); + return ((const struct argpar_item_opt *) item)->descr; +} + +ARGPAR_HIDDEN +const char *argpar_item_opt_arg(const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + ARGPAR_ASSERT(item->type == ARGPAR_ITEM_TYPE_OPT); + return ((const struct argpar_item_opt *) item)->arg; +} + +ARGPAR_HIDDEN +const char *argpar_item_non_opt_arg(const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + ARGPAR_ASSERT(item->type == ARGPAR_ITEM_TYPE_NON_OPT); + return ((const struct argpar_item_non_opt *) item)->arg; +} + +ARGPAR_HIDDEN +unsigned int argpar_item_non_opt_orig_index( + const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + ARGPAR_ASSERT(item->type == ARGPAR_ITEM_TYPE_NON_OPT); + return ((const struct argpar_item_non_opt *) item)->orig_index; +} + +ARGPAR_HIDDEN +unsigned int argpar_item_non_opt_non_opt_index( + const struct argpar_item * const item) +{ + ARGPAR_ASSERT(item); + ARGPAR_ASSERT(item->type == ARGPAR_ITEM_TYPE_NON_OPT); + return ((const struct argpar_item_non_opt *) item)->non_opt_index; +} + +ARGPAR_HIDDEN +void argpar_item_destroy(const struct argpar_item * const item) { if (!item) { goto end; } - if (item->type == BT_ARGPAR_ITEM_TYPE_OPT) { - struct bt_argpar_item_opt * const opt_item = (void *) item; + if (item->type == ARGPAR_ITEM_TYPE_OPT) { + struct argpar_item_opt * const opt_item = + (struct argpar_item_opt *) item; - g_free((void *) opt_item->arg); + free(opt_item->arg); } - g_free(item); + free((void *) item); end: return; } static -struct bt_argpar_item_opt *create_opt_item( - const struct bt_argpar_opt_descr * const descr, +struct argpar_item_opt *create_opt_item( + const struct argpar_opt_descr * const descr, const char * const arg) { - struct bt_argpar_item_opt *opt_item = - g_new0(struct bt_argpar_item_opt, 1); + struct argpar_item_opt *opt_item = + ARGPAR_ZALLOC(struct argpar_item_opt); if (!opt_item) { goto end; } - opt_item->base.type = BT_ARGPAR_ITEM_TYPE_OPT; + opt_item->base.type = ARGPAR_ITEM_TYPE_OPT; opt_item->descr = descr; if (arg) { - opt_item->arg = g_strdup(arg); + opt_item->arg = strdup(arg); if (!opt_item->arg) { goto error; } @@ -74,7 +278,7 @@ struct bt_argpar_item_opt *create_opt_item( goto end; error: - destroy_item(&opt_item->base); + argpar_item_destroy(&opt_item->base); opt_item = NULL; end: @@ -82,18 +286,18 @@ end: } static -struct bt_argpar_item_non_opt *create_non_opt_item(const char * const arg, +struct argpar_item_non_opt *create_non_opt_item(const char * const arg, const unsigned int orig_index, const unsigned int non_opt_index) { - struct bt_argpar_item_non_opt * const non_opt_item = - g_new0(struct bt_argpar_item_non_opt, 1); + struct argpar_item_non_opt * const non_opt_item = + ARGPAR_ZALLOC(struct argpar_item_non_opt); if (!non_opt_item) { goto end; } - non_opt_item->base.type = BT_ARGPAR_ITEM_TYPE_NON_OPT; + non_opt_item->base.type = ARGPAR_ITEM_TYPE_NON_OPT; non_opt_item->arg = arg; non_opt_item->orig_index = orig_index; non_opt_item->non_opt_index = non_opt_index; @@ -103,11 +307,11 @@ end: } static -const struct bt_argpar_opt_descr *find_descr( - const struct bt_argpar_opt_descr * const descrs, +const struct argpar_opt_descr *find_descr( + const struct argpar_opt_descr * const descrs, const char short_name, const char * const long_name) { - const struct bt_argpar_opt_descr *descr; + const struct argpar_opt_descr *descr; for (descr = descrs; descr->short_name || descr->long_name; descr++) { if (short_name && descr->short_name && @@ -127,86 +331,90 @@ end: enum parse_orig_arg_opt_ret { PARSE_ORIG_ARG_OPT_RET_OK, - PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT = -2, - PARSE_ORIG_ARG_OPT_RET_ERROR = -1, + PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT = -1, + PARSE_ORIG_ARG_OPT_RET_ERROR_MISSING_OPT_ARG = -2, + PARSE_ORIG_ARG_OPT_RET_ERROR_UNEXPECTED_OPT_ARG = -4, + PARSE_ORIG_ARG_OPT_RET_ERROR_MEMORY = -5, }; static enum parse_orig_arg_opt_ret parse_short_opts(const char * const short_opts, const char * const next_orig_arg, - const struct bt_argpar_opt_descr * const descrs, - struct bt_argpar_parse_ret * const parse_ret, - bool * const used_next_orig_arg) + const struct argpar_opt_descr * const descrs, + struct argpar_iter * const iter, + char ** const error, struct argpar_item ** const item) { enum parse_orig_arg_opt_ret ret = PARSE_ORIG_ARG_OPT_RET_OK; - const char *short_opt_ch = short_opts; + bool used_next_orig_arg = false; + const char *opt_arg = NULL; + const struct argpar_opt_descr *descr; + struct argpar_item_opt *opt_item; - if (strlen(short_opts) == 0) { - g_string_append(parse_ret->error, "Invalid argument"); - goto error; - } + ARGPAR_ASSERT(strlen(short_opts) != 0); - while (*short_opt_ch) { - const char *opt_arg = NULL; - const struct bt_argpar_opt_descr *descr; - struct bt_argpar_item_opt *opt_item; - - /* Find corresponding option descriptor */ - descr = find_descr(descrs, *short_opt_ch, NULL); - if (!descr) { - ret = PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT; - g_string_append_printf(parse_ret->error, - "Unknown option `-%c`", *short_opt_ch); - goto error; - } + if (!iter->short_opt_ch) { + iter->short_opt_ch = short_opts; + } - if (descr->with_arg) { - if (short_opt_ch[1]) { - /* `-oarg` form */ - opt_arg = &short_opt_ch[1]; - } else { - /* `-o arg` form */ - opt_arg = next_orig_arg; - *used_next_orig_arg = true; - } + /* Find corresponding option descriptor */ + descr = find_descr(descrs, *iter->short_opt_ch, NULL); + if (!descr) { + try_append_string_printf(error, "Unknown option `-%c`", + *iter->short_opt_ch); + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT; + goto error; + } - /* - * We accept `-o ''` (empty option's argument), - * but not `-o` alone if an option's argument is - * expected. - */ - if (!opt_arg || (short_opt_ch[1] && strlen(opt_arg) == 0)) { - g_string_append_printf(parse_ret->error, - "Missing required argument for option `-%c`", - *short_opt_ch); - *used_next_orig_arg = false; - goto error; - } + if (descr->with_arg) { + if (iter->short_opt_ch[1]) { + /* `-oarg` form */ + opt_arg = &iter->short_opt_ch[1]; + } else { + /* `-o arg` form */ + opt_arg = next_orig_arg; + used_next_orig_arg = true; } - /* Create and append option argument */ - opt_item = create_opt_item(descr, opt_arg); - if (!opt_item) { + /* + * We accept `-o ''` (empty option argument), but not + * `-o` alone if an option argument is expected. + */ + if (!opt_arg || (iter->short_opt_ch[1] && + strlen(opt_arg) == 0)) { + try_append_string_printf(error, + "Missing required argument for option `-%c`", + *iter->short_opt_ch); + used_next_orig_arg = false; + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_MISSING_OPT_ARG; goto error; } + } + + /* Create and append option argument */ + opt_item = create_opt_item(descr, opt_arg); + if (!opt_item) { + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_MEMORY; + goto error; + } - g_ptr_array_add(parse_ret->items, opt_item); + *item = &opt_item->base; + iter->short_opt_ch++; - if (descr->with_arg) { - /* Option has an argument: no more options */ - break; - } + if (descr->with_arg || !*iter->short_opt_ch) { + /* Option has an argument: no more options */ + iter->short_opt_ch = NULL; - /* Go to next short option */ - short_opt_ch++; + if (used_next_orig_arg) { + iter->i += 2; + } else { + iter->i++; + } } goto end; error: - if (ret == PARSE_ORIG_ARG_OPT_RET_OK) { - ret = PARSE_ORIG_ARG_OPT_RET_ERROR; - } + ARGPAR_ASSERT(ret != PARSE_ORIG_ARG_OPT_RET_OK); end: return ret; @@ -215,14 +423,14 @@ end: static enum parse_orig_arg_opt_ret parse_long_opt(const char * const long_opt_arg, const char * const next_orig_arg, - const struct bt_argpar_opt_descr * const descrs, - struct bt_argpar_parse_ret * const parse_ret, - bool * const used_next_orig_arg) + const struct argpar_opt_descr * const descrs, + struct argpar_iter * const iter, + char ** const error, struct argpar_item ** const item) { - const size_t max_len = 127; enum parse_orig_arg_opt_ret ret = PARSE_ORIG_ARG_OPT_RET_OK; - const struct bt_argpar_opt_descr *descr; - struct bt_argpar_item_opt *opt_item; + const struct argpar_opt_descr *descr; + struct argpar_item_opt *opt_item; + bool used_next_orig_arg = false; /* Option's argument, if any */ const char *opt_arg = NULL; @@ -230,16 +438,10 @@ enum parse_orig_arg_opt_ret parse_long_opt(const char * const long_opt_arg, /* Position of first `=`, if any */ const char *eq_pos; - /* Buffer holding option name when `long_opt_arg` contains `=` */ - char buf[max_len + 1]; - /* Option name */ const char *long_opt_name = long_opt_arg; - if (strlen(long_opt_arg) == 0) { - g_string_append(parse_ret->error, "Invalid argument"); - goto error; - } + ARGPAR_ASSERT(strlen(long_opt_arg) != 0); /* Find the first `=` in original argument */ eq_pos = strchr(long_opt_arg, '='); @@ -247,22 +449,26 @@ enum parse_orig_arg_opt_ret parse_long_opt(const char * const long_opt_arg, const size_t long_opt_name_size = eq_pos - long_opt_arg; /* Isolate the option name */ - if (long_opt_name_size > max_len) { - g_string_append_printf(parse_ret->error, - "Invalid argument `--%s`", long_opt_arg); - goto error; + while (long_opt_name_size > iter->tmp_buf.size - 1) { + iter->tmp_buf.size *= 2; + iter->tmp_buf.data = ARGPAR_REALLOC(iter->tmp_buf.data, + char, iter->tmp_buf.size); + if (!iter->tmp_buf.data) { + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_MEMORY; + goto error; + } } - memcpy(buf, long_opt_arg, long_opt_name_size); - buf[long_opt_name_size] = '\0'; - long_opt_name = buf; + memcpy(iter->tmp_buf.data, long_opt_arg, long_opt_name_size); + iter->tmp_buf.data[long_opt_name_size] = '\0'; + long_opt_name = iter->tmp_buf.data; } /* Find corresponding option descriptor */ descr = find_descr(descrs, '\0', long_opt_name); if (!descr) { - g_string_append_printf(parse_ret->error, - "Unknown option `--%s`", long_opt_name); + try_append_string_printf(error, "Unknown option `--%s`", + long_opt_name); ret = PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT; goto error; } @@ -275,15 +481,25 @@ enum parse_orig_arg_opt_ret parse_long_opt(const char * const long_opt_arg, } else { /* `--long-opt arg` style */ if (!next_orig_arg) { - g_string_append_printf(parse_ret->error, + try_append_string_printf(error, "Missing required argument for option `--%s`", long_opt_name); + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_MISSING_OPT_ARG; goto error; } opt_arg = next_orig_arg; - *used_next_orig_arg = true; + used_next_orig_arg = true; } + } else if (eq_pos) { + /* + * Unexpected `--opt=arg` style for a long option which + * doesn't accept an argument. + */ + try_append_string_printf(error, + "Unexpected argument for option `--%s`", long_opt_name); + ret = PARSE_ORIG_ARG_OPT_RET_ERROR_UNEXPECTED_OPT_ARG; + goto error; } /* Create and append option argument */ @@ -292,13 +508,17 @@ enum parse_orig_arg_opt_ret parse_long_opt(const char * const long_opt_arg, goto error; } - g_ptr_array_add(parse_ret->items, opt_item); + if (used_next_orig_arg) { + iter->i += 2; + } else { + iter->i++; + } + + *item = &opt_item->base; goto end; error: - if (ret == PARSE_ORIG_ARG_OPT_RET_OK) { - ret = PARSE_ORIG_ARG_OPT_RET_ERROR; - } + ARGPAR_ASSERT(ret != PARSE_ORIG_ARG_OPT_RET_OK); end: return ret; @@ -307,159 +527,177 @@ end: static enum parse_orig_arg_opt_ret parse_orig_arg_opt(const char * const orig_arg, const char * const next_orig_arg, - const struct bt_argpar_opt_descr * const descrs, - struct bt_argpar_parse_ret * const parse_ret, - bool * const used_next_orig_arg) + const struct argpar_opt_descr * const descrs, + struct argpar_iter * const iter, char ** const error, + struct argpar_item ** const item) { enum parse_orig_arg_opt_ret ret = PARSE_ORIG_ARG_OPT_RET_OK; - BT_ASSERT(orig_arg[0] == '-'); + ARGPAR_ASSERT(orig_arg[0] == '-'); if (orig_arg[1] == '-') { /* Long option */ ret = parse_long_opt(&orig_arg[2], - next_orig_arg, descrs, parse_ret, - used_next_orig_arg); + next_orig_arg, descrs, iter, error, item); } else { /* Short option */ ret = parse_short_opts(&orig_arg[1], - next_orig_arg, descrs, parse_ret, - used_next_orig_arg); + next_orig_arg, descrs, iter, error, item); } return ret; } static -void prepend_while_parsing_arg_to_error(GString * const error, +bool try_prepend_while_parsing_arg_to_error(char ** const error, const unsigned int i, const char * const arg) { - /* 🙁 There's no g_string_prepend_printf()! */ - GString * const tmp_str = g_string_new(NULL); + char *new_error; + bool success; - BT_ASSERT(error); - BT_ASSERT(arg); + if (!error) { + success = true; + goto end; + } - if (!tmp_str) { + ARGPAR_ASSERT(*error); + new_error = argpar_asprintf("While parsing argument #%u (`%s`): %s", + i + 1, arg, *error); + if (!new_error) { + success = false; goto end; } - g_string_append_printf(tmp_str, "While parsing argument #%u (`%s`): %s", - i + 1, arg, error->str); - g_string_assign(error, tmp_str->str); - g_string_free(tmp_str, TRUE); + free(*error); + *error = new_error; + success = true; end: - return; + return success; } -BT_HIDDEN -struct bt_argpar_parse_ret bt_argpar_parse(unsigned int argc, - const char * const *argv, - const struct bt_argpar_opt_descr * const descrs, - bool fail_on_unknown_opt) +ARGPAR_HIDDEN +struct argpar_iter *argpar_iter_create(const unsigned int argc, + const char * const * const argv, + const struct argpar_opt_descr * const descrs) { - struct bt_argpar_parse_ret parse_ret = { 0 }; - unsigned int i; - unsigned int non_opt_index = 0; + struct argpar_iter *iter = ARGPAR_ZALLOC(struct argpar_iter); - parse_ret.error = g_string_new(NULL); - if (!parse_ret.error) { - goto error; + if (!iter) { + goto end; } - parse_ret.items = g_ptr_array_new_with_free_func( - (GDestroyNotify) destroy_item); - if (!parse_ret.items) { - goto error; + iter->argc = argc; + iter->argv = argv; + iter->descrs = descrs; + iter->tmp_buf.size = 128; + iter->tmp_buf.data = ARGPAR_CALLOC(char, iter->tmp_buf.size); + if (!iter->tmp_buf.data) { + argpar_iter_destroy(iter); + iter = NULL; + goto end; } - for (i = 0; i < argc; i++) { - enum parse_orig_arg_opt_ret parse_orig_arg_opt_ret; - bool used_next_orig_arg = false; - const char * const orig_arg = argv[i]; - const char * const next_orig_arg = - i < argc - 1 ? argv[i + 1] : NULL; +end: + return iter; +} - if (orig_arg[0] != '-') { - /* Non-option argument */ - struct bt_argpar_item_non_opt *non_opt_item = - create_non_opt_item(orig_arg, i, non_opt_index); +ARGPAR_HIDDEN +void argpar_iter_destroy(struct argpar_iter * const iter) +{ + if (iter) { + free(iter->tmp_buf.data); + free(iter); + } +} - if (!non_opt_item) { - goto error; - } +ARGPAR_HIDDEN +enum argpar_iter_next_status argpar_iter_next( + struct argpar_iter * const iter, + const struct argpar_item ** const item, char ** const error) +{ + enum argpar_iter_next_status status; + enum parse_orig_arg_opt_ret parse_orig_arg_opt_ret; + const char *orig_arg; + const char *next_orig_arg; - non_opt_index++; - g_ptr_array_add(parse_ret.items, non_opt_item); - continue; - } + ARGPAR_ASSERT(iter->i <= iter->argc); - /* Option argument */ - parse_orig_arg_opt_ret = parse_orig_arg_opt(orig_arg, - next_orig_arg, descrs, &parse_ret, &used_next_orig_arg); - switch (parse_orig_arg_opt_ret) { - case PARSE_ORIG_ARG_OPT_RET_OK: - break; - case PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT: - BT_ASSERT(!used_next_orig_arg); + if (error) { + *error = NULL; + } - if (fail_on_unknown_opt) { - prepend_while_parsing_arg_to_error( - parse_ret.error, i, orig_arg); - goto error; - } + if (iter->i == iter->argc) { + status = ARGPAR_ITER_NEXT_STATUS_END; + goto end; + } + + orig_arg = iter->argv[iter->i]; + next_orig_arg = + iter->i < (iter->argc - 1) ? iter->argv[iter->i + 1] : NULL; - /* - * The current original argument is not - * considered ingested because it triggered an - * unknown option. - */ - parse_ret.ingested_orig_args = i; - g_string_free(parse_ret.error, TRUE); - parse_ret.error = NULL; + if (strcmp(orig_arg, "-") == 0 || strcmp(orig_arg, "--") == 0 || + orig_arg[0] != '-') { + /* Non-option argument */ + const struct argpar_item_non_opt * const non_opt_item = + create_non_opt_item(orig_arg, iter->i, + iter->non_opt_index); + + if (!non_opt_item) { + status = ARGPAR_ITER_NEXT_STATUS_ERROR_MEMORY; goto end; - case PARSE_ORIG_ARG_OPT_RET_ERROR: - prepend_while_parsing_arg_to_error( - parse_ret.error, i, orig_arg); - goto error; - default: - bt_common_abort(); } - if (used_next_orig_arg) { - i++; - } + iter->non_opt_index++; + iter->i++; + *item = &non_opt_item->base; + status = ARGPAR_ITER_NEXT_STATUS_OK; + goto end; } - parse_ret.ingested_orig_args = argc; - g_string_free(parse_ret.error, TRUE); - parse_ret.error = NULL; - goto end; + /* Option argument */ + parse_orig_arg_opt_ret = parse_orig_arg_opt(orig_arg, + next_orig_arg, iter->descrs, iter, error, + (struct argpar_item **) item); + switch (parse_orig_arg_opt_ret) { + case PARSE_ORIG_ARG_OPT_RET_OK: + status = ARGPAR_ITER_NEXT_STATUS_OK; + break; + case PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT: + case PARSE_ORIG_ARG_OPT_RET_ERROR_MISSING_OPT_ARG: + case PARSE_ORIG_ARG_OPT_RET_ERROR_UNEXPECTED_OPT_ARG: + try_prepend_while_parsing_arg_to_error(error, iter->i, + orig_arg); -error: - if (parse_ret.items) { - /* That's how we indicate that an error occured */ - g_ptr_array_free(parse_ret.items, TRUE); - parse_ret.items = NULL; + switch (parse_orig_arg_opt_ret) { + case PARSE_ORIG_ARG_OPT_RET_ERROR_UNKNOWN_OPT: + status = ARGPAR_ITER_NEXT_STATUS_ERROR_UNKNOWN_OPT; + break; + case PARSE_ORIG_ARG_OPT_RET_ERROR_MISSING_OPT_ARG: + status = ARGPAR_ITER_NEXT_STATUS_ERROR_MISSING_OPT_ARG; + break; + case PARSE_ORIG_ARG_OPT_RET_ERROR_UNEXPECTED_OPT_ARG: + status = ARGPAR_ITER_NEXT_STATUS_ERROR_UNEXPECTED_OPT_ARG; + break; + default: + abort(); + } + + break; + case PARSE_ORIG_ARG_OPT_RET_ERROR_MEMORY: + status = ARGPAR_ITER_NEXT_STATUS_ERROR_MEMORY; + break; + default: + abort(); } end: - return parse_ret; + return status; } -BT_HIDDEN -void bt_argpar_parse_ret_fini(struct bt_argpar_parse_ret *ret) +ARGPAR_HIDDEN +unsigned int argpar_iter_ingested_orig_args( + const struct argpar_iter * const iter) { - BT_ASSERT(ret); - - if (ret->items) { - g_ptr_array_free(ret->items, TRUE); - ret->items = NULL; - } - - if (ret->error) { - g_string_free(ret->error, TRUE); - ret->error = NULL; - } + return iter->i; }