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

   1 #define _BSD_SOURCE
   2 #include <unistd.h>
   3 #include <wchar.h>
   4 #include <string.h>
   5 #include <limits.h>
   6 #include <stdlib.h>
   7 #include "locale_impl.h"
   8 #include "stdio_impl.h"
   9
  10 char *optarg;
  11 int optind=1, opterr=1, optopt, __optpos, __optreset=0;
  12
  13 #define optpos __optpos
  14 weak_alias(__optreset, optreset);
  15
  16 void __getopt_msg(const char *a, const char *b, const char *c, size_t l)
  17 {
  18         FILE *f = stderr;
  19         b = __lctrans_cur(b);
  20         FLOCK(f);
  21         fputs(a, f)>=0
  22         && fwrite(b, strlen(b), 1, f)
  23         && fwrite(c, 1, l, f)==l
  24         && putc('\n', f);
  25         FUNLOCK(f);
  26 }
  27
  28 int getopt(int argc, char * const argv[], const char *optstring)
  29 {
  30         int i;
  31         wchar_t c, d;
  32         int k, l;
  33         char *optchar;
  34
  35         if (!optind || __optreset) {
  36                 __optreset = 0;
  37                 __optpos = 0;
  38                 optind = 1;
  39         }
  40
  41         if (optind >= argc || !argv[optind])
  42                 return -1;
  43
  44         if (argv[optind][0] != '-') {
  45                 if (optstring[0] == '-') {
  46                         optarg = argv[optind++];
  47                         return 1;
  48                 }
  49                 return -1;
  50         }
  51
  52         if (!argv[optind][1])
  53                 return -1;
  54
  55         if (argv[optind][1] == '-' && !argv[optind][2])
  56                 return optind++, -1;
  57
  58         if (!optpos) optpos++;
  59         if ((k = mbtowc(&c, argv[optind]+optpos, MB_LEN_MAX)) < 0) {
  60                 k = 1;
  61                 c = 0xfffd; /* replacement char */
  62         }
  63         optchar = argv[optind]+optpos;
  64         optpos += k;
  65
  66         if (!argv[optind][optpos]) {
  67                 optind++;
  68                 optpos = 0;
  69         }
  70
  71         if (optstring[0] == '-' || optstring[0] == '+')
  72                 optstring++;
  73
  74         i = 0;
  75         d = 0;
  76         do {
  77                 l = mbtowc(&d, optstring+i, MB_LEN_MAX);
  78                 if (l>0) i+=l; else i++;
  79         } while (l && d != c);
  80
  81         if (d != c || c == ':') {
  82                 optopt = c;
  83                 if (optstring[0] != ':' && opterr)
  84                         __getopt_msg(argv[0], ": unrecognized option: ", optchar, k);
  85                 return '?';
  86         }
  87         if (optstring[i] == ':') {
  88                 optarg = 0;
  89                 if (optstring[i+1] != ':' || optpos) {
  90                         optarg = argv[optind++];
  91                         if (optpos) optarg += optpos;
  92                         optpos = 0;
  93                 }
  94                 if (optind > argc) {
  95                         optopt = c;
  96                         if (optstring[0] == ':') return ':';
  97                         if (opterr) __getopt_msg(argv[0],
  98                                 ": option requires an argument: ",
  99                                 optchar, k);
 100                         return '?';
 101                 }
 102         }
 103         return c;
 104 }
 105
 106 weak_alias(getopt, __posix_getopt);