| 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 ambigious 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 | } |