123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466 |
- #include <assert.h>
- #include <errno.h>
- #include <stdarg.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include "getopt.h"
- #ifndef _DIAGASSERT
- #define _DIAGASSERT(e)
- #endif
- int opterr = 1;
- int optind = 1;
- int optopt = '?';
- int optreset;
- char *optarg;
- #define IGNORE_FIRST (*options == '-' || *options == '+')
- #define PRINT_ERROR ((opterr) && ((*options != ':') \
- || (IGNORE_FIRST && options[1] != ':')))
- #define IS_POSIXLY_CORRECT (getenv("POSIXLY_CORRECT") != NULL)
- #define PERMUTE (!IS_POSIXLY_CORRECT && !IGNORE_FIRST)
- #define IN_ORDER (!IS_POSIXLY_CORRECT && *options == '-')
- #define BADCH (int)'?'
- #define BADARG (int)':'
- #define INORDER (int)1
- #define EMSG ""
- static int getopt_internal(int, char * const *, const char *);
- static int gcd(int, int);
- static void permute_args(int, int, int, char * const *);
- static void xwarnx(const char *, ...);
- static char *place = EMSG;
- static int nonopt_start = -1;
- static int nonopt_end = -1;
- static const char recargchar[] = "option requires an argument -- %c";
- static const char recargstring[] = "option requires an argument -- %s";
- static const char ambig[] = "ambiguous option -- %.*s";
- static const char noarg[] = "option doesn't take an argument -- %.*s";
- static const char illoptchar[] = "illegal option -- %c";
- static const char illoptstring[] = "illegal option -- %s";
- static const char *progname;
- static void xwarnx(const char *fmt, ...) {
- va_list ap;
- va_start(ap, fmt);
- if (progname)
- (void) fprintf(stderr, "%s: ", progname);
- if (fmt)
- (void) vfprintf(stderr, fmt, ap);
- (void) fprintf(stderr, "\n");
- va_end(ap);
- }
- static int
- gcd(int a, int b)
- {
- int c;
- c = a % b;
- while (c != 0) {
- a = b;
- b = c;
- c = a % b;
- }
- return b;
- }
- static void
- permute_args(int nonopt_start, int nonopt_end, int opt_end, char * const *nargv)
- {
- int cstart, cyclelen, i, j, ncycle, nnonopts, nopts, pos;
- char *swap;
-
- nnonopts = nonopt_end - nonopt_start;
- nopts = opt_end - nonopt_end;
- ncycle = gcd(nnonopts, nopts);
- cyclelen = (opt_end - nonopt_start) / ncycle;
- for (i = 0; i < ncycle; i++) {
- cstart = nonopt_end+i;
- pos = cstart;
- for (j = 0; j < cyclelen; j++) {
- if (pos >= nonopt_end)
- pos -= nnonopts;
- else
- pos += nopts;
- swap = nargv[pos];
-
- ((char **) nargv)[pos] = nargv[cstart];
-
- ((char **)nargv)[cstart] = swap;
- }
- }
- }
- static int
- getopt_internal(int nargc, char * const *nargv, const char *options)
- {
- char *oli;
- int optchar;
- _DIAGASSERT(nargv != NULL);
- _DIAGASSERT(options != NULL);
- optarg = NULL;
-
- if (optind == 0)
- optind = 1;
- if (optreset)
- nonopt_start = nonopt_end = -1;
- start:
- if (optreset || !*place) {
- optreset = 0;
- if (optind >= nargc) {
- place = EMSG;
- if (nonopt_end != -1) {
-
- permute_args(nonopt_start, nonopt_end,
- optind, nargv);
- optind -= nonopt_end - nonopt_start;
- }
- else if (nonopt_start != -1) {
-
- optind = nonopt_start;
- }
- nonopt_start = nonopt_end = -1;
- return -1;
- }
- if (*(place = nargv[optind]) != '-') {
- place = EMSG;
- if (IN_ORDER) {
-
- optarg = nargv[optind++];
- return INORDER;
- }
- if (!PERMUTE) {
-
- return -1;
- }
-
- if (nonopt_start == -1)
- nonopt_start = optind;
- else if (nonopt_end != -1) {
- permute_args(nonopt_start, nonopt_end,
- optind, nargv);
- nonopt_start = optind -
- (nonopt_end - nonopt_start);
- nonopt_end = -1;
- }
- optind++;
-
- goto start;
- }
- if (nonopt_start != -1 && nonopt_end == -1)
- nonopt_end = optind;
- if (place[1] && *++place == '-') {
- place++;
- return -2;
- }
- }
- if ((optchar = (int)*place++) == (int)':' ||
- (oli = strchr(options + (IGNORE_FIRST ? 1 : 0), optchar)) == NULL) {
-
- if (!*place)
- ++optind;
- if (PRINT_ERROR)
- xwarnx(illoptchar, optchar);
- optopt = optchar;
- return BADCH;
- }
- if (optchar == 'W' && oli[1] == ';') {
-
- if (*place)
- return -2;
- if (++optind >= nargc) {
- place = EMSG;
- if (PRINT_ERROR)
- xwarnx(recargchar, optchar);
- optopt = optchar;
-
- return BADARG;
- } else
- place = nargv[optind];
-
- return -2;
- }
- if (*++oli != ':') {
- if (!*place)
- ++optind;
- } else {
- optarg = NULL;
- if (*place)
- optarg = place;
-
- else if (oli[1] != ':') {
- if (++optind >= nargc) {
- place = EMSG;
- if (PRINT_ERROR)
- xwarnx(recargchar, optchar);
- optopt = optchar;
-
- return BADARG;
- } else
- optarg = nargv[optind];
- }
- place = EMSG;
- ++optind;
- }
-
- return optchar;
- }
- int
- getopt(int nargc, char * const *nargv, const char *options)
- {
- int retval;
- progname = nargv[0];
- if ((retval = getopt_internal(nargc, nargv, options)) == -2) {
- ++optind;
-
- if (nonopt_end != -1) {
- permute_args(nonopt_start, nonopt_end, optind,
- nargv);
- optind -= nonopt_end - nonopt_start;
- }
- nonopt_start = nonopt_end = -1;
- retval = -1;
- }
- return retval;
- }
- int
- getopt_long(int nargc,
- char * const *nargv,
- const char *options,
- const struct option *long_options,
- int *idx)
- {
- int retval;
- _DIAGASSERT(nargv != NULL);
- _DIAGASSERT(options != NULL);
- _DIAGASSERT(long_options != NULL);
-
- progname = nargv[0];
- if ((retval = getopt_internal(nargc, nargv, options)) == -2) {
- char *current_argv, *has_equal;
- size_t current_argv_len;
- int i, match;
- current_argv = place;
- match = -1;
- optind++;
- place = EMSG;
- if (*current_argv == '\0') {
-
- if (nonopt_end != -1) {
- permute_args(nonopt_start, nonopt_end,
- optind, nargv);
- optind -= nonopt_end - nonopt_start;
- }
- nonopt_start = nonopt_end = -1;
- return -1;
- }
- if ((has_equal = strchr(current_argv, '=')) != NULL) {
-
- current_argv_len = has_equal - current_argv;
- has_equal++;
- } else
- current_argv_len = strlen(current_argv);
- for (i = 0; long_options[i].name; i++) {
-
- if (strncmp(current_argv, long_options[i].name,
- current_argv_len))
- continue;
- if (strlen(long_options[i].name) ==
- (unsigned)current_argv_len) {
-
- match = i;
- break;
- }
- if (match == -1)
- match = i;
- else {
-
- if (PRINT_ERROR)
- xwarnx(ambig, (int)current_argv_len,
- current_argv);
- optopt = 0;
- return BADCH;
- }
- }
- if (match != -1) {
- if (long_options[match].has_arg == no_argument
- && has_equal) {
- if (PRINT_ERROR)
- xwarnx(noarg, (int)current_argv_len,
- current_argv);
-
- if (long_options[match].flag == NULL)
- optopt = long_options[match].val;
- else
- optopt = 0;
-
- return BADARG;
- }
- if (long_options[match].has_arg == required_argument ||
- long_options[match].has_arg == optional_argument) {
- if (has_equal)
- optarg = has_equal;
- else if (long_options[match].has_arg ==
- required_argument) {
-
- optarg = nargv[optind++];
- }
- }
- if ((long_options[match].has_arg == required_argument)
- && (optarg == NULL)) {
-
- if (PRINT_ERROR)
- xwarnx(recargstring, current_argv);
-
- if (long_options[match].flag == NULL)
- optopt = long_options[match].val;
- else
- optopt = 0;
-
- --optind;
- return BADARG;
- }
- } else {
- if (PRINT_ERROR)
- xwarnx(illoptstring, current_argv);
- optopt = 0;
- return BADCH;
- }
- if (long_options[match].flag) {
- *long_options[match].flag = long_options[match].val;
- retval = 0;
- } else
- retval = long_options[match].val;
- if (idx)
- *idx = match;
- }
- return retval;
- }
|