#!/bin/sh

device="$1"
mode="$2"


case "$mode" in
	unbind)
		# ****************
		# * UNBIND BEGIN *
		# ****************

		echo "mode: unbind driver and bind vfio"

		id="$(lspci -n -s "$device" | cut -d' ' -f3 | tr ':' ' ')"
		echo "$device $id"
		echo "$id" > /sys/bus/pci/drivers/vfio-pci/new_id
		echo "$device" > /sys/bus/pci/devices/"$device"/driver/unbind
		echo "$device" > /sys/bus/pci/drivers/vfio-pci/bind
		echo "$id" > /sys/bus/pci/drivers/vfio-pci/remove_id

		# ****************
		# * UNBIND END   *
		# ****************
		;;

	bind)
		# ****************
		# * REBIND BEGIN *
		# ****************

		echo "mode: unbind vfio and bind driver"

		echo 1 > /sys/bus/pci/devices/"$device"/remove
		echo 1 > /sys/bus/pci/rescan

		# ****************
		# * REBIND END   *
		# ****************
		;;

	*)
		echo "error: unkown mode '$mode'"
		;;
esac