1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2024-2025 Meta Platforms, Inc. and affiliates. */ 3 #include <test_progs.h> 4 #include <network_helpers.h> 5 #include <sys/sysinfo.h> 6 7 #include "res_spin_lock.skel.h" 8 #include "res_spin_lock_fail.skel.h" 9 10 void test_res_spin_lock_failure(void) 11 { 12 RUN_TESTS(res_spin_lock_fail); 13 } 14 15 static volatile int skip; 16 17 static void *spin_lock_thread(void *arg) 18 { 19 int err, prog_fd = *(u32 *) arg; 20 LIBBPF_OPTS(bpf_test_run_opts, topts, 21 .data_in = &pkt_v4, 22 .data_size_in = sizeof(pkt_v4), 23 .repeat = 10000, 24 ); 25 26 while (!READ_ONCE(skip)) { 27 err = bpf_prog_test_run_opts(prog_fd, &topts); 28 ASSERT_OK(err, "test_run"); 29 ASSERT_OK(topts.retval, "test_run retval"); 30 } 31 pthread_exit(arg); 32 } 33 34 void test_res_spin_lock_success(void) 35 { 36 LIBBPF_OPTS(bpf_test_run_opts, topts, 37 .data_in = &pkt_v4, 38 .data_size_in = sizeof(pkt_v4), 39 .repeat = 1, 40 ); 41 struct res_spin_lock *skel; 42 pthread_t thread_id[16]; 43 int prog_fd, i, err; 44 void *ret; 45 46 if (get_nprocs() < 2) { 47 test__skip(); 48 return; 49 } 50 51 skel = res_spin_lock__open_and_load(); 52 if (!ASSERT_OK_PTR(skel, "res_spin_lock__open_and_load")) 53 return; 54 /* AA deadlock */ 55 prog_fd = bpf_program__fd(skel->progs.res_spin_lock_test); 56 err = bpf_prog_test_run_opts(prog_fd, &topts); 57 ASSERT_OK(err, "error"); 58 ASSERT_OK(topts.retval, "retval"); 59 60 prog_fd = bpf_program__fd(skel->progs.res_spin_lock_test_held_lock_max); 61 err = bpf_prog_test_run_opts(prog_fd, &topts); 62 ASSERT_OK(err, "error"); 63 ASSERT_OK(topts.retval, "retval"); 64 65 /* Multi-threaded ABBA deadlock. */ 66 67 prog_fd = bpf_program__fd(skel->progs.res_spin_lock_test_AB); 68 for (i = 0; i < 16; i++) { 69 int err; 70 71 err = pthread_create(&thread_id[i], NULL, &spin_lock_thread, &prog_fd); 72 if (!ASSERT_OK(err, "pthread_create")) 73 goto end; 74 } 75 76 topts.retval = 0; 77 topts.repeat = 1000; 78 int fd = bpf_program__fd(skel->progs.res_spin_lock_test_BA); 79 while (!topts.retval && !err && !READ_ONCE(skel->bss->err)) { 80 err = bpf_prog_test_run_opts(fd, &topts); 81 } 82 83 WRITE_ONCE(skip, true); 84 85 for (i = 0; i < 16; i++) { 86 if (!ASSERT_OK(pthread_join(thread_id[i], &ret), "pthread_join")) 87 goto end; 88 if (!ASSERT_EQ(ret, &prog_fd, "ret == prog_fd")) 89 goto end; 90 } 91 92 ASSERT_EQ(READ_ONCE(skel->bss->err), -EDEADLK, "timeout err"); 93 ASSERT_OK(err, "err"); 94 ASSERT_EQ(topts.retval, -EDEADLK, "timeout"); 95 end: 96 res_spin_lock__destroy(skel); 97 return; 98 } 99