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 } 73