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