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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#pragma once
#include <fstream>
#include <iostream>
#include <regex>
#include "ast/attachpoint_parser.h"
#include "ast/passes/c_macro_expansion.h"
#include "ast/passes/clang_build.h"
#include "ast/passes/clang_parser.h"
#include "ast/passes/codegen_llvm.h"
#include "ast/passes/field_analyser.h"
#include "ast/passes/fold_literals.h"
#include "ast/passes/map_sugar.h"
#include "ast/passes/named_param.h"
#include "ast/passes/parser.h"
#include "ast/passes/pid_filter_pass.h"
#include "ast/passes/probe_expansion.h"
#include "ast/passes/recursion_check.h"
#include "ast/passes/resolve_imports.h"
#include "ast/passes/resource_analyser.h"
#include "ast/passes/semantic_analyser.h"
#include "ast/passes/type_system.h"
#include "bpftrace.h"
#include "btf_common.h"
#include "driver.h"
#include "util/env.h"
#include "gtest/gtest.h"
#include "../mocks.h"
namespace bpftrace::test::codegen {
#define NAME (::testing::UnitTest::GetInstance()->current_test_info()->name())
class codegen_btf : public test_btf {};
static std::string get_expected(const std::string &name)
{
std::string fname = TEST_CODEGEN_LOCATION + name + ".ll";
std::ifstream file;
file.open(fname);
if (file.good())
return std::string((std::istreambuf_iterator<char>(file)),
(std::istreambuf_iterator<char>()));
throw std::runtime_error("Could not find codegen result for test: " + name);
}
// This is the lower level codegen test entrypoint.
//
// The contract is that the `bpftrace` must be completely initialized and ready
// to go (eg. members replaced with mocks as necessary) before calling into
// here.
static void test(BPFtrace &bpftrace,
const std::string &input,
const std::string &name)
{
ast::ASTContext ast("stdin", input);
// N.B. No tracepoint expansion.
std::stringstream out;
auto ok = ast::PassManager()
.put(ast)
.put(bpftrace)
.add(CreateParsePass())
.add(ast::CreateResolveImportsPass())
.add(ast::CreateImportInternalScriptsPass())
.add(ast::CreateMacroExpansionPass())
.add(ast::CreateParseAttachpointsPass())
.add(ast::CreateProbeExpansionPass())
.add(ast::CreateFieldAnalyserPass())
.add(ast::CreateClangParsePass())
.add(ast::CreateCMacroExpansionPass())
.add(ast::CreateFoldLiteralsPass())
.add(ast::CreateMapSugarPass())
.add(ast::CreateNamedParamsPass())
.add(ast::CreateLLVMInitPass())
.add(ast::CreateClangBuildPass())
.add(ast::CreateTypeSystemPass())
.add(ast::CreateSemanticPass())
.add(ast::CreatePidFilterPass())
.add(ast::CreateRecursionCheckPass())
.add(ast::CreateSemanticPass())
.add(ast::CreateResourcePass())
.add(ast::CreateCompilePass())
.add(ast::CreateDumpIRPass(out))
.run();
std::stringstream errs;
ast.diagnostics().emit(errs);
ASSERT_TRUE(ok && ast.diagnostics().ok()) << errs.str();
uint64_t update_tests = 0;
util::get_uint64_env_var("BPFTRACE_UPDATE_TESTS",
[&](uint64_t x) { update_tests = x; });
if (update_tests >= 1) {
std::cerr << "Running in update mode, test is skipped" << std::endl;
std::ofstream file(TEST_CODEGEN_LOCATION + name + ".ll");
file << out.str();
return;
}
std::string expected_output = get_expected(name);
EXPECT_EQ(expected_output, out.str())
<< "the following program failed: '" << input << "'";
}
// This is the common case codegen test entrypoint.
//
// Please prefer to use this interface.
static void test(const std::string &input,
const std::string &name,
bool safe_mode = true)
{
auto bpftrace = get_mock_bpftrace();
bpftrace->safe_mode_ = safe_mode;
test(*bpftrace, input, name);
}
} // namespace bpftrace::test::codegen
|
