summaryrefslogtreecommitdiffstats
path: root/tools/testing/selftests/bpf/progs/bpf_iter_vma_offset.c
blob: ee7455d2623aab28d307c10d09c4bd096b6de8d3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
#include "bpf_iter.h"
#include <bpf/bpf_helpers.h>

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

__u32 unique_tgid_cnt = 0;
uintptr_t address = 0;
uintptr_t offset = 0;
__u32 last_tgid = 0;
__u32 pid = 0;
__u32 page_shift = 0;

SEC("iter/task_vma")
int get_vma_offset(struct bpf_iter__task_vma *ctx)
{
	struct vm_area_struct *vma = ctx->vma;
	struct seq_file *seq = ctx->meta->seq;
	struct task_struct *task = ctx->task;

	if (task == NULL || vma == NULL)
		return 0;

	if (last_tgid != task->tgid)
		unique_tgid_cnt++;
	last_tgid = task->tgid;

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

	if (vma->vm_start <= address && vma->vm_end > address) {
		offset = address - vma->vm_start + (vma->vm_pgoff << page_shift);
		BPF_SEQ_PRINTF(seq, "OK\n");
	}
	return 0;
}