Header cleanup on two more networking files (move libbb.h to the top and
[oweals/busybox.git] / 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 /* Watchdog file descriptor */
19 static int fd;
20
21 static void watchdog_shutdown(int ATTRIBUTE_UNUSED unused)
22 {
23         write(fd, "V", 1);      /* Magic */
24         close(fd);
25         exit(0);
26 }
27
28 int watchdog_main(int argc, char **argv)
29 {
30         unsigned long timer_duration = 30; /* Userspace timer duration, in seconds */
31         char *t_arg;
32
33         if (bb_getopt_ulflags(argc, argv, "t:", &t_arg))
34                 timer_duration = bb_xgetlarg(t_arg, 10, 0, INT_MAX);
35
36         /* We're only interested in the watchdog device .. */
37         if (optind < argc - 1 || argc == 1)
38                 bb_show_usage();
39
40         bb_xdaemon(0, 1);
41
42         signal(SIGHUP, watchdog_shutdown);
43         signal(SIGINT, watchdog_shutdown);
44
45         fd = bb_xopen(argv[argc - 1], O_WRONLY);
46
47         while (1) {
48                 /*
49                  * Make sure we clear the counter before sleeping, as the counter value
50                  * is undefined at this point -- PFM
51                  */
52                 write(fd, "\0", 1);
53                 sleep(timer_duration);
54         }
55
56         watchdog_shutdown(0);
57
58         return EXIT_SUCCESS;
59 }