1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3 #include "bpf_iter.h"
4 #include <bpf/bpf_helpers.h>
5 
6 char _license[] SEC("license") = "GPL";
7 
8 __u32 unique_tgid_cnt = 0;
9 uintptr_t address = 0;
10 uintptr_t offset = 0;
11 __u32 last_tgid = 0;
12 __u32 pid = 0;
13 __u32 page_shift = 0;
14 
15 SEC("iter/task_vma")
get_vma_offset(struct bpf_iter__task_vma * ctx)16 int get_vma_offset(struct bpf_iter__task_vma *ctx)
17 {
18 	struct vm_area_struct *vma = ctx->vma;
19 	struct seq_file *seq = ctx->meta->seq;
20 	struct task_struct *task = ctx->task;
21 
22 	if (task == NULL || vma == NULL)
23 		return 0;
24 
25 	if (last_tgid != task->tgid)
26 		unique_tgid_cnt++;
27 	last_tgid = task->tgid;
28 
29 	if (task->tgid != pid)
30 		return 0;
31 
32 	if (vma->vm_start <= address && vma->vm_end > address) {
33 		offset = address - vma->vm_start + (vma->vm_pgoff << page_shift);
34 		BPF_SEQ_PRINTF(seq, "OK\n");
35 	}
36 	return 0;
37 }
38