This set provides an implementation of queued spin lock for arena. There is no support for resiliency and recovering from deadlocks yet. We will wait for the rqspinlock patch set to land before incorporating support. One minor change compared to the qspinlock algorithm in the kernel is that we don't have the trylock fallback when nesting count exceeds 4. The maximum number of supported CPUs is 1024, but this can be increased in the future if necessary. The API supports returning an error, so resiliency support can be added in the future. Callers are still expected to check for and handle any potential errors. Errors are returned when the spin loops time out, when the number of CPUs is greater than 1024, or when the extreme edge case of NMI interrupting NMI interrupting HardIRQ interrupting SoftIRQ task, all of them simultaneously in slow path, occurs, which is unsupported. Changelog: ---------- v2 -> v3 v2: https://lore.kernel.org/bpf/20250118162238.2621311-1-memxor@xxxxxxxxx * Rename to arena_spin_lock * Introduce cond_break_label macro to jump to label from cond_break. * Drop trylock fallback when nesting count exceeds 4. * Fix bug in try_cmpxchg implementation. * Add tests with critical sections of varying lengths. * Add comments for _Generic trick to drop __arena tag. * Fix bug due to qnodes being placed on first page, leading to CPU 0's node being indistinguishable from NULL. v1 -> v2 v1: https://lore.kernel.org/bpf/20250117223754.1020174-1-memxor@xxxxxxxxx * Fix definition of lock in selftest Kumar Kartikeya Dwivedi (3): selftests/bpf: Introduce cond_break_label selftests/bpf: Introduce arena spin lock selftests/bpf: Add tests for arena spin lock .../selftests/bpf/bpf_arena_spin_lock.h | 505 ++++++++++++++++++ tools/testing/selftests/bpf/bpf_atomic.h | 132 +++++ .../testing/selftests/bpf/bpf_experimental.h | 15 +- .../bpf/prog_tests/arena_spin_lock.c | 102 ++++ .../selftests/bpf/progs/arena_spin_lock.c | 51 ++ 5 files changed, 799 insertions(+), 6 deletions(-) create mode 100644 tools/testing/selftests/bpf/bpf_arena_spin_lock.h create mode 100644 tools/testing/selftests/bpf/bpf_atomic.h create mode 100644 tools/testing/selftests/bpf/prog_tests/arena_spin_lock.c create mode 100644 tools/testing/selftests/bpf/progs/arena_spin_lock.c base-commit: 42ba8a49d085e0c2ad50fb9a8ec954c9762b6e01 -- 2.47.1
