tests/lib/libpthread/t_sem.c

   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
 113 static sem_t sem;
 114
 115 ATF_TC(named);
 116 ATF_TC_HEAD(named, tc)
 117 {
 118         atf_tc_set_md_var(tc, "descr", "Checks named semaphores");
 119 }
 120 ATF_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
 137 ATF_TC(unnamed);
 138 ATF_TC_HEAD(unnamed, tc)
 139 {
 140         atf_tc_set_md_var(tc, "descr", "Checks unnamed semaphores");
 141 }
 142
 143 static void *
 144 entry(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
 156 ATF_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
 210 static void
 211 sighandler(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
 219 static void
 220 alarm_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
 230 static void *
 231 threadfunc(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
 257 static void
 258 before_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
 275 ATF_TC(before_start_no_threads);
 276 ATF_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 }
 282 ATF_TC_BODY(before_start_no_threads, tc)
 283 {
 284         before_start_test(false);
 285 }
 286
 287 ATF_TC(before_start_one_thread);
 288 ATF_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 }
 294 ATF_TC_BODY(before_start_one_thread, tc)
 295 {
 296         before_start_test(true);
 297 }
 298
 299 ATF_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 }