waitqueue.c 2.5 KB
Newer Older
B
bernard 已提交
1 2 3 4 5 6 7 8 9 10 11
#include <stdint.h>

#include <rthw.h>
#include <rtdevice.h>
#include <rtservice.h>

void rt_wqueue_add(rt_wqueue_t *queue, struct rt_wqueue_node *node)
{
    rt_base_t level;

    level = rt_hw_interrupt_disable();
12
    rt_list_insert_before(&(queue->waiting_list), &(node->list));
B
bernard 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
    rt_hw_interrupt_enable(level);
}

void rt_wqueue_remove(struct rt_wqueue_node *node)
{
    rt_base_t level;

    level = rt_hw_interrupt_disable();
    rt_list_remove(&(node->list));
    rt_hw_interrupt_enable(level);
}

int __wqueue_default_wake(struct rt_wqueue_node *wait, void *key)
{
    return 0;
}

void rt_wqueue_wakeup(rt_wqueue_t *queue, void *key)
{
    rt_base_t level;
    register int need_schedule = 0;

35
	rt_list_t *queue_list;
B
bernard 已提交
36 37 38
    struct rt_list_node *node;
    struct rt_wqueue_node *entry;

39
	queue_list = &(queue->waiting_list);
B
bernard 已提交
40 41

    level = rt_hw_interrupt_disable();
42 43 44 45
	/* set wakeup flag in the queue */
	queue->flag = RT_WQ_FLAG_WAKEUP;

    if (!(rt_list_isempty(queue_list)))
B
bernard 已提交
46
    {
47 48 49 50 51 52 53 54 55 56 57 58
	    for (node = queue_list->next; node != queue_list; node = node->next)
	    {
	        entry = rt_list_entry(node, struct rt_wqueue_node, list);
	        if (entry->wakeup(entry, key) == 0)
	        {
	            rt_thread_resume(entry->polling_thread);
	            need_schedule = 1;

	            rt_wqueue_remove(entry);
	            break;
	        }
	    }
B
bernard 已提交
59 60 61 62 63 64 65 66 67 68
    }
    rt_hw_interrupt_enable(level);

    if (need_schedule)
        rt_schedule();
}

int rt_wqueue_wait(rt_wqueue_t *queue, int condition, int msec)
{
    int tick;
69
    rt_thread_t tid = rt_thread_self();
B
bernard 已提交
70 71
    rt_timer_t  tmr = &(tid->thread_timer);
    struct rt_wqueue_node __wait;
72 73 74 75
    rt_base_t level;

    /* current context checking */
    RT_DEBUG_NOT_IN_INTERRUPT;
B
bernard 已提交
76 77 78 79 80 81 82 83 84 85 86

    tick = rt_tick_from_millisecond(msec);

    if ((condition) || (tick == 0))
        return 0;

    __wait.polling_thread = rt_thread_self();
    __wait.key = 0;
    __wait.wakeup = __wqueue_default_wake;
    rt_list_init(&__wait.list);

87
    level = rt_hw_interrupt_disable();
88 89 90 91 92 93
	if (queue->flag == RT_WQ_FLAG_WAKEUP)
	{
		/* already wakeup */
		goto __exit_wakeup;
	}

B
bernard 已提交
94 95 96 97 98 99 100 101 102 103 104 105
    rt_wqueue_add(queue, &__wait);
    rt_thread_suspend(tid);

    /* start timer */
    if (tick != RT_WAITING_FOREVER)
    {
        rt_timer_control(tmr,
                         RT_TIMER_CTRL_SET_TIME,
                         &tick);

        rt_timer_start(tmr);
    }
106
    rt_hw_interrupt_enable(level);
B
bernard 已提交
107 108 109

    rt_schedule();

110 111 112 113 114 115
    level = rt_hw_interrupt_disable();

__exit_wakeup:
	queue->flag = 0;
    rt_hw_interrupt_enable(level);

B
bernard 已提交
116 117 118 119
    rt_wqueue_remove(&__wait);

    return 0;
}