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