// 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";