Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
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"
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_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
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 * Skip this test if user's .perfconfig doesn't set [llvm] section
70 * and clang is not found in $PATH, and this is not perf test -v
71 */
72 if (!force && (verbose <= 0 &&
73 !llvm_param.user_set_param &&
74 llvm__search_clang())) {
75 pr_debug("No clang and no verbosive, skip this test\n");
76 return TEST_SKIP;
77 }
78
79 /*
80 * llvm is verbosity when error. Suppress all error output if
81 * not 'perf test -v'.
82 */
83 old_verbose = verbose;
84 if (verbose == 0)
85 verbose = -1;
86
87 *p_obj_buf = NULL;
88 *p_obj_buf_sz = 0;
89
90 if (!llvm_param.clang_bpf_cmd_template)
91 goto out;
92
93 if (!llvm_param.clang_opt)
94 llvm_param.clang_opt = strdup("");
95
96 err = asprintf(&tmpl_new, "echo '%s' | %s%s", source,
97 llvm_param.clang_bpf_cmd_template,
98 old_verbose ? "" : " 2>/dev/null");
99 if (err < 0)
100 goto out;
101 err = asprintf(&clang_opt_new, "-xc %s", llvm_param.clang_opt);
102 if (err < 0)
103 goto out;
104
105 tmpl_old = llvm_param.clang_bpf_cmd_template;
106 llvm_param.clang_bpf_cmd_template = tmpl_new;
107 clang_opt_old = llvm_param.clang_opt;
108 llvm_param.clang_opt = clang_opt_new;
109
110 err = llvm__compile_bpf("-", p_obj_buf, p_obj_buf_sz);
111
112 llvm_param.clang_bpf_cmd_template = tmpl_old;
113 llvm_param.clang_opt = clang_opt_old;
114
115 verbose = old_verbose;
116 if (err)
117 goto out;
118
119 ret = TEST_OK;
120out:
121 free(tmpl_new);
122 free(clang_opt_new);
123 if (ret != TEST_OK)
124 pr_debug("Failed to compile test case: '%s'\n", desc);
125 return ret;
126}
127
128int test__llvm(struct test *test __maybe_unused, int subtest)
129{
130 int ret;
131 void *obj_buf = NULL;
132 size_t obj_buf_sz = 0;
133 bool should_load_fail = false;
134
135 if ((subtest < 0) || (subtest >= __LLVM_TESTCASE_MAX))
136 return TEST_FAIL;
137
138 ret = test_llvm__fetch_bpf_obj(&obj_buf, &obj_buf_sz,
139 subtest, false, &should_load_fail);
140
141 if (ret == TEST_OK && !should_load_fail) {
142 ret = test__bpf_parsing(obj_buf, obj_buf_sz);
143 if (ret != TEST_OK) {
144 pr_debug("Failed to parse test case '%s'\n",
145 bpf_source_table[subtest].desc);
146 }
147 }
148 free(obj_buf);
149
150 return ret;
151}
152
153int test__llvm_subtest_get_nr(void)
154{
155 return __LLVM_TESTCASE_MAX;
156}
157
158const char *test__llvm_subtest_get_desc(int subtest)
159{
160 if ((subtest < 0) || (subtest >= __LLVM_TESTCASE_MAX))
161 return NULL;
162
163 return bpf_source_table[subtest].desc;
164}
165#else //HAVE_LIBBPF_SUPPORT
166int test__llvm(struct test *test __maybe_unused, int subtest __maybe_unused)
167{
168 return TEST_SKIP;
169}
170
171int test__llvm_subtest_get_nr(void)
172{
173 return 0;
174}
175
176const char *test__llvm_subtest_get_desc(int subtest __maybe_unused)
177{
178 return NULL;
179}
180#endif // HAVE_LIBBPF_SUPPORT