expose the interrup handler

Signed-off-by: Jorge Prendes <jorge.prendes@gmail.com>

Author: jprendes

src/hyperlight_wasm/src/sandbox/loaded_wasm_sandbox.rs

@@ -14,11 +14,15 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
+use std::sync::Arc;
+
 use hyperlight_host::func::{ParameterTuple, SupportedReturnType};
+// re-export the InterruptHandle trait as it's part of the public API
+pub use hyperlight_host::hypervisor::InterruptHandle;
 use hyperlight_host::sandbox::Callable;
 use hyperlight_host::sandbox_state::sandbox::{DevolvableSandbox, Sandbox};
 use hyperlight_host::sandbox_state::transition::Noop;
-use hyperlight_host::{MultiUseSandbox, Result, log_then_return};
+use hyperlight_host::{MultiUseSandbox, Result, log_then_return, new_error};
 
 use super::metrics::METRIC_TOTAL_LOADED_WASM_SANDBOXES;
 use super::wasm_sandbox::WasmSandbox;
@@ -72,6 +76,18 @@ impl LoadedWasmSandbox {
         metrics::counter!(METRIC_TOTAL_LOADED_WASM_SANDBOXES).increment(1);
         Ok(LoadedWasmSandbox { inner: Some(inner) })
     }
+
+    /// Get a handle to the interrupt handler for this sandbox,
+    /// capable of interrupting guest execution.
+    pub fn interrupt_handle(&self) -> Result<Arc<dyn InterruptHandle>> {
+        if let Some(inner) = &self.inner {
+            Ok(inner.interrupt_handle())
+        } else {
+            Err(new_error!(
+                "WasmSandbox is None, cannot get interrupt handle"
+            ))
+        }
+    }
 }
 
 impl Callable for LoadedWasmSandbox {

src/hyperlight_wasm/src/sandbox/wasm_sandbox.rs

@@ -148,6 +148,13 @@ mod tests {
 
         let mut loaded = loaded.load_module(run_wasm)?;
 
+        let interrupt = loaded.interrupt_handle()?;
+
+        std::thread::spawn(move || {
+            std::thread::sleep(std::time::Duration::from_millis(1000));
+            interrupt.kill();
+        });
+
         let result = loaded.call_guest_function::<i32>("KeepCPUBusy", 10000i32);
 
         match result {