LXR linux/tools/testing/selftests/net/udpgso

   1#!/bin/bash
   2# SPDX-License-Identifier: GPL-2.0
   3#
   4# Run a series of udpgso benchmarks
   5
   6readonly GREEN='\033[0;92m'
   7readonly YELLOW='\033[0;33m'
   8readonly RED='\033[0;31m'
   9readonly NC='\033[0m' # No Color
  10
  11readonly KSFT_PASS=0
  12readonly KSFT_FAIL=1
  13readonly KSFT_SKIP=4
  14
  15num_pass=0
  16num_err=0
  17num_skip=0
  18
  19kselftest_test_exitcode() {
  20        local -r exitcode=$1
  21
  22        if [[ ${exitcode} -eq ${KSFT_PASS} ]]; then
  23                num_pass=$(( $num_pass + 1 ))
  24        elif [[ ${exitcode} -eq ${KSFT_SKIP} ]]; then
  25                num_skip=$(( $num_skip + 1 ))
  26        else
  27                num_err=$(( $num_err + 1 ))
  28        fi
  29}
  30
  31kselftest_exit() {
  32        echo -e "$(basename $0): PASS=${num_pass} SKIP=${num_skip} FAIL=${num_err}"
  33
  34        if [[ $num_err -ne 0 ]]; then
  35                echo -e "$(basename $0): ${RED}FAIL${NC}"
  36                exit ${KSFT_FAIL}
  37        fi
  38
  39        if [[ $num_skip -ne 0 ]]; then
  40                echo -e "$(basename $0): ${YELLOW}SKIP${NC}"
  41                exit ${KSFT_SKIP}
  42        fi
  43
  44        echo -e "$(basename $0): ${GREEN}PASS${NC}"
  45        exit ${KSFT_PASS}
  46}
  47
  48wake_children() {
  49        local -r jobs="$(jobs -p)"
  50
  51        if [[ "${jobs}" != "" ]]; then
  52                kill -1 ${jobs} 2>/dev/null
  53        fi
  54}
  55trap wake_children EXIT
  56
  57run_one() {
  58        local -r args=$@
  59
  60        ./udpgso_bench_rx &
  61        ./udpgso_bench_rx -t &
  62
  63        ./udpgso_bench_tx ${args}
  64}
  65
  66run_in_netns() {
  67        local -r args=$@
  68
  69        ./in_netns.sh $0 __subprocess ${args}
  70        kselftest_test_exitcode $?
  71}
  72
  73run_udp() {
  74        local -r args=$@
  75
  76        echo "udp"
  77        run_in_netns ${args}
  78
  79        echo "udp gso"
  80        run_in_netns ${args} -S 0
  81
  82        echo "udp gso zerocopy"
  83        run_in_netns ${args} -S 0 -z
  84
  85        echo "udp gso timestamp"
  86        run_in_netns ${args} -S 0 -T
  87
  88        echo "udp gso zerocopy audit"
  89        run_in_netns ${args} -S 0 -z -a
  90
  91        echo "udp gso timestamp audit"
  92        run_in_netns ${args} -S 0 -T -a
  93
  94        echo "udp gso zerocopy timestamp audit"
  95        run_in_netns ${args} -S 0 -T -z -a
  96}
  97
  98run_tcp() {
  99        local -r args=$@
 100
 101        echo "tcp"
 102        run_in_netns ${args} -t
 103
 104        echo "tcp zerocopy"
 105        run_in_netns ${args} -t -z
 106
 107        # excluding for now because test fails intermittently
 108        # add -P option to include poll() to reduce possibility of lost messages
 109        #echo "tcp zerocopy audit"
 110        #run_in_netns ${args} -t -z -P -a
 111}
 112
 113run_all() {
 114        local -r core_args="-l 3"
 115        local -r ipv4_args="${core_args} -4 -D 127.0.0.1"
 116        local -r ipv6_args="${core_args} -6 -D ::1"
 117
 118        echo "ipv4"
 119        run_tcp "${ipv4_args}"
 120        run_udp "${ipv4_args}"
 121
 122        echo "ipv6"
 123        run_tcp "${ipv4_args}"
 124        run_udp "${ipv6_args}"
 125}
 126
 127if [[ $# -eq 0 ]]; then
 128        run_all
 129        kselftest_exit
 130elif [[ $1 == "__subprocess" ]]; then
 131        shift
 132        run_one $@
 133else
 134        run_in_netns $@
 135fi
 136