bpf/prog_tests/sockopt_qos_to_cc.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2021 Facebook */
0003 #include <test_progs.h>
0004 #include <netinet/tcp.h>
0005 #include "sockopt_qos_to_cc.skel.h"
0006
0007 static void run_setsockopt_test(int cg_fd, int sock_fd)
0008 {
0009     socklen_t optlen;
0010     char cc[16]; /* TCP_CA_NAME_MAX */
0011     int buf;
0012     int err = -1;
0013
0014     buf = 0x2D;
0015     err = setsockopt(sock_fd, SOL_IPV6, IPV6_TCLASS, &buf, sizeof(buf));
0016     if (!ASSERT_OK(err, "setsockopt(sock_fd, IPV6_TCLASS)"))
0017         return;
0018
0019     /* Verify the setsockopt cc change */
0020     optlen = sizeof(cc);
0021     err = getsockopt(sock_fd, SOL_TCP, TCP_CONGESTION, cc, &optlen);
0022     if (!ASSERT_OK(err, "getsockopt(sock_fd, TCP_CONGESTION)"))
0023         return;
0024
0025     if (!ASSERT_STREQ(cc, "reno", "getsockopt(sock_fd, TCP_CONGESTION)"))
0026         return;
0027 }
0028
0029 void test_sockopt_qos_to_cc(void)
0030 {
0031     struct sockopt_qos_to_cc *skel;
0032     char cc_cubic[16] = "cubic"; /* TCP_CA_NAME_MAX */
0033     int cg_fd = -1;
0034     int sock_fd = -1;
0035     int err;
0036
0037     cg_fd = test__join_cgroup("/sockopt_qos_to_cc");
0038     if (!ASSERT_GE(cg_fd, 0, "cg-join(sockopt_qos_to_cc)"))
0039         return;
0040
0041     skel = sockopt_qos_to_cc__open_and_load();
0042     if (!ASSERT_OK_PTR(skel, "skel"))
0043         goto done;
0044
0045     sock_fd = socket(AF_INET6, SOCK_STREAM, 0);
0046     if (!ASSERT_GE(sock_fd, 0, "v6 socket open"))
0047         goto done;
0048
0049     err = setsockopt(sock_fd, SOL_TCP, TCP_CONGESTION, &cc_cubic,
0050              sizeof(cc_cubic));
0051     if (!ASSERT_OK(err, "setsockopt(sock_fd, TCP_CONGESTION)"))
0052         goto done;
0053
0054     skel->links.sockopt_qos_to_cc =
0055         bpf_program__attach_cgroup(skel->progs.sockopt_qos_to_cc,
0056                        cg_fd);
0057     if (!ASSERT_OK_PTR(skel->links.sockopt_qos_to_cc,
0058                "prog_attach(sockopt_qos_to_cc)"))
0059         goto done;
0060
0061     run_setsockopt_test(cg_fd, sock_fd);
0062
0063 done:
0064     if (sock_fd != -1)
0065         close(sock_fd);
0066     if (cg_fd != -1)
0067         close(cg_fd);
0068     /* destroy can take null and error pointer */
0069     sockopt_qos_to_cc__destroy(skel);
0070 }