1 //===-- SymbolFileDWARFDebugMap.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 "SymbolFileDWARFDebugMap.h"
11 
12 #include "DWARFDebugAranges.h"
13 
14 #include "lldb/Core/Module.h"
15 #include "lldb/Core/ModuleList.h"
16 #include "lldb/Core/PluginManager.h"
17 #include "lldb/Core/RangeMap.h"
18 #include "lldb/Core/Section.h"
19 #include "lldb/Host/FileSystem.h"
20 #include "lldb/Utility/RegularExpression.h"
21 #include "lldb/Utility/Timer.h"
22 
23 //#define DEBUG_OSO_DMAP // DO NOT CHECKIN WITH THIS NOT COMMENTED OUT
24 #if defined(DEBUG_OSO_DMAP)
25 #include "lldb/Core/StreamFile.h"
26 #endif
27 
28 #include "lldb/Symbol/CompileUnit.h"
29 #include "lldb/Symbol/LineTable.h"
30 #include "lldb/Symbol/ObjectFile.h"
31 #include "lldb/Symbol/SymbolVendor.h"
32 #include "lldb/Symbol/TypeMap.h"
33 #include "lldb/Symbol/VariableList.h"
34 #include "llvm/Support/ScopedPrinter.h"
35 
36 #include "LogChannelDWARF.h"
37 #include "SymbolFileDWARF.h"
38 
39 using namespace lldb;
40 using namespace lldb_private;
41 
42 // Subclass lldb_private::Module so we can intercept the
43 // "Module::GetObjectFile()" (so we can fixup the object file sections) and
44 // also for "Module::GetSymbolVendor()" (so we can fixup the symbol file id.
45 
46 const SymbolFileDWARFDebugMap::FileRangeMap &
47 SymbolFileDWARFDebugMap::CompileUnitInfo::GetFileRangeMap(
48     SymbolFileDWARFDebugMap *exe_symfile) {
49   if (file_range_map_valid)
50     return file_range_map;
51 
52   file_range_map_valid = true;
53 
54   Module *oso_module = exe_symfile->GetModuleByCompUnitInfo(this);
55   if (!oso_module)
56     return file_range_map;
57 
58   ObjectFile *oso_objfile = oso_module->GetObjectFile();
59   if (!oso_objfile)
60     return file_range_map;
61 
62   Log *log(LogChannelDWARF::GetLogIfAll(DWARF_LOG_DEBUG_MAP));
63   if (log) {
64     ConstString object_name(oso_module->GetObjectName());
65     log->Printf(
66         "%p: SymbolFileDWARFDebugMap::CompileUnitInfo::GetFileRangeMap ('%s')",
67         static_cast<void *>(this),
68         oso_module->GetSpecificationDescription().c_str());
69   }
70 
71   std::vector<SymbolFileDWARFDebugMap::CompileUnitInfo *> cu_infos;
72   if (exe_symfile->GetCompUnitInfosForModule(oso_module, cu_infos)) {
73     for (auto comp_unit_info : cu_infos) {
74       Symtab *exe_symtab = exe_symfile->GetObjectFile()->GetSymtab();
75       ModuleSP oso_module_sp(oso_objfile->GetModule());
76       Symtab *oso_symtab = oso_objfile->GetSymtab();
77 
78       /// const uint32_t fun_resolve_flags = SymbolContext::Module |
79       /// eSymbolContextCompUnit | eSymbolContextFunction;
80       // SectionList *oso_sections = oso_objfile->Sections();
81       // Now we need to make sections that map from zero based object file
82       // addresses to where things ended up in the main executable.
83 
84       assert(comp_unit_info->first_symbol_index != UINT32_MAX);
85       // End index is one past the last valid symbol index
86       const uint32_t oso_end_idx = comp_unit_info->last_symbol_index + 1;
87       for (uint32_t idx = comp_unit_info->first_symbol_index +
88                           2; // Skip the N_SO and N_OSO
89            idx < oso_end_idx; ++idx) {
90         Symbol *exe_symbol = exe_symtab->SymbolAtIndex(idx);
91         if (exe_symbol) {
92           if (!exe_symbol->IsDebug())
93             continue;
94 
95           switch (exe_symbol->GetType()) {
96           default:
97             break;
98 
99           case eSymbolTypeCode: {
100             // For each N_FUN, or function that we run into in the debug map we
101             // make a new section that we add to the sections found in the .o
102             // file. This new section has the file address set to what the
103             // addresses are in the .o file, and the load address is adjusted
104             // to match where it ended up in the final executable! We do this
105             // before we parse any dwarf info so that when it goes get parsed
106             // all section/offset addresses that get registered will resolve
107             // correctly to the new addresses in the main executable.
108 
109             // First we find the original symbol in the .o file's symbol table
110             Symbol *oso_fun_symbol = oso_symtab->FindFirstSymbolWithNameAndType(
111                 exe_symbol->GetMangled().GetName(lldb::eLanguageTypeUnknown,
112                                                  Mangled::ePreferMangled),
113                 eSymbolTypeCode, Symtab::eDebugNo, Symtab::eVisibilityAny);
114             if (oso_fun_symbol) {
115               // Add the inverse OSO file address to debug map entry mapping
116               exe_symfile->AddOSOFileRange(
117                   this, exe_symbol->GetAddressRef().GetFileAddress(),
118                   exe_symbol->GetByteSize(),
119                   oso_fun_symbol->GetAddressRef().GetFileAddress(),
120                   oso_fun_symbol->GetByteSize());
121             }
122           } break;
123 
124           case eSymbolTypeData: {
125             // For each N_GSYM we remap the address for the global by making a
126             // new section that we add to the sections found in the .o file.
127             // This new section has the file address set to what the addresses
128             // are in the .o file, and the load address is adjusted to match
129             // where it ended up in the final executable! We do this before we
130             // parse any dwarf info so that when it goes get parsed all
131             // section/offset addresses that get registered will resolve
132             // correctly to the new addresses in the main executable. We
133             // initially set the section size to be 1 byte, but will need to
134             // fix up these addresses further after all globals have been
135             // parsed to span the gaps, or we can find the global variable
136             // sizes from the DWARF info as we are parsing.
137 
138             // Next we find the non-stab entry that corresponds to the N_GSYM
139             // in the .o file
140             Symbol *oso_gsym_symbol =
141                 oso_symtab->FindFirstSymbolWithNameAndType(
142                     exe_symbol->GetMangled().GetName(lldb::eLanguageTypeUnknown,
143                                                      Mangled::ePreferMangled),
144                     eSymbolTypeData, Symtab::eDebugNo, Symtab::eVisibilityAny);
145             if (exe_symbol && oso_gsym_symbol && exe_symbol->ValueIsAddress() &&
146                 oso_gsym_symbol->ValueIsAddress()) {
147               // Add the inverse OSO file address to debug map entry mapping
148               exe_symfile->AddOSOFileRange(
149                   this, exe_symbol->GetAddressRef().GetFileAddress(),
150                   exe_symbol->GetByteSize(),
151                   oso_gsym_symbol->GetAddressRef().GetFileAddress(),
152                   oso_gsym_symbol->GetByteSize());
153             }
154           } break;
155           }
156         }
157       }
158 
159       exe_symfile->FinalizeOSOFileRanges(this);
160       // We don't need the symbols anymore for the .o files
161       oso_objfile->ClearSymtab();
162     }
163   }
164   return file_range_map;
165 }
166 
167 class DebugMapModule : public Module {
168 public:
169   DebugMapModule(const ModuleSP &exe_module_sp, uint32_t cu_idx,
170                  const FileSpec &file_spec, const ArchSpec &arch,
171                  const ConstString *object_name, off_t object_offset,
172                  const llvm::sys::TimePoint<> object_mod_time)
173       : Module(file_spec, arch, object_name, object_offset, object_mod_time),
174         m_exe_module_wp(exe_module_sp), m_cu_idx(cu_idx) {}
175 
176   ~DebugMapModule() override = default;
177 
178   SymbolVendor *
179   GetSymbolVendor(bool can_create = true,
180                   lldb_private::Stream *feedback_strm = NULL) override {
181     // Scope for locker
182     if (m_symfile_ap.get() || !can_create)
183       return m_symfile_ap.get();
184 
185     ModuleSP exe_module_sp(m_exe_module_wp.lock());
186     if (exe_module_sp) {
187       // Now get the object file outside of a locking scope
188       ObjectFile *oso_objfile = GetObjectFile();
189       if (oso_objfile) {
190         std::lock_guard<std::recursive_mutex> guard(m_mutex);
191         SymbolVendor *symbol_vendor =
192             Module::GetSymbolVendor(can_create, feedback_strm);
193         if (symbol_vendor) {
194           // Set a pointer to this class to set our OSO DWARF file know that
195           // the DWARF is being used along with a debug map and that it will
196           // have the remapped sections that we do below.
197           SymbolFileDWARF *oso_symfile =
198               SymbolFileDWARFDebugMap::GetSymbolFileAsSymbolFileDWARF(
199                   symbol_vendor->GetSymbolFile());
200 
201           if (!oso_symfile)
202             return NULL;
203 
204           ObjectFile *exe_objfile = exe_module_sp->GetObjectFile();
205           SymbolVendor *exe_sym_vendor = exe_module_sp->GetSymbolVendor();
206 
207           if (exe_objfile && exe_sym_vendor) {
208             oso_symfile->SetDebugMapModule(exe_module_sp);
209             // Set the ID of the symbol file DWARF to the index of the OSO
210             // shifted left by 32 bits to provide a unique prefix for any
211             // UserID's that get created in the symbol file.
212             oso_symfile->SetID(((uint64_t)m_cu_idx + 1ull) << 32ull);
213           }
214           return symbol_vendor;
215         }
216       }
217     }
218     return NULL;
219   }
220 
221 protected:
222   ModuleWP m_exe_module_wp;
223   const uint32_t m_cu_idx;
224 };
225 
226 void SymbolFileDWARFDebugMap::Initialize() {
227   PluginManager::RegisterPlugin(GetPluginNameStatic(),
228                                 GetPluginDescriptionStatic(), CreateInstance);
229 }
230 
231 void SymbolFileDWARFDebugMap::Terminate() {
232   PluginManager::UnregisterPlugin(CreateInstance);
233 }
234 
235 lldb_private::ConstString SymbolFileDWARFDebugMap::GetPluginNameStatic() {
236   static ConstString g_name("dwarf-debugmap");
237   return g_name;
238 }
239 
240 const char *SymbolFileDWARFDebugMap::GetPluginDescriptionStatic() {
241   return "DWARF and DWARF3 debug symbol file reader (debug map).";
242 }
243 
244 SymbolFile *SymbolFileDWARFDebugMap::CreateInstance(ObjectFile *obj_file) {
245   return new SymbolFileDWARFDebugMap(obj_file);
246 }
247 
248 SymbolFileDWARFDebugMap::SymbolFileDWARFDebugMap(ObjectFile *ofile)
249     : SymbolFile(ofile), m_flags(), m_compile_unit_infos(), m_func_indexes(),
250       m_glob_indexes(),
251       m_supports_DW_AT_APPLE_objc_complete_type(eLazyBoolCalculate) {}
252 
253 SymbolFileDWARFDebugMap::~SymbolFileDWARFDebugMap() {}
254 
255 void SymbolFileDWARFDebugMap::InitializeObject() {}
256 
257 void SymbolFileDWARFDebugMap::InitOSO() {
258   if (m_flags.test(kHaveInitializedOSOs))
259     return;
260 
261   m_flags.set(kHaveInitializedOSOs);
262 
263   // If the object file has been stripped, there is no sense in looking further
264   // as all of the debug symbols for the debug map will not be available
265   if (m_obj_file->IsStripped())
266     return;
267 
268   // Also make sure the file type is some sort of executable. Core files, debug
269   // info files (dSYM), object files (.o files), and stub libraries all can
270   switch (m_obj_file->GetType()) {
271   case ObjectFile::eTypeInvalid:
272   case ObjectFile::eTypeCoreFile:
273   case ObjectFile::eTypeDebugInfo:
274   case ObjectFile::eTypeObjectFile:
275   case ObjectFile::eTypeStubLibrary:
276   case ObjectFile::eTypeUnknown:
277   case ObjectFile::eTypeJIT:
278     return;
279 
280   case ObjectFile::eTypeExecutable:
281   case ObjectFile::eTypeDynamicLinker:
282   case ObjectFile::eTypeSharedLibrary:
283     break;
284   }
285 
286   // In order to get the abilities of this plug-in, we look at the list of
287   // N_OSO entries (object files) from the symbol table and make sure that
288   // these files exist and also contain valid DWARF. If we get any of that then
289   // we return the abilities of the first N_OSO's DWARF.
290 
291   Symtab *symtab = m_obj_file->GetSymtab();
292   if (symtab) {
293     Log *log(LogChannelDWARF::GetLogIfAll(DWARF_LOG_DEBUG_MAP));
294 
295     std::vector<uint32_t> oso_indexes;
296     // When a mach-o symbol is encoded, the n_type field is encoded in bits
297     // 23:16, and the n_desc field is encoded in bits 15:0.
298     //
299     // To find all N_OSO entries that are part of the DWARF + debug map we find
300     // only object file symbols with the flags value as follows: bits 23:16 ==
301     // 0x66 (N_OSO) bits 15: 0 == 0x0001 (specifies this is a debug map object
302     // file)
303     const uint32_t k_oso_symbol_flags_value = 0x660001u;
304 
305     const uint32_t oso_index_count =
306         symtab->AppendSymbolIndexesWithTypeAndFlagsValue(
307             eSymbolTypeObjectFile, k_oso_symbol_flags_value, oso_indexes);
308 
309     if (oso_index_count > 0) {
310       symtab->AppendSymbolIndexesWithType(eSymbolTypeCode, Symtab::eDebugYes,
311                                           Symtab::eVisibilityAny,
312                                           m_func_indexes);
313       symtab->AppendSymbolIndexesWithType(eSymbolTypeData, Symtab::eDebugYes,
314                                           Symtab::eVisibilityAny,
315                                           m_glob_indexes);
316 
317       symtab->SortSymbolIndexesByValue(m_func_indexes, true);
318       symtab->SortSymbolIndexesByValue(m_glob_indexes, true);
319 
320       for (uint32_t sym_idx : m_func_indexes) {
321         const Symbol *symbol = symtab->SymbolAtIndex(sym_idx);
322         lldb::addr_t file_addr = symbol->GetAddressRef().GetFileAddress();
323         lldb::addr_t byte_size = symbol->GetByteSize();
324         DebugMap::Entry debug_map_entry(
325             file_addr, byte_size, OSOEntry(sym_idx, LLDB_INVALID_ADDRESS));
326         m_debug_map.Append(debug_map_entry);
327       }
328       for (uint32_t sym_idx : m_glob_indexes) {
329         const Symbol *symbol = symtab->SymbolAtIndex(sym_idx);
330         lldb::addr_t file_addr = symbol->GetAddressRef().GetFileAddress();
331         lldb::addr_t byte_size = symbol->GetByteSize();
332         DebugMap::Entry debug_map_entry(
333             file_addr, byte_size, OSOEntry(sym_idx, LLDB_INVALID_ADDRESS));
334         m_debug_map.Append(debug_map_entry);
335       }
336       m_debug_map.Sort();
337 
338       m_compile_unit_infos.resize(oso_index_count);
339 
340       for (uint32_t i = 0; i < oso_index_count; ++i) {
341         const uint32_t so_idx = oso_indexes[i] - 1;
342         const uint32_t oso_idx = oso_indexes[i];
343         const Symbol *so_symbol = symtab->SymbolAtIndex(so_idx);
344         const Symbol *oso_symbol = symtab->SymbolAtIndex(oso_idx);
345         if (so_symbol && oso_symbol &&
346             so_symbol->GetType() == eSymbolTypeSourceFile &&
347             oso_symbol->GetType() == eSymbolTypeObjectFile) {
348           m_compile_unit_infos[i].so_file.SetFile(
349               so_symbol->GetName().AsCString(), FileSpec::Style::native);
350           m_compile_unit_infos[i].oso_path = oso_symbol->GetName();
351           m_compile_unit_infos[i].oso_mod_time =
352               llvm::sys::toTimePoint(oso_symbol->GetIntegerValue(0));
353           uint32_t sibling_idx = so_symbol->GetSiblingIndex();
354           // The sibling index can't be less that or equal to the current index
355           // "i"
356           if (sibling_idx == UINT32_MAX) {
357             m_obj_file->GetModule()->ReportError(
358                 "N_SO in symbol with UID %u has invalid sibling in debug map, "
359                 "please file a bug and attach the binary listed in this error",
360                 so_symbol->GetID());
361           } else {
362             const Symbol *last_symbol = symtab->SymbolAtIndex(sibling_idx - 1);
363             m_compile_unit_infos[i].first_symbol_index = so_idx;
364             m_compile_unit_infos[i].last_symbol_index = sibling_idx - 1;
365             m_compile_unit_infos[i].first_symbol_id = so_symbol->GetID();
366             m_compile_unit_infos[i].last_symbol_id = last_symbol->GetID();
367 
368             if (log)
369               log->Printf("Initialized OSO 0x%8.8x: file=%s", i,
370                           oso_symbol->GetName().GetCString());
371           }
372         } else {
373           if (oso_symbol == NULL)
374             m_obj_file->GetModule()->ReportError(
375                 "N_OSO symbol[%u] can't be found, please file a bug and attach "
376                 "the binary listed in this error",
377                 oso_idx);
378           else if (so_symbol == NULL)
379             m_obj_file->GetModule()->ReportError(
380                 "N_SO not found for N_OSO symbol[%u], please file a bug and "
381                 "attach the binary listed in this error",
382                 oso_idx);
383           else if (so_symbol->GetType() != eSymbolTypeSourceFile)
384             m_obj_file->GetModule()->ReportError(
385                 "N_SO has incorrect symbol type (%u) for N_OSO symbol[%u], "
386                 "please file a bug and attach the binary listed in this error",
387                 so_symbol->GetType(), oso_idx);
388           else if (oso_symbol->GetType() != eSymbolTypeSourceFile)
389             m_obj_file->GetModule()->ReportError(
390                 "N_OSO has incorrect symbol type (%u) for N_OSO symbol[%u], "
391                 "please file a bug and attach the binary listed in this error",
392                 oso_symbol->GetType(), oso_idx);
393         }
394       }
395     }
396   }
397 }
398 
399 Module *SymbolFileDWARFDebugMap::GetModuleByOSOIndex(uint32_t oso_idx) {
400   const uint32_t cu_count = GetNumCompileUnits();
401   if (oso_idx < cu_count)
402     return GetModuleByCompUnitInfo(&m_compile_unit_infos[oso_idx]);
403   return NULL;
404 }
405 
406 Module *SymbolFileDWARFDebugMap::GetModuleByCompUnitInfo(
407     CompileUnitInfo *comp_unit_info) {
408   if (!comp_unit_info->oso_sp) {
409     auto pos = m_oso_map.find(
410         {comp_unit_info->oso_path, comp_unit_info->oso_mod_time});
411     if (pos != m_oso_map.end()) {
412       comp_unit_info->oso_sp = pos->second;
413     } else {
414       ObjectFile *obj_file = GetObjectFile();
415       comp_unit_info->oso_sp.reset(new OSOInfo());
416       m_oso_map[{comp_unit_info->oso_path, comp_unit_info->oso_mod_time}] =
417           comp_unit_info->oso_sp;
418       const char *oso_path = comp_unit_info->oso_path.GetCString();
419       FileSpec oso_file(oso_path);
420       ConstString oso_object;
421       if (FileSystem::Instance().Exists(oso_file)) {
422         // The modification time returned by the FS can have a higher precision
423         // than the one from the CU.
424         auto oso_mod_time = std::chrono::time_point_cast<std::chrono::seconds>(
425             FileSystem::Instance().GetModificationTime(oso_file));
426         if (oso_mod_time != comp_unit_info->oso_mod_time) {
427           obj_file->GetModule()->ReportError(
428               "debug map object file '%s' has changed (actual time is "
429               "%s, debug map time is %s"
430               ") since this executable was linked, file will be ignored",
431               oso_file.GetPath().c_str(), llvm::to_string(oso_mod_time).c_str(),
432               llvm::to_string(comp_unit_info->oso_mod_time).c_str());
433           return NULL;
434         }
435 
436       } else {
437         const bool must_exist = true;
438 
439         if (!ObjectFile::SplitArchivePathWithObject(oso_path, oso_file,
440                                                     oso_object, must_exist)) {
441           return NULL;
442         }
443       }
444       // Always create a new module for .o files. Why? Because we use the debug
445       // map, to add new sections to each .o file and even though a .o file
446       // might not have changed, the sections that get added to the .o file can
447       // change.
448       ArchSpec oso_arch;
449       // Only adopt the architecture from the module (not the vendor or OS)
450       // since .o files for "i386-apple-ios" will historically show up as "i386
451       // -apple-macosx" due to the lack of a LC_VERSION_MIN_MACOSX or
452       // LC_VERSION_MIN_IPHONEOS load command...
453       oso_arch.SetTriple(m_obj_file->GetModule()
454                              ->GetArchitecture()
455                              .GetTriple()
456                              .getArchName()
457                              .str()
458                              .c_str());
459       comp_unit_info->oso_sp->module_sp.reset(new DebugMapModule(
460           obj_file->GetModule(), GetCompUnitInfoIndex(comp_unit_info), oso_file,
461           oso_arch, oso_object ? &oso_object : NULL, 0,
462           oso_object ? comp_unit_info->oso_mod_time
463                      : llvm::sys::TimePoint<>()));
464     }
465   }
466   if (comp_unit_info->oso_sp)
467     return comp_unit_info->oso_sp->module_sp.get();
468   return NULL;
469 }
470 
471 bool SymbolFileDWARFDebugMap::GetFileSpecForSO(uint32_t oso_idx,
472                                                FileSpec &file_spec) {
473   if (oso_idx < m_compile_unit_infos.size()) {
474     if (m_compile_unit_infos[oso_idx].so_file) {
475       file_spec = m_compile_unit_infos[oso_idx].so_file;
476       return true;
477     }
478   }
479   return false;
480 }
481 
482 ObjectFile *SymbolFileDWARFDebugMap::GetObjectFileByOSOIndex(uint32_t oso_idx) {
483   Module *oso_module = GetModuleByOSOIndex(oso_idx);
484   if (oso_module)
485     return oso_module->GetObjectFile();
486   return NULL;
487 }
488 
489 SymbolFileDWARF *
490 SymbolFileDWARFDebugMap::GetSymbolFile(const SymbolContext &sc) {
491   CompileUnitInfo *comp_unit_info = GetCompUnitInfo(sc);
492   if (comp_unit_info)
493     return GetSymbolFileByCompUnitInfo(comp_unit_info);
494   return NULL;
495 }
496 
497 ObjectFile *SymbolFileDWARFDebugMap::GetObjectFileByCompUnitInfo(
498     CompileUnitInfo *comp_unit_info) {
499   Module *oso_module = GetModuleByCompUnitInfo(comp_unit_info);
500   if (oso_module)
501     return oso_module->GetObjectFile();
502   return NULL;
503 }
504 
505 uint32_t SymbolFileDWARFDebugMap::GetCompUnitInfoIndex(
506     const CompileUnitInfo *comp_unit_info) {
507   if (!m_compile_unit_infos.empty()) {
508     const CompileUnitInfo *first_comp_unit_info = &m_compile_unit_infos.front();
509     const CompileUnitInfo *last_comp_unit_info = &m_compile_unit_infos.back();
510     if (first_comp_unit_info <= comp_unit_info &&
511         comp_unit_info <= last_comp_unit_info)
512       return comp_unit_info - first_comp_unit_info;
513   }
514   return UINT32_MAX;
515 }
516 
517 SymbolFileDWARF *
518 SymbolFileDWARFDebugMap::GetSymbolFileByOSOIndex(uint32_t oso_idx) {
519   unsigned size = m_compile_unit_infos.size();
520   if (oso_idx < size)
521     return GetSymbolFileByCompUnitInfo(&m_compile_unit_infos[oso_idx]);
522   return NULL;
523 }
524 
525 SymbolFileDWARF *
526 SymbolFileDWARFDebugMap::GetSymbolFileAsSymbolFileDWARF(SymbolFile *sym_file) {
527   if (sym_file &&
528       sym_file->GetPluginName() == SymbolFileDWARF::GetPluginNameStatic())
529     return (SymbolFileDWARF *)sym_file;
530   return NULL;
531 }
532 
533 SymbolFileDWARF *SymbolFileDWARFDebugMap::GetSymbolFileByCompUnitInfo(
534     CompileUnitInfo *comp_unit_info) {
535   Module *oso_module = GetModuleByCompUnitInfo(comp_unit_info);
536   if (oso_module) {
537     SymbolVendor *sym_vendor = oso_module->GetSymbolVendor();
538     if (sym_vendor)
539       return GetSymbolFileAsSymbolFileDWARF(sym_vendor->GetSymbolFile());
540   }
541   return NULL;
542 }
543 
544 uint32_t SymbolFileDWARFDebugMap::CalculateAbilities() {
545   // In order to get the abilities of this plug-in, we look at the list of
546   // N_OSO entries (object files) from the symbol table and make sure that
547   // these files exist and also contain valid DWARF. If we get any of that then
548   // we return the abilities of the first N_OSO's DWARF.
549 
550   const uint32_t oso_index_count = GetNumCompileUnits();
551   if (oso_index_count > 0) {
552     InitOSO();
553     if (!m_compile_unit_infos.empty()) {
554       return SymbolFile::CompileUnits | SymbolFile::Functions |
555              SymbolFile::Blocks | SymbolFile::GlobalVariables |
556              SymbolFile::LocalVariables | SymbolFile::VariableTypes |
557              SymbolFile::LineTables;
558     }
559   }
560   return 0;
561 }
562 
563 uint32_t SymbolFileDWARFDebugMap::GetNumCompileUnits() {
564   InitOSO();
565   return m_compile_unit_infos.size();
566 }
567 
568 CompUnitSP SymbolFileDWARFDebugMap::ParseCompileUnitAtIndex(uint32_t cu_idx) {
569   CompUnitSP comp_unit_sp;
570   const uint32_t cu_count = GetNumCompileUnits();
571 
572   if (cu_idx < cu_count) {
573     Module *oso_module = GetModuleByCompUnitInfo(&m_compile_unit_infos[cu_idx]);
574     if (oso_module) {
575       FileSpec so_file_spec;
576       if (GetFileSpecForSO(cu_idx, so_file_spec)) {
577         // User zero as the ID to match the compile unit at offset zero in each
578         // .o file since each .o file can only have one compile unit for now.
579         lldb::user_id_t cu_id = 0;
580         m_compile_unit_infos[cu_idx].compile_unit_sp.reset(
581             new CompileUnit(m_obj_file->GetModule(), NULL, so_file_spec, cu_id,
582                             eLanguageTypeUnknown, eLazyBoolCalculate));
583 
584         if (m_compile_unit_infos[cu_idx].compile_unit_sp) {
585           // Let our symbol vendor know about this compile unit
586           m_obj_file->GetModule()->GetSymbolVendor()->SetCompileUnitAtIndex(
587               cu_idx, m_compile_unit_infos[cu_idx].compile_unit_sp);
588         }
589       }
590     }
591     comp_unit_sp = m_compile_unit_infos[cu_idx].compile_unit_sp;
592   }
593 
594   return comp_unit_sp;
595 }
596 
597 SymbolFileDWARFDebugMap::CompileUnitInfo *
598 SymbolFileDWARFDebugMap::GetCompUnitInfo(const SymbolContext &sc) {
599   const uint32_t cu_count = GetNumCompileUnits();
600   for (uint32_t i = 0; i < cu_count; ++i) {
601     if (sc.comp_unit == m_compile_unit_infos[i].compile_unit_sp.get())
602       return &m_compile_unit_infos[i];
603   }
604   return NULL;
605 }
606 
607 size_t SymbolFileDWARFDebugMap::GetCompUnitInfosForModule(
608     const lldb_private::Module *module,
609     std::vector<CompileUnitInfo *> &cu_infos) {
610   const uint32_t cu_count = GetNumCompileUnits();
611   for (uint32_t i = 0; i < cu_count; ++i) {
612     if (module == GetModuleByCompUnitInfo(&m_compile_unit_infos[i]))
613       cu_infos.push_back(&m_compile_unit_infos[i]);
614   }
615   return cu_infos.size();
616 }
617 
618 lldb::LanguageType
619 SymbolFileDWARFDebugMap::ParseCompileUnitLanguage(const SymbolContext &sc) {
620   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
621   if (oso_dwarf)
622     return oso_dwarf->ParseCompileUnitLanguage(sc);
623   return eLanguageTypeUnknown;
624 }
625 
626 size_t
627 SymbolFileDWARFDebugMap::ParseCompileUnitFunctions(const SymbolContext &sc) {
628   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
629   if (oso_dwarf)
630     return oso_dwarf->ParseCompileUnitFunctions(sc);
631   return 0;
632 }
633 
634 bool SymbolFileDWARFDebugMap::ParseCompileUnitLineTable(
635     const SymbolContext &sc) {
636   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
637   if (oso_dwarf)
638     return oso_dwarf->ParseCompileUnitLineTable(sc);
639   return false;
640 }
641 
642 bool SymbolFileDWARFDebugMap::ParseCompileUnitDebugMacros(
643     const SymbolContext &sc) {
644   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
645   if (oso_dwarf)
646     return oso_dwarf->ParseCompileUnitDebugMacros(sc);
647   return false;
648 }
649 
650 bool SymbolFileDWARFDebugMap::ParseCompileUnitSupportFiles(
651     const SymbolContext &sc, FileSpecList &support_files) {
652   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
653   if (oso_dwarf)
654     return oso_dwarf->ParseCompileUnitSupportFiles(sc, support_files);
655   return false;
656 }
657 
658 bool SymbolFileDWARFDebugMap::ParseCompileUnitIsOptimized(
659     const lldb_private::SymbolContext &sc) {
660   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
661   if (oso_dwarf)
662     return oso_dwarf->ParseCompileUnitIsOptimized(sc);
663   return false;
664 }
665 
666 bool SymbolFileDWARFDebugMap::ParseImportedModules(
667     const SymbolContext &sc, std::vector<ConstString> &imported_modules) {
668   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
669   if (oso_dwarf)
670     return oso_dwarf->ParseImportedModules(sc, imported_modules);
671   return false;
672 }
673 
674 size_t SymbolFileDWARFDebugMap::ParseFunctionBlocks(const SymbolContext &sc) {
675   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
676   if (oso_dwarf)
677     return oso_dwarf->ParseFunctionBlocks(sc);
678   return 0;
679 }
680 
681 size_t SymbolFileDWARFDebugMap::ParseTypes(const SymbolContext &sc) {
682   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
683   if (oso_dwarf)
684     return oso_dwarf->ParseTypes(sc);
685   return 0;
686 }
687 
688 size_t
689 SymbolFileDWARFDebugMap::ParseVariablesForContext(const SymbolContext &sc) {
690   SymbolFileDWARF *oso_dwarf = GetSymbolFile(sc);
691   if (oso_dwarf)
692     return oso_dwarf->ParseVariablesForContext(sc);
693   return 0;
694 }
695 
696 Type *SymbolFileDWARFDebugMap::ResolveTypeUID(lldb::user_id_t type_uid) {
697   const uint64_t oso_idx = GetOSOIndexFromUserID(type_uid);
698   SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
699   if (oso_dwarf)
700     return oso_dwarf->ResolveTypeUID(type_uid);
701   return NULL;
702 }
703 
704 llvm::Optional<SymbolFile::ArrayInfo>
705 SymbolFileDWARFDebugMap::GetDynamicArrayInfoForUID(
706     lldb::user_id_t type_uid, const lldb_private::ExecutionContext *exe_ctx) {
707   const uint64_t oso_idx = GetOSOIndexFromUserID(type_uid);
708   SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
709   if (oso_dwarf)
710     return oso_dwarf->GetDynamicArrayInfoForUID(type_uid, exe_ctx);
711   return llvm::None;
712 }
713 
714 bool SymbolFileDWARFDebugMap::CompleteType(CompilerType &compiler_type) {
715   bool success = false;
716   if (compiler_type) {
717     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
718       if (oso_dwarf->HasForwardDeclForClangType(compiler_type)) {
719         oso_dwarf->CompleteType(compiler_type);
720         success = true;
721         return true;
722       }
723       return false;
724     });
725   }
726   return success;
727 }
728 
729 uint32_t
730 SymbolFileDWARFDebugMap::ResolveSymbolContext(const Address &exe_so_addr,
731                                               SymbolContextItem resolve_scope,
732                                               SymbolContext &sc) {
733   uint32_t resolved_flags = 0;
734   Symtab *symtab = m_obj_file->GetSymtab();
735   if (symtab) {
736     const addr_t exe_file_addr = exe_so_addr.GetFileAddress();
737 
738     const DebugMap::Entry *debug_map_entry =
739         m_debug_map.FindEntryThatContains(exe_file_addr);
740     if (debug_map_entry) {
741 
742       sc.symbol =
743           symtab->SymbolAtIndex(debug_map_entry->data.GetExeSymbolIndex());
744 
745       if (sc.symbol != NULL) {
746         resolved_flags |= eSymbolContextSymbol;
747 
748         uint32_t oso_idx = 0;
749         CompileUnitInfo *comp_unit_info =
750             GetCompileUnitInfoForSymbolWithID(sc.symbol->GetID(), &oso_idx);
751         if (comp_unit_info) {
752           comp_unit_info->GetFileRangeMap(this);
753           Module *oso_module = GetModuleByCompUnitInfo(comp_unit_info);
754           if (oso_module) {
755             lldb::addr_t oso_file_addr =
756                 exe_file_addr - debug_map_entry->GetRangeBase() +
757                 debug_map_entry->data.GetOSOFileAddress();
758             Address oso_so_addr;
759             if (oso_module->ResolveFileAddress(oso_file_addr, oso_so_addr)) {
760               resolved_flags |=
761                   oso_module->GetSymbolVendor()->ResolveSymbolContext(
762                       oso_so_addr, resolve_scope, sc);
763             }
764           }
765         }
766       }
767     }
768   }
769   return resolved_flags;
770 }
771 
772 uint32_t SymbolFileDWARFDebugMap::ResolveSymbolContext(
773     const FileSpec &file_spec, uint32_t line, bool check_inlines,
774     SymbolContextItem resolve_scope, SymbolContextList &sc_list) {
775   const uint32_t initial = sc_list.GetSize();
776   const uint32_t cu_count = GetNumCompileUnits();
777 
778   for (uint32_t i = 0; i < cu_count; ++i) {
779     // If we are checking for inlines, then we need to look through all compile
780     // units no matter if "file_spec" matches.
781     bool resolve = check_inlines;
782 
783     if (!resolve) {
784       FileSpec so_file_spec;
785       if (GetFileSpecForSO(i, so_file_spec)) {
786         // Match the full path if the incoming file_spec has a directory (not
787         // just a basename)
788         const bool full_match = (bool)file_spec.GetDirectory();
789         resolve = FileSpec::Equal(file_spec, so_file_spec, full_match);
790       }
791     }
792     if (resolve) {
793       SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(i);
794       if (oso_dwarf)
795         oso_dwarf->ResolveSymbolContext(file_spec, line, check_inlines,
796                                         resolve_scope, sc_list);
797     }
798   }
799   return sc_list.GetSize() - initial;
800 }
801 
802 uint32_t SymbolFileDWARFDebugMap::PrivateFindGlobalVariables(
803     const ConstString &name, const CompilerDeclContext *parent_decl_ctx,
804     const std::vector<uint32_t>
805         &indexes, // Indexes into the symbol table that match "name"
806     uint32_t max_matches, VariableList &variables) {
807   const uint32_t original_size = variables.GetSize();
808   const size_t match_count = indexes.size();
809   for (size_t i = 0; i < match_count; ++i) {
810     uint32_t oso_idx;
811     CompileUnitInfo *comp_unit_info =
812         GetCompileUnitInfoForSymbolWithIndex(indexes[i], &oso_idx);
813     if (comp_unit_info) {
814       SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
815       if (oso_dwarf) {
816         if (oso_dwarf->FindGlobalVariables(name, parent_decl_ctx, max_matches,
817                                            variables))
818           if (variables.GetSize() > max_matches)
819             break;
820       }
821     }
822   }
823   return variables.GetSize() - original_size;
824 }
825 
826 uint32_t SymbolFileDWARFDebugMap::FindGlobalVariables(
827     const ConstString &name, const CompilerDeclContext *parent_decl_ctx,
828     uint32_t max_matches, VariableList &variables) {
829 
830   // Remember how many variables are in the list before we search.
831   const uint32_t original_size = variables.GetSize();
832 
833   uint32_t total_matches = 0;
834 
835   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
836     const uint32_t oso_matches = oso_dwarf->FindGlobalVariables(
837         name, parent_decl_ctx, max_matches, variables);
838     if (oso_matches > 0) {
839       total_matches += oso_matches;
840 
841       // Are we getting all matches?
842       if (max_matches == UINT32_MAX)
843         return false; // Yep, continue getting everything
844 
845       // If we have found enough matches, lets get out
846       if (max_matches >= total_matches)
847         return true;
848 
849       // Update the max matches for any subsequent calls to find globals in any
850       // other object files with DWARF
851       max_matches -= oso_matches;
852     }
853 
854     return false;
855   });
856 
857   // Return the number of variable that were appended to the list
858   return variables.GetSize() - original_size;
859 }
860 
861 uint32_t
862 SymbolFileDWARFDebugMap::FindGlobalVariables(const RegularExpression &regex,
863                                              uint32_t max_matches,
864                                              VariableList &variables) {
865   // Remember how many variables are in the list before we search.
866   const uint32_t original_size = variables.GetSize();
867 
868   uint32_t total_matches = 0;
869   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
870     const uint32_t oso_matches =
871         oso_dwarf->FindGlobalVariables(regex, max_matches, variables);
872     if (oso_matches > 0) {
873       total_matches += oso_matches;
874 
875       // Are we getting all matches?
876       if (max_matches == UINT32_MAX)
877         return false; // Yep, continue getting everything
878 
879       // If we have found enough matches, lets get out
880       if (max_matches >= total_matches)
881         return true;
882 
883       // Update the max matches for any subsequent calls to find globals in any
884       // other object files with DWARF
885       max_matches -= oso_matches;
886     }
887 
888     return false;
889   });
890 
891   // Return the number of variable that were appended to the list
892   return variables.GetSize() - original_size;
893 }
894 
895 int SymbolFileDWARFDebugMap::SymbolContainsSymbolWithIndex(
896     uint32_t *symbol_idx_ptr, const CompileUnitInfo *comp_unit_info) {
897   const uint32_t symbol_idx = *symbol_idx_ptr;
898 
899   if (symbol_idx < comp_unit_info->first_symbol_index)
900     return -1;
901 
902   if (symbol_idx <= comp_unit_info->last_symbol_index)
903     return 0;
904 
905   return 1;
906 }
907 
908 int SymbolFileDWARFDebugMap::SymbolContainsSymbolWithID(
909     user_id_t *symbol_idx_ptr, const CompileUnitInfo *comp_unit_info) {
910   const user_id_t symbol_id = *symbol_idx_ptr;
911 
912   if (symbol_id < comp_unit_info->first_symbol_id)
913     return -1;
914 
915   if (symbol_id <= comp_unit_info->last_symbol_id)
916     return 0;
917 
918   return 1;
919 }
920 
921 SymbolFileDWARFDebugMap::CompileUnitInfo *
922 SymbolFileDWARFDebugMap::GetCompileUnitInfoForSymbolWithIndex(
923     uint32_t symbol_idx, uint32_t *oso_idx_ptr) {
924   const uint32_t oso_index_count = m_compile_unit_infos.size();
925   CompileUnitInfo *comp_unit_info = NULL;
926   if (oso_index_count) {
927     comp_unit_info = (CompileUnitInfo *)bsearch(
928         &symbol_idx, &m_compile_unit_infos[0], m_compile_unit_infos.size(),
929         sizeof(CompileUnitInfo),
930         (ComparisonFunction)SymbolContainsSymbolWithIndex);
931   }
932 
933   if (oso_idx_ptr) {
934     if (comp_unit_info != NULL)
935       *oso_idx_ptr = comp_unit_info - &m_compile_unit_infos[0];
936     else
937       *oso_idx_ptr = UINT32_MAX;
938   }
939   return comp_unit_info;
940 }
941 
942 SymbolFileDWARFDebugMap::CompileUnitInfo *
943 SymbolFileDWARFDebugMap::GetCompileUnitInfoForSymbolWithID(
944     user_id_t symbol_id, uint32_t *oso_idx_ptr) {
945   const uint32_t oso_index_count = m_compile_unit_infos.size();
946   CompileUnitInfo *comp_unit_info = NULL;
947   if (oso_index_count) {
948     comp_unit_info = (CompileUnitInfo *)::bsearch(
949         &symbol_id, &m_compile_unit_infos[0], m_compile_unit_infos.size(),
950         sizeof(CompileUnitInfo),
951         (ComparisonFunction)SymbolContainsSymbolWithID);
952   }
953 
954   if (oso_idx_ptr) {
955     if (comp_unit_info != NULL)
956       *oso_idx_ptr = comp_unit_info - &m_compile_unit_infos[0];
957     else
958       *oso_idx_ptr = UINT32_MAX;
959   }
960   return comp_unit_info;
961 }
962 
963 static void RemoveFunctionsWithModuleNotEqualTo(const ModuleSP &module_sp,
964                                                 SymbolContextList &sc_list,
965                                                 uint32_t start_idx) {
966   // We found functions in .o files. Not all functions in the .o files will
967   // have made it into the final output file. The ones that did make it into
968   // the final output file will have a section whose module matches the module
969   // from the ObjectFile for this SymbolFile. When the modules don't match,
970   // then we have something that was in a .o file, but doesn't map to anything
971   // in the final executable.
972   uint32_t i = start_idx;
973   while (i < sc_list.GetSize()) {
974     SymbolContext sc;
975     sc_list.GetContextAtIndex(i, sc);
976     if (sc.function) {
977       const SectionSP section_sp(
978           sc.function->GetAddressRange().GetBaseAddress().GetSection());
979       if (section_sp->GetModule() != module_sp) {
980         sc_list.RemoveContextAtIndex(i);
981         continue;
982       }
983     }
984     ++i;
985   }
986 }
987 
988 uint32_t SymbolFileDWARFDebugMap::FindFunctions(
989     const ConstString &name, const CompilerDeclContext *parent_decl_ctx,
990     FunctionNameType name_type_mask, bool include_inlines, bool append,
991     SymbolContextList &sc_list) {
992   static Timer::Category func_cat(LLVM_PRETTY_FUNCTION);
993   Timer scoped_timer(func_cat,
994                      "SymbolFileDWARFDebugMap::FindFunctions (name = %s)",
995                      name.GetCString());
996 
997   uint32_t initial_size = 0;
998   if (append)
999     initial_size = sc_list.GetSize();
1000   else
1001     sc_list.Clear();
1002 
1003   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1004     uint32_t sc_idx = sc_list.GetSize();
1005     if (oso_dwarf->FindFunctions(name, parent_decl_ctx, name_type_mask,
1006                                  include_inlines, true, sc_list)) {
1007       RemoveFunctionsWithModuleNotEqualTo(m_obj_file->GetModule(), sc_list,
1008                                           sc_idx);
1009     }
1010     return false;
1011   });
1012 
1013   return sc_list.GetSize() - initial_size;
1014 }
1015 
1016 uint32_t SymbolFileDWARFDebugMap::FindFunctions(const RegularExpression &regex,
1017                                                 bool include_inlines,
1018                                                 bool append,
1019                                                 SymbolContextList &sc_list) {
1020   static Timer::Category func_cat(LLVM_PRETTY_FUNCTION);
1021   Timer scoped_timer(func_cat,
1022                      "SymbolFileDWARFDebugMap::FindFunctions (regex = '%s')",
1023                      regex.GetText().str().c_str());
1024 
1025   uint32_t initial_size = 0;
1026   if (append)
1027     initial_size = sc_list.GetSize();
1028   else
1029     sc_list.Clear();
1030 
1031   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1032     uint32_t sc_idx = sc_list.GetSize();
1033 
1034     if (oso_dwarf->FindFunctions(regex, include_inlines, true, sc_list)) {
1035       RemoveFunctionsWithModuleNotEqualTo(m_obj_file->GetModule(), sc_list,
1036                                           sc_idx);
1037     }
1038     return false;
1039   });
1040 
1041   return sc_list.GetSize() - initial_size;
1042 }
1043 
1044 size_t SymbolFileDWARFDebugMap::GetTypes(SymbolContextScope *sc_scope,
1045                                          lldb::TypeClass type_mask,
1046                                          TypeList &type_list) {
1047   static Timer::Category func_cat(LLVM_PRETTY_FUNCTION);
1048   Timer scoped_timer(func_cat,
1049                      "SymbolFileDWARFDebugMap::GetTypes (type_mask = 0x%8.8x)",
1050                      type_mask);
1051 
1052   uint32_t initial_size = type_list.GetSize();
1053   SymbolFileDWARF *oso_dwarf = NULL;
1054   if (sc_scope) {
1055     SymbolContext sc;
1056     sc_scope->CalculateSymbolContext(&sc);
1057 
1058     CompileUnitInfo *cu_info = GetCompUnitInfo(sc);
1059     if (cu_info) {
1060       oso_dwarf = GetSymbolFileByCompUnitInfo(cu_info);
1061       if (oso_dwarf)
1062         oso_dwarf->GetTypes(sc_scope, type_mask, type_list);
1063     }
1064   } else {
1065     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1066       oso_dwarf->GetTypes(sc_scope, type_mask, type_list);
1067       return false;
1068     });
1069   }
1070   return type_list.GetSize() - initial_size;
1071 }
1072 
1073 std::vector<lldb_private::CallEdge>
1074 SymbolFileDWARFDebugMap::ParseCallEdgesInFunction(UserID func_id) {
1075   uint32_t oso_idx = GetOSOIndexFromUserID(func_id.GetID());
1076   SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
1077   if (oso_dwarf)
1078     return oso_dwarf->ParseCallEdgesInFunction(func_id);
1079   return {};
1080 }
1081 
1082 TypeSP SymbolFileDWARFDebugMap::FindDefinitionTypeForDWARFDeclContext(
1083     const DWARFDeclContext &die_decl_ctx) {
1084   TypeSP type_sp;
1085   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1086     type_sp = oso_dwarf->FindDefinitionTypeForDWARFDeclContext(die_decl_ctx);
1087     return ((bool)type_sp);
1088   });
1089   return type_sp;
1090 }
1091 
1092 bool SymbolFileDWARFDebugMap::Supports_DW_AT_APPLE_objc_complete_type(
1093     SymbolFileDWARF *skip_dwarf_oso) {
1094   if (m_supports_DW_AT_APPLE_objc_complete_type == eLazyBoolCalculate) {
1095     m_supports_DW_AT_APPLE_objc_complete_type = eLazyBoolNo;
1096     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1097       if (skip_dwarf_oso != oso_dwarf &&
1098           oso_dwarf->Supports_DW_AT_APPLE_objc_complete_type(NULL)) {
1099         m_supports_DW_AT_APPLE_objc_complete_type = eLazyBoolYes;
1100         return true;
1101       }
1102       return false;
1103     });
1104   }
1105   return m_supports_DW_AT_APPLE_objc_complete_type == eLazyBoolYes;
1106 }
1107 
1108 TypeSP SymbolFileDWARFDebugMap::FindCompleteObjCDefinitionTypeForDIE(
1109     const DWARFDIE &die, const ConstString &type_name,
1110     bool must_be_implementation) {
1111   // If we have a debug map, we will have an Objective-C symbol whose name is
1112   // the type name and whose type is eSymbolTypeObjCClass. If we can find that
1113   // symbol and find its containing parent, we can locate the .o file that will
1114   // contain the implementation definition since it will be scoped inside the
1115   // N_SO and we can then locate the SymbolFileDWARF that corresponds to that
1116   // N_SO.
1117   SymbolFileDWARF *oso_dwarf = NULL;
1118   TypeSP type_sp;
1119   ObjectFile *module_objfile = m_obj_file->GetModule()->GetObjectFile();
1120   if (module_objfile) {
1121     Symtab *symtab = module_objfile->GetSymtab();
1122     if (symtab) {
1123       Symbol *objc_class_symbol = symtab->FindFirstSymbolWithNameAndType(
1124           type_name, eSymbolTypeObjCClass, Symtab::eDebugAny,
1125           Symtab::eVisibilityAny);
1126       if (objc_class_symbol) {
1127         // Get the N_SO symbol that contains the objective C class symbol as
1128         // this should be the .o file that contains the real definition...
1129         const Symbol *source_file_symbol = symtab->GetParent(objc_class_symbol);
1130 
1131         if (source_file_symbol &&
1132             source_file_symbol->GetType() == eSymbolTypeSourceFile) {
1133           const uint32_t source_file_symbol_idx =
1134               symtab->GetIndexForSymbol(source_file_symbol);
1135           if (source_file_symbol_idx != UINT32_MAX) {
1136             CompileUnitInfo *compile_unit_info =
1137                 GetCompileUnitInfoForSymbolWithIndex(source_file_symbol_idx,
1138                                                      NULL);
1139             if (compile_unit_info) {
1140               oso_dwarf = GetSymbolFileByCompUnitInfo(compile_unit_info);
1141               if (oso_dwarf) {
1142                 TypeSP type_sp(oso_dwarf->FindCompleteObjCDefinitionTypeForDIE(
1143                     die, type_name, must_be_implementation));
1144                 if (type_sp) {
1145                   return type_sp;
1146                 }
1147               }
1148             }
1149           }
1150         }
1151       }
1152     }
1153   }
1154 
1155   // Only search all .o files for the definition if we don't need the
1156   // implementation because otherwise, with a valid debug map we should have
1157   // the ObjC class symbol and the code above should have found it.
1158   if (!must_be_implementation) {
1159     TypeSP type_sp;
1160 
1161     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1162       type_sp = oso_dwarf->FindCompleteObjCDefinitionTypeForDIE(
1163           die, type_name, must_be_implementation);
1164       return (bool)type_sp;
1165     });
1166 
1167     return type_sp;
1168   }
1169   return TypeSP();
1170 }
1171 
1172 uint32_t SymbolFileDWARFDebugMap::FindTypes(
1173     const SymbolContext &sc, const ConstString &name,
1174     const CompilerDeclContext *parent_decl_ctx, bool append,
1175     uint32_t max_matches,
1176     llvm::DenseSet<lldb_private::SymbolFile *> &searched_symbol_files,
1177     TypeMap &types) {
1178   if (!append)
1179     types.Clear();
1180 
1181   const uint32_t initial_types_size = types.GetSize();
1182   SymbolFileDWARF *oso_dwarf;
1183 
1184   if (sc.comp_unit) {
1185     oso_dwarf = GetSymbolFile(sc);
1186     if (oso_dwarf)
1187       return oso_dwarf->FindTypes(sc, name, parent_decl_ctx, append,
1188                                   max_matches, searched_symbol_files, types);
1189   } else {
1190     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1191       oso_dwarf->FindTypes(sc, name, parent_decl_ctx, append, max_matches,
1192                            searched_symbol_files, types);
1193       return types.GetSize() >= max_matches;
1194     });
1195   }
1196 
1197   return types.GetSize() - initial_types_size;
1198 }
1199 
1200 //
1201 // uint32_t
1202 // SymbolFileDWARFDebugMap::FindTypes (const SymbolContext& sc, const
1203 // RegularExpression& regex, bool append, uint32_t max_matches, Type::Encoding
1204 // encoding, lldb::user_id_t udt_uid, TypeList& types)
1205 //{
1206 //  SymbolFileDWARF *oso_dwarf = GetSymbolFile (sc);
1207 //  if (oso_dwarf)
1208 //      return oso_dwarf->FindTypes (sc, regex, append, max_matches, encoding,
1209 //      udt_uid, types);
1210 //  return 0;
1211 //}
1212 
1213 CompilerDeclContext SymbolFileDWARFDebugMap::FindNamespace(
1214     const lldb_private::SymbolContext &sc,
1215     const lldb_private::ConstString &name,
1216     const CompilerDeclContext *parent_decl_ctx) {
1217   CompilerDeclContext matching_namespace;
1218   SymbolFileDWARF *oso_dwarf;
1219 
1220   if (sc.comp_unit) {
1221     oso_dwarf = GetSymbolFile(sc);
1222     if (oso_dwarf)
1223       matching_namespace = oso_dwarf->FindNamespace(sc, name, parent_decl_ctx);
1224   } else {
1225     ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1226       matching_namespace = oso_dwarf->FindNamespace(sc, name, parent_decl_ctx);
1227 
1228       return (bool)matching_namespace;
1229     });
1230   }
1231 
1232   return matching_namespace;
1233 }
1234 
1235 void SymbolFileDWARFDebugMap::DumpClangAST(Stream &s) {
1236   ForEachSymbolFile([&s](SymbolFileDWARF *oso_dwarf) -> bool {
1237     oso_dwarf->DumpClangAST(s);
1238     return true;
1239   });
1240 }
1241 
1242 //------------------------------------------------------------------
1243 // PluginInterface protocol
1244 //------------------------------------------------------------------
1245 lldb_private::ConstString SymbolFileDWARFDebugMap::GetPluginName() {
1246   return GetPluginNameStatic();
1247 }
1248 
1249 uint32_t SymbolFileDWARFDebugMap::GetPluginVersion() { return 1; }
1250 
1251 lldb::CompUnitSP
1252 SymbolFileDWARFDebugMap::GetCompileUnit(SymbolFileDWARF *oso_dwarf) {
1253   if (oso_dwarf) {
1254     const uint32_t cu_count = GetNumCompileUnits();
1255     for (uint32_t cu_idx = 0; cu_idx < cu_count; ++cu_idx) {
1256       SymbolFileDWARF *oso_symfile =
1257           GetSymbolFileByCompUnitInfo(&m_compile_unit_infos[cu_idx]);
1258       if (oso_symfile == oso_dwarf) {
1259         if (!m_compile_unit_infos[cu_idx].compile_unit_sp)
1260           m_compile_unit_infos[cu_idx].compile_unit_sp =
1261               ParseCompileUnitAtIndex(cu_idx);
1262 
1263         return m_compile_unit_infos[cu_idx].compile_unit_sp;
1264       }
1265     }
1266   }
1267   llvm_unreachable("this shouldn't happen");
1268 }
1269 
1270 SymbolFileDWARFDebugMap::CompileUnitInfo *
1271 SymbolFileDWARFDebugMap::GetCompileUnitInfo(SymbolFileDWARF *oso_dwarf) {
1272   if (oso_dwarf) {
1273     const uint32_t cu_count = GetNumCompileUnits();
1274     for (uint32_t cu_idx = 0; cu_idx < cu_count; ++cu_idx) {
1275       SymbolFileDWARF *oso_symfile =
1276           GetSymbolFileByCompUnitInfo(&m_compile_unit_infos[cu_idx]);
1277       if (oso_symfile == oso_dwarf) {
1278         return &m_compile_unit_infos[cu_idx];
1279       }
1280     }
1281   }
1282   return NULL;
1283 }
1284 
1285 void SymbolFileDWARFDebugMap::SetCompileUnit(SymbolFileDWARF *oso_dwarf,
1286                                              const CompUnitSP &cu_sp) {
1287   if (oso_dwarf) {
1288     const uint32_t cu_count = GetNumCompileUnits();
1289     for (uint32_t cu_idx = 0; cu_idx < cu_count; ++cu_idx) {
1290       SymbolFileDWARF *oso_symfile =
1291           GetSymbolFileByCompUnitInfo(&m_compile_unit_infos[cu_idx]);
1292       if (oso_symfile == oso_dwarf) {
1293         if (m_compile_unit_infos[cu_idx].compile_unit_sp) {
1294           assert(m_compile_unit_infos[cu_idx].compile_unit_sp.get() ==
1295                  cu_sp.get());
1296         } else {
1297           m_compile_unit_infos[cu_idx].compile_unit_sp = cu_sp;
1298           m_obj_file->GetModule()->GetSymbolVendor()->SetCompileUnitAtIndex(
1299               cu_idx, cu_sp);
1300         }
1301       }
1302     }
1303   }
1304 }
1305 
1306 CompilerDeclContext
1307 SymbolFileDWARFDebugMap::GetDeclContextForUID(lldb::user_id_t type_uid) {
1308   const uint64_t oso_idx = GetOSOIndexFromUserID(type_uid);
1309   SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
1310   if (oso_dwarf)
1311     return oso_dwarf->GetDeclContextForUID(type_uid);
1312   return CompilerDeclContext();
1313 }
1314 
1315 CompilerDeclContext
1316 SymbolFileDWARFDebugMap::GetDeclContextContainingUID(lldb::user_id_t type_uid) {
1317   const uint64_t oso_idx = GetOSOIndexFromUserID(type_uid);
1318   SymbolFileDWARF *oso_dwarf = GetSymbolFileByOSOIndex(oso_idx);
1319   if (oso_dwarf)
1320     return oso_dwarf->GetDeclContextContainingUID(type_uid);
1321   return CompilerDeclContext();
1322 }
1323 
1324 void SymbolFileDWARFDebugMap::ParseDeclsForContext(
1325     lldb_private::CompilerDeclContext decl_ctx) {
1326   ForEachSymbolFile([&](SymbolFileDWARF *oso_dwarf) -> bool {
1327     oso_dwarf->ParseDeclsForContext(decl_ctx);
1328     return true; // Keep iterating
1329   });
1330 }
1331 
1332 bool SymbolFileDWARFDebugMap::AddOSOFileRange(CompileUnitInfo *cu_info,
1333                                               lldb::addr_t exe_file_addr,
1334                                               lldb::addr_t exe_byte_size,
1335                                               lldb::addr_t oso_file_addr,
1336                                               lldb::addr_t oso_byte_size) {
1337   const uint32_t debug_map_idx =
1338       m_debug_map.FindEntryIndexThatContains(exe_file_addr);
1339   if (debug_map_idx != UINT32_MAX) {
1340     DebugMap::Entry *debug_map_entry =
1341         m_debug_map.FindEntryThatContains(exe_file_addr);
1342     debug_map_entry->data.SetOSOFileAddress(oso_file_addr);
1343     addr_t range_size = std::min<addr_t>(exe_byte_size, oso_byte_size);
1344     if (range_size == 0) {
1345       range_size = std::max<addr_t>(exe_byte_size, oso_byte_size);
1346       if (range_size == 0)
1347         range_size = 1;
1348     }
1349     cu_info->file_range_map.Append(
1350         FileRangeMap::Entry(oso_file_addr, range_size, exe_file_addr));
1351     return true;
1352   }
1353   return false;
1354 }
1355 
1356 void SymbolFileDWARFDebugMap::FinalizeOSOFileRanges(CompileUnitInfo *cu_info) {
1357   cu_info->file_range_map.Sort();
1358 #if defined(DEBUG_OSO_DMAP)
1359   const FileRangeMap &oso_file_range_map = cu_info->GetFileRangeMap(this);
1360   const size_t n = oso_file_range_map.GetSize();
1361   printf("SymbolFileDWARFDebugMap::FinalizeOSOFileRanges (cu_info = %p) %s\n",
1362          cu_info, cu_info->oso_sp->module_sp->GetFileSpec().GetPath().c_str());
1363   for (size_t i = 0; i < n; ++i) {
1364     const FileRangeMap::Entry &entry = oso_file_range_map.GetEntryRef(i);
1365     printf("oso [0x%16.16" PRIx64 " - 0x%16.16" PRIx64
1366            ") ==> exe [0x%16.16" PRIx64 " - 0x%16.16" PRIx64 ")\n",
1367            entry.GetRangeBase(), entry.GetRangeEnd(), entry.data,
1368            entry.data + entry.GetByteSize());
1369   }
1370 #endif
1371 }
1372 
1373 lldb::addr_t
1374 SymbolFileDWARFDebugMap::LinkOSOFileAddress(SymbolFileDWARF *oso_symfile,
1375                                             lldb::addr_t oso_file_addr) {
1376   CompileUnitInfo *cu_info = GetCompileUnitInfo(oso_symfile);
1377   if (cu_info) {
1378     const FileRangeMap::Entry *oso_range_entry =
1379         cu_info->GetFileRangeMap(this).FindEntryThatContains(oso_file_addr);
1380     if (oso_range_entry) {
1381       const DebugMap::Entry *debug_map_entry =
1382           m_debug_map.FindEntryThatContains(oso_range_entry->data);
1383       if (debug_map_entry) {
1384         const lldb::addr_t offset =
1385             oso_file_addr - oso_range_entry->GetRangeBase();
1386         const lldb::addr_t exe_file_addr =
1387             debug_map_entry->GetRangeBase() + offset;
1388         return exe_file_addr;
1389       }
1390     }
1391   }
1392   return LLDB_INVALID_ADDRESS;
1393 }
1394 
1395 bool SymbolFileDWARFDebugMap::LinkOSOAddress(Address &addr) {
1396   // Make sure this address hasn't been fixed already
1397   Module *exe_module = GetObjectFile()->GetModule().get();
1398   Module *addr_module = addr.GetModule().get();
1399   if (addr_module == exe_module)
1400     return true; // Address is already in terms of the main executable module
1401 
1402   CompileUnitInfo *cu_info = GetCompileUnitInfo(GetSymbolFileAsSymbolFileDWARF(
1403       addr_module->GetSymbolVendor()->GetSymbolFile()));
1404   if (cu_info) {
1405     const lldb::addr_t oso_file_addr = addr.GetFileAddress();
1406     const FileRangeMap::Entry *oso_range_entry =
1407         cu_info->GetFileRangeMap(this).FindEntryThatContains(oso_file_addr);
1408     if (oso_range_entry) {
1409       const DebugMap::Entry *debug_map_entry =
1410           m_debug_map.FindEntryThatContains(oso_range_entry->data);
1411       if (debug_map_entry) {
1412         const lldb::addr_t offset =
1413             oso_file_addr - oso_range_entry->GetRangeBase();
1414         const lldb::addr_t exe_file_addr =
1415             debug_map_entry->GetRangeBase() + offset;
1416         return exe_module->ResolveFileAddress(exe_file_addr, addr);
1417       }
1418     }
1419   }
1420   return true;
1421 }
1422 
1423 LineTable *SymbolFileDWARFDebugMap::LinkOSOLineTable(SymbolFileDWARF *oso_dwarf,
1424                                                      LineTable *line_table) {
1425   CompileUnitInfo *cu_info = GetCompileUnitInfo(oso_dwarf);
1426   if (cu_info)
1427     return line_table->LinkLineTable(cu_info->GetFileRangeMap(this));
1428   return NULL;
1429 }
1430 
1431 size_t
1432 SymbolFileDWARFDebugMap::AddOSOARanges(SymbolFileDWARF *dwarf2Data,
1433                                        DWARFDebugAranges *debug_aranges) {
1434   size_t num_line_entries_added = 0;
1435   if (debug_aranges && dwarf2Data) {
1436     CompileUnitInfo *compile_unit_info = GetCompileUnitInfo(dwarf2Data);
1437     if (compile_unit_info) {
1438       const FileRangeMap &file_range_map =
1439           compile_unit_info->GetFileRangeMap(this);
1440       for (size_t idx = 0; idx < file_range_map.GetSize(); idx++) {
1441         const FileRangeMap::Entry *entry = file_range_map.GetEntryAtIndex(idx);
1442         if (entry) {
1443           debug_aranges->AppendRange(dwarf2Data->GetID(), entry->GetRangeBase(),
1444                                      entry->GetRangeEnd());
1445           num_line_entries_added++;
1446         }
1447       }
1448     }
1449   }
1450   return num_line_entries_added;
1451 }
1452