diff options
Diffstat (limited to 'libbb')
-rw-r--r-- | libbb/getopt_ulflags.c | 108 |
1 files changed, 72 insertions, 36 deletions
diff --git a/libbb/getopt_ulflags.c b/libbb/getopt_ulflags.c index 2e2ee0b..5f35c58 100644 --- a/libbb/getopt_ulflags.c +++ b/libbb/getopt_ulflags.c @@ -197,8 +197,9 @@ Special characters: if (flags & BB_GETOPT_ERROR) bb_show_usage(); - "!" If previous point set BB_GETOPT_ERROR, don`t return and call - previous example internally + "?" A "ask" as the first char in a bb_opt_complementally group give: + if previous point set BB_GETOPT_ERROR, don`t return and + call previous example internally "*" A star after a char in bb_opt_complementally means that the option can occur multiple times: @@ -218,16 +219,38 @@ Special characters: $ grep -e user -e root /etc/passwd root:x:0:0:root:/root:/bin/bash user:x:500:500::/home/user:/bin/bash + + "?" A "ask" between main and group options causes the second of the two + to be depending required if first is given on the command line. + For example from "id" applet: + + // Don't allow -n -r -rn -ug -rug -nug -rnug + bb_opt_complementally = "?u~g:g~u:r?ug:n?ug"; + flags = bb_getopt_ulflags(argc, argv, "rnug"); + + This example allowed only: + $ id; id -u; id -g; id -ru; id -nu; id -rg; id -ng; id -rnu; id -rng + + "?" A "ask" between equivalent options in bb_opt_complementally means + requires this option always, checked after switch off from + complementally logic + For example from "start-stop-daemon" applet: + + // Don't allow -KS -SK, but -S or -K required + bb_opt_complementally = "?:K?K:S?S:K~S:S~K"; + flags = bb_getopt_ulflags(argc, argv, "KS...); + */ const char *bb_opt_complementally; typedef struct { - unsigned char opt; - char list_flg; + int opt; + int list_flg; unsigned long switch_on; unsigned long switch_off; unsigned long incongruously; + unsigned long requires; void **optarg; /* char **optarg or llist_t **optarg */ int *counter; } t_complementally; @@ -245,17 +268,20 @@ unsigned long bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) { unsigned long flags = 0; + unsigned long requires = 0; t_complementally complementally[sizeof(flags) * 8 + 1]; int c; const unsigned char *s; t_complementally *on_off; va_list p; const struct option *l_o; - char flg_show_usage_if_error = 0; - char flg_argv_is_opts = 0; unsigned long trigger; char **pargv = NULL; +#define SHOW_USAGE_IF_ERROR 1 +#define ARGV_IS_OPTS 2 + int spec_flgs = 0; + va_start (p, applet_opts); /* skip GNU extension */ @@ -265,16 +291,13 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) c = 0; on_off = complementally; + memset(on_off, 0, sizeof(complementally)); + for (; *s; s++) { if(c >= (int)(sizeof(flags)*8)) break; on_off->opt = *s; on_off->switch_on = (1 << c); - on_off->list_flg = 0; - on_off->switch_off = 0; - on_off->incongruously = 0; - on_off->optarg = NULL; - on_off->counter = NULL; if (s[1] == ':') { on_off->optarg = va_arg (p, void **); do @@ -284,9 +307,10 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) on_off++; c++; } - on_off->opt = 0; for(l_o = bb_applet_long_options; l_o->name; l_o++) { + if(l_o->flag) + continue; for(on_off = complementally; on_off->opt != 0; on_off++) if(on_off->opt == l_o->val) break; @@ -295,16 +319,8 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) break; on_off->opt = l_o->val; on_off->switch_on = (1 << c); - on_off->list_flg = 0; - on_off->switch_off = 0; - on_off->incongruously = 0; - on_off->counter = NULL; if(l_o->has_arg != no_argument) on_off->optarg = va_arg (p, void **); - else - on_off->optarg = NULL; - on_off++; - on_off->opt = 0; c++; } } @@ -318,12 +334,12 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) } if (c) continue; - if(*s == '!') { - flg_show_usage_if_error = '!'; + if(*s == '?') { + spec_flgs |= SHOW_USAGE_IF_ERROR; continue; } if(*s == '-') { - flg_argv_is_opts = '-'; + spec_flgs |= ARGV_IS_OPTS; continue; } for (on_off = complementally; on_off->opt; on_off++) @@ -331,18 +347,32 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) break; pair = on_off; for(s++; *s && *s != ':'; s++) { - if (*s == '-' || *s == '~') { + if (*s == '-' || *s == '~' || *s == '?') { c = *s; } else if(*s == '*') { pair->list_flg++; } else { - unsigned long *pair_switch = &(pair->switch_on); - if(c) - pair_switch = c == '-' ? &(pair->switch_off) : &(pair->incongruously); + unsigned long *pair_switch; + + switch(c) { + case '-': + pair_switch = &(pair->switch_off); + break; + case '~': + pair_switch = &(pair->incongruously); + break; + case '?': + pair_switch = &(pair->requires); + break; + default: + pair_switch = &(pair->switch_on); + } for (on_off = complementally; on_off->opt; on_off++) if (on_off->opt == *s) { if(pair_switch == &(on_off->switch_on)) on_off->counter = va_arg (p, int *); + else if(pair_switch == &(on_off->requires)) + requires |= on_off->switch_on; else *pair_switch |= on_off->switch_on; break; @@ -353,15 +383,16 @@ bb_getopt_ulflags (int argc, char **argv, const char *applet_opts, ...) } while ((c = getopt_long (argc, argv, applet_opts, - bb_applet_long_options, NULL)) > 0) { + bb_applet_long_options, NULL)) >= 0) { loop_arg_is_opt: for (on_off = complementally; on_off->opt != c; on_off++) { - if(!on_off->opt) + /* c==0 if long opt have non NULL flag */ + if(on_off->opt == 0 && c != 0) bb_show_usage (); } if(flags & on_off->incongruously) { - if(flg_show_usage_if_error) + if((spec_flgs & SHOW_USAGE_IF_ERROR)) bb_show_usage (); flags |= BB_GETOPT_ERROR; } @@ -377,15 +408,13 @@ loop_arg_is_opt: } else if (on_off->optarg) { *(char **)(on_off->optarg) = optarg; } - if(flg_argv_is_opts == 'p') + if(pargv != NULL) break; } - if(flg_argv_is_opts) { + if((spec_flgs & ARGV_IS_OPTS)) { /* process argv is option, for example "ps" applet */ - if(flg_argv_is_opts == '-') { - flg_argv_is_opts = 'p'; + if(pargv == NULL) pargv = argv + optind; - } while(*pargv) { c = **pargv; if(c == '\0') { @@ -396,6 +425,13 @@ loop_arg_is_opt: } } } - + /* check depending requires for given options */ + for (on_off = complementally; on_off->opt; on_off++) { + if(on_off->requires && (flags & on_off->switch_on) && + (flags & on_off->requires) == 0) + bb_show_usage (); + } + if(requires && (flags & requires) == 0) + bb_show_usage (); return flags; } |