72 lines
1.8 KiB
C
72 lines
1.8 KiB
C
|
// SPDX-License-Identifier: GPL-2.0
|
||
|
|
/*
|
||
|
|
* Test weak ksyms.
|
||
|
|
*
|
||
|
|
* Copyright (c) 2021 Google
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "vmlinux.h"
|
||
|
|
|
||
|
|
#include <bpf/bpf_helpers.h>
|
||
|
|
|
||
|
|
int out__existing_typed = -1;
|
||
|
|
__u64 out__existing_typeless = -1;
|
||
|
|
|
||
|
|
__u64 out__non_existent_typeless = -1;
|
||
|
|
__u64 out__non_existent_typed = -1;
|
||
|
|
|
||
|
|
/* existing weak symbols */
|
||
|
|
|
||
|
|
/* test existing weak symbols can be resolved. */
|
||
|
|
extern const struct rq runqueues __ksym __weak; /* typed */
|
||
|
|
extern const void bpf_prog_active __ksym __weak; /* typeless */
|
||
|
|
struct task_struct *bpf_task_acquire(struct task_struct *p) __ksym __weak;
|
||
|
|
void bpf_testmod_test_mod_kfunc(int i) __ksym __weak;
|
||
|
|
|
||
|
|
|
||
|
|
/* non-existent weak symbols. */
|
||
|
|
|
||
|
|
/* typeless symbols, default to zero. */
|
||
|
|
extern const void bpf_link_fops1 __ksym __weak;
|
||
|
|
|
||
|
|
/* typed symbols, default to zero. */
|
||
|
|
extern const int bpf_link_fops2 __ksym __weak;
|
||
|
|
void invalid_kfunc(void) __ksym __weak;
|
||
|
|
|
||
|
|
SEC("raw_tp/sys_enter")
|
||
|
|
int pass_handler(const void *ctx)
|
||
|
|
{
|
||
|
|
struct rq *rq;
|
||
|
|
|
||
|
|
/* tests existing symbols. */
|
||
|
|
rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, 0);
|
||
|
|
if (rq && bpf_ksym_exists(&runqueues))
|
||
|
|
out__existing_typed = rq->cpu;
|
||
|
|
out__existing_typeless = (__u64)&bpf_prog_active;
|
||
|
|
|
||
|
|
/* tests non-existent symbols. */
|
||
|
|
out__non_existent_typeless = (__u64)&bpf_link_fops1;
|
||
|
|
|
||
|
|
/* tests non-existent symbols. */
|
||
|
|
out__non_existent_typed = (__u64)&bpf_link_fops2;
|
||
|
|
|
||
|
|
if (&bpf_link_fops2) /* can't happen */
|
||
|
|
out__non_existent_typed = (__u64)bpf_per_cpu_ptr(&bpf_link_fops2, 0);
|
||
|
|
|
||
|
|
if (!bpf_ksym_exists(bpf_task_acquire))
|
||
|
|
/* dead code won't be seen by the verifier */
|
||
|
|
bpf_task_acquire(0);
|
||
|
|
|
||
|
|
if (!bpf_ksym_exists(bpf_testmod_test_mod_kfunc))
|
||
|
|
/* dead code won't be seen by the verifier */
|
||
|
|
bpf_testmod_test_mod_kfunc(0);
|
||
|
|
|
||
|
|
if (bpf_ksym_exists(invalid_kfunc))
|
||
|
|
/* dead code won't be seen by the verifier */
|
||
|
|
invalid_kfunc();
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char _license[] SEC("license") = "GPL";
|