X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/714e732504021937171665407133851ad9f3e68b..270869139db617e29a35bb9ded41087bb702f9ac:/lunaix-os/kernel/ds/semaphore.c

diff --git a/lunaix-os/kernel/ds/semaphore.c b/lunaix-os/kernel/ds/semaphore.c
index 84c0100..b4d0674 100644
--- a/lunaix-os/kernel/ds/semaphore.c
+++ b/lunaix-os/kernel/ds/semaphore.c
@@ -1,5 +1,5 @@
 #include <lunaix/ds/semaphore.h>
-#include <lunaix/sched.h>
+#include <lunaix/kpreempt.h>
 
 void
 sem_init(struct sem_t* sem, unsigned int initial)
@@ -12,6 +12,7 @@ sem_wait(struct sem_t* sem)
 {
     while (!atomic_load(&sem->counter)) {
         // FIXME: better thing like wait queue
+        preempt_current();
     }
     atomic_fetch_sub(&sem->counter, 1);
 }