1 | // SPDX-License-Identifier: GPL-2.0 |
2 | /* Copyright (c) 2020 Google */ |
3 | |
4 | #include "vmlinux.h" |
5 | |
6 | #include <bpf/bpf_helpers.h> |
7 | |
8 | __u64 out__runqueues_addr = -1; |
9 | __u64 out__bpf_prog_active_addr = -1; |
10 | |
11 | __u32 out__rq_cpu = -1; /* percpu struct fields */ |
12 | int out__bpf_prog_active = -1; /* percpu int */ |
13 | |
14 | __u32 out__this_rq_cpu = -1; |
15 | int out__this_bpf_prog_active = -1; |
16 | |
17 | __u32 out__cpu_0_rq_cpu = -1; /* cpu_rq(0)->cpu */ |
18 | |
19 | extern const struct rq runqueues __ksym; /* struct type global var. */ |
20 | extern const int bpf_prog_active __ksym; /* int type global var. */ |
21 | |
22 | SEC("raw_tp/sys_enter" ) |
23 | int handler(const void *ctx) |
24 | { |
25 | struct rq *rq; |
26 | int *active; |
27 | __u32 cpu; |
28 | |
29 | out__runqueues_addr = (__u64)&runqueues; |
30 | out__bpf_prog_active_addr = (__u64)&bpf_prog_active; |
31 | |
32 | cpu = bpf_get_smp_processor_id(); |
33 | |
34 | /* test bpf_per_cpu_ptr() */ |
35 | rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, cpu); |
36 | if (rq) |
37 | out__rq_cpu = rq->cpu; |
38 | active = (int *)bpf_per_cpu_ptr(&bpf_prog_active, cpu); |
39 | if (active) |
40 | out__bpf_prog_active = *active; |
41 | |
42 | rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, 0); |
43 | if (rq) /* should always be valid, but we can't spare the check. */ |
44 | out__cpu_0_rq_cpu = rq->cpu; |
45 | |
46 | /* test bpf_this_cpu_ptr */ |
47 | rq = (struct rq *)bpf_this_cpu_ptr(&runqueues); |
48 | out__this_rq_cpu = rq->cpu; |
49 | active = (int *)bpf_this_cpu_ptr(&bpf_prog_active); |
50 | out__this_bpf_prog_active = *active; |
51 | |
52 | return 0; |
53 | } |
54 | |
55 | char _license[] SEC("license" ) = "GPL" ; |
56 | |