1 use crate::arch::CurrentIrqArch; 2 3 pub mod softirq; 4 5 /// @brief 中断相关的操作 6 pub trait InterruptArch: Send + Sync { 7 /// @brief 使能中断 8 unsafe fn interrupt_enable(); 9 /// @brief 禁止中断 10 unsafe fn interrupt_disable(); 11 /// @brief 检查中断是否被禁止 12 fn is_irq_enabled() -> bool; 13 14 /// @brief 保存当前中断状态，并且禁止中断 15 unsafe fn save_and_disable_irq() -> IrqFlagsGuard; 16 unsafe fn restore_irq(flags: IrqFlags); 17 } 18 19 #[derive(Debug, Clone, Copy)] 20 pub struct IrqFlags { 21 flags: u64, 22 } 23 24 impl IrqFlags { 25 pub fn new(flags: u64) -> Self { 26 IrqFlags { flags } 27 } 28 29 pub fn flags(&self) -> u64 { 30 self.flags 31 } 32 } 33 34 /// @brief 当前中断状态的保护器，当该对象被drop时，会恢复之前的中断状态 35 /// 36 /// # Example 37 /// 38 /// ``` 39 /// use crate::arch::CurrentIrqArch; 40 /// 41 /// // disable irq and save irq state （这是唯一的获取IrqFlagsGuard的方法） 42 /// let guard = unsafe{CurrentIrqArch::save_and_disable_irq()}; 43 /// 44 /// // do something 45 /// 46 /// // 销毁guard时，会恢复之前的中断状态 47 /// drop(guard); 48 /// 49 /// ``` 50 #[derive(Debug)] 51 pub struct IrqFlagsGuard { 52 flags: IrqFlags, 53 } 54 55 impl IrqFlagsGuard { 56 /// @brief 创建IrqFlagsGuard对象 57 /// 58 /// # Safety 59 /// 60 /// 该函数不安全，因为它不会检查flags是否是一个有效的IrqFlags对象, 而当它被drop时，会恢复flags中的中断状态 61 /// 62 /// 该函数只应被`CurrentIrqArch::save_and_disable_irq`调用 63 pub unsafe fn new(flags: IrqFlags) -> Self { 64 IrqFlagsGuard { flags } 65 } 66 } 67 impl Drop for IrqFlagsGuard { 68 fn drop(&mut self) { 69 unsafe { 70 CurrentIrqArch::restore_irq(self.flags); 71 } 72 } 73 } 74