diff options
author | Denis Vlasenko <vda.linux@googlemail.com> | 2007-11-23 23:28:55 +0000 |
---|---|---|
committer | Denis Vlasenko <vda.linux@googlemail.com> | 2007-11-23 23:28:55 +0000 |
commit | 96702ca945a8deac1f989584f2b25d1a16d14b72 (patch) | |
tree | 4de934d94214ca3b46dbce6de8dabe4517bd5247 /shell/lash_unused.c | |
parent | 8ec6ee47f1e70ff25518ad6455e68d45d7ce1b87 (diff) | |
download | busybox-w32-96702ca945a8deac1f989584f2b25d1a16d14b72.tar.gz busybox-w32-96702ca945a8deac1f989584f2b25d1a16d14b72.tar.bz2 busybox-w32-96702ca945a8deac1f989584f2b25d1a16d14b72.zip |
kill lash. "lash" builtin still exists, but it runs hush.
Diffstat (limited to 'shell/lash_unused.c')
-rw-r--r-- | shell/lash_unused.c | 1570 |
1 files changed, 1570 insertions, 0 deletions
diff --git a/shell/lash_unused.c b/shell/lash_unused.c new file mode 100644 index 000000000..781dfdb5a --- /dev/null +++ b/shell/lash_unused.c | |||
@@ -0,0 +1,1570 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * lash -- the BusyBox Lame-Ass SHell | ||
4 | * | ||
5 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> | ||
6 | * | ||
7 | * Based in part on ladsh.c by Michael K. Johnson and Erik W. Troan, which is | ||
8 | * under the following liberal license: "We have placed this source code in the | ||
9 | * public domain. Use it in any project, free or commercial." | ||
10 | * | ||
11 | * Licensed under the GPL v2 or later, see the file LICENSE in this tarball. | ||
12 | */ | ||
13 | |||
14 | /* This shell's parsing engine is officially at a dead-end. Future | ||
15 | * work shell work should be done using hush, msh, or ash. This is | ||
16 | * still a very useful, small shell -- it just don't need any more | ||
17 | * features beyond what it already has... | ||
18 | */ | ||
19 | |||
20 | //For debugging/development on the shell only... | ||
21 | //#define DEBUG_SHELL | ||
22 | |||
23 | #include <getopt.h> | ||
24 | #include <glob.h> | ||
25 | |||
26 | #include "busybox.h" /* for struct bb_applet */ | ||
27 | |||
28 | #define expand_t glob_t | ||
29 | |||
30 | /* Always enable for the moment... */ | ||
31 | #define CONFIG_LASH_PIPE_N_REDIRECTS | ||
32 | #define CONFIG_LASH_JOB_CONTROL | ||
33 | #define ENABLE_LASH_PIPE_N_REDIRECTS 1 | ||
34 | #define ENABLE_LASH_JOB_CONTROL 1 | ||
35 | |||
36 | |||
37 | enum { MAX_READ = 128 }; /* size of input buffer for 'read' builtin */ | ||
38 | #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n" | ||
39 | |||
40 | |||
41 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
42 | enum redir_type { REDIRECT_INPUT, REDIRECT_OVERWRITE, | ||
43 | REDIRECT_APPEND | ||
44 | }; | ||
45 | #endif | ||
46 | |||
47 | enum { | ||
48 | DEFAULT_CONTEXT = 0x1, | ||
49 | IF_TRUE_CONTEXT = 0x2, | ||
50 | IF_FALSE_CONTEXT = 0x4, | ||
51 | THEN_EXP_CONTEXT = 0x8, | ||
52 | ELSE_EXP_CONTEXT = 0x10 | ||
53 | }; | ||
54 | |||
55 | #define LASH_OPT_DONE (1) | ||
56 | #define LASH_OPT_SAW_QUOTE (2) | ||
57 | |||
58 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
59 | struct redir_struct { | ||
60 | enum redir_type type; /* type of redirection */ | ||
61 | int fd; /* file descriptor being redirected */ | ||
62 | char *filename; /* file to redirect fd to */ | ||
63 | }; | ||
64 | #endif | ||
65 | |||
66 | struct child_prog { | ||
67 | pid_t pid; /* 0 if exited */ | ||
68 | char **argv; /* program name and arguments */ | ||
69 | int num_redirects; /* elements in redirection array */ | ||
70 | int is_stopped; /* is the program currently running? */ | ||
71 | struct job *family; /* pointer back to the child's parent job */ | ||
72 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
73 | struct redir_struct *redirects; /* I/O redirects */ | ||
74 | #endif | ||
75 | }; | ||
76 | |||
77 | struct jobset { | ||
78 | struct job *head; /* head of list of running jobs */ | ||
79 | struct job *fg; /* current foreground job */ | ||
80 | }; | ||
81 | |||
82 | struct job { | ||
83 | int jobid; /* job number */ | ||
84 | int num_progs; /* total number of programs in job */ | ||
85 | int running_progs; /* number of programs running */ | ||
86 | char *text; /* name of job */ | ||
87 | char *cmdbuf; /* buffer various argv's point into */ | ||
88 | pid_t pgrp; /* process group ID for the job */ | ||
89 | struct child_prog *progs; /* array of programs in job */ | ||
90 | struct job *next; /* to track background commands */ | ||
91 | int stopped_progs; /* number of programs alive, but stopped */ | ||
92 | unsigned int job_context; /* bitmask defining current context */ | ||
93 | struct jobset *job_list; | ||
94 | }; | ||
95 | |||
96 | struct built_in_command { | ||
97 | const char *cmd; /* name */ | ||
98 | const char *descr; /* description */ | ||
99 | int (*function) (struct child_prog *); /* function ptr */ | ||
100 | }; | ||
101 | |||
102 | /* function prototypes for builtins */ | ||
103 | static int builtin_cd(struct child_prog *cmd); | ||
104 | static int builtin_exec(struct child_prog *cmd); | ||
105 | static int builtin_exit(struct child_prog *cmd); | ||
106 | static int builtin_fg_bg(struct child_prog *cmd); | ||
107 | static int builtin_help(struct child_prog *cmd); | ||
108 | static int builtin_jobs(struct child_prog *dummy); | ||
109 | static int builtin_pwd(struct child_prog *dummy); | ||
110 | static int builtin_export(struct child_prog *cmd); | ||
111 | static int builtin_source(struct child_prog *cmd); | ||
112 | static int builtin_unset(struct child_prog *cmd); | ||
113 | static int builtin_read(struct child_prog *cmd); | ||
114 | |||
115 | |||
116 | /* function prototypes for shell stuff */ | ||
117 | static void checkjobs(struct jobset *job_list); | ||
118 | static void remove_job(struct jobset *j_list, struct job *job); | ||
119 | static int get_command_bufsiz(FILE * source, char *command); | ||
120 | static int parse_command(char **command_ptr, struct job *job, int *inbg); | ||
121 | static int run_command(struct job *newjob, int inbg, int outpipe[2]); | ||
122 | static int pseudo_exec(struct child_prog *cmd) ATTRIBUTE_NORETURN; | ||
123 | static int busy_loop(FILE * input); | ||
124 | |||
125 | |||
126 | /* Table of built-in functions (these are non-forking builtins, meaning they | ||
127 | * can change global variables in the parent shell process but they will not | ||
128 | * work with pipes and redirects; 'unset foo | whatever' will not work) */ | ||
129 | static const struct built_in_command bltins[] = { | ||
130 | {"bg" , "Resume a job in the background", builtin_fg_bg}, | ||
131 | {"cd" , "Change working directory", builtin_cd}, | ||
132 | {"exec" , "Exec command, replacing this shell with the exec'd process", builtin_exec}, | ||
133 | {"exit" , "Exit from shell()", builtin_exit}, | ||
134 | {"fg" , "Bring job into the foreground", builtin_fg_bg}, | ||
135 | {"jobs" , "Lists the active jobs", builtin_jobs}, | ||
136 | {"export", "Set environment variable", builtin_export}, | ||
137 | {"unset" , "Unset environment variable", builtin_unset}, | ||
138 | {"read" , "Input environment variable", builtin_read}, | ||
139 | {"." , "Source-in and run commands in a file", builtin_source}, | ||
140 | /* These were "forked applets", but distinction was nuked */ | ||
141 | /* Original comment retained: */ | ||
142 | /* Table of forking built-in functions (things that fork cannot change global | ||
143 | * variables in the parent process, such as the current working directory) */ | ||
144 | {"pwd" , "Print current directory", builtin_pwd}, | ||
145 | {"help" , "List shell built-in commands", builtin_help}, | ||
146 | /* to do: add ulimit */ | ||
147 | }; | ||
148 | |||
149 | |||
150 | #define VEC_LAST(v) v[ARRAY_SIZE(v)-1] | ||
151 | |||
152 | |||
153 | static int shell_context; /* Type prompt trigger (PS1 or PS2) */ | ||
154 | |||
155 | |||
156 | /* Globals that are static to this file */ | ||
157 | static char *cwd; | ||
158 | static char *local_pending_command; | ||
159 | static struct jobset job_list = { NULL, NULL }; | ||
160 | static int global_argc; | ||
161 | static char **global_argv; | ||
162 | static llist_t *close_me_list; | ||
163 | static int last_return_code; | ||
164 | static int last_bg_pid; | ||
165 | static unsigned int last_jobid; | ||
166 | static int shell_terminal; | ||
167 | static const char *PS1; | ||
168 | static const char *PS2 = "> "; | ||
169 | |||
170 | |||
171 | #ifdef DEBUG_SHELL | ||
172 | static inline void debug_printf(const char *format, ...) | ||
173 | { | ||
174 | va_list args; | ||
175 | va_start(args, format); | ||
176 | vfprintf(stderr, format, args); | ||
177 | va_end(args); | ||
178 | } | ||
179 | #else | ||
180 | static inline void debug_printf(const char ATTRIBUTE_UNUSED *format, ...) { } | ||
181 | #endif | ||
182 | |||
183 | /* | ||
184 | Most builtins need access to the struct child_prog that has | ||
185 | their arguments, previously coded as cmd->progs[0]. That coding | ||
186 | can exhibit a bug, if the builtin is not the first command in | ||
187 | a pipeline: "echo foo | exec sort" will attempt to exec foo. | ||
188 | |||
189 | builtin previous use notes | ||
190 | ------ ----------------- --------- | ||
191 | cd cmd->progs[0] | ||
192 | exec cmd->progs[0] squashed bug: didn't look for applets or forking builtins | ||
193 | exit cmd->progs[0] | ||
194 | fg_bg cmd->progs[0], job_list->head, job_list->fg | ||
195 | help 0 | ||
196 | jobs job_list->head | ||
197 | pwd 0 | ||
198 | export cmd->progs[0] | ||
199 | source cmd->progs[0] | ||
200 | unset cmd->progs[0] | ||
201 | read cmd->progs[0] | ||
202 | |||
203 | I added "struct job *family;" to struct child_prog, | ||
204 | and switched API to builtin_foo(struct child_prog *child); | ||
205 | So cmd->text becomes child->family->text | ||
206 | cmd->job_context becomes child->family->job_context | ||
207 | cmd->progs[0] becomes *child | ||
208 | job_list becomes child->family->job_list | ||
209 | */ | ||
210 | |||
211 | |||
212 | static void update_cwd(void) | ||
213 | { | ||
214 | cwd = xrealloc_getcwd_or_warn(cwd); | ||
215 | if (!cwd) | ||
216 | cwd = xstrdup(bb_msg_unknown); | ||
217 | } | ||
218 | |||
219 | /* built-in 'cd <path>' handler */ | ||
220 | static int builtin_cd(struct child_prog *child) | ||
221 | { | ||
222 | char *newdir; | ||
223 | |||
224 | if (child->argv[1] == NULL) | ||
225 | newdir = getenv("HOME"); | ||
226 | else | ||
227 | newdir = child->argv[1]; | ||
228 | if (chdir(newdir)) { | ||
229 | bb_perror_msg("cd: %s", newdir); | ||
230 | return EXIT_FAILURE; | ||
231 | } | ||
232 | update_cwd(); | ||
233 | return EXIT_SUCCESS; | ||
234 | } | ||
235 | |||
236 | /* built-in 'exec' handler */ | ||
237 | static int builtin_exec(struct child_prog *child) | ||
238 | { | ||
239 | if (child->argv[1] == NULL) | ||
240 | return EXIT_SUCCESS; /* Really? */ | ||
241 | child->argv++; | ||
242 | while (close_me_list) | ||
243 | close((long)llist_pop(&close_me_list)); | ||
244 | pseudo_exec(child); | ||
245 | /* never returns */ | ||
246 | } | ||
247 | |||
248 | /* built-in 'exit' handler */ | ||
249 | static int builtin_exit(struct child_prog *child) | ||
250 | { | ||
251 | if (child->argv[1] == NULL) | ||
252 | exit(EXIT_SUCCESS); | ||
253 | |||
254 | exit(atoi(child->argv[1])); | ||
255 | } | ||
256 | |||
257 | /* built-in 'fg' and 'bg' handler */ | ||
258 | static int builtin_fg_bg(struct child_prog *child) | ||
259 | { | ||
260 | int i, jobnum; | ||
261 | struct job *job; | ||
262 | |||
263 | /* If they gave us no args, assume they want the last backgrounded task */ | ||
264 | if (!child->argv[1]) { | ||
265 | for (job = child->family->job_list->head; job; job = job->next) { | ||
266 | if (job->jobid == last_jobid) { | ||
267 | goto found; | ||
268 | } | ||
269 | } | ||
270 | bb_error_msg("%s: no current job", child->argv[0]); | ||
271 | return EXIT_FAILURE; | ||
272 | } | ||
273 | if (sscanf(child->argv[1], "%%%d", &jobnum) != 1) { | ||
274 | bb_error_msg(bb_msg_invalid_arg, child->argv[1], child->argv[0]); | ||
275 | return EXIT_FAILURE; | ||
276 | } | ||
277 | for (job = child->family->job_list->head; job; job = job->next) { | ||
278 | if (job->jobid == jobnum) { | ||
279 | goto found; | ||
280 | } | ||
281 | } | ||
282 | bb_error_msg("%s: %d: no such job", child->argv[0], jobnum); | ||
283 | return EXIT_FAILURE; | ||
284 | found: | ||
285 | if (*child->argv[0] == 'f') { | ||
286 | /* Put the job into the foreground. */ | ||
287 | tcsetpgrp(shell_terminal, job->pgrp); | ||
288 | |||
289 | child->family->job_list->fg = job; | ||
290 | } | ||
291 | |||
292 | /* Restart the processes in the job */ | ||
293 | for (i = 0; i < job->num_progs; i++) | ||
294 | job->progs[i].is_stopped = 0; | ||
295 | |||
296 | job->stopped_progs = 0; | ||
297 | |||
298 | i = kill(- job->pgrp, SIGCONT); | ||
299 | if (i < 0) { | ||
300 | if (errno == ESRCH) { | ||
301 | remove_job(&job_list, job); | ||
302 | } else { | ||
303 | bb_perror_msg("kill (SIGCONT)"); | ||
304 | } | ||
305 | } | ||
306 | |||
307 | return EXIT_SUCCESS; | ||
308 | } | ||
309 | |||
310 | /* built-in 'help' handler */ | ||
311 | static int builtin_help(struct child_prog ATTRIBUTE_UNUSED *dummy) | ||
312 | { | ||
313 | const struct built_in_command *x; | ||
314 | |||
315 | printf("\nBuilt-in commands:\n" | ||
316 | "-------------------\n"); | ||
317 | for (x = bltins; x <= &VEC_LAST(bltins); x++) { | ||
318 | if (x->descr == NULL) | ||
319 | continue; | ||
320 | printf("%s\t%s\n", x->cmd, x->descr); | ||
321 | } | ||
322 | bb_putchar('\n'); | ||
323 | return EXIT_SUCCESS; | ||
324 | } | ||
325 | |||
326 | /* built-in 'jobs' handler */ | ||
327 | static int builtin_jobs(struct child_prog *child) | ||
328 | { | ||
329 | struct job *job; | ||
330 | const char *status_string; | ||
331 | |||
332 | for (job = child->family->job_list->head; job; job = job->next) { | ||
333 | if (job->running_progs == job->stopped_progs) | ||
334 | status_string = "Stopped"; | ||
335 | else | ||
336 | status_string = "Running"; | ||
337 | |||
338 | printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->text); | ||
339 | } | ||
340 | return EXIT_SUCCESS; | ||
341 | } | ||
342 | |||
343 | |||
344 | /* built-in 'pwd' handler */ | ||
345 | static int builtin_pwd(struct child_prog ATTRIBUTE_UNUSED *dummy) | ||
346 | { | ||
347 | update_cwd(); | ||
348 | puts(cwd); | ||
349 | return EXIT_SUCCESS; | ||
350 | } | ||
351 | |||
352 | /* built-in 'export VAR=value' handler */ | ||
353 | static int builtin_export(struct child_prog *child) | ||
354 | { | ||
355 | int res; | ||
356 | char *v = child->argv[1]; | ||
357 | |||
358 | if (v == NULL) { | ||
359 | char **e; | ||
360 | for (e = environ; *e; e++) { | ||
361 | puts(*e); | ||
362 | } | ||
363 | return 0; | ||
364 | } | ||
365 | res = putenv(v); | ||
366 | if (res) | ||
367 | bb_perror_msg("export"); | ||
368 | #if ENABLE_FEATURE_EDITING_FANCY_PROMPT | ||
369 | if (strncmp(v, "PS1=", 4) == 0) | ||
370 | PS1 = getenv("PS1"); | ||
371 | #endif | ||
372 | |||
373 | #if ENABLE_LOCALE_SUPPORT | ||
374 | // TODO: why getenv? "" would be just as good... | ||
375 | if (strncmp(v, "LC_ALL=", 7) == 0) | ||
376 | setlocale(LC_ALL, getenv("LC_ALL")); | ||
377 | if (strncmp(v, "LC_CTYPE=", 9) == 0) | ||
378 | setlocale(LC_CTYPE, getenv("LC_CTYPE")); | ||
379 | #endif | ||
380 | |||
381 | return res; | ||
382 | } | ||
383 | |||
384 | /* built-in 'read VAR' handler */ | ||
385 | static int builtin_read(struct child_prog *child) | ||
386 | { | ||
387 | int res = 0, len; | ||
388 | char *s; | ||
389 | char string[MAX_READ]; | ||
390 | |||
391 | if (child->argv[1]) { | ||
392 | /* argument (VAR) given: put "VAR=" into buffer */ | ||
393 | safe_strncpy(string, child->argv[1], MAX_READ-1); | ||
394 | len = strlen(string); | ||
395 | string[len++] = '='; | ||
396 | string[len] = '\0'; | ||
397 | fgets(&string[len], sizeof(string) - len, stdin); /* read string */ | ||
398 | res = strlen(string); | ||
399 | if (res > len) | ||
400 | string[--res] = '\0'; /* chomp trailing newline */ | ||
401 | /* | ||
402 | ** string should now contain "VAR=<value>" | ||
403 | ** copy it (putenv() won't do that, so we must make sure | ||
404 | ** the string resides in a static buffer!) | ||
405 | */ | ||
406 | res = -1; | ||
407 | s = strdup(string); | ||
408 | if (s) | ||
409 | res = putenv(s); | ||
410 | if (res) | ||
411 | bb_perror_msg("read"); | ||
412 | } else | ||
413 | fgets(string, sizeof(string), stdin); | ||
414 | |||
415 | return res; | ||
416 | } | ||
417 | |||
418 | /* Built-in '.' handler (read-in and execute commands from file) */ | ||
419 | static int builtin_source(struct child_prog *child) | ||
420 | { | ||
421 | FILE *input; | ||
422 | int status; | ||
423 | |||
424 | input = fopen_or_warn(child->argv[1], "r"); | ||
425 | if (!input) { | ||
426 | return EXIT_FAILURE; | ||
427 | } | ||
428 | |||
429 | llist_add_to(&close_me_list, (void *)(long)fileno(input)); | ||
430 | /* Now run the file */ | ||
431 | status = busy_loop(input); | ||
432 | fclose(input); | ||
433 | llist_pop(&close_me_list); | ||
434 | return status; | ||
435 | } | ||
436 | |||
437 | /* built-in 'unset VAR' handler */ | ||
438 | static int builtin_unset(struct child_prog *child) | ||
439 | { | ||
440 | if (child->argv[1] == NULL) { | ||
441 | printf(bb_msg_requires_arg, "unset"); | ||
442 | return EXIT_FAILURE; | ||
443 | } | ||
444 | unsetenv(child->argv[1]); | ||
445 | return EXIT_SUCCESS; | ||
446 | } | ||
447 | |||
448 | #if ENABLE_LASH_JOB_CONTROL | ||
449 | /* free up all memory from a job */ | ||
450 | static void free_job(struct job *cmd) | ||
451 | { | ||
452 | int i; | ||
453 | struct jobset *keep; | ||
454 | |||
455 | for (i = 0; i < cmd->num_progs; i++) { | ||
456 | free(cmd->progs[i].argv); | ||
457 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
458 | free(cmd->progs[i].redirects); | ||
459 | #endif | ||
460 | } | ||
461 | free(cmd->progs); | ||
462 | free(cmd->text); | ||
463 | free(cmd->cmdbuf); | ||
464 | keep = cmd->job_list; | ||
465 | memset(cmd, 0, sizeof(struct job)); | ||
466 | cmd->job_list = keep; | ||
467 | } | ||
468 | |||
469 | /* remove a job from a jobset */ | ||
470 | static void remove_job(struct jobset *j_list, struct job *job) | ||
471 | { | ||
472 | struct job *prevjob; | ||
473 | |||
474 | free_job(job); | ||
475 | if (job == j_list->head) { | ||
476 | j_list->head = job->next; | ||
477 | } else { | ||
478 | prevjob = j_list->head; | ||
479 | while (prevjob->next != job) | ||
480 | prevjob = prevjob->next; | ||
481 | prevjob->next = job->next; | ||
482 | } | ||
483 | |||
484 | if (j_list->head) | ||
485 | last_jobid = j_list->head->jobid; | ||
486 | else | ||
487 | last_jobid = 0; | ||
488 | |||
489 | free(job); | ||
490 | } | ||
491 | |||
492 | /* Checks to see if any background processes have exited -- if they | ||
493 | have, figure out why and see if a job has completed */ | ||
494 | static void checkjobs(struct jobset *j_list) | ||
495 | { | ||
496 | struct job *job; | ||
497 | pid_t childpid; | ||
498 | int status; | ||
499 | int prognum = 0; | ||
500 | |||
501 | while ((childpid = waitpid(-1, &status, WNOHANG | WUNTRACED)) > 0) { | ||
502 | for (job = j_list->head; job; job = job->next) { | ||
503 | prognum = 0; | ||
504 | while (prognum < job->num_progs && | ||
505 | job->progs[prognum].pid != childpid) prognum++; | ||
506 | if (prognum < job->num_progs) | ||
507 | break; | ||
508 | } | ||
509 | |||
510 | /* This happens on backticked commands */ | ||
511 | if (job == NULL) | ||
512 | return; | ||
513 | |||
514 | if (WIFEXITED(status) || WIFSIGNALED(status)) { | ||
515 | /* child exited */ | ||
516 | job->running_progs--; | ||
517 | job->progs[prognum].pid = 0; | ||
518 | |||
519 | if (!job->running_progs) { | ||
520 | printf(JOB_STATUS_FORMAT, job->jobid, "Done", job->text); | ||
521 | last_jobid = 0; | ||
522 | remove_job(j_list, job); | ||
523 | } | ||
524 | } else { | ||
525 | /* child stopped */ | ||
526 | job->stopped_progs++; | ||
527 | job->progs[prognum].is_stopped = 1; | ||
528 | } | ||
529 | } | ||
530 | |||
531 | if (childpid == -1 && errno != ECHILD) | ||
532 | bb_perror_msg("waitpid"); | ||
533 | } | ||
534 | #else | ||
535 | static void checkjobs(struct jobset *j_list) | ||
536 | { | ||
537 | } | ||
538 | static void free_job(struct job *cmd) | ||
539 | { | ||
540 | } | ||
541 | static void remove_job(struct jobset *j_list, struct job *job) | ||
542 | { | ||
543 | } | ||
544 | #endif | ||
545 | |||
546 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
547 | /* squirrel != NULL means we squirrel away copies of stdin, stdout, | ||
548 | * and stderr if they are redirected. */ | ||
549 | static int setup_redirects(struct child_prog *prog, int squirrel[]) | ||
550 | { | ||
551 | int i; | ||
552 | int openfd; | ||
553 | int mode = O_RDONLY; | ||
554 | struct redir_struct *redir = prog->redirects; | ||
555 | |||
556 | for (i = 0; i < prog->num_redirects; i++, redir++) { | ||
557 | switch (redir->type) { | ||
558 | case REDIRECT_INPUT: | ||
559 | mode = O_RDONLY; | ||
560 | break; | ||
561 | case REDIRECT_OVERWRITE: | ||
562 | mode = O_WRONLY | O_CREAT | O_TRUNC; | ||
563 | break; | ||
564 | case REDIRECT_APPEND: | ||
565 | mode = O_WRONLY | O_CREAT | O_APPEND; | ||
566 | break; | ||
567 | } | ||
568 | |||
569 | openfd = open3_or_warn(redir->filename, mode, 0666); | ||
570 | if (openfd < 0) { | ||
571 | /* this could get lost if stderr has been redirected, but | ||
572 | bash and ash both lose it as well (though zsh doesn't!) */ | ||
573 | return 1; | ||
574 | } | ||
575 | |||
576 | if (openfd != redir->fd) { | ||
577 | if (squirrel && redir->fd < 3) { | ||
578 | squirrel[redir->fd] = dup(redir->fd); | ||
579 | close_on_exec_on(squirrel[redir->fd]); | ||
580 | } | ||
581 | dup2(openfd, redir->fd); | ||
582 | close(openfd); | ||
583 | } | ||
584 | } | ||
585 | |||
586 | return 0; | ||
587 | } | ||
588 | |||
589 | static void restore_redirects(int squirrel[]) | ||
590 | { | ||
591 | int i, fd; | ||
592 | for (i = 0; i < 3; i++) { | ||
593 | fd = squirrel[i]; | ||
594 | if (fd != -1) { | ||
595 | /* No error checking. I sure wouldn't know what | ||
596 | * to do with an error if I found one! */ | ||
597 | dup2(fd, i); | ||
598 | close(fd); | ||
599 | } | ||
600 | } | ||
601 | } | ||
602 | #else | ||
603 | static inline int setup_redirects(struct child_prog *prog, int squirrel[]) | ||
604 | { | ||
605 | return 0; | ||
606 | } | ||
607 | static inline void restore_redirects(int squirrel[]) | ||
608 | { | ||
609 | } | ||
610 | #endif | ||
611 | |||
612 | static inline void cmdedit_set_initial_prompt(void) | ||
613 | { | ||
614 | #if !ENABLE_FEATURE_EDITING_FANCY_PROMPT | ||
615 | PS1 = NULL; | ||
616 | #else | ||
617 | PS1 = getenv("PS1"); | ||
618 | if (PS1 == 0) | ||
619 | PS1 = "\\w \\$ "; | ||
620 | #endif | ||
621 | } | ||
622 | |||
623 | static inline const char* setup_prompt_string(void) | ||
624 | { | ||
625 | #if !ENABLE_FEATURE_EDITING_FANCY_PROMPT | ||
626 | /* Set up the prompt */ | ||
627 | if (shell_context == 0) { | ||
628 | char *ns; | ||
629 | free((char*)PS1); | ||
630 | ns = xmalloc(strlen(cwd)+4); | ||
631 | sprintf(ns, "%s %c ", cwd, (geteuid() != 0) ? '$': '#'); | ||
632 | PS1 = ns; | ||
633 | return ns; | ||
634 | } else { | ||
635 | return PS2; | ||
636 | } | ||
637 | #else | ||
638 | return (shell_context == 0)? PS1 : PS2; | ||
639 | #endif | ||
640 | } | ||
641 | |||
642 | #if ENABLE_FEATURE_EDITING | ||
643 | static line_input_t *line_input_state; | ||
644 | #endif | ||
645 | |||
646 | static int get_command_bufsiz(FILE * source, char *command) | ||
647 | { | ||
648 | const char *prompt_str; | ||
649 | |||
650 | if (source == NULL) { | ||
651 | if (local_pending_command) { | ||
652 | /* a command specified (-c option): return it & mark it done */ | ||
653 | strncpy(command, local_pending_command, BUFSIZ); | ||
654 | local_pending_command = NULL; | ||
655 | return 0; | ||
656 | } | ||
657 | return 1; | ||
658 | } | ||
659 | |||
660 | if (source == stdin) { | ||
661 | prompt_str = setup_prompt_string(); | ||
662 | |||
663 | #if ENABLE_FEATURE_EDITING | ||
664 | /* | ||
665 | ** enable command line editing only while a command line | ||
666 | ** is actually being read; otherwise, we'll end up bequeathing | ||
667 | ** atexit() handlers and other unwanted stuff to our | ||
668 | ** child processes (rob@sysgo.de) | ||
669 | */ | ||
670 | read_line_input(prompt_str, command, BUFSIZ, line_input_state); | ||
671 | return 0; | ||
672 | #else | ||
673 | fputs(prompt_str, stdout); | ||
674 | #endif | ||
675 | } | ||
676 | |||
677 | if (!fgets(command, BUFSIZ - 2, source)) { | ||
678 | if (source == stdin) | ||
679 | bb_putchar('\n'); | ||
680 | return 1; | ||
681 | } | ||
682 | |||
683 | return 0; | ||
684 | } | ||
685 | |||
686 | static char * strsep_space(char *string, int * ix) | ||
687 | { | ||
688 | /* Short circuit the trivial case */ | ||
689 | if (!string || ! string[*ix]) | ||
690 | return NULL; | ||
691 | |||
692 | /* Find the end of the token. */ | ||
693 | while (string[*ix] && !isspace(string[*ix]) ) { | ||
694 | (*ix)++; | ||
695 | } | ||
696 | |||
697 | /* Find the end of any whitespace trailing behind | ||
698 | * the token and let that be part of the token */ | ||
699 | while (string[*ix] && (isspace)(string[*ix]) ) { | ||
700 | (*ix)++; | ||
701 | } | ||
702 | |||
703 | if (!*ix) { | ||
704 | /* Nothing useful was found */ | ||
705 | return NULL; | ||
706 | } | ||
707 | |||
708 | return xstrndup(string, *ix); | ||
709 | } | ||
710 | |||
711 | static int expand_arguments(char *command) | ||
712 | { | ||
713 | static const char out_of_space[] ALIGN1 = "out of space during expansion"; | ||
714 | |||
715 | int total_length = 0, length, i, retval, ix = 0; | ||
716 | expand_t expand_result; | ||
717 | char *tmpcmd, *cmd, *cmd_copy; | ||
718 | char *src, *dst, *var; | ||
719 | int flags = GLOB_NOCHECK | ||
720 | #ifdef GLOB_BRACE | ||
721 | | GLOB_BRACE | ||
722 | #endif | ||
723 | #ifdef GLOB_TILDE | ||
724 | | GLOB_TILDE | ||
725 | #endif | ||
726 | ; | ||
727 | |||
728 | /* get rid of the terminating \n */ | ||
729 | chomp(command); | ||
730 | |||
731 | /* Fix up escape sequences to be the Real Thing(tm) */ | ||
732 | while (command && command[ix]) { | ||
733 | if (command[ix] == '\\') { | ||
734 | const char *tmp = command+ix+1; | ||
735 | command[ix] = bb_process_escape_sequence( &tmp ); | ||
736 | memmove(command+ix + 1, tmp, strlen(tmp)+1); | ||
737 | } | ||
738 | ix++; | ||
739 | } | ||
740 | /* Use glob and then fixup environment variables and such */ | ||
741 | |||
742 | /* It turns out that glob is very stupid. We have to feed it one word at a | ||
743 | * time since it can't cope with a full string. Here we convert command | ||
744 | * (char*) into cmd (char**, one word per string) */ | ||
745 | |||
746 | /* We need a clean copy, so strsep can mess up the copy while | ||
747 | * we write stuff into the original (in a minute) */ | ||
748 | cmd = cmd_copy = xstrdup(command); | ||
749 | *command = '\0'; | ||
750 | for (ix = 0, tmpcmd = cmd; | ||
751 | (tmpcmd = strsep_space(cmd, &ix)) != NULL; cmd += ix, ix = 0) { | ||
752 | if (*tmpcmd == '\0') | ||
753 | break; | ||
754 | /* we need to trim() the result for glob! */ | ||
755 | trim(tmpcmd); | ||
756 | retval = glob(tmpcmd, flags, NULL, &expand_result); | ||
757 | free(tmpcmd); /* Free mem allocated by strsep_space */ | ||
758 | if (retval == GLOB_NOSPACE) { | ||
759 | /* Mem may have been allocated... */ | ||
760 | globfree(&expand_result); | ||
761 | bb_error_msg(out_of_space); | ||
762 | return FALSE; | ||
763 | } else if (retval != 0) { | ||
764 | /* Some other error. GLOB_NOMATCH shouldn't | ||
765 | * happen because of the GLOB_NOCHECK flag in | ||
766 | * the glob call. */ | ||
767 | bb_error_msg("syntax error"); | ||
768 | return FALSE; | ||
769 | } else { | ||
770 | /* Convert from char** (one word per string) to a simple char*, | ||
771 | * but don't overflow command which is BUFSIZ in length */ | ||
772 | for (i = 0; i < expand_result.gl_pathc; i++) { | ||
773 | length = strlen(expand_result.gl_pathv[i]); | ||
774 | if (total_length+length+1 >= BUFSIZ) { | ||
775 | bb_error_msg(out_of_space); | ||
776 | return FALSE; | ||
777 | } | ||
778 | strcat(command+total_length, " "); | ||
779 | total_length += 1; | ||
780 | strcat(command+total_length, expand_result.gl_pathv[i]); | ||
781 | total_length += length; | ||
782 | } | ||
783 | globfree(&expand_result); | ||
784 | } | ||
785 | } | ||
786 | free(cmd_copy); | ||
787 | trim(command); | ||
788 | |||
789 | /* Now do the shell variable substitutions which | ||
790 | * wordexp can't do for us, namely $? and $! */ | ||
791 | src = command; | ||
792 | while ((dst = strchr(src,'$')) != NULL) { | ||
793 | var = NULL; | ||
794 | switch (*(dst+1)) { | ||
795 | case '?': | ||
796 | var = itoa(last_return_code); | ||
797 | break; | ||
798 | case '!': | ||
799 | if (last_bg_pid == -1) | ||
800 | *var = '\0'; | ||
801 | else | ||
802 | var = itoa(last_bg_pid); | ||
803 | break; | ||
804 | /* Everything else like $$, $#, $[0-9], etc. should all be | ||
805 | * expanded by wordexp(), so we can in theory skip that stuff | ||
806 | * here, but just to be on the safe side (i.e., since uClibc | ||
807 | * wordexp doesn't do this stuff yet), lets leave it in for | ||
808 | * now. */ | ||
809 | case '$': | ||
810 | var = itoa(getpid()); | ||
811 | break; | ||
812 | case '#': | ||
813 | var = itoa(global_argc - 1); | ||
814 | break; | ||
815 | case '0':case '1':case '2':case '3':case '4': | ||
816 | case '5':case '6':case '7':case '8':case '9': | ||
817 | { | ||
818 | int ixx = *(dst+1)-48+1; | ||
819 | if (ixx >= global_argc) { | ||
820 | var = '\0'; | ||
821 | } else { | ||
822 | var = global_argv[ixx]; | ||
823 | } | ||
824 | } | ||
825 | break; | ||
826 | |||
827 | } | ||
828 | if (var) { | ||
829 | /* a single character construction was found, and | ||
830 | * already handled in the case statement */ | ||
831 | src = dst + 2; | ||
832 | } else { | ||
833 | /* Looks like an environment variable */ | ||
834 | char delim_hold; | ||
835 | int num_skip_chars = 0; | ||
836 | int dstlen = strlen(dst); | ||
837 | /* Is this a ${foo} type variable? */ | ||
838 | if (dstlen >= 2 && *(dst+1) == '{') { | ||
839 | src = strchr(dst+1, '}'); | ||
840 | num_skip_chars = 1; | ||
841 | } else { | ||
842 | src = dst + 1; | ||
843 | while ((isalnum)(*src) || *src == '_') src++; | ||
844 | } | ||
845 | if (src == NULL) { | ||
846 | src = dst+dstlen; | ||
847 | } | ||
848 | delim_hold = *src; | ||
849 | *src = '\0'; /* temporary */ | ||
850 | var = getenv(dst + 1 + num_skip_chars); | ||
851 | *src = delim_hold; | ||
852 | src += num_skip_chars; | ||
853 | } | ||
854 | if (var == NULL) { | ||
855 | /* Seems we got an un-expandable variable. So delete it. */ | ||
856 | var = (char*)""; | ||
857 | } | ||
858 | { | ||
859 | int subst_len = strlen(var); | ||
860 | int trail_len = strlen(src); | ||
861 | if (dst+subst_len+trail_len >= command+BUFSIZ) { | ||
862 | bb_error_msg(out_of_space); | ||
863 | return FALSE; | ||
864 | } | ||
865 | /* Move stuff to the end of the string to accommodate | ||
866 | * filling the created gap with the new stuff */ | ||
867 | memmove(dst+subst_len, src, trail_len+1); | ||
868 | /* Now copy in the new stuff */ | ||
869 | memcpy(dst, var, subst_len); | ||
870 | src = dst+subst_len; | ||
871 | } | ||
872 | } | ||
873 | |||
874 | return TRUE; | ||
875 | } | ||
876 | |||
877 | /* Return cmd->num_progs as 0 if no command is present (e.g. an empty | ||
878 | line). If a valid command is found, command_ptr is set to point to | ||
879 | the beginning of the next command (if the original command had more | ||
880 | then one job associated with it) or NULL if no more commands are | ||
881 | present. */ | ||
882 | static int parse_command(char **command_ptr, struct job *job, int *inbg) | ||
883 | { | ||
884 | char *command; | ||
885 | char *return_command = NULL; | ||
886 | char *src, *buf; | ||
887 | int argc_l; | ||
888 | int flag; | ||
889 | int argv_alloced; | ||
890 | char quote = '\0'; | ||
891 | struct child_prog *prog; | ||
892 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
893 | int i; | ||
894 | char *chptr; | ||
895 | #endif | ||
896 | |||
897 | /* skip leading white space */ | ||
898 | *command_ptr = skip_whitespace(*command_ptr); | ||
899 | |||
900 | /* this handles empty lines or leading '#' characters */ | ||
901 | if (!**command_ptr || (**command_ptr == '#')) { | ||
902 | job->num_progs = 0; | ||
903 | return 0; | ||
904 | } | ||
905 | |||
906 | *inbg = 0; | ||
907 | job->num_progs = 1; | ||
908 | job->progs = xmalloc(sizeof(*job->progs)); | ||
909 | |||
910 | /* We set the argv elements to point inside of this string. The | ||
911 | memory is freed by free_job(). Allocate twice the original | ||
912 | length in case we need to quote every single character. | ||
913 | |||
914 | Getting clean memory relieves us of the task of NULL | ||
915 | terminating things and makes the rest of this look a bit | ||
916 | cleaner (though it is, admittedly, a tad less efficient) */ | ||
917 | job->cmdbuf = command = xzalloc(2*strlen(*command_ptr) + 1); | ||
918 | job->text = NULL; | ||
919 | |||
920 | prog = job->progs; | ||
921 | prog->num_redirects = 0; | ||
922 | prog->is_stopped = 0; | ||
923 | prog->family = job; | ||
924 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
925 | prog->redirects = NULL; | ||
926 | #endif | ||
927 | |||
928 | argv_alloced = 5; | ||
929 | prog->argv = xmalloc(sizeof(*prog->argv) * argv_alloced); | ||
930 | prog->argv[0] = job->cmdbuf; | ||
931 | |||
932 | flag = argc_l = 0; | ||
933 | buf = command; | ||
934 | src = *command_ptr; | ||
935 | while (*src && !(flag & LASH_OPT_DONE)) { | ||
936 | if (quote == *src) { | ||
937 | quote = '\0'; | ||
938 | } else if (quote) { | ||
939 | if (*src == '\\') { | ||
940 | src++; | ||
941 | if (!*src) { | ||
942 | bb_error_msg("character expected after \\"); | ||
943 | free_job(job); | ||
944 | return 1; | ||
945 | } | ||
946 | |||
947 | /* in shell, "\'" should yield \' */ | ||
948 | if (*src != quote) { | ||
949 | *buf++ = '\\'; | ||
950 | *buf++ = '\\'; | ||
951 | } | ||
952 | } else if (*src == '*' || *src == '?' || *src == '[' || | ||
953 | *src == ']') *buf++ = '\\'; | ||
954 | *buf++ = *src; | ||
955 | } else if (isspace(*src)) { | ||
956 | if (*prog->argv[argc_l] || (flag & LASH_OPT_SAW_QUOTE)) { | ||
957 | buf++, argc_l++; | ||
958 | /* +1 here leaves room for the NULL which ends argv */ | ||
959 | if ((argc_l + 1) == argv_alloced) { | ||
960 | argv_alloced += 5; | ||
961 | prog->argv = xrealloc(prog->argv, | ||
962 | sizeof(*prog->argv) * argv_alloced); | ||
963 | } | ||
964 | prog->argv[argc_l] = buf; | ||
965 | flag ^= LASH_OPT_SAW_QUOTE; | ||
966 | } | ||
967 | } else | ||
968 | switch (*src) { | ||
969 | case '"': | ||
970 | case '\'': | ||
971 | quote = *src; | ||
972 | flag |= LASH_OPT_SAW_QUOTE; | ||
973 | break; | ||
974 | |||
975 | case '#': /* comment */ | ||
976 | if (*(src-1)== '$') | ||
977 | *buf++ = *src; | ||
978 | else | ||
979 | flag |= LASH_OPT_DONE; | ||
980 | break; | ||
981 | |||
982 | #if ENABLE_LASH_PIPE_N_REDIRECTS | ||
983 | case '>': /* redirects */ | ||
984 | case '<': | ||
985 | i = prog->num_redirects++; | ||
986 | prog->redirects = xrealloc(prog->redirects, | ||
987 | sizeof(*prog->redirects) * (i + 1)); | ||
988 | |||
989 | prog->redirects[i].fd = -1; | ||
990 | if (buf != prog->argv[argc_l]) { | ||
991 | /* the stuff before this character may be the file number | ||
992 | being redirected */ | ||
993 | prog->redirects[i].fd = | ||
994 | strtol(prog->argv[argc_l], &chptr, 10); | ||
995 | |||
996 | if (*chptr && *prog->argv[argc_l]) { | ||
997 | buf++, argc_l++; | ||
998 | prog->argv[argc_l] = buf; | ||
999 | } | ||
1000 | } | ||
1001 | |||
1002 | if (prog->redirects[i].fd == -1) { | ||
1003 | if (*src == '>') | ||
1004 | prog->redirects[i].fd = 1; | ||
1005 | else | ||
1006 | prog->redirects[i].fd = 0; | ||
1007 | } | ||
1008 | |||
1009 | if (*src++ == '>') { | ||
1010 | if (*src == '>') | ||
1011 | prog->redirects[i].type = | ||
1012 | REDIRECT_APPEND, src++; | ||
1013 | else | ||
1014 | prog->redirects[i].type = REDIRECT_OVERWRITE; | ||
1015 | } else { | ||
1016 | prog->redirects[i].type = REDIRECT_INPUT; | ||
1017 | } | ||
1018 | |||
1019 | /* This isn't POSIX sh compliant. Oh well. */ | ||
1020 | chptr = src; | ||
1021 | chptr = skip_whitespace(chptr); | ||
1022 | |||
1023 | if (!*chptr) { | ||
1024 | bb_error_msg("file name expected after %c", *(src-1)); | ||
1025 | free_job(job); | ||
1026 | job->num_progs = 0; | ||
1027 | return 1; | ||
1028 | } | ||
1029 | |||
1030 | prog->redirects[i].filename = buf; | ||
1031 | while (*chptr && !isspace(*chptr)) | ||
1032 | *buf++ = *chptr++; | ||
1033 | |||
1034 | src = chptr - 1; /* we src++ later */ | ||
1035 | prog->argv[argc_l] = ++buf; | ||
1036 | break; | ||
1037 | |||
1038 | case '|': /* pipe */ | ||
1039 | /* finish this command */ | ||
1040 | if (*prog->argv[argc_l] || flag & LASH_OPT_SAW_QUOTE) | ||
1041 | argc_l++; | ||
1042 | if (!argc_l) { | ||
1043 | goto empty_command_in_pipe; | ||
1044 | } | ||
1045 | prog->argv[argc_l] = NULL; | ||
1046 | |||
1047 | /* and start the next */ | ||
1048 | job->num_progs++; | ||
1049 | job->progs = xrealloc(job->progs, | ||
1050 | sizeof(*job->progs) * job->num_progs); | ||
1051 | prog = job->progs + (job->num_progs - 1); | ||
1052 | prog->num_redirects = 0; | ||
1053 | prog->redirects = NULL; | ||
1054 | prog->is_stopped = 0; | ||
1055 | prog->family = job; | ||
1056 | argc_l = 0; | ||
1057 | |||
1058 | argv_alloced = 5; | ||
1059 | prog->argv = xmalloc(sizeof(*prog->argv) * argv_alloced); | ||
1060 | prog->argv[0] = ++buf; | ||
1061 | |||
1062 | src++; | ||
1063 | src = skip_whitespace(src); | ||
1064 | |||
1065 | if (!*src) { | ||
1066 | empty_command_in_pipe: | ||
1067 | bb_error_msg("empty command in pipe"); | ||
1068 | free_job(job); | ||
1069 | job->num_progs = 0; | ||
1070 | return 1; | ||
1071 | } | ||
1072 | src--; /* we'll ++ it at the end of the loop */ | ||
1073 | |||
1074 | break; | ||
1075 | #endif | ||
1076 | |||
1077 | #if ENABLE_LASH_JOB_CONTROL | ||
1078 | case '&': /* background */ | ||
1079 | *inbg = 1; | ||
1080 | /* fallthrough */ | ||
1081 | #endif | ||
1082 | case ';': /* multiple commands */ | ||
1083 | flag |= LASH_OPT_DONE; | ||
1084 | return_command = *command_ptr + (src - *command_ptr) + 1; | ||
1085 | break; | ||
1086 | |||
1087 | case '\\': | ||
1088 | src++; | ||
1089 | if (!*src) { | ||
1090 | bb_error_msg("character expected after \\"); | ||
1091 | free_job(job); | ||
1092 | return 1; | ||
1093 | } | ||
1094 | if (*src == '*' || *src == '[' || *src == ']' | ||
1095 | || *src == '?') *buf++ = '\\'; | ||
1096 | /* fallthrough */ | ||
1097 | default: | ||
1098 | *buf++ = *src; | ||
1099 | } | ||
1100 | |||
1101 | src++; | ||
1102 | } | ||
1103 | |||
1104 | if (*prog->argv[argc_l] || flag & LASH_OPT_SAW_QUOTE) { | ||
1105 | argc_l++; | ||
1106 | } | ||
1107 | if (!argc_l) { | ||
1108 | free_job(job); | ||
1109 | return 0; | ||
1110 | } | ||
1111 | prog->argv[argc_l] = NULL; | ||
1112 | |||
1113 | if (!return_command) { | ||
1114 | job->text = xstrdup(*command_ptr); | ||
1115 | } else { | ||
1116 | /* This leaves any trailing spaces, which is a bit sloppy */ | ||
1117 | job->text = xstrndup(*command_ptr, return_command - *command_ptr); | ||
1118 | } | ||
1119 | |||
1120 | *command_ptr = return_command; | ||
1121 | |||
1122 | return 0; | ||
1123 | } | ||
1124 | |||
1125 | /* Run the child_prog, no matter what kind of command it uses. | ||
1126 | */ | ||
1127 | static int pseudo_exec(struct child_prog *child) | ||
1128 | { | ||
1129 | const struct built_in_command *x; | ||
1130 | |||
1131 | /* Check if the command matches any of the non-forking builtins. | ||
1132 | * Depending on context, this might be redundant. But it's | ||
1133 | * easier to waste a few CPU cycles than it is to figure out | ||
1134 | * if this is one of those cases. | ||
1135 | */ | ||
1136 | /* Check if the command matches any of the forking builtins. */ | ||
1137 | for (x = bltins; x <= &VEC_LAST(bltins); x++) { | ||
1138 | if (strcmp(child->argv[0], x->cmd) == 0) { | ||
1139 | _exit(x->function(child)); | ||
1140 | } | ||
1141 | } | ||
1142 | |||
1143 | /* Check if the command matches any busybox internal | ||
1144 | * commands ("applets") here. Following discussions from | ||
1145 | * November 2000 on busybox@busybox.net, don't use | ||
1146 | * bb_get_last_path_component_nostrip(). This way explicit | ||
1147 | * (with slashes) filenames will never be interpreted as an | ||
1148 | * applet, just like with builtins. This way the user can | ||
1149 | * override an applet with an explicit filename reference. | ||
1150 | * The only downside to this change is that an explicit | ||
1151 | * /bin/foo invocation will fork and exec /bin/foo, even if | ||
1152 | * /bin/foo is a symlink to busybox. | ||
1153 | */ | ||
1154 | if (ENABLE_FEATURE_SH_STANDALONE) { | ||
1155 | run_applet_and_exit(child->argv[0], child->argv); | ||
1156 | } | ||
1157 | |||
1158 | execvp(child->argv[0], child->argv); | ||
1159 | |||
1160 | /* Do not use bb_perror_msg_and_die() here, since we must not | ||
1161 | * call exit() but should call _exit() instead */ | ||
1162 | bb_simple_perror_msg(child->argv[0]); | ||
1163 | _exit(EXIT_FAILURE); | ||
1164 | } | ||
1165 | |||
1166 | static void insert_job(struct job *newjob, int inbg) | ||
1167 | { | ||
1168 | struct job *thejob; | ||
1169 | struct jobset *j_list = newjob->job_list; | ||
1170 | |||
1171 | /* find the ID for thejob to use */ | ||
1172 | newjob->jobid = 1; | ||
1173 | for (thejob = j_list->head; thejob; thejob = thejob->next) | ||
1174 | if (thejob->jobid >= newjob->jobid) | ||
1175 | newjob->jobid = thejob->jobid + 1; | ||
1176 | |||
1177 | /* add thejob to the list of running jobs */ | ||
1178 | if (!j_list->head) { | ||
1179 | thejob = j_list->head = xmalloc(sizeof(*thejob)); | ||
1180 | } else { | ||
1181 | for (thejob = j_list->head; thejob->next; thejob = thejob->next) /* nothing */; | ||
1182 | thejob->next = xmalloc(sizeof(*thejob)); | ||
1183 | thejob = thejob->next; | ||
1184 | } | ||
1185 | |||
1186 | *thejob = *newjob; /* physically copy the struct job */ | ||
1187 | thejob->next = NULL; | ||
1188 | thejob->running_progs = thejob->num_progs; | ||
1189 | thejob->stopped_progs = 0; | ||
1190 | |||
1191 | #if ENABLE_LASH_JOB_CONTROL | ||
1192 | if (inbg) { | ||
1193 | /* we don't wait for background thejobs to return -- append it | ||
1194 | to the list of backgrounded thejobs and leave it alone */ | ||
1195 | printf("[%d] %d\n", thejob->jobid, | ||
1196 | newjob->progs[newjob->num_progs - 1].pid); | ||
1197 | last_jobid = newjob->jobid; | ||
1198 | last_bg_pid = newjob->progs[newjob->num_progs - 1].pid; | ||
1199 | } else { | ||
1200 | newjob->job_list->fg = thejob; | ||
1201 | |||
1202 | /* move the new process group into the foreground */ | ||
1203 | /* Ignore errors since child could have already exited */ | ||
1204 | tcsetpgrp(shell_terminal, newjob->pgrp); | ||
1205 | } | ||
1206 | #endif | ||
1207 | } | ||
1208 | |||
1209 | static int run_command(struct job *newjob, int inbg, int outpipe[2]) | ||
1210 | { | ||
1211 | /* struct job *thejob; */ | ||
1212 | int i; | ||
1213 | int nextin, nextout; | ||
1214 | int pipefds[2]; /* pipefd[0] is for reading */ | ||
1215 | const struct built_in_command *x; | ||
1216 | struct child_prog *child; | ||
1217 | |||
1218 | nextin = 0; | ||
1219 | for (i = 0; i < newjob->num_progs; i++) { | ||
1220 | child = &(newjob->progs[i]); | ||
1221 | |||
1222 | nextout = 1; | ||
1223 | if ((i + 1) < newjob->num_progs) { | ||
1224 | xpipe(pipefds); | ||
1225 | nextout = pipefds[1]; | ||
1226 | } else if (outpipe[1] != -1) { | ||
1227 | nextout = outpipe[1]; | ||
1228 | } | ||
1229 | |||
1230 | /* Check if the command matches any non-forking builtins, | ||
1231 | * but only if this is a simple command. | ||
1232 | * Non-forking builtins within pipes have to fork anyway, | ||
1233 | * and are handled in pseudo_exec. "echo foo | read bar" | ||
1234 | * is doomed to failure, and doesn't work on bash, either. | ||
1235 | */ | ||
1236 | if (newjob->num_progs == 1) { | ||
1237 | int rcode; | ||
1238 | int squirrel[] = {-1, -1, -1}; | ||
1239 | |||
1240 | /* Check if the command sets an environment variable. */ | ||
1241 | if (strchr(child->argv[0], '=') != NULL) { | ||
1242 | child->argv[1] = child->argv[0]; | ||
1243 | return builtin_export(child); | ||
1244 | } | ||
1245 | |||
1246 | for (x = bltins; x <= &VEC_LAST(bltins); x++) { | ||
1247 | if (strcmp(child->argv[0], x->cmd) == 0) { | ||
1248 | setup_redirects(child, squirrel); | ||
1249 | rcode = x->function(child); | ||
1250 | restore_redirects(squirrel); | ||
1251 | return rcode; | ||
1252 | } | ||
1253 | } | ||
1254 | #if ENABLE_FEATURE_SH_STANDALONE | ||
1255 | { | ||
1256 | const struct bb_applet *a = find_applet_by_name(child->argv[i]); | ||
1257 | if (a && a->nofork) { | ||
1258 | setup_redirects(child, squirrel); | ||
1259 | rcode = run_nofork_applet(a, child->argv + i); | ||
1260 | restore_redirects(squirrel); | ||
1261 | return rcode; | ||
1262 | } | ||
1263 | } | ||
1264 | #endif | ||
1265 | } | ||
1266 | |||
1267 | #if BB_MMU | ||
1268 | child->pid = fork(); | ||
1269 | #else | ||
1270 | child->pid = vfork(); | ||
1271 | #endif | ||
1272 | if (!child->pid) { | ||
1273 | /* Set the handling for job control signals back to the default. */ | ||
1274 | signal(SIGINT, SIG_DFL); | ||
1275 | signal(SIGQUIT, SIG_DFL); | ||
1276 | signal(SIGTSTP, SIG_DFL); | ||
1277 | signal(SIGTTIN, SIG_DFL); | ||
1278 | signal(SIGTTOU, SIG_DFL); | ||
1279 | signal(SIGCHLD, SIG_DFL); | ||
1280 | |||
1281 | /* Close all open filehandles. */ | ||
1282 | while (close_me_list) | ||
1283 | close((long)llist_pop(&close_me_list)); | ||
1284 | |||
1285 | if (outpipe[1] != -1) { | ||
1286 | close(outpipe[0]); | ||
1287 | } | ||
1288 | if (nextin != 0) { | ||
1289 | dup2(nextin, 0); | ||
1290 | close(nextin); | ||
1291 | } | ||
1292 | |||
1293 | if (nextout != 1) { | ||
1294 | dup2(nextout, 1); | ||
1295 | dup2(nextout, 2); /* Really? */ | ||
1296 | close(nextout); | ||
1297 | close(pipefds[0]); | ||
1298 | } | ||
1299 | |||
1300 | /* explicit redirects override pipes */ | ||
1301 | setup_redirects(child,NULL); | ||
1302 | |||
1303 | pseudo_exec(child); | ||
1304 | } | ||
1305 | if (outpipe[1] != -1) { | ||
1306 | close(outpipe[1]); | ||
1307 | } | ||
1308 | |||
1309 | /* put our child in the process group whose leader is the | ||
1310 | first process in this pipe */ | ||
1311 | setpgid(child->pid, newjob->progs[0].pid); | ||
1312 | if (nextin != 0) | ||
1313 | close(nextin); | ||
1314 | if (nextout != 1) | ||
1315 | close(nextout); | ||
1316 | |||
1317 | /* If there isn't another process, nextin is garbage | ||
1318 | but it doesn't matter */ | ||
1319 | nextin = pipefds[0]; | ||
1320 | } | ||
1321 | |||
1322 | newjob->pgrp = newjob->progs[0].pid; | ||
1323 | |||
1324 | insert_job(newjob, inbg); | ||
1325 | |||
1326 | return 0; | ||
1327 | } | ||
1328 | |||
1329 | static int busy_loop(FILE * input) | ||
1330 | { | ||
1331 | char *command; | ||
1332 | char *next_command = NULL; | ||
1333 | struct job newjob; | ||
1334 | int i; | ||
1335 | int inbg = 0; | ||
1336 | int status; | ||
1337 | #if ENABLE_LASH_JOB_CONTROL | ||
1338 | pid_t parent_pgrp; | ||
1339 | /* save current owner of TTY so we can restore it on exit */ | ||
1340 | parent_pgrp = tcgetpgrp(shell_terminal); | ||
1341 | #endif | ||
1342 | newjob.job_list = &job_list; | ||
1343 | newjob.job_context = DEFAULT_CONTEXT; | ||
1344 | |||
1345 | command = xzalloc(BUFSIZ); | ||
1346 | |||
1347 | while (1) { | ||
1348 | if (!job_list.fg) { | ||
1349 | /* no job is in the foreground */ | ||
1350 | |||
1351 | /* see if any background processes have exited */ | ||
1352 | checkjobs(&job_list); | ||
1353 | |||
1354 | if (!next_command) { | ||
1355 | if (get_command_bufsiz(input, command)) | ||
1356 | break; | ||
1357 | next_command = command; | ||
1358 | } | ||
1359 | |||
1360 | if (!expand_arguments(next_command)) { | ||
1361 | free(command); | ||
1362 | command = xzalloc(BUFSIZ); | ||
1363 | next_command = NULL; | ||
1364 | continue; | ||
1365 | } | ||
1366 | |||
1367 | if (!parse_command(&next_command, &newjob, &inbg) && | ||
1368 | newjob.num_progs) { | ||
1369 | int pipefds[2] = { -1, -1 }; | ||
1370 | debug_printf("job=%p fed to run_command by busy_loop()'\n", | ||
1371 | &newjob); | ||
1372 | run_command(&newjob, inbg, pipefds); | ||
1373 | } | ||
1374 | else { | ||
1375 | free(command); | ||
1376 | command = xzalloc(BUFSIZ); | ||
1377 | next_command = NULL; | ||
1378 | } | ||
1379 | } else { | ||
1380 | /* a job is running in the foreground; wait for it */ | ||
1381 | i = 0; | ||
1382 | while (!job_list.fg->progs[i].pid || | ||
1383 | job_list.fg->progs[i].is_stopped == 1) i++; | ||
1384 | |||
1385 | if (waitpid(job_list.fg->progs[i].pid, &status, WUNTRACED) < 0) { | ||
1386 | if (errno != ECHILD) { | ||
1387 | bb_perror_msg_and_die("waitpid(%d)", job_list.fg->progs[i].pid); | ||
1388 | } | ||
1389 | } | ||
1390 | |||
1391 | if (WIFEXITED(status) || WIFSIGNALED(status)) { | ||
1392 | /* the child exited */ | ||
1393 | job_list.fg->running_progs--; | ||
1394 | job_list.fg->progs[i].pid = 0; | ||
1395 | |||
1396 | last_return_code = WEXITSTATUS(status); | ||
1397 | |||
1398 | if (!job_list.fg->running_progs) { | ||
1399 | /* child exited */ | ||
1400 | remove_job(&job_list, job_list.fg); | ||
1401 | job_list.fg = NULL; | ||
1402 | } | ||
1403 | } | ||
1404 | #if ENABLE_LASH_JOB_CONTROL | ||
1405 | else { | ||
1406 | /* the child was stopped */ | ||
1407 | job_list.fg->stopped_progs++; | ||
1408 | job_list.fg->progs[i].is_stopped = 1; | ||
1409 | |||
1410 | if (job_list.fg->stopped_progs == job_list.fg->running_progs) { | ||
1411 | printf("\n" JOB_STATUS_FORMAT, job_list.fg->jobid, | ||
1412 | "Stopped", job_list.fg->text); | ||
1413 | job_list.fg = NULL; | ||
1414 | } | ||
1415 | } | ||
1416 | |||
1417 | if (!job_list.fg) { | ||
1418 | /* move the shell to the foreground */ | ||
1419 | /* suppress messages when run from /linuxrc mag@sysgo.de */ | ||
1420 | if (tcsetpgrp(shell_terminal, getpgrp()) && errno != ENOTTY) | ||
1421 | bb_perror_msg("tcsetpgrp"); | ||
1422 | } | ||
1423 | #endif | ||
1424 | } | ||
1425 | } | ||
1426 | free(command); | ||
1427 | |||
1428 | #if ENABLE_LASH_JOB_CONTROL | ||
1429 | /* return controlling TTY back to parent process group before exiting */ | ||
1430 | if (tcsetpgrp(shell_terminal, parent_pgrp) && errno != ENOTTY) | ||
1431 | bb_perror_msg("tcsetpgrp"); | ||
1432 | #endif | ||
1433 | |||
1434 | /* return exit status if called with "-c" */ | ||
1435 | if (input == NULL && WIFEXITED(status)) | ||
1436 | return WEXITSTATUS(status); | ||
1437 | |||
1438 | return 0; | ||
1439 | } | ||
1440 | |||
1441 | #if ENABLE_FEATURE_CLEAN_UP | ||
1442 | static void free_memory(void) | ||
1443 | { | ||
1444 | free(cwd); | ||
1445 | |||
1446 | if (job_list.fg && !job_list.fg->running_progs) { | ||
1447 | remove_job(&job_list, job_list.fg); | ||
1448 | } | ||
1449 | } | ||
1450 | #else | ||
1451 | void free_memory(void); | ||
1452 | #endif | ||
1453 | |||
1454 | #if ENABLE_LASH_JOB_CONTROL | ||
1455 | /* Make sure we have a controlling tty. If we get started under a job | ||
1456 | * aware app (like bash for example), make sure we are now in charge so | ||
1457 | * we don't fight over who gets the foreground */ | ||
1458 | static void setup_job_control(void) | ||
1459 | { | ||
1460 | int status; | ||
1461 | pid_t shell_pgrp; | ||
1462 | |||
1463 | /* Loop until we are in the foreground. */ | ||
1464 | while ((status = tcgetpgrp(shell_terminal)) >= 0) { | ||
1465 | shell_pgrp = getpgrp(); | ||
1466 | if (status == shell_pgrp) { | ||
1467 | break; | ||
1468 | } | ||
1469 | kill(- shell_pgrp, SIGTTIN); | ||
1470 | } | ||
1471 | |||
1472 | /* Ignore interactive and job-control signals. */ | ||
1473 | signal(SIGINT, SIG_IGN); | ||
1474 | signal(SIGQUIT, SIG_IGN); | ||
1475 | signal(SIGTSTP, SIG_IGN); | ||
1476 | signal(SIGTTIN, SIG_IGN); | ||
1477 | signal(SIGTTOU, SIG_IGN); | ||
1478 | signal(SIGCHLD, SIG_IGN); | ||
1479 | |||
1480 | /* Put ourselves in our own process group. */ | ||
1481 | setsid(); | ||
1482 | shell_pgrp = getpid(); | ||
1483 | setpgid(shell_pgrp, shell_pgrp); | ||
1484 | |||
1485 | /* Grab control of the terminal. */ | ||
1486 | tcsetpgrp(shell_terminal, shell_pgrp); | ||
1487 | } | ||
1488 | #else | ||
1489 | static inline void setup_job_control(void) | ||
1490 | { | ||
1491 | } | ||
1492 | #endif | ||
1493 | |||
1494 | int lash_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
1495 | int lash_main(int argc, char **argv) | ||
1496 | { | ||
1497 | unsigned opt; | ||
1498 | FILE *input = stdin; | ||
1499 | |||
1500 | global_argc = argc; | ||
1501 | global_argv = argv; | ||
1502 | |||
1503 | #if ENABLE_FEATURE_EDITING | ||
1504 | line_input_state = new_line_input_t(FOR_SHELL); | ||
1505 | #endif | ||
1506 | |||
1507 | /* These variables need re-initializing when recursing */ | ||
1508 | last_jobid = 0; | ||
1509 | close_me_list = NULL; | ||
1510 | job_list.head = NULL; | ||
1511 | job_list.fg = NULL; | ||
1512 | last_return_code = 1; | ||
1513 | |||
1514 | if (global_argv[0] && global_argv[0][0] == '-') { | ||
1515 | FILE *prof_input; | ||
1516 | prof_input = fopen("/etc/profile", "r"); | ||
1517 | if (prof_input) { | ||
1518 | llist_add_to(&close_me_list, (void *)(long)fileno(prof_input)); | ||
1519 | /* Now run the file */ | ||
1520 | busy_loop(prof_input); | ||
1521 | fclose_if_not_stdin(prof_input); | ||
1522 | llist_pop(&close_me_list); | ||
1523 | } | ||
1524 | } | ||
1525 | |||
1526 | opt = getopt32(argv, "+ic:", &local_pending_command); | ||
1527 | #define LASH_OPT_i (1<<0) | ||
1528 | #define LASH_OPT_c (1<<1) | ||
1529 | if (opt & LASH_OPT_c) { | ||
1530 | input = NULL; | ||
1531 | optind++; | ||
1532 | global_argv += optind; | ||
1533 | } | ||
1534 | /* A shell is interactive if the `-i' flag was given, or if all of | ||
1535 | * the following conditions are met: | ||
1536 | * no -c command | ||
1537 | * no arguments remaining or the -s flag given | ||
1538 | * standard input is a terminal | ||
1539 | * standard output is a terminal | ||
1540 | * Refer to Posix.2, the description of the `sh' utility. */ | ||
1541 | if (global_argv[optind] == NULL && input == stdin | ||
1542 | && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO) | ||
1543 | ) { | ||
1544 | opt |= LASH_OPT_i; | ||
1545 | } | ||
1546 | setup_job_control(); | ||
1547 | if (opt & LASH_OPT_i) { | ||
1548 | /* Looks like they want an interactive shell */ | ||
1549 | if (!ENABLE_FEATURE_SH_EXTRA_QUIET) { | ||
1550 | printf("\n\n%s built-in shell (lash)\n" | ||
1551 | "Enter 'help' for a list of built-in commands.\n\n", | ||
1552 | bb_banner); | ||
1553 | } | ||
1554 | } else if (!local_pending_command && global_argv[optind]) { | ||
1555 | //printf( "optind=%d argv[optind]='%s'\n", optind, argv[optind]); | ||
1556 | input = xfopen(global_argv[optind], "r"); | ||
1557 | /* be lazy, never mark this closed */ | ||
1558 | llist_add_to(&close_me_list, (void *)(long)fileno(input)); | ||
1559 | } | ||
1560 | |||
1561 | /* initialize the cwd -- this is never freed...*/ | ||
1562 | update_cwd(); | ||
1563 | |||
1564 | if (ENABLE_FEATURE_CLEAN_UP) atexit(free_memory); | ||
1565 | |||
1566 | if (ENABLE_FEATURE_EDITING) cmdedit_set_initial_prompt(); | ||
1567 | else PS1 = NULL; | ||
1568 | |||
1569 | return busy_loop(input); | ||
1570 | } | ||