tools/testing/selftests/bpf/test_tcp_check_syncookie.sh

   1 #!/bin/sh
   2 # SPDX-License-Identifier: GPL-2.0
   3 # Copyright (c) 2018 Facebook
   4 # Copyright (c) 2019 Cloudflare
   5
   6 set -eu
   7
   8 wait_for_ip()
   9 {
  10         local _i
  11         printf "Wait for IP %s to become available " "$1"
  12         for _i in $(seq ${MAX_PING_TRIES}); do
  13                 printf "."
  14                 if ns1_exec ping -c 1 -W 1 "$1" >/dev/null 2>&1; then
  15                         echo " OK"
  16                         return
  17                 fi
  18                 sleep 1
  19         done
  20         echo 1>&2 "ERROR: Timeout waiting for test IP to become available."
  21         exit 1
  22 }
  23
  24 get_prog_id()
  25 {
  26         awk '/ id / {sub(/.* id /, "", $0); print($1)}'
  27 }
  28
  29 ns1_exec()
  30 {
  31         ip netns exec ns1 "$@"
  32 }
  33
  34 setup()
  35 {
  36         ip netns add ns1
  37         ns1_exec ip link set lo up
  38
  39         ns1_exec sysctl -w net.ipv4.tcp_syncookies=2
  40
  41         wait_for_ip 127.0.0.1
  42         wait_for_ip ::1
  43 }
  44
  45 cleanup()
  46 {
  47         ip netns del ns1 2>/dev/null || :
  48 }
  49
  50 main()
  51 {
  52         trap cleanup EXIT 2 3 6 15
  53         setup
  54
  55         printf "Testing clsact..."
  56         ns1_exec tc qdisc add dev "${TEST_IF}" clsact
  57         ns1_exec tc filter add dev "${TEST_IF}" ingress \
  58                 bpf obj "${BPF_PROG_OBJ}" sec "${CLSACT_SECTION}" da
  59
  60         BPF_PROG_ID=$(ns1_exec tc filter show dev "${TEST_IF}" ingress | \
  61                       get_prog_id)
  62         ns1_exec "${PROG}" "${BPF_PROG_ID}"
  63         ns1_exec tc qdisc del dev "${TEST_IF}" clsact
  64
  65         printf "Testing XDP..."
  66         ns1_exec ip link set "${TEST_IF}" xdp \
  67                 object "${BPF_PROG_OBJ}" section "${XDP_SECTION}"
  68         BPF_PROG_ID=$(ns1_exec ip link show "${TEST_IF}" | get_prog_id)
  69         ns1_exec "${PROG}" "${BPF_PROG_ID}"
  70 }
  71
  72 DIR=$(dirname $0)
  73 TEST_IF=lo
  74 MAX_PING_TRIES=5
  75 BPF_PROG_OBJ="${DIR}/test_tcp_check_syncookie_kern.o"
  76 CLSACT_SECTION="clsact/check_syncookie"
  77 XDP_SECTION="xdp/check_syncookie"
  78 BPF_PROG_ID=0
  79 PROG="${DIR}/test_tcp_check_syncookie_user"
  80
  81 main