// SPDX-License-Identifier: GPL-2.0

#include <test_progs.h>
#include <bpf/bpf.h>
#include <errno.h>
#include <sys/mman.h>
#include <unistd.h>

#define ARENA_PAGES 32

static char log_buf[16384];

static void test_arena_direct_value_one_past_end(void)
{
        char expected[128];
        __u32 arena_sz = ARENA_PAGES * getpagesize();
        struct bpf_insn insns[] = {
                BPF_LD_IMM64_RAW(BPF_REG_1, BPF_PSEUDO_MAP_VALUE, 0),
                BPF_MOV64_IMM(BPF_REG_0, 0),
                BPF_EXIT_INSN(),
        };
        LIBBPF_OPTS(bpf_map_create_opts, map_opts);
        LIBBPF_OPTS(bpf_prog_load_opts, prog_opts);
        void *arena;
        int map_fd, prog_fd;

        map_opts.map_flags = BPF_F_MMAPABLE;
        prog_opts.log_buf = log_buf;
        prog_opts.log_size = sizeof(log_buf);
        prog_opts.log_level = 1;

        map_fd = bpf_map_create(BPF_MAP_TYPE_ARENA, "arena_direct_value",
                                0, 0, ARENA_PAGES, &map_opts);
        if (map_fd < 0) {
                if (errno == EOPNOTSUPP) {
                        test__skip();
                        return;
                }
                ASSERT_GE(map_fd, 0, "bpf_map_create");
                return;
        }

        arena = mmap(NULL, arena_sz, PROT_READ | PROT_WRITE, MAP_SHARED, map_fd, 0);
        if (!ASSERT_NEQ(arena, MAP_FAILED, "arena_mmap"))
                goto cleanup;

        insns[0].imm = map_fd;
        insns[1].imm = arena_sz;

        prog_fd = bpf_prog_load(BPF_PROG_TYPE_RAW_TRACEPOINT,
                                "arena_direct_value", "GPL", insns,
                                ARRAY_SIZE(insns), &prog_opts);
        if (!ASSERT_LT(prog_fd, 0, "prog_load")) {
                close(prog_fd);
                goto cleanup;
        }

        snprintf(expected, sizeof(expected),
                 "invalid access to map value pointer, value_size=0 off=%u",
                 arena_sz);
        ASSERT_HAS_SUBSTR(log_buf, expected, "verifier_log");

cleanup:
        if (arena != MAP_FAILED)
                munmap(arena, arena_sz);
        close(map_fd);
}

void test_arena_direct_value(void)
{
        if (test__start_subtest("one_past_end"))
                test_arena_direct_value_one_past_end();
}