pthread_cond_timedwait.c 1.5 KB
Newer Older
R
Rich Felker 已提交
1 2
#include "pthread_impl.h"

3 4 5 6 7
struct cm {
	pthread_cond_t *c;
	pthread_mutex_t *m;
};

8 9
static void unwait(pthread_cond_t *c, pthread_mutex_t *m)
{
10 11
	/* Removing a waiter is non-trivial if we could be using requeue
	 * based broadcast signals, due to mutex access issues, etc. */
12

13 14 15 16
	if (c->_c_mutex == (void *)-1) {
		a_dec(&c->_c_waiters);
		return;
	}
17

18 19 20
	while (a_swap(&c->_c_lock, 1))
		__wait(&c->_c_lock, &c->_c_lockwait, 1, 1);

21 22
	if (c->_c_waiters2) c->_c_waiters2--;
	else a_dec(&m->_m_waiters);
23 24 25

	a_store(&c->_c_lock, 0);
	if (c->_c_lockwait) __wake(&c->_c_lock, 1, 1);
26 27
}

28
static void cleanup(void *p)
R
Rich Felker 已提交
29
{
30
	struct cm *cm = p;
31
	unwait(cm->c, cm->m);
32
	pthread_mutex_lock(cm->m);
R
Rich Felker 已提交
33 34 35 36
}

int pthread_cond_timedwait(pthread_cond_t *c, pthread_mutex_t *m, const struct timespec *ts)
{
37
	struct cm cm = { .c=c, .m=m };
38
	int r, e=0, seq;
R
Rich Felker 已提交
39

40 41
	if (m->_m_type && (m->_m_lock&INT_MAX) != __pthread_self()->tid)
		return EPERM;
42

43 44 45
	if (ts && ts->tv_nsec >= 1000000000UL)
		return EINVAL;

R
Rich Felker 已提交
46
	pthread_testcancel();
R
Rich Felker 已提交
47

R
Rich Felker 已提交
48
	a_inc(&c->_c_waiters);
49 50 51 52 53 54 55 56 57

	if (c->_c_mutex != (void *)-1) {
		c->_c_mutex = m;
		while (a_swap(&c->_c_lock, 1))
			__wait(&c->_c_lock, &c->_c_lockwait, 1, 1);
		c->_c_waiters2++;
		a_store(&c->_c_lock, 0);
		if (c->_c_lockwait) __wake(&c->_c_lock, 1, 1);
	}
58

59
	seq = c->_c_seq;
60

61
	pthread_mutex_unlock(m);
R
Rich Felker 已提交
62

63 64
	do e = __timedwait(&c->_c_seq, seq, c->_c_clock, ts, cleanup, &cm, 0);
	while (c->_c_seq == seq && (!e || e==EINTR));
65 66
	if (e == EINTR) e = 0;

67
	unwait(c, m);
R
Rich Felker 已提交
68 69 70 71 72

	if ((r=pthread_mutex_lock(m))) return r;

	return e;
}