bpf/prog_tests/find_vma.c

6 #include "find_vma.skel.h"
7 #include "find_vma_fail1.skel.h"
8 #include "find_vma_fail2.skel.h"
10 static void test_and_reset_skel(struct find_vma *skel, int expected_find_zero_ret, bool need_test)
13 		ASSERT_EQ(skel->bss->found_vm_exec, 1, "found_vm_exec");
14 		ASSERT_EQ(skel->data->find_addr_ret, 0, "find_addr_ret");
15 		ASSERT_EQ(skel->data->find_zero_ret, expected_find_zero_ret, "find_zero_ret");
16 		ASSERT_OK_PTR(strstr(skel->bss->d_iname, "test_progs"), "find_test_progs");
19 	skel->bss->found_vm_exec = 0;
20 	skel->data->find_addr_ret = -1;
21 	skel->data->find_zero_ret = -1;
22 	skel->bss->d_iname[0] = 0;
41 static bool find_vma_pe_condition(struct find_vma *skel)
43 	return skel->bss->found_vm_exec == 0 ||
44 		skel->data->find_addr_ret != 0 ||
45 		skel->data->find_zero_ret == -1 ||
46 		strcmp(skel->bss->d_iname, "test_progs") != 0;
49 static void test_find_vma_pe(struct find_vma *skel)
67 	link = bpf_program__attach_perf_event(skel->progs.handle_pe, pfd);
71 	for (i = 0; i < one_bn && find_vma_pe_condition(skel); ++i)
74 	test_and_reset_skel(skel, -EBUSY /* in nmi, irq_work is busy */, i == one_bn);
80 static void test_find_vma_kprobe(struct find_vma *skel)
84 	err = find_vma__attach(skel);
88 	getpgid(skel->bss->target_pid);
89 	test_and_reset_skel(skel, -ENOENT /* could not find vma for ptr 0 */, true);
94 	struct find_vma_fail1 *skel;
96 	skel = find_vma_fail1__open_and_load();
97 	if (!ASSERT_ERR_PTR(skel, "find_vma_fail1__open_and_load"))
98 		find_vma_fail1__destroy(skel);
103 	struct find_vma_fail2 *skel;
105 	skel = find_vma_fail2__open_and_load();
106 	if (!ASSERT_ERR_PTR(skel, "find_vma_fail2__open_and_load"))
107 		find_vma_fail2__destroy(skel);
112 	struct find_vma *skel;
114 	skel = find_vma__open_and_load();
115 	if (!ASSERT_OK_PTR(skel, "find_vma__open_and_load"))
118 	skel->bss->target_pid = getpid();
119 	skel->bss->addr = (__u64)(uintptr_t)test_find_vma_pe;
121 	test_find_vma_pe(skel);
122 	test_find_vma_kprobe(skel);
124 	find_vma__destroy(skel);