Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f699cf7a authored by Yonghong Song's avatar Yonghong Song Committed by Alexei Starovoitov
Browse files

tools/bpf: add two BPF_TASK_FD_QUERY tests in test_progs 



The new tests are added to query perf_event information
for raw_tracepoint and tracepoint attachment. For tracepoint,
both syscalls and non-syscalls tracepoints are queries as
they are treated slightly differently inside the kernel.

Signed-off-by: default avatarYonghong Song <yhs@fb.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent ecb96f7f

tools/testing/selftests/bpf/test_progs.c

+158 −0
Original line number Diff line number Diff line
@@ -1542,6 +1542,162 @@ static void test_get_stack_raw_tp(void) 
	bpf_object__close(obj);

}



static void test_task_fd_query_rawtp(void)

{

	const char *file = "./test_get_stack_rawtp.o";

	__u64 probe_offset, probe_addr;

	__u32 len, prog_id, fd_type;

	struct bpf_object *obj;

	int efd, err, prog_fd;

	__u32 duration = 0;

	char buf[256];



	err = bpf_prog_load(file, BPF_PROG_TYPE_RAW_TRACEPOINT, &obj, &prog_fd);

	if (CHECK(err, "prog_load raw tp", "err %d errno %d\n", err, errno))

		return;



	efd = bpf_raw_tracepoint_open("sys_enter", prog_fd);

	if (CHECK(efd < 0, "raw_tp_open", "err %d errno %d\n", efd, errno))

		goto close_prog;



	/* query (getpid(), efd) */

	len = sizeof(buf);

	err = bpf_task_fd_query(getpid(), efd, 0, buf, &len, &prog_id,

				&fd_type, &probe_offset, &probe_addr);

	if (CHECK(err < 0, "bpf_task_fd_query", "err %d errno %d\n", err,

		  errno))

		goto close_prog;



	err = fd_type == BPF_FD_TYPE_RAW_TRACEPOINT &&

	      strcmp(buf, "sys_enter") == 0;

	if (CHECK(!err, "check_results", "fd_type %d tp_name %s\n",

		  fd_type, buf))

		goto close_prog;



	/* test zero len */

	len = 0;

	err = bpf_task_fd_query(getpid(), efd, 0, buf, &len, &prog_id,

				&fd_type, &probe_offset, &probe_addr);

	if (CHECK(err < 0, "bpf_task_fd_query (len = 0)", "err %d errno %d\n",

		  err, errno))

		goto close_prog;

	err = fd_type == BPF_FD_TYPE_RAW_TRACEPOINT &&

	      len == strlen("sys_enter");

	if (CHECK(!err, "check_results", "fd_type %d len %u\n", fd_type, len))

		goto close_prog;



	/* test empty buffer */

	len = sizeof(buf);

	err = bpf_task_fd_query(getpid(), efd, 0, 0, &len, &prog_id,

				&fd_type, &probe_offset, &probe_addr);

	if (CHECK(err < 0, "bpf_task_fd_query (buf = 0)", "err %d errno %d\n",

		  err, errno))

		goto close_prog;

	err = fd_type == BPF_FD_TYPE_RAW_TRACEPOINT &&

	      len == strlen("sys_enter");

	if (CHECK(!err, "check_results", "fd_type %d len %u\n", fd_type, len))

		goto close_prog;



	/* test smaller buffer */

	len = 3;

	err = bpf_task_fd_query(getpid(), efd, 0, buf, &len, &prog_id,

				&fd_type, &probe_offset, &probe_addr);

	if (CHECK(err >= 0 || errno != ENOSPC, "bpf_task_fd_query (len = 3)",

		  "err %d errno %d\n", err, errno))

		goto close_prog;

	err = fd_type == BPF_FD_TYPE_RAW_TRACEPOINT &&

	      len == strlen("sys_enter") &&

	      strcmp(buf, "sy") == 0;

	if (CHECK(!err, "check_results", "fd_type %d len %u\n", fd_type, len))

		goto close_prog;



	goto close_prog_noerr;

close_prog:

	error_cnt++;

close_prog_noerr:

	bpf_object__close(obj);

}



