Add tests for the given scenario

Signed-off-by: Ludvig Liljenberg <[email protected]>

Author: ludfjig

src/hyperlight_host/tests/integration_test.rs

@@ -327,6 +327,43 @@ fn interrupt_custom_signal_no_and_retry_delay() {
     thread.join().expect("Thread should finish");
 }
 
+#[test]
+fn interrupt_spamming_host_call() {
+    let mut uninit = UninitializedSandbox::new(
+        GuestBinary::FilePath(callback_guest_as_string().unwrap()),
+        None,
+    )
+    .unwrap();
+
+    uninit
+        .register("HostFunc1", || {
+            // do nothing
+        })
+        .unwrap();
+    let mut sbox1: MultiUseSandbox = uninit.evolve(Noop::default()).unwrap();
+
+    let interrupt_handle = sbox1.interrupt_handle();
+
+    let barrier = Arc::new(Barrier::new(2));
+    let barrier2 = barrier.clone();
+
+    let thread = thread::spawn(move || {
+        barrier2.wait();
+        thread::sleep(Duration::from_secs(1));
+        interrupt_handle.kill();
+    });
+
+    barrier.wait();
+    // This guest call calls "HostFunc1" in a loop
+    let res = sbox1
+        .call_guest_function_by_name::<i32>("HostCallLoop", "HostFunc1".to_string())
+        .unwrap_err();
+
+    assert!(matches!(res, HyperlightError::ExecutionCanceledByHost()));
+
+    thread.join().expect("Thread should finish");
+}
+
 #[test]
 fn print_four_args_c_guest() {
     let path = c_simple_guest_as_string().unwrap();

src/tests/rust_guests/callbackguest/src/main.rs

@@ -157,6 +157,19 @@ fn call_host_spin(_: &FunctionCall) -> Result<Vec<u8>> {
     Ok(get_flatbuffer_result(()))
 }
 
+fn host_call_loop(function_call: &FunctionCall) -> Result<Vec<u8>> {
+    if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
+        loop {
+            call_host_function::<()>(message, None, ReturnType::Void).unwrap();
+        }
+    } else {
+        Err(HyperlightGuestError::new(
+            ErrorCode::GuestFunctionParameterTypeMismatch,
+            "Invalid parameters passed to host_call_loop".to_string(),
+        ))
+    }
+}
+
 #[no_mangle]
 pub extern "C" fn hyperlight_main() {
     let print_output_def = GuestFunctionDefinition::new(
@@ -234,6 +247,14 @@ pub extern "C" fn hyperlight_main() {
         call_host_spin as usize,
     );
     register_function(call_host_spin_def);
+
+    let host_call_loop_def = GuestFunctionDefinition::new(
+        "HostCallLoop".to_string(),
+        Vec::from(&[ParameterType::String]),
+        ReturnType::Void,
+        host_call_loop as usize,
+    );
+    register_function(host_call_loop_def);
 }
 
 #[no_mangle]