public-mirrors/linux
1
0
Fork 0
mirror of git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git synced 2025-08-05 16:54:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

selftests/bpf: Migrate tcpbpf_user.c to use BPF skeleton

Browse source 
Update tcpbpf_user.c to make use of the BPF skeleton. Doing this we can
simplify test_tcpbpf_user and reduce the overhead involved in setting up
the test.

In addition we can clean up the remaining bits such as the one remaining
CHECK_FAIL at the end of test_tcpbpf_user so that the function only makes
use of CHECK as needed.

Signed-off-by: Alexander Duyck <alexanderduyck@fb.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Martin KaFai Lau <kafai@fb.com>
Acked-by: Andrii Nakryiko <andrii@kernel.org>
Link: https://lore.kernel.org/bpf/160443931155.1086697.17869006617113525162.stgit@localhost.localdomain
This commit is contained in:
Alexander Duyck 2020-11-03 13:35:11 -08:00 • committed by Alexei Starovoitov
parent d3813ea14b
commit 0a099d1429
1 changed files with 14 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
tools/testing/selftests/bpf/prog_tests/tcpbpf_user.c
View file

@ -3,6 +3,7 @@
#include <network_helpers.h> #include <network_helpers.h>
#include "test_tcpbpf.h" #include "test_tcpbpf.h"
#include "test_tcpbpf_kern.skel.h"
#define LO_ADDR6 "::1" #define LO_ADDR6 "::1"
#define CG_NAME "/tcpbpf-user-test" #define CG_NAME "/tcpbpf-user-test"
@ -130,44 +131,30 @@ done:
void test_tcpbpf_user(void) void test_tcpbpf_user(void)
{ {
const char *file = "test_tcpbpf_kern.o"; struct test_tcpbpf_kern *skel;
int prog_fd, map_fd, sock_map_fd; int map_fd, sock_map_fd;
int error = EXIT_FAILURE;
struct bpf_object *obj;
int cg_fd = -1; int cg_fd = -1;
int rv;
skel = test_tcpbpf_kern__open_and_load();
if (CHECK(!skel, "open and load skel", "failed"))
return;
cg_fd = test__join_cgroup(CG_NAME); cg_fd = test__join_cgroup(CG_NAME);
if (cg_fd < 0) if (CHECK(cg_fd < 0, "test__join_cgroup(" CG_NAME ")",
"cg_fd:%d errno:%d", cg_fd, errno))
goto err; goto err;
if (bpf_prog_load(file, BPF_PROG_TYPE_SOCK_OPS, &obj, &prog_fd)) { map_fd = bpf_map__fd(skel->maps.global_map);
printf("FAILED: load_bpf_file failed for: %s\n", file); sock_map_fd = bpf_map__fd(skel->maps.sockopt_results);
goto err;
}
rv = bpf_prog_attach(prog_fd, cg_fd, BPF_CGROUP_SOCK_OPS, 0); skel->links.bpf_testcb = bpf_program__attach_cgroup(skel->progs.bpf_testcb, cg_fd);
if (rv) { if (!ASSERT_OK_PTR(skel->links.bpf_testcb, "attach_cgroup(bpf_testcb)"))
printf("FAILED: bpf_prog_attach: %d (%s)\n",
error, strerror(errno));
goto err;
}
map_fd = bpf_find_map(__func__, obj, "global_map");
if (map_fd < 0)
goto err;
sock_map_fd = bpf_find_map(__func__, obj, "sockopt_results");
if (sock_map_fd < 0)
goto err; goto err;
run_test(map_fd, sock_map_fd); run_test(map_fd, sock_map_fd);
error = 0;
err: err:
bpf_prog_detach(cg_fd, BPF_CGROUP_SOCK_OPS);
if (cg_fd != -1) if (cg_fd != -1)
close(cg_fd); close(cg_fd);
test_tcpbpf_kern__destroy(skel);
CHECK_FAIL(error);
} }