miscutils/watchdog.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini watchdog implementation for busybox
   4  *
   5  * Copyright (C) 2003  Paul Mundt <lethal@linux-sh.org>
   6  * Copyright (C) 2006  Bernhard Fischer <busybox@busybox.net>
   7  *
   8  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
   9  */
  10
  11 #include "busybox.h"
  12 #include <stdio.h>
  13 #include <fcntl.h>
  14 #include <unistd.h>
  15 #include <stdlib.h>
  16 #include <signal.h>
  17
  18 #define OPT_FOREGROUND 0x01
  19 #define OPT_TIMER      0x02
  20
  21 /* Watchdog file descriptor */
  22 static int fd;
  23
  24 static void watchdog_shutdown(int ATTRIBUTE_UNUSED unused)
  25 {
  26         write(fd, "V", 1);      /* Magic, see watchdog-api.txt in kernel */
  27         close(fd);
  28         exit(0);
  29 }
  30
  31 int watchdog_main(int argc, char **argv)
  32 {
  33         unsigned long opts;
  34         unsigned long timer_duration = 30; /* Userspace timer duration, in seconds */
  35         char *t_arg;
  36
  37         opts = bb_getopt_ulflags(argc, argv, "Ft:", &t_arg);
  38
  39         if (opts & OPT_TIMER)
  40                 timer_duration = bb_xgetlarg(t_arg, 10, 0, INT_MAX);
  41
  42         /* We're only interested in the watchdog device .. */
  43         if (optind < argc - 1 || argc == 1)
  44                 bb_show_usage();
  45
  46 #ifdef BB_NOMMU
  47         if (!(opts & OPT_FOREGROUND))
  48                 vfork_daemon_rexec(0, 1, argc, argv, "-F");
  49 #else
  50         bb_xdaemon(0, 1);
  51 #endif
  52
  53         signal(SIGHUP, watchdog_shutdown);
  54         signal(SIGINT, watchdog_shutdown);
  55
  56         fd = bb_xopen(argv[argc - 1], O_WRONLY);
  57
  58         while (1) {
  59                 /*
  60                  * Make sure we clear the counter before sleeping, as the counter value
  61                  * is undefined at this point -- PFM
  62                  */
  63                 write(fd, "\0", 1);
  64                 sleep(timer_duration);
  65         }
  66
  67         watchdog_shutdown(0);
  68
  69         return EXIT_SUCCESS;
  70 }