738c345b74
With CONFIG_X86_KERNEL_IBT enabled the test for kprobe with offset won't work because of the extra endbr instruction. As suggested by Andrii adding CONFIG_X86_KERNEL_IBT detection and using appropriate offset value based on that. Also removing test7 program, because it does the same as test6. Suggested-by: Andrii Nakryiko <andrii@kernel.org> Acked-by: Andrii Nakryiko <andrii@kernel.org> Signed-off-by: Jiri Olsa <jolsa@kernel.org> Link: https://lore.kernel.org/r/20220926153340.1621984-7-jolsa@kernel.org Signed-off-by: Alexei Starovoitov <ast@kernel.org>
86 lines
1.8 KiB
C
86 lines
1.8 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
#include <linux/bpf.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
#include <stdbool.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;
|
|
}
|