1#!/bin/bash
  2# SPDX-License-Identifier: GPL-2.0
  3
  4readonly SERVER_MAC="aa:00:00:00:00:02"
  5readonly CLIENT_MAC="aa:00:00:00:00:01"
  6readonly TESTS=("data" "ack" "flags" "tcp" "ip" "large")
  7readonly PROTOS=("ipv4" "ipv6")
  8dev=""
  9test="all"
 10proto="ipv4"
 11
 12run_test() {
 13  local server_pid=0
 14  local exit_code=0
 15  local protocol=$1
 16  local test=$2
 17  local ARGS=( "--${protocol}" "--dmac" "${SERVER_MAC}" \
 18  "--smac" "${CLIENT_MAC}" "--test" "${test}" "--verbose" )
 19
 20  setup_ns
 21  # Each test is run 3 times to deflake, because given the receive timing,
 22  # not all packets that should coalesce will be considered in the same flow
 23  # on every try.
 24  for tries in {1..3}; do
 25    # Actual test starts here
 26    ip netns exec $server_ns ./gro "${ARGS[@]}" "--rx" "--iface" "server" \
 27      1>>log.txt &
 28    server_pid=$!
 29    sleep 0.5  # to allow for socket init
 30    ip netns exec $client_ns ./gro "${ARGS[@]}" "--iface" "client" \
 31      1>>log.txt
 32    wait "${server_pid}"
 33    exit_code=$?
 34    if [[ ${test} == "large" && -n "${KSFT_MACHINE_SLOW}" && \
 35          ${exit_code} -ne 0 ]]; then
 36        echo "Ignoring errors due to slow environment" 1>&2
 37        exit_code=0
 38    fi
 39    if [[ "${exit_code}" -eq 0 ]]; then
 40        break;
 41    fi
 42  done
 43  cleanup_ns
 44  echo ${exit_code}
 45}
 46
 47run_all_tests() {
 48  local failed_tests=()
 49  for proto in "${PROTOS[@]}"; do
 50    for test in "${TESTS[@]}"; do
 51      echo "running test ${proto} ${test}" >&2
 52      exit_code=$(run_test $proto $test)
 53      if [[ "${exit_code}" -ne 0 ]]; then
 54        failed_tests+=("${proto}_${test}")
 55      fi;
 56    done;
 57  done
 58  if [[ ${#failed_tests[@]} -ne 0 ]]; then
 59    echo "failed tests: ${failed_tests[*]}. \
 60    Please see log.txt for more logs"
 61    exit 1
 62  else
 63    echo "All Tests Succeeded!"
 64  fi;
 65}
 66
 67usage() {
 68  echo "Usage: $0 \
 69  [-i <DEV>] \
 70  [-t data|ack|flags|tcp|ip|large] \
 71  [-p <ipv4|ipv6>]" 1>&2;
 72  exit 1;
 73}
 74
 75while getopts "i:t:p:" opt; do
 76  case "${opt}" in
 77    i)
 78      dev="${OPTARG}"
 79      ;;
 80    t)
 81      test="${OPTARG}"
 82      ;;
 83    p)
 84      proto="${OPTARG}"
 85      ;;
 86    *)
 87      usage
 88      ;;
 89  esac
 90done
 91
 92if [ -n "$dev" ]; then
 93	source setup_loopback.sh
 94else
 95	source setup_veth.sh
 96fi
 97
 98setup
 99trap cleanup EXIT
100if [[ "${test}" == "all" ]]; then
101  run_all_tests
102else
103  run_test "${proto}" "${test}"
104fi;