korg/linux
0
0
Fork 0
mirror of https://mirrors.bfsu.edu.cn/git/linux.git synced 2024-11-26 05:34:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

selftests/bpf: Add connmark read test

Browse Source 
Test that the prog can read from the connection mark. This test is nice
because it ensures progs can interact with netfilter subsystem
correctly.

Signed-off-by: Daniel Xu <dxu@dxuuu.xyz>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
Acked-by: Kumar Kartikeya Dwivedi <memxor@gmail.com>
Link: https://lore.kernel.org/bpf/d3bc620a491e4c626c20d80631063922cbe13e2b.1660254747.git.dxu@dxuuu.xyz
This commit is contained in:
Daniel Xu 2022-08-11 15:55:26 -06:00 committed by Daniel Borkmann
parent e81fbd4c1b
commit 99799de2cb
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/testing/selftests/bpf/prog_tests/bpf_nf.c
View File

@ -44,7 +44,7 @@ out:
static void test_bpf_nf_ct(int mode) static void test_bpf_nf_ct(int mode)
{ {
const char *iptables = "iptables -t raw %s PREROUTING -j CT"; const char *iptables = "iptables -t raw %s PREROUTING -j CONNMARK --set-mark 42/0";
int srv_fd = -1, client_fd = -1, srv_client_fd = -1; int srv_fd = -1, client_fd = -1, srv_client_fd = -1;
struct sockaddr_in peer_addr = {}; struct sockaddr_in peer_addr = {};
struct test_bpf_nf *skel; struct test_bpf_nf *skel;
@ -114,6 +114,7 @@ static void test_bpf_nf_ct(int mode)
/* expected status is IPS_SEEN_REPLY */ /* expected status is IPS_SEEN_REPLY */
ASSERT_EQ(skel->bss->test_status, 2, "Test for ct status update "); ASSERT_EQ(skel->bss->test_status, 2, "Test for ct status update ");
ASSERT_EQ(skel->data->test_exist_lookup, 0, "Test existing connection lookup"); ASSERT_EQ(skel->data->test_exist_lookup, 0, "Test existing connection lookup");
ASSERT_EQ(skel->bss->test_exist_lookup_mark, 43, "Test existing connection lookup ctmark");
end: end:
if (srv_client_fd != -1) if (srv_client_fd != -1)
close(srv_client_fd); close(srv_client_fd);

3
tools/testing/selftests/bpf/progs/test_bpf_nf.c
View File

@ -28,6 +28,7 @@ __be16 sport = 0;
__be32 daddr = 0; __be32 daddr = 0;
__be16 dport = 0; __be16 dport = 0;
int test_exist_lookup = -ENOENT; int test_exist_lookup = -ENOENT;
u32 test_exist_lookup_mark = 0;
struct nf_conn; struct nf_conn;
@ -174,6 +175,8 @@ nf_ct_test(struct nf_conn *(*lookup_fn)(void *, struct bpf_sock_tuple *, u32,
sizeof(opts_def)); sizeof(opts_def));
if (ct) { if (ct) {
test_exist_lookup = 0; test_exist_lookup = 0;
if (ct->mark == 42)
test_exist_lookup_mark = 43;
bpf_ct_release(ct); bpf_ct_release(ct);
} else { } else {
test_exist_lookup = opts_def.error; test_exist_lookup = opts_def.error;