Update riscv-rt/src/lib.rs

Co-authored-by: rmsyn <[email protected]>
rust-embedded · Jul 4, 2024 · bb996a4 · bb996a4
1 parent e6df749
commit bb996a4
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/riscv-rt/src/lib.rs b/riscv-rt/src/lib.rs
@@ -527,17 +527,13 @@ pub unsafe extern "C" fn start_trap_rust(trap_frame: *const TrapFrame) {
 
     match xcause::read().cause() {
         xcause::Trap::Interrupt(code) => _dispatch_core_interrupt(code),
-        xcause::Trap::Exception(code) => {
-            let trap_frame = &*trap_frame;
-            if code < __EXCEPTIONS.len() {
-                let h = &__EXCEPTIONS[code];
-                if let Some(handler) = h {
-                    handler(trap_frame);
-                } else {
-                    ExceptionHandler(trap_frame);
-                }
-            } else {
-                ExceptionHandler(trap_frame);
+        xcause::Trap::Exception(code) if code < __EXCEPTIONS.len() => {
+            match __EXCEPTIONS[code].as_ref() {
+                Some(handler) => handler(&*trap_frame),
+                None => ExceptionHandler(&*trap_frame),
+            }
+        }
+        xcause::Trap::Exception(_) => ExceptionHandler(&*trap_frame),
             }
         }
     }