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:Tn:NLf:i";
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 	{"gettimeout",          no_argument, NULL, 'T'},
31 	{"pretimeout",    required_argument, NULL, 'n'},
32 	{"getpretimeout",       no_argument, NULL, 'N'},
33 	{"gettimeleft",		no_argument, NULL, 'L'},
34 	{"file",          required_argument, NULL, 'f'},
35 	{"info",		no_argument, NULL, 'i'},
36 	{NULL,                  no_argument, NULL, 0x0}
37 };
38 
39 /*
40  * This function simply sends an IOCTL to the driver, which in turn ticks
41  * the PC Watchdog card to reset its internal timer so it doesn't trigger
42  * a computer reset.
43  */
44 static void keep_alive(void)
45 {
46 	int dummy;
47 	int ret;
48 
49 	ret = ioctl(fd, WDIOC_KEEPALIVE, &dummy);
50 	if (!ret)
51 		printf(".");
52 }
53 
54 /*
55  * The main program.  Run the program with "-d" to disable the card,
56  * or "-e" to enable the card.
57  */
58 
59 static void term(int sig)
60 {
61 	int ret = write(fd, &v, 1);
62 
63 	close(fd);
64 	if (ret < 0)
65 		printf("\nStopping watchdog ticks failed (%d)...\n", errno);
66 	else
67 		printf("\nStopping watchdog ticks...\n");
68 	exit(0);
69 }
70 
71 static void usage(char *progname)
72 {
73 	printf("Usage: %s [options]\n", progname);
74 	printf(" -f, --file\t\tOpen watchdog device file\n");
75 	printf("\t\t\tDefault is /dev/watchdog\n");
76 	printf(" -i, --info\t\tShow watchdog_info\n");
77 	printf(" -b, --bootstatus\tGet last boot status (Watchdog/POR)\n");
78 	printf(" -d, --disable\t\tTurn off the watchdog timer\n");
79 	printf(" -e, --enable\t\tTurn on the watchdog timer\n");
80 	printf(" -h, --help\t\tPrint the help message\n");
81 	printf(" -p, --pingrate=P\tSet ping rate to P seconds (default %d)\n",
82 	       DEFAULT_PING_RATE);
83 	printf(" -t, --timeout=T\tSet timeout to T seconds\n");
84 	printf(" -T, --gettimeout\tGet the timeout\n");
85 	printf(" -n, --pretimeout=T\tSet the pretimeout to T seconds\n");
86 	printf(" -N, --getpretimeout\tGet the pretimeout\n");
87 	printf(" -L, --gettimeleft\tGet the time left until timer expires\n");
88 	printf("\n");
89 	printf("Parameters are parsed left-to-right in real-time.\n");
90 	printf("Example: %s -d -t 10 -p 5 -e\n", progname);
91 	printf("Example: %s -t 12 -T -n 7 -N\n", progname);
92 }
93 
94 int main(int argc, char *argv[])
95 {
96 	int flags;
97 	unsigned int ping_rate = DEFAULT_PING_RATE;
98 	int ret;
99 	int c;
100 	int oneshot = 0;
101 	char *file = "/dev/watchdog";
102 	struct watchdog_info info;
103 
104 	setbuf(stdout, NULL);
105 
106 	while ((c = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
107 		if (c == 'f')
108 			file = optarg;
109 	}
110 
111 	fd = open(file, O_WRONLY);
112 
113 	if (fd == -1) {
114 		if (errno == ENOENT)
115 			printf("Watchdog device (%s) not found.\n", file);
116 		else if (errno == EACCES)
117 			printf("Run watchdog as root.\n");
118 		else
119 			printf("Watchdog device open failed %s\n",
120 				strerror(errno));
121 		exit(-1);
122 	}
123 
124 	/*
125 	 * Validate that `file` is a watchdog device
126 	 */
127 	ret = ioctl(fd, WDIOC_GETSUPPORT, &info);
128 	if (ret) {
129 		printf("WDIOC_GETSUPPORT error '%s'\n", strerror(errno));
130 		close(fd);
131 		exit(ret);
132 	}
133 
134 	optind = 0;
135 
136 	while ((c = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
137 		switch (c) {
138 		case 'b':
139 			flags = 0;
140 			oneshot = 1;
141 			ret = ioctl(fd, WDIOC_GETBOOTSTATUS, &flags);
142 			if (!ret)
143 				printf("Last boot is caused by: %s.\n", (flags != 0) ?
144 					"Watchdog" : "Power-On-Reset");
145 			else
146 				printf("WDIOC_GETBOOTSTATUS error '%s'\n", strerror(errno));
147 			break;
148 		case 'd':
149 			flags = WDIOS_DISABLECARD;
150 			ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
151 			if (!ret)
152 				printf("Watchdog card disabled.\n");
153 			else {
154 				printf("WDIOS_DISABLECARD error '%s'\n", strerror(errno));
155 				oneshot = 1;
156 			}
157 			break;
158 		case 'e':
159 			flags = WDIOS_ENABLECARD;
160 			ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
161 			if (!ret)
162 				printf("Watchdog card enabled.\n");
163 			else {
164 				printf("WDIOS_ENABLECARD error '%s'\n", strerror(errno));
165 				oneshot = 1;
166 			}
167 			break;
168 		case 'p':
169 			ping_rate = strtoul(optarg, NULL, 0);
170 			if (!ping_rate)
171 				ping_rate = DEFAULT_PING_RATE;
172 			printf("Watchdog ping rate set to %u seconds.\n", ping_rate);
173 			break;
174 		case 't':
175 			flags = strtoul(optarg, NULL, 0);
176 			ret = ioctl(fd, WDIOC_SETTIMEOUT, &flags);
177 			if (!ret)
178 				printf("Watchdog timeout set to %u seconds.\n", flags);
179 			else {
180 				printf("WDIOC_SETTIMEOUT error '%s'\n", strerror(errno));
181 				oneshot = 1;
182 			}
183 			break;
184 		case 'T':
185 			oneshot = 1;
186 			ret = ioctl(fd, WDIOC_GETTIMEOUT, &flags);
187 			if (!ret)
188 				printf("WDIOC_GETTIMEOUT returns %u seconds.\n", flags);
189 			else
190 				printf("WDIOC_GETTIMEOUT error '%s'\n", strerror(errno));
191 			break;
192 		case 'n':
193 			flags = strtoul(optarg, NULL, 0);
194 			ret = ioctl(fd, WDIOC_SETPRETIMEOUT, &flags);
195 			if (!ret)
196 				printf("Watchdog pretimeout set to %u seconds.\n", flags);
197 			else {
198 				printf("WDIOC_SETPRETIMEOUT error '%s'\n", strerror(errno));
199 				oneshot = 1;
200 			}
201 			break;
202 		case 'N':
203 			oneshot = 1;
204 			ret = ioctl(fd, WDIOC_GETPRETIMEOUT, &flags);
205 			if (!ret)
206 				printf("WDIOC_GETPRETIMEOUT returns %u seconds.\n", flags);
207 			else
208 				printf("WDIOC_GETPRETIMEOUT error '%s'\n", strerror(errno));
209 			break;
210 		case 'L':
211 			oneshot = 1;
212 			ret = ioctl(fd, WDIOC_GETTIMELEFT, &flags);
213 			if (!ret)
214 				printf("WDIOC_GETTIMELEFT returns %u seconds.\n", flags);
215 			else
216 				printf("WDIOC_GETTIMELEFT error '%s'\n", strerror(errno));
217 			break;
218 		case 'f':
219 			/* Handled above */
220 			break;
221 		case 'i':
222 			/*
223 			 * watchdog_info was obtained as part of file open
224 			 * validation. So we just show it here.
225 			 */
226 			oneshot = 1;
227 			printf("watchdog_info:\n");
228 			printf(" identity:\t\t%s\n", info.identity);
229 			printf(" firmware_version:\t%u\n",
230 			       info.firmware_version);
231 			printf(" options:\t\t%08x\n", info.options);
232 			break;
233 
234 		default:
235 			usage(argv[0]);
236 			goto end;
237 		}
238 	}
239 
240 	if (oneshot)
241 		goto end;
242 
243 	printf("Watchdog Ticking Away!\n");
244 
245 	signal(SIGINT, term);
246 
247 	while (1) {
248 		keep_alive();
249 		sleep(ping_rate);
250 	}
251 end:
252 	ret = write(fd, &v, 1);
253 	if (ret < 0)
254 		printf("Stopping watchdog ticks failed (%d)...\n", errno);
255 	close(fd);
256 	return 0;
257 }
258