Phase 4: Portable Architecture Implementation

- Firmware Analyzer: Runtime firmware signature computation (CRC32, SHA256) and pattern-based RPC handler address discovery
- Dynamic RPC Dispatcher: Version-aware dispatch with native firmware handlers (550/560/570/580) and emulation fallback
- Hardware Architecture Layer: Turing, Ampere, Blackwell configs with BAR layouts, doorbell offsets, MSI-X vectors
- Driver Version Adaptation Layer: Feature detection per driver version
- Declarative YAML configs for hardware (turing/ampere/blackwell) and versions (550/560/570/580)
- Kernel unit tests for GSP emulation (gsp_emulation_test.c)
- Test infrastructure scripts (test_gsp_emu.sh, test_guest_driver.sh, test_cuda_workloads.sh, test_graphics_workloads.sh, test_windows_guest.sh, test_vgpu_stack.sh)
- Updated kernel Makefile with new source files

1 files changed, 123 insertions, 101 deletions

diff --git a/scripts/test_vgpu_stack.sh b/scripts/test_vgpu_stack.sh
index d31444d..4df9cdc 100755
--- a/scripts/test_vgpu_stack.sh
+++ b/scripts/test_vgpu_stack.sh
@@ -1,127 +1,149 @@
 #!/bin/bash
-# test_vgpu_stack.sh - Comprehensive vGPU stack test
+# Comprehensive Test Suite for Open-IOV vGPU-IO
+# Runs all validation tests in sequence
 
-set -euo pipefail
+set -e
 
-BASE_DIR="$OPEN_IOV_ROOT"
-LIB_DIR="$BASE_DIR/userspace/lib"
-EXAMPLE="$BASE_DIR/userspace/examples/example_vgpu"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+OPEN_IOV_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
 
-echo "=========================================="
-echo "vGPU Stack Test Suite"
-echo "=========================================="
-
-# Test 1: Module loading
-echo "Test 1: Module loading"
-for mod in nvidia open_iov_vfio gsp_vgpu; do
-    if lsmod | grep -q "^$mod "; then
-        echo "  PASS: $mod loaded"
-    else
-        echo "  FAIL: $mod not loaded"
-        exit 1
-    fi
-done
+echo "=== Open-IOV vGPU-IO Comprehensive Test Suite ==="
+echo "Open-IOV root: $OPEN_IOV_ROOT"
+echo "Date: $(date)"
+echo ""
 
-# Test 2: Device nodes
-echo "Test 2: Device nodes"
-if [[ -c /dev/gsp-vgpu ]]; then
-    echo "  PASS: /dev/gsp-vgpu exists"
-else
-    echo "  FAIL: /dev/gsp-vgpu missing"
+# Check if we're root
+if [ "$EUID" -ne 0 ]; then
+    echo "ERROR: This script must be run as root"
     exit 1
 fi
 
-# Test 3: Library loading
-echo "Test 3: Library loading"
-if LD_LIBRARY_PATH="$LIB_DIR" ldd "$EXAMPLE" 2>&1 | grep -q "libiov.so"; then
-    echo "  PASS: libiov.so loads correctly"
-else
-    echo "  FAIL: libiov.so load failed"
-    exit 1
-fi
+# Parse arguments
+RUN_KERNEL_TESTS=true
+RUN_GUEST_TESTS=true
+RUN_CUDA_TESTS=true
+RUN_GRAPHICS_TESTS=true
+RUN_WINDOWS_TESTS=false
 
-# Test 4: Example app help
-echo "Test 4: Example app help"
-if LD_LIBRARY_PATH="$LIB_DIR" "$EXAMPLE" --help 2>&1 | grep -q "Usage"; then
-    echo "  PASS: example_vgpu runs"
-else
-    echo "  FAIL: example_vgpu failed"
-    exit 1
-fi
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        --kernel-only)
+            RUN_GUEST_TESTS=false
+            RUN_CUDA_TESTS=false
+            RUN_GRAPHICS_TESTS=false
+            shift
+            ;;
+        --guest-only)
+            RUN_KERNEL_TESTS=false
+            RUN_CUDA_TESTS=false
+            RUN_GRAPHICS_TESTS=false
+            shift
+            ;;
+        --no-windows)
+            RUN_WINDOWS_TESTS=false
+            shift
+            ;;
+        --windows)
+            RUN_WINDOWS_TESTS=true
+            shift
+            ;;
+        *)
+            echo "Unknown option: $1"
+            echo "Usage: $0 [--kernel-only] [--guest-only] [--no-windows] [--windows]"
+            exit 1
+            ;;
+    esac
+done
+
+RESULTS=()
 
-# Test 5: mdev creation (if GPU available)
-echo "Test 5: mdev creation"
-GPU_PCI=$(lspci -D -d 10de: | grep -E "VGA|3D controller" | head -1 | awk '{print $1}')
-if [[ -n "$GPU_PCI" ]]; then
-    echo "  Found GPU: $GPU_PCI"
-    MDEV_TYPE_PATH="/sys/bus/pci/devices/$GPU_PCI/mdev_supported_types/nvidia-vgpu-turing"
-    if [[ -d "$MDEV_TYPE_PATH" ]]; then
-        AVAIL=$(cat "$MDEV_TYPE_PATH/available_instances" 2>/dev/null || echo 0)
-        if [[ "$AVAIL" -gt 0 ]]; then
-            UUID=$(uuidgen)
-            echo "$UUID" | sudo tee "$MDEV_TYPE_PATH/create" > /dev/null
-            sleep 1
-            if [[ -d "/sys/bus/mdev/devices/$UUID" ]]; then
-                echo "  PASS: mdev created: $UUID"
-                # Cleanup
-                echo 1 | sudo tee "/sys/bus/mdev/devices/$UUID/remove" > /dev/null
-            else
-                echo "  FAIL: mdev creation failed"
-                exit 1
-            fi
-        else
-            echo "  SKIP: No available instances"
-        fi
+# Test 1: Kernel GSP Emulation Tests
+if [ "$RUN_KERNEL_TESTS" = true ]; then
+    echo ""
+    echo "========================================"
+    echo "TEST 1: Kernel GSP Emulation Unit Tests"
+    echo "========================================"
+    if "$SCRIPT_DIR/test_gsp_emu.sh"; then
+        RESULTS+=("Kernel GSP Emulation: PASS")
     else
