add-vfio.sh

#!/usr/bin/env bash

# add-vfio.sh: detect iommu groups and override the driver of these devices to vfio-pci

IsolatingDevice="$1"

IOMMUGroup=$(ls "/sys/bus/pci/devices/$IsolatingDevice/iommu_group/devices")

for Device in ${IOMMUGroup}; do
    DeviceDir="/sys/bus/pci/devices/$Device"
    echo "Isolating: $(lspci -nns $Device)"

    # Remove it from previous binded driver
    if [ -d "$DeviceDir/driver" ]; then
        echo "$Device" > "$DeviceDir/driver/unbind"
    fi

    # Bind it to vfio-pci
    echo 'vfio-pci' > "$DeviceDir/driver_override"
    echo "$Device" > /sys/bus/pci/drivers_probe
done