shell/ash.c

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