nsz Git - musl/blob - src/misc/getopt_long.c

   1 #define _GNU_SOURCE
   2 #include <stddef.h>
   3 #include <getopt.h>
   4 #include <stdio.h>
   5
   6 extern int __optpos, __optreset;
   7
   8 static void permute(char *const *argv, int dest, int src)
   9 {
  10         char **av = (char **)argv;
  11         char *tmp = av[src];
  12         int i;
  13         for (i=src; i>dest; i--)
  14                 av[i] = av[i-1];
  15         av[dest] = tmp;
  16 }
  17
  18 static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly);
  19
  20 static int __getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly)
  21 {
  22         int ret, skipped, resumed;
  23         if (!optind || __optreset) {
  24                 __optreset = 0;
  25                 __optpos = 0;
  26                 optind = 1;
  27         }
  28         if (optind >= argc || !argv[optind]) return -1;
  29         skipped = optind;
  30         if (optstring[0] != '+' && optstring[0] != '-') {
  31                 int i;
  32                 for (i=optind; ; i++) {
  33                         if (i >= argc || !argv[i]) return -1;
  34                         if (argv[i][0] != '-') continue;
  35                         if (!argv[i][1]) continue;
  36                         if (argv[i][1] == '-' && !argv[i][2]) return -1;
  37                         break;
  38                 }
  39                 optind = i;
  40         }
  41         resumed = optind;
  42         ret = __getopt_long_core(argc, argv, optstring, longopts, idx, longonly);
  43         if (resumed > skipped) {
  44                 int i, cnt = optind-resumed;
  45                 for (i=0; i<cnt; i++)
  46                         permute(argv, skipped, optind-1);
  47                 optind = skipped + cnt;
  48         }
  49         return ret;
  50 }
  51
  52 static int __getopt_long_core(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly)
  53 {
  54
  55         if (argv[optind][0] == '-' &&
  56                 ((longonly && argv[optind][1]) ||
  57                  (argv[optind][1] == '-' && argv[optind][2])))
  58         {
  59                 int i, cnt, match;
  60                 char *opt;
  61                 for (cnt=i=0; longopts[i].name; i++) {
  62                         const char *name = longopts[i].name;
  63                         opt = argv[optind]+1;
  64                         if (*opt == '-') opt++;
  65                         for (; *name && *name == *opt; name++, opt++);
  66                         if (*opt && *opt != '=') continue;
  67                         match = i;
  68                         if (!*name) {
  69                                 cnt = 1;
  70                                 break;
  71                         }
  72                         cnt++;
  73                 }
  74                 if (cnt==1) {
  75                         i = match;
  76                         optind++;
  77                         if (*opt == '=') {
  78                                 if (!longopts[i].has_arg) return '?';
  79                                 optarg = opt+1;
  80                         } else {
  81                                 if (longopts[i].has_arg == required_argument) {
  82                                         if (!(optarg = argv[optind]))
  83                                                 return ':';
  84                                         optind++;
  85                                 } else optarg = NULL;
  86                         }
  87                         if (idx) *idx = i;
  88                         if (longopts[i].flag) {
  89                                 *longopts[i].flag = longopts[i].val;
  90                                 return 0;
  91                         }
  92                         return longopts[i].val;
  93                 }
  94                 if (argv[optind][1] == '-') {
  95                         optind++;
  96                         return '?';
  97                 }
  98         }
  99         return getopt(argc, argv, optstring);
 100 }
 101
 102 int getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
 103 {
 104         return __getopt_long(argc, argv, optstring, longopts, idx, 0);
 105 }
 106
 107 int getopt_long_only(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
 108 {
 109         return __getopt_long(argc, argv, optstring, longopts, idx, 1);
 110 }