hush: rename ->o_quote to ->o_escape
[oweals/busybox.git] / shell / hush.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * 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.
7  *
8  * Copyright (C) 2000,2001  Larry Doolittle  <larry@doolittle.boa.org>
9  *
10  * Credits:
11  *      The parser routines proper are all original material, first
12  *      written Dec 2000 and Jan 2001 by Larry Doolittle.  The
13  *      execution engine, the builtins, and much of the underlying
14  *      support has been adapted from busybox-0.49pre's lash, which is
15  *      Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
16  *      written by Erik Andersen <andersen@codepoet.org>.  That, in turn,
17  *      is based in part on ladsh.c, by Michael K. Johnson and Erik W.
18  *      Troan, which they placed in the public domain.  I don't know
19  *      how much of the Johnson/Troan code has survived the repeated
20  *      rewrites.
21  *
22  * Other credits:
23  *      o_addchr() derived from similar w_addchar function in glibc-2.2.
24  *      setup_redirect(), redirect_opt_num(), and big chunks of main()
25  *      and many builtins derived from contributions by Erik Andersen.
26  *      Miscellaneous bugfixes from Matt Kraai.
27  *
28  * There are two big (and related) architecture differences between
29  * this parser and the lash parser.  One is that this version is
30  * actually designed from the ground up to understand nearly all
31  * of the Bourne grammar.  The second, consequential change is that
32  * the parser and input reader have been turned inside out.  Now,
33  * the parser is in control, and asks for input as needed.  The old
34  * way had the input reader in control, and it asked for parsing to
35  * take place as needed.  The new way makes it much easier to properly
36  * handle the recursion implicit in the various substitutions, especially
37  * across continuation lines.
38  *
39  * POSIX syntax not implemented:
40  *      aliases
41  *      <(list) and >(list) Process Substitution
42  *      Here Documents ( << word )
43  *      Functions
44  *      Tilde Expansion
45  *      Parameter Expansion for substring processing ${var#word} ${var%word}
46  *
47  * Bash stuff maybe optional enable:
48  *      &> and >& redirection of stdout+stderr
49  *      Brace expansion
50  *      reserved words: [[ ]] function select
51  *      substrings ${var:1:5}
52  *
53  * Major bugs:
54  *      job handling woefully incomplete and buggy (improved --vda)
55  * to-do:
56  *      port selected bugfixes from post-0.49 busybox lash - done?
57  *      change { and } from special chars to reserved words
58  *      builtins: return, trap, ulimit
59  *      test magic exec with redirection only
60  *      follow IFS rules more precisely, including update semantics
61  *      figure out what to do with backslash-newline
62  *      propagate syntax errors, die on resource errors?
63  *      continuation lines, both explicit and implicit - done?
64  *      maybe change charmap[] to use 2-bit entries
65  *
66  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
67  */
68
69 #include "busybox.h" /* for APPLET_IS_NOFORK/NOEXEC */
70 //TODO: pull in some .h and find out whether we have SINGLE_APPLET_MAIN?
71 //#include "applet_tables.h" doesn't work
72 #include <glob.h>
73 /* #include <dmalloc.h> */
74 #if ENABLE_HUSH_CASE
75 #include <fnmatch.h>
76 #endif
77
78 #include "math.h"
79
80 #define HUSH_VER_STR "0.92"
81
82 #if defined SINGLE_APPLET_MAIN
83 /* STANDALONE does not make sense, and won't compile */
84 #undef CONFIG_FEATURE_SH_STANDALONE
85 #undef ENABLE_FEATURE_SH_STANDALONE
86 #undef USE_FEATURE_SH_STANDALONE
87 #define SKIP_FEATURE_SH_STANDALONE(...) __VA_ARGS__
88 #define ENABLE_FEATURE_SH_STANDALONE 0
89 #define USE_FEATURE_SH_STANDALONE(...)
90 #define SKIP_FEATURE_SH_STANDALONE(...) __VA_ARGS__
91 #endif
92
93 #if !BB_MMU && ENABLE_HUSH_TICK
94 //#undef ENABLE_HUSH_TICK
95 //#define ENABLE_HUSH_TICK 0
96 #warning On NOMMU, hush command substitution is dangerous.
97 #warning Dont use it for commands which produce lots of output.
98 #warning For more info see shell/hush.c, generate_stream_from_list().
99 #endif
100
101 #if !ENABLE_HUSH_INTERACTIVE
102 #undef ENABLE_FEATURE_EDITING
103 #define ENABLE_FEATURE_EDITING 0
104 #undef ENABLE_FEATURE_EDITING_FANCY_PROMPT
105 #define ENABLE_FEATURE_EDITING_FANCY_PROMPT 0
106 #endif
107
108 /* Do we support ANY keywords? */
109 #if ENABLE_HUSH_IF || ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
110 #define HAS_KEYWORDS 1
111 #define IF_HAS_KEYWORDS(...) __VA_ARGS__
112 #define IF_HAS_NO_KEYWORDS(...)
113 #else
114 #define HAS_KEYWORDS 0
115 #define IF_HAS_KEYWORDS(...)
116 #define IF_HAS_NO_KEYWORDS(...) __VA_ARGS__
117 #endif
118
119 /* Keep unconditionally on for now */
120 #define HUSH_DEBUG 1
121 /* In progress... */
122 #define ENABLE_HUSH_FUNCTIONS 0
123
124
125 /* If you comment out one of these below, it will be #defined later
126  * to perform debug printfs to stderr: */
127 #define debug_printf(...)        do {} while (0)
128 /* Finer-grained debug switches */
129 #define debug_printf_parse(...)  do {} while (0)
130 #define debug_print_tree(a, b)   do {} while (0)
131 #define debug_printf_exec(...)   do {} while (0)
132 #define debug_printf_env(...)    do {} while (0)
133 #define debug_printf_jobs(...)   do {} while (0)
134 #define debug_printf_expand(...) do {} while (0)
135 #define debug_printf_glob(...)   do {} while (0)
136 #define debug_printf_list(...)   do {} while (0)
137 #define debug_printf_subst(...)  do {} while (0)
138 #define debug_printf_clean(...)  do {} while (0)
139
140 #ifndef debug_printf
141 #define debug_printf(...) fprintf(stderr, __VA_ARGS__)
142 #endif
143
144 #ifndef debug_printf_parse
145 #define debug_printf_parse(...) fprintf(stderr, __VA_ARGS__)
146 #endif
147
148 #ifndef debug_printf_exec
149 #define debug_printf_exec(...) fprintf(stderr, __VA_ARGS__)
150 #endif
151
152 #ifndef debug_printf_env
153 #define debug_printf_env(...) fprintf(stderr, __VA_ARGS__)
154 #endif
155
156 #ifndef debug_printf_jobs
157 #define debug_printf_jobs(...) fprintf(stderr, __VA_ARGS__)
158 #define DEBUG_JOBS 1
159 #else
160 #define DEBUG_JOBS 0
161 #endif
162
163 #ifndef debug_printf_expand
164 #define debug_printf_expand(...) fprintf(stderr, __VA_ARGS__)
165 #define DEBUG_EXPAND 1
166 #else
167 #define DEBUG_EXPAND 0
168 #endif
169
170 #ifndef debug_printf_glob
171 #define debug_printf_glob(...) fprintf(stderr, __VA_ARGS__)
172 #define DEBUG_GLOB 1
173 #else
174 #define DEBUG_GLOB 0
175 #endif
176
177 #ifndef debug_printf_list
178 #define debug_printf_list(...) fprintf(stderr, __VA_ARGS__)
179 #endif
180
181 #ifndef debug_printf_subst
182 #define debug_printf_subst(...) fprintf(stderr, __VA_ARGS__)
183 #endif
184
185 #ifndef debug_printf_clean
186 /* broken, of course, but OK for testing */
187 static const char *indenter(int i)
188 {
189         static const char blanks[] ALIGN1 =
190                 "                                    ";
191         return &blanks[sizeof(blanks) - i - 1];
192 }
193 #define debug_printf_clean(...) fprintf(stderr, __VA_ARGS__)
194 #define DEBUG_CLEAN 1
195 #endif
196
197 #if DEBUG_EXPAND
198 static void debug_print_strings(const char *prefix, char **vv)
199 {
200         fprintf(stderr, "%s:\n", prefix);
201         while (*vv)
202                 fprintf(stderr, " '%s'\n", *vv++);
203 }
204 #else
205 #define debug_print_strings(prefix, vv) ((void)0)
206 #endif
207
208 /*
209  * Leak hunting. Use hush_leaktool.sh for post-processing.
210  */
211 #ifdef FOR_HUSH_LEAKTOOL
212 /* suppress "warning: no previous prototype..." */
213 void *xxmalloc(int lineno, size_t size);
214 void *xxrealloc(int lineno, void *ptr, size_t size);
215 char *xxstrdup(int lineno, const char *str);
216 void xxfree(void *ptr);
217 void *xxmalloc(int lineno, size_t size)
218 {
219         void *ptr = xmalloc((size + 0xff) & ~0xff);
220         fprintf(stderr, "line %d: malloc %p\n", lineno, ptr);
221         return ptr;
222 }
223 void *xxrealloc(int lineno, void *ptr, size_t size)
224 {
225         ptr = xrealloc(ptr, (size + 0xff) & ~0xff);
226         fprintf(stderr, "line %d: realloc %p\n", lineno, ptr);
227         return ptr;
228 }
229 char *xxstrdup(int lineno, const char *str)
230 {
231         char *ptr = xstrdup(str);
232         fprintf(stderr, "line %d: strdup %p\n", lineno, ptr);
233         return ptr;
234 }
235 void xxfree(void *ptr)
236 {
237         fprintf(stderr, "free %p\n", ptr);
238         free(ptr);
239 }
240 #define xmalloc(s)     xxmalloc(__LINE__, s)
241 #define xrealloc(p, s) xxrealloc(__LINE__, p, s)
242 #define xstrdup(s)     xxstrdup(__LINE__, s)
243 #define free(p)        xxfree(p)
244 #endif
245
246
247 static const char hush_version_str[] ALIGN1 = "HUSH_VERSION="HUSH_VER_STR;
248
249 #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
250
251 #define SPECIAL_VAR_SYMBOL       3
252 #define PARSEFLAG_EXIT_FROM_LOOP 1
253
254 typedef enum redir_type {
255         REDIRECT_INPUT     = 1,
256         REDIRECT_OVERWRITE = 2,
257         REDIRECT_APPEND    = 3,
258         REDIRECT_HEREIS    = 4,
259         REDIRECT_IO        = 5
260 } redir_type;
261
262 /* The descrip member of this structure is only used to make
263  * debugging output pretty */
264 static const struct {
265         int mode;
266         signed char default_fd;
267         char descrip[3];
268 } redir_table[] = {
269         { 0,                         0, "()" },
270         { O_RDONLY,                  0, "<"  },
271         { O_CREAT|O_TRUNC|O_WRONLY,  1, ">"  },
272         { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
273         { O_RDONLY,                 -1, "<<" },
274         { O_RDWR,                    1, "<>" }
275 };
276
277 typedef enum pipe_style {
278         PIPE_SEQ = 1,
279         PIPE_AND = 2,
280         PIPE_OR  = 3,
281         PIPE_BG  = 4,
282 } pipe_style;
283
284 typedef enum reserved_style {
285         RES_NONE  = 0,
286 #if ENABLE_HUSH_IF
287         RES_IF    ,
288         RES_THEN  ,
289         RES_ELIF  ,
290         RES_ELSE  ,
291         RES_FI    ,
292 #endif
293 #if ENABLE_HUSH_LOOPS
294         RES_FOR   ,
295         RES_WHILE ,
296         RES_UNTIL ,
297         RES_DO    ,
298         RES_DONE  ,
299 #endif
300 #if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
301         RES_IN    ,
302 #endif
303 #if ENABLE_HUSH_CASE
304         RES_CASE  ,
305         /* two pseudo-keywords support contrived "case" syntax: */
306         RES_MATCH , /* "word)" */
307         RES_CASEI , /* "this command is inside CASE" */
308         RES_ESAC  ,
309 #endif
310         RES_XXXX  ,
311         RES_SNTX
312 } reserved_style;
313
314 struct redir_struct {
315         struct redir_struct *next;
316         char *rd_filename;          /* filename */
317         int fd;                     /* file descriptor being redirected */
318         int dup;                    /* -1, or file descriptor being duplicated */
319         smallint /*enum redir_type*/ rd_type;
320 };
321
322 struct command {
323         pid_t pid;                  /* 0 if exited */
324         int assignment_cnt;         /* how many argv[i] are assignments? */
325         smallint is_stopped;        /* is the command currently running? */
326         smallint grp_type;          /* GRP_xxx */
327         struct pipe *group;         /* if non-NULL, this "prog" is {} group,
328                                      * subshell, or a compound statement */
329         char **argv;                /* command name and arguments */
330         struct redir_struct *redirects; /* I/O redirections */
331 };
332 /* argv vector may contain variable references (^Cvar^C, ^C0^C etc)
333  * and on execution these are substituted with their values.
334  * Substitution can make _several_ words out of one argv[n]!
335  * Example: argv[0]=='.^C*^C.' here: echo .$*.
336  * References of the form ^C`cmd arg^C are `cmd arg` substitutions.
337  */
338 #define GRP_NORMAL   0
339 #define GRP_SUBSHELL 1
340 #if ENABLE_HUSH_FUNCTIONS
341 #define GRP_FUNCTION 2
342 #endif
343
344 struct pipe {
345         struct pipe *next;
346         int num_cmds;               /* total number of commands in job */
347         int alive_cmds;             /* number of commands running (not exited) */
348         int stopped_cmds;           /* number of commands alive, but stopped */
349 #if ENABLE_HUSH_JOB
350         int jobid;                  /* job number */
351         pid_t pgrp;                 /* process group ID for the job */
352         char *cmdtext;              /* name of job */
353 #endif
354         struct command *cmds;       /* array of commands in pipe */
355         smallint followup;          /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
356         IF_HAS_KEYWORDS(smallint pi_inverted;) /* "! cmd | cmd" */
357         IF_HAS_KEYWORDS(smallint res_word;) /* needed for if, for, while, until... */
358 };
359
360 /* This holds pointers to the various results of parsing */
361 struct parse_context {
362         struct command *command;
363         struct pipe *list_head;
364         struct pipe *pipe;
365         struct redir_struct *pending_redirect;
366 #if HAS_KEYWORDS
367         smallint ctx_res_w;
368         smallint ctx_inverted; /* "! cmd | cmd" */
369 #if ENABLE_HUSH_CASE
370         smallint ctx_dsemicolon; /* ";;" seen */
371 #endif
372         int old_flag; /* bitmask of FLAG_xxx, for figuring out valid reserved words */
373         struct parse_context *stack;
374 #endif
375 };
376
377 /* On program start, environ points to initial environment.
378  * putenv adds new pointers into it, unsetenv removes them.
379  * Neither of these (de)allocates the strings.
380  * setenv allocates new strings in malloc space and does putenv,
381  * and thus setenv is unusable (leaky) for shell's purposes */
382 #define setenv(...) setenv_is_leaky_dont_use()
383 struct variable {
384         struct variable *next;
385         char *varstr;        /* points to "name=" portion */
386         int max_len;         /* if > 0, name is part of initial env; else name is malloced */
387         smallint flg_export; /* putenv should be done on this var */
388         smallint flg_read_only;
389 };
390
391 typedef struct o_string {
392         char *data;
393         int length; /* position where data is appended */
394         int maxlen;
395         /* Protect newly added chars against globbing
396          * (by prepending \ to *, ?, [, \) */
397         smallint o_escape;
398         smallint o_glob;
399         smallint nonnull;
400         smallint has_empty_slot;
401         smallint o_assignment; /* 0:maybe, 1:yes, 2:no */
402 } o_string;
403 enum {
404         MAYBE_ASSIGNMENT = 0,
405         DEFINITELY_ASSIGNMENT = 1,
406         NOT_ASSIGNMENT = 2,
407         WORD_IS_KEYWORD = 3, /* not assigment, but next word may be: "if v=xyz cmd;" */
408 };
409 /* Used for initialization: o_string foo = NULL_O_STRING; */
410 #define NULL_O_STRING { NULL }
411
412 /* I can almost use ordinary FILE*.  Is open_memstream() universally
413  * available?  Where is it documented? */
414 typedef struct in_str {
415         const char *p;
416         /* eof_flag=1: last char in ->p is really an EOF */
417         char eof_flag; /* meaningless if ->p == NULL */
418         char peek_buf[2];
419 #if ENABLE_HUSH_INTERACTIVE
420         smallint promptme;
421         smallint promptmode; /* 0: PS1, 1: PS2 */
422 #endif
423         FILE *file;
424         int (*get) (struct in_str *);
425         int (*peek) (struct in_str *);
426 } in_str;
427 #define i_getch(input) ((input)->get(input))
428 #define i_peek(input) ((input)->peek(input))
429
430 enum {
431         CHAR_ORDINARY           = 0,
432         CHAR_ORDINARY_IF_QUOTED = 1, /* example: *, # */
433         CHAR_IFS                = 2, /* treated as ordinary if quoted */
434         CHAR_SPECIAL            = 3, /* \, $, ", maybe ` */
435 };
436
437 enum {
438         BC_BREAK = 1,
439         BC_CONTINUE = 2,
440 };
441
442
443 /* "Globals" within this file */
444 /* Sorted roughly by size (smaller offsets == smaller code) */
445 struct globals {
446 #if ENABLE_HUSH_INTERACTIVE
447         /* 'interactive_fd' is a fd# open to ctty, if we have one
448          * _AND_ if we decided to act interactively */
449         int interactive_fd;
450         const char *PS1;
451         const char *PS2;
452 #endif
453 #if ENABLE_FEATURE_EDITING
454         line_input_t *line_input_state;
455 #endif
456         pid_t root_pid;
457         pid_t last_bg_pid;
458 #if ENABLE_HUSH_JOB
459         int run_list_level;
460         pid_t saved_tty_pgrp;
461         int last_jobid;
462         struct pipe *job_list;
463         struct pipe *toplevel_list;
464 ////    smallint ctrl_z_flag;
465 #endif
466         smallint flag_SIGINT;
467 #if ENABLE_HUSH_LOOPS
468         smallint flag_break_continue;
469 #endif
470         smallint fake_mode;
471         /* These four support $?, $#, and $1 */
472         smalluint last_return_code;
473         /* is global_argv and global_argv[1..n] malloced? (note: not [0]) */
474         smalluint global_args_malloced;
475         /* how many non-NULL argv's we have. NB: $# + 1 */
476         int global_argc;
477         char **global_argv;
478 #if ENABLE_HUSH_LOOPS
479         unsigned depth_break_continue;
480         unsigned depth_of_loop;
481 #endif
482         const char *ifs;
483         const char *cwd;
484         struct variable *top_var; /* = &G.shell_ver (set in main()) */
485         struct variable shell_ver;
486 #if ENABLE_FEATURE_SH_STANDALONE
487         struct nofork_save_area nofork_save;
488 #endif
489 #if ENABLE_HUSH_JOB
490         sigjmp_buf toplevel_jb;
491 #endif
492         unsigned char charmap[256];
493         char user_input_buf[ENABLE_FEATURE_EDITING ? BUFSIZ : 2];
494         /* Signal and trap handling */
495 //      unsigned count_SIGCHLD;
496 //      unsigned handled_SIGCHLD;
497         /* which signals have non-DFL handler (even with no traps set)? */
498         unsigned non_DFL_mask;
499         char **traps; /* char *traps[NSIG] */
500         sigset_t blocked_set;
501         sigset_t inherited_set;
502 };
503 #define G (*ptr_to_globals)
504 /* Not #defining name to G.name - this quickly gets unwieldy
505  * (too many defines). Also, I actually prefer to see when a variable
506  * is global, thus "G." prefix is a useful hint */
507 #define INIT_G() do { \
508         SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
509 } while (0)
510
511
512 /* Function prototypes for builtins */
513 static int builtin_cd(char **argv);
514 static int builtin_echo(char **argv);
515 static int builtin_eval(char **argv);
516 static int builtin_exec(char **argv);
517 static int builtin_exit(char **argv);
518 static int builtin_export(char **argv);
519 #if ENABLE_HUSH_JOB
520 static int builtin_fg_bg(char **argv);
521 static int builtin_jobs(char **argv);
522 #endif
523 #if ENABLE_HUSH_HELP
524 static int builtin_help(char **argv);
525 #endif
526 static int builtin_pwd(char **argv);
527 static int builtin_read(char **argv);
528 static int builtin_test(char **argv);
529 static int builtin_trap(char **argv);
530 static int builtin_true(char **argv);
531 static int builtin_set(char **argv);
532 static int builtin_shift(char **argv);
533 static int builtin_source(char **argv);
534 static int builtin_umask(char **argv);
535 static int builtin_unset(char **argv);
536 static int builtin_wait(char **argv);
537 #if ENABLE_HUSH_LOOPS
538 static int builtin_break(char **argv);
539 static int builtin_continue(char **argv);
540 #endif
541 //static int builtin_not_written(char **argv);
542
543 /* Table of built-in functions.  They can be forked or not, depending on
544  * context: within pipes, they fork.  As simple commands, they do not.
545  * When used in non-forking context, they can change global variables
546  * in the parent shell process.  If forked, of course they cannot.
547  * For example, 'unset foo | whatever' will parse and run, but foo will
548  * still be set at the end. */
549 struct built_in_command {
550         const char *cmd;
551         int (*function)(char **argv);
552 #if ENABLE_HUSH_HELP
553         const char *descr;
554 #define BLTIN(cmd, func, help) { cmd, func, help }
555 #else
556 #define BLTIN(cmd, func, help) { cmd, func }
557 #endif
558 };
559
560 /* For now, echo and test are unconditionally enabled.
561  * Maybe make it configurable? */
562 static const struct built_in_command bltins[] = {
563         BLTIN("."     , builtin_source, "Run commands in a file"),
564         BLTIN(":"     , builtin_true, "No-op"),
565         BLTIN("["     , builtin_test, "Test condition"),
566 #if ENABLE_HUSH_JOB
567         BLTIN("bg"    , builtin_fg_bg, "Resume a job in the background"),
568 #endif
569 #if ENABLE_HUSH_LOOPS
570         BLTIN("break" , builtin_break, "Exit from a loop"),
571 #endif
572         BLTIN("cd"    , builtin_cd, "Change directory"),
573 #if ENABLE_HUSH_LOOPS
574         BLTIN("continue", builtin_continue, "Start new loop iteration"),
575 #endif
576         BLTIN("echo"  , builtin_echo, "Write to stdout"),
577         BLTIN("eval"  , builtin_eval, "Construct and run shell command"),
578         BLTIN("exec"  , builtin_exec, "Execute command, don't return to shell"),
579         BLTIN("exit"  , builtin_exit, "Exit"),
580         BLTIN("export", builtin_export, "Set environment variable"),
581 #if ENABLE_HUSH_JOB
582         BLTIN("fg"    , builtin_fg_bg, "Bring job into the foreground"),
583         BLTIN("jobs"  , builtin_jobs, "List active jobs"),
584 #endif
585         BLTIN("pwd"   , builtin_pwd, "Print current directory"),
586         BLTIN("read"  , builtin_read, "Input environment variable"),
587 //      BLTIN("return", builtin_not_written, "Return from a function"),
588         BLTIN("set"   , builtin_set, "Set/unset shell local variables"),
589         BLTIN("shift" , builtin_shift, "Shift positional parameters"),
590         BLTIN("test"  , builtin_test, "Test condition"),
591         BLTIN("trap"  , builtin_trap, "Trap signals"),
592 //      BLTIN("ulimit", builtin_not_written, "Control resource limits"),
593         BLTIN("umask" , builtin_umask, "Set file creation mask"),
594         BLTIN("unset" , builtin_unset, "Unset environment variable"),
595         BLTIN("wait"  , builtin_wait, "Wait for process"),
596 #if ENABLE_HUSH_HELP
597         BLTIN("help"  , builtin_help, "List shell built-in commands"),
598 #endif
599 };
600
601
602 /* Normal */
603 static void maybe_die(const char *notice, const char *msg)
604 {
605         /* Was using fancy stuff:
606          * (G.interactive_fd ? bb_error_msg : bb_error_msg_and_die)(...params...)
607          * but it SEGVs. ?! Oh well... explicit temp ptr works around that */
608         void FAST_FUNC (*fp)(const char *s, ...) = bb_error_msg_and_die;
609 #if ENABLE_HUSH_INTERACTIVE
610         fp = (G.interactive_fd ? bb_error_msg : bb_error_msg_and_die);
611 #endif
612         fp(msg ? "%s: %s" : notice, notice, msg);
613 }
614 #if 1
615 #define syntax(msg) maybe_die("syntax error", msg);
616 #else
617 /* Debug -- trick gcc to expand __LINE__ and convert to string */
618 #define __syntax(msg, line) maybe_die("syntax error hush.c:" # line, msg)
619 #define _syntax(msg, line) __syntax(msg, line)
620 #define syntax(msg) _syntax(msg, __LINE__)
621 #endif
622
623 static int glob_needed(const char *s)
624 {
625         while (*s) {
626                 if (*s == '\\')
627                         s++;
628                 if (*s == '*' || *s == '[' || *s == '?')
629                         return 1;
630                 s++;
631         }
632         return 0;
633 }
634
635 static int is_assignment(const char *s)
636 {
637         if (!s || !(isalpha(*s) || *s == '_'))
638                 return 0;
639         s++;
640         while (isalnum(*s) || *s == '_')
641                 s++;
642         return *s == '=';
643 }
644
645 /* Replace each \x with x in place, return ptr past NUL. */
646 static char *unbackslash(char *src)
647 {
648         char *dst = src;
649         while (1) {
650                 if (*src == '\\')
651                         src++;
652                 if ((*dst++ = *src++) == '\0')
653                         break;
654         }
655         return dst;
656 }
657
658 static char **add_strings_to_strings(char **strings, char **add, int need_to_dup)
659 {
660         int i;
661         unsigned count1;
662         unsigned count2;
663         char **v;
664
665         v = strings;
666         count1 = 0;
667         if (v) {
668                 while (*v) {
669                         count1++;
670                         v++;
671                 }
672         }
673         count2 = 0;
674         v = add;
675         while (*v) {
676                 count2++;
677                 v++;
678         }
679         v = xrealloc(strings, (count1 + count2 + 1) * sizeof(char*));
680         v[count1 + count2] = NULL;
681         i = count2;
682         while (--i >= 0)
683                 v[count1 + i] = (need_to_dup ? xstrdup(add[i]) : add[i]);
684         return v;
685 }
686
687 static char **add_string_to_strings(char **strings, char *add)
688 {
689         char *v[2];
690         v[0] = add;
691         v[1] = NULL;
692         return add_strings_to_strings(strings, v, /*dup:*/ 0);
693 }
694
695 static void putenv_all(char **strings)
696 {
697         if (!strings)
698                 return;
699         while (*strings) {
700                 debug_printf_env("putenv '%s'\n", *strings);
701                 putenv(*strings++);
702         }
703 }
704
705 static char **putenv_all_and_save_old(char **strings)
706 {
707         char **old = NULL;
708         char **s = strings;
709
710         if (!strings)
711                 return old;
712         while (*strings) {
713                 char *v, *eq;
714
715                 eq = strchr(*strings, '=');
716                 if (eq) {
717                         *eq = '\0';
718                         v = getenv(*strings);
719                         *eq = '=';
720                         if (v) {
721                                 /* v points to VAL in VAR=VAL, go back to VAR */
722                                 v -= (eq - *strings) + 1;
723                                 old = add_string_to_strings(old, v);
724                         }
725                 }
726                 strings++;
727         }
728         putenv_all(s);
729         return old;
730 }
731
732 static void free_strings_and_unsetenv(char **strings, int unset)
733 {
734         char **v;
735
736         if (!strings)
737                 return;
738
739         v = strings;
740         while (*v) {
741                 if (unset) {
742                         debug_printf_env("unsetenv '%s'\n", *v);
743                         bb_unsetenv(*v);
744                 }
745                 free(*v++);
746         }
747         free(strings);
748 }
749
750 static void free_strings(char **strings)
751 {
752         free_strings_and_unsetenv(strings, 0);
753 }
754
755
756 /* Basic theory of signal handling in shell
757  * ========================================
758  * This does not describe what hush does, rather, it is current understanding
759  * what it _should_ do. If it doesn't, it's a bug.
760  * http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#trap
761  *
762  * Signals are handled only after each pipe ("cmd | cmd | cmd" thing)
763  * is finished or backgrounded. It is the same in interactive and
764  * non-interactive shells, and is the same regardless of whether
765  * a user trap handler is installed or a shell special one is in effect.
766  * ^C or ^Z from keyboard seem to execute "at once" because it usually
767  * backgrounds (i.e. stops) or kills all members of currently running
768  * pipe.
769  *
770  * Wait builtin in interruptible by signals for which user trap is set
771  * or by SIGINT in interactive shell.
772  *
773  * Trap handlers will execute even within trap handlers. (right?)
774  *
775  * User trap handlers are forgotten when subshell ("(cmd)") is entered. [TODO]
776  *
777  * If job control is off, backgrounded commands ("cmd &")
778  * have SIGINT, SIGQUIT set to SIG_IGN.
779  *
780  * Commands run in command substitution ("`cmd`")
781  * have SIGTTIN, SIGTTOU, SIGTSTP set to SIG_IGN.
782  *
783  * Ordinary commands have signals set to SIG_IGN/DFL set as inherited
784  * by the shell from its parent.
785  *
786  * Siganls which differ from SIG_DFL action
787  * (note: child (i.e., [v]forked) shell is not an interactive shell):
788  *
789  * SIGQUIT: ignore
790  * SIGTERM (interactive): ignore
791  * SIGHUP (interactive):
792  *    send SIGCONT to stopped jobs, send SIGHUP to all jobs and exit
793  * SIGTTIN, SIGTTOU, SIGTSTP (if job control is on): ignore
794  *    (note that ^Z is handled not by trapping SIGTSTP, but by seeing
795  *    that all pipe members are stopped) (right?)
796  * SIGINT (interactive): wait for last pipe, ignore the rest
797  *    of the command line, show prompt. NB: ^C does not send SIGINT
798  *    to interactive shell while shell is waiting for a pipe,
799  *    since shell is bg'ed (is not in foreground process group).
800  *    (check/expand this)
801  *    Example 1: this waits 5 sec, but does not execute ls:
802  *    "echo $$; sleep 5; ls -l" + "kill -INT <pid>"
803  *    Example 2: this does not wait and does not execute ls:
804  *    "echo $$; sleep 5 & wait; ls -l" + "kill -INT <pid>"
805  *    Example 3: this does not wait 5 sec, but executes ls:
806  *    "sleep 5; ls -l" + press ^C
807  *
808  * (What happens to signals which are IGN on shell start?)
809  * (What happens with signal mask on shell start?)
810  *
811  * Implementation in hush
812  * ======================
813  * We use in-kernel pending signal mask to determine which signals were sent.
814  * We block all signals which we don't want to take action immediately,
815  * i.e. we block all signals which need to have special handling as described
816  * above, and all signals which have traps set.
817  * After each pipe execution, we extract any pending signals via sigtimedwait()
818  * and act on them.
819  *
820  * unsigned non_DFL_mask: a mask of such "special" signals
821  * sigset_t blocked_set:  current blocked signal set
822  *
823  * "trap - SIGxxx":
824  *    clear bit in blocked_set unless it is also in non_DFL
825  * "trap 'cmd' SIGxxx":
826  *    set bit in blocked_set (even if 'cmd' is '')
827  * after [v]fork, if we plan to be a shell:
828  *    nothing for {} child shell (say, "true | { true; true; } | true")
829  *    unset all traps if () shell. [TODO]
830  * after [v]fork, if we plan to exec:
831  *    POSIX says pending signal mask is cleared in child - no need to clear it.
832  *    Restore blocked signal set to one inherited by shell just prior to exec.
833  *
834  * Note: as a result, we do not use signal handlers much. The only uses
835  * are to count SIGCHLDs [disabled - bug somewhere, + bloat]
836  * and to restore tty pgrp on signal-induced exit.
837  *
838  * TODO: check/fix wait builtin to be interruptible.
839  */
840
841 //static void SIGCHLD_handler(int sig UNUSED_PARAM)
842 //{
843 //      G.count_SIGCHLD++;
844 //}
845
846 /* called once at shell init */
847 static void init_signal_mask(void)
848 {
849         unsigned sig;
850         unsigned mask = (1 << SIGQUIT);
851 #if ENABLE_HUSH_INTERACTIVE
852         if (G.interactive_fd) {
853                 mask = 0
854                         | (1 << SIGQUIT)
855                         | (1 << SIGTERM)
856                         | (1 << SIGHUP)
857 #if ENABLE_HUSH_JOB
858                         | (1 << SIGTTIN) | (1 << SIGTTOU) | (1 << SIGTSTP)
859 #endif
860                         | (1 << SIGINT)
861                 ;
862         }
863 #endif
864         G.non_DFL_mask = mask;
865
866         sigprocmask(SIG_SETMASK, NULL, &G.blocked_set);
867         sig = 0;
868         while (mask) {
869                 if (mask & 1)
870                         sigaddset(&G.blocked_set, sig);
871                 mask >>= 1;
872                 sig++;
873         }
874         sigdelset(&G.blocked_set, SIGCHLD);
875         sigprocmask(SIG_SETMASK, &G.blocked_set, &G.inherited_set);
876 }
877
878 static int check_and_run_traps(int sig)
879 {
880         static const struct timespec zero_timespec = { 0, 0 };
881         smalluint save_rcode;
882         int last_sig = 0;
883
884         if (sig)
885                 goto jump_in;
886         while (1) {
887                 sig = sigtimedwait(&G.blocked_set, NULL, &zero_timespec);
888                 if (sig <= 0)
889                         break;
890  jump_in:
891                 last_sig = sig;
892                 if (G.traps && G.traps[sig]) {
893                         if (G.traps[sig][0]) {
894                                 /* We have user-defined handler */
895                                 char *argv[] = { NULL, xstrdup(G.traps[sig]), NULL };
896                                 save_rcode = G.last_return_code;
897                                 builtin_eval(argv);
898                                 free(argv[1]);
899                                 G.last_return_code = save_rcode;
900                         } /* else: "" trap, ignoring signal */
901                         continue;
902                 }
903                 /* not a trap: special action */
904                 switch (sig) {
905 //              case SIGCHLD:
906 //                      G.count_SIGCHLD++;
907 //                      break;
908                 case SIGINT:
909                         bb_putchar('\n');
910                         G.flag_SIGINT = 1;
911                         break;
912 //TODO
913 //              case SIGHUP: ...
914 //                      break;
915                 default: /* SIGTERM, SIGQUIT, SIGTTIN, SIGTTOU, SIGTSTP */
916                         break;
917                 }
918         }
919         return last_sig;
920 }
921
922 #if ENABLE_HUSH_JOB
923
924 /* Restores tty foreground process group, and exits.
925  * May be called as signal handler for fatal signal
926  * (will faithfully resend signal to itself, producing correct exit state)
927  * or called directly with -EXITCODE.
928  * We also call it if xfunc is exiting. */
929 static void sigexit(int sig) NORETURN;
930 static void sigexit(int sig)
931 {
932         /* Disable all signals: job control, SIGPIPE, etc. */
933         sigprocmask_allsigs(SIG_BLOCK);
934
935 #if ENABLE_HUSH_INTERACTIVE
936         /* Careful: we can end up here after [v]fork. Do not restore
937          * tty pgrp then, only top-level shell process does that */
938         if (G.interactive_fd && getpid() == G.root_pid)
939                 tcsetpgrp(G.interactive_fd, G.saved_tty_pgrp);
940 #endif
941
942         /* Not a signal, just exit */
943         if (sig <= 0)
944                 _exit(- sig);
945
946         kill_myself_with_sig(sig); /* does not return */
947 }
948
949 /* helper */
950 static void maybe_set_sighandler(int sig)
951 {
952         void (*handler)(int);
953         /* non_DFL_mask'ed signals are, well, masked,
954          * no need to set handler for them.
955          */
956         if (!((G.non_DFL_mask >> sig) & 1)) {
957                 handler = signal(sig, sigexit);
958                 if (handler == SIG_IGN) /* oops... restore back to IGN! */
959                         signal(sig, handler);
960         }
961 }
962 /* Used only to set handler to restore pgrp on exit */
963 static void set_fatal_signals_to_sigexit(void)
964 {
965         if (HUSH_DEBUG) {
966                 maybe_set_sighandler(SIGILL );
967                 maybe_set_sighandler(SIGFPE );
968                 maybe_set_sighandler(SIGBUS );
969                 maybe_set_sighandler(SIGSEGV);
970                 maybe_set_sighandler(SIGTRAP);
971         } /* else: hush is perfect. what SEGV? */
972
973         maybe_set_sighandler(SIGABRT);
974
975         /* bash 3.2 seems to handle these just like 'fatal' ones */
976         maybe_set_sighandler(SIGPIPE);
977         maybe_set_sighandler(SIGALRM);
978         maybe_set_sighandler(SIGHUP );
979
980         /* if we aren't interactive... but in this case
981          * we never want to restore pgrp on exit, and this fn is not called */
982         /*maybe_set_sighandler(SIGTERM);*/
983         /*maybe_set_sighandler(SIGINT );*/
984 }
985 /* Used only to suppress ^Z in `cmd` */
986 static void set_jobctrl_signals_to_IGN(void)
987 {
988         bb_signals(0
989                 + (1 << SIGTSTP)
990                 + (1 << SIGTTIN)
991                 + (1 << SIGTTOU)
992                 , SIG_IGN);
993 }
994
995 #else /* !JOB */
996
997 #define set_fatal_signals_to_sigexit(handler) ((void)0)
998 #define set_jobctrl_signals_to_IGN(handler)  ((void)0)
999
1000 #endif /* JOB */
1001
1002 /* Restores tty foreground process group, and exits. */
1003 static void hush_exit(int exitcode) NORETURN;
1004 static void hush_exit(int exitcode)
1005 {
1006         if (G.traps && G.traps[0] && G.traps[0][0]) {
1007                 char *argv[] = { NULL, xstrdup(G.traps[0]), NULL };
1008                 builtin_eval(argv);
1009                 free(argv[1]);
1010         }
1011
1012 #if ENABLE_HUSH_JOB
1013         fflush(NULL); /* flush all streams */
1014         sigexit(- (exitcode & 0xff));
1015 #else
1016         exit(exitcode);
1017 #endif
1018 }
1019
1020
1021 static const char *set_cwd(void)
1022 {
1023         /* xrealloc_getcwd_or_warn(arg) calls free(arg),
1024          * we must not try to free(bb_msg_unknown) */
1025         if (G.cwd == bb_msg_unknown)
1026                 G.cwd = NULL;
1027         G.cwd = xrealloc_getcwd_or_warn((char *)G.cwd);
1028         if (!G.cwd)
1029                 G.cwd = bb_msg_unknown;
1030         return G.cwd;
1031 }
1032
1033
1034 /* Get/check local shell variables */
1035 static struct variable *get_local_var(const char *name)
1036 {
1037         struct variable *cur;
1038         int len;
1039
1040         if (!name)
1041                 return NULL;
1042         len = strlen(name);
1043         for (cur = G.top_var; cur; cur = cur->next) {
1044                 if (strncmp(cur->varstr, name, len) == 0 && cur->varstr[len] == '=')
1045                         return cur;
1046         }
1047         return NULL;
1048 }
1049
1050 /* Basically useful version until someone wants to get fancier,
1051  * see the bash man page under "Parameter Expansion" */
1052 static const char *lookup_param(const char *src)
1053 {
1054         struct variable *var = get_local_var(src);
1055         if (var)
1056                 return strchr(var->varstr, '=') + 1;
1057         return NULL;
1058 }
1059
1060 /* str holds "NAME=VAL" and is expected to be malloced.
1061  * We take ownership of it.
1062  * flg_export is used by:
1063  *  0: do not export
1064  *  1: export
1065  * -1: if NAME is set, leave export status alone
1066  *     if NAME is not set, do not export
1067  */
1068 static int set_local_var(char *str, int flg_export)
1069 {
1070         struct variable *cur;
1071         char *value;
1072         int name_len;
1073
1074         value = strchr(str, '=');
1075         if (!value) { /* not expected to ever happen? */
1076                 free(str);
1077                 return -1;
1078         }
1079
1080         name_len = value - str + 1; /* including '=' */
1081         cur = G.top_var; /* cannot be NULL (we have HUSH_VERSION and it's RO) */
1082         while (1) {
1083                 if (strncmp(cur->varstr, str, name_len) != 0) {
1084                         if (!cur->next) {
1085                                 /* Bail out. Note that now cur points
1086                                  * to last var in linked list */
1087                                 break;
1088                         }
1089                         cur = cur->next;
1090                         continue;
1091                 }
1092                 /* We found an existing var with this name */
1093                 *value = '\0';
1094                 if (cur->flg_read_only) {
1095                         bb_error_msg("%s: readonly variable", str);
1096                         free(str);
1097                         return -1;
1098                 }
1099                 debug_printf_env("%s: unsetenv '%s'\n", __func__, str);
1100                 unsetenv(str); /* just in case */
1101                 *value = '=';
1102                 if (strcmp(cur->varstr, str) == 0) {
1103  free_and_exp:
1104                         free(str);
1105                         goto exp;
1106                 }
1107                 if (cur->max_len >= strlen(str)) {
1108                         /* This one is from startup env, reuse space */
1109                         strcpy(cur->varstr, str);
1110                         goto free_and_exp;
1111                 }
1112                 /* max_len == 0 signifies "malloced" var, which we can
1113                  * (and has to) free */
1114                 if (!cur->max_len)
1115                         free(cur->varstr);
1116                 cur->max_len = 0;
1117                 goto set_str_and_exp;
1118         }
1119
1120         /* Not found - create next variable struct */
1121         cur->next = xzalloc(sizeof(*cur));
1122         cur = cur->next;
1123
1124  set_str_and_exp:
1125         cur->varstr = str;
1126  exp:
1127         if (flg_export == 1)
1128                 cur->flg_export = 1;
1129         if (cur->flg_export) {
1130                 debug_printf_env("%s: putenv '%s'\n", __func__, cur->varstr);
1131                 return putenv(cur->varstr);
1132         }
1133         return 0;
1134 }
1135
1136 static int unset_local_var(const char *name)
1137 {
1138         struct variable *cur;
1139         struct variable *prev = prev; /* for gcc */
1140         int name_len;
1141
1142         if (!name)
1143                 return EXIT_SUCCESS;
1144         name_len = strlen(name);
1145         cur = G.top_var;
1146         while (cur) {
1147                 if (strncmp(cur->varstr, name, name_len) == 0 && cur->varstr[name_len] == '=') {
1148                         if (cur->flg_read_only) {
1149                                 bb_error_msg("%s: readonly variable", name);
1150                                 return EXIT_FAILURE;
1151                         }
1152                         /* prev is ok to use here because 1st variable, HUSH_VERSION,
1153                          * is ro, and we cannot reach this code on the 1st pass */
1154                         prev->next = cur->next;
1155                         debug_printf_env("%s: unsetenv '%s'\n", __func__, cur->varstr);
1156                         bb_unsetenv(cur->varstr);
1157                         if (!cur->max_len)
1158                                 free(cur->varstr);
1159                         free(cur);
1160                         return EXIT_SUCCESS;
1161                 }
1162                 prev = cur;
1163                 cur = cur->next;
1164         }
1165         return EXIT_SUCCESS;
1166 }
1167
1168 #if ENABLE_SH_MATH_SUPPORT
1169 #define is_name(c)      ((c) == '_' || isalpha((unsigned char)(c)))
1170 #define is_in_name(c)   ((c) == '_' || isalnum((unsigned char)(c)))
1171 static char *endofname(const char *name)
1172 {
1173         char *p;
1174
1175         p = (char *) name;
1176         if (!is_name(*p))
1177                 return p;
1178         while (*++p) {
1179                 if (!is_in_name(*p))
1180                         break;
1181         }
1182         return p;
1183 }
1184
1185 static void arith_set_local_var(const char *name, const char *val, int flags)
1186 {
1187         /* arith code doesnt malloc space, so do it for it */
1188         char *var = xasprintf("%s=%s", name, val);
1189         set_local_var(var, flags);
1190 }
1191 #endif
1192
1193
1194 /*
1195  * in_str support
1196  */
1197 static int static_get(struct in_str *i)
1198 {
1199         int ch = *i->p++;
1200         if (ch == '\0') return EOF;
1201         return ch;
1202 }
1203
1204 static int static_peek(struct in_str *i)
1205 {
1206         return *i->p;
1207 }
1208
1209 #if ENABLE_HUSH_INTERACTIVE
1210
1211 static void cmdedit_set_initial_prompt(void)
1212 {
1213         if (ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
1214                 G.PS1 = getenv("PS1");
1215                 if (G.PS1 == NULL)
1216                         G.PS1 = "\\w \\$ ";
1217         } else
1218                 G.PS1 = NULL;
1219 }
1220
1221 static const char* setup_prompt_string(int promptmode)
1222 {
1223         const char *prompt_str;
1224         debug_printf("setup_prompt_string %d ", promptmode);
1225         if (!ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
1226                 /* Set up the prompt */
1227                 if (promptmode == 0) { /* PS1 */
1228                         free((char*)G.PS1);
1229                         G.PS1 = xasprintf("%s %c ", G.cwd, (geteuid() != 0) ? '$' : '#');
1230                         prompt_str = G.PS1;
1231                 } else
1232                         prompt_str = G.PS2;
1233         } else
1234                 prompt_str = (promptmode == 0) ? G.PS1 : G.PS2;
1235         debug_printf("result '%s'\n", prompt_str);
1236         return prompt_str;
1237 }
1238
1239 static void get_user_input(struct in_str *i)
1240 {
1241         int r;
1242         const char *prompt_str;
1243
1244         prompt_str = setup_prompt_string(i->promptmode);
1245 #if ENABLE_FEATURE_EDITING
1246         /* Enable command line editing only while a command line
1247          * is actually being read */
1248         do {
1249                 G.flag_SIGINT = 0;
1250                 /* buglet: SIGINT will not make new prompt to appear _at once_,
1251                  * only after <Enter>. (^C will work) */
1252                 r = read_line_input(prompt_str, G.user_input_buf, BUFSIZ-1, G.line_input_state);
1253                 /* catch *SIGINT* etc (^C is handled by read_line_input) */
1254                 check_and_run_traps(0);
1255         } while (r == 0 || G.flag_SIGINT); /* repeat if ^C or SIGINT */
1256         i->eof_flag = (r < 0);
1257         if (i->eof_flag) { /* EOF/error detected */
1258                 G.user_input_buf[0] = EOF; /* yes, it will be truncated, it's ok */
1259                 G.user_input_buf[1] = '\0';
1260         }
1261 #else
1262         do {
1263                 G.flag_SIGINT = 0;
1264                 fputs(prompt_str, stdout);
1265                 fflush(stdout);
1266                 G.user_input_buf[0] = r = fgetc(i->file);
1267                 /*G.user_input_buf[1] = '\0'; - already is and never changed */
1268 //do we need check_and_run_traps(0)? (maybe only if stdin)
1269         } while (G.flag_SIGINT);
1270         i->eof_flag = (r == EOF);
1271 #endif
1272         i->p = G.user_input_buf;
1273 }
1274
1275 #endif  /* INTERACTIVE */
1276
1277 /* This is the magic location that prints prompts
1278  * and gets data back from the user */
1279 static int file_get(struct in_str *i)
1280 {
1281         int ch;
1282
1283         /* If there is data waiting, eat it up */
1284         if (i->p && *i->p) {
1285 #if ENABLE_HUSH_INTERACTIVE
1286  take_cached:
1287 #endif
1288                 ch = *i->p++;
1289                 if (i->eof_flag && !*i->p)
1290                         ch = EOF;
1291         } else {
1292                 /* need to double check i->file because we might be doing something
1293                  * more complicated by now, like sourcing or substituting. */
1294 #if ENABLE_HUSH_INTERACTIVE
1295                 if (G.interactive_fd && i->promptme && i->file == stdin) {
1296                         do {
1297                                 get_user_input(i);
1298                         } while (!*i->p); /* need non-empty line */
1299                         i->promptmode = 1; /* PS2 */
1300                         i->promptme = 0;
1301                         goto take_cached;
1302                 }
1303 #endif
1304                 ch = fgetc(i->file);
1305         }
1306         debug_printf("file_get: got a '%c' %d\n", ch, ch);
1307 #if ENABLE_HUSH_INTERACTIVE
1308         if (ch == '\n')
1309                 i->promptme = 1;
1310 #endif
1311         return ch;
1312 }
1313
1314 /* All the callers guarantee this routine will never be
1315  * used right after a newline, so prompting is not needed.
1316  */
1317 static int file_peek(struct in_str *i)
1318 {
1319         int ch;
1320         if (i->p && *i->p) {
1321                 if (i->eof_flag && !i->p[1])
1322                         return EOF;
1323                 return *i->p;
1324         }
1325         ch = fgetc(i->file);
1326         i->eof_flag = (ch == EOF);
1327         i->peek_buf[0] = ch;
1328         i->peek_buf[1] = '\0';
1329         i->p = i->peek_buf;
1330         debug_printf("file_peek: got a '%c' %d\n", *i->p, *i->p);
1331         return ch;
1332 }
1333
1334 static void setup_file_in_str(struct in_str *i, FILE *f)
1335 {
1336         i->peek = file_peek;
1337         i->get = file_get;
1338 #if ENABLE_HUSH_INTERACTIVE
1339         i->promptme = 1;
1340         i->promptmode = 0; /* PS1 */
1341 #endif
1342         i->file = f;
1343         i->p = NULL;
1344 }
1345
1346 static void setup_string_in_str(struct in_str *i, const char *s)
1347 {
1348         i->peek = static_peek;
1349         i->get = static_get;
1350 #if ENABLE_HUSH_INTERACTIVE
1351         i->promptme = 1;
1352         i->promptmode = 0; /* PS1 */
1353 #endif
1354         i->p = s;
1355         i->eof_flag = 0;
1356 }
1357
1358
1359 /*
1360  * o_string support
1361  */
1362 #define B_CHUNK  (32 * sizeof(char*))
1363
1364 static void o_reset(o_string *o)
1365 {
1366         o->length = 0;
1367         o->nonnull = 0;
1368         if (o->data)
1369                 o->data[0] = '\0';
1370 }
1371
1372 static void o_free(o_string *o)
1373 {
1374         free(o->data);
1375         memset(o, 0, sizeof(*o));
1376 }
1377
1378 static void o_grow_by(o_string *o, int len)
1379 {
1380         if (o->length + len > o->maxlen) {
1381                 o->maxlen += (2*len > B_CHUNK ? 2*len : B_CHUNK);
1382                 o->data = xrealloc(o->data, 1 + o->maxlen);
1383         }
1384 }
1385
1386 static void o_addchr(o_string *o, int ch)
1387 {
1388         debug_printf("o_addchr: '%c' o->length=%d o=%p\n", ch, o->length, o);
1389         o_grow_by(o, 1);
1390         o->data[o->length] = ch;
1391         o->length++;
1392         o->data[o->length] = '\0';
1393 }
1394
1395 static void o_addstr(o_string *o, const char *str, int len)
1396 {
1397         o_grow_by(o, len);
1398         memcpy(&o->data[o->length], str, len);
1399         o->length += len;
1400         o->data[o->length] = '\0';
1401 }
1402
1403 static void o_addstrauto(o_string *o, const char *str)
1404 {
1405         o_addstr(o, str, strlen(str) + 1);
1406 }
1407
1408 static void o_addstr_duplicate_backslash(o_string *o, const char *str, int len)
1409 {
1410         while (len) {
1411                 o_addchr(o, *str);
1412                 if (*str++ == '\\'
1413                  && (*str != '*' && *str != '?' && *str != '[')
1414                 ) {
1415                         o_addchr(o, '\\');
1416                 }
1417                 len--;
1418         }
1419 }
1420
1421 /* My analysis of quoting semantics tells me that state information
1422  * is associated with a destination, not a source.
1423  */
1424 static void o_addqchr(o_string *o, int ch)
1425 {
1426         int sz = 1;
1427         char *found = strchr("*?[\\", ch);
1428         if (found)
1429                 sz++;
1430         o_grow_by(o, sz);
1431         if (found) {
1432                 o->data[o->length] = '\\';
1433                 o->length++;
1434         }
1435         o->data[o->length] = ch;
1436         o->length++;
1437         o->data[o->length] = '\0';
1438 }
1439
1440 static void o_addQchr(o_string *o, int ch)
1441 {
1442         int sz = 1;
1443         if (o->o_escape && strchr("*?[\\", ch)) {
1444                 sz++;
1445                 o->data[o->length] = '\\';
1446                 o->length++;
1447         }
1448         o_grow_by(o, sz);
1449         o->data[o->length] = ch;
1450         o->length++;
1451         o->data[o->length] = '\0';
1452 }
1453
1454 static void o_addQstr(o_string *o, const char *str, int len)
1455 {
1456         if (!o->o_escape) {
1457                 o_addstr(o, str, len);
1458                 return;
1459         }
1460         while (len) {
1461                 char ch;
1462                 int sz;
1463                 int ordinary_cnt = strcspn(str, "*?[\\");
1464                 if (ordinary_cnt > len) /* paranoia */
1465                         ordinary_cnt = len;
1466                 o_addstr(o, str, ordinary_cnt);
1467                 if (ordinary_cnt == len)
1468                         return;
1469                 str += ordinary_cnt;
1470                 len -= ordinary_cnt + 1; /* we are processing + 1 char below */
1471
1472                 ch = *str++;
1473                 sz = 1;
1474                 if (ch) { /* it is necessarily one of "*?[\\" */
1475                         sz++;
1476                         o->data[o->length] = '\\';
1477                         o->length++;
1478                 }
1479                 o_grow_by(o, sz);
1480                 o->data[o->length] = ch;
1481                 o->length++;
1482                 o->data[o->length] = '\0';
1483         }
1484 }
1485
1486 /* A special kind of o_string for $VAR and `cmd` expansion.
1487  * It contains char* list[] at the beginning, which is grown in 16 element
1488  * increments. Actual string data starts at the next multiple of 16 * (char*).
1489  * list[i] contains an INDEX (int!) into this string data.
1490  * It means that if list[] needs to grow, data needs to be moved higher up
1491  * but list[i]'s need not be modified.
1492  * NB: remembering how many list[i]'s you have there is crucial.
1493  * o_finalize_list() operation post-processes this structure - calculates
1494  * and stores actual char* ptrs in list[]. Oh, it NULL terminates it as well.
1495  */
1496 #if DEBUG_EXPAND || DEBUG_GLOB
1497 static void debug_print_list(const char *prefix, o_string *o, int n)
1498 {
1499         char **list = (char**)o->data;
1500         int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
1501         int i = 0;
1502         fprintf(stderr, "%s: list:%p n:%d string_start:%d length:%d maxlen:%d\n",
1503                         prefix, list, n, string_start, o->length, o->maxlen);
1504         while (i < n) {
1505                 fprintf(stderr, " list[%d]=%d '%s' %p\n", i, (int)list[i],
1506                                 o->data + (int)list[i] + string_start,
1507                                 o->data + (int)list[i] + string_start);
1508                 i++;
1509         }
1510         if (n) {
1511                 const char *p = o->data + (int)list[n - 1] + string_start;
1512                 fprintf(stderr, " total_sz:%ld\n", (long)((p + strlen(p) + 1) - o->data));
1513         }
1514 }
1515 #else
1516 #define debug_print_list(prefix, o, n) ((void)0)
1517 #endif
1518
1519 /* n = o_save_ptr_helper(str, n) "starts new string" by storing an index value
1520  * in list[n] so that it points past last stored byte so far.
1521  * It returns n+1. */
1522 static int o_save_ptr_helper(o_string *o, int n)
1523 {
1524         char **list = (char**)o->data;
1525         int string_start;
1526         int string_len;
1527
1528         if (!o->has_empty_slot) {
1529                 string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
1530                 string_len = o->length - string_start;
1531                 if (!(n & 0xf)) { /* 0, 0x10, 0x20...? */
1532                         debug_printf_list("list[%d]=%d string_start=%d (growing)\n", n, string_len, string_start);
1533                         /* list[n] points to string_start, make space for 16 more pointers */
1534                         o->maxlen += 0x10 * sizeof(list[0]);
1535                         o->data = xrealloc(o->data, o->maxlen + 1);
1536                         list = (char**)o->data;
1537                         memmove(list + n + 0x10, list + n, string_len);
1538                         o->length += 0x10 * sizeof(list[0]);
1539                 } else
1540                         debug_printf_list("list[%d]=%d string_start=%d\n", n, string_len, string_start);
1541         } else {
1542                 /* We have empty slot at list[n], reuse without growth */
1543                 string_start = ((n+1 + 0xf) & ~0xf) * sizeof(list[0]); /* NB: n+1! */
1544                 string_len = o->length - string_start;
1545                 debug_printf_list("list[%d]=%d string_start=%d (empty slot)\n", n, string_len, string_start);
1546                 o->has_empty_slot = 0;
1547         }
1548         list[n] = (char*)(ptrdiff_t)string_len;
1549         return n + 1;
1550 }
1551
1552 /* "What was our last o_save_ptr'ed position (byte offset relative o->data)?" */
1553 static int o_get_last_ptr(o_string *o, int n)
1554 {
1555         char **list = (char**)o->data;
1556         int string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
1557
1558         return ((int)(ptrdiff_t)list[n-1]) + string_start;
1559 }
1560
1561 /* o_glob performs globbing on last list[], saving each result
1562  * as a new list[]. */
1563 static int o_glob(o_string *o, int n)
1564 {
1565         glob_t globdata;
1566         int gr;
1567         char *pattern;
1568
1569         debug_printf_glob("start o_glob: n:%d o->data:%p\n", n, o->data);
1570         if (!o->data)
1571                 return o_save_ptr_helper(o, n);
1572         pattern = o->data + o_get_last_ptr(o, n);
1573         debug_printf_glob("glob pattern '%s'\n", pattern);
1574         if (!glob_needed(pattern)) {
1575  literal:
1576                 o->length = unbackslash(pattern) - o->data;
1577                 debug_printf_glob("glob pattern '%s' is literal\n", pattern);
1578                 return o_save_ptr_helper(o, n);
1579         }
1580
1581         memset(&globdata, 0, sizeof(globdata));
1582         gr = glob(pattern, 0, NULL, &globdata);
1583         debug_printf_glob("glob('%s'):%d\n", pattern, gr);
1584         if (gr == GLOB_NOSPACE)
1585                 bb_error_msg_and_die("out of memory during glob");
1586         if (gr == GLOB_NOMATCH) {
1587                 globfree(&globdata);
1588                 goto literal;
1589         }
1590         if (gr != 0) { /* GLOB_ABORTED ? */
1591 //TODO: testcase for bad glob pattern behavior
1592                 bb_error_msg("glob(3) error %d on '%s'", gr, pattern);
1593         }
1594         if (globdata.gl_pathv && globdata.gl_pathv[0]) {
1595                 char **argv = globdata.gl_pathv;
1596                 o->length = pattern - o->data; /* "forget" pattern */
1597                 while (1) {
1598                         o_addstrauto(o, *argv);
1599                         n = o_save_ptr_helper(o, n);
1600                         argv++;
1601                         if (!*argv)
1602                                 break;
1603                 }
1604         }
1605         globfree(&globdata);
1606         if (DEBUG_GLOB)
1607                 debug_print_list("o_glob returning", o, n);
1608         return n;
1609 }
1610
1611 /* If o->o_glob == 1, glob the string so far remembered.
1612  * Otherwise, just finish current list[] and start new */
1613 static int o_save_ptr(o_string *o, int n)
1614 {
1615         if (o->o_glob) { /* if globbing is requested */
1616                 /* If o->has_empty_slot, list[n] was already globbed
1617                  * (if it was requested back then when it was filled)
1618                  * so don't do that again! */
1619                 if (!o->has_empty_slot)
1620                         return o_glob(o, n); /* o_save_ptr_helper is inside */
1621         }
1622         return o_save_ptr_helper(o, n);
1623 }
1624
1625 /* "Please convert list[n] to real char* ptrs, and NULL terminate it." */
1626 static char **o_finalize_list(o_string *o, int n)
1627 {
1628         char **list;
1629         int string_start;
1630
1631         n = o_save_ptr(o, n); /* force growth for list[n] if necessary */
1632         if (DEBUG_EXPAND)
1633                 debug_print_list("finalized", o, n);
1634         debug_printf_expand("finalized n:%d\n", n);
1635         list = (char**)o->data;
1636         string_start = ((n + 0xf) & ~0xf) * sizeof(list[0]);
1637         list[--n] = NULL;
1638         while (n) {
1639                 n--;
1640                 list[n] = o->data + (int)(ptrdiff_t)list[n] + string_start;
1641         }
1642         return list;
1643 }
1644
1645
1646 /* Expansion can recurse */
1647 #if ENABLE_HUSH_TICK
1648 static int process_command_subs(o_string *dest,
1649                 struct in_str *input, const char *subst_end);
1650 #endif
1651 static char *expand_string_to_string(const char *str);
1652 static int parse_stream_dquoted(o_string *dest, struct in_str *input, int dquote_end);
1653
1654 /* expand_strvec_to_strvec() takes a list of strings, expands
1655  * all variable references within and returns a pointer to
1656  * a list of expanded strings, possibly with larger number
1657  * of strings. (Think VAR="a b"; echo $VAR).
1658  * This new list is allocated as a single malloc block.
1659  * NULL-terminated list of char* pointers is at the beginning of it,
1660  * followed by strings themself.
1661  * Caller can deallocate entire list by single free(list). */
1662
1663 /* Store given string, finalizing the word and starting new one whenever
1664  * we encounter IFS char(s). This is used for expanding variable values.
1665  * End-of-string does NOT finalize word: think about 'echo -$VAR-' */
1666 static int expand_on_ifs(o_string *output, int n, const char *str)
1667 {
1668         while (1) {
1669                 int word_len = strcspn(str, G.ifs);
1670                 if (word_len) {
1671                         if (output->o_escape || !output->o_glob)
1672                                 o_addQstr(output, str, word_len);
1673                         else /* protect backslashes against globbing up :) */
1674                                 o_addstr_duplicate_backslash(output, str, word_len);
1675                         str += word_len;
1676                 }
1677                 if (!*str)  /* EOL - do not finalize word */
1678                         break;
1679                 o_addchr(output, '\0');
1680                 debug_print_list("expand_on_ifs", output, n);
1681                 n = o_save_ptr(output, n);
1682                 str += strspn(str, G.ifs); /* skip ifs chars */
1683         }
1684         debug_print_list("expand_on_ifs[1]", output, n);
1685         return n;
1686 }
1687
1688 /* Expand all variable references in given string, adding words to list[]
1689  * at n, n+1,... positions. Return updated n (so that list[n] is next one
1690  * to be filled). This routine is extremely tricky: has to deal with
1691  * variables/parameters with whitespace, $* and $@, and constructs like
1692  * 'echo -$*-'. If you play here, you must run testsuite afterwards! */
1693 static int expand_vars_to_list(o_string *output, int n, char *arg, char or_mask)
1694 {
1695         /* or_mask is either 0 (normal case) or 0x80
1696          * (expansion of right-hand side of assignment == 1-element expand.
1697          * It will also do no globbing, and thus we must not backslash-quote!) */
1698
1699         char first_ch, ored_ch;
1700         int i;
1701         const char *val;
1702         char *p;
1703
1704         ored_ch = 0;
1705
1706         debug_printf_expand("expand_vars_to_list: arg '%s'\n", arg);
1707         debug_print_list("expand_vars_to_list", output, n);
1708         n = o_save_ptr(output, n);
1709         debug_print_list("expand_vars_to_list[0]", output, n);
1710
1711         while ((p = strchr(arg, SPECIAL_VAR_SYMBOL)) != NULL) {
1712 #if ENABLE_HUSH_TICK
1713                 o_string subst_result = NULL_O_STRING;
1714 #endif
1715 #if ENABLE_SH_MATH_SUPPORT
1716                 char arith_buf[sizeof(arith_t)*3 + 2];
1717 #endif
1718                 o_addstr(output, arg, p - arg);
1719                 debug_print_list("expand_vars_to_list[1]", output, n);
1720                 arg = ++p;
1721                 p = strchr(p, SPECIAL_VAR_SYMBOL);
1722
1723                 first_ch = arg[0] | or_mask; /* forced to "quoted" if or_mask = 0x80 */
1724                 /* "$@" is special. Even if quoted, it can still
1725                  * expand to nothing (not even an empty string) */
1726                 if ((first_ch & 0x7f) != '@')
1727                         ored_ch |= first_ch;
1728
1729                 val = NULL;
1730                 switch (first_ch & 0x7f) {
1731                 /* Highest bit in first_ch indicates that var is double-quoted */
1732                 case '$': /* pid */
1733                         val = utoa(G.root_pid);
1734                         break;
1735                 case '!': /* bg pid */
1736                         val = G.last_bg_pid ? utoa(G.last_bg_pid) : (char*)"";
1737                         break;
1738                 case '?': /* exitcode */
1739                         val = utoa(G.last_return_code);
1740                         break;
1741                 case '#': /* argc */
1742                         if (arg[1] != SPECIAL_VAR_SYMBOL)
1743                                 /* actually, it's a ${#var} */
1744                                 goto case_default;
1745                         val = utoa(G.global_argc ? G.global_argc-1 : 0);
1746                         break;
1747                 case '*':
1748                 case '@':
1749                         i = 1;
1750                         if (!G.global_argv[i])
1751                                 break;
1752                         ored_ch |= first_ch; /* do it for "$@" _now_, when we know it's not empty */
1753                         if (!(first_ch & 0x80)) { /* unquoted $* or $@ */
1754                                 smallint sv = output->o_escape;
1755                                 /* unquoted var's contents should be globbed, so don't escape */
1756                                 output->o_escape = 0;
1757                                 while (G.global_argv[i]) {
1758                                         n = expand_on_ifs(output, n, G.global_argv[i]);
1759                                         debug_printf_expand("expand_vars_to_list: argv %d (last %d)\n", i, G.global_argc - 1);
1760                                         if (G.global_argv[i++][0] && G.global_argv[i]) {
1761                                                 /* this argv[] is not empty and not last:
1762                                                  * put terminating NUL, start new word */
1763                                                 o_addchr(output, '\0');
1764                                                 debug_print_list("expand_vars_to_list[2]", output, n);
1765                                                 n = o_save_ptr(output, n);
1766                                                 debug_print_list("expand_vars_to_list[3]", output, n);
1767                                         }
1768                                 }
1769                                 output->o_escape = sv;
1770                         } else
1771                         /* If or_mask is nonzero, we handle assignment 'a=....$@.....'
1772                          * and in this case should treat it like '$*' - see 'else...' below */
1773                         if (first_ch == ('@'|0x80) && !or_mask) { /* quoted $@ */
1774                                 while (1) {
1775                                         o_addQstr(output, G.global_argv[i], strlen(G.global_argv[i]));
1776                                         if (++i >= G.global_argc)
1777                                                 break;
1778                                         o_addchr(output, '\0');
1779                                         debug_print_list("expand_vars_to_list[4]", output, n);
1780                                         n = o_save_ptr(output, n);
1781                                 }
1782                         } else { /* quoted $*: add as one word */
1783                                 while (1) {
1784                                         o_addQstr(output, G.global_argv[i], strlen(G.global_argv[i]));
1785                                         if (!G.global_argv[++i])
1786                                                 break;
1787                                         if (G.ifs[0])
1788                                                 o_addchr(output, G.ifs[0]);
1789                                 }
1790                         }
1791                         break;
1792                 case SPECIAL_VAR_SYMBOL: /* <SPECIAL_VAR_SYMBOL><SPECIAL_VAR_SYMBOL> */
1793                         /* "Empty variable", used to make "" etc to not disappear */
1794                         arg++;
1795                         ored_ch = 0x80;
1796                         break;
1797 #if ENABLE_HUSH_TICK
1798                 case '`': { /* <SPECIAL_VAR_SYMBOL>`cmd<SPECIAL_VAR_SYMBOL> */
1799                         struct in_str input;
1800                         *p = '\0';
1801                         arg++;
1802 //TODO: can we just stuff it into "output" directly?
1803                         debug_printf_subst("SUBST '%s' first_ch %x\n", arg, first_ch);
1804                         setup_string_in_str(&input, arg);
1805                         process_command_subs(&subst_result, &input, NULL);
1806                         debug_printf_subst("SUBST RES '%s'\n", subst_result.data);
1807                         val = subst_result.data;
1808                         goto store_val;
1809                 }
1810 #endif
1811 #if ENABLE_SH_MATH_SUPPORT
1812                 case '+': { /* <SPECIAL_VAR_SYMBOL>+cmd<SPECIAL_VAR_SYMBOL> */
1813                         arith_eval_hooks_t hooks;
1814                         arith_t res;
1815                         int errcode;
1816                         char *exp_str;
1817
1818                         arg++; /* skip '+' */
1819                         *p = '\0'; /* replace trailing <SPECIAL_VAR_SYMBOL> */
1820                         debug_printf_subst("ARITH '%s' first_ch %x\n", arg, first_ch);
1821
1822                         /* Optional: skip expansion if expr is simple ("a + 3", "i++" etc) */
1823                         exp_str = arg;
1824                         while (1) {
1825                                 unsigned char c = *exp_str++;
1826                                 if (c == '\0') {
1827                                         exp_str = NULL;
1828                                         goto skip_expand;
1829                                 }
1830                                 if (isdigit(c))
1831                                         continue;
1832                                 if (strchr(" \t+-*/%_", c) != NULL)
1833                                         continue;
1834                                 c |= 0x20; /* tolower */
1835                                 if (c >= 'a' && c <= 'z')
1836                                         continue;
1837                                 break;
1838                         }
1839                         /* We need to expand. Example: "echo $(($a + 1)) $((1 + $((2)) ))" */
1840                         {
1841                                 struct in_str input;
1842                                 o_string dest = NULL_O_STRING;
1843
1844                                 setup_string_in_str(&input, arg);
1845                                 parse_stream_dquoted(&dest, &input, EOF);
1846                                 //bb_error_msg("'%s' -> '%s'", arg, dest.data);
1847                                 exp_str = expand_string_to_string(dest.data);
1848                                 //bb_error_msg("'%s' -> '%s'", dest.data, exp_str);
1849                                 o_free(&dest);
1850                         }
1851  skip_expand:
1852                         hooks.lookupvar = lookup_param;
1853                         hooks.setvar = arith_set_local_var;
1854                         hooks.endofname = endofname;
1855                         res = arith(exp_str ? exp_str : arg, &errcode, &hooks);
1856                         free(exp_str);
1857
1858                         if (errcode < 0) {
1859                                 switch (errcode) {
1860                                 case -3: maybe_die("arith", "exponent less than 0"); break;
1861                                 case -2: maybe_die("arith", "divide by zero"); break;
1862                                 case -5: maybe_die("arith", "expression recursion loop detected"); break;
1863                                 default: maybe_die("arith", "syntax error"); break;
1864                                 }
1865                         }
1866                         debug_printf_subst("ARITH RES '"arith_t_fmt"'\n", res);
1867                         sprintf(arith_buf, arith_t_fmt, res);
1868                         val = arith_buf;
1869                         break;
1870                 }
1871 #endif
1872                 default: /* <SPECIAL_VAR_SYMBOL>varname<SPECIAL_VAR_SYMBOL> */
1873                 case_default: {
1874                         bool exp_len = false, exp_null = false;
1875                         char *var = arg, exp_save, exp_op, *exp_word;
1876                         size_t exp_off = 0;
1877                         *p = '\0';
1878                         arg[0] = first_ch & 0x7f;
1879
1880                         /* prepare for expansions */
1881                         if (var[0] == '#') {
1882                                 /* handle length expansion ${#var} */
1883                                 exp_len = true;
1884                                 ++var;
1885                         } else {
1886                                 /* maybe handle parameter expansion */
1887                                 exp_off = strcspn(var, ":-=+?");
1888                                 if (!var[exp_off])
1889                                         exp_off = 0;
1890                                 if (exp_off) {
1891                                         exp_save = var[exp_off];
1892                                         exp_null = exp_save == ':';
1893                                         exp_word = var + exp_off;
1894                                         if (exp_null) ++exp_word;
1895                                         exp_op = *exp_word++;
1896                                         var[exp_off] = '\0';
1897                                 }
1898                         }
1899
1900                         /* lookup the variable in question */
1901                         if (isdigit(var[0])) {
1902                                 /* handle_dollar() should have vetted var for us */
1903                                 i = xatoi_u(var);
1904                                 if (i < G.global_argc)
1905                                         val = G.global_argv[i];
1906                                 /* else val remains NULL: $N with too big N */
1907                         } else
1908                                 val = lookup_param(var);
1909
1910                         /* handle any expansions */
1911                         if (exp_len) {
1912                                 debug_printf_expand("expand: length of '%s' = ", val);
1913                                 val = utoa(val ? strlen(val) : 0);
1914                                 debug_printf_expand("%s\n", val);
1915                         } else if (exp_off) {
1916                                 /* we need to do an expansion */
1917                                 int exp_test = (!val || (exp_null && !val[0]));
1918                                 if (exp_op == '+')
1919                                         exp_test = !exp_test;
1920                                 debug_printf_expand("expand: op:%c (null:%s) test:%i\n", exp_op,
1921                                         exp_null ? "true" : "false", exp_test);
1922                                 if (exp_test) {
1923                                         if (exp_op == '?')
1924                                                 maybe_die(var, *exp_word ? exp_word : "parameter null or not set");
1925                                         else
1926                                                 val = exp_word;
1927
1928                                         if (exp_op == '=') {
1929                                                 if (isdigit(var[0]) || var[0] == '#') {
1930                                                         maybe_die(var, "special vars cannot assign in this way");
1931                                                         val = NULL;
1932                                                 } else {
1933                                                         char *new_var = xmalloc(strlen(var) + strlen(val) + 2);
1934                                                         sprintf(new_var, "%s=%s", var, val);
1935                                                         set_local_var(new_var, -1);
1936                                                 }
1937                                         }
1938                                 }
1939                                 var[exp_off] = exp_save;
1940                         }
1941
1942                         arg[0] = first_ch;
1943
1944 #if ENABLE_HUSH_TICK
1945  store_val:
1946 #endif
1947                         if (!(first_ch & 0x80)) { /* unquoted $VAR */
1948                                 debug_printf_expand("unquoted '%s', output->o_escape:%d\n", val, output->o_escape);
1949                                 if (val) {
1950                                         /* unquoted var's contents should be globbed, so don't escape */
1951                                         smallint sv = output->o_escape;
1952                                         output->o_escape = 0;
1953                                         n = expand_on_ifs(output, n, val);
1954                                         val = NULL;
1955                                         output->o_escape = sv;
1956                                 }
1957                         } else { /* quoted $VAR, val will be appended below */
1958                                 debug_printf_expand("quoted '%s', output->o_escape:%d\n", val, output->o_escape);
1959                         }
1960                 } /* default: */
1961                 } /* switch (char after <SPECIAL_VAR_SYMBOL>) */
1962
1963                 if (val) {
1964                         o_addQstr(output, val, strlen(val));
1965                 }
1966                 /* Do the check to avoid writing to a const string */
1967                 if (*p != SPECIAL_VAR_SYMBOL)
1968                         *p = SPECIAL_VAR_SYMBOL;
1969
1970 #if ENABLE_HUSH_TICK
1971                 o_free(&subst_result);
1972 #endif
1973                 arg = ++p;
1974         } /* end of "while (SPECIAL_VAR_SYMBOL is found) ..." */
1975
1976         if (arg[0]) {
1977                 debug_print_list("expand_vars_to_list[a]", output, n);
1978                 /* this part is literal, and it was already pre-quoted
1979                  * if needed (much earlier), do not use o_addQstr here! */
1980                 o_addstrauto(output, arg);
1981                 debug_print_list("expand_vars_to_list[b]", output, n);
1982         } else if (output->length == o_get_last_ptr(output, n) /* expansion is empty */
1983          && !(ored_ch & 0x80) /* and all vars were not quoted. */
1984         ) {
1985                 n--;
1986                 /* allow to reuse list[n] later without re-growth */
1987                 output->has_empty_slot = 1;
1988         } else {
1989                 o_addchr(output, '\0');
1990         }
1991         return n;
1992 }
1993
1994 static char **expand_variables(char **argv, int or_mask)
1995 {
1996         int n;
1997         char **list;
1998         char **v;
1999         o_string output = NULL_O_STRING;
2000
2001         if (or_mask & 0x100) {
2002                 output.o_escape = 1; /* protect against globbing for "$var" */
2003                 /* (unquoted $var will temporarily switch it off) */
2004                 output.o_glob = 1;
2005         }
2006
2007         n = 0;
2008         v = argv;
2009         while (*v) {
2010                 n = expand_vars_to_list(&output, n, *v, (char)or_mask);
2011                 v++;
2012         }
2013         debug_print_list("expand_variables", &output, n);
2014
2015         /* output.data (malloced in one block) gets returned in "list" */
2016         list = o_finalize_list(&output, n);
2017         debug_print_strings("expand_variables[1]", list);
2018         return list;
2019 }
2020
2021 static char **expand_strvec_to_strvec(char **argv)
2022 {
2023         return expand_variables(argv, 0x100);
2024 }
2025
2026 /* Used for expansion of right hand of assignments */
2027 /* NB: should NOT do globbing! "export v=/bin/c*; env | grep ^v=" outputs
2028  * "v=/bin/c*" */
2029 static char *expand_string_to_string(const char *str)
2030 {
2031         char *argv[2], **list;
2032
2033         argv[0] = (char*)str;
2034         argv[1] = NULL;
2035         list = expand_variables(argv, 0x80); /* 0x80: make one-element expansion */
2036         if (HUSH_DEBUG)
2037                 if (!list[0] || list[1])
2038                         bb_error_msg_and_die("BUG in varexp2");
2039         /* actually, just move string 2*sizeof(char*) bytes back */
2040         overlapping_strcpy((char*)list, list[0]);
2041         debug_printf_expand("string_to_string='%s'\n", (char*)list);
2042         return (char*)list;
2043 }
2044
2045 /* Used for "eval" builtin */
2046 static char* expand_strvec_to_string(char **argv)
2047 {
2048         char **list;
2049
2050         list = expand_variables(argv, 0x80);
2051         /* Convert all NULs to spaces */
2052         if (list[0]) {
2053                 int n = 1;
2054                 while (list[n]) {
2055                         if (HUSH_DEBUG)
2056                                 if (list[n-1] + strlen(list[n-1]) + 1 != list[n])
2057                                         bb_error_msg_and_die("BUG in varexp3");
2058                         list[n][-1] = ' '; /* TODO: or to G.ifs[0]? */
2059                         n++;
2060                 }
2061         }
2062         overlapping_strcpy((char*)list, list[0]);
2063         debug_printf_expand("strvec_to_string='%s'\n", (char*)list);
2064         return (char*)list;
2065 }
2066
2067 static char **expand_assignments(char **argv, int count)
2068 {
2069         int i;
2070         char **p = NULL;
2071         /* Expand assignments into one string each */
2072         for (i = 0; i < count; i++) {
2073                 p = add_string_to_strings(p, expand_string_to_string(argv[i]));
2074         }
2075         return p;
2076 }
2077
2078
2079 /* squirrel != NULL means we squirrel away copies of stdin, stdout,
2080  * and stderr if they are redirected. */
2081 static int setup_redirects(struct command *prog, int squirrel[])
2082 {
2083         int openfd, mode;
2084         struct redir_struct *redir;
2085
2086         for (redir = prog->redirects; redir; redir = redir->next) {
2087                 if (redir->dup == -1 && redir->rd_filename == NULL) {
2088                         /* something went wrong in the parse.  Pretend it didn't happen */
2089                         continue;
2090                 }
2091                 if (redir->dup == -1) {
2092                         char *p;
2093                         mode = redir_table[redir->rd_type].mode;
2094 //TODO: check redir for names like '\\'
2095                         p = expand_string_to_string(redir->rd_filename);
2096                         openfd = open_or_warn(p, mode);
2097                         free(p);
2098                         if (openfd < 0) {
2099                         /* this could get lost if stderr has been redirected, but
2100                            bash and ash both lose it as well (though zsh doesn't!) */
2101                                 return 1;
2102                         }
2103                 } else {
2104                         openfd = redir->dup;
2105                 }
2106
2107                 if (openfd != redir->fd) {
2108                         if (squirrel && redir->fd < 3) {
2109                                 squirrel[redir->fd] = dup(redir->fd);
2110                         }
2111                         if (openfd == -3) {
2112                                 //close(openfd); // close(-3) ??!
2113                         } else {
2114                                 dup2(openfd, redir->fd);
2115                                 if (redir->dup == -1)
2116                                         close(openfd);
2117                         }
2118                 }
2119         }
2120         return 0;
2121 }
2122
2123 static void restore_redirects(int squirrel[])
2124 {
2125         int i, fd;
2126         for (i = 0; i < 3; i++) {
2127                 fd = squirrel[i];
2128                 if (fd != -1) {
2129                         /* We simply die on error */
2130                         xmove_fd(fd, i);
2131                 }
2132         }
2133 }
2134
2135
2136 #if !defined(DEBUG_CLEAN)
2137 #define free_pipe_list(head, indent) free_pipe_list(head)
2138 #define free_pipe(pi, indent)        free_pipe(pi)
2139 #endif
2140 static int free_pipe_list(struct pipe *head, int indent);
2141
2142 /* return code is the exit status of the pipe */
2143 static int free_pipe(struct pipe *pi, int indent)
2144 {
2145         char **p;
2146         struct command *command;
2147         struct redir_struct *r, *rnext;
2148         int a, i, ret_code = 0;
2149
2150         if (pi->stopped_cmds > 0)
2151                 return ret_code;
2152         debug_printf_clean("%s run pipe: (pid %d)\n", indenter(indent), getpid());
2153         for (i = 0; i < pi->num_cmds; i++) {
2154                 command = &pi->cmds[i];
2155                 debug_printf_clean("%s  command %d:\n", indenter(indent), i);
2156                 if (command->argv) {
2157                         for (a = 0, p = command->argv; *p; a++, p++) {
2158                                 debug_printf_clean("%s   argv[%d] = %s\n", indenter(indent), a, *p);
2159                         }
2160                         free_strings(command->argv);
2161                         command->argv = NULL;
2162                 } else if (command->group) {
2163                         debug_printf_clean("%s   begin group (grp_type:%d)\n", indenter(indent), command->grp_type);
2164                         ret_code = free_pipe_list(command->group, indent+3);
2165                         debug_printf_clean("%s   end group\n", indenter(indent));
2166                 } else {
2167                         debug_printf_clean("%s   (nil)\n", indenter(indent));
2168                 }
2169                 for (r = command->redirects; r; r = rnext) {
2170                         debug_printf_clean("%s   redirect %d%s", indenter(indent), r->fd, redir_table[r->rd_type].descrip);
2171                         if (r->dup == -1) {
2172                                 /* guard against the case >$FOO, where foo is unset or blank */
2173                                 if (r->rd_filename) {
2174                                         debug_printf_clean(" %s\n", r->rd_filename);
2175                                         free(r->rd_filename);
2176                                         r->rd_filename = NULL;
2177                                 }
2178                         } else {
2179                                 debug_printf_clean("&%d\n", r->dup);
2180                         }
2181                         rnext = r->next;
2182                         free(r);
2183                 }
2184                 command->redirects = NULL;
2185         }
2186         free(pi->cmds);   /* children are an array, they get freed all at once */
2187         pi->cmds = NULL;
2188 #if ENABLE_HUSH_JOB
2189         free(pi->cmdtext);
2190         pi->cmdtext = NULL;
2191 #endif
2192         return ret_code;
2193 }
2194
2195 static int free_pipe_list(struct pipe *head, int indent)
2196 {
2197         int rcode = 0;   /* if list has no members */
2198         struct pipe *pi, *next;
2199
2200         for (pi = head; pi; pi = next) {
2201 #if HAS_KEYWORDS
2202                 debug_printf_clean("%s pipe reserved mode %d\n", indenter(indent), pi->res_word);
2203 #endif
2204                 rcode = free_pipe(pi, indent);
2205                 debug_printf_clean("%s pipe followup code %d\n", indenter(indent), pi->followup);
2206                 next = pi->next;
2207                 /*pi->next = NULL;*/
2208                 free(pi);
2209         }
2210         return rcode;
2211 }
2212
2213
2214 #if !BB_MMU
2215 typedef struct nommu_save_t {
2216         char **new_env;
2217         char **old_env;
2218         char **argv;
2219 } nommu_save_t;
2220 #else
2221 #define pseudo_exec_argv(nommu_save, argv, assignment_cnt, argv_expanded) \
2222         pseudo_exec_argv(argv, assignment_cnt, argv_expanded)
2223 #define pseudo_exec(nommu_save, command, argv_expanded) \
2224         pseudo_exec(command, argv_expanded)
2225 #endif
2226
2227 /* Called after [v]fork() in run_pipe(), or from builtin_exec().
2228  * Never returns.
2229  * XXX no exit() here.  If you don't exec, use _exit instead.
2230  * The at_exit handlers apparently confuse the calling process,
2231  * in particular stdin handling.  Not sure why? -- because of vfork! (vda) */
2232 static void pseudo_exec_argv(nommu_save_t *nommu_save, char **argv, int assignment_cnt, char **argv_expanded) NORETURN;
2233 static void pseudo_exec_argv(nommu_save_t *nommu_save, char **argv, int assignment_cnt, char **argv_expanded)
2234 {
2235         int rcode;
2236         char **new_env;
2237         const struct built_in_command *x;
2238
2239         /* If a variable is assigned in a forest, and nobody listens,
2240          * was it ever really set?
2241          */
2242         if (!argv[assignment_cnt])
2243                 _exit(EXIT_SUCCESS);
2244
2245         new_env = expand_assignments(argv, assignment_cnt);
2246 #if BB_MMU
2247         putenv_all(new_env);
2248         free(new_env); /* optional */
2249 #else
2250         nommu_save->new_env = new_env;
2251         nommu_save->old_env = putenv_all_and_save_old(new_env);
2252 #endif
2253         if (argv_expanded) {
2254                 argv = argv_expanded;
2255         } else {
2256                 argv = expand_strvec_to_strvec(argv);
2257 #if !BB_MMU
2258                 nommu_save->argv = argv;
2259 #endif
2260         }
2261
2262         /*
2263          * Check if the command matches any of the builtins.
2264          * Depending on context, this might be redundant.  But it's
2265          * easier to waste a few CPU cycles than it is to figure out
2266          * if this is one of those cases.
2267          */
2268         for (x = bltins; x != &bltins[ARRAY_SIZE(bltins)]; x++) {
2269                 if (strcmp(argv[0], x->cmd) == 0) {
2270                         debug_printf_exec("running builtin '%s'\n", argv[0]);
2271                         rcode = x->function(argv);
2272                         fflush(stdout);
2273                         _exit(rcode);
2274                 }
2275         }
2276
2277         /* Check if the command matches any busybox applets */
2278 #if ENABLE_FEATURE_SH_STANDALONE
2279         if (strchr(argv[0], '/') == NULL) {
2280                 int a = find_applet_by_name(argv[0]);
2281                 if (a >= 0) {
2282                         if (APPLET_IS_NOEXEC(a)) {
2283                                 debug_printf_exec("running applet '%s'\n", argv[0]);
2284 // is it ok that run_applet_no_and_exit() does exit(), not _exit()?
2285                                 run_applet_no_and_exit(a, argv);
2286                         }
2287                         /* re-exec ourselves with the new arguments */
2288                         debug_printf_exec("re-execing applet '%s'\n", argv[0]);
2289                         execvp(bb_busybox_exec_path, argv);
2290                         /* If they called chroot or otherwise made the binary no longer
2291                          * executable, fall through */
2292                 }
2293         }
2294 #endif
2295
2296         sigprocmask(SIG_SETMASK, &G.inherited_set, NULL);
2297
2298         debug_printf_exec("execing '%s'\n", argv[0]);
2299         execvp(argv[0], argv);
2300         bb_perror_msg("can't exec '%s'", argv[0]);
2301         _exit(EXIT_FAILURE);
2302 }
2303
2304 static int run_list(struct pipe *pi);
2305
2306 /* Called after [v]fork() in run_pipe()
2307  */
2308 static void pseudo_exec(nommu_save_t *nommu_save, struct command *command, char **argv_expanded) NORETURN;
2309 static void pseudo_exec(nommu_save_t *nommu_save, struct command *command, char **argv_expanded)
2310 {
2311         if (command->argv)
2312                 pseudo_exec_argv(nommu_save, command->argv, command->assignment_cnt, argv_expanded);
2313
2314         if (command->group) {
2315 #if !BB_MMU
2316                 bb_error_msg_and_die("nested lists are not supported on NOMMU");
2317 #else
2318                 int rcode;
2319                 debug_printf_exec("pseudo_exec: run_list\n");
2320                 rcode = run_list(command->group);
2321                 /* OK to leak memory by not calling free_pipe_list,
2322                  * since this process is about to exit */
2323                 _exit(rcode);
2324 #endif
2325         }
2326
2327         /* Can happen.  See what bash does with ">foo" by itself. */
2328         debug_printf("trying to pseudo_exec null command\n");
2329         _exit(EXIT_SUCCESS);
2330 }
2331
2332 #if ENABLE_HUSH_JOB
2333 static const char *get_cmdtext(struct pipe *pi)
2334 {
2335         char **argv;
2336         char *p;
2337         int len;
2338
2339         /* This is subtle. ->cmdtext is created only on first backgrounding.
2340          * (Think "cat, <ctrl-z>, fg, <ctrl-z>, fg, <ctrl-z>...." here...)
2341          * On subsequent bg argv is trashed, but we won't use it */
2342         if (pi->cmdtext)
2343                 return pi->cmdtext;
2344         argv = pi->cmds[0].argv;
2345         if (!argv || !argv[0]) {
2346                 pi->cmdtext = xzalloc(1);
2347                 return pi->cmdtext;
2348         }
2349
2350         len = 0;
2351         do len += strlen(*argv) + 1; while (*++argv);
2352         pi->cmdtext = p = xmalloc(len);
2353         argv = pi->cmds[0].argv;
2354         do {
2355                 len = strlen(*argv);
2356                 memcpy(p, *argv, len);
2357                 p += len;
2358                 *p++ = ' ';
2359         } while (*++argv);
2360         p[-1] = '\0';
2361         return pi->cmdtext;
2362 }
2363
2364 static void insert_bg_job(struct pipe *pi)
2365 {
2366         struct pipe *thejob;
2367         int i;
2368
2369         /* Linear search for the ID of the job to use */
2370         pi->jobid = 1;
2371         for (thejob = G.job_list; thejob; thejob = thejob->next)
2372                 if (thejob->jobid >= pi->jobid)
2373                         pi->jobid = thejob->jobid + 1;
2374
2375         /* Add thejob to the list of running jobs */
2376         if (!G.job_list) {
2377                 thejob = G.job_list = xmalloc(sizeof(*thejob));
2378         } else {
2379                 for (thejob = G.job_list; thejob->next; thejob = thejob->next)
2380                         continue;
2381                 thejob->next = xmalloc(sizeof(*thejob));
2382                 thejob = thejob->next;
2383         }
2384
2385         /* Physically copy the struct job */
2386         memcpy(thejob, pi, sizeof(struct pipe));
2387         thejob->cmds = xzalloc(sizeof(pi->cmds[0]) * pi->num_cmds);
2388         /* We cannot copy entire pi->cmds[] vector! Double free()s will happen */
2389         for (i = 0; i < pi->num_cmds; i++) {
2390 // TODO: do we really need to have so many fields which are just dead weight
2391 // at execution stage?
2392                 thejob->cmds[i].pid = pi->cmds[i].pid;
2393                 /* all other fields are not used and stay zero */
2394         }
2395         thejob->next = NULL;
2396         thejob->cmdtext = xstrdup(get_cmdtext(pi));
2397
2398         /* We don't wait for background thejobs to return -- append it
2399            to the list of backgrounded thejobs and leave it alone */
2400         if (G.interactive_fd)
2401                 printf("[%d] %d %s\n", thejob->jobid, thejob->cmds[0].pid, thejob->cmdtext);
2402         G.last_bg_pid = thejob->cmds[0].pid;
2403         G.last_jobid = thejob->jobid;
2404 }
2405
2406 static void remove_bg_job(struct pipe *pi)
2407 {
2408         struct pipe *prev_pipe;
2409
2410         if (pi == G.job_list) {
2411                 G.job_list = pi->next;
2412         } else {
2413                 prev_pipe = G.job_list;
2414                 while (prev_pipe->next != pi)
2415                         prev_pipe = prev_pipe->next;
2416                 prev_pipe->next = pi->next;
2417         }
2418         if (G.job_list)
2419                 G.last_jobid = G.job_list->jobid;
2420         else
2421                 G.last_jobid = 0;
2422 }
2423
2424 /* Remove a backgrounded job */
2425 static void delete_finished_bg_job(struct pipe *pi)
2426 {
2427         remove_bg_job(pi);
2428         pi->stopped_cmds = 0;
2429         free_pipe(pi, 0);
2430         free(pi);
2431 }
2432 #endif /* JOB */
2433
2434 /* Check to see if any processes have exited -- if they
2435  * have, figure out why and see if a job has completed */
2436 static int checkjobs(struct pipe* fg_pipe)
2437 {
2438         int attributes;
2439         int status;
2440 #if ENABLE_HUSH_JOB
2441         struct pipe *pi;
2442 #endif
2443         pid_t childpid;
2444         int rcode = 0;
2445
2446         debug_printf_jobs("checkjobs %p\n", fg_pipe);
2447
2448         errno = 0;
2449 //      if (G.handled_SIGCHLD == G.count_SIGCHLD)
2450 //              /* avoid doing syscall, nothing there anyway */
2451 //              return rcode;
2452
2453         attributes = WUNTRACED;
2454         if (fg_pipe == NULL)
2455                 attributes |= WNOHANG;
2456
2457 /* Do we do this right?
2458  * bash-3.00# sleep 20 | false
2459  * <ctrl-Z pressed>
2460  * [3]+  Stopped          sleep 20 | false
2461  * bash-3.00# echo $?
2462  * 1   <========== bg pipe is not fully done, but exitcode is already known!
2463  */
2464
2465 //FIXME: non-interactive bash does not continue even if all processes in fg pipe
2466 //are stopped. Testcase: "cat | cat" in a script (not on command line)
2467 // + killall -STOP cat
2468
2469  wait_more:
2470         while (1) {
2471                 int i;
2472                 int dead;
2473
2474 //              i = G.count_SIGCHLD;
2475                 childpid = waitpid(-1, &status, attributes);
2476                 if (childpid <= 0) {
2477                         if (childpid && errno != ECHILD)
2478                                 bb_perror_msg("waitpid");
2479 //                      else /* Until next SIGCHLD, waitpid's are useless */
2480 //                              G.handled_SIGCHLD = i;
2481                         break;
2482                 }
2483                 dead = WIFEXITED(status) || WIFSIGNALED(status);
2484
2485 #if DEBUG_JOBS
2486                 if (WIFSTOPPED(status))
2487                         debug_printf_jobs("pid %d stopped by sig %d (exitcode %d)\n",
2488                                         childpid, WSTOPSIG(status), WEXITSTATUS(status));
2489                 if (WIFSIGNALED(status))
2490                         debug_printf_jobs("pid %d killed by sig %d (exitcode %d)\n",
2491                                         childpid, WTERMSIG(status), WEXITSTATUS(status));
2492                 if (WIFEXITED(status))
2493                         debug_printf_jobs("pid %d exited, exitcode %d\n",
2494                                         childpid, WEXITSTATUS(status));
2495 #endif
2496                 /* Were we asked to wait for fg pipe? */
2497                 if (fg_pipe) {
2498                         for (i = 0; i < fg_pipe->num_cmds; i++) {
2499                                 debug_printf_jobs("check pid %d\n", fg_pipe->cmds[i].pid);
2500                                 if (fg_pipe->cmds[i].pid != childpid)
2501                                         continue;
2502                                 /* printf("process %d exit %d\n", i, WEXITSTATUS(status)); */
2503                                 if (dead) {
2504                                         fg_pipe->cmds[i].pid = 0;
2505                                         fg_pipe->alive_cmds--;
2506                                         if (i == fg_pipe->num_cmds - 1) {
2507                                                 /* last process gives overall exitstatus */
2508                                                 rcode = WEXITSTATUS(status);
2509                                                 IF_HAS_KEYWORDS(if (fg_pipe->pi_inverted) rcode = !rcode;)
2510                                         }
2511                                 } else {
2512                                         fg_pipe->cmds[i].is_stopped = 1;
2513                                         fg_pipe->stopped_cmds++;
2514                                 }
2515                                 debug_printf_jobs("fg_pipe: alive_cmds %d stopped_cmds %d\n",
2516                                                 fg_pipe->alive_cmds, fg_pipe->stopped_cmds);
2517                                 if (fg_pipe->alive_cmds - fg_pipe->stopped_cmds <= 0) {
2518                                         /* All processes in fg pipe have exited/stopped */
2519 #if ENABLE_HUSH_JOB
2520                                         if (fg_pipe->alive_cmds)
2521                                                 insert_bg_job(fg_pipe);
2522 #endif
2523                                         return rcode;
2524                                 }
2525                                 /* There are still running processes in the fg pipe */
2526                                 goto wait_more; /* do waitpid again */
2527                         }
2528                         /* it wasnt fg_pipe, look for process in bg pipes */
2529                 }
2530
2531 #if ENABLE_HUSH_JOB
2532                 /* We asked to wait for bg or orphaned children */
2533                 /* No need to remember exitcode in this case */
2534                 for (pi = G.job_list; pi; pi = pi->next) {
2535                         for (i = 0; i < pi->num_cmds; i++) {
2536                                 if (pi->cmds[i].pid == childpid)
2537                                         goto found_pi_and_prognum;
2538                         }
2539                 }
2540                 /* Happens when shell is used as init process (init=/bin/sh) */
2541                 debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
2542                 continue; /* do waitpid again */
2543
2544  found_pi_and_prognum:
2545                 if (dead) {
2546                         /* child exited */
2547                         pi->cmds[i].pid = 0;
2548                         pi->alive_cmds--;
2549                         if (!pi->alive_cmds) {
2550                                 if (G.interactive_fd)
2551                                         printf(JOB_STATUS_FORMAT, pi->jobid,
2552                                                         "Done", pi->cmdtext);
2553                                 delete_finished_bg_job(pi);
2554                         }
2555                 } else {
2556                         /* child stopped */
2557                         pi->cmds[i].is_stopped = 1;
2558                         pi->stopped_cmds++;
2559                 }
2560 #endif
2561         } /* while (waitpid succeeds)... */
2562
2563         return rcode;
2564 }
2565
2566 #if ENABLE_HUSH_JOB
2567 static int checkjobs_and_fg_shell(struct pipe* fg_pipe)
2568 {
2569         pid_t p;
2570         int rcode = checkjobs(fg_pipe);
2571         /* Job finished, move the shell to the foreground */
2572         p = getpgid(0); /* pgid of our process */
2573         debug_printf_jobs("fg'ing ourself: getpgid(0)=%d\n", (int)p);
2574         tcsetpgrp(G.interactive_fd, p);
2575         return rcode;
2576 }
2577 #endif
2578
2579 /* run_pipe() starts all the jobs, but doesn't wait for anything
2580  * to finish.  See checkjobs().
2581  *
2582  * return code is normally -1, when the caller has to wait for children
2583  * to finish to determine the exit status of the pipe.  If the pipe
2584  * is a simple builtin command, however, the action is done by the
2585  * time run_pipe returns, and the exit code is provided as the
2586  * return value.
2587  *
2588  * The input of the pipe is always stdin, the output is always
2589  * stdout.  The outpipe[] mechanism in BusyBox-0.48 lash is bogus,
2590  * because it tries to avoid running the command substitution in
2591  * subshell, when that is in fact necessary.  The subshell process
2592  * now has its stdout directed to the input of the appropriate pipe,
2593  * so this routine is noticeably simpler.
2594  *
2595  * Returns -1 only if started some children. IOW: we have to
2596  * mask out retvals of builtins etc with 0xff!
2597  */
2598 static int run_pipe(struct pipe *pi)
2599 {
2600         int i;
2601         int nextin;
2602         int pipefds[2];         /* pipefds[0] is for reading */
2603         struct command *command;
2604         char **argv_expanded;
2605         char **argv;
2606         const struct built_in_command *x;
2607         char *p;
2608         /* it is not always needed, but we aim to smaller code */
2609         int squirrel[] = { -1, -1, -1 };
2610         int rcode;
2611         const int single_and_fg = (pi->num_cmds == 1 && pi->followup != PIPE_BG);
2612
2613         debug_printf_exec("run_pipe start: single_and_fg=%d\n", single_and_fg);
2614
2615 #if ENABLE_HUSH_JOB
2616         pi->pgrp = -1;
2617 #endif
2618         pi->alive_cmds = 1;
2619         pi->stopped_cmds = 0;
2620
2621         /* Check if this is a simple builtin (not part of a pipe).
2622          * Builtins within pipes have to fork anyway, and are handled in
2623          * pseudo_exec.  "echo foo | read bar" doesn't work on bash, either.
2624          */
2625         command = &(pi->cmds[0]);
2626
2627 #if ENABLE_HUSH_FUNCTIONS
2628         if (single_and_fg && command->group && command->grp_type == GRP_FUNCTION) {
2629                 /* We "execute" function definition */
2630                 bb_error_msg("here we ought to remember function definition, and go on");
2631                 return EXIT_SUCCESS;
2632         }
2633 #endif
2634
2635         if (single_and_fg && command->group && command->grp_type == GRP_NORMAL) {
2636                 debug_printf("non-subshell grouping\n");
2637                 setup_redirects(command, squirrel);
2638                 debug_printf_exec(": run_list\n");
2639                 rcode = run_list(command->group) & 0xff;
2640                 restore_redirects(squirrel);
2641                 debug_printf_exec("run_pipe return %d\n", rcode);
2642                 IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
2643                 return rcode;
2644         }
2645
2646         argv = command->argv;
2647         argv_expanded = NULL;
2648
2649         if (single_and_fg && argv != NULL) {
2650                 char **new_env = NULL;
2651                 char **old_env = NULL;
2652
2653                 i = command->assignment_cnt;
2654                 if (i != 0 && argv[i] == NULL) {
2655                         /* assignments, but no command: set local environment */
2656                         for (i = 0; argv[i] != NULL; i++) {
2657                                 debug_printf("local environment set: %s\n", argv[i]);
2658                                 p = expand_string_to_string(argv[i]);
2659                                 set_local_var(p, 0);
2660                         }
2661                         return EXIT_SUCCESS; /* don't worry about errors in set_local_var() yet */
2662                 }
2663
2664                 /* Expand the rest into (possibly) many strings each */
2665                 argv_expanded = expand_strvec_to_strvec(argv + i);
2666
2667                 for (x = bltins; x != &bltins[ARRAY_SIZE(bltins)]; x++) {
2668                         if (strcmp(argv_expanded[0], x->cmd) != 0)
2669                                 continue;
2670                         if (x->function == builtin_exec && argv_expanded[1] == NULL) {
2671                                 debug_printf("exec with redirects only\n");
2672                                 setup_redirects(command, NULL);
2673                                 rcode = EXIT_SUCCESS;
2674                                 goto clean_up_and_ret1;
2675                         }
2676                         debug_printf("builtin inline %s\n", argv_expanded[0]);
2677                         /* XXX setup_redirects acts on file descriptors, not FILEs.
2678                          * This is perfect for work that comes after exec().
2679                          * Is it really safe for inline use?  Experimentally,
2680                          * things seem to work with glibc. */
2681                         setup_redirects(command, squirrel);
2682                         new_env = expand_assignments(argv, command->assignment_cnt);
2683                         old_env = putenv_all_and_save_old(new_env);
2684                         debug_printf_exec(": builtin '%s' '%s'...\n", x->cmd, argv_expanded[1]);
2685                         rcode = x->function(argv_expanded) & 0xff;
2686 #if ENABLE_FEATURE_SH_STANDALONE
2687  clean_up_and_ret:
2688 #endif
2689                         restore_redirects(squirrel);
2690                         free_strings_and_unsetenv(new_env, 1);
2691                         putenv_all(old_env);
2692                         free(old_env); /* not free_strings()! */
2693  clean_up_and_ret1:
2694                         free(argv_expanded);
2695                         IF_HAS_KEYWORDS(if (pi->pi_inverted) rcode = !rcode;)
2696                         debug_printf_exec("run_pipe return %d\n", rcode);
2697                         return rcode;
2698                 }
2699 #if ENABLE_FEATURE_SH_STANDALONE
2700                 i = find_applet_by_name(argv_expanded[0]);
2701                 if (i >= 0 && APPLET_IS_NOFORK(i)) {
2702                         setup_redirects(command, squirrel);
2703                         save_nofork_data(&G.nofork_save);
2704                         new_env = expand_assignments(argv, command->assignment_cnt);
2705                         old_env = putenv_all_and_save_old(new_env);
2706                         debug_printf_exec(": run_nofork_applet '%s' '%s'...\n", argv_expanded[0], argv_expanded[1]);
2707                         rcode = run_nofork_applet_prime(&G.nofork_save, i, argv_expanded);
2708                         goto clean_up_and_ret;
2709                 }
2710 #endif
2711         }
2712
2713         /* NB: argv_expanded may already be created, and that
2714          * might include `cmd` runs! Do not rerun it! We *must*
2715          * use argv_expanded if it's non-NULL */
2716
2717         /* Going to fork a child per each pipe member */
2718         pi->alive_cmds = 0;
2719         nextin = 0;
2720
2721         for (i = 0; i < pi->num_cmds; i++) {
2722 #if !BB_MMU
2723                 volatile nommu_save_t nommu_save;
2724                 nommu_save.new_env = NULL;
2725                 nommu_save.old_env = NULL;
2726                 nommu_save.argv = NULL;
2727 #endif
2728                 command = &(pi->cmds[i]);
2729                 if (command->argv) {
2730                         debug_printf_exec(": pipe member '%s' '%s'...\n", command->argv[0], command->argv[1]);
2731                 } else
2732                         debug_printf_exec(": pipe member with no argv\n");
2733
2734                 /* pipes are inserted between pairs of commands */
2735                 pipefds[0] = 0;
2736                 pipefds[1] = 1;
2737                 if ((i + 1) < pi->num_cmds)
2738                         xpipe(pipefds);
2739
2740                 command->pid = BB_MMU ? fork() : vfork();
2741                 if (!command->pid) { /* child */
2742 #if ENABLE_HUSH_JOB
2743                         die_sleep = 0; /* let nofork's xfuncs die */
2744
2745                         /* Every child adds itself to new process group
2746                          * with pgid == pid_of_first_child_in_pipe */
2747                         if (G.run_list_level == 1 && G.interactive_fd) {
2748                                 pid_t pgrp;
2749                                 pgrp = pi->pgrp;
2750                                 if (pgrp < 0) /* true for 1st process only */
2751                                         pgrp = getpid();
2752                                 if (setpgid(0, pgrp) == 0 && pi->followup != PIPE_BG) {
2753                                         /* We do it in *every* child, not just first,
2754                                          * to avoid races */
2755                                         tcsetpgrp(G.interactive_fd, pgrp);
2756                                 }
2757                         }
2758 #endif
2759                         xmove_fd(nextin, 0);
2760                         xmove_fd(pipefds[1], 1); /* write end */
2761                         if (pipefds[0] > 1)
2762                                 close(pipefds[0]); /* read end */
2763                         /* Like bash, explicit redirects override pipes,
2764                          * and the pipe fd is available for dup'ing. */
2765                         setup_redirects(command, NULL);
2766
2767                         /* Restore default handlers just prior to exec */
2768                         /*signal(SIGCHLD, SIG_DFL); - so far we don't have any handlers */
2769
2770                         /* Stores to nommu_save list of env vars putenv'ed
2771                          * (NOMMU, on MMU we don't need that) */
2772                         /* cast away volatility... */
2773                         pseudo_exec((nommu_save_t*) &nommu_save, command, argv_expanded);
2774                         /* pseudo_exec() does not return */
2775                 }
2776                 /* parent */
2777 #if !BB_MMU
2778                 /* Clean up after vforked child */
2779                 free(nommu_save.argv);
2780                 free_strings_and_unsetenv(nommu_save.new_env, 1);
2781                 putenv_all(nommu_save.old_env);
2782 #endif
2783                 free(argv_expanded);
2784                 argv_expanded = NULL;
2785                 if (command->pid < 0) { /* [v]fork failed */
2786                         /* Clearly indicate, was it fork or vfork */
2787                         bb_perror_msg(BB_MMU ? "fork" : "vfork");
2788                 } else {
2789                         pi->alive_cmds++;
2790 #if ENABLE_HUSH_JOB
2791                         /* Second and next children need to know pid of first one */
2792                         if (pi->pgrp < 0)
2793                                 pi->pgrp = command->pid;
2794 #endif
2795                 }
2796
2797                 if (i)
2798                         close(nextin);
2799                 if ((i + 1) < pi->num_cmds)
2800                         close(pipefds[1]); /* write end */
2801                 /* Pass read (output) pipe end to next iteration */
2802                 nextin = pipefds[0];
2803         }
2804
2805         if (!pi->alive_cmds) {
2806                 debug_printf_exec("run_pipe return 1 (all forks failed, no children)\n");
2807                 return 1;
2808         }
2809
2810         debug_printf_exec("run_pipe return -1 (%u children started)\n", pi->alive_cmds);
2811         return -1;
2812 }
2813
2814 #ifndef debug_print_tree
2815 static void debug_print_tree(struct pipe *pi, int lvl)
2816 {
2817         static const char *const PIPE[] = {
2818                 [PIPE_SEQ] = "SEQ",
2819                 [PIPE_AND] = "AND",
2820                 [PIPE_OR ] = "OR" ,
2821                 [PIPE_BG ] = "BG" ,
2822         };
2823         static const char *RES[] = {
2824                 [RES_NONE ] = "NONE" ,
2825 #if ENABLE_HUSH_IF
2826                 [RES_IF   ] = "IF"   ,
2827                 [RES_THEN ] = "THEN" ,
2828                 [RES_ELIF ] = "ELIF" ,
2829                 [RES_ELSE ] = "ELSE" ,
2830                 [RES_FI   ] = "FI"   ,
2831 #endif
2832 #if ENABLE_HUSH_LOOPS
2833                 [RES_FOR  ] = "FOR"  ,
2834                 [RES_WHILE] = "WHILE",
2835                 [RES_UNTIL] = "UNTIL",
2836                 [RES_DO   ] = "DO"   ,
2837                 [RES_DONE ] = "DONE" ,
2838 #endif
2839 #if ENABLE_HUSH_LOOPS || ENABLE_HUSH_CASE
2840                 [RES_IN   ] = "IN"   ,
2841 #endif
2842 #if ENABLE_HUSH_CASE
2843                 [RES_CASE ] = "CASE" ,
2844                 [RES_MATCH] = "MATCH",
2845                 [RES_CASEI] = "CASEI",
2846                 [RES_ESAC ] = "ESAC" ,
2847 #endif
2848                 [RES_XXXX ] = "XXXX" ,
2849                 [RES_SNTX ] = "SNTX" ,
2850         };
2851         static const char *const GRPTYPE[] = {
2852                 "()",
2853                 "{}",
2854 #if ENABLE_HUSH_FUNCTIONS
2855                 "func()",
2856 #endif
2857         };
2858
2859         int pin, prn;
2860
2861         pin = 0;
2862         while (pi) {
2863                 fprintf(stderr, "%*spipe %d res_word=%s followup=%d %s\n", lvl*2, "",
2864                                 pin, RES[pi->res_word], pi->followup, PIPE[pi->followup]);
2865                 prn = 0;
2866                 while (prn < pi->num_cmds) {
2867                         struct command *command = &pi->cmds[prn];
2868                         char **argv = command->argv;
2869
2870                         fprintf(stderr, "%*s prog %d assignment_cnt:%d", lvl*2, "", prn, command->assignment_cnt);
2871                         if (command->group) {
2872                                 fprintf(stderr, " group %s: (argv=%p)\n",
2873                                                 GRPTYPE[command->grp_type],
2874                                                 argv);
2875                                 debug_print_tree(command->group, lvl+1);
2876                                 prn++;
2877                                 continue;
2878                         }
2879                         if (argv) while (*argv) {
2880                                 fprintf(stderr, " '%s'", *argv);
2881                                 argv++;
2882                         }
2883                         fprintf(stderr, "\n");
2884                         prn++;
2885                 }
2886                 pi = pi->next;
2887                 pin++;
2888         }
2889 }
2890 #endif
2891
2892 /* NB: called by pseudo_exec, and therefore must not modify any
2893  * global data until exec/_exit (we can be a child after vfork!) */
2894 static int run_list(struct pipe *pi)
2895 {
2896 #if ENABLE_HUSH_CASE
2897         char *case_word = NULL;
2898 #endif
2899 #if ENABLE_HUSH_LOOPS
2900         struct pipe *loop_top = NULL;
2901         char *for_varname = NULL;
2902         char **for_lcur = NULL;
2903         char **for_list = NULL;
2904 #endif
2905         smallint flag_skip = 1;
2906         smalluint rcode = 0; /* probably just for compiler */
2907 #if ENABLE_HUSH_IF || ENABLE_HUSH_CASE
2908         smalluint cond_code = 0;
2909 #else
2910         enum { cond_code = 0, };
2911 #endif
2912         /*enum reserved_style*/ smallint rword = RES_NONE;
2913         /*enum reserved_style*/ smallint skip_more_for_this_rword = RES_XXXX;
2914
2915         debug_printf_exec("run_list start lvl %d\n", G.run_list_level + 1);
2916
2917 #if ENABLE_HUSH_LOOPS
2918         /* Check syntax for "for" */
2919         for (struct pipe *cpipe = pi; cpipe; cpipe = cpipe->next) {
2920                 if (cpipe->res_word != RES_FOR && cpipe->res_word != RES_IN)
2921                         continue;
2922                 /* current word is FOR or IN (BOLD in comments below) */
2923                 if (cpipe->next == NULL) {
2924                         syntax("malformed for");
2925                         debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
2926                         return 1;
2927                 }
2928                 /* "FOR v; do ..." and "for v IN a b; do..." are ok */
2929                 if (cpipe->next->res_word == RES_DO)
2930                         continue;
2931                 /* next word is not "do". It must be "in" then ("FOR v in ...") */
2932                 if (cpipe->res_word == RES_IN /* "for v IN a b; not_do..."? */
2933                  || cpipe->next->res_word != RES_IN /* FOR v not_do_and_not_in..."? */
2934                 ) {
2935                         syntax("malformed for");
2936                         debug_printf_exec("run_list lvl %d return 1\n", G.run_list_level);
2937                         return 1;
2938                 }
2939         }
2940 #endif
2941
2942         /* Past this point, all code paths should jump to ret: label
2943          * in order to return, no direct "return" statements please.
2944          * This helps to ensure that no memory is leaked. */
2945
2946 ////TODO: ctrl-Z handling needs re-thinking and re-testing
2947
2948 #if ENABLE_HUSH_JOB
2949         /* Example of nested list: "while true; do { sleep 1 | exit 2; } done".
2950          * We are saving state before entering outermost list ("while...done")
2951          * so that ctrl-Z will correctly background _entire_ outermost list,
2952          * not just a part of it (like "sleep 1 | exit 2") */
2953         if (++G.run_list_level == 1 && G.interactive_fd) {
2954                 if (sigsetjmp(G.toplevel_jb, 1)) {
2955                         /* ctrl-Z forked and we are parent; or ctrl-C.
2956                          * Sighandler has longjmped us here */
2957                         signal(SIGINT, SIG_IGN);
2958                         signal(SIGTSTP, SIG_IGN);
2959                         /* Restore level (we can be coming from deep inside
2960                          * nested levels) */
2961                         G.run_list_level = 1;
2962 #if ENABLE_FEATURE_SH_STANDALONE
2963                         if (G.nofork_save.saved) { /* if save area is valid */
2964                                 debug_printf_jobs("exiting nofork early\n");
2965                                 restore_nofork_data(&G.nofork_save);
2966                         }
2967 #endif
2968 ////                    if (G.ctrl_z_flag) {
2969 ////                            /* ctrl-Z has forked and stored pid of the child in pi->pid.
2970 ////                             * Remember this child as background job */
2971 ////                            insert_bg_job(pi);
2972 ////                    } else {
2973                                 /* ctrl-C. We just stop doing whatever we were doing */
2974                                 bb_putchar('\n');
2975 ////                    }
2976                         USE_HUSH_LOOPS(loop_top = NULL;)
2977                         USE_HUSH_LOOPS(G.depth_of_loop = 0;)
2978                         rcode = 0;
2979                         goto ret;
2980                 }
2981 ////            /* ctrl-Z handler will store pid etc in pi */
2982 ////            G.toplevel_list = pi;
2983 ////            G.ctrl_z_flag = 0;
2984 ////#if ENABLE_FEATURE_SH_STANDALONE
2985 ////            G.nofork_save.saved = 0; /* in case we will run a nofork later */
2986 ////#endif
2987 ////            signal_SA_RESTART_empty_mask(SIGTSTP, handler_ctrl_z);
2988 ////            signal(SIGINT, handler_ctrl_c);
2989         }
2990 #endif /* JOB */
2991
2992         /* Go through list of pipes, (maybe) executing them. */
2993         for (; pi; pi = USE_HUSH_LOOPS(rword == RES_DONE ? loop_top : ) pi->next) {
2994                 if (G.flag_SIGINT)
2995                         break;
2996
2997                 IF_HAS_KEYWORDS(rword = pi->res_word;)
2998                 IF_HAS_NO_KEYWORDS(rword = RES_NONE;)
2999                 debug_printf_exec(": rword=%d cond_code=%d skip_more=%d\n",
3000                                 rword, cond_code, skip_more_for_this_rword);
3001 #if ENABLE_HUSH_LOOPS
3002                 if ((rword == RES_WHILE || rword == RES_UNTIL || rword == RES_FOR)
3003                  && loop_top == NULL /* avoid bumping G.depth_of_loop twice */
3004                 ) {
3005                         /* start of a loop: remember where loop starts */
3006                         loop_top = pi;
3007                         G.depth_of_loop++;
3008                 }
3009 #endif
3010                 if (rword == skip_more_for_this_rword && flag_skip) {
3011                         if (pi->followup == PIPE_SEQ)
3012                                 flag_skip = 0;
3013                         /* it is "<false> && CMD" or "<true> || CMD"
3014                          * and we should not execute CMD */
3015                         continue;
3016                 }
3017                 flag_skip = 1;
3018                 skip_more_for_this_rword = RES_XXXX;
3019 #if ENABLE_HUSH_IF
3020                 if (cond_code) {
3021                         if (rword == RES_THEN) {
3022                                 /* "if <false> THEN cmd": skip cmd */
3023                                 continue;
3024                         }
3025                 } else {
3026                         if (rword == RES_ELSE || rword == RES_ELIF) {
3027                                 /* "if <true> then ... ELSE/ELIF cmd":
3028                                  * skip cmd and all following ones */
3029                                 break;
3030                         }
3031                 }
3032 #endif
3033 #if ENABLE_HUSH_LOOPS
3034                 if (rword == RES_FOR) { /* && pi->num_cmds - always == 1 */
3035                         if (!for_lcur) {
3036                                 /* first loop through for */
3037
3038                                 static const char encoded_dollar_at[] ALIGN1 = {
3039                                         SPECIAL_VAR_SYMBOL, '@' | 0x80, SPECIAL_VAR_SYMBOL, '\0'
3040                                 }; /* encoded representation of "$@" */
3041                                 static const char *const encoded_dollar_at_argv[] = {
3042                                         encoded_dollar_at, NULL
3043                                 }; /* argv list with one element: "$@" */
3044                                 char **vals;
3045
3046                                 vals = (char**)encoded_dollar_at_argv;
3047                                 if (pi->next->res_word == RES_IN) {
3048                                         /* if no variable values after "in" we skip "for" */
3049                                         if (!pi->next->cmds[0].argv)
3050                                                 break;
3051                                         vals = pi->next->cmds[0].argv;
3052                                 } /* else: "for var; do..." -> assume "$@" list */
3053                                 /* create list of variable values */
3054                                 debug_print_strings("for_list made from", vals);
3055                                 for_list = expand_strvec_to_strvec(vals);
3056                                 for_lcur = for_list;
3057                                 debug_print_strings("for_list", for_list);
3058                                 for_varname = pi->cmds[0].argv[0];
3059                                 pi->cmds[0].argv[0] = NULL;
3060                         }
3061                         free(pi->cmds[0].argv[0]);
3062                         if (!*for_lcur) {
3063                                 /* "for" loop is over, clean up */
3064                                 free(for_list);
3065                                 for_list = NULL;
3066                                 for_lcur = NULL;
3067                                 pi->cmds[0].argv[0] = for_varname;
3068                                 break;
3069                         }
3070                         /* insert next value from for_lcur */
3071 //TODO: does it need escaping?
3072                         pi->cmds[0].argv[0] = xasprintf("%s=%s", for_varname, *for_lcur++);
3073                         pi->cmds[0].assignment_cnt = 1;
3074                 }
3075                 if (rword == RES_IN) /* "for v IN list;..." - "in" has no cmds anyway */
3076                         continue;
3077                 if (rword == RES_DONE) {
3078                         continue; /* "done" has no cmds too */
3079                 }
3080 #endif
3081 #if ENABLE_HUSH_CASE
3082                 if (rword == RES_CASE) {
3083                         case_word = expand_strvec_to_string(pi->cmds->argv);
3084                         continue;
3085                 }
3086                 if (rword == RES_MATCH) {
3087                         char **argv;
3088
3089                         if (!case_word) /* "case ... matched_word) ... WORD)": we executed selected branch, stop */
3090                                 break;
3091                         /* all prev words didn't match, does this one match? */
3092                         argv = pi->cmds->argv;
3093                         while (*argv) {
3094                                 char *pattern = expand_string_to_string(*argv);
3095                                 /* TODO: which FNM_xxx flags to use? */
3096                                 cond_code = (fnmatch(pattern, case_word, /*flags:*/ 0) != 0);
3097                                 free(pattern);
3098                                 if (cond_code == 0) { /* match! we will execute this branch */
3099                                         free(case_word); /* make future "word)" stop */
3100                                         case_word = NULL;
3101                                         break;
3102                                 }
3103                                 argv++;
3104                         }
3105                         continue;
3106                 }
3107                 if (rword == RES_CASEI) { /* inside of a case branch */
3108                         if (cond_code != 0)
3109                                 continue; /* not matched yet, skip this pipe */
3110                 }
3111 #endif
3112                 /* Just pressing <enter> in shell should check for jobs.
3113                  * OTOH, in non-interactive shell this is useless
3114                  * and only leads to extra job checks */
3115                 if (pi->num_cmds == 0) {
3116                         if (G.interactive_fd)
3117                                 goto check_jobs_and_continue;
3118                         continue;
3119                 }
3120
3121                 /* After analyzing all keywords and conditions, we decided
3122                  * to execute this pipe. NB: have to do checkjobs(NULL)
3123                  * after run_pipe() to collect any background children,
3124                  * even if list execution is to be stopped. */
3125                 debug_printf_exec(": run_pipe with %d members\n", pi->num_cmds);
3126                 {
3127                         int r;
3128 #if ENABLE_HUSH_LOOPS
3129                         G.flag_break_continue = 0;
3130 #endif
3131                         rcode = r = run_pipe(pi); /* NB: rcode is a smallint */
3132                         if (r != -1) {
3133                                 /* we only ran a builtin: rcode is already known
3134                                  * and we don't need to wait for anything. */
3135                                 check_and_run_traps(0);
3136 #if ENABLE_HUSH_LOOPS
3137                                 /* was it "break" or "continue"? */
3138                                 if (G.flag_break_continue) {
3139                                         smallint fbc = G.flag_break_continue;
3140                                         /* we might fall into outer *loop*,
3141                                          * don't want to break it too */
3142                                         if (loop_top) {
3143                                                 G.depth_break_continue--;
3144                                                 if (G.depth_break_continue == 0)
3145                                                         G.flag_break_continue = 0;
3146                                                 /* else: e.g. "continue 2" should *break* once, *then* continue */
3147                                         } /* else: "while... do... { we are here (innermost list is not a loop!) };...done" */
3148                                         if (G.depth_break_continue != 0 || fbc == BC_BREAK)
3149                                                 goto check_jobs_and_break;
3150                                         /* "continue": simulate end of loop */
3151                                         rword = RES_DONE;
3152                                         continue;
3153                                 }
3154 #endif
3155                         } else if (pi->followup == PIPE_BG) {
3156                                 /* what does bash do with attempts to background builtins? */
3157                                 /* even bash 3.2 doesn't do that well with nested bg:
3158                                  * try "{ { sleep 10; echo DEEP; } & echo HERE; } &".
3159                                  * I'm NOT treating inner &'s as jobs */
3160                                 check_and_run_traps(0);
3161 #if ENABLE_HUSH_JOB
3162                                 if (G.run_list_level == 1)
3163                                         insert_bg_job(pi);
3164 #endif
3165                                 rcode = 0; /* EXIT_SUCCESS */
3166                         } else {
3167 #if ENABLE_HUSH_JOB
3168                                 if (G.run_list_level == 1 && G.interactive_fd) {
3169                                         /* waits for completion, then fg's main shell */
3170                                         rcode = checkjobs_and_fg_shell(pi);
3171                                         check_and_run_traps(0);
3172                                         debug_printf_exec(": checkjobs_and_fg_shell returned %d\n", rcode);
3173                                 } else
3174 #endif
3175                                 { /* this one just waits for completion */
3176                                         rcode = checkjobs(pi);
3177                                         check_and_run_traps(0);
3178                                         debug_printf_exec(": checkjobs returned %d\n", rcode);
3179                                 }
3180                         }
3181                 }
3182                 debug_printf_exec(": setting last_return_code=%d\n", rcode);
3183                 G.last_return_code = rcode;
3184
3185                 /* Analyze how result affects subsequent commands */
3186 #if ENABLE_HUSH_IF
3187                 if (rword == RES_IF || rword == RES_ELIF)
3188                         cond_code = rcode;
3189 #endif
3190 #if ENABLE_HUSH_LOOPS
3191                 if (rword == RES_WHILE) {
3192                         if (rcode) {
3193                                 rcode = 0; /* "while false; do...done" - exitcode 0 */
3194                                 goto check_jobs_and_break;
3195                         }
3196                 }
3197                 if (rword == RES_UNTIL) {
3198                         if (!rcode) {
3199  check_jobs_and_break:
3200                                 checkjobs(NULL);
3201                                 break;
3202                         }
3203                 }
3204 #endif
3205                 if ((rcode == 0 && pi->followup == PIPE_OR)
3206                  || (rcode != 0 && pi->followup == PIPE_AND)
3207                 ) {
3208                         skip_more_for_this_rword = rword;
3209                 }
3210
3211  check_jobs_and_continue:
3212                 checkjobs(NULL);
3213         } /* for (pi) */
3214
3215 #if ENABLE_HUSH_JOB
3216 ////    if (G.ctrl_z_flag) {
3217 ////            /* ctrl-Z forked somewhere in the past, we are the child,
3218 ////             * and now we completed running the list. Exit. */
3219 //////TODO: _exit?
3220 ////            exit(rcode);
3221 ////    }
3222  ret:
3223         G.run_list_level--;
3224 ////    if (!G.run_list_level && G.interactive_fd) {
3225 ////            signal(SIGTSTP, SIG_IGN);
3226 ////            signal(SIGINT, SIG_IGN);
3227 ////    }
3228 #endif
3229         debug_printf_exec("run_list lvl %d return %d\n", G.run_list_level + 1, rcode);
3230 #if ENABLE_HUSH_LOOPS
3231         if (loop_top)
3232                 G.depth_of_loop--;
3233         free(for_list);
3234 #endif
3235 #if ENABLE_HUSH_CASE
3236         free(case_word);
3237 #endif
3238         return rcode;
3239 }
3240
3241 /* Select which version we will use */
3242 static int run_and_free_list(struct pipe *pi)
3243 {
3244         int rcode = 0;
3245         debug_printf_exec("run_and_free_list entered\n");
3246         if (!G.fake_mode) {
3247                 debug_printf_exec(": run_list with %d members\n", pi->num_cmds);
3248                 rcode = run_list(pi);
3249         }
3250         /* free_pipe_list has the side effect of clearing memory.
3251          * In the long run that function can be merged with run_list,
3252          * but doing that now would hobble the debugging effort. */
3253         free_pipe_list(pi, /* indent: */ 0);
3254         debug_printf_exec("run_and_free_list return %d\n", rcode);
3255         return rcode;
3256 }
3257
3258
3259 /* Peek ahead in the in_str to find out if we have a "&n" construct,
3260  * as in "2>&1", that represents duplicating a file descriptor.
3261  * Return either -2 (syntax error), -1 (no &), or the number found.
3262  */
3263 static int redirect_dup_num(struct in_str *input)
3264 {
3265         int ch, d = 0, ok = 0;
3266         ch = i_peek(input);
3267         if (ch != '&') return -1;
3268
3269         i_getch(input);  /* get the & */
3270         ch = i_peek(input);
3271         if (ch == '-') {
3272                 i_getch(input);
3273                 return -3;  /* "-" represents "close me" */
3274         }
3275         while (isdigit(ch)) {
3276                 d = d*10 + (ch-'0');
3277                 ok = 1;
3278                 i_getch(input);
3279                 ch = i_peek(input);
3280         }
3281         if (ok) return d;
3282
3283         bb_error_msg("ambiguous redirect");
3284         return -2;
3285 }
3286
3287 /* The src parameter allows us to peek forward to a possible &n syntax
3288  * for file descriptor duplication, e.g., "2>&1".
3289  * Return code is 0 normally, 1 if a syntax error is detected in src.
3290  * Resource errors (in xmalloc) cause the process to exit */
3291 static int setup_redirect(struct parse_context *ctx, int fd, redir_type style,
3292         struct in_str *input)
3293 {
3294         struct command *command = ctx->command;
3295         struct redir_struct *redir = command->redirects;
3296         struct redir_struct *last_redir = NULL;
3297
3298         /* Create a new redir_struct and drop it onto the end of the linked list */
3299         while (redir) {
3300                 last_redir = redir;
3301                 redir = redir->next;
3302         }
3303         redir = xzalloc(sizeof(struct redir_struct));
3304         /* redir->next = NULL; */
3305         /* redir->rd_filename = NULL; */
3306         if (last_redir) {
3307                 last_redir->next = redir;
3308         } else {
3309                 command->redirects = redir;
3310         }
3311
3312         redir->rd_type = style;
3313         redir->fd = (fd == -1) ? redir_table[style].default_fd : fd;
3314
3315         debug_printf("Redirect type %d%s\n", redir->fd, redir_table[style].descrip);
3316
3317         /* Check for a '2>&1' type redirect */
3318         redir->dup = redirect_dup_num(input);
3319         if (redir->dup == -2)
3320                 return 1;  /* syntax error */
3321         if (redir->dup != -1) {
3322                 /* Erik had a check here that the file descriptor in question
3323                  * is legit; I postpone that to "run time"
3324                  * A "-" representation of "close me" shows up as a -3 here */
3325                 debug_printf("Duplicating redirect '%d>&%d'\n", redir->fd, redir->dup);
3326         } else {
3327                 /* We do _not_ try to open the file that src points to,
3328                  * since we need to return and let src be expanded first.
3329                  * Set ctx->pending_redirect, so we know what to do at the
3330                  * end of the next parsed word. */
3331                 ctx->pending_redirect = redir;
3332         }
3333         return 0;
3334 }
3335
3336
3337 static struct pipe *new_pipe(void)
3338 {
3339         struct pipe *pi;
3340         pi = xzalloc(sizeof(struct pipe));
3341         /*pi->followup = 0; - deliberately invalid value */
3342         /*pi->res_word = RES_NONE; - RES_NONE is 0 anyway */
3343         return pi;
3344 }
3345
3346 /* Command (member of a pipe) is complete. The only possible error here
3347  * is out of memory, in which case xmalloc exits. */
3348 static int done_command(struct parse_context *ctx)
3349 {
3350         /* The command is really already in the pipe structure, so
3351          * advance the pipe counter and make a new, null command. */
3352         struct pipe *pi = ctx->pipe;
3353         struct command *command = ctx->command;
3354
3355         if (command) {
3356                 if (command->group == NULL
3357                  && command->argv == NULL
3358                  && command->redirects == NULL
3359                 ) {
3360                         debug_printf_parse("done_command: skipping null cmd, num_cmds=%d\n", pi->num_cmds);
3361                         return pi->num_cmds;
3362                 }
3363                 pi->num_cmds++;
3364                 debug_printf_parse("done_command: ++num_cmds=%d\n", pi->num_cmds);
3365         } else {
3366                 debug_printf_parse("done_command: initializing, num_cmds=%d\n", pi->num_cmds);
3367         }
3368
3369         /* Only real trickiness here is that the uncommitted
3370          * command structure is not counted in pi->num_cmds. */
3371         pi->cmds = xrealloc(pi->cmds, sizeof(*pi->cmds) * (pi->num_cmds+1));
3372         command = &pi->cmds[pi->num_cmds];
3373         memset(command, 0, sizeof(*command));
3374
3375         ctx->command = command;
3376         /* but ctx->pipe and ctx->list_head remain unchanged */
3377
3378         return pi->num_cmds; /* used only for 0/nonzero check */
3379 }
3380
3381 static void done_pipe(struct parse_context *ctx, pipe_style type)
3382 {
3383         int not_null;
3384
3385         debug_printf_parse("done_pipe entered, followup %d\n", type);
3386         /* Close previous command */
3387         not_null = done_command(ctx);
3388         ctx->pipe->followup = type;
3389         IF_HAS_KEYWORDS(ctx->pipe->pi_inverted = ctx->ctx_inverted;)
3390         IF_HAS_KEYWORDS(ctx->ctx_inverted = 0;)
3391         IF_HAS_KEYWORDS(ctx->pipe->res_word = ctx->ctx_res_w;)
3392
3393         /* Without this check, even just <enter> on command line generates
3394          * tree of three NOPs (!). Which is harmless but annoying.
3395          * IOW: it is safe to do it unconditionally.
3396          * RES_NONE case is for "for a in; do ..." (empty IN set)
3397          * to work, possibly other cases too. */
3398         if (not_null IF_HAS_KEYWORDS(|| ctx->ctx_res_w != RES_NONE)) {
3399                 struct pipe *new_p;
3400                 debug_printf_parse("done_pipe: adding new pipe: "
3401                                 "not_null:%d ctx->ctx_res_w:%d\n",
3402                                 not_null, ctx->ctx_res_w);
3403                 new_p = new_pipe();
3404                 ctx->pipe->next = new_p;
3405                 ctx->pipe = new_p;
3406                 ctx->command = NULL; /* needed! */
3407                 /* RES_THEN, RES_DO etc are "sticky" -
3408                  * they remain set for commands inside if/while.
3409                  * This is used to control execution.
3410                  * RES_FOR and RES_IN are NOT sticky (needed to support
3411                  * cases where variable or value happens to match a keyword):
3412                  */
3413 #if ENABLE_HUSH_LOOPS
3414                 if (ctx->ctx_res_w == RES_FOR
3415                  || ctx->ctx_res_w == RES_IN)
3416                         ctx->ctx_res_w = RES_NONE;
3417 #endif
3418 #if ENABLE_HUSH_CASE
3419                 if (ctx->ctx_res_w == RES_MATCH)
3420                         ctx->ctx_res_w = RES_CASEI;
3421 #endif
3422                 /* Create the memory for command, roughly:
3423                  * ctx->pipe->cmds = new struct command;
3424                  * ctx->command = &ctx->pipe->cmds[0];
3425                  */
3426                 done_command(ctx);
3427         }
3428         debug_printf_parse("done_pipe return\n");
3429 }
3430
3431 static void initialize_context(struct parse_context *ctx)
3432 {
3433         memset(ctx, 0, sizeof(*ctx));
3434         ctx->pipe = ctx->list_head = new_pipe();
3435         /* Create the memory for command, roughly:
3436          * ctx->pipe->cmds = new struct command;
3437          * ctx->command = &ctx->pipe->cmds[0];
3438          */
3439         done_command(ctx);
3440 }
3441
3442
3443 /* If a reserved word is found and processed, parse context is modified
3444  * and 1 is returned.
3445  */
3446 #if HAS_KEYWORDS
3447 struct reserved_combo {
3448         char literal[6];
3449         unsigned char res;
3450         unsigned char assignment_flag;
3451         int flag;
3452 };
3453 enum {
3454         FLAG_END   = (1 << RES_NONE ),
3455 #if ENABLE_HUSH_IF
3456         FLAG_IF    = (1 << RES_IF   ),
3457         FLAG_THEN  = (1 << RES_THEN ),
3458         FLAG_ELIF  = (1 << RES_ELIF ),
3459         FLAG_ELSE  = (1 << RES_ELSE ),
3460         FLAG_FI    = (1 << RES_FI   ),
3461 #endif
3462 #if ENABLE_HUSH_LOOPS
3463         FLAG_FOR   = (1 << RES_FOR  ),
3464         FLAG_WHILE = (1 << RES_WHILE),
3465         FLAG_UNTIL = (1 << RES_UNTIL),
3466         FLAG_DO    = (1 << RES_DO   ),
3467         FLAG_DONE  = (1 << RES_DONE ),
3468         FLAG_IN    = (1 << RES_IN   ),
3469 #endif
3470 #if ENABLE_HUSH_CASE
3471         FLAG_MATCH = (1 << RES_MATCH),
3472         FLAG_ESAC  = (1 << RES_ESAC ),
3473 #endif
3474         FLAG_START = (1 << RES_XXXX ),
3475 };
3476
3477 static const struct reserved_combo* match_reserved_word(o_string *word)
3478 {
3479         /* Mostly a list of accepted follow-up reserved words.
3480          * FLAG_END means we are done with the sequence, and are ready
3481          * to turn the compound list into a command.
3482          * FLAG_START means the word must start a new compound list.
3483          */
3484         static const struct reserved_combo reserved_list[] = {
3485 #if ENABLE_HUSH_IF
3486                 { "!",     RES_NONE,  NOT_ASSIGNMENT , 0 },
3487                 { "if",    RES_IF,    WORD_IS_KEYWORD, FLAG_THEN | FLAG_START },
3488                 { "then",  RES_THEN,  WORD_IS_KEYWORD, FLAG_ELIF | FLAG_ELSE | FLAG_FI },
3489                 { "elif",  RES_ELIF,  WORD_IS_KEYWORD, FLAG_THEN },
3490                 { "else",  RES_ELSE,  WORD_IS_KEYWORD, FLAG_FI   },
3491                 { "fi",    RES_FI,    NOT_ASSIGNMENT , FLAG_END  },
3492 #endif
3493 #if ENABLE_HUSH_LOOPS
3494                 { "for",   RES_FOR,   NOT_ASSIGNMENT , FLAG_IN | FLAG_DO | FLAG_START },
3495                 { "while", RES_WHILE, WORD_IS_KEYWORD, FLAG_DO | FLAG_START },
3496                 { "until", RES_UNTIL, WORD_IS_KEYWORD, FLAG_DO | FLAG_START },
3497                 { "in",    RES_IN,    NOT_ASSIGNMENT , FLAG_DO   },
3498                 { "do",    RES_DO,    WORD_IS_KEYWORD, FLAG_DONE },
3499                 { "done",  RES_DONE,  NOT_ASSIGNMENT , FLAG_END  },
3500 #endif
3501 #if ENABLE_HUSH_CASE
3502                 { "case",  RES_CASE,  NOT_ASSIGNMENT , FLAG_MATCH | FLAG_START },
3503                 { "esac",  RES_ESAC,  NOT_ASSIGNMENT , FLAG_END  },
3504 #endif
3505         };
3506         const struct reserved_combo *r;
3507
3508         for (r = reserved_list; r < reserved_list + ARRAY_SIZE(reserved_list); r++) {
3509                 if (strcmp(word->data, r->literal) == 0)
3510                         return r;
3511         }
3512         return NULL;
3513 }
3514 static int reserved_word(o_string *word, struct parse_context *ctx)
3515 {
3516 #if ENABLE_HUSH_CASE
3517         static const struct reserved_combo reserved_match = {
3518                 "",        RES_MATCH, NOT_ASSIGNMENT , FLAG_MATCH | FLAG_ESAC
3519         };
3520 #endif
3521         const struct reserved_combo *r;
3522
3523         r = match_reserved_word(word);
3524         if (!r)
3525                 return 0;
3526
3527         debug_printf("found reserved word %s, res %d\n", r->literal, r->res);
3528 #if ENABLE_HUSH_CASE
3529         if (r->res == RES_IN && ctx->ctx_res_w == RES_CASE)
3530                 /* "case word IN ..." - IN part starts first match part */
3531                 r = &reserved_match;
3532         else
3533 #endif
3534         if (r->flag == 0) { /* '!' */
3535                 if (ctx->ctx_inverted) { /* bash doesn't accept '! ! true' */
3536                         syntax(NULL);
3537                         IF_HAS_KEYWORDS(ctx->ctx_res_w = RES_SNTX;)
3538                 }
3539                 ctx->ctx_inverted = 1;
3540                 return 1;
3541         }
3542         if (r->flag & FLAG_START) {
3543                 struct parse_context *new;
3544                 debug_printf("push stack\n");
3545                 new = xmalloc(sizeof(*new));
3546                 *new = *ctx;   /* physical copy */
3547                 initialize_context(ctx);
3548                 ctx->stack = new;
3549         } else if (/*ctx->ctx_res_w == RES_NONE ||*/ !(ctx->old_flag & (1 << r->res))) {
3550                 syntax(NULL);
3551                 ctx->ctx_res_w = RES_SNTX;
3552                 return 1;
3553         }
3554         ctx->ctx_res_w = r->res;
3555         ctx->old_flag = r->flag;
3556         if (ctx->old_flag & FLAG_END) {
3557                 struct parse_context *old;
3558                 debug_printf("pop stack\n");
3559                 done_pipe(ctx, PIPE_SEQ);
3560                 old = ctx->stack;
3561                 old->command->group = ctx->list_head;
3562                 old->command->grp_type = GRP_NORMAL;
3563                 *ctx = *old;   /* physical copy */
3564                 free(old);
3565         }
3566         word->o_assignment = r->assignment_flag;
3567         return 1;
3568 }
3569 #endif
3570
3571 //TODO: many, many callers don't check error from done_word()
3572
3573 /* Word is complete, look at it and update parsing context.
3574  * Normal return is 0. Syntax errors return 1. */
3575 static int done_word(o_string *word, struct parse_context *ctx)
3576 {
3577         struct command *command = ctx->command;
3578
3579         debug_printf_parse("done_word entered: '%s' %p\n", word->data, command);
3580         if (word->length == 0 && word->nonnull == 0) {
3581                 debug_printf_parse("done_word return 0: true null, ignored\n");
3582                 return 0;
3583         }
3584         /* If this word wasn't an assignment, next ones definitely
3585          * can't be assignments. Even if they look like ones. */
3586         if (word->o_assignment != DEFINITELY_ASSIGNMENT
3587          && word->o_assignment != WORD_IS_KEYWORD
3588         ) {
3589                 word->o_assignment = NOT_ASSIGNMENT;
3590         } else {
3591                 if (word->o_assignment == DEFINITELY_ASSIGNMENT)
3592                         command->assignment_cnt++;
3593                 word->o_assignment = MAYBE_ASSIGNMENT;
3594         }
3595
3596         if (ctx->pending_redirect) {
3597                 /* We do not glob in e.g. >*.tmp case. bash seems to glob here
3598                  * only if run as "bash", not "sh" */
3599                 ctx->pending_redirect->rd_filename = xstrdup(word->data);
3600                 word->o_assignment = NOT_ASSIGNMENT;
3601                 debug_printf("word stored in rd_filename: '%s'\n", word->data);
3602         } else {
3603                 /* "{ echo foo; } echo bar" - bad */
3604                 /* NB: bash allows e.g. "if true; then { echo foo; } fi". TODO? */
3605                 if (command->group) {
3606                         syntax(NULL);
3607                         debug_printf_parse("done_word return 1: syntax error, groups and arglists don't mix\n");
3608                         return 1;
3609                 }
3610 #if HAS_KEYWORDS
3611 #if ENABLE_HUSH_CASE
3612                 if (ctx->ctx_dsemicolon
3613                  && strcmp(word->data, "esac") != 0 /* not "... pattern) cmd;; esac" */
3614                 ) {
3615                         /* already done when ctx_dsemicolon was set to 1: */
3616                         /* ctx->ctx_res_w = RES_MATCH; */
3617                         ctx->ctx_dsemicolon = 0;
3618                 } else
3619 #endif
3620
3621                 if (!command->argv /* if it's the first word... */
3622 #if ENABLE_HUSH_LOOPS
3623                  && ctx->ctx_res_w != RES_FOR /* ...not after FOR or IN */
3624                  && ctx->ctx_res_w != RES_IN
3625 #endif
3626                 ) {
3627                         debug_printf_parse(": checking '%s' for reserved-ness\n", word->data);
3628                         if (reserved_word(word, ctx)) {
3629                                 o_reset(word);
3630                                 debug_printf_parse("done_word return %d\n", (ctx->ctx_res_w == RES_SNTX));
3631                                 return (ctx->ctx_res_w == RES_SNTX);
3632                         }
3633                 }
3634 #endif
3635                 if (word->nonnull /* word had "xx" or 'xx' at least as part of it. */
3636                  /* optimization: and if it's ("" or '') or ($v... or `cmd`...): */
3637                  && (word->data[0] == '\0' || word->data[0] == SPECIAL_VAR_SYMBOL)
3638                  /* (otherwise it's known to be not empty and is already safe) */
3639                 ) {
3640                         /* exclude "$@" - it can expand to no word despite "" */
3641                         char *p = word->data;
3642                         while (p[0] == SPECIAL_VAR_SYMBOL
3643                             && (p[1] & 0x7f) == '@'
3644                             && p[2] == SPECIAL_VAR_SYMBOL
3645                         ) {
3646                                 p += 3;
3647                         }
3648                         if (p == word->data || p[0] != '\0') {
3649                                 /* saw no "$@", or not only "$@" but some
3650                                  * real text is there too */
3651                                 /* insert "empty variable" reference, this makes
3652                                  * e.g. "", $empty"" etc to not disappear */
3653                                 o_addchr(word, SPECIAL_VAR_SYMBOL);
3654                                 o_addchr(word, SPECIAL_VAR_SYMBOL);
3655                         }
3656                 }
3657                 command->argv = add_string_to_strings(command->argv, xstrdup(word->data));
3658                 debug_print_strings("word appended to argv", command->argv);
3659         }
3660
3661         o_reset(word);
3662         ctx->pending_redirect = NULL;
3663
3664 #if ENABLE_HUSH_LOOPS
3665         /* Force FOR to have just one word (variable name) */
3666         /* NB: basically, this makes hush see "for v in ..." syntax as if
3667          * as it is "for v; in ...". FOR and IN become two pipe structs
3668          * in parse tree. */
3669         if (ctx->ctx_res_w == RES_FOR) {
3670 //TODO: check that command->argv[0] is a valid variable name!
3671                 done_pipe(ctx, PIPE_SEQ);
3672         }
3673 #endif
3674 #if ENABLE_HUSH_CASE
3675         /* Force CASE to have just one word */
3676         if (ctx->ctx_res_w == RES_CASE) {
3677                 done_pipe(ctx, PIPE_SEQ);
3678         }
3679 #endif
3680         debug_printf_parse("done_word return 0\n");
3681         return 0;
3682 }
3683
3684 /* If a redirect is immediately preceded by a number, that number is
3685  * supposed to tell which file descriptor to redirect.  This routine
3686  * looks for such preceding numbers.  In an ideal world this routine
3687  * needs to handle all the following classes of redirects...
3688  *     echo 2>foo     # redirects fd  2 to file "foo", nothing passed to echo
3689  *     echo 49>foo    # redirects fd 49 to file "foo", nothing passed to echo
3690  *     echo -2>foo    # redirects fd  1 to file "foo",    "-2" passed to echo
3691  *     echo 49x>foo   # redirects fd  1 to file "foo",   "49x" passed to echo
3692  * A -1 output from this program means no valid number was found, so the
3693  * caller should use the appropriate default for this redirection.
3694  */
3695 static int redirect_opt_num(o_string *o)
3696 {
3697         int num;
3698
3699         if (o->length == 0)
3700                 return -1;
3701         for (num = 0; num < o->length; num++) {
3702                 if (!isdigit(o->data[num])) {
3703                         return -1;
3704                 }
3705         }
3706         num = atoi(o->data);
3707         o_reset(o);
3708         return num;
3709 }
3710
3711 static int parse_stream(o_string *dest, struct parse_context *ctx,
3712                 struct in_str *input0, const char *end_trigger);
3713
3714 #if ENABLE_HUSH_TICK
3715 static FILE *generate_stream_from_list(struct pipe *head)
3716 {
3717         FILE *pf;
3718         int pid, channel[2];
3719
3720         xpipe(channel);
3721 /* *** NOMMU WARNING *** */
3722 /* By using vfork here, we suspend parent till child exits or execs.
3723  * If child will not do it before it fills the pipe, it can block forever
3724  * in write(STDOUT_FILENO), and parent (shell) will be also stuck.
3725  * Try this script:
3726  * yes "0123456789012345678901234567890" | dd bs=32 count=64k >TESTFILE
3727  * huge=`cat TESTFILE` # will block here forever
3728  * echo OK
3729  */
3730         pid = BB_MMU ? fork() : vfork();
3731         if (pid < 0)
3732                 bb_perror_msg_and_die(BB_MMU ? "fork" : "vfork");
3733         if (pid == 0) { /* child */
3734                 if (ENABLE_HUSH_JOB)
3735                         die_sleep = 0; /* let nofork's xfuncs die */
3736                 close(channel[0]); /* NB: close _first_, then move fd! */
3737                 xmove_fd(channel[1], 1);
3738                 /* Prevent it from trying to handle ctrl-z etc */
3739 #if ENABLE_HUSH_JOB
3740                 G.run_list_level = 1;
3741 #endif
3742                 /* Process substitution is not considered to be usual
3743                  * 'command execution'.
3744                  * SUSv3 says ctrl-Z should be ignored, ctrl-C should not.
3745                  */
3746                 set_jobctrl_signals_to_IGN();
3747
3748                 /* Note: freeing 'head' here would break NOMMU. */
3749                 _exit(run_list(head));
3750         }
3751         close(channel[1]);
3752         pf = fdopen(channel[0], "r");
3753         return pf;
3754         /* 'head' is freed by the caller */
3755 }
3756
3757 /* Return code is exit status of the process that is run. */
3758 static int process_command_subs(o_string *dest,
3759                 struct in_str *input,
3760                 const char *subst_end)
3761 {
3762         int retcode, ch, eol_cnt;
3763         o_string result = NULL_O_STRING;
3764         struct parse_context inner;
3765         FILE *p;
3766         struct in_str pipe_str;
3767
3768         initialize_context(&inner);
3769
3770         /* Recursion to generate command */
3771         retcode = parse_stream(&result, &inner, input, subst_end);
3772         if (retcode != 0)
3773                 return retcode;  /* syntax error or EOF */
3774         done_word(&result, &inner);
3775         done_pipe(&inner, PIPE_SEQ);
3776         o_free(&result);
3777
3778         p = generate_stream_from_list(inner.list_head);
3779         if (p == NULL)
3780                 return 1;
3781         close_on_exec_on(fileno(p));
3782         setup_file_in_str(&pipe_str, p);
3783
3784         /* Now send results of command back into original context */
3785         eol_cnt = 0;
3786         while ((ch = i_getch(&pipe_str)) != EOF) {
3787                 if (ch == '\n') {
3788                         eol_cnt++;
3789                         continue;
3790                 }
3791                 while (eol_cnt) {
3792                         o_addchr(dest, '\n');
3793                         eol_cnt--;
3794                 }
3795                 o_addQchr(dest, ch);
3796         }
3797
3798         debug_printf("done reading from pipe, pclose()ing\n");
3799         /* Note: we got EOF, and we just close the read end of the pipe.
3800          * We do not wait for the `cmd` child to terminate. bash and ash do.
3801          * Try this:
3802          * echo `echo Hi; exec 1>&-; sleep 2`
3803          */
3804         retcode = fclose(p);
3805         free_pipe_list(inner.list_head, /* indent: */ 0);
3806         debug_printf("closed FILE from child, retcode=%d\n", retcode);
3807         return retcode;
3808 }
3809 #endif
3810
3811 static int parse_group(o_string *dest, struct parse_context *ctx,
3812         struct in_str *input, int ch)
3813 {
3814         /* dest contains characters seen prior to ( or {.
3815          * Typically it's empty, but for functions defs,
3816          * it contains function name (without '()'). */
3817         int rcode;
3818         const char *endch = NULL;
3819         struct parse_context sub;
3820         struct command *command = ctx->command;
3821
3822         debug_printf_parse("parse_group entered\n");
3823 #if ENABLE_HUSH_FUNCTIONS
3824         if (ch == 'F') { /* function definition? */
3825                 bb_error_msg("aha '%s' is a function, parsing it...", dest->data);
3826                 //command->fname = dest->data;
3827                 command->grp_type = GRP_FUNCTION;
3828 //TODO: review every o_reset() location... do they handle all o_string fields correctly?
3829                 memset(dest, 0, sizeof(*dest));
3830         }
3831 #endif
3832         if (command->argv /* word [word](... */
3833          || dest->length /* word(... */
3834          || dest->nonnull /* ""(... */
3835         ) {
3836                 syntax(NULL);
3837                 debug_printf_parse("parse_group return 1: syntax error, groups and arglists don't mix\n");
3838                 return 1;
3839         }
3840         initialize_context(&sub);
3841         endch = "}";
3842         if (ch == '(') {
3843                 endch = ")";
3844                 command->grp_type = GRP_SUBSHELL;
3845         }
3846         rcode = parse_stream(dest, &sub, input, endch);
3847         if (rcode == 0) {
3848                 done_word(dest, &sub); /* finish off the final word in the subcontext */
3849                 done_pipe(&sub, PIPE_SEQ);  /* and the final command there, too */
3850                 command->group = sub.list_head;
3851         }
3852         debug_printf_parse("parse_group return %d\n", rcode);
3853         return rcode;
3854         /* command remains "open", available for possible redirects */
3855 }
3856
3857 #if ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT
3858 /* Subroutines for copying $(...) and `...` things */
3859 static void add_till_backquote(o_string *dest, struct in_str *input);
3860 /* '...' */
3861 static void add_till_single_quote(o_string *dest, struct in_str *input)
3862 {
3863         while (1) {
3864                 int ch = i_getch(input);
3865                 if (ch == EOF)
3866                         break;
3867                 if (ch == '\'')
3868                         break;
3869                 o_addchr(dest, ch);
3870         }
3871 }
3872 /* "...\"...`..`...." - do we need to handle "...$(..)..." too? */
3873 static void add_till_double_quote(o_string *dest, struct in_str *input)
3874 {
3875         while (1) {
3876                 int ch = i_getch(input);
3877                 if (ch == '"')
3878                         break;
3879                 if (ch == '\\') {  /* \x. Copy both chars. */
3880                         o_addchr(dest, ch);
3881                         ch = i_getch(input);
3882                 }
3883                 if (ch == EOF)
3884                         break;
3885                 o_addchr(dest, ch);
3886                 if (ch == '`') {
3887                         add_till_backquote(dest, input);
3888                         o_addchr(dest, ch);
3889                         continue;
3890                 }
3891                 //if (ch == '$') ...
3892         }
3893 }
3894 /* Process `cmd` - copy contents until "`" is seen. Complicated by
3895  * \` quoting.
3896  * "Within the backquoted style of command substitution, backslash
3897  * shall retain its literal meaning, except when followed by: '$', '`', or '\'.
3898  * The search for the matching backquote shall be satisfied by the first
3899  * backquote found without a preceding backslash; during this search,
3900  * if a non-escaped backquote is encountered within a shell comment,
3901  * a here-document, an embedded command substitution of the $(command)
3902  * form, or a quoted string, undefined results occur. A single-quoted
3903  * or double-quoted string that begins, but does not end, within the
3904  * "`...`" sequence produces undefined results."
3905  * Example                               Output
3906  * echo `echo '\'TEST\`echo ZZ\`BEST`    \TESTZZBEST
3907  */
3908 static void add_till_backquote(o_string *dest, struct in_str *input)
3909 {
3910         while (1) {
3911                 int ch = i_getch(input);
3912                 if (ch == '`')
3913                         break;
3914                 if (ch == '\\') {  /* \x. Copy both chars unless it is \` */
3915                         int ch2 = i_getch(input);
3916                         if (ch2 != '`' && ch2 != '$' && ch2 != '\\')
3917                                 o_addchr(dest, ch);
3918                         ch = ch2;
3919                 }
3920                 if (ch == EOF)
3921                         break;
3922                 o_addchr(dest, ch);
3923         }
3924 }
3925 /* Process $(cmd) - copy contents until ")" is seen. Complicated by
3926  * quoting and nested ()s.
3927  * "With the $(command) style of command substitution, all characters
3928  * following the open parenthesis to the matching closing parenthesis
3929  * constitute the command. Any valid shell script can be used for command,
3930  * except a script consisting solely of redirections which produces
3931  * unspecified results."
3932  * Example                              Output
3933  * echo $(echo '(TEST)' BEST)           (TEST) BEST
3934  * echo $(echo 'TEST)' BEST)            TEST) BEST
3935  * echo $(echo \(\(TEST\) BEST)         ((TEST) BEST
3936  */
3937 static void add_till_closing_paren(o_string *dest, struct in_str *input, bool dbl)
3938 {
3939         int count = 0;
3940         while (1) {
3941                 int ch = i_getch(input);
3942                 if (ch == EOF)
3943                         break;
3944                 if (ch == '(')
3945                         count++;
3946                 if (ch == ')')
3947                         if (--count < 0) {
3948                                 if (!dbl)
3949                                         break;
3950                                 if (i_peek(input) == ')') {
3951                                         i_getch(input);
3952                                         break;
3953                                 }
3954                         }
3955                 o_addchr(dest, ch);
3956                 if (ch == '\'') {
3957                         add_till_single_quote(dest, input);
3958                         o_addchr(dest, ch);
3959                         continue;
3960                 }
3961                 if (ch == '"') {
3962                         add_till_double_quote(dest, input);
3963                         o_addchr(dest, ch);
3964                         continue;
3965                 }
3966                 if (ch == '\\') { /* \x. Copy verbatim. Important for  \(, \) */
3967                         ch = i_getch(input);
3968                         if (ch == EOF)
3969                                 break;
3970                         o_addchr(dest, ch);
3971                         continue;
3972                 }
3973         }
3974 }
3975 #endif /* ENABLE_HUSH_TICK || ENABLE_SH_MATH_SUPPORT */
3976
3977 /* Return code: 0 for OK, 1 for syntax error */
3978 static int handle_dollar(o_string *dest, struct in_str *input)
3979 {
3980         int expansion;
3981         int ch = i_peek(input);  /* first character after the $ */
3982         unsigned char quote_mask = dest->o_escape ? 0x80 : 0;
3983
3984         debug_printf_parse("handle_dollar entered: ch='%c'\n", ch);
3985         if (isalpha(ch)) {
3986                 i_getch(input);
3987  make_var:
3988                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3989                 while (1) {
3990                         debug_printf_parse(": '%c'\n", ch);
3991                         o_addchr(dest, ch | quote_mask);
3992                         quote_mask = 0;
3993                         ch = i_peek(input);
3994                         if (!isalnum(ch) && ch != '_')
3995                                 break;
3996                         i_getch(input);
3997                 }
3998                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
3999         } else if (isdigit(ch)) {
4000  make_one_char_var:
4001                 i_getch(input);
4002                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4003                 debug_printf_parse(": '%c'\n", ch);
4004                 o_addchr(dest, ch | quote_mask);
4005                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4006         } else switch (ch) {
4007                 case '$': /* pid */
4008                 case '!': /* last bg pid */
4009                 case '?': /* last exit code */
4010                 case '#': /* number of args */
4011                 case '*': /* args */
4012                 case '@': /* args */
4013                         goto make_one_char_var;
4014                 case '{': {
4015                         bool first_char, all_digits;
4016
4017                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4018                         i_getch(input);
4019                         /* XXX maybe someone will try to escape the '}' */
4020                         expansion = 0;
4021                         first_char = true;
4022                         all_digits = false;
4023                         while (1) {
4024                                 ch = i_getch(input);
4025                                 if (ch == '}')
4026                                         break;
4027
4028                                 if (first_char) {
4029                                         if (ch == '#')
4030                                                 /* ${#var}: length of var contents */
4031                                                 goto char_ok;
4032                                         else if (isdigit(ch)) {
4033                                                 all_digits = true;
4034                                                 goto char_ok;
4035                                         }
4036                                 }
4037
4038                                 if (expansion < 2 &&
4039                                     ((all_digits && !isdigit(ch)) ||
4040                                      (!all_digits && !isalnum(ch) && ch != '_')))
4041                                 {
4042                                         /* handle parameter expansions
4043                                          * http://www.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
4044                                          */
4045                                         if (first_char)
4046                                                 goto case_default;
4047                                         switch (ch) {
4048                                                 case ':': /* null modifier */
4049                                                         if (expansion == 0) {
4050                                                                 debug_printf_parse(": null modifier\n");
4051                                                                 ++expansion;
4052                                                                 break;
4053                                                         }
4054                                                         goto case_default;
4055
4056 #if 0 /* not implemented yet :( */
4057                                                 case '#': /* remove prefix */
4058                                                 case '%': /* remove suffix */
4059                                                         if (expansion == 0) {
4060                                                                 debug_printf_parse(": remove suffix/prefix\n");
4061                                                                 expansion = 2;
4062                                                                 break;
4063                                                         }
4064                                                         goto case_default;
4065 #endif
4066
4067                                                 case '-': /* default value */
4068                                                 case '=': /* assign default */
4069                                                 case '+': /* alternative */
4070                                                 case '?': /* error indicate */
4071                                                         debug_printf_parse(": parameter expansion\n");
4072                                                         expansion = 2;
4073                                                         break;
4074
4075                                                 default:
4076                                                 case_default:
4077                                                         syntax("unterminated ${name}");
4078                                                         debug_printf_parse("handle_dollar return 1: unterminated ${name}\n");
4079                                                         return 1;
4080                                                 }
4081                                 }
4082
4083  char_ok:
4084                                 debug_printf_parse(": '%c'\n", ch);
4085                                 o_addchr(dest, ch | quote_mask);
4086                                 quote_mask = 0;
4087                                 first_char = false;
4088                         }
4089                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4090                         break;
4091                 }
4092                 case '(': {
4093                         i_getch(input);
4094
4095 #if ENABLE_SH_MATH_SUPPORT
4096                         if (i_peek(input) == '(') {
4097                                 i_getch(input);
4098                                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4099                                 o_addchr(dest, /*quote_mask |*/ '+');
4100                                 add_till_closing_paren(dest, input, true);
4101                                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4102                                 break;
4103                         }
4104 #endif
4105
4106 #if ENABLE_HUSH_TICK
4107                         //int pos = dest->length;
4108                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4109                         o_addchr(dest, quote_mask | '`');
4110                         add_till_closing_paren(dest, input, false);
4111                         //debug_printf_subst("SUBST RES2 '%s'\n", dest->data + pos);
4112                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4113 #endif
4114                         break;
4115                 }
4116                 case '_':
4117                         i_getch(input);
4118                         ch = i_peek(input);
4119                         if (isalnum(ch)) { /* it's $_name or $_123 */
4120                                 ch = '_';
4121                                 goto make_var;
4122                         }
4123                         /* else: it's $_ */
4124                 case '-':
4125                         /* still unhandled, but should be eventually */
4126                         bb_error_msg("unhandled syntax: $%c", ch);
4127                         return 1;
4128                         break;
4129                 default:
4130                         o_addQchr(dest, '$');
4131         }
4132         debug_printf_parse("handle_dollar return 0\n");
4133         return 0;
4134 }
4135
4136 static int parse_stream_dquoted(o_string *dest, struct in_str *input, int dquote_end)
4137 {
4138         int ch, m;
4139         int next;
4140
4141  again:
4142         ch = i_getch(input);
4143         if (ch == dquote_end) { /* may be only '"' or EOF */
4144                 dest->nonnull = 1;
4145                 if (dest->o_assignment == NOT_ASSIGNMENT)
4146                         dest->o_escape ^= 1;
4147                 debug_printf_parse("parse_stream_dquoted return 0\n");
4148                 return 0;
4149         }
4150         if (ch == EOF) {
4151                 syntax("unterminated \"");
4152                 debug_printf_parse("parse_stream_dquoted return 1: unterminated \"\n");
4153                 return 1;
4154         }
4155         next = '\0';
4156         m = G.charmap[ch];
4157         if (ch != '\n') {
4158                 next = i_peek(input);
4159         }
4160         debug_printf_parse(": ch=%c (%d) m=%d escape=%d\n",
4161                                         ch, ch, m, dest->o_escape);
4162         /* Basically, checking every CHAR_SPECIAL char except '"' */
4163         if (ch == '\\') {
4164                 if (next == EOF) {
4165                         syntax("\\<eof>");
4166                         debug_printf_parse("parse_stream_dquoted return 1: \\<eof>\n");
4167                         return 1;
4168                 }
4169                 /* bash:
4170                  * "The backslash retains its special meaning [in "..."]
4171                  * only when followed by one of the following characters:
4172                  * $, `, ", \, or <newline>.  A double quote may be quoted
4173                  * within double quotes by preceding it with a backslash.
4174                  * If enabled, history expansion will be performed unless
4175                  * an ! appearing in double quotes is escaped using
4176                  * a backslash. The backslash preceding the ! is not removed."
4177                  */
4178                 if (strchr("$`\"\\", next) != NULL) {
4179                         o_addqchr(dest, i_getch(input));
4180                 } else {
4181                         o_addqchr(dest, '\\');
4182                 }
4183                 goto again;
4184         }
4185         if (ch == '$') {
4186                 if (handle_dollar(dest, input) != 0) {
4187                         debug_printf_parse("parse_stream_dquoted return 1: handle_dollar returned non-0\n");
4188                         return 1;
4189                 }
4190                 goto again;
4191         }
4192 #if ENABLE_HUSH_TICK
4193         if (ch == '`') {
4194                 //int pos = dest->length;
4195                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4196                 o_addchr(dest, 0x80 | '`');
4197                 add_till_backquote(dest, input);
4198                 o_addchr(dest, SPECIAL_VAR_SYMBOL);
4199                 //debug_printf_subst("SUBST RES3 '%s'\n", dest->data + pos);
4200                 goto again;
4201         }
4202 #endif
4203         o_addQchr(dest, ch);
4204         if (ch == '='
4205          && (dest->o_assignment == MAYBE_ASSIGNMENT
4206             || dest->o_assignment == WORD_IS_KEYWORD)
4207          && is_assignment(dest->data)
4208         ) {
4209                 dest->o_assignment = DEFINITELY_ASSIGNMENT;
4210         }
4211         goto again;
4212 }
4213
4214 /* Scan input, call done_word() whenever full IFS delimited word was seen.
4215  * Call done_pipe if '\n' was seen (and end_trigger != NULL).
4216  * Return code is 0 if end_trigger char is met,
4217  * -1 on EOF (but if end_trigger == NULL then return 0),
4218  * 1 for syntax error */
4219 static int parse_stream(o_string *dest, struct parse_context *ctx,
4220                 struct in_str *input, const char *end_trigger)
4221 {
4222         int ch, m;
4223         int redir_fd;
4224         redir_type redir_style;
4225         int is_in_dquote;
4226         int next;
4227
4228         /* Double-quote state is handled in the state variable is_in_dquote.
4229          * A single-quote triggers a bypass of the main loop until its mate is
4230          * found.  When recursing, quote state is passed in via dest->o_escape. */
4231
4232         debug_printf_parse("parse_stream entered, end_trigger='%s' dest->o_assignment:%d\n", end_trigger, dest->o_assignment);
4233
4234         is_in_dquote = dest->o_escape;
4235         while (1) {
4236                 if (is_in_dquote) {
4237                         if (parse_stream_dquoted(dest, input, '"'))
4238                                 return 1; /* propagate parse error */
4239                         /* If we're here, we reached closing '"' */
4240                         is_in_dquote = 0;
4241                 }
4242                 m = CHAR_IFS;
4243                 next = '\0';
4244                 ch = i_getch(input);
4245                 if (ch != EOF) {
4246                         m = G.charmap[ch];
4247                         if (ch != '\n') {
4248                                 next = i_peek(input);
4249                         }
4250                 }
4251                 debug_printf_parse(": ch=%c (%d) m=%d escape=%d\n",
4252                                                 ch, ch, m, dest->o_escape);
4253                 if (m == CHAR_ORDINARY) {
4254                         o_addQchr(dest, ch);
4255                         if ((dest->o_assignment == MAYBE_ASSIGNMENT
4256                             || dest->o_assignment == WORD_IS_KEYWORD)
4257                          && ch == '='
4258                          && is_assignment(dest->data)
4259                         ) {
4260                                 dest->o_assignment = DEFINITELY_ASSIGNMENT;
4261                         }
4262                         continue;
4263                 }
4264                 /* m is SPECIAL ($,`), IFS, or ORDINARY_IF_QUOTED (*,#)
4265                  */
4266                 if (m == CHAR_IFS) {
4267                         if (done_word(dest, ctx)) {
4268                                 debug_printf_parse("parse_stream return 1: done_word!=0\n");
4269                                 return 1;
4270                         }
4271                         if (ch == EOF)
4272                                 break;
4273                         /* If we aren't performing a substitution, treat
4274                          * a newline as a command separator.
4275                          * [why don't we handle it exactly like ';'? --vda] */
4276                         if (end_trigger && ch == '\n') {
4277 #if ENABLE_HUSH_CASE
4278                                 /* "case ... in <newline> word) ..." -
4279                                  * newlines are ignored (but ';' wouldn't be) */
4280                                 if (dest->length == 0 // && argv[0] == NULL
4281                                  && ctx->ctx_res_w == RES_MATCH
4282                                 ) {
4283                                         continue;
4284                                 }
4285 #endif
4286                                 done_pipe(ctx, PIPE_SEQ);
4287                                 dest->o_assignment = MAYBE_ASSIGNMENT;
4288                         }
4289                 }
4290                 if (end_trigger) {
4291                         if (strchr(end_trigger, ch)) {
4292                                 /* Special case: (...word) makes last word terminate,
4293                                  * as if ';' is seen */
4294                                 if (ch == ')') {
4295                                         done_word(dest, ctx);
4296 //err chk?
4297                                         done_pipe(ctx, PIPE_SEQ);
4298                                         dest->o_assignment = MAYBE_ASSIGNMENT;
4299                                 }
4300                                 if (!HAS_KEYWORDS
4301                                  IF_HAS_KEYWORDS(|| (ctx->ctx_res_w == RES_NONE && ctx->old_flag == 0))
4302                                 ) {
4303                                         debug_printf_parse("parse_stream return 0: end_trigger char found\n");
4304                                         return 0;
4305                                 }
4306                         }
4307                 }
4308                 if (m == CHAR_IFS)
4309                         continue;
4310
4311                 /* m is SPECIAL (e.g. $,`) or ORDINARY_IF_QUOTED (*,#) */
4312
4313                 if (dest->o_assignment == MAYBE_ASSIGNMENT) {
4314                         /* ch is a special char and thus this word
4315                          * cannot be an assignment */
4316                         dest->o_assignment = NOT_ASSIGNMENT;
4317                 }
4318
4319                 switch (ch) {
4320                 case '#':
4321                         if (dest->length == 0) {
4322                                 while (1) {
4323                                         ch = i_peek(input);
4324                                         if (ch == EOF || ch == '\n')
4325                                                 break;
4326                                         i_getch(input);
4327                                 }
4328                         } else {
4329                                 o_addQchr(dest, ch);
4330                         }
4331                         break;
4332                 case '\\':
4333                         if (next == EOF) {
4334                                 syntax("\\<eof>");
4335                                 debug_printf_parse("parse_stream return 1: \\<eof>\n");
4336                                 return 1;
4337                         }
4338                         o_addchr(dest, '\\');
4339                         o_addchr(dest, i_getch(input));
4340                         break;
4341                 case '$':
4342                         if (handle_dollar(dest, input) != 0) {
4343                                 debug_printf_parse("parse_stream return 1: handle_dollar returned non-0\n");
4344                                 return 1;
4345                         }
4346                         break;
4347                 case '\'':
4348                         dest->nonnull = 1;
4349                         while (1) {
4350                                 ch = i_getch(input);
4351                                 if (ch == EOF) {
4352                                         syntax("unterminated '");
4353                                         debug_printf_parse("parse_stream return 1: unterminated '\n");
4354                                         return 1;
4355                                 }
4356                                 if (ch == '\'')
4357                                         break;
4358                                 if (dest->o_assignment == NOT_ASSIGNMENT)
4359                                         o_addqchr(dest, ch);
4360                                 else
4361                                         o_addchr(dest, ch);
4362                         }
4363                         break;
4364                 case '"':
4365                         dest->nonnull = 1;
4366                         is_in_dquote ^= 1; /* invert */
4367                         if (dest->o_assignment == NOT_ASSIGNMENT)
4368                                 dest->o_escape ^= 1;
4369                         break;
4370 #if ENABLE_HUSH_TICK
4371                 case '`': {
4372                         //int pos = dest->length;
4373                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4374                         o_addchr(dest, '`');
4375                         add_till_backquote(dest, input);
4376                         o_addchr(dest, SPECIAL_VAR_SYMBOL);
4377                         //debug_printf_subst("SUBST RES3 '%s'\n", dest->data + pos);
4378                         break;
4379                 }
4380 #endif
4381                 case '>':
4382                         redir_fd = redirect_opt_num(dest);
4383                         done_word(dest, ctx);
4384                         redir_style = REDIRECT_OVERWRITE;
4385                         if (next == '>') {
4386                                 redir_style = REDIRECT_APPEND;
4387                                 i_getch(input);
4388                         }
4389 #if 0
4390                         else if (next == '(') {
4391                                 syntax(">(process) not supported");
4392                                 debug_printf_parse("parse_stream return 1: >(process) not supported\n");
4393                                 return 1;
4394                         }
4395 #endif
4396                         setup_redirect(ctx, redir_fd, redir_style, input);
4397                         break;
4398                 case '<':
4399                         redir_fd = redirect_opt_num(dest);
4400                         done_word(dest, ctx);
4401                         redir_style = REDIRECT_INPUT;
4402                         if (next == '<') {
4403                                 redir_style = REDIRECT_HEREIS;
4404                                 i_getch(input);
4405                         } else if (next == '>') {
4406                                 redir_style = REDIRECT_IO;
4407                                 i_getch(input);
4408                         }
4409 #if 0
4410                         else if (next == '(') {
4411                                 syntax("<(process) not supported");
4412                                 debug_printf_parse("parse_stream return 1: <(process) not supported\n");
4413                                 return 1;
4414                         }
4415 #endif
4416                         setup_redirect(ctx, redir_fd, redir_style, input);
4417                         break;
4418                 case ';':
4419 #if ENABLE_HUSH_CASE
4420  case_semi:
4421 #endif
4422                         done_word(dest, ctx);
4423                         done_pipe(ctx, PIPE_SEQ);
4424 #if ENABLE_HUSH_CASE
4425                         /* Eat multiple semicolons, detect
4426                          * whether it means something special */
4427                         while (1) {
4428                                 ch = i_peek(input);
4429                                 if (ch != ';')
4430                                         break;
4431                                 i_getch(input);
4432                                 if (ctx->ctx_res_w == RES_CASEI) {
4433                                         ctx->ctx_dsemicolon = 1;
4434                                         ctx->ctx_res_w = RES_MATCH;
4435                                         break;
4436                                 }
4437                         }
4438 #endif
4439  new_cmd:
4440                         /* We just finished a cmd. New one may start
4441                          * with an assignment */
4442                         dest->o_assignment = MAYBE_ASSIGNMENT;
4443                         break;
4444                 case '&':
4445                         done_word(dest, ctx);
4446                         if (next == '&') {
4447                                 i_getch(input);
4448                                 done_pipe(ctx, PIPE_AND);
4449                         } else {
4450                                 done_pipe(ctx, PIPE_BG);
4451                         }
4452                         goto new_cmd;
4453                 case '|':
4454                         done_word(dest, ctx);
4455 #if ENABLE_HUSH_CASE
4456                         if (ctx->ctx_res_w == RES_MATCH)
4457                                 break; /* we are in case's "word | word)" */
4458 #endif
4459                         if (next == '|') { /* || */
4460                                 i_getch(input);
4461                                 done_pipe(ctx, PIPE_OR);
4462                         } else {
4463                                 /* we could pick up a file descriptor choice here
4464                                  * with redirect_opt_num(), but bash doesn't do it.
4465                                  * "echo foo 2| cat" yields "foo 2". */
4466                                 done_command(ctx);
4467                         }
4468                         goto new_cmd;
4469                 case '(':
4470 #if ENABLE_HUSH_CASE
4471                         /* "case... in [(]word)..." - skip '(' */
4472                         if (ctx->ctx_res_w == RES_MATCH
4473                          && ctx->command->argv == NULL /* not (word|(... */
4474                          && dest->length == 0 /* not word(... */
4475                          && dest->nonnull == 0 /* not ""(... */
4476                         ) {
4477                                 continue;
4478                         }
4479 #endif
4480 #if ENABLE_HUSH_FUNCTIONS
4481                         if (dest->length != 0 /* not just () but word() */
4482                          && dest->nonnull == 0 /* not a"b"c() */
4483                          && ctx->command->argv == NULL /* it's the first word */
4484 //TODO: "func ( ) {...}" - note spaces - is valid format too in bash
4485                          && i_peek(input) == ')'
4486                          && !match_reserved_word(dest)
4487                         ) {
4488                                 bb_error_msg("seems like a function definition");
4489                                 i_getch(input);
4490                                 do {
4491 //TODO: do it properly.
4492                                         ch = i_getch(input);
4493                                 } while (ch == ' ' || ch == '\n');
4494                                 if (ch != '{') {
4495                                         syntax("was expecting {");
4496                                         debug_printf_parse("parse_stream return 1\n");
4497                                         return 1;
4498                                 }
4499                                 ch = 'F'; /* magic value */
4500                         }
4501 #endif
4502                 case '{':
4503                         if (parse_group(dest, ctx, input, ch) != 0) {
4504                                 debug_printf_parse("parse_stream return 1: parse_group returned non-0\n");
4505                                 return 1;
4506                         }
4507                         goto new_cmd;
4508                 case ')':
4509 #if ENABLE_HUSH_CASE
4510                         if (ctx->ctx_res_w == RES_MATCH)
4511                                 goto case_semi;
4512 #endif
4513                 case '}':
4514                         /* proper use of this character is caught by end_trigger:
4515                          * if we see {, we call parse_group(..., end_trigger='}')
4516                          * and it will match } earlier (not here). */
4517                         syntax("unexpected } or )");
4518                         debug_printf_parse("parse_stream return 1: unexpected '}'\n");
4519                         return 1;
4520                 default:
4521                         if (HUSH_DEBUG)
4522                                 bb_error_msg_and_die("BUG: unexpected %c\n", ch);
4523                 }
4524         } /* while (1) */
4525         debug_printf_parse("parse_stream return %d\n", -(end_trigger != NULL));
4526         if (end_trigger)
4527                 return -1;
4528         return 0;
4529 }
4530
4531 static void set_in_charmap(const char *set, int code)
4532 {
4533         while (*set)
4534                 G.charmap[(unsigned char)*set++] = code;
4535 }
4536
4537 static void update_charmap(void)
4538 {
4539         G.ifs = getenv("IFS");
4540         if (G.ifs == NULL)
4541                 G.ifs = " \t\n";
4542         /* Precompute a list of 'flow through' behavior so it can be treated
4543          * quickly up front.  Computation is necessary because of IFS.
4544          * Special case handling of IFS == " \t\n" is not implemented.
4545          * The charmap[] array only really needs two bits each,
4546          * and on most machines that would be faster (reduced L1 cache use).
4547          */
4548         memset(G.charmap, CHAR_ORDINARY, sizeof(G.charmap));
4549 #if ENABLE_HUSH_TICK
4550         set_in_charmap("\\$\"`", CHAR_SPECIAL);
4551 #else
4552         set_in_charmap("\\$\"", CHAR_SPECIAL);
4553 #endif
4554         set_in_charmap("<>;&|(){}#'", CHAR_ORDINARY_IF_QUOTED);
4555         set_in_charmap(G.ifs, CHAR_IFS);  /* are ordinary if quoted */
4556 }
4557
4558 /* Most recursion does not come through here, the exception is
4559  * from builtin_source() and builtin_eval() */
4560 static int parse_and_run_stream(struct in_str *inp, int parse_flag)
4561 {
4562         struct parse_context ctx;
4563         o_string temp = NULL_O_STRING;
4564         int rcode;
4565
4566         do {
4567                 initialize_context(&ctx);
4568                 update_charmap();
4569 #if ENABLE_HUSH_INTERACTIVE
4570                 inp->promptmode = 0; /* PS1 */
4571 #endif
4572                 /* We will stop & execute after each ';' or '\n'.
4573                  * Example: "sleep 9999; echo TEST" + ctrl-C:
4574                  * TEST should be printed */
4575                 temp.o_assignment = MAYBE_ASSIGNMENT;
4576                 rcode = parse_stream(&temp, &ctx, inp, ";\n");
4577 #if HAS_KEYWORDS
4578                 if (rcode != 1 && ctx.old_flag != 0) {
4579                         syntax(NULL);
4580                 }
4581 #endif
4582                 if (rcode != 1 IF_HAS_KEYWORDS(&& ctx.old_flag == 0)) {
4583                         done_word(&temp, &ctx);
4584                         done_pipe(&ctx, PIPE_SEQ);
4585                         debug_print_tree(ctx.list_head, 0);
4586                         debug_printf_exec("parse_stream_outer: run_and_free_list\n");
4587                         run_and_free_list(ctx.list_head);
4588                 } else {
4589                         /* We arrive here also if rcode == 1 (error in parse_stream) */
4590 #if HAS_KEYWORDS
4591                         if (ctx.old_flag != 0) {
4592                                 free(ctx.stack);
4593                                 o_reset(&temp);
4594                         }
4595 #endif
4596                         /*temp.nonnull = 0; - o_free does it below */
4597                         /*temp.o_escape = 0; - o_free does it below */
4598                         free_pipe_list(ctx.list_head, /* indent: */ 0);
4599                         /* Discard all unprocessed line input, force prompt on */
4600                         inp->p = NULL;
4601 #if ENABLE_HUSH_INTERACTIVE
4602                         inp->promptme = 1;
4603 #endif
4604                 }
4605                 o_free(&temp);
4606                 /* loop on syntax errors, return on EOF: */
4607         } while (rcode != -1 && !(parse_flag & PARSEFLAG_EXIT_FROM_LOOP));
4608         return 0;
4609 }
4610
4611 static int parse_and_run_string(const char *s, int parse_flag)
4612 {
4613         struct in_str input;
4614         setup_string_in_str(&input, s);
4615         return parse_and_run_stream(&input, parse_flag);
4616 }
4617
4618 static int parse_and_run_file(FILE *f)
4619 {
4620         int rcode;
4621         struct in_str input;
4622         setup_file_in_str(&input, f);
4623         rcode = parse_and_run_stream(&input, 0 /* parse_flag */);
4624         return rcode;
4625 }
4626
4627 #if ENABLE_HUSH_JOB
4628 /* Make sure we have a controlling tty.  If we get started under a job
4629  * aware app (like bash for example), make sure we are now in charge so
4630  * we don't fight over who gets the foreground */
4631 static void setup_job_control(void)
4632 {
4633         pid_t shell_pgrp;
4634
4635         shell_pgrp = getpgrp();
4636
4637         /* If we were ran as 'hush &',
4638          * sleep until we are in the foreground.  */
4639         while (tcgetpgrp(G.interactive_fd) != shell_pgrp) {
4640                 /* Send TTIN to ourself (should stop us) */
4641                 kill(- shell_pgrp, SIGTTIN);
4642                 shell_pgrp = getpgrp();
4643         }
4644
4645         /* We _must_ restore tty pgrp on fatal signals */
4646         set_fatal_signals_to_sigexit();
4647
4648         /* Put ourselves in our own process group.  */
4649         bb_setpgrp(); /* is the same as setpgid(our_pid, our_pid); */
4650         /* Grab control of the terminal.  */
4651         tcsetpgrp(G.interactive_fd, getpid());
4652 }
4653 #endif
4654
4655 static int set_mode(const char cstate, const char mode)
4656 {
4657         int state = (cstate == '-' ? 1 : 0);
4658         switch (mode) {
4659                 case 'n': G.fake_mode = state; break;
4660                 case 'x': /*G.debug_mode = state;*/ break;
4661                 default:  return EXIT_FAILURE;
4662         }
4663         return EXIT_SUCCESS;
4664 }
4665
4666 int hush_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
4667 int hush_main(int argc, char **argv)
4668 {
4669         static const struct variable const_shell_ver = {
4670                 .next = NULL,
4671                 .varstr = (char*)hush_version_str,
4672                 .max_len = 1, /* 0 can provoke free(name) */
4673                 .flg_export = 1,
4674                 .flg_read_only = 1,
4675         };
4676
4677         int opt;
4678         FILE *input;
4679         char **e;
4680         struct variable *cur_var;
4681
4682         INIT_G();
4683
4684         G.root_pid = getpid();
4685
4686         /* Deal with HUSH_VERSION */
4687         G.shell_ver = const_shell_ver; /* copying struct here */
4688         G.top_var = &G.shell_ver;
4689         debug_printf_env("unsetenv '%s'\n", "HUSH_VERSION");
4690         unsetenv("HUSH_VERSION"); /* in case it exists in initial env */
4691         /* Initialize our shell local variables with the values
4692          * currently living in the environment */
4693         cur_var = G.top_var;
4694         e = environ;
4695         if (e) while (*e) {
4696                 char *value = strchr(*e, '=');
4697                 if (value) { /* paranoia */
4698                         cur_var->next = xzalloc(sizeof(*cur_var));
4699                         cur_var = cur_var->next;
4700                         cur_var->varstr = *e;
4701                         cur_var->max_len = strlen(*e);
4702                         cur_var->flg_export = 1;
4703                 }
4704                 e++;
4705         }
4706         debug_printf_env("putenv '%s'\n", hush_version_str);
4707         putenv((char *)hush_version_str); /* reinstate HUSH_VERSION */
4708
4709 #if ENABLE_FEATURE_EDITING
4710         G.line_input_state = new_line_input_t(FOR_SHELL);
4711 #endif
4712         /* XXX what should these be while sourcing /etc/profile? */
4713         G.global_argc = argc;
4714         G.global_argv = argv;
4715         /* Initialize some more globals to non-zero values */
4716         set_cwd();
4717 #if ENABLE_HUSH_INTERACTIVE
4718         if (ENABLE_FEATURE_EDITING)
4719                 cmdedit_set_initial_prompt();
4720         G.PS2 = "> ";
4721 #endif
4722
4723         if (EXIT_SUCCESS) /* otherwise is already done */
4724                 G.last_return_code = EXIT_SUCCESS;
4725
4726         if (argv[0] && argv[0][0] == '-') {
4727                 debug_printf("sourcing /etc/profile\n");
4728                 input = fopen_for_read("/etc/profile");
4729                 if (input != NULL) {
4730                         close_on_exec_on(fileno(input));
4731                         parse_and_run_file(input);
4732                         fclose(input);
4733                 }
4734         }
4735         input = stdin;
4736
4737         /* http://www.opengroup.org/onlinepubs/9699919799/utilities/sh.html */
4738         while ((opt = getopt(argc, argv, "c:xins")) > 0) {
4739                 switch (opt) {
4740                 case 'c':
4741                         G.global_argv = argv + optind;
4742                         if (!argv[optind]) {
4743                                 /* -c 'script' (no params): prevent empty $0 */
4744                                 *--G.global_argv = argv[0];
4745                                 optind--;
4746                         } /* else -c 'script' PAR0 PAR1: $0 is PAR0 */
4747                         G.global_argc = argc - optind;
4748                         opt = parse_and_run_string(optarg, 0 /* parse_flag */);
4749                         goto final_return;
4750                 case 'i':
4751                         /* Well, we cannot just declare interactiveness,
4752                          * we have to have some stuff (ctty, etc) */
4753                         /* G.interactive_fd++; */
4754                         break;
4755                 case 's':
4756                         /* "-s" means "read from stdin", but this is how we always
4757                          * operate, so simply do nothing here. */
4758                         break;
4759                 case 'n':
4760                 case 'x':
4761                         if (!set_mode('-', opt))
4762                                 break;
4763                 default:
4764 #ifndef BB_VER
4765                         fprintf(stderr, "Usage: sh [FILE]...\n"
4766                                         "   or: sh -c command [args]...\n\n");
4767                         exit(EXIT_FAILURE);
4768 #else
4769                         bb_show_usage();
4770 #endif
4771                 }
4772         }
4773 #if ENABLE_HUSH_JOB
4774         /* A shell is interactive if the '-i' flag was given, or if all of
4775          * the following conditions are met:
4776          *    no -c command
4777          *    no arguments remaining or the -s flag given
4778          *    standard input is a terminal
4779          *    standard output is a terminal
4780          *    Refer to Posix.2, the description of the 'sh' utility. */
4781         if (argv[optind] == NULL && input == stdin
4782          && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)
4783         ) {
4784                 G.saved_tty_pgrp = tcgetpgrp(STDIN_FILENO);
4785                 debug_printf("saved_tty_pgrp=%d\n", G.saved_tty_pgrp);
4786                 if (G.saved_tty_pgrp >= 0) {
4787                         /* try to dup to high fd#, >= 255 */
4788                         G.interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
4789                         if (G.interactive_fd < 0) {
4790                                 /* try to dup to any fd */
4791                                 G.interactive_fd = dup(STDIN_FILENO);
4792                                 if (G.interactive_fd < 0)
4793                                         /* give up */
4794                                         G.interactive_fd = 0;
4795                         }
4796                         // TODO: track & disallow any attempts of user
4797                         // to (inadvertently) close/redirect it
4798                 }
4799         }
4800         init_signal_mask(); /* note: ensures SIGCHLD is not masked */
4801         debug_printf("G.interactive_fd=%d\n", G.interactive_fd);
4802         if (G.interactive_fd) {
4803                 fcntl(G.interactive_fd, F_SETFD, FD_CLOEXEC);
4804                 /* Looks like they want an interactive shell */
4805                 setup_job_control();
4806                 /* -1 is special - makes xfuncs longjmp, not exit
4807                  * (we reset die_sleep = 0 whereever we [v]fork) */
4808                 die_sleep = -1;
4809                 if (setjmp(die_jmp)) {
4810                         /* xfunc has failed! die die die */
4811                         hush_exit(xfunc_error_retval);
4812                 }
4813         }
4814 #elif ENABLE_HUSH_INTERACTIVE
4815 /* no job control compiled, only prompt/line editing */
4816         if (argv[optind] == NULL && input == stdin
4817          && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)
4818         ) {
4819                 G.interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
4820                 if (G.interactive_fd < 0) {
4821                         /* try to dup to any fd */
4822                         G.interactive_fd = dup(STDIN_FILENO);
4823                         if (G.interactive_fd < 0)
4824                                 /* give up */
4825                                 G.interactive_fd = 0;
4826                 }
4827                 if (G.interactive_fd) {
4828                         fcntl(G.interactive_fd, F_SETFD, FD_CLOEXEC);
4829                 }
4830         }
4831         init_signal_mask(); /* note: ensures SIGCHLD is not masked */
4832 #endif
4833         /* POSIX allows shell to re-enable SIGCHLD
4834          * even if it was SIG_IGN on entry */
4835 //      G.count_SIGCHLD++; /* ensure it is != G.handled_SIGCHLD */
4836         signal(SIGCHLD, SIG_DFL); // SIGCHLD_handler);
4837
4838 #if ENABLE_HUSH_INTERACTIVE && !ENABLE_FEATURE_SH_EXTRA_QUIET
4839         if (G.interactive_fd) {
4840                 printf("\n\n%s hush - the humble shell v"HUSH_VER_STR"\n", bb_banner);
4841                 printf("Enter 'help' for a list of built-in commands.\n\n");
4842         }
4843 #endif
4844
4845         if (argv[optind] == NULL) {
4846                 opt = parse_and_run_file(stdin);
4847         } else {
4848                 debug_printf("\nrunning script '%s'\n", argv[optind]);
4849                 G.global_argv = argv + optind;
4850                 G.global_argc = argc - optind;
4851                 input = xfopen_for_read(argv[optind]);
4852                 fcntl(fileno(input), F_SETFD, FD_CLOEXEC);
4853                 opt = parse_and_run_file(input);
4854         }
4855
4856  final_return:
4857
4858 #if ENABLE_FEATURE_CLEAN_UP
4859         fclose(input);
4860         if (G.cwd != bb_msg_unknown)
4861                 free((char*)G.cwd);
4862         cur_var = G.top_var->next;
4863         while (cur_var) {
4864                 struct variable *tmp = cur_var;
4865                 if (!cur_var->max_len)
4866                         free(cur_var->varstr);
4867                 cur_var = cur_var->next;
4868                 free(tmp);
4869         }
4870 #endif
4871         hush_exit(opt ? opt : G.last_return_code);
4872 }
4873
4874
4875 #if ENABLE_LASH
4876 int lash_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
4877 int lash_main(int argc, char **argv)
4878 {
4879         //bb_error_msg("lash is deprecated, please use hush instead");
4880         return hush_main(argc, argv);
4881 }
4882 #endif
4883
4884
4885 /*
4886  * Built-ins
4887  */
4888 static int builtin_trap(char **argv)
4889 {
4890         int i;
4891         int sig;
4892         char *new_cmd;
4893
4894         if (!G.traps)
4895                 G.traps = xzalloc(sizeof(G.traps[0]) * NSIG);
4896
4897         if (!argv[1]) {
4898                 /* No args: print all trapped.  This isn't 100% correct as we should
4899                  * be escaping the cmd so that it can be pasted back in ...
4900                  */
4901                 for (i = 0; i < NSIG; ++i)
4902                         if (G.traps[i])
4903                                 printf("trap -- '%s' %s\n", G.traps[i], get_signame(i));
4904                 return EXIT_SUCCESS;
4905         }
4906
4907         new_cmd = NULL;
4908         i = 0;
4909         /* if first arg is decimal: reset all specified */
4910         sig = bb_strtou(*++argv, NULL, 10);
4911         if (errno == 0) {
4912                 int ret;
4913  set_all:
4914                 ret = EXIT_SUCCESS;
4915                 while (*argv) {
4916                         sig = get_signum(*argv++);
4917                         if (sig < 0 || sig >= NSIG) {
4918                                 ret = EXIT_FAILURE;
4919                                 /* mimic bash message exactly */
4920                                 bb_perror_msg("trap: %s: invalid signal specification", argv[i]);
4921                                 continue;
4922                         }
4923
4924                         free(G.traps[sig]);
4925                         G.traps[sig] = xstrdup(new_cmd);
4926
4927                         debug_printf("trap: setting SIG%s (%i) to '%s'",
4928                                 get_signame(sig), sig, G.traps[sig]);
4929
4930                         /* There is no signal for 0 (EXIT) */
4931                         if (sig == 0)
4932                                 continue;
4933
4934                         if (new_cmd) {
4935                                 sigaddset(&G.blocked_set, sig);
4936                         } else {
4937                                 /* there was a trap handler, we are removing it
4938                                  * (if sig has non-DFL handling,
4939                                  * we don't need to do anything) */
4940                                 if (sig < 32 && (G.non_DFL_mask & (1 << sig)))
4941                                         continue;
4942                                 sigdelset(&G.blocked_set, sig);
4943                         }
4944                         sigprocmask(SIG_SETMASK, &G.blocked_set, NULL);
4945                 }
4946                 return ret;
4947         }
4948
4949         /* first arg is "-": reset all specified to default */
4950         /* first arg is "": ignore all specified */
4951         /* everything else: execute first arg upon signal */
4952         if (!argv[1]) {
4953                 bb_error_msg("trap: invalid arguments");
4954                 return EXIT_FAILURE;
4955         }
4956         if (LONE_DASH(*argv))
4957                 /* nothing! */;
4958         else
4959                 new_cmd = *argv;
4960         argv++;
4961         goto set_all;
4962 }
4963
4964 static int builtin_true(char **argv UNUSED_PARAM)
4965 {
4966         return 0;
4967 }
4968
4969 static int builtin_test(char **argv)
4970 {
4971         int argc = 0;
4972         while (*argv) {
4973                 argc++;
4974                 argv++;
4975         }
4976         return test_main(argc, argv - argc);
4977 }
4978
4979 static int builtin_echo(char **argv)
4980 {
4981         int argc = 0;
4982         while (*argv) {
4983                 argc++;
4984                 argv++;
4985         }
4986         return echo_main(argc, argv - argc);
4987 }
4988
4989 static int builtin_eval(char **argv)
4990 {
4991         int rcode = EXIT_SUCCESS;
4992
4993         if (argv[1]) {
4994                 char *str = expand_strvec_to_string(argv + 1);
4995                 parse_and_run_string(str, PARSEFLAG_EXIT_FROM_LOOP);
4996                 free(str);
4997                 rcode = G.last_return_code;
4998         }
4999         return rcode;
5000 }
5001
5002 static int builtin_cd(char **argv)
5003 {
5004         const char *newdir;
5005         if (argv[1] == NULL) {
5006                 // bash does nothing (exitcode 0) if HOME is ""; if it's unset,
5007                 // bash says "bash: cd: HOME not set" and does nothing (exitcode 1)
5008                 newdir = getenv("HOME") ? : "/";
5009         } else
5010                 newdir = argv[1];
5011         if (chdir(newdir)) {
5012                 printf("cd: %s: %s\n", newdir, strerror(errno));
5013                 return EXIT_FAILURE;
5014         }
5015         set_cwd();
5016         return EXIT_SUCCESS;
5017 }
5018
5019 static int builtin_exec(char **argv)
5020 {
5021         if (argv[1] == NULL)
5022                 return EXIT_SUCCESS; /* bash does this */
5023         {
5024 #if !BB_MMU
5025                 nommu_save_t dummy;
5026 #endif
5027 // FIXME: if exec fails, bash does NOT exit! We do...
5028                 pseudo_exec_argv(&dummy, argv + 1, 0, NULL);
5029                 /* never returns */
5030         }
5031 }
5032
5033 static int builtin_exit(char **argv)
5034 {
5035 // TODO: bash does it ONLY on top-level sh exit (+interacive only?)
5036         //puts("exit"); /* bash does it */
5037 // TODO: warn if we have background jobs: "There are stopped jobs"
5038 // On second consecutive 'exit', exit anyway.
5039         if (argv[1] == NULL)
5040                 hush_exit(G.last_return_code);
5041         /* mimic bash: exit 123abc == exit 255 + error msg */
5042         xfunc_error_retval = 255;
5043         /* bash: exit -2 == exit 254, no error msg */
5044         hush_exit(xatoi(argv[1]) & 0xff);
5045 }
5046
5047 static int builtin_export(char **argv)
5048 {
5049         const char *value;
5050         char *name = argv[1];
5051
5052         if (name == NULL) {
5053                 // TODO:
5054                 // ash emits: export VAR='VAL'
5055                 // bash: declare -x VAR="VAL"
5056                 // (both also escape as needed (quotes, $, etc))
5057                 char **e = environ;
5058                 if (e)
5059                         while (*e)
5060                                 puts(*e++);
5061                 return EXIT_SUCCESS;
5062         }
5063
5064         value = strchr(name, '=');
5065         if (!value) {
5066                 /* They are exporting something without a =VALUE */
5067                 struct variable *var;
5068
5069                 var = get_local_var(name);
5070                 if (var) {
5071                         var->flg_export = 1;
5072                         debug_printf_env("%s: putenv '%s'\n", __func__, var->varstr);
5073                         putenv(var->varstr);
5074                 }
5075                 /* bash does not return an error when trying to export
5076                  * an undefined variable.  Do likewise. */
5077                 return EXIT_SUCCESS;
5078         }
5079
5080         set_local_var(xstrdup(name), 1);
5081         return EXIT_SUCCESS;
5082 }
5083
5084 #if ENABLE_HUSH_JOB
5085 /* built-in 'fg' and 'bg' handler */
5086 static int builtin_fg_bg(char **argv)
5087 {
5088         int i, jobnum;
5089         struct pipe *pi;
5090
5091         if (!G.interactive_fd)
5092                 return EXIT_FAILURE;
5093         /* If they gave us no args, assume they want the last backgrounded task */
5094         if (!argv[1]) {
5095                 for (pi = G.job_list; pi; pi = pi->next) {
5096                         if (pi->jobid == G.last_jobid) {
5097                                 goto found;
5098                         }
5099                 }
5100                 bb_error_msg("%s: no current job", argv[0]);
5101                 return EXIT_FAILURE;
5102         }
5103         if (sscanf(argv[1], "%%%d", &jobnum) != 1) {
5104                 bb_error_msg("%s: bad argument '%s'", argv[0], argv[1]);
5105                 return EXIT_FAILURE;
5106         }
5107         for (pi = G.job_list; pi; pi = pi->next) {
5108                 if (pi->jobid == jobnum) {
5109                         goto found;
5110                 }
5111         }
5112         bb_error_msg("%s: %d: no such job", argv[0], jobnum);
5113         return EXIT_FAILURE;
5114  found:
5115         // TODO: bash prints a string representation
5116         // of job being foregrounded (like "sleep 1 | cat")
5117         if (*argv[0] == 'f') {
5118                 /* Put the job into the foreground.  */
5119                 tcsetpgrp(G.interactive_fd, pi->pgrp);
5120         }
5121
5122         /* Restart the processes in the job */
5123         debug_printf_jobs("reviving %d procs, pgrp %d\n", pi->num_cmds, pi->pgrp);
5124         for (i = 0; i < pi->num_cmds; i++) {
5125                 debug_printf_jobs("reviving pid %d\n", pi->cmds[i].pid);
5126                 pi->cmds[i].is_stopped = 0;
5127         }
5128         pi->stopped_cmds = 0;
5129
5130         i = kill(- pi->pgrp, SIGCONT);
5131         if (i < 0) {
5132                 if (errno == ESRCH) {
5133                         delete_finished_bg_job(pi);
5134                         return EXIT_SUCCESS;
5135                 } else {
5136                         bb_perror_msg("kill (SIGCONT)");
5137                 }
5138         }
5139
5140         if (*argv[0] == 'f') {
5141                 remove_bg_job(pi);
5142                 return checkjobs_and_fg_shell(pi);
5143         }
5144         return EXIT_SUCCESS;
5145 }
5146 #endif
5147
5148 #if ENABLE_HUSH_HELP
5149 static int builtin_help(char **argv UNUSED_PARAM)
5150 {
5151         const struct built_in_command *x;
5152
5153         printf("\nBuilt-in commands:\n");
5154         printf("-------------------\n");
5155         for (x = bltins; x != &bltins[ARRAY_SIZE(bltins)]; x++) {
5156                 printf("%s\t%s\n", x->cmd, x->descr);
5157         }
5158         printf("\n\n");
5159         return EXIT_SUCCESS;
5160 }
5161 #endif
5162
5163 #if ENABLE_HUSH_JOB
5164 static int builtin_jobs(char **argv UNUSED_PARAM)
5165 {
5166         struct pipe *job;
5167         const char *status_string;
5168
5169         for (job = G.job_list; job; job = job->next) {
5170                 if (job->alive_cmds == job->stopped_cmds)
5171                         status_string = "Stopped";
5172                 else
5173                         status_string = "Running";
5174
5175                 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->cmdtext);
5176         }
5177         return EXIT_SUCCESS;
5178 }
5179 #endif
5180
5181 static int builtin_pwd(char **argv UNUSED_PARAM)
5182 {
5183         puts(set_cwd());
5184         return EXIT_SUCCESS;
5185 }
5186
5187 static int builtin_read(char **argv)
5188 {
5189         char *string;
5190         const char *name = argv[1] ? argv[1] : "REPLY";
5191
5192         string = xmalloc_reads(STDIN_FILENO, xasprintf("%s=", name), NULL);
5193         return set_local_var(string, 0);
5194 }
5195
5196 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#set
5197  * built-in 'set' handler
5198  * SUSv3 says:
5199  * set [-abCefhmnuvx] [-o option] [argument...]
5200  * set [+abCefhmnuvx] [+o option] [argument...]
5201  * set -- [argument...]
5202  * set -o
5203  * set +o
5204  * Implementations shall support the options in both their hyphen and
5205  * plus-sign forms. These options can also be specified as options to sh.
5206  * Examples:
5207  * Write out all variables and their values: set
5208  * Set $1, $2, and $3 and set "$#" to 3: set c a b
5209  * Turn on the -x and -v options: set -xv
5210  * Unset all positional parameters: set --
5211  * Set $1 to the value of x, even if it begins with '-' or '+': set -- "$x"
5212  * Set the positional parameters to the expansion of x, even if x expands
5213  * with a leading '-' or '+': set -- $x
5214  *
5215  * So far, we only support "set -- [argument...]" and some of the short names.
5216  */
5217 static int builtin_set(char **argv)
5218 {
5219         int n;
5220         char **pp, **g_argv;
5221         char *arg = *++argv;
5222
5223         if (arg == NULL) {
5224                 struct variable *e;
5225                 for (e = G.top_var; e; e = e->next)
5226                         puts(e->varstr);
5227                 return EXIT_SUCCESS;
5228         }
5229
5230         do {
5231                 if (!strcmp(arg, "--")) {
5232                         ++argv;
5233                         goto set_argv;
5234                 }
5235
5236                 if (arg[0] == '+' || arg[0] == '-') {
5237                         for (n = 1; arg[n]; ++n)
5238                                 if (set_mode(arg[0], arg[n]))
5239                                         goto error;
5240                         continue;
5241                 }
5242
5243                 break;
5244         } while ((arg = *++argv) != NULL);
5245         /* Now argv[0] is 1st argument */
5246
5247         /* Only reset global_argv if we didn't process anything */
5248         if (arg == NULL)
5249                 return EXIT_SUCCESS;
5250  set_argv:
5251
5252         /* NB: G.global_argv[0] ($0) is never freed/changed */
5253         g_argv = G.global_argv;
5254         if (G.global_args_malloced) {
5255                 pp = g_argv;
5256                 while (*++pp)
5257                         free(*pp);
5258                 g_argv[1] = NULL;
5259         } else {
5260                 G.global_args_malloced = 1;
5261                 pp = xzalloc(sizeof(pp[0]) * 2);
5262                 pp[0] = g_argv[0]; /* retain $0 */
5263                 g_argv = pp;
5264         }
5265         /* This realloc's G.global_argv */
5266         G.global_argv = pp = add_strings_to_strings(g_argv, argv, /*dup:*/ 1);
5267
5268         n = 1;
5269         while (*++pp)
5270                 n++;
5271         G.global_argc = n;
5272
5273         return EXIT_SUCCESS;
5274
5275         /* Nothing known, so abort */
5276  error:
5277         bb_error_msg("set: %s: invalid option", arg);
5278         return EXIT_FAILURE;
5279 }
5280
5281 static int builtin_shift(char **argv)
5282 {
5283         int n = 1;
5284         if (argv[1]) {
5285                 n = atoi(argv[1]);
5286         }
5287         if (n >= 0 && n < G.global_argc) {
5288                 if (G.global_args_malloced) {
5289                         int m = 1;
5290                         while (m <= n)
5291                                 free(G.global_argv[m++]);
5292                 }
5293                 G.global_argc -= n;
5294                 memmove(&G.global_argv[1], &G.global_argv[n+1],
5295                                 G.global_argc * sizeof(G.global_argv[0]));
5296                 return EXIT_SUCCESS;
5297         }
5298         return EXIT_FAILURE;
5299 }
5300
5301 static int builtin_source(char **argv)
5302 {
5303         FILE *input;
5304         int status;
5305
5306         if (argv[1] == NULL)
5307                 return EXIT_FAILURE;
5308
5309         /* XXX search through $PATH is missing */
5310         input = fopen_for_read(argv[1]);
5311         if (!input) {
5312                 bb_error_msg("can't open '%s'", argv[1]);
5313                 return EXIT_FAILURE;
5314         }
5315         close_on_exec_on(fileno(input));
5316
5317         /* Now run the file */
5318         /* XXX argv and argc are broken; need to save old G.global_argv
5319          * (pointer only is OK!) on this stack frame,
5320          * set G.global_argv=argv+1, recurse, and restore. */
5321         status = parse_and_run_file(input);
5322         fclose(input);
5323         return status;
5324 }
5325
5326 static int builtin_umask(char **argv)
5327 {
5328         mode_t new_umask;
5329         const char *arg = argv[1];
5330         if (arg) {
5331                 new_umask = bb_strtou(arg, NULL, 8);
5332                 if (errno)
5333                         return EXIT_FAILURE;
5334         } else {
5335                 new_umask = umask(0);
5336                 printf("%.3o\n", (unsigned) new_umask);
5337         }
5338         umask(new_umask);
5339         return EXIT_SUCCESS;
5340 }
5341
5342 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#unset */
5343 static int builtin_unset(char **argv)
5344 {
5345         size_t i;
5346         int ret;
5347         bool var = true;
5348
5349         if (!argv[1])
5350                 return EXIT_SUCCESS;
5351
5352         i = 0;
5353         if (argv[1][0] == '-') {
5354                 switch (argv[1][1]) {
5355                 case 'v': break;
5356                 case 'f': if (ENABLE_HUSH_FUNCTIONS) { var = false; break; }
5357                 default:
5358                         bb_error_msg("unset: %s: invalid option", argv[1]);
5359                         return EXIT_FAILURE;
5360                 }
5361                 ++i;
5362         }
5363
5364         ret = EXIT_SUCCESS;
5365         while (argv[++i]) {
5366                 if (var) {
5367                         if (unset_local_var(argv[i]))
5368                                 ret = EXIT_FAILURE;
5369                 }
5370 #if ENABLE_HUSH_FUNCTIONS
5371                 else
5372                         unset_local_func(argv[i]);
5373 #endif
5374         }
5375         return ret;
5376 }
5377
5378 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/wait.html */
5379 static int builtin_wait(char **argv)
5380 {
5381         int ret = EXIT_SUCCESS;
5382         int status, sig;
5383
5384         if (*++argv == NULL) {
5385                 /* Don't care about wait results */
5386                 /* Note 1: must wait until there are no more children */
5387                 /* Note 2: must be interruptible */
5388                 /* Examples:
5389                  * $ sleep 3 & sleep 6 & wait
5390                  * [1] 30934 sleep 3
5391                  * [2] 30935 sleep 6
5392                  * [1] Done                   sleep 3
5393                  * [2] Done                   sleep 6
5394                  * $ sleep 3 & sleep 6 & wait
5395                  * [1] 30936 sleep 3
5396                  * [2] 30937 sleep 6
5397                  * [1] Done                   sleep 3
5398                  * ^C <-- after ~4 sec from keyboard
5399                  * $
5400                  */
5401                 sigaddset(&G.blocked_set, SIGCHLD);
5402                 sigprocmask(SIG_SETMASK, &G.blocked_set, NULL);
5403                 while (1) {
5404                         checkjobs(NULL);
5405                         if (errno == ECHILD)
5406                                 break;
5407                         /* Wait for SIGCHLD or any other signal of interest */
5408                         /* sigtimedwait with infinite timeout: */
5409                         sig = sigwaitinfo(&G.blocked_set, NULL);
5410                         if (sig > 0) {
5411                                 sig = check_and_run_traps(sig);
5412                                 if (sig && sig != SIGCHLD) { /* see note 2 */
5413                                         ret = 128 + sig;
5414                                         break;
5415                                 }
5416                         }
5417                 }
5418                 sigdelset(&G.blocked_set, SIGCHLD);
5419                 sigprocmask(SIG_SETMASK, &G.blocked_set, NULL);
5420                 return ret;
5421         }
5422
5423         /* This is probably buggy wrt interruptible-ness */
5424         while (*argv) {
5425                 pid_t pid = bb_strtou(*argv, NULL, 10);
5426                 if (errno) {
5427                         /* mimic bash message */
5428                         bb_error_msg("wait: '%s': not a pid or valid job spec", *argv);
5429                         return EXIT_FAILURE;
5430                 }
5431                 if (waitpid(pid, &status, 0) == pid) {
5432                         if (WIFSIGNALED(status))
5433                                 ret = 128 + WTERMSIG(status);
5434                         else if (WIFEXITED(status))
5435                                 ret = WEXITSTATUS(status);
5436                         else /* wtf? */
5437                                 ret = EXIT_FAILURE;
5438                 } else {
5439                         bb_perror_msg("wait %s", *argv);
5440                         ret = 127;
5441                 }
5442                 argv++;
5443         }
5444
5445         return ret;
5446 }
5447
5448 #if ENABLE_HUSH_LOOPS
5449 static int builtin_break(char **argv)
5450 {
5451         if (G.depth_of_loop == 0) {
5452                 bb_error_msg("%s: only meaningful in a loop", argv[0]);
5453                 return EXIT_SUCCESS; /* bash compat */
5454         }
5455         G.flag_break_continue++; /* BC_BREAK = 1 */
5456         G.depth_break_continue = 1;
5457         if (argv[1]) {
5458                 G.depth_break_continue = bb_strtou(argv[1], NULL, 10);
5459                 if (errno || !G.depth_break_continue || argv[2]) {
5460                         bb_error_msg("%s: bad arguments", argv[0]);
5461                         G.flag_break_continue = BC_BREAK;
5462                         G.depth_break_continue = UINT_MAX;
5463                 }
5464         }
5465         if (G.depth_of_loop < G.depth_break_continue)
5466                 G.depth_break_continue = G.depth_of_loop;
5467         return EXIT_SUCCESS;
5468 }
5469
5470 static int builtin_continue(char **argv)
5471 {
5472         G.flag_break_continue = 1; /* BC_CONTINUE = 2 = 1+1 */
5473         return builtin_break(argv);
5474 }
5475 #endif