X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=ucw%2Fopt.c;h=9ef721d672d64d72c68d88fdd94a4e7e65657bb9;hb=db2e817025638b9738e5ff8f8f5ccc1fbf68a2f3;hp=01c660389b5d01e6d4f9101153a4ddb68420de77;hpb=25896ca4296e3d4dee930845f9ec5c428f842eae;p=libucw.git diff --git a/ucw/opt.c b/ucw/opt.c index 01c66038..9ef721d6 100644 --- a/ucw/opt.c +++ b/ucw/opt.c @@ -1,5 +1,5 @@ /* - * UCW Library -- Parsing of command line options + * UCW Library -- Parsing of command-line options * * (c) 2013 Jan Moskyto Matejka * (c) 2014 Martin Mares @@ -10,65 +10,31 @@ #include #include -#include -#include -#include +#include +#include #include #include #include #include -/*** - * Value flags defaults - * ~~~~~~~~~~~~~~~~~~~~ - * - * OPT_NO_VALUE for OPT_BOOL, OPT_SWITCH and OPT_INC - * OPT_MAYBE_VALUE for OPT_STRING, OPT_UNS, OPT_INT - * Some of the value flags (OPT_NO_VALUE, OPT_MAYBE_VALUE, OPT_REQUIRED_VALUE) - * must be specified for OPT_CALL and OPT_USER. - ***/ -static uns opt_default_value_flags[] = { +static uint opt_default_value_flags[] = { [OPT_CL_BOOL] = OPT_NO_VALUE, [OPT_CL_STATIC] = OPT_MAYBE_VALUE, + [OPT_CL_MULTIPLE] = OPT_REQUIRED_VALUE, [OPT_CL_SWITCH] = OPT_NO_VALUE, [OPT_CL_INC] = OPT_NO_VALUE, [OPT_CL_CALL] = 0, - [OPT_CL_USER] = 0, [OPT_CL_SECTION] = 0, [OPT_CL_HELP] = 0 }; -struct opt_precomputed { - struct opt_item * item; - const char * name; - short flags; - short count; -}; - -struct opt_context { - struct opt_precomputed * opts; - struct opt_precomputed ** shortopt; - struct opt_item ** hooks_before_arg; - struct opt_item ** hooks_before_value; - struct opt_item ** hooks_after_value; - short opt_count; - short hooks_before_arg_count; - short hooks_before_value_count; - short hooks_after_value_count; - int positional_max; - int positional_count; -}; - -static void opt_failure(const char * mesg, ...) FORMAT_CHECK(printf,1,2) NONRET; -static void opt_failure(const char * mesg, ...) { +void opt_failure(const char * mesg, ...) { va_list args; va_start(args, mesg); vfprintf(stderr, mesg, args); - fprintf(stderr, "\n"); - opt_usage(); + fprintf(stderr, "\nRun with --help for more information.\n"); exit(OPT_EXIT_BAD_ARGS); - va_end(args); // FIXME: Does this make a sense after exit()? } static char *opt_name(struct opt_context *oc, struct opt_precomputed *opt) @@ -86,130 +52,38 @@ static char *opt_name(struct opt_context *oc, struct opt_precomputed *opt) #define THIS_OPT opt_name(oc, opt) -#define FOREACHLINE(text) for (const char * begin = (text), * end = (text); (*end) && (end = strchrnul(begin, '\n')); begin = end+1) - -static inline uns uns_min(uns x, uns y) +void opt_precompute(struct opt_precomputed *opt, struct opt_item *item) { - return MIN(x, y); -} - -void opt_help(const struct opt_section * help) { - int sections_cnt = 0; - int lines_cnt = 0; + opt->item = item; + opt->count = 0; + opt->name = item->name; + uint flags = item->flags; - for (struct opt_item * item = help->opt; item->cls != OPT_CL_END; item++) { - if (item->flags & OPT_NO_HELP) continue; - if (item->cls == OPT_CL_SECTION) { - sections_cnt++; - continue; - } - if (!*(item->help)) { - lines_cnt++; - continue; - } - FOREACHLINE(item->help) - lines_cnt++; + if (item->letter >= OPT_POSITIONAL_TAIL) { + flags &= ~OPT_VALUE_FLAGS; + flags |= OPT_REQUIRED_VALUE; + } + if (!(flags & OPT_VALUE_FLAGS)) { + ASSERT(item->cls != OPT_CL_CALL); + flags |= opt_default_value_flags[item->cls]; } - struct opt_sectlist { - int pos; - struct opt_section * sect; - } sections[sections_cnt]; - int s = 0; - - const char *lines[lines_cnt][3]; - memset(lines, 0, sizeof(lines)); - int line = 0; - - int linelengths[3] = { -1, -1, -1 }; - - for (struct opt_item * item = help->opt; item->cls != OPT_CL_END; item++) { - if (item->flags & OPT_NO_HELP) continue; - - if (item->cls == OPT_CL_HELP) { - if (!*(item->help)) { - line++; - continue; - } -#define SPLITLINES(text) do { \ - FOREACHLINE(text) { \ - int cell = 0; \ - for (const char * b = begin, * e = begin; (e < end) && (e = strchrnul(b, '\t')) && (e > end ? (e = end) : end); b = e+1) { \ - lines[line][cell] = b; \ - if (cell >= 2) \ - break; \ - else \ - if (*e == '\t' && linelengths[cell] < (e - b)) \ - linelengths[cell] = e-b; \ - cell++; \ - } \ - line++; \ - } } while (0) - SPLITLINES(item->help); - continue; - } - - if (item->cls == OPT_CL_SECTION) { - sections[s++] = (struct opt_sectlist) { .pos = line, .sect = item->u.section }; - continue; - } - - uns valoff = strchrnul(item->help, '\t') - item->help; - uns eol = strchrnul(item->help, '\n') - item->help; - if (valoff > eol) - valoff = eol; -#define VAL(it) ((it->flags & OPT_REQUIRED_VALUE) ? stk_printf("=%.*s", valoff, item->help) : ((it->flags & OPT_NO_VALUE) ? "" : stk_printf("(=%.*s)", valoff, item->help))) - if (item->name) { - lines[line][1] = stk_printf("--%s%s", item->name, VAL(item)); - if (linelengths[1] < (int) strlen(lines[line][1])) - linelengths[1] = strlen(lines[line][1]); - lines[line][0] = ""; - if (linelengths[0] < 0) - linelengths[0] = 0; - } - if (item->letter) { - lines[line][0] = stk_printf("-%c,", item->letter); - if (linelengths[0] < (int) strlen(lines[line][0])) - linelengths[0] = strlen(lines[line][0]); - } -#undef VAL - - if (eol > valoff) { - lines[line][2] = item->help + valoff + 1; - } - - line++; + opt->flags = flags; +} - if (*(item->help + eol)) - SPLITLINES(item->help + eol + 1); - } -#undef SPLITLINES - - s = 0; -#define FIELD(k) linelengths[k], uns_min(strchrnul(lines[i][k], '\t') - lines[i][k], strchrnul(lines[i][k], '\n') - lines[i][k]), lines[i][k] -#define LASTFIELD(k) uns_min(strchrnul(lines[i][k], '\t') - lines[i][k], strchrnul(lines[i][k], '\n') - lines[i][k]), lines[i][k] - for (int i=0;ihook_count; i++) { + struct opt_item *hook = oc->hooks[i]; + if (hook->flags & event) { + void *data = (hook->flags & OPT_HOOK_INTERNAL) ? oc : hook->ptr; + hook->u.hook(item, event, value, data); } - if (lines[i][0] == NULL) - printf("\n"); - else if (linelengths[0] == -1 || lines[i][1] == NULL) - printf("%.*s\n", LASTFIELD(0)); - else if (linelengths[1] == -1 || lines[i][2] == NULL) - printf("%-*.*s %.*s\n", FIELD(0), LASTFIELD(1)); - else - printf("%-*.*s %-*.*s %.*s\n", FIELD(0), FIELD(1), LASTFIELD(2)); - } - while (s < sections_cnt && sections[s].pos == line) { - opt_help(sections[s].sect); - s++; } } static struct opt_precomputed * opt_find_item_longopt(struct opt_context * oc, char * str) { - uns len = strlen(str); + uint len = strlen(str); struct opt_precomputed * candidate = NULL; for (int i = 0; i < oc->opt_count; i++) { @@ -220,9 +94,22 @@ static struct opt_precomputed * opt_find_item_longopt(struct opt_context * oc, c if (!strncmp(opt->name, str, len)) { if (strlen(opt->name) == len) return opt; - } else if (opt->item->cls == OPT_CL_BOOL && !strncmp("no-", str, 3) && !strncmp(opt->name, str+3, len-3)) { - if (strlen(opt->name) == len-3) - return opt; + } else if (opt->item->cls == OPT_CL_BOOL) { + if (opt->flags & OPT_NEGATIVE) { + // If the option is called no-X, match X as well + if (!strncmp("no-", opt->name, 3) && !strncmp(opt->name+3, str, len)) { + if (strlen(opt->name) == len+3) + return opt; + } else + continue; + } else { + // Match no-X as well + if (!strncmp("no-", str, 3) && !strncmp(opt->name, str+3, len-3)) { + if (strlen(opt->name) == len-3) + return opt; + } else + continue; + } } else continue; @@ -238,25 +125,16 @@ static struct opt_precomputed * opt_find_item_longopt(struct opt_context * oc, c opt_failure("Invalid option --%s.", str); } -// FIXME: Use simple-lists? -#define OPT_PTR(type) ({ \ - type * ptr; \ - if (item->flags & OPT_MULTIPLE) { \ - struct { \ - cnode n; \ - type v; \ - } * n = xmalloc(sizeof(*n)); \ - clist_add_tail(item->ptr, &(n->n)); \ - ptr = &(n->v); \ - } else \ - ptr = item->ptr; \ - ptr; }) - static void opt_parse_value(struct opt_context * oc, struct opt_precomputed * opt, char * value) { struct opt_item * item = opt->item; - for (int i = 0; i < oc->hooks_before_value_count; i++) - oc->hooks_before_value[i]->u.call(item, value, oc->hooks_before_value[i]->ptr); + if (opt->count++ && (opt->flags & OPT_SINGLE)) + opt_failure("Option %s must be specified at most once.", THIS_OPT); + + if (opt->flags & OPT_LAST_ARG) + oc->stop_parsing = 1; + + opt_invoke_hooks(oc, OPT_HOOK_BEFORE_VALUE, item, value); switch (item->cls) { case OPT_CL_BOOL: @@ -268,8 +146,15 @@ static void opt_parse_value(struct opt_context * oc, struct opt_precomputed * op opt_failure("Boolean argument for %s has a strange value. Supported (case insensitive): 1/0, y/n, yes/no, true/false.", THIS_OPT); break; case OPT_CL_STATIC: + case OPT_CL_MULTIPLE: { char * e = NULL; + void * ptr; + if (item->cls == OPT_CL_STATIC) + ptr = item->ptr; + else + ptr = GARY_PUSH_GENERIC(*(void **)item->ptr); +#define OPT_PTR(type) ((type *) ptr) switch (item->type) { case CT_INT: if (!value) @@ -309,14 +194,21 @@ static void opt_parse_value(struct opt_context * oc, struct opt_precomputed * op else *OPT_PTR(const char *) = xstrdup(value); break; + case CT_USER: + { + char * e = item->u.utype->parser(value, ptr); + if (e) + opt_failure("Cannot parse the value of %s: %s", THIS_OPT, e); + break; + } default: ASSERT(0); } +#undef OPT_PTR break; } case OPT_CL_SWITCH: - // FIXME: Really? And who sets the default to -1? - if (*((int *)item->ptr) != -1) + if ((opt->flags & OPT_SINGLE) && *((int *)item->ptr) != -1) opt_failure("Multiple switches: %s", THIS_OPT); else *((int *)item->ptr) = item->u.value; @@ -328,38 +220,31 @@ static void opt_parse_value(struct opt_context * oc, struct opt_precomputed * op (*((int *)item->ptr))++; break; case OPT_CL_CALL: - item->u.call(item, value, item->ptr); - break; - case OPT_CL_USER: { - char * e = NULL; - e = item->u.utype->parser(value, OPT_PTR(void*)); - if (e) - opt_failure("Cannot parse the value of %s: %s", THIS_OPT, e); + void *data = (opt->flags & OPT_INTERNAL) ? oc : item->ptr; + item->u.call(item, value, data); break; } default: ASSERT(0); } - for (int i = 0;i < oc->hooks_after_value_count; i++) - oc->hooks_after_value[i]->u.call(item, value, oc->hooks_after_value[i]->ptr); + opt_invoke_hooks(oc, OPT_HOOK_AFTER_VALUE, item, value); } static int opt_longopt(struct opt_context * oc, char ** argv, int index) { int eaten = 0; char * name_in = argv[index] + 2; // skipping the -- on the beginning - uns pos = strchrnul(name_in, '=') - name_in; + uint pos = strchrnul(name_in, '=') - name_in; struct opt_precomputed * opt = opt_find_item_longopt(oc, strndupa(name_in, pos)); char * value = NULL; opt->flags |= OPT_SEEN_AS_LONG; - // FIXME: Move to opt_parse_value()? - if (opt->count++ && (opt->flags & OPT_SINGLE)) - opt_failure("Option %s must be specified at most once.", THIS_OPT); - - if (opt->item->cls == OPT_CL_BOOL && !strncmp(name_in, "no-", 3) && !strncmp(name_in+3, opt->item->name, pos-3)) { + if (opt->item->cls == OPT_CL_BOOL && + ((opt->flags & OPT_NEGATIVE) + ? (!strncmp(opt->item->name, "no-", 3) && !strncmp(name_in, opt->item->name + 3, pos-3)) + : (!strncmp(name_in, "no-", 3) && !strncmp(name_in+3, opt->item->name, pos-3)))) { if (name_in[pos]) opt_failure("Option --%s must not have any value.", name_in); value = "n"; @@ -398,10 +283,6 @@ static int opt_shortopt(struct opt_context * oc, char ** argv, int index) { opt->flags &= ~OPT_SEEN_AS_LONG; - if (opt->count && (opt->flags & OPT_SINGLE)) - opt_failure("Option -%c must be specified at most once.", o); - opt->count++; - if (opt->flags & OPT_NO_VALUE) opt_parse_value(oc, opt, NULL); else if (opt->flags & OPT_REQUIRED_VALUE) { @@ -430,13 +311,12 @@ static int opt_shortopt(struct opt_context * oc, char ** argv, int index) { static void opt_positional(struct opt_context * oc, char * value) { oc->positional_count++; - uns id = oc->positional_count > oc->positional_max ? OPT_POSITIONAL_TAIL : OPT_POSITIONAL(oc->positional_count); + uint id = oc->positional_count > oc->positional_max ? OPT_POSITIONAL_TAIL : OPT_POSITIONAL(oc->positional_count); struct opt_precomputed * opt = oc->shortopt[id]; - if (!opt || (opt->flags & OPT_SINGLE) && opt->count) + if (!opt) opt_failure("Too many positional arguments."); else { - opt->flags &= OPT_SEEN_AS_LONG; - opt->count++; + opt->flags &= ~OPT_SEEN_AS_LONG; opt_parse_value(oc, opt, value); } } @@ -446,16 +326,9 @@ static void opt_count_items(struct opt_context *oc, const struct opt_section *se for (const struct opt_item *item = sec->opt; item->cls != OPT_CL_END; item++) { if (item->cls == OPT_CL_SECTION) opt_count_items(oc, item->u.section); - else if (item->cls == OPT_CL_HOOK) { - if (item->flags & OPT_HOOK_BEFORE_ARG) - oc->hooks_before_arg_count++; - else if (item->flags & OPT_HOOK_BEFORE_VALUE) - oc->hooks_before_value_count++; - else if (item->flags & OPT_HOOK_AFTER_VALUE) - oc->hooks_after_value_count++; - else - ASSERT(0); - } else if (item->letter || item->name) { + else if (item->cls == OPT_CL_HOOK) + oc->hook_count++; + else if (item->letter || item->name) { oc->opt_count++; if (item->letter > OPT_POSITIONAL_TAIL) oc->positional_max++; @@ -463,46 +336,18 @@ static void opt_count_items(struct opt_context *oc, const struct opt_section *se } } -static void opt_add_default_flags(struct opt_precomputed *opt) -{ - struct opt_item *item = opt->item; - uns flags = opt->flags; - - if (item->letter >= OPT_POSITIONAL_TAIL) { - flags &= ~OPT_VALUE_FLAGS; - flags |= OPT_REQUIRED_VALUE; - } - if (!(flags & OPT_VALUE_FLAGS)) { - ASSERT(item->cls != OPT_CL_CALL && item->cls != OPT_CL_USER); - flags |= opt_default_value_flags[item->cls]; - } - - opt->flags = flags; -} - static void opt_prepare_items(struct opt_context *oc, const struct opt_section *sec) { for (struct opt_item *item = sec->opt; item->cls != OPT_CL_END; item++) { if (item->cls == OPT_CL_SECTION) opt_prepare_items(oc, item->u.section); - else if (item->cls == OPT_CL_HOOK) { - if (item->flags & OPT_HOOK_BEFORE_ARG) - oc->hooks_before_arg[oc->hooks_before_arg_count++] = item; - else if (item->flags & OPT_HOOK_BEFORE_VALUE) - oc->hooks_before_value[oc->hooks_before_value_count++] = item; - else if (item->flags & OPT_HOOK_AFTER_VALUE) - oc->hooks_after_value[oc->hooks_after_value_count++] = item; - else - ASSERT(0); - } else if (item->letter || item->name) { + else if (item->cls == OPT_CL_HOOK) + oc->hooks[oc->hook_count++] = item; + else if (item->letter || item->name) { struct opt_precomputed * opt = &oc->opts[oc->opt_count++]; - opt->item = item; - opt->flags = item->flags; - opt->count = 0; - opt->name = item->name; + opt_precompute(opt, item); if (item->letter) oc->shortopt[(int) item->letter] = opt; - opt_add_default_flags(opt); } } } @@ -514,7 +359,7 @@ static void opt_check_required(struct opt_context *oc) if (!opt->count && (opt->flags & OPT_REQUIRED)) { struct opt_item *item = opt->item; if (item->letter > OPT_POSITIONAL_TAIL) - opt_failure("Required positional argument #%d not found.", i - OPT_POSITIONAL_TAIL); + opt_failure("Required positional argument #%d not found.", item->letter - OPT_POSITIONAL_TAIL); else if (item->letter == OPT_POSITIONAL_TAIL) opt_failure("Required positional argument not found."); else if (item->letter && item->name) @@ -527,29 +372,26 @@ static void opt_check_required(struct opt_context *oc) } } -void opt_parse(const struct opt_section * options, char ** argv) { +int opt_parse(const struct opt_section * options, char ** argv) { struct opt_context * oc = alloca(sizeof(*oc)); memset(oc, 0, sizeof (*oc)); + oc->options = options; opt_count_items(oc, options); oc->opts = alloca(sizeof(*oc->opts) * oc->opt_count); - oc->shortopt = alloca(sizeof(*oc->shortopt) * (oc->positional_max + 257)); - memset(oc->shortopt, 0, sizeof(*oc->shortopt) * (oc->positional_max + 257)); - oc->hooks_before_arg = alloca(sizeof (*oc->hooks_before_arg) * oc->hooks_before_arg_count); - oc->hooks_before_value = alloca(sizeof (*oc->hooks_before_value) * oc->hooks_before_value_count); - oc->hooks_after_value = alloca(sizeof (*oc->hooks_after_value) * oc->hooks_after_value_count); + oc->shortopt = alloca(sizeof(*oc->shortopt) * (oc->positional_max + OPT_POSITIONAL_TAIL + 1)); + memset(oc->shortopt, 0, sizeof(*oc->shortopt) * (oc->positional_max + OPT_POSITIONAL_TAIL + 1)); + oc->hooks = alloca(sizeof (*oc->hooks) * oc->hook_count); oc->opt_count = 0; - oc->hooks_before_arg_count = 0; - oc->hooks_before_value_count = 0; - oc->hooks_after_value_count = 0; + oc->hook_count = 0; opt_prepare_items(oc, options); int force_positional = 0; - for (int i = 0; argv[i]; i++) { + int i; + for (i=0; argv[i] && !oc->stop_parsing; i++) { char *arg = argv[i]; - for (int j = 0; j < oc->hooks_before_arg_count; j++) - oc->hooks_before_arg[j]->u.call(NULL, NULL, oc->hooks_before_arg[j]->ptr); + opt_invoke_hooks(oc, OPT_HOOK_BEFORE_ARG, NULL, NULL); if (arg[0] != '-' || force_positional) opt_positional(oc, arg); else { @@ -566,70 +408,6 @@ void opt_parse(const struct opt_section * options, char ** argv) { } opt_check_required(oc); -} - -static void opt_conf_end_of_options(struct cf_context *cc) { - cf_load_default(cc); - if (cc->postpone_commit && cf_close_group()) - opt_failure("Loading of configuration failed"); -} - -void opt_conf_internal(struct opt_item * opt, const char * value, void * data UNUSED) { - struct cf_context *cc = cf_get_context(); - switch (opt->letter) { - case 'S': - cf_load_default(cc); - if (cf_set(value)) - opt_failure("Cannot set %s", value); - break; - case 'C': - if (cf_load(value)) - opt_failure("Cannot load config file %s", value); - break; -#ifdef CONFIG_UCW_DEBUG - case '0': - opt_conf_end_of_options(cc); - struct fastbuf *b = bfdopen(1, 4096); - cf_dump_sections(b); - bclose(b); - exit(0); - break; -#endif - } -} - -void opt_conf_hook_internal(struct opt_item * opt, const char * value UNUSED, void * data UNUSED) { - static enum { - OPT_CONF_HOOK_BEGIN, - OPT_CONF_HOOK_CONFIG, - OPT_CONF_HOOK_OTHERS - } state = OPT_CONF_HOOK_BEGIN; - - int confopt = 0; - - if (opt->letter == 'S' || opt->letter == 'C' || (opt->name && !strcmp(opt->name, "dumpconfig"))) - confopt = 1; - - switch (state) { - case OPT_CONF_HOOK_BEGIN: - if (confopt) - state = OPT_CONF_HOOK_CONFIG; - else { - opt_conf_end_of_options(cf_get_context()); - state = OPT_CONF_HOOK_OTHERS; - } - break; - case OPT_CONF_HOOK_CONFIG: - if (!confopt) { - opt_conf_end_of_options(cf_get_context()); - state = OPT_CONF_HOOK_OTHERS; - } - break; - case OPT_CONF_HOOK_OTHERS: - if (confopt) - opt_failure("Config options (-C, -S) must stand before other options."); - break; - default: - ASSERT(0); - } + opt_invoke_hooks(oc, OPT_HOOK_FINAL, NULL, NULL); + return i; }