95 lines
1.3 KiB
C
95 lines
1.3 KiB
C
#ifndef SYS_SPINLOCK_H
|
|
#define SYS_SPINLOCK_H
|
|
|
|
#include "generic/typedef.h"
|
|
|
|
struct __spinlock {
|
|
volatile u8 rwlock;
|
|
};
|
|
|
|
typedef struct __spinlock spinlock_t;
|
|
|
|
#if CPU_CORE_NUM > 1
|
|
|
|
|
|
#define preempt_disable() \
|
|
__local_irq_disable()
|
|
|
|
#define preempt_enable() \
|
|
__local_irq_enable()
|
|
|
|
#else
|
|
|
|
#define preempt_disable() \
|
|
local_irq_disable()
|
|
|
|
#define preempt_enable() \
|
|
local_irq_enable()
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
#if CPU_CORE_NUM > 1
|
|
|
|
#define spin_acquire(lock) \
|
|
do { \
|
|
arch_spin_lock(lock); \
|
|
}while(0)
|
|
|
|
#define spin_release(lock) \
|
|
do { \
|
|
arch_spin_unlock(lock); \
|
|
}while(0)
|
|
|
|
#else
|
|
|
|
#define spin_acquire(lock) \
|
|
do { \
|
|
while ((lock)->rwlock); \
|
|
(lock)->rwlock = 1; \
|
|
}while(0)
|
|
|
|
|
|
#define spin_release(lock) \
|
|
do { \
|
|
(lock)->rwlock = 0; \
|
|
}while(0)
|
|
|
|
#endif
|
|
|
|
|
|
#define DEFINE_SPINLOCK(x) \
|
|
spinlock_t x = { .rwlock = 0 }
|
|
|
|
|
|
__attribute__((always_inline))
|
|
static inline void spin_lock_init(spinlock_t *lock)
|
|
{
|
|
lock->rwlock = 0;
|
|
}
|
|
|
|
__attribute__((always_inline))
|
|
static inline void spin_lock(spinlock_t *lock)
|
|
{
|
|
preempt_disable();
|
|
/*ASSERT(spin_lock_cnt[current_cpu_id()] == 0);
|
|
spin_lock_cnt[current_cpu_id()] = 1;*/
|
|
spin_acquire(lock);
|
|
}
|
|
|
|
__attribute__((always_inline))
|
|
static inline void spin_unlock(spinlock_t *lock)
|
|
{
|
|
/*spin_lock_cnt[current_cpu_id()] = 0;*/
|
|
spin_release(lock);
|
|
preempt_enable();
|
|
}
|
|
|
|
|
|
#endif
|
|
|
|
|