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
41	wait_for_ip 127.0.0.1
42	wait_for_ip ::1
43}
44
45cleanup()
46{
47	ip netns del ns1 2>/dev/null || :
48}
49
50main()
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
72DIR=$(dirname $0)
73TEST_IF=lo
74MAX_PING_TRIES=5
75BPF_PROG_OBJ="${DIR}/test_tcp_check_syncookie_kern.o"
76CLSACT_SECTION="clsact/check_syncookie"
77XDP_SECTION="xdp/check_syncookie"
78BPF_PROG_ID=0
79PROG="${DIR}/test_tcp_check_syncookie_user"
80
81main
82