selftests/net/cmsg_so_mark.sh

0001 #!/bin/bash
0002 # SPDX-License-Identifier: GPL-2.0
0003
0004 NS=ns
0005 IP4=172.16.0.1/24
0006 TGT4=172.16.0.2
0007 IP6=2001:db8:1::1/64
0008 TGT6=2001:db8:1::2
0009 MARK=1000
0010
0011 cleanup()
0012 {
0013     ip netns del $NS
0014 }
0015
0016 trap cleanup EXIT
0017
0018 # Namespaces
0019 ip netns add $NS
0020
0021 ip netns exec $NS sysctl -w net.ipv4.ping_group_range='0 2147483647' > /dev/null
0022
0023 # Connectivity
0024 ip -netns $NS link add type dummy
0025 ip -netns $NS link set dev dummy0 up
0026 ip -netns $NS addr add $IP4 dev dummy0
0027 ip -netns $NS addr add $IP6 dev dummy0
0028
0029 ip -netns $NS rule add fwmark $MARK lookup 300
0030 ip -6 -netns $NS rule add fwmark $MARK lookup 300
0031 ip -netns $NS route add prohibit any table 300
0032 ip -6 -netns $NS route add prohibit any table 300
0033
0034 # Test
0035 BAD=0
0036 TOTAL=0
0037
0038 check_result() {
0039     ((TOTAL++))
0040     if [ $1 -ne $2 ]; then
0041         echo "  Case $3 returned $1, expected $2"
0042         ((BAD++))
0043     fi
0044 }
0045
0046 for ovr in setsock cmsg both; do
0047     for i in 4 6; do
0048         [ $i == 4 ] && TGT=$TGT4 || TGT=$TGT6
0049
0050         for p in u i r; do
0051             [ $p == "u" ] && prot=UDP
0052             [ $p == "i" ] && prot=ICMP
0053             [ $p == "r" ] && prot=RAW
0054
0055             [ $ovr == "setsock" ] && m="-M"
0056             [ $ovr == "cmsg" ]    && m="-m"
0057             [ $ovr == "both" ]    && m="-M $MARK -m"
0058
0059             ip netns exec $NS ./cmsg_sender -$i -p $p $m $((MARK + 1)) $TGT 1234
0060             check_result $? 0 "$prot $ovr - pass"
0061
0062             [ $ovr == "diff" ] && m="-M $((MARK + 1)) -m"
0063
0064             ip netns exec $NS ./cmsg_sender -$i -p $p $m $MARK -s $TGT 1234
0065             check_result $? 1 "$prot $ovr - rejection"
0066         done
0067     done
0068 done
0069
0070 # Summary
0071 if [ $BAD -ne 0 ]; then
0072     echo "FAIL - $BAD/$TOTAL cases failed"
0073     exit 1
0074 else
0075     echo "OK"
0076     exit 0
0077 fi