xref: /DragonOS/kernel/src/arch/riscv64/interrupt/mod.rs (revision 9fab312ea9921618629924ab15c28c2d255b21c6)
1 use riscv::register::{scause::Scause, sstatus::Sstatus};
2 use system_error::SystemError;
3 
4 use crate::{
5     driver::irqchip::riscv_intc::riscv_intc_init,
6     exception::{InterruptArch, IrqFlags, IrqFlagsGuard, IrqNumber},
7     libs::align::align_up,
8 };
9 
10 use super::cpu::STACK_ALIGN;
11 
12 pub(super) mod entry;
13 mod handle;
14 pub mod ipi;
15 
16 pub struct RiscV64InterruptArch;
17 
18 impl InterruptArch for RiscV64InterruptArch {
19     unsafe fn arch_irq_init() -> Result<(), SystemError> {
20         riscv_intc_init()?;
21 
22         Ok(())
23     }
24     unsafe fn interrupt_enable() {
25         riscv::interrupt::enable();
26     }
27 
28     unsafe fn interrupt_disable() {
29         riscv::interrupt::disable();
30     }
31 
32     fn is_irq_enabled() -> bool {
33         riscv::register::sstatus::read().sie()
34     }
35 
36     unsafe fn save_and_disable_irq() -> IrqFlagsGuard {
37         let sie = riscv::register::sstatus::read().sie();
38         IrqFlagsGuard::new(IrqFlags::new(sie.into()))
39     }
40 
41     unsafe fn restore_irq(flags: IrqFlags) {
42         let sie: bool = flags.flags() != 0;
43         if sie {
44             riscv::register::sstatus::set_sie();
45         } else {
46             riscv::register::sstatus::clear_sie();
47         }
48     }
49 
50     fn probe_total_irq_num() -> u32 {
51         // todo: 获取中断总数
52         256
53     }
54 
55     fn ack_bad_irq(irq: IrqNumber) {
56         todo!("ack_bad_irq: {}", irq.data());
57     }
58 }
59 
60 /// 中断栈帧结构体
61 #[repr(C)]
62 #[derive(Debug, Copy, Clone)]
63 pub struct TrapFrame {
64     pub epc: usize,
65     pub ra: usize,
66     pub sp: usize,
67     pub gp: usize,
68     pub tp: usize,
69     pub t0: usize,
70     pub t1: usize,
71     pub t2: usize,
72     pub s0: usize,
73     pub s1: usize,
74     pub a0: usize,
75     pub a1: usize,
76     pub a2: usize,
77     pub a3: usize,
78     pub a4: usize,
79     pub a5: usize,
80     pub a6: usize,
81     pub a7: usize,
82     pub s2: usize,
83     pub s3: usize,
84     pub s4: usize,
85     pub s5: usize,
86     pub s6: usize,
87     pub s7: usize,
88     pub s8: usize,
89     pub s9: usize,
90     pub s10: usize,
91     pub s11: usize,
92     pub t3: usize,
93     pub t4: usize,
94     pub t5: usize,
95     pub t6: usize,
96     // 以下是中断发生时自动保存的寄存器
97     pub status: Sstatus,
98     pub badaddr: usize,
99     pub cause: Scause,
100     /// a0 value before the syscall
101     pub origin_a0: usize,
102 }
103 
104 impl TrapFrame {
105     /// 中断栈帧结构体的大小
106     pub const SIZE: usize = core::mem::size_of::<TrapFrame>();
107 
108     /// 中断栈帧在栈上的大小
109     pub const SIZE_ON_STACK: usize = align_up(Self::SIZE, STACK_ALIGN);
110     /// 判断当前中断是否来自用户模式
111     pub fn is_from_user(&self) -> bool {
112         self.status.spp() == riscv::register::sstatus::SPP::User
113     }
114 
115     pub fn new() -> Self {
116         Self {
117             epc: 0,
118             ra: 0,
119             sp: 0,
120             gp: 0,
121             tp: 0,
122             t0: 0,
123             t1: 0,
124             t2: 0,
125             s0: 0,
126             s1: 0,
127             a0: 0,
128             a1: 0,
129             a2: 0,
130             a3: 0,
131             a4: 0,
132             a5: 0,
133             a6: 0,
134             a7: 0,
135             s2: 0,
136             s3: 0,
137             s4: 0,
138             s5: 0,
139             s6: 0,
140             s7: 0,
141             s8: 0,
142             s9: 0,
143             s10: 0,
144             s11: 0,
145             t3: 0,
146             t4: 0,
147             t5: 0,
148             t6: 0,
149             status: unsafe { core::mem::zeroed() },
150             badaddr: 0,
151             cause: unsafe { core::mem::zeroed() },
152             origin_a0: 0,
153         }
154     }
155 
156     pub fn set_return_value(&mut self, value: usize) {
157         self.a0 = value;
158     }
159 }
160