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