#include <linux/kconfig.h>
#include <linux/ptrace.h>
#include <linux/version.h>
#include <uapi/linux/bpf.h>
#include <linux/uio.h>
#include <linux/types.h>

static void *(*bpf_map_lookup_elem)(void *map, void *key) =
    (void *)1;
static int (*bpf_map_update_elem)(void *map, void *key, void *value, unsigned long long flags) =
    (void *)2;
static unsigned long long (*bpf_get_current_pid_tgid)(void) =
    (void *)14;
static unsigned long long (*bpf_tail_call)(void *ctx, void *map, void *index) =
    (void *)12;

struct map_t
{
    u32 map_type;
    u32 key_size;
    u32 value_size;
    u32 max_entries;
};

struct map_t __test_map __attribute__((section("maps/test_map"), used)) = {
    BPF_MAP_TYPE_ARRAY,
    sizeof(u32),
    sizeof(u32),
    1024};

struct map_t __map_combined_section1 __attribute__((section("maps"), used)) = {
    BPF_MAP_TYPE_ARRAY,
    sizeof(u32),
    8,
    1024};

struct map_t __map_combined_section2 __attribute__((section("maps"), used)) = {
    BPF_MAP_TYPE_ARRAY,
    sizeof(u32),
    12,
    1024};

struct map_t __test_hash_map __attribute__((section("maps/test_hash_map"), used)) = {
    BPF_MAP_TYPE_HASH,
    sizeof(u64),
    sizeof(u64),
    1024};

struct map_t __test_tailcall_map __attribute__((section("maps/test_tailcall"), used)) = {
        BPF_MAP_TYPE_PROG_ARRAY,
        4,
        4,
        32};

__attribute__((section("kprobe/test_program"), used)) int test_program(struct pt_regs *regs)
{
    (void)bpf_get_current_pid_tgid();
    return 0;
}

__attribute__((section("kprobe/test_program_tailcall_update_map"), used)) int test_program_tailcall_update_map(void *ctx)
{
    u32 value = 111;
    u32 idx = 150;
    bpf_map_update_elem(&__test_map, &idx, &value, BPF_ANY);
    return 0;
}

__attribute__((section("kprobe/test_program_tailcall"), used)) int test_program_tailcall(struct pt_regs *regs)
{
    bpf_tail_call(regs, &__test_tailcall_map, 0);
    return 0;
}

__attribute__((section("kprobe/test_program_map_update"), used)) int test_program_map_update(struct pt_regs *regs)
{
    u32 index = 0;
    u64 key = 0x12345;
    u32 *value = bpf_map_lookup_elem(&__test_map, &index);
    if (!value)
    {
        return 0;
    }
    else
    {
        u32 new_value = 1234;
        u64 new_value64 = 1234;
        bpf_map_update_elem(&__test_map, &index, &new_value, BPF_ANY);
        bpf_map_update_elem(&__test_hash_map, &key, &new_value64, BPF_ANY);
    }
    return 0;
}

char _license[] __attribute__((section("license"), used)) = "Proprietary";
uint32_t _version __attribute__((section("version"), used)) = 0xFFFFFFFE;