tools/testing/selftests/bpf/progs/test_ksyms_weak.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Test weak ksyms.
   4  *
   5  * Copyright (c) 2021 Google
   6  */
   7
   8 #include "vmlinux.h"
   9
  10 #include <bpf/bpf_helpers.h>
  11
  12 int out__existing_typed = -1;
  13 __u64 out__existing_typeless = -1;
  14
  15 __u64 out__non_existent_typeless = -1;
  16 __u64 out__non_existent_typed = -1;
  17
  18 /* existing weak symbols */
  19
  20 /* test existing weak symbols can be resolved. */
  21 extern const struct rq runqueues __ksym __weak; /* typed */
  22 extern const void bpf_prog_active __ksym __weak; /* typeless */
  23 struct task_struct *bpf_task_acquire(struct task_struct *p) __ksym __weak;
  24 void bpf_testmod_test_mod_kfunc(int i) __ksym __weak;
  25
  26
  27 /* non-existent weak symbols. */
  28
  29 /* typeless symbols, default to zero. */
  30 extern const void bpf_link_fops1 __ksym __weak;
  31
  32 /* typed symbols, default to zero. */
  33 extern const int bpf_link_fops2 __ksym __weak;
  34 void invalid_kfunc(void) __ksym __weak;
  35
  36 SEC("raw_tp/sys_enter")
  37 int pass_handler(const void *ctx)
  38 {
  39         struct rq *rq;
  40
  41         /* tests existing symbols. */
  42         rq = (struct rq *)bpf_per_cpu_ptr(&runqueues, 0);
  43         if (rq && bpf_ksym_exists(&runqueues))
  44                 out__existing_typed = rq->cpu;
  45         out__existing_typeless = (__u64)&bpf_prog_active;
  46
  47         /* tests non-existent symbols. */
  48         out__non_existent_typeless = (__u64)&bpf_link_fops1;
  49
  50         /* tests non-existent symbols. */
  51         out__non_existent_typed = (__u64)&bpf_link_fops2;
  52
  53         if (&bpf_link_fops2) /* can't happen */
  54                 out__non_existent_typed = (__u64)bpf_per_cpu_ptr(&bpf_link_fops2, 0);
  55
  56         if (!bpf_ksym_exists(bpf_task_acquire))
  57                 /* dead code won't be seen by the verifier */
  58                 bpf_task_acquire(0);
  59
  60         if (!bpf_ksym_exists(bpf_testmod_test_mod_kfunc))
  61                 /* dead code won't be seen by the verifier */
  62                 bpf_testmod_test_mod_kfunc(0);
  63
  64         if (bpf_ksym_exists(invalid_kfunc))
  65                 /* dead code won't be seen by the verifier */
  66                 invalid_kfunc();
  67
  68         return 0;
  69 }
  70
  71 char _license[] SEC("license") = "GPL";