Home | History | Annotate | Line # | Download | only in setjmp
      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