t_sem.c revision 272458
1/* $NetBSD: t_sem.c,v 1.7 2012/03/09 19:46:37 joerg 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.7 2012/03/09 19:46:37 joerg 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#include <atf-c/config.h> 103 104#include "h_common.h" 105 106#define NTHREADS 10 107 108#define _LIBC_R_ 109 110#define SEM_REQUIRE(x) \ 111 ATF_REQUIRE_EQ_MSG(x, 0, "%s", strerror(errno)) 112 113static sem_t sem; 114 115ATF_TC(named); 116ATF_TC_HEAD(named, tc) 117{ 118 atf_tc_set_md_var(tc, "descr", "Checks named semaphores"); 119} 120ATF_TC_BODY(named, tc) 121{ 122 sem_t *semp; 123 124 ATF_REQUIRE_MSG(-1 != sysconf(_SC_SEMAPHORES), "%s", strerror(errno)); 125 126 printf("Test begin\n"); 127 128 (void) sem_unlink("/foo"); 129 semp = sem_open("/foo", O_CREAT | O_EXCL, 0644, 0); 130 ATF_REQUIRE_MSG(semp != SEM_FAILED, "%s", strerror(errno)); 131 SEM_REQUIRE(sem_close(semp)); 132 SEM_REQUIRE(sem_unlink("/foo")); 133 134 printf("Test end\n"); 135} 136 137ATF_TC(unnamed); 138ATF_TC_HEAD(unnamed, tc) 139{ 140 atf_tc_set_md_var(tc, "descr", "Checks unnamed semaphores"); 141} 142 143static void * 144entry(void * a_arg) 145{ 146 pthread_t self = pthread_self(); 147 sem_t *semp = (sem_t *) a_arg; 148 149 printf("Thread %p waiting for semaphore...\n", self); 150 sem_wait(semp); 151 printf("Thread %p got semaphore\n", self); 152 153 return NULL; 154} 155 156ATF_TC_BODY(unnamed, tc) 157{ 158 sem_t sem_a, sem_b; 159 pthread_t threads[NTHREADS]; 160 unsigned i, j; 161 int val; 162 163 ATF_REQUIRE_MSG(-1 != sysconf(_SC_SEMAPHORES), "%s", strerror(errno)); 164 165 printf("Test begin\n"); 166 167 SEM_REQUIRE(sem_init(&sem_b, 0, 0)); 168 SEM_REQUIRE(sem_getvalue(&sem_b, &val)); 169 ATF_REQUIRE_EQ(0, val); 170 171 SEM_REQUIRE(sem_post(&sem_b)); 172 SEM_REQUIRE(sem_getvalue(&sem_b, &val)); 173 ATF_REQUIRE_EQ(1, val); 174 175 SEM_REQUIRE(sem_wait(&sem_b)); 176 ATF_REQUIRE_EQ(sem_trywait(&sem_b), -1); 177 ATF_REQUIRE_EQ(errno, EAGAIN); 178 SEM_REQUIRE(sem_post(&sem_b)); 179 SEM_REQUIRE(sem_trywait(&sem_b)); 180 SEM_REQUIRE(sem_post(&sem_b)); 181 SEM_REQUIRE(sem_wait(&sem_b)); 182 SEM_REQUIRE(sem_post(&sem_b)); 183 184 SEM_REQUIRE(sem_destroy(&sem_b)); 185 186 SEM_REQUIRE(sem_init(&sem_a, 0, 0)); 187 188 for (j = 0; j < 2; j++) { 189 for (i = 0; i < NTHREADS; i++) { 190 PTHREAD_REQUIRE(pthread_create(&threads[i], NULL, 191 entry, (void *) &sem_a)); 192 } 193 194 for (i = 0; i < NTHREADS; i++) { 195 usleep(10000); 196 printf("main loop %u: posting...\n", j+1); 197 SEM_REQUIRE(sem_post(&sem_a)); 198 } 199 200 for (i = 0; i < NTHREADS; i++) { 201 PTHREAD_REQUIRE(pthread_join(threads[i], NULL)); 202 } 203 } 204 205 SEM_REQUIRE(sem_destroy(&sem_a)); 206 207 printf("Test end\n"); 208} 209 210static void 211sighandler(int signo) 212{ 213 /* printf("signal %d\n", signo); */ 214 215 ATF_REQUIRE_EQ_MSG(signo, SIGALRM, "unexpected signal"); 216 SEM_REQUIRE(sem_post(&sem)); 217} 218 219static void 220alarm_ms(const int ms) 221{ 222 struct itimerval timer; 223 timer.it_interval.tv_sec = 0; 224 timer.it_interval.tv_usec = 0; 225 timer.it_value.tv_sec = 0; 226 timer.it_value.tv_usec = ms * 1000; 227 ATF_REQUIRE(setitimer(ITIMER_REAL, &timer, NULL) == 0); 228} 229 230static void * 231threadfunc(void *arg) 232{ 233 int i, ret; 234 235 printf("Entering loop\n"); 236 for (i = 0; i < 500; ) { 237 if ((i & 1) != 0) { 238 do { 239 ret = sem_wait(&sem); 240 } while (ret == -1 && errno == EINTR); 241 ATF_REQUIRE(ret == 0); 242 } else { 243 ret = sem_trywait(&sem); 244 if (ret == -1) { 245 ATF_REQUIRE(errno == EAGAIN); 246 continue; 247 } 248 } 249 printf("%s: %d\n", __func__, i); 250 alarm_ms(5); 251 i++; 252 } 253 254 return NULL; 255} 256 257static void 258before_start_test(const bool use_pthread) 259{ 260 pthread_t t; 261 262 SEM_REQUIRE(sem_init(&sem, 0, 0)); 263 ATF_REQUIRE(SIG_ERR != signal(SIGALRM, sighandler)); 264 265 alarm_ms(5); 266 267 if (use_pthread) { 268 PTHREAD_REQUIRE(pthread_create(&t, NULL, threadfunc, NULL)); 269 PTHREAD_REQUIRE(pthread_join(t, NULL)); 270 } else { 271 threadfunc(NULL); 272 } 273} 274 275ATF_TC(before_start_no_threads); 276ATF_TC_HEAD(before_start_no_threads, tc) 277{ 278 atf_tc_set_md_var(tc, "descr", "Checks using semaphores without any " 279 "thread running"); 280 atf_tc_set_md_var(tc, "timeout", "40"); 281} 282ATF_TC_BODY(before_start_no_threads, tc) 283{ 284 before_start_test(false); 285} 286 287ATF_TC(before_start_one_thread); 288ATF_TC_HEAD(before_start_one_thread, tc) 289{ 290 atf_tc_set_md_var(tc, "descr", "Checks using semaphores before " 291 "starting one thread"); 292 atf_tc_set_md_var(tc, "timeout", "40"); 293} 294ATF_TC_BODY(before_start_one_thread, tc) 295{ 296 before_start_test(true); 297} 298 299ATF_TP_ADD_TCS(tp) 300{ 301 ATF_TP_ADD_TC(tp, named); 302 ATF_TP_ADD_TC(tp, unnamed); 303 ATF_TP_ADD_TC(tp, before_start_no_threads); 304 ATF_TP_ADD_TC(tp, before_start_one_thread); 305 306 return atf_no_error(); 307} 308