tools/perf/tests/llvm.c

   1 #include <stdio.h>
   2 #include <bpf/libbpf.h>
   3 #include <util/llvm-utils.h>
   4 #include <util/cache.h>
   5 #include "llvm.h"
   6 #include "tests.h"
   7 #include "debug.h"
   8
   9 #ifdef HAVE_LIBBPF_SUPPORT
  10 static int test__bpf_parsing(void *obj_buf, size_t obj_buf_sz)
  11 {
  12         struct bpf_object *obj;
  13
  14         obj = bpf_object__open_buffer(obj_buf, obj_buf_sz, NULL);
  15         if (IS_ERR(obj))
  16                 return TEST_FAIL;
  17         bpf_object__close(obj);
  18         return TEST_OK;
  19 }
  20 #else
  21 static int test__bpf_parsing(void *obj_buf __maybe_unused,
  22                              size_t obj_buf_sz __maybe_unused)
  23 {
  24         pr_debug("Skip bpf parsing\n");
  25         return TEST_OK;
  26 }
  27 #endif
  28
  29 static struct {
  30         const char *source;
  31         const char *desc;
  32         bool should_load_fail;
  33 } bpf_source_table[__LLVM_TESTCASE_MAX] = {
  34         [LLVM_TESTCASE_BASE] = {
  35                 .source = test_llvm__bpf_base_prog,
  36                 .desc = "Basic BPF llvm compiling test",
  37         },
  38         [LLVM_TESTCASE_KBUILD] = {
  39                 .source = test_llvm__bpf_test_kbuild_prog,
  40                 .desc = "Test kbuild searching",
  41         },
  42         [LLVM_TESTCASE_BPF_PROLOGUE] = {
  43                 .source = test_llvm__bpf_test_prologue_prog,
  44                 .desc = "Compile source for BPF prologue generation test",
  45         },
  46         [LLVM_TESTCASE_BPF_RELOCATION] = {
  47                 .source = test_llvm__bpf_test_relocation,
  48                 .desc = "Compile source for BPF relocation test",
  49                 .should_load_fail = true,
  50         },
  51 };
  52
  53 int
  54 test_llvm__fetch_bpf_obj(void **p_obj_buf,
  55                          size_t *p_obj_buf_sz,
  56                          enum test_llvm__testcase idx,
  57                          bool force,
  58                          bool *should_load_fail)
  59 {
  60         const char *source;
  61         const char *desc;
  62         const char *tmpl_old, *clang_opt_old;
  63         char *tmpl_new = NULL, *clang_opt_new = NULL;
  64         int err, old_verbose, ret = TEST_FAIL;
  65
  66         if (idx >= __LLVM_TESTCASE_MAX)
  67                 return TEST_FAIL;
  68
  69         source = bpf_source_table[idx].source;
  70         desc = bpf_source_table[idx].desc;
  71         if (should_load_fail)
  72                 *should_load_fail = bpf_source_table[idx].should_load_fail;
  73
  74         /*
  75          * Skip this test if user's .perfconfig doesn't set [llvm] section
  76          * and clang is not found in $PATH, and this is not perf test -v
  77          */
  78         if (!force && (verbose == 0 &&
  79                        !llvm_param.user_set_param &&
  80                        llvm__search_clang())) {
  81                 pr_debug("No clang and no verbosive, skip this test\n");
  82                 return TEST_SKIP;
  83         }
  84
  85         /*
  86          * llvm is verbosity when error. Suppress all error output if
  87          * not 'perf test -v'.
  88          */
  89         old_verbose = verbose;
  90         if (verbose == 0)
  91                 verbose = -1;
  92
  93         *p_obj_buf = NULL;
  94         *p_obj_buf_sz = 0;
  95
  96         if (!llvm_param.clang_bpf_cmd_template)
  97                 goto out;
  98
  99         if (!llvm_param.clang_opt)
 100                 llvm_param.clang_opt = strdup("");
 101
 102         err = asprintf(&tmpl_new, "echo '%s' | %s%s", source,
 103                        llvm_param.clang_bpf_cmd_template,
 104                        old_verbose ? "" : " 2>/dev/null");
 105         if (err < 0)
 106                 goto out;
 107         err = asprintf(&clang_opt_new, "-xc %s", llvm_param.clang_opt);
 108         if (err < 0)
 109                 goto out;
 110
 111         tmpl_old = llvm_param.clang_bpf_cmd_template;
 112         llvm_param.clang_bpf_cmd_template = tmpl_new;
 113         clang_opt_old = llvm_param.clang_opt;
 114         llvm_param.clang_opt = clang_opt_new;
 115
 116         err = llvm__compile_bpf("-", p_obj_buf, p_obj_buf_sz);
 117
 118         llvm_param.clang_bpf_cmd_template = tmpl_old;
 119         llvm_param.clang_opt = clang_opt_old;
 120
 121         verbose = old_verbose;
 122         if (err)
 123                 goto out;
 124
 125         ret = TEST_OK;
 126 out:
 127         free(tmpl_new);
 128         free(clang_opt_new);
 129         if (ret != TEST_OK)
 130                 pr_debug("Failed to compile test case: '%s'\n", desc);
 131         return ret;
 132 }
 133
 134 int test__llvm(int subtest)
 135 {
 136         int ret;
 137         void *obj_buf = NULL;
 138         size_t obj_buf_sz = 0;
 139         bool should_load_fail = false;
 140
 141         if ((subtest < 0) || (subtest >= __LLVM_TESTCASE_MAX))
 142                 return TEST_FAIL;
 143
 144         ret = test_llvm__fetch_bpf_obj(&obj_buf, &obj_buf_sz,
 145                                        subtest, false, &should_load_fail);
 146
 147         if (ret == TEST_OK && !should_load_fail) {
 148                 ret = test__bpf_parsing(obj_buf, obj_buf_sz);
 149                 if (ret != TEST_OK) {
 150                         pr_debug("Failed to parse test case '%s'\n",
 151                                  bpf_source_table[subtest].desc);
 152                 }
 153         }
 154         free(obj_buf);
 155
 156         return ret;
 157 }
 158
 159 int test__llvm_subtest_get_nr(void)
 160 {
 161         return __LLVM_TESTCASE_MAX;
 162 }
 163
 164 const char *test__llvm_subtest_get_desc(int subtest)
 165 {
 166         if ((subtest < 0) || (subtest >= __LLVM_TESTCASE_MAX))
 167                 return NULL;
 168
 169         return bpf_source_table[subtest].desc;
 170 }