NetBSD/lib/libpthread/pthread_cond.c

443 lines
13 KiB
C

/* $NetBSD: pthread_cond.c,v 1.39 2007/11/19 15:14:12 ad Exp $ */
/*-
* Copyright (c) 2001, 2006, 2007 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Nathan J. Williams and Andrew Doran.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the NetBSD
* Foundation, Inc. and its contributors.
* 4. Neither the name of The NetBSD Foundation nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <sys/cdefs.h>
__RCSID("$NetBSD: pthread_cond.c,v 1.39 2007/11/19 15:14:12 ad Exp $");
#include <errno.h>
#include <sys/time.h>
#include <sys/types.h>
#include "pthread.h"
#include "pthread_int.h"
int _sys_nanosleep(const struct timespec *, struct timespec *);
extern int pthread__started;
static int pthread_cond_wait_nothread(pthread_t, pthread_mutex_t *,
const struct timespec *);
__strong_alias(__libc_cond_init,pthread_cond_init)
__strong_alias(__libc_cond_signal,pthread_cond_signal)
__strong_alias(__libc_cond_broadcast,pthread_cond_broadcast)
__strong_alias(__libc_cond_wait,pthread_cond_wait)
__strong_alias(__libc_cond_timedwait,pthread_cond_timedwait)
__strong_alias(__libc_cond_destroy,pthread_cond_destroy)
int
pthread_cond_init(pthread_cond_t *cond, const pthread_condattr_t *attr)
{
pthread__error(EINVAL, "Invalid condition variable attribute",
(attr == NULL) || (attr->ptca_magic == _PT_CONDATTR_MAGIC));
cond->ptc_magic = _PT_COND_MAGIC;
pthread_lockinit(&cond->ptc_lock);
PTQ_INIT(&cond->ptc_waiters);
cond->ptc_mutex = NULL;
return 0;
}
int
pthread_cond_destroy(pthread_cond_t *cond)
{
pthread__error(EINVAL, "Invalid condition variable",
cond->ptc_magic == _PT_COND_MAGIC);
pthread__error(EBUSY, "Destroying condition variable in use",
cond->ptc_mutex == NULL);
cond->ptc_magic = _PT_COND_DEAD;
return 0;
}
int
pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex)
{
pthread_t self;
pthread__error(EINVAL, "Invalid condition variable",
cond->ptc_magic == _PT_COND_MAGIC);
pthread__error(EINVAL, "Invalid mutex",
mutex->ptm_magic == _PT_MUTEX_MAGIC);
pthread__error(EPERM, "Mutex not locked in condition wait",
mutex->ptm_owner != NULL);
self = pthread__self();
/* Just hang out for a while if threads aren't running yet. */
if (__predict_false(pthread__started == 0))
return pthread_cond_wait_nothread(self, mutex, NULL);
if (__predict_false(self->pt_cancel))
pthread_exit(PTHREAD_CANCELED);
/*
* Note this thread as waiting on the CV. To ensure good
* performance it's critical that the spinlock is held for
* as short a time as possible - that means no system calls.
*/
pthread__spinlock(self, &cond->ptc_lock);
#ifdef ERRORCHECK
if (cond->ptc_mutex == NULL)
cond->ptc_mutex = mutex;
else {
pthread__error(EINVAL,
"Multiple mutexes used for condition wait",
cond->ptc_mutex == mutex);
}
#else
cond->ptc_mutex = mutex;
#endif
PTQ_INSERT_HEAD(&cond->ptc_waiters, self, pt_sleep);
self->pt_signalled = 0;
self->pt_sleeponq = 1;
self->pt_sleepobj = &cond->ptc_waiters;
pthread__spinunlock(self, &cond->ptc_lock);
/*
* Before releasing the mutex, note that this thread is
* about to block by setting the willpark flag. If there
* is a single waiter on the mutex, setting the flag will
* defer restarting it until calling into the kernel to
* park, saving a syscall & involuntary context switch.
*/
self->pt_willpark = 1;
pthread_mutex_unlock(mutex);
(void)pthread__park(self, &cond->ptc_lock, &cond->ptc_waiters,
NULL, 1, &mutex->ptm_blocked);
pthread_mutex_lock(mutex);
/*
* If we awoke abnormally the waiters list will have been
* made empty by the current thread (in pthread__park()),
* so we can check the value safely without locking.
*
* Otherwise, it will have been updated by whichever thread
* last issued a wakeup.
*/
if (PTQ_EMPTY(&cond->ptc_waiters) && cond->ptc_mutex != NULL) {
pthread__spinlock(self, &cond->ptc_lock);
if (PTQ_EMPTY(&cond->ptc_waiters))
cond->ptc_mutex = NULL;
pthread__spinunlock(self, &cond->ptc_lock);
}
/*
* If we have cancelled then exit. POSIX dictates that the
* mutex must be held when we action the cancellation.
*/
if (__predict_false(self->pt_cancel)) {
if (self->pt_signalled)
pthread_cond_signal(cond);
pthread_exit(PTHREAD_CANCELED);
}
return 0;
}
int
pthread_cond_timedwait(pthread_cond_t *cond, pthread_mutex_t *mutex,
const struct timespec *abstime)
{
pthread_t self;
int retval;
pthread__error(EINVAL, "Invalid condition variable",
cond->ptc_magic == _PT_COND_MAGIC);
pthread__error(EINVAL, "Invalid mutex",
mutex->ptm_magic == _PT_MUTEX_MAGIC);
pthread__error(EPERM, "Mutex not locked in condition wait",
mutex->ptm_owner != NULL);
pthread__error(EINVAL, "Invalid wait time",
(abstime->tv_sec >= 0) &&
(abstime->tv_nsec >= 0) && (abstime->tv_nsec < 1000000000));
self = pthread__self();
/* Just hang out for a while if threads aren't running yet. */
if (__predict_false(pthread__started == 0))
return pthread_cond_wait_nothread(self, mutex, abstime);
if (__predict_false(self->pt_cancel))
pthread_exit(PTHREAD_CANCELED);
/*
* Note this thread as waiting on the CV. To ensure good
* performance it's critical that the spinlock is held for
* as short a time as possible - that means no system calls.
*/
pthread__spinlock(self, &cond->ptc_lock);
#ifdef ERRORCHECK
if (cond->ptc_mutex == NULL)
cond->ptc_mutex = mutex;
else {
pthread__error(EINVAL,
"Multiple mutexes used for condition wait",
cond->ptc_mutex == mutex);
}
#else
cond->ptc_mutex = mutex;
#endif
PTQ_INSERT_HEAD(&cond->ptc_waiters, self, pt_sleep);
self->pt_signalled = 0;
self->pt_sleeponq = 1;
self->pt_sleepobj = &cond->ptc_waiters;
pthread__spinunlock(self, &cond->ptc_lock);
/*
* Before releasing the mutex, note that this thread is
* about to block by setting the willpark flag. If there
* is a single waiter on the mutex, setting the flag will
* defer restarting it until calling into the kernel to
* park, saving a syscall & involuntary context switch.
*/
self->pt_willpark = 1;
pthread_mutex_unlock(mutex);
retval = pthread__park(self, &cond->ptc_lock, &cond->ptc_waiters,
abstime, 1, &mutex->ptm_blocked);
pthread_mutex_lock(mutex);
/*
* If we awoke abnormally the waiters list will have been
* made empty by the current thread (in pthread__park()),
* so we can check the value safely without locking.
*
* Otherwise, it will have been updated by whichever thread
* last issued a wakeup.
*/
if (PTQ_EMPTY(&cond->ptc_waiters) && cond->ptc_mutex != NULL) {
pthread__spinlock(self, &cond->ptc_lock);
if (PTQ_EMPTY(&cond->ptc_waiters))
cond->ptc_mutex = NULL;
pthread__spinunlock(self, &cond->ptc_lock);
}
/*
* If we have cancelled then exit. POSIX dictates that the
* mutex must be held when we action the cancellation.
*/
if (__predict_false(self->pt_cancel | retval)) {
if (self->pt_signalled)
pthread_cond_signal(cond);
if (self->pt_cancel)
pthread_exit(PTHREAD_CANCELED);
}
return retval;
}
int
pthread_cond_signal(pthread_cond_t *cond)
{
pthread_t self, signaled;
pthread_mutex_t *mutex;
pthread__error(EINVAL, "Invalid condition variable",
cond->ptc_magic == _PT_COND_MAGIC);
if (PTQ_EMPTY(&cond->ptc_waiters))
return 0;
self = pthread__self();
pthread__spinlock(self, &cond->ptc_lock);
/*
* Find a thread that is still blocked (no pending wakeup).
* A wakeup can be pending if we have interrupted unpark_all
* as it releases the interlock.
*/
PTQ_FOREACH(signaled, &cond->ptc_waiters, pt_sleep) {
if (signaled->pt_sleepobj != NULL)
break;
}
if (__predict_false(signaled == NULL)) {
cond->ptc_mutex = NULL;
pthread__spinunlock(self, &cond->ptc_lock);
return 0;
}
/*
* Pull the thread off the queue, and set pt_signalled.
*
* After resuming execution, the thread must check to see if it
* has been restarted as a result of pthread_cond_signal(). If it
* has, but cannot take the wakeup (because of eg a timeout) then
* try to ensure that another thread sees it. This is necessary
* because there may be multiple waiters, and at least one should
* take the wakeup if possible.
*/
PTQ_REMOVE(&cond->ptc_waiters, signaled, pt_sleep);
mutex = cond->ptc_mutex;
if (PTQ_EMPTY(&cond->ptc_waiters))
cond->ptc_mutex = NULL;
signaled->pt_signalled = 1;
/*
* For all valid uses of pthread_cond_signal(), the caller will
* hold the mutex that the target is using to synchronize with.
* To avoid the target awakening and immediatley blocking on the
* mutex, transfer the thread to be awoken to the current thread's
* deferred wakeup list. The waiter will be set running when the
* caller (this thread) releases the mutex.
*/
if (mutex != NULL && self->pt_nwaiters < pthread__unpark_max &&
pthread__mutex_deferwake(self, mutex)) {
signaled->pt_sleepobj = NULL;
signaled->pt_sleeponq = 0;
pthread__spinunlock(self, &cond->ptc_lock);
self->pt_waiters[self->pt_nwaiters++] = signaled->pt_lid;
} else {
pthread__unpark(self, &cond->ptc_lock,
&cond->ptc_waiters, signaled);
}
return 0;
}
int
pthread_cond_broadcast(pthread_cond_t *cond)
{
pthread_t self, signaled, next;
pthread_mutex_t *mutex;
pthread__error(EINVAL, "Invalid condition variable",
cond->ptc_magic == _PT_COND_MAGIC);
if (PTQ_EMPTY(&cond->ptc_waiters))
return 0;
self = pthread__self();
pthread__spinlock(self, &cond->ptc_lock);
mutex = cond->ptc_mutex;
cond->ptc_mutex = NULL;
/*
* Try to defer waking threads (see pthread_cond_signal()).
* Only transfer waiters for which there is no pending wakeup.
*/
if (mutex != NULL && pthread__mutex_deferwake(self, mutex)) {
for (signaled = PTQ_FIRST(&cond->ptc_waiters);
signaled != NULL;
signaled = next) {
next = PTQ_NEXT(signaled, pt_sleep);
if (__predict_false(signaled->pt_sleepobj == NULL))
continue;
if (self->pt_nwaiters == pthread__unpark_max) {
/* Overflow, take the slow path. */
break;
}
PTQ_REMOVE(&cond->ptc_waiters, signaled, pt_sleep);
signaled->pt_sleepobj = NULL;
signaled->pt_sleeponq = 0;
self->pt_waiters[self->pt_nwaiters++] =
signaled->pt_lid;
}
if (signaled == NULL) {
/* Anything more to do? */
pthread__spinunlock(self, &cond->ptc_lock);
return 0;
}
}
pthread__unpark_all(self, &cond->ptc_lock, &cond->ptc_waiters);
return 0;
}
int
pthread_condattr_init(pthread_condattr_t *attr)
{
attr->ptca_magic = _PT_CONDATTR_MAGIC;
return 0;
}
int
pthread_condattr_destroy(pthread_condattr_t *attr)
{
pthread__error(EINVAL, "Invalid condition variable attribute",
attr->ptca_magic == _PT_CONDATTR_MAGIC);
attr->ptca_magic = _PT_CONDATTR_DEAD;
return 0;
}
/* Utility routine to hang out for a while if threads haven't started yet. */
static int
pthread_cond_wait_nothread(pthread_t self, pthread_mutex_t *mutex,
const struct timespec *abstime)
{
struct timespec now, diff;
int retval;
if (abstime == NULL) {
diff.tv_sec = 99999999;
diff.tv_nsec = 0;
} else {
clock_gettime(CLOCK_REALTIME, &now);
if (timespeccmp(abstime, &now, <))
timespecclear(&diff);
else
timespecsub(abstime, &now, &diff);
}
do {
pthread__testcancel(self);
pthread_mutex_unlock(mutex);
retval = _sys_nanosleep(&diff, NULL);
pthread_mutex_lock(mutex);
} while (abstime == NULL && retval == 0);
pthread__testcancel(self);
if (retval == 0)
return ETIMEDOUT;
else
/* spurious wakeup */
return 0;
}