blob: 8a4215d51d5ca792db8c519c07ee9887f707e9f2 [file] [log] [blame]
Eric Andersen25f27032001-04-26 23:22:31 +00001/* vi: set sw=4 ts=4: */
2/*
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003 * A prototype Bourne shell grammar parser.
4 * Intended to follow the original Thompson and Ritchie
5 * "small and simple is beautiful" philosophy, which
6 * incidentally is a good match to today's BusyBox.
Eric Andersen25f27032001-04-26 23:22:31 +00007 *
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +00008 * Copyright (C) 2000,2001 Larry Doolittle <larry@doolittle.boa.org>
Denis Vlasenkoc8d27332009-04-06 10:47:21 +00009 * Copyright (C) 2008,2009 Denys Vlasenko <vda.linux@googlemail.com>
Eric Andersen25f27032001-04-26 23:22:31 +000010 *
Denys Vlasenkobbecd742010-10-03 17:22:52 +020011 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
12 *
Eric Andersen25f27032001-04-26 23:22:31 +000013 * Credits:
14 * The parser routines proper are all original material, first
Eric Andersencb81e642003-07-14 21:21:08 +000015 * written Dec 2000 and Jan 2001 by Larry Doolittle. The
16 * execution engine, the builtins, and much of the underlying
17 * support has been adapted from busybox-0.49pre's lash, which is
Eric Andersenc7bda1c2004-03-15 08:29:22 +000018 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
Eric Andersencb81e642003-07-14 21:21:08 +000019 * written by Erik Andersen <andersen@codepoet.org>. That, in turn,
20 * is based in part on ladsh.c, by Michael K. Johnson and Erik W.
21 * Troan, which they placed in the public domain. I don't know
22 * how much of the Johnson/Troan code has survived the repeated
23 * rewrites.
24 *
Eric Andersen25f27032001-04-26 23:22:31 +000025 * Other credits:
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +000026 * o_addchr derived from similar w_addchar function in glibc-2.2.
Denis Vlasenko50f3aa42009-04-07 10:52:40 +000027 * parse_redirect, redirect_opt_num, and big chunks of main
Denis Vlasenko424f79b2009-03-22 14:23:34 +000028 * and many builtins derived from contributions by Erik Andersen.
29 * Miscellaneous bugfixes from Matt Kraai.
Eric Andersen25f27032001-04-26 23:22:31 +000030 *
31 * There are two big (and related) architecture differences between
32 * this parser and the lash parser. One is that this version is
33 * actually designed from the ground up to understand nearly all
34 * of the Bourne grammar. The second, consequential change is that
35 * the parser and input reader have been turned inside out. Now,
36 * the parser is in control, and asks for input as needed. The old
37 * way had the input reader in control, and it asked for parsing to
38 * take place as needed. The new way makes it much easier to properly
39 * handle the recursion implicit in the various substitutions, especially
40 * across continuation lines.
41 *
Denys Vlasenko349ef962010-05-21 15:46:24 +020042 * TODOs:
43 * grep for "TODO" and fix (some of them are easy)
44 * special variables (done: PWD, PPID, RANDOM)
45 * tilde expansion
Eric Andersen78a7c992001-05-15 16:30:25 +000046 * aliases
Denys Vlasenko349ef962010-05-21 15:46:24 +020047 * follow IFS rules more precisely, including update semantics
48 * builtins mandated by standards we don't support:
49 * [un]alias, command, fc, getopts, newgrp, readonly, times
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +020050 * make complex ${var%...} constructs support optional
51 * make here documents optional
Mike Frysinger25a6ca02009-03-28 13:59:26 +000052 *
Denys Vlasenkoadc0e202010-05-17 18:56:58 +020053 * Bash compat TODO:
54 * redirection of stdout+stderr: &> and >&
Denys Vlasenkoadc0e202010-05-17 18:56:58 +020055 * reserved words: function select
56 * advanced test: [[ ]]
Denys Vlasenkoadc0e202010-05-17 18:56:58 +020057 * process substitution: <(list) and >(list)
58 * =~: regex operator
Denys Vlasenko9ca656b2009-06-10 13:39:35 +020059 * let EXPR [EXPR...]
Denys Vlasenko349ef962010-05-21 15:46:24 +020060 * Each EXPR is an arithmetic expression (ARITHMETIC EVALUATION)
61 * If the last arg evaluates to 0, let returns 1; 0 otherwise.
62 * NB: let `echo 'a=a + 1'` - error (IOW: multi-word expansion is used)
Denys Vlasenko9ca656b2009-06-10 13:39:35 +020063 * ((EXPR))
Denys Vlasenko349ef962010-05-21 15:46:24 +020064 * The EXPR is evaluated according to ARITHMETIC EVALUATION.
65 * This is exactly equivalent to let "EXPR".
Denys Vlasenkoadc0e202010-05-17 18:56:58 +020066 * $[EXPR]: synonym for $((EXPR))
Denys Vlasenkobbecd742010-10-03 17:22:52 +020067 *
68 * Won't do:
69 * In bash, export builtin is special, its arguments are assignments
Denys Vlasenko08218012009-06-03 14:43:56 +020070 * and therefore expansion of them should be "one-word" expansion:
71 * $ export i=`echo 'a b'` # export has one arg: "i=a b"
72 * compare with:
73 * $ ls i=`echo 'a b'` # ls has two args: "i=a" and "b"
74 * ls: cannot access i=a: No such file or directory
75 * ls: cannot access b: No such file or directory
Denys Vlasenko9ca656b2009-06-10 13:39:35 +020076 * Note1: same applies to local builtin.
Denys Vlasenko08218012009-06-03 14:43:56 +020077 * Note2: bash 3.2.33(1) does this only if export word itself
78 * is not quoted:
79 * $ export i=`echo 'aaa bbb'`; echo "$i"
80 * aaa bbb
81 * $ "export" i=`echo 'aaa bbb'`; echo "$i"
82 * aaa
Eric Andersen25f27032001-04-26 23:22:31 +000083 */
Denys Vlasenko8da415e2010-12-05 01:30:14 +010084#if !(defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__) \
85 || defined(__APPLE__) \
86 )
87# include <malloc.h> /* for malloc_trim */
88#endif
Denis Vlasenkobe709c22008-07-28 00:01:16 +000089#include <glob.h>
90/* #include <dmalloc.h> */
91#if ENABLE_HUSH_CASE
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +000092# include <fnmatch.h>
Denis Vlasenkobe709c22008-07-28 00:01:16 +000093#endif
Denys Vlasenko03dad222010-01-12 23:29:57 +010094
Denys Vlasenko20704f02011-03-23 17:59:27 +010095#include "busybox.h" /* for APPLET_IS_NOFORK/NOEXEC */
96#include "unicode.h"
Denys Vlasenko03dad222010-01-12 23:29:57 +010097#include "shell_common.h"
Mike Frysinger98c52642009-04-02 10:02:37 +000098#include "math.h"
Mike Frysingera4f331d2009-04-07 06:03:22 +000099#include "match.h"
Denys Vlasenkocbe0b7f2009-10-09 22:00:58 +0200100#if ENABLE_HUSH_RANDOM_SUPPORT
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200101# include "random.h"
Denys Vlasenko76ace252009-10-12 15:25:01 +0200102#else
103# define CLEAR_RANDOM_T(rnd) ((void)0)
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200104#endif
Denis Vlasenko50f3aa42009-04-07 10:52:40 +0000105#ifndef PIPE_BUF
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200106# define PIPE_BUF 4096 /* amount of buffering in a pipe */
Denis Vlasenko50f3aa42009-04-07 10:52:40 +0000107#endif
Mike Frysinger98c52642009-04-02 10:02:37 +0000108
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200109//config:config HUSH
110//config: bool "hush"
111//config: default y
112//config: help
Denys Vlasenko771f1992010-07-16 14:31:34 +0200113//config: hush is a small shell (25k). It handles the normal flow control
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200114//config: constructs such as if/then/elif/else/fi, for/in/do/done, while loops,
115//config: case/esac. Redirections, here documents, $((arithmetic))
116//config: and functions are supported.
117//config:
118//config: It will compile and work on no-mmu systems.
119//config:
Denys Vlasenkoe2069fb2010-10-04 00:01:47 +0200120//config: It does not handle select, aliases, tilde expansion,
121//config: &>file and >&file redirection of stdout+stderr.
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200122//config:
123//config:config HUSH_BASH_COMPAT
124//config: bool "bash-compatible extensions"
125//config: default y
126//config: depends on HUSH
127//config: help
128//config: Enable bash-compatible extensions.
129//config:
Denys Vlasenko9e800222010-10-03 14:28:04 +0200130//config:config HUSH_BRACE_EXPANSION
131//config: bool "Brace expansion"
132//config: default y
133//config: depends on HUSH_BASH_COMPAT
134//config: help
135//config: Enable {abc,def} extension.
136//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200137//config:config HUSH_HELP
138//config: bool "help builtin"
139//config: default y
140//config: depends on HUSH
141//config: help
142//config: Enable help builtin in hush. Code size + ~1 kbyte.
143//config:
144//config:config HUSH_INTERACTIVE
145//config: bool "Interactive mode"
146//config: default y
147//config: depends on HUSH
148//config: help
149//config: Enable interactive mode (prompt and command editing).
150//config: Without this, hush simply reads and executes commands
151//config: from stdin just like a shell script from a file.
152//config: No prompt, no PS1/PS2 magic shell variables.
153//config:
Denys Vlasenko99862cb2010-09-12 17:34:13 +0200154//config:config HUSH_SAVEHISTORY
155//config: bool "Save command history to .hush_history"
156//config: default y
157//config: depends on HUSH_INTERACTIVE && FEATURE_EDITING_SAVEHISTORY
158//config: help
159//config: Enable history saving in hush.
160//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200161//config:config HUSH_JOB
162//config: bool "Job control"
163//config: default y
164//config: depends on HUSH_INTERACTIVE
165//config: help
166//config: Enable job control: Ctrl-Z backgrounds, Ctrl-C interrupts current
167//config: command (not entire shell), fg/bg builtins work. Without this option,
168//config: "cmd &" still works by simply spawning a process and immediately
169//config: prompting for next command (or executing next command in a script),
170//config: but no separate process group is formed.
171//config:
172//config:config HUSH_TICK
173//config: bool "Process substitution"
174//config: default y
175//config: depends on HUSH
176//config: help
177//config: Enable process substitution `command` and $(command) in hush.
178//config:
179//config:config HUSH_IF
180//config: bool "Support if/then/elif/else/fi"
181//config: default y
182//config: depends on HUSH
183//config: help
184//config: Enable if/then/elif/else/fi in hush.
185//config:
186//config:config HUSH_LOOPS
187//config: bool "Support for, while and until loops"
188//config: default y
189//config: depends on HUSH
190//config: help
191//config: Enable for, while and until loops in hush.
192//config:
193//config:config HUSH_CASE
194//config: bool "Support case ... esac statement"
195//config: default y
196//config: depends on HUSH
197//config: help
198//config: Enable case ... esac statement in hush. +400 bytes.
199//config:
200//config:config HUSH_FUNCTIONS
201//config: bool "Support funcname() { commands; } syntax"
202//config: default y
203//config: depends on HUSH
204//config: help
205//config: Enable support for shell functions in hush. +800 bytes.
206//config:
207//config:config HUSH_LOCAL
208//config: bool "Support local builtin"
209//config: default y
210//config: depends on HUSH_FUNCTIONS
211//config: help
212//config: Enable support for local variables in functions.
213//config:
214//config:config HUSH_RANDOM_SUPPORT
215//config: bool "Pseudorandom generator and $RANDOM variable"
216//config: default y
217//config: depends on HUSH
218//config: help
219//config: Enable pseudorandom generator and dynamic variable "$RANDOM".
220//config: Each read of "$RANDOM" will generate a new pseudorandom value.
221//config:
222//config:config HUSH_EXPORT_N
223//config: bool "Support 'export -n' option"
224//config: default y
225//config: depends on HUSH
226//config: help
227//config: export -n unexports variables. It is a bash extension.
228//config:
229//config:config HUSH_MODE_X
230//config: bool "Support 'hush -x' option and 'set -x' command"
231//config: default y
232//config: depends on HUSH
233//config: help
Denys Vlasenko29082232010-07-16 13:52:32 +0200234//config: This instructs hush to print commands before execution.
235//config: Adds ~300 bytes.
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200236//config:
Denys Vlasenko6adf2aa2010-07-16 19:26:38 +0200237//config:config MSH
238//config: bool "msh (deprecated: aliased to hush)"
239//config: default n
240//config: select HUSH
241//config: help
242//config: msh is deprecated and will be removed, please migrate to hush.
243//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200244
Denys Vlasenko20704f02011-03-23 17:59:27 +0100245//applet:IF_HUSH(APPLET(hush, BB_DIR_BIN, BB_SUID_DROP))
246//applet:IF_MSH(APPLET(msh, BB_DIR_BIN, BB_SUID_DROP))
247//applet:IF_FEATURE_SH_IS_HUSH(APPLET_ODDNAME(sh, hush, BB_DIR_BIN, BB_SUID_DROP, sh))
248//applet:IF_FEATURE_BASH_IS_HUSH(APPLET_ODDNAME(bash, hush, BB_DIR_BIN, BB_SUID_DROP, bash))
249
250//kbuild:lib-$(CONFIG_HUSH) += hush.o match.o shell_common.o
251//kbuild:lib-$(CONFIG_HUSH_RANDOM_SUPPORT) += random.o
252
Dan Fandrich89ca2f92010-11-28 01:54:39 +0100253/* -i (interactive) and -s (read stdin) are also accepted,
254 * but currently do nothing, therefore aren't shown in help.
255 * NOMMU-specific options are not meant to be used by users,
256 * therefore we don't show them either.
257 */
258//usage:#define hush_trivial_usage
Denys Vlasenkof58f7052011-05-12 02:10:33 +0200259//usage: "[-nxl] [-c 'SCRIPT' [ARG0 [ARGS]] / FILE [ARGS]]"
Denys Vlasenkob0b83432011-03-07 12:34:59 +0100260//usage:#define hush_full_usage "\n\n"
261//usage: "Unix shell interpreter"
262
Dan Fandrich89ca2f92010-11-28 01:54:39 +0100263//usage:#define msh_trivial_usage hush_trivial_usage
Denys Vlasenkob0b83432011-03-07 12:34:59 +0100264//usage:#define msh_full_usage hush_full_usage
265
266//usage:#if ENABLE_FEATURE_SH_IS_HUSH
267//usage:# define sh_trivial_usage hush_trivial_usage
268//usage:# define sh_full_usage hush_full_usage
269//usage:#endif
270//usage:#if ENABLE_FEATURE_BASH_IS_HUSH
271//usage:# define bash_trivial_usage hush_trivial_usage
272//usage:# define bash_full_usage hush_full_usage
273//usage:#endif
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200274
Denis Vlasenko1943aec2009-04-09 14:15:57 +0000275
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200276/* Build knobs */
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +0000277#define LEAK_HUNTING 0
278#define BUILD_AS_NOMMU 0
279/* Enable/disable sanity checks. Ok to enable in production,
280 * only adds a bit of bloat. Set to >1 to get non-production level verbosity.
281 * Keeping 1 for now even in released versions.
282 */
283#define HUSH_DEBUG 1
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200284/* Slightly bigger (+200 bytes), but faster hush.
285 * So far it only enables a trick with counting SIGCHLDs and forks,
286 * which allows us to do fewer waitpid's.
287 * (we can detect a case where neither forks were done nor SIGCHLDs happened
288 * and therefore waitpid will return the same result as last time)
289 */
290#define ENABLE_HUSH_FAST 0
Denys Vlasenko9297dbc2010-07-05 21:37:12 +0200291/* TODO: implement simplified code for users which do not need ${var%...} ops
292 * So far ${var%...} ops are always enabled:
293 */
294#define ENABLE_HUSH_DOLLAR_OPS 1
Denis Vlasenko1943aec2009-04-09 14:15:57 +0000295
296
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +0000297#if BUILD_AS_NOMMU
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000298# undef BB_MMU
299# undef USE_FOR_NOMMU
300# undef USE_FOR_MMU
301# define BB_MMU 0
302# define USE_FOR_NOMMU(...) __VA_ARGS__
303# define USE_FOR_MMU(...)
304#endif
305
Denys Vlasenko1fcbff22010-06-26 02:40:08 +0200306#include "NUM_APPLETS.h"
Denys Vlasenko14974842010-03-23 01:08:26 +0100307#if NUM_APPLETS == 1
Denis Vlasenko61befda2008-11-25 01:36:03 +0000308/* STANDALONE does not make sense, and won't compile */
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000309# undef CONFIG_FEATURE_SH_STANDALONE
310# undef ENABLE_FEATURE_SH_STANDALONE
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000311# undef IF_FEATURE_SH_STANDALONE
Denys Vlasenko14974842010-03-23 01:08:26 +0100312# undef IF_NOT_FEATURE_SH_STANDALONE
313# define ENABLE_FEATURE_SH_STANDALONE 0
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000314# define IF_FEATURE_SH_STANDALONE(...)
315# define IF_NOT_FEATURE_SH_STANDALONE(...) __VA_ARGS__
Denis Vlasenko61befda2008-11-25 01:36:03 +0000316#endif
317
Denis Vlasenko05743d72008-02-10 12:10:08 +0000318#if !ENABLE_HUSH_INTERACTIVE
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000319# undef ENABLE_FEATURE_EDITING
320# define ENABLE_FEATURE_EDITING 0
321# undef ENABLE_FEATURE_EDITING_FANCY_PROMPT
322# define ENABLE_FEATURE_EDITING_FANCY_PROMPT 0
Tanguy Pruvot8a6c2c22012-04-28 00:24:09 +0200323# undef ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
324# define ENABLE_FEATURE_EDITING_SAVE_ON_EXIT 0
Denis Vlasenko8412d792007-10-01 09:59:47 +0000325#endif
326
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000327/* Do we support ANY keywords? */
328#if ENABLE_HUSH_IF || ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000329# define HAS_KEYWORDS 1
330# define IF_HAS_KEYWORDS(...) __VA_ARGS__
331# define IF_HAS_NO_KEYWORDS(...)
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000332#else
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000333# define HAS_KEYWORDS 0
334# define IF_HAS_KEYWORDS(...)
335# define IF_HAS_NO_KEYWORDS(...) __VA_ARGS__
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000336#endif
Denis Vlasenko8412d792007-10-01 09:59:47 +0000337
Denis Vlasenkod01ff132007-05-02 21:40:23 +0000338/* If you comment out one of these below, it will be #defined later
339 * to perform debug printfs to stderr: */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000340#define debug_printf(...) do {} while (0)
Denis Vlasenko400c5b62007-05-04 13:07:27 +0000341/* Finer-grained debug switches */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000342#define debug_printf_parse(...) do {} while (0)
343#define debug_print_tree(a, b) do {} while (0)
344#define debug_printf_exec(...) do {} while (0)
Denis Vlasenkof886fd22008-10-13 12:36:05 +0000345#define debug_printf_env(...) do {} while (0)
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000346#define debug_printf_jobs(...) do {} while (0)
347#define debug_printf_expand(...) do {} while (0)
Denys Vlasenko1e811b12010-05-22 03:12:29 +0200348#define debug_printf_varexp(...) do {} while (0)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +0000349#define debug_printf_glob(...) do {} while (0)
350#define debug_printf_list(...) do {} while (0)
Denis Vlasenko30c9cc52008-06-17 07:24:29 +0000351#define debug_printf_subst(...) do {} while (0)
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000352#define debug_printf_clean(...) do {} while (0)
Denis Vlasenkod01ff132007-05-02 21:40:23 +0000353
Denis Vlasenkob6e65562009-04-03 16:49:04 +0000354#define ERR_PTR ((void*)(long)1)
355
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200356#define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000357
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200358#define _SPECIAL_VARS_STR "_*@$!?#"
359#define SPECIAL_VARS_STR ("_*@$!?#" + 1)
360#define NUMERIC_SPECVARS_STR ("_*@$!?#" + 3)
Denys Vlasenko36f774a2010-09-05 14:45:38 +0200361#if ENABLE_HUSH_BASH_COMPAT
362/* Support / and // replace ops */
363/* Note that // is stored as \ in "encoded" string representation */
364# define VAR_ENCODED_SUBST_OPS "\\/%#:-=+?"
365# define VAR_SUBST_OPS ("\\/%#:-=+?" + 1)
366# define MINUS_PLUS_EQUAL_QUESTION ("\\/%#:-=+?" + 5)
367#else
368# define VAR_ENCODED_SUBST_OPS "%#:-=+?"
369# define VAR_SUBST_OPS "%#:-=+?"
370# define MINUS_PLUS_EQUAL_QUESTION ("%#:-=+?" + 3)
371#endif
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200372
373#define SPECIAL_VAR_SYMBOL 3
Eric Andersen25f27032001-04-26 23:22:31 +0000374
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200375struct variable;
376
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000377static const char hush_version_str[] ALIGN1 = "HUSH_VERSION="BB_VER;
378
379/* This supports saving pointers malloced in vfork child,
Denis Vlasenkoc376db32009-04-15 21:49:48 +0000380 * to be freed in the parent.
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000381 */
382#if !BB_MMU
383typedef struct nommu_save_t {
384 char **new_env;
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200385 struct variable *old_vars;
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000386 char **argv;
Denis Vlasenko27014ed2009-04-15 21:48:23 +0000387 char **argv_from_re_execing;
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000388} nommu_save_t;
389#endif
390
Denys Vlasenko9b782552010-09-08 13:33:26 +0200391enum {
Eric Andersen25f27032001-04-26 23:22:31 +0000392 RES_NONE = 0,
Denis Vlasenko06810332007-05-21 23:30:54 +0000393#if ENABLE_HUSH_IF
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000394 RES_IF ,
395 RES_THEN ,
396 RES_ELIF ,
397 RES_ELSE ,
398 RES_FI ,
Denis Vlasenko06810332007-05-21 23:30:54 +0000399#endif
400#if ENABLE_HUSH_LOOPS
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000401 RES_FOR ,
402 RES_WHILE ,
403 RES_UNTIL ,
404 RES_DO ,
405 RES_DONE ,
Denis Vlasenkod91afa32008-07-29 11:10:01 +0000406#endif
407#if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000408 RES_IN ,
Denis Vlasenko06810332007-05-21 23:30:54 +0000409#endif
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000410#if ENABLE_HUSH_CASE
411 RES_CASE ,
Denys Vlasenkoe9bda902009-05-23 16:50:07 +0200412 /* three pseudo-keywords support contrived "case" syntax: */
413 RES_CASE_IN, /* "case ... IN", turns into RES_MATCH when IN is observed */
414 RES_MATCH , /* "word)" */
415 RES_CASE_BODY, /* "this command is inside CASE" */
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000416 RES_ESAC ,
417#endif
418 RES_XXXX ,
419 RES_SNTX
Denys Vlasenko9b782552010-09-08 13:33:26 +0200420};
Denis Vlasenkoc7985b72008-06-17 05:43:38 +0000421
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000422typedef struct o_string {
423 char *data;
424 int length; /* position where data is appended */
425 int maxlen;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +0200426 int o_expflags;
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000427 /* At least some part of the string was inside '' or "",
428 * possibly empty one: word"", wo''rd etc. */
Denys Vlasenko38292b62010-09-05 14:49:40 +0200429 smallint has_quoted_part;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000430 smallint has_empty_slot;
431 smallint o_assignment; /* 0:maybe, 1:yes, 2:no */
432} o_string;
433enum {
Denys Vlasenko0e13b402010-09-21 12:35:39 +0200434 EXP_FLAG_SINGLEWORD = 0x80, /* must be 0x80 */
435 EXP_FLAG_GLOB = 0x2,
436 /* Protect newly added chars against globbing
437 * by prepending \ to *, ?, [, \ */
438 EXP_FLAG_ESC_GLOB_CHARS = 0x1,
439};
440enum {
441 MAYBE_ASSIGNMENT = 0,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000442 DEFINITELY_ASSIGNMENT = 1,
Denys Vlasenko0e13b402010-09-21 12:35:39 +0200443 NOT_ASSIGNMENT = 2,
444 /* Not an assigment, but next word may be: "if v=xyz cmd;" */
445 WORD_IS_KEYWORD = 3,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000446};
447/* Used for initialization: o_string foo = NULL_O_STRING; */
448#define NULL_O_STRING { NULL }
449
Denys Vlasenko29f9b722011-05-14 11:27:36 +0200450#ifndef debug_printf_parse
451static const char *const assignment_flag[] = {
452 "MAYBE_ASSIGNMENT",
453 "DEFINITELY_ASSIGNMENT",
454 "NOT_ASSIGNMENT",
455 "WORD_IS_KEYWORD",
456};
457#endif
458
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000459typedef struct in_str {
460 const char *p;
461 /* eof_flag=1: last char in ->p is really an EOF */
462 char eof_flag; /* meaningless if ->p == NULL */
463 char peek_buf[2];
464#if ENABLE_HUSH_INTERACTIVE
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000465 smallint promptmode; /* 0: PS1, 1: PS2 */
466#endif
Denys Vlasenkocecbc982011-03-30 18:54:52 +0200467 int last_char;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000468 FILE *file;
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200469 int (*get) (struct in_str *) FAST_FUNC;
470 int (*peek) (struct in_str *) FAST_FUNC;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000471} in_str;
472#define i_getch(input) ((input)->get(input))
473#define i_peek(input) ((input)->peek(input))
474
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200475/* The descrip member of this structure is only used to make
476 * debugging output pretty */
477static const struct {
478 int mode;
479 signed char default_fd;
480 char descrip[3];
481} redir_table[] = {
482 { O_RDONLY, 0, "<" },
483 { O_CREAT|O_TRUNC|O_WRONLY, 1, ">" },
484 { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
485 { O_CREAT|O_RDWR, 1, "<>" },
486 { O_RDONLY, 0, "<<" },
487/* Should not be needed. Bogus default_fd helps in debugging */
488/* { O_RDONLY, 77, "<<" }, */
489};
490
Eric Andersen25f27032001-04-26 23:22:31 +0000491struct redir_struct {
Denis Vlasenko55789c62008-06-18 16:30:42 +0000492 struct redir_struct *next;
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000493 char *rd_filename; /* filename */
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000494 int rd_fd; /* fd to redirect */
495 /* fd to redirect to, or -3 if rd_fd is to be closed (n>&-) */
496 int rd_dup;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000497 smallint rd_type; /* (enum redir_type) */
498 /* note: for heredocs, rd_filename contains heredoc delimiter,
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000499 * and subsequently heredoc itself; and rd_dup is a bitmask:
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200500 * bit 0: do we need to trim leading tabs?
501 * bit 1: is heredoc quoted (<<'delim' syntax) ?
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000502 */
Eric Andersen25f27032001-04-26 23:22:31 +0000503};
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000504typedef enum redir_type {
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200505 REDIRECT_INPUT = 0,
506 REDIRECT_OVERWRITE = 1,
507 REDIRECT_APPEND = 2,
508 REDIRECT_IO = 3,
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000509 REDIRECT_HEREDOC = 4,
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200510 REDIRECT_HEREDOC2 = 5, /* REDIRECT_HEREDOC after heredoc is loaded */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +0000511
512 REDIRFD_CLOSE = -3,
513 REDIRFD_SYNTAX_ERR = -2,
Denis Vlasenko835fcfd2009-04-10 13:51:56 +0000514 REDIRFD_TO_FILE = -1,
515 /* otherwise, rd_fd is redirected to rd_dup */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +0000516
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000517 HEREDOC_SKIPTABS = 1,
518 HEREDOC_QUOTED = 2,
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000519} redir_type;
520
Eric Andersen25f27032001-04-26 23:22:31 +0000521
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000522struct command {
Denis Vlasenko0c886c62007-01-30 22:30:09 +0000523 pid_t pid; /* 0 if exited */
Denis Vlasenko2b576b82008-08-04 00:46:07 +0000524 int assignment_cnt; /* how many argv[i] are assignments? */
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200525 smallint cmd_type; /* CMD_xxx */
526#define CMD_NORMAL 0
527#define CMD_SUBSHELL 1
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200528#if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenkod383b492010-09-06 10:22:13 +0200529/* used for "[[ EXPR ]]" */
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200530# define CMD_SINGLEWORD_NOGLOB 2
Denis Vlasenkoed055212009-04-11 10:37:10 +0000531#endif
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200532#if ENABLE_HUSH_FUNCTIONS
533# define CMD_FUNCDEF 3
534#endif
535
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100536 smalluint cmd_exitcode;
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200537 /* if non-NULL, this "command" is { list }, ( list ), or a compound statement */
538 struct pipe *group;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000539#if !BB_MMU
540 char *group_as_string;
541#endif
Denis Vlasenkoed055212009-04-11 10:37:10 +0000542#if ENABLE_HUSH_FUNCTIONS
543 struct function *child_func;
544/* This field is used to prevent a bug here:
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200545 * while...do f1() {a;}; f1; f1() {b;}; f1; done
Denis Vlasenkoed055212009-04-11 10:37:10 +0000546 * When we execute "f1() {a;}" cmd, we create new function and clear
547 * cmd->group, cmd->group_as_string, cmd->argv[0].
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200548 * When we execute "f1() {b;}", we notice that f1 exists,
549 * and that its "parent cmd" struct is still "alive",
Denis Vlasenkoed055212009-04-11 10:37:10 +0000550 * we put those fields back into cmd->xxx
551 * (struct function has ->parent_cmd ptr to facilitate that).
552 * When we loop back, we can execute "f1() {a;}" again and set f1 correctly.
553 * Without this trick, loop would execute a;b;b;b;...
554 * instead of correct sequence a;b;a;b;...
555 * When command is freed, it severs the link
556 * (sets ->child_func->parent_cmd to NULL).
557 */
558#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000559 char **argv; /* command name and arguments */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000560/* argv vector may contain variable references (^Cvar^C, ^C0^C etc)
561 * and on execution these are substituted with their values.
562 * Substitution can make _several_ words out of one argv[n]!
563 * Example: argv[0]=='.^C*^C.' here: echo .$*.
Denis Vlasenkoc7985b72008-06-17 05:43:38 +0000564 * References of the form ^C`cmd arg^C are `cmd arg` substitutions.
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000565 */
Denis Vlasenkoed055212009-04-11 10:37:10 +0000566 struct redir_struct *redirects; /* I/O redirections */
567};
Denis Vlasenkof8c1f022009-04-17 11:55:42 +0000568/* Is there anything in this command at all? */
569#define IS_NULL_CMD(cmd) \
570 (!(cmd)->group && !(cmd)->argv && !(cmd)->redirects)
571
Eric Andersen25f27032001-04-26 23:22:31 +0000572struct pipe {
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000573 struct pipe *next;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000574 int num_cmds; /* total number of commands in pipe */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000575 int alive_cmds; /* number of commands running (not exited) */
576 int stopped_cmds; /* number of commands alive, but stopped */
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +0000577#if ENABLE_HUSH_JOB
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000578 int jobid; /* job number */
Denis Vlasenko0c886c62007-01-30 22:30:09 +0000579 pid_t pgrp; /* process group ID for the job */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000580 char *cmdtext; /* name of job */
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000581#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000582 struct command *cmds; /* array of commands in pipe */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000583 smallint followup; /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000584 IF_HAS_KEYWORDS(smallint pi_inverted;) /* "! cmd | cmd" */
585 IF_HAS_KEYWORDS(smallint res_word;) /* needed for if, for, while, until... */
Eric Andersen25f27032001-04-26 23:22:31 +0000586};
Denis Vlasenkoa2b11e32009-04-06 14:11:13 +0000587typedef enum pipe_style {
588 PIPE_SEQ = 1,
589 PIPE_AND = 2,
590 PIPE_OR = 3,
591 PIPE_BG = 4,
592} pipe_style;
Denis Vlasenkof8c1f022009-04-17 11:55:42 +0000593/* Is there anything in this pipe at all? */
594#define IS_NULL_PIPE(pi) \
595 ((pi)->num_cmds == 0 IF_HAS_KEYWORDS( && (pi)->res_word == RES_NONE))
Eric Andersen25f27032001-04-26 23:22:31 +0000596
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000597/* This holds pointers to the various results of parsing */
598struct parse_context {
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000599 /* linked list of pipes */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000600 struct pipe *list_head;
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000601 /* last pipe (being constructed right now) */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000602 struct pipe *pipe;
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000603 /* last command in pipe (being constructed right now) */
604 struct command *command;
605 /* last redirect in command->redirects list */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000606 struct redir_struct *pending_redirect;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000607#if !BB_MMU
608 o_string as_string;
609#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000610#if HAS_KEYWORDS
611 smallint ctx_res_w;
612 smallint ctx_inverted; /* "! cmd | cmd" */
613#if ENABLE_HUSH_CASE
614 smallint ctx_dsemicolon; /* ";;" seen */
615#endif
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000616 /* bitmask of FLAG_xxx, for figuring out valid reserved words */
617 int old_flag;
618 /* group we are enclosed in:
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000619 * example: "if pipe1; pipe2; then pipe3; fi"
620 * when we see "if" or "then", we malloc and copy current context,
621 * and make ->stack point to it. then we parse pipeN.
622 * when closing "then" / fi" / whatever is found,
623 * we move list_head into ->stack->command->group,
624 * copy ->stack into current context, and delete ->stack.
625 * (parsing of { list } and ( list ) doesn't use this method)
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000626 */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000627 struct parse_context *stack;
628#endif
629};
630
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000631/* On program start, environ points to initial environment.
632 * putenv adds new pointers into it, unsetenv removes them.
633 * Neither of these (de)allocates the strings.
634 * setenv allocates new strings in malloc space and does putenv,
635 * and thus setenv is unusable (leaky) for shell's purposes */
636#define setenv(...) setenv_is_leaky_dont_use()
637struct variable {
638 struct variable *next;
Denis Vlasenko28c0f0f2007-05-25 02:46:01 +0000639 char *varstr; /* points to "name=" portion */
Denys Vlasenko295fef82009-06-03 12:47:26 +0200640#if ENABLE_HUSH_LOCAL
641 unsigned func_nest_level;
642#endif
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000643 int max_len; /* if > 0, name is part of initial env; else name is malloced */
644 smallint flg_export; /* putenv should be done on this var */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000645 smallint flg_read_only;
Eric Andersen9ffb7dd2001-05-19 03:00:46 +0000646};
647
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000648enum {
Denis Vlasenkobcb25532008-07-28 23:04:34 +0000649 BC_BREAK = 1,
650 BC_CONTINUE = 2,
651};
652
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000653#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000654struct function {
655 struct function *next;
656 char *name;
Denis Vlasenkoed055212009-04-11 10:37:10 +0000657 struct command *parent_cmd;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000658 struct pipe *body;
Denys Vlasenkoc1947f12009-10-23 01:30:26 +0200659# if !BB_MMU
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000660 char *body_as_string;
Denys Vlasenkoc1947f12009-10-23 01:30:26 +0200661# endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000662};
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000663#endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000664
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000665
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100666/* set -/+o OPT support. (TODO: make it optional)
667 * bash supports the following opts:
668 * allexport off
669 * braceexpand on
670 * emacs on
671 * errexit off
672 * errtrace off
673 * functrace off
674 * hashall on
675 * histexpand off
676 * history on
677 * ignoreeof off
678 * interactive-comments on
679 * keyword off
680 * monitor on
681 * noclobber off
682 * noexec off
683 * noglob off
684 * nolog off
685 * notify off
686 * nounset off
687 * onecmd off
688 * physical off
689 * pipefail off
690 * posix off
691 * privileged off
692 * verbose off
693 * vi off
694 * xtrace off
695 */
Dan Fandrich85c62472010-11-20 13:05:17 -0800696static const char o_opt_strings[] ALIGN1 =
697 "pipefail\0"
698 "noexec\0"
699#if ENABLE_HUSH_MODE_X
700 "xtrace\0"
701#endif
702 ;
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100703enum {
704 OPT_O_PIPEFAIL,
Dan Fandrich85c62472010-11-20 13:05:17 -0800705 OPT_O_NOEXEC,
706#if ENABLE_HUSH_MODE_X
707 OPT_O_XTRACE,
708#endif
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100709 NUM_OPT_O
710};
711
712
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000713/* "Globals" within this file */
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000714/* Sorted roughly by size (smaller offsets == smaller code) */
715struct globals {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000716 /* interactive_fd != 0 means we are an interactive shell.
717 * If we are, then saved_tty_pgrp can also be != 0, meaning
718 * that controlling tty is available. With saved_tty_pgrp == 0,
719 * job control still works, but terminal signals
720 * (^C, ^Z, ^Y, ^\) won't work at all, and background
721 * process groups can only be created with "cmd &".
722 * With saved_tty_pgrp != 0, hush will use tcsetpgrp()
723 * to give tty to the foreground process group,
724 * and will take it back when the group is stopped (^Z)
725 * or killed (^C).
726 */
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000727#if ENABLE_HUSH_INTERACTIVE
728 /* 'interactive_fd' is a fd# open to ctty, if we have one
729 * _AND_ if we decided to act interactively */
730 int interactive_fd;
731 const char *PS1;
732 const char *PS2;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000733# define G_interactive_fd (G.interactive_fd)
Denis Vlasenko60b392f2009-04-03 19:14:32 +0000734#else
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000735# define G_interactive_fd 0
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000736#endif
737#if ENABLE_FEATURE_EDITING
738 line_input_t *line_input_state;
739#endif
Denis Vlasenkocc3f20b2008-06-23 22:31:52 +0000740 pid_t root_pid;
Denys Vlasenkodea47882009-10-09 15:40:49 +0200741 pid_t root_ppid;
Denis Vlasenko87a86552008-07-29 19:43:10 +0000742 pid_t last_bg_pid;
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200743#if ENABLE_HUSH_RANDOM_SUPPORT
744 random_t random_gen;
745#endif
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000746#if ENABLE_HUSH_JOB
747 int run_list_level;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000748 int last_jobid;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000749 pid_t saved_tty_pgrp;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000750 struct pipe *job_list;
Mike Frysinger38478a62009-05-20 04:48:06 -0400751# define G_saved_tty_pgrp (G.saved_tty_pgrp)
752#else
753# define G_saved_tty_pgrp 0
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000754#endif
Denys Vlasenko26777aa2010-11-22 23:49:10 +0100755 char o_opt[NUM_OPT_O];
Denys Vlasenko57542eb2010-11-28 03:59:30 +0100756#if ENABLE_HUSH_MODE_X
757# define G_x_mode (G.o_opt[OPT_O_XTRACE])
758#else
759# define G_x_mode 0
760#endif
Denis Vlasenko422cd7c2009-03-31 12:41:52 +0000761 smallint flag_SIGINT;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000762#if ENABLE_HUSH_LOOPS
Denis Vlasenkobcb25532008-07-28 23:04:34 +0000763 smallint flag_break_continue;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000764#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000765#if ENABLE_HUSH_FUNCTIONS
766 /* 0: outside of a function (or sourced file)
767 * -1: inside of a function, ok to use return builtin
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000768 * 1: return is invoked, skip all till end of func
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000769 */
770 smallint flag_return_in_progress;
771#endif
Denis Vlasenkoefea9d22009-04-09 13:43:11 +0000772 smallint exiting; /* used to prevent EXIT trap recursion */
Denis Vlasenkod5762932009-03-31 11:22:57 +0000773 /* These four support $?, $#, and $1 */
Denis Vlasenkoab2b0642009-04-06 18:42:11 +0000774 smalluint last_exitcode;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000775 /* are global_argv and global_argv[1..n] malloced? (note: not [0]) */
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +0000776 smalluint global_args_malloced;
Denis Vlasenkoe1300f62009-03-22 11:41:18 +0000777 /* how many non-NULL argv's we have. NB: $# + 1 */
778 int global_argc;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000779 char **global_argv;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000780#if !BB_MMU
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +0000781 char *argv0_for_re_execing;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000782#endif
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000783#if ENABLE_HUSH_LOOPS
Denis Vlasenko6a2d40f2008-07-28 23:07:06 +0000784 unsigned depth_break_continue;
Denis Vlasenkofcf37c32008-07-29 11:37:15 +0000785 unsigned depth_of_loop;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000786#endif
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000787 const char *ifs;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000788 const char *cwd;
Denys Vlasenko52e460b2010-09-16 16:12:00 +0200789 struct variable *top_var;
Denys Vlasenko29082232010-07-16 13:52:32 +0200790 char **expanded_assignments;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000791#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000792 struct function *top_func;
Denys Vlasenko295fef82009-06-03 12:47:26 +0200793# if ENABLE_HUSH_LOCAL
794 struct variable **shadowed_vars_pp;
795 unsigned func_nest_level;
796# endif
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000797#endif
Denis Vlasenkod5762932009-03-31 11:22:57 +0000798 /* Signal and trap handling */
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200799#if ENABLE_HUSH_FAST
800 unsigned count_SIGCHLD;
801 unsigned handled_SIGCHLD;
Denys Vlasenkoe2df5f42009-05-26 14:34:10 +0200802 smallint we_have_children;
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200803#endif
Denys Vlasenko10c01312011-05-11 11:49:21 +0200804 /* Which signals have non-DFL handler (even with no traps set)?
805 * Set at the start to:
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200806 * (SIGQUIT + maybe SPECIAL_INTERACTIVE_SIGS + maybe SPECIAL_JOBSTOP_SIGS)
Denys Vlasenko10c01312011-05-11 11:49:21 +0200807 * SPECIAL_INTERACTIVE_SIGS are cleared after fork.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200808 * The rest is cleared right before execv syscalls.
Denys Vlasenko10c01312011-05-11 11:49:21 +0200809 * Other than these two times, never modified.
810 */
811 unsigned special_sig_mask;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200812#if ENABLE_HUSH_JOB
813 unsigned fatal_sig_mask;
Denys Vlasenko75e77de2011-05-12 13:12:47 +0200814# define G_fatal_sig_mask G.fatal_sig_mask
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200815#else
Denys Vlasenko75e77de2011-05-12 13:12:47 +0200816# define G_fatal_sig_mask 0
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200817#endif
Denis Vlasenko7566bae2009-03-31 17:24:49 +0000818 char **traps; /* char *traps[NSIG] */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200819 sigset_t pending_set;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000820#if HUSH_DEBUG
821 unsigned long memleak_value;
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000822 int debug_indent;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000823#endif
Denys Vlasenko0806e402011-05-12 23:06:20 +0200824 struct sigaction sa;
Denys Vlasenkoaaa22d22009-10-19 16:34:39 +0200825 char user_input_buf[ENABLE_FEATURE_EDITING ? CONFIG_FEATURE_EDITING_MAX_LEN : 2];
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000826};
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000827#define G (*ptr_to_globals)
Denis Vlasenko87a86552008-07-29 19:43:10 +0000828/* Not #defining name to G.name - this quickly gets unwieldy
829 * (too many defines). Also, I actually prefer to see when a variable
830 * is global, thus "G." prefix is a useful hint */
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000831#define INIT_G() do { \
832 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
Denys Vlasenko0806e402011-05-12 23:06:20 +0200833 /* memset(&G.sa, 0, sizeof(G.sa)); */ \
834 sigfillset(&G.sa.sa_mask); \
835 G.sa.sa_flags = SA_RESTART; \
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000836} while (0)
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000837
838
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000839/* Function prototypes for builtins */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200840static int builtin_cd(char **argv) FAST_FUNC;
841static int builtin_echo(char **argv) FAST_FUNC;
842static int builtin_eval(char **argv) FAST_FUNC;
843static int builtin_exec(char **argv) FAST_FUNC;
844static int builtin_exit(char **argv) FAST_FUNC;
845static int builtin_export(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000846#if ENABLE_HUSH_JOB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200847static int builtin_fg_bg(char **argv) FAST_FUNC;
848static int builtin_jobs(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000849#endif
850#if ENABLE_HUSH_HELP
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200851static int builtin_help(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000852#endif
maxwen27116ba2015-08-14 21:41:28 +0200853#if MAX_HISTORY && ENABLE_FEATURE_EDITING
854static int builtin_history(char **argv) FAST_FUNC;
855#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +0200856#if ENABLE_HUSH_LOCAL
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200857static int builtin_local(char **argv) FAST_FUNC;
Denys Vlasenko295fef82009-06-03 12:47:26 +0200858#endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000859#if HUSH_DEBUG
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200860static int builtin_memleak(char **argv) FAST_FUNC;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000861#endif
Mike Frysinger4ebc76c2009-10-15 03:32:39 -0400862#if ENABLE_PRINTF
863static int builtin_printf(char **argv) FAST_FUNC;
864#endif
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200865static int builtin_pwd(char **argv) FAST_FUNC;
866static int builtin_read(char **argv) FAST_FUNC;
867static int builtin_set(char **argv) FAST_FUNC;
868static int builtin_shift(char **argv) FAST_FUNC;
869static int builtin_source(char **argv) FAST_FUNC;
870static int builtin_test(char **argv) FAST_FUNC;
871static int builtin_trap(char **argv) FAST_FUNC;
872static int builtin_type(char **argv) FAST_FUNC;
873static int builtin_true(char **argv) FAST_FUNC;
874static int builtin_umask(char **argv) FAST_FUNC;
875static int builtin_unset(char **argv) FAST_FUNC;
876static int builtin_wait(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000877#if ENABLE_HUSH_LOOPS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200878static int builtin_break(char **argv) FAST_FUNC;
879static int builtin_continue(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000880#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000881#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200882static int builtin_return(char **argv) FAST_FUNC;
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000883#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000884
885/* Table of built-in functions. They can be forked or not, depending on
886 * context: within pipes, they fork. As simple commands, they do not.
887 * When used in non-forking context, they can change global variables
888 * in the parent shell process. If forked, of course they cannot.
889 * For example, 'unset foo | whatever' will parse and run, but foo will
890 * still be set at the end. */
891struct built_in_command {
Denys Vlasenko17323a62010-01-28 01:57:05 +0100892 const char *b_cmd;
893 int (*b_function)(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000894#if ENABLE_HUSH_HELP
Denys Vlasenko17323a62010-01-28 01:57:05 +0100895 const char *b_descr;
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200896# define BLTIN(cmd, func, help) { cmd, func, help }
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000897#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200898# define BLTIN(cmd, func, help) { cmd, func }
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000899#endif
900};
901
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200902static const struct built_in_command bltins1[] = {
903 BLTIN("." , builtin_source , "Run commands in a file"),
904 BLTIN(":" , builtin_true , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000905#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200906 BLTIN("bg" , builtin_fg_bg , "Resume a job in the background"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000907#endif
908#if ENABLE_HUSH_LOOPS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200909 BLTIN("break" , builtin_break , "Exit from a loop"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000910#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200911 BLTIN("cd" , builtin_cd , "Change directory"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000912#if ENABLE_HUSH_LOOPS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200913 BLTIN("continue" , builtin_continue, "Start new loop iteration"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000914#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200915 BLTIN("eval" , builtin_eval , "Construct and run shell command"),
916 BLTIN("exec" , builtin_exec , "Execute command, don't return to shell"),
917 BLTIN("exit" , builtin_exit , "Exit"),
918 BLTIN("export" , builtin_export , "Set environment variables"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000919#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200920 BLTIN("fg" , builtin_fg_bg , "Bring job into the foreground"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000921#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000922#if ENABLE_HUSH_HELP
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200923 BLTIN("help" , builtin_help , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000924#endif
maxwen27116ba2015-08-14 21:41:28 +0200925#if MAX_HISTORY && ENABLE_FEATURE_EDITING
926 BLTIN("history" , builtin_history , "Show command history"),
927#endif
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000928#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200929 BLTIN("jobs" , builtin_jobs , "List jobs"),
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000930#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +0200931#if ENABLE_HUSH_LOCAL
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200932 BLTIN("local" , builtin_local , "Set local variables"),
Denys Vlasenko295fef82009-06-03 12:47:26 +0200933#endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000934#if HUSH_DEBUG
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200935 BLTIN("memleak" , builtin_memleak , NULL),
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000936#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200937 BLTIN("read" , builtin_read , "Input into variable"),
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000938#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200939 BLTIN("return" , builtin_return , "Return from a function"),
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000940#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200941 BLTIN("set" , builtin_set , "Set/unset positional parameters"),
942 BLTIN("shift" , builtin_shift , "Shift positional parameters"),
Denys Vlasenko82731b42010-05-17 17:49:52 +0200943#if ENABLE_HUSH_BASH_COMPAT
944 BLTIN("source" , builtin_source , "Run commands in a file"),
945#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200946 BLTIN("trap" , builtin_trap , "Trap signals"),
Denys Vlasenko651a2692010-03-23 16:25:17 +0100947 BLTIN("type" , builtin_type , "Show command type"),
Denys Vlasenkof3c742f2010-03-06 20:12:00 +0100948 BLTIN("ulimit" , shell_builtin_ulimit , "Control resource limits"),
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200949 BLTIN("umask" , builtin_umask , "Set file creation mask"),
950 BLTIN("unset" , builtin_unset , "Unset variables"),
951 BLTIN("wait" , builtin_wait , "Wait for process"),
952};
953/* For now, echo and test are unconditionally enabled.
954 * Maybe make it configurable? */
955static const struct built_in_command bltins2[] = {
956 BLTIN("[" , builtin_test , NULL),
957 BLTIN("echo" , builtin_echo , NULL),
Mike Frysinger4ebc76c2009-10-15 03:32:39 -0400958#if ENABLE_PRINTF
959 BLTIN("printf" , builtin_printf , NULL),
960#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200961 BLTIN("pwd" , builtin_pwd , NULL),
962 BLTIN("test" , builtin_test , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000963};
964
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000965
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000966/* Debug printouts.
967 */
968#if HUSH_DEBUG
969/* prevent disasters with G.debug_indent < 0 */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100970# define indent() fdprintf(2, "%*s", (G.debug_indent * 2) & 0xff, "")
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000971# define debug_enter() (G.debug_indent++)
972# define debug_leave() (G.debug_indent--)
973#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200974# define indent() ((void)0)
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000975# define debug_enter() ((void)0)
976# define debug_leave() ((void)0)
977#endif
978
979#ifndef debug_printf
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100980# define debug_printf(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000981#endif
982
983#ifndef debug_printf_parse
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100984# define debug_printf_parse(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000985#endif
986
987#ifndef debug_printf_exec
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100988#define debug_printf_exec(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000989#endif
990
991#ifndef debug_printf_env
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100992# define debug_printf_env(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000993#endif
994
995#ifndef debug_printf_jobs
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100996# define debug_printf_jobs(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000997# define DEBUG_JOBS 1
998#else
999# define DEBUG_JOBS 0
1000#endif
1001
1002#ifndef debug_printf_expand
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001003# define debug_printf_expand(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001004# define DEBUG_EXPAND 1
1005#else
1006# define DEBUG_EXPAND 0
1007#endif
1008
Denys Vlasenko1e811b12010-05-22 03:12:29 +02001009#ifndef debug_printf_varexp
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001010# define debug_printf_varexp(...) (indent(), fdprintf(2, __VA_ARGS__))
Denys Vlasenko1e811b12010-05-22 03:12:29 +02001011#endif
1012
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001013#ifndef debug_printf_glob
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001014# define debug_printf_glob(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001015# define DEBUG_GLOB 1
1016#else
1017# define DEBUG_GLOB 0
1018#endif
1019
1020#ifndef debug_printf_list
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001021# define debug_printf_list(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001022#endif
1023
1024#ifndef debug_printf_subst
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001025# define debug_printf_subst(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001026#endif
1027
1028#ifndef debug_printf_clean
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001029# define debug_printf_clean(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001030# define DEBUG_CLEAN 1
1031#else
1032# define DEBUG_CLEAN 0
1033#endif
1034
1035#if DEBUG_EXPAND
1036static void debug_print_strings(const char *prefix, char **vv)
1037{
1038 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001039 fdprintf(2, "%s:\n", prefix);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001040 while (*vv)
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001041 fdprintf(2, " '%s'\n", *vv++);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001042}
1043#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001044# define debug_print_strings(prefix, vv) ((void)0)
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001045#endif
1046
1047
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001048/* Leak hunting. Use hush_leaktool.sh for post-processing.
1049 */
1050#if LEAK_HUNTING
1051static void *xxmalloc(int lineno, size_t size)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001052{
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001053 void *ptr = xmalloc((size + 0xff) & ~0xff);
1054 fdprintf(2, "line %d: malloc %p\n", lineno, ptr);
1055 return ptr;
1056}
1057static void *xxrealloc(int lineno, void *ptr, size_t size)
1058{
1059 ptr = xrealloc(ptr, (size + 0xff) & ~0xff);
1060 fdprintf(2, "line %d: realloc %p\n", lineno, ptr);
1061 return ptr;
1062}
1063static char *xxstrdup(int lineno, const char *str)
1064{
1065 char *ptr = xstrdup(str);
1066 fdprintf(2, "line %d: strdup %p\n", lineno, ptr);
1067 return ptr;
1068}
1069static void xxfree(void *ptr)
1070{
1071 fdprintf(2, "free %p\n", ptr);
1072 free(ptr);
1073}
Denys Vlasenko8391c482010-05-22 17:50:43 +02001074# define xmalloc(s) xxmalloc(__LINE__, s)
1075# define xrealloc(p, s) xxrealloc(__LINE__, p, s)
1076# define xstrdup(s) xxstrdup(__LINE__, s)
1077# define free(p) xxfree(p)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001078#endif
1079
1080
1081/* Syntax and runtime errors. They always abort scripts.
1082 * In interactive use they usually discard unparsed and/or unexecuted commands
1083 * and return to the prompt.
1084 * HUSH_DEBUG >= 2 prints line number in this file where it was detected.
1085 */
1086#if HUSH_DEBUG < 2
Denys Vlasenko606291b2009-09-23 23:15:43 +02001087# define die_if_script(lineno, ...) die_if_script(__VA_ARGS__)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001088# define syntax_error(lineno, msg) syntax_error(msg)
1089# define syntax_error_at(lineno, msg) syntax_error_at(msg)
1090# define syntax_error_unterm_ch(lineno, ch) syntax_error_unterm_ch(ch)
1091# define syntax_error_unterm_str(lineno, s) syntax_error_unterm_str(s)
1092# define syntax_error_unexpected_ch(lineno, ch) syntax_error_unexpected_ch(ch)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001093#endif
1094
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001095static void die_if_script(unsigned lineno, const char *fmt, ...)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001096{
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001097 va_list p;
1098
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001099#if HUSH_DEBUG >= 2
1100 bb_error_msg("hush.c:%u", lineno);
1101#endif
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001102 va_start(p, fmt);
1103 bb_verror_msg(fmt, p, NULL);
1104 va_end(p);
1105 if (!G_interactive_fd)
1106 xfunc_die();
Mike Frysinger6379bb42009-03-28 18:55:03 +00001107}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001108
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001109static void syntax_error(unsigned lineno UNUSED_PARAM, const char *msg)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001110{
1111 if (msg)
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001112 bb_error_msg("syntax error: %s", msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001113 else
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001114 bb_error_msg("syntax error");
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001115}
1116
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001117static void syntax_error_at(unsigned lineno UNUSED_PARAM, const char *msg)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001118{
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001119 bb_error_msg("syntax error at '%s'", msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001120}
1121
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001122static void syntax_error_unterm_str(unsigned lineno UNUSED_PARAM, const char *s)
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001123{
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001124 bb_error_msg("syntax error: unterminated %s", s);
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001125}
1126
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001127static void syntax_error_unterm_ch(unsigned lineno, char ch)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001128{
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001129 char msg[2] = { ch, '\0' };
1130 syntax_error_unterm_str(lineno, msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001131}
1132
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001133static void syntax_error_unexpected_ch(unsigned lineno UNUSED_PARAM, int ch)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001134{
1135 char msg[2];
1136 msg[0] = ch;
1137 msg[1] = '\0';
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001138 bb_error_msg("syntax error: unexpected %s", ch == EOF ? "EOF" : msg);
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001139}
1140
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001141#if HUSH_DEBUG < 2
1142# undef die_if_script
1143# undef syntax_error
1144# undef syntax_error_at
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001145# undef syntax_error_unterm_ch
1146# undef syntax_error_unterm_str
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001147# undef syntax_error_unexpected_ch
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001148#else
Denys Vlasenko606291b2009-09-23 23:15:43 +02001149# define die_if_script(...) die_if_script(__LINE__, __VA_ARGS__)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001150# define syntax_error(msg) syntax_error(__LINE__, msg)
1151# define syntax_error_at(msg) syntax_error_at(__LINE__, msg)
1152# define syntax_error_unterm_ch(ch) syntax_error_unterm_ch(__LINE__, ch)
1153# define syntax_error_unterm_str(s) syntax_error_unterm_str(__LINE__, s)
1154# define syntax_error_unexpected_ch(ch) syntax_error_unexpected_ch(__LINE__, ch)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001155#endif
Eric Andersen25f27032001-04-26 23:22:31 +00001156
Denis Vlasenko552433b2009-04-04 19:29:21 +00001157
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001158#if ENABLE_HUSH_INTERACTIVE
1159static void cmdedit_update_prompt(void);
1160#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001161# define cmdedit_update_prompt() ((void)0)
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001162#endif
1163
1164
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001165/* Utility functions
1166 */
Denis Vlasenko55789c62008-06-18 16:30:42 +00001167/* Replace each \x with x in place, return ptr past NUL. */
1168static char *unbackslash(char *src)
1169{
Denys Vlasenko71885402009-09-24 01:44:13 +02001170 char *dst = src = strchrnul(src, '\\');
Denis Vlasenko55789c62008-06-18 16:30:42 +00001171 while (1) {
1172 if (*src == '\\')
1173 src++;
1174 if ((*dst++ = *src++) == '\0')
1175 break;
1176 }
1177 return dst;
1178}
1179
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001180static char **add_strings_to_strings(char **strings, char **add, int need_to_dup)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001181{
1182 int i;
1183 unsigned count1;
1184 unsigned count2;
1185 char **v;
1186
1187 v = strings;
1188 count1 = 0;
1189 if (v) {
1190 while (*v) {
1191 count1++;
1192 v++;
1193 }
1194 }
1195 count2 = 0;
1196 v = add;
1197 while (*v) {
1198 count2++;
1199 v++;
1200 }
1201 v = xrealloc(strings, (count1 + count2 + 1) * sizeof(char*));
1202 v[count1 + count2] = NULL;
1203 i = count2;
1204 while (--i >= 0)
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001205 v[count1 + i] = (need_to_dup ? xstrdup(add[i]) : add[i]);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001206 return v;
1207}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001208#if LEAK_HUNTING
Denis Vlasenkocc90f442009-04-08 16:40:34 +00001209static char **xx_add_strings_to_strings(int lineno, char **strings, char **add, int need_to_dup)
1210{
1211 char **ptr = add_strings_to_strings(strings, add, need_to_dup);
1212 fdprintf(2, "line %d: add_strings_to_strings %p\n", lineno, ptr);
1213 return ptr;
1214}
1215#define add_strings_to_strings(strings, add, need_to_dup) \
1216 xx_add_strings_to_strings(__LINE__, strings, add, need_to_dup)
1217#endif
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001218
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001219/* Note: takes ownership of "add" ptr (it is not strdup'ed) */
Denis Vlasenko22d10a02008-10-13 08:53:43 +00001220static char **add_string_to_strings(char **strings, char *add)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001221{
1222 char *v[2];
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001223 v[0] = add;
1224 v[1] = NULL;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001225 return add_strings_to_strings(strings, v, /*dup:*/ 0);
Denis Vlasenko22d10a02008-10-13 08:53:43 +00001226}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001227#if LEAK_HUNTING
Denis Vlasenkocc90f442009-04-08 16:40:34 +00001228static char **xx_add_string_to_strings(int lineno, char **strings, char *add)
1229{
1230 char **ptr = add_string_to_strings(strings, add);
1231 fdprintf(2, "line %d: add_string_to_strings %p\n", lineno, ptr);
1232 return ptr;
1233}
1234#define add_string_to_strings(strings, add) \
1235 xx_add_string_to_strings(__LINE__, strings, add)
1236#endif
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001237
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001238static void free_strings(char **strings)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001239{
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001240 char **v;
1241
1242 if (!strings)
1243 return;
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001244 v = strings;
1245 while (*v) {
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001246 free(*v);
1247 v++;
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001248 }
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001249 free(strings);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001250}
1251
Denis Vlasenko76d50412008-06-10 16:19:39 +00001252
Denis Vlasenko270b1c32009-04-17 18:54:50 +00001253/* Helpers for setting new $n and restoring them back
1254 */
1255typedef struct save_arg_t {
1256 char *sv_argv0;
1257 char **sv_g_argv;
1258 int sv_g_argc;
1259 smallint sv_g_malloced;
1260} save_arg_t;
1261
1262static void save_and_replace_G_args(save_arg_t *sv, char **argv)
1263{
1264 int n;
1265
1266 sv->sv_argv0 = argv[0];
1267 sv->sv_g_argv = G.global_argv;
1268 sv->sv_g_argc = G.global_argc;
1269 sv->sv_g_malloced = G.global_args_malloced;
1270
1271 argv[0] = G.global_argv[0]; /* retain $0 */
1272 G.global_argv = argv;
1273 G.global_args_malloced = 0;
1274
1275 n = 1;
1276 while (*++argv)
1277 n++;
1278 G.global_argc = n;
1279}
1280
1281static void restore_G_args(save_arg_t *sv, char **argv)
1282{
1283 char **pp;
1284
1285 if (G.global_args_malloced) {
1286 /* someone ran "set -- arg1 arg2 ...", undo */
1287 pp = G.global_argv;
1288 while (*++pp) /* note: does not free $0 */
1289 free(*pp);
1290 free(G.global_argv);
1291 }
1292 argv[0] = sv->sv_argv0;
1293 G.global_argv = sv->sv_g_argv;
1294 G.global_argc = sv->sv_g_argc;
1295 G.global_args_malloced = sv->sv_g_malloced;
1296}
1297
1298
Denis Vlasenkod5762932009-03-31 11:22:57 +00001299/* Basic theory of signal handling in shell
1300 * ========================================
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001301 * This does not describe what hush does, rather, it is current understanding
1302 * what it _should_ do. If it doesn't, it's a bug.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001303 * http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#trap
1304 *
1305 * Signals are handled only after each pipe ("cmd | cmd | cmd" thing)
1306 * is finished or backgrounded. It is the same in interactive and
1307 * non-interactive shells, and is the same regardless of whether
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001308 * a user trap handler is installed or a shell special one is in effect.
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02001309 * ^C or ^Z from keyboard seems to execute "at once" because it usually
Denis Vlasenkod5762932009-03-31 11:22:57 +00001310 * backgrounds (i.e. stops) or kills all members of currently running
1311 * pipe.
1312 *
maxwen27116ba2015-08-14 21:41:28 +02001313 * Wait builtin is interruptible by signals for which user trap is set
Denis Vlasenkod5762932009-03-31 11:22:57 +00001314 * or by SIGINT in interactive shell.
1315 *
1316 * Trap handlers will execute even within trap handlers. (right?)
1317 *
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001318 * User trap handlers are forgotten when subshell ("(cmd)") is entered,
1319 * except for handlers set to '' (empty string).
Denis Vlasenkod5762932009-03-31 11:22:57 +00001320 *
1321 * If job control is off, backgrounded commands ("cmd &")
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001322 * have SIGINT, SIGQUIT set to SIG_IGN.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001323 *
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001324 * Commands which are run in command substitution ("`cmd`")
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001325 * have SIGTTIN, SIGTTOU, SIGTSTP set to SIG_IGN.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001326 *
Denys Vlasenko4b7db4f2009-05-29 10:39:06 +02001327 * Ordinary commands have signals set to SIG_IGN/DFL as inherited
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001328 * by the shell from its parent.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001329 *
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001330 * Signals which differ from SIG_DFL action
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001331 * (note: child (i.e., [v]forked) shell is not an interactive shell):
Denis Vlasenkod5762932009-03-31 11:22:57 +00001332 *
1333 * SIGQUIT: ignore
1334 * SIGTERM (interactive): ignore
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001335 * SIGHUP (interactive):
1336 * send SIGCONT to stopped jobs, send SIGHUP to all jobs and exit
Denis Vlasenkod5762932009-03-31 11:22:57 +00001337 * SIGTTIN, SIGTTOU, SIGTSTP (if job control is on): ignore
Denis Vlasenkoc4ada792009-04-15 23:29:00 +00001338 * Note that ^Z is handled not by trapping SIGTSTP, but by seeing
1339 * that all pipe members are stopped. Try this in bash:
1340 * while :; do :; done - ^Z does not background it
1341 * (while :; do :; done) - ^Z backgrounds it
Denis Vlasenkod5762932009-03-31 11:22:57 +00001342 * SIGINT (interactive): wait for last pipe, ignore the rest
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001343 * of the command line, show prompt. NB: ^C does not send SIGINT
1344 * to interactive shell while shell is waiting for a pipe,
1345 * since shell is bg'ed (is not in foreground process group).
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001346 * Example 1: this waits 5 sec, but does not execute ls:
1347 * "echo $$; sleep 5; ls -l" + "kill -INT <pid>"
1348 * Example 2: this does not wait and does not execute ls:
1349 * "echo $$; sleep 5 & wait; ls -l" + "kill -INT <pid>"
1350 * Example 3: this does not wait 5 sec, but executes ls:
1351 * "sleep 5; ls -l" + press ^C
Denys Vlasenkob8709032011-05-08 21:20:01 +02001352 * Example 4: this does not wait and does not execute ls:
1353 * "sleep 5 & wait; ls -l" + press ^C
Denis Vlasenkod5762932009-03-31 11:22:57 +00001354 *
1355 * (What happens to signals which are IGN on shell start?)
1356 * (What happens with signal mask on shell start?)
1357 *
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001358 * Old implementation
1359 * ==================
Denis Vlasenkod5762932009-03-31 11:22:57 +00001360 * We use in-kernel pending signal mask to determine which signals were sent.
1361 * We block all signals which we don't want to take action immediately,
1362 * i.e. we block all signals which need to have special handling as described
1363 * above, and all signals which have traps set.
1364 * After each pipe execution, we extract any pending signals via sigtimedwait()
1365 * and act on them.
1366 *
Denys Vlasenko10c01312011-05-11 11:49:21 +02001367 * unsigned special_sig_mask: a mask of such "special" signals
Denis Vlasenkod5762932009-03-31 11:22:57 +00001368 * sigset_t blocked_set: current blocked signal set
1369 *
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001370 * "trap - SIGxxx":
Denys Vlasenko10c01312011-05-11 11:49:21 +02001371 * clear bit in blocked_set unless it is also in special_sig_mask
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001372 * "trap 'cmd' SIGxxx":
1373 * set bit in blocked_set (even if 'cmd' is '')
Denis Vlasenkod5762932009-03-31 11:22:57 +00001374 * after [v]fork, if we plan to be a shell:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001375 * unblock signals with special interactive handling
1376 * (child shell is not interactive),
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001377 * unset all traps except '' (note: regardless of child shell's type - {}, (), etc)
Denis Vlasenkod5762932009-03-31 11:22:57 +00001378 * after [v]fork, if we plan to exec:
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02001379 * POSIX says fork clears pending signal mask in child - no need to clear it.
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001380 * Restore blocked signal set to one inherited by shell just prior to exec.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001381 *
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001382 * Note: as a result, we do not use signal handlers much. The only uses
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001383 * are to count SIGCHLDs
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001384 * and to restore tty pgrp on signal-induced exit.
Denys Vlasenko4ea0ca82009-09-25 12:58:37 +02001385 *
Denys Vlasenko67f71862009-09-25 14:21:06 +02001386 * Note 2 (compat):
Denys Vlasenko4ea0ca82009-09-25 12:58:37 +02001387 * Standard says "When a subshell is entered, traps that are not being ignored
1388 * are set to the default actions". bash interprets it so that traps which
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001389 * are set to '' (ignore) are NOT reset to defaults. We do the same.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001390 *
1391 * Problem: the above approach makes it unwieldy to catch signals while
maxwen27116ba2015-08-14 21:41:28 +02001392 * we are in read builtin, or while we read commands from stdin:
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001393 * masked signals are not visible!
1394 *
1395 * New implementation
1396 * ==================
1397 * We record each signal we are interested in by installing signal handler
1398 * for them - a bit like emulating kernel pending signal mask in userspace.
1399 * We are interested in: signals which need to have special handling
1400 * as described above, and all signals which have traps set.
maxwen27116ba2015-08-14 21:41:28 +02001401 * Signals are recorded in pending_set.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001402 * After each pipe execution, we extract any pending signals
1403 * and act on them.
1404 *
1405 * unsigned special_sig_mask: a mask of shell-special signals.
1406 * unsigned fatal_sig_mask: a mask of signals on which we restore tty pgrp.
1407 * char *traps[sig] if trap for sig is set (even if it's '').
1408 * sigset_t pending_set: set of sigs we received.
1409 *
1410 * "trap - SIGxxx":
1411 * if sig is in special_sig_mask, set handler back to:
1412 * record_pending_signo, or to IGN if it's a tty stop signal
1413 * if sig is in fatal_sig_mask, set handler back to sigexit.
1414 * else: set handler back to SIG_DFL
1415 * "trap 'cmd' SIGxxx":
1416 * set handler to record_pending_signo.
1417 * "trap '' SIGxxx":
1418 * set handler to SIG_IGN.
1419 * after [v]fork, if we plan to be a shell:
1420 * set signals with special interactive handling to SIG_DFL
1421 * (because child shell is not interactive),
1422 * unset all traps except '' (note: regardless of child shell's type - {}, (), etc)
1423 * after [v]fork, if we plan to exec:
1424 * POSIX says fork clears pending signal mask in child - no need to clear it.
1425 *
1426 * To make wait builtin interruptible, we handle SIGCHLD as special signal,
1427 * otherwise (if we leave it SIG_DFL) sigsuspend in wait builtin will not wake up on it.
1428 *
1429 * Note (compat):
1430 * Standard says "When a subshell is entered, traps that are not being ignored
1431 * are set to the default actions". bash interprets it so that traps which
1432 * are set to '' (ignore) are NOT reset to defaults. We do the same.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001433 */
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001434enum {
1435 SPECIAL_INTERACTIVE_SIGS = 0
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001436 | (1 << SIGTERM)
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001437 | (1 << SIGINT)
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00001438 | (1 << SIGHUP)
1439 ,
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001440 SPECIAL_JOBSTOP_SIGS = 0
Mike Frysinger38478a62009-05-20 04:48:06 -04001441#if ENABLE_HUSH_JOB
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00001442 | (1 << SIGTTIN)
1443 | (1 << SIGTTOU)
1444 | (1 << SIGTSTP)
1445#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001446 ,
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001447};
Denis Vlasenkod5762932009-03-31 11:22:57 +00001448
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001449static void record_pending_signo(int sig)
Denys Vlasenko54e9e122011-05-09 00:52:15 +02001450{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001451 sigaddset(&G.pending_set, sig);
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001452#if ENABLE_HUSH_FAST
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001453 if (sig == SIGCHLD) {
1454 G.count_SIGCHLD++;
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001455//bb_error_msg("[%d] SIGCHLD_handler: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001456 }
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001457#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001458}
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001459
Denys Vlasenko0806e402011-05-12 23:06:20 +02001460static sighandler_t install_sighandler(int sig, sighandler_t handler)
1461{
1462 struct sigaction old_sa;
1463
1464 /* We could use signal() to install handlers... almost:
1465 * except that we need to mask ALL signals while handlers run.
1466 * I saw signal nesting in strace, race window isn't small.
1467 * SA_RESTART is also needed, but in Linux, signal()
1468 * sets SA_RESTART too.
1469 */
1470 /* memset(&G.sa, 0, sizeof(G.sa)); - already done */
1471 /* sigfillset(&G.sa.sa_mask); - already done */
1472 /* G.sa.sa_flags = SA_RESTART; - already done */
1473 G.sa.sa_handler = handler;
1474 sigaction(sig, &G.sa, &old_sa);
1475 return old_sa.sa_handler;
1476}
1477
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00001478#if ENABLE_HUSH_JOB
Denis Vlasenko25af86f2009-04-07 13:29:27 +00001479
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001480/* After [v]fork, in child: do not restore tty pgrp on xfunc death */
Denys Vlasenko8391c482010-05-22 17:50:43 +02001481# define disable_restore_tty_pgrp_on_exit() (die_sleep = 0)
Denis Vlasenko25af86f2009-04-07 13:29:27 +00001482/* After [v]fork, in parent: restore tty pgrp on xfunc death */
Denys Vlasenko8391c482010-05-22 17:50:43 +02001483# define enable_restore_tty_pgrp_on_exit() (die_sleep = -1)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001484
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001485/* Restores tty foreground process group, and exits.
1486 * May be called as signal handler for fatal signal
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001487 * (will resend signal to itself, producing correct exit state)
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001488 * or called directly with -EXITCODE.
1489 * We also call it if xfunc is exiting. */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +00001490static void sigexit(int sig) NORETURN;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001491static void sigexit(int sig)
1492{
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001493 /* Careful: we can end up here after [v]fork. Do not restore
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001494 * tty pgrp then, only top-level shell process does that */
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001495 if (G_saved_tty_pgrp && getpid() == G.root_pid) {
1496 /* Disable all signals: job control, SIGPIPE, etc.
1497 * Mostly paranoid measure, to prevent infinite SIGTTOU.
1498 */
1499 sigprocmask_allsigs(SIG_BLOCK);
Mike Frysinger38478a62009-05-20 04:48:06 -04001500 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001501 }
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001502
1503 /* Not a signal, just exit */
1504 if (sig <= 0)
1505 _exit(- sig);
1506
Denis Vlasenko400d8bb2008-02-24 13:36:01 +00001507 kill_myself_with_sig(sig); /* does not return */
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001508}
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001509#else
1510
Denys Vlasenko8391c482010-05-22 17:50:43 +02001511# define disable_restore_tty_pgrp_on_exit() ((void)0)
1512# define enable_restore_tty_pgrp_on_exit() ((void)0)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001513
Denis Vlasenkoe0755e52009-04-03 21:16:45 +00001514#endif
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001515
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001516static sighandler_t pick_sighandler(unsigned sig)
1517{
1518 sighandler_t handler = SIG_DFL;
1519 if (sig < sizeof(unsigned)*8) {
1520 unsigned sigmask = (1 << sig);
1521
1522#if ENABLE_HUSH_JOB
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001523 /* is sig fatal? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001524 if (G_fatal_sig_mask & sigmask)
1525 handler = sigexit;
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001526 else
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001527#endif
1528 /* sig has special handling? */
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001529 if (G.special_sig_mask & sigmask) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001530 handler = record_pending_signo;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001531 /* TTIN/TTOU/TSTP can't be set to record_pending_signo
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001532 * in order to ignore them: they will be raised
Denys Vlasenkof58f7052011-05-12 02:10:33 +02001533 * in an endless loop when we try to do some
1534 * terminal ioctls! We do have to _ignore_ these.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001535 */
1536 if (SPECIAL_JOBSTOP_SIGS & sigmask)
1537 handler = SIG_IGN;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001538 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001539 }
1540 return handler;
1541}
1542
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001543/* Restores tty foreground process group, and exits. */
1544static void hush_exit(int exitcode) NORETURN;
1545static void hush_exit(int exitcode)
1546{
Tanguy Pruvot8a6c2c22012-04-28 00:24:09 +02001547#if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
1548 save_history(G.line_input_state);
1549#endif
1550
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01001551 fflush_all();
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00001552 if (G.exiting <= 0 && G.traps && G.traps[0] && G.traps[0][0]) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001553 char *argv[3];
1554 /* argv[0] is unused */
1555 argv[1] = G.traps[0];
1556 argv[2] = NULL;
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001557 G.exiting = 1; /* prevent EXIT trap recursion */
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001558 /* Note: G.traps[0] is not cleared!
Denys Vlasenkode8c3f62010-09-12 16:13:44 +02001559 * "trap" will still show it, if executed
1560 * in the handler */
1561 builtin_eval(argv);
Denis Vlasenkod5762932009-03-31 11:22:57 +00001562 }
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001563
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001564#if ENABLE_FEATURE_CLEAN_UP
1565 {
1566 struct variable *cur_var;
1567 if (G.cwd != bb_msg_unknown)
1568 free((char*)G.cwd);
1569 cur_var = G.top_var;
1570 while (cur_var) {
1571 struct variable *tmp = cur_var;
1572 if (!cur_var->max_len)
1573 free(cur_var->varstr);
1574 cur_var = cur_var->next;
1575 free(tmp);
1576 }
1577 }
1578#endif
1579
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001580#if ENABLE_HUSH_JOB
Denys Vlasenko8131eea2009-11-02 14:19:51 +01001581 fflush_all();
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001582 sigexit(- (exitcode & 0xff));
1583#else
1584 exit(exitcode);
1585#endif
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001586}
1587
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02001588
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001589//TODO: return a mask of ALL handled sigs?
1590static int check_and_run_traps(void)
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001591{
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001592 int last_sig = 0;
1593
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001594 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001595 int sig;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02001596
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001597 if (sigisemptyset(&G.pending_set))
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001598 break;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001599 sig = 0;
1600 do {
1601 sig++;
1602 if (sigismember(&G.pending_set, sig)) {
1603 sigdelset(&G.pending_set, sig);
1604 goto got_sig;
1605 }
1606 } while (sig < NSIG);
1607 break;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001608 got_sig:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001609 if (G.traps && G.traps[sig]) {
1610 if (G.traps[sig][0]) {
1611 /* We have user-defined handler */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001612 smalluint save_rcode;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001613 char *argv[3];
1614 /* argv[0] is unused */
1615 argv[1] = G.traps[sig];
1616 argv[2] = NULL;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001617 save_rcode = G.last_exitcode;
1618 builtin_eval(argv);
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001619 G.last_exitcode = save_rcode;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001620 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001621 } /* else: "" trap, ignoring signal */
1622 continue;
1623 }
1624 /* not a trap: special action */
1625 switch (sig) {
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001626 case SIGINT:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001627 /* Builtin was ^C'ed, make it look prettier: */
1628 bb_putchar('\n');
1629 G.flag_SIGINT = 1;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001630 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001631 break;
1632#if ENABLE_HUSH_JOB
1633 case SIGHUP: {
1634 struct pipe *job;
1635 /* bash is observed to signal whole process groups,
1636 * not individual processes */
1637 for (job = G.job_list; job; job = job->next) {
1638 if (job->pgrp <= 0)
1639 continue;
1640 debug_printf_exec("HUPing pgrp %d\n", job->pgrp);
1641 if (kill(- job->pgrp, SIGHUP) == 0)
1642 kill(- job->pgrp, SIGCONT);
1643 }
1644 sigexit(SIGHUP);
1645 }
1646#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001647#if ENABLE_HUSH_FAST
1648 case SIGCHLD:
1649 G.count_SIGCHLD++;
1650//bb_error_msg("[%d] check_and_run_traps: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
1651 /* Note:
1652 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1653 * This simplifies wait builtin a bit.
1654 */
1655 break;
1656#endif
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001657 default: /* ignored: */
1658 /* SIGTERM, SIGQUIT, SIGTTIN, SIGTTOU, SIGTSTP */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001659 /* Note:
1660 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1661 * Example: wait is not interrupted by TERM
Denys Vlasenkob8709032011-05-08 21:20:01 +02001662 * in interactive shell, because TERM is ignored.
1663 */
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001664 break;
1665 }
1666 }
1667 return last_sig;
1668}
1669
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001670
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001671static const char *get_cwd(int force)
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001672{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001673 if (force || G.cwd == NULL) {
1674 /* xrealloc_getcwd_or_warn(arg) calls free(arg),
1675 * we must not try to free(bb_msg_unknown) */
1676 if (G.cwd == bb_msg_unknown)
1677 G.cwd = NULL;
1678 G.cwd = xrealloc_getcwd_or_warn((char *)G.cwd);
1679 if (!G.cwd)
1680 G.cwd = bb_msg_unknown;
1681 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00001682 return G.cwd;
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001683}
1684
Denis Vlasenko83506862007-11-23 13:11:42 +00001685
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001686/*
1687 * Shell and environment variable support
1688 */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001689static struct variable **get_ptr_to_local_var(const char *name, unsigned len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001690{
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001691 struct variable **pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001692 struct variable *cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001693
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001694 pp = &G.top_var;
1695 while ((cur = *pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001696 if (strncmp(cur->varstr, name, len) == 0 && cur->varstr[len] == '=')
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001697 return pp;
1698 pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001699 }
1700 return NULL;
1701}
1702
Denys Vlasenko03dad222010-01-12 23:29:57 +01001703static const char* FAST_FUNC get_local_var_value(const char *name)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001704{
Denys Vlasenko29082232010-07-16 13:52:32 +02001705 struct variable **vpp;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001706 unsigned len = strlen(name);
Denys Vlasenko29082232010-07-16 13:52:32 +02001707
1708 if (G.expanded_assignments) {
1709 char **cpp = G.expanded_assignments;
Denys Vlasenko29082232010-07-16 13:52:32 +02001710 while (*cpp) {
1711 char *cp = *cpp;
1712 if (strncmp(cp, name, len) == 0 && cp[len] == '=')
1713 return cp + len + 1;
1714 cpp++;
1715 }
1716 }
1717
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001718 vpp = get_ptr_to_local_var(name, len);
Denys Vlasenko29082232010-07-16 13:52:32 +02001719 if (vpp)
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001720 return (*vpp)->varstr + len + 1;
Denys Vlasenko29082232010-07-16 13:52:32 +02001721
Denys Vlasenkodea47882009-10-09 15:40:49 +02001722 if (strcmp(name, "PPID") == 0)
1723 return utoa(G.root_ppid);
1724 // bash compat: UID? EUID?
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001725#if ENABLE_HUSH_RANDOM_SUPPORT
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001726 if (strcmp(name, "RANDOM") == 0)
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001727 return utoa(next_random(&G.random_gen));
1728#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001729 return NULL;
1730}
1731
1732/* str holds "NAME=VAL" and is expected to be malloced.
Mike Frysinger6379bb42009-03-28 18:55:03 +00001733 * We take ownership of it.
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001734 * flg_export:
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001735 * 0: do not change export flag
1736 * (if creating new variable, flag will be 0)
1737 * 1: set export flag and putenv the variable
1738 * -1: clear export flag and unsetenv the variable
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001739 * flg_read_only is set only when we handle -R var=val
Mike Frysinger6379bb42009-03-28 18:55:03 +00001740 */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001741#if !BB_MMU && ENABLE_HUSH_LOCAL
1742/* all params are used */
1743#elif BB_MMU && ENABLE_HUSH_LOCAL
1744#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1745 set_local_var(str, flg_export, local_lvl)
1746#elif BB_MMU && !ENABLE_HUSH_LOCAL
1747#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001748 set_local_var(str, flg_export)
Denys Vlasenko295fef82009-06-03 12:47:26 +02001749#elif !BB_MMU && !ENABLE_HUSH_LOCAL
1750#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1751 set_local_var(str, flg_export, flg_read_only)
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001752#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +02001753static int set_local_var(char *str, int flg_export, int local_lvl, int flg_read_only)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001754{
Denys Vlasenko295fef82009-06-03 12:47:26 +02001755 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001756 struct variable *cur;
Denis Vlasenko950bd722009-04-21 11:23:56 +00001757 char *eq_sign;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001758 int name_len;
1759
Denis Vlasenko950bd722009-04-21 11:23:56 +00001760 eq_sign = strchr(str, '=');
1761 if (!eq_sign) { /* not expected to ever happen? */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001762 free(str);
1763 return -1;
1764 }
1765
Denis Vlasenko950bd722009-04-21 11:23:56 +00001766 name_len = eq_sign - str + 1; /* including '=' */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001767 var_pp = &G.top_var;
1768 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001769 if (strncmp(cur->varstr, str, name_len) != 0) {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001770 var_pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001771 continue;
1772 }
1773 /* We found an existing var with this name */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001774 if (cur->flg_read_only) {
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001775#if !BB_MMU
1776 if (!flg_read_only)
1777#endif
1778 bb_error_msg("%s: readonly variable", str);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001779 free(str);
1780 return -1;
1781 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001782 if (flg_export == -1) { // "&& cur->flg_export" ?
Denis Vlasenko950bd722009-04-21 11:23:56 +00001783 debug_printf_env("%s: unsetenv '%s'\n", __func__, str);
1784 *eq_sign = '\0';
1785 unsetenv(str);
1786 *eq_sign = '=';
1787 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001788#if ENABLE_HUSH_LOCAL
1789 if (cur->func_nest_level < local_lvl) {
1790 /* New variable is declared as local,
1791 * and existing one is global, or local
1792 * from enclosing function.
1793 * Remove and save old one: */
1794 *var_pp = cur->next;
1795 cur->next = *G.shadowed_vars_pp;
1796 *G.shadowed_vars_pp = cur;
1797 /* bash 3.2.33(1) and exported vars:
1798 * # export z=z
1799 * # f() { local z=a; env | grep ^z; }
1800 * # f
1801 * z=a
1802 * # env | grep ^z
1803 * z=z
1804 */
1805 if (cur->flg_export)
1806 flg_export = 1;
1807 break;
1808 }
1809#endif
Denis Vlasenko950bd722009-04-21 11:23:56 +00001810 if (strcmp(cur->varstr + name_len, eq_sign + 1) == 0) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001811 free_and_exp:
1812 free(str);
1813 goto exp;
1814 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001815 if (cur->max_len != 0) {
1816 if (cur->max_len >= strlen(str)) {
1817 /* This one is from startup env, reuse space */
1818 strcpy(cur->varstr, str);
1819 goto free_and_exp;
1820 }
1821 } else {
1822 /* max_len == 0 signifies "malloced" var, which we can
1823 * (and has to) free */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001824 free(cur->varstr);
Denys Vlasenko295fef82009-06-03 12:47:26 +02001825 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001826 cur->max_len = 0;
1827 goto set_str_and_exp;
1828 }
1829
Denys Vlasenko295fef82009-06-03 12:47:26 +02001830 /* Not found - create new variable struct */
1831 cur = xzalloc(sizeof(*cur));
1832#if ENABLE_HUSH_LOCAL
1833 cur->func_nest_level = local_lvl;
1834#endif
1835 cur->next = *var_pp;
1836 *var_pp = cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001837
1838 set_str_and_exp:
1839 cur->varstr = str;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001840#if !BB_MMU
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001841 cur->flg_read_only = flg_read_only;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001842#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001843 exp:
Mike Frysinger6379bb42009-03-28 18:55:03 +00001844 if (flg_export == 1)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001845 cur->flg_export = 1;
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001846 if (name_len == 4 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1847 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001848 if (cur->flg_export) {
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001849 if (flg_export == -1) {
1850 cur->flg_export = 0;
1851 /* unsetenv was already done */
1852 } else {
1853 debug_printf_env("%s: putenv '%s'\n", __func__, cur->varstr);
1854 return putenv(cur->varstr);
1855 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001856 }
1857 return 0;
1858}
1859
Denys Vlasenko6db47842009-09-05 20:15:17 +02001860/* Used at startup and after each cd */
1861static void set_pwd_var(int exp)
1862{
1863 set_local_var(xasprintf("PWD=%s", get_cwd(/*force:*/ 1)),
1864 /*exp:*/ exp, /*lvl:*/ 0, /*ro:*/ 0);
1865}
1866
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001867static int unset_local_var_len(const char *name, int name_len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001868{
1869 struct variable *cur;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001870 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001871
1872 if (!name)
Mike Frysingerd690f682009-03-30 06:50:54 +00001873 return EXIT_SUCCESS;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001874 var_pp = &G.top_var;
1875 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001876 if (strncmp(cur->varstr, name, name_len) == 0 && cur->varstr[name_len] == '=') {
1877 if (cur->flg_read_only) {
1878 bb_error_msg("%s: readonly variable", name);
Mike Frysingerd690f682009-03-30 06:50:54 +00001879 return EXIT_FAILURE;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001880 }
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001881 *var_pp = cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001882 debug_printf_env("%s: unsetenv '%s'\n", __func__, cur->varstr);
1883 bb_unsetenv(cur->varstr);
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001884 if (name_len == 3 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1885 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001886 if (!cur->max_len)
1887 free(cur->varstr);
1888 free(cur);
Mike Frysingerd690f682009-03-30 06:50:54 +00001889 return EXIT_SUCCESS;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001890 }
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001891 var_pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001892 }
Mike Frysingerd690f682009-03-30 06:50:54 +00001893 return EXIT_SUCCESS;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001894}
1895
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001896static int unset_local_var(const char *name)
1897{
1898 return unset_local_var_len(name, strlen(name));
1899}
1900
1901static void unset_vars(char **strings)
1902{
1903 char **v;
1904
1905 if (!strings)
1906 return;
1907 v = strings;
1908 while (*v) {
1909 const char *eq = strchrnul(*v, '=');
1910 unset_local_var_len(*v, (int)(eq - *v));
1911 v++;
1912 }
1913 free(strings);
1914}
1915
Denys Vlasenko03dad222010-01-12 23:29:57 +01001916static void FAST_FUNC set_local_var_from_halves(const char *name, const char *val)
Mike Frysinger98c52642009-04-02 10:02:37 +00001917{
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001918 char *var = xasprintf("%s=%s", name, val);
Denys Vlasenko03dad222010-01-12 23:29:57 +01001919 set_local_var(var, /*flags:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
Mike Frysinger98c52642009-04-02 10:02:37 +00001920}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001921
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001922
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001923/*
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001924 * Helpers for "var1=val1 var2=val2 cmd" feature
1925 */
1926static void add_vars(struct variable *var)
1927{
1928 struct variable *next;
1929
1930 while (var) {
1931 next = var->next;
1932 var->next = G.top_var;
1933 G.top_var = var;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001934 if (var->flg_export) {
1935 debug_printf_env("%s: restoring exported '%s'\n", __func__, var->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001936 putenv(var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001937 } else {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001938 debug_printf_env("%s: restoring variable '%s'\n", __func__, var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001939 }
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001940 var = next;
1941 }
1942}
1943
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001944static struct variable *set_vars_and_save_old(char **strings)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001945{
1946 char **s;
1947 struct variable *old = NULL;
1948
1949 if (!strings)
1950 return old;
1951 s = strings;
1952 while (*s) {
1953 struct variable *var_p;
1954 struct variable **var_pp;
1955 char *eq;
1956
1957 eq = strchr(*s, '=');
1958 if (eq) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001959 var_pp = get_ptr_to_local_var(*s, eq - *s);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001960 if (var_pp) {
1961 /* Remove variable from global linked list */
1962 var_p = *var_pp;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001963 debug_printf_env("%s: removing '%s'\n", __func__, var_p->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001964 *var_pp = var_p->next;
1965 /* Add it to returned list */
1966 var_p->next = old;
1967 old = var_p;
1968 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001969 set_local_var(*s, /*exp:*/ 1, /*lvl:*/ 0, /*ro:*/ 0);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001970 }
1971 s++;
1972 }
1973 return old;
1974}
1975
1976
1977/*
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001978 * in_str support
1979 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02001980static int FAST_FUNC static_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001981{
Denys Vlasenko8391c482010-05-22 17:50:43 +02001982 int ch = *i->p;
1983 if (ch != '\0') {
1984 i->p++;
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001985 i->last_char = ch;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00001986 return ch;
Denys Vlasenko8391c482010-05-22 17:50:43 +02001987 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00001988 return EOF;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001989}
1990
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02001991static int FAST_FUNC static_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001992{
1993 return *i->p;
1994}
1995
1996#if ENABLE_HUSH_INTERACTIVE
1997
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001998static void cmdedit_update_prompt(void)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001999{
Mike Frysingerec2c6552009-03-28 12:24:44 +00002000 if (ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002001 G.PS1 = get_local_var_value("PS1");
Mike Frysingerec2c6552009-03-28 12:24:44 +00002002 if (G.PS1 == NULL)
2003 G.PS1 = "\\w \\$ ";
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002004 G.PS2 = get_local_var_value("PS2");
Denys Vlasenko690ad242009-04-30 21:24:24 +02002005 } else {
Mike Frysingerec2c6552009-03-28 12:24:44 +00002006 G.PS1 = NULL;
Denys Vlasenko690ad242009-04-30 21:24:24 +02002007 }
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002008 if (G.PS2 == NULL)
2009 G.PS2 = "> ";
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002010}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002011
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002012static const char *setup_prompt_string(int promptmode)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002013{
2014 const char *prompt_str;
2015 debug_printf("setup_prompt_string %d ", promptmode);
Mike Frysingerec2c6552009-03-28 12:24:44 +00002016 if (!ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
2017 /* Set up the prompt */
2018 if (promptmode == 0) { /* PS1 */
2019 free((char*)G.PS1);
Denys Vlasenko6db47842009-09-05 20:15:17 +02002020 /* bash uses $PWD value, even if it is set by user.
2021 * It uses current dir only if PWD is unset.
2022 * We always use current dir. */
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02002023 G.PS1 = xasprintf("%s %c ", get_cwd(0), (geteuid() != 0) ? '$' : '#');
Mike Frysingerec2c6552009-03-28 12:24:44 +00002024 prompt_str = G.PS1;
2025 } else
2026 prompt_str = G.PS2;
2027 } else
2028 prompt_str = (promptmode == 0) ? G.PS1 : G.PS2;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002029 debug_printf("result '%s'\n", prompt_str);
2030 return prompt_str;
2031}
2032
2033static void get_user_input(struct in_str *i)
2034{
2035 int r;
2036 const char *prompt_str;
2037
2038 prompt_str = setup_prompt_string(i->promptmode);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002039# if ENABLE_FEATURE_EDITING
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002040 /* Enable command line editing only while a command line
2041 * is actually being read */
2042 do {
Denys Vlasenko20704f02011-03-23 17:59:27 +01002043 /* Unicode support should be activated even if LANG is set
2044 * _during_ shell execution, not only if it was set when
2045 * shell was started. Therefore, re-check LANG every time:
2046 */
maxwen27116ba2015-08-14 21:41:28 +02002047 const char *s = get_local_var_value("LC_ALL");
2048 if (!s) s = get_local_var_value("LC_CTYPE");
2049 if (!s) s = get_local_var_value("LANG");
2050 reinit_unicode(s);
Denys Vlasenko20704f02011-03-23 17:59:27 +01002051
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002052 G.flag_SIGINT = 0;
2053 /* buglet: SIGINT will not make new prompt to appear _at once_,
2054 * only after <Enter>. (^C will work) */
Denys Vlasenko66c5b122011-02-08 05:07:02 +01002055 r = read_line_input(G.line_input_state, prompt_str, G.user_input_buf, CONFIG_FEATURE_EDITING_MAX_LEN-1, /*timeout*/ -1);
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002056 /* catch *SIGINT* etc (^C is handled by read_line_input) */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002057 check_and_run_traps();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002058 } while (r == 0 || G.flag_SIGINT); /* repeat if ^C or SIGINT */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002059 i->eof_flag = (r < 0);
2060 if (i->eof_flag) { /* EOF/error detected */
2061 G.user_input_buf[0] = EOF; /* yes, it will be truncated, it's ok */
2062 G.user_input_buf[1] = '\0';
2063 }
Denys Vlasenko8391c482010-05-22 17:50:43 +02002064# else
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002065 do {
2066 G.flag_SIGINT = 0;
Denys Vlasenkob8709032011-05-08 21:20:01 +02002067 if (i->last_char == '\0' || i->last_char == '\n') {
2068 /* Why check_and_run_traps here? Try this interactively:
2069 * $ trap 'echo INT' INT; (sleep 2; kill -INT $$) &
2070 * $ <[enter], repeatedly...>
2071 * Without check_and_run_traps, handler never runs.
2072 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002073 check_and_run_traps();
Denys Vlasenkob8709032011-05-08 21:20:01 +02002074 fputs(prompt_str, stdout);
2075 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002076 fflush_all();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002077 G.user_input_buf[0] = r = fgetc(i->file);
2078 /*G.user_input_buf[1] = '\0'; - already is and never changed */
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002079 } while (G.flag_SIGINT);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002080 i->eof_flag = (r == EOF);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002081# endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002082 i->p = G.user_input_buf;
2083}
2084
2085#endif /* INTERACTIVE */
2086
2087/* This is the magic location that prints prompts
2088 * and gets data back from the user */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002089static int FAST_FUNC file_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002090{
2091 int ch;
2092
2093 /* If there is data waiting, eat it up */
2094 if (i->p && *i->p) {
2095#if ENABLE_HUSH_INTERACTIVE
2096 take_cached:
2097#endif
2098 ch = *i->p++;
2099 if (i->eof_flag && !*i->p)
2100 ch = EOF;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002101 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002102 } else {
2103 /* need to double check i->file because we might be doing something
2104 * more complicated by now, like sourcing or substituting. */
2105#if ENABLE_HUSH_INTERACTIVE
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002106 if (G_interactive_fd && i->file == stdin) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002107 do {
2108 get_user_input(i);
2109 } while (!*i->p); /* need non-empty line */
2110 i->promptmode = 1; /* PS2 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002111 goto take_cached;
2112 }
2113#endif
Denis Vlasenko913a2012009-04-05 22:17:04 +00002114 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002115 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002116 debug_printf("file_get: got '%c' %d\n", ch, ch);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02002117 i->last_char = ch;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002118 return ch;
2119}
2120
Denis Vlasenko913a2012009-04-05 22:17:04 +00002121/* All callers guarantee this routine will never
2122 * be used right after a newline, so prompting is not needed.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002123 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002124static int FAST_FUNC file_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002125{
2126 int ch;
2127 if (i->p && *i->p) {
2128 if (i->eof_flag && !i->p[1])
2129 return EOF;
2130 return *i->p;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002131 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002132 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002133 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002134 i->eof_flag = (ch == EOF);
2135 i->peek_buf[0] = ch;
2136 i->peek_buf[1] = '\0';
2137 i->p = i->peek_buf;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002138 debug_printf("file_peek: got '%c' %d\n", ch, ch);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002139 return ch;
2140}
2141
2142static void setup_file_in_str(struct in_str *i, FILE *f)
2143{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002144 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002145 i->peek = file_peek;
2146 i->get = file_get;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002147 /* i->promptmode = 0; - PS1 (memset did it) */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002148 i->file = f;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002149 /* i->p = NULL; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002150}
2151
2152static void setup_string_in_str(struct in_str *i, const char *s)
2153{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002154 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002155 i->peek = static_peek;
2156 i->get = static_get;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002157 /* i->promptmode = 0; - PS1 (memset did it) */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002158 i->p = s;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002159 /* i->eof_flag = 0; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002160}
2161
2162
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002163/*
2164 * o_string support
2165 */
2166#define B_CHUNK (32 * sizeof(char*))
Eric Andersen25f27032001-04-26 23:22:31 +00002167
Denis Vlasenko0b677d82009-04-10 13:49:10 +00002168static void o_reset_to_empty_unquoted(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002169{
2170 o->length = 0;
Denys Vlasenko38292b62010-09-05 14:49:40 +02002171 o->has_quoted_part = 0;
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002172 if (o->data)
2173 o->data[0] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002174}
2175
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002176static void o_free(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002177{
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002178 free(o->data);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002179 memset(o, 0, sizeof(*o));
Eric Andersen25f27032001-04-26 23:22:31 +00002180}
2181
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002182static ALWAYS_INLINE void o_free_unsafe(o_string *o)
2183{
2184 free(o->data);
2185}
2186
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002187static void o_grow_by(o_string *o, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002188{
2189 if (o->length + len > o->maxlen) {
2190 o->maxlen += (2*len > B_CHUNK ? 2*len : B_CHUNK);
2191 o->data = xrealloc(o->data, 1 + o->maxlen);
2192 }
2193}
2194
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002195static void o_addchr(o_string *o, int ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002196{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002197 debug_printf("o_addchr: '%c' o->length=%d o=%p\n", ch, o->length, o);
2198 o_grow_by(o, 1);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002199 o->data[o->length] = ch;
2200 o->length++;
2201 o->data[o->length] = '\0';
2202}
2203
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002204static void o_addblock(o_string *o, const char *str, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002205{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002206 o_grow_by(o, len);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002207 memcpy(&o->data[o->length], str, len);
2208 o->length += len;
2209 o->data[o->length] = '\0';
2210}
2211
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002212static void o_addstr(o_string *o, const char *str)
Mike Frysinger98c52642009-04-02 10:02:37 +00002213{
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002214 o_addblock(o, str, strlen(str));
2215}
Denys Vlasenko2e48d532010-05-22 17:30:39 +02002216
Denys Vlasenko1e811b12010-05-22 03:12:29 +02002217#if !BB_MMU
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00002218static void nommu_addchr(o_string *o, int ch)
2219{
2220 if (o)
2221 o_addchr(o, ch);
2222}
2223#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002224# define nommu_addchr(o, str) ((void)0)
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002225#endif
2226
2227static void o_addstr_with_NUL(o_string *o, const char *str)
2228{
2229 o_addblock(o, str, strlen(str) + 1);
Mike Frysinger98c52642009-04-02 10:02:37 +00002230}
2231
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002232/*
Denys Vlasenko238081f2010-10-03 14:26:26 +02002233 * HUSH_BRACE_EXPANSION code needs corresponding quoting on variable expansion side.
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002234 * Currently, "v='{q,w}'; echo $v" erroneously expands braces in $v.
2235 * Apparently, on unquoted $v bash still does globbing
2236 * ("v='*.txt'; echo $v" prints all .txt files),
2237 * but NOT brace expansion! Thus, there should be TWO independent
2238 * quoting mechanisms on $v expansion side: one protects
2239 * $v from brace expansion, and other additionally protects "$v" against globbing.
2240 * We have only second one.
2241 */
2242
Denys Vlasenko9e800222010-10-03 14:28:04 +02002243#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002244# define MAYBE_BRACES "{}"
2245#else
2246# define MAYBE_BRACES ""
2247#endif
2248
Eric Andersen25f27032001-04-26 23:22:31 +00002249/* My analysis of quoting semantics tells me that state information
2250 * is associated with a destination, not a source.
2251 */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002252static void o_addqchr(o_string *o, int ch)
Eric Andersen25f27032001-04-26 23:22:31 +00002253{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002254 int sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002255 char *found = strchr("*?[\\" MAYBE_BRACES, ch);
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002256 if (found)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002257 sz++;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002258 o_grow_by(o, sz);
2259 if (found) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002260 o->data[o->length] = '\\';
2261 o->length++;
Eric Andersen25f27032001-04-26 23:22:31 +00002262 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002263 o->data[o->length] = ch;
2264 o->length++;
2265 o->data[o->length] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002266}
2267
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002268static void o_addQchr(o_string *o, int ch)
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002269{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002270 int sz = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002271 if ((o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)
2272 && strchr("*?[\\" MAYBE_BRACES, ch)
2273 ) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002274 sz++;
2275 o->data[o->length] = '\\';
2276 o->length++;
2277 }
2278 o_grow_by(o, sz);
2279 o->data[o->length] = ch;
2280 o->length++;
2281 o->data[o->length] = '\0';
2282}
2283
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002284static void o_addqblock(o_string *o, const char *str, int len)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002285{
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002286 while (len) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002287 char ch;
2288 int sz;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002289 int ordinary_cnt = strcspn(str, "*?[\\" MAYBE_BRACES);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002290 if (ordinary_cnt > len) /* paranoia */
2291 ordinary_cnt = len;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002292 o_addblock(o, str, ordinary_cnt);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002293 if (ordinary_cnt == len)
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002294 return; /* NUL is already added by o_addblock */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002295 str += ordinary_cnt;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002296 len -= ordinary_cnt + 1; /* we are processing + 1 char below */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002297
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002298 ch = *str++;
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002299 sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002300 if (ch) { /* it is necessarily one of "*?[\\" MAYBE_BRACES */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002301 sz++;
2302 o->data[o->length] = '\\';
2303 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002304 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002305 o_grow_by(o, sz);
2306 o->data[o->length] = ch;
2307 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002308 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002309 o->data[o->length] = '\0';
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002310}
2311
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002312static void o_addQblock(o_string *o, const char *str, int len)
2313{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002314 if (!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002315 o_addblock(o, str, len);
2316 return;
2317 }
2318 o_addqblock(o, str, len);
2319}
2320
Denys Vlasenko38292b62010-09-05 14:49:40 +02002321static void o_addQstr(o_string *o, const char *str)
2322{
2323 o_addQblock(o, str, strlen(str));
2324}
2325
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002326/* A special kind of o_string for $VAR and `cmd` expansion.
2327 * It contains char* list[] at the beginning, which is grown in 16 element
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002328 * increments. Actual string data starts at the next multiple of 16 * (char*).
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002329 * list[i] contains an INDEX (int!) into this string data.
2330 * It means that if list[] needs to grow, data needs to be moved higher up
2331 * but list[i]'s need not be modified.
2332 * NB: remembering how many list[i]'s you have there is crucial.
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002333 * o_finalize_list() operation post-processes this structure - calculates
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002334 * and stores actual char* ptrs in list[]. Oh, it NULL terminates it as well.
2335 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002336#if DEBUG_EXPAND || DEBUG_GLOB
2337static void debug_print_list(const char *prefix, o_string *o, int n)
2338{
2339 char **list = (char**)o->data;
2340 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2341 int i = 0;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002342
2343 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002344 fdprintf(2, "%s: list:%p n:%d string_start:%d length:%d maxlen:%d glob:%d quoted:%d escape:%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002345 prefix, list, n, string_start, o->length, o->maxlen,
2346 !!(o->o_expflags & EXP_FLAG_GLOB),
2347 o->has_quoted_part,
2348 !!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002349 while (i < n) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002350 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002351 fdprintf(2, " list[%d]=%d '%s' %p\n", i, (int)(uintptr_t)list[i],
2352 o->data + (int)(uintptr_t)list[i] + string_start,
2353 o->data + (int)(uintptr_t)list[i] + string_start);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002354 i++;
2355 }
2356 if (n) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002357 const char *p = o->data + (int)(uintptr_t)list[n - 1] + string_start;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002358 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002359 fdprintf(2, " total_sz:%ld\n", (long)((p + strlen(p) + 1) - o->data));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002360 }
2361}
2362#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002363# define debug_print_list(prefix, o, n) ((void)0)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002364#endif
2365
2366/* n = o_save_ptr_helper(str, n) "starts new string" by storing an index value
2367 * in list[n] so that it points past last stored byte so far.
2368 * It returns n+1. */
2369static int o_save_ptr_helper(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002370{
2371 char **list = (char**)o->data;
Denis Vlasenko895bea22008-06-10 18:06:24 +00002372 int string_start;
2373 int string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002374
2375 if (!o->has_empty_slot) {
Denis Vlasenko895bea22008-06-10 18:06:24 +00002376 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2377 string_len = o->length - string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002378 if (!(n & 0xf)) { /* 0, 0x10, 0x20...? */
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002379 debug_printf_list("list[%d]=%d string_start=%d (growing)\n", n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002380 /* list[n] points to string_start, make space for 16 more pointers */
2381 o->maxlen += 0x10 * sizeof(list[0]);
2382 o->data = xrealloc(o->data, o->maxlen + 1);
Denis Vlasenko7049ff82008-06-25 09:53:17 +00002383 list = (char**)o->data;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002384 memmove(list + n + 0x10, list + n, string_len);
2385 o->length += 0x10 * sizeof(list[0]);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002386 } else {
2387 debug_printf_list("list[%d]=%d string_start=%d\n",
2388 n, string_len, string_start);
2389 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002390 } else {
2391 /* We have empty slot at list[n], reuse without growth */
Denis Vlasenko895bea22008-06-10 18:06:24 +00002392 string_start = ((n+1 + 0xf) & ~0xf) * sizeof(list[0]); /* NB: n+1! */
2393 string_len = o->length - string_start;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002394 debug_printf_list("list[%d]=%d string_start=%d (empty slot)\n",
2395 n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002396 o->has_empty_slot = 0;
2397 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002398 o->has_quoted_part = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002399 list[n] = (char*)(uintptr_t)string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002400 return n + 1;
2401}
2402
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002403/* "What was our last o_save_ptr'ed position (byte offset relative o->data)?" */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002404static int o_get_last_ptr(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002405{
2406 char **list = (char**)o->data;
2407 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2408
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002409 return ((int)(uintptr_t)list[n-1]) + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002410}
2411
Denys Vlasenko9e800222010-10-03 14:28:04 +02002412#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002413/* There in a GNU extension, GLOB_BRACE, but it is not usable:
2414 * first, it processes even {a} (no commas), second,
2415 * I didn't manage to make it return strings when they don't match
Denys Vlasenko160746b2009-11-16 05:51:18 +01002416 * existing files. Need to re-implement it.
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002417 */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002418
2419/* Helper */
2420static int glob_needed(const char *s)
2421{
2422 while (*s) {
2423 if (*s == '\\') {
2424 if (!s[1])
2425 return 0;
2426 s += 2;
2427 continue;
2428 }
2429 if (*s == '*' || *s == '[' || *s == '?' || *s == '{')
2430 return 1;
2431 s++;
2432 }
2433 return 0;
2434}
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002435/* Return pointer to next closing brace or to comma */
2436static const char *next_brace_sub(const char *cp)
2437{
2438 unsigned depth = 0;
2439 cp++;
2440 while (*cp != '\0') {
2441 if (*cp == '\\') {
2442 if (*++cp == '\0')
2443 break;
2444 cp++;
2445 continue;
Denys Vlasenko3581c622010-01-25 13:39:24 +01002446 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002447 if ((*cp == '}' && depth-- == 0) || (*cp == ',' && depth == 0))
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002448 break;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002449 if (*cp++ == '{')
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002450 depth++;
2451 }
2452
2453 return *cp != '\0' ? cp : NULL;
2454}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002455/* Recursive brace globber. Note: may garble pattern[]. */
2456static int glob_brace(char *pattern, o_string *o, int n)
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002457{
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002458 char *new_pattern_buf;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002459 const char *begin;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002460 const char *next;
2461 const char *rest;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002462 const char *p;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002463 size_t rest_len;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002464
2465 debug_printf_glob("glob_brace('%s')\n", pattern);
2466
2467 begin = pattern;
2468 while (1) {
2469 if (*begin == '\0')
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002470 goto simple_glob;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002471 if (*begin == '{') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002472 /* Find the first sub-pattern and at the same time
2473 * find the rest after the closing brace */
2474 next = next_brace_sub(begin);
2475 if (next == NULL) {
2476 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002477 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002478 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002479 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002480 /* "{abc}" with no commas - illegal
2481 * brace expr, disregard and skip it */
2482 begin = next + 1;
2483 continue;
2484 }
2485 break;
2486 }
2487 if (*begin == '\\' && begin[1] != '\0')
2488 begin++;
2489 begin++;
2490 }
2491 debug_printf_glob("begin:%s\n", begin);
2492 debug_printf_glob("next:%s\n", next);
2493
2494 /* Now find the end of the whole brace expression */
2495 rest = next;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002496 while (*rest != '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002497 rest = next_brace_sub(rest);
2498 if (rest == NULL) {
2499 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002500 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002501 }
2502 debug_printf_glob("rest:%s\n", rest);
2503 }
2504 rest_len = strlen(++rest) + 1;
2505
2506 /* We are sure the brace expression is well-formed */
2507
2508 /* Allocate working buffer large enough for our work */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002509 new_pattern_buf = xmalloc(strlen(pattern));
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002510
2511 /* We have a brace expression. BEGIN points to the opening {,
2512 * NEXT points past the terminator of the first element, and REST
2513 * points past the final }. We will accumulate result names from
2514 * recursive runs for each brace alternative in the buffer using
2515 * GLOB_APPEND. */
2516
2517 p = begin + 1;
2518 while (1) {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002519 /* Construct the new glob expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002520 memcpy(
2521 mempcpy(
2522 mempcpy(new_pattern_buf,
2523 /* We know the prefix for all sub-patterns */
2524 pattern, begin - pattern),
2525 p, next - p),
2526 rest, rest_len);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002527
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002528 /* Note: glob_brace() may garble new_pattern_buf[].
2529 * That's why we re-copy prefix every time (1st memcpy above).
2530 */
2531 n = glob_brace(new_pattern_buf, o, n);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002532 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002533 /* We saw the last entry */
2534 break;
2535 }
2536 p = next + 1;
2537 next = next_brace_sub(next);
2538 }
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002539 free(new_pattern_buf);
2540 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002541
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002542 simple_glob:
2543 {
2544 int gr;
2545 glob_t globdata;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002546
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002547 memset(&globdata, 0, sizeof(globdata));
2548 gr = glob(pattern, 0, NULL, &globdata);
2549 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
2550 if (gr != 0) {
2551 if (gr == GLOB_NOMATCH) {
2552 globfree(&globdata);
2553 /* NB: garbles parameter */
2554 unbackslash(pattern);
2555 o_addstr_with_NUL(o, pattern);
2556 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2557 return o_save_ptr_helper(o, n);
2558 }
2559 if (gr == GLOB_NOSPACE)
Tanguy Pruvot8aeb3712011-06-30 08:59:26 +02002560 bb_error_msg_and_die("%s", bb_msg_memory_exhausted);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002561 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2562 * but we didn't specify it. Paranoia again. */
2563 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
2564 }
2565 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2566 char **argv = globdata.gl_pathv;
2567 while (1) {
2568 o_addstr_with_NUL(o, *argv);
2569 n = o_save_ptr_helper(o, n);
2570 argv++;
2571 if (!*argv)
2572 break;
2573 }
2574 }
2575 globfree(&globdata);
2576 }
2577 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002578}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002579/* Performs globbing on last list[],
2580 * saving each result as a new list[].
2581 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002582static int perform_glob(o_string *o, int n)
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002583{
2584 char *pattern, *copy;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002585
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002586 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002587 if (!o->data)
2588 return o_save_ptr_helper(o, n);
2589 pattern = o->data + o_get_last_ptr(o, n);
2590 debug_printf_glob("glob pattern '%s'\n", pattern);
2591 if (!glob_needed(pattern)) {
2592 /* unbackslash last string in o in place, fix length */
2593 o->length = unbackslash(pattern) - o->data;
2594 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2595 return o_save_ptr_helper(o, n);
2596 }
2597
2598 copy = xstrdup(pattern);
2599 /* "forget" pattern in o */
2600 o->length = pattern - o->data;
2601 n = glob_brace(copy, o, n);
2602 free(copy);
2603 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002604 debug_print_list("perform_glob returning", o, n);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002605 return n;
2606}
2607
Denys Vlasenko238081f2010-10-03 14:26:26 +02002608#else /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002609
2610/* Helper */
2611static int glob_needed(const char *s)
2612{
2613 while (*s) {
2614 if (*s == '\\') {
2615 if (!s[1])
2616 return 0;
2617 s += 2;
2618 continue;
2619 }
2620 if (*s == '*' || *s == '[' || *s == '?')
2621 return 1;
2622 s++;
2623 }
2624 return 0;
2625}
2626/* Performs globbing on last list[],
2627 * saving each result as a new list[].
2628 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002629static int perform_glob(o_string *o, int n)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002630{
2631 glob_t globdata;
2632 int gr;
2633 char *pattern;
2634
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002635 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002636 if (!o->data)
2637 return o_save_ptr_helper(o, n);
2638 pattern = o->data + o_get_last_ptr(o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002639 debug_printf_glob("glob pattern '%s'\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002640 if (!glob_needed(pattern)) {
2641 literal:
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002642 /* unbackslash last string in o in place, fix length */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002643 o->length = unbackslash(pattern) - o->data;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002644 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002645 return o_save_ptr_helper(o, n);
2646 }
2647
2648 memset(&globdata, 0, sizeof(globdata));
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002649 /* Can't use GLOB_NOCHECK: it does not unescape the string.
2650 * If we glob "*.\*" and don't find anything, we need
2651 * to fall back to using literal "*.*", but GLOB_NOCHECK
2652 * will return "*.\*"!
2653 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002654 gr = glob(pattern, 0, NULL, &globdata);
2655 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002656 if (gr != 0) {
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002657 if (gr == GLOB_NOMATCH) {
2658 globfree(&globdata);
2659 goto literal;
2660 }
2661 if (gr == GLOB_NOSPACE)
Tanguy Pruvot8aeb3712011-06-30 08:59:26 +02002662 bb_error_msg_and_die("%s", bb_msg_memory_exhausted);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002663 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2664 * but we didn't specify it. Paranoia again. */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002665 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002666 }
2667 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2668 char **argv = globdata.gl_pathv;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002669 /* "forget" pattern in o */
2670 o->length = pattern - o->data;
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002671 while (1) {
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002672 o_addstr_with_NUL(o, *argv);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002673 n = o_save_ptr_helper(o, n);
2674 argv++;
2675 if (!*argv)
2676 break;
2677 }
2678 }
2679 globfree(&globdata);
2680 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002681 debug_print_list("perform_glob returning", o, n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002682 return n;
2683}
2684
Denys Vlasenko238081f2010-10-03 14:26:26 +02002685#endif /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002686
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002687/* If o->o_expflags & EXP_FLAG_GLOB, glob the string so far remembered.
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002688 * Otherwise, just finish current list[] and start new */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002689static int o_save_ptr(o_string *o, int n)
2690{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002691 if (o->o_expflags & EXP_FLAG_GLOB) {
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002692 /* If o->has_empty_slot, list[n] was already globbed
2693 * (if it was requested back then when it was filled)
2694 * so don't do that again! */
2695 if (!o->has_empty_slot)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002696 return perform_glob(o, n); /* o_save_ptr_helper is inside */
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002697 }
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002698 return o_save_ptr_helper(o, n);
2699}
2700
2701/* "Please convert list[n] to real char* ptrs, and NULL terminate it." */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002702static char **o_finalize_list(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002703{
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002704 char **list;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002705 int string_start;
2706
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002707 n = o_save_ptr(o, n); /* force growth for list[n] if necessary */
2708 if (DEBUG_EXPAND)
2709 debug_print_list("finalized", o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002710 debug_printf_expand("finalized n:%d\n", n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002711 list = (char**)o->data;
2712 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2713 list[--n] = NULL;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002714 while (n) {
2715 n--;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002716 list[n] = o->data + (int)(uintptr_t)list[n] + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002717 }
2718 return list;
2719}
2720
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002721static void free_pipe_list(struct pipe *pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002722
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002723/* Returns pi->next - next pipe in the list */
2724static struct pipe *free_pipe(struct pipe *pi)
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002725{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002726 struct pipe *next;
2727 int i;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002728
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002729 debug_printf_clean("free_pipe (pid %d)\n", getpid());
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002730 for (i = 0; i < pi->num_cmds; i++) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002731 struct command *command;
2732 struct redir_struct *r, *rnext;
2733
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002734 command = &pi->cmds[i];
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002735 debug_printf_clean(" command %d:\n", i);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002736 if (command->argv) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002737 if (DEBUG_CLEAN) {
2738 int a;
2739 char **p;
2740 for (a = 0, p = command->argv; *p; a++, p++) {
2741 debug_printf_clean(" argv[%d] = %s\n", a, *p);
2742 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002743 }
2744 free_strings(command->argv);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002745 //command->argv = NULL;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002746 }
2747 /* not "else if": on syntax error, we may have both! */
2748 if (command->group) {
Denys Vlasenko9d617c42009-06-09 18:40:52 +02002749 debug_printf_clean(" begin group (cmd_type:%d)\n",
2750 command->cmd_type);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002751 free_pipe_list(command->group);
2752 debug_printf_clean(" end group\n");
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002753 //command->group = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002754 }
Denis Vlasenkoed055212009-04-11 10:37:10 +00002755 /* else is crucial here.
2756 * If group != NULL, child_func is meaningless */
2757#if ENABLE_HUSH_FUNCTIONS
2758 else if (command->child_func) {
2759 debug_printf_exec("cmd %p releases child func at %p\n", command, command->child_func);
2760 command->child_func->parent_cmd = NULL;
2761 }
2762#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002763#if !BB_MMU
2764 free(command->group_as_string);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002765 //command->group_as_string = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002766#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002767 for (r = command->redirects; r; r = rnext) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002768 debug_printf_clean(" redirect %d%s",
2769 r->rd_fd, redir_table[r->rd_type].descrip);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002770 /* guard against the case >$FOO, where foo is unset or blank */
2771 if (r->rd_filename) {
2772 debug_printf_clean(" fname:'%s'\n", r->rd_filename);
2773 free(r->rd_filename);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002774 //r->rd_filename = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002775 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002776 debug_printf_clean(" rd_dup:%d\n", r->rd_dup);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002777 rnext = r->next;
2778 free(r);
2779 }
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002780 //command->redirects = NULL;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002781 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002782 free(pi->cmds); /* children are an array, they get freed all at once */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002783 //pi->cmds = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002784#if ENABLE_HUSH_JOB
2785 free(pi->cmdtext);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002786 //pi->cmdtext = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002787#endif
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002788
2789 next = pi->next;
2790 free(pi);
2791 return next;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002792}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002793
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002794static void free_pipe_list(struct pipe *pi)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002795{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002796 while (pi) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002797#if HAS_KEYWORDS
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002798 debug_printf_clean("pipe reserved word %d\n", pi->res_word);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002799#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002800 debug_printf_clean("pipe followup code %d\n", pi->followup);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002801 pi = free_pipe(pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002802 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002803}
2804
2805
Denys Vlasenkob36abf22010-09-05 14:50:59 +02002806/*** Parsing routines ***/
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002807
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002808#ifndef debug_print_tree
2809static void debug_print_tree(struct pipe *pi, int lvl)
2810{
2811 static const char *const PIPE[] = {
2812 [PIPE_SEQ] = "SEQ",
2813 [PIPE_AND] = "AND",
2814 [PIPE_OR ] = "OR" ,
2815 [PIPE_BG ] = "BG" ,
2816 };
2817 static const char *RES[] = {
2818 [RES_NONE ] = "NONE" ,
2819# if ENABLE_HUSH_IF
2820 [RES_IF ] = "IF" ,
2821 [RES_THEN ] = "THEN" ,
2822 [RES_ELIF ] = "ELIF" ,
2823 [RES_ELSE ] = "ELSE" ,
2824 [RES_FI ] = "FI" ,
2825# endif
2826# if ENABLE_HUSH_LOOPS
2827 [RES_FOR ] = "FOR" ,
2828 [RES_WHILE] = "WHILE",
2829 [RES_UNTIL] = "UNTIL",
2830 [RES_DO ] = "DO" ,
2831 [RES_DONE ] = "DONE" ,
2832# endif
2833# if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
2834 [RES_IN ] = "IN" ,
2835# endif
2836# if ENABLE_HUSH_CASE
2837 [RES_CASE ] = "CASE" ,
2838 [RES_CASE_IN ] = "CASE_IN" ,
2839 [RES_MATCH] = "MATCH",
2840 [RES_CASE_BODY] = "CASE_BODY",
2841 [RES_ESAC ] = "ESAC" ,
2842# endif
2843 [RES_XXXX ] = "XXXX" ,
2844 [RES_SNTX ] = "SNTX" ,
2845 };
2846 static const char *const CMDTYPE[] = {
2847 "{}",
2848 "()",
2849 "[noglob]",
2850# if ENABLE_HUSH_FUNCTIONS
2851 "func()",
2852# endif
2853 };
2854
2855 int pin, prn;
2856
2857 pin = 0;
2858 while (pi) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002859 fdprintf(2, "%*spipe %d res_word=%s followup=%d %s\n", lvl*2, "",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002860 pin, RES[pi->res_word], pi->followup, PIPE[pi->followup]);
2861 prn = 0;
2862 while (prn < pi->num_cmds) {
2863 struct command *command = &pi->cmds[prn];
2864 char **argv = command->argv;
2865
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002866 fdprintf(2, "%*s cmd %d assignment_cnt:%d",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002867 lvl*2, "", prn,
2868 command->assignment_cnt);
2869 if (command->group) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002870 fdprintf(2, " group %s: (argv=%p)%s%s\n",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002871 CMDTYPE[command->cmd_type],
2872 argv
2873# if !BB_MMU
2874 , " group_as_string:", command->group_as_string
2875# else
2876 , "", ""
2877# endif
2878 );
2879 debug_print_tree(command->group, lvl+1);
2880 prn++;
2881 continue;
2882 }
2883 if (argv) while (*argv) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002884 fdprintf(2, " '%s'", *argv);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002885 argv++;
2886 }
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002887 fdprintf(2, "\n");
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002888 prn++;
2889 }
2890 pi = pi->next;
2891 pin++;
2892 }
2893}
2894#endif /* debug_print_tree */
2895
Denis Vlasenkoac678ec2007-04-16 22:32:04 +00002896static struct pipe *new_pipe(void)
2897{
Eric Andersen25f27032001-04-26 23:22:31 +00002898 struct pipe *pi;
Denis Vlasenko3ac0e002007-04-28 16:45:22 +00002899 pi = xzalloc(sizeof(struct pipe));
Denis Vlasenkoa8442002008-06-14 11:00:17 +00002900 /*pi->followup = 0; - deliberately invalid value */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00002901 /*pi->res_word = RES_NONE; - RES_NONE is 0 anyway */
Eric Andersen25f27032001-04-26 23:22:31 +00002902 return pi;
2903}
2904
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002905/* Command (member of a pipe) is complete, or we start a new pipe
2906 * if ctx->command is NULL.
2907 * No errors possible here.
2908 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002909static int done_command(struct parse_context *ctx)
2910{
2911 /* The command is really already in the pipe structure, so
2912 * advance the pipe counter and make a new, null command. */
2913 struct pipe *pi = ctx->pipe;
2914 struct command *command = ctx->command;
2915
2916 if (command) {
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002917 if (IS_NULL_CMD(command)) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002918 debug_printf_parse("done_command: skipping null cmd, num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002919 goto clear_and_ret;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002920 }
2921 pi->num_cmds++;
2922 debug_printf_parse("done_command: ++num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002923 //debug_print_tree(ctx->list_head, 20);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002924 } else {
2925 debug_printf_parse("done_command: initializing, num_cmds=%d\n", pi->num_cmds);
2926 }
2927
2928 /* Only real trickiness here is that the uncommitted
2929 * command structure is not counted in pi->num_cmds. */
2930 pi->cmds = xrealloc(pi->cmds, sizeof(*pi->cmds) * (pi->num_cmds+1));
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002931 ctx->command = command = &pi->cmds[pi->num_cmds];
2932 clear_and_ret:
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002933 memset(command, 0, sizeof(*command));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002934 return pi->num_cmds; /* used only for 0/nonzero check */
2935}
2936
2937static void done_pipe(struct parse_context *ctx, pipe_style type)
2938{
2939 int not_null;
2940
2941 debug_printf_parse("done_pipe entered, followup %d\n", type);
2942 /* Close previous command */
2943 not_null = done_command(ctx);
2944 ctx->pipe->followup = type;
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002945#if HAS_KEYWORDS
2946 ctx->pipe->pi_inverted = ctx->ctx_inverted;
2947 ctx->ctx_inverted = 0;
2948 ctx->pipe->res_word = ctx->ctx_res_w;
2949#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002950
2951 /* Without this check, even just <enter> on command line generates
2952 * tree of three NOPs (!). Which is harmless but annoying.
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002953 * IOW: it is safe to do it unconditionally. */
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002954 if (not_null
Denis Vlasenko7f959372009-04-14 08:06:59 +00002955#if ENABLE_HUSH_IF
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002956 || ctx->ctx_res_w == RES_FI
Denis Vlasenko7f959372009-04-14 08:06:59 +00002957#endif
2958#if ENABLE_HUSH_LOOPS
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002959 || ctx->ctx_res_w == RES_DONE
2960 || ctx->ctx_res_w == RES_FOR
2961 || ctx->ctx_res_w == RES_IN
Denis Vlasenko7f959372009-04-14 08:06:59 +00002962#endif
2963#if ENABLE_HUSH_CASE
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002964 || ctx->ctx_res_w == RES_ESAC
2965#endif
2966 ) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002967 struct pipe *new_p;
2968 debug_printf_parse("done_pipe: adding new pipe: "
2969 "not_null:%d ctx->ctx_res_w:%d\n",
2970 not_null, ctx->ctx_res_w);
2971 new_p = new_pipe();
2972 ctx->pipe->next = new_p;
2973 ctx->pipe = new_p;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002974 /* RES_THEN, RES_DO etc are "sticky" -
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002975 * they remain set for pipes inside if/while.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002976 * This is used to control execution.
2977 * RES_FOR and RES_IN are NOT sticky (needed to support
2978 * cases where variable or value happens to match a keyword):
2979 */
2980#if ENABLE_HUSH_LOOPS
2981 if (ctx->ctx_res_w == RES_FOR
2982 || ctx->ctx_res_w == RES_IN)
2983 ctx->ctx_res_w = RES_NONE;
2984#endif
2985#if ENABLE_HUSH_CASE
2986 if (ctx->ctx_res_w == RES_MATCH)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02002987 ctx->ctx_res_w = RES_CASE_BODY;
2988 if (ctx->ctx_res_w == RES_CASE)
2989 ctx->ctx_res_w = RES_CASE_IN;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002990#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002991 ctx->command = NULL; /* trick done_command below */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002992 /* Create the memory for command, roughly:
2993 * ctx->pipe->cmds = new struct command;
2994 * ctx->command = &ctx->pipe->cmds[0];
2995 */
2996 done_command(ctx);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002997 //debug_print_tree(ctx->list_head, 10);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002998 }
2999 debug_printf_parse("done_pipe return\n");
3000}
3001
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003002static void initialize_context(struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003003{
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003004 memset(ctx, 0, sizeof(*ctx));
Denis Vlasenko1a735862007-05-23 00:32:25 +00003005 ctx->pipe = ctx->list_head = new_pipe();
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003006 /* Create the memory for command, roughly:
3007 * ctx->pipe->cmds = new struct command;
3008 * ctx->command = &ctx->pipe->cmds[0];
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003009 */
3010 done_command(ctx);
Eric Andersen25f27032001-04-26 23:22:31 +00003011}
3012
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003013/* If a reserved word is found and processed, parse context is modified
3014 * and 1 is returned.
Eric Andersen25f27032001-04-26 23:22:31 +00003015 */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003016#if HAS_KEYWORDS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003017struct reserved_combo {
3018 char literal[6];
3019 unsigned char res;
3020 unsigned char assignment_flag;
3021 int flag;
3022};
3023enum {
3024 FLAG_END = (1 << RES_NONE ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003025# if ENABLE_HUSH_IF
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003026 FLAG_IF = (1 << RES_IF ),
3027 FLAG_THEN = (1 << RES_THEN ),
3028 FLAG_ELIF = (1 << RES_ELIF ),
3029 FLAG_ELSE = (1 << RES_ELSE ),
3030 FLAG_FI = (1 << RES_FI ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003031# endif
3032# if ENABLE_HUSH_LOOPS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003033 FLAG_FOR = (1 << RES_FOR ),
3034 FLAG_WHILE = (1 << RES_WHILE),
3035 FLAG_UNTIL = (1 << RES_UNTIL),
3036 FLAG_DO = (1 << RES_DO ),
3037 FLAG_DONE = (1 << RES_DONE ),
3038 FLAG_IN = (1 << RES_IN ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003039# endif
3040# if ENABLE_HUSH_CASE
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003041 FLAG_MATCH = (1 << RES_MATCH),
3042 FLAG_ESAC = (1 << RES_ESAC ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003043# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003044 FLAG_START = (1 << RES_XXXX ),
3045};
3046
3047static const struct reserved_combo* match_reserved_word(o_string *word)
3048{
Eric Andersen25f27032001-04-26 23:22:31 +00003049 /* Mostly a list of accepted follow-up reserved words.
3050 * FLAG_END means we are done with the sequence, and are ready
3051 * to turn the compound list into a command.
3052 * FLAG_START means the word must start a new compound list.
3053 */
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003054 static const struct reserved_combo reserved_list[] = {
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003055# if ENABLE_HUSH_IF
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003056 { "!", RES_NONE, NOT_ASSIGNMENT , 0 },
3057 { "if", RES_IF, MAYBE_ASSIGNMENT, FLAG_THEN | FLAG_START },
3058 { "then", RES_THEN, MAYBE_ASSIGNMENT, FLAG_ELIF | FLAG_ELSE | FLAG_FI },
3059 { "elif", RES_ELIF, MAYBE_ASSIGNMENT, FLAG_THEN },
3060 { "else", RES_ELSE, MAYBE_ASSIGNMENT, FLAG_FI },
3061 { "fi", RES_FI, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003062# endif
3063# if ENABLE_HUSH_LOOPS
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003064 { "for", RES_FOR, NOT_ASSIGNMENT , FLAG_IN | FLAG_DO | FLAG_START },
3065 { "while", RES_WHILE, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3066 { "until", RES_UNTIL, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3067 { "in", RES_IN, NOT_ASSIGNMENT , FLAG_DO },
3068 { "do", RES_DO, MAYBE_ASSIGNMENT, FLAG_DONE },
3069 { "done", RES_DONE, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003070# endif
3071# if ENABLE_HUSH_CASE
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003072 { "case", RES_CASE, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_START },
3073 { "esac", RES_ESAC, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003074# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003075 };
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003076 const struct reserved_combo *r;
3077
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02003078 for (r = reserved_list; r < reserved_list + ARRAY_SIZE(reserved_list); r++) {
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003079 if (strcmp(word->data, r->literal) == 0)
3080 return r;
3081 }
3082 return NULL;
3083}
Denis Vlasenkobb929512009-04-16 10:59:40 +00003084/* Return 0: not a keyword, 1: keyword
3085 */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003086static int reserved_word(o_string *word, struct parse_context *ctx)
3087{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003088# if ENABLE_HUSH_CASE
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003089 static const struct reserved_combo reserved_match = {
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003090 "", RES_MATCH, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_ESAC
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003091 };
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003092# endif
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003093 const struct reserved_combo *r;
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003094
Denys Vlasenko38292b62010-09-05 14:49:40 +02003095 if (word->has_quoted_part)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003096 return 0;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003097 r = match_reserved_word(word);
3098 if (!r)
3099 return 0;
3100
3101 debug_printf("found reserved word %s, res %d\n", r->literal, r->res);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003102# if ENABLE_HUSH_CASE
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003103 if (r->res == RES_IN && ctx->ctx_res_w == RES_CASE_IN) {
3104 /* "case word IN ..." - IN part starts first MATCH part */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003105 r = &reserved_match;
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003106 } else
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003107# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003108 if (r->flag == 0) { /* '!' */
3109 if (ctx->ctx_inverted) { /* bash doesn't accept '! ! true' */
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003110 syntax_error("! ! command");
Denis Vlasenkobb929512009-04-16 10:59:40 +00003111 ctx->ctx_res_w = RES_SNTX;
Eric Andersen25f27032001-04-26 23:22:31 +00003112 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003113 ctx->ctx_inverted = 1;
Denis Vlasenko1a735862007-05-23 00:32:25 +00003114 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003115 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003116 if (r->flag & FLAG_START) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003117 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003118
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003119 old = xmalloc(sizeof(*old));
3120 debug_printf_parse("push stack %p\n", old);
3121 *old = *ctx; /* physical copy */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003122 initialize_context(ctx);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003123 ctx->stack = old;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003124 } else if (/*ctx->ctx_res_w == RES_NONE ||*/ !(ctx->old_flag & (1 << r->res))) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003125 syntax_error_at(word->data);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003126 ctx->ctx_res_w = RES_SNTX;
3127 return 1;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003128 } else {
3129 /* "{...} fi" is ok. "{...} if" is not
3130 * Example:
3131 * if { echo foo; } then { echo bar; } fi */
3132 if (ctx->command->group)
3133 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003134 }
Denis Vlasenkobb929512009-04-16 10:59:40 +00003135
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003136 ctx->ctx_res_w = r->res;
3137 ctx->old_flag = r->flag;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003138 word->o_assignment = r->assignment_flag;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003139 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
Denis Vlasenkobb929512009-04-16 10:59:40 +00003140
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003141 if (ctx->old_flag & FLAG_END) {
3142 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003143
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003144 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003145 debug_printf_parse("pop stack %p\n", ctx->stack);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003146 old = ctx->stack;
3147 old->command->group = ctx->list_head;
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003148 old->command->cmd_type = CMD_NORMAL;
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003149# if !BB_MMU
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003150 o_addstr(&old->as_string, ctx->as_string.data);
3151 o_free_unsafe(&ctx->as_string);
3152 old->command->group_as_string = xstrdup(old->as_string.data);
3153 debug_printf_parse("pop, remembering as:'%s'\n",
3154 old->command->group_as_string);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003155# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003156 *ctx = *old; /* physical copy */
3157 free(old);
3158 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003159 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003160}
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003161#endif /* HAS_KEYWORDS */
Eric Andersen25f27032001-04-26 23:22:31 +00003162
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003163/* Word is complete, look at it and update parsing context.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003164 * Normal return is 0. Syntax errors return 1.
3165 * Note: on return, word is reset, but not o_free'd!
3166 */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003167static int done_word(o_string *word, struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003168{
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003169 struct command *command = ctx->command;
Eric Andersen25f27032001-04-26 23:22:31 +00003170
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003171 debug_printf_parse("done_word entered: '%s' %p\n", word->data, command);
Denys Vlasenko38292b62010-09-05 14:49:40 +02003172 if (word->length == 0 && !word->has_quoted_part) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003173 debug_printf_parse("done_word return 0: true null, ignored\n");
3174 return 0;
Eric Andersen25f27032001-04-26 23:22:31 +00003175 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003176
Eric Andersen25f27032001-04-26 23:22:31 +00003177 if (ctx->pending_redirect) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003178 /* We do not glob in e.g. >*.tmp case. bash seems to glob here
3179 * only if run as "bash", not "sh" */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003180 /* http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3181 * "2.7 Redirection
3182 * ...the word that follows the redirection operator
3183 * shall be subjected to tilde expansion, parameter expansion,
3184 * command substitution, arithmetic expansion, and quote
3185 * removal. Pathname expansion shall not be performed
3186 * on the word by a non-interactive shell; an interactive
3187 * shell may perform it, but shall do so only when
3188 * the expansion would result in one word."
3189 */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003190 ctx->pending_redirect->rd_filename = xstrdup(word->data);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003191 /* Cater for >\file case:
3192 * >\a creates file a; >\\a, >"\a", >"\\a" create file \a
3193 * Same with heredocs:
3194 * for <<\H delim is H; <<\\H, <<"\H", <<"\\H" - \H
3195 */
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003196 if (ctx->pending_redirect->rd_type == REDIRECT_HEREDOC) {
3197 unbackslash(ctx->pending_redirect->rd_filename);
3198 /* Is it <<"HEREDOC"? */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003199 if (word->has_quoted_part) {
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003200 ctx->pending_redirect->rd_dup |= HEREDOC_QUOTED;
3201 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003202 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003203 debug_printf_parse("word stored in rd_filename: '%s'\n", word->data);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003204 ctx->pending_redirect = NULL;
Eric Andersen25f27032001-04-26 23:22:31 +00003205 } else {
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003206#if HAS_KEYWORDS
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003207# if ENABLE_HUSH_CASE
Denis Vlasenko757361f2008-07-14 08:26:47 +00003208 if (ctx->ctx_dsemicolon
3209 && strcmp(word->data, "esac") != 0 /* not "... pattern) cmd;; esac" */
3210 ) {
Denis Vlasenko395ae452008-07-14 06:29:38 +00003211 /* already done when ctx_dsemicolon was set to 1: */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003212 /* ctx->ctx_res_w = RES_MATCH; */
3213 ctx->ctx_dsemicolon = 0;
3214 } else
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003215# endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003216 if (!command->argv /* if it's the first word... */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003217# if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003218 && ctx->ctx_res_w != RES_FOR /* ...not after FOR or IN */
3219 && ctx->ctx_res_w != RES_IN
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003220# endif
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003221# if ENABLE_HUSH_CASE
3222 && ctx->ctx_res_w != RES_CASE
3223# endif
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003224 ) {
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003225 int reserved = reserved_word(word, ctx);
3226 debug_printf_parse("checking for reserved-ness: %d\n", reserved);
3227 if (reserved) {
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003228 o_reset_to_empty_unquoted(word);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003229 debug_printf_parse("done_word return %d\n",
3230 (ctx->ctx_res_w == RES_SNTX));
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003231 return (ctx->ctx_res_w == RES_SNTX);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003232 }
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003233# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003234 if (strcmp(word->data, "[[") == 0) {
3235 command->cmd_type = CMD_SINGLEWORD_NOGLOB;
3236 }
3237 /* fall through */
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003238# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003239 }
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003240#endif
Denis Vlasenkobb929512009-04-16 10:59:40 +00003241 if (command->group) {
3242 /* "{ echo foo; } echo bar" - bad */
3243 syntax_error_at(word->data);
3244 debug_printf_parse("done_word return 1: syntax error, "
3245 "groups and arglists don't mix\n");
3246 return 1;
3247 }
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003248
3249 /* If this word wasn't an assignment, next ones definitely
3250 * can't be assignments. Even if they look like ones. */
3251 if (word->o_assignment != DEFINITELY_ASSIGNMENT
3252 && word->o_assignment != WORD_IS_KEYWORD
3253 ) {
3254 word->o_assignment = NOT_ASSIGNMENT;
3255 } else {
3256 if (word->o_assignment == DEFINITELY_ASSIGNMENT) {
3257 command->assignment_cnt++;
3258 debug_printf_parse("++assignment_cnt=%d\n", command->assignment_cnt);
3259 }
3260 debug_printf_parse("word->o_assignment was:'%s'\n", assignment_flag[word->o_assignment]);
3261 word->o_assignment = MAYBE_ASSIGNMENT;
3262 }
3263 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
3264
Denys Vlasenko38292b62010-09-05 14:49:40 +02003265 if (word->has_quoted_part
Denis Vlasenko55789c62008-06-18 16:30:42 +00003266 /* optimization: and if it's ("" or '') or ($v... or `cmd`...): */
3267 && (word->data[0] == '\0' || word->data[0] == SPECIAL_VAR_SYMBOL)
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003268 /* (otherwise it's known to be not empty and is already safe) */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003269 ) {
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003270 /* exclude "$@" - it can expand to no word despite "" */
Denis Vlasenkoafdcd122008-07-05 17:40:04 +00003271 char *p = word->data;
3272 while (p[0] == SPECIAL_VAR_SYMBOL
3273 && (p[1] & 0x7f) == '@'
3274 && p[2] == SPECIAL_VAR_SYMBOL
3275 ) {
3276 p += 3;
3277 }
Denis Vlasenkoc1c63b62008-06-18 09:20:35 +00003278 }
Denis Vlasenko22d10a02008-10-13 08:53:43 +00003279 command->argv = add_string_to_strings(command->argv, xstrdup(word->data));
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003280 debug_print_strings("word appended to argv", command->argv);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003281 }
Eric Andersen25f27032001-04-26 23:22:31 +00003282
Denis Vlasenko06810332007-05-21 23:30:54 +00003283#if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003284 if (ctx->ctx_res_w == RES_FOR) {
Denys Vlasenko38292b62010-09-05 14:49:40 +02003285 if (word->has_quoted_part
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003286 || !is_well_formed_var_name(command->argv[0], '\0')
3287 ) {
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003288 /* bash says just "not a valid identifier" */
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003289 syntax_error("not a valid identifier in for");
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003290 return 1;
3291 }
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003292 /* Force FOR to have just one word (variable name) */
3293 /* NB: basically, this makes hush see "for v in ..."
3294 * syntax as if it is "for v; in ...". FOR and IN become
3295 * two pipe structs in parse tree. */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00003296 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003297 }
Denis Vlasenko06810332007-05-21 23:30:54 +00003298#endif
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003299#if ENABLE_HUSH_CASE
3300 /* Force CASE to have just one word */
3301 if (ctx->ctx_res_w == RES_CASE) {
3302 done_pipe(ctx, PIPE_SEQ);
3303 }
3304#endif
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003305
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003306 o_reset_to_empty_unquoted(word);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003307
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003308 debug_printf_parse("done_word return 0\n");
Eric Andersen25f27032001-04-26 23:22:31 +00003309 return 0;
3310}
3311
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003312
3313/* Peek ahead in the input to find out if we have a "&n" construct,
3314 * as in "2>&1", that represents duplicating a file descriptor.
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003315 * Return:
3316 * REDIRFD_CLOSE if >&- "close fd" construct is seen,
3317 * REDIRFD_SYNTAX_ERR if syntax error,
3318 * REDIRFD_TO_FILE if no & was seen,
3319 * or the number found.
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003320 */
3321#if BB_MMU
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003322#define parse_redir_right_fd(as_string, input) \
3323 parse_redir_right_fd(input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003324#endif
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003325static int parse_redir_right_fd(o_string *as_string, struct in_str *input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003326{
3327 int ch, d, ok;
3328
3329 ch = i_peek(input);
3330 if (ch != '&')
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003331 return REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003332
3333 ch = i_getch(input); /* get the & */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003334 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003335 ch = i_peek(input);
3336 if (ch == '-') {
3337 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003338 nommu_addchr(as_string, ch);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003339 return REDIRFD_CLOSE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003340 }
3341 d = 0;
3342 ok = 0;
3343 while (ch != EOF && isdigit(ch)) {
3344 d = d*10 + (ch-'0');
3345 ok = 1;
3346 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003347 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003348 ch = i_peek(input);
3349 }
3350 if (ok) return d;
3351
3352//TODO: this is the place to catch ">&file" bashism (redirect both fd 1 and 2)
3353
3354 bb_error_msg("ambiguous redirect");
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003355 return REDIRFD_SYNTAX_ERR;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003356}
3357
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003358/* Return code is 0 normal, 1 if a syntax error is detected
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003359 */
3360static int parse_redirect(struct parse_context *ctx,
3361 int fd,
3362 redir_type style,
3363 struct in_str *input)
3364{
3365 struct command *command = ctx->command;
3366 struct redir_struct *redir;
3367 struct redir_struct **redirp;
3368 int dup_num;
3369
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003370 dup_num = REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003371 if (style != REDIRECT_HEREDOC) {
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003372 /* Check for a '>&1' type redirect */
3373 dup_num = parse_redir_right_fd(&ctx->as_string, input);
3374 if (dup_num == REDIRFD_SYNTAX_ERR)
3375 return 1;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003376 } else {
3377 int ch = i_peek(input);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003378 dup_num = (ch == '-'); /* HEREDOC_SKIPTABS bit is 1 */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003379 if (dup_num) { /* <<-... */
3380 ch = i_getch(input);
3381 nommu_addchr(&ctx->as_string, ch);
3382 ch = i_peek(input);
3383 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003384 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003385
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003386 if (style == REDIRECT_OVERWRITE && dup_num == REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003387 int ch = i_peek(input);
3388 if (ch == '|') {
3389 /* >|FILE redirect ("clobbering" >).
3390 * Since we do not support "set -o noclobber" yet,
3391 * >| and > are the same for now. Just eat |.
3392 */
3393 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003394 nommu_addchr(&ctx->as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003395 }
3396 }
3397
3398 /* Create a new redir_struct and append it to the linked list */
3399 redirp = &command->redirects;
3400 while ((redir = *redirp) != NULL) {
3401 redirp = &(redir->next);
3402 }
3403 *redirp = redir = xzalloc(sizeof(*redir));
3404 /* redir->next = NULL; */
3405 /* redir->rd_filename = NULL; */
3406 redir->rd_type = style;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003407 redir->rd_fd = (fd == -1) ? redir_table[style].default_fd : fd;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003408
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003409 debug_printf_parse("redirect type %d %s\n", redir->rd_fd,
3410 redir_table[style].descrip);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003411
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003412 redir->rd_dup = dup_num;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003413 if (style != REDIRECT_HEREDOC && dup_num != REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003414 /* Erik had a check here that the file descriptor in question
3415 * is legit; I postpone that to "run time"
3416 * A "-" representation of "close me" shows up as a -3 here */
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003417 debug_printf_parse("duplicating redirect '%d>&%d'\n",
3418 redir->rd_fd, redir->rd_dup);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003419 } else {
3420 /* Set ctx->pending_redirect, so we know what to do at the
3421 * end of the next parsed word. */
3422 ctx->pending_redirect = redir;
3423 }
3424 return 0;
3425}
3426
Eric Andersen25f27032001-04-26 23:22:31 +00003427/* If a redirect is immediately preceded by a number, that number is
3428 * supposed to tell which file descriptor to redirect. This routine
3429 * looks for such preceding numbers. In an ideal world this routine
3430 * needs to handle all the following classes of redirects...
3431 * echo 2>foo # redirects fd 2 to file "foo", nothing passed to echo
3432 * echo 49>foo # redirects fd 49 to file "foo", nothing passed to echo
3433 * echo -2>foo # redirects fd 1 to file "foo", "-2" passed to echo
3434 * echo 49x>foo # redirects fd 1 to file "foo", "49x" passed to echo
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003435 *
3436 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3437 * "2.7 Redirection
3438 * ... If n is quoted, the number shall not be recognized as part of
3439 * the redirection expression. For example:
3440 * echo \2>a
3441 * writes the character 2 into file a"
Denys Vlasenko38292b62010-09-05 14:49:40 +02003442 * We are getting it right by setting ->has_quoted_part on any \<char>
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003443 *
3444 * A -1 return means no valid number was found,
3445 * the caller should use the appropriate default for this redirection.
Eric Andersen25f27032001-04-26 23:22:31 +00003446 */
3447static int redirect_opt_num(o_string *o)
3448{
3449 int num;
3450
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003451 if (o->data == NULL)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00003452 return -1;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003453 num = bb_strtou(o->data, NULL, 10);
3454 if (errno || num < 0)
3455 return -1;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003456 o_reset_to_empty_unquoted(o);
Eric Andersen25f27032001-04-26 23:22:31 +00003457 return num;
3458}
3459
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003460#if BB_MMU
3461#define fetch_till_str(as_string, input, word, skip_tabs) \
3462 fetch_till_str(input, word, skip_tabs)
3463#endif
3464static char *fetch_till_str(o_string *as_string,
3465 struct in_str *input,
3466 const char *word,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003467 int heredoc_flags)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003468{
3469 o_string heredoc = NULL_O_STRING;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003470 unsigned past_EOL;
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003471 int prev = 0; /* not \ */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003472 int ch;
3473
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003474 goto jump_in;
Denys Vlasenkob8709032011-05-08 21:20:01 +02003475
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003476 while (1) {
3477 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003478 if (ch != EOF)
3479 nommu_addchr(as_string, ch);
3480 if ((ch == '\n' || ch == EOF)
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003481 && ((heredoc_flags & HEREDOC_QUOTED) || prev != '\\')
3482 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003483 if (strcmp(heredoc.data + past_EOL, word) == 0) {
3484 heredoc.data[past_EOL] = '\0';
3485 debug_printf_parse("parsed heredoc '%s'\n", heredoc.data);
3486 return heredoc.data;
3487 }
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003488 while (ch == '\n') {
3489 o_addchr(&heredoc, ch);
3490 prev = ch;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003491 jump_in:
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003492 past_EOL = heredoc.length;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003493 do {
3494 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003495 if (ch != EOF)
3496 nommu_addchr(as_string, ch);
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003497 } while ((heredoc_flags & HEREDOC_SKIPTABS) && ch == '\t');
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003498 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003499 }
3500 if (ch == EOF) {
3501 o_free_unsafe(&heredoc);
3502 return NULL;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003503 }
3504 o_addchr(&heredoc, ch);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003505 nommu_addchr(as_string, ch);
Denys Vlasenkoc3adfac2010-09-06 11:46:03 +02003506 if (prev == '\\' && ch == '\\')
3507 /* Correctly handle foo\\<eol> (not a line cont.) */
3508 prev = 0; /* not \ */
3509 else
3510 prev = ch;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003511 }
3512}
3513
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003514/* Look at entire parse tree for not-yet-loaded REDIRECT_HEREDOCs
3515 * and load them all. There should be exactly heredoc_cnt of them.
3516 */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003517static int fetch_heredocs(int heredoc_cnt, struct parse_context *ctx, struct in_str *input)
3518{
3519 struct pipe *pi = ctx->list_head;
3520
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003521 while (pi && heredoc_cnt) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003522 int i;
3523 struct command *cmd = pi->cmds;
3524
3525 debug_printf_parse("fetch_heredocs: num_cmds:%d cmd argv0:'%s'\n",
3526 pi->num_cmds,
3527 cmd->argv ? cmd->argv[0] : "NONE");
3528 for (i = 0; i < pi->num_cmds; i++) {
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003529 struct redir_struct *redir = cmd->redirects;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003530
3531 debug_printf_parse("fetch_heredocs: %d cmd argv0:'%s'\n",
3532 i, cmd->argv ? cmd->argv[0] : "NONE");
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003533 while (redir) {
3534 if (redir->rd_type == REDIRECT_HEREDOC) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003535 char *p;
3536
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003537 redir->rd_type = REDIRECT_HEREDOC2;
Denys Vlasenko764b2f02009-06-07 16:05:04 +02003538 /* redir->rd_dup is (ab)used to indicate <<- */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003539 p = fetch_till_str(&ctx->as_string, input,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003540 redir->rd_filename, redir->rd_dup);
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003541 if (!p) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003542 syntax_error("unexpected EOF in here document");
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003543 return 1;
3544 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003545 free(redir->rd_filename);
3546 redir->rd_filename = p;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003547 heredoc_cnt--;
3548 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003549 redir = redir->next;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003550 }
3551 cmd++;
3552 }
3553 pi = pi->next;
3554 }
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003555#if 0
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003556 /* Should be 0. If it isn't, it's a parse error */
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003557 if (heredoc_cnt)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003558 bb_error_msg_and_die("heredoc BUG 2");
3559#endif
3560 return 0;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003561}
3562
3563
Denys Vlasenkob36abf22010-09-05 14:50:59 +02003564static int run_list(struct pipe *pi);
3565#if BB_MMU
3566#define parse_stream(pstring, input, end_trigger) \
3567 parse_stream(input, end_trigger)
3568#endif
3569static struct pipe *parse_stream(char **pstring,
3570 struct in_str *input,
3571 int end_trigger);
Denis Vlasenkoba7cf262007-05-25 14:34:30 +00003572
Eric Andersen25f27032001-04-26 23:22:31 +00003573
Denys Vlasenkoc2704542009-11-20 19:14:19 +01003574#if !ENABLE_HUSH_FUNCTIONS
3575#define parse_group(dest, ctx, input, ch) \
3576 parse_group(ctx, input, ch)
3577#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003578static int parse_group(o_string *dest, struct parse_context *ctx,
Eric Andersen25f27032001-04-26 23:22:31 +00003579 struct in_str *input, int ch)
3580{
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003581 /* dest contains characters seen prior to ( or {.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00003582 * Typically it's empty, but for function defs,
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003583 * it contains function name (without '()'). */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003584 struct pipe *pipe_list;
Denis Vlasenko240c2552009-04-03 03:45:05 +00003585 int endch;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003586 struct command *command = ctx->command;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003587
3588 debug_printf_parse("parse_group entered\n");
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003589#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenko38292b62010-09-05 14:49:40 +02003590 if (ch == '(' && !dest->has_quoted_part) {
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003591 if (dest->length)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003592 if (done_word(dest, ctx))
3593 return 1;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003594 if (!command->argv)
3595 goto skip; /* (... */
3596 if (command->argv[1]) { /* word word ... (... */
3597 syntax_error_unexpected_ch('(');
3598 return 1;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003599 }
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003600 /* it is "word(..." or "word (..." */
3601 do
3602 ch = i_getch(input);
3603 while (ch == ' ' || ch == '\t');
3604 if (ch != ')') {
3605 syntax_error_unexpected_ch(ch);
3606 return 1;
3607 }
3608 nommu_addchr(&ctx->as_string, ch);
3609 do
3610 ch = i_getch(input);
3611 while (ch == ' ' || ch == '\t' || ch == '\n');
3612 if (ch != '{') {
3613 syntax_error_unexpected_ch(ch);
3614 return 1;
3615 }
3616 nommu_addchr(&ctx->as_string, ch);
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003617 command->cmd_type = CMD_FUNCDEF;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003618 goto skip;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003619 }
3620#endif
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003621
3622#if 0 /* Prevented by caller */
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003623 if (command->argv /* word [word]{... */
3624 || dest->length /* word{... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003625 || dest->has_quoted_part /* ""{... */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003626 ) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003627 syntax_error(NULL);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003628 debug_printf_parse("parse_group return 1: "
3629 "syntax error, groups and arglists don't mix\n");
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003630 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003631 }
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003632#endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003633
3634#if ENABLE_HUSH_FUNCTIONS
3635 skip:
3636#endif
Denis Vlasenko240c2552009-04-03 03:45:05 +00003637 endch = '}';
Denis Vlasenko90e485c2007-05-23 15:22:50 +00003638 if (ch == '(') {
Denis Vlasenko240c2552009-04-03 03:45:05 +00003639 endch = ')';
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003640 command->cmd_type = CMD_SUBSHELL;
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00003641 } else {
3642 /* bash does not allow "{echo...", requires whitespace */
3643 ch = i_getch(input);
3644 if (ch != ' ' && ch != '\t' && ch != '\n') {
3645 syntax_error_unexpected_ch(ch);
3646 return 1;
3647 }
3648 nommu_addchr(&ctx->as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003649 }
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003650
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003651 {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003652#if BB_MMU
3653# define as_string NULL
3654#else
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003655 char *as_string = NULL;
3656#endif
3657 pipe_list = parse_stream(&as_string, input, endch);
3658#if !BB_MMU
3659 if (as_string)
3660 o_addstr(&ctx->as_string, as_string);
3661#endif
3662 /* empty ()/{} or parse error? */
3663 if (!pipe_list || pipe_list == ERR_PTR) {
Denis Vlasenkobb929512009-04-16 10:59:40 +00003664 /* parse_stream already emitted error msg */
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003665 if (!BB_MMU)
3666 free(as_string);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003667 debug_printf_parse("parse_group return 1: "
3668 "parse_stream returned %p\n", pipe_list);
3669 return 1;
3670 }
3671 command->group = pipe_list;
3672#if !BB_MMU
3673 as_string[strlen(as_string) - 1] = '\0'; /* plink ')' or '}' */
3674 command->group_as_string = as_string;
3675 debug_printf_parse("end of group, remembering as:'%s'\n",
3676 command->group_as_string);
3677#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003678#undef as_string
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00003679 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003680 debug_printf_parse("parse_group return 0\n");
3681 return 0;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003682 /* command remains "open", available for possible redirects */
Eric Andersen25f27032001-04-26 23:22:31 +00003683}
3684
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003685#if ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003686/* Subroutines for copying $(...) and `...` things */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003687static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003688/* '...' */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003689static int add_till_single_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003690{
3691 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003692 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003693 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003694 syntax_error_unterm_ch('\'');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003695 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003696 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003697 if (ch == '\'')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003698 return 1;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003699 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003700 }
3701}
3702/* "...\"...`..`...." - do we need to handle "...$(..)..." too? */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003703static int add_till_double_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003704{
3705 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003706 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003707 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003708 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003709 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003710 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003711 if (ch == '"')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003712 return 1;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003713 if (ch == '\\') { /* \x. Copy both chars. */
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003714 o_addchr(dest, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003715 ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003716 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003717 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003718 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003719 if (!add_till_backquote(dest, input, /*in_dquote:*/ 1))
3720 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003721 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003722 continue;
3723 }
Denis Vlasenko5703c222008-06-15 11:49:42 +00003724 //if (ch == '$') ...
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003725 }
3726}
3727/* Process `cmd` - copy contents until "`" is seen. Complicated by
3728 * \` quoting.
3729 * "Within the backquoted style of command substitution, backslash
3730 * shall retain its literal meaning, except when followed by: '$', '`', or '\'.
3731 * The search for the matching backquote shall be satisfied by the first
3732 * backquote found without a preceding backslash; during this search,
3733 * if a non-escaped backquote is encountered within a shell comment,
3734 * a here-document, an embedded command substitution of the $(command)
3735 * form, or a quoted string, undefined results occur. A single-quoted
3736 * or double-quoted string that begins, but does not end, within the
3737 * "`...`" sequence produces undefined results."
3738 * Example Output
3739 * echo `echo '\'TEST\`echo ZZ\`BEST` \TESTZZBEST
3740 */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003741static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003742{
3743 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003744 int ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003745 if (ch == '`')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003746 return 1;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003747 if (ch == '\\') {
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003748 /* \x. Copy both unless it is \`, \$, \\ and maybe \" */
3749 ch = i_getch(input);
3750 if (ch != '`'
3751 && ch != '$'
3752 && ch != '\\'
3753 && (!in_dquote || ch != '"')
3754 ) {
3755 o_addchr(dest, '\\');
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003756 }
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003757 }
3758 if (ch == EOF) {
3759 syntax_error_unterm_ch('`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003760 return 0;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003761 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003762 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003763 }
3764}
3765/* Process $(cmd) - copy contents until ")" is seen. Complicated by
3766 * quoting and nested ()s.
3767 * "With the $(command) style of command substitution, all characters
3768 * following the open parenthesis to the matching closing parenthesis
3769 * constitute the command. Any valid shell script can be used for command,
3770 * except a script consisting solely of redirections which produces
3771 * unspecified results."
3772 * Example Output
3773 * echo $(echo '(TEST)' BEST) (TEST) BEST
3774 * echo $(echo 'TEST)' BEST) TEST) BEST
3775 * echo $(echo \(\(TEST\) BEST) ((TEST) BEST
Denys Vlasenko74369502010-05-21 19:52:01 +02003776 *
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003777 * Also adapted to eat ${var%...} and $((...)) constructs, since ... part
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003778 * can contain arbitrary constructs, just like $(cmd).
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003779 * In bash compat mode, it needs to also be able to stop on ':' or '/'
3780 * for ${var:N[:M]} and ${var/P[/R]} parsing.
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003781 */
Denys Vlasenko74369502010-05-21 19:52:01 +02003782#define DOUBLE_CLOSE_CHAR_FLAG 0x80
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003783static int add_till_closing_bracket(o_string *dest, struct in_str *input, unsigned end_ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003784{
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003785 int ch;
Denys Vlasenko74369502010-05-21 19:52:01 +02003786 char dbl = end_ch & DOUBLE_CLOSE_CHAR_FLAG;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003787# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003788 char end_char2 = end_ch >> 8;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003789# endif
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003790 end_ch &= (DOUBLE_CLOSE_CHAR_FLAG - 1);
3791
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003792 while (1) {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003793 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003794 if (ch == EOF) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003795 syntax_error_unterm_ch(end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003796 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003797 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003798 if (ch == end_ch IF_HUSH_BASH_COMPAT( || ch == end_char2)) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003799 if (!dbl)
3800 break;
3801 /* we look for closing )) of $((EXPR)) */
3802 if (i_peek(input) == end_ch) {
3803 i_getch(input); /* eat second ')' */
3804 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003805 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003806 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003807 o_addchr(dest, ch);
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003808 if (ch == '(' || ch == '{') {
3809 ch = (ch == '(' ? ')' : '}');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003810 if (!add_till_closing_bracket(dest, input, ch))
3811 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003812 o_addchr(dest, ch);
3813 continue;
3814 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003815 if (ch == '\'') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003816 if (!add_till_single_quote(dest, input))
3817 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003818 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003819 continue;
3820 }
3821 if (ch == '"') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003822 if (!add_till_double_quote(dest, input))
3823 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003824 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003825 continue;
3826 }
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003827 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003828 if (!add_till_backquote(dest, input, /*in_dquote:*/ 0))
3829 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003830 o_addchr(dest, ch);
3831 continue;
3832 }
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003833 if (ch == '\\') {
3834 /* \x. Copy verbatim. Important for \(, \) */
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003835 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003836 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003837 syntax_error_unterm_ch(')');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003838 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003839 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003840 o_addchr(dest, ch);
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003841 continue;
3842 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003843 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003844 return ch;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003845}
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003846#endif /* ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS */
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003847
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00003848/* Return code: 0 for OK, 1 for syntax error */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003849#if BB_MMU
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003850#define parse_dollar(as_string, dest, input, quote_mask) \
3851 parse_dollar(dest, input, quote_mask)
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003852#define as_string NULL
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003853#endif
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003854static int parse_dollar(o_string *as_string,
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003855 o_string *dest,
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003856 struct in_str *input, unsigned char quote_mask)
Eric Andersen25f27032001-04-26 23:22:31 +00003857{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003858 int ch = i_peek(input); /* first character after the $ */
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003859
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003860 debug_printf_parse("parse_dollar entered: ch='%c'\n", ch);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003861 if (isalpha(ch)) {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003862 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003863 nommu_addchr(as_string, ch);
Denis Vlasenkod4981312008-07-31 10:34:48 +00003864 make_var:
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003865 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003866 while (1) {
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003867 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003868 o_addchr(dest, ch | quote_mask);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003869 quote_mask = 0;
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003870 ch = i_peek(input);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003871 if (!isalnum(ch) && ch != '_')
3872 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003873 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003874 nommu_addchr(as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003875 }
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003876 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003877 } else if (isdigit(ch)) {
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003878 make_one_char_var:
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003879 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003880 nommu_addchr(as_string, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003881 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003882 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003883 o_addchr(dest, ch | quote_mask);
3884 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003885 } else switch (ch) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003886 case '$': /* pid */
3887 case '!': /* last bg pid */
3888 case '?': /* last exit code */
3889 case '#': /* number of args */
3890 case '*': /* args */
3891 case '@': /* args */
3892 goto make_one_char_var;
3893 case '{': {
Mike Frysingeref3e7fd2009-06-01 14:13:39 -04003894 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3895
Denys Vlasenko74369502010-05-21 19:52:01 +02003896 ch = i_getch(input); /* eat '{' */
3897 nommu_addchr(as_string, ch);
3898
3899 ch = i_getch(input); /* first char after '{' */
Denys Vlasenko74369502010-05-21 19:52:01 +02003900 /* It should be ${?}, or ${#var},
3901 * or even ${?+subst} - operator acting on a special variable,
3902 * or the beginning of variable name.
3903 */
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003904 if (ch == EOF
3905 || (!strchr(_SPECIAL_VARS_STR, ch) && !isalnum(ch)) /* not one of those */
3906 ) {
Denys Vlasenko74369502010-05-21 19:52:01 +02003907 bad_dollar_syntax:
3908 syntax_error_unterm_str("${name}");
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003909 debug_printf_parse("parse_dollar return 0: unterminated ${name}\n");
3910 return 0;
Denys Vlasenko74369502010-05-21 19:52:01 +02003911 }
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003912 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003913 ch |= quote_mask;
3914
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003915 /* It's possible to just call add_till_closing_bracket() at this point.
Denys Vlasenko74369502010-05-21 19:52:01 +02003916 * However, this regresses some of our testsuite cases
3917 * which check invalid constructs like ${%}.
3918 * Oh well... let's check that the var name part is fine... */
3919
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003920 while (1) {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003921 unsigned pos;
3922
Denys Vlasenko74369502010-05-21 19:52:01 +02003923 o_addchr(dest, ch);
3924 debug_printf_parse(": '%c'\n", ch);
3925
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003926 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003927 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003928 if (ch == '}')
Mike Frysinger98c52642009-04-02 10:02:37 +00003929 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003930
Denys Vlasenko74369502010-05-21 19:52:01 +02003931 if (!isalnum(ch) && ch != '_') {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003932 unsigned end_ch;
3933 unsigned char last_ch;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003934 /* handle parameter expansions
3935 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
3936 */
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003937 if (!strchr(VAR_SUBST_OPS, ch)) /* ${var<bad_char>... */
Denys Vlasenko74369502010-05-21 19:52:01 +02003938 goto bad_dollar_syntax;
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003939
3940 /* Eat everything until closing '}' (or ':') */
3941 end_ch = '}';
3942 if (ENABLE_HUSH_BASH_COMPAT
3943 && ch == ':'
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003944 && !strchr(MINUS_PLUS_EQUAL_QUESTION, i_peek(input))
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003945 ) {
3946 /* It's ${var:N[:M]} thing */
3947 end_ch = '}' * 0x100 + ':';
3948 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003949 if (ENABLE_HUSH_BASH_COMPAT
3950 && ch == '/'
3951 ) {
3952 /* It's ${var/[/]pattern[/repl]} thing */
3953 if (i_peek(input) == '/') { /* ${var//pattern[/repl]}? */
3954 i_getch(input);
3955 nommu_addchr(as_string, '/');
3956 ch = '\\';
3957 }
3958 end_ch = '}' * 0x100 + '/';
3959 }
3960 o_addchr(dest, ch);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003961 again:
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003962 if (!BB_MMU)
3963 pos = dest->length;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003964#if ENABLE_HUSH_DOLLAR_OPS
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003965 last_ch = add_till_closing_bracket(dest, input, end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003966 if (last_ch == 0) /* error? */
3967 return 0;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003968#else
3969#error Simple code to only allow ${var} is not implemented
3970#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003971 if (as_string) {
3972 o_addstr(as_string, dest->data + pos);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003973 o_addchr(as_string, last_ch);
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003974 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003975
3976 if (ENABLE_HUSH_BASH_COMPAT && (end_ch & 0xff00)) {
3977 /* close the first block: */
3978 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003979 /* while parsing N from ${var:N[:M]}
3980 * or pattern from ${var/[/]pattern[/repl]} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003981 if ((end_ch & 0xff) == last_ch) {
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003982 /* got ':' or '/'- parse the rest */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003983 end_ch = '}';
3984 goto again;
3985 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003986 /* got '}' */
3987 if (end_ch == '}' * 0x100 + ':') {
3988 /* it's ${var:N} - emulate :999999999 */
3989 o_addstr(dest, "999999999");
3990 } /* else: it's ${var/[/]pattern} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003991 }
Denys Vlasenko74369502010-05-21 19:52:01 +02003992 break;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003993 }
Denys Vlasenko74369502010-05-21 19:52:01 +02003994 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003995 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3996 break;
3997 }
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003998#if ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003999 case '(': {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004000 unsigned pos;
4001
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004002 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004003 nommu_addchr(as_string, ch);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004004# if ENABLE_SH_MATH_SUPPORT
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004005 if (i_peek(input) == '(') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004006 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004007 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004008 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4009 o_addchr(dest, /*quote_mask |*/ '+');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004010 if (!BB_MMU)
4011 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004012 if (!add_till_closing_bracket(dest, input, ')' | DOUBLE_CLOSE_CHAR_FLAG))
4013 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004014 if (as_string) {
4015 o_addstr(as_string, dest->data + pos);
4016 o_addchr(as_string, ')');
4017 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004018 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004019 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00004020 break;
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00004021 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004022# endif
4023# if ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004024 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4025 o_addchr(dest, quote_mask | '`');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004026 if (!BB_MMU)
4027 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004028 if (!add_till_closing_bracket(dest, input, ')'))
4029 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004030 if (as_string) {
4031 o_addstr(as_string, dest->data + pos);
Denys Vlasenkob70cef72010-01-12 13:45:45 +01004032 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004033 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004034 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004035# endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004036 break;
4037 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004038#endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004039 case '_':
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004040 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004041 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004042 ch = i_peek(input);
4043 if (isalnum(ch)) { /* it's $_name or $_123 */
4044 ch = '_';
4045 goto make_var;
4046 }
4047 /* else: it's $_ */
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02004048 /* TODO: $_ and $-: */
4049 /* $_ Shell or shell script name; or last argument of last command
4050 * (if last command wasn't a pipe; if it was, bash sets $_ to "");
4051 * but in command's env, set to full pathname used to invoke it */
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004052 /* $- Option flags set by set builtin or shell options (-i etc) */
4053 default:
4054 o_addQchr(dest, '$');
Eric Andersen25f27032001-04-26 23:22:31 +00004055 }
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004056 debug_printf_parse("parse_dollar return 1 (ok)\n");
4057 return 1;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004058#undef as_string
Eric Andersen25f27032001-04-26 23:22:31 +00004059}
4060
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004061#if BB_MMU
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004062# if ENABLE_HUSH_BASH_COMPAT
4063#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4064 encode_string(dest, input, dquote_end, process_bkslash)
4065# else
4066/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4067#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4068 encode_string(dest, input, dquote_end)
4069# endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004070#define as_string NULL
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004071
4072#else /* !MMU */
4073
4074# if ENABLE_HUSH_BASH_COMPAT
4075/* all parameters are needed, no macro tricks */
4076# else
4077#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4078 encode_string(as_string, dest, input, dquote_end)
4079# endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004080#endif
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004081static int encode_string(o_string *as_string,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004082 o_string *dest,
4083 struct in_str *input,
Denys Vlasenko14e289b2010-09-10 10:15:18 +02004084 int dquote_end,
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004085 int process_bkslash)
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004086{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004087#if !ENABLE_HUSH_BASH_COMPAT
4088 const int process_bkslash = 1;
4089#endif
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004090 int ch;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004091 int next;
4092
4093 again:
4094 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004095 if (ch != EOF)
4096 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004097 if (ch == dquote_end) { /* may be only '"' or EOF */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004098 debug_printf_parse("encode_string return 1 (ok)\n");
4099 return 1;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004100 }
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004101 /* note: can't move it above ch == dquote_end check! */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004102 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004103 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004104 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004105 }
4106 next = '\0';
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004107 if (ch != '\n') {
4108 next = i_peek(input);
4109 }
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004110 debug_printf_parse("\" ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004111 ch, ch, !!(dest->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004112 if (process_bkslash && ch == '\\') {
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004113 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004114 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004115 xfunc_die();
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004116 }
4117 /* bash:
4118 * "The backslash retains its special meaning [in "..."]
4119 * only when followed by one of the following characters:
4120 * $, `, ", \, or <newline>. A double quote may be quoted
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02004121 * within double quotes by preceding it with a backslash."
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004122 * NB: in (unquoted) heredoc, above does not apply to ",
4123 * therefore we check for it by "next == dquote_end" cond.
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004124 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004125 if (next == dquote_end || strchr("$`\\\n", next)) {
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004126 ch = i_getch(input); /* eat next */
4127 if (ch == '\n')
4128 goto again; /* skip \<newline> */
Denys Vlasenko4f870492010-09-10 11:06:01 +02004129 } /* else: ch remains == '\\', and we double it below: */
4130 o_addqchr(dest, ch); /* \c if c is a glob char, else just c */
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004131 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004132 goto again;
4133 }
4134 if (ch == '$') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004135 if (!parse_dollar(as_string, dest, input, /*quote_mask:*/ 0x80)) {
4136 debug_printf_parse("encode_string return 0: "
4137 "parse_dollar returned 0 (error)\n");
4138 return 0;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004139 }
4140 goto again;
4141 }
4142#if ENABLE_HUSH_TICK
4143 if (ch == '`') {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004144 //unsigned pos = dest->length;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004145 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4146 o_addchr(dest, 0x80 | '`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004147 if (!add_till_backquote(dest, input, /*in_dquote:*/ dquote_end == '"'))
4148 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004149 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4150 //debug_printf_subst("SUBST RES3 '%s'\n", dest->data + pos);
Denis Vlasenkof328e002009-04-02 16:55:38 +00004151 goto again;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004152 }
4153#endif
Denis Vlasenkof328e002009-04-02 16:55:38 +00004154 o_addQchr(dest, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004155 goto again;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004156#undef as_string
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004157}
4158
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004159/*
4160 * Scan input until EOF or end_trigger char.
4161 * Return a list of pipes to execute, or NULL on EOF
4162 * or if end_trigger character is met.
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004163 * On syntax error, exit if shell is not interactive,
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004164 * reset parsing machinery and start parsing anew,
4165 * or return ERR_PTR.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004166 */
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004167static struct pipe *parse_stream(char **pstring,
4168 struct in_str *input,
4169 int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00004170{
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004171 struct parse_context ctx;
4172 o_string dest = NULL_O_STRING;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004173 int heredoc_cnt;
Eric Andersen25f27032001-04-26 23:22:31 +00004174
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004175 /* Single-quote triggers a bypass of the main loop until its mate is
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004176 * found. When recursing, quote state is passed in via dest->o_expflags.
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004177 */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004178 debug_printf_parse("parse_stream entered, end_trigger='%c'\n",
Denys Vlasenko90a99042009-09-06 02:36:23 +02004179 end_trigger ? end_trigger : 'X');
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004180 debug_enter();
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004181
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004182 /* If very first arg is "" or '', dest.data may end up NULL.
4183 * Preventing this: */
4184 o_addchr(&dest, '\0');
4185 dest.length = 0;
4186
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004187 /* We used to separate words on $IFS here. This was wrong.
4188 * $IFS is used only for word splitting when $var is expanded,
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004189 * here we should use blank chars as separators, not $IFS
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004190 */
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004191
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004192 if (MAYBE_ASSIGNMENT != 0)
4193 dest.o_assignment = MAYBE_ASSIGNMENT;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004194 initialize_context(&ctx);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004195 heredoc_cnt = 0;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004196 while (1) {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004197 const char *is_blank;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004198 const char *is_special;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004199 int ch;
4200 int next;
4201 int redir_fd;
4202 redir_type redir_style;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004203
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00004204 ch = i_getch(input);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004205 debug_printf_parse(": ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004206 ch, ch, !!(dest.o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004207 if (ch == EOF) {
4208 struct pipe *pi;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004209
4210 if (heredoc_cnt) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004211 syntax_error_unterm_str("here document");
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004212 goto parse_error;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004213 }
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004214 /* end_trigger == '}' case errors out earlier,
4215 * checking only ')' */
4216 if (end_trigger == ')') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004217 syntax_error_unterm_ch('(');
4218 goto parse_error;
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004219 }
4220
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004221 if (done_word(&dest, &ctx)) {
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004222 goto parse_error;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004223 }
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004224 o_free(&dest);
4225 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004226 pi = ctx.list_head;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004227 /* If we got nothing... */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004228 /* (this makes bare "&" cmd a no-op.
4229 * bash says: "syntax error near unexpected token '&'") */
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004230 if (pi->num_cmds == 0
Tanguy Pruvot823694d2012-11-18 13:20:29 +01004231 IF_HAS_KEYWORDS(&& pi->res_word == RES_NONE)
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004232 ) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004233 free_pipe_list(pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004234 pi = NULL;
4235 }
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004236#if !BB_MMU
4237 debug_printf_parse("as_string '%s'\n", ctx.as_string.data);
4238 if (pstring)
4239 *pstring = ctx.as_string.data;
4240 else
4241 o_free_unsafe(&ctx.as_string);
4242#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004243 debug_leave();
4244 debug_printf_parse("parse_stream return %p\n", pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004245 return pi;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004246 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004247 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004248
4249 next = '\0';
4250 if (ch != '\n')
4251 next = i_peek(input);
4252
4253 is_special = "{}<>;&|()#'" /* special outside of "str" */
4254 "\\$\"" IF_HUSH_TICK("`"); /* always special */
4255 /* Are { and } special here? */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004256 if (ctx.command->argv /* word [word]{... - non-special */
4257 || dest.length /* word{... - non-special */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004258 || dest.has_quoted_part /* ""{... - non-special */
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004259 || (next != ';' /* }; - special */
4260 && next != ')' /* }) - special */
4261 && next != '&' /* }& and }&& ... - special */
4262 && next != '|' /* }|| ... - special */
4263 && !strchr(defifs, next) /* {word - non-special */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004264 )
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004265 ) {
4266 /* They are not special, skip "{}" */
4267 is_special += 2;
4268 }
4269 is_special = strchr(is_special, ch);
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004270 is_blank = strchr(defifs, ch);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004271
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004272 if (!is_special && !is_blank) { /* ordinary char */
Denis Vlasenkobf25fbc2009-04-19 13:57:51 +00004273 ordinary_char:
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004274 o_addQchr(&dest, ch);
4275 if ((dest.o_assignment == MAYBE_ASSIGNMENT
4276 || dest.o_assignment == WORD_IS_KEYWORD)
Denis Vlasenko55789c62008-06-18 16:30:42 +00004277 && ch == '='
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004278 && is_well_formed_var_name(dest.data, '=')
Denis Vlasenko55789c62008-06-18 16:30:42 +00004279 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004280 dest.o_assignment = DEFINITELY_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004281 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko55789c62008-06-18 16:30:42 +00004282 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004283 continue;
4284 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004285
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004286 if (is_blank) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004287 if (done_word(&dest, &ctx)) {
4288 goto parse_error;
Eric Andersenaac75e52001-04-30 18:18:45 +00004289 }
Denis Vlasenko37181682009-04-03 03:19:15 +00004290 if (ch == '\n') {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004291 /* Is this a case when newline is simply ignored?
4292 * Some examples:
4293 * "cmd | <newline> cmd ..."
4294 * "case ... in <newline> word) ..."
4295 */
4296 if (IS_NULL_CMD(ctx.command)
4297 && dest.length == 0 && !dest.has_quoted_part
Denis Vlasenkof1736072008-07-31 10:09:26 +00004298 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004299 /* This newline can be ignored. But...
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004300 * Without check #1, interactive shell
4301 * ignores even bare <newline>,
4302 * and shows the continuation prompt:
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004303 * ps1_prompt$ <enter>
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004304 * ps2> _ <=== wrong, should be ps1
4305 * Without check #2, "cmd & <newline>"
4306 * is similarly mistreated.
4307 * (BTW, this makes "cmd & cmd"
4308 * and "cmd && cmd" non-orthogonal.
4309 * Really, ask yourself, why
4310 * "cmd && <newline>" doesn't start
4311 * cmd but waits for more input?
4312 * No reason...)
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004313 */
4314 struct pipe *pi = ctx.list_head;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004315 if (pi->num_cmds != 0 /* check #1 */
4316 && pi->followup != PIPE_BG /* check #2 */
4317 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004318 continue;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004319 }
Denis Vlasenkof1736072008-07-31 10:09:26 +00004320 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004321 /* Treat newline as a command separator. */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004322 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004323 debug_printf_parse("heredoc_cnt:%d\n", heredoc_cnt);
4324 if (heredoc_cnt) {
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004325 if (fetch_heredocs(heredoc_cnt, &ctx, input)) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004326 goto parse_error;
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004327 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004328 heredoc_cnt = 0;
4329 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004330 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004331 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004332 ch = ';';
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004333 /* note: if (is_blank) continue;
Denis Vlasenko240c2552009-04-03 03:45:05 +00004334 * will still trigger for us */
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004335 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004336 }
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004337
4338 /* "cmd}" or "cmd }..." without semicolon or &:
4339 * } is an ordinary char in this case, even inside { cmd; }
4340 * Pathological example: { ""}; } should exec "}" cmd
4341 */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004342 if (ch == '}') {
4343 if (!IS_NULL_CMD(ctx.command) /* cmd } */
4344 || dest.length != 0 /* word} */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004345 || dest.has_quoted_part /* ""} */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004346 ) {
4347 goto ordinary_char;
4348 }
4349 if (!IS_NULL_PIPE(ctx.pipe)) /* cmd | } */
4350 goto skip_end_trigger;
4351 /* else: } does terminate a group */
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004352 }
4353
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004354 if (end_trigger && end_trigger == ch
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004355 && (ch != ';' || heredoc_cnt == 0)
4356#if ENABLE_HUSH_CASE
4357 && (ch != ')'
4358 || ctx.ctx_res_w != RES_MATCH
Denys Vlasenko38292b62010-09-05 14:49:40 +02004359 || (!dest.has_quoted_part && strcmp(dest.data, "esac") == 0)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004360 )
4361#endif
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004362 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004363 if (heredoc_cnt) {
4364 /* This is technically valid:
4365 * { cat <<HERE; }; echo Ok
4366 * heredoc
4367 * heredoc
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004368 * HERE
4369 * but we don't support this.
4370 * We require heredoc to be in enclosing {}/(),
4371 * if any.
4372 */
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004373 syntax_error_unterm_str("here document");
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004374 goto parse_error;
4375 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004376 if (done_word(&dest, &ctx)) {
4377 goto parse_error;
4378 }
4379 done_pipe(&ctx, PIPE_SEQ);
4380 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004381 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004382 /* Do we sit outside of any if's, loops or case's? */
Denis Vlasenko37181682009-04-03 03:19:15 +00004383 if (!HAS_KEYWORDS
Tanguy Pruvot823694d2012-11-18 13:20:29 +01004384 IF_HAS_KEYWORDS(|| (ctx.ctx_res_w == RES_NONE && ctx.old_flag == 0))
Denis Vlasenko37181682009-04-03 03:19:15 +00004385 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004386 o_free(&dest);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004387#if !BB_MMU
4388 debug_printf_parse("as_string '%s'\n", ctx.as_string.data);
4389 if (pstring)
4390 *pstring = ctx.as_string.data;
4391 else
4392 o_free_unsafe(&ctx.as_string);
4393#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004394 debug_leave();
4395 debug_printf_parse("parse_stream return %p: "
4396 "end_trigger char found\n",
4397 ctx.list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004398 return ctx.list_head;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004399 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004400 }
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004401 skip_end_trigger:
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004402 if (is_blank)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004403 continue;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004404
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004405 /* Catch <, > before deciding whether this word is
4406 * an assignment. a=1 2>z b=2: b=2 is still assignment */
4407 switch (ch) {
4408 case '>':
4409 redir_fd = redirect_opt_num(&dest);
4410 if (done_word(&dest, &ctx)) {
4411 goto parse_error;
4412 }
4413 redir_style = REDIRECT_OVERWRITE;
4414 if (next == '>') {
4415 redir_style = REDIRECT_APPEND;
4416 ch = i_getch(input);
4417 nommu_addchr(&ctx.as_string, ch);
4418 }
4419#if 0
4420 else if (next == '(') {
4421 syntax_error(">(process) not supported");
4422 goto parse_error;
4423 }
4424#endif
4425 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4426 goto parse_error;
4427 continue; /* back to top of while (1) */
4428 case '<':
4429 redir_fd = redirect_opt_num(&dest);
4430 if (done_word(&dest, &ctx)) {
4431 goto parse_error;
4432 }
4433 redir_style = REDIRECT_INPUT;
4434 if (next == '<') {
4435 redir_style = REDIRECT_HEREDOC;
4436 heredoc_cnt++;
4437 debug_printf_parse("++heredoc_cnt=%d\n", heredoc_cnt);
4438 ch = i_getch(input);
4439 nommu_addchr(&ctx.as_string, ch);
4440 } else if (next == '>') {
4441 redir_style = REDIRECT_IO;
4442 ch = i_getch(input);
4443 nommu_addchr(&ctx.as_string, ch);
4444 }
4445#if 0
4446 else if (next == '(') {
4447 syntax_error("<(process) not supported");
4448 goto parse_error;
4449 }
4450#endif
4451 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4452 goto parse_error;
4453 continue; /* back to top of while (1) */
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004454 case '#':
4455 if (dest.length == 0 && !dest.has_quoted_part) {
4456 /* skip "#comment" */
4457 while (1) {
4458 ch = i_peek(input);
4459 if (ch == EOF || ch == '\n')
4460 break;
4461 i_getch(input);
4462 /* note: we do not add it to &ctx.as_string */
4463 }
4464 nommu_addchr(&ctx.as_string, '\n');
4465 continue; /* back to top of while (1) */
4466 }
4467 break;
4468 case '\\':
4469 if (next == '\n') {
4470 /* It's "\<newline>" */
4471#if !BB_MMU
4472 /* Remove trailing '\' from ctx.as_string */
4473 ctx.as_string.data[--ctx.as_string.length] = '\0';
4474#endif
4475 ch = i_getch(input); /* eat it */
4476 continue; /* back to top of while (1) */
4477 }
4478 break;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004479 }
4480
4481 if (dest.o_assignment == MAYBE_ASSIGNMENT
4482 /* check that we are not in word in "a=1 2>word b=1": */
4483 && !ctx.pending_redirect
4484 ) {
4485 /* ch is a special char and thus this word
4486 * cannot be an assignment */
4487 dest.o_assignment = NOT_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004488 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004489 }
4490
Denys Vlasenkocbfe6ad2009-08-12 19:47:44 +02004491 /* Note: nommu_addchr(&ctx.as_string, ch) is already done */
4492
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004493 switch (ch) {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004494 case '#': /* non-comment #: "echo a#b" etc */
4495 o_addQchr(&dest, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004496 break;
4497 case '\\':
4498 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004499 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004500 xfunc_die();
Eric Andersen25f27032001-04-26 23:22:31 +00004501 }
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004502 ch = i_getch(input);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004503 /* note: ch != '\n' (that case does not reach this place) */
4504 o_addchr(&dest, '\\');
4505 /*nommu_addchr(&ctx.as_string, '\\'); - already done */
4506 o_addchr(&dest, ch);
4507 nommu_addchr(&ctx.as_string, ch);
4508 /* Example: echo Hello \2>file
4509 * we need to know that word 2 is quoted */
4510 dest.has_quoted_part = 1;
Eric Andersen25f27032001-04-26 23:22:31 +00004511 break;
4512 case '$':
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004513 if (!parse_dollar(&ctx.as_string, &dest, input, /*quote_mask:*/ 0)) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004514 debug_printf_parse("parse_stream parse error: "
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004515 "parse_dollar returned 0 (error)\n");
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004516 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004517 }
Eric Andersen25f27032001-04-26 23:22:31 +00004518 break;
4519 case '\'':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004520 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004521 if (next == '\'' && !ctx.pending_redirect) {
4522 insert_empty_quoted_str_marker:
4523 nommu_addchr(&ctx.as_string, next);
4524 i_getch(input); /* eat second ' */
4525 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4526 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4527 } else {
4528 while (1) {
4529 ch = i_getch(input);
4530 if (ch == EOF) {
4531 syntax_error_unterm_ch('\'');
4532 goto parse_error;
4533 }
4534 nommu_addchr(&ctx.as_string, ch);
4535 if (ch == '\'')
4536 break;
4537 o_addqchr(&dest, ch);
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004538 }
Eric Andersen25f27032001-04-26 23:22:31 +00004539 }
Eric Andersen25f27032001-04-26 23:22:31 +00004540 break;
4541 case '"':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004542 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004543 if (next == '"' && !ctx.pending_redirect)
4544 goto insert_empty_quoted_str_marker;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004545 if (dest.o_assignment == NOT_ASSIGNMENT)
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004546 dest.o_expflags |= EXP_FLAG_ESC_GLOB_CHARS;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004547 if (!encode_string(&ctx.as_string, &dest, input, '"', /*process_bkslash:*/ 1))
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004548 goto parse_error;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004549 dest.o_expflags &= ~EXP_FLAG_ESC_GLOB_CHARS;
Eric Andersen25f27032001-04-26 23:22:31 +00004550 break;
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004551#if ENABLE_HUSH_TICK
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004552 case '`': {
Denys Vlasenko60a94142011-05-13 20:57:01 +02004553 USE_FOR_NOMMU(unsigned pos;)
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004554
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004555 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4556 o_addchr(&dest, '`');
Denys Vlasenko60a94142011-05-13 20:57:01 +02004557 USE_FOR_NOMMU(pos = dest.length;)
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004558 if (!add_till_backquote(&dest, input, /*in_dquote:*/ 0))
4559 goto parse_error;
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004560# if !BB_MMU
Denis Vlasenko5c090a92009-04-08 21:51:33 +00004561 o_addstr(&ctx.as_string, dest.data + pos);
4562 o_addchr(&ctx.as_string, '`');
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004563# endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004564 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4565 //debug_printf_subst("SUBST RES3 '%s'\n", dest.data + pos);
Eric Andersen25f27032001-04-26 23:22:31 +00004566 break;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004567 }
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004568#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004569 case ';':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004570#if ENABLE_HUSH_CASE
4571 case_semi:
4572#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004573 if (done_word(&dest, &ctx)) {
4574 goto parse_error;
4575 }
4576 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004577#if ENABLE_HUSH_CASE
4578 /* Eat multiple semicolons, detect
4579 * whether it means something special */
4580 while (1) {
4581 ch = i_peek(input);
4582 if (ch != ';')
4583 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004584 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004585 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004586 if (ctx.ctx_res_w == RES_CASE_BODY) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004587 ctx.ctx_dsemicolon = 1;
4588 ctx.ctx_res_w = RES_MATCH;
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004589 break;
4590 }
4591 }
4592#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004593 new_cmd:
4594 /* We just finished a cmd. New one may start
4595 * with an assignment */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004596 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004597 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Eric Andersen25f27032001-04-26 23:22:31 +00004598 break;
4599 case '&':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004600 if (done_word(&dest, &ctx)) {
4601 goto parse_error;
4602 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004603 if (next == '&') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004604 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004605 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004606 done_pipe(&ctx, PIPE_AND);
Eric Andersen25f27032001-04-26 23:22:31 +00004607 } else {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004608 done_pipe(&ctx, PIPE_BG);
Eric Andersen25f27032001-04-26 23:22:31 +00004609 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004610 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004611 case '|':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004612 if (done_word(&dest, &ctx)) {
4613 goto parse_error;
4614 }
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004615#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004616 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenkof1736072008-07-31 10:09:26 +00004617 break; /* we are in case's "word | word)" */
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004618#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004619 if (next == '|') { /* || */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004620 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004621 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004622 done_pipe(&ctx, PIPE_OR);
Eric Andersen25f27032001-04-26 23:22:31 +00004623 } else {
4624 /* we could pick up a file descriptor choice here
4625 * with redirect_opt_num(), but bash doesn't do it.
4626 * "echo foo 2| cat" yields "foo 2". */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004627 done_command(&ctx);
Denys Vlasenkob70cef72010-01-12 13:45:45 +01004628#if !BB_MMU
4629 o_reset_to_empty_unquoted(&ctx.as_string);
4630#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004631 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004632 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004633 case '(':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004634#if ENABLE_HUSH_CASE
Denis Vlasenkof1736072008-07-31 10:09:26 +00004635 /* "case... in [(]word)..." - skip '(' */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004636 if (ctx.ctx_res_w == RES_MATCH
4637 && ctx.command->argv == NULL /* not (word|(... */
4638 && dest.length == 0 /* not word(... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004639 && dest.has_quoted_part == 0 /* not ""(... */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004640 ) {
4641 continue;
4642 }
4643#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004644 case '{':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004645 if (parse_group(&dest, &ctx, input, ch) != 0) {
4646 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004647 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004648 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004649 case ')':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004650#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004651 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004652 goto case_semi;
4653#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004654 case '}':
Denis Vlasenkoc3735272008-10-09 12:58:26 +00004655 /* proper use of this character is caught by end_trigger:
4656 * if we see {, we call parse_group(..., end_trigger='}')
4657 * and it will match } earlier (not here). */
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00004658 syntax_error_unexpected_ch(ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004659 goto parse_error;
Eric Andersen25f27032001-04-26 23:22:31 +00004660 default:
Denis Vlasenko5ec61322008-06-24 00:50:07 +00004661 if (HUSH_DEBUG)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00004662 bb_error_msg_and_die("BUG: unexpected %c\n", ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004663 }
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004664 } /* while (1) */
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004665
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004666 parse_error:
4667 {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004668 struct parse_context *pctx;
4669 IF_HAS_KEYWORDS(struct parse_context *p2;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004670
4671 /* Clean up allocated tree.
Denys Vlasenko764b2f02009-06-07 16:05:04 +02004672 * Sample for finding leaks on syntax error recovery path.
4673 * Run it from interactive shell, watch pmap `pidof hush`.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004674 * while if false; then false; fi; do break; fi
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00004675 * Samples to catch leaks at execution:
4676 * while if (true | {true;}); then echo ok; fi; do break; done
4677 * while if (true | {true;}); then echo ok; fi; do (if echo ok; break; then :; fi) | cat; break; done
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004678 */
4679 pctx = &ctx;
4680 do {
4681 /* Update pipe/command counts,
4682 * otherwise freeing may miss some */
4683 done_pipe(pctx, PIPE_SEQ);
4684 debug_printf_clean("freeing list %p from ctx %p\n",
4685 pctx->list_head, pctx);
4686 debug_print_tree(pctx->list_head, 0);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004687 free_pipe_list(pctx->list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004688 debug_printf_clean("freed list %p\n", pctx->list_head);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004689#if !BB_MMU
4690 o_free_unsafe(&pctx->as_string);
4691#endif
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004692 IF_HAS_KEYWORDS(p2 = pctx->stack;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004693 if (pctx != &ctx) {
4694 free(pctx);
4695 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004696 IF_HAS_KEYWORDS(pctx = p2;)
4697 } while (HAS_KEYWORDS && pctx);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004698
Denys Vlasenkoa439fa92011-03-30 19:11:46 +02004699 o_free(&dest);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004700 G.last_exitcode = 1;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004701#if !BB_MMU
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004702 if (pstring)
4703 *pstring = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004704#endif
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004705 debug_leave();
4706 return ERR_PTR;
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004707 }
Eric Andersen25f27032001-04-26 23:22:31 +00004708}
4709
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004710
4711/*** Execution routines ***/
4712
4713/* Expansion can recurse, need forward decls: */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004714#if !ENABLE_HUSH_BASH_COMPAT
4715/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4716#define expand_string_to_string(str, do_unbackslash) \
4717 expand_string_to_string(str)
4718#endif
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004719static char *expand_string_to_string(const char *str, int do_unbackslash);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004720#if ENABLE_HUSH_TICK
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004721static int process_command_subs(o_string *dest, const char *s);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004722#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004723
4724/* expand_strvec_to_strvec() takes a list of strings, expands
4725 * all variable references within and returns a pointer to
4726 * a list of expanded strings, possibly with larger number
4727 * of strings. (Think VAR="a b"; echo $VAR).
4728 * This new list is allocated as a single malloc block.
4729 * NULL-terminated list of char* pointers is at the beginning of it,
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004730 * followed by strings themselves.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004731 * Caller can deallocate entire list by single free(list). */
4732
Denys Vlasenko238081f2010-10-03 14:26:26 +02004733/* A horde of its helpers come first: */
4734
4735static void o_addblock_duplicate_backslash(o_string *o, const char *str, int len)
4736{
4737 while (--len >= 0) {
Denys Vlasenko9e800222010-10-03 14:28:04 +02004738 char c = *str++;
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004739
Denys Vlasenko9e800222010-10-03 14:28:04 +02004740#if ENABLE_HUSH_BRACE_EXPANSION
4741 if (c == '{' || c == '}') {
4742 /* { -> \{, } -> \} */
4743 o_addchr(o, '\\');
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004744 /* And now we want to add { or } and continue:
4745 * o_addchr(o, c);
4746 * continue;
4747 * luckily, just falling throught achieves this.
4748 */
Denys Vlasenko9e800222010-10-03 14:28:04 +02004749 }
4750#endif
4751 o_addchr(o, c);
4752 if (c == '\\') {
Denys Vlasenko238081f2010-10-03 14:26:26 +02004753 /* \z -> \\\z; \<eol> -> \\<eol> */
4754 o_addchr(o, '\\');
4755 if (len) {
4756 len--;
4757 o_addchr(o, '\\');
4758 o_addchr(o, *str++);
4759 }
4760 }
4761 }
4762}
4763
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004764/* Store given string, finalizing the word and starting new one whenever
4765 * we encounter IFS char(s). This is used for expanding variable values.
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004766 * End-of-string does NOT finalize word: think about 'echo -$VAR-'.
4767 * Return in *ended_with_ifs:
4768 * 1 - ended with IFS char, else 0 (this includes case of empty str).
4769 */
4770static int expand_on_ifs(int *ended_with_ifs, o_string *output, int n, const char *str)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004771{
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004772 int last_is_ifs = 0;
4773
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004774 while (1) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004775 int word_len;
4776
4777 if (!*str) /* EOL - do not finalize word */
4778 break;
4779 word_len = strcspn(str, G.ifs);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004780 if (word_len) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004781 /* We have WORD_LEN leading non-IFS chars */
Denys Vlasenko238081f2010-10-03 14:26:26 +02004782 if (!(output->o_expflags & EXP_FLAG_GLOB)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004783 o_addblock(output, str, word_len);
Denys Vlasenko238081f2010-10-03 14:26:26 +02004784 } else {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004785 /* Protect backslashes against globbing up :)
Denys Vlasenkoa769e022010-09-10 10:12:34 +02004786 * Example: "v='\*'; echo b$v" prints "b\*"
4787 * (and does not try to glob on "*")
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004788 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004789 o_addblock_duplicate_backslash(output, str, word_len);
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004790 /*/ Why can't we do it easier? */
4791 /*o_addblock(output, str, word_len); - WRONG: "v='\*'; echo Z$v" prints "Z*" instead of "Z\*" */
4792 /*o_addqblock(output, str, word_len); - WRONG: "v='*'; echo Z$v" prints "Z*" instead of Z* files */
4793 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004794 last_is_ifs = 0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004795 str += word_len;
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004796 if (!*str) /* EOL - do not finalize word */
4797 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004798 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004799
4800 /* We know str here points to at least one IFS char */
4801 last_is_ifs = 1;
4802 str += strspn(str, G.ifs); /* skip IFS chars */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004803 if (!*str) /* EOL - do not finalize word */
4804 break;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004805
4806 /* Start new word... but not always! */
4807 /* Case "v=' a'; echo ''$v": we do need to finalize empty word: */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004808 if (output->has_quoted_part
4809 /* Case "v=' a'; echo $v":
4810 * here nothing precedes the space in $v expansion,
4811 * therefore we should not finish the word
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004812 * (IOW: if there *is* word to finalize, only then do it):
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004813 */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004814 || (n > 0 && output->data[output->length - 1])
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004815 ) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004816 o_addchr(output, '\0');
4817 debug_print_list("expand_on_ifs", output, n);
4818 n = o_save_ptr(output, n);
4819 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004820 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004821
4822 if (ended_with_ifs)
4823 *ended_with_ifs = last_is_ifs;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004824 debug_print_list("expand_on_ifs[1]", output, n);
4825 return n;
4826}
4827
4828/* Helper to expand $((...)) and heredoc body. These act as if
4829 * they are in double quotes, with the exception that they are not :).
4830 * Just the rules are similar: "expand only $var and `cmd`"
4831 *
4832 * Returns malloced string.
4833 * As an optimization, we return NULL if expansion is not needed.
4834 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004835#if !ENABLE_HUSH_BASH_COMPAT
4836/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4837#define encode_then_expand_string(str, process_bkslash, do_unbackslash) \
4838 encode_then_expand_string(str)
4839#endif
4840static char *encode_then_expand_string(const char *str, int process_bkslash, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004841{
4842 char *exp_str;
4843 struct in_str input;
4844 o_string dest = NULL_O_STRING;
4845
4846 if (!strchr(str, '$')
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02004847 && !strchr(str, '\\')
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004848#if ENABLE_HUSH_TICK
4849 && !strchr(str, '`')
4850#endif
4851 ) {
4852 return NULL;
4853 }
4854
4855 /* We need to expand. Example:
4856 * echo $(($a + `echo 1`)) $((1 + $((2)) ))
4857 */
4858 setup_string_in_str(&input, str);
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004859 encode_string(NULL, &dest, &input, EOF, process_bkslash);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004860//TODO: error check (encode_string returns 0 on error)?
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004861 //bb_error_msg("'%s' -> '%s'", str, dest.data);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004862 exp_str = expand_string_to_string(dest.data, /*unbackslash:*/ do_unbackslash);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004863 //bb_error_msg("'%s' -> '%s'", dest.data, exp_str);
4864 o_free_unsafe(&dest);
4865 return exp_str;
4866}
4867
4868#if ENABLE_SH_MATH_SUPPORT
Denys Vlasenko063847d2010-09-15 13:33:02 +02004869static arith_t expand_and_evaluate_arith(const char *arg, const char **errmsg_p)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004870{
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004871 arith_state_t math_state;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004872 arith_t res;
4873 char *exp_str;
4874
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004875 math_state.lookupvar = get_local_var_value;
4876 math_state.setvar = set_local_var_from_halves;
4877 //math_state.endofname = endofname;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004878 exp_str = encode_then_expand_string(arg, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004879 res = arith(&math_state, exp_str ? exp_str : arg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004880 free(exp_str);
Denys Vlasenko063847d2010-09-15 13:33:02 +02004881 if (errmsg_p)
4882 *errmsg_p = math_state.errmsg;
4883 if (math_state.errmsg)
4884 die_if_script(math_state.errmsg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004885 return res;
4886}
4887#endif
4888
4889#if ENABLE_HUSH_BASH_COMPAT
4890/* ${var/[/]pattern[/repl]} helpers */
4891static char *strstr_pattern(char *val, const char *pattern, int *size)
4892{
4893 while (1) {
4894 char *end = scan_and_match(val, pattern, SCAN_MOVE_FROM_RIGHT + SCAN_MATCH_LEFT_HALF);
4895 debug_printf_varexp("val:'%s' pattern:'%s' end:'%s'\n", val, pattern, end);
4896 if (end) {
4897 *size = end - val;
4898 return val;
4899 }
4900 if (*val == '\0')
4901 return NULL;
4902 /* Optimization: if "*pat" did not match the start of "string",
4903 * we know that "tring", "ring" etc will not match too:
4904 */
4905 if (pattern[0] == '*')
4906 return NULL;
4907 val++;
4908 }
4909}
4910static char *replace_pattern(char *val, const char *pattern, const char *repl, char exp_op)
4911{
4912 char *result = NULL;
4913 unsigned res_len = 0;
4914 unsigned repl_len = strlen(repl);
4915
4916 while (1) {
4917 int size;
4918 char *s = strstr_pattern(val, pattern, &size);
4919 if (!s)
4920 break;
4921
4922 result = xrealloc(result, res_len + (s - val) + repl_len + 1);
4923 memcpy(result + res_len, val, s - val);
4924 res_len += s - val;
4925 strcpy(result + res_len, repl);
4926 res_len += repl_len;
4927 debug_printf_varexp("val:'%s' s:'%s' result:'%s'\n", val, s, result);
4928
4929 val = s + size;
4930 if (exp_op == '/')
4931 break;
4932 }
4933 if (val[0] && result) {
4934 result = xrealloc(result, res_len + strlen(val) + 1);
4935 strcpy(result + res_len, val);
4936 debug_printf_varexp("val:'%s' result:'%s'\n", val, result);
4937 }
4938 debug_printf_varexp("result:'%s'\n", result);
4939 return result;
4940}
4941#endif
4942
4943/* Helper:
4944 * Handles <SPECIAL_VAR_SYMBOL>varname...<SPECIAL_VAR_SYMBOL> construct.
4945 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004946static NOINLINE const char *expand_one_var(char **to_be_freed_pp, char *arg, char **pp)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004947{
4948 const char *val = NULL;
4949 char *to_be_freed = NULL;
4950 char *p = *pp;
4951 char *var;
4952 char first_char;
4953 char exp_op;
4954 char exp_save = exp_save; /* for compiler */
4955 char *exp_saveptr; /* points to expansion operator */
4956 char *exp_word = exp_word; /* for compiler */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004957 char arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004958
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004959 *p = '\0'; /* replace trailing SPECIAL_VAR_SYMBOL */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004960 var = arg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004961 exp_saveptr = arg[1] ? strchr(VAR_ENCODED_SUBST_OPS, arg[1]) : NULL;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004962 arg0 = arg[0];
4963 first_char = arg[0] = arg0 & 0x7f;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004964 exp_op = 0;
4965
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004966 if (first_char == '#' /* ${#... */
4967 && arg[1] && !exp_saveptr /* not ${#} and not ${#<op_char>...} */
4968 ) {
4969 /* It must be length operator: ${#var} */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004970 var++;
4971 exp_op = 'L';
4972 } else {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004973 /* Maybe handle parameter expansion */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004974 if (exp_saveptr /* if 2nd char is one of expansion operators */
4975 && strchr(NUMERIC_SPECVARS_STR, first_char) /* 1st char is special variable */
4976 ) {
4977 /* ${?:0}, ${#[:]%0} etc */
4978 exp_saveptr = var + 1;
4979 } else {
4980 /* ${?}, ${var}, ${var:0}, ${var[:]%0} etc */
4981 exp_saveptr = var+1 + strcspn(var+1, VAR_ENCODED_SUBST_OPS);
4982 }
4983 exp_op = exp_save = *exp_saveptr;
4984 if (exp_op) {
4985 exp_word = exp_saveptr + 1;
4986 if (exp_op == ':') {
4987 exp_op = *exp_word++;
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004988//TODO: try ${var:} and ${var:bogus} in non-bash config
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004989 if (ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004990 && (!exp_op || !strchr(MINUS_PLUS_EQUAL_QUESTION, exp_op))
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004991 ) {
4992 /* oops... it's ${var:N[:M]}, not ${var:?xxx} or some such */
4993 exp_op = ':';
4994 exp_word--;
4995 }
4996 }
4997 *exp_saveptr = '\0';
4998 } /* else: it's not an expansion op, but bare ${var} */
4999 }
5000
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005001 /* Look up the variable in question */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005002 if (isdigit(var[0])) {
Denys Vlasenko77a7b552010-09-09 12:40:03 +02005003 /* parse_dollar should have vetted var for us */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005004 int n = xatoi_positive(var);
5005 if (n < G.global_argc)
5006 val = G.global_argv[n];
5007 /* else val remains NULL: $N with too big N */
5008 } else {
5009 switch (var[0]) {
5010 case '$': /* pid */
5011 val = utoa(G.root_pid);
5012 break;
5013 case '!': /* bg pid */
5014 val = G.last_bg_pid ? utoa(G.last_bg_pid) : "";
5015 break;
5016 case '?': /* exitcode */
5017 val = utoa(G.last_exitcode);
5018 break;
5019 case '#': /* argc */
5020 val = utoa(G.global_argc ? G.global_argc-1 : 0);
5021 break;
5022 default:
5023 val = get_local_var_value(var);
5024 }
5025 }
5026
5027 /* Handle any expansions */
5028 if (exp_op == 'L') {
5029 debug_printf_expand("expand: length(%s)=", val);
5030 val = utoa(val ? strlen(val) : 0);
5031 debug_printf_expand("%s\n", val);
5032 } else if (exp_op) {
5033 if (exp_op == '%' || exp_op == '#') {
5034 /* Standard-mandated substring removal ops:
5035 * ${parameter%word} - remove smallest suffix pattern
5036 * ${parameter%%word} - remove largest suffix pattern
5037 * ${parameter#word} - remove smallest prefix pattern
5038 * ${parameter##word} - remove largest prefix pattern
5039 *
5040 * Word is expanded to produce a glob pattern.
5041 * Then var's value is matched to it and matching part removed.
5042 */
5043 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005044 char *t;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005045 char *exp_exp_word;
5046 char *loc;
5047 unsigned scan_flags = pick_scan(exp_op, *exp_word);
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02005048 if (exp_op == *exp_word) /* ## or %% */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005049 exp_word++;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005050 exp_exp_word = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005051 if (exp_exp_word)
5052 exp_word = exp_exp_word;
Denys Vlasenko4f870492010-09-10 11:06:01 +02005053 /* HACK ALERT. We depend here on the fact that
5054 * G.global_argv and results of utoa and get_local_var_value
5055 * are actually in writable memory:
5056 * scan_and_match momentarily stores NULs there. */
5057 t = (char*)val;
5058 loc = scan_and_match(t, exp_word, scan_flags);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005059 //bb_error_msg("op:%c str:'%s' pat:'%s' res:'%s'",
Denys Vlasenko4f870492010-09-10 11:06:01 +02005060 // exp_op, t, exp_word, loc);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005061 free(exp_exp_word);
5062 if (loc) { /* match was found */
5063 if (scan_flags & SCAN_MATCH_LEFT_HALF) /* #[#] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005064 val = loc; /* take right part */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005065 else /* %[%] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005066 val = to_be_freed = xstrndup(val, loc - val); /* left */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005067 }
5068 }
5069 }
5070#if ENABLE_HUSH_BASH_COMPAT
5071 else if (exp_op == '/' || exp_op == '\\') {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005072 /* It's ${var/[/]pattern[/repl]} thing.
5073 * Note that in encoded form it has TWO parts:
5074 * var/pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenko4f870492010-09-10 11:06:01 +02005075 * and if // is used, it is encoded as \:
5076 * var\pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005077 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005078 /* Empty variable always gives nothing: */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005079 // "v=''; echo ${v/*/w}" prints "", not "w"
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005080 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005081 /* pattern uses non-standard expansion.
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005082 * repl should be unbackslashed and globbed
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005083 * by the usual expansion rules:
5084 * >az; >bz;
5085 * v='a bz'; echo "${v/a*z/a*z}" prints "a*z"
5086 * v='a bz'; echo "${v/a*z/\z}" prints "\z"
5087 * v='a bz'; echo ${v/a*z/a*z} prints "az"
5088 * v='a bz'; echo ${v/a*z/\z} prints "z"
5089 * (note that a*z _pattern_ is never globbed!)
5090 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005091 char *pattern, *repl, *t;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005092 pattern = encode_then_expand_string(exp_word, /*process_bkslash:*/ 0, /*unbackslash:*/ 0);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005093 if (!pattern)
5094 pattern = xstrdup(exp_word);
5095 debug_printf_varexp("pattern:'%s'->'%s'\n", exp_word, pattern);
5096 *p++ = SPECIAL_VAR_SYMBOL;
5097 exp_word = p;
5098 p = strchr(p, SPECIAL_VAR_SYMBOL);
5099 *p = '\0';
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005100 repl = encode_then_expand_string(exp_word, /*process_bkslash:*/ arg0 & 0x80, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005101 debug_printf_varexp("repl:'%s'->'%s'\n", exp_word, repl);
5102 /* HACK ALERT. We depend here on the fact that
5103 * G.global_argv and results of utoa and get_local_var_value
5104 * are actually in writable memory:
5105 * replace_pattern momentarily stores NULs there. */
5106 t = (char*)val;
5107 to_be_freed = replace_pattern(t,
5108 pattern,
5109 (repl ? repl : exp_word),
5110 exp_op);
5111 if (to_be_freed) /* at least one replace happened */
5112 val = to_be_freed;
5113 free(pattern);
5114 free(repl);
5115 }
5116 }
5117#endif
5118 else if (exp_op == ':') {
5119#if ENABLE_HUSH_BASH_COMPAT && ENABLE_SH_MATH_SUPPORT
5120 /* It's ${var:N[:M]} bashism.
5121 * Note that in encoded form it has TWO parts:
5122 * var:N<SPECIAL_VAR_SYMBOL>M<SPECIAL_VAR_SYMBOL>
5123 */
5124 arith_t beg, len;
Denys Vlasenko063847d2010-09-15 13:33:02 +02005125 const char *errmsg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005126
Denys Vlasenko063847d2010-09-15 13:33:02 +02005127 beg = expand_and_evaluate_arith(exp_word, &errmsg);
5128 if (errmsg)
5129 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005130 debug_printf_varexp("beg:'%s'=%lld\n", exp_word, (long long)beg);
5131 *p++ = SPECIAL_VAR_SYMBOL;
5132 exp_word = p;
5133 p = strchr(p, SPECIAL_VAR_SYMBOL);
5134 *p = '\0';
Denys Vlasenko063847d2010-09-15 13:33:02 +02005135 len = expand_and_evaluate_arith(exp_word, &errmsg);
5136 if (errmsg)
5137 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005138 debug_printf_varexp("len:'%s'=%lld\n", exp_word, (long long)len);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005139 if (len >= 0) { /* bash compat: len < 0 is illegal */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005140 if (beg < 0) /* bash compat */
5141 beg = 0;
5142 debug_printf_varexp("from val:'%s'\n", val);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005143 if (len == 0 || !val || beg >= strlen(val)) {
Denys Vlasenko063847d2010-09-15 13:33:02 +02005144 arith_err:
Denys Vlasenkob771c652010-09-13 00:34:26 +02005145 val = NULL;
5146 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005147 /* Paranoia. What if user entered 9999999999999
5148 * which fits in arith_t but not int? */
5149 if (len >= INT_MAX)
5150 len = INT_MAX;
5151 val = to_be_freed = xstrndup(val + beg, len);
5152 }
5153 debug_printf_varexp("val:'%s'\n", val);
5154 } else
5155#endif
5156 {
5157 die_if_script("malformed ${%s:...}", var);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005158 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005159 }
5160 } else { /* one of "-=+?" */
5161 /* Standard-mandated substitution ops:
5162 * ${var?word} - indicate error if unset
5163 * If var is unset, word (or a message indicating it is unset
5164 * if word is null) is written to standard error
5165 * and the shell exits with a non-zero exit status.
5166 * Otherwise, the value of var is substituted.
5167 * ${var-word} - use default value
5168 * If var is unset, word is substituted.
5169 * ${var=word} - assign and use default value
5170 * If var is unset, word is assigned to var.
5171 * In all cases, final value of var is substituted.
5172 * ${var+word} - use alternative value
5173 * If var is unset, null is substituted.
5174 * Otherwise, word is substituted.
5175 *
5176 * Word is subjected to tilde expansion, parameter expansion,
5177 * command substitution, and arithmetic expansion.
5178 * If word is not needed, it is not expanded.
5179 *
5180 * Colon forms (${var:-word}, ${var:=word} etc) do the same,
5181 * but also treat null var as if it is unset.
5182 */
5183 int use_word = (!val || ((exp_save == ':') && !val[0]));
5184 if (exp_op == '+')
5185 use_word = !use_word;
5186 debug_printf_expand("expand: op:%c (null:%s) test:%i\n", exp_op,
5187 (exp_save == ':') ? "true" : "false", use_word);
5188 if (use_word) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005189 to_be_freed = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005190 if (to_be_freed)
5191 exp_word = to_be_freed;
5192 if (exp_op == '?') {
5193 /* mimic bash message */
5194 die_if_script("%s: %s",
5195 var,
5196 exp_word[0] ? exp_word : "parameter null or not set"
5197 );
5198//TODO: how interactive bash aborts expansion mid-command?
5199 } else {
5200 val = exp_word;
5201 }
5202
5203 if (exp_op == '=') {
5204 /* ${var=[word]} or ${var:=[word]} */
5205 if (isdigit(var[0]) || var[0] == '#') {
5206 /* mimic bash message */
5207 die_if_script("$%s: cannot assign in this way", var);
5208 val = NULL;
5209 } else {
5210 char *new_var = xasprintf("%s=%s", var, val);
5211 set_local_var(new_var, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
5212 }
5213 }
5214 }
5215 } /* one of "-=+?" */
5216
5217 *exp_saveptr = exp_save;
5218 } /* if (exp_op) */
5219
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005220 arg[0] = arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005221
5222 *pp = p;
5223 *to_be_freed_pp = to_be_freed;
5224 return val;
5225}
5226
5227/* Expand all variable references in given string, adding words to list[]
5228 * at n, n+1,... positions. Return updated n (so that list[n] is next one
5229 * to be filled). This routine is extremely tricky: has to deal with
5230 * variables/parameters with whitespace, $* and $@, and constructs like
5231 * 'echo -$*-'. If you play here, you must run testsuite afterwards! */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005232static NOINLINE int expand_vars_to_list(o_string *output, int n, char *arg)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005233{
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005234 /* output->o_expflags & EXP_FLAG_SINGLEWORD (0x80) if we are in
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005235 * expansion of right-hand side of assignment == 1-element expand.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005236 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005237 char cant_be_null = 0; /* only bit 0x80 matters */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005238 int ended_in_ifs = 0; /* did last unquoted expansion end with IFS chars? */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005239 char *p;
5240
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005241 debug_printf_expand("expand_vars_to_list: arg:'%s' singleword:%x\n", arg,
5242 !!(output->o_expflags & EXP_FLAG_SINGLEWORD));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005243 debug_print_list("expand_vars_to_list", output, n);
5244 n = o_save_ptr(output, n);
5245 debug_print_list("expand_vars_to_list[0]", output, n);
5246
5247 while ((p = strchr(arg, SPECIAL_VAR_SYMBOL)) != NULL) {
5248 char first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005249 char *to_be_freed = NULL;
5250 const char *val = NULL;
5251#if ENABLE_HUSH_TICK
5252 o_string subst_result = NULL_O_STRING;
5253#endif
5254#if ENABLE_SH_MATH_SUPPORT
5255 char arith_buf[sizeof(arith_t)*3 + 2];
5256#endif
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005257
5258 if (ended_in_ifs) {
5259 o_addchr(output, '\0');
5260 n = o_save_ptr(output, n);
5261 ended_in_ifs = 0;
5262 }
5263
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005264 o_addblock(output, arg, p - arg);
5265 debug_print_list("expand_vars_to_list[1]", output, n);
5266 arg = ++p;
5267 p = strchr(p, SPECIAL_VAR_SYMBOL);
5268
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005269 /* Fetch special var name (if it is indeed one of them)
5270 * and quote bit, force the bit on if singleword expansion -
5271 * important for not getting v=$@ expand to many words. */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005272 first_ch = arg[0] | (output->o_expflags & EXP_FLAG_SINGLEWORD);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005273
5274 /* Is this variable quoted and thus expansion can't be null?
5275 * "$@" is special. Even if quoted, it can still
5276 * expand to nothing (not even an empty string),
5277 * thus it is excluded. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005278 if ((first_ch & 0x7f) != '@')
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005279 cant_be_null |= first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005280
5281 switch (first_ch & 0x7f) {
5282 /* Highest bit in first_ch indicates that var is double-quoted */
5283 case '*':
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005284 case '@': {
5285 int i;
5286 if (!G.global_argv[1])
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005287 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005288 i = 1;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005289 cant_be_null |= first_ch; /* do it for "$@" _now_, when we know it's not empty */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005290 if (!(first_ch & 0x80)) { /* unquoted $* or $@ */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005291 while (G.global_argv[i]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005292 n = expand_on_ifs(NULL, output, n, G.global_argv[i]);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005293 debug_printf_expand("expand_vars_to_list: argv %d (last %d)\n", i, G.global_argc - 1);
5294 if (G.global_argv[i++][0] && G.global_argv[i]) {
5295 /* this argv[] is not empty and not last:
5296 * put terminating NUL, start new word */
5297 o_addchr(output, '\0');
5298 debug_print_list("expand_vars_to_list[2]", output, n);
5299 n = o_save_ptr(output, n);
5300 debug_print_list("expand_vars_to_list[3]", output, n);
5301 }
5302 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005303 } else
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005304 /* If EXP_FLAG_SINGLEWORD, we handle assignment 'a=....$@.....'
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005305 * and in this case should treat it like '$*' - see 'else...' below */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005306 if (first_ch == ('@'|0x80) /* quoted $@ */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005307 && !(output->o_expflags & EXP_FLAG_SINGLEWORD) /* not v="$@" case */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005308 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005309 while (1) {
5310 o_addQstr(output, G.global_argv[i]);
5311 if (++i >= G.global_argc)
5312 break;
5313 o_addchr(output, '\0');
5314 debug_print_list("expand_vars_to_list[4]", output, n);
5315 n = o_save_ptr(output, n);
5316 }
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005317 } else { /* quoted $* (or v="$@" case): add as one word */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005318 while (1) {
5319 o_addQstr(output, G.global_argv[i]);
5320 if (!G.global_argv[++i])
5321 break;
5322 if (G.ifs[0])
5323 o_addchr(output, G.ifs[0]);
5324 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005325 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005326 }
5327 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005328 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005329 case SPECIAL_VAR_SYMBOL: /* <SPECIAL_VAR_SYMBOL><SPECIAL_VAR_SYMBOL> */
5330 /* "Empty variable", used to make "" etc to not disappear */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005331 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005332 arg++;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005333 cant_be_null = 0x80;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005334 break;
5335#if ENABLE_HUSH_TICK
5336 case '`': /* <SPECIAL_VAR_SYMBOL>`cmd<SPECIAL_VAR_SYMBOL> */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005337 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005338 arg++;
5339 /* Can't just stuff it into output o_string,
5340 * expanded result may need to be globbed
5341 * and $IFS-splitted */
5342 debug_printf_subst("SUBST '%s' first_ch %x\n", arg, first_ch);
5343 G.last_exitcode = process_command_subs(&subst_result, arg);
5344 debug_printf_subst("SUBST RES:%d '%s'\n", G.last_exitcode, subst_result.data);
5345 val = subst_result.data;
5346 goto store_val;
5347#endif
5348#if ENABLE_SH_MATH_SUPPORT
5349 case '+': { /* <SPECIAL_VAR_SYMBOL>+cmd<SPECIAL_VAR_SYMBOL> */
5350 arith_t res;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005351
5352 arg++; /* skip '+' */
5353 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
5354 debug_printf_subst("ARITH '%s' first_ch %x\n", arg, first_ch);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005355 res = expand_and_evaluate_arith(arg, NULL);
Denys Vlasenkobed7c812010-09-16 11:50:46 +02005356 debug_printf_subst("ARITH RES '"ARITH_FMT"'\n", res);
5357 sprintf(arith_buf, ARITH_FMT, res);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005358 val = arith_buf;
5359 break;
5360 }
5361#endif
5362 default:
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005363 val = expand_one_var(&to_be_freed, arg, &p);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005364 IF_HUSH_TICK(store_val:)
5365 if (!(first_ch & 0x80)) { /* unquoted $VAR */
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005366 debug_printf_expand("unquoted '%s', output->o_escape:%d\n", val,
5367 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005368 if (val && val[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005369 n = expand_on_ifs(&ended_in_ifs, output, n, val);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005370 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005371 }
5372 } else { /* quoted $VAR, val will be appended below */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005373 output->has_quoted_part = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005374 debug_printf_expand("quoted '%s', output->o_escape:%d\n", val,
5375 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005376 }
5377 break;
5378
5379 } /* switch (char after <SPECIAL_VAR_SYMBOL>) */
5380
5381 if (val && val[0]) {
5382 o_addQstr(output, val);
5383 }
5384 free(to_be_freed);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005385
5386 /* Restore NULL'ed SPECIAL_VAR_SYMBOL.
5387 * Do the check to avoid writing to a const string. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005388 if (*p != SPECIAL_VAR_SYMBOL)
5389 *p = SPECIAL_VAR_SYMBOL;
5390
5391#if ENABLE_HUSH_TICK
5392 o_free(&subst_result);
5393#endif
5394 arg = ++p;
5395 } /* end of "while (SPECIAL_VAR_SYMBOL is found) ..." */
5396
5397 if (arg[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005398 if (ended_in_ifs) {
5399 o_addchr(output, '\0');
5400 n = o_save_ptr(output, n);
5401 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005402 debug_print_list("expand_vars_to_list[a]", output, n);
5403 /* this part is literal, and it was already pre-quoted
5404 * if needed (much earlier), do not use o_addQstr here! */
5405 o_addstr_with_NUL(output, arg);
5406 debug_print_list("expand_vars_to_list[b]", output, n);
5407 } else if (output->length == o_get_last_ptr(output, n) /* expansion is empty */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005408 && !(cant_be_null & 0x80) /* and all vars were not quoted. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005409 ) {
5410 n--;
5411 /* allow to reuse list[n] later without re-growth */
5412 output->has_empty_slot = 1;
5413 } else {
5414 o_addchr(output, '\0');
5415 }
5416
5417 return n;
5418}
5419
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005420static char **expand_variables(char **argv, unsigned expflags)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005421{
5422 int n;
5423 char **list;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005424 o_string output = NULL_O_STRING;
5425
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005426 output.o_expflags = expflags;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005427
5428 n = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005429 while (*argv) {
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005430 n = expand_vars_to_list(&output, n, *argv);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005431 argv++;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005432 }
5433 debug_print_list("expand_variables", &output, n);
5434
5435 /* output.data (malloced in one block) gets returned in "list" */
5436 list = o_finalize_list(&output, n);
5437 debug_print_strings("expand_variables[1]", list);
5438 return list;
5439}
5440
5441static char **expand_strvec_to_strvec(char **argv)
5442{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005443 return expand_variables(argv, EXP_FLAG_GLOB | EXP_FLAG_ESC_GLOB_CHARS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005444}
5445
5446#if ENABLE_HUSH_BASH_COMPAT
5447static char **expand_strvec_to_strvec_singleword_noglob(char **argv)
5448{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005449 return expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005450}
5451#endif
5452
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005453/* Used for expansion of right hand of assignments,
5454 * $((...)), heredocs, variable espansion parts.
5455 *
5456 * NB: should NOT do globbing!
5457 * "export v=/bin/c*; env | grep ^v=" outputs "v=/bin/c*"
5458 */
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005459static char *expand_string_to_string(const char *str, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005460{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005461#if !ENABLE_HUSH_BASH_COMPAT
5462 const int do_unbackslash = 1;
5463#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005464 char *argv[2], **list;
5465
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005466 debug_printf_expand("string_to_string<='%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005467 /* This is generally an optimization, but it also
5468 * handles "", which otherwise trips over !list[0] check below.
5469 * (is this ever happens that we actually get str="" here?)
5470 */
5471 if (!strchr(str, SPECIAL_VAR_SYMBOL) && !strchr(str, '\\')) {
5472 //TODO: Can use on strings with \ too, just unbackslash() them?
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005473 debug_printf_expand("string_to_string(fast)=>'%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005474 return xstrdup(str);
5475 }
5476
5477 argv[0] = (char*)str;
5478 argv[1] = NULL;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005479 list = expand_variables(argv, do_unbackslash
5480 ? EXP_FLAG_ESC_GLOB_CHARS | EXP_FLAG_SINGLEWORD
5481 : EXP_FLAG_SINGLEWORD
5482 );
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005483 if (HUSH_DEBUG)
5484 if (!list[0] || list[1])
5485 bb_error_msg_and_die("BUG in varexp2");
5486 /* actually, just move string 2*sizeof(char*) bytes back */
5487 overlapping_strcpy((char*)list, list[0]);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005488 if (do_unbackslash)
5489 unbackslash((char*)list);
5490 debug_printf_expand("string_to_string=>'%s'\n", (char*)list);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005491 return (char*)list;
5492}
5493
5494/* Used for "eval" builtin */
5495static char* expand_strvec_to_string(char **argv)
5496{
5497 char **list;
5498
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005499 list = expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005500 /* Convert all NULs to spaces */
5501 if (list[0]) {
5502 int n = 1;
5503 while (list[n]) {
5504 if (HUSH_DEBUG)
5505 if (list[n-1] + strlen(list[n-1]) + 1 != list[n])
5506 bb_error_msg_and_die("BUG in varexp3");
5507 /* bash uses ' ' regardless of $IFS contents */
5508 list[n][-1] = ' ';
5509 n++;
5510 }
5511 }
5512 overlapping_strcpy((char*)list, list[0]);
5513 debug_printf_expand("strvec_to_string='%s'\n", (char*)list);
5514 return (char*)list;
5515}
5516
5517static char **expand_assignments(char **argv, int count)
5518{
5519 int i;
5520 char **p;
5521
5522 G.expanded_assignments = p = NULL;
5523 /* Expand assignments into one string each */
5524 for (i = 0; i < count; i++) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005525 G.expanded_assignments = p = add_string_to_strings(p, expand_string_to_string(argv[i], /*unbackslash:*/ 1));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005526 }
5527 G.expanded_assignments = NULL;
5528 return p;
5529}
5530
5531
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005532static void switch_off_special_sigs(unsigned mask)
5533{
5534 unsigned sig = 0;
5535 while ((mask >>= 1) != 0) {
5536 sig++;
5537 if (!(mask & 1))
5538 continue;
5539 if (G.traps) {
5540 if (G.traps[sig] && !G.traps[sig][0])
5541 /* trap is '', has to remain SIG_IGN */
5542 continue;
5543 free(G.traps[sig]);
5544 G.traps[sig] = NULL;
5545 }
5546 /* We are here only if no trap or trap was not '' */
Denys Vlasenko0806e402011-05-12 23:06:20 +02005547 install_sighandler(sig, SIG_DFL);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005548 }
5549}
5550
Denys Vlasenkob347df92011-08-09 22:49:15 +02005551#if BB_MMU
5552/* never called */
5553void re_execute_shell(char ***to_free, const char *s,
5554 char *g_argv0, char **g_argv,
5555 char **builtin_argv) NORETURN;
5556
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005557static void reset_traps_to_defaults(void)
5558{
5559 /* This function is always called in a child shell
5560 * after fork (not vfork, NOMMU doesn't use this function).
5561 */
5562 unsigned sig;
5563 unsigned mask;
5564
5565 /* Child shells are not interactive.
5566 * SIGTTIN/SIGTTOU/SIGTSTP should not have special handling.
5567 * Testcase: (while :; do :; done) + ^Z should background.
5568 * Same goes for SIGTERM, SIGHUP, SIGINT.
5569 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005570 mask = (G.special_sig_mask & SPECIAL_INTERACTIVE_SIGS) | G_fatal_sig_mask;
5571 if (!G.traps && !mask)
5572 return; /* already no traps and no special sigs */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005573
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005574 /* Switch off special sigs */
5575 switch_off_special_sigs(mask);
5576#if ENABLE_HUSH_JOB
5577 G_fatal_sig_mask = 0;
5578#endif
Denys Vlasenko10c01312011-05-11 11:49:21 +02005579 G.special_sig_mask &= ~SPECIAL_INTERACTIVE_SIGS;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02005580 /* SIGQUIT,SIGCHLD and maybe SPECIAL_JOBSTOP_SIGS
5581 * remain set in G.special_sig_mask */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005582
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005583 if (!G.traps)
5584 return;
5585
5586 /* Reset all sigs to default except ones with empty traps */
5587 for (sig = 0; sig < NSIG; sig++) {
5588 if (!G.traps[sig])
5589 continue; /* no trap: nothing to do */
5590 if (!G.traps[sig][0])
5591 continue; /* empty trap: has to remain SIG_IGN */
5592 /* sig has non-empty trap, reset it: */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005593 free(G.traps[sig]);
5594 G.traps[sig] = NULL;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005595 /* There is no signal for trap 0 (EXIT) */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005596 if (sig == 0)
5597 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02005598 install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005599 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005600}
5601
5602#else /* !BB_MMU */
5603
5604static void re_execute_shell(char ***to_free, const char *s,
5605 char *g_argv0, char **g_argv,
5606 char **builtin_argv) NORETURN;
5607static void re_execute_shell(char ***to_free, const char *s,
5608 char *g_argv0, char **g_argv,
5609 char **builtin_argv)
5610{
5611# define NOMMU_HACK_FMT ("-$%x:%x:%x:%x:%x:%llx" IF_HUSH_LOOPS(":%x"))
5612 /* delims + 2 * (number of bytes in printed hex numbers) */
5613 char param_buf[sizeof(NOMMU_HACK_FMT) + 2 * (sizeof(int)*6 + sizeof(long long)*1)];
5614 char *heredoc_argv[4];
5615 struct variable *cur;
5616# if ENABLE_HUSH_FUNCTIONS
5617 struct function *funcp;
5618# endif
5619 char **argv, **pp;
5620 unsigned cnt;
5621 unsigned long long empty_trap_mask;
5622
5623 if (!g_argv0) { /* heredoc */
5624 argv = heredoc_argv;
5625 argv[0] = (char *) G.argv0_for_re_execing;
5626 argv[1] = (char *) "-<";
5627 argv[2] = (char *) s;
5628 argv[3] = NULL;
5629 pp = &argv[3]; /* used as pointer to empty environment */
5630 goto do_exec;
5631 }
5632
5633 cnt = 0;
5634 pp = builtin_argv;
5635 if (pp) while (*pp++)
5636 cnt++;
5637
5638 empty_trap_mask = 0;
5639 if (G.traps) {
5640 int sig;
5641 for (sig = 1; sig < NSIG; sig++) {
5642 if (G.traps[sig] && !G.traps[sig][0])
5643 empty_trap_mask |= 1LL << sig;
5644 }
5645 }
5646
5647 sprintf(param_buf, NOMMU_HACK_FMT
5648 , (unsigned) G.root_pid
5649 , (unsigned) G.root_ppid
5650 , (unsigned) G.last_bg_pid
5651 , (unsigned) G.last_exitcode
5652 , cnt
5653 , empty_trap_mask
5654 IF_HUSH_LOOPS(, G.depth_of_loop)
5655 );
5656# undef NOMMU_HACK_FMT
5657 /* 1:hush 2:-$<pid>:<pid>:<exitcode>:<etc...> <vars...> <funcs...>
5658 * 3:-c 4:<cmd> 5:<arg0> <argN...> 6:NULL
5659 */
5660 cnt += 6;
5661 for (cur = G.top_var; cur; cur = cur->next) {
5662 if (!cur->flg_export || cur->flg_read_only)
5663 cnt += 2;
5664 }
5665# if ENABLE_HUSH_FUNCTIONS
5666 for (funcp = G.top_func; funcp; funcp = funcp->next)
5667 cnt += 3;
5668# endif
5669 pp = g_argv;
5670 while (*pp++)
5671 cnt++;
5672 *to_free = argv = pp = xzalloc(sizeof(argv[0]) * cnt);
5673 *pp++ = (char *) G.argv0_for_re_execing;
5674 *pp++ = param_buf;
5675 for (cur = G.top_var; cur; cur = cur->next) {
5676 if (strcmp(cur->varstr, hush_version_str) == 0)
5677 continue;
5678 if (cur->flg_read_only) {
5679 *pp++ = (char *) "-R";
5680 *pp++ = cur->varstr;
5681 } else if (!cur->flg_export) {
5682 *pp++ = (char *) "-V";
5683 *pp++ = cur->varstr;
5684 }
5685 }
5686# if ENABLE_HUSH_FUNCTIONS
5687 for (funcp = G.top_func; funcp; funcp = funcp->next) {
5688 *pp++ = (char *) "-F";
5689 *pp++ = funcp->name;
5690 *pp++ = funcp->body_as_string;
5691 }
5692# endif
5693 /* We can pass activated traps here. Say, -Tnn:trap_string
5694 *
5695 * However, POSIX says that subshells reset signals with traps
5696 * to SIG_DFL.
5697 * I tested bash-3.2 and it not only does that with true subshells
5698 * of the form ( list ), but with any forked children shells.
5699 * I set trap "echo W" WINCH; and then tried:
5700 *
5701 * { echo 1; sleep 20; echo 2; } &
5702 * while true; do echo 1; sleep 20; echo 2; break; done &
5703 * true | { echo 1; sleep 20; echo 2; } | cat
5704 *
5705 * In all these cases sending SIGWINCH to the child shell
5706 * did not run the trap. If I add trap "echo V" WINCH;
5707 * _inside_ group (just before echo 1), it works.
5708 *
5709 * I conclude it means we don't need to pass active traps here.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005710 */
5711 *pp++ = (char *) "-c";
5712 *pp++ = (char *) s;
5713 if (builtin_argv) {
5714 while (*++builtin_argv)
5715 *pp++ = *builtin_argv;
5716 *pp++ = (char *) "";
5717 }
5718 *pp++ = g_argv0;
5719 while (*g_argv)
5720 *pp++ = *g_argv++;
5721 /* *pp = NULL; - is already there */
5722 pp = environ;
5723
5724 do_exec:
5725 debug_printf_exec("re_execute_shell pid:%d cmd:'%s'\n", getpid(), s);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02005726 /* Don't propagate SIG_IGN to the child */
5727 if (SPECIAL_JOBSTOP_SIGS != 0)
5728 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005729 execve(bb_busybox_exec_path, argv, pp);
5730 /* Fallback. Useful for init=/bin/hush usage etc */
5731 if (argv[0][0] == '/')
5732 execve(argv[0], argv, pp);
5733 xfunc_error_retval = 127;
5734 bb_error_msg_and_die("can't re-execute the shell");
5735}
5736#endif /* !BB_MMU */
5737
5738
5739static int run_and_free_list(struct pipe *pi);
5740
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005741/* Executing from string: eval, sh -c '...'
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005742 * or from file: /etc/profile, . file, sh <script>, sh (intereactive)
5743 * end_trigger controls how often we stop parsing
5744 * NUL: parse all, execute, return
5745 * ';': parse till ';' or newline, execute, repeat till EOF
5746 */
5747static void parse_and_run_stream(struct in_str *inp, int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00005748{
Denys Vlasenko00243b02009-11-16 02:00:03 +01005749 /* Why we need empty flag?
5750 * An obscure corner case "false; ``; echo $?":
5751 * empty command in `` should still set $? to 0.
5752 * But we can't just set $? to 0 at the start,
5753 * this breaks "false; echo `echo $?`" case.
5754 */
5755 bool empty = 1;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005756 while (1) {
5757 struct pipe *pipe_list;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00005758
Denys Vlasenkoa1463192011-01-18 17:55:04 +01005759#if ENABLE_HUSH_INTERACTIVE
5760 if (end_trigger == ';')
5761 inp->promptmode = 0; /* PS1 */
5762#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005763 pipe_list = parse_stream(NULL, inp, end_trigger);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02005764 if (!pipe_list || pipe_list == ERR_PTR) { /* EOF/error */
5765 /* If we are in "big" script
5766 * (not in `cmd` or something similar)...
5767 */
5768 if (pipe_list == ERR_PTR && end_trigger == ';') {
5769 /* Discard cached input (rest of line) */
5770 int ch = inp->last_char;
5771 while (ch != EOF && ch != '\n') {
5772 //bb_error_msg("Discarded:'%c'", ch);
5773 ch = i_getch(inp);
5774 }
5775 /* Force prompt */
5776 inp->p = NULL;
5777 /* This stream isn't empty */
5778 empty = 0;
5779 continue;
5780 }
5781 if (!pipe_list && empty)
Denys Vlasenko00243b02009-11-16 02:00:03 +01005782 G.last_exitcode = 0;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005783 break;
Denys Vlasenko00243b02009-11-16 02:00:03 +01005784 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005785 debug_print_tree(pipe_list, 0);
5786 debug_printf_exec("parse_and_run_stream: run_and_free_list\n");
5787 run_and_free_list(pipe_list);
Denys Vlasenko00243b02009-11-16 02:00:03 +01005788 empty = 0;
Denys Vlasenko68d5cb52011-03-24 02:50:03 +01005789#if ENABLE_HUSH_FUNCTIONS
5790 if (G.flag_return_in_progress == 1)
5791 break;
5792#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005793 }
Eric Andersen25f27032001-04-26 23:22:31 +00005794}
5795
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005796static void parse_and_run_string(const char *s)
Eric Andersen25f27032001-04-26 23:22:31 +00005797{
5798 struct in_str input;
5799 setup_string_in_str(&input, s);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005800 parse_and_run_stream(&input, '\0');
Eric Andersen25f27032001-04-26 23:22:31 +00005801}
5802
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005803static void parse_and_run_file(FILE *f)
Eric Andersen25f27032001-04-26 23:22:31 +00005804{
Eric Andersen25f27032001-04-26 23:22:31 +00005805 struct in_str input;
5806 setup_file_in_str(&input, f);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005807 parse_and_run_stream(&input, ';');
Eric Andersen25f27032001-04-26 23:22:31 +00005808}
5809
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005810#if ENABLE_HUSH_TICK
5811static FILE *generate_stream_from_string(const char *s, pid_t *pid_p)
5812{
5813 pid_t pid;
5814 int channel[2];
5815# if !BB_MMU
5816 char **to_free = NULL;
5817# endif
5818
5819 xpipe(channel);
5820 pid = BB_MMU ? xfork() : xvfork();
5821 if (pid == 0) { /* child */
5822 disable_restore_tty_pgrp_on_exit();
5823 /* Process substitution is not considered to be usual
5824 * 'command execution'.
5825 * SUSv3 says ctrl-Z should be ignored, ctrl-C should not.
5826 */
5827 bb_signals(0
5828 + (1 << SIGTSTP)
5829 + (1 << SIGTTIN)
5830 + (1 << SIGTTOU)
5831 , SIG_IGN);
5832 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
5833 close(channel[0]); /* NB: close _first_, then move fd! */
5834 xmove_fd(channel[1], 1);
5835 /* Prevent it from trying to handle ctrl-z etc */
5836 IF_HUSH_JOB(G.run_list_level = 1;)
5837 /* Awful hack for `trap` or $(trap).
5838 *
5839 * http://www.opengroup.org/onlinepubs/009695399/utilities/trap.html
5840 * contains an example where "trap" is executed in a subshell:
5841 *
5842 * save_traps=$(trap)
5843 * ...
5844 * eval "$save_traps"
5845 *
5846 * Standard does not say that "trap" in subshell shall print
5847 * parent shell's traps. It only says that its output
5848 * must have suitable form, but then, in the above example
5849 * (which is not supposed to be normative), it implies that.
5850 *
5851 * bash (and probably other shell) does implement it
5852 * (traps are reset to defaults, but "trap" still shows them),
5853 * but as a result, "trap" logic is hopelessly messed up:
5854 *
5855 * # trap
5856 * trap -- 'echo Ho' SIGWINCH <--- we have a handler
5857 * # (trap) <--- trap is in subshell - no output (correct, traps are reset)
5858 * # true | trap <--- trap is in subshell - no output (ditto)
5859 * # echo `true | trap` <--- in subshell - output (but traps are reset!)
5860 * trap -- 'echo Ho' SIGWINCH
5861 * # echo `(trap)` <--- in subshell in subshell - output
5862 * trap -- 'echo Ho' SIGWINCH
5863 * # echo `true | (trap)` <--- in subshell in subshell in subshell - output!
5864 * trap -- 'echo Ho' SIGWINCH
5865 *
5866 * The rules when to forget and when to not forget traps
5867 * get really complex and nonsensical.
5868 *
5869 * Our solution: ONLY bare $(trap) or `trap` is special.
5870 */
5871 s = skip_whitespace(s);
5872 if (strncmp(s, "trap", 4) == 0
5873 && skip_whitespace(s + 4)[0] == '\0'
5874 ) {
5875 static const char *const argv[] = { NULL, NULL };
5876 builtin_trap((char**)argv);
5877 exit(0); /* not _exit() - we need to fflush */
5878 }
5879# if BB_MMU
5880 reset_traps_to_defaults();
5881 parse_and_run_string(s);
5882 _exit(G.last_exitcode);
5883# else
5884 /* We re-execute after vfork on NOMMU. This makes this script safe:
5885 * yes "0123456789012345678901234567890" | dd bs=32 count=64k >BIG
5886 * huge=`cat BIG` # was blocking here forever
5887 * echo OK
5888 */
5889 re_execute_shell(&to_free,
5890 s,
5891 G.global_argv[0],
5892 G.global_argv + 1,
5893 NULL);
5894# endif
5895 }
5896
5897 /* parent */
5898 *pid_p = pid;
5899# if ENABLE_HUSH_FAST
5900 G.count_SIGCHLD++;
5901//bb_error_msg("[%d] fork in generate_stream_from_string:"
5902// " G.count_SIGCHLD:%d G.handled_SIGCHLD:%d",
5903// getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
5904# endif
5905 enable_restore_tty_pgrp_on_exit();
5906# if !BB_MMU
5907 free(to_free);
5908# endif
5909 close(channel[1]);
5910 close_on_exec_on(channel[0]);
5911 return xfdopen_for_read(channel[0]);
5912}
5913
5914/* Return code is exit status of the process that is run. */
5915static int process_command_subs(o_string *dest, const char *s)
5916{
5917 FILE *fp;
5918 struct in_str pipe_str;
5919 pid_t pid;
5920 int status, ch, eol_cnt;
5921
5922 fp = generate_stream_from_string(s, &pid);
5923
5924 /* Now send results of command back into original context */
5925 setup_file_in_str(&pipe_str, fp);
5926 eol_cnt = 0;
5927 while ((ch = i_getch(&pipe_str)) != EOF) {
5928 if (ch == '\n') {
5929 eol_cnt++;
5930 continue;
5931 }
5932 while (eol_cnt) {
5933 o_addchr(dest, '\n');
5934 eol_cnt--;
5935 }
5936 o_addQchr(dest, ch);
5937 }
5938
5939 debug_printf("done reading from `cmd` pipe, closing it\n");
5940 fclose(fp);
5941 /* We need to extract exitcode. Test case
5942 * "true; echo `sleep 1; false` $?"
5943 * should print 1 */
5944 safe_waitpid(pid, &status, 0);
5945 debug_printf("child exited. returning its exitcode:%d\n", WEXITSTATUS(status));
5946 return WEXITSTATUS(status);
5947}
5948#endif /* ENABLE_HUSH_TICK */
5949
5950
5951static void setup_heredoc(struct redir_struct *redir)
5952{
5953 struct fd_pair pair;
5954 pid_t pid;
5955 int len, written;
5956 /* the _body_ of heredoc (misleading field name) */
5957 const char *heredoc = redir->rd_filename;
5958 char *expanded;
5959#if !BB_MMU
5960 char **to_free;
5961#endif
5962
5963 expanded = NULL;
5964 if (!(redir->rd_dup & HEREDOC_QUOTED)) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005965 expanded = encode_then_expand_string(heredoc, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005966 if (expanded)
5967 heredoc = expanded;
5968 }
5969 len = strlen(heredoc);
5970
5971 close(redir->rd_fd); /* often saves dup2+close in xmove_fd */
5972 xpiped_pair(pair);
5973 xmove_fd(pair.rd, redir->rd_fd);
5974
5975 /* Try writing without forking. Newer kernels have
5976 * dynamically growing pipes. Must use non-blocking write! */
5977 ndelay_on(pair.wr);
5978 while (1) {
5979 written = write(pair.wr, heredoc, len);
5980 if (written <= 0)
5981 break;
5982 len -= written;
5983 if (len == 0) {
5984 close(pair.wr);
5985 free(expanded);
5986 return;
5987 }
5988 heredoc += written;
5989 }
5990 ndelay_off(pair.wr);
5991
5992 /* Okay, pipe buffer was not big enough */
5993 /* Note: we must not create a stray child (bastard? :)
5994 * for the unsuspecting parent process. Child creates a grandchild
5995 * and exits before parent execs the process which consumes heredoc
5996 * (that exec happens after we return from this function) */
5997#if !BB_MMU
5998 to_free = NULL;
5999#endif
6000 pid = xvfork();
6001 if (pid == 0) {
6002 /* child */
6003 disable_restore_tty_pgrp_on_exit();
6004 pid = BB_MMU ? xfork() : xvfork();
6005 if (pid != 0)
6006 _exit(0);
6007 /* grandchild */
6008 close(redir->rd_fd); /* read side of the pipe */
6009#if BB_MMU
6010 full_write(pair.wr, heredoc, len); /* may loop or block */
6011 _exit(0);
6012#else
6013 /* Delegate blocking writes to another process */
6014 xmove_fd(pair.wr, STDOUT_FILENO);
6015 re_execute_shell(&to_free, heredoc, NULL, NULL, NULL);
6016#endif
6017 }
6018 /* parent */
6019#if ENABLE_HUSH_FAST
6020 G.count_SIGCHLD++;
6021//bb_error_msg("[%d] fork in setup_heredoc: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6022#endif
6023 enable_restore_tty_pgrp_on_exit();
6024#if !BB_MMU
6025 free(to_free);
6026#endif
6027 close(pair.wr);
6028 free(expanded);
6029 wait(NULL); /* wait till child has died */
6030}
6031
6032/* squirrel != NULL means we squirrel away copies of stdin, stdout,
6033 * and stderr if they are redirected. */
6034static int setup_redirects(struct command *prog, int squirrel[])
6035{
6036 int openfd, mode;
6037 struct redir_struct *redir;
6038
6039 for (redir = prog->redirects; redir; redir = redir->next) {
6040 if (redir->rd_type == REDIRECT_HEREDOC2) {
6041 /* rd_fd<<HERE case */
6042 if (squirrel && redir->rd_fd < 3
6043 && squirrel[redir->rd_fd] < 0
6044 ) {
6045 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6046 }
6047 /* for REDIRECT_HEREDOC2, rd_filename holds _contents_
6048 * of the heredoc */
6049 debug_printf_parse("set heredoc '%s'\n",
6050 redir->rd_filename);
6051 setup_heredoc(redir);
6052 continue;
6053 }
6054
6055 if (redir->rd_dup == REDIRFD_TO_FILE) {
6056 /* rd_fd<*>file case (<*> is <,>,>>,<>) */
6057 char *p;
6058 if (redir->rd_filename == NULL) {
6059 /* Something went wrong in the parse.
6060 * Pretend it didn't happen */
6061 bb_error_msg("bug in redirect parse");
6062 continue;
6063 }
6064 mode = redir_table[redir->rd_type].mode;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02006065 p = expand_string_to_string(redir->rd_filename, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006066 openfd = open_or_warn(p, mode);
6067 free(p);
6068 if (openfd < 0) {
6069 /* this could get lost if stderr has been redirected, but
6070 * bash and ash both lose it as well (though zsh doesn't!) */
6071//what the above comment tries to say?
6072 return 1;
6073 }
6074 } else {
6075 /* rd_fd<*>rd_dup or rd_fd<*>- cases */
6076 openfd = redir->rd_dup;
6077 }
6078
6079 if (openfd != redir->rd_fd) {
6080 if (squirrel && redir->rd_fd < 3
6081 && squirrel[redir->rd_fd] < 0
6082 ) {
6083 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6084 }
6085 if (openfd == REDIRFD_CLOSE) {
6086 /* "n>-" means "close me" */
6087 close(redir->rd_fd);
6088 } else {
6089 xdup2(openfd, redir->rd_fd);
6090 if (redir->rd_dup == REDIRFD_TO_FILE)
6091 close(openfd);
6092 }
6093 }
6094 }
6095 return 0;
6096}
6097
6098static void restore_redirects(int squirrel[])
6099{
6100 int i, fd;
6101 for (i = 0; i < 3; i++) {
6102 fd = squirrel[i];
6103 if (fd != -1) {
6104 /* We simply die on error */
6105 xmove_fd(fd, i);
6106 }
6107 }
6108}
6109
6110static char *find_in_path(const char *arg)
6111{
6112 char *ret = NULL;
6113 const char *PATH = get_local_var_value("PATH");
6114
6115 if (!PATH)
6116 return NULL;
6117
6118 while (1) {
6119 const char *end = strchrnul(PATH, ':');
6120 int sz = end - PATH; /* must be int! */
6121
6122 free(ret);
6123 if (sz != 0) {
6124 ret = xasprintf("%.*s/%s", sz, PATH, arg);
6125 } else {
6126 /* We have xxx::yyyy in $PATH,
6127 * it means "use current dir" */
6128 ret = xstrdup(arg);
6129 }
6130 if (access(ret, F_OK) == 0)
6131 break;
6132
6133 if (*end == '\0') {
6134 free(ret);
6135 return NULL;
6136 }
6137 PATH = end + 1;
6138 }
6139
6140 return ret;
6141}
6142
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006143static const struct built_in_command *find_builtin_helper(const char *name,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006144 const struct built_in_command *x,
6145 const struct built_in_command *end)
6146{
6147 while (x != end) {
6148 if (strcmp(name, x->b_cmd) != 0) {
6149 x++;
6150 continue;
6151 }
6152 debug_printf_exec("found builtin '%s'\n", name);
6153 return x;
6154 }
6155 return NULL;
6156}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006157static const struct built_in_command *find_builtin1(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006158{
6159 return find_builtin_helper(name, bltins1, &bltins1[ARRAY_SIZE(bltins1)]);
6160}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006161static const struct built_in_command *find_builtin(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006162{
6163 const struct built_in_command *x = find_builtin1(name);
6164 if (x)
6165 return x;
6166 return find_builtin_helper(name, bltins2, &bltins2[ARRAY_SIZE(bltins2)]);
6167}
6168
6169#if ENABLE_HUSH_FUNCTIONS
6170static struct function **find_function_slot(const char *name)
6171{
6172 struct function **funcpp = &G.top_func;
6173 while (*funcpp) {
6174 if (strcmp(name, (*funcpp)->name) == 0) {
6175 break;
6176 }
6177 funcpp = &(*funcpp)->next;
6178 }
6179 return funcpp;
6180}
6181
6182static const struct function *find_function(const char *name)
6183{
6184 const struct function *funcp = *find_function_slot(name);
6185 if (funcp)
6186 debug_printf_exec("found function '%s'\n", name);
6187 return funcp;
6188}
6189
6190/* Note: takes ownership on name ptr */
6191static struct function *new_function(char *name)
6192{
6193 struct function **funcpp = find_function_slot(name);
6194 struct function *funcp = *funcpp;
6195
6196 if (funcp != NULL) {
6197 struct command *cmd = funcp->parent_cmd;
6198 debug_printf_exec("func %p parent_cmd %p\n", funcp, cmd);
6199 if (!cmd) {
6200 debug_printf_exec("freeing & replacing function '%s'\n", funcp->name);
6201 free(funcp->name);
6202 /* Note: if !funcp->body, do not free body_as_string!
6203 * This is a special case of "-F name body" function:
6204 * body_as_string was not malloced! */
6205 if (funcp->body) {
6206 free_pipe_list(funcp->body);
6207# if !BB_MMU
6208 free(funcp->body_as_string);
6209# endif
6210 }
6211 } else {
6212 debug_printf_exec("reinserting in tree & replacing function '%s'\n", funcp->name);
6213 cmd->argv[0] = funcp->name;
6214 cmd->group = funcp->body;
6215# if !BB_MMU
6216 cmd->group_as_string = funcp->body_as_string;
6217# endif
6218 }
6219 } else {
6220 debug_printf_exec("remembering new function '%s'\n", name);
6221 funcp = *funcpp = xzalloc(sizeof(*funcp));
6222 /*funcp->next = NULL;*/
6223 }
6224
6225 funcp->name = name;
6226 return funcp;
6227}
6228
6229static void unset_func(const char *name)
6230{
6231 struct function **funcpp = find_function_slot(name);
6232 struct function *funcp = *funcpp;
6233
6234 if (funcp != NULL) {
6235 debug_printf_exec("freeing function '%s'\n", funcp->name);
6236 *funcpp = funcp->next;
6237 /* funcp is unlinked now, deleting it.
6238 * Note: if !funcp->body, the function was created by
6239 * "-F name body", do not free ->body_as_string
6240 * and ->name as they were not malloced. */
6241 if (funcp->body) {
6242 free_pipe_list(funcp->body);
6243 free(funcp->name);
6244# if !BB_MMU
6245 free(funcp->body_as_string);
6246# endif
6247 }
6248 free(funcp);
6249 }
6250}
6251
6252# if BB_MMU
6253#define exec_function(to_free, funcp, argv) \
6254 exec_function(funcp, argv)
6255# endif
6256static void exec_function(char ***to_free,
6257 const struct function *funcp,
6258 char **argv) NORETURN;
6259static void exec_function(char ***to_free,
6260 const struct function *funcp,
6261 char **argv)
6262{
6263# if BB_MMU
6264 int n = 1;
6265
6266 argv[0] = G.global_argv[0];
6267 G.global_argv = argv;
6268 while (*++argv)
6269 n++;
6270 G.global_argc = n;
6271 /* On MMU, funcp->body is always non-NULL */
6272 n = run_list(funcp->body);
6273 fflush_all();
6274 _exit(n);
6275# else
6276 re_execute_shell(to_free,
6277 funcp->body_as_string,
6278 G.global_argv[0],
6279 argv + 1,
6280 NULL);
6281# endif
6282}
6283
6284static int run_function(const struct function *funcp, char **argv)
6285{
6286 int rc;
6287 save_arg_t sv;
6288 smallint sv_flg;
6289
6290 save_and_replace_G_args(&sv, argv);
6291
6292 /* "we are in function, ok to use return" */
6293 sv_flg = G.flag_return_in_progress;
6294 G.flag_return_in_progress = -1;
6295# if ENABLE_HUSH_LOCAL
6296 G.func_nest_level++;
6297# endif
6298
6299 /* On MMU, funcp->body is always non-NULL */
6300# if !BB_MMU
6301 if (!funcp->body) {
6302 /* Function defined by -F */
6303 parse_and_run_string(funcp->body_as_string);
6304 rc = G.last_exitcode;
6305 } else
6306# endif
6307 {
6308 rc = run_list(funcp->body);
6309 }
6310
6311# if ENABLE_HUSH_LOCAL
6312 {
6313 struct variable *var;
6314 struct variable **var_pp;
6315
6316 var_pp = &G.top_var;
6317 while ((var = *var_pp) != NULL) {
6318 if (var->func_nest_level < G.func_nest_level) {
6319 var_pp = &var->next;
6320 continue;
6321 }
6322 /* Unexport */
6323 if (var->flg_export)
6324 bb_unsetenv(var->varstr);
6325 /* Remove from global list */
6326 *var_pp = var->next;
6327 /* Free */
6328 if (!var->max_len)
6329 free(var->varstr);
6330 free(var);
6331 }
6332 G.func_nest_level--;
6333 }
6334# endif
6335 G.flag_return_in_progress = sv_flg;
6336
6337 restore_G_args(&sv, argv);
6338
6339 return rc;
6340}
6341#endif /* ENABLE_HUSH_FUNCTIONS */
6342
6343
6344#if BB_MMU
6345#define exec_builtin(to_free, x, argv) \
6346 exec_builtin(x, argv)
6347#else
6348#define exec_builtin(to_free, x, argv) \
6349 exec_builtin(to_free, argv)
6350#endif
6351static void exec_builtin(char ***to_free,
6352 const struct built_in_command *x,
6353 char **argv) NORETURN;
6354static void exec_builtin(char ***to_free,
6355 const struct built_in_command *x,
6356 char **argv)
6357{
6358#if BB_MMU
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006359 int rcode;
6360 fflush_all();
6361 rcode = x->b_function(argv);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006362 fflush_all();
6363 _exit(rcode);
6364#else
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006365 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006366 /* On NOMMU, we must never block!
6367 * Example: { sleep 99 | read line; } & echo Ok
6368 */
6369 re_execute_shell(to_free,
6370 argv[0],
6371 G.global_argv[0],
6372 G.global_argv + 1,
6373 argv);
6374#endif
6375}
6376
6377
6378static void execvp_or_die(char **argv) NORETURN;
6379static void execvp_or_die(char **argv)
6380{
6381 debug_printf_exec("execing '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006382 /* Don't propagate SIG_IGN to the child */
6383 if (SPECIAL_JOBSTOP_SIGS != 0)
6384 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006385 execvp(argv[0], argv);
6386 bb_perror_msg("can't execute '%s'", argv[0]);
6387 _exit(127); /* bash compat */
6388}
6389
6390#if ENABLE_HUSH_MODE_X
6391static void dump_cmd_in_x_mode(char **argv)
6392{
6393 if (G_x_mode && argv) {
6394 /* We want to output the line in one write op */
6395 char *buf, *p;
6396 int len;
6397 int n;
6398
6399 len = 3;
6400 n = 0;
6401 while (argv[n])
6402 len += strlen(argv[n++]) + 1;
6403 buf = xmalloc(len);
6404 buf[0] = '+';
6405 p = buf + 1;
6406 n = 0;
6407 while (argv[n])
6408 p += sprintf(p, " %s", argv[n++]);
6409 *p++ = '\n';
6410 *p = '\0';
6411 fputs(buf, stderr);
6412 free(buf);
6413 }
6414}
6415#else
6416# define dump_cmd_in_x_mode(argv) ((void)0)
6417#endif
6418
6419#if BB_MMU
6420#define pseudo_exec_argv(nommu_save, argv, assignment_cnt, argv_expanded) \
6421 pseudo_exec_argv(argv, assignment_cnt, argv_expanded)
6422#define pseudo_exec(nommu_save, command, argv_expanded) \
6423 pseudo_exec(command, argv_expanded)
6424#endif
6425
6426/* Called after [v]fork() in run_pipe, or from builtin_exec.
6427 * Never returns.
6428 * Don't exit() here. If you don't exec, use _exit instead.
6429 * The at_exit handlers apparently confuse the calling process,
6430 * in particular stdin handling. Not sure why? -- because of vfork! (vda) */
6431static void pseudo_exec_argv(nommu_save_t *nommu_save,
6432 char **argv, int assignment_cnt,
6433 char **argv_expanded) NORETURN;
6434static NOINLINE void pseudo_exec_argv(nommu_save_t *nommu_save,
6435 char **argv, int assignment_cnt,
6436 char **argv_expanded)
6437{
6438 char **new_env;
6439
6440 new_env = expand_assignments(argv, assignment_cnt);
6441 dump_cmd_in_x_mode(new_env);
6442
6443 if (!argv[assignment_cnt]) {
6444 /* Case when we are here: ... | var=val | ...
6445 * (note that we do not exit early, i.e., do not optimize out
6446 * expand_assignments(): think about ... | var=`sleep 1` | ...
6447 */
6448 free_strings(new_env);
6449 _exit(EXIT_SUCCESS);
6450 }
6451
6452#if BB_MMU
6453 set_vars_and_save_old(new_env);
6454 free(new_env); /* optional */
6455 /* we can also destroy set_vars_and_save_old's return value,
6456 * to save memory */
6457#else
6458 nommu_save->new_env = new_env;
6459 nommu_save->old_vars = set_vars_and_save_old(new_env);
6460#endif
6461
6462 if (argv_expanded) {
6463 argv = argv_expanded;
6464 } else {
6465 argv = expand_strvec_to_strvec(argv + assignment_cnt);
6466#if !BB_MMU
6467 nommu_save->argv = argv;
6468#endif
6469 }
6470 dump_cmd_in_x_mode(argv);
6471
6472#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6473 if (strchr(argv[0], '/') != NULL)
6474 goto skip;
6475#endif
6476
6477 /* Check if the command matches any of the builtins.
6478 * Depending on context, this might be redundant. But it's
6479 * easier to waste a few CPU cycles than it is to figure out
6480 * if this is one of those cases.
6481 */
6482 {
6483 /* On NOMMU, it is more expensive to re-execute shell
6484 * just in order to run echo or test builtin.
6485 * It's better to skip it here and run corresponding
6486 * non-builtin later. */
6487 const struct built_in_command *x;
6488 x = BB_MMU ? find_builtin(argv[0]) : find_builtin1(argv[0]);
6489 if (x) {
6490 exec_builtin(&nommu_save->argv_from_re_execing, x, argv);
6491 }
6492 }
6493#if ENABLE_HUSH_FUNCTIONS
6494 /* Check if the command matches any functions */
6495 {
6496 const struct function *funcp = find_function(argv[0]);
6497 if (funcp) {
6498 exec_function(&nommu_save->argv_from_re_execing, funcp, argv);
6499 }
6500 }
6501#endif
6502
6503#if ENABLE_FEATURE_SH_STANDALONE
6504 /* Check if the command matches any busybox applets */
6505 {
6506 int a = find_applet_by_name(argv[0]);
6507 if (a >= 0) {
6508# if BB_MMU /* see above why on NOMMU it is not allowed */
6509 if (APPLET_IS_NOEXEC(a)) {
6510 debug_printf_exec("running applet '%s'\n", argv[0]);
6511 run_applet_no_and_exit(a, argv);
6512 }
6513# endif
6514 /* Re-exec ourselves */
6515 debug_printf_exec("re-execing applet '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006516 /* Don't propagate SIG_IGN to the child */
6517 if (SPECIAL_JOBSTOP_SIGS != 0)
6518 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006519 execv(bb_busybox_exec_path, argv);
6520 /* If they called chroot or otherwise made the binary no longer
6521 * executable, fall through */
6522 }
6523 }
6524#endif
6525
6526#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6527 skip:
6528#endif
6529 execvp_or_die(argv);
6530}
6531
6532/* Called after [v]fork() in run_pipe
6533 */
6534static void pseudo_exec(nommu_save_t *nommu_save,
6535 struct command *command,
6536 char **argv_expanded) NORETURN;
6537static void pseudo_exec(nommu_save_t *nommu_save,
6538 struct command *command,
6539 char **argv_expanded)
6540{
6541 if (command->argv) {
6542 pseudo_exec_argv(nommu_save, command->argv,
6543 command->assignment_cnt, argv_expanded);
6544 }
6545
6546 if (command->group) {
6547 /* Cases when we are here:
6548 * ( list )
6549 * { list } &
6550 * ... | ( list ) | ...
6551 * ... | { list } | ...
6552 */
6553#if BB_MMU
6554 int rcode;
6555 debug_printf_exec("pseudo_exec: run_list\n");
6556 reset_traps_to_defaults();
6557 rcode = run_list(command->group);
6558 /* OK to leak memory by not calling free_pipe_list,
6559 * since this process is about to exit */
6560 _exit(rcode);
6561#else
6562 re_execute_shell(&nommu_save->argv_from_re_execing,
6563 command->group_as_string,
6564 G.global_argv[0],
6565 G.global_argv + 1,
6566 NULL);
6567#endif
6568 }
6569
6570 /* Case when we are here: ... | >file */
6571 debug_printf_exec("pseudo_exec'ed null command\n");
6572 _exit(EXIT_SUCCESS);
6573}
6574
6575#if ENABLE_HUSH_JOB
6576static const char *get_cmdtext(struct pipe *pi)
6577{
6578 char **argv;
6579 char *p;
6580 int len;
6581
6582 /* This is subtle. ->cmdtext is created only on first backgrounding.
6583 * (Think "cat, <ctrl-z>, fg, <ctrl-z>, fg, <ctrl-z>...." here...)
6584 * On subsequent bg argv is trashed, but we won't use it */
6585 if (pi->cmdtext)
6586 return pi->cmdtext;
6587 argv = pi->cmds[0].argv;
6588 if (!argv || !argv[0]) {
6589 pi->cmdtext = xzalloc(1);
6590 return pi->cmdtext;
6591 }
6592
6593 len = 0;
6594 do {
6595 len += strlen(*argv) + 1;
6596 } while (*++argv);
6597 p = xmalloc(len);
6598 pi->cmdtext = p;
6599 argv = pi->cmds[0].argv;
6600 do {
6601 len = strlen(*argv);
6602 memcpy(p, *argv, len);
6603 p += len;
6604 *p++ = ' ';
6605 } while (*++argv);
6606 p[-1] = '\0';
6607 return pi->cmdtext;
6608}
6609
6610static void insert_bg_job(struct pipe *pi)
6611{
6612 struct pipe *job, **jobp;
6613 int i;
6614
6615 /* Linear search for the ID of the job to use */
6616 pi->jobid = 1;
6617 for (job = G.job_list; job; job = job->next)
6618 if (job->jobid >= pi->jobid)
6619 pi->jobid = job->jobid + 1;
6620
6621 /* Add job to the list of running jobs */
6622 jobp = &G.job_list;
6623 while ((job = *jobp) != NULL)
6624 jobp = &job->next;
6625 job = *jobp = xmalloc(sizeof(*job));
6626
6627 *job = *pi; /* physical copy */
6628 job->next = NULL;
6629 job->cmds = xzalloc(sizeof(pi->cmds[0]) * pi->num_cmds);
6630 /* Cannot copy entire pi->cmds[] vector! This causes double frees */
6631 for (i = 0; i < pi->num_cmds; i++) {
6632 job->cmds[i].pid = pi->cmds[i].pid;
6633 /* all other fields are not used and stay zero */
6634 }
6635 job->cmdtext = xstrdup(get_cmdtext(pi));
6636
6637 if (G_interactive_fd)
6638 printf("[%d] %d %s\n", job->jobid, job->cmds[0].pid, job->cmdtext);
6639 G.last_jobid = job->jobid;
6640}
6641
6642static void remove_bg_job(struct pipe *pi)
6643{
6644 struct pipe *prev_pipe;
6645
6646 if (pi == G.job_list) {
6647 G.job_list = pi->next;
6648 } else {
6649 prev_pipe = G.job_list;
6650 while (prev_pipe->next != pi)
6651 prev_pipe = prev_pipe->next;
6652 prev_pipe->next = pi->next;
6653 }
6654 if (G.job_list)
6655 G.last_jobid = G.job_list->jobid;
6656 else
6657 G.last_jobid = 0;
6658}
6659
6660/* Remove a backgrounded job */
6661static void delete_finished_bg_job(struct pipe *pi)
6662{
6663 remove_bg_job(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006664 free_pipe(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006665}
6666#endif /* JOB */
6667
6668/* Check to see if any processes have exited -- if they
6669 * have, figure out why and see if a job has completed */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02006670static int checkjobs(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006671{
6672 int attributes;
6673 int status;
6674#if ENABLE_HUSH_JOB
6675 struct pipe *pi;
6676#endif
6677 pid_t childpid;
6678 int rcode = 0;
6679
6680 debug_printf_jobs("checkjobs %p\n", fg_pipe);
6681
6682 attributes = WUNTRACED;
6683 if (fg_pipe == NULL)
6684 attributes |= WNOHANG;
6685
6686 errno = 0;
6687#if ENABLE_HUSH_FAST
6688 if (G.handled_SIGCHLD == G.count_SIGCHLD) {
6689//bb_error_msg("[%d] checkjobs: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d children?:%d fg_pipe:%p",
6690//getpid(), G.count_SIGCHLD, G.handled_SIGCHLD, G.we_have_children, fg_pipe);
6691 /* There was neither fork nor SIGCHLD since last waitpid */
6692 /* Avoid doing waitpid syscall if possible */
6693 if (!G.we_have_children) {
6694 errno = ECHILD;
6695 return -1;
6696 }
6697 if (fg_pipe == NULL) { /* is WNOHANG set? */
6698 /* We have children, but they did not exit
6699 * or stop yet (we saw no SIGCHLD) */
6700 return 0;
6701 }
6702 /* else: !WNOHANG, waitpid will block, can't short-circuit */
6703 }
6704#endif
6705
6706/* Do we do this right?
6707 * bash-3.00# sleep 20 | false
6708 * <ctrl-Z pressed>
6709 * [3]+ Stopped sleep 20 | false
6710 * bash-3.00# echo $?
6711 * 1 <========== bg pipe is not fully done, but exitcode is already known!
6712 * [hush 1.14.0: yes we do it right]
6713 */
6714 wait_more:
6715 while (1) {
6716 int i;
6717 int dead;
6718
6719#if ENABLE_HUSH_FAST
6720 i = G.count_SIGCHLD;
6721#endif
6722 childpid = waitpid(-1, &status, attributes);
6723 if (childpid <= 0) {
6724 if (childpid && errno != ECHILD)
6725 bb_perror_msg("waitpid");
6726#if ENABLE_HUSH_FAST
6727 else { /* Until next SIGCHLD, waitpid's are useless */
6728 G.we_have_children = (childpid == 0);
6729 G.handled_SIGCHLD = i;
6730//bb_error_msg("[%d] checkjobs: waitpid returned <= 0, G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6731 }
6732#endif
6733 break;
6734 }
6735 dead = WIFEXITED(status) || WIFSIGNALED(status);
6736
6737#if DEBUG_JOBS
6738 if (WIFSTOPPED(status))
6739 debug_printf_jobs("pid %d stopped by sig %d (exitcode %d)\n",
6740 childpid, WSTOPSIG(status), WEXITSTATUS(status));
6741 if (WIFSIGNALED(status))
6742 debug_printf_jobs("pid %d killed by sig %d (exitcode %d)\n",
6743 childpid, WTERMSIG(status), WEXITSTATUS(status));
6744 if (WIFEXITED(status))
6745 debug_printf_jobs("pid %d exited, exitcode %d\n",
6746 childpid, WEXITSTATUS(status));
6747#endif
6748 /* Were we asked to wait for fg pipe? */
6749 if (fg_pipe) {
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006750 i = fg_pipe->num_cmds;
6751 while (--i >= 0) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006752 debug_printf_jobs("check pid %d\n", fg_pipe->cmds[i].pid);
6753 if (fg_pipe->cmds[i].pid != childpid)
6754 continue;
6755 if (dead) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006756 int ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006757 fg_pipe->cmds[i].pid = 0;
6758 fg_pipe->alive_cmds--;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006759 ex = WEXITSTATUS(status);
6760 /* bash prints killer signal's name for *last*
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006761 * process in pipe (prints just newline for SIGINT/SIGPIPE).
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006762 * Mimic this. Example: "sleep 5" + (^\ or kill -QUIT)
6763 */
6764 if (WIFSIGNALED(status)) {
6765 int sig = WTERMSIG(status);
6766 if (i == fg_pipe->num_cmds-1)
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006767 /* TODO: use strsignal() instead for bash compat? but that's bloat... */
6768 printf("%s\n", sig == SIGINT || sig == SIGPIPE ? "" : get_signame(sig));
6769 /* TODO: if (WCOREDUMP(status)) + " (core dumped)"; */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006770 /* TODO: MIPS has 128 sigs (1..128), what if sig==128 here?
6771 * Maybe we need to use sig | 128? */
6772 ex = sig + 128;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006773 }
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006774 fg_pipe->cmds[i].cmd_exitcode = ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006775 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006776 fg_pipe->stopped_cmds++;
6777 }
6778 debug_printf_jobs("fg_pipe: alive_cmds %d stopped_cmds %d\n",
6779 fg_pipe->alive_cmds, fg_pipe->stopped_cmds);
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006780 if (fg_pipe->alive_cmds == fg_pipe->stopped_cmds) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006781 /* All processes in fg pipe have exited or stopped */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006782 i = fg_pipe->num_cmds;
6783 while (--i >= 0) {
6784 rcode = fg_pipe->cmds[i].cmd_exitcode;
6785 /* usually last process gives overall exitstatus,
6786 * but with "set -o pipefail", last *failed* process does */
6787 if (G.o_opt[OPT_O_PIPEFAIL] == 0 || rcode != 0)
6788 break;
6789 }
6790 IF_HAS_KEYWORDS(if (fg_pipe->pi_inverted) rcode = !rcode;)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006791/* Note: *non-interactive* bash does not continue if all processes in fg pipe
6792 * are stopped. Testcase: "cat | cat" in a script (not on command line!)
6793 * and "killall -STOP cat" */
6794 if (G_interactive_fd) {
6795#if ENABLE_HUSH_JOB
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006796 if (fg_pipe->alive_cmds != 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006797 insert_bg_job(fg_pipe);
6798#endif
6799 return rcode;
6800 }
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006801 if (fg_pipe->alive_cmds == 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006802 return rcode;
6803 }
6804 /* There are still running processes in the fg pipe */
6805 goto wait_more; /* do waitpid again */
6806 }
6807 /* it wasnt fg_pipe, look for process in bg pipes */
6808 }
6809
6810#if ENABLE_HUSH_JOB
6811 /* We asked to wait for bg or orphaned children */
6812 /* No need to remember exitcode in this case */
6813 for (pi = G.job_list; pi; pi = pi->next) {
6814 for (i = 0; i < pi->num_cmds; i++) {
6815 if (pi->cmds[i].pid == childpid)
6816 goto found_pi_and_prognum;
6817 }
6818 }
6819 /* Happens when shell is used as init process (init=/bin/sh) */
6820 debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
6821 continue; /* do waitpid again */
6822
6823 found_pi_and_prognum:
6824 if (dead) {
6825 /* child exited */
6826 pi->cmds[i].pid = 0;
6827 pi->alive_cmds--;
6828 if (!pi->alive_cmds) {
6829 if (G_interactive_fd)
6830 printf(JOB_STATUS_FORMAT, pi->jobid,
6831 "Done", pi->cmdtext);
6832 delete_finished_bg_job(pi);
6833 }
6834 } else {
6835 /* child stopped */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006836 pi->stopped_cmds++;
6837 }
6838#endif
6839 } /* while (waitpid succeeds)... */
6840
6841 return rcode;
6842}
6843
6844#if ENABLE_HUSH_JOB
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006845static int checkjobs_and_fg_shell(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006846{
6847 pid_t p;
6848 int rcode = checkjobs(fg_pipe);
6849 if (G_saved_tty_pgrp) {
6850 /* Job finished, move the shell to the foreground */
6851 p = getpgrp(); /* our process group id */
6852 debug_printf_jobs("fg'ing ourself: getpgrp()=%d\n", (int)p);
6853 tcsetpgrp(G_interactive_fd, p);
6854 }
6855 return rcode;
6856}
6857#endif
6858
6859/* Start all the jobs, but don't wait for anything to finish.
6860 * See checkjobs().
6861 *
6862 * Return code is normally -1, when the caller has to wait for children
6863 * to finish to determine the exit status of the pipe. If the pipe
6864 * is a simple builtin command, however, the action is done by the
6865 * time run_pipe returns, and the exit code is provided as the
6866 * return value.
6867 *
6868 * Returns -1 only if started some children. IOW: we have to
6869 * mask out retvals of builtins etc with 0xff!
6870 *
6871 * The only case when we do not need to [v]fork is when the pipe
6872 * is single, non-backgrounded, non-subshell command. Examples:
6873 * cmd ; ... { list } ; ...
6874 * cmd && ... { list } && ...
6875 * cmd || ... { list } || ...
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01006876 * If it is, then we can run cmd as a builtin, NOFORK,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006877 * or (if SH_STANDALONE) an applet, and we can run the { list }
6878 * with run_list. If it isn't one of these, we fork and exec cmd.
6879 *
6880 * Cases when we must fork:
6881 * non-single: cmd | cmd
6882 * backgrounded: cmd & { list } &
6883 * subshell: ( list ) [&]
6884 */
6885#if !ENABLE_HUSH_MODE_X
Denys Vlasenko26777aa2010-11-22 23:49:10 +01006886#define redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel, argv_expanded) \
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006887 redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel)
6888#endif
6889static int redirect_and_varexp_helper(char ***new_env_p,
6890 struct variable **old_vars_p,
6891 struct command *command,
6892 int squirrel[3],
6893 char **argv_expanded)
6894{
6895 /* setup_redirects acts on file descriptors, not FILEs.
6896 * This is perfect for work that comes after exec().
6897 * Is it really safe for inline use? Experimentally,
6898 * things seem to work. */
6899 int rcode = setup_redirects(command, squirrel);
6900 if (rcode == 0) {
6901 char **new_env = expand_assignments(command->argv, command->assignment_cnt);
6902 *new_env_p = new_env;
6903 dump_cmd_in_x_mode(new_env);
6904 dump_cmd_in_x_mode(argv_expanded);
6905 if (old_vars_p)
6906 *old_vars_p = set_vars_and_save_old(new_env);
6907 }
6908 return rcode;
6909}
6910static NOINLINE int run_pipe(struct pipe *pi)
6911{
6912 static const char *const null_ptr = NULL;
6913
6914 int cmd_no;
6915 int next_infd;
6916 struct command *command;
6917 char **argv_expanded;
6918 char **argv;
6919 /* it is not always needed, but we aim to smaller code */
6920 int squirrel[] = { -1, -1, -1 };
6921 int rcode;
6922
6923 debug_printf_exec("run_pipe start: members:%d\n", pi->num_cmds);
6924 debug_enter();
6925
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02006926 /* Testcase: set -- q w e; (IFS='' echo "$*"; IFS=''; echo "$*"); echo "$*"
6927 * Result should be 3 lines: q w e, qwe, q w e
6928 */
6929 G.ifs = get_local_var_value("IFS");
6930 if (!G.ifs)
6931 G.ifs = defifs;
6932
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006933 IF_HUSH_JOB(pi->pgrp = -1;)
6934 pi->stopped_cmds = 0;
6935 command = &pi->cmds[0];
6936 argv_expanded = NULL;
6937
6938 if (pi->num_cmds != 1
6939 || pi->followup == PIPE_BG
6940 || command->cmd_type == CMD_SUBSHELL
6941 ) {
6942 goto must_fork;
6943 }
6944
6945 pi->alive_cmds = 1;
6946
6947 debug_printf_exec(": group:%p argv:'%s'\n",
6948 command->group, command->argv ? command->argv[0] : "NONE");
6949
6950 if (command->group) {
6951#if ENABLE_HUSH_FUNCTIONS
6952 if (command->cmd_type == CMD_FUNCDEF) {
6953 /* "executing" func () { list } */
6954 struct function *funcp;
6955
6956 funcp = new_function(command->argv[0]);
6957 /* funcp->name is already set to argv[0] */
6958 funcp->body = command->group;
6959# if !BB_MMU
6960 funcp->body_as_string = command->group_as_string;
6961 command->group_as_string = NULL;
6962# endif
6963 command->group = NULL;
6964 command->argv[0] = NULL;
6965 debug_printf_exec("cmd %p has child func at %p\n", command, funcp);
6966 funcp->parent_cmd = command;
6967 command->child_func = funcp;
6968
6969 debug_printf_exec("run_pipe: return EXIT_SUCCESS\n");
6970 debug_leave();
6971 return EXIT_SUCCESS;
6972 }
6973#endif
6974 /* { list } */
6975 debug_printf("non-subshell group\n");
6976 rcode = 1; /* exitcode if redir failed */
6977 if (setup_redirects(command, squirrel) == 0) {
6978 debug_printf_exec(": run_list\n");
6979 rcode = run_list(command->group) & 0xff;
6980 }
6981 restore_redirects(squirrel);
6982 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
6983 debug_leave();
6984 debug_printf_exec("run_pipe: return %d\n", rcode);
6985 return rcode;
6986 }
6987
6988 argv = command->argv ? command->argv : (char **) &null_ptr;
6989 {
6990 const struct built_in_command *x;
6991#if ENABLE_HUSH_FUNCTIONS
6992 const struct function *funcp;
6993#else
6994 enum { funcp = 0 };
6995#endif
6996 char **new_env = NULL;
6997 struct variable *old_vars = NULL;
6998
6999 if (argv[command->assignment_cnt] == NULL) {
7000 /* Assignments, but no command */
7001 /* Ensure redirects take effect (that is, create files).
7002 * Try "a=t >file" */
7003#if 0 /* A few cases in testsuite fail with this code. FIXME */
7004 rcode = redirect_and_varexp_helper(&new_env, /*old_vars:*/ NULL, command, squirrel, /*argv_expanded:*/ NULL);
7005 /* Set shell variables */
7006 if (new_env) {
7007 argv = new_env;
7008 while (*argv) {
7009 set_local_var(*argv, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7010 /* Do we need to flag set_local_var() errors?
7011 * "assignment to readonly var" and "putenv error"
7012 */
7013 argv++;
7014 }
7015 }
7016 /* Redirect error sets $? to 1. Otherwise,
7017 * if evaluating assignment value set $?, retain it.
7018 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7019 if (rcode == 0)
7020 rcode = G.last_exitcode;
7021 /* Exit, _skipping_ variable restoring code: */
7022 goto clean_up_and_ret0;
7023
7024#else /* Older, bigger, but more correct code */
7025
7026 rcode = setup_redirects(command, squirrel);
7027 restore_redirects(squirrel);
7028 /* Set shell variables */
7029 if (G_x_mode)
7030 bb_putchar_stderr('+');
7031 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007032 char *p = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007033 if (G_x_mode)
7034 fprintf(stderr, " %s", p);
7035 debug_printf_exec("set shell var:'%s'->'%s'\n",
7036 *argv, p);
7037 set_local_var(p, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7038 /* Do we need to flag set_local_var() errors?
7039 * "assignment to readonly var" and "putenv error"
7040 */
7041 argv++;
7042 }
7043 if (G_x_mode)
7044 bb_putchar_stderr('\n');
7045 /* Redirect error sets $? to 1. Otherwise,
7046 * if evaluating assignment value set $?, retain it.
7047 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7048 if (rcode == 0)
7049 rcode = G.last_exitcode;
7050 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7051 debug_leave();
7052 debug_printf_exec("run_pipe: return %d\n", rcode);
7053 return rcode;
7054#endif
7055 }
7056
7057 /* Expand the rest into (possibly) many strings each */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007058#if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007059 if (command->cmd_type == CMD_SINGLEWORD_NOGLOB) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007060 argv_expanded = expand_strvec_to_strvec_singleword_noglob(argv + command->assignment_cnt);
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007061 } else
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007062#endif
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007063 {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007064 argv_expanded = expand_strvec_to_strvec(argv + command->assignment_cnt);
7065 }
7066
7067 /* if someone gives us an empty string: `cmd with empty output` */
7068 if (!argv_expanded[0]) {
7069 free(argv_expanded);
7070 debug_leave();
7071 return G.last_exitcode;
7072 }
7073
7074 x = find_builtin(argv_expanded[0]);
7075#if ENABLE_HUSH_FUNCTIONS
7076 funcp = NULL;
7077 if (!x)
7078 funcp = find_function(argv_expanded[0]);
7079#endif
7080 if (x || funcp) {
7081 if (!funcp) {
7082 if (x->b_function == builtin_exec && argv_expanded[1] == NULL) {
7083 debug_printf("exec with redirects only\n");
7084 rcode = setup_redirects(command, NULL);
7085 goto clean_up_and_ret1;
7086 }
7087 }
7088 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7089 if (rcode == 0) {
7090 if (!funcp) {
7091 debug_printf_exec(": builtin '%s' '%s'...\n",
7092 x->b_cmd, argv_expanded[1]);
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007093 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007094 rcode = x->b_function(argv_expanded) & 0xff;
7095 fflush_all();
7096 }
7097#if ENABLE_HUSH_FUNCTIONS
7098 else {
7099# if ENABLE_HUSH_LOCAL
7100 struct variable **sv;
7101 sv = G.shadowed_vars_pp;
7102 G.shadowed_vars_pp = &old_vars;
7103# endif
7104 debug_printf_exec(": function '%s' '%s'...\n",
7105 funcp->name, argv_expanded[1]);
7106 rcode = run_function(funcp, argv_expanded) & 0xff;
7107# if ENABLE_HUSH_LOCAL
7108 G.shadowed_vars_pp = sv;
7109# endif
7110 }
7111#endif
7112 }
7113 clean_up_and_ret:
7114 unset_vars(new_env);
7115 add_vars(old_vars);
7116/* clean_up_and_ret0: */
7117 restore_redirects(squirrel);
7118 clean_up_and_ret1:
7119 free(argv_expanded);
7120 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7121 debug_leave();
7122 debug_printf_exec("run_pipe return %d\n", rcode);
7123 return rcode;
7124 }
7125
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007126 if (ENABLE_FEATURE_SH_NOFORK) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007127 int n = find_applet_by_name(argv_expanded[0]);
7128 if (n >= 0 && APPLET_IS_NOFORK(n)) {
7129 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7130 if (rcode == 0) {
7131 debug_printf_exec(": run_nofork_applet '%s' '%s'...\n",
7132 argv_expanded[0], argv_expanded[1]);
7133 rcode = run_nofork_applet(n, argv_expanded);
7134 }
7135 goto clean_up_and_ret;
7136 }
7137 }
7138 /* It is neither builtin nor applet. We must fork. */
7139 }
7140
7141 must_fork:
7142 /* NB: argv_expanded may already be created, and that
7143 * might include `cmd` runs! Do not rerun it! We *must*
7144 * use argv_expanded if it's non-NULL */
7145
7146 /* Going to fork a child per each pipe member */
7147 pi->alive_cmds = 0;
7148 next_infd = 0;
7149
7150 cmd_no = 0;
7151 while (cmd_no < pi->num_cmds) {
7152 struct fd_pair pipefds;
7153#if !BB_MMU
7154 volatile nommu_save_t nommu_save;
7155 nommu_save.new_env = NULL;
7156 nommu_save.old_vars = NULL;
7157 nommu_save.argv = NULL;
7158 nommu_save.argv_from_re_execing = NULL;
7159#endif
7160 command = &pi->cmds[cmd_no];
7161 cmd_no++;
7162 if (command->argv) {
7163 debug_printf_exec(": pipe member '%s' '%s'...\n",
7164 command->argv[0], command->argv[1]);
7165 } else {
7166 debug_printf_exec(": pipe member with no argv\n");
7167 }
7168
7169 /* pipes are inserted between pairs of commands */
7170 pipefds.rd = 0;
7171 pipefds.wr = 1;
7172 if (cmd_no < pi->num_cmds)
7173 xpiped_pair(pipefds);
7174
7175 command->pid = BB_MMU ? fork() : vfork();
7176 if (!command->pid) { /* child */
7177#if ENABLE_HUSH_JOB
7178 disable_restore_tty_pgrp_on_exit();
7179 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
7180
7181 /* Every child adds itself to new process group
7182 * with pgid == pid_of_first_child_in_pipe */
7183 if (G.run_list_level == 1 && G_interactive_fd) {
7184 pid_t pgrp;
7185 pgrp = pi->pgrp;
7186 if (pgrp < 0) /* true for 1st process only */
7187 pgrp = getpid();
7188 if (setpgid(0, pgrp) == 0
7189 && pi->followup != PIPE_BG
7190 && G_saved_tty_pgrp /* we have ctty */
7191 ) {
7192 /* We do it in *every* child, not just first,
7193 * to avoid races */
7194 tcsetpgrp(G_interactive_fd, pgrp);
7195 }
7196 }
7197#endif
7198 if (pi->alive_cmds == 0 && pi->followup == PIPE_BG) {
7199 /* 1st cmd in backgrounded pipe
7200 * should have its stdin /dev/null'ed */
7201 close(0);
7202 if (open(bb_dev_null, O_RDONLY))
7203 xopen("/", O_RDONLY);
7204 } else {
7205 xmove_fd(next_infd, 0);
7206 }
7207 xmove_fd(pipefds.wr, 1);
7208 if (pipefds.rd > 1)
7209 close(pipefds.rd);
7210 /* Like bash, explicit redirects override pipes,
7211 * and the pipe fd is available for dup'ing. */
7212 if (setup_redirects(command, NULL))
7213 _exit(1);
7214
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007215 /* Stores to nommu_save list of env vars putenv'ed
7216 * (NOMMU, on MMU we don't need that) */
7217 /* cast away volatility... */
7218 pseudo_exec((nommu_save_t*) &nommu_save, command, argv_expanded);
7219 /* pseudo_exec() does not return */
7220 }
7221
7222 /* parent or error */
7223#if ENABLE_HUSH_FAST
7224 G.count_SIGCHLD++;
7225//bb_error_msg("[%d] fork in run_pipe: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
7226#endif
7227 enable_restore_tty_pgrp_on_exit();
7228#if !BB_MMU
7229 /* Clean up after vforked child */
7230 free(nommu_save.argv);
7231 free(nommu_save.argv_from_re_execing);
7232 unset_vars(nommu_save.new_env);
7233 add_vars(nommu_save.old_vars);
7234#endif
7235 free(argv_expanded);
7236 argv_expanded = NULL;
7237 if (command->pid < 0) { /* [v]fork failed */
7238 /* Clearly indicate, was it fork or vfork */
7239 bb_perror_msg(BB_MMU ? "vfork"+1 : "vfork");
7240 } else {
7241 pi->alive_cmds++;
7242#if ENABLE_HUSH_JOB
7243 /* Second and next children need to know pid of first one */
7244 if (pi->pgrp < 0)
7245 pi->pgrp = command->pid;
7246#endif
7247 }
7248
7249 if (cmd_no > 1)
7250 close(next_infd);
7251 if (cmd_no < pi->num_cmds)
7252 close(pipefds.wr);
7253 /* Pass read (output) pipe end to next iteration */
7254 next_infd = pipefds.rd;
7255 }
7256
7257 if (!pi->alive_cmds) {
7258 debug_leave();
7259 debug_printf_exec("run_pipe return 1 (all forks failed, no children)\n");
7260 return 1;
7261 }
7262
7263 debug_leave();
7264 debug_printf_exec("run_pipe return -1 (%u children started)\n", pi->alive_cmds);
7265 return -1;
7266}
7267
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007268/* NB: called by pseudo_exec, and therefore must not modify any
7269 * global data until exec/_exit (we can be a child after vfork!) */
7270static int run_list(struct pipe *pi)
7271{
7272#if ENABLE_HUSH_CASE
7273 char *case_word = NULL;
7274#endif
7275#if ENABLE_HUSH_LOOPS
7276 struct pipe *loop_top = NULL;
7277 char **for_lcur = NULL;
7278 char **for_list = NULL;
7279#endif
7280 smallint last_followup;
7281 smalluint rcode;
7282#if ENABLE_HUSH_IF || ENABLE_HUSH_CASE
7283 smalluint cond_code = 0;
7284#else
7285 enum { cond_code = 0 };
7286#endif
7287#if HAS_KEYWORDS
Denys Vlasenko9b782552010-09-08 13:33:26 +02007288 smallint rword; /* RES_foo */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007289 smallint last_rword; /* ditto */
7290#endif
7291
7292 debug_printf_exec("run_list start lvl %d\n", G.run_list_level);
7293 debug_enter();
7294
7295#if ENABLE_HUSH_LOOPS
7296 /* Check syntax for "for" */
Denys Vlasenko0d6a4ec2010-12-18 01:34:49 +01007297 {
7298 struct pipe *cpipe;
7299 for (cpipe = pi; cpipe; cpipe = cpipe->next) {
7300 if (cpipe->res_word != RES_FOR && cpipe->res_word != RES_IN)
7301 continue;
7302 /* current word is FOR or IN (BOLD in comments below) */
7303 if (cpipe->next == NULL) {
7304 syntax_error("malformed for");
7305 debug_leave();
7306 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7307 return 1;
7308 }
7309 /* "FOR v; do ..." and "for v IN a b; do..." are ok */
7310 if (cpipe->next->res_word == RES_DO)
7311 continue;
7312 /* next word is not "do". It must be "in" then ("FOR v in ...") */
7313 if (cpipe->res_word == RES_IN /* "for v IN a b; not_do..."? */
7314 || cpipe->next->res_word != RES_IN /* FOR v not_do_and_not_in..."? */
7315 ) {
7316 syntax_error("malformed for");
7317 debug_leave();
7318 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7319 return 1;
7320 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007321 }
7322 }
7323#endif
7324
7325 /* Past this point, all code paths should jump to ret: label
7326 * in order to return, no direct "return" statements please.
7327 * This helps to ensure that no memory is leaked. */
7328
7329#if ENABLE_HUSH_JOB
7330 G.run_list_level++;
7331#endif
7332
7333#if HAS_KEYWORDS
7334 rword = RES_NONE;
7335 last_rword = RES_XXXX;
7336#endif
7337 last_followup = PIPE_SEQ;
7338 rcode = G.last_exitcode;
7339
7340 /* Go through list of pipes, (maybe) executing them. */
7341 for (; pi; pi = IF_HUSH_LOOPS(rword == RES_DONE ? loop_top : ) pi->next) {
7342 if (G.flag_SIGINT)
7343 break;
7344
7345 IF_HAS_KEYWORDS(rword = pi->res_word;)
7346 debug_printf_exec(": rword=%d cond_code=%d last_rword=%d\n",
7347 rword, cond_code, last_rword);
7348#if ENABLE_HUSH_LOOPS
7349 if ((rword == RES_WHILE || rword == RES_UNTIL || rword == RES_FOR)
7350 && loop_top == NULL /* avoid bumping G.depth_of_loop twice */
7351 ) {
7352 /* start of a loop: remember where loop starts */
7353 loop_top = pi;
7354 G.depth_of_loop++;
7355 }
7356#endif
7357 /* Still in the same "if...", "then..." or "do..." branch? */
7358 if (IF_HAS_KEYWORDS(rword == last_rword &&) 1) {
7359 if ((rcode == 0 && last_followup == PIPE_OR)
7360 || (rcode != 0 && last_followup == PIPE_AND)
7361 ) {
7362 /* It is "<true> || CMD" or "<false> && CMD"
7363 * and we should not execute CMD */
7364 debug_printf_exec("skipped cmd because of || or &&\n");
7365 last_followup = pi->followup;
maxwen27116ba2015-08-14 21:41:28 +02007366 goto dont_check_jobs_but_continue;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007367 }
7368 }
7369 last_followup = pi->followup;
7370 IF_HAS_KEYWORDS(last_rword = rword;)
7371#if ENABLE_HUSH_IF
7372 if (cond_code) {
7373 if (rword == RES_THEN) {
7374 /* if false; then ... fi has exitcode 0! */
7375 G.last_exitcode = rcode = EXIT_SUCCESS;
7376 /* "if <false> THEN cmd": skip cmd */
7377 continue;
7378 }
7379 } else {
7380 if (rword == RES_ELSE || rword == RES_ELIF) {
7381 /* "if <true> then ... ELSE/ELIF cmd":
7382 * skip cmd and all following ones */
7383 break;
7384 }
7385 }
7386#endif
7387#if ENABLE_HUSH_LOOPS
7388 if (rword == RES_FOR) { /* && pi->num_cmds - always == 1 */
7389 if (!for_lcur) {
7390 /* first loop through for */
7391
7392 static const char encoded_dollar_at[] ALIGN1 = {
7393 SPECIAL_VAR_SYMBOL, '@' | 0x80, SPECIAL_VAR_SYMBOL, '\0'
7394 }; /* encoded representation of "$@" */
7395 static const char *const encoded_dollar_at_argv[] = {
7396 encoded_dollar_at, NULL
7397 }; /* argv list with one element: "$@" */
7398 char **vals;
7399
7400 vals = (char**)encoded_dollar_at_argv;
7401 if (pi->next->res_word == RES_IN) {
7402 /* if no variable values after "in" we skip "for" */
7403 if (!pi->next->cmds[0].argv) {
7404 G.last_exitcode = rcode = EXIT_SUCCESS;
7405 debug_printf_exec(": null FOR: exitcode EXIT_SUCCESS\n");
7406 break;
7407 }
7408 vals = pi->next->cmds[0].argv;
7409 } /* else: "for var; do..." -> assume "$@" list */
7410 /* create list of variable values */
7411 debug_print_strings("for_list made from", vals);
7412 for_list = expand_strvec_to_strvec(vals);
7413 for_lcur = for_list;
7414 debug_print_strings("for_list", for_list);
7415 }
7416 if (!*for_lcur) {
7417 /* "for" loop is over, clean up */
7418 free(for_list);
7419 for_list = NULL;
7420 for_lcur = NULL;
7421 break;
7422 }
7423 /* Insert next value from for_lcur */
7424 /* note: *for_lcur already has quotes removed, $var expanded, etc */
7425 set_local_var(xasprintf("%s=%s", pi->cmds[0].argv[0], *for_lcur++), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7426 continue;
7427 }
7428 if (rword == RES_IN) {
7429 continue; /* "for v IN list;..." - "in" has no cmds anyway */
7430 }
7431 if (rword == RES_DONE) {
7432 continue; /* "done" has no cmds too */
7433 }
7434#endif
7435#if ENABLE_HUSH_CASE
7436 if (rword == RES_CASE) {
7437 case_word = expand_strvec_to_string(pi->cmds->argv);
7438 continue;
7439 }
7440 if (rword == RES_MATCH) {
7441 char **argv;
7442
7443 if (!case_word) /* "case ... matched_word) ... WORD)": we executed selected branch, stop */
7444 break;
7445 /* all prev words didn't match, does this one match? */
7446 argv = pi->cmds->argv;
7447 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007448 char *pattern = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007449 /* TODO: which FNM_xxx flags to use? */
7450 cond_code = (fnmatch(pattern, case_word, /*flags:*/ 0) != 0);
7451 free(pattern);
7452 if (cond_code == 0) { /* match! we will execute this branch */
7453 free(case_word); /* make future "word)" stop */
7454 case_word = NULL;
7455 break;
7456 }
7457 argv++;
7458 }
7459 continue;
7460 }
7461 if (rword == RES_CASE_BODY) { /* inside of a case branch */
7462 if (cond_code != 0)
7463 continue; /* not matched yet, skip this pipe */
7464 }
7465#endif
7466 /* Just pressing <enter> in shell should check for jobs.
7467 * OTOH, in non-interactive shell this is useless
7468 * and only leads to extra job checks */
7469 if (pi->num_cmds == 0) {
7470 if (G_interactive_fd)
7471 goto check_jobs_and_continue;
7472 continue;
7473 }
7474
7475 /* After analyzing all keywords and conditions, we decided
7476 * to execute this pipe. NB: have to do checkjobs(NULL)
7477 * after run_pipe to collect any background children,
7478 * even if list execution is to be stopped. */
7479 debug_printf_exec(": run_pipe with %d members\n", pi->num_cmds);
7480 {
7481 int r;
7482#if ENABLE_HUSH_LOOPS
7483 G.flag_break_continue = 0;
7484#endif
7485 rcode = r = run_pipe(pi); /* NB: rcode is a smallint */
7486 if (r != -1) {
7487 /* We ran a builtin, function, or group.
7488 * rcode is already known
7489 * and we don't need to wait for anything. */
7490 G.last_exitcode = rcode;
7491 debug_printf_exec(": builtin/func exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007492 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007493#if ENABLE_HUSH_LOOPS
7494 /* Was it "break" or "continue"? */
7495 if (G.flag_break_continue) {
7496 smallint fbc = G.flag_break_continue;
7497 /* We might fall into outer *loop*,
7498 * don't want to break it too */
7499 if (loop_top) {
7500 G.depth_break_continue--;
7501 if (G.depth_break_continue == 0)
7502 G.flag_break_continue = 0;
7503 /* else: e.g. "continue 2" should *break* once, *then* continue */
7504 } /* else: "while... do... { we are here (innermost list is not a loop!) };...done" */
maxwen27116ba2015-08-14 21:41:28 +02007505 if (G.depth_break_continue != 0 || fbc == BC_BREAK) {
7506 checkjobs(NULL);
7507 break;
7508 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007509 /* "continue": simulate end of loop */
7510 rword = RES_DONE;
7511 continue;
7512 }
7513#endif
7514#if ENABLE_HUSH_FUNCTIONS
7515 if (G.flag_return_in_progress == 1) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007516 checkjobs(NULL);
7517 break;
7518 }
7519#endif
7520 } else if (pi->followup == PIPE_BG) {
7521 /* What does bash do with attempts to background builtins? */
7522 /* even bash 3.2 doesn't do that well with nested bg:
7523 * try "{ { sleep 10; echo DEEP; } & echo HERE; } &".
7524 * I'm NOT treating inner &'s as jobs */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007525 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007526#if ENABLE_HUSH_JOB
7527 if (G.run_list_level == 1)
7528 insert_bg_job(pi);
7529#endif
7530 /* Last command's pid goes to $! */
7531 G.last_bg_pid = pi->cmds[pi->num_cmds - 1].pid;
7532 G.last_exitcode = rcode = EXIT_SUCCESS;
7533 debug_printf_exec(": cmd&: exitcode EXIT_SUCCESS\n");
7534 } else {
7535#if ENABLE_HUSH_JOB
7536 if (G.run_list_level == 1 && G_interactive_fd) {
7537 /* Waits for completion, then fg's main shell */
7538 rcode = checkjobs_and_fg_shell(pi);
7539 debug_printf_exec(": checkjobs_and_fg_shell exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007540 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007541 } else
7542#endif
7543 { /* This one just waits for completion */
7544 rcode = checkjobs(pi);
7545 debug_printf_exec(": checkjobs exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007546 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007547 }
7548 G.last_exitcode = rcode;
7549 }
7550 }
7551
7552 /* Analyze how result affects subsequent commands */
7553#if ENABLE_HUSH_IF
7554 if (rword == RES_IF || rword == RES_ELIF)
7555 cond_code = rcode;
7556#endif
maxwen27116ba2015-08-14 21:41:28 +02007557 check_jobs_and_continue:
7558 checkjobs(NULL);
7559 dont_check_jobs_but_continue: ;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007560#if ENABLE_HUSH_LOOPS
7561 /* Beware of "while false; true; do ..."! */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007562 if (pi->next
7563 && (pi->next->res_word == RES_DO || pi->next->res_word == RES_DONE)
Denys Vlasenko56a3b822011-06-01 12:47:07 +02007564 /* check for RES_DONE is needed for "while ...; do \n done" case */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007565 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007566 if (rword == RES_WHILE) {
7567 if (rcode) {
7568 /* "while false; do...done" - exitcode 0 */
7569 G.last_exitcode = rcode = EXIT_SUCCESS;
7570 debug_printf_exec(": while expr is false: breaking (exitcode:EXIT_SUCCESS)\n");
maxwen27116ba2015-08-14 21:41:28 +02007571 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007572 }
7573 }
7574 if (rword == RES_UNTIL) {
7575 if (!rcode) {
7576 debug_printf_exec(": until expr is true: breaking\n");
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007577 break;
7578 }
7579 }
7580 }
7581#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007582 } /* for (pi) */
7583
7584#if ENABLE_HUSH_JOB
7585 G.run_list_level--;
7586#endif
7587#if ENABLE_HUSH_LOOPS
7588 if (loop_top)
7589 G.depth_of_loop--;
7590 free(for_list);
7591#endif
7592#if ENABLE_HUSH_CASE
7593 free(case_word);
7594#endif
7595 debug_leave();
7596 debug_printf_exec("run_list lvl %d return %d\n", G.run_list_level + 1, rcode);
7597 return rcode;
7598}
7599
7600/* Select which version we will use */
7601static int run_and_free_list(struct pipe *pi)
7602{
7603 int rcode = 0;
7604 debug_printf_exec("run_and_free_list entered\n");
Dan Fandrich85c62472010-11-20 13:05:17 -08007605 if (!G.o_opt[OPT_O_NOEXEC]) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007606 debug_printf_exec(": run_list: 1st pipe with %d cmds\n", pi->num_cmds);
7607 rcode = run_list(pi);
7608 }
7609 /* free_pipe_list has the side effect of clearing memory.
7610 * In the long run that function can be merged with run_list,
7611 * but doing that now would hobble the debugging effort. */
7612 free_pipe_list(pi);
7613 debug_printf_exec("run_and_free_list return %d\n", rcode);
7614 return rcode;
7615}
7616
7617
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007618static void install_sighandlers(unsigned mask)
Eric Andersen52a97ca2001-06-22 06:49:26 +00007619{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007620 sighandler_t old_handler;
7621 unsigned sig = 0;
7622 while ((mask >>= 1) != 0) {
7623 sig++;
7624 if (!(mask & 1))
7625 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02007626 old_handler = install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007627 /* POSIX allows shell to re-enable SIGCHLD
7628 * even if it was SIG_IGN on entry.
7629 * Therefore we skip IGN check for it:
7630 */
7631 if (sig == SIGCHLD)
7632 continue;
7633 if (old_handler == SIG_IGN) {
7634 /* oops... restore back to IGN, and record this fact */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007635 install_sighandler(sig, old_handler);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007636 if (!G.traps)
7637 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7638 free(G.traps[sig]);
7639 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
7640 }
7641 }
7642}
7643
7644/* Called a few times only (or even once if "sh -c") */
7645static void install_special_sighandlers(void)
7646{
Denis Vlasenkof9375282009-04-05 19:13:39 +00007647 unsigned mask;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007648
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007649 /* Which signals are shell-special? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007650 mask = (1 << SIGQUIT) | (1 << SIGCHLD);
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007651 if (G_interactive_fd) {
7652 mask |= SPECIAL_INTERACTIVE_SIGS;
7653 if (G_saved_tty_pgrp) /* we have ctty, job control sigs work */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007654 mask |= SPECIAL_JOBSTOP_SIGS;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007655 }
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007656 /* Careful, do not re-install handlers we already installed */
7657 if (G.special_sig_mask != mask) {
7658 unsigned diff = mask & ~G.special_sig_mask;
7659 G.special_sig_mask = mask;
7660 install_sighandlers(diff);
7661 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00007662}
7663
7664#if ENABLE_HUSH_JOB
7665/* helper */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007666/* Set handlers to restore tty pgrp and exit */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007667static void install_fatal_sighandlers(void)
Denis Vlasenkof9375282009-04-05 19:13:39 +00007668{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007669 unsigned mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007670
7671 /* We will restore tty pgrp on these signals */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007672 mask = 0
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007673 + (1 << SIGILL ) * HUSH_DEBUG
7674 + (1 << SIGFPE ) * HUSH_DEBUG
7675 + (1 << SIGBUS ) * HUSH_DEBUG
7676 + (1 << SIGSEGV) * HUSH_DEBUG
7677 + (1 << SIGTRAP) * HUSH_DEBUG
7678 + (1 << SIGABRT)
7679 /* bash 3.2 seems to handle these just like 'fatal' ones */
7680 + (1 << SIGPIPE)
7681 + (1 << SIGALRM)
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007682 /* if we are interactive, SIGHUP, SIGTERM and SIGINT are special sigs.
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007683 * if we aren't interactive... but in this case
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007684 * we never want to restore pgrp on exit, and this fn is not called
7685 */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007686 /*+ (1 << SIGHUP )*/
7687 /*+ (1 << SIGTERM)*/
7688 /*+ (1 << SIGINT )*/
7689 ;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007690 G_fatal_sig_mask = mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007691
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007692 install_sighandlers(mask);
Denis Vlasenkof9375282009-04-05 19:13:39 +00007693}
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00007694#endif
Eric Andersenada18ff2001-05-21 16:18:22 +00007695
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007696static int set_mode(int state, char mode, const char *o_opt)
Denis Vlasenkod5762932009-03-31 11:22:57 +00007697{
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007698 int idx;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007699 switch (mode) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007700 case 'n':
Dan Fandrich85c62472010-11-20 13:05:17 -08007701 G.o_opt[OPT_O_NOEXEC] = state;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007702 break;
7703 case 'x':
7704 IF_HUSH_MODE_X(G_x_mode = state;)
7705 break;
7706 case 'o':
7707 if (!o_opt) {
7708 /* "set -+o" without parameter.
7709 * in bash, set -o produces this output:
7710 * pipefail off
7711 * and set +o:
7712 * set +o pipefail
7713 * We always use the second form.
7714 */
7715 const char *p = o_opt_strings;
7716 idx = 0;
7717 while (*p) {
7718 printf("set %co %s\n", (G.o_opt[idx] ? '-' : '+'), p);
7719 idx++;
7720 p += strlen(p) + 1;
7721 }
7722 break;
7723 }
7724 idx = index_in_strings(o_opt_strings, o_opt);
7725 if (idx >= 0) {
7726 G.o_opt[idx] = state;
7727 break;
7728 }
7729 default:
7730 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007731 }
7732 return EXIT_SUCCESS;
7733}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00007734
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00007735int hush_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Matt Kraai2d91deb2001-08-01 17:21:35 +00007736int hush_main(int argc, char **argv)
Eric Andersen25f27032001-04-26 23:22:31 +00007737{
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007738 enum {
7739 OPT_login = (1 << 0),
7740 };
7741 unsigned flags;
Eric Andersen25f27032001-04-26 23:22:31 +00007742 int opt;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007743 unsigned builtin_argc;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007744 char **e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007745 struct variable *cur_var;
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007746 struct variable *shell_ver;
Eric Andersenbc604a22001-05-16 05:24:03 +00007747
Denis Vlasenko574f2f42008-02-27 18:41:59 +00007748 INIT_G();
Denys Vlasenko10c01312011-05-11 11:49:21 +02007749 if (EXIT_SUCCESS != 0) /* if EXIT_SUCCESS == 0, it is already done */
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007750 G.last_exitcode = EXIT_SUCCESS;
Denys Vlasenko10c01312011-05-11 11:49:21 +02007751#if ENABLE_HUSH_FAST
7752 G.count_SIGCHLD++; /* ensure it is != G.handled_SIGCHLD */
7753#endif
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007754#if !BB_MMU
7755 G.argv0_for_re_execing = argv[0];
7756#endif
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007757 /* Deal with HUSH_VERSION */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007758 shell_ver = xzalloc(sizeof(*shell_ver));
7759 shell_ver->flg_export = 1;
7760 shell_ver->flg_read_only = 1;
Denys Vlasenko4f870492010-09-10 11:06:01 +02007761 /* Code which handles ${var<op>...} needs writable values for all variables,
Denys Vlasenko36f774a2010-09-05 14:45:38 +02007762 * therefore we xstrdup: */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007763 shell_ver->varstr = xstrdup(hush_version_str);
Denys Vlasenko605067b2010-09-06 12:10:51 +02007764 /* Create shell local variables from the values
7765 * currently living in the environment */
Denis Vlasenkof886fd22008-10-13 12:36:05 +00007766 debug_printf_env("unsetenv '%s'\n", "HUSH_VERSION");
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007767 unsetenv("HUSH_VERSION"); /* in case it exists in initial env */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007768 G.top_var = shell_ver;
Denis Vlasenko87a86552008-07-29 19:43:10 +00007769 cur_var = G.top_var;
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007770 e = environ;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007771 if (e) while (*e) {
7772 char *value = strchr(*e, '=');
7773 if (value) { /* paranoia */
7774 cur_var->next = xzalloc(sizeof(*cur_var));
7775 cur_var = cur_var->next;
Denis Vlasenko28c0f0f2007-05-25 02:46:01 +00007776 cur_var->varstr = *e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007777 cur_var->max_len = strlen(*e);
7778 cur_var->flg_export = 1;
7779 }
7780 e++;
7781 }
Denys Vlasenko605067b2010-09-06 12:10:51 +02007782 /* (Re)insert HUSH_VERSION into env (AFTER we scanned the env!) */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007783 debug_printf_env("putenv '%s'\n", shell_ver->varstr);
7784 putenv(shell_ver->varstr);
Denys Vlasenko6db47842009-09-05 20:15:17 +02007785
7786 /* Export PWD */
7787 set_pwd_var(/*exp:*/ 1);
7788 /* bash also exports SHLVL and _,
7789 * and sets (but doesn't export) the following variables:
7790 * BASH=/bin/bash
7791 * BASH_VERSINFO=([0]="3" [1]="2" [2]="0" [3]="1" [4]="release" [5]="i386-pc-linux-gnu")
7792 * BASH_VERSION='3.2.0(1)-release'
7793 * HOSTTYPE=i386
7794 * MACHTYPE=i386-pc-linux-gnu
7795 * OSTYPE=linux-gnu
7796 * HOSTNAME=<xxxxxxxxxx>
Denys Vlasenkodea47882009-10-09 15:40:49 +02007797 * PPID=<NNNNN> - we also do it elsewhere
Denys Vlasenko6db47842009-09-05 20:15:17 +02007798 * EUID=<NNNNN>
7799 * UID=<NNNNN>
7800 * GROUPS=()
7801 * LINES=<NNN>
7802 * COLUMNS=<NNN>
7803 * BASH_ARGC=()
7804 * BASH_ARGV=()
7805 * BASH_LINENO=()
7806 * BASH_SOURCE=()
7807 * DIRSTACK=()
7808 * PIPESTATUS=([0]="0")
7809 * HISTFILE=/<xxx>/.bash_history
7810 * HISTFILESIZE=500
7811 * HISTSIZE=500
7812 * MAILCHECK=60
7813 * PATH=/usr/gnu/bin:/usr/local/bin:/bin:/usr/bin:.
7814 * SHELL=/bin/bash
7815 * SHELLOPTS=braceexpand:emacs:hashall:histexpand:history:interactive-comments:monitor
7816 * TERM=dumb
7817 * OPTERR=1
7818 * OPTIND=1
7819 * IFS=$' \t\n'
7820 * PS1='\s-\v\$ '
7821 * PS2='> '
7822 * PS4='+ '
7823 */
7824
Denis Vlasenko38f63192007-01-22 09:03:07 +00007825#if ENABLE_FEATURE_EDITING
Denis Vlasenko87a86552008-07-29 19:43:10 +00007826 G.line_input_state = new_line_input_t(FOR_SHELL);
Denis Vlasenko8e1c7152007-01-22 07:21:38 +00007827#endif
Denys Vlasenko99862cb2010-09-12 17:34:13 +02007828
Eric Andersen94ac2442001-05-22 19:05:18 +00007829 /* Initialize some more globals to non-zero values */
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00007830 cmdedit_update_prompt();
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +00007831
Denis Vlasenkoed782372009-04-10 00:45:02 +00007832 if (setjmp(die_jmp)) {
7833 /* xfunc has failed! die die die */
7834 /* no EXIT traps, this is an escape hatch! */
7835 G.exiting = 1;
7836 hush_exit(xfunc_error_retval);
7837 }
7838
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007839 /* Shell is non-interactive at first. We need to call
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007840 * install_special_sighandlers() if we are going to execute "sh <script>",
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007841 * "sh -c <cmds>" or login shell's /etc/profile and friends.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007842 * If we later decide that we are interactive, we run install_special_sighandlers()
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007843 * in order to intercept (more) signals.
7844 */
7845
7846 /* Parse options */
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007847 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/sh.html */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007848 flags = (argv[0] && argv[0][0] == '-') ? OPT_login : 0;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007849 builtin_argc = 0;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007850 while (1) {
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007851 opt = getopt(argc, argv, "+c:xinsl"
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007852#if !BB_MMU
Denis Vlasenkobc569742009-04-12 20:35:19 +00007853 "<:$:R:V:"
7854# if ENABLE_HUSH_FUNCTIONS
7855 "F:"
7856# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007857#endif
7858 );
7859 if (opt <= 0)
7860 break;
Eric Andersen25f27032001-04-26 23:22:31 +00007861 switch (opt) {
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007862 case 'c':
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007863 /* Possibilities:
7864 * sh ... -c 'script'
7865 * sh ... -c 'script' ARG0 [ARG1...]
7866 * On NOMMU, if builtin_argc != 0,
Denys Vlasenko17323a62010-01-28 01:57:05 +01007867 * sh ... -c 'builtin' BARGV... "" ARG0 [ARG1...]
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007868 * "" needs to be replaced with NULL
7869 * and BARGV vector fed to builtin function.
Denys Vlasenko17323a62010-01-28 01:57:05 +01007870 * Note: the form without ARG0 never happens:
7871 * sh ... -c 'builtin' BARGV... ""
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007872 */
Denys Vlasenkodea47882009-10-09 15:40:49 +02007873 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007874 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02007875 G.root_ppid = getppid();
7876 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00007877 G.global_argv = argv + optind;
7878 G.global_argc = argc - optind;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007879 if (builtin_argc) {
7880 /* -c 'builtin' [BARGV...] "" ARG0 [ARG1...] */
7881 const struct built_in_command *x;
7882
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007883 install_special_sighandlers();
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007884 x = find_builtin(optarg);
7885 if (x) { /* paranoia */
7886 G.global_argc -= builtin_argc; /* skip [BARGV...] "" */
7887 G.global_argv += builtin_argc;
7888 G.global_argv[-1] = NULL; /* replace "" */
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007889 fflush_all();
Denys Vlasenko17323a62010-01-28 01:57:05 +01007890 G.last_exitcode = x->b_function(argv + optind - 1);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007891 }
7892 goto final_return;
7893 }
7894 if (!G.global_argv[0]) {
7895 /* -c 'script' (no params): prevent empty $0 */
7896 G.global_argv--; /* points to argv[i] of 'script' */
7897 G.global_argv[0] = argv[0];
Denys Vlasenko5ae8f1c2010-05-22 06:32:11 +02007898 G.global_argc++;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007899 } /* else -c 'script' ARG0 [ARG1...]: $0 is ARG0 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007900 install_special_sighandlers();
Denis Vlasenkob6e65562009-04-03 16:49:04 +00007901 parse_and_run_string(optarg);
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007902 goto final_return;
7903 case 'i':
Denis Vlasenkoc666f712007-05-16 22:18:54 +00007904 /* Well, we cannot just declare interactiveness,
7905 * we have to have some stuff (ctty, etc) */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00007906 /* G_interactive_fd++; */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007907 break;
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007908 case 's':
7909 /* "-s" means "read from stdin", but this is how we always
7910 * operate, so simply do nothing here. */
7911 break;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007912 case 'l':
7913 flags |= OPT_login;
7914 break;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007915#if !BB_MMU
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007916 case '<': /* "big heredoc" support */
Denys Vlasenko729ecb82010-06-07 14:14:26 +02007917 full_write1_str(optarg);
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007918 _exit(0);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007919 case '$': {
7920 unsigned long long empty_trap_mask;
7921
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007922 G.root_pid = bb_strtou(optarg, &optarg, 16);
7923 optarg++;
Denys Vlasenkodea47882009-10-09 15:40:49 +02007924 G.root_ppid = bb_strtou(optarg, &optarg, 16);
7925 optarg++;
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007926 G.last_bg_pid = bb_strtou(optarg, &optarg, 16);
7927 optarg++;
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007928 G.last_exitcode = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007929 optarg++;
7930 builtin_argc = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007931 optarg++;
7932 empty_trap_mask = bb_strtoull(optarg, &optarg, 16);
7933 if (empty_trap_mask != 0) {
7934 int sig;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007935 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007936 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7937 for (sig = 1; sig < NSIG; sig++) {
7938 if (empty_trap_mask & (1LL << sig)) {
7939 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007940 install_sighandler(sig, SIG_IGN);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007941 }
7942 }
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007943 }
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007944# if ENABLE_HUSH_LOOPS
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007945 optarg++;
7946 G.depth_of_loop = bb_strtou(optarg, &optarg, 16);
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007947# endif
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007948 break;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007949 }
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007950 case 'R':
7951 case 'V':
Denys Vlasenko295fef82009-06-03 12:47:26 +02007952 set_local_var(xstrdup(optarg), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ opt == 'R');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007953 break;
Denis Vlasenkobc569742009-04-12 20:35:19 +00007954# if ENABLE_HUSH_FUNCTIONS
7955 case 'F': {
7956 struct function *funcp = new_function(optarg);
7957 /* funcp->name is already set to optarg */
7958 /* funcp->body is set to NULL. It's a special case. */
7959 funcp->body_as_string = argv[optind];
7960 optind++;
7961 break;
7962 }
7963# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007964#endif
Mike Frysingerad88d5a2009-03-28 13:44:51 +00007965 case 'n':
7966 case 'x':
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007967 if (set_mode(1, opt, NULL) == 0) /* no error */
Mike Frysingerad88d5a2009-03-28 13:44:51 +00007968 break;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007969 default:
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007970#ifndef BB_VER
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007971 fprintf(stderr, "Usage: sh [FILE]...\n"
7972 " or: sh -c command [args]...\n\n");
7973 exit(EXIT_FAILURE);
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007974#else
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007975 bb_show_usage();
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007976#endif
Eric Andersen25f27032001-04-26 23:22:31 +00007977 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00007978 } /* option parsing loop */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007979
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007980 /* Skip options. Try "hush -l": $1 should not be "-l"! */
7981 G.global_argc = argc - (optind - 1);
7982 G.global_argv = argv + (optind - 1);
7983 G.global_argv[0] = argv[0];
7984
Denys Vlasenkodea47882009-10-09 15:40:49 +02007985 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007986 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02007987 G.root_ppid = getppid();
7988 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00007989
7990 /* If we are login shell... */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007991 if (flags & OPT_login) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007992 FILE *input;
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007993 debug_printf("sourcing /etc/profile\n");
7994 input = fopen_for_read("/etc/profile");
7995 if (input != NULL) {
7996 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007997 install_special_sighandlers();
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007998 parse_and_run_file(input);
7999 fclose(input);
8000 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008001 /* bash: after sourcing /etc/profile,
8002 * tries to source (in the given order):
8003 * ~/.bash_profile, ~/.bash_login, ~/.profile,
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008004 * stopping on first found. --noprofile turns this off.
Denis Vlasenkof9375282009-04-05 19:13:39 +00008005 * bash also sources ~/.bash_logout on exit.
8006 * If called as sh, skips .bash_XXX files.
8007 */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008008 }
8009
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008010 if (G.global_argv[1]) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008011 FILE *input;
8012 /*
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00008013 * "bash <script>" (which is never interactive (unless -i?))
8014 * sources $BASH_ENV here (without scanning $PATH).
Denis Vlasenkof9375282009-04-05 19:13:39 +00008015 * If called as sh, does the same but with $ENV.
8016 */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008017 G.global_argc--;
8018 G.global_argv++;
8019 debug_printf("running script '%s'\n", G.global_argv[0]);
8020 input = xfopen_for_read(G.global_argv[0]);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008021 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008022 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008023 parse_and_run_file(input);
8024#if ENABLE_FEATURE_CLEAN_UP
8025 fclose(input);
8026#endif
8027 goto final_return;
8028 }
8029
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008030 /* Up to here, shell was non-interactive. Now it may become one.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008031 * NB: don't forget to (re)run install_special_sighandlers() as needed.
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008032 */
Denis Vlasenkof9375282009-04-05 19:13:39 +00008033
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008034 /* A shell is interactive if the '-i' flag was given,
8035 * or if all of the following conditions are met:
Denis Vlasenko55b2de72007-04-18 17:21:28 +00008036 * no -c command
Eric Andersen25f27032001-04-26 23:22:31 +00008037 * no arguments remaining or the -s flag given
8038 * standard input is a terminal
8039 * standard output is a terminal
Denis Vlasenkof9375282009-04-05 19:13:39 +00008040 * Refer to Posix.2, the description of the 'sh' utility.
8041 */
8042#if ENABLE_HUSH_JOB
8043 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Mike Frysinger38478a62009-05-20 04:48:06 -04008044 G_saved_tty_pgrp = tcgetpgrp(STDIN_FILENO);
8045 debug_printf("saved_tty_pgrp:%d\n", G_saved_tty_pgrp);
8046 if (G_saved_tty_pgrp < 0)
8047 G_saved_tty_pgrp = 0;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008048
8049 /* try to dup stdin to high fd#, >= 255 */
8050 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8051 if (G_interactive_fd < 0) {
8052 /* try to dup to any fd */
8053 G_interactive_fd = dup(STDIN_FILENO);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008054 if (G_interactive_fd < 0) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008055 /* give up */
8056 G_interactive_fd = 0;
Mike Frysinger38478a62009-05-20 04:48:06 -04008057 G_saved_tty_pgrp = 0;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00008058 }
8059 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008060// TODO: track & disallow any attempts of user
8061// to (inadvertently) close/redirect G_interactive_fd
Eric Andersen25f27032001-04-26 23:22:31 +00008062 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008063 debug_printf("interactive_fd:%d\n", G_interactive_fd);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008064 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008065 close_on_exec_on(G_interactive_fd);
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008066
Mike Frysinger38478a62009-05-20 04:48:06 -04008067 if (G_saved_tty_pgrp) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008068 /* If we were run as 'hush &', sleep until we are
8069 * in the foreground (tty pgrp == our pgrp).
8070 * If we get started under a job aware app (like bash),
8071 * make sure we are now in charge so we don't fight over
8072 * who gets the foreground */
8073 while (1) {
8074 pid_t shell_pgrp = getpgrp();
Mike Frysinger38478a62009-05-20 04:48:06 -04008075 G_saved_tty_pgrp = tcgetpgrp(G_interactive_fd);
8076 if (G_saved_tty_pgrp == shell_pgrp)
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008077 break;
8078 /* send TTIN to ourself (should stop us) */
8079 kill(- shell_pgrp, SIGTTIN);
8080 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008081 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008082
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008083 /* Install more signal handlers */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008084 install_special_sighandlers();
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008085
Mike Frysinger38478a62009-05-20 04:48:06 -04008086 if (G_saved_tty_pgrp) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008087 /* Set other signals to restore saved_tty_pgrp */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008088 install_fatal_sighandlers();
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008089 /* Put ourselves in our own process group
8090 * (bash, too, does this only if ctty is available) */
8091 bb_setpgrp(); /* is the same as setpgid(our_pid, our_pid); */
8092 /* Grab control of the terminal */
8093 tcsetpgrp(G_interactive_fd, getpid());
8094 }
Denis Vlasenko4ecfcdc2008-02-11 08:32:31 +00008095 /* -1 is special - makes xfuncs longjmp, not exit
Denis Vlasenkoc04163a2008-02-11 08:30:53 +00008096 * (we reset die_sleep = 0 whereever we [v]fork) */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00008097 enable_restore_tty_pgrp_on_exit(); /* sets die_sleep = -1 */
Tanguy Pruvot8a6c2c22012-04-28 00:24:09 +02008098
8099# if ENABLE_HUSH_SAVEHISTORY && MAX_HISTORY > 0
8100 {
8101 const char *hp = get_local_var_value("HISTFILE");
8102 if (!hp) {
8103 hp = get_local_var_value("HOME");
8104 if (hp)
8105 hp = concat_path_file(hp, ".hush_history");
8106 } else {
8107 hp = xstrdup(hp);
8108 }
8109 if (hp) {
8110 G.line_input_state->hist_file = hp;
8111 //set_local_var(xasprintf("HISTFILE=%s", ...));
8112 }
8113# if ENABLE_FEATURE_SH_HISTFILESIZE
8114 hp = get_local_var_value("HISTFILESIZE");
8115 G.line_input_state->max_history = size_from_HISTFILESIZE(hp);
8116# endif
8117 }
8118# endif
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008119 } else {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008120 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008121 }
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008122#elif ENABLE_HUSH_INTERACTIVE
Denis Vlasenkof9375282009-04-05 19:13:39 +00008123 /* No job control compiled in, only prompt/line editing */
8124 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008125 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8126 if (G_interactive_fd < 0) {
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008127 /* try to dup to any fd */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008128 G_interactive_fd = dup(STDIN_FILENO);
8129 if (G_interactive_fd < 0)
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008130 /* give up */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008131 G_interactive_fd = 0;
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008132 }
8133 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008134 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008135 close_on_exec_on(G_interactive_fd);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008136 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008137 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008138#else
8139 /* We have interactiveness code disabled */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008140 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008141#endif
8142 /* bash:
8143 * if interactive but not a login shell, sources ~/.bashrc
8144 * (--norc turns this off, --rcfile <file> overrides)
8145 */
8146
8147 if (!ENABLE_FEATURE_SH_EXTRA_QUIET && G_interactive_fd) {
Denys Vlasenkoc34c0332009-09-29 12:25:30 +02008148 /* note: ash and hush share this string */
8149 printf("\n\n%s %s\n"
8150 IF_HUSH_HELP("Enter 'help' for a list of built-in commands.\n")
8151 "\n",
8152 bb_banner,
8153 "hush - the humble shell"
8154 );
Mike Frysingerb2705e12009-03-23 08:44:02 +00008155 }
8156
Denis Vlasenkof9375282009-04-05 19:13:39 +00008157 parse_and_run_file(stdin);
Eric Andersen25f27032001-04-26 23:22:31 +00008158
Denis Vlasenkod76c0492007-05-25 02:16:25 +00008159 final_return:
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008160 hush_exit(G.last_exitcode);
Eric Andersen25f27032001-04-26 23:22:31 +00008161}
Denis Vlasenko96702ca2007-11-23 23:28:55 +00008162
8163
Denys Vlasenko1cc4b132009-08-21 00:05:51 +02008164#if ENABLE_MSH
8165int msh_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
8166int msh_main(int argc, char **argv)
8167{
8168 //bb_error_msg("msh is deprecated, please use hush instead");
8169 return hush_main(argc, argv);
8170}
8171#endif
8172
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008173
8174/*
8175 * Built-ins
8176 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008177static int FAST_FUNC builtin_true(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008178{
8179 return 0;
8180}
8181
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008182static int run_applet_main(char **argv, int (*applet_main_func)(int argc, char **argv))
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008183{
8184 int argc = 0;
8185 while (*argv) {
8186 argc++;
8187 argv++;
8188 }
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008189 return applet_main_func(argc, argv - argc);
Mike Frysingerccb19592009-10-15 03:31:15 -04008190}
8191
8192static int FAST_FUNC builtin_test(char **argv)
8193{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008194 return run_applet_main(argv, test_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008195}
8196
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008197static int FAST_FUNC builtin_echo(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008198{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008199 return run_applet_main(argv, echo_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008200}
8201
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008202#if ENABLE_PRINTF
8203static int FAST_FUNC builtin_printf(char **argv)
8204{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008205 return run_applet_main(argv, printf_main);
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008206}
8207#endif
8208
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008209static char **skip_dash_dash(char **argv)
8210{
8211 argv++;
8212 if (argv[0] && argv[0][0] == '-' && argv[0][1] == '-' && argv[0][2] == '\0')
8213 argv++;
8214 return argv;
8215}
8216
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008217static int FAST_FUNC builtin_eval(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008218{
8219 int rcode = EXIT_SUCCESS;
8220
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008221 argv = skip_dash_dash(argv);
8222 if (*argv) {
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008223 char *str = expand_strvec_to_string(argv);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008224 /* bash:
8225 * eval "echo Hi; done" ("done" is syntax error):
8226 * "echo Hi" will not execute too.
8227 */
8228 parse_and_run_string(str);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008229 free(str);
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008230 rcode = G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008231 }
8232 return rcode;
8233}
8234
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008235static int FAST_FUNC builtin_cd(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008236{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008237 const char *newdir;
8238
8239 argv = skip_dash_dash(argv);
8240 newdir = argv[0];
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008241 if (newdir == NULL) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008242 /* bash does nothing (exitcode 0) if HOME is ""; if it's unset,
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008243 * bash says "bash: cd: HOME not set" and does nothing
8244 * (exitcode 1)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008245 */
Denys Vlasenko90a99042009-09-06 02:36:23 +02008246 const char *home = get_local_var_value("HOME");
8247 newdir = home ? home : "/";
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008248 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008249 if (chdir(newdir)) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008250 /* Mimic bash message exactly */
8251 bb_perror_msg("cd: %s", newdir);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008252 return EXIT_FAILURE;
8253 }
Denys Vlasenko6db47842009-09-05 20:15:17 +02008254 /* Read current dir (get_cwd(1) is inside) and set PWD.
8255 * Note: do not enforce exporting. If PWD was unset or unexported,
8256 * set it again, but do not export. bash does the same.
8257 */
8258 set_pwd_var(/*exp:*/ 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008259 return EXIT_SUCCESS;
8260}
8261
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008262static int FAST_FUNC builtin_exec(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008263{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008264 argv = skip_dash_dash(argv);
8265 if (argv[0] == NULL)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008266 return EXIT_SUCCESS; /* bash does this */
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008267
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008268 /* Careful: we can end up here after [v]fork. Do not restore
8269 * tty pgrp then, only top-level shell process does that */
8270 if (G_saved_tty_pgrp && getpid() == G.root_pid)
8271 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
8272
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008273 /* TODO: if exec fails, bash does NOT exit! We do.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008274 * We'll need to undo trap cleanup (it's inside execvp_or_die)
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008275 * and tcsetpgrp, and this is inherently racy.
8276 */
8277 execvp_or_die(argv);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008278}
8279
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008280static int FAST_FUNC builtin_exit(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008281{
Denis Vlasenkocd418a22009-04-06 18:08:35 +00008282 debug_printf_exec("%s()\n", __func__);
Denis Vlasenko40e84372009-04-18 11:23:38 +00008283
8284 /* interactive bash:
8285 * # trap "echo EEE" EXIT
8286 * # exit
8287 * exit
8288 * There are stopped jobs.
8289 * (if there are _stopped_ jobs, running ones don't count)
8290 * # exit
8291 * exit
Tanguy Pruvot823694d2012-11-18 13:20:29 +01008292 * EEE (then bash exits)
Denis Vlasenko40e84372009-04-18 11:23:38 +00008293 *
Denys Vlasenkoa110c902010-09-12 15:38:04 +02008294 * TODO: we can use G.exiting = -1 as indicator "last cmd was exit"
Denis Vlasenko40e84372009-04-18 11:23:38 +00008295 */
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00008296
8297 /* note: EXIT trap is run by hush_exit */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008298 argv = skip_dash_dash(argv);
8299 if (argv[0] == NULL)
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008300 hush_exit(G.last_exitcode);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008301 /* mimic bash: exit 123abc == exit 255 + error msg */
8302 xfunc_error_retval = 255;
8303 /* bash: exit -2 == exit 254, no error msg */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008304 hush_exit(xatoi(argv[0]) & 0xff);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008305}
8306
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008307static void print_escaped(const char *s)
8308{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008309 if (*s == '\'')
8310 goto squote;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008311 do {
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008312 const char *p = strchrnul(s, '\'');
8313 /* print 'xxxx', possibly just '' */
8314 printf("'%.*s'", (int)(p - s), s);
8315 if (*p == '\0')
8316 break;
8317 s = p;
8318 squote:
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008319 /* s points to '; print "'''...'''" */
8320 putchar('"');
8321 do putchar('\''); while (*++s == '\'');
8322 putchar('"');
8323 } while (*s);
8324}
8325
Denys Vlasenko295fef82009-06-03 12:47:26 +02008326#if !ENABLE_HUSH_LOCAL
8327#define helper_export_local(argv, exp, lvl) \
8328 helper_export_local(argv, exp)
8329#endif
8330static void helper_export_local(char **argv, int exp, int lvl)
8331{
8332 do {
8333 char *name = *argv;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008334 char *name_end = strchrnul(name, '=');
Denys Vlasenko295fef82009-06-03 12:47:26 +02008335
8336 /* So far we do not check that name is valid (TODO?) */
8337
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008338 if (*name_end == '\0') {
8339 struct variable *var, **vpp;
Denys Vlasenko295fef82009-06-03 12:47:26 +02008340
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008341 vpp = get_ptr_to_local_var(name, name_end - name);
8342 var = vpp ? *vpp : NULL;
8343
Denys Vlasenko295fef82009-06-03 12:47:26 +02008344 if (exp == -1) { /* unexporting? */
8345 /* export -n NAME (without =VALUE) */
8346 if (var) {
8347 var->flg_export = 0;
8348 debug_printf_env("%s: unsetenv '%s'\n", __func__, name);
8349 unsetenv(name);
8350 } /* else: export -n NOT_EXISTING_VAR: no-op */
8351 continue;
8352 }
8353 if (exp == 1) { /* exporting? */
8354 /* export NAME (without =VALUE) */
8355 if (var) {
8356 var->flg_export = 1;
8357 debug_printf_env("%s: putenv '%s'\n", __func__, var->varstr);
8358 putenv(var->varstr);
8359 continue;
8360 }
8361 }
8362 /* Exporting non-existing variable.
8363 * bash does not put it in environment,
8364 * but remembers that it is exported,
8365 * and does put it in env when it is set later.
8366 * We just set it to "" and export. */
8367 /* Or, it's "local NAME" (without =VALUE).
8368 * bash sets the value to "". */
8369 name = xasprintf("%s=", name);
8370 } else {
8371 /* (Un)exporting/making local NAME=VALUE */
8372 name = xstrdup(name);
8373 }
8374 set_local_var(name, /*exp:*/ exp, /*lvl:*/ lvl, /*ro:*/ 0);
8375 } while (*++argv);
8376}
8377
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008378static int FAST_FUNC builtin_export(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008379{
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00008380 unsigned opt_unexport;
8381
Denys Vlasenkodf5131c2009-06-07 16:04:17 +02008382#if ENABLE_HUSH_EXPORT_N
8383 /* "!": do not abort on errors */
8384 opt_unexport = getopt32(argv, "!n");
8385 if (opt_unexport == (uint32_t)-1)
8386 return EXIT_FAILURE;
8387 argv += optind;
8388#else
8389 opt_unexport = 0;
8390 argv++;
8391#endif
8392
8393 if (argv[0] == NULL) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008394 char **e = environ;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008395 if (e) {
8396 while (*e) {
8397#if 0
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008398 puts(*e++);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008399#else
8400 /* ash emits: export VAR='VAL'
8401 * bash: declare -x VAR="VAL"
8402 * we follow ash example */
8403 const char *s = *e++;
8404 const char *p = strchr(s, '=');
8405
8406 if (!p) /* wtf? take next variable */
8407 continue;
8408 /* export var= */
8409 printf("export %.*s", (int)(p - s) + 1, s);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008410 print_escaped(p + 1);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008411 putchar('\n');
8412#endif
8413 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008414 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008415 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008416 return EXIT_SUCCESS;
8417 }
8418
Denys Vlasenko295fef82009-06-03 12:47:26 +02008419 helper_export_local(argv, (opt_unexport ? -1 : 1), 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008420
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008421 return EXIT_SUCCESS;
8422}
8423
Denys Vlasenko295fef82009-06-03 12:47:26 +02008424#if ENABLE_HUSH_LOCAL
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008425static int FAST_FUNC builtin_local(char **argv)
Denys Vlasenko295fef82009-06-03 12:47:26 +02008426{
8427 if (G.func_nest_level == 0) {
8428 bb_error_msg("%s: not in a function", argv[0]);
8429 return EXIT_FAILURE; /* bash compat */
8430 }
8431 helper_export_local(argv, 0, G.func_nest_level);
8432 return EXIT_SUCCESS;
8433}
8434#endif
8435
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008436static int FAST_FUNC builtin_trap(char **argv)
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008437{
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008438 int sig;
8439 char *new_cmd;
8440
8441 if (!G.traps)
8442 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
8443
8444 argv++;
8445 if (!*argv) {
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008446 int i;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008447 /* No args: print all trapped */
8448 for (i = 0; i < NSIG; ++i) {
8449 if (G.traps[i]) {
8450 printf("trap -- ");
8451 print_escaped(G.traps[i]);
Denys Vlasenkoe74aaf92009-09-27 02:05:45 +02008452 /* note: bash adds "SIG", but only if invoked
8453 * as "bash". If called as "sh", or if set -o posix,
8454 * then it prints short signal names.
8455 * We are printing short names: */
8456 printf(" %s\n", get_signame(i));
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008457 }
8458 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008459 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008460 return EXIT_SUCCESS;
8461 }
8462
8463 new_cmd = NULL;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008464 /* If first arg is a number: reset all specified signals */
8465 sig = bb_strtou(*argv, NULL, 10);
8466 if (errno == 0) {
8467 int ret;
8468 process_sig_list:
8469 ret = EXIT_SUCCESS;
8470 while (*argv) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008471 sighandler_t handler;
8472
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008473 sig = get_signum(*argv++);
8474 if (sig < 0 || sig >= NSIG) {
8475 ret = EXIT_FAILURE;
8476 /* Mimic bash message exactly */
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008477 bb_perror_msg("trap: %s: invalid signal specification", argv[-1]);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008478 continue;
8479 }
8480
8481 free(G.traps[sig]);
8482 G.traps[sig] = xstrdup(new_cmd);
8483
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008484 debug_printf("trap: setting SIG%s (%i) to '%s'\n",
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008485 get_signame(sig), sig, G.traps[sig]);
8486
8487 /* There is no signal for 0 (EXIT) */
8488 if (sig == 0)
8489 continue;
8490
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008491 if (new_cmd)
8492 handler = (new_cmd[0] ? record_pending_signo : SIG_IGN);
8493 else
8494 /* We are removing trap handler */
8495 handler = pick_sighandler(sig);
Denys Vlasenko0806e402011-05-12 23:06:20 +02008496 install_sighandler(sig, handler);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008497 }
8498 return ret;
8499 }
8500
8501 if (!argv[1]) { /* no second arg */
8502 bb_error_msg("trap: invalid arguments");
8503 return EXIT_FAILURE;
8504 }
8505
8506 /* First arg is "-": reset all specified to default */
8507 /* First arg is "--": skip it, the rest is "handler SIGs..." */
8508 /* Everything else: set arg as signal handler
8509 * (includes "" case, which ignores signal) */
8510 if (argv[0][0] == '-') {
8511 if (argv[0][1] == '\0') { /* "-" */
8512 /* new_cmd remains NULL: "reset these sigs" */
8513 goto reset_traps;
8514 }
8515 if (argv[0][1] == '-' && argv[0][2] == '\0') { /* "--" */
8516 argv++;
8517 }
8518 /* else: "-something", no special meaning */
8519 }
8520 new_cmd = *argv;
8521 reset_traps:
8522 argv++;
8523 goto process_sig_list;
8524}
8525
Mike Frysinger93cadc22009-05-27 17:06:25 -04008526/* http://www.opengroup.org/onlinepubs/9699919799/utilities/type.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008527static int FAST_FUNC builtin_type(char **argv)
Mike Frysinger93cadc22009-05-27 17:06:25 -04008528{
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008529 int ret = EXIT_SUCCESS;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008530
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008531 while (*++argv) {
Mike Frysinger93cadc22009-05-27 17:06:25 -04008532 const char *type;
Denys Vlasenko171932d2009-05-28 17:07:22 +02008533 char *path = NULL;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008534
8535 if (0) {} /* make conditional compile easier below */
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008536 /*else if (find_alias(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008537 type = "an alias";*/
8538#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008539 else if (find_function(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008540 type = "a function";
8541#endif
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008542 else if (find_builtin(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008543 type = "a shell builtin";
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008544 else if ((path = find_in_path(*argv)) != NULL)
8545 type = path;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008546 else {
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008547 bb_error_msg("type: %s: not found", *argv);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008548 ret = EXIT_FAILURE;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008549 continue;
8550 }
Mike Frysinger93cadc22009-05-27 17:06:25 -04008551
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008552 printf("%s is %s\n", *argv, type);
8553 free(path);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008554 }
8555
8556 return ret;
8557}
8558
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008559#if ENABLE_HUSH_JOB
8560/* built-in 'fg' and 'bg' handler */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008561static int FAST_FUNC builtin_fg_bg(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008562{
8563 int i, jobnum;
8564 struct pipe *pi;
8565
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008566 if (!G_interactive_fd)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008567 return EXIT_FAILURE;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008568
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008569 /* If they gave us no args, assume they want the last backgrounded task */
8570 if (!argv[1]) {
Denis Vlasenko87a86552008-07-29 19:43:10 +00008571 for (pi = G.job_list; pi; pi = pi->next) {
8572 if (pi->jobid == G.last_jobid) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008573 goto found;
8574 }
8575 }
8576 bb_error_msg("%s: no current job", argv[0]);
8577 return EXIT_FAILURE;
8578 }
8579 if (sscanf(argv[1], "%%%d", &jobnum) != 1) {
8580 bb_error_msg("%s: bad argument '%s'", argv[0], argv[1]);
8581 return EXIT_FAILURE;
8582 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008583 for (pi = G.job_list; pi; pi = pi->next) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008584 if (pi->jobid == jobnum) {
8585 goto found;
8586 }
8587 }
8588 bb_error_msg("%s: %d: no such job", argv[0], jobnum);
8589 return EXIT_FAILURE;
8590 found:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00008591 /* TODO: bash prints a string representation
8592 * of job being foregrounded (like "sleep 1 | cat") */
Mike Frysinger38478a62009-05-20 04:48:06 -04008593 if (argv[0][0] == 'f' && G_saved_tty_pgrp) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008594 /* Put the job into the foreground. */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008595 tcsetpgrp(G_interactive_fd, pi->pgrp);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008596 }
8597
8598 /* Restart the processes in the job */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008599 debug_printf_jobs("reviving %d procs, pgrp %d\n", pi->num_cmds, pi->pgrp);
8600 for (i = 0; i < pi->num_cmds; i++) {
8601 debug_printf_jobs("reviving pid %d\n", pi->cmds[i].pid);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008602 }
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008603 pi->stopped_cmds = 0;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008604
8605 i = kill(- pi->pgrp, SIGCONT);
8606 if (i < 0) {
8607 if (errno == ESRCH) {
8608 delete_finished_bg_job(pi);
8609 return EXIT_SUCCESS;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008610 }
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008611 bb_perror_msg("kill (SIGCONT)");
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008612 }
8613
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008614 if (argv[0][0] == 'f') {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008615 remove_bg_job(pi);
8616 return checkjobs_and_fg_shell(pi);
8617 }
8618 return EXIT_SUCCESS;
8619}
8620#endif
8621
8622#if ENABLE_HUSH_HELP
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008623static int FAST_FUNC builtin_help(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008624{
8625 const struct built_in_command *x;
8626
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008627 printf(
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008628 "Built-in commands:\n"
8629 "------------------\n");
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008630 for (x = bltins1; x != &bltins1[ARRAY_SIZE(bltins1)]; x++) {
Denys Vlasenko17323a62010-01-28 01:57:05 +01008631 if (x->b_descr)
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008632 printf("%-10s%s\n", x->b_cmd, x->b_descr);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008633 }
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008634 bb_putchar('\n');
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008635 return EXIT_SUCCESS;
8636}
8637#endif
8638
maxwen27116ba2015-08-14 21:41:28 +02008639#if MAX_HISTORY && ENABLE_FEATURE_EDITING
8640static int FAST_FUNC builtin_history(char **argv UNUSED_PARAM)
8641{
8642 show_history(G.line_input_state);
8643 return EXIT_SUCCESS;
8644}
8645#endif
8646
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008647#if ENABLE_HUSH_JOB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008648static int FAST_FUNC builtin_jobs(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008649{
8650 struct pipe *job;
8651 const char *status_string;
8652
Denis Vlasenko87a86552008-07-29 19:43:10 +00008653 for (job = G.job_list; job; job = job->next) {
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008654 if (job->alive_cmds == job->stopped_cmds)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008655 status_string = "Stopped";
8656 else
8657 status_string = "Running";
8658
8659 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->cmdtext);
8660 }
8661 return EXIT_SUCCESS;
8662}
8663#endif
8664
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008665#if HUSH_DEBUG
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008666static int FAST_FUNC builtin_memleak(char **argv UNUSED_PARAM)
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008667{
8668 void *p;
8669 unsigned long l;
8670
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008671# ifdef M_TRIM_THRESHOLD
Denys Vlasenko27726cb2009-09-12 14:48:33 +02008672 /* Optional. Reduces probability of false positives */
8673 malloc_trim(0);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008674# endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008675 /* Crude attempt to find where "free memory" starts,
8676 * sans fragmentation. */
8677 p = malloc(240);
8678 l = (unsigned long)p;
8679 free(p);
8680 p = malloc(3400);
8681 if (l < (unsigned long)p) l = (unsigned long)p;
8682 free(p);
8683
8684 if (!G.memleak_value)
8685 G.memleak_value = l;
Denys Vlasenko9038d6f2009-07-15 20:02:19 +02008686
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008687 l -= G.memleak_value;
8688 if ((long)l < 0)
8689 l = 0;
8690 l /= 1024;
8691 if (l > 127)
8692 l = 127;
8693
8694 /* Exitcode is "how many kilobytes we leaked since 1st call" */
8695 return l;
8696}
8697#endif
8698
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008699static int FAST_FUNC builtin_pwd(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008700{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008701 puts(get_cwd(0));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008702 return EXIT_SUCCESS;
8703}
8704
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008705/* Interruptibility of read builtin in bash
8706 * (tested on bash-4.2.8 by sending signals (not by ^C)):
8707 *
8708 * Empty trap makes read ignore corresponding signal, for any signal.
8709 *
8710 * SIGINT:
8711 * - terminates non-interactive shell;
8712 * - interrupts read in interactive shell;
8713 * if it has non-empty trap:
8714 * - executes trap and returns to command prompt in interactive shell;
8715 * - executes trap and returns to read in non-interactive shell;
8716 * SIGTERM:
8717 * - is ignored (does not interrupt) read in interactive shell;
8718 * - terminates non-interactive shell;
8719 * if it has non-empty trap:
8720 * - executes trap and returns to read;
8721 * SIGHUP:
8722 * - terminates shell (regardless of interactivity);
8723 * if it has non-empty trap:
8724 * - executes trap and returns to read;
8725 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008726static int FAST_FUNC builtin_read(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008727{
Denys Vlasenko03dad222010-01-12 23:29:57 +01008728 const char *r;
8729 char *opt_n = NULL;
8730 char *opt_p = NULL;
8731 char *opt_t = NULL;
8732 char *opt_u = NULL;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008733 const char *ifs;
Denys Vlasenko03dad222010-01-12 23:29:57 +01008734 int read_flags;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008735
Denys Vlasenko03dad222010-01-12 23:29:57 +01008736 /* "!": do not abort on errors.
8737 * Option string must start with "sr" to match BUILTIN_READ_xxx
8738 */
8739 read_flags = getopt32(argv, "!srn:p:t:u:", &opt_n, &opt_p, &opt_t, &opt_u);
8740 if (read_flags == (uint32_t)-1)
8741 return EXIT_FAILURE;
8742 argv += optind;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008743 ifs = get_local_var_value("IFS"); /* can be NULL */
8744
8745 again:
Denys Vlasenko03dad222010-01-12 23:29:57 +01008746 r = shell_builtin_read(set_local_var_from_halves,
8747 argv,
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008748 ifs,
Denys Vlasenko03dad222010-01-12 23:29:57 +01008749 read_flags,
8750 opt_n,
8751 opt_p,
8752 opt_t,
8753 opt_u
8754 );
8755
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008756 if ((uintptr_t)r == 1 && errno == EINTR) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008757 unsigned sig = check_and_run_traps();
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008758 if (sig && sig != SIGINT)
8759 goto again;
8760 }
8761
Denys Vlasenko03dad222010-01-12 23:29:57 +01008762 if ((uintptr_t)r > 1) {
8763 bb_error_msg("%s", r);
8764 r = (char*)(uintptr_t)1;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008765 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008766
Denys Vlasenko03dad222010-01-12 23:29:57 +01008767 return (uintptr_t)r;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008768}
8769
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008770/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#set
8771 * built-in 'set' handler
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008772 * SUSv3 says:
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008773 * set [-abCefhmnuvx] [-o option] [argument...]
8774 * set [+abCefhmnuvx] [+o option] [argument...]
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008775 * set -- [argument...]
8776 * set -o
8777 * set +o
8778 * Implementations shall support the options in both their hyphen and
8779 * plus-sign forms. These options can also be specified as options to sh.
8780 * Examples:
8781 * Write out all variables and their values: set
8782 * Set $1, $2, and $3 and set "$#" to 3: set c a b
8783 * Turn on the -x and -v options: set -xv
8784 * Unset all positional parameters: set --
8785 * Set $1 to the value of x, even if it begins with '-' or '+': set -- "$x"
8786 * Set the positional parameters to the expansion of x, even if x expands
8787 * with a leading '-' or '+': set -- $x
8788 *
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008789 * So far, we only support "set -- [argument...]" and some of the short names.
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008790 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008791static int FAST_FUNC builtin_set(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008792{
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008793 int n;
8794 char **pp, **g_argv;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008795 char *arg = *++argv;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008796
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008797 if (arg == NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008798 struct variable *e;
Denis Vlasenko87a86552008-07-29 19:43:10 +00008799 for (e = G.top_var; e; e = e->next)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008800 puts(e->varstr);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008801 return EXIT_SUCCESS;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008802 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008803
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008804 do {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008805 if (strcmp(arg, "--") == 0) {
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008806 ++argv;
8807 goto set_argv;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008808 }
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008809 if (arg[0] != '+' && arg[0] != '-')
8810 break;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008811 for (n = 1; arg[n]; ++n) {
8812 if (set_mode((arg[0] == '-'), arg[n], argv[1]))
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008813 goto error;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008814 if (arg[n] == 'o' && argv[1])
8815 argv++;
8816 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008817 } while ((arg = *++argv) != NULL);
8818 /* Now argv[0] is 1st argument */
8819
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008820 if (arg == NULL)
8821 return EXIT_SUCCESS;
8822 set_argv:
8823
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008824 /* NB: G.global_argv[0] ($0) is never freed/changed */
8825 g_argv = G.global_argv;
8826 if (G.global_args_malloced) {
8827 pp = g_argv;
8828 while (*++pp)
8829 free(*pp);
8830 g_argv[1] = NULL;
8831 } else {
8832 G.global_args_malloced = 1;
8833 pp = xzalloc(sizeof(pp[0]) * 2);
8834 pp[0] = g_argv[0]; /* retain $0 */
8835 g_argv = pp;
8836 }
8837 /* This realloc's G.global_argv */
8838 G.global_argv = pp = add_strings_to_strings(g_argv, argv, /*dup:*/ 1);
8839
8840 n = 1;
8841 while (*++pp)
8842 n++;
8843 G.global_argc = n;
8844
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008845 return EXIT_SUCCESS;
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008846
8847 /* Nothing known, so abort */
8848 error:
8849 bb_error_msg("set: %s: invalid option", arg);
8850 return EXIT_FAILURE;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008851}
8852
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008853static int FAST_FUNC builtin_shift(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008854{
8855 int n = 1;
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008856 argv = skip_dash_dash(argv);
8857 if (argv[0]) {
8858 n = atoi(argv[0]);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008859 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008860 if (n >= 0 && n < G.global_argc) {
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008861 if (G.global_args_malloced) {
8862 int m = 1;
8863 while (m <= n)
8864 free(G.global_argv[m++]);
8865 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008866 G.global_argc -= n;
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008867 memmove(&G.global_argv[1], &G.global_argv[n+1],
8868 G.global_argc * sizeof(G.global_argv[0]));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008869 return EXIT_SUCCESS;
8870 }
8871 return EXIT_FAILURE;
8872}
8873
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008874static int FAST_FUNC builtin_source(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008875{
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008876 char *arg_path, *filename;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008877 FILE *input;
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008878 save_arg_t sv;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008879#if ENABLE_HUSH_FUNCTIONS
8880 smallint sv_flg;
8881#endif
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008882
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008883 argv = skip_dash_dash(argv);
8884 filename = argv[0];
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008885 if (!filename) {
8886 /* bash says: "bash: .: filename argument required" */
8887 return 2; /* bash compat */
8888 }
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008889 arg_path = NULL;
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008890 if (!strchr(filename, '/')) {
8891 arg_path = find_in_path(filename);
8892 if (arg_path)
8893 filename = arg_path;
8894 }
8895 input = fopen_or_warn(filename, "r");
8896 free(arg_path);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008897 if (!input) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008898 /* bb_perror_msg("%s", *argv); - done by fopen_or_warn */
maxwen27116ba2015-08-14 21:41:28 +02008899 /* POSIX: non-interactive shell should abort here,
8900 * not merely fail. So far no one complained :)
8901 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008902 return EXIT_FAILURE;
8903 }
8904 close_on_exec_on(fileno(input));
8905
Mike Frysinger885b6f22009-04-18 21:04:25 +00008906#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008907 sv_flg = G.flag_return_in_progress;
8908 /* "we are inside sourced file, ok to use return" */
8909 G.flag_return_in_progress = -1;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008910#endif
maxwen27116ba2015-08-14 21:41:28 +02008911 if (argv[1])
8912 save_and_replace_G_args(&sv, argv);
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008913
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008914 parse_and_run_file(input);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008915 fclose(input);
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008916
maxwen27116ba2015-08-14 21:41:28 +02008917 if (argv[1])
8918 restore_G_args(&sv, argv);
Mike Frysinger885b6f22009-04-18 21:04:25 +00008919#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008920 G.flag_return_in_progress = sv_flg;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008921#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008922
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008923 return G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008924}
8925
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008926static int FAST_FUNC builtin_umask(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008927{
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008928 int rc;
8929 mode_t mask;
8930
8931 mask = umask(0);
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008932 argv = skip_dash_dash(argv);
8933 if (argv[0]) {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008934 mode_t old_mask = mask;
8935
8936 mask ^= 0777;
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008937 rc = bb_parse_mode(argv[0], &mask);
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008938 mask ^= 0777;
8939 if (rc == 0) {
8940 mask = old_mask;
8941 /* bash messages:
8942 * bash: umask: 'q': invalid symbolic mode operator
8943 * bash: umask: 999: octal number out of range
8944 */
Denys Vlasenko44c86ce2010-05-20 04:22:55 +02008945 bb_error_msg("%s: invalid mode '%s'", "umask", argv[0]);
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008946 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008947 } else {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008948 rc = 1;
8949 /* Mimic bash */
8950 printf("%04o\n", (unsigned) mask);
8951 /* fall through and restore mask which we set to 0 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008952 }
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008953 umask(mask);
8954
8955 return !rc; /* rc != 0 - success */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008956}
8957
Mike Frysingerd690f682009-03-30 06:50:54 +00008958/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#unset */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008959static int FAST_FUNC builtin_unset(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008960{
Mike Frysingerd690f682009-03-30 06:50:54 +00008961 int ret;
Denis Vlasenko28e67962009-04-26 23:22:40 +00008962 unsigned opts;
Mike Frysingerd690f682009-03-30 06:50:54 +00008963
Denis Vlasenko28e67962009-04-26 23:22:40 +00008964 /* "!": do not abort on errors */
8965 /* "+": stop at 1st non-option */
8966 opts = getopt32(argv, "!+vf");
8967 if (opts == (unsigned)-1)
8968 return EXIT_FAILURE;
8969 if (opts == 3) {
8970 bb_error_msg("unset: -v and -f are exclusive");
8971 return EXIT_FAILURE;
Mike Frysingerd690f682009-03-30 06:50:54 +00008972 }
Denis Vlasenko28e67962009-04-26 23:22:40 +00008973 argv += optind;
Mike Frysingerd690f682009-03-30 06:50:54 +00008974
8975 ret = EXIT_SUCCESS;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008976 while (*argv) {
Denis Vlasenko28e67962009-04-26 23:22:40 +00008977 if (!(opts & 2)) { /* not -f */
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008978 if (unset_local_var(*argv)) {
8979 /* unset <nonexistent_var> doesn't fail.
8980 * Error is when one tries to unset RO var.
8981 * Message was printed by unset_local_var. */
Mike Frysingerd690f682009-03-30 06:50:54 +00008982 ret = EXIT_FAILURE;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008983 }
Mike Frysingerd690f682009-03-30 06:50:54 +00008984 }
Denis Vlasenko40e84372009-04-18 11:23:38 +00008985#if ENABLE_HUSH_FUNCTIONS
8986 else {
8987 unset_func(*argv);
8988 }
8989#endif
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008990 argv++;
Mike Frysingerd690f682009-03-30 06:50:54 +00008991 }
8992 return ret;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008993}
Denis Vlasenkobcb25532008-07-28 23:04:34 +00008994
Mike Frysinger56bdea12009-03-28 20:01:58 +00008995/* http://www.opengroup.org/onlinepubs/9699919799/utilities/wait.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008996static int FAST_FUNC builtin_wait(char **argv)
Mike Frysinger56bdea12009-03-28 20:01:58 +00008997{
8998 int ret = EXIT_SUCCESS;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008999 int status;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009000
Denys Vlasenkob131cce2010-05-20 03:39:43 +02009001 argv = skip_dash_dash(argv);
9002 if (argv[0] == NULL) {
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009003 /* Don't care about wait results */
9004 /* Note 1: must wait until there are no more children */
9005 /* Note 2: must be interruptible */
9006 /* Examples:
9007 * $ sleep 3 & sleep 6 & wait
9008 * [1] 30934 sleep 3
9009 * [2] 30935 sleep 6
9010 * [1] Done sleep 3
9011 * [2] Done sleep 6
9012 * $ sleep 3 & sleep 6 & wait
9013 * [1] 30936 sleep 3
9014 * [2] 30937 sleep 6
9015 * [1] Done sleep 3
9016 * ^C <-- after ~4 sec from keyboard
9017 * $
9018 */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009019 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009020 int sig;
9021 sigset_t oldset, allsigs;
9022
9023 /* waitpid is not interruptible by SA_RESTARTed
9024 * signals which we use. Thus, this ugly dance:
9025 */
9026
9027 /* Make sure possible SIGCHLD is stored in kernel's
9028 * pending signal mask before we call waitpid.
9029 * Or else we may race with SIGCHLD, lose it,
9030 * and get stuck in sigwaitinfo...
9031 */
9032 sigfillset(&allsigs);
9033 sigprocmask(SIG_SETMASK, &allsigs, &oldset);
9034
9035 if (!sigisemptyset(&G.pending_set)) {
9036 /* Crap! we raced with some signal! */
9037 // sig = 0;
9038 goto restore;
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009039 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009040
9041 checkjobs(NULL); /* waitpid(WNOHANG) inside */
9042 if (errno == ECHILD) {
9043 sigprocmask(SIG_SETMASK, &oldset, NULL);
9044 break;
9045 }
9046
9047 /* Wait for SIGCHLD or any other signal */
9048 //sig = sigwaitinfo(&allsigs, NULL);
9049 /* It is vitally important for sigsuspend that SIGCHLD has non-DFL handler! */
9050 /* Note: sigsuspend invokes signal handler */
9051 sigsuspend(&oldset);
9052 restore:
9053 sigprocmask(SIG_SETMASK, &oldset, NULL);
9054
9055 /* So, did we get a signal? */
9056 //if (sig > 0)
9057 // raise(sig); /* run handler */
9058 sig = check_and_run_traps();
9059 if (sig /*&& sig != SIGCHLD - always true */) {
9060 /* see note 2 */
9061 ret = 128 + sig;
9062 break;
9063 }
9064 /* SIGCHLD, or no signal, or ignored one, such as SIGQUIT. Repeat */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009065 }
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009066 return ret;
9067 }
Mike Frysinger56bdea12009-03-28 20:01:58 +00009068
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009069 /* This is probably buggy wrt interruptible-ness */
Denis Vlasenkod5762932009-03-31 11:22:57 +00009070 while (*argv) {
9071 pid_t pid = bb_strtou(*argv, NULL, 10);
Mike Frysinger40b8dc42009-03-29 00:50:30 +00009072 if (errno) {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009073 /* mimic bash message */
9074 bb_error_msg("wait: '%s': not a pid or valid job spec", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009075 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00009076 }
9077 if (waitpid(pid, &status, 0) == pid) {
Mike Frysinger56bdea12009-03-28 20:01:58 +00009078 if (WIFSIGNALED(status))
9079 ret = 128 + WTERMSIG(status);
9080 else if (WIFEXITED(status))
9081 ret = WEXITSTATUS(status);
Denis Vlasenkod5762932009-03-31 11:22:57 +00009082 else /* wtf? */
Mike Frysinger56bdea12009-03-28 20:01:58 +00009083 ret = EXIT_FAILURE;
9084 } else {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009085 bb_perror_msg("wait %s", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009086 ret = 127;
9087 }
Denis Vlasenkod5762932009-03-31 11:22:57 +00009088 argv++;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009089 }
9090
9091 return ret;
9092}
9093
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009094#if ENABLE_HUSH_LOOPS || ENABLE_HUSH_FUNCTIONS
9095static unsigned parse_numeric_argv1(char **argv, unsigned def, unsigned def_min)
9096{
9097 if (argv[1]) {
9098 def = bb_strtou(argv[1], NULL, 10);
9099 if (errno || def < def_min || argv[2]) {
9100 bb_error_msg("%s: bad arguments", argv[0]);
9101 def = UINT_MAX;
9102 }
9103 }
9104 return def;
9105}
9106#endif
9107
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009108#if ENABLE_HUSH_LOOPS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009109static int FAST_FUNC builtin_break(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009110{
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009111 unsigned depth;
Denis Vlasenko87a86552008-07-29 19:43:10 +00009112 if (G.depth_of_loop == 0) {
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009113 bb_error_msg("%s: only meaningful in a loop", argv[0]);
Denis Vlasenkofcf37c32008-07-29 11:37:15 +00009114 return EXIT_SUCCESS; /* bash compat */
9115 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00009116 G.flag_break_continue++; /* BC_BREAK = 1 */
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009117
9118 G.depth_break_continue = depth = parse_numeric_argv1(argv, 1, 1);
9119 if (depth == UINT_MAX)
9120 G.flag_break_continue = BC_BREAK;
9121 if (G.depth_of_loop < depth)
Denis Vlasenko87a86552008-07-29 19:43:10 +00009122 G.depth_break_continue = G.depth_of_loop;
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009123
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009124 return EXIT_SUCCESS;
9125}
9126
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009127static int FAST_FUNC builtin_continue(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009128{
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009129 G.flag_break_continue = 1; /* BC_CONTINUE = 2 = 1+1 */
9130 return builtin_break(argv);
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009131}
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009132#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009133
9134#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009135static int FAST_FUNC builtin_return(char **argv)
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009136{
9137 int rc;
9138
9139 if (G.flag_return_in_progress != -1) {
9140 bb_error_msg("%s: not in a function or sourced script", argv[0]);
9141 return EXIT_FAILURE; /* bash compat */
9142 }
9143
9144 G.flag_return_in_progress = 1;
9145
9146 /* bash:
9147 * out of range: wraps around at 256, does not error out
9148 * non-numeric param:
9149 * f() { false; return qwe; }; f; echo $?
9150 * bash: return: qwe: numeric argument required <== we do this
9151 * 255 <== we also do this
9152 */
9153 rc = parse_numeric_argv1(argv, G.last_exitcode, 0);
9154 return rc;
9155}
9156#endif