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 }