#!/bin/bash# SPDX-License-Identifier: GPL-2.0 net_dir=$(dirname "$(readlink -e "${BASH_SOURCE[0]}")")source "$net_dir/lib/sh/defer.sh" ############################################################################### Defines : "${WAIT_TIMEOUT:=20}" # Whether to pause on after a failure.: "${PAUSE_ON_FAIL:=no}" BUSYWAIT_TIMEOUT=$((WAIT_TIMEOUT * 1000)) # ms # Kselftest framework constants.ksft_pass=0ksft_fail=1ksft_xfail=2ksft_skip=4 # namespace list created by setup_nsNS_LIST=() # Exit status to return at the end. Set in case one of the tests fails.EXIT_STATUS=0# Per-test return value. Clear at the beginning of each test.RET=0 ############################################################################### Helpers __ksft_status_merge(){ local a=$1; shift local b=$1; shift local -A weights local weight=0 local i for i in "$@"; do weights[$i]=$((weight++)) done if [[ ${weights[$a]} -ge ${weights[$b]} ]]; then echo "$a" return 0 else echo "$b" return 1 fi} ksft_status_merge(){ local a=$1; shift local b=$1; shift __ksft_status_merge "$a" "$b" \ $ksft_pass $ksft_xfail $ksft_skip $ksft_fail} ksft_exit_status_merge(){ local a=$1; shift local b=$1; shift __ksft_status_merge "$a" "$b" \ $ksft_xfail $ksft_pass $ksft_skip $ksft_fail} timestamp_ms(){ local now local seconds local nanoseconds now=$(date -u +%s:%N) || return seconds=${now%:*} nanoseconds=${now#*:} if [[ $nanoseconds =~ ^[0-9]+$ ]]; then nanoseconds=${nanoseconds:0:9} else nanoseconds=0 fi echo $((seconds * 1000 + 10#$nanoseconds / 1000000))} loopy_wait(){ local sleep_cmd=$1; shift local timeout_ms=$1; shift local start_time local current_time start_time=$(timestamp_ms) || return while true do local out if out=$("$@"); then echo -n "$out" return 0 fi current_time=$(timestamp_ms) || return if ((current_time - start_time > timeout_ms)); then echo -n "$out" return 1 fi $sleep_cmd done} busywait(){ local timeout_ms=$1; shift loopy_wait : "$timeout_ms" "$@"} # timeout in secondsslowwait(){ local timeout_sec=$1; shift loopy_wait "sleep 0.1" "$((timeout_sec * 1000))" "$@"} until_counter_is(){ local expr=$1; shift local current=$("$@") echo $((current)) ((current $expr))} busywait_for_counter(){ local timeout=$1; shift local delta=$1; shift local base=$("$@") busywait "$timeout" until_counter_is ">= $((base + delta))" "$@"} slowwait_for_counter(){ local timeout=$1; shift local delta=$1; shift local base=$("$@") slowwait "$timeout" until_counter_is ">= $((base + delta))" "$@"} # Check for existence of tools which are built as part of selftests# but may also already exist in $PATHcheck_gen_prog(){ local prog_name=$1; shift if ! which $prog_name >/dev/null 2>/dev/null; then PATH=$PWD:$PATH if ! which $prog_name >/dev/null; then echo "'$prog_name' command not found; skipping tests" exit $ksft_skip fi fi} remove_ns_list(){ local item=$1 local ns local ns_list=("${NS_LIST[@]}") NS_LIST=() for ns in "${ns_list[@]}"; do if [ "${ns}" != "${item}" ]; then NS_LIST+=("${ns}") fi done} cleanup_ns(){ local ns="" local ret=0 for ns in "$@"; do [ -z "${ns}" ] && continue ip netns pids "${ns}" 2> /dev/null | xargs -r kill || true ip netns delete "${ns}" &> /dev/null || true if ! busywait $BUSYWAIT_TIMEOUT ip netns list \| grep -vq "^$ns$" &> /dev/null; then echo "Warn: Failed to remove namespace $ns" ret=1 else remove_ns_list "${ns}" fi done return $ret} cleanup_all_ns(){ cleanup_ns "${NS_LIST[@]}"} # setup netns with given names as prefix. e.g# setup_ns local remotesetup_ns(){ local ns_name="" local ns_list=() for ns_name in "$@"; do # avoid conflicts with local var: internal error if [ "${ns_name}" = "ns_name" ]; then echo "Failed to setup namespace '${ns_name}': invalid name" cleanup_ns "${ns_list[@]}" exit $ksft_fail fi # Some test may setup/remove same netns multi times if [ -z "${!ns_name}" ]; then eval "${ns_name}=${ns_name,,}-$(mktemp -u XXXXXX)" else cleanup_ns "${!ns_name}" fi if ! ip netns add "${!ns_name}"; then echo "Failed to create namespace $ns_name" cleanup_ns "${ns_list[@]}" return $ksft_skip fi ip -n "${!ns_name}" link set lo up ip netns exec "${!ns_name}" sysctl -wq net.ipv4.conf.all.rp_filter=0 ip netns exec "${!ns_name}" sysctl -wq net.ipv4.conf.default.rp_filter=0 ns_list+=("${!ns_name}") done NS_LIST+=("${ns_list[@]}")} in_all_ns(){ local ret=0 local ns_list=("${NS_LIST[@]}") for ns in "${ns_list[@]}"; do ip netns exec "${ns}" "$@" (( ret = ret || $? )) done return "${ret}"} # Create netdevsim with given id and net namespace.create_netdevsim() { local id="$1" local ns="$2" modprobe netdevsim &> /dev/null udevadm settle echo "$id 1" | ip netns exec $ns tee /sys/bus/netdevsim/new_device >/dev/null local dev=$(ip netns exec $ns ls /sys/bus/netdevsim/devices/netdevsim$id/net) ip -netns $ns link set dev $dev name nsim$id ip -netns $ns link set dev nsim$id up echo nsim$id} create_netdevsim_port() { local nsim_id="$1" local ns="$2" local port_id="$3" local perm_addr="$4" local orig_dev local new_dev local nsim_path nsim_path="/sys/bus/netdevsim/devices/netdevsim$nsim_id" echo "$port_id $perm_addr" | ip netns exec "$ns" tee "$nsim_path"/new_port > /dev/null || return 1 orig_dev=$(ip netns exec "$ns" find "$nsim_path"/net/ -maxdepth 1 -name 'e*' | tail -n 1) orig_dev=$(basename "$orig_dev") new_dev="nsim${nsim_id}p$port_id" ip -netns "$ns" link set dev "$orig_dev" name "$new_dev" ip -netns "$ns" link set dev "$new_dev" up echo "$new_dev"} # Remove netdevsim with given id.cleanup_netdevsim() { local id="$1" if [ -d "/sys/bus/netdevsim/devices/netdevsim$id/net" ]; then echo "$id" > /sys/bus/netdevsim/del_device fi} tc_rule_stats_get(){ local dev=$1; shift local pref=$1; shift local dir=${1:-ingress}; shift local selector=${1:-.packets}; shift tc -j -s filter show dev $dev $dir pref $pref \ | jq ".[] | select(.options.actions) | .options.actions[].stats$selector"} tc_rule_handle_stats_get(){ local id=$1; shift local handle=$1; shift local selector=${1:-.packets}; shift local netns=${1:-""}; shift tc $netns -j -s filter show $id \ | jq ".[] | select(.options.handle == $handle) | \ .options.actions[0].stats$selector"} # attach a qdisc with two children match/no-match and a flower filter to matchtc_set_flower_counter() { local -r ns=$1 local -r ipver=$2 local -r dev=$3 local -r flower_expr=$4 tc -n $ns qdisc add dev $dev root handle 1: prio bands 2 \ priomap 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 tc -n $ns qdisc add dev $dev parent 1:1 handle 11: pfifo tc -n $ns qdisc add dev $dev parent 1:2 handle 12: pfifo tc -n $ns filter add dev $dev parent 1: protocol ipv$ipver \ flower $flower_expr classid 1:2} tc_get_flower_counter() { local -r ns=$1 local -r dev=$2 tc -n $ns -j -s qdisc show dev $dev handle 12: | jq .[0].packets} ret_set_ksft_status(){ local ksft_status=$1; shift local msg=$1; shift RET=$(ksft_status_merge $RET $ksft_status) if (( $? )); then retmsg=$msg fi} log_test_result(){ local test_name=$1; shift local opt_str=$1; shift local result=$1; shift local retmsg=$1 printf "TEST: %-60s [%s]\n" "$test_name $opt_str" "$result" if [[ $retmsg ]]; then printf "\t%s\n" "$retmsg" fi} pause_on_fail(){ if [[ $PAUSE_ON_FAIL == yes ]]; then echo "Hit enter to continue, 'q' to quit" read a [[ $a == q ]] && exit 1 fi} handle_test_result_pass(){ local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" " OK "} handle_test_result_fail(){ local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" FAIL "$retmsg" pause_on_fail} handle_test_result_xfail(){ local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" XFAIL "$retmsg" pause_on_fail} handle_test_result_skip(){ local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" SKIP "$retmsg"} log_test(){ local test_name=$1 local opt_str=$2 if [[ $# -eq 2 ]]; then opt_str="($opt_str)" fi if ((RET == ksft_pass)); then handle_test_result_pass "$test_name" "$opt_str" elif ((RET == ksft_xfail)); then handle_test_result_xfail "$test_name" "$opt_str" elif ((RET == ksft_skip)); then handle_test_result_skip "$test_name" "$opt_str" else handle_test_result_fail "$test_name" "$opt_str" fi EXIT_STATUS=$(ksft_exit_status_merge $EXIT_STATUS $RET) return $RET} log_test_skip(){ RET=$ksft_skip retmsg= log_test "$@"} log_test_xfail(){ RET=$ksft_xfail retmsg= log_test "$@"} log_info(){ local msg=$1 echo "INFO: $msg"} tests_run(){ local current_test for current_test in ${TESTS:-$ALL_TESTS}; do in_defer_scope \ $current_test done} # Whether FAILs should be interpreted as XFAILs. Internal.FAIL_TO_XFAIL= check_err(){ local err=$1 local msg=$2 if ((err)); then if [[ $FAIL_TO_XFAIL = yes ]]; then ret_set_ksft_status $ksft_xfail "$msg" else ret_set_ksft_status $ksft_fail "$msg" fi fi} check_fail(){ local err=$1 local msg=$2 check_err $((!err)) "$msg"} check_err_fail(){Lines 1–500 of 712. Load more
#!/bin/bash # SPDX-License-Identifier: GPL-2.0 net_dir=$(dirname "$(readlink -e "${BASH_SOURCE[0]}")") source "$net_dir/lib/sh/defer.sh" ############################################################################## # Defines : "${WAIT_TIMEOUT:=20}" # Whether to pause on after a failure. : "${PAUSE_ON_FAIL:=no}" BUSYWAIT_TIMEOUT=$((WAIT_TIMEOUT * 1000)) # ms # Kselftest framework constants. ksft_pass=0 ksft_fail=1 ksft_xfail=2 ksft_skip=4 # namespace list created by setup_ns NS_LIST=() # Exit status to return at the end. Set in case one of the tests fails. EXIT_STATUS=0 # Per-test return value. Clear at the beginning of each test. RET=0 ############################################################################## # Helpers __ksft_status_merge() { local a=$1; shift local b=$1; shift local -A weights local weight=0 local i for i in "$@"; do weights[$i]=$((weight++)) done if [[ ${weights[$a]} -ge ${weights[$b]} ]]; then echo "$a" return 0 else echo "$b" return 1 fi } ksft_status_merge() { local a=$1; shift local b=$1; shift __ksft_status_merge "$a" "$b" \ $ksft_pass $ksft_xfail $ksft_skip $ksft_fail } ksft_exit_status_merge() { local a=$1; shift local b=$1; shift __ksft_status_merge "$a" "$b" \ $ksft_xfail $ksft_pass $ksft_skip $ksft_fail } timestamp_ms() { local now local seconds local nanoseconds now=$(date -u +%s:%N) || return seconds=${now%:*} nanoseconds=${now#*:} if [[ $nanoseconds =~ ^[0-9]+$ ]]; then nanoseconds=${nanoseconds:0:9} else nanoseconds=0 fi echo $((seconds * 1000 + 10#$nanoseconds / 1000000)) } loopy_wait() { local sleep_cmd=$1; shift local timeout_ms=$1; shift local start_time local current_time start_time=$(timestamp_ms) || return while true do local out if out=$("$@"); then echo -n "$out" return 0 fi current_time=$(timestamp_ms) || return if ((current_time - start_time > timeout_ms)); then echo -n "$out" return 1 fi $sleep_cmd done } busywait() { local timeout_ms=$1; shift loopy_wait : "$timeout_ms" "$@" } # timeout in seconds slowwait() { local timeout_sec=$1; shift loopy_wait "sleep 0.1" "$((timeout_sec * 1000))" "$@" } until_counter_is() { local expr=$1; shift local current=$("$@") echo $((current)) ((current $expr)) } busywait_for_counter() { local timeout=$1; shift local delta=$1; shift local base=$("$@") busywait "$timeout" until_counter_is ">= $((base + delta))" "$@" } slowwait_for_counter() { local timeout=$1; shift local delta=$1; shift local base=$("$@") slowwait "$timeout" until_counter_is ">= $((base + delta))" "$@" } # Check for existence of tools which are built as part of selftests # but may also already exist in $PATH check_gen_prog() { local prog_name=$1; shift if ! which $prog_name >/dev/null 2>/dev/null; then PATH=$PWD:$PATH if ! which $prog_name >/dev/null; then echo "'$prog_name' command not found; skipping tests" exit $ksft_skip fi fi } remove_ns_list() { local item=$1 local ns local ns_list=("${NS_LIST[@]}") NS_LIST=() for ns in "${ns_list[@]}"; do if [ "${ns}" != "${item}" ]; then NS_LIST+=("${ns}") fi done } cleanup_ns() { local ns="" local ret=0 for ns in "$@"; do [ -z "${ns}" ] && continue ip netns pids "${ns}" 2> /dev/null | xargs -r kill || true ip netns delete "${ns}" &> /dev/null || true if ! busywait $BUSYWAIT_TIMEOUT ip netns list \| grep -vq "^$ns$" &> /dev/null; then echo "Warn: Failed to remove namespace $ns" ret=1 else remove_ns_list "${ns}" fi done return $ret } cleanup_all_ns() { cleanup_ns "${NS_LIST[@]}" } # setup netns with given names as prefix. e.g # setup_ns local remote setup_ns() { local ns_name="" local ns_list=() for ns_name in "$@"; do # avoid conflicts with local var: internal error if [ "${ns_name}" = "ns_name" ]; then echo "Failed to setup namespace '${ns_name}': invalid name" cleanup_ns "${ns_list[@]}" exit $ksft_fail fi # Some test may setup/remove same netns multi times if [ -z "${!ns_name}" ]; then eval "${ns_name}=${ns_name,,}-$(mktemp -u XXXXXX)" else cleanup_ns "${!ns_name}" fi if ! ip netns add "${!ns_name}"; then echo "Failed to create namespace $ns_name" cleanup_ns "${ns_list[@]}" return $ksft_skip fi ip -n "${!ns_name}" link set lo up ip netns exec "${!ns_name}" sysctl -wq net.ipv4.conf.all.rp_filter=0 ip netns exec "${!ns_name}" sysctl -wq net.ipv4.conf.default.rp_filter=0 ns_list+=("${!ns_name}") done NS_LIST+=("${ns_list[@]}") } in_all_ns() { local ret=0 local ns_list=("${NS_LIST[@]}") for ns in "${ns_list[@]}"; do ip netns exec "${ns}" "$@" (( ret = ret || $? )) done return "${ret}" } # Create netdevsim with given id and net namespace. create_netdevsim() { local id="$1" local ns="$2" modprobe netdevsim &> /dev/null udevadm settle echo "$id 1" | ip netns exec $ns tee /sys/bus/netdevsim/new_device >/dev/null local dev=$(ip netns exec $ns ls /sys/bus/netdevsim/devices/netdevsim$id/net) ip -netns $ns link set dev $dev name nsim$id ip -netns $ns link set dev nsim$id up echo nsim$id } create_netdevsim_port() { local nsim_id="$1" local ns="$2" local port_id="$3" local perm_addr="$4" local orig_dev local new_dev local nsim_path nsim_path="/sys/bus/netdevsim/devices/netdevsim$nsim_id" echo "$port_id $perm_addr" | ip netns exec "$ns" tee "$nsim_path"/new_port > /dev/null || return 1 orig_dev=$(ip netns exec "$ns" find "$nsim_path"/net/ -maxdepth 1 -name 'e*' | tail -n 1) orig_dev=$(basename "$orig_dev") new_dev="nsim${nsim_id}p$port_id" ip -netns "$ns" link set dev "$orig_dev" name "$new_dev" ip -netns "$ns" link set dev "$new_dev" up echo "$new_dev" } # Remove netdevsim with given id. cleanup_netdevsim() { local id="$1" if [ -d "/sys/bus/netdevsim/devices/netdevsim$id/net" ]; then echo "$id" > /sys/bus/netdevsim/del_device fi } tc_rule_stats_get() { local dev=$1; shift local pref=$1; shift local dir=${1:-ingress}; shift local selector=${1:-.packets}; shift tc -j -s filter show dev $dev $dir pref $pref \ | jq ".[] | select(.options.actions) | .options.actions[].stats$selector" } tc_rule_handle_stats_get() { local id=$1; shift local handle=$1; shift local selector=${1:-.packets}; shift local netns=${1:-""}; shift tc $netns -j -s filter show $id \ | jq ".[] | select(.options.handle == $handle) | \ .options.actions[0].stats$selector" } # attach a qdisc with two children match/no-match and a flower filter to match tc_set_flower_counter() { local -r ns=$1 local -r ipver=$2 local -r dev=$3 local -r flower_expr=$4 tc -n $ns qdisc add dev $dev root handle 1: prio bands 2 \ priomap 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 tc -n $ns qdisc add dev $dev parent 1:1 handle 11: pfifo tc -n $ns qdisc add dev $dev parent 1:2 handle 12: pfifo tc -n $ns filter add dev $dev parent 1: protocol ipv$ipver \ flower $flower_expr classid 1:2 } tc_get_flower_counter() { local -r ns=$1 local -r dev=$2 tc -n $ns -j -s qdisc show dev $dev handle 12: | jq .[0].packets } ret_set_ksft_status() { local ksft_status=$1; shift local msg=$1; shift RET=$(ksft_status_merge $RET $ksft_status) if (( $? )); then retmsg=$msg fi } log_test_result() { local test_name=$1; shift local opt_str=$1; shift local result=$1; shift local retmsg=$1 printf "TEST: %-60s [%s]\n" "$test_name $opt_str" "$result" if [[ $retmsg ]]; then printf "\t%s\n" "$retmsg" fi } pause_on_fail() { if [[ $PAUSE_ON_FAIL == yes ]]; then echo "Hit enter to continue, 'q' to quit" read a [[ $a == q ]] && exit 1 fi } handle_test_result_pass() { local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" " OK " } handle_test_result_fail() { local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" FAIL "$retmsg" pause_on_fail } handle_test_result_xfail() { local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" XFAIL "$retmsg" pause_on_fail } handle_test_result_skip() { local test_name=$1; shift local opt_str=$1; shift log_test_result "$test_name" "$opt_str" SKIP "$retmsg" } log_test() { local test_name=$1 local opt_str=$2 if [[ $# -eq 2 ]]; then opt_str="($opt_str)" fi if ((RET == ksft_pass)); then handle_test_result_pass "$test_name" "$opt_str" elif ((RET == ksft_xfail)); then handle_test_result_xfail "$test_name" "$opt_str" elif ((RET == ksft_skip)); then handle_test_result_skip "$test_name" "$opt_str" else handle_test_result_fail "$test_name" "$opt_str" fi EXIT_STATUS=$(ksft_exit_status_merge $EXIT_STATUS $RET) return $RET } log_test_skip() { RET=$ksft_skip retmsg= log_test "$@" } log_test_xfail() { RET=$ksft_xfail retmsg= log_test "$@" } log_info() { local msg=$1 echo "INFO: $msg" } tests_run() { local current_test for current_test in ${TESTS:-$ALL_TESTS}; do in_defer_scope \ $current_test done } # Whether FAILs should be interpreted as XFAILs. Internal. FAIL_TO_XFAIL= check_err() { local err=$1 local msg=$2 if ((err)); then if [[ $FAIL_TO_XFAIL = yes ]]; then ret_set_ksft_status $ksft_xfail "$msg" else ret_set_ksft_status $ksft_fail "$msg" fi fi } check_fail() { local err=$1 local msg=$2 check_err $((!err)) "$msg" } check_err_fail() {