wash/mango
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sched: implement waitqueues

Browse Source
This commit is contained in:
max 🍓
2023-05-10 20:29:57 +01:00
parent b3957b311a
commit 63f27adb1b
3 changed files with 100 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
sched/runqueue.c
View File

@@ -50,6 +50,7 @@ void rq_enqueue(struct runqueue *rq, struct thread *thr)
rq->rq_nthreads++;
rq->rq_readybits |= PRIO_MASK(thread_priority(thr));
thr->tr_rq = rq;
}
struct runqueue *cpu_rq(unsigned int cpu)

73
sched/wait.c Normal file
View File

@@ -0,0 +1,73 @@
#include <socks/sched.h>
#include <socks/cpu.h>
void wait_item_init(struct wait_item *item, struct thread *thr)
{
item->w_thread = thr;
item->w_entry = QUEUE_ENTRY_INIT;
}
void thread_wait_begin(struct wait_item *waiter, struct waitqueue *q)
{
unsigned long flags;
spin_lock_irqsave(&q->wq_lock, &flags);
queue_push_back(&q->wq_waiters, &waiter->w_entry);
waiter->w_thread->tr_state = THREAD_SLEEPING;
spin_unlock_irqrestore(&q->wq_lock, flags);
}
void thread_wait_end(struct wait_item *waiter, struct waitqueue *q)
{
waiter->w_thread->tr_state = THREAD_READY;
unsigned long flags;
spin_lock_irqsave(&q->wq_lock, &flags);
queue_delete(&q->wq_waiters, &waiter->w_entry);
spin_unlock_irqrestore(&q->wq_lock, flags);
}
void wakeup_queue(struct waitqueue *q)
{
unsigned long flags;
spin_lock_irqsave(&q->wq_lock, &flags);
struct queue_entry *ent = queue_pop_front(&q->wq_waiters);
while (ent) {
struct wait_item *waiter = QUEUE_CONTAINER(struct wait_item, w_entry, ent);
struct thread *thr = waiter->w_thread;
struct runqueue *rq = thr->tr_rq;
if (!rq) {
rq = cpu_rq(this_cpu());
}
thr->tr_state = THREAD_READY;
rq_enqueue(rq, thr);
ent = queue_pop_front(&q->wq_waiters);
}
spin_unlock_irqrestore(&q->wq_lock, flags);
}
void wakeup_one(struct waitqueue *q)
{
unsigned long flags;
spin_lock_irqsave(&q->wq_lock, &flags);
struct queue_entry *ent = queue_pop_front(&q->wq_waiters);
if(ent) {
struct wait_item *waiter = QUEUE_CONTAINER(struct wait_item, w_entry, ent);
struct thread *thr = waiter->w_thread;
struct runqueue *rq = thr->tr_rq;
if (!rq) {
rq = cpu_rq(this_cpu());
}
thr->tr_state = THREAD_READY;
rq_enqueue(rq, thr);
}
spin_unlock_irqrestore(&q->wq_lock, flags);
}