1 /* $NetBSD: t_sem.c,v 1.8 2014/11/04 00:20:19 justin 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 /*- 30 * Copyright (c)2004 YAMAMOTO Takashi, 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 * 42 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 43 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 44 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 45 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 46 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 47 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 48 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 49 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 50 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 51 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 52 * SUCH DAMAGE. 53 */ 54 55 /**************************************************************************** 56 * 57 * Copyright (C) 2000 Jason Evans <jasone@freebsd.org>. 58 * All rights reserved. 59 * 60 * Redistribution and use in source and binary forms, with or without 61 * modification, are permitted provided that the following conditions 62 * are met: 63 * 1. Redistributions of source code must retain the above copyright 64 * notice(s), this list of conditions and the following disclaimer as 65 * the first lines of this file unmodified other than the possible 66 * addition of one or more copyright notices. 67 * 2. Redistributions in binary form must reproduce the above copyright 68 * notice(s), this list of conditions and the following disclaimer in 69 * the documentation and/or other materials provided with the 70 * distribution. 71 * 72 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) ``AS IS'' AND ANY 73 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 74 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 75 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) BE 76 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 77 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 78 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 79 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 80 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE 81 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, 82 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 83 * 84 ****************************************************************************/ 85 86 #include <sys/cdefs.h> 87 __COPYRIGHT("@(#) Copyright (c) 2008, 2010\ 88 The NetBSD Foundation, inc. All rights reserved."); 89 __RCSID("$NetBSD: t_sem.c,v 1.8 2014/11/04 00:20:19 justin Exp $"); 90 91 #include <errno.h> 92 #include <fcntl.h> 93 #include <pthread.h> 94 #include <semaphore.h> 95 #include <signal.h> 96 #include <stdio.h> 97 #include <stdlib.h> 98 #include <string.h> 99 #include <unistd.h> 100 101 #include <atf-c.h> 102 103 #include "h_common.h" 104 105 #define NTHREADS 10 106 107 #define _LIBC_R_ 108 109 #define SEM_REQUIRE(x) \ 110 ATF_REQUIRE_EQ_MSG(x, 0, "%s", strerror(errno)) 111 112 static sem_t sem; 113 114 #ifdef __FreeBSD__ 115 #include <sys/time.h> 116 #endif 117 118 ATF_TC(named); 119 ATF_TC_HEAD(named, tc) 120 { 121 atf_tc_set_md_var(tc, "descr", "Checks named semaphores"); 122 } 123 ATF_TC_BODY(named, tc) 124 { 125 sem_t *semp; 126 127 ATF_REQUIRE_MSG(-1 != sysconf(_SC_SEMAPHORES), "%s", strerror(errno)); 128 129 printf("Test begin\n"); 130 131 (void) sem_unlink("/foo"); 132 semp = sem_open("/foo", O_CREAT | O_EXCL, 0644, 0); 133 ATF_REQUIRE_MSG(semp != SEM_FAILED, "%s", strerror(errno)); 134 SEM_REQUIRE(sem_close(semp)); 135 SEM_REQUIRE(sem_unlink("/foo")); 136 137 printf("Test end\n"); 138 } 139 140 ATF_TC(unnamed); 141 ATF_TC_HEAD(unnamed, tc) 142 { 143 atf_tc_set_md_var(tc, "descr", "Checks unnamed semaphores"); 144 } 145 146 static void * 147 entry(void * a_arg) 148 { 149 pthread_t self = pthread_self(); 150 sem_t *semp = (sem_t *) a_arg; 151 152 printf("Thread %p waiting for semaphore...\n", self); 153 sem_wait(semp); 154 printf("Thread %p got semaphore\n", self); 155 156 return NULL; 157 } 158 159 ATF_TC_BODY(unnamed, tc) 160 { 161 sem_t sem_a, sem_b; 162 pthread_t threads[NTHREADS]; 163 unsigned i, j; 164 int val; 165 166 ATF_REQUIRE_MSG(-1 != sysconf(_SC_SEMAPHORES), "%s", strerror(errno)); 167 168 printf("Test begin\n"); 169 170 SEM_REQUIRE(sem_init(&sem_b, 0, 0)); 171 SEM_REQUIRE(sem_getvalue(&sem_b, &val)); 172 ATF_REQUIRE_EQ(0, val); 173 174 SEM_REQUIRE(sem_post(&sem_b)); 175 SEM_REQUIRE(sem_getvalue(&sem_b, &val)); 176 ATF_REQUIRE_EQ(1, val); 177 178 SEM_REQUIRE(sem_wait(&sem_b)); 179 ATF_REQUIRE_EQ(sem_trywait(&sem_b), -1); 180 ATF_REQUIRE_EQ(errno, EAGAIN); 181 SEM_REQUIRE(sem_post(&sem_b)); 182 SEM_REQUIRE(sem_trywait(&sem_b)); 183 SEM_REQUIRE(sem_post(&sem_b)); 184 SEM_REQUIRE(sem_wait(&sem_b)); 185 SEM_REQUIRE(sem_post(&sem_b)); 186 187 SEM_REQUIRE(sem_destroy(&sem_b)); 188 189 SEM_REQUIRE(sem_init(&sem_a, 0, 0)); 190 191 for (j = 0; j < 2; j++) { 192 for (i = 0; i < NTHREADS; i++) { 193 PTHREAD_REQUIRE(pthread_create(&threads[i], NULL, 194 entry, (void *) &sem_a)); 195 } 196 197 for (i = 0; i < NTHREADS; i++) { 198 usleep(10000); 199 printf("main loop %u: posting...\n", j+1); 200 SEM_REQUIRE(sem_post(&sem_a)); 201 } 202 203 for (i = 0; i < NTHREADS; i++) { 204 PTHREAD_REQUIRE(pthread_join(threads[i], NULL)); 205 } 206 } 207 208 SEM_REQUIRE(sem_destroy(&sem_a)); 209 210 printf("Test end\n"); 211 } 212 213 static void 214 sighandler(int signo) 215 { 216 /* printf("signal %d\n", signo); */ 217 218 ATF_REQUIRE_EQ_MSG(signo, SIGALRM, "unexpected signal"); 219 SEM_REQUIRE(sem_post(&sem)); 220 } 221 222 static void 223 alarm_ms(const int ms) 224 { 225 struct itimerval timer; 226 timer.it_interval.tv_sec = 0; 227 timer.it_interval.tv_usec = 0; 228 timer.it_value.tv_sec = 0; 229 timer.it_value.tv_usec = ms * 1000; 230 ATF_REQUIRE(setitimer(ITIMER_REAL, &timer, NULL) == 0); 231 } 232 233 static void * 234 threadfunc(void *arg) 235 { 236 int i, ret; 237 238 printf("Entering loop\n"); 239 for (i = 0; i < 500; ) { 240 if ((i & 1) != 0) { 241 do { 242 ret = sem_wait(&sem); 243 } while (ret == -1 && errno == EINTR); 244 ATF_REQUIRE(ret == 0); 245 } else { 246 ret = sem_trywait(&sem); 247 if (ret == -1) { 248 ATF_REQUIRE(errno == EAGAIN); 249 continue; 250 } 251 } 252 printf("%s: %d\n", __func__, i); 253 alarm_ms(5); 254 i++; 255 } 256 257 return NULL; 258 } 259 260 static void 261 before_start_test(const bool use_pthread) 262 { 263 pthread_t t; 264 265 SEM_REQUIRE(sem_init(&sem, 0, 0)); 266 ATF_REQUIRE(SIG_ERR != signal(SIGALRM, sighandler)); 267 268 alarm_ms(5); 269 270 if (use_pthread) { 271 PTHREAD_REQUIRE(pthread_create(&t, NULL, threadfunc, NULL)); 272 PTHREAD_REQUIRE(pthread_join(t, NULL)); 273 } else { 274 threadfunc(NULL); 275 } 276 } 277 278 ATF_TC(before_start_no_threads); 279 ATF_TC_HEAD(before_start_no_threads, tc) 280 { 281 atf_tc_set_md_var(tc, "descr", "Checks using semaphores without any " 282 "thread running"); 283 atf_tc_set_md_var(tc, "timeout", "40"); 284 } 285 ATF_TC_BODY(before_start_no_threads, tc) 286 { 287 before_start_test(false); 288 } 289 290 ATF_TC(before_start_one_thread); 291 ATF_TC_HEAD(before_start_one_thread, tc) 292 { 293 atf_tc_set_md_var(tc, "descr", "Checks using semaphores before " 294 "starting one thread"); 295 atf_tc_set_md_var(tc, "timeout", "40"); 296 } 297 ATF_TC_BODY(before_start_one_thread, tc) 298 { 299 before_start_test(true); 300 } 301 302 ATF_TP_ADD_TCS(tp) 303 { 304 ATF_TP_ADD_TC(tp, named); 305 ATF_TP_ADD_TC(tp, unnamed); 306 ATF_TP_ADD_TC(tp, before_start_no_threads); 307 ATF_TP_ADD_TC(tp, before_start_one_thread); 308 309 return atf_no_error(); 310 } 311