tools/testing/selftests/bpf/prog_tests/ksyms.c

   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 }