1e3e02e1dSStanislav Fomichev // SPDX-License-Identifier: GPL-2.0 2e3e02e1dSStanislav Fomichev #include <test_progs.h> 3e3e02e1dSStanislav Fomichev #include "cgroup_helpers.h" 4e3e02e1dSStanislav Fomichev 5e3e02e1dSStanislav Fomichev #define SOL_CUSTOM 0xdeadbeef 6e3e02e1dSStanislav Fomichev #define CUSTOM_INHERIT1 0 7e3e02e1dSStanislav Fomichev #define CUSTOM_INHERIT2 1 8e3e02e1dSStanislav Fomichev #define CUSTOM_LISTENER 2 9e3e02e1dSStanislav Fomichev 10e3e02e1dSStanislav Fomichev static int connect_to_server(int server_fd) 11e3e02e1dSStanislav Fomichev { 12e3e02e1dSStanislav Fomichev struct sockaddr_storage addr; 13e3e02e1dSStanislav Fomichev socklen_t len = sizeof(addr); 14e3e02e1dSStanislav Fomichev int fd; 15e3e02e1dSStanislav Fomichev 16e3e02e1dSStanislav Fomichev fd = socket(AF_INET, SOCK_STREAM, 0); 17e3e02e1dSStanislav Fomichev if (fd < 0) { 18e3e02e1dSStanislav Fomichev log_err("Failed to create client socket"); 19e3e02e1dSStanislav Fomichev return -1; 20e3e02e1dSStanislav Fomichev } 21e3e02e1dSStanislav Fomichev 22e3e02e1dSStanislav Fomichev if (getsockname(server_fd, (struct sockaddr *)&addr, &len)) { 23e3e02e1dSStanislav Fomichev log_err("Failed to get server addr"); 24e3e02e1dSStanislav Fomichev goto out; 25e3e02e1dSStanislav Fomichev } 26e3e02e1dSStanislav Fomichev 27e3e02e1dSStanislav Fomichev if (connect(fd, (const struct sockaddr *)&addr, len) < 0) { 28e3e02e1dSStanislav Fomichev log_err("Fail to connect to server"); 29e3e02e1dSStanislav Fomichev goto out; 30e3e02e1dSStanislav Fomichev } 31e3e02e1dSStanislav Fomichev 32e3e02e1dSStanislav Fomichev return fd; 33e3e02e1dSStanislav Fomichev 34e3e02e1dSStanislav Fomichev out: 35e3e02e1dSStanislav Fomichev close(fd); 36e3e02e1dSStanislav Fomichev return -1; 37e3e02e1dSStanislav Fomichev } 38e3e02e1dSStanislav Fomichev 39e3e02e1dSStanislav Fomichev static int verify_sockopt(int fd, int optname, const char *msg, char expected) 40e3e02e1dSStanislav Fomichev { 41e3e02e1dSStanislav Fomichev socklen_t optlen = 1; 42e3e02e1dSStanislav Fomichev char buf = 0; 43e3e02e1dSStanislav Fomichev int err; 44e3e02e1dSStanislav Fomichev 45e3e02e1dSStanislav Fomichev err = getsockopt(fd, SOL_CUSTOM, optname, &buf, &optlen); 46e3e02e1dSStanislav Fomichev if (err) { 47e3e02e1dSStanislav Fomichev log_err("%s: failed to call getsockopt", msg); 48e3e02e1dSStanislav Fomichev return 1; 49e3e02e1dSStanislav Fomichev } 50e3e02e1dSStanislav Fomichev 51e3e02e1dSStanislav Fomichev printf("%s %d: got=0x%x ? expected=0x%x\n", msg, optname, buf, expected); 52e3e02e1dSStanislav Fomichev 53e3e02e1dSStanislav Fomichev if (buf != expected) { 54e3e02e1dSStanislav Fomichev log_err("%s: unexpected getsockopt value %d != %d", msg, 55e3e02e1dSStanislav Fomichev buf, expected); 56e3e02e1dSStanislav Fomichev return 1; 57e3e02e1dSStanislav Fomichev } 58e3e02e1dSStanislav Fomichev 59e3e02e1dSStanislav Fomichev return 0; 60e3e02e1dSStanislav Fomichev } 61e3e02e1dSStanislav Fomichev 62e3e02e1dSStanislav Fomichev static pthread_mutex_t server_started_mtx = PTHREAD_MUTEX_INITIALIZER; 63e3e02e1dSStanislav Fomichev static pthread_cond_t server_started = PTHREAD_COND_INITIALIZER; 64e3e02e1dSStanislav Fomichev 65e3e02e1dSStanislav Fomichev static void *server_thread(void *arg) 66e3e02e1dSStanislav Fomichev { 67e3e02e1dSStanislav Fomichev struct sockaddr_storage addr; 68e3e02e1dSStanislav Fomichev socklen_t len = sizeof(addr); 69e3e02e1dSStanislav Fomichev int fd = *(int *)arg; 70e3e02e1dSStanislav Fomichev int client_fd; 71e3e02e1dSStanislav Fomichev int err = 0; 72e3e02e1dSStanislav Fomichev 73e3e02e1dSStanislav Fomichev err = listen(fd, 1); 74e3e02e1dSStanislav Fomichev 75e3e02e1dSStanislav Fomichev pthread_mutex_lock(&server_started_mtx); 76e3e02e1dSStanislav Fomichev pthread_cond_signal(&server_started); 77e3e02e1dSStanislav Fomichev pthread_mutex_unlock(&server_started_mtx); 78e3e02e1dSStanislav Fomichev 79*675bc8abSWang Yufen if (!ASSERT_GE(err, 0, "listed on socket")) 80e3e02e1dSStanislav Fomichev return NULL; 81e3e02e1dSStanislav Fomichev 82e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_INHERIT1, "listen", 1); 83e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_INHERIT2, "listen", 1); 84e3e02e1dSStanislav Fomichev err += verify_sockopt(fd, CUSTOM_LISTENER, "listen", 1); 85e3e02e1dSStanislav Fomichev 86e3e02e1dSStanislav Fomichev client_fd = accept(fd, (struct sockaddr *)&addr, &len); 87*675bc8abSWang Yufen if (!ASSERT_GE(client_fd, 0, "accept client")) 88e3e02e1dSStanislav Fomichev return NULL; 89e3e02e1dSStanislav Fomichev 90e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT1, "accept", 1); 91e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT2, "accept", 1); 92e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_LISTENER, "accept", 0); 93e3e02e1dSStanislav Fomichev 94e3e02e1dSStanislav Fomichev close(client_fd); 95e3e02e1dSStanislav Fomichev 96e3e02e1dSStanislav Fomichev return (void *)(long)err; 97e3e02e1dSStanislav Fomichev } 98e3e02e1dSStanislav Fomichev 99e3e02e1dSStanislav Fomichev static int start_server(void) 100e3e02e1dSStanislav Fomichev { 101e3e02e1dSStanislav Fomichev struct sockaddr_in addr = { 102e3e02e1dSStanislav Fomichev .sin_family = AF_INET, 103e3e02e1dSStanislav Fomichev .sin_addr.s_addr = htonl(INADDR_LOOPBACK), 104e3e02e1dSStanislav Fomichev }; 105e3e02e1dSStanislav Fomichev char buf; 106e3e02e1dSStanislav Fomichev int err; 107e3e02e1dSStanislav Fomichev int fd; 108e3e02e1dSStanislav Fomichev int i; 109e3e02e1dSStanislav Fomichev 110e3e02e1dSStanislav Fomichev fd = socket(AF_INET, SOCK_STREAM, 0); 111e3e02e1dSStanislav Fomichev if (fd < 0) { 112e3e02e1dSStanislav Fomichev log_err("Failed to create server socket"); 113e3e02e1dSStanislav Fomichev return -1; 114e3e02e1dSStanislav Fomichev } 115e3e02e1dSStanislav Fomichev 116e3e02e1dSStanislav Fomichev for (i = CUSTOM_INHERIT1; i <= CUSTOM_LISTENER; i++) { 117e3e02e1dSStanislav Fomichev buf = 0x01; 118e3e02e1dSStanislav Fomichev err = setsockopt(fd, SOL_CUSTOM, i, &buf, 1); 119e3e02e1dSStanislav Fomichev if (err) { 120e3e02e1dSStanislav Fomichev log_err("Failed to call setsockopt(%d)", i); 121e3e02e1dSStanislav Fomichev close(fd); 122e3e02e1dSStanislav Fomichev return -1; 123e3e02e1dSStanislav Fomichev } 124e3e02e1dSStanislav Fomichev } 125e3e02e1dSStanislav Fomichev 126e3e02e1dSStanislav Fomichev if (bind(fd, (const struct sockaddr *)&addr, sizeof(addr)) < 0) { 127e3e02e1dSStanislav Fomichev log_err("Failed to bind socket"); 128e3e02e1dSStanislav Fomichev close(fd); 129e3e02e1dSStanislav Fomichev return -1; 130e3e02e1dSStanislav Fomichev } 131e3e02e1dSStanislav Fomichev 132e3e02e1dSStanislav Fomichev return fd; 133e3e02e1dSStanislav Fomichev } 134e3e02e1dSStanislav Fomichev 135a393ea80SKui-Feng Lee static int prog_attach(struct bpf_object *obj, int cgroup_fd, const char *title, 136a393ea80SKui-Feng Lee const char *prog_name) 137e3e02e1dSStanislav Fomichev { 138e3e02e1dSStanislav Fomichev enum bpf_attach_type attach_type; 139e3e02e1dSStanislav Fomichev enum bpf_prog_type prog_type; 140e3e02e1dSStanislav Fomichev struct bpf_program *prog; 141e3e02e1dSStanislav Fomichev int err; 142e3e02e1dSStanislav Fomichev 143e3e02e1dSStanislav Fomichev err = libbpf_prog_type_by_name(title, &prog_type, &attach_type); 144e3e02e1dSStanislav Fomichev if (err) { 145a393ea80SKui-Feng Lee log_err("Failed to deduct types for %s BPF program", prog_name); 146e3e02e1dSStanislav Fomichev return -1; 147e3e02e1dSStanislav Fomichev } 148e3e02e1dSStanislav Fomichev 149a393ea80SKui-Feng Lee prog = bpf_object__find_program_by_name(obj, prog_name); 150e3e02e1dSStanislav Fomichev if (!prog) { 151a393ea80SKui-Feng Lee log_err("Failed to find %s BPF program", prog_name); 152e3e02e1dSStanislav Fomichev return -1; 153e3e02e1dSStanislav Fomichev } 154e3e02e1dSStanislav Fomichev 155e3e02e1dSStanislav Fomichev err = bpf_prog_attach(bpf_program__fd(prog), cgroup_fd, 156e3e02e1dSStanislav Fomichev attach_type, 0); 157e3e02e1dSStanislav Fomichev if (err) { 158a393ea80SKui-Feng Lee log_err("Failed to attach %s BPF program", prog_name); 159e3e02e1dSStanislav Fomichev return -1; 160e3e02e1dSStanislav Fomichev } 161e3e02e1dSStanislav Fomichev 162e3e02e1dSStanislav Fomichev return 0; 163e3e02e1dSStanislav Fomichev } 164e3e02e1dSStanislav Fomichev 165e3e02e1dSStanislav Fomichev static void run_test(int cgroup_fd) 166e3e02e1dSStanislav Fomichev { 167e3e02e1dSStanislav Fomichev int server_fd = -1, client_fd; 168e3e02e1dSStanislav Fomichev struct bpf_object *obj; 169e3e02e1dSStanislav Fomichev void *server_err; 170e3e02e1dSStanislav Fomichev pthread_t tid; 171e3e02e1dSStanislav Fomichev int err; 172e3e02e1dSStanislav Fomichev 173afef88e6SDaniel Müller obj = bpf_object__open_file("sockopt_inherit.bpf.o", NULL); 174186d1a86SAndrii Nakryiko if (!ASSERT_OK_PTR(obj, "obj_open")) 175e3e02e1dSStanislav Fomichev return; 176e3e02e1dSStanislav Fomichev 177186d1a86SAndrii Nakryiko err = bpf_object__load(obj); 178186d1a86SAndrii Nakryiko if (!ASSERT_OK(err, "obj_load")) 179186d1a86SAndrii Nakryiko goto close_bpf_object; 180186d1a86SAndrii Nakryiko 181a393ea80SKui-Feng Lee err = prog_attach(obj, cgroup_fd, "cgroup/getsockopt", "_getsockopt"); 182*675bc8abSWang Yufen if (!ASSERT_OK(err, "prog_attach _getsockopt")) 183e3e02e1dSStanislav Fomichev goto close_bpf_object; 184e3e02e1dSStanislav Fomichev 185a393ea80SKui-Feng Lee err = prog_attach(obj, cgroup_fd, "cgroup/setsockopt", "_setsockopt"); 186*675bc8abSWang Yufen if (!ASSERT_OK(err, "prog_attach _setsockopt")) 187e3e02e1dSStanislav Fomichev goto close_bpf_object; 188e3e02e1dSStanislav Fomichev 189e3e02e1dSStanislav Fomichev server_fd = start_server(); 190*675bc8abSWang Yufen if (!ASSERT_GE(server_fd, 0, "start_server")) 191e3e02e1dSStanislav Fomichev goto close_bpf_object; 192e3e02e1dSStanislav Fomichev 193e81e7a53SStanislav Fomichev pthread_mutex_lock(&server_started_mtx); 194*675bc8abSWang Yufen if (!ASSERT_OK(pthread_create(&tid, NULL, server_thread, 195*675bc8abSWang Yufen (void *)&server_fd), "pthread_create")) { 19647bb27a2SChengfeng Ye pthread_mutex_unlock(&server_started_mtx); 197a2d074e4SBrian Vazquez goto close_server_fd; 19847bb27a2SChengfeng Ye } 199e3e02e1dSStanislav Fomichev pthread_cond_wait(&server_started, &server_started_mtx); 200e3e02e1dSStanislav Fomichev pthread_mutex_unlock(&server_started_mtx); 201e3e02e1dSStanislav Fomichev 202e3e02e1dSStanislav Fomichev client_fd = connect_to_server(server_fd); 203*675bc8abSWang Yufen if (!ASSERT_GE(client_fd, 0, "connect_to_server")) 204e3e02e1dSStanislav Fomichev goto close_server_fd; 205e3e02e1dSStanislav Fomichev 206*675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_INHERIT1, "connect", 0), "verify_sockopt1"); 207*675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_INHERIT2, "connect", 0), "verify_sockopt2"); 208*675bc8abSWang Yufen ASSERT_OK(verify_sockopt(client_fd, CUSTOM_LISTENER, "connect", 0), "verify_sockopt ener"); 209e3e02e1dSStanislav Fomichev 210e3e02e1dSStanislav Fomichev pthread_join(tid, &server_err); 211e3e02e1dSStanislav Fomichev 212e3e02e1dSStanislav Fomichev err = (int)(long)server_err; 213*675bc8abSWang Yufen ASSERT_OK(err, "pthread_join retval"); 214e3e02e1dSStanislav Fomichev 215e3e02e1dSStanislav Fomichev close(client_fd); 216e3e02e1dSStanislav Fomichev 217e3e02e1dSStanislav Fomichev close_server_fd: 218e3e02e1dSStanislav Fomichev close(server_fd); 219e3e02e1dSStanislav Fomichev close_bpf_object: 220e3e02e1dSStanislav Fomichev bpf_object__close(obj); 221e3e02e1dSStanislav Fomichev } 222e3e02e1dSStanislav Fomichev 223e3e02e1dSStanislav Fomichev void test_sockopt_inherit(void) 224e3e02e1dSStanislav Fomichev { 225e3e02e1dSStanislav Fomichev int cgroup_fd; 226e3e02e1dSStanislav Fomichev 227e3e02e1dSStanislav Fomichev cgroup_fd = test__join_cgroup("/sockopt_inherit"); 228*675bc8abSWang Yufen if (!ASSERT_GE(cgroup_fd, 0, "join_cgroup")) 229e3e02e1dSStanislav Fomichev return; 230e3e02e1dSStanislav Fomichev 231e3e02e1dSStanislav Fomichev run_test(cgroup_fd); 232e3e02e1dSStanislav Fomichev close(cgroup_fd); 233e3e02e1dSStanislav Fomichev } 234