358 lines
7.4 KiB
Bash
358 lines
7.4 KiB
Bash
#!/bin/bash
|
|
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
##############################################################################
|
|
# Defines
|
|
|
|
if [[ ! -v DEVLINK_DEV ]]; then
|
|
DEVLINK_DEV=$(devlink port show "${NETIFS[p1]}" -j \
|
|
| jq -r '.port | keys[]' | cut -d/ -f-2)
|
|
if [ -z "$DEVLINK_DEV" ]; then
|
|
echo "SKIP: ${NETIFS[p1]} has no devlink device registered for it"
|
|
exit 1
|
|
fi
|
|
if [[ "$(echo $DEVLINK_DEV | grep -c pci)" -eq 0 ]]; then
|
|
echo "SKIP: devlink device's bus is not PCI"
|
|
exit 1
|
|
fi
|
|
|
|
DEVLINK_VIDDID=$(lspci -s $(echo $DEVLINK_DEV | cut -d"/" -f2) \
|
|
-n | cut -d" " -f3)
|
|
fi
|
|
|
|
##############################################################################
|
|
# Sanity checks
|
|
|
|
devlink help 2>&1 | grep resource &> /dev/null
|
|
if [ $? -ne 0 ]; then
|
|
echo "SKIP: iproute2 too old, missing devlink resource support"
|
|
exit 1
|
|
fi
|
|
|
|
devlink help 2>&1 | grep trap &> /dev/null
|
|
if [ $? -ne 0 ]; then
|
|
echo "SKIP: iproute2 too old, missing devlink trap support"
|
|
exit 1
|
|
fi
|
|
|
|
##############################################################################
|
|
# Devlink helpers
|
|
|
|
devlink_resource_names_to_path()
|
|
{
|
|
local resource
|
|
local path=""
|
|
|
|
for resource in "${@}"; do
|
|
if [ "$path" == "" ]; then
|
|
path="$resource"
|
|
else
|
|
path="${path}/$resource"
|
|
fi
|
|
done
|
|
|
|
echo "$path"
|
|
}
|
|
|
|
devlink_resource_get()
|
|
{
|
|
local name=$1
|
|
local resource_name=.[][\"$DEVLINK_DEV\"]
|
|
|
|
resource_name="$resource_name | .[] | select (.name == \"$name\")"
|
|
|
|
shift
|
|
for resource in "${@}"; do
|
|
resource_name="${resource_name} | .[\"resources\"][] | \
|
|
select (.name == \"$resource\")"
|
|
done
|
|
|
|
devlink -j resource show "$DEVLINK_DEV" | jq "$resource_name"
|
|
}
|
|
|
|
devlink_resource_size_get()
|
|
{
|
|
local size=$(devlink_resource_get "$@" | jq '.["size_new"]')
|
|
|
|
if [ "$size" == "null" ]; then
|
|
devlink_resource_get "$@" | jq '.["size"]'
|
|
else
|
|
echo "$size"
|
|
fi
|
|
}
|
|
|
|
devlink_resource_size_set()
|
|
{
|
|
local new_size=$1
|
|
local path
|
|
|
|
shift
|
|
path=$(devlink_resource_names_to_path "$@")
|
|
devlink resource set "$DEVLINK_DEV" path "$path" size "$new_size"
|
|
check_err $? "Failed setting path $path to size $size"
|
|
}
|
|
|
|
devlink_reload()
|
|
{
|
|
local still_pending
|
|
|
|
devlink dev reload "$DEVLINK_DEV" &> /dev/null
|
|
check_err $? "Failed reload"
|
|
|
|
still_pending=$(devlink resource show "$DEVLINK_DEV" | \
|
|
grep -c "size_new")
|
|
check_err $still_pending "Failed reload - There are still unset sizes"
|
|
}
|
|
|
|
declare -A DEVLINK_ORIG
|
|
|
|
devlink_port_pool_threshold()
|
|
{
|
|
local port=$1; shift
|
|
local pool=$1; shift
|
|
|
|
devlink sb port pool show $port pool $pool -j \
|
|
| jq '.port_pool."'"$port"'"[].threshold'
|
|
}
|
|
|
|
devlink_port_pool_th_set()
|
|
{
|
|
local port=$1; shift
|
|
local pool=$1; shift
|
|
local th=$1; shift
|
|
local key="port_pool($port,$pool).threshold"
|
|
|
|
DEVLINK_ORIG[$key]=$(devlink_port_pool_threshold $port $pool)
|
|
devlink sb port pool set $port pool $pool th $th
|
|
}
|
|
|
|
devlink_port_pool_th_restore()
|
|
{
|
|
local port=$1; shift
|
|
local pool=$1; shift
|
|
local key="port_pool($port,$pool).threshold"
|
|
|
|
devlink sb port pool set $port pool $pool th ${DEVLINK_ORIG[$key]}
|
|
}
|
|
|
|
devlink_pool_size_thtype()
|
|
{
|
|
local pool=$1; shift
|
|
|
|
devlink sb pool show "$DEVLINK_DEV" pool $pool -j \
|
|
| jq -r '.pool[][] | (.size, .thtype)'
|
|
}
|
|
|
|
devlink_pool_size_thtype_set()
|
|
{
|
|
local pool=$1; shift
|
|
local thtype=$1; shift
|
|
local size=$1; shift
|
|
local key="pool($pool).size_thtype"
|
|
|
|
DEVLINK_ORIG[$key]=$(devlink_pool_size_thtype $pool)
|
|
devlink sb pool set "$DEVLINK_DEV" pool $pool size $size thtype $thtype
|
|
}
|
|
|
|
devlink_pool_size_thtype_restore()
|
|
{
|
|
local pool=$1; shift
|
|
local key="pool($pool).size_thtype"
|
|
local -a orig=(${DEVLINK_ORIG[$key]})
|
|
|
|
devlink sb pool set "$DEVLINK_DEV" pool $pool \
|
|
size ${orig[0]} thtype ${orig[1]}
|
|
}
|
|
|
|
devlink_tc_bind_pool_th()
|
|
{
|
|
local port=$1; shift
|
|
local tc=$1; shift
|
|
local dir=$1; shift
|
|
|
|
devlink sb tc bind show $port tc $tc type $dir -j \
|
|
| jq -r '.tc_bind[][] | (.pool, .threshold)'
|
|
}
|
|
|
|
devlink_tc_bind_pool_th_set()
|
|
{
|
|
local port=$1; shift
|
|
local tc=$1; shift
|
|
local dir=$1; shift
|
|
local pool=$1; shift
|
|
local th=$1; shift
|
|
local key="tc_bind($port,$dir,$tc).pool_th"
|
|
|
|
DEVLINK_ORIG[$key]=$(devlink_tc_bind_pool_th $port $tc $dir)
|
|
devlink sb tc bind set $port tc $tc type $dir pool $pool th $th
|
|
}
|
|
|
|
devlink_tc_bind_pool_th_restore()
|
|
{
|
|
local port=$1; shift
|
|
local tc=$1; shift
|
|
local dir=$1; shift
|
|
local key="tc_bind($port,$dir,$tc).pool_th"
|
|
local -a orig=(${DEVLINK_ORIG[$key]})
|
|
|
|
devlink sb tc bind set $port tc $tc type $dir \
|
|
pool ${orig[0]} th ${orig[1]}
|
|
}
|
|
|
|
devlink_traps_num_get()
|
|
{
|
|
devlink -j trap | jq '.[]["'$DEVLINK_DEV'"] | length'
|
|
}
|
|
|
|
devlink_traps_get()
|
|
{
|
|
devlink -j trap | jq -r '.[]["'$DEVLINK_DEV'"][].name'
|
|
}
|
|
|
|
devlink_trap_type_get()
|
|
{
|
|
local trap_name=$1; shift
|
|
|
|
devlink -j trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq -r '.[][][].type'
|
|
}
|
|
|
|
devlink_trap_action_set()
|
|
{
|
|
local trap_name=$1; shift
|
|
local action=$1; shift
|
|
|
|
# Pipe output to /dev/null to avoid expected warnings.
|
|
devlink trap set $DEVLINK_DEV trap $trap_name \
|
|
action $action &> /dev/null
|
|
}
|
|
|
|
devlink_trap_action_get()
|
|
{
|
|
local trap_name=$1; shift
|
|
|
|
devlink -j trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq -r '.[][][].action'
|
|
}
|
|
|
|
devlink_trap_group_get()
|
|
{
|
|
devlink -j trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq -r '.[][][].group'
|
|
}
|
|
|
|
devlink_trap_metadata_test()
|
|
{
|
|
local trap_name=$1; shift
|
|
local metadata=$1; shift
|
|
|
|
devlink -jv trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq -e '.[][][].metadata | contains(["'$metadata'"])' \
|
|
&> /dev/null
|
|
}
|
|
|
|
devlink_trap_rx_packets_get()
|
|
{
|
|
local trap_name=$1; shift
|
|
|
|
devlink -js trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq '.[][][]["stats"]["rx"]["packets"]'
|
|
}
|
|
|
|
devlink_trap_rx_bytes_get()
|
|
{
|
|
local trap_name=$1; shift
|
|
|
|
devlink -js trap show $DEVLINK_DEV trap $trap_name \
|
|
| jq '.[][][]["stats"]["rx"]["bytes"]'
|
|
}
|
|
|
|
devlink_trap_stats_idle_test()
|
|
{
|
|
local trap_name=$1; shift
|
|
local t0_packets t0_bytes
|
|
local t1_packets t1_bytes
|
|
|
|
t0_packets=$(devlink_trap_rx_packets_get $trap_name)
|
|
t0_bytes=$(devlink_trap_rx_bytes_get $trap_name)
|
|
|
|
sleep 1
|
|
|
|
t1_packets=$(devlink_trap_rx_packets_get $trap_name)
|
|
t1_bytes=$(devlink_trap_rx_bytes_get $trap_name)
|
|
|
|
if [[ $t0_packets -eq $t1_packets && $t0_bytes -eq $t1_bytes ]]; then
|
|
return 0
|
|
else
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
devlink_traps_enable_all()
|
|
{
|
|
local trap_name
|
|
|
|
for trap_name in $(devlink_traps_get); do
|
|
devlink_trap_action_set $trap_name "trap"
|
|
done
|
|
}
|
|
|
|
devlink_traps_disable_all()
|
|
{
|
|
for trap_name in $(devlink_traps_get); do
|
|
devlink_trap_action_set $trap_name "drop"
|
|
done
|
|
}
|
|
|
|
devlink_trap_groups_get()
|
|
{
|
|
devlink -j trap group | jq -r '.[]["'$DEVLINK_DEV'"][].name'
|
|
}
|
|
|
|
devlink_trap_group_action_set()
|
|
{
|
|
local group_name=$1; shift
|
|
local action=$1; shift
|
|
|
|
# Pipe output to /dev/null to avoid expected warnings.
|
|
devlink trap group set $DEVLINK_DEV group $group_name action $action \
|
|
&> /dev/null
|
|
}
|
|
|
|
devlink_trap_group_rx_packets_get()
|
|
{
|
|
local group_name=$1; shift
|
|
|
|
devlink -js trap group show $DEVLINK_DEV group $group_name \
|
|
| jq '.[][][]["stats"]["rx"]["packets"]'
|
|
}
|
|
|
|
devlink_trap_group_rx_bytes_get()
|
|
{
|
|
local group_name=$1; shift
|
|
|
|
devlink -js trap group show $DEVLINK_DEV group $group_name \
|
|
| jq '.[][][]["stats"]["rx"]["bytes"]'
|
|
}
|
|
|
|
devlink_trap_group_stats_idle_test()
|
|
{
|
|
local group_name=$1; shift
|
|
local t0_packets t0_bytes
|
|
local t1_packets t1_bytes
|
|
|
|
t0_packets=$(devlink_trap_group_rx_packets_get $group_name)
|
|
t0_bytes=$(devlink_trap_group_rx_bytes_get $group_name)
|
|
|
|
sleep 1
|
|
|
|
t1_packets=$(devlink_trap_group_rx_packets_get $group_name)
|
|
t1_bytes=$(devlink_trap_group_rx_bytes_get $group_name)
|
|
|
|
if [[ $t0_packets -eq $t1_packets && $t0_bytes -eq $t1_bytes ]]; then
|
|
return 0
|
|
else
|
|
return 1
|
|
fi
|
|
}
|