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 79e3e02e1dSStanislav Fomichev if (CHECK_FAIL(err < 0)) { 80e3e02e1dSStanislav Fomichev perror("Failed to listed on socket"); 81e3e02e1dSStanislav Fomichev return NULL; 82e3e02e1dSStanislav Fomichev } 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); 89e3e02e1dSStanislav Fomichev if (CHECK_FAIL(client_fd < 0)) { 90e3e02e1dSStanislav Fomichev perror("Failed to accept client"); 91e3e02e1dSStanislav Fomichev return NULL; 92e3e02e1dSStanislav Fomichev } 93e3e02e1dSStanislav Fomichev 94e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT1, "accept", 1); 95e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_INHERIT2, "accept", 1); 96e3e02e1dSStanislav Fomichev err += verify_sockopt(client_fd, CUSTOM_LISTENER, "accept", 0); 97e3e02e1dSStanislav Fomichev 98e3e02e1dSStanislav Fomichev close(client_fd); 99e3e02e1dSStanislav Fomichev 100e3e02e1dSStanislav Fomichev return (void *)(long)err; 101e3e02e1dSStanislav Fomichev } 102e3e02e1dSStanislav Fomichev 103e3e02e1dSStanislav Fomichev static int start_server(void) 104e3e02e1dSStanislav Fomichev { 105e3e02e1dSStanislav Fomichev struct sockaddr_in addr = { 106e3e02e1dSStanislav Fomichev .sin_family = AF_INET, 107e3e02e1dSStanislav Fomichev .sin_addr.s_addr = htonl(INADDR_LOOPBACK), 108e3e02e1dSStanislav Fomichev }; 109e3e02e1dSStanislav Fomichev char buf; 110e3e02e1dSStanislav Fomichev int err; 111e3e02e1dSStanislav Fomichev int fd; 112e3e02e1dSStanislav Fomichev int i; 113e3e02e1dSStanislav Fomichev 114e3e02e1dSStanislav Fomichev fd = socket(AF_INET, SOCK_STREAM, 0); 115e3e02e1dSStanislav Fomichev if (fd < 0) { 116e3e02e1dSStanislav Fomichev log_err("Failed to create server socket"); 117e3e02e1dSStanislav Fomichev return -1; 118e3e02e1dSStanislav Fomichev } 119e3e02e1dSStanislav Fomichev 120e3e02e1dSStanislav Fomichev for (i = CUSTOM_INHERIT1; i <= CUSTOM_LISTENER; i++) { 121e3e02e1dSStanislav Fomichev buf = 0x01; 122e3e02e1dSStanislav Fomichev err = setsockopt(fd, SOL_CUSTOM, i, &buf, 1); 123e3e02e1dSStanislav Fomichev if (err) { 124e3e02e1dSStanislav Fomichev log_err("Failed to call setsockopt(%d)", i); 125e3e02e1dSStanislav Fomichev close(fd); 126e3e02e1dSStanislav Fomichev return -1; 127e3e02e1dSStanislav Fomichev } 128e3e02e1dSStanislav Fomichev } 129e3e02e1dSStanislav Fomichev 130e3e02e1dSStanislav Fomichev if (bind(fd, (const struct sockaddr *)&addr, sizeof(addr)) < 0) { 131e3e02e1dSStanislav Fomichev log_err("Failed to bind socket"); 132e3e02e1dSStanislav Fomichev close(fd); 133e3e02e1dSStanislav Fomichev return -1; 134e3e02e1dSStanislav Fomichev } 135e3e02e1dSStanislav Fomichev 136e3e02e1dSStanislav Fomichev return fd; 137e3e02e1dSStanislav Fomichev } 138e3e02e1dSStanislav Fomichev 139e3e02e1dSStanislav Fomichev static int prog_attach(struct bpf_object *obj, int cgroup_fd, const char *title) 140e3e02e1dSStanislav Fomichev { 141e3e02e1dSStanislav Fomichev enum bpf_attach_type attach_type; 142e3e02e1dSStanislav Fomichev enum bpf_prog_type prog_type; 143e3e02e1dSStanislav Fomichev struct bpf_program *prog; 144e3e02e1dSStanislav Fomichev int err; 145e3e02e1dSStanislav Fomichev 146e3e02e1dSStanislav Fomichev err = libbpf_prog_type_by_name(title, &prog_type, &attach_type); 147e3e02e1dSStanislav Fomichev if (err) { 148e3e02e1dSStanislav Fomichev log_err("Failed to deduct types for %s BPF program", title); 149e3e02e1dSStanislav Fomichev return -1; 150e3e02e1dSStanislav Fomichev } 151e3e02e1dSStanislav Fomichev 152e3e02e1dSStanislav Fomichev prog = bpf_object__find_program_by_title(obj, title); 153e3e02e1dSStanislav Fomichev if (!prog) { 154e3e02e1dSStanislav Fomichev log_err("Failed to find %s BPF program", title); 155e3e02e1dSStanislav Fomichev return -1; 156e3e02e1dSStanislav Fomichev } 157e3e02e1dSStanislav Fomichev 158e3e02e1dSStanislav Fomichev err = bpf_prog_attach(bpf_program__fd(prog), cgroup_fd, 159e3e02e1dSStanislav Fomichev attach_type, 0); 160e3e02e1dSStanislav Fomichev if (err) { 161e3e02e1dSStanislav Fomichev log_err("Failed to attach %s BPF program", title); 162e3e02e1dSStanislav Fomichev return -1; 163e3e02e1dSStanislav Fomichev } 164e3e02e1dSStanislav Fomichev 165e3e02e1dSStanislav Fomichev return 0; 166e3e02e1dSStanislav Fomichev } 167e3e02e1dSStanislav Fomichev 168e3e02e1dSStanislav Fomichev static void run_test(int cgroup_fd) 169e3e02e1dSStanislav Fomichev { 170e3e02e1dSStanislav Fomichev struct bpf_prog_load_attr attr = { 171e3e02e1dSStanislav Fomichev .file = "./sockopt_inherit.o", 172e3e02e1dSStanislav Fomichev }; 173e3e02e1dSStanislav Fomichev int server_fd = -1, client_fd; 174e3e02e1dSStanislav Fomichev struct bpf_object *obj; 175e3e02e1dSStanislav Fomichev void *server_err; 176e3e02e1dSStanislav Fomichev pthread_t tid; 177e3e02e1dSStanislav Fomichev int ignored; 178e3e02e1dSStanislav Fomichev int err; 179e3e02e1dSStanislav Fomichev 180e3e02e1dSStanislav Fomichev err = bpf_prog_load_xattr(&attr, &obj, &ignored); 181e3e02e1dSStanislav Fomichev if (CHECK_FAIL(err)) 182e3e02e1dSStanislav Fomichev return; 183e3e02e1dSStanislav Fomichev 184e3e02e1dSStanislav Fomichev err = prog_attach(obj, cgroup_fd, "cgroup/getsockopt"); 185e3e02e1dSStanislav Fomichev if (CHECK_FAIL(err)) 186e3e02e1dSStanislav Fomichev goto close_bpf_object; 187e3e02e1dSStanislav Fomichev 188e3e02e1dSStanislav Fomichev err = prog_attach(obj, cgroup_fd, "cgroup/setsockopt"); 189e3e02e1dSStanislav Fomichev if (CHECK_FAIL(err)) 190e3e02e1dSStanislav Fomichev goto close_bpf_object; 191e3e02e1dSStanislav Fomichev 192e3e02e1dSStanislav Fomichev server_fd = start_server(); 193e3e02e1dSStanislav Fomichev if (CHECK_FAIL(server_fd < 0)) 194e3e02e1dSStanislav Fomichev goto close_bpf_object; 195e3e02e1dSStanislav Fomichev 196e81e7a53SStanislav Fomichev pthread_mutex_lock(&server_started_mtx); 197e3e02e1dSStanislav Fomichev if (CHECK_FAIL(pthread_create(&tid, NULL, server_thread, 198e3e02e1dSStanislav Fomichev (void *)&server_fd))) 199a2d074e4SBrian Vazquez goto close_server_fd; 200e3e02e1dSStanislav Fomichev pthread_cond_wait(&server_started, &server_started_mtx); 201e3e02e1dSStanislav Fomichev pthread_mutex_unlock(&server_started_mtx); 202e3e02e1dSStanislav Fomichev 203e3e02e1dSStanislav Fomichev client_fd = connect_to_server(server_fd); 204e3e02e1dSStanislav Fomichev if (CHECK_FAIL(client_fd < 0)) 205e3e02e1dSStanislav Fomichev goto close_server_fd; 206e3e02e1dSStanislav Fomichev 207e3e02e1dSStanislav Fomichev CHECK_FAIL(verify_sockopt(client_fd, CUSTOM_INHERIT1, "connect", 0)); 208e3e02e1dSStanislav Fomichev CHECK_FAIL(verify_sockopt(client_fd, CUSTOM_INHERIT2, "connect", 0)); 209e3e02e1dSStanislav Fomichev CHECK_FAIL(verify_sockopt(client_fd, CUSTOM_LISTENER, "connect", 0)); 210e3e02e1dSStanislav Fomichev 211e3e02e1dSStanislav Fomichev pthread_join(tid, &server_err); 212e3e02e1dSStanislav Fomichev 213e3e02e1dSStanislav Fomichev err = (int)(long)server_err; 214e3e02e1dSStanislav Fomichev CHECK_FAIL(err); 215e3e02e1dSStanislav Fomichev 216e3e02e1dSStanislav Fomichev close(client_fd); 217e3e02e1dSStanislav Fomichev 218e3e02e1dSStanislav Fomichev close_server_fd: 219e3e02e1dSStanislav Fomichev close(server_fd); 220e3e02e1dSStanislav Fomichev close_bpf_object: 221e3e02e1dSStanislav Fomichev bpf_object__close(obj); 222e3e02e1dSStanislav Fomichev } 223e3e02e1dSStanislav Fomichev 224e3e02e1dSStanislav Fomichev void test_sockopt_inherit(void) 225e3e02e1dSStanislav Fomichev { 226e3e02e1dSStanislav Fomichev int cgroup_fd; 227e3e02e1dSStanislav Fomichev 228e3e02e1dSStanislav Fomichev cgroup_fd = test__join_cgroup("/sockopt_inherit"); 229e3e02e1dSStanislav Fomichev if (CHECK_FAIL(cgroup_fd < 0)) 230e3e02e1dSStanislav Fomichev return; 231e3e02e1dSStanislav Fomichev 232e3e02e1dSStanislav Fomichev run_test(cgroup_fd); 233e3e02e1dSStanislav Fomichev close(cgroup_fd); 234e3e02e1dSStanislav Fomichev } 235