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 int __getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx, int longonly)
   9 {
  10         if (!optind || __optreset) {
  11                 __optreset = 0;
  12                 __optpos = 0;
  13                 optind = 1;
  14         }
  15         if (optind >= argc || !argv[optind]) return -1;
  16         if (argv[optind][0] == '-' &&
  17                 ((longonly && argv[optind][1]) ||
  18                  (argv[optind][1] == '-' && argv[optind][2])))
  19         {
  20                 int i;
  21                 for (i=0; longopts[i].name; i++) {
  22                         const char *name = longopts[i].name;
  23                         char *opt = argv[optind]+1;
  24                         if (*opt == '-') opt++;
  25                         for (; *name && *name == *opt; name++, opt++);
  26                         if (*name || (*opt && *opt != '=')) continue;
  27                         if (*opt == '=') {
  28                                 if (!longopts[i].has_arg) continue;
  29                                 optarg = opt+1;
  30                         } else {
  31                                 if (longopts[i].has_arg == required_argument) {
  32                                         if (!(optarg = argv[++optind]))
  33                                                 return ':';
  34                                 } else optarg = NULL;
  35                         }
  36                         optind++;
  37                         if (idx) *idx = i;
  38                         if (longopts[i].flag) {
  39                                 *longopts[i].flag = longopts[i].val;
  40                                 return 0;
  41                         }
  42                         return longopts[i].val;
  43                 }
  44                 if (argv[optind][1] == '-') {
  45                         optind++;
  46                         return '?';
  47                 }
  48         }
  49         return getopt(argc, argv, optstring);
  50 }
  51
  52 int getopt_long(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
  53 {
  54         return __getopt_long(argc, argv, optstring, longopts, idx, 0);
  55 }
  56
  57 int getopt_long_only(int argc, char *const *argv, const char *optstring, const struct option *longopts, int *idx)
  58 {
  59         return __getopt_long(argc, argv, optstring, longopts, idx, 1);
  60 }