/* timeout.c - Run command line with a timeout * * Copyright 2013 Rob Landley <rob@landley.net> * * No standard USE_TIMEOUT(NEWTOY(timeout, "<2^(foreground)(preserve-status)vk:s(signal):", TOYFLAG_USR|TOYFLAG_BIN|TOYFLAG_ARGFAIL(125))) config TIMEOUT bool "timeout" default y help usage: timeout [-k DURATION] [-s SIGNAL] DURATION COMMAND... Run command line as a child process, sending child a signal if the command doesn't exit soon enough. DURATION can be a decimal fraction. An optional suffix can be "m" (minutes), "h" (hours), "d" (days), or "s" (seconds, the default). -s Send specified signal (default TERM) -k Send KILL signal if child still running this long after first signal -v Verbose --foreground Don't create new process group --preserve-status Exit with the child's exit status */ #define FOR_timeout #include "toys.h" GLOBALS( char *s, *k; int nextsig; pid_t pid; struct timeval ktv; struct itimerval itv; ) static void handler(int i) { if (FLAG(v)) fprintf(stderr, "timeout pid %d signal %d\n", TT.pid, TT.nextsig); kill(TT.pid, TT.nextsig); if (TT.k) { TT.k = 0; TT.nextsig = SIGKILL; xsignal(SIGALRM, handler); TT.itv.it_value = TT.ktv; setitimer(ITIMER_REAL, &TT.itv, (void *)toybuf); } } // timeval inexplicably makes up a new type for microseconds, despite timespec's // nanoseconds field (needing to store 1000* the range) using "long". Bravo. void xparsetimeval(char *s, struct timeval *tv) { long ll; tv->tv_sec = xparsetime(s, 6, &ll); tv->tv_usec = ll; } void timeout_main(void) { toys.exitval = 125; // Parse early to get any errors out of the way. xparsetimeval(*toys.optargs, &TT.itv.it_value); if (TT.k) xparsetimeval(TT.k, &TT.ktv); TT.nextsig = SIGTERM; if (TT.s && -1 == (TT.nextsig = sig_to_num(TT.s))) error_exit("bad -s: '%s'", TT.s); if (!FLAG(foreground)) setpgid(0, 0); if (!(TT.pid = XVFORK())) xexec(toys.optargs+1); else { int status; xsignal(SIGALRM, handler); setitimer(ITIMER_REAL, &TT.itv, (void *)toybuf); while (-1 == waitpid(TT.pid, &status, 0) && errno == EINTR); if (WIFEXITED(status)) toys.exitval = WEXITSTATUS(status); else if (WTERMSIG(status)==SIGKILL) toys.exitval = 137; else toys.exitval = FLAG(preserve_status) ? 128+WTERMSIG(status) : 124; } }