diff options
Diffstat (limited to 'kernel/task')
-rw-r--r-- | kernel/task/include/task/lock.h | 4 | ||||
-rw-r--r-- | kernel/task/include/task/sched.h | 1 |
2 files changed, 5 insertions, 0 deletions
diff --git a/kernel/task/include/task/lock.h b/kernel/task/include/task/lock.h index bfb7994..eaa661c 100644 --- a/kernel/task/include/task/lock.h +++ b/kernel/task/include/task/lock.h @@ -9,11 +9,15 @@ typedef char spin_lock_t; static inline void spin_lock(struct spin_lock_t* lock) { while(*lock != SPIN_LOCK_UNLOCKED) {yield();}; + cli(); *lock = 1; + sei(); } static inline void spin_unlock(struct spin_lock_t* lock) { + cli(); *lock = SPIN_LOCK_UNLOCKED; + sei(); } #endif
\ No newline at end of file diff --git a/kernel/task/include/task/sched.h b/kernel/task/include/task/sched.h index 9029664..c14a85b 100644 --- a/kernel/task/include/task/sched.h +++ b/kernel/task/include/task/sched.h @@ -2,6 +2,7 @@ #define SCHED_H #include "collection/list.h" +#include "task/task.h" #include "mcu/task/task.h" /** |