Home | History | Annotate | Line # | Download | only in libpthread
      1 /* $NetBSD: t_sigmask.c,v 1.3 2013/10/19 17:45:01 christos Exp $ */
      2 
      3 /*
      4  * Copyright (c) 2008, 2010 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 #include <sys/cdefs.h>
     30 __COPYRIGHT("@(#) Copyright (c) 2008, 2010\
     31  The NetBSD Foundation, inc. All rights reserved.");
     32 __RCSID("$NetBSD: t_sigmask.c,v 1.3 2013/10/19 17:45:01 christos Exp $");
     33 
     34 /*
     35  * Regression test for pthread_sigmask when SA upcalls aren't started yet.
     36  *
     37  * Written by Christian Limpach <cl (at) NetBSD.org>, December 2003.
     38  * Public domain.
     39  */
     40 
     41 #include <errno.h>
     42 #include <pthread.h>
     43 #include <signal.h>
     44 #include <stdio.h>
     45 #include <unistd.h>
     46 
     47 #include <sys/resource.h>
     48 
     49 #include <atf-c.h>
     50 
     51 #include "h_common.h"
     52 
     53 static volatile sig_atomic_t flag;
     54 static volatile sig_atomic_t flag2;
     55 
     56 static volatile pthread_t thr_usr1;
     57 static volatile pthread_t thr_usr2;
     58 
     59 static sig_atomic_t count = 0;
     60 
     61 ATF_TC(upcalls_not_started);
     62 ATF_TC_HEAD(upcalls_not_started, tc)
     63 {
     64 	atf_tc_set_md_var(tc, "descr", "Checks pthread_sigmask when SA upcalls "
     65 	    "aren't started yet");
     66 }
     67 ATF_TC_BODY(upcalls_not_started, tc)
     68 {
     69 	sigset_t nset;
     70 	struct rlimit rlim;
     71 
     72 	rlim.rlim_cur = rlim.rlim_max = 0;
     73 	(void) setrlimit(RLIMIT_CORE, &rlim);
     74 
     75 	sigemptyset(&nset);
     76 	sigaddset(&nset, SIGFPE);
     77 	pthread_sigmask(SIG_BLOCK, &nset, NULL);
     78 
     79 	kill(getpid(), SIGFPE);
     80 }
     81 
     82 static void
     83 upcalls_not_started_handler1(int sig, siginfo_t *info, void *ctx)
     84 {
     85 
     86 	kill(getpid(), SIGUSR2);
     87 	/*
     88 	 * If the mask is properly set, SIGUSR2 will not be handled
     89 	 * until this handler returns.
     90 	 */
     91 	flag = 1;
     92 }
     93 
     94 static void
     95 upcalls_not_started_handler2(int sig, siginfo_t *info, void *ctx)
     96 {
     97 	if (flag == 1)
     98 		flag = 2;
     99 }
    100 
    101 ATF_TC(before_threads);
    102 ATF_TC_HEAD(before_threads, tc)
    103 {
    104 	atf_tc_set_md_var(tc, "descr", "Checks that signal masks are respected "
    105 	    "before threads are started");
    106 }
    107 ATF_TC_BODY(before_threads, tc)
    108 {
    109 	struct sigaction act;
    110 
    111 	act.sa_sigaction = upcalls_not_started_handler1;
    112 	sigemptyset(&act.sa_mask);
    113 	sigaddset(&act.sa_mask, SIGUSR2);
    114 	act.sa_flags = SA_SIGINFO;
    115 
    116 	ATF_REQUIRE_EQ(sigaction(SIGUSR1, &act, NULL), 0);
    117 
    118 	act.sa_sigaction = upcalls_not_started_handler2;
    119 	sigemptyset(&act.sa_mask);
    120 	act.sa_flags = SA_SIGINFO;
    121 	(void)sigaction(SIGUSR2, &act, NULL);
    122 
    123 	kill(getpid(), SIGUSR1);
    124 
    125 	ATF_REQUIRE_EQ(flag, 2);
    126 	printf("Success: Both handlers ran in order\n");
    127 }
    128 
    129 static void
    130 respected_while_running_handler1(int sig, siginfo_t *info, void *ctx)
    131 {
    132 
    133 	kill(getpid(), SIGUSR2);
    134 	/*
    135 	 * If the mask is properly set, SIGUSR2 will not be handled
    136 	 * by the current thread until this handler returns.
    137 	 */
    138 	flag = 1;
    139 	thr_usr1 = pthread_self();
    140 }
    141 
    142 static void
    143 respected_while_running_handler2(int sig, siginfo_t *info, void *ctx)
    144 {
    145 	if (flag == 1)
    146 		flag = 2;
    147 	flag2 = 1;
    148 	thr_usr2 = pthread_self();
    149 }
    150 
    151 static void *
    152 respected_while_running_threadroutine(void *arg)
    153 {
    154 
    155 	kill(getpid(), SIGUSR1);
    156 	sleep(1);
    157 
    158 	if (flag == 2)
    159 		printf("Success: Both handlers ran in order\n");
    160 	else if (flag == 1 && flag2 == 1 && thr_usr1 != thr_usr2)
    161 		printf("Success: Handlers were invoked by different threads\n");
    162 	else {
    163 		printf("Failure: flag=%d, flag2=%d, thr1=%p, thr2=%p\n",
    164 			(int)flag, (int)flag2, (void *)thr_usr1, (void *)thr_usr2);
    165 		atf_tc_fail("failure");
    166 	}
    167 
    168 	return NULL;
    169 }
    170 
    171 ATF_TC(respected_while_running);
    172 ATF_TC_HEAD(respected_while_running, tc)
    173 {
    174 	atf_tc_set_md_var(tc, "descr", "Checks that signal masks are respected "
    175 	    "while threads are running");
    176 }
    177 ATF_TC_BODY(respected_while_running, tc)
    178 {
    179 	struct sigaction act;
    180 	pthread_t thread;
    181 
    182 	act.sa_sigaction = respected_while_running_handler1;
    183 	sigemptyset(&act.sa_mask);
    184 	sigaddset(&act.sa_mask, SIGUSR2);
    185 	act.sa_flags = SA_SIGINFO;
    186 
    187 	ATF_REQUIRE_EQ(sigaction(SIGUSR1, &act, NULL), 0);
    188 
    189 	act.sa_sigaction = respected_while_running_handler2;
    190 	sigemptyset(&act.sa_mask);
    191 	act.sa_flags = SA_SIGINFO;
    192 	(void)sigaction(SIGUSR2, &act, NULL);
    193 
    194 	PTHREAD_REQUIRE(pthread_create(&thread, NULL,
    195 	    respected_while_running_threadroutine, NULL));
    196 	PTHREAD_REQUIRE(pthread_join(thread, NULL));
    197 }
    198 
    199 static void
    200 incorrect_mask_bug_handler(int sig)
    201 {
    202 	count++;
    203 }
    204 
    205 static void *
    206 incorrect_mask_bug_sleeper(void* arg)
    207 {
    208 	int i;
    209 	for (i = 0; i < 10; i++)
    210 		sleep(1);
    211 
    212 	atf_tc_fail("sleeper");
    213 }
    214 
    215 ATF_TC(incorrect_mask_bug);
    216 ATF_TC_HEAD(incorrect_mask_bug, tc)
    217 {
    218 	atf_tc_set_md_var(tc, "descr", "Checks for bug in libpthread where "
    219 	    "incorrect signal mask was used");
    220 }
    221 ATF_TC_BODY(incorrect_mask_bug, tc)
    222 {
    223 	pthread_t id;
    224 	struct sigaction act;
    225 
    226 	act.sa_sigaction = NULL;
    227 	sigemptyset(&act.sa_mask);
    228 	act.sa_flags = 0;
    229 	act.sa_handler = incorrect_mask_bug_handler;
    230 
    231 	ATF_REQUIRE_EQ_MSG(sigaction(SIGALRM, &act, NULL), 0, "%s",
    232 	    strerror(errno));
    233 
    234 	sigaddset(&act.sa_mask, SIGALRM);
    235 	PTHREAD_REQUIRE(pthread_sigmask(SIG_SETMASK, &act.sa_mask, NULL));
    236 
    237 	PTHREAD_REQUIRE(pthread_create(&id, NULL, incorrect_mask_bug_sleeper,
    238 	    NULL));
    239 	sleep(1);
    240 
    241 	sigemptyset(&act.sa_mask);
    242 	PTHREAD_REQUIRE(pthread_sigmask(SIG_SETMASK, &act.sa_mask, NULL));
    243 
    244 	for (;;) {
    245 		alarm(1);
    246 		if (select(1, NULL, NULL, NULL, NULL) == -1 && errno == EINTR)
    247 			if (count == 2)
    248 				return;
    249 	}
    250 }
    251 
    252 ATF_TP_ADD_TCS(tp)
    253 {
    254 
    255 	ATF_TP_ADD_TC(tp, upcalls_not_started);
    256 	ATF_TP_ADD_TC(tp, before_threads);
    257 	ATF_TP_ADD_TC(tp, respected_while_running);
    258 	ATF_TP_ADD_TC(tp, incorrect_mask_bug);
    259 
    260 	return atf_no_error();
    261 }
    262