1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2019 Facebook */
3 
4 #include <test_progs.h>
5 #include "test_ksyms.skel.h"
6 #include <sys/stat.h>
7 
8 static int duration;
9 
10 void test_ksyms(void)
11 {
12 	const char *btf_path = "/sys/kernel/btf/vmlinux";
13 	struct test_ksyms *skel;
14 	struct test_ksyms__data *data;
15 	__u64 link_fops_addr, per_cpu_start_addr;
16 	struct stat st;
17 	__u64 btf_size;
18 	int err;
19 
20 	err = kallsyms_find("bpf_link_fops", &link_fops_addr);
21 	if (CHECK(err == -EINVAL, "kallsyms_fopen", "failed to open: %d\n", errno))
22 		return;
23 	if (CHECK(err == -ENOENT, "ksym_find", "symbol 'bpf_link_fops' not found\n"))
24 		return;
25 
26 	err = kallsyms_find("__per_cpu_start", &per_cpu_start_addr);
27 	if (CHECK(err == -EINVAL, "kallsyms_fopen", "failed to open: %d\n", errno))
28 		return;
29 	if (CHECK(err == -ENOENT, "ksym_find", "symbol 'per_cpu_start' not found\n"))
30 		return;
31 
32 	if (CHECK(stat(btf_path, &st), "stat_btf", "err %d\n", errno))
33 		return;
34 	btf_size = st.st_size;
35 
36 	skel = test_ksyms__open_and_load();
37 	if (CHECK(!skel, "skel_open", "failed to open and load skeleton\n"))
38 		return;
39 
40 	err = test_ksyms__attach(skel);
41 	if (CHECK(err, "skel_attach", "skeleton attach failed: %d\n", err))
42 		goto cleanup;
43 
44 	/* trigger tracepoint */
45 	usleep(1);
46 
47 	data = skel->data;
48 	CHECK(data->out__bpf_link_fops != link_fops_addr, "bpf_link_fops",
49 	      "got 0x%llx, exp 0x%llx\n",
50 	      data->out__bpf_link_fops, link_fops_addr);
51 	CHECK(data->out__bpf_link_fops1 != 0, "bpf_link_fops1",
52 	      "got %llu, exp %llu\n", data->out__bpf_link_fops1, (__u64)0);
53 	CHECK(data->out__btf_size != btf_size, "btf_size",
54 	      "got %llu, exp %llu\n", data->out__btf_size, btf_size);
55 	CHECK(data->out__per_cpu_start != per_cpu_start_addr, "__per_cpu_start",
56 	      "got %llu, exp %llu\n", data->out__per_cpu_start,
57 	      per_cpu_start_addr);
58 
59 cleanup:
60 	test_ksyms__destroy(skel);
61 }
62