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
|
#include "ast/attachpoint_parser.h"
#include "ast/passes/clang_build.h"
#include "ast/passes/codegen_llvm.h"
#include "ast/passes/field_analyser.h"
#include "ast/passes/map_sugar.h"
#include "ast/passes/named_param.h"
#include "ast/passes/probe_expansion.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.h"
#include "btf_common.h"
#include "driver.h"
#include "mocks.h"
#include "gtest/gtest.h"
namespace bpftrace::test::probe {
using bpftrace::ast::AttachPoint;
using bpftrace::ast::AttachPointList;
using bpftrace::ast::Probe;
void gen_bytecode(const std::string &input, std::stringstream &out)
{
auto bpftrace = get_mock_bpftrace();
ast::ASTContext ast("stdin", input);
ast::CDefinitions no_c_defs; // Output from clang parser.
// N.B. No macro expansion.
auto ok = ast::PassManager()
.put(ast)
.put<BPFtrace>(*bpftrace)
.put(no_c_defs)
.add(CreateParsePass())
.add(ast::CreateResolveImportsPass())
.add(ast::CreateParseAttachpointsPass())
.add(ast::CreateProbeExpansionPass())
.add(CreateParseBTFPass())
.add(ast::CreateMapSugarPass())
.add(ast::CreateFieldAnalyserPass())
.add(ast::CreateNamedParamsPass())
.add(ast::CreateLLVMInitPass())
.add(ast::CreateClangBuildPass())
.add(ast::CreateTypeSystemPass())
.add(ast::CreateSemanticPass())
.add(ast::CreateResourcePass())
.add(ast::AllCompilePasses())
.run();
ASSERT_TRUE(ok && ast.diagnostics().ok());
auto &obj = ok->get<ast::BpfObject>();
out.write(obj.data.data(), obj.data.size());
}
void compare_bytecode(const std::string &input1, const std::string &input2)
{
std::stringstream expected_output1;
std::stringstream expected_output2;
gen_bytecode(input1, expected_output1);
gen_bytecode(input2, expected_output2);
EXPECT_EQ(expected_output1.str(), expected_output2.str());
}
TEST(probe, short_name)
{
compare_bytecode("tracepoint:sched:sched_one { args }",
"t:sched:sched_one { args }");
compare_bytecode("kprobe:f { pid }", "k:f { pid }");
compare_bytecode("kretprobe:f { pid }", "kr:f { pid }");
compare_bytecode("uprobe:sh:f { 1 }", "u:sh:f { 1 }");
compare_bytecode("profile:hz:997 { 1 }", "p:hz:997 { 1 }");
compare_bytecode("hardware:cache-references:1000000 { 1 }",
"h:cache-references:1000000 { 1 }");
compare_bytecode("software:faults:1000 { 1 }", "s:faults:1000 { 1 }");
compare_bytecode("interval:s:1 { 1 }", "i:s:1 { 1 }");
}
TEST(probe, case_insensitive)
{
compare_bytecode("tracepoint:sched:sched_one { args }",
"traCepoInt:sched:sched_one { args }");
compare_bytecode("kprobe:f { pid }", "KPROBE:f { pid }");
compare_bytecode("BEGIN { pid }", "begin { pid }");
}
class probe_btf : public test_btf {};
TEST_F(probe_btf, short_name)
{
compare_bytecode("fentry:func_1 { 1 }", "f:func_1 { 1 }");
compare_bytecode("fexit:func_1 { 1 }", "fr:func_1 { 1 }");
compare_bytecode("iter:task { 1 }", "it:task { 1 }");
compare_bytecode("iter:task_file { 1 }", "it:task_file { 1 }");
compare_bytecode("iter:task_vma { 1 }", "it:task_vma { 1 }");
}
} // namespace bpftrace::test::probe
|
