e43163ed1c
Adding get_func_ip tests for uprobe on function entry that validates that bpf_get_func_ip returns proper values from both uprobe and return uprobe. Tested-by: Alan Maguire <alan.maguire@oracle.com> Signed-off-by: Jiri Olsa <jolsa@kernel.org> Acked-by: Yonghong Song <yonghong.song@linux.dev> Link: https://lore.kernel.org/r/20230807085956.2344866-3-jolsa@kernel.org Signed-off-by: Martin KaFai Lau <martin.lau@kernel.org>
106 lines
2.3 KiB
C
106 lines
2.3 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
#include "vmlinux.h"
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
extern const void bpf_fentry_test1 __ksym;
|
|
extern const void bpf_fentry_test2 __ksym;
|
|
extern const void bpf_fentry_test3 __ksym;
|
|
extern const void bpf_fentry_test4 __ksym;
|
|
extern const void bpf_modify_return_test __ksym;
|
|
extern const void bpf_fentry_test6 __ksym;
|
|
extern const void bpf_fentry_test7 __ksym;
|
|
|
|
extern bool CONFIG_X86_KERNEL_IBT __kconfig __weak;
|
|
|
|
/* This function is here to have CONFIG_X86_KERNEL_IBT
|
|
* used and added to object BTF.
|
|
*/
|
|
int unused(void)
|
|
{
|
|
return CONFIG_X86_KERNEL_IBT ? 0 : 1;
|
|
}
|
|
|
|
__u64 test1_result = 0;
|
|
SEC("fentry/bpf_fentry_test1")
|
|
int BPF_PROG(test1, int a)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test1_result = (const void *) addr == &bpf_fentry_test1;
|
|
return 0;
|
|
}
|
|
|
|
__u64 test2_result = 0;
|
|
SEC("fexit/bpf_fentry_test2")
|
|
int BPF_PROG(test2, int a)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test2_result = (const void *) addr == &bpf_fentry_test2;
|
|
return 0;
|
|
}
|
|
|
|
__u64 test3_result = 0;
|
|
SEC("kprobe/bpf_fentry_test3")
|
|
int test3(struct pt_regs *ctx)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test3_result = (const void *) addr == &bpf_fentry_test3;
|
|
return 0;
|
|
}
|
|
|
|
__u64 test4_result = 0;
|
|
SEC("kretprobe/bpf_fentry_test4")
|
|
int BPF_KRETPROBE(test4)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test4_result = (const void *) addr == &bpf_fentry_test4;
|
|
return 0;
|
|
}
|
|
|
|
__u64 test5_result = 0;
|
|
SEC("fmod_ret/bpf_modify_return_test")
|
|
int BPF_PROG(test5, int a, int *b, int ret)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test5_result = (const void *) addr == &bpf_modify_return_test;
|
|
return ret;
|
|
}
|
|
|
|
__u64 test6_result = 0;
|
|
SEC("?kprobe")
|
|
int test6(struct pt_regs *ctx)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test6_result = (const void *) addr == 0;
|
|
return 0;
|
|
}
|
|
|
|
unsigned long uprobe_trigger;
|
|
|
|
__u64 test7_result = 0;
|
|
SEC("uprobe//proc/self/exe:uprobe_trigger")
|
|
int BPF_UPROBE(test7)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test7_result = (const void *) addr == (const void *) uprobe_trigger;
|
|
return 0;
|
|
}
|
|
|
|
__u64 test8_result = 0;
|
|
SEC("uretprobe//proc/self/exe:uprobe_trigger")
|
|
int BPF_URETPROBE(test8, int ret)
|
|
{
|
|
__u64 addr = bpf_get_func_ip(ctx);
|
|
|
|
test8_result = (const void *) addr == (const void *) uprobe_trigger;
|
|
return 0;
|
|
}
|