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