blob: f3a0c2648de87644b0dc39605274c8093772e742 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#!/bin/bash
uio_drivers="igb_uio uio_pci_generic vfio-pci"
tmpfile=$(mktemp)
function bind_drv() {
addr=$1
modalias=$(cat $selection/modalias)
native_drv=$(modprobe -R $modalias)
array=()
for drv in $native_drv $uio_drivers; do
if [ -e /sys/bus/pci/drivers/$drv ]; then
echo driver $drv
drv_desc=$(modinfo $drv | grep description: | sed -e 's/.*:[[:space:]]\+//' )
array+=("${drv}")
array+=("${drv_desc}")
fi
done
dialog --backtitle "PCI NIC Bind Utility" \
--clear \
--menu "Select kernel driver" 18 100 12 \
"${array[@]}" 2> $tmpfile
retval=$?
selection=$(cat $tmpfile)
rm $tmpfile
if [ $retval -ne 0 ]; then
return
fi
vd=$(cat /sys/bus/pci/devices/${addr}/vendor /sys/bus/pci/devices/${addr}/device)
echo $addr | tee /sys/bus/pci/devices/${addr}/driver/unbind > /dev/null 2> /dev/null
echo $vd | tee /sys/bus/pci/drivers/${selection}/new_id > /dev/null 2> /dev/null
echo $addr | tee /sys/bus/pci/drivers/${selection}/bind > /dev/null 2> /dev/null
}
function find_pci_slot() {
addr=$1
[ ! "$(ls -A /sys/bus/pci/slots )" ] && echo "No PCI slot data" && return
for slot in $(find /sys/bus/pci/slots/* -maxdepth 0 -exec basename {} \;); do
slot_addr=$(cat /sys/bus/pci/slots/$slot/address)
if [[ "${addr}" == *"${slot_addr}"* ]]; then
echo "PCI slot: ${slot}"
return
fi
done
echo "Unknown PCI slot"
}
! type -ap dialog > /dev/null && echo "Please install dialog (apt-get install dialog)" && exit
if [ $USER != "root" ] ; then
echo "Restarting script with sudo..."
sudo $0 ${*}
exit
fi
cd /sys/bus/pci/devices
while true; do
array=()
for addr in *; do
class=$(cat ${addr}/class)
if [ "$class" = "0x020000" ]; then
name=$(lspci -s $addr | sed -e 's/.*: //')
if [ -e "/sys/bus/pci/devices/$addr/driver" ]; then
drv=$(basename $(readlink -f /sys/bus/pci/devices/$addr/driver))
else
drv=" "
fi
slot=$(find_pci_slot ${addr})
array+=("${addr}")
array+=("${drv}|${name}")
array+=("${slot}")
fi
done
dialog --backtitle "PCI NIC Bind Utility" \
--item-help \
--clear \
--column-separator '|' \
--menu "Select NIC" 18 100 12 \
"${array[@]}" 2> $tmpfile
retval=$?
selection=$(cat $tmpfile)
rm $tmpfile
if [ $retval -ne 0 ]; then
exit
fi
bind_drv $selection
done
|
