[ci] fixup: Add checks for tracing features and run tracing tests

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

Author: dblnz

.github/workflows/dep_rust.yml

@@ -137,6 +137,7 @@ jobs:
           cargo check -p hyperlight-host --features crashdump
           cargo check -p hyperlight-host --features print_debug
           cargo check -p hyperlight-host --features gdb
+          cargo check -p hyperlight-host --features trace_guest,unwind_guest,mem_profile
 
           # without any features
           just test-compilation-no-default-features ${{ matrix.config }}
@@ -169,6 +170,13 @@ jobs:
           RUST_LOG: debug
         run: just test-rust-crashdump ${{ matrix.config }} ${{ matrix.hypervisor == 'mshv' && 'mshv2' || ''}}
 
+      - name: Run Rust Tracing tests - linux
+        if: runner.os == 'Linux'
+        env:
+          CARGO_TERM_COLOR: always
+          RUST_LOG: debug
+        run: just test-rust-tracing ${{ matrix.config }} ${{ matrix.hypervisor == 'mshv3' && 'mshv3' || ''}}
+
       - name: Download benchmarks from "latest"
         run: just bench-download ${{ runner.os }} ${{ matrix.hypervisor }} ${{ matrix.cpu}} dev-latest # compare to prerelease
         env:

Justfile

@@ -82,13 +82,17 @@ test-like-ci config=default-target hypervisor="kvm":
     cargo check -p hyperlight-host --features crashdump
     cargo check -p hyperlight-host --features print_debug
     cargo check -p hyperlight-host --features gdb
+    cargo check -p hyperlight-host --features trace_guest,unwind_guest,mem_profile
 
     @# without any driver (should fail to compile)
     just test-compilation-no-default-features {{config}}
 
     @# test the crashdump feature
     just test-rust-crashdump {{config}}
 
+    @# test the tracing related features
+    just test-rust-tracing {{config}} {{ if hypervisor == "mshv3" {"mshv3"} else {""} }}
+
 # runs all tests
 test target=default-target features="": (test-unit target features) (test-isolated target features) (test-integration "rust" target features) (test-integration "c" target features) (test-seccomp target features)
 
@@ -141,6 +145,25 @@ test-rust-gdb-debugging target=default-target features="":
 test-rust-crashdump target=default-target features="":
     cargo test --profile={{ if target == "debug" { "dev" } else { target } }} {{ if features =="" {'--features crashdump'} else { "--features crashdump," + features } }} -- test_crashdump
 
+# rust test for tracing
+test-rust-tracing target=default-target features="":
+    # Run tests for the tracing guest and macro
+    cargo test -p hyperlight-guest-tracing --profile={{ if target == "debug" { "dev" } else { target } }}
+    cargo test -p hyperlight-guest-tracing-macro --profile={{ if target == "debug" { "dev" } else { target } }}
+
+    # Prepare the tracing guest for testing
+    just build-rust-guests {{ target }} trace_guest
+    just move-rust-guests {{ target }}
+    # Run hello-world example with tracing enabled to get the trace output
+    # Capture the trace file path and print use it afterwards to run cargo run -p trace_dump
+    cargo run --profile={{ if target == "debug" { "dev" } else { target } }} --example hello-world --features {{ if features =="" {'trace_guest'} else { "trace_guest," + features } }} \
+        | sed -n 's/.*Creating trace file at: \(.*\)/\1/p' \
+        | xargs -I {} cargo run -p trace_dump ./{{ simpleguest_source }}/{{ target }}/simpleguest {} list_frames
+
+    # Rebuild the tracing guest without the tracing feature
+    # This is to ensure that the tracing feature does not affect the other tests
+    just build-rust-guests {{ target }}
+    just move-rust-guests {{ target }}
 
 ################
 ### LINTING ####