name : test_skb_cgroup_id.sh #!/bin/sh # SPDX-License-Identifier: GPL-2.0 # Copyright (c) 2018 Facebook set -eu wait_for_ip() { local _i echo -n "Wait for testing link-local IP to become available " for _i in $(seq ${MAX_PING_TRIES}); do echo -n "." if $PING6 -c 1 -W 1 ff02::1%${TEST_IF} >/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 } setup() { # Create testing interfaces not to interfere with current environment. ip link add dev ${TEST_IF} type veth peer name ${TEST_IF_PEER} ip link set ${TEST_IF} up ip link set ${TEST_IF_PEER} up wait_for_ip tc qdisc add dev ${TEST_IF} clsact tc filter add dev ${TEST_IF} egress bpf obj ${BPF_PROG_OBJ} \ sec ${BPF_PROG_SECTION} da BPF_PROG_ID=$(tc filter show dev ${TEST_IF} egress | \ awk '/ id / {sub(/.* id /, "", $0); print($1)}') } cleanup() { ip link del ${TEST_IF} 2>/dev/null || : ip link del ${TEST_IF_PEER} 2>/dev/null || : } main() { trap cleanup EXIT 2 3 6 15 setup ${PROG} ${TEST_IF} ${BPF_PROG_ID} } DIR=$(dirname $0) TEST_IF="test_cgid_1" TEST_IF_PEER="test_cgid_2" MAX_PING_TRIES=5 BPF_PROG_OBJ="${DIR}/test_skb_cgroup_id_kern.bpf.o" BPF_PROG_SECTION="cgroup_id_logger" BPF_PROG_ID=0 PROG="${DIR}/test_skb_cgroup_id_user" type ping6 >/dev/null 2>&1 && PING6="ping6" || PING6="ping -6" main