// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */

#include "vmlinux.h"
#include "bpf_experimental.h"
#include <bpf/bpf_helpers.h>
#include "bpf_misc.h"

pid_t target_pid = 0;
unsigned int vmas_seen = 0;

struct {
        __u64 vm_start;
        __u64 vm_end;
} vm_ranges[1000];

SEC("raw_tp/sys_enter")
int iter_task_vma_for_each(const void *ctx)
{
        struct task_struct *task = bpf_get_current_task_btf();
        struct vm_area_struct *vma;
        unsigned int seen = 0;

        if (task->pid != target_pid)
                return 0;

        if (vmas_seen)
                return 0;

        bpf_for_each(task_vma, vma, task, 0) {
                if (bpf_cmp_unlikely(seen, >=, 1000))
                        break;

                vm_ranges[seen].vm_start = vma->vm_start;
                vm_ranges[seen].vm_end = vma->vm_end;
                seen++;
        }

        vmas_seen = seen;
        return 0;
}

char _license[] SEC("license") = "GPL";