Home | History | Annotate | Line # | Download | only in libpthread 
      1 /* $NetBSD: t_barrier.c,v 1.2 2010/11/03 16:10:22 christos Exp $ */
      2 
      3 /*
      4  * Copyright (c) 2008 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 #include <sys/cdefs.h>
     30 __COPYRIGHT("@(#) Copyright (c) 2008\
     31  The NetBSD Foundation, inc. All rights reserved.");
     32 __RCSID("$NetBSD: t_barrier.c,v 1.2 2010/11/03 16:10:22 christos Exp $");
     33 
     34 #include <pthread.h>
     35 #include <stdio.h>
     36 #include <unistd.h>
     37 
     38 #include <atf-c.h>
     39 
     40 #include "h_common.h"
     41 
     42 #define COUNT 5
     43 
     44 pthread_barrier_t barrier;
     45 pthread_mutex_t mutex;
     46 int serial_count;
     47 int after_barrier_count;
     48 
     49 static void *
     50 threadfunc(void *arg)
     51 {
     52 	int which = (int)(long)arg;
     53 	int rv;
     54 
     55 	printf("thread %d entering barrier\n", which);
     56 	rv = pthread_barrier_wait(&barrier);
     57 	printf("thread %d leaving barrier -> %d\n", which, rv);
     58 
     59 	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
     60 	after_barrier_count++;
     61 	if (rv == PTHREAD_BARRIER_SERIAL_THREAD)
     62 		serial_count++;
     63 	PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
     64 
     65 	return NULL;
     66 }
     67 
     68 ATF_TC(barrier);
     69 ATF_TC_HEAD(barrier, tc)
     70 {
     71 	atf_tc_set_md_var(tc, "descr", "Checks barriers");
     72 }
     73 ATF_TC_BODY(barrier, tc)
     74 {
     75 	int i;
     76 	pthread_t new[COUNT];
     77 	void *joinval;
     78 
     79 	PTHREAD_REQUIRE(pthread_mutex_init(&mutex, NULL));
     80 	PTHREAD_REQUIRE(pthread_barrier_init(&barrier, NULL, COUNT));
     81 
     82 	for (i = 0; i < COUNT; i++) {
     83 		PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
     84 		ATF_REQUIRE_EQ(after_barrier_count, 0);
     85 		PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
     86 		PTHREAD_REQUIRE(pthread_create(&new[i], NULL, threadfunc,
     87 						(void *)(long)i));
     88 		sleep(2);
     89 	}
     90 
     91 	for (i = 0; i < COUNT; i++) {
     92 		PTHREAD_REQUIRE(pthread_join(new[i], &joinval));
     93 		PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
     94 		ATF_REQUIRE(after_barrier_count > i);
     95 		PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
     96 		printf("main joined with thread %d\n", i);
     97 	}
     98 
     99 	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
    100 	ATF_REQUIRE_EQ(after_barrier_count, COUNT);
    101 	PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
    102 	ATF_REQUIRE_EQ(serial_count, 1);
    103 }
    104 
    105 ATF_TP_ADD_TCS(tp)
    106 {
    107 	ATF_TP_ADD_TC(tp, barrier);
    108 
    109 	return atf_no_error();
    110 }
    111