#!/bin/bash
shopt -s nullglob
for g in /sys/kernel/iommu_groups/*; do
    echo "IOMMU Group ${g##*/}:"
    for d in $g/devices/*; do
        echo -e "\t$(lspci -nns ${d##*/})"
    done;
done;

# vfio.conf
options vfio-pci.ids=10de:2482,10de:228b

# vfio-pci.conf
vfio-pci

$ sudo vim /etc/modprobe.d/blacklist.conf

blacklist nvidia
blacklist nouveau

#!/bin/bash
shopt -s nullglob
for g in /sys/kernel/iommu_groups/*; do
    echo "IOMMU Group ${g##*/}:"
    for d in $g/devices/*; do
        echo -e "\t$(lspci -nns ${d##*/})"
    done;
done;

# qemu

OBJECT="$1"
OPERATION="$2"

if [[ $OBJECT == "win10" ]]; then
	case "$OPERATION" in
        	"prepare")
                systemctl start libvirt-nosleep@"$OBJECT"  2>&1 | tee -a /var/log/libvirt/custom_hooks.log
                /bin/vfio-startup.sh 2>&1 | tee -a /var/log/libvirt/custom_hooks.log
                ;;

            "release")
                systemctl stop libvirt-nosleep@"$OBJECT"  2>&1 | tee -a /var/log/libvirt/custom_hooks.log  
                /bin/vfio-teardown.sh 2>&1 | tee -a /var/log/libvirt/custom_hooks.log
                ;;
	esac
fi

set -x

# 关闭窗口管理器, 如
# systemctl stop sdm.service
killall bspwm

# 放置数据竞争
sleep 4

# 关闭nvidia驱动, 以便后续能启动vfio
modprobe -r nvidia_drm
modprobe -r nvidia_uvm
modprobe -r nvidia_modeset
modprobe -r nvidia

# 关闭console连接, 之后在console中就看不到光标闪烁了, echo 1能恢复
echo 0 > /sys/class/vtconsole/vtcon0/bind
echo 0 > /sys/class/vtconsole/vtcon1/bind

# 有的GPU没有就注释掉
# echo efi-framebuffer.0 > /sys/bus/platform/drivers/efi-framebuffer/unbind

sleep 2

# detach GPU, 01_00_0是GPU对应的id, 如
# $ lspci -nnk
# 01:00.0 VGA compatible controller [0300]: NVIDIA Corporation GA104 [GeForce RTX 3070 Ti] [10de:2482] (rev a1)
# 01:00.1 Audio device [0403]: NVIDIA Corporation GA104 High Definition Audio Controller [10de:228b] (rev a1)
virsh nodedev-detach pci_0000_01_00_0
virsh nodedev-detach pci_0000_01_00_1

# 启动vfio
modprobe vfio-pci

set -x

virsh nodedev-reattach pci_0000_01_00_0
virsh nodedev-reattach pci_0000_01_00_1

modprobe nvidia
modprobe nvidia_modeset
modprobe nvidia_uvm
modprobe nvidia_drm

echo 1 > /sys/class/vtconsole/vtcon0/bind
# 我这里echo 1到vtcon1会死机所以注释掉了, 问题不大
# echo 1 > /sys/class/vtconsole/vtcon1/bind

# echo efi-framebuffer.0 > /sys/bus/platform/drivers/efi-framebuffer/bind

# 重新启动窗口管理器
# systemctl start sdm.service