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