tools/testing/selftests/bpf/prog_tests/sockopt_qos_to_cc.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2021 Facebook */
   3 #include <test_progs.h>
   4 #include <netinet/tcp.h>
   5 #include "sockopt_qos_to_cc.skel.h"
   6
   7 static void run_setsockopt_test(int cg_fd, int sock_fd)
   8 {
   9         socklen_t optlen;
  10         char cc[16]; /* TCP_CA_NAME_MAX */
  11         int buf;
  12         int err = -1;
  13
  14         buf = 0x2D;
  15         err = setsockopt(sock_fd, SOL_IPV6, IPV6_TCLASS, &buf, sizeof(buf));
  16         if (!ASSERT_OK(err, "setsockopt(sock_fd, IPV6_TCLASS)"))
  17                 return;
  18
  19         /* Verify the setsockopt cc change */
  20         optlen = sizeof(cc);
  21         err = getsockopt(sock_fd, SOL_TCP, TCP_CONGESTION, cc, &optlen);
  22         if (!ASSERT_OK(err, "getsockopt(sock_fd, TCP_CONGESTION)"))
  23                 return;
  24
  25         if (!ASSERT_STREQ(cc, "reno", "getsockopt(sock_fd, TCP_CONGESTION)"))
  26                 return;
  27 }
  28
  29 void test_sockopt_qos_to_cc(void)
  30 {
  31         struct sockopt_qos_to_cc *skel;
  32         char cc_cubic[16] = "cubic"; /* TCP_CA_NAME_MAX */
  33         int cg_fd = -1;
  34         int sock_fd = -1;
  35         int err;
  36
  37         cg_fd = test__join_cgroup("/sockopt_qos_to_cc");
  38         if (!ASSERT_GE(cg_fd, 0, "cg-join(sockopt_qos_to_cc)"))
  39                 return;
  40
  41         skel = sockopt_qos_to_cc__open_and_load();
  42         if (!ASSERT_OK_PTR(skel, "skel"))
  43                 goto done;
  44
  45         skel->bss->page_size = sysconf(_SC_PAGESIZE);
  46
  47         sock_fd = socket(AF_INET6, SOCK_STREAM, 0);
  48         if (!ASSERT_GE(sock_fd, 0, "v6 socket open"))
  49                 goto done;
  50
  51         err = setsockopt(sock_fd, SOL_TCP, TCP_CONGESTION, &cc_cubic,
  52                          sizeof(cc_cubic));
  53         if (!ASSERT_OK(err, "setsockopt(sock_fd, TCP_CONGESTION)"))
  54                 goto done;
  55
  56         skel->links.sockopt_qos_to_cc =
  57                 bpf_program__attach_cgroup(skel->progs.sockopt_qos_to_cc,
  58                                            cg_fd);
  59         if (!ASSERT_OK_PTR(skel->links.sockopt_qos_to_cc,
  60                            "prog_attach(sockopt_qos_to_cc)"))
  61                 goto done;
  62
  63         run_setsockopt_test(cg_fd, sock_fd);
  64
  65 done:
  66         if (sock_fd != -1)
  67                 close(sock_fd);
  68         if (cg_fd != -1)
  69                 close(cg_fd);
  70         /* destroy can take null and error pointer */
  71         sockopt_qos_to_cc__destroy(skel);
  72 }