1#!/bin/sh
2# SPDX-License-Identifier: GPL-2.0
3# Copyright (c) 2018 Facebook
4# Copyright (c) 2019 Cloudflare
5
6set -eu
7
8wait_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
24get_prog_id()
25{
26	awk '/ id / {sub(/.* id /, "", $0); print($1)}'
27}
28
29ns1_exec()
30{
31	ip netns exec ns1 "$@"
32}
33
34setup()
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	ns1_exec sysctl -w net.ipv4.tcp_window_scaling=0
41	ns1_exec sysctl -w net.ipv4.tcp_timestamps=0
42	ns1_exec sysctl -w net.ipv4.tcp_sack=0
43
44	wait_for_ip 127.0.0.1
45	wait_for_ip ::1
46}
47
48cleanup()
49{
50	ip netns del ns1 2>/dev/null || :
51}
52
53main()
54{
55	trap cleanup EXIT 2 3 6 15
56	setup
57
58	printf "Testing clsact..."
59	ns1_exec tc qdisc add dev "${TEST_IF}" clsact
60	ns1_exec tc filter add dev "${TEST_IF}" ingress \
61		bpf obj "${BPF_PROG_OBJ}" sec "${CLSACT_SECTION}" da
62
63	BPF_PROG_ID=$(ns1_exec tc filter show dev "${TEST_IF}" ingress | \
64		      get_prog_id)
65	ns1_exec "${PROG}" "${BPF_PROG_ID}"
66	ns1_exec tc qdisc del dev "${TEST_IF}" clsact
67
68	printf "Testing XDP..."
69	ns1_exec ip link set "${TEST_IF}" xdp \
70		object "${BPF_PROG_OBJ}" section "${XDP_SECTION}"
71	BPF_PROG_ID=$(ns1_exec ip link show "${TEST_IF}" | get_prog_id)
72	ns1_exec "${PROG}" "${BPF_PROG_ID}"
73}
74
75DIR=$(dirname $0)
76TEST_IF=lo
77MAX_PING_TRIES=5
78BPF_PROG_OBJ="${DIR}/test_tcp_check_syncookie_kern.o"
79CLSACT_SECTION="tc"
80XDP_SECTION="xdp"
81BPF_PROG_ID=0
82PROG="${DIR}/test_tcp_check_syncookie_user"
83
84main
85