#include <sys/resource.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/wait.h>
#include <ctype.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "sh.h"
#include "tty.h"
#define PRUNNING 0
#define PEXITED 1
#define PSIGNALLED 2
#define PSTOPPED 3
typedef struct proc Proc;
struct proc {
Proc *next;
int state;
int status;
pid_t pid;
char command[48];
};
#define JP_NONE 0
#define JP_SHORT 1
#define JP_MEDIUM 2
#define JP_LONG 3
#define JP_PGRP 4
#define PJ_ON_FRONT 0
#define PJ_PAST_STOPPED 1
#define JF_STARTED 0x001
#define JF_WAITING 0x002
#define JF_W_ASYNCNOTIFY 0x004
#define JF_XXCOM 0x008
#define JF_FG 0x010
#define JF_SAVEDTTY 0x020
#define JF_CHANGED 0x040
#define JF_KNOWN 0x080
#define JF_ZOMBIE 0x100
#define JF_REMOVE 0x200
#define JF_USETTYMODE 0x400
#define JF_SAVEDTTYPGRP 0x800
#define JF_PIPEFAIL 0x1000
typedef struct job Job;
struct job {
Job *next;
int job;
int flags;
int state;
int status;
pid_t pgrp;
pid_t ppid;
int age;
struct timeval systime;
struct timeval usrtime;
Proc *proc_list;
Proc *last_proc;
Coproc_id coproc_id;
struct termios ttystate;
pid_t saved_ttypgrp;
};
#define JW_NONE 0x00
#define JW_INTERRUPT 0x01
#define JW_ASYNCNOTIFY 0x02
#define JW_STOPPEDWAIT 0x04
#define JL_OK 0
#define JL_NOSUCH 1
#define JL_AMBIG 2
#define JL_INVALID 3
static const char *const lookup_msgs[] = {
null,
"no such job",
"ambiguous",
"argument must be %job or process id",
NULL
};
struct timeval j_systime, j_usrtime;
static Job *job_list;
static Job *last_job;
static Job *async_job;
static pid_t async_pid;
static int nzombie;
int njobs;
static int child_max;
static volatile sig_atomic_t held_sigchld;
static struct shf *shl_j;
static int ttypgrp_ok;
static pid_t restore_ttypgrp = -1;
static pid_t our_pgrp;
static int const tt_sigs[] = { SIGTSTP, SIGTTIN, SIGTTOU };
static void j_set_async(Job *);
static void j_startjob(Job *);
static int j_waitj(Job *, int, const char *);
static void j_sigchld(int);
static void j_print(Job *, int, struct shf *);
static Job *j_lookup(const char *, int *);
static Job *new_job(void);
static Proc *new_proc(void);
static void check_job(Job *);
static void put_job(Job *, int);
static void remove_job(Job *, const char *);
static int kill_job(Job *, int);
void
j_init(int mflagset)
{
child_max = CHILD_MAX;
sigemptyset(&sm_default);
sigprocmask(SIG_SETMASK, &sm_default, NULL);
sigemptyset(&sm_sigchld);
sigaddset(&sm_sigchld, SIGCHLD);
setsig(&sigtraps[SIGCHLD], j_sigchld,
SS_RESTORE_ORIG|SS_FORCE|SS_SHTRAP);
if (!mflagset && Flag(FTALKING))
Flag(FMONITOR) = 1;
shl_j = shf_fdopen(2, SHF_WR, NULL);
if (Flag(FMONITOR) || Flag(FTALKING)) {
int i;
for (i = NELEM(tt_sigs); --i >= 0; ) {
sigtraps[tt_sigs[i]].flags |= TF_SHELL_USES;
setsig(&sigtraps[tt_sigs[i]], SIG_IGN,
SS_RESTORE_IGN|SS_FORCE);
}
}
if (Flag(FMONITOR))
j_change();
else if (Flag(FTALKING))
tty_init(true);
}
void
j_suspend(void)
{
struct sigaction sa, osa;
if (ttypgrp_ok) {
tcsetattr(tty_fd, TCSADRAIN, &tty_state);
if (restore_ttypgrp >= 0) {
if (tcsetpgrp(tty_fd, restore_ttypgrp) == -1) {
warningf(false, "%s: tcsetpgrp() failed: %s",
__func__, strerror(errno));
} else {
if (setpgid(0, restore_ttypgrp) == -1) {
warningf(false,
"%s: setpgid() failed: %s",
__func__, strerror(errno));
}
}
}
}
memset(&sa, 0, sizeof(sa));
sigemptyset(&sa.sa_mask);
sa.sa_handler = SIG_DFL;
sigaction(SIGTSTP, &sa, &osa);
kill(0, SIGTSTP);
sigaction(SIGTSTP, &osa, NULL);
if (ttypgrp_ok) {
if (restore_ttypgrp >= 0) {
if (setpgid(0, kshpid) == -1) {
warningf(false, "%s: setpgid() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
} else {
if (tcsetpgrp(tty_fd, kshpid) == -1) {
warningf(false,
"%s: tcsetpgrp() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
}
}
}
tty_init(true);
}
}
void
j_exit(void)
{
Job *j;
int killed = 0;
for (j = job_list; j != NULL; j = j->next) {
if (j->ppid == procpid &&
(j->state == PSTOPPED ||
(j->state == PRUNNING &&
((j->flags & JF_FG) ||
(Flag(FLOGIN) && !Flag(FNOHUP) && procpid == kshpid))))) {
killed = 1;
if (j->pgrp == 0)
kill_job(j, SIGHUP);
else
killpg(j->pgrp, SIGHUP);
if (j->state == PSTOPPED) {
if (j->pgrp == 0)
kill_job(j, SIGCONT);
else
killpg(j->pgrp, SIGCONT);
}
}
}
if (killed)
sleep(1);
j_notify();
if (kshpid == procpid && restore_ttypgrp >= 0) {
tcsetpgrp(tty_fd, restore_ttypgrp);
setpgid(0, restore_ttypgrp);
}
if (Flag(FMONITOR)) {
Flag(FMONITOR) = 0;
j_change();
}
}
void
j_change(void)
{
int i;
if (Flag(FMONITOR)) {
int use_tty;
if (Flag(FTALKING)) {
use_tty = 1;
tty_init(false);
} else
use_tty = 0;
ttypgrp_ok = use_tty && tty_fd >= 0 && tty_devtty;
if (ttypgrp_ok && (our_pgrp = getpgrp()) < 0) {
warningf(false, "%s: getpgrp() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
}
if (ttypgrp_ok) {
setsig(&sigtraps[SIGTTIN], SIG_DFL,
SS_RESTORE_ORIG|SS_FORCE);
while (1) {
pid_t ttypgrp;
if ((ttypgrp = tcgetpgrp(tty_fd)) == -1) {
warningf(false,
"%s: tcgetpgrp() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
break;
}
if (ttypgrp == our_pgrp)
break;
kill(0, SIGTTIN);
}
}
for (i = NELEM(tt_sigs); --i >= 0; )
setsig(&sigtraps[tt_sigs[i]], SIG_IGN,
SS_RESTORE_DFL|SS_FORCE);
if (ttypgrp_ok && our_pgrp != kshpid) {
if (setpgid(0, kshpid) == -1) {
warningf(false, "%s: setpgid() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
} else {
if (tcsetpgrp(tty_fd, kshpid) == -1) {
warningf(false,
"%s: tcsetpgrp() failed: %s",
__func__, strerror(errno));
ttypgrp_ok = 0;
} else
restore_ttypgrp = our_pgrp;
our_pgrp = kshpid;
}
}
if (use_tty) {
if (!ttypgrp_ok)
warningf(false,
"warning: won't have full job control");
}
if (tty_fd >= 0)
tcgetattr(tty_fd, &tty_state);
} else {
ttypgrp_ok = 0;
if (Flag(FTALKING))
for (i = NELEM(tt_sigs); --i >= 0; )
setsig(&sigtraps[tt_sigs[i]], SIG_IGN,
SS_RESTORE_IGN|SS_FORCE);
else
for (i = NELEM(tt_sigs); --i >= 0; ) {
if (sigtraps[tt_sigs[i]].flags &
(TF_ORIG_IGN | TF_ORIG_DFL))
setsig(&sigtraps[tt_sigs[i]],
(sigtraps[tt_sigs[i]].flags & TF_ORIG_IGN) ?
SIG_IGN : SIG_DFL,
SS_RESTORE_ORIG|SS_FORCE);
}
if (!Flag(FTALKING))
tty_close();
}
}
int
exchild(struct op *t, int flags, volatile int *xerrok,
int close_fd)
{
static Proc *last_proc;
int i;
sigset_t omask;
Proc *p;
Job *j;
int rv = 0;
int forksleep;
int ischild;
if (flags & XEXEC)
return execute(t, flags & (XEXEC | XERROK), xerrok);
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
p = new_proc();
p->next = NULL;
p->state = PRUNNING;
p->status = 0;
p->pid = 0;
if (flags&XPIPEI) {
if (!last_job)
internal_errorf("%s: XPIPEI and no last_job - pid %d",
__func__, (int) procpid);
j = last_job;
last_proc->next = p;
last_proc = p;
} else {
j = new_job();
j->flags = (flags & XXCOM) ? JF_XXCOM :
((flags & XBGND) ? 0 : (JF_FG|JF_USETTYMODE));
if (Flag(FPIPEFAIL))
j->flags |= JF_PIPEFAIL;
timerclear(&j->usrtime);
timerclear(&j->systime);
j->state = PRUNNING;
j->pgrp = 0;
j->ppid = procpid;
j->age = ++njobs;
j->proc_list = p;
j->coproc_id = 0;
last_job = j;
last_proc = p;
put_job(j, PJ_PAST_STOPPED);
}
snptreef(p->command, sizeof(p->command), "%T", t);
forksleep = 1;
while ((i = fork()) == -1 && errno == EAGAIN && forksleep < 32) {
if (intrsig)
break;
sleep(forksleep);
forksleep <<= 1;
}
if (i == -1) {
kill_job(j, SIGKILL);
remove_job(j, "fork failed");
sigprocmask(SIG_SETMASK, &omask, NULL);
errorf("cannot fork - try again");
}
ischild = i == 0;
if (ischild)
p->pid = procpid = getpid();
else
p->pid = i;
if (Flag(FMONITOR) && !(flags&XXCOM)) {
int dotty = 0;
if (j->pgrp == 0) {
j->pgrp = p->pid;
dotty = 1;
}
setpgid(p->pid, j->pgrp);
if (ttypgrp_ok && dotty && !(flags & XBGND))
tcsetpgrp(tty_fd, j->pgrp);
}
if (close_fd >= 0 && (((flags & XPCLOSE) && !ischild) ||
((flags & XCCLOSE) && ischild)))
close(close_fd);
if (ischild) {
if (flags & XCOPROC)
coproc_cleanup(false);
sigprocmask(SIG_SETMASK, &omask, NULL);
cleanup_parents_env();
if (Flag(FMONITOR) && !(flags & XXCOM)) {
for (i = NELEM(tt_sigs); --i >= 0; )
setsig(&sigtraps[tt_sigs[i]], SIG_DFL,
SS_RESTORE_DFL|SS_FORCE);
}
if (Flag(FBGNICE) && (flags & XBGND))
nice(4);
if ((flags & XBGND) && !Flag(FMONITOR)) {
setsig(&sigtraps[SIGINT], SIG_IGN,
SS_RESTORE_IGN|SS_FORCE);
setsig(&sigtraps[SIGQUIT], SIG_IGN,
SS_RESTORE_IGN|SS_FORCE);
if (!(flags & (XPIPEI | XCOPROC))) {
int fd = open("/dev/null", O_RDONLY);
if (fd != 0) {
(void) ksh_dup2(fd, 0, true);
close(fd);
}
}
}
remove_job(j, "child");
nzombie = 0;
ttypgrp_ok = 0;
Flag(FMONITOR) = 0;
Flag(FTALKING) = 0;
tty_close();
cleartraps();
execute(t, (flags & XERROK) | XEXEC, NULL);
internal_warningf("%s: execute() returned", __func__);
unwind(LLEAVE);
}
change_random();
if (!(flags & XPIPEO)) {
j_startjob(j);
if (flags & XCOPROC) {
j->coproc_id = coproc.id;
coproc.njobs++;
coproc.job = (void *) j;
}
if (flags & XBGND) {
j_set_async(j);
if (Flag(FTALKING)) {
shf_fprintf(shl_out, "[%d]", j->job);
for (p = j->proc_list; p; p = p->next)
shf_fprintf(shl_out, " %d", p->pid);
shf_putchar('\n', shl_out);
shf_flush(shl_out);
}
} else
rv = j_waitj(j, JW_NONE, "jw:last proc");
}
sigprocmask(SIG_SETMASK, &omask, NULL);
return rv;
}
void
startlast(void)
{
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
if (last_job) {
last_job->flags |= JF_WAITING;
j_startjob(last_job);
}
sigprocmask(SIG_SETMASK, &omask, NULL);
}
int
waitlast(void)
{
int rv;
Job *j;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
j = last_job;
if (!j || !(j->flags & JF_STARTED)) {
if (!j)
warningf(true, "%s: no last job", __func__);
else
internal_warningf("%s: not started", __func__);
sigprocmask(SIG_SETMASK, &omask, NULL);
return 125;
}
rv = j_waitj(j, JW_NONE, "jw:waitlast");
sigprocmask(SIG_SETMASK, &omask, NULL);
return rv;
}
int
waitfor(const char *cp, int *sigp)
{
int rv;
Job *j;
int ecode;
int flags = JW_INTERRUPT|JW_ASYNCNOTIFY;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
*sigp = 0;
if (cp == NULL) {
for (j = job_list; j; j = j->next)
if (j->ppid == procpid && j->state == PRUNNING)
break;
if (!j) {
sigprocmask(SIG_SETMASK, &omask, NULL);
return -1;
}
} else if ((j = j_lookup(cp, &ecode))) {
flags &= ~JW_ASYNCNOTIFY;
if (j->ppid != procpid) {
sigprocmask(SIG_SETMASK, &omask, NULL);
return -1;
}
} else {
sigprocmask(SIG_SETMASK, &omask, NULL);
if (ecode != JL_NOSUCH)
bi_errorf("%s: %s", cp, lookup_msgs[ecode]);
return -1;
}
rv = j_waitj(j, flags, "jw:waitfor");
sigprocmask(SIG_SETMASK, &omask, NULL);
if (rv < 0)
*sigp = 128 + -rv;
return rv;
}
int
j_kill(const char *cp, int sig)
{
Job *j;
int rv = 0;
int ecode;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
if ((j = j_lookup(cp, &ecode)) == NULL) {
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("%s: %s", cp, lookup_msgs[ecode]);
return 1;
}
if (j->pgrp == 0) {
if (kill_job(j, sig) < 0) {
bi_errorf("%s: %s", cp, strerror(errno));
rv = 1;
}
} else {
if (j->state == PSTOPPED && (sig == SIGTERM || sig == SIGHUP))
(void) killpg(j->pgrp, SIGCONT);
if (killpg(j->pgrp, sig) == -1) {
bi_errorf("%s: %s", cp, strerror(errno));
rv = 1;
}
}
sigprocmask(SIG_SETMASK, &omask, NULL);
return rv;
}
int
j_resume(const char *cp, int bg)
{
Job *j;
Proc *p;
int ecode;
int running;
int rv = 0;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
if ((j = j_lookup(cp, &ecode)) == NULL) {
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("%s: %s", cp, lookup_msgs[ecode]);
return 1;
}
if (j->pgrp == 0) {
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("job not job-controlled");
return 1;
}
if (bg)
shprintf("[%d] ", j->job);
running = 0;
for (p = j->proc_list; p != NULL; p = p->next) {
if (p->state == PSTOPPED) {
p->state = PRUNNING;
p->status = 0;
running = 1;
}
shprintf("%s%s", p->command, p->next ? "| " : "");
}
shprintf("\n");
shf_flush(shl_stdout);
if (running)
j->state = PRUNNING;
put_job(j, PJ_PAST_STOPPED);
if (bg)
j_set_async(j);
else {
if (j->state == PRUNNING) {
if (ttypgrp_ok && (j->flags & JF_SAVEDTTY))
tcsetattr(tty_fd, TCSADRAIN, &j->ttystate);
if (ttypgrp_ok &&
tcsetpgrp(tty_fd, (j->flags & JF_SAVEDTTYPGRP) ?
j->saved_ttypgrp : j->pgrp) == -1) {
if (j->flags & JF_SAVEDTTY)
tcsetattr(tty_fd, TCSADRAIN, &tty_state);
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("1st tcsetpgrp(%d, %d) failed: %s",
tty_fd,
(int) ((j->flags & JF_SAVEDTTYPGRP) ?
j->saved_ttypgrp : j->pgrp),
strerror(errno));
return 1;
}
}
j->flags |= JF_FG;
j->flags &= ~JF_KNOWN;
if (j == async_job)
async_job = NULL;
}
if (j->state == PRUNNING && killpg(j->pgrp, SIGCONT) == -1) {
int err = errno;
if (!bg) {
j->flags &= ~JF_FG;
if (ttypgrp_ok && (j->flags & JF_SAVEDTTY))
tcsetattr(tty_fd, TCSADRAIN, &tty_state);
if (ttypgrp_ok && tcsetpgrp(tty_fd, our_pgrp) == -1) {
warningf(true,
"fg: 2nd tcsetpgrp(%d, %d) failed: %s",
tty_fd, (int) our_pgrp,
strerror(errno));
}
}
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("cannot continue job %s: %s",
cp, strerror(err));
return 1;
}
if (!bg) {
if (ttypgrp_ok) {
j->flags &= ~(JF_SAVEDTTY | JF_SAVEDTTYPGRP);
}
rv = j_waitj(j, JW_NONE, "jw:resume");
}
sigprocmask(SIG_SETMASK, &omask, NULL);
return rv;
}
int
j_stopped_running(void)
{
Job *j;
int which = 0;
for (j = job_list; j != NULL; j = j->next) {
if (j->ppid == procpid && j->state == PSTOPPED)
which |= 1;
if (Flag(FLOGIN) && !Flag(FNOHUP) && procpid == kshpid &&
j->ppid == procpid && j->state == PRUNNING)
which |= 2;
}
if (which) {
shellf("You have %s%s%s jobs\n",
which & 1 ? "stopped" : "",
which == 3 ? " and " : "",
which & 2 ? "running" : "");
return 1;
}
return 0;
}
int
j_njobs(void)
{
Job *j;
int nj = 0;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
for (j = job_list; j; j = j->next)
nj++;
sigprocmask(SIG_SETMASK, &omask, NULL);
return nj;
}
int
j_jobs(const char *cp, int slp,
int nflag)
{
Job *j, *tmp;
int how;
int zflag = 0;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
if (nflag < 0) {
nflag = 0;
zflag = 1;
}
if (cp) {
int ecode;
if ((j = j_lookup(cp, &ecode)) == NULL) {
sigprocmask(SIG_SETMASK, &omask, NULL);
bi_errorf("%s: %s", cp, lookup_msgs[ecode]);
return 1;
}
} else
j = job_list;
how = slp == 0 ? JP_MEDIUM : (slp == 1 ? JP_LONG : JP_PGRP);
for (; j; j = j->next) {
if ((!(j->flags & JF_ZOMBIE) || zflag) &&
(!nflag || (j->flags & JF_CHANGED))) {
j_print(j, how, shl_stdout);
if (j->state == PEXITED || j->state == PSIGNALLED)
j->flags |= JF_REMOVE;
}
if (cp)
break;
}
for (j = job_list; j; j = tmp) {
tmp = j->next;
if (j->flags & JF_REMOVE)
remove_job(j, "jobs");
}
sigprocmask(SIG_SETMASK, &omask, NULL);
return 0;
}
void
j_notify(void)
{
Job *j, *tmp;
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
for (j = job_list; j; j = j->next) {
if (Flag(FMONITOR) && (j->flags & JF_CHANGED))
j_print(j, JP_MEDIUM, shl_out);
if (j->state == PEXITED || j->state == PSIGNALLED)
j->flags |= JF_REMOVE;
}
for (j = job_list; j; j = tmp) {
tmp = j->next;
if (j->flags & JF_REMOVE)
remove_job(j, "notify");
}
shf_flush(shl_out);
sigprocmask(SIG_SETMASK, &omask, NULL);
}
pid_t
j_async(void)
{
sigset_t omask;
sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
if (async_job)
async_job->flags |= JF_KNOWN;
sigprocmask(SIG_SETMASK, &omask, NULL);
return async_pid;
}
static void
j_set_async(Job *j)
{
Job *jl, *oldest;
if (async_job && (async_job->flags & (JF_KNOWN|JF_ZOMBIE)) == JF_ZOMBIE)
remove_job(async_job, "async");
if (!(j->flags & JF_STARTED)) {
internal_warningf("%s: job not started", __func__);
return;
}
async_job = j;
async_pid = j->last_proc->pid;
while (nzombie > child_max) {
oldest = NULL;
for (jl = job_list; jl; jl = jl->next)
if (jl != async_job && (jl->flags & JF_ZOMBIE) &&
(!oldest || jl->age < oldest->age))
oldest = jl;
if (!oldest) {
if (!(async_job->flags & JF_ZOMBIE) || nzombie != 1) {
internal_warningf("%s: bad nzombie (%d)",
__func__, nzombie);
nzombie = 0;
}
break;
}
remove_job(oldest, "zombie");
}
}
static void
j_startjob(Job *j)
{
Proc *p;
j->flags |= JF_STARTED;
for (p = j->proc_list; p->next; p = p->next)
;
j->last_proc = p;
if (held_sigchld) {
held_sigchld = 0;
kill(procpid, SIGCHLD);
}
}
static int
j_waitj(Job *j,
int flags,
const char *where)
{
int rv;
j->flags |= JF_WAITING;
if (flags & JW_ASYNCNOTIFY)
j->flags |= JF_W_ASYNCNOTIFY;
if (!Flag(FMONITOR))
flags |= JW_STOPPEDWAIT;
while ((volatile int) j->state == PRUNNING ||
((flags & JW_STOPPEDWAIT) && (volatile int) j->state == PSTOPPED)) {
sigsuspend(&sm_default);
if (fatal_trap) {
int oldf = j->flags & (JF_WAITING|JF_W_ASYNCNOTIFY);
j->flags &= ~(JF_WAITING|JF_W_ASYNCNOTIFY);
runtraps(TF_FATAL);
j->flags |= oldf;
}
if ((flags & JW_INTERRUPT) && (rv = trap_pending())) {
j->flags &= ~(JF_WAITING|JF_W_ASYNCNOTIFY);
return -rv;
}
}
j->flags &= ~(JF_WAITING|JF_W_ASYNCNOTIFY);
if (j->flags & JF_FG) {
int status;
j->flags &= ~JF_FG;
if (Flag(FMONITOR) && ttypgrp_ok && j->pgrp) {
if (j->state == PSTOPPED &&
(j->saved_ttypgrp = tcgetpgrp(tty_fd)) >= 0)
j->flags |= JF_SAVEDTTYPGRP;
if (tcsetpgrp(tty_fd, our_pgrp) == -1) {
warningf(true,
"%s: tcsetpgrp(%d, %d) failed: %s",
__func__, tty_fd, (int)our_pgrp,
strerror(errno));
}
if (j->state == PSTOPPED) {
j->flags |= JF_SAVEDTTY;
tcgetattr(tty_fd, &j->ttystate);
}
}
if (tty_fd >= 0) {
if (j->state == PEXITED && j->status == 0 &&
(j->flags & JF_USETTYMODE)) {
tcgetattr(tty_fd, &tty_state);
} else {
tcsetattr(tty_fd, TCSADRAIN, &tty_state);
if (j->state == PSTOPPED)
j->flags &= ~JF_USETTYMODE;
}
}
status = j->last_proc->status;
if (Flag(FMONITOR) && j->state == PSIGNALLED &&
WIFSIGNALED(status) &&
(sigtraps[WTERMSIG(status)].flags & TF_TTY_INTR))
trapsig(WTERMSIG(status));
}
j_usrtime = j->usrtime;
j_systime = j->systime;
if (j->flags & JF_PIPEFAIL) {
Proc *p;
int status;
rv = 0;
for (p = j->proc_list; p != NULL; p = p->next) {
switch (p->state) {
case PEXITED:
status = WEXITSTATUS(p->status);
break;
case PSIGNALLED:
status = 128 + WTERMSIG(p->status);
break;
default:
status = 0;
break;
}
if (status)
rv = status;
}
} else
rv = j->status;
if (!(flags & JW_ASYNCNOTIFY) &&
(!Flag(FMONITOR) || j->state != PSTOPPED)) {
j_print(j, JP_SHORT, shl_out);
shf_flush(shl_out);
}
if (j->state != PSTOPPED &&
(!Flag(FMONITOR) || !(flags & JW_ASYNCNOTIFY)))
remove_job(j, where);
return rv;
}
static void
j_sigchld(int sig)
{
int errno_ = errno;
Job *j;
Proc *p = NULL;
int pid;
int status;
struct rusage ru0, ru1;
for (j = job_list; j; j = j->next)
if (j->ppid == procpid && !(j->flags & JF_STARTED)) {
held_sigchld = 1;
goto finished;
}
getrusage(RUSAGE_CHILDREN, &ru0);
do {
pid = waitpid(-1, &status, (WNOHANG|WUNTRACED));
if (pid <= 0)
break;
getrusage(RUSAGE_CHILDREN, &ru1);
for (j = job_list; j != NULL; j = j->next)
for (p = j->proc_list; p != NULL; p = p->next)
if (p->pid == pid)
goto found;
found:
if (j == NULL) {
ru0 = ru1;
continue;
}
timeradd(&j->usrtime, &ru1.ru_utime, &j->usrtime);
timersub(&j->usrtime, &ru0.ru_utime, &j->usrtime);
timeradd(&j->systime, &ru1.ru_stime, &j->systime);
timersub(&j->systime, &ru0.ru_stime, &j->systime);
ru0 = ru1;
p->status = status;
if (WIFSTOPPED(status))
p->state = PSTOPPED;
else if (WIFSIGNALED(status))
p->state = PSIGNALLED;
else
p->state = PEXITED;
check_job(j);
} while (1);
finished:
errno = errno_;
}
static void
check_job(Job *j)
{
int jstate;
Proc *p;
if (!(j->flags & JF_STARTED)) {
internal_warningf("%s: job started (flags 0x%x)",
__func__, j->flags);
return;
}
jstate = PRUNNING;
for (p=j->proc_list; p != NULL; p = p->next) {
if (p->state == PRUNNING)
return;
if (p->state > jstate)
jstate = p->state;
}
j->state = jstate;
switch (j->last_proc->state) {
case PEXITED:
j->status = WEXITSTATUS(j->last_proc->status);
break;
case PSIGNALLED:
j->status = 128 + WTERMSIG(j->last_proc->status);
break;
default:
j->status = 0;
break;
}
if (j->state == PEXITED || j->state == PSIGNALLED) {
if (coproc.job == j) {
coproc.job = NULL;
coproc_write_close(coproc.write);
}
if (j->coproc_id && j->coproc_id == coproc.id &&
--coproc.njobs == 0)
coproc_readw_close(coproc.read);
}
j->flags |= JF_CHANGED;
if (Flag(FMONITOR) && !(j->flags & JF_XXCOM)) {
if (j->state == PSTOPPED)
put_job(j, PJ_ON_FRONT);
if (Flag(FNOTIFY) &&
(j->flags & (JF_WAITING|JF_W_ASYNCNOTIFY)) != JF_WAITING) {
{
struct env *ep;
int fd = 2;
for (ep = genv; ep; ep = ep->oenv)
if (ep->savefd && ep->savefd[2])
fd = ep->savefd[2];
shf_reopen(fd, SHF_WR, shl_j);
}
j_print(j, JP_MEDIUM, shl_j);
shf_flush(shl_j);
if (!(j->flags & JF_WAITING) && j->state != PSTOPPED)
remove_job(j, "notify");
}
}
if (!Flag(FMONITOR) && !(j->flags & (JF_WAITING|JF_FG)) &&
j->state != PSTOPPED) {
if (j == async_job || (j->flags & JF_KNOWN)) {
j->flags |= JF_ZOMBIE;
j->job = -1;
nzombie++;
} else
remove_job(j, "checkjob");
}
}
static void
j_print(Job *j, int how, struct shf *shf)
{
Proc *p;
int state;
int status;
int coredumped;
char jobchar = ' ';
char buf[64];
const char *filler;
int output = 0;
if (how == JP_PGRP) {
shf_fprintf(shf, "%d\n", j->pgrp ? j->pgrp :
(j->last_proc ? j->last_proc->pid : 0));
return;
}
j->flags &= ~JF_CHANGED;
filler = j->job > 10 ? "\n " : "\n ";
if (j == job_list)
jobchar = '+';
else if (j == job_list->next)
jobchar = '-';
for (p = j->proc_list; p != NULL;) {
coredumped = 0;
switch (p->state) {
case PRUNNING:
strlcpy(buf, "Running", sizeof buf);
break;
case PSTOPPED:
strlcpy(buf, sigtraps[WSTOPSIG(p->status)].mess,
sizeof buf);
break;
case PEXITED:
if (how == JP_SHORT)
buf[0] = '\0';
else if (WEXITSTATUS(p->status) == 0)
strlcpy(buf, "Done", sizeof buf);
else
shf_snprintf(buf, sizeof(buf), "Done (%d)",
WEXITSTATUS(p->status));
break;
case PSIGNALLED:
if (WCOREDUMP(p->status))
coredumped = 1;
if (how == JP_SHORT && !coredumped &&
(WTERMSIG(p->status) == SIGINT ||
WTERMSIG(p->status) == SIGPIPE)) {
buf[0] = '\0';
} else
strlcpy(buf, sigtraps[WTERMSIG(p->status)].mess,
sizeof buf);
break;
}
if (how != JP_SHORT) {
if (p == j->proc_list)
shf_fprintf(shf, "[%d] %c ", j->job, jobchar);
else
shf_fprintf(shf, "%s", filler);
}
if (how == JP_LONG)
shf_fprintf(shf, "%5d ", p->pid);
if (how == JP_SHORT) {
if (buf[0]) {
output = 1;
shf_fprintf(shf, "%s%s ",
buf, coredumped ? " (core dumped)" : "");
}
} else {
output = 1;
shf_fprintf(shf, "%-20s %s%s%s", buf, p->command,
p->next ? "|" : "",
coredumped ? " (core dumped)" : "");
}
state = p->state;
status = p->status;
p = p->next;
while (p && p->state == state && p->status == status) {
if (how == JP_LONG)
shf_fprintf(shf, "%s%5d %-20s %s%s", filler, p->pid,
" ", p->command, p->next ? "|" : "");
else if (how == JP_MEDIUM)
shf_fprintf(shf, " %s%s", p->command,
p->next ? "|" : "");
p = p->next;
}
}
if (output)
shf_fprintf(shf, "\n");
}
static Job *
j_lookup(const char *cp, int *ecodep)
{
Job *j, *last_match;
const char *errstr;
Proc *p;
int len, job = 0;
if (digit(*cp)) {
job = strtonum(cp, 1, INT_MAX, &errstr);
if (errstr) {
if (ecodep)
*ecodep = JL_NOSUCH;
return NULL;
}
for (j = job_list; j != NULL; j = j->next)
if (j->last_proc && j->last_proc->pid == job)
return j;
for (j = job_list; j != NULL; j = j->next)
if (j->pgrp && j->pgrp == job)
return j;
if (ecodep)
*ecodep = JL_NOSUCH;
return NULL;
}
if (*cp != '%') {
if (ecodep)
*ecodep = JL_INVALID;
return NULL;
}
switch (*++cp) {
case '\0':
case '+':
case '%':
if (job_list != NULL)
return job_list;
break;
case '-':
if (job_list != NULL && job_list->next)
return job_list->next;
break;
case '0': case '1': case '2': case '3': case '4':
case '5': case '6': case '7': case '8': case '9':
job = strtonum(cp, 1, INT_MAX, &errstr);
if (errstr)
break;
for (j = job_list; j != NULL; j = j->next)
if (j->job == job)
return j;
break;
case '?':
last_match = NULL;
for (j = job_list; j != NULL; j = j->next)
for (p = j->proc_list; p != NULL; p = p->next)
if (strstr(p->command, cp+1) != NULL) {
if (last_match) {
if (ecodep)
*ecodep = JL_AMBIG;
return NULL;
}
last_match = j;
}
if (last_match)
return last_match;
break;
default:
len = strlen(cp);
last_match = NULL;
for (j = job_list; j != NULL; j = j->next)
if (strncmp(cp, j->proc_list->command, len) == 0) {
if (last_match) {
if (ecodep)
*ecodep = JL_AMBIG;
return NULL;
}
last_match = j;
}
if (last_match)
return last_match;
break;
}
if (ecodep)
*ecodep = JL_NOSUCH;
return NULL;
}
static Job *free_jobs;
static Proc *free_procs;
static Job *
new_job(void)
{
int i;
Job *newj, *j;
if (free_jobs != NULL) {
newj = free_jobs;
free_jobs = free_jobs->next;
} else
newj = alloc(sizeof(Job), APERM);
for (i = 1; ; i++) {
for (j = job_list; j && j->job != i; j = j->next)
;
if (j == NULL)
break;
}
newj->job = i;
return newj;
}
static Proc *
new_proc(void)
{
Proc *p;
if (free_procs != NULL) {
p = free_procs;
free_procs = free_procs->next;
} else
p = alloc(sizeof(Proc), APERM);
return p;
}
static void
remove_job(Job *j, const char *where)
{
Proc *p, *tmp;
Job **prev, *curr;
prev = &job_list;
curr = *prev;
for (; curr != NULL && curr != j; prev = &curr->next, curr = *prev)
;
if (curr != j) {
internal_warningf("%s: job not found (%s)", __func__, where);
return;
}
*prev = curr->next;
for (p = j->proc_list; p != NULL; ) {
tmp = p;
p = p->next;
tmp->next = free_procs;
free_procs = tmp;
}
if ((j->flags & JF_ZOMBIE) && j->ppid == procpid)
--nzombie;
j->next = free_jobs;
free_jobs = j;
if (j == last_job)
last_job = NULL;
if (j == async_job)
async_job = NULL;
}
static void
put_job(Job *j, int where)
{
Job **prev, *curr;
prev = &job_list;
curr = job_list;
for (; curr && curr != j; prev = &curr->next, curr = *prev)
;
if (curr == j)
*prev = curr->next;
switch (where) {
case PJ_ON_FRONT:
j->next = job_list;
job_list = j;
break;
case PJ_PAST_STOPPED:
prev = &job_list;
curr = job_list;
for (; curr && curr->state == PSTOPPED; prev = &curr->next,
curr = *prev)
;
j->next = curr;
*prev = j;
break;
}
}
static int
kill_job(Job *j, int sig)
{
Proc *p;
int rval = 0;
for (p = j->proc_list; p != NULL; p = p->next)
if (p->pid != 0)
if (kill(p->pid, sig) == -1)
rval = -1;
return rval;
}