shell/hush.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * sh.c -- 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  *      b_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  * Bash grammar not implemented: (how many of these were in original sh?)
  40  *      $@ (those sure look like weird quoting rules)
  41  *      $_
  42  *      ! negation operator for pipes
  43  *      &> and >& redirection of stdout+stderr
  44  *      Brace Expansion
  45  *      Tilde Expansion
  46  *      fancy forms of Parameter Expansion
  47  *      aliases
  48  *      Arithmetic Expansion
  49  *      <(list) and >(list) Process Substitution
  50  *      reserved words: case, esac, select, function
  51  *      Here Documents ( << word )
  52  *      Functions
  53  * Major bugs:
  54  *      job handling woefully incomplete and buggy (improved --vda)
  55  *      reserved word execution woefully incomplete and buggy
  56  * to-do:
  57  *      port selected bugfixes from post-0.49 busybox lash - done?
  58  *      finish implementing reserved words: for, while, until, do, done
  59  *      change { and } from special chars to reserved words
  60  *      builtins: break, continue, eval, return, set, trap, ulimit
  61  *      test magic exec
  62  *      handle children going into background
  63  *      clean up recognition of null pipes
  64  *      check setting of global_argc and global_argv
  65  *      control-C handling, probably with longjmp
  66  *      follow IFS rules more precisely, including update semantics
  67  *      figure out what to do with backslash-newline
  68  *      explain why we use signal instead of sigaction
  69  *      propagate syntax errors, die on resource errors?
  70  *      continuation lines, both explicit and implicit - done?
  71  *      memory leak finding and plugging - done?
  72  *      more testing, especially quoting rules and redirection
  73  *      document how quoting rules not precisely followed for variable assignments
  74  *      maybe change charmap[] to use 2-bit entries
  75  *      (eventually) remove all the printf's
  76  *
  77  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
  78  */
  79
  80 #include "busybox.h"
  81 #include <glob.h>      /* glob, of course */
  82 #include <getopt.h>    /* should be pretty obvious */
  83 /* #include <dmalloc.h> */
  84 extern char **environ; /* This is in <unistd.h>, but protected with __USE_GNU */
  85
  86
  87 /* If you comment out one of these below, it will be #defined later
  88  * to perform debug printfs to stderr: */
  89 #define debug_printf(...)        do {} while (0)
  90 /* Finer-grained debug switches */
  91 #define debug_printf_parse(...)  do {} while (0)
  92 #define debug_print_tree(a, b)   do {} while (0)
  93 #define debug_printf_exec(...)   do {} while (0)
  94 #define debug_printf_jobs(...)   do {} while (0)
  95 #define debug_printf_expand(...) do {} while (0)
  96 #define debug_printf_clean(...)  do {} while (0)
  97
  98 #ifndef debug_printf
  99 #define debug_printf(...) fprintf(stderr, __VA_ARGS__)
 100 #endif
 101
 102 #ifndef debug_printf_parse
 103 #define debug_printf_parse(...) fprintf(stderr, __VA_ARGS__)
 104 #endif
 105
 106 #ifndef debug_printf_exec
 107 #define debug_printf_exec(...) fprintf(stderr, __VA_ARGS__)
 108 #endif
 109
 110 #ifndef debug_printf_jobs
 111 #define debug_printf_jobs(...) fprintf(stderr, __VA_ARGS__)
 112 #define DEBUG_SHELL_JOBS 1
 113 #endif
 114
 115 #ifndef debug_printf_expand
 116 #define debug_printf_expand(...) fprintf(stderr, __VA_ARGS__)
 117 #define DEBUG_EXPAND 1
 118 #endif
 119
 120 #ifndef debug_printf_clean
 121 /* broken, of course, but OK for testing */
 122 static const char *indenter(int i)
 123 {
 124         static const char blanks[] = "                                    ";
 125         return &blanks[sizeof(blanks) - i - 1];
 126 }
 127 #define debug_printf_clean(...) fprintf(stderr, __VA_ARGS__)
 128 #define DEBUG_CLEAN 1
 129 #endif
 130
 131
 132 #if !ENABLE_HUSH_INTERACTIVE
 133 #undef ENABLE_FEATURE_EDITING
 134 #define ENABLE_FEATURE_EDITING 0
 135 #undef ENABLE_FEATURE_EDITING_FANCY_PROMPT
 136 #define ENABLE_FEATURE_EDITING_FANCY_PROMPT 0
 137 #endif
 138
 139 #define SPECIAL_VAR_SYMBOL   3
 140 #define FLAG_EXIT_FROM_LOOP  1
 141 #define FLAG_PARSE_SEMICOLON (1 << 1)           /* symbol ';' is special for parser */
 142 #define FLAG_REPARSING       (1 << 2)           /* >= 2nd pass */
 143
 144 typedef enum {
 145         REDIRECT_INPUT     = 1,
 146         REDIRECT_OVERWRITE = 2,
 147         REDIRECT_APPEND    = 3,
 148         REDIRECT_HEREIS    = 4,
 149         REDIRECT_IO        = 5
 150 } redir_type;
 151
 152 /* The descrip member of this structure is only used to make debugging
 153  * output pretty */
 154 static const struct {
 155         int mode;
 156         signed char default_fd;
 157         char descrip[3];
 158 } redir_table[] = {
 159         { 0,                         0, "()" },
 160         { O_RDONLY,                  0, "<"  },
 161         { O_CREAT|O_TRUNC|O_WRONLY,  1, ">"  },
 162         { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
 163         { O_RDONLY,                 -1, "<<" },
 164         { O_RDWR,                    1, "<>" }
 165 };
 166
 167 typedef enum {
 168         PIPE_SEQ = 1,
 169         PIPE_AND = 2,
 170         PIPE_OR  = 3,
 171         PIPE_BG  = 4,
 172 } pipe_style;
 173
 174 /* might eventually control execution */
 175 typedef enum {
 176         RES_NONE  = 0,
 177         RES_IF    = 1,
 178         RES_THEN  = 2,
 179         RES_ELIF  = 3,
 180         RES_ELSE  = 4,
 181         RES_FI    = 5,
 182         RES_FOR   = 6,
 183         RES_WHILE = 7,
 184         RES_UNTIL = 8,
 185         RES_DO    = 9,
 186         RES_DONE  = 10,
 187         RES_XXXX  = 11,
 188         RES_IN    = 12,
 189         RES_SNTX  = 13
 190 } reserved_style;
 191 enum {
 192         FLAG_END   = (1 << RES_NONE ),
 193         FLAG_IF    = (1 << RES_IF   ),
 194         FLAG_THEN  = (1 << RES_THEN ),
 195         FLAG_ELIF  = (1 << RES_ELIF ),
 196         FLAG_ELSE  = (1 << RES_ELSE ),
 197         FLAG_FI    = (1 << RES_FI   ),
 198         FLAG_FOR   = (1 << RES_FOR  ),
 199         FLAG_WHILE = (1 << RES_WHILE),
 200         FLAG_UNTIL = (1 << RES_UNTIL),
 201         FLAG_DO    = (1 << RES_DO   ),
 202         FLAG_DONE  = (1 << RES_DONE ),
 203         FLAG_IN    = (1 << RES_IN   ),
 204         FLAG_START = (1 << RES_XXXX ),
 205 };
 206
 207 /* This holds pointers to the various results of parsing */
 208 struct p_context {
 209         struct child_prog *child;
 210         struct pipe *list_head;
 211         struct pipe *pipe;
 212         struct redir_struct *pending_redirect;
 213         reserved_style res_w;
 214         int old_flag;           /* for figuring out valid reserved words */
 215         struct p_context *stack;
 216         int parse_type;         /* define type of parser : ";$" common or special symbol */
 217         /* How about quoting status? */
 218 };
 219
 220 struct redir_struct {
 221         struct redir_struct *next;  /* pointer to the next redirect in the list */
 222         redir_type type;            /* type of redirection */
 223         int fd;                     /* file descriptor being redirected */
 224         int dup;                    /* -1, or file descriptor being duplicated */
 225         glob_t word;                /* *word.gl_pathv is the filename */
 226 };
 227
 228 struct child_prog {
 229         pid_t pid;                  /* 0 if exited */
 230         char **argv;                /* program name and arguments */
 231         struct pipe *group;         /* if non-NULL, first in group or subshell */
 232         int subshell;               /* flag, non-zero if group must be forked */
 233         struct redir_struct *redirects; /* I/O redirections */
 234         glob_t glob_result;         /* result of parameter globbing */
 235         int is_stopped;             /* is the program currently running? */
 236         struct pipe *family;        /* pointer back to the child's parent pipe */
 237         //sp counting seems to be broken... so commented out, grep for '//sp:'
 238         //sp: int sp;               /* number of SPECIAL_VAR_SYMBOL */
 239         int type;
 240 };
 241 /* argv vector may contain variable references (^Cvar^C, ^C0^C etc)
 242  * and on execution these are substituted with their values.
 243  * Substitution can make _several_ words out of one argv[n]!
 244  * Example: argv[0]=='.^C*^C.' here: echo .$*.
 245  */
 246
 247 struct pipe {
 248         struct pipe *next;
 249         int num_progs;              /* total number of programs in job */
 250         int running_progs;          /* number of programs running (not exited) */
 251         char *cmdbuf;               /* buffer various argv's point into */
 252 #if ENABLE_HUSH_JOB
 253         int jobid;                  /* job number */
 254         char *cmdtext;              /* name of job */
 255         pid_t pgrp;                 /* process group ID for the job */
 256 #endif
 257         struct child_prog *progs;   /* array of commands in pipe */
 258         int stopped_progs;          /* number of programs alive, but stopped */
 259         int job_context;            /* bitmask defining current context */
 260         pipe_style followup;        /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
 261         reserved_style r_mode;      /* supports if, for, while, until */
 262 };
 263
 264 struct close_me {
 265         struct close_me *next;
 266         int fd;
 267 };
 268
 269 struct variables {
 270         struct variables *next;
 271         const char *name;
 272         const char *value;
 273         int flg_export;
 274         int flg_read_only;
 275 };
 276
 277 typedef struct {
 278         char *data;
 279         int length;
 280         int maxlen;
 281         int quote;
 282         int nonnull;
 283 } o_string;
 284 #define NULL_O_STRING {NULL,0,0,0,0}
 285 /* used for initialization: o_string foo = NULL_O_STRING; */
 286
 287 /* I can almost use ordinary FILE *.  Is open_memstream() universally
 288  * available?  Where is it documented? */
 289 struct in_str {
 290         const char *p;
 291         /* eof_flag=1: last char in ->p is really an EOF */
 292         char eof_flag; /* meaningless if ->p == NULL */
 293         char peek_buf[2];
 294 #if ENABLE_HUSH_INTERACTIVE
 295         int __promptme;
 296         int promptmode;
 297 #endif
 298         FILE *file;
 299         int (*get) (struct in_str *);
 300         int (*peek) (struct in_str *);
 301 };
 302 #define b_getch(input) ((input)->get(input))
 303 #define b_peek(input) ((input)->peek(input))
 304
 305 enum {
 306         CHAR_ORDINARY           = 0,
 307         CHAR_ORDINARY_IF_QUOTED = 1, /* example: *, # */
 308         CHAR_IFS                = 2, /* treated as ordinary if quoted */
 309         CHAR_SPECIAL            = 3, /* example: $ */
 310 };
 311
 312
 313 /* "Globals" within this file */
 314
 315 #define HUSH_VER_STR "0.02"
 316 static const struct variables const_shell_ver = {
 317         NULL, "HUSH_VERSION", HUSH_VER_STR, 1, 1
 318 };
 319
 320 /* Sorted roughly by size (smaller offsets == smaller code) */
 321 struct globals {
 322 #if ENABLE_HUSH_INTERACTIVE
 323         /* 'interactive_fd' is a fd# open to ctty, if we have one
 324          * _AND_ if we decided to act interactively */
 325         int interactive_fd;
 326         const char *PS1;
 327         const char *PS2;
 328 #endif
 329 #if ENABLE_FEATURE_EDITING
 330         line_input_t *line_input_state;
 331 #endif
 332 #if ENABLE_HUSH_JOB
 333         int run_list_level;
 334         pid_t saved_task_pgrp;
 335         pid_t saved_tty_pgrp;
 336         int last_jobid;
 337         struct pipe *job_list;
 338         struct pipe *toplevel_list;
 339         smallint ctrl_z_flag;
 340 #endif
 341         smallint fake_mode;
 342         /* these three support $?, $#, and $1 */
 343         char **global_argv;
 344         int global_argc;
 345         int last_return_code;
 346         const char *ifs;
 347         struct close_me *close_me_head;
 348         const char *cwd;
 349         unsigned last_bg_pid;
 350         struct variables *top_vars; /* = &shell_ver (both are set in main()) */
 351         struct variables shell_ver; /* = const_shell_ver */
 352 #if ENABLE_FEATURE_SH_STANDALONE
 353         struct nofork_save_area nofork_save;
 354 #endif
 355 #if ENABLE_HUSH_JOB
 356         sigjmp_buf toplevel_jb;
 357 #endif
 358         unsigned char charmap[256];
 359         char user_input_buf[ENABLE_FEATURE_EDITING ? BUFSIZ : 2];
 360 };
 361
 362 #define G (*ptr_to_globals)
 363
 364 #if !ENABLE_HUSH_INTERACTIVE
 365 enum { interactive_fd = 0 };
 366 #endif
 367 #if !ENABLE_HUSH_JOB
 368 enum { run_list_level = 0 };
 369 #endif
 370
 371 #if ENABLE_HUSH_INTERACTIVE
 372 #define interactive_fd   (G.interactive_fd  )
 373 #define PS1              (G.PS1             )
 374 #define PS2              (G.PS2             )
 375 #endif
 376 #if ENABLE_FEATURE_EDITING
 377 #define line_input_state (G.line_input_state)
 378 #endif
 379 #if ENABLE_HUSH_JOB
 380 #define run_list_level   (G.run_list_level  )
 381 #define saved_task_pgrp  (G.saved_task_pgrp )
 382 #define saved_tty_pgrp   (G.saved_tty_pgrp  )
 383 #define last_jobid       (G.last_jobid      )
 384 #define job_list         (G.job_list        )
 385 #define toplevel_list    (G.toplevel_list   )
 386 #define toplevel_jb      (G.toplevel_jb     )
 387 #define ctrl_z_flag      (G.ctrl_z_flag     )
 388 #endif /* JOB */
 389 #define global_argv      (G.global_argv     )
 390 #define global_argc      (G.global_argc     )
 391 #define last_return_code (G.last_return_code)
 392 #define ifs              (G.ifs             )
 393 #define fake_mode        (G.fake_mode       )
 394 #define close_me_head    (G.close_me_head   )
 395 #define cwd              (G.cwd             )
 396 #define last_bg_pid      (G.last_bg_pid     )
 397 #define top_vars         (G.top_vars        )
 398 #define shell_ver        (G.shell_ver       )
 399 #if ENABLE_FEATURE_SH_STANDALONE
 400 #define nofork_save      (G.nofork_save     )
 401 #endif
 402 #if ENABLE_HUSH_JOB
 403 #define toplevel_jb      (G.toplevel_jb     )
 404 #endif
 405 #define charmap          (G.charmap         )
 406 #define user_input_buf   (G.user_input_buf  )
 407
 408
 409 #define B_CHUNK  100
 410 #define B_NOSPAC 1
 411 #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
 412
 413 static void __syntax(int line)
 414 {
 415         bb_error_msg("syntax error hush.c:%d", line);
 416 }
 417 #define syntax() __syntax(__LINE__)
 418
 419 /* Index of subroutines: */
 420 /*   function prototypes for builtins */
 421 static int builtin_cd(char **argv);
 422 static int builtin_eval(char **argv);
 423 static int builtin_exec(char **argv);
 424 static int builtin_exit(char **argv);
 425 static int builtin_export(char **argv);
 426 #if ENABLE_HUSH_JOB
 427 static int builtin_fg_bg(char **argv);
 428 static int builtin_jobs(char **argv);
 429 #endif
 430 static int builtin_help(char **argv);
 431 static int builtin_pwd(char **argv);
 432 static int builtin_read(char **argv);
 433 static int builtin_set(char **argv);
 434 static int builtin_shift(char **argv);
 435 static int builtin_source(char **argv);
 436 static int builtin_umask(char **argv);
 437 static int builtin_unset(char **argv);
 438 static int builtin_not_written(char **argv);
 439 /*   o_string manipulation: */
 440 static int b_check_space(o_string *o, int len);
 441 static int b_addchr(o_string *o, int ch);
 442 static void b_reset(o_string *o);
 443 static int b_addqchr(o_string *o, int ch, int quote);
 444 /*  in_str manipulations: */
 445 static int static_get(struct in_str *i);
 446 static int static_peek(struct in_str *i);
 447 static int file_get(struct in_str *i);
 448 static int file_peek(struct in_str *i);
 449 static void setup_file_in_str(struct in_str *i, FILE *f);
 450 static void setup_string_in_str(struct in_str *i, const char *s);
 451 /*  close_me manipulations: */
 452 static void mark_open(int fd);
 453 static void mark_closed(int fd);
 454 static void close_all(void);
 455 /*  "run" the final data structures: */
 456 #if !defined(DEBUG_CLEAN)
 457 #define free_pipe_list(head, indent) free_pipe_list(head)
 458 #define free_pipe(pi, indent)        free_pipe(pi)
 459 #endif
 460 static int free_pipe_list(struct pipe *head, int indent);
 461 static int free_pipe(struct pipe *pi, int indent);
 462 /*  really run the final data structures: */
 463 static int setup_redirects(struct child_prog *prog, int squirrel[]);
 464 static int run_list_real(struct pipe *pi);
 465 static void pseudo_exec_argv(char **argv) ATTRIBUTE_NORETURN;
 466 static void pseudo_exec(struct child_prog *child) ATTRIBUTE_NORETURN;
 467 static int run_pipe_real(struct pipe *pi);
 468 /*   extended glob support: */
 469 static int globhack(const char *src, int flags, glob_t *pglob);
 470 static int glob_needed(const char *s);
 471 static int xglob(o_string *dest, int flags, glob_t *pglob);
 472 /*   variable assignment: */
 473 static int is_assignment(const char *s);
 474 /*   data structure manipulation: */
 475 static int setup_redirect(struct p_context *ctx, int fd, redir_type style, struct in_str *input);
 476 static void initialize_context(struct p_context *ctx);
 477 static int done_word(o_string *dest, struct p_context *ctx);
 478 static int done_command(struct p_context *ctx);
 479 static int done_pipe(struct p_context *ctx, pipe_style type);
 480 /*   primary string parsing: */
 481 static int redirect_dup_num(struct in_str *input);
 482 static int redirect_opt_num(o_string *o);
 483 static int process_command_subs(o_string *dest, struct p_context *ctx, struct in_str *input, const char *subst_end);
 484 static int parse_group(o_string *dest, struct p_context *ctx, struct in_str *input, int ch);
 485 static const char *lookup_param(const char *src);
 486 static char *make_string(char **inp);
 487 static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input);
 488 static int parse_stream(o_string *dest, struct p_context *ctx, struct in_str *input0, const char *end_trigger);
 489 /*   setup: */
 490 static int parse_stream_outer(struct in_str *inp, int parse_flag);
 491 static int parse_string_outer(const char *s, int parse_flag);
 492 static int parse_file_outer(FILE *f);
 493 /*   job management: */
 494 static int checkjobs(struct pipe* fg_pipe);
 495 #if ENABLE_HUSH_JOB
 496 static int checkjobs_and_fg_shell(struct pipe* fg_pipe);
 497 static void insert_bg_job(struct pipe *pi);
 498 static void remove_bg_job(struct pipe *pi);
 499 static void delete_finished_bg_job(struct pipe *pi);
 500 #else
 501 int checkjobs_and_fg_shell(struct pipe* fg_pipe); /* never called */
 502 #endif
 503 /*     local variable support */
 504 static char **expand_variables_to_list(char **argv);
 505 /* used for expansion of right hand of assignments */
 506 static char *expand_variables_to_string(const char *str);
 507 static const char *get_local_var(const char *var);
 508 static int set_local_var(const char *s, int flg_export);
 509 static void unset_local_var(const char *name);
 510
 511 /* Table of built-in functions.  They can be forked or not, depending on
 512  * context: within pipes, they fork.  As simple commands, they do not.
 513  * When used in non-forking context, they can change global variables
 514  * in the parent shell process.  If forked, of course they cannot.
 515  * For example, 'unset foo | whatever' will parse and run, but foo will
 516  * still be set at the end. */
 517 struct built_in_command {
 518         const char *cmd;                /* name */
 519         const char *descr;              /* description */
 520         int (*function) (char **argv);  /* function ptr */
 521 };
 522
 523 static const struct built_in_command bltins[] = {
 524 #if ENABLE_HUSH_JOB
 525         { "bg", "Resume a job in the background", builtin_fg_bg },
 526 #endif
 527         { "break", "Exit for, while or until loop", builtin_not_written },
 528         { "cd", "Change working directory", builtin_cd },
 529         { "continue", "Continue for, while or until loop", builtin_not_written },
 530         { "eval", "Construct and run shell command", builtin_eval },
 531         { "exec", "Exec command, replacing this shell with the exec'd process",
 532                 builtin_exec },
 533         { "exit", "Exit from shell()", builtin_exit },
 534         { "export", "Set environment variable", builtin_export },
 535 #if ENABLE_HUSH_JOB
 536         { "fg", "Bring job into the foreground", builtin_fg_bg },
 537         { "jobs", "Lists the active jobs", builtin_jobs },
 538 #endif
 539         { "pwd", "Print current directory", builtin_pwd },
 540         { "read", "Input environment variable", builtin_read },
 541         { "return", "Return from a function", builtin_not_written },
 542         { "set", "Set/unset shell local variables", builtin_set },
 543         { "shift", "Shift positional parameters", builtin_shift },
 544         { "trap", "Trap signals", builtin_not_written },
 545         { "ulimit","Controls resource limits", builtin_not_written },
 546         { "umask","Sets file creation mask", builtin_umask },
 547         { "unset", "Unset environment variable", builtin_unset },
 548         { ".", "Source-in and run commands in a file", builtin_source },
 549         { "help", "List shell built-in commands", builtin_help },
 550         { NULL, NULL, NULL }
 551 };
 552
 553 #if ENABLE_HUSH_JOB
 554
 555 /* move to libbb? */
 556 static void signal_SA_RESTART(int sig, void (*handler)(int))
 557 {
 558         struct sigaction sa;
 559         sa.sa_handler = handler;
 560         sa.sa_flags = SA_RESTART;
 561         sigemptyset(&sa.sa_mask);
 562         sigaction(sig, &sa, NULL);
 563 }
 564
 565 /* Signals are grouped, we handle them in batches */
 566 static void set_fatal_sighandler(void (*handler)(int))
 567 {
 568         signal(SIGILL , handler);
 569         signal(SIGTRAP, handler);
 570         signal(SIGABRT, handler);
 571         signal(SIGFPE , handler);
 572         signal(SIGBUS , handler);
 573         signal(SIGSEGV, handler);
 574         /* bash 3.2 seems to handle these just like 'fatal' ones */
 575         signal(SIGHUP , handler);
 576         signal(SIGPIPE, handler);
 577         signal(SIGALRM, handler);
 578 }
 579 static void set_jobctrl_sighandler(void (*handler)(int))
 580 {
 581         signal(SIGTSTP, handler);
 582         signal(SIGTTIN, handler);
 583         signal(SIGTTOU, handler);
 584 }
 585 static void set_misc_sighandler(void (*handler)(int))
 586 {
 587         signal(SIGINT , handler);
 588         signal(SIGQUIT, handler);
 589         signal(SIGTERM, handler);
 590 }
 591 /* SIGCHLD is special and handled separately */
 592
 593 static void set_every_sighandler(void (*handler)(int))
 594 {
 595         set_fatal_sighandler(handler);
 596         set_jobctrl_sighandler(handler);
 597         set_misc_sighandler(handler);
 598         signal(SIGCHLD, handler);
 599 }
 600
 601 static void handler_ctrl_c(int sig)
 602 {
 603         debug_printf_jobs("got sig %d\n", sig);
 604 // as usual we can have all kinds of nasty problems with leaked malloc data here
 605         siglongjmp(toplevel_jb, 1);
 606 }
 607
 608 static void handler_ctrl_z(int sig)
 609 {
 610         pid_t pid;
 611
 612         debug_printf_jobs("got tty sig %d in pid %d\n", sig, getpid());
 613         pid = fork();
 614         if (pid < 0) /* can't fork. Pretend there was no ctrl-Z */
 615                 return;
 616         ctrl_z_flag = 1;
 617         if (!pid) { /* child */
 618                 setpgrp();
 619                 debug_printf_jobs("set pgrp for child %d ok\n", getpid());
 620                 set_every_sighandler(SIG_DFL);
 621                 raise(SIGTSTP); /* resend TSTP so that child will be stopped */
 622                 debug_printf_jobs("returning in child\n");
 623                 /* return to nofork, it will eventually exit now,
 624                  * not return back to shell */
 625                 return;
 626         }
 627         /* parent */
 628         /* finish filling up pipe info */
 629         toplevel_list->pgrp = pid; /* child is in its own pgrp */
 630         toplevel_list->progs[0].pid = pid;
 631         /* parent needs to longjmp out of running nofork.
 632          * we will "return" exitcode 0, with child put in background */
 633 // as usual we can have all kinds of nasty problems with leaked malloc data here
 634         debug_printf_jobs("siglongjmp in parent\n");
 635         siglongjmp(toplevel_jb, 1);
 636 }
 637
 638 /* Restores tty foreground process group, and exits.
 639  * May be called as signal handler for fatal signal
 640  * (will faithfully resend signal to itself, producing correct exit state)
 641  * or called directly with -EXITCODE.
 642  * We also call it if xfunc is exiting. */
 643 static void sigexit(int sig) ATTRIBUTE_NORETURN;
 644 static void sigexit(int sig)
 645 {
 646         sigset_t block_all;
 647
 648         /* Disable all signals: job control, SIGPIPE, etc. */
 649         sigfillset(&block_all);
 650         sigprocmask(SIG_SETMASK, &block_all, NULL);
 651
 652         if (interactive_fd)
 653                 tcsetpgrp(interactive_fd, saved_tty_pgrp);
 654
 655         /* Not a signal, just exit */
 656         if (sig <= 0)
 657                 _exit(- sig);
 658
 659         /* Enable only this sig and kill ourself with it */
 660         signal(sig, SIG_DFL);
 661         sigdelset(&block_all, sig);
 662         sigprocmask(SIG_SETMASK, &block_all, NULL);
 663         raise(sig);
 664         _exit(1); /* Should not reach it */
 665 }
 666
 667 /* Restores tty foreground process group, and exits. */
 668 static void hush_exit(int exitcode) ATTRIBUTE_NORETURN;
 669 static void hush_exit(int exitcode)
 670 {
 671         fflush(NULL); /* flush all streams */
 672         sigexit(- (exitcode & 0xff));
 673 }
 674
 675 #else /* !JOB */
 676
 677 #define set_fatal_sighandler(handler)   ((void)0)
 678 #define set_jobctrl_sighandler(handler) ((void)0)
 679 #define set_misc_sighandler(handler)    ((void)0)
 680 #define hush_exit(e)                    exit(e)
 681
 682 #endif /* JOB */
 683
 684
 685 static const char *set_cwd(void)
 686 {
 687         if (cwd == bb_msg_unknown)
 688                 cwd = NULL;     /* xrealloc_getcwd_or_warn(arg) calls free(arg)! */
 689         cwd = xrealloc_getcwd_or_warn((char *)cwd);
 690         if (!cwd)
 691                 cwd = bb_msg_unknown;
 692         return cwd;
 693 }
 694
 695 /* built-in 'eval' handler */
 696 static int builtin_eval(char **argv)
 697 {
 698         char *str = NULL;
 699         int rcode = EXIT_SUCCESS;
 700
 701         if (argv[1]) {
 702                 str = make_string(argv + 1);
 703                 parse_string_outer(str, FLAG_EXIT_FROM_LOOP |
 704                                         FLAG_PARSE_SEMICOLON);
 705                 free(str);
 706                 rcode = last_return_code;
 707         }
 708         return rcode;
 709 }
 710
 711 /* built-in 'cd <path>' handler */
 712 static int builtin_cd(char **argv)
 713 {
 714         char *newdir;
 715         if (argv[1] == NULL)
 716                 newdir = getenv("HOME");
 717         else
 718                 newdir = argv[1];
 719         if (chdir(newdir)) {
 720                 printf("cd: %s: %s\n", newdir, strerror(errno));
 721                 return EXIT_FAILURE;
 722         }
 723         set_cwd();
 724         return EXIT_SUCCESS;
 725 }
 726
 727 /* built-in 'exec' handler */
 728 static int builtin_exec(char **argv)
 729 {
 730         if (argv[1] == NULL)
 731                 return EXIT_SUCCESS;   /* Really? */
 732         pseudo_exec_argv(argv + 1);
 733         /* never returns */
 734 }
 735
 736 /* built-in 'exit' handler */
 737 static int builtin_exit(char **argv)
 738 {
 739 // TODO: bash does it ONLY on top-level sh exit (+interacive only?)
 740         //puts("exit"); /* bash does it */
 741 // TODO: warn if we have background jobs: "There are stopped jobs"
 742 // On second consecutive 'exit', exit anyway.
 743
 744         if (argv[1] == NULL)
 745                 hush_exit(last_return_code);
 746         /* mimic bash: exit 123abc == exit 255 + error msg */
 747         xfunc_error_retval = 255;
 748         /* bash: exit -2 == exit 254, no error msg */
 749         hush_exit(xatoi(argv[1]) & 0xff);
 750 }
 751
 752 /* built-in 'export VAR=value' handler */
 753 static int builtin_export(char **argv)
 754 {
 755         int res = 0;
 756         char *name = argv[1];
 757
 758         if (name == NULL) {
 759                 // TODO:
 760                 // ash emits: export VAR='VAL'
 761                 // bash: declare -x VAR="VAL"
 762                 // (both also escape as needed (quotes, $, etc))
 763                 char **e = environ;
 764                 if (e)
 765                         while (*e)
 766                                 puts(*e++);
 767                 return EXIT_SUCCESS;
 768         }
 769
 770         name = xstrdup(name);
 771         {
 772                 const char *value = strchr(name, '=');
 773
 774                 if (!value) {
 775                         char *tmp;
 776                         /* They are exporting something without an =VALUE */
 777
 778                         value = get_local_var(name);
 779                         if (value) {
 780                                 size_t ln = strlen(name);
 781
 782                                 tmp = xrealloc(name, ln+strlen(value)+2);
 783                                 sprintf(tmp+ln, "=%s", value);
 784                                 name = tmp;
 785                         } else {
 786                                 /* bash does not return an error when trying to export
 787                                  * an undefined variable.  Do likewise. */
 788                                 res = 1;
 789                         }
 790                 }
 791         }
 792         if (res < 0)
 793                 bb_perror_msg("export");
 794         else if (res == 0)
 795                 res = set_local_var(name, 1);
 796         else
 797                 res = 0;
 798         free(name);
 799         return res;
 800 }
 801
 802 #if ENABLE_HUSH_JOB
 803 /* built-in 'fg' and 'bg' handler */
 804 static int builtin_fg_bg(char **argv)
 805 {
 806         int i, jobnum;
 807         struct pipe *pi;
 808
 809         if (!interactive_fd)
 810                 return EXIT_FAILURE;
 811         /* If they gave us no args, assume they want the last backgrounded task */
 812         if (!argv[1]) {
 813                 for (pi = job_list; pi; pi = pi->next) {
 814                         if (pi->jobid == last_jobid) {
 815                                 goto found;
 816                         }
 817                 }
 818                 bb_error_msg("%s: no current job", argv[0]);
 819                 return EXIT_FAILURE;
 820         }
 821         if (sscanf(argv[1], "%%%d", &jobnum) != 1) {
 822                 bb_error_msg("%s: bad argument '%s'", argv[0], argv[1]);
 823                 return EXIT_FAILURE;
 824         }
 825         for (pi = job_list; pi; pi = pi->next) {
 826                 if (pi->jobid == jobnum) {
 827                         goto found;
 828                 }
 829         }
 830         bb_error_msg("%s: %d: no such job", argv[0], jobnum);
 831         return EXIT_FAILURE;
 832  found:
 833         // TODO: bash prints a string representation
 834         // of job being foregrounded (like "sleep 1 | cat")
 835         if (*argv[0] == 'f') {
 836                 /* Put the job into the foreground.  */
 837                 tcsetpgrp(interactive_fd, pi->pgrp);
 838         }
 839
 840         /* Restart the processes in the job */
 841         debug_printf_jobs("reviving %d procs, pgrp %d\n", pi->num_progs, pi->pgrp);
 842         for (i = 0; i < pi->num_progs; i++) {
 843                 debug_printf_jobs("reviving pid %d\n", pi->progs[i].pid);
 844                 pi->progs[i].is_stopped = 0;
 845         }
 846         pi->stopped_progs = 0;
 847
 848         i = kill(- pi->pgrp, SIGCONT);
 849         if (i < 0) {
 850                 if (errno == ESRCH) {
 851                         delete_finished_bg_job(pi);
 852                         return EXIT_SUCCESS;
 853                 } else {
 854                         bb_perror_msg("kill (SIGCONT)");
 855                 }
 856         }
 857
 858         if (*argv[0] == 'f') {
 859                 remove_bg_job(pi);
 860                 return checkjobs_and_fg_shell(pi);
 861         }
 862         return EXIT_SUCCESS;
 863 }
 864 #endif
 865
 866 /* built-in 'help' handler */
 867 static int builtin_help(char **argv ATTRIBUTE_UNUSED)
 868 {
 869         const struct built_in_command *x;
 870
 871         printf("\nBuilt-in commands:\n");
 872         printf("-------------------\n");
 873         for (x = bltins; x->cmd; x++) {
 874                 if (x->descr == NULL)
 875                         continue;
 876                 printf("%s\t%s\n", x->cmd, x->descr);
 877         }
 878         printf("\n\n");
 879         return EXIT_SUCCESS;
 880 }
 881
 882 #if ENABLE_HUSH_JOB
 883 /* built-in 'jobs' handler */
 884 static int builtin_jobs(char **argv ATTRIBUTE_UNUSED)
 885 {
 886         struct pipe *job;
 887         const char *status_string;
 888
 889         for (job = job_list; job; job = job->next) {
 890                 if (job->running_progs == job->stopped_progs)
 891                         status_string = "Stopped";
 892                 else
 893                         status_string = "Running";
 894
 895                 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->cmdtext);
 896         }
 897         return EXIT_SUCCESS;
 898 }
 899 #endif
 900
 901 /* built-in 'pwd' handler */
 902 static int builtin_pwd(char **argv ATTRIBUTE_UNUSED)
 903 {
 904         puts(set_cwd());
 905         return EXIT_SUCCESS;
 906 }
 907
 908 /* built-in 'read VAR' handler */
 909 static int builtin_read(char **argv)
 910 {
 911         char string[BUFSIZ];
 912         char *p;
 913         const char *name = argv[1] ? argv[1] : "REPLY";
 914         int name_len = strlen(name);
 915
 916         if (name_len >= sizeof(string) - 2)
 917                 return EXIT_FAILURE;
 918         strcpy(string, name);
 919         p = string + name_len;
 920         *p++ = '=';
 921         *p = '\0'; /* In case stdin has only EOF */
 922         /* read string. name_len+1 chars are already used by 'name=' */
 923         fgets(p, sizeof(string) - 1 - name_len, stdin);
 924         chomp(p);
 925         return set_local_var(string, 0);
 926 }
 927
 928 /* built-in 'set [VAR=value]' handler */
 929 static int builtin_set(char **argv)
 930 {
 931         char *temp = argv[1];
 932         struct variables *e;
 933
 934         if (temp == NULL)
 935                 for (e = top_vars; e; e = e->next)
 936                         printf("%s=%s\n", e->name, e->value);
 937         else
 938                 set_local_var(temp, 0);
 939
 940         return EXIT_SUCCESS;
 941 }
 942
 943
 944 /* Built-in 'shift' handler */
 945 static int builtin_shift(char **argv)
 946 {
 947         int n = 1;
 948         if (argv[1]) {
 949                 n = atoi(argv[1]);
 950         }
 951         if (n >= 0 && n < global_argc) {
 952                 /* XXX This probably breaks $0 */
 953                 global_argc -= n;
 954                 global_argv += n;
 955                 return EXIT_SUCCESS;
 956         }
 957         return EXIT_FAILURE;
 958 }
 959
 960 /* Built-in '.' handler (read-in and execute commands from file) */
 961 static int builtin_source(char **argv)
 962 {
 963         FILE *input;
 964         int status;
 965
 966         if (argv[1] == NULL)
 967                 return EXIT_FAILURE;
 968
 969         /* XXX search through $PATH is missing */
 970         input = fopen(argv[1], "r");
 971         if (!input) {
 972                 bb_error_msg("cannot open '%s'", argv[1]);
 973                 return EXIT_FAILURE;
 974         }
 975
 976         /* Now run the file */
 977         /* XXX argv and argc are broken; need to save old global_argv
 978          * (pointer only is OK!) on this stack frame,
 979          * set global_argv=argv+1, recurse, and restore. */
 980         mark_open(fileno(input));
 981         status = parse_file_outer(input);
 982         mark_closed(fileno(input));
 983         fclose(input);
 984         return status;
 985 }
 986
 987 static int builtin_umask(char **argv)
 988 {
 989         mode_t new_umask;
 990         const char *arg = argv[1];
 991         char *end;
 992         if (arg) {
 993                 new_umask = strtoul(arg, &end, 8);
 994                 if (*end != '\0' || end == arg) {
 995                         return EXIT_FAILURE;
 996                 }
 997         } else {
 998                 new_umask = umask(0);
 999                 printf("%.3o\n", (unsigned) new_umask);
1000         }
1001         umask(new_umask);
1002         return EXIT_SUCCESS;
1003 }
1004
1005 /* built-in 'unset VAR' handler */
1006 static int builtin_unset(char **argv)
1007 {
1008         /* bash always returns true */
1009         unset_local_var(argv[1]);
1010         return EXIT_SUCCESS;
1011 }
1012
1013 static int builtin_not_written(char **argv)
1014 {
1015         printf("builtin_%s not written\n", argv[0]);
1016         return EXIT_FAILURE;
1017 }
1018
1019 static int b_check_space(o_string *o, int len)
1020 {
1021         /* It would be easy to drop a more restrictive policy
1022          * in here, such as setting a maximum string length */
1023         if (o->length + len > o->maxlen) {
1024                 /* assert(data == NULL || o->maxlen != 0); */
1025                 o->maxlen += (2*len > B_CHUNK ? 2*len : B_CHUNK);
1026                 o->data = xrealloc(o->data, 1 + o->maxlen);
1027         }
1028         return o->data == NULL;
1029 }
1030
1031 static int b_addchr(o_string *o, int ch)
1032 {
1033         debug_printf("b_addchr: '%c' o->length=%d o=%p\n", ch, o->length, o);
1034         if (b_check_space(o, 1))
1035                 return B_NOSPAC;
1036         o->data[o->length] = ch;
1037         o->length++;
1038         o->data[o->length] = '\0';
1039         return 0;
1040 }
1041
1042 static void b_reset(o_string *o)
1043 {
1044         o->length = 0;
1045         o->nonnull = 0;
1046         if (o->data != NULL)
1047                 *o->data = '\0';
1048 }
1049
1050 static void b_free(o_string *o)
1051 {
1052         b_reset(o);
1053         free(o->data);
1054         o->data = NULL;
1055         o->maxlen = 0;
1056 }
1057
1058 /* My analysis of quoting semantics tells me that state information
1059  * is associated with a destination, not a source.
1060  */
1061 static int b_addqchr(o_string *o, int ch, int quote)
1062 {
1063         if (quote && strchr("*?[\\", ch)) {
1064                 int rc;
1065                 rc = b_addchr(o, '\\');
1066                 if (rc)
1067                         return rc;
1068         }
1069         return b_addchr(o, ch);
1070 }
1071
1072 static int static_get(struct in_str *i)
1073 {
1074         int ch = *i->p++;
1075         if (ch == '\0') return EOF;
1076         return ch;
1077 }
1078
1079 static int static_peek(struct in_str *i)
1080 {
1081         return *i->p;
1082 }
1083
1084 #if ENABLE_HUSH_INTERACTIVE
1085 #if ENABLE_FEATURE_EDITING
1086 static void cmdedit_set_initial_prompt(void)
1087 {
1088 #if !ENABLE_FEATURE_EDITING_FANCY_PROMPT
1089         PS1 = NULL;
1090 #else
1091         PS1 = getenv("PS1");
1092         if (PS1 == NULL)
1093                 PS1 = "\\w \\$ ";
1094 #endif
1095 }
1096 #endif /* EDITING */
1097
1098 static const char* setup_prompt_string(int promptmode)
1099 {
1100         const char *prompt_str;
1101         debug_printf("setup_prompt_string %d ", promptmode);
1102 #if !ENABLE_FEATURE_EDITING_FANCY_PROMPT
1103         /* Set up the prompt */
1104         if (promptmode == 1) {
1105                 char *ns;
1106                 free((char*)PS1);
1107                 ns = xmalloc(strlen(cwd)+4);
1108                 sprintf(ns, "%s %s", cwd, (geteuid() != 0) ? "$ " : "# ");
1109                 prompt_str = ns;
1110                 PS1 = ns;
1111         } else {
1112                 prompt_str = PS2;
1113         }
1114 #else
1115         prompt_str = (promptmode == 1) ? PS1 : PS2;
1116 #endif
1117         debug_printf("result %s\n", prompt_str);
1118         return prompt_str;
1119 }
1120
1121 static void get_user_input(struct in_str *i)
1122 {
1123         int r;
1124         const char *prompt_str;
1125
1126         prompt_str = setup_prompt_string(i->promptmode);
1127 #if ENABLE_FEATURE_EDITING
1128         /*
1129          ** enable command line editing only while a command line
1130          ** is actually being read; otherwise, we'll end up bequeathing
1131          ** atexit() handlers and other unwanted stuff to our
1132          ** child processes (rob@sysgo.de)
1133          */
1134         r = read_line_input(prompt_str, user_input_buf, BUFSIZ-1, line_input_state);
1135         i->eof_flag = (r < 0);
1136         if (i->eof_flag) { /* EOF/error detected */
1137                 user_input_buf[0] = EOF; /* yes, it will be truncated, it's ok */
1138                 user_input_buf[1] = '\0';
1139         }
1140 #else
1141         fputs(prompt_str, stdout);
1142         fflush(stdout);
1143         user_input_buf[0] = r = fgetc(i->file);
1144         /*user_input_buf[1] = '\0'; - already is and never changed */
1145         i->eof_flag = (r == EOF);
1146 #endif
1147         i->p = user_input_buf;
1148 }
1149 #endif  /* INTERACTIVE */
1150
1151 /* This is the magic location that prints prompts
1152  * and gets data back from the user */
1153 static int file_get(struct in_str *i)
1154 {
1155         int ch;
1156
1157         /* If there is data waiting, eat it up */
1158         if (i->p && *i->p) {
1159 #if ENABLE_HUSH_INTERACTIVE
1160  take_cached:
1161 #endif
1162                 ch = *i->p++;
1163                 if (i->eof_flag && !*i->p)
1164                         ch = EOF;
1165         } else {
1166                 /* need to double check i->file because we might be doing something
1167                  * more complicated by now, like sourcing or substituting. */
1168 #if ENABLE_HUSH_INTERACTIVE
1169                 if (interactive_fd && i->__promptme && i->file == stdin) {
1170                         do {
1171                                 get_user_input(i);
1172                         } while (!*i->p); /* need non-empty line */
1173                         i->promptmode = 2;
1174                         i->__promptme = 0;
1175                         goto take_cached;
1176                 }
1177 #endif
1178                 ch = fgetc(i->file);
1179         }
1180         debug_printf("file_get: got a '%c' %d\n", ch, ch);
1181 #if ENABLE_HUSH_INTERACTIVE
1182         if (ch == '\n')
1183                 i->__promptme = 1;
1184 #endif
1185         return ch;
1186 }
1187
1188 /* All the callers guarantee this routine will never be
1189  * used right after a newline, so prompting is not needed.
1190  */
1191 static int file_peek(struct in_str *i)
1192 {
1193         int ch;
1194         if (i->p && *i->p) {
1195                 if (i->eof_flag && !i->p[1])
1196                         return EOF;
1197                 return *i->p;
1198         }
1199         ch = fgetc(i->file);
1200         i->eof_flag = (ch == EOF);
1201         i->peek_buf[0] = ch;
1202         i->peek_buf[1] = '\0';
1203         i->p = i->peek_buf;
1204         debug_printf("file_peek: got a '%c' %d\n", *i->p, *i->p);
1205         return ch;
1206 }
1207
1208 static void setup_file_in_str(struct in_str *i, FILE *f)
1209 {
1210         i->peek = file_peek;
1211         i->get = file_get;
1212 #if ENABLE_HUSH_INTERACTIVE
1213         i->__promptme = 1;
1214         i->promptmode = 1;
1215 #endif
1216         i->file = f;
1217         i->p = NULL;
1218 }
1219
1220 static void setup_string_in_str(struct in_str *i, const char *s)
1221 {
1222         i->peek = static_peek;
1223         i->get = static_get;
1224 #if ENABLE_HUSH_INTERACTIVE
1225         i->__promptme = 1;
1226         i->promptmode = 1;
1227 #endif
1228         i->p = s;
1229         i->eof_flag = 0;
1230 }
1231
1232 static void mark_open(int fd)
1233 {
1234         struct close_me *new = xmalloc(sizeof(struct close_me));
1235         new->fd = fd;
1236         new->next = close_me_head;
1237         close_me_head = new;
1238 }
1239
1240 static void mark_closed(int fd)
1241 {
1242         struct close_me *tmp;
1243         if (close_me_head == NULL || close_me_head->fd != fd)
1244                 bb_error_msg_and_die("corrupt close_me");
1245         tmp = close_me_head;
1246         close_me_head = close_me_head->next;
1247         free(tmp);
1248 }
1249
1250 static void close_all(void)
1251 {
1252         struct close_me *c;
1253         for (c = close_me_head; c; c = c->next) {
1254                 close(c->fd);
1255         }
1256         close_me_head = NULL;
1257 }
1258
1259 /* squirrel != NULL means we squirrel away copies of stdin, stdout,
1260  * and stderr if they are redirected. */
1261 static int setup_redirects(struct child_prog *prog, int squirrel[])
1262 {
1263         int openfd, mode;
1264         struct redir_struct *redir;
1265
1266         for (redir = prog->redirects; redir; redir = redir->next) {
1267                 if (redir->dup == -1 && redir->word.gl_pathv == NULL) {
1268                         /* something went wrong in the parse.  Pretend it didn't happen */
1269                         continue;
1270                 }
1271                 if (redir->dup == -1) {
1272                         mode = redir_table[redir->type].mode;
1273                         openfd = open_or_warn(redir->word.gl_pathv[0], mode);
1274                         if (openfd < 0) {
1275                         /* this could get lost if stderr has been redirected, but
1276                            bash and ash both lose it as well (though zsh doesn't!) */
1277                                 return 1;
1278                         }
1279                 } else {
1280                         openfd = redir->dup;
1281                 }
1282
1283                 if (openfd != redir->fd) {
1284                         if (squirrel && redir->fd < 3) {
1285                                 squirrel[redir->fd] = dup(redir->fd);
1286                         }
1287                         if (openfd == -3) {
1288                                 close(openfd);
1289                         } else {
1290                                 dup2(openfd, redir->fd);
1291                                 if (redir->dup == -1)
1292                                         close(openfd);
1293                         }
1294                 }
1295         }
1296         return 0;
1297 }
1298
1299 static void restore_redirects(int squirrel[])
1300 {
1301         int i, fd;
1302         for (i = 0; i < 3; i++) {
1303                 fd = squirrel[i];
1304                 if (fd != -1) {
1305                         /* We simply die on error */
1306                         xmove_fd(fd, i);
1307                 }
1308         }
1309 }
1310
1311 /* never returns */
1312 /* XXX no exit() here.  If you don't exec, use _exit instead.
1313  * The at_exit handlers apparently confuse the calling process,
1314  * in particular stdin handling.  Not sure why? -- because of vfork! (vda) */
1315 static void pseudo_exec_argv(char **argv)
1316 {
1317         int i, rcode;
1318         char *p;
1319         const struct built_in_command *x;
1320
1321         for (i = 0; is_assignment(argv[i]); i++) {
1322                 debug_printf_exec("pid %d environment modification: %s\n",
1323                                 getpid(), argv[i]);
1324 // FIXME: vfork case??
1325                 p = expand_variables_to_string(argv[i]);
1326                 putenv(p == argv[i] ? xstrdup(p) : p);
1327         }
1328         argv += i;
1329         /* If a variable is assigned in a forest, and nobody listens,
1330          * was it ever really set?
1331          */
1332         if (argv[0] == NULL) {
1333                 _exit(EXIT_SUCCESS);
1334         }
1335
1336         argv = expand_variables_to_list(argv);
1337
1338         /*
1339          * Check if the command matches any of the builtins.
1340          * Depending on context, this might be redundant.  But it's
1341          * easier to waste a few CPU cycles than it is to figure out
1342          * if this is one of those cases.
1343          */
1344         for (x = bltins; x->cmd; x++) {
1345                 if (strcmp(argv[0], x->cmd) == 0) {
1346                         debug_printf_exec("running builtin '%s'\n", argv[0]);
1347                         rcode = x->function(argv);
1348                         fflush(stdout);
1349                         _exit(rcode);
1350                 }
1351         }
1352
1353         /* Check if the command matches any busybox applets */
1354 #if ENABLE_FEATURE_SH_STANDALONE
1355         if (strchr(argv[0], '/') == NULL) {
1356                 const struct bb_applet *a = find_applet_by_name(argv[0]);
1357                 if (a) {
1358                         if (a->noexec) {
1359                                 current_applet = a;
1360                                 debug_printf_exec("running applet '%s'\n", argv[0]);
1361 // is it ok that run_current_applet_and_exit() does exit(), not _exit()?
1362                                 run_current_applet_and_exit(argv);
1363                         }
1364                         /* re-exec ourselves with the new arguments */
1365                         debug_printf_exec("re-execing applet '%s'\n", argv[0]);
1366                         execvp(CONFIG_BUSYBOX_EXEC_PATH, argv);
1367                         /* If they called chroot or otherwise made the binary no longer
1368                          * executable, fall through */
1369                 }
1370         }
1371 #endif
1372
1373         debug_printf_exec("execing '%s'\n", argv[0]);
1374         execvp(argv[0], argv);
1375         bb_perror_msg("cannot exec '%s'", argv[0]);
1376         _exit(1);
1377 }
1378
1379 static void pseudo_exec(struct child_prog *child)
1380 {
1381 // FIXME: buggy wrt NOMMU! Must not modify any global data
1382 // until it does exec/_exit, but currently it does.
1383         int rcode;
1384
1385         if (child->argv) {
1386                 pseudo_exec_argv(child->argv);
1387         }
1388
1389         if (child->group) {
1390         // FIXME: do not modify globals! Think vfork!
1391 #if ENABLE_HUSH_INTERACTIVE
1392                 debug_printf_exec("pseudo_exec: setting interactive_fd=0\n");
1393                 interactive_fd = 0;    /* crucial!!!! */
1394 #endif
1395                 debug_printf_exec("pseudo_exec: run_list_real\n");
1396                 rcode = run_list_real(child->group);
1397                 /* OK to leak memory by not calling free_pipe_list,
1398                  * since this process is about to exit */
1399                 _exit(rcode);
1400         }
1401
1402         /* Can happen.  See what bash does with ">foo" by itself. */
1403         debug_printf("trying to pseudo_exec null command\n");
1404         _exit(EXIT_SUCCESS);
1405 }
1406
1407 #if ENABLE_HUSH_JOB
1408 static const char *get_cmdtext(struct pipe *pi)
1409 {
1410         char **argv;
1411         char *p;
1412         int len;
1413
1414         /* This is subtle. ->cmdtext is created only on first backgrounding.
1415          * (Think "cat, <ctrl-z>, fg, <ctrl-z>, fg, <ctrl-z>...." here...)
1416          * On subsequent bg argv is trashed, but we won't use it */
1417         if (pi->cmdtext)
1418                 return pi->cmdtext;
1419         argv = pi->progs[0].argv;
1420         if (!argv || !argv[0])
1421                 return (pi->cmdtext = xzalloc(1));
1422
1423         len = 0;
1424         do len += strlen(*argv) + 1; while (*++argv);
1425         pi->cmdtext = p = xmalloc(len);
1426         argv = pi->progs[0].argv;
1427         do {
1428                 len = strlen(*argv);
1429                 memcpy(p, *argv, len);
1430                 p += len;
1431                 *p++ = ' ';
1432         } while (*++argv);
1433         p[-1] = '\0';
1434         return pi->cmdtext;
1435 }
1436
1437 static void insert_bg_job(struct pipe *pi)
1438 {
1439         struct pipe *thejob;
1440         int i;
1441
1442         /* Linear search for the ID of the job to use */
1443         pi->jobid = 1;
1444         for (thejob = job_list; thejob; thejob = thejob->next)
1445                 if (thejob->jobid >= pi->jobid)
1446                         pi->jobid = thejob->jobid + 1;
1447
1448         /* Add thejob to the list of running jobs */
1449         if (!job_list) {
1450                 thejob = job_list = xmalloc(sizeof(*thejob));
1451         } else {
1452                 for (thejob = job_list; thejob->next; thejob = thejob->next)
1453                         continue;
1454                 thejob->next = xmalloc(sizeof(*thejob));
1455                 thejob = thejob->next;
1456         }
1457
1458         /* Physically copy the struct job */
1459         memcpy(thejob, pi, sizeof(struct pipe));
1460         thejob->progs = xzalloc(sizeof(pi->progs[0]) * pi->num_progs);
1461         /* We cannot copy entire pi->progs[] vector! Double free()s will happen */
1462         for (i = 0; i < pi->num_progs; i++) {
1463 // TODO: do we really need to have so many fields which are just dead weight
1464 // at execution stage?
1465                 thejob->progs[i].pid = pi->progs[i].pid;
1466                 /* all other fields are not used and stay zero */
1467         }
1468         thejob->next = NULL;
1469         thejob->cmdtext = xstrdup(get_cmdtext(pi));
1470
1471         /* We don't wait for background thejobs to return -- append it
1472            to the list of backgrounded thejobs and leave it alone */
1473         printf("[%d] %d %s\n", thejob->jobid, thejob->progs[0].pid, thejob->cmdtext);
1474         last_bg_pid = thejob->progs[0].pid;
1475         last_jobid = thejob->jobid;
1476 }
1477
1478 static void remove_bg_job(struct pipe *pi)
1479 {
1480         struct pipe *prev_pipe;
1481
1482         if (pi == job_list) {
1483                 job_list = pi->next;
1484         } else {
1485                 prev_pipe = job_list;
1486                 while (prev_pipe->next != pi)
1487                         prev_pipe = prev_pipe->next;
1488                 prev_pipe->next = pi->next;
1489         }
1490         if (job_list)
1491                 last_jobid = job_list->jobid;
1492         else
1493                 last_jobid = 0;
1494 }
1495
1496 /* remove a backgrounded job */
1497 static void delete_finished_bg_job(struct pipe *pi)
1498 {
1499         remove_bg_job(pi);
1500         pi->stopped_progs = 0;
1501         free_pipe(pi, 0);
1502         free(pi);
1503 }
1504 #endif /* JOB */
1505
1506 /* Checks to see if any processes have exited -- if they
1507    have, figure out why and see if a job has completed */
1508 static int checkjobs(struct pipe* fg_pipe)
1509 {
1510         int attributes;
1511         int status;
1512 #if ENABLE_HUSH_JOB
1513         int prognum = 0;
1514         struct pipe *pi;
1515 #endif
1516         pid_t childpid;
1517         int rcode = 0;
1518
1519         attributes = WUNTRACED;
1520         if (fg_pipe == NULL) {
1521                 attributes |= WNOHANG;
1522         }
1523
1524 /* Do we do this right?
1525  * bash-3.00# sleep 20 | false
1526  * <ctrl-Z pressed>
1527  * [3]+  Stopped          sleep 20 | false
1528  * bash-3.00# echo $?
1529  * 1   <========== bg pipe is not fully done, but exitcode is already known!
1530  */
1531
1532 //FIXME: non-interactive bash does not continue even if all processes in fg pipe
1533 //are stopped. Testcase: "cat | cat" in a script (not on command line)
1534 // + killall -STOP cat
1535
1536  wait_more:
1537         while ((childpid = waitpid(-1, &status, attributes)) > 0) {
1538                 const int dead = WIFEXITED(status) || WIFSIGNALED(status);
1539
1540 #ifdef DEBUG_SHELL_JOBS
1541                 if (WIFSTOPPED(status))
1542                         debug_printf_jobs("pid %d stopped by sig %d (exitcode %d)\n",
1543                                         childpid, WSTOPSIG(status), WEXITSTATUS(status));
1544                 if (WIFSIGNALED(status))
1545                         debug_printf_jobs("pid %d killed by sig %d (exitcode %d)\n",
1546                                         childpid, WTERMSIG(status), WEXITSTATUS(status));
1547                 if (WIFEXITED(status))
1548                         debug_printf_jobs("pid %d exited, exitcode %d\n",
1549                                         childpid, WEXITSTATUS(status));
1550 #endif
1551                 /* Were we asked to wait for fg pipe? */
1552                 if (fg_pipe) {
1553                         int i;
1554                         for (i = 0; i < fg_pipe->num_progs; i++) {
1555                                 debug_printf_jobs("check pid %d\n", fg_pipe->progs[i].pid);
1556                                 if (fg_pipe->progs[i].pid == childpid) {
1557                                         /* printf("process %d exit %d\n", i, WEXITSTATUS(status)); */
1558                                         if (dead) {
1559                                                 fg_pipe->progs[i].pid = 0;
1560                                                 fg_pipe->running_progs--;
1561                                                 if (i == fg_pipe->num_progs-1)
1562                                                         /* last process gives overall exitstatus */
1563                                                         rcode = WEXITSTATUS(status);
1564                                         } else {
1565                                                 fg_pipe->progs[i].is_stopped = 1;
1566                                                 fg_pipe->stopped_progs++;
1567                                         }
1568                                         debug_printf_jobs("fg_pipe: running_progs %d stopped_progs %d\n",
1569                                                         fg_pipe->running_progs, fg_pipe->stopped_progs);
1570                                         if (fg_pipe->running_progs - fg_pipe->stopped_progs <= 0) {
1571                                                 /* All processes in fg pipe have exited/stopped */
1572 #if ENABLE_HUSH_JOB
1573                                                 if (fg_pipe->running_progs)
1574                                                         insert_bg_job(fg_pipe);
1575 #endif
1576                                                 return rcode;
1577                                         }
1578                                         /* There are still running processes in the fg pipe */
1579                                         goto wait_more;
1580                                 }
1581                         }
1582                         /* fall through to searching process in bg pipes */
1583                 }
1584
1585 #if ENABLE_HUSH_JOB
1586                 /* We asked to wait for bg or orphaned children */
1587                 /* No need to remember exitcode in this case */
1588                 for (pi = job_list; pi; pi = pi->next) {
1589                         prognum = 0;
1590                         while (prognum < pi->num_progs) {
1591                                 if (pi->progs[prognum].pid == childpid)
1592                                         goto found_pi_and_prognum;
1593                                 prognum++;
1594                         }
1595                 }
1596 #endif
1597
1598                 /* Happens when shell is used as init process (init=/bin/sh) */
1599                 debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
1600                 goto wait_more;
1601
1602 #if ENABLE_HUSH_JOB
1603  found_pi_and_prognum:
1604                 if (dead) {
1605                         /* child exited */
1606                         pi->progs[prognum].pid = 0;
1607                         pi->running_progs--;
1608                         if (!pi->running_progs) {
1609                                 printf(JOB_STATUS_FORMAT, pi->jobid,
1610                                                         "Done", pi->cmdtext);
1611                                 delete_finished_bg_job(pi);
1612                         }
1613                 } else {
1614                         /* child stopped */
1615                         pi->stopped_progs++;
1616                         pi->progs[prognum].is_stopped = 1;
1617                 }
1618 #endif
1619         }
1620
1621         /* wait found no children or failed */
1622
1623         if (childpid && errno != ECHILD)
1624                 bb_perror_msg("waitpid");
1625         return rcode;
1626 }
1627
1628 #if ENABLE_HUSH_JOB
1629 static int checkjobs_and_fg_shell(struct pipe* fg_pipe)
1630 {
1631         pid_t p;
1632         int rcode = checkjobs(fg_pipe);
1633         /* Job finished, move the shell to the foreground */
1634         p = getpgid(0);
1635         debug_printf("fg'ing ourself: getpgid(0)=%d\n", (int)p);
1636         if (tcsetpgrp(interactive_fd, p) && errno != ENOTTY)
1637                 bb_perror_msg("tcsetpgrp-4a");
1638         return rcode;
1639 }
1640 #endif
1641
1642 /* run_pipe_real() starts all the jobs, but doesn't wait for anything
1643  * to finish.  See checkjobs().
1644  *
1645  * return code is normally -1, when the caller has to wait for children
1646  * to finish to determine the exit status of the pipe.  If the pipe
1647  * is a simple builtin command, however, the action is done by the
1648  * time run_pipe_real returns, and the exit code is provided as the
1649  * return value.
1650  *
1651  * The input of the pipe is always stdin, the output is always
1652  * stdout.  The outpipe[] mechanism in BusyBox-0.48 lash is bogus,
1653  * because it tries to avoid running the command substitution in
1654  * subshell, when that is in fact necessary.  The subshell process
1655  * now has its stdout directed to the input of the appropriate pipe,
1656  * so this routine is noticeably simpler.
1657  */
1658 static int run_pipe_real(struct pipe *pi)
1659 {
1660         int i;
1661         int nextin, nextout;
1662         int pipefds[2];                         /* pipefds[0] is for reading */
1663         struct child_prog *child;
1664         const struct built_in_command *x;
1665         char *p;
1666         /* it is not always needed, but we aim to smaller code */
1667         int squirrel[] = { -1, -1, -1 };
1668         int rcode;
1669         const int single_fg = (pi->num_progs == 1 && pi->followup != PIPE_BG);
1670
1671         debug_printf_exec("run_pipe_real start: single_fg=%d\n", single_fg);
1672
1673         nextin = 0;
1674 #if ENABLE_HUSH_JOB
1675         pi->pgrp = -1;
1676 #endif
1677         pi->running_progs = 1;
1678         pi->stopped_progs = 0;
1679
1680         /* Check if this is a simple builtin (not part of a pipe).
1681          * Builtins within pipes have to fork anyway, and are handled in
1682          * pseudo_exec.  "echo foo | read bar" doesn't work on bash, either.
1683          */
1684         child = &(pi->progs[0]);
1685         if (single_fg && child->group && child->subshell == 0) {
1686                 debug_printf("non-subshell grouping\n");
1687                 setup_redirects(child, squirrel);
1688                 debug_printf_exec(": run_list_real\n");
1689                 rcode = run_list_real(child->group);
1690                 restore_redirects(squirrel);
1691                 debug_printf_exec("run_pipe_real return %d\n", rcode);
1692                 return rcode;
1693         }
1694
1695         if (single_fg && child->argv != NULL) {
1696                 char **argv_expanded;
1697                 char **argv = child->argv;
1698
1699                 for (i = 0; is_assignment(argv[i]); i++)
1700                         continue;
1701                 if (i != 0 && argv[i] == NULL) {
1702                         /* assignments, but no command: set the local environment */
1703                         for (i = 0; argv[i] != NULL; i++) {
1704                                 /* Ok, this case is tricky.  We have to decide if this is a
1705                                  * local variable, or an already exported variable.  If it is
1706                                  * already exported, we have to export the new value.  If it is
1707                                  * not exported, we need only set this as a local variable.
1708                                  * This junk is all to decide whether or not to export this
1709                                  * variable. */
1710                                 int export_me = 0;
1711                                 char *name, *value;
1712                                 name = xstrdup(argv[i]);
1713                                 debug_printf("local environment set: %s\n", name);
1714                                 value = strchr(name, '=');
1715                                 if (value)
1716                                         *value = '\0';
1717                                 if (get_local_var(name)) {
1718                                         export_me = 1;
1719                                 }
1720                                 free(name);
1721                                 p = expand_variables_to_string(argv[i]);
1722                                 set_local_var(p, export_me);
1723                                 if (p != argv[i])
1724                                         free(p);
1725                         }
1726                         return EXIT_SUCCESS;   /* don't worry about errors in set_local_var() yet */
1727                 }
1728                 for (i = 0; is_assignment(argv[i]); i++) {
1729                         p = expand_variables_to_string(argv[i]);
1730                         if (p != argv[i]) {
1731                                 //sp: child->sp--;
1732                                 putenv(p);
1733                         } else {
1734                                 putenv(xstrdup(p));
1735                         }
1736                 }
1737                 for (x = bltins; x->cmd; x++) {
1738                         if (strcmp(argv[i], x->cmd) == 0) {
1739                                 if (x->function == builtin_exec && argv[i+1] == NULL) {
1740                                         debug_printf("magic exec\n");
1741                                         setup_redirects(child, NULL);
1742                                         return EXIT_SUCCESS;
1743                                 }
1744                                 debug_printf("builtin inline %s\n", argv[0]);
1745                                 /* XXX setup_redirects acts on file descriptors, not FILEs.
1746                                  * This is perfect for work that comes after exec().
1747                                  * Is it really safe for inline use?  Experimentally,
1748                                  * things seem to work with glibc. */
1749                                 setup_redirects(child, squirrel);
1750                                 debug_printf_exec(": builtin '%s' '%s'...\n", x->cmd, argv[i+1]);
1751                                 //sp: if (child->sp) /* btw we can do it unconditionally... */
1752                                 argv_expanded = expand_variables_to_list(argv + i);
1753                                 rcode = x->function(argv_expanded);
1754                                 free(argv_expanded);
1755                                 restore_redirects(squirrel);
1756                                 debug_printf_exec("run_pipe_real return %d\n", rcode);
1757                                 return rcode;
1758                         }
1759                 }
1760 #if ENABLE_FEATURE_SH_STANDALONE
1761                 {
1762                         const struct bb_applet *a = find_applet_by_name(argv[i]);
1763                         if (a && a->nofork) {
1764                                 setup_redirects(child, squirrel);
1765                                 save_nofork_data(&nofork_save);
1766                                 argv_expanded = argv + i;
1767                                 //sp: if (child->sp)
1768                                 argv_expanded = expand_variables_to_list(argv + i);
1769                                 debug_printf_exec(": run_nofork_applet '%s' '%s'...\n", argv_expanded[0], argv_expanded[1]);
1770                                 rcode = run_nofork_applet_prime(&nofork_save, a, argv_expanded);
1771                                 free(argv_expanded);
1772                                 restore_redirects(squirrel);
1773                                 debug_printf_exec("run_pipe_real return %d\n", rcode);
1774                                 return rcode;
1775                         }
1776                 }
1777 #endif
1778         }
1779
1780         /* Going to fork a child per each pipe member */
1781         pi->running_progs = 0;
1782
1783         /* Disable job control signals for shell (parent) and
1784          * for initial child code after fork */
1785         set_jobctrl_sighandler(SIG_IGN);
1786
1787         for (i = 0; i < pi->num_progs; i++) {
1788                 child = &(pi->progs[i]);
1789                 if (child->argv)
1790                         debug_printf_exec(": pipe member '%s' '%s'...\n", child->argv[0], child->argv[1]);
1791                 else
1792                         debug_printf_exec(": pipe member with no argv\n");
1793
1794                 /* pipes are inserted between pairs of commands */
1795                 if ((i + 1) < pi->num_progs) {
1796                         if (pipe(pipefds) < 0)
1797                                 bb_perror_msg_and_die("pipe");
1798                         nextout = pipefds[1];
1799                 } else {
1800                         nextout = 1;
1801                         pipefds[0] = -1;
1802                 }
1803
1804                 /* XXX test for failed fork()? */
1805 #if BB_MMU
1806                 child->pid = fork();
1807 #else
1808                 child->pid = vfork();
1809 #endif
1810                 if (!child->pid) { /* child */
1811                         /* Every child adds itself to new process group
1812                          * with pgid == pid of first child in pipe */
1813 #if ENABLE_HUSH_JOB
1814                         if (interactive_fd) {
1815                                 /* Don't do pgrp restore anymore on fatal signals */
1816                                 set_fatal_sighandler(SIG_DFL);
1817                                 if (pi->pgrp < 0) /* true for 1st process only */
1818                                         pi->pgrp = getpid();
1819                                 if (setpgid(0, pi->pgrp) == 0 && pi->followup != PIPE_BG) {
1820                                         /* We do it in *every* child, not just first,
1821                                          * to avoid races */
1822                                         tcsetpgrp(interactive_fd, pi->pgrp);
1823                                 }
1824                         }
1825 #endif
1826                         /* in non-interactive case fatal sigs are already SIG_DFL */
1827                         close_all();
1828                         if (nextin != 0) {
1829                                 dup2(nextin, 0);
1830                                 close(nextin);
1831                         }
1832                         if (nextout != 1) {
1833                                 dup2(nextout, 1);
1834                                 close(nextout);
1835                         }
1836                         if (pipefds[0] != -1) {
1837                                 close(pipefds[0]);  /* opposite end of our output pipe */
1838                         }
1839                         /* Like bash, explicit redirects override pipes,
1840                          * and the pipe fd is available for dup'ing. */
1841                         setup_redirects(child, NULL);
1842
1843                         /* Restore default handlers just prior to exec */
1844                         set_jobctrl_sighandler(SIG_DFL);
1845                         set_misc_sighandler(SIG_DFL);
1846                         signal(SIGCHLD, SIG_DFL);
1847                         pseudo_exec(child);
1848                 }
1849
1850                 pi->running_progs++;
1851
1852 #if ENABLE_HUSH_JOB
1853                 /* Second and next children need to know pid of first one */
1854                 if (pi->pgrp < 0)
1855                         pi->pgrp = child->pid;
1856 #endif
1857                 if (nextin != 0)
1858                         close(nextin);
1859                 if (nextout != 1)
1860                         close(nextout);
1861
1862                 /* If there isn't another process, nextin is garbage
1863                    but it doesn't matter */
1864                 nextin = pipefds[0];
1865         }
1866         debug_printf_exec("run_pipe_real return -1\n");
1867         return -1;
1868 }
1869
1870 #ifndef debug_print_tree
1871 static void debug_print_tree(struct pipe *pi, int lvl)
1872 {
1873         static const char *PIPE[] = {
1874                 [PIPE_SEQ] = "SEQ",
1875                 [PIPE_AND] = "AND",
1876                 [PIPE_OR ] = "OR" ,
1877                 [PIPE_BG ] = "BG" ,
1878         };
1879         static const char *RES[] = {
1880                 [RES_NONE ] = "NONE" ,
1881                 [RES_IF   ] = "IF"   ,
1882                 [RES_THEN ] = "THEN" ,
1883                 [RES_ELIF ] = "ELIF" ,
1884                 [RES_ELSE ] = "ELSE" ,
1885                 [RES_FI   ] = "FI"   ,
1886                 [RES_FOR  ] = "FOR"  ,
1887                 [RES_WHILE] = "WHILE",
1888                 [RES_UNTIL] = "UNTIL",
1889                 [RES_DO   ] = "DO"   ,
1890                 [RES_DONE ] = "DONE" ,
1891                 [RES_XXXX ] = "XXXX" ,
1892                 [RES_IN   ] = "IN"   ,
1893                 [RES_SNTX ] = "SNTX" ,
1894         };
1895
1896         int pin, prn;
1897         pin = 0;
1898         while (pi) {
1899                 fprintf(stderr, "%*spipe %d r_mode=%s followup=%d %s\n", lvl*2, "",
1900                                 pin, RES[pi->r_mode], pi->followup, PIPE[pi->followup]);
1901                 prn = 0;
1902                 while (prn < pi->num_progs) {
1903                         struct child_prog *child = &pi->progs[prn];
1904                         char **argv = child->argv;
1905
1906                         fprintf(stderr, "%*s prog %d", lvl*2, "", prn);
1907                         if (child->group) {
1908                                 fprintf(stderr, " group %s: (argv=%p)\n",
1909                                                 (child->subshell ? "()" : "{}"),
1910                                                 argv);
1911                                 debug_print_tree(child->group, lvl+1);
1912                                 prn++;
1913                                 continue;
1914                         }
1915                         if (argv) while (*argv) {
1916                                 fprintf(stderr, " '%s'", *argv);
1917                                 argv++;
1918                         }
1919                         fprintf(stderr, "\n");
1920                         prn++;
1921                 }
1922                 pi = pi->next;
1923                 pin++;
1924         }
1925 }
1926 #endif
1927
1928 /* NB: called by pseudo_exec, and therefore must not modify any
1929  * global data until exec/_exit (we can be a child after vfork!) */
1930 static int run_list_real(struct pipe *pi)
1931 {
1932         char *for_varname = NULL;
1933         char **for_lcur = NULL;
1934         char **for_list = NULL;
1935         struct pipe *rpipe;
1936         int flag_rep = 0;
1937         int save_num_progs;
1938         int flag_skip = 1;
1939         int rcode = 0; /* probably for gcc only */
1940         int flag_restore = 0;
1941         int if_code = 0, next_if_code = 0;  /* need double-buffer to handle elif */
1942         reserved_style rmode, skip_more_in_this_rmode = RES_XXXX;
1943
1944         debug_printf_exec("run_list_real start lvl %d\n", run_list_level + 1);
1945
1946         /* check syntax for "for" */
1947         for (rpipe = pi; rpipe; rpipe = rpipe->next) {
1948                 if ((rpipe->r_mode == RES_IN || rpipe->r_mode == RES_FOR)
1949                  && (rpipe->next == NULL)
1950                 ) {
1951                         syntax(); /* unterminated FOR (no IN or no commands after IN) */
1952                         debug_printf_exec("run_list_real lvl %d return 1\n", run_list_level);
1953                         return 1;
1954                 }
1955                 if ((rpipe->r_mode == RES_IN && rpipe->next->r_mode == RES_IN && rpipe->next->progs[0].argv != NULL)
1956                  || (rpipe->r_mode == RES_FOR && rpipe->next->r_mode != RES_IN)
1957                 ) {
1958                         /* TODO: what is tested in the first condition? */
1959                         syntax(); /* 2nd: malformed FOR (not followed by IN) */
1960                         debug_printf_exec("run_list_real lvl %d return 1\n", run_list_level);
1961                         return 1;
1962                 }
1963         }
1964
1965 #if ENABLE_HUSH_JOB
1966         /* Example of nested list: "while true; do { sleep 1 | exit 2; } done".
1967          * We are saving state before entering outermost list ("while...done")
1968          * so that ctrl-Z will correctly background _entire_ outermost list,
1969          * not just a part of it (like "sleep 1 | exit 2") */
1970         if (++run_list_level == 1 && interactive_fd) {
1971                 if (sigsetjmp(toplevel_jb, 1)) {
1972                         /* ctrl-Z forked and we are parent; or ctrl-C.
1973                          * Sighandler has longjmped us here */
1974                         signal(SIGINT, SIG_IGN);
1975                         signal(SIGTSTP, SIG_IGN);
1976                         /* Restore level (we can be coming from deep inside
1977                          * nested levels) */
1978                         run_list_level = 1;
1979 #if ENABLE_FEATURE_SH_STANDALONE
1980                         if (nofork_save.saved) { /* if save area is valid */
1981                                 debug_printf_jobs("exiting nofork early\n");
1982                                 restore_nofork_data(&nofork_save);
1983                         }
1984 #endif
1985                         if (ctrl_z_flag) {
1986                                 /* ctrl-Z has forked and stored pid of the child in pi->pid.
1987                                  * Remember this child as background job */
1988                                 insert_bg_job(pi);
1989                         } else {
1990                                 /* ctrl-C. We just stop doing whatever we were doing */
1991                                 putchar('\n');
1992                         }
1993                         rcode = 0;
1994                         goto ret;
1995                 }
1996                 /* ctrl-Z handler will store pid etc in pi */
1997                 toplevel_list = pi;
1998                 ctrl_z_flag = 0;
1999 #if ENABLE_FEATURE_SH_STANDALONE
2000                 nofork_save.saved = 0; /* in case we will run a nofork later */
2001 #endif
2002                 signal_SA_RESTART(SIGTSTP, handler_ctrl_z);
2003                 signal(SIGINT, handler_ctrl_c);
2004         }
2005 #endif
2006
2007         for (; pi; pi = flag_restore ? rpipe : pi->next) {
2008                 rmode = pi->r_mode;
2009                 if (rmode == RES_WHILE || rmode == RES_UNTIL || rmode == RES_FOR) {
2010                         flag_restore = 0;
2011                         if (!rpipe) {
2012                                 flag_rep = 0;
2013                                 rpipe = pi;
2014                         }
2015                 }
2016                 debug_printf_exec(": rmode=%d if_code=%d next_if_code=%d skip_more=%d\n",
2017                                 rmode, if_code, next_if_code, skip_more_in_this_rmode);
2018                 if (rmode == skip_more_in_this_rmode && flag_skip) {
2019                         if (pi->followup == PIPE_SEQ)
2020                                 flag_skip = 0;
2021                         continue;
2022                 }
2023                 flag_skip = 1;
2024                 skip_more_in_this_rmode = RES_XXXX;
2025                 if (rmode == RES_THEN || rmode == RES_ELSE)
2026                         if_code = next_if_code;
2027                 if (rmode == RES_THEN && if_code)
2028                         continue;
2029                 if (rmode == RES_ELSE && !if_code)
2030                         continue;
2031                 if (rmode == RES_ELIF && !if_code)
2032                         break;
2033                 if (rmode == RES_FOR && pi->num_progs) {
2034                         if (!for_lcur) {
2035                                 /* if no variable values after "in" we skip "for" */
2036                                 if (!pi->next->progs->argv)
2037                                         continue;
2038                                 /* create list of variable values */
2039                                 for_list = expand_variables_to_list(pi->next->progs->argv);
2040                                 for_lcur = for_list;
2041                                 for_varname = pi->progs->argv[0];
2042                                 pi->progs->argv[0] = NULL;
2043                                 flag_rep = 1;
2044                         }
2045                         free(pi->progs->argv[0]);
2046                         if (!*for_lcur) {
2047                                 free(for_list);
2048                                 for_lcur = NULL;
2049                                 flag_rep = 0;
2050                                 pi->progs->argv[0] = for_varname;
2051                                 pi->progs->glob_result.gl_pathv[0] = pi->progs->argv[0];
2052                                 continue;
2053                         }
2054                         /* insert next value from for_lcur */
2055                         /* vda: does it need escaping? */
2056                         pi->progs->argv[0] = xasprintf("%s=%s", for_varname, *for_lcur++);
2057                         pi->progs->glob_result.gl_pathv[0] = pi->progs->argv[0];
2058                 }
2059                 if (rmode == RES_IN)
2060                         continue;
2061                 if (rmode == RES_DO) {
2062                         if (!flag_rep)
2063                                 continue;
2064                 }
2065                 if (rmode == RES_DONE) {
2066                         if (flag_rep) {
2067                                 flag_restore = 1;
2068                         } else {
2069                                 rpipe = NULL;
2070                         }
2071                 }
2072                 if (pi->num_progs == 0)
2073                         continue;
2074                 save_num_progs = pi->num_progs; /* save number of programs */
2075                 debug_printf_exec(": run_pipe_real with %d members\n", pi->num_progs);
2076                 rcode = run_pipe_real(pi);
2077                 if (rcode != -1) {
2078                         /* We only ran a builtin: rcode was set by the return value
2079                          * of run_pipe_real(), and we don't need to wait for anything. */
2080                 } else if (pi->followup == PIPE_BG) {
2081                         /* What does bash do with attempts to background builtins? */
2082
2083                         /* Even bash 3.2 doesn't do that well with nested bg:
2084                          * try "{ { sleep 10; echo DEEP; } & echo HERE; } &".
2085                          * I'm considering NOT treating inner bgs as jobs -
2086                          * thus maybe "if (run_list_level == 1 && pi->followup == PIPE_BG)"
2087                          * above? */
2088 #if ENABLE_HUSH_JOB
2089                         insert_bg_job(pi);
2090 #endif
2091                         rcode = EXIT_SUCCESS;
2092                 } else {
2093 #if ENABLE_HUSH_JOB
2094                         /* Paranoia, just "interactive_fd" should be enough */
2095                         if (run_list_level == 1 && interactive_fd) {
2096                                 rcode = checkjobs_and_fg_shell(pi);
2097                         } else
2098 #endif
2099                         {
2100                                 rcode = checkjobs(pi);
2101                         }
2102                         debug_printf_exec(": checkjobs returned %d\n", rcode);
2103                 }
2104                 debug_printf_exec(": setting last_return_code=%d\n", rcode);
2105                 last_return_code = rcode;
2106                 pi->num_progs = save_num_progs; /* restore number of programs */
2107                 if (rmode == RES_IF || rmode == RES_ELIF)
2108                         next_if_code = rcode;  /* can be overwritten a number of times */
2109                 if (rmode == RES_WHILE)
2110                         flag_rep = !last_return_code;
2111                 if (rmode == RES_UNTIL)
2112                         flag_rep = last_return_code;
2113                 if ((rcode == EXIT_SUCCESS && pi->followup == PIPE_OR)
2114                  || (rcode != EXIT_SUCCESS && pi->followup == PIPE_AND)
2115                 ) {
2116                         skip_more_in_this_rmode = rmode;
2117                 }
2118                 checkjobs(NULL);
2119         }
2120
2121 #if ENABLE_HUSH_JOB
2122         if (ctrl_z_flag) {
2123                 /* ctrl-Z forked somewhere in the past, we are the child,
2124                  * and now we completed running the list. Exit. */
2125                 exit(rcode);
2126         }
2127  ret:
2128         run_list_level--;
2129 #endif
2130         debug_printf_exec("run_list_real lvl %d return %d\n", run_list_level + 1, rcode);
2131         return rcode;
2132 }
2133
2134 /* return code is the exit status of the pipe */
2135 static int free_pipe(struct pipe *pi, int indent)
2136 {
2137         char **p;
2138         struct child_prog *child;
2139         struct redir_struct *r, *rnext;
2140         int a, i, ret_code = 0;
2141
2142         if (pi->stopped_progs > 0)
2143                 return ret_code;
2144         debug_printf_clean("%s run pipe: (pid %d)\n", indenter(indent), getpid());
2145         for (i = 0; i < pi->num_progs; i++) {
2146                 child = &pi->progs[i];
2147                 debug_printf_clean("%s  command %d:\n", indenter(indent), i);
2148                 if (child->argv) {
2149                         for (a = 0, p = child->argv; *p; a++, p++) {
2150                                 debug_printf_clean("%s   argv[%d] = %s\n", indenter(indent), a, *p);
2151                         }
2152                         globfree(&child->glob_result);
2153                         child->argv = NULL;
2154                 } else if (child->group) {
2155                         debug_printf_clean("%s   begin group (subshell:%d)\n", indenter(indent), child->subshell);
2156                         ret_code = free_pipe_list(child->group, indent+3);
2157                         debug_printf_clean("%s   end group\n", indenter(indent));
2158                 } else {
2159                         debug_printf_clean("%s   (nil)\n", indenter(indent));
2160                 }
2161                 for (r = child->redirects; r; r = rnext) {
2162                         debug_printf_clean("%s   redirect %d%s", indenter(indent), r->fd, redir_table[r->type].descrip);
2163                         if (r->dup == -1) {
2164                                 /* guard against the case >$FOO, where foo is unset or blank */
2165                                 if (r->word.gl_pathv) {
2166                                         debug_printf_clean(" %s\n", *r->word.gl_pathv);
2167                                         globfree(&r->word);
2168                                 }
2169                         } else {
2170                                 debug_printf_clean("&%d\n", r->dup);
2171                         }
2172                         rnext = r->next;
2173                         free(r);
2174                 }
2175                 child->redirects = NULL;
2176         }
2177         free(pi->progs);   /* children are an array, they get freed all at once */
2178         pi->progs = NULL;
2179 #if ENABLE_HUSH_JOB
2180         free(pi->cmdtext);
2181         pi->cmdtext = NULL;
2182 #endif
2183         return ret_code;
2184 }
2185
2186 static int free_pipe_list(struct pipe *head, int indent)
2187 {
2188         int rcode = 0;   /* if list has no members */
2189         struct pipe *pi, *next;
2190
2191         for (pi = head; pi; pi = next) {
2192                 debug_printf_clean("%s pipe reserved mode %d\n", indenter(indent), pi->r_mode);
2193                 rcode = free_pipe(pi, indent);
2194                 debug_printf_clean("%s pipe followup code %d\n", indenter(indent), pi->followup);
2195                 next = pi->next;
2196                 /*pi->next = NULL;*/
2197                 free(pi);
2198         }
2199         return rcode;
2200 }
2201
2202 /* Select which version we will use */
2203 static int run_list(struct pipe *pi)
2204 {
2205         int rcode = 0;
2206         debug_printf_exec("run_list entered\n");
2207         if (fake_mode == 0) {
2208                 debug_printf_exec(": run_list_real with %d members\n", pi->num_progs);
2209                 rcode = run_list_real(pi);
2210         }
2211         /* free_pipe_list has the side effect of clearing memory.
2212          * In the long run that function can be merged with run_list_real,
2213          * but doing that now would hobble the debugging effort. */
2214         free_pipe_list(pi, 0);
2215         debug_printf_exec("run_list return %d\n", rcode);
2216         return rcode;
2217 }
2218
2219 /* The API for glob is arguably broken.  This routine pushes a non-matching
2220  * string into the output structure, removing non-backslashed backslashes.
2221  * If someone can prove me wrong, by performing this function within the
2222  * original glob(3) api, feel free to rewrite this routine into oblivion.
2223  * Return code (0 vs. GLOB_NOSPACE) matches glob(3).
2224  * XXX broken if the last character is '\\', check that before calling.
2225  */
2226 static int globhack(const char *src, int flags, glob_t *pglob)
2227 {
2228         int cnt = 0, pathc;
2229         const char *s;
2230         char *dest;
2231         for (cnt = 1, s = src; s && *s; s++) {
2232                 if (*s == '\\') s++;
2233                 cnt++;
2234         }
2235         dest = xmalloc(cnt);
2236         if (!(flags & GLOB_APPEND)) {
2237                 pglob->gl_pathv = NULL;
2238                 pglob->gl_pathc = 0;
2239                 pglob->gl_offs = 0;
2240                 pglob->gl_offs = 0;
2241         }
2242         pathc = ++pglob->gl_pathc;
2243         pglob->gl_pathv = xrealloc(pglob->gl_pathv, (pathc+1) * sizeof(*pglob->gl_pathv));
2244         pglob->gl_pathv[pathc-1] = dest;
2245         pglob->gl_pathv[pathc] = NULL;
2246         for (s = src; s && *s; s++, dest++) {
2247                 if (*s == '\\') s++;
2248                 *dest = *s;
2249         }
2250         *dest = '\0';
2251         return 0;
2252 }
2253
2254 /* XXX broken if the last character is '\\', check that before calling */
2255 static int glob_needed(const char *s)
2256 {
2257         for (; *s; s++) {
2258                 if (*s == '\\') s++;
2259                 if (strchr("*[?", *s)) return 1;
2260         }
2261         return 0;
2262 }
2263
2264 static int xglob(o_string *dest, int flags, glob_t *pglob)
2265 {
2266         int gr;
2267
2268         /* short-circuit for null word */
2269         /* we can code this better when the debug_printf's are gone */
2270         if (dest->length == 0) {
2271                 if (dest->nonnull) {
2272                         /* bash man page calls this an "explicit" null */
2273                         gr = globhack(dest->data, flags, pglob);
2274                         debug_printf("globhack returned %d\n", gr);
2275                 } else {
2276                         return 0;
2277                 }
2278         } else if (glob_needed(dest->data)) {
2279                 gr = glob(dest->data, flags, NULL, pglob);
2280                 debug_printf("glob returned %d\n", gr);
2281                 if (gr == GLOB_NOMATCH) {
2282                         /* quote removal, or more accurately, backslash removal */
2283                         gr = globhack(dest->data, flags, pglob);
2284                         debug_printf("globhack returned %d\n", gr);
2285                 }
2286         } else {
2287                 gr = globhack(dest->data, flags, pglob);
2288                 debug_printf("globhack returned %d\n", gr);
2289         }
2290         if (gr == GLOB_NOSPACE)
2291                 bb_error_msg_and_die("out of memory during glob");
2292         if (gr != 0) { /* GLOB_ABORTED ? */
2293                 bb_error_msg("glob(3) error %d", gr);
2294         }
2295         /* globprint(glob_target); */
2296         return gr;
2297 }
2298
2299 /* expand_variables_to_list() takes a list of strings, expands
2300  * all variable references within and returns a pointer to
2301  * a list of expanded strings, possibly with larger number
2302  * of strings. (Think VAR="a b"; echo $VAR).
2303  * This new list is allocated as a single malloc block.
2304  * NULL-terminated list of char* pointers is at the beginning of it,
2305  * followed by strings themself.
2306  * Caller can deallocate entire list by single free(list). */
2307
2308 /* Helpers first:
2309  * count_XXX estimates size of the block we need. It's okay
2310  * to over-estimate sizes a bit, if it makes code simpler */
2311 static int count_ifs(const char *str)
2312 {
2313         int cnt = 0;
2314         debug_printf_expand("count_ifs('%s') ifs='%s'", str, ifs);
2315         while (1) {
2316                 str += strcspn(str, ifs);
2317                 if (!*str) break;
2318                 str++; /* str += strspn(str, ifs); */
2319                 cnt++; /* cnt += strspn(str, ifs); - but this code is larger */
2320         }
2321         debug_printf_expand(" return %d\n", cnt);
2322         return cnt;
2323 }
2324
2325 static void count_var_expansion_space(int *countp, int *lenp, char *arg)
2326 {
2327         char first_ch;
2328         int i;
2329         int len = *lenp;
2330         int count = *countp;
2331         const char *val;
2332         char *p;
2333
2334         while ((p = strchr(arg, SPECIAL_VAR_SYMBOL))) {
2335                 len += p - arg;
2336                 arg = ++p;
2337                 p = strchr(p, SPECIAL_VAR_SYMBOL);
2338                 first_ch = arg[0];
2339
2340                 switch (first_ch & 0x7f) {
2341                 /* high bit in 1st_ch indicates that var is double-quoted */
2342                 case '$': /* pid */
2343                 case '!': /* bg pid */
2344                 case '?': /* exitcode */
2345                 case '#': /* argc */
2346                         len += sizeof(int)*3 + 1; /* enough for int */
2347                         break;
2348                 case '*':
2349                 case '@':
2350                         for (i = 1; i < global_argc; i++) {
2351                                 len += strlen(global_argv[i]) + 1;
2352                                 count++;
2353                                 if (!(first_ch & 0x80))
2354                                         count += count_ifs(global_argv[i]);
2355                         }
2356                         break;
2357                 default:
2358                         *p = '\0';
2359                         arg[0] = first_ch & 0x7f;
2360                         if (isdigit(arg[0])) {
2361                                 i = xatoi_u(arg);
2362                                 val = NULL;
2363                                 if (i < global_argc)
2364                                         val = global_argv[i];
2365                         } else
2366                                 val = lookup_param(arg);
2367                         arg[0] = first_ch;
2368                         *p = SPECIAL_VAR_SYMBOL;
2369
2370                         if (val) {
2371                                 len += strlen(val) + 1;
2372                                 if (!(first_ch & 0x80))
2373                                         count += count_ifs(val);
2374                         }
2375                 }
2376                 arg = ++p;
2377         }
2378
2379         len += strlen(arg) + 1;
2380         count++;
2381         *lenp = len;
2382         *countp = count;
2383 }
2384
2385 /* Store given string, finalizing the word and starting new one whenever
2386  * we encounter ifs char(s). This is used for expanding variable values.
2387  * End-of-string does NOT finalize word: think about 'echo -$VAR-' */
2388 static int expand_on_ifs(char **list, int n, char **posp, const char *str)
2389 {
2390         char *pos = *posp;
2391         while (1) {
2392                 int word_len = strcspn(str, ifs);
2393                 if (word_len) {
2394                         memcpy(pos, str, word_len); /* store non-ifs chars */
2395                         pos += word_len;
2396                         str += word_len;
2397                 }
2398                 if (!*str)  /* EOL - do not finalize word */
2399                         break;
2400                 *pos++ = '\0';
2401                 if (n) debug_printf_expand("expand_on_ifs finalized list[%d]=%p '%s' "
2402                         "strlen=%d next=%p pos=%p\n", n-1, list[n-1], list[n-1],
2403                         strlen(list[n-1]), list[n-1] + strlen(list[n-1]) + 1, pos);
2404                 list[n++] = pos;
2405                 str += strspn(str, ifs); /* skip ifs chars */
2406         }
2407         *posp = pos;
2408         return n;
2409 }
2410
2411 /* Expand all variable references in given string, adding words to list[]
2412  * at n, n+1,... positions. Return updated n (so that list[n] is next one
2413  * to be filled). This routine is extremely tricky: has to deal with
2414  * variables/parameters with whitespace, $* and $@, and constructs like
2415  * 'echo -$*-'. If you play here, you must run testsuite afterwards! */
2416 /* NB: another bug is that we cannot detect empty strings yet:
2417  * "" or $empty"" expands to zero words, has to expand to empty word */
2418 static int expand_vars_to_list(char **list, int n, char **posp, char *arg, char or_mask)
2419 {
2420         /* or_mask is either 0 (normal case) or 0x80
2421          * (expansion of right-hand side of assignment == 1-element expand) */
2422
2423         char first_ch, ored_ch;
2424         int i;
2425         const char *val;
2426         char *p;
2427         char *pos = *posp;
2428
2429         ored_ch = 0;
2430
2431         if (n) debug_printf_expand("expand_vars_to_list finalized list[%d]=%p '%s' "
2432                 "strlen=%d next=%p pos=%p\n", n-1, list[n-1], list[n-1],
2433                 strlen(list[n-1]), list[n-1] + strlen(list[n-1]) + 1, pos);
2434         list[n++] = pos;
2435
2436         while ((p = strchr(arg, SPECIAL_VAR_SYMBOL))) {
2437                 memcpy(pos, arg, p - arg);
2438                 pos += (p - arg);
2439                 arg = ++p;
2440                 p = strchr(p, SPECIAL_VAR_SYMBOL);
2441
2442                 first_ch = arg[0] | or_mask; /* forced to "quoted" if or_mask = 0x80 */
2443                 ored_ch |= first_ch;
2444                 val = NULL;
2445                 switch (first_ch & 0x7f) {
2446                 /* Highest bit in first_ch indicates that var is double-quoted */
2447                 case '$': /* pid */
2448                         /* FIXME: (echo $$) should still print pid of main shell */
2449                         val = utoa(getpid());
2450                         break;
2451                 case '!': /* bg pid */
2452                         val = last_bg_pid ? utoa(last_bg_pid) : (char*)"";
2453                         break;
2454                 case '?': /* exitcode */
2455                         val = utoa(last_return_code);
2456                         break;
2457                 case '#': /* argc */
2458                         val = utoa(global_argc ? global_argc-1 : 0);
2459                         break;
2460                 case '*':
2461                 case '@':
2462                         i = 1;
2463                         if (!(first_ch & 0x80)) { /* unquoted $* or $@ */
2464                                 while (i < global_argc) {
2465                                         n = expand_on_ifs(list, n, &pos, global_argv[i]);
2466                                         debug_printf_expand("expand_vars_to_list: argv %d (last %d)\n", i, global_argc-1);
2467                                         if (global_argv[i++][0] && i < global_argc) {
2468                                                 /* this argv[] is not empty and not last:
2469                                                  * put terminating NUL, start new word */
2470                                                 *pos++ = '\0';
2471                                                 if (n) debug_printf_expand("expand_vars_to_list 2 finalized list[%d]=%p '%s' "
2472                                                         "strlen=%d next=%p pos=%p\n", n-1, list[n-1], list[n-1],
2473                                                         strlen(list[n-1]), list[n-1] + strlen(list[n-1]) + 1, pos);
2474                                                 list[n++] = pos;
2475                                         }
2476                                 }
2477                         } else
2478                         /* If or_mask is nonzero, we handle assignment 'a=....$@.....'
2479                          * and in this case should theat it like '$*' */
2480                         if (first_ch == ('@'|0x80) && !or_mask) { /* quoted $@ */
2481                                 while (1) {
2482                                         strcpy(pos, global_argv[i]);
2483                                         pos += strlen(global_argv[i]);
2484                                         if (++i >= global_argc)
2485                                                 break;
2486                                         *pos++ = '\0';
2487                                         if (n) debug_printf_expand("expand_vars_to_list 3 finalized list[%d]=%p '%s' "
2488                                                 "strlen=%d next=%p pos=%p\n", n-1, list[n-1], list[n-1],
2489                                                         strlen(list[n-1]), list[n-1] + strlen(list[n-1]) + 1, pos);
2490                                         list[n++] = pos;
2491                                 }
2492                         } else { /* quoted $*: add as one word */
2493                                 while (1) {
2494                                         strcpy(pos, global_argv[i]);
2495                                         pos += strlen(global_argv[i]);
2496                                         if (++i >= global_argc)
2497                                                 break;
2498                                         if (ifs[0])
2499                                                 *pos++ = ifs[0];
2500                                 }
2501                         }
2502                         break;
2503                 default:
2504                         *p = '\0';
2505                         arg[0] = first_ch & 0x7f;
2506                         if (isdigit(arg[0])) {
2507                                 i = xatoi_u(arg);
2508                                 val = NULL;
2509                                 if (i < global_argc)
2510                                         val = global_argv[i];
2511                         } else
2512                                 val = lookup_param(arg);
2513                         arg[0] = first_ch;
2514                         *p = SPECIAL_VAR_SYMBOL;
2515                         if (!(first_ch & 0x80)) { /* unquoted $VAR */
2516                                 if (val) {
2517                                         n = expand_on_ifs(list, n, &pos, val);
2518                                         val = NULL;
2519                                 }
2520                         } /* else: quoted $VAR, val will be appended at pos */
2521                 }
2522                 if (val) {
2523                         strcpy(pos, val);
2524                         pos += strlen(val);
2525                 }
2526                 arg = ++p;
2527         }
2528         debug_printf_expand("expand_vars_to_list adding tail '%s' at %p\n", arg, pos);
2529         strcpy(pos, arg);
2530         pos += strlen(arg) + 1;
2531         if (pos == list[n-1] + 1) { /* expansion is empty */
2532                 if (!(ored_ch & 0x80)) { /* all vars were not quoted... */
2533                         debug_printf_expand("expand_vars_to_list list[%d] empty, going back\n", n);
2534                         pos--;
2535                         n--;
2536                 }
2537         }
2538
2539         *posp = pos;
2540         return n;
2541 }
2542
2543 static char **expand_variables(char **argv, char or_mask)
2544 {
2545         int n;
2546         int count = 1;
2547         int len = 0;
2548         char *pos, **v, **list;
2549
2550         v = argv;
2551         if (!*v) debug_printf_expand("count_var_expansion_space: "
2552                         "argv[0]=NULL count=%d len=%d alloc_space=%d\n",
2553                         count, len, sizeof(char*) * count + len);
2554         while (*v) {
2555                 count_var_expansion_space(&count, &len, *v);
2556                 debug_printf_expand("count_var_expansion_space: "
2557                         "'%s' count=%d len=%d alloc_space=%d\n",
2558                         *v, count, len, sizeof(char*) * count + len);
2559                 v++;
2560         }
2561         len += sizeof(char*) * count; /* total to alloc */
2562         list = xmalloc(len);
2563         pos = (char*)(list + count);
2564         debug_printf_expand("list=%p, list[0] should be %p\n", list, pos);
2565         n = 0;
2566         v = argv;
2567         while (*v)
2568                 n = expand_vars_to_list(list, n, &pos, *v++, or_mask);
2569
2570         if (n) debug_printf_expand("finalized list[%d]=%p '%s' "
2571                 "strlen=%d next=%p pos=%p\n", n-1, list[n-1], list[n-1],
2572                 strlen(list[n-1]), list[n-1] + strlen(list[n-1]) + 1, pos);
2573         list[n] = NULL;
2574
2575 #ifdef DEBUG_EXPAND
2576         {
2577                 int m = 0;
2578                 while (m <= n) {
2579                         debug_printf_expand("list[%d]=%p '%s'\n", m, list[m], list[m]);
2580                         m++;
2581                 }
2582                 debug_printf_expand("used_space=%d\n", pos - (char*)list);
2583         }
2584 #endif
2585         /* To be removed / made conditional later. */
2586         if (pos - (char*)list > len)
2587                 bb_error_msg_and_die("BUG in varexp");
2588         return list;
2589 }
2590
2591 static char **expand_variables_to_list(char **argv)
2592 {
2593         return expand_variables(argv, 0);
2594 }
2595
2596 static char *expand_variables_to_string(const char *str)
2597 {
2598         char *argv[2], **list;
2599
2600         argv[0] = (char*)str;
2601         argv[1] = NULL;
2602         list = expand_variables(argv, 0x80); /* 0x80: make one-element expansion */
2603         /* To be removed / made conditional later. */
2604         if (!list[0] || list[1])
2605                 bb_error_msg_and_die("BUG in varexp");
2606         /* actually, just move string 2*sizeof(char*) bytes back */
2607         strcpy((char*)list, list[0]);
2608         return (char*)list;
2609 }
2610
2611 /* This is used to get/check local shell variables */
2612 static const char *get_local_var(const char *s)
2613 {
2614         struct variables *cur;
2615
2616         if (!s)
2617                 return NULL;
2618         for (cur = top_vars; cur; cur = cur->next) {
2619                 if (strcmp(cur->name, s) == 0)
2620                         return cur->value;
2621         }
2622         return NULL;
2623 }
2624
2625 /* This is used to set local shell variables
2626    flg_export == 0 if only local (not exporting) variable
2627    flg_export == 1 if "new" exporting environ
2628    flg_export > 1  if current startup environ (not call putenv()) */
2629 static int set_local_var(const char *s, int flg_export)
2630 {
2631         char *name, *value;
2632         int result = 0;
2633         struct variables *cur;
2634
2635         name = xstrdup(s);
2636
2637         /* Assume when we enter this function that we are already in
2638          * NAME=VALUE format.  So the first order of business is to
2639          * split 's' on the '=' into 'name' and 'value' */
2640         value = strchr(name, '=');
2641         /*if (value == 0 && ++value == 0) ??? -vda */
2642         if (value == NULL || value[1] == '\0') {
2643                 free(name);
2644                 return -1;
2645         }
2646         *value++ = '\0';
2647
2648         for (cur = top_vars; cur; cur = cur->next) {
2649                 if (strcmp(cur->name, name) == 0) {
2650                         if (strcmp(cur->value, value) == 0) {
2651                                 if (flg_export && !cur->flg_export)
2652                                         cur->flg_export = flg_export;
2653                                 else
2654                                         result++;
2655                         } else if (cur->flg_read_only) {
2656                                 bb_error_msg("%s: readonly variable", name);
2657                                 result = -1;
2658                         } else {
2659                                 if (flg_export > 0 || cur->flg_export > 1)
2660                                         cur->flg_export = 1;
2661                                 free((char*)cur->value);
2662                                 cur->value = xstrdup(value);
2663                         }
2664                         goto skip;
2665                 }
2666         }
2667
2668         cur = xzalloc(sizeof(*cur));
2669         cur->name = xstrdup(name);
2670         cur->value = xstrdup(value);
2671         /*cur->next = 0;*/
2672         cur->flg_export = flg_export;
2673         /*cur->flg_read_only = 0;*/
2674         {
2675                 struct variables *bottom = top_vars;
2676                 while (bottom->next)
2677                         bottom = bottom->next;
2678                 bottom->next = cur;
2679         }
2680  skip:
2681         if (result == 0 && cur->flg_export == 1) {
2682                 *(value-1) = '=';
2683                 result = putenv(name);
2684         } else {
2685                 free(name);
2686                 if (result > 0)            /* equivalent to previous set */
2687                         result = 0;
2688         }
2689         return result;
2690 }
2691
2692 static void unset_local_var(const char *name)
2693 {
2694         struct variables *cur, *next;
2695
2696         if (!name)
2697                 return;
2698         for (cur = top_vars; cur; cur = cur->next) {
2699                 if (strcmp(cur->name, name) == 0) {
2700                         if (cur->flg_read_only) {
2701                                 bb_error_msg("%s: readonly variable", name);
2702                                 return;
2703                         }
2704                         if (cur->flg_export)
2705                                 unsetenv(cur->name);
2706                         free((char*)cur->name);
2707                         free((char*)cur->value);
2708                         next = top_vars;
2709                         while (next->next != cur)
2710                                 next = next->next;
2711                         next->next = cur->next;
2712                         free(cur);
2713                         return;
2714                 }
2715         }
2716 }
2717
2718 static int is_assignment(const char *s)
2719 {
2720         if (!s || !isalpha(*s))
2721                 return 0;
2722         s++;
2723         while (isalnum(*s) || *s == '_')
2724                 s++;
2725         return *s == '=';
2726 }
2727
2728 /* the src parameter allows us to peek forward to a possible &n syntax
2729  * for file descriptor duplication, e.g., "2>&1".
2730  * Return code is 0 normally, 1 if a syntax error is detected in src.
2731  * Resource errors (in xmalloc) cause the process to exit */
2732 static int setup_redirect(struct p_context *ctx, int fd, redir_type style,
2733         struct in_str *input)
2734 {
2735         struct child_prog *child = ctx->child;
2736         struct redir_struct *redir = child->redirects;
2737         struct redir_struct *last_redir = NULL;
2738
2739         /* Create a new redir_struct and drop it onto the end of the linked list */
2740         while (redir) {
2741                 last_redir = redir;
2742                 redir = redir->next;
2743         }
2744         redir = xmalloc(sizeof(struct redir_struct));
2745         redir->next = NULL;
2746         redir->word.gl_pathv = NULL;
2747         if (last_redir) {
2748                 last_redir->next = redir;
2749         } else {
2750                 child->redirects = redir;
2751         }
2752
2753         redir->type = style;
2754         redir->fd = (fd == -1) ? redir_table[style].default_fd : fd;
2755
2756         debug_printf("Redirect type %d%s\n", redir->fd, redir_table[style].descrip);
2757
2758         /* Check for a '2>&1' type redirect */
2759         redir->dup = redirect_dup_num(input);
2760         if (redir->dup == -2) return 1;  /* syntax error */
2761         if (redir->dup != -1) {
2762                 /* Erik had a check here that the file descriptor in question
2763                  * is legit; I postpone that to "run time"
2764                  * A "-" representation of "close me" shows up as a -3 here */
2765                 debug_printf("Duplicating redirect '%d>&%d'\n", redir->fd, redir->dup);
2766         } else {
2767                 /* We do _not_ try to open the file that src points to,
2768                  * since we need to return and let src be expanded first.
2769                  * Set ctx->pending_redirect, so we know what to do at the
2770                  * end of the next parsed word.
2771                  */
2772                 ctx->pending_redirect = redir;
2773         }
2774         return 0;
2775 }
2776
2777 static struct pipe *new_pipe(void)
2778 {
2779         struct pipe *pi;
2780         pi = xzalloc(sizeof(struct pipe));
2781         /*pi->num_progs = 0;*/
2782         /*pi->progs = NULL;*/
2783         /*pi->next = NULL;*/
2784         /*pi->followup = 0;  invalid */
2785         if (RES_NONE)
2786                 pi->r_mode = RES_NONE;
2787         return pi;
2788 }
2789
2790 static void initialize_context(struct p_context *ctx)
2791 {
2792         ctx->pipe = NULL;
2793         ctx->pending_redirect = NULL;
2794         ctx->child = NULL;
2795         ctx->list_head = new_pipe();
2796         ctx->pipe = ctx->list_head;
2797         ctx->res_w = RES_NONE;
2798         ctx->stack = NULL;
2799         ctx->old_flag = 0;
2800         done_command(ctx);   /* creates the memory for working child */
2801 }
2802
2803 /* normal return is 0
2804  * if a reserved word is found, and processed, return 1
2805  * should handle if, then, elif, else, fi, for, while, until, do, done.
2806  * case, function, and select are obnoxious, save those for later.
2807  */
2808 static int reserved_word(o_string *dest, struct p_context *ctx)
2809 {
2810         struct reserved_combo {
2811                 char literal[7];
2812                 unsigned char code;
2813                 int flag;
2814         };
2815         /* Mostly a list of accepted follow-up reserved words.
2816          * FLAG_END means we are done with the sequence, and are ready
2817          * to turn the compound list into a command.
2818          * FLAG_START means the word must start a new compound list.
2819          */
2820         static const struct reserved_combo reserved_list[] = {
2821                 { "if",    RES_IF,    FLAG_THEN | FLAG_START },
2822                 { "then",  RES_THEN,  FLAG_ELIF | FLAG_ELSE | FLAG_FI },
2823                 { "elif",  RES_ELIF,  FLAG_THEN },
2824                 { "else",  RES_ELSE,  FLAG_FI   },
2825                 { "fi",    RES_FI,    FLAG_END  },
2826                 { "for",   RES_FOR,   FLAG_IN   | FLAG_START },
2827                 { "while", RES_WHILE, FLAG_DO   | FLAG_START },
2828                 { "until", RES_UNTIL, FLAG_DO   | FLAG_START },
2829                 { "in",    RES_IN,    FLAG_DO   },
2830                 { "do",    RES_DO,    FLAG_DONE },
2831                 { "done",  RES_DONE,  FLAG_END  }
2832         };
2833         enum { NRES = sizeof(reserved_list)/sizeof(reserved_list[0]) };
2834         const struct reserved_combo *r;
2835
2836         for (r = reserved_list; r < reserved_list + NRES; r++) {
2837                 if (strcmp(dest->data, r->literal) == 0) {
2838                         debug_printf("found reserved word %s, code %d\n", r->literal, r->code);
2839                         if (r->flag & FLAG_START) {
2840                                 struct p_context *new = xmalloc(sizeof(struct p_context));
2841                                 debug_printf("push stack\n");
2842                                 if (ctx->res_w == RES_IN || ctx->res_w == RES_FOR) {
2843                                         syntax();
2844                                         free(new);
2845                                         ctx->res_w = RES_SNTX;
2846                                         b_reset(dest);
2847                                         return 1;
2848                                 }
2849                                 *new = *ctx;   /* physical copy */
2850                                 initialize_context(ctx);
2851                                 ctx->stack = new;
2852                         } else if (ctx->res_w == RES_NONE || !(ctx->old_flag & (1 << r->code))) {
2853                                 syntax();
2854                                 ctx->res_w = RES_SNTX;
2855                                 b_reset(dest);
2856                                 return 1;
2857                         }
2858                         ctx->res_w = r->code;
2859                         ctx->old_flag = r->flag;
2860                         if (ctx->old_flag & FLAG_END) {
2861                                 struct p_context *old;
2862                                 debug_printf("pop stack\n");
2863                                 done_pipe(ctx, PIPE_SEQ);
2864                                 old = ctx->stack;
2865                                 old->child->group = ctx->list_head;
2866                                 old->child->subshell = 0;
2867                                 *ctx = *old;   /* physical copy */
2868                                 free(old);
2869                         }
2870                         b_reset(dest);
2871                         return 1;
2872                 }
2873         }
2874         return 0;
2875 }
2876
2877 /* Normal return is 0.
2878  * Syntax or xglob errors return 1. */
2879 static int done_word(o_string *dest, struct p_context *ctx)
2880 {
2881         struct child_prog *child = ctx->child;
2882         glob_t *glob_target;
2883         int gr, flags = 0;
2884
2885         debug_printf_parse("done_word entered: '%s' %p\n", dest->data, child);
2886         if (dest->length == 0 && !dest->nonnull) {
2887                 debug_printf_parse("done_word return 0: true null, ignored\n");
2888                 return 0;
2889         }
2890         if (ctx->pending_redirect) {
2891                 glob_target = &ctx->pending_redirect->word;
2892         } else {
2893                 if (child->group) {
2894                         syntax();
2895                         debug_printf_parse("done_word return 1: syntax error, groups and arglists don't mix\n");
2896                         return 1;
2897                 }
2898                 if (!child->argv && (ctx->parse_type & FLAG_PARSE_SEMICOLON)) {
2899                         debug_printf_parse(": checking '%s' for reserved-ness\n", dest->data);
2900                         if (reserved_word(dest, ctx)) {
2901                                 debug_printf_parse("done_word return %d\n", (ctx->res_w == RES_SNTX));
2902                                 return (ctx->res_w == RES_SNTX);
2903                         }
2904                 }
2905                 glob_target = &child->glob_result;
2906                 if (child->argv)
2907                         flags |= GLOB_APPEND;
2908         }
2909         gr = xglob(dest, flags, glob_target);
2910         if (gr != 0) {
2911                 debug_printf_parse("done_word return 1: xglob returned %d\n", gr);
2912                 return 1;
2913         }
2914
2915         b_reset(dest);
2916         if (ctx->pending_redirect) {
2917                 ctx->pending_redirect = NULL;
2918                 if (glob_target->gl_pathc != 1) {
2919                         bb_error_msg("ambiguous redirect");
2920                         debug_printf_parse("done_word return 1: ambiguous redirect\n");
2921                         return 1;
2922                 }
2923         } else {
2924                 child->argv = glob_target->gl_pathv;
2925         }
2926         if (ctx->res_w == RES_FOR) {
2927                 done_word(dest, ctx);
2928                 done_pipe(ctx, PIPE_SEQ);
2929         }
2930         debug_printf_parse("done_word return 0\n");
2931         return 0;
2932 }
2933
2934 /* The only possible error here is out of memory, in which case
2935  * xmalloc exits. */
2936 static int done_command(struct p_context *ctx)
2937 {
2938         /* The child is really already in the pipe structure, so
2939          * advance the pipe counter and make a new, null child. */
2940         struct pipe *pi = ctx->pipe;
2941         struct child_prog *child = ctx->child;
2942
2943         if (child) {
2944                 if (child->group == NULL
2945                  && child->argv == NULL
2946                  && child->redirects == NULL
2947                 ) {
2948                         debug_printf_parse("done_command: skipping null cmd, num_progs=%d\n", pi->num_progs);
2949                         return pi->num_progs;
2950                 }
2951                 pi->num_progs++;
2952                 debug_printf_parse("done_command: ++num_progs=%d\n", pi->num_progs);
2953         } else {
2954                 debug_printf_parse("done_command: initializing, num_progs=%d\n", pi->num_progs);
2955         }
2956
2957         /* Only real trickiness here is that the uncommitted
2958          * child structure is not counted in pi->num_progs. */
2959         pi->progs = xrealloc(pi->progs, sizeof(*pi->progs) * (pi->num_progs+1));
2960         child = &pi->progs[pi->num_progs];
2961
2962         memset(child, 0, sizeof(*child));
2963         /*child->redirects = NULL;*/
2964         /*child->argv = NULL;*/
2965         /*child->is_stopped = 0;*/
2966         /*child->group = NULL;*/
2967         /*child->glob_result.gl_pathv = NULL;*/
2968         child->family = pi;
2969         //sp: /*child->sp = 0;*/
2970         child->type = ctx->parse_type;
2971
2972         ctx->child = child;
2973         /* but ctx->pipe and ctx->list_head remain unchanged */
2974
2975         return pi->num_progs; /* used only for 0/nonzero check */
2976 }
2977
2978 static int done_pipe(struct p_context *ctx, pipe_style type)
2979 {
2980         struct pipe *new_p;
2981         int not_null;
2982
2983         debug_printf_parse("done_pipe entered, followup %d\n", type);
2984         not_null = done_command(ctx);  /* implicit closure of previous command */
2985         ctx->pipe->followup = type;
2986         ctx->pipe->r_mode = ctx->res_w;
2987         /* Without this check, even just <enter> on command line generates
2988          * tree of three NOPs (!). Which is harmless but annoying.
2989          * IOW: it is safe to do it unconditionally. */
2990         if (not_null) {
2991                 new_p = new_pipe();
2992                 ctx->pipe->next = new_p;
2993                 ctx->pipe = new_p;
2994                 ctx->child = NULL;
2995                 done_command(ctx);  /* set up new pipe to accept commands */
2996         }
2997         debug_printf_parse("done_pipe return 0\n");
2998         return 0;
2999 }
3000
3001 /* peek ahead in the in_str to find out if we have a "&n" construct,
3002  * as in "2>&1", that represents duplicating a file descriptor.
3003  * returns either -2 (syntax error), -1 (no &), or the number found.
3004  */
3005 static int redirect_dup_num(struct in_str *input)
3006 {
3007         int ch, d = 0, ok = 0;
3008         ch = b_peek(input);
3009         if (ch != '&') return -1;
3010
3011         b_getch(input);  /* get the & */
3012         ch = b_peek(input);
3013         if (ch == '-') {
3014                 b_getch(input);
3015                 return -3;  /* "-" represents "close me" */
3016         }
3017         while (isdigit(ch)) {
3018                 d = d*10 + (ch-'0');
3019                 ok = 1;
3020                 b_getch(input);
3021                 ch = b_peek(input);
3022         }
3023         if (ok) return d;
3024
3025         bb_error_msg("ambiguous redirect");
3026         return -2;
3027 }
3028
3029 /* If a redirect is immediately preceded by a number, that number is
3030  * supposed to tell which file descriptor to redirect.  This routine
3031  * looks for such preceding numbers.  In an ideal world this routine
3032  * needs to handle all the following classes of redirects...
3033  *     echo 2>foo     # redirects fd  2 to file "foo", nothing passed to echo
3034  *     echo 49>foo    # redirects fd 49 to file "foo", nothing passed to echo
3035  *     echo -2>foo    # redirects fd  1 to file "foo",    "-2" passed to echo
3036  *     echo 49x>foo   # redirects fd  1 to file "foo",   "49x" passed to echo
3037  * A -1 output from this program means no valid number was found, so the
3038  * caller should use the appropriate default for this redirection.
3039  */
3040 static int redirect_opt_num(o_string *o)
3041 {
3042         int num;
3043
3044         if (o->length == 0)
3045                 return -1;
3046         for (num = 0; num < o->length; num++) {
3047                 if (!isdigit(*(o->data + num))) {
3048                         return -1;
3049                 }
3050         }
3051         /* reuse num (and save an int) */
3052         num = atoi(o->data);
3053         b_reset(o);
3054         return num;
3055 }
3056
3057 static FILE *generate_stream_from_list(struct pipe *head)
3058 {
3059         FILE *pf;
3060         int pid, channel[2];
3061         if (pipe(channel) < 0) bb_perror_msg_and_die("pipe");
3062 #if BB_MMU
3063         pid = fork();
3064 #else
3065         pid = vfork();
3066 #endif
3067         if (pid < 0) {
3068                 bb_perror_msg_and_die("fork");
3069         } else if (pid == 0) {
3070                 close(channel[0]);
3071                 if (channel[1] != 1) {
3072                         dup2(channel[1], 1);
3073                         close(channel[1]);
3074                 }
3075                 _exit(run_list_real(head));   /* leaks memory */
3076         }
3077         debug_printf("forked child %d\n", pid);
3078         close(channel[1]);
3079         pf = fdopen(channel[0], "r");
3080         debug_printf("pipe on FILE *%p\n", pf);
3081         return pf;
3082 }
3083
3084 /* Return code is exit status of the process that is run. */
3085 static int process_command_subs(o_string *dest, struct p_context *ctx,
3086         struct in_str *input, const char *subst_end)
3087 {
3088         int retcode, ch, eol_cnt;
3089         o_string result = NULL_O_STRING;
3090         struct p_context inner;
3091         FILE *p;
3092         struct in_str pipe_str;
3093
3094         initialize_context(&inner);
3095
3096         /* recursion to generate command */
3097         retcode = parse_stream(&result, &inner, input, subst_end);
3098         if (retcode != 0) return retcode;  /* syntax error or EOF */
3099         done_word(&result, &inner);
3100         done_pipe(&inner, PIPE_SEQ);
3101         b_free(&result);
3102
3103         p = generate_stream_from_list(inner.list_head);
3104         if (p == NULL) return 1;
3105         mark_open(fileno(p));
3106         setup_file_in_str(&pipe_str, p);
3107
3108         /* now send results of command back into original context */
3109         eol_cnt = 0;
3110         while ((ch = b_getch(&pipe_str)) != EOF) {
3111                 if (ch == '\n') {
3112                         eol_cnt++;
3113                         continue;
3114                 }
3115                 while (eol_cnt) {
3116                         b_addqchr(dest, '\n', dest->quote);
3117                         eol_cnt--;
3118                 }
3119                 b_addqchr(dest, ch, dest->quote);
3120         }
3121
3122         debug_printf("done reading from pipe, pclose()ing\n");
3123         /* This is the step that wait()s for the child.  Should be pretty
3124          * safe, since we just read an EOF from its stdout.  We could try
3125          * to better, by using wait(), and keeping track of background jobs
3126          * at the same time.  That would be a lot of work, and contrary
3127          * to the KISS philosophy of this program. */
3128         mark_closed(fileno(p));
3129         retcode = pclose(p);
3130         free_pipe_list(inner.list_head, 0);
3131         debug_printf("pclosed, retcode=%d\n", retcode);
3132         return retcode;
3133 }
3134
3135 static int parse_group(o_string *dest, struct p_context *ctx,
3136         struct in_str *input, int ch)
3137 {
3138         int rcode;
3139         const char *endch = NULL;
3140         struct p_context sub;
3141         struct child_prog *child = ctx->child;
3142
3143         debug_printf_parse("parse_group entered\n");
3144         if (child->argv) {
3145                 syntax();
3146                 debug_printf_parse("parse_group return 1: syntax error, groups and arglists don't mix\n");
3147                 return 1;
3148         }
3149         initialize_context(&sub);
3150         switch (ch) {
3151         case '(':
3152                 endch = ")";
3153                 child->subshell = 1;
3154                 break;
3155         case '{':
3156                 endch = "}";
3157                 break;
3158         default:
3159                 syntax();   /* really logic error */
3160         }
3161         rcode = parse_stream(dest, &sub, input, endch);
3162         done_word(dest, &sub); /* finish off the final word in the subcontext */
3163         done_pipe(&sub, PIPE_SEQ);  /* and the final command there, too */
3164         child->group = sub.list_head;
3165
3166         debug_printf_parse("parse_group return %d\n", rcode);
3167         return rcode;
3168         /* child remains "open", available for possible redirects */
3169 }
3170
3171 /* Basically useful version until someone wants to get fancier,
3172  * see the bash man page under "Parameter Expansion" */
3173 static const char *lookup_param(const char *src)
3174 {
3175         const char *p = NULL;
3176         if (src) {
3177                 p = getenv(src);
3178                 if (!p)
3179                         p = get_local_var(src);
3180         }
3181         return p;
3182 }
3183
3184 /* Make new string for parser */
3185 static char* make_string(char **inp)
3186 {
3187         char *p;
3188         char *str = NULL;
3189         int n;
3190         int val_len;
3191         int len = 0;
3192
3193         for (n = 0; inp[n]; n++) {
3194                 p = expand_variables_to_string(inp[n]);
3195                 val_len = strlen(p);
3196                 str = xrealloc(str, len + val_len + 3); /* +3: space, '\n', <nul>*/
3197                 str[len++] = ' ';
3198                 strcpy(str + len, p);
3199                 len += val_len;
3200                 if (p != inp[n]) free(p);
3201         }
3202         /* We do not check for case where loop had no iterations at all
3203          * - cannot happen? */
3204         str[len] = '\n';
3205         str[len+1] = '\0';
3206         return str;
3207 }
3208
3209 /* return code: 0 for OK, 1 for syntax error */
3210 static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input)
3211 {
3212         int ch = b_peek(input);  /* first character after the $ */
3213         unsigned char quote_mask = dest->quote ? 0x80 : 0;
3214
3215         debug_printf_parse("handle_dollar entered: ch='%c'\n", ch);
3216         if (isalpha(ch)) {
3217                 b_addchr(dest, SPECIAL_VAR_SYMBOL);
3218                 //sp: ctx->child->sp++;
3219                 while (1) {
3220                         debug_printf_parse(": '%c'\n", ch);
3221                         b_getch(input);
3222                         b_addchr(dest, ch | quote_mask);
3223                         quote_mask = 0;
3224                         ch = b_peek(input);
3225                         if (!isalnum(ch) && ch != '_')
3226                                 break;
3227                 }
3228                 b_addchr(dest, SPECIAL_VAR_SYMBOL);
3229         } else if (isdigit(ch)) {
3230  make_one_char_var:
3231                 b_addchr(dest, SPECIAL_VAR_SYMBOL);
3232                 //sp: ctx->child->sp++;
3233                 debug_printf_parse(": '%c'\n", ch);
3234                 b_getch(input);
3235                 b_addchr(dest, ch | quote_mask);
3236                 b_addchr(dest, SPECIAL_VAR_SYMBOL);
3237         } else switch (ch) {
3238                 case '$': /* pid */
3239                 case '!': /* last bg pid */
3240                 case '?': /* last exit code */
3241                 case '#': /* number of args */
3242                 case '*': /* args */
3243                 case '@': /* args */
3244                         goto make_one_char_var;
3245                 case '{':
3246                         b_addchr(dest, SPECIAL_VAR_SYMBOL);
3247                         //sp: ctx->child->sp++;
3248                         b_getch(input);
3249                         /* XXX maybe someone will try to escape the '}' */
3250                         while (1) {
3251                                 ch = b_getch(input);
3252                                 if (ch == EOF) {
3253                                         syntax();
3254                                         debug_printf_parse("handle_dollar return 1: unterminated ${name}\n");
3255                                         return 1;
3256                                 }
3257                                 if (ch == '}')
3258                                         break;
3259                                 debug_printf_parse(": '%c'\n", ch);
3260                                 b_addchr(dest, ch | quote_mask);
3261                                 quote_mask = 0;
3262                         }
3263                         b_addchr(dest, SPECIAL_VAR_SYMBOL);
3264                         break;
3265                 case '(':
3266                         b_getch(input);
3267                         process_command_subs(dest, ctx, input, ")");
3268                         break;
3269                 case '-':
3270                 case '_':
3271                         /* still unhandled, but should be eventually */
3272                         bb_error_msg("unhandled syntax: $%c", ch);
3273                         return 1;
3274                         break;
3275                 default:
3276                         b_addqchr(dest, '$', dest->quote);
3277         }
3278         debug_printf_parse("handle_dollar return 0\n");
3279         return 0;
3280 }
3281
3282 /* return code is 0 for normal exit, 1 for syntax error */
3283 static int parse_stream(o_string *dest, struct p_context *ctx,
3284         struct in_str *input, const char *end_trigger)
3285 {
3286         int ch, m;
3287         int redir_fd;
3288         redir_type redir_style;
3289         int next;
3290
3291         /* Only double-quote state is handled in the state variable dest->quote.
3292          * A single-quote triggers a bypass of the main loop until its mate is
3293          * found.  When recursing, quote state is passed in via dest->quote. */
3294
3295         debug_printf_parse("parse_stream entered, end_trigger='%s'\n", end_trigger);
3296
3297         while ((ch = b_getch(input)) != EOF) {
3298                 m = charmap[ch];
3299                 next = (ch == '\n') ? '\0' : b_peek(input);
3300                 debug_printf_parse(": ch=%c (%d) m=%d quote=%d\n",
3301                                                 ch, ch, m, dest->quote);
3302                 if (m == CHAR_ORDINARY
3303                  || (m != CHAR_SPECIAL && dest->quote)
3304                 ) {
3305                         b_addqchr(dest, ch, dest->quote);
3306                         continue;
3307                 }
3308                 if (m == CHAR_IFS) {
3309                         if (done_word(dest, ctx)) {
3310                                 debug_printf_parse("parse_stream return 1: done_word!=0\n");
3311                                 return 1;
3312                         }
3313                         /* If we aren't performing a substitution, treat
3314                          * a newline as a command separator.
3315                          * [why we don't handle it exactly like ';'? --vda] */
3316                         if (end_trigger && ch == '\n') {
3317                                 done_pipe(ctx, PIPE_SEQ);
3318                         }
3319                 }
3320                 if ((end_trigger && strchr(end_trigger, ch))
3321                  && !dest->quote && ctx->res_w == RES_NONE
3322                 ) {
3323                         debug_printf_parse("parse_stream return 0: end_trigger char found\n");
3324                         return 0;
3325                 }
3326                 if (m == CHAR_IFS)
3327                         continue;
3328                 switch (ch) {
3329                 case '#':
3330                         if (dest->length == 0 && !dest->quote) {
3331                                 while (1) {
3332                                         ch = b_peek(input);
3333                                         if (ch == EOF || ch == '\n')
3334                                                 break;
3335                                         b_getch(input);
3336                                 }
3337                         } else {
3338                                 b_addqchr(dest, ch, dest->quote);
3339                         }
3340                         break;
3341                 case '\\':
3342                         if (next == EOF) {
3343                                 syntax();
3344                                 debug_printf_parse("parse_stream return 1: \\<eof>\n");
3345                                 return 1;
3346                         }
3347                         b_addqchr(dest, '\\', dest->quote);
3348                         b_addqchr(dest, b_getch(input), dest->quote);
3349                         break;
3350                 case '$':
3351                         if (handle_dollar(dest, ctx, input) != 0) {
3352                                 debug_printf_parse("parse_stream return 1: handle_dollar returned non-0\n");
3353                                 return 1;
3354                         }
3355                         break;
3356                 case '\'':
3357                         dest->nonnull = 1;
3358                         while (1) {
3359                                 ch = b_getch(input);
3360                                 if (ch == EOF || ch == '\'')
3361                                         break;
3362                                 b_addchr(dest, ch);
3363                         }
3364                         if (ch == EOF) {
3365                                 syntax();
3366                                 debug_printf_parse("parse_stream return 1: unterminated '\n");
3367                                 return 1;
3368                         }
3369                         break;
3370                 case '"':
3371                         dest->nonnull = 1;
3372                         dest->quote = !dest->quote;
3373                         break;
3374                 case '`':
3375                         process_command_subs(dest, ctx, input, "`");
3376                         break;
3377                 case '>':
3378                         redir_fd = redirect_opt_num(dest);
3379                         done_word(dest, ctx);
3380                         redir_style = REDIRECT_OVERWRITE;
3381                         if (next == '>') {
3382                                 redir_style = REDIRECT_APPEND;
3383                                 b_getch(input);
3384                         } else if (next == '(') {
3385                                 syntax();   /* until we support >(list) Process Substitution */
3386                                 debug_printf_parse("parse_stream return 1: >(process) not supported\n");
3387                                 return 1;
3388                         }
3389                         setup_redirect(ctx, redir_fd, redir_style, input);
3390                         break;
3391                 case '<':
3392                         redir_fd = redirect_opt_num(dest);
3393                         done_word(dest, ctx);
3394                         redir_style = REDIRECT_INPUT;
3395                         if (next == '<') {
3396                                 redir_style = REDIRECT_HEREIS;
3397                                 b_getch(input);
3398                         } else if (next == '>') {
3399                                 redir_style = REDIRECT_IO;
3400                                 b_getch(input);
3401                         } else if (next == '(') {
3402                                 syntax();   /* until we support <(list) Process Substitution */
3403                                 debug_printf_parse("parse_stream return 1: <(process) not supported\n");
3404                                 return 1;
3405                         }
3406                         setup_redirect(ctx, redir_fd, redir_style, input);
3407                         break;
3408                 case ';':
3409                         done_word(dest, ctx);
3410                         done_pipe(ctx, PIPE_SEQ);
3411                         break;
3412                 case '&':
3413                         done_word(dest, ctx);
3414                         if (next == '&') {
3415                                 b_getch(input);
3416                                 done_pipe(ctx, PIPE_AND);
3417                         } else {
3418                                 done_pipe(ctx, PIPE_BG);
3419                         }
3420                         break;
3421                 case '|':
3422                         done_word(dest, ctx);
3423                         if (next == '|') {
3424                                 b_getch(input);
3425                                 done_pipe(ctx, PIPE_OR);
3426                         } else {
3427                                 /* we could pick up a file descriptor choice here
3428                                  * with redirect_opt_num(), but bash doesn't do it.
3429                                  * "echo foo 2| cat" yields "foo 2". */
3430                                 done_command(ctx);
3431                         }
3432                         break;
3433                 case '(':
3434                 case '{':
3435                         if (parse_group(dest, ctx, input, ch) != 0) {
3436                                 debug_printf_parse("parse_stream return 1: parse_group returned non-0\n");
3437                                 return 1;
3438                         }
3439                         break;
3440                 case ')':
3441                 case '}':
3442                         syntax();   /* Proper use of this character is caught by end_trigger */
3443                         debug_printf_parse("parse_stream return 1: unexpected '}'\n");
3444                         return 1;
3445                 default:
3446                         syntax();   /* this is really an internal logic error */
3447                         debug_printf_parse("parse_stream return 1: internal logic error\n");
3448                         return 1;
3449                 }
3450         }
3451         /* Complain if quote?  No, maybe we just finished a command substitution
3452          * that was quoted.  Example:
3453          * $ echo "`cat foo` plus more"
3454          * and we just got the EOF generated by the subshell that ran "cat foo"
3455          * The only real complaint is if we got an EOF when end_trigger != NULL,
3456          * that is, we were really supposed to get end_trigger, and never got
3457          * one before the EOF.  Can't use the standard "syntax error" return code,
3458          * so that parse_stream_outer can distinguish the EOF and exit smoothly. */
3459         debug_printf_parse("parse_stream return %d\n", -(end_trigger != NULL));
3460         if (end_trigger)
3461                 return -1;
3462         return 0;
3463 }
3464
3465 static void set_in_charmap(const char *set, int code)
3466 {
3467         while (*set)
3468                 charmap[(unsigned char)*set++] = code;
3469 }
3470
3471 static void update_charmap(void)
3472 {
3473         /* char *ifs and char charmap[256] are both globals. */
3474         ifs = getenv("IFS");
3475         if (ifs == NULL)
3476                 ifs = " \t\n";
3477         /* Precompute a list of 'flow through' behavior so it can be treated
3478          * quickly up front.  Computation is necessary because of IFS.
3479          * Special case handling of IFS == " \t\n" is not implemented.
3480          * The charmap[] array only really needs two bits each,
3481          * and on most machines that would be faster (reduced L1 cache use).
3482          */
3483         memset(charmap, CHAR_ORDINARY, sizeof(charmap));
3484         set_in_charmap("\\$\"`", CHAR_SPECIAL);
3485         set_in_charmap("<>;&|(){}#'", CHAR_ORDINARY_IF_QUOTED);
3486         set_in_charmap(ifs, CHAR_IFS);  /* also flow through if quoted */
3487 }
3488
3489 /* most recursion does not come through here, the exception is
3490  * from builtin_source() */
3491 static int parse_stream_outer(struct in_str *inp, int parse_flag)
3492 {
3493         struct p_context ctx;
3494         o_string temp = NULL_O_STRING;
3495         int rcode;
3496         do {
3497                 ctx.parse_type = parse_flag;
3498                 initialize_context(&ctx);
3499                 update_charmap();
3500                 if (!(parse_flag & FLAG_PARSE_SEMICOLON) || (parse_flag & FLAG_REPARSING))
3501                         set_in_charmap(";$&|", CHAR_ORDINARY);
3502 #if ENABLE_HUSH_INTERACTIVE
3503                 inp->promptmode = 1;
3504 #endif
3505                 /* We will stop & execute after each ';' or '\n'.
3506                  * Example: "sleep 9999; echo TEST" + ctrl-C:
3507                  * TEST should be printed */
3508                 rcode = parse_stream(&temp, &ctx, inp, ";\n");
3509                 if (rcode != 1 && ctx.old_flag != 0) {
3510                         syntax();
3511                 }
3512                 if (rcode != 1 && ctx.old_flag == 0) {
3513                         done_word(&temp, &ctx);
3514                         done_pipe(&ctx, PIPE_SEQ);
3515                         debug_print_tree(ctx.list_head, 0);
3516                         debug_printf_exec("parse_stream_outer: run_list\n");
3517                         run_list(ctx.list_head);
3518                 } else {
3519                         if (ctx.old_flag != 0) {
3520                                 free(ctx.stack);
3521                                 b_reset(&temp);
3522                         }
3523                         temp.nonnull = 0;
3524                         temp.quote = 0;
3525                         inp->p = NULL;
3526                         free_pipe_list(ctx.list_head, 0);
3527                 }
3528                 b_free(&temp);
3529         } while (rcode != -1 && !(parse_flag & FLAG_EXIT_FROM_LOOP));   /* loop on syntax errors, return on EOF */
3530         return 0;
3531 }
3532
3533 static int parse_string_outer(const char *s, int parse_flag)
3534 {
3535         struct in_str input;
3536         setup_string_in_str(&input, s);
3537         return parse_stream_outer(&input, parse_flag);
3538 }
3539
3540 static int parse_file_outer(FILE *f)
3541 {
3542         int rcode;
3543         struct in_str input;
3544         setup_file_in_str(&input, f);
3545         rcode = parse_stream_outer(&input, FLAG_PARSE_SEMICOLON);
3546         return rcode;
3547 }
3548
3549 #if ENABLE_HUSH_JOB
3550 /* Make sure we have a controlling tty.  If we get started under a job
3551  * aware app (like bash for example), make sure we are now in charge so
3552  * we don't fight over who gets the foreground */
3553 static void setup_job_control(void)
3554 {
3555         pid_t shell_pgrp;
3556
3557         saved_task_pgrp = shell_pgrp = getpgrp();
3558         debug_printf_jobs("saved_task_pgrp=%d\n", saved_task_pgrp);
3559         fcntl(interactive_fd, F_SETFD, FD_CLOEXEC);
3560
3561         /* If we were ran as 'hush &',
3562          * sleep until we are in the foreground.  */
3563         while (tcgetpgrp(interactive_fd) != shell_pgrp) {
3564                 /* Send TTIN to ourself (should stop us) */
3565                 kill(- shell_pgrp, SIGTTIN);
3566                 shell_pgrp = getpgrp();
3567         }
3568
3569         /* Ignore job-control and misc signals.  */
3570         set_jobctrl_sighandler(SIG_IGN);
3571         set_misc_sighandler(SIG_IGN);
3572 //huh?  signal(SIGCHLD, SIG_IGN);
3573
3574         /* We _must_ restore tty pgrp on fatal signals */
3575         set_fatal_sighandler(sigexit);
3576
3577         /* Put ourselves in our own process group.  */
3578         setpgrp(); /* is the same as setpgid(our_pid, our_pid); */
3579         /* Grab control of the terminal.  */
3580         tcsetpgrp(interactive_fd, getpid());
3581 }
3582 #endif
3583
3584 int hush_main(int argc, char **argv);
3585 int hush_main(int argc, char **argv)
3586 {
3587         int opt;
3588         FILE *input;
3589         char **e;
3590
3591         PTR_TO_GLOBALS = xzalloc(sizeof(G));
3592         top_vars = &shell_ver;
3593         shell_ver = const_shell_ver; /* copying struct here */
3594
3595 #if ENABLE_FEATURE_EDITING
3596         line_input_state = new_line_input_t(FOR_SHELL);
3597 #endif
3598         /* XXX what should these be while sourcing /etc/profile? */
3599         global_argc = argc;
3600         global_argv = argv;
3601         /* Initialize some more globals to non-zero values */
3602         set_cwd();
3603 #if ENABLE_HUSH_INTERACTIVE
3604 #if ENABLE_FEATURE_EDITING
3605         cmdedit_set_initial_prompt();
3606 #endif
3607         PS2 = "> ";
3608 #endif
3609
3610         /* initialize our shell local variables with the values
3611          * currently living in the environment */
3612         e = environ;
3613         if (e)
3614                 while (*e)
3615                         set_local_var(*e++, 2);   /* without call putenv() */
3616
3617         last_return_code = EXIT_SUCCESS;
3618
3619         if (argv[0] && argv[0][0] == '-') {
3620                 debug_printf("sourcing /etc/profile\n");
3621                 input = fopen("/etc/profile", "r");
3622                 if (input != NULL) {
3623                         mark_open(fileno(input));
3624                         parse_file_outer(input);
3625                         mark_closed(fileno(input));
3626                         fclose(input);
3627                 }
3628         }
3629         input = stdin;
3630
3631         while ((opt = getopt(argc, argv, "c:xif")) > 0) {
3632                 switch (opt) {
3633                 case 'c':
3634                         global_argv = argv + optind;
3635                         global_argc = argc - optind;
3636                         opt = parse_string_outer(optarg, FLAG_PARSE_SEMICOLON);
3637                         goto final_return;
3638                 case 'i':
3639                         /* Well, we cannot just declare interactiveness,
3640                          * we have to have some stuff (ctty, etc) */
3641                         /* interactive_fd++; */
3642                         break;
3643                 case 'f':
3644                         fake_mode = 1;
3645                         break;
3646                 default:
3647 #ifndef BB_VER
3648                         fprintf(stderr, "Usage: sh [FILE]...\n"
3649                                         "   or: sh -c command [args]...\n\n");
3650                         exit(EXIT_FAILURE);
3651 #else
3652                         bb_show_usage();
3653 #endif
3654                 }
3655         }
3656 #if ENABLE_HUSH_JOB
3657         /* A shell is interactive if the '-i' flag was given, or if all of
3658          * the following conditions are met:
3659          *    no -c command
3660          *    no arguments remaining or the -s flag given
3661          *    standard input is a terminal
3662          *    standard output is a terminal
3663          *    Refer to Posix.2, the description of the 'sh' utility. */
3664         if (argv[optind] == NULL && input == stdin
3665          && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)
3666         ) {
3667                 saved_tty_pgrp = tcgetpgrp(STDIN_FILENO);
3668                 debug_printf("saved_tty_pgrp=%d\n", saved_tty_pgrp);
3669                 if (saved_tty_pgrp >= 0) {
3670                         /* try to dup to high fd#, >= 255 */
3671                         interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
3672                         if (interactive_fd < 0) {
3673                                 /* try to dup to any fd */
3674                                 interactive_fd = dup(STDIN_FILENO);
3675                                 if (interactive_fd < 0)
3676                                         /* give up */
3677                                         interactive_fd = 0;
3678                         }
3679                         // TODO: track & disallow any attempts of user
3680                         // to (inadvertently) close/redirect it
3681                 }
3682         }
3683         debug_printf("interactive_fd=%d\n", interactive_fd);
3684         if (interactive_fd) {
3685                 /* Looks like they want an interactive shell */
3686                 setup_job_control();
3687                 /* Make xfuncs do cleanup on exit */
3688                 die_sleep = -1; /* flag */
3689 // FIXME: should we reset die_sleep = 0 whereever we fork?
3690                 if (setjmp(die_jmp)) {
3691                         /* xfunc has failed! die die die */
3692                         hush_exit(xfunc_error_retval);
3693                 }
3694 #if !ENABLE_FEATURE_SH_EXTRA_QUIET
3695                 printf("\n\n%s hush - the humble shell v"HUSH_VER_STR"\n", BB_BANNER);
3696                 printf("Enter 'help' for a list of built-in commands.\n\n");
3697 #endif
3698         }
3699 #elif ENABLE_HUSH_INTERACTIVE
3700 /* no job control compiled, only prompt/line editing */
3701         if (argv[optind] == NULL && input == stdin
3702          && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO)
3703         ) {
3704                 interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255);
3705                 if (interactive_fd < 0) {
3706                         /* try to dup to any fd */
3707                         interactive_fd = dup(STDIN_FILENO);
3708                         if (interactive_fd < 0)
3709                                 /* give up */
3710                                 interactive_fd = 0;
3711                 }
3712         }
3713
3714 #endif
3715
3716         if (argv[optind] == NULL) {
3717                 opt = parse_file_outer(stdin);
3718                 goto final_return;
3719         }
3720
3721         debug_printf("\nrunning script '%s'\n", argv[optind]);
3722         global_argv = argv + optind;
3723         global_argc = argc - optind;
3724         input = xfopen(argv[optind], "r");
3725         opt = parse_file_outer(input);
3726
3727 #if ENABLE_FEATURE_CLEAN_UP
3728         fclose(input);
3729         if (cwd != bb_msg_unknown)
3730                 free((char*)cwd);
3731         {
3732                 struct variables *cur, *tmp;
3733                 for (cur = top_vars; cur; cur = tmp) {
3734                         tmp = cur->next;
3735                         if (!cur->flg_read_only) {
3736                                 free((char*)cur->name);
3737                                 free((char*)cur->value);
3738                                 free(cur);
3739                         }
3740                 }
3741         }
3742 #endif
3743
3744  final_return:
3745         hush_exit(opt ? opt : last_return_code);
3746 }