// SPDX-License-Identifier: GPL-2.0 /* Copyright (c) 2025 Meta Platforms, Inc. and affiliates. */ #include <vmlinux.h> #include <bpf/bpf_tracing.h> #include <bpf/bpf_helpers.h> #include "bpf_misc.h" #include "bpf_arena_spin_lock.h" struct { __uint(type, BPF_MAP_TYPE_ARENA); __uint(map_flags, BPF_F_MMAPABLE); __uint(max_entries, 100); /* number of pages */ #ifdef __TARGET_ARCH_arm64 __ulong(map_extra, 0x1ull << 32); /* start of mmap() region */ #else __ulong(map_extra, 0x1ull << 44); /* start of mmap() region */ #endif } arena SEC(".maps"); int cs_count; #if defined(ENABLE_ATOMICS_TESTS) && defined(__BPF_FEATURE_ADDR_SPACE_CAST) arena_spinlock_t __arena lock; int test_skip = 1; #else int test_skip = 2; #endif int counter; int limit; SEC("tc") int prog(void *ctx) { int ret = -2; #if defined(ENABLE_ATOMICS_TESTS) && defined(__BPF_FEATURE_ADDR_SPACE_CAST) unsigned long flags; if ((ret = arena_spin_lock_irqsave(&lock, flags))) { if (ret == -EOPNOTSUPP) test_skip = 3; return ret; } if (counter != limit) counter++; bpf_repeat(cs_count); ret = 0; arena_spin_unlock_irqrestore(&lock, flags); #endif return ret; } char _license[] SEC("license") = "GPL";