Use DUCT-TAPE to fix irq test

Signed-off-by: Michał Szaknis <[email protected]>

lib/armv9a/src/lib.rs

@@ -11,3 +11,16 @@ pub use regs::*;
 pub const fn bits_in_reg(mask: u64, val: u64) -> u64 {
     (val << (mask.trailing_zeros())) & mask
 }
+
+pub fn is_irq_pending() -> bool {
+    let val: u64;
+
+    unsafe {
+        core::arch::asm!(
+            "mrs {}, ISR_EL1",
+            out(reg) val
+        )
+    }
+
+    val != 0
+}

rmm/src/rsi/mod.rs

@@ -8,6 +8,7 @@ pub mod ripas;
 pub mod version;
 
 use alloc::vec::Vec;
+use armv9a::is_irq_pending;
 
 use crate::define_interface;
 use crate::event::RsiHandle;
@@ -214,6 +215,14 @@ pub fn set_event_handler(rsi: &mut RsiHandle) {
         {
             let (token_part, token_left) = get_token_part(&rd, rec, buffer_size)?;
 
+            if is_irq_pending() {
+                error!("IRQ is pending while fetching token");
+                set_reg(rec, 0, INCOMPLETE)?;
+                set_reg(rec, 1, 0)?;
+                ret[0] = rmi::SUCCESS_REC_ENTER;
+                return Ok(());
+            }
+
             unsafe {
                 let pa_ptr = attest_pa as *mut u8;
                 core::ptr::copy(token_part.as_ptr(), pa_ptr.add(pa_offset), token_part.len());