It turns out job control in both hush and lash was broken by the
[oweals/busybox.git] / 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.
13  *      The execution engine, the builtins, and much of the underlying
14  *      support has been adapted from busybox-0.49pre's lash,
15  *      which is Copyright (C) 2000 by Lineo, Inc., and
16  *      written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>.
17  *      That, in turn, is based in part on ladsh.c, by Michael K. Johnson and
18  *      Erik W. Troan, which they placed in the public domain.  I don't know
19  *      how much of the Johnson/Troan code has survived the repeated rewrites.
20  * Other credits:
21  *      simple_itoa() was lifted from boa-0.93.15
22  *      b_addchr() derived from similar w_addchar function in glibc-2.2
23  *      setup_redirect(), redirect_opt_num(), and big chunks of main()
24  *        and many builtins derived from contributions by Erik Andersen
25  *      miscellaneous bugfixes from Matt Kraai
26  *
27  * There are two big (and related) architecture differences between
28  * this parser and the lash parser.  One is that this version is
29  * actually designed from the ground up to understand nearly all
30  * of the Bourne grammar.  The second, consequential change is that
31  * the parser and input reader have been turned inside out.  Now,
32  * the parser is in control, and asks for input as needed.  The old
33  * way had the input reader in control, and it asked for parsing to
34  * take place as needed.  The new way makes it much easier to properly
35  * handle the recursion implicit in the various substitutions, especially
36  * across continuation lines.
37  *
38  * Bash grammar not implemented: (how many of these were in original sh?)
39  *      $@ (those sure look like weird quoting rules)
40  *      $_
41  *      ! negation operator for pipes
42  *      &> and >& redirection of stdout+stderr
43  *      Brace Expansion
44  *      Tilde Expansion
45  *      fancy forms of Parameter Expansion
46  *      aliases
47  *      Arithmetic Expansion
48  *      <(list) and >(list) Process Substitution
49  *      reserved words: case, esac, select, function
50  *      Here Documents ( << word )
51  *      Functions
52  * Major bugs:
53  *      job handling woefully incomplete and buggy
54  *      reserved word execution woefully incomplete and buggy
55  * to-do:
56  *      port selected bugfixes from post-0.49 busybox lash - done?
57  *      finish implementing reserved words: for, while, until, do, done
58  *      change { and } from special chars to reserved words
59  *      builtins: break, continue, eval, return, set, trap, ulimit
60  *      test magic exec
61  *      handle children going into background
62  *      clean up recognition of null pipes
63  *      check setting of global_argc and global_argv
64  *      control-C handling, probably with longjmp
65  *      follow IFS rules more precisely, including update semantics
66  *      figure out what to do with backslash-newline
67  *      explain why we use signal instead of sigaction
68  *      propagate syntax errors, die on resource errors?
69  *      continuation lines, both explicit and implicit - done?
70  *      memory leak finding and plugging - done?
71  *      more testing, especially quoting rules and redirection
72  *      document how quoting rules not precisely followed for variable assignments
73  *      maybe change map[] to use 2-bit entries
74  *      (eventually) remove all the printf's
75  *
76  * This program is free software; you can redistribute it and/or modify
77  * it under the terms of the GNU General Public License as published by
78  * the Free Software Foundation; either version 2 of the License, or
79  * (at your option) any later version.
80  *
81  * This program is distributed in the hope that it will be useful,
82  * but WITHOUT ANY WARRANTY; without even the implied warranty of
83  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
84  * General Public License for more details.
85  *
86  * You should have received a copy of the GNU General Public License
87  * along with this program; if not, write to the Free Software
88  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
89  */
90 #include <ctype.h>     /* isalpha, isdigit */
91 #include <unistd.h>    /* getpid */
92 #include <stdlib.h>    /* getenv, atoi */
93 #include <string.h>    /* strchr */
94 #include <stdio.h>     /* popen etc. */
95 #include <glob.h>      /* glob, of course */
96 #include <stdarg.h>    /* va_list */
97 #include <errno.h>
98 #include <fcntl.h>
99 #include <getopt.h>    /* should be pretty obvious */
100
101 #include <sys/stat.h>  /* ulimit */
102 #include <sys/types.h>
103 #include <sys/wait.h>
104 #include <signal.h>
105
106 /* #include <dmalloc.h> */
107 /* #define DEBUG_SHELL */
108
109 #ifdef BB_VER
110 #include "busybox.h"
111 #include "cmdedit.h"
112 #else
113 #define applet_name "hush"
114 #include "standalone.h"
115 #define shell_main main
116 #define BB_FEATURE_SH_SIMPLE_PROMPT
117 #endif
118
119 typedef enum {
120         REDIRECT_INPUT     = 1,
121         REDIRECT_OVERWRITE = 2,
122         REDIRECT_APPEND    = 3,
123         REDIRECT_HEREIS    = 4,
124         REDIRECT_IO        = 5
125 } redir_type;
126
127 /* The descrip member of this structure is only used to make debugging
128  * output pretty */
129 struct {int mode; int default_fd; char *descrip;} redir_table[] = {
130         { 0,                         0, "()" },
131         { O_RDONLY,                  0, "<"  },
132         { O_CREAT|O_TRUNC|O_WRONLY,  1, ">"  },
133         { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
134         { O_RDONLY,                 -1, "<<" },
135         { O_RDWR,                    1, "<>" }
136 };
137
138 typedef enum {
139         PIPE_SEQ = 1,
140         PIPE_AND = 2,
141         PIPE_OR  = 3,
142         PIPE_BG  = 4,
143 } pipe_style;
144
145 /* might eventually control execution */
146 typedef enum {
147         RES_NONE  = 0,
148         RES_IF    = 1,
149         RES_THEN  = 2,
150         RES_ELIF  = 3,
151         RES_ELSE  = 4,
152         RES_FI    = 5,
153         RES_FOR   = 6,
154         RES_WHILE = 7,
155         RES_UNTIL = 8,
156         RES_DO    = 9,
157         RES_DONE  = 10,
158         RES_XXXX  = 11,
159         RES_SNTX  = 12
160 } reserved_style;
161 #define FLAG_END   (1<<RES_NONE)
162 #define FLAG_IF    (1<<RES_IF)
163 #define FLAG_THEN  (1<<RES_THEN)
164 #define FLAG_ELIF  (1<<RES_ELIF)
165 #define FLAG_ELSE  (1<<RES_ELSE)
166 #define FLAG_FI    (1<<RES_FI)
167 #define FLAG_FOR   (1<<RES_FOR)
168 #define FLAG_WHILE (1<<RES_WHILE)
169 #define FLAG_UNTIL (1<<RES_UNTIL)
170 #define FLAG_DO    (1<<RES_DO)
171 #define FLAG_DONE  (1<<RES_DONE)
172 #define FLAG_START (1<<RES_XXXX)
173
174 /* This holds pointers to the various results of parsing */
175 struct p_context {
176         struct child_prog *child;
177         struct pipe *list_head;
178         struct pipe *pipe;
179         struct redir_struct *pending_redirect;
180         reserved_style w;
181         int old_flag;                           /* for figuring out valid reserved words */
182         struct p_context *stack;
183         /* How about quoting status? */
184 };
185
186 struct redir_struct {
187         redir_type type;                        /* type of redirection */
188         int fd;                                         /* file descriptor being redirected */
189         int dup;                                        /* -1, or file descriptor being duplicated */
190         struct redir_struct *next;      /* pointer to the next redirect in the list */ 
191         glob_t word;                            /* *word.gl_pathv is the filename */
192 };
193
194 struct child_prog {
195         pid_t pid;                                      /* 0 if exited */
196         char **argv;                            /* program name and arguments */
197         struct pipe *group;                     /* if non-NULL, first in group or subshell */
198         int subshell;                           /* flag, non-zero if group must be forked */
199         struct redir_struct *redirects; /* I/O redirections */
200         glob_t glob_result;                     /* result of parameter globbing */
201         int is_stopped;                         /* is the program currently running? */
202         struct pipe *family;            /* pointer back to the child's parent pipe */
203 };
204
205 struct pipe {
206         int jobid;                                      /* job number */
207         int num_progs;                          /* total number of programs in job */
208         int running_progs;                      /* number of programs running */
209         char *text;                                     /* name of job */
210         char *cmdbuf;                           /* buffer various argv's point into */
211         pid_t pgrp;                                     /* process group ID for the job */
212         struct child_prog *progs;       /* array of commands in pipe */
213         struct pipe *next;                      /* to track background commands */
214         int stopped_progs;                      /* number of programs alive, but stopped */
215         int job_context;                        /* bitmask defining current context */
216         pipe_style followup;            /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
217         reserved_style r_mode;          /* supports if, for, while, until */
218 };
219
220 struct jobset {
221         struct pipe *head;                      /* head of list of running jobs */
222         struct pipe *fg;                        /* current foreground job */
223 };
224
225 struct close_me {
226         int fd;
227         struct close_me *next;
228 };
229
230 /* globals, connect us to the outside world
231  * the first three support $?, $#, and $1 */
232 char **global_argv;
233 unsigned int global_argc;
234 unsigned int last_return_code;
235 extern char **environ; /* This is in <unistd.h>, but protected with __USE_GNU */
236  
237 /* Variables we export */
238 unsigned int shell_context;  /* Used in cmdedit.c to reset the
239                               * context when someone hits ^C */
240
241 /* "globals" within this file */
242 static char *ifs;
243 static char map[256];
244 static int fake_mode;
245 static int interactive;
246 static struct close_me *close_me_head;
247 static const char *cwd;
248 static struct jobset *job_list;
249 static unsigned int last_bg_pid;
250 static char *PS1;
251 static char *PS2;
252 static char **__shell_local_env;
253
254 #define B_CHUNK (100)
255 #define B_NOSPAC 1
256 #define MAX_LINE 256       /* for cwd */
257 #define MAX_READ 256       /* for builtin_read */
258
259 typedef struct {
260         char *data;
261         int length;
262         int maxlen;
263         int quote;
264         int nonnull;
265 } o_string;
266 #define NULL_O_STRING {NULL,0,0,0,0}
267 /* used for initialization:
268         o_string foo = NULL_O_STRING; */
269
270 /* I can almost use ordinary FILE *.  Is open_memstream() universally
271  * available?  Where is it documented? */
272 struct in_str {
273         const char *p;
274         int __promptme;
275         int promptmode;
276         FILE *file;
277         int (*get) (struct in_str *);
278         int (*peek) (struct in_str *);
279 };
280 #define b_getch(input) ((input)->get(input))
281 #define b_peek(input) ((input)->peek(input))
282
283 #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
284
285 struct built_in_command {
286         char *cmd;                                      /* name */
287         char *descr;                            /* description */
288         int (*function) (struct child_prog *);  /* function ptr */
289 };
290
291 /* belongs in busybox.h */
292 static inline int max(int a, int b) {
293         return (a>b)?a:b;
294 }
295
296 /* This should be in utility.c */
297 #ifdef DEBUG_SHELL
298 static void debug_printf(const char *format, ...)
299 {
300         va_list args;
301         va_start(args, format);
302         vfprintf(stderr, format, args);
303         va_end(args);
304 }
305 #else
306 static void debug_printf(const char *format, ...) { }
307 #endif
308 #define final_printf debug_printf
309
310 void __syntax(char *file, int line) {
311         fprintf(stderr,"syntax error %s:%d\n",file,line);
312 }
313 #define syntax() __syntax(__FILE__, __LINE__)
314
315 /* Index of subroutines: */
316 /*   function prototypes for builtins */
317 static int builtin_cd(struct child_prog *child);
318 static int builtin_env(struct child_prog *child);
319 static int builtin_exec(struct child_prog *child);
320 static int builtin_exit(struct child_prog *child);
321 static int builtin_export(struct child_prog *child);
322 static int builtin_fg_bg(struct child_prog *child);
323 static int builtin_help(struct child_prog *child);
324 static int builtin_jobs(struct child_prog *child);
325 static int builtin_pwd(struct child_prog *child);
326 static int builtin_read(struct child_prog *child);
327 static int builtin_set(struct child_prog *child);
328 static int builtin_shift(struct child_prog *child);
329 static int builtin_source(struct child_prog *child);
330 static int builtin_umask(struct child_prog *child);
331 static int builtin_unset(struct child_prog *child);
332 static int builtin_not_written(struct child_prog *child);
333 /*   o_string manipulation: */
334 static int b_check_space(o_string *o, int len);
335 static int b_addchr(o_string *o, int ch);
336 static void b_reset(o_string *o);
337 static int b_addqchr(o_string *o, int ch, int quote);
338 static int b_adduint(o_string *o, unsigned int i);
339 /*  in_str manipulations: */
340 static int static_get(struct in_str *i);
341 static int static_peek(struct in_str *i);
342 static int file_get(struct in_str *i);
343 static int file_peek(struct in_str *i);
344 static void setup_file_in_str(struct in_str *i, FILE *f);
345 static void setup_string_in_str(struct in_str *i, const char *s);
346 /*  close_me manipulations: */
347 static void mark_open(int fd);
348 static void mark_closed(int fd);
349 static void close_all();
350 /*  "run" the final data structures: */
351 static char *indenter(int i);
352 static int run_list_test(struct pipe *head, int indent);
353 static int run_pipe_test(struct pipe *pi, int indent);
354 /*  really run the final data structures: */
355 static int setup_redirects(struct child_prog *prog, int squirrel[]);
356 static int pipe_wait(struct pipe *pi);
357 static int run_list_real(struct pipe *pi);
358 static void pseudo_exec(struct child_prog *child) __attribute__ ((noreturn));
359 static int run_pipe_real(struct pipe *pi);
360 /*   extended glob support: */
361 static int globhack(const char *src, int flags, glob_t *pglob);
362 static int glob_needed(const char *s);
363 static int xglob(o_string *dest, int flags, glob_t *pglob);
364 /*   variable assignment: */
365 static int set_local_var(const char *s);
366 static int is_assignment(const char *s);
367 /*   data structure manipulation: */
368 static int setup_redirect(struct p_context *ctx, int fd, redir_type style, struct in_str *input);
369 static void initialize_context(struct p_context *ctx);
370 static int done_word(o_string *dest, struct p_context *ctx);
371 static int done_command(struct p_context *ctx);
372 static int done_pipe(struct p_context *ctx, pipe_style type);
373 /*   primary string parsing: */
374 static int redirect_dup_num(struct in_str *input);
375 static int redirect_opt_num(o_string *o);
376 static int process_command_subs(o_string *dest, struct p_context *ctx, struct in_str *input, int subst_end);
377 static int parse_group(o_string *dest, struct p_context *ctx, struct in_str *input, int ch);
378 static void lookup_param(o_string *dest, struct p_context *ctx, o_string *src);
379 static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input);
380 static int parse_string(o_string *dest, struct p_context *ctx, const char *src);
381 static int parse_stream(o_string *dest, struct p_context *ctx, struct in_str *input0, int end_trigger);
382 /*   setup: */
383 static int parse_stream_outer(struct in_str *inp);
384 static int parse_string_outer(const char *s);
385 static int parse_file_outer(FILE *f);
386 /*   job management: */
387 static void checkjobs();
388 static void insert_bg_job(struct pipe *pi);
389 static void remove_bg_job(struct pipe *pi);
390 static void free_pipe(struct pipe *pi);
391 /*     local variable support */
392 static char *get_local_var(const char *var);
393 static int   set_local_var(const char *s);
394 static void  unset_local_var(const char *name);
395
396
397 /* Table of built-in functions.  They can be forked or not, depending on
398  * context: within pipes, they fork.  As simple commands, they do not.
399  * When used in non-forking context, they can change global variables
400  * in the parent shell process.  If forked, of course they can not.
401  * For example, 'unset foo | whatever' will parse and run, but foo will
402  * still be set at the end. */
403 static struct built_in_command bltins[] = {
404         {"bg", "Resume a job in the background", builtin_fg_bg},
405         {"break", "Exit for, while or until loop", builtin_not_written},
406         {"cd", "Change working directory", builtin_cd},
407         {"continue", "Continue for, while or until loop", builtin_not_written},
408         {"env", "Print all environment variables", builtin_env},
409         {"eval", "Construct and run shell command", builtin_not_written},
410         {"exec", "Exec command, replacing this shell with the exec'd process", 
411                 builtin_exec},
412         {"exit", "Exit from shell()", builtin_exit},
413         {"export", "Set environment variable", builtin_export},
414         {"fg", "Bring job into the foreground", builtin_fg_bg},
415         {"jobs", "Lists the active jobs", builtin_jobs},
416         {"pwd", "Print current directory", builtin_pwd},
417         {"read", "Input environment variable", builtin_read},
418         {"return", "Return from a function", builtin_not_written},
419         {"set", "Set/unset shell local variables", builtin_set},
420         {"shift", "Shift positional parameters", builtin_shift},
421         {"trap", "Trap signals", builtin_not_written},
422         {"ulimit","Controls resource limits", builtin_not_written},
423         {"umask","Sets file creation mask", builtin_umask},
424         {"unset", "Unset environment variable", builtin_unset},
425         {".", "Source-in and run commands in a file", builtin_source},
426         {"help", "List shell built-in commands", builtin_help},
427         {NULL, NULL, NULL}
428 };
429
430 /* built-in 'cd <path>' handler */
431 static int builtin_cd(struct child_prog *child)
432 {
433         char *newdir;
434         if (child->argv[1] == NULL)
435                 newdir = getenv("HOME");
436         else
437                 newdir = child->argv[1];
438         if (chdir(newdir)) {
439                 printf("cd: %s: %s\n", newdir, strerror(errno));
440                 return EXIT_FAILURE;
441         }
442         cwd = xgetcwd((char *)cwd);
443         if (!cwd)
444                 cwd = unknown;
445         return EXIT_SUCCESS;
446 }
447
448 /* built-in 'env' handler */
449 static int builtin_env(struct child_prog *dummy)
450 {
451         char **e = environ;
452         if (e == NULL) return EXIT_FAILURE;
453         for (; *e; e++) {
454                 puts(*e);
455         }
456         return EXIT_SUCCESS;
457 }
458
459 /* built-in 'exec' handler */
460 static int builtin_exec(struct child_prog *child)
461 {
462         if (child->argv[1] == NULL)
463                 return EXIT_SUCCESS;   /* Really? */
464         child->argv++;
465         pseudo_exec(child);
466         /* never returns */
467 }
468
469 /* built-in 'exit' handler */
470 static int builtin_exit(struct child_prog *child)
471 {
472         if (child->argv[1] == NULL)
473                 exit(last_return_code);
474         exit (atoi(child->argv[1]));
475 }
476
477 /* built-in 'export VAR=value' handler */
478 static int builtin_export(struct child_prog *child)
479 {
480         int res;
481         char *value, *name = child->argv[1];
482
483         if (name == NULL) {
484                 return (builtin_env(child));
485         }
486
487         value = strchr(name, '=');
488         if (!value) {
489                 /* They are exporting something without an =VALUE.
490                  * Assume this is a local shell variable they are exporting */
491                 name = get_local_var(name);
492                 if (! name ) { 
493                         error_msg("export failed");
494                         return (EXIT_FAILURE);
495                 }
496                 /* FIXME -- I leak memory!!!!! */
497                 value = malloc(strlen(child->argv[1]) + strlen(name) + 2);
498                 sprintf(value, "%s=%s", child->argv[1], name);
499         } else {
500                 /* Bourne shells always put exported variables into the 
501                  * local shell variable list.  Do that first... */
502                 set_local_var(name);
503                 /* FIXME -- I leak memory!!!!! */
504                 value = strdup(name);
505         }
506
507         /* FIXME -- I leak memory!!!!!
508          * It seems most putenv implementations place the very char* pointer
509          * we pass in directly into the environ array, so the memory holding
510          * this string has to be persistant.  We can't even use the memory for
511          * the local shell variable list, since where that memory is keeps 
512          * changing due to reallocs... */
513         res = putenv(value);
514         if (res)
515                 perror_msg("export");
516         return (res);
517 }
518
519 /* built-in 'fg' and 'bg' handler */
520 static int builtin_fg_bg(struct child_prog *child)
521 {
522         int i, jobnum;
523         struct pipe *pi=NULL;
524
525         /* If they gave us no args, assume they want the last backgrounded task */
526         if (!child->argv[1]) {
527                 for (pi = job_list->head; pi; pi = pi->next) {
528                         if (pi->progs && pi->progs->pid == last_bg_pid) {
529                                 break;
530                         }
531                 }
532                 if (!pi) {
533                         error_msg("%s: no current job", child->argv[0]);
534                         return EXIT_FAILURE;
535                 }
536         } else {
537                 if (sscanf(child->argv[1], "%%%d", &jobnum) != 1) {
538                         error_msg("%s: bad argument '%s'", child->argv[0], child->argv[1]);
539                         return EXIT_FAILURE;
540                 }
541
542                 for (pi = job_list->head; pi; pi = pi->next) {
543                         if (pi->jobid == jobnum) {
544                                 break;
545                         }
546                 }
547                 if (!pi) {
548                         error_msg("%s: %d: no such job", child->argv[0], jobnum);
549                         return EXIT_FAILURE;
550                 }
551         }
552         if (*child->argv[0] == 'f') {
553                 /* Make this job the foreground job */
554                 signal(SIGTTOU, SIG_IGN);
555                 /* suppress messages when run from /linuxrc mag@sysgo.de */
556                 if (tcsetpgrp(0, pi->pgrp) && errno != ENOTTY)
557                         perror_msg("tcsetpgrp"); 
558                 signal(SIGTTOU, SIG_DFL);
559                 job_list->fg = pi;
560         }
561
562         /* Restart the processes in the job */
563         for (i = 0; i < pi->num_progs; i++)
564                 pi->progs[i].is_stopped = 0;
565
566         kill(-pi->pgrp, SIGCONT);
567
568         pi->stopped_progs = 0;
569         return EXIT_SUCCESS;
570 }
571
572 /* built-in 'help' handler */
573 static int builtin_help(struct child_prog *dummy)
574 {
575         struct built_in_command *x;
576
577         printf("\nBuilt-in commands:\n");
578         printf("-------------------\n");
579         for (x = bltins; x->cmd; x++) {
580                 if (x->descr==NULL)
581                         continue;
582                 printf("%s\t%s\n", x->cmd, x->descr);
583         }
584         printf("\n\n");
585         return EXIT_SUCCESS;
586 }
587
588 /* built-in 'jobs' handler */
589 static int builtin_jobs(struct child_prog *child)
590 {
591         struct pipe *job;
592         char *status_string;
593
594         for (job = job_list->head; job; job = job->next) {
595                 if (job->running_progs == job->stopped_progs)
596                         status_string = "Stopped";
597                 else
598                         status_string = "Running";
599                 printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->text);
600         }
601         return EXIT_SUCCESS;
602 }
603
604
605 /* built-in 'pwd' handler */
606 static int builtin_pwd(struct child_prog *dummy)
607 {
608         cwd = xgetcwd((char *)cwd);
609         if (!cwd)
610                 cwd = unknown;
611         puts(cwd);
612         return EXIT_SUCCESS;
613 }
614
615 /* built-in 'read VAR' handler */
616 static int builtin_read(struct child_prog *child)
617 {
618         int res = 0, len, newlen;
619         char *s;
620         char string[MAX_READ];
621
622         if (child->argv[1]) {
623                 /* argument (VAR) given: put "VAR=" into buffer */
624                 strcpy(string, child->argv[1]);
625                 len = strlen(string);
626                 string[len++] = '=';
627                 string[len]   = '\0';
628                 /* XXX would it be better to go through in_str? */
629                 fgets(&string[len], sizeof(string) - len, stdin);       /* read string */
630                 newlen = strlen(string);
631                 if(newlen > len)
632                         string[--newlen] = '\0';        /* chomp trailing newline */
633                 /*
634                 ** string should now contain "VAR=<value>"
635                 ** copy it (putenv() won't do that, so we must make sure
636                 ** the string resides in a static buffer!)
637                 */
638                 res = -1;
639                 if((s = strdup(string)))
640                         res = putenv(s);
641                 if (res)
642                         fprintf(stderr, "read: %s\n", strerror(errno));
643         }
644         else
645                 fgets(string, sizeof(string), stdin);
646
647         return (res);
648 }
649
650 /* built-in 'set VAR=value' handler */
651 static int builtin_set(struct child_prog *child)
652 {
653         int res;
654         char *temp = child->argv[1];
655
656         if (child->argv[1] == NULL) {
657                 char **e = __shell_local_env;
658                 if (e == NULL) return EXIT_FAILURE;
659                 for (; *e; e++) {
660                         puts(*e);
661                 }
662                 return EXIT_SUCCESS;
663         }
664         res = set_local_var(temp);
665         if (res)
666                 fprintf(stderr, "set: %s\n", strerror(errno));
667         return (res);
668 }
669
670
671 /* Built-in 'shift' handler */
672 static int builtin_shift(struct child_prog *child)
673 {
674         int n=1;
675         if (child->argv[1]) {
676                 n=atoi(child->argv[1]);
677         }
678         if (n>=0 && n<global_argc) {
679                 /* XXX This probably breaks $0 */
680                 global_argc -= n;
681                 global_argv += n;
682                 return EXIT_SUCCESS;
683         } else {
684                 return EXIT_FAILURE;
685         }
686 }
687
688 /* Built-in '.' handler (read-in and execute commands from file) */
689 static int builtin_source(struct child_prog *child)
690 {
691         FILE *input;
692         int status;
693
694         if (child->argv[1] == NULL)
695                 return EXIT_FAILURE;
696
697         /* XXX search through $PATH is missing */
698         input = fopen(child->argv[1], "r");
699         if (!input) {
700                 fprintf(stderr, "Couldn't open file '%s'\n", child->argv[1]);
701                 return EXIT_FAILURE;
702         }
703
704         /* Now run the file */
705         /* XXX argv and argc are broken; need to save old global_argv
706          * (pointer only is OK!) on this stack frame,
707          * set global_argv=child->argv+1, recurse, and restore. */
708         mark_open(fileno(input));
709         status = parse_file_outer(input);
710         mark_closed(fileno(input));
711         fclose(input);
712         return (status);
713 }
714
715 static int builtin_umask(struct child_prog *child)
716 {
717         mode_t new_umask;
718         const char *arg = child->argv[1];
719         char *end;
720         if (arg) {
721                 new_umask=strtoul(arg, &end, 8);
722                 if (*end!='\0' || end == arg) {
723                         return EXIT_FAILURE;
724                 }
725         } else {
726                 printf("%.3o\n", (unsigned int) (new_umask=umask(0)));
727         }
728         umask(new_umask);
729         return EXIT_SUCCESS;
730 }
731
732 /* built-in 'unset VAR' handler */
733 static int builtin_unset(struct child_prog *child)
734 {
735         if (child->argv[1] == NULL) {
736                 fprintf(stderr, "unset: parameter required.\n");
737                 return EXIT_FAILURE;
738         }
739         unsetenv(child->argv[1]);
740         unset_local_var(child->argv[1]);
741         return EXIT_SUCCESS;
742 }
743
744 static int builtin_not_written(struct child_prog *child)
745 {
746         printf("builtin_%s not written\n",child->argv[0]);
747         return EXIT_FAILURE;
748 }
749
750 static int b_check_space(o_string *o, int len)
751 {
752         /* It would be easy to drop a more restrictive policy
753          * in here, such as setting a maximum string length */
754         if (o->length + len > o->maxlen) {
755                 char *old_data = o->data;
756                 /* assert (data == NULL || o->maxlen != 0); */
757                 o->maxlen += max(2*len, B_CHUNK);
758                 o->data = realloc(o->data, 1 + o->maxlen);
759                 if (o->data == NULL) {
760                         free(old_data);
761                 }
762         }
763         return o->data == NULL;
764 }
765
766 static int b_addchr(o_string *o, int ch)
767 {
768         debug_printf("b_addchr: %c %d %p\n", ch, o->length, o);
769         if (b_check_space(o, 1)) return B_NOSPAC;
770         o->data[o->length] = ch;
771         o->length++;
772         o->data[o->length] = '\0';
773         return 0;
774 }
775
776 static void b_reset(o_string *o)
777 {
778         o->length = 0;
779         o->nonnull = 0;
780         if (o->data != NULL) *o->data = '\0';
781 }
782
783 static void b_free(o_string *o)
784 {
785         b_reset(o);
786         if (o->data != NULL) free(o->data);
787         o->data = NULL;
788         o->maxlen = 0;
789 }
790
791 /* My analysis of quoting semantics tells me that state information
792  * is associated with a destination, not a source.
793  */
794 static int b_addqchr(o_string *o, int ch, int quote)
795 {
796         if (quote && strchr("*?[\\",ch)) {
797                 int rc;
798                 rc = b_addchr(o, '\\');
799                 if (rc) return rc;
800         }
801         return b_addchr(o, ch);
802 }
803
804 /* belongs in utility.c */
805 char *simple_itoa(unsigned int i)
806 {
807         /* 21 digits plus null terminator, good for 64-bit or smaller ints */
808         static char local[22];
809         char *p = &local[21];
810         *p-- = '\0';
811         do {
812                 *p-- = '0' + i % 10;
813                 i /= 10;
814         } while (i > 0);
815         return p + 1;
816 }
817
818 static int b_adduint(o_string *o, unsigned int i)
819 {
820         int r;
821         char *p = simple_itoa(i);
822         /* no escape checking necessary */
823         do r=b_addchr(o, *p++); while (r==0 && *p);
824         return r;
825 }
826
827 static int static_get(struct in_str *i)
828 {
829         int ch=*i->p++;
830         if (ch=='\0') return EOF;
831         return ch;
832 }
833
834 static int static_peek(struct in_str *i)
835 {
836         return *i->p;
837 }
838
839 static inline void cmdedit_set_initial_prompt(void)
840 {
841 #ifdef BB_FEATURE_SH_SIMPLE_PROMPT
842         PS1 = NULL;
843 #else
844         PS1 = getenv("PS1");
845         if(PS1==0)
846                 PS1 = "\\w \\$ ";
847 #endif  
848 }
849
850 static inline void setup_prompt_string(int promptmode, char **prompt_str)
851 {
852         debug_printf("setup_prompt_string %d ",promptmode);
853 #ifdef BB_FEATURE_SH_SIMPLE_PROMPT
854         /* Set up the prompt */
855         if (promptmode == 1) {
856                 if (PS1)
857                         free(PS1);
858                 PS1=xmalloc(strlen(cwd)+4);
859                 sprintf(PS1, "%s %s", cwd, ( geteuid() != 0 ) ?  "$ ":"# ");
860                 *prompt_str = PS1;
861         } else {
862                 *prompt_str = PS2;
863         }
864 #else
865         *prompt_str = (promptmode==0)? PS1 : PS2;
866 #endif
867         debug_printf("result %s\n",*prompt_str);
868 }
869
870 static void get_user_input(struct in_str *i)
871 {
872         char *prompt_str;
873         static char the_command[BUFSIZ];
874
875         setup_prompt_string(i->promptmode, &prompt_str);
876 #ifdef BB_FEATURE_COMMAND_EDITING
877         /*
878          ** enable command line editing only while a command line
879          ** is actually being read; otherwise, we'll end up bequeathing
880          ** atexit() handlers and other unwanted stuff to our
881          ** child processes (rob@sysgo.de)
882          */
883         cmdedit_read_input(prompt_str, the_command);
884         cmdedit_terminate();
885 #else
886         fputs(prompt_str, stdout);
887         fflush(stdout);
888         the_command[0]=fgetc(i->file);
889         the_command[1]='\0';
890 #endif
891         i->p = the_command;
892 }
893
894 /* This is the magic location that prints prompts 
895  * and gets data back from the user */
896 static int file_get(struct in_str *i)
897 {
898         int ch;
899
900         ch = 0;
901         /* If there is data waiting, eat it up */
902         if (i->p && *i->p) {
903                 ch=*i->p++;
904         } else {
905                 /* need to double check i->file because we might be doing something
906                  * more complicated by now, like sourcing or substituting. */
907                 if (i->__promptme && interactive && i->file == stdin) {
908                         get_user_input(i);
909                         i->promptmode=2;
910                         i->__promptme = 0;
911                         if (i->p && *i->p) {
912                                 ch=*i->p++;
913                         }
914                 } else {
915                         ch = fgetc(i->file);
916                 }
917
918                 debug_printf("b_getch: got a %d\n", ch);
919         }
920         if (ch == '\n') i->__promptme=1;
921         return ch;
922 }
923
924 /* All the callers guarantee this routine will never be
925  * used right after a newline, so prompting is not needed.
926  */
927 static int file_peek(struct in_str *i)
928 {
929         if (i->p && *i->p) {
930                 return *i->p;
931         } else {
932                 static char buffer[2];
933                 buffer[0] = fgetc(i->file);
934                 buffer[1] = '\0';
935                 i->p = buffer;
936                 debug_printf("b_peek: got a %d\n", *i->p);
937                 return *i->p; 
938         }
939 }
940
941 static void setup_file_in_str(struct in_str *i, FILE *f)
942 {
943         i->peek = file_peek;
944         i->get = file_get;
945         i->__promptme=1;
946         i->promptmode=1;
947         i->file = f;
948         i->p = NULL;
949 }
950
951 static void setup_string_in_str(struct in_str *i, const char *s)
952 {
953         i->peek = static_peek;
954         i->get = static_get;
955         i->__promptme=1;
956         i->promptmode=1;
957         i->p = s;
958 }
959
960 static void mark_open(int fd)
961 {
962         struct close_me *new = xmalloc(sizeof(struct close_me));
963         new->fd = fd;
964         new->next = close_me_head;
965         close_me_head = new;
966 }
967
968 static void mark_closed(int fd)
969 {
970         struct close_me *tmp;
971         if (close_me_head == NULL || close_me_head->fd != fd)
972                 error_msg_and_die("corrupt close_me");
973         tmp = close_me_head;
974         close_me_head = close_me_head->next;
975         free(tmp);
976 }
977
978 static void close_all()
979 {
980         struct close_me *c;
981         for (c=close_me_head; c; c=c->next) {
982                 close(c->fd);
983         }
984         close_me_head = NULL;
985 }
986
987 /* squirrel != NULL means we squirrel away copies of stdin, stdout,
988  * and stderr if they are redirected. */
989 static int setup_redirects(struct child_prog *prog, int squirrel[])
990 {
991         int openfd, mode;
992         struct redir_struct *redir;
993
994         for (redir=prog->redirects; redir; redir=redir->next) {
995                 if (redir->dup == -1) {
996                         mode=redir_table[redir->type].mode;
997                         openfd = open(redir->word.gl_pathv[0], mode, 0666);
998                         if (openfd < 0) {
999                         /* this could get lost if stderr has been redirected, but
1000                            bash and ash both lose it as well (though zsh doesn't!) */
1001                                 fprintf(stderr,"error opening %s: %s\n", redir->word.gl_pathv[0],
1002                                         strerror(errno));
1003                                 return 1;
1004                         }
1005                 } else {
1006                         openfd = redir->dup;
1007                 }
1008
1009                 if (openfd != redir->fd) {
1010                         if (squirrel && redir->fd < 3) {
1011                                 squirrel[redir->fd] = dup(redir->fd);
1012                         }
1013                         if (openfd == -3) {
1014                                 close(openfd);
1015                         } else {
1016                                 dup2(openfd, redir->fd);
1017                                 close(openfd);
1018                         }
1019                 }
1020         }
1021         return 0;
1022 }
1023
1024 static void restore_redirects(int squirrel[])
1025 {
1026         int i, fd;
1027         for (i=0; i<3; i++) {
1028                 fd = squirrel[i];
1029                 if (fd != -1) {
1030                         /* No error checking.  I sure wouldn't know what
1031                          * to do with an error if I found one! */
1032                         dup2(fd, i);
1033                         close(fd);
1034                 }
1035         }
1036 }
1037
1038 /* XXX this definitely needs some more thought, work, and
1039  * cribbing from other shells */
1040 static int pipe_wait(struct pipe *pi)
1041 {
1042         int rcode=0, i, pid, running, status;
1043         running = pi->num_progs;
1044         while (running) {
1045                 pid=waitpid(-1, &status, 0);
1046                 if (pid < 0) perror_msg_and_die("waitpid");
1047                 for (i=0; i < pi->num_progs; i++) {
1048                         if (pi->progs[i].pid == pid) {
1049                                 if (i==pi->num_progs-1) rcode=WEXITSTATUS(status);
1050                                 pi->progs[i].pid = 0;
1051                                 running--;
1052                                 break;
1053                         }
1054                 }
1055         }
1056         return rcode;
1057 }
1058
1059 /* very simple version for testing */
1060 static void pseudo_exec(struct child_prog *child)
1061 {
1062         int i, rcode;
1063         struct built_in_command *x;
1064         if (child->argv) {
1065                 for (i=0; is_assignment(child->argv[i]); i++) {
1066                         putenv(strdup(child->argv[i]));
1067                 }
1068                 child->argv+=i;  /* XXX this hack isn't so horrible, since we are about
1069                                         to exit, and therefore don't need to keep data
1070                                         structures consistent for free() use. */
1071                 /* If a variable is assigned in a forest, and nobody listens,
1072                  * was it ever really set?
1073                  */
1074                 if (child->argv[0] == NULL) exit(EXIT_SUCCESS);
1075
1076                 /*
1077                  * Check if the command matches any of the builtins.
1078                  * Depending on context, this might be redundant.  But it's
1079                  * easier to waste a few CPU cycles than it is to figure out
1080                  * if this is one of those cases.
1081                  */
1082                 for (x = bltins; x->cmd; x++) {
1083                         if (strcmp(child->argv[0], x->cmd) == 0 ) {
1084                                 debug_printf("builtin exec %s\n", child->argv[0]);
1085                                 exit(x->function(child));
1086                         }
1087                 }
1088
1089                 /* Check if the command matches any busybox internal commands
1090                  * ("applets") here.  
1091                  * FIXME: This feature is not 100% safe, since
1092                  * BusyBox is not fully reentrant, so we have no guarantee the things
1093                  * from the .bss are still zeroed, or that things from .data are still
1094                  * at their defaults.  We could exec ourself from /proc/self/exe, but I
1095                  * really dislike relying on /proc for things.  We could exec ourself
1096                  * from global_argv[0], but if we are in a chroot, we may not be able
1097                  * to find ourself... */ 
1098 #ifdef BB_FEATURE_SH_STANDALONE_SHELL
1099                 {
1100                         int argc_l;
1101                         char** argv_l=child->argv;
1102                         char *name = child->argv[0];
1103
1104 #ifdef BB_FEATURE_SH_APPLETS_ALWAYS_WIN
1105                         /* Following discussions from November 2000 on the busybox mailing
1106                          * list, the default configuration, (without
1107                          * get_last_path_component()) lets the user force use of an
1108                          * external command by specifying the full (with slashes) filename.
1109                          * If you enable BB_FEATURE_SH_APPLETS_ALWAYS_WIN, then applets
1110                          * _aways_ override external commands, so if you want to run
1111                          * /bin/cat, it will use BusyBox cat even if /bin/cat exists on the
1112                          * filesystem and is _not_ busybox.  Some systems may want this,
1113                          * most do not.  */
1114                         name = get_last_path_component(name);
1115 #endif
1116                         /* Count argc for use in a second... */
1117                         for(argc_l=0;*argv_l!=NULL; argv_l++, argc_l++);
1118                         optind = 1;
1119                         debug_printf("running applet %s\n", name);
1120                         run_applet_by_name(name, argc_l, child->argv);
1121                 }
1122 #endif
1123                 debug_printf("exec of %s\n",child->argv[0]);
1124                 execvp(child->argv[0],child->argv);
1125                 perror("execvp");
1126                 exit(1);
1127         } else if (child->group) {
1128                 debug_printf("runtime nesting to group\n");
1129                 interactive=0;    /* crucial!!!! */
1130                 rcode = run_list_real(child->group);
1131                 /* OK to leak memory by not calling run_list_test,
1132                  * since this process is about to exit */
1133                 exit(rcode);
1134         } else {
1135                 /* Can happen.  See what bash does with ">foo" by itself. */
1136                 debug_printf("trying to pseudo_exec null command\n");
1137                 exit(EXIT_SUCCESS);
1138         }
1139 }
1140
1141 static void insert_bg_job(struct pipe *pi)
1142 {
1143         struct pipe *thejob;
1144
1145         /* Linear search for the ID of the job to use */
1146         pi->jobid = 1;
1147         for (thejob = job_list->head; thejob; thejob = thejob->next)
1148                 if (thejob->jobid >= pi->jobid)
1149                         pi->jobid = thejob->jobid + 1;
1150
1151         /* add thejob to the list of running jobs */
1152         if (!job_list->head) {
1153                 thejob = job_list->head = xmalloc(sizeof(*thejob));
1154         } else {
1155                 for (thejob = job_list->head; thejob->next; thejob = thejob->next) /* nothing */;
1156                 thejob->next = xmalloc(sizeof(*thejob));
1157                 thejob = thejob->next;
1158         }
1159
1160         /* physically copy the struct job */
1161         memcpy(thejob, pi, sizeof(struct pipe));
1162         thejob->next = NULL;
1163         //thejob->num_progs = 0;
1164         thejob->running_progs = thejob->num_progs;
1165         thejob->stopped_progs = 0;
1166         thejob->text = xmalloc(MAX_LINE);
1167
1168         //if (pi->progs[0] && pi->progs[0].argv && pi->progs[0].argv[0])
1169         {
1170                 char *bar=thejob->text;
1171                 char **foo=pi->progs[0].argv;
1172                 while(foo && *foo) {
1173                         bar += sprintf(bar, "%s ", *foo++);
1174                 }
1175         }
1176
1177         /* we don't wait for background thejobs to return -- append it 
1178            to the list of backgrounded thejobs and leave it alone */
1179         printf("[%d] %d\n", thejob->jobid, thejob->progs[0].pid);
1180         last_bg_pid = thejob->progs[0].pid;
1181 }
1182
1183 /* remove a backgrounded job from a jobset */
1184 static void remove_bg_job(struct pipe *pi)
1185 {
1186         struct pipe *prev_pipe;
1187
1188         if (pi == job_list->head) {
1189                 job_list->head = pi->next;
1190         } else {
1191                 prev_pipe = job_list->head;
1192                 while (prev_pipe->next != pi)
1193                         prev_pipe = prev_pipe->next;
1194                 prev_pipe->next = pi->next;
1195         }
1196
1197         free_pipe(pi);
1198         free(pi);
1199 }
1200
1201 /* free up all memory from a pipe */
1202 static void free_pipe(struct pipe *pi)
1203 {
1204         int i;
1205
1206         for (i = 0; i < pi->num_progs; i++) {
1207                 free(pi->progs[i].argv);
1208                 if (pi->progs[i].redirects)
1209                         free(pi->progs[i].redirects);
1210         }
1211         if (pi->progs)
1212                 free(pi->progs);
1213         if (pi->text)
1214                 free(pi->text);
1215         if (pi->cmdbuf)
1216                 free(pi->cmdbuf);
1217         memset(pi, 0, sizeof(struct pipe));
1218 }
1219
1220
1221 /* Checks to see if any background processes have exited -- if they 
1222    have, figure out why and see if a job has completed */
1223 static void checkjobs()
1224 {
1225         int status;
1226         int prognum = 0;
1227         struct pipe *pi;
1228         pid_t childpid;
1229
1230         while ((childpid = waitpid(-1, &status, WNOHANG | WUNTRACED)) > 0) {
1231                 for (pi = job_list->head; pi; pi = pi->next) {
1232                         prognum = 0;
1233                         while (prognum < pi->num_progs &&
1234                                    pi->progs[prognum].pid != childpid) prognum++;
1235                         if (prognum < pi->num_progs)
1236                                 break;
1237                 }
1238
1239                 if (WIFEXITED(status) || WIFSIGNALED(status)) {
1240                         /* child exited */
1241                         pi->running_progs--;
1242                         pi->progs[prognum].pid = 0;
1243
1244                         if (!pi->running_progs) {
1245                                 printf(JOB_STATUS_FORMAT, pi->jobid, "Done", pi->text);
1246                                 remove_bg_job(pi);
1247                         }
1248                 } else {
1249                         if(pi==NULL)
1250                                 break;
1251                         /* child stopped */
1252                         pi->stopped_progs++;
1253                         pi->progs[prognum].is_stopped = 1;
1254
1255                         if (pi->stopped_progs == pi->num_progs) {
1256                                 printf(JOB_STATUS_FORMAT, pi->jobid, "Stopped", pi->text);
1257                         }
1258                 }
1259         }
1260
1261         if (childpid == -1 && errno != ECHILD)
1262                 perror_msg("waitpid");
1263
1264         /* move the shell to the foreground */
1265         if (tcsetpgrp(0, getpgrp()) && errno != ENOTTY)
1266                 perror_msg("tcsetpgrp"); 
1267 }
1268
1269 /* run_pipe_real() starts all the jobs, but doesn't wait for anything
1270  * to finish.  See pipe_wait().
1271  *
1272  * return code is normally -1, when the caller has to wait for children
1273  * to finish to determine the exit status of the pipe.  If the pipe
1274  * is a simple builtin command, however, the action is done by the
1275  * time run_pipe_real returns, and the exit code is provided as the
1276  * return value.
1277  *
1278  * The input of the pipe is always stdin, the output is always
1279  * stdout.  The outpipe[] mechanism in BusyBox-0.48 lash is bogus,
1280  * because it tries to avoid running the command substitution in
1281  * subshell, when that is in fact necessary.  The subshell process
1282  * now has its stdout directed to the input of the appropriate pipe,
1283  * so this routine is noticeably simpler.
1284  */
1285 static int run_pipe_real(struct pipe *pi)
1286 {
1287         int i;
1288         int ctty;
1289         int nextin, nextout;
1290         int pipefds[2];                         /* pipefds[0] is for reading */
1291         struct child_prog *child;
1292         struct built_in_command *x;
1293
1294         ctty = -1;
1295         nextin = 0;
1296         pi->pgrp = 0;
1297
1298         /* Check if we are supposed to run in the foreground */
1299         if (interactive && pi->followup!=PIPE_BG) {
1300                 if ((pi->pgrp = tcgetpgrp(ctty = 2)) < 0
1301                                 && (pi->pgrp = tcgetpgrp(ctty = 0)) < 0
1302                                 && (pi->pgrp = tcgetpgrp(ctty = 1)) < 0)
1303                         return errno = ENOTTY, -1;
1304
1305                 if (pi->pgrp < 0 && pi->pgrp != getpgrp())
1306                         return errno = EPERM, -1;
1307         }
1308
1309         /* Check if this is a simple builtin (not part of a pipe).
1310          * Builtins within pipes have to fork anyway, and are handled in
1311          * pseudo_exec.  "echo foo | read bar" doesn't work on bash, either.
1312          */
1313         if (pi->num_progs == 1 && pi->progs[0].argv != NULL) {
1314                 child = & (pi->progs[0]);
1315                 if (child->group && ! child->subshell) {
1316                         int squirrel[] = {-1, -1, -1};
1317                         int rcode;
1318                         debug_printf("non-subshell grouping\n");
1319                         setup_redirects(child, squirrel);
1320                         /* XXX could we merge code with following builtin case,
1321                          * by creating a pseudo builtin that calls run_list_real? */
1322                         rcode = run_list_real(child->group);
1323                         restore_redirects(squirrel);
1324                         return rcode;
1325                 }
1326                 for (i=0; is_assignment(child->argv[i]); i++) { /* nothing */ }
1327                 if (i!=0 && child->argv[i]==NULL) {
1328                         /* assignments, but no command: set the local environment */
1329                         for (i=0; child->argv[i]!=NULL; i++) {
1330                                 set_local_var(child->argv[i]);
1331                         }
1332                         return EXIT_SUCCESS;   /* don't worry about errors in set_local_var() yet */
1333                 }
1334                 for (x = bltins; x->cmd; x++) {
1335                         if (strcmp(child->argv[i], x->cmd) == 0 ) {
1336                                 int squirrel[] = {-1, -1, -1};
1337                                 int rcode;
1338                                 if (x->function == builtin_exec && child->argv[i+1]==NULL) {
1339                                         debug_printf("magic exec\n");
1340                                         setup_redirects(child,NULL);
1341                                         return EXIT_SUCCESS;
1342                                 }
1343                                 debug_printf("builtin inline %s\n", child->argv[0]);
1344                                 /* XXX setup_redirects acts on file descriptors, not FILEs.
1345                                  * This is perfect for work that comes after exec().
1346                                  * Is it really safe for inline use?  Experimentally,
1347                                  * things seem to work with glibc. */
1348                                 setup_redirects(child, squirrel);
1349                                 for (i=0; is_assignment(child->argv[i]); i++) {
1350                                         putenv(strdup(child->argv[i]));
1351                                 }
1352                                 child->argv+=i;  /* XXX horrible hack */
1353                                 rcode = x->function(child);
1354                                 child->argv-=i;  /* XXX restore hack so free() can work right */
1355                                 restore_redirects(squirrel);
1356                                 return rcode;
1357                         }
1358                 }
1359         }
1360
1361         for (i = 0; i < pi->num_progs; i++) {
1362                 child = & (pi->progs[i]);
1363
1364                 /* pipes are inserted between pairs of commands */
1365                 if ((i + 1) < pi->num_progs) {
1366                         if (pipe(pipefds)<0) perror_msg_and_die("pipe");
1367                         nextout = pipefds[1];
1368                 } else {
1369                         nextout=1;
1370                         pipefds[0] = -1;
1371                 }
1372
1373                 /* XXX test for failed fork()? */
1374                 if (!(child->pid = fork())) {
1375
1376                         signal(SIGTTOU, SIG_DFL);
1377                         
1378                         close_all();
1379
1380                         if (nextin != 0) {
1381                                 dup2(nextin, 0);
1382                                 close(nextin);
1383                         }
1384                         if (nextout != 1) {
1385                                 dup2(nextout, 1);
1386                                 close(nextout);
1387                         }
1388                         if (pipefds[0]!=-1) {
1389                                 close(pipefds[0]);  /* opposite end of our output pipe */
1390                         }
1391
1392                         /* Like bash, explicit redirects override pipes,
1393                          * and the pipe fd is available for dup'ing. */
1394                         setup_redirects(child,NULL);
1395                         
1396                         if (pi->followup!=PIPE_BG) {
1397                                 /* If we (the child) win the race, put ourselves in the process
1398                                  * group whose leader is the first process in this pipe. */
1399                                 if (pi->pgrp < 0) {
1400                                         pi->pgrp = child->pid;
1401                                 }
1402                                 if (setpgid(0, pi->pgrp) == 0) {
1403                                         signal(SIGTTOU, SIG_IGN);
1404                                         tcsetpgrp(ctty, pi->pgrp);
1405                                         signal(SIGTTOU, SIG_DFL);
1406                                 }
1407                         }
1408
1409                         pseudo_exec(child);
1410                 }
1411                 /* Put our child in the process group whose leader is the
1412                  * first process in this pipe. */
1413                 if (pi->pgrp < 0) {
1414                         pi->pgrp = child->pid;
1415                 }
1416                 /* Don't check for errors.  The child may be dead already,
1417                  * in which case setpgid returns error code EACCES. */
1418                 setpgid(child->pid, pi->pgrp);
1419
1420                 if (nextin != 0)
1421                         close(nextin);
1422                 if (nextout != 1)
1423                         close(nextout);
1424
1425                 /* If there isn't another process, nextin is garbage 
1426                    but it doesn't matter */
1427                 nextin = pipefds[0];
1428         }
1429         return -1;
1430 }
1431
1432 static int run_list_real(struct pipe *pi)
1433 {
1434         int rcode=0;
1435         int if_code=0, next_if_code=0;  /* need double-buffer to handle elif */
1436         reserved_style rmode, skip_more_in_this_rmode=RES_XXXX;
1437         for (;pi;pi=pi->next) {
1438                 rmode = pi->r_mode;
1439                 debug_printf("rmode=%d  if_code=%d  next_if_code=%d skip_more=%d\n", rmode, if_code, next_if_code, skip_more_in_this_rmode);
1440                 if (rmode == skip_more_in_this_rmode) continue;
1441                 skip_more_in_this_rmode = RES_XXXX;
1442                 if (rmode == RES_THEN || rmode == RES_ELSE) if_code = next_if_code;
1443                 if (rmode == RES_THEN &&  if_code) continue;
1444                 if (rmode == RES_ELSE && !if_code) continue;
1445                 if (rmode == RES_ELIF && !if_code) continue;
1446                 if (pi->num_progs == 0) continue;
1447                 rcode = run_pipe_real(pi);
1448                 if (rcode!=-1) {
1449                         /* We only ran a builtin: rcode was set by the return value
1450                          * of run_pipe_real(), and we don't need to wait for anything. */
1451                 } else if (pi->followup==PIPE_BG) {
1452                         /* XXX check bash's behavior with nontrivial pipes */
1453                         /* XXX compute jobid */
1454                         /* XXX what does bash do with attempts to background builtins? */
1455                         insert_bg_job(pi);
1456                         rcode = EXIT_SUCCESS;
1457                 } else {
1458
1459                         if (interactive) {
1460                                 /* move the new process group into the foreground */
1461                                 /* suppress messages when run from /linuxrc mag@sysgo.de */
1462                                 if (tcsetpgrp(0, pi->pgrp) && errno != ENOTTY)
1463                                         perror_msg("tcsetpgrp");
1464                                 rcode = pipe_wait(pi);
1465                                 if (tcsetpgrp(0, getpgrp()) && errno != ENOTTY)
1466                                         perror_msg("tcsetpgrp");
1467                         } else {
1468                                 rcode = pipe_wait(pi);
1469                         }
1470                 }
1471                 last_return_code=rcode;
1472                 if ( rmode == RES_IF || rmode == RES_ELIF )
1473                         next_if_code=rcode;  /* can be overwritten a number of times */
1474                 if ( (rcode==EXIT_SUCCESS && pi->followup==PIPE_OR) ||
1475                      (rcode!=EXIT_SUCCESS && pi->followup==PIPE_AND) )
1476                         skip_more_in_this_rmode=rmode;
1477         }
1478         checkjobs();
1479         return rcode;
1480 }
1481
1482 /* broken, of course, but OK for testing */
1483 static char *indenter(int i)
1484 {
1485         static char blanks[]="                                    ";
1486         return &blanks[sizeof(blanks)-i-1];
1487 }
1488
1489 /* return code is the exit status of the pipe */
1490 static int run_pipe_test(struct pipe *pi, int indent)
1491 {
1492         char **p;
1493         struct child_prog *child;
1494         struct redir_struct *r, *rnext;
1495         int a, i, ret_code=0;
1496         char *ind = indenter(indent);
1497         final_printf("%s run pipe: (pid %d)\n",ind,getpid());
1498         for (i=0; i<pi->num_progs; i++) {
1499                 child = &pi->progs[i];
1500                 final_printf("%s  command %d:\n",ind,i);
1501                 if (child->argv) {
1502                         for (a=0,p=child->argv; *p; a++,p++) {
1503                                 final_printf("%s   argv[%d] = %s\n",ind,a,*p);
1504                         }
1505                         globfree(&child->glob_result);
1506                         child->argv=NULL;
1507                 } else if (child->group) {
1508                         final_printf("%s   begin group (subshell:%d)\n",ind, child->subshell);
1509                         ret_code = run_list_test(child->group,indent+3);
1510                         final_printf("%s   end group\n",ind);
1511                 } else {
1512                         final_printf("%s   (nil)\n",ind);
1513                 }
1514                 for (r=child->redirects; r; r=rnext) {
1515                         final_printf("%s   redirect %d%s", ind, r->fd, redir_table[r->type].descrip);
1516                         if (r->dup == -1) {
1517                                 final_printf(" %s\n", *r->word.gl_pathv);
1518                                 globfree(&r->word);
1519                         } else {
1520                                 final_printf("&%d\n", r->dup);
1521                         }
1522                         rnext=r->next;
1523                         free(r);
1524                 }
1525                 child->redirects=NULL;
1526         }
1527         free(pi->progs);   /* children are an array, they get freed all at once */
1528         pi->progs=NULL;
1529         return ret_code;
1530 }
1531
1532 static int run_list_test(struct pipe *head, int indent)
1533 {
1534         int rcode=0;   /* if list has no members */
1535         struct pipe *pi, *next;
1536         char *ind = indenter(indent);
1537         for (pi=head; pi; pi=next) {
1538                 if (pi->num_progs == 0) break;
1539                 final_printf("%s pipe reserved mode %d\n", ind, pi->r_mode);
1540                 rcode = run_pipe_test(pi, indent);
1541                 final_printf("%s pipe followup code %d\n", ind, pi->followup);
1542                 next=pi->next;
1543                 pi->next=NULL;
1544                 free(pi);
1545         }
1546         return rcode;   
1547 }
1548
1549 /* Select which version we will use */
1550 static int run_list(struct pipe *pi)
1551 {
1552         int rcode=0;
1553         if (fake_mode==0) {
1554                 rcode = run_list_real(pi);
1555         } 
1556         /* run_list_test has the side effect of clearing memory
1557          * In the long run that function can be merged with run_list_real,
1558          * but doing that now would hobble the debugging effort. */
1559         run_list_test(pi,0);
1560         return rcode;
1561 }
1562
1563 /* The API for glob is arguably broken.  This routine pushes a non-matching
1564  * string into the output structure, removing non-backslashed backslashes.
1565  * If someone can prove me wrong, by performing this function within the
1566  * original glob(3) api, feel free to rewrite this routine into oblivion.
1567  * Return code (0 vs. GLOB_NOSPACE) matches glob(3).
1568  * XXX broken if the last character is '\\', check that before calling.
1569  */
1570 static int globhack(const char *src, int flags, glob_t *pglob)
1571 {
1572         int cnt, pathc;
1573         const char *s;
1574         char *dest;
1575         for (cnt=1, s=src; *s; s++) {
1576                 if (*s == '\\') s++;
1577                 cnt++;
1578         }
1579         dest = malloc(cnt);
1580         if (!dest) return GLOB_NOSPACE;
1581         if (!(flags & GLOB_APPEND)) {
1582                 pglob->gl_pathv=NULL;
1583                 pglob->gl_pathc=0;
1584                 pglob->gl_offs=0;
1585                 pglob->gl_offs=0;
1586         }
1587         pathc = ++pglob->gl_pathc;
1588         pglob->gl_pathv = realloc(pglob->gl_pathv, (pathc+1)*sizeof(*pglob->gl_pathv));
1589         if (pglob->gl_pathv == NULL) return GLOB_NOSPACE;
1590         pglob->gl_pathv[pathc-1]=dest;
1591         pglob->gl_pathv[pathc]=NULL;
1592         for (s=src; *s; s++, dest++) {
1593                 if (*s == '\\') s++;
1594                 *dest = *s;
1595         }
1596         *dest='\0';
1597         return 0;
1598 }
1599
1600 /* XXX broken if the last character is '\\', check that before calling */
1601 static int glob_needed(const char *s)
1602 {
1603         for (; *s; s++) {
1604                 if (*s == '\\') s++;
1605                 if (strchr("*[?",*s)) return 1;
1606         }
1607         return 0;
1608 }
1609
1610 #if 0
1611 static void globprint(glob_t *pglob)
1612 {
1613         int i;
1614         debug_printf("glob_t at %p:\n", pglob);
1615         debug_printf("  gl_pathc=%d  gl_pathv=%p  gl_offs=%d  gl_flags=%d\n",
1616                 pglob->gl_pathc, pglob->gl_pathv, pglob->gl_offs, pglob->gl_flags);
1617         for (i=0; i<pglob->gl_pathc; i++)
1618                 debug_printf("pglob->gl_pathv[%d] = %p = %s\n", i,
1619                         pglob->gl_pathv[i], pglob->gl_pathv[i]);
1620 }
1621 #endif
1622
1623 static int xglob(o_string *dest, int flags, glob_t *pglob)
1624 {
1625         int gr;
1626
1627         /* short-circuit for null word */
1628         /* we can code this better when the debug_printf's are gone */
1629         if (dest->length == 0) {
1630                 if (dest->nonnull) {
1631                         /* bash man page calls this an "explicit" null */
1632                         gr = globhack(dest->data, flags, pglob);
1633                         debug_printf("globhack returned %d\n",gr);
1634                 } else {
1635                         return 0;
1636                 }
1637         } else if (glob_needed(dest->data)) {
1638                 gr = glob(dest->data, flags, NULL, pglob);
1639                 debug_printf("glob returned %d\n",gr);
1640                 if (gr == GLOB_NOMATCH) {
1641                         /* quote removal, or more accurately, backslash removal */
1642                         gr = globhack(dest->data, flags, pglob);
1643                         debug_printf("globhack returned %d\n",gr);
1644                 }
1645         } else {
1646                 gr = globhack(dest->data, flags, pglob);
1647                 debug_printf("globhack returned %d\n",gr);
1648         }
1649         if (gr == GLOB_NOSPACE) {
1650                 fprintf(stderr,"out of memory during glob\n");
1651                 exit(1);
1652         }
1653         if (gr != 0) { /* GLOB_ABORTED ? */
1654                 fprintf(stderr,"glob(3) error %d\n",gr);
1655         }
1656         /* globprint(glob_target); */
1657         return gr;
1658 }
1659
1660 /* This is used to get/check local shell variables */
1661 static char *get_local_var(const char *s)
1662 {
1663         char **p;
1664         int len;
1665
1666         if (!s)
1667                 return NULL;
1668         if (!__shell_local_env)
1669                 return NULL;
1670         len = strlen(s);
1671
1672         for (p = __shell_local_env; *p; p++) {
1673                 if (memcmp(s, *p, len) == 0 && (*p)[len] == '=') {
1674                         return *p + len + 1;
1675                 }
1676         }
1677         return NULL;
1678 }
1679
1680 /* This is used to set local shell variables */
1681 static int set_local_var(const char *s)
1682 {
1683         char **ep;
1684         char *tmp,*name, *value;
1685         size_t size;
1686         size_t namelen;
1687         size_t vallen;
1688         int result=0;
1689
1690         name=tmp=strdup(s);
1691
1692         /* Assume when we enter this function that we are already in
1693          * NAME=VALUE format.  So the first order of business is to
1694          * split 's' on the '=' into 'name' and 'value' */ 
1695         value = strchr(name, '=');
1696         if (!value) {
1697                 result = -1;
1698                 goto done_already;
1699         }
1700         *value='\0';
1701         ++value;
1702
1703         namelen = strlen (name);
1704         vallen = strlen (value);
1705
1706         /* Now see how many local environment entries we have, and check
1707          * if we match an existing environment entry (so we can overwrite it) */
1708         size = 0;
1709         for (ep = __shell_local_env; ep && *ep != NULL; ++ep) {
1710                 if (!memcmp (*ep, name, namelen) && (*ep)[namelen] == '=')
1711                         break;
1712                 else
1713                         ++size;
1714         }
1715
1716         if (ep == NULL || *ep == NULL) {
1717                 static char **last_environ = NULL;
1718                 char **new_environ = (char **) malloc((size + 2) * sizeof(char *));
1719                 if (new_environ == NULL) {
1720                         result = -1;
1721                         goto done_already;
1722                 }
1723                 memcpy((__ptr_t) new_environ, (__ptr_t) __shell_local_env, 
1724                                 size * sizeof(char *));
1725
1726                 new_environ[size] = malloc (namelen + 1 + vallen + 1);
1727                 if (new_environ[size] == NULL) {
1728                         free (new_environ);
1729                         errno=ENOMEM;
1730                         result = -1;
1731                         goto done_already;
1732                 }
1733                 memcpy (new_environ[size], name, namelen);
1734                 new_environ[size][namelen] = '=';
1735                 memcpy (&new_environ[size][namelen + 1], value, vallen + 1);
1736
1737                 new_environ[size + 1] = NULL;
1738
1739                 if (last_environ != NULL)
1740                         free ((__ptr_t) last_environ);
1741                 last_environ = new_environ;
1742                 __shell_local_env = new_environ;
1743         }
1744         else {
1745                 size_t len = strlen (*ep);
1746                 if (len < namelen + 1 + vallen) {
1747                         char *new = malloc (namelen + 1 + vallen + 1);
1748                         if (new == NULL) {
1749                                 result = -1;
1750                                 goto done_already;
1751                         }
1752                         *ep = new;
1753                         memcpy (*ep, name, namelen);
1754                         (*ep)[namelen] = '=';
1755                 }
1756                 memcpy (&(*ep)[namelen + 1], value, vallen + 1);
1757         }
1758
1759         /* One last little detail...  If this variable is already
1760          * in the environment we must set it there as well... */
1761         tmp = getenv(name);
1762         if (tmp) { 
1763                 /* FIXME -- I leak memory!!!!! */
1764                 putenv(strdup(s));
1765         }
1766
1767 done_already:
1768         free(name);
1769         return result;
1770 }
1771
1772 static void unset_local_var(const char *name)
1773 {
1774         char **ep, **dp;
1775         size_t namelen;
1776
1777         if (!name)
1778                 return;
1779         namelen = strlen(name);
1780         for (dp = ep = __shell_local_env; ep && *ep != NULL; ++ep) {
1781                 if (memcmp (*ep, name, namelen)==0 && (*ep)[namelen] == '=') {
1782                         *dp = *ep;
1783                         ++dp;
1784                         *ep = NULL;
1785                         break;
1786                 }
1787         }
1788 }
1789
1790 static int is_assignment(const char *s)
1791 {
1792         if (s==NULL || !isalpha(*s)) return 0;
1793         ++s;
1794         while(isalnum(*s) || *s=='_') ++s;
1795         return *s=='=';
1796 }
1797
1798 /* the src parameter allows us to peek forward to a possible &n syntax
1799  * for file descriptor duplication, e.g., "2>&1".
1800  * Return code is 0 normally, 1 if a syntax error is detected in src.
1801  * Resource errors (in xmalloc) cause the process to exit */
1802 static int setup_redirect(struct p_context *ctx, int fd, redir_type style,
1803         struct in_str *input)
1804 {
1805         struct child_prog *child=ctx->child;
1806         struct redir_struct *redir = child->redirects;
1807         struct redir_struct *last_redir=NULL;
1808
1809         /* Create a new redir_struct and drop it onto the end of the linked list */
1810         while(redir) {
1811                 last_redir=redir;
1812                 redir=redir->next;
1813         }
1814         redir = xmalloc(sizeof(struct redir_struct));
1815         redir->next=NULL;
1816         if (last_redir) {
1817                 last_redir->next=redir;
1818         } else {
1819                 child->redirects=redir;
1820         }
1821
1822         redir->type=style;
1823         redir->fd= (fd==-1) ? redir_table[style].default_fd : fd ;
1824
1825         debug_printf("Redirect type %d%s\n", redir->fd, redir_table[style].descrip);
1826
1827         /* Check for a '2>&1' type redirect */ 
1828         redir->dup = redirect_dup_num(input);
1829         if (redir->dup == -2) return 1;  /* syntax error */
1830         if (redir->dup != -1) {
1831                 /* Erik had a check here that the file descriptor in question
1832                  * is legit; I postpone that to "run time"
1833                  * A "-" representation of "close me" shows up as a -3 here */
1834                 debug_printf("Duplicating redirect '%d>&%d'\n", redir->fd, redir->dup);
1835         } else {
1836                 /* We do _not_ try to open the file that src points to,
1837                  * since we need to return and let src be expanded first.
1838                  * Set ctx->pending_redirect, so we know what to do at the
1839                  * end of the next parsed word.
1840                  */
1841                 ctx->pending_redirect = redir;
1842         }
1843         return 0;
1844 }
1845
1846 struct pipe *new_pipe(void) {
1847         struct pipe *pi;
1848         pi = xmalloc(sizeof(struct pipe));
1849         pi->num_progs = 0;
1850         pi->progs = NULL;
1851         pi->next = NULL;
1852         pi->followup = 0;  /* invalid */
1853         return pi;
1854 }
1855
1856 static void initialize_context(struct p_context *ctx)
1857 {
1858         ctx->pipe=NULL;
1859         ctx->pending_redirect=NULL;
1860         ctx->child=NULL;
1861         ctx->list_head=new_pipe();
1862         ctx->pipe=ctx->list_head;
1863         ctx->w=RES_NONE;
1864         ctx->stack=NULL;
1865         done_command(ctx);   /* creates the memory for working child */
1866 }
1867
1868 /* normal return is 0
1869  * if a reserved word is found, and processed, return 1
1870  * should handle if, then, elif, else, fi, for, while, until, do, done.
1871  * case, function, and select are obnoxious, save those for later.
1872  */
1873 int reserved_word(o_string *dest, struct p_context *ctx)
1874 {
1875         struct reserved_combo {
1876                 char *literal;
1877                 int code;
1878                 long flag;
1879         };
1880         /* Mostly a list of accepted follow-up reserved words.
1881          * FLAG_END means we are done with the sequence, and are ready
1882          * to turn the compound list into a command.
1883          * FLAG_START means the word must start a new compound list.
1884          */
1885         static struct reserved_combo reserved_list[] = {
1886                 { "if",    RES_IF,    FLAG_THEN | FLAG_START },
1887                 { "then",  RES_THEN,  FLAG_ELIF | FLAG_ELSE | FLAG_FI },
1888                 { "elif",  RES_ELIF,  FLAG_THEN },
1889                 { "else",  RES_ELSE,  FLAG_FI   },
1890                 { "fi",    RES_FI,    FLAG_END  },
1891                 { "for",   RES_FOR,   FLAG_DO   | FLAG_START },
1892                 { "while", RES_WHILE, FLAG_DO   | FLAG_START },
1893                 { "until", RES_UNTIL, FLAG_DO   | FLAG_START },
1894                 { "do",    RES_DO,    FLAG_DONE },
1895                 { "done",  RES_DONE,  FLAG_END  }
1896         };
1897         struct reserved_combo *r;
1898         for (r=reserved_list;
1899 #define NRES sizeof(reserved_list)/sizeof(struct reserved_combo)
1900                 r<reserved_list+NRES; r++) {
1901                 if (strcmp(dest->data, r->literal) == 0) {
1902                         debug_printf("found reserved word %s, code %d\n",r->literal,r->code);
1903                         if (r->flag & FLAG_START) {
1904                                 struct p_context *new = xmalloc(sizeof(struct p_context));
1905                                 debug_printf("push stack\n");
1906                                 *new = *ctx;   /* physical copy */
1907                                 initialize_context(ctx);
1908                                 ctx->stack=new;
1909                         } else if ( ctx->w == RES_NONE || ! (ctx->old_flag & (1<<r->code))) {
1910                                 syntax();
1911                                 ctx->w = RES_SNTX;
1912                                 b_reset (dest);
1913                                 return 1;
1914                         }
1915                         ctx->w=r->code;
1916                         ctx->old_flag = r->flag;
1917                         if (ctx->old_flag & FLAG_END) {
1918                                 struct p_context *old;
1919                                 debug_printf("pop stack\n");
1920                                 old = ctx->stack;
1921                                 old->child->group = ctx->list_head;
1922                                 *ctx = *old;   /* physical copy */
1923                                 free(old);
1924                         }
1925                         b_reset (dest);
1926                         return 1;
1927                 }
1928         }
1929         return 0;
1930 }
1931
1932 /* normal return is 0.
1933  * Syntax or xglob errors return 1. */
1934 static int done_word(o_string *dest, struct p_context *ctx)
1935 {
1936         struct child_prog *child=ctx->child;
1937         glob_t *glob_target;
1938         int gr, flags = 0;
1939
1940         debug_printf("done_word: %s %p\n", dest->data, child);
1941         if (dest->length == 0 && !dest->nonnull) {
1942                 debug_printf("  true null, ignored\n");
1943                 return 0;
1944         }
1945         if (ctx->pending_redirect) {
1946                 glob_target = &ctx->pending_redirect->word;
1947         } else {
1948                 if (child->group) {
1949                         syntax();
1950                         return 1;  /* syntax error, groups and arglists don't mix */
1951                 }
1952                 if (!child->argv) {
1953                         debug_printf("checking %s for reserved-ness\n",dest->data);
1954                         if (reserved_word(dest,ctx)) return ctx->w==RES_SNTX;
1955                 }
1956                 glob_target = &child->glob_result;
1957                 if (child->argv) flags |= GLOB_APPEND;
1958         }
1959         gr = xglob(dest, flags, glob_target);
1960         if (gr != 0) return 1;
1961
1962         b_reset(dest);
1963         if (ctx->pending_redirect) {
1964                 ctx->pending_redirect=NULL;
1965                 if (glob_target->gl_pathc != 1) {
1966                         fprintf(stderr, "ambiguous redirect\n");
1967                         return 1;
1968                 }
1969         } else {
1970                 child->argv = glob_target->gl_pathv;
1971         }
1972         return 0;
1973 }
1974
1975 /* The only possible error here is out of memory, in which case
1976  * xmalloc exits. */
1977 static int done_command(struct p_context *ctx)
1978 {
1979         /* The child is really already in the pipe structure, so
1980          * advance the pipe counter and make a new, null child.
1981          * Only real trickiness here is that the uncommitted
1982          * child structure, to which ctx->child points, is not
1983          * counted in pi->num_progs. */
1984         struct pipe *pi=ctx->pipe;
1985         struct child_prog *prog=ctx->child;
1986
1987         if (prog && prog->group == NULL
1988                  && prog->argv == NULL
1989                  && prog->redirects == NULL) {
1990                 debug_printf("done_command: skipping null command\n");
1991                 return 0;
1992         } else if (prog) {
1993                 pi->num_progs++;
1994                 debug_printf("done_command: num_progs incremented to %d\n",pi->num_progs);
1995         } else {
1996                 debug_printf("done_command: initializing\n");
1997         }
1998         pi->progs = xrealloc(pi->progs, sizeof(*pi->progs) * (pi->num_progs+1));
1999
2000         prog = pi->progs + pi->num_progs;
2001         prog->redirects = NULL;
2002         prog->argv = NULL;
2003         prog->is_stopped = 0;
2004         prog->group = NULL;
2005         prog->glob_result.gl_pathv = NULL;
2006         prog->family = pi;
2007
2008         ctx->child=prog;
2009         /* but ctx->pipe and ctx->list_head remain unchanged */
2010         return 0;
2011 }
2012
2013 static int done_pipe(struct p_context *ctx, pipe_style type)
2014 {
2015         struct pipe *new_p;
2016         done_command(ctx);  /* implicit closure of previous command */
2017         debug_printf("done_pipe, type %d\n", type);
2018         ctx->pipe->followup = type;
2019         ctx->pipe->r_mode = ctx->w;
2020         new_p=new_pipe();
2021         ctx->pipe->next = new_p;
2022         ctx->pipe = new_p;
2023         ctx->child = NULL;
2024         done_command(ctx);  /* set up new pipe to accept commands */
2025         return 0;
2026 }
2027
2028 /* peek ahead in the in_str to find out if we have a "&n" construct,
2029  * as in "2>&1", that represents duplicating a file descriptor.
2030  * returns either -2 (syntax error), -1 (no &), or the number found.
2031  */
2032 static int redirect_dup_num(struct in_str *input)
2033 {
2034         int ch, d=0, ok=0;
2035         ch = b_peek(input);
2036         if (ch != '&') return -1;
2037
2038         b_getch(input);  /* get the & */
2039         ch=b_peek(input);
2040         if (ch == '-') {
2041                 b_getch(input);
2042                 return -3;  /* "-" represents "close me" */
2043         }
2044         while (isdigit(ch)) {
2045                 d = d*10+(ch-'0');
2046                 ok=1;
2047                 b_getch(input);
2048                 ch = b_peek(input);
2049         }
2050         if (ok) return d;
2051
2052         fprintf(stderr, "ambiguous redirect\n");
2053         return -2;
2054 }
2055
2056 /* If a redirect is immediately preceded by a number, that number is
2057  * supposed to tell which file descriptor to redirect.  This routine
2058  * looks for such preceding numbers.  In an ideal world this routine
2059  * needs to handle all the following classes of redirects...
2060  *     echo 2>foo     # redirects fd  2 to file "foo", nothing passed to echo
2061  *     echo 49>foo    # redirects fd 49 to file "foo", nothing passed to echo
2062  *     echo -2>foo    # redirects fd  1 to file "foo",    "-2" passed to echo
2063  *     echo 49x>foo   # redirects fd  1 to file "foo",   "49x" passed to echo
2064  * A -1 output from this program means no valid number was found, so the
2065  * caller should use the appropriate default for this redirection.
2066  */
2067 static int redirect_opt_num(o_string *o)
2068 {
2069         int num;
2070
2071         if (o->length==0) return -1;
2072         for(num=0; num<o->length; num++) {
2073                 if (!isdigit(*(o->data+num))) {
2074                         return -1;
2075                 }
2076         }
2077         /* reuse num (and save an int) */
2078         num=atoi(o->data);
2079         b_reset(o);
2080         return num;
2081 }
2082
2083 FILE *generate_stream_from_list(struct pipe *head)
2084 {
2085         FILE *pf;
2086 #if 1
2087         int pid, channel[2];
2088         if (pipe(channel)<0) perror_msg_and_die("pipe");
2089         pid=fork();
2090         if (pid<0) {
2091                 perror_msg_and_die("fork");
2092         } else if (pid==0) {
2093                 close(channel[0]);
2094                 if (channel[1] != 1) {
2095                         dup2(channel[1],1);
2096                         close(channel[1]);
2097                 }
2098 #if 0
2099 #define SURROGATE "surrogate response"
2100                 write(1,SURROGATE,sizeof(SURROGATE));
2101                 exit(run_list(head));
2102 #else
2103                 exit(run_list_real(head));   /* leaks memory */
2104 #endif
2105         }
2106         debug_printf("forked child %d\n",pid);
2107         close(channel[1]);
2108         pf = fdopen(channel[0],"r");
2109         debug_printf("pipe on FILE *%p\n",pf);
2110 #else
2111         run_list_test(head,0);
2112         pf=popen("echo surrogate response","r");
2113         debug_printf("started fake pipe on FILE *%p\n",pf);
2114 #endif
2115         return pf;
2116 }
2117
2118 /* this version hacked for testing purposes */
2119 /* return code is exit status of the process that is run. */
2120 static int process_command_subs(o_string *dest, struct p_context *ctx, struct in_str *input, int subst_end)
2121 {
2122         int retcode;
2123         o_string result=NULL_O_STRING;
2124         struct p_context inner;
2125         FILE *p;
2126         struct in_str pipe_str;
2127         initialize_context(&inner);
2128
2129         /* recursion to generate command */
2130         retcode = parse_stream(&result, &inner, input, subst_end);
2131         if (retcode != 0) return retcode;  /* syntax error or EOF */
2132         done_word(&result, &inner);
2133         done_pipe(&inner, PIPE_SEQ);
2134         b_free(&result);
2135
2136         p=generate_stream_from_list(inner.list_head);
2137         if (p==NULL) return 1;
2138         mark_open(fileno(p));
2139         setup_file_in_str(&pipe_str, p);
2140
2141         /* now send results of command back into original context */
2142         retcode = parse_stream(dest, ctx, &pipe_str, '\0');
2143         /* XXX In case of a syntax error, should we try to kill the child?
2144          * That would be tough to do right, so just read until EOF. */
2145         if (retcode == 1) {
2146                 while (b_getch(&pipe_str)!=EOF) { /* discard */ };
2147         }
2148
2149         debug_printf("done reading from pipe, pclose()ing\n");
2150         /* This is the step that wait()s for the child.  Should be pretty
2151          * safe, since we just read an EOF from its stdout.  We could try
2152          * to better, by using wait(), and keeping track of background jobs
2153          * at the same time.  That would be a lot of work, and contrary
2154          * to the KISS philosophy of this program. */
2155         mark_closed(fileno(p));
2156         retcode=pclose(p);
2157         debug_printf("pclosed, retcode=%d\n",retcode);
2158         /* XXX this process fails to trim a single trailing newline */
2159         return retcode;
2160 }
2161
2162 static int parse_group(o_string *dest, struct p_context *ctx,
2163         struct in_str *input, int ch)
2164 {
2165         int rcode, endch=0;
2166         struct p_context sub;
2167         struct child_prog *child = ctx->child;
2168         if (child->argv) {
2169                 syntax();
2170                 return 1;  /* syntax error, groups and arglists don't mix */
2171         }
2172         initialize_context(&sub);
2173         switch(ch) {
2174                 case '(': endch=')'; child->subshell=1; break;
2175                 case '{': endch='}'; break;
2176                 default: syntax();   /* really logic error */
2177         }
2178         rcode=parse_stream(dest,&sub,input,endch);
2179         done_word(dest,&sub); /* finish off the final word in the subcontext */
2180         done_pipe(&sub, PIPE_SEQ);  /* and the final command there, too */
2181         child->group = sub.list_head;
2182         return rcode;
2183         /* child remains "open", available for possible redirects */
2184 }
2185
2186 /* basically useful version until someone wants to get fancier,
2187  * see the bash man page under "Parameter Expansion" */
2188 static void lookup_param(o_string *dest, struct p_context *ctx, o_string *src)
2189 {
2190         const char *p=NULL;
2191         if (src->data) { 
2192                 p = getenv(src->data);
2193                 if (!p) 
2194                         p = get_local_var(src->data);
2195         }
2196         if (p) parse_string(dest, ctx, p);   /* recursion */
2197         b_free(src);
2198 }
2199
2200 /* return code: 0 for OK, 1 for syntax error */
2201 static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input)
2202 {
2203         int i, advance=0;
2204         o_string alt=NULL_O_STRING;
2205         char sep[]=" ";
2206         int ch = input->peek(input);  /* first character after the $ */
2207         debug_printf("handle_dollar: ch=%c\n",ch);
2208         if (isalpha(ch)) {
2209                 while(ch=b_peek(input),isalnum(ch) || ch=='_') {
2210                         b_getch(input);
2211                         b_addchr(&alt,ch);
2212                 }
2213                 lookup_param(dest, ctx, &alt);
2214         } else if (isdigit(ch)) {
2215                 i = ch-'0';  /* XXX is $0 special? */
2216                 if (i<global_argc) {
2217                         parse_string(dest, ctx, global_argv[i]); /* recursion */
2218                 }
2219                 advance = 1;
2220         } else switch (ch) {
2221                 case '$':
2222                         b_adduint(dest,getpid());
2223                         advance = 1;
2224                         break;
2225                 case '!':
2226                         if (last_bg_pid > 0) b_adduint(dest, last_bg_pid);
2227                         advance = 1;
2228                         break;
2229                 case '?':
2230                         b_adduint(dest,last_return_code);
2231                         advance = 1;
2232                         break;
2233                 case '#':
2234                         b_adduint(dest,global_argc ? global_argc-1 : 0);
2235                         advance = 1;
2236                         break;
2237                 case '{':
2238                         b_getch(input);
2239                         /* XXX maybe someone will try to escape the '}' */
2240                         while(ch=b_getch(input),ch!=EOF && ch!='}') {
2241                                 b_addchr(&alt,ch);
2242                         }
2243                         if (ch != '}') {
2244                                 syntax();
2245                                 return 1;
2246                         }
2247                         lookup_param(dest, ctx, &alt);
2248                         break;
2249                 case '(':
2250                         b_getch(input);
2251                         process_command_subs(dest, ctx, input, ')');
2252                         break;
2253                 case '*':
2254                         sep[0]=ifs[0];
2255                         for (i=1; i<global_argc; i++) {
2256                                 parse_string(dest, ctx, global_argv[i]);
2257                                 if (i+1 < global_argc) parse_string(dest, ctx, sep);
2258                         }
2259                         break;
2260                 case '@':
2261                 case '-':
2262                 case '_':
2263                         /* still unhandled, but should be eventually */
2264                         fprintf(stderr,"unhandled syntax: $%c\n",ch);
2265                         return 1;
2266                         break;
2267                 default:
2268                         b_addqchr(dest,'$',dest->quote);
2269         }
2270         /* Eat the character if the flag was set.  If the compiler
2271          * is smart enough, we could substitute "b_getch(input);"
2272          * for all the "advance = 1;" above, and also end up with
2273          * a nice size-optimized program.  Hah!  That'll be the day.
2274          */
2275         if (advance) b_getch(input);
2276         return 0;
2277 }
2278
2279 int parse_string(o_string *dest, struct p_context *ctx, const char *src)
2280 {
2281         struct in_str foo;
2282         setup_string_in_str(&foo, src);
2283         return parse_stream(dest, ctx, &foo, '\0');
2284 }
2285
2286 /* return code is 0 for normal exit, 1 for syntax error */
2287 int parse_stream(o_string *dest, struct p_context *ctx,
2288         struct in_str *input, int end_trigger)
2289 {
2290         unsigned int ch, m;
2291         int redir_fd;
2292         redir_type redir_style;
2293         int next;
2294
2295         /* Only double-quote state is handled in the state variable dest->quote.
2296          * A single-quote triggers a bypass of the main loop until its mate is
2297          * found.  When recursing, quote state is passed in via dest->quote. */
2298
2299         debug_printf("parse_stream, end_trigger=%d\n",end_trigger);
2300         while ((ch=b_getch(input))!=EOF) {
2301                 m = map[ch];
2302                 next = (ch == '\n') ? 0 : b_peek(input);
2303                 debug_printf("parse_stream: ch=%c (%d) m=%d quote=%d\n",
2304                         ch,ch,m,dest->quote);
2305                 if (m==0 || ((m==1 || m==2) && dest->quote)) {
2306                         b_addqchr(dest, ch, dest->quote);
2307                 } else {
2308                         if (m==2) {  /* unquoted IFS */
2309                                 done_word(dest, ctx);
2310                                 /* If we aren't performing a substitution, treat a newline as a
2311                                  * command separator.  */
2312                                 if (end_trigger != '\0' && ch=='\n')
2313                                         done_pipe(ctx,PIPE_SEQ);
2314                         }
2315                         if (ch == end_trigger && !dest->quote && ctx->w==RES_NONE) {
2316                                 debug_printf("leaving parse_stream\n");
2317                                 return 0;
2318                         }
2319 #if 0
2320                         if (ch=='\n') {
2321                                 /* Yahoo!  Time to run with it! */
2322                                 done_pipe(ctx,PIPE_SEQ);
2323                                 run_list(ctx->list_head);
2324                                 initialize_context(ctx);
2325                         }
2326 #endif
2327                         if (m!=2) switch (ch) {
2328                 case '#':
2329                         if (dest->length == 0 && !dest->quote) {
2330                                 while(ch=b_peek(input),ch!=EOF && ch!='\n') { b_getch(input); }
2331                         } else {
2332                                 b_addqchr(dest, ch, dest->quote);
2333                         }
2334                         break;
2335                 case '\\':
2336                         if (next == EOF) {
2337                                 syntax();
2338                                 return 1;
2339                         }
2340                         b_addqchr(dest, '\\', dest->quote);
2341                         b_addqchr(dest, b_getch(input), dest->quote);
2342                         break;
2343                 case '$':
2344                         if (handle_dollar(dest, ctx, input)!=0) return 1;
2345                         break;
2346                 case '\'':
2347                         dest->nonnull = 1;
2348                         while(ch=b_getch(input),ch!=EOF && ch!='\'') {
2349                                 b_addchr(dest,ch);
2350                         }
2351                         if (ch==EOF) {
2352                                 syntax();
2353                                 return 1;
2354                         }
2355                         break;
2356                 case '"':
2357                         dest->nonnull = 1;
2358                         dest->quote = !dest->quote;
2359                         break;
2360                 case '`':
2361                         process_command_subs(dest, ctx, input, '`');
2362                         break;
2363                 case '>':
2364                         redir_fd = redirect_opt_num(dest);
2365                         done_word(dest, ctx);
2366                         redir_style=REDIRECT_OVERWRITE;
2367                         if (next == '>') {
2368                                 redir_style=REDIRECT_APPEND;
2369                                 b_getch(input);
2370                         } else if (next == '(') {
2371                                 syntax();   /* until we support >(list) Process Substitution */
2372                                 return 1;
2373                         }
2374                         setup_redirect(ctx, redir_fd, redir_style, input);
2375                         break;
2376                 case '<':
2377                         redir_fd = redirect_opt_num(dest);
2378                         done_word(dest, ctx);
2379                         redir_style=REDIRECT_INPUT;
2380                         if (next == '<') {
2381                                 redir_style=REDIRECT_HEREIS;
2382                                 b_getch(input);
2383                         } else if (next == '>') {
2384                                 redir_style=REDIRECT_IO;
2385                                 b_getch(input);
2386                         } else if (next == '(') {
2387                                 syntax();   /* until we support <(list) Process Substitution */
2388                                 return 1;
2389                         }
2390                         setup_redirect(ctx, redir_fd, redir_style, input);
2391                         break;
2392                 case ';':
2393                         done_word(dest, ctx);
2394                         done_pipe(ctx,PIPE_SEQ);
2395                         break;
2396                 case '&':
2397                         done_word(dest, ctx);
2398                         if (next=='&') {
2399                                 b_getch(input);
2400                                 done_pipe(ctx,PIPE_AND);
2401                         } else {
2402                                 done_pipe(ctx,PIPE_BG);
2403                         }
2404                         break;
2405                 case '|':
2406                         done_word(dest, ctx);
2407                         if (next=='|') {
2408                                 b_getch(input);
2409                                 done_pipe(ctx,PIPE_OR);
2410                         } else {
2411                                 /* we could pick up a file descriptor choice here
2412                                  * with redirect_opt_num(), but bash doesn't do it.
2413                                  * "echo foo 2| cat" yields "foo 2". */
2414                                 done_command(ctx);
2415                         }
2416                         break;
2417                 case '(':
2418                 case '{':
2419                         if (parse_group(dest, ctx, input, ch)!=0) return 1;
2420                         break;
2421                 case ')':
2422                 case '}':
2423                         syntax();   /* Proper use of this character caught by end_trigger */
2424                         return 1;
2425                         break;
2426                 default:
2427                         syntax();   /* this is really an internal logic error */
2428                         return 1;
2429                         }
2430                 }
2431         }
2432         /* complain if quote?  No, maybe we just finished a command substitution
2433          * that was quoted.  Example:
2434          * $ echo "`cat foo` plus more" 
2435          * and we just got the EOF generated by the subshell that ran "cat foo"
2436          * The only real complaint is if we got an EOF when end_trigger != '\0',
2437          * that is, we were really supposed to get end_trigger, and never got
2438          * one before the EOF.  Can't use the standard "syntax error" return code,
2439          * so that parse_stream_outer can distinguish the EOF and exit smoothly. */
2440         if (end_trigger != '\0') return -1;
2441         return 0;
2442 }
2443
2444 void mapset(const unsigned char *set, int code)
2445 {
2446         const unsigned char *s;
2447         for (s=set; *s; s++) map[*s] = code;
2448 }
2449
2450 void update_ifs_map(void)
2451 {
2452         /* char *ifs and char map[256] are both globals. */
2453         ifs = getenv("IFS");
2454         if (ifs == NULL) ifs=" \t\n";
2455         /* Precompute a list of 'flow through' behavior so it can be treated
2456          * quickly up front.  Computation is necessary because of IFS.
2457          * Special case handling of IFS == " \t\n" is not implemented.
2458          * The map[] array only really needs two bits each, and on most machines
2459          * that would be faster because of the reduced L1 cache footprint.
2460          */
2461         memset(map,0,256);        /* most characters flow through always */
2462         mapset("\\$'\"`", 3);     /* never flow through */
2463         mapset("<>;&|(){}#", 1);  /* flow through if quoted */
2464         mapset(ifs, 2);           /* also flow through if quoted */
2465 }
2466
2467 /* most recursion does not come through here, the exeception is
2468  * from builtin_source() */
2469 int parse_stream_outer(struct in_str *inp)
2470 {
2471
2472         struct p_context ctx;
2473         o_string temp=NULL_O_STRING;
2474         int rcode;
2475         do {
2476                 initialize_context(&ctx);
2477                 update_ifs_map();
2478                 inp->promptmode=1;
2479                 rcode = parse_stream(&temp, &ctx, inp, '\n');
2480                 done_word(&temp, &ctx);
2481                 done_pipe(&ctx,PIPE_SEQ);
2482                 run_list(ctx.list_head);
2483         } while (rcode != -1);   /* loop on syntax errors, return on EOF */
2484         return 0;
2485 }
2486
2487 static int parse_string_outer(const char *s)
2488 {
2489         struct in_str input;
2490         setup_string_in_str(&input, s);
2491         return parse_stream_outer(&input);
2492 }
2493
2494 static int parse_file_outer(FILE *f)
2495 {
2496         int rcode;
2497         struct in_str input;
2498         setup_file_in_str(&input, f);
2499         rcode = parse_stream_outer(&input);
2500         return rcode;
2501 }
2502
2503 int shell_main(int argc, char **argv)
2504 {
2505         int opt;
2506         FILE *input;
2507         struct jobset joblist_end = { NULL, NULL };
2508
2509         /* (re?) initialize globals */
2510         ifs=NULL;
2511         fake_mode=0;
2512         interactive=0;
2513         close_me_head = NULL;
2514         job_list = &joblist_end;
2515         last_bg_pid=0;
2516         PS2 = "> ";
2517         __shell_local_env = 0;
2518
2519         last_return_code=EXIT_SUCCESS;
2520
2521         /* XXX what should these be while sourcing /etc/profile? */
2522         global_argc = argc;
2523         global_argv = argv;
2524
2525         /* If we get started under a job aware app (like bash 
2526          * for example), make sure we are now in charge so we 
2527          * don't fight over who gets the foreground */
2528         do { 
2529                 pid_t initialpgrp;
2530                 initialpgrp = tcgetpgrp(fileno(stderr));
2531                 if (initialpgrp < 0) {
2532                         error_msg_and_die("sh: can't access tty; job control disabled\n");
2533                 }
2534                 if (initialpgrp == -1)
2535                         initialpgrp = getpgrp();
2536                 else if (initialpgrp != getpgrp()) {
2537                         killpg(initialpgrp, SIGTTIN);
2538                         continue;
2539                 }
2540         } while (0);
2541         /* don't pay any attention to this signal; it just confuses 
2542            things and isn't really meant for shells anyway */
2543         signal(SIGTTOU, SIG_IGN);
2544         setpgid(0, getpid());
2545         tcsetpgrp(fileno(stderr), getpid());
2546
2547         if (argv[0] && argv[0][0] == '-') {
2548                 debug_printf("\nsourcing /etc/profile\n");
2549                 input = xfopen("/etc/profile", "r");
2550                 mark_open(fileno(input));
2551                 parse_file_outer(input);
2552                 mark_closed(fileno(input));
2553                 fclose(input);
2554         }
2555         input=stdin;
2556         
2557         /* initialize the cwd -- this is never freed...*/
2558         cwd = xgetcwd(0);
2559         if (!cwd)
2560                 cwd = unknown;
2561 #ifdef BB_FEATURE_COMMAND_EDITING
2562         cmdedit_set_initial_prompt();
2563 #else
2564         PS1 = NULL;
2565 #endif
2566         
2567         while ((opt = getopt(argc, argv, "c:xif")) > 0) {
2568                 switch (opt) {
2569                         case 'c':
2570                                 {
2571                                         global_argv = argv+optind;
2572                                         global_argc = argc-optind;
2573                                         opt = parse_string_outer(optarg);
2574                                         goto final_return;
2575                                 }
2576                                 break;
2577                         case 'i':
2578                                 interactive++;
2579                                 break;
2580                         case 'f':
2581                                 fake_mode++;
2582                                 break;
2583                         default:
2584                                 fprintf(stderr, "Usage: sh [FILE]...\n"
2585                                                 "   or: sh -c command [args]...\n\n");
2586                                 exit(EXIT_FAILURE);
2587                 }
2588         }
2589         /* A shell is interactive if the `-i' flag was given, or if all of
2590          * the following conditions are met:
2591          *        no -c command
2592          *    no arguments remaining or the -s flag given
2593          *    standard input is a terminal
2594          *    standard output is a terminal
2595          *    Refer to Posix.2, the description of the `sh' utility. */
2596         if (argv[optind]==NULL && input==stdin &&
2597                         isatty(fileno(stdin)) && isatty(fileno(stdout))) {
2598                 interactive++;
2599         }
2600
2601         debug_printf("\ninteractive=%d\n", interactive);
2602         if (interactive) {
2603                 /* Looks like they want an interactive shell */
2604                 fprintf(stdout, "\nhush -- the humble shell v0.01 (testing)\n\n");
2605                 opt=parse_file_outer(stdin);
2606                 goto final_return;
2607         }
2608
2609         debug_printf("\nrunning script '%s'\n", argv[optind]);
2610         global_argv = argv+optind;
2611         global_argc = argc-optind;
2612         input = xfopen(argv[optind], "r");
2613         opt = parse_file_outer(input);
2614
2615 #ifdef BB_FEATURE_CLEAN_UP
2616         fclose(input.file);
2617 #endif
2618
2619 final_return:
2620         return(opt?opt:last_return_code);
2621 }