fix(virtq): add instrumentation to virtq host calls

Signed-off-by: Tomasz Andrzejak <andreiltd@gmail.com>

Author: andreiltd

fuzz/fuzz_targets/host_call.rs

@@ -61,6 +61,8 @@ fuzz_target!(
                 HyperlightError::GuestError(ErrorCode::HostFunctionError, msg) if msg.contains("The number of arguments to the function is wrong") => {}
                 HyperlightError::ParameterValueConversionFailure(_, _) => {},
                 HyperlightError::GuestError(ErrorCode::HostFunctionError, msg) if msg.contains("Failed To Convert Parameter Value") => {}
+                HyperlightError::GuestError(ErrorCode::HostFunctionError, msg) if msg.contains("The parameter value type is unexpected") => {}
+                HyperlightError::GuestError(ErrorCode::HostFunctionError, msg) if msg.contains("The return value type is unexpected") => {}
 
                 // any other error should be reported
                 _ => panic!("Guest Aborted with Unexpected Error: {:?}", e),

src/hyperlight_guest/src/virtq/context.rs

@@ -33,6 +33,7 @@ use hyperlight_common::virtq::msg::{MsgKind, VirtqMsgHeader};
 use hyperlight_common::virtq::{
     self, BufferPool, Layout, Notifier, QueueStats, RecyclePool, Token, VirtqProducer,
 };
+use tracing::instrument;
 
 use super::GuestMemOps;
 use crate::bail;
@@ -120,6 +121,7 @@ impl GuestContext {
     ///
     /// The reply guard is checked before submitting the readwrite chain
     /// to ensure G2H capacity is reserved for pending responses.
+    #[instrument(skip_all, level = "Info")]
     pub fn call_host_function<T: TryFrom<ReturnValue>>(
         &mut self,
         function_name: &str,

src/hyperlight_host/tests/integration_test.rs

@@ -716,10 +716,10 @@ fn log_message() {
     // follows:
     //  - logs from trace level tracing spans created as logs because of the tracing `log` feature
     //    - 4 from evolve call (generic_init + hyperlight_main)
-    //    - 8 from guest call
+    //    - 4 from guest call (call_host_function + read_n_bytes_from_user_memory)
     // and are multiplied because we make 6 calls to `log_test_messages`
     // NOTE: These numbers need to be updated if log messages or spans are added/removed
-    let num_fixed_trace_log = 12 * 6;
+    let num_fixed_trace_log = 8 * 6;
 
     // Calculate fixed info logs
     // - 4 logs per iteration from infrastructure at Info level (internal_dispatch_function)