1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3 4############################################################################## 5# Defines 6 7: "${WAIT_TIMEOUT:=20}" 8 9BUSYWAIT_TIMEOUT=$((WAIT_TIMEOUT * 1000)) # ms 10 11# Kselftest framework constants. 12ksft_pass=0 13ksft_fail=1 14ksft_xfail=2 15ksft_skip=4 16 17# namespace list created by setup_ns 18NS_LIST=() 19 20############################################################################## 21# Helpers 22 23__ksft_status_merge() 24{ 25 local a=$1; shift 26 local b=$1; shift 27 local -A weights 28 local weight=0 29 30 local i 31 for i in "$@"; do 32 weights[$i]=$((weight++)) 33 done 34 35 if [[ ${weights[$a]} > ${weights[$b]} ]]; then 36 echo "$a" 37 return 0 38 else 39 echo "$b" 40 return 1 41 fi 42} 43 44ksft_status_merge() 45{ 46 local a=$1; shift 47 local b=$1; shift 48 49 __ksft_status_merge "$a" "$b" \ 50 $ksft_pass $ksft_xfail $ksft_skip $ksft_fail 51} 52 53ksft_exit_status_merge() 54{ 55 local a=$1; shift 56 local b=$1; shift 57 58 __ksft_status_merge "$a" "$b" \ 59 $ksft_xfail $ksft_pass $ksft_skip $ksft_fail 60} 61 62loopy_wait() 63{ 64 local sleep_cmd=$1; shift 65 local timeout_ms=$1; shift 66 67 local start_time="$(date -u +%s%3N)" 68 while true 69 do 70 local out 71 if out=$("$@"); then 72 echo -n "$out" 73 return 0 74 fi 75 76 local current_time="$(date -u +%s%3N)" 77 if ((current_time - start_time > timeout_ms)); then 78 echo -n "$out" 79 return 1 80 fi 81 82 $sleep_cmd 83 done 84} 85 86busywait() 87{ 88 local timeout_ms=$1; shift 89 90 loopy_wait : "$timeout_ms" "$@" 91} 92 93# timeout in seconds 94slowwait() 95{ 96 local timeout_sec=$1; shift 97 98 loopy_wait "sleep 0.1" "$((timeout_sec * 1000))" "$@" 99} 100 101until_counter_is() 102{ 103 local expr=$1; shift 104 local current=$("$@") 105 106 echo $((current)) 107 ((current $expr)) 108} 109 110busywait_for_counter() 111{ 112 local timeout=$1; shift 113 local delta=$1; shift 114 115 local base=$("$@") 116 busywait "$timeout" until_counter_is ">= $((base + delta))" "$@" 117} 118 119slowwait_for_counter() 120{ 121 local timeout=$1; shift 122 local delta=$1; shift 123 124 local base=$("$@") 125 slowwait "$timeout" until_counter_is ">= $((base + delta))" "$@" 126} 127 128# Check for existence of tools which are built as part of selftests 129# but may also already exist in $PATH 130check_gen_prog() 131{ 132 local prog_name=$1; shift 133 134 if ! which $prog_name >/dev/null 2>/dev/null; then 135 PATH=$PWD:$PATH 136 if ! which $prog_name >/dev/null; then 137 echo "'$prog_name' command not found; skipping tests" 138 exit $ksft_skip 139 fi 140 fi 141} 142 143remove_ns_list() 144{ 145 local item=$1 146 local ns 147 local ns_list=("${NS_LIST[@]}") 148 NS_LIST=() 149 150 for ns in "${ns_list[@]}"; do 151 if [ "${ns}" != "${item}" ]; then 152 NS_LIST+=("${ns}") 153 fi 154 done 155} 156 157cleanup_ns() 158{ 159 local ns="" 160 local ret=0 161 162 for ns in "$@"; do 163 [ -z "${ns}" ] && continue 164 ip netns pids "${ns}" 2> /dev/null | xargs -r kill || true 165 ip netns delete "${ns}" &> /dev/null || true 166 if ! busywait $BUSYWAIT_TIMEOUT ip netns list \| grep -vq "^$ns$" &> /dev/null; then 167 echo "Warn: Failed to remove namespace $ns" 168 ret=1 169 else 170 remove_ns_list "${ns}" 171 fi 172 done 173 174 return $ret 175} 176 177cleanup_all_ns() 178{ 179 cleanup_ns "${NS_LIST[@]}" 180} 181 182# setup netns with given names as prefix. e.g 183# setup_ns local remote 184setup_ns() 185{ 186 local ns_name="" 187 local ns_list=() 188 for ns_name in "$@"; do 189 # avoid conflicts with local var: internal error 190 if [ "${ns_name}" = "ns_name" ]; then 191 echo "Failed to setup namespace '${ns_name}': invalid name" 192 cleanup_ns "${ns_list[@]}" 193 exit $ksft_fail 194 fi 195 196 # Some test may setup/remove same netns multi times 197 if [ -z "${!ns_name}" ]; then 198 eval "${ns_name}=${ns_name,,}-$(mktemp -u XXXXXX)" 199 else 200 cleanup_ns "${!ns_name}" 201 fi 202 203 if ! ip netns add "${!ns_name}"; then 204 echo "Failed to create namespace $ns_name" 205 cleanup_ns "${ns_list[@]}" 206 return $ksft_skip 207 fi 208 ip -n "${!ns_name}" link set lo up 209 ns_list+=("${!ns_name}") 210 done 211 NS_LIST+=("${ns_list[@]}") 212} 213 214tc_rule_stats_get() 215{ 216 local dev=$1; shift 217 local pref=$1; shift 218 local dir=${1:-ingress}; shift 219 local selector=${1:-.packets}; shift 220 221 tc -j -s filter show dev $dev $dir pref $pref \ 222 | jq ".[1].options.actions[].stats$selector" 223} 224 225tc_rule_handle_stats_get() 226{ 227 local id=$1; shift 228 local handle=$1; shift 229 local selector=${1:-.packets}; shift 230 local netns=${1:-""}; shift 231 232 tc $netns -j -s filter show $id \ 233 | jq ".[] | select(.options.handle == $handle) | \ 234 .options.actions[0].stats$selector" 235} 236