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