1// SPDX-License-Identifier: GPL-2.0
2#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#else
132#define test_function_body()
133#endif
134
135void test_get_func_ip_test(void)
136{
137	test_function_entry();
138	test_function_body();
139}
140