bpf/prog_tests/netns_cookie.c

0001 // SPDX-License-Identifier: GPL-2.0
0002
0003 #include <test_progs.h>
0004 #include "netns_cookie_prog.skel.h"
0005 #include "network_helpers.h"
0006
0007 #ifndef SO_NETNS_COOKIE
0008 #define SO_NETNS_COOKIE 71
0009 #endif
0010
0011 static int duration;
0012
0013 void test_netns_cookie(void)
0014 {
0015     int server_fd = -1, client_fd = -1, cgroup_fd = -1;
0016     int err, val, ret, map, verdict;
0017     struct netns_cookie_prog *skel;
0018     uint64_t cookie_expected_value;
0019     socklen_t vallen = sizeof(cookie_expected_value);
0020     static const char send_msg[] = "message";
0021
0022     skel = netns_cookie_prog__open_and_load();
0023     if (!ASSERT_OK_PTR(skel, "skel_open"))
0024         return;
0025
0026     cgroup_fd = test__join_cgroup("/netns_cookie");
0027     if (CHECK(cgroup_fd < 0, "join_cgroup", "cgroup creation failed\n"))
0028         goto done;
0029
0030     skel->links.get_netns_cookie_sockops = bpf_program__attach_cgroup(
0031         skel->progs.get_netns_cookie_sockops, cgroup_fd);
0032     if (!ASSERT_OK_PTR(skel->links.get_netns_cookie_sockops, "prog_attach"))
0033         goto done;
0034
0035     verdict = bpf_program__fd(skel->progs.get_netns_cookie_sk_msg);
0036     map = bpf_map__fd(skel->maps.sock_map);
0037     err = bpf_prog_attach(verdict, map, BPF_SK_MSG_VERDICT, 0);
0038     if (!ASSERT_OK(err, "prog_attach"))
0039         goto done;
0040
0041     server_fd = start_server(AF_INET6, SOCK_STREAM, "::1", 0, 0);
0042     if (CHECK(server_fd < 0, "start_server", "errno %d\n", errno))
0043         goto done;
0044
0045     client_fd = connect_to_fd(server_fd, 0);
0046     if (CHECK(client_fd < 0, "connect_to_fd", "errno %d\n", errno))
0047         goto done;
0048
0049     ret = send(client_fd, send_msg, sizeof(send_msg), 0);
0050     if (CHECK(ret != sizeof(send_msg), "send(msg)", "ret:%d\n", ret))
0051         goto done;
0052
0053     err = bpf_map_lookup_elem(bpf_map__fd(skel->maps.sockops_netns_cookies),
0054                   &client_fd, &val);
0055     if (!ASSERT_OK(err, "map_lookup(sockops_netns_cookies)"))
0056         goto done;
0057
0058     err = getsockopt(client_fd, SOL_SOCKET, SO_NETNS_COOKIE,
0059              &cookie_expected_value, &vallen);
0060     if (!ASSERT_OK(err, "getsockopt"))
0061         goto done;
0062
0063     ASSERT_EQ(val, cookie_expected_value, "cookie_value");
0064
0065     err = bpf_map_lookup_elem(bpf_map__fd(skel->maps.sk_msg_netns_cookies),
0066                   &client_fd, &val);
0067     if (!ASSERT_OK(err, "map_lookup(sk_msg_netns_cookies)"))
0068         goto done;
0069
0070     ASSERT_EQ(val, cookie_expected_value, "cookie_value");
0071
0072 done:
0073     if (server_fd != -1)
0074         close(server_fd);
0075     if (client_fd != -1)
0076         close(client_fd);
0077     if (cgroup_fd != -1)
0078         close(cgroup_fd);
0079     netns_cookie_prog__destroy(skel);
0080 }