103 строки
2.4 KiB
C
103 строки
2.4 KiB
C
/*
|
|
* Semaphore implementation Copyright (c) 2001 Matthew Wilcox, Hewlett-Packard
|
|
*/
|
|
|
|
#include <linux/sched.h>
|
|
#include <linux/spinlock.h>
|
|
#include <linux/errno.h>
|
|
#include <linux/init.h>
|
|
|
|
/*
|
|
* Semaphores are complex as we wish to avoid using two variables.
|
|
* `count' has multiple roles, depending on its value. If it is positive
|
|
* or zero, there are no waiters. The functions here will never be
|
|
* called; see <asm/semaphore.h>
|
|
*
|
|
* When count is -1 it indicates there is at least one task waiting
|
|
* for the semaphore.
|
|
*
|
|
* When count is less than that, there are '- count - 1' wakeups
|
|
* pending. ie if it has value -3, there are 2 wakeups pending.
|
|
*
|
|
* Note that these functions are only called when there is contention
|
|
* on the lock, and as such all this is the "non-critical" part of the
|
|
* whole semaphore business. The critical part is the inline stuff in
|
|
* <asm/semaphore.h> where we want to avoid any extra jumps and calls.
|
|
*/
|
|
void __up(struct semaphore *sem)
|
|
{
|
|
sem->count--;
|
|
wake_up(&sem->wait);
|
|
}
|
|
|
|
#define wakers(count) (-1 - count)
|
|
|
|
#define DOWN_HEAD \
|
|
int ret = 0; \
|
|
DECLARE_WAITQUEUE(wait, current); \
|
|
\
|
|
/* Note that someone is waiting */ \
|
|
if (sem->count == 0) \
|
|
sem->count = -1; \
|
|
\
|
|
/* protected by the sentry still -- use unlocked version */ \
|
|
wait.flags = WQ_FLAG_EXCLUSIVE; \
|
|
__add_wait_queue_tail(&sem->wait, &wait); \
|
|
lost_race: \
|
|
spin_unlock_irq(&sem->sentry); \
|
|
|
|
#define DOWN_TAIL \
|
|
spin_lock_irq(&sem->sentry); \
|
|
if (wakers(sem->count) == 0 && ret == 0) \
|
|
goto lost_race; /* Someone stole our wakeup */ \
|
|
__remove_wait_queue(&sem->wait, &wait); \
|
|
current->state = TASK_RUNNING; \
|
|
if (!waitqueue_active(&sem->wait) && (sem->count < 0)) \
|
|
sem->count = wakers(sem->count);
|
|
|
|
#define UPDATE_COUNT \
|
|
sem->count += (sem->count < 0) ? 1 : - 1;
|
|
|
|
|
|
void __sched __down(struct semaphore * sem)
|
|
{
|
|
DOWN_HEAD
|
|
|
|
for(;;) {
|
|
set_task_state(current, TASK_UNINTERRUPTIBLE);
|
|
/* we can _read_ this without the sentry */
|
|
if (sem->count != -1)
|
|
break;
|
|
schedule();
|
|
}
|
|
|
|
DOWN_TAIL
|
|
UPDATE_COUNT
|
|
}
|
|
|
|
int __sched __down_interruptible(struct semaphore * sem)
|
|
{
|
|
DOWN_HEAD
|
|
|
|
for(;;) {
|
|
set_task_state(current, TASK_INTERRUPTIBLE);
|
|
/* we can _read_ this without the sentry */
|
|
if (sem->count != -1)
|
|
break;
|
|
|
|
if (signal_pending(current)) {
|
|
ret = -EINTR;
|
|
break;
|
|
}
|
|
schedule();
|
|
}
|
|
|
|
DOWN_TAIL
|
|
|
|
if (!ret) {
|
|
UPDATE_COUNT
|
|
}
|
|
|
|
return ret;
|
|
}
|