1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/bpf.h> 3 #include <bpf/bpf_helpers.h> 4 #include <bpf/bpf_tracing.h> 5 #include <stdbool.h> 6 #include "bpf_kfuncs.h" 7 8 char _license[] SEC("license") = "GPL"; 9 10 int pid = 0; 11 12 __u64 test_kprobe_1_result = 0; 13 __u64 test_kprobe_2_result = 0; 14 __u64 test_kprobe_3_result = 0; 15 16 /* 17 * No tests in here, just to trigger 'bpf_fentry_test*' 18 * through tracing test_run 19 */ 20 SEC("fentry/bpf_modify_return_test") 21 int BPF_PROG(trigger) 22 { 23 return 0; 24 } 25 26 static int check_cookie(__u64 val, __u64 *result) 27 { 28 __u64 *cookie; 29 30 if (bpf_get_current_pid_tgid() >> 32 != pid) 31 return 1; 32 33 cookie = bpf_session_cookie(); 34 35 if (bpf_session_is_return()) 36 *result = *cookie == val ? val : 0; 37 else 38 *cookie = val; 39 return 0; 40 } 41 42 SEC("kprobe.session/bpf_fentry_test1") 43 int test_kprobe_1(struct pt_regs *ctx) 44 { 45 return check_cookie(1, &test_kprobe_1_result); 46 } 47 48 SEC("kprobe.session/bpf_fentry_test1") 49 int test_kprobe_2(struct pt_regs *ctx) 50 { 51 return check_cookie(2, &test_kprobe_2_result); 52 } 53 54 SEC("kprobe.session/bpf_fentry_test1") 55 int test_kprobe_3(struct pt_regs *ctx) 56 { 57 return check_cookie(3, &test_kprobe_3_result); 58 } 59