static void test_task_fd_query_tp_core(const char *probe_name,

				       const char *tp_name)

{

	const char *file = "./test_tracepoint.o";

	int err, bytes, efd, prog_fd, pmu_fd;

	struct perf_event_attr attr = {};

	__u64 probe_offset, probe_addr;

	__u32 len, prog_id, fd_type;

	struct bpf_object *obj;

	__u32 duration = 0;

	char buf[256];



	err = bpf_prog_load(file, BPF_PROG_TYPE_TRACEPOINT, &obj, &prog_fd);

	if (CHECK(err, "bpf_prog_load", "err %d errno %d\n", err, errno))

		goto close_prog;



	snprintf(buf, sizeof(buf),

		 "/sys/kernel/debug/tracing/events/%s/id", probe_name);

	efd = open(buf, O_RDONLY, 0);

	if (CHECK(efd < 0, "open", "err %d errno %d\n", efd, errno))

		goto close_prog;

	bytes = read(efd, buf, sizeof(buf));

	close(efd);

	if (CHECK(bytes <= 0 || bytes >= sizeof(buf), "read",

		  "bytes %d errno %d\n", bytes, errno))

		goto close_prog;



	attr.config = strtol(buf, NULL, 0);

	attr.type = PERF_TYPE_TRACEPOINT;

	attr.sample_type = PERF_SAMPLE_RAW;

	attr.sample_period = 1;

	attr.wakeup_events = 1;

	pmu_fd = syscall(__NR_perf_event_open, &attr, -1 /* pid */,

			 0 /* cpu 0 */, -1 /* group id */,

			 0 /* flags */);

	if (CHECK(err, "perf_event_open", "err %d errno %d\n", err, errno))

		goto close_pmu;



	err = ioctl(pmu_fd, PERF_EVENT_IOC_ENABLE, 0);

	if (CHECK(err, "perf_event_ioc_enable", "err %d errno %d\n", err,

		  errno))

		goto close_pmu;



	err = ioctl(pmu_fd, PERF_EVENT_IOC_SET_BPF, prog_fd);

	if (CHECK(err, "perf_event_ioc_set_bpf", "err %d errno %d\n", err,

		  errno))

		goto close_pmu;



	/* query (getpid(), pmu_fd) */

	len = sizeof(buf);

	err = bpf_task_fd_query(getpid(), pmu_fd, 0, buf, &len, &prog_id,

				&fd_type, &probe_offset, &probe_addr);

	if (CHECK(err < 0, "bpf_task_fd_query", "err %d errno %d\n", err,

		  errno))

		goto close_pmu;



	err = (fd_type == BPF_FD_TYPE_TRACEPOINT) && !strcmp(buf, tp_name);

	if (CHECK(!err, "check_results", "fd_type %d tp_name %s\n",

		  fd_type, buf))

		goto close_pmu;



	close(pmu_fd);

	goto close_prog_noerr;



close_pmu:

	close(pmu_fd);

close_prog:

	error_cnt++;

close_prog_noerr:

	bpf_object__close(obj);

}



static void test_task_fd_query_tp(void)

{

	test_task_fd_query_tp_core("sched/sched_switch",

				   "sched_switch");

	test_task_fd_query_tp_core("syscalls/sys_enter_read",

				   "sys_enter_read");

}



int main(void)

{

	jit_enabled = is_jit_enabled();
@@ -1561,6 +1717,8 @@ int main(void) 
	test_stacktrace_build_id_nmi();

	test_stacktrace_map_raw_tp();

	test_get_stack_raw_tp();

	test_task_fd_query_rawtp();

	test_task_fd_query_tp();



	printf("Summary: %d PASSED, %d FAILED\n", pass_cnt, error_cnt);

	return error_cnt ? EXIT_FAILURE : EXIT_SUCCESS;