1e3e02e1dSStanislav Fomichev // SPDX-License-Identifier: GPL-2.0
2e3e02e1dSStanislav Fomichev #include <test_progs.h>
3e3e02e1dSStanislav Fomichev #include "cgroup_helpers.h"
4e3e02e1dSStanislav Fomichev
5*e01b4a72SStanislav Fomichev #include "sockopt_inherit.skel.h"
6*e01b4a72SStanislav Fomichev
7e3e02e1dSStanislav Fomichev #define SOL_CUSTOM 0xdeadbeef
8e3e02e1dSStanislav Fomichev #define CUSTOM_INHERIT1 0
9e3e02e1dSStanislav Fomichev #define CUSTOM_INHERIT2 1
10e3e02e1dSStanislav Fomichev #define CUSTOM_LISTENER 2
11e3e02e1dSStanislav Fomichev
connect_to_server(int server_fd)12e3e02e1dSStanislav Fomichev static int connect_to_server(int server_fd)
13e3e02e1dSStanislav Fomichev {
14e3e02e1dSStanislav Fomichev struct sockaddr_storage addr;
15e3e02e1dSStanislav Fomichev socklen_t len = sizeof(addr);
16e3e02e1dSStanislav Fomichev int fd;
17e3e02e1dSStanislav Fomichev
18e3e02e1dSStanislav Fomichev fd = socket(AF_INET, SOCK_STREAM, 0);
19e3e02e1dSStanislav Fomichev if (fd < 0) {
20e3e02e1dSStanislav Fomichev log_err("Failed to create client socket");
21e3e02e1dSStanislav Fomichev return -1;
22e3e02e1dSStanislav Fomichev }
23e3e02e1dSStanislav Fomichev
24e3e02e1dSStanislav Fomichev if (getsockname(server_fd, (struct sockaddr *)&addr, &len)) {
25e3e02e1dSStanislav Fomichev log_err("Failed to get server addr");
26e3e02e1dSStanislav Fomichev goto out;
27e3e02e1dSStanislav Fomichev }
28e3e02e1dSStanislav Fomichev
29e3e02e1dSStanislav Fomichev if (connect(fd, (const struct sockaddr *)&addr, len) < 0) {
30e3e02e1dSStanislav Fomichev log_err("Fail to connect to server");
31e3e02e1dSStanislav Fomichev goto out;
32e3e02e1dSStanislav Fomichev }
33e3e02e1dSStanislav Fomichev
34e3e02e1dSStanislav Fomichev return fd;
35e3e02e1dSStanislav Fomichev
36e3e02e1dSStanislav Fomichev out:
37e3e02e1dSStanislav Fomichev close(fd);
38e3e02e1dSStanislav Fomichev return -1;
39e3e02e1dSStanislav Fomichev }
40e3e02e1dSStanislav Fomichev
verify_sockopt(int fd,int optname,const char * msg,char expected)41e3e02e1dSStanislav Fomichev static int verify_sockopt(int fd, int optname, const char *msg, char expected)
42e3e02e1dSStanislav Fomichev {
43e3e02e1dSStanislav Fomichev socklen_t optlen = 1;
44e3e02e1dSStanislav Fomichev char buf = 0;
45e3e02e1dSStanislav Fomichev int err;
46e3e02e1dSStanislav Fomichev
47e3e02e1dSStanislav Fomichev err = getsockopt(fd, SOL_CUSTOM, optname, &buf, &optlen);
48e3e02e1dSStanislav Fomichev if (err) {
49e3e02e1dSStanislav Fomichev log_err("%s: failed to call getsockopt", msg);
50e3e02e1dSStanislav Fomichev return 1;
51e3e02e1dSStanislav Fomichev }
52e3e02e1dSStanislav Fomichev
53e3e02e1dSStanislav Fomichev printf("%s %d: got=0x%x ? expected=0x%x\n", msg, optname, buf, expected);
54e3e02e1dSStanislav Fomichev
55e3e02e1dSStanislav Fomichev if (buf != expected) {
56e3e02e1dSStanislav Fomichev log_err("%s: unexpected getsockopt value %d != %d", msg,
57e3e02e1dSStanislav Fomichev buf, expected);
58e3e02e1dSStanislav Fomichev return 1;
59e3e02e1dSStanislav Fomichev }
60e3e02e1dSStanislav Fomichev
61e3e02e1dSStanislav Fomichev return 0;
62e3e02e1dSStanislav Fomichev }
63e3e02e1dSStanislav Fomichev
64e3e02e1dSStanislav Fomichev static pthread_mutex_t server_started_mtx = PTHREAD_MUTEX_INITIALIZER;
65e3e02e1dSStanislav Fomichev static pthread_cond_t server_started = PTHREAD_COND_INITIALIZER;
66e3e02e1dSStanislav Fomichev
server_thread(void * arg)67e3e02e1dSStanislav Fomichev static void *server_thread(void *arg)
68e3e02e1dSStanislav Fomichev {
69e3e02e1dSStanislav Fomichev struct sockaddr_storage addr;
70e3e02e1dSStanislav Fomichev socklen_t len = sizeof(addr);
71e3e02e1dSStanislav Fomichev int fd = *(int *)arg;
72e3e02e1dSStanislav Fomichev int client_fd;
73e3e02e1dSStanislav Fomichev int err = 0;
74e3e02e1dSStanislav Fomichev
75e3e02e1dSStanislav Fomichev err = listen(fd, 1);
76e3e02e1dSStanislav Fomichev
77e3e02e1dSStanislav Fomichev pthread_mutex_lock(&server_started_mtx);
78e3e02e1dSStanislav Fomichev pthread_cond_signal(&server_started);
79e3e02e1dSStanislav Fomichev pthread_mutex_unlock(&server_started_mtx);
80e3e02e1dSStanislav Fomichev
81675bc8abSWang Yufen if (!ASSERT_GE(err, 0, "listed on socket"))
82e3e02e1dSStanislav Fomichev return NULL;
83e3e02e1dSStanislav Fomichev
84e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_INHERIT1, "listen", 1);
85e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_INHERIT2, "listen", 1);
86e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_LISTENER, "listen", 1);
87e3e02e1dSStanislav Fomichev
88e3e02e1dSStanislav Fomichev client_fd = accept(fd, (struct sockaddr *)&addr, &len);
89675bc8abSWang Yufen if (!ASSERT_GE(client_fd, 0, "accept client"))
90e3e02e1dSStanislav Fomichev return NULL;
91e3e02e1dSStanislav Fomichev
92e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT1, "accept", 1);
93e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT2, "accept", 1);
94e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_LISTENER, "accept", 0);
95e3e02e1dSStanislav Fomichev
96e3e02e1dSStanislav Fomichev close(client_fd);
97e3e02e1dSStanislav Fomichev
98e3e02e1dSStanislav Fomichev return (void *)(long)err;
99e3e02e1dSStanislav Fomichev }
100e3e02e1dSStanislav Fomichev
start_server(void)101e3e02e1dSStanislav Fomichev static int start_server(void)
102e3e02e1dSStanislav Fomichev {
103e3e02e1dSStanislav Fomichev struct sockaddr_in addr = {
104e3e02e1dSStanislav Fomichev .sin_family = AF_INET,
105e3e02e1dSStanislav Fomichev .sin_addr.s_addr = htonl(INADDR_LOOPBACK),
106e3e02e1dSStanislav Fomichev };
107e3e02e1dSStanislav Fomichev char buf;
108e3e02e1dSStanislav Fomichev int err;
109e3e02e1dSStanislav Fomichev int fd;
110e3e02e1dSStanislav Fomichev int i;
111e3e02e1dSStanislav Fomichev
112e3e02e1dSStanislav Fomichev fd = socket(AF_INET, SOCK_STREAM, 0);
113e3e02e1dSStanislav Fomichev if (fd < 0) {
114e3e02e1dSStanislav Fomichev log_err("Failed to create server socket");
115e3e02e1dSStanislav Fomichev return -1;
116e3e02e1dSStanislav Fomichev }
117e3e02e1dSStanislav Fomichev
118e3e02e1dSStanislav Fomichev for (i = CUSTOM_INHERIT1; i <= CUSTOM_LISTENER; i++) {
119e3e02e1dSStanislav Fomichev buf = 0x01;
120e3e02e1dSStanislav Fomichev err = setsockopt(fd, SOL_CUSTOM, i, &buf, 1);
121e3e02e1dSStanislav Fomichev if (err) {
122e3e02e1dSStanislav Fomichev log_err("Failed to call setsockopt(%d)", i);
123e3e02e1dSStanislav Fomichev close(fd);
124e3e02e1dSStanislav Fomichev return -1;
125e3e02e1dSStanislav Fomichev }
126e3e02e1dSStanislav Fomichev }
127e3e02e1dSStanislav Fomichev
128e3e02e1dSStanislav Fomichev if (bind(fd, (const struct sockaddr *)&addr, sizeof(addr)) < 0) {
129e3e02e1dSStanislav Fomichev log_err("Failed to bind socket");
130e3e02e1dSStanislav Fomichev close(fd);
131e3e02e1dSStanislav Fomichev return -1;
132e3e02e1dSStanislav Fomichev }
133e3e02e1dSStanislav Fomichev
134e3e02e1dSStanislav Fomichev return fd;
135e3e02e1dSStanislav Fomichev }
136e3e02e1dSStanislav Fomichev
run_test(int cgroup_fd)137e3e02e1dSStanislav Fomichev static void run_test(int cgroup_fd)
138e3e02e1dSStanislav Fomichev {
139*e01b4a72SStanislav Fomichev struct bpf_link *link_getsockopt = NULL;
140*e01b4a72SStanislav Fomichev struct bpf_link *link_setsockopt = NULL;
141e3e02e1dSStanislav Fomichev int server_fd = -1, client_fd;
142*e01b4a72SStanislav Fomichev struct sockopt_inherit *obj;
143e3e02e1dSStanislav Fomichev void *server_err;
144e3e02e1dSStanislav Fomichev pthread_t tid;
145e3e02e1dSStanislav Fomichev int err;
146e3e02e1dSStanislav Fomichev
147*e01b4a72SStanislav Fomichev obj = sockopt_inherit__open_and_load();
148*e01b4a72SStanislav Fomichev if (!ASSERT_OK_PTR(obj, "skel-load"))
149e3e02e1dSStanislav Fomichev return;
150e3e02e1dSStanislav Fomichev
151*e01b4a72SStanislav Fomichev obj->bss->page_size = sysconf(_SC_PAGESIZE);
152*e01b4a72SStanislav Fomichev
153*e01b4a72SStanislav Fomichev link_getsockopt = bpf_program__attach_cgroup(obj->progs._getsockopt,
154*e01b4a72SStanislav Fomichev cgroup_fd);
155*e01b4a72SStanislav Fomichev if (!ASSERT_OK_PTR(link_getsockopt, "cg-attach-getsockopt"))
156186d1a86SAndrii Nakryiko goto close_bpf_object;
157186d1a86SAndrii Nakryiko
158*e01b4a72SStanislav Fomichev link_setsockopt = bpf_program__attach_cgroup(obj->progs._setsockopt,
159*e01b4a72SStanislav Fomichev cgroup_fd);
160*e01b4a72SStanislav Fomichev if (!ASSERT_OK_PTR(link_setsockopt, "cg-attach-setsockopt"))
161e3e02e1dSStanislav Fomichev goto close_bpf_object;
162e3e02e1dSStanislav Fomichev
163e3e02e1dSStanislav Fomichev server_fd = start_server();
164675bc8abSWang Yufen if (!ASSERT_GE(server_fd, 0, "start_server"))
165e3e02e1dSStanislav Fomichev goto close_bpf_object;
166e3e02e1dSStanislav Fomichev
167e81e7a53SStanislav Fomichev pthread_mutex_lock(&server_started_mtx);
168675bc8abSWang Yufen if (!ASSERT_OK(pthread_create(&tid, NULL, server_thread,
169675bc8abSWang Yufen (void *)&server_fd), "pthread_create")) {
17047bb27a2SChengfeng Ye pthread_mutex_unlock(&server_started_mtx);
171a2d074e4SBrian Vazquez goto close_server_fd;
17247bb27a2SChengfeng Ye }
173e3e02e1dSStanislav Fomichev pthread_cond_wait(&server_started, &server_started_mtx);
174e3e02e1dSStanislav Fomichev pthread_mutex_unlock(&server_started_mtx);
175e3e02e1dSStanislav Fomichev
176e3e02e1dSStanislav Fomichev client_fd = connect_to_server(server_fd);
177675bc8abSWang Yufen if (!ASSERT_GE(client_fd, 0, "connect_to_server"))
178e3e02e1dSStanislav Fomichev goto close_server_fd;
179e3e02e1dSStanislav Fomichev
180675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_INHERIT1, "connect", 0), "verify_sockopt1");
181675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_INHERIT2, "connect", 0), "verify_sockopt2");
182675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_LISTENER, "connect", 0), "verify_sockopt ener");
183e3e02e1dSStanislav Fomichev
184e3e02e1dSStanislav Fomichev pthread_join(tid, &server_err);
185e3e02e1dSStanislav Fomichev
186e3e02e1dSStanislav Fomichev err = (int)(long)server_err;
187675bc8abSWang Yufen ASSERT_OK(err, "pthread_join retval");
188e3e02e1dSStanislav Fomichev
189e3e02e1dSStanislav Fomichev close(client_fd);
190e3e02e1dSStanislav Fomichev
191e3e02e1dSStanislav Fomichev close_server_fd:
192e3e02e1dSStanislav Fomichev close(server_fd);
193e3e02e1dSStanislav Fomichev close_bpf_object:
194*e01b4a72SStanislav Fomichev bpf_link__destroy(link_getsockopt);
195*e01b4a72SStanislav Fomichev bpf_link__destroy(link_setsockopt);
196*e01b4a72SStanislav Fomichev
197*e01b4a72SStanislav Fomichev sockopt_inherit__destroy(obj);
198e3e02e1dSStanislav Fomichev }
199e3e02e1dSStanislav Fomichev
test_sockopt_inherit(void)200e3e02e1dSStanislav Fomichev void test_sockopt_inherit(void)
201e3e02e1dSStanislav Fomichev {
202e3e02e1dSStanislav Fomichev int cgroup_fd;
203e3e02e1dSStanislav Fomichev
204e3e02e1dSStanislav Fomichev cgroup_fd = test__join_cgroup("/sockopt_inherit");
205675bc8abSWang Yufen if (!ASSERT_GE(cgroup_fd, 0, "join_cgroup"))
206e3e02e1dSStanislav Fomichev return;
207e3e02e1dSStanislav Fomichev
208e3e02e1dSStanislav Fomichev run_test(cgroup_fd);
209e3e02e1dSStanislav Fomichev close(cgroup_fd);
210e3e02e1dSStanislav Fomichev }
211