Attach to two bpf progs with fentry.multi, fexit.multi, and fsession.multi progs.
Assisted-by: Codex:gpt-5.5 Signed-off-by: Leon Hwang <[email protected]> --- .../selftests/bpf/prog_tests/tracing_multi.c | 70 +++++++++++++++++++ .../selftests/bpf/progs/tracing_multi_bpf.c | 52 ++++++++++++++ 2 files changed, 122 insertions(+) create mode 100644 tools/testing/selftests/bpf/progs/tracing_multi_bpf.c diff --git a/tools/testing/selftests/bpf/prog_tests/tracing_multi.c b/tools/testing/selftests/bpf/prog_tests/tracing_multi.c index 0aa9532a05cf..5ecb751af9be 100644 --- a/tools/testing/selftests/bpf/prog_tests/tracing_multi.c +++ b/tools/testing/selftests/bpf/prog_tests/tracing_multi.c @@ -12,6 +12,7 @@ #include "tracing_multi_verifier.skel.h" #include "tracing_multi_bench.skel.h" #include "tracing_multi_rollback.skel.h" +#include "tracing_multi_bpf.skel.h" #include "trace_helpers.h" static __u64 bpf_fentry_test_cookies[] = { @@ -288,6 +289,73 @@ static void test_link_api_ids(bool test_cookies) free(ids); } +static int run_bpf_target(struct bpf_program *prog, __u32 retval, const char *name) +{ + LIBBPF_OPTS(bpf_test_run_opts, topts); + int err; + + err = bpf_prog_test_run_opts(bpf_program__fd(prog), &topts); + if (!ASSERT_OK(err, name)) + return err; + + if (!ASSERT_EQ(topts.retval, retval, name)) + return -EINVAL; + + return 0; +} + +static void test_link_api_bpf_prog(void) +{ + const char *funcs[] = { "target_1", "target_2" }; + LIBBPF_OPTS(bpf_tracing_multi_opts, opts); + struct tracing_multi_bpf *skel = NULL; + int fds[ARRAY_SIZE(funcs)]; + struct bpf_link *link; + + skel = tracing_multi_bpf__open_and_load(); + if (!ASSERT_OK_PTR(skel, "tracing_multi_bpf__open_and_load")) + return; + + fds[0] = bpf_program__fd(skel->progs.target_1); + fds[1] = bpf_program__fd(skel->progs.target_2); + + opts.fds = fds; + opts.funcs = funcs; + opts.cnt = ARRAY_SIZE(fds); + + link = bpf_program__attach_tracing_multi(skel->progs.test_fentry, NULL, &opts); + if (!ASSERT_OK_PTR(link, "attach_fentry")) + goto cleanup; + skel->links.test_fentry = link; + + link = bpf_program__attach_tracing_multi(skel->progs.test_fexit, NULL, &opts); + if (!ASSERT_OK_PTR(link, "attach_fexit")) + goto cleanup; + skel->links.test_fexit = link; + + link = bpf_program__attach_tracing_multi(skel->progs.test_fsession, NULL, &opts); + if (!ASSERT_OK_PTR(link, "attach_fsession")) + goto cleanup; + skel->links.test_fsession = link; + + if (run_bpf_target(skel->progs.target_1, 1, "target_1")) + goto cleanup; + if (run_bpf_target(skel->progs.target_2, 2, "target_2")) + goto cleanup; + + ASSERT_EQ(skel->bss->target_1_result, 1, "target_1_result"); + ASSERT_EQ(skel->bss->target_2_result, 1, "target_2_result"); + ASSERT_EQ(skel->bss->test_result_fentry, ARRAY_SIZE(funcs), "test_result_fentry"); + ASSERT_EQ(skel->bss->test_result_fexit, ARRAY_SIZE(funcs), "test_result_fexit"); + ASSERT_EQ(skel->bss->test_result_fsession_entry, ARRAY_SIZE(funcs), + "test_result_fsession_entry"); + ASSERT_EQ(skel->bss->test_result_fsession_exit, ARRAY_SIZE(funcs), + "test_result_fsession_exit"); + +cleanup: + tracing_multi_bpf__destroy(skel); +} + static void test_module_skel_api(void) { struct tracing_multi_module *skel = NULL; @@ -1026,4 +1094,6 @@ void test_tracing_multi_test(void) RUN_TESTS(tracing_multi_verifier); if (test__start_subtest("fentry_after_multi")) test_fentry_after_multi(); + if (test__start_subtest("link_api_bpf_prog")) + test_link_api_bpf_prog(); } diff --git a/tools/testing/selftests/bpf/progs/tracing_multi_bpf.c b/tools/testing/selftests/bpf/progs/tracing_multi_bpf.c new file mode 100644 index 000000000000..3e02602558ad --- /dev/null +++ b/tools/testing/selftests/bpf/progs/tracing_multi_bpf.c @@ -0,0 +1,52 @@ +// SPDX-License-Identifier: GPL-2.0 +#include <vmlinux.h> +#include <bpf/bpf_helpers.h> +#include <bpf/bpf_tracing.h> + +char _license[] SEC("license") = "GPL"; + +__u64 test_result_fentry; +__u64 test_result_fexit; +__u64 test_result_fsession_entry; +__u64 test_result_fsession_exit; +__u64 target_1_result; +__u64 target_2_result; + +SEC("syscall") +int target_1(void *ctx) +{ + target_1_result++; + return 1; +} + +SEC("syscall") +int target_2(void *ctx) +{ + target_2_result++; + return 2; +} + +SEC("fentry.multi") +int BPF_PROG(test_fentry) +{ + test_result_fentry++; + return 0; +} + +SEC("fexit.multi") +int BPF_PROG(test_fexit) +{ + test_result_fexit++; + return 0; +} + +SEC("fsession.multi") +int BPF_PROG(test_fsession) +{ + if (bpf_session_is_return(ctx)) + test_result_fsession_exit++; + else + test_result_fsession_entry++; + + return 0; +} -- 2.55.0
