73 lines · c
1// SPDX-License-Identifier: GPL-2.02/* Copyright (c) 2021 Facebook */3#include <test_progs.h>4#include <netinet/tcp.h>5#include "sockopt_qos_to_cc.skel.h"6 7static 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 29void 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 65done: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