140 lines · c
1// SPDX-License-Identifier: GPL-2.02#include <test_progs.h>3#include "get_func_ip_test.skel.h"4#include "get_func_ip_uprobe_test.skel.h"5 6static noinline void uprobe_trigger(void)7{8}9 10static void test_function_entry(void)11{12 struct get_func_ip_test *skel = NULL;13 int err, prog_fd;14 LIBBPF_OPTS(bpf_test_run_opts, topts);15 16 skel = get_func_ip_test__open();17 if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open"))18 return;19 20 err = get_func_ip_test__load(skel);21 if (!ASSERT_OK(err, "get_func_ip_test__load"))22 goto cleanup;23 24 err = get_func_ip_test__attach(skel);25 if (!ASSERT_OK(err, "get_func_ip_test__attach"))26 goto cleanup;27 28 skel->bss->uprobe_trigger = (unsigned long) uprobe_trigger;29 30 prog_fd = bpf_program__fd(skel->progs.test1);31 err = bpf_prog_test_run_opts(prog_fd, &topts);32 ASSERT_OK(err, "test_run");33 ASSERT_EQ(topts.retval, 0, "test_run");34 35 prog_fd = bpf_program__fd(skel->progs.test5);36 err = bpf_prog_test_run_opts(prog_fd, &topts);37 38 ASSERT_OK(err, "test_run");39 40 uprobe_trigger();41 42 ASSERT_EQ(skel->bss->test1_result, 1, "test1_result");43 ASSERT_EQ(skel->bss->test2_result, 1, "test2_result");44 ASSERT_EQ(skel->bss->test3_result, 1, "test3_result");45 ASSERT_EQ(skel->bss->test4_result, 1, "test4_result");46 ASSERT_EQ(skel->bss->test5_result, 1, "test5_result");47 ASSERT_EQ(skel->bss->test7_result, 1, "test7_result");48 ASSERT_EQ(skel->bss->test8_result, 1, "test8_result");49 50cleanup:51 get_func_ip_test__destroy(skel);52}53 54#ifdef __x86_64__55extern void uprobe_trigger_body(void);56asm(57".globl uprobe_trigger_body\n"58".type uprobe_trigger_body, @function\n"59"uprobe_trigger_body:\n"60" nop\n"61" ret\n"62);63 64static void test_function_body_kprobe(void)65{66 struct get_func_ip_test *skel = NULL;67 LIBBPF_OPTS(bpf_test_run_opts, topts);68 LIBBPF_OPTS(bpf_kprobe_opts, kopts);69 struct bpf_link *link6 = NULL;70 int err, prog_fd;71 72 skel = get_func_ip_test__open();73 if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open"))74 return;75 76 /* test6 is x86_64 specific and is disabled by default,77 * enable it for body test.78 */79 bpf_program__set_autoload(skel->progs.test6, true);80 81 err = get_func_ip_test__load(skel);82 if (!ASSERT_OK(err, "get_func_ip_test__load"))83 goto cleanup;84 85 kopts.offset = skel->kconfig->CONFIG_X86_KERNEL_IBT ? 9 : 5;86 87 link6 = bpf_program__attach_kprobe_opts(skel->progs.test6, "bpf_fentry_test6", &kopts);88 if (!ASSERT_OK_PTR(link6, "link6"))89 goto cleanup;90 91 prog_fd = bpf_program__fd(skel->progs.test1);92 err = bpf_prog_test_run_opts(prog_fd, &topts);93 ASSERT_OK(err, "test_run");94 ASSERT_EQ(topts.retval, 0, "test_run");95 96 ASSERT_EQ(skel->bss->test6_result, 1, "test6_result");97 98cleanup:99 bpf_link__destroy(link6);100 get_func_ip_test__destroy(skel);101}102 103static void test_function_body_uprobe(void)104{105 struct get_func_ip_uprobe_test *skel = NULL;106 int err;107 108 skel = get_func_ip_uprobe_test__open_and_load();109 if (!ASSERT_OK_PTR(skel, "get_func_ip_uprobe_test__open_and_load"))110 return;111 112 err = get_func_ip_uprobe_test__attach(skel);113 if (!ASSERT_OK(err, "get_func_ip_test__attach"))114 goto cleanup;115 116 skel->bss->uprobe_trigger_body = (unsigned long) uprobe_trigger_body;117 118 uprobe_trigger_body();119 120 ASSERT_EQ(skel->bss->test1_result, 1, "test1_result");121 122cleanup:123 get_func_ip_uprobe_test__destroy(skel);124}125 126static void test_function_body(void)127{128 test_function_body_kprobe();129 test_function_body_uprobe();130}131#else132#define test_function_body()133#endif134 135void test_get_func_ip_test(void)136{137 test_function_entry();138 test_function_body();139}140