| // SPDX-License-Identifier: GPL-2.0 |
| #include <test_progs.h> |
| #include "get_func_ip_test.skel.h" |
| |
| static void test_function_entry(void) |
| { |
| struct get_func_ip_test *skel = NULL; |
| int err, prog_fd; |
| LIBBPF_OPTS(bpf_test_run_opts, topts); |
| |
| skel = get_func_ip_test__open(); |
| if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open")) |
| return; |
| |
| err = get_func_ip_test__load(skel); |
| if (!ASSERT_OK(err, "get_func_ip_test__load")) |
| goto cleanup; |
| |
| err = get_func_ip_test__attach(skel); |
| if (!ASSERT_OK(err, "get_func_ip_test__attach")) |
| goto cleanup; |
| |
| prog_fd = bpf_program__fd(skel->progs.test1); |
| err = bpf_prog_test_run_opts(prog_fd, &topts); |
| ASSERT_OK(err, "test_run"); |
| ASSERT_EQ(topts.retval, 0, "test_run"); |
| |
| prog_fd = bpf_program__fd(skel->progs.test5); |
| err = bpf_prog_test_run_opts(prog_fd, &topts); |
| |
| ASSERT_OK(err, "test_run"); |
| |
| ASSERT_EQ(skel->bss->test1_result, 1, "test1_result"); |
| ASSERT_EQ(skel->bss->test2_result, 1, "test2_result"); |
| ASSERT_EQ(skel->bss->test3_result, 1, "test3_result"); |
| ASSERT_EQ(skel->bss->test4_result, 1, "test4_result"); |
| ASSERT_EQ(skel->bss->test5_result, 1, "test5_result"); |
| |
| cleanup: |
| get_func_ip_test__destroy(skel); |
| } |
| |
| /* test6 is x86_64 specific because of the instruction |
| * offset, disabling it for all other archs |
| */ |
| #ifdef __x86_64__ |
| static void test_function_body(void) |
| { |
| struct get_func_ip_test *skel = NULL; |
| LIBBPF_OPTS(bpf_test_run_opts, topts); |
| LIBBPF_OPTS(bpf_kprobe_opts, kopts); |
| struct bpf_link *link6 = NULL; |
| int err, prog_fd; |
| |
| skel = get_func_ip_test__open(); |
| if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open")) |
| return; |
| |
| bpf_program__set_autoload(skel->progs.test6, true); |
| |
| err = get_func_ip_test__load(skel); |
| if (!ASSERT_OK(err, "get_func_ip_test__load")) |
| goto cleanup; |
| |
| kopts.offset = skel->kconfig->CONFIG_X86_KERNEL_IBT ? 9 : 5; |
| |
| link6 = bpf_program__attach_kprobe_opts(skel->progs.test6, "bpf_fentry_test6", &kopts); |
| if (!ASSERT_OK_PTR(link6, "link6")) |
| goto cleanup; |
| |
| prog_fd = bpf_program__fd(skel->progs.test1); |
| err = bpf_prog_test_run_opts(prog_fd, &topts); |
| ASSERT_OK(err, "test_run"); |
| ASSERT_EQ(topts.retval, 0, "test_run"); |
| |
| ASSERT_EQ(skel->bss->test6_result, 1, "test6_result"); |
| |
| cleanup: |
| bpf_link__destroy(link6); |
| get_func_ip_test__destroy(skel); |
| } |
| #else |
| #define test_function_body() |
| #endif |
| |
| void test_get_func_ip_test(void) |
| { |
| test_function_entry(); |
| test_function_body(); |
| } |