1 files changed, 163 insertions, 0 deletions

diff --git a/tools/testing/selftests/net/forwarding/devlink_lib.sh b/tools/testing/selftests/net/forwarding/devlink_lib.sh
index 2b9296f6aa07..13d03a6d85ba 100644
--- a/tools/testing/selftests/net/forwarding/devlink_lib.sh
+++ b/tools/testing/selftests/net/forwarding/devlink_lib.sh
@@ -29,6 +29,12 @@ if [ $? -ne 0 ]; then
         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
 
@@ -192,3 +198,160 @@ devlink_tc_bind_pool_th_restore()
         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
+}