Skip to content

Commit

Permalink
selftests/bpf: add two skmsg verdict tests for BPF_F_PERMANENT flag
Browse files Browse the repository at this point in the history
Add two tests for BPF_F_PERMANENT flag in sockmap_basic.c.

Signed-off-by: Liu Jian <[email protected]>
  • Loading branch information
liujian56 authored and d-e-s-o committed Nov 9, 2023
1 parent 5db027f commit b6d86cf
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions tools/testing/selftests/bpf/prog_tests/sockmap_basic.c
Original file line number Diff line number Diff line change
Expand Up @@ -525,7 +525,7 @@ static void test_sockmap_skb_verdict_peek(void)
test_sockmap_pass_prog__destroy(pass);
}

static void test_sockmap_msg_verdict(bool is_ingress)
static void test_sockmap_msg_verdict(bool is_ingress, bool is_permanent)
{
int key, sent, recvd, recv_fd;
int err, map, verdict, s, c0, c1, p0, p1;
Expand Down Expand Up @@ -577,11 +577,18 @@ static void test_sockmap_msg_verdict(bool is_ingress)
skel->bss->skmsg_redir_key = 2;
}

if (is_permanent)
skel->bss->skmsg_redir_flags |= BPF_F_PERMANENT;

sent = xsend(p1, &buf, sizeof(buf), 0);
ASSERT_EQ(sent, sizeof(buf), "xsend(p1)");
recvd = recv_timeout(recv_fd, &buf, sizeof(buf), SOCK_NONBLOCK, IO_TIMEOUT_SEC);
ASSERT_EQ(recvd, sizeof(buf), "recv_timeout(recv_fd)");

sent = xsend(p1, &buf, sizeof(buf), 0);
ASSERT_EQ(sent, sizeof(buf), "xsend(p1)");
recvd = recv_timeout(recv_fd, &buf, sizeof(buf), SOCK_NONBLOCK, IO_TIMEOUT_SEC);
ASSERT_EQ(recvd, sizeof(buf), "recv_timeout(recv_fd)");
out_close:
close(c0);
close(p0);
Expand Down Expand Up @@ -634,7 +641,11 @@ void test_sockmap_basic(void)
if (test__start_subtest("sockmap skb_verdict msg_f_peek"))
test_sockmap_skb_verdict_peek();
if (test__start_subtest("sockmap msg_verdict"))
test_sockmap_msg_verdict(false);
test_sockmap_msg_verdict(false, false);
if (test__start_subtest("sockmap msg_verdict ingress"))
test_sockmap_msg_verdict(true);
test_sockmap_msg_verdict(true, false);
if (test__start_subtest("sockmap msg_verdict permanent"))
test_sockmap_msg_verdict(false, true);
if (test__start_subtest("sockmap msg_verdict ingress permanent"))
test_sockmap_msg_verdict(true, true);
}

0 comments on commit b6d86cf

Please sign in to comment.