about summary refs log tree commit diff
path: root/src/thread/pthread_barrier_wait.c
blob: 7bfadb9532195bfad63561032e5e3f2fde376f12 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include "pthread_impl.h"

int pthread_barrier_wait(pthread_barrier_t *b)
{
	int cur;

	/* Trivial case: count was set at 1 */
	if (!b->_b_limit) return PTHREAD_BARRIER_SERIAL_THREAD;

	/* Wait for anyone still suspended at previous use of barrier */
	while ((cur=b->_b_left))
		__wait(&b->_b_left, &b->_b_waiters, cur, 0);

	/* If we are the last to reach barrier, reset it and wake others */
	if (a_fetch_add(&b->_b_count, 1) == b->_b_limit) {
		b->_b_left = b->_b_limit;
		b->_b_count = 0;
		__wake(&b->_b_count, -1, 0);
		return PTHREAD_BARRIER_SERIAL_THREAD;
	}

	/* Wait for our peers to reach the barrier */
	while ((cur=b->_b_count))
		__wait(&b->_b_count, 0, cur, 0);

	/* If we're the last to wake up and barrier is awaiting reuse */
	if (a_fetch_add(&b->_b_left, -1) == 1 && b->_b_waiters)
		__wake(&b->_b_left, -1, 0);

	return 0;
}