contrib/netbsd-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 #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 }