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