blob: 92d79018019fca42014e981d379307c143f5a639 [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,
445 /* Not an assigment, but next word may be: "if v=xyz cmd;" */
446 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
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001482/* After [v]fork, in child: do not restore tty pgrp on xfunc death */
Denys Vlasenko8391c482010-05-22 17:50:43 +02001483# define disable_restore_tty_pgrp_on_exit() (die_sleep = 0)
Denis Vlasenko25af86f2009-04-07 13:29:27 +00001484/* After [v]fork, in parent: restore tty pgrp on xfunc death */
Denys Vlasenko8391c482010-05-22 17:50:43 +02001485# define enable_restore_tty_pgrp_on_exit() (die_sleep = -1)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001486
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001487/* Restores tty foreground process group, and exits.
1488 * May be called as signal handler for fatal signal
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001489 * (will resend signal to itself, producing correct exit state)
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001490 * or called directly with -EXITCODE.
1491 * We also call it if xfunc is exiting. */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +00001492static void sigexit(int sig) NORETURN;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001493static void sigexit(int sig)
1494{
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001495 /* Careful: we can end up here after [v]fork. Do not restore
Denis Vlasenko7b830e72009-03-31 13:05:32 +00001496 * tty pgrp then, only top-level shell process does that */
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001497 if (G_saved_tty_pgrp && getpid() == G.root_pid) {
1498 /* Disable all signals: job control, SIGPIPE, etc.
1499 * Mostly paranoid measure, to prevent infinite SIGTTOU.
1500 */
1501 sigprocmask_allsigs(SIG_BLOCK);
Mike Frysinger38478a62009-05-20 04:48:06 -04001502 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
Denys Vlasenkoebc1ee22011-05-12 10:59:18 +02001503 }
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001504
1505 /* Not a signal, just exit */
1506 if (sig <= 0)
1507 _exit(- sig);
1508
Denis Vlasenko400d8bb2008-02-24 13:36:01 +00001509 kill_myself_with_sig(sig); /* does not return */
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00001510}
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001511#else
1512
Denys Vlasenko8391c482010-05-22 17:50:43 +02001513# define disable_restore_tty_pgrp_on_exit() ((void)0)
1514# define enable_restore_tty_pgrp_on_exit() ((void)0)
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00001515
Denis Vlasenkoe0755e52009-04-03 21:16:45 +00001516#endif
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001517
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001518static sighandler_t pick_sighandler(unsigned sig)
1519{
1520 sighandler_t handler = SIG_DFL;
1521 if (sig < sizeof(unsigned)*8) {
1522 unsigned sigmask = (1 << sig);
1523
1524#if ENABLE_HUSH_JOB
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001525 /* is sig fatal? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001526 if (G_fatal_sig_mask & sigmask)
1527 handler = sigexit;
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001528 else
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001529#endif
1530 /* sig has special handling? */
Denys Vlasenko75e77de2011-05-12 13:12:47 +02001531 if (G.special_sig_mask & sigmask) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001532 handler = record_pending_signo;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001533 /* TTIN/TTOU/TSTP can't be set to record_pending_signo
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001534 * in order to ignore them: they will be raised
Denys Vlasenkof58f7052011-05-12 02:10:33 +02001535 * in an endless loop when we try to do some
1536 * terminal ioctls! We do have to _ignore_ these.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001537 */
1538 if (SPECIAL_JOBSTOP_SIGS & sigmask)
1539 handler = SIG_IGN;
Denys Vlasenko0c40a732011-05-12 09:50:12 +02001540 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001541 }
1542 return handler;
1543}
1544
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001545/* Restores tty foreground process group, and exits. */
1546static void hush_exit(int exitcode) NORETURN;
1547static void hush_exit(int exitcode)
1548{
Denys Vlasenkobede2152011-09-04 16:12:33 +02001549#if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
1550 save_history(G.line_input_state);
1551#endif
1552
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01001553 fflush_all();
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00001554 if (G.exiting <= 0 && G.traps && G.traps[0] && G.traps[0][0]) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001555 char *argv[3];
1556 /* argv[0] is unused */
1557 argv[1] = G.traps[0];
1558 argv[2] = NULL;
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001559 G.exiting = 1; /* prevent EXIT trap recursion */
Denys Vlasenkoa110c902010-09-12 15:38:04 +02001560 /* Note: G.traps[0] is not cleared!
Denys Vlasenkode8c3f62010-09-12 16:13:44 +02001561 * "trap" will still show it, if executed
1562 * in the handler */
1563 builtin_eval(argv);
Denis Vlasenkod5762932009-03-31 11:22:57 +00001564 }
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001565
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01001566#if ENABLE_FEATURE_CLEAN_UP
1567 {
1568 struct variable *cur_var;
1569 if (G.cwd != bb_msg_unknown)
1570 free((char*)G.cwd);
1571 cur_var = G.top_var;
1572 while (cur_var) {
1573 struct variable *tmp = cur_var;
1574 if (!cur_var->max_len)
1575 free(cur_var->varstr);
1576 cur_var = cur_var->next;
1577 free(tmp);
1578 }
1579 }
1580#endif
1581
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001582#if ENABLE_HUSH_JOB
Denys Vlasenko8131eea2009-11-02 14:19:51 +01001583 fflush_all();
Denis Vlasenkoabedaac2009-03-31 12:03:40 +00001584 sigexit(- (exitcode & 0xff));
1585#else
1586 exit(exitcode);
1587#endif
Mike Frysinger9f8128f2009-03-29 23:49:37 +00001588}
1589
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02001590
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001591//TODO: return a mask of ALL handled sigs?
1592static int check_and_run_traps(void)
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001593{
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001594 int last_sig = 0;
1595
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001596 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001597 int sig;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02001598
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001599 if (sigisemptyset(&G.pending_set))
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001600 break;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001601 sig = 0;
1602 do {
1603 sig++;
1604 if (sigismember(&G.pending_set, sig)) {
1605 sigdelset(&G.pending_set, sig);
1606 goto got_sig;
1607 }
1608 } while (sig < NSIG);
1609 break;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001610 got_sig:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001611 if (G.traps && G.traps[sig]) {
1612 if (G.traps[sig][0]) {
1613 /* We have user-defined handler */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001614 smalluint save_rcode;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001615 char *argv[3];
1616 /* argv[0] is unused */
1617 argv[1] = G.traps[sig];
1618 argv[2] = NULL;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001619 save_rcode = G.last_exitcode;
1620 builtin_eval(argv);
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001621 G.last_exitcode = save_rcode;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001622 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001623 } /* else: "" trap, ignoring signal */
1624 continue;
1625 }
1626 /* not a trap: special action */
1627 switch (sig) {
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001628 case SIGINT:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001629 /* Builtin was ^C'ed, make it look prettier: */
1630 bb_putchar('\n');
1631 G.flag_SIGINT = 1;
Denys Vlasenkob8709032011-05-08 21:20:01 +02001632 last_sig = sig;
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001633 break;
1634#if ENABLE_HUSH_JOB
1635 case SIGHUP: {
1636 struct pipe *job;
1637 /* bash is observed to signal whole process groups,
1638 * not individual processes */
1639 for (job = G.job_list; job; job = job->next) {
1640 if (job->pgrp <= 0)
1641 continue;
1642 debug_printf_exec("HUPing pgrp %d\n", job->pgrp);
1643 if (kill(- job->pgrp, SIGHUP) == 0)
1644 kill(- job->pgrp, SIGCONT);
1645 }
1646 sigexit(SIGHUP);
1647 }
1648#endif
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001649#if ENABLE_HUSH_FAST
1650 case SIGCHLD:
1651 G.count_SIGCHLD++;
1652//bb_error_msg("[%d] check_and_run_traps: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
1653 /* Note:
1654 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1655 * This simplifies wait builtin a bit.
1656 */
1657 break;
1658#endif
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001659 default: /* ignored: */
1660 /* SIGTERM, SIGQUIT, SIGTTIN, SIGTTOU, SIGTSTP */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02001661 /* Note:
1662 * We dont do 'last_sig = sig' here -> NOT returning this sig.
1663 * Example: wait is not interrupted by TERM
Denys Vlasenkob8709032011-05-08 21:20:01 +02001664 * in interactive shell, because TERM is ignored.
1665 */
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00001666 break;
1667 }
1668 }
1669 return last_sig;
1670}
1671
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00001672
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001673static const char *get_cwd(int force)
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001674{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02001675 if (force || G.cwd == NULL) {
1676 /* xrealloc_getcwd_or_warn(arg) calls free(arg),
1677 * we must not try to free(bb_msg_unknown) */
1678 if (G.cwd == bb_msg_unknown)
1679 G.cwd = NULL;
1680 G.cwd = xrealloc_getcwd_or_warn((char *)G.cwd);
1681 if (!G.cwd)
1682 G.cwd = bb_msg_unknown;
1683 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00001684 return G.cwd;
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00001685}
1686
Denis Vlasenko83506862007-11-23 13:11:42 +00001687
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001688/*
1689 * Shell and environment variable support
1690 */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001691static struct variable **get_ptr_to_local_var(const char *name, unsigned len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001692{
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001693 struct variable **pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001694 struct variable *cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001695
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001696 pp = &G.top_var;
1697 while ((cur = *pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001698 if (strncmp(cur->varstr, name, len) == 0 && cur->varstr[len] == '=')
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001699 return pp;
1700 pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001701 }
1702 return NULL;
1703}
1704
Denys Vlasenko03dad222010-01-12 23:29:57 +01001705static const char* FAST_FUNC get_local_var_value(const char *name)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001706{
Denys Vlasenko29082232010-07-16 13:52:32 +02001707 struct variable **vpp;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001708 unsigned len = strlen(name);
Denys Vlasenko29082232010-07-16 13:52:32 +02001709
1710 if (G.expanded_assignments) {
1711 char **cpp = G.expanded_assignments;
Denys Vlasenko29082232010-07-16 13:52:32 +02001712 while (*cpp) {
1713 char *cp = *cpp;
1714 if (strncmp(cp, name, len) == 0 && cp[len] == '=')
1715 return cp + len + 1;
1716 cpp++;
1717 }
1718 }
1719
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001720 vpp = get_ptr_to_local_var(name, len);
Denys Vlasenko29082232010-07-16 13:52:32 +02001721 if (vpp)
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001722 return (*vpp)->varstr + len + 1;
Denys Vlasenko29082232010-07-16 13:52:32 +02001723
Denys Vlasenkodea47882009-10-09 15:40:49 +02001724 if (strcmp(name, "PPID") == 0)
1725 return utoa(G.root_ppid);
1726 // bash compat: UID? EUID?
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001727#if ENABLE_HUSH_RANDOM_SUPPORT
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001728 if (strcmp(name, "RANDOM") == 0)
Denys Vlasenko20b3d142009-10-09 20:59:39 +02001729 return utoa(next_random(&G.random_gen));
1730#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001731 return NULL;
1732}
1733
1734/* str holds "NAME=VAL" and is expected to be malloced.
Mike Frysinger6379bb42009-03-28 18:55:03 +00001735 * We take ownership of it.
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001736 * flg_export:
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001737 * 0: do not change export flag
1738 * (if creating new variable, flag will be 0)
1739 * 1: set export flag and putenv the variable
1740 * -1: clear export flag and unsetenv the variable
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001741 * flg_read_only is set only when we handle -R var=val
Mike Frysinger6379bb42009-03-28 18:55:03 +00001742 */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001743#if !BB_MMU && ENABLE_HUSH_LOCAL
1744/* all params are used */
1745#elif BB_MMU && ENABLE_HUSH_LOCAL
1746#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1747 set_local_var(str, flg_export, local_lvl)
1748#elif BB_MMU && !ENABLE_HUSH_LOCAL
1749#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001750 set_local_var(str, flg_export)
Denys Vlasenko295fef82009-06-03 12:47:26 +02001751#elif !BB_MMU && !ENABLE_HUSH_LOCAL
1752#define set_local_var(str, flg_export, local_lvl, flg_read_only) \
1753 set_local_var(str, flg_export, flg_read_only)
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001754#endif
Denys Vlasenko295fef82009-06-03 12:47:26 +02001755static int set_local_var(char *str, int flg_export, int local_lvl, int flg_read_only)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001756{
Denys Vlasenko295fef82009-06-03 12:47:26 +02001757 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001758 struct variable *cur;
Denis Vlasenko950bd722009-04-21 11:23:56 +00001759 char *eq_sign;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001760 int name_len;
1761
Denis Vlasenko950bd722009-04-21 11:23:56 +00001762 eq_sign = strchr(str, '=');
1763 if (!eq_sign) { /* not expected to ever happen? */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001764 free(str);
1765 return -1;
1766 }
1767
Denis Vlasenko950bd722009-04-21 11:23:56 +00001768 name_len = eq_sign - str + 1; /* including '=' */
Denys Vlasenko295fef82009-06-03 12:47:26 +02001769 var_pp = &G.top_var;
1770 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001771 if (strncmp(cur->varstr, str, name_len) != 0) {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001772 var_pp = &cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001773 continue;
1774 }
1775 /* We found an existing var with this name */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001776 if (cur->flg_read_only) {
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001777#if !BB_MMU
1778 if (!flg_read_only)
1779#endif
1780 bb_error_msg("%s: readonly variable", str);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001781 free(str);
1782 return -1;
1783 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001784 if (flg_export == -1) { // "&& cur->flg_export" ?
Denis Vlasenko950bd722009-04-21 11:23:56 +00001785 debug_printf_env("%s: unsetenv '%s'\n", __func__, str);
1786 *eq_sign = '\0';
1787 unsetenv(str);
1788 *eq_sign = '=';
1789 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001790#if ENABLE_HUSH_LOCAL
1791 if (cur->func_nest_level < local_lvl) {
1792 /* New variable is declared as local,
1793 * and existing one is global, or local
1794 * from enclosing function.
1795 * Remove and save old one: */
1796 *var_pp = cur->next;
1797 cur->next = *G.shadowed_vars_pp;
1798 *G.shadowed_vars_pp = cur;
1799 /* bash 3.2.33(1) and exported vars:
1800 * # export z=z
1801 * # f() { local z=a; env | grep ^z; }
1802 * # f
1803 * z=a
1804 * # env | grep ^z
1805 * z=z
1806 */
1807 if (cur->flg_export)
1808 flg_export = 1;
1809 break;
1810 }
1811#endif
Denis Vlasenko950bd722009-04-21 11:23:56 +00001812 if (strcmp(cur->varstr + name_len, eq_sign + 1) == 0) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001813 free_and_exp:
1814 free(str);
1815 goto exp;
1816 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001817 if (cur->max_len != 0) {
1818 if (cur->max_len >= strlen(str)) {
1819 /* This one is from startup env, reuse space */
1820 strcpy(cur->varstr, str);
1821 goto free_and_exp;
1822 }
1823 } else {
1824 /* max_len == 0 signifies "malloced" var, which we can
1825 * (and has to) free */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001826 free(cur->varstr);
Denys Vlasenko295fef82009-06-03 12:47:26 +02001827 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001828 cur->max_len = 0;
1829 goto set_str_and_exp;
1830 }
1831
Denys Vlasenko295fef82009-06-03 12:47:26 +02001832 /* Not found - create new variable struct */
1833 cur = xzalloc(sizeof(*cur));
1834#if ENABLE_HUSH_LOCAL
1835 cur->func_nest_level = local_lvl;
1836#endif
1837 cur->next = *var_pp;
1838 *var_pp = cur;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001839
1840 set_str_and_exp:
1841 cur->varstr = str;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001842#if !BB_MMU
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00001843 cur->flg_read_only = flg_read_only;
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00001844#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001845 exp:
Mike Frysinger6379bb42009-03-28 18:55:03 +00001846 if (flg_export == 1)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001847 cur->flg_export = 1;
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001848 if (name_len == 4 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1849 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001850 if (cur->flg_export) {
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00001851 if (flg_export == -1) {
1852 cur->flg_export = 0;
1853 /* unsetenv was already done */
1854 } else {
1855 debug_printf_env("%s: putenv '%s'\n", __func__, cur->varstr);
1856 return putenv(cur->varstr);
1857 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001858 }
1859 return 0;
1860}
1861
Denys Vlasenko6db47842009-09-05 20:15:17 +02001862/* Used at startup and after each cd */
1863static void set_pwd_var(int exp)
1864{
1865 set_local_var(xasprintf("PWD=%s", get_cwd(/*force:*/ 1)),
1866 /*exp:*/ exp, /*lvl:*/ 0, /*ro:*/ 0);
1867}
1868
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001869static int unset_local_var_len(const char *name, int name_len)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001870{
1871 struct variable *cur;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001872 struct variable **var_pp;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001873
1874 if (!name)
Mike Frysingerd690f682009-03-30 06:50:54 +00001875 return EXIT_SUCCESS;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001876 var_pp = &G.top_var;
1877 while ((cur = *var_pp) != NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001878 if (strncmp(cur->varstr, name, name_len) == 0 && cur->varstr[name_len] == '=') {
1879 if (cur->flg_read_only) {
1880 bb_error_msg("%s: readonly variable", name);
Mike Frysingerd690f682009-03-30 06:50:54 +00001881 return EXIT_FAILURE;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001882 }
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001883 *var_pp = cur->next;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001884 debug_printf_env("%s: unsetenv '%s'\n", __func__, cur->varstr);
1885 bb_unsetenv(cur->varstr);
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00001886 if (name_len == 3 && cur->varstr[0] == 'P' && cur->varstr[1] == 'S')
1887 cmdedit_update_prompt();
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001888 if (!cur->max_len)
1889 free(cur->varstr);
1890 free(cur);
Mike Frysingerd690f682009-03-30 06:50:54 +00001891 return EXIT_SUCCESS;
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 }
Mike Frysingerd690f682009-03-30 06:50:54 +00001895 return EXIT_SUCCESS;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001896}
1897
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001898static int unset_local_var(const char *name)
1899{
1900 return unset_local_var_len(name, strlen(name));
1901}
1902
1903static void unset_vars(char **strings)
1904{
1905 char **v;
1906
1907 if (!strings)
1908 return;
1909 v = strings;
1910 while (*v) {
1911 const char *eq = strchrnul(*v, '=');
1912 unset_local_var_len(*v, (int)(eq - *v));
1913 v++;
1914 }
1915 free(strings);
1916}
1917
Denys Vlasenko03dad222010-01-12 23:29:57 +01001918static void FAST_FUNC set_local_var_from_halves(const char *name, const char *val)
Mike Frysinger98c52642009-04-02 10:02:37 +00001919{
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001920 char *var = xasprintf("%s=%s", name, val);
Denys Vlasenko03dad222010-01-12 23:29:57 +01001921 set_local_var(var, /*flags:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
Mike Frysinger98c52642009-04-02 10:02:37 +00001922}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001923
Denis Vlasenkob29eb6e2009-04-02 13:46:27 +00001924
Denis Vlasenko424f79b2009-03-22 14:23:34 +00001925/*
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001926 * Helpers for "var1=val1 var2=val2 cmd" feature
1927 */
1928static void add_vars(struct variable *var)
1929{
1930 struct variable *next;
1931
1932 while (var) {
1933 next = var->next;
1934 var->next = G.top_var;
1935 G.top_var = var;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001936 if (var->flg_export) {
1937 debug_printf_env("%s: restoring exported '%s'\n", __func__, var->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001938 putenv(var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001939 } else {
Denys Vlasenko295fef82009-06-03 12:47:26 +02001940 debug_printf_env("%s: restoring variable '%s'\n", __func__, var->varstr);
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001941 }
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001942 var = next;
1943 }
1944}
1945
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001946static struct variable *set_vars_and_save_old(char **strings)
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001947{
1948 char **s;
1949 struct variable *old = NULL;
1950
1951 if (!strings)
1952 return old;
1953 s = strings;
1954 while (*s) {
1955 struct variable *var_p;
1956 struct variable **var_pp;
1957 char *eq;
1958
1959 eq = strchr(*s, '=');
1960 if (eq) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02001961 var_pp = get_ptr_to_local_var(*s, eq - *s);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001962 if (var_pp) {
1963 /* Remove variable from global linked list */
1964 var_p = *var_pp;
Denys Vlasenkoacdc49c2009-05-04 01:58:10 +02001965 debug_printf_env("%s: removing '%s'\n", __func__, var_p->varstr);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001966 *var_pp = var_p->next;
1967 /* Add it to returned list */
1968 var_p->next = old;
1969 old = var_p;
1970 }
Denys Vlasenko295fef82009-06-03 12:47:26 +02001971 set_local_var(*s, /*exp:*/ 1, /*lvl:*/ 0, /*ro:*/ 0);
Denys Vlasenkocb6ff252009-05-04 00:14:30 +02001972 }
1973 s++;
1974 }
1975 return old;
1976}
1977
1978
1979/*
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02001980 * Unicode helper
1981 */
1982static void reinit_unicode_for_hush(void)
1983{
1984 /* Unicode support should be activated even if LANG is set
1985 * _during_ shell execution, not only if it was set when
1986 * shell was started. Therefore, re-check LANG every time:
1987 */
Denys Vlasenko841f8332014-08-13 10:09:49 +02001988 if (ENABLE_FEATURE_CHECK_UNICODE_IN_ENV
1989 || ENABLE_UNICODE_USING_LOCALE
1990 ) {
1991 const char *s = get_local_var_value("LC_ALL");
1992 if (!s) s = get_local_var_value("LC_CTYPE");
1993 if (!s) s = get_local_var_value("LANG");
1994 reinit_unicode(s);
1995 }
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02001996}
1997
1998
1999/*
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002000 * in_str support
2001 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002002static int FAST_FUNC static_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002003{
Denys Vlasenko8391c482010-05-22 17:50:43 +02002004 int ch = *i->p;
2005 if (ch != '\0') {
2006 i->p++;
Denys Vlasenkocecbc982011-03-30 18:54:52 +02002007 i->last_char = ch;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002008 return ch;
Denys Vlasenko8391c482010-05-22 17:50:43 +02002009 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002010 return EOF;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002011}
2012
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002013static int FAST_FUNC static_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002014{
2015 return *i->p;
2016}
2017
2018#if ENABLE_HUSH_INTERACTIVE
2019
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002020static void cmdedit_update_prompt(void)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002021{
Mike Frysingerec2c6552009-03-28 12:24:44 +00002022 if (ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002023 G.PS1 = get_local_var_value("PS1");
Mike Frysingerec2c6552009-03-28 12:24:44 +00002024 if (G.PS1 == NULL)
2025 G.PS1 = "\\w \\$ ";
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002026 G.PS2 = get_local_var_value("PS2");
Denys Vlasenko690ad242009-04-30 21:24:24 +02002027 } else {
Mike Frysingerec2c6552009-03-28 12:24:44 +00002028 G.PS1 = NULL;
Denys Vlasenko690ad242009-04-30 21:24:24 +02002029 }
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00002030 if (G.PS2 == NULL)
2031 G.PS2 = "> ";
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002032}
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002033
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002034static const char *setup_prompt_string(int promptmode)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002035{
2036 const char *prompt_str;
2037 debug_printf("setup_prompt_string %d ", promptmode);
Mike Frysingerec2c6552009-03-28 12:24:44 +00002038 if (!ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
2039 /* Set up the prompt */
2040 if (promptmode == 0) { /* PS1 */
2041 free((char*)G.PS1);
Denys Vlasenko6db47842009-09-05 20:15:17 +02002042 /* bash uses $PWD value, even if it is set by user.
2043 * It uses current dir only if PWD is unset.
2044 * We always use current dir. */
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02002045 G.PS1 = xasprintf("%s %c ", get_cwd(0), (geteuid() != 0) ? '$' : '#');
Mike Frysingerec2c6552009-03-28 12:24:44 +00002046 prompt_str = G.PS1;
2047 } else
2048 prompt_str = G.PS2;
2049 } else
2050 prompt_str = (promptmode == 0) ? G.PS1 : G.PS2;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002051 debug_printf("result '%s'\n", prompt_str);
2052 return prompt_str;
2053}
2054
2055static void get_user_input(struct in_str *i)
2056{
2057 int r;
2058 const char *prompt_str;
2059
2060 prompt_str = setup_prompt_string(i->promptmode);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002061# if ENABLE_FEATURE_EDITING
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002062 /* Enable command line editing only while a command line
2063 * is actually being read */
2064 do {
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02002065 reinit_unicode_for_hush();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002066 G.flag_SIGINT = 0;
2067 /* buglet: SIGINT will not make new prompt to appear _at once_,
2068 * only after <Enter>. (^C will work) */
Denys Vlasenko66c5b122011-02-08 05:07:02 +01002069 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 +00002070 /* catch *SIGINT* etc (^C is handled by read_line_input) */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002071 check_and_run_traps();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002072 } while (r == 0 || G.flag_SIGINT); /* repeat if ^C or SIGINT */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002073 i->eof_flag = (r < 0);
2074 if (i->eof_flag) { /* EOF/error detected */
2075 G.user_input_buf[0] = EOF; /* yes, it will be truncated, it's ok */
2076 G.user_input_buf[1] = '\0';
2077 }
Denys Vlasenko8391c482010-05-22 17:50:43 +02002078# else
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002079 do {
2080 G.flag_SIGINT = 0;
Denys Vlasenkob8709032011-05-08 21:20:01 +02002081 if (i->last_char == '\0' || i->last_char == '\n') {
2082 /* Why check_and_run_traps here? Try this interactively:
2083 * $ trap 'echo INT' INT; (sleep 2; kill -INT $$) &
2084 * $ <[enter], repeatedly...>
2085 * Without check_and_run_traps, handler never runs.
2086 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02002087 check_and_run_traps();
Denys Vlasenkob8709032011-05-08 21:20:01 +02002088 fputs(prompt_str, stdout);
2089 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002090 fflush_all();
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002091 G.user_input_buf[0] = r = fgetc(i->file);
2092 /*G.user_input_buf[1] = '\0'; - already is and never changed */
Denis Vlasenko422cd7c2009-03-31 12:41:52 +00002093 } while (G.flag_SIGINT);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002094 i->eof_flag = (r == EOF);
Denys Vlasenko8391c482010-05-22 17:50:43 +02002095# endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002096 i->p = G.user_input_buf;
2097}
2098
2099#endif /* INTERACTIVE */
2100
2101/* This is the magic location that prints prompts
2102 * and gets data back from the user */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002103static int FAST_FUNC file_get(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002104{
2105 int ch;
2106
2107 /* If there is data waiting, eat it up */
2108 if (i->p && *i->p) {
2109#if ENABLE_HUSH_INTERACTIVE
2110 take_cached:
2111#endif
2112 ch = *i->p++;
2113 if (i->eof_flag && !*i->p)
2114 ch = EOF;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002115 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002116 } else {
2117 /* need to double check i->file because we might be doing something
2118 * more complicated by now, like sourcing or substituting. */
2119#if ENABLE_HUSH_INTERACTIVE
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002120 if (G_interactive_fd && i->file == stdin) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002121 do {
2122 get_user_input(i);
2123 } while (!*i->p); /* need non-empty line */
2124 i->promptmode = 1; /* PS2 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002125 goto take_cached;
2126 }
2127#endif
Denis Vlasenko913a2012009-04-05 22:17:04 +00002128 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002129 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002130 debug_printf("file_get: got '%c' %d\n", ch, ch);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02002131 i->last_char = ch;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002132 return ch;
2133}
2134
Denis Vlasenko913a2012009-04-05 22:17:04 +00002135/* All callers guarantee this routine will never
2136 * be used right after a newline, so prompting is not needed.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002137 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02002138static int FAST_FUNC file_peek(struct in_str *i)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002139{
2140 int ch;
2141 if (i->p && *i->p) {
2142 if (i->eof_flag && !i->p[1])
2143 return EOF;
2144 return *i->p;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002145 /* note: ch is never NUL */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002146 }
Denis Vlasenko913a2012009-04-05 22:17:04 +00002147 do ch = fgetc(i->file); while (ch == '\0');
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002148 i->eof_flag = (ch == EOF);
2149 i->peek_buf[0] = ch;
2150 i->peek_buf[1] = '\0';
2151 i->p = i->peek_buf;
Denis Vlasenko913a2012009-04-05 22:17:04 +00002152 debug_printf("file_peek: got '%c' %d\n", ch, ch);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002153 return ch;
2154}
2155
2156static void setup_file_in_str(struct in_str *i, FILE *f)
2157{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002158 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002159 i->peek = file_peek;
2160 i->get = file_get;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002161 /* i->promptmode = 0; - PS1 (memset did it) */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002162 i->file = f;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002163 /* i->p = NULL; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002164}
2165
2166static void setup_string_in_str(struct in_str *i, const char *s)
2167{
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002168 memset(i, 0, sizeof(*i));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002169 i->peek = static_peek;
2170 i->get = static_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->p = s;
Denys Vlasenkoa1463192011-01-18 17:55:04 +01002173 /* i->eof_flag = 0; */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002174}
2175
2176
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002177/*
2178 * o_string support
2179 */
2180#define B_CHUNK (32 * sizeof(char*))
Eric Andersen25f27032001-04-26 23:22:31 +00002181
Denis Vlasenko0b677d82009-04-10 13:49:10 +00002182static void o_reset_to_empty_unquoted(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002183{
2184 o->length = 0;
Denys Vlasenko38292b62010-09-05 14:49:40 +02002185 o->has_quoted_part = 0;
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002186 if (o->data)
2187 o->data[0] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002188}
2189
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002190static void o_free(o_string *o)
Eric Andersen25f27032001-04-26 23:22:31 +00002191{
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002192 free(o->data);
Denis Vlasenkod65ea392007-10-01 10:02:25 +00002193 memset(o, 0, sizeof(*o));
Eric Andersen25f27032001-04-26 23:22:31 +00002194}
2195
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002196static ALWAYS_INLINE void o_free_unsafe(o_string *o)
2197{
2198 free(o->data);
2199}
2200
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002201static void o_grow_by(o_string *o, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002202{
2203 if (o->length + len > o->maxlen) {
2204 o->maxlen += (2*len > B_CHUNK ? 2*len : B_CHUNK);
2205 o->data = xrealloc(o->data, 1 + o->maxlen);
2206 }
2207}
2208
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002209static void o_addchr(o_string *o, int ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002210{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002211 debug_printf("o_addchr: '%c' o->length=%d o=%p\n", ch, o->length, o);
2212 o_grow_by(o, 1);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002213 o->data[o->length] = ch;
2214 o->length++;
2215 o->data[o->length] = '\0';
2216}
2217
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002218static void o_addblock(o_string *o, const char *str, int len)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002219{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002220 o_grow_by(o, len);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002221 memcpy(&o->data[o->length], str, len);
2222 o->length += len;
2223 o->data[o->length] = '\0';
2224}
2225
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002226static void o_addstr(o_string *o, const char *str)
Mike Frysinger98c52642009-04-02 10:02:37 +00002227{
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002228 o_addblock(o, str, strlen(str));
2229}
Denys Vlasenko2e48d532010-05-22 17:30:39 +02002230
Denys Vlasenko1e811b12010-05-22 03:12:29 +02002231#if !BB_MMU
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00002232static void nommu_addchr(o_string *o, int ch)
2233{
2234 if (o)
2235 o_addchr(o, ch);
2236}
2237#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002238# define nommu_addchr(o, str) ((void)0)
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002239#endif
2240
2241static void o_addstr_with_NUL(o_string *o, const char *str)
2242{
2243 o_addblock(o, str, strlen(str) + 1);
Mike Frysinger98c52642009-04-02 10:02:37 +00002244}
2245
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002246/*
Denys Vlasenko238081f2010-10-03 14:26:26 +02002247 * HUSH_BRACE_EXPANSION code needs corresponding quoting on variable expansion side.
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002248 * Currently, "v='{q,w}'; echo $v" erroneously expands braces in $v.
2249 * Apparently, on unquoted $v bash still does globbing
2250 * ("v='*.txt'; echo $v" prints all .txt files),
2251 * but NOT brace expansion! Thus, there should be TWO independent
2252 * quoting mechanisms on $v expansion side: one protects
2253 * $v from brace expansion, and other additionally protects "$v" against globbing.
2254 * We have only second one.
2255 */
2256
Denys Vlasenko9e800222010-10-03 14:28:04 +02002257#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002258# define MAYBE_BRACES "{}"
2259#else
2260# define MAYBE_BRACES ""
2261#endif
2262
Eric Andersen25f27032001-04-26 23:22:31 +00002263/* My analysis of quoting semantics tells me that state information
2264 * is associated with a destination, not a source.
2265 */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002266static void o_addqchr(o_string *o, int ch)
Eric Andersen25f27032001-04-26 23:22:31 +00002267{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002268 int sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002269 char *found = strchr("*?[\\" MAYBE_BRACES, ch);
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002270 if (found)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002271 sz++;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00002272 o_grow_by(o, sz);
2273 if (found) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002274 o->data[o->length] = '\\';
2275 o->length++;
Eric Andersen25f27032001-04-26 23:22:31 +00002276 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002277 o->data[o->length] = ch;
2278 o->length++;
2279 o->data[o->length] = '\0';
Eric Andersen25f27032001-04-26 23:22:31 +00002280}
2281
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002282static void o_addQchr(o_string *o, int ch)
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002283{
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002284 int sz = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002285 if ((o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)
2286 && strchr("*?[\\" MAYBE_BRACES, ch)
2287 ) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002288 sz++;
2289 o->data[o->length] = '\\';
2290 o->length++;
2291 }
2292 o_grow_by(o, sz);
2293 o->data[o->length] = ch;
2294 o->length++;
2295 o->data[o->length] = '\0';
2296}
2297
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002298static void o_addqblock(o_string *o, const char *str, int len)
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002299{
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002300 while (len) {
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002301 char ch;
2302 int sz;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002303 int ordinary_cnt = strcspn(str, "*?[\\" MAYBE_BRACES);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002304 if (ordinary_cnt > len) /* paranoia */
2305 ordinary_cnt = len;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002306 o_addblock(o, str, ordinary_cnt);
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002307 if (ordinary_cnt == len)
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002308 return; /* NUL is already added by o_addblock */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002309 str += ordinary_cnt;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002310 len -= ordinary_cnt + 1; /* we are processing + 1 char below */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002311
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002312 ch = *str++;
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002313 sz = 1;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002314 if (ch) { /* it is necessarily one of "*?[\\" MAYBE_BRACES */
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002315 sz++;
2316 o->data[o->length] = '\\';
2317 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002318 }
Denis Vlasenko7e3d33b2008-06-12 13:31:04 +00002319 o_grow_by(o, sz);
2320 o->data[o->length] = ch;
2321 o->length++;
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002322 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002323 o->data[o->length] = '\0';
Denis Vlasenko87f40ba2008-06-10 22:39:37 +00002324}
2325
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002326static void o_addQblock(o_string *o, const char *str, int len)
2327{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002328 if (!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002329 o_addblock(o, str, len);
2330 return;
2331 }
2332 o_addqblock(o, str, len);
2333}
2334
Denys Vlasenko38292b62010-09-05 14:49:40 +02002335static void o_addQstr(o_string *o, const char *str)
2336{
2337 o_addQblock(o, str, strlen(str));
2338}
2339
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002340/* A special kind of o_string for $VAR and `cmd` expansion.
2341 * It contains char* list[] at the beginning, which is grown in 16 element
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002342 * increments. Actual string data starts at the next multiple of 16 * (char*).
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002343 * list[i] contains an INDEX (int!) into this string data.
2344 * It means that if list[] needs to grow, data needs to be moved higher up
2345 * but list[i]'s need not be modified.
2346 * NB: remembering how many list[i]'s you have there is crucial.
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002347 * o_finalize_list() operation post-processes this structure - calculates
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002348 * and stores actual char* ptrs in list[]. Oh, it NULL terminates it as well.
2349 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002350#if DEBUG_EXPAND || DEBUG_GLOB
2351static void debug_print_list(const char *prefix, o_string *o, int n)
2352{
2353 char **list = (char**)o->data;
2354 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2355 int i = 0;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002356
2357 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002358 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 +02002359 prefix, list, n, string_start, o->length, o->maxlen,
2360 !!(o->o_expflags & EXP_FLAG_GLOB),
2361 o->has_quoted_part,
2362 !!(o->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002363 while (i < n) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002364 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002365 fdprintf(2, " list[%d]=%d '%s' %p\n", i, (int)(uintptr_t)list[i],
2366 o->data + (int)(uintptr_t)list[i] + string_start,
2367 o->data + (int)(uintptr_t)list[i] + string_start);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002368 i++;
2369 }
2370 if (n) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002371 const char *p = o->data + (int)(uintptr_t)list[n - 1] + string_start;
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002372 indent();
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002373 fdprintf(2, " total_sz:%ld\n", (long)((p + strlen(p) + 1) - o->data));
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002374 }
2375}
2376#else
Denys Vlasenko28a105d2009-06-01 11:26:30 +02002377# define debug_print_list(prefix, o, n) ((void)0)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002378#endif
2379
2380/* n = o_save_ptr_helper(str, n) "starts new string" by storing an index value
2381 * in list[n] so that it points past last stored byte so far.
2382 * It returns n+1. */
2383static int o_save_ptr_helper(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002384{
2385 char **list = (char**)o->data;
Denis Vlasenko895bea22008-06-10 18:06:24 +00002386 int string_start;
2387 int string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002388
2389 if (!o->has_empty_slot) {
Denis Vlasenko895bea22008-06-10 18:06:24 +00002390 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2391 string_len = o->length - string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002392 if (!(n & 0xf)) { /* 0, 0x10, 0x20...? */
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002393 debug_printf_list("list[%d]=%d string_start=%d (growing)\n", n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002394 /* list[n] points to string_start, make space for 16 more pointers */
2395 o->maxlen += 0x10 * sizeof(list[0]);
2396 o->data = xrealloc(o->data, o->maxlen + 1);
Denis Vlasenko7049ff82008-06-25 09:53:17 +00002397 list = (char**)o->data;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002398 memmove(list + n + 0x10, list + n, string_len);
2399 o->length += 0x10 * sizeof(list[0]);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002400 } else {
2401 debug_printf_list("list[%d]=%d string_start=%d\n",
2402 n, string_len, string_start);
2403 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002404 } else {
2405 /* We have empty slot at list[n], reuse without growth */
Denis Vlasenko895bea22008-06-10 18:06:24 +00002406 string_start = ((n+1 + 0xf) & ~0xf) * sizeof(list[0]); /* NB: n+1! */
2407 string_len = o->length - string_start;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00002408 debug_printf_list("list[%d]=%d string_start=%d (empty slot)\n",
2409 n, string_len, string_start);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002410 o->has_empty_slot = 0;
2411 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02002412 o->has_quoted_part = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002413 list[n] = (char*)(uintptr_t)string_len;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002414 return n + 1;
2415}
2416
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002417/* "What was our last o_save_ptr'ed position (byte offset relative o->data)?" */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002418static int o_get_last_ptr(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002419{
2420 char **list = (char**)o->data;
2421 int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2422
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002423 return ((int)(uintptr_t)list[n-1]) + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002424}
2425
Denys Vlasenko9e800222010-10-03 14:28:04 +02002426#if ENABLE_HUSH_BRACE_EXPANSION
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002427/* There in a GNU extension, GLOB_BRACE, but it is not usable:
2428 * first, it processes even {a} (no commas), second,
2429 * I didn't manage to make it return strings when they don't match
Denys Vlasenko160746b2009-11-16 05:51:18 +01002430 * existing files. Need to re-implement it.
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002431 */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002432
2433/* Helper */
2434static int glob_needed(const char *s)
2435{
2436 while (*s) {
2437 if (*s == '\\') {
2438 if (!s[1])
2439 return 0;
2440 s += 2;
2441 continue;
2442 }
2443 if (*s == '*' || *s == '[' || *s == '?' || *s == '{')
2444 return 1;
2445 s++;
2446 }
2447 return 0;
2448}
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002449/* Return pointer to next closing brace or to comma */
2450static const char *next_brace_sub(const char *cp)
2451{
2452 unsigned depth = 0;
2453 cp++;
2454 while (*cp != '\0') {
2455 if (*cp == '\\') {
2456 if (*++cp == '\0')
2457 break;
2458 cp++;
2459 continue;
Denys Vlasenko3581c622010-01-25 13:39:24 +01002460 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002461 if ((*cp == '}' && depth-- == 0) || (*cp == ',' && depth == 0))
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002462 break;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002463 if (*cp++ == '{')
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002464 depth++;
2465 }
2466
2467 return *cp != '\0' ? cp : NULL;
2468}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002469/* Recursive brace globber. Note: may garble pattern[]. */
2470static int glob_brace(char *pattern, o_string *o, int n)
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002471{
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002472 char *new_pattern_buf;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002473 const char *begin;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002474 const char *next;
2475 const char *rest;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002476 const char *p;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002477 size_t rest_len;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002478
2479 debug_printf_glob("glob_brace('%s')\n", pattern);
2480
2481 begin = pattern;
2482 while (1) {
2483 if (*begin == '\0')
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002484 goto simple_glob;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002485 if (*begin == '{') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002486 /* Find the first sub-pattern and at the same time
2487 * find the rest after the closing brace */
2488 next = next_brace_sub(begin);
2489 if (next == NULL) {
2490 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002491 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002492 }
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002493 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002494 /* "{abc}" with no commas - illegal
2495 * brace expr, disregard and skip it */
2496 begin = next + 1;
2497 continue;
2498 }
2499 break;
2500 }
2501 if (*begin == '\\' && begin[1] != '\0')
2502 begin++;
2503 begin++;
2504 }
2505 debug_printf_glob("begin:%s\n", begin);
2506 debug_printf_glob("next:%s\n", next);
2507
2508 /* Now find the end of the whole brace expression */
2509 rest = next;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002510 while (*rest != '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002511 rest = next_brace_sub(rest);
2512 if (rest == NULL) {
2513 /* An illegal expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002514 goto simple_glob;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002515 }
2516 debug_printf_glob("rest:%s\n", rest);
2517 }
2518 rest_len = strlen(++rest) + 1;
2519
2520 /* We are sure the brace expression is well-formed */
2521
2522 /* Allocate working buffer large enough for our work */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002523 new_pattern_buf = xmalloc(strlen(pattern));
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002524
2525 /* We have a brace expression. BEGIN points to the opening {,
2526 * NEXT points past the terminator of the first element, and REST
2527 * points past the final }. We will accumulate result names from
2528 * recursive runs for each brace alternative in the buffer using
2529 * GLOB_APPEND. */
2530
2531 p = begin + 1;
2532 while (1) {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002533 /* Construct the new glob expression */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002534 memcpy(
2535 mempcpy(
2536 mempcpy(new_pattern_buf,
2537 /* We know the prefix for all sub-patterns */
2538 pattern, begin - pattern),
2539 p, next - p),
2540 rest, rest_len);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002541
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002542 /* Note: glob_brace() may garble new_pattern_buf[].
2543 * That's why we re-copy prefix every time (1st memcpy above).
2544 */
2545 n = glob_brace(new_pattern_buf, o, n);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02002546 if (*next == '}') {
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002547 /* We saw the last entry */
2548 break;
2549 }
2550 p = next + 1;
2551 next = next_brace_sub(next);
2552 }
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002553 free(new_pattern_buf);
2554 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002555
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002556 simple_glob:
2557 {
2558 int gr;
2559 glob_t globdata;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002560
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002561 memset(&globdata, 0, sizeof(globdata));
2562 gr = glob(pattern, 0, NULL, &globdata);
2563 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
2564 if (gr != 0) {
2565 if (gr == GLOB_NOMATCH) {
2566 globfree(&globdata);
2567 /* NB: garbles parameter */
2568 unbackslash(pattern);
2569 o_addstr_with_NUL(o, pattern);
2570 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2571 return o_save_ptr_helper(o, n);
2572 }
2573 if (gr == GLOB_NOSPACE)
2574 bb_error_msg_and_die(bb_msg_memory_exhausted);
2575 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2576 * but we didn't specify it. Paranoia again. */
2577 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
2578 }
2579 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2580 char **argv = globdata.gl_pathv;
2581 while (1) {
2582 o_addstr_with_NUL(o, *argv);
2583 n = o_save_ptr_helper(o, n);
2584 argv++;
2585 if (!*argv)
2586 break;
2587 }
2588 }
2589 globfree(&globdata);
2590 }
2591 return n;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002592}
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002593/* Performs globbing on last list[],
2594 * saving each result as a new list[].
2595 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002596static int perform_glob(o_string *o, int n)
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002597{
2598 char *pattern, *copy;
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002599
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002600 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002601 if (!o->data)
2602 return o_save_ptr_helper(o, n);
2603 pattern = o->data + o_get_last_ptr(o, n);
2604 debug_printf_glob("glob pattern '%s'\n", pattern);
2605 if (!glob_needed(pattern)) {
2606 /* unbackslash last string in o in place, fix length */
2607 o->length = unbackslash(pattern) - o->data;
2608 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
2609 return o_save_ptr_helper(o, n);
2610 }
2611
2612 copy = xstrdup(pattern);
2613 /* "forget" pattern in o */
2614 o->length = pattern - o->data;
2615 n = glob_brace(copy, o, n);
2616 free(copy);
2617 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002618 debug_print_list("perform_glob returning", o, n);
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002619 return n;
2620}
2621
Denys Vlasenko238081f2010-10-03 14:26:26 +02002622#else /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002623
2624/* Helper */
2625static int glob_needed(const char *s)
2626{
2627 while (*s) {
2628 if (*s == '\\') {
2629 if (!s[1])
2630 return 0;
2631 s += 2;
2632 continue;
2633 }
2634 if (*s == '*' || *s == '[' || *s == '?')
2635 return 1;
2636 s++;
2637 }
2638 return 0;
2639}
2640/* Performs globbing on last list[],
2641 * saving each result as a new list[].
2642 */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002643static int perform_glob(o_string *o, int n)
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002644{
2645 glob_t globdata;
2646 int gr;
2647 char *pattern;
2648
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002649 debug_printf_glob("start perform_glob: n:%d o->data:%p\n", n, o->data);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002650 if (!o->data)
2651 return o_save_ptr_helper(o, n);
2652 pattern = o->data + o_get_last_ptr(o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002653 debug_printf_glob("glob pattern '%s'\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002654 if (!glob_needed(pattern)) {
2655 literal:
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002656 /* unbackslash last string in o in place, fix length */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002657 o->length = unbackslash(pattern) - o->data;
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002658 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002659 return o_save_ptr_helper(o, n);
2660 }
2661
2662 memset(&globdata, 0, sizeof(globdata));
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002663 /* Can't use GLOB_NOCHECK: it does not unescape the string.
2664 * If we glob "*.\*" and don't find anything, we need
2665 * to fall back to using literal "*.*", but GLOB_NOCHECK
2666 * will return "*.\*"!
2667 */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002668 gr = glob(pattern, 0, NULL, &globdata);
2669 debug_printf_glob("glob('%s'):%d\n", pattern, gr);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002670 if (gr != 0) {
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002671 if (gr == GLOB_NOMATCH) {
2672 globfree(&globdata);
2673 goto literal;
2674 }
2675 if (gr == GLOB_NOSPACE)
2676 bb_error_msg_and_die(bb_msg_memory_exhausted);
Denys Vlasenko5b2db972009-11-16 05:49:36 +01002677 /* GLOB_ABORTED? Only happens with GLOB_ERR flag,
2678 * but we didn't specify it. Paranoia again. */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002679 bb_error_msg_and_die("glob error %d on '%s'", gr, pattern);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002680 }
2681 if (globdata.gl_pathv && globdata.gl_pathv[0]) {
2682 char **argv = globdata.gl_pathv;
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002683 /* "forget" pattern in o */
2684 o->length = pattern - o->data;
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002685 while (1) {
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002686 o_addstr_with_NUL(o, *argv);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002687 n = o_save_ptr_helper(o, n);
2688 argv++;
2689 if (!*argv)
2690 break;
2691 }
2692 }
2693 globfree(&globdata);
2694 if (DEBUG_GLOB)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002695 debug_print_list("perform_glob returning", o, n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002696 return n;
2697}
2698
Denys Vlasenko238081f2010-10-03 14:26:26 +02002699#endif /* !HUSH_BRACE_EXPANSION */
Denys Vlasenkof3e28182009-11-17 03:35:31 +01002700
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002701/* If o->o_expflags & EXP_FLAG_GLOB, glob the string so far remembered.
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002702 * Otherwise, just finish current list[] and start new */
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002703static int o_save_ptr(o_string *o, int n)
2704{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02002705 if (o->o_expflags & EXP_FLAG_GLOB) {
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002706 /* If o->has_empty_slot, list[n] was already globbed
2707 * (if it was requested back then when it was filled)
2708 * so don't do that again! */
2709 if (!o->has_empty_slot)
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02002710 return perform_glob(o, n); /* o_save_ptr_helper is inside */
Denis Vlasenkoa8b6dff2009-03-20 12:05:14 +00002711 }
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002712 return o_save_ptr_helper(o, n);
2713}
2714
2715/* "Please convert list[n] to real char* ptrs, and NULL terminate it." */
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00002716static char **o_finalize_list(o_string *o, int n)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002717{
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002718 char **list;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002719 int string_start;
2720
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002721 n = o_save_ptr(o, n); /* force growth for list[n] if necessary */
2722 if (DEBUG_EXPAND)
2723 debug_print_list("finalized", o, n);
Denis Vlasenko30c9cc52008-06-17 07:24:29 +00002724 debug_printf_expand("finalized n:%d\n", n);
Denis Vlasenkob61e13d2008-06-17 05:11:43 +00002725 list = (char**)o->data;
2726 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
2727 list[--n] = NULL;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002728 while (n) {
2729 n--;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002730 list[n] = o->data + (int)(uintptr_t)list[n] + string_start;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00002731 }
2732 return list;
2733}
2734
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002735static void free_pipe_list(struct pipe *pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002736
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002737/* Returns pi->next - next pipe in the list */
2738static struct pipe *free_pipe(struct pipe *pi)
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002739{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002740 struct pipe *next;
2741 int i;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002742
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002743 debug_printf_clean("free_pipe (pid %d)\n", getpid());
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002744 for (i = 0; i < pi->num_cmds; i++) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002745 struct command *command;
2746 struct redir_struct *r, *rnext;
2747
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002748 command = &pi->cmds[i];
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002749 debug_printf_clean(" command %d:\n", i);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002750 if (command->argv) {
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002751 if (DEBUG_CLEAN) {
2752 int a;
2753 char **p;
2754 for (a = 0, p = command->argv; *p; a++, p++) {
2755 debug_printf_clean(" argv[%d] = %s\n", a, *p);
2756 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002757 }
2758 free_strings(command->argv);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002759 //command->argv = NULL;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00002760 }
2761 /* not "else if": on syntax error, we may have both! */
2762 if (command->group) {
Denys Vlasenko9d617c42009-06-09 18:40:52 +02002763 debug_printf_clean(" begin group (cmd_type:%d)\n",
2764 command->cmd_type);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002765 free_pipe_list(command->group);
2766 debug_printf_clean(" end group\n");
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002767 //command->group = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002768 }
Denis Vlasenkoed055212009-04-11 10:37:10 +00002769 /* else is crucial here.
2770 * If group != NULL, child_func is meaningless */
2771#if ENABLE_HUSH_FUNCTIONS
2772 else if (command->child_func) {
2773 debug_printf_exec("cmd %p releases child func at %p\n", command, command->child_func);
2774 command->child_func->parent_cmd = NULL;
2775 }
2776#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002777#if !BB_MMU
2778 free(command->group_as_string);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002779 //command->group_as_string = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00002780#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002781 for (r = command->redirects; r; r = rnext) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002782 debug_printf_clean(" redirect %d%s",
2783 r->rd_fd, redir_table[r->rd_type].descrip);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002784 /* guard against the case >$FOO, where foo is unset or blank */
2785 if (r->rd_filename) {
2786 debug_printf_clean(" fname:'%s'\n", r->rd_filename);
2787 free(r->rd_filename);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002788 //r->rd_filename = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002789 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00002790 debug_printf_clean(" rd_dup:%d\n", r->rd_dup);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002791 rnext = r->next;
2792 free(r);
2793 }
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002794 //command->redirects = NULL;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002795 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002796 free(pi->cmds); /* children are an array, they get freed all at once */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002797 //pi->cmds = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002798#if ENABLE_HUSH_JOB
2799 free(pi->cmdtext);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002800 //pi->cmdtext = NULL;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002801#endif
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002802
2803 next = pi->next;
2804 free(pi);
2805 return next;
Denis Vlasenkof886fd22008-10-13 12:36:05 +00002806}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002807
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002808static void free_pipe_list(struct pipe *pi)
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002809{
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002810 while (pi) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002811#if HAS_KEYWORDS
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002812 debug_printf_clean("pipe reserved word %d\n", pi->res_word);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002813#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00002814 debug_printf_clean("pipe followup code %d\n", pi->followup);
Denys Vlasenko27c56f12010-09-07 09:56:34 +02002815 pi = free_pipe(pi);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002816 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002817}
2818
2819
Denys Vlasenkob36abf22010-09-05 14:50:59 +02002820/*** Parsing routines ***/
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00002821
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002822#ifndef debug_print_tree
2823static void debug_print_tree(struct pipe *pi, int lvl)
2824{
2825 static const char *const PIPE[] = {
2826 [PIPE_SEQ] = "SEQ",
2827 [PIPE_AND] = "AND",
2828 [PIPE_OR ] = "OR" ,
2829 [PIPE_BG ] = "BG" ,
2830 };
2831 static const char *RES[] = {
2832 [RES_NONE ] = "NONE" ,
2833# if ENABLE_HUSH_IF
2834 [RES_IF ] = "IF" ,
2835 [RES_THEN ] = "THEN" ,
2836 [RES_ELIF ] = "ELIF" ,
2837 [RES_ELSE ] = "ELSE" ,
2838 [RES_FI ] = "FI" ,
2839# endif
2840# if ENABLE_HUSH_LOOPS
2841 [RES_FOR ] = "FOR" ,
2842 [RES_WHILE] = "WHILE",
2843 [RES_UNTIL] = "UNTIL",
2844 [RES_DO ] = "DO" ,
2845 [RES_DONE ] = "DONE" ,
2846# endif
2847# if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
2848 [RES_IN ] = "IN" ,
2849# endif
2850# if ENABLE_HUSH_CASE
2851 [RES_CASE ] = "CASE" ,
2852 [RES_CASE_IN ] = "CASE_IN" ,
2853 [RES_MATCH] = "MATCH",
2854 [RES_CASE_BODY] = "CASE_BODY",
2855 [RES_ESAC ] = "ESAC" ,
2856# endif
2857 [RES_XXXX ] = "XXXX" ,
2858 [RES_SNTX ] = "SNTX" ,
2859 };
2860 static const char *const CMDTYPE[] = {
2861 "{}",
2862 "()",
2863 "[noglob]",
2864# if ENABLE_HUSH_FUNCTIONS
2865 "func()",
2866# endif
2867 };
2868
2869 int pin, prn;
2870
2871 pin = 0;
2872 while (pi) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002873 fdprintf(2, "%*spipe %d res_word=%s followup=%d %s\n", lvl*2, "",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002874 pin, RES[pi->res_word], pi->followup, PIPE[pi->followup]);
2875 prn = 0;
2876 while (prn < pi->num_cmds) {
2877 struct command *command = &pi->cmds[prn];
2878 char **argv = command->argv;
2879
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002880 fdprintf(2, "%*s cmd %d assignment_cnt:%d",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002881 lvl*2, "", prn,
2882 command->assignment_cnt);
2883 if (command->group) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002884 fdprintf(2, " group %s: (argv=%p)%s%s\n",
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002885 CMDTYPE[command->cmd_type],
2886 argv
2887# if !BB_MMU
2888 , " group_as_string:", command->group_as_string
2889# else
2890 , "", ""
2891# endif
2892 );
2893 debug_print_tree(command->group, lvl+1);
2894 prn++;
2895 continue;
2896 }
2897 if (argv) while (*argv) {
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002898 fdprintf(2, " '%s'", *argv);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002899 argv++;
2900 }
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01002901 fdprintf(2, "\n");
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01002902 prn++;
2903 }
2904 pi = pi->next;
2905 pin++;
2906 }
2907}
2908#endif /* debug_print_tree */
2909
Denis Vlasenkoac678ec2007-04-16 22:32:04 +00002910static struct pipe *new_pipe(void)
2911{
Eric Andersen25f27032001-04-26 23:22:31 +00002912 struct pipe *pi;
Denis Vlasenko3ac0e002007-04-28 16:45:22 +00002913 pi = xzalloc(sizeof(struct pipe));
Denis Vlasenkoa8442002008-06-14 11:00:17 +00002914 /*pi->followup = 0; - deliberately invalid value */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00002915 /*pi->res_word = RES_NONE; - RES_NONE is 0 anyway */
Eric Andersen25f27032001-04-26 23:22:31 +00002916 return pi;
2917}
2918
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002919/* Command (member of a pipe) is complete, or we start a new pipe
2920 * if ctx->command is NULL.
2921 * No errors possible here.
2922 */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002923static int done_command(struct parse_context *ctx)
2924{
2925 /* The command is really already in the pipe structure, so
2926 * advance the pipe counter and make a new, null command. */
2927 struct pipe *pi = ctx->pipe;
2928 struct command *command = ctx->command;
2929
2930 if (command) {
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002931 if (IS_NULL_CMD(command)) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002932 debug_printf_parse("done_command: skipping null cmd, num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002933 goto clear_and_ret;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002934 }
2935 pi->num_cmds++;
2936 debug_printf_parse("done_command: ++num_cmds=%d\n", pi->num_cmds);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002937 //debug_print_tree(ctx->list_head, 20);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002938 } else {
2939 debug_printf_parse("done_command: initializing, num_cmds=%d\n", pi->num_cmds);
2940 }
2941
2942 /* Only real trickiness here is that the uncommitted
2943 * command structure is not counted in pi->num_cmds. */
2944 pi->cmds = xrealloc(pi->cmds, sizeof(*pi->cmds) * (pi->num_cmds+1));
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002945 ctx->command = command = &pi->cmds[pi->num_cmds];
2946 clear_and_ret:
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002947 memset(command, 0, sizeof(*command));
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002948 return pi->num_cmds; /* used only for 0/nonzero check */
2949}
2950
2951static void done_pipe(struct parse_context *ctx, pipe_style type)
2952{
2953 int not_null;
2954
2955 debug_printf_parse("done_pipe entered, followup %d\n", type);
2956 /* Close previous command */
2957 not_null = done_command(ctx);
2958 ctx->pipe->followup = type;
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002959#if HAS_KEYWORDS
2960 ctx->pipe->pi_inverted = ctx->ctx_inverted;
2961 ctx->ctx_inverted = 0;
2962 ctx->pipe->res_word = ctx->ctx_res_w;
2963#endif
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002964
2965 /* Without this check, even just <enter> on command line generates
2966 * tree of three NOPs (!). Which is harmless but annoying.
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002967 * IOW: it is safe to do it unconditionally. */
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002968 if (not_null
Denis Vlasenko7f959372009-04-14 08:06:59 +00002969#if ENABLE_HUSH_IF
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002970 || ctx->ctx_res_w == RES_FI
Denis Vlasenko7f959372009-04-14 08:06:59 +00002971#endif
2972#if ENABLE_HUSH_LOOPS
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002973 || ctx->ctx_res_w == RES_DONE
2974 || ctx->ctx_res_w == RES_FOR
2975 || ctx->ctx_res_w == RES_IN
Denis Vlasenko7f959372009-04-14 08:06:59 +00002976#endif
2977#if ENABLE_HUSH_CASE
Denis Vlasenkocd418a22009-04-06 18:08:35 +00002978 || ctx->ctx_res_w == RES_ESAC
2979#endif
2980 ) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002981 struct pipe *new_p;
2982 debug_printf_parse("done_pipe: adding new pipe: "
2983 "not_null:%d ctx->ctx_res_w:%d\n",
2984 not_null, ctx->ctx_res_w);
2985 new_p = new_pipe();
2986 ctx->pipe->next = new_p;
2987 ctx->pipe = new_p;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002988 /* RES_THEN, RES_DO etc are "sticky" -
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00002989 * they remain set for pipes inside if/while.
Denis Vlasenko424f79b2009-03-22 14:23:34 +00002990 * This is used to control execution.
2991 * RES_FOR and RES_IN are NOT sticky (needed to support
2992 * cases where variable or value happens to match a keyword):
2993 */
2994#if ENABLE_HUSH_LOOPS
2995 if (ctx->ctx_res_w == RES_FOR
2996 || ctx->ctx_res_w == RES_IN)
2997 ctx->ctx_res_w = RES_NONE;
2998#endif
2999#if ENABLE_HUSH_CASE
3000 if (ctx->ctx_res_w == RES_MATCH)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003001 ctx->ctx_res_w = RES_CASE_BODY;
3002 if (ctx->ctx_res_w == RES_CASE)
3003 ctx->ctx_res_w = RES_CASE_IN;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003004#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003005 ctx->command = NULL; /* trick done_command below */
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003006 /* Create the memory for command, roughly:
3007 * ctx->pipe->cmds = new struct command;
3008 * ctx->command = &ctx->pipe->cmds[0];
3009 */
3010 done_command(ctx);
Denis Vlasenkocd418a22009-04-06 18:08:35 +00003011 //debug_print_tree(ctx->list_head, 10);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00003012 }
3013 debug_printf_parse("done_pipe return\n");
3014}
3015
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003016static void initialize_context(struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003017{
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003018 memset(ctx, 0, sizeof(*ctx));
Denis Vlasenko1a735862007-05-23 00:32:25 +00003019 ctx->pipe = ctx->list_head = new_pipe();
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003020 /* Create the memory for command, roughly:
3021 * ctx->pipe->cmds = new struct command;
3022 * ctx->command = &ctx->pipe->cmds[0];
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003023 */
3024 done_command(ctx);
Eric Andersen25f27032001-04-26 23:22:31 +00003025}
3026
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003027/* If a reserved word is found and processed, parse context is modified
3028 * and 1 is returned.
Eric Andersen25f27032001-04-26 23:22:31 +00003029 */
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003030#if HAS_KEYWORDS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003031struct reserved_combo {
3032 char literal[6];
3033 unsigned char res;
3034 unsigned char assignment_flag;
3035 int flag;
3036};
3037enum {
3038 FLAG_END = (1 << RES_NONE ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003039# if ENABLE_HUSH_IF
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003040 FLAG_IF = (1 << RES_IF ),
3041 FLAG_THEN = (1 << RES_THEN ),
3042 FLAG_ELIF = (1 << RES_ELIF ),
3043 FLAG_ELSE = (1 << RES_ELSE ),
3044 FLAG_FI = (1 << RES_FI ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003045# endif
3046# if ENABLE_HUSH_LOOPS
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003047 FLAG_FOR = (1 << RES_FOR ),
3048 FLAG_WHILE = (1 << RES_WHILE),
3049 FLAG_UNTIL = (1 << RES_UNTIL),
3050 FLAG_DO = (1 << RES_DO ),
3051 FLAG_DONE = (1 << RES_DONE ),
3052 FLAG_IN = (1 << RES_IN ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003053# endif
3054# if ENABLE_HUSH_CASE
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003055 FLAG_MATCH = (1 << RES_MATCH),
3056 FLAG_ESAC = (1 << RES_ESAC ),
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003057# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003058 FLAG_START = (1 << RES_XXXX ),
3059};
3060
3061static const struct reserved_combo* match_reserved_word(o_string *word)
3062{
Eric Andersen25f27032001-04-26 23:22:31 +00003063 /* Mostly a list of accepted follow-up reserved words.
3064 * FLAG_END means we are done with the sequence, and are ready
3065 * to turn the compound list into a command.
3066 * FLAG_START means the word must start a new compound list.
3067 */
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003068 static const struct reserved_combo reserved_list[] = {
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003069# if ENABLE_HUSH_IF
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003070 { "!", RES_NONE, NOT_ASSIGNMENT , 0 },
3071 { "if", RES_IF, MAYBE_ASSIGNMENT, FLAG_THEN | FLAG_START },
3072 { "then", RES_THEN, MAYBE_ASSIGNMENT, FLAG_ELIF | FLAG_ELSE | FLAG_FI },
3073 { "elif", RES_ELIF, MAYBE_ASSIGNMENT, FLAG_THEN },
3074 { "else", RES_ELSE, MAYBE_ASSIGNMENT, FLAG_FI },
3075 { "fi", RES_FI, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003076# endif
3077# if ENABLE_HUSH_LOOPS
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003078 { "for", RES_FOR, NOT_ASSIGNMENT , FLAG_IN | FLAG_DO | FLAG_START },
3079 { "while", RES_WHILE, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3080 { "until", RES_UNTIL, MAYBE_ASSIGNMENT, FLAG_DO | FLAG_START },
3081 { "in", RES_IN, NOT_ASSIGNMENT , FLAG_DO },
3082 { "do", RES_DO, MAYBE_ASSIGNMENT, FLAG_DONE },
3083 { "done", RES_DONE, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003084# endif
3085# if ENABLE_HUSH_CASE
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003086 { "case", RES_CASE, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_START },
3087 { "esac", RES_ESAC, NOT_ASSIGNMENT , FLAG_END },
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003088# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003089 };
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003090 const struct reserved_combo *r;
3091
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02003092 for (r = reserved_list; r < reserved_list + ARRAY_SIZE(reserved_list); r++) {
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003093 if (strcmp(word->data, r->literal) == 0)
3094 return r;
3095 }
3096 return NULL;
3097}
Denis Vlasenkobb929512009-04-16 10:59:40 +00003098/* Return 0: not a keyword, 1: keyword
3099 */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003100static int reserved_word(o_string *word, struct parse_context *ctx)
3101{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003102# if ENABLE_HUSH_CASE
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003103 static const struct reserved_combo reserved_match = {
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003104 "", RES_MATCH, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_ESAC
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003105 };
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003106# endif
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003107 const struct reserved_combo *r;
Denis Vlasenkoc72c1ed2007-01-30 22:31:26 +00003108
Denys Vlasenko38292b62010-09-05 14:49:40 +02003109 if (word->has_quoted_part)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003110 return 0;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003111 r = match_reserved_word(word);
3112 if (!r)
3113 return 0;
3114
3115 debug_printf("found reserved word %s, res %d\n", r->literal, r->res);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003116# if ENABLE_HUSH_CASE
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003117 if (r->res == RES_IN && ctx->ctx_res_w == RES_CASE_IN) {
3118 /* "case word IN ..." - IN part starts first MATCH part */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003119 r = &reserved_match;
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003120 } else
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003121# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003122 if (r->flag == 0) { /* '!' */
3123 if (ctx->ctx_inverted) { /* bash doesn't accept '! ! true' */
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003124 syntax_error("! ! command");
Denis Vlasenkobb929512009-04-16 10:59:40 +00003125 ctx->ctx_res_w = RES_SNTX;
Eric Andersen25f27032001-04-26 23:22:31 +00003126 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003127 ctx->ctx_inverted = 1;
Denis Vlasenko1a735862007-05-23 00:32:25 +00003128 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003129 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003130 if (r->flag & FLAG_START) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003131 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003132
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003133 old = xmalloc(sizeof(*old));
3134 debug_printf_parse("push stack %p\n", old);
3135 *old = *ctx; /* physical copy */
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003136 initialize_context(ctx);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003137 ctx->stack = old;
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003138 } else if (/*ctx->ctx_res_w == RES_NONE ||*/ !(ctx->old_flag & (1 << r->res))) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003139 syntax_error_at(word->data);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003140 ctx->ctx_res_w = RES_SNTX;
3141 return 1;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003142 } else {
3143 /* "{...} fi" is ok. "{...} if" is not
3144 * Example:
3145 * if { echo foo; } then { echo bar; } fi */
3146 if (ctx->command->group)
3147 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003148 }
Denis Vlasenkobb929512009-04-16 10:59:40 +00003149
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003150 ctx->ctx_res_w = r->res;
3151 ctx->old_flag = r->flag;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003152 word->o_assignment = r->assignment_flag;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003153 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
Denis Vlasenkobb929512009-04-16 10:59:40 +00003154
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003155 if (ctx->old_flag & FLAG_END) {
3156 struct parse_context *old;
Denis Vlasenkobb929512009-04-16 10:59:40 +00003157
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003158 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003159 debug_printf_parse("pop stack %p\n", ctx->stack);
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003160 old = ctx->stack;
3161 old->command->group = ctx->list_head;
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003162 old->command->cmd_type = CMD_NORMAL;
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003163# if !BB_MMU
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003164 o_addstr(&old->as_string, ctx->as_string.data);
3165 o_free_unsafe(&ctx->as_string);
3166 old->command->group_as_string = xstrdup(old->as_string.data);
3167 debug_printf_parse("pop, remembering as:'%s'\n",
3168 old->command->group_as_string);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003169# endif
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003170 *ctx = *old; /* physical copy */
3171 free(old);
3172 }
Denis Vlasenkoc3735272008-10-09 12:58:26 +00003173 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003174}
Denys Vlasenkoc0836532009-10-19 13:13:06 +02003175#endif /* HAS_KEYWORDS */
Eric Andersen25f27032001-04-26 23:22:31 +00003176
Denis Vlasenkoa8442002008-06-14 11:00:17 +00003177/* Word is complete, look at it and update parsing context.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003178 * Normal return is 0. Syntax errors return 1.
3179 * Note: on return, word is reset, but not o_free'd!
3180 */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003181static int done_word(o_string *word, struct parse_context *ctx)
Eric Andersen25f27032001-04-26 23:22:31 +00003182{
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003183 struct command *command = ctx->command;
Eric Andersen25f27032001-04-26 23:22:31 +00003184
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003185 debug_printf_parse("done_word entered: '%s' %p\n", word->data, command);
Denys Vlasenko38292b62010-09-05 14:49:40 +02003186 if (word->length == 0 && !word->has_quoted_part) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003187 debug_printf_parse("done_word return 0: true null, ignored\n");
3188 return 0;
Eric Andersen25f27032001-04-26 23:22:31 +00003189 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00003190
Eric Andersen25f27032001-04-26 23:22:31 +00003191 if (ctx->pending_redirect) {
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003192 /* We do not glob in e.g. >*.tmp case. bash seems to glob here
3193 * only if run as "bash", not "sh" */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003194 /* http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3195 * "2.7 Redirection
3196 * ...the word that follows the redirection operator
3197 * shall be subjected to tilde expansion, parameter expansion,
3198 * command substitution, arithmetic expansion, and quote
3199 * removal. Pathname expansion shall not be performed
3200 * on the word by a non-interactive shell; an interactive
3201 * shell may perform it, but shall do so only when
3202 * the expansion would result in one word."
3203 */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003204 ctx->pending_redirect->rd_filename = xstrdup(word->data);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003205 /* Cater for >\file case:
3206 * >\a creates file a; >\\a, >"\a", >"\\a" create file \a
3207 * Same with heredocs:
3208 * for <<\H delim is H; <<\\H, <<"\H", <<"\\H" - \H
3209 */
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003210 if (ctx->pending_redirect->rd_type == REDIRECT_HEREDOC) {
3211 unbackslash(ctx->pending_redirect->rd_filename);
3212 /* Is it <<"HEREDOC"? */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003213 if (word->has_quoted_part) {
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02003214 ctx->pending_redirect->rd_dup |= HEREDOC_QUOTED;
3215 }
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003216 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003217 debug_printf_parse("word stored in rd_filename: '%s'\n", word->data);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003218 ctx->pending_redirect = NULL;
Eric Andersen25f27032001-04-26 23:22:31 +00003219 } else {
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003220#if HAS_KEYWORDS
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003221# if ENABLE_HUSH_CASE
Denis Vlasenko757361f2008-07-14 08:26:47 +00003222 if (ctx->ctx_dsemicolon
3223 && strcmp(word->data, "esac") != 0 /* not "... pattern) cmd;; esac" */
3224 ) {
Denis Vlasenko395ae452008-07-14 06:29:38 +00003225 /* already done when ctx_dsemicolon was set to 1: */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003226 /* ctx->ctx_res_w = RES_MATCH; */
3227 ctx->ctx_dsemicolon = 0;
3228 } else
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003229# endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003230 if (!command->argv /* if it's the first word... */
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003231# if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003232 && ctx->ctx_res_w != RES_FOR /* ...not after FOR or IN */
3233 && ctx->ctx_res_w != RES_IN
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003234# endif
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02003235# if ENABLE_HUSH_CASE
3236 && ctx->ctx_res_w != RES_CASE
3237# endif
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003238 ) {
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003239 int reserved = reserved_word(word, ctx);
3240 debug_printf_parse("checking for reserved-ness: %d\n", reserved);
3241 if (reserved) {
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003242 o_reset_to_empty_unquoted(word);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003243 debug_printf_parse("done_word return %d\n",
3244 (ctx->ctx_res_w == RES_SNTX));
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003245 return (ctx->ctx_res_w == RES_SNTX);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003246 }
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003247# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003248 if (strcmp(word->data, "[[") == 0) {
3249 command->cmd_type = CMD_SINGLEWORD_NOGLOB;
3250 }
3251 /* fall through */
Denys Vlasenko9ca656b2009-06-10 13:39:35 +02003252# endif
Eric Andersen25f27032001-04-26 23:22:31 +00003253 }
Denis Vlasenko5ec61322008-06-24 00:50:07 +00003254#endif
Denis Vlasenkobb929512009-04-16 10:59:40 +00003255 if (command->group) {
3256 /* "{ echo foo; } echo bar" - bad */
3257 syntax_error_at(word->data);
3258 debug_printf_parse("done_word return 1: syntax error, "
3259 "groups and arglists don't mix\n");
3260 return 1;
3261 }
Denys Vlasenko29f9b722011-05-14 11:27:36 +02003262
3263 /* If this word wasn't an assignment, next ones definitely
3264 * can't be assignments. Even if they look like ones. */
3265 if (word->o_assignment != DEFINITELY_ASSIGNMENT
3266 && word->o_assignment != WORD_IS_KEYWORD
3267 ) {
3268 word->o_assignment = NOT_ASSIGNMENT;
3269 } else {
3270 if (word->o_assignment == DEFINITELY_ASSIGNMENT) {
3271 command->assignment_cnt++;
3272 debug_printf_parse("++assignment_cnt=%d\n", command->assignment_cnt);
3273 }
3274 debug_printf_parse("word->o_assignment was:'%s'\n", assignment_flag[word->o_assignment]);
3275 word->o_assignment = MAYBE_ASSIGNMENT;
3276 }
3277 debug_printf_parse("word->o_assignment='%s'\n", assignment_flag[word->o_assignment]);
3278
Denys Vlasenko38292b62010-09-05 14:49:40 +02003279 if (word->has_quoted_part
Denis Vlasenko55789c62008-06-18 16:30:42 +00003280 /* optimization: and if it's ("" or '') or ($v... or `cmd`...): */
3281 && (word->data[0] == '\0' || word->data[0] == SPECIAL_VAR_SYMBOL)
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003282 /* (otherwise it's known to be not empty and is already safe) */
Denis Vlasenkoab876cd2008-06-18 16:29:32 +00003283 ) {
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003284 /* exclude "$@" - it can expand to no word despite "" */
Denis Vlasenkoafdcd122008-07-05 17:40:04 +00003285 char *p = word->data;
3286 while (p[0] == SPECIAL_VAR_SYMBOL
3287 && (p[1] & 0x7f) == '@'
3288 && p[2] == SPECIAL_VAR_SYMBOL
3289 ) {
3290 p += 3;
3291 }
Denis Vlasenkoc1c63b62008-06-18 09:20:35 +00003292 }
Denis Vlasenko22d10a02008-10-13 08:53:43 +00003293 command->argv = add_string_to_strings(command->argv, xstrdup(word->data));
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003294 debug_print_strings("word appended to argv", command->argv);
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003295 }
Eric Andersen25f27032001-04-26 23:22:31 +00003296
Denis Vlasenko06810332007-05-21 23:30:54 +00003297#if ENABLE_HUSH_LOOPS
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003298 if (ctx->ctx_res_w == RES_FOR) {
Denys Vlasenko38292b62010-09-05 14:49:40 +02003299 if (word->has_quoted_part
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003300 || !is_well_formed_var_name(command->argv[0], '\0')
3301 ) {
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003302 /* bash says just "not a valid identifier" */
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003303 syntax_error("not a valid identifier in for");
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003304 return 1;
3305 }
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003306 /* Force FOR to have just one word (variable name) */
3307 /* NB: basically, this makes hush see "for v in ..."
3308 * syntax as if it is "for v; in ...". FOR and IN become
3309 * two pipe structs in parse tree. */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00003310 done_pipe(ctx, PIPE_SEQ);
Denis Vlasenko733e3fb2008-07-06 10:01:13 +00003311 }
Denis Vlasenko06810332007-05-21 23:30:54 +00003312#endif
Denis Vlasenko17f02e72008-07-14 04:32:29 +00003313#if ENABLE_HUSH_CASE
3314 /* Force CASE to have just one word */
3315 if (ctx->ctx_res_w == RES_CASE) {
3316 done_pipe(ctx, PIPE_SEQ);
3317 }
3318#endif
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003319
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003320 o_reset_to_empty_unquoted(word);
Denis Vlasenko1fd1ea42009-04-10 12:03:20 +00003321
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003322 debug_printf_parse("done_word return 0\n");
Eric Andersen25f27032001-04-26 23:22:31 +00003323 return 0;
3324}
3325
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003326
3327/* Peek ahead in the input to find out if we have a "&n" construct,
3328 * as in "2>&1", that represents duplicating a file descriptor.
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003329 * Return:
3330 * REDIRFD_CLOSE if >&- "close fd" construct is seen,
3331 * REDIRFD_SYNTAX_ERR if syntax error,
3332 * REDIRFD_TO_FILE if no & was seen,
3333 * or the number found.
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003334 */
3335#if BB_MMU
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003336#define parse_redir_right_fd(as_string, input) \
3337 parse_redir_right_fd(input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003338#endif
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003339static int parse_redir_right_fd(o_string *as_string, struct in_str *input)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003340{
3341 int ch, d, ok;
3342
3343 ch = i_peek(input);
3344 if (ch != '&')
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003345 return REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003346
3347 ch = i_getch(input); /* get the & */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003348 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003349 ch = i_peek(input);
3350 if (ch == '-') {
3351 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003352 nommu_addchr(as_string, ch);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003353 return REDIRFD_CLOSE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003354 }
3355 d = 0;
3356 ok = 0;
3357 while (ch != EOF && isdigit(ch)) {
3358 d = d*10 + (ch-'0');
3359 ok = 1;
3360 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003361 nommu_addchr(as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003362 ch = i_peek(input);
3363 }
3364 if (ok) return d;
3365
3366//TODO: this is the place to catch ">&file" bashism (redirect both fd 1 and 2)
3367
3368 bb_error_msg("ambiguous redirect");
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003369 return REDIRFD_SYNTAX_ERR;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003370}
3371
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003372/* Return code is 0 normal, 1 if a syntax error is detected
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003373 */
3374static int parse_redirect(struct parse_context *ctx,
3375 int fd,
3376 redir_type style,
3377 struct in_str *input)
3378{
3379 struct command *command = ctx->command;
3380 struct redir_struct *redir;
3381 struct redir_struct **redirp;
3382 int dup_num;
3383
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003384 dup_num = REDIRFD_TO_FILE;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003385 if (style != REDIRECT_HEREDOC) {
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003386 /* Check for a '>&1' type redirect */
3387 dup_num = parse_redir_right_fd(&ctx->as_string, input);
3388 if (dup_num == REDIRFD_SYNTAX_ERR)
3389 return 1;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003390 } else {
3391 int ch = i_peek(input);
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003392 dup_num = (ch == '-'); /* HEREDOC_SKIPTABS bit is 1 */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003393 if (dup_num) { /* <<-... */
3394 ch = i_getch(input);
3395 nommu_addchr(&ctx->as_string, ch);
3396 ch = i_peek(input);
3397 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003398 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003399
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003400 if (style == REDIRECT_OVERWRITE && dup_num == REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003401 int ch = i_peek(input);
3402 if (ch == '|') {
3403 /* >|FILE redirect ("clobbering" >).
3404 * Since we do not support "set -o noclobber" yet,
3405 * >| and > are the same for now. Just eat |.
3406 */
3407 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003408 nommu_addchr(&ctx->as_string, ch);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003409 }
3410 }
3411
3412 /* Create a new redir_struct and append it to the linked list */
3413 redirp = &command->redirects;
3414 while ((redir = *redirp) != NULL) {
3415 redirp = &(redir->next);
3416 }
3417 *redirp = redir = xzalloc(sizeof(*redir));
3418 /* redir->next = NULL; */
3419 /* redir->rd_filename = NULL; */
3420 redir->rd_type = style;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003421 redir->rd_fd = (fd == -1) ? redir_table[style].default_fd : fd;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003422
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003423 debug_printf_parse("redirect type %d %s\n", redir->rd_fd,
3424 redir_table[style].descrip);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003425
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003426 redir->rd_dup = dup_num;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00003427 if (style != REDIRECT_HEREDOC && dup_num != REDIRFD_TO_FILE) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003428 /* Erik had a check here that the file descriptor in question
3429 * is legit; I postpone that to "run time"
3430 * A "-" representation of "close me" shows up as a -3 here */
Denis Vlasenko02d6f1a2009-04-07 19:56:55 +00003431 debug_printf_parse("duplicating redirect '%d>&%d'\n",
3432 redir->rd_fd, redir->rd_dup);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003433 } else {
3434 /* Set ctx->pending_redirect, so we know what to do at the
3435 * end of the next parsed word. */
3436 ctx->pending_redirect = redir;
3437 }
3438 return 0;
3439}
3440
Eric Andersen25f27032001-04-26 23:22:31 +00003441/* If a redirect is immediately preceded by a number, that number is
3442 * supposed to tell which file descriptor to redirect. This routine
3443 * looks for such preceding numbers. In an ideal world this routine
3444 * needs to handle all the following classes of redirects...
3445 * echo 2>foo # redirects fd 2 to file "foo", nothing passed to echo
3446 * echo 49>foo # redirects fd 49 to file "foo", nothing passed to echo
3447 * echo -2>foo # redirects fd 1 to file "foo", "-2" passed to echo
3448 * echo 49x>foo # redirects fd 1 to file "foo", "49x" passed to echo
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003449 *
3450 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html
3451 * "2.7 Redirection
3452 * ... If n is quoted, the number shall not be recognized as part of
3453 * the redirection expression. For example:
3454 * echo \2>a
3455 * writes the character 2 into file a"
Denys Vlasenko38292b62010-09-05 14:49:40 +02003456 * We are getting it right by setting ->has_quoted_part on any \<char>
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003457 *
3458 * A -1 return means no valid number was found,
3459 * the caller should use the appropriate default for this redirection.
Eric Andersen25f27032001-04-26 23:22:31 +00003460 */
3461static int redirect_opt_num(o_string *o)
3462{
3463 int num;
3464
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003465 if (o->data == NULL)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00003466 return -1;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003467 num = bb_strtou(o->data, NULL, 10);
3468 if (errno || num < 0)
3469 return -1;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00003470 o_reset_to_empty_unquoted(o);
Eric Andersen25f27032001-04-26 23:22:31 +00003471 return num;
3472}
3473
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003474#if BB_MMU
3475#define fetch_till_str(as_string, input, word, skip_tabs) \
3476 fetch_till_str(input, word, skip_tabs)
3477#endif
3478static char *fetch_till_str(o_string *as_string,
3479 struct in_str *input,
3480 const char *word,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003481 int heredoc_flags)
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003482{
3483 o_string heredoc = NULL_O_STRING;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003484 unsigned past_EOL;
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003485 int prev = 0; /* not \ */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003486 int ch;
3487
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003488 goto jump_in;
Denys Vlasenkob8709032011-05-08 21:20:01 +02003489
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003490 while (1) {
3491 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003492 if (ch != EOF)
3493 nommu_addchr(as_string, ch);
3494 if ((ch == '\n' || ch == EOF)
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003495 && ((heredoc_flags & HEREDOC_QUOTED) || prev != '\\')
3496 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003497 if (strcmp(heredoc.data + past_EOL, word) == 0) {
3498 heredoc.data[past_EOL] = '\0';
3499 debug_printf_parse("parsed heredoc '%s'\n", heredoc.data);
3500 return heredoc.data;
3501 }
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003502 while (ch == '\n') {
3503 o_addchr(&heredoc, ch);
3504 prev = ch;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003505 jump_in:
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003506 past_EOL = heredoc.length;
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003507 do {
3508 ch = i_getch(input);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003509 if (ch != EOF)
3510 nommu_addchr(as_string, ch);
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003511 } while ((heredoc_flags & HEREDOC_SKIPTABS) && ch == '\t');
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003512 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003513 }
3514 if (ch == EOF) {
3515 o_free_unsafe(&heredoc);
3516 return NULL;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003517 }
3518 o_addchr(&heredoc, ch);
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02003519 nommu_addchr(as_string, ch);
Denys Vlasenkoc3adfac2010-09-06 11:46:03 +02003520 if (prev == '\\' && ch == '\\')
3521 /* Correctly handle foo\\<eol> (not a line cont.) */
3522 prev = 0; /* not \ */
3523 else
3524 prev = ch;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003525 }
3526}
3527
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003528/* Look at entire parse tree for not-yet-loaded REDIRECT_HEREDOCs
3529 * and load them all. There should be exactly heredoc_cnt of them.
3530 */
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003531static int fetch_heredocs(int heredoc_cnt, struct parse_context *ctx, struct in_str *input)
3532{
3533 struct pipe *pi = ctx->list_head;
3534
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003535 while (pi && heredoc_cnt) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003536 int i;
3537 struct command *cmd = pi->cmds;
3538
3539 debug_printf_parse("fetch_heredocs: num_cmds:%d cmd argv0:'%s'\n",
3540 pi->num_cmds,
3541 cmd->argv ? cmd->argv[0] : "NONE");
3542 for (i = 0; i < pi->num_cmds; i++) {
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003543 struct redir_struct *redir = cmd->redirects;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003544
3545 debug_printf_parse("fetch_heredocs: %d cmd argv0:'%s'\n",
3546 i, cmd->argv ? cmd->argv[0] : "NONE");
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003547 while (redir) {
3548 if (redir->rd_type == REDIRECT_HEREDOC) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003549 char *p;
3550
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003551 redir->rd_type = REDIRECT_HEREDOC2;
Denys Vlasenko764b2f02009-06-07 16:05:04 +02003552 /* redir->rd_dup is (ab)used to indicate <<- */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003553 p = fetch_till_str(&ctx->as_string, input,
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02003554 redir->rd_filename, redir->rd_dup);
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003555 if (!p) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003556 syntax_error("unexpected EOF in here document");
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003557 return 1;
3558 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003559 free(redir->rd_filename);
3560 redir->rd_filename = p;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003561 heredoc_cnt--;
3562 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003563 redir = redir->next;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003564 }
3565 cmd++;
3566 }
3567 pi = pi->next;
3568 }
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003569#if 0
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003570 /* Should be 0. If it isn't, it's a parse error */
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00003571 if (heredoc_cnt)
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003572 bb_error_msg_and_die("heredoc BUG 2");
3573#endif
3574 return 0;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00003575}
3576
3577
Denys Vlasenkob36abf22010-09-05 14:50:59 +02003578static int run_list(struct pipe *pi);
3579#if BB_MMU
3580#define parse_stream(pstring, input, end_trigger) \
3581 parse_stream(input, end_trigger)
3582#endif
3583static struct pipe *parse_stream(char **pstring,
3584 struct in_str *input,
3585 int end_trigger);
Denis Vlasenkoba7cf262007-05-25 14:34:30 +00003586
Eric Andersen25f27032001-04-26 23:22:31 +00003587
Denys Vlasenkoc2704542009-11-20 19:14:19 +01003588#if !ENABLE_HUSH_FUNCTIONS
3589#define parse_group(dest, ctx, input, ch) \
3590 parse_group(ctx, input, ch)
3591#endif
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003592static int parse_group(o_string *dest, struct parse_context *ctx,
Eric Andersen25f27032001-04-26 23:22:31 +00003593 struct in_str *input, int ch)
3594{
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003595 /* dest contains characters seen prior to ( or {.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00003596 * Typically it's empty, but for function defs,
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003597 * it contains function name (without '()'). */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003598 struct pipe *pipe_list;
Denis Vlasenko240c2552009-04-03 03:45:05 +00003599 int endch;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003600 struct command *command = ctx->command;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003601
3602 debug_printf_parse("parse_group entered\n");
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003603#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenko38292b62010-09-05 14:49:40 +02003604 if (ch == '(' && !dest->has_quoted_part) {
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003605 if (dest->length)
Denis Vlasenkobb929512009-04-16 10:59:40 +00003606 if (done_word(dest, ctx))
3607 return 1;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003608 if (!command->argv)
3609 goto skip; /* (... */
3610 if (command->argv[1]) { /* word word ... (... */
3611 syntax_error_unexpected_ch('(');
3612 return 1;
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003613 }
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003614 /* it is "word(..." or "word (..." */
3615 do
3616 ch = i_getch(input);
3617 while (ch == ' ' || ch == '\t');
3618 if (ch != ')') {
3619 syntax_error_unexpected_ch(ch);
3620 return 1;
3621 }
3622 nommu_addchr(&ctx->as_string, ch);
3623 do
3624 ch = i_getch(input);
3625 while (ch == ' ' || ch == '\t' || ch == '\n');
3626 if (ch != '{') {
3627 syntax_error_unexpected_ch(ch);
3628 return 1;
3629 }
3630 nommu_addchr(&ctx->as_string, ch);
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003631 command->cmd_type = CMD_FUNCDEF;
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00003632 goto skip;
Denis Vlasenko371de4a2008-10-14 12:43:13 +00003633 }
3634#endif
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003635
3636#if 0 /* Prevented by caller */
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003637 if (command->argv /* word [word]{... */
3638 || dest->length /* word{... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02003639 || dest->has_quoted_part /* ""{... */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003640 ) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00003641 syntax_error(NULL);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003642 debug_printf_parse("parse_group return 1: "
3643 "syntax error, groups and arglists don't mix\n");
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00003644 return 1;
Eric Andersen25f27032001-04-26 23:22:31 +00003645 }
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01003646#endif
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003647
3648#if ENABLE_HUSH_FUNCTIONS
3649 skip:
3650#endif
Denis Vlasenko240c2552009-04-03 03:45:05 +00003651 endch = '}';
Denis Vlasenko90e485c2007-05-23 15:22:50 +00003652 if (ch == '(') {
Denis Vlasenko240c2552009-04-03 03:45:05 +00003653 endch = ')';
Denys Vlasenko9d617c42009-06-09 18:40:52 +02003654 command->cmd_type = CMD_SUBSHELL;
Denis Vlasenkof8c1f022009-04-17 11:55:42 +00003655 } else {
3656 /* bash does not allow "{echo...", requires whitespace */
3657 ch = i_getch(input);
3658 if (ch != ' ' && ch != '\t' && ch != '\n') {
3659 syntax_error_unexpected_ch(ch);
3660 return 1;
3661 }
3662 nommu_addchr(&ctx->as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003663 }
Denis Vlasenkob7d8c0d2009-04-10 19:05:43 +00003664
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003665 {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003666#if BB_MMU
3667# define as_string NULL
3668#else
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003669 char *as_string = NULL;
3670#endif
3671 pipe_list = parse_stream(&as_string, input, endch);
3672#if !BB_MMU
3673 if (as_string)
3674 o_addstr(&ctx->as_string, as_string);
3675#endif
3676 /* empty ()/{} or parse error? */
3677 if (!pipe_list || pipe_list == ERR_PTR) {
Denis Vlasenkobb929512009-04-16 10:59:40 +00003678 /* parse_stream already emitted error msg */
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003679 if (!BB_MMU)
3680 free(as_string);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00003681 debug_printf_parse("parse_group return 1: "
3682 "parse_stream returned %p\n", pipe_list);
3683 return 1;
3684 }
3685 command->group = pipe_list;
3686#if !BB_MMU
3687 as_string[strlen(as_string) - 1] = '\0'; /* plink ')' or '}' */
3688 command->group_as_string = as_string;
3689 debug_printf_parse("end of group, remembering as:'%s'\n",
3690 command->group_as_string);
3691#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003692#undef as_string
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00003693 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00003694 debug_printf_parse("parse_group return 0\n");
3695 return 0;
Denis Vlasenko9af22c72008-10-09 12:54:58 +00003696 /* command remains "open", available for possible redirects */
Eric Andersen25f27032001-04-26 23:22:31 +00003697}
3698
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003699#if ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003700/* Subroutines for copying $(...) and `...` things */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003701static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003702/* '...' */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003703static int add_till_single_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003704{
3705 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003706 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003707 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003708 syntax_error_unterm_ch('\'');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003709 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003710 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003711 if (ch == '\'')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003712 return 1;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003713 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003714 }
3715}
3716/* "...\"...`..`...." - do we need to handle "...$(..)..." too? */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003717static int add_till_double_quote(o_string *dest, struct in_str *input)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003718{
3719 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003720 int ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003721 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003722 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003723 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003724 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003725 if (ch == '"')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003726 return 1;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003727 if (ch == '\\') { /* \x. Copy both chars. */
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003728 o_addchr(dest, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003729 ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003730 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003731 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003732 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003733 if (!add_till_backquote(dest, input, /*in_dquote:*/ 1))
3734 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003735 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003736 continue;
3737 }
Denis Vlasenko5703c222008-06-15 11:49:42 +00003738 //if (ch == '$') ...
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003739 }
3740}
3741/* Process `cmd` - copy contents until "`" is seen. Complicated by
3742 * \` quoting.
3743 * "Within the backquoted style of command substitution, backslash
3744 * shall retain its literal meaning, except when followed by: '$', '`', or '\'.
3745 * The search for the matching backquote shall be satisfied by the first
3746 * backquote found without a preceding backslash; during this search,
3747 * if a non-escaped backquote is encountered within a shell comment,
3748 * a here-document, an embedded command substitution of the $(command)
3749 * form, or a quoted string, undefined results occur. A single-quoted
3750 * or double-quoted string that begins, but does not end, within the
3751 * "`...`" sequence produces undefined results."
3752 * Example Output
3753 * echo `echo '\'TEST\`echo ZZ\`BEST` \TESTZZBEST
3754 */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003755static int add_till_backquote(o_string *dest, struct in_str *input, int in_dquote)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003756{
3757 while (1) {
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003758 int ch = i_getch(input);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003759 if (ch == '`')
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003760 return 1;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003761 if (ch == '\\') {
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003762 /* \x. Copy both unless it is \`, \$, \\ and maybe \" */
3763 ch = i_getch(input);
3764 if (ch != '`'
3765 && ch != '$'
3766 && ch != '\\'
3767 && (!in_dquote || ch != '"')
3768 ) {
3769 o_addchr(dest, '\\');
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003770 }
Denys Vlasenkoacd5bc82010-09-12 15:05:39 +02003771 }
3772 if (ch == EOF) {
3773 syntax_error_unterm_ch('`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003774 return 0;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003775 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003776 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003777 }
3778}
3779/* Process $(cmd) - copy contents until ")" is seen. Complicated by
3780 * quoting and nested ()s.
3781 * "With the $(command) style of command substitution, all characters
3782 * following the open parenthesis to the matching closing parenthesis
3783 * constitute the command. Any valid shell script can be used for command,
3784 * except a script consisting solely of redirections which produces
3785 * unspecified results."
3786 * Example Output
3787 * echo $(echo '(TEST)' BEST) (TEST) BEST
3788 * echo $(echo 'TEST)' BEST) TEST) BEST
3789 * echo $(echo \(\(TEST\) BEST) ((TEST) BEST
Denys Vlasenko74369502010-05-21 19:52:01 +02003790 *
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003791 * Also adapted to eat ${var%...} and $((...)) constructs, since ... part
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003792 * can contain arbitrary constructs, just like $(cmd).
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003793 * In bash compat mode, it needs to also be able to stop on ':' or '/'
3794 * for ${var:N[:M]} and ${var/P[/R]} parsing.
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003795 */
Denys Vlasenko74369502010-05-21 19:52:01 +02003796#define DOUBLE_CLOSE_CHAR_FLAG 0x80
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003797static int add_till_closing_bracket(o_string *dest, struct in_str *input, unsigned end_ch)
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003798{
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003799 int ch;
Denys Vlasenko74369502010-05-21 19:52:01 +02003800 char dbl = end_ch & DOUBLE_CLOSE_CHAR_FLAG;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003801# if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003802 char end_char2 = end_ch >> 8;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003803# endif
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003804 end_ch &= (DOUBLE_CLOSE_CHAR_FLAG - 1);
3805
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003806 while (1) {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003807 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003808 if (ch == EOF) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003809 syntax_error_unterm_ch(end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003810 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003811 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003812 if (ch == end_ch IF_HUSH_BASH_COMPAT( || ch == end_char2)) {
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003813 if (!dbl)
3814 break;
3815 /* we look for closing )) of $((EXPR)) */
3816 if (i_peek(input) == end_ch) {
3817 i_getch(input); /* eat second ')' */
3818 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003819 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003820 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003821 o_addchr(dest, ch);
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003822 if (ch == '(' || ch == '{') {
3823 ch = (ch == '(' ? ')' : '}');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003824 if (!add_till_closing_bracket(dest, input, ch))
3825 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003826 o_addchr(dest, ch);
3827 continue;
3828 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003829 if (ch == '\'') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003830 if (!add_till_single_quote(dest, input))
3831 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003832 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003833 continue;
3834 }
3835 if (ch == '"') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003836 if (!add_till_double_quote(dest, input))
3837 return 0;
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003838 o_addchr(dest, ch);
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003839 continue;
3840 }
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003841 if (ch == '`') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003842 if (!add_till_backquote(dest, input, /*in_dquote:*/ 0))
3843 return 0;
Denys Vlasenkoa6ad3972010-05-22 00:26:06 +02003844 o_addchr(dest, ch);
3845 continue;
3846 }
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003847 if (ch == '\\') {
3848 /* \x. Copy verbatim. Important for \(, \) */
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003849 ch = i_getch(input);
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003850 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00003851 syntax_error_unterm_ch(')');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003852 return 0;
Denis Vlasenko5c090a92009-04-08 21:51:33 +00003853 }
Denis Vlasenko82dfec32008-06-16 12:47:11 +00003854 o_addchr(dest, ch);
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00003855 continue;
3856 }
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003857 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003858 return ch;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003859}
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003860#endif /* ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_DOLLAR_OPS */
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00003861
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00003862/* Return code: 0 for OK, 1 for syntax error */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003863#if BB_MMU
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003864#define parse_dollar(as_string, dest, input, quote_mask) \
3865 parse_dollar(dest, input, quote_mask)
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003866#define as_string NULL
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003867#endif
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003868static int parse_dollar(o_string *as_string,
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003869 o_string *dest,
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003870 struct in_str *input, unsigned char quote_mask)
Eric Andersen25f27032001-04-26 23:22:31 +00003871{
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003872 int ch = i_peek(input); /* first character after the $ */
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003873
Denys Vlasenko2e48d532010-05-22 17:30:39 +02003874 debug_printf_parse("parse_dollar entered: ch='%c'\n", ch);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003875 if (isalpha(ch)) {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003876 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003877 nommu_addchr(as_string, ch);
Denis Vlasenkod4981312008-07-31 10:34:48 +00003878 make_var:
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003879 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003880 while (1) {
Denis Vlasenkoe0a33672007-05-10 23:06:55 +00003881 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003882 o_addchr(dest, ch | quote_mask);
Denis Vlasenko1f4cf512007-05-16 10:39:24 +00003883 quote_mask = 0;
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003884 ch = i_peek(input);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003885 if (!isalnum(ch) && ch != '_')
3886 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003887 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003888 nommu_addchr(as_string, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00003889 }
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003890 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003891 } else if (isdigit(ch)) {
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003892 make_one_char_var:
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00003893 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003894 nommu_addchr(as_string, ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003895 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenko602d13c2007-05-13 18:34:53 +00003896 debug_printf_parse(": '%c'\n", ch);
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00003897 o_addchr(dest, ch | quote_mask);
3898 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00003899 } else switch (ch) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003900 case '$': /* pid */
3901 case '!': /* last bg pid */
3902 case '?': /* last exit code */
3903 case '#': /* number of args */
3904 case '*': /* args */
3905 case '@': /* args */
3906 goto make_one_char_var;
3907 case '{': {
Mike Frysingeref3e7fd2009-06-01 14:13:39 -04003908 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3909
Denys Vlasenko74369502010-05-21 19:52:01 +02003910 ch = i_getch(input); /* eat '{' */
3911 nommu_addchr(as_string, ch);
3912
3913 ch = i_getch(input); /* first char after '{' */
Denys Vlasenko74369502010-05-21 19:52:01 +02003914 /* It should be ${?}, or ${#var},
3915 * or even ${?+subst} - operator acting on a special variable,
3916 * or the beginning of variable name.
3917 */
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003918 if (ch == EOF
3919 || (!strchr(_SPECIAL_VARS_STR, ch) && !isalnum(ch)) /* not one of those */
3920 ) {
Denys Vlasenko74369502010-05-21 19:52:01 +02003921 bad_dollar_syntax:
3922 syntax_error_unterm_str("${name}");
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003923 debug_printf_parse("parse_dollar return 0: unterminated ${name}\n");
3924 return 0;
Denys Vlasenko74369502010-05-21 19:52:01 +02003925 }
Denys Vlasenko101a4e32010-09-09 14:04:57 +02003926 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003927 ch |= quote_mask;
3928
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003929 /* It's possible to just call add_till_closing_bracket() at this point.
Denys Vlasenko74369502010-05-21 19:52:01 +02003930 * However, this regresses some of our testsuite cases
3931 * which check invalid constructs like ${%}.
3932 * Oh well... let's check that the var name part is fine... */
3933
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003934 while (1) {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003935 unsigned pos;
3936
Denys Vlasenko74369502010-05-21 19:52:01 +02003937 o_addchr(dest, ch);
3938 debug_printf_parse(": '%c'\n", ch);
3939
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003940 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00003941 nommu_addchr(as_string, ch);
Denys Vlasenko74369502010-05-21 19:52:01 +02003942 if (ch == '}')
Mike Frysinger98c52642009-04-02 10:02:37 +00003943 break;
Mike Frysinger98c52642009-04-02 10:02:37 +00003944
Denys Vlasenko74369502010-05-21 19:52:01 +02003945 if (!isalnum(ch) && ch != '_') {
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003946 unsigned end_ch;
3947 unsigned char last_ch;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00003948 /* handle parameter expansions
3949 * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
3950 */
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003951 if (!strchr(VAR_SUBST_OPS, ch)) /* ${var<bad_char>... */
Denys Vlasenko74369502010-05-21 19:52:01 +02003952 goto bad_dollar_syntax;
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003953
3954 /* Eat everything until closing '}' (or ':') */
3955 end_ch = '}';
3956 if (ENABLE_HUSH_BASH_COMPAT
3957 && ch == ':'
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003958 && !strchr(MINUS_PLUS_EQUAL_QUESTION, i_peek(input))
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003959 ) {
3960 /* It's ${var:N[:M]} thing */
3961 end_ch = '}' * 0x100 + ':';
3962 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003963 if (ENABLE_HUSH_BASH_COMPAT
3964 && ch == '/'
3965 ) {
3966 /* It's ${var/[/]pattern[/repl]} thing */
3967 if (i_peek(input) == '/') { /* ${var//pattern[/repl]}? */
3968 i_getch(input);
3969 nommu_addchr(as_string, '/');
3970 ch = '\\';
3971 }
3972 end_ch = '}' * 0x100 + '/';
3973 }
3974 o_addchr(dest, ch);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003975 again:
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003976 if (!BB_MMU)
3977 pos = dest->length;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003978#if ENABLE_HUSH_DOLLAR_OPS
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003979 last_ch = add_till_closing_bracket(dest, input, end_ch);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01003980 if (last_ch == 0) /* error? */
3981 return 0;
Denys Vlasenko9297dbc2010-07-05 21:37:12 +02003982#else
3983#error Simple code to only allow ${var} is not implemented
3984#endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003985 if (as_string) {
3986 o_addstr(as_string, dest->data + pos);
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003987 o_addchr(as_string, last_ch);
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02003988 }
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003989
3990 if (ENABLE_HUSH_BASH_COMPAT && (end_ch & 0xff00)) {
3991 /* close the first block: */
3992 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003993 /* while parsing N from ${var:N[:M]}
3994 * or pattern from ${var/[/]pattern[/repl]} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003995 if ((end_ch & 0xff) == last_ch) {
Denys Vlasenko36f774a2010-09-05 14:45:38 +02003996 /* got ':' or '/'- parse the rest */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02003997 end_ch = '}';
3998 goto again;
3999 }
Denys Vlasenko36f774a2010-09-05 14:45:38 +02004000 /* got '}' */
4001 if (end_ch == '}' * 0x100 + ':') {
4002 /* it's ${var:N} - emulate :999999999 */
4003 o_addstr(dest, "999999999");
4004 } /* else: it's ${var/[/]pattern} */
Denys Vlasenko1e811b12010-05-22 03:12:29 +02004005 }
Denys Vlasenko74369502010-05-21 19:52:01 +02004006 break;
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004007 }
Denys Vlasenko74369502010-05-21 19:52:01 +02004008 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004009 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4010 break;
4011 }
Denys Vlasenkoc0836532009-10-19 13:13:06 +02004012#if ENABLE_SH_MATH_SUPPORT || ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004013 case '(': {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004014 unsigned pos;
4015
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004016 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004017 nommu_addchr(as_string, ch);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004018# if ENABLE_SH_MATH_SUPPORT
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004019 if (i_peek(input) == '(') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004020 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004021 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004022 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4023 o_addchr(dest, /*quote_mask |*/ '+');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004024 if (!BB_MMU)
4025 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004026 if (!add_till_closing_bracket(dest, input, ')' | DOUBLE_CLOSE_CHAR_FLAG))
4027 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004028 if (as_string) {
4029 o_addstr(as_string, dest->data + pos);
4030 o_addchr(as_string, ')');
4031 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004032 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004033 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Eric Andersen25f27032001-04-26 23:22:31 +00004034 break;
Denis Vlasenko76db5ad2008-06-12 12:58:20 +00004035 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004036# endif
4037# if ENABLE_HUSH_TICK
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004038 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4039 o_addchr(dest, quote_mask | '`');
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004040 if (!BB_MMU)
4041 pos = dest->length;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004042 if (!add_till_closing_bracket(dest, input, ')'))
4043 return 0; /* error */
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00004044 if (as_string) {
4045 o_addstr(as_string, dest->data + pos);
Denys Vlasenkob70cef72010-01-12 13:45:45 +01004046 o_addchr(as_string, ')');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00004047 }
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004048 o_addchr(dest, SPECIAL_VAR_SYMBOL);
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004049# endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004050 break;
4051 }
Denis Vlasenkod85a5df2009-04-05 08:43:57 +00004052#endif
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004053 case '_':
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004054 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004055 nommu_addchr(as_string, ch);
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004056 ch = i_peek(input);
4057 if (isalnum(ch)) { /* it's $_name or $_123 */
4058 ch = '_';
4059 goto make_var;
4060 }
4061 /* else: it's $_ */
Denys Vlasenko69b1cef2009-09-21 10:21:44 +02004062 /* TODO: $_ and $-: */
4063 /* $_ Shell or shell script name; or last argument of last command
4064 * (if last command wasn't a pipe; if it was, bash sets $_ to "");
4065 * but in command's env, set to full pathname used to invoke it */
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004066 /* $- Option flags set by set builtin or shell options (-i etc) */
4067 default:
4068 o_addQchr(dest, '$');
Eric Andersen25f27032001-04-26 23:22:31 +00004069 }
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004070 debug_printf_parse("parse_dollar return 1 (ok)\n");
4071 return 1;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004072#undef as_string
Eric Andersen25f27032001-04-26 23:22:31 +00004073}
4074
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004075#if BB_MMU
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004076# if ENABLE_HUSH_BASH_COMPAT
4077#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4078 encode_string(dest, input, dquote_end, process_bkslash)
4079# else
4080/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4081#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4082 encode_string(dest, input, dquote_end)
4083# endif
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004084#define as_string NULL
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004085
4086#else /* !MMU */
4087
4088# if ENABLE_HUSH_BASH_COMPAT
4089/* all parameters are needed, no macro tricks */
4090# else
4091#define encode_string(as_string, dest, input, dquote_end, process_bkslash) \
4092 encode_string(as_string, dest, input, dquote_end)
4093# endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004094#endif
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004095static int encode_string(o_string *as_string,
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004096 o_string *dest,
4097 struct in_str *input,
Denys Vlasenko14e289b2010-09-10 10:15:18 +02004098 int dquote_end,
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004099 int process_bkslash)
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004100{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004101#if !ENABLE_HUSH_BASH_COMPAT
4102 const int process_bkslash = 1;
4103#endif
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004104 int ch;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004105 int next;
4106
4107 again:
4108 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004109 if (ch != EOF)
4110 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004111 if (ch == dquote_end) { /* may be only '"' or EOF */
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004112 debug_printf_parse("encode_string return 1 (ok)\n");
4113 return 1;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004114 }
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004115 /* note: can't move it above ch == dquote_end check! */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004116 if (ch == EOF) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004117 syntax_error_unterm_ch('"');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004118 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004119 }
4120 next = '\0';
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004121 if (ch != '\n') {
4122 next = i_peek(input);
4123 }
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004124 debug_printf_parse("\" ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004125 ch, ch, !!(dest->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004126 if (process_bkslash && ch == '\\') {
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004127 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004128 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004129 xfunc_die();
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004130 }
4131 /* bash:
4132 * "The backslash retains its special meaning [in "..."]
4133 * only when followed by one of the following characters:
4134 * $, `, ", \, or <newline>. A double quote may be quoted
Denys Vlasenkoe640cb42009-05-28 16:49:11 +02004135 * within double quotes by preceding it with a backslash."
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004136 * NB: in (unquoted) heredoc, above does not apply to ",
4137 * therefore we check for it by "next == dquote_end" cond.
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004138 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004139 if (next == dquote_end || strchr("$`\\\n", next)) {
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004140 ch = i_getch(input); /* eat next */
4141 if (ch == '\n')
4142 goto again; /* skip \<newline> */
Denys Vlasenko4f870492010-09-10 11:06:01 +02004143 } /* else: ch remains == '\\', and we double it below: */
4144 o_addqchr(dest, ch); /* \c if c is a glob char, else just c */
Denys Vlasenko850b15b2010-09-09 12:58:19 +02004145 nommu_addchr(as_string, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004146 goto again;
4147 }
4148 if (ch == '$') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004149 if (!parse_dollar(as_string, dest, input, /*quote_mask:*/ 0x80)) {
4150 debug_printf_parse("encode_string return 0: "
4151 "parse_dollar returned 0 (error)\n");
4152 return 0;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004153 }
4154 goto again;
4155 }
4156#if ENABLE_HUSH_TICK
4157 if (ch == '`') {
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004158 //unsigned pos = dest->length;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004159 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4160 o_addchr(dest, 0x80 | '`');
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004161 if (!add_till_backquote(dest, input, /*in_dquote:*/ dquote_end == '"'))
4162 return 0; /* error */
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004163 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4164 //debug_printf_subst("SUBST RES3 '%s'\n", dest->data + pos);
Denis Vlasenkof328e002009-04-02 16:55:38 +00004165 goto again;
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004166 }
4167#endif
Denis Vlasenkof328e002009-04-02 16:55:38 +00004168 o_addQchr(dest, ch);
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004169 goto again;
Denys Vlasenkoddc62f62010-05-22 00:53:32 +02004170#undef as_string
Denis Vlasenko2f1d3942009-04-02 16:31:29 +00004171}
4172
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004173/*
4174 * Scan input until EOF or end_trigger char.
4175 * Return a list of pipes to execute, or NULL on EOF
4176 * or if end_trigger character is met.
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004177 * On syntax error, exit if shell is not interactive,
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004178 * reset parsing machinery and start parsing anew,
4179 * or return ERR_PTR.
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004180 */
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004181static struct pipe *parse_stream(char **pstring,
4182 struct in_str *input,
4183 int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00004184{
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004185 struct parse_context ctx;
4186 o_string dest = NULL_O_STRING;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004187 int heredoc_cnt;
Eric Andersen25f27032001-04-26 23:22:31 +00004188
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004189 /* Single-quote triggers a bypass of the main loop until its mate is
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004190 * found. When recursing, quote state is passed in via dest->o_expflags.
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004191 */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004192 debug_printf_parse("parse_stream entered, end_trigger='%c'\n",
Denys Vlasenko90a99042009-09-06 02:36:23 +02004193 end_trigger ? end_trigger : 'X');
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004194 debug_enter();
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004195
Denys Vlasenkof37eb392009-10-18 11:46:35 +02004196 /* If very first arg is "" or '', dest.data may end up NULL.
4197 * Preventing this: */
4198 o_addchr(&dest, '\0');
4199 dest.length = 0;
4200
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004201 /* We used to separate words on $IFS here. This was wrong.
4202 * $IFS is used only for word splitting when $var is expanded,
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004203 * here we should use blank chars as separators, not $IFS
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004204 */
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004205
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004206 if (MAYBE_ASSIGNMENT != 0)
4207 dest.o_assignment = MAYBE_ASSIGNMENT;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004208 initialize_context(&ctx);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004209 heredoc_cnt = 0;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004210 while (1) {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004211 const char *is_blank;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004212 const char *is_special;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004213 int ch;
4214 int next;
4215 int redir_fd;
4216 redir_type redir_style;
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004217
Denis Vlasenko46ccdcb2008-06-10 18:05:12 +00004218 ch = i_getch(input);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004219 debug_printf_parse(": ch=%c (%d) escape=%d\n",
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02004220 ch, ch, !!(dest.o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004221 if (ch == EOF) {
4222 struct pipe *pi;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004223
4224 if (heredoc_cnt) {
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004225 syntax_error_unterm_str("here document");
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004226 goto parse_error;
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004227 }
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004228 /* end_trigger == '}' case errors out earlier,
4229 * checking only ')' */
4230 if (end_trigger == ')') {
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004231 syntax_error_unterm_ch('(');
4232 goto parse_error;
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004233 }
4234
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004235 if (done_word(&dest, &ctx)) {
Denys Vlasenkob1cfc452009-05-02 17:18:34 +02004236 goto parse_error;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004237 }
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004238 o_free(&dest);
4239 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004240 pi = ctx.list_head;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004241 /* If we got nothing... */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004242 /* (this makes bare "&" cmd a no-op.
4243 * bash says: "syntax error near unexpected token '&'") */
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004244 if (pi->num_cmds == 0
Denys Vlasenko60cb48c2013-01-14 15:57:44 +01004245 IF_HAS_KEYWORDS(&& pi->res_word == RES_NONE)
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004246 ) {
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004247 free_pipe_list(pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004248 pi = NULL;
4249 }
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004250#if !BB_MMU
4251 debug_printf_parse("as_string '%s'\n", ctx.as_string.data);
4252 if (pstring)
4253 *pstring = ctx.as_string.data;
4254 else
4255 o_free_unsafe(&ctx.as_string);
4256#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004257 debug_leave();
4258 debug_printf_parse("parse_stream return %p\n", pi);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004259 return pi;
Denis Vlasenko1a735862007-05-23 00:32:25 +00004260 }
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004261 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004262
4263 next = '\0';
4264 if (ch != '\n')
4265 next = i_peek(input);
4266
4267 is_special = "{}<>;&|()#'" /* special outside of "str" */
4268 "\\$\"" IF_HUSH_TICK("`"); /* always special */
4269 /* Are { and } special here? */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004270 if (ctx.command->argv /* word [word]{... - non-special */
4271 || dest.length /* word{... - non-special */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004272 || dest.has_quoted_part /* ""{... - non-special */
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004273 || (next != ';' /* }; - special */
4274 && next != ')' /* }) - special */
4275 && next != '&' /* }& and }&& ... - special */
4276 && next != '|' /* }|| ... - special */
4277 && !strchr(defifs, next) /* {word - non-special */
Denys Vlasenko3227d3f2010-05-17 09:49:47 +02004278 )
Denys Vlasenkod8389ad2009-11-16 03:18:46 +01004279 ) {
4280 /* They are not special, skip "{}" */
4281 is_special += 2;
4282 }
4283 is_special = strchr(is_special, ch);
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004284 is_blank = strchr(defifs, ch);
Denis Vlasenko6da69cd2009-04-04 12:12:58 +00004285
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004286 if (!is_special && !is_blank) { /* ordinary char */
Denis Vlasenkobf25fbc2009-04-19 13:57:51 +00004287 ordinary_char:
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004288 o_addQchr(&dest, ch);
4289 if ((dest.o_assignment == MAYBE_ASSIGNMENT
4290 || dest.o_assignment == WORD_IS_KEYWORD)
Denis Vlasenko55789c62008-06-18 16:30:42 +00004291 && ch == '='
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004292 && is_well_formed_var_name(dest.data, '=')
Denis Vlasenko55789c62008-06-18 16:30:42 +00004293 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004294 dest.o_assignment = DEFINITELY_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004295 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko55789c62008-06-18 16:30:42 +00004296 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004297 continue;
4298 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004299
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004300 if (is_blank) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004301 if (done_word(&dest, &ctx)) {
4302 goto parse_error;
Eric Andersenaac75e52001-04-30 18:18:45 +00004303 }
Denis Vlasenko37181682009-04-03 03:19:15 +00004304 if (ch == '\n') {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004305 /* Is this a case when newline is simply ignored?
4306 * Some examples:
4307 * "cmd | <newline> cmd ..."
4308 * "case ... in <newline> word) ..."
4309 */
4310 if (IS_NULL_CMD(ctx.command)
4311 && dest.length == 0 && !dest.has_quoted_part
Denis Vlasenkof1736072008-07-31 10:09:26 +00004312 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004313 /* This newline can be ignored. But...
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004314 * Without check #1, interactive shell
4315 * ignores even bare <newline>,
4316 * and shows the continuation prompt:
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004317 * ps1_prompt$ <enter>
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004318 * ps2> _ <=== wrong, should be ps1
4319 * Without check #2, "cmd & <newline>"
4320 * is similarly mistreated.
4321 * (BTW, this makes "cmd & cmd"
4322 * and "cmd && cmd" non-orthogonal.
4323 * Really, ask yourself, why
4324 * "cmd && <newline>" doesn't start
4325 * cmd but waits for more input?
4326 * No reason...)
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004327 */
4328 struct pipe *pi = ctx.list_head;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004329 if (pi->num_cmds != 0 /* check #1 */
4330 && pi->followup != PIPE_BG /* check #2 */
4331 ) {
Denys Vlasenko642e71a2011-01-07 15:16:05 +01004332 continue;
Denys Vlasenko98c46d12011-01-18 17:30:07 +01004333 }
Denis Vlasenkof1736072008-07-31 10:09:26 +00004334 }
Denis Vlasenko240c2552009-04-03 03:45:05 +00004335 /* Treat newline as a command separator. */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004336 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004337 debug_printf_parse("heredoc_cnt:%d\n", heredoc_cnt);
4338 if (heredoc_cnt) {
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004339 if (fetch_heredocs(heredoc_cnt, &ctx, input)) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004340 goto parse_error;
Denis Vlasenko3dfb0352009-04-08 09:29:14 +00004341 }
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004342 heredoc_cnt = 0;
4343 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004344 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004345 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004346 ch = ';';
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004347 /* note: if (is_blank) continue;
Denis Vlasenko240c2552009-04-03 03:45:05 +00004348 * will still trigger for us */
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004349 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004350 }
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004351
4352 /* "cmd}" or "cmd }..." without semicolon or &:
4353 * } is an ordinary char in this case, even inside { cmd; }
4354 * Pathological example: { ""}; } should exec "}" cmd
4355 */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004356 if (ch == '}') {
4357 if (!IS_NULL_CMD(ctx.command) /* cmd } */
4358 || dest.length != 0 /* word} */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004359 || dest.has_quoted_part /* ""} */
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004360 ) {
4361 goto ordinary_char;
4362 }
4363 if (!IS_NULL_PIPE(ctx.pipe)) /* cmd | } */
4364 goto skip_end_trigger;
4365 /* else: } does terminate a group */
Denis Vlasenko9f8d9382009-04-19 14:03:11 +00004366 }
4367
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004368 if (end_trigger && end_trigger == ch
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004369 && (ch != ';' || heredoc_cnt == 0)
4370#if ENABLE_HUSH_CASE
4371 && (ch != ')'
4372 || ctx.ctx_res_w != RES_MATCH
Denys Vlasenko38292b62010-09-05 14:49:40 +02004373 || (!dest.has_quoted_part && strcmp(dest.data, "esac") == 0)
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004374 )
4375#endif
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004376 ) {
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004377 if (heredoc_cnt) {
4378 /* This is technically valid:
4379 * { cat <<HERE; }; echo Ok
4380 * heredoc
4381 * heredoc
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004382 * HERE
4383 * but we don't support this.
4384 * We require heredoc to be in enclosing {}/(),
4385 * if any.
4386 */
Denis Vlasenkod68ae082009-04-09 20:41:34 +00004387 syntax_error_unterm_str("here document");
Denis Vlasenko6c9be7f2009-04-07 02:29:51 +00004388 goto parse_error;
4389 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004390 if (done_word(&dest, &ctx)) {
4391 goto parse_error;
4392 }
4393 done_pipe(&ctx, PIPE_SEQ);
4394 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004395 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenko240c2552009-04-03 03:45:05 +00004396 /* Do we sit outside of any if's, loops or case's? */
Denis Vlasenko37181682009-04-03 03:19:15 +00004397 if (!HAS_KEYWORDS
Denys Vlasenko60cb48c2013-01-14 15:57:44 +01004398 IF_HAS_KEYWORDS(|| (ctx.ctx_res_w == RES_NONE && ctx.old_flag == 0))
Denis Vlasenko37181682009-04-03 03:19:15 +00004399 ) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004400 o_free(&dest);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004401#if !BB_MMU
4402 debug_printf_parse("as_string '%s'\n", ctx.as_string.data);
4403 if (pstring)
4404 *pstring = ctx.as_string.data;
4405 else
4406 o_free_unsafe(&ctx.as_string);
4407#endif
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004408 debug_leave();
4409 debug_printf_parse("parse_stream return %p: "
4410 "end_trigger char found\n",
4411 ctx.list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004412 return ctx.list_head;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004413 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004414 }
Denis Vlasenkodcd78c42009-04-19 23:07:51 +00004415 skip_end_trigger:
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004416 if (is_blank)
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004417 continue;
Denis Vlasenko55789c62008-06-18 16:30:42 +00004418
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004419 /* Catch <, > before deciding whether this word is
4420 * an assignment. a=1 2>z b=2: b=2 is still assignment */
4421 switch (ch) {
4422 case '>':
4423 redir_fd = redirect_opt_num(&dest);
4424 if (done_word(&dest, &ctx)) {
4425 goto parse_error;
4426 }
4427 redir_style = REDIRECT_OVERWRITE;
4428 if (next == '>') {
4429 redir_style = REDIRECT_APPEND;
4430 ch = i_getch(input);
4431 nommu_addchr(&ctx.as_string, ch);
4432 }
4433#if 0
4434 else if (next == '(') {
4435 syntax_error(">(process) not supported");
4436 goto parse_error;
4437 }
4438#endif
4439 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4440 goto parse_error;
4441 continue; /* back to top of while (1) */
4442 case '<':
4443 redir_fd = redirect_opt_num(&dest);
4444 if (done_word(&dest, &ctx)) {
4445 goto parse_error;
4446 }
4447 redir_style = REDIRECT_INPUT;
4448 if (next == '<') {
4449 redir_style = REDIRECT_HEREDOC;
4450 heredoc_cnt++;
4451 debug_printf_parse("++heredoc_cnt=%d\n", heredoc_cnt);
4452 ch = i_getch(input);
4453 nommu_addchr(&ctx.as_string, ch);
4454 } else if (next == '>') {
4455 redir_style = REDIRECT_IO;
4456 ch = i_getch(input);
4457 nommu_addchr(&ctx.as_string, ch);
4458 }
4459#if 0
4460 else if (next == '(') {
4461 syntax_error("<(process) not supported");
4462 goto parse_error;
4463 }
4464#endif
4465 if (parse_redirect(&ctx, redir_fd, redir_style, input))
4466 goto parse_error;
4467 continue; /* back to top of while (1) */
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004468 case '#':
4469 if (dest.length == 0 && !dest.has_quoted_part) {
4470 /* skip "#comment" */
4471 while (1) {
4472 ch = i_peek(input);
4473 if (ch == EOF || ch == '\n')
4474 break;
4475 i_getch(input);
4476 /* note: we do not add it to &ctx.as_string */
4477 }
4478 nommu_addchr(&ctx.as_string, '\n');
4479 continue; /* back to top of while (1) */
4480 }
4481 break;
4482 case '\\':
4483 if (next == '\n') {
4484 /* It's "\<newline>" */
4485#if !BB_MMU
4486 /* Remove trailing '\' from ctx.as_string */
4487 ctx.as_string.data[--ctx.as_string.length] = '\0';
4488#endif
4489 ch = i_getch(input); /* eat it */
4490 continue; /* back to top of while (1) */
4491 }
4492 break;
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004493 }
4494
4495 if (dest.o_assignment == MAYBE_ASSIGNMENT
4496 /* check that we are not in word in "a=1 2>word b=1": */
4497 && !ctx.pending_redirect
4498 ) {
4499 /* ch is a special char and thus this word
4500 * cannot be an assignment */
4501 dest.o_assignment = NOT_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004502 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Denis Vlasenkoc96865f2009-04-10 00:20:58 +00004503 }
4504
Denys Vlasenkocbfe6ad2009-08-12 19:47:44 +02004505 /* Note: nommu_addchr(&ctx.as_string, ch) is already done */
4506
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004507 switch (ch) {
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004508 case '#': /* non-comment #: "echo a#b" etc */
4509 o_addQchr(&dest, ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004510 break;
4511 case '\\':
4512 if (next == EOF) {
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00004513 syntax_error("\\<eof>");
Denis Vlasenko0b677d82009-04-10 13:49:10 +00004514 xfunc_die();
Eric Andersen25f27032001-04-26 23:22:31 +00004515 }
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004516 ch = i_getch(input);
Denys Vlasenko7b4c0fd2010-11-22 17:58:14 +01004517 /* note: ch != '\n' (that case does not reach this place) */
4518 o_addchr(&dest, '\\');
4519 /*nommu_addchr(&ctx.as_string, '\\'); - already done */
4520 o_addchr(&dest, ch);
4521 nommu_addchr(&ctx.as_string, ch);
4522 /* Example: echo Hello \2>file
4523 * we need to know that word 2 is quoted */
4524 dest.has_quoted_part = 1;
Eric Andersen25f27032001-04-26 23:22:31 +00004525 break;
4526 case '$':
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004527 if (!parse_dollar(&ctx.as_string, &dest, input, /*quote_mask:*/ 0)) {
Denis Vlasenkoa24c8ca2009-04-04 15:24:40 +00004528 debug_printf_parse("parse_stream parse error: "
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004529 "parse_dollar returned 0 (error)\n");
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004530 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004531 }
Eric Andersen25f27032001-04-26 23:22:31 +00004532 break;
4533 case '\'':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004534 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004535 if (next == '\'' && !ctx.pending_redirect) {
4536 insert_empty_quoted_str_marker:
4537 nommu_addchr(&ctx.as_string, next);
4538 i_getch(input); /* eat second ' */
4539 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4540 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4541 } else {
4542 while (1) {
4543 ch = i_getch(input);
4544 if (ch == EOF) {
4545 syntax_error_unterm_ch('\'');
4546 goto parse_error;
4547 }
4548 nommu_addchr(&ctx.as_string, ch);
4549 if (ch == '\'')
4550 break;
4551 o_addqchr(&dest, ch);
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004552 }
Eric Andersen25f27032001-04-26 23:22:31 +00004553 }
Eric Andersen25f27032001-04-26 23:22:31 +00004554 break;
4555 case '"':
Denys Vlasenko38292b62010-09-05 14:49:40 +02004556 dest.has_quoted_part = 1;
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004557 if (next == '"' && !ctx.pending_redirect)
4558 goto insert_empty_quoted_str_marker;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004559 if (dest.o_assignment == NOT_ASSIGNMENT)
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004560 dest.o_expflags |= EXP_FLAG_ESC_GLOB_CHARS;
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004561 if (!encode_string(&ctx.as_string, &dest, input, '"', /*process_bkslash:*/ 1))
Denys Vlasenko77a7b552010-09-09 12:40:03 +02004562 goto parse_error;
Denys Vlasenko5b6210c2010-09-09 13:32:21 +02004563 dest.o_expflags &= ~EXP_FLAG_ESC_GLOB_CHARS;
Eric Andersen25f27032001-04-26 23:22:31 +00004564 break;
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004565#if ENABLE_HUSH_TICK
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004566 case '`': {
Denys Vlasenko60a94142011-05-13 20:57:01 +02004567 USE_FOR_NOMMU(unsigned pos;)
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004568
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004569 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4570 o_addchr(&dest, '`');
Denys Vlasenko60a94142011-05-13 20:57:01 +02004571 USE_FOR_NOMMU(pos = dest.length;)
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004572 if (!add_till_backquote(&dest, input, /*in_dquote:*/ 0))
4573 goto parse_error;
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004574# if !BB_MMU
Denis Vlasenko5c090a92009-04-08 21:51:33 +00004575 o_addstr(&ctx.as_string, dest.data + pos);
4576 o_addchr(&ctx.as_string, '`');
Denys Vlasenko2e48d532010-05-22 17:30:39 +02004577# endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004578 o_addchr(&dest, SPECIAL_VAR_SYMBOL);
4579 //debug_printf_subst("SUBST RES3 '%s'\n", dest.data + pos);
Eric Andersen25f27032001-04-26 23:22:31 +00004580 break;
Denis Vlasenko7b4f3f12008-06-10 18:04:32 +00004581 }
Denis Vlasenko14b5dd92007-05-20 21:51:38 +00004582#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004583 case ';':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004584#if ENABLE_HUSH_CASE
4585 case_semi:
4586#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004587 if (done_word(&dest, &ctx)) {
4588 goto parse_error;
4589 }
4590 done_pipe(&ctx, PIPE_SEQ);
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004591#if ENABLE_HUSH_CASE
4592 /* Eat multiple semicolons, detect
4593 * whether it means something special */
4594 while (1) {
4595 ch = i_peek(input);
4596 if (ch != ';')
4597 break;
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004598 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004599 nommu_addchr(&ctx.as_string, ch);
Denys Vlasenkoe9bda902009-05-23 16:50:07 +02004600 if (ctx.ctx_res_w == RES_CASE_BODY) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004601 ctx.ctx_dsemicolon = 1;
4602 ctx.ctx_res_w = RES_MATCH;
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004603 break;
4604 }
4605 }
4606#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004607 new_cmd:
4608 /* We just finished a cmd. New one may start
4609 * with an assignment */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004610 dest.o_assignment = MAYBE_ASSIGNMENT;
Denys Vlasenko29f9b722011-05-14 11:27:36 +02004611 debug_printf_parse("dest.o_assignment='%s'\n", assignment_flag[dest.o_assignment]);
Eric Andersen25f27032001-04-26 23:22:31 +00004612 break;
4613 case '&':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004614 if (done_word(&dest, &ctx)) {
4615 goto parse_error;
4616 }
Denis Vlasenkobb81c582007-01-30 22:32:09 +00004617 if (next == '&') {
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004618 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004619 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004620 done_pipe(&ctx, PIPE_AND);
Eric Andersen25f27032001-04-26 23:22:31 +00004621 } else {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004622 done_pipe(&ctx, PIPE_BG);
Eric Andersen25f27032001-04-26 23:22:31 +00004623 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004624 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004625 case '|':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004626 if (done_word(&dest, &ctx)) {
4627 goto parse_error;
4628 }
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004629#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004630 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenkof1736072008-07-31 10:09:26 +00004631 break; /* we are in case's "word | word)" */
Denis Vlasenkofbeeb322008-07-31 00:17:01 +00004632#endif
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004633 if (next == '|') { /* || */
Denis Vlasenko609f2ab2009-04-04 23:15:14 +00004634 ch = i_getch(input);
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00004635 nommu_addchr(&ctx.as_string, ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004636 done_pipe(&ctx, PIPE_OR);
Eric Andersen25f27032001-04-26 23:22:31 +00004637 } else {
4638 /* we could pick up a file descriptor choice here
4639 * with redirect_opt_num(), but bash doesn't do it.
4640 * "echo foo 2| cat" yields "foo 2". */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004641 done_command(&ctx);
Denys Vlasenkob70cef72010-01-12 13:45:45 +01004642#if !BB_MMU
4643 o_reset_to_empty_unquoted(&ctx.as_string);
4644#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004645 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004646 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004647 case '(':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004648#if ENABLE_HUSH_CASE
Denis Vlasenkof1736072008-07-31 10:09:26 +00004649 /* "case... in [(]word)..." - skip '(' */
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004650 if (ctx.ctx_res_w == RES_MATCH
4651 && ctx.command->argv == NULL /* not (word|(... */
4652 && dest.length == 0 /* not word(... */
Denys Vlasenko38292b62010-09-05 14:49:40 +02004653 && dest.has_quoted_part == 0 /* not ""(... */
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004654 ) {
4655 continue;
4656 }
4657#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004658 case '{':
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004659 if (parse_group(&dest, &ctx, input, ch) != 0) {
4660 goto parse_error;
Denis Vlasenkoe725bfe2007-05-03 22:45:39 +00004661 }
Denis Vlasenko2b576b82008-08-04 00:46:07 +00004662 goto new_cmd;
Eric Andersen25f27032001-04-26 23:22:31 +00004663 case ')':
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004664#if ENABLE_HUSH_CASE
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004665 if (ctx.ctx_res_w == RES_MATCH)
Denis Vlasenko17f02e72008-07-14 04:32:29 +00004666 goto case_semi;
4667#endif
Eric Andersen25f27032001-04-26 23:22:31 +00004668 case '}':
Denis Vlasenkoc3735272008-10-09 12:58:26 +00004669 /* proper use of this character is caught by end_trigger:
4670 * if we see {, we call parse_group(..., end_trigger='}')
4671 * and it will match } earlier (not here). */
Denis Vlasenkoc0ea3292009-04-10 21:22:02 +00004672 syntax_error_unexpected_ch(ch);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004673 goto parse_error;
Eric Andersen25f27032001-04-26 23:22:31 +00004674 default:
Denis Vlasenko5ec61322008-06-24 00:50:07 +00004675 if (HUSH_DEBUG)
Denis Vlasenko90e485c2007-05-23 15:22:50 +00004676 bb_error_msg_and_die("BUG: unexpected %c\n", ch);
Eric Andersen25f27032001-04-26 23:22:31 +00004677 }
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00004678 } /* while (1) */
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004679
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004680 parse_error:
4681 {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004682 struct parse_context *pctx;
4683 IF_HAS_KEYWORDS(struct parse_context *p2;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004684
4685 /* Clean up allocated tree.
Denys Vlasenko764b2f02009-06-07 16:05:04 +02004686 * Sample for finding leaks on syntax error recovery path.
4687 * Run it from interactive shell, watch pmap `pidof hush`.
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004688 * while if false; then false; fi; do break; fi
Denis Vlasenkocc4c6932009-04-05 07:38:48 +00004689 * Samples to catch leaks at execution:
4690 * while if (true | {true;}); then echo ok; fi; do break; done
4691 * 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 +00004692 */
4693 pctx = &ctx;
4694 do {
4695 /* Update pipe/command counts,
4696 * otherwise freeing may miss some */
4697 done_pipe(pctx, PIPE_SEQ);
4698 debug_printf_clean("freeing list %p from ctx %p\n",
4699 pctx->list_head, pctx);
4700 debug_print_tree(pctx->list_head, 0);
Denis Vlasenko0701dca2009-04-11 10:38:47 +00004701 free_pipe_list(pctx->list_head);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004702 debug_printf_clean("freed list %p\n", pctx->list_head);
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004703#if !BB_MMU
4704 o_free_unsafe(&pctx->as_string);
4705#endif
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004706 IF_HAS_KEYWORDS(p2 = pctx->stack;)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00004707 if (pctx != &ctx) {
4708 free(pctx);
4709 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00004710 IF_HAS_KEYWORDS(pctx = p2;)
4711 } while (HAS_KEYWORDS && pctx);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004712
Denys Vlasenkoa439fa92011-03-30 19:11:46 +02004713 o_free(&dest);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004714 G.last_exitcode = 1;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004715#if !BB_MMU
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004716 if (pstring)
4717 *pstring = NULL;
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00004718#endif
Denys Vlasenkocecbc982011-03-30 18:54:52 +02004719 debug_leave();
4720 return ERR_PTR;
Denis Vlasenko027e3fd2009-04-02 22:50:40 +00004721 }
Eric Andersen25f27032001-04-26 23:22:31 +00004722}
4723
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004724
4725/*** Execution routines ***/
4726
4727/* Expansion can recurse, need forward decls: */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004728#if !ENABLE_HUSH_BASH_COMPAT
4729/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4730#define expand_string_to_string(str, do_unbackslash) \
4731 expand_string_to_string(str)
4732#endif
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004733static char *expand_string_to_string(const char *str, int do_unbackslash);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004734#if ENABLE_HUSH_TICK
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004735static int process_command_subs(o_string *dest, const char *s);
Denys Vlasenko26777aa2010-11-22 23:49:10 +01004736#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004737
4738/* expand_strvec_to_strvec() takes a list of strings, expands
4739 * all variable references within and returns a pointer to
4740 * a list of expanded strings, possibly with larger number
4741 * of strings. (Think VAR="a b"; echo $VAR).
4742 * This new list is allocated as a single malloc block.
4743 * NULL-terminated list of char* pointers is at the beginning of it,
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004744 * followed by strings themselves.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004745 * Caller can deallocate entire list by single free(list). */
4746
Denys Vlasenko238081f2010-10-03 14:26:26 +02004747/* A horde of its helpers come first: */
4748
4749static void o_addblock_duplicate_backslash(o_string *o, const char *str, int len)
4750{
4751 while (--len >= 0) {
Denys Vlasenko9e800222010-10-03 14:28:04 +02004752 char c = *str++;
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004753
Denys Vlasenko9e800222010-10-03 14:28:04 +02004754#if ENABLE_HUSH_BRACE_EXPANSION
4755 if (c == '{' || c == '}') {
4756 /* { -> \{, } -> \} */
4757 o_addchr(o, '\\');
Denys Vlasenko957f79f2010-10-03 17:15:50 +02004758 /* And now we want to add { or } and continue:
4759 * o_addchr(o, c);
4760 * continue;
4761 * luckily, just falling throught achieves this.
4762 */
Denys Vlasenko9e800222010-10-03 14:28:04 +02004763 }
4764#endif
4765 o_addchr(o, c);
4766 if (c == '\\') {
Denys Vlasenko238081f2010-10-03 14:26:26 +02004767 /* \z -> \\\z; \<eol> -> \\<eol> */
4768 o_addchr(o, '\\');
4769 if (len) {
4770 len--;
4771 o_addchr(o, '\\');
4772 o_addchr(o, *str++);
4773 }
4774 }
4775 }
4776}
4777
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004778/* Store given string, finalizing the word and starting new one whenever
4779 * we encounter IFS char(s). This is used for expanding variable values.
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004780 * End-of-string does NOT finalize word: think about 'echo -$VAR-'.
4781 * Return in *ended_with_ifs:
4782 * 1 - ended with IFS char, else 0 (this includes case of empty str).
4783 */
4784static int expand_on_ifs(int *ended_with_ifs, o_string *output, int n, const char *str)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004785{
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004786 int last_is_ifs = 0;
4787
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004788 while (1) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004789 int word_len;
4790
4791 if (!*str) /* EOL - do not finalize word */
4792 break;
4793 word_len = strcspn(str, G.ifs);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004794 if (word_len) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004795 /* We have WORD_LEN leading non-IFS chars */
Denys Vlasenko238081f2010-10-03 14:26:26 +02004796 if (!(output->o_expflags & EXP_FLAG_GLOB)) {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004797 o_addblock(output, str, word_len);
Denys Vlasenko238081f2010-10-03 14:26:26 +02004798 } else {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004799 /* Protect backslashes against globbing up :)
Denys Vlasenkoa769e022010-09-10 10:12:34 +02004800 * Example: "v='\*'; echo b$v" prints "b\*"
4801 * (and does not try to glob on "*")
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004802 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004803 o_addblock_duplicate_backslash(output, str, word_len);
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02004804 /*/ Why can't we do it easier? */
4805 /*o_addblock(output, str, word_len); - WRONG: "v='\*'; echo Z$v" prints "Z*" instead of "Z\*" */
4806 /*o_addqblock(output, str, word_len); - WRONG: "v='*'; echo Z$v" prints "Z*" instead of Z* files */
4807 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004808 last_is_ifs = 0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004809 str += word_len;
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004810 if (!*str) /* EOL - do not finalize word */
4811 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004812 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004813
4814 /* We know str here points to at least one IFS char */
4815 last_is_ifs = 1;
4816 str += strspn(str, G.ifs); /* skip IFS chars */
4817 if (!*str) /* EOL - do not finalize word */
4818 break;
4819
4820 /* Start new word... but not always! */
4821 /* Case "v=' a'; echo ''$v": we do need to finalize empty word: */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004822 if (output->has_quoted_part
4823 /* Case "v=' a'; echo $v":
4824 * here nothing precedes the space in $v expansion,
4825 * therefore we should not finish the word
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004826 * (IOW: if there *is* word to finalize, only then do it):
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004827 */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004828 || (n > 0 && output->data[output->length - 1])
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004829 ) {
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02004830 o_addchr(output, '\0');
4831 debug_print_list("expand_on_ifs", output, n);
4832 n = o_save_ptr(output, n);
4833 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004834 }
Denys Vlasenko6e42b892011-08-01 18:16:43 +02004835
4836 if (ended_with_ifs)
4837 *ended_with_ifs = last_is_ifs;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004838 debug_print_list("expand_on_ifs[1]", output, n);
4839 return n;
4840}
4841
4842/* Helper to expand $((...)) and heredoc body. These act as if
4843 * they are in double quotes, with the exception that they are not :).
4844 * Just the rules are similar: "expand only $var and `cmd`"
4845 *
4846 * Returns malloced string.
4847 * As an optimization, we return NULL if expansion is not needed.
4848 */
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004849#if !ENABLE_HUSH_BASH_COMPAT
4850/* only ${var/pattern/repl} (its pattern part) needs additional mode */
4851#define encode_then_expand_string(str, process_bkslash, do_unbackslash) \
4852 encode_then_expand_string(str)
4853#endif
4854static char *encode_then_expand_string(const char *str, int process_bkslash, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004855{
4856 char *exp_str;
4857 struct in_str input;
4858 o_string dest = NULL_O_STRING;
4859
4860 if (!strchr(str, '$')
Denys Vlasenko77b32cc2010-09-06 11:27:32 +02004861 && !strchr(str, '\\')
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004862#if ENABLE_HUSH_TICK
4863 && !strchr(str, '`')
4864#endif
4865 ) {
4866 return NULL;
4867 }
4868
4869 /* We need to expand. Example:
4870 * echo $(($a + `echo 1`)) $((1 + $((2)) ))
4871 */
4872 setup_string_in_str(&input, str);
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004873 encode_string(NULL, &dest, &input, EOF, process_bkslash);
Denys Vlasenko3eab24e2011-03-24 05:25:59 +01004874//TODO: error check (encode_string returns 0 on error)?
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004875 //bb_error_msg("'%s' -> '%s'", str, dest.data);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02004876 exp_str = expand_string_to_string(dest.data, /*unbackslash:*/ do_unbackslash);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004877 //bb_error_msg("'%s' -> '%s'", dest.data, exp_str);
4878 o_free_unsafe(&dest);
4879 return exp_str;
4880}
4881
4882#if ENABLE_SH_MATH_SUPPORT
Denys Vlasenko063847d2010-09-15 13:33:02 +02004883static arith_t expand_and_evaluate_arith(const char *arg, const char **errmsg_p)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004884{
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004885 arith_state_t math_state;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004886 arith_t res;
4887 char *exp_str;
4888
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004889 math_state.lookupvar = get_local_var_value;
4890 math_state.setvar = set_local_var_from_halves;
4891 //math_state.endofname = endofname;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02004892 exp_str = encode_then_expand_string(arg, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenko06d44d72010-09-13 12:49:03 +02004893 res = arith(&math_state, exp_str ? exp_str : arg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004894 free(exp_str);
Denys Vlasenko063847d2010-09-15 13:33:02 +02004895 if (errmsg_p)
4896 *errmsg_p = math_state.errmsg;
4897 if (math_state.errmsg)
4898 die_if_script(math_state.errmsg);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004899 return res;
4900}
4901#endif
4902
4903#if ENABLE_HUSH_BASH_COMPAT
4904/* ${var/[/]pattern[/repl]} helpers */
4905static char *strstr_pattern(char *val, const char *pattern, int *size)
4906{
4907 while (1) {
4908 char *end = scan_and_match(val, pattern, SCAN_MOVE_FROM_RIGHT + SCAN_MATCH_LEFT_HALF);
4909 debug_printf_varexp("val:'%s' pattern:'%s' end:'%s'\n", val, pattern, end);
4910 if (end) {
4911 *size = end - val;
4912 return val;
4913 }
4914 if (*val == '\0')
4915 return NULL;
4916 /* Optimization: if "*pat" did not match the start of "string",
4917 * we know that "tring", "ring" etc will not match too:
4918 */
4919 if (pattern[0] == '*')
4920 return NULL;
4921 val++;
4922 }
4923}
4924static char *replace_pattern(char *val, const char *pattern, const char *repl, char exp_op)
4925{
4926 char *result = NULL;
4927 unsigned res_len = 0;
4928 unsigned repl_len = strlen(repl);
4929
4930 while (1) {
4931 int size;
4932 char *s = strstr_pattern(val, pattern, &size);
4933 if (!s)
4934 break;
4935
4936 result = xrealloc(result, res_len + (s - val) + repl_len + 1);
4937 memcpy(result + res_len, val, s - val);
4938 res_len += s - val;
4939 strcpy(result + res_len, repl);
4940 res_len += repl_len;
4941 debug_printf_varexp("val:'%s' s:'%s' result:'%s'\n", val, s, result);
4942
4943 val = s + size;
4944 if (exp_op == '/')
4945 break;
4946 }
4947 if (val[0] && result) {
4948 result = xrealloc(result, res_len + strlen(val) + 1);
4949 strcpy(result + res_len, val);
4950 debug_printf_varexp("val:'%s' result:'%s'\n", val, result);
4951 }
4952 debug_printf_varexp("result:'%s'\n", result);
4953 return result;
4954}
4955#endif
4956
4957/* Helper:
4958 * Handles <SPECIAL_VAR_SYMBOL>varname...<SPECIAL_VAR_SYMBOL> construct.
4959 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004960static NOINLINE const char *expand_one_var(char **to_be_freed_pp, char *arg, char **pp)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004961{
4962 const char *val = NULL;
4963 char *to_be_freed = NULL;
4964 char *p = *pp;
4965 char *var;
4966 char first_char;
4967 char exp_op;
4968 char exp_save = exp_save; /* for compiler */
4969 char *exp_saveptr; /* points to expansion operator */
4970 char *exp_word = exp_word; /* for compiler */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004971 char arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004972
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004973 *p = '\0'; /* replace trailing SPECIAL_VAR_SYMBOL */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004974 var = arg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004975 exp_saveptr = arg[1] ? strchr(VAR_ENCODED_SUBST_OPS, arg[1]) : NULL;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02004976 arg0 = arg[0];
4977 first_char = arg[0] = arg0 & 0x7f;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004978 exp_op = 0;
4979
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004980 if (first_char == '#' /* ${#... */
4981 && arg[1] && !exp_saveptr /* not ${#} and not ${#<op_char>...} */
4982 ) {
4983 /* It must be length operator: ${#var} */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004984 var++;
4985 exp_op = 'L';
4986 } else {
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02004987 /* Maybe handle parameter expansion */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02004988 if (exp_saveptr /* if 2nd char is one of expansion operators */
4989 && strchr(NUMERIC_SPECVARS_STR, first_char) /* 1st char is special variable */
4990 ) {
4991 /* ${?:0}, ${#[:]%0} etc */
4992 exp_saveptr = var + 1;
4993 } else {
4994 /* ${?}, ${var}, ${var:0}, ${var[:]%0} etc */
4995 exp_saveptr = var+1 + strcspn(var+1, VAR_ENCODED_SUBST_OPS);
4996 }
4997 exp_op = exp_save = *exp_saveptr;
4998 if (exp_op) {
4999 exp_word = exp_saveptr + 1;
5000 if (exp_op == ':') {
5001 exp_op = *exp_word++;
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005002//TODO: try ${var:} and ${var:bogus} in non-bash config
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005003 if (ENABLE_HUSH_BASH_COMPAT
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005004 && (!exp_op || !strchr(MINUS_PLUS_EQUAL_QUESTION, exp_op))
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005005 ) {
5006 /* oops... it's ${var:N[:M]}, not ${var:?xxx} or some such */
5007 exp_op = ':';
5008 exp_word--;
5009 }
5010 }
5011 *exp_saveptr = '\0';
5012 } /* else: it's not an expansion op, but bare ${var} */
5013 }
5014
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005015 /* Look up the variable in question */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005016 if (isdigit(var[0])) {
Denys Vlasenko77a7b552010-09-09 12:40:03 +02005017 /* parse_dollar should have vetted var for us */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005018 int n = xatoi_positive(var);
5019 if (n < G.global_argc)
5020 val = G.global_argv[n];
5021 /* else val remains NULL: $N with too big N */
5022 } else {
5023 switch (var[0]) {
5024 case '$': /* pid */
5025 val = utoa(G.root_pid);
5026 break;
5027 case '!': /* bg pid */
5028 val = G.last_bg_pid ? utoa(G.last_bg_pid) : "";
5029 break;
5030 case '?': /* exitcode */
5031 val = utoa(G.last_exitcode);
5032 break;
5033 case '#': /* argc */
5034 val = utoa(G.global_argc ? G.global_argc-1 : 0);
5035 break;
5036 default:
5037 val = get_local_var_value(var);
5038 }
5039 }
5040
5041 /* Handle any expansions */
5042 if (exp_op == 'L') {
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02005043 reinit_unicode_for_hush();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005044 debug_printf_expand("expand: length(%s)=", val);
Denys Vlasenkoc538d5b2014-08-13 09:57:44 +02005045 val = utoa(val ? unicode_strlen(val) : 0);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005046 debug_printf_expand("%s\n", val);
5047 } else if (exp_op) {
5048 if (exp_op == '%' || exp_op == '#') {
5049 /* Standard-mandated substring removal ops:
5050 * ${parameter%word} - remove smallest suffix pattern
5051 * ${parameter%%word} - remove largest suffix pattern
5052 * ${parameter#word} - remove smallest prefix pattern
5053 * ${parameter##word} - remove largest prefix pattern
5054 *
5055 * Word is expanded to produce a glob pattern.
5056 * Then var's value is matched to it and matching part removed.
5057 */
5058 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005059 char *t;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005060 char *exp_exp_word;
5061 char *loc;
5062 unsigned scan_flags = pick_scan(exp_op, *exp_word);
Denys Vlasenkoe4dcba12010-10-28 18:57:19 +02005063 if (exp_op == *exp_word) /* ## or %% */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005064 exp_word++;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005065 exp_exp_word = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005066 if (exp_exp_word)
5067 exp_word = exp_exp_word;
Denys Vlasenko4f870492010-09-10 11:06:01 +02005068 /* HACK ALERT. We depend here on the fact that
5069 * G.global_argv and results of utoa and get_local_var_value
5070 * are actually in writable memory:
5071 * scan_and_match momentarily stores NULs there. */
5072 t = (char*)val;
5073 loc = scan_and_match(t, exp_word, scan_flags);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005074 //bb_error_msg("op:%c str:'%s' pat:'%s' res:'%s'",
Denys Vlasenko4f870492010-09-10 11:06:01 +02005075 // exp_op, t, exp_word, loc);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005076 free(exp_exp_word);
5077 if (loc) { /* match was found */
5078 if (scan_flags & SCAN_MATCH_LEFT_HALF) /* #[#] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005079 val = loc; /* take right part */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005080 else /* %[%] */
Denys Vlasenko4f870492010-09-10 11:06:01 +02005081 val = to_be_freed = xstrndup(val, loc - val); /* left */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005082 }
5083 }
5084 }
5085#if ENABLE_HUSH_BASH_COMPAT
5086 else if (exp_op == '/' || exp_op == '\\') {
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005087 /* It's ${var/[/]pattern[/repl]} thing.
5088 * Note that in encoded form it has TWO parts:
5089 * var/pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenko4f870492010-09-10 11:06:01 +02005090 * and if // is used, it is encoded as \:
5091 * var\pattern<SPECIAL_VAR_SYMBOL>repl<SPECIAL_VAR_SYMBOL>
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005092 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005093 /* Empty variable always gives nothing: */
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005094 // "v=''; echo ${v/*/w}" prints "", not "w"
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005095 if (val && val[0]) {
Denys Vlasenko4f870492010-09-10 11:06:01 +02005096 /* pattern uses non-standard expansion.
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005097 * repl should be unbackslashed and globbed
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005098 * by the usual expansion rules:
5099 * >az; >bz;
5100 * v='a bz'; echo "${v/a*z/a*z}" prints "a*z"
5101 * v='a bz'; echo "${v/a*z/\z}" prints "\z"
5102 * v='a bz'; echo ${v/a*z/a*z} prints "az"
5103 * v='a bz'; echo ${v/a*z/\z} prints "z"
5104 * (note that a*z _pattern_ is never globbed!)
5105 */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005106 char *pattern, *repl, *t;
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005107 pattern = encode_then_expand_string(exp_word, /*process_bkslash:*/ 0, /*unbackslash:*/ 0);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005108 if (!pattern)
5109 pattern = xstrdup(exp_word);
5110 debug_printf_varexp("pattern:'%s'->'%s'\n", exp_word, pattern);
5111 *p++ = SPECIAL_VAR_SYMBOL;
5112 exp_word = p;
5113 p = strchr(p, SPECIAL_VAR_SYMBOL);
5114 *p = '\0';
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005115 repl = encode_then_expand_string(exp_word, /*process_bkslash:*/ arg0 & 0x80, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005116 debug_printf_varexp("repl:'%s'->'%s'\n", exp_word, repl);
5117 /* HACK ALERT. We depend here on the fact that
5118 * G.global_argv and results of utoa and get_local_var_value
5119 * are actually in writable memory:
5120 * replace_pattern momentarily stores NULs there. */
5121 t = (char*)val;
5122 to_be_freed = replace_pattern(t,
5123 pattern,
5124 (repl ? repl : exp_word),
5125 exp_op);
5126 if (to_be_freed) /* at least one replace happened */
5127 val = to_be_freed;
5128 free(pattern);
5129 free(repl);
5130 }
5131 }
5132#endif
5133 else if (exp_op == ':') {
5134#if ENABLE_HUSH_BASH_COMPAT && ENABLE_SH_MATH_SUPPORT
5135 /* It's ${var:N[:M]} bashism.
5136 * Note that in encoded form it has TWO parts:
5137 * var:N<SPECIAL_VAR_SYMBOL>M<SPECIAL_VAR_SYMBOL>
5138 */
5139 arith_t beg, len;
Denys Vlasenko063847d2010-09-15 13:33:02 +02005140 const char *errmsg;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005141
Denys Vlasenko063847d2010-09-15 13:33:02 +02005142 beg = expand_and_evaluate_arith(exp_word, &errmsg);
5143 if (errmsg)
5144 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005145 debug_printf_varexp("beg:'%s'=%lld\n", exp_word, (long long)beg);
5146 *p++ = SPECIAL_VAR_SYMBOL;
5147 exp_word = p;
5148 p = strchr(p, SPECIAL_VAR_SYMBOL);
5149 *p = '\0';
Denys Vlasenko063847d2010-09-15 13:33:02 +02005150 len = expand_and_evaluate_arith(exp_word, &errmsg);
5151 if (errmsg)
5152 goto arith_err;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005153 debug_printf_varexp("len:'%s'=%lld\n", exp_word, (long long)len);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005154 if (len >= 0) { /* bash compat: len < 0 is illegal */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005155 if (beg < 0) /* bash compat */
5156 beg = 0;
5157 debug_printf_varexp("from val:'%s'\n", val);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005158 if (len == 0 || !val || beg >= strlen(val)) {
Denys Vlasenko063847d2010-09-15 13:33:02 +02005159 arith_err:
Denys Vlasenkob771c652010-09-13 00:34:26 +02005160 val = NULL;
5161 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005162 /* Paranoia. What if user entered 9999999999999
5163 * which fits in arith_t but not int? */
5164 if (len >= INT_MAX)
5165 len = INT_MAX;
5166 val = to_be_freed = xstrndup(val + beg, len);
5167 }
5168 debug_printf_varexp("val:'%s'\n", val);
5169 } else
5170#endif
5171 {
5172 die_if_script("malformed ${%s:...}", var);
Denys Vlasenkob771c652010-09-13 00:34:26 +02005173 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005174 }
5175 } else { /* one of "-=+?" */
5176 /* Standard-mandated substitution ops:
5177 * ${var?word} - indicate error if unset
5178 * If var is unset, word (or a message indicating it is unset
5179 * if word is null) is written to standard error
5180 * and the shell exits with a non-zero exit status.
5181 * Otherwise, the value of var is substituted.
5182 * ${var-word} - use default value
5183 * If var is unset, word is substituted.
5184 * ${var=word} - assign and use default value
5185 * If var is unset, word is assigned to var.
5186 * In all cases, final value of var is substituted.
5187 * ${var+word} - use alternative value
5188 * If var is unset, null is substituted.
5189 * Otherwise, word is substituted.
5190 *
5191 * Word is subjected to tilde expansion, parameter expansion,
5192 * command substitution, and arithmetic expansion.
5193 * If word is not needed, it is not expanded.
5194 *
5195 * Colon forms (${var:-word}, ${var:=word} etc) do the same,
5196 * but also treat null var as if it is unset.
5197 */
5198 int use_word = (!val || ((exp_save == ':') && !val[0]));
5199 if (exp_op == '+')
5200 use_word = !use_word;
5201 debug_printf_expand("expand: op:%c (null:%s) test:%i\n", exp_op,
5202 (exp_save == ':') ? "true" : "false", use_word);
5203 if (use_word) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005204 to_be_freed = encode_then_expand_string(exp_word, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005205 if (to_be_freed)
5206 exp_word = to_be_freed;
5207 if (exp_op == '?') {
5208 /* mimic bash message */
5209 die_if_script("%s: %s",
5210 var,
5211 exp_word[0] ? exp_word : "parameter null or not set"
5212 );
5213//TODO: how interactive bash aborts expansion mid-command?
5214 } else {
5215 val = exp_word;
5216 }
5217
5218 if (exp_op == '=') {
5219 /* ${var=[word]} or ${var:=[word]} */
5220 if (isdigit(var[0]) || var[0] == '#') {
5221 /* mimic bash message */
5222 die_if_script("$%s: cannot assign in this way", var);
5223 val = NULL;
5224 } else {
5225 char *new_var = xasprintf("%s=%s", var, val);
5226 set_local_var(new_var, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
5227 }
5228 }
5229 }
5230 } /* one of "-=+?" */
5231
5232 *exp_saveptr = exp_save;
5233 } /* if (exp_op) */
5234
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005235 arg[0] = arg0;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005236
5237 *pp = p;
5238 *to_be_freed_pp = to_be_freed;
5239 return val;
5240}
5241
5242/* Expand all variable references in given string, adding words to list[]
5243 * at n, n+1,... positions. Return updated n (so that list[n] is next one
5244 * to be filled). This routine is extremely tricky: has to deal with
5245 * variables/parameters with whitespace, $* and $@, and constructs like
5246 * 'echo -$*-'. If you play here, you must run testsuite afterwards! */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005247static NOINLINE int expand_vars_to_list(o_string *output, int n, char *arg)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005248{
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005249 /* output->o_expflags & EXP_FLAG_SINGLEWORD (0x80) if we are in
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005250 * expansion of right-hand side of assignment == 1-element expand.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005251 */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005252 char cant_be_null = 0; /* only bit 0x80 matters */
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005253 int ended_in_ifs = 0; /* did last unquoted expansion end with IFS chars? */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005254 char *p;
5255
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005256 debug_printf_expand("expand_vars_to_list: arg:'%s' singleword:%x\n", arg,
5257 !!(output->o_expflags & EXP_FLAG_SINGLEWORD));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005258 debug_print_list("expand_vars_to_list", output, n);
5259 n = o_save_ptr(output, n);
5260 debug_print_list("expand_vars_to_list[0]", output, n);
5261
5262 while ((p = strchr(arg, SPECIAL_VAR_SYMBOL)) != NULL) {
5263 char first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005264 char *to_be_freed = NULL;
5265 const char *val = NULL;
5266#if ENABLE_HUSH_TICK
5267 o_string subst_result = NULL_O_STRING;
5268#endif
5269#if ENABLE_SH_MATH_SUPPORT
5270 char arith_buf[sizeof(arith_t)*3 + 2];
5271#endif
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005272
5273 if (ended_in_ifs) {
5274 o_addchr(output, '\0');
5275 n = o_save_ptr(output, n);
5276 ended_in_ifs = 0;
5277 }
5278
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005279 o_addblock(output, arg, p - arg);
5280 debug_print_list("expand_vars_to_list[1]", output, n);
5281 arg = ++p;
5282 p = strchr(p, SPECIAL_VAR_SYMBOL);
5283
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005284 /* Fetch special var name (if it is indeed one of them)
5285 * and quote bit, force the bit on if singleword expansion -
5286 * important for not getting v=$@ expand to many words. */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005287 first_ch = arg[0] | (output->o_expflags & EXP_FLAG_SINGLEWORD);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005288
5289 /* Is this variable quoted and thus expansion can't be null?
5290 * "$@" is special. Even if quoted, it can still
5291 * expand to nothing (not even an empty string),
5292 * thus it is excluded. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005293 if ((first_ch & 0x7f) != '@')
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005294 cant_be_null |= first_ch;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005295
5296 switch (first_ch & 0x7f) {
5297 /* Highest bit in first_ch indicates that var is double-quoted */
5298 case '*':
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005299 case '@': {
5300 int i;
5301 if (!G.global_argv[1])
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005302 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005303 i = 1;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005304 cant_be_null |= first_ch; /* do it for "$@" _now_, when we know it's not empty */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005305 if (!(first_ch & 0x80)) { /* unquoted $* or $@ */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005306 while (G.global_argv[i]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005307 n = expand_on_ifs(NULL, output, n, G.global_argv[i]);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005308 debug_printf_expand("expand_vars_to_list: argv %d (last %d)\n", i, G.global_argc - 1);
5309 if (G.global_argv[i++][0] && G.global_argv[i]) {
5310 /* this argv[] is not empty and not last:
5311 * put terminating NUL, start new word */
5312 o_addchr(output, '\0');
5313 debug_print_list("expand_vars_to_list[2]", output, n);
5314 n = o_save_ptr(output, n);
5315 debug_print_list("expand_vars_to_list[3]", output, n);
5316 }
5317 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005318 } else
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005319 /* If EXP_FLAG_SINGLEWORD, we handle assignment 'a=....$@.....'
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005320 * and in this case should treat it like '$*' - see 'else...' below */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005321 if (first_ch == ('@'|0x80) /* quoted $@ */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005322 && !(output->o_expflags & EXP_FLAG_SINGLEWORD) /* not v="$@" case */
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005323 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005324 while (1) {
5325 o_addQstr(output, G.global_argv[i]);
5326 if (++i >= G.global_argc)
5327 break;
5328 o_addchr(output, '\0');
5329 debug_print_list("expand_vars_to_list[4]", output, n);
5330 n = o_save_ptr(output, n);
5331 }
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005332 } else { /* quoted $* (or v="$@" case): add as one word */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005333 while (1) {
5334 o_addQstr(output, G.global_argv[i]);
5335 if (!G.global_argv[++i])
5336 break;
5337 if (G.ifs[0])
5338 o_addchr(output, G.ifs[0]);
5339 }
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005340 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005341 }
5342 break;
Denys Vlasenkoc49d2d92010-09-06 10:26:37 +02005343 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005344 case SPECIAL_VAR_SYMBOL: /* <SPECIAL_VAR_SYMBOL><SPECIAL_VAR_SYMBOL> */
5345 /* "Empty variable", used to make "" etc to not disappear */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005346 output->has_quoted_part = 1;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005347 arg++;
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005348 cant_be_null = 0x80;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005349 break;
5350#if ENABLE_HUSH_TICK
5351 case '`': /* <SPECIAL_VAR_SYMBOL>`cmd<SPECIAL_VAR_SYMBOL> */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005352 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005353 arg++;
5354 /* Can't just stuff it into output o_string,
5355 * expanded result may need to be globbed
5356 * and $IFS-splitted */
5357 debug_printf_subst("SUBST '%s' first_ch %x\n", arg, first_ch);
5358 G.last_exitcode = process_command_subs(&subst_result, arg);
5359 debug_printf_subst("SUBST RES:%d '%s'\n", G.last_exitcode, subst_result.data);
5360 val = subst_result.data;
5361 goto store_val;
5362#endif
5363#if ENABLE_SH_MATH_SUPPORT
5364 case '+': { /* <SPECIAL_VAR_SYMBOL>+cmd<SPECIAL_VAR_SYMBOL> */
5365 arith_t res;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005366
5367 arg++; /* skip '+' */
5368 *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
5369 debug_printf_subst("ARITH '%s' first_ch %x\n", arg, first_ch);
Denys Vlasenko063847d2010-09-15 13:33:02 +02005370 res = expand_and_evaluate_arith(arg, NULL);
Denys Vlasenkobed7c812010-09-16 11:50:46 +02005371 debug_printf_subst("ARITH RES '"ARITH_FMT"'\n", res);
5372 sprintf(arith_buf, ARITH_FMT, res);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005373 val = arith_buf;
5374 break;
5375 }
5376#endif
5377 default:
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005378 val = expand_one_var(&to_be_freed, arg, &p);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005379 IF_HUSH_TICK(store_val:)
5380 if (!(first_ch & 0x80)) { /* unquoted $VAR */
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005381 debug_printf_expand("unquoted '%s', output->o_escape:%d\n", val,
5382 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005383 if (val && val[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005384 n = expand_on_ifs(&ended_in_ifs, output, n, val);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005385 val = NULL;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005386 }
5387 } else { /* quoted $VAR, val will be appended below */
Denys Vlasenko4fb53fb2011-08-01 14:06:20 +02005388 output->has_quoted_part = 1;
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005389 debug_printf_expand("quoted '%s', output->o_escape:%d\n", val,
5390 !!(output->o_expflags & EXP_FLAG_ESC_GLOB_CHARS));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005391 }
5392 break;
5393
5394 } /* switch (char after <SPECIAL_VAR_SYMBOL>) */
5395
5396 if (val && val[0]) {
5397 o_addQstr(output, val);
5398 }
5399 free(to_be_freed);
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005400
5401 /* Restore NULL'ed SPECIAL_VAR_SYMBOL.
5402 * Do the check to avoid writing to a const string. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005403 if (*p != SPECIAL_VAR_SYMBOL)
5404 *p = SPECIAL_VAR_SYMBOL;
5405
5406#if ENABLE_HUSH_TICK
5407 o_free(&subst_result);
5408#endif
5409 arg = ++p;
5410 } /* end of "while (SPECIAL_VAR_SYMBOL is found) ..." */
5411
5412 if (arg[0]) {
Denys Vlasenko6e42b892011-08-01 18:16:43 +02005413 if (ended_in_ifs) {
5414 o_addchr(output, '\0');
5415 n = o_save_ptr(output, n);
5416 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005417 debug_print_list("expand_vars_to_list[a]", output, n);
5418 /* this part is literal, and it was already pre-quoted
5419 * if needed (much earlier), do not use o_addQstr here! */
5420 o_addstr_with_NUL(output, arg);
5421 debug_print_list("expand_vars_to_list[b]", output, n);
5422 } else if (output->length == o_get_last_ptr(output, n) /* expansion is empty */
Denys Vlasenkobfc02a72010-09-09 14:38:46 +02005423 && !(cant_be_null & 0x80) /* and all vars were not quoted. */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005424 ) {
5425 n--;
5426 /* allow to reuse list[n] later without re-growth */
5427 output->has_empty_slot = 1;
5428 } else {
5429 o_addchr(output, '\0');
5430 }
5431
5432 return n;
5433}
5434
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005435static char **expand_variables(char **argv, unsigned expflags)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005436{
5437 int n;
5438 char **list;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005439 o_string output = NULL_O_STRING;
5440
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005441 output.o_expflags = expflags;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005442
5443 n = 0;
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005444 while (*argv) {
Denys Vlasenko95d48f22010-09-08 13:58:55 +02005445 n = expand_vars_to_list(&output, n, *argv);
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02005446 argv++;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005447 }
5448 debug_print_list("expand_variables", &output, n);
5449
5450 /* output.data (malloced in one block) gets returned in "list" */
5451 list = o_finalize_list(&output, n);
5452 debug_print_strings("expand_variables[1]", list);
5453 return list;
5454}
5455
5456static char **expand_strvec_to_strvec(char **argv)
5457{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005458 return expand_variables(argv, EXP_FLAG_GLOB | EXP_FLAG_ESC_GLOB_CHARS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005459}
5460
5461#if ENABLE_HUSH_BASH_COMPAT
5462static char **expand_strvec_to_strvec_singleword_noglob(char **argv)
5463{
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005464 return expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005465}
5466#endif
5467
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02005468/* Used for expansion of right hand of assignments,
5469 * $((...)), heredocs, variable espansion parts.
5470 *
5471 * NB: should NOT do globbing!
5472 * "export v=/bin/c*; env | grep ^v=" outputs "v=/bin/c*"
5473 */
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005474static char *expand_string_to_string(const char *str, int do_unbackslash)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005475{
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005476#if !ENABLE_HUSH_BASH_COMPAT
5477 const int do_unbackslash = 1;
5478#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005479 char *argv[2], **list;
5480
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005481 debug_printf_expand("string_to_string<='%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005482 /* This is generally an optimization, but it also
5483 * handles "", which otherwise trips over !list[0] check below.
5484 * (is this ever happens that we actually get str="" here?)
5485 */
5486 if (!strchr(str, SPECIAL_VAR_SYMBOL) && !strchr(str, '\\')) {
5487 //TODO: Can use on strings with \ too, just unbackslash() them?
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005488 debug_printf_expand("string_to_string(fast)=>'%s'\n", str);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005489 return xstrdup(str);
5490 }
5491
5492 argv[0] = (char*)str;
5493 argv[1] = NULL;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005494 list = expand_variables(argv, do_unbackslash
5495 ? EXP_FLAG_ESC_GLOB_CHARS | EXP_FLAG_SINGLEWORD
5496 : EXP_FLAG_SINGLEWORD
5497 );
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005498 if (HUSH_DEBUG)
5499 if (!list[0] || list[1])
5500 bb_error_msg_and_die("BUG in varexp2");
5501 /* actually, just move string 2*sizeof(char*) bytes back */
5502 overlapping_strcpy((char*)list, list[0]);
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005503 if (do_unbackslash)
5504 unbackslash((char*)list);
5505 debug_printf_expand("string_to_string=>'%s'\n", (char*)list);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005506 return (char*)list;
5507}
5508
5509/* Used for "eval" builtin */
5510static char* expand_strvec_to_string(char **argv)
5511{
5512 char **list;
5513
Denys Vlasenko5b686cb2010-09-08 13:44:34 +02005514 list = expand_variables(argv, EXP_FLAG_SINGLEWORD);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005515 /* Convert all NULs to spaces */
5516 if (list[0]) {
5517 int n = 1;
5518 while (list[n]) {
5519 if (HUSH_DEBUG)
5520 if (list[n-1] + strlen(list[n-1]) + 1 != list[n])
5521 bb_error_msg_and_die("BUG in varexp3");
5522 /* bash uses ' ' regardless of $IFS contents */
5523 list[n][-1] = ' ';
5524 n++;
5525 }
5526 }
5527 overlapping_strcpy((char*)list, list[0]);
5528 debug_printf_expand("strvec_to_string='%s'\n", (char*)list);
5529 return (char*)list;
5530}
5531
5532static char **expand_assignments(char **argv, int count)
5533{
5534 int i;
5535 char **p;
5536
5537 G.expanded_assignments = p = NULL;
5538 /* Expand assignments into one string each */
5539 for (i = 0; i < count; i++) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02005540 G.expanded_assignments = p = add_string_to_strings(p, expand_string_to_string(argv[i], /*unbackslash:*/ 1));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005541 }
5542 G.expanded_assignments = NULL;
5543 return p;
5544}
5545
5546
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005547static void switch_off_special_sigs(unsigned mask)
5548{
5549 unsigned sig = 0;
5550 while ((mask >>= 1) != 0) {
5551 sig++;
5552 if (!(mask & 1))
5553 continue;
5554 if (G.traps) {
5555 if (G.traps[sig] && !G.traps[sig][0])
5556 /* trap is '', has to remain SIG_IGN */
5557 continue;
5558 free(G.traps[sig]);
5559 G.traps[sig] = NULL;
5560 }
5561 /* We are here only if no trap or trap was not '' */
Denys Vlasenko0806e402011-05-12 23:06:20 +02005562 install_sighandler(sig, SIG_DFL);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005563 }
5564}
5565
Denys Vlasenkob347df92011-08-09 22:49:15 +02005566#if BB_MMU
5567/* never called */
5568void re_execute_shell(char ***to_free, const char *s,
5569 char *g_argv0, char **g_argv,
5570 char **builtin_argv) NORETURN;
5571
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005572static void reset_traps_to_defaults(void)
5573{
5574 /* This function is always called in a child shell
5575 * after fork (not vfork, NOMMU doesn't use this function).
5576 */
5577 unsigned sig;
5578 unsigned mask;
5579
5580 /* Child shells are not interactive.
5581 * SIGTTIN/SIGTTOU/SIGTSTP should not have special handling.
5582 * Testcase: (while :; do :; done) + ^Z should background.
5583 * Same goes for SIGTERM, SIGHUP, SIGINT.
5584 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005585 mask = (G.special_sig_mask & SPECIAL_INTERACTIVE_SIGS) | G_fatal_sig_mask;
5586 if (!G.traps && !mask)
5587 return; /* already no traps and no special sigs */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005588
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005589 /* Switch off special sigs */
5590 switch_off_special_sigs(mask);
5591#if ENABLE_HUSH_JOB
5592 G_fatal_sig_mask = 0;
5593#endif
Denys Vlasenko10c01312011-05-11 11:49:21 +02005594 G.special_sig_mask &= ~SPECIAL_INTERACTIVE_SIGS;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02005595 /* SIGQUIT,SIGCHLD and maybe SPECIAL_JOBSTOP_SIGS
5596 * remain set in G.special_sig_mask */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005597
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005598 if (!G.traps)
5599 return;
5600
5601 /* Reset all sigs to default except ones with empty traps */
5602 for (sig = 0; sig < NSIG; sig++) {
5603 if (!G.traps[sig])
5604 continue; /* no trap: nothing to do */
5605 if (!G.traps[sig][0])
5606 continue; /* empty trap: has to remain SIG_IGN */
5607 /* sig has non-empty trap, reset it: */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005608 free(G.traps[sig]);
5609 G.traps[sig] = NULL;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02005610 /* There is no signal for trap 0 (EXIT) */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005611 if (sig == 0)
5612 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02005613 install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005614 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005615}
5616
5617#else /* !BB_MMU */
5618
5619static void re_execute_shell(char ***to_free, const char *s,
5620 char *g_argv0, char **g_argv,
5621 char **builtin_argv) NORETURN;
5622static void re_execute_shell(char ***to_free, const char *s,
5623 char *g_argv0, char **g_argv,
5624 char **builtin_argv)
5625{
5626# define NOMMU_HACK_FMT ("-$%x:%x:%x:%x:%x:%llx" IF_HUSH_LOOPS(":%x"))
5627 /* delims + 2 * (number of bytes in printed hex numbers) */
5628 char param_buf[sizeof(NOMMU_HACK_FMT) + 2 * (sizeof(int)*6 + sizeof(long long)*1)];
5629 char *heredoc_argv[4];
5630 struct variable *cur;
5631# if ENABLE_HUSH_FUNCTIONS
5632 struct function *funcp;
5633# endif
5634 char **argv, **pp;
5635 unsigned cnt;
5636 unsigned long long empty_trap_mask;
5637
5638 if (!g_argv0) { /* heredoc */
5639 argv = heredoc_argv;
5640 argv[0] = (char *) G.argv0_for_re_execing;
5641 argv[1] = (char *) "-<";
5642 argv[2] = (char *) s;
5643 argv[3] = NULL;
5644 pp = &argv[3]; /* used as pointer to empty environment */
5645 goto do_exec;
5646 }
5647
5648 cnt = 0;
5649 pp = builtin_argv;
5650 if (pp) while (*pp++)
5651 cnt++;
5652
5653 empty_trap_mask = 0;
5654 if (G.traps) {
5655 int sig;
5656 for (sig = 1; sig < NSIG; sig++) {
5657 if (G.traps[sig] && !G.traps[sig][0])
5658 empty_trap_mask |= 1LL << sig;
5659 }
5660 }
5661
5662 sprintf(param_buf, NOMMU_HACK_FMT
5663 , (unsigned) G.root_pid
5664 , (unsigned) G.root_ppid
5665 , (unsigned) G.last_bg_pid
5666 , (unsigned) G.last_exitcode
5667 , cnt
5668 , empty_trap_mask
5669 IF_HUSH_LOOPS(, G.depth_of_loop)
5670 );
5671# undef NOMMU_HACK_FMT
5672 /* 1:hush 2:-$<pid>:<pid>:<exitcode>:<etc...> <vars...> <funcs...>
5673 * 3:-c 4:<cmd> 5:<arg0> <argN...> 6:NULL
5674 */
5675 cnt += 6;
5676 for (cur = G.top_var; cur; cur = cur->next) {
5677 if (!cur->flg_export || cur->flg_read_only)
5678 cnt += 2;
5679 }
5680# if ENABLE_HUSH_FUNCTIONS
5681 for (funcp = G.top_func; funcp; funcp = funcp->next)
5682 cnt += 3;
5683# endif
5684 pp = g_argv;
5685 while (*pp++)
5686 cnt++;
5687 *to_free = argv = pp = xzalloc(sizeof(argv[0]) * cnt);
5688 *pp++ = (char *) G.argv0_for_re_execing;
5689 *pp++ = param_buf;
5690 for (cur = G.top_var; cur; cur = cur->next) {
5691 if (strcmp(cur->varstr, hush_version_str) == 0)
5692 continue;
5693 if (cur->flg_read_only) {
5694 *pp++ = (char *) "-R";
5695 *pp++ = cur->varstr;
5696 } else if (!cur->flg_export) {
5697 *pp++ = (char *) "-V";
5698 *pp++ = cur->varstr;
5699 }
5700 }
5701# if ENABLE_HUSH_FUNCTIONS
5702 for (funcp = G.top_func; funcp; funcp = funcp->next) {
5703 *pp++ = (char *) "-F";
5704 *pp++ = funcp->name;
5705 *pp++ = funcp->body_as_string;
5706 }
5707# endif
5708 /* We can pass activated traps here. Say, -Tnn:trap_string
5709 *
5710 * However, POSIX says that subshells reset signals with traps
5711 * to SIG_DFL.
5712 * I tested bash-3.2 and it not only does that with true subshells
5713 * of the form ( list ), but with any forked children shells.
5714 * I set trap "echo W" WINCH; and then tried:
5715 *
5716 * { echo 1; sleep 20; echo 2; } &
5717 * while true; do echo 1; sleep 20; echo 2; break; done &
5718 * true | { echo 1; sleep 20; echo 2; } | cat
5719 *
5720 * In all these cases sending SIGWINCH to the child shell
5721 * did not run the trap. If I add trap "echo V" WINCH;
5722 * _inside_ group (just before echo 1), it works.
5723 *
5724 * I conclude it means we don't need to pass active traps here.
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005725 */
5726 *pp++ = (char *) "-c";
5727 *pp++ = (char *) s;
5728 if (builtin_argv) {
5729 while (*++builtin_argv)
5730 *pp++ = *builtin_argv;
5731 *pp++ = (char *) "";
5732 }
5733 *pp++ = g_argv0;
5734 while (*g_argv)
5735 *pp++ = *g_argv++;
5736 /* *pp = NULL; - is already there */
5737 pp = environ;
5738
5739 do_exec:
5740 debug_printf_exec("re_execute_shell pid:%d cmd:'%s'\n", getpid(), s);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02005741 /* Don't propagate SIG_IGN to the child */
5742 if (SPECIAL_JOBSTOP_SIGS != 0)
5743 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005744 execve(bb_busybox_exec_path, argv, pp);
5745 /* Fallback. Useful for init=/bin/hush usage etc */
5746 if (argv[0][0] == '/')
5747 execve(argv[0], argv, pp);
5748 xfunc_error_retval = 127;
5749 bb_error_msg_and_die("can't re-execute the shell");
5750}
5751#endif /* !BB_MMU */
5752
5753
5754static int run_and_free_list(struct pipe *pi);
5755
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005756/* Executing from string: eval, sh -c '...'
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005757 * or from file: /etc/profile, . file, sh <script>, sh (intereactive)
5758 * end_trigger controls how often we stop parsing
5759 * NUL: parse all, execute, return
5760 * ';': parse till ';' or newline, execute, repeat till EOF
5761 */
5762static void parse_and_run_stream(struct in_str *inp, int end_trigger)
Eric Andersen25f27032001-04-26 23:22:31 +00005763{
Denys Vlasenko00243b02009-11-16 02:00:03 +01005764 /* Why we need empty flag?
5765 * An obscure corner case "false; ``; echo $?":
5766 * empty command in `` should still set $? to 0.
5767 * But we can't just set $? to 0 at the start,
5768 * this breaks "false; echo `echo $?`" case.
5769 */
5770 bool empty = 1;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005771 while (1) {
5772 struct pipe *pipe_list;
Denis Vlasenkof8d01d32008-06-14 17:13:20 +00005773
Denys Vlasenkoa1463192011-01-18 17:55:04 +01005774#if ENABLE_HUSH_INTERACTIVE
5775 if (end_trigger == ';')
5776 inp->promptmode = 0; /* PS1 */
5777#endif
Denis Vlasenko9aa7d6f2009-04-04 22:47:50 +00005778 pipe_list = parse_stream(NULL, inp, end_trigger);
Denys Vlasenkocecbc982011-03-30 18:54:52 +02005779 if (!pipe_list || pipe_list == ERR_PTR) { /* EOF/error */
5780 /* If we are in "big" script
5781 * (not in `cmd` or something similar)...
5782 */
5783 if (pipe_list == ERR_PTR && end_trigger == ';') {
5784 /* Discard cached input (rest of line) */
5785 int ch = inp->last_char;
5786 while (ch != EOF && ch != '\n') {
5787 //bb_error_msg("Discarded:'%c'", ch);
5788 ch = i_getch(inp);
5789 }
5790 /* Force prompt */
5791 inp->p = NULL;
5792 /* This stream isn't empty */
5793 empty = 0;
5794 continue;
5795 }
5796 if (!pipe_list && empty)
Denys Vlasenko00243b02009-11-16 02:00:03 +01005797 G.last_exitcode = 0;
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005798 break;
Denys Vlasenko00243b02009-11-16 02:00:03 +01005799 }
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005800 debug_print_tree(pipe_list, 0);
5801 debug_printf_exec("parse_and_run_stream: run_and_free_list\n");
5802 run_and_free_list(pipe_list);
Denys Vlasenko00243b02009-11-16 02:00:03 +01005803 empty = 0;
Denys Vlasenko68d5cb52011-03-24 02:50:03 +01005804#if ENABLE_HUSH_FUNCTIONS
5805 if (G.flag_return_in_progress == 1)
5806 break;
5807#endif
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005808 }
Eric Andersen25f27032001-04-26 23:22:31 +00005809}
5810
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005811static void parse_and_run_string(const char *s)
Eric Andersen25f27032001-04-26 23:22:31 +00005812{
5813 struct in_str input;
5814 setup_string_in_str(&input, s);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005815 parse_and_run_stream(&input, '\0');
Eric Andersen25f27032001-04-26 23:22:31 +00005816}
5817
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005818static void parse_and_run_file(FILE *f)
Eric Andersen25f27032001-04-26 23:22:31 +00005819{
Eric Andersen25f27032001-04-26 23:22:31 +00005820 struct in_str input;
5821 setup_file_in_str(&input, f);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00005822 parse_and_run_stream(&input, ';');
Eric Andersen25f27032001-04-26 23:22:31 +00005823}
5824
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005825#if ENABLE_HUSH_TICK
5826static FILE *generate_stream_from_string(const char *s, pid_t *pid_p)
5827{
5828 pid_t pid;
5829 int channel[2];
5830# if !BB_MMU
5831 char **to_free = NULL;
5832# endif
5833
5834 xpipe(channel);
5835 pid = BB_MMU ? xfork() : xvfork();
5836 if (pid == 0) { /* child */
5837 disable_restore_tty_pgrp_on_exit();
5838 /* Process substitution is not considered to be usual
5839 * 'command execution'.
5840 * SUSv3 says ctrl-Z should be ignored, ctrl-C should not.
5841 */
5842 bb_signals(0
5843 + (1 << SIGTSTP)
5844 + (1 << SIGTTIN)
5845 + (1 << SIGTTOU)
5846 , SIG_IGN);
5847 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
5848 close(channel[0]); /* NB: close _first_, then move fd! */
5849 xmove_fd(channel[1], 1);
5850 /* Prevent it from trying to handle ctrl-z etc */
5851 IF_HUSH_JOB(G.run_list_level = 1;)
5852 /* Awful hack for `trap` or $(trap).
5853 *
5854 * http://www.opengroup.org/onlinepubs/009695399/utilities/trap.html
5855 * contains an example where "trap" is executed in a subshell:
5856 *
5857 * save_traps=$(trap)
5858 * ...
5859 * eval "$save_traps"
5860 *
5861 * Standard does not say that "trap" in subshell shall print
5862 * parent shell's traps. It only says that its output
5863 * must have suitable form, but then, in the above example
5864 * (which is not supposed to be normative), it implies that.
5865 *
5866 * bash (and probably other shell) does implement it
5867 * (traps are reset to defaults, but "trap" still shows them),
5868 * but as a result, "trap" logic is hopelessly messed up:
5869 *
5870 * # trap
5871 * trap -- 'echo Ho' SIGWINCH <--- we have a handler
5872 * # (trap) <--- trap is in subshell - no output (correct, traps are reset)
5873 * # true | trap <--- trap is in subshell - no output (ditto)
5874 * # echo `true | trap` <--- in subshell - output (but traps are reset!)
5875 * trap -- 'echo Ho' SIGWINCH
5876 * # echo `(trap)` <--- in subshell in subshell - output
5877 * trap -- 'echo Ho' SIGWINCH
5878 * # echo `true | (trap)` <--- in subshell in subshell in subshell - output!
5879 * trap -- 'echo Ho' SIGWINCH
5880 *
5881 * The rules when to forget and when to not forget traps
5882 * get really complex and nonsensical.
5883 *
5884 * Our solution: ONLY bare $(trap) or `trap` is special.
5885 */
5886 s = skip_whitespace(s);
5887 if (strncmp(s, "trap", 4) == 0
5888 && skip_whitespace(s + 4)[0] == '\0'
5889 ) {
5890 static const char *const argv[] = { NULL, NULL };
5891 builtin_trap((char**)argv);
5892 exit(0); /* not _exit() - we need to fflush */
5893 }
5894# if BB_MMU
5895 reset_traps_to_defaults();
5896 parse_and_run_string(s);
5897 _exit(G.last_exitcode);
5898# else
5899 /* We re-execute after vfork on NOMMU. This makes this script safe:
5900 * yes "0123456789012345678901234567890" | dd bs=32 count=64k >BIG
5901 * huge=`cat BIG` # was blocking here forever
5902 * echo OK
5903 */
5904 re_execute_shell(&to_free,
5905 s,
5906 G.global_argv[0],
5907 G.global_argv + 1,
5908 NULL);
5909# endif
5910 }
5911
5912 /* parent */
5913 *pid_p = pid;
5914# if ENABLE_HUSH_FAST
5915 G.count_SIGCHLD++;
5916//bb_error_msg("[%d] fork in generate_stream_from_string:"
5917// " G.count_SIGCHLD:%d G.handled_SIGCHLD:%d",
5918// getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
5919# endif
5920 enable_restore_tty_pgrp_on_exit();
5921# if !BB_MMU
5922 free(to_free);
5923# endif
5924 close(channel[1]);
5925 close_on_exec_on(channel[0]);
5926 return xfdopen_for_read(channel[0]);
5927}
5928
5929/* Return code is exit status of the process that is run. */
5930static int process_command_subs(o_string *dest, const char *s)
5931{
5932 FILE *fp;
5933 struct in_str pipe_str;
5934 pid_t pid;
5935 int status, ch, eol_cnt;
5936
5937 fp = generate_stream_from_string(s, &pid);
5938
5939 /* Now send results of command back into original context */
5940 setup_file_in_str(&pipe_str, fp);
5941 eol_cnt = 0;
5942 while ((ch = i_getch(&pipe_str)) != EOF) {
5943 if (ch == '\n') {
5944 eol_cnt++;
5945 continue;
5946 }
5947 while (eol_cnt) {
5948 o_addchr(dest, '\n');
5949 eol_cnt--;
5950 }
5951 o_addQchr(dest, ch);
5952 }
5953
5954 debug_printf("done reading from `cmd` pipe, closing it\n");
5955 fclose(fp);
5956 /* We need to extract exitcode. Test case
5957 * "true; echo `sleep 1; false` $?"
5958 * should print 1 */
5959 safe_waitpid(pid, &status, 0);
5960 debug_printf("child exited. returning its exitcode:%d\n", WEXITSTATUS(status));
5961 return WEXITSTATUS(status);
5962}
5963#endif /* ENABLE_HUSH_TICK */
5964
5965
5966static void setup_heredoc(struct redir_struct *redir)
5967{
5968 struct fd_pair pair;
5969 pid_t pid;
5970 int len, written;
5971 /* the _body_ of heredoc (misleading field name) */
5972 const char *heredoc = redir->rd_filename;
5973 char *expanded;
5974#if !BB_MMU
5975 char **to_free;
5976#endif
5977
5978 expanded = NULL;
5979 if (!(redir->rd_dup & HEREDOC_QUOTED)) {
Denys Vlasenkod98e5c62010-09-10 10:44:23 +02005980 expanded = encode_then_expand_string(heredoc, /*process_bkslash:*/ 1, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02005981 if (expanded)
5982 heredoc = expanded;
5983 }
5984 len = strlen(heredoc);
5985
5986 close(redir->rd_fd); /* often saves dup2+close in xmove_fd */
5987 xpiped_pair(pair);
5988 xmove_fd(pair.rd, redir->rd_fd);
5989
5990 /* Try writing without forking. Newer kernels have
5991 * dynamically growing pipes. Must use non-blocking write! */
5992 ndelay_on(pair.wr);
5993 while (1) {
5994 written = write(pair.wr, heredoc, len);
5995 if (written <= 0)
5996 break;
5997 len -= written;
5998 if (len == 0) {
5999 close(pair.wr);
6000 free(expanded);
6001 return;
6002 }
6003 heredoc += written;
6004 }
6005 ndelay_off(pair.wr);
6006
6007 /* Okay, pipe buffer was not big enough */
6008 /* Note: we must not create a stray child (bastard? :)
6009 * for the unsuspecting parent process. Child creates a grandchild
6010 * and exits before parent execs the process which consumes heredoc
6011 * (that exec happens after we return from this function) */
6012#if !BB_MMU
6013 to_free = NULL;
6014#endif
6015 pid = xvfork();
6016 if (pid == 0) {
6017 /* child */
6018 disable_restore_tty_pgrp_on_exit();
6019 pid = BB_MMU ? xfork() : xvfork();
6020 if (pid != 0)
6021 _exit(0);
6022 /* grandchild */
6023 close(redir->rd_fd); /* read side of the pipe */
6024#if BB_MMU
6025 full_write(pair.wr, heredoc, len); /* may loop or block */
6026 _exit(0);
6027#else
6028 /* Delegate blocking writes to another process */
6029 xmove_fd(pair.wr, STDOUT_FILENO);
6030 re_execute_shell(&to_free, heredoc, NULL, NULL, NULL);
6031#endif
6032 }
6033 /* parent */
6034#if ENABLE_HUSH_FAST
6035 G.count_SIGCHLD++;
6036//bb_error_msg("[%d] fork in setup_heredoc: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6037#endif
6038 enable_restore_tty_pgrp_on_exit();
6039#if !BB_MMU
6040 free(to_free);
6041#endif
6042 close(pair.wr);
6043 free(expanded);
6044 wait(NULL); /* wait till child has died */
6045}
6046
6047/* squirrel != NULL means we squirrel away copies of stdin, stdout,
6048 * and stderr if they are redirected. */
6049static int setup_redirects(struct command *prog, int squirrel[])
6050{
6051 int openfd, mode;
6052 struct redir_struct *redir;
6053
6054 for (redir = prog->redirects; redir; redir = redir->next) {
6055 if (redir->rd_type == REDIRECT_HEREDOC2) {
6056 /* rd_fd<<HERE case */
6057 if (squirrel && redir->rd_fd < 3
6058 && squirrel[redir->rd_fd] < 0
6059 ) {
6060 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6061 }
6062 /* for REDIRECT_HEREDOC2, rd_filename holds _contents_
6063 * of the heredoc */
6064 debug_printf_parse("set heredoc '%s'\n",
6065 redir->rd_filename);
6066 setup_heredoc(redir);
6067 continue;
6068 }
6069
6070 if (redir->rd_dup == REDIRFD_TO_FILE) {
6071 /* rd_fd<*>file case (<*> is <,>,>>,<>) */
6072 char *p;
6073 if (redir->rd_filename == NULL) {
6074 /* Something went wrong in the parse.
6075 * Pretend it didn't happen */
6076 bb_error_msg("bug in redirect parse");
6077 continue;
6078 }
6079 mode = redir_table[redir->rd_type].mode;
Denys Vlasenkoebee4102010-09-10 10:17:53 +02006080 p = expand_string_to_string(redir->rd_filename, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006081 openfd = open_or_warn(p, mode);
6082 free(p);
6083 if (openfd < 0) {
6084 /* this could get lost if stderr has been redirected, but
6085 * bash and ash both lose it as well (though zsh doesn't!) */
6086//what the above comment tries to say?
6087 return 1;
6088 }
6089 } else {
6090 /* rd_fd<*>rd_dup or rd_fd<*>- cases */
6091 openfd = redir->rd_dup;
6092 }
6093
6094 if (openfd != redir->rd_fd) {
6095 if (squirrel && redir->rd_fd < 3
6096 && squirrel[redir->rd_fd] < 0
6097 ) {
6098 squirrel[redir->rd_fd] = dup(redir->rd_fd);
6099 }
6100 if (openfd == REDIRFD_CLOSE) {
6101 /* "n>-" means "close me" */
6102 close(redir->rd_fd);
6103 } else {
6104 xdup2(openfd, redir->rd_fd);
6105 if (redir->rd_dup == REDIRFD_TO_FILE)
6106 close(openfd);
6107 }
6108 }
6109 }
6110 return 0;
6111}
6112
6113static void restore_redirects(int squirrel[])
6114{
6115 int i, fd;
6116 for (i = 0; i < 3; i++) {
6117 fd = squirrel[i];
6118 if (fd != -1) {
6119 /* We simply die on error */
6120 xmove_fd(fd, i);
6121 }
6122 }
6123}
6124
6125static char *find_in_path(const char *arg)
6126{
6127 char *ret = NULL;
6128 const char *PATH = get_local_var_value("PATH");
6129
6130 if (!PATH)
6131 return NULL;
6132
6133 while (1) {
6134 const char *end = strchrnul(PATH, ':');
6135 int sz = end - PATH; /* must be int! */
6136
6137 free(ret);
6138 if (sz != 0) {
6139 ret = xasprintf("%.*s/%s", sz, PATH, arg);
6140 } else {
6141 /* We have xxx::yyyy in $PATH,
6142 * it means "use current dir" */
6143 ret = xstrdup(arg);
6144 }
6145 if (access(ret, F_OK) == 0)
6146 break;
6147
6148 if (*end == '\0') {
6149 free(ret);
6150 return NULL;
6151 }
6152 PATH = end + 1;
6153 }
6154
6155 return ret;
6156}
6157
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006158static const struct built_in_command *find_builtin_helper(const char *name,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006159 const struct built_in_command *x,
6160 const struct built_in_command *end)
6161{
6162 while (x != end) {
6163 if (strcmp(name, x->b_cmd) != 0) {
6164 x++;
6165 continue;
6166 }
6167 debug_printf_exec("found builtin '%s'\n", name);
6168 return x;
6169 }
6170 return NULL;
6171}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006172static const struct built_in_command *find_builtin1(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006173{
6174 return find_builtin_helper(name, bltins1, &bltins1[ARRAY_SIZE(bltins1)]);
6175}
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006176static const struct built_in_command *find_builtin(const char *name)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006177{
6178 const struct built_in_command *x = find_builtin1(name);
6179 if (x)
6180 return x;
6181 return find_builtin_helper(name, bltins2, &bltins2[ARRAY_SIZE(bltins2)]);
6182}
6183
6184#if ENABLE_HUSH_FUNCTIONS
6185static struct function **find_function_slot(const char *name)
6186{
6187 struct function **funcpp = &G.top_func;
6188 while (*funcpp) {
6189 if (strcmp(name, (*funcpp)->name) == 0) {
6190 break;
6191 }
6192 funcpp = &(*funcpp)->next;
6193 }
6194 return funcpp;
6195}
6196
6197static const struct function *find_function(const char *name)
6198{
6199 const struct function *funcp = *find_function_slot(name);
6200 if (funcp)
6201 debug_printf_exec("found function '%s'\n", name);
6202 return funcp;
6203}
6204
6205/* Note: takes ownership on name ptr */
6206static struct function *new_function(char *name)
6207{
6208 struct function **funcpp = find_function_slot(name);
6209 struct function *funcp = *funcpp;
6210
6211 if (funcp != NULL) {
6212 struct command *cmd = funcp->parent_cmd;
6213 debug_printf_exec("func %p parent_cmd %p\n", funcp, cmd);
6214 if (!cmd) {
6215 debug_printf_exec("freeing & replacing function '%s'\n", funcp->name);
6216 free(funcp->name);
6217 /* Note: if !funcp->body, do not free body_as_string!
6218 * This is a special case of "-F name body" function:
6219 * body_as_string was not malloced! */
6220 if (funcp->body) {
6221 free_pipe_list(funcp->body);
6222# if !BB_MMU
6223 free(funcp->body_as_string);
6224# endif
6225 }
6226 } else {
6227 debug_printf_exec("reinserting in tree & replacing function '%s'\n", funcp->name);
6228 cmd->argv[0] = funcp->name;
6229 cmd->group = funcp->body;
6230# if !BB_MMU
6231 cmd->group_as_string = funcp->body_as_string;
6232# endif
6233 }
6234 } else {
6235 debug_printf_exec("remembering new function '%s'\n", name);
6236 funcp = *funcpp = xzalloc(sizeof(*funcp));
6237 /*funcp->next = NULL;*/
6238 }
6239
6240 funcp->name = name;
6241 return funcp;
6242}
6243
6244static void unset_func(const char *name)
6245{
6246 struct function **funcpp = find_function_slot(name);
6247 struct function *funcp = *funcpp;
6248
6249 if (funcp != NULL) {
6250 debug_printf_exec("freeing function '%s'\n", funcp->name);
6251 *funcpp = funcp->next;
6252 /* funcp is unlinked now, deleting it.
6253 * Note: if !funcp->body, the function was created by
6254 * "-F name body", do not free ->body_as_string
6255 * and ->name as they were not malloced. */
6256 if (funcp->body) {
6257 free_pipe_list(funcp->body);
6258 free(funcp->name);
6259# if !BB_MMU
6260 free(funcp->body_as_string);
6261# endif
6262 }
6263 free(funcp);
6264 }
6265}
6266
6267# if BB_MMU
6268#define exec_function(to_free, funcp, argv) \
6269 exec_function(funcp, argv)
6270# endif
6271static void exec_function(char ***to_free,
6272 const struct function *funcp,
6273 char **argv) NORETURN;
6274static void exec_function(char ***to_free,
6275 const struct function *funcp,
6276 char **argv)
6277{
6278# if BB_MMU
6279 int n = 1;
6280
6281 argv[0] = G.global_argv[0];
6282 G.global_argv = argv;
6283 while (*++argv)
6284 n++;
6285 G.global_argc = n;
6286 /* On MMU, funcp->body is always non-NULL */
6287 n = run_list(funcp->body);
6288 fflush_all();
6289 _exit(n);
6290# else
6291 re_execute_shell(to_free,
6292 funcp->body_as_string,
6293 G.global_argv[0],
6294 argv + 1,
6295 NULL);
6296# endif
6297}
6298
6299static int run_function(const struct function *funcp, char **argv)
6300{
6301 int rc;
6302 save_arg_t sv;
6303 smallint sv_flg;
6304
6305 save_and_replace_G_args(&sv, argv);
6306
6307 /* "we are in function, ok to use return" */
6308 sv_flg = G.flag_return_in_progress;
6309 G.flag_return_in_progress = -1;
6310# if ENABLE_HUSH_LOCAL
6311 G.func_nest_level++;
6312# endif
6313
6314 /* On MMU, funcp->body is always non-NULL */
6315# if !BB_MMU
6316 if (!funcp->body) {
6317 /* Function defined by -F */
6318 parse_and_run_string(funcp->body_as_string);
6319 rc = G.last_exitcode;
6320 } else
6321# endif
6322 {
6323 rc = run_list(funcp->body);
6324 }
6325
6326# if ENABLE_HUSH_LOCAL
6327 {
6328 struct variable *var;
6329 struct variable **var_pp;
6330
6331 var_pp = &G.top_var;
6332 while ((var = *var_pp) != NULL) {
6333 if (var->func_nest_level < G.func_nest_level) {
6334 var_pp = &var->next;
6335 continue;
6336 }
6337 /* Unexport */
6338 if (var->flg_export)
6339 bb_unsetenv(var->varstr);
6340 /* Remove from global list */
6341 *var_pp = var->next;
6342 /* Free */
6343 if (!var->max_len)
6344 free(var->varstr);
6345 free(var);
6346 }
6347 G.func_nest_level--;
6348 }
6349# endif
6350 G.flag_return_in_progress = sv_flg;
6351
6352 restore_G_args(&sv, argv);
6353
6354 return rc;
6355}
6356#endif /* ENABLE_HUSH_FUNCTIONS */
6357
6358
6359#if BB_MMU
6360#define exec_builtin(to_free, x, argv) \
6361 exec_builtin(x, argv)
6362#else
6363#define exec_builtin(to_free, x, argv) \
6364 exec_builtin(to_free, argv)
6365#endif
6366static void exec_builtin(char ***to_free,
6367 const struct built_in_command *x,
6368 char **argv) NORETURN;
6369static void exec_builtin(char ***to_free,
6370 const struct built_in_command *x,
6371 char **argv)
6372{
6373#if BB_MMU
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006374 int rcode;
6375 fflush_all();
6376 rcode = x->b_function(argv);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006377 fflush_all();
6378 _exit(rcode);
6379#else
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01006380 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006381 /* On NOMMU, we must never block!
6382 * Example: { sleep 99 | read line; } & echo Ok
6383 */
6384 re_execute_shell(to_free,
6385 argv[0],
6386 G.global_argv[0],
6387 G.global_argv + 1,
6388 argv);
6389#endif
6390}
6391
6392
6393static void execvp_or_die(char **argv) NORETURN;
6394static void execvp_or_die(char **argv)
6395{
6396 debug_printf_exec("execing '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006397 /* Don't propagate SIG_IGN to the child */
6398 if (SPECIAL_JOBSTOP_SIGS != 0)
6399 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006400 execvp(argv[0], argv);
6401 bb_perror_msg("can't execute '%s'", argv[0]);
6402 _exit(127); /* bash compat */
6403}
6404
6405#if ENABLE_HUSH_MODE_X
6406static void dump_cmd_in_x_mode(char **argv)
6407{
6408 if (G_x_mode && argv) {
6409 /* We want to output the line in one write op */
6410 char *buf, *p;
6411 int len;
6412 int n;
6413
6414 len = 3;
6415 n = 0;
6416 while (argv[n])
6417 len += strlen(argv[n++]) + 1;
6418 buf = xmalloc(len);
6419 buf[0] = '+';
6420 p = buf + 1;
6421 n = 0;
6422 while (argv[n])
6423 p += sprintf(p, " %s", argv[n++]);
6424 *p++ = '\n';
6425 *p = '\0';
6426 fputs(buf, stderr);
6427 free(buf);
6428 }
6429}
6430#else
6431# define dump_cmd_in_x_mode(argv) ((void)0)
6432#endif
6433
6434#if BB_MMU
6435#define pseudo_exec_argv(nommu_save, argv, assignment_cnt, argv_expanded) \
6436 pseudo_exec_argv(argv, assignment_cnt, argv_expanded)
6437#define pseudo_exec(nommu_save, command, argv_expanded) \
6438 pseudo_exec(command, argv_expanded)
6439#endif
6440
6441/* Called after [v]fork() in run_pipe, or from builtin_exec.
6442 * Never returns.
6443 * Don't exit() here. If you don't exec, use _exit instead.
6444 * The at_exit handlers apparently confuse the calling process,
6445 * in particular stdin handling. Not sure why? -- because of vfork! (vda) */
6446static void pseudo_exec_argv(nommu_save_t *nommu_save,
6447 char **argv, int assignment_cnt,
6448 char **argv_expanded) NORETURN;
6449static NOINLINE void pseudo_exec_argv(nommu_save_t *nommu_save,
6450 char **argv, int assignment_cnt,
6451 char **argv_expanded)
6452{
6453 char **new_env;
6454
6455 new_env = expand_assignments(argv, assignment_cnt);
6456 dump_cmd_in_x_mode(new_env);
6457
6458 if (!argv[assignment_cnt]) {
6459 /* Case when we are here: ... | var=val | ...
6460 * (note that we do not exit early, i.e., do not optimize out
6461 * expand_assignments(): think about ... | var=`sleep 1` | ...
6462 */
6463 free_strings(new_env);
6464 _exit(EXIT_SUCCESS);
6465 }
6466
6467#if BB_MMU
6468 set_vars_and_save_old(new_env);
6469 free(new_env); /* optional */
6470 /* we can also destroy set_vars_and_save_old's return value,
6471 * to save memory */
6472#else
6473 nommu_save->new_env = new_env;
6474 nommu_save->old_vars = set_vars_and_save_old(new_env);
6475#endif
6476
6477 if (argv_expanded) {
6478 argv = argv_expanded;
6479 } else {
6480 argv = expand_strvec_to_strvec(argv + assignment_cnt);
6481#if !BB_MMU
6482 nommu_save->argv = argv;
6483#endif
6484 }
6485 dump_cmd_in_x_mode(argv);
6486
6487#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6488 if (strchr(argv[0], '/') != NULL)
6489 goto skip;
6490#endif
6491
6492 /* Check if the command matches any of the builtins.
6493 * Depending on context, this might be redundant. But it's
6494 * easier to waste a few CPU cycles than it is to figure out
6495 * if this is one of those cases.
6496 */
6497 {
6498 /* On NOMMU, it is more expensive to re-execute shell
6499 * just in order to run echo or test builtin.
6500 * It's better to skip it here and run corresponding
6501 * non-builtin later. */
6502 const struct built_in_command *x;
6503 x = BB_MMU ? find_builtin(argv[0]) : find_builtin1(argv[0]);
6504 if (x) {
6505 exec_builtin(&nommu_save->argv_from_re_execing, x, argv);
6506 }
6507 }
6508#if ENABLE_HUSH_FUNCTIONS
6509 /* Check if the command matches any functions */
6510 {
6511 const struct function *funcp = find_function(argv[0]);
6512 if (funcp) {
6513 exec_function(&nommu_save->argv_from_re_execing, funcp, argv);
6514 }
6515 }
6516#endif
6517
6518#if ENABLE_FEATURE_SH_STANDALONE
6519 /* Check if the command matches any busybox applets */
6520 {
6521 int a = find_applet_by_name(argv[0]);
6522 if (a >= 0) {
6523# if BB_MMU /* see above why on NOMMU it is not allowed */
6524 if (APPLET_IS_NOEXEC(a)) {
6525 debug_printf_exec("running applet '%s'\n", argv[0]);
6526 run_applet_no_and_exit(a, argv);
6527 }
6528# endif
6529 /* Re-exec ourselves */
6530 debug_printf_exec("re-execing applet '%s'\n", argv[0]);
Denys Vlasenko75e77de2011-05-12 13:12:47 +02006531 /* Don't propagate SIG_IGN to the child */
6532 if (SPECIAL_JOBSTOP_SIGS != 0)
6533 switch_off_special_sigs(G.special_sig_mask & SPECIAL_JOBSTOP_SIGS);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006534 execv(bb_busybox_exec_path, argv);
6535 /* If they called chroot or otherwise made the binary no longer
6536 * executable, fall through */
6537 }
6538 }
6539#endif
6540
6541#if ENABLE_FEATURE_SH_STANDALONE || BB_MMU
6542 skip:
6543#endif
6544 execvp_or_die(argv);
6545}
6546
6547/* Called after [v]fork() in run_pipe
6548 */
6549static void pseudo_exec(nommu_save_t *nommu_save,
6550 struct command *command,
6551 char **argv_expanded) NORETURN;
6552static void pseudo_exec(nommu_save_t *nommu_save,
6553 struct command *command,
6554 char **argv_expanded)
6555{
6556 if (command->argv) {
6557 pseudo_exec_argv(nommu_save, command->argv,
6558 command->assignment_cnt, argv_expanded);
6559 }
6560
6561 if (command->group) {
6562 /* Cases when we are here:
6563 * ( list )
6564 * { list } &
6565 * ... | ( list ) | ...
6566 * ... | { list } | ...
6567 */
6568#if BB_MMU
6569 int rcode;
6570 debug_printf_exec("pseudo_exec: run_list\n");
6571 reset_traps_to_defaults();
6572 rcode = run_list(command->group);
6573 /* OK to leak memory by not calling free_pipe_list,
6574 * since this process is about to exit */
6575 _exit(rcode);
6576#else
6577 re_execute_shell(&nommu_save->argv_from_re_execing,
6578 command->group_as_string,
6579 G.global_argv[0],
6580 G.global_argv + 1,
6581 NULL);
6582#endif
6583 }
6584
6585 /* Case when we are here: ... | >file */
6586 debug_printf_exec("pseudo_exec'ed null command\n");
6587 _exit(EXIT_SUCCESS);
6588}
6589
6590#if ENABLE_HUSH_JOB
6591static const char *get_cmdtext(struct pipe *pi)
6592{
6593 char **argv;
6594 char *p;
6595 int len;
6596
6597 /* This is subtle. ->cmdtext is created only on first backgrounding.
6598 * (Think "cat, <ctrl-z>, fg, <ctrl-z>, fg, <ctrl-z>...." here...)
6599 * On subsequent bg argv is trashed, but we won't use it */
6600 if (pi->cmdtext)
6601 return pi->cmdtext;
6602 argv = pi->cmds[0].argv;
6603 if (!argv || !argv[0]) {
6604 pi->cmdtext = xzalloc(1);
6605 return pi->cmdtext;
6606 }
6607
6608 len = 0;
6609 do {
6610 len += strlen(*argv) + 1;
6611 } while (*++argv);
6612 p = xmalloc(len);
6613 pi->cmdtext = p;
6614 argv = pi->cmds[0].argv;
6615 do {
6616 len = strlen(*argv);
6617 memcpy(p, *argv, len);
6618 p += len;
6619 *p++ = ' ';
6620 } while (*++argv);
6621 p[-1] = '\0';
6622 return pi->cmdtext;
6623}
6624
6625static void insert_bg_job(struct pipe *pi)
6626{
6627 struct pipe *job, **jobp;
6628 int i;
6629
6630 /* Linear search for the ID of the job to use */
6631 pi->jobid = 1;
6632 for (job = G.job_list; job; job = job->next)
6633 if (job->jobid >= pi->jobid)
6634 pi->jobid = job->jobid + 1;
6635
6636 /* Add job to the list of running jobs */
6637 jobp = &G.job_list;
6638 while ((job = *jobp) != NULL)
6639 jobp = &job->next;
6640 job = *jobp = xmalloc(sizeof(*job));
6641
6642 *job = *pi; /* physical copy */
6643 job->next = NULL;
6644 job->cmds = xzalloc(sizeof(pi->cmds[0]) * pi->num_cmds);
6645 /* Cannot copy entire pi->cmds[] vector! This causes double frees */
6646 for (i = 0; i < pi->num_cmds; i++) {
6647 job->cmds[i].pid = pi->cmds[i].pid;
6648 /* all other fields are not used and stay zero */
6649 }
6650 job->cmdtext = xstrdup(get_cmdtext(pi));
6651
6652 if (G_interactive_fd)
6653 printf("[%d] %d %s\n", job->jobid, job->cmds[0].pid, job->cmdtext);
6654 G.last_jobid = job->jobid;
6655}
6656
6657static void remove_bg_job(struct pipe *pi)
6658{
6659 struct pipe *prev_pipe;
6660
6661 if (pi == G.job_list) {
6662 G.job_list = pi->next;
6663 } else {
6664 prev_pipe = G.job_list;
6665 while (prev_pipe->next != pi)
6666 prev_pipe = prev_pipe->next;
6667 prev_pipe->next = pi->next;
6668 }
6669 if (G.job_list)
6670 G.last_jobid = G.job_list->jobid;
6671 else
6672 G.last_jobid = 0;
6673}
6674
6675/* Remove a backgrounded job */
6676static void delete_finished_bg_job(struct pipe *pi)
6677{
6678 remove_bg_job(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006679 free_pipe(pi);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006680}
6681#endif /* JOB */
6682
6683/* Check to see if any processes have exited -- if they
6684 * have, figure out why and see if a job has completed */
Denys Vlasenko27c56f12010-09-07 09:56:34 +02006685static int checkjobs(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006686{
6687 int attributes;
6688 int status;
6689#if ENABLE_HUSH_JOB
6690 struct pipe *pi;
6691#endif
6692 pid_t childpid;
6693 int rcode = 0;
6694
6695 debug_printf_jobs("checkjobs %p\n", fg_pipe);
6696
6697 attributes = WUNTRACED;
6698 if (fg_pipe == NULL)
6699 attributes |= WNOHANG;
6700
6701 errno = 0;
6702#if ENABLE_HUSH_FAST
6703 if (G.handled_SIGCHLD == G.count_SIGCHLD) {
6704//bb_error_msg("[%d] checkjobs: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d children?:%d fg_pipe:%p",
6705//getpid(), G.count_SIGCHLD, G.handled_SIGCHLD, G.we_have_children, fg_pipe);
6706 /* There was neither fork nor SIGCHLD since last waitpid */
6707 /* Avoid doing waitpid syscall if possible */
6708 if (!G.we_have_children) {
6709 errno = ECHILD;
6710 return -1;
6711 }
6712 if (fg_pipe == NULL) { /* is WNOHANG set? */
6713 /* We have children, but they did not exit
6714 * or stop yet (we saw no SIGCHLD) */
6715 return 0;
6716 }
6717 /* else: !WNOHANG, waitpid will block, can't short-circuit */
6718 }
6719#endif
6720
6721/* Do we do this right?
6722 * bash-3.00# sleep 20 | false
6723 * <ctrl-Z pressed>
6724 * [3]+ Stopped sleep 20 | false
6725 * bash-3.00# echo $?
6726 * 1 <========== bg pipe is not fully done, but exitcode is already known!
6727 * [hush 1.14.0: yes we do it right]
6728 */
6729 wait_more:
6730 while (1) {
6731 int i;
6732 int dead;
6733
6734#if ENABLE_HUSH_FAST
6735 i = G.count_SIGCHLD;
6736#endif
6737 childpid = waitpid(-1, &status, attributes);
6738 if (childpid <= 0) {
6739 if (childpid && errno != ECHILD)
6740 bb_perror_msg("waitpid");
6741#if ENABLE_HUSH_FAST
6742 else { /* Until next SIGCHLD, waitpid's are useless */
6743 G.we_have_children = (childpid == 0);
6744 G.handled_SIGCHLD = i;
6745//bb_error_msg("[%d] checkjobs: waitpid returned <= 0, G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
6746 }
6747#endif
6748 break;
6749 }
6750 dead = WIFEXITED(status) || WIFSIGNALED(status);
6751
6752#if DEBUG_JOBS
6753 if (WIFSTOPPED(status))
6754 debug_printf_jobs("pid %d stopped by sig %d (exitcode %d)\n",
6755 childpid, WSTOPSIG(status), WEXITSTATUS(status));
6756 if (WIFSIGNALED(status))
6757 debug_printf_jobs("pid %d killed by sig %d (exitcode %d)\n",
6758 childpid, WTERMSIG(status), WEXITSTATUS(status));
6759 if (WIFEXITED(status))
6760 debug_printf_jobs("pid %d exited, exitcode %d\n",
6761 childpid, WEXITSTATUS(status));
6762#endif
6763 /* Were we asked to wait for fg pipe? */
6764 if (fg_pipe) {
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006765 i = fg_pipe->num_cmds;
6766 while (--i >= 0) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006767 debug_printf_jobs("check pid %d\n", fg_pipe->cmds[i].pid);
6768 if (fg_pipe->cmds[i].pid != childpid)
6769 continue;
6770 if (dead) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006771 int ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006772 fg_pipe->cmds[i].pid = 0;
6773 fg_pipe->alive_cmds--;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006774 ex = WEXITSTATUS(status);
6775 /* bash prints killer signal's name for *last*
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006776 * process in pipe (prints just newline for SIGINT/SIGPIPE).
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006777 * Mimic this. Example: "sleep 5" + (^\ or kill -QUIT)
6778 */
6779 if (WIFSIGNALED(status)) {
6780 int sig = WTERMSIG(status);
6781 if (i == fg_pipe->num_cmds-1)
Denys Vlasenko7c6f2462011-02-14 17:17:10 +01006782 /* TODO: use strsignal() instead for bash compat? but that's bloat... */
6783 printf("%s\n", sig == SIGINT || sig == SIGPIPE ? "" : get_signame(sig));
6784 /* TODO: if (WCOREDUMP(status)) + " (core dumped)"; */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006785 /* TODO: MIPS has 128 sigs (1..128), what if sig==128 here?
6786 * Maybe we need to use sig | 128? */
6787 ex = sig + 128;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006788 }
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006789 fg_pipe->cmds[i].cmd_exitcode = ex;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006790 } else {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006791 fg_pipe->stopped_cmds++;
6792 }
6793 debug_printf_jobs("fg_pipe: alive_cmds %d stopped_cmds %d\n",
6794 fg_pipe->alive_cmds, fg_pipe->stopped_cmds);
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006795 if (fg_pipe->alive_cmds == fg_pipe->stopped_cmds) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006796 /* All processes in fg pipe have exited or stopped */
Denys Vlasenko6696eac2010-11-14 02:01:50 +01006797 i = fg_pipe->num_cmds;
6798 while (--i >= 0) {
6799 rcode = fg_pipe->cmds[i].cmd_exitcode;
6800 /* usually last process gives overall exitstatus,
6801 * but with "set -o pipefail", last *failed* process does */
6802 if (G.o_opt[OPT_O_PIPEFAIL] == 0 || rcode != 0)
6803 break;
6804 }
6805 IF_HAS_KEYWORDS(if (fg_pipe->pi_inverted) rcode = !rcode;)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006806/* Note: *non-interactive* bash does not continue if all processes in fg pipe
6807 * are stopped. Testcase: "cat | cat" in a script (not on command line!)
6808 * and "killall -STOP cat" */
6809 if (G_interactive_fd) {
6810#if ENABLE_HUSH_JOB
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006811 if (fg_pipe->alive_cmds != 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006812 insert_bg_job(fg_pipe);
6813#endif
6814 return rcode;
6815 }
Denys Vlasenkoc08c3f52010-11-14 01:59:55 +01006816 if (fg_pipe->alive_cmds == 0)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006817 return rcode;
6818 }
6819 /* There are still running processes in the fg pipe */
6820 goto wait_more; /* do waitpid again */
6821 }
6822 /* it wasnt fg_pipe, look for process in bg pipes */
6823 }
6824
6825#if ENABLE_HUSH_JOB
6826 /* We asked to wait for bg or orphaned children */
6827 /* No need to remember exitcode in this case */
6828 for (pi = G.job_list; pi; pi = pi->next) {
6829 for (i = 0; i < pi->num_cmds; i++) {
6830 if (pi->cmds[i].pid == childpid)
6831 goto found_pi_and_prognum;
6832 }
6833 }
6834 /* Happens when shell is used as init process (init=/bin/sh) */
6835 debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
6836 continue; /* do waitpid again */
6837
6838 found_pi_and_prognum:
6839 if (dead) {
6840 /* child exited */
6841 pi->cmds[i].pid = 0;
6842 pi->alive_cmds--;
6843 if (!pi->alive_cmds) {
6844 if (G_interactive_fd)
6845 printf(JOB_STATUS_FORMAT, pi->jobid,
6846 "Done", pi->cmdtext);
6847 delete_finished_bg_job(pi);
6848 }
6849 } else {
6850 /* child stopped */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006851 pi->stopped_cmds++;
6852 }
6853#endif
6854 } /* while (waitpid succeeds)... */
6855
6856 return rcode;
6857}
6858
6859#if ENABLE_HUSH_JOB
Denys Vlasenkoda463fb2010-09-07 09:53:50 +02006860static int checkjobs_and_fg_shell(struct pipe *fg_pipe)
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006861{
6862 pid_t p;
6863 int rcode = checkjobs(fg_pipe);
6864 if (G_saved_tty_pgrp) {
6865 /* Job finished, move the shell to the foreground */
6866 p = getpgrp(); /* our process group id */
6867 debug_printf_jobs("fg'ing ourself: getpgrp()=%d\n", (int)p);
6868 tcsetpgrp(G_interactive_fd, p);
6869 }
6870 return rcode;
6871}
6872#endif
6873
6874/* Start all the jobs, but don't wait for anything to finish.
6875 * See checkjobs().
6876 *
6877 * Return code is normally -1, when the caller has to wait for children
6878 * to finish to determine the exit status of the pipe. If the pipe
6879 * is a simple builtin command, however, the action is done by the
6880 * time run_pipe returns, and the exit code is provided as the
6881 * return value.
6882 *
6883 * Returns -1 only if started some children. IOW: we have to
6884 * mask out retvals of builtins etc with 0xff!
6885 *
6886 * The only case when we do not need to [v]fork is when the pipe
6887 * is single, non-backgrounded, non-subshell command. Examples:
6888 * cmd ; ... { list } ; ...
6889 * cmd && ... { list } && ...
6890 * cmd || ... { list } || ...
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01006891 * If it is, then we can run cmd as a builtin, NOFORK,
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006892 * or (if SH_STANDALONE) an applet, and we can run the { list }
6893 * with run_list. If it isn't one of these, we fork and exec cmd.
6894 *
6895 * Cases when we must fork:
6896 * non-single: cmd | cmd
6897 * backgrounded: cmd & { list } &
6898 * subshell: ( list ) [&]
6899 */
6900#if !ENABLE_HUSH_MODE_X
Denys Vlasenko26777aa2010-11-22 23:49:10 +01006901#define redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel, argv_expanded) \
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006902 redirect_and_varexp_helper(new_env_p, old_vars_p, command, squirrel)
6903#endif
6904static int redirect_and_varexp_helper(char ***new_env_p,
6905 struct variable **old_vars_p,
6906 struct command *command,
6907 int squirrel[3],
6908 char **argv_expanded)
6909{
6910 /* setup_redirects acts on file descriptors, not FILEs.
6911 * This is perfect for work that comes after exec().
6912 * Is it really safe for inline use? Experimentally,
6913 * things seem to work. */
6914 int rcode = setup_redirects(command, squirrel);
6915 if (rcode == 0) {
6916 char **new_env = expand_assignments(command->argv, command->assignment_cnt);
6917 *new_env_p = new_env;
6918 dump_cmd_in_x_mode(new_env);
6919 dump_cmd_in_x_mode(argv_expanded);
6920 if (old_vars_p)
6921 *old_vars_p = set_vars_and_save_old(new_env);
6922 }
6923 return rcode;
6924}
6925static NOINLINE int run_pipe(struct pipe *pi)
6926{
6927 static const char *const null_ptr = NULL;
6928
6929 int cmd_no;
6930 int next_infd;
6931 struct command *command;
6932 char **argv_expanded;
6933 char **argv;
6934 /* it is not always needed, but we aim to smaller code */
6935 int squirrel[] = { -1, -1, -1 };
6936 int rcode;
6937
6938 debug_printf_exec("run_pipe start: members:%d\n", pi->num_cmds);
6939 debug_enter();
6940
Denys Vlasenko1fd3d942010-09-08 13:31:53 +02006941 /* Testcase: set -- q w e; (IFS='' echo "$*"; IFS=''; echo "$*"); echo "$*"
6942 * Result should be 3 lines: q w e, qwe, q w e
6943 */
6944 G.ifs = get_local_var_value("IFS");
6945 if (!G.ifs)
6946 G.ifs = defifs;
6947
Denys Vlasenkob36abf22010-09-05 14:50:59 +02006948 IF_HUSH_JOB(pi->pgrp = -1;)
6949 pi->stopped_cmds = 0;
6950 command = &pi->cmds[0];
6951 argv_expanded = NULL;
6952
6953 if (pi->num_cmds != 1
6954 || pi->followup == PIPE_BG
6955 || command->cmd_type == CMD_SUBSHELL
6956 ) {
6957 goto must_fork;
6958 }
6959
6960 pi->alive_cmds = 1;
6961
6962 debug_printf_exec(": group:%p argv:'%s'\n",
6963 command->group, command->argv ? command->argv[0] : "NONE");
6964
6965 if (command->group) {
6966#if ENABLE_HUSH_FUNCTIONS
6967 if (command->cmd_type == CMD_FUNCDEF) {
6968 /* "executing" func () { list } */
6969 struct function *funcp;
6970
6971 funcp = new_function(command->argv[0]);
6972 /* funcp->name is already set to argv[0] */
6973 funcp->body = command->group;
6974# if !BB_MMU
6975 funcp->body_as_string = command->group_as_string;
6976 command->group_as_string = NULL;
6977# endif
6978 command->group = NULL;
6979 command->argv[0] = NULL;
6980 debug_printf_exec("cmd %p has child func at %p\n", command, funcp);
6981 funcp->parent_cmd = command;
6982 command->child_func = funcp;
6983
6984 debug_printf_exec("run_pipe: return EXIT_SUCCESS\n");
6985 debug_leave();
6986 return EXIT_SUCCESS;
6987 }
6988#endif
6989 /* { list } */
6990 debug_printf("non-subshell group\n");
6991 rcode = 1; /* exitcode if redir failed */
6992 if (setup_redirects(command, squirrel) == 0) {
6993 debug_printf_exec(": run_list\n");
6994 rcode = run_list(command->group) & 0xff;
6995 }
6996 restore_redirects(squirrel);
6997 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
6998 debug_leave();
6999 debug_printf_exec("run_pipe: return %d\n", rcode);
7000 return rcode;
7001 }
7002
7003 argv = command->argv ? command->argv : (char **) &null_ptr;
7004 {
7005 const struct built_in_command *x;
7006#if ENABLE_HUSH_FUNCTIONS
7007 const struct function *funcp;
7008#else
7009 enum { funcp = 0 };
7010#endif
7011 char **new_env = NULL;
7012 struct variable *old_vars = NULL;
7013
7014 if (argv[command->assignment_cnt] == NULL) {
7015 /* Assignments, but no command */
7016 /* Ensure redirects take effect (that is, create files).
7017 * Try "a=t >file" */
7018#if 0 /* A few cases in testsuite fail with this code. FIXME */
7019 rcode = redirect_and_varexp_helper(&new_env, /*old_vars:*/ NULL, command, squirrel, /*argv_expanded:*/ NULL);
7020 /* Set shell variables */
7021 if (new_env) {
7022 argv = new_env;
7023 while (*argv) {
7024 set_local_var(*argv, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7025 /* Do we need to flag set_local_var() errors?
7026 * "assignment to readonly var" and "putenv error"
7027 */
7028 argv++;
7029 }
7030 }
7031 /* Redirect error sets $? to 1. Otherwise,
7032 * if evaluating assignment value set $?, retain it.
7033 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7034 if (rcode == 0)
7035 rcode = G.last_exitcode;
7036 /* Exit, _skipping_ variable restoring code: */
7037 goto clean_up_and_ret0;
7038
7039#else /* Older, bigger, but more correct code */
7040
7041 rcode = setup_redirects(command, squirrel);
7042 restore_redirects(squirrel);
7043 /* Set shell variables */
7044 if (G_x_mode)
7045 bb_putchar_stderr('+');
7046 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007047 char *p = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007048 if (G_x_mode)
7049 fprintf(stderr, " %s", p);
7050 debug_printf_exec("set shell var:'%s'->'%s'\n",
7051 *argv, p);
7052 set_local_var(p, /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7053 /* Do we need to flag set_local_var() errors?
7054 * "assignment to readonly var" and "putenv error"
7055 */
7056 argv++;
7057 }
7058 if (G_x_mode)
7059 bb_putchar_stderr('\n');
7060 /* Redirect error sets $? to 1. Otherwise,
7061 * if evaluating assignment value set $?, retain it.
7062 * Try "false; q=`exit 2`; echo $?" - should print 2: */
7063 if (rcode == 0)
7064 rcode = G.last_exitcode;
7065 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7066 debug_leave();
7067 debug_printf_exec("run_pipe: return %d\n", rcode);
7068 return rcode;
7069#endif
7070 }
7071
7072 /* Expand the rest into (possibly) many strings each */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007073#if ENABLE_HUSH_BASH_COMPAT
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007074 if (command->cmd_type == CMD_SINGLEWORD_NOGLOB) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007075 argv_expanded = expand_strvec_to_strvec_singleword_noglob(argv + command->assignment_cnt);
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007076 } else
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007077#endif
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007078 {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007079 argv_expanded = expand_strvec_to_strvec(argv + command->assignment_cnt);
7080 }
7081
7082 /* if someone gives us an empty string: `cmd with empty output` */
7083 if (!argv_expanded[0]) {
7084 free(argv_expanded);
7085 debug_leave();
7086 return G.last_exitcode;
7087 }
7088
7089 x = find_builtin(argv_expanded[0]);
7090#if ENABLE_HUSH_FUNCTIONS
7091 funcp = NULL;
7092 if (!x)
7093 funcp = find_function(argv_expanded[0]);
7094#endif
7095 if (x || funcp) {
7096 if (!funcp) {
7097 if (x->b_function == builtin_exec && argv_expanded[1] == NULL) {
7098 debug_printf("exec with redirects only\n");
7099 rcode = setup_redirects(command, NULL);
7100 goto clean_up_and_ret1;
7101 }
7102 }
7103 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7104 if (rcode == 0) {
7105 if (!funcp) {
7106 debug_printf_exec(": builtin '%s' '%s'...\n",
7107 x->b_cmd, argv_expanded[1]);
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007108 fflush_all();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007109 rcode = x->b_function(argv_expanded) & 0xff;
7110 fflush_all();
7111 }
7112#if ENABLE_HUSH_FUNCTIONS
7113 else {
7114# if ENABLE_HUSH_LOCAL
7115 struct variable **sv;
7116 sv = G.shadowed_vars_pp;
7117 G.shadowed_vars_pp = &old_vars;
7118# endif
7119 debug_printf_exec(": function '%s' '%s'...\n",
7120 funcp->name, argv_expanded[1]);
7121 rcode = run_function(funcp, argv_expanded) & 0xff;
7122# if ENABLE_HUSH_LOCAL
7123 G.shadowed_vars_pp = sv;
7124# endif
7125 }
7126#endif
7127 }
7128 clean_up_and_ret:
7129 unset_vars(new_env);
7130 add_vars(old_vars);
7131/* clean_up_and_ret0: */
7132 restore_redirects(squirrel);
7133 clean_up_and_ret1:
7134 free(argv_expanded);
7135 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
7136 debug_leave();
7137 debug_printf_exec("run_pipe return %d\n", rcode);
7138 return rcode;
7139 }
7140
Denys Vlasenkob72baeb2011-02-02 18:38:57 +01007141 if (ENABLE_FEATURE_SH_NOFORK) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007142 int n = find_applet_by_name(argv_expanded[0]);
7143 if (n >= 0 && APPLET_IS_NOFORK(n)) {
7144 rcode = redirect_and_varexp_helper(&new_env, &old_vars, command, squirrel, argv_expanded);
7145 if (rcode == 0) {
7146 debug_printf_exec(": run_nofork_applet '%s' '%s'...\n",
7147 argv_expanded[0], argv_expanded[1]);
7148 rcode = run_nofork_applet(n, argv_expanded);
7149 }
7150 goto clean_up_and_ret;
7151 }
7152 }
7153 /* It is neither builtin nor applet. We must fork. */
7154 }
7155
7156 must_fork:
7157 /* NB: argv_expanded may already be created, and that
7158 * might include `cmd` runs! Do not rerun it! We *must*
7159 * use argv_expanded if it's non-NULL */
7160
7161 /* Going to fork a child per each pipe member */
7162 pi->alive_cmds = 0;
7163 next_infd = 0;
7164
7165 cmd_no = 0;
7166 while (cmd_no < pi->num_cmds) {
7167 struct fd_pair pipefds;
7168#if !BB_MMU
7169 volatile nommu_save_t nommu_save;
7170 nommu_save.new_env = NULL;
7171 nommu_save.old_vars = NULL;
7172 nommu_save.argv = NULL;
7173 nommu_save.argv_from_re_execing = NULL;
7174#endif
7175 command = &pi->cmds[cmd_no];
7176 cmd_no++;
7177 if (command->argv) {
7178 debug_printf_exec(": pipe member '%s' '%s'...\n",
7179 command->argv[0], command->argv[1]);
7180 } else {
7181 debug_printf_exec(": pipe member with no argv\n");
7182 }
7183
7184 /* pipes are inserted between pairs of commands */
7185 pipefds.rd = 0;
7186 pipefds.wr = 1;
7187 if (cmd_no < pi->num_cmds)
7188 xpiped_pair(pipefds);
7189
7190 command->pid = BB_MMU ? fork() : vfork();
7191 if (!command->pid) { /* child */
7192#if ENABLE_HUSH_JOB
7193 disable_restore_tty_pgrp_on_exit();
7194 CLEAR_RANDOM_T(&G.random_gen); /* or else $RANDOM repeats in child */
7195
7196 /* Every child adds itself to new process group
7197 * with pgid == pid_of_first_child_in_pipe */
7198 if (G.run_list_level == 1 && G_interactive_fd) {
7199 pid_t pgrp;
7200 pgrp = pi->pgrp;
7201 if (pgrp < 0) /* true for 1st process only */
7202 pgrp = getpid();
7203 if (setpgid(0, pgrp) == 0
7204 && pi->followup != PIPE_BG
7205 && G_saved_tty_pgrp /* we have ctty */
7206 ) {
7207 /* We do it in *every* child, not just first,
7208 * to avoid races */
7209 tcsetpgrp(G_interactive_fd, pgrp);
7210 }
7211 }
7212#endif
7213 if (pi->alive_cmds == 0 && pi->followup == PIPE_BG) {
7214 /* 1st cmd in backgrounded pipe
7215 * should have its stdin /dev/null'ed */
7216 close(0);
7217 if (open(bb_dev_null, O_RDONLY))
7218 xopen("/", O_RDONLY);
7219 } else {
7220 xmove_fd(next_infd, 0);
7221 }
7222 xmove_fd(pipefds.wr, 1);
7223 if (pipefds.rd > 1)
7224 close(pipefds.rd);
7225 /* Like bash, explicit redirects override pipes,
7226 * and the pipe fd is available for dup'ing. */
7227 if (setup_redirects(command, NULL))
7228 _exit(1);
7229
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007230 /* Stores to nommu_save list of env vars putenv'ed
7231 * (NOMMU, on MMU we don't need that) */
7232 /* cast away volatility... */
7233 pseudo_exec((nommu_save_t*) &nommu_save, command, argv_expanded);
7234 /* pseudo_exec() does not return */
7235 }
7236
7237 /* parent or error */
7238#if ENABLE_HUSH_FAST
7239 G.count_SIGCHLD++;
7240//bb_error_msg("[%d] fork in run_pipe: G.count_SIGCHLD:%d G.handled_SIGCHLD:%d", getpid(), G.count_SIGCHLD, G.handled_SIGCHLD);
7241#endif
7242 enable_restore_tty_pgrp_on_exit();
7243#if !BB_MMU
7244 /* Clean up after vforked child */
7245 free(nommu_save.argv);
7246 free(nommu_save.argv_from_re_execing);
7247 unset_vars(nommu_save.new_env);
7248 add_vars(nommu_save.old_vars);
7249#endif
7250 free(argv_expanded);
7251 argv_expanded = NULL;
7252 if (command->pid < 0) { /* [v]fork failed */
7253 /* Clearly indicate, was it fork or vfork */
7254 bb_perror_msg(BB_MMU ? "vfork"+1 : "vfork");
7255 } else {
7256 pi->alive_cmds++;
7257#if ENABLE_HUSH_JOB
7258 /* Second and next children need to know pid of first one */
7259 if (pi->pgrp < 0)
7260 pi->pgrp = command->pid;
7261#endif
7262 }
7263
7264 if (cmd_no > 1)
7265 close(next_infd);
7266 if (cmd_no < pi->num_cmds)
7267 close(pipefds.wr);
7268 /* Pass read (output) pipe end to next iteration */
7269 next_infd = pipefds.rd;
7270 }
7271
7272 if (!pi->alive_cmds) {
7273 debug_leave();
7274 debug_printf_exec("run_pipe return 1 (all forks failed, no children)\n");
7275 return 1;
7276 }
7277
7278 debug_leave();
7279 debug_printf_exec("run_pipe return -1 (%u children started)\n", pi->alive_cmds);
7280 return -1;
7281}
7282
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007283/* NB: called by pseudo_exec, and therefore must not modify any
7284 * global data until exec/_exit (we can be a child after vfork!) */
7285static int run_list(struct pipe *pi)
7286{
7287#if ENABLE_HUSH_CASE
7288 char *case_word = NULL;
7289#endif
7290#if ENABLE_HUSH_LOOPS
7291 struct pipe *loop_top = NULL;
7292 char **for_lcur = NULL;
7293 char **for_list = NULL;
7294#endif
7295 smallint last_followup;
7296 smalluint rcode;
7297#if ENABLE_HUSH_IF || ENABLE_HUSH_CASE
7298 smalluint cond_code = 0;
7299#else
7300 enum { cond_code = 0 };
7301#endif
7302#if HAS_KEYWORDS
Denys Vlasenko9b782552010-09-08 13:33:26 +02007303 smallint rword; /* RES_foo */
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007304 smallint last_rword; /* ditto */
7305#endif
7306
7307 debug_printf_exec("run_list start lvl %d\n", G.run_list_level);
7308 debug_enter();
7309
7310#if ENABLE_HUSH_LOOPS
7311 /* Check syntax for "for" */
Denys Vlasenko0d6a4ec2010-12-18 01:34:49 +01007312 {
7313 struct pipe *cpipe;
7314 for (cpipe = pi; cpipe; cpipe = cpipe->next) {
7315 if (cpipe->res_word != RES_FOR && cpipe->res_word != RES_IN)
7316 continue;
7317 /* current word is FOR or IN (BOLD in comments below) */
7318 if (cpipe->next == NULL) {
7319 syntax_error("malformed for");
7320 debug_leave();
7321 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7322 return 1;
7323 }
7324 /* "FOR v; do ..." and "for v IN a b; do..." are ok */
7325 if (cpipe->next->res_word == RES_DO)
7326 continue;
7327 /* next word is not "do". It must be "in" then ("FOR v in ...") */
7328 if (cpipe->res_word == RES_IN /* "for v IN a b; not_do..."? */
7329 || cpipe->next->res_word != RES_IN /* FOR v not_do_and_not_in..."? */
7330 ) {
7331 syntax_error("malformed for");
7332 debug_leave();
7333 debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
7334 return 1;
7335 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007336 }
7337 }
7338#endif
7339
7340 /* Past this point, all code paths should jump to ret: label
7341 * in order to return, no direct "return" statements please.
7342 * This helps to ensure that no memory is leaked. */
7343
7344#if ENABLE_HUSH_JOB
7345 G.run_list_level++;
7346#endif
7347
7348#if HAS_KEYWORDS
7349 rword = RES_NONE;
7350 last_rword = RES_XXXX;
7351#endif
7352 last_followup = PIPE_SEQ;
7353 rcode = G.last_exitcode;
7354
7355 /* Go through list of pipes, (maybe) executing them. */
7356 for (; pi; pi = IF_HUSH_LOOPS(rword == RES_DONE ? loop_top : ) pi->next) {
7357 if (G.flag_SIGINT)
7358 break;
7359
7360 IF_HAS_KEYWORDS(rword = pi->res_word;)
7361 debug_printf_exec(": rword=%d cond_code=%d last_rword=%d\n",
7362 rword, cond_code, last_rword);
7363#if ENABLE_HUSH_LOOPS
7364 if ((rword == RES_WHILE || rword == RES_UNTIL || rword == RES_FOR)
7365 && loop_top == NULL /* avoid bumping G.depth_of_loop twice */
7366 ) {
7367 /* start of a loop: remember where loop starts */
7368 loop_top = pi;
7369 G.depth_of_loop++;
7370 }
7371#endif
7372 /* Still in the same "if...", "then..." or "do..." branch? */
7373 if (IF_HAS_KEYWORDS(rword == last_rword &&) 1) {
7374 if ((rcode == 0 && last_followup == PIPE_OR)
7375 || (rcode != 0 && last_followup == PIPE_AND)
7376 ) {
7377 /* It is "<true> || CMD" or "<false> && CMD"
7378 * and we should not execute CMD */
7379 debug_printf_exec("skipped cmd because of || or &&\n");
7380 last_followup = pi->followup;
Denys Vlasenko3beab832013-04-07 18:16:58 +02007381 goto dont_check_jobs_but_continue;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007382 }
7383 }
7384 last_followup = pi->followup;
7385 IF_HAS_KEYWORDS(last_rword = rword;)
7386#if ENABLE_HUSH_IF
7387 if (cond_code) {
7388 if (rword == RES_THEN) {
7389 /* if false; then ... fi has exitcode 0! */
7390 G.last_exitcode = rcode = EXIT_SUCCESS;
7391 /* "if <false> THEN cmd": skip cmd */
7392 continue;
7393 }
7394 } else {
7395 if (rword == RES_ELSE || rword == RES_ELIF) {
7396 /* "if <true> then ... ELSE/ELIF cmd":
7397 * skip cmd and all following ones */
7398 break;
7399 }
7400 }
7401#endif
7402#if ENABLE_HUSH_LOOPS
7403 if (rword == RES_FOR) { /* && pi->num_cmds - always == 1 */
7404 if (!for_lcur) {
7405 /* first loop through for */
7406
7407 static const char encoded_dollar_at[] ALIGN1 = {
7408 SPECIAL_VAR_SYMBOL, '@' | 0x80, SPECIAL_VAR_SYMBOL, '\0'
7409 }; /* encoded representation of "$@" */
7410 static const char *const encoded_dollar_at_argv[] = {
7411 encoded_dollar_at, NULL
7412 }; /* argv list with one element: "$@" */
7413 char **vals;
7414
7415 vals = (char**)encoded_dollar_at_argv;
7416 if (pi->next->res_word == RES_IN) {
7417 /* if no variable values after "in" we skip "for" */
7418 if (!pi->next->cmds[0].argv) {
7419 G.last_exitcode = rcode = EXIT_SUCCESS;
7420 debug_printf_exec(": null FOR: exitcode EXIT_SUCCESS\n");
7421 break;
7422 }
7423 vals = pi->next->cmds[0].argv;
7424 } /* else: "for var; do..." -> assume "$@" list */
7425 /* create list of variable values */
7426 debug_print_strings("for_list made from", vals);
7427 for_list = expand_strvec_to_strvec(vals);
7428 for_lcur = for_list;
7429 debug_print_strings("for_list", for_list);
7430 }
7431 if (!*for_lcur) {
7432 /* "for" loop is over, clean up */
7433 free(for_list);
7434 for_list = NULL;
7435 for_lcur = NULL;
7436 break;
7437 }
7438 /* Insert next value from for_lcur */
7439 /* note: *for_lcur already has quotes removed, $var expanded, etc */
7440 set_local_var(xasprintf("%s=%s", pi->cmds[0].argv[0], *for_lcur++), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ 0);
7441 continue;
7442 }
7443 if (rword == RES_IN) {
7444 continue; /* "for v IN list;..." - "in" has no cmds anyway */
7445 }
7446 if (rword == RES_DONE) {
7447 continue; /* "done" has no cmds too */
7448 }
7449#endif
7450#if ENABLE_HUSH_CASE
7451 if (rword == RES_CASE) {
7452 case_word = expand_strvec_to_string(pi->cmds->argv);
7453 continue;
7454 }
7455 if (rword == RES_MATCH) {
7456 char **argv;
7457
7458 if (!case_word) /* "case ... matched_word) ... WORD)": we executed selected branch, stop */
7459 break;
7460 /* all prev words didn't match, does this one match? */
7461 argv = pi->cmds->argv;
7462 while (*argv) {
Denys Vlasenkoebee4102010-09-10 10:17:53 +02007463 char *pattern = expand_string_to_string(*argv, /*unbackslash:*/ 1);
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007464 /* TODO: which FNM_xxx flags to use? */
7465 cond_code = (fnmatch(pattern, case_word, /*flags:*/ 0) != 0);
7466 free(pattern);
7467 if (cond_code == 0) { /* match! we will execute this branch */
7468 free(case_word); /* make future "word)" stop */
7469 case_word = NULL;
7470 break;
7471 }
7472 argv++;
7473 }
7474 continue;
7475 }
7476 if (rword == RES_CASE_BODY) { /* inside of a case branch */
7477 if (cond_code != 0)
7478 continue; /* not matched yet, skip this pipe */
7479 }
7480#endif
7481 /* Just pressing <enter> in shell should check for jobs.
7482 * OTOH, in non-interactive shell this is useless
7483 * and only leads to extra job checks */
7484 if (pi->num_cmds == 0) {
7485 if (G_interactive_fd)
7486 goto check_jobs_and_continue;
7487 continue;
7488 }
7489
7490 /* After analyzing all keywords and conditions, we decided
7491 * to execute this pipe. NB: have to do checkjobs(NULL)
7492 * after run_pipe to collect any background children,
7493 * even if list execution is to be stopped. */
7494 debug_printf_exec(": run_pipe with %d members\n", pi->num_cmds);
7495 {
7496 int r;
7497#if ENABLE_HUSH_LOOPS
7498 G.flag_break_continue = 0;
7499#endif
7500 rcode = r = run_pipe(pi); /* NB: rcode is a smallint */
7501 if (r != -1) {
7502 /* We ran a builtin, function, or group.
7503 * rcode is already known
7504 * and we don't need to wait for anything. */
7505 G.last_exitcode = rcode;
7506 debug_printf_exec(": builtin/func exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007507 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007508#if ENABLE_HUSH_LOOPS
7509 /* Was it "break" or "continue"? */
7510 if (G.flag_break_continue) {
7511 smallint fbc = G.flag_break_continue;
7512 /* We might fall into outer *loop*,
7513 * don't want to break it too */
7514 if (loop_top) {
7515 G.depth_break_continue--;
7516 if (G.depth_break_continue == 0)
7517 G.flag_break_continue = 0;
7518 /* else: e.g. "continue 2" should *break* once, *then* continue */
7519 } /* else: "while... do... { we are here (innermost list is not a loop!) };...done" */
Denys Vlasenko3beab832013-04-07 18:16:58 +02007520 if (G.depth_break_continue != 0 || fbc == BC_BREAK) {
7521 checkjobs(NULL);
7522 break;
7523 }
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007524 /* "continue": simulate end of loop */
7525 rword = RES_DONE;
7526 continue;
7527 }
7528#endif
7529#if ENABLE_HUSH_FUNCTIONS
7530 if (G.flag_return_in_progress == 1) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007531 checkjobs(NULL);
7532 break;
7533 }
7534#endif
7535 } else if (pi->followup == PIPE_BG) {
7536 /* What does bash do with attempts to background builtins? */
7537 /* even bash 3.2 doesn't do that well with nested bg:
7538 * try "{ { sleep 10; echo DEEP; } & echo HERE; } &".
7539 * I'm NOT treating inner &'s as jobs */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007540 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007541#if ENABLE_HUSH_JOB
7542 if (G.run_list_level == 1)
7543 insert_bg_job(pi);
7544#endif
7545 /* Last command's pid goes to $! */
7546 G.last_bg_pid = pi->cmds[pi->num_cmds - 1].pid;
7547 G.last_exitcode = rcode = EXIT_SUCCESS;
7548 debug_printf_exec(": cmd&: exitcode EXIT_SUCCESS\n");
7549 } else {
7550#if ENABLE_HUSH_JOB
7551 if (G.run_list_level == 1 && G_interactive_fd) {
7552 /* Waits for completion, then fg's main shell */
7553 rcode = checkjobs_and_fg_shell(pi);
7554 debug_printf_exec(": checkjobs_and_fg_shell exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007555 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007556 } else
7557#endif
7558 { /* This one just waits for completion */
7559 rcode = checkjobs(pi);
7560 debug_printf_exec(": checkjobs exitcode %d\n", rcode);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007561 check_and_run_traps();
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007562 }
7563 G.last_exitcode = rcode;
7564 }
7565 }
7566
7567 /* Analyze how result affects subsequent commands */
7568#if ENABLE_HUSH_IF
7569 if (rword == RES_IF || rword == RES_ELIF)
7570 cond_code = rcode;
7571#endif
Denys Vlasenko3beab832013-04-07 18:16:58 +02007572 check_jobs_and_continue:
7573 checkjobs(NULL);
7574 dont_check_jobs_but_continue: ;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007575#if ENABLE_HUSH_LOOPS
7576 /* Beware of "while false; true; do ..."! */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007577 if (pi->next
7578 && (pi->next->res_word == RES_DO || pi->next->res_word == RES_DONE)
Denys Vlasenko56a3b822011-06-01 12:47:07 +02007579 /* check for RES_DONE is needed for "while ...; do \n done" case */
Denys Vlasenko00ae9892011-05-31 17:35:45 +02007580 ) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007581 if (rword == RES_WHILE) {
7582 if (rcode) {
7583 /* "while false; do...done" - exitcode 0 */
7584 G.last_exitcode = rcode = EXIT_SUCCESS;
7585 debug_printf_exec(": while expr is false: breaking (exitcode:EXIT_SUCCESS)\n");
Denys Vlasenko3beab832013-04-07 18:16:58 +02007586 break;
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007587 }
7588 }
7589 if (rword == RES_UNTIL) {
7590 if (!rcode) {
7591 debug_printf_exec(": until expr is true: breaking\n");
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007592 break;
7593 }
7594 }
7595 }
7596#endif
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007597 } /* for (pi) */
7598
7599#if ENABLE_HUSH_JOB
7600 G.run_list_level--;
7601#endif
7602#if ENABLE_HUSH_LOOPS
7603 if (loop_top)
7604 G.depth_of_loop--;
7605 free(for_list);
7606#endif
7607#if ENABLE_HUSH_CASE
7608 free(case_word);
7609#endif
7610 debug_leave();
7611 debug_printf_exec("run_list lvl %d return %d\n", G.run_list_level + 1, rcode);
7612 return rcode;
7613}
7614
7615/* Select which version we will use */
7616static int run_and_free_list(struct pipe *pi)
7617{
7618 int rcode = 0;
7619 debug_printf_exec("run_and_free_list entered\n");
Dan Fandrich85c62472010-11-20 13:05:17 -08007620 if (!G.o_opt[OPT_O_NOEXEC]) {
Denys Vlasenkob36abf22010-09-05 14:50:59 +02007621 debug_printf_exec(": run_list: 1st pipe with %d cmds\n", pi->num_cmds);
7622 rcode = run_list(pi);
7623 }
7624 /* free_pipe_list has the side effect of clearing memory.
7625 * In the long run that function can be merged with run_list,
7626 * but doing that now would hobble the debugging effort. */
7627 free_pipe_list(pi);
7628 debug_printf_exec("run_and_free_list return %d\n", rcode);
7629 return rcode;
7630}
7631
7632
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007633static void install_sighandlers(unsigned mask)
Eric Andersen52a97ca2001-06-22 06:49:26 +00007634{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007635 sighandler_t old_handler;
7636 unsigned sig = 0;
7637 while ((mask >>= 1) != 0) {
7638 sig++;
7639 if (!(mask & 1))
7640 continue;
Denys Vlasenko0806e402011-05-12 23:06:20 +02007641 old_handler = install_sighandler(sig, pick_sighandler(sig));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007642 /* POSIX allows shell to re-enable SIGCHLD
7643 * even if it was SIG_IGN on entry.
7644 * Therefore we skip IGN check for it:
7645 */
7646 if (sig == SIGCHLD)
7647 continue;
7648 if (old_handler == SIG_IGN) {
7649 /* oops... restore back to IGN, and record this fact */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007650 install_sighandler(sig, old_handler);
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007651 if (!G.traps)
7652 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7653 free(G.traps[sig]);
7654 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
7655 }
7656 }
7657}
7658
7659/* Called a few times only (or even once if "sh -c") */
7660static void install_special_sighandlers(void)
7661{
Denis Vlasenkof9375282009-04-05 19:13:39 +00007662 unsigned mask;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007663
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007664 /* Which signals are shell-special? */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007665 mask = (1 << SIGQUIT) | (1 << SIGCHLD);
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007666 if (G_interactive_fd) {
7667 mask |= SPECIAL_INTERACTIVE_SIGS;
7668 if (G_saved_tty_pgrp) /* we have ctty, job control sigs work */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007669 mask |= SPECIAL_JOBSTOP_SIGS;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007670 }
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007671 /* Careful, do not re-install handlers we already installed */
7672 if (G.special_sig_mask != mask) {
7673 unsigned diff = mask & ~G.special_sig_mask;
7674 G.special_sig_mask = mask;
7675 install_sighandlers(diff);
7676 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00007677}
7678
7679#if ENABLE_HUSH_JOB
7680/* helper */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007681/* Set handlers to restore tty pgrp and exit */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007682static void install_fatal_sighandlers(void)
Denis Vlasenkof9375282009-04-05 19:13:39 +00007683{
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007684 unsigned mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007685
7686 /* We will restore tty pgrp on these signals */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007687 mask = 0
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007688 + (1 << SIGILL ) * HUSH_DEBUG
7689 + (1 << SIGFPE ) * HUSH_DEBUG
7690 + (1 << SIGBUS ) * HUSH_DEBUG
7691 + (1 << SIGSEGV) * HUSH_DEBUG
7692 + (1 << SIGTRAP) * HUSH_DEBUG
7693 + (1 << SIGABRT)
7694 /* bash 3.2 seems to handle these just like 'fatal' ones */
7695 + (1 << SIGPIPE)
7696 + (1 << SIGALRM)
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007697 /* if we are interactive, SIGHUP, SIGTERM and SIGINT are special sigs.
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007698 * if we aren't interactive... but in this case
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007699 * we never want to restore pgrp on exit, and this fn is not called
7700 */
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007701 /*+ (1 << SIGHUP )*/
7702 /*+ (1 << SIGTERM)*/
7703 /*+ (1 << SIGINT )*/
7704 ;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007705 G_fatal_sig_mask = mask;
Denys Vlasenko54e9e122011-05-09 00:52:15 +02007706
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007707 install_sighandlers(mask);
Denis Vlasenkof9375282009-04-05 19:13:39 +00007708}
Denis Vlasenkob81b3df2007-04-28 16:48:04 +00007709#endif
Eric Andersenada18ff2001-05-21 16:18:22 +00007710
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007711static int set_mode(int state, char mode, const char *o_opt)
Denis Vlasenkod5762932009-03-31 11:22:57 +00007712{
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007713 int idx;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007714 switch (mode) {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007715 case 'n':
Dan Fandrich85c62472010-11-20 13:05:17 -08007716 G.o_opt[OPT_O_NOEXEC] = state;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007717 break;
7718 case 'x':
7719 IF_HUSH_MODE_X(G_x_mode = state;)
7720 break;
7721 case 'o':
7722 if (!o_opt) {
7723 /* "set -+o" without parameter.
7724 * in bash, set -o produces this output:
7725 * pipefail off
7726 * and set +o:
7727 * set +o pipefail
7728 * We always use the second form.
7729 */
7730 const char *p = o_opt_strings;
7731 idx = 0;
7732 while (*p) {
7733 printf("set %co %s\n", (G.o_opt[idx] ? '-' : '+'), p);
7734 idx++;
7735 p += strlen(p) + 1;
7736 }
7737 break;
7738 }
7739 idx = index_in_strings(o_opt_strings, o_opt);
7740 if (idx >= 0) {
7741 G.o_opt[idx] = state;
7742 break;
7743 }
7744 default:
7745 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00007746 }
7747 return EXIT_SUCCESS;
7748}
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00007749
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00007750int hush_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Matt Kraai2d91deb2001-08-01 17:21:35 +00007751int hush_main(int argc, char **argv)
Eric Andersen25f27032001-04-26 23:22:31 +00007752{
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007753 enum {
7754 OPT_login = (1 << 0),
7755 };
7756 unsigned flags;
Eric Andersen25f27032001-04-26 23:22:31 +00007757 int opt;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007758 unsigned builtin_argc;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007759 char **e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007760 struct variable *cur_var;
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007761 struct variable *shell_ver;
Eric Andersenbc604a22001-05-16 05:24:03 +00007762
Denis Vlasenko574f2f42008-02-27 18:41:59 +00007763 INIT_G();
Denys Vlasenko10c01312011-05-11 11:49:21 +02007764 if (EXIT_SUCCESS != 0) /* if EXIT_SUCCESS == 0, it is already done */
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007765 G.last_exitcode = EXIT_SUCCESS;
Denys Vlasenko10c01312011-05-11 11:49:21 +02007766#if ENABLE_HUSH_FAST
7767 G.count_SIGCHLD++; /* ensure it is != G.handled_SIGCHLD */
7768#endif
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007769#if !BB_MMU
7770 G.argv0_for_re_execing = argv[0];
7771#endif
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007772 /* Deal with HUSH_VERSION */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007773 shell_ver = xzalloc(sizeof(*shell_ver));
7774 shell_ver->flg_export = 1;
7775 shell_ver->flg_read_only = 1;
Denys Vlasenko4f870492010-09-10 11:06:01 +02007776 /* Code which handles ${var<op>...} needs writable values for all variables,
Denys Vlasenko36f774a2010-09-05 14:45:38 +02007777 * therefore we xstrdup: */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007778 shell_ver->varstr = xstrdup(hush_version_str);
Denys Vlasenko605067b2010-09-06 12:10:51 +02007779 /* Create shell local variables from the values
7780 * currently living in the environment */
Denis Vlasenkof886fd22008-10-13 12:36:05 +00007781 debug_printf_env("unsetenv '%s'\n", "HUSH_VERSION");
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007782 unsetenv("HUSH_VERSION"); /* in case it exists in initial env */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007783 G.top_var = shell_ver;
Denis Vlasenko87a86552008-07-29 19:43:10 +00007784 cur_var = G.top_var;
Denis Vlasenko0a83fc32007-05-25 11:12:32 +00007785 e = environ;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007786 if (e) while (*e) {
7787 char *value = strchr(*e, '=');
7788 if (value) { /* paranoia */
7789 cur_var->next = xzalloc(sizeof(*cur_var));
7790 cur_var = cur_var->next;
Denis Vlasenko28c0f0f2007-05-25 02:46:01 +00007791 cur_var->varstr = *e;
Denis Vlasenkod76c0492007-05-25 02:16:25 +00007792 cur_var->max_len = strlen(*e);
7793 cur_var->flg_export = 1;
7794 }
7795 e++;
7796 }
Denys Vlasenko605067b2010-09-06 12:10:51 +02007797 /* (Re)insert HUSH_VERSION into env (AFTER we scanned the env!) */
Denys Vlasenko75eb9d22010-12-21 21:18:12 +01007798 debug_printf_env("putenv '%s'\n", shell_ver->varstr);
7799 putenv(shell_ver->varstr);
Denys Vlasenko6db47842009-09-05 20:15:17 +02007800
7801 /* Export PWD */
7802 set_pwd_var(/*exp:*/ 1);
Denys Vlasenko3fa97af2014-04-15 11:43:29 +02007803
7804#if ENABLE_HUSH_BASH_COMPAT
7805 /* Set (but not export) HOSTNAME unless already set */
7806 if (!get_local_var_value("HOSTNAME")) {
7807 struct utsname uts;
7808 uname(&uts);
7809 set_local_var_from_halves("HOSTNAME", uts.nodename);
7810 }
Denys Vlasenko6db47842009-09-05 20:15:17 +02007811 /* bash also exports SHLVL and _,
7812 * and sets (but doesn't export) the following variables:
7813 * BASH=/bin/bash
7814 * BASH_VERSINFO=([0]="3" [1]="2" [2]="0" [3]="1" [4]="release" [5]="i386-pc-linux-gnu")
7815 * BASH_VERSION='3.2.0(1)-release'
7816 * HOSTTYPE=i386
7817 * MACHTYPE=i386-pc-linux-gnu
7818 * OSTYPE=linux-gnu
Denys Vlasenkodea47882009-10-09 15:40:49 +02007819 * PPID=<NNNNN> - we also do it elsewhere
Denys Vlasenko6db47842009-09-05 20:15:17 +02007820 * EUID=<NNNNN>
7821 * UID=<NNNNN>
7822 * GROUPS=()
7823 * LINES=<NNN>
7824 * COLUMNS=<NNN>
7825 * BASH_ARGC=()
7826 * BASH_ARGV=()
7827 * BASH_LINENO=()
7828 * BASH_SOURCE=()
7829 * DIRSTACK=()
7830 * PIPESTATUS=([0]="0")
7831 * HISTFILE=/<xxx>/.bash_history
7832 * HISTFILESIZE=500
7833 * HISTSIZE=500
7834 * MAILCHECK=60
7835 * PATH=/usr/gnu/bin:/usr/local/bin:/bin:/usr/bin:.
7836 * SHELL=/bin/bash
7837 * SHELLOPTS=braceexpand:emacs:hashall:histexpand:history:interactive-comments:monitor
7838 * TERM=dumb
7839 * OPTERR=1
7840 * OPTIND=1
7841 * IFS=$' \t\n'
7842 * PS1='\s-\v\$ '
7843 * PS2='> '
7844 * PS4='+ '
7845 */
Denys Vlasenko3fa97af2014-04-15 11:43:29 +02007846#endif
Denys Vlasenko6db47842009-09-05 20:15:17 +02007847
Denis Vlasenko38f63192007-01-22 09:03:07 +00007848#if ENABLE_FEATURE_EDITING
Denys Vlasenkoe45af7a2011-09-04 16:15:24 +02007849 G.line_input_state = new_line_input_t(FOR_SHELL);
Denis Vlasenko8e1c7152007-01-22 07:21:38 +00007850#endif
Denys Vlasenko99862cb2010-09-12 17:34:13 +02007851
Eric Andersen94ac2442001-05-22 19:05:18 +00007852 /* Initialize some more globals to non-zero values */
Mike Frysinger67c1c7b2009-04-24 06:26:18 +00007853 cmdedit_update_prompt();
Denis Vlasenkoc8be5ee2007-05-17 15:38:46 +00007854
Denis Vlasenkoed782372009-04-10 00:45:02 +00007855 if (setjmp(die_jmp)) {
7856 /* xfunc has failed! die die die */
7857 /* no EXIT traps, this is an escape hatch! */
7858 G.exiting = 1;
7859 hush_exit(xfunc_error_retval);
7860 }
7861
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007862 /* Shell is non-interactive at first. We need to call
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007863 * install_special_sighandlers() if we are going to execute "sh <script>",
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007864 * "sh -c <cmds>" or login shell's /etc/profile and friends.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007865 * If we later decide that we are interactive, we run install_special_sighandlers()
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00007866 * in order to intercept (more) signals.
7867 */
7868
7869 /* Parse options */
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007870 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/sh.html */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007871 flags = (argv[0] && argv[0][0] == '-') ? OPT_login : 0;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007872 builtin_argc = 0;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007873 while (1) {
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007874 opt = getopt(argc, argv, "+c:xinsl"
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007875#if !BB_MMU
Denis Vlasenkobc569742009-04-12 20:35:19 +00007876 "<:$:R:V:"
7877# if ENABLE_HUSH_FUNCTIONS
7878 "F:"
7879# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007880#endif
7881 );
7882 if (opt <= 0)
7883 break;
Eric Andersen25f27032001-04-26 23:22:31 +00007884 switch (opt) {
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007885 case 'c':
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007886 /* Possibilities:
7887 * sh ... -c 'script'
7888 * sh ... -c 'script' ARG0 [ARG1...]
7889 * On NOMMU, if builtin_argc != 0,
Denys Vlasenko17323a62010-01-28 01:57:05 +01007890 * sh ... -c 'builtin' BARGV... "" ARG0 [ARG1...]
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007891 * "" needs to be replaced with NULL
7892 * and BARGV vector fed to builtin function.
Denys Vlasenko17323a62010-01-28 01:57:05 +01007893 * Note: the form without ARG0 never happens:
7894 * sh ... -c 'builtin' BARGV... ""
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007895 */
Denys Vlasenkodea47882009-10-09 15:40:49 +02007896 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00007897 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02007898 G.root_ppid = getppid();
7899 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00007900 G.global_argv = argv + optind;
7901 G.global_argc = argc - optind;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007902 if (builtin_argc) {
7903 /* -c 'builtin' [BARGV...] "" ARG0 [ARG1...] */
7904 const struct built_in_command *x;
7905
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007906 install_special_sighandlers();
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007907 x = find_builtin(optarg);
7908 if (x) { /* paranoia */
7909 G.global_argc -= builtin_argc; /* skip [BARGV...] "" */
7910 G.global_argv += builtin_argc;
7911 G.global_argv[-1] = NULL; /* replace "" */
Denys Vlasenko8ee2ada2011-02-07 02:03:51 +01007912 fflush_all();
Denys Vlasenko17323a62010-01-28 01:57:05 +01007913 G.last_exitcode = x->b_function(argv + optind - 1);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007914 }
7915 goto final_return;
7916 }
7917 if (!G.global_argv[0]) {
7918 /* -c 'script' (no params): prevent empty $0 */
7919 G.global_argv--; /* points to argv[i] of 'script' */
7920 G.global_argv[0] = argv[0];
Denys Vlasenko5ae8f1c2010-05-22 06:32:11 +02007921 G.global_argc++;
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007922 } /* else -c 'script' ARG0 [ARG1...]: $0 is ARG0 */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007923 install_special_sighandlers();
Denis Vlasenkob6e65562009-04-03 16:49:04 +00007924 parse_and_run_string(optarg);
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007925 goto final_return;
7926 case 'i':
Denis Vlasenkoc666f712007-05-16 22:18:54 +00007927 /* Well, we cannot just declare interactiveness,
7928 * we have to have some stuff (ctty, etc) */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00007929 /* G_interactive_fd++; */
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007930 break;
Mike Frysinger19a7ea12009-03-28 13:02:11 +00007931 case 's':
7932 /* "-s" means "read from stdin", but this is how we always
7933 * operate, so simply do nothing here. */
7934 break;
Denys Vlasenkof58f7052011-05-12 02:10:33 +02007935 case 'l':
7936 flags |= OPT_login;
7937 break;
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007938#if !BB_MMU
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007939 case '<': /* "big heredoc" support */
Denys Vlasenko729ecb82010-06-07 14:14:26 +02007940 full_write1_str(optarg);
Denis Vlasenko50f3aa42009-04-07 10:52:40 +00007941 _exit(0);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007942 case '$': {
7943 unsigned long long empty_trap_mask;
7944
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007945 G.root_pid = bb_strtou(optarg, &optarg, 16);
7946 optarg++;
Denys Vlasenkodea47882009-10-09 15:40:49 +02007947 G.root_ppid = bb_strtou(optarg, &optarg, 16);
7948 optarg++;
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007949 G.last_bg_pid = bb_strtou(optarg, &optarg, 16);
7950 optarg++;
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00007951 G.last_exitcode = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02007952 optarg++;
7953 builtin_argc = bb_strtou(optarg, &optarg, 16);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007954 optarg++;
7955 empty_trap_mask = bb_strtoull(optarg, &optarg, 16);
7956 if (empty_trap_mask != 0) {
7957 int sig;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02007958 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007959 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
7960 for (sig = 1; sig < NSIG; sig++) {
7961 if (empty_trap_mask & (1LL << sig)) {
7962 G.traps[sig] = xzalloc(1); /* == xstrdup(""); */
Denys Vlasenko0806e402011-05-12 23:06:20 +02007963 install_sighandler(sig, SIG_IGN);
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007964 }
7965 }
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007966 }
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007967# if ENABLE_HUSH_LOOPS
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007968 optarg++;
7969 G.depth_of_loop = bb_strtou(optarg, &optarg, 16);
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00007970# endif
Denis Vlasenko34e573d2009-04-06 12:56:28 +00007971 break;
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01007972 }
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007973 case 'R':
7974 case 'V':
Denys Vlasenko295fef82009-06-03 12:47:26 +02007975 set_local_var(xstrdup(optarg), /*exp:*/ 0, /*lvl:*/ 0, /*ro:*/ opt == 'R');
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007976 break;
Denis Vlasenkobc569742009-04-12 20:35:19 +00007977# if ENABLE_HUSH_FUNCTIONS
7978 case 'F': {
7979 struct function *funcp = new_function(optarg);
7980 /* funcp->name is already set to optarg */
7981 /* funcp->body is set to NULL. It's a special case. */
7982 funcp->body_as_string = argv[optind];
7983 optind++;
7984 break;
7985 }
7986# endif
Denis Vlasenko0bb4a232009-04-05 01:42:59 +00007987#endif
Mike Frysingerad88d5a2009-03-28 13:44:51 +00007988 case 'n':
7989 case 'x':
Denys Vlasenko6696eac2010-11-14 02:01:50 +01007990 if (set_mode(1, opt, NULL) == 0) /* no error */
Mike Frysingerad88d5a2009-03-28 13:44:51 +00007991 break;
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007992 default:
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007993#ifndef BB_VER
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007994 fprintf(stderr, "Usage: sh [FILE]...\n"
7995 " or: sh -c command [args]...\n\n");
7996 exit(EXIT_FAILURE);
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007997#else
Denis Vlasenkofbf6dea2007-04-13 19:56:56 +00007998 bb_show_usage();
Eric Andersen9ffb7dd2001-05-19 03:00:46 +00007999#endif
Eric Andersen25f27032001-04-26 23:22:31 +00008000 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008001 } /* option parsing loop */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008002
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008003 /* Skip options. Try "hush -l": $1 should not be "-l"! */
8004 G.global_argc = argc - (optind - 1);
8005 G.global_argv = argv + (optind - 1);
8006 G.global_argv[0] = argv[0];
8007
Denys Vlasenkodea47882009-10-09 15:40:49 +02008008 if (!G.root_pid) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008009 G.root_pid = getpid();
Denys Vlasenkodea47882009-10-09 15:40:49 +02008010 G.root_ppid = getppid();
8011 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008012
8013 /* If we are login shell... */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008014 if (flags & OPT_login) {
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008015 FILE *input;
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008016 debug_printf("sourcing /etc/profile\n");
8017 input = fopen_for_read("/etc/profile");
8018 if (input != NULL) {
8019 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008020 install_special_sighandlers();
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008021 parse_and_run_file(input);
8022 fclose(input);
8023 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008024 /* bash: after sourcing /etc/profile,
8025 * tries to source (in the given order):
8026 * ~/.bash_profile, ~/.bash_login, ~/.profile,
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008027 * stopping on first found. --noprofile turns this off.
Denis Vlasenkof9375282009-04-05 19:13:39 +00008028 * bash also sources ~/.bash_logout on exit.
8029 * If called as sh, skips .bash_XXX files.
8030 */
Denis Vlasenko46f9b6d2009-04-05 10:39:03 +00008031 }
8032
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008033 if (G.global_argv[1]) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008034 FILE *input;
8035 /*
Denis Vlasenkod3f973e2009-04-06 10:21:42 +00008036 * "bash <script>" (which is never interactive (unless -i?))
8037 * sources $BASH_ENV here (without scanning $PATH).
Denis Vlasenkof9375282009-04-05 19:13:39 +00008038 * If called as sh, does the same but with $ENV.
8039 */
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008040 G.global_argc--;
8041 G.global_argv++;
8042 debug_printf("running script '%s'\n", G.global_argv[0]);
8043 input = xfopen_for_read(G.global_argv[0]);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008044 close_on_exec_on(fileno(input));
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008045 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008046 parse_and_run_file(input);
8047#if ENABLE_FEATURE_CLEAN_UP
8048 fclose(input);
8049#endif
8050 goto final_return;
8051 }
8052
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008053 /* Up to here, shell was non-interactive. Now it may become one.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008054 * NB: don't forget to (re)run install_special_sighandlers() as needed.
Denis Vlasenkoc4a7af52009-04-05 20:33:27 +00008055 */
Denis Vlasenkof9375282009-04-05 19:13:39 +00008056
Denys Vlasenko28a105d2009-06-01 11:26:30 +02008057 /* A shell is interactive if the '-i' flag was given,
8058 * or if all of the following conditions are met:
Denis Vlasenko55b2de72007-04-18 17:21:28 +00008059 * no -c command
Eric Andersen25f27032001-04-26 23:22:31 +00008060 * no arguments remaining or the -s flag given
8061 * standard input is a terminal
8062 * standard output is a terminal
Denis Vlasenkof9375282009-04-05 19:13:39 +00008063 * Refer to Posix.2, the description of the 'sh' utility.
8064 */
8065#if ENABLE_HUSH_JOB
8066 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Mike Frysinger38478a62009-05-20 04:48:06 -04008067 G_saved_tty_pgrp = tcgetpgrp(STDIN_FILENO);
8068 debug_printf("saved_tty_pgrp:%d\n", G_saved_tty_pgrp);
8069 if (G_saved_tty_pgrp < 0)
8070 G_saved_tty_pgrp = 0;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008071
8072 /* try to dup stdin to high fd#, >= 255 */
8073 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8074 if (G_interactive_fd < 0) {
8075 /* try to dup to any fd */
8076 G_interactive_fd = dup(STDIN_FILENO);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008077 if (G_interactive_fd < 0) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008078 /* give up */
8079 G_interactive_fd = 0;
Mike Frysinger38478a62009-05-20 04:48:06 -04008080 G_saved_tty_pgrp = 0;
Denis Vlasenko54e7ffb2007-04-21 00:03:36 +00008081 }
8082 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008083// TODO: track & disallow any attempts of user
8084// to (inadvertently) close/redirect G_interactive_fd
Eric Andersen25f27032001-04-26 23:22:31 +00008085 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008086 debug_printf("interactive_fd:%d\n", G_interactive_fd);
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008087 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008088 close_on_exec_on(G_interactive_fd);
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008089
Mike Frysinger38478a62009-05-20 04:48:06 -04008090 if (G_saved_tty_pgrp) {
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008091 /* If we were run as 'hush &', sleep until we are
8092 * in the foreground (tty pgrp == our pgrp).
8093 * If we get started under a job aware app (like bash),
8094 * make sure we are now in charge so we don't fight over
8095 * who gets the foreground */
8096 while (1) {
8097 pid_t shell_pgrp = getpgrp();
Mike Frysinger38478a62009-05-20 04:48:06 -04008098 G_saved_tty_pgrp = tcgetpgrp(G_interactive_fd);
8099 if (G_saved_tty_pgrp == shell_pgrp)
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008100 break;
8101 /* send TTIN to ourself (should stop us) */
8102 kill(- shell_pgrp, SIGTTIN);
8103 }
Denis Vlasenkof9375282009-04-05 19:13:39 +00008104 }
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008105
Denys Vlasenkof58f7052011-05-12 02:10:33 +02008106 /* Install more signal handlers */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008107 install_special_sighandlers();
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 /* Set other signals to restore saved_tty_pgrp */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008111 install_fatal_sighandlers();
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008112 /* Put ourselves in our own process group
8113 * (bash, too, does this only if ctty is available) */
8114 bb_setpgrp(); /* is the same as setpgid(our_pid, our_pid); */
8115 /* Grab control of the terminal */
8116 tcsetpgrp(G_interactive_fd, getpid());
8117 }
Denis Vlasenko4ecfcdc2008-02-11 08:32:31 +00008118 /* -1 is special - makes xfuncs longjmp, not exit
Denis Vlasenkoc04163a2008-02-11 08:30:53 +00008119 * (we reset die_sleep = 0 whereever we [v]fork) */
Denis Vlasenkoaf07b7c2009-04-07 13:26:18 +00008120 enable_restore_tty_pgrp_on_exit(); /* sets die_sleep = -1 */
Denys Vlasenko4840ae82011-09-04 15:28:03 +02008121
8122# if ENABLE_HUSH_SAVEHISTORY && MAX_HISTORY > 0
8123 {
8124 const char *hp = get_local_var_value("HISTFILE");
8125 if (!hp) {
8126 hp = get_local_var_value("HOME");
8127 if (hp)
8128 hp = concat_path_file(hp, ".hush_history");
8129 } else {
8130 hp = xstrdup(hp);
8131 }
8132 if (hp) {
8133 G.line_input_state->hist_file = hp;
Denys Vlasenko4840ae82011-09-04 15:28:03 +02008134 //set_local_var(xasprintf("HISTFILE=%s", ...));
8135 }
8136# if ENABLE_FEATURE_SH_HISTFILESIZE
8137 hp = get_local_var_value("HISTFILESIZE");
8138 G.line_input_state->max_history = size_from_HISTFILESIZE(hp);
8139# endif
8140 }
8141# endif
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008142 } else {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008143 install_special_sighandlers();
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008144 }
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008145#elif ENABLE_HUSH_INTERACTIVE
Denis Vlasenkof9375282009-04-05 19:13:39 +00008146 /* No job control compiled in, only prompt/line editing */
8147 if (isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)) {
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008148 G_interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
8149 if (G_interactive_fd < 0) {
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008150 /* try to dup to any fd */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008151 G_interactive_fd = dup(STDIN_FILENO);
8152 if (G_interactive_fd < 0)
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008153 /* give up */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008154 G_interactive_fd = 0;
Denis Vlasenkoe3f2f892007-04-28 16:48:27 +00008155 }
8156 }
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008157 if (G_interactive_fd) {
Denis Vlasenkof9375282009-04-05 19:13:39 +00008158 close_on_exec_on(G_interactive_fd);
Denis Vlasenkof9375282009-04-05 19:13:39 +00008159 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008160 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008161#else
8162 /* We have interactiveness code disabled */
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008163 install_special_sighandlers();
Denis Vlasenkof9375282009-04-05 19:13:39 +00008164#endif
8165 /* bash:
8166 * if interactive but not a login shell, sources ~/.bashrc
8167 * (--norc turns this off, --rcfile <file> overrides)
8168 */
8169
8170 if (!ENABLE_FEATURE_SH_EXTRA_QUIET && G_interactive_fd) {
Denys Vlasenkoc34c0332009-09-29 12:25:30 +02008171 /* note: ash and hush share this string */
8172 printf("\n\n%s %s\n"
8173 IF_HUSH_HELP("Enter 'help' for a list of built-in commands.\n")
8174 "\n",
8175 bb_banner,
8176 "hush - the humble shell"
8177 );
Mike Frysingerb2705e12009-03-23 08:44:02 +00008178 }
8179
Denis Vlasenkof9375282009-04-05 19:13:39 +00008180 parse_and_run_file(stdin);
Eric Andersen25f27032001-04-26 23:22:31 +00008181
Denis Vlasenkod76c0492007-05-25 02:16:25 +00008182 final_return:
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008183 hush_exit(G.last_exitcode);
Eric Andersen25f27032001-04-26 23:22:31 +00008184}
Denis Vlasenko96702ca2007-11-23 23:28:55 +00008185
8186
Denys Vlasenko1cc4b132009-08-21 00:05:51 +02008187#if ENABLE_MSH
8188int msh_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
8189int msh_main(int argc, char **argv)
8190{
8191 //bb_error_msg("msh is deprecated, please use hush instead");
8192 return hush_main(argc, argv);
8193}
8194#endif
8195
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008196
8197/*
8198 * Built-ins
8199 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008200static int FAST_FUNC builtin_true(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008201{
8202 return 0;
8203}
8204
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008205static int run_applet_main(char **argv, int (*applet_main_func)(int argc, char **argv))
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008206{
8207 int argc = 0;
8208 while (*argv) {
8209 argc++;
8210 argv++;
8211 }
Denys Vlasenko8bc7f2c2009-10-19 13:20:52 +02008212 return applet_main_func(argc, argv - argc);
Mike Frysingerccb19592009-10-15 03:31:15 -04008213}
8214
8215static int FAST_FUNC builtin_test(char **argv)
8216{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008217 return run_applet_main(argv, test_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008218}
8219
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008220static int FAST_FUNC builtin_echo(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008221{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008222 return run_applet_main(argv, echo_main);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008223}
8224
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008225#if ENABLE_PRINTF
8226static int FAST_FUNC builtin_printf(char **argv)
8227{
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008228 return run_applet_main(argv, printf_main);
Mike Frysinger4ebc76c2009-10-15 03:32:39 -04008229}
8230#endif
8231
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008232static char **skip_dash_dash(char **argv)
8233{
8234 argv++;
8235 if (argv[0] && argv[0][0] == '-' && argv[0][1] == '-' && argv[0][2] == '\0')
8236 argv++;
8237 return argv;
8238}
8239
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008240static int FAST_FUNC builtin_eval(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008241{
8242 int rcode = EXIT_SUCCESS;
8243
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008244 argv = skip_dash_dash(argv);
8245 if (*argv) {
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008246 char *str = expand_strvec_to_string(argv);
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008247 /* bash:
8248 * eval "echo Hi; done" ("done" is syntax error):
8249 * "echo Hi" will not execute too.
8250 */
8251 parse_and_run_string(str);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008252 free(str);
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008253 rcode = G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008254 }
8255 return rcode;
8256}
8257
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008258static int FAST_FUNC builtin_cd(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008259{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008260 const char *newdir;
8261
8262 argv = skip_dash_dash(argv);
8263 newdir = argv[0];
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008264 if (newdir == NULL) {
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008265 /* bash does nothing (exitcode 0) if HOME is ""; if it's unset,
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008266 * bash says "bash: cd: HOME not set" and does nothing
8267 * (exitcode 1)
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008268 */
Denys Vlasenko90a99042009-09-06 02:36:23 +02008269 const char *home = get_local_var_value("HOME");
8270 newdir = home ? home : "/";
Denis Vlasenkob0a64782009-04-06 11:33:07 +00008271 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008272 if (chdir(newdir)) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008273 /* Mimic bash message exactly */
8274 bb_perror_msg("cd: %s", newdir);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008275 return EXIT_FAILURE;
8276 }
Denys Vlasenko6db47842009-09-05 20:15:17 +02008277 /* Read current dir (get_cwd(1) is inside) and set PWD.
8278 * Note: do not enforce exporting. If PWD was unset or unexported,
8279 * set it again, but do not export. bash does the same.
8280 */
8281 set_pwd_var(/*exp:*/ 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008282 return EXIT_SUCCESS;
8283}
8284
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008285static int FAST_FUNC builtin_exec(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008286{
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008287 argv = skip_dash_dash(argv);
8288 if (argv[0] == NULL)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008289 return EXIT_SUCCESS; /* bash does this */
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008290
Denys Vlasenkof37eb392009-10-18 11:46:35 +02008291 /* Careful: we can end up here after [v]fork. Do not restore
8292 * tty pgrp then, only top-level shell process does that */
8293 if (G_saved_tty_pgrp && getpid() == G.root_pid)
8294 tcsetpgrp(G_interactive_fd, G_saved_tty_pgrp);
8295
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008296 /* TODO: if exec fails, bash does NOT exit! We do.
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008297 * We'll need to undo trap cleanup (it's inside execvp_or_die)
Denys Vlasenko3ef4f772009-10-19 23:09:06 +02008298 * and tcsetpgrp, and this is inherently racy.
8299 */
8300 execvp_or_die(argv);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008301}
8302
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008303static int FAST_FUNC builtin_exit(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008304{
Denis Vlasenkocd418a22009-04-06 18:08:35 +00008305 debug_printf_exec("%s()\n", __func__);
Denis Vlasenko40e84372009-04-18 11:23:38 +00008306
8307 /* interactive bash:
8308 * # trap "echo EEE" EXIT
8309 * # exit
8310 * exit
8311 * There are stopped jobs.
8312 * (if there are _stopped_ jobs, running ones don't count)
8313 * # exit
8314 * exit
Denys Vlasenko6830ade2013-01-15 13:58:01 +01008315 * EEE (then bash exits)
Denis Vlasenko40e84372009-04-18 11:23:38 +00008316 *
Denys Vlasenkoa110c902010-09-12 15:38:04 +02008317 * TODO: we can use G.exiting = -1 as indicator "last cmd was exit"
Denis Vlasenko40e84372009-04-18 11:23:38 +00008318 */
Denis Vlasenkoefea9d22009-04-09 13:43:11 +00008319
8320 /* note: EXIT trap is run by hush_exit */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008321 argv = skip_dash_dash(argv);
8322 if (argv[0] == NULL)
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008323 hush_exit(G.last_exitcode);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008324 /* mimic bash: exit 123abc == exit 255 + error msg */
8325 xfunc_error_retval = 255;
8326 /* bash: exit -2 == exit 254, no error msg */
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008327 hush_exit(xatoi(argv[0]) & 0xff);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008328}
8329
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008330static void print_escaped(const char *s)
8331{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008332 if (*s == '\'')
8333 goto squote;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008334 do {
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008335 const char *p = strchrnul(s, '\'');
8336 /* print 'xxxx', possibly just '' */
8337 printf("'%.*s'", (int)(p - s), s);
8338 if (*p == '\0')
8339 break;
8340 s = p;
8341 squote:
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008342 /* s points to '; print "'''...'''" */
8343 putchar('"');
8344 do putchar('\''); while (*++s == '\'');
8345 putchar('"');
8346 } while (*s);
8347}
8348
Denys Vlasenko295fef82009-06-03 12:47:26 +02008349#if !ENABLE_HUSH_LOCAL
8350#define helper_export_local(argv, exp, lvl) \
8351 helper_export_local(argv, exp)
8352#endif
8353static void helper_export_local(char **argv, int exp, int lvl)
8354{
8355 do {
8356 char *name = *argv;
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008357 char *name_end = strchrnul(name, '=');
Denys Vlasenko295fef82009-06-03 12:47:26 +02008358
8359 /* So far we do not check that name is valid (TODO?) */
8360
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008361 if (*name_end == '\0') {
8362 struct variable *var, **vpp;
Denys Vlasenko295fef82009-06-03 12:47:26 +02008363
Denys Vlasenko27c56f12010-09-07 09:56:34 +02008364 vpp = get_ptr_to_local_var(name, name_end - name);
8365 var = vpp ? *vpp : NULL;
8366
Denys Vlasenko295fef82009-06-03 12:47:26 +02008367 if (exp == -1) { /* unexporting? */
8368 /* export -n NAME (without =VALUE) */
8369 if (var) {
8370 var->flg_export = 0;
8371 debug_printf_env("%s: unsetenv '%s'\n", __func__, name);
8372 unsetenv(name);
8373 } /* else: export -n NOT_EXISTING_VAR: no-op */
8374 continue;
8375 }
8376 if (exp == 1) { /* exporting? */
8377 /* export NAME (without =VALUE) */
8378 if (var) {
8379 var->flg_export = 1;
8380 debug_printf_env("%s: putenv '%s'\n", __func__, var->varstr);
8381 putenv(var->varstr);
8382 continue;
8383 }
8384 }
8385 /* Exporting non-existing variable.
8386 * bash does not put it in environment,
8387 * but remembers that it is exported,
8388 * and does put it in env when it is set later.
8389 * We just set it to "" and export. */
8390 /* Or, it's "local NAME" (without =VALUE).
8391 * bash sets the value to "". */
8392 name = xasprintf("%s=", name);
8393 } else {
8394 /* (Un)exporting/making local NAME=VALUE */
8395 name = xstrdup(name);
8396 }
8397 set_local_var(name, /*exp:*/ exp, /*lvl:*/ lvl, /*ro:*/ 0);
8398 } while (*++argv);
8399}
8400
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008401static int FAST_FUNC builtin_export(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008402{
Denis Vlasenkoad4bd052009-04-20 22:04:21 +00008403 unsigned opt_unexport;
8404
Denys Vlasenkodf5131c2009-06-07 16:04:17 +02008405#if ENABLE_HUSH_EXPORT_N
8406 /* "!": do not abort on errors */
8407 opt_unexport = getopt32(argv, "!n");
8408 if (opt_unexport == (uint32_t)-1)
8409 return EXIT_FAILURE;
8410 argv += optind;
8411#else
8412 opt_unexport = 0;
8413 argv++;
8414#endif
8415
8416 if (argv[0] == NULL) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008417 char **e = environ;
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008418 if (e) {
8419 while (*e) {
8420#if 0
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008421 puts(*e++);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008422#else
8423 /* ash emits: export VAR='VAL'
8424 * bash: declare -x VAR="VAL"
8425 * we follow ash example */
8426 const char *s = *e++;
8427 const char *p = strchr(s, '=');
8428
8429 if (!p) /* wtf? take next variable */
8430 continue;
8431 /* export var= */
8432 printf("export %.*s", (int)(p - s) + 1, s);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008433 print_escaped(p + 1);
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008434 putchar('\n');
8435#endif
8436 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008437 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko0b677d82009-04-10 13:49:10 +00008438 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008439 return EXIT_SUCCESS;
8440 }
8441
Denys Vlasenko295fef82009-06-03 12:47:26 +02008442 helper_export_local(argv, (opt_unexport ? -1 : 1), 0);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008443
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008444 return EXIT_SUCCESS;
8445}
8446
Denys Vlasenko295fef82009-06-03 12:47:26 +02008447#if ENABLE_HUSH_LOCAL
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008448static int FAST_FUNC builtin_local(char **argv)
Denys Vlasenko295fef82009-06-03 12:47:26 +02008449{
8450 if (G.func_nest_level == 0) {
8451 bb_error_msg("%s: not in a function", argv[0]);
8452 return EXIT_FAILURE; /* bash compat */
8453 }
8454 helper_export_local(argv, 0, G.func_nest_level);
8455 return EXIT_SUCCESS;
8456}
8457#endif
8458
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008459static int FAST_FUNC builtin_trap(char **argv)
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008460{
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008461 int sig;
8462 char *new_cmd;
8463
8464 if (!G.traps)
8465 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
8466
8467 argv++;
8468 if (!*argv) {
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008469 int i;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008470 /* No args: print all trapped */
8471 for (i = 0; i < NSIG; ++i) {
8472 if (G.traps[i]) {
8473 printf("trap -- ");
8474 print_escaped(G.traps[i]);
Denys Vlasenkoe74aaf92009-09-27 02:05:45 +02008475 /* note: bash adds "SIG", but only if invoked
8476 * as "bash". If called as "sh", or if set -o posix,
8477 * then it prints short signal names.
8478 * We are printing short names: */
8479 printf(" %s\n", get_signame(i));
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008480 }
8481 }
Denys Vlasenko8131eea2009-11-02 14:19:51 +01008482 /*fflush_all(); - done after each builtin anyway */
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008483 return EXIT_SUCCESS;
8484 }
8485
8486 new_cmd = NULL;
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008487 /* If first arg is a number: reset all specified signals */
8488 sig = bb_strtou(*argv, NULL, 10);
8489 if (errno == 0) {
8490 int ret;
8491 process_sig_list:
8492 ret = EXIT_SUCCESS;
8493 while (*argv) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008494 sighandler_t handler;
8495
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008496 sig = get_signum(*argv++);
8497 if (sig < 0 || sig >= NSIG) {
8498 ret = EXIT_FAILURE;
8499 /* Mimic bash message exactly */
Denis Vlasenko6008d8a2009-04-18 13:05:10 +00008500 bb_perror_msg("trap: %s: invalid signal specification", argv[-1]);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008501 continue;
8502 }
8503
8504 free(G.traps[sig]);
8505 G.traps[sig] = xstrdup(new_cmd);
8506
Denys Vlasenkoe89a2412010-01-12 15:19:31 +01008507 debug_printf("trap: setting SIG%s (%i) to '%s'\n",
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008508 get_signame(sig), sig, G.traps[sig]);
8509
8510 /* There is no signal for 0 (EXIT) */
8511 if (sig == 0)
8512 continue;
8513
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008514 if (new_cmd)
8515 handler = (new_cmd[0] ? record_pending_signo : SIG_IGN);
8516 else
8517 /* We are removing trap handler */
8518 handler = pick_sighandler(sig);
Denys Vlasenko0806e402011-05-12 23:06:20 +02008519 install_sighandler(sig, handler);
Denis Vlasenko38e626d2009-04-18 12:58:19 +00008520 }
8521 return ret;
8522 }
8523
8524 if (!argv[1]) { /* no second arg */
8525 bb_error_msg("trap: invalid arguments");
8526 return EXIT_FAILURE;
8527 }
8528
8529 /* First arg is "-": reset all specified to default */
8530 /* First arg is "--": skip it, the rest is "handler SIGs..." */
8531 /* Everything else: set arg as signal handler
8532 * (includes "" case, which ignores signal) */
8533 if (argv[0][0] == '-') {
8534 if (argv[0][1] == '\0') { /* "-" */
8535 /* new_cmd remains NULL: "reset these sigs" */
8536 goto reset_traps;
8537 }
8538 if (argv[0][1] == '-' && argv[0][2] == '\0') { /* "--" */
8539 argv++;
8540 }
8541 /* else: "-something", no special meaning */
8542 }
8543 new_cmd = *argv;
8544 reset_traps:
8545 argv++;
8546 goto process_sig_list;
8547}
8548
Mike Frysinger93cadc22009-05-27 17:06:25 -04008549/* http://www.opengroup.org/onlinepubs/9699919799/utilities/type.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008550static int FAST_FUNC builtin_type(char **argv)
Mike Frysinger93cadc22009-05-27 17:06:25 -04008551{
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008552 int ret = EXIT_SUCCESS;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008553
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008554 while (*++argv) {
Mike Frysinger93cadc22009-05-27 17:06:25 -04008555 const char *type;
Denys Vlasenko171932d2009-05-28 17:07:22 +02008556 char *path = NULL;
Mike Frysinger93cadc22009-05-27 17:06:25 -04008557
8558 if (0) {} /* make conditional compile easier below */
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008559 /*else if (find_alias(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008560 type = "an alias";*/
8561#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008562 else if (find_function(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008563 type = "a function";
8564#endif
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008565 else if (find_builtin(*argv))
Mike Frysinger93cadc22009-05-27 17:06:25 -04008566 type = "a shell builtin";
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008567 else if ((path = find_in_path(*argv)) != NULL)
8568 type = path;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008569 else {
Denys Vlasenkodd6b2112009-05-28 09:45:50 +02008570 bb_error_msg("type: %s: not found", *argv);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008571 ret = EXIT_FAILURE;
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008572 continue;
8573 }
Mike Frysinger93cadc22009-05-27 17:06:25 -04008574
Denys Vlasenko5d7cca22009-05-28 09:58:43 +02008575 printf("%s is %s\n", *argv, type);
8576 free(path);
Mike Frysinger93cadc22009-05-27 17:06:25 -04008577 }
8578
8579 return ret;
8580}
8581
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008582#if ENABLE_HUSH_JOB
8583/* built-in 'fg' and 'bg' handler */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008584static int FAST_FUNC builtin_fg_bg(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008585{
8586 int i, jobnum;
8587 struct pipe *pi;
8588
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008589 if (!G_interactive_fd)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008590 return EXIT_FAILURE;
Denis Vlasenkoc8653f62009-04-27 23:29:14 +00008591
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008592 /* If they gave us no args, assume they want the last backgrounded task */
8593 if (!argv[1]) {
Denis Vlasenko87a86552008-07-29 19:43:10 +00008594 for (pi = G.job_list; pi; pi = pi->next) {
8595 if (pi->jobid == G.last_jobid) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008596 goto found;
8597 }
8598 }
8599 bb_error_msg("%s: no current job", argv[0]);
8600 return EXIT_FAILURE;
8601 }
8602 if (sscanf(argv[1], "%%%d", &jobnum) != 1) {
8603 bb_error_msg("%s: bad argument '%s'", argv[0], argv[1]);
8604 return EXIT_FAILURE;
8605 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008606 for (pi = G.job_list; pi; pi = pi->next) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008607 if (pi->jobid == jobnum) {
8608 goto found;
8609 }
8610 }
8611 bb_error_msg("%s: %d: no such job", argv[0], jobnum);
8612 return EXIT_FAILURE;
8613 found:
Denis Vlasenko6b9e0532009-04-18 01:23:21 +00008614 /* TODO: bash prints a string representation
8615 * of job being foregrounded (like "sleep 1 | cat") */
Mike Frysinger38478a62009-05-20 04:48:06 -04008616 if (argv[0][0] == 'f' && G_saved_tty_pgrp) {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008617 /* Put the job into the foreground. */
Denis Vlasenko60b392f2009-04-03 19:14:32 +00008618 tcsetpgrp(G_interactive_fd, pi->pgrp);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008619 }
8620
8621 /* Restart the processes in the job */
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008622 debug_printf_jobs("reviving %d procs, pgrp %d\n", pi->num_cmds, pi->pgrp);
8623 for (i = 0; i < pi->num_cmds; i++) {
8624 debug_printf_jobs("reviving pid %d\n", pi->cmds[i].pid);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008625 }
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008626 pi->stopped_cmds = 0;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008627
8628 i = kill(- pi->pgrp, SIGCONT);
8629 if (i < 0) {
8630 if (errno == ESRCH) {
8631 delete_finished_bg_job(pi);
8632 return EXIT_SUCCESS;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008633 }
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008634 bb_perror_msg("kill (SIGCONT)");
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008635 }
8636
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008637 if (argv[0][0] == 'f') {
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008638 remove_bg_job(pi);
8639 return checkjobs_and_fg_shell(pi);
8640 }
8641 return EXIT_SUCCESS;
8642}
8643#endif
8644
8645#if ENABLE_HUSH_HELP
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008646static int FAST_FUNC builtin_help(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008647{
8648 const struct built_in_command *x;
8649
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008650 printf(
Denis Vlasenko34d4d892009-04-04 20:24:37 +00008651 "Built-in commands:\n"
8652 "------------------\n");
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008653 for (x = bltins1; x != &bltins1[ARRAY_SIZE(bltins1)]; x++) {
Denys Vlasenko17323a62010-01-28 01:57:05 +01008654 if (x->b_descr)
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008655 printf("%-10s%s\n", x->b_cmd, x->b_descr);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008656 }
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008657 bb_putchar('\n');
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008658 return EXIT_SUCCESS;
8659}
8660#endif
8661
Denys Vlasenkoff463a82013-05-12 02:45:23 +02008662#if MAX_HISTORY && ENABLE_FEATURE_EDITING
Flemming Madsend96ffda2013-04-07 18:47:24 +02008663static int FAST_FUNC builtin_history(char **argv UNUSED_PARAM)
8664{
8665 show_history(G.line_input_state);
8666 return EXIT_SUCCESS;
8667}
8668#endif
8669
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008670#if ENABLE_HUSH_JOB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008671static int FAST_FUNC builtin_jobs(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008672{
8673 struct pipe *job;
8674 const char *status_string;
8675
Denis Vlasenko87a86552008-07-29 19:43:10 +00008676 for (job = G.job_list; job; job = job->next) {
Denis Vlasenko9af22c72008-10-09 12:54:58 +00008677 if (job->alive_cmds == job->stopped_cmds)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008678 status_string = "Stopped";
8679 else
8680 status_string = "Running";
8681
8682 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->cmdtext);
8683 }
8684 return EXIT_SUCCESS;
8685}
8686#endif
8687
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008688#if HUSH_DEBUG
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008689static int FAST_FUNC builtin_memleak(char **argv UNUSED_PARAM)
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008690{
8691 void *p;
8692 unsigned long l;
8693
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008694# ifdef M_TRIM_THRESHOLD
Denys Vlasenko27726cb2009-09-12 14:48:33 +02008695 /* Optional. Reduces probability of false positives */
8696 malloc_trim(0);
Denys Vlasenkoc0836532009-10-19 13:13:06 +02008697# endif
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008698 /* Crude attempt to find where "free memory" starts,
8699 * sans fragmentation. */
8700 p = malloc(240);
8701 l = (unsigned long)p;
8702 free(p);
8703 p = malloc(3400);
8704 if (l < (unsigned long)p) l = (unsigned long)p;
8705 free(p);
8706
8707 if (!G.memleak_value)
8708 G.memleak_value = l;
Denys Vlasenko9038d6f2009-07-15 20:02:19 +02008709
Denis Vlasenkoc73b70c2009-04-08 11:48:57 +00008710 l -= G.memleak_value;
8711 if ((long)l < 0)
8712 l = 0;
8713 l /= 1024;
8714 if (l > 127)
8715 l = 127;
8716
8717 /* Exitcode is "how many kilobytes we leaked since 1st call" */
8718 return l;
8719}
8720#endif
8721
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008722static int FAST_FUNC builtin_pwd(char **argv UNUSED_PARAM)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008723{
Denys Vlasenkod6b05eb2009-06-06 20:59:55 +02008724 puts(get_cwd(0));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008725 return EXIT_SUCCESS;
8726}
8727
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008728/* Interruptibility of read builtin in bash
8729 * (tested on bash-4.2.8 by sending signals (not by ^C)):
8730 *
8731 * Empty trap makes read ignore corresponding signal, for any signal.
8732 *
8733 * SIGINT:
8734 * - terminates non-interactive shell;
8735 * - interrupts read in interactive shell;
8736 * if it has non-empty trap:
8737 * - executes trap and returns to command prompt in interactive shell;
8738 * - executes trap and returns to read in non-interactive shell;
8739 * SIGTERM:
8740 * - is ignored (does not interrupt) read in interactive shell;
8741 * - terminates non-interactive shell;
8742 * if it has non-empty trap:
8743 * - executes trap and returns to read;
8744 * SIGHUP:
8745 * - terminates shell (regardless of interactivity);
8746 * if it has non-empty trap:
8747 * - executes trap and returns to read;
8748 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008749static int FAST_FUNC builtin_read(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008750{
Denys Vlasenko03dad222010-01-12 23:29:57 +01008751 const char *r;
8752 char *opt_n = NULL;
8753 char *opt_p = NULL;
8754 char *opt_t = NULL;
8755 char *opt_u = NULL;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008756 const char *ifs;
Denys Vlasenko03dad222010-01-12 23:29:57 +01008757 int read_flags;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008758
Denys Vlasenko03dad222010-01-12 23:29:57 +01008759 /* "!": do not abort on errors.
8760 * Option string must start with "sr" to match BUILTIN_READ_xxx
8761 */
8762 read_flags = getopt32(argv, "!srn:p:t:u:", &opt_n, &opt_p, &opt_t, &opt_u);
8763 if (read_flags == (uint32_t)-1)
8764 return EXIT_FAILURE;
8765 argv += optind;
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008766 ifs = get_local_var_value("IFS"); /* can be NULL */
8767
8768 again:
Denys Vlasenko03dad222010-01-12 23:29:57 +01008769 r = shell_builtin_read(set_local_var_from_halves,
8770 argv,
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008771 ifs,
Denys Vlasenko03dad222010-01-12 23:29:57 +01008772 read_flags,
8773 opt_n,
8774 opt_p,
8775 opt_t,
8776 opt_u
8777 );
8778
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008779 if ((uintptr_t)r == 1 && errno == EINTR) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02008780 unsigned sig = check_and_run_traps();
Denys Vlasenko80542ba2011-05-08 21:23:43 +02008781 if (sig && sig != SIGINT)
8782 goto again;
8783 }
8784
Denys Vlasenko03dad222010-01-12 23:29:57 +01008785 if ((uintptr_t)r > 1) {
8786 bb_error_msg("%s", r);
8787 r = (char*)(uintptr_t)1;
Denis Vlasenko05d3b7c2009-04-09 19:16:15 +00008788 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008789
Denys Vlasenko03dad222010-01-12 23:29:57 +01008790 return (uintptr_t)r;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008791}
8792
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008793/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#set
8794 * built-in 'set' handler
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008795 * SUSv3 says:
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008796 * set [-abCefhmnuvx] [-o option] [argument...]
8797 * set [+abCefhmnuvx] [+o option] [argument...]
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008798 * set -- [argument...]
8799 * set -o
8800 * set +o
8801 * Implementations shall support the options in both their hyphen and
8802 * plus-sign forms. These options can also be specified as options to sh.
8803 * Examples:
8804 * Write out all variables and their values: set
8805 * Set $1, $2, and $3 and set "$#" to 3: set c a b
8806 * Turn on the -x and -v options: set -xv
8807 * Unset all positional parameters: set --
8808 * Set $1 to the value of x, even if it begins with '-' or '+': set -- "$x"
8809 * Set the positional parameters to the expansion of x, even if x expands
8810 * with a leading '-' or '+': set -- $x
8811 *
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008812 * So far, we only support "set -- [argument...]" and some of the short names.
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008813 */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008814static int FAST_FUNC builtin_set(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008815{
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008816 int n;
8817 char **pp, **g_argv;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008818 char *arg = *++argv;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008819
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008820 if (arg == NULL) {
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008821 struct variable *e;
Denis Vlasenko87a86552008-07-29 19:43:10 +00008822 for (e = G.top_var; e; e = e->next)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008823 puts(e->varstr);
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008824 return EXIT_SUCCESS;
Denis Vlasenko11fb7cf2009-03-20 10:13:08 +00008825 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008826
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008827 do {
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008828 if (strcmp(arg, "--") == 0) {
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008829 ++argv;
8830 goto set_argv;
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008831 }
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008832 if (arg[0] != '+' && arg[0] != '-')
8833 break;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008834 for (n = 1; arg[n]; ++n) {
8835 if (set_mode((arg[0] == '-'), arg[n], argv[1]))
Denis Vlasenko6ba6f542009-04-10 21:57:50 +00008836 goto error;
Denys Vlasenko6696eac2010-11-14 02:01:50 +01008837 if (arg[n] == 'o' && argv[1])
8838 argv++;
8839 }
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008840 } while ((arg = *++argv) != NULL);
8841 /* Now argv[0] is 1st argument */
8842
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008843 if (arg == NULL)
8844 return EXIT_SUCCESS;
8845 set_argv:
8846
Denis Vlasenko424f79b2009-03-22 14:23:34 +00008847 /* NB: G.global_argv[0] ($0) is never freed/changed */
8848 g_argv = G.global_argv;
8849 if (G.global_args_malloced) {
8850 pp = g_argv;
8851 while (*++pp)
8852 free(*pp);
8853 g_argv[1] = NULL;
8854 } else {
8855 G.global_args_malloced = 1;
8856 pp = xzalloc(sizeof(pp[0]) * 2);
8857 pp[0] = g_argv[0]; /* retain $0 */
8858 g_argv = pp;
8859 }
8860 /* This realloc's G.global_argv */
8861 G.global_argv = pp = add_strings_to_strings(g_argv, argv, /*dup:*/ 1);
8862
8863 n = 1;
8864 while (*++pp)
8865 n++;
8866 G.global_argc = n;
8867
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008868 return EXIT_SUCCESS;
Mike Frysingerad88d5a2009-03-28 13:44:51 +00008869
8870 /* Nothing known, so abort */
8871 error:
8872 bb_error_msg("set: %s: invalid option", arg);
8873 return EXIT_FAILURE;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008874}
8875
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008876static int FAST_FUNC builtin_shift(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008877{
8878 int n = 1;
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008879 argv = skip_dash_dash(argv);
8880 if (argv[0]) {
8881 n = atoi(argv[0]);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008882 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008883 if (n >= 0 && n < G.global_argc) {
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008884 if (G.global_args_malloced) {
8885 int m = 1;
8886 while (m <= n)
8887 free(G.global_argv[m++]);
8888 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00008889 G.global_argc -= n;
Denis Vlasenkoe1300f62009-03-22 11:41:18 +00008890 memmove(&G.global_argv[1], &G.global_argv[n+1],
8891 G.global_argc * sizeof(G.global_argv[0]));
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008892 return EXIT_SUCCESS;
8893 }
8894 return EXIT_FAILURE;
8895}
8896
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008897static int FAST_FUNC builtin_source(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008898{
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008899 char *arg_path, *filename;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008900 FILE *input;
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008901 save_arg_t sv;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008902#if ENABLE_HUSH_FUNCTIONS
8903 smallint sv_flg;
8904#endif
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008905
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008906 argv = skip_dash_dash(argv);
8907 filename = argv[0];
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008908 if (!filename) {
8909 /* bash says: "bash: .: filename argument required" */
8910 return 2; /* bash compat */
8911 }
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008912 arg_path = NULL;
Denys Vlasenkoe66cf822010-05-18 09:12:53 +02008913 if (!strchr(filename, '/')) {
8914 arg_path = find_in_path(filename);
8915 if (arg_path)
8916 filename = arg_path;
8917 }
8918 input = fopen_or_warn(filename, "r");
8919 free(arg_path);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008920 if (!input) {
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008921 /* bb_perror_msg("%s", *argv); - done by fopen_or_warn */
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008922 /* POSIX: non-interactive shell should abort here,
8923 * not merely fail. So far no one complained :)
8924 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008925 return EXIT_FAILURE;
8926 }
8927 close_on_exec_on(fileno(input));
8928
Mike Frysinger885b6f22009-04-18 21:04:25 +00008929#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008930 sv_flg = G.flag_return_in_progress;
8931 /* "we are inside sourced file, ok to use return" */
8932 G.flag_return_in_progress = -1;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008933#endif
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008934 if (argv[1])
8935 save_and_replace_G_args(&sv, argv);
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008936
Denis Vlasenkob6e65562009-04-03 16:49:04 +00008937 parse_and_run_file(input);
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008938 fclose(input);
Denis Vlasenko270b1c32009-04-17 18:54:50 +00008939
Denys Vlasenko88b532d2013-03-17 14:11:04 +01008940 if (argv[1])
8941 restore_G_args(&sv, argv);
Mike Frysinger885b6f22009-04-18 21:04:25 +00008942#if ENABLE_HUSH_FUNCTIONS
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008943 G.flag_return_in_progress = sv_flg;
Mike Frysinger885b6f22009-04-18 21:04:25 +00008944#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00008945
Denis Vlasenkoab2b0642009-04-06 18:42:11 +00008946 return G.last_exitcode;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008947}
8948
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008949static int FAST_FUNC builtin_umask(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008950{
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008951 int rc;
8952 mode_t mask;
8953
8954 mask = umask(0);
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008955 argv = skip_dash_dash(argv);
8956 if (argv[0]) {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008957 mode_t old_mask = mask;
8958
8959 mask ^= 0777;
Denys Vlasenkob131cce2010-05-20 03:39:43 +02008960 rc = bb_parse_mode(argv[0], &mask);
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008961 mask ^= 0777;
8962 if (rc == 0) {
8963 mask = old_mask;
8964 /* bash messages:
8965 * bash: umask: 'q': invalid symbolic mode operator
8966 * bash: umask: 999: octal number out of range
8967 */
Denys Vlasenko44c86ce2010-05-20 04:22:55 +02008968 bb_error_msg("%s: invalid mode '%s'", "umask", argv[0]);
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008969 }
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008970 } else {
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008971 rc = 1;
8972 /* Mimic bash */
8973 printf("%04o\n", (unsigned) mask);
8974 /* fall through and restore mask which we set to 0 */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008975 }
Denis Vlasenkoeb858492009-04-18 02:06:54 +00008976 umask(mask);
8977
8978 return !rc; /* rc != 0 - success */
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008979}
8980
Mike Frysingerd690f682009-03-30 06:50:54 +00008981/* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#unset */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02008982static int FAST_FUNC builtin_unset(char **argv)
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00008983{
Mike Frysingerd690f682009-03-30 06:50:54 +00008984 int ret;
Denis Vlasenko28e67962009-04-26 23:22:40 +00008985 unsigned opts;
Mike Frysingerd690f682009-03-30 06:50:54 +00008986
Denis Vlasenko28e67962009-04-26 23:22:40 +00008987 /* "!": do not abort on errors */
8988 /* "+": stop at 1st non-option */
8989 opts = getopt32(argv, "!+vf");
8990 if (opts == (unsigned)-1)
8991 return EXIT_FAILURE;
8992 if (opts == 3) {
8993 bb_error_msg("unset: -v and -f are exclusive");
8994 return EXIT_FAILURE;
Mike Frysingerd690f682009-03-30 06:50:54 +00008995 }
Denis Vlasenko28e67962009-04-26 23:22:40 +00008996 argv += optind;
Mike Frysingerd690f682009-03-30 06:50:54 +00008997
8998 ret = EXIT_SUCCESS;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00008999 while (*argv) {
Denis Vlasenko28e67962009-04-26 23:22:40 +00009000 if (!(opts & 2)) { /* not -f */
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009001 if (unset_local_var(*argv)) {
9002 /* unset <nonexistent_var> doesn't fail.
9003 * Error is when one tries to unset RO var.
9004 * Message was printed by unset_local_var. */
Mike Frysingerd690f682009-03-30 06:50:54 +00009005 ret = EXIT_FAILURE;
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009006 }
Mike Frysingerd690f682009-03-30 06:50:54 +00009007 }
Denis Vlasenko40e84372009-04-18 11:23:38 +00009008#if ENABLE_HUSH_FUNCTIONS
9009 else {
9010 unset_func(*argv);
9011 }
9012#endif
Denis Vlasenkobfbc9712009-04-06 12:04:42 +00009013 argv++;
Mike Frysingerd690f682009-03-30 06:50:54 +00009014 }
9015 return ret;
Denis Vlasenkoc7985b72008-06-17 05:43:38 +00009016}
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009017
Mike Frysinger56bdea12009-03-28 20:01:58 +00009018/* http://www.opengroup.org/onlinepubs/9699919799/utilities/wait.html */
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009019static int FAST_FUNC builtin_wait(char **argv)
Mike Frysinger56bdea12009-03-28 20:01:58 +00009020{
9021 int ret = EXIT_SUCCESS;
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009022 int status;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009023
Denys Vlasenkob131cce2010-05-20 03:39:43 +02009024 argv = skip_dash_dash(argv);
9025 if (argv[0] == NULL) {
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009026 /* Don't care about wait results */
9027 /* Note 1: must wait until there are no more children */
9028 /* Note 2: must be interruptible */
9029 /* Examples:
9030 * $ sleep 3 & sleep 6 & wait
9031 * [1] 30934 sleep 3
9032 * [2] 30935 sleep 6
9033 * [1] Done sleep 3
9034 * [2] Done sleep 6
9035 * $ sleep 3 & sleep 6 & wait
9036 * [1] 30936 sleep 3
9037 * [2] 30937 sleep 6
9038 * [1] Done sleep 3
9039 * ^C <-- after ~4 sec from keyboard
9040 * $
9041 */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009042 while (1) {
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009043 int sig;
9044 sigset_t oldset, allsigs;
9045
9046 /* waitpid is not interruptible by SA_RESTARTed
9047 * signals which we use. Thus, this ugly dance:
9048 */
9049
9050 /* Make sure possible SIGCHLD is stored in kernel's
9051 * pending signal mask before we call waitpid.
9052 * Or else we may race with SIGCHLD, lose it,
9053 * and get stuck in sigwaitinfo...
9054 */
9055 sigfillset(&allsigs);
9056 sigprocmask(SIG_SETMASK, &allsigs, &oldset);
9057
9058 if (!sigisemptyset(&G.pending_set)) {
9059 /* Crap! we raced with some signal! */
9060 // sig = 0;
9061 goto restore;
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009062 }
Denys Vlasenko9d6cbaf2011-05-11 23:56:11 +02009063
9064 checkjobs(NULL); /* waitpid(WNOHANG) inside */
9065 if (errno == ECHILD) {
9066 sigprocmask(SIG_SETMASK, &oldset, NULL);
9067 break;
9068 }
9069
9070 /* Wait for SIGCHLD or any other signal */
9071 //sig = sigwaitinfo(&allsigs, NULL);
9072 /* It is vitally important for sigsuspend that SIGCHLD has non-DFL handler! */
9073 /* Note: sigsuspend invokes signal handler */
9074 sigsuspend(&oldset);
9075 restore:
9076 sigprocmask(SIG_SETMASK, &oldset, NULL);
9077
9078 /* So, did we get a signal? */
9079 //if (sig > 0)
9080 // raise(sig); /* run handler */
9081 sig = check_and_run_traps();
9082 if (sig /*&& sig != SIGCHLD - always true */) {
9083 /* see note 2 */
9084 ret = 128 + sig;
9085 break;
9086 }
9087 /* SIGCHLD, or no signal, or ignored one, such as SIGQUIT. Repeat */
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009088 }
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009089 return ret;
9090 }
Mike Frysinger56bdea12009-03-28 20:01:58 +00009091
Denis Vlasenko7566bae2009-03-31 17:24:49 +00009092 /* This is probably buggy wrt interruptible-ness */
Denis Vlasenkod5762932009-03-31 11:22:57 +00009093 while (*argv) {
9094 pid_t pid = bb_strtou(*argv, NULL, 10);
Mike Frysinger40b8dc42009-03-29 00:50:30 +00009095 if (errno) {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009096 /* mimic bash message */
9097 bb_error_msg("wait: '%s': not a pid or valid job spec", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009098 return EXIT_FAILURE;
Denis Vlasenkod5762932009-03-31 11:22:57 +00009099 }
9100 if (waitpid(pid, &status, 0) == pid) {
Mike Frysinger56bdea12009-03-28 20:01:58 +00009101 if (WIFSIGNALED(status))
9102 ret = 128 + WTERMSIG(status);
9103 else if (WIFEXITED(status))
9104 ret = WEXITSTATUS(status);
Denis Vlasenkod5762932009-03-31 11:22:57 +00009105 else /* wtf? */
Mike Frysinger56bdea12009-03-28 20:01:58 +00009106 ret = EXIT_FAILURE;
9107 } else {
Denis Vlasenkod5762932009-03-31 11:22:57 +00009108 bb_perror_msg("wait %s", *argv);
Mike Frysinger56bdea12009-03-28 20:01:58 +00009109 ret = 127;
9110 }
Denis Vlasenkod5762932009-03-31 11:22:57 +00009111 argv++;
Mike Frysinger56bdea12009-03-28 20:01:58 +00009112 }
9113
9114 return ret;
9115}
9116
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009117#if ENABLE_HUSH_LOOPS || ENABLE_HUSH_FUNCTIONS
9118static unsigned parse_numeric_argv1(char **argv, unsigned def, unsigned def_min)
9119{
9120 if (argv[1]) {
9121 def = bb_strtou(argv[1], NULL, 10);
9122 if (errno || def < def_min || argv[2]) {
9123 bb_error_msg("%s: bad arguments", argv[0]);
9124 def = UINT_MAX;
9125 }
9126 }
9127 return def;
9128}
9129#endif
9130
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009131#if ENABLE_HUSH_LOOPS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009132static int FAST_FUNC builtin_break(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009133{
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009134 unsigned depth;
Denis Vlasenko87a86552008-07-29 19:43:10 +00009135 if (G.depth_of_loop == 0) {
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009136 bb_error_msg("%s: only meaningful in a loop", argv[0]);
Denis Vlasenkofcf37c32008-07-29 11:37:15 +00009137 return EXIT_SUCCESS; /* bash compat */
9138 }
Denis Vlasenko87a86552008-07-29 19:43:10 +00009139 G.flag_break_continue++; /* BC_BREAK = 1 */
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009140
9141 G.depth_break_continue = depth = parse_numeric_argv1(argv, 1, 1);
9142 if (depth == UINT_MAX)
9143 G.flag_break_continue = BC_BREAK;
9144 if (G.depth_of_loop < depth)
Denis Vlasenko87a86552008-07-29 19:43:10 +00009145 G.depth_break_continue = G.depth_of_loop;
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009146
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009147 return EXIT_SUCCESS;
9148}
9149
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009150static int FAST_FUNC builtin_continue(char **argv)
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009151{
Denis Vlasenko4f504a92008-07-29 19:48:30 +00009152 G.flag_break_continue = 1; /* BC_CONTINUE = 2 = 1+1 */
9153 return builtin_break(argv);
Denis Vlasenkobcb25532008-07-28 23:04:34 +00009154}
Denis Vlasenkodadfb492008-07-29 10:16:05 +00009155#endif
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009156
9157#if ENABLE_HUSH_FUNCTIONS
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +02009158static int FAST_FUNC builtin_return(char **argv)
Denis Vlasenko3d40d8e2009-04-17 23:44:18 +00009159{
9160 int rc;
9161
9162 if (G.flag_return_in_progress != -1) {
9163 bb_error_msg("%s: not in a function or sourced script", argv[0]);
9164 return EXIT_FAILURE; /* bash compat */
9165 }
9166
9167 G.flag_return_in_progress = 1;
9168
9169 /* bash:
9170 * out of range: wraps around at 256, does not error out
9171 * non-numeric param:
9172 * f() { false; return qwe; }; f; echo $?
9173 * bash: return: qwe: numeric argument required <== we do this
9174 * 255 <== we also do this
9175 */
9176 rc = parse_numeric_argv1(argv, G.last_exitcode, 0);
9177 return rc;
9178}
9179#endif