1 //===-- IRExecutionUnit.cpp -------------------------------------*- C++ -*-===//
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 //===----------------------------------------------------------------------===//
8
9 #include "llvm/ExecutionEngine/ExecutionEngine.h"
10 #include "llvm/ExecutionEngine/ObjectCache.h"
11 #include "llvm/IR/Constants.h"
12 #include "llvm/IR/LLVMContext.h"
13 #include "llvm/IR/Module.h"
14 #include "llvm/Support/SourceMgr.h"
15 #include "llvm/Support/raw_ostream.h"
16
17 #include "lldb/Core/Debugger.h"
18 #include "lldb/Core/Disassembler.h"
19 #include "lldb/Core/Module.h"
20 #include "lldb/Core/Section.h"
21 #include "lldb/Expression/IRExecutionUnit.h"
22 #include "lldb/Symbol/CompileUnit.h"
23 #include "lldb/Symbol/SymbolContext.h"
24 #include "lldb/Symbol/SymbolFile.h"
25 #include "lldb/Symbol/SymbolVendor.h"
26 #include "lldb/Target/ExecutionContext.h"
27 #include "lldb/Target/LanguageRuntime.h"
28 #include "lldb/Target/Target.h"
29 #include "lldb/Utility/DataBufferHeap.h"
30 #include "lldb/Utility/DataExtractor.h"
31 #include "lldb/Utility/LLDBAssert.h"
32 #include "lldb/Utility/Log.h"
33
34 #include "lldb/../../source/Plugins/Language/CPlusPlus/CPlusPlusLanguage.h"
35 #include "lldb/../../source/Plugins/ObjectFile/JIT/ObjectFileJIT.h"
36
37 using namespace lldb_private;
38
IRExecutionUnit(std::unique_ptr<llvm::LLVMContext> & context_up,std::unique_ptr<llvm::Module> & module_up,ConstString & name,const lldb::TargetSP & target_sp,const SymbolContext & sym_ctx,std::vector<std::string> & cpu_features)39 IRExecutionUnit::IRExecutionUnit(std::unique_ptr<llvm::LLVMContext> &context_up,
40 std::unique_ptr<llvm::Module> &module_up,
41 ConstString &name,
42 const lldb::TargetSP &target_sp,
43 const SymbolContext &sym_ctx,
44 std::vector<std::string> &cpu_features)
45 : IRMemoryMap(target_sp), m_context_up(context_up.release()),
46 m_module_up(module_up.release()), m_module(m_module_up.get()),
47 m_cpu_features(cpu_features), m_name(name), m_sym_ctx(sym_ctx),
48 m_did_jit(false), m_function_load_addr(LLDB_INVALID_ADDRESS),
49 m_function_end_load_addr(LLDB_INVALID_ADDRESS),
50 m_reported_allocations(false) {}
51
WriteNow(const uint8_t * bytes,size_t size,Status & error)52 lldb::addr_t IRExecutionUnit::WriteNow(const uint8_t *bytes, size_t size,
53 Status &error) {
54 const bool zero_memory = false;
55 lldb::addr_t allocation_process_addr =
56 Malloc(size, 8, lldb::ePermissionsWritable | lldb::ePermissionsReadable,
57 eAllocationPolicyMirror, zero_memory, error);
58
59 if (!error.Success())
60 return LLDB_INVALID_ADDRESS;
61
62 WriteMemory(allocation_process_addr, bytes, size, error);
63
64 if (!error.Success()) {
65 Status err;
66 Free(allocation_process_addr, err);
67
68 return LLDB_INVALID_ADDRESS;
69 }
70
71 if (Log *log =
72 lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS)) {
73 DataBufferHeap my_buffer(size, 0);
74 Status err;
75 ReadMemory(my_buffer.GetBytes(), allocation_process_addr, size, err);
76
77 if (err.Success()) {
78 DataExtractor my_extractor(my_buffer.GetBytes(), my_buffer.GetByteSize(),
79 lldb::eByteOrderBig, 8);
80 my_extractor.PutToLog(log, 0, my_buffer.GetByteSize(),
81 allocation_process_addr, 16,
82 DataExtractor::TypeUInt8);
83 }
84 }
85
86 return allocation_process_addr;
87 }
88
FreeNow(lldb::addr_t allocation)89 void IRExecutionUnit::FreeNow(lldb::addr_t allocation) {
90 if (allocation == LLDB_INVALID_ADDRESS)
91 return;
92
93 Status err;
94
95 Free(allocation, err);
96 }
97
DisassembleFunction(Stream & stream,lldb::ProcessSP & process_wp)98 Status IRExecutionUnit::DisassembleFunction(Stream &stream,
99 lldb::ProcessSP &process_wp) {
100 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
101
102 ExecutionContext exe_ctx(process_wp);
103
104 Status ret;
105
106 ret.Clear();
107
108 lldb::addr_t func_local_addr = LLDB_INVALID_ADDRESS;
109 lldb::addr_t func_remote_addr = LLDB_INVALID_ADDRESS;
110
111 for (JittedFunction &function : m_jitted_functions) {
112 if (function.m_name == m_name) {
113 func_local_addr = function.m_local_addr;
114 func_remote_addr = function.m_remote_addr;
115 }
116 }
117
118 if (func_local_addr == LLDB_INVALID_ADDRESS) {
119 ret.SetErrorToGenericError();
120 ret.SetErrorStringWithFormat("Couldn't find function %s for disassembly",
121 m_name.AsCString());
122 return ret;
123 }
124
125 LLDB_LOGF(log,
126 "Found function, has local address 0x%" PRIx64
127 " and remote address 0x%" PRIx64,
128 (uint64_t)func_local_addr, (uint64_t)func_remote_addr);
129
130 std::pair<lldb::addr_t, lldb::addr_t> func_range;
131
132 func_range = GetRemoteRangeForLocal(func_local_addr);
133
134 if (func_range.first == 0 && func_range.second == 0) {
135 ret.SetErrorToGenericError();
136 ret.SetErrorStringWithFormat("Couldn't find code range for function %s",
137 m_name.AsCString());
138 return ret;
139 }
140
141 LLDB_LOGF(log, "Function's code range is [0x%" PRIx64 "+0x%" PRIx64 "]",
142 func_range.first, func_range.second);
143
144 Target *target = exe_ctx.GetTargetPtr();
145 if (!target) {
146 ret.SetErrorToGenericError();
147 ret.SetErrorString("Couldn't find the target");
148 return ret;
149 }
150
151 lldb::DataBufferSP buffer_sp(new DataBufferHeap(func_range.second, 0));
152
153 Process *process = exe_ctx.GetProcessPtr();
154 Status err;
155 process->ReadMemory(func_remote_addr, buffer_sp->GetBytes(),
156 buffer_sp->GetByteSize(), err);
157
158 if (!err.Success()) {
159 ret.SetErrorToGenericError();
160 ret.SetErrorStringWithFormat("Couldn't read from process: %s",
161 err.AsCString("unknown error"));
162 return ret;
163 }
164
165 ArchSpec arch(target->GetArchitecture());
166
167 const char *plugin_name = nullptr;
168 const char *flavor_string = nullptr;
169 lldb::DisassemblerSP disassembler_sp =
170 Disassembler::FindPlugin(arch, flavor_string, plugin_name);
171
172 if (!disassembler_sp) {
173 ret.SetErrorToGenericError();
174 ret.SetErrorStringWithFormat(
175 "Unable to find disassembler plug-in for %s architecture.",
176 arch.GetArchitectureName());
177 return ret;
178 }
179
180 if (!process) {
181 ret.SetErrorToGenericError();
182 ret.SetErrorString("Couldn't find the process");
183 return ret;
184 }
185
186 DataExtractor extractor(buffer_sp, process->GetByteOrder(),
187 target->GetArchitecture().GetAddressByteSize());
188
189 if (log) {
190 LLDB_LOGF(log, "Function data has contents:");
191 extractor.PutToLog(log, 0, extractor.GetByteSize(), func_remote_addr, 16,
192 DataExtractor::TypeUInt8);
193 }
194
195 disassembler_sp->DecodeInstructions(Address(func_remote_addr), extractor, 0,
196 UINT32_MAX, false, false);
197
198 InstructionList &instruction_list = disassembler_sp->GetInstructionList();
199 instruction_list.Dump(&stream, true, true, &exe_ctx);
200 return ret;
201 }
202
ReportInlineAsmError(const llvm::SMDiagnostic & diagnostic,void * Context,unsigned LocCookie)203 static void ReportInlineAsmError(const llvm::SMDiagnostic &diagnostic,
204 void *Context, unsigned LocCookie) {
205 Status *err = static_cast<Status *>(Context);
206
207 if (err && err->Success()) {
208 err->SetErrorToGenericError();
209 err->SetErrorStringWithFormat("Inline assembly error: %s",
210 diagnostic.getMessage().str().c_str());
211 }
212 }
213
ReportSymbolLookupError(ConstString name)214 void IRExecutionUnit::ReportSymbolLookupError(ConstString name) {
215 m_failed_lookups.push_back(name);
216 }
217
GetRunnableInfo(Status & error,lldb::addr_t & func_addr,lldb::addr_t & func_end)218 void IRExecutionUnit::GetRunnableInfo(Status &error, lldb::addr_t &func_addr,
219 lldb::addr_t &func_end) {
220 lldb::ProcessSP process_sp(GetProcessWP().lock());
221
222 static std::recursive_mutex s_runnable_info_mutex;
223
224 func_addr = LLDB_INVALID_ADDRESS;
225 func_end = LLDB_INVALID_ADDRESS;
226
227 if (!process_sp) {
228 error.SetErrorToGenericError();
229 error.SetErrorString("Couldn't write the JIT compiled code into the "
230 "process because the process is invalid");
231 return;
232 }
233
234 if (m_did_jit) {
235 func_addr = m_function_load_addr;
236 func_end = m_function_end_load_addr;
237
238 return;
239 };
240
241 std::lock_guard<std::recursive_mutex> guard(s_runnable_info_mutex);
242
243 m_did_jit = true;
244
245 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
246
247 std::string error_string;
248
249 if (log) {
250 std::string s;
251 llvm::raw_string_ostream oss(s);
252
253 m_module->print(oss, nullptr);
254
255 oss.flush();
256
257 LLDB_LOGF(log, "Module being sent to JIT: \n%s", s.c_str());
258 }
259
260 m_module_up->getContext().setInlineAsmDiagnosticHandler(ReportInlineAsmError,
261 &error);
262
263 llvm::EngineBuilder builder(std::move(m_module_up));
264 llvm::Triple triple(m_module->getTargetTriple());
265
266 builder.setEngineKind(llvm::EngineKind::JIT)
267 .setErrorStr(&error_string)
268 .setRelocationModel(triple.isOSBinFormatMachO()
269 ? llvm::Reloc::PIC_
270 : llvm::Reloc::Static)
271 .setMCJITMemoryManager(
272 std::unique_ptr<MemoryManager>(new MemoryManager(*this)))
273 .setOptLevel(llvm::CodeGenOpt::Less);
274
275 llvm::StringRef mArch;
276 llvm::StringRef mCPU;
277 llvm::SmallVector<std::string, 0> mAttrs;
278
279 for (std::string &feature : m_cpu_features)
280 mAttrs.push_back(feature);
281
282 llvm::TargetMachine *target_machine =
283 builder.selectTarget(triple, mArch, mCPU, mAttrs);
284
285 m_execution_engine_up.reset(builder.create(target_machine));
286
287 if (!m_execution_engine_up) {
288 error.SetErrorToGenericError();
289 error.SetErrorStringWithFormat("Couldn't JIT the function: %s",
290 error_string.c_str());
291 return;
292 }
293
294 m_strip_underscore =
295 (m_execution_engine_up->getDataLayout().getGlobalPrefix() == '_');
296
297 class ObjectDumper : public llvm::ObjectCache {
298 public:
299 void notifyObjectCompiled(const llvm::Module *module,
300 llvm::MemoryBufferRef object) override {
301 int fd = 0;
302 llvm::SmallVector<char, 256> result_path;
303 std::string object_name_model =
304 "jit-object-" + module->getModuleIdentifier() + "-%%%.o";
305 (void)llvm::sys::fs::createUniqueFile(object_name_model, fd, result_path);
306 llvm::raw_fd_ostream fds(fd, true);
307 fds.write(object.getBufferStart(), object.getBufferSize());
308 }
309
310 std::unique_ptr<llvm::MemoryBuffer>
311 getObject(const llvm::Module *module) override {
312 // Return nothing - we're just abusing the object-cache mechanism to dump
313 // objects.
314 return nullptr;
315 }
316 };
317
318 if (process_sp->GetTarget().GetEnableSaveObjects()) {
319 m_object_cache_up = std::make_unique<ObjectDumper>();
320 m_execution_engine_up->setObjectCache(m_object_cache_up.get());
321 }
322
323 // Make sure we see all sections, including ones that don't have
324 // relocations...
325 m_execution_engine_up->setProcessAllSections(true);
326
327 m_execution_engine_up->DisableLazyCompilation();
328
329 for (llvm::Function &function : *m_module) {
330 if (function.isDeclaration() || function.hasPrivateLinkage())
331 continue;
332
333 const bool external =
334 function.hasExternalLinkage() || function.hasLinkOnceODRLinkage();
335
336 void *fun_ptr = m_execution_engine_up->getPointerToFunction(&function);
337
338 if (!error.Success()) {
339 // We got an error through our callback!
340 return;
341 }
342
343 if (!fun_ptr) {
344 error.SetErrorToGenericError();
345 error.SetErrorStringWithFormat(
346 "'%s' was in the JITted module but wasn't lowered",
347 function.getName().str().c_str());
348 return;
349 }
350 m_jitted_functions.push_back(JittedFunction(
351 function.getName().str().c_str(), external, reinterpret_cast<uintptr_t>(fun_ptr)));
352 }
353
354 CommitAllocations(process_sp);
355 ReportAllocations(*m_execution_engine_up);
356
357 // We have to do this after calling ReportAllocations because for the MCJIT,
358 // getGlobalValueAddress will cause the JIT to perform all relocations. That
359 // can only be done once, and has to happen after we do the remapping from
360 // local -> remote. That means we don't know the local address of the
361 // Variables, but we don't need that for anything, so that's okay.
362
363 std::function<void(llvm::GlobalValue &)> RegisterOneValue = [this](
364 llvm::GlobalValue &val) {
365 if (val.hasExternalLinkage() && !val.isDeclaration()) {
366 uint64_t var_ptr_addr =
367 m_execution_engine_up->getGlobalValueAddress(val.getName().str());
368
369 lldb::addr_t remote_addr = GetRemoteAddressForLocal(var_ptr_addr);
370
371 // This is a really unfortunae API that sometimes returns local addresses
372 // and sometimes returns remote addresses, based on whether the variable
373 // was relocated during ReportAllocations or not.
374
375 if (remote_addr == LLDB_INVALID_ADDRESS) {
376 remote_addr = var_ptr_addr;
377 }
378
379 if (var_ptr_addr != 0)
380 m_jitted_global_variables.push_back(JittedGlobalVariable(
381 val.getName().str().c_str(), LLDB_INVALID_ADDRESS, remote_addr));
382 }
383 };
384
385 for (llvm::GlobalVariable &global_var : m_module->getGlobalList()) {
386 RegisterOneValue(global_var);
387 }
388
389 for (llvm::GlobalAlias &global_alias : m_module->getAliasList()) {
390 RegisterOneValue(global_alias);
391 }
392
393 WriteData(process_sp);
394
395 if (m_failed_lookups.size()) {
396 StreamString ss;
397
398 ss.PutCString("Couldn't lookup symbols:\n");
399
400 bool emitNewLine = false;
401
402 for (ConstString failed_lookup : m_failed_lookups) {
403 if (emitNewLine)
404 ss.PutCString("\n");
405 emitNewLine = true;
406 ss.PutCString(" ");
407 ss.PutCString(Mangled(failed_lookup)
408 .GetDemangledName(lldb::eLanguageTypeObjC_plus_plus)
409 .AsCString());
410 }
411
412 m_failed_lookups.clear();
413
414 error.SetErrorString(ss.GetString());
415
416 return;
417 }
418
419 m_function_load_addr = LLDB_INVALID_ADDRESS;
420 m_function_end_load_addr = LLDB_INVALID_ADDRESS;
421
422 for (JittedFunction &jitted_function : m_jitted_functions) {
423 jitted_function.m_remote_addr =
424 GetRemoteAddressForLocal(jitted_function.m_local_addr);
425
426 if (!m_name.IsEmpty() && jitted_function.m_name == m_name) {
427 AddrRange func_range =
428 GetRemoteRangeForLocal(jitted_function.m_local_addr);
429 m_function_end_load_addr = func_range.first + func_range.second;
430 m_function_load_addr = jitted_function.m_remote_addr;
431 }
432 }
433
434 if (log) {
435 LLDB_LOGF(log, "Code can be run in the target.");
436
437 StreamString disassembly_stream;
438
439 Status err = DisassembleFunction(disassembly_stream, process_sp);
440
441 if (!err.Success()) {
442 LLDB_LOGF(log, "Couldn't disassemble function : %s",
443 err.AsCString("unknown error"));
444 } else {
445 LLDB_LOGF(log, "Function disassembly:\n%s", disassembly_stream.GetData());
446 }
447
448 LLDB_LOGF(log, "Sections: ");
449 for (AllocationRecord &record : m_records) {
450 if (record.m_process_address != LLDB_INVALID_ADDRESS) {
451 record.dump(log);
452
453 DataBufferHeap my_buffer(record.m_size, 0);
454 Status err;
455 ReadMemory(my_buffer.GetBytes(), record.m_process_address,
456 record.m_size, err);
457
458 if (err.Success()) {
459 DataExtractor my_extractor(my_buffer.GetBytes(),
460 my_buffer.GetByteSize(),
461 lldb::eByteOrderBig, 8);
462 my_extractor.PutToLog(log, 0, my_buffer.GetByteSize(),
463 record.m_process_address, 16,
464 DataExtractor::TypeUInt8);
465 }
466 } else {
467 record.dump(log);
468
469 DataExtractor my_extractor((const void *)record.m_host_address,
470 record.m_size, lldb::eByteOrderBig, 8);
471 my_extractor.PutToLog(log, 0, record.m_size, record.m_host_address, 16,
472 DataExtractor::TypeUInt8);
473 }
474 }
475 }
476
477 func_addr = m_function_load_addr;
478 func_end = m_function_end_load_addr;
479
480 return;
481 }
482
~IRExecutionUnit()483 IRExecutionUnit::~IRExecutionUnit() {
484 m_module_up.reset();
485 m_execution_engine_up.reset();
486 m_context_up.reset();
487 }
488
MemoryManager(IRExecutionUnit & parent)489 IRExecutionUnit::MemoryManager::MemoryManager(IRExecutionUnit &parent)
490 : m_default_mm_up(new llvm::SectionMemoryManager()), m_parent(parent) {}
491
~MemoryManager()492 IRExecutionUnit::MemoryManager::~MemoryManager() {}
493
GetSectionTypeFromSectionName(const llvm::StringRef & name,IRExecutionUnit::AllocationKind alloc_kind)494 lldb::SectionType IRExecutionUnit::GetSectionTypeFromSectionName(
495 const llvm::StringRef &name, IRExecutionUnit::AllocationKind alloc_kind) {
496 lldb::SectionType sect_type = lldb::eSectionTypeCode;
497 switch (alloc_kind) {
498 case AllocationKind::Stub:
499 sect_type = lldb::eSectionTypeCode;
500 break;
501 case AllocationKind::Code:
502 sect_type = lldb::eSectionTypeCode;
503 break;
504 case AllocationKind::Data:
505 sect_type = lldb::eSectionTypeData;
506 break;
507 case AllocationKind::Global:
508 sect_type = lldb::eSectionTypeData;
509 break;
510 case AllocationKind::Bytes:
511 sect_type = lldb::eSectionTypeOther;
512 break;
513 }
514
515 if (!name.empty()) {
516 if (name.equals("__text") || name.equals(".text"))
517 sect_type = lldb::eSectionTypeCode;
518 else if (name.equals("__data") || name.equals(".data"))
519 sect_type = lldb::eSectionTypeCode;
520 else if (name.startswith("__debug_") || name.startswith(".debug_")) {
521 const uint32_t name_idx = name[0] == '_' ? 8 : 7;
522 llvm::StringRef dwarf_name(name.substr(name_idx));
523 switch (dwarf_name[0]) {
524 case 'a':
525 if (dwarf_name.equals("abbrev"))
526 sect_type = lldb::eSectionTypeDWARFDebugAbbrev;
527 else if (dwarf_name.equals("aranges"))
528 sect_type = lldb::eSectionTypeDWARFDebugAranges;
529 else if (dwarf_name.equals("addr"))
530 sect_type = lldb::eSectionTypeDWARFDebugAddr;
531 break;
532
533 case 'f':
534 if (dwarf_name.equals("frame"))
535 sect_type = lldb::eSectionTypeDWARFDebugFrame;
536 break;
537
538 case 'i':
539 if (dwarf_name.equals("info"))
540 sect_type = lldb::eSectionTypeDWARFDebugInfo;
541 break;
542
543 case 'l':
544 if (dwarf_name.equals("line"))
545 sect_type = lldb::eSectionTypeDWARFDebugLine;
546 else if (dwarf_name.equals("loc"))
547 sect_type = lldb::eSectionTypeDWARFDebugLoc;
548 else if (dwarf_name.equals("loclists"))
549 sect_type = lldb::eSectionTypeDWARFDebugLocLists;
550 break;
551
552 case 'm':
553 if (dwarf_name.equals("macinfo"))
554 sect_type = lldb::eSectionTypeDWARFDebugMacInfo;
555 break;
556
557 case 'p':
558 if (dwarf_name.equals("pubnames"))
559 sect_type = lldb::eSectionTypeDWARFDebugPubNames;
560 else if (dwarf_name.equals("pubtypes"))
561 sect_type = lldb::eSectionTypeDWARFDebugPubTypes;
562 break;
563
564 case 's':
565 if (dwarf_name.equals("str"))
566 sect_type = lldb::eSectionTypeDWARFDebugStr;
567 else if (dwarf_name.equals("str_offsets"))
568 sect_type = lldb::eSectionTypeDWARFDebugStrOffsets;
569 break;
570
571 case 'r':
572 if (dwarf_name.equals("ranges"))
573 sect_type = lldb::eSectionTypeDWARFDebugRanges;
574 break;
575
576 default:
577 break;
578 }
579 } else if (name.startswith("__apple_") || name.startswith(".apple_"))
580 sect_type = lldb::eSectionTypeInvalid;
581 else if (name.equals("__objc_imageinfo"))
582 sect_type = lldb::eSectionTypeOther;
583 }
584 return sect_type;
585 }
586
allocateCodeSection(uintptr_t Size,unsigned Alignment,unsigned SectionID,llvm::StringRef SectionName)587 uint8_t *IRExecutionUnit::MemoryManager::allocateCodeSection(
588 uintptr_t Size, unsigned Alignment, unsigned SectionID,
589 llvm::StringRef SectionName) {
590 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
591
592 uint8_t *return_value = m_default_mm_up->allocateCodeSection(
593 Size, Alignment, SectionID, SectionName);
594
595 m_parent.m_records.push_back(AllocationRecord(
596 (uintptr_t)return_value,
597 lldb::ePermissionsReadable | lldb::ePermissionsExecutable,
598 GetSectionTypeFromSectionName(SectionName, AllocationKind::Code), Size,
599 Alignment, SectionID, SectionName.str().c_str()));
600
601 LLDB_LOGF(log,
602 "IRExecutionUnit::allocateCodeSection(Size=0x%" PRIx64
603 ", Alignment=%u, SectionID=%u) = %p",
604 (uint64_t)Size, Alignment, SectionID, (void *)return_value);
605
606 if (m_parent.m_reported_allocations) {
607 Status err;
608 lldb::ProcessSP process_sp =
609 m_parent.GetBestExecutionContextScope()->CalculateProcess();
610
611 m_parent.CommitOneAllocation(process_sp, err, m_parent.m_records.back());
612 }
613
614 return return_value;
615 }
616
allocateDataSection(uintptr_t Size,unsigned Alignment,unsigned SectionID,llvm::StringRef SectionName,bool IsReadOnly)617 uint8_t *IRExecutionUnit::MemoryManager::allocateDataSection(
618 uintptr_t Size, unsigned Alignment, unsigned SectionID,
619 llvm::StringRef SectionName, bool IsReadOnly) {
620 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
621
622 uint8_t *return_value = m_default_mm_up->allocateDataSection(
623 Size, Alignment, SectionID, SectionName, IsReadOnly);
624
625 uint32_t permissions = lldb::ePermissionsReadable;
626 if (!IsReadOnly)
627 permissions |= lldb::ePermissionsWritable;
628 m_parent.m_records.push_back(AllocationRecord(
629 (uintptr_t)return_value, permissions,
630 GetSectionTypeFromSectionName(SectionName, AllocationKind::Data), Size,
631 Alignment, SectionID, SectionName.str().c_str()));
632 LLDB_LOGF(log,
633 "IRExecutionUnit::allocateDataSection(Size=0x%" PRIx64
634 ", Alignment=%u, SectionID=%u) = %p",
635 (uint64_t)Size, Alignment, SectionID, (void *)return_value);
636
637 if (m_parent.m_reported_allocations) {
638 Status err;
639 lldb::ProcessSP process_sp =
640 m_parent.GetBestExecutionContextScope()->CalculateProcess();
641
642 m_parent.CommitOneAllocation(process_sp, err, m_parent.m_records.back());
643 }
644
645 return return_value;
646 }
647
648 static ConstString
FindBestAlternateMangledName(ConstString demangled,const lldb::LanguageType & lang_type,const SymbolContext & sym_ctx)649 FindBestAlternateMangledName(ConstString demangled,
650 const lldb::LanguageType &lang_type,
651 const SymbolContext &sym_ctx) {
652 CPlusPlusLanguage::MethodName cpp_name(demangled);
653 std::string scope_qualified_name = cpp_name.GetScopeQualifiedName();
654
655 if (!scope_qualified_name.size())
656 return ConstString();
657
658 if (!sym_ctx.module_sp)
659 return ConstString();
660
661 lldb_private::SymbolFile *sym_file = sym_ctx.module_sp->GetSymbolFile();
662 if (!sym_file)
663 return ConstString();
664
665 std::vector<ConstString> alternates;
666 sym_file->GetMangledNamesForFunction(scope_qualified_name, alternates);
667
668 std::vector<ConstString> param_and_qual_matches;
669 std::vector<ConstString> param_matches;
670 for (size_t i = 0; i < alternates.size(); i++) {
671 ConstString alternate_mangled_name = alternates[i];
672 Mangled mangled(alternate_mangled_name);
673 ConstString demangled = mangled.GetDemangledName(lang_type);
674
675 CPlusPlusLanguage::MethodName alternate_cpp_name(demangled);
676 if (!cpp_name.IsValid())
677 continue;
678
679 if (alternate_cpp_name.GetArguments() == cpp_name.GetArguments()) {
680 if (alternate_cpp_name.GetQualifiers() == cpp_name.GetQualifiers())
681 param_and_qual_matches.push_back(alternate_mangled_name);
682 else
683 param_matches.push_back(alternate_mangled_name);
684 }
685 }
686
687 if (param_and_qual_matches.size())
688 return param_and_qual_matches[0]; // It is assumed that there will be only
689 // one!
690 else if (param_matches.size())
691 return param_matches[0]; // Return one of them as a best match
692 else
693 return ConstString();
694 }
695
696 struct IRExecutionUnit::SearchSpec {
697 ConstString name;
698 lldb::FunctionNameType mask;
699
SearchSpecIRExecutionUnit::SearchSpec700 SearchSpec(ConstString n,
701 lldb::FunctionNameType m = lldb::eFunctionNameTypeFull)
702 : name(n), mask(m) {}
703 };
704
CollectCandidateCNames(std::vector<IRExecutionUnit::SearchSpec> & C_specs,ConstString name)705 void IRExecutionUnit::CollectCandidateCNames(
706 std::vector<IRExecutionUnit::SearchSpec> &C_specs,
707 ConstString name) {
708 if (m_strip_underscore && name.AsCString()[0] == '_')
709 C_specs.insert(C_specs.begin(), ConstString(&name.AsCString()[1]));
710 C_specs.push_back(SearchSpec(name));
711 }
712
CollectCandidateCPlusPlusNames(std::vector<IRExecutionUnit::SearchSpec> & CPP_specs,const std::vector<SearchSpec> & C_specs,const SymbolContext & sc)713 void IRExecutionUnit::CollectCandidateCPlusPlusNames(
714 std::vector<IRExecutionUnit::SearchSpec> &CPP_specs,
715 const std::vector<SearchSpec> &C_specs, const SymbolContext &sc) {
716 for (const SearchSpec &C_spec : C_specs) {
717 ConstString name = C_spec.name;
718
719 if (CPlusPlusLanguage::IsCPPMangledName(name.GetCString())) {
720 Mangled mangled(name);
721 ConstString demangled =
722 mangled.GetDemangledName(lldb::eLanguageTypeC_plus_plus);
723
724 if (demangled) {
725 ConstString best_alternate_mangled_name = FindBestAlternateMangledName(
726 demangled, lldb::eLanguageTypeC_plus_plus, sc);
727
728 if (best_alternate_mangled_name) {
729 CPP_specs.push_back(best_alternate_mangled_name);
730 }
731 }
732 }
733
734 std::set<ConstString> alternates;
735 CPlusPlusLanguage::FindAlternateFunctionManglings(name, alternates);
736 CPP_specs.insert(CPP_specs.end(), alternates.begin(), alternates.end());
737 }
738 }
739
CollectFallbackNames(std::vector<SearchSpec> & fallback_specs,const std::vector<SearchSpec> & C_specs)740 void IRExecutionUnit::CollectFallbackNames(
741 std::vector<SearchSpec> &fallback_specs,
742 const std::vector<SearchSpec> &C_specs) {
743 // As a last-ditch fallback, try the base name for C++ names. It's terrible,
744 // but the DWARF doesn't always encode "extern C" correctly.
745
746 for (const SearchSpec &C_spec : C_specs) {
747 ConstString name = C_spec.name;
748
749 if (CPlusPlusLanguage::IsCPPMangledName(name.GetCString())) {
750 Mangled mangled_name(name);
751 ConstString demangled_name =
752 mangled_name.GetDemangledName(lldb::eLanguageTypeC_plus_plus);
753 if (!demangled_name.IsEmpty()) {
754 const char *demangled_cstr = demangled_name.AsCString();
755 const char *lparen_loc = strchr(demangled_cstr, '(');
756 if (lparen_loc) {
757 llvm::StringRef base_name(demangled_cstr,
758 lparen_loc - demangled_cstr);
759 fallback_specs.push_back(ConstString(base_name));
760 }
761 }
762 }
763 }
764 }
765
FindInSymbols(const std::vector<IRExecutionUnit::SearchSpec> & specs,const lldb_private::SymbolContext & sc,bool & symbol_was_missing_weak)766 lldb::addr_t IRExecutionUnit::FindInSymbols(
767 const std::vector<IRExecutionUnit::SearchSpec> &specs,
768 const lldb_private::SymbolContext &sc,
769 bool &symbol_was_missing_weak) {
770 symbol_was_missing_weak = false;
771 Target *target = sc.target_sp.get();
772
773 if (!target) {
774 // we shouldn't be doing any symbol lookup at all without a target
775 return LLDB_INVALID_ADDRESS;
776 }
777
778 for (const SearchSpec &spec : specs) {
779 SymbolContextList sc_list;
780
781 lldb::addr_t best_internal_load_address = LLDB_INVALID_ADDRESS;
782
783 std::function<bool(lldb::addr_t &, SymbolContextList &,
784 const lldb_private::SymbolContext &)>
785 get_external_load_address = [&best_internal_load_address, target,
786 &symbol_was_missing_weak](
787 lldb::addr_t &load_address, SymbolContextList &sc_list,
788 const lldb_private::SymbolContext &sc) -> lldb::addr_t {
789 load_address = LLDB_INVALID_ADDRESS;
790
791 if (sc_list.GetSize() == 0)
792 return false;
793
794 // missing_weak_symbol will be true only if we found only weak undefined
795 // references to this symbol.
796 symbol_was_missing_weak = true;
797 for (auto candidate_sc : sc_list.SymbolContexts()) {
798 // Only symbols can be weak undefined:
799 if (!candidate_sc.symbol)
800 symbol_was_missing_weak = false;
801 else if (candidate_sc.symbol->GetType() != lldb::eSymbolTypeUndefined
802 || !candidate_sc.symbol->IsWeak())
803 symbol_was_missing_weak = false;
804
805 const bool is_external =
806 (candidate_sc.function) ||
807 (candidate_sc.symbol && candidate_sc.symbol->IsExternal());
808 if (candidate_sc.symbol) {
809 load_address = candidate_sc.symbol->ResolveCallableAddress(*target);
810
811 if (load_address == LLDB_INVALID_ADDRESS) {
812 if (target->GetProcessSP())
813 load_address =
814 candidate_sc.symbol->GetAddress().GetLoadAddress(target);
815 else
816 load_address = candidate_sc.symbol->GetAddress().GetFileAddress();
817 }
818 }
819
820 if (load_address == LLDB_INVALID_ADDRESS && candidate_sc.function) {
821 if (target->GetProcessSP())
822 load_address = candidate_sc.function->GetAddressRange()
823 .GetBaseAddress()
824 .GetLoadAddress(target);
825 else
826 load_address = candidate_sc.function->GetAddressRange()
827 .GetBaseAddress()
828 .GetFileAddress();
829 }
830
831 if (load_address != LLDB_INVALID_ADDRESS) {
832 if (is_external) {
833 return true;
834 } else if (best_internal_load_address == LLDB_INVALID_ADDRESS) {
835 best_internal_load_address = load_address;
836 load_address = LLDB_INVALID_ADDRESS;
837 }
838 }
839 }
840
841 // You test the address of a weak symbol against NULL to see if it is
842 // present. So we should return 0 for a missing weak symbol.
843 if (symbol_was_missing_weak) {
844 load_address = 0;
845 return true;
846 }
847
848 return false;
849 };
850
851 if (sc.module_sp) {
852 sc.module_sp->FindFunctions(spec.name, nullptr, spec.mask,
853 true, // include_symbols
854 false, // include_inlines
855 sc_list);
856 }
857
858 lldb::addr_t load_address = LLDB_INVALID_ADDRESS;
859
860 if (get_external_load_address(load_address, sc_list, sc)) {
861 return load_address;
862 } else {
863 sc_list.Clear();
864 }
865
866 if (sc_list.GetSize() == 0 && sc.target_sp) {
867 sc.target_sp->GetImages().FindFunctions(spec.name, spec.mask,
868 true, // include_symbols
869 false, // include_inlines
870 sc_list);
871 }
872
873 if (get_external_load_address(load_address, sc_list, sc)) {
874 return load_address;
875 } else {
876 sc_list.Clear();
877 }
878
879 if (sc_list.GetSize() == 0 && sc.target_sp) {
880 sc.target_sp->GetImages().FindSymbolsWithNameAndType(
881 spec.name, lldb::eSymbolTypeAny, sc_list);
882 }
883
884 if (get_external_load_address(load_address, sc_list, sc)) {
885 return load_address;
886 }
887 // if there are any searches we try after this, add an sc_list.Clear() in
888 // an "else" clause here
889
890 if (best_internal_load_address != LLDB_INVALID_ADDRESS) {
891 return best_internal_load_address;
892 }
893 }
894
895 return LLDB_INVALID_ADDRESS;
896 }
897
898 lldb::addr_t
FindInRuntimes(const std::vector<SearchSpec> & specs,const lldb_private::SymbolContext & sc)899 IRExecutionUnit::FindInRuntimes(const std::vector<SearchSpec> &specs,
900 const lldb_private::SymbolContext &sc) {
901 lldb::TargetSP target_sp = sc.target_sp;
902
903 if (!target_sp) {
904 return LLDB_INVALID_ADDRESS;
905 }
906
907 lldb::ProcessSP process_sp = sc.target_sp->GetProcessSP();
908
909 if (!process_sp) {
910 return LLDB_INVALID_ADDRESS;
911 }
912
913 for (const SearchSpec &spec : specs) {
914 for (LanguageRuntime *runtime : process_sp->GetLanguageRuntimes()) {
915 lldb::addr_t symbol_load_addr = runtime->LookupRuntimeSymbol(spec.name);
916
917 if (symbol_load_addr != LLDB_INVALID_ADDRESS)
918 return symbol_load_addr;
919 }
920 }
921
922 return LLDB_INVALID_ADDRESS;
923 }
924
FindInUserDefinedSymbols(const std::vector<SearchSpec> & specs,const lldb_private::SymbolContext & sc)925 lldb::addr_t IRExecutionUnit::FindInUserDefinedSymbols(
926 const std::vector<SearchSpec> &specs,
927 const lldb_private::SymbolContext &sc) {
928 lldb::TargetSP target_sp = sc.target_sp;
929
930 for (const SearchSpec &spec : specs) {
931 lldb::addr_t symbol_load_addr = target_sp->GetPersistentSymbol(spec.name);
932
933 if (symbol_load_addr != LLDB_INVALID_ADDRESS)
934 return symbol_load_addr;
935 }
936
937 return LLDB_INVALID_ADDRESS;
938 }
939
940 lldb::addr_t
FindSymbol(lldb_private::ConstString name,bool & missing_weak)941 IRExecutionUnit::FindSymbol(lldb_private::ConstString name, bool &missing_weak) {
942 std::vector<SearchSpec> candidate_C_names;
943 std::vector<SearchSpec> candidate_CPlusPlus_names;
944
945 CollectCandidateCNames(candidate_C_names, name);
946
947 lldb::addr_t ret = FindInSymbols(candidate_C_names, m_sym_ctx, missing_weak);
948 if (ret != LLDB_INVALID_ADDRESS)
949 return ret;
950
951 // If we find the symbol in runtimes or user defined symbols it can't be
952 // a missing weak symbol.
953 missing_weak = false;
954 ret = FindInRuntimes(candidate_C_names, m_sym_ctx);
955 if (ret != LLDB_INVALID_ADDRESS)
956 return ret;
957
958 ret = FindInUserDefinedSymbols(candidate_C_names, m_sym_ctx);
959 if (ret != LLDB_INVALID_ADDRESS)
960 return ret;
961
962 CollectCandidateCPlusPlusNames(candidate_CPlusPlus_names, candidate_C_names,
963 m_sym_ctx);
964 ret = FindInSymbols(candidate_CPlusPlus_names, m_sym_ctx, missing_weak);
965 if (ret != LLDB_INVALID_ADDRESS)
966 return ret;
967
968 std::vector<SearchSpec> candidate_fallback_names;
969
970 CollectFallbackNames(candidate_fallback_names, candidate_C_names);
971 ret = FindInSymbols(candidate_fallback_names, m_sym_ctx, missing_weak);
972
973 return ret;
974 }
975
GetStaticInitializers(std::vector<lldb::addr_t> & static_initializers)976 void IRExecutionUnit::GetStaticInitializers(
977 std::vector<lldb::addr_t> &static_initializers) {
978 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
979
980 llvm::GlobalVariable *global_ctors =
981 m_module->getNamedGlobal("llvm.global_ctors");
982 if (!global_ctors) {
983 LLDB_LOG(log, "Couldn't find llvm.global_ctors.");
984 return;
985 }
986 auto *ctor_array =
987 llvm::dyn_cast<llvm::ConstantArray>(global_ctors->getInitializer());
988 if (!ctor_array) {
989 LLDB_LOG(log, "llvm.global_ctors not a ConstantArray.");
990 return;
991 }
992
993 for (llvm::Use &ctor_use : ctor_array->operands()) {
994 auto *ctor_struct = llvm::dyn_cast<llvm::ConstantStruct>(ctor_use);
995 if (!ctor_struct)
996 continue;
997 // this is standardized
998 lldbassert(ctor_struct->getNumOperands() == 3);
999 auto *ctor_function =
1000 llvm::dyn_cast<llvm::Function>(ctor_struct->getOperand(1));
1001 if (!ctor_function) {
1002 LLDB_LOG(log, "global_ctor doesn't contain an llvm::Function");
1003 continue;
1004 }
1005
1006 ConstString ctor_function_name(ctor_function->getName().str());
1007 LLDB_LOG(log, "Looking for callable jitted function with name {0}.",
1008 ctor_function_name);
1009
1010 for (JittedFunction &jitted_function : m_jitted_functions) {
1011 if (ctor_function_name != jitted_function.m_name)
1012 continue;
1013 if (jitted_function.m_remote_addr == LLDB_INVALID_ADDRESS) {
1014 LLDB_LOG(log, "Found jitted function with invalid address.");
1015 continue;
1016 }
1017 static_initializers.push_back(jitted_function.m_remote_addr);
1018 LLDB_LOG(log, "Calling function at address {0:x}.",
1019 jitted_function.m_remote_addr);
1020 break;
1021 }
1022 }
1023 }
1024
1025 llvm::JITSymbol
findSymbol(const std::string & Name)1026 IRExecutionUnit::MemoryManager::findSymbol(const std::string &Name) {
1027 bool missing_weak = false;
1028 uint64_t addr = GetSymbolAddressAndPresence(Name, missing_weak);
1029 // This is a weak symbol:
1030 if (missing_weak)
1031 return llvm::JITSymbol(addr,
1032 llvm::JITSymbolFlags::Exported | llvm::JITSymbolFlags::Weak);
1033 else
1034 return llvm::JITSymbol(addr, llvm::JITSymbolFlags::Exported);
1035 }
1036
1037 uint64_t
getSymbolAddress(const std::string & Name)1038 IRExecutionUnit::MemoryManager::getSymbolAddress(const std::string &Name) {
1039 bool missing_weak = false;
1040 return GetSymbolAddressAndPresence(Name, missing_weak);
1041 }
1042
1043 uint64_t
GetSymbolAddressAndPresence(const std::string & Name,bool & missing_weak)1044 IRExecutionUnit::MemoryManager::GetSymbolAddressAndPresence(
1045 const std::string &Name, bool &missing_weak) {
1046 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
1047
1048 ConstString name_cs(Name.c_str());
1049
1050 lldb::addr_t ret = m_parent.FindSymbol(name_cs, missing_weak);
1051
1052 if (ret == LLDB_INVALID_ADDRESS) {
1053 LLDB_LOGF(log,
1054 "IRExecutionUnit::getSymbolAddress(Name=\"%s\") = <not found>",
1055 Name.c_str());
1056
1057 m_parent.ReportSymbolLookupError(name_cs);
1058 return 0;
1059 } else {
1060 LLDB_LOGF(log, "IRExecutionUnit::getSymbolAddress(Name=\"%s\") = %" PRIx64,
1061 Name.c_str(), ret);
1062 return ret;
1063 }
1064 }
1065
getPointerToNamedFunction(const std::string & Name,bool AbortOnFailure)1066 void *IRExecutionUnit::MemoryManager::getPointerToNamedFunction(
1067 const std::string &Name, bool AbortOnFailure) {
1068 return (void *)getSymbolAddress(Name);
1069 }
1070
1071 lldb::addr_t
GetRemoteAddressForLocal(lldb::addr_t local_address)1072 IRExecutionUnit::GetRemoteAddressForLocal(lldb::addr_t local_address) {
1073 Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
1074
1075 for (AllocationRecord &record : m_records) {
1076 if (local_address >= record.m_host_address &&
1077 local_address < record.m_host_address + record.m_size) {
1078 if (record.m_process_address == LLDB_INVALID_ADDRESS)
1079 return LLDB_INVALID_ADDRESS;
1080
1081 lldb::addr_t ret =
1082 record.m_process_address + (local_address - record.m_host_address);
1083
1084 LLDB_LOGF(log,
1085 "IRExecutionUnit::GetRemoteAddressForLocal() found 0x%" PRIx64
1086 " in [0x%" PRIx64 "..0x%" PRIx64 "], and returned 0x%" PRIx64
1087 " from [0x%" PRIx64 "..0x%" PRIx64 "].",
1088 local_address, (uint64_t)record.m_host_address,
1089 (uint64_t)record.m_host_address + (uint64_t)record.m_size, ret,
1090 record.m_process_address,
1091 record.m_process_address + record.m_size);
1092
1093 return ret;
1094 }
1095 }
1096
1097 return LLDB_INVALID_ADDRESS;
1098 }
1099
1100 IRExecutionUnit::AddrRange
GetRemoteRangeForLocal(lldb::addr_t local_address)1101 IRExecutionUnit::GetRemoteRangeForLocal(lldb::addr_t local_address) {
1102 for (AllocationRecord &record : m_records) {
1103 if (local_address >= record.m_host_address &&
1104 local_address < record.m_host_address + record.m_size) {
1105 if (record.m_process_address == LLDB_INVALID_ADDRESS)
1106 return AddrRange(0, 0);
1107
1108 return AddrRange(record.m_process_address, record.m_size);
1109 }
1110 }
1111
1112 return AddrRange(0, 0);
1113 }
1114
CommitOneAllocation(lldb::ProcessSP & process_sp,Status & error,AllocationRecord & record)1115 bool IRExecutionUnit::CommitOneAllocation(lldb::ProcessSP &process_sp,
1116 Status &error,
1117 AllocationRecord &record) {
1118 if (record.m_process_address != LLDB_INVALID_ADDRESS) {
1119 return true;
1120 }
1121
1122 switch (record.m_sect_type) {
1123 case lldb::eSectionTypeInvalid:
1124 case lldb::eSectionTypeDWARFDebugAbbrev:
1125 case lldb::eSectionTypeDWARFDebugAddr:
1126 case lldb::eSectionTypeDWARFDebugAranges:
1127 case lldb::eSectionTypeDWARFDebugCuIndex:
1128 case lldb::eSectionTypeDWARFDebugFrame:
1129 case lldb::eSectionTypeDWARFDebugInfo:
1130 case lldb::eSectionTypeDWARFDebugLine:
1131 case lldb::eSectionTypeDWARFDebugLoc:
1132 case lldb::eSectionTypeDWARFDebugLocLists:
1133 case lldb::eSectionTypeDWARFDebugMacInfo:
1134 case lldb::eSectionTypeDWARFDebugPubNames:
1135 case lldb::eSectionTypeDWARFDebugPubTypes:
1136 case lldb::eSectionTypeDWARFDebugRanges:
1137 case lldb::eSectionTypeDWARFDebugStr:
1138 case lldb::eSectionTypeDWARFDebugStrOffsets:
1139 case lldb::eSectionTypeDWARFAppleNames:
1140 case lldb::eSectionTypeDWARFAppleTypes:
1141 case lldb::eSectionTypeDWARFAppleNamespaces:
1142 case lldb::eSectionTypeDWARFAppleObjC:
1143 case lldb::eSectionTypeDWARFGNUDebugAltLink:
1144 error.Clear();
1145 break;
1146 default:
1147 const bool zero_memory = false;
1148 record.m_process_address =
1149 Malloc(record.m_size, record.m_alignment, record.m_permissions,
1150 eAllocationPolicyProcessOnly, zero_memory, error);
1151 break;
1152 }
1153
1154 return error.Success();
1155 }
1156
CommitAllocations(lldb::ProcessSP & process_sp)1157 bool IRExecutionUnit::CommitAllocations(lldb::ProcessSP &process_sp) {
1158 bool ret = true;
1159
1160 lldb_private::Status err;
1161
1162 for (AllocationRecord &record : m_records) {
1163 ret = CommitOneAllocation(process_sp, err, record);
1164
1165 if (!ret) {
1166 break;
1167 }
1168 }
1169
1170 if (!ret) {
1171 for (AllocationRecord &record : m_records) {
1172 if (record.m_process_address != LLDB_INVALID_ADDRESS) {
1173 Free(record.m_process_address, err);
1174 record.m_process_address = LLDB_INVALID_ADDRESS;
1175 }
1176 }
1177 }
1178
1179 return ret;
1180 }
1181
ReportAllocations(llvm::ExecutionEngine & engine)1182 void IRExecutionUnit::ReportAllocations(llvm::ExecutionEngine &engine) {
1183 m_reported_allocations = true;
1184
1185 for (AllocationRecord &record : m_records) {
1186 if (record.m_process_address == LLDB_INVALID_ADDRESS)
1187 continue;
1188
1189 if (record.m_section_id == eSectionIDInvalid)
1190 continue;
1191
1192 engine.mapSectionAddress((void *)record.m_host_address,
1193 record.m_process_address);
1194 }
1195
1196 // Trigger re-application of relocations.
1197 engine.finalizeObject();
1198 }
1199
WriteData(lldb::ProcessSP & process_sp)1200 bool IRExecutionUnit::WriteData(lldb::ProcessSP &process_sp) {
1201 bool wrote_something = false;
1202 for (AllocationRecord &record : m_records) {
1203 if (record.m_process_address != LLDB_INVALID_ADDRESS) {
1204 lldb_private::Status err;
1205 WriteMemory(record.m_process_address, (uint8_t *)record.m_host_address,
1206 record.m_size, err);
1207 if (err.Success())
1208 wrote_something = true;
1209 }
1210 }
1211 return wrote_something;
1212 }
1213
dump(Log * log)1214 void IRExecutionUnit::AllocationRecord::dump(Log *log) {
1215 if (!log)
1216 return;
1217
1218 LLDB_LOGF(log,
1219 "[0x%llx+0x%llx]->0x%llx (alignment %d, section ID %d, name %s)",
1220 (unsigned long long)m_host_address, (unsigned long long)m_size,
1221 (unsigned long long)m_process_address, (unsigned)m_alignment,
1222 (unsigned)m_section_id, m_name.c_str());
1223 }
1224
GetByteOrder() const1225 lldb::ByteOrder IRExecutionUnit::GetByteOrder() const {
1226 ExecutionContext exe_ctx(GetBestExecutionContextScope());
1227 return exe_ctx.GetByteOrder();
1228 }
1229
GetAddressByteSize() const1230 uint32_t IRExecutionUnit::GetAddressByteSize() const {
1231 ExecutionContext exe_ctx(GetBestExecutionContextScope());
1232 return exe_ctx.GetAddressByteSize();
1233 }
1234
PopulateSymtab(lldb_private::ObjectFile * obj_file,lldb_private::Symtab & symtab)1235 void IRExecutionUnit::PopulateSymtab(lldb_private::ObjectFile *obj_file,
1236 lldb_private::Symtab &symtab) {
1237 // No symbols yet...
1238 }
1239
PopulateSectionList(lldb_private::ObjectFile * obj_file,lldb_private::SectionList & section_list)1240 void IRExecutionUnit::PopulateSectionList(
1241 lldb_private::ObjectFile *obj_file,
1242 lldb_private::SectionList §ion_list) {
1243 for (AllocationRecord &record : m_records) {
1244 if (record.m_size > 0) {
1245 lldb::SectionSP section_sp(new lldb_private::Section(
1246 obj_file->GetModule(), obj_file, record.m_section_id,
1247 ConstString(record.m_name), record.m_sect_type,
1248 record.m_process_address, record.m_size,
1249 record.m_host_address, // file_offset (which is the host address for
1250 // the data)
1251 record.m_size, // file_size
1252 0,
1253 record.m_permissions)); // flags
1254 section_list.AddSection(section_sp);
1255 }
1256 }
1257 }
1258
GetArchitecture()1259 ArchSpec IRExecutionUnit::GetArchitecture() {
1260 ExecutionContext exe_ctx(GetBestExecutionContextScope());
1261 if(Target *target = exe_ctx.GetTargetPtr())
1262 return target->GetArchitecture();
1263 return ArchSpec();
1264 }
1265
GetJITModule()1266 lldb::ModuleSP IRExecutionUnit::GetJITModule() {
1267 ExecutionContext exe_ctx(GetBestExecutionContextScope());
1268 Target *target = exe_ctx.GetTargetPtr();
1269 if (!target)
1270 return nullptr;
1271
1272 auto Delegate = std::static_pointer_cast<lldb_private::ObjectFileJITDelegate>(
1273 shared_from_this());
1274
1275 lldb::ModuleSP jit_module_sp =
1276 lldb_private::Module::CreateModuleFromObjectFile<ObjectFileJIT>(Delegate);
1277 if (!jit_module_sp)
1278 return nullptr;
1279
1280 bool changed = false;
1281 jit_module_sp->SetLoadAddress(*target, 0, true, changed);
1282 return jit_module_sp;
1283 }
1284