1/* $NetBSD: t_setjmp.c,v 1.2 2017/01/14 21:08:17 christos Exp $ */
2
3/*-
4 * Copyright (c) 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/*
30 * Copyright (c) 1994 Christopher G. Demetriou
31 * All rights reserved.
32 *
33 * Redistribution and use in source and binary forms, with or without
34 * modification, are permitted provided that the following conditions
35 * are met:
36 * 1. Redistributions of source code must retain the above copyright
37 *    notice, this list of conditions and the following disclaimer.
38 * 2. Redistributions in binary form must reproduce the above copyright
39 *    notice, this list of conditions and the following disclaimer in the
40 *    documentation and/or other materials provided with the distribution.
41 * 3. All advertising materials mentioning features or use of this software
42 *    must display the following acknowledgement:
43 *          This product includes software developed for the
44 *          NetBSD Project.  See http://www.NetBSD.org/ for
45 *          information about NetBSD.
46 * 4. The name of the author may not be used to endorse or promote products
47 *    derived from this software without specific prior written permission.
48 *
49 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
50 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
51 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
52 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
53 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
54 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
55 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
56 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
57 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
58 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
59 *
60 * <<Id: LICENSE,v 1.2 2000/06/14 15:57:33 cgd Exp>>
61 */
62
63#include <sys/cdefs.h>
64__COPYRIGHT("@(#) Copyright (c) 2008\
65 The NetBSD Foundation, inc. All rights reserved.");
66__RCSID("$NetBSD: t_setjmp.c,v 1.2 2017/01/14 21:08:17 christos Exp $");
67
68#include <sys/types.h>
69
70#include <errno.h>
71#include <setjmp.h>
72#include <signal.h>
73#include <stdbool.h>
74#include <stdio.h>
75#include <stdlib.h>
76#include <string.h>
77#include <unistd.h>
78
79#include <atf-c.h>
80
81#define REQUIRE_ERRNO(x) ATF_REQUIRE_MSG(x, "%s", strerror(errno))
82
83#define TEST_SETJMP 0
84#define TEST_U_SETJMP 1
85#define TEST_SIGSETJMP_SAVE 2
86#define TEST_SIGSETJMP_NOSAVE 3
87#define TEST_LONGJMP_ZERO 4
88#define TEST_U_LONGJMP_ZERO 5
89
90static int expectsignal;
91
92static void
93aborthandler(int signo __unused)
94{
95	ATF_REQUIRE_MSG(expectsignal, "kill(SIGABRT) succeeded");
96	atf_tc_pass();
97}
98
99static void
100h_check(int test)
101{
102	struct sigaction sa;
103	jmp_buf jb;
104	sigjmp_buf sjb;
105	sigset_t ss;
106	int i, x;
107	volatile bool did_longjmp;
108
109	i = getpid();
110	did_longjmp = false;
111
112	if (test == TEST_SETJMP || test == TEST_SIGSETJMP_SAVE ||
113	    test == TEST_LONGJMP_ZERO)
114		expectsignal = 0;
115	else if (test == TEST_U_SETJMP || test == TEST_SIGSETJMP_NOSAVE ||
116	    test == TEST_U_LONGJMP_ZERO)
117		expectsignal = 1;
118	else
119		atf_tc_fail("unknown test");
120
121	sa.sa_handler = aborthandler;
122	sigemptyset(&sa.sa_mask);
123	sa.sa_flags = 0;
124	REQUIRE_ERRNO(sigaction(SIGABRT, &sa, NULL) != -1);
125	REQUIRE_ERRNO(sigemptyset(&ss) != -1);
126	REQUIRE_ERRNO(sigaddset(&ss, SIGABRT) != -1);
127	REQUIRE_ERRNO(sigprocmask(SIG_BLOCK, &ss, NULL) != -1);
128
129	if (test == TEST_SETJMP || test == TEST_LONGJMP_ZERO)
130		x = setjmp(jb);
131	else if (test == TEST_U_SETJMP || test == TEST_U_LONGJMP_ZERO)
132		x = _setjmp(jb);
133	else
134		x = sigsetjmp(sjb, !expectsignal);
135
136	if (x != 0) {
137		if (test == TEST_LONGJMP_ZERO || test == TEST_U_LONGJMP_ZERO)
138			ATF_REQUIRE_MSG(x == 1, "setjmp returned wrong value");
139		else
140			ATF_REQUIRE_MSG(x == i, "setjmp returned wrong value");
141
142		kill(i, SIGABRT);
143		ATF_REQUIRE_MSG(!expectsignal, "kill(SIGABRT) failed");
144		atf_tc_pass();
145	} else if (did_longjmp) {
146		atf_tc_fail("setjmp returned zero after longjmp");
147	}
148
149	REQUIRE_ERRNO(sigprocmask(SIG_UNBLOCK, &ss, NULL) != -1);
150
151	did_longjmp = true;
152	if (test == TEST_SETJMP)
153		longjmp(jb, i);
154	else if (test == TEST_LONGJMP_ZERO)
155		longjmp(jb, 0);
156	else if (test == TEST_U_SETJMP)
157		_longjmp(jb, i);
158	else if (test == TEST_U_LONGJMP_ZERO)
159		_longjmp(jb, 0);
160	else
161		siglongjmp(sjb, i);
162
163	atf_tc_fail("jmp failed");
164}
165
166ATF_TC(setjmp);
167ATF_TC_HEAD(setjmp, tc)
168{
169	atf_tc_set_md_var(tc, "descr", "Checks setjmp(3)");
170}
171ATF_TC_BODY(setjmp, tc)
172{
173	h_check(TEST_SETJMP);
174}
175
176ATF_TC(_setjmp);
177ATF_TC_HEAD(_setjmp, tc)
178{
179	atf_tc_set_md_var(tc, "descr", "Checks _setjmp(3)");
180}
181ATF_TC_BODY(_setjmp, tc)
182{
183	h_check(TEST_U_SETJMP);
184}
185
186ATF_TC(sigsetjmp_save);
187ATF_TC_HEAD(sigsetjmp_save, tc)
188{
189	atf_tc_set_md_var(tc, "descr", "Checks sigsetjmp(3) with savemask enabled");
190}
191ATF_TC_BODY(sigsetjmp_save, tc)
192{
193	h_check(TEST_SIGSETJMP_SAVE);
194}
195
196ATF_TC(sigsetjmp_nosave);
197ATF_TC_HEAD(sigsetjmp_nosave, tc)
198{
199	atf_tc_set_md_var(tc, "descr", "Checks sigsetjmp(3) with savemask disabled");
200}
201ATF_TC_BODY(sigsetjmp_nosave, tc)
202{
203	h_check(TEST_SIGSETJMP_NOSAVE);
204}
205
206ATF_TC(longjmp_zero);
207ATF_TC_HEAD(longjmp_zero, tc)
208{
209	atf_tc_set_md_var(tc, "descr", "Checks longjmp(3) with a zero value");
210}
211ATF_TC_BODY(longjmp_zero, tc)
212{
213	h_check(TEST_LONGJMP_ZERO);
214}
215
216ATF_TC(_longjmp_zero);
217ATF_TC_HEAD(_longjmp_zero, tc)
218{
219	atf_tc_set_md_var(tc, "descr", "Checks _longjmp(3) with a zero value");
220}
221ATF_TC_BODY(_longjmp_zero, tc)
222{
223	h_check(TEST_U_LONGJMP_ZERO);
224}
225
226ATF_TP_ADD_TCS(tp)
227{
228	ATF_TP_ADD_TC(tp, setjmp);
229	ATF_TP_ADD_TC(tp, _setjmp);
230	ATF_TP_ADD_TC(tp, sigsetjmp_save);
231	ATF_TP_ADD_TC(tp, sigsetjmp_nosave);
232	ATF_TP_ADD_TC(tp, longjmp_zero);
233	ATF_TP_ADD_TC(tp, _longjmp_zero);
234
235	return atf_no_error();
236}
237