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

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include "vmlinux.h"
   3 #include <bpf/bpf_helpers.h>
   4 #include <bpf/bpf_tracing.h>
   5
   6 char _license[] SEC("license") = "GPL";
   7
   8 extern int bpf_fentry_test1(int a) __ksym;
   9 extern int bpf_modify_return_test(int a, int *b) __ksym;
  10
  11 extern const void bpf_fentry_test2 __ksym;
  12 extern const void bpf_fentry_test3 __ksym;
  13 extern const void bpf_fentry_test4 __ksym;
  14
  15 extern bool CONFIG_X86_KERNEL_IBT __kconfig __weak;
  16
  17 /* This function is here to have CONFIG_X86_KERNEL_IBT
  18  * used and added to object BTF.
  19  */
  20 int unused(void)
  21 {
  22         return CONFIG_X86_KERNEL_IBT ? 0 : 1;
  23 }
  24
  25 __u64 test1_result = 0;
  26 SEC("fentry/bpf_fentry_test1")
  27 int BPF_PROG(test1, int a)
  28 {
  29         __u64 addr = bpf_get_func_ip(ctx);
  30
  31         test1_result = (const void *) addr == &bpf_fentry_test1;
  32         return 0;
  33 }
  34
  35 __u64 test2_result = 0;
  36 SEC("fexit/bpf_fentry_test2")
  37 int BPF_PROG(test2, int a)
  38 {
  39         __u64 addr = bpf_get_func_ip(ctx);
  40
  41         test2_result = (const void *) addr == &bpf_fentry_test2;
  42         return 0;
  43 }
  44
  45 __u64 test3_result = 0;
  46 SEC("kprobe/bpf_fentry_test3")
  47 int test3(struct pt_regs *ctx)
  48 {
  49         __u64 addr = bpf_get_func_ip(ctx);
  50
  51         test3_result = (const void *) addr == &bpf_fentry_test3;
  52         return 0;
  53 }
  54
  55 __u64 test4_result = 0;
  56 SEC("kretprobe/bpf_fentry_test4")
  57 int BPF_KRETPROBE(test4)
  58 {
  59         __u64 addr = bpf_get_func_ip(ctx);
  60
  61         test4_result = (const void *) addr == &bpf_fentry_test4;
  62         return 0;
  63 }
  64
  65 __u64 test5_result = 0;
  66 SEC("fmod_ret/bpf_modify_return_test")
  67 int BPF_PROG(test5, int a, int *b, int ret)
  68 {
  69         __u64 addr = bpf_get_func_ip(ctx);
  70
  71         test5_result = (const void *) addr == &bpf_modify_return_test;
  72         return ret;
  73 }
  74
  75 __u64 test6_result = 0;
  76 SEC("?kprobe")
  77 int test6(struct pt_regs *ctx)
  78 {
  79         __u64 addr = bpf_get_func_ip(ctx);
  80
  81         test6_result = (const void *) addr == 0;
  82         return 0;
  83 }
  84
  85 unsigned long uprobe_trigger;
  86
  87 __u64 test7_result = 0;
  88 SEC("uprobe//proc/self/exe:uprobe_trigger")
  89 int BPF_UPROBE(test7)
  90 {
  91         __u64 addr = bpf_get_func_ip(ctx);
  92
  93         test7_result = (const void *) addr == (const void *) uprobe_trigger;
  94         return 0;
  95 }
  96
  97 __u64 test8_result = 0;
  98 SEC("uretprobe//proc/self/exe:uprobe_trigger")
  99 int BPF_URETPROBE(test8, int ret)
 100 {
 101         __u64 addr = bpf_get_func_ip(ctx);
 102
 103         test8_result = (const void *) addr == (const void *) uprobe_trigger;
 104         return 0;
 105 }