1 // SPDX-License-Identifier: GPL-2.0 2 #include <test_progs.h> 3 #include <network_helpers.h> 4 5 #include "test_tcpbpf.h" 6 #include "test_tcpbpf_kern.skel.h" 7 8 #define LO_ADDR6 "::1" 9 #define CG_NAME "/tcpbpf-user-test" 10 11 static __u32 duration; 12 13 static void verify_result(struct tcpbpf_globals *result) 14 { 15 __u32 expected_events = ((1 << BPF_SOCK_OPS_TIMEOUT_INIT) | 16 (1 << BPF_SOCK_OPS_RWND_INIT) | 17 (1 << BPF_SOCK_OPS_TCP_CONNECT_CB) | 18 (1 << BPF_SOCK_OPS_ACTIVE_ESTABLISHED_CB) | 19 (1 << BPF_SOCK_OPS_PASSIVE_ESTABLISHED_CB) | 20 (1 << BPF_SOCK_OPS_NEEDS_ECN) | 21 (1 << BPF_SOCK_OPS_STATE_CB) | 22 (1 << BPF_SOCK_OPS_TCP_LISTEN_CB)); 23 24 /* check global map */ 25 CHECK(expected_events != result->event_map, "event_map", 26 "unexpected event_map: actual 0x%08x != expected 0x%08x\n", 27 result->event_map, expected_events); 28 29 ASSERT_EQ(result->bytes_received, 501, "bytes_received"); 30 ASSERT_EQ(result->bytes_acked, 1002, "bytes_acked"); 31 ASSERT_EQ(result->data_segs_in, 1, "data_segs_in"); 32 ASSERT_EQ(result->data_segs_out, 1, "data_segs_out"); 33 ASSERT_EQ(result->bad_cb_test_rv, 0x80, "bad_cb_test_rv"); 34 ASSERT_EQ(result->good_cb_test_rv, 0, "good_cb_test_rv"); 35 ASSERT_EQ(result->num_listen, 1, "num_listen"); 36 37 /* 3 comes from one listening socket + both ends of the connection */ 38 ASSERT_EQ(result->num_close_events, 3, "num_close_events"); 39 40 /* check setsockopt for SAVE_SYN */ 41 ASSERT_EQ(result->tcp_save_syn, 0, "tcp_save_syn"); 42 43 /* check getsockopt for SAVED_SYN */ 44 ASSERT_EQ(result->tcp_saved_syn, 1, "tcp_saved_syn"); 45 } 46 47 static void run_test(struct tcpbpf_globals *result) 48 { 49 int listen_fd = -1, cli_fd = -1, accept_fd = -1; 50 char buf[1000]; 51 int err = -1; 52 int i, rv; 53 54 listen_fd = start_server(AF_INET6, SOCK_STREAM, LO_ADDR6, 0, 0); 55 if (CHECK(listen_fd == -1, "start_server", "listen_fd:%d errno:%d\n", 56 listen_fd, errno)) 57 goto done; 58 59 cli_fd = connect_to_fd(listen_fd, 0); 60 if (CHECK(cli_fd == -1, "connect_to_fd(listen_fd)", 61 "cli_fd:%d errno:%d\n", cli_fd, errno)) 62 goto done; 63 64 accept_fd = accept(listen_fd, NULL, NULL); 65 if (CHECK(accept_fd == -1, "accept(listen_fd)", 66 "accept_fd:%d errno:%d\n", accept_fd, errno)) 67 goto done; 68 69 /* Send 1000B of '+'s from cli_fd -> accept_fd */ 70 for (i = 0; i < 1000; i++) 71 buf[i] = '+'; 72 73 rv = send(cli_fd, buf, 1000, 0); 74 if (CHECK(rv != 1000, "send(cli_fd)", "rv:%d errno:%d\n", rv, errno)) 75 goto done; 76 77 rv = recv(accept_fd, buf, 1000, 0); 78 if (CHECK(rv != 1000, "recv(accept_fd)", "rv:%d errno:%d\n", rv, errno)) 79 goto done; 80 81 /* Send 500B of '.'s from accept_fd ->cli_fd */ 82 for (i = 0; i < 500; i++) 83 buf[i] = '.'; 84 85 rv = send(accept_fd, buf, 500, 0); 86 if (CHECK(rv != 500, "send(accept_fd)", "rv:%d errno:%d\n", rv, errno)) 87 goto done; 88 89 rv = recv(cli_fd, buf, 500, 0); 90 if (CHECK(rv != 500, "recv(cli_fd)", "rv:%d errno:%d\n", rv, errno)) 91 goto done; 92 93 /* 94 * shutdown accept first to guarantee correct ordering for 95 * bytes_received and bytes_acked when we go to verify the results. 96 */ 97 shutdown(accept_fd, SHUT_WR); 98 err = recv(cli_fd, buf, 1, 0); 99 if (CHECK(err, "recv(cli_fd) for fin", "err:%d errno:%d\n", err, errno)) 100 goto done; 101 102 shutdown(cli_fd, SHUT_WR); 103 err = recv(accept_fd, buf, 1, 0); 104 CHECK(err, "recv(accept_fd) for fin", "err:%d errno:%d\n", err, errno); 105 done: 106 if (accept_fd != -1) 107 close(accept_fd); 108 if (cli_fd != -1) 109 close(cli_fd); 110 if (listen_fd != -1) 111 close(listen_fd); 112 113 if (!err) 114 verify_result(result); 115 } 116 117 void test_tcpbpf_user(void) 118 { 119 struct test_tcpbpf_kern *skel; 120 int cg_fd = -1; 121 122 skel = test_tcpbpf_kern__open_and_load(); 123 if (CHECK(!skel, "open and load skel", "failed")) 124 return; 125 126 cg_fd = test__join_cgroup(CG_NAME); 127 if (CHECK(cg_fd < 0, "test__join_cgroup(" CG_NAME ")", 128 "cg_fd:%d errno:%d", cg_fd, errno)) 129 goto err; 130 131 skel->links.bpf_testcb = bpf_program__attach_cgroup(skel->progs.bpf_testcb, cg_fd); 132 if (!ASSERT_OK_PTR(skel->links.bpf_testcb, "attach_cgroup(bpf_testcb)")) 133 goto err; 134 135 run_test(&skel->bss->global); 136 137 err: 138 if (cg_fd != -1) 139 close(cg_fd); 140 test_tcpbpf_kern__destroy(skel); 141 } 142