mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
2dfc8186d6
Add some basic selftests for qspinlock built over BPF arena using cond_break_label macro. Signed-off-by: Kumar Kartikeya Dwivedi <memxor@gmail.com> Link: https://lore.kernel.org/r/20250306035431.2186189-4-memxor@gmail.com Signed-off-by: Alexei Starovoitov <ast@kernel.org>
51 lines
1.1 KiB
C
51 lines
1.1 KiB
C
// 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)))
|
|
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";
|