1 //===-- IRExecutionUnit.cpp -------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/ExecutionEngine/ExecutionEngine.h"
11 #include "llvm/IR/LLVMContext.h"
12 #include "llvm/IR/Module.h"
13 #include "llvm/Support/SourceMgr.h"
14 #include "llvm/Support/raw_ostream.h"
15 
16 #include "lldb/Core/DataBufferHeap.h"
17 #include "lldb/Core/DataExtractor.h"
18 #include "lldb/Core/Debugger.h"
19 #include "lldb/Core/Disassembler.h"
20 #include "lldb/Core/Log.h"
21 #include "lldb/Core/Module.h"
22 #include "lldb/Core/Section.h"
23 #include "lldb/Symbol/SymbolContext.h"
24 #include "lldb/Expression/IRExecutionUnit.h"
25 #include "lldb/Target/ExecutionContext.h"
26 #include "lldb/Target/Target.h"
27 #include "lldb/Target/ObjCLanguageRuntime.h"
28 
29 using namespace lldb_private;
30 
31 IRExecutionUnit::IRExecutionUnit (std::unique_ptr<llvm::LLVMContext> &context_ap,
32                                   std::unique_ptr<llvm::Module> &module_ap,
33                                   ConstString &name,
34                                   const lldb::TargetSP &target_sp,
35                                   std::vector<std::string> &cpu_features) :
36     IRMemoryMap(target_sp),
37     m_context_ap(context_ap.release()),
38     m_module_ap(module_ap.release()),
39     m_module(m_module_ap.get()),
40     m_cpu_features(cpu_features),
41     m_name(name),
42     m_did_jit(false),
43     m_function_load_addr(LLDB_INVALID_ADDRESS),
44     m_function_end_load_addr(LLDB_INVALID_ADDRESS)
45 {
46 }
47 
48 lldb::addr_t
49 IRExecutionUnit::WriteNow (const uint8_t *bytes,
50                            size_t size,
51                            Error &error)
52 {
53     lldb::addr_t allocation_process_addr = Malloc (size,
54                                                    8,
55                                                    lldb::ePermissionsWritable | lldb::ePermissionsReadable,
56                                                    eAllocationPolicyMirror,
57                                                    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     {
66         Error err;
67         Free (allocation_process_addr, err);
68 
69         return LLDB_INVALID_ADDRESS;
70     }
71 
72     if (Log *log = lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS))
73     {
74         DataBufferHeap my_buffer(size, 0);
75         Error err;
76         ReadMemory(my_buffer.GetBytes(), allocation_process_addr, size, err);
77 
78         if (err.Success())
79         {
80             DataExtractor my_extractor(my_buffer.GetBytes(), my_buffer.GetByteSize(), lldb::eByteOrderBig, 8);
81             my_extractor.PutToLog(log, 0, my_buffer.GetByteSize(), allocation_process_addr, 16, DataExtractor::TypeUInt8);
82         }
83     }
84 
85     return allocation_process_addr;
86 }
87 
88 void
89 IRExecutionUnit::FreeNow (lldb::addr_t allocation)
90 {
91     if (allocation == LLDB_INVALID_ADDRESS)
92         return;
93 
94     Error err;
95 
96     Free(allocation, err);
97 }
98 
99 Error
100 IRExecutionUnit::DisassembleFunction (Stream &stream,
101                                       lldb::ProcessSP &process_wp)
102 {
103     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
104 
105     ExecutionContext exe_ctx(process_wp);
106 
107     Error ret;
108 
109     ret.Clear();
110 
111     lldb::addr_t func_local_addr = LLDB_INVALID_ADDRESS;
112     lldb::addr_t func_remote_addr = LLDB_INVALID_ADDRESS;
113 
114     for (JittedFunction &function : m_jitted_functions)
115     {
116         if (strstr(function.m_name.c_str(), m_name.AsCString()))
117         {
118             func_local_addr = function.m_local_addr;
119             func_remote_addr = function.m_remote_addr;
120         }
121     }
122 
123     if (func_local_addr == LLDB_INVALID_ADDRESS)
124     {
125         ret.SetErrorToGenericError();
126         ret.SetErrorStringWithFormat("Couldn't find function %s for disassembly", m_name.AsCString());
127         return ret;
128     }
129 
130     if (log)
131         log->Printf("Found function, has local address 0x%" PRIx64 " and remote address 0x%" PRIx64, (uint64_t)func_local_addr, (uint64_t)func_remote_addr);
132 
133     std::pair <lldb::addr_t, lldb::addr_t> func_range;
134 
135     func_range = GetRemoteRangeForLocal(func_local_addr);
136 
137     if (func_range.first == 0 && func_range.second == 0)
138     {
139         ret.SetErrorToGenericError();
140         ret.SetErrorStringWithFormat("Couldn't find code range for function %s", m_name.AsCString());
141         return ret;
142     }
143 
144     if (log)
145         log->Printf("Function's code range is [0x%" PRIx64 "+0x%" PRIx64 "]", func_range.first, func_range.second);
146 
147     Target *target = exe_ctx.GetTargetPtr();
148     if (!target)
149     {
150         ret.SetErrorToGenericError();
151         ret.SetErrorString("Couldn't find the target");
152         return ret;
153     }
154 
155     lldb::DataBufferSP buffer_sp(new DataBufferHeap(func_range.second, 0));
156 
157     Process *process = exe_ctx.GetProcessPtr();
158     Error err;
159     process->ReadMemory(func_remote_addr, buffer_sp->GetBytes(), buffer_sp->GetByteSize(), err);
160 
161     if (!err.Success())
162     {
163         ret.SetErrorToGenericError();
164         ret.SetErrorStringWithFormat("Couldn't read from process: %s", err.AsCString("unknown error"));
165         return ret;
166     }
167 
168     ArchSpec arch(target->GetArchitecture());
169 
170     const char *plugin_name = NULL;
171     const char *flavor_string = NULL;
172     lldb::DisassemblerSP disassembler_sp = Disassembler::FindPlugin(arch, flavor_string, plugin_name);
173 
174     if (!disassembler_sp)
175     {
176         ret.SetErrorToGenericError();
177         ret.SetErrorStringWithFormat("Unable to find disassembler plug-in for %s architecture.", arch.GetArchitectureName());
178         return ret;
179     }
180 
181     if (!process)
182     {
183         ret.SetErrorToGenericError();
184         ret.SetErrorString("Couldn't find the process");
185         return ret;
186     }
187 
188     DataExtractor extractor(buffer_sp,
189                             process->GetByteOrder(),
190                             target->GetArchitecture().GetAddressByteSize());
191 
192     if (log)
193     {
194         log->Printf("Function data has contents:");
195         extractor.PutToLog (log,
196                             0,
197                             extractor.GetByteSize(),
198                             func_remote_addr,
199                             16,
200                             DataExtractor::TypeUInt8);
201     }
202 
203     disassembler_sp->DecodeInstructions (Address (func_remote_addr), extractor, 0, UINT32_MAX, false, false);
204 
205     InstructionList &instruction_list = disassembler_sp->GetInstructionList();
206     instruction_list.Dump(&stream, true, true, &exe_ctx);
207 
208     // FIXME: The DisassemblerLLVMC has a reference cycle and won't go away if it has any active instructions.
209     // I'll fix that but for now, just clear the list and it will go away nicely.
210     disassembler_sp->GetInstructionList().Clear();
211     return ret;
212 }
213 
214 static void ReportInlineAsmError(const llvm::SMDiagnostic &diagnostic, void *Context, unsigned LocCookie)
215 {
216     Error *err = static_cast<Error*>(Context);
217 
218     if (err && err->Success())
219     {
220         err->SetErrorToGenericError();
221         err->SetErrorStringWithFormat("Inline assembly error: %s", diagnostic.getMessage().str().c_str());
222     }
223 }
224 
225 void
226 IRExecutionUnit::ReportSymbolLookupError(const ConstString &name)
227 {
228     m_failed_lookups.push_back(name);
229 }
230 
231 void
232 IRExecutionUnit::GetRunnableInfo(Error &error,
233                                  lldb::addr_t &func_addr,
234                                  lldb::addr_t &func_end)
235 {
236     lldb::ProcessSP process_sp(GetProcessWP().lock());
237 
238     static Mutex s_runnable_info_mutex(Mutex::Type::eMutexTypeRecursive);
239 
240     func_addr = LLDB_INVALID_ADDRESS;
241     func_end = LLDB_INVALID_ADDRESS;
242 
243     if (!process_sp)
244     {
245         error.SetErrorToGenericError();
246         error.SetErrorString("Couldn't write the JIT compiled code into the process because the process is invalid");
247         return;
248     }
249 
250     if (m_did_jit)
251     {
252         func_addr = m_function_load_addr;
253         func_end = m_function_end_load_addr;
254 
255         return;
256     };
257 
258     Mutex::Locker runnable_info_mutex_locker(s_runnable_info_mutex);
259 
260     m_did_jit = true;
261 
262     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
263 
264     std::string error_string;
265 
266     if (log)
267     {
268         std::string s;
269         llvm::raw_string_ostream oss(s);
270 
271         m_module->print(oss, NULL);
272 
273         oss.flush();
274 
275         log->Printf ("Module being sent to JIT: \n%s", s.c_str());
276     }
277 
278     llvm::Triple triple(m_module->getTargetTriple());
279     llvm::Function *function = m_module->getFunction (m_name.AsCString());
280     llvm::Reloc::Model relocModel;
281     llvm::CodeModel::Model codeModel;
282 
283     if (triple.isOSBinFormatELF())
284     {
285         relocModel = llvm::Reloc::Static;
286         // This will be small for 32-bit and large for 64-bit.
287         codeModel = llvm::CodeModel::JITDefault;
288     }
289     else
290     {
291         relocModel = llvm::Reloc::PIC_;
292         codeModel = llvm::CodeModel::Small;
293     }
294 
295     m_module_ap->getContext().setInlineAsmDiagnosticHandler(ReportInlineAsmError, &error);
296 
297     llvm::EngineBuilder builder(std::move(m_module_ap));
298 
299     builder.setEngineKind(llvm::EngineKind::JIT)
300     .setErrorStr(&error_string)
301     .setRelocationModel(relocModel)
302     .setMCJITMemoryManager(std::unique_ptr<MemoryManager>(new MemoryManager(*this)))
303     .setCodeModel(codeModel)
304     .setOptLevel(llvm::CodeGenOpt::Less);
305 
306     llvm::StringRef mArch;
307     llvm::StringRef mCPU;
308     llvm::SmallVector<std::string, 0> mAttrs;
309 
310     for (std::string &feature : m_cpu_features)
311         mAttrs.push_back(feature);
312 
313     llvm::TargetMachine *target_machine = builder.selectTarget(triple,
314                                                                mArch,
315                                                                mCPU,
316                                                                mAttrs);
317 
318     m_execution_engine_ap.reset(builder.create(target_machine));
319 
320     if (!m_execution_engine_ap.get())
321     {
322         error.SetErrorToGenericError();
323         error.SetErrorStringWithFormat("Couldn't JIT the function: %s", error_string.c_str());
324         return;
325     }
326 
327     // Make sure we see all sections, including ones that don't have relocations...
328     m_execution_engine_ap->setProcessAllSections(true);
329 
330     m_execution_engine_ap->DisableLazyCompilation();
331 
332     // We don't actually need the function pointer here, this just forces it to get resolved.
333 
334     void *fun_ptr = m_execution_engine_ap->getPointerToFunction(function);
335 
336     if (!error.Success())
337     {
338         // We got an error through our callback!
339         return;
340     }
341 
342     if (!function)
343     {
344         error.SetErrorToGenericError();
345         error.SetErrorStringWithFormat("Couldn't find '%s' in the JITted module", m_name.AsCString());
346         return;
347     }
348 
349     if (!fun_ptr)
350     {
351         error.SetErrorToGenericError();
352         error.SetErrorStringWithFormat("'%s' was in the JITted module but wasn't lowered", m_name.AsCString());
353         return;
354     }
355 
356     m_jitted_functions.push_back (JittedFunction(m_name.AsCString(), (lldb::addr_t)fun_ptr));
357 
358     CommitAllocations(process_sp);
359     ReportAllocations(*m_execution_engine_ap);
360     WriteData(process_sp);
361 
362     if (m_failed_lookups.size())
363     {
364         StreamString ss;
365 
366         ss.PutCString("Couldn't lookup symbols:\n");
367 
368         bool emitNewLine = false;
369 
370         for (const ConstString &failed_lookup : m_failed_lookups)
371         {
372             if (emitNewLine)
373                 ss.PutCString("\n");
374             emitNewLine = true;
375             ss.PutCString("  ");
376             ss.PutCString(Mangled(failed_lookup).GetDemangledName(lldb::eLanguageTypeObjC_plus_plus).AsCString());
377         }
378 
379         m_failed_lookups.clear();
380 
381         error.SetErrorString(ss.GetData());
382 
383         return;
384     }
385 
386     m_function_load_addr = LLDB_INVALID_ADDRESS;
387     m_function_end_load_addr = LLDB_INVALID_ADDRESS;
388 
389     for (JittedFunction &jitted_function : m_jitted_functions)
390     {
391         jitted_function.m_remote_addr = GetRemoteAddressForLocal (jitted_function.m_local_addr);
392 
393         if (!jitted_function.m_name.compare(m_name.AsCString()))
394         {
395             AddrRange func_range = GetRemoteRangeForLocal(jitted_function.m_local_addr);
396             m_function_end_load_addr = func_range.first + func_range.second;
397             m_function_load_addr = jitted_function.m_remote_addr;
398         }
399     }
400 
401     if (log)
402     {
403         log->Printf("Code can be run in the target.");
404 
405         StreamString disassembly_stream;
406 
407         Error err = DisassembleFunction(disassembly_stream, process_sp);
408 
409         if (!err.Success())
410         {
411             log->Printf("Couldn't disassemble function : %s", err.AsCString("unknown error"));
412         }
413         else
414         {
415             log->Printf("Function disassembly:\n%s", disassembly_stream.GetData());
416         }
417 
418         log->Printf("Sections: ");
419         for (AllocationRecord &record : m_records)
420         {
421             if (record.m_process_address != LLDB_INVALID_ADDRESS)
422             {
423                 record.dump(log);
424 
425                 DataBufferHeap my_buffer(record.m_size, 0);
426                 Error err;
427                 ReadMemory(my_buffer.GetBytes(), record.m_process_address, record.m_size, err);
428 
429                 if (err.Success())
430                 {
431                     DataExtractor my_extractor(my_buffer.GetBytes(), my_buffer.GetByteSize(), lldb::eByteOrderBig, 8);
432                     my_extractor.PutToLog(log, 0, my_buffer.GetByteSize(), record.m_process_address, 16, DataExtractor::TypeUInt8);
433                 }
434             }
435         }
436     }
437 
438     func_addr = m_function_load_addr;
439     func_end = m_function_end_load_addr;
440 
441     return;
442 }
443 
444 IRExecutionUnit::~IRExecutionUnit ()
445 {
446     m_module_ap.reset();
447     m_execution_engine_ap.reset();
448     m_context_ap.reset();
449 }
450 
451 IRExecutionUnit::MemoryManager::MemoryManager (IRExecutionUnit &parent) :
452     m_default_mm_ap (new llvm::SectionMemoryManager()),
453     m_parent (parent)
454 {
455 }
456 
457 IRExecutionUnit::MemoryManager::~MemoryManager ()
458 {
459 }
460 
461 lldb::SectionType
462 IRExecutionUnit::GetSectionTypeFromSectionName (const llvm::StringRef &name, IRExecutionUnit::AllocationKind alloc_kind)
463 {
464     lldb::SectionType sect_type = lldb::eSectionTypeCode;
465     switch (alloc_kind)
466     {
467         case AllocationKind::Stub:  sect_type = lldb::eSectionTypeCode; break;
468         case AllocationKind::Code:  sect_type = lldb::eSectionTypeCode; break;
469         case AllocationKind::Data:  sect_type = lldb::eSectionTypeData; break;
470         case AllocationKind::Global:sect_type = lldb::eSectionTypeData; break;
471         case AllocationKind::Bytes: sect_type = lldb::eSectionTypeOther; break;
472     }
473 
474     if (!name.empty())
475     {
476         if (name.equals("__text") || name.equals(".text"))
477             sect_type = lldb::eSectionTypeCode;
478         else if (name.equals("__data") || name.equals(".data"))
479             sect_type = lldb::eSectionTypeCode;
480         else if (name.startswith("__debug_") || name.startswith(".debug_"))
481         {
482             const uint32_t name_idx = name[0] == '_' ? 8 : 7;
483             llvm::StringRef dwarf_name(name.substr(name_idx));
484             switch (dwarf_name[0])
485             {
486                 case 'a':
487                     if (dwarf_name.equals("abbrev"))
488                         sect_type = lldb::eSectionTypeDWARFDebugAbbrev;
489                     else if (dwarf_name.equals("aranges"))
490                         sect_type = lldb::eSectionTypeDWARFDebugAranges;
491                     else if (dwarf_name.equals("addr"))
492                         sect_type = lldb::eSectionTypeDWARFDebugAddr;
493                     break;
494 
495                 case 'f':
496                     if (dwarf_name.equals("frame"))
497                         sect_type = lldb::eSectionTypeDWARFDebugFrame;
498                     break;
499 
500                 case 'i':
501                     if (dwarf_name.equals("info"))
502                         sect_type = lldb::eSectionTypeDWARFDebugInfo;
503                     break;
504 
505                 case 'l':
506                     if (dwarf_name.equals("line"))
507                         sect_type = lldb::eSectionTypeDWARFDebugLine;
508                     else if (dwarf_name.equals("loc"))
509                         sect_type = lldb::eSectionTypeDWARFDebugLoc;
510                     break;
511 
512                 case 'm':
513                     if (dwarf_name.equals("macinfo"))
514                         sect_type = lldb::eSectionTypeDWARFDebugMacInfo;
515                     break;
516 
517                 case 'p':
518                     if (dwarf_name.equals("pubnames"))
519                         sect_type = lldb::eSectionTypeDWARFDebugPubNames;
520                     else if (dwarf_name.equals("pubtypes"))
521                         sect_type = lldb::eSectionTypeDWARFDebugPubTypes;
522                     break;
523 
524                 case 's':
525                     if (dwarf_name.equals("str"))
526                         sect_type = lldb::eSectionTypeDWARFDebugStr;
527                     else if (dwarf_name.equals("str_offsets"))
528                         sect_type = lldb::eSectionTypeDWARFDebugStrOffsets;
529                     break;
530 
531                 case 'r':
532                     if (dwarf_name.equals("ranges"))
533                         sect_type = lldb::eSectionTypeDWARFDebugRanges;
534                     break;
535 
536                 default:
537                     break;
538             }
539         }
540         else if (name.startswith("__apple_") || name.startswith(".apple_"))
541         {
542 #if 0
543             const uint32_t name_idx = name[0] == '_' ? 8 : 7;
544             llvm::StringRef apple_name(name.substr(name_idx));
545             switch (apple_name[0])
546             {
547                 case 'n':
548                     if (apple_name.equals("names"))
549                         sect_type = lldb::eSectionTypeDWARFAppleNames;
550                     else if (apple_name.equals("namespac") || apple_name.equals("namespaces"))
551                         sect_type = lldb::eSectionTypeDWARFAppleNamespaces;
552                     break;
553                 case 't':
554                     if (apple_name.equals("types"))
555                         sect_type = lldb::eSectionTypeDWARFAppleTypes;
556                     break;
557                 case 'o':
558                     if (apple_name.equals("objc"))
559                         sect_type = lldb::eSectionTypeDWARFAppleObjC;
560                     break;
561                 default:
562                     break;
563             }
564 #else
565             sect_type = lldb::eSectionTypeInvalid;
566 #endif
567         }
568         else if (name.equals("__objc_imageinfo"))
569             sect_type = lldb::eSectionTypeOther;
570     }
571     return sect_type;
572 }
573 
574 uint8_t *
575 IRExecutionUnit::MemoryManager::allocateCodeSection(uintptr_t Size,
576                                                     unsigned Alignment,
577                                                     unsigned SectionID,
578                                                     llvm::StringRef SectionName)
579 {
580     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
581 
582     uint8_t *return_value = m_default_mm_ap->allocateCodeSection(Size, Alignment, SectionID, SectionName);
583 
584     m_parent.m_records.push_back(AllocationRecord((uintptr_t)return_value,
585                                                   lldb::ePermissionsReadable | lldb::ePermissionsExecutable,
586                                                   GetSectionTypeFromSectionName (SectionName, AllocationKind::Code),
587                                                   Size,
588                                                   Alignment,
589                                                   SectionID,
590                                                   SectionName.str().c_str()));
591 
592     if (log)
593     {
594         log->Printf("IRExecutionUnit::allocateCodeSection(Size=0x%" PRIx64 ", Alignment=%u, SectionID=%u) = %p",
595                     (uint64_t)Size, Alignment, SectionID, (void *)return_value);
596     }
597 
598     return return_value;
599 }
600 
601 uint8_t *
602 IRExecutionUnit::MemoryManager::allocateDataSection(uintptr_t Size,
603                                                     unsigned Alignment,
604                                                     unsigned SectionID,
605                                                     llvm::StringRef SectionName,
606                                                     bool IsReadOnly)
607 {
608     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
609 
610     uint8_t *return_value = m_default_mm_ap->allocateDataSection(Size, Alignment, SectionID, SectionName, IsReadOnly);
611 
612     uint32_t permissions = lldb::ePermissionsReadable;
613     if (!IsReadOnly)
614         permissions |= lldb::ePermissionsWritable;
615     m_parent.m_records.push_back(AllocationRecord((uintptr_t)return_value,
616                                                   permissions,
617                                                   GetSectionTypeFromSectionName (SectionName, AllocationKind::Data),
618                                                   Size,
619                                                   Alignment,
620                                                   SectionID,
621                                                   SectionName.str().c_str()));
622     if (log)
623     {
624         log->Printf("IRExecutionUnit::allocateDataSection(Size=0x%" PRIx64 ", Alignment=%u, SectionID=%u) = %p",
625                     (uint64_t)Size, Alignment, SectionID, (void *)return_value);
626     }
627 
628     return return_value;
629 }
630 
631 uint64_t
632 IRExecutionUnit::MemoryManager::getSymbolAddress(const std::string &Name)
633 {
634     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
635 
636     SymbolContextList sc_list;
637 
638     ExecutionContextScope *exe_scope = m_parent.GetBestExecutionContextScope();
639 
640     lldb::TargetSP target_sp = exe_scope->CalculateTarget();
641 
642     const char *name = Name.c_str();
643 
644     ConstString bare_name_cs(name);
645     ConstString name_cs;
646 
647     if (name[0] == '_')
648         name_cs = ConstString(name + 1);
649 
650     if (!target_sp)
651     {
652         if (log)
653             log->Printf("IRExecutionUnit::getSymbolAddress(Name=\"%s\") = <no target>",
654                         Name.c_str());
655 
656         m_parent.ReportSymbolLookupError(name_cs);
657 
658         return 0xbad0bad0;
659     }
660 
661     uint32_t num_matches = 0;
662     lldb::ProcessSP process_sp = exe_scope->CalculateProcess();
663 
664     if (!name_cs.IsEmpty())
665     {
666         target_sp->GetImages().FindSymbolsWithNameAndType(name_cs, lldb::eSymbolTypeAny, sc_list);
667         num_matches = sc_list.GetSize();
668     }
669 
670     if (!num_matches)
671     {
672         target_sp->GetImages().FindSymbolsWithNameAndType(bare_name_cs, lldb::eSymbolTypeAny, sc_list);
673         num_matches = sc_list.GetSize();
674     }
675 
676     lldb::addr_t symbol_load_addr = LLDB_INVALID_ADDRESS;
677 
678     for (uint32_t i=0; i<num_matches && (symbol_load_addr == 0 || symbol_load_addr == LLDB_INVALID_ADDRESS); i++)
679     {
680         SymbolContext sym_ctx;
681         sc_list.GetContextAtIndex(i, sym_ctx);
682 
683         symbol_load_addr = sym_ctx.symbol->ResolveCallableAddress(*target_sp);
684 
685         if (symbol_load_addr == LLDB_INVALID_ADDRESS)
686             symbol_load_addr = sym_ctx.symbol->GetAddress().GetLoadAddress(target_sp.get());
687     }
688 
689     if (symbol_load_addr == LLDB_INVALID_ADDRESS && process_sp && name_cs)
690     {
691         // Try the Objective-C language runtime.
692 
693         ObjCLanguageRuntime *runtime = process_sp->GetObjCLanguageRuntime();
694 
695         if (runtime)
696             symbol_load_addr = runtime->LookupRuntimeSymbol(name_cs);
697     }
698 
699     if (symbol_load_addr == LLDB_INVALID_ADDRESS)
700     {
701         if (log)
702             log->Printf("IRExecutionUnit::getSymbolAddress(Name=\"%s\") = <not found>",
703                         name);
704 
705         m_parent.ReportSymbolLookupError(bare_name_cs);
706 
707         return 0xbad0bad0;
708     }
709 
710     if (log)
711         log->Printf("IRExecutionUnit::getSymbolAddress(Name=\"%s\") = %" PRIx64,
712                     name,
713                     symbol_load_addr);
714 
715     if (symbol_load_addr == 0)
716         return 0xbad00add;
717 
718     return symbol_load_addr;
719 }
720 
721 void *
722 IRExecutionUnit::MemoryManager::getPointerToNamedFunction(const std::string &Name,
723                                                           bool AbortOnFailure) {
724     assert (sizeof(void *) == 8);
725 
726     return (void*)getSymbolAddress(Name);
727 }
728 
729 lldb::addr_t
730 IRExecutionUnit::GetRemoteAddressForLocal (lldb::addr_t local_address)
731 {
732     Log *log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
733 
734     for (AllocationRecord &record : m_records)
735     {
736         if (local_address >= record.m_host_address &&
737             local_address < record.m_host_address + record.m_size)
738         {
739             if (record.m_process_address == LLDB_INVALID_ADDRESS)
740                 return LLDB_INVALID_ADDRESS;
741 
742             lldb::addr_t ret = record.m_process_address + (local_address - record.m_host_address);
743 
744             if (log)
745             {
746                 log->Printf("IRExecutionUnit::GetRemoteAddressForLocal() found 0x%" PRIx64 " in [0x%" PRIx64 "..0x%" PRIx64 "], and returned 0x%" PRIx64 " from [0x%" PRIx64 "..0x%" PRIx64 "].",
747                             local_address,
748                             (uint64_t)record.m_host_address,
749                             (uint64_t)record.m_host_address + (uint64_t)record.m_size,
750                             ret,
751                             record.m_process_address,
752                             record.m_process_address + record.m_size);
753             }
754 
755             return ret;
756         }
757     }
758 
759     return LLDB_INVALID_ADDRESS;
760 }
761 
762 IRExecutionUnit::AddrRange
763 IRExecutionUnit::GetRemoteRangeForLocal (lldb::addr_t local_address)
764 {
765     for (AllocationRecord &record : m_records)
766     {
767         if (local_address >= record.m_host_address &&
768             local_address < record.m_host_address + record.m_size)
769         {
770             if (record.m_process_address == LLDB_INVALID_ADDRESS)
771                 return AddrRange(0, 0);
772 
773             return AddrRange(record.m_process_address, record.m_size);
774         }
775     }
776 
777     return AddrRange (0, 0);
778 }
779 
780 bool
781 IRExecutionUnit::CommitAllocations (lldb::ProcessSP &process_sp)
782 {
783     bool ret = true;
784 
785     lldb_private::Error err;
786 
787     for (AllocationRecord &record : m_records)
788     {
789         if (record.m_process_address != LLDB_INVALID_ADDRESS)
790             continue;
791 
792         switch (record.m_sect_type)
793         {
794         case lldb::eSectionTypeInvalid:
795         case lldb::eSectionTypeDWARFDebugAbbrev:
796         case lldb::eSectionTypeDWARFDebugAddr:
797         case lldb::eSectionTypeDWARFDebugAranges:
798         case lldb::eSectionTypeDWARFDebugFrame:
799         case lldb::eSectionTypeDWARFDebugInfo:
800         case lldb::eSectionTypeDWARFDebugLine:
801         case lldb::eSectionTypeDWARFDebugLoc:
802         case lldb::eSectionTypeDWARFDebugMacInfo:
803         case lldb::eSectionTypeDWARFDebugPubNames:
804         case lldb::eSectionTypeDWARFDebugPubTypes:
805         case lldb::eSectionTypeDWARFDebugRanges:
806         case lldb::eSectionTypeDWARFDebugStr:
807         case lldb::eSectionTypeDWARFDebugStrOffsets:
808         case lldb::eSectionTypeDWARFAppleNames:
809         case lldb::eSectionTypeDWARFAppleTypes:
810         case lldb::eSectionTypeDWARFAppleNamespaces:
811         case lldb::eSectionTypeDWARFAppleObjC:
812             err.Clear();
813             break;
814         default:
815             record.m_process_address = Malloc (record.m_size,
816                                                record.m_alignment,
817                                                record.m_permissions,
818                                                eAllocationPolicyProcessOnly,
819                                                err);
820             break;
821         }
822 
823         if (!err.Success())
824         {
825             ret = false;
826             break;
827         }
828     }
829 
830     if (!ret)
831     {
832         for (AllocationRecord &record : m_records)
833         {
834             if (record.m_process_address != LLDB_INVALID_ADDRESS)
835             {
836                 Free(record.m_process_address, err);
837                 record.m_process_address = LLDB_INVALID_ADDRESS;
838             }
839         }
840     }
841 
842     return ret;
843 }
844 
845 void
846 IRExecutionUnit::ReportAllocations (llvm::ExecutionEngine &engine)
847 {
848     for (AllocationRecord &record : m_records)
849     {
850         if (record.m_process_address == LLDB_INVALID_ADDRESS)
851             continue;
852 
853         if (record.m_section_id == eSectionIDInvalid)
854             continue;
855 
856         engine.mapSectionAddress((void*)record.m_host_address, record.m_process_address);
857     }
858 
859     // Trigger re-application of relocations.
860     engine.finalizeObject();
861 }
862 
863 bool
864 IRExecutionUnit::WriteData (lldb::ProcessSP &process_sp)
865 {
866     bool wrote_something = false;
867     for (AllocationRecord &record : m_records)
868     {
869         if (record.m_process_address != LLDB_INVALID_ADDRESS)
870         {
871             lldb_private::Error err;
872             WriteMemory (record.m_process_address, (uint8_t*)record.m_host_address, record.m_size, err);
873             if (err.Success())
874                 wrote_something = true;
875         }
876     }
877     return wrote_something;
878 }
879 
880 void
881 IRExecutionUnit::AllocationRecord::dump (Log *log)
882 {
883     if (!log)
884         return;
885 
886     log->Printf("[0x%llx+0x%llx]->0x%llx (alignment %d, section ID %d)",
887                 (unsigned long long)m_host_address,
888                 (unsigned long long)m_size,
889                 (unsigned long long)m_process_address,
890                 (unsigned)m_alignment,
891                 (unsigned)m_section_id);
892 }
893 
894 
895 lldb::ByteOrder
896 IRExecutionUnit::GetByteOrder () const
897 {
898     ExecutionContext exe_ctx (GetBestExecutionContextScope());
899     return exe_ctx.GetByteOrder();
900 }
901 
902 uint32_t
903 IRExecutionUnit::GetAddressByteSize () const
904 {
905     ExecutionContext exe_ctx (GetBestExecutionContextScope());
906     return exe_ctx.GetAddressByteSize();
907 }
908 
909 void
910 IRExecutionUnit::PopulateSymtab (lldb_private::ObjectFile *obj_file,
911                                  lldb_private::Symtab &symtab)
912 {
913     // No symbols yet...
914 }
915 
916 
917 void
918 IRExecutionUnit::PopulateSectionList (lldb_private::ObjectFile *obj_file,
919                                       lldb_private::SectionList &section_list)
920 {
921     for (AllocationRecord &record : m_records)
922     {
923         if (record.m_size > 0)
924         {
925             lldb::SectionSP section_sp (new lldb_private::Section (obj_file->GetModule(),
926                                                                    obj_file,
927                                                                    record.m_section_id,
928                                                                    ConstString(record.m_name),
929                                                                    record.m_sect_type,
930                                                                    record.m_process_address,
931                                                                    record.m_size,
932                                                                    record.m_host_address,   // file_offset (which is the host address for the data)
933                                                                    record.m_size,           // file_size
934                                                                    0,
935                                                                    record.m_permissions));  // flags
936             section_list.AddSection (section_sp);
937         }
938     }
939 }
940 
941 bool
942 IRExecutionUnit::GetArchitecture (lldb_private::ArchSpec &arch)
943 {
944     ExecutionContext exe_ctx (GetBestExecutionContextScope());
945     Target *target = exe_ctx.GetTargetPtr();
946     if (target)
947         arch = target->GetArchitecture();
948     else
949         arch.Clear();
950     return arch.IsValid();
951 }
952 
953 lldb::ModuleSP
954 IRExecutionUnit::GetJITModule ()
955 {
956     ExecutionContext exe_ctx (GetBestExecutionContextScope());
957     Target *target = exe_ctx.GetTargetPtr();
958     if (target)
959     {
960         lldb::ModuleSP jit_module_sp = lldb_private::Module::CreateJITModule (std::static_pointer_cast<lldb_private::ObjectFileJITDelegate>(shared_from_this()));
961         if (jit_module_sp)
962         {
963             bool changed = false;
964             jit_module_sp->SetLoadAddress(*target, 0, true, changed);
965         }
966         return jit_module_sp;
967     }
968     return lldb::ModuleSP();
969 }
970