xref: /netbsd-src/tests/lib/libpthread/t_barrier.c (revision 0f10aa9dce10d06b907b06520ae9c3443b22e983)
1*0f10aa9dSchristos /* $NetBSD: t_barrier.c,v 1.2 2010/11/03 16:10:22 christos Exp $ */
2ac555471Sjmmv 
3ac555471Sjmmv /*
4ac555471Sjmmv  * Copyright (c) 2008 The NetBSD Foundation, Inc.
5ac555471Sjmmv  * All rights reserved.
6ac555471Sjmmv  *
7ac555471Sjmmv  * Redistribution and use in source and binary forms, with or without
8ac555471Sjmmv  * modification, are permitted provided that the following conditions
9ac555471Sjmmv  * are met:
10ac555471Sjmmv  * 1. Redistributions of source code must retain the above copyright
11ac555471Sjmmv  *    notice, this list of conditions and the following disclaimer.
12ac555471Sjmmv  * 2. Redistributions in binary form must reproduce the above copyright
13ac555471Sjmmv  *    notice, this list of conditions and the following disclaimer in the
14ac555471Sjmmv  *    documentation and/or other materials provided with the distribution.
15ac555471Sjmmv  *
16ac555471Sjmmv  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17ac555471Sjmmv  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18ac555471Sjmmv  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19ac555471Sjmmv  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20ac555471Sjmmv  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21ac555471Sjmmv  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22ac555471Sjmmv  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23ac555471Sjmmv  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24ac555471Sjmmv  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25ac555471Sjmmv  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26ac555471Sjmmv  * POSSIBILITY OF SUCH DAMAGE.
27ac555471Sjmmv  */
28ac555471Sjmmv 
29ac555471Sjmmv #include <sys/cdefs.h>
30ac555471Sjmmv __COPYRIGHT("@(#) Copyright (c) 2008\
31ac555471Sjmmv  The NetBSD Foundation, inc. All rights reserved.");
32*0f10aa9dSchristos __RCSID("$NetBSD: t_barrier.c,v 1.2 2010/11/03 16:10:22 christos Exp $");
33ac555471Sjmmv 
34ac555471Sjmmv #include <pthread.h>
35ac555471Sjmmv #include <stdio.h>
36ac555471Sjmmv #include <unistd.h>
37ac555471Sjmmv 
38ac555471Sjmmv #include <atf-c.h>
39ac555471Sjmmv 
40ac555471Sjmmv #include "h_common.h"
41ac555471Sjmmv 
42ac555471Sjmmv #define COUNT 5
43ac555471Sjmmv 
44ac555471Sjmmv pthread_barrier_t barrier;
45ac555471Sjmmv pthread_mutex_t mutex;
46ac555471Sjmmv int serial_count;
47ac555471Sjmmv int after_barrier_count;
48ac555471Sjmmv 
49ac555471Sjmmv static void *
threadfunc(void * arg)50ac555471Sjmmv threadfunc(void *arg)
51ac555471Sjmmv {
52ac555471Sjmmv 	int which = (int)(long)arg;
53*0f10aa9dSchristos 	int rv;
54ac555471Sjmmv 
55ac555471Sjmmv 	printf("thread %d entering barrier\n", which);
56*0f10aa9dSchristos 	rv = pthread_barrier_wait(&barrier);
57*0f10aa9dSchristos 	printf("thread %d leaving barrier -> %d\n", which, rv);
58ac555471Sjmmv 
59ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
60ac555471Sjmmv 	after_barrier_count++;
61*0f10aa9dSchristos 	if (rv == PTHREAD_BARRIER_SERIAL_THREAD)
62ac555471Sjmmv 		serial_count++;
63ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
64ac555471Sjmmv 
65ac555471Sjmmv 	return NULL;
66ac555471Sjmmv }
67ac555471Sjmmv 
68ac555471Sjmmv ATF_TC(barrier);
ATF_TC_HEAD(barrier,tc)69ac555471Sjmmv ATF_TC_HEAD(barrier, tc)
70ac555471Sjmmv {
71ac555471Sjmmv 	atf_tc_set_md_var(tc, "descr", "Checks barriers");
72ac555471Sjmmv }
ATF_TC_BODY(barrier,tc)73ac555471Sjmmv ATF_TC_BODY(barrier, tc)
74ac555471Sjmmv {
75ac555471Sjmmv 	int i;
76ac555471Sjmmv 	pthread_t new[COUNT];
77ac555471Sjmmv 	void *joinval;
78ac555471Sjmmv 
79ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_mutex_init(&mutex, NULL));
80ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_barrier_init(&barrier, NULL, COUNT));
81ac555471Sjmmv 
82ac555471Sjmmv 	for (i = 0; i < COUNT; i++) {
83ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
84ac555471Sjmmv 		ATF_REQUIRE_EQ(after_barrier_count, 0);
85ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
86ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_create(&new[i], NULL, threadfunc,
87ac555471Sjmmv 						(void *)(long)i));
88ac555471Sjmmv 		sleep(2);
89ac555471Sjmmv 	}
90ac555471Sjmmv 
91ac555471Sjmmv 	for (i = 0; i < COUNT; i++) {
92ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_join(new[i], &joinval));
93ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
94ac555471Sjmmv 		ATF_REQUIRE(after_barrier_count > i);
95ac555471Sjmmv 		PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
96ac555471Sjmmv 		printf("main joined with thread %d\n", i);
97ac555471Sjmmv 	}
98ac555471Sjmmv 
99ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
100ac555471Sjmmv 	ATF_REQUIRE_EQ(after_barrier_count, COUNT);
101ac555471Sjmmv 	PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
102ac555471Sjmmv 	ATF_REQUIRE_EQ(serial_count, 1);
103ac555471Sjmmv }
104ac555471Sjmmv 
ATF_TP_ADD_TCS(tp)105ac555471Sjmmv ATF_TP_ADD_TCS(tp)
106ac555471Sjmmv {
107ac555471Sjmmv 	ATF_TP_ADD_TC(tp, barrier);
108ac555471Sjmmv 
109ac555471Sjmmv 	return atf_no_error();
110ac555471Sjmmv }
111