1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright Amazon.com Inc. or its affiliates. */
3 #include <sys/socket.h>
4 #include <sys/un.h>
5 #include <test_progs.h>
6 #include "bpf_iter_setsockopt_unix.skel.h"
7
8 #define NR_CASES 5
9
create_unix_socket(struct bpf_iter_setsockopt_unix * skel)10 static int create_unix_socket(struct bpf_iter_setsockopt_unix *skel)
11 {
12 struct sockaddr_un addr = {
13 .sun_family = AF_UNIX,
14 .sun_path = "",
15 };
16 socklen_t len;
17 int fd, err;
18
19 fd = socket(AF_UNIX, SOCK_STREAM, 0);
20 if (!ASSERT_NEQ(fd, -1, "socket"))
21 return -1;
22
23 len = offsetof(struct sockaddr_un, sun_path);
24 err = bind(fd, (struct sockaddr *)&addr, len);
25 if (!ASSERT_OK(err, "bind"))
26 return -1;
27
28 len = sizeof(addr);
29 err = getsockname(fd, (struct sockaddr *)&addr, &len);
30 if (!ASSERT_OK(err, "getsockname"))
31 return -1;
32
33 memcpy(&skel->bss->sun_path, &addr.sun_path,
34 len - offsetof(struct sockaddr_un, sun_path));
35
36 return fd;
37 }
38
test_sndbuf(struct bpf_iter_setsockopt_unix * skel,int fd)39 static void test_sndbuf(struct bpf_iter_setsockopt_unix *skel, int fd)
40 {
41 socklen_t optlen;
42 int i, err;
43
44 for (i = 0; i < NR_CASES; i++) {
45 if (!ASSERT_NEQ(skel->data->sndbuf_getsockopt[i], -1,
46 "bpf_(get|set)sockopt"))
47 return;
48
49 err = setsockopt(fd, SOL_SOCKET, SO_SNDBUF,
50 &(skel->data->sndbuf_setsockopt[i]),
51 sizeof(skel->data->sndbuf_setsockopt[i]));
52 if (!ASSERT_OK(err, "setsockopt"))
53 return;
54
55 optlen = sizeof(skel->bss->sndbuf_getsockopt_expected[i]);
56 err = getsockopt(fd, SOL_SOCKET, SO_SNDBUF,
57 &(skel->bss->sndbuf_getsockopt_expected[i]),
58 &optlen);
59 if (!ASSERT_OK(err, "getsockopt"))
60 return;
61
62 if (!ASSERT_EQ(skel->data->sndbuf_getsockopt[i],
63 skel->bss->sndbuf_getsockopt_expected[i],
64 "bpf_(get|set)sockopt"))
65 return;
66 }
67 }
68
test_bpf_iter_setsockopt_unix(void)69 void test_bpf_iter_setsockopt_unix(void)
70 {
71 struct bpf_iter_setsockopt_unix *skel;
72 int err, unix_fd, iter_fd;
73 char buf;
74
75 skel = bpf_iter_setsockopt_unix__open_and_load();
76 if (!ASSERT_OK_PTR(skel, "open_and_load"))
77 return;
78
79 unix_fd = create_unix_socket(skel);
80 if (!ASSERT_NEQ(unix_fd, -1, "create_unix_server"))
81 goto destroy;
82
83 skel->links.change_sndbuf = bpf_program__attach_iter(skel->progs.change_sndbuf, NULL);
84 if (!ASSERT_OK_PTR(skel->links.change_sndbuf, "bpf_program__attach_iter"))
85 goto destroy;
86
87 iter_fd = bpf_iter_create(bpf_link__fd(skel->links.change_sndbuf));
88 if (!ASSERT_GE(iter_fd, 0, "bpf_iter_create"))
89 goto destroy;
90
91 while ((err = read(iter_fd, &buf, sizeof(buf))) == -1 &&
92 errno == EAGAIN)
93 ;
94 if (!ASSERT_OK(err, "read iter error"))
95 goto destroy;
96
97 test_sndbuf(skel, unix_fd);
98 destroy:
99 bpf_iter_setsockopt_unix__destroy(skel);
100 }
101