blob: 7c65a6a1f92edaa6ff0f5eeb648241d34a641b1e [file] [log] [blame]
"Robert P. J. Day"63fc1a92006-07-02 19:47:05 +00001/* vi: set sw=4 ts=4: */
Eric Andersena1f16bb2000-08-21 22:02:34 +00002/*
3 * getopt.c - Enhanced implementation of BSD getopt(1)
4 * Copyright (c) 1997, 1998, 1999, 2000 Frodo Looijaard <frodol@dds.nl>
5 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02006 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
Eric Andersena1f16bb2000-08-21 22:02:34 +00007 */
8
9/*
10 * Version 1.0-b4: Tue Sep 23 1997. First public release.
11 * Version 1.0: Wed Nov 19 1997.
12 * Bumped up the version number to 1.0
13 * Fixed minor typo (CSH instead of TCSH)
14 * Version 1.0.1: Tue Jun 3 1998
15 * Fixed sizeof instead of strlen bug
16 * Bumped up the version number to 1.0.1
17 * Version 1.0.2: Thu Jun 11 1998 (not present)
18 * Fixed gcc-2.8.1 warnings
19 * Fixed --version/-V option (not present)
20 * Version 1.0.5: Tue Jun 22 1999
21 * Make -u option work (not present)
22 * Version 1.0.6: Tue Jun 27 2000
23 * No important changes
24 * Version 1.1.0: Tue Jun 30 2000
Denis Vlasenkob44c7902008-03-17 09:29:43 +000025 * Added NLS support (partly written by Arkadiusz Mickiewicz
Eric Andersena1f16bb2000-08-21 22:02:34 +000026 * <misiek@misiek.eu.org>)
27 * Ported to Busybox - Alfred M. Szmidt <ams@trillian.itslinux.org>
Denys Vlasenko488dd702011-05-29 04:24:13 +020028 * Removed --version/-V and --help/-h
Eric Andersenaff114c2004-04-14 17:51:38 +000029 * Removed parse_error(), using bb_error_msg() from Busybox instead
Eric Andersena1f16bb2000-08-21 22:02:34 +000030 * Replaced our_malloc with xmalloc and our_realloc with xrealloc
Eric Andersena1f16bb2000-08-21 22:02:34 +000031 */
Denys Vlasenkodd898c92016-11-23 11:46:32 +010032//config:config GETOPT
Denys Vlasenko4eed2c62017-07-18 22:01:24 +020033//config: bool "getopt (5.6 kb)"
Denys Vlasenkodd898c92016-11-23 11:46:32 +010034//config: default y
35//config: help
36//config: The getopt utility is used to break up (parse) options in command
37//config: lines to make it easy to write complex shell scripts that also check
38//config: for legal (and illegal) options. If you want to write horribly
39//config: complex shell scripts, or use some horribly complex shell script
40//config: written by others, this utility may be for you. Most people will
41//config: wisely leave this disabled.
42//config:
43//config:config FEATURE_GETOPT_LONG
44//config: bool "Support option -l"
45//config: default y if LONG_OPTS
46//config: depends on GETOPT
47//config: help
48//config: Enable support for long options (option -l).
49
50//applet:IF_GETOPT(APPLET(getopt, BB_DIR_BIN, BB_SUID_DROP))
51
52//kbuild:lib-$(CONFIG_GETOPT) += getopt.o
Eric Andersena1f16bb2000-08-21 22:02:34 +000053
Pere Orga5bc8c002011-04-11 03:29:49 +020054//usage:#define getopt_trivial_usage
Denys Vlasenko594db1e2012-02-09 18:17:29 +010055//usage: "[OPTIONS] [--] OPTSTRING PARAMS"
Pere Orga5bc8c002011-04-11 03:29:49 +020056//usage:#define getopt_full_usage "\n\n"
Pere Orga5bc8c002011-04-11 03:29:49 +020057//usage: IF_LONG_OPTS(
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010058//usage: IF_FEATURE_GETOPT_LONG(
59//usage: " -a,--alternative Allow long options starting with single -\n"
Denys Vlasenkobbc7bee2017-01-21 02:49:58 +010060//usage: " -l,--longoptions LOPT[,...] Long options to recognize\n"
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010061//usage: )
Denys Vlasenkobbc7bee2017-01-21 02:49:58 +010062//usage: " -n,--name PROGNAME The name under which errors are reported"
63//usage: "\n -o,--options OPTSTRING Short options to recognize"
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010064//usage: "\n -q,--quiet No error messages on unrecognized options"
Pere Orga5bc8c002011-04-11 03:29:49 +020065//usage: "\n -Q,--quiet-output No normal output"
Denys Vlasenkobbc7bee2017-01-21 02:49:58 +010066//usage: "\n -s,--shell SHELL Set shell quoting conventions"
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010067//usage: "\n -T,--test Version test (exits with 4)"
68//usage: "\n -u,--unquoted Don't quote output"
Pere Orga5bc8c002011-04-11 03:29:49 +020069//usage: )
70//usage: IF_NOT_LONG_OPTS(
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010071//usage: IF_FEATURE_GETOPT_LONG(
72//usage: " -a Allow long options starting with single -\n"
73//usage: " -l LOPT[,...] Long options to recognize\n"
74//usage: )
75//usage: " -n PROGNAME The name under which errors are reported"
76//usage: "\n -o OPTSTRING Short options to recognize"
77//usage: "\n -q No error messages on unrecognized options"
Pere Orga5bc8c002011-04-11 03:29:49 +020078//usage: "\n -Q No normal output"
Denys Vlasenko594db1e2012-02-09 18:17:29 +010079//usage: "\n -s SHELL Set shell quoting conventions"
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010080//usage: "\n -T Version test (exits with 4)"
81//usage: "\n -u Don't quote output"
Pere Orga5bc8c002011-04-11 03:29:49 +020082//usage: )
Denys Vlasenkoee3bc702013-01-24 11:36:00 +010083//usage: IF_FEATURE_GETOPT_LONG( /* example uses -l, needs FEATURE_GETOPT_LONG */
Denys Vlasenko594db1e2012-02-09 18:17:29 +010084//usage: "\n"
85//usage: "\nExample:"
86//usage: "\n"
Denys Vlasenko817c2302012-02-09 18:39:16 +010087//usage: "\nO=`getopt -l bb: -- ab:c:: \"$@\"` || exit 1"
Denys Vlasenko594db1e2012-02-09 18:17:29 +010088//usage: "\neval set -- \"$O\""
89//usage: "\nwhile true; do"
90//usage: "\n case \"$1\" in"
91//usage: "\n -a) echo A; shift;;"
92//usage: "\n -b|--bb) echo \"B:'$2'\"; shift 2;;"
93//usage: "\n -c) case \"$2\" in"
94//usage: "\n \"\") echo C; shift 2;;"
95//usage: "\n *) echo \"C:'$2'\"; shift 2;;"
96//usage: "\n esac;;"
97//usage: "\n --) shift; break;;"
98//usage: "\n *) echo Error; exit 1;;"
99//usage: "\n esac"
100//usage: "\ndone"
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100101//usage: )
Pere Orga5bc8c002011-04-11 03:29:49 +0200102//usage:
103//usage:#define getopt_example_usage
104//usage: "$ cat getopt.test\n"
105//usage: "#!/bin/sh\n"
106//usage: "GETOPT=`getopt -o ab:c:: --long a-long,b-long:,c-long:: \\\n"
107//usage: " -n 'example.busybox' -- \"$@\"`\n"
108//usage: "if [ $? != 0 ]; then exit 1; fi\n"
109//usage: "eval set -- \"$GETOPT\"\n"
110//usage: "while true; do\n"
111//usage: " case $1 in\n"
112//usage: " -a|--a-long) echo \"Option a\"; shift;;\n"
113//usage: " -b|--b-long) echo \"Option b, argument '$2'\"; shift 2;;\n"
114//usage: " -c|--c-long)\n"
115//usage: " case \"$2\" in\n"
116//usage: " \"\") echo \"Option c, no argument\"; shift 2;;\n"
117//usage: " *) echo \"Option c, argument '$2'\"; shift 2;;\n"
118//usage: " esac;;\n"
119//usage: " --) shift; break;;\n"
120//usage: " *) echo \"Internal error!\"; exit 1;;\n"
121//usage: " esac\n"
122//usage: "done\n"
123
Denys Vlasenko488dd702011-05-29 04:24:13 +0200124#if ENABLE_FEATURE_GETOPT_LONG
125# include <getopt.h>
126#endif
Denis Vlasenkob6adbf12007-05-26 19:00:18 +0000127#include "libbb.h"
Eric Andersena1f16bb2000-08-21 22:02:34 +0000128
129/* NON_OPT is the code that is returned when a non-option is found in '+'
130 mode */
Rob Landleybc68cd12006-03-10 19:22:06 +0000131enum {
132 NON_OPT = 1,
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200133#if ENABLE_FEATURE_GETOPT_LONG
Eric Andersena1f16bb2000-08-21 22:02:34 +0000134/* LONG_OPT is the code that is returned when a long option is found. */
Rob Landleybc68cd12006-03-10 19:22:06 +0000135 LONG_OPT = 2
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000136#endif
Rob Landleybc68cd12006-03-10 19:22:06 +0000137};
Eric Andersena1f16bb2000-08-21 22:02:34 +0000138
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000139/* For finding activated option flags. Must match getopt32 call! */
140enum {
141 OPT_o = 0x1, // -o
142 OPT_n = 0x2, // -n
143 OPT_q = 0x4, // -q
144 OPT_Q = 0x8, // -Q
145 OPT_s = 0x10, // -s
146 OPT_T = 0x20, // -T
147 OPT_u = 0x40, // -u
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200148#if ENABLE_FEATURE_GETOPT_LONG
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000149 OPT_a = 0x80, // -a
150 OPT_l = 0x100, // -l
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000151#endif
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000152 SHELL_IS_TCSH = 0x8000, /* hijack this bit for other purposes */
153};
Eric Andersena1f16bb2000-08-21 22:02:34 +0000154
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000155/* 0 is getopt_long, 1 is getopt_long_only */
156#define alternative (option_mask32 & OPT_a)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000157
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000158#define quiet_errors (option_mask32 & OPT_q)
159#define quiet_output (option_mask32 & OPT_Q)
160#define quote (!(option_mask32 & OPT_u))
161#define shell_TCSH (option_mask32 & SHELL_IS_TCSH)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000162
163/*
164 * This function 'normalizes' a single argument: it puts single quotes around
165 * it and escapes other special characters. If quote is false, it just
166 * returns its argument.
167 * Bash only needs special treatment for single quotes; tcsh also recognizes
168 * exclamation marks within single quotes, and nukes whitespace.
169 * This function returns a pointer to a buffer that is overwritten by
170 * each call.
171 */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000172static const char *normalize(const char *arg)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000173{
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000174 char *bufptr;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000175#if ENABLE_FEATURE_CLEAN_UP
176 static char *BUFFER = NULL;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000177 free(BUFFER);
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000178#else
179 char *BUFFER;
180#endif
Eric Andersena1f16bb2000-08-21 22:02:34 +0000181
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000182 if (!quote) { /* Just copy arg */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000183 BUFFER = xstrdup(arg);
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000184 return BUFFER;
185 }
Eric Andersena1f16bb2000-08-21 22:02:34 +0000186
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000187 /* Each character in arg may take up to four characters in the result:
188 For a quote we need a closing quote, a backslash, a quote and an
189 opening quote! We need also the global opening and closing quote,
190 and one extra character for '\0'. */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000191 BUFFER = xmalloc(strlen(arg)*4 + 3);
Eric Andersena1f16bb2000-08-21 22:02:34 +0000192
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000193 bufptr = BUFFER;
194 *bufptr ++= '\'';
Eric Andersena1f16bb2000-08-21 22:02:34 +0000195
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000196 while (*arg) {
197 if (*arg == '\'') {
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000198 /* Quote: replace it with: '\'' */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000199 *bufptr ++= '\'';
200 *bufptr ++= '\\';
201 *bufptr ++= '\'';
202 *bufptr ++= '\'';
203 } else if (shell_TCSH && *arg == '!') {
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000204 /* Exclamation mark: replace it with: \! */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000205 *bufptr ++= '\'';
206 *bufptr ++= '\\';
207 *bufptr ++= '!';
208 *bufptr ++= '\'';
209 } else if (shell_TCSH && *arg == '\n') {
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000210 /* Newline: replace it with: \n */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000211 *bufptr ++= '\\';
212 *bufptr ++= 'n';
213 } else if (shell_TCSH && isspace(*arg)) {
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000214 /* Non-newline whitespace: replace it with \<ws> */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000215 *bufptr ++= '\'';
216 *bufptr ++= '\\';
217 *bufptr ++= *arg;
218 *bufptr ++= '\'';
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000219 } else
220 /* Just copy */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000221 *bufptr ++= *arg;
222 arg++;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000223 }
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000224 *bufptr ++= '\'';
225 *bufptr ++= '\0';
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000226 return BUFFER;
Eric Andersena1f16bb2000-08-21 22:02:34 +0000227}
228
229/*
230 * Generate the output. argv[0] is the program name (used for reporting errors).
231 * argv[1..] contains the options to be parsed. argc must be the number of
232 * elements in argv (ie. 1 if there are no options, only the program name),
233 * optstr must contain the short options, and longopts the long options.
234 * Other settings are found in global variables.
235 */
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200236#if !ENABLE_FEATURE_GETOPT_LONG
Denis Vlasenkod1660cb2008-10-20 07:52:33 +0000237#define generate_output(argv,argc,optstr,longopts) \
238 generate_output(argv,argc,optstr)
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000239#endif
240static int generate_output(char **argv, int argc, const char *optstr, const struct option *longopts)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000241{
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000242 int exit_code = 0; /* We assume everything will be OK */
Eric Andersena1f16bb2000-08-21 22:02:34 +0000243
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000244 if (quiet_errors) /* No error reporting from getopt(3) */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000245 opterr = 0;
Denis Vlasenko97728162008-01-28 22:57:10 +0000246
Denis Vlasenko7effd7a2008-10-21 12:36:43 +0000247 /* We used it already in main() in getopt32(),
248 * we *must* reset getopt(3): */
Kaarle Ritvanen835ad3a2017-04-12 00:58:46 +0300249 GETOPT_RESET();
Denis Vlasenko7effd7a2008-10-21 12:36:43 +0000250
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000251 while (1) {
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200252#if ENABLE_FEATURE_GETOPT_LONG
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100253 int longindex;
254 int opt = alternative
255 ? getopt_long_only(argc, argv, optstr, longopts, &longindex)
256 : getopt_long(argc, argv, optstr, longopts, &longindex)
257 ;
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000258#else
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100259 int opt = getopt(argc, argv, optstr);
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000260#endif
Denis Vlasenko6b06cb82008-05-15 21:30:45 +0000261 if (opt == -1)
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000262 break;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000263 if (opt == '?' || opt == ':' )
264 exit_code = 1;
265 else if (!quiet_output) {
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200266#if ENABLE_FEATURE_GETOPT_LONG
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000267 if (opt == LONG_OPT) {
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000268 printf(" --%s", longopts[longindex].name);
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000269 if (longopts[longindex].has_arg)
270 printf(" %s",
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000271 normalize(optarg ? optarg : ""));
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000272 } else
273#endif
274 if (opt == NON_OPT)
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000275 printf(" %s", normalize(optarg));
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000276 else {
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100277 const char *charptr;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000278 printf(" -%c", opt);
Denis Vlasenko6b06cb82008-05-15 21:30:45 +0000279 charptr = strchr(optstr, opt);
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100280 if (charptr && *++charptr == ':')
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000281 printf(" %s",
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000282 normalize(optarg ? optarg : ""));
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000283 }
284 }
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000285 }
Eric Andersena1f16bb2000-08-21 22:02:34 +0000286
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000287 if (!quiet_output) {
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100288 unsigned idx;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000289 printf(" --");
Denys Vlasenkoee3bc702013-01-24 11:36:00 +0100290 idx = optind;
291 while (argv[idx])
292 printf(" %s", normalize(argv[idx++]));
Denis Vlasenko4daad902007-09-27 10:20:47 +0000293 bb_putchar('\n');
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000294 }
295 return exit_code;
Eric Andersena1f16bb2000-08-21 22:02:34 +0000296}
297
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200298#if ENABLE_FEATURE_GETOPT_LONG
Eric Andersena1f16bb2000-08-21 22:02:34 +0000299/*
300 * Register several long options. options is a string of long options,
301 * separated by commas or whitespace.
302 * This nukes options!
303 */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000304static struct option *add_long_options(struct option *long_options, char *options)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000305{
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000306 int long_nr = 0;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000307 int arg_opt, tlen;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000308 char *tokptr = strtok(options, ", \t\n");
309
310 if (long_options)
311 while (long_options[long_nr].name)
312 long_nr++;
313
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000314 while (tokptr) {
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000315 arg_opt = no_argument;
316 tlen = strlen(tokptr);
317 if (tlen) {
318 tlen--;
319 if (tokptr[tlen] == ':') {
320 arg_opt = required_argument;
321 if (tlen && tokptr[tlen-1] == ':') {
322 tlen--;
323 arg_opt = optional_argument;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000324 }
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000325 tokptr[tlen] = '\0';
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000326 if (tlen == 0)
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000327 bb_error_msg_and_die("empty long option specified");
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000328 }
Denis Vlasenkodeeed592008-07-08 05:14:36 +0000329 long_options = xrealloc_vector(long_options, 4, long_nr);
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000330 long_options[long_nr].has_arg = arg_opt;
Denis Vlasenko27842282008-08-04 13:20:36 +0000331 /*long_options[long_nr].flag = NULL; - xrealloc_vector did it */
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000332 long_options[long_nr].val = LONG_OPT;
333 long_options[long_nr].name = xstrdup(tokptr);
334 long_nr++;
Denis Vlasenko27842282008-08-04 13:20:36 +0000335 /*memset(&long_options[long_nr], 0, sizeof(long_options[0])); - xrealloc_vector did it */
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000336 }
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000337 tokptr = strtok(NULL, ", \t\n");
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000338 }
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000339 return long_options;
Eric Andersena1f16bb2000-08-21 22:02:34 +0000340}
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000341#endif
Eric Andersena1f16bb2000-08-21 22:02:34 +0000342
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000343static void set_shell(const char *new_shell)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000344{
Denys Vlasenko1d3a04a2016-11-28 01:22:57 +0100345 if (strcmp(new_shell, "bash") == 0 || strcmp(new_shell, "sh") == 0)
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000346 return;
Denys Vlasenko1d3a04a2016-11-28 01:22:57 +0100347 if (strcmp(new_shell, "tcsh") == 0 || strcmp(new_shell, "csh") == 0)
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000348 option_mask32 |= SHELL_IS_TCSH;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000349 else
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000350 bb_error_msg("unknown shell '%s', assuming bash", new_shell);
Eric Andersena1f16bb2000-08-21 22:02:34 +0000351}
352
353
354/* Exit codes:
Eric Andersenaff114c2004-04-14 17:51:38 +0000355 * 0) No errors, successful operation.
Eric Andersena1f16bb2000-08-21 22:02:34 +0000356 * 1) getopt(3) returned an error.
357 * 2) A problem with parameter parsing for getopt(1).
358 * 3) Internal error, out of memory
359 * 4) Returned for -T
360 */
361
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200362#if ENABLE_FEATURE_GETOPT_LONG
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000363static const char getopt_longopts[] ALIGN1 =
Denis Vlasenkobdc88fd2007-07-23 17:14:14 +0000364 "options\0" Required_argument "o"
365 "longoptions\0" Required_argument "l"
366 "quiet\0" No_argument "q"
367 "quiet-output\0" No_argument "Q"
368 "shell\0" Required_argument "s"
369 "test\0" No_argument "T"
370 "unquoted\0" No_argument "u"
371 "alternative\0" No_argument "a"
372 "name\0" Required_argument "n"
Denis Vlasenko990d0f62007-07-24 15:54:42 +0000373 ;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000374#endif
Eric Andersena1f16bb2000-08-21 22:02:34 +0000375
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000376int getopt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
377int getopt_main(int argc, char **argv)
Eric Andersena1f16bb2000-08-21 22:02:34 +0000378{
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100379 int n;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000380 char *optstr = NULL;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000381 char *name = NULL;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000382 unsigned opt;
383 const char *compatible;
Denis Vlasenko4b924f32007-05-30 00:29:55 +0000384 char *s_arg;
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200385#if ENABLE_FEATURE_GETOPT_LONG
Denis Vlasenko80d14be2007-04-10 23:03:30 +0000386 struct option *long_options = NULL;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000387 llist_t *l_arg = NULL;
388#endif
Eric Andersena1f16bb2000-08-21 22:02:34 +0000389
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000390 compatible = getenv("GETOPT_COMPATIBLE"); /* used as yes/no flag */
Eric Andersena1f16bb2000-08-21 22:02:34 +0000391
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100392 if (!argv[1]) {
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000393 if (compatible) {
394 /* For some reason, the original getopt gave no error
Denys Vlasenko60cb48c2013-01-14 15:57:44 +0100395 * when there were no arguments. */
Denys Vlasenkod60752f2015-10-07 22:42:45 +0200396 puts(" --");
Denis Vlasenkof47ff102006-09-22 08:42:06 +0000397 return 0;
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000398 }
399 bb_error_msg_and_die("missing optstring argument");
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000400 }
Eric Andersena1f16bb2000-08-21 22:02:34 +0000401
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000402 if (argv[1][0] != '-' || compatible) {
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100403 char *s = argv[1];
Eric Andersena1f16bb2000-08-21 22:02:34 +0000404
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000405 option_mask32 |= OPT_u; /* quoting off */
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100406 s = xstrdup(s + strspn(s, "-+"));
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000407 argv[1] = argv[0];
408 return generate_output(argv+1, argc-1, s, long_options);
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000409 }
Eric Andersena1f16bb2000-08-21 22:02:34 +0000410
Denys Vlasenkof3b92d32009-06-19 12:10:38 +0200411#if !ENABLE_FEATURE_GETOPT_LONG
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000412 opt = getopt32(argv, "+o:n:qQs:Tu", &optstr, &name, &s_arg);
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000413#else
Denis Vlasenkobdc88fd2007-07-23 17:14:14 +0000414 applet_long_options = getopt_longopts;
Denys Vlasenko237bedd2016-07-06 21:58:02 +0200415 opt = getopt32(argv, "+o:n:qQs:Tual:*",
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000416 &optstr, &name, &s_arg, &l_arg);
417 /* Effectuate the read options for the applet itself */
418 while (l_arg) {
Denis Vlasenkod50dda82008-06-15 05:40:56 +0000419 long_options = add_long_options(long_options, llist_pop(&l_arg));
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000420 }
421#endif
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000422
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000423 if (opt & OPT_s) {
424 set_shell(s_arg);
425 }
426
427 if (opt & OPT_T) {
428 return 4;
429 }
430
431 /* All options controlling the applet have now been parsed */
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100432 n = optind - 1;
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000433 if (!optstr) {
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100434 optstr = argv[++n];
435 if (!optstr)
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000436 bb_error_msg_and_die("missing optstring argument");
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000437 }
Denis Vlasenko9c146a92007-04-07 10:25:04 +0000438
Denys Vlasenko594db1e2012-02-09 18:17:29 +0100439 argv[n] = name ? name : argv[0];
440 return generate_output(argv + n, argc - n, optstr, long_options);
Eric Andersena1f16bb2000-08-21 22:02:34 +0000441}