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