Add timeout, factoring out common code from sleep.

2 files changed, 82 insertions, 21 deletions

diff --git a/toys/other/timeout.c b/toys/other/timeout.c
new file mode 100644
index 00000000..4b8a87df
--- /dev/null
+++ b/toys/other/timeout.c
@@ -0,0 +1,74 @@
+/* timeout.c - Run command line with a timeout
+ *
+ * Copyright 2013 Rob Landley <rob@landley.net>
+ *
+ * No standard
+
+USE_TIMEOUT(NEWTOY(timeout, "<2^k:s: ", TOYFLAG_BIN))
+
+config TIMEOUT
+  bool "timeout"
+  default y
+  depends on TOYBOX_FLOAT
+  help
+    usage: timeout [-k LENGTH] [-s SIGNAL] LENGTH COMMAND...
+
+    Run command line as a child process, sending child a signal if the
+    command doesn't exit soon enough.
+
+    Length 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.
+*/
+
+#define FOR_timeout
+#include "toys.h"
+
+GLOBALS(
+  char *s_signal;
+  char *k_timeout;
+
+  int nextsig;
+  pid_t pid;
+  struct timeval ktv;
+  struct itimerval itv;
+)
+
+static void handler(int i)
+{
+  kill(TT.pid, TT.nextsig);
+  
+  if (TT.k_timeout) {
+    TT.k_timeout = 0;
+    TT.nextsig = SIGKILL;
+    signal(SIGALRM, handler);
+    TT.itv.it_value = TT.ktv;
+    setitimer(ITIMER_REAL, &TT.itv, (void *)&toybuf);
+  }
+}
+
+void timeout_main(void)
+{
+  // Parse early to get any errors out of the way.
+  TT.itv.it_value.tv_sec = xparsetime(*toys.optargs, 1000000, &TT.itv.it_value.tv_usec);
+
+  if (TT.k_timeout)
+    TT.ktv.tv_sec = xparsetime(TT.k_timeout, 1000000, &TT.ktv.tv_usec);
+  TT.nextsig = SIGTERM;
+  if (TT.s_signal && -1 == (TT.nextsig = sig_to_num(TT.s_signal)))
+    error_exit("bad -s: '%s'", TT.s_signal);
+
+  if (!(TT.pid = fork())) xexec_optargs(1);
+  else {
+    int status;
+
+    signal(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 (WIFSIGNALED(status)) toys.exitval = WTERMSIG(status);
+    else toys.exitval = 1;
+  }
+}
diff --git a/toys/posix/sleep.c b/toys/posix/sleep.c
index a83f1e56..c7b8bbf1 100644
--- a/toys/posix/sleep.c
+++ b/toys/posix/sleep.c
@@ -11,39 +11,26 @@ config SLEEP
   bool "sleep"
   default y
   help
-    usage: sleep SECONDS
+    usage: sleep LENGTH
+
+    Wait before exiting. An optional suffix can be "m" (minutes), "h" (hours),
+    "d" (days), or "s" (seconds, the default).
 
-    Wait before exiting.
 
 config SLEEP_FLOAT
   bool
   default y
   depends on SLEEP && TOYBOX_FLOAT
   help
-    The delay can be a decimal fraction. An optional suffix can be "m"
-    (minutes), "h" (hours), "d" (days), or "s" (seconds, the default).
+    Length can be a decimal fraction.
 */
 
 #include "toys.h"
 
 void sleep_main(void)
 {
+  struct timespec tv;
 
-  if (!CFG_TOYBOX_FLOAT) toys.exitval = sleep(atol(*toys.optargs));
-  else {
-    char *arg;
-    double d = strtod(*toys.optargs, &arg);
-    struct timespec tv;
-
-    // Parse suffix
-    if (*arg) {
-      int ismhd[]={1,60,3600,86400};
-      char *smhd = "smhd", *c = strchr(smhd, *arg);
-      if (!c) error_exit("Unknown suffix '%c'", *arg);
-      d *= ismhd[c-smhd];
-    }
-
-    tv.tv_nsec=1000000000*(d-(tv.tv_sec = (unsigned long)d));
-    toys.exitval = !!nanosleep(&tv, NULL);
-  }
+  tv.tv_sec = xparsetime(*toys.optargs, 1000000000, &tv.tv_nsec);
+  toys.exitval = !!nanosleep(&tv, NULL);
 }