1
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"
12static int test__bpf_parsing(void *obj_buf, size_t obj_buf_sz)
13{
14 struct bpf_object *obj;
15
16 obj = bpf_object__open_mem(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
23static 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
47int
48test_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
70
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
80
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;
119out:
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
127static int test__llvm(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#endif
152
153static int test__llvm__bpf_base_prog(struct test_suite *test __maybe_unused,
154 int subtest __maybe_unused)
155{
156#ifdef HAVE_LIBBPF_SUPPORT
157 return test__llvm(LLVM_TESTCASE_BASE);
158#else
159 pr_debug("Skip LLVM test because BPF support is not compiled\n");
160 return TEST_SKIP;
161#endif
162}
163
164static int test__llvm__bpf_test_kbuild_prog(struct test_suite *test __maybe_unused,
165 int subtest __maybe_unused)
166{
167#ifdef HAVE_LIBBPF_SUPPORT
168 return test__llvm(LLVM_TESTCASE_KBUILD);
169#else
170 pr_debug("Skip LLVM test because BPF support is not compiled\n");
171 return TEST_SKIP;
172#endif
173}
174
175static int test__llvm__bpf_test_prologue_prog(struct test_suite *test __maybe_unused,
176 int subtest __maybe_unused)
177{
178#ifdef HAVE_LIBBPF_SUPPORT
179 return test__llvm(LLVM_TESTCASE_BPF_PROLOGUE);
180#else
181 pr_debug("Skip LLVM test because BPF support is not compiled\n");
182 return TEST_SKIP;
183#endif
184}
185
186static int test__llvm__bpf_test_relocation(struct test_suite *test __maybe_unused,
187 int subtest __maybe_unused)
188{
189#ifdef HAVE_LIBBPF_SUPPORT
190 return test__llvm(LLVM_TESTCASE_BPF_RELOCATION);
191#else
192 pr_debug("Skip LLVM test because BPF support is not compiled\n");
193 return TEST_SKIP;
194#endif
195}
196
197
198static struct test_case llvm_tests[] = {
199#ifdef HAVE_LIBBPF_SUPPORT
200 TEST_CASE("Basic BPF llvm compile", llvm__bpf_base_prog),
201 TEST_CASE("kbuild searching", llvm__bpf_test_kbuild_prog),
202 TEST_CASE("Compile source for BPF prologue generation",
203 llvm__bpf_test_prologue_prog),
204 TEST_CASE("Compile source for BPF relocation", llvm__bpf_test_relocation),
205#else
206 TEST_CASE_REASON("Basic BPF llvm compile", llvm__bpf_base_prog, "not compiled in"),
207 TEST_CASE_REASON("kbuild searching", llvm__bpf_test_kbuild_prog, "not compiled in"),
208 TEST_CASE_REASON("Compile source for BPF prologue generation",
209 llvm__bpf_test_prologue_prog, "not compiled in"),
210 TEST_CASE_REASON("Compile source for BPF relocation",
211 llvm__bpf_test_relocation, "not compiled in"),
212#endif
213 { .name = NULL, }
214};
215
216struct test_suite suite__llvm = {
217 .desc = "LLVM search and compile",
218 .test_cases = llvm_tests,
219};
220
