[sanitizer] Improve FreeBSD ASLR detection
[llvm-project.git] / lldb / unittests / Expression / DWARFExpressionTest.cpp
blobfc4a63ce6a7b42a6a6c96e5be670ab68b6ae5432
1 //===-- DWARFExpressionTest.cpp -------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
9 #include "lldb/Expression/DWARFExpression.h"
10 #include "Plugins/Platform/Linux/PlatformLinux.h"
11 #include "Plugins/TypeSystem/Clang/TypeSystemClang.h"
12 #include "TestingSupport/Symbol/YAMLModuleTester.h"
13 #include "lldb/Core/Value.h"
14 #include "lldb/Core/Debugger.h"
15 #include "lldb/Core/dwarf.h"
16 #include "lldb/Host/HostInfo.h"
17 #include "lldb/Symbol/ObjectFile.h"
18 #include "lldb/Utility/Reproducer.h"
19 #include "lldb/Utility/StreamString.h"
20 #include "llvm/ADT/StringExtras.h"
21 #include "llvm/Testing/Support/Error.h"
22 #include "gtest/gtest.h"
24 using namespace lldb_private;
26 static llvm::Expected<Scalar> Evaluate(llvm::ArrayRef<uint8_t> expr,
27 lldb::ModuleSP module_sp = {},
28 DWARFUnit *unit = nullptr,
29 ExecutionContext *exe_ctx = nullptr) {
30 DataExtractor extractor(expr.data(), expr.size(), lldb::eByteOrderLittle,
31 /*addr_size*/ 4);
32 Value result;
33 Status status;
34 if (!DWARFExpression::Evaluate(exe_ctx, /*reg_ctx*/ nullptr, module_sp,
35 extractor, unit, lldb::eRegisterKindLLDB,
36 /*initial_value_ptr*/ nullptr,
37 /*object_address_ptr*/ nullptr, result,
38 &status))
39 return status.ToError();
41 switch (result.GetValueType()) {
42 case Value::ValueType::Scalar:
43 return result.GetScalar();
44 case Value::ValueType::LoadAddress:
45 return LLDB_INVALID_ADDRESS;
46 case Value::ValueType::HostAddress: {
47 // Convert small buffers to scalars to simplify the tests.
48 DataBufferHeap &buf = result.GetBuffer();
49 if (buf.GetByteSize() <= 8) {
50 uint64_t val = 0;
51 memcpy(&val, buf.GetBytes(), buf.GetByteSize());
52 return Scalar(llvm::APInt(buf.GetByteSize()*8, val, false));
55 LLVM_FALLTHROUGH;
56 default:
57 return status.ToError();
61 class DWARFExpressionTester : public YAMLModuleTester {
62 public:
63 using YAMLModuleTester::YAMLModuleTester;
64 llvm::Expected<Scalar> Eval(llvm::ArrayRef<uint8_t> expr) {
65 return ::Evaluate(expr, m_module_sp, m_dwarf_unit);
69 /// Unfortunately Scalar's operator==() is really picky.
70 static Scalar GetScalar(unsigned bits, uint64_t value, bool sign) {
71 Scalar scalar(value);
72 scalar.TruncOrExtendTo(bits, sign);
73 return scalar;
76 /// This is needed for the tests that use a mock process.
77 class DWARFExpressionMockProcessTest : public ::testing::Test {
78 public:
79 void SetUp() override {
80 llvm::cantFail(repro::Reproducer::Initialize(repro::ReproducerMode::Off, {}));
81 FileSystem::Initialize();
82 HostInfo::Initialize();
83 platform_linux::PlatformLinux::Initialize();
85 void TearDown() override {
86 platform_linux::PlatformLinux::Terminate();
87 HostInfo::Terminate();
88 FileSystem::Terminate();
89 repro::Reproducer::Terminate();
93 TEST(DWARFExpression, DW_OP_pick) {
94 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit1, DW_OP_lit0, DW_OP_pick, 0}),
95 llvm::HasValue(0));
96 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit1, DW_OP_lit0, DW_OP_pick, 1}),
97 llvm::HasValue(1));
98 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit1, DW_OP_lit0, DW_OP_pick, 2}),
99 llvm::Failed());
102 TEST(DWARFExpression, DW_OP_const) {
103 // Extend to address size.
104 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const1u, 0x88}), llvm::HasValue(0x88));
105 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const1s, 0x88}),
106 llvm::HasValue(0xffffff88));
107 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const2u, 0x47, 0x88}),
108 llvm::HasValue(0x8847));
109 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const2s, 0x47, 0x88}),
110 llvm::HasValue(0xffff8847));
111 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const4u, 0x44, 0x42, 0x47, 0x88}),
112 llvm::HasValue(0x88474244));
113 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const4s, 0x44, 0x42, 0x47, 0x88}),
114 llvm::HasValue(0x88474244));
116 // Truncate to address size.
117 EXPECT_THAT_EXPECTED(
118 Evaluate({DW_OP_const8u, 0x00, 0x11, 0x22, 0x33, 0x44, 0x42, 0x47, 0x88}),
119 llvm::HasValue(0x33221100));
120 EXPECT_THAT_EXPECTED(
121 Evaluate({DW_OP_const8s, 0x00, 0x11, 0x22, 0x33, 0x44, 0x42, 0x47, 0x88}),
122 llvm::HasValue(0x33221100));
124 // Don't truncate to address size for compatibility with clang (pr48087).
125 EXPECT_THAT_EXPECTED(
126 Evaluate({DW_OP_constu, 0x81, 0x82, 0x84, 0x88, 0x90, 0xa0, 0x40}),
127 llvm::HasValue(0x01010101010101));
128 EXPECT_THAT_EXPECTED(
129 Evaluate({DW_OP_consts, 0x81, 0x82, 0x84, 0x88, 0x90, 0xa0, 0x40}),
130 llvm::HasValue(0xffff010101010101));
133 TEST(DWARFExpression, DW_OP_convert) {
134 /// Auxiliary debug info.
135 const char *yamldata = R"(
136 --- !ELF
137 FileHeader:
138 Class: ELFCLASS64
139 Data: ELFDATA2LSB
140 Type: ET_EXEC
141 Machine: EM_386
142 DWARF:
143 debug_abbrev:
144 - Table:
145 - Code: 0x00000001
146 Tag: DW_TAG_compile_unit
147 Children: DW_CHILDREN_yes
148 Attributes:
149 - Attribute: DW_AT_language
150 Form: DW_FORM_data2
151 - Code: 0x00000002
152 Tag: DW_TAG_base_type
153 Children: DW_CHILDREN_no
154 Attributes:
155 - Attribute: DW_AT_encoding
156 Form: DW_FORM_data1
157 - Attribute: DW_AT_byte_size
158 Form: DW_FORM_data1
159 debug_info:
160 - Version: 4
161 AddrSize: 8
162 Entries:
163 - AbbrCode: 0x00000001
164 Values:
165 - Value: 0x000000000000000C
166 # 0x0000000e:
167 - AbbrCode: 0x00000002
168 Values:
169 - Value: 0x0000000000000007 # DW_ATE_unsigned
170 - Value: 0x0000000000000004
171 # 0x00000011:
172 - AbbrCode: 0x00000002
173 Values:
174 - Value: 0x0000000000000007 # DW_ATE_unsigned
175 - Value: 0x0000000000000008
176 # 0x00000014:
177 - AbbrCode: 0x00000002
178 Values:
179 - Value: 0x0000000000000005 # DW_ATE_signed
180 - Value: 0x0000000000000008
181 # 0x00000017:
182 - AbbrCode: 0x00000002
183 Values:
184 - Value: 0x0000000000000008 # DW_ATE_unsigned_char
185 - Value: 0x0000000000000001
186 # 0x0000001a:
187 - AbbrCode: 0x00000002
188 Values:
189 - Value: 0x0000000000000006 # DW_ATE_signed_char
190 - Value: 0x0000000000000001
191 # 0x0000001d:
192 - AbbrCode: 0x00000002
193 Values:
194 - Value: 0x000000000000000b # DW_ATE_numeric_string
195 - Value: 0x0000000000000001
196 - AbbrCode: 0x00000000
198 uint8_t offs_uint32_t = 0x0000000e;
199 uint8_t offs_uint64_t = 0x00000011;
200 uint8_t offs_sint64_t = 0x00000014;
201 uint8_t offs_uchar = 0x00000017;
202 uint8_t offs_schar = 0x0000001a;
204 DWARFExpressionTester t(yamldata);
205 ASSERT_TRUE((bool)t.GetDwarfUnit());
207 // Constant is given as little-endian.
208 bool is_signed = true;
209 bool not_signed = false;
212 // Positive tests.
215 // Leave as is.
216 EXPECT_THAT_EXPECTED(
217 t.Eval({DW_OP_const4u, 0x11, 0x22, 0x33, 0x44, //
218 DW_OP_convert, offs_uint32_t, DW_OP_stack_value}),
219 llvm::HasValue(GetScalar(64, 0x44332211, not_signed)));
221 // Zero-extend to 64 bits.
222 EXPECT_THAT_EXPECTED(
223 t.Eval({DW_OP_const4u, 0x11, 0x22, 0x33, 0x44, //
224 DW_OP_convert, offs_uint64_t, DW_OP_stack_value}),
225 llvm::HasValue(GetScalar(64, 0x44332211, not_signed)));
227 // Sign-extend to 64 bits.
228 EXPECT_THAT_EXPECTED(
229 t.Eval({DW_OP_const4s, 0xcc, 0xdd, 0xee, 0xff, //
230 DW_OP_convert, offs_sint64_t, DW_OP_stack_value}),
231 llvm::HasValue(GetScalar(64, 0xffffffffffeeddcc, is_signed)));
233 // Sign-extend, then truncate.
234 EXPECT_THAT_EXPECTED(
235 t.Eval({DW_OP_const4s, 0xcc, 0xdd, 0xee, 0xff, //
236 DW_OP_convert, offs_sint64_t, //
237 DW_OP_convert, offs_uint32_t, DW_OP_stack_value}),
238 llvm::HasValue(GetScalar(32, 0xffeeddcc, not_signed)));
240 // Truncate to default unspecified (pointer-sized) type.
241 EXPECT_THAT_EXPECTED(t.Eval({DW_OP_const4s, 0xcc, 0xdd, 0xee, 0xff, //
242 DW_OP_convert, offs_sint64_t, //
243 DW_OP_convert, 0x00, DW_OP_stack_value}),
244 llvm::HasValue(GetScalar(32, 0xffeeddcc, not_signed)));
246 // Truncate to 8 bits.
247 EXPECT_THAT_EXPECTED(t.Eval({DW_OP_const4s, 'A', 'B', 'C', 'D', DW_OP_convert,
248 offs_uchar, DW_OP_stack_value}),
249 llvm::HasValue(GetScalar(8, 'A', not_signed)));
251 // Also truncate to 8 bits.
252 EXPECT_THAT_EXPECTED(t.Eval({DW_OP_const4s, 'A', 'B', 'C', 'D', DW_OP_convert,
253 offs_schar, DW_OP_stack_value}),
254 llvm::HasValue(GetScalar(8, 'A', is_signed)));
257 // Errors.
260 // No Module.
261 EXPECT_THAT_ERROR(Evaluate({DW_OP_const1s, 'X', DW_OP_convert, 0x00}, nullptr,
262 t.GetDwarfUnit())
263 .takeError(),
264 llvm::Failed());
266 // No DIE.
267 EXPECT_THAT_ERROR(
268 t.Eval({DW_OP_const1s, 'X', DW_OP_convert, 0x01}).takeError(),
269 llvm::Failed());
271 // Unsupported.
272 EXPECT_THAT_ERROR(
273 t.Eval({DW_OP_const1s, 'X', DW_OP_convert, 0x1d}).takeError(),
274 llvm::Failed());
277 TEST(DWARFExpression, DW_OP_stack_value) {
278 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_stack_value}), llvm::Failed());
281 TEST(DWARFExpression, DW_OP_piece) {
282 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_const2u, 0x11, 0x22, DW_OP_piece, 2,
283 DW_OP_const2u, 0x33, 0x44, DW_OP_piece, 2}),
284 llvm::HasValue(GetScalar(32, 0x44332211, true)));
285 EXPECT_THAT_EXPECTED(
286 Evaluate({DW_OP_piece, 1, DW_OP_const1u, 0xff, DW_OP_piece, 1}),
287 // Note that the "00" should really be "undef", but we can't
288 // represent that yet.
289 llvm::HasValue(GetScalar(16, 0xff00, true)));
292 TEST(DWARFExpression, DW_OP_implicit_value) {
293 unsigned char bytes = 4;
295 EXPECT_THAT_EXPECTED(
296 Evaluate({DW_OP_implicit_value, bytes, 0x11, 0x22, 0x33, 0x44}),
297 llvm::HasValue(GetScalar(8 * bytes, 0x44332211, true)));
300 TEST(DWARFExpression, DW_OP_unknown) {
301 EXPECT_THAT_EXPECTED(
302 Evaluate({0xff}),
303 llvm::FailedWithMessage(
304 "Unhandled opcode DW_OP_unknown_ff in DWARFExpression"));
307 TEST_F(DWARFExpressionMockProcessTest, DW_OP_deref) {
308 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit0, DW_OP_deref}), llvm::Failed());
310 struct MockProcess : Process {
311 using Process::Process;
312 llvm::StringRef GetPluginName() override { return "mock process"; }
313 bool CanDebug(lldb::TargetSP target,
314 bool plugin_specified_by_name) override {
315 return false;
317 Status DoDestroy() override { return {}; }
318 void RefreshStateAfterStop() override {}
319 bool DoUpdateThreadList(ThreadList &old_thread_list,
320 ThreadList &new_thread_list) override {
321 return false;
323 size_t DoReadMemory(lldb::addr_t vm_addr, void *buf, size_t size,
324 Status &error) override {
325 for (size_t i = 0; i < size; ++i)
326 ((char *)buf)[i] = (vm_addr + i) & 0xff;
327 error.Clear();
328 return size;
332 // Set up a mock process.
333 ArchSpec arch("i386-pc-linux");
334 Platform::SetHostPlatform(
335 platform_linux::PlatformLinux::CreateInstance(true, &arch));
336 lldb::DebuggerSP debugger_sp = Debugger::CreateInstance();
337 ASSERT_TRUE(debugger_sp);
338 lldb::TargetSP target_sp;
339 lldb::PlatformSP platform_sp;
340 debugger_sp->GetTargetList().CreateTarget(
341 *debugger_sp, "", arch, eLoadDependentsNo, platform_sp, target_sp);
342 ASSERT_TRUE(target_sp);
343 ASSERT_TRUE(target_sp->GetArchitecture().IsValid());
344 ASSERT_TRUE(platform_sp);
345 lldb::ListenerSP listener_sp(Listener::MakeListener("dummy"));
346 lldb::ProcessSP process_sp =
347 std::make_shared<MockProcess>(target_sp, listener_sp);
348 ASSERT_TRUE(process_sp);
350 ExecutionContext exe_ctx(process_sp);
351 // Implicit location: *0x4.
352 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit4, DW_OP_deref, DW_OP_stack_value},
353 {}, {}, &exe_ctx),
354 llvm::HasValue(GetScalar(32, 0x07060504, false)));
355 // Memory location: *(*0x4).
356 // Evaluate returns LLDB_INVALID_ADDRESS for all load addresses.
357 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit4, DW_OP_deref}, {}, {}, &exe_ctx),
358 llvm::HasValue(Scalar(LLDB_INVALID_ADDRESS)));
359 // Memory location: *0x4.
360 // Evaluate returns LLDB_INVALID_ADDRESS for all load addresses.
361 EXPECT_THAT_EXPECTED(Evaluate({DW_OP_lit4}, {}, {}, &exe_ctx),
362 llvm::HasValue(Scalar(4)));
363 // Implicit location: *0x4.
364 // Evaluate returns LLDB_INVALID_ADDRESS for all load addresses.
365 EXPECT_THAT_EXPECTED(
366 Evaluate({DW_OP_lit4, DW_OP_deref, DW_OP_stack_value}, {}, {}, &exe_ctx),
367 llvm::HasValue(GetScalar(32, 0x07060504, false)));