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