#!/bin/bash

# Get GPU PCI bus IDs and names from nvidia-smi
nvidia-smi --query-gpu=pci.bus_id,gpu_name --format=csv,noheader | while IFS=, read -r BUSID NAME; do
    # Remove whitespace
    BUSID=$(echo $BUSID | xargs)
    NAME=$(echo $NAME | xargs | tr ' ' '_')
    # Format PCI path (nvidia-smi uses XXXX:YY:ZZ.Z, sysfs uses 0000:YY:ZZ.Z)
    SYSFSID="0000:${BUSID#*:}"
    ROMPATH="/sys/bus/pci/devices/${SYSFSID}/rom"
    OUTFILE="vbios_${SYSFSID}_${NAME}.rom"
    if [ -f "$ROMPATH" ]; then
        echo "Dumping vBIOS for $SYSFSID ($NAME)..."
        sudo sh -c "echo 1 > $ROMPATH"
        sudo cat "$ROMPATH" > "$OUTFILE"
        sudo sh -c "echo 0 > $ROMPATH"
        ls -lh "$OUTFILE"
    else
        echo "ROM file not found for $SYSFSID ($NAME)"
    fi
done