lib/getopt_long.c

   1 /*
   2  * getopt.c
   3  *
   4  * getopt_long(), or at least a common subset thereof:
   5  *
   6  * - Option reordering is not supported
   7  * - -W foo is not supported
   8  * - First optstring character "-" not supported.
   9  */
  10
  11 #include <stdint.h>
  12 #include <unistd.h>
  13 #include <string.h>
  14 #include <getopt.h>
  15
  16 char *optarg;
  17 int optind, opterr, optopt;
  18 static struct getopt_private_state {
  19         const char *optptr;
  20         const char *last_optstring;
  21         char *const *last_argv;
  22 } pvt;
  23
  24 static inline const char *option_matches(const char *arg_str,
  25                                          const char *opt_name)
  26 {
  27         while (*arg_str != '\0' && *arg_str != '=') {
  28                 if (*arg_str++ != *opt_name++)
  29                         return NULL;
  30         }
  31
  32         if (*opt_name)
  33                 return NULL;
  34
  35         return arg_str;
  36 }
  37
  38 int getopt_long_only(int argc, char *const *argv, const char *optstring,
  39                 const struct option *longopts, int *longindex)
  40 {
  41         const char *carg;
  42         const char *osptr;
  43         int opt;
  44
  45         /* getopt() relies on a number of different global state
  46            variables, which can make this really confusing if there is
  47            more than one use of getopt() in the same program.  This
  48            attempts to detect that situation by detecting if the
  49            "optstring" or "argv" argument have changed since last time
  50            we were called; if so, reinitialize the query state. */
  51
  52         if (optstring != pvt.last_optstring || argv != pvt.last_argv ||
  53             optind < 1 || optind > argc) {
  54                 /* optind doesn't match the current query */
  55                 pvt.last_optstring = optstring;
  56                 pvt.last_argv = argv;
  57                 optind = 1;
  58                 pvt.optptr = NULL;
  59         }
  60
  61         carg = argv[optind];
  62
  63         /* First, eliminate all non-option cases */
  64
  65         if (!carg || carg[0] != '-' || !carg[1])
  66                 return -1;
  67
  68         if (carg[1] == '-') {
  69                 const struct option *lo;
  70                 const char *opt_end = NULL;
  71
  72                 optind++;
  73
  74                 /* Either it's a long option, or it's -- */
  75                 if (!carg[2]) {
  76                         /* It's -- */
  77                         return -1;
  78                 }
  79
  80                 for (lo = longopts; lo->name; lo++) {
  81                         if ((opt_end = option_matches(carg+2, lo->name)))
  82                             break;
  83                 }
  84                 if (!opt_end)
  85                         return '?';
  86
  87                 if (longindex)
  88                         *longindex = lo-longopts;
  89
  90                 if (*opt_end == '=') {
  91                         if (lo->has_arg)
  92                                 optarg = (char *)opt_end+1;
  93                         else
  94                                 return '?';
  95                 } else if (lo->has_arg == 1) {
  96                         if (!(optarg = argv[optind]))
  97                                 return '?';
  98                         optind++;
  99                 }
 100
 101                 if (lo->flag) {
 102                         *lo->flag = lo->val;
 103                         return 0;
 104                 } else {
 105                         return lo->val;
 106                 }
 107         }
 108
 109         if ((uintptr_t) (pvt.optptr - carg) > (uintptr_t) strlen(carg)) {
 110                 /* Someone frobbed optind, change to new opt. */
 111                 pvt.optptr = carg + 1;
 112         }
 113
 114         opt = *pvt.optptr++;
 115
 116         if (opt != ':' && (osptr = strchr(optstring, opt))) {
 117                 if (osptr[1] == ':') {
 118                         if (*pvt.optptr) {
 119                                 /* Argument-taking option with attached
 120                                    argument */
 121                                 optarg = (char *)pvt.optptr;
 122                                 optind++;
 123                         } else {
 124                                 /* Argument-taking option with non-attached
 125                                    argument */
 126                                 if (argv[optind + 1]) {
 127                                         optarg = (char *)argv[optind+1];
 128                                         optind += 2;
 129                                 } else {
 130                                         /* Missing argument */
 131                                         optind++;
 132                                         return (optstring[0] == ':')
 133                                                 ? ':' : '?';
 134                                 }
 135                         }
 136                         return opt;
 137                 } else {
 138                         /* Non-argument-taking option */
 139                         /* pvt.optptr will remember the exact position to
 140                            resume at */
 141                         if (!*pvt.optptr)
 142                                 optind++;
 143                         return opt;
 144                 }
 145         } else {
 146                 /* Unknown option */
 147                 optopt = opt;
 148                 if (!*pvt.optptr)
 149                         optind++;
 150                 return '?';
 151         }
 152 }