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