Home | History | Annotate | Line # | Download | only in sh
jobs.c revision 1.34
      1 /*	$NetBSD: jobs.c,v 1.34 2000/05/13 20:59:41 elric Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1991, 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * This code is derived from software contributed to Berkeley by
      8  * Kenneth Almquist.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  * 3. All advertising materials mentioning features or use of this software
     19  *    must display the following acknowledgement:
     20  *	This product includes software developed by the University of
     21  *	California, Berkeley and its contributors.
     22  * 4. Neither the name of the University nor the names of its contributors
     23  *    may be used to endorse or promote products derived from this software
     24  *    without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     36  * SUCH DAMAGE.
     37  */
     38 
     39 #include <sys/cdefs.h>
     40 #ifndef lint
     41 #if 0
     42 static char sccsid[] = "@(#)jobs.c	8.5 (Berkeley) 5/4/95";
     43 #else
     44 __RCSID("$NetBSD: jobs.c,v 1.34 2000/05/13 20:59:41 elric Exp $");
     45 #endif
     46 #endif /* not lint */
     47 
     48 #include <fcntl.h>
     49 #include <signal.h>
     50 #include <errno.h>
     51 #include <unistd.h>
     52 #include <stdlib.h>
     53 #include <paths.h>
     54 #include <sys/types.h>
     55 #include <sys/param.h>
     56 #ifdef BSD
     57 #include <sys/wait.h>
     58 #include <sys/time.h>
     59 #include <sys/resource.h>
     60 #endif
     61 #include <sys/ioctl.h>
     62 
     63 #include "shell.h"
     64 #if JOBS
     65 #if OLD_TTY_DRIVER
     66 #include "sgtty.h"
     67 #else
     68 #include <termios.h>
     69 #endif
     70 #undef CEOF			/* syntax.h redefines this */
     71 #endif
     72 #include "redir.h"
     73 #include "show.h"
     74 #include "main.h"
     75 #include "parser.h"
     76 #include "nodes.h"
     77 #include "jobs.h"
     78 #include "options.h"
     79 #include "trap.h"
     80 #include "syntax.h"
     81 #include "input.h"
     82 #include "output.h"
     83 #include "memalloc.h"
     84 #include "error.h"
     85 #include "mystring.h"
     86 
     87 
     88 struct job *jobtab;		/* array of jobs */
     89 int njobs;			/* size of array */
     90 MKINIT short backgndpid = -1;	/* pid of last background process */
     91 #if JOBS
     92 int initialpgrp;		/* pgrp of shell on invocation */
     93 short curjob;			/* current job */
     94 #endif
     95 
     96 STATIC void restartjob __P((struct job *));
     97 STATIC void freejob __P((struct job *));
     98 STATIC struct job *getjob __P((char *));
     99 STATIC int dowait __P((int, struct job *));
    100 STATIC int onsigchild __P((void));
    101 STATIC int waitproc __P((int, int *));
    102 STATIC void cmdtxt __P((union node *));
    103 STATIC void cmdputs __P((const char *));
    104 
    105 
    106 /*
    107  * Turn job control on and off.
    108  *
    109  * Note:  This code assumes that the third arg to ioctl is a character
    110  * pointer, which is true on Berkeley systems but not System V.  Since
    111  * System V doesn't have job control yet, this isn't a problem now.
    112  */
    113 
    114 MKINIT int jobctl;
    115 
    116 void
    117 setjobctl(on)
    118 	int on;
    119 {
    120 #ifdef OLD_TTY_DRIVER
    121 	int ldisc;
    122 #endif
    123 
    124 	if (on == jobctl || rootshell == 0)
    125 		return;
    126 	if (on) {
    127 		do { /* while we are in the background */
    128 #ifdef OLD_TTY_DRIVER
    129 			if (ioctl(2, TIOCGPGRP, (char *)&initialpgrp) < 0) {
    130 #else
    131 			initialpgrp = tcgetpgrp(2);
    132 			if (initialpgrp < 0) {
    133 #endif
    134 				out2str("sh: can't access tty; job control turned off\n");
    135 				mflag = 0;
    136 				return;
    137 			}
    138 			if (initialpgrp == -1)
    139 				initialpgrp = getpgrp();
    140 			else if (initialpgrp != getpgrp()) {
    141 				killpg(initialpgrp, SIGTTIN);
    142 				continue;
    143 			}
    144 		} while (0);
    145 #ifdef OLD_TTY_DRIVER
    146 		if (ioctl(2, TIOCGETD, (char *)&ldisc) < 0 || ldisc != NTTYDISC) {
    147 			out2str("sh: need new tty driver to run job control; job control turned off\n");
    148 			mflag = 0;
    149 			return;
    150 		}
    151 #endif
    152 		setsignal(SIGTSTP, 0);
    153 		setsignal(SIGTTOU, 0);
    154 		setsignal(SIGTTIN, 0);
    155 		setpgid(0, rootpid);
    156 #ifdef OLD_TTY_DRIVER
    157 		ioctl(2, TIOCSPGRP, (char *)&rootpid);
    158 #else
    159 		tcsetpgrp(2, rootpid);
    160 #endif
    161 	} else { /* turning job control off */
    162 		setpgid(0, initialpgrp);
    163 #ifdef OLD_TTY_DRIVER
    164 		ioctl(2, TIOCSPGRP, (char *)&initialpgrp);
    165 #else
    166 		tcsetpgrp(2, initialpgrp);
    167 #endif
    168 		setsignal(SIGTSTP, 0);
    169 		setsignal(SIGTTOU, 0);
    170 		setsignal(SIGTTIN, 0);
    171 	}
    172 	jobctl = on;
    173 }
    174 
    175 
    176 #ifdef mkinit
    177 INCLUDE <stdlib.h>
    178 
    179 SHELLPROC {
    180 	backgndpid = -1;
    181 #if JOBS
    182 	jobctl = 0;
    183 #endif
    184 }
    185 
    186 #endif
    187 
    188 
    189 
    190 #if JOBS
    191 int
    192 fgcmd(argc, argv)
    193 	int argc;
    194 	char **argv;
    195 {
    196 	struct job *jp;
    197 	int pgrp;
    198 	int status;
    199 
    200 	jp = getjob(argv[1]);
    201 	if (jp->jobctl == 0)
    202 		error("job not created under job control");
    203 	pgrp = jp->ps[0].pid;
    204 #ifdef OLD_TTY_DRIVER
    205 	ioctl(2, TIOCSPGRP, (char *)&pgrp);
    206 #else
    207 	tcsetpgrp(2, pgrp);
    208 #endif
    209 	restartjob(jp);
    210 	INTOFF;
    211 	status = waitforjob(jp);
    212 	INTON;
    213 	return status;
    214 }
    215 
    216 
    217 int
    218 bgcmd(argc, argv)
    219 	int argc;
    220 	char **argv;
    221 {
    222 	struct job *jp;
    223 
    224 	do {
    225 		jp = getjob(*++argv);
    226 		if (jp->jobctl == 0)
    227 			error("job not created under job control");
    228 		restartjob(jp);
    229 	} while (--argc > 1);
    230 	return 0;
    231 }
    232 
    233 
    234 STATIC void
    235 restartjob(jp)
    236 	struct job *jp;
    237 {
    238 	struct procstat *ps;
    239 	int i;
    240 
    241 	if (jp->state == JOBDONE)
    242 		return;
    243 	INTOFF;
    244 	killpg(jp->ps[0].pid, SIGCONT);
    245 	for (ps = jp->ps, i = jp->nprocs ; --i >= 0 ; ps++) {
    246 		if (WIFSTOPPED(ps->status)) {
    247 			ps->status = -1;
    248 			jp->state = 0;
    249 		}
    250 	}
    251 	INTON;
    252 }
    253 #endif
    254 
    255 
    256 int
    257 jobscmd(argc, argv)
    258 	int argc;
    259 	char **argv;
    260 {
    261 	showjobs(0);
    262 	return 0;
    263 }
    264 
    265 
    266 /*
    267  * Print a list of jobs.  If "change" is nonzero, only print jobs whose
    268  * statuses have changed since the last call to showjobs.
    269  *
    270  * If the shell is interrupted in the process of creating a job, the
    271  * result may be a job structure containing zero processes.  Such structures
    272  * will be freed here.
    273  */
    274 
    275 void
    276 showjobs(change)
    277 	int change;
    278 {
    279 	int jobno;
    280 	int procno;
    281 	int i;
    282 	struct job *jp;
    283 	struct procstat *ps;
    284 	int col;
    285 	char s[64];
    286 
    287 	TRACE(("showjobs(%d) called\n", change));
    288 	while (dowait(0, (struct job *)NULL) > 0);
    289 	for (jobno = 1, jp = jobtab ; jobno <= njobs ; jobno++, jp++) {
    290 		if (! jp->used)
    291 			continue;
    292 		if (jp->nprocs == 0) {
    293 			freejob(jp);
    294 			continue;
    295 		}
    296 		if (change && ! jp->changed)
    297 			continue;
    298 		procno = jp->nprocs;
    299 		for (ps = jp->ps ; ; ps++) {	/* for each process */
    300 			if (ps == jp->ps)
    301 				fmtstr(s, 64, "[%d] %ld ", jobno,
    302 				    (long)ps->pid);
    303 			else
    304 				fmtstr(s, 64, "    %ld ",
    305 				    (long)ps->pid);
    306 			out1str(s);
    307 			col = strlen(s);
    308 			s[0] = '\0';
    309 			if (ps->status == -1) {
    310 				/* don't print anything */
    311 			} else if (WIFEXITED(ps->status)) {
    312 				fmtstr(s, 64, "Exit %d",
    313 				       WEXITSTATUS(ps->status));
    314 			} else {
    315 #if JOBS
    316 				if (WIFSTOPPED(ps->status))
    317 					i = WSTOPSIG(ps->status);
    318 				else /* WIFSIGNALED(ps->status) */
    319 #endif
    320 					i = WTERMSIG(ps->status);
    321 				if ((i & 0x7F) < NSIG && sys_siglist[i & 0x7F])
    322 					scopy(sys_siglist[i & 0x7F], s);
    323 				else
    324 					fmtstr(s, 64, "Signal %d", i & 0x7F);
    325 				if (WCOREDUMP(ps->status))
    326 					strcat(s, " (core dumped)");
    327 			}
    328 			out1str(s);
    329 			col += strlen(s);
    330 			do {
    331 				out1c(' ');
    332 				col++;
    333 			} while (col < 30);
    334 			out1str(ps->cmd);
    335 			out1c('\n');
    336 			if (--procno <= 0)
    337 				break;
    338 		}
    339 		jp->changed = 0;
    340 		if (jp->state == JOBDONE) {
    341 			freejob(jp);
    342 		}
    343 	}
    344 }
    345 
    346 
    347 /*
    348  * Mark a job structure as unused.
    349  */
    350 
    351 STATIC void
    352 freejob(jp)
    353 	struct job *jp;
    354 	{
    355 	struct procstat *ps;
    356 	int i;
    357 
    358 	INTOFF;
    359 	for (i = jp->nprocs, ps = jp->ps ; --i >= 0 ; ps++) {
    360 		if (ps->cmd != nullstr)
    361 			ckfree(ps->cmd);
    362 	}
    363 	if (jp->ps != &jp->ps0)
    364 		ckfree(jp->ps);
    365 	jp->used = 0;
    366 #if JOBS
    367 	if (curjob == jp - jobtab + 1)
    368 		curjob = 0;
    369 #endif
    370 	INTON;
    371 }
    372 
    373 
    374 
    375 int
    376 waitcmd(argc, argv)
    377 	int argc;
    378 	char **argv;
    379 {
    380 	struct job *job;
    381 	int status, retval;
    382 	struct job *jp;
    383 
    384 	if (argc > 1) {
    385 		job = getjob(argv[1]);
    386 	} else {
    387 		job = NULL;
    388 	}
    389 	for (;;) {	/* loop until process terminated or stopped */
    390 		if (job != NULL) {
    391 			if (job->state) {
    392 				status = job->ps[job->nprocs - 1].status;
    393 				if (WIFEXITED(status))
    394 					retval = WEXITSTATUS(status);
    395 #if JOBS
    396 				else if (WIFSTOPPED(status))
    397 					retval = WSTOPSIG(status) + 128;
    398 #endif
    399 				else {
    400 					/* XXX: limits number of signals */
    401 					retval = WTERMSIG(status) + 128;
    402 				}
    403 				if (! iflag)
    404 					freejob(job);
    405 				return retval;
    406 			}
    407 		} else {
    408 			for (jp = jobtab ; ; jp++) {
    409 				if (jp >= jobtab + njobs) {	/* no running procs */
    410 					return 0;
    411 				}
    412 				if (jp->used && jp->state == 0)
    413 					break;
    414 			}
    415 		}
    416 		dowait(1, (struct job *)NULL);
    417 	}
    418 }
    419 
    420 
    421 
    422 int
    423 jobidcmd(argc, argv)
    424 	int argc;
    425 	char **argv;
    426 {
    427 	struct job *jp;
    428 	int i;
    429 
    430 	jp = getjob(argv[1]);
    431 	for (i = 0 ; i < jp->nprocs ; ) {
    432 		out1fmt("%ld", (long)jp->ps[i].pid);
    433 		out1c(++i < jp->nprocs? ' ' : '\n');
    434 	}
    435 	return 0;
    436 }
    437 
    438 
    439 
    440 /*
    441  * Convert a job name to a job structure.
    442  */
    443 
    444 STATIC struct job *
    445 getjob(name)
    446 	char *name;
    447 	{
    448 	int jobno;
    449 	struct job *jp;
    450 	int pid;
    451 	int i;
    452 
    453 	if (name == NULL) {
    454 #if JOBS
    455 currentjob:
    456 		if ((jobno = curjob) == 0 || jobtab[jobno - 1].used == 0)
    457 			error("No current job");
    458 		return &jobtab[jobno - 1];
    459 #else
    460 		error("No current job");
    461 #endif
    462 	} else if (name[0] == '%') {
    463 		if (is_digit(name[1])) {
    464 			jobno = number(name + 1);
    465 			if (jobno > 0 && jobno <= njobs
    466 			 && jobtab[jobno - 1].used != 0)
    467 				return &jobtab[jobno - 1];
    468 #if JOBS
    469 		} else if (name[1] == '%' && name[2] == '\0') {
    470 			goto currentjob;
    471 #endif
    472 		} else {
    473 			struct job *found = NULL;
    474 			for (jp = jobtab, i = njobs ; --i >= 0 ; jp++) {
    475 				if (jp->used && jp->nprocs > 0
    476 				 && prefix(name + 1, jp->ps[0].cmd)) {
    477 					if (found)
    478 						error("%s: ambiguous", name);
    479 					found = jp;
    480 				}
    481 			}
    482 			if (found)
    483 				return found;
    484 		}
    485 	} else if (is_number(name)) {
    486 		pid = number(name);
    487 		for (jp = jobtab, i = njobs ; --i >= 0 ; jp++) {
    488 			if (jp->used && jp->nprocs > 0
    489 			 && jp->ps[jp->nprocs - 1].pid == pid)
    490 				return jp;
    491 		}
    492 	}
    493 	error("No such job: %s", name);
    494 	/* NOTREACHED */
    495 }
    496 
    497 
    498 
    499 /*
    500  * Return a new job structure,
    501  */
    502 
    503 struct job *
    504 makejob(node, nprocs)
    505 	union node *node;
    506 	int nprocs;
    507 {
    508 	int i;
    509 	struct job *jp;
    510 
    511 	for (i = njobs, jp = jobtab ; ; jp++) {
    512 		if (--i < 0) {
    513 			INTOFF;
    514 			if (njobs == 0) {
    515 				jobtab = ckmalloc(4 * sizeof jobtab[0]);
    516 			} else {
    517 				jp = ckmalloc((njobs + 4) * sizeof jobtab[0]);
    518 				memcpy(jp, jobtab, njobs * sizeof jp[0]);
    519 				/* Relocate `ps' pointers */
    520 				for (i = 0; i < njobs; i++)
    521 					if (jp[i].ps == &jobtab[i].ps0)
    522 						jp[i].ps = &jp[i].ps0;
    523 				ckfree(jobtab);
    524 				jobtab = jp;
    525 			}
    526 			jp = jobtab + njobs;
    527 			for (i = 4 ; --i >= 0 ; jobtab[njobs++].used = 0);
    528 			INTON;
    529 			break;
    530 		}
    531 		if (jp->used == 0)
    532 			break;
    533 	}
    534 	INTOFF;
    535 	jp->state = 0;
    536 	jp->used = 1;
    537 	jp->changed = 0;
    538 	jp->nprocs = 0;
    539 #if JOBS
    540 	jp->jobctl = jobctl;
    541 #endif
    542 	if (nprocs > 1) {
    543 		jp->ps = ckmalloc(nprocs * sizeof (struct procstat));
    544 	} else {
    545 		jp->ps = &jp->ps0;
    546 	}
    547 	INTON;
    548 	TRACE(("makejob(0x%lx, %d) returns %%%d\n", (long)node, nprocs,
    549 	    jp - jobtab + 1));
    550 	return jp;
    551 }
    552 
    553 
    554 /*
    555  * Fork of a subshell.  If we are doing job control, give the subshell its
    556  * own process group.  Jp is a job structure that the job is to be added to.
    557  * N is the command that will be evaluated by the child.  Both jp and n may
    558  * be NULL.  The mode parameter can be one of the following:
    559  *	FORK_FG - Fork off a foreground process.
    560  *	FORK_BG - Fork off a background process.
    561  *	FORK_NOJOB - Like FORK_FG, but don't give the process its own
    562  *		     process group even if job control is on.
    563  *
    564  * When job control is turned off, background processes have their standard
    565  * input redirected to /dev/null (except for the second and later processes
    566  * in a pipeline).
    567  */
    568 
    569 int
    570 forkshell(jp, n, mode)
    571 	union node *n;
    572 	struct job *jp;
    573 	int mode;
    574 {
    575 	int pid;
    576 
    577 	TRACE(("forkshell(%%%d, 0x%lx, %d) called\n", jp - jobtab, (long)n,
    578 	    mode));
    579 	INTOFF;
    580 	switch (pid = fork()) {
    581 	case -1:
    582 		TRACE(("Fork failed, errno=%d\n", errno));
    583 		INTON;
    584 		error("Cannot fork");
    585 		break;
    586 	case 0:
    587 		forkchild(jp, n, mode, 0);
    588 		return 0;
    589 	default:
    590 		return forkparent(jp, n, mode, pid);
    591 	}
    592 }
    593 
    594 int
    595 forkparent(jp, n, mode, pid)
    596 	union node *n;
    597 	struct job *jp;
    598 	int mode;
    599 	pid_t pid;
    600 {
    601 	int pgrp;
    602 
    603 	if (rootshell && mode != FORK_NOJOB && mflag) {
    604 		if (jp == NULL || jp->nprocs == 0)
    605 			pgrp = pid;
    606 		else
    607 			pgrp = jp->ps[0].pid;
    608 		setpgid(pid, pgrp);
    609 	}
    610 	if (mode == FORK_BG)
    611 		backgndpid = pid;		/* set $! */
    612 	if (jp) {
    613 		struct procstat *ps = &jp->ps[jp->nprocs++];
    614 		ps->pid = pid;
    615 		ps->status = -1;
    616 		ps->cmd = nullstr;
    617 		if (iflag && rootshell && n)
    618 			ps->cmd = commandtext(n);
    619 	}
    620 	INTON;
    621 	TRACE(("In parent shell:  child = %d\n", pid));
    622 	return pid;
    623 }
    624 
    625 void
    626 forkchild(jp, n, mode, vforked)
    627 	union node *n;
    628 	struct job *jp;
    629 	int mode;
    630 	int vforked;
    631 {
    632 	struct job *p;
    633 	int wasroot;
    634 	int i;
    635 	int pgrp;
    636 	const char *devnull = _PATH_DEVNULL;
    637 	const char *nullerr = "Can't open %s";
    638 
    639 	TRACE(("Child shell %d\n", getpid()));
    640 	wasroot = rootshell;
    641 	if (!vforked) {
    642 		rootshell = 0;
    643 		for (i = njobs, p = jobtab ; --i >= 0 ; p++)
    644 			if (p->used)
    645 				freejob(p);
    646 	}
    647 	closescript(vforked);
    648 	INTON;
    649 	clear_traps(vforked);
    650 #if JOBS
    651 	if (!vforked)
    652 		jobctl = 0;		/* do job control only in root shell */
    653 	if (wasroot && mode != FORK_NOJOB && mflag) {
    654 		if (jp == NULL || jp->nprocs == 0)
    655 			pgrp = getpid();
    656 		else
    657 			pgrp = jp->ps[0].pid;
    658 		setpgid(0, pgrp);
    659 		if (mode == FORK_FG) {
    660 			/*** this causes superfluous TIOCSPGRPS ***/
    661 #ifdef OLD_TTY_DRIVER
    662 			if (ioctl(2, TIOCSPGRP, (char *)&pgrp) < 0)
    663 				error("TIOCSPGRP failed, errno=%d", errno);
    664 #else
    665 			if (tcsetpgrp(2, pgrp) < 0)
    666 				error("tcsetpgrp failed, errno=%d", errno);
    667 #endif
    668 		}
    669 		setsignal(SIGTSTP, vforked);
    670 		setsignal(SIGTTOU, vforked);
    671 	} else if (mode == FORK_BG) {
    672 		ignoresig(SIGINT, vforked);
    673 		ignoresig(SIGQUIT, vforked);
    674 		if ((jp == NULL || jp->nprocs == 0) &&
    675 		    ! fd0_redirected_p ()) {
    676 			close(0);
    677 			if (open(devnull, O_RDONLY) != 0)
    678 				error(nullerr, devnull);
    679 		}
    680 	}
    681 #else
    682 	if (mode == FORK_BG) {
    683 		ignoresig(SIGINT, vforked);
    684 		ignoresig(SIGQUIT, vforked);
    685 		if ((jp == NULL || jp->nprocs == 0) &&
    686 		    ! fd0_redirected_p ()) {
    687 			close(0);
    688 			if (open(devnull, O_RDONLY) != 0)
    689 				error(nullerr, devnull);
    690 		}
    691 	}
    692 #endif
    693 	if (wasroot && iflag) {
    694 		setsignal(SIGINT, vforked);
    695 		setsignal(SIGQUIT, vforked);
    696 		setsignal(SIGTERM, vforked);
    697 	}
    698 }
    699 
    700 
    701 
    702 /*
    703  * Wait for job to finish.
    704  *
    705  * Under job control we have the problem that while a child process is
    706  * running interrupts generated by the user are sent to the child but not
    707  * to the shell.  This means that an infinite loop started by an inter-
    708  * active user may be hard to kill.  With job control turned off, an
    709  * interactive user may place an interactive program inside a loop.  If
    710  * the interactive program catches interrupts, the user doesn't want
    711  * these interrupts to also abort the loop.  The approach we take here
    712  * is to have the shell ignore interrupt signals while waiting for a
    713  * forground process to terminate, and then send itself an interrupt
    714  * signal if the child process was terminated by an interrupt signal.
    715  * Unfortunately, some programs want to do a bit of cleanup and then
    716  * exit on interrupt; unless these processes terminate themselves by
    717  * sending a signal to themselves (instead of calling exit) they will
    718  * confuse this approach.
    719  */
    720 
    721 int
    722 waitforjob(jp)
    723 	struct job *jp;
    724 	{
    725 #if JOBS
    726 	int mypgrp = getpgrp();
    727 #endif
    728 	int status;
    729 	int st;
    730 
    731 	INTOFF;
    732 	TRACE(("waitforjob(%%%d) called\n", jp - jobtab + 1));
    733 	while (jp->state == 0) {
    734 		dowait(1, jp);
    735 	}
    736 #if JOBS
    737 	if (jp->jobctl) {
    738 #ifdef OLD_TTY_DRIVER
    739 		if (ioctl(2, TIOCSPGRP, (char *)&mypgrp) < 0)
    740 			error("TIOCSPGRP failed, errno=%d\n", errno);
    741 #else
    742 		if (tcsetpgrp(2, mypgrp) < 0)
    743 			error("tcsetpgrp failed, errno=%d\n", errno);
    744 #endif
    745 	}
    746 	if (jp->state == JOBSTOPPED)
    747 		curjob = jp - jobtab + 1;
    748 #endif
    749 	status = jp->ps[jp->nprocs - 1].status;
    750 	/* convert to 8 bits */
    751 	if (WIFEXITED(status))
    752 		st = WEXITSTATUS(status);
    753 #if JOBS
    754 	else if (WIFSTOPPED(status))
    755 		st = WSTOPSIG(status) + 128;
    756 #endif
    757 	else
    758 		st = WTERMSIG(status) + 128;
    759 #if JOBS
    760 	if (jp->jobctl) {
    761 		/*
    762 		 * This is truly gross.
    763 		 * If we're doing job control, then we did a TIOCSPGRP which
    764 		 * caused us (the shell) to no longer be in the controlling
    765 		 * session -- so we wouldn't have seen any ^C/SIGINT.  So, we
    766 		 * intuit from the subprocess exit status whether a SIGINT
    767 		 * occured, and if so interrupt ourselves.  Yuck.  - mycroft
    768 		 */
    769 		if (WIFSIGNALED(status) && WTERMSIG(status) == SIGINT)
    770 			raise(SIGINT);
    771 	}
    772 #endif
    773 	if (! JOBS || jp->state == JOBDONE)
    774 		freejob(jp);
    775 	INTON;
    776 	return st;
    777 }
    778 
    779 
    780 
    781 /*
    782  * Wait for a process to terminate.
    783  */
    784 
    785 STATIC int
    786 dowait(block, job)
    787 	int block;
    788 	struct job *job;
    789 {
    790 	int pid;
    791 	int status;
    792 	struct procstat *sp;
    793 	struct job *jp;
    794 	struct job *thisjob;
    795 	int done;
    796 	int stopped;
    797 	int core;
    798 	int sig;
    799 
    800 	TRACE(("dowait(%d) called\n", block));
    801 	do {
    802 		pid = waitproc(block, &status);
    803 		TRACE(("wait returns %d, status=%d\n", pid, status));
    804 	} while (pid == -1 && errno == EINTR);
    805 	if (pid <= 0)
    806 		return pid;
    807 	INTOFF;
    808 	thisjob = NULL;
    809 	for (jp = jobtab ; jp < jobtab + njobs ; jp++) {
    810 		if (jp->used) {
    811 			done = 1;
    812 			stopped = 1;
    813 			for (sp = jp->ps ; sp < jp->ps + jp->nprocs ; sp++) {
    814 				if (sp->pid == -1)
    815 					continue;
    816 				if (sp->pid == pid) {
    817 					TRACE(("Changing status of proc %d from 0x%x to 0x%x\n", pid, sp->status, status));
    818 					sp->status = status;
    819 					thisjob = jp;
    820 				}
    821 				if (sp->status == -1)
    822 					stopped = 0;
    823 				else if (WIFSTOPPED(sp->status))
    824 					done = 0;
    825 			}
    826 			if (stopped) {		/* stopped or done */
    827 				int state = done? JOBDONE : JOBSTOPPED;
    828 				if (jp->state != state) {
    829 					TRACE(("Job %d: changing state from %d to %d\n", jp - jobtab + 1, jp->state, state));
    830 					jp->state = state;
    831 #if JOBS
    832 					if (done && curjob == jp - jobtab + 1)
    833 						curjob = 0;		/* no current job */
    834 #endif
    835 				}
    836 			}
    837 		}
    838 	}
    839 	INTON;
    840 	if (! rootshell || ! iflag || (job && thisjob == job)) {
    841 		core = WCOREDUMP(status);
    842 #if JOBS
    843 		if (WIFSTOPPED(status)) sig = WSTOPSIG(status);
    844 		else
    845 #endif
    846 		if (WIFEXITED(status)) sig = 0;
    847 		else sig = WTERMSIG(status);
    848 
    849 		if (sig != 0 && sig != SIGINT && sig != SIGPIPE) {
    850 			if (thisjob != job)
    851 				outfmt(out2, "%d: ", pid);
    852 #if JOBS
    853 			if (sig == SIGTSTP && rootshell && iflag)
    854 				outfmt(out2, "%%%ld ",
    855 				    (long)(job - jobtab + 1));
    856 #endif
    857 			if (sig < NSIG && sys_siglist[sig])
    858 				out2str(sys_siglist[sig]);
    859 			else
    860 				outfmt(out2, "Signal %d", sig);
    861 			if (core)
    862 				out2str(" - core dumped");
    863 			out2c('\n');
    864 			flushout(&errout);
    865 		} else {
    866 			TRACE(("Not printing status: status=%d, sig=%d\n",
    867 			       status, sig));
    868 		}
    869 	} else {
    870 		TRACE(("Not printing status, rootshell=%d, job=0x%x\n", rootshell, job));
    871 		if (thisjob)
    872 			thisjob->changed = 1;
    873 	}
    874 	return pid;
    875 }
    876 
    877 
    878 
    879 /*
    880  * Do a wait system call.  If job control is compiled in, we accept
    881  * stopped processes.  If block is zero, we return a value of zero
    882  * rather than blocking.
    883  *
    884  * System V doesn't have a non-blocking wait system call.  It does
    885  * have a SIGCLD signal that is sent to a process when one of it's
    886  * children dies.  The obvious way to use SIGCLD would be to install
    887  * a handler for SIGCLD which simply bumped a counter when a SIGCLD
    888  * was received, and have waitproc bump another counter when it got
    889  * the status of a process.  Waitproc would then know that a wait
    890  * system call would not block if the two counters were different.
    891  * This approach doesn't work because if a process has children that
    892  * have not been waited for, System V will send it a SIGCLD when it
    893  * installs a signal handler for SIGCLD.  What this means is that when
    894  * a child exits, the shell will be sent SIGCLD signals continuously
    895  * until is runs out of stack space, unless it does a wait call before
    896  * restoring the signal handler.  The code below takes advantage of
    897  * this (mis)feature by installing a signal handler for SIGCLD and
    898  * then checking to see whether it was called.  If there are any
    899  * children to be waited for, it will be.
    900  *
    901  * If neither SYSV nor BSD is defined, we don't implement nonblocking
    902  * waits at all.  In this case, the user will not be informed when
    903  * a background process until the next time she runs a real program
    904  * (as opposed to running a builtin command or just typing return),
    905  * and the jobs command may give out of date information.
    906  */
    907 
    908 #ifdef SYSV
    909 STATIC int gotsigchild;
    910 
    911 STATIC int onsigchild() {
    912 	gotsigchild = 1;
    913 }
    914 #endif
    915 
    916 
    917 STATIC int
    918 waitproc(block, status)
    919 	int block;
    920 	int *status;
    921 {
    922 #ifdef BSD
    923 	int flags;
    924 
    925 #if JOBS
    926 	flags = WUNTRACED;
    927 #else
    928 	flags = 0;
    929 #endif
    930 	if (block == 0)
    931 		flags |= WNOHANG;
    932 	return wait3(status, flags, (struct rusage *)NULL);
    933 #else
    934 #ifdef SYSV
    935 	int (*save)();
    936 
    937 	if (block == 0) {
    938 		gotsigchild = 0;
    939 		save = signal(SIGCLD, onsigchild);
    940 		signal(SIGCLD, save);
    941 		if (gotsigchild == 0)
    942 			return 0;
    943 	}
    944 	return wait(status);
    945 #else
    946 	if (block == 0)
    947 		return 0;
    948 	return wait(status);
    949 #endif
    950 #endif
    951 }
    952 
    953 /*
    954  * return 1 if there are stopped jobs, otherwise 0
    955  */
    956 int job_warning = 0;
    957 int
    958 stoppedjobs()
    959 {
    960 	int jobno;
    961 	struct job *jp;
    962 
    963 	if (job_warning)
    964 		return (0);
    965 	for (jobno = 1, jp = jobtab; jobno <= njobs; jobno++, jp++) {
    966 		if (jp->used == 0)
    967 			continue;
    968 		if (jp->state == JOBSTOPPED) {
    969 			out2str("You have stopped jobs.\n");
    970 			job_warning = 2;
    971 			return (1);
    972 		}
    973 	}
    974 
    975 	return (0);
    976 }
    977 
    978 /*
    979  * Return a string identifying a command (to be printed by the
    980  * jobs command.
    981  */
    982 
    983 STATIC char *cmdnextc;
    984 STATIC int cmdnleft;
    985 #define MAXCMDTEXT	200
    986 
    987 char *
    988 commandtext(n)
    989 	union node *n;
    990 	{
    991 	char *name;
    992 
    993 	cmdnextc = name = ckmalloc(MAXCMDTEXT);
    994 	cmdnleft = MAXCMDTEXT - 4;
    995 	cmdtxt(n);
    996 	*cmdnextc = '\0';
    997 	return name;
    998 }
    999 
   1000 
   1001 STATIC void
   1002 cmdtxt(n)
   1003 	union node *n;
   1004 	{
   1005 	union node *np;
   1006 	struct nodelist *lp;
   1007 	const char *p;
   1008 	int i;
   1009 	char s[2];
   1010 
   1011 	if (n == NULL)
   1012 		return;
   1013 	switch (n->type) {
   1014 	case NSEMI:
   1015 		cmdtxt(n->nbinary.ch1);
   1016 		cmdputs("; ");
   1017 		cmdtxt(n->nbinary.ch2);
   1018 		break;
   1019 	case NAND:
   1020 		cmdtxt(n->nbinary.ch1);
   1021 		cmdputs(" && ");
   1022 		cmdtxt(n->nbinary.ch2);
   1023 		break;
   1024 	case NOR:
   1025 		cmdtxt(n->nbinary.ch1);
   1026 		cmdputs(" || ");
   1027 		cmdtxt(n->nbinary.ch2);
   1028 		break;
   1029 	case NPIPE:
   1030 		for (lp = n->npipe.cmdlist ; lp ; lp = lp->next) {
   1031 			cmdtxt(lp->n);
   1032 			if (lp->next)
   1033 				cmdputs(" | ");
   1034 		}
   1035 		break;
   1036 	case NSUBSHELL:
   1037 		cmdputs("(");
   1038 		cmdtxt(n->nredir.n);
   1039 		cmdputs(")");
   1040 		break;
   1041 	case NREDIR:
   1042 	case NBACKGND:
   1043 		cmdtxt(n->nredir.n);
   1044 		break;
   1045 	case NIF:
   1046 		cmdputs("if ");
   1047 		cmdtxt(n->nif.test);
   1048 		cmdputs("; then ");
   1049 		cmdtxt(n->nif.ifpart);
   1050 		cmdputs("...");
   1051 		break;
   1052 	case NWHILE:
   1053 		cmdputs("while ");
   1054 		goto until;
   1055 	case NUNTIL:
   1056 		cmdputs("until ");
   1057 until:
   1058 		cmdtxt(n->nbinary.ch1);
   1059 		cmdputs("; do ");
   1060 		cmdtxt(n->nbinary.ch2);
   1061 		cmdputs("; done");
   1062 		break;
   1063 	case NFOR:
   1064 		cmdputs("for ");
   1065 		cmdputs(n->nfor.var);
   1066 		cmdputs(" in ...");
   1067 		break;
   1068 	case NCASE:
   1069 		cmdputs("case ");
   1070 		cmdputs(n->ncase.expr->narg.text);
   1071 		cmdputs(" in ...");
   1072 		break;
   1073 	case NDEFUN:
   1074 		cmdputs(n->narg.text);
   1075 		cmdputs("() ...");
   1076 		break;
   1077 	case NCMD:
   1078 		for (np = n->ncmd.args ; np ; np = np->narg.next) {
   1079 			cmdtxt(np);
   1080 			if (np->narg.next)
   1081 				cmdputs(" ");
   1082 		}
   1083 		for (np = n->ncmd.redirect ; np ; np = np->nfile.next) {
   1084 			cmdputs(" ");
   1085 			cmdtxt(np);
   1086 		}
   1087 		break;
   1088 	case NARG:
   1089 		cmdputs(n->narg.text);
   1090 		break;
   1091 	case NTO:
   1092 		p = ">";  i = 1;  goto redir;
   1093 	case NAPPEND:
   1094 		p = ">>";  i = 1;  goto redir;
   1095 	case NTOFD:
   1096 		p = ">&";  i = 1;  goto redir;
   1097 	case NFROM:
   1098 		p = "<";  i = 0;  goto redir;
   1099 	case NFROMFD:
   1100 		p = "<&";  i = 0;  goto redir;
   1101 	case NFROMTO:
   1102 		p = "<>";  i = 0;  goto redir;
   1103 redir:
   1104 		if (n->nfile.fd != i) {
   1105 			s[0] = n->nfile.fd + '0';
   1106 			s[1] = '\0';
   1107 			cmdputs(s);
   1108 		}
   1109 		cmdputs(p);
   1110 		if (n->type == NTOFD || n->type == NFROMFD) {
   1111 			s[0] = n->ndup.dupfd + '0';
   1112 			s[1] = '\0';
   1113 			cmdputs(s);
   1114 		} else {
   1115 			cmdtxt(n->nfile.fname);
   1116 		}
   1117 		break;
   1118 	case NHERE:
   1119 	case NXHERE:
   1120 		cmdputs("<<...");
   1121 		break;
   1122 	default:
   1123 		cmdputs("???");
   1124 		break;
   1125 	}
   1126 }
   1127 
   1128 
   1129 
   1130 STATIC void
   1131 cmdputs(s)
   1132 	const char *s;
   1133 	{
   1134 	const char *p;
   1135 	char *q;
   1136 	char c;
   1137 	int subtype = 0;
   1138 
   1139 	if (cmdnleft <= 0)
   1140 		return;
   1141 	p = s;
   1142 	q = cmdnextc;
   1143 	while ((c = *p++) != '\0') {
   1144 		if (c == CTLESC)
   1145 			*q++ = *p++;
   1146 		else if (c == CTLVAR) {
   1147 			*q++ = '$';
   1148 			if (--cmdnleft > 0)
   1149 				*q++ = '{';
   1150 			subtype = *p++;
   1151 		} else if (c == '=' && subtype != 0) {
   1152 			*q++ = "}-+?="[(subtype & VSTYPE) - VSNORMAL];
   1153 			subtype = 0;
   1154 		} else if (c == CTLENDVAR) {
   1155 			*q++ = '}';
   1156 		} else if (c == CTLBACKQ || c == CTLBACKQ+CTLQUOTE)
   1157 			cmdnleft++;		/* ignore it */
   1158 		else
   1159 			*q++ = c;
   1160 		if (--cmdnleft <= 0) {
   1161 			*q++ = '.';
   1162 			*q++ = '.';
   1163 			*q++ = '.';
   1164 			break;
   1165 		}
   1166 	}
   1167 	cmdnextc = q;
   1168 }
   1169