1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2023 Facebook */
3 #include <test_progs.h>
4 #include <bpf/libbpf.h>
5 #include <sys/types.h>
6 #include <sys/socket.h>
7 #include "sk_storage_omem_uncharge.skel.h"
8
test_sk_storage_omem_uncharge(void)9 void test_sk_storage_omem_uncharge(void)
10 {
11 struct sk_storage_omem_uncharge *skel;
12 int sk_fd = -1, map_fd, err, value;
13 socklen_t optlen;
14
15 skel = sk_storage_omem_uncharge__open_and_load();
16 if (!ASSERT_OK_PTR(skel, "skel open_and_load"))
17 return;
18 map_fd = bpf_map__fd(skel->maps.sk_storage);
19
20 /* A standalone socket not binding to addr:port,
21 * so nentns is not needed.
22 */
23 sk_fd = socket(AF_INET6, SOCK_STREAM, 0);
24 if (!ASSERT_GE(sk_fd, 0, "socket"))
25 goto done;
26
27 optlen = sizeof(skel->bss->cookie);
28 err = getsockopt(sk_fd, SOL_SOCKET, SO_COOKIE, &skel->bss->cookie, &optlen);
29 if (!ASSERT_OK(err, "getsockopt(SO_COOKIE)"))
30 goto done;
31
32 value = 0;
33 err = bpf_map_update_elem(map_fd, &sk_fd, &value, 0);
34 if (!ASSERT_OK(err, "bpf_map_update_elem(value=0)"))
35 goto done;
36
37 value = 0xdeadbeef;
38 err = bpf_map_update_elem(map_fd, &sk_fd, &value, 0);
39 if (!ASSERT_OK(err, "bpf_map_update_elem(value=0xdeadbeef)"))
40 goto done;
41
42 err = sk_storage_omem_uncharge__attach(skel);
43 if (!ASSERT_OK(err, "attach"))
44 goto done;
45
46 close(sk_fd);
47 sk_fd = -1;
48
49 ASSERT_EQ(skel->bss->cookie_found, 2, "cookie_found");
50 ASSERT_EQ(skel->bss->omem, 0, "omem");
51
52 done:
53 sk_storage_omem_uncharge__destroy(skel);
54 if (sk_fd != -1)
55 close(sk_fd);
56 }
57