@@ -476,7 +476,7 @@ static void test_sockmap_skb_verdict_fionread(bool pass_prog)
test_sockmap_drop_prog__destroy(drop);
}
-static void test_sockmap_msg_verdict(bool is_ingress, bool is_permanent)
+static void test_sockmap_msg_verdict(bool is_ingress, bool is_permanent, bool is_self)
{
int key, sent, recvd, recv_fd;
int err, map, verdict, s, c0, c1, p0, p1;
@@ -519,13 +519,23 @@ static void test_sockmap_msg_verdict(bool is_ingress, bool is_permanent)
goto out_close;
if (is_ingress) {
- recv_fd = c1;
skel->bss->skmsg_redir_flags = BPF_F_INGRESS;
- skel->bss->skmsg_redir_key = 1;
+ if (is_self) {
+ skel->bss->skmsg_redir_key = 0;
+ recv_fd = p1;
+ } else {
+ skel->bss->skmsg_redir_key = 1;
+ recv_fd = c1;
+ }
} else {
- recv_fd = c0;
skel->bss->skmsg_redir_flags = 0;
- skel->bss->skmsg_redir_key = 2;
+ if (is_self) {
+ skel->bss->skmsg_redir_key = 0;
+ recv_fd = c1;
+ } else {
+ skel->bss->skmsg_redir_key = 2;
+ recv_fd = c0;
+ }
}
if (is_permanent)
@@ -590,11 +600,15 @@ void test_sockmap_basic(void)
if (test__start_subtest("sockmap skb_verdict fionread on drop"))
test_sockmap_skb_verdict_fionread(false);
if (test__start_subtest("sockmap msg_verdict"))
- test_sockmap_msg_verdict(false, false);
+ test_sockmap_msg_verdict(false, false, false);
if (test__start_subtest("sockmap msg_verdict ingress"))
- test_sockmap_msg_verdict(true, false);
+ test_sockmap_msg_verdict(true, false, false);
if (test__start_subtest("sockmap msg_verdict permanent"))
- test_sockmap_msg_verdict(false, true);
+ test_sockmap_msg_verdict(false, true, false);
if (test__start_subtest("sockmap msg_verdict ingress permanent"))
- test_sockmap_msg_verdict(true, true);
+ test_sockmap_msg_verdict(true, true, false);
+ if (test__start_subtest("sockmap msg_verdict permanent self"))
+ test_sockmap_msg_verdict(false, true, true);
+ if (test__start_subtest("sockmap msg_verdict ingress permanent self"))
+ test_sockmap_msg_verdict(true, true, true);
}
Add tests for verdict skmsg to itself in sockmap_basic.c Signed-off-by: Liu Jian <liujian56@huawei.com> --- .../selftests/bpf/prog_tests/sockmap_basic.c | 32 +++++++++++++------ 1 file changed, 23 insertions(+), 9 deletions(-)