*: fix places where we were still using malloc/realloc
[oweals/busybox.git] / shell / ash.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * ash shell port for busybox
4  *
5  * Copyright (c) 1989, 1991, 1993, 1994
6  *      The Regents of the University of California.  All rights reserved.
7  *
8  * Copyright (c) 1997-2005 Herbert Xu <herbert@gondor.apana.org.au>
9  * was re-ported from NetBSD and debianized.
10  *
11  * This code is derived from software contributed to Berkeley by
12  * Kenneth Almquist.
13  *
14  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
15  *
16  * Original BSD copyright notice is retained at the end of this file.
17  */
18
19 /*
20  * The following should be set to reflect the type of system you have:
21  *      JOBS -> 1 if you have Berkeley job control, 0 otherwise.
22  *      define SYSV if you are running under System V.
23  *      define DEBUG=1 to compile in debugging ('set -o debug' to turn on)
24  *      define DEBUG=2 to compile in and turn on debugging.
25  *
26  * When debugging is on, debugging info will be written to ./trace and
27  * a quit signal will generate a core dump.
28  */
29 #define DEBUG 0
30 /* Tweak debug output verbosity here */
31 #define DEBUG_TIME 0
32 #define DEBUG_PID 1
33 #define DEBUG_SIG 1
34
35 #define PROFILE 0
36
37 #define IFS_BROKEN
38
39 #define JOBS ENABLE_ASH_JOB_CONTROL
40
41 #if DEBUG
42 # ifndef _GNU_SOURCE
43 #  define _GNU_SOURCE
44 # endif
45 #endif
46
47 #include "busybox.h" /* for applet_names */
48 //TODO: pull in some .h and find out do we have SINGLE_APPLET_MAIN?
49 //#include "applet_tables.h" doesn't work
50 #include <paths.h>
51 #include <setjmp.h>
52 #include <fnmatch.h>
53 #include "math.h"
54 #if ENABLE_ASH_RANDOM_SUPPORT
55 # include "random.h"
56 #else
57 # define CLEAR_RANDOM_T(rnd) ((void)0)
58 #endif
59
60 #if defined SINGLE_APPLET_MAIN
61 /* STANDALONE does not make sense, and won't compile */
62 # undef CONFIG_FEATURE_SH_STANDALONE
63 # undef ENABLE_FEATURE_SH_STANDALONE
64 # undef IF_FEATURE_SH_STANDALONE
65 # undef IF_NOT_FEATURE_SH_STANDALONE(...)
66 # define ENABLE_FEATURE_SH_STANDALONE 0
67 # define IF_FEATURE_SH_STANDALONE(...)
68 # define IF_NOT_FEATURE_SH_STANDALONE(...) __VA_ARGS__
69 #endif
70
71 #ifndef PIPE_BUF
72 # define PIPE_BUF 4096           /* amount of buffering in a pipe */
73 #endif
74
75 #if defined(__uClinux__)
76 # error "Do not even bother, ash will not run on NOMMU machine"
77 #endif
78
79
80 /* ============ Hash table sizes. Configurable. */
81
82 #define VTABSIZE 39
83 #define ATABSIZE 39
84 #define CMDTABLESIZE 31         /* should be prime */
85
86
87 /* ============ Shell options */
88
89 static const char *const optletters_optnames[] = {
90         "e"   "errexit",
91         "f"   "noglob",
92         "I"   "ignoreeof",
93         "i"   "interactive",
94         "m"   "monitor",
95         "n"   "noexec",
96         "s"   "stdin",
97         "x"   "xtrace",
98         "v"   "verbose",
99         "C"   "noclobber",
100         "a"   "allexport",
101         "b"   "notify",
102         "u"   "nounset",
103         "\0"  "vi"
104 #if ENABLE_ASH_BASH_COMPAT
105         ,"\0"  "pipefail"
106 #endif
107 #if DEBUG
108         ,"\0"  "nolog"
109         ,"\0"  "debug"
110 #endif
111 };
112
113 #define optletters(n)  optletters_optnames[n][0]
114 #define optnames(n)   (optletters_optnames[n] + 1)
115
116 enum { NOPTS = ARRAY_SIZE(optletters_optnames) };
117
118
119 /* ============ Misc data */
120
121 static const char homestr[] ALIGN1 = "HOME";
122 static const char snlfmt[] ALIGN1 = "%s\n";
123 static const char msg_illnum[] ALIGN1 = "Illegal number: %s";
124
125 /*
126  * We enclose jmp_buf in a structure so that we can declare pointers to
127  * jump locations.  The global variable handler contains the location to
128  * jump to when an exception occurs, and the global variable exception_type
129  * contains a code identifying the exception.  To implement nested
130  * exception handlers, the user should save the value of handler on entry
131  * to an inner scope, set handler to point to a jmploc structure for the
132  * inner scope, and restore handler on exit from the scope.
133  */
134 struct jmploc {
135         jmp_buf loc;
136 };
137
138 struct globals_misc {
139         /* pid of main shell */
140         int rootpid;
141         /* shell level: 0 for the main shell, 1 for its children, and so on */
142         int shlvl;
143 #define rootshell (!shlvl)
144         char *minusc;  /* argument to -c option */
145
146         char *curdir; // = nullstr;     /* current working directory */
147         char *physdir; // = nullstr;    /* physical working directory */
148
149         char *arg0; /* value of $0 */
150
151         struct jmploc *exception_handler;
152
153         volatile int suppress_int; /* counter */
154         volatile /*sig_atomic_t*/ smallint pending_int; /* 1 = got SIGINT */
155         /* last pending signal */
156         volatile /*sig_atomic_t*/ smallint pending_sig;
157         smallint exception_type; /* kind of exception (0..5) */
158         /* exceptions */
159 #define EXINT 0         /* SIGINT received */
160 #define EXERROR 1       /* a generic error */
161 #define EXSHELLPROC 2   /* execute a shell procedure */
162 #define EXEXEC 3        /* command execution failed */
163 #define EXEXIT 4        /* exit the shell */
164 #define EXSIG 5         /* trapped signal in wait(1) */
165
166         smallint isloginsh;
167         char nullstr[1];        /* zero length string */
168
169         char optlist[NOPTS];
170 #define eflag optlist[0]
171 #define fflag optlist[1]
172 #define Iflag optlist[2]
173 #define iflag optlist[3]
174 #define mflag optlist[4]
175 #define nflag optlist[5]
176 #define sflag optlist[6]
177 #define xflag optlist[7]
178 #define vflag optlist[8]
179 #define Cflag optlist[9]
180 #define aflag optlist[10]
181 #define bflag optlist[11]
182 #define uflag optlist[12]
183 #define viflag optlist[13]
184 #if ENABLE_ASH_BASH_COMPAT
185 # define pipefail optlist[14]
186 #else
187 # define pipefail 0
188 #endif
189 #if DEBUG
190 # define nolog optlist[14 + ENABLE_ASH_BASH_COMPAT]
191 # define debug optlist[15 + ENABLE_ASH_BASH_COMPAT]
192 #endif
193
194         /* trap handler commands */
195         /*
196          * Sigmode records the current value of the signal handlers for the various
197          * modes.  A value of zero means that the current handler is not known.
198          * S_HARD_IGN indicates that the signal was ignored on entry to the shell.
199          */
200         char sigmode[NSIG - 1];
201 #define S_DFL      1            /* default signal handling (SIG_DFL) */
202 #define S_CATCH    2            /* signal is caught */
203 #define S_IGN      3            /* signal is ignored (SIG_IGN) */
204 #define S_HARD_IGN 4            /* signal is ignored permenantly */
205
206         /* indicates specified signal received */
207         uint8_t gotsig[NSIG - 1]; /* offset by 1: "signal" 0 is meaningless */
208         char *trap[NSIG];
209         char **trap_ptr;        /* used only by "trap hack" */
210
211         /* Rarely referenced stuff */
212 #if ENABLE_ASH_RANDOM_SUPPORT
213         random_t random_gen;
214 #endif
215         pid_t backgndpid;        /* pid of last background process */
216         smallint job_warning;    /* user was warned about stopped jobs (can be 2, 1 or 0). */
217 };
218 extern struct globals_misc *const ash_ptr_to_globals_misc;
219 #define G_misc (*ash_ptr_to_globals_misc)
220 #define rootpid     (G_misc.rootpid    )
221 #define shlvl       (G_misc.shlvl      )
222 #define minusc      (G_misc.minusc     )
223 #define curdir      (G_misc.curdir     )
224 #define physdir     (G_misc.physdir    )
225 #define arg0        (G_misc.arg0       )
226 #define exception_handler (G_misc.exception_handler)
227 #define exception_type    (G_misc.exception_type   )
228 #define suppress_int      (G_misc.suppress_int     )
229 #define pending_int       (G_misc.pending_int      )
230 #define pending_sig       (G_misc.pending_sig      )
231 #define isloginsh   (G_misc.isloginsh  )
232 #define nullstr     (G_misc.nullstr    )
233 #define optlist     (G_misc.optlist    )
234 #define sigmode     (G_misc.sigmode    )
235 #define gotsig      (G_misc.gotsig     )
236 #define trap        (G_misc.trap       )
237 #define trap_ptr    (G_misc.trap_ptr   )
238 #define random_gen  (G_misc.random_gen )
239 #define backgndpid  (G_misc.backgndpid )
240 #define job_warning (G_misc.job_warning)
241 #define INIT_G_misc() do { \
242         (*(struct globals_misc**)&ash_ptr_to_globals_misc) = xzalloc(sizeof(G_misc)); \
243         barrier(); \
244         curdir = nullstr; \
245         physdir = nullstr; \
246         trap_ptr = trap; \
247 } while (0)
248
249
250 /* ============ DEBUG */
251 #if DEBUG
252 static void trace_printf(const char *fmt, ...);
253 static void trace_vprintf(const char *fmt, va_list va);
254 # define TRACE(param)    trace_printf param
255 # define TRACEV(param)   trace_vprintf param
256 # define close(fd) do { \
257         int dfd = (fd); \
258         if (close(dfd) < 0) \
259                 bb_error_msg("bug on %d: closing %d(0x%x)", \
260                         __LINE__, dfd, dfd); \
261 } while (0)
262 #else
263 # define TRACE(param)
264 # define TRACEV(param)
265 #endif
266
267
268 /* ============ Utility functions */
269 #define xbarrier() do { __asm__ __volatile__ ("": : :"memory"); } while (0)
270
271 static int isdigit_str9(const char *str)
272 {
273         int maxlen = 9 + 1; /* max 9 digits: 999999999 */
274         while (--maxlen && isdigit(*str))
275                 str++;
276         return (*str == '\0');
277 }
278
279
280 /* ============ Interrupts / exceptions */
281 /*
282  * These macros allow the user to suspend the handling of interrupt signals
283  * over a period of time.  This is similar to SIGHOLD or to sigblock, but
284  * much more efficient and portable.  (But hacking the kernel is so much
285  * more fun than worrying about efficiency and portability. :-))
286  */
287 #define INT_OFF do { \
288         suppress_int++; \
289         xbarrier(); \
290 } while (0)
291
292 /*
293  * Called to raise an exception.  Since C doesn't include exceptions, we
294  * just do a longjmp to the exception handler.  The type of exception is
295  * stored in the global variable "exception_type".
296  */
297 static void raise_exception(int) NORETURN;
298 static void
299 raise_exception(int e)
300 {
301 #if DEBUG
302         if (exception_handler == NULL)
303                 abort();
304 #endif
305         INT_OFF;
306         exception_type = e;
307         longjmp(exception_handler->loc, 1);
308 }
309 #if DEBUG
310 #define raise_exception(e) do { \
311         TRACE(("raising exception %d on line %d\n", (e), __LINE__)); \
312         raise_exception(e); \
313 } while (0)
314 #endif
315
316 /*
317  * Called from trap.c when a SIGINT is received.  (If the user specifies
318  * that SIGINT is to be trapped or ignored using the trap builtin, then
319  * this routine is not called.)  Suppressint is nonzero when interrupts
320  * are held using the INT_OFF macro.  (The test for iflag is just
321  * defensive programming.)
322  */
323 static void raise_interrupt(void) NORETURN;
324 static void
325 raise_interrupt(void)
326 {
327         int ex_type;
328
329         pending_int = 0;
330         /* Signal is not automatically unmasked after it is raised,
331          * do it ourself - unmask all signals */
332         sigprocmask_allsigs(SIG_UNBLOCK);
333         /* pending_sig = 0; - now done in onsig() */
334
335         ex_type = EXSIG;
336         if (gotsig[SIGINT - 1] && !trap[SIGINT]) {
337                 if (!(rootshell && iflag)) {
338                         /* Kill ourself with SIGINT */
339                         signal(SIGINT, SIG_DFL);
340                         raise(SIGINT);
341                 }
342                 ex_type = EXINT;
343         }
344         raise_exception(ex_type);
345         /* NOTREACHED */
346 }
347 #if DEBUG
348 #define raise_interrupt() do { \
349         TRACE(("raising interrupt on line %d\n", __LINE__)); \
350         raise_interrupt(); \
351 } while (0)
352 #endif
353
354 static IF_ASH_OPTIMIZE_FOR_SIZE(inline) void
355 int_on(void)
356 {
357         xbarrier();
358         if (--suppress_int == 0 && pending_int) {
359                 raise_interrupt();
360         }
361 }
362 #define INT_ON int_on()
363 static IF_ASH_OPTIMIZE_FOR_SIZE(inline) void
364 force_int_on(void)
365 {
366         xbarrier();
367         suppress_int = 0;
368         if (pending_int)
369                 raise_interrupt();
370 }
371 #define FORCE_INT_ON force_int_on()
372
373 #define SAVE_INT(v) ((v) = suppress_int)
374
375 #define RESTORE_INT(v) do { \
376         xbarrier(); \
377         suppress_int = (v); \
378         if (suppress_int == 0 && pending_int) \
379                 raise_interrupt(); \
380 } while (0)
381
382
383 /* ============ Stdout/stderr output */
384
385 static void
386 outstr(const char *p, FILE *file)
387 {
388         INT_OFF;
389         fputs(p, file);
390         INT_ON;
391 }
392
393 static void
394 flush_stdout_stderr(void)
395 {
396         INT_OFF;
397         fflush_all();
398         INT_ON;
399 }
400
401 static void
402 outcslow(int c, FILE *dest)
403 {
404         INT_OFF;
405         putc(c, dest);
406         fflush(dest);
407         INT_ON;
408 }
409
410 static int out1fmt(const char *, ...) __attribute__((__format__(__printf__,1,2)));
411 static int
412 out1fmt(const char *fmt, ...)
413 {
414         va_list ap;
415         int r;
416
417         INT_OFF;
418         va_start(ap, fmt);
419         r = vprintf(fmt, ap);
420         va_end(ap);
421         INT_ON;
422         return r;
423 }
424
425 static int fmtstr(char *, size_t, const char *, ...) __attribute__((__format__(__printf__,3,4)));
426 static int
427 fmtstr(char *outbuf, size_t length, const char *fmt, ...)
428 {
429         va_list ap;
430         int ret;
431
432         va_start(ap, fmt);
433         INT_OFF;
434         ret = vsnprintf(outbuf, length, fmt, ap);
435         va_end(ap);
436         INT_ON;
437         return ret;
438 }
439
440 static void
441 out1str(const char *p)
442 {
443         outstr(p, stdout);
444 }
445
446 static void
447 out2str(const char *p)
448 {
449         outstr(p, stderr);
450         flush_stdout_stderr();
451 }
452
453
454 /* ============ Parser structures */
455
456 /* control characters in argument strings */
457 #define CTL_FIRST CTLESC
458 #define CTLESC       ((unsigned char)'\201')    /* escape next character */
459 #define CTLVAR       ((unsigned char)'\202')    /* variable defn */
460 #define CTLENDVAR    ((unsigned char)'\203')
461 #define CTLBACKQ     ((unsigned char)'\204')
462 #define CTLQUOTE 01             /* ored with CTLBACKQ code if in quotes */
463 /*      CTLBACKQ | CTLQUOTE == '\205' */
464 #define CTLARI       ((unsigned char)'\206')    /* arithmetic expression */
465 #define CTLENDARI    ((unsigned char)'\207')
466 #define CTLQUOTEMARK ((unsigned char)'\210')
467 #define CTL_LAST CTLQUOTEMARK
468
469 /* variable substitution byte (follows CTLVAR) */
470 #define VSTYPE  0x0f            /* type of variable substitution */
471 #define VSNUL   0x10            /* colon--treat the empty string as unset */
472 #define VSQUOTE 0x80            /* inside double quotes--suppress splitting */
473
474 /* values of VSTYPE field */
475 #define VSNORMAL        0x1     /* normal variable:  $var or ${var} */
476 #define VSMINUS         0x2     /* ${var-text} */
477 #define VSPLUS          0x3     /* ${var+text} */
478 #define VSQUESTION      0x4     /* ${var?message} */
479 #define VSASSIGN        0x5     /* ${var=text} */
480 #define VSTRIMRIGHT     0x6     /* ${var%pattern} */
481 #define VSTRIMRIGHTMAX  0x7     /* ${var%%pattern} */
482 #define VSTRIMLEFT      0x8     /* ${var#pattern} */
483 #define VSTRIMLEFTMAX   0x9     /* ${var##pattern} */
484 #define VSLENGTH        0xa     /* ${#var} */
485 #if ENABLE_ASH_BASH_COMPAT
486 #define VSSUBSTR        0xc     /* ${var:position:length} */
487 #define VSREPLACE       0xd     /* ${var/pattern/replacement} */
488 #define VSREPLACEALL    0xe     /* ${var//pattern/replacement} */
489 #endif
490
491 static const char dolatstr[] ALIGN1 = {
492         CTLVAR, VSNORMAL|VSQUOTE, '@', '=', '\0'
493 };
494
495 #define NCMD      0
496 #define NPIPE     1
497 #define NREDIR    2
498 #define NBACKGND  3
499 #define NSUBSHELL 4
500 #define NAND      5
501 #define NOR       6
502 #define NSEMI     7
503 #define NIF       8
504 #define NWHILE    9
505 #define NUNTIL   10
506 #define NFOR     11
507 #define NCASE    12
508 #define NCLIST   13
509 #define NDEFUN   14
510 #define NARG     15
511 #define NTO      16
512 #if ENABLE_ASH_BASH_COMPAT
513 #define NTO2     17
514 #endif
515 #define NCLOBBER 18
516 #define NFROM    19
517 #define NFROMTO  20
518 #define NAPPEND  21
519 #define NTOFD    22
520 #define NFROMFD  23
521 #define NHERE    24
522 #define NXHERE   25
523 #define NNOT     26
524 #define N_NUMBER 27
525
526 union node;
527
528 struct ncmd {
529         smallint type; /* Nxxxx */
530         union node *assign;
531         union node *args;
532         union node *redirect;
533 };
534
535 struct npipe {
536         smallint type;
537         smallint pipe_backgnd;
538         struct nodelist *cmdlist;
539 };
540
541 struct nredir {
542         smallint type;
543         union node *n;
544         union node *redirect;
545 };
546
547 struct nbinary {
548         smallint type;
549         union node *ch1;
550         union node *ch2;
551 };
552
553 struct nif {
554         smallint type;
555         union node *test;
556         union node *ifpart;
557         union node *elsepart;
558 };
559
560 struct nfor {
561         smallint type;
562         union node *args;
563         union node *body;
564         char *var;
565 };
566
567 struct ncase {
568         smallint type;
569         union node *expr;
570         union node *cases;
571 };
572
573 struct nclist {
574         smallint type;
575         union node *next;
576         union node *pattern;
577         union node *body;
578 };
579
580 struct narg {
581         smallint type;
582         union node *next;
583         char *text;
584         struct nodelist *backquote;
585 };
586
587 /* nfile and ndup layout must match!
588  * NTOFD (>&fdnum) uses ndup structure, but we may discover mid-flight
589  * that it is actually NTO2 (>&file), and change its type.
590  */
591 struct nfile {
592         smallint type;
593         union node *next;
594         int fd;
595         int _unused_dupfd;
596         union node *fname;
597         char *expfname;
598 };
599
600 struct ndup {
601         smallint type;
602         union node *next;
603         int fd;
604         int dupfd;
605         union node *vname;
606         char *_unused_expfname;
607 };
608
609 struct nhere {
610         smallint type;
611         union node *next;
612         int fd;
613         union node *doc;
614 };
615
616 struct nnot {
617         smallint type;
618         union node *com;
619 };
620
621 union node {
622         smallint type;
623         struct ncmd ncmd;
624         struct npipe npipe;
625         struct nredir nredir;
626         struct nbinary nbinary;
627         struct nif nif;
628         struct nfor nfor;
629         struct ncase ncase;
630         struct nclist nclist;
631         struct narg narg;
632         struct nfile nfile;
633         struct ndup ndup;
634         struct nhere nhere;
635         struct nnot nnot;
636 };
637
638 /*
639  * NODE_EOF is returned by parsecmd when it encounters an end of file.
640  * It must be distinct from NULL.
641  */
642 #define NODE_EOF ((union node *) -1L)
643
644 struct nodelist {
645         struct nodelist *next;
646         union node *n;
647 };
648
649 struct funcnode {
650         int count;
651         union node n;
652 };
653
654 /*
655  * Free a parse tree.
656  */
657 static void
658 freefunc(struct funcnode *f)
659 {
660         if (f && --f->count < 0)
661                 free(f);
662 }
663
664
665 /* ============ Debugging output */
666
667 #if DEBUG
668
669 static FILE *tracefile;
670
671 static void
672 trace_printf(const char *fmt, ...)
673 {
674         va_list va;
675
676         if (debug != 1)
677                 return;
678         if (DEBUG_TIME)
679                 fprintf(tracefile, "%u ", (int) time(NULL));
680         if (DEBUG_PID)
681                 fprintf(tracefile, "[%u] ", (int) getpid());
682         if (DEBUG_SIG)
683                 fprintf(tracefile, "pending s:%d i:%d(supp:%d) ", pending_sig, pending_int, suppress_int);
684         va_start(va, fmt);
685         vfprintf(tracefile, fmt, va);
686         va_end(va);
687 }
688
689 static void
690 trace_vprintf(const char *fmt, va_list va)
691 {
692         if (debug != 1)
693                 return;
694         if (DEBUG_TIME)
695                 fprintf(tracefile, "%u ", (int) time(NULL));
696         if (DEBUG_PID)
697                 fprintf(tracefile, "[%u] ", (int) getpid());
698         if (DEBUG_SIG)
699                 fprintf(tracefile, "pending s:%d i:%d(supp:%d) ", pending_sig, pending_int, suppress_int);
700         vfprintf(tracefile, fmt, va);
701 }
702
703 static void
704 trace_puts(const char *s)
705 {
706         if (debug != 1)
707                 return;
708         fputs(s, tracefile);
709 }
710
711 static void
712 trace_puts_quoted(char *s)
713 {
714         char *p;
715         char c;
716
717         if (debug != 1)
718                 return;
719         putc('"', tracefile);
720         for (p = s; *p; p++) {
721                 switch ((unsigned char)*p) {
722                 case '\n': c = 'n'; goto backslash;
723                 case '\t': c = 't'; goto backslash;
724                 case '\r': c = 'r'; goto backslash;
725                 case '\"': c = '\"'; goto backslash;
726                 case '\\': c = '\\'; goto backslash;
727                 case CTLESC: c = 'e'; goto backslash;
728                 case CTLVAR: c = 'v'; goto backslash;
729                 case CTLVAR+CTLQUOTE: c = 'V'; goto backslash;
730                 case CTLBACKQ: c = 'q'; goto backslash;
731                 case CTLBACKQ+CTLQUOTE: c = 'Q'; goto backslash;
732  backslash:
733                         putc('\\', tracefile);
734                         putc(c, tracefile);
735                         break;
736                 default:
737                         if (*p >= ' ' && *p <= '~')
738                                 putc(*p, tracefile);
739                         else {
740                                 putc('\\', tracefile);
741                                 putc((*p >> 6) & 03, tracefile);
742                                 putc((*p >> 3) & 07, tracefile);
743                                 putc(*p & 07, tracefile);
744                         }
745                         break;
746                 }
747         }
748         putc('"', tracefile);
749 }
750
751 static void
752 trace_puts_args(char **ap)
753 {
754         if (debug != 1)
755                 return;
756         if (!*ap)
757                 return;
758         while (1) {
759                 trace_puts_quoted(*ap);
760                 if (!*++ap) {
761                         putc('\n', tracefile);
762                         break;
763                 }
764                 putc(' ', tracefile);
765         }
766 }
767
768 static void
769 opentrace(void)
770 {
771         char s[100];
772 #ifdef O_APPEND
773         int flags;
774 #endif
775
776         if (debug != 1) {
777                 if (tracefile)
778                         fflush(tracefile);
779                 /* leave open because libedit might be using it */
780                 return;
781         }
782         strcpy(s, "./trace");
783         if (tracefile) {
784                 if (!freopen(s, "a", tracefile)) {
785                         fprintf(stderr, "Can't re-open %s\n", s);
786                         debug = 0;
787                         return;
788                 }
789         } else {
790                 tracefile = fopen(s, "a");
791                 if (tracefile == NULL) {
792                         fprintf(stderr, "Can't open %s\n", s);
793                         debug = 0;
794                         return;
795                 }
796         }
797 #ifdef O_APPEND
798         flags = fcntl(fileno(tracefile), F_GETFL);
799         if (flags >= 0)
800                 fcntl(fileno(tracefile), F_SETFL, flags | O_APPEND);
801 #endif
802         setlinebuf(tracefile);
803         fputs("\nTracing started.\n", tracefile);
804 }
805
806 static void
807 indent(int amount, char *pfx, FILE *fp)
808 {
809         int i;
810
811         for (i = 0; i < amount; i++) {
812                 if (pfx && i == amount - 1)
813                         fputs(pfx, fp);
814                 putc('\t', fp);
815         }
816 }
817
818 /* little circular references here... */
819 static void shtree(union node *n, int ind, char *pfx, FILE *fp);
820
821 static void
822 sharg(union node *arg, FILE *fp)
823 {
824         char *p;
825         struct nodelist *bqlist;
826         unsigned char subtype;
827
828         if (arg->type != NARG) {
829                 out1fmt("<node type %d>\n", arg->type);
830                 abort();
831         }
832         bqlist = arg->narg.backquote;
833         for (p = arg->narg.text; *p; p++) {
834                 switch ((unsigned char)*p) {
835                 case CTLESC:
836                         putc(*++p, fp);
837                         break;
838                 case CTLVAR:
839                         putc('$', fp);
840                         putc('{', fp);
841                         subtype = *++p;
842                         if (subtype == VSLENGTH)
843                                 putc('#', fp);
844
845                         while (*p != '=')
846                                 putc(*p++, fp);
847
848                         if (subtype & VSNUL)
849                                 putc(':', fp);
850
851                         switch (subtype & VSTYPE) {
852                         case VSNORMAL:
853                                 putc('}', fp);
854                                 break;
855                         case VSMINUS:
856                                 putc('-', fp);
857                                 break;
858                         case VSPLUS:
859                                 putc('+', fp);
860                                 break;
861                         case VSQUESTION:
862                                 putc('?', fp);
863                                 break;
864                         case VSASSIGN:
865                                 putc('=', fp);
866                                 break;
867                         case VSTRIMLEFT:
868                                 putc('#', fp);
869                                 break;
870                         case VSTRIMLEFTMAX:
871                                 putc('#', fp);
872                                 putc('#', fp);
873                                 break;
874                         case VSTRIMRIGHT:
875                                 putc('%', fp);
876                                 break;
877                         case VSTRIMRIGHTMAX:
878                                 putc('%', fp);
879                                 putc('%', fp);
880                                 break;
881                         case VSLENGTH:
882                                 break;
883                         default:
884                                 out1fmt("<subtype %d>", subtype);
885                         }
886                         break;
887                 case CTLENDVAR:
888                         putc('}', fp);
889                         break;
890                 case CTLBACKQ:
891                 case CTLBACKQ|CTLQUOTE:
892                         putc('$', fp);
893                         putc('(', fp);
894                         shtree(bqlist->n, -1, NULL, fp);
895                         putc(')', fp);
896                         break;
897                 default:
898                         putc(*p, fp);
899                         break;
900                 }
901         }
902 }
903
904 static void
905 shcmd(union node *cmd, FILE *fp)
906 {
907         union node *np;
908         int first;
909         const char *s;
910         int dftfd;
911
912         first = 1;
913         for (np = cmd->ncmd.args; np; np = np->narg.next) {
914                 if (!first)
915                         putc(' ', fp);
916                 sharg(np, fp);
917                 first = 0;
918         }
919         for (np = cmd->ncmd.redirect; np; np = np->nfile.next) {
920                 if (!first)
921                         putc(' ', fp);
922                 dftfd = 0;
923                 switch (np->nfile.type) {
924                 case NTO:      s = ">>"+1; dftfd = 1; break;
925                 case NCLOBBER: s = ">|"; dftfd = 1; break;
926                 case NAPPEND:  s = ">>"; dftfd = 1; break;
927 #if ENABLE_ASH_BASH_COMPAT
928                 case NTO2:
929 #endif
930                 case NTOFD:    s = ">&"; dftfd = 1; break;
931                 case NFROM:    s = "<"; break;
932                 case NFROMFD:  s = "<&"; break;
933                 case NFROMTO:  s = "<>"; break;
934                 default:       s = "*error*"; break;
935                 }
936                 if (np->nfile.fd != dftfd)
937                         fprintf(fp, "%d", np->nfile.fd);
938                 fputs(s, fp);
939                 if (np->nfile.type == NTOFD || np->nfile.type == NFROMFD) {
940                         fprintf(fp, "%d", np->ndup.dupfd);
941                 } else {
942                         sharg(np->nfile.fname, fp);
943                 }
944                 first = 0;
945         }
946 }
947
948 static void
949 shtree(union node *n, int ind, char *pfx, FILE *fp)
950 {
951         struct nodelist *lp;
952         const char *s;
953
954         if (n == NULL)
955                 return;
956
957         indent(ind, pfx, fp);
958
959         if (n == NODE_EOF) {
960                 fputs("<EOF>", fp);
961                 return;
962         }
963
964         switch (n->type) {
965         case NSEMI:
966                 s = "; ";
967                 goto binop;
968         case NAND:
969                 s = " && ";
970                 goto binop;
971         case NOR:
972                 s = " || ";
973  binop:
974                 shtree(n->nbinary.ch1, ind, NULL, fp);
975                 /* if (ind < 0) */
976                         fputs(s, fp);
977                 shtree(n->nbinary.ch2, ind, NULL, fp);
978                 break;
979         case NCMD:
980                 shcmd(n, fp);
981                 if (ind >= 0)
982                         putc('\n', fp);
983                 break;
984         case NPIPE:
985                 for (lp = n->npipe.cmdlist; lp; lp = lp->next) {
986                         shtree(lp->n, 0, NULL, fp);
987                         if (lp->next)
988                                 fputs(" | ", fp);
989                 }
990                 if (n->npipe.pipe_backgnd)
991                         fputs(" &", fp);
992                 if (ind >= 0)
993                         putc('\n', fp);
994                 break;
995         default:
996                 fprintf(fp, "<node type %d>", n->type);
997                 if (ind >= 0)
998                         putc('\n', fp);
999                 break;
1000         }
1001 }
1002
1003 static void
1004 showtree(union node *n)
1005 {
1006         trace_puts("showtree called\n");
1007         shtree(n, 1, NULL, stderr);
1008 }
1009
1010 #endif /* DEBUG */
1011
1012
1013 /* ============ Parser data */
1014
1015 /*
1016  * ash_vmsg() needs parsefile->fd, hence parsefile definition is moved up.
1017  */
1018 struct strlist {
1019         struct strlist *next;
1020         char *text;
1021 };
1022
1023 struct alias;
1024
1025 struct strpush {
1026         struct strpush *prev;   /* preceding string on stack */
1027         char *prev_string;
1028         int prev_left_in_line;
1029 #if ENABLE_ASH_ALIAS
1030         struct alias *ap;       /* if push was associated with an alias */
1031 #endif
1032         char *string;           /* remember the string since it may change */
1033 };
1034
1035 struct parsefile {
1036         struct parsefile *prev; /* preceding file on stack */
1037         int linno;              /* current line */
1038         int fd;                 /* file descriptor (or -1 if string) */
1039         int left_in_line;       /* number of chars left in this line */
1040         int left_in_buffer;     /* number of chars left in this buffer past the line */
1041         char *next_to_pgetc;    /* next char in buffer */
1042         char *buf;              /* input buffer */
1043         struct strpush *strpush; /* for pushing strings at this level */
1044         struct strpush basestrpush; /* so pushing one is fast */
1045 };
1046
1047 static struct parsefile basepf;        /* top level input file */
1048 static struct parsefile *g_parsefile = &basepf;  /* current input file */
1049 static int startlinno;                 /* line # where last token started */
1050 static char *commandname;              /* currently executing command */
1051 static struct strlist *cmdenviron;     /* environment for builtin command */
1052 static uint8_t exitstatus;             /* exit status of last command */
1053
1054
1055 /* ============ Message printing */
1056
1057 static void
1058 ash_vmsg(const char *msg, va_list ap)
1059 {
1060         fprintf(stderr, "%s: ", arg0);
1061         if (commandname) {
1062                 if (strcmp(arg0, commandname))
1063                         fprintf(stderr, "%s: ", commandname);
1064                 if (!iflag || g_parsefile->fd)
1065                         fprintf(stderr, "line %d: ", startlinno);
1066         }
1067         vfprintf(stderr, msg, ap);
1068         outcslow('\n', stderr);
1069 }
1070
1071 /*
1072  * Exverror is called to raise the error exception.  If the second argument
1073  * is not NULL then error prints an error message using printf style
1074  * formatting.  It then raises the error exception.
1075  */
1076 static void ash_vmsg_and_raise(int, const char *, va_list) NORETURN;
1077 static void
1078 ash_vmsg_and_raise(int cond, const char *msg, va_list ap)
1079 {
1080 #if DEBUG
1081         if (msg) {
1082                 TRACE(("ash_vmsg_and_raise(%d, \"", cond));
1083                 TRACEV((msg, ap));
1084                 TRACE(("\") pid=%d\n", getpid()));
1085         } else
1086                 TRACE(("ash_vmsg_and_raise(%d, NULL) pid=%d\n", cond, getpid()));
1087         if (msg)
1088 #endif
1089                 ash_vmsg(msg, ap);
1090
1091         flush_stdout_stderr();
1092         raise_exception(cond);
1093         /* NOTREACHED */
1094 }
1095
1096 static void ash_msg_and_raise_error(const char *, ...) NORETURN;
1097 static void
1098 ash_msg_and_raise_error(const char *msg, ...)
1099 {
1100         va_list ap;
1101
1102         va_start(ap, msg);
1103         ash_vmsg_and_raise(EXERROR, msg, ap);
1104         /* NOTREACHED */
1105         va_end(ap);
1106 }
1107
1108 static void raise_error_syntax(const char *) NORETURN;
1109 static void
1110 raise_error_syntax(const char *msg)
1111 {
1112         ash_msg_and_raise_error("syntax error: %s", msg);
1113         /* NOTREACHED */
1114 }
1115
1116 static void ash_msg_and_raise(int, const char *, ...) NORETURN;
1117 static void
1118 ash_msg_and_raise(int cond, const char *msg, ...)
1119 {
1120         va_list ap;
1121
1122         va_start(ap, msg);
1123         ash_vmsg_and_raise(cond, msg, ap);
1124         /* NOTREACHED */
1125         va_end(ap);
1126 }
1127
1128 /*
1129  * error/warning routines for external builtins
1130  */
1131 static void
1132 ash_msg(const char *fmt, ...)
1133 {
1134         va_list ap;
1135
1136         va_start(ap, fmt);
1137         ash_vmsg(fmt, ap);
1138         va_end(ap);
1139 }
1140
1141 /*
1142  * Return a string describing an error.  The returned string may be a
1143  * pointer to a static buffer that will be overwritten on the next call.
1144  * Action describes the operation that got the error.
1145  */
1146 static const char *
1147 errmsg(int e, const char *em)
1148 {
1149         if (e == ENOENT || e == ENOTDIR) {
1150                 return em;
1151         }
1152         return strerror(e);
1153 }
1154
1155
1156 /* ============ Memory allocation */
1157
1158 #if 0
1159 /* I consider these wrappers nearly useless:
1160  * ok, they return you to nearest exception handler, but
1161  * how much memory do you leak in the process, making
1162  * memory starvation worse?
1163  */
1164 static void *
1165 ckrealloc(void * p, size_t nbytes)
1166 {
1167         p = realloc(p, nbytes);
1168         if (!p)
1169                 ash_msg_and_raise_error(bb_msg_memory_exhausted);
1170         return p;
1171 }
1172
1173 static void *
1174 ckmalloc(size_t nbytes)
1175 {
1176         return ckrealloc(NULL, nbytes);
1177 }
1178
1179 static void *
1180 ckzalloc(size_t nbytes)
1181 {
1182         return memset(ckmalloc(nbytes), 0, nbytes);
1183 }
1184
1185 static char *
1186 ckstrdup(const char *s)
1187 {
1188         char *p = strdup(s);
1189         if (!p)
1190                 ash_msg_and_raise_error(bb_msg_memory_exhausted);
1191         return p;
1192 }
1193 #else
1194 /* Using bbox equivalents. They exit if out of memory */
1195 # define ckrealloc xrealloc
1196 # define ckmalloc  xmalloc
1197 # define ckzalloc  xzalloc
1198 # define ckstrdup  xstrdup
1199 #endif
1200
1201 /*
1202  * It appears that grabstackstr() will barf with such alignments
1203  * because stalloc() will return a string allocated in a new stackblock.
1204  */
1205 #define SHELL_ALIGN(nbytes) (((nbytes) + SHELL_SIZE) & ~SHELL_SIZE)
1206 enum {
1207         /* Most machines require the value returned from malloc to be aligned
1208          * in some way.  The following macro will get this right
1209          * on many machines.  */
1210         SHELL_SIZE = sizeof(union { int i; char *cp; double d; }) - 1,
1211         /* Minimum size of a block */
1212         MINSIZE = SHELL_ALIGN(504),
1213 };
1214
1215 struct stack_block {
1216         struct stack_block *prev;
1217         char space[MINSIZE];
1218 };
1219
1220 struct stackmark {
1221         struct stack_block *stackp;
1222         char *stacknxt;
1223         size_t stacknleft;
1224         struct stackmark *marknext;
1225 };
1226
1227
1228 struct globals_memstack {
1229         struct stack_block *g_stackp; // = &stackbase;
1230         struct stackmark *markp;
1231         char *g_stacknxt; // = stackbase.space;
1232         char *sstrend; // = stackbase.space + MINSIZE;
1233         size_t g_stacknleft; // = MINSIZE;
1234         int    herefd; // = -1;
1235         struct stack_block stackbase;
1236 };
1237 extern struct globals_memstack *const ash_ptr_to_globals_memstack;
1238 #define G_memstack (*ash_ptr_to_globals_memstack)
1239 #define g_stackp     (G_memstack.g_stackp    )
1240 #define markp        (G_memstack.markp       )
1241 #define g_stacknxt   (G_memstack.g_stacknxt  )
1242 #define sstrend      (G_memstack.sstrend     )
1243 #define g_stacknleft (G_memstack.g_stacknleft)
1244 #define herefd       (G_memstack.herefd      )
1245 #define stackbase    (G_memstack.stackbase   )
1246 #define INIT_G_memstack() do { \
1247         (*(struct globals_memstack**)&ash_ptr_to_globals_memstack) = xzalloc(sizeof(G_memstack)); \
1248         barrier(); \
1249         g_stackp = &stackbase; \
1250         g_stacknxt = stackbase.space; \
1251         g_stacknleft = MINSIZE; \
1252         sstrend = stackbase.space + MINSIZE; \
1253         herefd = -1; \
1254 } while (0)
1255
1256
1257 #define stackblock()     ((void *)g_stacknxt)
1258 #define stackblocksize() g_stacknleft
1259
1260 /*
1261  * Parse trees for commands are allocated in lifo order, so we use a stack
1262  * to make this more efficient, and also to avoid all sorts of exception
1263  * handling code to handle interrupts in the middle of a parse.
1264  *
1265  * The size 504 was chosen because the Ultrix malloc handles that size
1266  * well.
1267  */
1268 static void *
1269 stalloc(size_t nbytes)
1270 {
1271         char *p;
1272         size_t aligned;
1273
1274         aligned = SHELL_ALIGN(nbytes);
1275         if (aligned > g_stacknleft) {
1276                 size_t len;
1277                 size_t blocksize;
1278                 struct stack_block *sp;
1279
1280                 blocksize = aligned;
1281                 if (blocksize < MINSIZE)
1282                         blocksize = MINSIZE;
1283                 len = sizeof(struct stack_block) - MINSIZE + blocksize;
1284                 if (len < blocksize)
1285                         ash_msg_and_raise_error(bb_msg_memory_exhausted);
1286                 INT_OFF;
1287                 sp = ckmalloc(len);
1288                 sp->prev = g_stackp;
1289                 g_stacknxt = sp->space;
1290                 g_stacknleft = blocksize;
1291                 sstrend = g_stacknxt + blocksize;
1292                 g_stackp = sp;
1293                 INT_ON;
1294         }
1295         p = g_stacknxt;
1296         g_stacknxt += aligned;
1297         g_stacknleft -= aligned;
1298         return p;
1299 }
1300
1301 static void *
1302 stzalloc(size_t nbytes)
1303 {
1304         return memset(stalloc(nbytes), 0, nbytes);
1305 }
1306
1307 static void
1308 stunalloc(void *p)
1309 {
1310 #if DEBUG
1311         if (!p || (g_stacknxt < (char *)p) || ((char *)p < g_stackp->space)) {
1312                 write(STDERR_FILENO, "stunalloc\n", 10);
1313                 abort();
1314         }
1315 #endif
1316         g_stacknleft += g_stacknxt - (char *)p;
1317         g_stacknxt = p;
1318 }
1319
1320 /*
1321  * Like strdup but works with the ash stack.
1322  */
1323 static char *
1324 ststrdup(const char *p)
1325 {
1326         size_t len = strlen(p) + 1;
1327         return memcpy(stalloc(len), p, len);
1328 }
1329
1330 static void
1331 setstackmark(struct stackmark *mark)
1332 {
1333         mark->stackp = g_stackp;
1334         mark->stacknxt = g_stacknxt;
1335         mark->stacknleft = g_stacknleft;
1336         mark->marknext = markp;
1337         markp = mark;
1338 }
1339
1340 static void
1341 popstackmark(struct stackmark *mark)
1342 {
1343         struct stack_block *sp;
1344
1345         if (!mark->stackp)
1346                 return;
1347
1348         INT_OFF;
1349         markp = mark->marknext;
1350         while (g_stackp != mark->stackp) {
1351                 sp = g_stackp;
1352                 g_stackp = sp->prev;
1353                 free(sp);
1354         }
1355         g_stacknxt = mark->stacknxt;
1356         g_stacknleft = mark->stacknleft;
1357         sstrend = mark->stacknxt + mark->stacknleft;
1358         INT_ON;
1359 }
1360
1361 /*
1362  * When the parser reads in a string, it wants to stick the string on the
1363  * stack and only adjust the stack pointer when it knows how big the
1364  * string is.  Stackblock (defined in stack.h) returns a pointer to a block
1365  * of space on top of the stack and stackblocklen returns the length of
1366  * this block.  Growstackblock will grow this space by at least one byte,
1367  * possibly moving it (like realloc).  Grabstackblock actually allocates the
1368  * part of the block that has been used.
1369  */
1370 static void
1371 growstackblock(void)
1372 {
1373         size_t newlen;
1374
1375         newlen = g_stacknleft * 2;
1376         if (newlen < g_stacknleft)
1377                 ash_msg_and_raise_error(bb_msg_memory_exhausted);
1378         if (newlen < 128)
1379                 newlen += 128;
1380
1381         if (g_stacknxt == g_stackp->space && g_stackp != &stackbase) {
1382                 struct stack_block *oldstackp;
1383                 struct stackmark *xmark;
1384                 struct stack_block *sp;
1385                 struct stack_block *prevstackp;
1386                 size_t grosslen;
1387
1388                 INT_OFF;
1389                 oldstackp = g_stackp;
1390                 sp = g_stackp;
1391                 prevstackp = sp->prev;
1392                 grosslen = newlen + sizeof(struct stack_block) - MINSIZE;
1393                 sp = ckrealloc(sp, grosslen);
1394                 sp->prev = prevstackp;
1395                 g_stackp = sp;
1396                 g_stacknxt = sp->space;
1397                 g_stacknleft = newlen;
1398                 sstrend = sp->space + newlen;
1399
1400                 /*
1401                  * Stack marks pointing to the start of the old block
1402                  * must be relocated to point to the new block
1403                  */
1404                 xmark = markp;
1405                 while (xmark != NULL && xmark->stackp == oldstackp) {
1406                         xmark->stackp = g_stackp;
1407                         xmark->stacknxt = g_stacknxt;
1408                         xmark->stacknleft = g_stacknleft;
1409                         xmark = xmark->marknext;
1410                 }
1411                 INT_ON;
1412         } else {
1413                 char *oldspace = g_stacknxt;
1414                 size_t oldlen = g_stacknleft;
1415                 char *p = stalloc(newlen);
1416
1417                 /* free the space we just allocated */
1418                 g_stacknxt = memcpy(p, oldspace, oldlen);
1419                 g_stacknleft += newlen;
1420         }
1421 }
1422
1423 static void
1424 grabstackblock(size_t len)
1425 {
1426         len = SHELL_ALIGN(len);
1427         g_stacknxt += len;
1428         g_stacknleft -= len;
1429 }
1430
1431 /*
1432  * The following routines are somewhat easier to use than the above.
1433  * The user declares a variable of type STACKSTR, which may be declared
1434  * to be a register.  The macro STARTSTACKSTR initializes things.  Then
1435  * the user uses the macro STPUTC to add characters to the string.  In
1436  * effect, STPUTC(c, p) is the same as *p++ = c except that the stack is
1437  * grown as necessary.  When the user is done, she can just leave the
1438  * string there and refer to it using stackblock().  Or she can allocate
1439  * the space for it using grabstackstr().  If it is necessary to allow
1440  * someone else to use the stack temporarily and then continue to grow
1441  * the string, the user should use grabstack to allocate the space, and
1442  * then call ungrabstr(p) to return to the previous mode of operation.
1443  *
1444  * USTPUTC is like STPUTC except that it doesn't check for overflow.
1445  * CHECKSTACKSPACE can be called before USTPUTC to ensure that there
1446  * is space for at least one character.
1447  */
1448 static void *
1449 growstackstr(void)
1450 {
1451         size_t len = stackblocksize();
1452         if (herefd >= 0 && len >= 1024) {
1453                 full_write(herefd, stackblock(), len);
1454                 return stackblock();
1455         }
1456         growstackblock();
1457         return (char *)stackblock() + len;
1458 }
1459
1460 /*
1461  * Called from CHECKSTRSPACE.
1462  */
1463 static char *
1464 makestrspace(size_t newlen, char *p)
1465 {
1466         size_t len = p - g_stacknxt;
1467         size_t size = stackblocksize();
1468
1469         for (;;) {
1470                 size_t nleft;
1471
1472                 size = stackblocksize();
1473                 nleft = size - len;
1474                 if (nleft >= newlen)
1475                         break;
1476                 growstackblock();
1477         }
1478         return (char *)stackblock() + len;
1479 }
1480
1481 static char *
1482 stack_nputstr(const char *s, size_t n, char *p)
1483 {
1484         p = makestrspace(n, p);
1485         p = (char *)memcpy(p, s, n) + n;
1486         return p;
1487 }
1488
1489 static char *
1490 stack_putstr(const char *s, char *p)
1491 {
1492         return stack_nputstr(s, strlen(s), p);
1493 }
1494
1495 static char *
1496 _STPUTC(int c, char *p)
1497 {
1498         if (p == sstrend)
1499                 p = growstackstr();
1500         *p++ = c;
1501         return p;
1502 }
1503
1504 #define STARTSTACKSTR(p)        ((p) = stackblock())
1505 #define STPUTC(c, p)            ((p) = _STPUTC((c), (p)))
1506 #define CHECKSTRSPACE(n, p) do { \
1507         char *q = (p); \
1508         size_t l = (n); \
1509         size_t m = sstrend - q; \
1510         if (l > m) \
1511                 (p) = makestrspace(l, q); \
1512 } while (0)
1513 #define USTPUTC(c, p)           (*(p)++ = (c))
1514 #define STACKSTRNUL(p) do { \
1515         if ((p) == sstrend) \
1516                 (p) = growstackstr(); \
1517         *(p) = '\0'; \
1518 } while (0)
1519 #define STUNPUTC(p)             (--(p))
1520 #define STTOPC(p)               ((p)[-1])
1521 #define STADJUST(amount, p)     ((p) += (amount))
1522
1523 #define grabstackstr(p)         stalloc((char *)(p) - (char *)stackblock())
1524 #define ungrabstackstr(s, p)    stunalloc(s)
1525 #define stackstrend()           ((void *)sstrend)
1526
1527
1528 /* ============ String helpers */
1529
1530 /*
1531  * prefix -- see if pfx is a prefix of string.
1532  */
1533 static char *
1534 prefix(const char *string, const char *pfx)
1535 {
1536         while (*pfx) {
1537                 if (*pfx++ != *string++)
1538                         return NULL;
1539         }
1540         return (char *) string;
1541 }
1542
1543 /*
1544  * Check for a valid number.  This should be elsewhere.
1545  */
1546 static int
1547 is_number(const char *p)
1548 {
1549         do {
1550                 if (!isdigit(*p))
1551                         return 0;
1552         } while (*++p != '\0');
1553         return 1;
1554 }
1555
1556 /*
1557  * Convert a string of digits to an integer, printing an error message on
1558  * failure.
1559  */
1560 static int
1561 number(const char *s)
1562 {
1563         if (!is_number(s))
1564                 ash_msg_and_raise_error(msg_illnum, s);
1565         return atoi(s);
1566 }
1567
1568 /*
1569  * Produce a possibly single quoted string suitable as input to the shell.
1570  * The return string is allocated on the stack.
1571  */
1572 static char *
1573 single_quote(const char *s)
1574 {
1575         char *p;
1576
1577         STARTSTACKSTR(p);
1578
1579         do {
1580                 char *q;
1581                 size_t len;
1582
1583                 len = strchrnul(s, '\'') - s;
1584
1585                 q = p = makestrspace(len + 3, p);
1586
1587                 *q++ = '\'';
1588                 q = (char *)memcpy(q, s, len) + len;
1589                 *q++ = '\'';
1590                 s += len;
1591
1592                 STADJUST(q - p, p);
1593
1594                 if (*s != '\'')
1595                         break;
1596                 len = 0;
1597                 do len++; while (*++s == '\'');
1598
1599                 q = p = makestrspace(len + 3, p);
1600
1601                 *q++ = '"';
1602                 q = (char *)memcpy(q, s - len, len) + len;
1603                 *q++ = '"';
1604
1605                 STADJUST(q - p, p);
1606         } while (*s);
1607
1608         USTPUTC('\0', p);
1609
1610         return stackblock();
1611 }
1612
1613
1614 /* ============ nextopt */
1615
1616 static char **argptr;                  /* argument list for builtin commands */
1617 static char *optionarg;                /* set by nextopt (like getopt) */
1618 static char *optptr;                   /* used by nextopt */
1619
1620 /*
1621  * XXX - should get rid of. Have all builtins use getopt(3).
1622  * The library getopt must have the BSD extension static variable
1623  * "optreset", otherwise it can't be used within the shell safely.
1624  *
1625  * Standard option processing (a la getopt) for builtin routines.
1626  * The only argument that is passed to nextopt is the option string;
1627  * the other arguments are unnecessary. It returns the character,
1628  * or '\0' on end of input.
1629  */
1630 static int
1631 nextopt(const char *optstring)
1632 {
1633         char *p;
1634         const char *q;
1635         char c;
1636
1637         p = optptr;
1638         if (p == NULL || *p == '\0') {
1639                 /* We ate entire "-param", take next one */
1640                 p = *argptr;
1641                 if (p == NULL)
1642                         return '\0';
1643                 if (*p != '-')
1644                         return '\0';
1645                 if (*++p == '\0') /* just "-" ? */
1646                         return '\0';
1647                 argptr++;
1648                 if (LONE_DASH(p)) /* "--" ? */
1649                         return '\0';
1650                 /* p => next "-param" */
1651         }
1652         /* p => some option char in the middle of a "-param" */
1653         c = *p++;
1654         for (q = optstring; *q != c;) {
1655                 if (*q == '\0')
1656                         ash_msg_and_raise_error("illegal option -%c", c);
1657                 if (*++q == ':')
1658                         q++;
1659         }
1660         if (*++q == ':') {
1661                 if (*p == '\0') {
1662                         p = *argptr++;
1663                         if (p == NULL)
1664                                 ash_msg_and_raise_error("no arg for -%c option", c);
1665                 }
1666                 optionarg = p;
1667                 p = NULL;
1668         }
1669         optptr = p;
1670         return c;
1671 }
1672
1673
1674 /* ============ Shell variables */
1675
1676 /*
1677  * The parsefile structure pointed to by the global variable parsefile
1678  * contains information about the current file being read.
1679  */
1680 struct shparam {
1681         int nparam;             /* # of positional parameters (without $0) */
1682 #if ENABLE_ASH_GETOPTS
1683         int optind;             /* next parameter to be processed by getopts */
1684         int optoff;             /* used by getopts */
1685 #endif
1686         unsigned char malloced; /* if parameter list dynamically allocated */
1687         char **p;               /* parameter list */
1688 };
1689
1690 /*
1691  * Free the list of positional parameters.
1692  */
1693 static void
1694 freeparam(volatile struct shparam *param)
1695 {
1696         if (param->malloced) {
1697                 char **ap, **ap1;
1698                 ap = ap1 = param->p;
1699                 while (*ap)
1700                         free(*ap++);
1701                 free(ap1);
1702         }
1703 }
1704
1705 #if ENABLE_ASH_GETOPTS
1706 static void FAST_FUNC getoptsreset(const char *value);
1707 #endif
1708
1709 struct var {
1710         struct var *next;               /* next entry in hash list */
1711         int flags;                      /* flags are defined above */
1712         const char *text;               /* name=value */
1713         void (*func)(const char *) FAST_FUNC; /* function to be called when  */
1714                                         /* the variable gets set/unset */
1715 };
1716
1717 struct localvar {
1718         struct localvar *next;          /* next local variable in list */
1719         struct var *vp;                 /* the variable that was made local */
1720         int flags;                      /* saved flags */
1721         const char *text;               /* saved text */
1722 };
1723
1724 /* flags */
1725 #define VEXPORT         0x01    /* variable is exported */
1726 #define VREADONLY       0x02    /* variable cannot be modified */
1727 #define VSTRFIXED       0x04    /* variable struct is statically allocated */
1728 #define VTEXTFIXED      0x08    /* text is statically allocated */
1729 #define VSTACK          0x10    /* text is allocated on the stack */
1730 #define VUNSET          0x20    /* the variable is not set */
1731 #define VNOFUNC         0x40    /* don't call the callback function */
1732 #define VNOSET          0x80    /* do not set variable - just readonly test */
1733 #define VNOSAVE         0x100   /* when text is on the heap before setvareq */
1734 #if ENABLE_ASH_RANDOM_SUPPORT
1735 # define VDYNAMIC       0x200   /* dynamic variable */
1736 #else
1737 # define VDYNAMIC       0
1738 #endif
1739
1740 #ifdef IFS_BROKEN
1741 static const char defifsvar[] ALIGN1 = "IFS= \t\n";
1742 #define defifs (defifsvar + 4)
1743 #else
1744 static const char defifs[] ALIGN1 = " \t\n";
1745 #endif
1746
1747
1748 /* Need to be before varinit_data[] */
1749 #if ENABLE_LOCALE_SUPPORT
1750 static void FAST_FUNC
1751 change_lc_all(const char *value)
1752 {
1753         if (value && *value != '\0')
1754                 setlocale(LC_ALL, value);
1755 }
1756 static void FAST_FUNC
1757 change_lc_ctype(const char *value)
1758 {
1759         if (value && *value != '\0')
1760                 setlocale(LC_CTYPE, value);
1761 }
1762 #endif
1763 #if ENABLE_ASH_MAIL
1764 static void chkmail(void);
1765 static void changemail(const char *) FAST_FUNC;
1766 #endif
1767 static void changepath(const char *) FAST_FUNC;
1768 #if ENABLE_ASH_RANDOM_SUPPORT
1769 static void change_random(const char *) FAST_FUNC;
1770 #endif
1771
1772 static const struct {
1773         int flags;
1774         const char *text;
1775         void (*func)(const char *) FAST_FUNC;
1776 } varinit_data[] = {
1777 #ifdef IFS_BROKEN
1778         { VSTRFIXED|VTEXTFIXED       , defifsvar   , NULL            },
1779 #else
1780         { VSTRFIXED|VTEXTFIXED|VUNSET, "IFS\0"     , NULL            },
1781 #endif
1782 #if ENABLE_ASH_MAIL
1783         { VSTRFIXED|VTEXTFIXED|VUNSET, "MAIL\0"    , changemail      },
1784         { VSTRFIXED|VTEXTFIXED|VUNSET, "MAILPATH\0", changemail      },
1785 #endif
1786         { VSTRFIXED|VTEXTFIXED       , bb_PATH_root_path, changepath },
1787         { VSTRFIXED|VTEXTFIXED       , "PS1=$ "    , NULL            },
1788         { VSTRFIXED|VTEXTFIXED       , "PS2=> "    , NULL            },
1789         { VSTRFIXED|VTEXTFIXED       , "PS4=+ "    , NULL            },
1790 #if ENABLE_ASH_GETOPTS
1791         { VSTRFIXED|VTEXTFIXED       , "OPTIND=1"  , getoptsreset    },
1792 #endif
1793 #if ENABLE_ASH_RANDOM_SUPPORT
1794         { VSTRFIXED|VTEXTFIXED|VUNSET|VDYNAMIC, "RANDOM\0", change_random },
1795 #endif
1796 #if ENABLE_LOCALE_SUPPORT
1797         { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_ALL\0"  , change_lc_all   },
1798         { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_CTYPE\0", change_lc_ctype },
1799 #endif
1800 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
1801         { VSTRFIXED|VTEXTFIXED|VUNSET, "HISTFILE\0", NULL            },
1802 #endif
1803 };
1804
1805 struct redirtab;
1806
1807 struct globals_var {
1808         struct shparam shellparam;      /* $@ current positional parameters */
1809         struct redirtab *redirlist;
1810         int g_nullredirs;
1811         int preverrout_fd;   /* save fd2 before print debug if xflag is set. */
1812         struct var *vartab[VTABSIZE];
1813         struct var varinit[ARRAY_SIZE(varinit_data)];
1814 };
1815 extern struct globals_var *const ash_ptr_to_globals_var;
1816 #define G_var (*ash_ptr_to_globals_var)
1817 #define shellparam    (G_var.shellparam   )
1818 //#define redirlist     (G_var.redirlist    )
1819 #define g_nullredirs  (G_var.g_nullredirs )
1820 #define preverrout_fd (G_var.preverrout_fd)
1821 #define vartab        (G_var.vartab       )
1822 #define varinit       (G_var.varinit      )
1823 #define INIT_G_var() do { \
1824         unsigned i; \
1825         (*(struct globals_var**)&ash_ptr_to_globals_var) = xzalloc(sizeof(G_var)); \
1826         barrier(); \
1827         for (i = 0; i < ARRAY_SIZE(varinit_data); i++) { \
1828                 varinit[i].flags = varinit_data[i].flags; \
1829                 varinit[i].text  = varinit_data[i].text; \
1830                 varinit[i].func  = varinit_data[i].func; \
1831         } \
1832 } while (0)
1833
1834 #define vifs      varinit[0]
1835 #if ENABLE_ASH_MAIL
1836 # define vmail    (&vifs)[1]
1837 # define vmpath   (&vmail)[1]
1838 # define vpath    (&vmpath)[1]
1839 #else
1840 # define vpath    (&vifs)[1]
1841 #endif
1842 #define vps1      (&vpath)[1]
1843 #define vps2      (&vps1)[1]
1844 #define vps4      (&vps2)[1]
1845 #if ENABLE_ASH_GETOPTS
1846 # define voptind  (&vps4)[1]
1847 # if ENABLE_ASH_RANDOM_SUPPORT
1848 #  define vrandom (&voptind)[1]
1849 # endif
1850 #else
1851 # if ENABLE_ASH_RANDOM_SUPPORT
1852 #  define vrandom (&vps4)[1]
1853 # endif
1854 #endif
1855
1856 /*
1857  * The following macros access the values of the above variables.
1858  * They have to skip over the name.  They return the null string
1859  * for unset variables.
1860  */
1861 #define ifsval()        (vifs.text + 4)
1862 #define ifsset()        ((vifs.flags & VUNSET) == 0)
1863 #if ENABLE_ASH_MAIL
1864 # define mailval()      (vmail.text + 5)
1865 # define mpathval()     (vmpath.text + 9)
1866 # define mpathset()     ((vmpath.flags & VUNSET) == 0)
1867 #endif
1868 #define pathval()       (vpath.text + 5)
1869 #define ps1val()        (vps1.text + 4)
1870 #define ps2val()        (vps2.text + 4)
1871 #define ps4val()        (vps4.text + 4)
1872 #if ENABLE_ASH_GETOPTS
1873 # define optindval()    (voptind.text + 7)
1874 #endif
1875
1876
1877 #define is_name(c)      ((c) == '_' || isalpha((unsigned char)(c)))
1878 #define is_in_name(c)   ((c) == '_' || isalnum((unsigned char)(c)))
1879
1880 #if ENABLE_ASH_GETOPTS
1881 static void FAST_FUNC
1882 getoptsreset(const char *value)
1883 {
1884         shellparam.optind = number(value);
1885         shellparam.optoff = -1;
1886 }
1887 #endif
1888
1889 /*
1890  * Return of a legal variable name (a letter or underscore followed by zero or
1891  * more letters, underscores, and digits).
1892  */
1893 static char *
1894 endofname(const char *name)
1895 {
1896         char *p;
1897
1898         p = (char *) name;
1899         if (!is_name(*p))
1900                 return p;
1901         while (*++p) {
1902                 if (!is_in_name(*p))
1903                         break;
1904         }
1905         return p;
1906 }
1907
1908 /*
1909  * Compares two strings up to the first = or '\0'.  The first
1910  * string must be terminated by '='; the second may be terminated by
1911  * either '=' or '\0'.
1912  */
1913 static int
1914 varcmp(const char *p, const char *q)
1915 {
1916         int c, d;
1917
1918         while ((c = *p) == (d = *q)) {
1919                 if (!c || c == '=')
1920                         goto out;
1921                 p++;
1922                 q++;
1923         }
1924         if (c == '=')
1925                 c = '\0';
1926         if (d == '=')
1927                 d = '\0';
1928  out:
1929         return c - d;
1930 }
1931
1932 static int
1933 varequal(const char *a, const char *b)
1934 {
1935         return !varcmp(a, b);
1936 }
1937
1938 /*
1939  * Find the appropriate entry in the hash table from the name.
1940  */
1941 static struct var **
1942 hashvar(const char *p)
1943 {
1944         unsigned hashval;
1945
1946         hashval = ((unsigned char) *p) << 4;
1947         while (*p && *p != '=')
1948                 hashval += (unsigned char) *p++;
1949         return &vartab[hashval % VTABSIZE];
1950 }
1951
1952 static int
1953 vpcmp(const void *a, const void *b)
1954 {
1955         return varcmp(*(const char **)a, *(const char **)b);
1956 }
1957
1958 /*
1959  * This routine initializes the builtin variables.
1960  */
1961 static void
1962 initvar(void)
1963 {
1964         struct var *vp;
1965         struct var *end;
1966         struct var **vpp;
1967
1968         /*
1969          * PS1 depends on uid
1970          */
1971 #if ENABLE_FEATURE_EDITING && ENABLE_FEATURE_EDITING_FANCY_PROMPT
1972         vps1.text = "PS1=\\w \\$ ";
1973 #else
1974         if (!geteuid())
1975                 vps1.text = "PS1=# ";
1976 #endif
1977         vp = varinit;
1978         end = vp + ARRAY_SIZE(varinit);
1979         do {
1980                 vpp = hashvar(vp->text);
1981                 vp->next = *vpp;
1982                 *vpp = vp;
1983         } while (++vp < end);
1984 }
1985
1986 static struct var **
1987 findvar(struct var **vpp, const char *name)
1988 {
1989         for (; *vpp; vpp = &(*vpp)->next) {
1990                 if (varequal((*vpp)->text, name)) {
1991                         break;
1992                 }
1993         }
1994         return vpp;
1995 }
1996
1997 /*
1998  * Find the value of a variable.  Returns NULL if not set.
1999  */
2000 static const char *
2001 lookupvar(const char *name)
2002 {
2003         struct var *v;
2004
2005         v = *findvar(hashvar(name), name);
2006         if (v) {
2007 #if ENABLE_ASH_RANDOM_SUPPORT
2008         /*
2009          * Dynamic variables are implemented roughly the same way they are
2010          * in bash. Namely, they're "special" so long as they aren't unset.
2011          * As soon as they're unset, they're no longer dynamic, and dynamic
2012          * lookup will no longer happen at that point. -- PFM.
2013          */
2014                 if ((v->flags & VDYNAMIC))
2015                         (*v->func)(NULL);
2016 #endif
2017                 if (!(v->flags & VUNSET))
2018                         return strchrnul(v->text, '=') + 1;
2019         }
2020         return NULL;
2021 }
2022
2023 /*
2024  * Search the environment of a builtin command.
2025  */
2026 static const char *
2027 bltinlookup(const char *name)
2028 {
2029         struct strlist *sp;
2030
2031         for (sp = cmdenviron; sp; sp = sp->next) {
2032                 if (varequal(sp->text, name))
2033                         return strchrnul(sp->text, '=') + 1;
2034         }
2035         return lookupvar(name);
2036 }
2037
2038 /*
2039  * Same as setvar except that the variable and value are passed in
2040  * the first argument as name=value.  Since the first argument will
2041  * be actually stored in the table, it should not be a string that
2042  * will go away.
2043  * Called with interrupts off.
2044  */
2045 static void
2046 setvareq(char *s, int flags)
2047 {
2048         struct var *vp, **vpp;
2049
2050         vpp = hashvar(s);
2051         flags |= (VEXPORT & (((unsigned) (1 - aflag)) - 1));
2052         vp = *findvar(vpp, s);
2053         if (vp) {
2054                 if ((vp->flags & (VREADONLY|VDYNAMIC)) == VREADONLY) {
2055                         const char *n;
2056
2057                         if (flags & VNOSAVE)
2058                                 free(s);
2059                         n = vp->text;
2060                         ash_msg_and_raise_error("%.*s: is read only", strchrnul(n, '=') - n, n);
2061                 }
2062
2063                 if (flags & VNOSET)
2064                         return;
2065
2066                 if (vp->func && (flags & VNOFUNC) == 0)
2067                         (*vp->func)(strchrnul(s, '=') + 1);
2068
2069                 if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0)
2070                         free((char*)vp->text);
2071
2072                 flags |= vp->flags & ~(VTEXTFIXED|VSTACK|VNOSAVE|VUNSET);
2073         } else {
2074                 if (flags & VNOSET)
2075                         return;
2076                 /* not found */
2077                 vp = ckzalloc(sizeof(*vp));
2078                 vp->next = *vpp;
2079                 /*vp->func = NULL; - ckzalloc did it */
2080                 *vpp = vp;
2081         }
2082         if (!(flags & (VTEXTFIXED|VSTACK|VNOSAVE)))
2083                 s = ckstrdup(s);
2084         vp->text = s;
2085         vp->flags = flags;
2086 }
2087
2088 /*
2089  * Set the value of a variable.  The flags argument is ored with the
2090  * flags of the variable.  If val is NULL, the variable is unset.
2091  */
2092 static void
2093 setvar(const char *name, const char *val, int flags)
2094 {
2095         char *p, *q;
2096         size_t namelen;
2097         char *nameeq;
2098         size_t vallen;
2099
2100         q = endofname(name);
2101         p = strchrnul(q, '=');
2102         namelen = p - name;
2103         if (!namelen || p != q)
2104                 ash_msg_and_raise_error("%.*s: bad variable name", namelen, name);
2105         vallen = 0;
2106         if (val == NULL) {
2107                 flags |= VUNSET;
2108         } else {
2109                 vallen = strlen(val);
2110         }
2111         INT_OFF;
2112         nameeq = ckmalloc(namelen + vallen + 2);
2113         p = (char *)memcpy(nameeq, name, namelen) + namelen;
2114         if (val) {
2115                 *p++ = '=';
2116                 p = (char *)memcpy(p, val, vallen) + vallen;
2117         }
2118         *p = '\0';
2119         setvareq(nameeq, flags | VNOSAVE);
2120         INT_ON;
2121 }
2122
2123 #if ENABLE_ASH_GETOPTS
2124 /*
2125  * Safe version of setvar, returns 1 on success 0 on failure.
2126  */
2127 static int
2128 setvarsafe(const char *name, const char *val, int flags)
2129 {
2130         int err;
2131         volatile int saveint;
2132         struct jmploc *volatile savehandler = exception_handler;
2133         struct jmploc jmploc;
2134
2135         SAVE_INT(saveint);
2136         if (setjmp(jmploc.loc))
2137                 err = 1;
2138         else {
2139                 exception_handler = &jmploc;
2140                 setvar(name, val, flags);
2141                 err = 0;
2142         }
2143         exception_handler = savehandler;
2144         RESTORE_INT(saveint);
2145         return err;
2146 }
2147 #endif
2148
2149 /*
2150  * Unset the specified variable.
2151  */
2152 static int
2153 unsetvar(const char *s)
2154 {
2155         struct var **vpp;
2156         struct var *vp;
2157         int retval;
2158
2159         vpp = findvar(hashvar(s), s);
2160         vp = *vpp;
2161         retval = 2;
2162         if (vp) {
2163                 int flags = vp->flags;
2164
2165                 retval = 1;
2166                 if (flags & VREADONLY)
2167                         goto out;
2168 #if ENABLE_ASH_RANDOM_SUPPORT
2169                 vp->flags &= ~VDYNAMIC;
2170 #endif
2171                 if (flags & VUNSET)
2172                         goto ok;
2173                 if ((flags & VSTRFIXED) == 0) {
2174                         INT_OFF;
2175                         if ((flags & (VTEXTFIXED|VSTACK)) == 0)
2176                                 free((char*)vp->text);
2177                         *vpp = vp->next;
2178                         free(vp);
2179                         INT_ON;
2180                 } else {
2181                         setvar(s, 0, 0);
2182                         vp->flags &= ~VEXPORT;
2183                 }
2184  ok:
2185                 retval = 0;
2186         }
2187  out:
2188         return retval;
2189 }
2190
2191 /*
2192  * Process a linked list of variable assignments.
2193  */
2194 static void
2195 listsetvar(struct strlist *list_set_var, int flags)
2196 {
2197         struct strlist *lp = list_set_var;
2198
2199         if (!lp)
2200                 return;
2201         INT_OFF;
2202         do {
2203                 setvareq(lp->text, flags);
2204                 lp = lp->next;
2205         } while (lp);
2206         INT_ON;
2207 }
2208
2209 /*
2210  * Generate a list of variables satisfying the given conditions.
2211  */
2212 static char **
2213 listvars(int on, int off, char ***end)
2214 {
2215         struct var **vpp;
2216         struct var *vp;
2217         char **ep;
2218         int mask;
2219
2220         STARTSTACKSTR(ep);
2221         vpp = vartab;
2222         mask = on | off;
2223         do {
2224                 for (vp = *vpp; vp; vp = vp->next) {
2225                         if ((vp->flags & mask) == on) {
2226                                 if (ep == stackstrend())
2227                                         ep = growstackstr();
2228                                 *ep++ = (char *) vp->text;
2229                         }
2230                 }
2231         } while (++vpp < vartab + VTABSIZE);
2232         if (ep == stackstrend())
2233                 ep = growstackstr();
2234         if (end)
2235                 *end = ep;
2236         *ep++ = NULL;
2237         return grabstackstr(ep);
2238 }
2239
2240
2241 /* ============ Path search helper
2242  *
2243  * The variable path (passed by reference) should be set to the start
2244  * of the path before the first call; path_advance will update
2245  * this value as it proceeds.  Successive calls to path_advance will return
2246  * the possible path expansions in sequence.  If an option (indicated by
2247  * a percent sign) appears in the path entry then the global variable
2248  * pathopt will be set to point to it; otherwise pathopt will be set to
2249  * NULL.
2250  */
2251 static const char *pathopt;     /* set by path_advance */
2252
2253 static char *
2254 path_advance(const char **path, const char *name)
2255 {
2256         const char *p;
2257         char *q;
2258         const char *start;
2259         size_t len;
2260
2261         if (*path == NULL)
2262                 return NULL;
2263         start = *path;
2264         for (p = start; *p && *p != ':' && *p != '%'; p++)
2265                 continue;
2266         len = p - start + strlen(name) + 2;     /* "2" is for '/' and '\0' */
2267         while (stackblocksize() < len)
2268                 growstackblock();
2269         q = stackblock();
2270         if (p != start) {
2271                 memcpy(q, start, p - start);
2272                 q += p - start;
2273                 *q++ = '/';
2274         }
2275         strcpy(q, name);
2276         pathopt = NULL;
2277         if (*p == '%') {
2278                 pathopt = ++p;
2279                 while (*p && *p != ':')
2280                         p++;
2281         }
2282         if (*p == ':')
2283                 *path = p + 1;
2284         else
2285                 *path = NULL;
2286         return stalloc(len);
2287 }
2288
2289
2290 /* ============ Prompt */
2291
2292 static smallint doprompt;                   /* if set, prompt the user */
2293 static smallint needprompt;                 /* true if interactive and at start of line */
2294
2295 #if ENABLE_FEATURE_EDITING
2296 static line_input_t *line_input_state;
2297 static const char *cmdedit_prompt;
2298 static void
2299 putprompt(const char *s)
2300 {
2301         if (ENABLE_ASH_EXPAND_PRMT) {
2302                 free((char*)cmdedit_prompt);
2303                 cmdedit_prompt = ckstrdup(s);
2304                 return;
2305         }
2306         cmdedit_prompt = s;
2307 }
2308 #else
2309 static void
2310 putprompt(const char *s)
2311 {
2312         out2str(s);
2313 }
2314 #endif
2315
2316 #if ENABLE_ASH_EXPAND_PRMT
2317 /* expandstr() needs parsing machinery, so it is far away ahead... */
2318 static const char *expandstr(const char *ps);
2319 #else
2320 #define expandstr(s) s
2321 #endif
2322
2323 static void
2324 setprompt(int whichprompt)
2325 {
2326         const char *prompt;
2327 #if ENABLE_ASH_EXPAND_PRMT
2328         struct stackmark smark;
2329 #endif
2330
2331         needprompt = 0;
2332
2333         switch (whichprompt) {
2334         case 1:
2335                 prompt = ps1val();
2336                 break;
2337         case 2:
2338                 prompt = ps2val();
2339                 break;
2340         default:                        /* 0 */
2341                 prompt = nullstr;
2342         }
2343 #if ENABLE_ASH_EXPAND_PRMT
2344         setstackmark(&smark);
2345         stalloc(stackblocksize());
2346 #endif
2347         putprompt(expandstr(prompt));
2348 #if ENABLE_ASH_EXPAND_PRMT
2349         popstackmark(&smark);
2350 #endif
2351 }
2352
2353
2354 /* ============ The cd and pwd commands */
2355
2356 #define CD_PHYSICAL 1
2357 #define CD_PRINT 2
2358
2359 static int
2360 cdopt(void)
2361 {
2362         int flags = 0;
2363         int i, j;
2364
2365         j = 'L';
2366         while ((i = nextopt("LP")) != '\0') {
2367                 if (i != j) {
2368                         flags ^= CD_PHYSICAL;
2369                         j = i;
2370                 }
2371         }
2372
2373         return flags;
2374 }
2375
2376 /*
2377  * Update curdir (the name of the current directory) in response to a
2378  * cd command.
2379  */
2380 static const char *
2381 updatepwd(const char *dir)
2382 {
2383         char *new;
2384         char *p;
2385         char *cdcomppath;
2386         const char *lim;
2387
2388         cdcomppath = ststrdup(dir);
2389         STARTSTACKSTR(new);
2390         if (*dir != '/') {
2391                 if (curdir == nullstr)
2392                         return 0;
2393                 new = stack_putstr(curdir, new);
2394         }
2395         new = makestrspace(strlen(dir) + 2, new);
2396         lim = (char *)stackblock() + 1;
2397         if (*dir != '/') {
2398                 if (new[-1] != '/')
2399                         USTPUTC('/', new);
2400                 if (new > lim && *lim == '/')
2401                         lim++;
2402         } else {
2403                 USTPUTC('/', new);
2404                 cdcomppath++;
2405                 if (dir[1] == '/' && dir[2] != '/') {
2406                         USTPUTC('/', new);
2407                         cdcomppath++;
2408                         lim++;
2409                 }
2410         }
2411         p = strtok(cdcomppath, "/");
2412         while (p) {
2413                 switch (*p) {
2414                 case '.':
2415                         if (p[1] == '.' && p[2] == '\0') {
2416                                 while (new > lim) {
2417                                         STUNPUTC(new);
2418                                         if (new[-1] == '/')
2419                                                 break;
2420                                 }
2421                                 break;
2422                         }
2423                         if (p[1] == '\0')
2424                                 break;
2425                         /* fall through */
2426                 default:
2427                         new = stack_putstr(p, new);
2428                         USTPUTC('/', new);
2429                 }
2430                 p = strtok(0, "/");
2431         }
2432         if (new > lim)
2433                 STUNPUTC(new);
2434         *new = 0;
2435         return stackblock();
2436 }
2437
2438 /*
2439  * Find out what the current directory is. If we already know the current
2440  * directory, this routine returns immediately.
2441  */
2442 static char *
2443 getpwd(void)
2444 {
2445         char *dir = getcwd(NULL, 0); /* huh, using glibc extension? */
2446         return dir ? dir : nullstr;
2447 }
2448
2449 static void
2450 setpwd(const char *val, int setold)
2451 {
2452         char *oldcur, *dir;
2453
2454         oldcur = dir = curdir;
2455
2456         if (setold) {
2457                 setvar("OLDPWD", oldcur, VEXPORT);
2458         }
2459         INT_OFF;
2460         if (physdir != nullstr) {
2461                 if (physdir != oldcur)
2462                         free(physdir);
2463                 physdir = nullstr;
2464         }
2465         if (oldcur == val || !val) {
2466                 char *s = getpwd();
2467                 physdir = s;
2468                 if (!val)
2469                         dir = s;
2470         } else
2471                 dir = ckstrdup(val);
2472         if (oldcur != dir && oldcur != nullstr) {
2473                 free(oldcur);
2474         }
2475         curdir = dir;
2476         INT_ON;
2477         setvar("PWD", dir, VEXPORT);
2478 }
2479
2480 static void hashcd(void);
2481
2482 /*
2483  * Actually do the chdir.  We also call hashcd to let the routines in exec.c
2484  * know that the current directory has changed.
2485  */
2486 static int
2487 docd(const char *dest, int flags)
2488 {
2489         const char *dir = 0;
2490         int err;
2491
2492         TRACE(("docd(\"%s\", %d) called\n", dest, flags));
2493
2494         INT_OFF;
2495         if (!(flags & CD_PHYSICAL)) {
2496                 dir = updatepwd(dest);
2497                 if (dir)
2498                         dest = dir;
2499         }
2500         err = chdir(dest);
2501         if (err)
2502                 goto out;
2503         setpwd(dir, 1);
2504         hashcd();
2505  out:
2506         INT_ON;
2507         return err;
2508 }
2509
2510 static int FAST_FUNC
2511 cdcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
2512 {
2513         const char *dest;
2514         const char *path;
2515         const char *p;
2516         char c;
2517         struct stat statb;
2518         int flags;
2519
2520         flags = cdopt();
2521         dest = *argptr;
2522         if (!dest)
2523                 dest = bltinlookup(homestr);
2524         else if (LONE_DASH(dest)) {
2525                 dest = bltinlookup("OLDPWD");
2526                 flags |= CD_PRINT;
2527         }
2528         if (!dest)
2529                 dest = nullstr;
2530         if (*dest == '/')
2531                 goto step7;
2532         if (*dest == '.') {
2533                 c = dest[1];
2534  dotdot:
2535                 switch (c) {
2536                 case '\0':
2537                 case '/':
2538                         goto step6;
2539                 case '.':
2540                         c = dest[2];
2541                         if (c != '.')
2542                                 goto dotdot;
2543                 }
2544         }
2545         if (!*dest)
2546                 dest = ".";
2547         path = bltinlookup("CDPATH");
2548         if (!path) {
2549  step6:
2550  step7:
2551                 p = dest;
2552                 goto docd;
2553         }
2554         do {
2555                 c = *path;
2556                 p = path_advance(&path, dest);
2557                 if (stat(p, &statb) >= 0 && S_ISDIR(statb.st_mode)) {
2558                         if (c && c != ':')
2559                                 flags |= CD_PRINT;
2560  docd:
2561                         if (!docd(p, flags))
2562                                 goto out;
2563                         break;
2564                 }
2565         } while (path);
2566         ash_msg_and_raise_error("can't cd to %s", dest);
2567         /* NOTREACHED */
2568  out:
2569         if (flags & CD_PRINT)
2570                 out1fmt(snlfmt, curdir);
2571         return 0;
2572 }
2573
2574 static int FAST_FUNC
2575 pwdcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
2576 {
2577         int flags;
2578         const char *dir = curdir;
2579
2580         flags = cdopt();
2581         if (flags) {
2582                 if (physdir == nullstr)
2583                         setpwd(dir, 0);
2584                 dir = physdir;
2585         }
2586         out1fmt(snlfmt, dir);
2587         return 0;
2588 }
2589
2590
2591 /* ============ ... */
2592
2593
2594 #define IBUFSIZ COMMON_BUFSIZE
2595 /* buffer for top level input file */
2596 #define basebuf bb_common_bufsiz1
2597
2598 /* Syntax classes */
2599 #define CWORD     0             /* character is nothing special */
2600 #define CNL       1             /* newline character */
2601 #define CBACK     2             /* a backslash character */
2602 #define CSQUOTE   3             /* single quote */
2603 #define CDQUOTE   4             /* double quote */
2604 #define CENDQUOTE 5             /* a terminating quote */
2605 #define CBQUOTE   6             /* backwards single quote */
2606 #define CVAR      7             /* a dollar sign */
2607 #define CENDVAR   8             /* a '}' character */
2608 #define CLP       9             /* a left paren in arithmetic */
2609 #define CRP      10             /* a right paren in arithmetic */
2610 #define CENDFILE 11             /* end of file */
2611 #define CCTL     12             /* like CWORD, except it must be escaped */
2612 #define CSPCL    13             /* these terminate a word */
2613 #define CIGN     14             /* character should be ignored */
2614
2615 #define PEOF     256
2616 #if ENABLE_ASH_ALIAS
2617 # define PEOA    257
2618 #endif
2619
2620 #define USE_SIT_FUNCTION ENABLE_ASH_OPTIMIZE_FOR_SIZE
2621
2622 #if ENABLE_SH_MATH_SUPPORT
2623 # define SIT_ITEM(a,b,c,d) (a | (b << 4) | (c << 8) | (d << 12))
2624 #else
2625 # define SIT_ITEM(a,b,c,d) (a | (b << 4) | (c << 8))
2626 #endif
2627 static const uint16_t S_I_T[] = {
2628 #if ENABLE_ASH_ALIAS
2629         SIT_ITEM(CSPCL   , CIGN     , CIGN , CIGN   ),    /* 0, PEOA */
2630 #endif
2631         SIT_ITEM(CSPCL   , CWORD    , CWORD, CWORD  ),    /* 1, ' ' */
2632         SIT_ITEM(CNL     , CNL      , CNL  , CNL    ),    /* 2, \n */
2633         SIT_ITEM(CWORD   , CCTL     , CCTL , CWORD  ),    /* 3, !*-/:=?[]~ */
2634         SIT_ITEM(CDQUOTE , CENDQUOTE, CWORD, CWORD  ),    /* 4, '"' */
2635         SIT_ITEM(CVAR    , CVAR     , CWORD, CVAR   ),    /* 5, $ */
2636         SIT_ITEM(CSQUOTE , CWORD    , CENDQUOTE, CWORD),  /* 6, "'" */
2637         SIT_ITEM(CSPCL   , CWORD    , CWORD, CLP    ),    /* 7, ( */
2638         SIT_ITEM(CSPCL   , CWORD    , CWORD, CRP    ),    /* 8, ) */
2639         SIT_ITEM(CBACK   , CBACK    , CCTL , CBACK  ),    /* 9, \ */
2640         SIT_ITEM(CBQUOTE , CBQUOTE  , CWORD, CBQUOTE),    /* 10, ` */
2641         SIT_ITEM(CENDVAR , CENDVAR  , CWORD, CENDVAR),    /* 11, } */
2642 #if !USE_SIT_FUNCTION
2643         SIT_ITEM(CENDFILE, CENDFILE , CENDFILE, CENDFILE),/* 12, PEOF */
2644         SIT_ITEM(CWORD   , CWORD    , CWORD, CWORD  ),    /* 13, 0-9A-Za-z */
2645         SIT_ITEM(CCTL    , CCTL     , CCTL , CCTL   )     /* 14, CTLESC ... */
2646 #endif
2647 #undef SIT_ITEM
2648 };
2649 /* Constants below must match table above */
2650 enum {
2651 #if ENABLE_ASH_ALIAS
2652         CSPCL_CIGN_CIGN_CIGN               , /*  0 */
2653 #endif
2654         CSPCL_CWORD_CWORD_CWORD            , /*  1 */
2655         CNL_CNL_CNL_CNL                    , /*  2 */
2656         CWORD_CCTL_CCTL_CWORD              , /*  3 */
2657         CDQUOTE_CENDQUOTE_CWORD_CWORD      , /*  4 */
2658         CVAR_CVAR_CWORD_CVAR               , /*  5 */
2659         CSQUOTE_CWORD_CENDQUOTE_CWORD      , /*  6 */
2660         CSPCL_CWORD_CWORD_CLP              , /*  7 */
2661         CSPCL_CWORD_CWORD_CRP              , /*  8 */
2662         CBACK_CBACK_CCTL_CBACK             , /*  9 */
2663         CBQUOTE_CBQUOTE_CWORD_CBQUOTE      , /* 10 */
2664         CENDVAR_CENDVAR_CWORD_CENDVAR      , /* 11 */
2665         CENDFILE_CENDFILE_CENDFILE_CENDFILE, /* 12 */
2666         CWORD_CWORD_CWORD_CWORD            , /* 13 */
2667         CCTL_CCTL_CCTL_CCTL                , /* 14 */
2668 };
2669
2670 /* c in SIT(c, syntax) must be an *unsigned char* or PEOA or PEOF,
2671  * caller must ensure proper cast on it if c is *char_ptr!
2672  */
2673 /* Values for syntax param */
2674 #define BASESYNTAX 0    /* not in quotes */
2675 #define DQSYNTAX   1    /* in double quotes */
2676 #define SQSYNTAX   2    /* in single quotes */
2677 #define ARISYNTAX  3    /* in arithmetic */
2678 #define PSSYNTAX   4    /* prompt. never passed to SIT() */
2679
2680 #if USE_SIT_FUNCTION
2681
2682 static int
2683 SIT(int c, int syntax)
2684 {
2685         static const char spec_symbls[] ALIGN1 = "\t\n !\"$&'()*-/:;<=>?[\\]`|}~";
2686 # if ENABLE_ASH_ALIAS
2687         static const uint8_t syntax_index_table[] ALIGN1 = {
2688                 1, 2, 1, 3, 4, 5, 1, 6,         /* "\t\n !\"$&'" */
2689                 7, 8, 3, 3, 3, 3, 1, 1,         /* "()*-/:;<" */
2690                 3, 1, 3, 3, 9, 3, 10, 1,        /* "=>?[\\]`|" */
2691                 11, 3                           /* "}~" */
2692         };
2693 # else
2694         static const uint8_t syntax_index_table[] ALIGN1 = {
2695                 0, 1, 0, 2, 3, 4, 0, 5,         /* "\t\n !\"$&'" */
2696                 6, 7, 2, 2, 2, 2, 0, 0,         /* "()*-/:;<" */
2697                 2, 0, 2, 2, 8, 2, 9, 0,         /* "=>?[\\]`|" */
2698                 10, 2                           /* "}~" */
2699         };
2700 # endif
2701         const char *s;
2702         int indx;
2703
2704         if (c == PEOF)
2705                 return CENDFILE;
2706 # if ENABLE_ASH_ALIAS
2707         if (c == PEOA)
2708                 indx = 0;
2709         else
2710 # endif
2711         {
2712                 /* Cast is purely for paranoia here,
2713                  * just in case someone passed signed char to us */
2714                 if ((unsigned char)c >= CTL_FIRST
2715                  && (unsigned char)c <= CTL_LAST
2716                 ) {
2717                         return CCTL;
2718                 }
2719                 s = strchrnul(spec_symbls, c);
2720                 if (*s == '\0')
2721                         return CWORD;
2722                 indx = syntax_index_table[s - spec_symbls];
2723         }
2724         return (S_I_T[indx] >> (syntax*4)) & 0xf;
2725 }
2726
2727 #else   /* !USE_SIT_FUNCTION */
2728
2729 static const uint8_t syntax_index_table[] = {
2730         /* BASESYNTAX_DQSYNTAX_SQSYNTAX_ARISYNTAX */
2731         /*   0      */ CWORD_CWORD_CWORD_CWORD,
2732         /*   1      */ CWORD_CWORD_CWORD_CWORD,
2733         /*   2      */ CWORD_CWORD_CWORD_CWORD,
2734         /*   3      */ CWORD_CWORD_CWORD_CWORD,
2735         /*   4      */ CWORD_CWORD_CWORD_CWORD,
2736         /*   5      */ CWORD_CWORD_CWORD_CWORD,
2737         /*   6      */ CWORD_CWORD_CWORD_CWORD,
2738         /*   7      */ CWORD_CWORD_CWORD_CWORD,
2739         /*   8      */ CWORD_CWORD_CWORD_CWORD,
2740         /*   9 "\t" */ CSPCL_CWORD_CWORD_CWORD,
2741         /*  10 "\n" */ CNL_CNL_CNL_CNL,
2742         /*  11      */ CWORD_CWORD_CWORD_CWORD,
2743         /*  12      */ CWORD_CWORD_CWORD_CWORD,
2744         /*  13      */ CWORD_CWORD_CWORD_CWORD,
2745         /*  14      */ CWORD_CWORD_CWORD_CWORD,
2746         /*  15      */ CWORD_CWORD_CWORD_CWORD,
2747         /*  16      */ CWORD_CWORD_CWORD_CWORD,
2748         /*  17      */ CWORD_CWORD_CWORD_CWORD,
2749         /*  18      */ CWORD_CWORD_CWORD_CWORD,
2750         /*  19      */ CWORD_CWORD_CWORD_CWORD,
2751         /*  20      */ CWORD_CWORD_CWORD_CWORD,
2752         /*  21      */ CWORD_CWORD_CWORD_CWORD,
2753         /*  22      */ CWORD_CWORD_CWORD_CWORD,
2754         /*  23      */ CWORD_CWORD_CWORD_CWORD,
2755         /*  24      */ CWORD_CWORD_CWORD_CWORD,
2756         /*  25      */ CWORD_CWORD_CWORD_CWORD,
2757         /*  26      */ CWORD_CWORD_CWORD_CWORD,
2758         /*  27      */ CWORD_CWORD_CWORD_CWORD,
2759         /*  28      */ CWORD_CWORD_CWORD_CWORD,
2760         /*  29      */ CWORD_CWORD_CWORD_CWORD,
2761         /*  30      */ CWORD_CWORD_CWORD_CWORD,
2762         /*  31      */ CWORD_CWORD_CWORD_CWORD,
2763         /*  32  " " */ CSPCL_CWORD_CWORD_CWORD,
2764         /*  33  "!" */ CWORD_CCTL_CCTL_CWORD,
2765         /*  34  """ */ CDQUOTE_CENDQUOTE_CWORD_CWORD,
2766         /*  35  "#" */ CWORD_CWORD_CWORD_CWORD,
2767         /*  36  "$" */ CVAR_CVAR_CWORD_CVAR,
2768         /*  37  "%" */ CWORD_CWORD_CWORD_CWORD,
2769         /*  38  "&" */ CSPCL_CWORD_CWORD_CWORD,
2770         /*  39  "'" */ CSQUOTE_CWORD_CENDQUOTE_CWORD,
2771         /*  40  "(" */ CSPCL_CWORD_CWORD_CLP,
2772         /*  41  ")" */ CSPCL_CWORD_CWORD_CRP,
2773         /*  42  "*" */ CWORD_CCTL_CCTL_CWORD,
2774         /*  43  "+" */ CWORD_CWORD_CWORD_CWORD,
2775         /*  44  "," */ CWORD_CWORD_CWORD_CWORD,
2776         /*  45  "-" */ CWORD_CCTL_CCTL_CWORD,
2777         /*  46  "." */ CWORD_CWORD_CWORD_CWORD,
2778         /*  47  "/" */ CWORD_CCTL_CCTL_CWORD,
2779         /*  48  "0" */ CWORD_CWORD_CWORD_CWORD,
2780         /*  49  "1" */ CWORD_CWORD_CWORD_CWORD,
2781         /*  50  "2" */ CWORD_CWORD_CWORD_CWORD,
2782         /*  51  "3" */ CWORD_CWORD_CWORD_CWORD,
2783         /*  52  "4" */ CWORD_CWORD_CWORD_CWORD,
2784         /*  53  "5" */ CWORD_CWORD_CWORD_CWORD,
2785         /*  54  "6" */ CWORD_CWORD_CWORD_CWORD,
2786         /*  55  "7" */ CWORD_CWORD_CWORD_CWORD,
2787         /*  56  "8" */ CWORD_CWORD_CWORD_CWORD,
2788         /*  57  "9" */ CWORD_CWORD_CWORD_CWORD,
2789         /*  58  ":" */ CWORD_CCTL_CCTL_CWORD,
2790         /*  59  ";" */ CSPCL_CWORD_CWORD_CWORD,
2791         /*  60  "<" */ CSPCL_CWORD_CWORD_CWORD,
2792         /*  61  "=" */ CWORD_CCTL_CCTL_CWORD,
2793         /*  62  ">" */ CSPCL_CWORD_CWORD_CWORD,
2794         /*  63  "?" */ CWORD_CCTL_CCTL_CWORD,
2795         /*  64  "@" */ CWORD_CWORD_CWORD_CWORD,
2796         /*  65  "A" */ CWORD_CWORD_CWORD_CWORD,
2797         /*  66  "B" */ CWORD_CWORD_CWORD_CWORD,
2798         /*  67  "C" */ CWORD_CWORD_CWORD_CWORD,
2799         /*  68  "D" */ CWORD_CWORD_CWORD_CWORD,
2800         /*  69  "E" */ CWORD_CWORD_CWORD_CWORD,
2801         /*  70  "F" */ CWORD_CWORD_CWORD_CWORD,
2802         /*  71  "G" */ CWORD_CWORD_CWORD_CWORD,
2803         /*  72  "H" */ CWORD_CWORD_CWORD_CWORD,
2804         /*  73  "I" */ CWORD_CWORD_CWORD_CWORD,
2805         /*  74  "J" */ CWORD_CWORD_CWORD_CWORD,
2806         /*  75  "K" */ CWORD_CWORD_CWORD_CWORD,
2807         /*  76  "L" */ CWORD_CWORD_CWORD_CWORD,
2808         /*  77  "M" */ CWORD_CWORD_CWORD_CWORD,
2809         /*  78  "N" */ CWORD_CWORD_CWORD_CWORD,
2810         /*  79  "O" */ CWORD_CWORD_CWORD_CWORD,
2811         /*  80  "P" */ CWORD_CWORD_CWORD_CWORD,
2812         /*  81  "Q" */ CWORD_CWORD_CWORD_CWORD,
2813         /*  82  "R" */ CWORD_CWORD_CWORD_CWORD,
2814         /*  83  "S" */ CWORD_CWORD_CWORD_CWORD,
2815         /*  84  "T" */ CWORD_CWORD_CWORD_CWORD,
2816         /*  85  "U" */ CWORD_CWORD_CWORD_CWORD,
2817         /*  86  "V" */ CWORD_CWORD_CWORD_CWORD,
2818         /*  87  "W" */ CWORD_CWORD_CWORD_CWORD,
2819         /*  88  "X" */ CWORD_CWORD_CWORD_CWORD,
2820         /*  89  "Y" */ CWORD_CWORD_CWORD_CWORD,
2821         /*  90  "Z" */ CWORD_CWORD_CWORD_CWORD,
2822         /*  91  "[" */ CWORD_CCTL_CCTL_CWORD,
2823         /*  92  "\" */ CBACK_CBACK_CCTL_CBACK,
2824         /*  93  "]" */ CWORD_CCTL_CCTL_CWORD,
2825         /*  94  "^" */ CWORD_CWORD_CWORD_CWORD,
2826         /*  95  "_" */ CWORD_CWORD_CWORD_CWORD,
2827         /*  96  "`" */ CBQUOTE_CBQUOTE_CWORD_CBQUOTE,
2828         /*  97  "a" */ CWORD_CWORD_CWORD_CWORD,
2829         /*  98  "b" */ CWORD_CWORD_CWORD_CWORD,
2830         /*  99  "c" */ CWORD_CWORD_CWORD_CWORD,
2831         /* 100  "d" */ CWORD_CWORD_CWORD_CWORD,
2832         /* 101  "e" */ CWORD_CWORD_CWORD_CWORD,
2833         /* 102  "f" */ CWORD_CWORD_CWORD_CWORD,
2834         /* 103  "g" */ CWORD_CWORD_CWORD_CWORD,
2835         /* 104  "h" */ CWORD_CWORD_CWORD_CWORD,
2836         /* 105  "i" */ CWORD_CWORD_CWORD_CWORD,
2837         /* 106  "j" */ CWORD_CWORD_CWORD_CWORD,
2838         /* 107  "k" */ CWORD_CWORD_CWORD_CWORD,
2839         /* 108  "l" */ CWORD_CWORD_CWORD_CWORD,
2840         /* 109  "m" */ CWORD_CWORD_CWORD_CWORD,
2841         /* 110  "n" */ CWORD_CWORD_CWORD_CWORD,
2842         /* 111  "o" */ CWORD_CWORD_CWORD_CWORD,
2843         /* 112  "p" */ CWORD_CWORD_CWORD_CWORD,
2844         /* 113  "q" */ CWORD_CWORD_CWORD_CWORD,
2845         /* 114  "r" */ CWORD_CWORD_CWORD_CWORD,
2846         /* 115  "s" */ CWORD_CWORD_CWORD_CWORD,
2847         /* 116  "t" */ CWORD_CWORD_CWORD_CWORD,
2848         /* 117  "u" */ CWORD_CWORD_CWORD_CWORD,
2849         /* 118  "v" */ CWORD_CWORD_CWORD_CWORD,
2850         /* 119  "w" */ CWORD_CWORD_CWORD_CWORD,
2851         /* 120  "x" */ CWORD_CWORD_CWORD_CWORD,
2852         /* 121  "y" */ CWORD_CWORD_CWORD_CWORD,
2853         /* 122  "z" */ CWORD_CWORD_CWORD_CWORD,
2854         /* 123  "{" */ CWORD_CWORD_CWORD_CWORD,
2855         /* 124  "|" */ CSPCL_CWORD_CWORD_CWORD,
2856         /* 125  "}" */ CENDVAR_CENDVAR_CWORD_CENDVAR,
2857         /* 126  "~" */ CWORD_CCTL_CCTL_CWORD,
2858         /* 127  del */ CWORD_CWORD_CWORD_CWORD,
2859         /* 128 0x80 */ CWORD_CWORD_CWORD_CWORD,
2860         /* 129 CTLESC       */ CCTL_CCTL_CCTL_CCTL,
2861         /* 130 CTLVAR       */ CCTL_CCTL_CCTL_CCTL,
2862         /* 131 CTLENDVAR    */ CCTL_CCTL_CCTL_CCTL,
2863         /* 132 CTLBACKQ     */ CCTL_CCTL_CCTL_CCTL,
2864         /* 133 CTLQUOTE     */ CCTL_CCTL_CCTL_CCTL,
2865         /* 134 CTLARI       */ CCTL_CCTL_CCTL_CCTL,
2866         /* 135 CTLENDARI    */ CCTL_CCTL_CCTL_CCTL,
2867         /* 136 CTLQUOTEMARK */ CCTL_CCTL_CCTL_CCTL,
2868         /* 137      */ CWORD_CWORD_CWORD_CWORD,
2869         /* 138      */ CWORD_CWORD_CWORD_CWORD,
2870         /* 139      */ CWORD_CWORD_CWORD_CWORD,
2871         /* 140      */ CWORD_CWORD_CWORD_CWORD,
2872         /* 141      */ CWORD_CWORD_CWORD_CWORD,
2873         /* 142      */ CWORD_CWORD_CWORD_CWORD,
2874         /* 143      */ CWORD_CWORD_CWORD_CWORD,
2875         /* 144      */ CWORD_CWORD_CWORD_CWORD,
2876         /* 145      */ CWORD_CWORD_CWORD_CWORD,
2877         /* 146      */ CWORD_CWORD_CWORD_CWORD,
2878         /* 147      */ CWORD_CWORD_CWORD_CWORD,
2879         /* 148      */ CWORD_CWORD_CWORD_CWORD,
2880         /* 149      */ CWORD_CWORD_CWORD_CWORD,
2881         /* 150      */ CWORD_CWORD_CWORD_CWORD,
2882         /* 151      */ CWORD_CWORD_CWORD_CWORD,
2883         /* 152      */ CWORD_CWORD_CWORD_CWORD,
2884         /* 153      */ CWORD_CWORD_CWORD_CWORD,
2885         /* 154      */ CWORD_CWORD_CWORD_CWORD,
2886         /* 155      */ CWORD_CWORD_CWORD_CWORD,
2887         /* 156      */ CWORD_CWORD_CWORD_CWORD,
2888         /* 157      */ CWORD_CWORD_CWORD_CWORD,
2889         /* 158      */ CWORD_CWORD_CWORD_CWORD,
2890         /* 159      */ CWORD_CWORD_CWORD_CWORD,
2891         /* 160      */ CWORD_CWORD_CWORD_CWORD,
2892         /* 161      */ CWORD_CWORD_CWORD_CWORD,
2893         /* 162      */ CWORD_CWORD_CWORD_CWORD,
2894         /* 163      */ CWORD_CWORD_CWORD_CWORD,
2895         /* 164      */ CWORD_CWORD_CWORD_CWORD,
2896         /* 165      */ CWORD_CWORD_CWORD_CWORD,
2897         /* 166      */ CWORD_CWORD_CWORD_CWORD,
2898         /* 167      */ CWORD_CWORD_CWORD_CWORD,
2899         /* 168      */ CWORD_CWORD_CWORD_CWORD,
2900         /* 169      */ CWORD_CWORD_CWORD_CWORD,
2901         /* 170      */ CWORD_CWORD_CWORD_CWORD,
2902         /* 171      */ CWORD_CWORD_CWORD_CWORD,
2903         /* 172      */ CWORD_CWORD_CWORD_CWORD,
2904         /* 173      */ CWORD_CWORD_CWORD_CWORD,
2905         /* 174      */ CWORD_CWORD_CWORD_CWORD,
2906         /* 175      */ CWORD_CWORD_CWORD_CWORD,
2907         /* 176      */ CWORD_CWORD_CWORD_CWORD,
2908         /* 177      */ CWORD_CWORD_CWORD_CWORD,
2909         /* 178      */ CWORD_CWORD_CWORD_CWORD,
2910         /* 179      */ CWORD_CWORD_CWORD_CWORD,
2911         /* 180      */ CWORD_CWORD_CWORD_CWORD,
2912         /* 181      */ CWORD_CWORD_CWORD_CWORD,
2913         /* 182      */ CWORD_CWORD_CWORD_CWORD,
2914         /* 183      */ CWORD_CWORD_CWORD_CWORD,
2915         /* 184      */ CWORD_CWORD_CWORD_CWORD,
2916         /* 185      */ CWORD_CWORD_CWORD_CWORD,
2917         /* 186      */ CWORD_CWORD_CWORD_CWORD,
2918         /* 187      */ CWORD_CWORD_CWORD_CWORD,
2919         /* 188      */ CWORD_CWORD_CWORD_CWORD,
2920         /* 189      */ CWORD_CWORD_CWORD_CWORD,
2921         /* 190      */ CWORD_CWORD_CWORD_CWORD,
2922         /* 191      */ CWORD_CWORD_CWORD_CWORD,
2923         /* 192      */ CWORD_CWORD_CWORD_CWORD,
2924         /* 193      */ CWORD_CWORD_CWORD_CWORD,
2925         /* 194      */ CWORD_CWORD_CWORD_CWORD,
2926         /* 195      */ CWORD_CWORD_CWORD_CWORD,
2927         /* 196      */ CWORD_CWORD_CWORD_CWORD,
2928         /* 197      */ CWORD_CWORD_CWORD_CWORD,
2929         /* 198      */ CWORD_CWORD_CWORD_CWORD,
2930         /* 199      */ CWORD_CWORD_CWORD_CWORD,
2931         /* 200      */ CWORD_CWORD_CWORD_CWORD,
2932         /* 201      */ CWORD_CWORD_CWORD_CWORD,
2933         /* 202      */ CWORD_CWORD_CWORD_CWORD,
2934         /* 203      */ CWORD_CWORD_CWORD_CWORD,
2935         /* 204      */ CWORD_CWORD_CWORD_CWORD,
2936         /* 205      */ CWORD_CWORD_CWORD_CWORD,
2937         /* 206      */ CWORD_CWORD_CWORD_CWORD,
2938         /* 207      */ CWORD_CWORD_CWORD_CWORD,
2939         /* 208      */ CWORD_CWORD_CWORD_CWORD,
2940         /* 209      */ CWORD_CWORD_CWORD_CWORD,
2941         /* 210      */ CWORD_CWORD_CWORD_CWORD,
2942         /* 211      */ CWORD_CWORD_CWORD_CWORD,
2943         /* 212      */ CWORD_CWORD_CWORD_CWORD,
2944         /* 213      */ CWORD_CWORD_CWORD_CWORD,
2945         /* 214      */ CWORD_CWORD_CWORD_CWORD,
2946         /* 215      */ CWORD_CWORD_CWORD_CWORD,
2947         /* 216      */ CWORD_CWORD_CWORD_CWORD,
2948         /* 217      */ CWORD_CWORD_CWORD_CWORD,
2949         /* 218      */ CWORD_CWORD_CWORD_CWORD,
2950         /* 219      */ CWORD_CWORD_CWORD_CWORD,
2951         /* 220      */ CWORD_CWORD_CWORD_CWORD,
2952         /* 221      */ CWORD_CWORD_CWORD_CWORD,
2953         /* 222      */ CWORD_CWORD_CWORD_CWORD,
2954         /* 223      */ CWORD_CWORD_CWORD_CWORD,
2955         /* 224      */ CWORD_CWORD_CWORD_CWORD,
2956         /* 225      */ CWORD_CWORD_CWORD_CWORD,
2957         /* 226      */ CWORD_CWORD_CWORD_CWORD,
2958         /* 227      */ CWORD_CWORD_CWORD_CWORD,
2959         /* 228      */ CWORD_CWORD_CWORD_CWORD,
2960         /* 229      */ CWORD_CWORD_CWORD_CWORD,
2961         /* 230      */ CWORD_CWORD_CWORD_CWORD,
2962         /* 231      */ CWORD_CWORD_CWORD_CWORD,
2963         /* 232      */ CWORD_CWORD_CWORD_CWORD,
2964         /* 233      */ CWORD_CWORD_CWORD_CWORD,
2965         /* 234      */ CWORD_CWORD_CWORD_CWORD,
2966         /* 235      */ CWORD_CWORD_CWORD_CWORD,
2967         /* 236      */ CWORD_CWORD_CWORD_CWORD,
2968         /* 237      */ CWORD_CWORD_CWORD_CWORD,
2969         /* 238      */ CWORD_CWORD_CWORD_CWORD,
2970         /* 239      */ CWORD_CWORD_CWORD_CWORD,
2971         /* 230      */ CWORD_CWORD_CWORD_CWORD,
2972         /* 241      */ CWORD_CWORD_CWORD_CWORD,
2973         /* 242      */ CWORD_CWORD_CWORD_CWORD,
2974         /* 243      */ CWORD_CWORD_CWORD_CWORD,
2975         /* 244      */ CWORD_CWORD_CWORD_CWORD,
2976         /* 245      */ CWORD_CWORD_CWORD_CWORD,
2977         /* 246      */ CWORD_CWORD_CWORD_CWORD,
2978         /* 247      */ CWORD_CWORD_CWORD_CWORD,
2979         /* 248      */ CWORD_CWORD_CWORD_CWORD,
2980         /* 249      */ CWORD_CWORD_CWORD_CWORD,
2981         /* 250      */ CWORD_CWORD_CWORD_CWORD,
2982         /* 251      */ CWORD_CWORD_CWORD_CWORD,
2983         /* 252      */ CWORD_CWORD_CWORD_CWORD,
2984         /* 253      */ CWORD_CWORD_CWORD_CWORD,
2985         /* 254      */ CWORD_CWORD_CWORD_CWORD,
2986         /* 255      */ CWORD_CWORD_CWORD_CWORD,
2987         /* PEOF */     CENDFILE_CENDFILE_CENDFILE_CENDFILE,
2988 # if ENABLE_ASH_ALIAS
2989         /* PEOA */     CSPCL_CIGN_CIGN_CIGN,
2990 # endif
2991 };
2992
2993 # define SIT(c, syntax) ((S_I_T[syntax_index_table[c]] >> ((syntax)*4)) & 0xf)
2994
2995 #endif  /* !USE_SIT_FUNCTION */
2996
2997
2998 /* ============ Alias handling */
2999
3000 #if ENABLE_ASH_ALIAS
3001
3002 #define ALIASINUSE 1
3003 #define ALIASDEAD  2
3004
3005 struct alias {
3006         struct alias *next;
3007         char *name;
3008         char *val;
3009         int flag;
3010 };
3011
3012
3013 static struct alias **atab; // [ATABSIZE];
3014 #define INIT_G_alias() do { \
3015         atab = xzalloc(ATABSIZE * sizeof(atab[0])); \
3016 } while (0)
3017
3018
3019 static struct alias **
3020 __lookupalias(const char *name) {
3021         unsigned int hashval;
3022         struct alias **app;
3023         const char *p;
3024         unsigned int ch;
3025
3026         p = name;
3027
3028         ch = (unsigned char)*p;
3029         hashval = ch << 4;
3030         while (ch) {
3031                 hashval += ch;
3032                 ch = (unsigned char)*++p;
3033         }
3034         app = &atab[hashval % ATABSIZE];
3035
3036         for (; *app; app = &(*app)->next) {
3037                 if (strcmp(name, (*app)->name) == 0) {
3038                         break;
3039                 }
3040         }
3041
3042         return app;
3043 }
3044
3045 static struct alias *
3046 lookupalias(const char *name, int check)
3047 {
3048         struct alias *ap = *__lookupalias(name);
3049
3050         if (check && ap && (ap->flag & ALIASINUSE))
3051                 return NULL;
3052         return ap;
3053 }
3054
3055 static struct alias *
3056 freealias(struct alias *ap)
3057 {
3058         struct alias *next;
3059
3060         if (ap->flag & ALIASINUSE) {
3061                 ap->flag |= ALIASDEAD;
3062                 return ap;
3063         }
3064
3065         next = ap->next;
3066         free(ap->name);
3067         free(ap->val);
3068         free(ap);
3069         return next;
3070 }
3071
3072 static void
3073 setalias(const char *name, const char *val)
3074 {
3075         struct alias *ap, **app;
3076
3077         app = __lookupalias(name);
3078         ap = *app;
3079         INT_OFF;
3080         if (ap) {
3081                 if (!(ap->flag & ALIASINUSE)) {
3082                         free(ap->val);
3083                 }
3084                 ap->val = ckstrdup(val);
3085                 ap->flag &= ~ALIASDEAD;
3086         } else {
3087                 /* not found */
3088                 ap = ckzalloc(sizeof(struct alias));
3089                 ap->name = ckstrdup(name);
3090                 ap->val = ckstrdup(val);
3091                 /*ap->flag = 0; - ckzalloc did it */
3092                 /*ap->next = NULL;*/
3093                 *app = ap;
3094         }
3095         INT_ON;
3096 }
3097
3098 static int
3099 unalias(const char *name)
3100 {
3101         struct alias **app;
3102
3103         app = __lookupalias(name);
3104
3105         if (*app) {
3106                 INT_OFF;
3107                 *app = freealias(*app);
3108                 INT_ON;
3109                 return 0;
3110         }
3111
3112         return 1;
3113 }
3114
3115 static void
3116 rmaliases(void)
3117 {
3118         struct alias *ap, **app;
3119         int i;
3120
3121         INT_OFF;
3122         for (i = 0; i < ATABSIZE; i++) {
3123                 app = &atab[i];
3124                 for (ap = *app; ap; ap = *app) {
3125                         *app = freealias(*app);
3126                         if (ap == *app) {
3127                                 app = &ap->next;
3128                         }
3129                 }
3130         }
3131         INT_ON;
3132 }
3133
3134 static void
3135 printalias(const struct alias *ap)
3136 {
3137         out1fmt("%s=%s\n", ap->name, single_quote(ap->val));
3138 }
3139
3140 /*
3141  * TODO - sort output
3142  */
3143 static int FAST_FUNC
3144 aliascmd(int argc UNUSED_PARAM, char **argv)
3145 {
3146         char *n, *v;
3147         int ret = 0;
3148         struct alias *ap;
3149
3150         if (!argv[1]) {
3151                 int i;
3152
3153                 for (i = 0; i < ATABSIZE; i++) {
3154                         for (ap = atab[i]; ap; ap = ap->next) {
3155                                 printalias(ap);
3156                         }
3157                 }
3158                 return 0;
3159         }
3160         while ((n = *++argv) != NULL) {
3161                 v = strchr(n+1, '=');
3162                 if (v == NULL) { /* n+1: funny ksh stuff */
3163                         ap = *__lookupalias(n);
3164                         if (ap == NULL) {
3165                                 fprintf(stderr, "%s: %s not found\n", "alias", n);
3166                                 ret = 1;
3167                         } else
3168                                 printalias(ap);
3169                 } else {
3170                         *v++ = '\0';
3171                         setalias(n, v);
3172                 }
3173         }
3174
3175         return ret;
3176 }
3177
3178 static int FAST_FUNC
3179 unaliascmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
3180 {
3181         int i;
3182
3183         while ((i = nextopt("a")) != '\0') {
3184                 if (i == 'a') {
3185                         rmaliases();
3186                         return 0;
3187                 }
3188         }
3189         for (i = 0; *argptr; argptr++) {
3190                 if (unalias(*argptr)) {
3191                         fprintf(stderr, "%s: %s not found\n", "unalias", *argptr);
3192                         i = 1;
3193                 }
3194         }
3195
3196         return i;
3197 }
3198
3199 #endif /* ASH_ALIAS */
3200
3201
3202 /* ============ jobs.c */
3203
3204 /* Mode argument to forkshell.  Don't change FORK_FG or FORK_BG. */
3205 #define FORK_FG    0
3206 #define FORK_BG    1
3207 #define FORK_NOJOB 2
3208
3209 /* mode flags for showjob(s) */
3210 #define SHOW_ONLY_PGID  0x01    /* show only pgid (jobs -p) */
3211 #define SHOW_PIDS       0x02    /* show individual pids, not just one line per job */
3212 #define SHOW_CHANGED    0x04    /* only jobs whose state has changed */
3213
3214 /*
3215  * A job structure contains information about a job.  A job is either a
3216  * single process or a set of processes contained in a pipeline.  In the
3217  * latter case, pidlist will be non-NULL, and will point to a -1 terminated
3218  * array of pids.
3219  */
3220 struct procstat {
3221         pid_t   ps_pid;         /* process id */
3222         int     ps_status;      /* last process status from wait() */
3223         char    *ps_cmd;        /* text of command being run */
3224 };
3225
3226 struct job {
3227         struct procstat ps0;    /* status of process */
3228         struct procstat *ps;    /* status or processes when more than one */
3229 #if JOBS
3230         int stopstatus;         /* status of a stopped job */
3231 #endif
3232         uint32_t
3233                 nprocs: 16,     /* number of processes */
3234                 state: 8,
3235 #define JOBRUNNING      0       /* at least one proc running */
3236 #define JOBSTOPPED      1       /* all procs are stopped */
3237 #define JOBDONE         2       /* all procs are completed */
3238 #if JOBS
3239                 sigint: 1,      /* job was killed by SIGINT */
3240                 jobctl: 1,      /* job running under job control */
3241 #endif
3242                 waited: 1,      /* true if this entry has been waited for */
3243                 used: 1,        /* true if this entry is in used */
3244                 changed: 1;     /* true if status has changed */
3245         struct job *prev_job;   /* previous job */
3246 };
3247
3248 static struct job *makejob(/*union node *,*/ int);
3249 static int forkshell(struct job *, union node *, int);
3250 static int waitforjob(struct job *);
3251
3252 #if !JOBS
3253 enum { doing_jobctl = 0 };
3254 #define setjobctl(on) do {} while (0)
3255 #else
3256 static smallint doing_jobctl; //references:8
3257 static void setjobctl(int);
3258 #endif
3259
3260 /*
3261  * Ignore a signal.
3262  */
3263 static void
3264 ignoresig(int signo)
3265 {
3266         /* Avoid unnecessary system calls. Is it already SIG_IGNed? */
3267         if (sigmode[signo - 1] != S_IGN && sigmode[signo - 1] != S_HARD_IGN) {
3268                 /* No, need to do it */
3269                 signal(signo, SIG_IGN);
3270         }
3271         sigmode[signo - 1] = S_HARD_IGN;
3272 }
3273
3274 /*
3275  * Signal handler. Only one usage site - in setsignal()
3276  */
3277 static void
3278 onsig(int signo)
3279 {
3280         gotsig[signo - 1] = 1;
3281
3282         if (signo == SIGINT && !trap[SIGINT]) {
3283                 if (!suppress_int) {
3284                         pending_sig = 0;
3285                         raise_interrupt(); /* does not return */
3286                 }
3287                 pending_int = 1;
3288         } else {
3289                 pending_sig = signo;
3290         }
3291 }
3292
3293 /*
3294  * Set the signal handler for the specified signal.  The routine figures
3295  * out what it should be set to.
3296  */
3297 static void
3298 setsignal(int signo)
3299 {
3300         char *t;
3301         char cur_act, new_act;
3302         struct sigaction act;
3303
3304         t = trap[signo];
3305         new_act = S_DFL;
3306         if (t != NULL) { /* trap for this sig is set */
3307                 new_act = S_CATCH;
3308                 if (t[0] == '\0') /* trap is "": ignore this sig */
3309                         new_act = S_IGN;
3310         }
3311
3312         if (rootshell && new_act == S_DFL) {
3313                 switch (signo) {
3314                 case SIGINT:
3315                         if (iflag || minusc || sflag == 0)
3316                                 new_act = S_CATCH;
3317                         break;
3318                 case SIGQUIT:
3319 #if DEBUG
3320                         if (debug)
3321                                 break;
3322 #endif
3323                         /* man bash:
3324                          * "In all cases, bash ignores SIGQUIT. Non-builtin
3325                          * commands run by bash have signal handlers
3326                          * set to the values inherited by the shell
3327                          * from its parent". */
3328                         new_act = S_IGN;
3329                         break;
3330                 case SIGTERM:
3331                         if (iflag)
3332                                 new_act = S_IGN;
3333                         break;
3334 #if JOBS
3335                 case SIGTSTP:
3336                 case SIGTTOU:
3337                         if (mflag)
3338                                 new_act = S_IGN;
3339                         break;
3340 #endif
3341                 }
3342         }
3343 //TODO: if !rootshell, we reset SIGQUIT to DFL,
3344 //whereas we have to restore it to what shell got on entry
3345 //from the parent. See comment above
3346
3347         t = &sigmode[signo - 1];
3348         cur_act = *t;
3349         if (cur_act == 0) {
3350                 /* current setting is not yet known */
3351                 if (sigaction(signo, NULL, &act)) {
3352                         /* pretend it worked; maybe we should give a warning,
3353                          * but other shells don't. We don't alter sigmode,
3354                          * so we retry every time.
3355                          * btw, in Linux it never fails. --vda */
3356                         return;
3357                 }
3358                 if (act.sa_handler == SIG_IGN) {
3359                         cur_act = S_HARD_IGN;
3360                         if (mflag
3361                          && (signo == SIGTSTP || signo == SIGTTIN || signo == SIGTTOU)
3362                         ) {
3363                                 cur_act = S_IGN;   /* don't hard ignore these */
3364                         }
3365                 }
3366         }
3367         if (cur_act == S_HARD_IGN || cur_act == new_act)
3368                 return;
3369
3370         act.sa_handler = SIG_DFL;
3371         switch (new_act) {
3372         case S_CATCH:
3373                 act.sa_handler = onsig;
3374                 act.sa_flags = 0; /* matters only if !DFL and !IGN */
3375                 sigfillset(&act.sa_mask); /* ditto */
3376                 break;
3377         case S_IGN:
3378                 act.sa_handler = SIG_IGN;
3379                 break;
3380         }
3381         sigaction_set(signo, &act);
3382
3383         *t = new_act;
3384 }
3385
3386 /* mode flags for set_curjob */
3387 #define CUR_DELETE 2
3388 #define CUR_RUNNING 1
3389 #define CUR_STOPPED 0
3390
3391 /* mode flags for dowait */
3392 #define DOWAIT_NONBLOCK WNOHANG
3393 #define DOWAIT_BLOCK    0
3394
3395 #if JOBS
3396 /* pgrp of shell on invocation */
3397 static int initialpgrp; //references:2
3398 static int ttyfd = -1; //5
3399 #endif
3400 /* array of jobs */
3401 static struct job *jobtab; //5
3402 /* size of array */
3403 static unsigned njobs; //4
3404 /* current job */
3405 static struct job *curjob; //lots
3406 /* number of presumed living untracked jobs */
3407 static int jobless; //4
3408
3409 static void
3410 set_curjob(struct job *jp, unsigned mode)
3411 {
3412         struct job *jp1;
3413         struct job **jpp, **curp;
3414
3415         /* first remove from list */
3416         jpp = curp = &curjob;
3417         do {
3418                 jp1 = *jpp;
3419                 if (jp1 == jp)
3420                         break;
3421                 jpp = &jp1->prev_job;
3422         } while (1);
3423         *jpp = jp1->prev_job;
3424
3425         /* Then re-insert in correct position */
3426         jpp = curp;
3427         switch (mode) {
3428         default:
3429 #if DEBUG
3430                 abort();
3431 #endif
3432         case CUR_DELETE:
3433                 /* job being deleted */
3434                 break;
3435         case CUR_RUNNING:
3436                 /* newly created job or backgrounded job,
3437                    put after all stopped jobs. */
3438                 do {
3439                         jp1 = *jpp;
3440 #if JOBS
3441                         if (!jp1 || jp1->state != JOBSTOPPED)
3442 #endif
3443                                 break;
3444                         jpp = &jp1->prev_job;
3445                 } while (1);
3446                 /* FALLTHROUGH */
3447 #if JOBS
3448         case CUR_STOPPED:
3449 #endif
3450                 /* newly stopped job - becomes curjob */
3451                 jp->prev_job = *jpp;
3452                 *jpp = jp;
3453                 break;
3454         }
3455 }
3456
3457 #if JOBS || DEBUG
3458 static int
3459 jobno(const struct job *jp)
3460 {
3461         return jp - jobtab + 1;
3462 }
3463 #endif
3464
3465 /*
3466  * Convert a job name to a job structure.
3467  */
3468 #if !JOBS
3469 #define getjob(name, getctl) getjob(name)
3470 #endif
3471 static struct job *
3472 getjob(const char *name, int getctl)
3473 {
3474         struct job *jp;
3475         struct job *found;
3476         const char *err_msg = "%s: no such job";
3477         unsigned num;
3478         int c;
3479         const char *p;
3480         char *(*match)(const char *, const char *);
3481
3482         jp = curjob;
3483         p = name;
3484         if (!p)
3485                 goto currentjob;
3486
3487         if (*p != '%')
3488                 goto err;
3489
3490         c = *++p;
3491         if (!c)
3492                 goto currentjob;
3493
3494         if (!p[1]) {
3495                 if (c == '+' || c == '%') {
3496  currentjob:
3497                         err_msg = "No current job";
3498                         goto check;
3499                 }
3500                 if (c == '-') {
3501                         if (jp)
3502                                 jp = jp->prev_job;
3503                         err_msg = "No previous job";
3504  check:
3505                         if (!jp)
3506                                 goto err;
3507                         goto gotit;
3508                 }
3509         }
3510
3511         if (is_number(p)) {
3512                 num = atoi(p);
3513                 if (num < njobs) {
3514                         jp = jobtab + num - 1;
3515                         if (jp->used)
3516                                 goto gotit;
3517                         goto err;
3518                 }
3519         }
3520
3521         match = prefix;
3522         if (*p == '?') {
3523                 match = strstr;
3524                 p++;
3525         }
3526
3527         found = NULL;
3528         while (jp) {
3529                 if (match(jp->ps[0].ps_cmd, p)) {
3530                         if (found)
3531                                 goto err;
3532                         found = jp;
3533                         err_msg = "%s: ambiguous";
3534                 }
3535                 jp = jp->prev_job;
3536         }
3537         if (!found)
3538                 goto err;
3539         jp = found;
3540
3541  gotit:
3542 #if JOBS
3543         err_msg = "job %s not created under job control";
3544         if (getctl && jp->jobctl == 0)
3545                 goto err;
3546 #endif
3547         return jp;
3548  err:
3549         ash_msg_and_raise_error(err_msg, name);
3550 }
3551
3552 /*
3553  * Mark a job structure as unused.
3554  */
3555 static void
3556 freejob(struct job *jp)
3557 {
3558         struct procstat *ps;
3559         int i;
3560
3561         INT_OFF;
3562         for (i = jp->nprocs, ps = jp->ps; --i >= 0; ps++) {
3563                 if (ps->ps_cmd != nullstr)
3564                         free(ps->ps_cmd);
3565         }
3566         if (jp->ps != &jp->ps0)
3567                 free(jp->ps);
3568         jp->used = 0;
3569         set_curjob(jp, CUR_DELETE);
3570         INT_ON;
3571 }
3572
3573 #if JOBS
3574 static void
3575 xtcsetpgrp(int fd, pid_t pgrp)
3576 {
3577         if (tcsetpgrp(fd, pgrp))
3578                 ash_msg_and_raise_error("can't set tty process group (%m)");
3579 }
3580
3581 /*
3582  * Turn job control on and off.
3583  *
3584  * Note:  This code assumes that the third arg to ioctl is a character
3585  * pointer, which is true on Berkeley systems but not System V.  Since
3586  * System V doesn't have job control yet, this isn't a problem now.
3587  *
3588  * Called with interrupts off.
3589  */
3590 static void
3591 setjobctl(int on)
3592 {
3593         int fd;
3594         int pgrp;
3595
3596         if (on == doing_jobctl || rootshell == 0)
3597                 return;
3598         if (on) {
3599                 int ofd;
3600                 ofd = fd = open(_PATH_TTY, O_RDWR);
3601                 if (fd < 0) {
3602         /* BTW, bash will try to open(ttyname(0)) if open("/dev/tty") fails.
3603          * That sometimes helps to acquire controlling tty.
3604          * Obviously, a workaround for bugs when someone
3605          * failed to provide a controlling tty to bash! :) */
3606                         fd = 2;
3607                         while (!isatty(fd))
3608                                 if (--fd < 0)
3609                                         goto out;
3610                 }
3611                 fd = fcntl(fd, F_DUPFD, 10);
3612                 if (ofd >= 0)
3613                         close(ofd);
3614                 if (fd < 0)
3615                         goto out;
3616                 /* fd is a tty at this point */
3617                 close_on_exec_on(fd);
3618                 do { /* while we are in the background */
3619                         pgrp = tcgetpgrp(fd);
3620                         if (pgrp < 0) {
3621  out:
3622                                 ash_msg("can't access tty; job control turned off");
3623                                 mflag = on = 0;
3624                                 goto close;
3625                         }
3626                         if (pgrp == getpgrp())
3627                                 break;
3628                         killpg(0, SIGTTIN);
3629                 } while (1);
3630                 initialpgrp = pgrp;
3631
3632                 setsignal(SIGTSTP);
3633                 setsignal(SIGTTOU);
3634                 setsignal(SIGTTIN);
3635                 pgrp = rootpid;
3636                 setpgid(0, pgrp);
3637                 xtcsetpgrp(fd, pgrp);
3638         } else {
3639                 /* turning job control off */
3640                 fd = ttyfd;
3641                 pgrp = initialpgrp;
3642                 /* was xtcsetpgrp, but this can make exiting ash
3643                  * loop forever if pty is already deleted */
3644                 tcsetpgrp(fd, pgrp);
3645                 setpgid(0, pgrp);
3646                 setsignal(SIGTSTP);
3647                 setsignal(SIGTTOU);
3648                 setsignal(SIGTTIN);
3649  close:
3650                 if (fd >= 0)
3651                         close(fd);
3652                 fd = -1;
3653         }
3654         ttyfd = fd;
3655         doing_jobctl = on;
3656 }
3657
3658 static int FAST_FUNC
3659 killcmd(int argc, char **argv)
3660 {
3661         int i = 1;
3662         if (argv[1] && strcmp(argv[1], "-l") != 0) {
3663                 do {
3664                         if (argv[i][0] == '%') {
3665                                 struct job *jp = getjob(argv[i], 0);
3666                                 unsigned pid = jp->ps[0].ps_pid;
3667                                 /* Enough space for ' -NNN<nul>' */
3668                                 argv[i] = alloca(sizeof(int)*3 + 3);
3669                                 /* kill_main has matching code to expect
3670                                  * leading space. Needed to not confuse
3671                                  * negative pids with "kill -SIGNAL_NO" syntax */
3672                                 sprintf(argv[i], " -%u", pid);
3673                         }
3674                 } while (argv[++i]);
3675         }
3676         return kill_main(argc, argv);
3677 }
3678
3679 static void
3680 showpipe(struct job *jp /*, FILE *out*/)
3681 {
3682         struct procstat *ps;
3683         struct procstat *psend;
3684
3685         psend = jp->ps + jp->nprocs;
3686         for (ps = jp->ps + 1; ps < psend; ps++)
3687                 printf(" | %s", ps->ps_cmd);
3688         outcslow('\n', stdout);
3689         flush_stdout_stderr();
3690 }
3691
3692
3693 static int
3694 restartjob(struct job *jp, int mode)
3695 {
3696         struct procstat *ps;
3697         int i;
3698         int status;
3699         pid_t pgid;
3700
3701         INT_OFF;
3702         if (jp->state == JOBDONE)
3703                 goto out;
3704         jp->state = JOBRUNNING;
3705         pgid = jp->ps[0].ps_pid;
3706         if (mode == FORK_FG)
3707                 xtcsetpgrp(ttyfd, pgid);
3708         killpg(pgid, SIGCONT);
3709         ps = jp->ps;
3710         i = jp->nprocs;
3711         do {
3712                 if (WIFSTOPPED(ps->ps_status)) {
3713                         ps->ps_status = -1;
3714                 }
3715                 ps++;
3716         } while (--i);
3717  out:
3718         status = (mode == FORK_FG) ? waitforjob(jp) : 0;
3719         INT_ON;
3720         return status;
3721 }
3722
3723 static int FAST_FUNC
3724 fg_bgcmd(int argc UNUSED_PARAM, char **argv)
3725 {
3726         struct job *jp;
3727         int mode;
3728         int retval;
3729
3730         mode = (**argv == 'f') ? FORK_FG : FORK_BG;
3731         nextopt(nullstr);
3732         argv = argptr;
3733         do {
3734                 jp = getjob(*argv, 1);
3735                 if (mode == FORK_BG) {
3736                         set_curjob(jp, CUR_RUNNING);
3737                         printf("[%d] ", jobno(jp));
3738                 }
3739                 out1str(jp->ps[0].ps_cmd);
3740                 showpipe(jp /*, stdout*/);
3741                 retval = restartjob(jp, mode);
3742         } while (*argv && *++argv);
3743         return retval;
3744 }
3745 #endif
3746
3747 static int
3748 sprint_status(char *s, int status, int sigonly)
3749 {
3750         int col;
3751         int st;
3752
3753         col = 0;
3754         if (!WIFEXITED(status)) {
3755 #if JOBS
3756                 if (WIFSTOPPED(status))
3757                         st = WSTOPSIG(status);
3758                 else
3759 #endif
3760                         st = WTERMSIG(status);
3761                 if (sigonly) {
3762                         if (st == SIGINT || st == SIGPIPE)
3763                                 goto out;
3764 #if JOBS
3765                         if (WIFSTOPPED(status))
3766                                 goto out;
3767 #endif
3768                 }
3769                 st &= 0x7f;
3770                 col = fmtstr(s, 32, strsignal(st));
3771                 if (WCOREDUMP(status)) {
3772                         col += fmtstr(s + col, 16, " (core dumped)");
3773                 }
3774         } else if (!sigonly) {
3775                 st = WEXITSTATUS(status);
3776                 if (st)
3777                         col = fmtstr(s, 16, "Done(%d)", st);
3778                 else
3779                         col = fmtstr(s, 16, "Done");
3780         }
3781  out:
3782         return col;
3783 }
3784
3785 static int
3786 dowait(int wait_flags, struct job *job)
3787 {
3788         int pid;
3789         int status;
3790         struct job *jp;
3791         struct job *thisjob;
3792         int state;
3793
3794         TRACE(("dowait(0x%x) called\n", wait_flags));
3795
3796         /* Do a wait system call. If job control is compiled in, we accept
3797          * stopped processes. wait_flags may have WNOHANG, preventing blocking.
3798          * NB: _not_ safe_waitpid, we need to detect EINTR */
3799         if (doing_jobctl)
3800                 wait_flags |= WUNTRACED;
3801         pid = waitpid(-1, &status, wait_flags);
3802         TRACE(("wait returns pid=%d, status=0x%x, errno=%d(%s)\n",
3803                                 pid, status, errno, strerror(errno)));
3804         if (pid <= 0)
3805                 return pid;
3806
3807         INT_OFF;
3808         thisjob = NULL;
3809         for (jp = curjob; jp; jp = jp->prev_job) {
3810                 struct procstat *ps;
3811                 struct procstat *psend;
3812                 if (jp->state == JOBDONE)
3813                         continue;
3814                 state = JOBDONE;
3815                 ps = jp->ps;
3816                 psend = ps + jp->nprocs;
3817                 do {
3818                         if (ps->ps_pid == pid) {
3819                                 TRACE(("Job %d: changing status of proc %d "
3820                                         "from 0x%x to 0x%x\n",
3821                                         jobno(jp), pid, ps->ps_status, status));
3822                                 ps->ps_status = status;
3823                                 thisjob = jp;
3824                         }
3825                         if (ps->ps_status == -1)
3826                                 state = JOBRUNNING;
3827 #if JOBS
3828                         if (state == JOBRUNNING)
3829                                 continue;
3830                         if (WIFSTOPPED(ps->ps_status)) {
3831                                 jp->stopstatus = ps->ps_status;
3832                                 state = JOBSTOPPED;
3833                         }
3834 #endif
3835                 } while (++ps < psend);
3836                 if (thisjob)
3837                         goto gotjob;
3838         }
3839 #if JOBS
3840         if (!WIFSTOPPED(status))
3841 #endif
3842                 jobless--;
3843         goto out;
3844
3845  gotjob:
3846         if (state != JOBRUNNING) {
3847                 thisjob->changed = 1;
3848
3849                 if (thisjob->state != state) {
3850                         TRACE(("Job %d: changing state from %d to %d\n",
3851                                 jobno(thisjob), thisjob->state, state));
3852                         thisjob->state = state;
3853 #if JOBS
3854                         if (state == JOBSTOPPED) {
3855                                 set_curjob(thisjob, CUR_STOPPED);
3856                         }
3857 #endif
3858                 }
3859         }
3860
3861  out:
3862         INT_ON;
3863
3864         if (thisjob && thisjob == job) {
3865                 char s[48 + 1];
3866                 int len;
3867
3868                 len = sprint_status(s, status, 1);
3869                 if (len) {
3870                         s[len] = '\n';
3871                         s[len + 1] = '\0';
3872                         out2str(s);
3873                 }
3874         }
3875         return pid;
3876 }
3877
3878 static int
3879 blocking_wait_with_raise_on_sig(struct job *job)
3880 {
3881         pid_t pid = dowait(DOWAIT_BLOCK, job);
3882         if (pid <= 0 && pending_sig)
3883                 raise_exception(EXSIG);
3884         return pid;
3885 }
3886
3887 #if JOBS
3888 static void
3889 showjob(FILE *out, struct job *jp, int mode)
3890 {
3891         struct procstat *ps;
3892         struct procstat *psend;
3893         int col;
3894         int indent_col;
3895         char s[80];
3896
3897         ps = jp->ps;
3898
3899         if (mode & SHOW_ONLY_PGID) { /* jobs -p */
3900                 /* just output process (group) id of pipeline */
3901                 fprintf(out, "%d\n", ps->ps_pid);
3902                 return;
3903         }
3904
3905         col = fmtstr(s, 16, "[%d]   ", jobno(jp));
3906         indent_col = col;
3907
3908         if (jp == curjob)
3909                 s[col - 3] = '+';
3910         else if (curjob && jp == curjob->prev_job)
3911                 s[col - 3] = '-';
3912
3913         if (mode & SHOW_PIDS)
3914                 col += fmtstr(s + col, 16, "%d ", ps->ps_pid);
3915
3916         psend = ps + jp->nprocs;
3917
3918         if (jp->state == JOBRUNNING) {
3919                 strcpy(s + col, "Running");
3920                 col += sizeof("Running") - 1;
3921         } else {
3922                 int status = psend[-1].ps_status;
3923                 if (jp->state == JOBSTOPPED)
3924                         status = jp->stopstatus;
3925                 col += sprint_status(s + col, status, 0);
3926         }
3927         /* By now, "[JOBID]*  [maybe PID] STATUS" is printed */
3928
3929         /* This loop either prints "<cmd1> | <cmd2> | <cmd3>" line
3930          * or prints several "PID             | <cmdN>" lines,
3931          * depending on SHOW_PIDS bit.
3932          * We do not print status of individual processes
3933          * between PID and <cmdN>. bash does it, but not very well:
3934          * first line shows overall job status, not process status,
3935          * making it impossible to know 1st process status.
3936          */
3937         goto start;
3938         do {
3939                 /* for each process */
3940                 s[0] = '\0';
3941                 col = 33;
3942                 if (mode & SHOW_PIDS)
3943                         col = fmtstr(s, 48, "\n%*c%d ", indent_col, ' ', ps->ps_pid) - 1;
3944  start:
3945                 fprintf(out, "%s%*c%s%s",
3946                                 s,
3947                                 33 - col >= 0 ? 33 - col : 0, ' ',
3948                                 ps == jp->ps ? "" : "| ",
3949                                 ps->ps_cmd
3950                 );
3951         } while (++ps != psend);
3952         outcslow('\n', out);
3953
3954         jp->changed = 0;
3955
3956         if (jp->state == JOBDONE) {
3957                 TRACE(("showjob: freeing job %d\n", jobno(jp)));
3958                 freejob(jp);
3959         }
3960 }
3961
3962 /*
3963  * Print a list of jobs.  If "change" is nonzero, only print jobs whose
3964  * statuses have changed since the last call to showjobs.
3965  */
3966 static void
3967 showjobs(FILE *out, int mode)
3968 {
3969         struct job *jp;
3970
3971         TRACE(("showjobs(0x%x) called\n", mode));
3972
3973         /* Handle all finished jobs */
3974         while (dowait(DOWAIT_NONBLOCK, NULL) > 0)
3975                 continue;
3976
3977         for (jp = curjob; jp; jp = jp->prev_job) {
3978                 if (!(mode & SHOW_CHANGED) || jp->changed) {
3979                         showjob(out, jp, mode);
3980                 }
3981         }
3982 }
3983
3984 static int FAST_FUNC
3985 jobscmd(int argc UNUSED_PARAM, char **argv)
3986 {
3987         int mode, m;
3988
3989         mode = 0;
3990         while ((m = nextopt("lp")) != '\0') {
3991                 if (m == 'l')
3992                         mode |= SHOW_PIDS;
3993                 else
3994                         mode |= SHOW_ONLY_PGID;
3995         }
3996
3997         argv = argptr;
3998         if (*argv) {
3999                 do
4000                         showjob(stdout, getjob(*argv, 0), mode);
4001                 while (*++argv);
4002         } else {
4003                 showjobs(stdout, mode);
4004         }
4005
4006         return 0;
4007 }
4008 #endif /* JOBS */
4009
4010 /* Called only on finished or stopped jobs (no members are running) */
4011 static int
4012 getstatus(struct job *job)
4013 {
4014         int status;
4015         int retval;
4016         struct procstat *ps;
4017
4018         /* Fetch last member's status */
4019         ps = job->ps + job->nprocs - 1;
4020         status = ps->ps_status;
4021         if (pipefail) {
4022                 /* "set -o pipefail" mode: use last _nonzero_ status */
4023                 while (status == 0 && --ps >= job->ps)
4024                         status = ps->ps_status;
4025         }
4026
4027         retval = WEXITSTATUS(status);
4028         if (!WIFEXITED(status)) {
4029 #if JOBS
4030                 retval = WSTOPSIG(status);
4031                 if (!WIFSTOPPED(status))
4032 #endif
4033                 {
4034                         /* XXX: limits number of signals */
4035                         retval = WTERMSIG(status);
4036 #if JOBS
4037                         if (retval == SIGINT)
4038                                 job->sigint = 1;
4039 #endif
4040                 }
4041                 retval += 128;
4042         }
4043         TRACE(("getstatus: job %d, nproc %d, status 0x%x, retval 0x%x\n",
4044                 jobno(job), job->nprocs, status, retval));
4045         return retval;
4046 }
4047
4048 static int FAST_FUNC
4049 waitcmd(int argc UNUSED_PARAM, char **argv)
4050 {
4051         struct job *job;
4052         int retval;
4053         struct job *jp;
4054
4055         if (pending_sig)
4056                 raise_exception(EXSIG);
4057
4058         nextopt(nullstr);
4059         retval = 0;
4060
4061         argv = argptr;
4062         if (!*argv) {
4063                 /* wait for all jobs */
4064                 for (;;) {
4065                         jp = curjob;
4066                         while (1) {
4067                                 if (!jp) /* no running procs */
4068                                         goto ret;
4069                                 if (jp->state == JOBRUNNING)
4070                                         break;
4071                                 jp->waited = 1;
4072                                 jp = jp->prev_job;
4073                         }
4074         /* man bash:
4075          * "When bash is waiting for an asynchronous command via
4076          * the wait builtin, the reception of a signal for which a trap
4077          * has been set will cause the wait builtin to return immediately
4078          * with an exit status greater than 128, immediately after which
4079          * the trap is executed."
4080          * Do we do it that way? */
4081                         blocking_wait_with_raise_on_sig(NULL);
4082                 }
4083         }
4084
4085         retval = 127;
4086         do {
4087                 if (**argv != '%') {
4088                         pid_t pid = number(*argv);
4089                         job = curjob;
4090                         while (1) {
4091                                 if (!job)
4092                                         goto repeat;
4093                                 if (job->ps[job->nprocs - 1].ps_pid == pid)
4094                                         break;
4095                                 job = job->prev_job;
4096                         }
4097                 } else
4098                         job = getjob(*argv, 0);
4099                 /* loop until process terminated or stopped */
4100                 while (job->state == JOBRUNNING)
4101                         blocking_wait_with_raise_on_sig(NULL);
4102                 job->waited = 1;
4103                 retval = getstatus(job);
4104  repeat: ;
4105         } while (*++argv);
4106
4107  ret:
4108         return retval;
4109 }
4110
4111 static struct job *
4112 growjobtab(void)
4113 {
4114         size_t len;
4115         ptrdiff_t offset;
4116         struct job *jp, *jq;
4117
4118         len = njobs * sizeof(*jp);
4119         jq = jobtab;
4120         jp = ckrealloc(jq, len + 4 * sizeof(*jp));
4121
4122         offset = (char *)jp - (char *)jq;
4123         if (offset) {
4124                 /* Relocate pointers */
4125                 size_t l = len;
4126
4127                 jq = (struct job *)((char *)jq + l);
4128                 while (l) {
4129                         l -= sizeof(*jp);
4130                         jq--;
4131 #define joff(p) ((struct job *)((char *)(p) + l))
4132 #define jmove(p) (p) = (void *)((char *)(p) + offset)
4133                         if (joff(jp)->ps == &jq->ps0)
4134                                 jmove(joff(jp)->ps);
4135                         if (joff(jp)->prev_job)
4136                                 jmove(joff(jp)->prev_job);
4137                 }
4138                 if (curjob)
4139                         jmove(curjob);
4140 #undef joff
4141 #undef jmove
4142         }
4143
4144         njobs += 4;
4145         jobtab = jp;
4146         jp = (struct job *)((char *)jp + len);
4147         jq = jp + 3;
4148         do {
4149                 jq->used = 0;
4150         } while (--jq >= jp);
4151         return jp;
4152 }
4153
4154 /*
4155  * Return a new job structure.
4156  * Called with interrupts off.
4157  */
4158 static struct job *
4159 makejob(/*union node *node,*/ int nprocs)
4160 {
4161         int i;
4162         struct job *jp;
4163
4164         for (i = njobs, jp = jobtab; ; jp++) {
4165                 if (--i < 0) {
4166                         jp = growjobtab();
4167                         break;
4168                 }
4169                 if (jp->used == 0)
4170                         break;
4171                 if (jp->state != JOBDONE || !jp->waited)
4172                         continue;
4173 #if JOBS
4174                 if (doing_jobctl)
4175                         continue;
4176 #endif
4177                 freejob(jp);
4178                 break;
4179         }
4180         memset(jp, 0, sizeof(*jp));
4181 #if JOBS
4182         /* jp->jobctl is a bitfield.
4183          * "jp->jobctl |= jobctl" likely to give awful code */
4184         if (doing_jobctl)
4185                 jp->jobctl = 1;
4186 #endif
4187         jp->prev_job = curjob;
4188         curjob = jp;
4189         jp->used = 1;
4190         jp->ps = &jp->ps0;
4191         if (nprocs > 1) {
4192                 jp->ps = ckmalloc(nprocs * sizeof(struct procstat));
4193         }
4194         TRACE(("makejob(%d) returns %%%d\n", nprocs,
4195                                 jobno(jp)));
4196         return jp;
4197 }
4198
4199 #if JOBS
4200 /*
4201  * Return a string identifying a command (to be printed by the
4202  * jobs command).
4203  */
4204 static char *cmdnextc;
4205
4206 static void
4207 cmdputs(const char *s)
4208 {
4209         static const char vstype[VSTYPE + 1][3] = {
4210                 "", "}", "-", "+", "?", "=",
4211                 "%", "%%", "#", "##"
4212                 IF_ASH_BASH_COMPAT(, ":", "/", "//")
4213         };
4214
4215         const char *p, *str;
4216         char cc[2];
4217         char *nextc;
4218         unsigned char c;
4219         unsigned char subtype = 0;
4220         int quoted = 0;
4221
4222         cc[1] = '\0';
4223         nextc = makestrspace((strlen(s) + 1) * 8, cmdnextc);
4224         p = s;
4225         while ((c = *p++) != '\0') {
4226                 str = NULL;
4227                 switch (c) {
4228                 case CTLESC:
4229                         c = *p++;
4230                         break;
4231                 case CTLVAR:
4232                         subtype = *p++;
4233                         if ((subtype & VSTYPE) == VSLENGTH)
4234                                 str = "${#";
4235                         else
4236                                 str = "${";
4237                         if (!(subtype & VSQUOTE) == !(quoted & 1))
4238                                 goto dostr;
4239                         quoted ^= 1;
4240                         c = '"';
4241                         break;
4242                 case CTLENDVAR:
4243                         str = "\"}" + !(quoted & 1);
4244                         quoted >>= 1;
4245                         subtype = 0;
4246                         goto dostr;
4247                 case CTLBACKQ:
4248                         str = "$(...)";
4249                         goto dostr;
4250                 case CTLBACKQ+CTLQUOTE:
4251                         str = "\"$(...)\"";
4252                         goto dostr;
4253 #if ENABLE_SH_MATH_SUPPORT
4254                 case CTLARI:
4255                         str = "$((";
4256                         goto dostr;
4257                 case CTLENDARI:
4258                         str = "))";
4259                         goto dostr;
4260 #endif
4261                 case CTLQUOTEMARK:
4262                         quoted ^= 1;
4263                         c = '"';
4264                         break;
4265                 case '=':
4266                         if (subtype == 0)
4267                                 break;
4268                         if ((subtype & VSTYPE) != VSNORMAL)
4269                                 quoted <<= 1;
4270                         str = vstype[subtype & VSTYPE];
4271                         if (subtype & VSNUL)
4272                                 c = ':';
4273                         else
4274                                 goto checkstr;
4275                         break;
4276                 case '\'':
4277                 case '\\':
4278                 case '"':
4279                 case '$':
4280                         /* These can only happen inside quotes */
4281                         cc[0] = c;
4282                         str = cc;
4283                         c = '\\';
4284                         break;
4285                 default:
4286                         break;
4287                 }
4288                 USTPUTC(c, nextc);
4289  checkstr:
4290                 if (!str)
4291                         continue;
4292  dostr:
4293                 while ((c = *str++) != '\0') {
4294                         USTPUTC(c, nextc);
4295                 }
4296         } /* while *p++ not NUL */
4297
4298         if (quoted & 1) {
4299                 USTPUTC('"', nextc);
4300         }
4301         *nextc = 0;
4302         cmdnextc = nextc;
4303 }
4304
4305 /* cmdtxt() and cmdlist() call each other */
4306 static void cmdtxt(union node *n);
4307
4308 static void
4309 cmdlist(union node *np, int sep)
4310 {
4311         for (; np; np = np->narg.next) {
4312                 if (!sep)
4313                         cmdputs(" ");
4314                 cmdtxt(np);
4315                 if (sep && np->narg.next)
4316                         cmdputs(" ");
4317         }
4318 }
4319
4320 static void
4321 cmdtxt(union node *n)
4322 {
4323         union node *np;
4324         struct nodelist *lp;
4325         const char *p;
4326
4327         if (!n)
4328                 return;
4329         switch (n->type) {
4330         default:
4331 #if DEBUG
4332                 abort();
4333 #endif
4334         case NPIPE:
4335                 lp = n->npipe.cmdlist;
4336                 for (;;) {
4337                         cmdtxt(lp->n);
4338                         lp = lp->next;
4339                         if (!lp)
4340                                 break;
4341                         cmdputs(" | ");
4342                 }
4343                 break;
4344         case NSEMI:
4345                 p = "; ";
4346                 goto binop;
4347         case NAND:
4348                 p = " && ";
4349                 goto binop;
4350         case NOR:
4351                 p = " || ";
4352  binop:
4353                 cmdtxt(n->nbinary.ch1);
4354                 cmdputs(p);
4355                 n = n->nbinary.ch2;
4356                 goto donode;
4357         case NREDIR:
4358         case NBACKGND:
4359                 n = n->nredir.n;
4360                 goto donode;
4361         case NNOT:
4362                 cmdputs("!");
4363                 n = n->nnot.com;
4364  donode:
4365                 cmdtxt(n);
4366                 break;
4367         case NIF:
4368                 cmdputs("if ");
4369                 cmdtxt(n->nif.test);
4370                 cmdputs("; then ");
4371                 if (n->nif.elsepart) {
4372                         cmdtxt(n->nif.ifpart);
4373                         cmdputs("; else ");
4374                         n = n->nif.elsepart;
4375                 } else {
4376                         n = n->nif.ifpart;
4377                 }
4378                 p = "; fi";
4379                 goto dotail;
4380         case NSUBSHELL:
4381                 cmdputs("(");
4382                 n = n->nredir.n;
4383                 p = ")";
4384                 goto dotail;
4385         case NWHILE:
4386                 p = "while ";
4387                 goto until;
4388         case NUNTIL:
4389                 p = "until ";
4390  until:
4391                 cmdputs(p);
4392                 cmdtxt(n->nbinary.ch1);
4393                 n = n->nbinary.ch2;
4394                 p = "; done";
4395  dodo:
4396                 cmdputs("; do ");
4397  dotail:
4398                 cmdtxt(n);
4399                 goto dotail2;
4400         case NFOR:
4401                 cmdputs("for ");
4402                 cmdputs(n->nfor.var);
4403                 cmdputs(" in ");
4404                 cmdlist(n->nfor.args, 1);
4405                 n = n->nfor.body;
4406                 p = "; done";
4407                 goto dodo;
4408         case NDEFUN:
4409                 cmdputs(n->narg.text);
4410                 p = "() { ... }";
4411                 goto dotail2;
4412         case NCMD:
4413                 cmdlist(n->ncmd.args, 1);
4414                 cmdlist(n->ncmd.redirect, 0);
4415                 break;
4416         case NARG:
4417                 p = n->narg.text;
4418  dotail2:
4419                 cmdputs(p);
4420                 break;
4421         case NHERE:
4422         case NXHERE:
4423                 p = "<<...";
4424                 goto dotail2;
4425         case NCASE:
4426                 cmdputs("case ");
4427                 cmdputs(n->ncase.expr->narg.text);
4428                 cmdputs(" in ");
4429                 for (np = n->ncase.cases; np; np = np->nclist.next) {
4430                         cmdtxt(np->nclist.pattern);
4431                         cmdputs(") ");
4432                         cmdtxt(np->nclist.body);
4433                         cmdputs(";; ");
4434                 }
4435                 p = "esac";
4436                 goto dotail2;
4437         case NTO:
4438                 p = ">";
4439                 goto redir;
4440         case NCLOBBER:
4441                 p = ">|";
4442                 goto redir;
4443         case NAPPEND:
4444                 p = ">>";
4445                 goto redir;
4446 #if ENABLE_ASH_BASH_COMPAT
4447         case NTO2:
4448 #endif
4449         case NTOFD:
4450                 p = ">&";
4451                 goto redir;
4452         case NFROM:
4453                 p = "<";
4454                 goto redir;
4455         case NFROMFD:
4456                 p = "<&";
4457                 goto redir;
4458         case NFROMTO:
4459                 p = "<>";
4460  redir:
4461                 cmdputs(utoa(n->nfile.fd));
4462                 cmdputs(p);
4463                 if (n->type == NTOFD || n->type == NFROMFD) {
4464                         cmdputs(utoa(n->ndup.dupfd));
4465                         break;
4466                 }
4467                 n = n->nfile.fname;
4468                 goto donode;
4469         }
4470 }
4471
4472 static char *
4473 commandtext(union node *n)
4474 {
4475         char *name;
4476
4477         STARTSTACKSTR(cmdnextc);
4478         cmdtxt(n);
4479         name = stackblock();
4480         TRACE(("commandtext: name %p, end %p\n\t\"%s\"\n",
4481                         name, cmdnextc, cmdnextc));
4482         return ckstrdup(name);
4483 }
4484 #endif /* JOBS */
4485
4486 /*
4487  * Fork off a subshell.  If we are doing job control, give the subshell its
4488  * own process group.  Jp is a job structure that the job is to be added to.
4489  * N is the command that will be evaluated by the child.  Both jp and n may
4490  * be NULL.  The mode parameter can be one of the following:
4491  *      FORK_FG - Fork off a foreground process.
4492  *      FORK_BG - Fork off a background process.
4493  *      FORK_NOJOB - Like FORK_FG, but don't give the process its own
4494  *                   process group even if job control is on.
4495  *
4496  * When job control is turned off, background processes have their standard
4497  * input redirected to /dev/null (except for the second and later processes
4498  * in a pipeline).
4499  *
4500  * Called with interrupts off.
4501  */
4502 /*
4503  * Clear traps on a fork.
4504  */
4505 static void
4506 clear_traps(void)
4507 {
4508         char **tp;
4509
4510         for (tp = trap; tp < &trap[NSIG]; tp++) {
4511                 if (*tp && **tp) {      /* trap not NULL or "" (SIG_IGN) */
4512                         INT_OFF;
4513                         if (trap_ptr == trap)
4514                                 free(*tp);
4515                         /* else: it "belongs" to trap_ptr vector, don't free */
4516                         *tp = NULL;
4517                         if ((tp - trap) != 0)
4518                                 setsignal(tp - trap);
4519                         INT_ON;
4520                 }
4521         }
4522 }
4523
4524 /* Lives far away from here, needed for forkchild */
4525 static void closescript(void);
4526
4527 /* Called after fork(), in child */
4528 static NOINLINE void
4529 forkchild(struct job *jp, union node *n, int mode)
4530 {
4531         int oldlvl;
4532
4533         TRACE(("Child shell %d\n", getpid()));
4534         oldlvl = shlvl;
4535         shlvl++;
4536
4537         /* man bash: "Non-builtin commands run by bash have signal handlers
4538          * set to the values inherited by the shell from its parent".
4539          * Do we do it correctly? */
4540
4541         closescript();
4542
4543         if (mode == FORK_NOJOB          /* is it `xxx` ? */
4544          && n && n->type == NCMD        /* is it single cmd? */
4545         /* && n->ncmd.args->type == NARG - always true? */
4546          && strcmp(n->ncmd.args->narg.text, "trap") == 0
4547          && n->ncmd.args->narg.next == NULL /* "trap" with no arguments */
4548         /* && n->ncmd.args->narg.backquote == NULL - do we need to check this? */
4549         ) {
4550                 TRACE(("Trap hack\n"));
4551                 /* Awful hack for `trap` or $(trap).
4552                  *
4553                  * http://www.opengroup.org/onlinepubs/009695399/utilities/trap.html
4554                  * contains an example where "trap" is executed in a subshell:
4555                  *
4556                  * save_traps=$(trap)
4557                  * ...
4558                  * eval "$save_traps"
4559                  *
4560                  * Standard does not say that "trap" in subshell shall print
4561                  * parent shell's traps. It only says that its output
4562                  * must have suitable form, but then, in the above example
4563                  * (which is not supposed to be normative), it implies that.
4564                  *
4565                  * bash (and probably other shell) does implement it
4566                  * (traps are reset to defaults, but "trap" still shows them),
4567                  * but as a result, "trap" logic is hopelessly messed up:
4568                  *
4569                  * # trap
4570                  * trap -- 'echo Ho' SIGWINCH  <--- we have a handler
4571                  * # (trap)        <--- trap is in subshell - no output (correct, traps are reset)
4572                  * # true | trap   <--- trap is in subshell - no output (ditto)
4573                  * # echo `true | trap`    <--- in subshell - output (but traps are reset!)
4574                  * trap -- 'echo Ho' SIGWINCH
4575                  * # echo `(trap)`         <--- in subshell in subshell - output
4576                  * trap -- 'echo Ho' SIGWINCH
4577                  * # echo `true | (trap)`  <--- in subshell in subshell in subshell - output!
4578                  * trap -- 'echo Ho' SIGWINCH
4579                  *
4580                  * The rules when to forget and when to not forget traps
4581                  * get really complex and nonsensical.
4582                  *
4583                  * Our solution: ONLY bare $(trap) or `trap` is special.
4584                  */
4585                 /* Save trap handler strings for trap builtin to print */
4586                 trap_ptr = memcpy(xmalloc(sizeof(trap)), trap, sizeof(trap));
4587                 /* Fall through into clearing traps */
4588         }
4589         clear_traps();
4590 #if JOBS
4591         /* do job control only in root shell */
4592         doing_jobctl = 0;
4593         if (mode != FORK_NOJOB && jp->jobctl && !oldlvl) {
4594                 pid_t pgrp;
4595
4596                 if (jp->nprocs == 0)
4597                         pgrp = getpid();
4598                 else
4599                         pgrp = jp->ps[0].ps_pid;
4600                 /* this can fail because we are doing it in the parent also */
4601                 setpgid(0, pgrp);
4602                 if (mode == FORK_FG)
4603                         xtcsetpgrp(ttyfd, pgrp);
4604                 setsignal(SIGTSTP);
4605                 setsignal(SIGTTOU);
4606         } else
4607 #endif
4608         if (mode == FORK_BG) {
4609                 /* man bash: "When job control is not in effect,
4610                  * asynchronous commands ignore SIGINT and SIGQUIT" */
4611                 ignoresig(SIGINT);
4612                 ignoresig(SIGQUIT);
4613                 if (jp->nprocs == 0) {
4614                         close(0);
4615                         if (open(bb_dev_null, O_RDONLY) != 0)
4616                                 ash_msg_and_raise_error("can't open '%s'", bb_dev_null);
4617                 }
4618         }
4619         if (!oldlvl) {
4620                 if (iflag) { /* why if iflag only? */
4621                         setsignal(SIGINT);
4622                         setsignal(SIGTERM);
4623                 }
4624                 /* man bash:
4625                  * "In all cases, bash ignores SIGQUIT. Non-builtin
4626                  * commands run by bash have signal handlers
4627                  * set to the values inherited by the shell
4628                  * from its parent".
4629                  * Take care of the second rule: */
4630                 setsignal(SIGQUIT);
4631         }
4632 #if JOBS
4633         if (n && n->type == NCMD
4634          && strcmp(n->ncmd.args->narg.text, "jobs") == 0
4635         ) {
4636                 TRACE(("Job hack\n"));
4637                 /* "jobs": we do not want to clear job list for it,
4638                  * instead we remove only _its_ own_ job from job list.
4639                  * This makes "jobs .... | cat" more useful.
4640                  */
4641                 freejob(curjob);
4642                 return;
4643         }
4644 #endif
4645         for (jp = curjob; jp; jp = jp->prev_job)
4646                 freejob(jp);
4647         jobless = 0;
4648 }
4649
4650 /* Called after fork(), in parent */
4651 #if !JOBS
4652 #define forkparent(jp, n, mode, pid) forkparent(jp, mode, pid)
4653 #endif
4654 static void
4655 forkparent(struct job *jp, union node *n, int mode, pid_t pid)
4656 {
4657         TRACE(("In parent shell: child = %d\n", pid));
4658         if (!jp) {
4659                 while (jobless && dowait(DOWAIT_NONBLOCK, NULL) > 0)
4660                         continue;
4661                 jobless++;
4662                 return;
4663         }
4664 #if JOBS
4665         if (mode != FORK_NOJOB && jp->jobctl) {
4666                 int pgrp;
4667
4668                 if (jp->nprocs == 0)
4669                         pgrp = pid;
4670                 else
4671                         pgrp = jp->ps[0].ps_pid;
4672                 /* This can fail because we are doing it in the child also */
4673                 setpgid(pid, pgrp);
4674         }
4675 #endif
4676         if (mode == FORK_BG) {
4677                 backgndpid = pid;               /* set $! */
4678                 set_curjob(jp, CUR_RUNNING);
4679         }
4680         if (jp) {
4681                 struct procstat *ps = &jp->ps[jp->nprocs++];
4682                 ps->ps_pid = pid;
4683                 ps->ps_status = -1;
4684                 ps->ps_cmd = nullstr;
4685 #if JOBS
4686                 if (doing_jobctl && n)
4687                         ps->ps_cmd = commandtext(n);
4688 #endif
4689         }
4690 }
4691
4692 static int
4693 forkshell(struct job *jp, union node *n, int mode)
4694 {
4695         int pid;
4696
4697         TRACE(("forkshell(%%%d, %p, %d) called\n", jobno(jp), n, mode));
4698         pid = fork();
4699         if (pid < 0) {
4700                 TRACE(("Fork failed, errno=%d", errno));
4701                 if (jp)
4702                         freejob(jp);
4703                 ash_msg_and_raise_error("can't fork");
4704         }
4705         if (pid == 0) {
4706                 CLEAR_RANDOM_T(&random_gen); /* or else $RANDOM repeats in child */
4707                 forkchild(jp, n, mode);
4708         } else {
4709                 forkparent(jp, n, mode, pid);
4710         }
4711         return pid;
4712 }
4713
4714 /*
4715  * Wait for job to finish.
4716  *
4717  * Under job control we have the problem that while a child process
4718  * is running interrupts generated by the user are sent to the child
4719  * but not to the shell.  This means that an infinite loop started by
4720  * an interactive user may be hard to kill.  With job control turned off,
4721  * an interactive user may place an interactive program inside a loop.
4722  * If the interactive program catches interrupts, the user doesn't want
4723  * these interrupts to also abort the loop.  The approach we take here
4724  * is to have the shell ignore interrupt signals while waiting for a
4725  * foreground process to terminate, and then send itself an interrupt
4726  * signal if the child process was terminated by an interrupt signal.
4727  * Unfortunately, some programs want to do a bit of cleanup and then
4728  * exit on interrupt; unless these processes terminate themselves by
4729  * sending a signal to themselves (instead of calling exit) they will
4730  * confuse this approach.
4731  *
4732  * Called with interrupts off.
4733  */
4734 static int
4735 waitforjob(struct job *jp)
4736 {
4737         int st;
4738
4739         TRACE(("waitforjob(%%%d) called\n", jobno(jp)));
4740
4741         INT_OFF;
4742         while (jp->state == JOBRUNNING) {
4743                 /* In non-interactive shells, we _can_ get
4744                  * a keyboard signal here and be EINTRed,
4745                  * but we just loop back, waiting for command to complete.
4746                  *
4747                  * man bash:
4748                  * "If bash is waiting for a command to complete and receives
4749                  * a signal for which a trap has been set, the trap
4750                  * will not be executed until the command completes."
4751                  *
4752                  * Reality is that even if trap is not set, bash
4753                  * will not act on the signal until command completes.
4754                  * Try this. sleep5intoff.c:
4755                  * #include <signal.h>
4756                  * #include <unistd.h>
4757                  * int main() {
4758                  *         sigset_t set;
4759                  *         sigemptyset(&set);
4760                  *         sigaddset(&set, SIGINT);
4761                  *         sigaddset(&set, SIGQUIT);
4762                  *         sigprocmask(SIG_BLOCK, &set, NULL);
4763                  *         sleep(5);
4764                  *         return 0;
4765                  * }
4766                  * $ bash -c './sleep5intoff; echo hi'
4767                  * ^C^C^C^C <--- pressing ^C once a second
4768                  * $ _
4769                  * $ bash -c './sleep5intoff; echo hi'
4770                  * ^\^\^\^\hi <--- pressing ^\ (SIGQUIT)
4771                  * $ _
4772                  */
4773                 dowait(DOWAIT_BLOCK, jp);
4774         }
4775         INT_ON;
4776
4777         st = getstatus(jp);
4778 #if JOBS
4779         if (jp->jobctl) {
4780                 xtcsetpgrp(ttyfd, rootpid);
4781                 /*
4782                  * This is truly gross.
4783                  * If we're doing job control, then we did a TIOCSPGRP which
4784                  * caused us (the shell) to no longer be in the controlling
4785                  * session -- so we wouldn't have seen any ^C/SIGINT.  So, we
4786                  * intuit from the subprocess exit status whether a SIGINT
4787                  * occurred, and if so interrupt ourselves.  Yuck.  - mycroft
4788                  */
4789                 if (jp->sigint) /* TODO: do the same with all signals */
4790                         raise(SIGINT); /* ... by raise(jp->sig) instead? */
4791         }
4792         if (jp->state == JOBDONE)
4793 #endif
4794                 freejob(jp);
4795         return st;
4796 }
4797
4798 /*
4799  * return 1 if there are stopped jobs, otherwise 0
4800  */
4801 static int
4802 stoppedjobs(void)
4803 {
4804         struct job *jp;
4805         int retval;
4806
4807         retval = 0;
4808         if (job_warning)
4809                 goto out;
4810         jp = curjob;
4811         if (jp && jp->state == JOBSTOPPED) {
4812                 out2str("You have stopped jobs.\n");
4813                 job_warning = 2;
4814                 retval++;
4815         }
4816  out:
4817         return retval;
4818 }
4819
4820
4821 /* ============ redir.c
4822  *
4823  * Code for dealing with input/output redirection.
4824  */
4825
4826 #define EMPTY -2                /* marks an unused slot in redirtab */
4827 #define CLOSED -3               /* marks a slot of previously-closed fd */
4828
4829 /*
4830  * Open a file in noclobber mode.
4831  * The code was copied from bash.
4832  */
4833 static int
4834 noclobberopen(const char *fname)
4835 {
4836         int r, fd;
4837         struct stat finfo, finfo2;
4838
4839         /*
4840          * If the file exists and is a regular file, return an error
4841          * immediately.
4842          */
4843         r = stat(fname, &finfo);
4844         if (r == 0 && S_ISREG(finfo.st_mode)) {
4845                 errno = EEXIST;
4846                 return -1;
4847         }
4848
4849         /*
4850          * If the file was not present (r != 0), make sure we open it
4851          * exclusively so that if it is created before we open it, our open
4852          * will fail.  Make sure that we do not truncate an existing file.
4853          * Note that we don't turn on O_EXCL unless the stat failed -- if the
4854          * file was not a regular file, we leave O_EXCL off.
4855          */
4856         if (r != 0)
4857                 return open(fname, O_WRONLY|O_CREAT|O_EXCL, 0666);
4858         fd = open(fname, O_WRONLY|O_CREAT, 0666);
4859
4860         /* If the open failed, return the file descriptor right away. */
4861         if (fd < 0)
4862                 return fd;
4863
4864         /*
4865          * OK, the open succeeded, but the file may have been changed from a
4866          * non-regular file to a regular file between the stat and the open.
4867          * We are assuming that the O_EXCL open handles the case where FILENAME
4868          * did not exist and is symlinked to an existing file between the stat
4869          * and open.
4870          */
4871
4872         /*
4873          * If we can open it and fstat the file descriptor, and neither check
4874          * revealed that it was a regular file, and the file has not been
4875          * replaced, return the file descriptor.
4876          */
4877         if (fstat(fd, &finfo2) == 0 && !S_ISREG(finfo2.st_mode)
4878          && finfo.st_dev == finfo2.st_dev && finfo.st_ino == finfo2.st_ino)
4879                 return fd;
4880
4881         /* The file has been replaced.  badness. */
4882         close(fd);
4883         errno = EEXIST;
4884         return -1;
4885 }
4886
4887 /*
4888  * Handle here documents.  Normally we fork off a process to write the
4889  * data to a pipe.  If the document is short, we can stuff the data in
4890  * the pipe without forking.
4891  */
4892 /* openhere needs this forward reference */
4893 static void expandhere(union node *arg, int fd);
4894 static int
4895 openhere(union node *redir)
4896 {
4897         int pip[2];
4898         size_t len = 0;
4899
4900         if (pipe(pip) < 0)
4901                 ash_msg_and_raise_error("pipe call failed");
4902         if (redir->type == NHERE) {
4903                 len = strlen(redir->nhere.doc->narg.text);
4904                 if (len <= PIPE_BUF) {
4905                         full_write(pip[1], redir->nhere.doc->narg.text, len);
4906                         goto out;
4907                 }
4908         }
4909         if (forkshell((struct job *)NULL, (union node *)NULL, FORK_NOJOB) == 0) {
4910                 /* child */
4911                 close(pip[0]);
4912                 ignoresig(SIGINT);  //signal(SIGINT, SIG_IGN);
4913                 ignoresig(SIGQUIT); //signal(SIGQUIT, SIG_IGN);
4914                 ignoresig(SIGHUP);  //signal(SIGHUP, SIG_IGN);
4915                 ignoresig(SIGTSTP); //signal(SIGTSTP, SIG_IGN);
4916                 signal(SIGPIPE, SIG_DFL);
4917                 if (redir->type == NHERE)
4918                         full_write(pip[1], redir->nhere.doc->narg.text, len);
4919                 else /* NXHERE */
4920                         expandhere(redir->nhere.doc, pip[1]);
4921                 _exit(EXIT_SUCCESS);
4922         }
4923  out:
4924         close(pip[1]);
4925         return pip[0];
4926 }
4927
4928 static int
4929 openredirect(union node *redir)
4930 {
4931         char *fname;
4932         int f;
4933
4934         switch (redir->nfile.type) {
4935         case NFROM:
4936                 fname = redir->nfile.expfname;
4937                 f = open(fname, O_RDONLY);
4938                 if (f < 0)
4939                         goto eopen;
4940                 break;
4941         case NFROMTO:
4942                 fname = redir->nfile.expfname;
4943                 f = open(fname, O_RDWR|O_CREAT|O_TRUNC, 0666);
4944                 if (f < 0)
4945                         goto ecreate;
4946                 break;
4947         case NTO:
4948 #if ENABLE_ASH_BASH_COMPAT
4949         case NTO2:
4950 #endif
4951                 /* Take care of noclobber mode. */
4952                 if (Cflag) {
4953                         fname = redir->nfile.expfname;
4954                         f = noclobberopen(fname);
4955                         if (f < 0)
4956                                 goto ecreate;
4957                         break;
4958                 }
4959                 /* FALLTHROUGH */
4960         case NCLOBBER:
4961                 fname = redir->nfile.expfname;
4962                 f = open(fname, O_WRONLY|O_CREAT|O_TRUNC, 0666);
4963                 if (f < 0)
4964                         goto ecreate;
4965                 break;
4966         case NAPPEND:
4967                 fname = redir->nfile.expfname;
4968                 f = open(fname, O_WRONLY|O_CREAT|O_APPEND, 0666);
4969                 if (f < 0)
4970                         goto ecreate;
4971                 break;
4972         default:
4973 #if DEBUG
4974                 abort();
4975 #endif
4976                 /* Fall through to eliminate warning. */
4977 /* Our single caller does this itself */
4978 //      case NTOFD:
4979 //      case NFROMFD:
4980 //              f = -1;
4981 //              break;
4982         case NHERE:
4983         case NXHERE:
4984                 f = openhere(redir);
4985                 break;
4986         }
4987
4988         return f;
4989  ecreate:
4990         ash_msg_and_raise_error("can't create %s: %s", fname, errmsg(errno, "nonexistent directory"));
4991  eopen:
4992         ash_msg_and_raise_error("can't open %s: %s", fname, errmsg(errno, "no such file"));
4993 }
4994
4995 /*
4996  * Copy a file descriptor to be >= to.  Returns -1
4997  * if the source file descriptor is closed, EMPTY if there are no unused
4998  * file descriptors left.
4999  */
5000 /* 0x800..00: bit to set in "to" to request dup2 instead of fcntl(F_DUPFD).
5001  * old code was doing close(to) prior to copyfd() to achieve the same */
5002 enum {
5003         COPYFD_EXACT   = (int)~(INT_MAX),
5004         COPYFD_RESTORE = (int)((unsigned)COPYFD_EXACT >> 1),
5005 };
5006 static int
5007 copyfd(int from, int to)
5008 {
5009         int newfd;
5010
5011         if (to & COPYFD_EXACT) {
5012                 to &= ~COPYFD_EXACT;
5013                 /*if (from != to)*/
5014                         newfd = dup2(from, to);
5015         } else {
5016                 newfd = fcntl(from, F_DUPFD, to);
5017         }
5018         if (newfd < 0) {
5019                 if (errno == EMFILE)
5020                         return EMPTY;
5021                 /* Happens when source fd is not open: try "echo >&99" */
5022                 ash_msg_and_raise_error("%d: %m", from);
5023         }
5024         return newfd;
5025 }
5026
5027 /* Struct def and variable are moved down to the first usage site */
5028 struct two_fd_t {
5029         int orig, copy;
5030 };
5031 struct redirtab {
5032         struct redirtab *next;
5033         int nullredirs;
5034         int pair_count;
5035         struct two_fd_t two_fd[];
5036 };
5037 #define redirlist (G_var.redirlist)
5038
5039 static int need_to_remember(struct redirtab *rp, int fd)
5040 {
5041         int i;
5042
5043         if (!rp) /* remembering was not requested */
5044                 return 0;
5045
5046         for (i = 0; i < rp->pair_count; i++) {
5047                 if (rp->two_fd[i].orig == fd) {
5048                         /* already remembered */
5049                         return 0;
5050                 }
5051         }
5052         return 1;
5053 }
5054
5055 /* "hidden" fd is a fd used to read scripts, or a copy of such */
5056 static int is_hidden_fd(struct redirtab *rp, int fd)
5057 {
5058         int i;
5059         struct parsefile *pf;
5060
5061         if (fd == -1)
5062                 return 0;
5063         pf = g_parsefile;
5064         while (pf) {
5065                 if (fd == pf->fd) {
5066                         return 1;
5067                 }
5068                 pf = pf->prev;
5069         }
5070         if (!rp)
5071                 return 0;
5072         fd |= COPYFD_RESTORE;
5073         for (i = 0; i < rp->pair_count; i++) {
5074                 if (rp->two_fd[i].copy == fd) {
5075                         return 1;
5076                 }
5077         }
5078         return 0;
5079 }
5080
5081 /*
5082  * Process a list of redirection commands.  If the REDIR_PUSH flag is set,
5083  * old file descriptors are stashed away so that the redirection can be
5084  * undone by calling popredir.  If the REDIR_BACKQ flag is set, then the
5085  * standard output, and the standard error if it becomes a duplicate of
5086  * stdout, is saved in memory.
5087  */
5088 /* flags passed to redirect */
5089 #define REDIR_PUSH    01        /* save previous values of file descriptors */
5090 #define REDIR_SAVEFD2 03        /* set preverrout */
5091 static void
5092 redirect(union node *redir, int flags)
5093 {
5094         struct redirtab *sv;
5095         int sv_pos;
5096         int i;
5097         int fd;
5098         int newfd;
5099         int copied_fd2 = -1;
5100
5101         g_nullredirs++;
5102         if (!redir) {
5103                 return;
5104         }
5105
5106         sv = NULL;
5107         sv_pos = 0;
5108         INT_OFF;
5109         if (flags & REDIR_PUSH) {
5110                 union node *tmp = redir;
5111                 do {
5112                         sv_pos++;
5113 #if ENABLE_ASH_BASH_COMPAT
5114                         if (redir->nfile.type == NTO2)
5115                                 sv_pos++;
5116 #endif
5117                         tmp = tmp->nfile.next;
5118                 } while (tmp);
5119                 sv = ckmalloc(sizeof(*sv) + sv_pos * sizeof(sv->two_fd[0]));
5120                 sv->next = redirlist;
5121                 sv->pair_count = sv_pos;
5122                 redirlist = sv;
5123                 sv->nullredirs = g_nullredirs - 1;
5124                 g_nullredirs = 0;
5125                 while (sv_pos > 0) {
5126                         sv_pos--;
5127                         sv->two_fd[sv_pos].orig = sv->two_fd[sv_pos].copy = EMPTY;
5128                 }
5129         }
5130
5131         do {
5132                 fd = redir->nfile.fd;
5133                 if (redir->nfile.type == NTOFD || redir->nfile.type == NFROMFD) {
5134                         int right_fd = redir->ndup.dupfd;
5135                         /* redirect from/to same file descriptor? */
5136                         if (right_fd == fd)
5137                                 continue;
5138                         /* echo >&10 and 10 is a fd opened to the sh script? */
5139                         if (is_hidden_fd(sv, right_fd)) {
5140                                 errno = EBADF; /* as if it is closed */
5141                                 ash_msg_and_raise_error("%d: %m", right_fd);
5142                         }
5143                         newfd = -1;
5144                 } else {
5145                         newfd = openredirect(redir); /* always >= 0 */
5146                         if (fd == newfd) {
5147                                 /* Descriptor wasn't open before redirect.
5148                                  * Mark it for close in the future */
5149                                 if (need_to_remember(sv, fd)) {
5150                                         goto remember_to_close;
5151                                 }
5152                                 continue;
5153                         }
5154                 }
5155 #if ENABLE_ASH_BASH_COMPAT
5156  redirect_more:
5157 #endif
5158                 if (need_to_remember(sv, fd)) {
5159                         /* Copy old descriptor */
5160                         i = fcntl(fd, F_DUPFD, 10);
5161 /* You'd expect copy to be CLOEXECed. Currently these extra "saved" fds
5162  * are closed in popredir() in the child, preventing them from leaking
5163  * into child. (popredir() also cleans up the mess in case of failures)
5164  */
5165                         if (i == -1) {
5166                                 i = errno;
5167                                 if (i != EBADF) {
5168                                         /* Strange error (e.g. "too many files" EMFILE?) */
5169                                         if (newfd >= 0)
5170                                                 close(newfd);
5171                                         errno = i;
5172                                         ash_msg_and_raise_error("%d: %m", fd);
5173                                         /* NOTREACHED */
5174                                 }
5175                                 /* EBADF: it is not open - good, remember to close it */
5176  remember_to_close:
5177                                 i = CLOSED;
5178                         } else { /* fd is open, save its copy */
5179                                 /* "exec fd>&-" should not close fds
5180                                  * which point to script file(s).
5181                                  * Force them to be restored afterwards */
5182                                 if (is_hidden_fd(sv, fd))
5183                                         i |= COPYFD_RESTORE;
5184                         }
5185                         if (fd == 2)
5186                                 copied_fd2 = i;
5187                         sv->two_fd[sv_pos].orig = fd;
5188                         sv->two_fd[sv_pos].copy = i;
5189                         sv_pos++;
5190                 }
5191                 if (newfd < 0) {
5192                         /* NTOFD/NFROMFD: copy redir->ndup.dupfd to fd */
5193                         if (redir->ndup.dupfd < 0) { /* "fd>&-" */
5194                                 /* Don't want to trigger debugging */
5195                                 if (fd != -1)
5196                                         close(fd);
5197                         } else {
5198                                 copyfd(redir->ndup.dupfd, fd | COPYFD_EXACT);
5199                         }
5200                 } else if (fd != newfd) { /* move newfd to fd */
5201                         copyfd(newfd, fd | COPYFD_EXACT);
5202 #if ENABLE_ASH_BASH_COMPAT
5203                         if (!(redir->nfile.type == NTO2 && fd == 2))
5204 #endif
5205                                 close(newfd);
5206                 }
5207 #if ENABLE_ASH_BASH_COMPAT
5208                 if (redir->nfile.type == NTO2 && fd == 1) {
5209                         /* We already redirected it to fd 1, now copy it to 2 */
5210                         newfd = 1;
5211                         fd = 2;
5212                         goto redirect_more;
5213                 }
5214 #endif
5215         } while ((redir = redir->nfile.next) != NULL);
5216
5217         INT_ON;
5218         if ((flags & REDIR_SAVEFD2) && copied_fd2 >= 0)
5219                 preverrout_fd = copied_fd2;
5220 }
5221
5222 /*
5223  * Undo the effects of the last redirection.
5224  */
5225 static void
5226 popredir(int drop, int restore)
5227 {
5228         struct redirtab *rp;
5229         int i;
5230
5231         if (--g_nullredirs >= 0)
5232                 return;
5233         INT_OFF;
5234         rp = redirlist;
5235         for (i = 0; i < rp->pair_count; i++) {
5236                 int fd = rp->two_fd[i].orig;
5237                 int copy = rp->two_fd[i].copy;
5238                 if (copy == CLOSED) {
5239                         if (!drop)
5240                                 close(fd);
5241                         continue;
5242                 }
5243                 if (copy != EMPTY) {
5244                         if (!drop || (restore && (copy & COPYFD_RESTORE))) {
5245                                 copy &= ~COPYFD_RESTORE;
5246                                 /*close(fd);*/
5247                                 copyfd(copy, fd | COPYFD_EXACT);
5248                         }
5249                         close(copy & ~COPYFD_RESTORE);
5250                 }
5251         }
5252         redirlist = rp->next;
5253         g_nullredirs = rp->nullredirs;
5254         free(rp);
5255         INT_ON;
5256 }
5257
5258 /*
5259  * Undo all redirections.  Called on error or interrupt.
5260  */
5261
5262 /*
5263  * Discard all saved file descriptors.
5264  */
5265 static void
5266 clearredir(int drop)
5267 {
5268         for (;;) {
5269                 g_nullredirs = 0;
5270                 if (!redirlist)
5271                         break;
5272                 popredir(drop, /*restore:*/ 0);
5273         }
5274 }
5275
5276 static int
5277 redirectsafe(union node *redir, int flags)
5278 {
5279         int err;
5280         volatile int saveint;
5281         struct jmploc *volatile savehandler = exception_handler;
5282         struct jmploc jmploc;
5283
5284         SAVE_INT(saveint);
5285         /* "echo 9>/dev/null; echo >&9; echo result: $?" - result should be 1, not 2! */
5286         err = setjmp(jmploc.loc); // huh?? was = setjmp(jmploc.loc) * 2;
5287         if (!err) {
5288                 exception_handler = &jmploc;
5289                 redirect(redir, flags);
5290         }
5291         exception_handler = savehandler;
5292         if (err && exception_type != EXERROR)
5293                 longjmp(exception_handler->loc, 1);
5294         RESTORE_INT(saveint);
5295         return err;
5296 }
5297
5298
5299 /* ============ Routines to expand arguments to commands
5300  *
5301  * We have to deal with backquotes, shell variables, and file metacharacters.
5302  */
5303
5304 #if ENABLE_SH_MATH_SUPPORT
5305 static arith_t
5306 ash_arith(const char *s)
5307 {
5308         arith_eval_hooks_t math_hooks;
5309         arith_t result;
5310         int errcode = 0;
5311
5312         math_hooks.lookupvar = lookupvar;
5313         math_hooks.setvar = setvar;
5314         math_hooks.endofname = endofname;
5315
5316         INT_OFF;
5317         result = arith(s, &errcode, &math_hooks);
5318         if (errcode < 0) {
5319                 if (errcode == -3)
5320                         ash_msg_and_raise_error("exponent less than 0");
5321                 if (errcode == -2)
5322                         ash_msg_and_raise_error("divide by zero");
5323                 if (errcode == -5)
5324                         ash_msg_and_raise_error("expression recursion loop detected");
5325                 raise_error_syntax(s);
5326         }
5327         INT_ON;
5328
5329         return result;
5330 }
5331 #endif
5332
5333 /*
5334  * expandarg flags
5335  */
5336 #define EXP_FULL        0x1     /* perform word splitting & file globbing */
5337 #define EXP_TILDE       0x2     /* do normal tilde expansion */
5338 #define EXP_VARTILDE    0x4     /* expand tildes in an assignment */
5339 #define EXP_REDIR       0x8     /* file glob for a redirection (1 match only) */
5340 #define EXP_CASE        0x10    /* keeps quotes around for CASE pattern */
5341 #define EXP_RECORD      0x20    /* need to record arguments for ifs breakup */
5342 #define EXP_VARTILDE2   0x40    /* expand tildes after colons only */
5343 #define EXP_WORD        0x80    /* expand word in parameter expansion */
5344 #define EXP_QWORD       0x100   /* expand word in quoted parameter expansion */
5345 /*
5346  * rmescape() flags
5347  */
5348 #define RMESCAPE_ALLOC  0x1     /* Allocate a new string */
5349 #define RMESCAPE_GLOB   0x2     /* Add backslashes for glob */
5350 #define RMESCAPE_QUOTED 0x4     /* Remove CTLESC unless in quotes */
5351 #define RMESCAPE_GROW   0x8     /* Grow strings instead of stalloc */
5352 #define RMESCAPE_HEAP   0x10    /* Malloc strings instead of stalloc */
5353
5354 /*
5355  * Structure specifying which parts of the string should be searched
5356  * for IFS characters.
5357  */
5358 struct ifsregion {
5359         struct ifsregion *next; /* next region in list */
5360         int begoff;             /* offset of start of region */
5361         int endoff;             /* offset of end of region */
5362         int nulonly;            /* search for nul bytes only */
5363 };
5364
5365 struct arglist {
5366         struct strlist *list;
5367         struct strlist **lastp;
5368 };
5369
5370 /* output of current string */
5371 static char *expdest;
5372 /* list of back quote expressions */
5373 static struct nodelist *argbackq;
5374 /* first struct in list of ifs regions */
5375 static struct ifsregion ifsfirst;
5376 /* last struct in list */
5377 static struct ifsregion *ifslastp;
5378 /* holds expanded arg list */
5379 static struct arglist exparg;
5380
5381 /*
5382  * Our own itoa().
5383  */
5384 static int
5385 cvtnum(arith_t num)
5386 {
5387         int len;
5388
5389         expdest = makestrspace(32, expdest);
5390         len = fmtstr(expdest, 32, arith_t_fmt, num);
5391         STADJUST(len, expdest);
5392         return len;
5393 }
5394
5395 static size_t
5396 esclen(const char *start, const char *p)
5397 {
5398         size_t esc = 0;
5399
5400         while (p > start && (unsigned char)*--p == CTLESC) {
5401                 esc++;
5402         }
5403         return esc;
5404 }
5405
5406 /*
5407  * Remove any CTLESC characters from a string.
5408  */
5409 static char *
5410 rmescapes(char *str, int flag)
5411 {
5412         static const char qchars[] ALIGN1 = { CTLESC, CTLQUOTEMARK, '\0' };
5413
5414         char *p, *q, *r;
5415         unsigned inquotes;
5416         unsigned protect_against_glob;
5417         unsigned globbing;
5418
5419         p = strpbrk(str, qchars);
5420         if (!p)
5421                 return str;
5422
5423         q = p;
5424         r = str;
5425         if (flag & RMESCAPE_ALLOC) {
5426                 size_t len = p - str;
5427                 size_t fulllen = len + strlen(p) + 1;
5428
5429                 if (flag & RMESCAPE_GROW) {
5430                         r = makestrspace(fulllen, expdest);
5431                 } else if (flag & RMESCAPE_HEAP) {
5432                         r = ckmalloc(fulllen);
5433                 } else {
5434                         r = stalloc(fulllen);
5435                 }
5436                 q = r;
5437                 if (len > 0) {
5438                         q = (char *)memcpy(q, str, len) + len;
5439                 }
5440         }
5441
5442         inquotes = (flag & RMESCAPE_QUOTED) ^ RMESCAPE_QUOTED;
5443         globbing = flag & RMESCAPE_GLOB;
5444         protect_against_glob = globbing;
5445         while (*p) {
5446                 if ((unsigned char)*p == CTLQUOTEMARK) {
5447 // TODO: if no RMESCAPE_QUOTED in flags, inquotes never becomes 0
5448 // (alternates between RMESCAPE_QUOTED and ~RMESCAPE_QUOTED). Is it ok?
5449 // Note: both inquotes and protect_against_glob only affect whether
5450 // CTLESC,<ch> gets converted to <ch> or to \<ch>
5451                         inquotes = ~inquotes;
5452                         p++;
5453                         protect_against_glob = globbing;
5454                         continue;
5455                 }
5456                 if (*p == '\\') {
5457                         /* naked back slash */
5458                         protect_against_glob = 0;
5459                         goto copy;
5460                 }
5461                 if ((unsigned char)*p == CTLESC) {
5462                         p++;
5463                         if (protect_against_glob && inquotes && *p != '/') {
5464                                 *q++ = '\\';
5465                         }
5466                 }
5467                 protect_against_glob = globbing;
5468  copy:
5469                 *q++ = *p++;
5470         }
5471         *q = '\0';
5472         if (flag & RMESCAPE_GROW) {
5473                 expdest = r;
5474                 STADJUST(q - r + 1, expdest);
5475         }
5476         return r;
5477 }
5478 #define pmatch(a, b) !fnmatch((a), (b), 0)
5479
5480 /*
5481  * Prepare a pattern for a expmeta (internal glob(3)) call.
5482  *
5483  * Returns an stalloced string.
5484  */
5485 static char *
5486 preglob(const char *pattern, int quoted, int flag)
5487 {
5488         flag |= RMESCAPE_GLOB;
5489         if (quoted) {
5490                 flag |= RMESCAPE_QUOTED;
5491         }
5492         return rmescapes((char *)pattern, flag);
5493 }
5494
5495 /*
5496  * Put a string on the stack.
5497  */
5498 static void
5499 memtodest(const char *p, size_t len, int syntax, int quotes)
5500 {
5501         char *q = expdest;
5502
5503         q = makestrspace(quotes ? len * 2 : len, q);
5504
5505         while (len--) {
5506                 unsigned char c = *p++;
5507                 if (c == '\0')
5508                         continue;
5509                 if (quotes) {
5510                         int n = SIT(c, syntax);
5511                         if (n == CCTL || n == CBACK)
5512                                 USTPUTC(CTLESC, q);
5513                 }
5514                 USTPUTC(c, q);
5515         }
5516
5517         expdest = q;
5518 }
5519
5520 static void
5521 strtodest(const char *p, int syntax, int quotes)
5522 {
5523         memtodest(p, strlen(p), syntax, quotes);
5524 }
5525
5526 /*
5527  * Record the fact that we have to scan this region of the
5528  * string for IFS characters.
5529  */
5530 static void
5531 recordregion(int start, int end, int nulonly)
5532 {
5533         struct ifsregion *ifsp;
5534
5535         if (ifslastp == NULL) {
5536                 ifsp = &ifsfirst;
5537         } else {
5538                 INT_OFF;
5539                 ifsp = ckzalloc(sizeof(*ifsp));
5540                 /*ifsp->next = NULL; - ckzalloc did it */
5541                 ifslastp->next = ifsp;
5542                 INT_ON;
5543         }
5544         ifslastp = ifsp;
5545         ifslastp->begoff = start;
5546         ifslastp->endoff = end;
5547         ifslastp->nulonly = nulonly;
5548 }
5549
5550 static void
5551 removerecordregions(int endoff)
5552 {
5553         if (ifslastp == NULL)
5554                 return;
5555
5556         if (ifsfirst.endoff > endoff) {
5557                 while (ifsfirst.next != NULL) {
5558                         struct ifsregion *ifsp;
5559                         INT_OFF;
5560                         ifsp = ifsfirst.next->next;
5561                         free(ifsfirst.next);
5562                         ifsfirst.next = ifsp;
5563                         INT_ON;
5564                 }
5565                 if (ifsfirst.begoff > endoff)
5566                         ifslastp = NULL;
5567                 else {
5568                         ifslastp = &ifsfirst;
5569                         ifsfirst.endoff = endoff;
5570                 }
5571                 return;
5572         }
5573
5574         ifslastp = &ifsfirst;
5575         while (ifslastp->next && ifslastp->next->begoff < endoff)
5576                 ifslastp=ifslastp->next;
5577         while (ifslastp->next != NULL) {
5578                 struct ifsregion *ifsp;
5579                 INT_OFF;
5580                 ifsp = ifslastp->next->next;
5581                 free(ifslastp->next);
5582                 ifslastp->next = ifsp;
5583                 INT_ON;
5584         }
5585         if (ifslastp->endoff > endoff)
5586                 ifslastp->endoff = endoff;
5587 }
5588
5589 static char *
5590 exptilde(char *startp, char *p, int flags)
5591 {
5592         unsigned char c;
5593         char *name;
5594         struct passwd *pw;
5595         const char *home;
5596         int quotes = flags & (EXP_FULL | EXP_CASE | EXP_REDIR);
5597         int startloc;
5598
5599         name = p + 1;
5600
5601         while ((c = *++p) != '\0') {
5602                 switch (c) {
5603                 case CTLESC:
5604                         return startp;
5605                 case CTLQUOTEMARK:
5606                         return startp;
5607                 case ':':
5608                         if (flags & EXP_VARTILDE)
5609                                 goto done;
5610                         break;
5611                 case '/':
5612                 case CTLENDVAR:
5613                         goto done;
5614                 }
5615         }
5616  done:
5617         *p = '\0';
5618         if (*name == '\0') {
5619                 home = lookupvar(homestr);
5620         } else {
5621                 pw = getpwnam(name);
5622                 if (pw == NULL)
5623                         goto lose;
5624                 home = pw->pw_dir;
5625         }
5626         if (!home || !*home)
5627                 goto lose;
5628         *p = c;
5629         startloc = expdest - (char *)stackblock();
5630         strtodest(home, SQSYNTAX, quotes);
5631         recordregion(startloc, expdest - (char *)stackblock(), 0);
5632         return p;
5633  lose:
5634         *p = c;
5635         return startp;
5636 }
5637
5638 /*
5639  * Execute a command inside back quotes.  If it's a builtin command, we
5640  * want to save its output in a block obtained from malloc.  Otherwise
5641  * we fork off a subprocess and get the output of the command via a pipe.
5642  * Should be called with interrupts off.
5643  */
5644 struct backcmd {                /* result of evalbackcmd */
5645         int fd;                 /* file descriptor to read from */
5646         int nleft;              /* number of chars in buffer */
5647         char *buf;              /* buffer */
5648         struct job *jp;         /* job structure for command */
5649 };
5650
5651 /* These forward decls are needed to use "eval" code for backticks handling: */
5652 static uint8_t back_exitstatus; /* exit status of backquoted command */
5653 #define EV_EXIT 01              /* exit after evaluating tree */
5654 static void evaltree(union node *, int);
5655
5656 static void FAST_FUNC
5657 evalbackcmd(union node *n, struct backcmd *result)
5658 {
5659         int saveherefd;
5660
5661         result->fd = -1;
5662         result->buf = NULL;
5663         result->nleft = 0;
5664         result->jp = NULL;
5665         if (n == NULL)
5666                 goto out;
5667
5668         saveherefd = herefd;
5669         herefd = -1;
5670
5671         {
5672                 int pip[2];
5673                 struct job *jp;
5674
5675                 if (pipe(pip) < 0)
5676                         ash_msg_and_raise_error("pipe call failed");
5677                 jp = makejob(/*n,*/ 1);
5678                 if (forkshell(jp, n, FORK_NOJOB) == 0) {
5679                         FORCE_INT_ON;
5680                         close(pip[0]);
5681                         if (pip[1] != 1) {
5682                                 /*close(1);*/
5683                                 copyfd(pip[1], 1 | COPYFD_EXACT);
5684                                 close(pip[1]);
5685                         }
5686                         eflag = 0;
5687                         evaltree(n, EV_EXIT); /* actually evaltreenr... */
5688                         /* NOTREACHED */
5689                 }
5690                 close(pip[1]);
5691                 result->fd = pip[0];
5692                 result->jp = jp;
5693         }
5694         herefd = saveherefd;
5695  out:
5696         TRACE(("evalbackcmd done: fd=%d buf=0x%x nleft=%d jp=0x%x\n",
5697                 result->fd, result->buf, result->nleft, result->jp));
5698 }
5699
5700 /*
5701  * Expand stuff in backwards quotes.
5702  */
5703 static void
5704 expbackq(union node *cmd, int quoted, int quotes)
5705 {
5706         struct backcmd in;
5707         int i;
5708         char buf[128];
5709         char *p;
5710         char *dest;
5711         int startloc;
5712         int syntax = quoted ? DQSYNTAX : BASESYNTAX;
5713         struct stackmark smark;
5714
5715         INT_OFF;
5716         setstackmark(&smark);
5717         dest = expdest;
5718         startloc = dest - (char *)stackblock();
5719         grabstackstr(dest);
5720         evalbackcmd(cmd, &in);
5721         popstackmark(&smark);
5722
5723         p = in.buf;
5724         i = in.nleft;
5725         if (i == 0)
5726                 goto read;
5727         for (;;) {
5728                 memtodest(p, i, syntax, quotes);
5729  read:
5730                 if (in.fd < 0)
5731                         break;
5732                 i = nonblock_safe_read(in.fd, buf, sizeof(buf));
5733                 TRACE(("expbackq: read returns %d\n", i));
5734                 if (i <= 0)
5735                         break;
5736                 p = buf;
5737         }
5738
5739         free(in.buf);
5740         if (in.fd >= 0) {
5741                 close(in.fd);
5742                 back_exitstatus = waitforjob(in.jp);
5743         }
5744         INT_ON;
5745
5746         /* Eat all trailing newlines */
5747         dest = expdest;
5748         for (; dest > (char *)stackblock() && dest[-1] == '\n';)
5749                 STUNPUTC(dest);
5750         expdest = dest;
5751
5752         if (quoted == 0)
5753                 recordregion(startloc, dest - (char *)stackblock(), 0);
5754         TRACE(("evalbackq: size=%d: \"%.*s\"\n",
5755                 (dest - (char *)stackblock()) - startloc,
5756                 (dest - (char *)stackblock()) - startloc,
5757                 stackblock() + startloc));
5758 }
5759
5760 #if ENABLE_SH_MATH_SUPPORT
5761 /*
5762  * Expand arithmetic expression.  Backup to start of expression,
5763  * evaluate, place result in (backed up) result, adjust string position.
5764  */
5765 static void
5766 expari(int quotes)
5767 {
5768         char *p, *start;
5769         int begoff;
5770         int flag;
5771         int len;
5772
5773         /* ifsfree(); */
5774
5775         /*
5776          * This routine is slightly over-complicated for
5777          * efficiency.  Next we scan backwards looking for the
5778          * start of arithmetic.
5779          */
5780         start = stackblock();
5781         p = expdest - 1;
5782         *p = '\0';
5783         p--;
5784         do {
5785                 int esc;
5786
5787                 while ((unsigned char)*p != CTLARI) {
5788                         p--;
5789 #if DEBUG
5790                         if (p < start) {
5791                                 ash_msg_and_raise_error("missing CTLARI (shouldn't happen)");
5792                         }
5793 #endif
5794                 }
5795
5796                 esc = esclen(start, p);
5797                 if (!(esc % 2)) {
5798                         break;
5799                 }
5800
5801                 p -= esc + 1;
5802         } while (1);
5803
5804         begoff = p - start;
5805
5806         removerecordregions(begoff);
5807
5808         flag = p[1];
5809
5810         expdest = p;
5811
5812         if (quotes)
5813                 rmescapes(p + 2, 0);
5814
5815         len = cvtnum(ash_arith(p + 2));
5816
5817         if (flag != '"')
5818                 recordregion(begoff, begoff + len, 0);
5819 }
5820 #endif
5821
5822 /* argstr needs it */
5823 static char *evalvar(char *p, int flags, struct strlist *var_str_list);
5824
5825 /*
5826  * Perform variable and command substitution.  If EXP_FULL is set, output CTLESC
5827  * characters to allow for further processing.  Otherwise treat
5828  * $@ like $* since no splitting will be performed.
5829  *
5830  * var_str_list (can be NULL) is a list of "VAR=val" strings which take precedence
5831  * over shell varables. Needed for "A=a B=$A; echo $B" case - we use it
5832  * for correct expansion of "B=$A" word.
5833  */
5834 static void
5835 argstr(char *p, int flags, struct strlist *var_str_list)
5836 {
5837         static const char spclchars[] ALIGN1 = {
5838                 '=',
5839                 ':',
5840                 CTLQUOTEMARK,
5841                 CTLENDVAR,
5842                 CTLESC,
5843                 CTLVAR,
5844                 CTLBACKQ,
5845                 CTLBACKQ | CTLQUOTE,
5846 #if ENABLE_SH_MATH_SUPPORT
5847                 CTLENDARI,
5848 #endif
5849                 '\0'
5850         };
5851         const char *reject = spclchars;
5852         int quotes = flags & (EXP_FULL | EXP_CASE | EXP_REDIR); /* do CTLESC */
5853         int breakall = flags & EXP_WORD;
5854         int inquotes;
5855         size_t length;
5856         int startloc;
5857
5858         if (!(flags & EXP_VARTILDE)) {
5859                 reject += 2;
5860         } else if (flags & EXP_VARTILDE2) {
5861                 reject++;
5862         }
5863         inquotes = 0;
5864         length = 0;
5865         if (flags & EXP_TILDE) {
5866                 char *q;
5867
5868                 flags &= ~EXP_TILDE;
5869  tilde:
5870                 q = p;
5871                 if (*q == CTLESC && (flags & EXP_QWORD))
5872                         q++;
5873                 if (*q == '~')
5874                         p = exptilde(p, q, flags);
5875         }
5876  start:
5877         startloc = expdest - (char *)stackblock();
5878         for (;;) {
5879                 unsigned char c;
5880
5881                 length += strcspn(p + length, reject);
5882                 c = p[length];
5883                 if (c) {
5884                         if (!(c & 0x80)
5885 #if ENABLE_SH_MATH_SUPPORT
5886                          || c == CTLENDARI
5887 #endif
5888                         ) {
5889                                 /* c == '=' || c == ':' || c == CTLENDARI */
5890                                 length++;
5891                         }
5892                 }
5893                 if (length > 0) {
5894                         int newloc;
5895                         expdest = stack_nputstr(p, length, expdest);
5896                         newloc = expdest - (char *)stackblock();
5897                         if (breakall && !inquotes && newloc > startloc) {
5898                                 recordregion(startloc, newloc, 0);
5899                         }
5900                         startloc = newloc;
5901                 }
5902                 p += length + 1;
5903                 length = 0;
5904
5905                 switch (c) {
5906                 case '\0':
5907                         goto breakloop;
5908                 case '=':
5909                         if (flags & EXP_VARTILDE2) {
5910                                 p--;
5911                                 continue;
5912                         }
5913                         flags |= EXP_VARTILDE2;
5914                         reject++;
5915                         /* fall through */
5916                 case ':':
5917                         /*
5918                          * sort of a hack - expand tildes in variable
5919                          * assignments (after the first '=' and after ':'s).
5920                          */
5921                         if (*--p == '~') {
5922                                 goto tilde;
5923                         }
5924                         continue;
5925                 }
5926
5927                 switch (c) {
5928                 case CTLENDVAR: /* ??? */
5929                         goto breakloop;
5930                 case CTLQUOTEMARK:
5931                         /* "$@" syntax adherence hack */
5932                         if (!inquotes
5933                          && memcmp(p, dolatstr, 4) == 0
5934                          && (  p[4] == CTLQUOTEMARK
5935                             || (p[4] == CTLENDVAR && p[5] == CTLQUOTEMARK)
5936                             )
5937                         ) {
5938                                 p = evalvar(p + 1, flags, /* var_str_list: */ NULL) + 1;
5939                                 goto start;
5940                         }
5941                         inquotes = !inquotes;
5942  addquote:
5943                         if (quotes) {
5944                                 p--;
5945                                 length++;
5946                                 startloc++;
5947                         }
5948                         break;
5949                 case CTLESC:
5950                         startloc++;
5951                         length++;
5952                         goto addquote;
5953                 case CTLVAR:
5954                         p = evalvar(p, flags, var_str_list);
5955                         goto start;
5956                 case CTLBACKQ:
5957                         c = '\0';
5958                 case CTLBACKQ|CTLQUOTE:
5959                         expbackq(argbackq->n, c, quotes);
5960                         argbackq = argbackq->next;
5961                         goto start;
5962 #if ENABLE_SH_MATH_SUPPORT
5963                 case CTLENDARI:
5964                         p--;
5965                         expari(quotes);
5966                         goto start;
5967 #endif
5968                 }
5969         }
5970  breakloop:
5971         ;
5972 }
5973
5974 static char *
5975 scanleft(char *startp, char *rmesc, char *rmescend UNUSED_PARAM, char *str, int quotes,
5976         int zero)
5977 {
5978 // This commented out code was added by James Simmons <jsimmons@infradead.org>
5979 // as part of a larger change when he added support for ${var/a/b}.
5980 // However, it broke # and % operators:
5981 //
5982 //var=ababcdcd
5983 //                 ok       bad
5984 //echo ${var#ab}   abcdcd   abcdcd
5985 //echo ${var##ab}  abcdcd   abcdcd
5986 //echo ${var#a*b}  abcdcd   ababcdcd  (!)
5987 //echo ${var##a*b} cdcd     cdcd
5988 //echo ${var#?}    babcdcd  ababcdcd  (!)
5989 //echo ${var##?}   babcdcd  babcdcd
5990 //echo ${var#*}    ababcdcd babcdcd   (!)
5991 //echo ${var##*}
5992 //echo ${var%cd}   ababcd   ababcd
5993 //echo ${var%%cd}  ababcd   abab      (!)
5994 //echo ${var%c*d}  ababcd   ababcd
5995 //echo ${var%%c*d} abab     ababcdcd  (!)
5996 //echo ${var%?}    ababcdc  ababcdc
5997 //echo ${var%%?}   ababcdc  ababcdcd  (!)
5998 //echo ${var%*}    ababcdcd ababcdcd
5999 //echo ${var%%*}
6000 //
6001 // Commenting it back out helped. Remove it completely if it really
6002 // is not needed.
6003
6004         char *loc, *loc2; //, *full;
6005         char c;
6006
6007         loc = startp;
6008         loc2 = rmesc;
6009         do {
6010                 int match; // = strlen(str);
6011                 const char *s = loc2;
6012
6013                 c = *loc2;
6014                 if (zero) {
6015                         *loc2 = '\0';
6016                         s = rmesc;
6017                 }
6018                 match = pmatch(str, s); // this line was deleted
6019
6020 //              // chop off end if its '*'
6021 //              full = strrchr(str, '*');
6022 //              if (full && full != str)
6023 //                      match--;
6024 //
6025 //              // If str starts with '*' replace with s.
6026 //              if ((*str == '*') && strlen(s) >= match) {
6027 //                      full = xstrdup(s);
6028 //                      strncpy(full+strlen(s)-match+1, str+1, match-1);
6029 //              } else
6030 //                      full = xstrndup(str, match);
6031 //              match = strncmp(s, full, strlen(full));
6032 //              free(full);
6033 //
6034                 *loc2 = c;
6035                 if (match) // if (!match)
6036                         return loc;
6037                 if (quotes && (unsigned char)*loc == CTLESC)
6038                         loc++;
6039                 loc++;
6040                 loc2++;
6041         } while (c);
6042         return 0;
6043 }
6044
6045 static char *
6046 scanright(char *startp, char *rmesc, char *rmescend, char *str, int quotes,
6047         int zero)
6048 {
6049         int esc = 0;
6050         char *loc;
6051         char *loc2;
6052
6053         for (loc = str - 1, loc2 = rmescend; loc >= startp; loc2--) {
6054                 int match;
6055                 char c = *loc2;
6056                 const char *s = loc2;
6057                 if (zero) {
6058                         *loc2 = '\0';
6059                         s = rmesc;
6060                 }
6061                 match = pmatch(str, s);
6062                 *loc2 = c;
6063                 if (match)
6064                         return loc;
6065                 loc--;
6066                 if (quotes) {
6067                         if (--esc < 0) {
6068                                 esc = esclen(startp, loc);
6069                         }
6070                         if (esc % 2) {
6071                                 esc--;
6072                                 loc--;
6073                         }
6074                 }
6075         }
6076         return 0;
6077 }
6078
6079 static void varunset(const char *, const char *, const char *, int) NORETURN;
6080 static void
6081 varunset(const char *end, const char *var, const char *umsg, int varflags)
6082 {
6083         const char *msg;
6084         const char *tail;
6085
6086         tail = nullstr;
6087         msg = "parameter not set";
6088         if (umsg) {
6089                 if ((unsigned char)*end == CTLENDVAR) {
6090                         if (varflags & VSNUL)
6091                                 tail = " or null";
6092                 } else {
6093                         msg = umsg;
6094                 }
6095         }
6096         ash_msg_and_raise_error("%.*s: %s%s", end - var - 1, var, msg, tail);
6097 }
6098
6099 #if ENABLE_ASH_BASH_COMPAT
6100 static char *
6101 parse_sub_pattern(char *arg, int inquotes)
6102 {
6103         char *idx, *repl = NULL;
6104         unsigned char c;
6105
6106         idx = arg;
6107         while (1) {
6108                 c = *arg;
6109                 if (!c)
6110                         break;
6111                 if (c == '/') {
6112                         /* Only the first '/' seen is our separator */
6113                         if (!repl) {
6114                                 repl = idx + 1;
6115                                 c = '\0';
6116                         }
6117                 }
6118                 *idx++ = c;
6119                 if (!inquotes && c == '\\' && arg[1] == '\\')
6120                         arg++; /* skip both \\, not just first one */
6121                 arg++;
6122         }
6123         *idx = c; /* NUL */
6124
6125         return repl;
6126 }
6127 #endif /* ENABLE_ASH_BASH_COMPAT */
6128
6129 static const char *
6130 subevalvar(char *p, char *str, int strloc, int subtype,
6131                 int startloc, int varflags, int quotes, struct strlist *var_str_list)
6132 {
6133         struct nodelist *saveargbackq = argbackq;
6134         char *startp;
6135         char *loc;
6136         char *rmesc, *rmescend;
6137         IF_ASH_BASH_COMPAT(char *repl = NULL;)
6138         IF_ASH_BASH_COMPAT(char null = '\0';)
6139         IF_ASH_BASH_COMPAT(int pos, len, orig_len;)
6140         int saveherefd = herefd;
6141         int amount, workloc, resetloc;
6142         int zero;
6143         char *(*scan)(char*, char*, char*, char*, int, int);
6144
6145         herefd = -1;
6146         argstr(p, (subtype != VSASSIGN && subtype != VSQUESTION) ? EXP_CASE : 0,
6147                         var_str_list);
6148         STPUTC('\0', expdest);
6149         herefd = saveherefd;
6150         argbackq = saveargbackq;
6151         startp = (char *)stackblock() + startloc;
6152
6153         switch (subtype) {
6154         case VSASSIGN:
6155                 setvar(str, startp, 0);
6156                 amount = startp - expdest;
6157                 STADJUST(amount, expdest);
6158                 return startp;
6159
6160 #if ENABLE_ASH_BASH_COMPAT
6161         case VSSUBSTR:
6162                 loc = str = stackblock() + strloc;
6163                 /* Read POS in ${var:POS:LEN} */
6164                 pos = atoi(loc); /* number(loc) errors out on "1:4" */
6165                 len = str - startp - 1;
6166
6167                 /* *loc != '\0', guaranteed by parser */
6168                 if (quotes) {
6169                         char *ptr;
6170
6171                         /* Adjust the length by the number of escapes */
6172                         for (ptr = startp; ptr < (str - 1); ptr++) {
6173                                 if ((unsigned char)*ptr == CTLESC) {
6174                                         len--;
6175                                         ptr++;
6176                                 }
6177                         }
6178                 }
6179                 orig_len = len;
6180
6181                 if (*loc++ == ':') {
6182                         /* ${var::LEN} */
6183                         len = number(loc);
6184                 } else {
6185                         /* Skip POS in ${var:POS:LEN} */
6186                         len = orig_len;
6187                         while (*loc && *loc != ':') {
6188                                 /* TODO?
6189                                  * bash complains on: var=qwe; echo ${var:1a:123}
6190                                 if (!isdigit(*loc))
6191                                         ash_msg_and_raise_error(msg_illnum, str);
6192                                  */
6193                                 loc++;
6194                         }
6195                         if (*loc++ == ':') {
6196                                 len = number(loc);
6197                         }
6198                 }
6199                 if (pos >= orig_len) {
6200                         pos = 0;
6201                         len = 0;
6202                 }
6203                 if (len > (orig_len - pos))
6204                         len = orig_len - pos;
6205
6206                 for (str = startp; pos; str++, pos--) {
6207                         if (quotes && (unsigned char)*str == CTLESC)
6208                                 str++;
6209                 }
6210                 for (loc = startp; len; len--) {
6211                         if (quotes && (unsigned char)*str == CTLESC)
6212                                 *loc++ = *str++;
6213                         *loc++ = *str++;
6214                 }
6215                 *loc = '\0';
6216                 amount = loc - expdest;
6217                 STADJUST(amount, expdest);
6218                 return loc;
6219 #endif
6220
6221         case VSQUESTION:
6222                 varunset(p, str, startp, varflags);
6223                 /* NOTREACHED */
6224         }
6225         resetloc = expdest - (char *)stackblock();
6226
6227         /* We'll comeback here if we grow the stack while handling
6228          * a VSREPLACE or VSREPLACEALL, since our pointers into the
6229          * stack will need rebasing, and we'll need to remove our work
6230          * areas each time
6231          */
6232  IF_ASH_BASH_COMPAT(restart:)
6233
6234         amount = expdest - ((char *)stackblock() + resetloc);
6235         STADJUST(-amount, expdest);
6236         startp = (char *)stackblock() + startloc;
6237
6238         rmesc = startp;
6239         rmescend = (char *)stackblock() + strloc;
6240         if (quotes) {
6241                 rmesc = rmescapes(startp, RMESCAPE_ALLOC | RMESCAPE_GROW);
6242                 if (rmesc != startp) {
6243                         rmescend = expdest;
6244                         startp = (char *)stackblock() + startloc;
6245                 }
6246         }
6247         rmescend--;
6248         str = (char *)stackblock() + strloc;
6249         preglob(str, varflags & VSQUOTE, 0);
6250         workloc = expdest - (char *)stackblock();
6251
6252 #if ENABLE_ASH_BASH_COMPAT
6253         if (subtype == VSREPLACE || subtype == VSREPLACEALL) {
6254                 char *idx, *end, *restart_detect;
6255
6256                 if (!repl) {
6257                         repl = parse_sub_pattern(str, varflags & VSQUOTE);
6258                         if (!repl)
6259                                 repl = &null;
6260                 }
6261
6262                 /* If there's no pattern to match, return the expansion unmolested */
6263                 if (*str == '\0')
6264                         return 0;
6265
6266                 len = 0;
6267                 idx = startp;
6268                 end = str - 1;
6269                 while (idx < end) {
6270                         loc = scanright(idx, rmesc, rmescend, str, quotes, 1);
6271                         if (!loc) {
6272                                 /* No match, advance */
6273                                 restart_detect = stackblock();
6274                                 STPUTC(*idx, expdest);
6275                                 if (quotes && (unsigned char)*idx == CTLESC) {
6276                                         idx++;
6277                                         len++;
6278                                         STPUTC(*idx, expdest);
6279                                 }
6280                                 if (stackblock() != restart_detect)
6281                                         goto restart;
6282                                 idx++;
6283                                 len++;
6284                                 rmesc++;
6285                                 continue;
6286                         }
6287
6288                         if (subtype == VSREPLACEALL) {
6289                                 while (idx < loc) {
6290                                         if (quotes && (unsigned char)*idx == CTLESC)
6291                                                 idx++;
6292                                         idx++;
6293                                         rmesc++;
6294                                 }
6295                         } else {
6296                                 idx = loc;
6297                         }
6298
6299                         for (loc = repl; *loc; loc++) {
6300                                 restart_detect = stackblock();
6301                                 STPUTC(*loc, expdest);
6302                                 if (stackblock() != restart_detect)
6303                                         goto restart;
6304                                 len++;
6305                         }
6306
6307                         if (subtype == VSREPLACE) {
6308                                 while (*idx) {
6309                                         restart_detect = stackblock();
6310                                         STPUTC(*idx, expdest);
6311                                         if (stackblock() != restart_detect)
6312                                                 goto restart;
6313                                         len++;
6314                                         idx++;
6315                                 }
6316                                 break;
6317                         }
6318                 }
6319
6320                 /* We've put the replaced text into a buffer at workloc, now
6321                  * move it to the right place and adjust the stack.
6322                  */
6323                 startp = stackblock() + startloc;
6324                 STPUTC('\0', expdest);
6325                 memmove(startp, stackblock() + workloc, len);
6326                 startp[len++] = '\0';
6327                 amount = expdest - ((char *)stackblock() + startloc + len - 1);
6328                 STADJUST(-amount, expdest);
6329                 return startp;
6330         }
6331 #endif /* ENABLE_ASH_BASH_COMPAT */
6332
6333         subtype -= VSTRIMRIGHT;
6334 #if DEBUG
6335         if (subtype < 0 || subtype > 7)
6336                 abort();
6337 #endif
6338         /* zero = subtype == VSTRIMLEFT || subtype == VSTRIMLEFTMAX */
6339         zero = subtype >> 1;
6340         /* VSTRIMLEFT/VSTRIMRIGHTMAX -> scanleft */
6341         scan = (subtype & 1) ^ zero ? scanleft : scanright;
6342
6343         loc = scan(startp, rmesc, rmescend, str, quotes, zero);
6344         if (loc) {
6345                 if (zero) {
6346                         memmove(startp, loc, str - loc);
6347                         loc = startp + (str - loc) - 1;
6348                 }
6349                 *loc = '\0';
6350                 amount = loc - expdest;
6351                 STADJUST(amount, expdest);
6352         }
6353         return loc;
6354 }
6355
6356 /*
6357  * Add the value of a specialized variable to the stack string.
6358  * name parameter (examples):
6359  * ash -c 'echo $1'      name:'1='
6360  * ash -c 'echo $qwe'    name:'qwe='
6361  * ash -c 'echo $$'      name:'$='
6362  * ash -c 'echo ${$}'    name:'$='
6363  * ash -c 'echo ${$##q}' name:'$=q'
6364  * ash -c 'echo ${#$}'   name:'$='
6365  * note: examples with bad shell syntax:
6366  * ash -c 'echo ${#$1}'  name:'$=1'
6367  * ash -c 'echo ${#1#}'  name:'1=#'
6368  */
6369 static NOINLINE ssize_t
6370 varvalue(char *name, int varflags, int flags, struct strlist *var_str_list)
6371 {
6372         const char *p;
6373         int num;
6374         int i;
6375         int sepq = 0;
6376         ssize_t len = 0;
6377         int subtype = varflags & VSTYPE;
6378         int quotes = flags & (EXP_FULL | EXP_CASE | EXP_REDIR);
6379         int quoted = varflags & VSQUOTE;
6380         int syntax = quoted ? DQSYNTAX : BASESYNTAX;
6381
6382         switch (*name) {
6383         case '$':
6384                 num = rootpid;
6385                 goto numvar;
6386         case '?':
6387                 num = exitstatus;
6388                 goto numvar;
6389         case '#':
6390                 num = shellparam.nparam;
6391                 goto numvar;
6392         case '!':
6393                 num = backgndpid;
6394                 if (num == 0)
6395                         return -1;
6396  numvar:
6397                 len = cvtnum(num);
6398                 goto check_1char_name;
6399         case '-':
6400                 expdest = makestrspace(NOPTS, expdest);
6401                 for (i = NOPTS - 1; i >= 0; i--) {
6402                         if (optlist[i]) {
6403                                 USTPUTC(optletters(i), expdest);
6404                                 len++;
6405                         }
6406                 }
6407  check_1char_name:
6408 #if 0
6409                 /* handles cases similar to ${#$1} */
6410                 if (name[2] != '\0')
6411                         raise_error_syntax("bad substitution");
6412 #endif
6413                 break;
6414         case '@': {
6415                 char **ap;
6416                 int sep;
6417
6418                 if (quoted && (flags & EXP_FULL)) {
6419                         /* note: this is not meant as PEOF value */
6420                         sep = 1 << CHAR_BIT;
6421                         goto param;
6422                 }
6423                 /* fall through */
6424         case '*':
6425                 sep = ifsset() ? (unsigned char)(ifsval()[0]) : ' ';
6426                 i = SIT(sep, syntax);
6427                 if (quotes && (i == CCTL || i == CBACK))
6428                         sepq = 1;
6429  param:
6430                 ap = shellparam.p;
6431                 if (!ap)
6432                         return -1;
6433                 while ((p = *ap++) != NULL) {
6434                         size_t partlen;
6435
6436                         partlen = strlen(p);
6437                         len += partlen;
6438
6439                         if (!(subtype == VSPLUS || subtype == VSLENGTH))
6440                                 memtodest(p, partlen, syntax, quotes);
6441
6442                         if (*ap && sep) {
6443                                 char *q;
6444
6445                                 len++;
6446                                 if (subtype == VSPLUS || subtype == VSLENGTH) {
6447                                         continue;
6448                                 }
6449                                 q = expdest;
6450                                 if (sepq)
6451                                         STPUTC(CTLESC, q);
6452                                 /* note: may put NUL despite sep != 0
6453                                  * (see sep = 1 << CHAR_BIT above) */
6454                                 STPUTC(sep, q);
6455                                 expdest = q;
6456                         }
6457                 }
6458                 return len;
6459         } /* case '@' and '*' */
6460         case '0':
6461         case '1':
6462         case '2':
6463         case '3':
6464         case '4':
6465         case '5':
6466         case '6':
6467         case '7':
6468         case '8':
6469         case '9':
6470                 num = atoi(name); /* number(name) fails on ${N#str} etc */
6471                 if (num < 0 || num > shellparam.nparam)
6472                         return -1;
6473                 p = num ? shellparam.p[num - 1] : arg0;
6474                 goto value;
6475         default:
6476                 /* NB: name has form "VAR=..." */
6477
6478                 /* "A=a B=$A" case: var_str_list is a list of "A=a" strings
6479                  * which should be considered before we check variables. */
6480                 if (var_str_list) {
6481                         unsigned name_len = (strchrnul(name, '=') - name) + 1;
6482                         p = NULL;
6483                         do {
6484                                 char *str, *eq;
6485                                 str = var_str_list->text;
6486                                 eq = strchr(str, '=');
6487                                 if (!eq) /* stop at first non-assignment */
6488                                         break;
6489                                 eq++;
6490                                 if (name_len == (unsigned)(eq - str)
6491                                  && strncmp(str, name, name_len) == 0
6492                                 ) {
6493                                         p = eq;
6494                                         /* goto value; - WRONG! */
6495                                         /* think "A=1 A=2 B=$A" */
6496                                 }
6497                                 var_str_list = var_str_list->next;
6498                         } while (var_str_list);
6499                         if (p)
6500                                 goto value;
6501                 }
6502                 p = lookupvar(name);
6503  value:
6504                 if (!p)
6505                         return -1;
6506
6507                 len = strlen(p);
6508                 if (!(subtype == VSPLUS || subtype == VSLENGTH))
6509                         memtodest(p, len, syntax, quotes);
6510                 return len;
6511         }
6512
6513         if (subtype == VSPLUS || subtype == VSLENGTH)
6514                 STADJUST(-len, expdest);
6515         return len;
6516 }
6517
6518 /*
6519  * Expand a variable, and return a pointer to the next character in the
6520  * input string.
6521  */
6522 static char *
6523 evalvar(char *p, int flags, struct strlist *var_str_list)
6524 {
6525         char varflags;
6526         char subtype;
6527         char quoted;
6528         char easy;
6529         char *var;
6530         int patloc;
6531         int startloc;
6532         ssize_t varlen;
6533
6534         varflags = (unsigned char) *p++;
6535         subtype = varflags & VSTYPE;
6536         quoted = varflags & VSQUOTE;
6537         var = p;
6538         easy = (!quoted || (*var == '@' && shellparam.nparam));
6539         startloc = expdest - (char *)stackblock();
6540         p = strchr(p, '=') + 1;
6541
6542  again:
6543         varlen = varvalue(var, varflags, flags, var_str_list);
6544         if (varflags & VSNUL)
6545                 varlen--;
6546
6547         if (subtype == VSPLUS) {
6548                 varlen = -1 - varlen;
6549                 goto vsplus;
6550         }
6551
6552         if (subtype == VSMINUS) {
6553  vsplus:
6554                 if (varlen < 0) {
6555                         argstr(
6556                                 p, flags | EXP_TILDE |
6557                                         (quoted ? EXP_QWORD : EXP_WORD),
6558                                 var_str_list
6559                         );
6560                         goto end;
6561                 }
6562                 if (easy)
6563                         goto record;
6564                 goto end;
6565         }
6566
6567         if (subtype == VSASSIGN || subtype == VSQUESTION) {
6568                 if (varlen < 0) {
6569                         if (subevalvar(p, var, /* strloc: */ 0,
6570                                         subtype, startloc, varflags,
6571                                         /* quotes: */ 0,
6572                                         var_str_list)
6573                         ) {
6574                                 varflags &= ~VSNUL;
6575                                 /*
6576                                  * Remove any recorded regions beyond
6577                                  * start of variable
6578                                  */
6579                                 removerecordregions(startloc);
6580                                 goto again;
6581                         }
6582                         goto end;
6583                 }
6584                 if (easy)
6585                         goto record;
6586                 goto end;
6587         }
6588
6589         if (varlen < 0 && uflag)
6590                 varunset(p, var, 0, 0);
6591
6592         if (subtype == VSLENGTH) {
6593                 cvtnum(varlen > 0 ? varlen : 0);
6594                 goto record;
6595         }
6596
6597         if (subtype == VSNORMAL) {
6598                 if (easy)
6599                         goto record;
6600                 goto end;
6601         }
6602
6603 #if DEBUG
6604         switch (subtype) {
6605         case VSTRIMLEFT:
6606         case VSTRIMLEFTMAX:
6607         case VSTRIMRIGHT:
6608         case VSTRIMRIGHTMAX:
6609 #if ENABLE_ASH_BASH_COMPAT
6610         case VSSUBSTR:
6611         case VSREPLACE:
6612         case VSREPLACEALL:
6613 #endif
6614                 break;
6615         default:
6616                 abort();
6617         }
6618 #endif
6619
6620         if (varlen >= 0) {
6621                 /*
6622                  * Terminate the string and start recording the pattern
6623                  * right after it
6624                  */
6625                 STPUTC('\0', expdest);
6626                 patloc = expdest - (char *)stackblock();
6627                 if (0 == subevalvar(p, /* str: */ NULL, patloc, subtype,
6628                                 startloc, varflags,
6629 //TODO: | EXP_REDIR too? All other such places do it too
6630                                 /* quotes: */ flags & (EXP_FULL | EXP_CASE),
6631                                 var_str_list)
6632                 ) {
6633                         int amount = expdest - (
6634                                 (char *)stackblock() + patloc - 1
6635                         );
6636                         STADJUST(-amount, expdest);
6637                 }
6638                 /* Remove any recorded regions beyond start of variable */
6639                 removerecordregions(startloc);
6640  record:
6641                 recordregion(startloc, expdest - (char *)stackblock(), quoted);
6642         }
6643
6644  end:
6645         if (subtype != VSNORMAL) {      /* skip to end of alternative */
6646                 int nesting = 1;
6647                 for (;;) {
6648                         unsigned char c = *p++;
6649                         if (c == CTLESC)
6650                                 p++;
6651                         else if (c == CTLBACKQ || c == (CTLBACKQ|CTLQUOTE)) {
6652                                 if (varlen >= 0)
6653                                         argbackq = argbackq->next;
6654                         } else if (c == CTLVAR) {
6655                                 if ((*p++ & VSTYPE) != VSNORMAL)
6656                                         nesting++;
6657                         } else if (c == CTLENDVAR) {
6658                                 if (--nesting == 0)
6659                                         break;
6660                         }
6661                 }
6662         }
6663         return p;
6664 }
6665
6666 /*
6667  * Break the argument string into pieces based upon IFS and add the
6668  * strings to the argument list.  The regions of the string to be
6669  * searched for IFS characters have been stored by recordregion.
6670  */
6671 static void
6672 ifsbreakup(char *string, struct arglist *arglist)
6673 {
6674         struct ifsregion *ifsp;
6675         struct strlist *sp;
6676         char *start;
6677         char *p;
6678         char *q;
6679         const char *ifs, *realifs;
6680         int ifsspc;
6681         int nulonly;
6682
6683         start = string;
6684         if (ifslastp != NULL) {
6685                 ifsspc = 0;
6686                 nulonly = 0;
6687                 realifs = ifsset() ? ifsval() : defifs;
6688                 ifsp = &ifsfirst;
6689                 do {
6690                         p = string + ifsp->begoff;
6691                         nulonly = ifsp->nulonly;
6692                         ifs = nulonly ? nullstr : realifs;
6693                         ifsspc = 0;
6694                         while (p < string + ifsp->endoff) {
6695                                 q = p;
6696                                 if ((unsigned char)*p == CTLESC)
6697                                         p++;
6698                                 if (!strchr(ifs, *p)) {
6699                                         p++;
6700                                         continue;
6701                                 }
6702                                 if (!nulonly)
6703                                         ifsspc = (strchr(defifs, *p) != NULL);
6704                                 /* Ignore IFS whitespace at start */
6705                                 if (q == start && ifsspc) {
6706                                         p++;
6707                                         start = p;
6708                                         continue;
6709                                 }
6710                                 *q = '\0';
6711                                 sp = stzalloc(sizeof(*sp));
6712                                 sp->text = start;
6713                                 *arglist->lastp = sp;
6714                                 arglist->lastp = &sp->next;
6715                                 p++;
6716                                 if (!nulonly) {
6717                                         for (;;) {
6718                                                 if (p >= string + ifsp->endoff) {
6719                                                         break;
6720                                                 }
6721                                                 q = p;
6722                                                 if ((unsigned char)*p == CTLESC)
6723                                                         p++;
6724                                                 if (strchr(ifs, *p) == NULL) {
6725                                                         p = q;
6726                                                         break;
6727                                                 }
6728                                                 if (strchr(defifs, *p) == NULL) {
6729                                                         if (ifsspc) {
6730                                                                 p++;
6731                                                                 ifsspc = 0;
6732                                                         } else {
6733                                                                 p = q;
6734                                                                 break;
6735                                                         }
6736                                                 } else
6737                                                         p++;
6738                                         }
6739                                 }
6740                                 start = p;
6741                         } /* while */
6742                         ifsp = ifsp->next;
6743                 } while (ifsp != NULL);
6744                 if (nulonly)
6745                         goto add;
6746         }
6747
6748         if (!*start)
6749                 return;
6750
6751  add:
6752         sp = stzalloc(sizeof(*sp));
6753         sp->text = start;
6754         *arglist->lastp = sp;
6755         arglist->lastp = &sp->next;
6756 }
6757
6758 static void
6759 ifsfree(void)
6760 {
6761         struct ifsregion *p;
6762
6763         INT_OFF;
6764         p = ifsfirst.next;
6765         do {
6766                 struct ifsregion *ifsp;
6767                 ifsp = p->next;
6768                 free(p);
6769                 p = ifsp;
6770         } while (p);
6771         ifslastp = NULL;
6772         ifsfirst.next = NULL;
6773         INT_ON;
6774 }
6775
6776 /*
6777  * Add a file name to the list.
6778  */
6779 static void
6780 addfname(const char *name)
6781 {
6782         struct strlist *sp;
6783
6784         sp = stzalloc(sizeof(*sp));
6785         sp->text = ststrdup(name);
6786         *exparg.lastp = sp;
6787         exparg.lastp = &sp->next;
6788 }
6789
6790 static char *expdir;
6791
6792 /*
6793  * Do metacharacter (i.e. *, ?, [...]) expansion.
6794  */
6795 static void
6796 expmeta(char *enddir, char *name)
6797 {
6798         char *p;
6799         const char *cp;
6800         char *start;
6801         char *endname;
6802         int metaflag;
6803         struct stat statb;
6804         DIR *dirp;
6805         struct dirent *dp;
6806         int atend;
6807         int matchdot;
6808
6809         metaflag = 0;
6810         start = name;
6811         for (p = name; *p; p++) {
6812                 if (*p == '*' || *p == '?')
6813                         metaflag = 1;
6814                 else if (*p == '[') {
6815                         char *q = p + 1;
6816                         if (*q == '!')
6817                                 q++;
6818                         for (;;) {
6819                                 if (*q == '\\')
6820                                         q++;
6821                                 if (*q == '/' || *q == '\0')
6822                                         break;
6823                                 if (*++q == ']') {
6824                                         metaflag = 1;
6825                                         break;
6826                                 }
6827                         }
6828                 } else if (*p == '\\')
6829                         p++;
6830                 else if (*p == '/') {
6831                         if (metaflag)
6832                                 goto out;
6833                         start = p + 1;
6834                 }
6835         }
6836  out:
6837         if (metaflag == 0) {    /* we've reached the end of the file name */
6838                 if (enddir != expdir)
6839                         metaflag++;
6840                 p = name;
6841                 do {
6842                         if (*p == '\\')
6843                                 p++;
6844                         *enddir++ = *p;
6845                 } while (*p++);
6846                 if (metaflag == 0 || lstat(expdir, &statb) >= 0)
6847                         addfname(expdir);
6848                 return;
6849         }
6850         endname = p;
6851         if (name < start) {
6852                 p = name;
6853                 do {
6854                         if (*p == '\\')
6855                                 p++;
6856                         *enddir++ = *p++;
6857                 } while (p < start);
6858         }
6859         if (enddir == expdir) {
6860                 cp = ".";
6861         } else if (enddir == expdir + 1 && *expdir == '/') {
6862                 cp = "/";
6863         } else {
6864                 cp = expdir;
6865                 enddir[-1] = '\0';
6866         }
6867         dirp = opendir(cp);
6868         if (dirp == NULL)
6869                 return;
6870         if (enddir != expdir)
6871                 enddir[-1] = '/';
6872         if (*endname == 0) {
6873                 atend = 1;
6874         } else {
6875                 atend = 0;
6876                 *endname++ = '\0';
6877         }
6878         matchdot = 0;
6879         p = start;
6880         if (*p == '\\')
6881                 p++;
6882         if (*p == '.')
6883                 matchdot++;
6884         while (!pending_int && (dp = readdir(dirp)) != NULL) {
6885                 if (dp->d_name[0] == '.' && !matchdot)
6886                         continue;
6887                 if (pmatch(start, dp->d_name)) {
6888                         if (atend) {
6889                                 strcpy(enddir, dp->d_name);
6890                                 addfname(expdir);
6891                         } else {
6892                                 for (p = enddir, cp = dp->d_name; (*p++ = *cp++) != '\0';)
6893                                         continue;
6894                                 p[-1] = '/';
6895                                 expmeta(p, endname);
6896                         }
6897                 }
6898         }
6899         closedir(dirp);
6900         if (!atend)
6901                 endname[-1] = '/';
6902 }
6903
6904 static struct strlist *
6905 msort(struct strlist *list, int len)
6906 {
6907         struct strlist *p, *q = NULL;
6908         struct strlist **lpp;
6909         int half;
6910         int n;
6911
6912         if (len <= 1)
6913                 return list;
6914         half = len >> 1;
6915         p = list;
6916         for (n = half; --n >= 0;) {
6917                 q = p;
6918                 p = p->next;
6919         }
6920         q->next = NULL;                 /* terminate first half of list */
6921         q = msort(list, half);          /* sort first half of list */
6922         p = msort(p, len - half);               /* sort second half */
6923         lpp = &list;
6924         for (;;) {
6925 #if ENABLE_LOCALE_SUPPORT
6926                 if (strcoll(p->text, q->text) < 0)
6927 #else
6928                 if (strcmp(p->text, q->text) < 0)
6929 #endif
6930                                                 {
6931                         *lpp = p;
6932                         lpp = &p->next;
6933                         p = *lpp;
6934                         if (p == NULL) {
6935                                 *lpp = q;
6936                                 break;
6937                         }
6938                 } else {
6939                         *lpp = q;
6940                         lpp = &q->next;
6941                         q = *lpp;
6942                         if (q == NULL) {
6943                                 *lpp = p;
6944                                 break;
6945                         }
6946                 }
6947         }
6948         return list;
6949 }
6950
6951 /*
6952  * Sort the results of file name expansion.  It calculates the number of
6953  * strings to sort and then calls msort (short for merge sort) to do the
6954  * work.
6955  */
6956 static struct strlist *
6957 expsort(struct strlist *str)
6958 {
6959         int len;
6960         struct strlist *sp;
6961
6962         len = 0;
6963         for (sp = str; sp; sp = sp->next)
6964                 len++;
6965         return msort(str, len);
6966 }
6967
6968 static void
6969 expandmeta(struct strlist *str /*, int flag*/)
6970 {
6971         static const char metachars[] ALIGN1 = {
6972                 '*', '?', '[', 0
6973         };
6974         /* TODO - EXP_REDIR */
6975
6976         while (str) {
6977                 struct strlist **savelastp;
6978                 struct strlist *sp;
6979                 char *p;
6980
6981                 if (fflag)
6982                         goto nometa;
6983                 if (!strpbrk(str->text, metachars))
6984                         goto nometa;
6985                 savelastp = exparg.lastp;
6986
6987                 INT_OFF;
6988                 p = preglob(str->text, 0, RMESCAPE_ALLOC | RMESCAPE_HEAP);
6989                 {
6990                         int i = strlen(str->text);
6991                         expdir = ckmalloc(i < 2048 ? 2048 : i); /* XXX */
6992                 }
6993
6994                 expmeta(expdir, p);
6995                 free(expdir);
6996                 if (p != str->text)
6997                         free(p);
6998                 INT_ON;
6999                 if (exparg.lastp == savelastp) {
7000                         /*
7001                          * no matches
7002                          */
7003  nometa:
7004                         *exparg.lastp = str;
7005                         rmescapes(str->text, 0);
7006                         exparg.lastp = &str->next;
7007                 } else {
7008                         *exparg.lastp = NULL;
7009                         *savelastp = sp = expsort(*savelastp);
7010                         while (sp->next != NULL)
7011                                 sp = sp->next;
7012                         exparg.lastp = &sp->next;
7013                 }
7014                 str = str->next;
7015         }
7016 }
7017
7018 /*
7019  * Perform variable substitution and command substitution on an argument,
7020  * placing the resulting list of arguments in arglist.  If EXP_FULL is true,
7021  * perform splitting and file name expansion.  When arglist is NULL, perform
7022  * here document expansion.
7023  */
7024 static void
7025 expandarg(union node *arg, struct arglist *arglist, int flag)
7026 {
7027         struct strlist *sp;
7028         char *p;
7029
7030         argbackq = arg->narg.backquote;
7031         STARTSTACKSTR(expdest);
7032         ifsfirst.next = NULL;
7033         ifslastp = NULL;
7034         argstr(arg->narg.text, flag,
7035                         /* var_str_list: */ arglist ? arglist->list : NULL);
7036         p = _STPUTC('\0', expdest);
7037         expdest = p - 1;
7038         if (arglist == NULL) {
7039                 return;                 /* here document expanded */
7040         }
7041         p = grabstackstr(p);
7042         exparg.lastp = &exparg.list;
7043         /*
7044          * TODO - EXP_REDIR
7045          */
7046         if (flag & EXP_FULL) {
7047                 ifsbreakup(p, &exparg);
7048                 *exparg.lastp = NULL;
7049                 exparg.lastp = &exparg.list;
7050                 expandmeta(exparg.list /*, flag*/);
7051         } else {
7052                 if (flag & EXP_REDIR) /*XXX - for now, just remove escapes */
7053                         rmescapes(p, 0);
7054                 sp = stzalloc(sizeof(*sp));
7055                 sp->text = p;
7056                 *exparg.lastp = sp;
7057                 exparg.lastp = &sp->next;
7058         }
7059         if (ifsfirst.next)
7060                 ifsfree();
7061         *exparg.lastp = NULL;
7062         if (exparg.list) {
7063                 *arglist->lastp = exparg.list;
7064                 arglist->lastp = exparg.lastp;
7065         }
7066 }
7067
7068 /*
7069  * Expand shell variables and backquotes inside a here document.
7070  */
7071 static void
7072 expandhere(union node *arg, int fd)
7073 {
7074         herefd = fd;
7075         expandarg(arg, (struct arglist *)NULL, 0);
7076         full_write(fd, stackblock(), expdest - (char *)stackblock());
7077 }
7078
7079 /*
7080  * Returns true if the pattern matches the string.
7081  */
7082 static int
7083 patmatch(char *pattern, const char *string)
7084 {
7085         return pmatch(preglob(pattern, 0, 0), string);
7086 }
7087
7088 /*
7089  * See if a pattern matches in a case statement.
7090  */
7091 static int
7092 casematch(union node *pattern, char *val)
7093 {
7094         struct stackmark smark;
7095         int result;
7096
7097         setstackmark(&smark);
7098         argbackq = pattern->narg.backquote;
7099         STARTSTACKSTR(expdest);
7100         ifslastp = NULL;
7101         argstr(pattern->narg.text, EXP_TILDE | EXP_CASE,
7102                         /* var_str_list: */ NULL);
7103         STACKSTRNUL(expdest);
7104         result = patmatch(stackblock(), val);
7105         popstackmark(&smark);
7106         return result;
7107 }
7108
7109
7110 /* ============ find_command */
7111
7112 struct builtincmd {
7113         const char *name;
7114         int (*builtin)(int, char **) FAST_FUNC;
7115         /* unsigned flags; */
7116 };
7117 #define IS_BUILTIN_SPECIAL(b) ((b)->name[0] & 1)
7118 /* "regular" builtins always take precedence over commands,
7119  * regardless of PATH=....%builtin... position */
7120 #define IS_BUILTIN_REGULAR(b) ((b)->name[0] & 2)
7121 #define IS_BUILTIN_ASSIGN(b)  ((b)->name[0] & 4)
7122
7123 struct cmdentry {
7124         smallint cmdtype;       /* CMDxxx */
7125         union param {
7126                 int index;
7127                 /* index >= 0 for commands without path (slashes) */
7128                 /* (TODO: what exactly does the value mean? PATH position?) */
7129                 /* index == -1 for commands with slashes */
7130                 /* index == (-2 - applet_no) for NOFORK applets */
7131                 const struct builtincmd *cmd;
7132                 struct funcnode *func;
7133         } u;
7134 };
7135 /* values of cmdtype */
7136 #define CMDUNKNOWN      -1      /* no entry in table for command */
7137 #define CMDNORMAL       0       /* command is an executable program */
7138 #define CMDFUNCTION     1       /* command is a shell function */
7139 #define CMDBUILTIN      2       /* command is a shell builtin */
7140
7141 /* action to find_command() */
7142 #define DO_ERR          0x01    /* prints errors */
7143 #define DO_ABS          0x02    /* checks absolute paths */
7144 #define DO_NOFUNC       0x04    /* don't return shell functions, for command */
7145 #define DO_ALTPATH      0x08    /* using alternate path */
7146 #define DO_ALTBLTIN     0x20    /* %builtin in alt. path */
7147
7148 static void find_command(char *, struct cmdentry *, int, const char *);
7149
7150
7151 /* ============ Hashing commands */
7152
7153 /*
7154  * When commands are first encountered, they are entered in a hash table.
7155  * This ensures that a full path search will not have to be done for them
7156  * on each invocation.
7157  *
7158  * We should investigate converting to a linear search, even though that
7159  * would make the command name "hash" a misnomer.
7160  */
7161
7162 struct tblentry {
7163         struct tblentry *next;  /* next entry in hash chain */
7164         union param param;      /* definition of builtin function */
7165         smallint cmdtype;       /* CMDxxx */
7166         char rehash;            /* if set, cd done since entry created */
7167         char cmdname[1];        /* name of command */
7168 };
7169
7170 static struct tblentry **cmdtable;
7171 #define INIT_G_cmdtable() do { \
7172         cmdtable = xzalloc(CMDTABLESIZE * sizeof(cmdtable[0])); \
7173 } while (0)
7174
7175 static int builtinloc = -1;     /* index in path of %builtin, or -1 */
7176
7177
7178 static void
7179 tryexec(IF_FEATURE_SH_STANDALONE(int applet_no,) char *cmd, char **argv, char **envp)
7180 {
7181         int repeated = 0;
7182
7183 #if ENABLE_FEATURE_SH_STANDALONE
7184         if (applet_no >= 0) {
7185                 if (APPLET_IS_NOEXEC(applet_no)) {
7186                         while (*envp)
7187                                 putenv(*envp++);
7188                         run_applet_no_and_exit(applet_no, argv);
7189                 }
7190                 /* re-exec ourselves with the new arguments */
7191                 execve(bb_busybox_exec_path, argv, envp);
7192                 /* If they called chroot or otherwise made the binary no longer
7193                  * executable, fall through */
7194         }
7195 #endif
7196
7197  repeat:
7198 #ifdef SYSV
7199         do {
7200                 execve(cmd, argv, envp);
7201         } while (errno == EINTR);
7202 #else
7203         execve(cmd, argv, envp);
7204 #endif
7205         if (repeated) {
7206                 free(argv);
7207                 return;
7208         }
7209         if (errno == ENOEXEC) {
7210                 char **ap;
7211                 char **new;
7212
7213                 for (ap = argv; *ap; ap++)
7214                         continue;
7215                 ap = new = ckmalloc((ap - argv + 2) * sizeof(ap[0]));
7216                 ap[1] = cmd;
7217                 ap[0] = cmd = (char *)DEFAULT_SHELL;
7218                 ap += 2;
7219                 argv++;
7220                 while ((*ap++ = *argv++) != NULL)
7221                         continue;
7222                 argv = new;
7223                 repeated++;
7224                 goto repeat;
7225         }
7226 }
7227
7228 /*
7229  * Exec a program.  Never returns.  If you change this routine, you may
7230  * have to change the find_command routine as well.
7231  */
7232 static void shellexec(char **, const char *, int) NORETURN;
7233 static void
7234 shellexec(char **argv, const char *path, int idx)
7235 {
7236         char *cmdname;
7237         int e;
7238         char **envp;
7239         int exerrno;
7240 #if ENABLE_FEATURE_SH_STANDALONE
7241         int applet_no = -1;
7242 #endif
7243
7244         clearredir(/*drop:*/ 1);
7245         envp = listvars(VEXPORT, VUNSET, 0);
7246         if (strchr(argv[0], '/') != NULL
7247 #if ENABLE_FEATURE_SH_STANDALONE
7248          || (applet_no = find_applet_by_name(argv[0])) >= 0
7249 #endif
7250         ) {
7251                 tryexec(IF_FEATURE_SH_STANDALONE(applet_no,) argv[0], argv, envp);
7252                 e = errno;
7253         } else {
7254                 e = ENOENT;
7255                 while ((cmdname = path_advance(&path, argv[0])) != NULL) {
7256                         if (--idx < 0 && pathopt == NULL) {
7257                                 tryexec(IF_FEATURE_SH_STANDALONE(-1,) cmdname, argv, envp);
7258                                 if (errno != ENOENT && errno != ENOTDIR)
7259                                         e = errno;
7260                         }
7261                         stunalloc(cmdname);
7262                 }
7263         }
7264
7265         /* Map to POSIX errors */
7266         switch (e) {
7267         case EACCES:
7268                 exerrno = 126;
7269                 break;
7270         case ENOENT:
7271                 exerrno = 127;
7272                 break;
7273         default:
7274                 exerrno = 2;
7275                 break;
7276         }
7277         exitstatus = exerrno;
7278         TRACE(("shellexec failed for %s, errno %d, suppress_int %d\n",
7279                 argv[0], e, suppress_int));
7280         ash_msg_and_raise(EXEXEC, "%s: %s", argv[0], errmsg(e, "not found"));
7281         /* NOTREACHED */
7282 }
7283
7284 static void
7285 printentry(struct tblentry *cmdp)
7286 {
7287         int idx;
7288         const char *path;
7289         char *name;
7290
7291         idx = cmdp->param.index;
7292         path = pathval();
7293         do {
7294                 name = path_advance(&path, cmdp->cmdname);
7295                 stunalloc(name);
7296         } while (--idx >= 0);
7297         out1fmt("%s%s\n", name, (cmdp->rehash ? "*" : nullstr));
7298 }
7299
7300 /*
7301  * Clear out command entries.  The argument specifies the first entry in
7302  * PATH which has changed.
7303  */
7304 static void
7305 clearcmdentry(int firstchange)
7306 {
7307         struct tblentry **tblp;
7308         struct tblentry **pp;
7309         struct tblentry *cmdp;
7310
7311         INT_OFF;
7312         for (tblp = cmdtable; tblp < &cmdtable[CMDTABLESIZE]; tblp++) {
7313                 pp = tblp;
7314                 while ((cmdp = *pp) != NULL) {
7315                         if ((cmdp->cmdtype == CMDNORMAL &&
7316                              cmdp->param.index >= firstchange)
7317                          || (cmdp->cmdtype == CMDBUILTIN &&
7318                              builtinloc >= firstchange)
7319                         ) {
7320                                 *pp = cmdp->next;
7321                                 free(cmdp);
7322                         } else {
7323                                 pp = &cmdp->next;
7324                         }
7325                 }
7326         }
7327         INT_ON;
7328 }
7329
7330 /*
7331  * Locate a command in the command hash table.  If "add" is nonzero,
7332  * add the command to the table if it is not already present.  The
7333  * variable "lastcmdentry" is set to point to the address of the link
7334  * pointing to the entry, so that delete_cmd_entry can delete the
7335  * entry.
7336  *
7337  * Interrupts must be off if called with add != 0.
7338  */
7339 static struct tblentry **lastcmdentry;
7340
7341 static struct tblentry *
7342 cmdlookup(const char *name, int add)
7343 {
7344         unsigned int hashval;
7345         const char *p;
7346         struct tblentry *cmdp;
7347         struct tblentry **pp;
7348
7349         p = name;
7350         hashval = (unsigned char)*p << 4;
7351         while (*p)
7352                 hashval += (unsigned char)*p++;
7353         hashval &= 0x7FFF;
7354         pp = &cmdtable[hashval % CMDTABLESIZE];
7355         for (cmdp = *pp; cmdp; cmdp = cmdp->next) {
7356                 if (strcmp(cmdp->cmdname, name) == 0)
7357                         break;
7358                 pp = &cmdp->next;
7359         }
7360         if (add && cmdp == NULL) {
7361                 cmdp = *pp = ckzalloc(sizeof(struct tblentry)
7362                                 + strlen(name)
7363                                 /* + 1 - already done because
7364                                  * tblentry::cmdname is char[1] */);
7365                 /*cmdp->next = NULL; - ckzalloc did it */
7366                 cmdp->cmdtype = CMDUNKNOWN;
7367                 strcpy(cmdp->cmdname, name);
7368         }
7369         lastcmdentry = pp;
7370         return cmdp;
7371 }
7372
7373 /*
7374  * Delete the command entry returned on the last lookup.
7375  */
7376 static void
7377 delete_cmd_entry(void)
7378 {
7379         struct tblentry *cmdp;
7380
7381         INT_OFF;
7382         cmdp = *lastcmdentry;
7383         *lastcmdentry = cmdp->next;
7384         if (cmdp->cmdtype == CMDFUNCTION)
7385                 freefunc(cmdp->param.func);
7386         free(cmdp);
7387         INT_ON;
7388 }
7389
7390 /*
7391  * Add a new command entry, replacing any existing command entry for
7392  * the same name - except special builtins.
7393  */
7394 static void
7395 addcmdentry(char *name, struct cmdentry *entry)
7396 {
7397         struct tblentry *cmdp;
7398
7399         cmdp = cmdlookup(name, 1);
7400         if (cmdp->cmdtype == CMDFUNCTION) {
7401                 freefunc(cmdp->param.func);
7402         }
7403         cmdp->cmdtype = entry->cmdtype;
7404         cmdp->param = entry->u;
7405         cmdp->rehash = 0;
7406 }
7407
7408 static int FAST_FUNC
7409 hashcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
7410 {
7411         struct tblentry **pp;
7412         struct tblentry *cmdp;
7413         int c;
7414         struct cmdentry entry;
7415         char *name;
7416
7417         if (nextopt("r") != '\0') {
7418                 clearcmdentry(0);
7419                 return 0;
7420         }
7421
7422         if (*argptr == NULL) {
7423                 for (pp = cmdtable; pp < &cmdtable[CMDTABLESIZE]; pp++) {
7424                         for (cmdp = *pp; cmdp; cmdp = cmdp->next) {
7425                                 if (cmdp->cmdtype == CMDNORMAL)
7426                                         printentry(cmdp);
7427                         }
7428                 }
7429                 return 0;
7430         }
7431
7432         c = 0;
7433         while ((name = *argptr) != NULL) {
7434                 cmdp = cmdlookup(name, 0);
7435                 if (cmdp != NULL
7436                  && (cmdp->cmdtype == CMDNORMAL
7437                      || (cmdp->cmdtype == CMDBUILTIN && builtinloc >= 0))
7438                 ) {
7439                         delete_cmd_entry();
7440                 }
7441                 find_command(name, &entry, DO_ERR, pathval());
7442                 if (entry.cmdtype == CMDUNKNOWN)
7443                         c = 1;
7444                 argptr++;
7445         }
7446         return c;
7447 }
7448
7449 /*
7450  * Called when a cd is done.  Marks all commands so the next time they
7451  * are executed they will be rehashed.
7452  */
7453 static void
7454 hashcd(void)
7455 {
7456         struct tblentry **pp;
7457         struct tblentry *cmdp;
7458
7459         for (pp = cmdtable; pp < &cmdtable[CMDTABLESIZE]; pp++) {
7460                 for (cmdp = *pp; cmdp; cmdp = cmdp->next) {
7461                         if (cmdp->cmdtype == CMDNORMAL
7462                          || (cmdp->cmdtype == CMDBUILTIN
7463                              && !IS_BUILTIN_REGULAR(cmdp->param.cmd)
7464                              && builtinloc > 0)
7465                         ) {
7466                                 cmdp->rehash = 1;
7467                         }
7468                 }
7469         }
7470 }
7471
7472 /*
7473  * Fix command hash table when PATH changed.
7474  * Called before PATH is changed.  The argument is the new value of PATH;
7475  * pathval() still returns the old value at this point.
7476  * Called with interrupts off.
7477  */
7478 static void FAST_FUNC
7479 changepath(const char *new)
7480 {
7481         const char *old;
7482         int firstchange;
7483         int idx;
7484         int idx_bltin;
7485
7486         old = pathval();
7487         firstchange = 9999;     /* assume no change */
7488         idx = 0;
7489         idx_bltin = -1;
7490         for (;;) {
7491                 if (*old != *new) {
7492                         firstchange = idx;
7493                         if ((*old == '\0' && *new == ':')
7494                          || (*old == ':' && *new == '\0'))
7495                                 firstchange++;
7496                         old = new;      /* ignore subsequent differences */
7497                 }
7498                 if (*new == '\0')
7499                         break;
7500                 if (*new == '%' && idx_bltin < 0 && prefix(new + 1, "builtin"))
7501                         idx_bltin = idx;
7502                 if (*new == ':')
7503                         idx++;
7504                 new++, old++;
7505         }
7506         if (builtinloc < 0 && idx_bltin >= 0)
7507                 builtinloc = idx_bltin;             /* zap builtins */
7508         if (builtinloc >= 0 && idx_bltin < 0)
7509                 firstchange = 0;
7510         clearcmdentry(firstchange);
7511         builtinloc = idx_bltin;
7512 }
7513
7514 #define TEOF 0
7515 #define TNL 1
7516 #define TREDIR 2
7517 #define TWORD 3
7518 #define TSEMI 4
7519 #define TBACKGND 5
7520 #define TAND 6
7521 #define TOR 7
7522 #define TPIPE 8
7523 #define TLP 9
7524 #define TRP 10
7525 #define TENDCASE 11
7526 #define TENDBQUOTE 12
7527 #define TNOT 13
7528 #define TCASE 14
7529 #define TDO 15
7530 #define TDONE 16
7531 #define TELIF 17
7532 #define TELSE 18
7533 #define TESAC 19
7534 #define TFI 20
7535 #define TFOR 21
7536 #define TIF 22
7537 #define TIN 23
7538 #define TTHEN 24
7539 #define TUNTIL 25
7540 #define TWHILE 26
7541 #define TBEGIN 27
7542 #define TEND 28
7543 typedef smallint token_id_t;
7544
7545 /* first char is indicating which tokens mark the end of a list */
7546 static const char *const tokname_array[] = {
7547         "\1end of file",
7548         "\0newline",
7549         "\0redirection",
7550         "\0word",
7551         "\0;",
7552         "\0&",
7553         "\0&&",
7554         "\0||",
7555         "\0|",
7556         "\0(",
7557         "\1)",
7558         "\1;;",
7559         "\1`",
7560 #define KWDOFFSET 13
7561         /* the following are keywords */
7562         "\0!",
7563         "\0case",
7564         "\1do",
7565         "\1done",
7566         "\1elif",
7567         "\1else",
7568         "\1esac",
7569         "\1fi",
7570         "\0for",
7571         "\0if",
7572         "\0in",
7573         "\1then",
7574         "\0until",
7575         "\0while",
7576         "\0{",
7577         "\1}",
7578 };
7579
7580 static const char *
7581 tokname(int tok)
7582 {
7583         static char buf[16];
7584
7585 //try this:
7586 //if (tok < TSEMI) return tokname_array[tok] + 1;
7587 //sprintf(buf, "\"%s\"", tokname_array[tok] + 1);
7588 //return buf;
7589
7590         if (tok >= TSEMI)
7591                 buf[0] = '"';
7592         sprintf(buf + (tok >= TSEMI), "%s%c",
7593                         tokname_array[tok] + 1, (tok >= TSEMI ? '"' : 0));
7594         return buf;
7595 }
7596
7597 /* Wrapper around strcmp for qsort/bsearch/... */
7598 static int
7599 pstrcmp(const void *a, const void *b)
7600 {
7601         return strcmp((char*) a, (*(char**) b) + 1);
7602 }
7603
7604 static const char *const *
7605 findkwd(const char *s)
7606 {
7607         return bsearch(s, tokname_array + KWDOFFSET,
7608                         ARRAY_SIZE(tokname_array) - KWDOFFSET,
7609                         sizeof(tokname_array[0]), pstrcmp);
7610 }
7611
7612 /*
7613  * Locate and print what a word is...
7614  */
7615 static int
7616 describe_command(char *command, int describe_command_verbose)
7617 {
7618         struct cmdentry entry;
7619         struct tblentry *cmdp;
7620 #if ENABLE_ASH_ALIAS
7621         const struct alias *ap;
7622 #endif
7623         const char *path = pathval();
7624
7625         if (describe_command_verbose) {
7626                 out1str(command);
7627         }
7628
7629         /* First look at the keywords */
7630         if (findkwd(command)) {
7631                 out1str(describe_command_verbose ? " is a shell keyword" : command);
7632                 goto out;
7633         }
7634
7635 #if ENABLE_ASH_ALIAS
7636         /* Then look at the aliases */
7637         ap = lookupalias(command, 0);
7638         if (ap != NULL) {
7639                 if (!describe_command_verbose) {
7640                         out1str("alias ");
7641                         printalias(ap);
7642                         return 0;
7643                 }
7644                 out1fmt(" is an alias for %s", ap->val);
7645                 goto out;
7646         }
7647 #endif
7648         /* Then check if it is a tracked alias */
7649         cmdp = cmdlookup(command, 0);
7650         if (cmdp != NULL) {
7651                 entry.cmdtype = cmdp->cmdtype;
7652                 entry.u = cmdp->param;
7653         } else {
7654                 /* Finally use brute force */
7655                 find_command(command, &entry, DO_ABS, path);
7656         }
7657
7658         switch (entry.cmdtype) {
7659         case CMDNORMAL: {
7660                 int j = entry.u.index;
7661                 char *p;
7662                 if (j < 0) {
7663                         p = command;
7664                 } else {
7665                         do {
7666                                 p = path_advance(&path, command);
7667                                 stunalloc(p);
7668                         } while (--j >= 0);
7669                 }
7670                 if (describe_command_verbose) {
7671                         out1fmt(" is%s %s",
7672                                 (cmdp ? " a tracked alias for" : nullstr), p
7673                         );
7674                 } else {
7675                         out1str(p);
7676                 }
7677                 break;
7678         }
7679
7680         case CMDFUNCTION:
7681                 if (describe_command_verbose) {
7682                         out1str(" is a shell function");
7683                 } else {
7684                         out1str(command);
7685                 }
7686                 break;
7687
7688         case CMDBUILTIN:
7689                 if (describe_command_verbose) {
7690                         out1fmt(" is a %sshell builtin",
7691                                 IS_BUILTIN_SPECIAL(entry.u.cmd) ?
7692                                         "special " : nullstr
7693                         );
7694                 } else {
7695                         out1str(command);
7696                 }
7697                 break;
7698
7699         default:
7700                 if (describe_command_verbose) {
7701                         out1str(": not found\n");
7702                 }
7703                 return 127;
7704         }
7705  out:
7706         out1str("\n");
7707         return 0;
7708 }
7709
7710 static int FAST_FUNC
7711 typecmd(int argc UNUSED_PARAM, char **argv)
7712 {
7713         int i = 1;
7714         int err = 0;
7715         int verbose = 1;
7716
7717         /* type -p ... ? (we don't bother checking for 'p') */
7718         if (argv[1] && argv[1][0] == '-') {
7719                 i++;
7720                 verbose = 0;
7721         }
7722         while (argv[i]) {
7723                 err |= describe_command(argv[i++], verbose);
7724         }
7725         return err;
7726 }
7727
7728 #if ENABLE_ASH_CMDCMD
7729 static int FAST_FUNC
7730 commandcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
7731 {
7732         int c;
7733         enum {
7734                 VERIFY_BRIEF = 1,
7735                 VERIFY_VERBOSE = 2,
7736         } verify = 0;
7737
7738         while ((c = nextopt("pvV")) != '\0')
7739                 if (c == 'V')
7740                         verify |= VERIFY_VERBOSE;
7741                 else if (c == 'v')
7742                         verify |= VERIFY_BRIEF;
7743 #if DEBUG
7744                 else if (c != 'p')
7745                         abort();
7746 #endif
7747         /* Mimic bash: just "command -v" doesn't complain, it's a nop */
7748         if (verify && (*argptr != NULL)) {
7749                 return describe_command(*argptr, verify - VERIFY_BRIEF);
7750         }
7751
7752         return 0;
7753 }
7754 #endif
7755
7756
7757 /* ============ eval.c */
7758
7759 static int funcblocksize;       /* size of structures in function */
7760 static int funcstringsize;      /* size of strings in node */
7761 static void *funcblock;         /* block to allocate function from */
7762 static char *funcstring;        /* block to allocate strings from */
7763
7764 /* flags in argument to evaltree */
7765 #define EV_EXIT    01           /* exit after evaluating tree */
7766 #define EV_TESTED  02           /* exit status is checked; ignore -e flag */
7767 #define EV_BACKCMD 04           /* command executing within back quotes */
7768
7769 static const uint8_t nodesize[N_NUMBER] = {
7770         [NCMD     ] = SHELL_ALIGN(sizeof(struct ncmd)),
7771         [NPIPE    ] = SHELL_ALIGN(sizeof(struct npipe)),
7772         [NREDIR   ] = SHELL_ALIGN(sizeof(struct nredir)),
7773         [NBACKGND ] = SHELL_ALIGN(sizeof(struct nredir)),
7774         [NSUBSHELL] = SHELL_ALIGN(sizeof(struct nredir)),
7775         [NAND     ] = SHELL_ALIGN(sizeof(struct nbinary)),
7776         [NOR      ] = SHELL_ALIGN(sizeof(struct nbinary)),
7777         [NSEMI    ] = SHELL_ALIGN(sizeof(struct nbinary)),
7778         [NIF      ] = SHELL_ALIGN(sizeof(struct nif)),
7779         [NWHILE   ] = SHELL_ALIGN(sizeof(struct nbinary)),
7780         [NUNTIL   ] = SHELL_ALIGN(sizeof(struct nbinary)),
7781         [NFOR     ] = SHELL_ALIGN(sizeof(struct nfor)),
7782         [NCASE    ] = SHELL_ALIGN(sizeof(struct ncase)),
7783         [NCLIST   ] = SHELL_ALIGN(sizeof(struct nclist)),
7784         [NDEFUN   ] = SHELL_ALIGN(sizeof(struct narg)),
7785         [NARG     ] = SHELL_ALIGN(sizeof(struct narg)),
7786         [NTO      ] = SHELL_ALIGN(sizeof(struct nfile)),
7787 #if ENABLE_ASH_BASH_COMPAT
7788         [NTO2     ] = SHELL_ALIGN(sizeof(struct nfile)),
7789 #endif
7790         [NCLOBBER ] = SHELL_ALIGN(sizeof(struct nfile)),
7791         [NFROM    ] = SHELL_ALIGN(sizeof(struct nfile)),
7792         [NFROMTO  ] = SHELL_ALIGN(sizeof(struct nfile)),
7793         [NAPPEND  ] = SHELL_ALIGN(sizeof(struct nfile)),
7794         [NTOFD    ] = SHELL_ALIGN(sizeof(struct ndup)),
7795         [NFROMFD  ] = SHELL_ALIGN(sizeof(struct ndup)),
7796         [NHERE    ] = SHELL_ALIGN(sizeof(struct nhere)),
7797         [NXHERE   ] = SHELL_ALIGN(sizeof(struct nhere)),
7798         [NNOT     ] = SHELL_ALIGN(sizeof(struct nnot)),
7799 };
7800
7801 static void calcsize(union node *n);
7802
7803 static void
7804 sizenodelist(struct nodelist *lp)
7805 {
7806         while (lp) {
7807                 funcblocksize += SHELL_ALIGN(sizeof(struct nodelist));
7808                 calcsize(lp->n);
7809                 lp = lp->next;
7810         }
7811 }
7812
7813 static void
7814 calcsize(union node *n)
7815 {
7816         if (n == NULL)
7817                 return;
7818         funcblocksize += nodesize[n->type];
7819         switch (n->type) {
7820         case NCMD:
7821                 calcsize(n->ncmd.redirect);
7822                 calcsize(n->ncmd.args);
7823                 calcsize(n->ncmd.assign);
7824                 break;
7825         case NPIPE:
7826                 sizenodelist(n->npipe.cmdlist);
7827                 break;
7828         case NREDIR:
7829         case NBACKGND:
7830         case NSUBSHELL:
7831                 calcsize(n->nredir.redirect);
7832                 calcsize(n->nredir.n);
7833                 break;
7834         case NAND:
7835         case NOR:
7836         case NSEMI:
7837         case NWHILE:
7838         case NUNTIL:
7839                 calcsize(n->nbinary.ch2);
7840                 calcsize(n->nbinary.ch1);
7841                 break;
7842         case NIF:
7843                 calcsize(n->nif.elsepart);
7844                 calcsize(n->nif.ifpart);
7845                 calcsize(n->nif.test);
7846                 break;
7847         case NFOR:
7848                 funcstringsize += strlen(n->nfor.var) + 1;
7849                 calcsize(n->nfor.body);
7850                 calcsize(n->nfor.args);
7851                 break;
7852         case NCASE:
7853                 calcsize(n->ncase.cases);
7854                 calcsize(n->ncase.expr);
7855                 break;
7856         case NCLIST:
7857                 calcsize(n->nclist.body);
7858                 calcsize(n->nclist.pattern);
7859                 calcsize(n->nclist.next);
7860                 break;
7861         case NDEFUN:
7862         case NARG:
7863                 sizenodelist(n->narg.backquote);
7864                 funcstringsize += strlen(n->narg.text) + 1;
7865                 calcsize(n->narg.next);
7866                 break;
7867         case NTO:
7868 #if ENABLE_ASH_BASH_COMPAT
7869         case NTO2:
7870 #endif
7871         case NCLOBBER:
7872         case NFROM:
7873         case NFROMTO:
7874         case NAPPEND:
7875                 calcsize(n->nfile.fname);
7876                 calcsize(n->nfile.next);
7877                 break;
7878         case NTOFD:
7879         case NFROMFD:
7880                 calcsize(n->ndup.vname);
7881                 calcsize(n->ndup.next);
7882         break;
7883         case NHERE:
7884         case NXHERE:
7885                 calcsize(n->nhere.doc);
7886                 calcsize(n->nhere.next);
7887                 break;
7888         case NNOT:
7889                 calcsize(n->nnot.com);
7890                 break;
7891         };
7892 }
7893
7894 static char *
7895 nodeckstrdup(char *s)
7896 {
7897         char *rtn = funcstring;
7898
7899         strcpy(funcstring, s);
7900         funcstring += strlen(s) + 1;
7901         return rtn;
7902 }
7903
7904 static union node *copynode(union node *);
7905
7906 static struct nodelist *
7907 copynodelist(struct nodelist *lp)
7908 {
7909         struct nodelist *start;
7910         struct nodelist **lpp;
7911
7912         lpp = &start;
7913         while (lp) {
7914                 *lpp = funcblock;
7915                 funcblock = (char *) funcblock + SHELL_ALIGN(sizeof(struct nodelist));
7916                 (*lpp)->n = copynode(lp->n);
7917                 lp = lp->next;
7918                 lpp = &(*lpp)->next;
7919         }
7920         *lpp = NULL;
7921         return start;
7922 }
7923
7924 static union node *
7925 copynode(union node *n)
7926 {
7927         union node *new;
7928
7929         if (n == NULL)
7930                 return NULL;
7931         new = funcblock;
7932         funcblock = (char *) funcblock + nodesize[n->type];
7933
7934         switch (n->type) {
7935         case NCMD:
7936                 new->ncmd.redirect = copynode(n->ncmd.redirect);
7937                 new->ncmd.args = copynode(n->ncmd.args);
7938                 new->ncmd.assign = copynode(n->ncmd.assign);
7939                 break;
7940         case NPIPE:
7941                 new->npipe.cmdlist = copynodelist(n->npipe.cmdlist);
7942                 new->npipe.pipe_backgnd = n->npipe.pipe_backgnd;
7943                 break;
7944         case NREDIR:
7945         case NBACKGND:
7946         case NSUBSHELL:
7947                 new->nredir.redirect = copynode(n->nredir.redirect);
7948                 new->nredir.n = copynode(n->nredir.n);
7949                 break;
7950         case NAND:
7951         case NOR:
7952         case NSEMI:
7953         case NWHILE:
7954         case NUNTIL:
7955                 new->nbinary.ch2 = copynode(n->nbinary.ch2);
7956                 new->nbinary.ch1 = copynode(n->nbinary.ch1);
7957                 break;
7958         case NIF:
7959                 new->nif.elsepart = copynode(n->nif.elsepart);
7960                 new->nif.ifpart = copynode(n->nif.ifpart);
7961                 new->nif.test = copynode(n->nif.test);
7962                 break;
7963         case NFOR:
7964                 new->nfor.var = nodeckstrdup(n->nfor.var);
7965                 new->nfor.body = copynode(n->nfor.body);
7966                 new->nfor.args = copynode(n->nfor.args);
7967                 break;
7968         case NCASE:
7969                 new->ncase.cases = copynode(n->ncase.cases);
7970                 new->ncase.expr = copynode(n->ncase.expr);
7971                 break;
7972         case NCLIST:
7973                 new->nclist.body = copynode(n->nclist.body);
7974                 new->nclist.pattern = copynode(n->nclist.pattern);
7975                 new->nclist.next = copynode(n->nclist.next);
7976                 break;
7977         case NDEFUN:
7978         case NARG:
7979                 new->narg.backquote = copynodelist(n->narg.backquote);
7980                 new->narg.text = nodeckstrdup(n->narg.text);
7981                 new->narg.next = copynode(n->narg.next);
7982                 break;
7983         case NTO:
7984 #if ENABLE_ASH_BASH_COMPAT
7985         case NTO2:
7986 #endif
7987         case NCLOBBER:
7988         case NFROM:
7989         case NFROMTO:
7990         case NAPPEND:
7991                 new->nfile.fname = copynode(n->nfile.fname);
7992                 new->nfile.fd = n->nfile.fd;
7993                 new->nfile.next = copynode(n->nfile.next);
7994                 break;
7995         case NTOFD:
7996         case NFROMFD:
7997                 new->ndup.vname = copynode(n->ndup.vname);
7998                 new->ndup.dupfd = n->ndup.dupfd;
7999                 new->ndup.fd = n->ndup.fd;
8000                 new->ndup.next = copynode(n->ndup.next);
8001                 break;
8002         case NHERE:
8003         case NXHERE:
8004                 new->nhere.doc = copynode(n->nhere.doc);
8005                 new->nhere.fd = n->nhere.fd;
8006                 new->nhere.next = copynode(n->nhere.next);
8007                 break;
8008         case NNOT:
8009                 new->nnot.com = copynode(n->nnot.com);
8010                 break;
8011         };
8012         new->type = n->type;
8013         return new;
8014 }
8015
8016 /*
8017  * Make a copy of a parse tree.
8018  */
8019 static struct funcnode *
8020 copyfunc(union node *n)
8021 {
8022         struct funcnode *f;
8023         size_t blocksize;
8024
8025         funcblocksize = offsetof(struct funcnode, n);
8026         funcstringsize = 0;
8027         calcsize(n);
8028         blocksize = funcblocksize;
8029         f = ckmalloc(blocksize + funcstringsize);
8030         funcblock = (char *) f + offsetof(struct funcnode, n);
8031         funcstring = (char *) f + blocksize;
8032         copynode(n);
8033         f->count = 0;
8034         return f;
8035 }
8036
8037 /*
8038  * Define a shell function.
8039  */
8040 static void
8041 defun(char *name, union node *func)
8042 {
8043         struct cmdentry entry;
8044
8045         INT_OFF;
8046         entry.cmdtype = CMDFUNCTION;
8047         entry.u.func = copyfunc(func);
8048         addcmdentry(name, &entry);
8049         INT_ON;
8050 }
8051
8052 /* Reasons for skipping commands (see comment on breakcmd routine) */
8053 #define SKIPBREAK      (1 << 0)
8054 #define SKIPCONT       (1 << 1)
8055 #define SKIPFUNC       (1 << 2)
8056 #define SKIPFILE       (1 << 3)
8057 #define SKIPEVAL       (1 << 4)
8058 static smallint evalskip;       /* set to SKIPxxx if we are skipping commands */
8059 static int skipcount;           /* number of levels to skip */
8060 static int funcnest;            /* depth of function calls */
8061 static int loopnest;            /* current loop nesting level */
8062
8063 /* Forward decl way out to parsing code - dotrap needs it */
8064 static int evalstring(char *s, int mask);
8065
8066 /* Called to execute a trap.
8067  * Single callsite - at the end of evaltree().
8068  * If we return non-zero, exaltree raises EXEXIT exception.
8069  *
8070  * Perhaps we should avoid entering new trap handlers
8071  * while we are executing a trap handler. [is it a TODO?]
8072  */
8073 static int
8074 dotrap(void)
8075 {
8076         uint8_t *g;
8077         int sig;
8078         uint8_t savestatus;
8079
8080         savestatus = exitstatus;
8081         pending_sig = 0;
8082         xbarrier();
8083
8084         TRACE(("dotrap entered\n"));
8085         for (sig = 1, g = gotsig; sig < NSIG; sig++, g++) {
8086                 int want_exexit;
8087                 char *t;
8088
8089                 if (*g == 0)
8090                         continue;
8091                 t = trap[sig];
8092                 /* non-trapped SIGINT is handled separately by raise_interrupt,
8093                  * don't upset it by resetting gotsig[SIGINT-1] */
8094                 if (sig == SIGINT && !t)
8095                         continue;
8096
8097                 TRACE(("sig %d is active, will run handler '%s'\n", sig, t));
8098                 *g = 0;
8099                 if (!t)
8100                         continue;
8101                 want_exexit = evalstring(t, SKIPEVAL);
8102                 exitstatus = savestatus;
8103                 if (want_exexit) {
8104                         TRACE(("dotrap returns %d\n", want_exexit));
8105                         return want_exexit;
8106                 }
8107         }
8108
8109         TRACE(("dotrap returns 0\n"));
8110         return 0;
8111 }
8112
8113 /* forward declarations - evaluation is fairly recursive business... */
8114 static void evalloop(union node *, int);
8115 static void evalfor(union node *, int);
8116 static void evalcase(union node *, int);
8117 static void evalsubshell(union node *, int);
8118 static void expredir(union node *);
8119 static void evalpipe(union node *, int);
8120 static void evalcommand(union node *, int);
8121 static int evalbltin(const struct builtincmd *, int, char **);
8122 static void prehash(union node *);
8123
8124 /*
8125  * Evaluate a parse tree.  The value is left in the global variable
8126  * exitstatus.
8127  */
8128 static void
8129 evaltree(union node *n, int flags)
8130 {
8131         struct jmploc *volatile savehandler = exception_handler;
8132         struct jmploc jmploc;
8133         int checkexit = 0;
8134         void (*evalfn)(union node *, int);
8135         int status;
8136         int int_level;
8137
8138         SAVE_INT(int_level);
8139
8140         if (n == NULL) {
8141                 TRACE(("evaltree(NULL) called\n"));
8142                 goto out1;
8143         }
8144         TRACE(("evaltree(%p: %d, %d) called\n", n, n->type, flags));
8145
8146         exception_handler = &jmploc;
8147         {
8148                 int err = setjmp(jmploc.loc);
8149                 if (err) {
8150                         /* if it was a signal, check for trap handlers */
8151                         if (exception_type == EXSIG) {
8152                                 TRACE(("exception %d (EXSIG) in evaltree, err=%d\n",
8153                                                 exception_type, err));
8154                                 goto out;
8155                         }
8156                         /* continue on the way out */
8157                         TRACE(("exception %d in evaltree, propagating err=%d\n",
8158                                         exception_type, err));
8159                         exception_handler = savehandler;
8160                         longjmp(exception_handler->loc, err);
8161                 }
8162         }
8163
8164         switch (n->type) {
8165         default:
8166 #if DEBUG
8167                 out1fmt("Node type = %d\n", n->type);
8168                 fflush_all();
8169                 break;
8170 #endif
8171         case NNOT:
8172                 evaltree(n->nnot.com, EV_TESTED);
8173                 status = !exitstatus;
8174                 goto setstatus;
8175         case NREDIR:
8176                 expredir(n->nredir.redirect);
8177                 status = redirectsafe(n->nredir.redirect, REDIR_PUSH);
8178                 if (!status) {
8179                         evaltree(n->nredir.n, flags & EV_TESTED);
8180                         status = exitstatus;
8181                 }
8182                 popredir(/*drop:*/ 0, /*restore:*/ 0 /* not sure */);
8183                 goto setstatus;
8184         case NCMD:
8185                 evalfn = evalcommand;
8186  checkexit:
8187                 if (eflag && !(flags & EV_TESTED))
8188                         checkexit = ~0;
8189                 goto calleval;
8190         case NFOR:
8191                 evalfn = evalfor;
8192                 goto calleval;
8193         case NWHILE:
8194         case NUNTIL:
8195                 evalfn = evalloop;
8196                 goto calleval;
8197         case NSUBSHELL:
8198         case NBACKGND:
8199                 evalfn = evalsubshell;
8200                 goto calleval;
8201         case NPIPE:
8202                 evalfn = evalpipe;
8203                 goto checkexit;
8204         case NCASE:
8205                 evalfn = evalcase;
8206                 goto calleval;
8207         case NAND:
8208         case NOR:
8209         case NSEMI: {
8210
8211 #if NAND + 1 != NOR
8212 #error NAND + 1 != NOR
8213 #endif
8214 #if NOR + 1 != NSEMI
8215 #error NOR + 1 != NSEMI
8216 #endif
8217                 unsigned is_or = n->type - NAND;
8218                 evaltree(
8219                         n->nbinary.ch1,
8220                         (flags | ((is_or >> 1) - 1)) & EV_TESTED
8221                 );
8222                 if (!exitstatus == is_or)
8223                         break;
8224                 if (!evalskip) {
8225                         n = n->nbinary.ch2;
8226  evaln:
8227                         evalfn = evaltree;
8228  calleval:
8229                         evalfn(n, flags);
8230                         break;
8231                 }
8232                 break;
8233         }
8234         case NIF:
8235                 evaltree(n->nif.test, EV_TESTED);
8236                 if (evalskip)
8237                         break;
8238                 if (exitstatus == 0) {
8239                         n = n->nif.ifpart;
8240                         goto evaln;
8241                 }
8242                 if (n->nif.elsepart) {
8243                         n = n->nif.elsepart;
8244                         goto evaln;
8245                 }
8246                 goto success;
8247         case NDEFUN:
8248                 defun(n->narg.text, n->narg.next);
8249  success:
8250                 status = 0;
8251  setstatus:
8252                 exitstatus = status;
8253                 break;
8254         }
8255
8256  out:
8257         exception_handler = savehandler;
8258  out1:
8259         if (checkexit & exitstatus)
8260                 evalskip |= SKIPEVAL;
8261         else if (pending_sig && dotrap())
8262                 goto exexit;
8263
8264         if (flags & EV_EXIT) {
8265  exexit:
8266                 raise_exception(EXEXIT);
8267         }
8268
8269         RESTORE_INT(int_level);
8270         TRACE(("leaving evaltree (no interrupts)\n"));
8271 }
8272
8273 #if !defined(__alpha__) || (defined(__GNUC__) && __GNUC__ >= 3)
8274 static
8275 #endif
8276 void evaltreenr(union node *, int) __attribute__ ((alias("evaltree"),__noreturn__));
8277
8278 static void
8279 evalloop(union node *n, int flags)
8280 {
8281         int status;
8282
8283         loopnest++;
8284         status = 0;
8285         flags &= EV_TESTED;
8286         for (;;) {
8287                 int i;
8288
8289                 evaltree(n->nbinary.ch1, EV_TESTED);
8290                 if (evalskip) {
8291  skipping:
8292                         if (evalskip == SKIPCONT && --skipcount <= 0) {
8293                                 evalskip = 0;
8294                                 continue;
8295                         }
8296                         if (evalskip == SKIPBREAK && --skipcount <= 0)
8297                                 evalskip = 0;
8298                         break;
8299                 }
8300                 i = exitstatus;
8301                 if (n->type != NWHILE)
8302                         i = !i;
8303                 if (i != 0)
8304                         break;
8305                 evaltree(n->nbinary.ch2, flags);
8306                 status = exitstatus;
8307                 if (evalskip)
8308                         goto skipping;
8309         }
8310         loopnest--;
8311         exitstatus = status;
8312 }
8313
8314 static void
8315 evalfor(union node *n, int flags)
8316 {
8317         struct arglist arglist;
8318         union node *argp;
8319         struct strlist *sp;
8320         struct stackmark smark;
8321
8322         setstackmark(&smark);
8323         arglist.list = NULL;
8324         arglist.lastp = &arglist.list;
8325         for (argp = n->nfor.args; argp; argp = argp->narg.next) {
8326                 expandarg(argp, &arglist, EXP_FULL | EXP_TILDE | EXP_RECORD);
8327                 /* XXX */
8328                 if (evalskip)
8329                         goto out;
8330         }
8331         *arglist.lastp = NULL;
8332
8333         exitstatus = 0;
8334         loopnest++;
8335         flags &= EV_TESTED;
8336         for (sp = arglist.list; sp; sp = sp->next) {
8337                 setvar(n->nfor.var, sp->text, 0);
8338                 evaltree(n->nfor.body, flags);
8339                 if (evalskip) {
8340                         if (evalskip == SKIPCONT && --skipcount <= 0) {
8341                                 evalskip = 0;
8342                                 continue;
8343                         }
8344                         if (evalskip == SKIPBREAK && --skipcount <= 0)
8345                                 evalskip = 0;
8346                         break;
8347                 }
8348         }
8349         loopnest--;
8350  out:
8351         popstackmark(&smark);
8352 }
8353
8354 static void
8355 evalcase(union node *n, int flags)
8356 {
8357         union node *cp;
8358         union node *patp;
8359         struct arglist arglist;
8360         struct stackmark smark;
8361
8362         setstackmark(&smark);
8363         arglist.list = NULL;
8364         arglist.lastp = &arglist.list;
8365         expandarg(n->ncase.expr, &arglist, EXP_TILDE);
8366         exitstatus = 0;
8367         for (cp = n->ncase.cases; cp && evalskip == 0; cp = cp->nclist.next) {
8368                 for (patp = cp->nclist.pattern; patp; patp = patp->narg.next) {
8369                         if (casematch(patp, arglist.list->text)) {
8370                                 if (evalskip == 0) {
8371                                         evaltree(cp->nclist.body, flags);
8372                                 }
8373                                 goto out;
8374                         }
8375                 }
8376         }
8377  out:
8378         popstackmark(&smark);
8379 }
8380
8381 /*
8382  * Kick off a subshell to evaluate a tree.
8383  */
8384 static void
8385 evalsubshell(union node *n, int flags)
8386 {
8387         struct job *jp;
8388         int backgnd = (n->type == NBACKGND);
8389         int status;
8390
8391         expredir(n->nredir.redirect);
8392         if (!backgnd && flags & EV_EXIT && !trap[0])
8393                 goto nofork;
8394         INT_OFF;
8395         jp = makejob(/*n,*/ 1);
8396         if (forkshell(jp, n, backgnd) == 0) {
8397                 INT_ON;
8398                 flags |= EV_EXIT;
8399                 if (backgnd)
8400                         flags &=~ EV_TESTED;
8401  nofork:
8402                 redirect(n->nredir.redirect, 0);
8403                 evaltreenr(n->nredir.n, flags);
8404                 /* never returns */
8405         }
8406         status = 0;
8407         if (!backgnd)
8408                 status = waitforjob(jp);
8409         exitstatus = status;
8410         INT_ON;
8411 }
8412
8413 /*
8414  * Compute the names of the files in a redirection list.
8415  */
8416 static void fixredir(union node *, const char *, int);
8417 static void
8418 expredir(union node *n)
8419 {
8420         union node *redir;
8421
8422         for (redir = n; redir; redir = redir->nfile.next) {
8423                 struct arglist fn;
8424
8425                 fn.list = NULL;
8426                 fn.lastp = &fn.list;
8427                 switch (redir->type) {
8428                 case NFROMTO:
8429                 case NFROM:
8430                 case NTO:
8431 #if ENABLE_ASH_BASH_COMPAT
8432                 case NTO2:
8433 #endif
8434                 case NCLOBBER:
8435                 case NAPPEND:
8436                         expandarg(redir->nfile.fname, &fn, EXP_TILDE | EXP_REDIR);
8437 #if ENABLE_ASH_BASH_COMPAT
8438  store_expfname:
8439 #endif
8440                         redir->nfile.expfname = fn.list->text;
8441                         break;
8442                 case NFROMFD:
8443                 case NTOFD: /* >& */
8444                         if (redir->ndup.vname) {
8445                                 expandarg(redir->ndup.vname, &fn, EXP_FULL | EXP_TILDE);
8446                                 if (fn.list == NULL)
8447                                         ash_msg_and_raise_error("redir error");
8448 #if ENABLE_ASH_BASH_COMPAT
8449 //FIXME: we used expandarg with different args!
8450                                 if (!isdigit_str9(fn.list->text)) {
8451                                         /* >&file, not >&fd */
8452                                         if (redir->nfile.fd != 1) /* 123>&file - BAD */
8453                                                 ash_msg_and_raise_error("redir error");
8454                                         redir->type = NTO2;
8455                                         goto store_expfname;
8456                                 }
8457 #endif
8458                                 fixredir(redir, fn.list->text, 1);
8459                         }
8460                         break;
8461                 }
8462         }
8463 }
8464
8465 /*
8466  * Evaluate a pipeline.  All the processes in the pipeline are children
8467  * of the process creating the pipeline.  (This differs from some versions
8468  * of the shell, which make the last process in a pipeline the parent
8469  * of all the rest.)
8470  */
8471 static void
8472 evalpipe(union node *n, int flags)
8473 {
8474         struct job *jp;
8475         struct nodelist *lp;
8476         int pipelen;
8477         int prevfd;
8478         int pip[2];
8479
8480         TRACE(("evalpipe(0x%lx) called\n", (long)n));
8481         pipelen = 0;
8482         for (lp = n->npipe.cmdlist; lp; lp = lp->next)
8483                 pipelen++;
8484         flags |= EV_EXIT;
8485         INT_OFF;
8486         jp = makejob(/*n,*/ pipelen);
8487         prevfd = -1;
8488         for (lp = n->npipe.cmdlist; lp; lp = lp->next) {
8489                 prehash(lp->n);
8490                 pip[1] = -1;
8491                 if (lp->next) {
8492                         if (pipe(pip) < 0) {
8493                                 close(prevfd);
8494                                 ash_msg_and_raise_error("pipe call failed");
8495                         }
8496                 }
8497                 if (forkshell(jp, lp->n, n->npipe.pipe_backgnd) == 0) {
8498                         INT_ON;
8499                         if (pip[1] >= 0) {
8500                                 close(pip[0]);
8501                         }
8502                         if (prevfd > 0) {
8503                                 dup2(prevfd, 0);
8504                                 close(prevfd);
8505                         }
8506                         if (pip[1] > 1) {
8507                                 dup2(pip[1], 1);
8508                                 close(pip[1]);
8509                         }
8510                         evaltreenr(lp->n, flags);
8511                         /* never returns */
8512                 }
8513                 if (prevfd >= 0)
8514                         close(prevfd);
8515                 prevfd = pip[0];
8516                 /* Don't want to trigger debugging */
8517                 if (pip[1] != -1)
8518                         close(pip[1]);
8519         }
8520         if (n->npipe.pipe_backgnd == 0) {
8521                 exitstatus = waitforjob(jp);
8522                 TRACE(("evalpipe:  job done exit status %d\n", exitstatus));
8523         }
8524         INT_ON;
8525 }
8526
8527 /*
8528  * Controls whether the shell is interactive or not.
8529  */
8530 static void
8531 setinteractive(int on)
8532 {
8533         static smallint is_interactive;
8534
8535         if (++on == is_interactive)
8536                 return;
8537         is_interactive = on;
8538         setsignal(SIGINT);
8539         setsignal(SIGQUIT);
8540         setsignal(SIGTERM);
8541 #if !ENABLE_FEATURE_SH_EXTRA_QUIET
8542         if (is_interactive > 1) {
8543                 /* Looks like they want an interactive shell */
8544                 static smallint did_banner;
8545
8546                 if (!did_banner) {
8547                         /* note: ash and hush share this string */
8548                         out1fmt("\n\n%s %s\n"
8549                                 "Enter 'help' for a list of built-in commands."
8550                                 "\n\n",
8551                                 bb_banner,
8552                                 "built-in shell (ash)"
8553                         );
8554                         did_banner = 1;
8555                 }
8556         }
8557 #endif
8558 }
8559
8560 static void
8561 optschanged(void)
8562 {
8563 #if DEBUG
8564         opentrace();
8565 #endif
8566         setinteractive(iflag);
8567         setjobctl(mflag);
8568 #if ENABLE_FEATURE_EDITING_VI
8569         if (viflag)
8570                 line_input_state->flags |= VI_MODE;
8571         else
8572                 line_input_state->flags &= ~VI_MODE;
8573 #else
8574         viflag = 0; /* forcibly keep the option off */
8575 #endif
8576 }
8577
8578 static struct localvar *localvars;
8579
8580 /*
8581  * Called after a function returns.
8582  * Interrupts must be off.
8583  */
8584 static void
8585 poplocalvars(void)
8586 {
8587         struct localvar *lvp;
8588         struct var *vp;
8589
8590         while ((lvp = localvars) != NULL) {
8591                 localvars = lvp->next;
8592                 vp = lvp->vp;
8593                 TRACE(("poplocalvar %s\n", vp ? vp->text : "-"));
8594                 if (vp == NULL) {       /* $- saved */
8595                         memcpy(optlist, lvp->text, sizeof(optlist));
8596                         free((char*)lvp->text);
8597                         optschanged();
8598                 } else if ((lvp->flags & (VUNSET|VSTRFIXED)) == VUNSET) {
8599                         unsetvar(vp->text);
8600                 } else {
8601                         if (vp->func)
8602                                 (*vp->func)(strchrnul(lvp->text, '=') + 1);
8603                         if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0)
8604                                 free((char*)vp->text);
8605                         vp->flags = lvp->flags;
8606                         vp->text = lvp->text;
8607                 }
8608                 free(lvp);
8609         }
8610 }
8611
8612 static int
8613 evalfun(struct funcnode *func, int argc, char **argv, int flags)
8614 {
8615         volatile struct shparam saveparam;
8616         struct localvar *volatile savelocalvars;
8617         struct jmploc *volatile savehandler;
8618         struct jmploc jmploc;
8619         int e;
8620
8621         saveparam = shellparam;
8622         savelocalvars = localvars;
8623         e = setjmp(jmploc.loc);
8624         if (e) {
8625                 goto funcdone;
8626         }
8627         INT_OFF;
8628         savehandler = exception_handler;
8629         exception_handler = &jmploc;
8630         localvars = NULL;
8631         shellparam.malloced = 0;
8632         func->count++;
8633         funcnest++;
8634         INT_ON;
8635         shellparam.nparam = argc - 1;
8636         shellparam.p = argv + 1;
8637 #if ENABLE_ASH_GETOPTS
8638         shellparam.optind = 1;
8639         shellparam.optoff = -1;
8640 #endif
8641         evaltree(&func->n, flags & EV_TESTED);
8642  funcdone:
8643         INT_OFF;
8644         funcnest--;
8645         freefunc(func);
8646         poplocalvars();
8647         localvars = savelocalvars;
8648         freeparam(&shellparam);
8649         shellparam = saveparam;
8650         exception_handler = savehandler;
8651         INT_ON;
8652         evalskip &= ~SKIPFUNC;
8653         return e;
8654 }
8655
8656 #if ENABLE_ASH_CMDCMD
8657 static char **
8658 parse_command_args(char **argv, const char **path)
8659 {
8660         char *cp, c;
8661
8662         for (;;) {
8663                 cp = *++argv;
8664                 if (!cp)
8665                         return 0;
8666                 if (*cp++ != '-')
8667                         break;
8668                 c = *cp++;
8669                 if (!c)
8670                         break;
8671                 if (c == '-' && !*cp) {
8672                         argv++;
8673                         break;
8674                 }
8675                 do {
8676                         switch (c) {
8677                         case 'p':
8678                                 *path = bb_default_path;
8679                                 break;
8680                         default:
8681                                 /* run 'typecmd' for other options */
8682                                 return 0;
8683                         }
8684                         c = *cp++;
8685                 } while (c);
8686         }
8687         return argv;
8688 }
8689 #endif
8690
8691 /*
8692  * Make a variable a local variable.  When a variable is made local, it's
8693  * value and flags are saved in a localvar structure.  The saved values
8694  * will be restored when the shell function returns.  We handle the name
8695  * "-" as a special case.
8696  */
8697 static void
8698 mklocal(char *name)
8699 {
8700         struct localvar *lvp;
8701         struct var **vpp;
8702         struct var *vp;
8703
8704         INT_OFF;
8705         lvp = ckzalloc(sizeof(struct localvar));
8706         if (LONE_DASH(name)) {
8707                 char *p;
8708                 p = ckmalloc(sizeof(optlist));
8709                 lvp->text = memcpy(p, optlist, sizeof(optlist));
8710                 vp = NULL;
8711         } else {
8712                 char *eq;
8713
8714                 vpp = hashvar(name);
8715                 vp = *findvar(vpp, name);
8716                 eq = strchr(name, '=');
8717                 if (vp == NULL) {
8718                         if (eq)
8719                                 setvareq(name, VSTRFIXED);
8720                         else
8721                                 setvar(name, NULL, VSTRFIXED);
8722                         vp = *vpp;      /* the new variable */
8723                         lvp->flags = VUNSET;
8724                 } else {
8725                         lvp->text = vp->text;
8726                         lvp->flags = vp->flags;
8727                         vp->flags |= VSTRFIXED|VTEXTFIXED;
8728                         if (eq)
8729                                 setvareq(name, 0);
8730                 }
8731         }
8732         lvp->vp = vp;
8733         lvp->next = localvars;
8734         localvars = lvp;
8735         INT_ON;
8736 }
8737
8738 /*
8739  * The "local" command.
8740  */
8741 static int FAST_FUNC
8742 localcmd(int argc UNUSED_PARAM, char **argv)
8743 {
8744         char *name;
8745
8746         argv = argptr;
8747         while ((name = *argv++) != NULL) {
8748                 mklocal(name);
8749         }
8750         return 0;
8751 }
8752
8753 static int FAST_FUNC
8754 falsecmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
8755 {
8756         return 1;
8757 }
8758
8759 static int FAST_FUNC
8760 truecmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
8761 {
8762         return 0;
8763 }
8764
8765 static int FAST_FUNC
8766 execcmd(int argc UNUSED_PARAM, char **argv)
8767 {
8768         if (argv[1]) {
8769                 iflag = 0;              /* exit on error */
8770                 mflag = 0;
8771                 optschanged();
8772                 shellexec(argv + 1, pathval(), 0);
8773         }
8774         return 0;
8775 }
8776
8777 /*
8778  * The return command.
8779  */
8780 static int FAST_FUNC
8781 returncmd(int argc UNUSED_PARAM, char **argv)
8782 {
8783         /*
8784          * If called outside a function, do what ksh does;
8785          * skip the rest of the file.
8786          */
8787         evalskip = funcnest ? SKIPFUNC : SKIPFILE;
8788         return argv[1] ? number(argv[1]) : exitstatus;
8789 }
8790
8791 /* Forward declarations for builtintab[] */
8792 static int breakcmd(int, char **) FAST_FUNC;
8793 static int dotcmd(int, char **) FAST_FUNC;
8794 static int evalcmd(int, char **) FAST_FUNC;
8795 static int exitcmd(int, char **) FAST_FUNC;
8796 static int exportcmd(int, char **) FAST_FUNC;
8797 #if ENABLE_ASH_GETOPTS
8798 static int getoptscmd(int, char **) FAST_FUNC;
8799 #endif
8800 #if !ENABLE_FEATURE_SH_EXTRA_QUIET
8801 static int helpcmd(int, char **) FAST_FUNC;
8802 #endif
8803 #if ENABLE_SH_MATH_SUPPORT
8804 static int letcmd(int, char **) FAST_FUNC;
8805 #endif
8806 static int readcmd(int, char **) FAST_FUNC;
8807 static int setcmd(int, char **) FAST_FUNC;
8808 static int shiftcmd(int, char **) FAST_FUNC;
8809 static int timescmd(int, char **) FAST_FUNC;
8810 static int trapcmd(int, char **) FAST_FUNC;
8811 static int umaskcmd(int, char **) FAST_FUNC;
8812 static int unsetcmd(int, char **) FAST_FUNC;
8813 static int ulimitcmd(int, char **) FAST_FUNC;
8814
8815 #define BUILTIN_NOSPEC          "0"
8816 #define BUILTIN_SPECIAL         "1"
8817 #define BUILTIN_REGULAR         "2"
8818 #define BUILTIN_SPEC_REG        "3"
8819 #define BUILTIN_ASSIGN          "4"
8820 #define BUILTIN_SPEC_ASSG       "5"
8821 #define BUILTIN_REG_ASSG        "6"
8822 #define BUILTIN_SPEC_REG_ASSG   "7"
8823
8824 /* Stubs for calling non-FAST_FUNC's */
8825 #if ENABLE_ASH_BUILTIN_ECHO
8826 static int FAST_FUNC echocmd(int argc, char **argv)   { return echo_main(argc, argv); }
8827 #endif
8828 #if ENABLE_ASH_BUILTIN_PRINTF
8829 static int FAST_FUNC printfcmd(int argc, char **argv) { return printf_main(argc, argv); }
8830 #endif
8831 #if ENABLE_ASH_BUILTIN_TEST
8832 static int FAST_FUNC testcmd(int argc, char **argv)   { return test_main(argc, argv); }
8833 #endif
8834
8835 /* Keep these in proper order since it is searched via bsearch() */
8836 static const struct builtincmd builtintab[] = {
8837         { BUILTIN_SPEC_REG      ".", dotcmd },
8838         { BUILTIN_SPEC_REG      ":", truecmd },
8839 #if ENABLE_ASH_BUILTIN_TEST
8840         { BUILTIN_REGULAR       "[", testcmd },
8841 #if ENABLE_ASH_BASH_COMPAT
8842         { BUILTIN_REGULAR       "[[", testcmd },
8843 #endif
8844 #endif
8845 #if ENABLE_ASH_ALIAS
8846         { BUILTIN_REG_ASSG      "alias", aliascmd },
8847 #endif
8848 #if JOBS
8849         { BUILTIN_REGULAR       "bg", fg_bgcmd },
8850 #endif
8851         { BUILTIN_SPEC_REG      "break", breakcmd },
8852         { BUILTIN_REGULAR       "cd", cdcmd },
8853         { BUILTIN_NOSPEC        "chdir", cdcmd },
8854 #if ENABLE_ASH_CMDCMD
8855         { BUILTIN_REGULAR       "command", commandcmd },
8856 #endif
8857         { BUILTIN_SPEC_REG      "continue", breakcmd },
8858 #if ENABLE_ASH_BUILTIN_ECHO
8859         { BUILTIN_REGULAR       "echo", echocmd },
8860 #endif
8861         { BUILTIN_SPEC_REG      "eval", evalcmd },
8862         { BUILTIN_SPEC_REG      "exec", execcmd },
8863         { BUILTIN_SPEC_REG      "exit", exitcmd },
8864         { BUILTIN_SPEC_REG_ASSG "export", exportcmd },
8865         { BUILTIN_REGULAR       "false", falsecmd },
8866 #if JOBS
8867         { BUILTIN_REGULAR       "fg", fg_bgcmd },
8868 #endif
8869 #if ENABLE_ASH_GETOPTS
8870         { BUILTIN_REGULAR       "getopts", getoptscmd },
8871 #endif
8872         { BUILTIN_NOSPEC        "hash", hashcmd },
8873 #if !ENABLE_FEATURE_SH_EXTRA_QUIET
8874         { BUILTIN_NOSPEC        "help", helpcmd },
8875 #endif
8876 #if JOBS
8877         { BUILTIN_REGULAR       "jobs", jobscmd },
8878         { BUILTIN_REGULAR       "kill", killcmd },
8879 #endif
8880 #if ENABLE_SH_MATH_SUPPORT
8881         { BUILTIN_NOSPEC        "let", letcmd },
8882 #endif
8883         { BUILTIN_ASSIGN        "local", localcmd },
8884 #if ENABLE_ASH_BUILTIN_PRINTF
8885         { BUILTIN_REGULAR       "printf", printfcmd },
8886 #endif
8887         { BUILTIN_NOSPEC        "pwd", pwdcmd },
8888         { BUILTIN_REGULAR       "read", readcmd },
8889         { BUILTIN_SPEC_REG_ASSG "readonly", exportcmd },
8890         { BUILTIN_SPEC_REG      "return", returncmd },
8891         { BUILTIN_SPEC_REG      "set", setcmd },
8892         { BUILTIN_SPEC_REG      "shift", shiftcmd },
8893         { BUILTIN_SPEC_REG      "source", dotcmd },
8894 #if ENABLE_ASH_BUILTIN_TEST
8895         { BUILTIN_REGULAR       "test", testcmd },
8896 #endif
8897         { BUILTIN_SPEC_REG      "times", timescmd },
8898         { BUILTIN_SPEC_REG      "trap", trapcmd },
8899         { BUILTIN_REGULAR       "true", truecmd },
8900         { BUILTIN_NOSPEC        "type", typecmd },
8901         { BUILTIN_NOSPEC        "ulimit", ulimitcmd },
8902         { BUILTIN_REGULAR       "umask", umaskcmd },
8903 #if ENABLE_ASH_ALIAS
8904         { BUILTIN_REGULAR       "unalias", unaliascmd },
8905 #endif
8906         { BUILTIN_SPEC_REG      "unset", unsetcmd },
8907         { BUILTIN_REGULAR       "wait", waitcmd },
8908 };
8909
8910 /* Should match the above table! */
8911 #define COMMANDCMD (builtintab + \
8912         2 + \
8913         1 * ENABLE_ASH_BUILTIN_TEST + \
8914         1 * ENABLE_ASH_BUILTIN_TEST * ENABLE_ASH_BASH_COMPAT + \
8915         1 * ENABLE_ASH_ALIAS + \
8916         1 * ENABLE_ASH_JOB_CONTROL + \
8917         3)
8918 #define EXECCMD (builtintab + \
8919         2 + \
8920         1 * ENABLE_ASH_BUILTIN_TEST + \
8921         1 * ENABLE_ASH_BUILTIN_TEST * ENABLE_ASH_BASH_COMPAT + \
8922         1 * ENABLE_ASH_ALIAS + \
8923         1 * ENABLE_ASH_JOB_CONTROL + \
8924         3 + \
8925         1 * ENABLE_ASH_CMDCMD + \
8926         1 + \
8927         ENABLE_ASH_BUILTIN_ECHO + \
8928         1)
8929
8930 /*
8931  * Search the table of builtin commands.
8932  */
8933 static struct builtincmd *
8934 find_builtin(const char *name)
8935 {
8936         struct builtincmd *bp;
8937
8938         bp = bsearch(
8939                 name, builtintab, ARRAY_SIZE(builtintab), sizeof(builtintab[0]),
8940                 pstrcmp
8941         );
8942         return bp;
8943 }
8944
8945 /*
8946  * Execute a simple command.
8947  */
8948 static int
8949 isassignment(const char *p)
8950 {
8951         const char *q = endofname(p);
8952         if (p == q)
8953                 return 0;
8954         return *q == '=';
8955 }
8956 static int FAST_FUNC
8957 bltincmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
8958 {
8959         /* Preserve exitstatus of a previous possible redirection
8960          * as POSIX mandates */
8961         return back_exitstatus;
8962 }
8963 static void
8964 evalcommand(union node *cmd, int flags)
8965 {
8966         static const struct builtincmd null_bltin = {
8967                 "\0\0", bltincmd /* why three NULs? */
8968         };
8969         struct stackmark smark;
8970         union node *argp;
8971         struct arglist arglist;
8972         struct arglist varlist;
8973         char **argv;
8974         int argc;
8975         const struct strlist *sp;
8976         struct cmdentry cmdentry;
8977         struct job *jp;
8978         char *lastarg;
8979         const char *path;
8980         int spclbltin;
8981         int status;
8982         char **nargv;
8983         struct builtincmd *bcmd;
8984         smallint cmd_is_exec;
8985         smallint pseudovarflag = 0;
8986
8987         /* First expand the arguments. */
8988         TRACE(("evalcommand(0x%lx, %d) called\n", (long)cmd, flags));
8989         setstackmark(&smark);
8990         back_exitstatus = 0;
8991
8992         cmdentry.cmdtype = CMDBUILTIN;
8993         cmdentry.u.cmd = &null_bltin;
8994         varlist.lastp = &varlist.list;
8995         *varlist.lastp = NULL;
8996         arglist.lastp = &arglist.list;
8997         *arglist.lastp = NULL;
8998
8999         argc = 0;
9000         if (cmd->ncmd.args) {
9001                 bcmd = find_builtin(cmd->ncmd.args->narg.text);
9002                 pseudovarflag = bcmd && IS_BUILTIN_ASSIGN(bcmd);
9003         }
9004
9005         for (argp = cmd->ncmd.args; argp; argp = argp->narg.next) {
9006                 struct strlist **spp;
9007
9008                 spp = arglist.lastp;
9009                 if (pseudovarflag && isassignment(argp->narg.text))
9010                         expandarg(argp, &arglist, EXP_VARTILDE);
9011                 else
9012                         expandarg(argp, &arglist, EXP_FULL | EXP_TILDE);
9013
9014                 for (sp = *spp; sp; sp = sp->next)
9015                         argc++;
9016         }
9017
9018         argv = nargv = stalloc(sizeof(char *) * (argc + 1));
9019         for (sp = arglist.list; sp; sp = sp->next) {
9020                 TRACE(("evalcommand arg: %s\n", sp->text));
9021                 *nargv++ = sp->text;
9022         }
9023         *nargv = NULL;
9024
9025         lastarg = NULL;
9026         if (iflag && funcnest == 0 && argc > 0)
9027                 lastarg = nargv[-1];
9028
9029         preverrout_fd = 2;
9030         expredir(cmd->ncmd.redirect);
9031         status = redirectsafe(cmd->ncmd.redirect, REDIR_PUSH | REDIR_SAVEFD2);
9032
9033         path = vpath.text;
9034         for (argp = cmd->ncmd.assign; argp; argp = argp->narg.next) {
9035                 struct strlist **spp;
9036                 char *p;
9037
9038                 spp = varlist.lastp;
9039                 expandarg(argp, &varlist, EXP_VARTILDE);
9040
9041                 /*
9042                  * Modify the command lookup path, if a PATH= assignment
9043                  * is present
9044                  */
9045                 p = (*spp)->text;
9046                 if (varequal(p, path))
9047                         path = p;
9048         }
9049
9050         /* Print the command if xflag is set. */
9051         if (xflag) {
9052                 int n;
9053                 const char *p = " %s";
9054
9055                 p++;
9056                 fdprintf(preverrout_fd, p, expandstr(ps4val()));
9057
9058                 sp = varlist.list;
9059                 for (n = 0; n < 2; n++) {
9060                         while (sp) {
9061                                 fdprintf(preverrout_fd, p, sp->text);
9062                                 sp = sp->next;
9063                                 if (*p == '%') {
9064                                         p--;
9065                                 }
9066                         }
9067                         sp = arglist.list;
9068                 }
9069                 safe_write(preverrout_fd, "\n", 1);
9070         }
9071
9072         cmd_is_exec = 0;
9073         spclbltin = -1;
9074
9075         /* Now locate the command. */
9076         if (argc) {
9077                 const char *oldpath;
9078                 int cmd_flag = DO_ERR;
9079
9080                 path += 5;
9081                 oldpath = path;
9082                 for (;;) {
9083                         find_command(argv[0], &cmdentry, cmd_flag, path);
9084                         if (cmdentry.cmdtype == CMDUNKNOWN) {
9085                                 flush_stdout_stderr();
9086                                 status = 127;
9087                                 goto bail;
9088                         }
9089
9090                         /* implement bltin and command here */
9091                         if (cmdentry.cmdtype != CMDBUILTIN)
9092                                 break;
9093                         if (spclbltin < 0)
9094                                 spclbltin = IS_BUILTIN_SPECIAL(cmdentry.u.cmd);
9095                         if (cmdentry.u.cmd == EXECCMD)
9096                                 cmd_is_exec = 1;
9097 #if ENABLE_ASH_CMDCMD
9098                         if (cmdentry.u.cmd == COMMANDCMD) {
9099                                 path = oldpath;
9100                                 nargv = parse_command_args(argv, &path);
9101                                 if (!nargv)
9102                                         break;
9103                                 argc -= nargv - argv;
9104                                 argv = nargv;
9105                                 cmd_flag |= DO_NOFUNC;
9106                         } else
9107 #endif
9108                                 break;
9109                 }
9110         }
9111
9112         if (status) {
9113                 /* We have a redirection error. */
9114                 if (spclbltin > 0)
9115                         raise_exception(EXERROR);
9116  bail:
9117                 exitstatus = status;
9118                 goto out;
9119         }
9120
9121         /* Execute the command. */
9122         switch (cmdentry.cmdtype) {
9123         default:
9124
9125 #if ENABLE_FEATURE_SH_NOFORK
9126 /* Hmmm... shouldn't it happen somewhere in forkshell() instead?
9127  * Why "fork off a child process if necessary" doesn't apply to NOFORK? */
9128         {
9129                 /* find_command() encodes applet_no as (-2 - applet_no) */
9130                 int applet_no = (- cmdentry.u.index - 2);
9131                 if (applet_no >= 0 && APPLET_IS_NOFORK(applet_no)) {
9132                         listsetvar(varlist.list, VEXPORT|VSTACK);
9133                         /* run <applet>_main() */
9134                         exitstatus = run_nofork_applet(applet_no, argv);
9135                         break;
9136                 }
9137         }
9138 #endif
9139                 /* Fork off a child process if necessary. */
9140                 if (!(flags & EV_EXIT) || trap[0]) {
9141                         INT_OFF;
9142                         jp = makejob(/*cmd,*/ 1);
9143                         if (forkshell(jp, cmd, FORK_FG) != 0) {
9144                                 exitstatus = waitforjob(jp);
9145                                 INT_ON;
9146                                 TRACE(("forked child exited with %d\n", exitstatus));
9147                                 break;
9148                         }
9149                         FORCE_INT_ON;
9150                 }
9151                 listsetvar(varlist.list, VEXPORT|VSTACK);
9152                 shellexec(argv, path, cmdentry.u.index);
9153                 /* NOTREACHED */
9154
9155         case CMDBUILTIN:
9156                 cmdenviron = varlist.list;
9157                 if (cmdenviron) {
9158                         struct strlist *list = cmdenviron;
9159                         int i = VNOSET;
9160                         if (spclbltin > 0 || argc == 0) {
9161                                 i = 0;
9162                                 if (cmd_is_exec && argc > 1)
9163                                         i = VEXPORT;
9164                         }
9165                         listsetvar(list, i);
9166                 }
9167                 /* Tight loop with builtins only:
9168                  * "while kill -0 $child; do true; done"
9169                  * will never exit even if $child died, unless we do this
9170                  * to reap the zombie and make kill detect that it's gone: */
9171                 dowait(DOWAIT_NONBLOCK, NULL);
9172
9173                 if (evalbltin(cmdentry.u.cmd, argc, argv)) {
9174                         int exit_status;
9175                         int i = exception_type;
9176                         if (i == EXEXIT)
9177                                 goto raise;
9178                         exit_status = 2;
9179                         if (i == EXINT)
9180                                 exit_status = 128 + SIGINT;
9181                         if (i == EXSIG)
9182                                 exit_status = 128 + pending_sig;
9183                         exitstatus = exit_status;
9184                         if (i == EXINT || spclbltin > 0) {
9185  raise:
9186                                 longjmp(exception_handler->loc, 1);
9187                         }
9188                         FORCE_INT_ON;
9189                 }
9190                 break;
9191
9192         case CMDFUNCTION:
9193                 listsetvar(varlist.list, 0);
9194                 /* See above for the rationale */
9195                 dowait(DOWAIT_NONBLOCK, NULL);
9196                 if (evalfun(cmdentry.u.func, argc, argv, flags))
9197                         goto raise;
9198                 break;
9199         }
9200
9201  out:
9202         popredir(/*drop:*/ cmd_is_exec, /*restore:*/ cmd_is_exec);
9203         if (lastarg) {
9204                 /* dsl: I think this is intended to be used to support
9205                  * '_' in 'vi' command mode during line editing...
9206                  * However I implemented that within libedit itself.
9207                  */
9208                 setvar("_", lastarg, 0);
9209         }
9210         popstackmark(&smark);
9211 }
9212
9213 static int
9214 evalbltin(const struct builtincmd *cmd, int argc, char **argv)
9215 {
9216         char *volatile savecmdname;
9217         struct jmploc *volatile savehandler;
9218         struct jmploc jmploc;
9219         int i;
9220
9221         savecmdname = commandname;
9222         i = setjmp(jmploc.loc);
9223         if (i)
9224                 goto cmddone;
9225         savehandler = exception_handler;
9226         exception_handler = &jmploc;
9227         commandname = argv[0];
9228         argptr = argv + 1;
9229         optptr = NULL;                  /* initialize nextopt */
9230         exitstatus = (*cmd->builtin)(argc, argv);
9231         flush_stdout_stderr();
9232  cmddone:
9233         exitstatus |= ferror(stdout);
9234         clearerr(stdout);
9235         commandname = savecmdname;
9236         exception_handler = savehandler;
9237
9238         return i;
9239 }
9240
9241 static int
9242 goodname(const char *p)
9243 {
9244         return !*endofname(p);
9245 }
9246
9247
9248 /*
9249  * Search for a command.  This is called before we fork so that the
9250  * location of the command will be available in the parent as well as
9251  * the child.  The check for "goodname" is an overly conservative
9252  * check that the name will not be subject to expansion.
9253  */
9254 static void
9255 prehash(union node *n)
9256 {
9257         struct cmdentry entry;
9258
9259         if (n->type == NCMD && n->ncmd.args && goodname(n->ncmd.args->narg.text))
9260                 find_command(n->ncmd.args->narg.text, &entry, 0, pathval());
9261 }
9262
9263
9264 /* ============ Builtin commands
9265  *
9266  * Builtin commands whose functions are closely tied to evaluation
9267  * are implemented here.
9268  */
9269
9270 /*
9271  * Handle break and continue commands.  Break, continue, and return are
9272  * all handled by setting the evalskip flag.  The evaluation routines
9273  * above all check this flag, and if it is set they start skipping
9274  * commands rather than executing them.  The variable skipcount is
9275  * the number of loops to break/continue, or the number of function
9276  * levels to return.  (The latter is always 1.)  It should probably
9277  * be an error to break out of more loops than exist, but it isn't
9278  * in the standard shell so we don't make it one here.
9279  */
9280 static int FAST_FUNC
9281 breakcmd(int argc UNUSED_PARAM, char **argv)
9282 {
9283         int n = argv[1] ? number(argv[1]) : 1;
9284
9285         if (n <= 0)
9286                 ash_msg_and_raise_error(msg_illnum, argv[1]);
9287         if (n > loopnest)
9288                 n = loopnest;
9289         if (n > 0) {
9290                 evalskip = (**argv == 'c') ? SKIPCONT : SKIPBREAK;
9291                 skipcount = n;
9292         }
9293         return 0;
9294 }
9295
9296
9297 /* ============ input.c
9298  *
9299  * This implements the input routines used by the parser.
9300  */
9301
9302 enum {
9303         INPUT_PUSH_FILE = 1,
9304         INPUT_NOFILE_OK = 2,
9305 };
9306
9307 static smallint checkkwd;
9308 /* values of checkkwd variable */
9309 #define CHKALIAS        0x1
9310 #define CHKKWD          0x2
9311 #define CHKNL           0x4
9312
9313 /*
9314  * Push a string back onto the input at this current parsefile level.
9315  * We handle aliases this way.
9316  */
9317 #if !ENABLE_ASH_ALIAS
9318 #define pushstring(s, ap) pushstring(s)
9319 #endif
9320 static void
9321 pushstring(char *s, struct alias *ap)
9322 {
9323         struct strpush *sp;
9324         int len;
9325
9326         len = strlen(s);
9327         INT_OFF;
9328         if (g_parsefile->strpush) {
9329                 sp = ckzalloc(sizeof(*sp));
9330                 sp->prev = g_parsefile->strpush;
9331         } else {
9332                 sp = &(g_parsefile->basestrpush);
9333         }
9334         g_parsefile->strpush = sp;
9335         sp->prev_string = g_parsefile->next_to_pgetc;
9336         sp->prev_left_in_line = g_parsefile->left_in_line;
9337 #if ENABLE_ASH_ALIAS
9338         sp->ap = ap;
9339         if (ap) {
9340                 ap->flag |= ALIASINUSE;
9341                 sp->string = s;
9342         }
9343 #endif
9344         g_parsefile->next_to_pgetc = s;
9345         g_parsefile->left_in_line = len;
9346         INT_ON;
9347 }
9348
9349 static void
9350 popstring(void)
9351 {
9352         struct strpush *sp = g_parsefile->strpush;
9353
9354         INT_OFF;
9355 #if ENABLE_ASH_ALIAS
9356         if (sp->ap) {
9357                 if (g_parsefile->next_to_pgetc[-1] == ' '
9358                  || g_parsefile->next_to_pgetc[-1] == '\t'
9359                 ) {
9360                         checkkwd |= CHKALIAS;
9361                 }
9362                 if (sp->string != sp->ap->val) {
9363                         free(sp->string);
9364                 }
9365                 sp->ap->flag &= ~ALIASINUSE;
9366                 if (sp->ap->flag & ALIASDEAD) {
9367                         unalias(sp->ap->name);
9368                 }
9369         }
9370 #endif
9371         g_parsefile->next_to_pgetc = sp->prev_string;
9372         g_parsefile->left_in_line = sp->prev_left_in_line;
9373         g_parsefile->strpush = sp->prev;
9374         if (sp != &(g_parsefile->basestrpush))
9375                 free(sp);
9376         INT_ON;
9377 }
9378
9379 //FIXME: BASH_COMPAT with "...&" does TWO pungetc():
9380 //it peeks whether it is &>, and then pushes back both chars.
9381 //This function needs to save last *next_to_pgetc to buf[0]
9382 //to make two pungetc() reliable. Currently,
9383 // pgetc (out of buf: does preadfd), pgetc, pungetc, pungetc won't work...
9384 static int
9385 preadfd(void)
9386 {
9387         int nr;
9388         char *buf = g_parsefile->buf;
9389
9390         g_parsefile->next_to_pgetc = buf;
9391 #if ENABLE_FEATURE_EDITING
9392  retry:
9393         if (!iflag || g_parsefile->fd != STDIN_FILENO)
9394                 nr = nonblock_safe_read(g_parsefile->fd, buf, BUFSIZ - 1);
9395         else {
9396 #if ENABLE_FEATURE_TAB_COMPLETION
9397                 line_input_state->path_lookup = pathval();
9398 #endif
9399                 nr = read_line_input(cmdedit_prompt, buf, BUFSIZ, line_input_state);
9400                 if (nr == 0) {
9401                         /* Ctrl+C pressed */
9402                         if (trap[SIGINT]) {
9403                                 buf[0] = '\n';
9404                                 buf[1] = '\0';
9405                                 raise(SIGINT);
9406                                 return 1;
9407                         }
9408                         goto retry;
9409                 }
9410                 if (nr < 0 && errno == 0) {
9411                         /* Ctrl+D pressed */
9412                         nr = 0;
9413                 }
9414         }
9415 #else
9416         nr = nonblock_safe_read(g_parsefile->fd, buf, BUFSIZ - 1);
9417 #endif
9418
9419 #if 0
9420 /* nonblock_safe_read() handles this problem */
9421         if (nr < 0) {
9422                 if (parsefile->fd == 0 && errno == EWOULDBLOCK) {
9423                         int flags = fcntl(0, F_GETFL);
9424                         if (flags >= 0 && (flags & O_NONBLOCK)) {
9425                                 flags &= ~O_NONBLOCK;
9426                                 if (fcntl(0, F_SETFL, flags) >= 0) {
9427                                         out2str("sh: turning off NDELAY mode\n");
9428                                         goto retry;
9429                                 }
9430                         }
9431                 }
9432         }
9433 #endif
9434         return nr;
9435 }
9436
9437 /*
9438  * Refill the input buffer and return the next input character:
9439  *
9440  * 1) If a string was pushed back on the input, pop it;
9441  * 2) If an EOF was pushed back (g_parsefile->left_in_line < -BIGNUM)
9442  *    or we are reading from a string so we can't refill the buffer,
9443  *    return EOF.
9444  * 3) If there is more stuff in this buffer, use it else call read to fill it.
9445  * 4) Process input up to the next newline, deleting nul characters.
9446  */
9447 //#define pgetc_debug(...) bb_error_msg(__VA_ARGS__)
9448 #define pgetc_debug(...) ((void)0)
9449 static int
9450 preadbuffer(void)
9451 {
9452         char *q;
9453         int more;
9454
9455         while (g_parsefile->strpush) {
9456 #if ENABLE_ASH_ALIAS
9457                 if (g_parsefile->left_in_line == -1
9458                  && g_parsefile->strpush->ap
9459                  && g_parsefile->next_to_pgetc[-1] != ' '
9460                  && g_parsefile->next_to_pgetc[-1] != '\t'
9461                 ) {
9462                         pgetc_debug("preadbuffer PEOA");
9463                         return PEOA;
9464                 }
9465 #endif
9466                 popstring();
9467                 /* try "pgetc" now: */
9468                 pgetc_debug("preadbuffer internal pgetc at %d:%p'%s'",
9469                                 g_parsefile->left_in_line,
9470                                 g_parsefile->next_to_pgetc,
9471                                 g_parsefile->next_to_pgetc);
9472                 if (--g_parsefile->left_in_line >= 0)
9473                         return (unsigned char)(*g_parsefile->next_to_pgetc++);
9474         }
9475         /* on both branches above g_parsefile->left_in_line < 0.
9476          * "pgetc" needs refilling.
9477          */
9478
9479         /* -90 is our -BIGNUM. Below we use -99 to mark "EOF on read",
9480          * pungetc() may increment it a few times.
9481          * Assuming it won't increment it to less than -90.
9482          */
9483         if (g_parsefile->left_in_line < -90 || g_parsefile->buf == NULL) {
9484                 pgetc_debug("preadbuffer PEOF1");
9485                 /* even in failure keep left_in_line and next_to_pgetc
9486                  * in lock step, for correct multi-layer pungetc.
9487                  * left_in_line was decremented before preadbuffer(),
9488                  * must inc next_to_pgetc: */
9489                 g_parsefile->next_to_pgetc++;
9490                 return PEOF;
9491         }
9492
9493         more = g_parsefile->left_in_buffer;
9494         if (more <= 0) {
9495                 flush_stdout_stderr();
9496  again:
9497                 more = preadfd();
9498                 if (more <= 0) {
9499                         /* don't try reading again */
9500                         g_parsefile->left_in_line = -99;
9501                         pgetc_debug("preadbuffer PEOF2");
9502                         g_parsefile->next_to_pgetc++;
9503                         return PEOF;
9504                 }
9505         }
9506
9507         /* Find out where's the end of line.
9508          * Set g_parsefile->left_in_line
9509          * and g_parsefile->left_in_buffer acordingly.
9510          * NUL chars are deleted.
9511          */
9512         q = g_parsefile->next_to_pgetc;
9513         for (;;) {
9514                 char c;
9515
9516                 more--;
9517
9518                 c = *q;
9519                 if (c == '\0') {
9520                         memmove(q, q + 1, more);
9521                 } else {
9522                         q++;
9523                         if (c == '\n') {
9524                                 g_parsefile->left_in_line = q - g_parsefile->next_to_pgetc - 1;
9525                                 break;
9526                         }
9527                 }
9528
9529                 if (more <= 0) {
9530                         g_parsefile->left_in_line = q - g_parsefile->next_to_pgetc - 1;
9531                         if (g_parsefile->left_in_line < 0)
9532                                 goto again;
9533                         break;
9534                 }
9535         }
9536         g_parsefile->left_in_buffer = more;
9537
9538         if (vflag) {
9539                 char save = *q;
9540                 *q = '\0';
9541                 out2str(g_parsefile->next_to_pgetc);
9542                 *q = save;
9543         }
9544
9545         pgetc_debug("preadbuffer at %d:%p'%s'",
9546                         g_parsefile->left_in_line,
9547                         g_parsefile->next_to_pgetc,
9548                         g_parsefile->next_to_pgetc);
9549         return (unsigned char)*g_parsefile->next_to_pgetc++;
9550 }
9551
9552 #define pgetc_as_macro() \
9553         (--g_parsefile->left_in_line >= 0 \
9554         ? (unsigned char)*g_parsefile->next_to_pgetc++ \
9555         : preadbuffer() \
9556         )
9557
9558 static int
9559 pgetc(void)
9560 {
9561         pgetc_debug("pgetc_fast at %d:%p'%s'",
9562                         g_parsefile->left_in_line,
9563                         g_parsefile->next_to_pgetc,
9564                         g_parsefile->next_to_pgetc);
9565         return pgetc_as_macro();
9566 }
9567
9568 #if ENABLE_ASH_OPTIMIZE_FOR_SIZE
9569 # define pgetc_fast() pgetc()
9570 #else
9571 # define pgetc_fast() pgetc_as_macro()
9572 #endif
9573
9574 #if ENABLE_ASH_ALIAS
9575 static int
9576 pgetc_without_PEOA(void)
9577 {
9578         int c;
9579         do {
9580                 pgetc_debug("pgetc_fast at %d:%p'%s'",
9581                                 g_parsefile->left_in_line,
9582                                 g_parsefile->next_to_pgetc,
9583                                 g_parsefile->next_to_pgetc);
9584                 c = pgetc_fast();
9585         } while (c == PEOA);
9586         return c;
9587 }
9588 #else
9589 # define pgetc_without_PEOA() pgetc()
9590 #endif
9591
9592 /*
9593  * Read a line from the script.
9594  */
9595 static char *
9596 pfgets(char *line, int len)
9597 {
9598         char *p = line;
9599         int nleft = len;
9600         int c;
9601
9602         while (--nleft > 0) {
9603                 c = pgetc_without_PEOA();
9604                 if (c == PEOF) {
9605                         if (p == line)
9606                                 return NULL;
9607                         break;
9608                 }
9609                 *p++ = c;
9610                 if (c == '\n')
9611                         break;
9612         }
9613         *p = '\0';
9614         return line;
9615 }
9616
9617 /*
9618  * Undo the last call to pgetc.  Only one character may be pushed back.
9619  * PEOF may be pushed back.
9620  */
9621 static void
9622 pungetc(void)
9623 {
9624         g_parsefile->left_in_line++;
9625         g_parsefile->next_to_pgetc--;
9626         pgetc_debug("pushed back to %d:%p'%s'",
9627                         g_parsefile->left_in_line,
9628                         g_parsefile->next_to_pgetc,
9629                         g_parsefile->next_to_pgetc);
9630 }
9631
9632 /*
9633  * To handle the "." command, a stack of input files is used.  Pushfile
9634  * adds a new entry to the stack and popfile restores the previous level.
9635  */
9636 static void
9637 pushfile(void)
9638 {
9639         struct parsefile *pf;
9640
9641         pf = ckzalloc(sizeof(*pf));
9642         pf->prev = g_parsefile;
9643         pf->fd = -1;
9644         /*pf->strpush = NULL; - ckzalloc did it */
9645         /*pf->basestrpush.prev = NULL;*/
9646         g_parsefile = pf;
9647 }
9648
9649 static void
9650 popfile(void)
9651 {
9652         struct parsefile *pf = g_parsefile;
9653
9654         INT_OFF;
9655         if (pf->fd >= 0)
9656                 close(pf->fd);
9657         free(pf->buf);
9658         while (pf->strpush)
9659                 popstring();
9660         g_parsefile = pf->prev;
9661         free(pf);
9662         INT_ON;
9663 }
9664
9665 /*
9666  * Return to top level.
9667  */
9668 static void
9669 popallfiles(void)
9670 {
9671         while (g_parsefile != &basepf)
9672                 popfile();
9673 }
9674
9675 /*
9676  * Close the file(s) that the shell is reading commands from.  Called
9677  * after a fork is done.
9678  */
9679 static void
9680 closescript(void)
9681 {
9682         popallfiles();
9683         if (g_parsefile->fd > 0) {
9684                 close(g_parsefile->fd);
9685                 g_parsefile->fd = 0;
9686         }
9687 }
9688
9689 /*
9690  * Like setinputfile, but takes an open file descriptor.  Call this with
9691  * interrupts off.
9692  */
9693 static void
9694 setinputfd(int fd, int push)
9695 {
9696         close_on_exec_on(fd);
9697         if (push) {
9698                 pushfile();
9699                 g_parsefile->buf = NULL;
9700         }
9701         g_parsefile->fd = fd;
9702         if (g_parsefile->buf == NULL)
9703                 g_parsefile->buf = ckmalloc(IBUFSIZ);
9704         g_parsefile->left_in_buffer = 0;
9705         g_parsefile->left_in_line = 0;
9706         g_parsefile->linno = 1;
9707 }
9708
9709 /*
9710  * Set the input to take input from a file.  If push is set, push the
9711  * old input onto the stack first.
9712  */
9713 static int
9714 setinputfile(const char *fname, int flags)
9715 {
9716         int fd;
9717         int fd2;
9718
9719         INT_OFF;
9720         fd = open(fname, O_RDONLY);
9721         if (fd < 0) {
9722                 if (flags & INPUT_NOFILE_OK)
9723                         goto out;
9724                 ash_msg_and_raise_error("can't open '%s'", fname);
9725         }
9726         if (fd < 10) {
9727                 fd2 = copyfd(fd, 10);
9728                 close(fd);
9729                 if (fd2 < 0)
9730                         ash_msg_and_raise_error("out of file descriptors");
9731                 fd = fd2;
9732         }
9733         setinputfd(fd, flags & INPUT_PUSH_FILE);
9734  out:
9735         INT_ON;
9736         return fd;
9737 }
9738
9739 /*
9740  * Like setinputfile, but takes input from a string.
9741  */
9742 static void
9743 setinputstring(char *string)
9744 {
9745         INT_OFF;
9746         pushfile();
9747         g_parsefile->next_to_pgetc = string;
9748         g_parsefile->left_in_line = strlen(string);
9749         g_parsefile->buf = NULL;
9750         g_parsefile->linno = 1;
9751         INT_ON;
9752 }
9753
9754
9755 /* ============ mail.c
9756  *
9757  * Routines to check for mail.
9758  */
9759
9760 #if ENABLE_ASH_MAIL
9761
9762 #define MAXMBOXES 10
9763
9764 /* times of mailboxes */
9765 static time_t mailtime[MAXMBOXES];
9766 /* Set if MAIL or MAILPATH is changed. */
9767 static smallint mail_var_path_changed;
9768
9769 /*
9770  * Print appropriate message(s) if mail has arrived.
9771  * If mail_var_path_changed is set,
9772  * then the value of MAIL has mail_var_path_changed,
9773  * so we just update the values.
9774  */
9775 static void
9776 chkmail(void)
9777 {
9778         const char *mpath;
9779         char *p;
9780         char *q;
9781         time_t *mtp;
9782         struct stackmark smark;
9783         struct stat statb;
9784
9785         setstackmark(&smark);
9786         mpath = mpathset() ? mpathval() : mailval();
9787         for (mtp = mailtime; mtp < mailtime + MAXMBOXES; mtp++) {
9788                 p = path_advance(&mpath, nullstr);
9789                 if (p == NULL)
9790                         break;
9791                 if (*p == '\0')
9792                         continue;
9793                 for (q = p; *q; q++)
9794                         continue;
9795 #if DEBUG
9796                 if (q[-1] != '/')
9797                         abort();
9798 #endif
9799                 q[-1] = '\0';                   /* delete trailing '/' */
9800                 if (stat(p, &statb) < 0) {
9801                         *mtp = 0;
9802                         continue;
9803                 }
9804                 if (!mail_var_path_changed && statb.st_mtime != *mtp) {
9805                         fprintf(
9806                                 stderr, snlfmt,
9807                                 pathopt ? pathopt : "you have mail"
9808                         );
9809                 }
9810                 *mtp = statb.st_mtime;
9811         }
9812         mail_var_path_changed = 0;
9813         popstackmark(&smark);
9814 }
9815
9816 static void FAST_FUNC
9817 changemail(const char *val UNUSED_PARAM)
9818 {
9819         mail_var_path_changed = 1;
9820 }
9821
9822 #endif /* ASH_MAIL */
9823
9824
9825 /* ============ ??? */
9826
9827 /*
9828  * Set the shell parameters.
9829  */
9830 static void
9831 setparam(char **argv)
9832 {
9833         char **newparam;
9834         char **ap;
9835         int nparam;
9836
9837         for (nparam = 0; argv[nparam]; nparam++)
9838                 continue;
9839         ap = newparam = ckmalloc((nparam + 1) * sizeof(*ap));
9840         while (*argv) {
9841                 *ap++ = ckstrdup(*argv++);
9842         }
9843         *ap = NULL;
9844         freeparam(&shellparam);
9845         shellparam.malloced = 1;
9846         shellparam.nparam = nparam;
9847         shellparam.p = newparam;
9848 #if ENABLE_ASH_GETOPTS
9849         shellparam.optind = 1;
9850         shellparam.optoff = -1;
9851 #endif
9852 }
9853
9854 /*
9855  * Process shell options.  The global variable argptr contains a pointer
9856  * to the argument list; we advance it past the options.
9857  *
9858  * SUSv3 section 2.8.1 "Consequences of Shell Errors" says:
9859  * For a non-interactive shell, an error condition encountered
9860  * by a special built-in ... shall cause the shell to write a diagnostic message
9861  * to standard error and exit as shown in the following table:
9862  * Error                                           Special Built-In
9863  * ...
9864  * Utility syntax error (option or operand error)  Shall exit
9865  * ...
9866  * However, in bug 1142 (http://busybox.net/bugs/view.php?id=1142)
9867  * we see that bash does not do that (set "finishes" with error code 1 instead,
9868  * and shell continues), and people rely on this behavior!
9869  * Testcase:
9870  * set -o barfoo 2>/dev/null
9871  * echo $?
9872  *
9873  * Oh well. Let's mimic that.
9874  */
9875 static int
9876 plus_minus_o(char *name, int val)
9877 {
9878         int i;
9879
9880         if (name) {
9881                 for (i = 0; i < NOPTS; i++) {
9882                         if (strcmp(name, optnames(i)) == 0) {
9883                                 optlist[i] = val;
9884                                 return 0;
9885                         }
9886                 }
9887                 ash_msg("illegal option %co %s", val ? '-' : '+', name);
9888                 return 1;
9889         }
9890         for (i = 0; i < NOPTS; i++) {
9891                 if (val) {
9892                         out1fmt("%-16s%s\n", optnames(i), optlist[i] ? "on" : "off");
9893                 } else {
9894                         out1fmt("set %co %s\n", optlist[i] ? '-' : '+', optnames(i));
9895                 }
9896         }
9897         return 0;
9898 }
9899 static void
9900 setoption(int flag, int val)
9901 {
9902         int i;
9903
9904         for (i = 0; i < NOPTS; i++) {
9905                 if (optletters(i) == flag) {
9906                         optlist[i] = val;
9907                         return;
9908                 }
9909         }
9910         ash_msg_and_raise_error("illegal option %c%c", val ? '-' : '+', flag);
9911         /* NOTREACHED */
9912 }
9913 static int
9914 options(int cmdline)
9915 {
9916         char *p;
9917         int val;
9918         int c;
9919
9920         if (cmdline)
9921                 minusc = NULL;
9922         while ((p = *argptr) != NULL) {
9923                 c = *p++;
9924                 if (c != '-' && c != '+')
9925                         break;
9926                 argptr++;
9927                 val = 0; /* val = 0 if c == '+' */
9928                 if (c == '-') {
9929                         val = 1;
9930                         if (p[0] == '\0' || LONE_DASH(p)) {
9931                                 if (!cmdline) {
9932                                         /* "-" means turn off -x and -v */
9933                                         if (p[0] == '\0')
9934                                                 xflag = vflag = 0;
9935                                         /* "--" means reset params */
9936                                         else if (*argptr == NULL)
9937                                                 setparam(argptr);
9938                                 }
9939                                 break;    /* "-" or  "--" terminates options */
9940                         }
9941                 }
9942                 /* first char was + or - */
9943                 while ((c = *p++) != '\0') {
9944                         /* bash 3.2 indeed handles -c CMD and +c CMD the same */
9945                         if (c == 'c' && cmdline) {
9946                                 minusc = p;     /* command is after shell args */
9947                         } else if (c == 'o') {
9948                                 if (plus_minus_o(*argptr, val)) {
9949                                         /* it already printed err message */
9950                                         return 1; /* error */
9951                                 }
9952                                 if (*argptr)
9953                                         argptr++;
9954                         } else if (cmdline && (c == 'l')) { /* -l or +l == --login */
9955                                 isloginsh = 1;
9956                         /* bash does not accept +-login, we also won't */
9957                         } else if (cmdline && val && (c == '-')) { /* long options */
9958                                 if (strcmp(p, "login") == 0)
9959                                         isloginsh = 1;
9960                                 break;
9961                         } else {
9962                                 setoption(c, val);
9963                         }
9964                 }
9965         }
9966         return 0;
9967 }
9968
9969 /*
9970  * The shift builtin command.
9971  */
9972 static int FAST_FUNC
9973 shiftcmd(int argc UNUSED_PARAM, char **argv)
9974 {
9975         int n;
9976         char **ap1, **ap2;
9977
9978         n = 1;
9979         if (argv[1])
9980                 n = number(argv[1]);
9981         if (n > shellparam.nparam)
9982                 n = 0; /* bash compat, was = shellparam.nparam; */
9983         INT_OFF;
9984         shellparam.nparam -= n;
9985         for (ap1 = shellparam.p; --n >= 0; ap1++) {
9986                 if (shellparam.malloced)
9987                         free(*ap1);
9988         }
9989         ap2 = shellparam.p;
9990         while ((*ap2++ = *ap1++) != NULL)
9991                 continue;
9992 #if ENABLE_ASH_GETOPTS
9993         shellparam.optind = 1;
9994         shellparam.optoff = -1;
9995 #endif
9996         INT_ON;
9997         return 0;
9998 }
9999
10000 /*
10001  * POSIX requires that 'set' (but not export or readonly) output the
10002  * variables in lexicographic order - by the locale's collating order (sigh).
10003  * Maybe we could keep them in an ordered balanced binary tree
10004  * instead of hashed lists.
10005  * For now just roll 'em through qsort for printing...
10006  */
10007 static int
10008 showvars(const char *sep_prefix, int on, int off)
10009 {
10010         const char *sep;
10011         char **ep, **epend;
10012
10013         ep = listvars(on, off, &epend);
10014         qsort(ep, epend - ep, sizeof(char *), vpcmp);
10015
10016         sep = *sep_prefix ? " " : sep_prefix;
10017
10018         for (; ep < epend; ep++) {
10019                 const char *p;
10020                 const char *q;
10021
10022                 p = strchrnul(*ep, '=');
10023                 q = nullstr;
10024                 if (*p)
10025                         q = single_quote(++p);
10026                 out1fmt("%s%s%.*s%s\n", sep_prefix, sep, (int)(p - *ep), *ep, q);
10027         }
10028         return 0;
10029 }
10030
10031 /*
10032  * The set command builtin.
10033  */
10034 static int FAST_FUNC
10035 setcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
10036 {
10037         int retval;
10038
10039         if (!argv[1])
10040                 return showvars(nullstr, 0, VUNSET);
10041         INT_OFF;
10042         retval = 1;
10043         if (!options(0)) { /* if no parse error... */
10044                 retval = 0;
10045                 optschanged();
10046                 if (*argptr != NULL) {
10047                         setparam(argptr);
10048                 }
10049         }
10050         INT_ON;
10051         return retval;
10052 }
10053
10054 #if ENABLE_ASH_RANDOM_SUPPORT
10055 static void FAST_FUNC
10056 change_random(const char *value)
10057 {
10058         uint32_t t;
10059
10060         if (value == NULL) {
10061                 /* "get", generate */
10062                 t = next_random(&random_gen);
10063                 /* set without recursion */
10064                 setvar(vrandom.text, utoa(t), VNOFUNC);
10065                 vrandom.flags &= ~VNOFUNC;
10066         } else {
10067                 /* set/reset */
10068                 t = strtoul(value, NULL, 10);
10069                 INIT_RANDOM_T(&random_gen, (t ? t : 1), t);
10070         }
10071 }
10072 #endif
10073
10074 #if ENABLE_ASH_GETOPTS
10075 static int
10076 getopts(char *optstr, char *optvar, char **optfirst, int *param_optind, int *optoff)
10077 {
10078         char *p, *q;
10079         char c = '?';
10080         int done = 0;
10081         int err = 0;
10082         char s[12];
10083         char **optnext;
10084
10085         if (*param_optind < 1)
10086                 return 1;
10087         optnext = optfirst + *param_optind - 1;
10088
10089         if (*param_optind <= 1 || *optoff < 0 || (int)strlen(optnext[-1]) < *optoff)
10090                 p = NULL;
10091         else
10092                 p = optnext[-1] + *optoff;
10093         if (p == NULL || *p == '\0') {
10094                 /* Current word is done, advance */
10095                 p = *optnext;
10096                 if (p == NULL || *p != '-' || *++p == '\0') {
10097  atend:
10098                         p = NULL;
10099                         done = 1;
10100                         goto out;
10101                 }
10102                 optnext++;
10103                 if (LONE_DASH(p))        /* check for "--" */
10104                         goto atend;
10105         }
10106
10107         c = *p++;
10108         for (q = optstr; *q != c;) {
10109                 if (*q == '\0') {
10110                         if (optstr[0] == ':') {
10111                                 s[0] = c;
10112                                 s[1] = '\0';
10113                                 err |= setvarsafe("OPTARG", s, 0);
10114                         } else {
10115                                 fprintf(stderr, "Illegal option -%c\n", c);
10116                                 unsetvar("OPTARG");
10117                         }
10118                         c = '?';
10119                         goto out;
10120                 }
10121                 if (*++q == ':')
10122                         q++;
10123         }
10124
10125         if (*++q == ':') {
10126                 if (*p == '\0' && (p = *optnext) == NULL) {
10127                         if (optstr[0] == ':') {
10128                                 s[0] = c;
10129                                 s[1] = '\0';
10130                                 err |= setvarsafe("OPTARG", s, 0);
10131                                 c = ':';
10132                         } else {
10133                                 fprintf(stderr, "No arg for -%c option\n", c);
10134                                 unsetvar("OPTARG");
10135                                 c = '?';
10136                         }
10137                         goto out;
10138                 }
10139
10140                 if (p == *optnext)
10141                         optnext++;
10142                 err |= setvarsafe("OPTARG", p, 0);
10143                 p = NULL;
10144         } else
10145                 err |= setvarsafe("OPTARG", nullstr, 0);
10146  out:
10147         *optoff = p ? p - *(optnext - 1) : -1;
10148         *param_optind = optnext - optfirst + 1;
10149         fmtstr(s, sizeof(s), "%d", *param_optind);
10150         err |= setvarsafe("OPTIND", s, VNOFUNC);
10151         s[0] = c;
10152         s[1] = '\0';
10153         err |= setvarsafe(optvar, s, 0);
10154         if (err) {
10155                 *param_optind = 1;
10156                 *optoff = -1;
10157                 flush_stdout_stderr();
10158                 raise_exception(EXERROR);
10159         }
10160         return done;
10161 }
10162
10163 /*
10164  * The getopts builtin.  Shellparam.optnext points to the next argument
10165  * to be processed.  Shellparam.optptr points to the next character to
10166  * be processed in the current argument.  If shellparam.optnext is NULL,
10167  * then it's the first time getopts has been called.
10168  */
10169 static int FAST_FUNC
10170 getoptscmd(int argc, char **argv)
10171 {
10172         char **optbase;
10173
10174         if (argc < 3)
10175                 ash_msg_and_raise_error("usage: getopts optstring var [arg]");
10176         if (argc == 3) {
10177                 optbase = shellparam.p;
10178                 if (shellparam.optind > shellparam.nparam + 1) {
10179                         shellparam.optind = 1;
10180                         shellparam.optoff = -1;
10181                 }
10182         } else {
10183                 optbase = &argv[3];
10184                 if (shellparam.optind > argc - 2) {
10185                         shellparam.optind = 1;
10186                         shellparam.optoff = -1;
10187                 }
10188         }
10189
10190         return getopts(argv[1], argv[2], optbase, &shellparam.optind,
10191                         &shellparam.optoff);
10192 }
10193 #endif /* ASH_GETOPTS */
10194
10195
10196 /* ============ Shell parser */
10197
10198 struct heredoc {
10199         struct heredoc *next;   /* next here document in list */
10200         union node *here;       /* redirection node */
10201         char *eofmark;          /* string indicating end of input */
10202         smallint striptabs;     /* if set, strip leading tabs */
10203 };
10204
10205 static smallint tokpushback;           /* last token pushed back */
10206 static smallint parsebackquote;        /* nonzero if we are inside backquotes */
10207 static smallint quoteflag;             /* set if (part of) last token was quoted */
10208 static token_id_t lasttoken;           /* last token read (integer id Txxx) */
10209 static struct heredoc *heredoclist;    /* list of here documents to read */
10210 static char *wordtext;                 /* text of last word returned by readtoken */
10211 static struct nodelist *backquotelist;
10212 static union node *redirnode;
10213 static struct heredoc *heredoc;
10214
10215 /*
10216  * Called when an unexpected token is read during the parse.  The argument
10217  * is the token that is expected, or -1 if more than one type of token can
10218  * occur at this point.
10219  */
10220 static void raise_error_unexpected_syntax(int) NORETURN;
10221 static void
10222 raise_error_unexpected_syntax(int token)
10223 {
10224         char msg[64];
10225         int l;
10226
10227         l = sprintf(msg, "unexpected %s", tokname(lasttoken));
10228         if (token >= 0)
10229                 sprintf(msg + l, " (expecting %s)", tokname(token));
10230         raise_error_syntax(msg);
10231         /* NOTREACHED */
10232 }
10233
10234 #define EOFMARKLEN 79
10235
10236 /* parsing is heavily cross-recursive, need these forward decls */
10237 static union node *andor(void);
10238 static union node *pipeline(void);
10239 static union node *parse_command(void);
10240 static void parseheredoc(void);
10241 static char peektoken(void);
10242 static int readtoken(void);
10243
10244 static union node *
10245 list(int nlflag)
10246 {
10247         union node *n1, *n2, *n3;
10248         int tok;
10249
10250         checkkwd = CHKNL | CHKKWD | CHKALIAS;
10251         if (nlflag == 2 && peektoken())
10252                 return NULL;
10253         n1 = NULL;
10254         for (;;) {
10255                 n2 = andor();
10256                 tok = readtoken();
10257                 if (tok == TBACKGND) {
10258                         if (n2->type == NPIPE) {
10259                                 n2->npipe.pipe_backgnd = 1;
10260                         } else {
10261                                 if (n2->type != NREDIR) {
10262                                         n3 = stzalloc(sizeof(struct nredir));
10263                                         n3->nredir.n = n2;
10264                                         /*n3->nredir.redirect = NULL; - stzalloc did it */
10265                                         n2 = n3;
10266                                 }
10267                                 n2->type = NBACKGND;
10268                         }
10269                 }
10270                 if (n1 == NULL) {
10271                         n1 = n2;
10272                 } else {
10273                         n3 = stzalloc(sizeof(struct nbinary));
10274                         n3->type = NSEMI;
10275                         n3->nbinary.ch1 = n1;
10276                         n3->nbinary.ch2 = n2;
10277                         n1 = n3;
10278                 }
10279                 switch (tok) {
10280                 case TBACKGND:
10281                 case TSEMI:
10282                         tok = readtoken();
10283                         /* fall through */
10284                 case TNL:
10285                         if (tok == TNL) {
10286                                 parseheredoc();
10287                                 if (nlflag == 1)
10288                                         return n1;
10289                         } else {
10290                                 tokpushback = 1;
10291                         }
10292                         checkkwd = CHKNL | CHKKWD | CHKALIAS;
10293                         if (peektoken())
10294                                 return n1;
10295                         break;
10296                 case TEOF:
10297                         if (heredoclist)
10298                                 parseheredoc();
10299                         else
10300                                 pungetc();              /* push back EOF on input */
10301                         return n1;
10302                 default:
10303                         if (nlflag == 1)
10304                                 raise_error_unexpected_syntax(-1);
10305                         tokpushback = 1;
10306                         return n1;
10307                 }
10308         }
10309 }
10310
10311 static union node *
10312 andor(void)
10313 {
10314         union node *n1, *n2, *n3;
10315         int t;
10316
10317         n1 = pipeline();
10318         for (;;) {
10319                 t = readtoken();
10320                 if (t == TAND) {
10321                         t = NAND;
10322                 } else if (t == TOR) {
10323                         t = NOR;
10324                 } else {
10325                         tokpushback = 1;
10326                         return n1;
10327                 }
10328                 checkkwd = CHKNL | CHKKWD | CHKALIAS;
10329                 n2 = pipeline();
10330                 n3 = stzalloc(sizeof(struct nbinary));
10331                 n3->type = t;
10332                 n3->nbinary.ch1 = n1;
10333                 n3->nbinary.ch2 = n2;
10334                 n1 = n3;
10335         }
10336 }
10337
10338 static union node *
10339 pipeline(void)
10340 {
10341         union node *n1, *n2, *pipenode;
10342         struct nodelist *lp, *prev;
10343         int negate;
10344
10345         negate = 0;
10346         TRACE(("pipeline: entered\n"));
10347         if (readtoken() == TNOT) {
10348                 negate = !negate;
10349                 checkkwd = CHKKWD | CHKALIAS;
10350         } else
10351                 tokpushback = 1;
10352         n1 = parse_command();
10353         if (readtoken() == TPIPE) {
10354                 pipenode = stzalloc(sizeof(struct npipe));
10355                 pipenode->type = NPIPE;
10356                 /*pipenode->npipe.pipe_backgnd = 0; - stzalloc did it */
10357                 lp = stzalloc(sizeof(struct nodelist));
10358                 pipenode->npipe.cmdlist = lp;
10359                 lp->n = n1;
10360                 do {
10361                         prev = lp;
10362                         lp = stzalloc(sizeof(struct nodelist));
10363                         checkkwd = CHKNL | CHKKWD | CHKALIAS;
10364                         lp->n = parse_command();
10365                         prev->next = lp;
10366                 } while (readtoken() == TPIPE);
10367                 lp->next = NULL;
10368                 n1 = pipenode;
10369         }
10370         tokpushback = 1;
10371         if (negate) {
10372                 n2 = stzalloc(sizeof(struct nnot));
10373                 n2->type = NNOT;
10374                 n2->nnot.com = n1;
10375                 return n2;
10376         }
10377         return n1;
10378 }
10379
10380 static union node *
10381 makename(void)
10382 {
10383         union node *n;
10384
10385         n = stzalloc(sizeof(struct narg));
10386         n->type = NARG;
10387         /*n->narg.next = NULL; - stzalloc did it */
10388         n->narg.text = wordtext;
10389         n->narg.backquote = backquotelist;
10390         return n;
10391 }
10392
10393 static void
10394 fixredir(union node *n, const char *text, int err)
10395 {
10396         int fd;
10397
10398         TRACE(("Fix redir %s %d\n", text, err));
10399         if (!err)
10400                 n->ndup.vname = NULL;
10401
10402         fd = bb_strtou(text, NULL, 10);
10403         if (!errno && fd >= 0)
10404                 n->ndup.dupfd = fd;
10405         else if (LONE_DASH(text))
10406                 n->ndup.dupfd = -1;
10407         else {
10408                 if (err)
10409                         raise_error_syntax("bad fd number");
10410                 n->ndup.vname = makename();
10411         }
10412 }
10413
10414 /*
10415  * Returns true if the text contains nothing to expand (no dollar signs
10416  * or backquotes).
10417  */
10418 static int
10419 noexpand(const char *text)
10420 {
10421         unsigned char c;
10422
10423         while ((c = *text++) != '\0') {
10424                 if (c == CTLQUOTEMARK)
10425                         continue;
10426                 if (c == CTLESC)
10427                         text++;
10428                 else if (SIT(c, BASESYNTAX) == CCTL)
10429                         return 0;
10430         }
10431         return 1;
10432 }
10433
10434 static void
10435 parsefname(void)
10436 {
10437         union node *n = redirnode;
10438
10439         if (readtoken() != TWORD)
10440                 raise_error_unexpected_syntax(-1);
10441         if (n->type == NHERE) {
10442                 struct heredoc *here = heredoc;
10443                 struct heredoc *p;
10444                 int i;
10445
10446                 if (quoteflag == 0)
10447                         n->type = NXHERE;
10448                 TRACE(("Here document %d\n", n->type));
10449                 if (!noexpand(wordtext) || (i = strlen(wordtext)) == 0 || i > EOFMARKLEN)
10450                         raise_error_syntax("illegal eof marker for << redirection");
10451                 rmescapes(wordtext, 0);
10452                 here->eofmark = wordtext;
10453                 here->next = NULL;
10454                 if (heredoclist == NULL)
10455                         heredoclist = here;
10456                 else {
10457                         for (p = heredoclist; p->next; p = p->next)
10458                                 continue;
10459                         p->next = here;
10460                 }
10461         } else if (n->type == NTOFD || n->type == NFROMFD) {
10462                 fixredir(n, wordtext, 0);
10463         } else {
10464                 n->nfile.fname = makename();
10465         }
10466 }
10467
10468 static union node *
10469 simplecmd(void)
10470 {
10471         union node *args, **app;
10472         union node *n = NULL;
10473         union node *vars, **vpp;
10474         union node **rpp, *redir;
10475         int savecheckkwd;
10476 #if ENABLE_ASH_BASH_COMPAT
10477         smallint double_brackets_flag = 0;
10478 #endif
10479
10480         args = NULL;
10481         app = &args;
10482         vars = NULL;
10483         vpp = &vars;
10484         redir = NULL;
10485         rpp = &redir;
10486
10487         savecheckkwd = CHKALIAS;
10488         for (;;) {
10489                 int t;
10490                 checkkwd = savecheckkwd;
10491                 t = readtoken();
10492                 switch (t) {
10493 #if ENABLE_ASH_BASH_COMPAT
10494                 case TAND: /* "&&" */
10495                 case TOR: /* "||" */
10496                         if (!double_brackets_flag) {
10497                                 tokpushback = 1;
10498                                 goto out;
10499                         }
10500                         wordtext = (char *) (t == TAND ? "-a" : "-o");
10501 #endif
10502                 case TWORD:
10503                         n = stzalloc(sizeof(struct narg));
10504                         n->type = NARG;
10505                         /*n->narg.next = NULL; - stzalloc did it */
10506                         n->narg.text = wordtext;
10507 #if ENABLE_ASH_BASH_COMPAT
10508                         if (strcmp("[[", wordtext) == 0)
10509                                 double_brackets_flag = 1;
10510                         else if (strcmp("]]", wordtext) == 0)
10511                                 double_brackets_flag = 0;
10512 #endif
10513                         n->narg.backquote = backquotelist;
10514                         if (savecheckkwd && isassignment(wordtext)) {
10515                                 *vpp = n;
10516                                 vpp = &n->narg.next;
10517                         } else {
10518                                 *app = n;
10519                                 app = &n->narg.next;
10520                                 savecheckkwd = 0;
10521                         }
10522                         break;
10523                 case TREDIR:
10524                         *rpp = n = redirnode;
10525                         rpp = &n->nfile.next;
10526                         parsefname();   /* read name of redirection file */
10527                         break;
10528                 case TLP:
10529                         if (args && app == &args->narg.next
10530                          && !vars && !redir
10531                         ) {
10532                                 struct builtincmd *bcmd;
10533                                 const char *name;
10534
10535                                 /* We have a function */
10536                                 if (readtoken() != TRP)
10537                                         raise_error_unexpected_syntax(TRP);
10538                                 name = n->narg.text;
10539                                 if (!goodname(name)
10540                                  || ((bcmd = find_builtin(name)) && IS_BUILTIN_SPECIAL(bcmd))
10541                                 ) {
10542                                         raise_error_syntax("bad function name");
10543                                 }
10544                                 n->type = NDEFUN;
10545                                 checkkwd = CHKNL | CHKKWD | CHKALIAS;
10546                                 n->narg.next = parse_command();
10547                                 return n;
10548                         }
10549                         /* fall through */
10550                 default:
10551                         tokpushback = 1;
10552                         goto out;
10553                 }
10554         }
10555  out:
10556         *app = NULL;
10557         *vpp = NULL;
10558         *rpp = NULL;
10559         n = stzalloc(sizeof(struct ncmd));
10560         n->type = NCMD;
10561         n->ncmd.args = args;
10562         n->ncmd.assign = vars;
10563         n->ncmd.redirect = redir;
10564         return n;
10565 }
10566
10567 static union node *
10568 parse_command(void)
10569 {
10570         union node *n1, *n2;
10571         union node *ap, **app;
10572         union node *cp, **cpp;
10573         union node *redir, **rpp;
10574         union node **rpp2;
10575         int t;
10576
10577         redir = NULL;
10578         rpp2 = &redir;
10579
10580         switch (readtoken()) {
10581         default:
10582                 raise_error_unexpected_syntax(-1);
10583                 /* NOTREACHED */
10584         case TIF:
10585                 n1 = stzalloc(sizeof(struct nif));
10586                 n1->type = NIF;
10587                 n1->nif.test = list(0);
10588                 if (readtoken() != TTHEN)
10589                         raise_error_unexpected_syntax(TTHEN);
10590                 n1->nif.ifpart = list(0);
10591                 n2 = n1;
10592                 while (readtoken() == TELIF) {
10593                         n2->nif.elsepart = stzalloc(sizeof(struct nif));
10594                         n2 = n2->nif.elsepart;
10595                         n2->type = NIF;
10596                         n2->nif.test = list(0);
10597                         if (readtoken() != TTHEN)
10598                                 raise_error_unexpected_syntax(TTHEN);
10599                         n2->nif.ifpart = list(0);
10600                 }
10601                 if (lasttoken == TELSE)
10602                         n2->nif.elsepart = list(0);
10603                 else {
10604                         n2->nif.elsepart = NULL;
10605                         tokpushback = 1;
10606                 }
10607                 t = TFI;
10608                 break;
10609         case TWHILE:
10610         case TUNTIL: {
10611                 int got;
10612                 n1 = stzalloc(sizeof(struct nbinary));
10613                 n1->type = (lasttoken == TWHILE) ? NWHILE : NUNTIL;
10614                 n1->nbinary.ch1 = list(0);
10615                 got = readtoken();
10616                 if (got != TDO) {
10617                         TRACE(("expecting DO got %s %s\n", tokname(got),
10618                                         got == TWORD ? wordtext : ""));
10619                         raise_error_unexpected_syntax(TDO);
10620                 }
10621                 n1->nbinary.ch2 = list(0);
10622                 t = TDONE;
10623                 break;
10624         }
10625         case TFOR:
10626                 if (readtoken() != TWORD || quoteflag || !goodname(wordtext))
10627                         raise_error_syntax("bad for loop variable");
10628                 n1 = stzalloc(sizeof(struct nfor));
10629                 n1->type = NFOR;
10630                 n1->nfor.var = wordtext;
10631                 checkkwd = CHKKWD | CHKALIAS;
10632                 if (readtoken() == TIN) {
10633                         app = &ap;
10634                         while (readtoken() == TWORD) {
10635                                 n2 = stzalloc(sizeof(struct narg));
10636                                 n2->type = NARG;
10637                                 /*n2->narg.next = NULL; - stzalloc did it */
10638                                 n2->narg.text = wordtext;
10639                                 n2->narg.backquote = backquotelist;
10640                                 *app = n2;
10641                                 app = &n2->narg.next;
10642                         }
10643                         *app = NULL;
10644                         n1->nfor.args = ap;
10645                         if (lasttoken != TNL && lasttoken != TSEMI)
10646                                 raise_error_unexpected_syntax(-1);
10647                 } else {
10648                         n2 = stzalloc(sizeof(struct narg));
10649                         n2->type = NARG;
10650                         /*n2->narg.next = NULL; - stzalloc did it */
10651                         n2->narg.text = (char *)dolatstr;
10652                         /*n2->narg.backquote = NULL;*/
10653                         n1->nfor.args = n2;
10654                         /*
10655                          * Newline or semicolon here is optional (but note
10656                          * that the original Bourne shell only allowed NL).
10657                          */
10658                         if (lasttoken != TNL && lasttoken != TSEMI)
10659                                 tokpushback = 1;
10660                 }
10661                 checkkwd = CHKNL | CHKKWD | CHKALIAS;
10662                 if (readtoken() != TDO)
10663                         raise_error_unexpected_syntax(TDO);
10664                 n1->nfor.body = list(0);
10665                 t = TDONE;
10666                 break;
10667         case TCASE:
10668                 n1 = stzalloc(sizeof(struct ncase));
10669                 n1->type = NCASE;
10670                 if (readtoken() != TWORD)
10671                         raise_error_unexpected_syntax(TWORD);
10672                 n1->ncase.expr = n2 = stzalloc(sizeof(struct narg));
10673                 n2->type = NARG;
10674                 /*n2->narg.next = NULL; - stzalloc did it */
10675                 n2->narg.text = wordtext;
10676                 n2->narg.backquote = backquotelist;
10677                 do {
10678                         checkkwd = CHKKWD | CHKALIAS;
10679                 } while (readtoken() == TNL);
10680                 if (lasttoken != TIN)
10681                         raise_error_unexpected_syntax(TIN);
10682                 cpp = &n1->ncase.cases;
10683  next_case:
10684                 checkkwd = CHKNL | CHKKWD;
10685                 t = readtoken();
10686                 while (t != TESAC) {
10687                         if (lasttoken == TLP)
10688                                 readtoken();
10689                         *cpp = cp = stzalloc(sizeof(struct nclist));
10690                         cp->type = NCLIST;
10691                         app = &cp->nclist.pattern;
10692                         for (;;) {
10693                                 *app = ap = stzalloc(sizeof(struct narg));
10694                                 ap->type = NARG;
10695                                 /*ap->narg.next = NULL; - stzalloc did it */
10696                                 ap->narg.text = wordtext;
10697                                 ap->narg.backquote = backquotelist;
10698                                 if (readtoken() != TPIPE)
10699                                         break;
10700                                 app = &ap->narg.next;
10701                                 readtoken();
10702                         }
10703                         //ap->narg.next = NULL;
10704                         if (lasttoken != TRP)
10705                                 raise_error_unexpected_syntax(TRP);
10706                         cp->nclist.body = list(2);
10707
10708                         cpp = &cp->nclist.next;
10709
10710                         checkkwd = CHKNL | CHKKWD;
10711                         t = readtoken();
10712                         if (t != TESAC) {
10713                                 if (t != TENDCASE)
10714                                         raise_error_unexpected_syntax(TENDCASE);
10715                                 goto next_case;
10716                         }
10717                 }
10718                 *cpp = NULL;
10719                 goto redir;
10720         case TLP:
10721                 n1 = stzalloc(sizeof(struct nredir));
10722                 n1->type = NSUBSHELL;
10723                 n1->nredir.n = list(0);
10724                 /*n1->nredir.redirect = NULL; - stzalloc did it */
10725                 t = TRP;
10726                 break;
10727         case TBEGIN:
10728                 n1 = list(0);
10729                 t = TEND;
10730                 break;
10731         case TWORD:
10732         case TREDIR:
10733                 tokpushback = 1;
10734                 return simplecmd();
10735         }
10736
10737         if (readtoken() != t)
10738                 raise_error_unexpected_syntax(t);
10739
10740  redir:
10741         /* Now check for redirection which may follow command */
10742         checkkwd = CHKKWD | CHKALIAS;
10743         rpp = rpp2;
10744         while (readtoken() == TREDIR) {
10745                 *rpp = n2 = redirnode;
10746                 rpp = &n2->nfile.next;
10747                 parsefname();
10748         }
10749         tokpushback = 1;
10750         *rpp = NULL;
10751         if (redir) {
10752                 if (n1->type != NSUBSHELL) {
10753                         n2 = stzalloc(sizeof(struct nredir));
10754                         n2->type = NREDIR;
10755                         n2->nredir.n = n1;
10756                         n1 = n2;
10757                 }
10758                 n1->nredir.redirect = redir;
10759         }
10760         return n1;
10761 }
10762
10763 #if ENABLE_ASH_BASH_COMPAT
10764 static int decode_dollar_squote(void)
10765 {
10766         static const char C_escapes[] ALIGN1 = "nrbtfav""x\\01234567";
10767         int c, cnt;
10768         char *p;
10769         char buf[4];
10770
10771         c = pgetc();
10772         p = strchr(C_escapes, c);
10773         if (p) {
10774                 buf[0] = c;
10775                 p = buf;
10776                 cnt = 3;
10777                 if ((unsigned char)(c - '0') <= 7) { /* \ooo */
10778                         do {
10779                                 c = pgetc();
10780                                 *++p = c;
10781                         } while ((unsigned char)(c - '0') <= 7 && --cnt);
10782                         pungetc();
10783                 } else if (c == 'x') { /* \xHH */
10784                         do {
10785                                 c = pgetc();
10786                                 *++p = c;
10787                         } while (isxdigit(c) && --cnt);
10788                         pungetc();
10789                         if (cnt == 3) { /* \x but next char is "bad" */
10790                                 c = 'x';
10791                                 goto unrecognized;
10792                         }
10793                 } else { /* simple seq like \\ or \t */
10794                         p++;
10795                 }
10796                 *p = '\0';
10797                 p = buf;
10798                 c = bb_process_escape_sequence((void*)&p);
10799         } else { /* unrecognized "\z": print both chars unless ' or " */
10800                 if (c != '\'' && c != '"') {
10801  unrecognized:
10802                         c |= 0x100; /* "please encode \, then me" */
10803                 }
10804         }
10805         return c;
10806 }
10807 #endif
10808
10809 /*
10810  * If eofmark is NULL, read a word or a redirection symbol.  If eofmark
10811  * is not NULL, read a here document.  In the latter case, eofmark is the
10812  * word which marks the end of the document and striptabs is true if
10813  * leading tabs should be stripped from the document.  The argument c
10814  * is the first character of the input token or document.
10815  *
10816  * Because C does not have internal subroutines, I have simulated them
10817  * using goto's to implement the subroutine linkage.  The following macros
10818  * will run code that appears at the end of readtoken1.
10819  */
10820 #define CHECKEND()      {goto checkend; checkend_return:;}
10821 #define PARSEREDIR()    {goto parseredir; parseredir_return:;}
10822 #define PARSESUB()      {goto parsesub; parsesub_return:;}
10823 #define PARSEBACKQOLD() {oldstyle = 1; goto parsebackq; parsebackq_oldreturn:;}
10824 #define PARSEBACKQNEW() {oldstyle = 0; goto parsebackq; parsebackq_newreturn:;}
10825 #define PARSEARITH()    {goto parsearith; parsearith_return:;}
10826 static int
10827 readtoken1(int c, int syntax, char *eofmark, int striptabs)
10828 {
10829         /* NB: syntax parameter fits into smallint */
10830         /* c parameter is an unsigned char or PEOF or PEOA */
10831         char *out;
10832         int len;
10833         char line[EOFMARKLEN + 1];
10834         struct nodelist *bqlist;
10835         smallint quotef;
10836         smallint dblquote;
10837         smallint oldstyle;
10838         smallint prevsyntax; /* syntax before arithmetic */
10839 #if ENABLE_ASH_EXPAND_PRMT
10840         smallint pssyntax;   /* we are expanding a prompt string */
10841 #endif
10842         int varnest;         /* levels of variables expansion */
10843         int arinest;         /* levels of arithmetic expansion */
10844         int parenlevel;      /* levels of parens in arithmetic */
10845         int dqvarnest;       /* levels of variables expansion within double quotes */
10846
10847         IF_ASH_BASH_COMPAT(smallint bash_dollar_squote = 0;)
10848
10849 #if __GNUC__
10850         /* Avoid longjmp clobbering */
10851         (void) &out;
10852         (void) &quotef;
10853         (void) &dblquote;
10854         (void) &varnest;
10855         (void) &arinest;
10856         (void) &parenlevel;
10857         (void) &dqvarnest;
10858         (void) &oldstyle;
10859         (void) &prevsyntax;
10860         (void) &syntax;
10861 #endif
10862         startlinno = g_parsefile->linno;
10863         bqlist = NULL;
10864         quotef = 0;
10865         oldstyle = 0;
10866         prevsyntax = 0;
10867 #if ENABLE_ASH_EXPAND_PRMT
10868         pssyntax = (syntax == PSSYNTAX);
10869         if (pssyntax)
10870                 syntax = DQSYNTAX;
10871 #endif
10872         dblquote = (syntax == DQSYNTAX);
10873         varnest = 0;
10874         arinest = 0;
10875         parenlevel = 0;
10876         dqvarnest = 0;
10877
10878         STARTSTACKSTR(out);
10879  loop:
10880         /* For each line, until end of word */
10881         {
10882                 CHECKEND();     /* set c to PEOF if at end of here document */
10883                 for (;;) {      /* until end of line or end of word */
10884                         CHECKSTRSPACE(4, out);  /* permit 4 calls to USTPUTC */
10885                         switch (SIT(c, syntax)) {
10886                         case CNL:       /* '\n' */
10887                                 if (syntax == BASESYNTAX)
10888                                         goto endword;   /* exit outer loop */
10889                                 USTPUTC(c, out);
10890                                 g_parsefile->linno++;
10891                                 if (doprompt)
10892                                         setprompt(2);
10893                                 c = pgetc();
10894                                 goto loop;              /* continue outer loop */
10895                         case CWORD:
10896                                 USTPUTC(c, out);
10897                                 break;
10898                         case CCTL:
10899                                 if (eofmark == NULL || dblquote)
10900                                         USTPUTC(CTLESC, out);
10901 #if ENABLE_ASH_BASH_COMPAT
10902                                 if (c == '\\' && bash_dollar_squote) {
10903                                         c = decode_dollar_squote();
10904                                         if (c & 0x100) {
10905                                                 USTPUTC('\\', out);
10906                                                 c = (unsigned char)c;
10907                                         }
10908                                 }
10909 #endif
10910                                 USTPUTC(c, out);
10911                                 break;
10912                         case CBACK:     /* backslash */
10913                                 c = pgetc_without_PEOA();
10914                                 if (c == PEOF) {
10915                                         USTPUTC(CTLESC, out);
10916                                         USTPUTC('\\', out);
10917                                         pungetc();
10918                                 } else if (c == '\n') {
10919                                         if (doprompt)
10920                                                 setprompt(2);
10921                                 } else {
10922 #if ENABLE_ASH_EXPAND_PRMT
10923                                         if (c == '$' && pssyntax) {
10924                                                 USTPUTC(CTLESC, out);
10925                                                 USTPUTC('\\', out);
10926                                         }
10927 #endif
10928                                         if (dblquote && c != '\\'
10929                                          && c != '`' && c != '$'
10930                                          && (c != '"' || eofmark != NULL)
10931                                         ) {
10932                                                 USTPUTC(CTLESC, out);
10933                                                 USTPUTC('\\', out);
10934                                         }
10935                                         if (SIT(c, SQSYNTAX) == CCTL)
10936                                                 USTPUTC(CTLESC, out);
10937                                         USTPUTC(c, out);
10938                                         quotef = 1;
10939                                 }
10940                                 break;
10941                         case CSQUOTE:
10942                                 syntax = SQSYNTAX;
10943  quotemark:
10944                                 if (eofmark == NULL) {
10945                                         USTPUTC(CTLQUOTEMARK, out);
10946                                 }
10947                                 break;
10948                         case CDQUOTE:
10949                                 syntax = DQSYNTAX;
10950                                 dblquote = 1;
10951                                 goto quotemark;
10952                         case CENDQUOTE:
10953                                 IF_ASH_BASH_COMPAT(bash_dollar_squote = 0;)
10954                                 if (eofmark != NULL && arinest == 0
10955                                  && varnest == 0
10956                                 ) {
10957                                         USTPUTC(c, out);
10958                                 } else {
10959                                         if (dqvarnest == 0) {
10960                                                 syntax = BASESYNTAX;
10961                                                 dblquote = 0;
10962                                         }
10963                                         quotef = 1;
10964                                         goto quotemark;
10965                                 }
10966                                 break;
10967                         case CVAR:      /* '$' */
10968                                 PARSESUB();             /* parse substitution */
10969                                 break;
10970                         case CENDVAR:   /* '}' */
10971                                 if (varnest > 0) {
10972                                         varnest--;
10973                                         if (dqvarnest > 0) {
10974                                                 dqvarnest--;
10975                                         }
10976                                         USTPUTC(CTLENDVAR, out);
10977                                 } else {
10978                                         USTPUTC(c, out);
10979                                 }
10980                                 break;
10981 #if ENABLE_SH_MATH_SUPPORT
10982                         case CLP:       /* '(' in arithmetic */
10983                                 parenlevel++;
10984                                 USTPUTC(c, out);
10985                                 break;
10986                         case CRP:       /* ')' in arithmetic */
10987                                 if (parenlevel > 0) {
10988                                         USTPUTC(c, out);
10989                                         --parenlevel;
10990                                 } else {
10991                                         if (pgetc() == ')') {
10992                                                 if (--arinest == 0) {
10993                                                         USTPUTC(CTLENDARI, out);
10994                                                         syntax = prevsyntax;
10995                                                         dblquote = (syntax == DQSYNTAX);
10996                                                 } else
10997                                                         USTPUTC(')', out);
10998                                         } else {
10999                                                 /*
11000                                                  * unbalanced parens
11001                                                  *  (don't 2nd guess - no error)
11002                                                  */
11003                                                 pungetc();
11004                                                 USTPUTC(')', out);
11005                                         }
11006                                 }
11007                                 break;
11008 #endif
11009                         case CBQUOTE:   /* '`' */
11010                                 PARSEBACKQOLD();
11011                                 break;
11012                         case CENDFILE:
11013                                 goto endword;           /* exit outer loop */
11014                         case CIGN:
11015                                 break;
11016                         default:
11017                                 if (varnest == 0) {
11018 #if ENABLE_ASH_BASH_COMPAT
11019                                         if (c == '&') {
11020                                                 if (pgetc() == '>')
11021                                                         c = 0x100 + '>'; /* flag &> */
11022                                                 pungetc();
11023                                         }
11024 #endif
11025                                         goto endword;   /* exit outer loop */
11026                                 }
11027                                 IF_ASH_ALIAS(if (c != PEOA))
11028                                         USTPUTC(c, out);
11029
11030                         }
11031                         c = pgetc_fast();
11032                 } /* for (;;) */
11033         }
11034  endword:
11035 #if ENABLE_SH_MATH_SUPPORT
11036         if (syntax == ARISYNTAX)
11037                 raise_error_syntax("missing '))'");
11038 #endif
11039         if (syntax != BASESYNTAX && !parsebackquote && eofmark == NULL)
11040                 raise_error_syntax("unterminated quoted string");
11041         if (varnest != 0) {
11042                 startlinno = g_parsefile->linno;
11043                 /* { */
11044                 raise_error_syntax("missing '}'");
11045         }
11046         USTPUTC('\0', out);
11047         len = out - (char *)stackblock();
11048         out = stackblock();
11049         if (eofmark == NULL) {
11050                 if ((c == '>' || c == '<' IF_ASH_BASH_COMPAT( || c == 0x100 + '>'))
11051                  && quotef == 0
11052                 ) {
11053                         if (isdigit_str9(out)) {
11054                                 PARSEREDIR(); /* passed as params: out, c */
11055                                 lasttoken = TREDIR;
11056                                 return lasttoken;
11057                         }
11058                         /* else: non-number X seen, interpret it
11059                          * as "NNNX>file" = "NNNX >file" */
11060                 }
11061                 pungetc();
11062         }
11063         quoteflag = quotef;
11064         backquotelist = bqlist;
11065         grabstackblock(len);
11066         wordtext = out;
11067         lasttoken = TWORD;
11068         return lasttoken;
11069 /* end of readtoken routine */
11070
11071 /*
11072  * Check to see whether we are at the end of the here document.  When this
11073  * is called, c is set to the first character of the next input line.  If
11074  * we are at the end of the here document, this routine sets the c to PEOF.
11075  */
11076 checkend: {
11077         if (eofmark) {
11078 #if ENABLE_ASH_ALIAS
11079                 if (c == PEOA)
11080                         c = pgetc_without_PEOA();
11081 #endif
11082                 if (striptabs) {
11083                         while (c == '\t') {
11084                                 c = pgetc_without_PEOA();
11085                         }
11086                 }
11087                 if (c == *eofmark) {
11088                         if (pfgets(line, sizeof(line)) != NULL) {
11089                                 char *p, *q;
11090
11091                                 p = line;
11092                                 for (q = eofmark + 1; *q && *p == *q; p++, q++)
11093                                         continue;
11094                                 if (*p == '\n' && *q == '\0') {
11095                                         c = PEOF;
11096                                         g_parsefile->linno++;
11097                                         needprompt = doprompt;
11098                                 } else {
11099                                         pushstring(line, NULL);
11100                                 }
11101                         }
11102                 }
11103         }
11104         goto checkend_return;
11105 }
11106
11107 /*
11108  * Parse a redirection operator.  The variable "out" points to a string
11109  * specifying the fd to be redirected.  The variable "c" contains the
11110  * first character of the redirection operator.
11111  */
11112 parseredir: {
11113         /* out is already checked to be a valid number or "" */
11114         int fd = (*out == '\0' ? -1 : atoi(out));
11115         union node *np;
11116
11117         np = stzalloc(sizeof(struct nfile));
11118         if (c == '>') {
11119                 np->nfile.fd = 1;
11120                 c = pgetc();
11121                 if (c == '>')
11122                         np->type = NAPPEND;
11123                 else if (c == '|')
11124                         np->type = NCLOBBER;
11125                 else if (c == '&')
11126                         np->type = NTOFD;
11127                         /* it also can be NTO2 (>&file), but we can't figure it out yet */
11128                 else {
11129                         np->type = NTO;
11130                         pungetc();
11131                 }
11132         }
11133 #if ENABLE_ASH_BASH_COMPAT
11134         else if (c == 0x100 + '>') { /* this flags &> redirection */
11135                 np->nfile.fd = 1;
11136                 pgetc(); /* this is '>', no need to check */
11137                 np->type = NTO2;
11138         }
11139 #endif
11140         else { /* c == '<' */
11141                 /*np->nfile.fd = 0; - stzalloc did it */
11142                 c = pgetc();
11143                 switch (c) {
11144                 case '<':
11145                         if (sizeof(struct nfile) != sizeof(struct nhere)) {
11146                                 np = stzalloc(sizeof(struct nhere));
11147                                 /*np->nfile.fd = 0; - stzalloc did it */
11148                         }
11149                         np->type = NHERE;
11150                         heredoc = stzalloc(sizeof(struct heredoc));
11151                         heredoc->here = np;
11152                         c = pgetc();
11153                         if (c == '-') {
11154                                 heredoc->striptabs = 1;
11155                         } else {
11156                                 /*heredoc->striptabs = 0; - stzalloc did it */
11157                                 pungetc();
11158                         }
11159                         break;
11160
11161                 case '&':
11162                         np->type = NFROMFD;
11163                         break;
11164
11165                 case '>':
11166                         np->type = NFROMTO;
11167                         break;
11168
11169                 default:
11170                         np->type = NFROM;
11171                         pungetc();
11172                         break;
11173                 }
11174         }
11175         if (fd >= 0)
11176                 np->nfile.fd = fd;
11177         redirnode = np;
11178         goto parseredir_return;
11179 }
11180
11181 /*
11182  * Parse a substitution.  At this point, we have read the dollar sign
11183  * and nothing else.
11184  */
11185
11186 /* is_special(c) evaluates to 1 for c in "!#$*-0123456789?@"; 0 otherwise
11187  * (assuming ascii char codes, as the original implementation did) */
11188 #define is_special(c) \
11189         (((unsigned)(c) - 33 < 32) \
11190                         && ((0xc1ff920dU >> ((unsigned)(c) - 33)) & 1))
11191 parsesub: {
11192         unsigned char subtype;
11193         int typeloc;
11194         int flags;
11195         char *p;
11196         static const char types[] ALIGN1 = "}-+?=";
11197
11198         c = pgetc();
11199         if (c > 255 /* PEOA or PEOF */
11200          || (c != '(' && c != '{' && !is_name(c) && !is_special(c))
11201         ) {
11202 #if ENABLE_ASH_BASH_COMPAT
11203                 if (c == '\'')
11204                         bash_dollar_squote = 1;
11205                 else
11206 #endif
11207                         USTPUTC('$', out);
11208                 pungetc();
11209         } else if (c == '(') {  /* $(command) or $((arith)) */
11210                 if (pgetc() == '(') {
11211 #if ENABLE_SH_MATH_SUPPORT
11212                         PARSEARITH();
11213 #else
11214                         raise_error_syntax("you disabled math support for $((arith)) syntax");
11215 #endif
11216                 } else {
11217                         pungetc();
11218                         PARSEBACKQNEW();
11219                 }
11220         } else {
11221                 USTPUTC(CTLVAR, out);
11222                 typeloc = out - (char *)stackblock();
11223                 USTPUTC(VSNORMAL, out);
11224                 subtype = VSNORMAL;
11225                 if (c == '{') {
11226                         c = pgetc();
11227                         if (c == '#') {
11228                                 c = pgetc();
11229                                 if (c == '}')
11230                                         c = '#';
11231                                 else
11232                                         subtype = VSLENGTH;
11233                         } else
11234                                 subtype = 0;
11235                 }
11236                 if (c <= 255 /* not PEOA or PEOF */ && is_name(c)) {
11237                         do {
11238                                 STPUTC(c, out);
11239                                 c = pgetc();
11240                         } while (c <= 255 /* not PEOA or PEOF */ && is_in_name(c));
11241                 } else if (isdigit(c)) {
11242                         do {
11243                                 STPUTC(c, out);
11244                                 c = pgetc();
11245                         } while (isdigit(c));
11246                 } else if (is_special(c)) {
11247                         USTPUTC(c, out);
11248                         c = pgetc();
11249                 } else {
11250  badsub:
11251                         raise_error_syntax("bad substitution");
11252                 }
11253                 if (c != '}' && subtype == VSLENGTH)
11254                         goto badsub;
11255
11256                 STPUTC('=', out);
11257                 flags = 0;
11258                 if (subtype == 0) {
11259                         switch (c) {
11260                         case ':':
11261                                 c = pgetc();
11262 #if ENABLE_ASH_BASH_COMPAT
11263                                 if (c == ':' || c == '$' || isdigit(c)) {
11264                                         pungetc();
11265                                         subtype = VSSUBSTR;
11266                                         break;
11267                                 }
11268 #endif
11269                                 flags = VSNUL;
11270                                 /*FALLTHROUGH*/
11271                         default:
11272                                 p = strchr(types, c);
11273                                 if (p == NULL)
11274                                         goto badsub;
11275                                 subtype = p - types + VSNORMAL;
11276                                 break;
11277                         case '%':
11278                         case '#': {
11279                                 int cc = c;
11280                                 subtype = c == '#' ? VSTRIMLEFT : VSTRIMRIGHT;
11281                                 c = pgetc();
11282                                 if (c == cc)
11283                                         subtype++;
11284                                 else
11285                                         pungetc();
11286                                 break;
11287                         }
11288 #if ENABLE_ASH_BASH_COMPAT
11289                         case '/':
11290                                 subtype = VSREPLACE;
11291                                 c = pgetc();
11292                                 if (c == '/')
11293                                         subtype++; /* VSREPLACEALL */
11294                                 else
11295                                         pungetc();
11296                                 break;
11297 #endif
11298                         }
11299                 } else {
11300                         pungetc();
11301                 }
11302                 if (dblquote || arinest)
11303                         flags |= VSQUOTE;
11304                 ((unsigned char *)stackblock())[typeloc] = subtype | flags;
11305                 if (subtype != VSNORMAL) {
11306                         varnest++;
11307                         if (dblquote || arinest) {
11308                                 dqvarnest++;
11309                         }
11310                 }
11311         }
11312         goto parsesub_return;
11313 }
11314
11315 /*
11316  * Called to parse command substitutions.  Newstyle is set if the command
11317  * is enclosed inside $(...); nlpp is a pointer to the head of the linked
11318  * list of commands (passed by reference), and savelen is the number of
11319  * characters on the top of the stack which must be preserved.
11320  */
11321 parsebackq: {
11322         struct nodelist **nlpp;
11323         smallint savepbq;
11324         union node *n;
11325         char *volatile str;
11326         struct jmploc jmploc;
11327         struct jmploc *volatile savehandler;
11328         size_t savelen;
11329         smallint saveprompt = 0;
11330
11331 #ifdef __GNUC__
11332         (void) &saveprompt;
11333 #endif
11334         savepbq = parsebackquote;
11335         if (setjmp(jmploc.loc)) {
11336                 free(str);
11337                 parsebackquote = 0;
11338                 exception_handler = savehandler;
11339                 longjmp(exception_handler->loc, 1);
11340         }
11341         INT_OFF;
11342         str = NULL;
11343         savelen = out - (char *)stackblock();
11344         if (savelen > 0) {
11345                 str = ckmalloc(savelen);
11346                 memcpy(str, stackblock(), savelen);
11347         }
11348         savehandler = exception_handler;
11349         exception_handler = &jmploc;
11350         INT_ON;
11351         if (oldstyle) {
11352                 /* We must read until the closing backquote, giving special
11353                    treatment to some slashes, and then push the string and
11354                    reread it as input, interpreting it normally.  */
11355                 char *pout;
11356                 int pc;
11357                 size_t psavelen;
11358                 char *pstr;
11359
11360
11361                 STARTSTACKSTR(pout);
11362                 for (;;) {
11363                         if (needprompt) {
11364                                 setprompt(2);
11365                         }
11366                         pc = pgetc();
11367                         switch (pc) {
11368                         case '`':
11369                                 goto done;
11370
11371                         case '\\':
11372                                 pc = pgetc();
11373                                 if (pc == '\n') {
11374                                         g_parsefile->linno++;
11375                                         if (doprompt)
11376                                                 setprompt(2);
11377                                         /*
11378                                          * If eating a newline, avoid putting
11379                                          * the newline into the new character
11380                                          * stream (via the STPUTC after the
11381                                          * switch).
11382                                          */
11383                                         continue;
11384                                 }
11385                                 if (pc != '\\' && pc != '`' && pc != '$'
11386                                  && (!dblquote || pc != '"')
11387                                 ) {
11388                                         STPUTC('\\', pout);
11389                                 }
11390                                 if (pc <= 255 /* not PEOA or PEOF */) {
11391                                         break;
11392                                 }
11393                                 /* fall through */
11394
11395                         case PEOF:
11396                         IF_ASH_ALIAS(case PEOA:)
11397                                 startlinno = g_parsefile->linno;
11398                                 raise_error_syntax("EOF in backquote substitution");
11399
11400                         case '\n':
11401                                 g_parsefile->linno++;
11402                                 needprompt = doprompt;
11403                                 break;
11404
11405                         default:
11406                                 break;
11407                         }
11408                         STPUTC(pc, pout);
11409                 }
11410  done:
11411                 STPUTC('\0', pout);
11412                 psavelen = pout - (char *)stackblock();
11413                 if (psavelen > 0) {
11414                         pstr = grabstackstr(pout);
11415                         setinputstring(pstr);
11416                 }
11417         }
11418         nlpp = &bqlist;
11419         while (*nlpp)
11420                 nlpp = &(*nlpp)->next;
11421         *nlpp = stzalloc(sizeof(**nlpp));
11422         /* (*nlpp)->next = NULL; - stzalloc did it */
11423         parsebackquote = oldstyle;
11424
11425         if (oldstyle) {
11426                 saveprompt = doprompt;
11427                 doprompt = 0;
11428         }
11429
11430         n = list(2);
11431
11432         if (oldstyle)
11433                 doprompt = saveprompt;
11434         else if (readtoken() != TRP)
11435                 raise_error_unexpected_syntax(TRP);
11436
11437         (*nlpp)->n = n;
11438         if (oldstyle) {
11439                 /*
11440                  * Start reading from old file again, ignoring any pushed back
11441                  * tokens left from the backquote parsing
11442                  */
11443                 popfile();
11444                 tokpushback = 0;
11445         }
11446         while (stackblocksize() <= savelen)
11447                 growstackblock();
11448         STARTSTACKSTR(out);
11449         if (str) {
11450                 memcpy(out, str, savelen);
11451                 STADJUST(savelen, out);
11452                 INT_OFF;
11453                 free(str);
11454                 str = NULL;
11455                 INT_ON;
11456         }
11457         parsebackquote = savepbq;
11458         exception_handler = savehandler;
11459         if (arinest || dblquote)
11460                 USTPUTC(CTLBACKQ | CTLQUOTE, out);
11461         else
11462                 USTPUTC(CTLBACKQ, out);
11463         if (oldstyle)
11464                 goto parsebackq_oldreturn;
11465         goto parsebackq_newreturn;
11466 }
11467
11468 #if ENABLE_SH_MATH_SUPPORT
11469 /*
11470  * Parse an arithmetic expansion (indicate start of one and set state)
11471  */
11472 parsearith: {
11473         if (++arinest == 1) {
11474                 prevsyntax = syntax;
11475                 syntax = ARISYNTAX;
11476                 USTPUTC(CTLARI, out);
11477                 if (dblquote)
11478                         USTPUTC('"', out);
11479                 else
11480                         USTPUTC(' ', out);
11481         } else {
11482                 /*
11483                  * we collapse embedded arithmetic expansion to
11484                  * parenthesis, which should be equivalent
11485                  */
11486                 USTPUTC('(', out);
11487         }
11488         goto parsearith_return;
11489 }
11490 #endif
11491
11492 } /* end of readtoken */
11493
11494 /*
11495  * Read the next input token.
11496  * If the token is a word, we set backquotelist to the list of cmds in
11497  *      backquotes.  We set quoteflag to true if any part of the word was
11498  *      quoted.
11499  * If the token is TREDIR, then we set redirnode to a structure containing
11500  *      the redirection.
11501  * In all cases, the variable startlinno is set to the number of the line
11502  *      on which the token starts.
11503  *
11504  * [Change comment:  here documents and internal procedures]
11505  * [Readtoken shouldn't have any arguments.  Perhaps we should make the
11506  *  word parsing code into a separate routine.  In this case, readtoken
11507  *  doesn't need to have any internal procedures, but parseword does.
11508  *  We could also make parseoperator in essence the main routine, and
11509  *  have parseword (readtoken1?) handle both words and redirection.]
11510  */
11511 #define NEW_xxreadtoken
11512 #ifdef NEW_xxreadtoken
11513 /* singles must be first! */
11514 static const char xxreadtoken_chars[7] ALIGN1 = {
11515         '\n', '(', ')', /* singles */
11516         '&', '|', ';',  /* doubles */
11517         0
11518 };
11519
11520 #define xxreadtoken_singles 3
11521 #define xxreadtoken_doubles 3
11522
11523 static const char xxreadtoken_tokens[] ALIGN1 = {
11524         TNL, TLP, TRP,          /* only single occurrence allowed */
11525         TBACKGND, TPIPE, TSEMI, /* if single occurrence */
11526         TEOF,                   /* corresponds to trailing nul */
11527         TAND, TOR, TENDCASE     /* if double occurrence */
11528 };
11529
11530 static int
11531 xxreadtoken(void)
11532 {
11533         int c;
11534
11535         if (tokpushback) {
11536                 tokpushback = 0;
11537                 return lasttoken;
11538         }
11539         if (needprompt) {
11540                 setprompt(2);
11541         }
11542         startlinno = g_parsefile->linno;
11543         for (;;) {                      /* until token or start of word found */
11544                 c = pgetc_fast();
11545                 if (c == ' ' || c == '\t' IF_ASH_ALIAS( || c == PEOA))
11546                         continue;
11547
11548                 if (c == '#') {
11549                         while ((c = pgetc()) != '\n' && c != PEOF)
11550                                 continue;
11551                         pungetc();
11552                 } else if (c == '\\') {
11553                         if (pgetc() != '\n') {
11554                                 pungetc();
11555                                 break; /* return readtoken1(...) */
11556                         }
11557                         startlinno = ++g_parsefile->linno;
11558                         if (doprompt)
11559                                 setprompt(2);
11560                 } else {
11561                         const char *p;
11562
11563                         p = xxreadtoken_chars + sizeof(xxreadtoken_chars) - 1;
11564                         if (c != PEOF) {
11565                                 if (c == '\n') {
11566                                         g_parsefile->linno++;
11567                                         needprompt = doprompt;
11568                                 }
11569
11570                                 p = strchr(xxreadtoken_chars, c);
11571                                 if (p == NULL)
11572                                         break; /* return readtoken1(...) */
11573
11574                                 if ((int)(p - xxreadtoken_chars) >= xxreadtoken_singles) {
11575                                         int cc = pgetc();
11576                                         if (cc == c) {    /* double occurrence? */
11577                                                 p += xxreadtoken_doubles + 1;
11578                                         } else {
11579                                                 pungetc();
11580 #if ENABLE_ASH_BASH_COMPAT
11581                                                 if (c == '&' && cc == '>') /* &> */
11582                                                         break; /* return readtoken1(...) */
11583 #endif
11584                                         }
11585                                 }
11586                         }
11587                         lasttoken = xxreadtoken_tokens[p - xxreadtoken_chars];
11588                         return lasttoken;
11589                 }
11590         } /* for (;;) */
11591
11592         return readtoken1(c, BASESYNTAX, (char *) NULL, 0);
11593 }
11594 #else /* old xxreadtoken */
11595 #define RETURN(token)   return lasttoken = token
11596 static int
11597 xxreadtoken(void)
11598 {
11599         int c;
11600
11601         if (tokpushback) {
11602                 tokpushback = 0;
11603                 return lasttoken;
11604         }
11605         if (needprompt) {
11606                 setprompt(2);
11607         }
11608         startlinno = g_parsefile->linno;
11609         for (;;) {      /* until token or start of word found */
11610                 c = pgetc_fast();
11611                 switch (c) {
11612                 case ' ': case '\t':
11613                 IF_ASH_ALIAS(case PEOA:)
11614                         continue;
11615                 case '#':
11616                         while ((c = pgetc()) != '\n' && c != PEOF)
11617                                 continue;
11618                         pungetc();
11619                         continue;
11620                 case '\\':
11621                         if (pgetc() == '\n') {
11622                                 startlinno = ++g_parsefile->linno;
11623                                 if (doprompt)
11624                                         setprompt(2);
11625                                 continue;
11626                         }
11627                         pungetc();
11628                         goto breakloop;
11629                 case '\n':
11630                         g_parsefile->linno++;
11631                         needprompt = doprompt;
11632                         RETURN(TNL);
11633                 case PEOF:
11634                         RETURN(TEOF);
11635                 case '&':
11636                         if (pgetc() == '&')
11637                                 RETURN(TAND);
11638                         pungetc();
11639                         RETURN(TBACKGND);
11640                 case '|':
11641                         if (pgetc() == '|')
11642                                 RETURN(TOR);
11643                         pungetc();
11644                         RETURN(TPIPE);
11645                 case ';':
11646                         if (pgetc() == ';')
11647                                 RETURN(TENDCASE);
11648                         pungetc();
11649                         RETURN(TSEMI);
11650                 case '(':
11651                         RETURN(TLP);
11652                 case ')':
11653                         RETURN(TRP);
11654                 default:
11655                         goto breakloop;
11656                 }
11657         }
11658  breakloop:
11659         return readtoken1(c, BASESYNTAX, (char *)NULL, 0);
11660 #undef RETURN
11661 }
11662 #endif /* old xxreadtoken */
11663
11664 static int
11665 readtoken(void)
11666 {
11667         int t;
11668 #if DEBUG
11669         smallint alreadyseen = tokpushback;
11670 #endif
11671
11672 #if ENABLE_ASH_ALIAS
11673  top:
11674 #endif
11675
11676         t = xxreadtoken();
11677
11678         /*
11679          * eat newlines
11680          */
11681         if (checkkwd & CHKNL) {
11682                 while (t == TNL) {
11683                         parseheredoc();
11684                         t = xxreadtoken();
11685                 }
11686         }
11687
11688         if (t != TWORD || quoteflag) {
11689                 goto out;
11690         }
11691
11692         /*
11693          * check for keywords
11694          */
11695         if (checkkwd & CHKKWD) {
11696                 const char *const *pp;
11697
11698                 pp = findkwd(wordtext);
11699                 if (pp) {
11700                         lasttoken = t = pp - tokname_array;
11701                         TRACE(("keyword %s recognized\n", tokname(t)));
11702                         goto out;
11703                 }
11704         }
11705
11706         if (checkkwd & CHKALIAS) {
11707 #if ENABLE_ASH_ALIAS
11708                 struct alias *ap;
11709                 ap = lookupalias(wordtext, 1);
11710                 if (ap != NULL) {
11711                         if (*ap->val) {
11712                                 pushstring(ap->val, ap);
11713                         }
11714                         goto top;
11715                 }
11716 #endif
11717         }
11718  out:
11719         checkkwd = 0;
11720 #if DEBUG
11721         if (!alreadyseen)
11722                 TRACE(("token %s %s\n", tokname(t), t == TWORD ? wordtext : ""));
11723         else
11724                 TRACE(("reread token %s %s\n", tokname(t), t == TWORD ? wordtext : ""));
11725 #endif
11726         return t;
11727 }
11728
11729 static char
11730 peektoken(void)
11731 {
11732         int t;
11733
11734         t = readtoken();
11735         tokpushback = 1;
11736         return tokname_array[t][0];
11737 }
11738
11739 /*
11740  * Read and parse a command.  Returns NODE_EOF on end of file.
11741  * (NULL is a valid parse tree indicating a blank line.)
11742  */
11743 static union node *
11744 parsecmd(int interact)
11745 {
11746         int t;
11747
11748         tokpushback = 0;
11749         doprompt = interact;
11750         if (doprompt)
11751                 setprompt(doprompt);
11752         needprompt = 0;
11753         t = readtoken();
11754         if (t == TEOF)
11755                 return NODE_EOF;
11756         if (t == TNL)
11757                 return NULL;
11758         tokpushback = 1;
11759         return list(1);
11760 }
11761
11762 /*
11763  * Input any here documents.
11764  */
11765 static void
11766 parseheredoc(void)
11767 {
11768         struct heredoc *here;
11769         union node *n;
11770
11771         here = heredoclist;
11772         heredoclist = NULL;
11773
11774         while (here) {
11775                 if (needprompt) {
11776                         setprompt(2);
11777                 }
11778                 readtoken1(pgetc(), here->here->type == NHERE? SQSYNTAX : DQSYNTAX,
11779                                 here->eofmark, here->striptabs);
11780                 n = stzalloc(sizeof(struct narg));
11781                 n->narg.type = NARG;
11782                 /*n->narg.next = NULL; - stzalloc did it */
11783                 n->narg.text = wordtext;
11784                 n->narg.backquote = backquotelist;
11785                 here->here->nhere.doc = n;
11786                 here = here->next;
11787         }
11788 }
11789
11790
11791 /*
11792  * called by editline -- any expansions to the prompt should be added here.
11793  */
11794 #if ENABLE_ASH_EXPAND_PRMT
11795 static const char *
11796 expandstr(const char *ps)
11797 {
11798         union node n;
11799
11800         /* XXX Fix (char *) cast. It _is_ a bug. ps is variable's value,
11801          * and token processing _can_ alter it (delete NULs etc). */
11802         setinputstring((char *)ps);
11803         readtoken1(pgetc(), PSSYNTAX, nullstr, 0);
11804         popfile();
11805
11806         n.narg.type = NARG;
11807         n.narg.next = NULL;
11808         n.narg.text = wordtext;
11809         n.narg.backquote = backquotelist;
11810
11811         expandarg(&n, NULL, 0);
11812         return stackblock();
11813 }
11814 #endif
11815
11816 /*
11817  * Execute a command or commands contained in a string.
11818  */
11819 static int
11820 evalstring(char *s, int mask)
11821 {
11822         union node *n;
11823         struct stackmark smark;
11824         int skip;
11825
11826         setinputstring(s);
11827         setstackmark(&smark);
11828
11829         skip = 0;
11830         while ((n = parsecmd(0)) != NODE_EOF) {
11831                 evaltree(n, 0);
11832                 popstackmark(&smark);
11833                 skip = evalskip;
11834                 if (skip)
11835                         break;
11836         }
11837         popfile();
11838
11839         skip &= mask;
11840         evalskip = skip;
11841         return skip;
11842 }
11843
11844 /*
11845  * The eval command.
11846  */
11847 static int FAST_FUNC
11848 evalcmd(int argc UNUSED_PARAM, char **argv)
11849 {
11850         char *p;
11851         char *concat;
11852
11853         if (argv[1]) {
11854                 p = argv[1];
11855                 argv += 2;
11856                 if (argv[0]) {
11857                         STARTSTACKSTR(concat);
11858                         for (;;) {
11859                                 concat = stack_putstr(p, concat);
11860                                 p = *argv++;
11861                                 if (p == NULL)
11862                                         break;
11863                                 STPUTC(' ', concat);
11864                         }
11865                         STPUTC('\0', concat);
11866                         p = grabstackstr(concat);
11867                 }
11868                 evalstring(p, ~SKIPEVAL);
11869
11870         }
11871         return exitstatus;
11872 }
11873
11874 /*
11875  * Read and execute commands.
11876  * "Top" is nonzero for the top level command loop;
11877  * it turns on prompting if the shell is interactive.
11878  */
11879 static int
11880 cmdloop(int top)
11881 {
11882         union node *n;
11883         struct stackmark smark;
11884         int inter;
11885         int numeof = 0;
11886
11887         TRACE(("cmdloop(%d) called\n", top));
11888         for (;;) {
11889                 int skip;
11890
11891                 setstackmark(&smark);
11892 #if JOBS
11893                 if (doing_jobctl)
11894                         showjobs(stderr, SHOW_CHANGED);
11895 #endif
11896                 inter = 0;
11897                 if (iflag && top) {
11898                         inter++;
11899 #if ENABLE_ASH_MAIL
11900                         chkmail();
11901 #endif
11902                 }
11903                 n = parsecmd(inter);
11904 #if DEBUG
11905                 if (DEBUG > 2 && debug && (n != NODE_EOF))
11906                         showtree(n);
11907 #endif
11908                 if (n == NODE_EOF) {
11909                         if (!top || numeof >= 50)
11910                                 break;
11911                         if (!stoppedjobs()) {
11912                                 if (!Iflag)
11913                                         break;
11914                                 out2str("\nUse \"exit\" to leave shell.\n");
11915                         }
11916                         numeof++;
11917                 } else if (nflag == 0) {
11918                         /* job_warning can only be 2,1,0. Here 2->1, 1/0->0 */
11919                         job_warning >>= 1;
11920                         numeof = 0;
11921                         evaltree(n, 0);
11922                 }
11923                 popstackmark(&smark);
11924                 skip = evalskip;
11925
11926                 if (skip) {
11927                         evalskip = 0;
11928                         return skip & SKIPEVAL;
11929                 }
11930         }
11931         return 0;
11932 }
11933
11934 /*
11935  * Take commands from a file.  To be compatible we should do a path
11936  * search for the file, which is necessary to find sub-commands.
11937  */
11938 static char *
11939 find_dot_file(char *name)
11940 {
11941         char *fullname;
11942         const char *path = pathval();
11943         struct stat statb;
11944
11945         /* don't try this for absolute or relative paths */
11946         if (strchr(name, '/'))
11947                 return name;
11948
11949         /* IIRC standards do not say whether . is to be searched.
11950          * And it is even smaller this way, making it unconditional for now:
11951          */
11952         if (1) { /* ENABLE_ASH_BASH_COMPAT */
11953                 fullname = name;
11954                 goto try_cur_dir;
11955         }
11956
11957         while ((fullname = path_advance(&path, name)) != NULL) {
11958  try_cur_dir:
11959                 if ((stat(fullname, &statb) == 0) && S_ISREG(statb.st_mode)) {
11960                         /*
11961                          * Don't bother freeing here, since it will
11962                          * be freed by the caller.
11963                          */
11964                         return fullname;
11965                 }
11966                 if (fullname != name)
11967                         stunalloc(fullname);
11968         }
11969
11970         /* not found in the PATH */
11971         ash_msg_and_raise_error("%s: not found", name);
11972         /* NOTREACHED */
11973 }
11974
11975 static int FAST_FUNC
11976 dotcmd(int argc, char **argv)
11977 {
11978         struct strlist *sp;
11979         volatile struct shparam saveparam;
11980         int status = 0;
11981
11982         for (sp = cmdenviron; sp; sp = sp->next)
11983                 setvareq(ckstrdup(sp->text), VSTRFIXED | VTEXTFIXED);
11984
11985         if (argv[1]) {        /* That's what SVR2 does */
11986                 char *fullname = find_dot_file(argv[1]);
11987                 argv += 2;
11988                 argc -= 2;
11989                 if (argc) { /* argc > 0, argv[0] != NULL */
11990                         saveparam = shellparam;
11991                         shellparam.malloced = 0;
11992                         shellparam.nparam = argc;
11993                         shellparam.p = argv;
11994                 };
11995
11996                 setinputfile(fullname, INPUT_PUSH_FILE);
11997                 commandname = fullname;
11998                 cmdloop(0);
11999                 popfile();
12000
12001                 if (argc) {
12002                         freeparam(&shellparam);
12003                         shellparam = saveparam;
12004                 };
12005                 status = exitstatus;
12006         }
12007         return status;
12008 }
12009
12010 static int FAST_FUNC
12011 exitcmd(int argc UNUSED_PARAM, char **argv)
12012 {
12013         if (stoppedjobs())
12014                 return 0;
12015         if (argv[1])
12016                 exitstatus = number(argv[1]);
12017         raise_exception(EXEXIT);
12018         /* NOTREACHED */
12019 }
12020
12021 /*
12022  * Read a file containing shell functions.
12023  */
12024 static void
12025 readcmdfile(char *name)
12026 {
12027         setinputfile(name, INPUT_PUSH_FILE);
12028         cmdloop(0);
12029         popfile();
12030 }
12031
12032
12033 /* ============ find_command inplementation */
12034
12035 /*
12036  * Resolve a command name.  If you change this routine, you may have to
12037  * change the shellexec routine as well.
12038  */
12039 static void
12040 find_command(char *name, struct cmdentry *entry, int act, const char *path)
12041 {
12042         struct tblentry *cmdp;
12043         int idx;
12044         int prev;
12045         char *fullname;
12046         struct stat statb;
12047         int e;
12048         int updatetbl;
12049         struct builtincmd *bcmd;
12050
12051         /* If name contains a slash, don't use PATH or hash table */
12052         if (strchr(name, '/') != NULL) {
12053                 entry->u.index = -1;
12054                 if (act & DO_ABS) {
12055                         while (stat(name, &statb) < 0) {
12056 #ifdef SYSV
12057                                 if (errno == EINTR)
12058                                         continue;
12059 #endif
12060                                 entry->cmdtype = CMDUNKNOWN;
12061                                 return;
12062                         }
12063                 }
12064                 entry->cmdtype = CMDNORMAL;
12065                 return;
12066         }
12067
12068 /* #if ENABLE_FEATURE_SH_STANDALONE... moved after builtin check */
12069
12070         updatetbl = (path == pathval());
12071         if (!updatetbl) {
12072                 act |= DO_ALTPATH;
12073                 if (strstr(path, "%builtin") != NULL)
12074                         act |= DO_ALTBLTIN;
12075         }
12076
12077         /* If name is in the table, check answer will be ok */
12078         cmdp = cmdlookup(name, 0);
12079         if (cmdp != NULL) {
12080                 int bit;
12081
12082                 switch (cmdp->cmdtype) {
12083                 default:
12084 #if DEBUG
12085                         abort();
12086 #endif
12087                 case CMDNORMAL:
12088                         bit = DO_ALTPATH;
12089                         break;
12090                 case CMDFUNCTION:
12091                         bit = DO_NOFUNC;
12092                         break;
12093                 case CMDBUILTIN:
12094                         bit = DO_ALTBLTIN;
12095                         break;
12096                 }
12097                 if (act & bit) {
12098                         updatetbl = 0;
12099                         cmdp = NULL;
12100                 } else if (cmdp->rehash == 0)
12101                         /* if not invalidated by cd, we're done */
12102                         goto success;
12103         }
12104
12105         /* If %builtin not in path, check for builtin next */
12106         bcmd = find_builtin(name);
12107         if (bcmd) {
12108                 if (IS_BUILTIN_REGULAR(bcmd))
12109                         goto builtin_success;
12110                 if (act & DO_ALTPATH) {
12111                         if (!(act & DO_ALTBLTIN))
12112                                 goto builtin_success;
12113                 } else if (builtinloc <= 0) {
12114                         goto builtin_success;
12115                 }
12116         }
12117
12118 #if ENABLE_FEATURE_SH_STANDALONE
12119         {
12120                 int applet_no = find_applet_by_name(name);
12121                 if (applet_no >= 0) {
12122                         entry->cmdtype = CMDNORMAL;
12123                         entry->u.index = -2 - applet_no;
12124                         return;
12125                 }
12126         }
12127 #endif
12128
12129         /* We have to search path. */
12130         prev = -1;              /* where to start */
12131         if (cmdp && cmdp->rehash) {     /* doing a rehash */
12132                 if (cmdp->cmdtype == CMDBUILTIN)
12133                         prev = builtinloc;
12134                 else
12135                         prev = cmdp->param.index;
12136         }
12137
12138         e = ENOENT;
12139         idx = -1;
12140  loop:
12141         while ((fullname = path_advance(&path, name)) != NULL) {
12142                 stunalloc(fullname);
12143                 /* NB: code below will still use fullname
12144                  * despite it being "unallocated" */
12145                 idx++;
12146                 if (pathopt) {
12147                         if (prefix(pathopt, "builtin")) {
12148                                 if (bcmd)
12149                                         goto builtin_success;
12150                                 continue;
12151                         }
12152                         if ((act & DO_NOFUNC)
12153                          || !prefix(pathopt, "func")
12154                         ) {     /* ignore unimplemented options */
12155                                 continue;
12156                         }
12157                 }
12158                 /* if rehash, don't redo absolute path names */
12159                 if (fullname[0] == '/' && idx <= prev) {
12160                         if (idx < prev)
12161                                 continue;
12162                         TRACE(("searchexec \"%s\": no change\n", name));
12163                         goto success;
12164                 }
12165                 while (stat(fullname, &statb) < 0) {
12166 #ifdef SYSV
12167                         if (errno == EINTR)
12168                                 continue;
12169 #endif
12170                         if (errno != ENOENT && errno != ENOTDIR)
12171                                 e = errno;
12172                         goto loop;
12173                 }
12174                 e = EACCES;     /* if we fail, this will be the error */
12175                 if (!S_ISREG(statb.st_mode))
12176                         continue;
12177                 if (pathopt) {          /* this is a %func directory */
12178                         stalloc(strlen(fullname) + 1);
12179                         /* NB: stalloc will return space pointed by fullname
12180                          * (because we don't have any intervening allocations
12181                          * between stunalloc above and this stalloc) */
12182                         readcmdfile(fullname);
12183                         cmdp = cmdlookup(name, 0);
12184                         if (cmdp == NULL || cmdp->cmdtype != CMDFUNCTION)
12185                                 ash_msg_and_raise_error("%s not defined in %s", name, fullname);
12186                         stunalloc(fullname);
12187                         goto success;
12188                 }
12189                 TRACE(("searchexec \"%s\" returns \"%s\"\n", name, fullname));
12190                 if (!updatetbl) {
12191                         entry->cmdtype = CMDNORMAL;
12192                         entry->u.index = idx;
12193                         return;
12194                 }
12195                 INT_OFF;
12196                 cmdp = cmdlookup(name, 1);
12197                 cmdp->cmdtype = CMDNORMAL;
12198                 cmdp->param.index = idx;
12199                 INT_ON;
12200                 goto success;
12201         }
12202
12203         /* We failed.  If there was an entry for this command, delete it */
12204         if (cmdp && updatetbl)
12205                 delete_cmd_entry();
12206         if (act & DO_ERR)
12207                 ash_msg("%s: %s", name, errmsg(e, "not found"));
12208         entry->cmdtype = CMDUNKNOWN;
12209         return;
12210
12211  builtin_success:
12212         if (!updatetbl) {
12213                 entry->cmdtype = CMDBUILTIN;
12214                 entry->u.cmd = bcmd;
12215                 return;
12216         }
12217         INT_OFF;
12218         cmdp = cmdlookup(name, 1);
12219         cmdp->cmdtype = CMDBUILTIN;
12220         cmdp->param.cmd = bcmd;
12221         INT_ON;
12222  success:
12223         cmdp->rehash = 0;
12224         entry->cmdtype = cmdp->cmdtype;
12225         entry->u = cmdp->param;
12226 }
12227
12228
12229 /* ============ trap.c */
12230
12231 /*
12232  * The trap builtin.
12233  */
12234 static int FAST_FUNC
12235 trapcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12236 {
12237         char *action;
12238         char **ap;
12239         int signo;
12240
12241         nextopt(nullstr);
12242         ap = argptr;
12243         if (!*ap) {
12244                 for (signo = 0; signo < NSIG; signo++) {
12245                         char *tr = trap_ptr[signo];
12246                         if (tr) {
12247                                 /* note: bash adds "SIG", but only if invoked
12248                                  * as "bash". If called as "sh", or if set -o posix,
12249                                  * then it prints short signal names.
12250                                  * We are printing short names: */
12251                                 out1fmt("trap -- %s %s\n",
12252                                                 single_quote(tr),
12253                                                 get_signame(signo));
12254                 /* trap_ptr != trap only if we are in special-cased `trap` code.
12255                  * In this case, we will exit very soon, no need to free(). */
12256                                 /* if (trap_ptr != trap && tp[0]) */
12257                                 /*      free(tr); */
12258                         }
12259                 }
12260                 /*
12261                 if (trap_ptr != trap) {
12262                         free(trap_ptr);
12263                         trap_ptr = trap;
12264                 }
12265                 */
12266                 return 0;
12267         }
12268
12269         action = NULL;
12270         if (ap[1])
12271                 action = *ap++;
12272         while (*ap) {
12273                 signo = get_signum(*ap);
12274                 if (signo < 0)
12275                         ash_msg_and_raise_error("%s: bad trap", *ap);
12276                 INT_OFF;
12277                 if (action) {
12278                         if (LONE_DASH(action))
12279                                 action = NULL;
12280                         else
12281                                 action = ckstrdup(action);
12282                 }
12283                 free(trap[signo]);
12284                 trap[signo] = action;
12285                 if (signo != 0)
12286                         setsignal(signo);
12287                 INT_ON;
12288                 ap++;
12289         }
12290         return 0;
12291 }
12292
12293
12294 /* ============ Builtins */
12295
12296 #if !ENABLE_FEATURE_SH_EXTRA_QUIET
12297 /*
12298  * Lists available builtins
12299  */
12300 static int FAST_FUNC
12301 helpcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12302 {
12303         unsigned col;
12304         unsigned i;
12305
12306         out1fmt(
12307                 "Built-in commands:\n"
12308                 "------------------\n");
12309         for (col = 0, i = 0; i < ARRAY_SIZE(builtintab); i++) {
12310                 col += out1fmt("%c%s", ((col == 0) ? '\t' : ' '),
12311                                         builtintab[i].name + 1);
12312                 if (col > 60) {
12313                         out1fmt("\n");
12314                         col = 0;
12315                 }
12316         }
12317 #if ENABLE_FEATURE_SH_STANDALONE
12318         {
12319                 const char *a = applet_names;
12320                 while (*a) {
12321                         col += out1fmt("%c%s", ((col == 0) ? '\t' : ' '), a);
12322                         if (col > 60) {
12323                                 out1fmt("\n");
12324                                 col = 0;
12325                         }
12326                         a += strlen(a) + 1;
12327                 }
12328         }
12329 #endif
12330         out1fmt("\n\n");
12331         return EXIT_SUCCESS;
12332 }
12333 #endif /* FEATURE_SH_EXTRA_QUIET */
12334
12335 /*
12336  * The export and readonly commands.
12337  */
12338 static int FAST_FUNC
12339 exportcmd(int argc UNUSED_PARAM, char **argv)
12340 {
12341         struct var *vp;
12342         char *name;
12343         const char *p;
12344         char **aptr;
12345         int flag = argv[0][0] == 'r' ? VREADONLY : VEXPORT;
12346
12347         if (nextopt("p") != 'p') {
12348                 aptr = argptr;
12349                 name = *aptr;
12350                 if (name) {
12351                         do {
12352                                 p = strchr(name, '=');
12353                                 if (p != NULL) {
12354                                         p++;
12355                                 } else {
12356                                         vp = *findvar(hashvar(name), name);
12357                                         if (vp) {
12358                                                 vp->flags |= flag;
12359                                                 continue;
12360                                         }
12361                                 }
12362                                 setvar(name, p, flag);
12363                         } while ((name = *++aptr) != NULL);
12364                         return 0;
12365                 }
12366         }
12367         showvars(argv[0], flag, 0);
12368         return 0;
12369 }
12370
12371 /*
12372  * Delete a function if it exists.
12373  */
12374 static void
12375 unsetfunc(const char *name)
12376 {
12377         struct tblentry *cmdp;
12378
12379         cmdp = cmdlookup(name, 0);
12380         if (cmdp!= NULL && cmdp->cmdtype == CMDFUNCTION)
12381                 delete_cmd_entry();
12382 }
12383
12384 /*
12385  * The unset builtin command.  We unset the function before we unset the
12386  * variable to allow a function to be unset when there is a readonly variable
12387  * with the same name.
12388  */
12389 static int FAST_FUNC
12390 unsetcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12391 {
12392         char **ap;
12393         int i;
12394         int flag = 0;
12395         int ret = 0;
12396
12397         while ((i = nextopt("vf")) != '\0') {
12398                 flag = i;
12399         }
12400
12401         for (ap = argptr; *ap; ap++) {
12402                 if (flag != 'f') {
12403                         i = unsetvar(*ap);
12404                         ret |= i;
12405                         if (!(i & 2))
12406                                 continue;
12407                 }
12408                 if (flag != 'v')
12409                         unsetfunc(*ap);
12410         }
12411         return ret & 1;
12412 }
12413
12414
12415 /*      setmode.c      */
12416
12417 #include <sys/times.h>
12418
12419 static const unsigned char timescmd_str[] ALIGN1 = {
12420         ' ',  offsetof(struct tms, tms_utime),
12421         '\n', offsetof(struct tms, tms_stime),
12422         ' ',  offsetof(struct tms, tms_cutime),
12423         '\n', offsetof(struct tms, tms_cstime),
12424         0
12425 };
12426
12427 static int FAST_FUNC
12428 timescmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12429 {
12430         long clk_tck, s, t;
12431         const unsigned char *p;
12432         struct tms buf;
12433
12434         clk_tck = sysconf(_SC_CLK_TCK);
12435         times(&buf);
12436
12437         p = timescmd_str;
12438         do {
12439                 t = *(clock_t *)(((char *) &buf) + p[1]);
12440                 s = t / clk_tck;
12441                 out1fmt("%ldm%ld.%.3lds%c",
12442                         s/60, s%60,
12443                         ((t - s * clk_tck) * 1000) / clk_tck,
12444                         p[0]);
12445         } while (*(p += 2));
12446
12447         return 0;
12448 }
12449
12450 #if ENABLE_SH_MATH_SUPPORT
12451 /*
12452  * The let builtin. partial stolen from GNU Bash, the Bourne Again SHell.
12453  * Copyright (C) 1987, 1989, 1991 Free Software Foundation, Inc.
12454  *
12455  * Copyright (C) 2003 Vladimir Oleynik <dzo@simtreas.ru>
12456  */
12457 static int FAST_FUNC
12458 letcmd(int argc UNUSED_PARAM, char **argv)
12459 {
12460         arith_t i;
12461
12462         argv++;
12463         if (!*argv)
12464                 ash_msg_and_raise_error("expression expected");
12465         do {
12466                 i = ash_arith(*argv);
12467         } while (*++argv);
12468
12469         return !i;
12470 }
12471 #endif /* SH_MATH_SUPPORT */
12472
12473
12474 /* ============ miscbltin.c
12475  *
12476  * Miscellaneous builtins.
12477  */
12478
12479 #undef rflag
12480
12481 #if defined(__GLIBC__) && __GLIBC__ == 2 && __GLIBC_MINOR__ < 1
12482 typedef enum __rlimit_resource rlim_t;
12483 #endif
12484
12485 /*
12486  * The read builtin. Options:
12487  *      -r              Do not interpret '\' specially
12488  *      -s              Turn off echo (tty only)
12489  *      -n NCHARS       Read NCHARS max
12490  *      -p PROMPT       Display PROMPT on stderr (if input is from tty)
12491  *      -t SECONDS      Timeout after SECONDS (tty or pipe only)
12492  *      -u FD           Read from given FD instead of fd 0
12493  * This uses unbuffered input, which may be avoidable in some cases.
12494  * TODO: bash also has:
12495  *      -a ARRAY        Read into array[0],[1],etc
12496  *      -d DELIM        End on DELIM char, not newline
12497  *      -e              Use line editing (tty only)
12498  */
12499 static int FAST_FUNC
12500 readcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12501 {
12502         static const char *const arg_REPLY[] = { "REPLY", NULL };
12503
12504         char **ap;
12505         int backslash;
12506         char c;
12507         int rflag;
12508         char *prompt;
12509         const char *ifs;
12510         char *p;
12511         int startword;
12512         int status;
12513         int i;
12514         int fd = 0;
12515 #if ENABLE_ASH_READ_NCHARS
12516         int nchars = 0; /* if != 0, -n is in effect */
12517         int silent = 0;
12518         struct termios tty, old_tty;
12519 #endif
12520 #if ENABLE_ASH_READ_TIMEOUT
12521         unsigned end_ms = 0;
12522         unsigned timeout = 0;
12523 #endif
12524
12525         rflag = 0;
12526         prompt = NULL;
12527         while ((i = nextopt("p:u:r"
12528                 IF_ASH_READ_TIMEOUT("t:")
12529                 IF_ASH_READ_NCHARS("n:s")
12530         )) != '\0') {
12531                 switch (i) {
12532                 case 'p':
12533                         prompt = optionarg;
12534                         break;
12535 #if ENABLE_ASH_READ_NCHARS
12536                 case 'n':
12537                         nchars = bb_strtou(optionarg, NULL, 10);
12538                         if (nchars < 0 || errno)
12539                                 ash_msg_and_raise_error("invalid count");
12540                         /* nchars == 0: off (bash 3.2 does this too) */
12541                         break;
12542                 case 's':
12543                         silent = 1;
12544                         break;
12545 #endif
12546 #if ENABLE_ASH_READ_TIMEOUT
12547                 case 't':
12548                         timeout = bb_strtou(optionarg, NULL, 10);
12549                         if (errno || timeout > UINT_MAX / 2048)
12550                                 ash_msg_and_raise_error("invalid timeout");
12551                         timeout *= 1000;
12552 #if 0 /* even bash have no -t N.NNN support */
12553                         ts.tv_sec = bb_strtou(optionarg, &p, 10);
12554                         ts.tv_usec = 0;
12555                         /* EINVAL means number is ok, but not terminated by NUL */
12556                         if (*p == '.' && errno == EINVAL) {
12557                                 char *p2;
12558                                 if (*++p) {
12559                                         int scale;
12560                                         ts.tv_usec = bb_strtou(p, &p2, 10);
12561                                         if (errno)
12562                                                 ash_msg_and_raise_error("invalid timeout");
12563                                         scale = p2 - p;
12564                                         /* normalize to usec */
12565                                         if (scale > 6)
12566                                                 ash_msg_and_raise_error("invalid timeout");
12567                                         while (scale++ < 6)
12568                                                 ts.tv_usec *= 10;
12569                                 }
12570                         } else if (ts.tv_sec < 0 || errno) {
12571                                 ash_msg_and_raise_error("invalid timeout");
12572                         }
12573                         if (!(ts.tv_sec | ts.tv_usec)) { /* both are 0? */
12574                                 ash_msg_and_raise_error("invalid timeout");
12575                         }
12576 #endif /* if 0 */
12577                         break;
12578 #endif
12579                 case 'r':
12580                         rflag = 1;
12581                         break;
12582                 case 'u':
12583                         fd = bb_strtou(optionarg, NULL, 10);
12584                         if (fd < 0 || errno)
12585                                 ash_msg_and_raise_error("invalid file descriptor");
12586                         break;
12587                 default:
12588                         break;
12589                 }
12590         }
12591         if (prompt && isatty(fd)) {
12592                 out2str(prompt);
12593         }
12594         ap = argptr;
12595         if (*ap == NULL)
12596                 ap = (char**)arg_REPLY;
12597         ifs = bltinlookup("IFS");
12598         if (ifs == NULL)
12599                 ifs = defifs;
12600 #if ENABLE_ASH_READ_NCHARS
12601         tcgetattr(fd, &tty);
12602         old_tty = tty;
12603         if (nchars || silent) {
12604                 if (nchars) {
12605                         tty.c_lflag &= ~ICANON;
12606                         tty.c_cc[VMIN] = nchars < 256 ? nchars : 255;
12607                 }
12608                 if (silent) {
12609                         tty.c_lflag &= ~(ECHO | ECHOK | ECHONL);
12610                 }
12611                 /* if tcgetattr failed, tcsetattr will fail too.
12612                  * Ignoring, it's harmless. */
12613                 tcsetattr(fd, TCSANOW, &tty);
12614         }
12615 #endif
12616
12617         status = 0;
12618         startword = 2;
12619         backslash = 0;
12620 #if ENABLE_ASH_READ_TIMEOUT
12621         if (timeout) /* NB: ensuring end_ms is nonzero */
12622                 end_ms = ((unsigned)(monotonic_us() / 1000) + timeout) | 1;
12623 #endif
12624         STARTSTACKSTR(p);
12625         do {
12626                 const char *is_ifs;
12627
12628 #if ENABLE_ASH_READ_TIMEOUT
12629                 if (end_ms) {
12630                         struct pollfd pfd[1];
12631                         pfd[0].fd = fd;
12632                         pfd[0].events = POLLIN;
12633                         timeout = end_ms - (unsigned)(monotonic_us() / 1000);
12634                         if ((int)timeout <= 0 /* already late? */
12635                          || safe_poll(pfd, 1, timeout) != 1 /* no? wait... */
12636                         ) { /* timed out! */
12637 #if ENABLE_ASH_READ_NCHARS
12638                                 tcsetattr(fd, TCSANOW, &old_tty);
12639 #endif
12640                                 return 1;
12641                         }
12642                 }
12643 #endif
12644                 if (nonblock_safe_read(fd, &c, 1) != 1) {
12645                         status = 1;
12646                         break;
12647                 }
12648                 if (c == '\0')
12649                         continue;
12650                 if (backslash) {
12651                         backslash = 0;
12652                         if (c != '\n')
12653                                 goto put;
12654                         continue;
12655                 }
12656                 if (!rflag && c == '\\') {
12657                         backslash = 1;
12658                         continue;
12659                 }
12660                 if (c == '\n')
12661                         break;
12662                 /* $IFS splitting */
12663 /* http://www.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#tag_18_06_05 */
12664                 is_ifs = strchr(ifs, c);
12665                 if (startword && is_ifs) {
12666                         if (isspace(c))
12667                                 continue;
12668                         /* it is a non-space ifs char */
12669                         startword--;
12670                         if (startword == 1) /* first one? */
12671                                 continue; /* yes, it is not next word yet */
12672                 }
12673                 startword = 0;
12674                 if (ap[1] != NULL && is_ifs) {
12675                         const char *beg;
12676                         STACKSTRNUL(p);
12677                         beg = stackblock();
12678                         setvar(*ap, beg, 0);
12679                         ap++;
12680                         /* can we skip one non-space ifs char? (2: yes) */
12681                         startword = isspace(c) ? 2 : 1;
12682                         STARTSTACKSTR(p);
12683                         continue;
12684                 }
12685  put:
12686                 STPUTC(c, p);
12687         }
12688 /* end of do {} while: */
12689 #if ENABLE_ASH_READ_NCHARS
12690         while (--nchars);
12691 #else
12692         while (1);
12693 #endif
12694
12695 #if ENABLE_ASH_READ_NCHARS
12696         tcsetattr(fd, TCSANOW, &old_tty);
12697 #endif
12698
12699         STACKSTRNUL(p);
12700         /* Remove trailing space ifs chars */
12701         while ((char *)stackblock() <= --p && isspace(*p) && strchr(ifs, *p) != NULL)
12702                 *p = '\0';
12703         setvar(*ap, stackblock(), 0);
12704         while (*++ap != NULL)
12705                 setvar(*ap, nullstr, 0);
12706         return status;
12707 }
12708
12709 static int FAST_FUNC
12710 umaskcmd(int argc UNUSED_PARAM, char **argv)
12711 {
12712         static const char permuser[3] ALIGN1 = "ugo";
12713         static const char permmode[3] ALIGN1 = "rwx";
12714         static const short permmask[] ALIGN2 = {
12715                 S_IRUSR, S_IWUSR, S_IXUSR,
12716                 S_IRGRP, S_IWGRP, S_IXGRP,
12717                 S_IROTH, S_IWOTH, S_IXOTH
12718         };
12719
12720         /* TODO: use bb_parse_mode() instead */
12721
12722         char *ap;
12723         mode_t mask;
12724         int i;
12725         int symbolic_mode = 0;
12726
12727         while (nextopt("S") != '\0') {
12728                 symbolic_mode = 1;
12729         }
12730
12731         INT_OFF;
12732         mask = umask(0);
12733         umask(mask);
12734         INT_ON;
12735
12736         ap = *argptr;
12737         if (ap == NULL) {
12738                 if (symbolic_mode) {
12739                         char buf[18];
12740                         char *p = buf;
12741
12742                         for (i = 0; i < 3; i++) {
12743                                 int j;
12744
12745                                 *p++ = permuser[i];
12746                                 *p++ = '=';
12747                                 for (j = 0; j < 3; j++) {
12748                                         if ((mask & permmask[3 * i + j]) == 0) {
12749                                                 *p++ = permmode[j];
12750                                         }
12751                                 }
12752                                 *p++ = ',';
12753                         }
12754                         *--p = 0;
12755                         puts(buf);
12756                 } else {
12757                         out1fmt("%.4o\n", mask);
12758                 }
12759         } else {
12760                 if (isdigit((unsigned char) *ap)) {
12761                         mask = 0;
12762                         do {
12763                                 if (*ap >= '8' || *ap < '0')
12764                                         ash_msg_and_raise_error(msg_illnum, argv[1]);
12765                                 mask = (mask << 3) + (*ap - '0');
12766                         } while (*++ap != '\0');
12767                         umask(mask);
12768                 } else {
12769                         mask = ~mask & 0777;
12770                         if (!bb_parse_mode(ap, &mask)) {
12771                                 ash_msg_and_raise_error("illegal mode: %s", ap);
12772                         }
12773                         umask(~mask & 0777);
12774                 }
12775         }
12776         return 0;
12777 }
12778
12779 /*
12780  * ulimit builtin
12781  *
12782  * This code, originally by Doug Gwyn, Doug Kingston, Eric Gisin, and
12783  * Michael Rendell was ripped from pdksh 5.0.8 and hacked for use with
12784  * ash by J.T. Conklin.
12785  *
12786  * Public domain.
12787  */
12788 struct limits {
12789         uint8_t cmd;          /* RLIMIT_xxx fit into it */
12790         uint8_t factor_shift; /* shift by to get rlim_{cur,max} values */
12791         char    option;
12792 };
12793
12794 static const struct limits limits_tbl[] = {
12795 #ifdef RLIMIT_CPU
12796         { RLIMIT_CPU,        0, 't' },
12797 #endif
12798 #ifdef RLIMIT_FSIZE
12799         { RLIMIT_FSIZE,      9, 'f' },
12800 #endif
12801 #ifdef RLIMIT_DATA
12802         { RLIMIT_DATA,      10, 'd' },
12803 #endif
12804 #ifdef RLIMIT_STACK
12805         { RLIMIT_STACK,     10, 's' },
12806 #endif
12807 #ifdef RLIMIT_CORE
12808         { RLIMIT_CORE,       9, 'c' },
12809 #endif
12810 #ifdef RLIMIT_RSS
12811         { RLIMIT_RSS,       10, 'm' },
12812 #endif
12813 #ifdef RLIMIT_MEMLOCK
12814         { RLIMIT_MEMLOCK,   10, 'l' },
12815 #endif
12816 #ifdef RLIMIT_NPROC
12817         { RLIMIT_NPROC,      0, 'p' },
12818 #endif
12819 #ifdef RLIMIT_NOFILE
12820         { RLIMIT_NOFILE,     0, 'n' },
12821 #endif
12822 #ifdef RLIMIT_AS
12823         { RLIMIT_AS,        10, 'v' },
12824 #endif
12825 #ifdef RLIMIT_LOCKS
12826         { RLIMIT_LOCKS,      0, 'w' },
12827 #endif
12828 };
12829 static const char limits_name[] =
12830 #ifdef RLIMIT_CPU
12831         "time(seconds)" "\0"
12832 #endif
12833 #ifdef RLIMIT_FSIZE
12834         "file(blocks)" "\0"
12835 #endif
12836 #ifdef RLIMIT_DATA
12837         "data(kb)" "\0"
12838 #endif
12839 #ifdef RLIMIT_STACK
12840         "stack(kb)" "\0"
12841 #endif
12842 #ifdef RLIMIT_CORE
12843         "coredump(blocks)" "\0"
12844 #endif
12845 #ifdef RLIMIT_RSS
12846         "memory(kb)" "\0"
12847 #endif
12848 #ifdef RLIMIT_MEMLOCK
12849         "locked memory(kb)" "\0"
12850 #endif
12851 #ifdef RLIMIT_NPROC
12852         "process" "\0"
12853 #endif
12854 #ifdef RLIMIT_NOFILE
12855         "nofiles" "\0"
12856 #endif
12857 #ifdef RLIMIT_AS
12858         "vmemory(kb)" "\0"
12859 #endif
12860 #ifdef RLIMIT_LOCKS
12861         "locks" "\0"
12862 #endif
12863 ;
12864
12865 enum limtype { SOFT = 0x1, HARD = 0x2 };
12866
12867 static void
12868 printlim(enum limtype how, const struct rlimit *limit,
12869                         const struct limits *l)
12870 {
12871         rlim_t val;
12872
12873         val = limit->rlim_max;
12874         if (how & SOFT)
12875                 val = limit->rlim_cur;
12876
12877         if (val == RLIM_INFINITY)
12878                 out1fmt("unlimited\n");
12879         else {
12880                 val >>= l->factor_shift;
12881                 out1fmt("%lld\n", (long long) val);
12882         }
12883 }
12884
12885 static int FAST_FUNC
12886 ulimitcmd(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
12887 {
12888         rlim_t val;
12889         enum limtype how = SOFT | HARD;
12890         const struct limits *l;
12891         int set, all = 0;
12892         int optc, what;
12893         struct rlimit limit;
12894
12895         what = 'f';
12896         while ((optc = nextopt("HSa"
12897 #ifdef RLIMIT_CPU
12898                                 "t"
12899 #endif
12900 #ifdef RLIMIT_FSIZE
12901                                 "f"
12902 #endif
12903 #ifdef RLIMIT_DATA
12904                                 "d"
12905 #endif
12906 #ifdef RLIMIT_STACK
12907                                 "s"
12908 #endif
12909 #ifdef RLIMIT_CORE
12910                                 "c"
12911 #endif
12912 #ifdef RLIMIT_RSS
12913                                 "m"
12914 #endif
12915 #ifdef RLIMIT_MEMLOCK
12916                                 "l"
12917 #endif
12918 #ifdef RLIMIT_NPROC
12919                                 "p"
12920 #endif
12921 #ifdef RLIMIT_NOFILE
12922                                 "n"
12923 #endif
12924 #ifdef RLIMIT_AS
12925                                 "v"
12926 #endif
12927 #ifdef RLIMIT_LOCKS
12928                                 "w"
12929 #endif
12930                                         )) != '\0')
12931                 switch (optc) {
12932                 case 'H':
12933                         how = HARD;
12934                         break;
12935                 case 'S':
12936                         how = SOFT;
12937                         break;
12938                 case 'a':
12939                         all = 1;
12940                         break;
12941                 default:
12942                         what = optc;
12943                 }
12944
12945         for (l = limits_tbl; l->option != what; l++)
12946                 continue;
12947
12948         set = *argptr ? 1 : 0;
12949         val = 0;
12950         if (set) {
12951                 char *p = *argptr;
12952
12953                 if (all || argptr[1])
12954                         ash_msg_and_raise_error("too many arguments");
12955                 if (strncmp(p, "unlimited\n", 9) == 0)
12956                         val = RLIM_INFINITY;
12957                 else {
12958                         if (sizeof(val) == sizeof(int))
12959                                 val = bb_strtou(p, NULL, 10);
12960                         else if (sizeof(val) == sizeof(long))
12961                                 val = bb_strtoul(p, NULL, 10);
12962                         else
12963                                 val = bb_strtoull(p, NULL, 10);
12964                         if (errno)
12965                                 ash_msg_and_raise_error("bad number");
12966                         val <<= l->factor_shift;
12967                 }
12968         }
12969         if (all) {
12970                 const char *lname = limits_name;
12971                 for (l = limits_tbl; l != &limits_tbl[ARRAY_SIZE(limits_tbl)]; l++) {
12972                         getrlimit(l->cmd, &limit);
12973                         out1fmt("%-20s ", lname);
12974                         lname += strlen(lname) + 1;
12975                         printlim(how, &limit, l);
12976                 }
12977                 return 0;
12978         }
12979
12980         getrlimit(l->cmd, &limit);
12981         if (set) {
12982                 if (how & HARD)
12983                         limit.rlim_max = val;
12984                 if (how & SOFT)
12985                         limit.rlim_cur = val;
12986                 if (setrlimit(l->cmd, &limit) < 0)
12987                         ash_msg_and_raise_error("error setting limit (%m)");
12988         } else {
12989                 printlim(how, &limit, l);
12990         }
12991         return 0;
12992 }
12993
12994 /* ============ main() and helpers */
12995
12996 /*
12997  * Called to exit the shell.
12998  */
12999 static void exitshell(void) NORETURN;
13000 static void
13001 exitshell(void)
13002 {
13003         struct jmploc loc;
13004         char *p;
13005         int status;
13006
13007         status = exitstatus;
13008         TRACE(("pid %d, exitshell(%d)\n", getpid(), status));
13009         if (setjmp(loc.loc)) {
13010                 if (exception_type == EXEXIT)
13011 /* dash bug: it just does _exit(exitstatus) here
13012  * but we have to do setjobctl(0) first!
13013  * (bug is still not fixed in dash-0.5.3 - if you run dash
13014  * under Midnight Commander, on exit from dash MC is backgrounded) */
13015                         status = exitstatus;
13016                 goto out;
13017         }
13018         exception_handler = &loc;
13019         p = trap[0];
13020         if (p) {
13021                 trap[0] = NULL;
13022                 evalstring(p, 0);
13023                 free(p);
13024         }
13025         flush_stdout_stderr();
13026  out:
13027         setjobctl(0);
13028         _exit(status);
13029         /* NOTREACHED */
13030 }
13031
13032 static void
13033 init(void)
13034 {
13035         /* from input.c: */
13036         basepf.next_to_pgetc = basepf.buf = basebuf;
13037
13038         /* from trap.c: */
13039         signal(SIGCHLD, SIG_DFL);
13040         /* bash re-enables SIGHUP which is SIG_IGNed on entry.
13041          * Try: "trap '' HUP; bash; echo RET" and type "kill -HUP $$"
13042          */
13043         signal(SIGHUP, SIG_DFL);
13044
13045         /* from var.c: */
13046         {
13047                 char **envp;
13048                 const char *p;
13049                 struct stat st1, st2;
13050
13051                 initvar();
13052                 for (envp = environ; envp && *envp; envp++) {
13053                         if (strchr(*envp, '=')) {
13054                                 setvareq(*envp, VEXPORT|VTEXTFIXED);
13055                         }
13056                 }
13057
13058                 setvar("PPID", utoa(getppid()), 0);
13059
13060                 p = lookupvar("PWD");
13061                 if (p)
13062                         if (*p != '/' || stat(p, &st1) || stat(".", &st2)
13063                          || st1.st_dev != st2.st_dev || st1.st_ino != st2.st_ino)
13064                                 p = '\0';
13065                 setpwd(p, 0);
13066         }
13067 }
13068
13069 /*
13070  * Process the shell command line arguments.
13071  */
13072 static void
13073 procargs(char **argv)
13074 {
13075         int i;
13076         const char *xminusc;
13077         char **xargv;
13078
13079         xargv = argv;
13080         arg0 = xargv[0];
13081         /* if (xargv[0]) - mmm, this is always true! */
13082                 xargv++;
13083         for (i = 0; i < NOPTS; i++)
13084                 optlist[i] = 2;
13085         argptr = xargv;
13086         if (options(1)) {
13087                 /* it already printed err message */
13088                 raise_exception(EXERROR);
13089         }
13090         xargv = argptr;
13091         xminusc = minusc;
13092         if (*xargv == NULL) {
13093                 if (xminusc)
13094                         ash_msg_and_raise_error(bb_msg_requires_arg, "-c");
13095                 sflag = 1;
13096         }
13097         if (iflag == 2 && sflag == 1 && isatty(0) && isatty(1))
13098                 iflag = 1;
13099         if (mflag == 2)
13100                 mflag = iflag;
13101         for (i = 0; i < NOPTS; i++)
13102                 if (optlist[i] == 2)
13103                         optlist[i] = 0;
13104 #if DEBUG == 2
13105         debug = 1;
13106 #endif
13107         /* POSIX 1003.2: first arg after -c cmd is $0, remainder $1... */
13108         if (xminusc) {
13109                 minusc = *xargv++;
13110                 if (*xargv)
13111                         goto setarg0;
13112         } else if (!sflag) {
13113                 setinputfile(*xargv, 0);
13114  setarg0:
13115                 arg0 = *xargv++;
13116                 commandname = arg0;
13117         }
13118
13119         shellparam.p = xargv;
13120 #if ENABLE_ASH_GETOPTS
13121         shellparam.optind = 1;
13122         shellparam.optoff = -1;
13123 #endif
13124         /* assert(shellparam.malloced == 0 && shellparam.nparam == 0); */
13125         while (*xargv) {
13126                 shellparam.nparam++;
13127                 xargv++;
13128         }
13129         optschanged();
13130 }
13131
13132 /*
13133  * Read /etc/profile or .profile.
13134  */
13135 static void
13136 read_profile(const char *name)
13137 {
13138         int skip;
13139
13140         if (setinputfile(name, INPUT_PUSH_FILE | INPUT_NOFILE_OK) < 0)
13141                 return;
13142         skip = cmdloop(0);
13143         popfile();
13144         if (skip)
13145                 exitshell();
13146 }
13147
13148 /*
13149  * This routine is called when an error or an interrupt occurs in an
13150  * interactive shell and control is returned to the main command loop.
13151  */
13152 static void
13153 reset(void)
13154 {
13155         /* from eval.c: */
13156         evalskip = 0;
13157         loopnest = 0;
13158         /* from input.c: */
13159         g_parsefile->left_in_buffer = 0;
13160         g_parsefile->left_in_line = 0;      /* clear input buffer */
13161         popallfiles();
13162         /* from parser.c: */
13163         tokpushback = 0;
13164         checkkwd = 0;
13165         /* from redir.c: */
13166         clearredir(/*drop:*/ 0);
13167 }
13168
13169 #if PROFILE
13170 static short profile_buf[16384];
13171 extern int etext();
13172 #endif
13173
13174 /*
13175  * Main routine.  We initialize things, parse the arguments, execute
13176  * profiles if we're a login shell, and then call cmdloop to execute
13177  * commands.  The setjmp call sets up the location to jump to when an
13178  * exception occurs.  When an exception occurs the variable "state"
13179  * is used to figure out how far we had gotten.
13180  */
13181 int ash_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
13182 int ash_main(int argc UNUSED_PARAM, char **argv)
13183 {
13184         const char *shinit;
13185         volatile smallint state;
13186         struct jmploc jmploc;
13187         struct stackmark smark;
13188
13189         /* Initialize global data */
13190         INIT_G_misc();
13191         INIT_G_memstack();
13192         INIT_G_var();
13193 #if ENABLE_ASH_ALIAS
13194         INIT_G_alias();
13195 #endif
13196         INIT_G_cmdtable();
13197
13198 #if PROFILE
13199         monitor(4, etext, profile_buf, sizeof(profile_buf), 50);
13200 #endif
13201
13202 #if ENABLE_FEATURE_EDITING
13203         line_input_state = new_line_input_t(FOR_SHELL | WITH_PATH_LOOKUP);
13204 #endif
13205         state = 0;
13206         if (setjmp(jmploc.loc)) {
13207                 smallint e;
13208                 smallint s;
13209
13210                 reset();
13211
13212                 e = exception_type;
13213                 if (e == EXERROR)
13214                         exitstatus = 2;
13215                 s = state;
13216                 if (e == EXEXIT || s == 0 || iflag == 0 || shlvl)
13217                         exitshell();
13218                 if (e == EXINT)
13219                         outcslow('\n', stderr);
13220
13221                 popstackmark(&smark);
13222                 FORCE_INT_ON; /* enable interrupts */
13223                 if (s == 1)
13224                         goto state1;
13225                 if (s == 2)
13226                         goto state2;
13227                 if (s == 3)
13228                         goto state3;
13229                 goto state4;
13230         }
13231         exception_handler = &jmploc;
13232 #if DEBUG
13233         opentrace();
13234         TRACE(("Shell args: "));
13235         trace_puts_args(argv);
13236 #endif
13237         rootpid = getpid();
13238
13239         init();
13240         setstackmark(&smark);
13241         procargs(argv);
13242
13243 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
13244         if (iflag) {
13245                 const char *hp = lookupvar("HISTFILE");
13246
13247                 if (hp == NULL) {
13248                         hp = lookupvar("HOME");
13249                         if (hp != NULL) {
13250                                 char *defhp = concat_path_file(hp, ".ash_history");
13251                                 setvar("HISTFILE", defhp, 0);
13252                                 free(defhp);
13253                         }
13254                 }
13255         }
13256 #endif
13257         if (/* argv[0] && */ argv[0][0] == '-')
13258                 isloginsh = 1;
13259         if (isloginsh) {
13260                 state = 1;
13261                 read_profile("/etc/profile");
13262  state1:
13263                 state = 2;
13264                 read_profile(".profile");
13265         }
13266  state2:
13267         state = 3;
13268         if (
13269 #ifndef linux
13270          getuid() == geteuid() && getgid() == getegid() &&
13271 #endif
13272          iflag
13273         ) {
13274                 shinit = lookupvar("ENV");
13275                 if (shinit != NULL && *shinit != '\0') {
13276                         read_profile(shinit);
13277                 }
13278         }
13279  state3:
13280         state = 4;
13281         if (minusc) {
13282                 /* evalstring pushes parsefile stack.
13283                  * Ensure we don't falsely claim that 0 (stdin)
13284                  * is one of stacked source fds.
13285                  * Testcase: ash -c 'exec 1>&0' must not complain. */
13286                 if (!sflag)
13287                         g_parsefile->fd = -1;
13288                 evalstring(minusc, 0);
13289         }
13290
13291         if (sflag || minusc == NULL) {
13292 #if defined MAX_HISTORY && MAX_HISTORY > 0 && ENABLE_FEATURE_EDITING_SAVEHISTORY
13293                 if (iflag) {
13294                         const char *hp = lookupvar("HISTFILE");
13295                         if (hp)
13296                                 line_input_state->hist_file = hp;
13297                 }
13298 #endif
13299  state4: /* XXX ??? - why isn't this before the "if" statement */
13300                 cmdloop(1);
13301         }
13302 #if PROFILE
13303         monitor(0);
13304 #endif
13305 #ifdef GPROF
13306         {
13307                 extern void _mcleanup(void);
13308                 _mcleanup();
13309         }
13310 #endif
13311         exitshell();
13312         /* NOTREACHED */
13313 }
13314
13315
13316 /*-
13317  * Copyright (c) 1989, 1991, 1993, 1994
13318  *      The Regents of the University of California.  All rights reserved.
13319  *
13320  * This code is derived from software contributed to Berkeley by
13321  * Kenneth Almquist.
13322  *
13323  * Redistribution and use in source and binary forms, with or without
13324  * modification, are permitted provided that the following conditions
13325  * are met:
13326  * 1. Redistributions of source code must retain the above copyright
13327  *    notice, this list of conditions and the following disclaimer.
13328  * 2. Redistributions in binary form must reproduce the above copyright
13329  *    notice, this list of conditions and the following disclaimer in the
13330  *    documentation and/or other materials provided with the distribution.
13331  * 3. Neither the name of the University nor the names of its contributors
13332  *    may be used to endorse or promote products derived from this software
13333  *    without specific prior written permission.
13334  *
13335  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
13336  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
13337  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
13338  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
13339  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
13340  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
13341  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
13342  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
13343  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
13344  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
13345  * SUCH DAMAGE.
13346  */