mirror of
https://github.com/hermitcore/libhermit.git
synced 2025-03-30 00:00:15 +01:00
137 lines
3 KiB
Bash
137 lines
3 KiB
Bash
#!/bin/bash
|
|
#
|
|
# Initialize SRIOV virtual devices
|
|
#
|
|
# This is usually run automatically by systemd after a hardware activation
|
|
# event in udev has triggered a start of the rdma.service unit
|
|
port=1
|
|
|
|
function __get_parent_pci_dev()
|
|
{
|
|
pushd /sys/bus/pci/devices/$pci_dev >/dev/null 2>&1
|
|
ppci_dev=`ls -l physfn | cut -f 2 -d '/'`
|
|
popd >/dev/null 2>&1
|
|
}
|
|
|
|
function __get_parent_ib_dev()
|
|
{
|
|
ib_dev=`ls -l | awk '/'$ppci_dev'/ { print $9 }'`
|
|
}
|
|
|
|
function __get_parent_net_dev()
|
|
{
|
|
for netdev in /sys/bus/pci/devices/$ppci_dev/net/* ; do
|
|
if [ "$port" -eq `cat $netdev/dev_port` ]; then
|
|
netdev=`basename $netdev`
|
|
break
|
|
fi
|
|
done
|
|
}
|
|
|
|
function __get_vf_num()
|
|
{
|
|
pushd /sys/bus/pci/devices/$ppci_dev >/dev/null 2>&1
|
|
vf=`ls -l virtfn* | awk '/'$pci_dev'/ { print $9 }' | sed -e 's/virtfn//'`
|
|
popd >/dev/null 2>&1
|
|
}
|
|
|
|
function __en_sriov_set_vf()
|
|
{
|
|
pci_dev=$1
|
|
shift
|
|
[ "$1" = "port" ] && port=$2 && shift 2
|
|
# We find our parent device by the netdev registered port number,
|
|
# however, the netdev port numbers start at 0 while the port
|
|
# numbers on the card start at 1, so we subtract 1 from our
|
|
# configured port number to get the netdev number
|
|
let port--
|
|
# Now we need to fill in the necessary information to pass to the ip
|
|
# command
|
|
__get_parent_pci_dev
|
|
__get_parent_net_dev
|
|
__get_vf_num
|
|
# The rest is easy. Either the user passed valid arguments as options
|
|
# or they didn't
|
|
ip link set dev $netdev vf $vf $*
|
|
}
|
|
|
|
function __ib_sriov_set_vf()
|
|
{
|
|
pci_dev=$1
|
|
shift
|
|
[ "$1" = "port" ] && port=$2 && shift 2
|
|
guid=""
|
|
__get_parent_pci_dev
|
|
__get_parent_ib_dev
|
|
[ -f $ib_dev/iov/$pci_dev/ports/$port/gid_idx/0 ] || return
|
|
while [ -n "$1" ]; do
|
|
case $1 in
|
|
guid)
|
|
guid=$2
|
|
shift 2
|
|
;;
|
|
pkey)
|
|
shift 1
|
|
break
|
|
;;
|
|
*)
|
|
echo "Unknown option in $src"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
if [ -n "$guid" ]; then
|
|
guid_idx=`cat "$ib_dev/iov/$pci_dev/ports/$port/gid_idx/0"`
|
|
echo "$guid" > "$ib_dev/iov/ports/$port/admin_guids/$guid_idx"
|
|
fi
|
|
i=0
|
|
while [ -n "$1" ]; do
|
|
for pkey in $ib_dev/iov/ports/$port/pkeys/*; do
|
|
if [ `cat $pkey` = "$1" ]; then
|
|
echo `basename $pkey` > $ib_dev/iov/$pci_dev/ports/$port/pkey_idx/$i
|
|
let i++
|
|
break
|
|
fi
|
|
done
|
|
shift
|
|
done
|
|
}
|
|
|
|
[ -d /sys/class/infiniband ] || return
|
|
pushd /sys/class/infiniband >/dev/null 2>&1
|
|
|
|
if [ -z "$*" ]; then
|
|
src=/etc/rdma/sriov-vfs
|
|
[ -f "$src" ] || return
|
|
grep -v "^#" $src | while read -a args; do
|
|
# When we use read -a to read into an array, the index starts at
|
|
# 0, unlike below where the arg count starts at 1
|
|
port=1
|
|
next_arg=1
|
|
[ "${args[$next_arg]}" = "port" ] && next_arg=3
|
|
case ${args[$next_arg]} in
|
|
guid|pkey)
|
|
__ib_sriov_set_vf ${args[*]}
|
|
;;
|
|
mac|vlan|rate|spoofchk|enable)
|
|
__en_sriov_set_vf ${args[*]}
|
|
;;
|
|
*)
|
|
;;
|
|
esac
|
|
done
|
|
else
|
|
[ "$2" = "port" ] && next_arg=$4 || next_arg=$2
|
|
case $next_arg in
|
|
guid|pkey)
|
|
__ib_sriov_set_vf $*
|
|
;;
|
|
mac|vlan|rate|spoofchk|enable)
|
|
__en_sriov_set_vf $*
|
|
;;
|
|
*)
|
|
;;
|
|
esac
|
|
fi
|
|
|
|
popd >/dev/null 2>&1
|