Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 | // SPDX-License-Identifier: GPL-2.0 /* Copyright (c) 2020, Tessares SA. */ /* Copyright (c) 2022, SUSE. */ #include <test_progs.h> #include "cgroup_helpers.h" #include "network_helpers.h" #include "mptcp_sock.skel.h" #ifndef TCP_CA_NAME_MAX #define TCP_CA_NAME_MAX 16 #endif struct mptcp_storage { __u32 invoked; __u32 is_mptcp; struct sock *sk; __u32 token; struct sock *first; char ca_name[TCP_CA_NAME_MAX]; }; static int verify_tsk(int map_fd, int client_fd) { int err, cfd = client_fd; struct mptcp_storage val; err = bpf_map_lookup_elem(map_fd, &cfd, &val); if (!ASSERT_OK(err, "bpf_map_lookup_elem")) return err; if (!ASSERT_EQ(val.invoked, 1, "unexpected invoked count")) err++; if (!ASSERT_EQ(val.is_mptcp, 0, "unexpected is_mptcp")) err++; return err; } static void get_msk_ca_name(char ca_name[]) { size_t len; int fd; fd = open("/proc/sys/net/ipv4/tcp_congestion_control", O_RDONLY); if (!ASSERT_GE(fd, 0, "failed to open tcp_congestion_control")) return; len = read(fd, ca_name, TCP_CA_NAME_MAX); if (!ASSERT_GT(len, 0, "failed to read ca_name")) goto err; if (len > 0 && ca_name[len - 1] == '\n') ca_name[len - 1] = '\0'; err: close(fd); } static int verify_msk(int map_fd, int client_fd, __u32 token) { char ca_name[TCP_CA_NAME_MAX]; int err, cfd = client_fd; struct mptcp_storage val; if (!ASSERT_GT(token, 0, "invalid token")) return -1; get_msk_ca_name(ca_name); err = bpf_map_lookup_elem(map_fd, &cfd, &val); if (!ASSERT_OK(err, "bpf_map_lookup_elem")) return err; if (!ASSERT_EQ(val.invoked, 1, "unexpected invoked count")) err++; if (!ASSERT_EQ(val.is_mptcp, 1, "unexpected is_mptcp")) err++; if (!ASSERT_EQ(val.token, token, "unexpected token")) err++; if (!ASSERT_EQ(val.first, val.sk, "unexpected first")) err++; if (!ASSERT_STRNEQ(val.ca_name, ca_name, TCP_CA_NAME_MAX, "unexpected ca_name")) err++; return err; } static int run_test(int cgroup_fd, int server_fd, bool is_mptcp) { int client_fd, prog_fd, map_fd, err; struct mptcp_sock *sock_skel; sock_skel = mptcp_sock__open_and_load(); if (!ASSERT_OK_PTR(sock_skel, "skel_open_load")) return -EIO; err = mptcp_sock__attach(sock_skel); if (!ASSERT_OK(err, "skel_attach")) goto out; prog_fd = bpf_program__fd(sock_skel->progs._sockops); if (!ASSERT_GE(prog_fd, 0, "bpf_program__fd")) { err = -EIO; goto out; } map_fd = bpf_map__fd(sock_skel->maps.socket_storage_map); if (!ASSERT_GE(map_fd, 0, "bpf_map__fd")) { err = -EIO; goto out; } err = bpf_prog_attach(prog_fd, cgroup_fd, BPF_CGROUP_SOCK_OPS, 0); if (!ASSERT_OK(err, "bpf_prog_attach")) goto out; client_fd = connect_to_fd(server_fd, 0); if (!ASSERT_GE(client_fd, 0, "connect to fd")) { err = -EIO; goto out; } err += is_mptcp ? verify_msk(map_fd, client_fd, sock_skel->bss->token) : verify_tsk(map_fd, client_fd); close(client_fd); out: mptcp_sock__destroy(sock_skel); return err; } static void test_base(void) { int server_fd, cgroup_fd; cgroup_fd = test__join_cgroup("/mptcp"); if (!ASSERT_GE(cgroup_fd, 0, "test__join_cgroup")) return; /* without MPTCP */ server_fd = start_server(AF_INET, SOCK_STREAM, NULL, 0, 0); if (!ASSERT_GE(server_fd, 0, "start_server")) goto with_mptcp; ASSERT_OK(run_test(cgroup_fd, server_fd, false), "run_test tcp"); close(server_fd); with_mptcp: /* with MPTCP */ server_fd = start_mptcp_server(AF_INET, NULL, 0, 0); if (!ASSERT_GE(server_fd, 0, "start_mptcp_server")) goto close_cgroup_fd; ASSERT_OK(run_test(cgroup_fd, server_fd, true), "run_test mptcp"); close(server_fd); close_cgroup_fd: close(cgroup_fd); } void test_mptcp(void) { if (test__start_subtest("base")) test_base(); } |