1 /* $NetBSD: t_threadjmp.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_threadjmp.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 <stdio.h>
74 #include <stdlib.h>
75 #include <string.h>
76 #include <unistd.h>
77 
78 #include <pthread.h>
79 
80 #include <atf-c.h>
81 
82 #define REQUIRE_ERRNO(x) ATF_REQUIRE_MSG(x, "%s", strerror(errno))
83 
84 #define TEST_SETJMP 0
85 #define TEST_U_SETJMP 1
86 #define TEST_SIGSETJMP_SAVE 2
87 #define TEST_SIGSETJMP_NOSAVE 3
88 
89 static pthread_t myself = NULL;
90 
91 static int expectsignal;
92 
93 static void
94 aborthandler(int signo __unused)
95 {
96 	ATF_REQUIRE(myself == pthread_self());
97 	ATF_REQUIRE_MSG(expectsignal, "kill(SIGABRT) succeeded");
98 	atf_tc_pass();
99 }
100 
101 static void
102 h_check(int test)
103 {
104 	struct sigaction sa;
105 	jmp_buf jb;
106 	sigjmp_buf sjb;
107 	sigset_t ss;
108 	int i, x;
109 
110 	myself = pthread_self();
111 	i = getpid();
112 
113 	if (test == TEST_SETJMP || test == TEST_SIGSETJMP_SAVE)
114 		expectsignal = 0;
115 	else if (test == TEST_U_SETJMP || test == TEST_SIGSETJMP_NOSAVE)
116 		expectsignal = 1;
117 	else
118 		atf_tc_fail("unknown test");
119 
120 	sa.sa_handler = aborthandler;
121 	sigemptyset(&sa.sa_mask);
122 	sa.sa_flags = 0;
123 	REQUIRE_ERRNO(sigaction(SIGABRT, &sa, NULL) != -1);
124 	REQUIRE_ERRNO(sigemptyset(&ss) != -1);
125 	REQUIRE_ERRNO(sigaddset(&ss, SIGABRT) != -1);
126 	REQUIRE_ERRNO(sigprocmask(SIG_BLOCK, &ss, NULL) != -1);
127 	ATF_REQUIRE(myself == pthread_self());
128 
129 	if (test == TEST_SETJMP)
130 		x = setjmp(jb);
131 	else if (test == TEST_U_SETJMP)
132 		x = _setjmp(jb);
133 	else
134 		x = sigsetjmp(sjb, !expectsignal);
135 
136 	if (x != 0) {
137 		ATF_REQUIRE(myself == pthread_self());
138 		ATF_REQUIRE_MSG(x == i, "setjmp returned wrong value");
139 		kill(i, SIGABRT);
140 		ATF_REQUIRE_MSG(!expectsignal, "kill(SIGABRT) failed");
141 		ATF_REQUIRE(myself == pthread_self());
142 		atf_tc_pass();
143 	}
144 
145 	ATF_REQUIRE(myself == pthread_self());
146 	REQUIRE_ERRNO(sigprocmask(SIG_UNBLOCK, &ss, NULL) != -1);
147 
148 	if (test == TEST_SETJMP)
149 		longjmp(jb, i);
150 	else if (test == TEST_U_SETJMP)
151 		_longjmp(jb, i);
152 	else
153 		siglongjmp(sjb, i);
154 
155 	atf_tc_fail("jmp failed");
156 }
157 
158 ATF_TC(setjmp);
159 ATF_TC_HEAD(setjmp, tc)
160 {
161 	atf_tc_set_md_var(tc, "descr",
162 	    "Checks pthread_self() and setjmp(3)");
163 }
164 ATF_TC_BODY(setjmp, tc)
165 {
166 	h_check(TEST_SETJMP);
167 }
168 
169 ATF_TC(_setjmp);
170 ATF_TC_HEAD(_setjmp, tc)
171 {
172 	atf_tc_set_md_var(tc, "descr",
173 	    "Checks pthread_self() and _setjmp(3)");
174 }
175 ATF_TC_BODY(_setjmp, tc)
176 {
177 	h_check(TEST_U_SETJMP);
178 }
179 
180 ATF_TC(sigsetjmp_save);
181 ATF_TC_HEAD(sigsetjmp_save, tc)
182 {
183 	atf_tc_set_md_var(tc, "descr",
184 	    "Checks pthread_self() and sigsetjmp(3) with savemask enabled");
185 }
186 ATF_TC_BODY(sigsetjmp_save, tc)
187 {
188 	h_check(TEST_SIGSETJMP_SAVE);
189 }
190 
191 ATF_TC(sigsetjmp_nosave);
192 ATF_TC_HEAD(sigsetjmp_nosave, tc)
193 {
194 	atf_tc_set_md_var(tc, "descr",
195 	    "Checks pthread_self() and sigsetjmp(3) with savemask disabled");
196 }
197 ATF_TC_BODY(sigsetjmp_nosave, tc)
198 {
199 	h_check(TEST_SIGSETJMP_NOSAVE);
200 }
201 
202 ATF_TP_ADD_TCS(tp)
203 {
204 
205 	/*
206 	 * These test cases try to verify setjmp and friends in a program
207 	 * linked with pthreads, and verify that pthread_self() stays
208 	 * consistent throughout the program. A setcontext() call invoked
209 	 * by *setjmp() might clobber the TLS special register used to
210 	 * implement pthread_self().
211 	 */
212 	ATF_TP_ADD_TC(tp, setjmp);
213 	ATF_TP_ADD_TC(tp, _setjmp);
214 	ATF_TP_ADD_TC(tp, sigsetjmp_save);
215 	ATF_TP_ADD_TC(tp, sigsetjmp_nosave);
216 
217 	return atf_no_error();
218 }
219