1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3 4# Uncomment to see generated bytecode 5#VERBOSE=verbose 6 7NS1=lwt_ns1 8NS2=lwt_ns2 9VETH0=tst_lwt1a 10VETH1=tst_lwt1b 11VETH2=tst_lwt2a 12VETH3=tst_lwt2b 13IPVETH0="192.168.254.1" 14IPVETH1="192.168.254.2" 15IPVETH1b="192.168.254.3" 16 17IPVETH2="192.168.111.1" 18IPVETH3="192.168.111.2" 19 20IP_LOCAL="192.168.99.1" 21 22PROG_SRC="test_lwt_bpf.c" 23BPF_PROG="test_lwt_bpf.o" 24TRACE_ROOT=/sys/kernel/tracing 25CONTEXT_INFO=$(cat ${TRACE_ROOT}/trace_options | grep context) 26 27function lookup_mac() 28{ 29 set +x 30 if [ ! -z "$2" ]; then 31 MAC=$(ip netns exec $2 ip link show $1 | grep ether | awk '{print $2}') 32 else 33 MAC=$(ip link show $1 | grep ether | awk '{print $2}') 34 fi 35 MAC="${MAC//:/}" 36 echo "0x${MAC:10:2}${MAC:8:2}${MAC:6:2}${MAC:4:2}${MAC:2:2}${MAC:0:2}" 37 set -x 38} 39 40function cleanup { 41 set +ex 42 rm $BPF_PROG 2> /dev/null 43 ip link del $VETH0 2> /dev/null 44 ip link del $VETH1 2> /dev/null 45 ip link del $VETH2 2> /dev/null 46 ip link del $VETH3 2> /dev/null 47 ip netns exec $NS1 killall netserver 48 ip netns delete $NS1 2> /dev/null 49 ip netns delete $NS2 2> /dev/null 50 set -ex 51} 52 53function setup_one_veth { 54 ip netns add $1 55 ip link add $2 type veth peer name $3 56 ip link set dev $2 up 57 ip addr add $4/24 dev $2 58 ip link set $3 netns $1 59 ip netns exec $1 ip link set dev $3 up 60 ip netns exec $1 ip addr add $5/24 dev $3 61 62 if [ "$6" ]; then 63 ip netns exec $1 ip addr add $6/32 dev $3 64 fi 65} 66 67function get_trace { 68 set +x 69 cat ${TRACE_ROOT}/trace | grep -v '^#' 70 set -x 71} 72 73function cleanup_routes { 74 ip route del ${IPVETH1}/32 dev $VETH0 2> /dev/null || true 75 ip route del table local local ${IP_LOCAL}/32 dev lo 2> /dev/null || true 76} 77 78function install_test { 79 cleanup_routes 80 cp /dev/null ${TRACE_ROOT}/trace 81 82 OPTS="encap bpf headroom 14 $1 obj $BPF_PROG section $2 $VERBOSE" 83 84 if [ "$1" == "in" ]; then 85 ip route add table local local ${IP_LOCAL}/32 $OPTS dev lo 86 else 87 ip route add ${IPVETH1}/32 $OPTS dev $VETH0 88 fi 89} 90 91function remove_prog { 92 if [ "$1" == "in" ]; then 93 ip route del table local local ${IP_LOCAL}/32 dev lo 94 else 95 ip route del ${IPVETH1}/32 dev $VETH0 96 fi 97} 98 99function filter_trace { 100 # Add newline to allow starting EXPECT= variables on newline 101 NL=$'\n' 102 echo "${NL}$*" | sed -e 's/bpf_trace_printk: //g' 103} 104 105function expect_fail { 106 set +x 107 echo "FAIL:" 108 echo "Expected: $1" 109 echo "Got: $2" 110 set -x 111 exit 1 112} 113 114function match_trace { 115 set +x 116 RET=0 117 TRACE=$1 118 EXPECT=$2 119 GOT="$(filter_trace "$TRACE")" 120 121 [ "$GOT" != "$EXPECT" ] && { 122 expect_fail "$EXPECT" "$GOT" 123 RET=1 124 } 125 set -x 126 return $RET 127} 128 129function test_start { 130 set +x 131 echo "----------------------------------------------------------------" 132 echo "Starting test: $*" 133 echo "----------------------------------------------------------------" 134 set -x 135} 136 137function failure { 138 get_trace 139 echo "FAIL: $*" 140 exit 1 141} 142 143function test_ctx_xmit { 144 test_start "test_ctx on lwt xmit" 145 install_test xmit test_ctx 146 ping -c 3 $IPVETH1 || { 147 failure "test_ctx xmit: packets are dropped" 148 } 149 match_trace "$(get_trace)" " 150len 84 hash 0 protocol 8 151cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX 152len 84 hash 0 protocol 8 153cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX 154len 84 hash 0 protocol 8 155cb 1234 ingress_ifindex 0 ifindex $DST_IFINDEX" || exit 1 156 remove_prog xmit 157} 158 159function test_ctx_out { 160 test_start "test_ctx on lwt out" 161 install_test out test_ctx 162 ping -c 3 $IPVETH1 || { 163 failure "test_ctx out: packets are dropped" 164 } 165 match_trace "$(get_trace)" " 166len 84 hash 0 protocol 8 167cb 1234 ingress_ifindex 0 ifindex 0 168len 84 hash 0 protocol 8 169cb 1234 ingress_ifindex 0 ifindex 0 170len 84 hash 0 protocol 8 171cb 1234 ingress_ifindex 0 ifindex 0" || exit 1 172 remove_prog out 173} 174 175function test_ctx_in { 176 test_start "test_ctx on lwt in" 177 install_test in test_ctx 178 ping -c 3 $IP_LOCAL || { 179 failure "test_ctx out: packets are dropped" 180 } 181 # We will both request & reply packets as the packets will 182 # be from $IP_LOCAL => $IP_LOCAL 183 match_trace "$(get_trace)" " 184len 84 hash 0 protocol 8 185cb 1234 ingress_ifindex 1 ifindex 1 186len 84 hash 0 protocol 8 187cb 1234 ingress_ifindex 1 ifindex 1 188len 84 hash 0 protocol 8 189cb 1234 ingress_ifindex 1 ifindex 1 190len 84 hash 0 protocol 8 191cb 1234 ingress_ifindex 1 ifindex 1 192len 84 hash 0 protocol 8 193cb 1234 ingress_ifindex 1 ifindex 1 194len 84 hash 0 protocol 8 195cb 1234 ingress_ifindex 1 ifindex 1" || exit 1 196 remove_prog in 197} 198 199function test_data { 200 test_start "test_data on lwt $1" 201 install_test $1 test_data 202 ping -c 3 $IPVETH1 || { 203 failure "test_data ${1}: packets are dropped" 204 } 205 match_trace "$(get_trace)" " 206src: 1fea8c0 dst: 2fea8c0 207src: 1fea8c0 dst: 2fea8c0 208src: 1fea8c0 dst: 2fea8c0" || exit 1 209 remove_prog $1 210} 211 212function test_data_in { 213 test_start "test_data on lwt in" 214 install_test in test_data 215 ping -c 3 $IP_LOCAL || { 216 failure "test_data in: packets are dropped" 217 } 218 # We will both request & reply packets as the packets will 219 # be from $IP_LOCAL => $IP_LOCAL 220 match_trace "$(get_trace)" " 221src: 163a8c0 dst: 163a8c0 222src: 163a8c0 dst: 163a8c0 223src: 163a8c0 dst: 163a8c0 224src: 163a8c0 dst: 163a8c0 225src: 163a8c0 dst: 163a8c0 226src: 163a8c0 dst: 163a8c0" || exit 1 227 remove_prog in 228} 229 230function test_cb { 231 test_start "test_cb on lwt $1" 232 install_test $1 test_cb 233 ping -c 3 $IPVETH1 || { 234 failure "test_cb ${1}: packets are dropped" 235 } 236 match_trace "$(get_trace)" " 237cb0: 0 cb1: 0 cb2: 0 238cb3: 0 cb4: 0 239cb0: 0 cb1: 0 cb2: 0 240cb3: 0 cb4: 0 241cb0: 0 cb1: 0 cb2: 0 242cb3: 0 cb4: 0" || exit 1 243 remove_prog $1 244} 245 246function test_cb_in { 247 test_start "test_cb on lwt in" 248 install_test in test_cb 249 ping -c 3 $IP_LOCAL || { 250 failure "test_cb in: packets are dropped" 251 } 252 # We will both request & reply packets as the packets will 253 # be from $IP_LOCAL => $IP_LOCAL 254 match_trace "$(get_trace)" " 255cb0: 0 cb1: 0 cb2: 0 256cb3: 0 cb4: 0 257cb0: 0 cb1: 0 cb2: 0 258cb3: 0 cb4: 0 259cb0: 0 cb1: 0 cb2: 0 260cb3: 0 cb4: 0 261cb0: 0 cb1: 0 cb2: 0 262cb3: 0 cb4: 0 263cb0: 0 cb1: 0 cb2: 0 264cb3: 0 cb4: 0 265cb0: 0 cb1: 0 cb2: 0 266cb3: 0 cb4: 0" || exit 1 267 remove_prog in 268} 269 270function test_drop_all { 271 test_start "test_drop_all on lwt $1" 272 install_test $1 drop_all 273 ping -c 3 $IPVETH1 && { 274 failure "test_drop_all ${1}: Unexpected success of ping" 275 } 276 match_trace "$(get_trace)" " 277dropping with: 2 278dropping with: 2 279dropping with: 2" || exit 1 280 remove_prog $1 281} 282 283function test_drop_all_in { 284 test_start "test_drop_all on lwt in" 285 install_test in drop_all 286 ping -c 3 $IP_LOCAL && { 287 failure "test_drop_all in: Unexpected success of ping" 288 } 289 match_trace "$(get_trace)" " 290dropping with: 2 291dropping with: 2 292dropping with: 2" || exit 1 293 remove_prog in 294} 295 296function test_push_ll_and_redirect { 297 test_start "test_push_ll_and_redirect on lwt xmit" 298 install_test xmit push_ll_and_redirect 299 ping -c 3 $IPVETH1 || { 300 failure "Redirected packets appear to be dropped" 301 } 302 match_trace "$(get_trace)" " 303redirected to $DST_IFINDEX 304redirected to $DST_IFINDEX 305redirected to $DST_IFINDEX" || exit 1 306 remove_prog xmit 307} 308 309function test_no_l2_and_redirect { 310 test_start "test_no_l2_and_redirect on lwt xmit" 311 install_test xmit fill_garbage_and_redirect 312 ping -c 3 $IPVETH1 && { 313 failure "Unexpected success despite lack of L2 header" 314 } 315 match_trace "$(get_trace)" " 316redirected to $DST_IFINDEX 317redirected to $DST_IFINDEX 318redirected to $DST_IFINDEX" || exit 1 319 remove_prog xmit 320} 321 322function test_rewrite { 323 test_start "test_rewrite on lwt xmit" 324 install_test xmit test_rewrite 325 ping -c 3 $IPVETH1 || { 326 failure "Rewritten packets appear to be dropped" 327 } 328 match_trace "$(get_trace)" " 329out: rewriting from 2fea8c0 to 3fea8c0 330out: rewriting from 2fea8c0 to 3fea8c0 331out: rewriting from 2fea8c0 to 3fea8c0" || exit 1 332 remove_prog out 333} 334 335function test_fill_garbage { 336 test_start "test_fill_garbage on lwt xmit" 337 install_test xmit fill_garbage 338 ping -c 3 $IPVETH1 && { 339 failure "test_drop_all ${1}: Unexpected success of ping" 340 } 341 match_trace "$(get_trace)" " 342Set initial 96 bytes of header to FF 343Set initial 96 bytes of header to FF 344Set initial 96 bytes of header to FF" || exit 1 345 remove_prog xmit 346} 347 348function test_netperf_nop { 349 test_start "test_netperf_nop on lwt xmit" 350 install_test xmit nop 351 netperf -H $IPVETH1 -t TCP_STREAM || { 352 failure "packets appear to be dropped" 353 } 354 match_trace "$(get_trace)" ""|| exit 1 355 remove_prog xmit 356} 357 358function test_netperf_redirect { 359 test_start "test_netperf_redirect on lwt xmit" 360 install_test xmit push_ll_and_redirect_silent 361 netperf -H $IPVETH1 -t TCP_STREAM || { 362 failure "Rewritten packets appear to be dropped" 363 } 364 match_trace "$(get_trace)" ""|| exit 1 365 remove_prog xmit 366} 367 368cleanup 369setup_one_veth $NS1 $VETH0 $VETH1 $IPVETH0 $IPVETH1 $IPVETH1b 370setup_one_veth $NS2 $VETH2 $VETH3 $IPVETH2 $IPVETH3 371ip netns exec $NS1 netserver 372echo 1 > ${TRACE_ROOT}/tracing_on 373echo nocontext-info > ${TRACE_ROOT}/trace_options 374 375DST_MAC=$(lookup_mac $VETH1 $NS1) 376SRC_MAC=$(lookup_mac $VETH0) 377DST_IFINDEX=$(cat /sys/class/net/$VETH0/ifindex) 378 379CLANG_OPTS="-O2 --target=bpf -I ../include/" 380CLANG_OPTS+=" -DSRC_MAC=$SRC_MAC -DDST_MAC=$DST_MAC -DDST_IFINDEX=$DST_IFINDEX" 381clang $CLANG_OPTS -c $PROG_SRC -o $BPF_PROG 382 383test_ctx_xmit 384test_ctx_out 385test_ctx_in 386test_data "xmit" 387test_data "out" 388test_data_in 389test_cb "xmit" 390test_cb "out" 391test_cb_in 392test_drop_all "xmit" 393test_drop_all "out" 394test_drop_all_in 395test_rewrite 396test_push_ll_and_redirect 397test_no_l2_and_redirect 398test_fill_garbage 399test_netperf_nop 400test_netperf_redirect 401 402cleanup 403echo 0 > ${TRACE_ROOT}/tracing_on 404echo $CONTEXT_INFO > ${TRACE_ROOT}/trace_options 405exit 0 406