tests/lib/libpthread/t_sem.c

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