[PATCHv2 RFC bpf-next 7/7] selftests/bpf: Add ftrace probe test
From: Jiri Olsa <jolsa@kernel.org>
Date: 2021-04-13 12:16:18
Also in:
bpf
Subsystem:
bpf [general] (safe dynamic programs and tools), bpf [selftests] (test runners & infrastructure), kernel selftest framework, the rest · Maintainers:
Alexei Starovoitov, Daniel Borkmann, Andrii Nakryiko, Eduard Zingerman, Kumar Kartikeya Dwivedi, Shuah Khan, Linus Torvalds
Adding simple ftrace probe test that configures ftrace probe and verifies the 'ip' argument matches the probed functions addresses. Signed-off-by: Jiri Olsa <jolsa@kernel.org> --- .../selftests/bpf/prog_tests/ftrace_test.c | 48 +++++++++++++++++++ .../testing/selftests/bpf/progs/ftrace_test.c | 17 +++++++ 2 files changed, 65 insertions(+) create mode 100644 tools/testing/selftests/bpf/prog_tests/ftrace_test.c create mode 100644 tools/testing/selftests/bpf/progs/ftrace_test.c
diff --git a/tools/testing/selftests/bpf/prog_tests/ftrace_test.c b/tools/testing/selftests/bpf/prog_tests/ftrace_test.c
new file mode 100644
index 000000000000..34d6dbe88251
--- /dev/null
+++ b/tools/testing/selftests/bpf/prog_tests/ftrace_test.c@@ -0,0 +1,48 @@ +// SPDX-License-Identifier: GPL-2.0 +#include <test_progs.h> +#include "ftrace_test.skel.h" + +void test_ftrace_test(void) +{ + struct ftrace_test *ftrace_skel = NULL; + __u32 duration = 0, retval; + int err, prog_fd; + __u64 *ips; + int idx, i; + + ftrace_skel = ftrace_test__open_and_load(); + if (!ASSERT_OK_PTR(ftrace_skel, "ftrace_skel_load")) + goto cleanup; + + err = ftrace_test__attach(ftrace_skel); + if (!ASSERT_OK(err, "ftrace_attach")) + goto cleanup; + + prog_fd = bpf_program__fd(ftrace_skel->progs.test); + err = bpf_prog_test_run(prog_fd, 1, NULL, 0, + NULL, NULL, &retval, &duration); + ASSERT_OK(err || retval, "test_run"); + + ips = ftrace_skel->bss->ips; + idx = ftrace_skel->bss->idx; + + if (!ASSERT_EQ(idx, 8, "idx")) + goto cleanup; + + for (i = 0; i < 8; i++) { + unsigned long long addr; + char func[50]; + + snprintf(func, sizeof(func), "bpf_fentry_test%d", i + 1); + + err = kallsyms_find(func, &addr); + if (!ASSERT_OK(err, "kallsyms_find")) + goto cleanup; + + if (!ASSERT_EQ(ips[i], addr, "ips_addr")) + goto cleanup; + } + +cleanup: + ftrace_test__destroy(ftrace_skel); +}
diff --git a/tools/testing/selftests/bpf/progs/ftrace_test.c b/tools/testing/selftests/bpf/progs/ftrace_test.c
new file mode 100644
index 000000000000..b2a55aa10318
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/ftrace_test.c@@ -0,0 +1,17 @@ +// SPDX-License-Identifier: GPL-2.0 +#include <linux/bpf.h> +#include <bpf/bpf_helpers.h> +#include <bpf/bpf_tracing.h> + +char _license[] SEC("license") = "GPL"; + +__u64 ips[9] = { }; +unsigned int idx = 0; + +SEC("fentry.ftrace/bpf_fentry_test*") +int BPF_PROG(test, __u64 ip, __u64 parent_ip) +{ + if (idx >= 0 && idx < 8) + ips[idx++] = ip; + return 0; +}
--
2.30.2