[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

@@ -160,7 +160,7 @@ mod tests {
                 None,
                 None,
             )
-            .unwrap();
+                .unwrap();
 
             make_get_pid_syscall_func.register(&mut usbox, "MakeGetpidSyscall")?;
 
@@ -196,7 +196,7 @@ mod tests {
                 None,
                 None,
             )
-            .unwrap();
+                .unwrap();
 
             make_get_pid_syscall_func.register_with_extra_allowed_syscalls(
                 &mut usbox,
@@ -228,7 +228,7 @@ mod tests {
                 None,
                 None,
             )
-            .unwrap()
+                .unwrap()
         };
 
         // test_function0
@@ -345,7 +345,7 @@ mod tests {
             // just use the built-in host print function
             None,
         )
-        .unwrap();
+            .unwrap();
         test_call_guest_function_by_name(u_sbox);
     }
 
@@ -365,7 +365,7 @@ mod tests {
             Some(crate::SandboxRunOptions::RunInProcess(true)),
             None,
         )
-        .unwrap();
+            .unwrap();
         test_call_guest_function_by_name(u_sbox);
     }
 
@@ -378,7 +378,7 @@ mod tests {
             Some(crate::SandboxRunOptions::RunInProcess(false)),
             None,
         )
-        .unwrap();
+            .unwrap();
         test_call_guest_function_by_name(u_sbox);
     }
 
@@ -444,7 +444,7 @@ mod tests {
             None,
             None,
         )
-        .unwrap();
+            .unwrap();
 
         // Make this host call run for 5 seconds
 
@@ -480,4 +480,33 @@ 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/callbackguest/Cargo.lock

@@ -637,6 +637,11 @@ 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 +1099,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]