selftests: net: test IPV6_TCLASS

Test setting IPV6_TCLASS via setsockopt and cmsg
across socket types.

Output without the kernel support (this series):

  Case TCLASS ICMP cmsg - packet data returned 1, expected 0
  Case TCLASS ICMP cmsg - rejection returned 0, expected 1
  Case TCLASS ICMP diff - pass returned 1, expected 0
  Case TCLASS ICMP diff - packet data returned 1, expected 0
  Case TCLASS ICMP diff - rejection returned 0, expected 1

Signed-off-by: Jakub Kicinski <kuba@kernel.org>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/tools/testing/selftests/net/cmsg_ipv6.sh b/tools/testing/selftests/net/cmsg_ipv6.sh
index fb5a8ab..f7bb6ce 100755 (executable)
--- a/tools/testing/selftests/net/cmsg_ipv6.sh
+++ b/tools/testing/selftests/net/cmsg_ipv6.sh
@@ -1,12 +1,16 @@
 #!/bin/bash
 # SPDX-License-Identifier: GPL-2.0
 
+ksft_skip=4
+
 NS=ns
 IP6=2001:db8:1::1/64
 TGT6=2001:db8:1::2
+TMPF=`mktemp`
 
 cleanup()
 {
+    rm -f $TMPF
     ip netns del $NS
 }
 
@@ -14,6 +18,12 @@ trap cleanup EXIT
 
 NSEXE="ip netns exec $NS"
 
+tcpdump -h | grep immediate-mode >> /dev/null
+if [ $? -ne 0 ]; then
+    echo "SKIP - tcpdump with --immediate-mode option required"
+    exit $ksft_skip
+fi
+
 # Namespaces
 ip netns add $NS
 
@@ -55,6 +65,47 @@ for ovr in setsock cmsg both diff; do
     done
 done
 
+# IPV6_TCLASS
+TOS=0x10
+TOS2=0x20
+
+ip -6 -netns $NS rule add tos $TOS lookup 300
+ip -6 -netns $NS route add table 300 prohibit any
+
+for ovr in setsock cmsg both diff; do
+    for p in u i r; do
+       [ $p == "u" ] && prot=UDP
+       [ $p == "i" ] && prot=ICMP
+       [ $p == "r" ] && prot=RAW
+
+       [ $ovr == "setsock" ] && m="-C"
+       [ $ovr == "cmsg" ]    && m="-c"
+       [ $ovr == "both" ]    && m="-C $((TOS2)) -c"
+       [ $ovr == "diff" ]    && m="-C $((TOS )) -c"
+
+       $NSEXE nohup tcpdump --immediate-mode -p -ni dummy0 -w $TMPF -c 4 2> /dev/null &
+       BG=$!
+       sleep 0.05
+
+       $NSEXE ./cmsg_sender -6 -p $p $m $((TOS2)) $TGT6 1234
+       check_result $? 0 "TCLASS $prot $ovr - pass"
+
+       while [ -d /proc/$BG ]; do
+           $NSEXE ./cmsg_sender -6 -p u $TGT6 1234
+       done
+
+       tcpdump -r $TMPF -v 2>&1 | grep "class $TOS2" >> /dev/null
+       check_result $? 0 "TCLASS $prot $ovr - packet data"
+       rm $TMPF
+
+       [ $ovr == "both" ]    && m="-C $((TOS )) -c"
+       [ $ovr == "diff" ]    && m="-C $((TOS2)) -c"
+
+       $NSEXE ./cmsg_sender -6 -p $p $m $((TOS)) -s $TGT6 1234
+       check_result $? 1 "TCLASS $prot $ovr - rejection"
+    done
+done
+
 # Summary
 if [ $BAD -ne 0 ]; then
     echo "FAIL - $BAD/$TOTAL cases failed"

diff --git a/tools/testing/selftests/net/cmsg_sender.c b/tools/testing/selftests/net/cmsg_sender.c
index 844ca61..4033cf9 100644 (file)
--- a/tools/testing/selftests/net/cmsg_sender.c
+++ b/tools/testing/selftests/net/cmsg_sender.c
@@ -46,6 +46,7 @@ struct options {
        struct {
                unsigned int mark;
                unsigned int dontfrag;
+               unsigned int tclass;
        } sockopt;
        struct {
                unsigned int family;
@@ -62,6 +63,7 @@ struct options {
        } ts;
        struct {
                struct option_cmsg_u32 dontfrag;
+               struct option_cmsg_u32 tclass;
        } v6;
 } opt = {
        .size = 13,
@@ -91,6 +93,8 @@ static void __attribute__((noreturn)) cs_usage(const char *bin)
               "\t\t-t      Enable time stamp reporting\n"
               "\t\t-f val  Set don't fragment via cmsg\n"
               "\t\t-F val  Set don't fragment via setsockopt\n"
+              "\t\t-c val  Set TCLASS via cmsg\n"
+              "\t\t-C val  Set TCLASS via setsockopt\n"
               "");
        exit(ERN_HELP);
 }
@@ -99,7 +103,7 @@ static void cs_parse_args(int argc, char *argv[])
 {
        char o;
 
-       while ((o = getopt(argc, argv, "46sS:p:m:M:d:tf:F:")) != -1) {
+       while ((o = getopt(argc, argv, "46sS:p:m:M:d:tf:F:c:C:")) != -1) {
                switch (o) {
                case 's':
                        opt.silent_send = true;
@@ -147,6 +151,13 @@ static void cs_parse_args(int argc, char *argv[])
                case 'F':
                        opt.sockopt.dontfrag = atoi(optarg);
                        break;
+               case 'c':
+                       opt.v6.tclass.ena = true;
+                       opt.v6.tclass.val = atoi(optarg);
+                       break;
+               case 'C':
+                       opt.sockopt.tclass = atoi(optarg);
+                       break;
                }
        }
 
@@ -202,6 +213,8 @@ cs_write_cmsg(int fd, struct msghdr *msg, char *cbuf, size_t cbuf_sz)
                          SOL_SOCKET, SO_MARK, &opt.mark);
        ca_write_cmsg_u32(cbuf, cbuf_sz, &cmsg_len,
                          SOL_IPV6, IPV6_DONTFRAG, &opt.v6.dontfrag);
+       ca_write_cmsg_u32(cbuf, cbuf_sz, &cmsg_len,
+                         SOL_IPV6, IPV6_TCLASS, &opt.v6.tclass);
 
        if (opt.txtime.ena) {
                struct sock_txtime so_txtime = {
@@ -343,6 +356,10 @@ static void ca_set_sockopts(int fd)
            setsockopt(fd, SOL_IPV6, IPV6_DONTFRAG,
                       &opt.sockopt.dontfrag, sizeof(opt.sockopt.dontfrag)))
                error(ERN_SOCKOPT, errno, "setsockopt IPV6_DONTFRAG");
+       if (opt.sockopt.tclass &&
+           setsockopt(fd, SOL_IPV6, IPV6_TCLASS,
+                      &opt.sockopt.tclass, sizeof(opt.sockopt.tclass)))
+               error(ERN_SOCKOPT, errno, "setsockopt IPV6_TCLASS");
 }
 
 int main(int argc, char *argv[])