fix(sbi): 正确设置 mstatus.spp

Author: YdrMaster

rustsbi-qemu/src/execute/transfer_trap.rs

@@ -7,21 +7,31 @@ pub(super) fn should_transfer_trap(ctx: &Context) -> bool {
 
 pub(super) fn do_transfer_trap(ctx: &mut Context, cause: scause::Trap) {
     unsafe {
-        // 填写陷入原因
+        // 向 S 转发陷入
+        mstatus::set_mpp(mstatus::MPP::Supervisor);
+        // 转发陷入源状态
+        let spp = match (ctx.mstatus >> 11) & 0b11 {
+            // U
+            0b00 => mstatus::SPP::User,
+            // S
+            0b01 => mstatus::SPP::Supervisor,
+            // H/M
+            mpp => unreachable!("invalid mpp: {mpp:#x} to delegate"),
+        };
+        mstatus::set_spp(spp);
+        // 转发陷入原因
         scause::set(cause);
-        // 填写陷入附加信息
+        // 转发陷入附加信息
         stval::write(mtval::read());
-        // 填写 S 态层需要返回到的地址
+        // 转发陷入地址
         sepc::write(ctx.mepc);
-        // 设置中断位
-        mstatus::set_mpp(mstatus::MPP::Supervisor);
-        mstatus::set_spp(mstatus::SPP::Supervisor);
+        // 设置 S 中断状态
         if mstatus::read().sie() {
-            mstatus::set_spie()
+            mstatus::set_spie();
+            mstatus::clear_sie();
         }
-        mstatus::clear_sie();
         core::arch::asm!("csrr {}, mstatus", out(reg) ctx.mstatus);
-        // 设置返回地址，返回到S层
+        // 设置返回地址，返回到 S
         // TODO Vectored stvec?
         ctx.mepc = stvec::read().address();
     }

test-kernel/src/main.rs

@@ -89,7 +89,7 @@ extern "C" fn primary_rust_main(hartid: usize, dtb_pa: usize) -> ! {
     unsafe { stvec::write(start_trap as usize, TrapMode::Direct) };
     test::trap_delegate(hartid);
 
-    test::start_stop_harts(hartid, smp);
+    test::hsm(hartid, smp);
 
     sbi::system_reset(sbi::RESET_TYPE_SHUTDOWN, sbi::RESET_REASON_NO_REASON);
     unreachable!()

test-kernel/src/test.rs

@@ -79,7 +79,7 @@ pub(crate) fn trap_delegate(hartid: usize) {
 }
 
 /// 所有副核：启动 -> 不可恢复休眠 -> 唤醒 -> 可恢复休眠 -> 唤醒 -> 关闭。
-pub(crate) fn start_stop_harts(hartid: usize, smp: usize) {
+pub(crate) fn hsm(hartid: usize, smp: usize) {
     const SUSPENDED: sbi::SbiRet = sbi::SbiRet {
         error: sbi::RET_SUCCESS,
         value: sbi::HART_STATE_SUSPENDED,
@@ -124,7 +124,7 @@ pub(crate) fn start_stop_harts(hartid: usize, smp: usize) {
 
     println!(
         "
-[test-kernel] Testing start harts"
+[test-kernel] Testing hsm: start, stop, suspend and resume"
     );
 
     // 启动副核