tjh.dev / kernel
Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel os linux
fork atom

selftests/bpf: add cgroup_skb netns cookie tests

Add netns cookie test that verifies the helper is now supported and work
in the context of cgroup_skb programs.

Signed-off-by: Mahe Tardy <mahe.tardy@gmail.com>
Link: https://lore.kernel.org/r/20250225125031.258740-2-mahe.tardy@gmail.com
Signed-off-by: Alexei Starovoitov <ast@kernel.org>

authored by

Mahe Tardy and committed by
Alexei Starovoitov 9138048b c221d374

+23 -7
+14 -7
tools/testing/selftests/bpf/prog_tests/netns_cookie.c
··· 33 33 34 34 skel->links.get_netns_cookie_sockops = bpf_program__attach_cgroup( 35 35 skel->progs.get_netns_cookie_sockops, cgroup_fd); 36 - if (!ASSERT_OK_PTR(skel->links.get_netns_cookie_sockops, "prog_attach")) 36 + if (!ASSERT_OK_PTR(skel->links.get_netns_cookie_sockops, "prog_attach_sockops")) 37 37 goto done; 38 38 39 39 verdict = bpf_program__fd(skel->progs.get_netns_cookie_sk_msg); 40 40 map = bpf_map__fd(skel->maps.sock_map); 41 41 err = bpf_prog_attach(verdict, map, BPF_SK_MSG_VERDICT, 0); 42 - if (!ASSERT_OK(err, "prog_attach")) 42 + if (!ASSERT_OK(err, "prog_attach_sk_msg")) 43 43 goto done; 44 44 45 45 tc_fd = bpf_program__fd(skel->progs.get_netns_cookie_tcx); 46 46 err = bpf_prog_attach_opts(tc_fd, loopback, BPF_TCX_INGRESS, &opta); 47 - if (!ASSERT_OK(err, "prog_attach")) 47 + if (!ASSERT_OK(err, "prog_attach_tcx")) 48 48 goto done; 49 + 50 + skel->links.get_netns_cookie_cgroup_skb = bpf_program__attach_cgroup( 51 + skel->progs.get_netns_cookie_cgroup_skb, cgroup_fd); 52 + if (!ASSERT_OK_PTR(skel->links.get_netns_cookie_cgroup_skb, "prog_attach_cgroup_skb")) 53 + goto cleanup_tc; 49 54 50 55 server_fd = start_server(AF_INET6, SOCK_STREAM, "::1", 0, 0); 51 56 if (CHECK(server_fd < 0, "start_server", "errno %d\n", errno)) ··· 74 69 if (!ASSERT_OK(err, "getsockopt")) 75 70 goto cleanup_tc; 76 71 77 - ASSERT_EQ(val, cookie_expected_value, "cookie_value"); 72 + ASSERT_EQ(val, cookie_expected_value, "cookie_value_sockops"); 78 73 79 74 err = bpf_map_lookup_elem(bpf_map__fd(skel->maps.sk_msg_netns_cookies), 80 75 &client_fd, &val); 81 76 if (!ASSERT_OK(err, "map_lookup(sk_msg_netns_cookies)")) 82 77 goto cleanup_tc; 83 78 84 - ASSERT_EQ(val, cookie_expected_value, "cookie_value"); 85 - ASSERT_EQ(skel->bss->tcx_init_netns_cookie, cookie_expected_value, "cookie_value"); 86 - ASSERT_EQ(skel->bss->tcx_netns_cookie, cookie_expected_value, "cookie_value"); 79 + ASSERT_EQ(val, cookie_expected_value, "cookie_value_sk_msg"); 80 + ASSERT_EQ(skel->bss->tcx_init_netns_cookie, cookie_expected_value, "cookie_value_init_tcx"); 81 + ASSERT_EQ(skel->bss->tcx_netns_cookie, cookie_expected_value, "cookie_value_tcx"); 82 + ASSERT_EQ(skel->bss->cgroup_skb_init_netns_cookie, cookie_expected_value, "cookie_value_init_cgroup_skb"); 83 + ASSERT_EQ(skel->bss->cgroup_skb_netns_cookie, cookie_expected_value, "cookie_value_cgroup_skb"); 87 84 88 85 cleanup_tc: 89 86 err = bpf_prog_detach_opts(tc_fd, loopback, BPF_TCX_INGRESS, &optd);