1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#include "bpfbytecode.h"
#include "ast/passes/clang_build.h"
#include "ast/passes/codegen_llvm.h"
#include "ast/passes/parser.h"
#include "ast/passes/semantic_analyser.h"
#include "ast/passes/type_system.h"
#include "mocks.h"
#include "gtest/gtest.h"
namespace bpftrace::test::bpfbytecode {
BpfBytecode codegen(const std::string &input)
{
auto bpftrace = get_mock_bpftrace();
ast::ASTContext ast("stdin", input);
auto ok = ast::PassManager()
.put(ast)
.put<BPFtrace>(*bpftrace)
.add(ast::AllParsePasses())
.add(ast::CreateLLVMInitPass())
.add(ast::CreateClangBuildPass())
.add(ast::CreateTypeSystemPass())
.add(ast::CreateSemanticPass())
.add(ast::AllCompilePasses())
.run();
EXPECT_TRUE(ok && ast.diagnostics().ok());
auto &output = ok->get<BpfBytecode>();
return std::move(output);
}
TEST(bpfbytecode, create_programs)
{
auto bytecode = codegen("kprobe:foo { 1 }");
Probe foo;
foo.type = ProbeType::kprobe;
foo.name = "kprobe:foo";
foo.index = 1;
auto &program = bytecode.getProgramForProbe(foo);
EXPECT_EQ(std::string_view{ bpf_program__name(program.bpf_prog()) },
"kprobe_foo_1");
EXPECT_EQ(std::string_view{ bpf_program__section_name(program.bpf_prog()) },
"s_kprobe_foo_1");
}
} // namespace bpftrace::test::bpfbytecode
|
