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}

 1// SPDX-License-Identifier: GPL-2.0
 2#include <test_progs.h>
 3#include "get_func_ip_test.skel.h"
 
 
 
 
 
 4
 5static void test_function_entry(void)
 6{
 7	struct get_func_ip_test *skel = NULL;
 8	int err, prog_fd;
 9	LIBBPF_OPTS(bpf_test_run_opts, topts);
10
11	skel = get_func_ip_test__open();
12	if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open"))
13		return;
14
15	err = get_func_ip_test__load(skel);
16	if (!ASSERT_OK(err, "get_func_ip_test__load"))
17		goto cleanup;
18
19	err = get_func_ip_test__attach(skel);
20	if (!ASSERT_OK(err, "get_func_ip_test__attach"))
21		goto cleanup;
22
 
 
23	prog_fd = bpf_program__fd(skel->progs.test1);
24	err = bpf_prog_test_run_opts(prog_fd, &topts);
25	ASSERT_OK(err, "test_run");
26	ASSERT_EQ(topts.retval, 0, "test_run");
27
28	prog_fd = bpf_program__fd(skel->progs.test5);
29	err = bpf_prog_test_run_opts(prog_fd, &topts);
30
31	ASSERT_OK(err, "test_run");
32
 
 
33	ASSERT_EQ(skel->bss->test1_result, 1, "test1_result");
34	ASSERT_EQ(skel->bss->test2_result, 1, "test2_result");
35	ASSERT_EQ(skel->bss->test3_result, 1, "test3_result");
36	ASSERT_EQ(skel->bss->test4_result, 1, "test4_result");
37	ASSERT_EQ(skel->bss->test5_result, 1, "test5_result");
 
 
38
39cleanup:
40	get_func_ip_test__destroy(skel);
41}
42
43/* test6 is x86_64 specific because of the instruction
44 * offset, disabling it for all other archs
45 */
46#ifdef __x86_64__
47static void test_function_body(void)
 
 
 
 
 
 
 
 
 
48{
49	struct get_func_ip_test *skel = NULL;
50	LIBBPF_OPTS(bpf_test_run_opts, topts);
51	LIBBPF_OPTS(bpf_kprobe_opts, kopts);
52	struct bpf_link *link6 = NULL;
53	int err, prog_fd;
54
55	skel = get_func_ip_test__open();
56	if (!ASSERT_OK_PTR(skel, "get_func_ip_test__open"))
57		return;
58
 
 
 
59	bpf_program__set_autoload(skel->progs.test6, true);
60
61	err = get_func_ip_test__load(skel);
62	if (!ASSERT_OK(err, "get_func_ip_test__load"))
63		goto cleanup;
64
65	kopts.offset = skel->kconfig->CONFIG_X86_KERNEL_IBT ? 9 : 5;
66
67	link6 = bpf_program__attach_kprobe_opts(skel->progs.test6, "bpf_fentry_test6", &kopts);
68	if (!ASSERT_OK_PTR(link6, "link6"))
69		goto cleanup;
70
71	prog_fd = bpf_program__fd(skel->progs.test1);
72	err = bpf_prog_test_run_opts(prog_fd, &topts);
73	ASSERT_OK(err, "test_run");
74	ASSERT_EQ(topts.retval, 0, "test_run");
75
76	ASSERT_EQ(skel->bss->test6_result, 1, "test6_result");
77
78cleanup:
79	bpf_link__destroy(link6);
80	get_func_ip_test__destroy(skel);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
81}
82#else
83#define test_function_body()
84#endif
85
86void test_get_func_ip_test(void)
87{
88	test_function_entry();
89	test_function_body();
90}