feat: add invalid parma error for ipi ext

Signed-off-by: Woshiluo Luo <woshiluo.luo@outlook.com>

prototyper/src/board.rs

@@ -40,12 +40,15 @@ impl<const N: usize> Display for StringInline<N> {
     }
 }
 
+type CpuEnableList = [bool; trap_stack::NUM_HART_MAX];
+
 pub struct BoardInfo {
     pub memory_range: Option<Range<usize>>,
     pub console: Option<(BaseAddress, MachineConsoleType)>,
     pub reset: Option<BaseAddress>,
     pub ipi: Option<BaseAddress>,
     pub cpu_num: Option<usize>,
+    pub cpu_enabled: Option<CpuEnableList>,
     pub model: StringInline<128>,
 }
 
@@ -56,6 +59,7 @@ impl BoardInfo {
             console: None,
             reset: None,
             ipi: None,
+            cpu_enabled: None,
             cpu_num: None,
             model: StringInline(0, [0u8; 128]),
         }
@@ -213,6 +217,16 @@ impl Board {
 
         // TODO: Need a better extension initialization method
         extensions::init(&tree.cpus.cpu);
+
+        // Find which hart is enabled by fdt
+        let mut cpu_list: CpuEnableList = [false; trap_stack::NUM_HART_MAX];
+        for cpu_iter in tree.cpus.cpu.iter() {
+            use dt::Cpu;
+            let cpu = cpu_iter.deserialize::<Cpu>();
+            let hart_id = cpu.reg.iter().next().unwrap().0.start;
+            cpu_list.get_mut(hart_id).map(|x| *x = true);
+        }
+        self.info.cpu_enabled = Some(cpu_list);
     }
 
     fn sbi_init(&mut self) {

prototyper/src/sbi/ipi.rs

@@ -75,11 +75,31 @@ impl<T: IpiDevice> rustsbi::Ipi for SbiIpi<T> {
                 continue;
             }
 
+            // There are 3 situation to return invalid_param
+            // 1. We can not get hsm, which usually means this hart_id is bigger than MAX_HART_ID
+            // 2. BOARD hasn't init or this hart_id is not enabled by device tree
+            // 3. this hart is not in a state which allow ipi
+            // In the next loop, we'll assume that all of above situation will not happend and
+            // directly send ipi
             let Some(hsm) = remote_hsm(hart_id) else {
-                continue;
+                return SbiRet::invalid_param();
             };
 
+            if unsafe {
+                BOARD
+                    .info
+                    .cpu_enabled
+                    .is_none_or(|list| list.get(hart_id).is_none_or(|res| !(*res)))
+            } {
+                return SbiRet::invalid_param();
+            }
+
             if !hsm.allow_ipi() {
+                return SbiRet::invalid_param();
+            }
+        }
+        for hart_id in 0..=self.max_hart_id {
+            if !hart_mask.has_bit(hart_id) {
                 continue;
             }