| 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 | */ | 
| … |  | … |  | 
| 63 | #include <getopt.h> | 65 | #include <getopt.h> | 
| 64 | #else | 66 | #else | 
| 65 | #include "getopt.h" | 67 | #include "getopt.h" | 
| 66 | #endif | 68 | #endif | 
| 67 |  | 69 |  | 
|  |  | 70 | #include "util-linux-compat.h" | 
| 68 | #include "nls.h" | 71 | #include "nls.h" | 
| 69 | #include "xalloc.h" | 72 | #include "xalloc.h" | 
| 70 |  | 73 |  | 
| 71 | /* 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 '+' | 
| 72 | * mode */ | 75 | * mode */ | 
| 73 | #define NON_OPT 1 | 76 | #define NON_OPT 1 | 
| 74 | /* 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. */ | 
| 75 | #define LONG_OPT 2 | 78 | #define LONG_OPT 0 | 
| 76 |  | 79 |  | 
| 77 | /* The shells recognized. */ | 80 | /* The shells recognized. */ | 
| 78 | typedef enum { BASH, TCSH } shell_t; | 81 | typedef enum { BASH, TCSH } shell_t; | 
| 79 |  | 82 |  | 
| 80 |  | 83 |  | 
| … |  | … |  | 
| 90 |  | 93 |  | 
| 91 | /* Function prototypes */ | 94 | /* Function prototypes */ | 
| 92 | static const char *normalize(const char *arg); | 95 | static const char *normalize(const char *arg); | 
| 93 | static int generate_output(char *argv[], int argc, const char *optstr, | 96 | static int generate_output(char *argv[], int argc, const char *optstr, | 
| 94 | const struct option *longopts); | 97 | const struct option *longopts); | 
| 95 | int main(int argc, char *argv[]); |  |  | 
| 96 | static void parse_error(const char *message); | 98 | static void parse_error(const char *message); | 
| 97 | static void add_long_options(char *options); | 99 | static void add_long_options(char *options); | 
| 98 | static void add_longopt(const char *name, int has_arg); | 100 | static void add_longopt(const char *name, int has_arg); | 
| 99 | static void print_help(void); | 101 | static void print_help(void); | 
| 100 | static void set_shell(const char *new_shell); | 102 | static void set_shell(const char *new_shell); | 
| … |  | … |  | 
| 122 | strcpy(BUFFER, arg); | 124 | strcpy(BUFFER, arg); | 
| 123 | return BUFFER; | 125 | return BUFFER; | 
| 124 | } | 126 | } | 
| 125 |  | 127 |  | 
| 126 | /* | 128 | /* | 
| 127 | * Each character in arg may take upto four characters in the | 129 | * Each character in arg may take up to four characters in the | 
| 128 | * 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 | 
| 129 | * 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 | 
| 130 | * quote, and one extra character for '\0'. | 132 | * quote, and one extra character for '\0'. | 
| 131 | */ | 133 | */ | 
| 132 | BUFFER = xmalloc(strlen(arg) * 4 + 3); | 134 | BUFFER = xmalloc(strlen(arg) * 4 + 3); | 
| … |  | … |  | 
| 165 | *bufptr++ = '\''; | 167 | *bufptr++ = '\''; | 
| 166 | *bufptr++ = '\0'; | 168 | *bufptr++ = '\0'; | 
| 167 | return BUFFER; | 169 | return BUFFER; | 
| 168 | } | 170 | } | 
| 169 |  | 171 |  | 
| 170 | /* | 172 | /* | 
| 171 | * 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). | 
| 172 | * 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 | 
| 173 | * 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), | 
| 174 | * optstr must contain the short options, and longopts the long options. | 176 | * optstr must contain the short options, and longopts the long options. | 
| 175 | * Other settings are found in global variables. | 177 | * Other settings are found in global variables. | 
| … |  | … |  | 
| 222 | * 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. | 
| 223 | */ | 225 | */ | 
| 224 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) | 226 | static void __attribute__ ((__noreturn__)) parse_error(const char *message) | 
| 225 | { | 227 | { | 
| 226 | if (message) | 228 | if (message) | 
| 227 | fprintf(stderr, "%s: %s\n", program_invocation_short_name, message); | 229 | warnx("%s", message); | 
| 228 | fprintf(stderr, _("Try `%s --help' for more information.\n"), | 230 | fprintf(stderr, _("Try `%s --help' for more information.\n"), | 
| 229 | program_invocation_short_name); | 231 | program_invocation_short_name); | 
| 230 | exit(PARAMETER_EXIT_CODE); | 232 | exit(PARAMETER_EXIT_CODE); | 
| 231 | } | 233 | } | 
| 232 |  | 234 |  | 
| … |  | … |  | 
| 238 |  | 240 |  | 
| 239 | /* Register a long option. The contents of name is copied. */ | 241 | /* Register a long option. The contents of name is copied. */ | 
| 240 | static void add_longopt(const char *name, int has_arg) | 242 | static void add_longopt(const char *name, int has_arg) | 
| 241 | { | 243 | { | 
| 242 | char *tmp; | 244 | char *tmp; | 
|  |  | 245 | static int flag; | 
|  |  | 246 |  | 
| 243 | if (!name) { | 247 | if (!name) { | 
| 244 | /* init */ | 248 | /* init */ | 
| 245 | free(long_options); | 249 | free(long_options); | 
| 246 | long_options = NULL; | 250 | long_options = NULL; | 
| 247 | long_options_length = 0; | 251 | long_options_length = 0; | 
| … |  | … |  | 
| 261 | long_options[long_options_nr].val = 0; | 265 | long_options[long_options_nr].val = 0; | 
| 262 |  | 266 |  | 
| 263 | if (long_options_nr && name) { | 267 | if (long_options_nr && name) { | 
| 264 | /* Not for init! */ | 268 | /* Not for init! */ | 
| 265 | long_options[long_options_nr - 1].has_arg = has_arg; | 269 | long_options[long_options_nr - 1].has_arg = has_arg; | 
| 266 | long_options[long_options_nr - 1].flag = NULL; | 270 | long_options[long_options_nr - 1].flag = &flag; | 
| 267 | long_options[long_options_nr - 1].val = LONG_OPT; | 271 | long_options[long_options_nr - 1].val = long_options_nr; | 
| 268 | tmp = xmalloc(strlen(name) + 1); | 272 | tmp = xmalloc(strlen(name) + 1); | 
| 269 | strcpy(tmp, name); | 273 | strcpy(tmp, name); | 
| 270 | long_options[long_options_nr - 1].name = tmp; | 274 | long_options[long_options_nr - 1].name = tmp; | 
| 271 | } | 275 | } | 
| 272 | long_options_nr++; | 276 | long_options_nr++; | 
| 273 | } | 277 | } | 
| 274 |  | 278 |  | 
| 275 |  | 279 |  | 
| 276 | /* | 280 | /* | 
| 277 | * Register several long options. options is a string of long options, | 281 | * Register several long options. options is a string of long options, | 
| 278 | * separated by commas or whitespace. This nukes options! | 282 | * separated by commas or whitespace. This nukes options! | 
| 279 | */ | 283 | */ | 
| 280 | static void add_long_options(char *options) | 284 | static void add_long_options(char *options) | 
| 281 | { | 285 | { | 
| … |  | … |  | 
| 318 | ("unknown shell after -s or --shell argument")); | 322 | ("unknown shell after -s or --shell argument")); | 
| 319 | } | 323 | } | 
| 320 |  | 324 |  | 
| 321 | static void __attribute__ ((__noreturn__)) print_help(void) | 325 | static void __attribute__ ((__noreturn__)) print_help(void) | 
| 322 | { | 326 | { | 
| 323 | fputs(_("Usage: getopt optstring parameters\n"), stderr); | 327 | fputs(USAGE_HEADER, stderr); | 
| 324 | fputs(_("       getopt [options] [--] optstring parameters\n"), stderr); | 328 | fprintf(stderr, _( | 
| 325 | fputs(_("       getopt [options] -o|--options optstring [options] [--]\n"), stderr); | 329 | " %1$s optstring parameters\n" | 
| 326 | fputs(_("              parameters\n"), stderr); | 330 | " %1$s [options] [--] optstring parameters\n" | 
| 327 | fputs(_("\nOptions:\n"), stderr); | 331 | " %1$s [options] -o|--options optstring [options] [--] parameters\n"), | 
|  |  | 332 | program_invocation_short_name); | 
|  |  | 333 |  | 
|  |  | 334 | fputs(USAGE_OPTIONS, stderr); | 
| 328 | fputs(_(" -a, --alternative            Allow long options starting with single -\n"), stderr); | 335 | fputs(_(" -a, --alternative            Allow long options starting with single -\n"), stderr); | 
| 329 | fputs(_(" -h, --help                   This small usage guide\n"), stderr); |  |  | 
| 330 | fputs(_(" -l, --longoptions <longopts> Long options to be recognized\n"), stderr); | 336 | fputs(_(" -l, --longoptions <longopts> Long options to be recognized\n"), stderr); | 
| 331 | fputs(_(" -n, --name <progname>        The name under which errors are reported\n"), stderr); | 337 | fputs(_(" -n, --name <progname>        The name under which errors are reported\n"), stderr); | 
| 332 | fputs(_(" -o, --options <optstring>    Short options to be recognized\n"), stderr); | 338 | fputs(_(" -o, --options <optstring>    Short options to be recognized\n"), stderr); | 
| 333 | fputs(_(" -q, --quiet                  Disable error reporting by getopt(3)\n"), stderr); | 339 | fputs(_(" -q, --quiet                  Disable error reporting by getopt(3)\n"), stderr); | 
| 334 | fputs(_(" -Q, --quiet-output           No normal output\n"), stderr); | 340 | fputs(_(" -Q, --quiet-output           No normal output\n"), stderr); | 
| 335 | fputs(_(" -s, --shell <shell>          Set shell quoting conventions\n"), stderr); | 341 | fputs(_(" -s, --shell <shell>          Set shell quoting conventions\n"), stderr); | 
| 336 | fputs(_(" -T, --test                   Test for getopt(1) version\n"), stderr); | 342 | fputs(_(" -T, --test                   Test for getopt(1) version\n"), stderr); | 
| 337 | fputs(_(" -u, --unquoted               Do not quote the output\n"), stderr); | 343 | fputs(_(" -u, --unquoted               Do not quote the output\n"), stderr); | 
| 338 | fputs(_(" -V, --version                Output version information\n"), stderr); | 344 | fputs(USAGE_SEPARATOR, stderr); | 
| 339 | fputc('\n', stderr); | 345 | fputs(USAGE_HELP, stderr); | 
| 340 |  | 346 | fputs(USAGE_VERSION, stderr); | 
|  |  | 347 | fprintf(stderr, USAGE_MAN_TAIL("getopt(1)")); | 
| 341 | exit(PARAMETER_EXIT_CODE); | 348 | exit(PARAMETER_EXIT_CODE); | 
| 342 | } | 349 | } | 
| 343 |  | 350 |  | 
| 344 | int main(int argc, char *argv[]) | 351 | int main(int argc, char *argv[]) | 
| 345 | { | 352 | { | 
| … |  | … |  | 
| 362 | {"alternative", no_argument, NULL, 'a'}, | 369 | {"alternative", no_argument, NULL, 'a'}, | 
| 363 | {"name", required_argument, NULL, 'n'}, | 370 | {"name", required_argument, NULL, 'n'}, | 
| 364 | {"version", no_argument, NULL, 'V'}, | 371 | {"version", no_argument, NULL, 'V'}, | 
| 365 | {NULL, 0, NULL, 0} | 372 | {NULL, 0, NULL, 0} | 
| 366 | }; | 373 | }; | 
|  |  | 374 |  | 
| 367 | setlocale(LC_ALL, ""); | 375 | setlocale(LC_ALL, ""); | 
| 368 | bindtextdomain(PACKAGE, LOCALEDIR); | 376 | bindtextdomain(PACKAGE, LOCALEDIR); | 
| 369 | textdomain(PACKAGE); | 377 | textdomain(PACKAGE); | 
| 370 |  | 378 |  | 
| 371 | init_longopt(); | 379 | init_longopt(); | 
| … |  | … |  | 
| 384 | return EXIT_SUCCESS; | 392 | return EXIT_SUCCESS; | 
| 385 | } else | 393 | } else | 
| 386 | parse_error(_("missing optstring argument")); | 394 | parse_error(_("missing optstring argument")); | 
| 387 | } | 395 | } | 
| 388 |  | 396 |  | 
| 389 |  |  |  | 
| 390 | if (argv[1][0] != '-' || compatible) { | 397 | if (argv[1][0] != '-' || compatible) { | 
| 391 | quote = 0; | 398 | quote = 0; | 
| 392 | optstr = xmalloc(strlen(argv[1]) + 1); | 399 | optstr = xmalloc(strlen(argv[1]) + 1); | 
| 393 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); | 400 | strcpy(optstr, argv[1] + strspn(argv[1], "-+")); | 
| 394 | argv[1] = argv[0]; | 401 | argv[1] = argv[0]; | 
| … |  | … |  | 
| 430 | return TEST_EXIT_CODE; | 437 | return TEST_EXIT_CODE; | 
| 431 | case 'u': | 438 | case 'u': | 
| 432 | quote = 0; | 439 | quote = 0; | 
| 433 | break; | 440 | break; | 
| 434 | case 'V': | 441 | case 'V': | 
| 435 | printf(_("%s (enhanced) %s\n"), program_invocation_short_name, program_version); | 442 | printf(UTIL_LINUX_VERSION); | 
| 436 | return EXIT_SUCCESS; | 443 | return EXIT_SUCCESS; | 
| 437 | case '?': | 444 | case '?': | 
| 438 | case ':': | 445 | case ':': | 
| 439 | parse_error(NULL); | 446 | parse_error(NULL); | 
| 440 | default: | 447 | default: | 
| 441 | parse_error(_("internal error, contact the author.")); | 448 | parse_error(_("internal error, contact the author.")); | 
| 442 | } | 449 | } | 
| 443 |  | 450 |  | 
| 444 | if (!optstr) { | 451 | if (!optstr) { | 
| 445 | if (optind >= argc) | 452 | if (optind >= argc) | 
| 446 | parse_error(_("missing optstring argument")); | 453 | parse_error(_("missing optstring argument")); | 
| 447 | else { | 454 | else { | 
| 448 | optstr = xmalloc(strlen(argv[optind]) + 1); | 455 | optstr = xmalloc(strlen(argv[optind]) + 1); | 
| … |  | … |  | 
| 452 | } | 459 | } | 
| 453 | if (name) | 460 | if (name) | 
| 454 | argv[optind - 1] = name; | 461 | argv[optind - 1] = name; | 
| 455 | else | 462 | else | 
| 456 | argv[optind - 1] = argv[0]; | 463 | argv[optind - 1] = argv[0]; | 
|  |  | 464 |  | 
| 457 | return generate_output(argv + optind - 1, argc-optind + 1, | 465 | return generate_output(argv + optind - 1, argc-optind + 1, | 
| 458 | optstr, long_options); | 466 | optstr, long_options); | 
| 459 | } | 467 | } |