Quellcode-Bibliothek test_ksyms_btf.c

// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2020 Google */

#include "vmlinux.h"

#include <bpf/bpf_helpers.h>

__u64 out__runqueues_addr = -1;
__u64 out__bpf_prog_active_addr = -1;

__u32 out__rq_cpu = -1; /* percpu struct fields */
int out__bpf_prog_active = -1; /* percpu int */

__u32 out__this_rq_cpu = -1;
int out__this_bpf_prog_active = -1;

__u32 out__cpu_0_rq_cpu = -1; /* cpu_rq(0)->cpu */

extern const struct rq runqueues __ksym; /* struct type global var. */
extern const int bpf_prog_active __ksym; /* int type global var. */

SEC("raw_tp/sys_enter")
int handler(const void *ctx)
{
 struct rq *rq;
 int *active;
 __u32 cpu;

 out__runqueues_addr = (__u64)&runqueues;
 out__bpf_prog_active_addr = (__u64)&bpf_prog_active;

 cpu = bpf_get_smp_processor_id();

 /* test bpf_per_cpu_ptr() */
 rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, cpu);
 if (rq)
  out__rq_cpu = rq->cpu;
 active = (int *)bpf_per_cpu_ptr(&bpf_prog_active, cpu);
 if (active)
  out__bpf_prog_active = *active;

 rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, 0);
 if (rq) /* should always be valid, but we can't spare the check. */
  out__cpu_0_rq_cpu = rq->cpu;

 /* test bpf_this_cpu_ptr */
 rq = (struct rq *)bpf_this_cpu_ptr(&runqueues);
 out__this_rq_cpu = rq->cpu;
 active = (int *)bpf_this_cpu_ptr(&bpf_prog_active);
 out__this_bpf_prog_active = *active;

 return 0;
}

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