1 | /* |
1 | /* |
2 | * getopt.c - Enhanced implementation of BSD getopt(1) |
2 | * getopt.c - Enhanced implementation of BSD getopt(1) |
3 | * Copyright (c) 1997-2005 Frodo Looijaard <frodo@frodo.looijaard.name> |
3 | * Copyright (c) 1997-2014 Frodo Looijaard <frodo@frodo.looijaard.name> |
4 | * |
4 | * |
5 | * This program is free software; you can redistribute it and/or modify |
5 | * This program is free software; you can redistribute it and/or modify |
6 | * it under the terms of the GNU General Public License as published by |
6 | * it under the terms of the GNU General Public License as published by |
7 | * the Free Software Foundation; either version 2 of the License, or |
7 | * the Free Software Foundation; either version 2 of the License, or |
8 | * (at your option) any later version. |
8 | * (at your option) any later version. |
… | |
… | |
10 | * This program is distributed in the hope that it will be useful, |
10 | * This program is distributed in the hope that it will be useful, |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
13 | * GNU General Public License for more details. |
13 | * GNU General Public License for more details. |
14 | * |
14 | * |
15 | * You should have received a copy of the GNU General Public License |
15 | * You should have received a copy of the GNU General Public License along |
16 | * along with this program; if not, write to the Free Software |
16 | * with this program; if not, write to the Free Software Foundation, Inc., |
17 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
18 | */ |
18 | */ |
19 | |
19 | |
20 | /* |
20 | /* |
21 | * Version 1.0-b4: Tue Sep 23 1997. First public release. |
21 | * Version 1.0-b4: Tue Sep 23 1997. First public release. |
22 | * Version 1.0: Wed Nov 19 1997. |
22 | * Version 1.0: Wed Nov 19 1997. |
23 | * Bumped up the version number to 1.0 |
23 | * Bumped up the version number to 1.0 |
24 | * Fixed minor typo (CSH instead of TCSH) |
24 | * Fixed minor typo (CSH instead of TCSH) |
25 | * Version 1.0.1: Tue Jun 3 1998 |
25 | * Version 1.0.1: Tue Jun 3 1998 |
26 | * Fixed sizeof instead of strlen bug |
26 | * Fixed sizeof instead of strlen bug |
27 | * Bumped up the version number to 1.0.1 |
27 | * Bumped up the version number to 1.0.1 |
… | |
… | |
31 | * Version 1.0.5: Tue Jun 22 1999 |
31 | * Version 1.0.5: Tue Jun 22 1999 |
32 | * Make -u option work (not present) |
32 | * Make -u option work (not present) |
33 | * Version 1.0.6: Tue Jun 27 2000 |
33 | * Version 1.0.6: Tue Jun 27 2000 |
34 | * No important changes |
34 | * No important changes |
35 | * Version 1.1.0: Tue Jun 30 2000 |
35 | * Version 1.1.0: Tue Jun 30 2000 |
36 | * Added NLS support (partly written by Arkadiusz Miśkiewicz |
36 | * Added NLS support (partly written by Arkadiusz Miśkiewicz |
37 | * <misiek@pld.org.pl>) |
37 | * <misiek@pld.org.pl>) |
38 | * Version 1.1.4: Mon Nov 7 2005 |
38 | * Version 1.1.4: Mon Nov 7 2005 |
39 | * Fixed a few type's in the manpage |
39 | * Fixed a few type's in the manpage |
40 | * Version 1.1.5: Sun Aug 12 2012 |
40 | * Version 1.1.5: Sun Aug 12 2012 |
41 | * Sync with util-linux-2.21, fixed build problems, many new translations |
41 | * Sync with util-linux-2.21, fixed build problems, many new translations |
|
|
42 | * Version 1.1.6: ??? 2014 |
|
|
43 | * Sync with util-linux git 20141120, detect ambiguous long options, fix |
|
|
44 | * backslash problem in tcsh |
42 | */ |
45 | */ |
43 | |
46 | |
44 | /* Exit codes: |
47 | /* Exit codes: |
45 | * 0) No errors, succesful operation. |
48 | * 0) No errors, successful operation. |
46 | * 1) getopt(3) returned an error. |
49 | * 1) getopt(3) returned an error. |
47 | * 2) A problem with parameter parsing for getopt(1). |
50 | * 2) A problem with parameter parsing for getopt(1). |
48 | * 3) Internal error, out of memory |
51 | * 3) Internal error, out of memory |
49 | * 4) Returned for -T |
52 | * 4) Returned for -T |
50 | */ |
53 | */ |
… | |
… | |
67 | |
70 | |
68 | #include "util-linux-compat.h" |
71 | #include "util-linux-compat.h" |
69 | #include "nls.h" |
72 | #include "nls.h" |
70 | #include "xalloc.h" |
73 | #include "xalloc.h" |
71 | |
74 | |
72 | /* NON_OPT is the code that is returned when a non-option is found in '+' |
75 | /* NON_OPT is the code that is returned when a non-option is found in '+' |
73 | * mode */ |
76 | * mode */ |
74 | #define NON_OPT 1 |
77 | #define NON_OPT 1 |
75 | /* LONG_OPT is the code that is returned when a long option is found. */ |
78 | /* LONG_OPT is the code that is returned when a long option is found. */ |
76 | #define LONG_OPT 2 |
79 | #define LONG_OPT 0 |
77 | |
80 | |
78 | /* The shells recognized. */ |
81 | /* The shells recognized. */ |
79 | typedef enum { BASH, TCSH } shell_t; |
82 | typedef enum { BASH, TCSH } shell_t; |
80 | |
83 | |
81 | |
84 | |
… | |
… | |
91 | |
94 | |
92 | /* Function prototypes */ |
95 | /* Function prototypes */ |
93 | static const char *normalize(const char *arg); |
96 | static const char *normalize(const char *arg); |
94 | static int generate_output(char *argv[], int argc, const char *optstr, |
97 | static int generate_output(char *argv[], int argc, const char *optstr, |
95 | const struct option *longopts); |
98 | const struct option *longopts); |
96 | int main(int argc, char *argv[]); |
|
|
97 | static void parse_error(const char *message); |
99 | static void parse_error(const char *message); |
98 | static void add_long_options(char *options); |
100 | static void add_long_options(char *options); |
99 | static void add_longopt(const char *name, int has_arg); |
101 | static void add_longopt(const char *name, int has_arg); |
100 | static void print_help(void); |
102 | static void print_help(void); |
101 | static void set_shell(const char *new_shell); |
103 | static void set_shell(const char *new_shell); |
… | |
… | |
123 | strcpy(BUFFER, arg); |
125 | strcpy(BUFFER, arg); |
124 | return BUFFER; |
126 | return BUFFER; |
125 | } |
127 | } |
126 | |
128 | |
127 | /* |
129 | /* |
128 | * Each character in arg may take upto four characters in the |
130 | * Each character in arg may take up to four characters in the |
129 | * result: For a quote we need a closing quote, a backslash, a quote |
131 | * result: For a quote we need a closing quote, a backslash, a quote |
130 | * and an opening quote! We need also the global opening and closing |
132 | * and an opening quote! We need also the global opening and closing |
131 | * quote, and one extra character for '\0'. |
133 | * quote, and one extra character for '\0'. |
132 | */ |
134 | */ |
133 | BUFFER = xmalloc(strlen(arg) * 4 + 3); |
135 | BUFFER = xmalloc(strlen(arg) * 4 + 3); |
… | |
… | |
140 | /* Quote: replace it with: '\'' */ |
142 | /* Quote: replace it with: '\'' */ |
141 | *bufptr++ = '\''; |
143 | *bufptr++ = '\''; |
142 | *bufptr++ = '\\'; |
144 | *bufptr++ = '\\'; |
143 | *bufptr++ = '\''; |
145 | *bufptr++ = '\''; |
144 | *bufptr++ = '\''; |
146 | *bufptr++ = '\''; |
|
|
147 | } else if (shell == TCSH && *argptr == '\\') { |
|
|
148 | /* Backslash: replace it with: '\\' */ |
|
|
149 | *bufptr++ = '\\'; |
|
|
150 | *bufptr++ = '\\'; |
145 | } else if (shell == TCSH && *argptr == '!') { |
151 | } else if (shell == TCSH && *argptr == '!') { |
146 | /* Exclamation mark: replace it with: \! */ |
152 | /* Exclamation mark: replace it with: \! */ |
147 | *bufptr++ = '\''; |
153 | *bufptr++ = '\''; |
148 | *bufptr++ = '\\'; |
154 | *bufptr++ = '\\'; |
149 | *bufptr++ = '!'; |
155 | *bufptr++ = '!'; |
… | |
… | |
166 | *bufptr++ = '\''; |
172 | *bufptr++ = '\''; |
167 | *bufptr++ = '\0'; |
173 | *bufptr++ = '\0'; |
168 | return BUFFER; |
174 | return BUFFER; |
169 | } |
175 | } |
170 | |
176 | |
171 | /* |
177 | /* |
172 | * Generate the output. argv[0] is the program name (used for reporting errors). |
178 | * Generate the output. argv[0] is the program name (used for reporting errors). |
173 | * argv[1..] contains the options to be parsed. argc must be the number of |
179 | * argv[1..] contains the options to be parsed. argc must be the number of |
174 | * elements in argv (ie. 1 if there are no options, only the program name), |
180 | * elements in argv (ie. 1 if there are no options, only the program name), |
175 | * optstr must contain the short options, and longopts the long options. |
181 | * optstr must contain the short options, and longopts the long options. |
176 | * Other settings are found in global variables. |
182 | * Other settings are found in global variables. |
… | |
… | |
223 | * we already sent a message, we just exit with a helpful hint. |
229 | * we already sent a message, we just exit with a helpful hint. |
224 | */ |
230 | */ |
225 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) |
231 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) |
226 | { |
232 | { |
227 | if (message) |
233 | if (message) |
228 | fprintf(stderr, "%s: %s\n", program_invocation_short_name, message); |
234 | warnx("%s", message); |
229 | fprintf(stderr, _("Try `%s --help' for more information.\n"), |
235 | fprintf(stderr, _("Try `%s --help' for more information.\n"), |
230 | program_invocation_short_name); |
236 | program_invocation_short_name); |
231 | exit(PARAMETER_EXIT_CODE); |
237 | exit(PARAMETER_EXIT_CODE); |
232 | } |
238 | } |
233 | |
239 | |
… | |
… | |
239 | |
245 | |
240 | /* Register a long option. The contents of name is copied. */ |
246 | /* Register a long option. The contents of name is copied. */ |
241 | static void add_longopt(const char *name, int has_arg) |
247 | static void add_longopt(const char *name, int has_arg) |
242 | { |
248 | { |
243 | char *tmp; |
249 | char *tmp; |
|
|
250 | static int flag; |
|
|
251 | |
244 | if (!name) { |
252 | if (!name) { |
245 | /* init */ |
253 | /* init */ |
246 | free(long_options); |
254 | free(long_options); |
247 | long_options = NULL; |
255 | long_options = NULL; |
248 | long_options_length = 0; |
256 | long_options_length = 0; |
… | |
… | |
262 | long_options[long_options_nr].val = 0; |
270 | long_options[long_options_nr].val = 0; |
263 | |
271 | |
264 | if (long_options_nr && name) { |
272 | if (long_options_nr && name) { |
265 | /* Not for init! */ |
273 | /* Not for init! */ |
266 | long_options[long_options_nr - 1].has_arg = has_arg; |
274 | long_options[long_options_nr - 1].has_arg = has_arg; |
267 | long_options[long_options_nr - 1].flag = NULL; |
275 | long_options[long_options_nr - 1].flag = &flag; |
268 | long_options[long_options_nr - 1].val = LONG_OPT; |
276 | long_options[long_options_nr - 1].val = long_options_nr; |
269 | tmp = xmalloc(strlen(name) + 1); |
277 | tmp = xmalloc(strlen(name) + 1); |
270 | strcpy(tmp, name); |
278 | strcpy(tmp, name); |
271 | long_options[long_options_nr - 1].name = tmp; |
279 | long_options[long_options_nr - 1].name = tmp; |
272 | } |
280 | } |
273 | long_options_nr++; |
281 | long_options_nr++; |
274 | } |
282 | } |
275 | |
283 | |
276 | |
284 | |
277 | /* |
285 | /* |
278 | * Register several long options. options is a string of long options, |
286 | * Register several long options. options is a string of long options, |
279 | * separated by commas or whitespace. This nukes options! |
287 | * separated by commas or whitespace. This nukes options! |
280 | */ |
288 | */ |
281 | static void add_long_options(char *options) |
289 | static void add_long_options(char *options) |
282 | { |
290 | { |
… | |
… | |
366 | {"alternative", no_argument, NULL, 'a'}, |
374 | {"alternative", no_argument, NULL, 'a'}, |
367 | {"name", required_argument, NULL, 'n'}, |
375 | {"name", required_argument, NULL, 'n'}, |
368 | {"version", no_argument, NULL, 'V'}, |
376 | {"version", no_argument, NULL, 'V'}, |
369 | {NULL, 0, NULL, 0} |
377 | {NULL, 0, NULL, 0} |
370 | }; |
378 | }; |
|
|
379 | |
371 | setlocale(LC_ALL, ""); |
380 | setlocale(LC_ALL, ""); |
372 | bindtextdomain(PACKAGE, LOCALEDIR); |
381 | bindtextdomain(PACKAGE, LOCALEDIR); |
373 | textdomain(PACKAGE); |
382 | textdomain(PACKAGE); |
374 | |
383 | |
375 | init_longopt(); |
384 | init_longopt(); |
… | |
… | |
388 | return EXIT_SUCCESS; |
397 | return EXIT_SUCCESS; |
389 | } else |
398 | } else |
390 | parse_error(_("missing optstring argument")); |
399 | parse_error(_("missing optstring argument")); |
391 | } |
400 | } |
392 | |
401 | |
393 | |
|
|
394 | if (argv[1][0] != '-' || compatible) { |
402 | if (argv[1][0] != '-' || compatible) { |
395 | quote = 0; |
403 | quote = 0; |
396 | optstr = xmalloc(strlen(argv[1]) + 1); |
404 | optstr = xmalloc(strlen(argv[1]) + 1); |
397 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); |
405 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); |
398 | argv[1] = argv[0]; |
406 | argv[1] = argv[0]; |
… | |
… | |
442 | case ':': |
450 | case ':': |
443 | parse_error(NULL); |
451 | parse_error(NULL); |
444 | default: |
452 | default: |
445 | parse_error(_("internal error, contact the author.")); |
453 | parse_error(_("internal error, contact the author.")); |
446 | } |
454 | } |
447 | |
455 | |
448 | if (!optstr) { |
456 | if (!optstr) { |
449 | if (optind >= argc) |
457 | if (optind >= argc) |
450 | parse_error(_("missing optstring argument")); |
458 | parse_error(_("missing optstring argument")); |
451 | else { |
459 | else { |
452 | optstr = xmalloc(strlen(argv[optind]) + 1); |
460 | optstr = xmalloc(strlen(argv[optind]) + 1); |
… | |
… | |
456 | } |
464 | } |
457 | if (name) |
465 | if (name) |
458 | argv[optind - 1] = name; |
466 | argv[optind - 1] = name; |
459 | else |
467 | else |
460 | argv[optind - 1] = argv[0]; |
468 | argv[optind - 1] = argv[0]; |
|
|
469 | |
461 | return generate_output(argv + optind - 1, argc-optind + 1, |
470 | return generate_output(argv + optind - 1, argc-optind + 1, |
462 | optstr, long_options); |
471 | optstr, long_options); |
463 | } |
472 | } |