xref: /DragonOS/kernel/src/libs/spinlock.rs (revision 66f67c6a95b8aad85cfd2146a86e5e3e6a3568e7)
1 #![allow(dead_code)]
2 use core::ptr::read_volatile;
3 
4 use crate::arch::x86_64::asm::irqflags::{local_irq_restore, local_irq_save};
5 use crate::include::bindings::bindings::{spin_lock, spin_unlock, spinlock_t};
6 
7 /// @brief 保存中断状态到flags中,关闭中断,并对自旋锁加锁
8 pub fn spin_lock_irqsave(lock: *mut spinlock_t, flags: &mut u64) {
9     local_irq_save(flags);
10     unsafe {
11         spin_lock(lock);
12     }
13 }
14 
15 /// @brief 恢复rflags以及中断状态并解锁自旋锁
16 #[no_mangle]
17 pub fn spin_unlock_irqrestore(lock: *mut spinlock_t, flags: &u64) {
18     unsafe {
19         spin_unlock(lock);
20     }
21     // kdebug!("123");
22     local_irq_restore(flags);
23 }
24 
25 /// 判断一个自旋锁是否已经被加锁
26 pub fn spin_is_locked(lock: &spinlock_t) -> bool {
27     let val = unsafe { read_volatile(&lock.lock as *const i8) };
28 
29     return if val == 0 { true } else { false };
30 }
31