Kernel/linux-next
1
0
Fork 0
mirror of https://git.kernel.org/pub/scm/linux/kernel/git/next/linux-next.git synced 2025-01-16 05:26:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

selftests/bpf: Add start_test helper in bpf_tcp_ca

Browse Source 
For moving the "if (sk_stg_map)" block out of do_test(), extract the
code before this block as a new function start_test(). It creates
server-side and client-side sockets and returns them to the caller.

Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
Link: https://lore.kernel.org/bpf/48f2921ff9be958f5d3d28fe6bb7269a61cafa9f.1717054461.git.tanggeliang@kylinos.cn
This commit is contained in:
Geliang Tang 2024-05-30 15:41:09 +08:00 committed by Daniel Borkmann
parent 9abdfd8a21
commit fee97d0c9a
1 changed files with 33 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c
View File

@ -39,6 +39,34 @@ static int settcpca(int fd, const char *tcp_ca)
return 0;
}
static bool start_test(char *addr_str,
const struct network_helper_opts *srv_opts,
const struct network_helper_opts *cli_opts,
int *srv_fd, int *cli_fd)
{
*srv_fd = start_server_str(AF_INET6, SOCK_STREAM, addr_str, 0, srv_opts);
if (!ASSERT_NEQ(*srv_fd, -1, "start_server_str"))
goto err;
/* connect to server */
*cli_fd = connect_to_fd_opts(*srv_fd, cli_opts);
if (!ASSERT_NEQ(*cli_fd, -1, "connect_to_fd_opts"))
goto err;
return true;
err:
if (*srv_fd != -1) {
close(*srv_fd);
*srv_fd = -1;
}
if (*cli_fd != -1) {
close(*cli_fd);
*cli_fd = -1;
}
return false;
}
static void do_test(const struct network_helper_opts *opts,
const struct network_helper_opts *cli_opts,
const struct bpf_map *sk_stg_map)
@ -46,13 +74,7 @@ static void do_test(const struct network_helper_opts *opts,
int lfd = -1, fd = -1;
int err;
lfd = start_server_str(AF_INET6, SOCK_STREAM, NULL, 0, opts);
if (!ASSERT_NEQ(lfd, -1, "socket"))
return;
/* connect to server */
fd = connect_to_fd_opts(lfd, cli_opts);
if (!ASSERT_NEQ(fd, -1, "connect_to_fd_opts"))
if (!start_test(NULL, opts, cli_opts, &lfd, &fd))
goto done;
if (sk_stg_map) {
@ -68,8 +90,10 @@ static void do_test(const struct network_helper_opts *opts,
ASSERT_OK(send_recv_data(lfd, fd, total_bytes), "send_recv_data");
done:
close(lfd);
close(fd);
if (lfd != -1)
close(lfd);
if (fd != -1)
close(fd);
}
static int cc_cb(int fd, void *opts)