// 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";
