blob: eabe83ac63e33dbcfc87b765a3353fa83704e46c [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 Vlasenko3fa97af2014-04-15 11:43:29 +020094#include <sys/utsname.h> /* for setting $HOSTNAME */
Denys Vlasenko03dad222010-01-12 23:29:57 +010095
Denys Vlasenko20704f02011-03-23 17:59:27 +010096#include "busybox.h" /* for APPLET_IS_NOFORK/NOEXEC */
97#include "unicode.h"
Denys Vlasenko03dad222010-01-12 23:29:57 +010098#include "shell_common.h"
Mike Frysinger98c52642009-04-02 10:02:37 +000099#include "math.h"
Mike Frysingera4f331d2009-04-07 06:03:22 +0000100#include "match.h"
Denys Vlasenkocbe0b7f2009-10-09 22:00:58 +0200101#if ENABLE_HUSH_RANDOM_SUPPORT
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200102# include "random.h"
Denys Vlasenko76ace252009-10-12 15:25:01 +0200103#else
104# define CLEAR_RANDOM_T(rnd) ((void)0)
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200105#endif
Denis Vlasenko50f3aa42009-04-07 10:52:40 +0000106#ifndef PIPE_BUF
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200107# define PIPE_BUF 4096 /* amount of buffering in a pipe */
Denis Vlasenko50f3aa42009-04-07 10:52:40 +0000108#endif
Mike Frysinger98c52642009-04-02 10:02:37 +0000109
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200110//config:config HUSH
111//config: bool "hush"
112//config: default y
113//config: help
Denys Vlasenko771f1992010-07-16 14:31:34 +0200114//config: hush is a small shell (25k). It handles the normal flow control
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200115//config: constructs such as if/then/elif/else/fi, for/in/do/done, while loops,
116//config: case/esac. Redirections, here documents, $((arithmetic))
117//config: and functions are supported.
118//config:
119//config: It will compile and work on no-mmu systems.
120//config:
Denys Vlasenkoe2069fb2010-10-04 00:01:47 +0200121//config: It does not handle select, aliases, tilde expansion,
122//config: &>file and >&file redirection of stdout+stderr.
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200123//config:
124//config:config HUSH_BASH_COMPAT
125//config: bool "bash-compatible extensions"
126//config: default y
127//config: depends on HUSH
128//config: help
129//config: Enable bash-compatible extensions.
130//config:
Denys Vlasenko9e800222010-10-03 14:28:04 +0200131//config:config HUSH_BRACE_EXPANSION
132//config: bool "Brace expansion"
133//config: default y
134//config: depends on HUSH_BASH_COMPAT
135//config: help
136//config: Enable {abc,def} extension.
137//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200138//config:config HUSH_HELP
139//config: bool "help builtin"
140//config: default y
141//config: depends on HUSH
142//config: help
143//config: Enable help builtin in hush. Code size + ~1 kbyte.
144//config:
145//config:config HUSH_INTERACTIVE
146//config: bool "Interactive mode"
147//config: default y
148//config: depends on HUSH
149//config: help
150//config: Enable interactive mode (prompt and command editing).
151//config: Without this, hush simply reads and executes commands
152//config: from stdin just like a shell script from a file.
153//config: No prompt, no PS1/PS2 magic shell variables.
154//config:
Denys Vlasenko99862cb2010-09-12 17:34:13 +0200155//config:config HUSH_SAVEHISTORY
156//config: bool "Save command history to .hush_history"
157//config: default y
158//config: depends on HUSH_INTERACTIVE && FEATURE_EDITING_SAVEHISTORY
159//config: help
160//config: Enable history saving in hush.
161//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200162//config:config HUSH_JOB
163//config: bool "Job control"
164//config: default y
165//config: depends on HUSH_INTERACTIVE
166//config: help
167//config: Enable job control: Ctrl-Z backgrounds, Ctrl-C interrupts current
168//config: command (not entire shell), fg/bg builtins work. Without this option,
169//config: "cmd &" still works by simply spawning a process and immediately
170//config: prompting for next command (or executing next command in a script),
171//config: but no separate process group is formed.
172//config:
173//config:config HUSH_TICK
174//config: bool "Process substitution"
175//config: default y
176//config: depends on HUSH
177//config: help
178//config: Enable process substitution `command` and $(command) in hush.
179//config:
180//config:config HUSH_IF
181//config: bool "Support if/then/elif/else/fi"
182//config: default y
183//config: depends on HUSH
184//config: help
185//config: Enable if/then/elif/else/fi in hush.
186//config:
187//config:config HUSH_LOOPS
188//config: bool "Support for, while and until loops"
189//config: default y
190//config: depends on HUSH
191//config: help
192//config: Enable for, while and until loops in hush.
193//config:
194//config:config HUSH_CASE
195//config: bool "Support case ... esac statement"
196//config: default y
197//config: depends on HUSH
198//config: help
199//config: Enable case ... esac statement in hush. +400 bytes.
200//config:
201//config:config HUSH_FUNCTIONS
202//config: bool "Support funcname() { commands; } syntax"
203//config: default y
204//config: depends on HUSH
205//config: help
206//config: Enable support for shell functions in hush. +800 bytes.
207//config:
208//config:config HUSH_LOCAL
209//config: bool "Support local builtin"
210//config: default y
211//config: depends on HUSH_FUNCTIONS
212//config: help
213//config: Enable support for local variables in functions.
214//config:
215//config:config HUSH_RANDOM_SUPPORT
216//config: bool "Pseudorandom generator and $RANDOM variable"
217//config: default y
218//config: depends on HUSH
219//config: help
220//config: Enable pseudorandom generator and dynamic variable "$RANDOM".
221//config: Each read of "$RANDOM" will generate a new pseudorandom value.
222//config:
223//config:config HUSH_EXPORT_N
224//config: bool "Support 'export -n' option"
225//config: default y
226//config: depends on HUSH
227//config: help
228//config: export -n unexports variables. It is a bash extension.
229//config:
230//config:config HUSH_MODE_X
231//config: bool "Support 'hush -x' option and 'set -x' command"
232//config: default y
233//config: depends on HUSH
234//config: help
Denys Vlasenko29082232010-07-16 13:52:32 +0200235//config: This instructs hush to print commands before execution.
236//config: Adds ~300 bytes.
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200237//config:
Denys Vlasenko6adf2aa2010-07-16 19:26:38 +0200238//config:config MSH
239//config: bool "msh (deprecated: aliased to hush)"
240//config: default n
241//config: select HUSH
242//config: help
243//config: msh is deprecated and will be removed, please migrate to hush.
244//config:
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200245
Denys Vlasenko20704f02011-03-23 17:59:27 +0100246//applet:IF_HUSH(APPLET(hush, BB_DIR_BIN, BB_SUID_DROP))
247//applet:IF_MSH(APPLET(msh, BB_DIR_BIN, BB_SUID_DROP))
248//applet:IF_FEATURE_SH_IS_HUSH(APPLET_ODDNAME(sh, hush, BB_DIR_BIN, BB_SUID_DROP, sh))
249//applet:IF_FEATURE_BASH_IS_HUSH(APPLET_ODDNAME(bash, hush, BB_DIR_BIN, BB_SUID_DROP, bash))
250
251//kbuild:lib-$(CONFIG_HUSH) += hush.o match.o shell_common.o
252//kbuild:lib-$(CONFIG_HUSH_RANDOM_SUPPORT) += random.o
253
Dan Fandrich89ca2f92010-11-28 01:54:39 +0100254/* -i (interactive) and -s (read stdin) are also accepted,
255 * but currently do nothing, therefore aren't shown in help.
256 * NOMMU-specific options are not meant to be used by users,
257 * therefore we don't show them either.
258 */
259//usage:#define hush_trivial_usage
Denys Vlasenkof58f7052011-05-12 02:10:33 +0200260//usage: "[-nxl] [-c 'SCRIPT' [ARG0 [ARGS]] / FILE [ARGS]]"
Denys Vlasenkob0b83432011-03-07 12:34:59 +0100261//usage:#define hush_full_usage "\n\n"
262//usage: "Unix shell interpreter"
263
Dan Fandrich89ca2f92010-11-28 01:54:39 +0100264//usage:#define msh_trivial_usage hush_trivial_usage
Denys Vlasenkob0b83432011-03-07 12:34:59 +0100265//usage:#define msh_full_usage hush_full_usage
266
267//usage:#if ENABLE_FEATURE_SH_IS_HUSH
268//usage:# define sh_trivial_usage hush_trivial_usage
269//usage:# define sh_full_usage hush_full_usage
270//usage:#endif
271//usage:#if ENABLE_FEATURE_BASH_IS_HUSH
272//usage:# define bash_trivial_usage hush_trivial_usage
273//usage:# define bash_full_usage hush_full_usage
274//usage:#endif
Denys Vlasenko202a2d12010-07-16 12:36:14 +0200275
Denis Vlasenko1943aec2009-04-09 14:15:57 +0000276
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200277/* Build knobs */
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +0000278#define LEAK_HUNTING 0
279#define BUILD_AS_NOMMU 0
280/* Enable/disable sanity checks. Ok to enable in production,
281 * only adds a bit of bloat. Set to >1 to get non-production level verbosity.
282 * Keeping 1 for now even in released versions.
283 */
284#define HUSH_DEBUG 1
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200285/* Slightly bigger (+200 bytes), but faster hush.
286 * So far it only enables a trick with counting SIGCHLDs and forks,
287 * which allows us to do fewer waitpid's.
288 * (we can detect a case where neither forks were done nor SIGCHLDs happened
289 * and therefore waitpid will return the same result as last time)
290 */
291#define ENABLE_HUSH_FAST 0
Denys Vlasenko9297dbc2010-07-05 21:37:12 +0200292/* TODO: implement simplified code for users which do not need ${var%...} ops
293 * So far ${var%...} ops are always enabled:
294 */
295#define ENABLE_HUSH_DOLLAR_OPS 1
Denis Vlasenko1943aec2009-04-09 14:15:57 +0000296
297
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +0000298#if BUILD_AS_NOMMU
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000299# undef BB_MMU
300# undef USE_FOR_NOMMU
301# undef USE_FOR_MMU
302# define BB_MMU 0
303# define USE_FOR_NOMMU(...) __VA_ARGS__
304# define USE_FOR_MMU(...)
305#endif
306
Denys Vlasenko1fcbff22010-06-26 02:40:08 +0200307#include "NUM_APPLETS.h"
Denys Vlasenko14974842010-03-23 01:08:26 +0100308#if NUM_APPLETS == 1
Denis Vlasenko61befda2008-11-25 01:36:03 +0000309/* STANDALONE does not make sense, and won't compile */
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000310# undef CONFIG_FEATURE_SH_STANDALONE
311# undef ENABLE_FEATURE_SH_STANDALONE
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000312# undef IF_FEATURE_SH_STANDALONE
Denys Vlasenko14974842010-03-23 01:08:26 +0100313# undef IF_NOT_FEATURE_SH_STANDALONE
314# define ENABLE_FEATURE_SH_STANDALONE 0
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000315# define IF_FEATURE_SH_STANDALONE(...)
316# define IF_NOT_FEATURE_SH_STANDALONE(...) __VA_ARGS__
Denis Vlasenko61befda2008-11-25 01:36:03 +0000317#endif
318
Denis Vlasenko05743d72008-02-10 12:10:08 +0000319#if !ENABLE_HUSH_INTERACTIVE
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000320# undef ENABLE_FEATURE_EDITING
321# define ENABLE_FEATURE_EDITING 0
322# undef ENABLE_FEATURE_EDITING_FANCY_PROMPT
323# define ENABLE_FEATURE_EDITING_FANCY_PROMPT 0
Denys Vlasenko8cab6672012-04-20 14:48:00 +0200324# undef ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
325# define ENABLE_FEATURE_EDITING_SAVE_ON_EXIT 0
Denis Vlasenko8412d792007-10-01 09:59:47 +0000326#endif
327
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000328/* Do we support ANY keywords? */
329#if ENABLE_HUSH_IF || ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000330# define HAS_KEYWORDS 1
331# define IF_HAS_KEYWORDS(...) __VA_ARGS__
332# define IF_HAS_NO_KEYWORDS(...)
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000333#else
Denis Vlasenkoce4acbb2009-04-10 23:23:41 +0000334# define HAS_KEYWORDS 0
335# define IF_HAS_KEYWORDS(...)
336# define IF_HAS_NO_KEYWORDS(...) __VA_ARGS__
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000337#endif
Denis Vlasenko8412d792007-10-01 09:59:47 +0000338
Denis Vlasenkod01ff132007-05-02 21:40:23 +0000339/* If you comment out one of these below, it will be #defined later
340 * to perform debug printfs to stderr: */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000341#define debug_printf(...) do {} while (0)
Denis Vlasenko400c5b62007-05-04 13:07:27 +0000342/* Finer-grained debug switches */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000343#define debug_printf_parse(...) do {} while (0)
344#define debug_print_tree(a, b) do {} while (0)
345#define debug_printf_exec(...) do {} while (0)
Denis Vlasenkof886fd22008-10-13 12:36:05 +0000346#define debug_printf_env(...) do {} while (0)
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000347#define debug_printf_jobs(...) do {} while (0)
348#define debug_printf_expand(...) do {} while (0)
Denys Vlasenko1e811b12010-05-22 03:12:29 +0200349#define debug_printf_varexp(...) do {} while (0)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +0000350#define debug_printf_glob(...) do {} while (0)
351#define debug_printf_list(...) do {} while (0)
Denis Vlasenko30c9cc52008-06-17 07:24:29 +0000352#define debug_printf_subst(...) do {} while (0)
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000353#define debug_printf_clean(...) do {} while (0)
Denis Vlasenkod01ff132007-05-02 21:40:23 +0000354
Denis Vlasenkob6e65562009-04-03 16:49:04 +0000355#define ERR_PTR ((void*)(long)1)
356
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200357#define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000358
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200359#define _SPECIAL_VARS_STR "_*@$!?#"
360#define SPECIAL_VARS_STR ("_*@$!?#" + 1)
361#define NUMERIC_SPECVARS_STR ("_*@$!?#" + 3)
Denys Vlasenko36f774a2010-09-05 14:45:38 +0200362#if ENABLE_HUSH_BASH_COMPAT
363/* Support / and // replace ops */
364/* Note that // is stored as \ in "encoded" string representation */
365# define VAR_ENCODED_SUBST_OPS "\\/%#:-=+?"
366# define VAR_SUBST_OPS ("\\/%#:-=+?" + 1)
367# define MINUS_PLUS_EQUAL_QUESTION ("\\/%#:-=+?" + 5)
368#else
369# define VAR_ENCODED_SUBST_OPS "%#:-=+?"
370# define VAR_SUBST_OPS "%#:-=+?"
371# define MINUS_PLUS_EQUAL_QUESTION ("%#:-=+?" + 3)
372#endif
Denys Vlasenkoe85248a2010-05-22 06:20:26 +0200373
374#define SPECIAL_VAR_SYMBOL 3
Eric Andersen25f27032001-04-26 23:22:31 +0000375
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200376struct variable;
377
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000378static const char hush_version_str[] ALIGN1 = "HUSH_VERSION="BB_VER;
379
380/* This supports saving pointers malloced in vfork child,
Denis Vlasenkoc376db32009-04-15 21:49:48 +0000381 * to be freed in the parent.
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000382 */
383#if !BB_MMU
384typedef struct nommu_save_t {
385 char **new_env;
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200386 struct variable *old_vars;
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000387 char **argv;
Denis Vlasenko27014ed2009-04-15 21:48:23 +0000388 char **argv_from_re_execing;
Denis Vlasenkocc90f442009-04-08 16:40:34 +0000389} nommu_save_t;
390#endif
391
Denys Vlasenko9b782552010-09-08 13:33:26 +0200392enum {
Eric Andersen25f27032001-04-26 23:22:31 +0000393 RES_NONE = 0,
Denis Vlasenko06810332007-05-21 23:30:54 +0000394#if ENABLE_HUSH_IF
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000395 RES_IF ,
396 RES_THEN ,
397 RES_ELIF ,
398 RES_ELSE ,
399 RES_FI ,
Denis Vlasenko06810332007-05-21 23:30:54 +0000400#endif
401#if ENABLE_HUSH_LOOPS
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000402 RES_FOR ,
403 RES_WHILE ,
404 RES_UNTIL ,
405 RES_DO ,
406 RES_DONE ,
Denis Vlasenkod91afa32008-07-29 11:10:01 +0000407#endif
408#if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000409 RES_IN ,
Denis Vlasenko06810332007-05-21 23:30:54 +0000410#endif
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000411#if ENABLE_HUSH_CASE
412 RES_CASE ,
Denys Vlasenkoe9bda902009-05-23 16:50:07 +0200413 /* three pseudo-keywords support contrived "case" syntax: */
414 RES_CASE_IN, /* "case ... IN", turns into RES_MATCH when IN is observed */
415 RES_MATCH , /* "word)" */
416 RES_CASE_BODY, /* "this command is inside CASE" */
Denis Vlasenko17f02e72008-07-14 04:32:29 +0000417 RES_ESAC ,
418#endif
419 RES_XXXX ,
420 RES_SNTX
Denys Vlasenko9b782552010-09-08 13:33:26 +0200421};
Denis Vlasenkoc7985b72008-06-17 05:43:38 +0000422
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000423typedef struct o_string {
424 char *data;
425 int length; /* position where data is appended */
426 int maxlen;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +0200427 int o_expflags;
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000428 /* At least some part of the string was inside '' or "",
429 * possibly empty one: word"", wo''rd etc. */
Denys Vlasenko38292b62010-09-05 14:49:40 +0200430 smallint has_quoted_part;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000431 smallint has_empty_slot;
432 smallint o_assignment; /* 0:maybe, 1:yes, 2:no */
433} o_string;
434enum {
Denys Vlasenko0e13b402010-09-21 12:35:39 +0200435 EXP_FLAG_SINGLEWORD = 0x80, /* must be 0x80 */
436 EXP_FLAG_GLOB = 0x2,
437 /* Protect newly added chars against globbing
438 * by prepending \ to *, ?, [, \ */
439 EXP_FLAG_ESC_GLOB_CHARS = 0x1,
440};
441enum {
442 MAYBE_ASSIGNMENT = 0,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000443 DEFINITELY_ASSIGNMENT = 1,
Denys Vlasenko0e13b402010-09-21 12:35:39 +0200444 NOT_ASSIGNMENT = 2,
Maninder Singh97c64912015-05-25 13:46:36 +0200445 /* Not an assignment, but next word may be: "if v=xyz cmd;" */
Denys Vlasenko0e13b402010-09-21 12:35:39 +0200446 WORD_IS_KEYWORD = 3,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000447};
448/* Used for initialization: o_string foo = NULL_O_STRING; */
449#define NULL_O_STRING { NULL }
450
Denys Vlasenko29f9b722011-05-14 11:27:36 +0200451#ifndef debug_printf_parse
452static const char *const assignment_flag[] = {
453 "MAYBE_ASSIGNMENT",
454 "DEFINITELY_ASSIGNMENT",
455 "NOT_ASSIGNMENT",
456 "WORD_IS_KEYWORD",
457};
458#endif
459
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000460typedef struct in_str {
461 const char *p;
462 /* eof_flag=1: last char in ->p is really an EOF */
463 char eof_flag; /* meaningless if ->p == NULL */
464 char peek_buf[2];
465#if ENABLE_HUSH_INTERACTIVE
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000466 smallint promptmode; /* 0: PS1, 1: PS2 */
467#endif
Denys Vlasenkocecbc982011-03-30 18:54:52 +0200468 int last_char;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000469 FILE *file;
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200470 int (*get) (struct in_str *) FAST_FUNC;
471 int (*peek) (struct in_str *) FAST_FUNC;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000472} in_str;
473#define i_getch(input) ((input)->get(input))
474#define i_peek(input) ((input)->peek(input))
475
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200476/* The descrip member of this structure is only used to make
477 * debugging output pretty */
478static const struct {
479 int mode;
480 signed char default_fd;
481 char descrip[3];
482} redir_table[] = {
483 { O_RDONLY, 0, "<" },
484 { O_CREAT|O_TRUNC|O_WRONLY, 1, ">" },
485 { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
486 { O_CREAT|O_RDWR, 1, "<>" },
487 { O_RDONLY, 0, "<<" },
488/* Should not be needed. Bogus default_fd helps in debugging */
489/* { O_RDONLY, 77, "<<" }, */
490};
491
Eric Andersen25f27032001-04-26 23:22:31 +0000492struct redir_struct {
Denis Vlasenko55789c62008-06-18 16:30:42 +0000493 struct redir_struct *next;
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000494 char *rd_filename; /* filename */
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000495 int rd_fd; /* fd to redirect */
496 /* fd to redirect to, or -3 if rd_fd is to be closed (n>&-) */
497 int rd_dup;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000498 smallint rd_type; /* (enum redir_type) */
499 /* note: for heredocs, rd_filename contains heredoc delimiter,
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000500 * and subsequently heredoc itself; and rd_dup is a bitmask:
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200501 * bit 0: do we need to trim leading tabs?
502 * bit 1: is heredoc quoted (<<'delim' syntax) ?
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000503 */
Eric Andersen25f27032001-04-26 23:22:31 +0000504};
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000505typedef enum redir_type {
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200506 REDIRECT_INPUT = 0,
507 REDIRECT_OVERWRITE = 1,
508 REDIRECT_APPEND = 2,
509 REDIRECT_IO = 3,
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000510 REDIRECT_HEREDOC = 4,
Denys Vlasenko764b2f02009-06-07 16:05:04 +0200511 REDIRECT_HEREDOC2 = 5, /* REDIRECT_HEREDOC after heredoc is loaded */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +0000512
513 REDIRFD_CLOSE = -3,
514 REDIRFD_SYNTAX_ERR = -2,
Denis Vlasenko835fcfd2009-04-10 13:51:56 +0000515 REDIRFD_TO_FILE = -1,
516 /* otherwise, rd_fd is redirected to rd_dup */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +0000517
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +0000518 HEREDOC_SKIPTABS = 1,
519 HEREDOC_QUOTED = 2,
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +0000520} redir_type;
521
Eric Andersen25f27032001-04-26 23:22:31 +0000522
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000523struct command {
Denis Vlasenko0c886c62007-01-30 22:30:09 +0000524 pid_t pid; /* 0 if exited */
Denis Vlasenko2b576b82008-08-04 00:46:07 +0000525 int assignment_cnt; /* how many argv[i] are assignments? */
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200526 smallint cmd_type; /* CMD_xxx */
527#define CMD_NORMAL 0
528#define CMD_SUBSHELL 1
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200529#if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenkod383b492010-09-06 10:22:13 +0200530/* used for "[[ EXPR ]]" */
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200531# define CMD_SINGLEWORD_NOGLOB 2
Denis Vlasenkoed055212009-04-11 10:37:10 +0000532#endif
Denys Vlasenko9ca656b2009-06-10 13:39:35 +0200533#if ENABLE_HUSH_FUNCTIONS
534# define CMD_FUNCDEF 3
535#endif
536
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100537 smalluint cmd_exitcode;
Denys Vlasenkocb6ff252009-05-04 00:14:30 +0200538 /* if non-NULL, this "command" is { list }, ( list ), or a compound statement */
539 struct pipe *group;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000540#if !BB_MMU
541 char *group_as_string;
542#endif
Denis Vlasenkoed055212009-04-11 10:37:10 +0000543#if ENABLE_HUSH_FUNCTIONS
544 struct function *child_func;
545/* This field is used to prevent a bug here:
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200546 * while...do f1() {a;}; f1; f1() {b;}; f1; done
Denis Vlasenkoed055212009-04-11 10:37:10 +0000547 * When we execute "f1() {a;}" cmd, we create new function and clear
548 * cmd->group, cmd->group_as_string, cmd->argv[0].
Denys Vlasenko9d617c42009-06-09 18:40:52 +0200549 * When we execute "f1() {b;}", we notice that f1 exists,
550 * and that its "parent cmd" struct is still "alive",
Denis Vlasenkoed055212009-04-11 10:37:10 +0000551 * we put those fields back into cmd->xxx
552 * (struct function has ->parent_cmd ptr to facilitate that).
553 * When we loop back, we can execute "f1() {a;}" again and set f1 correctly.
554 * Without this trick, loop would execute a;b;b;b;...
555 * instead of correct sequence a;b;a;b;...
556 * When command is freed, it severs the link
557 * (sets ->child_func->parent_cmd to NULL).
558 */
559#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000560 char **argv; /* command name and arguments */
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000561/* argv vector may contain variable references (^Cvar^C, ^C0^C etc)
562 * and on execution these are substituted with their values.
563 * Substitution can make _several_ words out of one argv[n]!
564 * Example: argv[0]=='.^C*^C.' here: echo .$*.
Denis Vlasenkoc7985b72008-06-17 05:43:38 +0000565 * References of the form ^C`cmd arg^C are `cmd arg` substitutions.
Denis Vlasenko03eb8bf2007-05-14 16:19:34 +0000566 */
Denis Vlasenkoed055212009-04-11 10:37:10 +0000567 struct redir_struct *redirects; /* I/O redirections */
568};
Denis Vlasenkof8c1f022009-04-17 11:55:42 +0000569/* Is there anything in this command at all? */
570#define IS_NULL_CMD(cmd) \
571 (!(cmd)->group && !(cmd)->argv && !(cmd)->redirects)
572
Eric Andersen25f27032001-04-26 23:22:31 +0000573struct pipe {
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000574 struct pipe *next;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000575 int num_cmds; /* total number of commands in pipe */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000576 int alive_cmds; /* number of commands running (not exited) */
577 int stopped_cmds; /* number of commands alive, but stopped */
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +0000578#if ENABLE_HUSH_JOB
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000579 int jobid; /* job number */
Denis Vlasenko0c886c62007-01-30 22:30:09 +0000580 pid_t pgrp; /* process group ID for the job */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000581 char *cmdtext; /* name of job */
Denis Vlasenkob81b3df2007-04-28 16:48:04 +0000582#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000583 struct command *cmds; /* array of commands in pipe */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000584 smallint followup; /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
Denis Vlasenko5ec61322008-06-24 00:50:07 +0000585 IF_HAS_KEYWORDS(smallint pi_inverted;) /* "! cmd | cmd" */
586 IF_HAS_KEYWORDS(smallint res_word;) /* needed for if, for, while, until... */
Eric Andersen25f27032001-04-26 23:22:31 +0000587};
Denis Vlasenkoa2b11e32009-04-06 14:11:13 +0000588typedef enum pipe_style {
589 PIPE_SEQ = 1,
590 PIPE_AND = 2,
591 PIPE_OR = 3,
592 PIPE_BG = 4,
593} pipe_style;
Denis Vlasenkof8c1f022009-04-17 11:55:42 +0000594/* Is there anything in this pipe at all? */
595#define IS_NULL_PIPE(pi) \
596 ((pi)->num_cmds == 0 IF_HAS_KEYWORDS( && (pi)->res_word == RES_NONE))
Eric Andersen25f27032001-04-26 23:22:31 +0000597
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000598/* This holds pointers to the various results of parsing */
599struct parse_context {
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000600 /* linked list of pipes */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000601 struct pipe *list_head;
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000602 /* last pipe (being constructed right now) */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000603 struct pipe *pipe;
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000604 /* last command in pipe (being constructed right now) */
605 struct command *command;
606 /* last redirect in command->redirects list */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000607 struct redir_struct *pending_redirect;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +0000608#if !BB_MMU
609 o_string as_string;
610#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000611#if HAS_KEYWORDS
612 smallint ctx_res_w;
613 smallint ctx_inverted; /* "! cmd | cmd" */
614#if ENABLE_HUSH_CASE
615 smallint ctx_dsemicolon; /* ";;" seen */
616#endif
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000617 /* bitmask of FLAG_xxx, for figuring out valid reserved words */
618 int old_flag;
619 /* group we are enclosed in:
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000620 * example: "if pipe1; pipe2; then pipe3; fi"
621 * when we see "if" or "then", we malloc and copy current context,
622 * and make ->stack point to it. then we parse pipeN.
623 * when closing "then" / fi" / whatever is found,
624 * we move list_head into ->stack->command->group,
625 * copy ->stack into current context, and delete ->stack.
626 * (parsing of { list } and ( list ) doesn't use this method)
Denis Vlasenkof9f74292009-04-03 00:07:05 +0000627 */
Denis Vlasenko9af22c72008-10-09 12:54:58 +0000628 struct parse_context *stack;
629#endif
630};
631
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000632/* On program start, environ points to initial environment.
633 * putenv adds new pointers into it, unsetenv removes them.
634 * Neither of these (de)allocates the strings.
635 * setenv allocates new strings in malloc space and does putenv,
636 * and thus setenv is unusable (leaky) for shell's purposes */
637#define setenv(...) setenv_is_leaky_dont_use()
638struct variable {
639 struct variable *next;
Denis Vlasenko28c0f0f2007-05-25 02:46:01 +0000640 char *varstr; /* points to "name=" portion */
Denys Vlasenko295fef82009-06-03 12:47:26 +0200641#if ENABLE_HUSH_LOCAL
642 unsigned func_nest_level;
643#endif
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000644 int max_len; /* if > 0, name is part of initial env; else name is malloced */
645 smallint flg_export; /* putenv should be done on this var */
Denis Vlasenko219e88d2007-05-21 10:18:23 +0000646 smallint flg_read_only;
Eric Andersen9ffb7dd2001-05-19 03:00:46 +0000647};
648
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000649enum {
Denis Vlasenkobcb25532008-07-28 23:04:34 +0000650 BC_BREAK = 1,
651 BC_CONTINUE = 2,
652};
653
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000654#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000655struct function {
656 struct function *next;
657 char *name;
Denis Vlasenkoed055212009-04-11 10:37:10 +0000658 struct command *parent_cmd;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000659 struct pipe *body;
Denys Vlasenkoc1947f12009-10-23 01:30:26 +0200660# if !BB_MMU
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000661 char *body_as_string;
Denys Vlasenkoc1947f12009-10-23 01:30:26 +0200662# endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000663};
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000664#endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000665
Denis Vlasenkod76c0492007-05-25 02:16:25 +0000666
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100667/* set -/+o OPT support. (TODO: make it optional)
668 * bash supports the following opts:
669 * allexport off
670 * braceexpand on
671 * emacs on
672 * errexit off
673 * errtrace off
674 * functrace off
675 * hashall on
676 * histexpand off
677 * history on
678 * ignoreeof off
679 * interactive-comments on
680 * keyword off
681 * monitor on
682 * noclobber off
683 * noexec off
684 * noglob off
685 * nolog off
686 * notify off
687 * nounset off
688 * onecmd off
689 * physical off
690 * pipefail off
691 * posix off
692 * privileged off
693 * verbose off
694 * vi off
695 * xtrace off
696 */
Dan Fandrich85c62472010-11-20 13:05:17 -0800697static const char o_opt_strings[] ALIGN1 =
698 "pipefail\0"
699 "noexec\0"
700#if ENABLE_HUSH_MODE_X
701 "xtrace\0"
702#endif
703 ;
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100704enum {
705 OPT_O_PIPEFAIL,
Dan Fandrich85c62472010-11-20 13:05:17 -0800706 OPT_O_NOEXEC,
707#if ENABLE_HUSH_MODE_X
708 OPT_O_XTRACE,
709#endif
Denys Vlasenko6696eac2010-11-14 02:01:50 +0100710 NUM_OPT_O
711};
712
713
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000714/* "Globals" within this file */
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000715/* Sorted roughly by size (smaller offsets == smaller code) */
716struct globals {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000717 /* interactive_fd != 0 means we are an interactive shell.
718 * If we are, then saved_tty_pgrp can also be != 0, meaning
719 * that controlling tty is available. With saved_tty_pgrp == 0,
720 * job control still works, but terminal signals
721 * (^C, ^Z, ^Y, ^\) won't work at all, and background
722 * process groups can only be created with "cmd &".
723 * With saved_tty_pgrp != 0, hush will use tcsetpgrp()
724 * to give tty to the foreground process group,
725 * and will take it back when the group is stopped (^Z)
726 * or killed (^C).
727 */
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000728#if ENABLE_HUSH_INTERACTIVE
729 /* 'interactive_fd' is a fd# open to ctty, if we have one
730 * _AND_ if we decided to act interactively */
731 int interactive_fd;
732 const char *PS1;
733 const char *PS2;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000734# define G_interactive_fd (G.interactive_fd)
Denis Vlasenko60b392f2009-04-03 19:14:32 +0000735#else
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000736# define G_interactive_fd 0
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000737#endif
738#if ENABLE_FEATURE_EDITING
739 line_input_t *line_input_state;
740#endif
Denis Vlasenkocc3f20b2008-06-23 22:31:52 +0000741 pid_t root_pid;
Denys Vlasenkodea47882009-10-09 15:40:49 +0200742 pid_t root_ppid;
Denis Vlasenko87a86552008-07-29 19:43:10 +0000743 pid_t last_bg_pid;
Denys Vlasenko20b3d142009-10-09 20:59:39 +0200744#if ENABLE_HUSH_RANDOM_SUPPORT
745 random_t random_gen;
746#endif
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000747#if ENABLE_HUSH_JOB
748 int run_list_level;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000749 int last_jobid;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000750 pid_t saved_tty_pgrp;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000751 struct pipe *job_list;
Mike Frysinger38478a62009-05-20 04:48:06 -0400752# define G_saved_tty_pgrp (G.saved_tty_pgrp)
753#else
754# define G_saved_tty_pgrp 0
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000755#endif
Denys Vlasenko26777aa2010-11-22 23:49:10 +0100756 char o_opt[NUM_OPT_O];
Denys Vlasenko57542eb2010-11-28 03:59:30 +0100757#if ENABLE_HUSH_MODE_X
758# define G_x_mode (G.o_opt[OPT_O_XTRACE])
759#else
760# define G_x_mode 0
761#endif
Denis Vlasenko422cd7c2009-03-31 12:41:52 +0000762 smallint flag_SIGINT;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000763#if ENABLE_HUSH_LOOPS
Denis Vlasenkobcb25532008-07-28 23:04:34 +0000764 smallint flag_break_continue;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000765#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000766#if ENABLE_HUSH_FUNCTIONS
767 /* 0: outside of a function (or sourced file)
768 * -1: inside of a function, ok to use return builtin
Denis Vlasenkoc8653f62009-04-27 23:29:14 +0000769 * 1: return is invoked, skip all till end of func
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000770 */
771 smallint flag_return_in_progress;
772#endif
Denis Vlasenkoefea9d22009-04-09 13:43:11 +0000773 smallint exiting; /* used to prevent EXIT trap recursion */
Denis Vlasenkod5762932009-03-31 11:22:57 +0000774 /* These four support $?, $#, and $1 */
Denis Vlasenkoab2b0642009-04-06 18:42:11 +0000775 smalluint last_exitcode;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000776 /* are global_argv and global_argv[1..n] malloced? (note: not [0]) */
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +0000777 smalluint global_args_malloced;
Denis Vlasenkoe1300f62009-03-22 11:41:18 +0000778 /* how many non-NULL argv's we have. NB: $# + 1 */
779 int global_argc;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000780 char **global_argv;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000781#if !BB_MMU
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +0000782 char *argv0_for_re_execing;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +0000783#endif
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000784#if ENABLE_HUSH_LOOPS
Denis Vlasenko6a2d40f2008-07-28 23:07:06 +0000785 unsigned depth_break_continue;
Denis Vlasenkofcf37c32008-07-29 11:37:15 +0000786 unsigned depth_of_loop;
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000787#endif
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000788 const char *ifs;
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000789 const char *cwd;
Denys Vlasenko52e460b2010-09-16 16:12:00 +0200790 struct variable *top_var;
Denys Vlasenko29082232010-07-16 13:52:32 +0200791 char **expanded_assignments;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000792#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +0000793 struct function *top_func;
Denys Vlasenko295fef82009-06-03 12:47:26 +0200794# if ENABLE_HUSH_LOCAL
795 struct variable **shadowed_vars_pp;
796 unsigned func_nest_level;
797# endif
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +0000798#endif
Denis Vlasenkod5762932009-03-31 11:22:57 +0000799 /* Signal and trap handling */
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200800#if ENABLE_HUSH_FAST
801 unsigned count_SIGCHLD;
802 unsigned handled_SIGCHLD;
Denys Vlasenkoe2df5f42009-05-26 14:34:10 +0200803 smallint we_have_children;
Denys Vlasenko8d7be232009-05-25 16:38:32 +0200804#endif
Denys Vlasenko10c01312011-05-11 11:49:21 +0200805 /* Which signals have non-DFL handler (even with no traps set)?
806 * Set at the start to:
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200807 * (SIGQUIT + maybe SPECIAL_INTERACTIVE_SIGS + maybe SPECIAL_JOBSTOP_SIGS)
Denys Vlasenko10c01312011-05-11 11:49:21 +0200808 * SPECIAL_INTERACTIVE_SIGS are cleared after fork.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200809 * The rest is cleared right before execv syscalls.
Denys Vlasenko10c01312011-05-11 11:49:21 +0200810 * Other than these two times, never modified.
811 */
812 unsigned special_sig_mask;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200813#if ENABLE_HUSH_JOB
814 unsigned fatal_sig_mask;
Denys Vlasenko75e77de2011-05-12 13:12:47 +0200815# define G_fatal_sig_mask G.fatal_sig_mask
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200816#else
Denys Vlasenko75e77de2011-05-12 13:12:47 +0200817# define G_fatal_sig_mask 0
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200818#endif
Denis Vlasenko7566bae2009-03-31 17:24:49 +0000819 char **traps; /* char *traps[NSIG] */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +0200820 sigset_t pending_set;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000821#if HUSH_DEBUG
822 unsigned long memleak_value;
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000823 int debug_indent;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000824#endif
Denys Vlasenko0806e402011-05-12 23:06:20 +0200825 struct sigaction sa;
Denys Vlasenkoaaa22d22009-10-19 16:34:39 +0200826 char user_input_buf[ENABLE_FEATURE_EDITING ? CONFIG_FEATURE_EDITING_MAX_LEN : 2];
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000827};
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000828#define G (*ptr_to_globals)
Denis Vlasenko87a86552008-07-29 19:43:10 +0000829/* Not #defining name to G.name - this quickly gets unwieldy
830 * (too many defines). Also, I actually prefer to see when a variable
831 * is global, thus "G." prefix is a useful hint */
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000832#define INIT_G() do { \
833 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
Denys Vlasenko0806e402011-05-12 23:06:20 +0200834 /* memset(&G.sa, 0, sizeof(G.sa)); */ \
835 sigfillset(&G.sa.sa_mask); \
836 G.sa.sa_flags = SA_RESTART; \
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000837} while (0)
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000838
839
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000840/* Function prototypes for builtins */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200841static int builtin_cd(char **argv) FAST_FUNC;
842static int builtin_echo(char **argv) FAST_FUNC;
843static int builtin_eval(char **argv) FAST_FUNC;
844static int builtin_exec(char **argv) FAST_FUNC;
845static int builtin_exit(char **argv) FAST_FUNC;
846static int builtin_export(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000847#if ENABLE_HUSH_JOB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200848static int builtin_fg_bg(char **argv) FAST_FUNC;
849static int builtin_jobs(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000850#endif
851#if ENABLE_HUSH_HELP
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200852static int builtin_help(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000853#endif
Denys Vlasenkoff463a82013-05-12 02:45:23 +0200854#if MAX_HISTORY && ENABLE_FEATURE_EDITING
Flemming Madsend96ffda2013-04-07 18:47:24 +0200855static int builtin_history(char **argv) FAST_FUNC;
856#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +0200857#if ENABLE_HUSH_LOCAL
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200858static int builtin_local(char **argv) FAST_FUNC;
Denys Vlasenko295fef82009-06-03 12:47:26 +0200859#endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000860#if HUSH_DEBUG
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200861static int builtin_memleak(char **argv) FAST_FUNC;
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000862#endif
Mike Frysinger4ebc76c2009-10-15 03:32:39 -0400863#if ENABLE_PRINTF
864static int builtin_printf(char **argv) FAST_FUNC;
865#endif
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200866static int builtin_pwd(char **argv) FAST_FUNC;
867static int builtin_read(char **argv) FAST_FUNC;
868static int builtin_set(char **argv) FAST_FUNC;
869static int builtin_shift(char **argv) FAST_FUNC;
870static int builtin_source(char **argv) FAST_FUNC;
871static int builtin_test(char **argv) FAST_FUNC;
872static int builtin_trap(char **argv) FAST_FUNC;
873static int builtin_type(char **argv) FAST_FUNC;
874static int builtin_true(char **argv) FAST_FUNC;
875static int builtin_umask(char **argv) FAST_FUNC;
876static int builtin_unset(char **argv) FAST_FUNC;
877static int builtin_wait(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000878#if ENABLE_HUSH_LOOPS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200879static int builtin_break(char **argv) FAST_FUNC;
880static int builtin_continue(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000881#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000882#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200883static int builtin_return(char **argv) FAST_FUNC;
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000884#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000885
886/* Table of built-in functions. They can be forked or not, depending on
887 * context: within pipes, they fork. As simple commands, they do not.
888 * When used in non-forking context, they can change global variables
889 * in the parent shell process. If forked, of course they cannot.
890 * For example, 'unset foo | whatever' will parse and run, but foo will
891 * still be set at the end. */
892struct built_in_command {
Denys Vlasenko17323a62010-01-28 01:57:05 +0100893 const char *b_cmd;
894 int (*b_function)(char **argv) FAST_FUNC;
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000895#if ENABLE_HUSH_HELP
Denys Vlasenko17323a62010-01-28 01:57:05 +0100896 const char *b_descr;
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200897# define BLTIN(cmd, func, help) { cmd, func, help }
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000898#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200899# define BLTIN(cmd, func, help) { cmd, func }
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000900#endif
901};
902
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200903static const struct built_in_command bltins1[] = {
904 BLTIN("." , builtin_source , "Run commands in a file"),
905 BLTIN(":" , builtin_true , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000906#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200907 BLTIN("bg" , builtin_fg_bg , "Resume a job in the background"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000908#endif
909#if ENABLE_HUSH_LOOPS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200910 BLTIN("break" , builtin_break , "Exit from a loop"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000911#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200912 BLTIN("cd" , builtin_cd , "Change directory"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000913#if ENABLE_HUSH_LOOPS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200914 BLTIN("continue" , builtin_continue, "Start new loop iteration"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000915#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200916 BLTIN("eval" , builtin_eval , "Construct and run shell command"),
917 BLTIN("exec" , builtin_exec , "Execute command, don't return to shell"),
918 BLTIN("exit" , builtin_exit , "Exit"),
919 BLTIN("export" , builtin_export , "Set environment variables"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000920#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200921 BLTIN("fg" , builtin_fg_bg , "Bring job into the foreground"),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000922#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000923#if ENABLE_HUSH_HELP
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200924 BLTIN("help" , builtin_help , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000925#endif
Denys Vlasenkoff463a82013-05-12 02:45:23 +0200926#if MAX_HISTORY && ENABLE_FEATURE_EDITING
Flemming Madsend96ffda2013-04-07 18:47:24 +0200927 BLTIN("history" , builtin_history , "Show command history"),
928#endif
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000929#if ENABLE_HUSH_JOB
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200930 BLTIN("jobs" , builtin_jobs , "List jobs"),
Denis Vlasenko34d4d892009-04-04 20:24:37 +0000931#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +0200932#if ENABLE_HUSH_LOCAL
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200933 BLTIN("local" , builtin_local , "Set local variables"),
Denys Vlasenko295fef82009-06-03 12:47:26 +0200934#endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000935#if HUSH_DEBUG
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200936 BLTIN("memleak" , builtin_memleak , NULL),
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +0000937#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200938 BLTIN("read" , builtin_read , "Input into variable"),
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000939#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200940 BLTIN("return" , builtin_return , "Return from a function"),
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +0000941#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200942 BLTIN("set" , builtin_set , "Set/unset positional parameters"),
943 BLTIN("shift" , builtin_shift , "Shift positional parameters"),
Denys Vlasenko82731b42010-05-17 17:49:52 +0200944#if ENABLE_HUSH_BASH_COMPAT
945 BLTIN("source" , builtin_source , "Run commands in a file"),
946#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200947 BLTIN("trap" , builtin_trap , "Trap signals"),
Denys Vlasenko2bba5912014-03-14 12:43:57 +0100948 BLTIN("true" , builtin_true , NULL),
Denys Vlasenko651a2692010-03-23 16:25:17 +0100949 BLTIN("type" , builtin_type , "Show command type"),
Denys Vlasenkof3c742f2010-03-06 20:12:00 +0100950 BLTIN("ulimit" , shell_builtin_ulimit , "Control resource limits"),
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200951 BLTIN("umask" , builtin_umask , "Set file creation mask"),
952 BLTIN("unset" , builtin_unset , "Unset variables"),
953 BLTIN("wait" , builtin_wait , "Wait for process"),
954};
955/* For now, echo and test are unconditionally enabled.
956 * Maybe make it configurable? */
957static const struct built_in_command bltins2[] = {
958 BLTIN("[" , builtin_test , NULL),
959 BLTIN("echo" , builtin_echo , NULL),
Mike Frysinger4ebc76c2009-10-15 03:32:39 -0400960#if ENABLE_PRINTF
961 BLTIN("printf" , builtin_printf , NULL),
962#endif
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +0200963 BLTIN("pwd" , builtin_pwd , NULL),
964 BLTIN("test" , builtin_test , NULL),
Denis Vlasenko424f79b2009-03-22 14:23:34 +0000965};
966
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +0000967
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000968/* Debug printouts.
969 */
970#if HUSH_DEBUG
971/* prevent disasters with G.debug_indent < 0 */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100972# define indent() fdprintf(2, "%*s", (G.debug_indent * 2) & 0xff, "")
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000973# define debug_enter() (G.debug_indent++)
974# define debug_leave() (G.debug_indent--)
975#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +0200976# define indent() ((void)0)
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000977# define debug_enter() ((void)0)
978# define debug_leave() ((void)0)
979#endif
980
981#ifndef debug_printf
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100982# define debug_printf(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000983#endif
984
985#ifndef debug_printf_parse
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100986# define debug_printf_parse(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000987#endif
988
989#ifndef debug_printf_exec
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100990#define debug_printf_exec(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000991#endif
992
993#ifndef debug_printf_env
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100994# define debug_printf_env(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000995#endif
996
997#ifndef debug_printf_jobs
Denys Vlasenko75eb9d22010-12-21 21:18:12 +0100998# define debug_printf_jobs(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +0000999# define DEBUG_JOBS 1
1000#else
1001# define DEBUG_JOBS 0
1002#endif
1003
1004#ifndef debug_printf_expand
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001005# define debug_printf_expand(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001006# define DEBUG_EXPAND 1
1007#else
1008# define DEBUG_EXPAND 0
1009#endif
1010
Denys Vlasenko1e811b12010-05-22 03:12:29 +02001011#ifndef debug_printf_varexp
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001012# define debug_printf_varexp(...) (indent(), fdprintf(2, __VA_ARGS__))
Denys Vlasenko1e811b12010-05-22 03:12:29 +02001013#endif
1014
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001015#ifndef debug_printf_glob
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001016# define debug_printf_glob(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001017# define DEBUG_GLOB 1
1018#else
1019# define DEBUG_GLOB 0
1020#endif
1021
1022#ifndef debug_printf_list
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001023# define debug_printf_list(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001024#endif
1025
1026#ifndef debug_printf_subst
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001027# define debug_printf_subst(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001028#endif
1029
1030#ifndef debug_printf_clean
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001031# define debug_printf_clean(...) (indent(), fdprintf(2, __VA_ARGS__))
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001032# define DEBUG_CLEAN 1
1033#else
1034# define DEBUG_CLEAN 0
1035#endif
1036
1037#if DEBUG_EXPAND
1038static void debug_print_strings(const char *prefix, char **vv)
1039{
1040 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001041 fdprintf(2, "%s:\n", prefix);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001042 while (*vv)
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001043 fdprintf(2, " '%s'\n", *vv++);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001044}
1045#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001046# define debug_print_strings(prefix, vv) ((void)0)
Denis Vlasenko0701dca2009-04-11 10:38:47 +00001047#endif
1048
1049
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001050/* Leak hunting. Use hush_leaktool.sh for post-processing.
1051 */
1052#if LEAK_HUNTING
1053static void *xxmalloc(int lineno, size_t size)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001054{
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001055 void *ptr = xmalloc((size + 0xff) & ~0xff);
1056 fdprintf(2, "line %d: malloc %p\n", lineno, ptr);
1057 return ptr;
1058}
1059static void *xxrealloc(int lineno, void *ptr, size_t size)
1060{
1061 ptr = xrealloc(ptr, (size + 0xff) & ~0xff);
1062 fdprintf(2, "line %d: realloc %p\n", lineno, ptr);
1063 return ptr;
1064}
1065static char *xxstrdup(int lineno, const char *str)
1066{
1067 char *ptr = xstrdup(str);
1068 fdprintf(2, "line %d: strdup %p\n", lineno, ptr);
1069 return ptr;
1070}
1071static void xxfree(void *ptr)
1072{
1073 fdprintf(2, "free %p\n", ptr);
1074 free(ptr);
1075}
Denys Vlasenko8391c482010-05-22 17:50:43 +02001076# define xmalloc(s) xxmalloc(__LINE__, s)
1077# define xrealloc(p, s) xxrealloc(__LINE__, p, s)
1078# define xstrdup(s) xxstrdup(__LINE__, s)
1079# define free(p) xxfree(p)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001080#endif
1081
1082
1083/* Syntax and runtime errors. They always abort scripts.
1084 * In interactive use they usually discard unparsed and/or unexecuted commands
1085 * and return to the prompt.
1086 * HUSH_DEBUG >= 2 prints line number in this file where it was detected.
1087 */
1088#if HUSH_DEBUG < 2
Denys Vlasenko606291b2009-09-23 23:15:43 +02001089# define die_if_script(lineno, ...) die_if_script(__VA_ARGS__)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001090# define syntax_error(lineno, msg) syntax_error(msg)
1091# define syntax_error_at(lineno, msg) syntax_error_at(msg)
1092# define syntax_error_unterm_ch(lineno, ch) syntax_error_unterm_ch(ch)
1093# define syntax_error_unterm_str(lineno, s) syntax_error_unterm_str(s)
1094# define syntax_error_unexpected_ch(lineno, ch) syntax_error_unexpected_ch(ch)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001095#endif
1096
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001097static void die_if_script(unsigned lineno, const char *fmt, ...)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001098{
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001099 va_list p;
1100
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001101#if HUSH_DEBUG >= 2
1102 bb_error_msg("hush.c:%u", lineno);
1103#endif
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001104 va_start(p, fmt);
1105 bb_verror_msg(fmt, p, NULL);
1106 va_end(p);
1107 if (!G_interactive_fd)
1108 xfunc_die();
Mike Frysinger6379bb42009-03-28 18:55:03 +00001109}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001110
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001111static void syntax_error(unsigned lineno UNUSED_PARAM, const char *msg)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001112{
1113 if (msg)
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001114 bb_error_msg("syntax error: %s", msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001115 else
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001116 bb_error_msg("syntax error");
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001117}
1118
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001119static void syntax_error_at(unsigned lineno UNUSED_PARAM, const char *msg)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001120{
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001121 bb_error_msg("syntax error at '%s'", msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001122}
1123
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001124static void syntax_error_unterm_str(unsigned lineno UNUSED_PARAM, const char *s)
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001125{
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001126 bb_error_msg("syntax error: unterminated %s", s);
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001127}
1128
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001129static void syntax_error_unterm_ch(unsigned lineno, char ch)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001130{
Mike Frysinger6a46ab82009-06-01 14:14:36 -04001131 char msg[2] = { ch, '\0' };
1132 syntax_error_unterm_str(lineno, msg);
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001133}
1134
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001135static void syntax_error_unexpected_ch(unsigned lineno UNUSED_PARAM, int ch)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001136{
1137 char msg[2];
1138 msg[0] = ch;
1139 msg[1] = '\0';
Denys Vlasenkocecbc982011-03-30 18:54:52 +02001140 bb_error_msg("syntax error: unexpected %s", ch == EOF ? "EOF" : msg);
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001141}
1142
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001143#if HUSH_DEBUG < 2
1144# undef die_if_script
1145# undef syntax_error
1146# undef syntax_error_at
Denis Vlasenkod68ae082009-04-09 20:41:34 +00001147# undef syntax_error_unterm_ch
1148# undef syntax_error_unterm_str
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001149# undef syntax_error_unexpected_ch
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001150#else
Denys Vlasenko606291b2009-09-23 23:15:43 +02001151# define die_if_script(...) die_if_script(__LINE__, __VA_ARGS__)
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00001152# define syntax_error(msg) syntax_error(__LINE__, msg)
1153# define syntax_error_at(msg) syntax_error_at(__LINE__, msg)
1154# define syntax_error_unterm_ch(ch) syntax_error_unterm_ch(__LINE__, ch)
1155# define syntax_error_unterm_str(s) syntax_error_unterm_str(__LINE__, s)
1156# define syntax_error_unexpected_ch(ch) syntax_error_unexpected_ch(__LINE__, ch)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00001157#endif
Eric Andersen25f27032001-04-26 23:22:31 +00001158
Denis Vlasenko552433b2009-04-04 19:29:21 +00001159
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001160#if ENABLE_HUSH_INTERACTIVE
1161static void cmdedit_update_prompt(void);
1162#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001163# define cmdedit_update_prompt() ((void)0)
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001164#endif
1165
1166
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001167/* Utility functions
1168 */
Denis Vlasenko55789c62008-06-18 16:30:42 +00001169/* Replace each \x with x in place, return ptr past NUL. */
1170static char *unbackslash(char *src)
1171{
Denys Vlasenko71885402009-09-24 01:44:13 +02001172 char *dst = src = strchrnul(src, '\\');
Denis Vlasenko55789c62008-06-18 16:30:42 +00001173 while (1) {
1174 if (*src == '\\')
1175 src++;
1176 if ((*dst++ = *src++) == '\0')
1177 break;
1178 }
1179 return dst;
1180}
1181
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001182static char **add_strings_to_strings(char **strings, char **add, int need_to_dup)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001183{
1184 int i;
1185 unsigned count1;
1186 unsigned count2;
1187 char **v;
1188
1189 v = strings;
1190 count1 = 0;
1191 if (v) {
1192 while (*v) {
1193 count1++;
1194 v++;
1195 }
1196 }
1197 count2 = 0;
1198 v = add;
1199 while (*v) {
1200 count2++;
1201 v++;
1202 }
1203 v = xrealloc(strings, (count1 + count2 + 1) * sizeof(char*));
1204 v[count1 + count2] = NULL;
1205 i = count2;
1206 while (--i >= 0)
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001207 v[count1 + i] = (need_to_dup ? xstrdup(add[i]) : add[i]);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001208 return v;
1209}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001210#if LEAK_HUNTING
Denis Vlasenkocc90f442009-04-08 16:40:34 +00001211static char **xx_add_strings_to_strings(int lineno, char **strings, char **add, int need_to_dup)
1212{
1213 char **ptr = add_strings_to_strings(strings, add, need_to_dup);
1214 fdprintf(2, "line %d: add_strings_to_strings %p\n", lineno, ptr);
1215 return ptr;
1216}
1217#define add_strings_to_strings(strings, add, need_to_dup) \
1218 xx_add_strings_to_strings(__LINE__, strings, add, need_to_dup)
1219#endif
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001220
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001221/* Note: takes ownership of "add" ptr (it is not strdup'ed) */
Denis Vlasenko22d10a02008-10-13 08:53:43 +00001222static char **add_string_to_strings(char **strings, char *add)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001223{
1224 char *v[2];
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001225 v[0] = add;
1226 v[1] = NULL;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00001227 return add_strings_to_strings(strings, v, /*dup:*/ 0);
Denis Vlasenko22d10a02008-10-13 08:53:43 +00001228}
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00001229#if LEAK_HUNTING
Denis Vlasenkocc90f442009-04-08 16:40:34 +00001230static char **xx_add_string_to_strings(int lineno, char **strings, char *add)
1231{
1232 char **ptr = add_string_to_strings(strings, add);
1233 fdprintf(2, "line %d: add_string_to_strings %p\n", lineno, ptr);
1234 return ptr;
1235}
1236#define add_string_to_strings(strings, add) \
1237 xx_add_string_to_strings(__LINE__, strings, add)
1238#endif
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001239
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001240static void free_strings(char **strings)
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001241{
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001242 char **v;
1243
1244 if (!strings)
1245 return;
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001246 v = strings;
1247 while (*v) {
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001248 free(*v);
1249 v++;
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001250 }
Denis Vlasenkoafd7a8d2008-10-09 16:29:44 +00001251 free(strings);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00001252}
1253
Denis Vlasenko76d50412008-06-10 16:19:39 +00001254
Denis Vlasenko270b1c32009-04-17 18:54:50 +00001255/* Helpers for setting new $n and restoring them back
1256 */
1257typedef struct save_arg_t {
1258 char *sv_argv0;
1259 char **sv_g_argv;
1260 int sv_g_argc;
1261 smallint sv_g_malloced;
1262} save_arg_t;
1263
1264static void save_and_replace_G_args(save_arg_t *sv, char **argv)
1265{
1266 int n;
1267
1268 sv->sv_argv0 = argv[0];
1269 sv->sv_g_argv = G.global_argv;
1270 sv->sv_g_argc = G.global_argc;
1271 sv->sv_g_malloced = G.global_args_malloced;
1272
1273 argv[0] = G.global_argv[0]; /* retain $0 */
1274 G.global_argv = argv;
1275 G.global_args_malloced = 0;
1276
1277 n = 1;
1278 while (*++argv)
1279 n++;
1280 G.global_argc = n;
1281}
1282
1283static void restore_G_args(save_arg_t *sv, char **argv)
1284{
1285 char **pp;
1286
1287 if (G.global_args_malloced) {
1288 /* someone ran "set -- arg1 arg2 ...", undo */
1289 pp = G.global_argv;
1290 while (*++pp) /* note: does not free $0 */
1291 free(*pp);
1292 free(G.global_argv);
1293 }
1294 argv[0] = sv->sv_argv0;
1295 G.global_argv = sv->sv_g_argv;
1296 G.global_argc = sv->sv_g_argc;
1297 G.global_args_malloced = sv->sv_g_malloced;
1298}
1299
1300
Denis Vlasenkod5762932009-03-31 11:22:57 +00001301/* Basic theory of signal handling in shell
1302 * ========================================
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001303 * This does not describe what hush does, rather, it is current understanding
1304 * what it _should_ do. If it doesn't, it's a bug.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001305 * http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#trap
1306 *
1307 * Signals are handled only after each pipe ("cmd | cmd | cmd" thing)
1308 * is finished or backgrounded. It is the same in interactive and
1309 * non-interactive shells, and is the same regardless of whether
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001310 * a user trap handler is installed or a shell special one is in effect.
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02001311 * ^C or ^Z from keyboard seems to execute "at once" because it usually
Denis Vlasenkod5762932009-03-31 11:22:57 +00001312 * backgrounds (i.e. stops) or kills all members of currently running
1313 * pipe.
1314 *
Denys Vlasenko8bd810b2013-11-28 01:50:01 +01001315 * Wait builtin is interruptible by signals for which user trap is set
Denis Vlasenkod5762932009-03-31 11:22:57 +00001316 * or by SIGINT in interactive shell.
1317 *
1318 * Trap handlers will execute even within trap handlers. (right?)
1319 *
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001320 * User trap handlers are forgotten when subshell ("(cmd)") is entered,
1321 * except for handlers set to '' (empty string).
Denis Vlasenkod5762932009-03-31 11:22:57 +00001322 *
1323 * If job control is off, backgrounded commands ("cmd &")
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001324 * have SIGINT, SIGQUIT set to SIG_IGN.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001325 *
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001326 * Commands which are run in command substitution ("`cmd`")
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001327 * have SIGTTIN, SIGTTOU, SIGTSTP set to SIG_IGN.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001328 *
Denys Vlasenko4b7db4f2009-05-29 10:39:06 +02001329 * Ordinary commands have signals set to SIG_IGN/DFL as inherited
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001330 * by the shell from its parent.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001331 *
Denys Vlasenko28a105d2009-06-01 11:26:30 +02001332 * Signals which differ from SIG_DFL action
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001333 * (note: child (i.e., [v]forked) shell is not an interactive shell):
Denis Vlasenkod5762932009-03-31 11:22:57 +00001334 *
1335 * SIGQUIT: ignore
1336 * SIGTERM (interactive): ignore
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001337 * SIGHUP (interactive):
1338 * send SIGCONT to stopped jobs, send SIGHUP to all jobs and exit
Denis Vlasenkod5762932009-03-31 11:22:57 +00001339 * SIGTTIN, SIGTTOU, SIGTSTP (if job control is on): ignore
Denis Vlasenkoc4ada792009-04-15 23:29:00 +00001340 * Note that ^Z is handled not by trapping SIGTSTP, but by seeing
1341 * that all pipe members are stopped. Try this in bash:
1342 * while :; do :; done - ^Z does not background it
1343 * (while :; do :; done) - ^Z backgrounds it
Denis Vlasenkod5762932009-03-31 11:22:57 +00001344 * SIGINT (interactive): wait for last pipe, ignore the rest
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001345 * of the command line, show prompt. NB: ^C does not send SIGINT
1346 * to interactive shell while shell is waiting for a pipe,
1347 * since shell is bg'ed (is not in foreground process group).
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001348 * Example 1: this waits 5 sec, but does not execute ls:
1349 * "echo $$; sleep 5; ls -l" + "kill -INT <pid>"
1350 * Example 2: this does not wait and does not execute ls:
1351 * "echo $$; sleep 5 & wait; ls -l" + "kill -INT <pid>"
1352 * Example 3: this does not wait 5 sec, but executes ls:
1353 * "sleep 5; ls -l" + press ^C
Denys Vlasenkob8709032011-05-08 21:20:01 +02001354 * Example 4: this does not wait and does not execute ls:
1355 * "sleep 5 & wait; ls -l" + press ^C
Denis Vlasenkod5762932009-03-31 11:22:57 +00001356 *
1357 * (What happens to signals which are IGN on shell start?)
1358 * (What happens with signal mask on shell start?)
1359 *
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001360 * Old implementation
1361 * ==================
Denis Vlasenkod5762932009-03-31 11:22:57 +00001362 * We use in-kernel pending signal mask to determine which signals were sent.
1363 * We block all signals which we don't want to take action immediately,
1364 * i.e. we block all signals which need to have special handling as described
1365 * above, and all signals which have traps set.
1366 * After each pipe execution, we extract any pending signals via sigtimedwait()
1367 * and act on them.
1368 *
Denys Vlasenko10c01312011-05-11 11:49:21 +02001369 * unsigned special_sig_mask: a mask of such "special" signals
Denis Vlasenkod5762932009-03-31 11:22:57 +00001370 * sigset_t blocked_set: current blocked signal set
1371 *
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001372 * "trap - SIGxxx":
Denys Vlasenko10c01312011-05-11 11:49:21 +02001373 * clear bit in blocked_set unless it is also in special_sig_mask
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001374 * "trap 'cmd' SIGxxx":
1375 * set bit in blocked_set (even if 'cmd' is '')
Denis Vlasenkod5762932009-03-31 11:22:57 +00001376 * after [v]fork, if we plan to be a shell:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001377 * unblock signals with special interactive handling
1378 * (child shell is not interactive),
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001379 * unset all traps except '' (note: regardless of child shell's type - {}, (), etc)
Denis Vlasenkod5762932009-03-31 11:22:57 +00001380 * after [v]fork, if we plan to exec:
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02001381 * POSIX says fork clears pending signal mask in child - no need to clear it.
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001382 * Restore blocked signal set to one inherited by shell just prior to exec.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001383 *
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001384 * Note: as a result, we do not use signal handlers much. The only uses
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001385 * are to count SIGCHLDs
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001386 * and to restore tty pgrp on signal-induced exit.
Denys Vlasenko4ea0ca82009-09-25 12:58:37 +02001387 *
Denys Vlasenko67f71862009-09-25 14:21:06 +02001388 * Note 2 (compat):
Denys Vlasenko4ea0ca82009-09-25 12:58:37 +02001389 * Standard says "When a subshell is entered, traps that are not being ignored
1390 * are set to the default actions". bash interprets it so that traps which
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01001391 * are set to '' (ignore) are NOT reset to defaults. We do the same.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001392 *
1393 * Problem: the above approach makes it unwieldy to catch signals while
Denys Vlasenkoe95738f2013-07-08 03:13:08 +02001394 * we are in read builtin, or while we read commands from stdin:
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001395 * masked signals are not visible!
1396 *
1397 * New implementation
1398 * ==================
1399 * We record each signal we are interested in by installing signal handler
1400 * for them - a bit like emulating kernel pending signal mask in userspace.
1401 * We are interested in: signals which need to have special handling
1402 * as described above, and all signals which have traps set.
Denys Vlasenko8bd810b2013-11-28 01:50:01 +01001403 * Signals are recorded in pending_set.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001404 * After each pipe execution, we extract any pending signals
1405 * and act on them.
1406 *
1407 * unsigned special_sig_mask: a mask of shell-special signals.
1408 * unsigned fatal_sig_mask: a mask of signals on which we restore tty pgrp.
1409 * char *traps[sig] if trap for sig is set (even if it's '').
1410 * sigset_t pending_set: set of sigs we received.
1411 *
1412 * "trap - SIGxxx":
1413 * if sig is in special_sig_mask, set handler back to:
1414 * record_pending_signo, or to IGN if it's a tty stop signal
1415 * if sig is in fatal_sig_mask, set handler back to sigexit.
1416 * else: set handler back to SIG_DFL
1417 * "trap 'cmd' SIGxxx":
1418 * set handler to record_pending_signo.
1419 * "trap '' SIGxxx":
1420 * set handler to SIG_IGN.
1421 * after [v]fork, if we plan to be a shell:
1422 * set signals with special interactive handling to SIG_DFL
1423 * (because child shell is not interactive),
1424 * unset all traps except '' (note: regardless of child shell's type - {}, (), etc)
1425 * after [v]fork, if we plan to exec:
1426 * POSIX says fork clears pending signal mask in child - no need to clear it.
1427 *
1428 * To make wait builtin interruptible, we handle SIGCHLD as special signal,
1429 * otherwise (if we leave it SIG_DFL) sigsuspend in wait builtin will not wake up on it.
1430 *
1431 * Note (compat):
1432 * Standard says "When a subshell is entered, traps that are not being ignored
1433 * are set to the default actions". bash interprets it so that traps which
1434 * are set to '' (ignore) are NOT reset to defaults. We do the same.
Denis Vlasenkod5762932009-03-31 11:22:57 +00001435 */
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001436enum {
1437 SPECIAL_INTERACTIVE_SIGS = 0
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001438 | (1 << SIGTERM)
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001439 | (1 << SIGINT)
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00001440 | (1 << SIGHUP)
1441 ,
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001442 SPECIAL_JOBSTOP_SIGS = 0
Mike Frysinger38478a62009-05-20 04:48:06 -04001443#if ENABLE_HUSH_JOB
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00001444 | (1 << SIGTTIN)
1445 | (1 << SIGTTOU)
1446 | (1 << SIGTSTP)
1447#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001448 ,
Denis Vlasenkoe4bd4f22009-04-17 13:52:51 +00001449};
Denis Vlasenkod5762932009-03-31 11:22:57 +00001450
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001451static void record_pending_signo(int sig)
Denys Vlasenko54e9e122011-05-09 00:52:15 +02001452{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001453 sigaddset(&G.pending_set, sig);
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001454#if ENABLE_HUSH_FAST
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001455 if (sig == SIGCHLD) {
1456 G.count_SIGCHLD++;
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001457//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 +02001458 }
Denys Vlasenko8d7be232009-05-25 16:38:32 +02001459#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001460}
Denis Vlasenko7566bae2009-03-31 17:24:49 +00001461
Denys Vlasenko0806e402011-05-12 23:06:20 +02001462static sighandler_t install_sighandler(int sig, sighandler_t handler)
1463{
1464 struct sigaction old_sa;
1465
1466 /* We could use signal() to install handlers... almost:
1467 * except that we need to mask ALL signals while handlers run.
1468 * I saw signal nesting in strace, race window isn't small.
1469 * SA_RESTART is also needed, but in Linux, signal()
1470 * sets SA_RESTART too.
1471 */
1472 /* memset(&G.sa, 0, sizeof(G.sa)); - already done */
1473 /* sigfillset(&G.sa.sa_mask); - already done */
1474 /* G.sa.sa_flags = SA_RESTART; - already done */
1475 G.sa.sa_handler = handler;
1476 sigaction(sig, &G.sa, &old_sa);
1477 return old_sa.sa_handler;
1478}
1479
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00001480#if ENABLE_HUSH_JOB
Denis Vlasenko25af86f2009-04-07 13:29:27 +00001481
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02001482static void xfunc_has_died(void);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001483/* After [v]fork, in child: do not restore tty pgrp on xfunc death */
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02001484# define disable_restore_tty_pgrp_on_exit() (die_func = NULL)
Denis Vlasenko25af86f2009-04-07 13:29:27 +00001485/* After [v]fork, in parent: restore tty pgrp on xfunc death */
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02001486# define enable_restore_tty_pgrp_on_exit() (die_func = xfunc_has_died)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001487
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001488/* Restores tty foreground process group, and exits.
1489 * May be called as signal handler for fatal signal
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001490 * (will resend signal to itself, producing correct exit state)
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001491 * or called directly with -EXITCODE.
1492 * We also call it if xfunc is exiting. */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +00001493static void sigexit(int sig) NORETURN;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001494static void sigexit(int sig)
1495{
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001496 /* Careful: we can end up here after [v]fork. Do not restore
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001497 * tty pgrp then, only top-level shell process does that */
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001498 if (G_saved_tty_pgrp && getpid() == G.root_pid) {
1499 /* Disable all signals: job control, SIGPIPE, etc.
1500 * Mostly paranoid measure, to prevent infinite SIGTTOU.
1501 */
1502 sigprocmask_allsigs(SIG_BLOCK);
Mike Frysinger38478a62009-05-20 04:48:06 -04001503 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001504 }
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001505
1506 /* Not a signal, just exit */
1507 if (sig <= 0)
1508 _exit(- sig);
1509
Denis Vlasenko400d8bb2008-02-24 13:36:01 +00001510 kill_myself_with_sig(sig); /* does not return */
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001511}
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001512#else
1513
Denys Vlasenko8391c482010-05-22 17:50:43 +02001514# define disable_restore_tty_pgrp_on_exit() ((void)0)
1515# define enable_restore_tty_pgrp_on_exit() ((void)0)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001516
Denis Vlasenkoe0755e52009-04-03 21:16:45 +00001517#endif
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001518
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001519static sighandler_t pick_sighandler(unsigned sig)
1520{
1521 sighandler_t handler = SIG_DFL;
1522 if (sig < sizeof(unsigned)*8) {
1523 unsigned sigmask = (1 << sig);
1524
1525#if ENABLE_HUSH_JOB
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001526 /* is sig fatal? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001527 if (G_fatal_sig_mask & sigmask)
1528 handler = sigexit;
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001529 else
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001530#endif
1531 /* sig has special handling? */
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001532 if (G.special_sig_mask & sigmask) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001533 handler = record_pending_signo;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001534 /* TTIN/TTOU/TSTP can't be set to record_pending_signo
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001535 * in order to ignore them: they will be raised
Denys Vlasenkof58f7052011-05-12 02:10:33 +02001536 * in an endless loop when we try to do some
1537 * terminal ioctls! We do have to _ignore_ these.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001538 */
1539 if (SPECIAL_JOBSTOP_SIGS & sigmask)
1540 handler = SIG_IGN;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001541 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001542 }
1543 return handler;
1544}
1545
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001546/* Restores tty foreground process group, and exits. */
1547static void hush_exit(int exitcode) NORETURN;
1548static void hush_exit(int exitcode)
1549{
Denys Vlasenkobede2152011-09-04 16:12:33 +02001550#if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
1551 save_history(G.line_input_state);
1552#endif
1553
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01001554 fflush_all();
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00001555 if (G.exiting <= 0 && G.traps && G.traps[0] && G.traps[0][0]) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001556 char *argv[3];
1557 /* argv[0] is unused */
1558 argv[1] = G.traps[0];
1559 argv[2] = NULL;
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001560 G.exiting = 1; /* prevent EXIT trap recursion */
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001561 /* Note: G.traps[0] is not cleared!
Denys Vlasenkode8c3f62010-09-12 16:13:44 +02001562 * "trap" will still show it, if executed
1563 * in the handler */
1564 builtin_eval(argv);
Denis Vlasenkod5762932009-03-31 11:22:57 +00001565 }
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001566
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001567#if ENABLE_FEATURE_CLEAN_UP
1568 {
1569 struct variable *cur_var;
1570 if (G.cwd != bb_msg_unknown)
1571 free((char*)G.cwd);
1572 cur_var = G.top_var;
1573 while (cur_var) {
1574 struct variable *tmp = cur_var;
1575 if (!cur_var->max_len)
1576 free(cur_var->varstr);
1577 cur_var = cur_var->next;
1578 free(tmp);
1579 }
1580 }
1581#endif
1582
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001583#if ENABLE_HUSH_JOB
Denys Vlasenko8131eea2009-11-02 14:19:51 +01001584 fflush_all();
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001585 sigexit(- (exitcode & 0xff));
1586#else
1587 exit(exitcode);
1588#endif
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001589}
1590
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02001591static void xfunc_has_died(void) NORETURN;
1592static void xfunc_has_died(void)
1593{
1594 /* xfunc has failed! die die die */
1595 /* no EXIT traps, this is an escape hatch! */
1596 G.exiting = 1;
1597 hush_exit(xfunc_error_retval);
1598}
1599
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02001600
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001601//TODO: return a mask of ALL handled sigs?
1602static int check_and_run_traps(void)
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001603{
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001604 int last_sig = 0;
1605
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001606 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001607 int sig;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02001608
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001609 if (sigisemptyset(&G.pending_set))
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001610 break;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001611 sig = 0;
1612 do {
1613 sig++;
1614 if (sigismember(&G.pending_set, sig)) {
1615 sigdelset(&G.pending_set, sig);
1616 goto got_sig;
1617 }
1618 } while (sig < NSIG);
1619 break;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001620 got_sig:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001621 if (G.traps && G.traps[sig]) {
1622 if (G.traps[sig][0]) {
1623 /* We have user-defined handler */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001624 smalluint save_rcode;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001625 char *argv[3];
1626 /* argv[0] is unused */
1627 argv[1] = G.traps[sig];
1628 argv[2] = NULL;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001629 save_rcode = G.last_exitcode;
1630 builtin_eval(argv);
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001631 G.last_exitcode = save_rcode;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001632 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001633 } /* else: "" trap, ignoring signal */
1634 continue;
1635 }
1636 /* not a trap: special action */
1637 switch (sig) {
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001638 case SIGINT:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001639 /* Builtin was ^C'ed, make it look prettier: */
1640 bb_putchar('\n');
1641 G.flag_SIGINT = 1;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001642 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001643 break;
1644#if ENABLE_HUSH_JOB
1645 case SIGHUP: {
1646 struct pipe *job;
1647 /* bash is observed to signal whole process groups,
1648 * not individual processes */
1649 for (job = G.job_list; job; job = job->next) {
1650 if (job->pgrp <= 0)
1651 continue;
1652 debug_printf_exec("HUPing pgrp %d\n", job->pgrp);
1653 if (kill(- job->pgrp, SIGHUP) == 0)
1654 kill(- job->pgrp, SIGCONT);
1655 }
1656 sigexit(SIGHUP);
1657 }
1658#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001659#if ENABLE_HUSH_FAST
1660 case SIGCHLD:
1661 G.count_SIGCHLD++;
1662//bb_error_msg("[%d] check_and_run_traps: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
1663 /* Note:
1664 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1665 * This simplifies wait builtin a bit.
1666 */
1667 break;
1668#endif
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001669 default: /* ignored: */
1670 /* SIGTERM, SIGQUIT, SIGTTIN, SIGTTOU, SIGTSTP */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001671 /* Note:
1672 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1673 * Example: wait is not interrupted by TERM
Denys Vlasenkob8709032011-05-08 21:20:01 +02001674 * in interactive shell, because TERM is ignored.
1675 */
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001676 break;
1677 }
1678 }
1679 return last_sig;
1680}
1681
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001682
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001683static const char *get_cwd(int force)
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001684{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001685 if (force || G.cwd == NULL) {
1686 /* xrealloc_getcwd_or_warn(arg) calls free(arg),
1687 * we must not try to free(bb_msg_unknown) */
1688 if (G.cwd == bb_msg_unknown)
1689 G.cwd = NULL;
1690 G.cwd = xrealloc_getcwd_or_warn((char *)G.cwd);
1691 if (!G.cwd)
1692 G.cwd = bb_msg_unknown;
1693 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00001694 return G.cwd;
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001695}
1696
Denis Vlasenko83506862007-11-23 13:11:42 +00001697
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001698/*
1699 * Shell and environment variable support
1700 */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001701static struct variable **get_ptr_to_local_var(const char *name, unsigned len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001702{
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001703 struct variable **pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001704 struct variable *cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001705
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001706 pp = &G.top_var;
1707 while ((cur = *pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001708 if (strncmp(cur->varstr, name, len) == 0 && cur->varstr[len] == '=')
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001709 return pp;
1710 pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001711 }
1712 return NULL;
1713}
1714
Denys Vlasenko03dad222010-01-12 23:29:57 +01001715static const char* FAST_FUNC get_local_var_value(const char *name)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001716{
Denys Vlasenko29082232010-07-16 13:52:32 +02001717 struct variable **vpp;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001718 unsigned len = strlen(name);
Denys Vlasenko29082232010-07-16 13:52:32 +02001719
1720 if (G.expanded_assignments) {
1721 char **cpp = G.expanded_assignments;
Denys Vlasenko29082232010-07-16 13:52:32 +02001722 while (*cpp) {
1723 char *cp = *cpp;
1724 if (strncmp(cp, name, len) == 0 && cp[len] == '=')
1725 return cp + len + 1;
1726 cpp++;
1727 }
1728 }
1729
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001730 vpp = get_ptr_to_local_var(name, len);
Denys Vlasenko29082232010-07-16 13:52:32 +02001731 if (vpp)
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001732 return (*vpp)->varstr + len + 1;
Denys Vlasenko29082232010-07-16 13:52:32 +02001733
Denys Vlasenkodea47882009-10-09 15:40:49 +02001734 if (strcmp(name, "PPID") == 0)
1735 return utoa(G.root_ppid);
1736 // bash compat: UID? EUID?
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001737#if ENABLE_HUSH_RANDOM_SUPPORT
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001738 if (strcmp(name, "RANDOM") == 0)
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001739 return utoa(next_random(&G.random_gen));
1740#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001741 return NULL;
1742}
1743
1744/* str holds "NAME=VAL" and is expected to be malloced.
Mike Frysinger6379bb42009-03-28 18:55:03 +00001745 * We take ownership of it.
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001746 * flg_export:
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001747 * 0: do not change export flag
1748 * (if creating new variable, flag will be 0)
1749 * 1: set export flag and putenv the variable
1750 * -1: clear export flag and unsetenv the variable
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001751 * flg_read_only is set only when we handle -R var=val
Mike Frysinger6379bb42009-03-28 18:55:03 +00001752 */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001753#if !BB_MMU && ENABLE_HUSH_LOCAL
1754/* all params are used */
1755#elif BB_MMU && ENABLE_HUSH_LOCAL
1756#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1757 set_local_var(str, flg_export, local_lvl)
1758#elif BB_MMU && !ENABLE_HUSH_LOCAL
1759#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001760 set_local_var(str, flg_export)
Denys Vlasenko295fef82009-06-03 12:47:26 +02001761#elif !BB_MMU && !ENABLE_HUSH_LOCAL
1762#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1763 set_local_var(str, flg_export, flg_read_only)
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001764#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +02001765static int set_local_var(char *str, int flg_export, int local_lvl, int flg_read_only)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001766{
Denys Vlasenko295fef82009-06-03 12:47:26 +02001767 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001768 struct variable *cur;
Denis Vlasenko950bd722009-04-21 11:23:56 +00001769 char *eq_sign;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001770 int name_len;
1771
Denis Vlasenko950bd722009-04-21 11:23:56 +00001772 eq_sign = strchr(str, '=');
1773 if (!eq_sign) { /* not expected to ever happen? */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001774 free(str);
1775 return -1;
1776 }
1777
Denis Vlasenko950bd722009-04-21 11:23:56 +00001778 name_len = eq_sign - str + 1; /* including '=' */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001779 var_pp = &G.top_var;
1780 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001781 if (strncmp(cur->varstr, str, name_len) != 0) {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001782 var_pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001783 continue;
1784 }
1785 /* We found an existing var with this name */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001786 if (cur->flg_read_only) {
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001787#if !BB_MMU
1788 if (!flg_read_only)
1789#endif
1790 bb_error_msg("%s: readonly variable", str);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001791 free(str);
1792 return -1;
1793 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001794 if (flg_export == -1) { // "&& cur->flg_export" ?
Denis Vlasenko950bd722009-04-21 11:23:56 +00001795 debug_printf_env("%s: unsetenv '%s'\n", __func__, str);
1796 *eq_sign = '\0';
1797 unsetenv(str);
1798 *eq_sign = '=';
1799 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001800#if ENABLE_HUSH_LOCAL
1801 if (cur->func_nest_level < local_lvl) {
1802 /* New variable is declared as local,
1803 * and existing one is global, or local
1804 * from enclosing function.
1805 * Remove and save old one: */
1806 *var_pp = cur->next;
1807 cur->next = *G.shadowed_vars_pp;
1808 *G.shadowed_vars_pp = cur;
1809 /* bash 3.2.33(1) and exported vars:
1810 * # export z=z
1811 * # f() { local z=a; env | grep ^z; }
1812 * # f
1813 * z=a
1814 * # env | grep ^z
1815 * z=z
1816 */
1817 if (cur->flg_export)
1818 flg_export = 1;
1819 break;
1820 }
1821#endif
Denis Vlasenko950bd722009-04-21 11:23:56 +00001822 if (strcmp(cur->varstr + name_len, eq_sign + 1) == 0) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001823 free_and_exp:
1824 free(str);
1825 goto exp;
1826 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001827 if (cur->max_len != 0) {
1828 if (cur->max_len >= strlen(str)) {
1829 /* This one is from startup env, reuse space */
1830 strcpy(cur->varstr, str);
1831 goto free_and_exp;
1832 }
1833 } else {
1834 /* max_len == 0 signifies "malloced" var, which we can
1835 * (and has to) free */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001836 free(cur->varstr);
Denys Vlasenko295fef82009-06-03 12:47:26 +02001837 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001838 cur->max_len = 0;
1839 goto set_str_and_exp;
1840 }
1841
Denys Vlasenko295fef82009-06-03 12:47:26 +02001842 /* Not found - create new variable struct */
1843 cur = xzalloc(sizeof(*cur));
1844#if ENABLE_HUSH_LOCAL
1845 cur->func_nest_level = local_lvl;
1846#endif
1847 cur->next = *var_pp;
1848 *var_pp = cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001849
1850 set_str_and_exp:
1851 cur->varstr = str;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001852#if !BB_MMU
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001853 cur->flg_read_only = flg_read_only;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001854#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001855 exp:
Mike Frysinger6379bb42009-03-28 18:55:03 +00001856 if (flg_export == 1)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001857 cur->flg_export = 1;
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001858 if (name_len == 4 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1859 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001860 if (cur->flg_export) {
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001861 if (flg_export == -1) {
1862 cur->flg_export = 0;
1863 /* unsetenv was already done */
1864 } else {
1865 debug_printf_env("%s: putenv '%s'\n", __func__, cur->varstr);
1866 return putenv(cur->varstr);
1867 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001868 }
1869 return 0;
1870}
1871
Denys Vlasenko6db47842009-09-05 20:15:17 +02001872/* Used at startup and after each cd */
1873static void set_pwd_var(int exp)
1874{
1875 set_local_var(xasprintf("PWD=%s", get_cwd(/*force:*/ 1)),
1876 /*exp:*/ exp, /*lvl:*/ 0, /*ro:*/ 0);
1877}
1878
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001879static int unset_local_var_len(const char *name, int name_len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001880{
1881 struct variable *cur;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001882 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001883
1884 if (!name)
Mike Frysingerd690f682009-03-30 06:50:54 +00001885 return EXIT_SUCCESS;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001886 var_pp = &G.top_var;
1887 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001888 if (strncmp(cur->varstr, name, name_len) == 0 && cur->varstr[name_len] == '=') {
1889 if (cur->flg_read_only) {
1890 bb_error_msg("%s: readonly variable", name);
Mike Frysingerd690f682009-03-30 06:50:54 +00001891 return EXIT_FAILURE;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001892 }
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001893 *var_pp = cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001894 debug_printf_env("%s: unsetenv '%s'\n", __func__, cur->varstr);
1895 bb_unsetenv(cur->varstr);
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001896 if (name_len == 3 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1897 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001898 if (!cur->max_len)
1899 free(cur->varstr);
1900 free(cur);
Mike Frysingerd690f682009-03-30 06:50:54 +00001901 return EXIT_SUCCESS;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001902 }
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001903 var_pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001904 }
Mike Frysingerd690f682009-03-30 06:50:54 +00001905 return EXIT_SUCCESS;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001906}
1907
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001908static int unset_local_var(const char *name)
1909{
1910 return unset_local_var_len(name, strlen(name));
1911}
1912
1913static void unset_vars(char **strings)
1914{
1915 char **v;
1916
1917 if (!strings)
1918 return;
1919 v = strings;
1920 while (*v) {
1921 const char *eq = strchrnul(*v, '=');
1922 unset_local_var_len(*v, (int)(eq - *v));
1923 v++;
1924 }
1925 free(strings);
1926}
1927
Denys Vlasenko03dad222010-01-12 23:29:57 +01001928static void FAST_FUNC set_local_var_from_halves(const char *name, const char *val)
Mike Frysinger98c52642009-04-02 10:02:37 +00001929{
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001930 char *var = xasprintf("%s=%s", name, val);
Denys Vlasenko03dad222010-01-12 23:29:57 +01001931 set_local_var(var, /*flags:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
Mike Frysinger98c52642009-04-02 10:02:37 +00001932}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001933
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001934
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001935/*
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001936 * Helpers for "var1=val1 var2=val2 cmd" feature
1937 */
1938static void add_vars(struct variable *var)
1939{
1940 struct variable *next;
1941
1942 while (var) {
1943 next = var->next;
1944 var->next = G.top_var;
1945 G.top_var = var;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001946 if (var->flg_export) {
1947 debug_printf_env("%s: restoring exported '%s'\n", __func__, var->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001948 putenv(var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001949 } else {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001950 debug_printf_env("%s: restoring variable '%s'\n", __func__, var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001951 }
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001952 var = next;
1953 }
1954}
1955
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001956static struct variable *set_vars_and_save_old(char **strings)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001957{
1958 char **s;
1959 struct variable *old = NULL;
1960
1961 if (!strings)
1962 return old;
1963 s = strings;
1964 while (*s) {
1965 struct variable *var_p;
1966 struct variable **var_pp;
1967 char *eq;
1968
1969 eq = strchr(*s, '=');
1970 if (eq) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001971 var_pp = get_ptr_to_local_var(*s, eq - *s);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001972 if (var_pp) {
1973 /* Remove variable from global linked list */
1974 var_p = *var_pp;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001975 debug_printf_env("%s: removing '%s'\n", __func__, var_p->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001976 *var_pp = var_p->next;
1977 /* Add it to returned list */
1978 var_p->next = old;
1979 old = var_p;
1980 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001981 set_local_var(*s, /*exp:*/ 1, /*lvl:*/ 0, /*ro:*/ 0);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001982 }
1983 s++;
1984 }
1985 return old;
1986}
1987
1988
1989/*
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02001990 * Unicode helper
1991 */
1992static void reinit_unicode_for_hush(void)
1993{
1994 /* Unicode support should be activated even if LANG is set
1995 * _during_ shell execution, not only if it was set when
1996 * shell was started. Therefore, re-check LANG every time:
1997 */
Denys Vlasenko841f8332014-08-13 10:09:49 +02001998 if (ENABLE_FEATURE_CHECK_UNICODE_IN_ENV
1999 || ENABLE_UNICODE_USING_LOCALE
2000 ) {
2001 const char *s = get_local_var_value("LC_ALL");
2002 if (!s) s = get_local_var_value("LC_CTYPE");
2003 if (!s) s = get_local_var_value("LANG");
2004 reinit_unicode(s);
2005 }
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02002006}
2007
2008
2009/*
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002010 * in_str support
2011 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002012static int FAST_FUNC static_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002013{
Denys Vlasenko8391c482010-05-22 17:50:43 +02002014 int ch = *i->p;
2015 if (ch != '\0') {
2016 i->p++;
Denys Vlasenkocecbc982011-03-30 18:54:52 +02002017 i->last_char = ch;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002018 return ch;
Denys Vlasenko8391c482010-05-22 17:50:43 +02002019 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002020 return EOF;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002021}
2022
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002023static int FAST_FUNC static_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002024{
2025 return *i->p;
2026}
2027
2028#if ENABLE_HUSH_INTERACTIVE
2029
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002030static void cmdedit_update_prompt(void)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002031{
Mike Frysingerec2c6552009-03-28 12:24:44 +00002032 if (ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002033 G.PS1 = get_local_var_value("PS1");
Mike Frysingerec2c6552009-03-28 12:24:44 +00002034 if (G.PS1 == NULL)
2035 G.PS1 = "\\w \\$ ";
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002036 G.PS2 = get_local_var_value("PS2");
Denys Vlasenko690ad242009-04-30 21:24:24 +02002037 } else {
Mike Frysingerec2c6552009-03-28 12:24:44 +00002038 G.PS1 = NULL;
Denys Vlasenko690ad242009-04-30 21:24:24 +02002039 }
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002040 if (G.PS2 == NULL)
2041 G.PS2 = "> ";
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002042}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002043
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002044static const char *setup_prompt_string(int promptmode)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002045{
2046 const char *prompt_str;
2047 debug_printf("setup_prompt_string %d ", promptmode);
Mike Frysingerec2c6552009-03-28 12:24:44 +00002048 if (!ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
2049 /* Set up the prompt */
2050 if (promptmode == 0) { /* PS1 */
2051 free((char*)G.PS1);
Denys Vlasenko6db47842009-09-05 20:15:17 +02002052 /* bash uses $PWD value, even if it is set by user.
2053 * It uses current dir only if PWD is unset.
2054 * We always use current dir. */
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02002055 G.PS1 = xasprintf("%s %c ", get_cwd(0), (geteuid() != 0) ? '$' : '#');
Mike Frysingerec2c6552009-03-28 12:24:44 +00002056 prompt_str = G.PS1;
2057 } else
2058 prompt_str = G.PS2;
2059 } else
2060 prompt_str = (promptmode == 0) ? G.PS1 : G.PS2;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002061 debug_printf("result '%s'\n", prompt_str);
2062 return prompt_str;
2063}
2064
2065static void get_user_input(struct in_str *i)
2066{
2067 int r;
2068 const char *prompt_str;
2069
2070 prompt_str = setup_prompt_string(i->promptmode);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002071# if ENABLE_FEATURE_EDITING
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002072 /* Enable command line editing only while a command line
2073 * is actually being read */
2074 do {
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02002075 reinit_unicode_for_hush();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002076 G.flag_SIGINT = 0;
2077 /* buglet: SIGINT will not make new prompt to appear _at once_,
2078 * only after <Enter>. (^C will work) */
Denys Vlasenko66c5b122011-02-08 05:07:02 +01002079 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 +00002080 /* catch *SIGINT* etc (^C is handled by read_line_input) */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002081 check_and_run_traps();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002082 } while (r == 0 || G.flag_SIGINT); /* repeat if ^C or SIGINT */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002083 i->eof_flag = (r < 0);
2084 if (i->eof_flag) { /* EOF/error detected */
2085 G.user_input_buf[0] = EOF; /* yes, it will be truncated, it's ok */
2086 G.user_input_buf[1] = '\0';
2087 }
Denys Vlasenko8391c482010-05-22 17:50:43 +02002088# else
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002089 do {
2090 G.flag_SIGINT = 0;
Denys Vlasenkob8709032011-05-08 21:20:01 +02002091 if (i->last_char == '\0' || i->last_char == '\n') {
2092 /* Why check_and_run_traps here? Try this interactively:
2093 * $ trap 'echo INT' INT; (sleep 2; kill -INT $$) &
2094 * $ <[enter], repeatedly...>
2095 * Without check_and_run_traps, handler never runs.
2096 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002097 check_and_run_traps();
Denys Vlasenkob8709032011-05-08 21:20:01 +02002098 fputs(prompt_str, stdout);
2099 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002100 fflush_all();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002101 G.user_input_buf[0] = r = fgetc(i->file);
2102 /*G.user_input_buf[1] = '\0'; - already is and never changed */
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002103 } while (G.flag_SIGINT);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002104 i->eof_flag = (r == EOF);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002105# endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002106 i->p = G.user_input_buf;
2107}
2108
2109#endif /* INTERACTIVE */
2110
2111/* This is the magic location that prints prompts
2112 * and gets data back from the user */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002113static int FAST_FUNC file_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002114{
2115 int ch;
2116
2117 /* If there is data waiting, eat it up */
2118 if (i->p && *i->p) {
2119#if ENABLE_HUSH_INTERACTIVE
2120 take_cached:
2121#endif
2122 ch = *i->p++;
2123 if (i->eof_flag && !*i->p)
2124 ch = EOF;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002125 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002126 } else {
2127 /* need to double check i->file because we might be doing something
2128 * more complicated by now, like sourcing or substituting. */
2129#if ENABLE_HUSH_INTERACTIVE
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002130 if (G_interactive_fd && i->file == stdin) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002131 do {
2132 get_user_input(i);
2133 } while (!*i->p); /* need non-empty line */
2134 i->promptmode = 1; /* PS2 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002135 goto take_cached;
2136 }
2137#endif
Denis Vlasenko913a2012009-04-05 22:17:04 +00002138 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002139 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002140 debug_printf("file_get: got '%c' %d\n", ch, ch);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02002141 i->last_char = ch;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002142 return ch;
2143}
2144
Denis Vlasenko913a2012009-04-05 22:17:04 +00002145/* All callers guarantee this routine will never
2146 * be used right after a newline, so prompting is not needed.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002147 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002148static int FAST_FUNC file_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002149{
2150 int ch;
2151 if (i->p && *i->p) {
2152 if (i->eof_flag && !i->p[1])
2153 return EOF;
2154 return *i->p;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002155 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002156 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002157 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002158 i->eof_flag = (ch == EOF);
2159 i->peek_buf[0] = ch;
2160 i->peek_buf[1] = '\0';
2161 i->p = i->peek_buf;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002162 debug_printf("file_peek: got '%c' %d\n", ch, ch);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002163 return ch;
2164}
2165
2166static void setup_file_in_str(struct in_str *i, FILE *f)
2167{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002168 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002169 i->peek = file_peek;
2170 i->get = file_get;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002171 /* i->promptmode = 0; - PS1 (memset did it) */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002172 i->file = f;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002173 /* i->p = NULL; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002174}
2175
2176static void setup_string_in_str(struct in_str *i, const char *s)
2177{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002178 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002179 i->peek = static_peek;
2180 i->get = static_get;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002181 /* i->promptmode = 0; - PS1 (memset did it) */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002182 i->p = s;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002183 /* i->eof_flag = 0; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002184}
2185
2186
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002187/*
2188 * o_string support
2189 */
2190#define B_CHUNK (32 * sizeof(char*))
Eric Andersen25f27032001-04-26 23:22:31 +00002191
Denis Vlasenko0b677d82009-04-10 13:49:10 +00002192static void o_reset_to_empty_unquoted(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002193{
2194 o->length = 0;
Denys Vlasenko38292b62010-09-05 14:49:40 +02002195 o->has_quoted_part = 0;
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002196 if (o->data)
2197 o->data[0] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002198}
2199
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002200static void o_free(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002201{
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002202 free(o->data);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002203 memset(o, 0, sizeof(*o));
Eric Andersen25f27032001-04-26 23:22:31 +00002204}
2205
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002206static ALWAYS_INLINE void o_free_unsafe(o_string *o)
2207{
2208 free(o->data);
2209}
2210
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002211static void o_grow_by(o_string *o, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002212{
2213 if (o->length + len > o->maxlen) {
2214 o->maxlen += (2*len > B_CHUNK ? 2*len : B_CHUNK);
2215 o->data = xrealloc(o->data, 1 + o->maxlen);
2216 }
2217}
2218
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002219static void o_addchr(o_string *o, int ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002220{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002221 debug_printf("o_addchr: '%c' o->length=%d o=%p\n", ch, o->length, o);
2222 o_grow_by(o, 1);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002223 o->data[o->length] = ch;
2224 o->length++;
2225 o->data[o->length] = '\0';
2226}
2227
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002228static void o_addblock(o_string *o, const char *str, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002229{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002230 o_grow_by(o, len);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002231 memcpy(&o->data[o->length], str, len);
2232 o->length += len;
2233 o->data[o->length] = '\0';
2234}
2235
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002236static void o_addstr(o_string *o, const char *str)
Mike Frysinger98c52642009-04-02 10:02:37 +00002237{
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002238 o_addblock(o, str, strlen(str));
2239}
Denys Vlasenko2e48d532010-05-22 17:30:39 +02002240
Denys Vlasenko1e811b12010-05-22 03:12:29 +02002241#if !BB_MMU
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00002242static void nommu_addchr(o_string *o, int ch)
2243{
2244 if (o)
2245 o_addchr(o, ch);
2246}
2247#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002248# define nommu_addchr(o, str) ((void)0)
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002249#endif
2250
2251static void o_addstr_with_NUL(o_string *o, const char *str)
2252{
2253 o_addblock(o, str, strlen(str) + 1);
Mike Frysinger98c52642009-04-02 10:02:37 +00002254}
2255
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002256/*
Denys Vlasenko238081f2010-10-03 14:26:26 +02002257 * HUSH_BRACE_EXPANSION code needs corresponding quoting on variable expansion side.
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002258 * Currently, "v='{q,w}'; echo $v" erroneously expands braces in $v.
2259 * Apparently, on unquoted $v bash still does globbing
2260 * ("v='*.txt'; echo $v" prints all .txt files),
2261 * but NOT brace expansion! Thus, there should be TWO independent
2262 * quoting mechanisms on $v expansion side: one protects
2263 * $v from brace expansion, and other additionally protects "$v" against globbing.
2264 * We have only second one.
2265 */
2266
Denys Vlasenko9e800222010-10-03 14:28:04 +02002267#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002268# define MAYBE_BRACES "{}"
2269#else
2270# define MAYBE_BRACES ""
2271#endif
2272
Eric Andersen25f27032001-04-26 23:22:31 +00002273/* My analysis of quoting semantics tells me that state information
2274 * is associated with a destination, not a source.
2275 */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002276static void o_addqchr(o_string *o, int ch)
Eric Andersen25f27032001-04-26 23:22:31 +00002277{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002278 int sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002279 char *found = strchr("*?[\\" MAYBE_BRACES, ch);
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002280 if (found)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002281 sz++;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002282 o_grow_by(o, sz);
2283 if (found) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002284 o->data[o->length] = '\\';
2285 o->length++;
Eric Andersen25f27032001-04-26 23:22:31 +00002286 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002287 o->data[o->length] = ch;
2288 o->length++;
2289 o->data[o->length] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002290}
2291
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002292static void o_addQchr(o_string *o, int ch)
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002293{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002294 int sz = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002295 if ((o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)
2296 && strchr("*?[\\" MAYBE_BRACES, ch)
2297 ) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002298 sz++;
2299 o->data[o->length] = '\\';
2300 o->length++;
2301 }
2302 o_grow_by(o, sz);
2303 o->data[o->length] = ch;
2304 o->length++;
2305 o->data[o->length] = '\0';
2306}
2307
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002308static void o_addqblock(o_string *o, const char *str, int len)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002309{
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002310 while (len) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002311 char ch;
2312 int sz;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002313 int ordinary_cnt = strcspn(str, "*?[\\" MAYBE_BRACES);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002314 if (ordinary_cnt > len) /* paranoia */
2315 ordinary_cnt = len;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002316 o_addblock(o, str, ordinary_cnt);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002317 if (ordinary_cnt == len)
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002318 return; /* NUL is already added by o_addblock */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002319 str += ordinary_cnt;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002320 len -= ordinary_cnt + 1; /* we are processing + 1 char below */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002321
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002322 ch = *str++;
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002323 sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002324 if (ch) { /* it is necessarily one of "*?[\\" MAYBE_BRACES */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002325 sz++;
2326 o->data[o->length] = '\\';
2327 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002328 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002329 o_grow_by(o, sz);
2330 o->data[o->length] = ch;
2331 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002332 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002333 o->data[o->length] = '\0';
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002334}
2335
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002336static void o_addQblock(o_string *o, const char *str, int len)
2337{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002338 if (!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002339 o_addblock(o, str, len);
2340 return;
2341 }
2342 o_addqblock(o, str, len);
2343}
2344
Denys Vlasenko38292b62010-09-05 14:49:40 +02002345static void o_addQstr(o_string *o, const char *str)
2346{
2347 o_addQblock(o, str, strlen(str));
2348}
2349
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002350/* A special kind of o_string for $VAR and `cmd` expansion.
2351 * It contains char* list[] at the beginning, which is grown in 16 element
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002352 * increments. Actual string data starts at the next multiple of 16 * (char*).
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002353 * list[i] contains an INDEX (int!) into this string data.
2354 * It means that if list[] needs to grow, data needs to be moved higher up
2355 * but list[i]'s need not be modified.
2356 * NB: remembering how many list[i]'s you have there is crucial.
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002357 * o_finalize_list() operation post-processes this structure - calculates
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002358 * and stores actual char* ptrs in list[]. Oh, it NULL terminates it as well.
2359 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002360#if DEBUG_EXPAND || DEBUG_GLOB
2361static void debug_print_list(const char *prefix, o_string *o, int n)
2362{
2363 char **list = (char**)o->data;
2364 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2365 int i = 0;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002366
2367 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002368 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 +02002369 prefix, list, n, string_start, o->length, o->maxlen,
2370 !!(o->o_expflags & EXP_FLAG_GLOB),
2371 o->has_quoted_part,
2372 !!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002373 while (i < n) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002374 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002375 fdprintf(2, " list[%d]=%d '%s' %p\n", i, (int)(uintptr_t)list[i],
2376 o->data + (int)(uintptr_t)list[i] + string_start,
2377 o->data + (int)(uintptr_t)list[i] + string_start);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002378 i++;
2379 }
2380 if (n) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002381 const char *p = o->data + (int)(uintptr_t)list[n - 1] + string_start;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002382 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002383 fdprintf(2, " total_sz:%ld\n", (long)((p + strlen(p) + 1) - o->data));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002384 }
2385}
2386#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002387# define debug_print_list(prefix, o, n) ((void)0)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002388#endif
2389
2390/* n = o_save_ptr_helper(str, n) "starts new string" by storing an index value
2391 * in list[n] so that it points past last stored byte so far.
2392 * It returns n+1. */
2393static int o_save_ptr_helper(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002394{
2395 char **list = (char**)o->data;
Denis Vlasenko895bea22008-06-10 18:06:24 +00002396 int string_start;
2397 int string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002398
2399 if (!o->has_empty_slot) {
Denis Vlasenko895bea22008-06-10 18:06:24 +00002400 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2401 string_len = o->length - string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002402 if (!(n & 0xf)) { /* 0, 0x10, 0x20...? */
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002403 debug_printf_list("list[%d]=%d string_start=%d (growing)\n", n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002404 /* list[n] points to string_start, make space for 16 more pointers */
2405 o->maxlen += 0x10 * sizeof(list[0]);
2406 o->data = xrealloc(o->data, o->maxlen + 1);
Denis Vlasenko7049ff82008-06-25 09:53:17 +00002407 list = (char**)o->data;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002408 memmove(list + n + 0x10, list + n, string_len);
2409 o->length += 0x10 * sizeof(list[0]);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002410 } else {
2411 debug_printf_list("list[%d]=%d string_start=%d\n",
2412 n, string_len, string_start);
2413 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002414 } else {
2415 /* We have empty slot at list[n], reuse without growth */
Denis Vlasenko895bea22008-06-10 18:06:24 +00002416 string_start = ((n+1 + 0xf) & ~0xf) * sizeof(list[0]); /* NB: n+1! */
2417 string_len = o->length - string_start;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002418 debug_printf_list("list[%d]=%d string_start=%d (empty slot)\n",
2419 n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002420 o->has_empty_slot = 0;
2421 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002422 o->has_quoted_part = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002423 list[n] = (char*)(uintptr_t)string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002424 return n + 1;
2425}
2426
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002427/* "What was our last o_save_ptr'ed position (byte offset relative o->data)?" */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002428static int o_get_last_ptr(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002429{
2430 char **list = (char**)o->data;
2431 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2432
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002433 return ((int)(uintptr_t)list[n-1]) + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002434}
2435
Denys Vlasenko9e800222010-10-03 14:28:04 +02002436#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002437/* There in a GNU extension, GLOB_BRACE, but it is not usable:
2438 * first, it processes even {a} (no commas), second,
2439 * I didn't manage to make it return strings when they don't match
Denys Vlasenko160746b2009-11-16 05:51:18 +01002440 * existing files. Need to re-implement it.
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002441 */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002442
2443/* Helper */
2444static int glob_needed(const char *s)
2445{
2446 while (*s) {
2447 if (*s == '\\') {
2448 if (!s[1])
2449 return 0;
2450 s += 2;
2451 continue;
2452 }
2453 if (*s == '*' || *s == '[' || *s == '?' || *s == '{')
2454 return 1;
2455 s++;
2456 }
2457 return 0;
2458}
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002459/* Return pointer to next closing brace or to comma */
2460static const char *next_brace_sub(const char *cp)
2461{
2462 unsigned depth = 0;
2463 cp++;
2464 while (*cp != '\0') {
2465 if (*cp == '\\') {
2466 if (*++cp == '\0')
2467 break;
2468 cp++;
2469 continue;
Denys Vlasenko3581c622010-01-25 13:39:24 +01002470 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002471 if ((*cp == '}' && depth-- == 0) || (*cp == ',' && depth == 0))
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002472 break;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002473 if (*cp++ == '{')
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002474 depth++;
2475 }
2476
2477 return *cp != '\0' ? cp : NULL;
2478}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002479/* Recursive brace globber. Note: may garble pattern[]. */
2480static int glob_brace(char *pattern, o_string *o, int n)
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002481{
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002482 char *new_pattern_buf;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002483 const char *begin;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002484 const char *next;
2485 const char *rest;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002486 const char *p;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002487 size_t rest_len;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002488
2489 debug_printf_glob("glob_brace('%s')\n", pattern);
2490
2491 begin = pattern;
2492 while (1) {
2493 if (*begin == '\0')
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002494 goto simple_glob;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002495 if (*begin == '{') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002496 /* Find the first sub-pattern and at the same time
2497 * find the rest after the closing brace */
2498 next = next_brace_sub(begin);
2499 if (next == NULL) {
2500 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002501 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002502 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002503 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002504 /* "{abc}" with no commas - illegal
2505 * brace expr, disregard and skip it */
2506 begin = next + 1;
2507 continue;
2508 }
2509 break;
2510 }
2511 if (*begin == '\\' && begin[1] != '\0')
2512 begin++;
2513 begin++;
2514 }
2515 debug_printf_glob("begin:%s\n", begin);
2516 debug_printf_glob("next:%s\n", next);
2517
2518 /* Now find the end of the whole brace expression */
2519 rest = next;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002520 while (*rest != '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002521 rest = next_brace_sub(rest);
2522 if (rest == NULL) {
2523 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002524 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002525 }
2526 debug_printf_glob("rest:%s\n", rest);
2527 }
2528 rest_len = strlen(++rest) + 1;
2529
2530 /* We are sure the brace expression is well-formed */
2531
2532 /* Allocate working buffer large enough for our work */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002533 new_pattern_buf = xmalloc(strlen(pattern));
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002534
2535 /* We have a brace expression. BEGIN points to the opening {,
2536 * NEXT points past the terminator of the first element, and REST
2537 * points past the final }. We will accumulate result names from
2538 * recursive runs for each brace alternative in the buffer using
2539 * GLOB_APPEND. */
2540
2541 p = begin + 1;
2542 while (1) {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002543 /* Construct the new glob expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002544 memcpy(
2545 mempcpy(
2546 mempcpy(new_pattern_buf,
2547 /* We know the prefix for all sub-patterns */
2548 pattern, begin - pattern),
2549 p, next - p),
2550 rest, rest_len);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002551
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002552 /* Note: glob_brace() may garble new_pattern_buf[].
2553 * That's why we re-copy prefix every time (1st memcpy above).
2554 */
2555 n = glob_brace(new_pattern_buf, o, n);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002556 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002557 /* We saw the last entry */
2558 break;
2559 }
2560 p = next + 1;
2561 next = next_brace_sub(next);
2562 }
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002563 free(new_pattern_buf);
2564 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002565
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002566 simple_glob:
2567 {
2568 int gr;
2569 glob_t globdata;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002570
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002571 memset(&globdata, 0, sizeof(globdata));
2572 gr = glob(pattern, 0, NULL, &globdata);
2573 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
2574 if (gr != 0) {
2575 if (gr == GLOB_NOMATCH) {
2576 globfree(&globdata);
2577 /* NB: garbles parameter */
2578 unbackslash(pattern);
2579 o_addstr_with_NUL(o, pattern);
2580 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2581 return o_save_ptr_helper(o, n);
2582 }
2583 if (gr == GLOB_NOSPACE)
2584 bb_error_msg_and_die(bb_msg_memory_exhausted);
2585 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2586 * but we didn't specify it. Paranoia again. */
2587 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
2588 }
2589 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2590 char **argv = globdata.gl_pathv;
2591 while (1) {
2592 o_addstr_with_NUL(o, *argv);
2593 n = o_save_ptr_helper(o, n);
2594 argv++;
2595 if (!*argv)
2596 break;
2597 }
2598 }
2599 globfree(&globdata);
2600 }
2601 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002602}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002603/* Performs globbing on last list[],
2604 * saving each result as a new list[].
2605 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002606static int perform_glob(o_string *o, int n)
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002607{
2608 char *pattern, *copy;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002609
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002610 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002611 if (!o->data)
2612 return o_save_ptr_helper(o, n);
2613 pattern = o->data + o_get_last_ptr(o, n);
2614 debug_printf_glob("glob pattern '%s'\n", pattern);
2615 if (!glob_needed(pattern)) {
2616 /* unbackslash last string in o in place, fix length */
2617 o->length = unbackslash(pattern) - o->data;
2618 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2619 return o_save_ptr_helper(o, n);
2620 }
2621
2622 copy = xstrdup(pattern);
2623 /* "forget" pattern in o */
2624 o->length = pattern - o->data;
2625 n = glob_brace(copy, o, n);
2626 free(copy);
2627 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002628 debug_print_list("perform_glob returning", o, n);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002629 return n;
2630}
2631
Denys Vlasenko238081f2010-10-03 14:26:26 +02002632#else /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002633
2634/* Helper */
2635static int glob_needed(const char *s)
2636{
2637 while (*s) {
2638 if (*s == '\\') {
2639 if (!s[1])
2640 return 0;
2641 s += 2;
2642 continue;
2643 }
2644 if (*s == '*' || *s == '[' || *s == '?')
2645 return 1;
2646 s++;
2647 }
2648 return 0;
2649}
2650/* Performs globbing on last list[],
2651 * saving each result as a new list[].
2652 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002653static int perform_glob(o_string *o, int n)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002654{
2655 glob_t globdata;
2656 int gr;
2657 char *pattern;
2658
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002659 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002660 if (!o->data)
2661 return o_save_ptr_helper(o, n);
2662 pattern = o->data + o_get_last_ptr(o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002663 debug_printf_glob("glob pattern '%s'\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002664 if (!glob_needed(pattern)) {
2665 literal:
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002666 /* unbackslash last string in o in place, fix length */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002667 o->length = unbackslash(pattern) - o->data;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002668 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002669 return o_save_ptr_helper(o, n);
2670 }
2671
2672 memset(&globdata, 0, sizeof(globdata));
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002673 /* Can't use GLOB_NOCHECK: it does not unescape the string.
2674 * If we glob "*.\*" and don't find anything, we need
2675 * to fall back to using literal "*.*", but GLOB_NOCHECK
2676 * will return "*.\*"!
2677 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002678 gr = glob(pattern, 0, NULL, &globdata);
2679 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002680 if (gr != 0) {
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002681 if (gr == GLOB_NOMATCH) {
2682 globfree(&globdata);
2683 goto literal;
2684 }
2685 if (gr == GLOB_NOSPACE)
2686 bb_error_msg_and_die(bb_msg_memory_exhausted);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002687 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2688 * but we didn't specify it. Paranoia again. */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002689 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002690 }
2691 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2692 char **argv = globdata.gl_pathv;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002693 /* "forget" pattern in o */
2694 o->length = pattern - o->data;
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002695 while (1) {
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002696 o_addstr_with_NUL(o, *argv);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002697 n = o_save_ptr_helper(o, n);
2698 argv++;
2699 if (!*argv)
2700 break;
2701 }
2702 }
2703 globfree(&globdata);
2704 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002705 debug_print_list("perform_glob returning", o, n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002706 return n;
2707}
2708
Denys Vlasenko238081f2010-10-03 14:26:26 +02002709#endif /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002710
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002711/* If o->o_expflags & EXP_FLAG_GLOB, glob the string so far remembered.
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002712 * Otherwise, just finish current list[] and start new */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002713static int o_save_ptr(o_string *o, int n)
2714{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002715 if (o->o_expflags & EXP_FLAG_GLOB) {
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002716 /* If o->has_empty_slot, list[n] was already globbed
2717 * (if it was requested back then when it was filled)
2718 * so don't do that again! */
2719 if (!o->has_empty_slot)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002720 return perform_glob(o, n); /* o_save_ptr_helper is inside */
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002721 }
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002722 return o_save_ptr_helper(o, n);
2723}
2724
2725/* "Please convert list[n] to real char* ptrs, and NULL terminate it." */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002726static char **o_finalize_list(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002727{
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002728 char **list;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002729 int string_start;
2730
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002731 n = o_save_ptr(o, n); /* force growth for list[n] if necessary */
2732 if (DEBUG_EXPAND)
2733 debug_print_list("finalized", o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002734 debug_printf_expand("finalized n:%d\n", n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002735 list = (char**)o->data;
2736 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2737 list[--n] = NULL;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002738 while (n) {
2739 n--;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002740 list[n] = o->data + (int)(uintptr_t)list[n] + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002741 }
2742 return list;
2743}
2744
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002745static void free_pipe_list(struct pipe *pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002746
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002747/* Returns pi->next - next pipe in the list */
2748static struct pipe *free_pipe(struct pipe *pi)
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002749{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002750 struct pipe *next;
2751 int i;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002752
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002753 debug_printf_clean("free_pipe (pid %d)\n", getpid());
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002754 for (i = 0; i < pi->num_cmds; i++) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002755 struct command *command;
2756 struct redir_struct *r, *rnext;
2757
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002758 command = &pi->cmds[i];
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002759 debug_printf_clean(" command %d:\n", i);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002760 if (command->argv) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002761 if (DEBUG_CLEAN) {
2762 int a;
2763 char **p;
2764 for (a = 0, p = command->argv; *p; a++, p++) {
2765 debug_printf_clean(" argv[%d] = %s\n", a, *p);
2766 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002767 }
2768 free_strings(command->argv);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002769 //command->argv = NULL;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002770 }
2771 /* not "else if": on syntax error, we may have both! */
2772 if (command->group) {
Denys Vlasenko9d617c42009-06-09 18:40:52 +02002773 debug_printf_clean(" begin group (cmd_type:%d)\n",
2774 command->cmd_type);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002775 free_pipe_list(command->group);
2776 debug_printf_clean(" end group\n");
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002777 //command->group = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002778 }
Denis Vlasenkoed055212009-04-11 10:37:10 +00002779 /* else is crucial here.
2780 * If group != NULL, child_func is meaningless */
2781#if ENABLE_HUSH_FUNCTIONS
2782 else if (command->child_func) {
2783 debug_printf_exec("cmd %p releases child func at %p\n", command, command->child_func);
2784 command->child_func->parent_cmd = NULL;
2785 }
2786#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002787#if !BB_MMU
2788 free(command->group_as_string);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002789 //command->group_as_string = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002790#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002791 for (r = command->redirects; r; r = rnext) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002792 debug_printf_clean(" redirect %d%s",
2793 r->rd_fd, redir_table[r->rd_type].descrip);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002794 /* guard against the case >$FOO, where foo is unset or blank */
2795 if (r->rd_filename) {
2796 debug_printf_clean(" fname:'%s'\n", r->rd_filename);
2797 free(r->rd_filename);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002798 //r->rd_filename = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002799 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002800 debug_printf_clean(" rd_dup:%d\n", r->rd_dup);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002801 rnext = r->next;
2802 free(r);
2803 }
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002804 //command->redirects = NULL;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002805 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002806 free(pi->cmds); /* children are an array, they get freed all at once */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002807 //pi->cmds = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002808#if ENABLE_HUSH_JOB
2809 free(pi->cmdtext);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002810 //pi->cmdtext = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002811#endif
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002812
2813 next = pi->next;
2814 free(pi);
2815 return next;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002816}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002817
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002818static void free_pipe_list(struct pipe *pi)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002819{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002820 while (pi) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002821#if HAS_KEYWORDS
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002822 debug_printf_clean("pipe reserved word %d\n", pi->res_word);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002823#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002824 debug_printf_clean("pipe followup code %d\n", pi->followup);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002825 pi = free_pipe(pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002826 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002827}
2828
2829
Denys Vlasenkob36abf22010-09-05 14:50:59 +02002830/*** Parsing routines ***/
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002831
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002832#ifndef debug_print_tree
2833static void debug_print_tree(struct pipe *pi, int lvl)
2834{
2835 static const char *const PIPE[] = {
2836 [PIPE_SEQ] = "SEQ",
2837 [PIPE_AND] = "AND",
2838 [PIPE_OR ] = "OR" ,
2839 [PIPE_BG ] = "BG" ,
2840 };
2841 static const char *RES[] = {
2842 [RES_NONE ] = "NONE" ,
2843# if ENABLE_HUSH_IF
2844 [RES_IF ] = "IF" ,
2845 [RES_THEN ] = "THEN" ,
2846 [RES_ELIF ] = "ELIF" ,
2847 [RES_ELSE ] = "ELSE" ,
2848 [RES_FI ] = "FI" ,
2849# endif
2850# if ENABLE_HUSH_LOOPS
2851 [RES_FOR ] = "FOR" ,
2852 [RES_WHILE] = "WHILE",
2853 [RES_UNTIL] = "UNTIL",
2854 [RES_DO ] = "DO" ,
2855 [RES_DONE ] = "DONE" ,
2856# endif
2857# if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
2858 [RES_IN ] = "IN" ,
2859# endif
2860# if ENABLE_HUSH_CASE
2861 [RES_CASE ] = "CASE" ,
2862 [RES_CASE_IN ] = "CASE_IN" ,
2863 [RES_MATCH] = "MATCH",
2864 [RES_CASE_BODY] = "CASE_BODY",
2865 [RES_ESAC ] = "ESAC" ,
2866# endif
2867 [RES_XXXX ] = "XXXX" ,
2868 [RES_SNTX ] = "SNTX" ,
2869 };
2870 static const char *const CMDTYPE[] = {
2871 "{}",
2872 "()",
2873 "[noglob]",
2874# if ENABLE_HUSH_FUNCTIONS
2875 "func()",
2876# endif
2877 };
2878
2879 int pin, prn;
2880
2881 pin = 0;
2882 while (pi) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002883 fdprintf(2, "%*spipe %d res_word=%s followup=%d %s\n", lvl*2, "",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002884 pin, RES[pi->res_word], pi->followup, PIPE[pi->followup]);
2885 prn = 0;
2886 while (prn < pi->num_cmds) {
2887 struct command *command = &pi->cmds[prn];
2888 char **argv = command->argv;
2889
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002890 fdprintf(2, "%*s cmd %d assignment_cnt:%d",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002891 lvl*2, "", prn,
2892 command->assignment_cnt);
2893 if (command->group) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002894 fdprintf(2, " group %s: (argv=%p)%s%s\n",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002895 CMDTYPE[command->cmd_type],
2896 argv
2897# if !BB_MMU
2898 , " group_as_string:", command->group_as_string
2899# else
2900 , "", ""
2901# endif
2902 );
2903 debug_print_tree(command->group, lvl+1);
2904 prn++;
2905 continue;
2906 }
2907 if (argv) while (*argv) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002908 fdprintf(2, " '%s'", *argv);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002909 argv++;
2910 }
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002911 fdprintf(2, "\n");
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002912 prn++;
2913 }
2914 pi = pi->next;
2915 pin++;
2916 }
2917}
2918#endif /* debug_print_tree */
2919
Denis Vlasenkoac678ec2007-04-16 22:32:04 +00002920static struct pipe *new_pipe(void)
2921{
Eric Andersen25f27032001-04-26 23:22:31 +00002922 struct pipe *pi;
Denis Vlasenko3ac0e002007-04-28 16:45:22 +00002923 pi = xzalloc(sizeof(struct pipe));
Denis Vlasenkoa8442002008-06-14 11:00:17 +00002924 /*pi->followup = 0; - deliberately invalid value */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00002925 /*pi->res_word = RES_NONE; - RES_NONE is 0 anyway */
Eric Andersen25f27032001-04-26 23:22:31 +00002926 return pi;
2927}
2928
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002929/* Command (member of a pipe) is complete, or we start a new pipe
2930 * if ctx->command is NULL.
2931 * No errors possible here.
2932 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002933static int done_command(struct parse_context *ctx)
2934{
2935 /* The command is really already in the pipe structure, so
2936 * advance the pipe counter and make a new, null command. */
2937 struct pipe *pi = ctx->pipe;
2938 struct command *command = ctx->command;
2939
2940 if (command) {
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002941 if (IS_NULL_CMD(command)) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002942 debug_printf_parse("done_command: skipping null cmd, num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002943 goto clear_and_ret;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002944 }
2945 pi->num_cmds++;
2946 debug_printf_parse("done_command: ++num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002947 //debug_print_tree(ctx->list_head, 20);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002948 } else {
2949 debug_printf_parse("done_command: initializing, num_cmds=%d\n", pi->num_cmds);
2950 }
2951
2952 /* Only real trickiness here is that the uncommitted
2953 * command structure is not counted in pi->num_cmds. */
2954 pi->cmds = xrealloc(pi->cmds, sizeof(*pi->cmds) * (pi->num_cmds+1));
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002955 ctx->command = command = &pi->cmds[pi->num_cmds];
2956 clear_and_ret:
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002957 memset(command, 0, sizeof(*command));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002958 return pi->num_cmds; /* used only for 0/nonzero check */
2959}
2960
2961static void done_pipe(struct parse_context *ctx, pipe_style type)
2962{
2963 int not_null;
2964
2965 debug_printf_parse("done_pipe entered, followup %d\n", type);
2966 /* Close previous command */
2967 not_null = done_command(ctx);
2968 ctx->pipe->followup = type;
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002969#if HAS_KEYWORDS
2970 ctx->pipe->pi_inverted = ctx->ctx_inverted;
2971 ctx->ctx_inverted = 0;
2972 ctx->pipe->res_word = ctx->ctx_res_w;
2973#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002974
2975 /* Without this check, even just <enter> on command line generates
2976 * tree of three NOPs (!). Which is harmless but annoying.
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002977 * IOW: it is safe to do it unconditionally. */
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002978 if (not_null
Denis Vlasenko7f959372009-04-14 08:06:59 +00002979#if ENABLE_HUSH_IF
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002980 || ctx->ctx_res_w == RES_FI
Denis Vlasenko7f959372009-04-14 08:06:59 +00002981#endif
2982#if ENABLE_HUSH_LOOPS
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002983 || ctx->ctx_res_w == RES_DONE
2984 || ctx->ctx_res_w == RES_FOR
2985 || ctx->ctx_res_w == RES_IN
Denis Vlasenko7f959372009-04-14 08:06:59 +00002986#endif
2987#if ENABLE_HUSH_CASE
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002988 || ctx->ctx_res_w == RES_ESAC
2989#endif
2990 ) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002991 struct pipe *new_p;
2992 debug_printf_parse("done_pipe: adding new pipe: "
2993 "not_null:%d ctx->ctx_res_w:%d\n",
2994 not_null, ctx->ctx_res_w);
2995 new_p = new_pipe();
2996 ctx->pipe->next = new_p;
2997 ctx->pipe = new_p;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002998 /* RES_THEN, RES_DO etc are "sticky" -
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002999 * they remain set for pipes inside if/while.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003000 * This is used to control execution.
3001 * RES_FOR and RES_IN are NOT sticky (needed to support
3002 * cases where variable or value happens to match a keyword):
3003 */
3004#if ENABLE_HUSH_LOOPS
3005 if (ctx->ctx_res_w == RES_FOR
3006 || ctx->ctx_res_w == RES_IN)
3007 ctx->ctx_res_w = RES_NONE;
3008#endif
3009#if ENABLE_HUSH_CASE
3010 if (ctx->ctx_res_w == RES_MATCH)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003011 ctx->ctx_res_w = RES_CASE_BODY;
3012 if (ctx->ctx_res_w == RES_CASE)
3013 ctx->ctx_res_w = RES_CASE_IN;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003014#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003015 ctx->command = NULL; /* trick done_command below */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003016 /* Create the memory for command, roughly:
3017 * ctx->pipe->cmds = new struct command;
3018 * ctx->command = &ctx->pipe->cmds[0];
3019 */
3020 done_command(ctx);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00003021 //debug_print_tree(ctx->list_head, 10);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003022 }
3023 debug_printf_parse("done_pipe return\n");
3024}
3025
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003026static void initialize_context(struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003027{
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003028 memset(ctx, 0, sizeof(*ctx));
Denis Vlasenko1a735862007-05-23 00:32:25 +00003029 ctx->pipe = ctx->list_head = new_pipe();
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003030 /* Create the memory for command, roughly:
3031 * ctx->pipe->cmds = new struct command;
3032 * ctx->command = &ctx->pipe->cmds[0];
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003033 */
3034 done_command(ctx);
Eric Andersen25f27032001-04-26 23:22:31 +00003035}
3036
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003037/* If a reserved word is found and processed, parse context is modified
3038 * and 1 is returned.
Eric Andersen25f27032001-04-26 23:22:31 +00003039 */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003040#if HAS_KEYWORDS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003041struct reserved_combo {
3042 char literal[6];
3043 unsigned char res;
3044 unsigned char assignment_flag;
3045 int flag;
3046};
3047enum {
3048 FLAG_END = (1 << RES_NONE ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003049# if ENABLE_HUSH_IF
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003050 FLAG_IF = (1 << RES_IF ),
3051 FLAG_THEN = (1 << RES_THEN ),
3052 FLAG_ELIF = (1 << RES_ELIF ),
3053 FLAG_ELSE = (1 << RES_ELSE ),
3054 FLAG_FI = (1 << RES_FI ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003055# endif
3056# if ENABLE_HUSH_LOOPS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003057 FLAG_FOR = (1 << RES_FOR ),
3058 FLAG_WHILE = (1 << RES_WHILE),
3059 FLAG_UNTIL = (1 << RES_UNTIL),
3060 FLAG_DO = (1 << RES_DO ),
3061 FLAG_DONE = (1 << RES_DONE ),
3062 FLAG_IN = (1 << RES_IN ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003063# endif
3064# if ENABLE_HUSH_CASE
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003065 FLAG_MATCH = (1 << RES_MATCH),
3066 FLAG_ESAC = (1 << RES_ESAC ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003067# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003068 FLAG_START = (1 << RES_XXXX ),
3069};
3070
3071static const struct reserved_combo* match_reserved_word(o_string *word)
3072{
Eric Andersen25f27032001-04-26 23:22:31 +00003073 /* Mostly a list of accepted follow-up reserved words.
3074 * FLAG_END means we are done with the sequence, and are ready
3075 * to turn the compound list into a command.
3076 * FLAG_START means the word must start a new compound list.
3077 */
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003078 static const struct reserved_combo reserved_list[] = {
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003079# if ENABLE_HUSH_IF
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003080 { "!", RES_NONE, NOT_ASSIGNMENT , 0 },
3081 { "if", RES_IF, MAYBE_ASSIGNMENT, FLAG_THEN | FLAG_START },
3082 { "then", RES_THEN, MAYBE_ASSIGNMENT, FLAG_ELIF | FLAG_ELSE | FLAG_FI },
3083 { "elif", RES_ELIF, MAYBE_ASSIGNMENT, FLAG_THEN },
3084 { "else", RES_ELSE, MAYBE_ASSIGNMENT, FLAG_FI },
3085 { "fi", RES_FI, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003086# endif
3087# if ENABLE_HUSH_LOOPS
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003088 { "for", RES_FOR, NOT_ASSIGNMENT , FLAG_IN | FLAG_DO | FLAG_START },
3089 { "while", RES_WHILE, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3090 { "until", RES_UNTIL, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3091 { "in", RES_IN, NOT_ASSIGNMENT , FLAG_DO },
3092 { "do", RES_DO, MAYBE_ASSIGNMENT, FLAG_DONE },
3093 { "done", RES_DONE, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003094# endif
3095# if ENABLE_HUSH_CASE
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003096 { "case", RES_CASE, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_START },
3097 { "esac", RES_ESAC, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003098# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003099 };
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003100 const struct reserved_combo *r;
3101
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02003102 for (r = reserved_list; r < reserved_list + ARRAY_SIZE(reserved_list); r++) {
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003103 if (strcmp(word->data, r->literal) == 0)
3104 return r;
3105 }
3106 return NULL;
3107}
Denis Vlasenkobb929512009-04-16 10:59:40 +00003108/* Return 0: not a keyword, 1: keyword
3109 */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003110static int reserved_word(o_string *word, struct parse_context *ctx)
3111{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003112# if ENABLE_HUSH_CASE
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003113 static const struct reserved_combo reserved_match = {
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003114 "", RES_MATCH, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_ESAC
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003115 };
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003116# endif
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003117 const struct reserved_combo *r;
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003118
Denys Vlasenko38292b62010-09-05 14:49:40 +02003119 if (word->has_quoted_part)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003120 return 0;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003121 r = match_reserved_word(word);
3122 if (!r)
3123 return 0;
3124
3125 debug_printf("found reserved word %s, res %d\n", r->literal, r->res);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003126# if ENABLE_HUSH_CASE
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003127 if (r->res == RES_IN && ctx->ctx_res_w == RES_CASE_IN) {
3128 /* "case word IN ..." - IN part starts first MATCH part */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003129 r = &reserved_match;
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003130 } else
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003131# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003132 if (r->flag == 0) { /* '!' */
3133 if (ctx->ctx_inverted) { /* bash doesn't accept '! ! true' */
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003134 syntax_error("! ! command");
Denis Vlasenkobb929512009-04-16 10:59:40 +00003135 ctx->ctx_res_w = RES_SNTX;
Eric Andersen25f27032001-04-26 23:22:31 +00003136 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003137 ctx->ctx_inverted = 1;
Denis Vlasenko1a735862007-05-23 00:32:25 +00003138 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003139 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003140 if (r->flag & FLAG_START) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003141 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003142
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003143 old = xmalloc(sizeof(*old));
3144 debug_printf_parse("push stack %p\n", old);
3145 *old = *ctx; /* physical copy */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003146 initialize_context(ctx);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003147 ctx->stack = old;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003148 } else if (/*ctx->ctx_res_w == RES_NONE ||*/ !(ctx->old_flag & (1 << r->res))) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003149 syntax_error_at(word->data);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003150 ctx->ctx_res_w = RES_SNTX;
3151 return 1;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003152 } else {
3153 /* "{...} fi" is ok. "{...} if" is not
3154 * Example:
3155 * if { echo foo; } then { echo bar; } fi */
3156 if (ctx->command->group)
3157 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003158 }
Denis Vlasenkobb929512009-04-16 10:59:40 +00003159
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003160 ctx->ctx_res_w = r->res;
3161 ctx->old_flag = r->flag;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003162 word->o_assignment = r->assignment_flag;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003163 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
Denis Vlasenkobb929512009-04-16 10:59:40 +00003164
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003165 if (ctx->old_flag & FLAG_END) {
3166 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003167
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003168 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003169 debug_printf_parse("pop stack %p\n", ctx->stack);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003170 old = ctx->stack;
3171 old->command->group = ctx->list_head;
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003172 old->command->cmd_type = CMD_NORMAL;
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003173# if !BB_MMU
Denys Vlasenkob5be13c2015-09-04 06:22:10 +02003174 /* At this point, the compound command's string is in
3175 * ctx->as_string... except for the leading keyword!
3176 * Consider this example: "echo a | if true; then echo a; fi"
3177 * ctx->as_string will contain "true; then echo a; fi",
3178 * with "if " remaining in old->as_string!
3179 */
3180 {
3181 char *str;
3182 int len = old->as_string.length;
3183 /* Concatenate halves */
3184 o_addstr(&old->as_string, ctx->as_string.data);
3185 o_free_unsafe(&ctx->as_string);
3186 /* Find where leading keyword starts in first half */
3187 str = old->as_string.data + len;
3188 if (str > old->as_string.data)
3189 str--; /* skip whitespace after keyword */
3190 while (str > old->as_string.data && isalpha(str[-1]))
3191 str--;
3192 /* Ugh, we're done with this horrid hack */
3193 old->command->group_as_string = xstrdup(str);
3194 debug_printf_parse("pop, remembering as:'%s'\n",
3195 old->command->group_as_string);
3196 }
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003197# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003198 *ctx = *old; /* physical copy */
3199 free(old);
3200 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003201 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003202}
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003203#endif /* HAS_KEYWORDS */
Eric Andersen25f27032001-04-26 23:22:31 +00003204
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003205/* Word is complete, look at it and update parsing context.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003206 * Normal return is 0. Syntax errors return 1.
3207 * Note: on return, word is reset, but not o_free'd!
3208 */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003209static int done_word(o_string *word, struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003210{
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003211 struct command *command = ctx->command;
Eric Andersen25f27032001-04-26 23:22:31 +00003212
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003213 debug_printf_parse("done_word entered: '%s' %p\n", word->data, command);
Denys Vlasenko38292b62010-09-05 14:49:40 +02003214 if (word->length == 0 && !word->has_quoted_part) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003215 debug_printf_parse("done_word return 0: true null, ignored\n");
3216 return 0;
Eric Andersen25f27032001-04-26 23:22:31 +00003217 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003218
Eric Andersen25f27032001-04-26 23:22:31 +00003219 if (ctx->pending_redirect) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003220 /* We do not glob in e.g. >*.tmp case. bash seems to glob here
3221 * only if run as "bash", not "sh" */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003222 /* http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3223 * "2.7 Redirection
3224 * ...the word that follows the redirection operator
3225 * shall be subjected to tilde expansion, parameter expansion,
3226 * command substitution, arithmetic expansion, and quote
3227 * removal. Pathname expansion shall not be performed
3228 * on the word by a non-interactive shell; an interactive
3229 * shell may perform it, but shall do so only when
3230 * the expansion would result in one word."
3231 */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003232 ctx->pending_redirect->rd_filename = xstrdup(word->data);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003233 /* Cater for >\file case:
3234 * >\a creates file a; >\\a, >"\a", >"\\a" create file \a
3235 * Same with heredocs:
3236 * for <<\H delim is H; <<\\H, <<"\H", <<"\\H" - \H
3237 */
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003238 if (ctx->pending_redirect->rd_type == REDIRECT_HEREDOC) {
3239 unbackslash(ctx->pending_redirect->rd_filename);
3240 /* Is it <<"HEREDOC"? */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003241 if (word->has_quoted_part) {
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003242 ctx->pending_redirect->rd_dup |= HEREDOC_QUOTED;
3243 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003244 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003245 debug_printf_parse("word stored in rd_filename: '%s'\n", word->data);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003246 ctx->pending_redirect = NULL;
Eric Andersen25f27032001-04-26 23:22:31 +00003247 } else {
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003248#if HAS_KEYWORDS
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003249# if ENABLE_HUSH_CASE
Denis Vlasenko757361f2008-07-14 08:26:47 +00003250 if (ctx->ctx_dsemicolon
3251 && strcmp(word->data, "esac") != 0 /* not "... pattern) cmd;; esac" */
3252 ) {
Denis Vlasenko395ae452008-07-14 06:29:38 +00003253 /* already done when ctx_dsemicolon was set to 1: */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003254 /* ctx->ctx_res_w = RES_MATCH; */
3255 ctx->ctx_dsemicolon = 0;
3256 } else
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003257# endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003258 if (!command->argv /* if it's the first word... */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003259# if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003260 && ctx->ctx_res_w != RES_FOR /* ...not after FOR or IN */
3261 && ctx->ctx_res_w != RES_IN
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003262# endif
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003263# if ENABLE_HUSH_CASE
3264 && ctx->ctx_res_w != RES_CASE
3265# endif
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003266 ) {
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003267 int reserved = reserved_word(word, ctx);
3268 debug_printf_parse("checking for reserved-ness: %d\n", reserved);
3269 if (reserved) {
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003270 o_reset_to_empty_unquoted(word);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003271 debug_printf_parse("done_word return %d\n",
3272 (ctx->ctx_res_w == RES_SNTX));
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003273 return (ctx->ctx_res_w == RES_SNTX);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003274 }
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003275# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003276 if (strcmp(word->data, "[[") == 0) {
3277 command->cmd_type = CMD_SINGLEWORD_NOGLOB;
3278 }
3279 /* fall through */
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003280# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003281 }
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003282#endif
Denis Vlasenkobb929512009-04-16 10:59:40 +00003283 if (command->group) {
3284 /* "{ echo foo; } echo bar" - bad */
3285 syntax_error_at(word->data);
3286 debug_printf_parse("done_word return 1: syntax error, "
3287 "groups and arglists don't mix\n");
3288 return 1;
3289 }
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003290
3291 /* If this word wasn't an assignment, next ones definitely
3292 * can't be assignments. Even if they look like ones. */
3293 if (word->o_assignment != DEFINITELY_ASSIGNMENT
3294 && word->o_assignment != WORD_IS_KEYWORD
3295 ) {
3296 word->o_assignment = NOT_ASSIGNMENT;
3297 } else {
3298 if (word->o_assignment == DEFINITELY_ASSIGNMENT) {
3299 command->assignment_cnt++;
3300 debug_printf_parse("++assignment_cnt=%d\n", command->assignment_cnt);
3301 }
3302 debug_printf_parse("word->o_assignment was:'%s'\n", assignment_flag[word->o_assignment]);
3303 word->o_assignment = MAYBE_ASSIGNMENT;
3304 }
3305 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
3306
Denys Vlasenko38292b62010-09-05 14:49:40 +02003307 if (word->has_quoted_part
Denis Vlasenko55789c62008-06-18 16:30:42 +00003308 /* optimization: and if it's ("" or '') or ($v... or `cmd`...): */
3309 && (word->data[0] == '\0' || word->data[0] == SPECIAL_VAR_SYMBOL)
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003310 /* (otherwise it's known to be not empty and is already safe) */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003311 ) {
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003312 /* exclude "$@" - it can expand to no word despite "" */
Denis Vlasenkoafdcd122008-07-05 17:40:04 +00003313 char *p = word->data;
3314 while (p[0] == SPECIAL_VAR_SYMBOL
3315 && (p[1] & 0x7f) == '@'
3316 && p[2] == SPECIAL_VAR_SYMBOL
3317 ) {
3318 p += 3;
3319 }
Denis Vlasenkoc1c63b62008-06-18 09:20:35 +00003320 }
Denis Vlasenko22d10a02008-10-13 08:53:43 +00003321 command->argv = add_string_to_strings(command->argv, xstrdup(word->data));
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003322 debug_print_strings("word appended to argv", command->argv);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003323 }
Eric Andersen25f27032001-04-26 23:22:31 +00003324
Denis Vlasenko06810332007-05-21 23:30:54 +00003325#if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003326 if (ctx->ctx_res_w == RES_FOR) {
Denys Vlasenko38292b62010-09-05 14:49:40 +02003327 if (word->has_quoted_part
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003328 || !is_well_formed_var_name(command->argv[0], '\0')
3329 ) {
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003330 /* bash says just "not a valid identifier" */
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003331 syntax_error("not a valid identifier in for");
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003332 return 1;
3333 }
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003334 /* Force FOR to have just one word (variable name) */
3335 /* NB: basically, this makes hush see "for v in ..."
3336 * syntax as if it is "for v; in ...". FOR and IN become
3337 * two pipe structs in parse tree. */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00003338 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003339 }
Denis Vlasenko06810332007-05-21 23:30:54 +00003340#endif
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003341#if ENABLE_HUSH_CASE
3342 /* Force CASE to have just one word */
3343 if (ctx->ctx_res_w == RES_CASE) {
3344 done_pipe(ctx, PIPE_SEQ);
3345 }
3346#endif
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003347
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003348 o_reset_to_empty_unquoted(word);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003349
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003350 debug_printf_parse("done_word return 0\n");
Eric Andersen25f27032001-04-26 23:22:31 +00003351 return 0;
3352}
3353
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003354
3355/* Peek ahead in the input to find out if we have a "&n" construct,
3356 * as in "2>&1", that represents duplicating a file descriptor.
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003357 * Return:
3358 * REDIRFD_CLOSE if >&- "close fd" construct is seen,
3359 * REDIRFD_SYNTAX_ERR if syntax error,
3360 * REDIRFD_TO_FILE if no & was seen,
3361 * or the number found.
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003362 */
3363#if BB_MMU
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003364#define parse_redir_right_fd(as_string, input) \
3365 parse_redir_right_fd(input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003366#endif
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003367static int parse_redir_right_fd(o_string *as_string, struct in_str *input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003368{
3369 int ch, d, ok;
3370
3371 ch = i_peek(input);
3372 if (ch != '&')
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003373 return REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003374
3375 ch = i_getch(input); /* get the & */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003376 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003377 ch = i_peek(input);
3378 if (ch == '-') {
3379 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003380 nommu_addchr(as_string, ch);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003381 return REDIRFD_CLOSE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003382 }
3383 d = 0;
3384 ok = 0;
3385 while (ch != EOF && isdigit(ch)) {
3386 d = d*10 + (ch-'0');
3387 ok = 1;
3388 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003389 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003390 ch = i_peek(input);
3391 }
3392 if (ok) return d;
3393
3394//TODO: this is the place to catch ">&file" bashism (redirect both fd 1 and 2)
3395
3396 bb_error_msg("ambiguous redirect");
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003397 return REDIRFD_SYNTAX_ERR;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003398}
3399
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003400/* Return code is 0 normal, 1 if a syntax error is detected
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003401 */
3402static int parse_redirect(struct parse_context *ctx,
3403 int fd,
3404 redir_type style,
3405 struct in_str *input)
3406{
3407 struct command *command = ctx->command;
3408 struct redir_struct *redir;
3409 struct redir_struct **redirp;
3410 int dup_num;
3411
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003412 dup_num = REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003413 if (style != REDIRECT_HEREDOC) {
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003414 /* Check for a '>&1' type redirect */
3415 dup_num = parse_redir_right_fd(&ctx->as_string, input);
3416 if (dup_num == REDIRFD_SYNTAX_ERR)
3417 return 1;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003418 } else {
3419 int ch = i_peek(input);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003420 dup_num = (ch == '-'); /* HEREDOC_SKIPTABS bit is 1 */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003421 if (dup_num) { /* <<-... */
3422 ch = i_getch(input);
3423 nommu_addchr(&ctx->as_string, ch);
3424 ch = i_peek(input);
3425 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003426 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003427
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003428 if (style == REDIRECT_OVERWRITE && dup_num == REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003429 int ch = i_peek(input);
3430 if (ch == '|') {
3431 /* >|FILE redirect ("clobbering" >).
3432 * Since we do not support "set -o noclobber" yet,
3433 * >| and > are the same for now. Just eat |.
3434 */
3435 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003436 nommu_addchr(&ctx->as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003437 }
3438 }
3439
3440 /* Create a new redir_struct and append it to the linked list */
3441 redirp = &command->redirects;
3442 while ((redir = *redirp) != NULL) {
3443 redirp = &(redir->next);
3444 }
3445 *redirp = redir = xzalloc(sizeof(*redir));
3446 /* redir->next = NULL; */
3447 /* redir->rd_filename = NULL; */
3448 redir->rd_type = style;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003449 redir->rd_fd = (fd == -1) ? redir_table[style].default_fd : fd;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003450
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003451 debug_printf_parse("redirect type %d %s\n", redir->rd_fd,
3452 redir_table[style].descrip);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003453
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003454 redir->rd_dup = dup_num;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003455 if (style != REDIRECT_HEREDOC && dup_num != REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003456 /* Erik had a check here that the file descriptor in question
3457 * is legit; I postpone that to "run time"
3458 * A "-" representation of "close me" shows up as a -3 here */
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003459 debug_printf_parse("duplicating redirect '%d>&%d'\n",
3460 redir->rd_fd, redir->rd_dup);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003461 } else {
3462 /* Set ctx->pending_redirect, so we know what to do at the
3463 * end of the next parsed word. */
3464 ctx->pending_redirect = redir;
3465 }
3466 return 0;
3467}
3468
Eric Andersen25f27032001-04-26 23:22:31 +00003469/* If a redirect is immediately preceded by a number, that number is
3470 * supposed to tell which file descriptor to redirect. This routine
3471 * looks for such preceding numbers. In an ideal world this routine
3472 * needs to handle all the following classes of redirects...
3473 * echo 2>foo # redirects fd 2 to file "foo", nothing passed to echo
3474 * echo 49>foo # redirects fd 49 to file "foo", nothing passed to echo
3475 * echo -2>foo # redirects fd 1 to file "foo", "-2" passed to echo
3476 * echo 49x>foo # redirects fd 1 to file "foo", "49x" passed to echo
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003477 *
3478 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3479 * "2.7 Redirection
3480 * ... If n is quoted, the number shall not be recognized as part of
3481 * the redirection expression. For example:
3482 * echo \2>a
3483 * writes the character 2 into file a"
Denys Vlasenko38292b62010-09-05 14:49:40 +02003484 * We are getting it right by setting ->has_quoted_part on any \<char>
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003485 *
3486 * A -1 return means no valid number was found,
3487 * the caller should use the appropriate default for this redirection.
Eric Andersen25f27032001-04-26 23:22:31 +00003488 */
3489static int redirect_opt_num(o_string *o)
3490{
3491 int num;
3492
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003493 if (o->data == NULL)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00003494 return -1;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003495 num = bb_strtou(o->data, NULL, 10);
3496 if (errno || num < 0)
3497 return -1;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003498 o_reset_to_empty_unquoted(o);
Eric Andersen25f27032001-04-26 23:22:31 +00003499 return num;
3500}
3501
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003502#if BB_MMU
3503#define fetch_till_str(as_string, input, word, skip_tabs) \
3504 fetch_till_str(input, word, skip_tabs)
3505#endif
3506static char *fetch_till_str(o_string *as_string,
3507 struct in_str *input,
3508 const char *word,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003509 int heredoc_flags)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003510{
3511 o_string heredoc = NULL_O_STRING;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003512 unsigned past_EOL;
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003513 int prev = 0; /* not \ */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003514 int ch;
3515
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003516 goto jump_in;
Denys Vlasenkob8709032011-05-08 21:20:01 +02003517
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003518 while (1) {
3519 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003520 if (ch != EOF)
3521 nommu_addchr(as_string, ch);
3522 if ((ch == '\n' || ch == EOF)
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003523 && ((heredoc_flags & HEREDOC_QUOTED) || prev != '\\')
3524 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003525 if (strcmp(heredoc.data + past_EOL, word) == 0) {
3526 heredoc.data[past_EOL] = '\0';
3527 debug_printf_parse("parsed heredoc '%s'\n", heredoc.data);
3528 return heredoc.data;
3529 }
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003530 while (ch == '\n') {
3531 o_addchr(&heredoc, ch);
3532 prev = ch;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003533 jump_in:
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003534 past_EOL = heredoc.length;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003535 do {
3536 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003537 if (ch != EOF)
3538 nommu_addchr(as_string, ch);
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003539 } while ((heredoc_flags & HEREDOC_SKIPTABS) && ch == '\t');
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003540 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003541 }
3542 if (ch == EOF) {
3543 o_free_unsafe(&heredoc);
3544 return NULL;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003545 }
3546 o_addchr(&heredoc, ch);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003547 nommu_addchr(as_string, ch);
Denys Vlasenkoc3adfac2010-09-06 11:46:03 +02003548 if (prev == '\\' && ch == '\\')
3549 /* Correctly handle foo\\<eol> (not a line cont.) */
3550 prev = 0; /* not \ */
3551 else
3552 prev = ch;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003553 }
3554}
3555
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003556/* Look at entire parse tree for not-yet-loaded REDIRECT_HEREDOCs
3557 * and load them all. There should be exactly heredoc_cnt of them.
3558 */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003559static int fetch_heredocs(int heredoc_cnt, struct parse_context *ctx, struct in_str *input)
3560{
3561 struct pipe *pi = ctx->list_head;
3562
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003563 while (pi && heredoc_cnt) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003564 int i;
3565 struct command *cmd = pi->cmds;
3566
3567 debug_printf_parse("fetch_heredocs: num_cmds:%d cmd argv0:'%s'\n",
3568 pi->num_cmds,
3569 cmd->argv ? cmd->argv[0] : "NONE");
3570 for (i = 0; i < pi->num_cmds; i++) {
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003571 struct redir_struct *redir = cmd->redirects;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003572
3573 debug_printf_parse("fetch_heredocs: %d cmd argv0:'%s'\n",
3574 i, cmd->argv ? cmd->argv[0] : "NONE");
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003575 while (redir) {
3576 if (redir->rd_type == REDIRECT_HEREDOC) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003577 char *p;
3578
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003579 redir->rd_type = REDIRECT_HEREDOC2;
Denys Vlasenko764b2f02009-06-07 16:05:04 +02003580 /* redir->rd_dup is (ab)used to indicate <<- */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003581 p = fetch_till_str(&ctx->as_string, input,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003582 redir->rd_filename, redir->rd_dup);
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003583 if (!p) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003584 syntax_error("unexpected EOF in here document");
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003585 return 1;
3586 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003587 free(redir->rd_filename);
3588 redir->rd_filename = p;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003589 heredoc_cnt--;
3590 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003591 redir = redir->next;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003592 }
3593 cmd++;
3594 }
3595 pi = pi->next;
3596 }
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003597#if 0
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003598 /* Should be 0. If it isn't, it's a parse error */
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003599 if (heredoc_cnt)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003600 bb_error_msg_and_die("heredoc BUG 2");
3601#endif
3602 return 0;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003603}
3604
3605
Denys Vlasenkob36abf22010-09-05 14:50:59 +02003606static int run_list(struct pipe *pi);
3607#if BB_MMU
3608#define parse_stream(pstring, input, end_trigger) \
3609 parse_stream(input, end_trigger)
3610#endif
3611static struct pipe *parse_stream(char **pstring,
3612 struct in_str *input,
3613 int end_trigger);
Denis Vlasenkoba7cf262007-05-25 14:34:30 +00003614
Eric Andersen25f27032001-04-26 23:22:31 +00003615
Denys Vlasenkoc2704542009-11-20 19:14:19 +01003616#if !ENABLE_HUSH_FUNCTIONS
3617#define parse_group(dest, ctx, input, ch) \
3618 parse_group(ctx, input, ch)
3619#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003620static int parse_group(o_string *dest, struct parse_context *ctx,
Eric Andersen25f27032001-04-26 23:22:31 +00003621 struct in_str *input, int ch)
3622{
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003623 /* dest contains characters seen prior to ( or {.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00003624 * Typically it's empty, but for function defs,
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003625 * it contains function name (without '()'). */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003626 struct pipe *pipe_list;
Denis Vlasenko240c2552009-04-03 03:45:05 +00003627 int endch;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003628 struct command *command = ctx->command;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003629
3630 debug_printf_parse("parse_group entered\n");
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003631#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenko38292b62010-09-05 14:49:40 +02003632 if (ch == '(' && !dest->has_quoted_part) {
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003633 if (dest->length)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003634 if (done_word(dest, ctx))
3635 return 1;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003636 if (!command->argv)
3637 goto skip; /* (... */
3638 if (command->argv[1]) { /* word word ... (... */
3639 syntax_error_unexpected_ch('(');
3640 return 1;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003641 }
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003642 /* it is "word(..." or "word (..." */
3643 do
3644 ch = i_getch(input);
3645 while (ch == ' ' || ch == '\t');
3646 if (ch != ')') {
3647 syntax_error_unexpected_ch(ch);
3648 return 1;
3649 }
3650 nommu_addchr(&ctx->as_string, ch);
3651 do
3652 ch = i_getch(input);
3653 while (ch == ' ' || ch == '\t' || ch == '\n');
3654 if (ch != '{') {
3655 syntax_error_unexpected_ch(ch);
3656 return 1;
3657 }
3658 nommu_addchr(&ctx->as_string, ch);
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003659 command->cmd_type = CMD_FUNCDEF;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003660 goto skip;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003661 }
3662#endif
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003663
3664#if 0 /* Prevented by caller */
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003665 if (command->argv /* word [word]{... */
3666 || dest->length /* word{... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003667 || dest->has_quoted_part /* ""{... */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003668 ) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003669 syntax_error(NULL);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003670 debug_printf_parse("parse_group return 1: "
3671 "syntax error, groups and arglists don't mix\n");
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003672 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003673 }
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003674#endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003675
3676#if ENABLE_HUSH_FUNCTIONS
3677 skip:
3678#endif
Denis Vlasenko240c2552009-04-03 03:45:05 +00003679 endch = '}';
Denis Vlasenko90e485c2007-05-23 15:22:50 +00003680 if (ch == '(') {
Denis Vlasenko240c2552009-04-03 03:45:05 +00003681 endch = ')';
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003682 command->cmd_type = CMD_SUBSHELL;
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00003683 } else {
3684 /* bash does not allow "{echo...", requires whitespace */
3685 ch = i_getch(input);
3686 if (ch != ' ' && ch != '\t' && ch != '\n') {
3687 syntax_error_unexpected_ch(ch);
3688 return 1;
3689 }
3690 nommu_addchr(&ctx->as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003691 }
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003692
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003693 {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003694#if BB_MMU
3695# define as_string NULL
3696#else
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003697 char *as_string = NULL;
3698#endif
3699 pipe_list = parse_stream(&as_string, input, endch);
3700#if !BB_MMU
3701 if (as_string)
3702 o_addstr(&ctx->as_string, as_string);
3703#endif
3704 /* empty ()/{} or parse error? */
3705 if (!pipe_list || pipe_list == ERR_PTR) {
Denis Vlasenkobb929512009-04-16 10:59:40 +00003706 /* parse_stream already emitted error msg */
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003707 if (!BB_MMU)
3708 free(as_string);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003709 debug_printf_parse("parse_group return 1: "
3710 "parse_stream returned %p\n", pipe_list);
3711 return 1;
3712 }
3713 command->group = pipe_list;
3714#if !BB_MMU
3715 as_string[strlen(as_string) - 1] = '\0'; /* plink ')' or '}' */
3716 command->group_as_string = as_string;
3717 debug_printf_parse("end of group, remembering as:'%s'\n",
3718 command->group_as_string);
3719#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003720#undef as_string
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00003721 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003722 debug_printf_parse("parse_group return 0\n");
3723 return 0;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003724 /* command remains "open", available for possible redirects */
Eric Andersen25f27032001-04-26 23:22:31 +00003725}
3726
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003727#if ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003728/* Subroutines for copying $(...) and `...` things */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003729static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003730/* '...' */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003731static int add_till_single_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003732{
3733 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003734 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003735 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003736 syntax_error_unterm_ch('\'');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003737 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003738 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003739 if (ch == '\'')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003740 return 1;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003741 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003742 }
3743}
3744/* "...\"...`..`...." - do we need to handle "...$(..)..." too? */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003745static int add_till_double_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003746{
3747 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003748 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003749 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003750 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003751 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003752 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003753 if (ch == '"')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003754 return 1;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003755 if (ch == '\\') { /* \x. Copy both chars. */
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003756 o_addchr(dest, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003757 ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003758 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003759 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003760 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003761 if (!add_till_backquote(dest, input, /*in_dquote:*/ 1))
3762 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003763 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003764 continue;
3765 }
Denis Vlasenko5703c222008-06-15 11:49:42 +00003766 //if (ch == '$') ...
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003767 }
3768}
3769/* Process `cmd` - copy contents until "`" is seen. Complicated by
3770 * \` quoting.
3771 * "Within the backquoted style of command substitution, backslash
3772 * shall retain its literal meaning, except when followed by: '$', '`', or '\'.
3773 * The search for the matching backquote shall be satisfied by the first
3774 * backquote found without a preceding backslash; during this search,
3775 * if a non-escaped backquote is encountered within a shell comment,
3776 * a here-document, an embedded command substitution of the $(command)
3777 * form, or a quoted string, undefined results occur. A single-quoted
3778 * or double-quoted string that begins, but does not end, within the
3779 * "`...`" sequence produces undefined results."
3780 * Example Output
3781 * echo `echo '\'TEST\`echo ZZ\`BEST` \TESTZZBEST
3782 */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003783static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003784{
3785 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003786 int ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003787 if (ch == '`')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003788 return 1;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003789 if (ch == '\\') {
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003790 /* \x. Copy both unless it is \`, \$, \\ and maybe \" */
3791 ch = i_getch(input);
3792 if (ch != '`'
3793 && ch != '$'
3794 && ch != '\\'
3795 && (!in_dquote || ch != '"')
3796 ) {
3797 o_addchr(dest, '\\');
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003798 }
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003799 }
3800 if (ch == EOF) {
3801 syntax_error_unterm_ch('`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003802 return 0;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003803 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003804 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003805 }
3806}
3807/* Process $(cmd) - copy contents until ")" is seen. Complicated by
3808 * quoting and nested ()s.
3809 * "With the $(command) style of command substitution, all characters
3810 * following the open parenthesis to the matching closing parenthesis
3811 * constitute the command. Any valid shell script can be used for command,
3812 * except a script consisting solely of redirections which produces
3813 * unspecified results."
3814 * Example Output
3815 * echo $(echo '(TEST)' BEST) (TEST) BEST
3816 * echo $(echo 'TEST)' BEST) TEST) BEST
3817 * echo $(echo \(\(TEST\) BEST) ((TEST) BEST
Denys Vlasenko74369502010-05-21 19:52:01 +02003818 *
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003819 * Also adapted to eat ${var%...} and $((...)) constructs, since ... part
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003820 * can contain arbitrary constructs, just like $(cmd).
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003821 * In bash compat mode, it needs to also be able to stop on ':' or '/'
3822 * for ${var:N[:M]} and ${var/P[/R]} parsing.
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003823 */
Denys Vlasenko74369502010-05-21 19:52:01 +02003824#define DOUBLE_CLOSE_CHAR_FLAG 0x80
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003825static int add_till_closing_bracket(o_string *dest, struct in_str *input, unsigned end_ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003826{
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003827 int ch;
Denys Vlasenko74369502010-05-21 19:52:01 +02003828 char dbl = end_ch & DOUBLE_CLOSE_CHAR_FLAG;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003829# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003830 char end_char2 = end_ch >> 8;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003831# endif
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003832 end_ch &= (DOUBLE_CLOSE_CHAR_FLAG - 1);
3833
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003834 while (1) {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003835 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003836 if (ch == EOF) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003837 syntax_error_unterm_ch(end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003838 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003839 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003840 if (ch == end_ch IF_HUSH_BASH_COMPAT( || ch == end_char2)) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003841 if (!dbl)
3842 break;
3843 /* we look for closing )) of $((EXPR)) */
3844 if (i_peek(input) == end_ch) {
3845 i_getch(input); /* eat second ')' */
3846 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003847 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003848 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003849 o_addchr(dest, ch);
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003850 if (ch == '(' || ch == '{') {
3851 ch = (ch == '(' ? ')' : '}');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003852 if (!add_till_closing_bracket(dest, input, ch))
3853 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003854 o_addchr(dest, ch);
3855 continue;
3856 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003857 if (ch == '\'') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003858 if (!add_till_single_quote(dest, input))
3859 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003860 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003861 continue;
3862 }
3863 if (ch == '"') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003864 if (!add_till_double_quote(dest, input))
3865 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003866 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003867 continue;
3868 }
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003869 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003870 if (!add_till_backquote(dest, input, /*in_dquote:*/ 0))
3871 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003872 o_addchr(dest, ch);
3873 continue;
3874 }
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003875 if (ch == '\\') {
3876 /* \x. Copy verbatim. Important for \(, \) */
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003877 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003878 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003879 syntax_error_unterm_ch(')');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003880 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003881 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003882 o_addchr(dest, ch);
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003883 continue;
3884 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003885 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003886 return ch;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003887}
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003888#endif /* ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS */
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003889
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00003890/* Return code: 0 for OK, 1 for syntax error */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003891#if BB_MMU
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003892#define parse_dollar(as_string, dest, input, quote_mask) \
3893 parse_dollar(dest, input, quote_mask)
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003894#define as_string NULL
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003895#endif
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003896static int parse_dollar(o_string *as_string,
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003897 o_string *dest,
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003898 struct in_str *input, unsigned char quote_mask)
Eric Andersen25f27032001-04-26 23:22:31 +00003899{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003900 int ch = i_peek(input); /* first character after the $ */
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003901
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003902 debug_printf_parse("parse_dollar entered: ch='%c'\n", ch);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003903 if (isalpha(ch)) {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003904 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003905 nommu_addchr(as_string, ch);
Denis Vlasenkod4981312008-07-31 10:34:48 +00003906 make_var:
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003907 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003908 while (1) {
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003909 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003910 o_addchr(dest, ch | quote_mask);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003911 quote_mask = 0;
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003912 ch = i_peek(input);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003913 if (!isalnum(ch) && ch != '_')
3914 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003915 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003916 nommu_addchr(as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003917 }
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003918 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003919 } else if (isdigit(ch)) {
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003920 make_one_char_var:
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003921 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003922 nommu_addchr(as_string, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003923 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003924 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003925 o_addchr(dest, ch | quote_mask);
3926 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003927 } else switch (ch) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003928 case '$': /* pid */
3929 case '!': /* last bg pid */
3930 case '?': /* last exit code */
3931 case '#': /* number of args */
3932 case '*': /* args */
3933 case '@': /* args */
3934 goto make_one_char_var;
3935 case '{': {
Mike Frysingeref3e7fd2009-06-01 14:13:39 -04003936 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3937
Denys Vlasenko74369502010-05-21 19:52:01 +02003938 ch = i_getch(input); /* eat '{' */
3939 nommu_addchr(as_string, ch);
3940
3941 ch = i_getch(input); /* first char after '{' */
Denys Vlasenko74369502010-05-21 19:52:01 +02003942 /* It should be ${?}, or ${#var},
3943 * or even ${?+subst} - operator acting on a special variable,
3944 * or the beginning of variable name.
3945 */
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003946 if (ch == EOF
3947 || (!strchr(_SPECIAL_VARS_STR, ch) && !isalnum(ch)) /* not one of those */
3948 ) {
Denys Vlasenko74369502010-05-21 19:52:01 +02003949 bad_dollar_syntax:
3950 syntax_error_unterm_str("${name}");
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003951 debug_printf_parse("parse_dollar return 0: unterminated ${name}\n");
3952 return 0;
Denys Vlasenko74369502010-05-21 19:52:01 +02003953 }
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003954 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003955 ch |= quote_mask;
3956
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003957 /* It's possible to just call add_till_closing_bracket() at this point.
Denys Vlasenko74369502010-05-21 19:52:01 +02003958 * However, this regresses some of our testsuite cases
3959 * which check invalid constructs like ${%}.
3960 * Oh well... let's check that the var name part is fine... */
3961
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003962 while (1) {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003963 unsigned pos;
3964
Denys Vlasenko74369502010-05-21 19:52:01 +02003965 o_addchr(dest, ch);
3966 debug_printf_parse(": '%c'\n", ch);
3967
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003968 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003969 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003970 if (ch == '}')
Mike Frysinger98c52642009-04-02 10:02:37 +00003971 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003972
Denys Vlasenko74369502010-05-21 19:52:01 +02003973 if (!isalnum(ch) && ch != '_') {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003974 unsigned end_ch;
3975 unsigned char last_ch;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003976 /* handle parameter expansions
3977 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
3978 */
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003979 if (!strchr(VAR_SUBST_OPS, ch)) /* ${var<bad_char>... */
Denys Vlasenko74369502010-05-21 19:52:01 +02003980 goto bad_dollar_syntax;
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003981
3982 /* Eat everything until closing '}' (or ':') */
3983 end_ch = '}';
3984 if (ENABLE_HUSH_BASH_COMPAT
3985 && ch == ':'
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003986 && !strchr(MINUS_PLUS_EQUAL_QUESTION, i_peek(input))
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003987 ) {
3988 /* It's ${var:N[:M]} thing */
3989 end_ch = '}' * 0x100 + ':';
3990 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003991 if (ENABLE_HUSH_BASH_COMPAT
3992 && ch == '/'
3993 ) {
3994 /* It's ${var/[/]pattern[/repl]} thing */
3995 if (i_peek(input) == '/') { /* ${var//pattern[/repl]}? */
3996 i_getch(input);
3997 nommu_addchr(as_string, '/');
3998 ch = '\\';
3999 }
4000 end_ch = '}' * 0x100 + '/';
4001 }
4002 o_addchr(dest, ch);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004003 again:
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004004 if (!BB_MMU)
4005 pos = dest->length;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02004006#if ENABLE_HUSH_DOLLAR_OPS
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004007 last_ch = add_till_closing_bracket(dest, input, end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004008 if (last_ch == 0) /* error? */
4009 return 0;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02004010#else
4011#error Simple code to only allow ${var} is not implemented
4012#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004013 if (as_string) {
4014 o_addstr(as_string, dest->data + pos);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004015 o_addchr(as_string, last_ch);
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004016 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004017
4018 if (ENABLE_HUSH_BASH_COMPAT && (end_ch & 0xff00)) {
4019 /* close the first block: */
4020 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denys Vlasenko36f774a2010-09-05 14:45:38 +02004021 /* while parsing N from ${var:N[:M]}
4022 * or pattern from ${var/[/]pattern[/repl]} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004023 if ((end_ch & 0xff) == last_ch) {
Denys Vlasenko36f774a2010-09-05 14:45:38 +02004024 /* got ':' or '/'- parse the rest */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004025 end_ch = '}';
4026 goto again;
4027 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02004028 /* got '}' */
4029 if (end_ch == '}' * 0x100 + ':') {
4030 /* it's ${var:N} - emulate :999999999 */
4031 o_addstr(dest, "999999999");
4032 } /* else: it's ${var/[/]pattern} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004033 }
Denys Vlasenko74369502010-05-21 19:52:01 +02004034 break;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004035 }
Denys Vlasenko74369502010-05-21 19:52:01 +02004036 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004037 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4038 break;
4039 }
Denys Vlasenkoc0836532009-10-19 13:13:06 +02004040#if ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004041 case '(': {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004042 unsigned pos;
4043
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004044 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004045 nommu_addchr(as_string, ch);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004046# if ENABLE_SH_MATH_SUPPORT
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004047 if (i_peek(input) == '(') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004048 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004049 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004050 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4051 o_addchr(dest, /*quote_mask |*/ '+');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004052 if (!BB_MMU)
4053 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004054 if (!add_till_closing_bracket(dest, input, ')' | DOUBLE_CLOSE_CHAR_FLAG))
4055 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004056 if (as_string) {
4057 o_addstr(as_string, dest->data + pos);
4058 o_addchr(as_string, ')');
4059 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004060 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004061 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00004062 break;
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00004063 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004064# endif
4065# if ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004066 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4067 o_addchr(dest, quote_mask | '`');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004068 if (!BB_MMU)
4069 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004070 if (!add_till_closing_bracket(dest, input, ')'))
4071 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004072 if (as_string) {
4073 o_addstr(as_string, dest->data + pos);
Denys Vlasenkob70cef72010-01-12 13:45:45 +01004074 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004075 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004076 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004077# endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004078 break;
4079 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004080#endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004081 case '_':
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004082 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004083 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004084 ch = i_peek(input);
4085 if (isalnum(ch)) { /* it's $_name or $_123 */
4086 ch = '_';
4087 goto make_var;
4088 }
4089 /* else: it's $_ */
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02004090 /* TODO: $_ and $-: */
4091 /* $_ Shell or shell script name; or last argument of last command
4092 * (if last command wasn't a pipe; if it was, bash sets $_ to "");
4093 * but in command's env, set to full pathname used to invoke it */
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004094 /* $- Option flags set by set builtin or shell options (-i etc) */
4095 default:
4096 o_addQchr(dest, '$');
Eric Andersen25f27032001-04-26 23:22:31 +00004097 }
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004098 debug_printf_parse("parse_dollar return 1 (ok)\n");
4099 return 1;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004100#undef as_string
Eric Andersen25f27032001-04-26 23:22:31 +00004101}
4102
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004103#if BB_MMU
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004104# if ENABLE_HUSH_BASH_COMPAT
4105#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4106 encode_string(dest, input, dquote_end, process_bkslash)
4107# else
4108/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4109#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4110 encode_string(dest, input, dquote_end)
4111# endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004112#define as_string NULL
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004113
4114#else /* !MMU */
4115
4116# if ENABLE_HUSH_BASH_COMPAT
4117/* all parameters are needed, no macro tricks */
4118# else
4119#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4120 encode_string(as_string, dest, input, dquote_end)
4121# endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004122#endif
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004123static int encode_string(o_string *as_string,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004124 o_string *dest,
4125 struct in_str *input,
Denys Vlasenko14e289b2010-09-10 10:15:18 +02004126 int dquote_end,
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004127 int process_bkslash)
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004128{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004129#if !ENABLE_HUSH_BASH_COMPAT
4130 const int process_bkslash = 1;
4131#endif
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004132 int ch;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004133 int next;
4134
4135 again:
4136 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004137 if (ch != EOF)
4138 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004139 if (ch == dquote_end) { /* may be only '"' or EOF */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004140 debug_printf_parse("encode_string return 1 (ok)\n");
4141 return 1;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004142 }
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004143 /* note: can't move it above ch == dquote_end check! */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004144 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004145 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004146 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004147 }
4148 next = '\0';
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004149 if (ch != '\n') {
4150 next = i_peek(input);
4151 }
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004152 debug_printf_parse("\" ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004153 ch, ch, !!(dest->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004154 if (process_bkslash && ch == '\\') {
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004155 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004156 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004157 xfunc_die();
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004158 }
4159 /* bash:
4160 * "The backslash retains its special meaning [in "..."]
4161 * only when followed by one of the following characters:
4162 * $, `, ", \, or <newline>. A double quote may be quoted
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02004163 * within double quotes by preceding it with a backslash."
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004164 * NB: in (unquoted) heredoc, above does not apply to ",
4165 * therefore we check for it by "next == dquote_end" cond.
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004166 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004167 if (next == dquote_end || strchr("$`\\\n", next)) {
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004168 ch = i_getch(input); /* eat next */
4169 if (ch == '\n')
4170 goto again; /* skip \<newline> */
Denys Vlasenko4f870492010-09-10 11:06:01 +02004171 } /* else: ch remains == '\\', and we double it below: */
4172 o_addqchr(dest, ch); /* \c if c is a glob char, else just c */
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004173 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004174 goto again;
4175 }
4176 if (ch == '$') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004177 if (!parse_dollar(as_string, dest, input, /*quote_mask:*/ 0x80)) {
4178 debug_printf_parse("encode_string return 0: "
4179 "parse_dollar returned 0 (error)\n");
4180 return 0;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004181 }
4182 goto again;
4183 }
4184#if ENABLE_HUSH_TICK
4185 if (ch == '`') {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004186 //unsigned pos = dest->length;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004187 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4188 o_addchr(dest, 0x80 | '`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004189 if (!add_till_backquote(dest, input, /*in_dquote:*/ dquote_end == '"'))
4190 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004191 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4192 //debug_printf_subst("SUBST RES3 '%s'\n", dest->data + pos);
Denis Vlasenkof328e002009-04-02 16:55:38 +00004193 goto again;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004194 }
4195#endif
Denis Vlasenkof328e002009-04-02 16:55:38 +00004196 o_addQchr(dest, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004197 goto again;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004198#undef as_string
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004199}
4200
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004201/*
4202 * Scan input until EOF or end_trigger char.
4203 * Return a list of pipes to execute, or NULL on EOF
4204 * or if end_trigger character is met.
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004205 * On syntax error, exit if shell is not interactive,
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004206 * reset parsing machinery and start parsing anew,
4207 * or return ERR_PTR.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004208 */
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004209static struct pipe *parse_stream(char **pstring,
4210 struct in_str *input,
4211 int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00004212{
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004213 struct parse_context ctx;
4214 o_string dest = NULL_O_STRING;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004215 int heredoc_cnt;
Eric Andersen25f27032001-04-26 23:22:31 +00004216
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004217 /* Single-quote triggers a bypass of the main loop until its mate is
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004218 * found. When recursing, quote state is passed in via dest->o_expflags.
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004219 */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004220 debug_printf_parse("parse_stream entered, end_trigger='%c'\n",
Denys Vlasenko90a99042009-09-06 02:36:23 +02004221 end_trigger ? end_trigger : 'X');
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004222 debug_enter();
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004223
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004224 /* If very first arg is "" or '', dest.data may end up NULL.
4225 * Preventing this: */
4226 o_addchr(&dest, '\0');
4227 dest.length = 0;
4228
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004229 /* We used to separate words on $IFS here. This was wrong.
4230 * $IFS is used only for word splitting when $var is expanded,
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004231 * here we should use blank chars as separators, not $IFS
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004232 */
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004233
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004234 if (MAYBE_ASSIGNMENT != 0)
4235 dest.o_assignment = MAYBE_ASSIGNMENT;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004236 initialize_context(&ctx);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004237 heredoc_cnt = 0;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004238 while (1) {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004239 const char *is_blank;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004240 const char *is_special;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004241 int ch;
4242 int next;
4243 int redir_fd;
4244 redir_type redir_style;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004245
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00004246 ch = i_getch(input);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004247 debug_printf_parse(": ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004248 ch, ch, !!(dest.o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004249 if (ch == EOF) {
4250 struct pipe *pi;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004251
4252 if (heredoc_cnt) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004253 syntax_error_unterm_str("here document");
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004254 goto parse_error;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004255 }
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004256 /* end_trigger == '}' case errors out earlier,
4257 * checking only ')' */
4258 if (end_trigger == ')') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004259 syntax_error_unterm_ch('(');
4260 goto parse_error;
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004261 }
4262
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004263 if (done_word(&dest, &ctx)) {
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004264 goto parse_error;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004265 }
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004266 o_free(&dest);
4267 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004268 pi = ctx.list_head;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004269 /* If we got nothing... */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004270 /* (this makes bare "&" cmd a no-op.
4271 * bash says: "syntax error near unexpected token '&'") */
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004272 if (pi->num_cmds == 0
Denys Vlasenko60cb48c2013-01-14 15:57:44 +01004273 IF_HAS_KEYWORDS(&& pi->res_word == RES_NONE)
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004274 ) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004275 free_pipe_list(pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004276 pi = NULL;
4277 }
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004278#if !BB_MMU
Denys Vlasenkob5be13c2015-09-04 06:22:10 +02004279 debug_printf_parse("as_string1 '%s'\n", ctx.as_string.data);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004280 if (pstring)
4281 *pstring = ctx.as_string.data;
4282 else
4283 o_free_unsafe(&ctx.as_string);
4284#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004285 debug_leave();
4286 debug_printf_parse("parse_stream return %p\n", pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004287 return pi;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004288 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004289 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004290
4291 next = '\0';
4292 if (ch != '\n')
4293 next = i_peek(input);
4294
4295 is_special = "{}<>;&|()#'" /* special outside of "str" */
4296 "\\$\"" IF_HUSH_TICK("`"); /* always special */
4297 /* Are { and } special here? */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004298 if (ctx.command->argv /* word [word]{... - non-special */
4299 || dest.length /* word{... - non-special */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004300 || dest.has_quoted_part /* ""{... - non-special */
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004301 || (next != ';' /* }; - special */
4302 && next != ')' /* }) - special */
4303 && next != '&' /* }& and }&& ... - special */
4304 && next != '|' /* }|| ... - special */
4305 && !strchr(defifs, next) /* {word - non-special */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004306 )
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004307 ) {
4308 /* They are not special, skip "{}" */
4309 is_special += 2;
4310 }
4311 is_special = strchr(is_special, ch);
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004312 is_blank = strchr(defifs, ch);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004313
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004314 if (!is_special && !is_blank) { /* ordinary char */
Denis Vlasenkobf25fbc2009-04-19 13:57:51 +00004315 ordinary_char:
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004316 o_addQchr(&dest, ch);
4317 if ((dest.o_assignment == MAYBE_ASSIGNMENT
4318 || dest.o_assignment == WORD_IS_KEYWORD)
Denis Vlasenko55789c62008-06-18 16:30:42 +00004319 && ch == '='
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004320 && is_well_formed_var_name(dest.data, '=')
Denis Vlasenko55789c62008-06-18 16:30:42 +00004321 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004322 dest.o_assignment = DEFINITELY_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004323 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko55789c62008-06-18 16:30:42 +00004324 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004325 continue;
4326 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004327
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004328 if (is_blank) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004329 if (done_word(&dest, &ctx)) {
4330 goto parse_error;
Eric Andersenaac75e52001-04-30 18:18:45 +00004331 }
Denis Vlasenko37181682009-04-03 03:19:15 +00004332 if (ch == '\n') {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004333 /* Is this a case when newline is simply ignored?
4334 * Some examples:
4335 * "cmd | <newline> cmd ..."
4336 * "case ... in <newline> word) ..."
4337 */
4338 if (IS_NULL_CMD(ctx.command)
4339 && dest.length == 0 && !dest.has_quoted_part
Denis Vlasenkof1736072008-07-31 10:09:26 +00004340 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004341 /* This newline can be ignored. But...
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004342 * Without check #1, interactive shell
4343 * ignores even bare <newline>,
4344 * and shows the continuation prompt:
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004345 * ps1_prompt$ <enter>
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004346 * ps2> _ <=== wrong, should be ps1
4347 * Without check #2, "cmd & <newline>"
4348 * is similarly mistreated.
4349 * (BTW, this makes "cmd & cmd"
4350 * and "cmd && cmd" non-orthogonal.
4351 * Really, ask yourself, why
4352 * "cmd && <newline>" doesn't start
4353 * cmd but waits for more input?
4354 * No reason...)
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004355 */
4356 struct pipe *pi = ctx.list_head;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004357 if (pi->num_cmds != 0 /* check #1 */
4358 && pi->followup != PIPE_BG /* check #2 */
4359 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004360 continue;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004361 }
Denis Vlasenkof1736072008-07-31 10:09:26 +00004362 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004363 /* Treat newline as a command separator. */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004364 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004365 debug_printf_parse("heredoc_cnt:%d\n", heredoc_cnt);
4366 if (heredoc_cnt) {
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004367 if (fetch_heredocs(heredoc_cnt, &ctx, input)) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004368 goto parse_error;
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004369 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004370 heredoc_cnt = 0;
4371 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004372 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004373 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004374 ch = ';';
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004375 /* note: if (is_blank) continue;
Denis Vlasenko240c2552009-04-03 03:45:05 +00004376 * will still trigger for us */
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004377 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004378 }
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004379
4380 /* "cmd}" or "cmd }..." without semicolon or &:
4381 * } is an ordinary char in this case, even inside { cmd; }
4382 * Pathological example: { ""}; } should exec "}" cmd
4383 */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004384 if (ch == '}') {
4385 if (!IS_NULL_CMD(ctx.command) /* cmd } */
4386 || dest.length != 0 /* word} */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004387 || dest.has_quoted_part /* ""} */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004388 ) {
4389 goto ordinary_char;
4390 }
4391 if (!IS_NULL_PIPE(ctx.pipe)) /* cmd | } */
4392 goto skip_end_trigger;
4393 /* else: } does terminate a group */
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004394 }
4395
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004396 if (end_trigger && end_trigger == ch
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004397 && (ch != ';' || heredoc_cnt == 0)
4398#if ENABLE_HUSH_CASE
4399 && (ch != ')'
4400 || ctx.ctx_res_w != RES_MATCH
Denys Vlasenko38292b62010-09-05 14:49:40 +02004401 || (!dest.has_quoted_part && strcmp(dest.data, "esac") == 0)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004402 )
4403#endif
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004404 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004405 if (heredoc_cnt) {
4406 /* This is technically valid:
4407 * { cat <<HERE; }; echo Ok
4408 * heredoc
4409 * heredoc
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004410 * HERE
4411 * but we don't support this.
4412 * We require heredoc to be in enclosing {}/(),
4413 * if any.
4414 */
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004415 syntax_error_unterm_str("here document");
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004416 goto parse_error;
4417 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004418 if (done_word(&dest, &ctx)) {
4419 goto parse_error;
4420 }
4421 done_pipe(&ctx, PIPE_SEQ);
4422 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004423 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004424 /* Do we sit outside of any if's, loops or case's? */
Denis Vlasenko37181682009-04-03 03:19:15 +00004425 if (!HAS_KEYWORDS
Denys Vlasenko60cb48c2013-01-14 15:57:44 +01004426 IF_HAS_KEYWORDS(|| (ctx.ctx_res_w == RES_NONE && ctx.old_flag == 0))
Denis Vlasenko37181682009-04-03 03:19:15 +00004427 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004428 o_free(&dest);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004429#if !BB_MMU
Denys Vlasenkob5be13c2015-09-04 06:22:10 +02004430 debug_printf_parse("as_string2 '%s'\n", ctx.as_string.data);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004431 if (pstring)
4432 *pstring = ctx.as_string.data;
4433 else
4434 o_free_unsafe(&ctx.as_string);
4435#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004436 debug_leave();
4437 debug_printf_parse("parse_stream return %p: "
4438 "end_trigger char found\n",
4439 ctx.list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004440 return ctx.list_head;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004441 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004442 }
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004443 skip_end_trigger:
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004444 if (is_blank)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004445 continue;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004446
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004447 /* Catch <, > before deciding whether this word is
4448 * an assignment. a=1 2>z b=2: b=2 is still assignment */
4449 switch (ch) {
4450 case '>':
4451 redir_fd = redirect_opt_num(&dest);
4452 if (done_word(&dest, &ctx)) {
4453 goto parse_error;
4454 }
4455 redir_style = REDIRECT_OVERWRITE;
4456 if (next == '>') {
4457 redir_style = REDIRECT_APPEND;
4458 ch = i_getch(input);
4459 nommu_addchr(&ctx.as_string, ch);
4460 }
4461#if 0
4462 else if (next == '(') {
4463 syntax_error(">(process) not supported");
4464 goto parse_error;
4465 }
4466#endif
4467 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4468 goto parse_error;
4469 continue; /* back to top of while (1) */
4470 case '<':
4471 redir_fd = redirect_opt_num(&dest);
4472 if (done_word(&dest, &ctx)) {
4473 goto parse_error;
4474 }
4475 redir_style = REDIRECT_INPUT;
4476 if (next == '<') {
4477 redir_style = REDIRECT_HEREDOC;
4478 heredoc_cnt++;
4479 debug_printf_parse("++heredoc_cnt=%d\n", heredoc_cnt);
4480 ch = i_getch(input);
4481 nommu_addchr(&ctx.as_string, ch);
4482 } else if (next == '>') {
4483 redir_style = REDIRECT_IO;
4484 ch = i_getch(input);
4485 nommu_addchr(&ctx.as_string, ch);
4486 }
4487#if 0
4488 else if (next == '(') {
4489 syntax_error("<(process) not supported");
4490 goto parse_error;
4491 }
4492#endif
4493 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4494 goto parse_error;
4495 continue; /* back to top of while (1) */
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004496 case '#':
4497 if (dest.length == 0 && !dest.has_quoted_part) {
4498 /* skip "#comment" */
4499 while (1) {
4500 ch = i_peek(input);
4501 if (ch == EOF || ch == '\n')
4502 break;
4503 i_getch(input);
4504 /* note: we do not add it to &ctx.as_string */
4505 }
4506 nommu_addchr(&ctx.as_string, '\n');
4507 continue; /* back to top of while (1) */
4508 }
4509 break;
4510 case '\\':
4511 if (next == '\n') {
4512 /* It's "\<newline>" */
4513#if !BB_MMU
4514 /* Remove trailing '\' from ctx.as_string */
4515 ctx.as_string.data[--ctx.as_string.length] = '\0';
4516#endif
4517 ch = i_getch(input); /* eat it */
4518 continue; /* back to top of while (1) */
4519 }
4520 break;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004521 }
4522
4523 if (dest.o_assignment == MAYBE_ASSIGNMENT
4524 /* check that we are not in word in "a=1 2>word b=1": */
4525 && !ctx.pending_redirect
4526 ) {
4527 /* ch is a special char and thus this word
4528 * cannot be an assignment */
4529 dest.o_assignment = NOT_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004530 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004531 }
4532
Denys Vlasenkocbfe6ad2009-08-12 19:47:44 +02004533 /* Note: nommu_addchr(&ctx.as_string, ch) is already done */
4534
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004535 switch (ch) {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004536 case '#': /* non-comment #: "echo a#b" etc */
4537 o_addQchr(&dest, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004538 break;
4539 case '\\':
4540 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004541 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004542 xfunc_die();
Eric Andersen25f27032001-04-26 23:22:31 +00004543 }
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004544 ch = i_getch(input);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004545 /* note: ch != '\n' (that case does not reach this place) */
4546 o_addchr(&dest, '\\');
4547 /*nommu_addchr(&ctx.as_string, '\\'); - already done */
4548 o_addchr(&dest, ch);
4549 nommu_addchr(&ctx.as_string, ch);
4550 /* Example: echo Hello \2>file
4551 * we need to know that word 2 is quoted */
4552 dest.has_quoted_part = 1;
Eric Andersen25f27032001-04-26 23:22:31 +00004553 break;
4554 case '$':
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004555 if (!parse_dollar(&ctx.as_string, &dest, input, /*quote_mask:*/ 0)) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004556 debug_printf_parse("parse_stream parse error: "
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004557 "parse_dollar returned 0 (error)\n");
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004558 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004559 }
Eric Andersen25f27032001-04-26 23:22:31 +00004560 break;
4561 case '\'':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004562 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004563 if (next == '\'' && !ctx.pending_redirect) {
4564 insert_empty_quoted_str_marker:
4565 nommu_addchr(&ctx.as_string, next);
4566 i_getch(input); /* eat second ' */
4567 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4568 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4569 } else {
4570 while (1) {
4571 ch = i_getch(input);
4572 if (ch == EOF) {
4573 syntax_error_unterm_ch('\'');
4574 goto parse_error;
4575 }
4576 nommu_addchr(&ctx.as_string, ch);
4577 if (ch == '\'')
4578 break;
4579 o_addqchr(&dest, ch);
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004580 }
Eric Andersen25f27032001-04-26 23:22:31 +00004581 }
Eric Andersen25f27032001-04-26 23:22:31 +00004582 break;
4583 case '"':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004584 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004585 if (next == '"' && !ctx.pending_redirect)
4586 goto insert_empty_quoted_str_marker;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004587 if (dest.o_assignment == NOT_ASSIGNMENT)
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004588 dest.o_expflags |= EXP_FLAG_ESC_GLOB_CHARS;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004589 if (!encode_string(&ctx.as_string, &dest, input, '"', /*process_bkslash:*/ 1))
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004590 goto parse_error;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004591 dest.o_expflags &= ~EXP_FLAG_ESC_GLOB_CHARS;
Eric Andersen25f27032001-04-26 23:22:31 +00004592 break;
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004593#if ENABLE_HUSH_TICK
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004594 case '`': {
Denys Vlasenko60a94142011-05-13 20:57:01 +02004595 USE_FOR_NOMMU(unsigned pos;)
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004596
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004597 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4598 o_addchr(&dest, '`');
Denys Vlasenko60a94142011-05-13 20:57:01 +02004599 USE_FOR_NOMMU(pos = dest.length;)
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004600 if (!add_till_backquote(&dest, input, /*in_dquote:*/ 0))
4601 goto parse_error;
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004602# if !BB_MMU
Denis Vlasenko5c090a92009-04-08 21:51:33 +00004603 o_addstr(&ctx.as_string, dest.data + pos);
4604 o_addchr(&ctx.as_string, '`');
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004605# endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004606 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4607 //debug_printf_subst("SUBST RES3 '%s'\n", dest.data + pos);
Eric Andersen25f27032001-04-26 23:22:31 +00004608 break;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004609 }
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004610#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004611 case ';':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004612#if ENABLE_HUSH_CASE
4613 case_semi:
4614#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004615 if (done_word(&dest, &ctx)) {
4616 goto parse_error;
4617 }
4618 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004619#if ENABLE_HUSH_CASE
4620 /* Eat multiple semicolons, detect
4621 * whether it means something special */
4622 while (1) {
4623 ch = i_peek(input);
4624 if (ch != ';')
4625 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004626 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004627 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004628 if (ctx.ctx_res_w == RES_CASE_BODY) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004629 ctx.ctx_dsemicolon = 1;
4630 ctx.ctx_res_w = RES_MATCH;
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004631 break;
4632 }
4633 }
4634#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004635 new_cmd:
4636 /* We just finished a cmd. New one may start
4637 * with an assignment */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004638 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004639 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Eric Andersen25f27032001-04-26 23:22:31 +00004640 break;
4641 case '&':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004642 if (done_word(&dest, &ctx)) {
4643 goto parse_error;
4644 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004645 if (next == '&') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004646 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004647 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004648 done_pipe(&ctx, PIPE_AND);
Eric Andersen25f27032001-04-26 23:22:31 +00004649 } else {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004650 done_pipe(&ctx, PIPE_BG);
Eric Andersen25f27032001-04-26 23:22:31 +00004651 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004652 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004653 case '|':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004654 if (done_word(&dest, &ctx)) {
4655 goto parse_error;
4656 }
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004657#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004658 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenkof1736072008-07-31 10:09:26 +00004659 break; /* we are in case's "word | word)" */
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004660#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004661 if (next == '|') { /* || */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004662 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004663 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004664 done_pipe(&ctx, PIPE_OR);
Eric Andersen25f27032001-04-26 23:22:31 +00004665 } else {
4666 /* we could pick up a file descriptor choice here
4667 * with redirect_opt_num(), but bash doesn't do it.
4668 * "echo foo 2| cat" yields "foo 2". */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004669 done_command(&ctx);
Eric Andersen25f27032001-04-26 23:22:31 +00004670 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004671 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004672 case '(':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004673#if ENABLE_HUSH_CASE
Denis Vlasenkof1736072008-07-31 10:09:26 +00004674 /* "case... in [(]word)..." - skip '(' */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004675 if (ctx.ctx_res_w == RES_MATCH
4676 && ctx.command->argv == NULL /* not (word|(... */
4677 && dest.length == 0 /* not word(... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004678 && dest.has_quoted_part == 0 /* not ""(... */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004679 ) {
4680 continue;
4681 }
4682#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004683 case '{':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004684 if (parse_group(&dest, &ctx, input, ch) != 0) {
4685 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004686 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004687 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004688 case ')':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004689#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004690 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004691 goto case_semi;
4692#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004693 case '}':
Denis Vlasenkoc3735272008-10-09 12:58:26 +00004694 /* proper use of this character is caught by end_trigger:
4695 * if we see {, we call parse_group(..., end_trigger='}')
4696 * and it will match } earlier (not here). */
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00004697 syntax_error_unexpected_ch(ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004698 goto parse_error;
Eric Andersen25f27032001-04-26 23:22:31 +00004699 default:
Denis Vlasenko5ec61322008-06-24 00:50:07 +00004700 if (HUSH_DEBUG)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00004701 bb_error_msg_and_die("BUG: unexpected %c\n", ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004702 }
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004703 } /* while (1) */
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004704
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004705 parse_error:
4706 {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004707 struct parse_context *pctx;
4708 IF_HAS_KEYWORDS(struct parse_context *p2;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004709
4710 /* Clean up allocated tree.
Denys Vlasenko764b2f02009-06-07 16:05:04 +02004711 * Sample for finding leaks on syntax error recovery path.
4712 * Run it from interactive shell, watch pmap `pidof hush`.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004713 * while if false; then false; fi; do break; fi
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00004714 * Samples to catch leaks at execution:
4715 * while if (true | {true;}); then echo ok; fi; do break; done
4716 * 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 +00004717 */
4718 pctx = &ctx;
4719 do {
4720 /* Update pipe/command counts,
4721 * otherwise freeing may miss some */
4722 done_pipe(pctx, PIPE_SEQ);
4723 debug_printf_clean("freeing list %p from ctx %p\n",
4724 pctx->list_head, pctx);
4725 debug_print_tree(pctx->list_head, 0);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004726 free_pipe_list(pctx->list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004727 debug_printf_clean("freed list %p\n", pctx->list_head);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004728#if !BB_MMU
4729 o_free_unsafe(&pctx->as_string);
4730#endif
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004731 IF_HAS_KEYWORDS(p2 = pctx->stack;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004732 if (pctx != &ctx) {
4733 free(pctx);
4734 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004735 IF_HAS_KEYWORDS(pctx = p2;)
4736 } while (HAS_KEYWORDS && pctx);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004737
Denys Vlasenkoa439fa92011-03-30 19:11:46 +02004738 o_free(&dest);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004739 G.last_exitcode = 1;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004740#if !BB_MMU
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004741 if (pstring)
4742 *pstring = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004743#endif
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004744 debug_leave();
4745 return ERR_PTR;
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004746 }
Eric Andersen25f27032001-04-26 23:22:31 +00004747}
4748
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004749
4750/*** Execution routines ***/
4751
4752/* Expansion can recurse, need forward decls: */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004753#if !ENABLE_HUSH_BASH_COMPAT
4754/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4755#define expand_string_to_string(str, do_unbackslash) \
4756 expand_string_to_string(str)
4757#endif
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004758static char *expand_string_to_string(const char *str, int do_unbackslash);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004759#if ENABLE_HUSH_TICK
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004760static int process_command_subs(o_string *dest, const char *s);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004761#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004762
4763/* expand_strvec_to_strvec() takes a list of strings, expands
4764 * all variable references within and returns a pointer to
4765 * a list of expanded strings, possibly with larger number
4766 * of strings. (Think VAR="a b"; echo $VAR).
4767 * This new list is allocated as a single malloc block.
4768 * NULL-terminated list of char* pointers is at the beginning of it,
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004769 * followed by strings themselves.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004770 * Caller can deallocate entire list by single free(list). */
4771
Denys Vlasenko238081f2010-10-03 14:26:26 +02004772/* A horde of its helpers come first: */
4773
4774static void o_addblock_duplicate_backslash(o_string *o, const char *str, int len)
4775{
4776 while (--len >= 0) {
Denys Vlasenko9e800222010-10-03 14:28:04 +02004777 char c = *str++;
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004778
Denys Vlasenko9e800222010-10-03 14:28:04 +02004779#if ENABLE_HUSH_BRACE_EXPANSION
4780 if (c == '{' || c == '}') {
4781 /* { -> \{, } -> \} */
4782 o_addchr(o, '\\');
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004783 /* And now we want to add { or } and continue:
4784 * o_addchr(o, c);
4785 * continue;
4786 * luckily, just falling throught achieves this.
4787 */
Denys Vlasenko9e800222010-10-03 14:28:04 +02004788 }
4789#endif
4790 o_addchr(o, c);
4791 if (c == '\\') {
Denys Vlasenko238081f2010-10-03 14:26:26 +02004792 /* \z -> \\\z; \<eol> -> \\<eol> */
4793 o_addchr(o, '\\');
4794 if (len) {
4795 len--;
4796 o_addchr(o, '\\');
4797 o_addchr(o, *str++);
4798 }
4799 }
4800 }
4801}
4802
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004803/* Store given string, finalizing the word and starting new one whenever
4804 * we encounter IFS char(s). This is used for expanding variable values.
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004805 * End-of-string does NOT finalize word: think about 'echo -$VAR-'.
4806 * Return in *ended_with_ifs:
4807 * 1 - ended with IFS char, else 0 (this includes case of empty str).
4808 */
4809static int expand_on_ifs(int *ended_with_ifs, o_string *output, int n, const char *str)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004810{
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004811 int last_is_ifs = 0;
4812
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004813 while (1) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004814 int word_len;
4815
4816 if (!*str) /* EOL - do not finalize word */
4817 break;
4818 word_len = strcspn(str, G.ifs);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004819 if (word_len) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004820 /* We have WORD_LEN leading non-IFS chars */
Denys Vlasenko238081f2010-10-03 14:26:26 +02004821 if (!(output->o_expflags & EXP_FLAG_GLOB)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004822 o_addblock(output, str, word_len);
Denys Vlasenko238081f2010-10-03 14:26:26 +02004823 } else {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004824 /* Protect backslashes against globbing up :)
Denys Vlasenkoa769e022010-09-10 10:12:34 +02004825 * Example: "v='\*'; echo b$v" prints "b\*"
4826 * (and does not try to glob on "*")
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004827 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004828 o_addblock_duplicate_backslash(output, str, word_len);
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004829 /*/ Why can't we do it easier? */
4830 /*o_addblock(output, str, word_len); - WRONG: "v='\*'; echo Z$v" prints "Z*" instead of "Z\*" */
4831 /*o_addqblock(output, str, word_len); - WRONG: "v='*'; echo Z$v" prints "Z*" instead of Z* files */
4832 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004833 last_is_ifs = 0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004834 str += word_len;
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004835 if (!*str) /* EOL - do not finalize word */
4836 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004837 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004838
4839 /* We know str here points to at least one IFS char */
4840 last_is_ifs = 1;
4841 str += strspn(str, G.ifs); /* skip IFS chars */
4842 if (!*str) /* EOL - do not finalize word */
4843 break;
4844
4845 /* Start new word... but not always! */
4846 /* Case "v=' a'; echo ''$v": we do need to finalize empty word: */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004847 if (output->has_quoted_part
4848 /* Case "v=' a'; echo $v":
4849 * here nothing precedes the space in $v expansion,
4850 * therefore we should not finish the word
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004851 * (IOW: if there *is* word to finalize, only then do it):
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004852 */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004853 || (n > 0 && output->data[output->length - 1])
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004854 ) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004855 o_addchr(output, '\0');
4856 debug_print_list("expand_on_ifs", output, n);
4857 n = o_save_ptr(output, n);
4858 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004859 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004860
4861 if (ended_with_ifs)
4862 *ended_with_ifs = last_is_ifs;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004863 debug_print_list("expand_on_ifs[1]", output, n);
4864 return n;
4865}
4866
4867/* Helper to expand $((...)) and heredoc body. These act as if
4868 * they are in double quotes, with the exception that they are not :).
4869 * Just the rules are similar: "expand only $var and `cmd`"
4870 *
4871 * Returns malloced string.
4872 * As an optimization, we return NULL if expansion is not needed.
4873 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004874#if !ENABLE_HUSH_BASH_COMPAT
4875/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4876#define encode_then_expand_string(str, process_bkslash, do_unbackslash) \
4877 encode_then_expand_string(str)
4878#endif
4879static char *encode_then_expand_string(const char *str, int process_bkslash, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004880{
4881 char *exp_str;
4882 struct in_str input;
4883 o_string dest = NULL_O_STRING;
4884
4885 if (!strchr(str, '$')
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02004886 && !strchr(str, '\\')
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004887#if ENABLE_HUSH_TICK
4888 && !strchr(str, '`')
4889#endif
4890 ) {
4891 return NULL;
4892 }
4893
4894 /* We need to expand. Example:
4895 * echo $(($a + `echo 1`)) $((1 + $((2)) ))
4896 */
4897 setup_string_in_str(&input, str);
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004898 encode_string(NULL, &dest, &input, EOF, process_bkslash);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004899//TODO: error check (encode_string returns 0 on error)?
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004900 //bb_error_msg("'%s' -> '%s'", str, dest.data);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004901 exp_str = expand_string_to_string(dest.data, /*unbackslash:*/ do_unbackslash);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004902 //bb_error_msg("'%s' -> '%s'", dest.data, exp_str);
4903 o_free_unsafe(&dest);
4904 return exp_str;
4905}
4906
4907#if ENABLE_SH_MATH_SUPPORT
Denys Vlasenko063847d2010-09-15 13:33:02 +02004908static arith_t expand_and_evaluate_arith(const char *arg, const char **errmsg_p)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004909{
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004910 arith_state_t math_state;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004911 arith_t res;
4912 char *exp_str;
4913
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004914 math_state.lookupvar = get_local_var_value;
4915 math_state.setvar = set_local_var_from_halves;
4916 //math_state.endofname = endofname;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004917 exp_str = encode_then_expand_string(arg, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004918 res = arith(&math_state, exp_str ? exp_str : arg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004919 free(exp_str);
Denys Vlasenko063847d2010-09-15 13:33:02 +02004920 if (errmsg_p)
4921 *errmsg_p = math_state.errmsg;
4922 if (math_state.errmsg)
4923 die_if_script(math_state.errmsg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004924 return res;
4925}
4926#endif
4927
4928#if ENABLE_HUSH_BASH_COMPAT
4929/* ${var/[/]pattern[/repl]} helpers */
4930static char *strstr_pattern(char *val, const char *pattern, int *size)
4931{
4932 while (1) {
4933 char *end = scan_and_match(val, pattern, SCAN_MOVE_FROM_RIGHT + SCAN_MATCH_LEFT_HALF);
4934 debug_printf_varexp("val:'%s' pattern:'%s' end:'%s'\n", val, pattern, end);
4935 if (end) {
4936 *size = end - val;
4937 return val;
4938 }
4939 if (*val == '\0')
4940 return NULL;
4941 /* Optimization: if "*pat" did not match the start of "string",
4942 * we know that "tring", "ring" etc will not match too:
4943 */
4944 if (pattern[0] == '*')
4945 return NULL;
4946 val++;
4947 }
4948}
4949static char *replace_pattern(char *val, const char *pattern, const char *repl, char exp_op)
4950{
4951 char *result = NULL;
4952 unsigned res_len = 0;
4953 unsigned repl_len = strlen(repl);
4954
4955 while (1) {
4956 int size;
4957 char *s = strstr_pattern(val, pattern, &size);
4958 if (!s)
4959 break;
4960
4961 result = xrealloc(result, res_len + (s - val) + repl_len + 1);
4962 memcpy(result + res_len, val, s - val);
4963 res_len += s - val;
4964 strcpy(result + res_len, repl);
4965 res_len += repl_len;
4966 debug_printf_varexp("val:'%s' s:'%s' result:'%s'\n", val, s, result);
4967
4968 val = s + size;
4969 if (exp_op == '/')
4970 break;
4971 }
4972 if (val[0] && result) {
4973 result = xrealloc(result, res_len + strlen(val) + 1);
4974 strcpy(result + res_len, val);
4975 debug_printf_varexp("val:'%s' result:'%s'\n", val, result);
4976 }
4977 debug_printf_varexp("result:'%s'\n", result);
4978 return result;
4979}
4980#endif
4981
4982/* Helper:
4983 * Handles <SPECIAL_VAR_SYMBOL>varname...<SPECIAL_VAR_SYMBOL> construct.
4984 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004985static NOINLINE const char *expand_one_var(char **to_be_freed_pp, char *arg, char **pp)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004986{
4987 const char *val = NULL;
4988 char *to_be_freed = NULL;
4989 char *p = *pp;
4990 char *var;
4991 char first_char;
4992 char exp_op;
4993 char exp_save = exp_save; /* for compiler */
4994 char *exp_saveptr; /* points to expansion operator */
4995 char *exp_word = exp_word; /* for compiler */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004996 char arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004997
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004998 *p = '\0'; /* replace trailing SPECIAL_VAR_SYMBOL */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004999 var = arg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005000 exp_saveptr = arg[1] ? strchr(VAR_ENCODED_SUBST_OPS, arg[1]) : NULL;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005001 arg0 = arg[0];
5002 first_char = arg[0] = arg0 & 0x7f;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005003 exp_op = 0;
5004
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005005 if (first_char == '#' /* ${#... */
5006 && arg[1] && !exp_saveptr /* not ${#} and not ${#<op_char>...} */
5007 ) {
5008 /* It must be length operator: ${#var} */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005009 var++;
5010 exp_op = 'L';
5011 } else {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005012 /* Maybe handle parameter expansion */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005013 if (exp_saveptr /* if 2nd char is one of expansion operators */
5014 && strchr(NUMERIC_SPECVARS_STR, first_char) /* 1st char is special variable */
5015 ) {
5016 /* ${?:0}, ${#[:]%0} etc */
5017 exp_saveptr = var + 1;
5018 } else {
5019 /* ${?}, ${var}, ${var:0}, ${var[:]%0} etc */
5020 exp_saveptr = var+1 + strcspn(var+1, VAR_ENCODED_SUBST_OPS);
5021 }
5022 exp_op = exp_save = *exp_saveptr;
5023 if (exp_op) {
5024 exp_word = exp_saveptr + 1;
5025 if (exp_op == ':') {
5026 exp_op = *exp_word++;
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005027//TODO: try ${var:} and ${var:bogus} in non-bash config
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005028 if (ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005029 && (!exp_op || !strchr(MINUS_PLUS_EQUAL_QUESTION, exp_op))
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005030 ) {
5031 /* oops... it's ${var:N[:M]}, not ${var:?xxx} or some such */
5032 exp_op = ':';
5033 exp_word--;
5034 }
5035 }
5036 *exp_saveptr = '\0';
5037 } /* else: it's not an expansion op, but bare ${var} */
5038 }
5039
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005040 /* Look up the variable in question */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005041 if (isdigit(var[0])) {
Denys Vlasenko77a7b552010-09-09 12:40:03 +02005042 /* parse_dollar should have vetted var for us */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005043 int n = xatoi_positive(var);
5044 if (n < G.global_argc)
5045 val = G.global_argv[n];
5046 /* else val remains NULL: $N with too big N */
5047 } else {
5048 switch (var[0]) {
5049 case '$': /* pid */
5050 val = utoa(G.root_pid);
5051 break;
5052 case '!': /* bg pid */
5053 val = G.last_bg_pid ? utoa(G.last_bg_pid) : "";
5054 break;
5055 case '?': /* exitcode */
5056 val = utoa(G.last_exitcode);
5057 break;
5058 case '#': /* argc */
5059 val = utoa(G.global_argc ? G.global_argc-1 : 0);
5060 break;
5061 default:
5062 val = get_local_var_value(var);
5063 }
5064 }
5065
5066 /* Handle any expansions */
5067 if (exp_op == 'L') {
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02005068 reinit_unicode_for_hush();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005069 debug_printf_expand("expand: length(%s)=", val);
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02005070 val = utoa(val ? unicode_strlen(val) : 0);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005071 debug_printf_expand("%s\n", val);
5072 } else if (exp_op) {
5073 if (exp_op == '%' || exp_op == '#') {
5074 /* Standard-mandated substring removal ops:
5075 * ${parameter%word} - remove smallest suffix pattern
5076 * ${parameter%%word} - remove largest suffix pattern
5077 * ${parameter#word} - remove smallest prefix pattern
5078 * ${parameter##word} - remove largest prefix pattern
5079 *
5080 * Word is expanded to produce a glob pattern.
5081 * Then var's value is matched to it and matching part removed.
5082 */
5083 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005084 char *t;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005085 char *exp_exp_word;
5086 char *loc;
5087 unsigned scan_flags = pick_scan(exp_op, *exp_word);
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02005088 if (exp_op == *exp_word) /* ## or %% */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005089 exp_word++;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005090 exp_exp_word = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005091 if (exp_exp_word)
5092 exp_word = exp_exp_word;
Denys Vlasenko4f870492010-09-10 11:06:01 +02005093 /* HACK ALERT. We depend here on the fact that
5094 * G.global_argv and results of utoa and get_local_var_value
5095 * are actually in writable memory:
5096 * scan_and_match momentarily stores NULs there. */
5097 t = (char*)val;
5098 loc = scan_and_match(t, exp_word, scan_flags);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005099 //bb_error_msg("op:%c str:'%s' pat:'%s' res:'%s'",
Denys Vlasenko4f870492010-09-10 11:06:01 +02005100 // exp_op, t, exp_word, loc);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005101 free(exp_exp_word);
5102 if (loc) { /* match was found */
5103 if (scan_flags & SCAN_MATCH_LEFT_HALF) /* #[#] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005104 val = loc; /* take right part */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005105 else /* %[%] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005106 val = to_be_freed = xstrndup(val, loc - val); /* left */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005107 }
5108 }
5109 }
5110#if ENABLE_HUSH_BASH_COMPAT
5111 else if (exp_op == '/' || exp_op == '\\') {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005112 /* It's ${var/[/]pattern[/repl]} thing.
5113 * Note that in encoded form it has TWO parts:
5114 * var/pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenko4f870492010-09-10 11:06:01 +02005115 * and if // is used, it is encoded as \:
5116 * var\pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005117 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005118 /* Empty variable always gives nothing: */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005119 // "v=''; echo ${v/*/w}" prints "", not "w"
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005120 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005121 /* pattern uses non-standard expansion.
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005122 * repl should be unbackslashed and globbed
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005123 * by the usual expansion rules:
5124 * >az; >bz;
5125 * v='a bz'; echo "${v/a*z/a*z}" prints "a*z"
5126 * v='a bz'; echo "${v/a*z/\z}" prints "\z"
5127 * v='a bz'; echo ${v/a*z/a*z} prints "az"
5128 * v='a bz'; echo ${v/a*z/\z} prints "z"
5129 * (note that a*z _pattern_ is never globbed!)
5130 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005131 char *pattern, *repl, *t;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005132 pattern = encode_then_expand_string(exp_word, /*process_bkslash:*/ 0, /*unbackslash:*/ 0);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005133 if (!pattern)
5134 pattern = xstrdup(exp_word);
5135 debug_printf_varexp("pattern:'%s'->'%s'\n", exp_word, pattern);
5136 *p++ = SPECIAL_VAR_SYMBOL;
5137 exp_word = p;
5138 p = strchr(p, SPECIAL_VAR_SYMBOL);
5139 *p = '\0';
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005140 repl = encode_then_expand_string(exp_word, /*process_bkslash:*/ arg0 & 0x80, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005141 debug_printf_varexp("repl:'%s'->'%s'\n", exp_word, repl);
5142 /* HACK ALERT. We depend here on the fact that
5143 * G.global_argv and results of utoa and get_local_var_value
5144 * are actually in writable memory:
5145 * replace_pattern momentarily stores NULs there. */
5146 t = (char*)val;
5147 to_be_freed = replace_pattern(t,
5148 pattern,
5149 (repl ? repl : exp_word),
5150 exp_op);
5151 if (to_be_freed) /* at least one replace happened */
5152 val = to_be_freed;
5153 free(pattern);
5154 free(repl);
5155 }
5156 }
5157#endif
5158 else if (exp_op == ':') {
5159#if ENABLE_HUSH_BASH_COMPAT && ENABLE_SH_MATH_SUPPORT
5160 /* It's ${var:N[:M]} bashism.
5161 * Note that in encoded form it has TWO parts:
5162 * var:N<SPECIAL_VAR_SYMBOL>M<SPECIAL_VAR_SYMBOL>
5163 */
5164 arith_t beg, len;
Denys Vlasenko063847d2010-09-15 13:33:02 +02005165 const char *errmsg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005166
Denys Vlasenko063847d2010-09-15 13:33:02 +02005167 beg = expand_and_evaluate_arith(exp_word, &errmsg);
5168 if (errmsg)
5169 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005170 debug_printf_varexp("beg:'%s'=%lld\n", exp_word, (long long)beg);
5171 *p++ = SPECIAL_VAR_SYMBOL;
5172 exp_word = p;
5173 p = strchr(p, SPECIAL_VAR_SYMBOL);
5174 *p = '\0';
Denys Vlasenko063847d2010-09-15 13:33:02 +02005175 len = expand_and_evaluate_arith(exp_word, &errmsg);
5176 if (errmsg)
5177 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005178 debug_printf_varexp("len:'%s'=%lld\n", exp_word, (long long)len);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005179 if (len >= 0) { /* bash compat: len < 0 is illegal */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005180 if (beg < 0) /* bash compat */
5181 beg = 0;
5182 debug_printf_varexp("from val:'%s'\n", val);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005183 if (len == 0 || !val || beg >= strlen(val)) {
Denys Vlasenko063847d2010-09-15 13:33:02 +02005184 arith_err:
Denys Vlasenkob771c652010-09-13 00:34:26 +02005185 val = NULL;
5186 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005187 /* Paranoia. What if user entered 9999999999999
5188 * which fits in arith_t but not int? */
5189 if (len >= INT_MAX)
5190 len = INT_MAX;
5191 val = to_be_freed = xstrndup(val + beg, len);
5192 }
5193 debug_printf_varexp("val:'%s'\n", val);
5194 } else
5195#endif
5196 {
5197 die_if_script("malformed ${%s:...}", var);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005198 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005199 }
5200 } else { /* one of "-=+?" */
5201 /* Standard-mandated substitution ops:
5202 * ${var?word} - indicate error if unset
5203 * If var is unset, word (or a message indicating it is unset
5204 * if word is null) is written to standard error
5205 * and the shell exits with a non-zero exit status.
5206 * Otherwise, the value of var is substituted.
5207 * ${var-word} - use default value
5208 * If var is unset, word is substituted.
5209 * ${var=word} - assign and use default value
5210 * If var is unset, word is assigned to var.
5211 * In all cases, final value of var is substituted.
5212 * ${var+word} - use alternative value
5213 * If var is unset, null is substituted.
5214 * Otherwise, word is substituted.
5215 *
5216 * Word is subjected to tilde expansion, parameter expansion,
5217 * command substitution, and arithmetic expansion.
5218 * If word is not needed, it is not expanded.
5219 *
5220 * Colon forms (${var:-word}, ${var:=word} etc) do the same,
5221 * but also treat null var as if it is unset.
5222 */
5223 int use_word = (!val || ((exp_save == ':') && !val[0]));
5224 if (exp_op == '+')
5225 use_word = !use_word;
5226 debug_printf_expand("expand: op:%c (null:%s) test:%i\n", exp_op,
5227 (exp_save == ':') ? "true" : "false", use_word);
5228 if (use_word) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005229 to_be_freed = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005230 if (to_be_freed)
5231 exp_word = to_be_freed;
5232 if (exp_op == '?') {
5233 /* mimic bash message */
5234 die_if_script("%s: %s",
5235 var,
5236 exp_word[0] ? exp_word : "parameter null or not set"
5237 );
5238//TODO: how interactive bash aborts expansion mid-command?
5239 } else {
5240 val = exp_word;
5241 }
5242
5243 if (exp_op == '=') {
5244 /* ${var=[word]} or ${var:=[word]} */
5245 if (isdigit(var[0]) || var[0] == '#') {
5246 /* mimic bash message */
5247 die_if_script("$%s: cannot assign in this way", var);
5248 val = NULL;
5249 } else {
5250 char *new_var = xasprintf("%s=%s", var, val);
5251 set_local_var(new_var, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
5252 }
5253 }
5254 }
5255 } /* one of "-=+?" */
5256
5257 *exp_saveptr = exp_save;
5258 } /* if (exp_op) */
5259
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005260 arg[0] = arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005261
5262 *pp = p;
5263 *to_be_freed_pp = to_be_freed;
5264 return val;
5265}
5266
5267/* Expand all variable references in given string, adding words to list[]
5268 * at n, n+1,... positions. Return updated n (so that list[n] is next one
5269 * to be filled). This routine is extremely tricky: has to deal with
5270 * variables/parameters with whitespace, $* and $@, and constructs like
5271 * 'echo -$*-'. If you play here, you must run testsuite afterwards! */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005272static NOINLINE int expand_vars_to_list(o_string *output, int n, char *arg)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005273{
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005274 /* output->o_expflags & EXP_FLAG_SINGLEWORD (0x80) if we are in
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005275 * expansion of right-hand side of assignment == 1-element expand.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005276 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005277 char cant_be_null = 0; /* only bit 0x80 matters */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005278 int ended_in_ifs = 0; /* did last unquoted expansion end with IFS chars? */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005279 char *p;
5280
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005281 debug_printf_expand("expand_vars_to_list: arg:'%s' singleword:%x\n", arg,
5282 !!(output->o_expflags & EXP_FLAG_SINGLEWORD));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005283 debug_print_list("expand_vars_to_list", output, n);
5284 n = o_save_ptr(output, n);
5285 debug_print_list("expand_vars_to_list[0]", output, n);
5286
5287 while ((p = strchr(arg, SPECIAL_VAR_SYMBOL)) != NULL) {
5288 char first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005289 char *to_be_freed = NULL;
5290 const char *val = NULL;
5291#if ENABLE_HUSH_TICK
5292 o_string subst_result = NULL_O_STRING;
5293#endif
5294#if ENABLE_SH_MATH_SUPPORT
5295 char arith_buf[sizeof(arith_t)*3 + 2];
5296#endif
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005297
5298 if (ended_in_ifs) {
5299 o_addchr(output, '\0');
5300 n = o_save_ptr(output, n);
5301 ended_in_ifs = 0;
5302 }
5303
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005304 o_addblock(output, arg, p - arg);
5305 debug_print_list("expand_vars_to_list[1]", output, n);
5306 arg = ++p;
5307 p = strchr(p, SPECIAL_VAR_SYMBOL);
5308
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005309 /* Fetch special var name (if it is indeed one of them)
5310 * and quote bit, force the bit on if singleword expansion -
5311 * important for not getting v=$@ expand to many words. */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005312 first_ch = arg[0] | (output->o_expflags & EXP_FLAG_SINGLEWORD);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005313
5314 /* Is this variable quoted and thus expansion can't be null?
5315 * "$@" is special. Even if quoted, it can still
5316 * expand to nothing (not even an empty string),
5317 * thus it is excluded. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005318 if ((first_ch & 0x7f) != '@')
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005319 cant_be_null |= first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005320
5321 switch (first_ch & 0x7f) {
5322 /* Highest bit in first_ch indicates that var is double-quoted */
5323 case '*':
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005324 case '@': {
5325 int i;
5326 if (!G.global_argv[1])
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005327 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005328 i = 1;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005329 cant_be_null |= first_ch; /* do it for "$@" _now_, when we know it's not empty */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005330 if (!(first_ch & 0x80)) { /* unquoted $* or $@ */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005331 while (G.global_argv[i]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005332 n = expand_on_ifs(NULL, output, n, G.global_argv[i]);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005333 debug_printf_expand("expand_vars_to_list: argv %d (last %d)\n", i, G.global_argc - 1);
5334 if (G.global_argv[i++][0] && G.global_argv[i]) {
5335 /* this argv[] is not empty and not last:
5336 * put terminating NUL, start new word */
5337 o_addchr(output, '\0');
5338 debug_print_list("expand_vars_to_list[2]", output, n);
5339 n = o_save_ptr(output, n);
5340 debug_print_list("expand_vars_to_list[3]", output, n);
5341 }
5342 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005343 } else
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005344 /* If EXP_FLAG_SINGLEWORD, we handle assignment 'a=....$@.....'
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005345 * and in this case should treat it like '$*' - see 'else...' below */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005346 if (first_ch == ('@'|0x80) /* quoted $@ */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005347 && !(output->o_expflags & EXP_FLAG_SINGLEWORD) /* not v="$@" case */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005348 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005349 while (1) {
5350 o_addQstr(output, G.global_argv[i]);
5351 if (++i >= G.global_argc)
5352 break;
5353 o_addchr(output, '\0');
5354 debug_print_list("expand_vars_to_list[4]", output, n);
5355 n = o_save_ptr(output, n);
5356 }
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005357 } else { /* quoted $* (or v="$@" case): add as one word */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005358 while (1) {
5359 o_addQstr(output, G.global_argv[i]);
5360 if (!G.global_argv[++i])
5361 break;
5362 if (G.ifs[0])
5363 o_addchr(output, G.ifs[0]);
5364 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005365 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005366 }
5367 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005368 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005369 case SPECIAL_VAR_SYMBOL: /* <SPECIAL_VAR_SYMBOL><SPECIAL_VAR_SYMBOL> */
5370 /* "Empty variable", used to make "" etc to not disappear */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005371 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005372 arg++;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005373 cant_be_null = 0x80;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005374 break;
5375#if ENABLE_HUSH_TICK
5376 case '`': /* <SPECIAL_VAR_SYMBOL>`cmd<SPECIAL_VAR_SYMBOL> */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005377 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005378 arg++;
5379 /* Can't just stuff it into output o_string,
5380 * expanded result may need to be globbed
5381 * and $IFS-splitted */
5382 debug_printf_subst("SUBST '%s' first_ch %x\n", arg, first_ch);
5383 G.last_exitcode = process_command_subs(&subst_result, arg);
5384 debug_printf_subst("SUBST RES:%d '%s'\n", G.last_exitcode, subst_result.data);
5385 val = subst_result.data;
5386 goto store_val;
5387#endif
5388#if ENABLE_SH_MATH_SUPPORT
5389 case '+': { /* <SPECIAL_VAR_SYMBOL>+cmd<SPECIAL_VAR_SYMBOL> */
5390 arith_t res;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005391
5392 arg++; /* skip '+' */
5393 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
5394 debug_printf_subst("ARITH '%s' first_ch %x\n", arg, first_ch);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005395 res = expand_and_evaluate_arith(arg, NULL);
Denys Vlasenkobed7c812010-09-16 11:50:46 +02005396 debug_printf_subst("ARITH RES '"ARITH_FMT"'\n", res);
5397 sprintf(arith_buf, ARITH_FMT, res);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005398 val = arith_buf;
5399 break;
5400 }
5401#endif
5402 default:
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005403 val = expand_one_var(&to_be_freed, arg, &p);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005404 IF_HUSH_TICK(store_val:)
5405 if (!(first_ch & 0x80)) { /* unquoted $VAR */
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005406 debug_printf_expand("unquoted '%s', output->o_escape:%d\n", val,
5407 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005408 if (val && val[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005409 n = expand_on_ifs(&ended_in_ifs, output, n, val);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005410 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005411 }
5412 } else { /* quoted $VAR, val will be appended below */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005413 output->has_quoted_part = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005414 debug_printf_expand("quoted '%s', output->o_escape:%d\n", val,
5415 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005416 }
5417 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005418 } /* switch (char after <SPECIAL_VAR_SYMBOL>) */
5419
5420 if (val && val[0]) {
5421 o_addQstr(output, val);
5422 }
5423 free(to_be_freed);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005424
5425 /* Restore NULL'ed SPECIAL_VAR_SYMBOL.
5426 * Do the check to avoid writing to a const string. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005427 if (*p != SPECIAL_VAR_SYMBOL)
5428 *p = SPECIAL_VAR_SYMBOL;
5429
5430#if ENABLE_HUSH_TICK
5431 o_free(&subst_result);
5432#endif
5433 arg = ++p;
5434 } /* end of "while (SPECIAL_VAR_SYMBOL is found) ..." */
5435
5436 if (arg[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005437 if (ended_in_ifs) {
5438 o_addchr(output, '\0');
5439 n = o_save_ptr(output, n);
5440 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005441 debug_print_list("expand_vars_to_list[a]", output, n);
5442 /* this part is literal, and it was already pre-quoted
5443 * if needed (much earlier), do not use o_addQstr here! */
5444 o_addstr_with_NUL(output, arg);
5445 debug_print_list("expand_vars_to_list[b]", output, n);
5446 } else if (output->length == o_get_last_ptr(output, n) /* expansion is empty */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005447 && !(cant_be_null & 0x80) /* and all vars were not quoted. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005448 ) {
5449 n--;
5450 /* allow to reuse list[n] later without re-growth */
5451 output->has_empty_slot = 1;
5452 } else {
5453 o_addchr(output, '\0');
5454 }
5455
5456 return n;
5457}
5458
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005459static char **expand_variables(char **argv, unsigned expflags)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005460{
5461 int n;
5462 char **list;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005463 o_string output = NULL_O_STRING;
5464
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005465 output.o_expflags = expflags;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005466
5467 n = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005468 while (*argv) {
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005469 n = expand_vars_to_list(&output, n, *argv);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005470 argv++;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005471 }
5472 debug_print_list("expand_variables", &output, n);
5473
5474 /* output.data (malloced in one block) gets returned in "list" */
5475 list = o_finalize_list(&output, n);
5476 debug_print_strings("expand_variables[1]", list);
5477 return list;
5478}
5479
5480static char **expand_strvec_to_strvec(char **argv)
5481{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005482 return expand_variables(argv, EXP_FLAG_GLOB | EXP_FLAG_ESC_GLOB_CHARS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005483}
5484
5485#if ENABLE_HUSH_BASH_COMPAT
5486static char **expand_strvec_to_strvec_singleword_noglob(char **argv)
5487{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005488 return expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005489}
5490#endif
5491
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005492/* Used for expansion of right hand of assignments,
5493 * $((...)), heredocs, variable espansion parts.
5494 *
5495 * NB: should NOT do globbing!
5496 * "export v=/bin/c*; env | grep ^v=" outputs "v=/bin/c*"
5497 */
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005498static char *expand_string_to_string(const char *str, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005499{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005500#if !ENABLE_HUSH_BASH_COMPAT
5501 const int do_unbackslash = 1;
5502#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005503 char *argv[2], **list;
5504
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005505 debug_printf_expand("string_to_string<='%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005506 /* This is generally an optimization, but it also
5507 * handles "", which otherwise trips over !list[0] check below.
5508 * (is this ever happens that we actually get str="" here?)
5509 */
5510 if (!strchr(str, SPECIAL_VAR_SYMBOL) && !strchr(str, '\\')) {
5511 //TODO: Can use on strings with \ too, just unbackslash() them?
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005512 debug_printf_expand("string_to_string(fast)=>'%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005513 return xstrdup(str);
5514 }
5515
5516 argv[0] = (char*)str;
5517 argv[1] = NULL;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005518 list = expand_variables(argv, do_unbackslash
5519 ? EXP_FLAG_ESC_GLOB_CHARS | EXP_FLAG_SINGLEWORD
5520 : EXP_FLAG_SINGLEWORD
5521 );
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005522 if (HUSH_DEBUG)
5523 if (!list[0] || list[1])
5524 bb_error_msg_and_die("BUG in varexp2");
5525 /* actually, just move string 2*sizeof(char*) bytes back */
5526 overlapping_strcpy((char*)list, list[0]);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005527 if (do_unbackslash)
5528 unbackslash((char*)list);
5529 debug_printf_expand("string_to_string=>'%s'\n", (char*)list);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005530 return (char*)list;
5531}
5532
5533/* Used for "eval" builtin */
5534static char* expand_strvec_to_string(char **argv)
5535{
5536 char **list;
5537
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005538 list = expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005539 /* Convert all NULs to spaces */
5540 if (list[0]) {
5541 int n = 1;
5542 while (list[n]) {
5543 if (HUSH_DEBUG)
5544 if (list[n-1] + strlen(list[n-1]) + 1 != list[n])
5545 bb_error_msg_and_die("BUG in varexp3");
5546 /* bash uses ' ' regardless of $IFS contents */
5547 list[n][-1] = ' ';
5548 n++;
5549 }
5550 }
5551 overlapping_strcpy((char*)list, list[0]);
5552 debug_printf_expand("strvec_to_string='%s'\n", (char*)list);
5553 return (char*)list;
5554}
5555
5556static char **expand_assignments(char **argv, int count)
5557{
5558 int i;
5559 char **p;
5560
5561 G.expanded_assignments = p = NULL;
5562 /* Expand assignments into one string each */
5563 for (i = 0; i < count; i++) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005564 G.expanded_assignments = p = add_string_to_strings(p, expand_string_to_string(argv[i], /*unbackslash:*/ 1));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005565 }
5566 G.expanded_assignments = NULL;
5567 return p;
5568}
5569
5570
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005571static void switch_off_special_sigs(unsigned mask)
5572{
5573 unsigned sig = 0;
5574 while ((mask >>= 1) != 0) {
5575 sig++;
5576 if (!(mask & 1))
5577 continue;
5578 if (G.traps) {
5579 if (G.traps[sig] && !G.traps[sig][0])
5580 /* trap is '', has to remain SIG_IGN */
5581 continue;
5582 free(G.traps[sig]);
5583 G.traps[sig] = NULL;
5584 }
5585 /* We are here only if no trap or trap was not '' */
Denys Vlasenko0806e402011-05-12 23:06:20 +02005586 install_sighandler(sig, SIG_DFL);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005587 }
5588}
5589
Denys Vlasenkob347df92011-08-09 22:49:15 +02005590#if BB_MMU
5591/* never called */
5592void re_execute_shell(char ***to_free, const char *s,
5593 char *g_argv0, char **g_argv,
5594 char **builtin_argv) NORETURN;
5595
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005596static void reset_traps_to_defaults(void)
5597{
5598 /* This function is always called in a child shell
5599 * after fork (not vfork, NOMMU doesn't use this function).
5600 */
5601 unsigned sig;
5602 unsigned mask;
5603
5604 /* Child shells are not interactive.
5605 * SIGTTIN/SIGTTOU/SIGTSTP should not have special handling.
5606 * Testcase: (while :; do :; done) + ^Z should background.
5607 * Same goes for SIGTERM, SIGHUP, SIGINT.
5608 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005609 mask = (G.special_sig_mask & SPECIAL_INTERACTIVE_SIGS) | G_fatal_sig_mask;
5610 if (!G.traps && !mask)
5611 return; /* already no traps and no special sigs */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005612
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005613 /* Switch off special sigs */
5614 switch_off_special_sigs(mask);
5615#if ENABLE_HUSH_JOB
5616 G_fatal_sig_mask = 0;
5617#endif
Denys Vlasenko10c01312011-05-11 11:49:21 +02005618 G.special_sig_mask &= ~SPECIAL_INTERACTIVE_SIGS;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02005619 /* SIGQUIT,SIGCHLD and maybe SPECIAL_JOBSTOP_SIGS
5620 * remain set in G.special_sig_mask */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005621
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005622 if (!G.traps)
5623 return;
5624
5625 /* Reset all sigs to default except ones with empty traps */
5626 for (sig = 0; sig < NSIG; sig++) {
5627 if (!G.traps[sig])
5628 continue; /* no trap: nothing to do */
5629 if (!G.traps[sig][0])
5630 continue; /* empty trap: has to remain SIG_IGN */
5631 /* sig has non-empty trap, reset it: */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005632 free(G.traps[sig]);
5633 G.traps[sig] = NULL;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005634 /* There is no signal for trap 0 (EXIT) */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005635 if (sig == 0)
5636 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02005637 install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005638 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005639}
5640
5641#else /* !BB_MMU */
5642
5643static void re_execute_shell(char ***to_free, const char *s,
5644 char *g_argv0, char **g_argv,
5645 char **builtin_argv) NORETURN;
5646static void re_execute_shell(char ***to_free, const char *s,
5647 char *g_argv0, char **g_argv,
5648 char **builtin_argv)
5649{
5650# define NOMMU_HACK_FMT ("-$%x:%x:%x:%x:%x:%llx" IF_HUSH_LOOPS(":%x"))
5651 /* delims + 2 * (number of bytes in printed hex numbers) */
5652 char param_buf[sizeof(NOMMU_HACK_FMT) + 2 * (sizeof(int)*6 + sizeof(long long)*1)];
5653 char *heredoc_argv[4];
5654 struct variable *cur;
5655# if ENABLE_HUSH_FUNCTIONS
5656 struct function *funcp;
5657# endif
5658 char **argv, **pp;
5659 unsigned cnt;
5660 unsigned long long empty_trap_mask;
5661
5662 if (!g_argv0) { /* heredoc */
5663 argv = heredoc_argv;
5664 argv[0] = (char *) G.argv0_for_re_execing;
5665 argv[1] = (char *) "-<";
5666 argv[2] = (char *) s;
5667 argv[3] = NULL;
5668 pp = &argv[3]; /* used as pointer to empty environment */
5669 goto do_exec;
5670 }
5671
5672 cnt = 0;
5673 pp = builtin_argv;
5674 if (pp) while (*pp++)
5675 cnt++;
5676
5677 empty_trap_mask = 0;
5678 if (G.traps) {
5679 int sig;
5680 for (sig = 1; sig < NSIG; sig++) {
5681 if (G.traps[sig] && !G.traps[sig][0])
5682 empty_trap_mask |= 1LL << sig;
5683 }
5684 }
5685
5686 sprintf(param_buf, NOMMU_HACK_FMT
5687 , (unsigned) G.root_pid
5688 , (unsigned) G.root_ppid
5689 , (unsigned) G.last_bg_pid
5690 , (unsigned) G.last_exitcode
5691 , cnt
5692 , empty_trap_mask
5693 IF_HUSH_LOOPS(, G.depth_of_loop)
5694 );
5695# undef NOMMU_HACK_FMT
5696 /* 1:hush 2:-$<pid>:<pid>:<exitcode>:<etc...> <vars...> <funcs...>
5697 * 3:-c 4:<cmd> 5:<arg0> <argN...> 6:NULL
5698 */
5699 cnt += 6;
5700 for (cur = G.top_var; cur; cur = cur->next) {
5701 if (!cur->flg_export || cur->flg_read_only)
5702 cnt += 2;
5703 }
5704# if ENABLE_HUSH_FUNCTIONS
5705 for (funcp = G.top_func; funcp; funcp = funcp->next)
5706 cnt += 3;
5707# endif
5708 pp = g_argv;
5709 while (*pp++)
5710 cnt++;
5711 *to_free = argv = pp = xzalloc(sizeof(argv[0]) * cnt);
5712 *pp++ = (char *) G.argv0_for_re_execing;
5713 *pp++ = param_buf;
5714 for (cur = G.top_var; cur; cur = cur->next) {
5715 if (strcmp(cur->varstr, hush_version_str) == 0)
5716 continue;
5717 if (cur->flg_read_only) {
5718 *pp++ = (char *) "-R";
5719 *pp++ = cur->varstr;
5720 } else if (!cur->flg_export) {
5721 *pp++ = (char *) "-V";
5722 *pp++ = cur->varstr;
5723 }
5724 }
5725# if ENABLE_HUSH_FUNCTIONS
5726 for (funcp = G.top_func; funcp; funcp = funcp->next) {
5727 *pp++ = (char *) "-F";
5728 *pp++ = funcp->name;
5729 *pp++ = funcp->body_as_string;
5730 }
5731# endif
5732 /* We can pass activated traps here. Say, -Tnn:trap_string
5733 *
5734 * However, POSIX says that subshells reset signals with traps
5735 * to SIG_DFL.
5736 * I tested bash-3.2 and it not only does that with true subshells
5737 * of the form ( list ), but with any forked children shells.
5738 * I set trap "echo W" WINCH; and then tried:
5739 *
5740 * { echo 1; sleep 20; echo 2; } &
5741 * while true; do echo 1; sleep 20; echo 2; break; done &
5742 * true | { echo 1; sleep 20; echo 2; } | cat
5743 *
5744 * In all these cases sending SIGWINCH to the child shell
5745 * did not run the trap. If I add trap "echo V" WINCH;
5746 * _inside_ group (just before echo 1), it works.
5747 *
5748 * I conclude it means we don't need to pass active traps here.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005749 */
5750 *pp++ = (char *) "-c";
5751 *pp++ = (char *) s;
5752 if (builtin_argv) {
5753 while (*++builtin_argv)
5754 *pp++ = *builtin_argv;
5755 *pp++ = (char *) "";
5756 }
5757 *pp++ = g_argv0;
5758 while (*g_argv)
5759 *pp++ = *g_argv++;
5760 /* *pp = NULL; - is already there */
5761 pp = environ;
5762
5763 do_exec:
5764 debug_printf_exec("re_execute_shell pid:%d cmd:'%s'\n", getpid(), s);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02005765 /* Don't propagate SIG_IGN to the child */
5766 if (SPECIAL_JOBSTOP_SIGS != 0)
5767 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005768 execve(bb_busybox_exec_path, argv, pp);
5769 /* Fallback. Useful for init=/bin/hush usage etc */
5770 if (argv[0][0] == '/')
5771 execve(argv[0], argv, pp);
5772 xfunc_error_retval = 127;
5773 bb_error_msg_and_die("can't re-execute the shell");
5774}
5775#endif /* !BB_MMU */
5776
5777
5778static int run_and_free_list(struct pipe *pi);
5779
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005780/* Executing from string: eval, sh -c '...'
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005781 * or from file: /etc/profile, . file, sh <script>, sh (intereactive)
5782 * end_trigger controls how often we stop parsing
5783 * NUL: parse all, execute, return
5784 * ';': parse till ';' or newline, execute, repeat till EOF
5785 */
5786static void parse_and_run_stream(struct in_str *inp, int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00005787{
Denys Vlasenko00243b02009-11-16 02:00:03 +01005788 /* Why we need empty flag?
5789 * An obscure corner case "false; ``; echo $?":
5790 * empty command in `` should still set $? to 0.
5791 * But we can't just set $? to 0 at the start,
5792 * this breaks "false; echo `echo $?`" case.
5793 */
5794 bool empty = 1;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005795 while (1) {
5796 struct pipe *pipe_list;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00005797
Denys Vlasenkoa1463192011-01-18 17:55:04 +01005798#if ENABLE_HUSH_INTERACTIVE
5799 if (end_trigger == ';')
5800 inp->promptmode = 0; /* PS1 */
5801#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005802 pipe_list = parse_stream(NULL, inp, end_trigger);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02005803 if (!pipe_list || pipe_list == ERR_PTR) { /* EOF/error */
5804 /* If we are in "big" script
5805 * (not in `cmd` or something similar)...
5806 */
5807 if (pipe_list == ERR_PTR && end_trigger == ';') {
5808 /* Discard cached input (rest of line) */
5809 int ch = inp->last_char;
5810 while (ch != EOF && ch != '\n') {
5811 //bb_error_msg("Discarded:'%c'", ch);
5812 ch = i_getch(inp);
5813 }
5814 /* Force prompt */
5815 inp->p = NULL;
5816 /* This stream isn't empty */
5817 empty = 0;
5818 continue;
5819 }
5820 if (!pipe_list && empty)
Denys Vlasenko00243b02009-11-16 02:00:03 +01005821 G.last_exitcode = 0;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005822 break;
Denys Vlasenko00243b02009-11-16 02:00:03 +01005823 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005824 debug_print_tree(pipe_list, 0);
5825 debug_printf_exec("parse_and_run_stream: run_and_free_list\n");
5826 run_and_free_list(pipe_list);
Denys Vlasenko00243b02009-11-16 02:00:03 +01005827 empty = 0;
Denys Vlasenko68d5cb52011-03-24 02:50:03 +01005828#if ENABLE_HUSH_FUNCTIONS
5829 if (G.flag_return_in_progress == 1)
5830 break;
5831#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005832 }
Eric Andersen25f27032001-04-26 23:22:31 +00005833}
5834
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005835static void parse_and_run_string(const char *s)
Eric Andersen25f27032001-04-26 23:22:31 +00005836{
5837 struct in_str input;
5838 setup_string_in_str(&input, s);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005839 parse_and_run_stream(&input, '\0');
Eric Andersen25f27032001-04-26 23:22:31 +00005840}
5841
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005842static void parse_and_run_file(FILE *f)
Eric Andersen25f27032001-04-26 23:22:31 +00005843{
Eric Andersen25f27032001-04-26 23:22:31 +00005844 struct in_str input;
5845 setup_file_in_str(&input, f);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005846 parse_and_run_stream(&input, ';');
Eric Andersen25f27032001-04-26 23:22:31 +00005847}
5848
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005849#if ENABLE_HUSH_TICK
5850static FILE *generate_stream_from_string(const char *s, pid_t *pid_p)
5851{
5852 pid_t pid;
5853 int channel[2];
5854# if !BB_MMU
5855 char **to_free = NULL;
5856# endif
5857
5858 xpipe(channel);
5859 pid = BB_MMU ? xfork() : xvfork();
5860 if (pid == 0) { /* child */
5861 disable_restore_tty_pgrp_on_exit();
5862 /* Process substitution is not considered to be usual
5863 * 'command execution'.
5864 * SUSv3 says ctrl-Z should be ignored, ctrl-C should not.
5865 */
5866 bb_signals(0
5867 + (1 << SIGTSTP)
5868 + (1 << SIGTTIN)
5869 + (1 << SIGTTOU)
5870 , SIG_IGN);
5871 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
5872 close(channel[0]); /* NB: close _first_, then move fd! */
5873 xmove_fd(channel[1], 1);
5874 /* Prevent it from trying to handle ctrl-z etc */
5875 IF_HUSH_JOB(G.run_list_level = 1;)
5876 /* Awful hack for `trap` or $(trap).
5877 *
5878 * http://www.opengroup.org/onlinepubs/009695399/utilities/trap.html
5879 * contains an example where "trap" is executed in a subshell:
5880 *
5881 * save_traps=$(trap)
5882 * ...
5883 * eval "$save_traps"
5884 *
5885 * Standard does not say that "trap" in subshell shall print
5886 * parent shell's traps. It only says that its output
5887 * must have suitable form, but then, in the above example
5888 * (which is not supposed to be normative), it implies that.
5889 *
5890 * bash (and probably other shell) does implement it
5891 * (traps are reset to defaults, but "trap" still shows them),
5892 * but as a result, "trap" logic is hopelessly messed up:
5893 *
5894 * # trap
5895 * trap -- 'echo Ho' SIGWINCH <--- we have a handler
5896 * # (trap) <--- trap is in subshell - no output (correct, traps are reset)
5897 * # true | trap <--- trap is in subshell - no output (ditto)
5898 * # echo `true | trap` <--- in subshell - output (but traps are reset!)
5899 * trap -- 'echo Ho' SIGWINCH
5900 * # echo `(trap)` <--- in subshell in subshell - output
5901 * trap -- 'echo Ho' SIGWINCH
5902 * # echo `true | (trap)` <--- in subshell in subshell in subshell - output!
5903 * trap -- 'echo Ho' SIGWINCH
5904 *
5905 * The rules when to forget and when to not forget traps
5906 * get really complex and nonsensical.
5907 *
5908 * Our solution: ONLY bare $(trap) or `trap` is special.
5909 */
5910 s = skip_whitespace(s);
Denys Vlasenko8dff01d2015-03-12 17:48:34 +01005911 if (is_prefixed_with(s, "trap")
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005912 && skip_whitespace(s + 4)[0] == '\0'
5913 ) {
5914 static const char *const argv[] = { NULL, NULL };
5915 builtin_trap((char**)argv);
5916 exit(0); /* not _exit() - we need to fflush */
5917 }
5918# if BB_MMU
5919 reset_traps_to_defaults();
5920 parse_and_run_string(s);
5921 _exit(G.last_exitcode);
5922# else
5923 /* We re-execute after vfork on NOMMU. This makes this script safe:
5924 * yes "0123456789012345678901234567890" | dd bs=32 count=64k >BIG
5925 * huge=`cat BIG` # was blocking here forever
5926 * echo OK
5927 */
5928 re_execute_shell(&to_free,
5929 s,
5930 G.global_argv[0],
5931 G.global_argv + 1,
5932 NULL);
5933# endif
5934 }
5935
5936 /* parent */
5937 *pid_p = pid;
5938# if ENABLE_HUSH_FAST
5939 G.count_SIGCHLD++;
5940//bb_error_msg("[%d] fork in generate_stream_from_string:"
5941// " G.count_SIGCHLD:%d G.handled_SIGCHLD:%d",
5942// getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
5943# endif
5944 enable_restore_tty_pgrp_on_exit();
5945# if !BB_MMU
5946 free(to_free);
5947# endif
5948 close(channel[1]);
5949 close_on_exec_on(channel[0]);
5950 return xfdopen_for_read(channel[0]);
5951}
5952
5953/* Return code is exit status of the process that is run. */
5954static int process_command_subs(o_string *dest, const char *s)
5955{
5956 FILE *fp;
5957 struct in_str pipe_str;
5958 pid_t pid;
5959 int status, ch, eol_cnt;
5960
5961 fp = generate_stream_from_string(s, &pid);
5962
5963 /* Now send results of command back into original context */
5964 setup_file_in_str(&pipe_str, fp);
5965 eol_cnt = 0;
5966 while ((ch = i_getch(&pipe_str)) != EOF) {
5967 if (ch == '\n') {
5968 eol_cnt++;
5969 continue;
5970 }
5971 while (eol_cnt) {
5972 o_addchr(dest, '\n');
5973 eol_cnt--;
5974 }
5975 o_addQchr(dest, ch);
5976 }
5977
5978 debug_printf("done reading from `cmd` pipe, closing it\n");
5979 fclose(fp);
5980 /* We need to extract exitcode. Test case
5981 * "true; echo `sleep 1; false` $?"
5982 * should print 1 */
5983 safe_waitpid(pid, &status, 0);
5984 debug_printf("child exited. returning its exitcode:%d\n", WEXITSTATUS(status));
5985 return WEXITSTATUS(status);
5986}
5987#endif /* ENABLE_HUSH_TICK */
5988
5989
5990static void setup_heredoc(struct redir_struct *redir)
5991{
5992 struct fd_pair pair;
5993 pid_t pid;
5994 int len, written;
5995 /* the _body_ of heredoc (misleading field name) */
5996 const char *heredoc = redir->rd_filename;
5997 char *expanded;
5998#if !BB_MMU
5999 char **to_free;
6000#endif
6001
6002 expanded = NULL;
6003 if (!(redir->rd_dup & HEREDOC_QUOTED)) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02006004 expanded = encode_then_expand_string(heredoc, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006005 if (expanded)
6006 heredoc = expanded;
6007 }
6008 len = strlen(heredoc);
6009
6010 close(redir->rd_fd); /* often saves dup2+close in xmove_fd */
6011 xpiped_pair(pair);
6012 xmove_fd(pair.rd, redir->rd_fd);
6013
6014 /* Try writing without forking. Newer kernels have
6015 * dynamically growing pipes. Must use non-blocking write! */
6016 ndelay_on(pair.wr);
6017 while (1) {
6018 written = write(pair.wr, heredoc, len);
6019 if (written <= 0)
6020 break;
6021 len -= written;
6022 if (len == 0) {
6023 close(pair.wr);
6024 free(expanded);
6025 return;
6026 }
6027 heredoc += written;
6028 }
6029 ndelay_off(pair.wr);
6030
6031 /* Okay, pipe buffer was not big enough */
6032 /* Note: we must not create a stray child (bastard? :)
6033 * for the unsuspecting parent process. Child creates a grandchild
6034 * and exits before parent execs the process which consumes heredoc
6035 * (that exec happens after we return from this function) */
6036#if !BB_MMU
6037 to_free = NULL;
6038#endif
6039 pid = xvfork();
6040 if (pid == 0) {
6041 /* child */
6042 disable_restore_tty_pgrp_on_exit();
6043 pid = BB_MMU ? xfork() : xvfork();
6044 if (pid != 0)
6045 _exit(0);
6046 /* grandchild */
6047 close(redir->rd_fd); /* read side of the pipe */
6048#if BB_MMU
6049 full_write(pair.wr, heredoc, len); /* may loop or block */
6050 _exit(0);
6051#else
6052 /* Delegate blocking writes to another process */
6053 xmove_fd(pair.wr, STDOUT_FILENO);
6054 re_execute_shell(&to_free, heredoc, NULL, NULL, NULL);
6055#endif
6056 }
6057 /* parent */
6058#if ENABLE_HUSH_FAST
6059 G.count_SIGCHLD++;
6060//bb_error_msg("[%d] fork in setup_heredoc: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6061#endif
6062 enable_restore_tty_pgrp_on_exit();
6063#if !BB_MMU
6064 free(to_free);
6065#endif
6066 close(pair.wr);
6067 free(expanded);
6068 wait(NULL); /* wait till child has died */
6069}
6070
6071/* squirrel != NULL means we squirrel away copies of stdin, stdout,
6072 * and stderr if they are redirected. */
6073static int setup_redirects(struct command *prog, int squirrel[])
6074{
6075 int openfd, mode;
6076 struct redir_struct *redir;
6077
6078 for (redir = prog->redirects; redir; redir = redir->next) {
6079 if (redir->rd_type == REDIRECT_HEREDOC2) {
6080 /* rd_fd<<HERE case */
6081 if (squirrel && redir->rd_fd < 3
6082 && squirrel[redir->rd_fd] < 0
6083 ) {
6084 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6085 }
6086 /* for REDIRECT_HEREDOC2, rd_filename holds _contents_
6087 * of the heredoc */
6088 debug_printf_parse("set heredoc '%s'\n",
6089 redir->rd_filename);
6090 setup_heredoc(redir);
6091 continue;
6092 }
6093
6094 if (redir->rd_dup == REDIRFD_TO_FILE) {
6095 /* rd_fd<*>file case (<*> is <,>,>>,<>) */
6096 char *p;
6097 if (redir->rd_filename == NULL) {
6098 /* Something went wrong in the parse.
6099 * Pretend it didn't happen */
6100 bb_error_msg("bug in redirect parse");
6101 continue;
6102 }
6103 mode = redir_table[redir->rd_type].mode;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02006104 p = expand_string_to_string(redir->rd_filename, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006105 openfd = open_or_warn(p, mode);
6106 free(p);
6107 if (openfd < 0) {
6108 /* this could get lost if stderr has been redirected, but
6109 * bash and ash both lose it as well (though zsh doesn't!) */
6110//what the above comment tries to say?
6111 return 1;
6112 }
6113 } else {
6114 /* rd_fd<*>rd_dup or rd_fd<*>- cases */
6115 openfd = redir->rd_dup;
6116 }
6117
6118 if (openfd != redir->rd_fd) {
6119 if (squirrel && redir->rd_fd < 3
6120 && squirrel[redir->rd_fd] < 0
6121 ) {
6122 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6123 }
6124 if (openfd == REDIRFD_CLOSE) {
6125 /* "n>-" means "close me" */
6126 close(redir->rd_fd);
6127 } else {
6128 xdup2(openfd, redir->rd_fd);
6129 if (redir->rd_dup == REDIRFD_TO_FILE)
6130 close(openfd);
6131 }
6132 }
6133 }
6134 return 0;
6135}
6136
6137static void restore_redirects(int squirrel[])
6138{
6139 int i, fd;
6140 for (i = 0; i < 3; i++) {
6141 fd = squirrel[i];
6142 if (fd != -1) {
6143 /* We simply die on error */
6144 xmove_fd(fd, i);
6145 }
6146 }
6147}
6148
6149static char *find_in_path(const char *arg)
6150{
6151 char *ret = NULL;
6152 const char *PATH = get_local_var_value("PATH");
6153
6154 if (!PATH)
6155 return NULL;
6156
6157 while (1) {
6158 const char *end = strchrnul(PATH, ':');
6159 int sz = end - PATH; /* must be int! */
6160
6161 free(ret);
6162 if (sz != 0) {
6163 ret = xasprintf("%.*s/%s", sz, PATH, arg);
6164 } else {
6165 /* We have xxx::yyyy in $PATH,
6166 * it means "use current dir" */
6167 ret = xstrdup(arg);
6168 }
6169 if (access(ret, F_OK) == 0)
6170 break;
6171
6172 if (*end == '\0') {
6173 free(ret);
6174 return NULL;
6175 }
6176 PATH = end + 1;
6177 }
6178
6179 return ret;
6180}
6181
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006182static const struct built_in_command *find_builtin_helper(const char *name,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006183 const struct built_in_command *x,
6184 const struct built_in_command *end)
6185{
6186 while (x != end) {
6187 if (strcmp(name, x->b_cmd) != 0) {
6188 x++;
6189 continue;
6190 }
6191 debug_printf_exec("found builtin '%s'\n", name);
6192 return x;
6193 }
6194 return NULL;
6195}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006196static const struct built_in_command *find_builtin1(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006197{
6198 return find_builtin_helper(name, bltins1, &bltins1[ARRAY_SIZE(bltins1)]);
6199}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006200static const struct built_in_command *find_builtin(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006201{
6202 const struct built_in_command *x = find_builtin1(name);
6203 if (x)
6204 return x;
6205 return find_builtin_helper(name, bltins2, &bltins2[ARRAY_SIZE(bltins2)]);
6206}
6207
6208#if ENABLE_HUSH_FUNCTIONS
6209static struct function **find_function_slot(const char *name)
6210{
6211 struct function **funcpp = &G.top_func;
6212 while (*funcpp) {
6213 if (strcmp(name, (*funcpp)->name) == 0) {
6214 break;
6215 }
6216 funcpp = &(*funcpp)->next;
6217 }
6218 return funcpp;
6219}
6220
6221static const struct function *find_function(const char *name)
6222{
6223 const struct function *funcp = *find_function_slot(name);
6224 if (funcp)
6225 debug_printf_exec("found function '%s'\n", name);
6226 return funcp;
6227}
6228
6229/* Note: takes ownership on name ptr */
6230static struct function *new_function(char *name)
6231{
6232 struct function **funcpp = find_function_slot(name);
6233 struct function *funcp = *funcpp;
6234
6235 if (funcp != NULL) {
6236 struct command *cmd = funcp->parent_cmd;
6237 debug_printf_exec("func %p parent_cmd %p\n", funcp, cmd);
6238 if (!cmd) {
6239 debug_printf_exec("freeing & replacing function '%s'\n", funcp->name);
6240 free(funcp->name);
6241 /* Note: if !funcp->body, do not free body_as_string!
6242 * This is a special case of "-F name body" function:
6243 * body_as_string was not malloced! */
6244 if (funcp->body) {
6245 free_pipe_list(funcp->body);
6246# if !BB_MMU
6247 free(funcp->body_as_string);
6248# endif
6249 }
6250 } else {
6251 debug_printf_exec("reinserting in tree & replacing function '%s'\n", funcp->name);
6252 cmd->argv[0] = funcp->name;
6253 cmd->group = funcp->body;
6254# if !BB_MMU
6255 cmd->group_as_string = funcp->body_as_string;
6256# endif
6257 }
6258 } else {
6259 debug_printf_exec("remembering new function '%s'\n", name);
6260 funcp = *funcpp = xzalloc(sizeof(*funcp));
6261 /*funcp->next = NULL;*/
6262 }
6263
6264 funcp->name = name;
6265 return funcp;
6266}
6267
6268static void unset_func(const char *name)
6269{
6270 struct function **funcpp = find_function_slot(name);
6271 struct function *funcp = *funcpp;
6272
6273 if (funcp != NULL) {
6274 debug_printf_exec("freeing function '%s'\n", funcp->name);
6275 *funcpp = funcp->next;
6276 /* funcp is unlinked now, deleting it.
6277 * Note: if !funcp->body, the function was created by
6278 * "-F name body", do not free ->body_as_string
6279 * and ->name as they were not malloced. */
6280 if (funcp->body) {
6281 free_pipe_list(funcp->body);
6282 free(funcp->name);
6283# if !BB_MMU
6284 free(funcp->body_as_string);
6285# endif
6286 }
6287 free(funcp);
6288 }
6289}
6290
6291# if BB_MMU
6292#define exec_function(to_free, funcp, argv) \
6293 exec_function(funcp, argv)
6294# endif
6295static void exec_function(char ***to_free,
6296 const struct function *funcp,
6297 char **argv) NORETURN;
6298static void exec_function(char ***to_free,
6299 const struct function *funcp,
6300 char **argv)
6301{
6302# if BB_MMU
6303 int n = 1;
6304
6305 argv[0] = G.global_argv[0];
6306 G.global_argv = argv;
6307 while (*++argv)
6308 n++;
6309 G.global_argc = n;
6310 /* On MMU, funcp->body is always non-NULL */
6311 n = run_list(funcp->body);
6312 fflush_all();
6313 _exit(n);
6314# else
6315 re_execute_shell(to_free,
6316 funcp->body_as_string,
6317 G.global_argv[0],
6318 argv + 1,
6319 NULL);
6320# endif
6321}
6322
6323static int run_function(const struct function *funcp, char **argv)
6324{
6325 int rc;
6326 save_arg_t sv;
6327 smallint sv_flg;
6328
6329 save_and_replace_G_args(&sv, argv);
6330
6331 /* "we are in function, ok to use return" */
6332 sv_flg = G.flag_return_in_progress;
6333 G.flag_return_in_progress = -1;
6334# if ENABLE_HUSH_LOCAL
6335 G.func_nest_level++;
6336# endif
6337
6338 /* On MMU, funcp->body is always non-NULL */
6339# if !BB_MMU
6340 if (!funcp->body) {
6341 /* Function defined by -F */
6342 parse_and_run_string(funcp->body_as_string);
6343 rc = G.last_exitcode;
6344 } else
6345# endif
6346 {
6347 rc = run_list(funcp->body);
6348 }
6349
6350# if ENABLE_HUSH_LOCAL
6351 {
6352 struct variable *var;
6353 struct variable **var_pp;
6354
6355 var_pp = &G.top_var;
6356 while ((var = *var_pp) != NULL) {
6357 if (var->func_nest_level < G.func_nest_level) {
6358 var_pp = &var->next;
6359 continue;
6360 }
6361 /* Unexport */
6362 if (var->flg_export)
6363 bb_unsetenv(var->varstr);
6364 /* Remove from global list */
6365 *var_pp = var->next;
6366 /* Free */
6367 if (!var->max_len)
6368 free(var->varstr);
6369 free(var);
6370 }
6371 G.func_nest_level--;
6372 }
6373# endif
6374 G.flag_return_in_progress = sv_flg;
6375
6376 restore_G_args(&sv, argv);
6377
6378 return rc;
6379}
6380#endif /* ENABLE_HUSH_FUNCTIONS */
6381
6382
6383#if BB_MMU
6384#define exec_builtin(to_free, x, argv) \
6385 exec_builtin(x, argv)
6386#else
6387#define exec_builtin(to_free, x, argv) \
6388 exec_builtin(to_free, argv)
6389#endif
6390static void exec_builtin(char ***to_free,
6391 const struct built_in_command *x,
6392 char **argv) NORETURN;
6393static void exec_builtin(char ***to_free,
6394 const struct built_in_command *x,
6395 char **argv)
6396{
6397#if BB_MMU
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006398 int rcode;
6399 fflush_all();
6400 rcode = x->b_function(argv);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006401 fflush_all();
6402 _exit(rcode);
6403#else
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006404 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006405 /* On NOMMU, we must never block!
6406 * Example: { sleep 99 | read line; } & echo Ok
6407 */
6408 re_execute_shell(to_free,
6409 argv[0],
6410 G.global_argv[0],
6411 G.global_argv + 1,
6412 argv);
6413#endif
6414}
6415
6416
6417static void execvp_or_die(char **argv) NORETURN;
6418static void execvp_or_die(char **argv)
6419{
6420 debug_printf_exec("execing '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006421 /* Don't propagate SIG_IGN to the child */
6422 if (SPECIAL_JOBSTOP_SIGS != 0)
6423 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006424 execvp(argv[0], argv);
6425 bb_perror_msg("can't execute '%s'", argv[0]);
6426 _exit(127); /* bash compat */
6427}
6428
6429#if ENABLE_HUSH_MODE_X
6430static void dump_cmd_in_x_mode(char **argv)
6431{
6432 if (G_x_mode && argv) {
6433 /* We want to output the line in one write op */
6434 char *buf, *p;
6435 int len;
6436 int n;
6437
6438 len = 3;
6439 n = 0;
6440 while (argv[n])
6441 len += strlen(argv[n++]) + 1;
6442 buf = xmalloc(len);
6443 buf[0] = '+';
6444 p = buf + 1;
6445 n = 0;
6446 while (argv[n])
6447 p += sprintf(p, " %s", argv[n++]);
6448 *p++ = '\n';
6449 *p = '\0';
6450 fputs(buf, stderr);
6451 free(buf);
6452 }
6453}
6454#else
6455# define dump_cmd_in_x_mode(argv) ((void)0)
6456#endif
6457
6458#if BB_MMU
6459#define pseudo_exec_argv(nommu_save, argv, assignment_cnt, argv_expanded) \
6460 pseudo_exec_argv(argv, assignment_cnt, argv_expanded)
6461#define pseudo_exec(nommu_save, command, argv_expanded) \
6462 pseudo_exec(command, argv_expanded)
6463#endif
6464
6465/* Called after [v]fork() in run_pipe, or from builtin_exec.
6466 * Never returns.
6467 * Don't exit() here. If you don't exec, use _exit instead.
6468 * The at_exit handlers apparently confuse the calling process,
6469 * in particular stdin handling. Not sure why? -- because of vfork! (vda) */
6470static void pseudo_exec_argv(nommu_save_t *nommu_save,
6471 char **argv, int assignment_cnt,
6472 char **argv_expanded) NORETURN;
6473static NOINLINE void pseudo_exec_argv(nommu_save_t *nommu_save,
6474 char **argv, int assignment_cnt,
6475 char **argv_expanded)
6476{
6477 char **new_env;
6478
6479 new_env = expand_assignments(argv, assignment_cnt);
6480 dump_cmd_in_x_mode(new_env);
6481
6482 if (!argv[assignment_cnt]) {
6483 /* Case when we are here: ... | var=val | ...
6484 * (note that we do not exit early, i.e., do not optimize out
6485 * expand_assignments(): think about ... | var=`sleep 1` | ...
6486 */
6487 free_strings(new_env);
6488 _exit(EXIT_SUCCESS);
6489 }
6490
6491#if BB_MMU
6492 set_vars_and_save_old(new_env);
6493 free(new_env); /* optional */
6494 /* we can also destroy set_vars_and_save_old's return value,
6495 * to save memory */
6496#else
6497 nommu_save->new_env = new_env;
6498 nommu_save->old_vars = set_vars_and_save_old(new_env);
6499#endif
6500
6501 if (argv_expanded) {
6502 argv = argv_expanded;
6503 } else {
6504 argv = expand_strvec_to_strvec(argv + assignment_cnt);
6505#if !BB_MMU
6506 nommu_save->argv = argv;
6507#endif
6508 }
6509 dump_cmd_in_x_mode(argv);
6510
6511#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6512 if (strchr(argv[0], '/') != NULL)
6513 goto skip;
6514#endif
6515
6516 /* Check if the command matches any of the builtins.
6517 * Depending on context, this might be redundant. But it's
6518 * easier to waste a few CPU cycles than it is to figure out
6519 * if this is one of those cases.
6520 */
6521 {
6522 /* On NOMMU, it is more expensive to re-execute shell
6523 * just in order to run echo or test builtin.
6524 * It's better to skip it here and run corresponding
6525 * non-builtin later. */
6526 const struct built_in_command *x;
6527 x = BB_MMU ? find_builtin(argv[0]) : find_builtin1(argv[0]);
6528 if (x) {
6529 exec_builtin(&nommu_save->argv_from_re_execing, x, argv);
6530 }
6531 }
6532#if ENABLE_HUSH_FUNCTIONS
6533 /* Check if the command matches any functions */
6534 {
6535 const struct function *funcp = find_function(argv[0]);
6536 if (funcp) {
6537 exec_function(&nommu_save->argv_from_re_execing, funcp, argv);
6538 }
6539 }
6540#endif
6541
6542#if ENABLE_FEATURE_SH_STANDALONE
6543 /* Check if the command matches any busybox applets */
6544 {
6545 int a = find_applet_by_name(argv[0]);
6546 if (a >= 0) {
6547# if BB_MMU /* see above why on NOMMU it is not allowed */
6548 if (APPLET_IS_NOEXEC(a)) {
6549 debug_printf_exec("running applet '%s'\n", argv[0]);
6550 run_applet_no_and_exit(a, argv);
6551 }
6552# endif
6553 /* Re-exec ourselves */
6554 debug_printf_exec("re-execing applet '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006555 /* Don't propagate SIG_IGN to the child */
6556 if (SPECIAL_JOBSTOP_SIGS != 0)
6557 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006558 execv(bb_busybox_exec_path, argv);
6559 /* If they called chroot or otherwise made the binary no longer
6560 * executable, fall through */
6561 }
6562 }
6563#endif
6564
6565#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6566 skip:
6567#endif
6568 execvp_or_die(argv);
6569}
6570
6571/* Called after [v]fork() in run_pipe
6572 */
6573static void pseudo_exec(nommu_save_t *nommu_save,
6574 struct command *command,
6575 char **argv_expanded) NORETURN;
6576static void pseudo_exec(nommu_save_t *nommu_save,
6577 struct command *command,
6578 char **argv_expanded)
6579{
6580 if (command->argv) {
6581 pseudo_exec_argv(nommu_save, command->argv,
6582 command->assignment_cnt, argv_expanded);
6583 }
6584
6585 if (command->group) {
6586 /* Cases when we are here:
6587 * ( list )
6588 * { list } &
6589 * ... | ( list ) | ...
6590 * ... | { list } | ...
6591 */
6592#if BB_MMU
6593 int rcode;
6594 debug_printf_exec("pseudo_exec: run_list\n");
6595 reset_traps_to_defaults();
6596 rcode = run_list(command->group);
6597 /* OK to leak memory by not calling free_pipe_list,
6598 * since this process is about to exit */
6599 _exit(rcode);
6600#else
6601 re_execute_shell(&nommu_save->argv_from_re_execing,
6602 command->group_as_string,
6603 G.global_argv[0],
6604 G.global_argv + 1,
6605 NULL);
6606#endif
6607 }
6608
6609 /* Case when we are here: ... | >file */
6610 debug_printf_exec("pseudo_exec'ed null command\n");
6611 _exit(EXIT_SUCCESS);
6612}
6613
6614#if ENABLE_HUSH_JOB
6615static const char *get_cmdtext(struct pipe *pi)
6616{
6617 char **argv;
6618 char *p;
6619 int len;
6620
6621 /* This is subtle. ->cmdtext is created only on first backgrounding.
6622 * (Think "cat, <ctrl-z>, fg, <ctrl-z>, fg, <ctrl-z>...." here...)
6623 * On subsequent bg argv is trashed, but we won't use it */
6624 if (pi->cmdtext)
6625 return pi->cmdtext;
6626 argv = pi->cmds[0].argv;
6627 if (!argv || !argv[0]) {
6628 pi->cmdtext = xzalloc(1);
6629 return pi->cmdtext;
6630 }
6631
6632 len = 0;
6633 do {
6634 len += strlen(*argv) + 1;
6635 } while (*++argv);
6636 p = xmalloc(len);
6637 pi->cmdtext = p;
6638 argv = pi->cmds[0].argv;
6639 do {
6640 len = strlen(*argv);
6641 memcpy(p, *argv, len);
6642 p += len;
6643 *p++ = ' ';
6644 } while (*++argv);
6645 p[-1] = '\0';
6646 return pi->cmdtext;
6647}
6648
6649static void insert_bg_job(struct pipe *pi)
6650{
6651 struct pipe *job, **jobp;
6652 int i;
6653
6654 /* Linear search for the ID of the job to use */
6655 pi->jobid = 1;
6656 for (job = G.job_list; job; job = job->next)
6657 if (job->jobid >= pi->jobid)
6658 pi->jobid = job->jobid + 1;
6659
6660 /* Add job to the list of running jobs */
6661 jobp = &G.job_list;
6662 while ((job = *jobp) != NULL)
6663 jobp = &job->next;
6664 job = *jobp = xmalloc(sizeof(*job));
6665
6666 *job = *pi; /* physical copy */
6667 job->next = NULL;
6668 job->cmds = xzalloc(sizeof(pi->cmds[0]) * pi->num_cmds);
6669 /* Cannot copy entire pi->cmds[] vector! This causes double frees */
6670 for (i = 0; i < pi->num_cmds; i++) {
6671 job->cmds[i].pid = pi->cmds[i].pid;
6672 /* all other fields are not used and stay zero */
6673 }
6674 job->cmdtext = xstrdup(get_cmdtext(pi));
6675
6676 if (G_interactive_fd)
6677 printf("[%d] %d %s\n", job->jobid, job->cmds[0].pid, job->cmdtext);
6678 G.last_jobid = job->jobid;
6679}
6680
6681static void remove_bg_job(struct pipe *pi)
6682{
6683 struct pipe *prev_pipe;
6684
6685 if (pi == G.job_list) {
6686 G.job_list = pi->next;
6687 } else {
6688 prev_pipe = G.job_list;
6689 while (prev_pipe->next != pi)
6690 prev_pipe = prev_pipe->next;
6691 prev_pipe->next = pi->next;
6692 }
6693 if (G.job_list)
6694 G.last_jobid = G.job_list->jobid;
6695 else
6696 G.last_jobid = 0;
6697}
6698
6699/* Remove a backgrounded job */
6700static void delete_finished_bg_job(struct pipe *pi)
6701{
6702 remove_bg_job(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006703 free_pipe(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006704}
6705#endif /* JOB */
6706
6707/* Check to see if any processes have exited -- if they
6708 * have, figure out why and see if a job has completed */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02006709static int checkjobs(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006710{
6711 int attributes;
6712 int status;
6713#if ENABLE_HUSH_JOB
6714 struct pipe *pi;
6715#endif
6716 pid_t childpid;
6717 int rcode = 0;
6718
6719 debug_printf_jobs("checkjobs %p\n", fg_pipe);
6720
6721 attributes = WUNTRACED;
6722 if (fg_pipe == NULL)
6723 attributes |= WNOHANG;
6724
6725 errno = 0;
6726#if ENABLE_HUSH_FAST
6727 if (G.handled_SIGCHLD == G.count_SIGCHLD) {
6728//bb_error_msg("[%d] checkjobs: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d children?:%d fg_pipe:%p",
6729//getpid(), G.count_SIGCHLD, G.handled_SIGCHLD, G.we_have_children, fg_pipe);
6730 /* There was neither fork nor SIGCHLD since last waitpid */
6731 /* Avoid doing waitpid syscall if possible */
6732 if (!G.we_have_children) {
6733 errno = ECHILD;
6734 return -1;
6735 }
6736 if (fg_pipe == NULL) { /* is WNOHANG set? */
6737 /* We have children, but they did not exit
6738 * or stop yet (we saw no SIGCHLD) */
6739 return 0;
6740 }
6741 /* else: !WNOHANG, waitpid will block, can't short-circuit */
6742 }
6743#endif
6744
6745/* Do we do this right?
6746 * bash-3.00# sleep 20 | false
6747 * <ctrl-Z pressed>
6748 * [3]+ Stopped sleep 20 | false
6749 * bash-3.00# echo $?
6750 * 1 <========== bg pipe is not fully done, but exitcode is already known!
6751 * [hush 1.14.0: yes we do it right]
6752 */
6753 wait_more:
6754 while (1) {
6755 int i;
6756 int dead;
6757
6758#if ENABLE_HUSH_FAST
6759 i = G.count_SIGCHLD;
6760#endif
6761 childpid = waitpid(-1, &status, attributes);
6762 if (childpid <= 0) {
6763 if (childpid && errno != ECHILD)
6764 bb_perror_msg("waitpid");
6765#if ENABLE_HUSH_FAST
6766 else { /* Until next SIGCHLD, waitpid's are useless */
6767 G.we_have_children = (childpid == 0);
6768 G.handled_SIGCHLD = i;
6769//bb_error_msg("[%d] checkjobs: waitpid returned <= 0, G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6770 }
6771#endif
6772 break;
6773 }
6774 dead = WIFEXITED(status) || WIFSIGNALED(status);
6775
6776#if DEBUG_JOBS
6777 if (WIFSTOPPED(status))
6778 debug_printf_jobs("pid %d stopped by sig %d (exitcode %d)\n",
6779 childpid, WSTOPSIG(status), WEXITSTATUS(status));
6780 if (WIFSIGNALED(status))
6781 debug_printf_jobs("pid %d killed by sig %d (exitcode %d)\n",
6782 childpid, WTERMSIG(status), WEXITSTATUS(status));
6783 if (WIFEXITED(status))
6784 debug_printf_jobs("pid %d exited, exitcode %d\n",
6785 childpid, WEXITSTATUS(status));
6786#endif
6787 /* Were we asked to wait for fg pipe? */
6788 if (fg_pipe) {
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006789 i = fg_pipe->num_cmds;
6790 while (--i >= 0) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006791 debug_printf_jobs("check pid %d\n", fg_pipe->cmds[i].pid);
6792 if (fg_pipe->cmds[i].pid != childpid)
6793 continue;
6794 if (dead) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006795 int ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006796 fg_pipe->cmds[i].pid = 0;
6797 fg_pipe->alive_cmds--;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006798 ex = WEXITSTATUS(status);
6799 /* bash prints killer signal's name for *last*
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006800 * process in pipe (prints just newline for SIGINT/SIGPIPE).
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006801 * Mimic this. Example: "sleep 5" + (^\ or kill -QUIT)
6802 */
6803 if (WIFSIGNALED(status)) {
6804 int sig = WTERMSIG(status);
6805 if (i == fg_pipe->num_cmds-1)
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006806 /* TODO: use strsignal() instead for bash compat? but that's bloat... */
Denys Vlasenkod60752f2015-10-07 22:42:45 +02006807 puts(sig == SIGINT || sig == SIGPIPE ? "" : get_signame(sig));
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006808 /* TODO: if (WCOREDUMP(status)) + " (core dumped)"; */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006809 /* TODO: MIPS has 128 sigs (1..128), what if sig==128 here?
6810 * Maybe we need to use sig | 128? */
6811 ex = sig + 128;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006812 }
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006813 fg_pipe->cmds[i].cmd_exitcode = ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006814 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006815 fg_pipe->stopped_cmds++;
6816 }
6817 debug_printf_jobs("fg_pipe: alive_cmds %d stopped_cmds %d\n",
6818 fg_pipe->alive_cmds, fg_pipe->stopped_cmds);
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006819 if (fg_pipe->alive_cmds == fg_pipe->stopped_cmds) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006820 /* All processes in fg pipe have exited or stopped */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006821 i = fg_pipe->num_cmds;
6822 while (--i >= 0) {
6823 rcode = fg_pipe->cmds[i].cmd_exitcode;
6824 /* usually last process gives overall exitstatus,
6825 * but with "set -o pipefail", last *failed* process does */
6826 if (G.o_opt[OPT_O_PIPEFAIL] == 0 || rcode != 0)
6827 break;
6828 }
6829 IF_HAS_KEYWORDS(if (fg_pipe->pi_inverted) rcode = !rcode;)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006830/* Note: *non-interactive* bash does not continue if all processes in fg pipe
6831 * are stopped. Testcase: "cat | cat" in a script (not on command line!)
6832 * and "killall -STOP cat" */
6833 if (G_interactive_fd) {
6834#if ENABLE_HUSH_JOB
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006835 if (fg_pipe->alive_cmds != 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006836 insert_bg_job(fg_pipe);
6837#endif
6838 return rcode;
6839 }
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006840 if (fg_pipe->alive_cmds == 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006841 return rcode;
6842 }
6843 /* There are still running processes in the fg pipe */
6844 goto wait_more; /* do waitpid again */
6845 }
6846 /* it wasnt fg_pipe, look for process in bg pipes */
6847 }
6848
6849#if ENABLE_HUSH_JOB
6850 /* We asked to wait for bg or orphaned children */
6851 /* No need to remember exitcode in this case */
6852 for (pi = G.job_list; pi; pi = pi->next) {
6853 for (i = 0; i < pi->num_cmds; i++) {
6854 if (pi->cmds[i].pid == childpid)
6855 goto found_pi_and_prognum;
6856 }
6857 }
6858 /* Happens when shell is used as init process (init=/bin/sh) */
6859 debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
6860 continue; /* do waitpid again */
6861
6862 found_pi_and_prognum:
6863 if (dead) {
6864 /* child exited */
6865 pi->cmds[i].pid = 0;
6866 pi->alive_cmds--;
6867 if (!pi->alive_cmds) {
6868 if (G_interactive_fd)
6869 printf(JOB_STATUS_FORMAT, pi->jobid,
6870 "Done", pi->cmdtext);
6871 delete_finished_bg_job(pi);
6872 }
6873 } else {
6874 /* child stopped */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006875 pi->stopped_cmds++;
6876 }
6877#endif
6878 } /* while (waitpid succeeds)... */
6879
6880 return rcode;
6881}
6882
6883#if ENABLE_HUSH_JOB
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006884static int checkjobs_and_fg_shell(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006885{
6886 pid_t p;
6887 int rcode = checkjobs(fg_pipe);
6888 if (G_saved_tty_pgrp) {
6889 /* Job finished, move the shell to the foreground */
6890 p = getpgrp(); /* our process group id */
6891 debug_printf_jobs("fg'ing ourself: getpgrp()=%d\n", (int)p);
6892 tcsetpgrp(G_interactive_fd, p);
6893 }
6894 return rcode;
6895}
6896#endif
6897
6898/* Start all the jobs, but don't wait for anything to finish.
6899 * See checkjobs().
6900 *
6901 * Return code is normally -1, when the caller has to wait for children
6902 * to finish to determine the exit status of the pipe. If the pipe
6903 * is a simple builtin command, however, the action is done by the
6904 * time run_pipe returns, and the exit code is provided as the
6905 * return value.
6906 *
6907 * Returns -1 only if started some children. IOW: we have to
6908 * mask out retvals of builtins etc with 0xff!
6909 *
6910 * The only case when we do not need to [v]fork is when the pipe
6911 * is single, non-backgrounded, non-subshell command. Examples:
6912 * cmd ; ... { list } ; ...
6913 * cmd && ... { list } && ...
6914 * cmd || ... { list } || ...
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01006915 * If it is, then we can run cmd as a builtin, NOFORK,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006916 * or (if SH_STANDALONE) an applet, and we can run the { list }
6917 * with run_list. If it isn't one of these, we fork and exec cmd.
6918 *
6919 * Cases when we must fork:
6920 * non-single: cmd | cmd
6921 * backgrounded: cmd & { list } &
6922 * subshell: ( list ) [&]
6923 */
6924#if !ENABLE_HUSH_MODE_X
Denys Vlasenko26777aa2010-11-22 23:49:10 +01006925#define redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel, argv_expanded) \
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006926 redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel)
6927#endif
6928static int redirect_and_varexp_helper(char ***new_env_p,
6929 struct variable **old_vars_p,
6930 struct command *command,
6931 int squirrel[3],
6932 char **argv_expanded)
6933{
6934 /* setup_redirects acts on file descriptors, not FILEs.
6935 * This is perfect for work that comes after exec().
6936 * Is it really safe for inline use? Experimentally,
6937 * things seem to work. */
6938 int rcode = setup_redirects(command, squirrel);
6939 if (rcode == 0) {
6940 char **new_env = expand_assignments(command->argv, command->assignment_cnt);
6941 *new_env_p = new_env;
6942 dump_cmd_in_x_mode(new_env);
6943 dump_cmd_in_x_mode(argv_expanded);
6944 if (old_vars_p)
6945 *old_vars_p = set_vars_and_save_old(new_env);
6946 }
6947 return rcode;
6948}
6949static NOINLINE int run_pipe(struct pipe *pi)
6950{
6951 static const char *const null_ptr = NULL;
6952
6953 int cmd_no;
6954 int next_infd;
6955 struct command *command;
6956 char **argv_expanded;
6957 char **argv;
6958 /* it is not always needed, but we aim to smaller code */
6959 int squirrel[] = { -1, -1, -1 };
6960 int rcode;
6961
6962 debug_printf_exec("run_pipe start: members:%d\n", pi->num_cmds);
6963 debug_enter();
6964
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02006965 /* Testcase: set -- q w e; (IFS='' echo "$*"; IFS=''; echo "$*"); echo "$*"
6966 * Result should be 3 lines: q w e, qwe, q w e
6967 */
6968 G.ifs = get_local_var_value("IFS");
6969 if (!G.ifs)
6970 G.ifs = defifs;
6971
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006972 IF_HUSH_JOB(pi->pgrp = -1;)
6973 pi->stopped_cmds = 0;
6974 command = &pi->cmds[0];
6975 argv_expanded = NULL;
6976
6977 if (pi->num_cmds != 1
6978 || pi->followup == PIPE_BG
6979 || command->cmd_type == CMD_SUBSHELL
6980 ) {
6981 goto must_fork;
6982 }
6983
6984 pi->alive_cmds = 1;
6985
6986 debug_printf_exec(": group:%p argv:'%s'\n",
6987 command->group, command->argv ? command->argv[0] : "NONE");
6988
6989 if (command->group) {
6990#if ENABLE_HUSH_FUNCTIONS
6991 if (command->cmd_type == CMD_FUNCDEF) {
6992 /* "executing" func () { list } */
6993 struct function *funcp;
6994
6995 funcp = new_function(command->argv[0]);
6996 /* funcp->name is already set to argv[0] */
6997 funcp->body = command->group;
6998# if !BB_MMU
6999 funcp->body_as_string = command->group_as_string;
7000 command->group_as_string = NULL;
7001# endif
7002 command->group = NULL;
7003 command->argv[0] = NULL;
7004 debug_printf_exec("cmd %p has child func at %p\n", command, funcp);
7005 funcp->parent_cmd = command;
7006 command->child_func = funcp;
7007
7008 debug_printf_exec("run_pipe: return EXIT_SUCCESS\n");
7009 debug_leave();
7010 return EXIT_SUCCESS;
7011 }
7012#endif
7013 /* { list } */
7014 debug_printf("non-subshell group\n");
7015 rcode = 1; /* exitcode if redir failed */
7016 if (setup_redirects(command, squirrel) == 0) {
7017 debug_printf_exec(": run_list\n");
7018 rcode = run_list(command->group) & 0xff;
7019 }
7020 restore_redirects(squirrel);
7021 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7022 debug_leave();
7023 debug_printf_exec("run_pipe: return %d\n", rcode);
7024 return rcode;
7025 }
7026
7027 argv = command->argv ? command->argv : (char **) &null_ptr;
7028 {
7029 const struct built_in_command *x;
7030#if ENABLE_HUSH_FUNCTIONS
7031 const struct function *funcp;
7032#else
7033 enum { funcp = 0 };
7034#endif
7035 char **new_env = NULL;
7036 struct variable *old_vars = NULL;
7037
7038 if (argv[command->assignment_cnt] == NULL) {
7039 /* Assignments, but no command */
7040 /* Ensure redirects take effect (that is, create files).
7041 * Try "a=t >file" */
7042#if 0 /* A few cases in testsuite fail with this code. FIXME */
7043 rcode = redirect_and_varexp_helper(&new_env, /*old_vars:*/ NULL, command, squirrel, /*argv_expanded:*/ NULL);
7044 /* Set shell variables */
7045 if (new_env) {
7046 argv = new_env;
7047 while (*argv) {
7048 set_local_var(*argv, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7049 /* Do we need to flag set_local_var() errors?
7050 * "assignment to readonly var" and "putenv error"
7051 */
7052 argv++;
7053 }
7054 }
7055 /* Redirect error sets $? to 1. Otherwise,
7056 * if evaluating assignment value set $?, retain it.
7057 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7058 if (rcode == 0)
7059 rcode = G.last_exitcode;
7060 /* Exit, _skipping_ variable restoring code: */
7061 goto clean_up_and_ret0;
7062
7063#else /* Older, bigger, but more correct code */
7064
7065 rcode = setup_redirects(command, squirrel);
7066 restore_redirects(squirrel);
7067 /* Set shell variables */
7068 if (G_x_mode)
7069 bb_putchar_stderr('+');
7070 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007071 char *p = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007072 if (G_x_mode)
7073 fprintf(stderr, " %s", p);
7074 debug_printf_exec("set shell var:'%s'->'%s'\n",
7075 *argv, p);
7076 set_local_var(p, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7077 /* Do we need to flag set_local_var() errors?
7078 * "assignment to readonly var" and "putenv error"
7079 */
7080 argv++;
7081 }
7082 if (G_x_mode)
7083 bb_putchar_stderr('\n');
7084 /* Redirect error sets $? to 1. Otherwise,
7085 * if evaluating assignment value set $?, retain it.
7086 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7087 if (rcode == 0)
7088 rcode = G.last_exitcode;
7089 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7090 debug_leave();
7091 debug_printf_exec("run_pipe: return %d\n", rcode);
7092 return rcode;
7093#endif
7094 }
7095
7096 /* Expand the rest into (possibly) many strings each */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007097#if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007098 if (command->cmd_type == CMD_SINGLEWORD_NOGLOB) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007099 argv_expanded = expand_strvec_to_strvec_singleword_noglob(argv + command->assignment_cnt);
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007100 } else
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007101#endif
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007102 {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007103 argv_expanded = expand_strvec_to_strvec(argv + command->assignment_cnt);
7104 }
7105
7106 /* if someone gives us an empty string: `cmd with empty output` */
7107 if (!argv_expanded[0]) {
7108 free(argv_expanded);
7109 debug_leave();
7110 return G.last_exitcode;
7111 }
7112
7113 x = find_builtin(argv_expanded[0]);
7114#if ENABLE_HUSH_FUNCTIONS
7115 funcp = NULL;
7116 if (!x)
7117 funcp = find_function(argv_expanded[0]);
7118#endif
7119 if (x || funcp) {
7120 if (!funcp) {
7121 if (x->b_function == builtin_exec && argv_expanded[1] == NULL) {
7122 debug_printf("exec with redirects only\n");
7123 rcode = setup_redirects(command, NULL);
7124 goto clean_up_and_ret1;
7125 }
7126 }
7127 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7128 if (rcode == 0) {
7129 if (!funcp) {
7130 debug_printf_exec(": builtin '%s' '%s'...\n",
7131 x->b_cmd, argv_expanded[1]);
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007132 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007133 rcode = x->b_function(argv_expanded) & 0xff;
7134 fflush_all();
7135 }
7136#if ENABLE_HUSH_FUNCTIONS
7137 else {
7138# if ENABLE_HUSH_LOCAL
7139 struct variable **sv;
7140 sv = G.shadowed_vars_pp;
7141 G.shadowed_vars_pp = &old_vars;
7142# endif
7143 debug_printf_exec(": function '%s' '%s'...\n",
7144 funcp->name, argv_expanded[1]);
7145 rcode = run_function(funcp, argv_expanded) & 0xff;
7146# if ENABLE_HUSH_LOCAL
7147 G.shadowed_vars_pp = sv;
7148# endif
7149 }
7150#endif
7151 }
7152 clean_up_and_ret:
7153 unset_vars(new_env);
7154 add_vars(old_vars);
7155/* clean_up_and_ret0: */
7156 restore_redirects(squirrel);
7157 clean_up_and_ret1:
7158 free(argv_expanded);
7159 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7160 debug_leave();
7161 debug_printf_exec("run_pipe return %d\n", rcode);
7162 return rcode;
7163 }
7164
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007165 if (ENABLE_FEATURE_SH_NOFORK) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007166 int n = find_applet_by_name(argv_expanded[0]);
7167 if (n >= 0 && APPLET_IS_NOFORK(n)) {
7168 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7169 if (rcode == 0) {
7170 debug_printf_exec(": run_nofork_applet '%s' '%s'...\n",
7171 argv_expanded[0], argv_expanded[1]);
7172 rcode = run_nofork_applet(n, argv_expanded);
7173 }
7174 goto clean_up_and_ret;
7175 }
7176 }
7177 /* It is neither builtin nor applet. We must fork. */
7178 }
7179
7180 must_fork:
7181 /* NB: argv_expanded may already be created, and that
7182 * might include `cmd` runs! Do not rerun it! We *must*
7183 * use argv_expanded if it's non-NULL */
7184
7185 /* Going to fork a child per each pipe member */
7186 pi->alive_cmds = 0;
7187 next_infd = 0;
7188
7189 cmd_no = 0;
7190 while (cmd_no < pi->num_cmds) {
7191 struct fd_pair pipefds;
7192#if !BB_MMU
7193 volatile nommu_save_t nommu_save;
7194 nommu_save.new_env = NULL;
7195 nommu_save.old_vars = NULL;
7196 nommu_save.argv = NULL;
7197 nommu_save.argv_from_re_execing = NULL;
7198#endif
7199 command = &pi->cmds[cmd_no];
7200 cmd_no++;
7201 if (command->argv) {
7202 debug_printf_exec(": pipe member '%s' '%s'...\n",
7203 command->argv[0], command->argv[1]);
7204 } else {
7205 debug_printf_exec(": pipe member with no argv\n");
7206 }
7207
7208 /* pipes are inserted between pairs of commands */
7209 pipefds.rd = 0;
7210 pipefds.wr = 1;
7211 if (cmd_no < pi->num_cmds)
7212 xpiped_pair(pipefds);
7213
7214 command->pid = BB_MMU ? fork() : vfork();
7215 if (!command->pid) { /* child */
7216#if ENABLE_HUSH_JOB
7217 disable_restore_tty_pgrp_on_exit();
7218 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
7219
7220 /* Every child adds itself to new process group
7221 * with pgid == pid_of_first_child_in_pipe */
7222 if (G.run_list_level == 1 && G_interactive_fd) {
7223 pid_t pgrp;
7224 pgrp = pi->pgrp;
7225 if (pgrp < 0) /* true for 1st process only */
7226 pgrp = getpid();
7227 if (setpgid(0, pgrp) == 0
7228 && pi->followup != PIPE_BG
7229 && G_saved_tty_pgrp /* we have ctty */
7230 ) {
7231 /* We do it in *every* child, not just first,
7232 * to avoid races */
7233 tcsetpgrp(G_interactive_fd, pgrp);
7234 }
7235 }
7236#endif
7237 if (pi->alive_cmds == 0 && pi->followup == PIPE_BG) {
7238 /* 1st cmd in backgrounded pipe
7239 * should have its stdin /dev/null'ed */
7240 close(0);
7241 if (open(bb_dev_null, O_RDONLY))
7242 xopen("/", O_RDONLY);
7243 } else {
7244 xmove_fd(next_infd, 0);
7245 }
7246 xmove_fd(pipefds.wr, 1);
7247 if (pipefds.rd > 1)
7248 close(pipefds.rd);
7249 /* Like bash, explicit redirects override pipes,
7250 * and the pipe fd is available for dup'ing. */
7251 if (setup_redirects(command, NULL))
7252 _exit(1);
7253
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007254 /* Stores to nommu_save list of env vars putenv'ed
7255 * (NOMMU, on MMU we don't need that) */
7256 /* cast away volatility... */
7257 pseudo_exec((nommu_save_t*) &nommu_save, command, argv_expanded);
7258 /* pseudo_exec() does not return */
7259 }
7260
7261 /* parent or error */
7262#if ENABLE_HUSH_FAST
7263 G.count_SIGCHLD++;
7264//bb_error_msg("[%d] fork in run_pipe: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
7265#endif
7266 enable_restore_tty_pgrp_on_exit();
7267#if !BB_MMU
7268 /* Clean up after vforked child */
7269 free(nommu_save.argv);
7270 free(nommu_save.argv_from_re_execing);
7271 unset_vars(nommu_save.new_env);
7272 add_vars(nommu_save.old_vars);
7273#endif
7274 free(argv_expanded);
7275 argv_expanded = NULL;
7276 if (command->pid < 0) { /* [v]fork failed */
7277 /* Clearly indicate, was it fork or vfork */
7278 bb_perror_msg(BB_MMU ? "vfork"+1 : "vfork");
7279 } else {
7280 pi->alive_cmds++;
7281#if ENABLE_HUSH_JOB
7282 /* Second and next children need to know pid of first one */
7283 if (pi->pgrp < 0)
7284 pi->pgrp = command->pid;
7285#endif
7286 }
7287
7288 if (cmd_no > 1)
7289 close(next_infd);
7290 if (cmd_no < pi->num_cmds)
7291 close(pipefds.wr);
7292 /* Pass read (output) pipe end to next iteration */
7293 next_infd = pipefds.rd;
7294 }
7295
7296 if (!pi->alive_cmds) {
7297 debug_leave();
7298 debug_printf_exec("run_pipe return 1 (all forks failed, no children)\n");
7299 return 1;
7300 }
7301
7302 debug_leave();
7303 debug_printf_exec("run_pipe return -1 (%u children started)\n", pi->alive_cmds);
7304 return -1;
7305}
7306
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007307/* NB: called by pseudo_exec, and therefore must not modify any
7308 * global data until exec/_exit (we can be a child after vfork!) */
7309static int run_list(struct pipe *pi)
7310{
7311#if ENABLE_HUSH_CASE
7312 char *case_word = NULL;
7313#endif
7314#if ENABLE_HUSH_LOOPS
7315 struct pipe *loop_top = NULL;
7316 char **for_lcur = NULL;
7317 char **for_list = NULL;
7318#endif
7319 smallint last_followup;
7320 smalluint rcode;
7321#if ENABLE_HUSH_IF || ENABLE_HUSH_CASE
7322 smalluint cond_code = 0;
7323#else
7324 enum { cond_code = 0 };
7325#endif
7326#if HAS_KEYWORDS
Denys Vlasenko9b782552010-09-08 13:33:26 +02007327 smallint rword; /* RES_foo */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007328 smallint last_rword; /* ditto */
7329#endif
7330
7331 debug_printf_exec("run_list start lvl %d\n", G.run_list_level);
7332 debug_enter();
7333
7334#if ENABLE_HUSH_LOOPS
7335 /* Check syntax for "for" */
Denys Vlasenko0d6a4ec2010-12-18 01:34:49 +01007336 {
7337 struct pipe *cpipe;
7338 for (cpipe = pi; cpipe; cpipe = cpipe->next) {
7339 if (cpipe->res_word != RES_FOR && cpipe->res_word != RES_IN)
7340 continue;
7341 /* current word is FOR or IN (BOLD in comments below) */
7342 if (cpipe->next == NULL) {
7343 syntax_error("malformed for");
7344 debug_leave();
7345 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7346 return 1;
7347 }
7348 /* "FOR v; do ..." and "for v IN a b; do..." are ok */
7349 if (cpipe->next->res_word == RES_DO)
7350 continue;
7351 /* next word is not "do". It must be "in" then ("FOR v in ...") */
7352 if (cpipe->res_word == RES_IN /* "for v IN a b; not_do..."? */
7353 || cpipe->next->res_word != RES_IN /* FOR v not_do_and_not_in..."? */
7354 ) {
7355 syntax_error("malformed for");
7356 debug_leave();
7357 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7358 return 1;
7359 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007360 }
7361 }
7362#endif
7363
7364 /* Past this point, all code paths should jump to ret: label
7365 * in order to return, no direct "return" statements please.
7366 * This helps to ensure that no memory is leaked. */
7367
7368#if ENABLE_HUSH_JOB
7369 G.run_list_level++;
7370#endif
7371
7372#if HAS_KEYWORDS
7373 rword = RES_NONE;
7374 last_rword = RES_XXXX;
7375#endif
7376 last_followup = PIPE_SEQ;
7377 rcode = G.last_exitcode;
7378
7379 /* Go through list of pipes, (maybe) executing them. */
7380 for (; pi; pi = IF_HUSH_LOOPS(rword == RES_DONE ? loop_top : ) pi->next) {
7381 if (G.flag_SIGINT)
7382 break;
7383
7384 IF_HAS_KEYWORDS(rword = pi->res_word;)
7385 debug_printf_exec(": rword=%d cond_code=%d last_rword=%d\n",
7386 rword, cond_code, last_rword);
7387#if ENABLE_HUSH_LOOPS
7388 if ((rword == RES_WHILE || rword == RES_UNTIL || rword == RES_FOR)
7389 && loop_top == NULL /* avoid bumping G.depth_of_loop twice */
7390 ) {
7391 /* start of a loop: remember where loop starts */
7392 loop_top = pi;
7393 G.depth_of_loop++;
7394 }
7395#endif
7396 /* Still in the same "if...", "then..." or "do..." branch? */
7397 if (IF_HAS_KEYWORDS(rword == last_rword &&) 1) {
7398 if ((rcode == 0 && last_followup == PIPE_OR)
7399 || (rcode != 0 && last_followup == PIPE_AND)
7400 ) {
7401 /* It is "<true> || CMD" or "<false> && CMD"
7402 * and we should not execute CMD */
7403 debug_printf_exec("skipped cmd because of || or &&\n");
7404 last_followup = pi->followup;
Denys Vlasenko3beab832013-04-07 18:16:58 +02007405 goto dont_check_jobs_but_continue;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007406 }
7407 }
7408 last_followup = pi->followup;
7409 IF_HAS_KEYWORDS(last_rword = rword;)
7410#if ENABLE_HUSH_IF
7411 if (cond_code) {
7412 if (rword == RES_THEN) {
7413 /* if false; then ... fi has exitcode 0! */
7414 G.last_exitcode = rcode = EXIT_SUCCESS;
7415 /* "if <false> THEN cmd": skip cmd */
7416 continue;
7417 }
7418 } else {
7419 if (rword == RES_ELSE || rword == RES_ELIF) {
7420 /* "if <true> then ... ELSE/ELIF cmd":
7421 * skip cmd and all following ones */
7422 break;
7423 }
7424 }
7425#endif
7426#if ENABLE_HUSH_LOOPS
7427 if (rword == RES_FOR) { /* && pi->num_cmds - always == 1 */
7428 if (!for_lcur) {
7429 /* first loop through for */
7430
7431 static const char encoded_dollar_at[] ALIGN1 = {
7432 SPECIAL_VAR_SYMBOL, '@' | 0x80, SPECIAL_VAR_SYMBOL, '\0'
7433 }; /* encoded representation of "$@" */
7434 static const char *const encoded_dollar_at_argv[] = {
7435 encoded_dollar_at, NULL
7436 }; /* argv list with one element: "$@" */
7437 char **vals;
7438
7439 vals = (char**)encoded_dollar_at_argv;
7440 if (pi->next->res_word == RES_IN) {
7441 /* if no variable values after "in" we skip "for" */
7442 if (!pi->next->cmds[0].argv) {
7443 G.last_exitcode = rcode = EXIT_SUCCESS;
7444 debug_printf_exec(": null FOR: exitcode EXIT_SUCCESS\n");
7445 break;
7446 }
7447 vals = pi->next->cmds[0].argv;
7448 } /* else: "for var; do..." -> assume "$@" list */
7449 /* create list of variable values */
7450 debug_print_strings("for_list made from", vals);
7451 for_list = expand_strvec_to_strvec(vals);
7452 for_lcur = for_list;
7453 debug_print_strings("for_list", for_list);
7454 }
7455 if (!*for_lcur) {
7456 /* "for" loop is over, clean up */
7457 free(for_list);
7458 for_list = NULL;
7459 for_lcur = NULL;
7460 break;
7461 }
7462 /* Insert next value from for_lcur */
7463 /* note: *for_lcur already has quotes removed, $var expanded, etc */
7464 set_local_var(xasprintf("%s=%s", pi->cmds[0].argv[0], *for_lcur++), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7465 continue;
7466 }
7467 if (rword == RES_IN) {
7468 continue; /* "for v IN list;..." - "in" has no cmds anyway */
7469 }
7470 if (rword == RES_DONE) {
7471 continue; /* "done" has no cmds too */
7472 }
7473#endif
7474#if ENABLE_HUSH_CASE
7475 if (rword == RES_CASE) {
7476 case_word = expand_strvec_to_string(pi->cmds->argv);
7477 continue;
7478 }
7479 if (rword == RES_MATCH) {
7480 char **argv;
7481
7482 if (!case_word) /* "case ... matched_word) ... WORD)": we executed selected branch, stop */
7483 break;
7484 /* all prev words didn't match, does this one match? */
7485 argv = pi->cmds->argv;
7486 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007487 char *pattern = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007488 /* TODO: which FNM_xxx flags to use? */
7489 cond_code = (fnmatch(pattern, case_word, /*flags:*/ 0) != 0);
7490 free(pattern);
7491 if (cond_code == 0) { /* match! we will execute this branch */
7492 free(case_word); /* make future "word)" stop */
7493 case_word = NULL;
7494 break;
7495 }
7496 argv++;
7497 }
7498 continue;
7499 }
7500 if (rword == RES_CASE_BODY) { /* inside of a case branch */
7501 if (cond_code != 0)
7502 continue; /* not matched yet, skip this pipe */
7503 }
7504#endif
7505 /* Just pressing <enter> in shell should check for jobs.
7506 * OTOH, in non-interactive shell this is useless
7507 * and only leads to extra job checks */
7508 if (pi->num_cmds == 0) {
7509 if (G_interactive_fd)
7510 goto check_jobs_and_continue;
7511 continue;
7512 }
7513
7514 /* After analyzing all keywords and conditions, we decided
7515 * to execute this pipe. NB: have to do checkjobs(NULL)
7516 * after run_pipe to collect any background children,
7517 * even if list execution is to be stopped. */
7518 debug_printf_exec(": run_pipe with %d members\n", pi->num_cmds);
7519 {
7520 int r;
7521#if ENABLE_HUSH_LOOPS
7522 G.flag_break_continue = 0;
7523#endif
7524 rcode = r = run_pipe(pi); /* NB: rcode is a smallint */
7525 if (r != -1) {
7526 /* We ran a builtin, function, or group.
7527 * rcode is already known
7528 * and we don't need to wait for anything. */
7529 G.last_exitcode = rcode;
7530 debug_printf_exec(": builtin/func exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007531 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007532#if ENABLE_HUSH_LOOPS
7533 /* Was it "break" or "continue"? */
7534 if (G.flag_break_continue) {
7535 smallint fbc = G.flag_break_continue;
7536 /* We might fall into outer *loop*,
7537 * don't want to break it too */
7538 if (loop_top) {
7539 G.depth_break_continue--;
7540 if (G.depth_break_continue == 0)
7541 G.flag_break_continue = 0;
7542 /* else: e.g. "continue 2" should *break* once, *then* continue */
7543 } /* else: "while... do... { we are here (innermost list is not a loop!) };...done" */
Denys Vlasenko3beab832013-04-07 18:16:58 +02007544 if (G.depth_break_continue != 0 || fbc == BC_BREAK) {
7545 checkjobs(NULL);
7546 break;
7547 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007548 /* "continue": simulate end of loop */
7549 rword = RES_DONE;
7550 continue;
7551 }
7552#endif
7553#if ENABLE_HUSH_FUNCTIONS
7554 if (G.flag_return_in_progress == 1) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007555 checkjobs(NULL);
7556 break;
7557 }
7558#endif
7559 } else if (pi->followup == PIPE_BG) {
7560 /* What does bash do with attempts to background builtins? */
7561 /* even bash 3.2 doesn't do that well with nested bg:
7562 * try "{ { sleep 10; echo DEEP; } & echo HERE; } &".
7563 * I'm NOT treating inner &'s as jobs */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007564 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007565#if ENABLE_HUSH_JOB
7566 if (G.run_list_level == 1)
7567 insert_bg_job(pi);
7568#endif
7569 /* Last command's pid goes to $! */
7570 G.last_bg_pid = pi->cmds[pi->num_cmds - 1].pid;
7571 G.last_exitcode = rcode = EXIT_SUCCESS;
7572 debug_printf_exec(": cmd&: exitcode EXIT_SUCCESS\n");
7573 } else {
7574#if ENABLE_HUSH_JOB
7575 if (G.run_list_level == 1 && G_interactive_fd) {
7576 /* Waits for completion, then fg's main shell */
7577 rcode = checkjobs_and_fg_shell(pi);
7578 debug_printf_exec(": checkjobs_and_fg_shell exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007579 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007580 } else
7581#endif
7582 { /* This one just waits for completion */
7583 rcode = checkjobs(pi);
7584 debug_printf_exec(": checkjobs exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007585 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007586 }
7587 G.last_exitcode = rcode;
7588 }
7589 }
7590
7591 /* Analyze how result affects subsequent commands */
7592#if ENABLE_HUSH_IF
7593 if (rword == RES_IF || rword == RES_ELIF)
7594 cond_code = rcode;
7595#endif
Denys Vlasenko3beab832013-04-07 18:16:58 +02007596 check_jobs_and_continue:
7597 checkjobs(NULL);
7598 dont_check_jobs_but_continue: ;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007599#if ENABLE_HUSH_LOOPS
7600 /* Beware of "while false; true; do ..."! */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007601 if (pi->next
7602 && (pi->next->res_word == RES_DO || pi->next->res_word == RES_DONE)
Denys Vlasenko56a3b822011-06-01 12:47:07 +02007603 /* check for RES_DONE is needed for "while ...; do \n done" case */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007604 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007605 if (rword == RES_WHILE) {
7606 if (rcode) {
7607 /* "while false; do...done" - exitcode 0 */
7608 G.last_exitcode = rcode = EXIT_SUCCESS;
7609 debug_printf_exec(": while expr is false: breaking (exitcode:EXIT_SUCCESS)\n");
Denys Vlasenko3beab832013-04-07 18:16:58 +02007610 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007611 }
7612 }
7613 if (rword == RES_UNTIL) {
7614 if (!rcode) {
7615 debug_printf_exec(": until expr is true: breaking\n");
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007616 break;
7617 }
7618 }
7619 }
7620#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007621 } /* for (pi) */
7622
7623#if ENABLE_HUSH_JOB
7624 G.run_list_level--;
7625#endif
7626#if ENABLE_HUSH_LOOPS
7627 if (loop_top)
7628 G.depth_of_loop--;
7629 free(for_list);
7630#endif
7631#if ENABLE_HUSH_CASE
7632 free(case_word);
7633#endif
7634 debug_leave();
7635 debug_printf_exec("run_list lvl %d return %d\n", G.run_list_level + 1, rcode);
7636 return rcode;
7637}
7638
7639/* Select which version we will use */
7640static int run_and_free_list(struct pipe *pi)
7641{
7642 int rcode = 0;
7643 debug_printf_exec("run_and_free_list entered\n");
Dan Fandrich85c62472010-11-20 13:05:17 -08007644 if (!G.o_opt[OPT_O_NOEXEC]) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007645 debug_printf_exec(": run_list: 1st pipe with %d cmds\n", pi->num_cmds);
7646 rcode = run_list(pi);
7647 }
7648 /* free_pipe_list has the side effect of clearing memory.
7649 * In the long run that function can be merged with run_list,
7650 * but doing that now would hobble the debugging effort. */
7651 free_pipe_list(pi);
7652 debug_printf_exec("run_and_free_list return %d\n", rcode);
7653 return rcode;
7654}
7655
7656
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007657static void install_sighandlers(unsigned mask)
Eric Andersen52a97ca2001-06-22 06:49:26 +00007658{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007659 sighandler_t old_handler;
7660 unsigned sig = 0;
7661 while ((mask >>= 1) != 0) {
7662 sig++;
7663 if (!(mask & 1))
7664 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02007665 old_handler = install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007666 /* POSIX allows shell to re-enable SIGCHLD
7667 * even if it was SIG_IGN on entry.
7668 * Therefore we skip IGN check for it:
7669 */
7670 if (sig == SIGCHLD)
7671 continue;
7672 if (old_handler == SIG_IGN) {
7673 /* oops... restore back to IGN, and record this fact */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007674 install_sighandler(sig, old_handler);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007675 if (!G.traps)
7676 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7677 free(G.traps[sig]);
7678 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
7679 }
7680 }
7681}
7682
7683/* Called a few times only (or even once if "sh -c") */
7684static void install_special_sighandlers(void)
7685{
Denis Vlasenkof9375282009-04-05 19:13:39 +00007686 unsigned mask;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007687
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007688 /* Which signals are shell-special? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007689 mask = (1 << SIGQUIT) | (1 << SIGCHLD);
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007690 if (G_interactive_fd) {
7691 mask |= SPECIAL_INTERACTIVE_SIGS;
7692 if (G_saved_tty_pgrp) /* we have ctty, job control sigs work */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007693 mask |= SPECIAL_JOBSTOP_SIGS;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007694 }
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007695 /* Careful, do not re-install handlers we already installed */
7696 if (G.special_sig_mask != mask) {
7697 unsigned diff = mask & ~G.special_sig_mask;
7698 G.special_sig_mask = mask;
7699 install_sighandlers(diff);
7700 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00007701}
7702
7703#if ENABLE_HUSH_JOB
7704/* helper */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007705/* Set handlers to restore tty pgrp and exit */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007706static void install_fatal_sighandlers(void)
Denis Vlasenkof9375282009-04-05 19:13:39 +00007707{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007708 unsigned mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007709
7710 /* We will restore tty pgrp on these signals */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007711 mask = 0
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007712 + (1 << SIGILL ) * HUSH_DEBUG
7713 + (1 << SIGFPE ) * HUSH_DEBUG
7714 + (1 << SIGBUS ) * HUSH_DEBUG
7715 + (1 << SIGSEGV) * HUSH_DEBUG
7716 + (1 << SIGTRAP) * HUSH_DEBUG
7717 + (1 << SIGABRT)
7718 /* bash 3.2 seems to handle these just like 'fatal' ones */
7719 + (1 << SIGPIPE)
7720 + (1 << SIGALRM)
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007721 /* if we are interactive, SIGHUP, SIGTERM and SIGINT are special sigs.
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007722 * if we aren't interactive... but in this case
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007723 * we never want to restore pgrp on exit, and this fn is not called
7724 */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007725 /*+ (1 << SIGHUP )*/
7726 /*+ (1 << SIGTERM)*/
7727 /*+ (1 << SIGINT )*/
7728 ;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007729 G_fatal_sig_mask = mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007730
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007731 install_sighandlers(mask);
Denis Vlasenkof9375282009-04-05 19:13:39 +00007732}
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00007733#endif
Eric Andersenada18ff2001-05-21 16:18:22 +00007734
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007735static int set_mode(int state, char mode, const char *o_opt)
Denis Vlasenkod5762932009-03-31 11:22:57 +00007736{
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007737 int idx;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007738 switch (mode) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007739 case 'n':
Dan Fandrich85c62472010-11-20 13:05:17 -08007740 G.o_opt[OPT_O_NOEXEC] = state;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007741 break;
7742 case 'x':
7743 IF_HUSH_MODE_X(G_x_mode = state;)
7744 break;
7745 case 'o':
7746 if (!o_opt) {
7747 /* "set -+o" without parameter.
7748 * in bash, set -o produces this output:
7749 * pipefail off
7750 * and set +o:
7751 * set +o pipefail
7752 * We always use the second form.
7753 */
7754 const char *p = o_opt_strings;
7755 idx = 0;
7756 while (*p) {
7757 printf("set %co %s\n", (G.o_opt[idx] ? '-' : '+'), p);
7758 idx++;
7759 p += strlen(p) + 1;
7760 }
7761 break;
7762 }
7763 idx = index_in_strings(o_opt_strings, o_opt);
7764 if (idx >= 0) {
7765 G.o_opt[idx] = state;
7766 break;
7767 }
7768 default:
7769 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007770 }
7771 return EXIT_SUCCESS;
7772}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00007773
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00007774int hush_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Matt Kraai2d91deb2001-08-01 17:21:35 +00007775int hush_main(int argc, char **argv)
Eric Andersen25f27032001-04-26 23:22:31 +00007776{
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007777 enum {
7778 OPT_login = (1 << 0),
7779 };
7780 unsigned flags;
Eric Andersen25f27032001-04-26 23:22:31 +00007781 int opt;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007782 unsigned builtin_argc;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007783 char **e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007784 struct variable *cur_var;
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007785 struct variable *shell_ver;
Eric Andersenbc604a22001-05-16 05:24:03 +00007786
Denis Vlasenko574f2f42008-02-27 18:41:59 +00007787 INIT_G();
Denys Vlasenko10c01312011-05-11 11:49:21 +02007788 if (EXIT_SUCCESS != 0) /* if EXIT_SUCCESS == 0, it is already done */
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007789 G.last_exitcode = EXIT_SUCCESS;
Denys Vlasenko10c01312011-05-11 11:49:21 +02007790#if ENABLE_HUSH_FAST
7791 G.count_SIGCHLD++; /* ensure it is != G.handled_SIGCHLD */
7792#endif
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007793#if !BB_MMU
7794 G.argv0_for_re_execing = argv[0];
7795#endif
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007796 /* Deal with HUSH_VERSION */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007797 shell_ver = xzalloc(sizeof(*shell_ver));
7798 shell_ver->flg_export = 1;
7799 shell_ver->flg_read_only = 1;
Denys Vlasenko4f870492010-09-10 11:06:01 +02007800 /* Code which handles ${var<op>...} needs writable values for all variables,
Denys Vlasenko36f774a2010-09-05 14:45:38 +02007801 * therefore we xstrdup: */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007802 shell_ver->varstr = xstrdup(hush_version_str);
Denys Vlasenko605067b2010-09-06 12:10:51 +02007803 /* Create shell local variables from the values
7804 * currently living in the environment */
Denis Vlasenkof886fd22008-10-13 12:36:05 +00007805 debug_printf_env("unsetenv '%s'\n", "HUSH_VERSION");
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007806 unsetenv("HUSH_VERSION"); /* in case it exists in initial env */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007807 G.top_var = shell_ver;
Denis Vlasenko87a86552008-07-29 19:43:10 +00007808 cur_var = G.top_var;
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007809 e = environ;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007810 if (e) while (*e) {
7811 char *value = strchr(*e, '=');
7812 if (value) { /* paranoia */
7813 cur_var->next = xzalloc(sizeof(*cur_var));
7814 cur_var = cur_var->next;
Denis Vlasenko28c0f0f2007-05-25 02:46:01 +00007815 cur_var->varstr = *e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007816 cur_var->max_len = strlen(*e);
7817 cur_var->flg_export = 1;
7818 }
7819 e++;
7820 }
Denys Vlasenko605067b2010-09-06 12:10:51 +02007821 /* (Re)insert HUSH_VERSION into env (AFTER we scanned the env!) */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007822 debug_printf_env("putenv '%s'\n", shell_ver->varstr);
7823 putenv(shell_ver->varstr);
Denys Vlasenko6db47842009-09-05 20:15:17 +02007824
7825 /* Export PWD */
7826 set_pwd_var(/*exp:*/ 1);
Denys Vlasenko3fa97af2014-04-15 11:43:29 +02007827
7828#if ENABLE_HUSH_BASH_COMPAT
7829 /* Set (but not export) HOSTNAME unless already set */
7830 if (!get_local_var_value("HOSTNAME")) {
7831 struct utsname uts;
7832 uname(&uts);
7833 set_local_var_from_halves("HOSTNAME", uts.nodename);
7834 }
Denys Vlasenko6db47842009-09-05 20:15:17 +02007835 /* bash also exports SHLVL and _,
7836 * and sets (but doesn't export) the following variables:
7837 * BASH=/bin/bash
7838 * BASH_VERSINFO=([0]="3" [1]="2" [2]="0" [3]="1" [4]="release" [5]="i386-pc-linux-gnu")
7839 * BASH_VERSION='3.2.0(1)-release'
7840 * HOSTTYPE=i386
7841 * MACHTYPE=i386-pc-linux-gnu
7842 * OSTYPE=linux-gnu
Denys Vlasenkodea47882009-10-09 15:40:49 +02007843 * PPID=<NNNNN> - we also do it elsewhere
Denys Vlasenko6db47842009-09-05 20:15:17 +02007844 * EUID=<NNNNN>
7845 * UID=<NNNNN>
7846 * GROUPS=()
7847 * LINES=<NNN>
7848 * COLUMNS=<NNN>
7849 * BASH_ARGC=()
7850 * BASH_ARGV=()
7851 * BASH_LINENO=()
7852 * BASH_SOURCE=()
7853 * DIRSTACK=()
7854 * PIPESTATUS=([0]="0")
7855 * HISTFILE=/<xxx>/.bash_history
7856 * HISTFILESIZE=500
7857 * HISTSIZE=500
7858 * MAILCHECK=60
7859 * PATH=/usr/gnu/bin:/usr/local/bin:/bin:/usr/bin:.
7860 * SHELL=/bin/bash
7861 * SHELLOPTS=braceexpand:emacs:hashall:histexpand:history:interactive-comments:monitor
7862 * TERM=dumb
7863 * OPTERR=1
7864 * OPTIND=1
7865 * IFS=$' \t\n'
7866 * PS1='\s-\v\$ '
7867 * PS2='> '
7868 * PS4='+ '
7869 */
Denys Vlasenko3fa97af2014-04-15 11:43:29 +02007870#endif
Denys Vlasenko6db47842009-09-05 20:15:17 +02007871
Denis Vlasenko38f63192007-01-22 09:03:07 +00007872#if ENABLE_FEATURE_EDITING
Denys Vlasenkoe45af7a2011-09-04 16:15:24 +02007873 G.line_input_state = new_line_input_t(FOR_SHELL);
Denis Vlasenko8e1c7152007-01-22 07:21:38 +00007874#endif
Denys Vlasenko99862cb2010-09-12 17:34:13 +02007875
Eric Andersen94ac2442001-05-22 19:05:18 +00007876 /* Initialize some more globals to non-zero values */
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00007877 cmdedit_update_prompt();
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +00007878
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02007879 die_func = xfunc_has_died;
Denis Vlasenkoed782372009-04-10 00:45:02 +00007880
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007881 /* Shell is non-interactive at first. We need to call
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007882 * install_special_sighandlers() if we are going to execute "sh <script>",
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007883 * "sh -c <cmds>" or login shell's /etc/profile and friends.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007884 * If we later decide that we are interactive, we run install_special_sighandlers()
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007885 * in order to intercept (more) signals.
7886 */
7887
7888 /* Parse options */
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007889 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/sh.html */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007890 flags = (argv[0] && argv[0][0] == '-') ? OPT_login : 0;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007891 builtin_argc = 0;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007892 while (1) {
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007893 opt = getopt(argc, argv, "+c:xinsl"
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007894#if !BB_MMU
Denis Vlasenkobc569742009-04-12 20:35:19 +00007895 "<:$:R:V:"
7896# if ENABLE_HUSH_FUNCTIONS
7897 "F:"
7898# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007899#endif
7900 );
7901 if (opt <= 0)
7902 break;
Eric Andersen25f27032001-04-26 23:22:31 +00007903 switch (opt) {
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007904 case 'c':
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007905 /* Possibilities:
7906 * sh ... -c 'script'
7907 * sh ... -c 'script' ARG0 [ARG1...]
7908 * On NOMMU, if builtin_argc != 0,
Denys Vlasenko17323a62010-01-28 01:57:05 +01007909 * sh ... -c 'builtin' BARGV... "" ARG0 [ARG1...]
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007910 * "" needs to be replaced with NULL
7911 * and BARGV vector fed to builtin function.
Denys Vlasenko17323a62010-01-28 01:57:05 +01007912 * Note: the form without ARG0 never happens:
7913 * sh ... -c 'builtin' BARGV... ""
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007914 */
Denys Vlasenkodea47882009-10-09 15:40:49 +02007915 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007916 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02007917 G.root_ppid = getppid();
7918 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00007919 G.global_argv = argv + optind;
7920 G.global_argc = argc - optind;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007921 if (builtin_argc) {
7922 /* -c 'builtin' [BARGV...] "" ARG0 [ARG1...] */
7923 const struct built_in_command *x;
7924
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007925 install_special_sighandlers();
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007926 x = find_builtin(optarg);
7927 if (x) { /* paranoia */
7928 G.global_argc -= builtin_argc; /* skip [BARGV...] "" */
7929 G.global_argv += builtin_argc;
7930 G.global_argv[-1] = NULL; /* replace "" */
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007931 fflush_all();
Denys Vlasenko17323a62010-01-28 01:57:05 +01007932 G.last_exitcode = x->b_function(argv + optind - 1);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007933 }
7934 goto final_return;
7935 }
7936 if (!G.global_argv[0]) {
7937 /* -c 'script' (no params): prevent empty $0 */
7938 G.global_argv--; /* points to argv[i] of 'script' */
7939 G.global_argv[0] = argv[0];
Denys Vlasenko5ae8f1c2010-05-22 06:32:11 +02007940 G.global_argc++;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007941 } /* else -c 'script' ARG0 [ARG1...]: $0 is ARG0 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007942 install_special_sighandlers();
Denis Vlasenkob6e65562009-04-03 16:49:04 +00007943 parse_and_run_string(optarg);
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007944 goto final_return;
7945 case 'i':
Denis Vlasenkoc666f712007-05-16 22:18:54 +00007946 /* Well, we cannot just declare interactiveness,
7947 * we have to have some stuff (ctty, etc) */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00007948 /* G_interactive_fd++; */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007949 break;
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007950 case 's':
7951 /* "-s" means "read from stdin", but this is how we always
7952 * operate, so simply do nothing here. */
7953 break;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007954 case 'l':
7955 flags |= OPT_login;
7956 break;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007957#if !BB_MMU
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007958 case '<': /* "big heredoc" support */
Denys Vlasenko729ecb82010-06-07 14:14:26 +02007959 full_write1_str(optarg);
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007960 _exit(0);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007961 case '$': {
7962 unsigned long long empty_trap_mask;
7963
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007964 G.root_pid = bb_strtou(optarg, &optarg, 16);
7965 optarg++;
Denys Vlasenkodea47882009-10-09 15:40:49 +02007966 G.root_ppid = bb_strtou(optarg, &optarg, 16);
7967 optarg++;
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007968 G.last_bg_pid = bb_strtou(optarg, &optarg, 16);
7969 optarg++;
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007970 G.last_exitcode = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007971 optarg++;
7972 builtin_argc = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007973 optarg++;
7974 empty_trap_mask = bb_strtoull(optarg, &optarg, 16);
7975 if (empty_trap_mask != 0) {
7976 int sig;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007977 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007978 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7979 for (sig = 1; sig < NSIG; sig++) {
7980 if (empty_trap_mask & (1LL << sig)) {
7981 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007982 install_sighandler(sig, SIG_IGN);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007983 }
7984 }
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007985 }
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007986# if ENABLE_HUSH_LOOPS
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007987 optarg++;
7988 G.depth_of_loop = bb_strtou(optarg, &optarg, 16);
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007989# endif
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007990 break;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007991 }
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007992 case 'R':
7993 case 'V':
Denys Vlasenko295fef82009-06-03 12:47:26 +02007994 set_local_var(xstrdup(optarg), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ opt == 'R');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007995 break;
Denis Vlasenkobc569742009-04-12 20:35:19 +00007996# if ENABLE_HUSH_FUNCTIONS
7997 case 'F': {
7998 struct function *funcp = new_function(optarg);
7999 /* funcp->name is already set to optarg */
8000 /* funcp->body is set to NULL. It's a special case. */
8001 funcp->body_as_string = argv[optind];
8002 optind++;
8003 break;
8004 }
8005# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00008006#endif
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008007 case 'n':
8008 case 'x':
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008009 if (set_mode(1, opt, NULL) == 0) /* no error */
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008010 break;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00008011 default:
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00008012#ifndef BB_VER
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00008013 fprintf(stderr, "Usage: sh [FILE]...\n"
8014 " or: sh -c command [args]...\n\n");
8015 exit(EXIT_FAILURE);
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00008016#else
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00008017 bb_show_usage();
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00008018#endif
Eric Andersen25f27032001-04-26 23:22:31 +00008019 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008020 } /* option parsing loop */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008021
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008022 /* Skip options. Try "hush -l": $1 should not be "-l"! */
8023 G.global_argc = argc - (optind - 1);
8024 G.global_argv = argv + (optind - 1);
8025 G.global_argv[0] = argv[0];
8026
Denys Vlasenkodea47882009-10-09 15:40:49 +02008027 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008028 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02008029 G.root_ppid = getppid();
8030 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008031
8032 /* If we are login shell... */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008033 if (flags & OPT_login) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008034 FILE *input;
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008035 debug_printf("sourcing /etc/profile\n");
8036 input = fopen_for_read("/etc/profile");
8037 if (input != NULL) {
8038 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008039 install_special_sighandlers();
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008040 parse_and_run_file(input);
8041 fclose(input);
8042 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008043 /* bash: after sourcing /etc/profile,
8044 * tries to source (in the given order):
8045 * ~/.bash_profile, ~/.bash_login, ~/.profile,
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008046 * stopping on first found. --noprofile turns this off.
Denis Vlasenkof9375282009-04-05 19:13:39 +00008047 * bash also sources ~/.bash_logout on exit.
8048 * If called as sh, skips .bash_XXX files.
8049 */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008050 }
8051
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008052 if (G.global_argv[1]) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008053 FILE *input;
8054 /*
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00008055 * "bash <script>" (which is never interactive (unless -i?))
8056 * sources $BASH_ENV here (without scanning $PATH).
Denis Vlasenkof9375282009-04-05 19:13:39 +00008057 * If called as sh, does the same but with $ENV.
8058 */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008059 G.global_argc--;
8060 G.global_argv++;
8061 debug_printf("running script '%s'\n", G.global_argv[0]);
8062 input = xfopen_for_read(G.global_argv[0]);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008063 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008064 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008065 parse_and_run_file(input);
8066#if ENABLE_FEATURE_CLEAN_UP
8067 fclose(input);
8068#endif
8069 goto final_return;
8070 }
8071
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008072 /* Up to here, shell was non-interactive. Now it may become one.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008073 * NB: don't forget to (re)run install_special_sighandlers() as needed.
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008074 */
Denis Vlasenkof9375282009-04-05 19:13:39 +00008075
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008076 /* A shell is interactive if the '-i' flag was given,
8077 * or if all of the following conditions are met:
Denis Vlasenko55b2de72007-04-18 17:21:28 +00008078 * no -c command
Eric Andersen25f27032001-04-26 23:22:31 +00008079 * no arguments remaining or the -s flag given
8080 * standard input is a terminal
8081 * standard output is a terminal
Denis Vlasenkof9375282009-04-05 19:13:39 +00008082 * Refer to Posix.2, the description of the 'sh' utility.
8083 */
8084#if ENABLE_HUSH_JOB
8085 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Mike Frysinger38478a62009-05-20 04:48:06 -04008086 G_saved_tty_pgrp = tcgetpgrp(STDIN_FILENO);
8087 debug_printf("saved_tty_pgrp:%d\n", G_saved_tty_pgrp);
8088 if (G_saved_tty_pgrp < 0)
8089 G_saved_tty_pgrp = 0;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008090
8091 /* try to dup stdin to high fd#, >= 255 */
8092 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8093 if (G_interactive_fd < 0) {
8094 /* try to dup to any fd */
8095 G_interactive_fd = dup(STDIN_FILENO);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008096 if (G_interactive_fd < 0) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008097 /* give up */
8098 G_interactive_fd = 0;
Mike Frysinger38478a62009-05-20 04:48:06 -04008099 G_saved_tty_pgrp = 0;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00008100 }
8101 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008102// TODO: track & disallow any attempts of user
8103// to (inadvertently) close/redirect G_interactive_fd
Eric Andersen25f27032001-04-26 23:22:31 +00008104 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008105 debug_printf("interactive_fd:%d\n", G_interactive_fd);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008106 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008107 close_on_exec_on(G_interactive_fd);
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008108
Mike Frysinger38478a62009-05-20 04:48:06 -04008109 if (G_saved_tty_pgrp) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008110 /* If we were run as 'hush &', sleep until we are
8111 * in the foreground (tty pgrp == our pgrp).
8112 * If we get started under a job aware app (like bash),
8113 * make sure we are now in charge so we don't fight over
8114 * who gets the foreground */
8115 while (1) {
8116 pid_t shell_pgrp = getpgrp();
Mike Frysinger38478a62009-05-20 04:48:06 -04008117 G_saved_tty_pgrp = tcgetpgrp(G_interactive_fd);
8118 if (G_saved_tty_pgrp == shell_pgrp)
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008119 break;
8120 /* send TTIN to ourself (should stop us) */
8121 kill(- shell_pgrp, SIGTTIN);
8122 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008123 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008124
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008125 /* Install more signal handlers */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008126 install_special_sighandlers();
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008127
Mike Frysinger38478a62009-05-20 04:48:06 -04008128 if (G_saved_tty_pgrp) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008129 /* Set other signals to restore saved_tty_pgrp */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008130 install_fatal_sighandlers();
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008131 /* Put ourselves in our own process group
8132 * (bash, too, does this only if ctty is available) */
8133 bb_setpgrp(); /* is the same as setpgid(our_pid, our_pid); */
8134 /* Grab control of the terminal */
8135 tcsetpgrp(G_interactive_fd, getpid());
8136 }
Denys Vlasenko550bf5b2015-10-09 16:42:57 +02008137 enable_restore_tty_pgrp_on_exit();
Denys Vlasenko4840ae82011-09-04 15:28:03 +02008138
8139# if ENABLE_HUSH_SAVEHISTORY && MAX_HISTORY > 0
8140 {
8141 const char *hp = get_local_var_value("HISTFILE");
8142 if (!hp) {
8143 hp = get_local_var_value("HOME");
8144 if (hp)
8145 hp = concat_path_file(hp, ".hush_history");
8146 } else {
8147 hp = xstrdup(hp);
8148 }
8149 if (hp) {
8150 G.line_input_state->hist_file = hp;
Denys Vlasenko4840ae82011-09-04 15:28:03 +02008151 //set_local_var(xasprintf("HISTFILE=%s", ...));
8152 }
8153# if ENABLE_FEATURE_SH_HISTFILESIZE
8154 hp = get_local_var_value("HISTFILESIZE");
8155 G.line_input_state->max_history = size_from_HISTFILESIZE(hp);
8156# endif
8157 }
8158# endif
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008159 } else {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008160 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008161 }
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008162#elif ENABLE_HUSH_INTERACTIVE
Denis Vlasenkof9375282009-04-05 19:13:39 +00008163 /* No job control compiled in, only prompt/line editing */
8164 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008165 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8166 if (G_interactive_fd < 0) {
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008167 /* try to dup to any fd */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008168 G_interactive_fd = dup(STDIN_FILENO);
8169 if (G_interactive_fd < 0)
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008170 /* give up */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008171 G_interactive_fd = 0;
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008172 }
8173 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008174 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008175 close_on_exec_on(G_interactive_fd);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008176 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008177 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008178#else
8179 /* We have interactiveness code disabled */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008180 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008181#endif
8182 /* bash:
8183 * if interactive but not a login shell, sources ~/.bashrc
8184 * (--norc turns this off, --rcfile <file> overrides)
8185 */
8186
8187 if (!ENABLE_FEATURE_SH_EXTRA_QUIET && G_interactive_fd) {
Denys Vlasenkoc34c0332009-09-29 12:25:30 +02008188 /* note: ash and hush share this string */
8189 printf("\n\n%s %s\n"
8190 IF_HUSH_HELP("Enter 'help' for a list of built-in commands.\n")
8191 "\n",
8192 bb_banner,
8193 "hush - the humble shell"
8194 );
Mike Frysingerb2705e12009-03-23 08:44:02 +00008195 }
8196
Denis Vlasenkof9375282009-04-05 19:13:39 +00008197 parse_and_run_file(stdin);
Eric Andersen25f27032001-04-26 23:22:31 +00008198
Denis Vlasenkod76c0492007-05-25 02:16:25 +00008199 final_return:
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008200 hush_exit(G.last_exitcode);
Eric Andersen25f27032001-04-26 23:22:31 +00008201}
Denis Vlasenko96702ca2007-11-23 23:28:55 +00008202
8203
Denys Vlasenko1cc4b132009-08-21 00:05:51 +02008204#if ENABLE_MSH
8205int msh_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
8206int msh_main(int argc, char **argv)
8207{
8208 //bb_error_msg("msh is deprecated, please use hush instead");
8209 return hush_main(argc, argv);
8210}
8211#endif
8212
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008213
8214/*
8215 * Built-ins
8216 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008217static int FAST_FUNC builtin_true(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008218{
8219 return 0;
8220}
8221
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008222static int run_applet_main(char **argv, int (*applet_main_func)(int argc, char **argv))
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008223{
8224 int argc = 0;
8225 while (*argv) {
8226 argc++;
8227 argv++;
8228 }
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008229 return applet_main_func(argc, argv - argc);
Mike Frysingerccb19592009-10-15 03:31:15 -04008230}
8231
8232static int FAST_FUNC builtin_test(char **argv)
8233{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008234 return run_applet_main(argv, test_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008235}
8236
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008237static int FAST_FUNC builtin_echo(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008238{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008239 return run_applet_main(argv, echo_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008240}
8241
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008242#if ENABLE_PRINTF
8243static int FAST_FUNC builtin_printf(char **argv)
8244{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008245 return run_applet_main(argv, printf_main);
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008246}
8247#endif
8248
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008249static char **skip_dash_dash(char **argv)
8250{
8251 argv++;
8252 if (argv[0] && argv[0][0] == '-' && argv[0][1] == '-' && argv[0][2] == '\0')
8253 argv++;
8254 return argv;
8255}
8256
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008257static int FAST_FUNC builtin_eval(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008258{
8259 int rcode = EXIT_SUCCESS;
8260
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008261 argv = skip_dash_dash(argv);
8262 if (*argv) {
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008263 char *str = expand_strvec_to_string(argv);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008264 /* bash:
8265 * eval "echo Hi; done" ("done" is syntax error):
8266 * "echo Hi" will not execute too.
8267 */
8268 parse_and_run_string(str);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008269 free(str);
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008270 rcode = G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008271 }
8272 return rcode;
8273}
8274
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008275static int FAST_FUNC builtin_cd(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008276{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008277 const char *newdir;
8278
8279 argv = skip_dash_dash(argv);
8280 newdir = argv[0];
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008281 if (newdir == NULL) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008282 /* bash does nothing (exitcode 0) if HOME is ""; if it's unset,
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008283 * bash says "bash: cd: HOME not set" and does nothing
8284 * (exitcode 1)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008285 */
Denys Vlasenko90a99042009-09-06 02:36:23 +02008286 const char *home = get_local_var_value("HOME");
8287 newdir = home ? home : "/";
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008288 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008289 if (chdir(newdir)) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008290 /* Mimic bash message exactly */
8291 bb_perror_msg("cd: %s", newdir);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008292 return EXIT_FAILURE;
8293 }
Denys Vlasenko6db47842009-09-05 20:15:17 +02008294 /* Read current dir (get_cwd(1) is inside) and set PWD.
8295 * Note: do not enforce exporting. If PWD was unset or unexported,
8296 * set it again, but do not export. bash does the same.
8297 */
8298 set_pwd_var(/*exp:*/ 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008299 return EXIT_SUCCESS;
8300}
8301
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008302static int FAST_FUNC builtin_exec(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008303{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008304 argv = skip_dash_dash(argv);
8305 if (argv[0] == NULL)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008306 return EXIT_SUCCESS; /* bash does this */
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008307
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008308 /* Careful: we can end up here after [v]fork. Do not restore
8309 * tty pgrp then, only top-level shell process does that */
8310 if (G_saved_tty_pgrp && getpid() == G.root_pid)
8311 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
8312
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008313 /* TODO: if exec fails, bash does NOT exit! We do.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008314 * We'll need to undo trap cleanup (it's inside execvp_or_die)
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008315 * and tcsetpgrp, and this is inherently racy.
8316 */
8317 execvp_or_die(argv);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008318}
8319
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008320static int FAST_FUNC builtin_exit(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008321{
Denis Vlasenkocd418a22009-04-06 18:08:35 +00008322 debug_printf_exec("%s()\n", __func__);
Denis Vlasenko40e84372009-04-18 11:23:38 +00008323
8324 /* interactive bash:
8325 * # trap "echo EEE" EXIT
8326 * # exit
8327 * exit
8328 * There are stopped jobs.
8329 * (if there are _stopped_ jobs, running ones don't count)
8330 * # exit
8331 * exit
Denys Vlasenko6830ade2013-01-15 13:58:01 +01008332 * EEE (then bash exits)
Denis Vlasenko40e84372009-04-18 11:23:38 +00008333 *
Denys Vlasenkoa110c902010-09-12 15:38:04 +02008334 * TODO: we can use G.exiting = -1 as indicator "last cmd was exit"
Denis Vlasenko40e84372009-04-18 11:23:38 +00008335 */
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00008336
8337 /* note: EXIT trap is run by hush_exit */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008338 argv = skip_dash_dash(argv);
8339 if (argv[0] == NULL)
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008340 hush_exit(G.last_exitcode);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008341 /* mimic bash: exit 123abc == exit 255 + error msg */
8342 xfunc_error_retval = 255;
8343 /* bash: exit -2 == exit 254, no error msg */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008344 hush_exit(xatoi(argv[0]) & 0xff);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008345}
8346
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008347static void print_escaped(const char *s)
8348{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008349 if (*s == '\'')
8350 goto squote;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008351 do {
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008352 const char *p = strchrnul(s, '\'');
8353 /* print 'xxxx', possibly just '' */
8354 printf("'%.*s'", (int)(p - s), s);
8355 if (*p == '\0')
8356 break;
8357 s = p;
8358 squote:
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008359 /* s points to '; print "'''...'''" */
8360 putchar('"');
8361 do putchar('\''); while (*++s == '\'');
8362 putchar('"');
8363 } while (*s);
8364}
8365
Denys Vlasenko295fef82009-06-03 12:47:26 +02008366#if !ENABLE_HUSH_LOCAL
8367#define helper_export_local(argv, exp, lvl) \
8368 helper_export_local(argv, exp)
8369#endif
8370static void helper_export_local(char **argv, int exp, int lvl)
8371{
8372 do {
8373 char *name = *argv;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008374 char *name_end = strchrnul(name, '=');
Denys Vlasenko295fef82009-06-03 12:47:26 +02008375
8376 /* So far we do not check that name is valid (TODO?) */
8377
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008378 if (*name_end == '\0') {
8379 struct variable *var, **vpp;
Denys Vlasenko295fef82009-06-03 12:47:26 +02008380
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008381 vpp = get_ptr_to_local_var(name, name_end - name);
8382 var = vpp ? *vpp : NULL;
8383
Denys Vlasenko295fef82009-06-03 12:47:26 +02008384 if (exp == -1) { /* unexporting? */
8385 /* export -n NAME (without =VALUE) */
8386 if (var) {
8387 var->flg_export = 0;
8388 debug_printf_env("%s: unsetenv '%s'\n", __func__, name);
8389 unsetenv(name);
8390 } /* else: export -n NOT_EXISTING_VAR: no-op */
8391 continue;
8392 }
8393 if (exp == 1) { /* exporting? */
8394 /* export NAME (without =VALUE) */
8395 if (var) {
8396 var->flg_export = 1;
8397 debug_printf_env("%s: putenv '%s'\n", __func__, var->varstr);
8398 putenv(var->varstr);
8399 continue;
8400 }
8401 }
8402 /* Exporting non-existing variable.
8403 * bash does not put it in environment,
8404 * but remembers that it is exported,
8405 * and does put it in env when it is set later.
8406 * We just set it to "" and export. */
8407 /* Or, it's "local NAME" (without =VALUE).
8408 * bash sets the value to "". */
8409 name = xasprintf("%s=", name);
8410 } else {
8411 /* (Un)exporting/making local NAME=VALUE */
8412 name = xstrdup(name);
8413 }
8414 set_local_var(name, /*exp:*/ exp, /*lvl:*/ lvl, /*ro:*/ 0);
8415 } while (*++argv);
8416}
8417
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008418static int FAST_FUNC builtin_export(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008419{
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00008420 unsigned opt_unexport;
8421
Denys Vlasenkodf5131c2009-06-07 16:04:17 +02008422#if ENABLE_HUSH_EXPORT_N
8423 /* "!": do not abort on errors */
8424 opt_unexport = getopt32(argv, "!n");
8425 if (opt_unexport == (uint32_t)-1)
8426 return EXIT_FAILURE;
8427 argv += optind;
8428#else
8429 opt_unexport = 0;
8430 argv++;
8431#endif
8432
8433 if (argv[0] == NULL) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008434 char **e = environ;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008435 if (e) {
8436 while (*e) {
8437#if 0
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008438 puts(*e++);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008439#else
8440 /* ash emits: export VAR='VAL'
8441 * bash: declare -x VAR="VAL"
8442 * we follow ash example */
8443 const char *s = *e++;
8444 const char *p = strchr(s, '=');
8445
8446 if (!p) /* wtf? take next variable */
8447 continue;
8448 /* export var= */
8449 printf("export %.*s", (int)(p - s) + 1, s);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008450 print_escaped(p + 1);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008451 putchar('\n');
8452#endif
8453 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008454 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008455 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008456 return EXIT_SUCCESS;
8457 }
8458
Denys Vlasenko295fef82009-06-03 12:47:26 +02008459 helper_export_local(argv, (opt_unexport ? -1 : 1), 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008460
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008461 return EXIT_SUCCESS;
8462}
8463
Denys Vlasenko295fef82009-06-03 12:47:26 +02008464#if ENABLE_HUSH_LOCAL
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008465static int FAST_FUNC builtin_local(char **argv)
Denys Vlasenko295fef82009-06-03 12:47:26 +02008466{
8467 if (G.func_nest_level == 0) {
8468 bb_error_msg("%s: not in a function", argv[0]);
8469 return EXIT_FAILURE; /* bash compat */
8470 }
8471 helper_export_local(argv, 0, G.func_nest_level);
8472 return EXIT_SUCCESS;
8473}
8474#endif
8475
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008476static int FAST_FUNC builtin_trap(char **argv)
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008477{
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008478 int sig;
8479 char *new_cmd;
8480
8481 if (!G.traps)
8482 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
8483
8484 argv++;
8485 if (!*argv) {
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008486 int i;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008487 /* No args: print all trapped */
8488 for (i = 0; i < NSIG; ++i) {
8489 if (G.traps[i]) {
8490 printf("trap -- ");
8491 print_escaped(G.traps[i]);
Denys Vlasenkoe74aaf92009-09-27 02:05:45 +02008492 /* note: bash adds "SIG", but only if invoked
8493 * as "bash". If called as "sh", or if set -o posix,
8494 * then it prints short signal names.
8495 * We are printing short names: */
8496 printf(" %s\n", get_signame(i));
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008497 }
8498 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008499 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008500 return EXIT_SUCCESS;
8501 }
8502
8503 new_cmd = NULL;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008504 /* If first arg is a number: reset all specified signals */
8505 sig = bb_strtou(*argv, NULL, 10);
8506 if (errno == 0) {
8507 int ret;
8508 process_sig_list:
8509 ret = EXIT_SUCCESS;
8510 while (*argv) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008511 sighandler_t handler;
8512
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008513 sig = get_signum(*argv++);
8514 if (sig < 0 || sig >= NSIG) {
8515 ret = EXIT_FAILURE;
8516 /* Mimic bash message exactly */
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008517 bb_perror_msg("trap: %s: invalid signal specification", argv[-1]);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008518 continue;
8519 }
8520
8521 free(G.traps[sig]);
8522 G.traps[sig] = xstrdup(new_cmd);
8523
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008524 debug_printf("trap: setting SIG%s (%i) to '%s'\n",
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008525 get_signame(sig), sig, G.traps[sig]);
8526
8527 /* There is no signal for 0 (EXIT) */
8528 if (sig == 0)
8529 continue;
8530
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008531 if (new_cmd)
8532 handler = (new_cmd[0] ? record_pending_signo : SIG_IGN);
8533 else
8534 /* We are removing trap handler */
8535 handler = pick_sighandler(sig);
Denys Vlasenko0806e402011-05-12 23:06:20 +02008536 install_sighandler(sig, handler);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008537 }
8538 return ret;
8539 }
8540
8541 if (!argv[1]) { /* no second arg */
8542 bb_error_msg("trap: invalid arguments");
8543 return EXIT_FAILURE;
8544 }
8545
8546 /* First arg is "-": reset all specified to default */
8547 /* First arg is "--": skip it, the rest is "handler SIGs..." */
8548 /* Everything else: set arg as signal handler
8549 * (includes "" case, which ignores signal) */
8550 if (argv[0][0] == '-') {
8551 if (argv[0][1] == '\0') { /* "-" */
8552 /* new_cmd remains NULL: "reset these sigs" */
8553 goto reset_traps;
8554 }
8555 if (argv[0][1] == '-' && argv[0][2] == '\0') { /* "--" */
8556 argv++;
8557 }
8558 /* else: "-something", no special meaning */
8559 }
8560 new_cmd = *argv;
8561 reset_traps:
8562 argv++;
8563 goto process_sig_list;
8564}
8565
Mike Frysinger93cadc22009-05-27 17:06:25 -04008566/* http://www.opengroup.org/onlinepubs/9699919799/utilities/type.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008567static int FAST_FUNC builtin_type(char **argv)
Mike Frysinger93cadc22009-05-27 17:06:25 -04008568{
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008569 int ret = EXIT_SUCCESS;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008570
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008571 while (*++argv) {
Mike Frysinger93cadc22009-05-27 17:06:25 -04008572 const char *type;
Denys Vlasenko171932d2009-05-28 17:07:22 +02008573 char *path = NULL;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008574
8575 if (0) {} /* make conditional compile easier below */
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008576 /*else if (find_alias(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008577 type = "an alias";*/
8578#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008579 else if (find_function(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008580 type = "a function";
8581#endif
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008582 else if (find_builtin(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008583 type = "a shell builtin";
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008584 else if ((path = find_in_path(*argv)) != NULL)
8585 type = path;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008586 else {
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008587 bb_error_msg("type: %s: not found", *argv);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008588 ret = EXIT_FAILURE;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008589 continue;
8590 }
Mike Frysinger93cadc22009-05-27 17:06:25 -04008591
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008592 printf("%s is %s\n", *argv, type);
8593 free(path);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008594 }
8595
8596 return ret;
8597}
8598
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008599#if ENABLE_HUSH_JOB
8600/* built-in 'fg' and 'bg' handler */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008601static int FAST_FUNC builtin_fg_bg(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008602{
8603 int i, jobnum;
8604 struct pipe *pi;
8605
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008606 if (!G_interactive_fd)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008607 return EXIT_FAILURE;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008608
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008609 /* If they gave us no args, assume they want the last backgrounded task */
8610 if (!argv[1]) {
Denis Vlasenko87a86552008-07-29 19:43:10 +00008611 for (pi = G.job_list; pi; pi = pi->next) {
8612 if (pi->jobid == G.last_jobid) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008613 goto found;
8614 }
8615 }
8616 bb_error_msg("%s: no current job", argv[0]);
8617 return EXIT_FAILURE;
8618 }
8619 if (sscanf(argv[1], "%%%d", &jobnum) != 1) {
8620 bb_error_msg("%s: bad argument '%s'", argv[0], argv[1]);
8621 return EXIT_FAILURE;
8622 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008623 for (pi = G.job_list; pi; pi = pi->next) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008624 if (pi->jobid == jobnum) {
8625 goto found;
8626 }
8627 }
8628 bb_error_msg("%s: %d: no such job", argv[0], jobnum);
8629 return EXIT_FAILURE;
8630 found:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00008631 /* TODO: bash prints a string representation
8632 * of job being foregrounded (like "sleep 1 | cat") */
Mike Frysinger38478a62009-05-20 04:48:06 -04008633 if (argv[0][0] == 'f' && G_saved_tty_pgrp) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008634 /* Put the job into the foreground. */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008635 tcsetpgrp(G_interactive_fd, pi->pgrp);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008636 }
8637
8638 /* Restart the processes in the job */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008639 debug_printf_jobs("reviving %d procs, pgrp %d\n", pi->num_cmds, pi->pgrp);
8640 for (i = 0; i < pi->num_cmds; i++) {
8641 debug_printf_jobs("reviving pid %d\n", pi->cmds[i].pid);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008642 }
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008643 pi->stopped_cmds = 0;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008644
8645 i = kill(- pi->pgrp, SIGCONT);
8646 if (i < 0) {
8647 if (errno == ESRCH) {
8648 delete_finished_bg_job(pi);
8649 return EXIT_SUCCESS;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008650 }
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008651 bb_perror_msg("kill (SIGCONT)");
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008652 }
8653
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008654 if (argv[0][0] == 'f') {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008655 remove_bg_job(pi);
8656 return checkjobs_and_fg_shell(pi);
8657 }
8658 return EXIT_SUCCESS;
8659}
8660#endif
8661
8662#if ENABLE_HUSH_HELP
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008663static int FAST_FUNC builtin_help(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008664{
8665 const struct built_in_command *x;
8666
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008667 printf(
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008668 "Built-in commands:\n"
8669 "------------------\n");
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008670 for (x = bltins1; x != &bltins1[ARRAY_SIZE(bltins1)]; x++) {
Denys Vlasenko17323a62010-01-28 01:57:05 +01008671 if (x->b_descr)
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008672 printf("%-10s%s\n", x->b_cmd, x->b_descr);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008673 }
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008674 bb_putchar('\n');
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008675 return EXIT_SUCCESS;
8676}
8677#endif
8678
Denys Vlasenkoff463a82013-05-12 02:45:23 +02008679#if MAX_HISTORY && ENABLE_FEATURE_EDITING
Flemming Madsend96ffda2013-04-07 18:47:24 +02008680static int FAST_FUNC builtin_history(char **argv UNUSED_PARAM)
8681{
8682 show_history(G.line_input_state);
8683 return EXIT_SUCCESS;
8684}
8685#endif
8686
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008687#if ENABLE_HUSH_JOB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008688static int FAST_FUNC builtin_jobs(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008689{
8690 struct pipe *job;
8691 const char *status_string;
8692
Denis Vlasenko87a86552008-07-29 19:43:10 +00008693 for (job = G.job_list; job; job = job->next) {
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008694 if (job->alive_cmds == job->stopped_cmds)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008695 status_string = "Stopped";
8696 else
8697 status_string = "Running";
8698
8699 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->cmdtext);
8700 }
8701 return EXIT_SUCCESS;
8702}
8703#endif
8704
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008705#if HUSH_DEBUG
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008706static int FAST_FUNC builtin_memleak(char **argv UNUSED_PARAM)
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008707{
8708 void *p;
8709 unsigned long l;
8710
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008711# ifdef M_TRIM_THRESHOLD
Denys Vlasenko27726cb2009-09-12 14:48:33 +02008712 /* Optional. Reduces probability of false positives */
8713 malloc_trim(0);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008714# endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008715 /* Crude attempt to find where "free memory" starts,
8716 * sans fragmentation. */
8717 p = malloc(240);
8718 l = (unsigned long)p;
8719 free(p);
8720 p = malloc(3400);
8721 if (l < (unsigned long)p) l = (unsigned long)p;
8722 free(p);
8723
8724 if (!G.memleak_value)
8725 G.memleak_value = l;
Denys Vlasenko9038d6f2009-07-15 20:02:19 +02008726
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008727 l -= G.memleak_value;
8728 if ((long)l < 0)
8729 l = 0;
8730 l /= 1024;
8731 if (l > 127)
8732 l = 127;
8733
8734 /* Exitcode is "how many kilobytes we leaked since 1st call" */
8735 return l;
8736}
8737#endif
8738
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008739static int FAST_FUNC builtin_pwd(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008740{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008741 puts(get_cwd(0));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008742 return EXIT_SUCCESS;
8743}
8744
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008745/* Interruptibility of read builtin in bash
8746 * (tested on bash-4.2.8 by sending signals (not by ^C)):
8747 *
8748 * Empty trap makes read ignore corresponding signal, for any signal.
8749 *
8750 * SIGINT:
8751 * - terminates non-interactive shell;
8752 * - interrupts read in interactive shell;
8753 * if it has non-empty trap:
8754 * - executes trap and returns to command prompt in interactive shell;
8755 * - executes trap and returns to read in non-interactive shell;
8756 * SIGTERM:
8757 * - is ignored (does not interrupt) read in interactive shell;
8758 * - terminates non-interactive shell;
8759 * if it has non-empty trap:
8760 * - executes trap and returns to read;
8761 * SIGHUP:
8762 * - terminates shell (regardless of interactivity);
8763 * if it has non-empty trap:
8764 * - executes trap and returns to read;
8765 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008766static int FAST_FUNC builtin_read(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008767{
Denys Vlasenko03dad222010-01-12 23:29:57 +01008768 const char *r;
8769 char *opt_n = NULL;
8770 char *opt_p = NULL;
8771 char *opt_t = NULL;
8772 char *opt_u = NULL;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008773 const char *ifs;
Denys Vlasenko03dad222010-01-12 23:29:57 +01008774 int read_flags;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008775
Denys Vlasenko03dad222010-01-12 23:29:57 +01008776 /* "!": do not abort on errors.
8777 * Option string must start with "sr" to match BUILTIN_READ_xxx
8778 */
8779 read_flags = getopt32(argv, "!srn:p:t:u:", &opt_n, &opt_p, &opt_t, &opt_u);
8780 if (read_flags == (uint32_t)-1)
8781 return EXIT_FAILURE;
8782 argv += optind;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008783 ifs = get_local_var_value("IFS"); /* can be NULL */
8784
8785 again:
Denys Vlasenko03dad222010-01-12 23:29:57 +01008786 r = shell_builtin_read(set_local_var_from_halves,
8787 argv,
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008788 ifs,
Denys Vlasenko03dad222010-01-12 23:29:57 +01008789 read_flags,
8790 opt_n,
8791 opt_p,
8792 opt_t,
8793 opt_u
8794 );
8795
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008796 if ((uintptr_t)r == 1 && errno == EINTR) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008797 unsigned sig = check_and_run_traps();
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008798 if (sig && sig != SIGINT)
8799 goto again;
8800 }
8801
Denys Vlasenko03dad222010-01-12 23:29:57 +01008802 if ((uintptr_t)r > 1) {
8803 bb_error_msg("%s", r);
8804 r = (char*)(uintptr_t)1;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008805 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008806
Denys Vlasenko03dad222010-01-12 23:29:57 +01008807 return (uintptr_t)r;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008808}
8809
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008810/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#set
8811 * built-in 'set' handler
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008812 * SUSv3 says:
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008813 * set [-abCefhmnuvx] [-o option] [argument...]
8814 * set [+abCefhmnuvx] [+o option] [argument...]
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008815 * set -- [argument...]
8816 * set -o
8817 * set +o
8818 * Implementations shall support the options in both their hyphen and
8819 * plus-sign forms. These options can also be specified as options to sh.
8820 * Examples:
8821 * Write out all variables and their values: set
8822 * Set $1, $2, and $3 and set "$#" to 3: set c a b
8823 * Turn on the -x and -v options: set -xv
8824 * Unset all positional parameters: set --
8825 * Set $1 to the value of x, even if it begins with '-' or '+': set -- "$x"
8826 * Set the positional parameters to the expansion of x, even if x expands
8827 * with a leading '-' or '+': set -- $x
8828 *
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008829 * So far, we only support "set -- [argument...]" and some of the short names.
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008830 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008831static int FAST_FUNC builtin_set(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008832{
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008833 int n;
8834 char **pp, **g_argv;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008835 char *arg = *++argv;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008836
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008837 if (arg == NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008838 struct variable *e;
Denis Vlasenko87a86552008-07-29 19:43:10 +00008839 for (e = G.top_var; e; e = e->next)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008840 puts(e->varstr);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008841 return EXIT_SUCCESS;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008842 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008843
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008844 do {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008845 if (strcmp(arg, "--") == 0) {
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008846 ++argv;
8847 goto set_argv;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008848 }
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008849 if (arg[0] != '+' && arg[0] != '-')
8850 break;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008851 for (n = 1; arg[n]; ++n) {
8852 if (set_mode((arg[0] == '-'), arg[n], argv[1]))
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008853 goto error;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008854 if (arg[n] == 'o' && argv[1])
8855 argv++;
8856 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008857 } while ((arg = *++argv) != NULL);
8858 /* Now argv[0] is 1st argument */
8859
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008860 if (arg == NULL)
8861 return EXIT_SUCCESS;
8862 set_argv:
8863
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008864 /* NB: G.global_argv[0] ($0) is never freed/changed */
8865 g_argv = G.global_argv;
8866 if (G.global_args_malloced) {
8867 pp = g_argv;
8868 while (*++pp)
8869 free(*pp);
8870 g_argv[1] = NULL;
8871 } else {
8872 G.global_args_malloced = 1;
8873 pp = xzalloc(sizeof(pp[0]) * 2);
8874 pp[0] = g_argv[0]; /* retain $0 */
8875 g_argv = pp;
8876 }
8877 /* This realloc's G.global_argv */
8878 G.global_argv = pp = add_strings_to_strings(g_argv, argv, /*dup:*/ 1);
8879
8880 n = 1;
8881 while (*++pp)
8882 n++;
8883 G.global_argc = n;
8884
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008885 return EXIT_SUCCESS;
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008886
8887 /* Nothing known, so abort */
8888 error:
8889 bb_error_msg("set: %s: invalid option", arg);
8890 return EXIT_FAILURE;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008891}
8892
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008893static int FAST_FUNC builtin_shift(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008894{
8895 int n = 1;
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008896 argv = skip_dash_dash(argv);
8897 if (argv[0]) {
8898 n = atoi(argv[0]);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008899 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008900 if (n >= 0 && n < G.global_argc) {
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008901 if (G.global_args_malloced) {
8902 int m = 1;
8903 while (m <= n)
8904 free(G.global_argv[m++]);
8905 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008906 G.global_argc -= n;
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008907 memmove(&G.global_argv[1], &G.global_argv[n+1],
8908 G.global_argc * sizeof(G.global_argv[0]));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008909 return EXIT_SUCCESS;
8910 }
8911 return EXIT_FAILURE;
8912}
8913
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008914static int FAST_FUNC builtin_source(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008915{
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008916 char *arg_path, *filename;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008917 FILE *input;
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008918 save_arg_t sv;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008919#if ENABLE_HUSH_FUNCTIONS
8920 smallint sv_flg;
8921#endif
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008922
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008923 argv = skip_dash_dash(argv);
8924 filename = argv[0];
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008925 if (!filename) {
8926 /* bash says: "bash: .: filename argument required" */
8927 return 2; /* bash compat */
8928 }
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008929 arg_path = NULL;
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008930 if (!strchr(filename, '/')) {
8931 arg_path = find_in_path(filename);
8932 if (arg_path)
8933 filename = arg_path;
8934 }
8935 input = fopen_or_warn(filename, "r");
8936 free(arg_path);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008937 if (!input) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008938 /* bb_perror_msg("%s", *argv); - done by fopen_or_warn */
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008939 /* POSIX: non-interactive shell should abort here,
8940 * not merely fail. So far no one complained :)
8941 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008942 return EXIT_FAILURE;
8943 }
8944 close_on_exec_on(fileno(input));
8945
Mike Frysinger885b6f22009-04-18 21:04:25 +00008946#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008947 sv_flg = G.flag_return_in_progress;
8948 /* "we are inside sourced file, ok to use return" */
8949 G.flag_return_in_progress = -1;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008950#endif
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008951 if (argv[1])
8952 save_and_replace_G_args(&sv, argv);
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008953
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008954 parse_and_run_file(input);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008955 fclose(input);
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008956
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008957 if (argv[1])
8958 restore_G_args(&sv, argv);
Mike Frysinger885b6f22009-04-18 21:04:25 +00008959#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008960 G.flag_return_in_progress = sv_flg;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008961#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008962
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008963 return G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008964}
8965
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008966static int FAST_FUNC builtin_umask(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008967{
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008968 int rc;
8969 mode_t mask;
8970
Denys Vlasenko5711a2a2015-10-07 17:55:33 +02008971 rc = 1;
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008972 mask = umask(0);
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008973 argv = skip_dash_dash(argv);
8974 if (argv[0]) {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008975 mode_t old_mask = mask;
8976
Denys Vlasenko6283f982015-10-07 16:56:20 +02008977 /* numeric umasks are taken as-is */
8978 /* symbolic umasks are inverted: "umask a=rx" calls umask(222) */
8979 if (!isdigit(argv[0][0]))
8980 mask ^= 0777;
Denys Vlasenko5711a2a2015-10-07 17:55:33 +02008981 mask = bb_parse_mode(argv[0], mask);
Denys Vlasenko6283f982015-10-07 16:56:20 +02008982 if (!isdigit(argv[0][0]))
8983 mask ^= 0777;
Denys Vlasenko5711a2a2015-10-07 17:55:33 +02008984 if ((unsigned)mask > 0777) {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008985 mask = old_mask;
8986 /* bash messages:
8987 * bash: umask: 'q': invalid symbolic mode operator
8988 * bash: umask: 999: octal number out of range
8989 */
Denys Vlasenko44c86ce2010-05-20 04:22:55 +02008990 bb_error_msg("%s: invalid mode '%s'", "umask", argv[0]);
Denys Vlasenko5711a2a2015-10-07 17:55:33 +02008991 rc = 0;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008992 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008993 } else {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008994 /* Mimic bash */
8995 printf("%04o\n", (unsigned) mask);
8996 /* fall through and restore mask which we set to 0 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008997 }
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008998 umask(mask);
8999
9000 return !rc; /* rc != 0 - success */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00009001}
9002
Mike Frysingerd690f682009-03-30 06:50:54 +00009003/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#unset */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009004static int FAST_FUNC builtin_unset(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00009005{
Mike Frysingerd690f682009-03-30 06:50:54 +00009006 int ret;
Denis Vlasenko28e67962009-04-26 23:22:40 +00009007 unsigned opts;
Mike Frysingerd690f682009-03-30 06:50:54 +00009008
Denis Vlasenko28e67962009-04-26 23:22:40 +00009009 /* "!": do not abort on errors */
9010 /* "+": stop at 1st non-option */
9011 opts = getopt32(argv, "!+vf");
9012 if (opts == (unsigned)-1)
9013 return EXIT_FAILURE;
9014 if (opts == 3) {
9015 bb_error_msg("unset: -v and -f are exclusive");
9016 return EXIT_FAILURE;
Mike Frysingerd690f682009-03-30 06:50:54 +00009017 }
Denis Vlasenko28e67962009-04-26 23:22:40 +00009018 argv += optind;
Mike Frysingerd690f682009-03-30 06:50:54 +00009019
9020 ret = EXIT_SUCCESS;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009021 while (*argv) {
Denis Vlasenko28e67962009-04-26 23:22:40 +00009022 if (!(opts & 2)) { /* not -f */
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009023 if (unset_local_var(*argv)) {
9024 /* unset <nonexistent_var> doesn't fail.
9025 * Error is when one tries to unset RO var.
9026 * Message was printed by unset_local_var. */
Mike Frysingerd690f682009-03-30 06:50:54 +00009027 ret = EXIT_FAILURE;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009028 }
Mike Frysingerd690f682009-03-30 06:50:54 +00009029 }
Denis Vlasenko40e84372009-04-18 11:23:38 +00009030#if ENABLE_HUSH_FUNCTIONS
9031 else {
9032 unset_func(*argv);
9033 }
9034#endif
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009035 argv++;
Mike Frysingerd690f682009-03-30 06:50:54 +00009036 }
9037 return ret;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00009038}
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009039
Mike Frysinger56bdea12009-03-28 20:01:58 +00009040/* http://www.opengroup.org/onlinepubs/9699919799/utilities/wait.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009041static int FAST_FUNC builtin_wait(char **argv)
Mike Frysinger56bdea12009-03-28 20:01:58 +00009042{
9043 int ret = EXIT_SUCCESS;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009044 int status;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009045
Denys Vlasenkob131cce2010-05-20 03:39:43 +02009046 argv = skip_dash_dash(argv);
9047 if (argv[0] == NULL) {
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009048 /* Don't care about wait results */
9049 /* Note 1: must wait until there are no more children */
9050 /* Note 2: must be interruptible */
9051 /* Examples:
9052 * $ sleep 3 & sleep 6 & wait
9053 * [1] 30934 sleep 3
9054 * [2] 30935 sleep 6
9055 * [1] Done sleep 3
9056 * [2] Done sleep 6
9057 * $ sleep 3 & sleep 6 & wait
9058 * [1] 30936 sleep 3
9059 * [2] 30937 sleep 6
9060 * [1] Done sleep 3
9061 * ^C <-- after ~4 sec from keyboard
9062 * $
9063 */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009064 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009065 int sig;
9066 sigset_t oldset, allsigs;
9067
9068 /* waitpid is not interruptible by SA_RESTARTed
9069 * signals which we use. Thus, this ugly dance:
9070 */
9071
9072 /* Make sure possible SIGCHLD is stored in kernel's
9073 * pending signal mask before we call waitpid.
9074 * Or else we may race with SIGCHLD, lose it,
9075 * and get stuck in sigwaitinfo...
9076 */
9077 sigfillset(&allsigs);
9078 sigprocmask(SIG_SETMASK, &allsigs, &oldset);
9079
9080 if (!sigisemptyset(&G.pending_set)) {
9081 /* Crap! we raced with some signal! */
9082 // sig = 0;
9083 goto restore;
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009084 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009085
9086 checkjobs(NULL); /* waitpid(WNOHANG) inside */
9087 if (errno == ECHILD) {
9088 sigprocmask(SIG_SETMASK, &oldset, NULL);
9089 break;
9090 }
9091
9092 /* Wait for SIGCHLD or any other signal */
9093 //sig = sigwaitinfo(&allsigs, NULL);
9094 /* It is vitally important for sigsuspend that SIGCHLD has non-DFL handler! */
9095 /* Note: sigsuspend invokes signal handler */
9096 sigsuspend(&oldset);
9097 restore:
9098 sigprocmask(SIG_SETMASK, &oldset, NULL);
9099
9100 /* So, did we get a signal? */
9101 //if (sig > 0)
9102 // raise(sig); /* run handler */
9103 sig = check_and_run_traps();
9104 if (sig /*&& sig != SIGCHLD - always true */) {
9105 /* see note 2 */
9106 ret = 128 + sig;
9107 break;
9108 }
9109 /* SIGCHLD, or no signal, or ignored one, such as SIGQUIT. Repeat */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009110 }
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009111 return ret;
9112 }
Mike Frysinger56bdea12009-03-28 20:01:58 +00009113
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009114 /* This is probably buggy wrt interruptible-ness */
Denis Vlasenkod5762932009-03-31 11:22:57 +00009115 while (*argv) {
9116 pid_t pid = bb_strtou(*argv, NULL, 10);
Mike Frysinger40b8dc42009-03-29 00:50:30 +00009117 if (errno) {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009118 /* mimic bash message */
9119 bb_error_msg("wait: '%s': not a pid or valid job spec", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009120 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00009121 }
9122 if (waitpid(pid, &status, 0) == pid) {
Denys Vlasenko85378cd2015-10-11 21:47:11 +02009123 ret = WEXITSTATUS(status);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009124 if (WIFSIGNALED(status))
9125 ret = 128 + WTERMSIG(status);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009126 } else {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009127 bb_perror_msg("wait %s", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009128 ret = 127;
9129 }
Denis Vlasenkod5762932009-03-31 11:22:57 +00009130 argv++;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009131 }
9132
9133 return ret;
9134}
9135
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009136#if ENABLE_HUSH_LOOPS || ENABLE_HUSH_FUNCTIONS
9137static unsigned parse_numeric_argv1(char **argv, unsigned def, unsigned def_min)
9138{
9139 if (argv[1]) {
9140 def = bb_strtou(argv[1], NULL, 10);
9141 if (errno || def < def_min || argv[2]) {
9142 bb_error_msg("%s: bad arguments", argv[0]);
9143 def = UINT_MAX;
9144 }
9145 }
9146 return def;
9147}
9148#endif
9149
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009150#if ENABLE_HUSH_LOOPS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009151static int FAST_FUNC builtin_break(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009152{
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009153 unsigned depth;
Denis Vlasenko87a86552008-07-29 19:43:10 +00009154 if (G.depth_of_loop == 0) {
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009155 bb_error_msg("%s: only meaningful in a loop", argv[0]);
Denis Vlasenkofcf37c32008-07-29 11:37:15 +00009156 return EXIT_SUCCESS; /* bash compat */
9157 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00009158 G.flag_break_continue++; /* BC_BREAK = 1 */
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009159
9160 G.depth_break_continue = depth = parse_numeric_argv1(argv, 1, 1);
9161 if (depth == UINT_MAX)
9162 G.flag_break_continue = BC_BREAK;
9163 if (G.depth_of_loop < depth)
Denis Vlasenko87a86552008-07-29 19:43:10 +00009164 G.depth_break_continue = G.depth_of_loop;
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009165
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009166 return EXIT_SUCCESS;
9167}
9168
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009169static int FAST_FUNC builtin_continue(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009170{
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009171 G.flag_break_continue = 1; /* BC_CONTINUE = 2 = 1+1 */
9172 return builtin_break(argv);
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009173}
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009174#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009175
9176#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009177static int FAST_FUNC builtin_return(char **argv)
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009178{
9179 int rc;
9180
9181 if (G.flag_return_in_progress != -1) {
9182 bb_error_msg("%s: not in a function or sourced script", argv[0]);
9183 return EXIT_FAILURE; /* bash compat */
9184 }
9185
9186 G.flag_return_in_progress = 1;
9187
9188 /* bash:
9189 * out of range: wraps around at 256, does not error out
9190 * non-numeric param:
9191 * f() { false; return qwe; }; f; echo $?
9192 * bash: return: qwe: numeric argument required <== we do this
9193 * 255 <== we also do this
9194 */
9195 rc = parse_numeric_argv1(argv, G.last_exitcode, 0);
9196 return rc;
9197}
9198#endif