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