From 54e7ffb3a405ab058d12fbed9a63314ba996be90 Mon Sep 17 00:00:00 2001 From: Denis Vlasenko Date: Sat, 21 Apr 2007 00:03:36 +0000 Subject: hush: begin fixing non-functional job control --- shell/hush.c | 270 +++++++++++++++++++++++++++++++++++++++++++---------------- 1 file changed, 198 insertions(+), 72 deletions(-) (limited to 'shell/hush.c') diff --git a/shell/hush.c b/shell/hush.c index e2ff012d7..7792887a5 100644 --- a/shell/hush.c +++ b/shell/hush.c @@ -81,9 +81,6 @@ #include /* glob, of course */ #include /* should be pretty obvious */ -//#include -//#include - /* #include */ /* #define DEBUG_SHELL */ @@ -91,7 +88,7 @@ #define SPECIAL_VAR_SYMBOL 03 #define FLAG_EXIT_FROM_LOOP 1 #define FLAG_PARSE_SEMICOLON (1 << 1) /* symbol ';' is special for parser */ -#define FLAG_REPARSING (1 << 2) /* >=2nd pass */ +#define FLAG_REPARSING (1 << 2) /* >=2nd pass */ typedef enum { REDIRECT_INPUT = 1, @@ -193,7 +190,7 @@ struct child_prog { struct pipe { int jobid; /* job number */ int num_progs; /* total number of programs in job */ - int running_progs; /* number of programs running */ + int running_progs; /* number of programs running (not exited) */ char *text; /* name of job */ char *cmdbuf; /* buffer various argv's point into */ pid_t pgrp; /* process group ID for the job */ @@ -229,13 +226,17 @@ extern char **environ; /* This is in , but protected with __USE_GNU */ static const char *ifs; static unsigned char map[256]; static int fake_mode; -static int interactive; static struct close_me *close_me_head; static const char *cwd; static struct pipe *job_list; static unsigned last_bg_pid; static int last_jobid; -static unsigned shell_terminal; +/* 'interactive_fd' is a fd# open to ctty, if we have one + * _AND_ if we decided to mess with job control */ +static int interactive_fd; +static pid_t saved_task_pgrp; +static pid_t saved_tty_pgrp; + static const char *PS1; static const char *PS2; static struct variables shell_ver = { "HUSH_VERSION", "0.01", 1, 1, 0 }; @@ -308,7 +309,7 @@ static void __syntax(const char *file, int line) { bb_error_msg("syntax error %s:%d", file, line); } -// NB: was __FILE__, but that produces full path sometimes, so... +/* NB: was __FILE__, but that produces full path sometimes, so... */ #define syntax() __syntax("hush.c", __LINE__) /* Index of subroutines: */ @@ -425,6 +426,84 @@ static const struct built_in_command bltins[] = { { NULL, NULL, NULL } }; +/* Restores tty foreground process group, and exits. + * May be called as signal handler for fatal signal + * (will faithfully resend signal to itself, producing correct exit state) + * or called directly with -EXITCODE. + * We also call it if xfunc is exiting. */ +static void sigexit(int sig) ATTRIBUTE_NORETURN; +static void sigexit(int sig) +{ + sigset_t block_all; + + /* Disable all signals: job control, SIGPIPE, etc. */ + sigfillset(&block_all); + sigprocmask(SIG_SETMASK, &block_all, NULL); + + if (interactive_fd) { + if (sig > 0) { + enum { KILLED = sizeof("Killed by signal ")-1 }; + char buf[KILLED + sizeof(int)*3 + 1]; + char *p; + + /* bash actually says "Illegal instruction" and the like */ + strcpy(buf, "Killed by signal "); + p = utoa_to_buf(sig, buf+KILLED, sizeof(buf)-KILLED); + *p++ = '\n'; + write(interactive_fd, buf, p-buf); + } + tcsetpgrp(interactive_fd, saved_tty_pgrp); + } + + /* Not a signal, just exit */ + if (sig <= 0) + _exit(- sig); + + /* Enable only this sig and kill ourself with it */ + signal(sig, SIG_DFL); + sigdelset(&block_all, sig); + sigprocmask(SIG_SETMASK, &block_all, NULL); + raise(sig); + _exit(1); /* Should not reach it */ +} + +/* Restores tty foreground process group, and exits. */ +static void hush_exit(int exitcode) ATTRIBUTE_NORETURN; +static void hush_exit(int exitcode) +{ + fflush(NULL); /* flush all streams */ + sigexit(- (exitcode & 0xff)); +} + +/* Signals are grouped, we handle them in batches */ +static void set_fatal_sighandler(void (*handler)(int)) +{ + signal(SIGILL , handler); + signal(SIGTRAP, handler); + signal(SIGABRT, handler); + signal(SIGFPE , handler); + signal(SIGBUS , handler); + signal(SIGSEGV, handler); + /* bash 3.2 seems to handle these just like 'fatal' ones, + * but _without_ printing signal name. TODO: mimic this too? */ + signal(SIGHUP , handler); + signal(SIGPIPE, handler); + signal(SIGALRM, handler); +} +static void set_jobctrl_sighandler(void (*handler)(int)) +{ + signal(SIGTSTP, handler); + signal(SIGTTIN, handler); + signal(SIGTTOU, handler); +} +static void set_misc_sighandler(void (*handler)(int)) +{ + signal(SIGINT , handler); + signal(SIGQUIT, handler); + signal(SIGTERM, handler); +} +/* SIGCHLD is special and handled separately */ + static const char *set_cwd(void) { if (cwd == bb_msg_unknown) @@ -493,13 +572,15 @@ static int builtin_exec(struct child_prog *child) /* built-in 'exit' handler */ static int builtin_exit(struct child_prog *child) { - /* bash prints "exit\n" here, then: */ +// TODO: bash does it ONLY on top-level sh exit (+interacive only?) + //puts("exit"); /* bash does it */ + if (child->argv[1] == NULL) - exit(last_return_code); + hush_exit(last_return_code); /* mimic bash: exit 123abc == exit 255 + error msg */ xfunc_error_retval = 255; /* bash: exit -2 == exit 254, no error msg */ - exit(xatoi(child->argv[1])); + hush_exit(xatoi(child->argv[1])); } /* built-in 'export VAR=value' handler */ @@ -555,7 +636,7 @@ static int builtin_fg_bg(struct child_prog *child) int i, jobnum; struct pipe *pi; - if (!interactive) + if (!interactive_fd) return EXIT_FAILURE; /* If they gave us no args, assume they want the last backgrounded task */ if (!child->argv[1]) { @@ -581,7 +662,7 @@ static int builtin_fg_bg(struct child_prog *child) found: if (*child->argv[0] == 'f') { /* Put the job into the foreground. */ - tcsetpgrp(shell_terminal, pi->pgrp); + tcsetpgrp(interactive_fd, pi->pgrp); } /* Restart the processes in the job */ @@ -915,8 +996,8 @@ static int file_get(struct in_str *i) } else { /* need to double check i->file because we might be doing something * more complicated by now, like sourcing or substituting. */ - if (i->__promptme && interactive && i->file == stdin) { - while (!i->p || !(interactive && strlen(i->p))) { + if (i->__promptme && interactive_fd && i->file == stdin) { + while (!i->p || !(interactive_fd && strlen(i->p))) { get_user_input(i); } i->promptmode = 2; @@ -1116,7 +1197,7 @@ static void pseudo_exec(struct child_prog *child) if (child->group) { debug_printf("runtime nesting to group\n"); - interactive = 0; /* crucial!!!! */ + interactive_fd = 0; /* crucial!!!! */ rcode = run_list_real(child->group); /* OK to leak memory by not calling free_pipe_list, * since this process is about to exit */ @@ -1203,25 +1284,39 @@ static int checkjobs(struct pipe* fg_pipe) int prognum = 0; struct pipe *pi; pid_t childpid; + int rcode = 0; attributes = WUNTRACED; +//WUNTRACED?? huh, what will happed on Ctrl-Z? fg waiting code +//doesn't seem to be ready for stopped children! (only exiting ones)... if (fg_pipe == NULL) { attributes |= WNOHANG; } + wait_more: while ((childpid = waitpid(-1, &status, attributes)) > 0) { + /* Were we asked to wait for fg pipe? */ if (fg_pipe) { - int i, rcode = 0; + int i; for (i = 0; i < fg_pipe->num_progs; i++) { if (fg_pipe->progs[i].pid == childpid) { + /* printf("process %d exit %d\n", i, WEXITSTATUS(status)); */ + fg_pipe->progs[i].pid = 0; if (i == fg_pipe->num_progs-1) + /* last process gives overall exitstatus */ rcode = WEXITSTATUS(status); - fg_pipe->num_progs--; - return rcode; + if (--fg_pipe->running_progs <= 0) + /* All processes in fg pipe have exited */ + return rcode; + /* There are still running processes in the fg pipe */ + goto wait_more; } } } + /* We asked to wait for bg or orphaned children */ + /* No need to remember exitcode in this case */ + for (pi = job_list; pi; pi = pi->next) { prognum = 0; while (prognum < pi->num_progs) { @@ -1257,9 +1352,9 @@ static int checkjobs(struct pipe* fg_pipe) bb_perror_msg("waitpid"); /* move the shell to the foreground */ - //if (interactive && tcsetpgrp(shell_terminal, getpgid(0))) + //if (interactive_fd && tcsetpgrp(interactive_fd, getpgid(0))) // bb_perror_msg("tcsetpgrp-2"); - return -1; + return rcode; } /* run_pipe_real() starts all the jobs, but doesn't wait for anything @@ -1376,6 +1471,8 @@ static int run_pipe_real(struct pipe *pi) } #if ENABLE_FEATURE_SH_STANDALONE { +// FIXME: applet runs like part of shell - for example, it ignores +// SIGINT! Try to Ctrl-C out of "rm -i"... doesn't work const struct bb_applet *a = find_applet_by_name(child->argv[i]); if (a && a->nofork) { setup_redirects(child, squirrel); @@ -1387,6 +1484,11 @@ static int run_pipe_real(struct pipe *pi) #endif } + /* Disable job control signals for shell (parent) and + * for initial child code after fork */ + set_jobctrl_sighandler(SIG_IGN); + + pi->running_progs = 0; for (i = 0; i < pi->num_progs; i++) { child = &(pi->progs[i]); @@ -1406,18 +1508,22 @@ static int run_pipe_real(struct pipe *pi) #else child->pid = vfork(); #endif - if (!child->pid) { - /* Set the handling for job control signals back to the default. */ - signal(SIGINT, SIG_DFL); - signal(SIGQUIT, SIG_DFL); - signal(SIGTERM, SIG_DFL); - signal(SIGTSTP, SIG_DFL); - signal(SIGTTIN, SIG_DFL); - signal(SIGTTOU, SIG_DFL); - signal(SIGCHLD, SIG_DFL); + if (!child->pid) { /* child */ + /* Every child adds itself to new process group + * with pgid == pid of first child in pipe */ + if (interactive_fd) { + if (pi->pgrp < 0) /* true for 1st process only */ + pi->pgrp = getpid(); + if (setpgid(0, pi->pgrp) == 0 && pi->followup != PIPE_BG) { + /* We do it in *every* child, not just first, + * to avoid races */ + tcsetpgrp(interactive_fd, pi->pgrp); + } + /* Don't do pgrp restore anymore on fatal signals */ + set_fatal_sighandler(SIG_DFL); + } close_all(); - if (nextin != 0) { dup2(nextin, 0); close(nextin); @@ -1429,33 +1535,28 @@ static int run_pipe_real(struct pipe *pi) if (pipefds[0] != -1) { close(pipefds[0]); /* opposite end of our output pipe */ } - /* Like bash, explicit redirects override pipes, * and the pipe fd is available for dup'ing. */ setup_redirects(child, NULL); - if (interactive && pi->followup != PIPE_BG) { - /* If we (the child) win the race, put ourselves in the process - * group whose leader is the first process in this pipe. */ - if (pi->pgrp < 0) { - pi->pgrp = getpid(); - } - if (setpgid(0, pi->pgrp) == 0) { - tcsetpgrp(2, pi->pgrp); - } - } - + /* Restore default handlers just prior to exec */ + set_jobctrl_sighandler(SIG_DFL); + set_misc_sighandler(SIG_DFL); + signal(SIGCHLD, SIG_DFL); pseudo_exec(child); } - /* put our child in the process group whose leader is the - first process in this pipe */ - if (pi->pgrp < 0) { + pi->running_progs++; + + /* Second and next children need to know ipd of first one */ + if (pi->pgrp < 0) pi->pgrp = child->pid; - } + /* Don't check for errors. The child may be dead already, * in which case setpgid returns error code EACCES. */ - setpgid(child->pid, pi->pgrp); + //why we do it at all?? child does it itself + //if (interactive_fd) + // setpgid(child->pid, pi->pgrp); if (nextin != 0) close(nextin); @@ -1580,14 +1681,14 @@ static int run_list_real(struct pipe *pi) insert_bg_job(pi); rcode = EXIT_SUCCESS; } else { - if (interactive) { - /* move the new process group into the foreground */ - if (tcsetpgrp(shell_terminal, pi->pgrp) && errno != ENOTTY) - bb_perror_msg("tcsetpgrp-3"); + if (interactive_fd) { + pid_t p; rcode = checkjobs(pi); /* move the shell to the foreground */ - if (tcsetpgrp(shell_terminal, getpgid(0)) && errno != ENOTTY) + p = getpgid(0); + if (tcsetpgrp(interactive_fd, p) && errno != ENOTTY) bb_perror_msg("tcsetpgrp-4"); + debug_printf("getpgid(0)=%d\n", (int)p); } else { rcode = checkjobs(pi); } @@ -2660,28 +2761,33 @@ static int parse_file_outer(FILE *f) * we don't fight over who gets the foreground */ static void setup_job_control(void) { - /*static --why?? */ pid_t shell_pgrp; + pid_t shell_pgrp; + + fcntl(interactive_fd, F_SETFD, FD_CLOEXEC); /* Loop until we are in the foreground. */ - while (tcgetpgrp(shell_terminal) != (shell_pgrp = getpgrp())) + while (1) { + shell_pgrp = getpgrp(); + if (tcgetpgrp(interactive_fd) == shell_pgrp) + break; +// and this does... what? need a comment here kill(- shell_pgrp, SIGTTIN); + } + + /* Ignore job-control and misc signals. */ + set_jobctrl_sighandler(SIG_IGN); + set_misc_sighandler(SIG_IGN); +//huh? signal(SIGCHLD, SIG_IGN); - /* Ignore interactive and job-control signals. */ - signal(SIGINT, SIG_IGN); - signal(SIGQUIT, SIG_IGN); - signal(SIGTERM, SIG_IGN); - signal(SIGTSTP, SIG_IGN); - signal(SIGTTIN, SIG_IGN); - signal(SIGTTOU, SIG_IGN); - signal(SIGCHLD, SIG_IGN); + /* We _must_ do cleanup on fatal signals */ + set_fatal_sighandler(sigexit); /* Put ourselves in our own process group. */ - setsid(); shell_pgrp = getpid(); - setpgid(shell_pgrp, shell_pgrp); + setpgrp(); /* is the same as setpgid(shell_pgrp, shell_pgrp); */ /* Grab control of the terminal. */ - tcsetpgrp(shell_terminal, shell_pgrp); + tcsetpgrp(interactive_fd, shell_pgrp); } int hush_main(int argc, char **argv); @@ -2705,7 +2811,7 @@ int hush_main(int argc, char **argv) ifs = NULL; /* map[] is taken care of with call to update_ifs_map() */ fake_mode = 0; - interactive = 0; + interactive_fd = 0; close_me_head = NULL; last_bg_pid = 0; job_list = NULL; @@ -2747,7 +2853,7 @@ int hush_main(int argc, char **argv) opt = parse_string_outer(optarg, FLAG_PARSE_SEMICOLON); goto final_return; case 'i': - interactive++; + /*interactive_fd++;*/ //huh?? break; case 'f': fake_mode++; @@ -2772,11 +2878,25 @@ int hush_main(int argc, char **argv) if (argv[optind] == NULL && input == stdin && isatty(STDIN_FILENO) && isatty(STDOUT_FILENO) ) { - interactive++; + saved_tty_pgrp = tcgetpgrp(STDIN_FILENO); + debug_printf("saved_tty_pgrp=%d\n", saved_tty_pgrp); + if (saved_tty_pgrp >= 0) { + saved_task_pgrp = getpgrp(); + debug_printf("saved_task_pgrp=%d\n", saved_task_pgrp); + /* try to dup to high fd#, >= 255 */ + interactive_fd = fcntl(STDIN_FILENO, F_DUPFD, 255); + if (interactive_fd < 0) { + /* try to dup to any fd */ + interactive_fd = dup(STDIN_FILENO); + if (interactive_fd < 0) + /* give up */ + interactive_fd = 0; + } + } } - debug_printf("\ninteractive=%d\n", interactive); - if (interactive) { + debug_printf("\ninteractive_fd=%d\n", interactive_fd); + if (interactive_fd) { /* Looks like they want an interactive shell */ #if !ENABLE_FEATURE_SH_EXTRA_QUIET printf( "\n\n%s hush - the humble shell v0.01 (testing)\n", @@ -2784,6 +2904,12 @@ int hush_main(int argc, char **argv) printf( "Enter 'help' for a list of built-in commands.\n\n"); #endif setup_job_control(); + /* Make xfuncs do cleanup on exit */ + die_sleep = -1; /* flag */ + if (setjmp(die_jmp)) { + /* xfunc has failed! die die die */ + hush_exit(xfunc_error_retval); + } } if (argv[optind] == NULL) { @@ -2815,7 +2941,7 @@ int hush_main(int argc, char **argv) #endif final_return: - return opt ? opt : last_return_code; + hush_exit(opt ? opt : last_return_code); } static char *insert_var_value(char *inp) -- cgit v1.2.3