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);
|
---|