1/* timeout.c - Run command line with a timeout
2 *
3 * Copyright 2013 Rob Landley <rob@landley.net>
4 *
5 * No standard
6
7USE_TIMEOUT(NEWTOY(timeout, "<2^vk:s: ", TOYFLAG_BIN))
8
9config TIMEOUT
10  bool "timeout"
11  default y
12  depends on TOYBOX_FLOAT
13  help
14    usage: timeout [-k LENGTH] [-s SIGNAL] LENGTH COMMAND...
15
16    Run command line as a child process, sending child a signal if the
17    command doesn't exit soon enough.
18
19    Length can be a decimal fraction. An optional suffix can be "m"
20    (minutes), "h" (hours), "d" (days), or "s" (seconds, the default).
21
22    -s	Send specified signal (default TERM)
23    -k	Send KILL signal if child still running this long after first signal.
24    -v	Verbose
25*/
26
27#define FOR_timeout
28#include "toys.h"
29
30GLOBALS(
31  char *s_signal;
32  char *k_timeout;
33
34  int nextsig;
35  pid_t pid;
36  struct timeval ktv;
37  struct itimerval itv;
38)
39
40static void handler(int i)
41{
42  fprintf(stderr, "timeout pid %d signal %d\n", TT.pid, TT.nextsig);
43  kill(TT.pid, TT.nextsig);
44
45  if (TT.k_timeout) {
46    TT.k_timeout = 0;
47    TT.nextsig = SIGKILL;
48    xsignal(SIGALRM, handler);
49    TT.itv.it_value = TT.ktv;
50    setitimer(ITIMER_REAL, &TT.itv, (void *)toybuf);
51  }
52}
53
54void timeout_main(void)
55{
56  // Parse early to get any errors out of the way.
57  TT.itv.it_value.tv_sec = xparsetime(*toys.optargs, 1000000, &TT.itv.it_value.tv_usec);
58
59  if (TT.k_timeout)
60    TT.ktv.tv_sec = xparsetime(TT.k_timeout, 1000000, &TT.ktv.tv_usec);
61  TT.nextsig = SIGTERM;
62  if (TT.s_signal && -1 == (TT.nextsig = sig_to_num(TT.s_signal)))
63    error_exit("bad -s: '%s'", TT.s_signal);
64
65  if (!(TT.pid = xfork())) xexec(toys.optargs+1);
66  else {
67    int status;
68
69    xsignal(SIGALRM, handler);
70    setitimer(ITIMER_REAL, &TT.itv, (void *)toybuf);
71    while (-1 == waitpid(TT.pid, &status, 0) && errno == EINTR);
72    toys.exitval = WIFEXITED(status)
73      ? WEXITSTATUS(status) : WTERMSIG(status) + 127;
74  }
75}
76