[test] added test for exception handling

- added new guest fxn to simpleguest that triggers an invalid
opcode exception
- added test in guest_dispatch that calls the new guest fxn
to test exception handling

Signed-off-by: danbugs <[email protected]>

Author: danbugs

src/hyperlight_host/src/func/guest_dispatch.rs

@@ -480,4 +480,36 @@ mod tests {
             ),
         }
     }
+
+    #[test]
+    fn test_trigger_exception_on_guest() {
+        let usbox = UninitializedSandbox::new(
+            GuestBinary::FilePath(simple_guest_as_string().expect("Guest Binary Missing")),
+            None,
+            None,
+            None,
+        )
+        .unwrap();
+
+        let mut multi_use_sandbox: MultiUseSandbox = usbox.evolve(Noop::default()).unwrap();
+
+        let res = multi_use_sandbox.call_guest_function_by_name(
+            "TriggerException",
+            ReturnType::Void,
+            None,
+        );
+
+        assert!(res.is_err());
+
+        match res.unwrap_err() {
+            HyperlightError::GuestAborted(_, msg) => {
+                // msg should indicate we got an invalid opcode exception
+                assert!(msg.contains("EXCEPTION: 0x6"));
+            }
+            e => panic!(
+                "Expected HyperlightError::GuestExecutionError but got {:?}",
+                e
+            ),
+        }
+    }
 }

src/tests/rust_guests/simpleguest/src/main.rs

@@ -637,6 +637,13 @@ fn log_message(function_call: &FunctionCall) -> Result<Vec<u8>> {
     }
 }
 
+fn trigger_exception(_: &FunctionCall) -> Result<Vec<u8>> {
+    unsafe {
+        core::arch::asm!("ud2");
+    } // trigger an undefined instruction exception
+    Ok(get_flatbuffer_result_from_void())
+}
+
 static mut COUNTER: i32 = 0;
 
 fn add_to_static(function_call: &FunctionCall) -> Result<Vec<u8>> {
@@ -1094,6 +1101,14 @@ pub extern "C" fn hyperlight_main() {
         add as i64,
     );
     register_function(add_def);
+
+    let trigger_exception_def = GuestFunctionDefinition::new(
+        "TriggerException".to_string(),
+        Vec::new(),
+        ReturnType::Void,
+        trigger_exception as i64,
+    );
+    register_function(trigger_exception_def);
 }
 
 #[no_mangle]