GNU Linux-libre 4.14.266-gnu1
[releases.git] / tools / testing / selftests / watchdog / watchdog-test.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Watchdog Driver Test Program
4  */
5
6 #include <errno.h>
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <string.h>
10 #include <unistd.h>
11 #include <fcntl.h>
12 #include <signal.h>
13 #include <getopt.h>
14 #include <sys/ioctl.h>
15 #include <linux/types.h>
16 #include <linux/watchdog.h>
17
18 #define DEFAULT_PING_RATE       1
19
20 int fd;
21 const char v = 'V';
22 static const char sopts[] = "bdehp:t:";
23 static const struct option lopts[] = {
24         {"bootstatus",          no_argument, NULL, 'b'},
25         {"disable",             no_argument, NULL, 'd'},
26         {"enable",              no_argument, NULL, 'e'},
27         {"help",                no_argument, NULL, 'h'},
28         {"pingrate",      required_argument, NULL, 'p'},
29         {"timeout",       required_argument, NULL, 't'},
30         {NULL,                  no_argument, NULL, 0x0}
31 };
32
33 /*
34  * This function simply sends an IOCTL to the driver, which in turn ticks
35  * the PC Watchdog card to reset its internal timer so it doesn't trigger
36  * a computer reset.
37  */
38 static void keep_alive(void)
39 {
40         int dummy;
41         int ret;
42
43         ret = ioctl(fd, WDIOC_KEEPALIVE, &dummy);
44         if (!ret)
45                 printf(".");
46 }
47
48 /*
49  * The main program.  Run the program with "-d" to disable the card,
50  * or "-e" to enable the card.
51  */
52
53 static void term(int sig)
54 {
55         int ret = write(fd, &v, 1);
56
57         close(fd);
58         if (ret < 0)
59                 printf("\nStopping watchdog ticks failed (%d)...\n", errno);
60         else
61                 printf("\nStopping watchdog ticks...\n");
62         exit(0);
63 }
64
65 static void usage(char *progname)
66 {
67         printf("Usage: %s [options]\n", progname);
68         printf(" -b, --bootstatus    Get last boot status (Watchdog/POR)\n");
69         printf(" -d, --disable       Turn off the watchdog timer\n");
70         printf(" -e, --enable        Turn on the watchdog timer\n");
71         printf(" -h, --help          Print the help message\n");
72         printf(" -p, --pingrate=P    Set ping rate to P seconds (default %d)\n", DEFAULT_PING_RATE);
73         printf(" -t, --timeout=T     Set timeout to T seconds\n");
74         printf("\n");
75         printf("Parameters are parsed left-to-right in real-time.\n");
76         printf("Example: %s -d -t 10 -p 5 -e\n", progname);
77 }
78
79 int main(int argc, char *argv[])
80 {
81         int flags;
82         unsigned int ping_rate = DEFAULT_PING_RATE;
83         int ret;
84         int c;
85         int oneshot = 0;
86
87         setbuf(stdout, NULL);
88
89         fd = open("/dev/watchdog", O_WRONLY);
90
91         if (fd == -1) {
92                 if (errno == ENOENT)
93                         printf("Watchdog device not enabled.\n");
94                 else if (errno == EACCES)
95                         printf("Run watchdog as root.\n");
96                 else
97                         printf("Watchdog device open failed %s\n",
98                                 strerror(errno));
99                 exit(-1);
100         }
101
102         while ((c = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
103                 switch (c) {
104                 case 'b':
105                         flags = 0;
106                         oneshot = 1;
107                         ret = ioctl(fd, WDIOC_GETBOOTSTATUS, &flags);
108                         if (!ret)
109                                 printf("Last boot is caused by: %s.\n", (flags != 0) ?
110                                         "Watchdog" : "Power-On-Reset");
111                         else
112                                 printf("WDIOC_GETBOOTSTATUS error '%s'\n", strerror(errno));
113                         break;
114                 case 'd':
115                         flags = WDIOS_DISABLECARD;
116                         ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
117                         if (!ret)
118                                 printf("Watchdog card disabled.\n");
119                         else
120                                 printf("WDIOS_DISABLECARD error '%s'\n", strerror(errno));
121                         break;
122                 case 'e':
123                         flags = WDIOS_ENABLECARD;
124                         ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
125                         if (!ret)
126                                 printf("Watchdog card enabled.\n");
127                         else
128                                 printf("WDIOS_ENABLECARD error '%s'\n", strerror(errno));
129                         break;
130                 case 'p':
131                         ping_rate = strtoul(optarg, NULL, 0);
132                         if (!ping_rate)
133                                 ping_rate = DEFAULT_PING_RATE;
134                         printf("Watchdog ping rate set to %u seconds.\n", ping_rate);
135                         break;
136                 case 't':
137                         flags = strtoul(optarg, NULL, 0);
138                         ret = ioctl(fd, WDIOC_SETTIMEOUT, &flags);
139                         if (!ret)
140                                 printf("Watchdog timeout set to %u seconds.\n", flags);
141                         else
142                                 printf("WDIOC_SETTIMEOUT error '%s'\n", strerror(errno));
143                         break;
144                 default:
145                         usage(argv[0]);
146                         goto end;
147                 }
148         }
149
150         if (oneshot)
151                 goto end;
152
153         printf("Watchdog Ticking Away!\n");
154
155         signal(SIGINT, term);
156
157         while (1) {
158                 keep_alive();
159                 sleep(ping_rate);
160         }
161 end:
162         ret = write(fd, &v, 1);
163         if (ret < 0)
164                 printf("Stopping watchdog ticks failed (%d)...\n", errno);
165         close(fd);
166         return 0;
167 }