[trace-guest] add spans in simpleguest sample

Signed-off-by: Doru Blânzeanu <[email protected]>

Author: dblnz

src/tests/rust_guests/simpleguest/Cargo.lock

@@ -9,6 +9,7 @@ hyperlight-guest-bin = { path = "../../../hyperlight_guest_bin" }
 hyperlight-common = { path = "../../../hyperlight_common", default-features = false }
 hyperlight-guest-tracing = { path = "../../../hyperlight_guest_tracing" }
 log = {version = "0.4", default-features = false }
+tracing = { version = "0.1.41", default-features = false, features = ["attributes"] }
 
 [features]
 default = []

src/tests/rust_guests/simpleguest/Cargo.toml

@@ -51,6 +51,7 @@ use hyperlight_guest_bin::host_comm::{
 use hyperlight_guest_bin::memory::malloc;
 use hyperlight_guest_bin::{MIN_STACK_ADDRESS, guest_logger};
 use log::{LevelFilter, error};
+use tracing::{Span, instrument};
 
 extern crate hyperlight_guest;
 
@@ -89,6 +90,7 @@ fn echo_float(function_call: &FunctionCall) -> Result<Vec<u8>> {
     }
 }
 
+#[instrument(skip_all, parent = Span::current(), level= "Trace")]
 fn print_output(message: &str) -> Result<Vec<u8>> {
     let res = call_host_function::<i32>(
         "HostPrint",
@@ -99,6 +101,7 @@ fn print_output(message: &str) -> Result<Vec<u8>> {
     Ok(get_flatbuffer_result(res))
 }
 
+#[instrument(skip_all, parent = Span::current(), level= "Trace")]
 fn simple_print_output(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = function_call.parameters.clone().unwrap()[0].clone() {
         print_output(&message)
@@ -865,6 +868,7 @@ fn exec_mapped_buffer(function_call: &FunctionCall) -> Result<Vec<u8>> {
 }
 
 #[no_mangle]
+#[instrument(skip_all, parent = Span::current(), level= "Trace")]
 pub extern "C" fn hyperlight_main() {
     let twenty_four_k_in_def = GuestFunctionDefinition::new(
         "24K_in_8K_out".to_string(),
@@ -1321,6 +1325,7 @@ pub extern "C" fn hyperlight_main() {
 }
 
 #[no_mangle]
+#[instrument(skip_all, parent = Span::current(), level= "Trace")]
 pub fn guest_dispatch_function(function_call: FunctionCall) -> Result<Vec<u8>> {
     // This test checks the stack behavior of the input/output buffer
     // by calling the host before serializing the function call.
@@ -1366,6 +1371,7 @@ pub fn guest_dispatch_function(function_call: FunctionCall) -> Result<Vec<u8>> {
 }
 
 // Interprets the given guest function call as a host function call and dispatches it to the host.
+#[instrument(skip_all, parent = Span::current(), level= "Trace")]
 fn fuzz_host_function(func: FunctionCall) -> Result<Vec<u8>> {
     let mut params = func.parameters.unwrap();
     // first parameter must be string (the name of the host function to call)