1// SPDX-License-Identifier: GPL-2.0
  2/* Copyright (c) 2021 Facebook */
  3#include <test_progs.h>
  4#include <sys/types.h>
  5#include <unistd.h>
  6#include "find_vma.skel.h"
  7#include "find_vma_fail1.skel.h"
  8#include "find_vma_fail2.skel.h"
  9
 10static void test_and_reset_skel(struct find_vma *skel, int expected_find_zero_ret, bool need_test)
 11{
 12	if (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");
 17	}
 18
 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;
 23}
 24
 25static int open_pe(void)
 26{
 27	struct perf_event_attr attr = {0};
 28	int pfd;
 29
 30	/* create perf event */
 31	attr.size = sizeof(attr);
 32	attr.type = PERF_TYPE_SOFTWARE;
 33	attr.config = PERF_COUNT_SW_CPU_CLOCK;
 34	attr.freq = 1;
 35	attr.sample_freq = 1000;
 36	pfd = syscall(__NR_perf_event_open, &attr, 0, -1, -1, PERF_FLAG_FD_CLOEXEC);
 37
 38	return pfd >= 0 ? pfd : -errno;
 39}
 40
 41static bool find_vma_pe_condition(struct find_vma *skel)
 42{
 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;
 47}
 48
 49static void test_find_vma_pe(struct find_vma *skel)
 50{
 51	struct bpf_link *link = NULL;
 52	volatile int j = 0;
 53	int pfd, i;
 54	const int one_bn = 1000000000;
 55
 56	pfd = open_pe();
 57	if (pfd < 0) {
 58		if (pfd == -ENOENT || pfd == -EOPNOTSUPP) {
 59			printf("%s:SKIP:no PERF_COUNT_HW_CPU_CYCLES\n", __func__);
 60			test__skip();
 61			goto cleanup;
 62		}
 63		if (!ASSERT_GE(pfd, 0, "perf_event_open"))
 64			goto cleanup;
 65	}
 66
 67	link = bpf_program__attach_perf_event(skel->progs.handle_pe, pfd);
 68	if (!ASSERT_OK_PTR(link, "attach_perf_event"))
 69		goto cleanup;
 70
 71	for (i = 0; i < one_bn && find_vma_pe_condition(skel); ++i)
 72		++j;
 73
 74	test_and_reset_skel(skel, -EBUSY /* in nmi, irq_work is busy */, i == one_bn);
 75cleanup:
 76	bpf_link__destroy(link);
 77	close(pfd);
 78}
 79
 80static void test_find_vma_kprobe(struct find_vma *skel)
 81{
 82	int err;
 83
 84	err = find_vma__attach(skel);
 85	if (!ASSERT_OK(err, "get_branch_snapshot__attach"))
 86		return;
 87
 88	getpgid(skel->bss->target_pid);
 89	test_and_reset_skel(skel, -ENOENT /* could not find vma for ptr 0 */, true);
 90}
 91
 92static void test_illegal_write_vma(void)
 93{
 94	struct find_vma_fail1 *skel;
 95
 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);
 99}
100
101static void test_illegal_write_task(void)
102{
103	struct find_vma_fail2 *skel;
104
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);
108}
109
110void serial_test_find_vma(void)
111{
112	struct find_vma *skel;
113
114	skel = find_vma__open_and_load();
115	if (!ASSERT_OK_PTR(skel, "find_vma__open_and_load"))
116		return;
117
118	skel->bss->target_pid = getpid();
119	skel->bss->addr = (__u64)(uintptr_t)test_find_vma_pe;
120
121	test_find_vma_pe(skel);
122	test_find_vma_kprobe(skel);
123
124	find_vma__destroy(skel);
125	test_illegal_write_vma();
126	test_illegal_write_task();
127}