-        echo "  SKIP: nvidia-vgpu-turing type not available"
+        RESULTS+=("Kernel GSP Emulation: FAIL")
     fi
-else
-    echo "  SKIP: No NVIDIA GPU found"
 fi
 
-# Test 6: Kernel log check
-echo "Test 6: Kernel log check"
-if dmesg -T | grep -q "GSP vGPU-IO driver loaded"; then
-    echo "  PASS: gsp-vgpu driver loaded successfully"
-else
-    echo "  WARN: gsp-vgpu load message not found in dmesg"
+# Test 2: Guest Driver Integration
+if [ "$RUN_GUEST_TESTS" = true ]; then
+    echo ""
+    echo "========================================"
+    echo "TEST 2: Guest Driver Integration"
+    echo "========================================"
+    if "$SCRIPT_DIR/test_guest_driver.sh"; then
+        RESULTS+=("Guest Driver Integration: PASS")
+    else
+        RESULTS+=("Guest Driver Integration: FAIL")
+    fi
 fi
 
-if dmesg -T | grep -q "NVIDIA vGPU VFIO driver loaded"; then
-    echo "  PASS: open-iov-vfio driver loaded successfully"
-else
-    echo "  WARN: open-iov-vfio load message not found in dmesg"
+# Test 3: CUDA Workloads
+if [ "$RUN_CUDA_TESTS" = true ]; then
+    echo ""
+    echo "========================================"
+    echo "TEST 3: CUDA Workload Validation"
+    echo "========================================"
+    if "$SCRIPT_DIR/test_cuda_workloads.sh"; then
+        RESULTS+=("CUDA Workloads: PASS")
+    else
+        RESULTS+=("CUDA Workloads: FAIL")
+    fi
 fi
 
-# Test 7: Firmware verification
-echo "Test 7: Firmware verification"
-FW="$BASE_DIR/firmware/570-GSP/gsp-570.144.fwimage.bin"
-if [[ -f "$FW" ]]; then
-    SIZE=$(stat -c%s "$FW")
-    EXPECTED=29622272  # ~28.5MB
-    if [[ "$SIZE" -eq "$EXPECTED" ]]; then
-        echo "  PASS: Firmware size correct ($SIZE bytes)"
+# Test 4: Graphics Workloads
+if [ "$RUN_GRAPHICS_TESTS" = true ]; then
+    echo ""
+    echo "========================================"
+    echo "TEST 4: Graphics Workload Validation"
+    echo "========================================"
+    if "$SCRIPT_DIR/test_graphics_workloads.sh"; then
+        RESULTS+=("Graphics Workloads: PASS")
     else
-        echo "  WARN: Firmware size mismatch: $SIZE (expected $EXPECTED)"
+        RESULTS+=("Graphics Workloads: FAIL")
     fi
-    
-    # Check ELF header
-    if file "$FW" | grep -q "ELF 64-bit LSB executable, UCB RISC-V"; then
-        echo "  PASS: Firmware is valid RISC-V ELF"
+fi
+
+# Test 5: Windows Guest (optional)
+if [ "$RUN_WINDOWS_TESTS" = true ]; then
+    echo ""
+    echo "========================================"
+    echo "TEST 5: Windows Guest Validation"
+    echo "========================================"
+    if "$SCRIPT_DIR/test_windows_guest.sh"; then
+        RESULTS+=("Windows Guest: PASS")
     else
-        echo "  FAIL: Firmware is not valid RISC-V ELF"
-        exit 1
+        RESULTS+=("Windows Guest: FAIL")
     fi
-else
-    echo "  FAIL: Firmware not found"
-    exit 1
 fi
 
+# Summary
 echo ""
-echo "=========================================="
-echo "All Tests Passed!"
-echo "=========================================="
+echo "========================================"
+echo "TEST SUMMARY"
+echo "========================================"
+for result in "${RESULTS[@]}"; do
+    echo "  $result"
+done
+
+# Check for failures
+FAIL_COUNT=0
+for result in "${RESULTS[@]}"; do
+    if [[ "$result" == *"FAIL"* ]]; then
+        FAIL_COUNT=$((FAIL_COUNT + 1))
+    fi
+done
+
 echo ""
-echo "Stack is ready for QEMU guest launch."
-echo "Run: $BASE_DIR/scripts/create_mdev.sh"
-echo "Then: $BASE_DIR/scripts/launch_qemu_vgpu.sh"
\ No newline at end of file
+if [ $FAIL_COUNT -eq 0 ]; then
+    echo "ALL TESTS PASSED ✓"
+    exit 0
+else
+    echo "$FAIL_COUNT TEST(S) FAILED ✗"
+    exit 1
+fi
\ No newline at end of file