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
 8static int duration;
 9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10void 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
59cleanup:
60	test_ksyms__destroy(skel);
61}

 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
 8static int duration;
 9
10static __u64 kallsyms_find(const char *sym)
11{
12	char type, name[500];
13	__u64 addr, res = 0;
14	FILE *f;
15
16	f = fopen("/proc/kallsyms", "r");
17	if (CHECK(!f, "kallsyms_fopen", "failed to open: %d\n", errno))
18		return 0;
19
20	while (fscanf(f, "%llx %c %499s%*[^\n]\n", &addr, &type, name) > 0) {
21		if (strcmp(name, sym) == 0) {
22			res = addr;
23			goto out;
24		}
25	}
26
27	CHECK(false, "not_found", "symbol %s not found\n", sym);
28out:
29	fclose(f);
30	return res;
31}
32
33void test_ksyms(void)
34{
35	__u64 link_fops_addr = kallsyms_find("bpf_link_fops");
36	const char *btf_path = "/sys/kernel/btf/vmlinux";
37	struct test_ksyms *skel;
38	struct test_ksyms__data *data;
 
39	struct stat st;
40	__u64 btf_size;
41	int err;
42
 
 
 
 
 
 
 
 
 
 
 
 
43	if (CHECK(stat(btf_path, &st), "stat_btf", "err %d\n", errno))
44		return;
45	btf_size = st.st_size;
46
47	skel = test_ksyms__open_and_load();
48	if (CHECK(!skel, "skel_open", "failed to open and load skeleton\n"))
49		return;
50
51	err = test_ksyms__attach(skel);
52	if (CHECK(err, "skel_attach", "skeleton attach failed: %d\n", err))
53		goto cleanup;
54
55	/* trigger tracepoint */
56	usleep(1);
57
58	data = skel->data;
59	CHECK(data->out__bpf_link_fops != link_fops_addr, "bpf_link_fops",
60	      "got 0x%llx, exp 0x%llx\n",
61	      data->out__bpf_link_fops, link_fops_addr);
62	CHECK(data->out__bpf_link_fops1 != 0, "bpf_link_fops1",
63	      "got %llu, exp %llu\n", data->out__bpf_link_fops1, (__u64)0);
64	CHECK(data->out__btf_size != btf_size, "btf_size",
65	      "got %llu, exp %llu\n", data->out__btf_size, btf_size);
66	CHECK(data->out__per_cpu_start != 0, "__per_cpu_start",
67	      "got %llu, exp %llu\n", data->out__per_cpu_start, (__u64)0);
 
68
69cleanup:
70	test_ksyms__destroy(skel);
71}