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