1 // SPDX-License-Identifier: GPL-2.0+
3 * sh.c -- a prototype Bourne shell grammar parser
4 * Intended to follow the original Thompson and Ritchie
5 * "small and simple is beautiful" philosophy, which
6 * incidentally is a good match to today's BusyBox.
8 * Copyright (C) 2000,2001 Larry Doolittle <larry@doolittle.boa.org>
11 * The parser routines proper are all original material, first
12 * written Dec 2000 and Jan 2001 by Larry Doolittle.
13 * The execution engine, the builtins, and much of the underlying
14 * support has been adapted from busybox-0.49pre's lash,
15 * which is Copyright (C) 2000 by Lineo, Inc., and
16 * written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>.
17 * That, in turn, is based in part on ladsh.c, by Michael K. Johnson and
18 * Erik W. Troan, which they placed in the public domain. I don't know
19 * how much of the Johnson/Troan code has survived the repeated rewrites.
21 * b_addchr() derived from similar w_addchar function in glibc-2.2
22 * setup_redirect(), redirect_opt_num(), and big chunks of main()
23 * and many builtins derived from contributions by Erik Andersen
24 * miscellaneous bugfixes from Matt Kraai
26 * There are two big (and related) architecture differences between
27 * this parser and the lash parser. One is that this version is
28 * actually designed from the ground up to understand nearly all
29 * of the Bourne grammar. The second, consequential change is that
30 * the parser and input reader have been turned inside out. Now,
31 * the parser is in control, and asks for input as needed. The old
32 * way had the input reader in control, and it asked for parsing to
33 * take place as needed. The new way makes it much easier to properly
34 * handle the recursion implicit in the various substitutions, especially
35 * across continuation lines.
37 * Bash grammar not implemented: (how many of these were in original sh?)
38 * $@ (those sure look like weird quoting rules)
40 * ! negation operator for pipes
41 * &> and >& redirection of stdout+stderr
44 * fancy forms of Parameter Expansion
46 * Arithmetic Expansion
47 * <(list) and >(list) Process Substitution
48 * reserved words: case, esac, select, function
49 * Here Documents ( << word )
52 * job handling woefully incomplete and buggy
53 * reserved word execution woefully incomplete and buggy
55 * port selected bugfixes from post-0.49 busybox lash - done?
56 * finish implementing reserved words: for, while, until, do, done
57 * change { and } from special chars to reserved words
58 * builtins: break, continue, eval, return, set, trap, ulimit
60 * handle children going into background
61 * clean up recognition of null pipes
62 * check setting of global_argc and global_argv
63 * control-C handling, probably with longjmp
64 * follow IFS rules more precisely, including update semantics
65 * figure out what to do with backslash-newline
66 * explain why we use signal instead of sigaction
67 * propagate syntax errors, die on resource errors?
68 * continuation lines, both explicit and implicit - done?
69 * memory leak finding and plugging - done?
70 * more testing, especially quoting rules and redirection
71 * document how quoting rules not precisely followed for variable assignments
72 * maybe change map[] to use 2-bit entries
73 * (eventually) remove all the printf's
78 #include <common.h> /* readline */
80 #include <malloc.h> /* malloc, free, realloc*/
81 #include <linux/ctype.h> /* isalpha, isdigit */
83 #include <bootretry.h>
86 #include <command.h> /* find_cmd */
87 #ifndef CONFIG_SYS_PROMPT_HUSH_PS2
88 #define CONFIG_SYS_PROMPT_HUSH_PS2 "> "
92 #include <ctype.h> /* isalpha, isdigit */
93 #include <unistd.h> /* getpid */
94 #include <stdlib.h> /* getenv, atoi */
95 #include <string.h> /* strchr */
96 #include <stdio.h> /* popen etc. */
97 #include <glob.h> /* glob, of course */
98 #include <stdarg.h> /* va_list */
101 #include <getopt.h> /* should be pretty obvious */
103 #include <sys/stat.h> /* ulimit */
104 #include <sys/types.h>
105 #include <sys/wait.h>
108 /* #include <dmalloc.h> */
114 #define applet_name "hush"
115 #include "standalone.h"
116 #define hush_main main
117 #undef CONFIG_FEATURE_SH_FANCY_PROMPT
121 #define SPECIAL_VAR_SYMBOL 03
122 #define SUBSTED_VAR_SYMBOL 04
124 #define FLAG_EXIT_FROM_LOOP 1
125 #define FLAG_PARSE_SEMICOLON (1 << 1) /* symbol ';' is special for parser */
126 #define FLAG_REPARSING (1 << 2) /* >= 2nd pass */
131 DECLARE_GLOBAL_DATA_PTR
;
133 #define EXIT_SUCCESS 0
135 #define syntax() syntax_err()
136 #define xstrdup strdup
137 #define error_msg printf
141 REDIRECT_OVERWRITE
= 2,
147 /* The descrip member of this structure is only used to make debugging
149 struct {int mode
; int default_fd
; char *descrip
;} redir_table
[] = {
151 { O_RDONLY
, 0, "<" },
152 { O_CREAT
|O_TRUNC
|O_WRONLY
, 1, ">" },
153 { O_CREAT
|O_APPEND
|O_WRONLY
, 1, ">>" },
154 { O_RDONLY
, -1, "<<" },
166 /* might eventually control execution */
183 #define FLAG_END (1<<RES_NONE)
184 #define FLAG_IF (1<<RES_IF)
185 #define FLAG_THEN (1<<RES_THEN)
186 #define FLAG_ELIF (1<<RES_ELIF)
187 #define FLAG_ELSE (1<<RES_ELSE)
188 #define FLAG_FI (1<<RES_FI)
189 #define FLAG_FOR (1<<RES_FOR)
190 #define FLAG_WHILE (1<<RES_WHILE)
191 #define FLAG_UNTIL (1<<RES_UNTIL)
192 #define FLAG_DO (1<<RES_DO)
193 #define FLAG_DONE (1<<RES_DONE)
194 #define FLAG_IN (1<<RES_IN)
195 #define FLAG_START (1<<RES_XXXX)
197 /* This holds pointers to the various results of parsing */
199 struct child_prog
*child
;
200 struct pipe
*list_head
;
203 struct redir_struct
*pending_redirect
;
206 int old_flag
; /* for figuring out valid reserved words */
207 struct p_context
*stack
;
208 int type
; /* define type of parser : ";$" common or special symbol */
209 /* How about quoting status? */
213 struct redir_struct
{
214 redir_type type
; /* type of redirection */
215 int fd
; /* file descriptor being redirected */
216 int dup
; /* -1, or file descriptor being duplicated */
217 struct redir_struct
*next
; /* pointer to the next redirect in the list */
218 glob_t word
; /* *word.gl_pathv is the filename */
224 pid_t pid
; /* 0 if exited */
226 char **argv
; /* program name and arguments */
227 /* was quoted when parsed; copy of struct o_string.nonnull field */
230 int argc
; /* number of program arguments */
232 struct pipe
*group
; /* if non-NULL, first in group or subshell */
234 int subshell
; /* flag, non-zero if group must be forked */
235 struct redir_struct
*redirects
; /* I/O redirections */
236 glob_t glob_result
; /* result of parameter globbing */
237 int is_stopped
; /* is the program currently running? */
238 struct pipe
*family
; /* pointer back to the child's parent pipe */
240 int sp
; /* number of SPECIAL_VAR_SYMBOL */
246 int jobid
; /* job number */
248 int num_progs
; /* total number of programs in job */
250 int running_progs
; /* number of programs running */
251 char *text
; /* name of job */
252 char *cmdbuf
; /* buffer various argv's point into */
253 pid_t pgrp
; /* process group ID for the job */
255 struct child_prog
*progs
; /* array of commands in pipe */
256 struct pipe
*next
; /* to track background commands */
258 int stopped_progs
; /* number of programs alive, but stopped */
259 int job_context
; /* bitmask defining current context */
261 pipe_style followup
; /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
262 reserved_style r_mode
; /* supports if, for, while, until */
268 struct close_me
*next
;
277 struct variables
*next
;
280 /* globals, connect us to the outside world
281 * the first three support $?, $#, and $1 */
284 unsigned int global_argc
;
286 static unsigned int last_return_code
;
288 extern char **environ
; /* This is in <unistd.h>, but protected with __USE_GNU */
291 /* "globals" within this file */
293 static char map
[256];
295 static int fake_mode
;
296 static int interactive
;
297 static struct close_me
*close_me_head
;
298 static const char *cwd
;
299 static struct pipe
*job_list
;
300 static unsigned int last_bg_pid
;
301 static unsigned int last_jobid
;
302 static unsigned int shell_terminal
;
305 struct variables shell_ver
= { "HUSH_VERSION", "0.01", 1, 1, 0 };
306 struct variables
*top_vars
= &shell_ver
;
308 static int flag_repeat
= 0;
309 static int do_repeat
= 0;
310 static struct variables
*top_vars
= NULL
;
311 #endif /*__U_BOOT__ */
313 #define B_CHUNK (100)
323 #define NULL_O_STRING {NULL,0,0,0,0}
324 /* used for initialization:
325 o_string foo = NULL_O_STRING; */
327 /* I can almost use ordinary FILE *. Is open_memstream() universally
328 * available? Where is it documented? */
339 int (*get
) (struct in_str
*);
340 int (*peek
) (struct in_str
*);
342 #define b_getch(input) ((input)->get(input))
343 #define b_peek(input) ((input)->peek(input))
346 #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
348 struct built_in_command
{
349 char *cmd
; /* name */
350 char *descr
; /* description */
351 int (*function
) (struct child_prog
*); /* function ptr */
355 /* define DEBUG_SHELL for debugging output (obviously ;-)) */
360 /* This should be in utility.c */
363 static void debug_printf(const char *format
, ...)
366 va_start(args
, format
);
367 vfprintf(stderr
, format
, args
);
371 #define debug_printf(fmt,args...) printf (fmt ,##args)
374 static inline void debug_printf(const char *format
, ...) { }
376 #define final_printf debug_printf
379 static void syntax_err(void) {
380 printf("syntax error\n");
383 static void __syntax(char *file
, int line
) {
384 error_msg("syntax error %s:%d", file
, line
);
386 #define syntax() __syntax(__FILE__, __LINE__)
390 static void *xmalloc(size_t size
);
391 static void *xrealloc(void *ptr
, size_t size
);
393 /* Index of subroutines: */
394 /* function prototypes for builtins */
395 static int builtin_cd(struct child_prog
*child
);
396 static int builtin_env(struct child_prog
*child
);
397 static int builtin_eval(struct child_prog
*child
);
398 static int builtin_exec(struct child_prog
*child
);
399 static int builtin_exit(struct child_prog
*child
);
400 static int builtin_export(struct child_prog
*child
);
401 static int builtin_fg_bg(struct child_prog
*child
);
402 static int builtin_help(struct child_prog
*child
);
403 static int builtin_jobs(struct child_prog
*child
);
404 static int builtin_pwd(struct child_prog
*child
);
405 static int builtin_read(struct child_prog
*child
);
406 static int builtin_set(struct child_prog
*child
);
407 static int builtin_shift(struct child_prog
*child
);
408 static int builtin_source(struct child_prog
*child
);
409 static int builtin_umask(struct child_prog
*child
);
410 static int builtin_unset(struct child_prog
*child
);
411 static int builtin_not_written(struct child_prog
*child
);
413 /* o_string manipulation: */
414 static int b_check_space(o_string
*o
, int len
);
415 static int b_addchr(o_string
*o
, int ch
);
416 static void b_reset(o_string
*o
);
417 static int b_addqchr(o_string
*o
, int ch
, int quote
);
419 static int b_adduint(o_string
*o
, unsigned int i
);
421 /* in_str manipulations: */
422 static int static_get(struct in_str
*i
);
423 static int static_peek(struct in_str
*i
);
424 static int file_get(struct in_str
*i
);
425 static int file_peek(struct in_str
*i
);
427 static void setup_file_in_str(struct in_str
*i
, FILE *f
);
429 static void setup_file_in_str(struct in_str
*i
);
431 static void setup_string_in_str(struct in_str
*i
, const char *s
);
433 /* close_me manipulations: */
434 static void mark_open(int fd
);
435 static void mark_closed(int fd
);
436 static void close_all(void);
438 /* "run" the final data structures: */
439 static char *indenter(int i
);
440 static int free_pipe_list(struct pipe
*head
, int indent
);
441 static int free_pipe(struct pipe
*pi
, int indent
);
442 /* really run the final data structures: */
444 static int setup_redirects(struct child_prog
*prog
, int squirrel
[]);
446 static int run_list_real(struct pipe
*pi
);
448 static void pseudo_exec(struct child_prog
*child
) __attribute__ ((noreturn
));
450 static int run_pipe_real(struct pipe
*pi
);
451 /* extended glob support: */
453 static int globhack(const char *src
, int flags
, glob_t
*pglob
);
454 static int glob_needed(const char *s
);
455 static int xglob(o_string
*dest
, int flags
, glob_t
*pglob
);
457 /* variable assignment: */
458 static int is_assignment(const char *s
);
459 /* data structure manipulation: */
461 static int setup_redirect(struct p_context
*ctx
, int fd
, redir_type style
, struct in_str
*input
);
463 static void initialize_context(struct p_context
*ctx
);
464 static int done_word(o_string
*dest
, struct p_context
*ctx
);
465 static int done_command(struct p_context
*ctx
);
466 static int done_pipe(struct p_context
*ctx
, pipe_style type
);
467 /* primary string parsing: */
469 static int redirect_dup_num(struct in_str
*input
);
470 static int redirect_opt_num(o_string
*o
);
471 static int process_command_subs(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input
, int subst_end
);
472 static int parse_group(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input
, int ch
);
474 static char *lookup_param(char *src
);
475 static char *make_string(char **inp
, int *nonnull
);
476 static int handle_dollar(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input
);
478 static int parse_string(o_string
*dest
, struct p_context
*ctx
, const char *src
);
480 static int parse_stream(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input0
, int end_trigger
);
482 static int parse_stream_outer(struct in_str
*inp
, int flag
);
484 static int parse_string_outer(const char *s
, int flag
);
485 static int parse_file_outer(FILE *f
);
488 /* job management: */
489 static int checkjobs(struct pipe
* fg_pipe
);
490 static void insert_bg_job(struct pipe
*pi
);
491 static void remove_bg_job(struct pipe
*pi
);
493 /* local variable support */
494 static char **make_list_in(char **inp
, char *name
);
495 static char *insert_var_value(char *inp
);
496 static char *insert_var_value_sub(char *inp
, int tag_subst
);
499 /* Table of built-in functions. They can be forked or not, depending on
500 * context: within pipes, they fork. As simple commands, they do not.
501 * When used in non-forking context, they can change global variables
502 * in the parent shell process. If forked, of course they can not.
503 * For example, 'unset foo | whatever' will parse and run, but foo will
504 * still be set at the end. */
505 static struct built_in_command bltins
[] = {
506 {"bg", "Resume a job in the background", builtin_fg_bg
},
507 {"break", "Exit for, while or until loop", builtin_not_written
},
508 {"cd", "Change working directory", builtin_cd
},
509 {"continue", "Continue for, while or until loop", builtin_not_written
},
510 {"env", "Print all environment variables", builtin_env
},
511 {"eval", "Construct and run shell command", builtin_eval
},
512 {"exec", "Exec command, replacing this shell with the exec'd process",
514 {"exit", "Exit from shell()", builtin_exit
},
515 {"export", "Set environment variable", builtin_export
},
516 {"fg", "Bring job into the foreground", builtin_fg_bg
},
517 {"jobs", "Lists the active jobs", builtin_jobs
},
518 {"pwd", "Print current directory", builtin_pwd
},
519 {"read", "Input environment variable", builtin_read
},
520 {"return", "Return from a function", builtin_not_written
},
521 {"set", "Set/unset shell local variables", builtin_set
},
522 {"shift", "Shift positional parameters", builtin_shift
},
523 {"trap", "Trap signals", builtin_not_written
},
524 {"ulimit","Controls resource limits", builtin_not_written
},
525 {"umask","Sets file creation mask", builtin_umask
},
526 {"unset", "Unset environment variable", builtin_unset
},
527 {".", "Source-in and run commands in a file", builtin_source
},
528 {"help", "List shell built-in commands", builtin_help
},
532 static const char *set_cwd(void)
535 cwd
= NULL
; /* xgetcwd(arg) called free(arg) */
536 cwd
= xgetcwd((char *)cwd
);
542 /* built-in 'eval' handler */
543 static int builtin_eval(struct child_prog
*child
)
546 int rcode
= EXIT_SUCCESS
;
548 if (child
->argv
[1]) {
549 str
= make_string(child
->argv
+ 1);
550 parse_string_outer(str
, FLAG_EXIT_FROM_LOOP
|
551 FLAG_PARSE_SEMICOLON
);
553 rcode
= last_return_code
;
558 /* built-in 'cd <path>' handler */
559 static int builtin_cd(struct child_prog
*child
)
562 if (child
->argv
[1] == NULL
)
563 newdir
= env_get("HOME");
565 newdir
= child
->argv
[1];
567 printf("cd: %s: %s\n", newdir
, strerror(errno
));
574 /* built-in 'env' handler */
575 static int builtin_env(struct child_prog
*dummy
)
578 if (e
== NULL
) return EXIT_FAILURE
;
585 /* built-in 'exec' handler */
586 static int builtin_exec(struct child_prog
*child
)
588 if (child
->argv
[1] == NULL
)
589 return EXIT_SUCCESS
; /* Really? */
595 /* built-in 'exit' handler */
596 static int builtin_exit(struct child_prog
*child
)
598 if (child
->argv
[1] == NULL
)
599 exit(last_return_code
);
600 exit (atoi(child
->argv
[1]));
603 /* built-in 'export VAR=value' handler */
604 static int builtin_export(struct child_prog
*child
)
607 char *name
= child
->argv
[1];
610 return (builtin_env(child
));
616 char *value
= strchr(name
, '=');
620 /* They are exporting something without an =VALUE */
622 value
= get_local_var(name
);
624 size_t ln
= strlen(name
);
626 tmp
= realloc(name
, ln
+strlen(value
)+2);
630 sprintf(tmp
+ln
, "=%s", value
);
634 /* bash does not return an error when trying to export
635 * an undefined variable. Do likewise. */
641 perror_msg("export");
643 res
= set_local_var(name
, 1);
650 /* built-in 'fg' and 'bg' handler */
651 static int builtin_fg_bg(struct child_prog
*child
)
654 struct pipe
*pi
=NULL
;
658 /* If they gave us no args, assume they want the last backgrounded task */
659 if (!child
->argv
[1]) {
660 for (pi
= job_list
; pi
; pi
= pi
->next
) {
661 if (pi
->jobid
== last_jobid
) {
666 error_msg("%s: no current job", child
->argv
[0]);
670 if (sscanf(child
->argv
[1], "%%%d", &jobnum
) != 1) {
671 error_msg("%s: bad argument '%s'", child
->argv
[0], child
->argv
[1]);
674 for (pi
= job_list
; pi
; pi
= pi
->next
) {
675 if (pi
->jobid
== jobnum
) {
680 error_msg("%s: %d: no such job", child
->argv
[0], jobnum
);
685 if (*child
->argv
[0] == 'f') {
686 /* Put the job into the foreground. */
687 tcsetpgrp(shell_terminal
, pi
->pgrp
);
690 /* Restart the processes in the job */
691 for (i
= 0; i
< pi
->num_progs
; i
++)
692 pi
->progs
[i
].is_stopped
= 0;
694 if ( (i
=kill(- pi
->pgrp
, SIGCONT
)) < 0) {
698 perror_msg("kill (SIGCONT)");
702 pi
->stopped_progs
= 0;
706 /* built-in 'help' handler */
707 static int builtin_help(struct child_prog
*dummy
)
709 struct built_in_command
*x
;
711 printf("\nBuilt-in commands:\n");
712 printf("-------------------\n");
713 for (x
= bltins
; x
->cmd
; x
++) {
716 printf("%s\t%s\n", x
->cmd
, x
->descr
);
722 /* built-in 'jobs' handler */
723 static int builtin_jobs(struct child_prog
*child
)
728 for (job
= job_list
; job
; job
= job
->next
) {
729 if (job
->running_progs
== job
->stopped_progs
)
730 status_string
= "Stopped";
732 status_string
= "Running";
734 printf(JOB_STATUS_FORMAT
, job
->jobid
, status_string
, job
->text
);
740 /* built-in 'pwd' handler */
741 static int builtin_pwd(struct child_prog
*dummy
)
747 /* built-in 'read VAR' handler */
748 static int builtin_read(struct child_prog
*child
)
752 if (child
->argv
[1]) {
756 string
[0] = 0; /* In case stdin has only EOF */
758 fgets(string
, sizeof(string
), stdin
);
760 var
= malloc(strlen(child
->argv
[1])+strlen(string
)+2);
762 sprintf(var
, "%s=%s", child
->argv
[1], string
);
763 res
= set_local_var(var
, 0);
767 fprintf(stderr
, "read: %m\n");
768 free(var
); /* So not move up to avoid breaking errno */
771 do res
=getchar(); while(res
!='\n' && res
!=EOF
);
776 /* built-in 'set VAR=value' handler */
777 static int builtin_set(struct child_prog
*child
)
779 char *temp
= child
->argv
[1];
783 for(e
= top_vars
; e
; e
=e
->next
)
784 printf("%s=%s\n", e
->name
, e
->value
);
786 set_local_var(temp
, 0);
792 /* Built-in 'shift' handler */
793 static int builtin_shift(struct child_prog
*child
)
796 if (child
->argv
[1]) {
797 n
=atoi(child
->argv
[1]);
799 if (n
>=0 && n
<global_argc
) {
800 /* XXX This probably breaks $0 */
809 /* Built-in '.' handler (read-in and execute commands from file) */
810 static int builtin_source(struct child_prog
*child
)
815 if (child
->argv
[1] == NULL
)
818 /* XXX search through $PATH is missing */
819 input
= fopen(child
->argv
[1], "r");
821 error_msg("Couldn't open file '%s'", child
->argv
[1]);
825 /* Now run the file */
826 /* XXX argv and argc are broken; need to save old global_argv
827 * (pointer only is OK!) on this stack frame,
828 * set global_argv=child->argv+1, recurse, and restore. */
829 mark_open(fileno(input
));
830 status
= parse_file_outer(input
);
831 mark_closed(fileno(input
));
836 static int builtin_umask(struct child_prog
*child
)
839 const char *arg
= child
->argv
[1];
842 new_umask
=strtoul(arg
, &end
, 8);
843 if (*end
!='\0' || end
== arg
) {
847 printf("%.3o\n", (unsigned int) (new_umask
=umask(0)));
853 /* built-in 'unset VAR' handler */
854 static int builtin_unset(struct child_prog
*child
)
856 /* bash returned already true */
857 unset_local_var(child
->argv
[1]);
861 static int builtin_not_written(struct child_prog
*child
)
863 printf("builtin_%s not written\n",child
->argv
[0]);
868 static int b_check_space(o_string
*o
, int len
)
870 /* It would be easy to drop a more restrictive policy
871 * in here, such as setting a maximum string length */
872 if (o
->length
+ len
> o
->maxlen
) {
873 char *old_data
= o
->data
;
874 /* assert (data == NULL || o->maxlen != 0); */
875 o
->maxlen
+= max(2*len
, B_CHUNK
);
876 o
->data
= realloc(o
->data
, 1 + o
->maxlen
);
877 if (o
->data
== NULL
) {
881 return o
->data
== NULL
;
884 static int b_addchr(o_string
*o
, int ch
)
886 debug_printf("b_addchr: %c %d %p\n", ch
, o
->length
, o
);
887 if (b_check_space(o
, 1)) return B_NOSPAC
;
888 o
->data
[o
->length
] = ch
;
890 o
->data
[o
->length
] = '\0';
894 static void b_reset(o_string
*o
)
898 if (o
->data
!= NULL
) *o
->data
= '\0';
901 static void b_free(o_string
*o
)
909 /* My analysis of quoting semantics tells me that state information
910 * is associated with a destination, not a source.
912 static int b_addqchr(o_string
*o
, int ch
, int quote
)
914 if (quote
&& strchr("*?[\\",ch
)) {
916 rc
= b_addchr(o
, '\\');
919 return b_addchr(o
, ch
);
923 static int b_adduint(o_string
*o
, unsigned int i
)
926 char *p
= simple_itoa(i
);
927 /* no escape checking necessary */
928 do r
=b_addchr(o
, *p
++); while (r
==0 && *p
);
933 static int static_get(struct in_str
*i
)
936 if (ch
=='\0') return EOF
;
940 static int static_peek(struct in_str
*i
)
946 static inline void cmdedit_set_initial_prompt(void)
948 #ifndef CONFIG_FEATURE_SH_FANCY_PROMPT
951 PS1
= env_get("PS1");
957 static inline void setup_prompt_string(int promptmode
, char **prompt_str
)
959 debug_printf("setup_prompt_string %d ",promptmode
);
960 #ifndef CONFIG_FEATURE_SH_FANCY_PROMPT
961 /* Set up the prompt */
962 if (promptmode
== 1) {
964 PS1
=xmalloc(strlen(cwd
)+4);
965 sprintf(PS1
, "%s %s", cwd
, ( geteuid() != 0 ) ? "$ ":"# ");
971 *prompt_str
= (promptmode
==1)? PS1
: PS2
;
973 debug_printf("result %s\n",*prompt_str
);
978 static int uboot_cli_readline(struct in_str
*i
)
981 char __maybe_unused
*ps_prompt
= NULL
;
983 if (i
->promptmode
== 1)
984 prompt
= CONFIG_SYS_PROMPT
;
986 prompt
= CONFIG_SYS_PROMPT_HUSH_PS2
;
988 #ifdef CONFIG_CMDLINE_PS_SUPPORT
989 if (i
->promptmode
== 1)
990 ps_prompt
= env_get("PS1");
992 ps_prompt
= env_get("PS2");
997 return cli_readline(prompt
);
1001 static void get_user_input(struct in_str
*i
)
1005 static char the_command
[BUFSIZ
];
1007 setup_prompt_string(i
->promptmode
, &prompt_str
);
1008 #ifdef CONFIG_FEATURE_COMMAND_EDITING
1010 ** enable command line editing only while a command line
1011 ** is actually being read; otherwise, we'll end up bequeathing
1012 ** atexit() handlers and other unwanted stuff to our
1013 ** child processes (rob@sysgo.de)
1015 cmdedit_read_input(prompt_str
, the_command
);
1017 fputs(prompt_str
, stdout
);
1019 the_command
[0]=fgetc(i
->file
);
1020 the_command
[1]='\0';
1026 static char the_command
[CONFIG_SYS_CBSIZE
+ 1];
1028 bootretry_reset_cmd_timeout();
1030 n
= uboot_cli_readline(i
);
1032 #ifdef CONFIG_BOOT_RETRY_TIME
1034 puts("\nTimeout waiting for command\n");
1035 # ifdef CONFIG_RESET_TO_RETRY
1036 do_reset(NULL
, 0, 0, NULL
);
1038 # error "This currently only works with CONFIG_RESET_TO_RETRY enabled"
1046 n
= strlen(console_buffer
);
1047 console_buffer
[n
] = '\n';
1048 console_buffer
[n
+1]= '\0';
1049 if (had_ctrlc()) flag_repeat
= 0;
1052 if (i
->promptmode
== 1) {
1053 if (console_buffer
[0] == '\n'&& flag_repeat
== 0) {
1054 strcpy(the_command
,console_buffer
);
1057 if (console_buffer
[0] != '\n') {
1058 strcpy(the_command
,console_buffer
);
1068 if (console_buffer
[0] != '\n') {
1069 if (strlen(the_command
) + strlen(console_buffer
)
1070 < CONFIG_SYS_CBSIZE
) {
1071 n
= strlen(the_command
);
1072 the_command
[n
-1] = ' ';
1073 strcpy(&the_command
[n
],console_buffer
);
1076 the_command
[0] = '\n';
1077 the_command
[1] = '\0';
1081 if (i
->__promptme
== 0) {
1082 the_command
[0] = '\n';
1083 the_command
[1] = '\0';
1085 i
->p
= console_buffer
;
1090 /* This is the magic location that prints prompts
1091 * and gets data back from the user */
1092 static int file_get(struct in_str
*i
)
1097 /* If there is data waiting, eat it up */
1098 if (i
->p
&& *i
->p
) {
1101 /* need to double check i->file because we might be doing something
1102 * more complicated by now, like sourcing or substituting. */
1104 if (i
->__promptme
&& interactive
&& i
->file
== stdin
) {
1105 while(! i
->p
|| (interactive
&& strlen(i
->p
)==0) ) {
1107 while(! i
->p
|| strlen(i
->p
)==0 ) {
1115 if (i
->p
&& *i
->p
) {
1120 ch
= fgetc(i
->file
);
1124 debug_printf("b_getch: got a %d\n", ch
);
1127 if (ch
== '\n') i
->__promptme
=1;
1132 /* All the callers guarantee this routine will never be
1133 * used right after a newline, so prompting is not needed.
1135 static int file_peek(struct in_str
*i
)
1138 if (i
->p
&& *i
->p
) {
1143 i
->peek_buf
[0] = fgetc(i
->file
);
1144 i
->peek_buf
[1] = '\0';
1146 debug_printf("b_peek: got a %d\n", *i
->p
);
1153 static void setup_file_in_str(struct in_str
*i
, FILE *f
)
1155 static void setup_file_in_str(struct in_str
*i
)
1158 i
->peek
= file_peek
;
1168 static void setup_string_in_str(struct in_str
*i
, const char *s
)
1170 i
->peek
= static_peek
;
1171 i
->get
= static_get
;
1178 static void mark_open(int fd
)
1180 struct close_me
*new = xmalloc(sizeof(struct close_me
));
1182 new->next
= close_me_head
;
1183 close_me_head
= new;
1186 static void mark_closed(int fd
)
1188 struct close_me
*tmp
;
1189 if (close_me_head
== NULL
|| close_me_head
->fd
!= fd
)
1190 error_msg_and_die("corrupt close_me");
1191 tmp
= close_me_head
;
1192 close_me_head
= close_me_head
->next
;
1196 static void close_all(void)
1199 for (c
=close_me_head
; c
; c
=c
->next
) {
1202 close_me_head
= NULL
;
1205 /* squirrel != NULL means we squirrel away copies of stdin, stdout,
1206 * and stderr if they are redirected. */
1207 static int setup_redirects(struct child_prog
*prog
, int squirrel
[])
1210 struct redir_struct
*redir
;
1212 for (redir
=prog
->redirects
; redir
; redir
=redir
->next
) {
1213 if (redir
->dup
== -1 && redir
->word
.gl_pathv
== NULL
) {
1214 /* something went wrong in the parse. Pretend it didn't happen */
1217 if (redir
->dup
== -1) {
1218 mode
=redir_table
[redir
->type
].mode
;
1219 openfd
= open(redir
->word
.gl_pathv
[0], mode
, 0666);
1221 /* this could get lost if stderr has been redirected, but
1222 bash and ash both lose it as well (though zsh doesn't!) */
1223 perror_msg("error opening %s", redir
->word
.gl_pathv
[0]);
1227 openfd
= redir
->dup
;
1230 if (openfd
!= redir
->fd
) {
1231 if (squirrel
&& redir
->fd
< 3) {
1232 squirrel
[redir
->fd
] = dup(redir
->fd
);
1237 dup2(openfd
, redir
->fd
);
1238 if (redir
->dup
== -1)
1246 static void restore_redirects(int squirrel
[])
1249 for (i
=0; i
<3; i
++) {
1252 /* No error checking. I sure wouldn't know what
1253 * to do with an error if I found one! */
1261 /* XXX no exit() here. If you don't exec, use _exit instead.
1262 * The at_exit handlers apparently confuse the calling process,
1263 * in particular stdin handling. Not sure why? */
1264 static void pseudo_exec(struct child_prog
*child
)
1268 struct built_in_command
*x
;
1270 for (i
=0; is_assignment(child
->argv
[i
]); i
++) {
1271 debug_printf("pid %d environment modification: %s\n",getpid(),child
->argv
[i
]);
1272 p
= insert_var_value(child
->argv
[i
]);
1274 if (p
!= child
->argv
[i
]) free(p
);
1276 child
->argv
+=i
; /* XXX this hack isn't so horrible, since we are about
1277 to exit, and therefore don't need to keep data
1278 structures consistent for free() use. */
1279 /* If a variable is assigned in a forest, and nobody listens,
1280 * was it ever really set?
1282 if (child
->argv
[0] == NULL
) {
1283 _exit(EXIT_SUCCESS
);
1287 * Check if the command matches any of the builtins.
1288 * Depending on context, this might be redundant. But it's
1289 * easier to waste a few CPU cycles than it is to figure out
1290 * if this is one of those cases.
1292 for (x
= bltins
; x
->cmd
; x
++) {
1293 if (strcmp(child
->argv
[0], x
->cmd
) == 0 ) {
1294 debug_printf("builtin exec %s\n", child
->argv
[0]);
1295 rcode
= x
->function(child
);
1301 /* Check if the command matches any busybox internal commands
1303 * FIXME: This feature is not 100% safe, since
1304 * BusyBox is not fully reentrant, so we have no guarantee the things
1305 * from the .bss are still zeroed, or that things from .data are still
1306 * at their defaults. We could exec ourself from /proc/self/exe, but I
1307 * really dislike relying on /proc for things. We could exec ourself
1308 * from global_argv[0], but if we are in a chroot, we may not be able
1309 * to find ourself... */
1310 #ifdef CONFIG_FEATURE_SH_STANDALONE_SHELL
1313 char** argv_l
=child
->argv
;
1314 char *name
= child
->argv
[0];
1316 #ifdef CONFIG_FEATURE_SH_APPLETS_ALWAYS_WIN
1317 /* Following discussions from November 2000 on the busybox mailing
1318 * list, the default configuration, (without
1319 * get_last_path_component()) lets the user force use of an
1320 * external command by specifying the full (with slashes) filename.
1321 * If you enable CONFIG_FEATURE_SH_APPLETS_ALWAYS_WIN then applets
1322 * _aways_ override external commands, so if you want to run
1323 * /bin/cat, it will use BusyBox cat even if /bin/cat exists on the
1324 * filesystem and is _not_ busybox. Some systems may want this,
1326 name
= get_last_path_component(name
);
1328 /* Count argc for use in a second... */
1329 for(argc_l
=0;*argv_l
!=NULL
; argv_l
++, argc_l
++);
1331 debug_printf("running applet %s\n", name
);
1332 run_applet_by_name(name
, argc_l
, child
->argv
);
1335 debug_printf("exec of %s\n",child
->argv
[0]);
1336 execvp(child
->argv
[0],child
->argv
);
1337 perror_msg("couldn't exec: %s",child
->argv
[0]);
1339 } else if (child
->group
) {
1340 debug_printf("runtime nesting to group\n");
1341 interactive
=0; /* crucial!!!! */
1342 rcode
= run_list_real(child
->group
);
1343 /* OK to leak memory by not calling free_pipe_list,
1344 * since this process is about to exit */
1347 /* Can happen. See what bash does with ">foo" by itself. */
1348 debug_printf("trying to pseudo_exec null command\n");
1349 _exit(EXIT_SUCCESS
);
1353 static void insert_bg_job(struct pipe
*pi
)
1355 struct pipe
*thejob
;
1357 /* Linear search for the ID of the job to use */
1359 for (thejob
= job_list
; thejob
; thejob
= thejob
->next
)
1360 if (thejob
->jobid
>= pi
->jobid
)
1361 pi
->jobid
= thejob
->jobid
+ 1;
1363 /* add thejob to the list of running jobs */
1365 thejob
= job_list
= xmalloc(sizeof(*thejob
));
1367 for (thejob
= job_list
; thejob
->next
; thejob
= thejob
->next
) /* nothing */;
1368 thejob
->next
= xmalloc(sizeof(*thejob
));
1369 thejob
= thejob
->next
;
1372 /* physically copy the struct job */
1373 memcpy(thejob
, pi
, sizeof(struct pipe
));
1374 thejob
->next
= NULL
;
1375 thejob
->running_progs
= thejob
->num_progs
;
1376 thejob
->stopped_progs
= 0;
1377 thejob
->text
= xmalloc(BUFSIZ
); /* cmdedit buffer size */
1379 /*if (pi->progs[0] && pi->progs[0].argv && pi->progs[0].argv[0]) */
1381 char *bar
=thejob
->text
;
1382 char **foo
=pi
->progs
[0].argv
;
1383 while(foo
&& *foo
) {
1384 bar
+= sprintf(bar
, "%s ", *foo
++);
1388 /* we don't wait for background thejobs to return -- append it
1389 to the list of backgrounded thejobs and leave it alone */
1390 printf("[%d] %d\n", thejob
->jobid
, thejob
->progs
[0].pid
);
1391 last_bg_pid
= thejob
->progs
[0].pid
;
1392 last_jobid
= thejob
->jobid
;
1395 /* remove a backgrounded job */
1396 static void remove_bg_job(struct pipe
*pi
)
1398 struct pipe
*prev_pipe
;
1400 if (pi
== job_list
) {
1401 job_list
= pi
->next
;
1403 prev_pipe
= job_list
;
1404 while (prev_pipe
->next
!= pi
)
1405 prev_pipe
= prev_pipe
->next
;
1406 prev_pipe
->next
= pi
->next
;
1409 last_jobid
= job_list
->jobid
;
1413 pi
->stopped_progs
= 0;
1418 /* Checks to see if any processes have exited -- if they
1419 have, figure out why and see if a job has completed */
1420 static int checkjobs(struct pipe
* fg_pipe
)
1428 attributes
= WUNTRACED
;
1429 if (fg_pipe
==NULL
) {
1430 attributes
|= WNOHANG
;
1433 while ((childpid
= waitpid(-1, &status
, attributes
)) > 0) {
1436 for (i
=0; i
< fg_pipe
->num_progs
; i
++) {
1437 if (fg_pipe
->progs
[i
].pid
== childpid
) {
1438 if (i
==fg_pipe
->num_progs
-1)
1439 rcode
=WEXITSTATUS(status
);
1440 (fg_pipe
->num_progs
)--;
1446 for (pi
= job_list
; pi
; pi
= pi
->next
) {
1448 while (prognum
< pi
->num_progs
&& pi
->progs
[prognum
].pid
!= childpid
) {
1451 if (prognum
< pi
->num_progs
)
1456 debug_printf("checkjobs: pid %d was not in our list!\n", childpid
);
1460 if (WIFEXITED(status
) || WIFSIGNALED(status
)) {
1462 pi
->running_progs
--;
1463 pi
->progs
[prognum
].pid
= 0;
1465 if (!pi
->running_progs
) {
1466 printf(JOB_STATUS_FORMAT
, pi
->jobid
, "Done", pi
->text
);
1471 pi
->stopped_progs
++;
1472 pi
->progs
[prognum
].is_stopped
= 1;
1475 /* Printing this stuff is a pain, since it tends to
1476 * overwrite the prompt an inconveinient moments. So
1478 if (pi
->stopped_progs
== pi
->num_progs
) {
1479 printf("\n"JOB_STATUS_FORMAT
, pi
->jobid
, "Stopped", pi
->text
);
1485 if (childpid
== -1 && errno
!= ECHILD
)
1486 perror_msg("waitpid");
1488 /* move the shell to the foreground */
1489 /*if (interactive && tcsetpgrp(shell_terminal, getpgid(0))) */
1490 /* perror_msg("tcsetpgrp-2"); */
1494 /* Figure out our controlling tty, checking in order stderr,
1495 * stdin, and stdout. If check_pgrp is set, also check that
1496 * we belong to the foreground process group associated with
1497 * that tty. The value of shell_terminal is needed in order to call
1498 * tcsetpgrp(shell_terminal, ...); */
1499 void controlling_tty(int check_pgrp
)
1503 if ((curpgrp
= tcgetpgrp(shell_terminal
= 2)) < 0
1504 && (curpgrp
= tcgetpgrp(shell_terminal
= 0)) < 0
1505 && (curpgrp
= tcgetpgrp(shell_terminal
= 1)) < 0)
1506 goto shell_terminal_error
;
1508 if (check_pgrp
&& curpgrp
!= getpgid(0))
1509 goto shell_terminal_error
;
1513 shell_terminal_error
:
1514 shell_terminal
= -1;
1519 /* run_pipe_real() starts all the jobs, but doesn't wait for anything
1520 * to finish. See checkjobs().
1522 * return code is normally -1, when the caller has to wait for children
1523 * to finish to determine the exit status of the pipe. If the pipe
1524 * is a simple builtin command, however, the action is done by the
1525 * time run_pipe_real returns, and the exit code is provided as the
1528 * The input of the pipe is always stdin, the output is always
1529 * stdout. The outpipe[] mechanism in BusyBox-0.48 lash is bogus,
1530 * because it tries to avoid running the command substitution in
1531 * subshell, when that is in fact necessary. The subshell process
1532 * now has its stdout directed to the input of the appropriate pipe,
1533 * so this routine is noticeably simpler.
1535 static int run_pipe_real(struct pipe
*pi
)
1539 int nextin
, nextout
;
1540 int pipefds
[2]; /* pipefds[0] is for reading */
1541 struct child_prog
*child
;
1542 struct built_in_command
*x
;
1545 /* Avoid longjmp clobbering */
1553 int flag
= do_repeat
? CMD_FLAG_REPEAT
: 0;
1554 struct child_prog
*child
;
1557 /* Avoid longjmp clobbering */
1562 #endif /* __U_BOOT__ */
1569 /* Check if this is a simple builtin (not part of a pipe).
1570 * Builtins within pipes have to fork anyway, and are handled in
1571 * pseudo_exec. "echo foo | read bar" doesn't work on bash, either.
1573 if (pi
->num_progs
== 1) child
= & (pi
->progs
[0]);
1575 if (pi
->num_progs
== 1 && child
->group
&& child
->subshell
== 0) {
1576 int squirrel
[] = {-1, -1, -1};
1578 debug_printf("non-subshell grouping\n");
1579 setup_redirects(child
, squirrel
);
1580 /* XXX could we merge code with following builtin case,
1581 * by creating a pseudo builtin that calls run_list_real? */
1582 rcode
= run_list_real(child
->group
);
1583 restore_redirects(squirrel
);
1585 if (pi
->num_progs
== 1 && child
->group
) {
1587 debug_printf("non-subshell grouping\n");
1588 rcode
= run_list_real(child
->group
);
1591 } else if (pi
->num_progs
== 1 && pi
->progs
[0].argv
!= NULL
) {
1592 for (i
=0; is_assignment(child
->argv
[i
]); i
++) { /* nothing */ }
1593 if (i
!=0 && child
->argv
[i
]==NULL
) {
1594 /* assignments, but no command: set the local environment */
1595 for (i
=0; child
->argv
[i
]!=NULL
; i
++) {
1597 /* Ok, this case is tricky. We have to decide if this is a
1598 * local variable, or an already exported variable. If it is
1599 * already exported, we have to export the new value. If it is
1600 * not exported, we need only set this as a local variable.
1601 * This junk is all to decide whether or not to export this
1605 name
= xstrdup(child
->argv
[i
]);
1606 debug_printf("Local environment set: %s\n", name
);
1607 value
= strchr(name
, '=');
1611 if ( get_local_var(name
)) {
1616 p
= insert_var_value(child
->argv
[i
]);
1617 set_local_var(p
, export_me
);
1618 if (p
!= child
->argv
[i
]) free(p
);
1620 return EXIT_SUCCESS
; /* don't worry about errors in set_local_var() yet */
1622 for (i
= 0; is_assignment(child
->argv
[i
]); i
++) {
1623 p
= insert_var_value(child
->argv
[i
]);
1627 set_local_var(p
, 0);
1629 if (p
!= child
->argv
[i
]) {
1637 str
= make_string(child
->argv
+ i
,
1638 child
->argv_nonnull
+ i
);
1639 parse_string_outer(str
, FLAG_EXIT_FROM_LOOP
| FLAG_REPARSING
);
1641 return last_return_code
;
1644 for (x
= bltins
; x
->cmd
; x
++) {
1645 if (strcmp(child
->argv
[i
], x
->cmd
) == 0 ) {
1646 int squirrel
[] = {-1, -1, -1};
1648 if (x
->function
== builtin_exec
&& child
->argv
[i
+1]==NULL
) {
1649 debug_printf("magic exec\n");
1650 setup_redirects(child
,NULL
);
1651 return EXIT_SUCCESS
;
1653 debug_printf("builtin inline %s\n", child
->argv
[0]);
1654 /* XXX setup_redirects acts on file descriptors, not FILEs.
1655 * This is perfect for work that comes after exec().
1656 * Is it really safe for inline use? Experimentally,
1657 * things seem to work with glibc. */
1658 setup_redirects(child
, squirrel
);
1660 child
->argv
+= i
; /* XXX horrible hack */
1661 rcode
= x
->function(child
);
1662 /* XXX restore hack so free() can work right */
1664 restore_redirects(squirrel
);
1669 /* check ";", because ,example , argv consist from
1670 * "help;flinfo" must not execute
1672 if (strchr(child
->argv
[i
], ';')) {
1673 printf("Unknown command '%s' - try 'help' or use "
1674 "'run' command\n", child
->argv
[i
]);
1677 /* Process the command */
1678 return cmd_process(flag
, child
->argc
, child
->argv
,
1679 &flag_repeat
, NULL
);
1684 for (i
= 0; i
< pi
->num_progs
; i
++) {
1685 child
= & (pi
->progs
[i
]);
1687 /* pipes are inserted between pairs of commands */
1688 if ((i
+ 1) < pi
->num_progs
) {
1689 if (pipe(pipefds
)<0) perror_msg_and_die("pipe");
1690 nextout
= pipefds
[1];
1696 /* XXX test for failed fork()? */
1697 if (!(child
->pid
= fork())) {
1698 /* Set the handling for job control signals back to the default. */
1699 signal(SIGINT
, SIG_DFL
);
1700 signal(SIGQUIT
, SIG_DFL
);
1701 signal(SIGTERM
, SIG_DFL
);
1702 signal(SIGTSTP
, SIG_DFL
);
1703 signal(SIGTTIN
, SIG_DFL
);
1704 signal(SIGTTOU
, SIG_DFL
);
1705 signal(SIGCHLD
, SIG_DFL
);
1717 if (pipefds
[0]!=-1) {
1718 close(pipefds
[0]); /* opposite end of our output pipe */
1721 /* Like bash, explicit redirects override pipes,
1722 * and the pipe fd is available for dup'ing. */
1723 setup_redirects(child
,NULL
);
1725 if (interactive
&& pi
->followup
!=PIPE_BG
) {
1726 /* If we (the child) win the race, put ourselves in the process
1727 * group whose leader is the first process in this pipe. */
1729 pi
->pgrp
= getpid();
1731 if (setpgid(0, pi
->pgrp
) == 0) {
1732 tcsetpgrp(2, pi
->pgrp
);
1740 /* put our child in the process group whose leader is the
1741 first process in this pipe */
1743 pi
->pgrp
= child
->pid
;
1745 /* Don't check for errors. The child may be dead already,
1746 * in which case setpgid returns error code EACCES. */
1747 setpgid(child
->pid
, pi
->pgrp
);
1754 /* If there isn't another process, nextin is garbage
1755 but it doesn't matter */
1756 nextin
= pipefds
[0];
1762 static int run_list_real(struct pipe
*pi
)
1764 char *save_name
= NULL
;
1766 char **save_list
= NULL
;
1772 int rcode
=0, flag_skip
=1;
1773 int flag_restore
= 0;
1774 int if_code
=0, next_if_code
=0; /* need double-buffer to handle elif */
1775 reserved_style rmode
, skip_more_in_this_rmode
=RES_XXXX
;
1776 /* check syntax for "for" */
1777 for (rpipe
= pi
; rpipe
; rpipe
= rpipe
->next
) {
1778 if ((rpipe
->r_mode
== RES_IN
||
1779 rpipe
->r_mode
== RES_FOR
) &&
1780 (rpipe
->next
== NULL
)) {
1787 if ((rpipe
->r_mode
== RES_IN
&&
1788 (rpipe
->next
->r_mode
== RES_IN
&&
1789 rpipe
->next
->progs
->argv
!= NULL
))||
1790 (rpipe
->r_mode
== RES_FOR
&&
1791 rpipe
->next
->r_mode
!= RES_IN
)) {
1799 for (; pi
; pi
= (flag_restore
!= 0) ? rpipe
: pi
->next
) {
1800 if (pi
->r_mode
== RES_WHILE
|| pi
->r_mode
== RES_UNTIL
||
1801 pi
->r_mode
== RES_FOR
) {
1805 if ((had_ctrlc())) {
1816 debug_printf("rmode=%d if_code=%d next_if_code=%d skip_more=%d\n", rmode
, if_code
, next_if_code
, skip_more_in_this_rmode
);
1817 if (rmode
== skip_more_in_this_rmode
&& flag_skip
) {
1818 if (pi
->followup
== PIPE_SEQ
) flag_skip
=0;
1822 skip_more_in_this_rmode
= RES_XXXX
;
1823 if (rmode
== RES_THEN
|| rmode
== RES_ELSE
) if_code
= next_if_code
;
1824 if (rmode
== RES_THEN
&& if_code
) continue;
1825 if (rmode
== RES_ELSE
&& !if_code
) continue;
1826 if (rmode
== RES_ELIF
&& !if_code
) break;
1827 if (rmode
== RES_FOR
&& pi
->num_progs
) {
1829 /* if no variable values after "in" we skip "for" */
1830 if (!pi
->next
->progs
->argv
) continue;
1831 /* create list of variable values */
1832 list
= make_list_in(pi
->next
->progs
->argv
,
1833 pi
->progs
->argv
[0]);
1835 save_name
= pi
->progs
->argv
[0];
1836 pi
->progs
->argv
[0] = NULL
;
1840 free(pi
->progs
->argv
[0]);
1844 pi
->progs
->argv
[0] = save_name
;
1846 pi
->progs
->glob_result
.gl_pathv
[0] =
1851 /* insert new value from list for variable */
1852 free(pi
->progs
->argv
[0]);
1853 pi
->progs
->argv
[0] = *list
++;
1855 pi
->progs
->glob_result
.gl_pathv
[0] =
1860 if (rmode
== RES_IN
) continue;
1861 if (rmode
== RES_DO
) {
1862 if (!flag_rep
) continue;
1864 if (rmode
== RES_DONE
) {
1871 if (pi
->num_progs
== 0) continue;
1873 save_num_progs
= pi
->num_progs
; /* save number of programs */
1875 rcode
= run_pipe_real(pi
);
1876 debug_printf("run_pipe_real returned %d\n",rcode
);
1879 /* We only ran a builtin: rcode was set by the return value
1880 * of run_pipe_real(), and we don't need to wait for anything. */
1881 } else if (pi
->followup
==PIPE_BG
) {
1882 /* XXX check bash's behavior with nontrivial pipes */
1883 /* XXX compute jobid */
1884 /* XXX what does bash do with attempts to background builtins? */
1886 rcode
= EXIT_SUCCESS
;
1889 /* move the new process group into the foreground */
1890 if (tcsetpgrp(shell_terminal
, pi
->pgrp
) && errno
!= ENOTTY
)
1891 perror_msg("tcsetpgrp-3");
1892 rcode
= checkjobs(pi
);
1893 /* move the shell to the foreground */
1894 if (tcsetpgrp(shell_terminal
, getpgid(0)) && errno
!= ENOTTY
)
1895 perror_msg("tcsetpgrp-4");
1897 rcode
= checkjobs(pi
);
1899 debug_printf("checkjobs returned %d\n",rcode
);
1901 last_return_code
=rcode
;
1904 last_return_code
= -rcode
- 2;
1905 return -2; /* exit */
1907 last_return_code
=(rcode
== 0) ? 0 : 1;
1910 pi
->num_progs
= save_num_progs
; /* restore number of programs */
1912 if ( rmode
== RES_IF
|| rmode
== RES_ELIF
)
1913 next_if_code
=rcode
; /* can be overwritten a number of times */
1914 if (rmode
== RES_WHILE
)
1915 flag_rep
= !last_return_code
;
1916 if (rmode
== RES_UNTIL
)
1917 flag_rep
= last_return_code
;
1918 if ( (rcode
==EXIT_SUCCESS
&& pi
->followup
==PIPE_OR
) ||
1919 (rcode
!=EXIT_SUCCESS
&& pi
->followup
==PIPE_AND
) )
1920 skip_more_in_this_rmode
=rmode
;
1928 /* broken, of course, but OK for testing */
1929 static char *indenter(int i
)
1931 static char blanks
[]=" ";
1932 return &blanks
[sizeof(blanks
)-i
-1];
1935 /* return code is the exit status of the pipe */
1936 static int free_pipe(struct pipe
*pi
, int indent
)
1939 struct child_prog
*child
;
1941 struct redir_struct
*r
, *rnext
;
1943 int a
, i
, ret_code
=0;
1944 char *ind
= indenter(indent
);
1947 if (pi
->stopped_progs
> 0)
1949 final_printf("%s run pipe: (pid %d)\n",ind
,getpid());
1951 for (i
=0; i
<pi
->num_progs
; i
++) {
1952 child
= &pi
->progs
[i
];
1953 final_printf("%s command %d:\n",ind
,i
);
1955 for (a
=0,p
=child
->argv
; *p
; a
++,p
++) {
1956 final_printf("%s argv[%d] = %s\n",ind
,a
,*p
);
1959 globfree(&child
->glob_result
);
1961 for (a
= 0; a
< child
->argc
; a
++) {
1962 free(child
->argv
[a
]);
1965 free(child
->argv_nonnull
);
1969 } else if (child
->group
) {
1971 final_printf("%s begin group (subshell:%d)\n",ind
, child
->subshell
);
1973 ret_code
= free_pipe_list(child
->group
,indent
+3);
1974 final_printf("%s end group\n",ind
);
1976 final_printf("%s (nil)\n",ind
);
1979 for (r
=child
->redirects
; r
; r
=rnext
) {
1980 final_printf("%s redirect %d%s", ind
, r
->fd
, redir_table
[r
->type
].descrip
);
1982 /* guard against the case >$FOO, where foo is unset or blank */
1983 if (r
->word
.gl_pathv
) {
1984 final_printf(" %s\n", *r
->word
.gl_pathv
);
1988 final_printf("&%d\n", r
->dup
);
1993 child
->redirects
=NULL
;
1996 free(pi
->progs
); /* children are an array, they get freed all at once */
2001 static int free_pipe_list(struct pipe
*head
, int indent
)
2003 int rcode
=0; /* if list has no members */
2004 struct pipe
*pi
, *next
;
2005 char *ind
= indenter(indent
);
2006 for (pi
=head
; pi
; pi
=next
) {
2007 final_printf("%s pipe reserved mode %d\n", ind
, pi
->r_mode
);
2008 rcode
= free_pipe(pi
, indent
);
2009 final_printf("%s pipe followup code %d\n", ind
, pi
->followup
);
2017 /* Select which version we will use */
2018 static int run_list(struct pipe
*pi
)
2024 rcode
= run_list_real(pi
);
2028 /* free_pipe_list has the side effect of clearing memory
2029 * In the long run that function can be merged with run_list_real,
2030 * but doing that now would hobble the debugging effort. */
2031 free_pipe_list(pi
,0);
2035 /* The API for glob is arguably broken. This routine pushes a non-matching
2036 * string into the output structure, removing non-backslashed backslashes.
2037 * If someone can prove me wrong, by performing this function within the
2038 * original glob(3) api, feel free to rewrite this routine into oblivion.
2039 * Return code (0 vs. GLOB_NOSPACE) matches glob(3).
2040 * XXX broken if the last character is '\\', check that before calling.
2043 static int globhack(const char *src
, int flags
, glob_t
*pglob
)
2048 for (cnt
=1, s
=src
; s
&& *s
; s
++) {
2049 if (*s
== '\\') s
++;
2053 if (!dest
) return GLOB_NOSPACE
;
2054 if (!(flags
& GLOB_APPEND
)) {
2055 pglob
->gl_pathv
=NULL
;
2060 pathc
= ++pglob
->gl_pathc
;
2061 pglob
->gl_pathv
= realloc(pglob
->gl_pathv
, (pathc
+1)*sizeof(*pglob
->gl_pathv
));
2062 if (pglob
->gl_pathv
== NULL
) return GLOB_NOSPACE
;
2063 pglob
->gl_pathv
[pathc
-1]=dest
;
2064 pglob
->gl_pathv
[pathc
]=NULL
;
2065 for (s
=src
; s
&& *s
; s
++, dest
++) {
2066 if (*s
== '\\') s
++;
2073 /* XXX broken if the last character is '\\', check that before calling */
2074 static int glob_needed(const char *s
)
2077 if (*s
== '\\') s
++;
2078 if (strchr("*[?",*s
)) return 1;
2084 static void globprint(glob_t
*pglob
)
2087 debug_printf("glob_t at %p:\n", pglob
);
2088 debug_printf(" gl_pathc=%d gl_pathv=%p gl_offs=%d gl_flags=%d\n",
2089 pglob
->gl_pathc
, pglob
->gl_pathv
, pglob
->gl_offs
, pglob
->gl_flags
);
2090 for (i
=0; i
<pglob
->gl_pathc
; i
++)
2091 debug_printf("pglob->gl_pathv[%d] = %p = %s\n", i
,
2092 pglob
->gl_pathv
[i
], pglob
->gl_pathv
[i
]);
2096 static int xglob(o_string
*dest
, int flags
, glob_t
*pglob
)
2100 /* short-circuit for null word */
2101 /* we can code this better when the debug_printf's are gone */
2102 if (dest
->length
== 0) {
2103 if (dest
->nonnull
) {
2104 /* bash man page calls this an "explicit" null */
2105 gr
= globhack(dest
->data
, flags
, pglob
);
2106 debug_printf("globhack returned %d\n",gr
);
2110 } else if (glob_needed(dest
->data
)) {
2111 gr
= glob(dest
->data
, flags
, NULL
, pglob
);
2112 debug_printf("glob returned %d\n",gr
);
2113 if (gr
== GLOB_NOMATCH
) {
2114 /* quote removal, or more accurately, backslash removal */
2115 gr
= globhack(dest
->data
, flags
, pglob
);
2116 debug_printf("globhack returned %d\n",gr
);
2119 gr
= globhack(dest
->data
, flags
, pglob
);
2120 debug_printf("globhack returned %d\n",gr
);
2122 if (gr
== GLOB_NOSPACE
)
2123 error_msg_and_die("out of memory during glob");
2124 if (gr
!= 0) { /* GLOB_ABORTED ? */
2125 error_msg("glob(3) error %d",gr
);
2127 /* globprint(glob_target); */
2133 static char *get_dollar_var(char ch
);
2136 /* This is used to get/check local shell variables */
2137 char *get_local_var(const char *s
)
2139 struct variables
*cur
;
2146 return get_dollar_var(s
[1]);
2149 for (cur
= top_vars
; cur
; cur
=cur
->next
)
2150 if(strcmp(cur
->name
, s
)==0)
2155 /* This is used to set local shell variables
2156 flg_export==0 if only local (not exporting) variable
2157 flg_export==1 if "new" exporting environ
2158 flg_export>1 if current startup environ (not call putenv()) */
2159 int set_local_var(const char *s
, int flg_export
)
2163 struct variables
*cur
;
2166 /* might be possible! */
2174 if (env_get(name
) != NULL
) {
2176 "There is a global environment variable with the same name.\n");
2181 /* Assume when we enter this function that we are already in
2182 * NAME=VALUE format. So the first order of business is to
2183 * split 's' on the '=' into 'name' and 'value' */
2184 value
= strchr(name
, '=');
2185 if (value
== NULL
|| *(value
+ 1) == 0) {
2191 for(cur
= top_vars
; cur
; cur
= cur
->next
) {
2192 if(strcmp(cur
->name
, name
)==0)
2197 if(strcmp(cur
->value
, value
)==0) {
2198 if(flg_export
>0 && cur
->flg_export
==0)
2199 cur
->flg_export
=flg_export
;
2203 if(cur
->flg_read_only
) {
2204 error_msg("%s: readonly variable", name
);
2207 if(flg_export
>0 || cur
->flg_export
>1)
2211 cur
->value
= strdup(value
);
2215 cur
= malloc(sizeof(struct variables
));
2219 cur
->name
= strdup(name
);
2220 if (cur
->name
== NULL
) {
2224 struct variables
*bottom
= top_vars
;
2225 cur
->value
= strdup(value
);
2227 cur
->flg_export
= flg_export
;
2228 cur
->flg_read_only
= 0;
2229 while(bottom
->next
) bottom
=bottom
->next
;
2236 if(result
==0 && cur
->flg_export
==1) {
2238 result
= putenv(name
);
2243 if(result
>0) /* equivalent to previous set */
2250 void unset_local_var(const char *name
)
2252 struct variables
*cur
;
2255 for (cur
= top_vars
; cur
; cur
=cur
->next
) {
2256 if(strcmp(cur
->name
, name
)==0)
2260 struct variables
*next
= top_vars
;
2261 if(cur
->flg_read_only
) {
2262 error_msg("%s: readonly variable", name
);
2267 unenv_set(cur
->name
);
2271 while (next
->next
!= cur
)
2273 next
->next
= cur
->next
;
2280 static int is_assignment(const char *s
)
2285 if (!isalpha(*s
)) return 0;
2287 while(isalnum(*s
) || *s
=='_') ++s
;
2292 /* the src parameter allows us to peek forward to a possible &n syntax
2293 * for file descriptor duplication, e.g., "2>&1".
2294 * Return code is 0 normally, 1 if a syntax error is detected in src.
2295 * Resource errors (in xmalloc) cause the process to exit */
2296 static int setup_redirect(struct p_context
*ctx
, int fd
, redir_type style
,
2297 struct in_str
*input
)
2299 struct child_prog
*child
=ctx
->child
;
2300 struct redir_struct
*redir
= child
->redirects
;
2301 struct redir_struct
*last_redir
=NULL
;
2303 /* Create a new redir_struct and drop it onto the end of the linked list */
2308 redir
= xmalloc(sizeof(struct redir_struct
));
2310 redir
->word
.gl_pathv
=NULL
;
2312 last_redir
->next
=redir
;
2314 child
->redirects
=redir
;
2318 redir
->fd
= (fd
==-1) ? redir_table
[style
].default_fd
: fd
;
2320 debug_printf("Redirect type %d%s\n", redir
->fd
, redir_table
[style
].descrip
);
2322 /* Check for a '2>&1' type redirect */
2323 redir
->dup
= redirect_dup_num(input
);
2324 if (redir
->dup
== -2) return 1; /* syntax error */
2325 if (redir
->dup
!= -1) {
2326 /* Erik had a check here that the file descriptor in question
2327 * is legit; I postpone that to "run time"
2328 * A "-" representation of "close me" shows up as a -3 here */
2329 debug_printf("Duplicating redirect '%d>&%d'\n", redir
->fd
, redir
->dup
);
2331 /* We do _not_ try to open the file that src points to,
2332 * since we need to return and let src be expanded first.
2333 * Set ctx->pending_redirect, so we know what to do at the
2334 * end of the next parsed word.
2336 ctx
->pending_redirect
= redir
;
2342 static struct pipe
*new_pipe(void)
2345 pi
= xmalloc(sizeof(struct pipe
));
2349 pi
->followup
= 0; /* invalid */
2350 pi
->r_mode
= RES_NONE
;
2354 static void initialize_context(struct p_context
*ctx
)
2358 ctx
->pending_redirect
=NULL
;
2361 ctx
->list_head
=new_pipe();
2362 ctx
->pipe
=ctx
->list_head
;
2368 done_command(ctx
); /* creates the memory for working child */
2371 /* normal return is 0
2372 * if a reserved word is found, and processed, return 1
2373 * should handle if, then, elif, else, fi, for, while, until, do, done.
2374 * case, function, and select are obnoxious, save those for later.
2376 struct reserved_combo
{
2381 /* Mostly a list of accepted follow-up reserved words.
2382 * FLAG_END means we are done with the sequence, and are ready
2383 * to turn the compound list into a command.
2384 * FLAG_START means the word must start a new compound list.
2386 static struct reserved_combo reserved_list
[] = {
2387 { "if", RES_IF
, FLAG_THEN
| FLAG_START
},
2388 { "then", RES_THEN
, FLAG_ELIF
| FLAG_ELSE
| FLAG_FI
},
2389 { "elif", RES_ELIF
, FLAG_THEN
},
2390 { "else", RES_ELSE
, FLAG_FI
},
2391 { "fi", RES_FI
, FLAG_END
},
2392 { "for", RES_FOR
, FLAG_IN
| FLAG_START
},
2393 { "while", RES_WHILE
, FLAG_DO
| FLAG_START
},
2394 { "until", RES_UNTIL
, FLAG_DO
| FLAG_START
},
2395 { "in", RES_IN
, FLAG_DO
},
2396 { "do", RES_DO
, FLAG_DONE
},
2397 { "done", RES_DONE
, FLAG_END
}
2399 #define NRES (sizeof(reserved_list)/sizeof(struct reserved_combo))
2401 static int reserved_word(o_string
*dest
, struct p_context
*ctx
)
2403 struct reserved_combo
*r
;
2404 for (r
=reserved_list
;
2405 r
<reserved_list
+NRES
; r
++) {
2406 if (strcmp(dest
->data
, r
->literal
) == 0) {
2407 debug_printf("found reserved word %s, code %d\n",r
->literal
,r
->code
);
2408 if (r
->flag
& FLAG_START
) {
2409 struct p_context
*new = xmalloc(sizeof(struct p_context
));
2410 debug_printf("push stack\n");
2411 if (ctx
->w
== RES_IN
|| ctx
->w
== RES_FOR
) {
2418 *new = *ctx
; /* physical copy */
2419 initialize_context(ctx
);
2421 } else if ( ctx
->w
== RES_NONE
|| ! (ctx
->old_flag
& (1<<r
->code
))) {
2428 ctx
->old_flag
= r
->flag
;
2429 if (ctx
->old_flag
& FLAG_END
) {
2430 struct p_context
*old
;
2431 debug_printf("pop stack\n");
2432 done_pipe(ctx
,PIPE_SEQ
);
2434 old
->child
->group
= ctx
->list_head
;
2436 old
->child
->subshell
= 0;
2438 *ctx
= *old
; /* physical copy */
2448 /* normal return is 0.
2449 * Syntax or xglob errors return 1. */
2450 static int done_word(o_string
*dest
, struct p_context
*ctx
)
2452 struct child_prog
*child
=ctx
->child
;
2454 glob_t
*glob_target
;
2461 debug_printf("done_word: %s %p\n", dest
->data
, child
);
2462 if (dest
->length
== 0 && !dest
->nonnull
) {
2463 debug_printf(" true null, ignored\n");
2467 if (ctx
->pending_redirect
) {
2468 glob_target
= &ctx
->pending_redirect
->word
;
2473 return 1; /* syntax error, groups and arglists don't mix */
2475 if (!child
->argv
&& (ctx
->type
& FLAG_PARSE_SEMICOLON
)) {
2476 debug_printf("checking %s for reserved-ness\n",dest
->data
);
2477 if (reserved_word(dest
,ctx
)) return ctx
->w
==RES_SNTX
;
2480 glob_target
= &child
->glob_result
;
2481 if (child
->argv
) flags
|= GLOB_APPEND
;
2483 for (cnt
= 1, s
= dest
->data
; s
&& *s
; s
++) {
2484 if (*s
== '\\') s
++;
2489 if ( child
->argv
== NULL
) {
2492 argc
= ++child
->argc
;
2493 child
->argv
= realloc(child
->argv
, (argc
+1)*sizeof(*child
->argv
));
2494 if (child
->argv
== NULL
) {
2498 child
->argv_nonnull
= realloc(child
->argv_nonnull
,
2499 (argc
+1)*sizeof(*child
->argv_nonnull
));
2500 if (child
->argv_nonnull
== NULL
) {
2504 child
->argv
[argc
-1]=str
;
2505 child
->argv_nonnull
[argc
-1] = dest
->nonnull
;
2506 child
->argv
[argc
]=NULL
;
2507 child
->argv_nonnull
[argc
] = 0;
2508 for (s
= dest
->data
; s
&& *s
; s
++,str
++) {
2509 if (*s
== '\\') s
++;
2516 gr
= xglob(dest
, flags
, glob_target
);
2517 if (gr
!= 0) return 1;
2522 if (ctx
->pending_redirect
) {
2523 ctx
->pending_redirect
=NULL
;
2524 if (glob_target
->gl_pathc
!= 1) {
2525 error_msg("ambiguous redirect");
2529 child
->argv
= glob_target
->gl_pathv
;
2532 if (ctx
->w
== RES_FOR
) {
2533 done_word(dest
,ctx
);
2534 done_pipe(ctx
,PIPE_SEQ
);
2539 /* The only possible error here is out of memory, in which case
2541 static int done_command(struct p_context
*ctx
)
2543 /* The child is really already in the pipe structure, so
2544 * advance the pipe counter and make a new, null child.
2545 * Only real trickiness here is that the uncommitted
2546 * child structure, to which ctx->child points, is not
2547 * counted in pi->num_progs. */
2548 struct pipe
*pi
=ctx
->pipe
;
2549 struct child_prog
*prog
=ctx
->child
;
2551 if (prog
&& prog
->group
== NULL
2552 && prog
->argv
== NULL
2554 && prog
->redirects
== NULL
) {
2558 debug_printf("done_command: skipping null command\n");
2562 debug_printf("done_command: num_progs incremented to %d\n",pi
->num_progs
);
2564 debug_printf("done_command: initializing\n");
2566 pi
->progs
= xrealloc(pi
->progs
, sizeof(*pi
->progs
) * (pi
->num_progs
+1));
2568 prog
= pi
->progs
+ pi
->num_progs
;
2570 prog
->redirects
= NULL
;
2573 prog
->argv_nonnull
= NULL
;
2575 prog
->is_stopped
= 0;
2579 prog
->glob_result
.gl_pathv
= NULL
;
2584 prog
->type
= ctx
->type
;
2586 /* but ctx->pipe and ctx->list_head remain unchanged */
2590 static int done_pipe(struct p_context
*ctx
, pipe_style type
)
2593 done_command(ctx
); /* implicit closure of previous command */
2594 debug_printf("done_pipe, type %d\n", type
);
2595 ctx
->pipe
->followup
= type
;
2596 ctx
->pipe
->r_mode
= ctx
->w
;
2598 ctx
->pipe
->next
= new_p
;
2601 done_command(ctx
); /* set up new pipe to accept commands */
2606 /* peek ahead in the in_str to find out if we have a "&n" construct,
2607 * as in "2>&1", that represents duplicating a file descriptor.
2608 * returns either -2 (syntax error), -1 (no &), or the number found.
2610 static int redirect_dup_num(struct in_str
*input
)
2614 if (ch
!= '&') return -1;
2616 b_getch(input
); /* get the & */
2620 return -3; /* "-" represents "close me" */
2622 while (isdigit(ch
)) {
2630 error_msg("ambiguous redirect");
2634 /* If a redirect is immediately preceded by a number, that number is
2635 * supposed to tell which file descriptor to redirect. This routine
2636 * looks for such preceding numbers. In an ideal world this routine
2637 * needs to handle all the following classes of redirects...
2638 * echo 2>foo # redirects fd 2 to file "foo", nothing passed to echo
2639 * echo 49>foo # redirects fd 49 to file "foo", nothing passed to echo
2640 * echo -2>foo # redirects fd 1 to file "foo", "-2" passed to echo
2641 * echo 49x>foo # redirects fd 1 to file "foo", "49x" passed to echo
2642 * A -1 output from this program means no valid number was found, so the
2643 * caller should use the appropriate default for this redirection.
2645 static int redirect_opt_num(o_string
*o
)
2649 if (o
->length
==0) return -1;
2650 for(num
=0; num
<o
->length
; num
++) {
2651 if (!isdigit(*(o
->data
+num
))) {
2655 /* reuse num (and save an int) */
2661 FILE *generate_stream_from_list(struct pipe
*head
)
2665 int pid
, channel
[2];
2666 if (pipe(channel
)<0) perror_msg_and_die("pipe");
2669 perror_msg_and_die("fork");
2670 } else if (pid
==0) {
2672 if (channel
[1] != 1) {
2677 #define SURROGATE "surrogate response"
2678 write(1,SURROGATE
,sizeof(SURROGATE
));
2679 _exit(run_list(head
));
2681 _exit(run_list_real(head
)); /* leaks memory */
2684 debug_printf("forked child %d\n",pid
);
2686 pf
= fdopen(channel
[0],"r");
2687 debug_printf("pipe on FILE *%p\n",pf
);
2689 free_pipe_list(head
,0);
2690 pf
=popen("echo surrogate response","r");
2691 debug_printf("started fake pipe on FILE *%p\n",pf
);
2696 /* this version hacked for testing purposes */
2697 /* return code is exit status of the process that is run. */
2698 static int process_command_subs(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input
, int subst_end
)
2701 o_string result
=NULL_O_STRING
;
2702 struct p_context inner
;
2704 struct in_str pipe_str
;
2705 initialize_context(&inner
);
2707 /* recursion to generate command */
2708 retcode
= parse_stream(&result
, &inner
, input
, subst_end
);
2709 if (retcode
!= 0) return retcode
; /* syntax error or EOF */
2710 done_word(&result
, &inner
);
2711 done_pipe(&inner
, PIPE_SEQ
);
2714 p
=generate_stream_from_list(inner
.list_head
);
2715 if (p
==NULL
) return 1;
2716 mark_open(fileno(p
));
2717 setup_file_in_str(&pipe_str
, p
);
2719 /* now send results of command back into original context */
2720 retcode
= parse_stream(dest
, ctx
, &pipe_str
, '\0');
2721 /* XXX In case of a syntax error, should we try to kill the child?
2722 * That would be tough to do right, so just read until EOF. */
2724 while (b_getch(&pipe_str
)!=EOF
) { /* discard */ };
2727 debug_printf("done reading from pipe, pclose()ing\n");
2728 /* This is the step that wait()s for the child. Should be pretty
2729 * safe, since we just read an EOF from its stdout. We could try
2730 * to better, by using wait(), and keeping track of background jobs
2731 * at the same time. That would be a lot of work, and contrary
2732 * to the KISS philosophy of this program. */
2733 mark_closed(fileno(p
));
2735 free_pipe_list(inner
.list_head
,0);
2736 debug_printf("pclosed, retcode=%d\n",retcode
);
2737 /* XXX this process fails to trim a single trailing newline */
2741 static int parse_group(o_string
*dest
, struct p_context
*ctx
,
2742 struct in_str
*input
, int ch
)
2745 struct p_context sub
;
2746 struct child_prog
*child
= ctx
->child
;
2749 return 1; /* syntax error, groups and arglists don't mix */
2751 initialize_context(&sub
);
2753 case '(': endch
=')'; child
->subshell
=1; break;
2754 case '{': endch
='}'; break;
2755 default: syntax(); /* really logic error */
2757 rcode
=parse_stream(dest
,&sub
,input
,endch
);
2758 done_word(dest
,&sub
); /* finish off the final word in the subcontext */
2759 done_pipe(&sub
, PIPE_SEQ
); /* and the final command there, too */
2760 child
->group
= sub
.list_head
;
2762 /* child remains "open", available for possible redirects */
2766 /* basically useful version until someone wants to get fancier,
2767 * see the bash man page under "Parameter Expansion" */
2768 static char *lookup_param(char *src
)
2772 char *default_val
= NULL
;
2774 int expand_empty
= 0;
2779 sep
= strchr(src
, ':');
2783 if (*(sep
+ 1) == '-')
2784 default_val
= sep
+2;
2785 if (*(sep
+ 1) == '=') {
2786 default_val
= sep
+2;
2789 if (*(sep
+ 1) == '+') {
2790 default_val
= sep
+2;
2797 p
= get_local_var(src
);
2799 if (!p
|| strlen(p
) == 0) {
2802 char *var
= malloc(strlen(src
)+strlen(default_val
)+2);
2804 sprintf(var
, "%s=%s", src
, default_val
);
2805 set_local_var(var
, 0);
2809 } else if (expand_empty
) {
2820 static char *get_dollar_var(char ch
)
2822 static char buf
[40];
2827 sprintf(buf
, "%u", (unsigned int)last_return_code
);
2836 /* return code: 0 for OK, 1 for syntax error */
2837 static int handle_dollar(o_string
*dest
, struct p_context
*ctx
, struct in_str
*input
)
2847 int ch
= input
->peek(input
); /* first character after the $ */
2848 debug_printf("handle_dollar: ch=%c\n",ch
);
2850 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2852 while(ch
=b_peek(input
),isalnum(ch
) || ch
=='_') {
2856 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2858 } else if (isdigit(ch
)) {
2859 i
= ch
-'0'; /* XXX is $0 special? */
2860 if (i
<global_argc
) {
2861 parse_string(dest
, ctx
, global_argv
[i
]); /* recursion */
2865 } else switch (ch
) {
2868 b_adduint(dest
,getpid());
2872 if (last_bg_pid
> 0) b_adduint(dest
, last_bg_pid
);
2878 b_adduint(dest
,last_return_code
);
2881 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2882 b_addchr(dest
, '$');
2883 b_addchr(dest
, '?');
2884 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2890 b_adduint(dest
,global_argc
? global_argc
-1 : 0);
2895 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2898 /* XXX maybe someone will try to escape the '}' */
2899 while(ch
=b_getch(input
),ch
!=EOF
&& ch
!='}') {
2906 b_addchr(dest
, SPECIAL_VAR_SYMBOL
);
2911 process_command_subs(dest
, ctx
, input
, ')');
2915 for (i
=1; i
<global_argc
; i
++) {
2916 parse_string(dest
, ctx
, global_argv
[i
]);
2917 if (i
+1 < global_argc
) parse_string(dest
, ctx
, sep
);
2923 /* still unhandled, but should be eventually */
2924 error_msg("unhandled syntax: $%c",ch
);
2929 b_addqchr(dest
,'$',dest
->quote
);
2931 /* Eat the character if the flag was set. If the compiler
2932 * is smart enough, we could substitute "b_getch(input);"
2933 * for all the "advance = 1;" above, and also end up with
2934 * a nice size-optimized program. Hah! That'll be the day.
2936 if (advance
) b_getch(input
);
2941 int parse_string(o_string
*dest
, struct p_context
*ctx
, const char *src
)
2944 setup_string_in_str(&foo
, src
);
2945 return parse_stream(dest
, ctx
, &foo
, '\0');
2949 /* return code is 0 for normal exit, 1 for syntax error */
2950 static int parse_stream(o_string
*dest
, struct p_context
*ctx
,
2951 struct in_str
*input
, int end_trigger
)
2956 redir_type redir_style
;
2960 /* Only double-quote state is handled in the state variable dest->quote.
2961 * A single-quote triggers a bypass of the main loop until its mate is
2962 * found. When recursing, quote state is passed in via dest->quote. */
2964 debug_printf("parse_stream, end_trigger=%d\n",end_trigger
);
2965 while ((ch
=b_getch(input
))!=EOF
) {
2968 if (input
->__promptme
== 0) return 1;
2970 next
= (ch
== '\n') ? 0 : b_peek(input
);
2972 debug_printf("parse_stream: ch=%c (%d) m=%d quote=%d - %c\n",
2973 ch
>= ' ' ? ch
: '.', ch
, m
,
2974 dest
->quote
, ctx
->stack
== NULL
? '*' : '.');
2976 if (m
==0 || ((m
==1 || m
==2) && dest
->quote
)) {
2977 b_addqchr(dest
, ch
, dest
->quote
);
2979 if (m
==2) { /* unquoted IFS */
2980 if (done_word(dest
, ctx
)) {
2983 /* If we aren't performing a substitution, treat a newline as a
2984 * command separator. */
2985 if (end_trigger
!= '\0' && ch
=='\n')
2986 done_pipe(ctx
,PIPE_SEQ
);
2988 if (ch
== end_trigger
&& !dest
->quote
&& ctx
->w
==RES_NONE
) {
2989 debug_printf("leaving parse_stream (triggered)\n");
2994 /* Yahoo! Time to run with it! */
2995 done_pipe(ctx
,PIPE_SEQ
);
2996 run_list(ctx
->list_head
);
2997 initialize_context(ctx
);
3000 if (m
!=2) switch (ch
) {
3002 if (dest
->length
== 0 && !dest
->quote
) {
3003 while(ch
=b_peek(input
),ch
!=EOF
&& ch
!='\n') { b_getch(input
); }
3005 b_addqchr(dest
, ch
, dest
->quote
);
3013 b_addqchr(dest
, '\\', dest
->quote
);
3014 b_addqchr(dest
, b_getch(input
), dest
->quote
);
3017 if (handle_dollar(dest
, ctx
, input
)!=0) return 1;
3021 while(ch
=b_getch(input
),ch
!=EOF
&& ch
!='\'') {
3023 if(input
->__promptme
== 0) return 1;
3034 dest
->quote
= !dest
->quote
;
3038 process_command_subs(dest
, ctx
, input
, '`');
3041 redir_fd
= redirect_opt_num(dest
);
3042 done_word(dest
, ctx
);
3043 redir_style
=REDIRECT_OVERWRITE
;
3045 redir_style
=REDIRECT_APPEND
;
3047 } else if (next
== '(') {
3048 syntax(); /* until we support >(list) Process Substitution */
3051 setup_redirect(ctx
, redir_fd
, redir_style
, input
);
3054 redir_fd
= redirect_opt_num(dest
);
3055 done_word(dest
, ctx
);
3056 redir_style
=REDIRECT_INPUT
;
3058 redir_style
=REDIRECT_HEREIS
;
3060 } else if (next
== '>') {
3061 redir_style
=REDIRECT_IO
;
3063 } else if (next
== '(') {
3064 syntax(); /* until we support <(list) Process Substitution */
3067 setup_redirect(ctx
, redir_fd
, redir_style
, input
);
3071 done_word(dest
, ctx
);
3072 done_pipe(ctx
,PIPE_SEQ
);
3075 done_word(dest
, ctx
);
3078 done_pipe(ctx
,PIPE_AND
);
3081 done_pipe(ctx
,PIPE_BG
);
3089 done_word(dest
, ctx
);
3092 done_pipe(ctx
,PIPE_OR
);
3094 /* we could pick up a file descriptor choice here
3095 * with redirect_opt_num(), but bash doesn't do it.
3096 * "echo foo 2| cat" yields "foo 2". */
3108 if (parse_group(dest
, ctx
, input
, ch
)!=0) return 1;
3112 syntax(); /* Proper use of this character caught by end_trigger */
3116 case SUBSTED_VAR_SYMBOL
:
3118 while (ch
= b_getch(input
), ch
!= EOF
&&
3119 ch
!= SUBSTED_VAR_SYMBOL
) {
3120 debug_printf("subst, pass=%d\n", ch
);
3121 if (input
->__promptme
== 0)
3125 debug_printf("subst, term=%d\n", ch
);
3132 syntax(); /* this is really an internal logic error */
3137 /* complain if quote? No, maybe we just finished a command substitution
3138 * that was quoted. Example:
3139 * $ echo "`cat foo` plus more"
3140 * and we just got the EOF generated by the subshell that ran "cat foo"
3141 * The only real complaint is if we got an EOF when end_trigger != '\0',
3142 * that is, we were really supposed to get end_trigger, and never got
3143 * one before the EOF. Can't use the standard "syntax error" return code,
3144 * so that parse_stream_outer can distinguish the EOF and exit smoothly. */
3145 debug_printf("leaving parse_stream (EOF)\n");
3146 if (end_trigger
!= '\0') return -1;
3150 static void mapset(const unsigned char *set
, int code
)
3152 const unsigned char *s
;
3153 for (s
=set
; *s
; s
++) map
[*s
] = code
;
3156 static void update_ifs_map(void)
3158 /* char *ifs and char map[256] are both globals. */
3159 ifs
= (uchar
*)env_get("IFS");
3160 if (ifs
== NULL
) ifs
=(uchar
*)" \t\n";
3161 /* Precompute a list of 'flow through' behavior so it can be treated
3162 * quickly up front. Computation is necessary because of IFS.
3163 * Special case handling of IFS == " \t\n" is not implemented.
3164 * The map[] array only really needs two bits each, and on most machines
3165 * that would be faster because of the reduced L1 cache footprint.
3167 memset(map
,0,sizeof(map
)); /* most characters flow through always */
3169 mapset((uchar
*)"\\$'\"`", 3); /* never flow through */
3170 mapset((uchar
*)"<>;&|(){}#", 1); /* flow through if quoted */
3173 uchar subst
[2] = {SUBSTED_VAR_SYMBOL
, 0};
3174 mapset(subst
, 3); /* never flow through */
3176 mapset((uchar
*)"\\$'\"", 3); /* never flow through */
3177 mapset((uchar
*)";&|#", 1); /* flow through if quoted */
3179 mapset(ifs
, 2); /* also flow through if quoted */
3182 /* most recursion does not come through here, the exeception is
3183 * from builtin_source() */
3184 static int parse_stream_outer(struct in_str
*inp
, int flag
)
3187 struct p_context ctx
;
3188 o_string temp
=NULL_O_STRING
;
3195 initialize_context(&ctx
);
3197 if (!(flag
& FLAG_PARSE_SEMICOLON
) || (flag
& FLAG_REPARSING
)) mapset((uchar
*)";$&|", 0);
3199 rcode
= parse_stream(&temp
, &ctx
, inp
,
3200 flag
& FLAG_CONT_ON_NEWLINE
? -1 : '\n');
3202 if (rcode
== 1) flag_repeat
= 0;
3204 if (rcode
!= 1 && ctx
.old_flag
!= 0) {
3210 if (rcode
!= 1 && ctx
.old_flag
== 0) {
3211 done_word(&temp
, &ctx
);
3212 done_pipe(&ctx
,PIPE_SEQ
);
3214 run_list(ctx
.list_head
);
3216 code
= run_list(ctx
.list_head
);
3217 if (code
== -2) { /* exit */
3220 /* XXX hackish way to not allow exit from main loop */
3221 if (inp
->peek
== file_peek
) {
3222 printf("exit not allowed from main input shell.\n");
3231 if (ctx
.old_flag
!= 0) {
3236 if (inp
->__promptme
== 0) printf("<INTERRUPT>\n");
3237 inp
->__promptme
= 1;
3242 free_pipe_list(ctx
.list_head
,0);
3245 /* loop on syntax errors, return on EOF */
3246 } while (rcode
!= -1 && !(flag
& FLAG_EXIT_FROM_LOOP
) &&
3247 (inp
->peek
!= static_peek
|| b_peek(inp
)));
3251 return (code
!= 0) ? 1 : 0;
3252 #endif /* __U_BOOT__ */
3256 static int parse_string_outer(const char *s
, int flag
)
3258 int parse_string_outer(const char *s
, int flag
)
3259 #endif /* __U_BOOT__ */
3261 struct in_str input
;
3269 if (!(p
= strchr(s
, '\n')) || *++p
) {
3270 p
= xmalloc(strlen(s
) + 2);
3273 setup_string_in_str(&input
, p
);
3274 rcode
= parse_stream_outer(&input
, flag
);
3279 setup_string_in_str(&input
, s
);
3280 return parse_stream_outer(&input
, flag
);
3287 static int parse_file_outer(FILE *f
)
3289 int parse_file_outer(void)
3293 struct in_str input
;
3295 setup_file_in_str(&input
, f
);
3297 setup_file_in_str(&input
);
3299 rcode
= parse_stream_outer(&input
, FLAG_PARSE_SEMICOLON
);
3304 #ifdef CONFIG_NEEDS_MANUAL_RELOC
3305 static void u_boot_hush_reloc(void)
3308 struct reserved_combo
*r
;
3310 for (r
=reserved_list
; r
<reserved_list
+NRES
; r
++) {
3311 addr
= (ulong
) (r
->literal
) + gd
->reloc_off
;
3312 r
->literal
= (char *)addr
;
3317 int u_boot_hush_start(void)
3319 if (top_vars
== NULL
) {
3320 top_vars
= malloc(sizeof(struct variables
));
3321 top_vars
->name
= "HUSH_VERSION";
3322 top_vars
->value
= "0.01";
3323 top_vars
->next
= NULL
;
3324 top_vars
->flg_export
= 0;
3325 top_vars
->flg_read_only
= 1;
3326 #ifdef CONFIG_NEEDS_MANUAL_RELOC
3327 u_boot_hush_reloc();
3333 static void *xmalloc(size_t size
)
3337 if (!(p
= malloc(size
))) {
3338 printf("ERROR : memory not allocated\n");
3344 static void *xrealloc(void *ptr
, size_t size
)
3348 if (!(p
= realloc(ptr
, size
))) {
3349 printf("ERROR : memory not allocated\n");
3354 #endif /* __U_BOOT__ */
3357 /* Make sure we have a controlling tty. If we get started under a job
3358 * aware app (like bash for example), make sure we are now in charge so
3359 * we don't fight over who gets the foreground */
3360 static void setup_job_control(void)
3362 static pid_t shell_pgrp
;
3363 /* Loop until we are in the foreground. */
3364 while (tcgetpgrp (shell_terminal
) != (shell_pgrp
= getpgrp ()))
3365 kill (- shell_pgrp
, SIGTTIN
);
3367 /* Ignore interactive and job-control signals. */
3368 signal(SIGINT
, SIG_IGN
);
3369 signal(SIGQUIT
, SIG_IGN
);
3370 signal(SIGTERM
, SIG_IGN
);
3371 signal(SIGTSTP
, SIG_IGN
);
3372 signal(SIGTTIN
, SIG_IGN
);
3373 signal(SIGTTOU
, SIG_IGN
);
3374 signal(SIGCHLD
, SIG_IGN
);
3376 /* Put ourselves in our own process group. */
3378 shell_pgrp
= getpid ();
3379 setpgid (shell_pgrp
, shell_pgrp
);
3381 /* Grab control of the terminal. */
3382 tcsetpgrp(shell_terminal
, shell_pgrp
);
3385 int hush_main(int argc
, char * const *argv
)
3391 /* XXX what should these be while sourcing /etc/profile? */
3395 /* (re?) initialize globals. Sometimes hush_main() ends up calling
3396 * hush_main(), therefore we cannot rely on the BSS to zero out this
3397 * stuff. Reset these to 0 every time. */
3399 /* map[] is taken care of with call to update_ifs_map() */
3402 close_me_head
= NULL
;
3407 /* Initialize some more globals to non-zero values */
3409 #ifdef CONFIG_FEATURE_COMMAND_EDITING
3410 cmdedit_set_initial_prompt();
3416 /* initialize our shell local variables with the values
3417 * currently living in the environment */
3420 set_local_var(*e
, 2); /* without call putenv() */
3423 last_return_code
=EXIT_SUCCESS
;
3426 if (argv
[0] && argv
[0][0] == '-') {
3427 debug_printf("\nsourcing /etc/profile\n");
3428 if ((input
= fopen("/etc/profile", "r")) != NULL
) {
3429 mark_open(fileno(input
));
3430 parse_file_outer(input
);
3431 mark_closed(fileno(input
));
3437 while ((opt
= getopt(argc
, argv
, "c:xif")) > 0) {
3441 global_argv
= argv
+optind
;
3442 global_argc
= argc
-optind
;
3443 opt
= parse_string_outer(optarg
, FLAG_PARSE_SEMICOLON
);
3455 fprintf(stderr
, "Usage: sh [FILE]...\n"
3456 " or: sh -c command [args]...\n\n");
3463 /* A shell is interactive if the `-i' flag was given, or if all of
3464 * the following conditions are met:
3466 * no arguments remaining or the -s flag given
3467 * standard input is a terminal
3468 * standard output is a terminal
3469 * Refer to Posix.2, the description of the `sh' utility. */
3470 if (argv
[optind
]==NULL
&& input
==stdin
&&
3471 isatty(fileno(stdin
)) && isatty(fileno(stdout
))) {
3475 debug_printf("\ninteractive=%d\n", interactive
);
3477 /* Looks like they want an interactive shell */
3478 #ifndef CONFIG_FEATURE_SH_EXTRA_QUIET
3479 printf( "\n\n" BB_BANNER
" hush - the humble shell v0.01 (testing)\n");
3480 printf( "Enter 'help' for a list of built-in commands.\n\n");
3482 setup_job_control();
3485 if (argv
[optind
]==NULL
) {
3486 opt
=parse_file_outer(stdin
);
3490 debug_printf("\nrunning script '%s'\n", argv
[optind
]);
3491 global_argv
= argv
+optind
;
3492 global_argc
= argc
-optind
;
3493 input
= xfopen(argv
[optind
], "r");
3494 opt
= parse_file_outer(input
);
3496 #ifdef CONFIG_FEATURE_CLEAN_UP
3498 if (cwd
&& cwd
!= unknown
)
3501 struct variables
*cur
, *tmp
;
3502 for(cur
= top_vars
; cur
; cur
= tmp
) {
3504 if (!cur
->flg_read_only
) {
3514 return(opt
?opt
:last_return_code
);
3518 static char *insert_var_value(char *inp
)
3520 return insert_var_value_sub(inp
, 0);
3523 static char *insert_var_value_sub(char *inp
, int tag_subst
)
3525 int res_str_len
= 0;
3528 char *p
, *p1
, *res_str
= NULL
;
3530 while ((p
= strchr(inp
, SPECIAL_VAR_SYMBOL
))) {
3531 /* check the beginning of the string for normal characters */
3533 /* copy any characters to the result string */
3535 res_str
= xrealloc(res_str
, (res_str_len
+ len
));
3536 strncpy((res_str
+ res_str_len
), inp
, len
);
3540 /* find the ending marker */
3541 p
= strchr(inp
, SPECIAL_VAR_SYMBOL
);
3543 /* look up the value to substitute */
3544 if ((p1
= lookup_param(inp
))) {
3546 len
= res_str_len
+ strlen(p1
) + 2;
3548 len
= res_str_len
+ strlen(p1
);
3549 res_str
= xrealloc(res_str
, (1 + len
));
3552 * copy the variable value to the result
3555 strcpy((res_str
+ res_str_len
+ 1), p1
);
3558 * mark the replaced text to be accepted as
3561 res_str
[res_str_len
] = SUBSTED_VAR_SYMBOL
;
3562 res_str
[res_str_len
+ 1 + strlen(p1
)] =
3566 * copy the variable value to the result
3569 strcpy((res_str
+ res_str_len
), p1
);
3573 *p
= SPECIAL_VAR_SYMBOL
;
3578 res_str
= xrealloc(res_str
, (1 + res_str_len
+ strlen(inp
)));
3579 strcpy((res_str
+ res_str_len
), inp
);
3580 while ((p
= strchr(res_str
, '\n'))) {
3584 return (res_str
== NULL
) ? inp
: res_str
;
3587 static char **make_list_in(char **inp
, char *name
)
3590 int name_len
= strlen(name
);
3595 /* create list of variable values */
3596 list
= xmalloc(sizeof(*list
));
3597 for (i
= 0; inp
[i
]; i
++) {
3598 p3
= insert_var_value(inp
[i
]);
3605 if ((p2
= strchr(p1
, ' '))) {
3611 /* we use n + 2 in realloc for list,because we add
3612 * new element and then we will add NULL element */
3613 list
= xrealloc(list
, sizeof(*list
) * (n
+ 2));
3614 list
[n
] = xmalloc(2 + name_len
+ len
);
3615 strcpy(list
[n
], name
);
3616 strcat(list
[n
], "=");
3617 strncat(list
[n
], p1
, len
);
3618 list
[n
++][name_len
+ len
+ 1] = '\0';
3621 if (p3
!= inp
[i
]) free(p3
);
3628 * Make new string for parser
3629 * inp - array of argument strings to flatten
3630 * nonnull - indicates argument was quoted when originally parsed
3632 static char *make_string(char **inp
, int *nonnull
)
3641 noeval_str
= get_local_var("HUSH_NO_EVAL");
3642 if (noeval_str
!= NULL
&& *noeval_str
!= '0' && *noeval_str
!= '\0')
3644 for (n
= 0; inp
[n
]; n
++) {
3645 p
= insert_var_value_sub(inp
[n
], noeval
);
3646 str
= xrealloc(str
, (len
+ strlen(p
) + (2 * nonnull
[n
])));
3657 len
= strlen(str
) + 3;
3658 if (p
!= inp
[n
]) free(p
);
3661 *(str
+ len
) = '\n';
3662 *(str
+ len
+ 1) = '\0';
3667 static int do_showvar(struct cmd_tbl
*cmdtp
, int flag
, int argc
,
3672 struct variables
*cur
;
3674 if (argc
== 1) { /* Print all env variables */
3675 for (cur
= top_vars
; cur
; cur
= cur
->next
) {
3676 printf ("%s=%s\n", cur
->name
, cur
->value
);
3678 puts ("\n ** Abort\n");
3684 for (i
= 1; i
< argc
; ++i
) { /* print single env variables */
3685 char *name
= argv
[i
];
3688 for (cur
= top_vars
; cur
; cur
= cur
->next
) {
3689 if(strcmp (cur
->name
, name
) == 0) {
3691 printf ("%s=%s\n", cur
->name
, cur
->value
);
3694 puts ("\n ** Abort\n");
3699 printf ("## Error: \"%s\" not defined\n", name
);
3707 showvar
, CONFIG_SYS_MAXARGS
, 1, do_showvar
,
3708 "print local hushshell variables",
3709 "\n - print values of all hushshell variables\n"
3710 "showvar name ...\n"
3711 " - print value of hushshell variable 'name'"
3715 /****************************************************************************/