1/*	$NetBSD: t_swwdog.c,v 1.7 2017/01/13 21:30:39 christos Exp $	*/
2
3/*
4 * Copyright (c) 2010 Antti Kantee.  All Rights Reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28#include <sys/types.h>
29#include <sys/wait.h>
30#include <sys/wdog.h>
31
32#include <assert.h>
33#include <atf-c.h>
34#include <err.h>
35#include <errno.h>
36#include <fcntl.h>
37#include <stdio.h>
38#include <stdlib.h>
39#include <string.h>
40#include <unistd.h>
41#include <signal.h>
42
43#include <rump/rump.h>
44#include <rump/rump_syscalls.h>
45
46#include "h_macros.h"
47
48static volatile sig_atomic_t tcount;
49
50static void
51sigcount(int sig)
52{
53
54	assert(sig == SIGUSR1);
55	tcount++;
56}
57
58/*
59 * Since we are testing for swwdog's ability to reboot/panic, we need
60 * to fork and monitor the exit status from the parent and report
61 * something sensible back to atf.
62 */
63static int
64testbody(int max)
65{
66	char wname[WDOG_NAMESIZE];
67	struct wdog_conf wc;
68	struct wdog_mode wm;
69	pid_t p1, p2;
70	int status;
71	int fd;
72
73	signal(SIGUSR1, sigcount);
74
75	switch ((p1 = fork())) {
76	case 0:
77		break;
78	case -1:
79		atf_tc_fail_errno("fork");
80		break;
81	default:
82		p2 = wait(&status);
83		ATF_REQUIRE_EQ(p1, p2);
84		ATF_REQUIRE_EQ(tcount, max);
85		return status;
86	}
87
88	rump_init();
89
90	fd = rump_sys_open("/dev/watchdog", O_RDWR);
91	if (fd == -1)
92		err(1, "open watchdog");
93
94	wc.wc_count = 1;
95	wc.wc_names = wname;
96
97	if (rump_sys_ioctl(fd, WDOGIOC_GWDOGS, &wc) == -1)
98		err(1, "can't fetch watchdog names");
99
100	if (wc.wc_count) {
101		assert(wc.wc_count == 1);
102
103		strlcpy(wm.wm_name, wc.wc_names, sizeof(wm.wm_name));
104		wm.wm_mode = WDOG_MODE_ETICKLE;
105		wm.wm_period = 1;
106		if (rump_sys_ioctl(fd, WDOGIOC_SMODE, &wm) == -1)
107			atf_tc_fail_errno("failed to set tickle");
108
109		usleep(400000);
110		if (max == 1)
111			rump_sys_ioctl(fd, WDOGIOC_TICKLE);
112		else {
113			wm.wm_mode = WDOG_MODE_DISARMED;
114			rump_sys_ioctl(fd, WDOGIOC_SMODE, &wm);
115		}
116		kill(getppid(), SIGUSR1);
117
118		sleep(2);
119		printf("staying alive\n");
120		kill(getppid(), SIGUSR1);
121		_exit(2);
122	}
123	/* fail */
124	printf("no watchdog registered!\n");
125	_exit(1);
126}
127
128ATF_TC(reboot);
129ATF_TC_HEAD(reboot, tc)
130{
131
132	atf_tc_set_md_var(tc, "descr", "check swwdog reboot capability");
133}
134
135ATF_TC_BODY(reboot, tc)
136{
137	extern bool rumpns_swwdog_reboot;
138	int status;
139
140	/* XXX: should use sysctl */
141	rumpns_swwdog_reboot = true;
142	status = testbody(1);
143
144	ATF_REQUIRE(WIFEXITED(status));
145	ATF_REQUIRE_EQ(WEXITSTATUS(status), 0);
146}
147
148ATF_TC(panic);
149ATF_TC_HEAD(panic, tc)
150{
151
152	atf_tc_set_md_var(tc, "descr", "check swwdog panic capability");
153}
154
155ATF_TC_BODY(panic, tc)
156{
157	extern bool rumpns_swwdog_reboot;
158	int status;
159
160	/* XXX: should use sysctl */
161	rumpns_swwdog_reboot = false;
162	status = testbody(1);
163
164	ATF_REQUIRE(WIFSIGNALED(status));
165	ATF_REQUIRE_EQ(WTERMSIG(status), SIGABRT);
166}
167
168ATF_TC(disarm);
169ATF_TC_HEAD(disarm, tc)
170{
171
172	atf_tc_set_md_var(tc, "descr", "check swwdog disarm capability");
173}
174
175ATF_TC_BODY(disarm, tc)
176{
177	int status;
178
179	status = testbody(2);
180
181	ATF_REQUIRE(WIFEXITED(status));
182	ATF_REQUIRE_EQ(WEXITSTATUS(status), 2);
183}
184
185ATF_TP_ADD_TCS(tp)
186{
187
188	ATF_TP_ADD_TC(tp, panic);
189	ATF_TP_ADD_TC(tp, reboot);
190	ATF_TP_ADD_TC(tp, disarm);
191
192	return atf_no_error();
193}
194