mirror of https://gitee.com/openkylin/linux.git
401 lines
9.0 KiB
Bash
401 lines
9.0 KiB
Bash
#!/bin/bash
|
|
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
# Uncomment to see generated bytecode
|
|
#VERBOSE=verbose
|
|
|
|
NS1=lwt_ns1
|
|
NS2=lwt_ns2
|
|
VETH0=tst_lwt1a
|
|
VETH1=tst_lwt1b
|
|
VETH2=tst_lwt2a
|
|
VETH3=tst_lwt2b
|
|
IPVETH0="192.168.254.1"
|
|
IPVETH1="192.168.254.2"
|
|
IPVETH1b="192.168.254.3"
|
|
|
|
IPVETH2="192.168.111.1"
|
|
IPVETH3="192.168.111.2"
|
|
|
|
IP_LOCAL="192.168.99.1"
|
|
|
|
TRACE_ROOT=/sys/kernel/debug/tracing
|
|
|
|
function lookup_mac()
|
|
{
|
|
set +x
|
|
if [ ! -z "$2" ]; then
|
|
MAC=$(ip netns exec $2 ip link show $1 | grep ether | awk '{print $2}')
|
|
else
|
|
MAC=$(ip link show $1 | grep ether | awk '{print $2}')
|
|
fi
|
|
MAC="${MAC//:/}"
|
|
echo "0x${MAC:10:2}${MAC:8:2}${MAC:6:2}${MAC:4:2}${MAC:2:2}${MAC:0:2}"
|
|
set -x
|
|
}
|
|
|
|
function cleanup {
|
|
set +ex
|
|
rm test_lwt_bpf.o 2> /dev/null
|
|
ip link del $VETH0 2> /dev/null
|
|
ip link del $VETH1 2> /dev/null
|
|
ip link del $VETH2 2> /dev/null
|
|
ip link del $VETH3 2> /dev/null
|
|
ip netns exec $NS1 killall netserver
|
|
ip netns delete $NS1 2> /dev/null
|
|
ip netns delete $NS2 2> /dev/null
|
|
set -ex
|
|
}
|
|
|
|
function setup_one_veth {
|
|
ip netns add $1
|
|
ip link add $2 type veth peer name $3
|
|
ip link set dev $2 up
|
|
ip addr add $4/24 dev $2
|
|
ip link set $3 netns $1
|
|
ip netns exec $1 ip link set dev $3 up
|
|
ip netns exec $1 ip addr add $5/24 dev $3
|
|
|
|
if [ "$6" ]; then
|
|
ip netns exec $1 ip addr add $6/32 dev $3
|
|
fi
|
|
}
|
|
|
|
function get_trace {
|
|
set +x
|
|
cat ${TRACE_ROOT}/trace | grep -v '^#'
|
|
set -x
|
|
}
|
|
|
|
function cleanup_routes {
|
|
ip route del ${IPVETH1}/32 dev $VETH0 2> /dev/null || true
|
|
ip route del table local local ${IP_LOCAL}/32 dev lo 2> /dev/null || true
|
|
}
|
|
|
|
function install_test {
|
|
cleanup_routes
|
|
cp /dev/null ${TRACE_ROOT}/trace
|
|
|
|
OPTS="encap bpf headroom 14 $1 obj test_lwt_bpf.o section $2 $VERBOSE"
|
|
|
|
if [ "$1" == "in" ]; then
|
|
ip route add table local local ${IP_LOCAL}/32 $OPTS dev lo
|
|
else
|
|
ip route add ${IPVETH1}/32 $OPTS dev $VETH0
|
|
fi
|
|
}
|
|
|
|
function remove_prog {
|
|
if [ "$1" == "in" ]; then
|
|
ip route del table local local ${IP_LOCAL}/32 dev lo
|
|
else
|
|
ip route del ${IPVETH1}/32 dev $VETH0
|
|
fi
|
|
}
|
|
|
|
function filter_trace {
|
|
# Add newline to allow starting EXPECT= variables on newline
|
|
NL=$'\n'
|
|
echo "${NL}$*" | sed -e 's/^.*: : //g'
|
|
}
|
|
|
|
function expect_fail {
|
|
set +x
|
|
echo "FAIL:"
|
|
echo "Expected: $1"
|
|
echo "Got: $2"
|
|
set -x
|
|
exit 1
|
|
}
|
|
|
|
function match_trace {
|
|
set +x
|
|
RET=0
|
|
TRACE=$1
|
|
EXPECT=$2
|
|
GOT="$(filter_trace "$TRACE")"
|
|
|
|
[ "$GOT" != "$EXPECT" ] && {
|
|
expect_fail "$EXPECT" "$GOT"
|
|
RET=1
|
|
}
|
|
set -x
|
|
return $RET
|
|
}
|
|
|
|
function test_start {
|
|
set +x
|
|
echo "----------------------------------------------------------------"
|
|
echo "Starting test: $*"
|
|
echo "----------------------------------------------------------------"
|
|
set -x
|
|
}
|
|
|
|
function failure {
|
|
get_trace
|
|
echo "FAIL: $*"
|
|
exit 1
|
|
}
|
|
|
|
function test_ctx_xmit {
|
|
test_start "test_ctx on lwt xmit"
|
|
install_test xmit test_ctx
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "test_ctx xmit: packets are dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX" || exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
function test_ctx_out {
|
|
test_start "test_ctx on lwt out"
|
|
install_test out test_ctx
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "test_ctx out: packets are dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
len 84 hash 0 protocol 0
|
|
cb 1234 ingress_ifindex 0 ifindex 0
|
|
len 84 hash 0 protocol 0
|
|
cb 1234 ingress_ifindex 0 ifindex 0
|
|
len 84 hash 0 protocol 0
|
|
cb 1234 ingress_ifindex 0 ifindex 0" || exit 1
|
|
remove_prog out
|
|
}
|
|
|
|
function test_ctx_in {
|
|
test_start "test_ctx on lwt in"
|
|
install_test in test_ctx
|
|
ping -c 3 $IP_LOCAL || {
|
|
failure "test_ctx out: packets are dropped"
|
|
}
|
|
# We will both request & reply packets as the packets will
|
|
# be from $IP_LOCAL => $IP_LOCAL
|
|
match_trace "$(get_trace)" "
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1
|
|
len 84 hash 0 protocol 8
|
|
cb 1234 ingress_ifindex 1 ifindex 1" || exit 1
|
|
remove_prog in
|
|
}
|
|
|
|
function test_data {
|
|
test_start "test_data on lwt $1"
|
|
install_test $1 test_data
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "test_data ${1}: packets are dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
src: 1fea8c0 dst: 2fea8c0
|
|
src: 1fea8c0 dst: 2fea8c0
|
|
src: 1fea8c0 dst: 2fea8c0" || exit 1
|
|
remove_prog $1
|
|
}
|
|
|
|
function test_data_in {
|
|
test_start "test_data on lwt in"
|
|
install_test in test_data
|
|
ping -c 3 $IP_LOCAL || {
|
|
failure "test_data in: packets are dropped"
|
|
}
|
|
# We will both request & reply packets as the packets will
|
|
# be from $IP_LOCAL => $IP_LOCAL
|
|
match_trace "$(get_trace)" "
|
|
src: 163a8c0 dst: 163a8c0
|
|
src: 163a8c0 dst: 163a8c0
|
|
src: 163a8c0 dst: 163a8c0
|
|
src: 163a8c0 dst: 163a8c0
|
|
src: 163a8c0 dst: 163a8c0
|
|
src: 163a8c0 dst: 163a8c0" || exit 1
|
|
remove_prog in
|
|
}
|
|
|
|
function test_cb {
|
|
test_start "test_cb on lwt $1"
|
|
install_test $1 test_cb
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "test_cb ${1}: packets are dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0" || exit 1
|
|
remove_prog $1
|
|
}
|
|
|
|
function test_cb_in {
|
|
test_start "test_cb on lwt in"
|
|
install_test in test_cb
|
|
ping -c 3 $IP_LOCAL || {
|
|
failure "test_cb in: packets are dropped"
|
|
}
|
|
# We will both request & reply packets as the packets will
|
|
# be from $IP_LOCAL => $IP_LOCAL
|
|
match_trace "$(get_trace)" "
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0
|
|
cb0: 0 cb1: 0 cb2: 0
|
|
cb3: 0 cb4: 0" || exit 1
|
|
remove_prog in
|
|
}
|
|
|
|
function test_drop_all {
|
|
test_start "test_drop_all on lwt $1"
|
|
install_test $1 drop_all
|
|
ping -c 3 $IPVETH1 && {
|
|
failure "test_drop_all ${1}: Unexpected success of ping"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
dropping with: 2
|
|
dropping with: 2
|
|
dropping with: 2" || exit 1
|
|
remove_prog $1
|
|
}
|
|
|
|
function test_drop_all_in {
|
|
test_start "test_drop_all on lwt in"
|
|
install_test in drop_all
|
|
ping -c 3 $IP_LOCAL && {
|
|
failure "test_drop_all in: Unexpected success of ping"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
dropping with: 2
|
|
dropping with: 2
|
|
dropping with: 2" || exit 1
|
|
remove_prog in
|
|
}
|
|
|
|
function test_push_ll_and_redirect {
|
|
test_start "test_push_ll_and_redirect on lwt xmit"
|
|
install_test xmit push_ll_and_redirect
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "Redirected packets appear to be dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
redirected to $DST_IFINDEX
|
|
redirected to $DST_IFINDEX
|
|
redirected to $DST_IFINDEX" || exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
function test_no_l2_and_redirect {
|
|
test_start "test_no_l2_and_redirect on lwt xmit"
|
|
install_test xmit fill_garbage_and_redirect
|
|
ping -c 3 $IPVETH1 && {
|
|
failure "Unexpected success despite lack of L2 header"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
redirected to $DST_IFINDEX
|
|
redirected to $DST_IFINDEX
|
|
redirected to $DST_IFINDEX" || exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
function test_rewrite {
|
|
test_start "test_rewrite on lwt xmit"
|
|
install_test xmit test_rewrite
|
|
ping -c 3 $IPVETH1 || {
|
|
failure "Rewritten packets appear to be dropped"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
out: rewriting from 2fea8c0 to 3fea8c0
|
|
out: rewriting from 2fea8c0 to 3fea8c0
|
|
out: rewriting from 2fea8c0 to 3fea8c0" || exit 1
|
|
remove_prog out
|
|
}
|
|
|
|
function test_fill_garbage {
|
|
test_start "test_fill_garbage on lwt xmit"
|
|
install_test xmit fill_garbage
|
|
ping -c 3 $IPVETH1 && {
|
|
failure "test_drop_all ${1}: Unexpected success of ping"
|
|
}
|
|
match_trace "$(get_trace)" "
|
|
Set initial 96 bytes of header to FF
|
|
Set initial 96 bytes of header to FF
|
|
Set initial 96 bytes of header to FF" || exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
function test_netperf_nop {
|
|
test_start "test_netperf_nop on lwt xmit"
|
|
install_test xmit nop
|
|
netperf -H $IPVETH1 -t TCP_STREAM || {
|
|
failure "packets appear to be dropped"
|
|
}
|
|
match_trace "$(get_trace)" ""|| exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
function test_netperf_redirect {
|
|
test_start "test_netperf_redirect on lwt xmit"
|
|
install_test xmit push_ll_and_redirect_silent
|
|
netperf -H $IPVETH1 -t TCP_STREAM || {
|
|
failure "Rewritten packets appear to be dropped"
|
|
}
|
|
match_trace "$(get_trace)" ""|| exit 1
|
|
remove_prog xmit
|
|
}
|
|
|
|
cleanup
|
|
setup_one_veth $NS1 $VETH0 $VETH1 $IPVETH0 $IPVETH1 $IPVETH1b
|
|
setup_one_veth $NS2 $VETH2 $VETH3 $IPVETH2 $IPVETH3
|
|
ip netns exec $NS1 netserver
|
|
echo 1 > ${TRACE_ROOT}/tracing_on
|
|
|
|
DST_MAC=$(lookup_mac $VETH1 $NS1)
|
|
SRC_MAC=$(lookup_mac $VETH0)
|
|
DST_IFINDEX=$(cat /sys/class/net/$VETH0/ifindex)
|
|
|
|
CLANG_OPTS="-O2 -target bpf -I ../include/"
|
|
CLANG_OPTS+=" -DSRC_MAC=$SRC_MAC -DDST_MAC=$DST_MAC -DDST_IFINDEX=$DST_IFINDEX"
|
|
clang $CLANG_OPTS -c test_lwt_bpf.c -o test_lwt_bpf.o
|
|
|
|
test_ctx_xmit
|
|
test_ctx_out
|
|
test_ctx_in
|
|
test_data "xmit"
|
|
test_data "out"
|
|
test_data_in
|
|
test_cb "xmit"
|
|
test_cb "out"
|
|
test_cb_in
|
|
test_drop_all "xmit"
|
|
test_drop_all "out"
|
|
test_drop_all_in
|
|
test_rewrite
|
|
test_push_ll_and_redirect
|
|
test_no_l2_and_redirect
|
|
test_fill_garbage
|
|
test_netperf_nop
|
|
test_netperf_redirect
|
|
|
|
cleanup
|
|
echo 0 > ${TRACE_ROOT}/tracing_on
|
|
exit 0
|