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