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
 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 for
 92	 * 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}