Add debugging logs for testing purpose

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

Author: ludfjig

src/hyperlight_host/src/func/call_ctx.rs

@@ -239,6 +239,10 @@ mod tests {
 
     #[test]
     fn ensure_multiusesandbox_multi_calls_dont_reset_state() {
+        env_logger::builder()
+            .filter_module("hyperlight", log::LevelFilter::Trace)
+            .filter_level(log::LevelFilter::Info)
+            .init();
         let sandbox = TestSandbox::new();
         let result = sandbox.call_add_to_static_multiple_times(5);
         assert!(result.is_ok());

src/hyperlight_host/src/hypervisor/whp.rs

@@ -39,6 +39,7 @@ use crate::mem::memory_region::{MemoryRegion, MemoryRegionFlags};
 use crate::{new_error, Result};
 
 pub(crate) fn is_hypervisor_present() -> bool {
+    println!("whp in_is_hypervisor_present");
     let mut capability: WHV_CAPABILITY = Default::default();
     let written_size: Option<*mut u32> = None;
 
@@ -88,6 +89,8 @@ unsafe impl Sync for WhpVm {}
 
 impl WhpVm {
     pub(crate) fn new(mmap_file_handle: HandleWrapper) -> Result<Self> {
+        println!("whp in_is_hypervisor_present");
+
         const NUM_CPU: u32 = 1;
         let partition = unsafe {
             let partition = WHvCreatePartition()?;
@@ -111,6 +114,8 @@ impl WhpVm {
 
     /// Helper for setting arbitrary registers.
     fn set_registers(&self, registers: &[(WHV_REGISTER_NAME, WHV_REGISTER_VALUE)]) -> Result<()> {
+        println!("whp set_registers");
+
         let register_count = registers.len();
         let mut register_names: Vec<WHV_REGISTER_NAME> = vec![];
         let mut register_values: Vec<WHV_REGISTER_VALUE> = vec![];
@@ -136,6 +141,8 @@ impl WhpVm {
 
 impl Vm for WhpVm {
     fn get_regs(&self) -> Result<CommonRegisters> {
+        println!("whp get_regs");
+
         let mut whv_regs_values: [WHV_REGISTER_VALUE; WHP_REGS_NAMES_LEN] =
             unsafe { std::mem::zeroed() };
 
@@ -164,12 +171,14 @@ impl Vm for WhpVm {
     }
 
     fn set_regs(&self, regs: &CommonRegisters) -> Result<()> {
+        println!("whp set_regs");
         let whp_regs: [(WHV_REGISTER_NAME, WHV_REGISTER_VALUE); WHP_REGS_NAMES_LEN] = regs.into();
         self.set_registers(&whp_regs)?;
         Ok(())
     }
 
     fn get_sregs(&self) -> Result<CommonSpecialRegisters> {
+        println!("whp get_sregs");
         let mut whp_sregs_values: [WHV_REGISTER_VALUE; WHP_SREGS_NAMES_LEN] =
             unsafe { std::mem::zeroed() };
 
@@ -198,12 +207,14 @@ impl Vm for WhpVm {
     }
 
     fn set_sregs(&self, sregs: &CommonSpecialRegisters) -> Result<()> {
+        println!("whp set_sregs");
         let whp_regs: [(WHV_REGISTER_NAME, WHV_REGISTER_VALUE); WHP_SREGS_NAMES_LEN] = sregs.into();
         self.set_registers(&whp_regs)?;
         Ok(())
     }
 
     fn get_fpu(&self) -> Result<CommonFpu> {
+        println!("whp get_fpu");
         let mut whp_fpu_values: [WHV_REGISTER_VALUE; WHP_FPU_NAMES_LEN] =
             unsafe { std::mem::zeroed() };
 
@@ -227,12 +238,14 @@ impl Vm for WhpVm {
     }
 
     fn set_fpu(&self, fpu: &CommonFpu) -> Result<()> {
+        println!("whp set_fpu");
         let whp_fpu: [(WHV_REGISTER_NAME, WHV_REGISTER_VALUE); WHP_FPU_NAMES_LEN] = fpu.into();
         self.set_registers(&whp_fpu)?;
         Ok(())
     }
 
     unsafe fn map_memory(&mut self, regions: &[MemoryRegion]) -> Result<()> {
+        println!("whp map_memory");
         if regions.is_empty() {
             return Err(new_error!("No memory regions to map"));
         }
@@ -302,6 +315,8 @@ impl Vm for WhpVm {
 
     #[expect(non_upper_case_globals, reason = "Windows API constant are lower case")]
     fn run_vcpu(&mut self) -> Result<HyperlightExit> {
+        println!("whp run_vpu");
+
         let mut exit_context: WHV_RUN_VP_EXIT_CONTEXT = Default::default();
 
         unsafe {
@@ -359,6 +374,7 @@ impl Vm for WhpVm {
     }
 
     fn get_partition_handle(&self) -> WHV_PARTITION_HANDLE {
+        println!("whp get_partition_handle");
         self.partition
     }
 }
@@ -372,6 +388,7 @@ impl Drop for WhpVm {
 }
 
 unsafe fn try_load_whv_map_gpa_range2() -> Result<WHvMapGpaRange2Func> {
+    println!("whp try_load_whv_map_gpa_range2");
     let library = unsafe {
         LoadLibraryExA(
             s!("winhvplatform.dll"),