1 /* $NetBSD: t_once.c,v 1.1 2010/07/16 15:42:53 jmmv 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 #include <sys/cdefs.h> 30 __COPYRIGHT("@(#) Copyright (c) 2008\ 31 The NetBSD Foundation, inc. All rights reserved."); 32 __RCSID("$NetBSD: t_once.c,v 1.1 2010/07/16 15:42:53 jmmv Exp $"); 33 34 #include <pthread.h> 35 #include <signal.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 39 #include <atf-c.h> 40 41 #include "h_common.h" 42 43 static pthread_once_t once = PTHREAD_ONCE_INIT; 44 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; 45 static int x; 46 47 #define NTHREADS 25 48 49 static void 50 ofunc(void) 51 { 52 printf("Variable x has value %d\n", x); 53 x++; 54 } 55 56 ATF_TC(once1); 57 ATF_TC_HEAD(once1, tc) 58 { 59 atf_tc_set_md_var(tc, "descr", "Checks pthread_once()"); 60 } 61 ATF_TC_BODY(once1, tc) 62 { 63 64 printf("1: Test 1 of pthread_once()\n"); 65 66 PTHREAD_REQUIRE(pthread_once(&once, ofunc)); 67 PTHREAD_REQUIRE(pthread_once(&once, ofunc)); 68 69 printf("1: X has value %d\n",x ); 70 ATF_REQUIRE_EQ(x, 1); 71 } 72 73 static void 74 once2_ofunc(void) 75 { 76 x++; 77 printf("ofunc: Variable x has value %d\n", x); 78 x++; 79 } 80 81 static void * 82 once2_threadfunc(void *arg) 83 { 84 int num; 85 86 PTHREAD_REQUIRE(pthread_once(&once, once2_ofunc)); 87 88 num = *(int *)arg; 89 printf("Thread %d sees x with value %d\n", num, x); 90 ATF_REQUIRE_EQ(x, 2); 91 92 return NULL; 93 } 94 95 ATF_TC(once2); 96 ATF_TC_HEAD(once2, tc) 97 { 98 atf_tc_set_md_var(tc, "descr", "Checks pthread_once()"); 99 } 100 ATF_TC_BODY(once2, tc) 101 { 102 pthread_t threads[NTHREADS]; 103 int id[NTHREADS]; 104 int i; 105 106 printf("1: Test 2 of pthread_once()\n"); 107 108 for (i=0; i < NTHREADS; i++) { 109 id[i] = i; 110 PTHREAD_REQUIRE(pthread_create(&threads[i], NULL, once2_threadfunc, &id[i])); 111 } 112 113 for (i=0; i < NTHREADS; i++) 114 PTHREAD_REQUIRE(pthread_join(threads[i], NULL)); 115 116 printf("1: X has value %d\n",x ); 117 ATF_REQUIRE_EQ(x, 2); 118 } 119 120 static void 121 once3_cleanup(void *m) 122 { 123 pthread_mutex_t *mu = m; 124 125 PTHREAD_REQUIRE(pthread_mutex_unlock(mu)); 126 } 127 128 static void 129 once3_ofunc(void) 130 { 131 pthread_testcancel(); 132 } 133 134 static void * 135 once3_threadfunc(void *arg) 136 { 137 PTHREAD_REQUIRE(pthread_mutex_lock(&mutex)); 138 pthread_cleanup_push(once3_cleanup, &mutex); 139 PTHREAD_REQUIRE(pthread_once(&once, once3_ofunc)); 140 pthread_cleanup_pop(1); 141 142 return NULL; 143 } 144 145 static void 146 handler(int sig, siginfo_t *info, void *ctx) 147 { 148 atf_tc_fail("Signal handler was called; " 149 "main thread deadlocked in pthread_once()"); 150 } 151 152 ATF_TC(once3); 153 ATF_TC_HEAD(once3, tc) 154 { 155 atf_tc_set_md_var(tc, "descr", "Checks pthread_once()"); 156 } 157 ATF_TC_BODY(once3, tc) 158 { 159 pthread_t thread; 160 struct sigaction act; 161 struct itimerval it; 162 printf("Test 3 of pthread_once() (test versus cancellation)\n"); 163 164 act.sa_sigaction = handler; 165 sigemptyset(&act.sa_mask); 166 act.sa_flags = SA_SIGINFO; 167 sigaction(SIGALRM, &act, NULL); 168 169 timerclear(&it.it_value); 170 it.it_value.tv_usec = 500000; 171 timerclear(&it.it_interval); 172 setitimer(ITIMER_REAL, &it, NULL); 173 174 PTHREAD_REQUIRE(pthread_mutex_lock(&mutex)); 175 PTHREAD_REQUIRE(pthread_create(&thread, NULL, once3_threadfunc, NULL)); 176 PTHREAD_REQUIRE(pthread_cancel(thread)); 177 PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex)); 178 PTHREAD_REQUIRE(pthread_join(thread, NULL)); 179 180 PTHREAD_REQUIRE(pthread_once(&once, ofunc)); 181 182 /* Cancel timer */ 183 timerclear(&it.it_value); 184 setitimer(ITIMER_REAL, &it, NULL); 185 186 printf("Test succeeded\n"); 187 } 188 189 ATF_TP_ADD_TCS(tp) 190 { 191 ATF_TP_ADD_TC(tp, once1); 192 ATF_TP_ADD_TC(tp, once2); 193 ATF_TP_ADD_TC(tp, once3); 194 195 return atf_no_error(); 196 } 197