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 */ unused(void)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") BPF_PROG(test1,int a)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") BPF_PROG(test2,int a)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") test3(struct pt_regs * ctx)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") BPF_KRETPROBE(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") BPF_PROG(test5,int a,int * b,int ret)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") test6(struct pt_regs * ctx)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") BPF_UPROBE(test7)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") BPF_URETPROBE(test8,int ret)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 } 106