179 lines
6.9 KiB
C
179 lines
6.9 KiB
C
#ifndef _LINUX_WAIT_H
|
|
#define _LINUX_WAIT_H
|
|
/*
|
|
* Linux wait queue related types and methods
|
|
*/
|
|
#include <linux/list.h>
|
|
#include <linux/stddef.h>
|
|
#include <linux/spinlock.h>
|
|
#include <asm/current.h>
|
|
|
|
|
|
|
|
#include <syscall.h>
|
|
|
|
typedef struct __wait_queue wait_queue_t;
|
|
typedef int (*wait_queue_func_t)(wait_queue_t *wait, unsigned mode, int flags, void *key);
|
|
int default_wake_function(wait_queue_t *wait, unsigned mode, int flags, void *key);
|
|
|
|
typedef struct __wait_queue_head wait_queue_head_t;
|
|
|
|
struct __wait_queue
|
|
{
|
|
wait_queue_func_t func;
|
|
struct list_head task_list;
|
|
evhandle_t evnt;
|
|
};
|
|
|
|
struct __wait_queue_head
|
|
{
|
|
spinlock_t lock;
|
|
struct list_head task_list;
|
|
};
|
|
static inline int waitqueue_active(wait_queue_head_t *q)
|
|
{
|
|
return !list_empty(&q->task_list);
|
|
}
|
|
|
|
extern void add_wait_queue(wait_queue_head_t *q, wait_queue_t *wait);
|
|
extern void add_wait_queue_exclusive(wait_queue_head_t *q, wait_queue_t *wait);
|
|
extern void remove_wait_queue(wait_queue_head_t *q, wait_queue_t *wait);
|
|
|
|
static inline void __add_wait_queue(wait_queue_head_t *head, wait_queue_t *new)
|
|
{
|
|
list_add(&new->task_list, &head->task_list);
|
|
}
|
|
|
|
/*
|
|
#define __wait_event(wq, condition) \
|
|
do { \
|
|
DEFINE_WAIT(__wait); \
|
|
\
|
|
for (;;) { \
|
|
prepare_to_wait(&wq, &__wait, TASK_UNINTERRUPTIBLE); \
|
|
if (condition) \
|
|
break; \
|
|
schedule(); \
|
|
} \
|
|
finish_wait(&wq, &__wait); \
|
|
} while (0)
|
|
|
|
*/
|
|
|
|
#define wait_event_timeout(wq, condition, timeout) \
|
|
({ \
|
|
long __ret = timeout; \
|
|
do{ \
|
|
wait_queue_t __wait = { \
|
|
.task_list = LIST_HEAD_INIT(__wait.task_list), \
|
|
.evnt = CreateEvent(NULL, MANUAL_DESTROY), \
|
|
}; \
|
|
unsigned long flags; \
|
|
\
|
|
spin_lock_irqsave(&wq.lock, flags); \
|
|
if (list_empty(&__wait.task_list)) \
|
|
__add_wait_queue(&wq, &__wait); \
|
|
spin_unlock_irqrestore(&wq.lock, flags); \
|
|
\
|
|
for(;;){ \
|
|
if (condition) \
|
|
break; \
|
|
WaitEventTimeout(__wait.evnt, timeout); \
|
|
}; \
|
|
if (!list_empty(&__wait.task_list)) { \
|
|
spin_lock_irqsave(&wq.lock, flags); \
|
|
list_del_init(&__wait.task_list); \
|
|
spin_unlock_irqrestore(&wq.lock, flags); \
|
|
}; \
|
|
DestroyEvent(__wait.evnt); \
|
|
} while (0); \
|
|
__ret; \
|
|
})
|
|
|
|
#define wait_event_interruptible_timeout(wq, condition, timeout) \
|
|
wait_event_timeout(wq, condition, timeout)
|
|
|
|
|
|
#define wait_event(wq, condition) \
|
|
do{ \
|
|
wait_queue_t __wait = { \
|
|
.task_list = LIST_HEAD_INIT(__wait.task_list), \
|
|
.evnt = CreateEvent(NULL, MANUAL_DESTROY), \
|
|
}; \
|
|
unsigned long flags; \
|
|
\
|
|
spin_lock_irqsave(&wq.lock, flags); \
|
|
if (list_empty(&__wait.task_list)) \
|
|
__add_wait_queue(&wq, &__wait); \
|
|
spin_unlock_irqrestore(&wq.lock, flags); \
|
|
\
|
|
for(;;){ \
|
|
if (condition) \
|
|
break; \
|
|
WaitEvent(__wait.evnt); \
|
|
}; \
|
|
if (!list_empty_careful(&__wait.task_list)) { \
|
|
spin_lock_irqsave(&wq.lock, flags); \
|
|
list_del_init(&__wait.task_list); \
|
|
spin_unlock_irqrestore(&wq.lock, flags); \
|
|
}; \
|
|
DestroyEvent(__wait.evnt); \
|
|
} while (0)
|
|
|
|
#define wait_event_interruptible(wq, condition) \
|
|
({ \
|
|
int __ret = 0; \
|
|
if (!(condition)) \
|
|
wait_event(wq, condition); \
|
|
__ret; \
|
|
})
|
|
|
|
|
|
static inline
|
|
void wake_up_all(wait_queue_head_t *q)
|
|
{
|
|
wait_queue_t *curr;
|
|
unsigned long flags;
|
|
|
|
spin_lock_irqsave(&q->lock, flags);
|
|
list_for_each_entry(curr, &q->task_list, task_list)
|
|
{
|
|
// printf("raise event \n");
|
|
|
|
kevent_t event;
|
|
event.code = -1;
|
|
RaiseEvent(curr->evnt, 0, &event);
|
|
}
|
|
spin_unlock_irqrestore(&q->lock, flags);
|
|
}
|
|
|
|
|
|
static inline void
|
|
init_waitqueue_head(wait_queue_head_t *q)
|
|
{
|
|
spin_lock_init(&q->lock);
|
|
INIT_LIST_HEAD(&q->task_list);
|
|
};
|
|
|
|
|
|
//struct completion {
|
|
// unsigned int done;
|
|
// wait_queue_head_t wait;
|
|
//};
|
|
|
|
int autoremove_wake_function(wait_queue_t *wait, unsigned mode, int sync, void *key);
|
|
|
|
|
|
#define DEFINE_WAIT_FUNC(name, function) \
|
|
wait_queue_t name = { \
|
|
.func = function, \
|
|
.task_list = LIST_HEAD_INIT((name).task_list), \
|
|
.evnt = CreateEvent(NULL, MANUAL_DESTROY), \
|
|
}
|
|
|
|
#define DEFINE_WAIT(name) DEFINE_WAIT_FUNC(name, autoremove_wake_function)
|
|
|
|
|
|
#endif
|
|
|