#!/bin/bash
# Verify GSP firmware for TU104
# Usage: ./verify_gsp_firmware.sh

set -euo pipefail

FIRMWARE_DIR="/lib/firmware/nvidia/tu104/gsp"

echo "=== GSP Firmware Verification (TU104) ==="
echo ""

if [[ ! -d "$FIRMWARE_DIR" ]]; then
    echo "ERROR: Firmware directory not found: $FIRMWARE_DIR"
    echo "Install firmware with: rzsh nvidia vgpu firmware install"
    exit 1
fi

echo "Firmware directory: $FIRMWARE_DIR"
echo ""

# Check required files
required_files=(
    "gsp-535.113.01.bin"
    "gsp-570.144.bin"
    "booter_load-535.113.01.bin"
    "booter_load-570.144.bin"
    "booter_unload-535.113.01.bin"
    "booter_unload-570.144.bin"
    "bootloader-535.113.01.bin"
    "bootloader-570.144.bin"
)

all_ok=true
for file in "${required_files[@]}"; do
    if [[ -f "$FIRMWARE_DIR/$file" ]]; then
        size=$(stat -c%s "$FIRMWARE_DIR/$file")
        echo "  ✓ $file ($size bytes)"
    else
        echo "  ✗ $file MISSING"
        all_ok=false
    fi
done

echo ""
if $all_ok; then
    echo "✓ All required firmware files present"
else
    echo "✗ Some firmware files missing"
    echo ""
    echo "To install firmware:"
    echo "  1. Download from NVIDIA or copy from nvidia-linux-firmware repo"
    echo "  2. sudo cp -r nvidia/tu104/gsp/* /lib/firmware/nvidia/tu104/gsp/"
    echo "  3. sudo update-initramfs -u"
fi

echo ""
echo "Current kernel module version:"
if lsmod | grep -q nvidia; then
    cat /proc/driver/nvidia/version 2>/dev/null | head -1
fi