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 |
42 | */ |
44 | */ |
43 | |
45 | |
44 | /* Exit codes: |
46 | /* Exit codes: |
45 | * 0) No errors, succesful operation. |
47 | * 0) No errors, successful operation. |
46 | * 1) getopt(3) returned an error. |
48 | * 1) getopt(3) returned an error. |
47 | * 2) A problem with parameter parsing for getopt(1). |
49 | * 2) A problem with parameter parsing for getopt(1). |
48 | * 3) Internal error, out of memory |
50 | * 3) Internal error, out of memory |
49 | * 4) Returned for -T |
51 | * 4) Returned for -T |
50 | */ |
52 | */ |
… | |
… | |
67 | |
69 | |
68 | #include "util-linux-compat.h" |
70 | #include "util-linux-compat.h" |
69 | #include "nls.h" |
71 | #include "nls.h" |
70 | #include "xalloc.h" |
72 | #include "xalloc.h" |
71 | |
73 | |
72 | /* NON_OPT is the code that is returned when a non-option is found in '+' |
74 | /* NON_OPT is the code that is returned when a non-option is found in '+' |
73 | * mode */ |
75 | * mode */ |
74 | #define NON_OPT 1 |
76 | #define NON_OPT 1 |
75 | /* LONG_OPT is the code that is returned when a long option is found. */ |
77 | /* LONG_OPT is the code that is returned when a long option is found. */ |
76 | #define LONG_OPT 2 |
78 | #define LONG_OPT 0 |
77 | |
79 | |
78 | /* The shells recognized. */ |
80 | /* The shells recognized. */ |
79 | typedef enum { BASH, TCSH } shell_t; |
81 | typedef enum { BASH, TCSH } shell_t; |
80 | |
82 | |
81 | |
83 | |
… | |
… | |
91 | |
93 | |
92 | /* Function prototypes */ |
94 | /* Function prototypes */ |
93 | static const char *normalize(const char *arg); |
95 | static const char *normalize(const char *arg); |
94 | static int generate_output(char *argv[], int argc, const char *optstr, |
96 | static int generate_output(char *argv[], int argc, const char *optstr, |
95 | const struct option *longopts); |
97 | const struct option *longopts); |
96 | int main(int argc, char *argv[]); |
|
|
97 | static void parse_error(const char *message); |
98 | static void parse_error(const char *message); |
98 | static void add_long_options(char *options); |
99 | static void add_long_options(char *options); |
99 | static void add_longopt(const char *name, int has_arg); |
100 | static void add_longopt(const char *name, int has_arg); |
100 | static void print_help(void); |
101 | static void print_help(void); |
101 | static void set_shell(const char *new_shell); |
102 | static void set_shell(const char *new_shell); |
… | |
… | |
123 | strcpy(BUFFER, arg); |
124 | strcpy(BUFFER, arg); |
124 | return BUFFER; |
125 | return BUFFER; |
125 | } |
126 | } |
126 | |
127 | |
127 | /* |
128 | /* |
128 | * Each character in arg may take upto four characters in the |
129 | * 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 |
130 | * 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 |
131 | * and an opening quote! We need also the global opening and closing |
131 | * quote, and one extra character for '\0'. |
132 | * quote, and one extra character for '\0'. |
132 | */ |
133 | */ |
133 | BUFFER = xmalloc(strlen(arg) * 4 + 3); |
134 | BUFFER = xmalloc(strlen(arg) * 4 + 3); |
… | |
… | |
166 | *bufptr++ = '\''; |
167 | *bufptr++ = '\''; |
167 | *bufptr++ = '\0'; |
168 | *bufptr++ = '\0'; |
168 | return BUFFER; |
169 | return BUFFER; |
169 | } |
170 | } |
170 | |
171 | |
171 | /* |
172 | /* |
172 | * Generate the output. argv[0] is the program name (used for reporting errors). |
173 | * 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 |
174 | * 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), |
175 | * 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. |
176 | * optstr must contain the short options, and longopts the long options. |
176 | * Other settings are found in global variables. |
177 | * Other settings are found in global variables. |
… | |
… | |
223 | * we already sent a message, we just exit with a helpful hint. |
224 | * we already sent a message, we just exit with a helpful hint. |
224 | */ |
225 | */ |
225 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) |
226 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) |
226 | { |
227 | { |
227 | if (message) |
228 | if (message) |
228 | fprintf(stderr, "%s: %s\n", program_invocation_short_name, message); |
229 | warnx("%s", message); |
229 | fprintf(stderr, _("Try `%s --help' for more information.\n"), |
230 | fprintf(stderr, _("Try `%s --help' for more information.\n"), |
230 | program_invocation_short_name); |
231 | program_invocation_short_name); |
231 | exit(PARAMETER_EXIT_CODE); |
232 | exit(PARAMETER_EXIT_CODE); |
232 | } |
233 | } |
233 | |
234 | |
… | |
… | |
239 | |
240 | |
240 | /* Register a long option. The contents of name is copied. */ |
241 | /* Register a long option. The contents of name is copied. */ |
241 | static void add_longopt(const char *name, int has_arg) |
242 | static void add_longopt(const char *name, int has_arg) |
242 | { |
243 | { |
243 | char *tmp; |
244 | char *tmp; |
|
|
245 | static int flag; |
|
|
246 | |
244 | if (!name) { |
247 | if (!name) { |
245 | /* init */ |
248 | /* init */ |
246 | free(long_options); |
249 | free(long_options); |
247 | long_options = NULL; |
250 | long_options = NULL; |
248 | long_options_length = 0; |
251 | long_options_length = 0; |
… | |
… | |
262 | long_options[long_options_nr].val = 0; |
265 | long_options[long_options_nr].val = 0; |
263 | |
266 | |
264 | if (long_options_nr && name) { |
267 | if (long_options_nr && name) { |
265 | /* Not for init! */ |
268 | /* Not for init! */ |
266 | long_options[long_options_nr - 1].has_arg = has_arg; |
269 | long_options[long_options_nr - 1].has_arg = has_arg; |
267 | long_options[long_options_nr - 1].flag = NULL; |
270 | long_options[long_options_nr - 1].flag = &flag; |
268 | long_options[long_options_nr - 1].val = LONG_OPT; |
271 | long_options[long_options_nr - 1].val = long_options_nr; |
269 | tmp = xmalloc(strlen(name) + 1); |
272 | tmp = xmalloc(strlen(name) + 1); |
270 | strcpy(tmp, name); |
273 | strcpy(tmp, name); |
271 | long_options[long_options_nr - 1].name = tmp; |
274 | long_options[long_options_nr - 1].name = tmp; |
272 | } |
275 | } |
273 | long_options_nr++; |
276 | long_options_nr++; |
274 | } |
277 | } |
275 | |
278 | |
276 | |
279 | |
277 | /* |
280 | /* |
278 | * Register several long options. options is a string of long options, |
281 | * Register several long options. options is a string of long options, |
279 | * separated by commas or whitespace. This nukes options! |
282 | * separated by commas or whitespace. This nukes options! |
280 | */ |
283 | */ |
281 | static void add_long_options(char *options) |
284 | static void add_long_options(char *options) |
282 | { |
285 | { |
… | |
… | |
366 | {"alternative", no_argument, NULL, 'a'}, |
369 | {"alternative", no_argument, NULL, 'a'}, |
367 | {"name", required_argument, NULL, 'n'}, |
370 | {"name", required_argument, NULL, 'n'}, |
368 | {"version", no_argument, NULL, 'V'}, |
371 | {"version", no_argument, NULL, 'V'}, |
369 | {NULL, 0, NULL, 0} |
372 | {NULL, 0, NULL, 0} |
370 | }; |
373 | }; |
|
|
374 | |
371 | setlocale(LC_ALL, ""); |
375 | setlocale(LC_ALL, ""); |
372 | bindtextdomain(PACKAGE, LOCALEDIR); |
376 | bindtextdomain(PACKAGE, LOCALEDIR); |
373 | textdomain(PACKAGE); |
377 | textdomain(PACKAGE); |
374 | |
378 | |
375 | init_longopt(); |
379 | init_longopt(); |
… | |
… | |
388 | return EXIT_SUCCESS; |
392 | return EXIT_SUCCESS; |
389 | } else |
393 | } else |
390 | parse_error(_("missing optstring argument")); |
394 | parse_error(_("missing optstring argument")); |
391 | } |
395 | } |
392 | |
396 | |
393 | |
|
|
394 | if (argv[1][0] != '-' || compatible) { |
397 | if (argv[1][0] != '-' || compatible) { |
395 | quote = 0; |
398 | quote = 0; |
396 | optstr = xmalloc(strlen(argv[1]) + 1); |
399 | optstr = xmalloc(strlen(argv[1]) + 1); |
397 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); |
400 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); |
398 | argv[1] = argv[0]; |
401 | argv[1] = argv[0]; |
… | |
… | |
442 | case ':': |
445 | case ':': |
443 | parse_error(NULL); |
446 | parse_error(NULL); |
444 | default: |
447 | default: |
445 | parse_error(_("internal error, contact the author.")); |
448 | parse_error(_("internal error, contact the author.")); |
446 | } |
449 | } |
447 | |
450 | |
448 | if (!optstr) { |
451 | if (!optstr) { |
449 | if (optind >= argc) |
452 | if (optind >= argc) |
450 | parse_error(_("missing optstring argument")); |
453 | parse_error(_("missing optstring argument")); |
451 | else { |
454 | else { |
452 | optstr = xmalloc(strlen(argv[optind]) + 1); |
455 | optstr = xmalloc(strlen(argv[optind]) + 1); |
… | |
… | |
456 | } |
459 | } |
457 | if (name) |
460 | if (name) |
458 | argv[optind - 1] = name; |
461 | argv[optind - 1] = name; |
459 | else |
462 | else |
460 | argv[optind - 1] = argv[0]; |
463 | argv[optind - 1] = argv[0]; |
|
|
464 | |
461 | return generate_output(argv + optind - 1, argc-optind + 1, |
465 | return generate_output(argv + optind - 1, argc-optind + 1, |
462 | optstr, long_options); |
466 | optstr, long_options); |
463 | } |
467 | } |