180814287SRaphael Isemann //===-- SymbolFilePDB.cpp -------------------------------------------------===//
274e08ca0SZachary Turner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
674e08ca0SZachary Turner //
774e08ca0SZachary Turner //===----------------------------------------------------------------------===//
874e08ca0SZachary Turner 
974e08ca0SZachary Turner #include "SymbolFilePDB.h"
1074e08ca0SZachary Turner 
11c1e530eeSAleksandr Urakov #include "PDBASTParser.h"
12c1e530eeSAleksandr Urakov #include "PDBLocationToDWARFExpression.h"
13c1e530eeSAleksandr Urakov 
1442dff790SZachary Turner #include "clang/Lex/Lexer.h"
1542dff790SZachary Turner 
1674e08ca0SZachary Turner #include "lldb/Core/Module.h"
1774e08ca0SZachary Turner #include "lldb/Core/PluginManager.h"
186e3b0cc2SRaphael Isemann #include "lldb/Symbol/TypeSystemClang.h"
1974e08ca0SZachary Turner #include "lldb/Symbol/CompileUnit.h"
2074e08ca0SZachary Turner #include "lldb/Symbol/LineTable.h"
2174e08ca0SZachary Turner #include "lldb/Symbol/ObjectFile.h"
2274e08ca0SZachary Turner #include "lldb/Symbol/SymbolContext.h"
2310a02577SAaron Smith #include "lldb/Symbol/SymbolVendor.h"
24ec40f818SAaron Smith #include "lldb/Symbol/TypeList.h"
25308e39caSAaron Smith #include "lldb/Symbol/TypeMap.h"
26cab0d23fSAaron Smith #include "lldb/Symbol/Variable.h"
270e252e38SAlex Langford #include "lldb/Utility/Log.h"
2886e9434dSAaron Smith #include "lldb/Utility/RegularExpression.h"
2974e08ca0SZachary Turner 
30b8d8c62bSPavel Labath #include "llvm/DebugInfo/PDB/GenericError.h"
311f8552abSAaron Smith #include "llvm/DebugInfo/PDB/IPDBDataStream.h"
3274e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/IPDBEnumChildren.h"
3374e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/IPDBLineNumber.h"
34308e39caSAaron Smith #include "llvm/DebugInfo/PDB/IPDBSectionContrib.h"
3574e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/IPDBSourceFile.h"
361f8552abSAaron Smith #include "llvm/DebugInfo/PDB/IPDBTable.h"
3774e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbol.h"
387ac1c780SAaron Smith #include "llvm/DebugInfo/PDB/PDBSymbolBlock.h"
3974e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
4074e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolCompilandDetails.h"
411f8552abSAaron Smith #include "llvm/DebugInfo/PDB/PDBSymbolData.h"
4274e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
4374e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolFunc.h"
4474e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugEnd.h"
4574e08ca0SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugStart.h"
467ac1c780SAaron Smith #include "llvm/DebugInfo/PDB/PDBSymbolPublicSymbol.h"
4742dff790SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolTypeEnum.h"
4842dff790SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolTypeTypedef.h"
4942dff790SZachary Turner #include "llvm/DebugInfo/PDB/PDBSymbolTypeUDT.h"
5042dff790SZachary Turner 
51672d2c12SJonas Devlieghere #include "Plugins/Language/CPlusPlus/CPlusPlusLanguage.h"
52c1e530eeSAleksandr Urakov #include "Plugins/Language/CPlusPlus/MSVCUndecoratedNameParser.h"
53307f5ae8SZachary Turner #include "Plugins/SymbolFile/NativePDB/SymbolFileNativePDB.h"
5442dff790SZachary Turner 
5542dff790SZachary Turner #include <regex>
5674e08ca0SZachary Turner 
5710a02577SAaron Smith using namespace lldb;
5874e08ca0SZachary Turner using namespace lldb_private;
5954fd7ff6SZachary Turner using namespace llvm::pdb;
6074e08ca0SZachary Turner 
617d71dd92SAdrian Prantl char SymbolFilePDB::ID;
627d71dd92SAdrian Prantl 
63b9c1b51eSKate Stone namespace {
64b9c1b51eSKate Stone lldb::LanguageType TranslateLanguage(PDB_Lang lang) {
65b9c1b51eSKate Stone   switch (lang) {
6654fd7ff6SZachary Turner   case PDB_Lang::Cpp:
6774e08ca0SZachary Turner     return lldb::LanguageType::eLanguageTypeC_plus_plus;
6854fd7ff6SZachary Turner   case PDB_Lang::C:
6974e08ca0SZachary Turner     return lldb::LanguageType::eLanguageTypeC;
700561be6cSNathan Lanza   case PDB_Lang::Swift:
710561be6cSNathan Lanza     return lldb::LanguageType::eLanguageTypeSwift;
7274e08ca0SZachary Turner   default:
7374e08ca0SZachary Turner     return lldb::LanguageType::eLanguageTypeUnknown;
7474e08ca0SZachary Turner   }
7574e08ca0SZachary Turner }
767e8c7beaSZachary Turner 
77b9c1b51eSKate Stone bool ShouldAddLine(uint32_t requested_line, uint32_t actual_line,
78b9c1b51eSKate Stone                    uint32_t addr_length) {
79b9c1b51eSKate Stone   return ((requested_line == 0 || actual_line == requested_line) &&
80b9c1b51eSKate Stone           addr_length > 0);
817e8c7beaSZachary Turner }
82c8316ed2SAaron Smith } // namespace
8374e08ca0SZachary Turner 
84307f5ae8SZachary Turner static bool ShouldUseNativeReader() {
8571970b72SGreg Clayton #if defined(_WIN32)
86307f5ae8SZachary Turner   llvm::StringRef use_native = ::getenv("LLDB_USE_NATIVE_PDB_READER");
87307f5ae8SZachary Turner   return use_native.equals_lower("on") || use_native.equals_lower("yes") ||
88307f5ae8SZachary Turner          use_native.equals_lower("1") || use_native.equals_lower("true");
8971970b72SGreg Clayton #else
9071970b72SGreg Clayton   return true;
9171970b72SGreg Clayton #endif
92307f5ae8SZachary Turner }
93307f5ae8SZachary Turner 
94b9c1b51eSKate Stone void SymbolFilePDB::Initialize() {
95307f5ae8SZachary Turner   if (ShouldUseNativeReader()) {
96307f5ae8SZachary Turner     npdb::SymbolFileNativePDB::Initialize();
97307f5ae8SZachary Turner   } else {
98b9c1b51eSKate Stone     PluginManager::RegisterPlugin(GetPluginNameStatic(),
99b9c1b51eSKate Stone                                   GetPluginDescriptionStatic(), CreateInstance,
10074e08ca0SZachary Turner                                   DebuggerInitialize);
10174e08ca0SZachary Turner   }
102307f5ae8SZachary Turner }
10374e08ca0SZachary Turner 
104b9c1b51eSKate Stone void SymbolFilePDB::Terminate() {
105307f5ae8SZachary Turner   if (ShouldUseNativeReader()) {
106307f5ae8SZachary Turner     npdb::SymbolFileNativePDB::Terminate();
107307f5ae8SZachary Turner   } else {
10874e08ca0SZachary Turner     PluginManager::UnregisterPlugin(CreateInstance);
10974e08ca0SZachary Turner   }
110307f5ae8SZachary Turner }
11174e08ca0SZachary Turner 
112b9c1b51eSKate Stone void SymbolFilePDB::DebuggerInitialize(lldb_private::Debugger &debugger) {}
11374e08ca0SZachary Turner 
114b9c1b51eSKate Stone lldb_private::ConstString SymbolFilePDB::GetPluginNameStatic() {
11574e08ca0SZachary Turner   static ConstString g_name("pdb");
11674e08ca0SZachary Turner   return g_name;
11774e08ca0SZachary Turner }
11874e08ca0SZachary Turner 
119b9c1b51eSKate Stone const char *SymbolFilePDB::GetPluginDescriptionStatic() {
12074e08ca0SZachary Turner   return "Microsoft PDB debug symbol file reader.";
12174e08ca0SZachary Turner }
12274e08ca0SZachary Turner 
12374e08ca0SZachary Turner lldb_private::SymbolFile *
124d2deeb44SPavel Labath SymbolFilePDB::CreateInstance(ObjectFileSP objfile_sp) {
125d2deeb44SPavel Labath   return new SymbolFilePDB(std::move(objfile_sp));
12674e08ca0SZachary Turner }
12774e08ca0SZachary Turner 
128d2deeb44SPavel Labath SymbolFilePDB::SymbolFilePDB(lldb::ObjectFileSP objfile_sp)
129d2deeb44SPavel Labath     : SymbolFile(std::move(objfile_sp)), m_session_up(), m_global_scope_up() {}
13074e08ca0SZachary Turner 
131b9c1b51eSKate Stone SymbolFilePDB::~SymbolFilePDB() {}
13274e08ca0SZachary Turner 
133b9c1b51eSKate Stone uint32_t SymbolFilePDB::CalculateAbilities() {
1341f8552abSAaron Smith   uint32_t abilities = 0;
135d2deeb44SPavel Labath   if (!m_objfile_sp)
1361f8552abSAaron Smith     return 0;
1371f8552abSAaron Smith 
138b9c1b51eSKate Stone   if (!m_session_up) {
13974e08ca0SZachary Turner     // Lazily load and match the PDB file, but only do this once.
140d2deeb44SPavel Labath     std::string exePath = m_objfile_sp->GetFileSpec().GetPath();
141b9c1b51eSKate Stone     auto error = loadDataForEXE(PDB_ReaderType::DIA, llvm::StringRef(exePath),
142b9c1b51eSKate Stone                                 m_session_up);
143b9c1b51eSKate Stone     if (error) {
1444fd6a960SZachary Turner       llvm::consumeError(std::move(error));
145d2deeb44SPavel Labath       auto module_sp = m_objfile_sp->GetModule();
1461f8552abSAaron Smith       if (!module_sp)
1471f8552abSAaron Smith         return 0;
1481f8552abSAaron Smith       // See if any symbol file is specified through `--symfile` option.
1491f8552abSAaron Smith       FileSpec symfile = module_sp->GetSymbolFileFileSpec();
1501f8552abSAaron Smith       if (!symfile)
1511f8552abSAaron Smith         return 0;
1521f8552abSAaron Smith       error = loadDataForPDB(PDB_ReaderType::DIA,
153c8316ed2SAaron Smith                              llvm::StringRef(symfile.GetPath()), m_session_up);
1541f8552abSAaron Smith       if (error) {
1551f8552abSAaron Smith         llvm::consumeError(std::move(error));
15674e08ca0SZachary Turner         return 0;
15774e08ca0SZachary Turner       }
158b8d8c62bSPavel Labath     }
1591f8552abSAaron Smith   }
160d5a925f4SAaron Smith   if (!m_session_up)
1611f8552abSAaron Smith     return 0;
1621f8552abSAaron Smith 
1631f8552abSAaron Smith   auto enum_tables_up = m_session_up->getEnumTables();
1641f8552abSAaron Smith   if (!enum_tables_up)
1651f8552abSAaron Smith     return 0;
1661f8552abSAaron Smith   while (auto table_up = enum_tables_up->getNext()) {
1671f8552abSAaron Smith     if (table_up->getItemCount() == 0)
1681f8552abSAaron Smith       continue;
1691f8552abSAaron Smith     auto type = table_up->getTableType();
1701f8552abSAaron Smith     switch (type) {
1711f8552abSAaron Smith     case PDB_TableType::Symbols:
1721f8552abSAaron Smith       // This table represents a store of symbols with types listed in
1731f8552abSAaron Smith       // PDBSym_Type
174c8316ed2SAaron Smith       abilities |= (CompileUnits | Functions | Blocks | GlobalVariables |
175c8316ed2SAaron Smith                     LocalVariables | VariableTypes);
1761f8552abSAaron Smith       break;
1771f8552abSAaron Smith     case PDB_TableType::LineNumbers:
1781f8552abSAaron Smith       abilities |= LineTables;
1791f8552abSAaron Smith       break;
180c8316ed2SAaron Smith     default:
181c8316ed2SAaron Smith       break;
1821f8552abSAaron Smith     }
1831f8552abSAaron Smith   }
1841f8552abSAaron Smith   return abilities;
18574e08ca0SZachary Turner }
18674e08ca0SZachary Turner 
187b9c1b51eSKate Stone void SymbolFilePDB::InitializeObject() {
188d2deeb44SPavel Labath   lldb::addr_t obj_load_address =
189d2deeb44SPavel Labath       m_objfile_sp->GetBaseAddress().GetFileAddress();
190c8316ed2SAaron Smith   lldbassert(obj_load_address && obj_load_address != LLDB_INVALID_ADDRESS);
19174e08ca0SZachary Turner   m_session_up->setLoadAddress(obj_load_address);
19210a02577SAaron Smith   if (!m_global_scope_up)
19310a02577SAaron Smith     m_global_scope_up = m_session_up->getGlobalScope();
19410a02577SAaron Smith   lldbassert(m_global_scope_up.get());
19574e08ca0SZachary Turner }
19674e08ca0SZachary Turner 
197e0119909SPavel Labath uint32_t SymbolFilePDB::CalculateNumCompileUnits() {
19810a02577SAaron Smith   auto compilands = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
19910a02577SAaron Smith   if (!compilands)
20010a02577SAaron Smith     return 0;
20110a02577SAaron Smith 
20210a02577SAaron Smith   // The linker could link *.dll (compiland language = LINK), or import
20305097246SAdrian Prantl   // *.dll. For example, a compiland with name `Import:KERNEL32.dll` could be
20405097246SAdrian Prantl   // found as a child of the global scope (PDB executable). Usually, such
20505097246SAdrian Prantl   // compilands contain `thunk` symbols in which we are not interested for
20605097246SAdrian Prantl   // now. However we still count them in the compiland list. If we perform
20705097246SAdrian Prantl   // any compiland related activity, like finding symbols through
20805097246SAdrian Prantl   // llvm::pdb::IPDBSession methods, such compilands will all be searched
20905097246SAdrian Prantl   // automatically no matter whether we include them or not.
210e0119909SPavel Labath   uint32_t compile_unit_count = compilands->getChildCount();
21174e08ca0SZachary Turner 
212b9c1b51eSKate Stone   // The linker can inject an additional "dummy" compilation unit into the
2139d0eb996SAdrian McCarthy   // PDB. Ignore this special compile unit for our purposes, if it is there.
2149d0eb996SAdrian McCarthy   // It is always the last one.
215e0119909SPavel Labath   auto last_compiland_up = compilands->getChildAtIndex(compile_unit_count - 1);
21610a02577SAaron Smith   lldbassert(last_compiland_up.get());
21710a02577SAaron Smith   std::string name = last_compiland_up->getName();
21874e08ca0SZachary Turner   if (name == "* Linker *")
219e0119909SPavel Labath     --compile_unit_count;
220e0119909SPavel Labath   return compile_unit_count;
22174e08ca0SZachary Turner }
22274e08ca0SZachary Turner 
22310a02577SAaron Smith void SymbolFilePDB::GetCompileUnitIndex(
224c8316ed2SAaron Smith     const llvm::pdb::PDBSymbolCompiland &pdb_compiland, uint32_t &index) {
22510a02577SAaron Smith   auto results_up = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
22610a02577SAaron Smith   if (!results_up)
22710a02577SAaron Smith     return;
228e664b5dcSAaron Smith   auto uid = pdb_compiland.getSymIndexId();
229fbdf0b93SRaphael Isemann   for (uint32_t cu_idx = 0; cu_idx < GetNumCompileUnits(); ++cu_idx) {
23010a02577SAaron Smith     auto compiland_up = results_up->getChildAtIndex(cu_idx);
23110a02577SAaron Smith     if (!compiland_up)
23210a02577SAaron Smith       continue;
23310a02577SAaron Smith     if (compiland_up->getSymIndexId() == uid) {
23410a02577SAaron Smith       index = cu_idx;
23510a02577SAaron Smith       return;
23610a02577SAaron Smith     }
23710a02577SAaron Smith   }
23810a02577SAaron Smith   index = UINT32_MAX;
23910a02577SAaron Smith   return;
24010a02577SAaron Smith }
24110a02577SAaron Smith 
24210a02577SAaron Smith std::unique_ptr<llvm::pdb::PDBSymbolCompiland>
24310a02577SAaron Smith SymbolFilePDB::GetPDBCompilandByUID(uint32_t uid) {
24410a02577SAaron Smith   return m_session_up->getConcreteSymbolById<PDBSymbolCompiland>(uid);
24510a02577SAaron Smith }
24610a02577SAaron Smith 
247b9c1b51eSKate Stone lldb::CompUnitSP SymbolFilePDB::ParseCompileUnitAtIndex(uint32_t index) {
24810a02577SAaron Smith   if (index >= GetNumCompileUnits())
24910a02577SAaron Smith     return CompUnitSP();
25074e08ca0SZachary Turner 
25110a02577SAaron Smith   // Assuming we always retrieve same compilands listed in same order through
25210a02577SAaron Smith   // `PDBSymbolExe::findAllChildren` method, otherwise using `index` to get a
25310a02577SAaron Smith   // compile unit makes no sense.
25410a02577SAaron Smith   auto results = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
25510a02577SAaron Smith   if (!results)
25610a02577SAaron Smith     return CompUnitSP();
25710a02577SAaron Smith   auto compiland_up = results->getChildAtIndex(index);
25810a02577SAaron Smith   if (!compiland_up)
25910a02577SAaron Smith     return CompUnitSP();
26010a02577SAaron Smith   return ParseCompileUnitForUID(compiland_up->getSymIndexId(), index);
26174e08ca0SZachary Turner }
26274e08ca0SZachary Turner 
263863f8c18SZachary Turner lldb::LanguageType SymbolFilePDB::ParseLanguage(CompileUnit &comp_unit) {
264656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
265863f8c18SZachary Turner   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
26610a02577SAaron Smith   if (!compiland_up)
26774e08ca0SZachary Turner     return lldb::eLanguageTypeUnknown;
26810a02577SAaron Smith   auto details = compiland_up->findOneChild<PDBSymbolCompilandDetails>();
26974e08ca0SZachary Turner   if (!details)
27074e08ca0SZachary Turner     return lldb::eLanguageTypeUnknown;
27174e08ca0SZachary Turner   return TranslateLanguage(details->getLanguage());
27274e08ca0SZachary Turner }
27374e08ca0SZachary Turner 
274863f8c18SZachary Turner lldb_private::Function *
275863f8c18SZachary Turner SymbolFilePDB::ParseCompileUnitFunctionForPDBFunc(const PDBSymbolFunc &pdb_func,
276863f8c18SZachary Turner                                                   CompileUnit &comp_unit) {
277863f8c18SZachary Turner   if (FunctionSP result = comp_unit.FindFunctionByUID(pdb_func.getSymIndexId()))
278a5235af9SAleksandr Urakov     return result.get();
279a5235af9SAleksandr Urakov 
280e664b5dcSAaron Smith   auto file_vm_addr = pdb_func.getVirtualAddress();
281308e39caSAaron Smith   if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
2827ac1c780SAaron Smith     return nullptr;
2837ac1c780SAaron Smith 
284e664b5dcSAaron Smith   auto func_length = pdb_func.getLength();
285c8316ed2SAaron Smith   AddressRange func_range =
286863f8c18SZachary Turner       AddressRange(file_vm_addr, func_length,
287863f8c18SZachary Turner                    GetObjectFile()->GetModule()->GetSectionList());
2887ac1c780SAaron Smith   if (!func_range.GetBaseAddress().IsValid())
2897ac1c780SAaron Smith     return nullptr;
2907ac1c780SAaron Smith 
291e664b5dcSAaron Smith   lldb_private::Type *func_type = ResolveTypeUID(pdb_func.getSymIndexId());
2927ac1c780SAaron Smith   if (!func_type)
2937ac1c780SAaron Smith     return nullptr;
2947ac1c780SAaron Smith 
295e664b5dcSAaron Smith   user_id_t func_type_uid = pdb_func.getSignatureId();
296f76fe682SAaron Smith 
2977ac1c780SAaron Smith   Mangled mangled = GetMangledForPDBFunc(pdb_func);
2987ac1c780SAaron Smith 
299c8316ed2SAaron Smith   FunctionSP func_sp =
300863f8c18SZachary Turner       std::make_shared<Function>(&comp_unit, pdb_func.getSymIndexId(),
301c8316ed2SAaron Smith                                  func_type_uid, mangled, func_type, func_range);
3027ac1c780SAaron Smith 
303863f8c18SZachary Turner   comp_unit.AddFunction(func_sp);
304c68925abSZachary Turner 
305d0050d1bSNathan Lanza   LanguageType lang = ParseLanguage(comp_unit);
3060e252e38SAlex Langford   auto type_system_or_err = GetTypeSystemForLanguage(lang);
3070e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
3080e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
3090e252e38SAlex Langford                    std::move(err), "Unable to parse PDBFunc");
310c68925abSZachary Turner     return nullptr;
3110e252e38SAlex Langford   }
3120e252e38SAlex Langford 
3136e3b0cc2SRaphael Isemann   TypeSystemClang *clang_type_system =
3146e3b0cc2SRaphael Isemann     llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
315c68925abSZachary Turner   if (!clang_type_system)
316c68925abSZachary Turner     return nullptr;
317c68925abSZachary Turner   clang_type_system->GetPDBParser()->GetDeclForSymbol(pdb_func);
318c68925abSZachary Turner 
3197ac1c780SAaron Smith   return func_sp.get();
3207ac1c780SAaron Smith }
3217ac1c780SAaron Smith 
322863f8c18SZachary Turner size_t SymbolFilePDB::ParseFunctions(CompileUnit &comp_unit) {
323656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
3247ac1c780SAaron Smith   size_t func_added = 0;
325863f8c18SZachary Turner   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
3267ac1c780SAaron Smith   if (!compiland_up)
3277ac1c780SAaron Smith     return 0;
3287ac1c780SAaron Smith   auto results_up = compiland_up->findAllChildren<PDBSymbolFunc>();
3297ac1c780SAaron Smith   if (!results_up)
3307ac1c780SAaron Smith     return 0;
3317ac1c780SAaron Smith   while (auto pdb_func_up = results_up->getNext()) {
332863f8c18SZachary Turner     auto func_sp = comp_unit.FindFunctionByUID(pdb_func_up->getSymIndexId());
3337ac1c780SAaron Smith     if (!func_sp) {
334863f8c18SZachary Turner       if (ParseCompileUnitFunctionForPDBFunc(*pdb_func_up, comp_unit))
3357ac1c780SAaron Smith         ++func_added;
3367ac1c780SAaron Smith     }
3377ac1c780SAaron Smith   }
3387ac1c780SAaron Smith   return func_added;
33974e08ca0SZachary Turner }
34074e08ca0SZachary Turner 
341863f8c18SZachary Turner bool SymbolFilePDB::ParseLineTable(CompileUnit &comp_unit) {
342656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
343863f8c18SZachary Turner   if (comp_unit.GetLineTable())
34410a02577SAaron Smith     return true;
345863f8c18SZachary Turner   return ParseCompileUnitLineTable(comp_unit, 0);
34674e08ca0SZachary Turner }
34774e08ca0SZachary Turner 
348863f8c18SZachary Turner bool SymbolFilePDB::ParseDebugMacros(CompileUnit &comp_unit) {
34974e08ca0SZachary Turner   // PDB doesn't contain information about macros
35074e08ca0SZachary Turner   return false;
35174e08ca0SZachary Turner }
35274e08ca0SZachary Turner 
353863f8c18SZachary Turner bool SymbolFilePDB::ParseSupportFiles(
354863f8c18SZachary Turner     CompileUnit &comp_unit, lldb_private::FileSpecList &support_files) {
35574e08ca0SZachary Turner 
356b9c1b51eSKate Stone   // In theory this is unnecessary work for us, because all of this information
3579d0eb996SAdrian McCarthy   // is easily (and quickly) accessible from DebugInfoPDB, so caching it a
3589d0eb996SAdrian McCarthy   // second time seems like a waste.  Unfortunately, there's no good way around
3599d0eb996SAdrian McCarthy   // this short of a moderate refactor since SymbolVendor depends on being able
3609d0eb996SAdrian McCarthy   // to cache this list.
361656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
362863f8c18SZachary Turner   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
36310a02577SAaron Smith   if (!compiland_up)
36474e08ca0SZachary Turner     return false;
36510a02577SAaron Smith   auto files = m_session_up->getSourceFilesForCompiland(*compiland_up);
36674e08ca0SZachary Turner   if (!files || files->getChildCount() == 0)
36774e08ca0SZachary Turner     return false;
36874e08ca0SZachary Turner 
369b9c1b51eSKate Stone   while (auto file = files->getNext()) {
3708f3be7a3SJonas Devlieghere     FileSpec spec(file->getFileName(), FileSpec::Style::windows);
37110a02577SAaron Smith     support_files.AppendIfUnique(spec);
37274e08ca0SZachary Turner   }
3739ea80d25SPavel Labath 
37474e08ca0SZachary Turner   return true;
37574e08ca0SZachary Turner }
37674e08ca0SZachary Turner 
377b9c1b51eSKate Stone bool SymbolFilePDB::ParseImportedModules(
378b9c1b51eSKate Stone     const lldb_private::SymbolContext &sc,
3790f30a3b6SAdrian Prantl     std::vector<SourceModule> &imported_modules) {
38074e08ca0SZachary Turner   // PDB does not yet support module debug info
38174e08ca0SZachary Turner   return false;
38274e08ca0SZachary Turner }
38374e08ca0SZachary Turner 
384c8316ed2SAaron Smith static size_t ParseFunctionBlocksForPDBSymbol(
385ffc1b8fdSZachary Turner     uint64_t func_file_vm_addr, const llvm::pdb::PDBSymbol *pdb_symbol,
386ffc1b8fdSZachary Turner     lldb_private::Block *parent_block, bool is_top_parent) {
3877ac1c780SAaron Smith   assert(pdb_symbol && parent_block);
3887ac1c780SAaron Smith 
3897ac1c780SAaron Smith   size_t num_added = 0;
3907ac1c780SAaron Smith   switch (pdb_symbol->getSymTag()) {
3917ac1c780SAaron Smith   case PDB_SymType::Block:
3927ac1c780SAaron Smith   case PDB_SymType::Function: {
3937ac1c780SAaron Smith     Block *block = nullptr;
3947ac1c780SAaron Smith     auto &raw_sym = pdb_symbol->getRawSymbol();
3957ac1c780SAaron Smith     if (auto *pdb_func = llvm::dyn_cast<PDBSymbolFunc>(pdb_symbol)) {
3967ac1c780SAaron Smith       if (pdb_func->hasNoInlineAttribute())
3977ac1c780SAaron Smith         break;
3987ac1c780SAaron Smith       if (is_top_parent)
3997ac1c780SAaron Smith         block = parent_block;
4007ac1c780SAaron Smith       else
4017ac1c780SAaron Smith         break;
4027ac1c780SAaron Smith     } else if (llvm::dyn_cast<PDBSymbolBlock>(pdb_symbol)) {
4037ac1c780SAaron Smith       auto uid = pdb_symbol->getSymIndexId();
4047ac1c780SAaron Smith       if (parent_block->FindBlockByID(uid))
4057ac1c780SAaron Smith         break;
4067ac1c780SAaron Smith       if (raw_sym.getVirtualAddress() < func_file_vm_addr)
4077ac1c780SAaron Smith         break;
4087ac1c780SAaron Smith 
4097ac1c780SAaron Smith       auto block_sp = std::make_shared<Block>(pdb_symbol->getSymIndexId());
4107ac1c780SAaron Smith       parent_block->AddChild(block_sp);
4117ac1c780SAaron Smith       block = block_sp.get();
4127ac1c780SAaron Smith     } else
4137ac1c780SAaron Smith       llvm_unreachable("Unexpected PDB symbol!");
4147ac1c780SAaron Smith 
415c8316ed2SAaron Smith     block->AddRange(Block::Range(
416c8316ed2SAaron Smith         raw_sym.getVirtualAddress() - func_file_vm_addr, raw_sym.getLength()));
4177ac1c780SAaron Smith     block->FinalizeRanges();
4187ac1c780SAaron Smith     ++num_added;
4197ac1c780SAaron Smith 
4207ac1c780SAaron Smith     auto results_up = pdb_symbol->findAllChildren();
4217ac1c780SAaron Smith     if (!results_up)
4227ac1c780SAaron Smith       break;
4237ac1c780SAaron Smith     while (auto symbol_up = results_up->getNext()) {
424c8316ed2SAaron Smith       num_added += ParseFunctionBlocksForPDBSymbol(
425ffc1b8fdSZachary Turner           func_file_vm_addr, symbol_up.get(), block, false);
4267ac1c780SAaron Smith     }
4277ac1c780SAaron Smith   } break;
428c8316ed2SAaron Smith   default:
429c8316ed2SAaron Smith     break;
4307ac1c780SAaron Smith   }
4317ac1c780SAaron Smith   return num_added;
4327ac1c780SAaron Smith }
4337ac1c780SAaron Smith 
434ffc1b8fdSZachary Turner size_t SymbolFilePDB::ParseBlocksRecursive(Function &func) {
435656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
4367ac1c780SAaron Smith   size_t num_added = 0;
437ffc1b8fdSZachary Turner   auto uid = func.GetID();
4387ac1c780SAaron Smith   auto pdb_func_up = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(uid);
4397ac1c780SAaron Smith   if (!pdb_func_up)
4407ac1c780SAaron Smith     return 0;
441ffc1b8fdSZachary Turner   Block &parent_block = func.GetBlock(false);
442ffc1b8fdSZachary Turner   num_added = ParseFunctionBlocksForPDBSymbol(
443ffc1b8fdSZachary Turner       pdb_func_up->getVirtualAddress(), pdb_func_up.get(), &parent_block, true);
4447ac1c780SAaron Smith   return num_added;
445b9c1b51eSKate Stone }
446b9c1b51eSKate Stone 
447863f8c18SZachary Turner size_t SymbolFilePDB::ParseTypes(CompileUnit &comp_unit) {
448656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
44966b84079SAaron Smith 
45066b84079SAaron Smith   size_t num_added = 0;
451ac0d41c7SZachary Turner   auto compiland = GetPDBCompilandByUID(comp_unit.GetID());
45266b84079SAaron Smith   if (!compiland)
45366b84079SAaron Smith     return 0;
45466b84079SAaron Smith 
45566b84079SAaron Smith   auto ParseTypesByTagFn = [&num_added, this](const PDBSymbol &raw_sym) {
45666b84079SAaron Smith     std::unique_ptr<IPDBEnumSymbols> results;
45766b84079SAaron Smith     PDB_SymType tags_to_search[] = {PDB_SymType::Enum, PDB_SymType::Typedef,
45866b84079SAaron Smith                                     PDB_SymType::UDT};
45966b84079SAaron Smith     for (auto tag : tags_to_search) {
46066b84079SAaron Smith       results = raw_sym.findAllChildren(tag);
46166b84079SAaron Smith       if (!results || results->getChildCount() == 0)
46266b84079SAaron Smith         continue;
46366b84079SAaron Smith       while (auto symbol = results->getNext()) {
46466b84079SAaron Smith         switch (symbol->getSymTag()) {
465ec40f818SAaron Smith         case PDB_SymType::Enum:
466ec40f818SAaron Smith         case PDB_SymType::UDT:
467ec40f818SAaron Smith         case PDB_SymType::Typedef:
468ec40f818SAaron Smith           break;
469ec40f818SAaron Smith         default:
470ec40f818SAaron Smith           continue;
471ec40f818SAaron Smith         }
472ec40f818SAaron Smith 
473ec40f818SAaron Smith         // This should cause the type to get cached and stored in the `m_types`
474ec40f818SAaron Smith         // lookup.
4757d2a74fcSAleksandr Urakov         if (auto type = ResolveTypeUID(symbol->getSymIndexId())) {
4767d2a74fcSAleksandr Urakov           // Resolve the type completely to avoid a completion
4777d2a74fcSAleksandr Urakov           // (and so a list change, which causes an iterators invalidation)
4787d2a74fcSAleksandr Urakov           // during a TypeList dumping
4797d2a74fcSAleksandr Urakov           type->GetFullCompilerType();
480ec40f818SAaron Smith           ++num_added;
481ec40f818SAaron Smith         }
48266b84079SAaron Smith       }
4837d2a74fcSAleksandr Urakov     }
48466b84079SAaron Smith   };
48566b84079SAaron Smith 
48666b84079SAaron Smith   ParseTypesByTagFn(*compiland);
48766b84079SAaron Smith 
48866b84079SAaron Smith   // Also parse global types particularly coming from this compiland.
48905097246SAdrian Prantl   // Unfortunately, PDB has no compiland information for each global type. We
49005097246SAdrian Prantl   // have to parse them all. But ensure we only do this once.
49166b84079SAaron Smith   static bool parse_all_global_types = false;
49266b84079SAaron Smith   if (!parse_all_global_types) {
49366b84079SAaron Smith     ParseTypesByTagFn(*m_global_scope_up);
49466b84079SAaron Smith     parse_all_global_types = true;
49566b84079SAaron Smith   }
496ec40f818SAaron Smith   return num_added;
49774e08ca0SZachary Turner }
49874e08ca0SZachary Turner 
49974e08ca0SZachary Turner size_t
500b9c1b51eSKate Stone SymbolFilePDB::ParseVariablesForContext(const lldb_private::SymbolContext &sc) {
501656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
502cab0d23fSAaron Smith   if (!sc.comp_unit)
503cab0d23fSAaron Smith     return 0;
504cab0d23fSAaron Smith 
505cab0d23fSAaron Smith   size_t num_added = 0;
506cab0d23fSAaron Smith   if (sc.function) {
507cab0d23fSAaron Smith     auto pdb_func = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(
508cab0d23fSAaron Smith         sc.function->GetID());
509cab0d23fSAaron Smith     if (!pdb_func)
510cab0d23fSAaron Smith       return 0;
511cab0d23fSAaron Smith 
512cab0d23fSAaron Smith     num_added += ParseVariables(sc, *pdb_func);
513cab0d23fSAaron Smith     sc.function->GetBlock(false).SetDidParseVariables(true, true);
514cab0d23fSAaron Smith   } else if (sc.comp_unit) {
515cab0d23fSAaron Smith     auto compiland = GetPDBCompilandByUID(sc.comp_unit->GetID());
516cab0d23fSAaron Smith     if (!compiland)
517cab0d23fSAaron Smith       return 0;
518cab0d23fSAaron Smith 
519cab0d23fSAaron Smith     if (sc.comp_unit->GetVariableList(false))
520cab0d23fSAaron Smith       return 0;
521cab0d23fSAaron Smith 
522cab0d23fSAaron Smith     auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
523cab0d23fSAaron Smith     if (results && results->getChildCount()) {
524cab0d23fSAaron Smith       while (auto result = results->getNext()) {
525356aa4a9SAleksandr Urakov         auto cu_id = GetCompilandId(*result);
526cab0d23fSAaron Smith         // FIXME: We are not able to determine variable's compile unit.
527cab0d23fSAaron Smith         if (cu_id == 0)
528cab0d23fSAaron Smith           continue;
529cab0d23fSAaron Smith 
530cab0d23fSAaron Smith         if (cu_id == sc.comp_unit->GetID())
531cab0d23fSAaron Smith           num_added += ParseVariables(sc, *result);
532cab0d23fSAaron Smith       }
533cab0d23fSAaron Smith     }
534cab0d23fSAaron Smith 
535cab0d23fSAaron Smith     // FIXME: A `file static` or `global constant` variable appears both in
536cab0d23fSAaron Smith     // compiland's children and global scope's children with unexpectedly
537cab0d23fSAaron Smith     // different symbol's Id making it ambiguous.
538cab0d23fSAaron Smith 
539cab0d23fSAaron Smith     // FIXME: 'local constant', for example, const char var[] = "abc", declared
540cab0d23fSAaron Smith     // in a function scope, can't be found in PDB.
541cab0d23fSAaron Smith 
542cab0d23fSAaron Smith     // Parse variables in this compiland.
543cab0d23fSAaron Smith     num_added += ParseVariables(sc, *compiland);
544cab0d23fSAaron Smith   }
545cab0d23fSAaron Smith 
546cab0d23fSAaron Smith   return num_added;
54774e08ca0SZachary Turner }
54874e08ca0SZachary Turner 
549b9c1b51eSKate Stone lldb_private::Type *SymbolFilePDB::ResolveTypeUID(lldb::user_id_t type_uid) {
550656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
55142dff790SZachary Turner   auto find_result = m_types.find(type_uid);
55242dff790SZachary Turner   if (find_result != m_types.end())
55342dff790SZachary Turner     return find_result->second.get();
55442dff790SZachary Turner 
5550e252e38SAlex Langford   auto type_system_or_err =
556b9c1b51eSKate Stone       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
5570e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
5580e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
5590e252e38SAlex Langford                    std::move(err), "Unable to ResolveTypeUID");
5600e252e38SAlex Langford     return nullptr;
5610e252e38SAlex Langford   }
5620e252e38SAlex Langford 
5636e3b0cc2SRaphael Isemann   TypeSystemClang *clang_type_system =
5646e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
56542dff790SZachary Turner   if (!clang_type_system)
56674e08ca0SZachary Turner     return nullptr;
567709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_type_system->GetPDBParser();
56842dff790SZachary Turner   if (!pdb)
56942dff790SZachary Turner     return nullptr;
57042dff790SZachary Turner 
57142dff790SZachary Turner   auto pdb_type = m_session_up->getSymbolById(type_uid);
57242dff790SZachary Turner   if (pdb_type == nullptr)
57342dff790SZachary Turner     return nullptr;
57442dff790SZachary Turner 
57542dff790SZachary Turner   lldb::TypeSP result = pdb->CreateLLDBTypeFromPDBType(*pdb_type);
576d5a925f4SAaron Smith   if (result) {
57742dff790SZachary Turner     m_types.insert(std::make_pair(type_uid, result));
578f46e8974SPavel Labath     GetTypeList().Insert(result);
579ec40f818SAaron Smith   }
58042dff790SZachary Turner   return result.get();
58174e08ca0SZachary Turner }
58274e08ca0SZachary Turner 
583eca07c59SAdrian Prantl llvm::Optional<SymbolFile::ArrayInfo> SymbolFilePDB::GetDynamicArrayInfoForUID(
584eca07c59SAdrian Prantl     lldb::user_id_t type_uid, const lldb_private::ExecutionContext *exe_ctx) {
585eca07c59SAdrian Prantl   return llvm::None;
586eca07c59SAdrian Prantl }
587eca07c59SAdrian Prantl 
588b9c1b51eSKate Stone bool SymbolFilePDB::CompleteType(lldb_private::CompilerType &compiler_type) {
5897d2a74fcSAleksandr Urakov   std::lock_guard<std::recursive_mutex> guard(
5907d2a74fcSAleksandr Urakov       GetObjectFile()->GetModule()->GetMutex());
5917d2a74fcSAleksandr Urakov 
5920e252e38SAlex Langford   auto type_system_or_err =
5930e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
5940e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
5950e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
5960e252e38SAlex Langford                    std::move(err), "Unable to get dynamic array info for UID");
5970e252e38SAlex Langford     return false;
5980e252e38SAlex Langford   }
5990e252e38SAlex Langford 
6006e3b0cc2SRaphael Isemann   TypeSystemClang *clang_ast_ctx =
6016e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
6020e252e38SAlex Langford 
6037d2a74fcSAleksandr Urakov   if (!clang_ast_ctx)
60474e08ca0SZachary Turner     return false;
6057d2a74fcSAleksandr Urakov 
606709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
6077d2a74fcSAleksandr Urakov   if (!pdb)
6087d2a74fcSAleksandr Urakov     return false;
6097d2a74fcSAleksandr Urakov 
6107d2a74fcSAleksandr Urakov   return pdb->CompleteTypeFromPDB(compiler_type);
61174e08ca0SZachary Turner }
61274e08ca0SZachary Turner 
613b9c1b51eSKate Stone lldb_private::CompilerDecl SymbolFilePDB::GetDeclForUID(lldb::user_id_t uid) {
6140e252e38SAlex Langford   auto type_system_or_err =
6150e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
6160e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
6170e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
6180e252e38SAlex Langford                    std::move(err), "Unable to get decl for UID");
6190e252e38SAlex Langford     return CompilerDecl();
6200e252e38SAlex Langford   }
6210e252e38SAlex Langford 
6226e3b0cc2SRaphael Isemann   TypeSystemClang *clang_ast_ctx =
6236e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
624709426b3SAleksandr Urakov   if (!clang_ast_ctx)
625709426b3SAleksandr Urakov     return CompilerDecl();
626709426b3SAleksandr Urakov 
627709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
628709426b3SAleksandr Urakov   if (!pdb)
629709426b3SAleksandr Urakov     return CompilerDecl();
630709426b3SAleksandr Urakov 
631709426b3SAleksandr Urakov   auto symbol = m_session_up->getSymbolById(uid);
632709426b3SAleksandr Urakov   if (!symbol)
633709426b3SAleksandr Urakov     return CompilerDecl();
634709426b3SAleksandr Urakov 
635709426b3SAleksandr Urakov   auto decl = pdb->GetDeclForSymbol(*symbol);
636709426b3SAleksandr Urakov   if (!decl)
637709426b3SAleksandr Urakov     return CompilerDecl();
638709426b3SAleksandr Urakov 
639709426b3SAleksandr Urakov   return CompilerDecl(clang_ast_ctx, decl);
64074e08ca0SZachary Turner }
64174e08ca0SZachary Turner 
64274e08ca0SZachary Turner lldb_private::CompilerDeclContext
643b9c1b51eSKate Stone SymbolFilePDB::GetDeclContextForUID(lldb::user_id_t uid) {
6440e252e38SAlex Langford   auto type_system_or_err =
6450e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
6460e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
6470e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
6480e252e38SAlex Langford                    std::move(err), "Unable to get DeclContext for UID");
6490e252e38SAlex Langford     return CompilerDeclContext();
6500e252e38SAlex Langford   }
6510e252e38SAlex Langford 
6526e3b0cc2SRaphael Isemann   TypeSystemClang *clang_ast_ctx =
6536e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
654709426b3SAleksandr Urakov   if (!clang_ast_ctx)
655709426b3SAleksandr Urakov     return CompilerDeclContext();
656709426b3SAleksandr Urakov 
657709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
658709426b3SAleksandr Urakov   if (!pdb)
659709426b3SAleksandr Urakov     return CompilerDeclContext();
660709426b3SAleksandr Urakov 
661709426b3SAleksandr Urakov   auto symbol = m_session_up->getSymbolById(uid);
662709426b3SAleksandr Urakov   if (!symbol)
663709426b3SAleksandr Urakov     return CompilerDeclContext();
664709426b3SAleksandr Urakov 
665709426b3SAleksandr Urakov   auto decl_context = pdb->GetDeclContextForSymbol(*symbol);
666709426b3SAleksandr Urakov   if (!decl_context)
667709426b3SAleksandr Urakov     return GetDeclContextContainingUID(uid);
668709426b3SAleksandr Urakov 
66942ec584aSRaphael Isemann   return clang_ast_ctx->CreateDeclContext(decl_context);
67074e08ca0SZachary Turner }
67174e08ca0SZachary Turner 
67274e08ca0SZachary Turner lldb_private::CompilerDeclContext
673b9c1b51eSKate Stone SymbolFilePDB::GetDeclContextContainingUID(lldb::user_id_t uid) {
6740e252e38SAlex Langford   auto type_system_or_err =
6750e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
6760e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
6770e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
6780e252e38SAlex Langford                    std::move(err), "Unable to get DeclContext containing UID");
6790e252e38SAlex Langford     return CompilerDeclContext();
6800e252e38SAlex Langford   }
6810e252e38SAlex Langford 
6826e3b0cc2SRaphael Isemann   TypeSystemClang *clang_ast_ctx =
6836e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
684709426b3SAleksandr Urakov   if (!clang_ast_ctx)
685709426b3SAleksandr Urakov     return CompilerDeclContext();
686709426b3SAleksandr Urakov 
687709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
688709426b3SAleksandr Urakov   if (!pdb)
689709426b3SAleksandr Urakov     return CompilerDeclContext();
690709426b3SAleksandr Urakov 
691709426b3SAleksandr Urakov   auto symbol = m_session_up->getSymbolById(uid);
692709426b3SAleksandr Urakov   if (!symbol)
693709426b3SAleksandr Urakov     return CompilerDeclContext();
694709426b3SAleksandr Urakov 
695709426b3SAleksandr Urakov   auto decl_context = pdb->GetDeclContextContainingSymbol(*symbol);
696709426b3SAleksandr Urakov   assert(decl_context);
697709426b3SAleksandr Urakov 
69842ec584aSRaphael Isemann   return clang_ast_ctx->CreateDeclContext(decl_context);
69974e08ca0SZachary Turner }
70074e08ca0SZachary Turner 
701b9c1b51eSKate Stone void SymbolFilePDB::ParseDeclsForContext(
702709426b3SAleksandr Urakov     lldb_private::CompilerDeclContext decl_ctx) {
7030e252e38SAlex Langford   auto type_system_or_err =
7040e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
7050e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
7060e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
7070e252e38SAlex Langford                    std::move(err), "Unable to parse decls for context");
7080e252e38SAlex Langford     return;
7090e252e38SAlex Langford   }
7100e252e38SAlex Langford 
7116e3b0cc2SRaphael Isemann   TypeSystemClang *clang_ast_ctx =
7126e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
713709426b3SAleksandr Urakov   if (!clang_ast_ctx)
714709426b3SAleksandr Urakov     return;
715709426b3SAleksandr Urakov 
716709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
717709426b3SAleksandr Urakov   if (!pdb)
718709426b3SAleksandr Urakov     return;
719709426b3SAleksandr Urakov 
720709426b3SAleksandr Urakov   pdb->ParseDeclsForDeclContext(
721709426b3SAleksandr Urakov       static_cast<clang::DeclContext *>(decl_ctx.GetOpaqueDeclContext()));
722709426b3SAleksandr Urakov }
72374e08ca0SZachary Turner 
72474e08ca0SZachary Turner uint32_t
725b9c1b51eSKate Stone SymbolFilePDB::ResolveSymbolContext(const lldb_private::Address &so_addr,
726991e4453SZachary Turner                                     SymbolContextItem resolve_scope,
727b9c1b51eSKate Stone                                     lldb_private::SymbolContext &sc) {
728656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
7297ac1c780SAaron Smith   uint32_t resolved_flags = 0;
7304d4d63eeSPavel Labath   if (resolve_scope & eSymbolContextCompUnit ||
7314d4d63eeSPavel Labath       resolve_scope & eSymbolContextVariable ||
7324d4d63eeSPavel Labath       resolve_scope & eSymbolContextFunction ||
7334d4d63eeSPavel Labath       resolve_scope & eSymbolContextBlock ||
7347ac1c780SAaron Smith       resolve_scope & eSymbolContextLineEntry) {
7357ac1c780SAaron Smith     auto cu_sp = GetCompileUnitContainsAddress(so_addr);
7367ac1c780SAaron Smith     if (!cu_sp) {
73733cdbff2SRichard Trieu       if (resolved_flags & eSymbolContextVariable) {
7387ac1c780SAaron Smith         // TODO: Resolve variables
7397ac1c780SAaron Smith       }
7407ac1c780SAaron Smith       return 0;
7417ac1c780SAaron Smith     }
7427ac1c780SAaron Smith     sc.comp_unit = cu_sp.get();
7437ac1c780SAaron Smith     resolved_flags |= eSymbolContextCompUnit;
7447ac1c780SAaron Smith     lldbassert(sc.module_sp == cu_sp->GetModule());
7459ea80d25SPavel Labath   }
7467ac1c780SAaron Smith 
747398f81b3SAleksandr Urakov   if (resolve_scope & eSymbolContextFunction ||
748398f81b3SAleksandr Urakov       resolve_scope & eSymbolContextBlock) {
7499ea80d25SPavel Labath     addr_t file_vm_addr = so_addr.GetFileAddress();
7509ea80d25SPavel Labath     auto symbol_up =
7519ea80d25SPavel Labath         m_session_up->findSymbolByAddress(file_vm_addr, PDB_SymType::Function);
7529ea80d25SPavel Labath     if (symbol_up) {
7537ac1c780SAaron Smith       auto *pdb_func = llvm::dyn_cast<PDBSymbolFunc>(symbol_up.get());
7547ac1c780SAaron Smith       assert(pdb_func);
7557ac1c780SAaron Smith       auto func_uid = pdb_func->getSymIndexId();
7567ac1c780SAaron Smith       sc.function = sc.comp_unit->FindFunctionByUID(func_uid).get();
7577ac1c780SAaron Smith       if (sc.function == nullptr)
758863f8c18SZachary Turner         sc.function =
759863f8c18SZachary Turner             ParseCompileUnitFunctionForPDBFunc(*pdb_func, *sc.comp_unit);
7607ac1c780SAaron Smith       if (sc.function) {
7617ac1c780SAaron Smith         resolved_flags |= eSymbolContextFunction;
7627ac1c780SAaron Smith         if (resolve_scope & eSymbolContextBlock) {
763398f81b3SAleksandr Urakov           auto block_symbol = m_session_up->findSymbolByAddress(
764398f81b3SAleksandr Urakov               file_vm_addr, PDB_SymType::Block);
765398f81b3SAleksandr Urakov           auto block_id = block_symbol ? block_symbol->getSymIndexId()
766398f81b3SAleksandr Urakov                                        : sc.function->GetID();
767398f81b3SAleksandr Urakov           sc.block = sc.function->GetBlock(true).FindBlockByID(block_id);
7687ac1c780SAaron Smith           if (sc.block)
7697ac1c780SAaron Smith             resolved_flags |= eSymbolContextBlock;
7707ac1c780SAaron Smith         }
7717ac1c780SAaron Smith       }
7727ac1c780SAaron Smith     }
7737ac1c780SAaron Smith   }
7747ac1c780SAaron Smith 
7757ac1c780SAaron Smith   if (resolve_scope & eSymbolContextLineEntry) {
7767ac1c780SAaron Smith     if (auto *line_table = sc.comp_unit->GetLineTable()) {
7777ac1c780SAaron Smith       Address addr(so_addr);
7787ac1c780SAaron Smith       if (line_table->FindLineEntryByAddress(addr, sc.line_entry))
7797ac1c780SAaron Smith         resolved_flags |= eSymbolContextLineEntry;
7807ac1c780SAaron Smith     }
7817ac1c780SAaron Smith   }
7829ea80d25SPavel Labath 
7837ac1c780SAaron Smith   return resolved_flags;
78474e08ca0SZachary Turner }
78574e08ca0SZachary Turner 
786b9c1b51eSKate Stone uint32_t SymbolFilePDB::ResolveSymbolContext(
787b9c1b51eSKate Stone     const lldb_private::FileSpec &file_spec, uint32_t line, bool check_inlines,
788991e4453SZachary Turner     SymbolContextItem resolve_scope, lldb_private::SymbolContextList &sc_list) {
789656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
79010a02577SAaron Smith   const size_t old_size = sc_list.GetSize();
791b9c1b51eSKate Stone   if (resolve_scope & lldb::eSymbolContextCompUnit) {
792b9c1b51eSKate Stone     // Locate all compilation units with line numbers referencing the specified
7939d0eb996SAdrian McCarthy     // file.  For example, if `file_spec` is <vector>, then this should return
7949d0eb996SAdrian McCarthy     // all source files and header files that reference <vector>, either
7959d0eb996SAdrian McCarthy     // directly or indirectly.
796b9c1b51eSKate Stone     auto compilands = m_session_up->findCompilandsForSourceFile(
797b9c1b51eSKate Stone         file_spec.GetPath(), PDB_NameSearchFlags::NS_CaseInsensitive);
79874e08ca0SZachary Turner 
79910a02577SAaron Smith     if (!compilands)
80010a02577SAaron Smith       return 0;
80110a02577SAaron Smith 
8029d0eb996SAdrian McCarthy     // For each one, either find its previously parsed data or parse it afresh
8039d0eb996SAdrian McCarthy     // and add it to the symbol context list.
804b9c1b51eSKate Stone     while (auto compiland = compilands->getNext()) {
80505097246SAdrian Prantl       // If we're not checking inlines, then don't add line information for
80605097246SAdrian Prantl       // this file unless the FileSpec matches. For inline functions, we don't
80705097246SAdrian Prantl       // have to match the FileSpec since they could be defined in headers
80805097246SAdrian Prantl       // other than file specified in FileSpec.
809b9c1b51eSKate Stone       if (!check_inlines) {
810487b0c6bSAaron Smith         std::string source_file = compiland->getSourceFileFullPath();
81110a02577SAaron Smith         if (source_file.empty())
81210a02577SAaron Smith           continue;
8138f3be7a3SJonas Devlieghere         FileSpec this_spec(source_file, FileSpec::Style::windows);
81410a02577SAaron Smith         bool need_full_match = !file_spec.GetDirectory().IsEmpty();
81510a02577SAaron Smith         if (FileSpec::Compare(file_spec, this_spec, need_full_match) != 0)
81674e08ca0SZachary Turner           continue;
81774e08ca0SZachary Turner       }
81874e08ca0SZachary Turner 
81974e08ca0SZachary Turner       SymbolContext sc;
82010a02577SAaron Smith       auto cu = ParseCompileUnitForUID(compiland->getSymIndexId());
821d5a925f4SAaron Smith       if (!cu)
82210a02577SAaron Smith         continue;
82374e08ca0SZachary Turner       sc.comp_unit = cu.get();
82474e08ca0SZachary Turner       sc.module_sp = cu->GetModule();
82574e08ca0SZachary Turner 
826b9c1b51eSKate Stone       // If we were asked to resolve line entries, add all entries to the line
8279d0eb996SAdrian McCarthy       // table that match the requested line (or all lines if `line` == 0).
8287ac1c780SAaron Smith       if (resolve_scope & (eSymbolContextFunction | eSymbolContextBlock |
8297ac1c780SAaron Smith                            eSymbolContextLineEntry)) {
830863f8c18SZachary Turner         bool has_line_table = ParseCompileUnitLineTable(*sc.comp_unit, line);
8317ac1c780SAaron Smith 
8327ac1c780SAaron Smith         if ((resolve_scope & eSymbolContextLineEntry) && !has_line_table) {
8337ac1c780SAaron Smith           // The query asks for line entries, but we can't get them for the
83405097246SAdrian Prantl           // compile unit. This is not normal for `line` = 0. So just assert
83505097246SAdrian Prantl           // it.
836f76fe682SAaron Smith           assert(line && "Couldn't get all line entries!\n");
8377ac1c780SAaron Smith 
8387ac1c780SAaron Smith           // Current compiland does not have the requested line. Search next.
8397ac1c780SAaron Smith           continue;
8407ac1c780SAaron Smith         }
8417ac1c780SAaron Smith 
8427ac1c780SAaron Smith         if (resolve_scope & (eSymbolContextFunction | eSymbolContextBlock)) {
8437ac1c780SAaron Smith           if (!has_line_table)
8447ac1c780SAaron Smith             continue;
8457ac1c780SAaron Smith 
8467ac1c780SAaron Smith           auto *line_table = sc.comp_unit->GetLineTable();
8477ac1c780SAaron Smith           lldbassert(line_table);
8487ac1c780SAaron Smith 
8497ac1c780SAaron Smith           uint32_t num_line_entries = line_table->GetSize();
8507ac1c780SAaron Smith           // Skip the terminal line entry.
8517ac1c780SAaron Smith           --num_line_entries;
8527ac1c780SAaron Smith 
85305097246SAdrian Prantl           // If `line `!= 0, see if we can resolve function for each line entry
85405097246SAdrian Prantl           // in the line table.
8557ac1c780SAaron Smith           for (uint32_t line_idx = 0; line && line_idx < num_line_entries;
8567ac1c780SAaron Smith                ++line_idx) {
8577ac1c780SAaron Smith             if (!line_table->GetLineEntryAtIndex(line_idx, sc.line_entry))
8587ac1c780SAaron Smith               continue;
8597ac1c780SAaron Smith 
8607ac1c780SAaron Smith             auto file_vm_addr =
8617ac1c780SAaron Smith                 sc.line_entry.range.GetBaseAddress().GetFileAddress();
862308e39caSAaron Smith             if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
8637ac1c780SAaron Smith               continue;
8647ac1c780SAaron Smith 
865c8316ed2SAaron Smith             auto symbol_up = m_session_up->findSymbolByAddress(
866c8316ed2SAaron Smith                 file_vm_addr, PDB_SymType::Function);
8677ac1c780SAaron Smith             if (symbol_up) {
8687ac1c780SAaron Smith               auto func_uid = symbol_up->getSymIndexId();
8697ac1c780SAaron Smith               sc.function = sc.comp_unit->FindFunctionByUID(func_uid).get();
8707ac1c780SAaron Smith               if (sc.function == nullptr) {
8717ac1c780SAaron Smith                 auto pdb_func = llvm::dyn_cast<PDBSymbolFunc>(symbol_up.get());
8727ac1c780SAaron Smith                 assert(pdb_func);
873863f8c18SZachary Turner                 sc.function = ParseCompileUnitFunctionForPDBFunc(*pdb_func,
874863f8c18SZachary Turner                                                                  *sc.comp_unit);
8757ac1c780SAaron Smith               }
8767ac1c780SAaron Smith               if (sc.function && (resolve_scope & eSymbolContextBlock)) {
8777ac1c780SAaron Smith                 Block &block = sc.function->GetBlock(true);
8787ac1c780SAaron Smith                 sc.block = block.FindBlockByID(sc.function->GetID());
8797ac1c780SAaron Smith               }
8807ac1c780SAaron Smith             }
8817ac1c780SAaron Smith             sc_list.Append(sc);
8827ac1c780SAaron Smith           }
8837ac1c780SAaron Smith         } else if (has_line_table) {
8847ac1c780SAaron Smith           // We can parse line table for the compile unit. But no query to
8857ac1c780SAaron Smith           // resolve function or block. We append `sc` to the list anyway.
8867ac1c780SAaron Smith           sc_list.Append(sc);
8877ac1c780SAaron Smith         }
8887ac1c780SAaron Smith       } else {
8897ac1c780SAaron Smith         // No query for line entry, function or block. But we have a valid
8907ac1c780SAaron Smith         // compile unit, append `sc` to the list.
8917ac1c780SAaron Smith         sc_list.Append(sc);
8927ac1c780SAaron Smith       }
89374e08ca0SZachary Turner     }
89474e08ca0SZachary Turner   }
89510a02577SAaron Smith   return sc_list.GetSize() - old_size;
89674e08ca0SZachary Turner }
89774e08ca0SZachary Turner 
898cab0d23fSAaron Smith std::string SymbolFilePDB::GetMangledForPDBData(const PDBSymbolData &pdb_data) {
899356aa4a9SAleksandr Urakov   // Cache public names at first
900356aa4a9SAleksandr Urakov   if (m_public_names.empty())
901356aa4a9SAleksandr Urakov     if (auto result_up =
902356aa4a9SAleksandr Urakov             m_global_scope_up->findAllChildren(PDB_SymType::PublicSymbol))
903356aa4a9SAleksandr Urakov       while (auto symbol_up = result_up->getNext())
904356aa4a9SAleksandr Urakov         if (auto addr = symbol_up->getRawSymbol().getVirtualAddress())
905356aa4a9SAleksandr Urakov           m_public_names[addr] = symbol_up->getRawSymbol().getName();
906cab0d23fSAaron Smith 
907356aa4a9SAleksandr Urakov   // Look up the name in the cache
908356aa4a9SAleksandr Urakov   return m_public_names.lookup(pdb_data.getVirtualAddress());
909cab0d23fSAaron Smith }
910cab0d23fSAaron Smith 
911cab0d23fSAaron Smith VariableSP SymbolFilePDB::ParseVariableForPDBData(
912cab0d23fSAaron Smith     const lldb_private::SymbolContext &sc,
913cab0d23fSAaron Smith     const llvm::pdb::PDBSymbolData &pdb_data) {
914cab0d23fSAaron Smith   VariableSP var_sp;
915cab0d23fSAaron Smith   uint32_t var_uid = pdb_data.getSymIndexId();
916cab0d23fSAaron Smith   auto result = m_variables.find(var_uid);
917cab0d23fSAaron Smith   if (result != m_variables.end())
918cab0d23fSAaron Smith     return result->second;
919cab0d23fSAaron Smith 
920cab0d23fSAaron Smith   ValueType scope = eValueTypeInvalid;
921cab0d23fSAaron Smith   bool is_static_member = false;
922cab0d23fSAaron Smith   bool is_external = false;
923cab0d23fSAaron Smith   bool is_artificial = false;
924cab0d23fSAaron Smith 
925cab0d23fSAaron Smith   switch (pdb_data.getDataKind()) {
926cab0d23fSAaron Smith   case PDB_DataKind::Global:
927cab0d23fSAaron Smith     scope = eValueTypeVariableGlobal;
928cab0d23fSAaron Smith     is_external = true;
929cab0d23fSAaron Smith     break;
930cab0d23fSAaron Smith   case PDB_DataKind::Local:
931cab0d23fSAaron Smith     scope = eValueTypeVariableLocal;
932cab0d23fSAaron Smith     break;
933cab0d23fSAaron Smith   case PDB_DataKind::FileStatic:
934cab0d23fSAaron Smith     scope = eValueTypeVariableStatic;
935cab0d23fSAaron Smith     break;
936cab0d23fSAaron Smith   case PDB_DataKind::StaticMember:
937cab0d23fSAaron Smith     is_static_member = true;
938cab0d23fSAaron Smith     scope = eValueTypeVariableStatic;
939cab0d23fSAaron Smith     break;
940cab0d23fSAaron Smith   case PDB_DataKind::Member:
941cab0d23fSAaron Smith     scope = eValueTypeVariableStatic;
942cab0d23fSAaron Smith     break;
943cab0d23fSAaron Smith   case PDB_DataKind::Param:
944cab0d23fSAaron Smith     scope = eValueTypeVariableArgument;
945cab0d23fSAaron Smith     break;
946cab0d23fSAaron Smith   case PDB_DataKind::Constant:
947cab0d23fSAaron Smith     scope = eValueTypeConstResult;
948cab0d23fSAaron Smith     break;
949cab0d23fSAaron Smith   default:
950cab0d23fSAaron Smith     break;
951cab0d23fSAaron Smith   }
952cab0d23fSAaron Smith 
953cab0d23fSAaron Smith   switch (pdb_data.getLocationType()) {
954cab0d23fSAaron Smith   case PDB_LocType::TLS:
955cab0d23fSAaron Smith     scope = eValueTypeVariableThreadLocal;
956cab0d23fSAaron Smith     break;
957cab0d23fSAaron Smith   case PDB_LocType::RegRel: {
958cab0d23fSAaron Smith     // It is a `this` pointer.
959cab0d23fSAaron Smith     if (pdb_data.getDataKind() == PDB_DataKind::ObjectPtr) {
960cab0d23fSAaron Smith       scope = eValueTypeVariableArgument;
961cab0d23fSAaron Smith       is_artificial = true;
962cab0d23fSAaron Smith     }
963cab0d23fSAaron Smith   } break;
964cab0d23fSAaron Smith   default:
965cab0d23fSAaron Smith     break;
966cab0d23fSAaron Smith   }
967cab0d23fSAaron Smith 
968cab0d23fSAaron Smith   Declaration decl;
969cab0d23fSAaron Smith   if (!is_artificial && !pdb_data.isCompilerGenerated()) {
970cab0d23fSAaron Smith     if (auto lines = pdb_data.getLineNumbers()) {
971cab0d23fSAaron Smith       if (auto first_line = lines->getNext()) {
972cab0d23fSAaron Smith         uint32_t src_file_id = first_line->getSourceFileId();
973cab0d23fSAaron Smith         auto src_file = m_session_up->getSourceFileById(src_file_id);
974cab0d23fSAaron Smith         if (src_file) {
9758f3be7a3SJonas Devlieghere           FileSpec spec(src_file->getFileName());
976cab0d23fSAaron Smith           decl.SetFile(spec);
977cab0d23fSAaron Smith           decl.SetColumn(first_line->getColumnNumber());
978cab0d23fSAaron Smith           decl.SetLine(first_line->getLineNumber());
979cab0d23fSAaron Smith         }
980cab0d23fSAaron Smith       }
981cab0d23fSAaron Smith     }
982cab0d23fSAaron Smith   }
983cab0d23fSAaron Smith 
984cab0d23fSAaron Smith   Variable::RangeList ranges;
985cab0d23fSAaron Smith   SymbolContextScope *context_scope = sc.comp_unit;
986758657e5SAleksandr Urakov   if (scope == eValueTypeVariableLocal || scope == eValueTypeVariableArgument) {
987cab0d23fSAaron Smith     if (sc.function) {
988758657e5SAleksandr Urakov       Block &function_block = sc.function->GetBlock(true);
989758657e5SAleksandr Urakov       Block *block =
990758657e5SAleksandr Urakov           function_block.FindBlockByID(pdb_data.getLexicalParentId());
991758657e5SAleksandr Urakov       if (!block)
992758657e5SAleksandr Urakov         block = &function_block;
993758657e5SAleksandr Urakov 
994758657e5SAleksandr Urakov       context_scope = block;
995758657e5SAleksandr Urakov 
996758657e5SAleksandr Urakov       for (size_t i = 0, num_ranges = block->GetNumRanges(); i < num_ranges;
997758657e5SAleksandr Urakov            ++i) {
998758657e5SAleksandr Urakov         AddressRange range;
999758657e5SAleksandr Urakov         if (!block->GetRangeAtIndex(i, range))
1000758657e5SAleksandr Urakov           continue;
1001758657e5SAleksandr Urakov 
1002758657e5SAleksandr Urakov         ranges.Append(range.GetBaseAddress().GetFileAddress(),
1003758657e5SAleksandr Urakov                       range.GetByteSize());
1004758657e5SAleksandr Urakov       }
1005cab0d23fSAaron Smith     }
1006cab0d23fSAaron Smith   }
1007cab0d23fSAaron Smith 
1008cab0d23fSAaron Smith   SymbolFileTypeSP type_sp =
1009cab0d23fSAaron Smith       std::make_shared<SymbolFileType>(*this, pdb_data.getTypeId());
1010cab0d23fSAaron Smith 
1011cab0d23fSAaron Smith   auto var_name = pdb_data.getName();
1012cab0d23fSAaron Smith   auto mangled = GetMangledForPDBData(pdb_data);
1013cab0d23fSAaron Smith   auto mangled_cstr = mangled.empty() ? nullptr : mangled.c_str();
1014cab0d23fSAaron Smith 
1015924d5608SJonas Devlieghere   bool is_constant;
1016924d5608SJonas Devlieghere   DWARFExpression location = ConvertPDBLocationToDWARFExpression(
1017758657e5SAleksandr Urakov       GetObjectFile()->GetModule(), pdb_data, ranges, is_constant);
1018cab0d23fSAaron Smith 
1019cab0d23fSAaron Smith   var_sp = std::make_shared<Variable>(
1020cab0d23fSAaron Smith       var_uid, var_name.c_str(), mangled_cstr, type_sp, scope, context_scope,
1021cab0d23fSAaron Smith       ranges, &decl, location, is_external, is_artificial, is_static_member);
1022924d5608SJonas Devlieghere   var_sp->SetLocationIsConstantValueData(is_constant);
1023cab0d23fSAaron Smith 
1024cab0d23fSAaron Smith   m_variables.insert(std::make_pair(var_uid, var_sp));
1025cab0d23fSAaron Smith   return var_sp;
1026cab0d23fSAaron Smith }
1027cab0d23fSAaron Smith 
1028cab0d23fSAaron Smith size_t
1029cab0d23fSAaron Smith SymbolFilePDB::ParseVariables(const lldb_private::SymbolContext &sc,
1030cab0d23fSAaron Smith                               const llvm::pdb::PDBSymbol &pdb_symbol,
1031cab0d23fSAaron Smith                               lldb_private::VariableList *variable_list) {
1032cab0d23fSAaron Smith   size_t num_added = 0;
1033cab0d23fSAaron Smith 
1034cab0d23fSAaron Smith   if (auto pdb_data = llvm::dyn_cast<PDBSymbolData>(&pdb_symbol)) {
1035cab0d23fSAaron Smith     VariableListSP local_variable_list_sp;
1036cab0d23fSAaron Smith 
1037cab0d23fSAaron Smith     auto result = m_variables.find(pdb_data->getSymIndexId());
1038cab0d23fSAaron Smith     if (result != m_variables.end()) {
1039cab0d23fSAaron Smith       if (variable_list)
1040cab0d23fSAaron Smith         variable_list->AddVariableIfUnique(result->second);
1041cab0d23fSAaron Smith     } else {
1042cab0d23fSAaron Smith       // Prepare right VariableList for this variable.
1043cab0d23fSAaron Smith       if (auto lexical_parent = pdb_data->getLexicalParent()) {
1044cab0d23fSAaron Smith         switch (lexical_parent->getSymTag()) {
1045cab0d23fSAaron Smith         case PDB_SymType::Exe:
1046cab0d23fSAaron Smith           assert(sc.comp_unit);
1047cab0d23fSAaron Smith           LLVM_FALLTHROUGH;
1048cab0d23fSAaron Smith         case PDB_SymType::Compiland: {
1049cab0d23fSAaron Smith           if (sc.comp_unit) {
1050cab0d23fSAaron Smith             local_variable_list_sp = sc.comp_unit->GetVariableList(false);
1051cab0d23fSAaron Smith             if (!local_variable_list_sp) {
1052cab0d23fSAaron Smith               local_variable_list_sp = std::make_shared<VariableList>();
1053cab0d23fSAaron Smith               sc.comp_unit->SetVariableList(local_variable_list_sp);
1054cab0d23fSAaron Smith             }
1055cab0d23fSAaron Smith           }
1056cab0d23fSAaron Smith         } break;
1057cab0d23fSAaron Smith         case PDB_SymType::Block:
1058cab0d23fSAaron Smith         case PDB_SymType::Function: {
1059cab0d23fSAaron Smith           if (sc.function) {
1060cab0d23fSAaron Smith             Block *block = sc.function->GetBlock(true).FindBlockByID(
1061cab0d23fSAaron Smith                 lexical_parent->getSymIndexId());
1062cab0d23fSAaron Smith             if (block) {
1063cab0d23fSAaron Smith               local_variable_list_sp = block->GetBlockVariableList(false);
1064cab0d23fSAaron Smith               if (!local_variable_list_sp) {
1065cab0d23fSAaron Smith                 local_variable_list_sp = std::make_shared<VariableList>();
1066cab0d23fSAaron Smith                 block->SetVariableList(local_variable_list_sp);
1067cab0d23fSAaron Smith               }
1068cab0d23fSAaron Smith             }
1069cab0d23fSAaron Smith           }
1070cab0d23fSAaron Smith         } break;
1071cab0d23fSAaron Smith         default:
1072cab0d23fSAaron Smith           break;
1073cab0d23fSAaron Smith         }
1074cab0d23fSAaron Smith       }
1075cab0d23fSAaron Smith 
1076cab0d23fSAaron Smith       if (local_variable_list_sp) {
1077cab0d23fSAaron Smith         if (auto var_sp = ParseVariableForPDBData(sc, *pdb_data)) {
1078cab0d23fSAaron Smith           local_variable_list_sp->AddVariableIfUnique(var_sp);
1079cab0d23fSAaron Smith           if (variable_list)
1080cab0d23fSAaron Smith             variable_list->AddVariableIfUnique(var_sp);
1081cab0d23fSAaron Smith           ++num_added;
1082c68925abSZachary Turner           PDBASTParser *ast = GetPDBAstParser();
1083c68925abSZachary Turner           if (ast)
1084c68925abSZachary Turner             ast->GetDeclForSymbol(*pdb_data);
1085cab0d23fSAaron Smith         }
1086cab0d23fSAaron Smith       }
1087cab0d23fSAaron Smith     }
1088cab0d23fSAaron Smith   }
1089cab0d23fSAaron Smith 
1090cab0d23fSAaron Smith   if (auto results = pdb_symbol.findAllChildren()) {
1091cab0d23fSAaron Smith     while (auto result = results->getNext())
1092cab0d23fSAaron Smith       num_added += ParseVariables(sc, *result, variable_list);
1093cab0d23fSAaron Smith   }
1094cab0d23fSAaron Smith 
1095cab0d23fSAaron Smith   return num_added;
1096cab0d23fSAaron Smith }
1097cab0d23fSAaron Smith 
10981ad655e2SAdrian Prantl void SymbolFilePDB::FindGlobalVariables(
10990e4c4821SAdrian Prantl     lldb_private::ConstString name,
110034cda14bSPavel Labath     const lldb_private::CompilerDeclContext *parent_decl_ctx,
1101b9c1b51eSKate Stone     uint32_t max_matches, lldb_private::VariableList &variables) {
1102656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1103cab0d23fSAaron Smith   if (!DeclContextMatchesThisSymbolFile(parent_decl_ctx))
11041ad655e2SAdrian Prantl     return;
1105cab0d23fSAaron Smith   if (name.IsEmpty())
11061ad655e2SAdrian Prantl     return;
1107cab0d23fSAaron Smith 
1108709426b3SAleksandr Urakov   auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
1109cab0d23fSAaron Smith   if (!results)
11101ad655e2SAdrian Prantl     return;
1111cab0d23fSAaron Smith 
1112cab0d23fSAaron Smith   uint32_t matches = 0;
1113cab0d23fSAaron Smith   size_t old_size = variables.GetSize();
1114cab0d23fSAaron Smith   while (auto result = results->getNext()) {
1115cab0d23fSAaron Smith     auto pdb_data = llvm::dyn_cast<PDBSymbolData>(result.get());
1116cab0d23fSAaron Smith     if (max_matches > 0 && matches >= max_matches)
1117cab0d23fSAaron Smith       break;
1118cab0d23fSAaron Smith 
1119cab0d23fSAaron Smith     SymbolContext sc;
1120d2deeb44SPavel Labath     sc.module_sp = m_objfile_sp->GetModule();
1121cab0d23fSAaron Smith     lldbassert(sc.module_sp.get());
1122cab0d23fSAaron Smith 
1123709426b3SAleksandr Urakov     if (!name.GetStringRef().equals(
1124c1e530eeSAleksandr Urakov             MSVCUndecoratedNameParser::DropScope(pdb_data->getName())))
1125709426b3SAleksandr Urakov       continue;
1126709426b3SAleksandr Urakov 
1127356aa4a9SAleksandr Urakov     sc.comp_unit = ParseCompileUnitForUID(GetCompilandId(*pdb_data)).get();
1128356aa4a9SAleksandr Urakov     // FIXME: We are not able to determine the compile unit.
1129356aa4a9SAleksandr Urakov     if (sc.comp_unit == nullptr)
1130356aa4a9SAleksandr Urakov       continue;
1131356aa4a9SAleksandr Urakov 
1132a5235af9SAleksandr Urakov     if (parent_decl_ctx && GetDeclContextContainingUID(
1133a5235af9SAleksandr Urakov                                result->getSymIndexId()) != *parent_decl_ctx)
1134709426b3SAleksandr Urakov       continue;
1135709426b3SAleksandr Urakov 
1136cab0d23fSAaron Smith     ParseVariables(sc, *pdb_data, &variables);
1137cab0d23fSAaron Smith     matches = variables.GetSize() - old_size;
1138cab0d23fSAaron Smith   }
113974e08ca0SZachary Turner }
114074e08ca0SZachary Turner 
11411ad655e2SAdrian Prantl void SymbolFilePDB::FindGlobalVariables(
11421ad655e2SAdrian Prantl     const lldb_private::RegularExpression &regex, uint32_t max_matches,
1143b9c1b51eSKate Stone     lldb_private::VariableList &variables) {
1144656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1145cab0d23fSAaron Smith   if (!regex.IsValid())
11461ad655e2SAdrian Prantl     return;
1147cab0d23fSAaron Smith   auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
1148cab0d23fSAaron Smith   if (!results)
11491ad655e2SAdrian Prantl     return;
1150cab0d23fSAaron Smith 
1151cab0d23fSAaron Smith   uint32_t matches = 0;
1152cab0d23fSAaron Smith   size_t old_size = variables.GetSize();
1153cab0d23fSAaron Smith   while (auto pdb_data = results->getNext()) {
1154cab0d23fSAaron Smith     if (max_matches > 0 && matches >= max_matches)
1155cab0d23fSAaron Smith       break;
1156cab0d23fSAaron Smith 
1157cab0d23fSAaron Smith     auto var_name = pdb_data->getName();
1158cab0d23fSAaron Smith     if (var_name.empty())
1159cab0d23fSAaron Smith       continue;
1160cab0d23fSAaron Smith     if (!regex.Execute(var_name))
1161cab0d23fSAaron Smith       continue;
1162cab0d23fSAaron Smith     SymbolContext sc;
1163d2deeb44SPavel Labath     sc.module_sp = m_objfile_sp->GetModule();
1164cab0d23fSAaron Smith     lldbassert(sc.module_sp.get());
1165cab0d23fSAaron Smith 
1166356aa4a9SAleksandr Urakov     sc.comp_unit = ParseCompileUnitForUID(GetCompilandId(*pdb_data)).get();
1167cab0d23fSAaron Smith     // FIXME: We are not able to determine the compile unit.
1168cab0d23fSAaron Smith     if (sc.comp_unit == nullptr)
1169cab0d23fSAaron Smith       continue;
1170cab0d23fSAaron Smith 
1171cab0d23fSAaron Smith     ParseVariables(sc, *pdb_data, &variables);
1172cab0d23fSAaron Smith     matches = variables.GetSize() - old_size;
1173cab0d23fSAaron Smith   }
1174b9c1b51eSKate Stone }
1175b9c1b51eSKate Stone 
1176e664b5dcSAaron Smith bool SymbolFilePDB::ResolveFunction(const llvm::pdb::PDBSymbolFunc &pdb_func,
11777ac1c780SAaron Smith                                     bool include_inlines,
11787ac1c780SAaron Smith                                     lldb_private::SymbolContextList &sc_list) {
11797ac1c780SAaron Smith   lldb_private::SymbolContext sc;
1180a3a8cc80SAaron Smith   sc.comp_unit = ParseCompileUnitForUID(pdb_func.getCompilandId()).get();
11817ac1c780SAaron Smith   if (!sc.comp_unit)
11827ac1c780SAaron Smith     return false;
11837ac1c780SAaron Smith   sc.module_sp = sc.comp_unit->GetModule();
1184863f8c18SZachary Turner   sc.function = ParseCompileUnitFunctionForPDBFunc(pdb_func, *sc.comp_unit);
11857ac1c780SAaron Smith   if (!sc.function)
11867ac1c780SAaron Smith     return false;
11877ac1c780SAaron Smith 
11887ac1c780SAaron Smith   sc_list.Append(sc);
11897ac1c780SAaron Smith   return true;
11907ac1c780SAaron Smith }
11917ac1c780SAaron Smith 
11927ac1c780SAaron Smith bool SymbolFilePDB::ResolveFunction(uint32_t uid, bool include_inlines,
11937ac1c780SAaron Smith                                     lldb_private::SymbolContextList &sc_list) {
1194c8316ed2SAaron Smith   auto pdb_func_up = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(uid);
11957ac1c780SAaron Smith   if (!pdb_func_up && !(include_inlines && pdb_func_up->hasInlineAttribute()))
11967ac1c780SAaron Smith     return false;
1197e664b5dcSAaron Smith   return ResolveFunction(*pdb_func_up, include_inlines, sc_list);
11987ac1c780SAaron Smith }
11997ac1c780SAaron Smith 
12007ac1c780SAaron Smith void SymbolFilePDB::CacheFunctionNames() {
12017ac1c780SAaron Smith   if (!m_func_full_names.IsEmpty())
12027ac1c780SAaron Smith     return;
12037ac1c780SAaron Smith 
12047ac1c780SAaron Smith   std::map<uint64_t, uint32_t> addr_ids;
12057ac1c780SAaron Smith 
12067ac1c780SAaron Smith   if (auto results_up = m_global_scope_up->findAllChildren<PDBSymbolFunc>()) {
12077ac1c780SAaron Smith     while (auto pdb_func_up = results_up->getNext()) {
1208f76fe682SAaron Smith       if (pdb_func_up->isCompilerGenerated())
1209f76fe682SAaron Smith         continue;
1210f76fe682SAaron Smith 
12117ac1c780SAaron Smith       auto name = pdb_func_up->getName();
12127ac1c780SAaron Smith       auto demangled_name = pdb_func_up->getUndecoratedName();
12137ac1c780SAaron Smith       if (name.empty() && demangled_name.empty())
12147ac1c780SAaron Smith         continue;
12157ac1c780SAaron Smith 
1216f76fe682SAaron Smith       auto uid = pdb_func_up->getSymIndexId();
12177ac1c780SAaron Smith       if (!demangled_name.empty() && pdb_func_up->getVirtualAddress())
12187ac1c780SAaron Smith         addr_ids.insert(std::make_pair(pdb_func_up->getVirtualAddress(), uid));
12197ac1c780SAaron Smith 
12207ac1c780SAaron Smith       if (auto parent = pdb_func_up->getClassParent()) {
12217ac1c780SAaron Smith 
12227ac1c780SAaron Smith         // PDB have symbols for class/struct methods or static methods in Enum
12237ac1c780SAaron Smith         // Class. We won't bother to check if the parent is UDT or Enum here.
12247ac1c780SAaron Smith         m_func_method_names.Append(ConstString(name), uid);
12257ac1c780SAaron Smith 
122605097246SAdrian Prantl         // To search a method name, like NS::Class:MemberFunc, LLDB searches
122705097246SAdrian Prantl         // its base name, i.e. MemberFunc by default. Since PDBSymbolFunc does
1228fa001767SAdrian Prantl         // not have information of this, we extract base names and cache them
122905097246SAdrian Prantl         // by our own effort.
1230c1e530eeSAleksandr Urakov         llvm::StringRef basename = MSVCUndecoratedNameParser::DropScope(name);
12317ac1c780SAaron Smith         if (!basename.empty())
12327ac1c780SAaron Smith           m_func_base_names.Append(ConstString(basename), uid);
12337ac1c780SAaron Smith         else {
12347ac1c780SAaron Smith           m_func_base_names.Append(ConstString(name), uid);
12357ac1c780SAaron Smith         }
12367ac1c780SAaron Smith 
12377ac1c780SAaron Smith         if (!demangled_name.empty())
12387ac1c780SAaron Smith           m_func_full_names.Append(ConstString(demangled_name), uid);
12397ac1c780SAaron Smith 
12407ac1c780SAaron Smith       } else {
12417ac1c780SAaron Smith         // Handle not-method symbols.
12427ac1c780SAaron Smith 
1243c1e530eeSAleksandr Urakov         // The function name might contain namespace, or its lexical scope.
1244c1e530eeSAleksandr Urakov         llvm::StringRef basename = MSVCUndecoratedNameParser::DropScope(name);
1245c1e530eeSAleksandr Urakov         if (!basename.empty())
1246c1e530eeSAleksandr Urakov           m_func_base_names.Append(ConstString(basename), uid);
1247c1e530eeSAleksandr Urakov         else
12487ac1c780SAaron Smith           m_func_base_names.Append(ConstString(name), uid);
12497ac1c780SAaron Smith 
12507ac1c780SAaron Smith         if (name == "main") {
12517ac1c780SAaron Smith           m_func_full_names.Append(ConstString(name), uid);
12527ac1c780SAaron Smith 
12537ac1c780SAaron Smith           if (!demangled_name.empty() && name != demangled_name) {
12547ac1c780SAaron Smith             m_func_full_names.Append(ConstString(demangled_name), uid);
12557ac1c780SAaron Smith             m_func_base_names.Append(ConstString(demangled_name), uid);
12567ac1c780SAaron Smith           }
12577ac1c780SAaron Smith         } else if (!demangled_name.empty()) {
12587ac1c780SAaron Smith           m_func_full_names.Append(ConstString(demangled_name), uid);
12597ac1c780SAaron Smith         } else {
12607ac1c780SAaron Smith           m_func_full_names.Append(ConstString(name), uid);
12617ac1c780SAaron Smith         }
12627ac1c780SAaron Smith       }
12637ac1c780SAaron Smith     }
12647ac1c780SAaron Smith   }
12657ac1c780SAaron Smith 
12667ac1c780SAaron Smith   if (auto results_up =
12677ac1c780SAaron Smith           m_global_scope_up->findAllChildren<PDBSymbolPublicSymbol>()) {
12687ac1c780SAaron Smith     while (auto pub_sym_up = results_up->getNext()) {
12697ac1c780SAaron Smith       if (!pub_sym_up->isFunction())
12707ac1c780SAaron Smith         continue;
12717ac1c780SAaron Smith       auto name = pub_sym_up->getName();
12727ac1c780SAaron Smith       if (name.empty())
12737ac1c780SAaron Smith         continue;
12747ac1c780SAaron Smith 
12757ac1c780SAaron Smith       if (CPlusPlusLanguage::IsCPPMangledName(name.c_str())) {
12767ac1c780SAaron Smith         auto vm_addr = pub_sym_up->getVirtualAddress();
12777ac1c780SAaron Smith 
12787ac1c780SAaron Smith         // PDB public symbol has mangled name for its associated function.
12797ac1c780SAaron Smith         if (vm_addr && addr_ids.find(vm_addr) != addr_ids.end()) {
12807ac1c780SAaron Smith           // Cache mangled name.
12817ac1c780SAaron Smith           m_func_full_names.Append(ConstString(name), addr_ids[vm_addr]);
12827ac1c780SAaron Smith         }
12837ac1c780SAaron Smith       }
12847ac1c780SAaron Smith     }
12857ac1c780SAaron Smith   }
12867ac1c780SAaron Smith   // Sort them before value searching is working properly
12877ac1c780SAaron Smith   m_func_full_names.Sort();
12887ac1c780SAaron Smith   m_func_full_names.SizeToFit();
12897ac1c780SAaron Smith   m_func_method_names.Sort();
12907ac1c780SAaron Smith   m_func_method_names.SizeToFit();
12917ac1c780SAaron Smith   m_func_base_names.Sort();
12927ac1c780SAaron Smith   m_func_base_names.SizeToFit();
12937ac1c780SAaron Smith }
12947ac1c780SAaron Smith 
12951ad655e2SAdrian Prantl void SymbolFilePDB::FindFunctions(
12960e4c4821SAdrian Prantl     lldb_private::ConstString name,
1297b9c1b51eSKate Stone     const lldb_private::CompilerDeclContext *parent_decl_ctx,
12981ad655e2SAdrian Prantl     FunctionNameType name_type_mask, bool include_inlines,
1299b9c1b51eSKate Stone     lldb_private::SymbolContextList &sc_list) {
1300656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
13017ac1c780SAaron Smith   lldbassert((name_type_mask & eFunctionNameTypeAuto) == 0);
13027ac1c780SAaron Smith 
13037ac1c780SAaron Smith   if (name_type_mask == eFunctionNameTypeNone)
13041ad655e2SAdrian Prantl     return;
13057ac1c780SAaron Smith   if (!DeclContextMatchesThisSymbolFile(parent_decl_ctx))
13061ad655e2SAdrian Prantl     return;
13077ac1c780SAaron Smith   if (name.IsEmpty())
13081ad655e2SAdrian Prantl     return;
13097ac1c780SAaron Smith 
13104d4d63eeSPavel Labath   if (name_type_mask & eFunctionNameTypeFull ||
13114d4d63eeSPavel Labath       name_type_mask & eFunctionNameTypeBase ||
13127ac1c780SAaron Smith       name_type_mask & eFunctionNameTypeMethod) {
13137ac1c780SAaron Smith     CacheFunctionNames();
13147ac1c780SAaron Smith 
13157ac1c780SAaron Smith     std::set<uint32_t> resolved_ids;
1316a5235af9SAleksandr Urakov     auto ResolveFn = [this, &name, parent_decl_ctx, include_inlines, &sc_list,
1317a5235af9SAleksandr Urakov                       &resolved_ids](UniqueCStringMap<uint32_t> &Names) {
13187ac1c780SAaron Smith       std::vector<uint32_t> ids;
1319a5235af9SAleksandr Urakov       if (!Names.GetValues(name, ids))
1320a5235af9SAleksandr Urakov         return;
1321a5235af9SAleksandr Urakov 
1322a5235af9SAleksandr Urakov       for (uint32_t id : ids) {
1323a5235af9SAleksandr Urakov         if (resolved_ids.find(id) != resolved_ids.end())
1324a5235af9SAleksandr Urakov           continue;
1325a5235af9SAleksandr Urakov 
1326a5235af9SAleksandr Urakov         if (parent_decl_ctx &&
1327a5235af9SAleksandr Urakov             GetDeclContextContainingUID(id) != *parent_decl_ctx)
1328a5235af9SAleksandr Urakov           continue;
1329a5235af9SAleksandr Urakov 
13307ac1c780SAaron Smith         if (ResolveFunction(id, include_inlines, sc_list))
13317ac1c780SAaron Smith           resolved_ids.insert(id);
13327ac1c780SAaron Smith       }
13337ac1c780SAaron Smith     };
13347ac1c780SAaron Smith     if (name_type_mask & eFunctionNameTypeFull) {
13357ac1c780SAaron Smith       ResolveFn(m_func_full_names);
1336a5235af9SAleksandr Urakov       ResolveFn(m_func_base_names);
1337a5235af9SAleksandr Urakov       ResolveFn(m_func_method_names);
13387ac1c780SAaron Smith     }
13391ad655e2SAdrian Prantl     if (name_type_mask & eFunctionNameTypeBase)
13407ac1c780SAaron Smith       ResolveFn(m_func_base_names);
13411ad655e2SAdrian Prantl     if (name_type_mask & eFunctionNameTypeMethod)
13427ac1c780SAaron Smith       ResolveFn(m_func_method_names);
13437ac1c780SAaron Smith   }
13447ac1c780SAaron Smith }
134574e08ca0SZachary Turner 
13461ad655e2SAdrian Prantl void SymbolFilePDB::FindFunctions(const lldb_private::RegularExpression &regex,
13471ad655e2SAdrian Prantl                                   bool include_inlines,
1348b9c1b51eSKate Stone                                   lldb_private::SymbolContextList &sc_list) {
1349656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
13507ac1c780SAaron Smith   if (!regex.IsValid())
13511ad655e2SAdrian Prantl     return;
13527ac1c780SAaron Smith 
13537ac1c780SAaron Smith   CacheFunctionNames();
13547ac1c780SAaron Smith 
13557ac1c780SAaron Smith   std::set<uint32_t> resolved_ids;
1356c8316ed2SAaron Smith   auto ResolveFn = [&regex, include_inlines, &sc_list, &resolved_ids,
1357c8316ed2SAaron Smith                     this](UniqueCStringMap<uint32_t> &Names) {
13587ac1c780SAaron Smith     std::vector<uint32_t> ids;
13597ac1c780SAaron Smith     if (Names.GetValues(regex, ids)) {
13607ac1c780SAaron Smith       for (auto id : ids) {
13617ac1c780SAaron Smith         if (resolved_ids.find(id) == resolved_ids.end())
13627ac1c780SAaron Smith           if (ResolveFunction(id, include_inlines, sc_list))
13637ac1c780SAaron Smith             resolved_ids.insert(id);
13647ac1c780SAaron Smith       }
13657ac1c780SAaron Smith     }
13667ac1c780SAaron Smith   };
13677ac1c780SAaron Smith   ResolveFn(m_func_full_names);
13687ac1c780SAaron Smith   ResolveFn(m_func_base_names);
136974e08ca0SZachary Turner }
137074e08ca0SZachary Turner 
1371b9c1b51eSKate Stone void SymbolFilePDB::GetMangledNamesForFunction(
1372b9c1b51eSKate Stone     const std::string &scope_qualified_name,
1373b9c1b51eSKate Stone     std::vector<lldb_private::ConstString> &mangled_names) {}
137474e08ca0SZachary Turner 
13758cfb12b9SAleksandr Urakov void SymbolFilePDB::AddSymbols(lldb_private::Symtab &symtab) {
13768cfb12b9SAleksandr Urakov   std::set<lldb::addr_t> sym_addresses;
13778cfb12b9SAleksandr Urakov   for (size_t i = 0; i < symtab.GetNumSymbols(); i++)
13788cfb12b9SAleksandr Urakov     sym_addresses.insert(symtab.SymbolAtIndex(i)->GetFileAddress());
13798cfb12b9SAleksandr Urakov 
13808cfb12b9SAleksandr Urakov   auto results = m_global_scope_up->findAllChildren<PDBSymbolPublicSymbol>();
13818cfb12b9SAleksandr Urakov   if (!results)
13828cfb12b9SAleksandr Urakov     return;
13838cfb12b9SAleksandr Urakov 
1384d2deeb44SPavel Labath   auto section_list = m_objfile_sp->GetSectionList();
13858cfb12b9SAleksandr Urakov   if (!section_list)
13868cfb12b9SAleksandr Urakov     return;
13878cfb12b9SAleksandr Urakov 
13888cfb12b9SAleksandr Urakov   while (auto pub_symbol = results->getNext()) {
13897db8b5c4SPavel Labath     auto section_id = pub_symbol->getAddressSection();
13908cfb12b9SAleksandr Urakov 
13917db8b5c4SPavel Labath     auto section = section_list->FindSectionByID(section_id);
13928cfb12b9SAleksandr Urakov     if (!section)
13938cfb12b9SAleksandr Urakov       continue;
13948cfb12b9SAleksandr Urakov 
13958cfb12b9SAleksandr Urakov     auto offset = pub_symbol->getAddressOffset();
13968cfb12b9SAleksandr Urakov 
13978cfb12b9SAleksandr Urakov     auto file_addr = section->GetFileAddress() + offset;
13988cfb12b9SAleksandr Urakov     if (sym_addresses.find(file_addr) != sym_addresses.end())
13998cfb12b9SAleksandr Urakov       continue;
14008cfb12b9SAleksandr Urakov     sym_addresses.insert(file_addr);
14018cfb12b9SAleksandr Urakov 
14028cfb12b9SAleksandr Urakov     auto size = pub_symbol->getLength();
14038cfb12b9SAleksandr Urakov     symtab.AddSymbol(
14048cfb12b9SAleksandr Urakov         Symbol(pub_symbol->getSymIndexId(),   // symID
14058cfb12b9SAleksandr Urakov                pub_symbol->getName().c_str(), // name
14068cfb12b9SAleksandr Urakov                pub_symbol->isCode() ? eSymbolTypeCode : eSymbolTypeData, // type
14078cfb12b9SAleksandr Urakov                true,      // external
14088cfb12b9SAleksandr Urakov                false,     // is_debug
14098cfb12b9SAleksandr Urakov                false,     // is_trampoline
14108cfb12b9SAleksandr Urakov                false,     // is_artificial
14118cfb12b9SAleksandr Urakov                section,   // section_sp
14128cfb12b9SAleksandr Urakov                offset,    // value
14138cfb12b9SAleksandr Urakov                size,      // size
14148cfb12b9SAleksandr Urakov                size != 0, // size_is_valid
14158cfb12b9SAleksandr Urakov                false,     // contains_linker_annotations
14168cfb12b9SAleksandr Urakov                0          // flags
14178cfb12b9SAleksandr Urakov                ));
14188cfb12b9SAleksandr Urakov   }
14198cfb12b9SAleksandr Urakov 
14208cfb12b9SAleksandr Urakov   symtab.CalculateSymbolSizes();
14218cfb12b9SAleksandr Urakov   symtab.Finalize();
14228cfb12b9SAleksandr Urakov }
14238cfb12b9SAleksandr Urakov 
1424796cd312SAdrian Prantl void SymbolFilePDB::FindTypes(
14250e4c4821SAdrian Prantl     lldb_private::ConstString name,
1426d4d428efSAdrian Prantl     const lldb_private::CompilerDeclContext *parent_decl_ctx,
1427b9c1b51eSKate Stone     uint32_t max_matches,
142874e08ca0SZachary Turner     llvm::DenseSet<lldb_private::SymbolFile *> &searched_symbol_files,
1429b9c1b51eSKate Stone     lldb_private::TypeMap &types) {
1430656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
143142dff790SZachary Turner   if (!name)
1432796cd312SAdrian Prantl     return;
14337ac1c780SAaron Smith   if (!DeclContextMatchesThisSymbolFile(parent_decl_ctx))
1434796cd312SAdrian Prantl     return;
143542dff790SZachary Turner 
143642dff790SZachary Turner   searched_symbol_files.clear();
143742dff790SZachary Turner   searched_symbol_files.insert(this);
143842dff790SZachary Turner 
143986e9434dSAaron Smith   // There is an assumption 'name' is not a regex
1440c1e530eeSAleksandr Urakov   FindTypesByName(name.GetStringRef(), parent_decl_ctx, max_matches, types);
144142dff790SZachary Turner }
144242dff790SZachary Turner 
14434911023fSZachary Turner void SymbolFilePDB::DumpClangAST(Stream &s) {
14440e252e38SAlex Langford   auto type_system_or_err =
14450e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
14460e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
14470e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
14480e252e38SAlex Langford                    std::move(err), "Unable to dump ClangAST");
14494911023fSZachary Turner     return;
14500e252e38SAlex Langford   }
14510e252e38SAlex Langford 
14520e252e38SAlex Langford   auto *clang_type_system =
14536e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
14540e252e38SAlex Langford   if (!clang_type_system)
14550e252e38SAlex Langford     return;
14560e252e38SAlex Langford   clang_type_system->Dump(s);
14574911023fSZachary Turner }
14584911023fSZachary Turner 
1459c8316ed2SAaron Smith void SymbolFilePDB::FindTypesByRegex(
1460c8316ed2SAaron Smith     const lldb_private::RegularExpression &regex, uint32_t max_matches,
1461b9c1b51eSKate Stone     lldb_private::TypeMap &types) {
1462b9c1b51eSKate Stone   // When searching by regex, we need to go out of our way to limit the search
14639d0eb996SAdrian McCarthy   // space as much as possible since this searches EVERYTHING in the PDB,
14649d0eb996SAdrian McCarthy   // manually doing regex comparisons.  PDB library isn't optimized for regex
14659d0eb996SAdrian McCarthy   // searches or searches across multiple symbol types at the same time, so the
1466b9c1b51eSKate Stone   // best we can do is to search enums, then typedefs, then classes one by one,
14679d0eb996SAdrian McCarthy   // and do a regex comparison against each of them.
1468b9c1b51eSKate Stone   PDB_SymType tags_to_search[] = {PDB_SymType::Enum, PDB_SymType::Typedef,
1469b9c1b51eSKate Stone                                   PDB_SymType::UDT};
147054fd7ff6SZachary Turner   std::unique_ptr<IPDBEnumSymbols> results;
147142dff790SZachary Turner 
147242dff790SZachary Turner   uint32_t matches = 0;
147342dff790SZachary Turner 
1474b9c1b51eSKate Stone   for (auto tag : tags_to_search) {
147510a02577SAaron Smith     results = m_global_scope_up->findAllChildren(tag);
147610a02577SAaron Smith     if (!results)
147710a02577SAaron Smith       continue;
147810a02577SAaron Smith 
1479b9c1b51eSKate Stone     while (auto result = results->getNext()) {
148042dff790SZachary Turner       if (max_matches > 0 && matches >= max_matches)
148142dff790SZachary Turner         break;
148242dff790SZachary Turner 
148342dff790SZachary Turner       std::string type_name;
148454fd7ff6SZachary Turner       if (auto enum_type = llvm::dyn_cast<PDBSymbolTypeEnum>(result.get()))
148542dff790SZachary Turner         type_name = enum_type->getName();
1486b9c1b51eSKate Stone       else if (auto typedef_type =
1487b9c1b51eSKate Stone                    llvm::dyn_cast<PDBSymbolTypeTypedef>(result.get()))
148842dff790SZachary Turner         type_name = typedef_type->getName();
148954fd7ff6SZachary Turner       else if (auto class_type = llvm::dyn_cast<PDBSymbolTypeUDT>(result.get()))
149042dff790SZachary Turner         type_name = class_type->getName();
1491b9c1b51eSKate Stone       else {
14929d0eb996SAdrian McCarthy         // We're looking only for types that have names.  Skip symbols, as well
14939d0eb996SAdrian McCarthy         // as unnamed types such as arrays, pointers, etc.
149442dff790SZachary Turner         continue;
149542dff790SZachary Turner       }
149642dff790SZachary Turner 
149786e9434dSAaron Smith       if (!regex.Execute(type_name))
149842dff790SZachary Turner         continue;
149942dff790SZachary Turner 
1500b9c1b51eSKate Stone       // This should cause the type to get cached and stored in the `m_types`
1501b9c1b51eSKate Stone       // lookup.
150242dff790SZachary Turner       if (!ResolveTypeUID(result->getSymIndexId()))
150342dff790SZachary Turner         continue;
150442dff790SZachary Turner 
150542dff790SZachary Turner       auto iter = m_types.find(result->getSymIndexId());
150642dff790SZachary Turner       if (iter == m_types.end())
150742dff790SZachary Turner         continue;
150842dff790SZachary Turner       types.Insert(iter->second);
150942dff790SZachary Turner       ++matches;
151042dff790SZachary Turner     }
151142dff790SZachary Turner   }
151242dff790SZachary Turner }
151342dff790SZachary Turner 
1514709426b3SAleksandr Urakov void SymbolFilePDB::FindTypesByName(
1515c1e530eeSAleksandr Urakov     llvm::StringRef name,
1516709426b3SAleksandr Urakov     const lldb_private::CompilerDeclContext *parent_decl_ctx,
1517709426b3SAleksandr Urakov     uint32_t max_matches, lldb_private::TypeMap &types) {
151854fd7ff6SZachary Turner   std::unique_ptr<IPDBEnumSymbols> results;
1519f76fe682SAaron Smith   if (name.empty())
1520f76fe682SAaron Smith     return;
1521709426b3SAleksandr Urakov   results = m_global_scope_up->findAllChildren(PDB_SymType::None);
152210a02577SAaron Smith   if (!results)
152310a02577SAaron Smith     return;
152442dff790SZachary Turner 
152542dff790SZachary Turner   uint32_t matches = 0;
152642dff790SZachary Turner 
1527b9c1b51eSKate Stone   while (auto result = results->getNext()) {
152842dff790SZachary Turner     if (max_matches > 0 && matches >= max_matches)
152942dff790SZachary Turner       break;
1530709426b3SAleksandr Urakov 
1531c1e530eeSAleksandr Urakov     if (MSVCUndecoratedNameParser::DropScope(
1532c1e530eeSAleksandr Urakov             result->getRawSymbol().getName()) != name)
1533709426b3SAleksandr Urakov       continue;
1534709426b3SAleksandr Urakov 
1535b9c1b51eSKate Stone     switch (result->getSymTag()) {
153654fd7ff6SZachary Turner     case PDB_SymType::Enum:
153754fd7ff6SZachary Turner     case PDB_SymType::UDT:
153854fd7ff6SZachary Turner     case PDB_SymType::Typedef:
153942dff790SZachary Turner       break;
154042dff790SZachary Turner     default:
154105097246SAdrian Prantl       // We're looking only for types that have names.  Skip symbols, as well
154205097246SAdrian Prantl       // as unnamed types such as arrays, pointers, etc.
154342dff790SZachary Turner       continue;
154442dff790SZachary Turner     }
154542dff790SZachary Turner 
1546b9c1b51eSKate Stone     // This should cause the type to get cached and stored in the `m_types`
1547b9c1b51eSKate Stone     // lookup.
154842dff790SZachary Turner     if (!ResolveTypeUID(result->getSymIndexId()))
154942dff790SZachary Turner       continue;
155042dff790SZachary Turner 
1551a5235af9SAleksandr Urakov     if (parent_decl_ctx && GetDeclContextContainingUID(
1552a5235af9SAleksandr Urakov                                result->getSymIndexId()) != *parent_decl_ctx)
1553709426b3SAleksandr Urakov       continue;
1554709426b3SAleksandr Urakov 
155542dff790SZachary Turner     auto iter = m_types.find(result->getSymIndexId());
155642dff790SZachary Turner     if (iter == m_types.end())
155742dff790SZachary Turner       continue;
155842dff790SZachary Turner     types.Insert(iter->second);
155942dff790SZachary Turner     ++matches;
156042dff790SZachary Turner   }
156174e08ca0SZachary Turner }
156274e08ca0SZachary Turner 
15633b73dcdcSAdrian Prantl void SymbolFilePDB::FindTypes(
15643b73dcdcSAdrian Prantl     llvm::ArrayRef<CompilerContext> pattern, LanguageSet languages,
15653b73dcdcSAdrian Prantl     llvm::DenseSet<SymbolFile *> &searched_symbol_files,
1566796cd312SAdrian Prantl     lldb_private::TypeMap &types) {}
156774e08ca0SZachary Turner 
1568c8316ed2SAaron Smith void SymbolFilePDB::GetTypesForPDBSymbol(const llvm::pdb::PDBSymbol &pdb_symbol,
15697ac1c780SAaron Smith                                          uint32_t type_mask,
15707ac1c780SAaron Smith                                          TypeCollection &type_collection) {
15717ac1c780SAaron Smith   bool can_parse = false;
1572e664b5dcSAaron Smith   switch (pdb_symbol.getSymTag()) {
15737ac1c780SAaron Smith   case PDB_SymType::ArrayType:
15747ac1c780SAaron Smith     can_parse = ((type_mask & eTypeClassArray) != 0);
15757ac1c780SAaron Smith     break;
15767ac1c780SAaron Smith   case PDB_SymType::BuiltinType:
15777ac1c780SAaron Smith     can_parse = ((type_mask & eTypeClassBuiltin) != 0);
15787ac1c780SAaron Smith     break;
15797ac1c780SAaron Smith   case PDB_SymType::Enum:
15807ac1c780SAaron Smith     can_parse = ((type_mask & eTypeClassEnumeration) != 0);
15817ac1c780SAaron Smith     break;
15827ac1c780SAaron Smith   case PDB_SymType::Function:
15837ac1c780SAaron Smith   case PDB_SymType::FunctionSig:
15847ac1c780SAaron Smith     can_parse = ((type_mask & eTypeClassFunction) != 0);
15857ac1c780SAaron Smith     break;
15867ac1c780SAaron Smith   case PDB_SymType::PointerType:
15877ac1c780SAaron Smith     can_parse = ((type_mask & (eTypeClassPointer | eTypeClassBlockPointer |
15887ac1c780SAaron Smith                                eTypeClassMemberPointer)) != 0);
15897ac1c780SAaron Smith     break;
15907ac1c780SAaron Smith   case PDB_SymType::Typedef:
15917ac1c780SAaron Smith     can_parse = ((type_mask & eTypeClassTypedef) != 0);
15927ac1c780SAaron Smith     break;
15937ac1c780SAaron Smith   case PDB_SymType::UDT: {
1594e664b5dcSAaron Smith     auto *udt = llvm::dyn_cast<PDBSymbolTypeUDT>(&pdb_symbol);
15957ac1c780SAaron Smith     assert(udt);
15967ac1c780SAaron Smith     can_parse = (udt->getUdtKind() != PDB_UdtType::Interface &&
15977ac1c780SAaron Smith                  ((type_mask & (eTypeClassClass | eTypeClassStruct |
15987ac1c780SAaron Smith                                 eTypeClassUnion)) != 0));
15997ac1c780SAaron Smith   } break;
1600c8316ed2SAaron Smith   default:
1601c8316ed2SAaron Smith     break;
16027ac1c780SAaron Smith   }
16037ac1c780SAaron Smith 
16047ac1c780SAaron Smith   if (can_parse) {
1605e664b5dcSAaron Smith     if (auto *type = ResolveTypeUID(pdb_symbol.getSymIndexId())) {
16067ac1c780SAaron Smith       auto result =
16077ac1c780SAaron Smith           std::find(type_collection.begin(), type_collection.end(), type);
16087ac1c780SAaron Smith       if (result == type_collection.end())
16097ac1c780SAaron Smith         type_collection.push_back(type);
16107ac1c780SAaron Smith     }
16117ac1c780SAaron Smith   }
16127ac1c780SAaron Smith 
1613e664b5dcSAaron Smith   auto results_up = pdb_symbol.findAllChildren();
16147ac1c780SAaron Smith   while (auto symbol_up = results_up->getNext())
1615e664b5dcSAaron Smith     GetTypesForPDBSymbol(*symbol_up, type_mask, type_collection);
16167ac1c780SAaron Smith }
16177ac1c780SAaron Smith 
1618796cd312SAdrian Prantl void SymbolFilePDB::GetTypes(lldb_private::SymbolContextScope *sc_scope,
1619117b1fa1SZachary Turner                              TypeClass type_mask,
1620b9c1b51eSKate Stone                              lldb_private::TypeList &type_list) {
1621656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
16227ac1c780SAaron Smith   TypeCollection type_collection;
1623c8316ed2SAaron Smith   CompileUnit *cu =
1624c8316ed2SAaron Smith       sc_scope ? sc_scope->CalculateSymbolContextCompileUnit() : nullptr;
16257ac1c780SAaron Smith   if (cu) {
16267ac1c780SAaron Smith     auto compiland_up = GetPDBCompilandByUID(cu->GetID());
1627e664b5dcSAaron Smith     if (!compiland_up)
1628796cd312SAdrian Prantl       return;
1629e664b5dcSAaron Smith     GetTypesForPDBSymbol(*compiland_up, type_mask, type_collection);
16307ac1c780SAaron Smith   } else {
16317ac1c780SAaron Smith     for (uint32_t cu_idx = 0; cu_idx < GetNumCompileUnits(); ++cu_idx) {
16327ac1c780SAaron Smith       auto cu_sp = ParseCompileUnitAtIndex(cu_idx);
1633d5a925f4SAaron Smith       if (cu_sp) {
1634e664b5dcSAaron Smith         if (auto compiland_up = GetPDBCompilandByUID(cu_sp->GetID()))
1635e664b5dcSAaron Smith           GetTypesForPDBSymbol(*compiland_up, type_mask, type_collection);
16367ac1c780SAaron Smith       }
16377ac1c780SAaron Smith     }
16387ac1c780SAaron Smith   }
16397ac1c780SAaron Smith 
16407ac1c780SAaron Smith   for (auto type : type_collection) {
16417ac1c780SAaron Smith     type->GetForwardCompilerType();
16427ac1c780SAaron Smith     type_list.Insert(type->shared_from_this());
16437ac1c780SAaron Smith   }
164474e08ca0SZachary Turner }
164574e08ca0SZachary Turner 
16460e252e38SAlex Langford llvm::Expected<lldb_private::TypeSystem &>
1647b9c1b51eSKate Stone SymbolFilePDB::GetTypeSystemForLanguage(lldb::LanguageType language) {
16480e252e38SAlex Langford   auto type_system_or_err =
1649d2deeb44SPavel Labath       m_objfile_sp->GetModule()->GetTypeSystemForLanguage(language);
16500e252e38SAlex Langford   if (type_system_or_err) {
16510e252e38SAlex Langford     type_system_or_err->SetSymbolFile(this);
16520e252e38SAlex Langford   }
16530e252e38SAlex Langford   return type_system_or_err;
165474e08ca0SZachary Turner }
165574e08ca0SZachary Turner 
1656c68925abSZachary Turner PDBASTParser *SymbolFilePDB::GetPDBAstParser() {
16570e252e38SAlex Langford   auto type_system_or_err =
16580e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
16590e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
16600e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
16610e252e38SAlex Langford                    std::move(err), "Unable to get PDB AST parser");
16620e252e38SAlex Langford     return nullptr;
16630e252e38SAlex Langford   }
16640e252e38SAlex Langford 
16650e252e38SAlex Langford   auto *clang_type_system =
16666e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
1667c68925abSZachary Turner   if (!clang_type_system)
1668c68925abSZachary Turner     return nullptr;
1669c68925abSZachary Turner 
1670c68925abSZachary Turner   return clang_type_system->GetPDBParser();
1671c68925abSZachary Turner }
1672c68925abSZachary Turner 
1673c68925abSZachary Turner 
1674b9c1b51eSKate Stone lldb_private::CompilerDeclContext SymbolFilePDB::FindNamespace(
16750e4c4821SAdrian Prantl     lldb_private::ConstString name,
1676b9c1b51eSKate Stone     const lldb_private::CompilerDeclContext *parent_decl_ctx) {
1677656ddeb2SPavel Labath   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
16780e252e38SAlex Langford   auto type_system_or_err =
16790e252e38SAlex Langford       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
16800e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
16810e252e38SAlex Langford     LLDB_LOG_ERROR(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
16820e252e38SAlex Langford                    std::move(err), "Unable to find namespace {}",
16830e252e38SAlex Langford                    name.AsCString());
16840e252e38SAlex Langford     return CompilerDeclContext();
16850e252e38SAlex Langford   }
16860e252e38SAlex Langford 
16870e252e38SAlex Langford   auto *clang_type_system =
16886e3b0cc2SRaphael Isemann       llvm::dyn_cast_or_null<TypeSystemClang>(&type_system_or_err.get());
1689709426b3SAleksandr Urakov   if (!clang_type_system)
1690709426b3SAleksandr Urakov     return CompilerDeclContext();
1691709426b3SAleksandr Urakov 
1692709426b3SAleksandr Urakov   PDBASTParser *pdb = clang_type_system->GetPDBParser();
1693709426b3SAleksandr Urakov   if (!pdb)
1694709426b3SAleksandr Urakov     return CompilerDeclContext();
1695709426b3SAleksandr Urakov 
1696709426b3SAleksandr Urakov   clang::DeclContext *decl_context = nullptr;
1697709426b3SAleksandr Urakov   if (parent_decl_ctx)
1698709426b3SAleksandr Urakov     decl_context = static_cast<clang::DeclContext *>(
1699709426b3SAleksandr Urakov         parent_decl_ctx->GetOpaqueDeclContext());
1700709426b3SAleksandr Urakov 
1701709426b3SAleksandr Urakov   auto namespace_decl =
1702709426b3SAleksandr Urakov       pdb->FindNamespaceDecl(decl_context, name.GetStringRef());
1703709426b3SAleksandr Urakov   if (!namespace_decl)
1704709426b3SAleksandr Urakov     return CompilerDeclContext();
1705709426b3SAleksandr Urakov 
170642ec584aSRaphael Isemann   return clang_type_system->CreateDeclContext(namespace_decl);
170774e08ca0SZachary Turner }
170874e08ca0SZachary Turner 
1709b9c1b51eSKate Stone lldb_private::ConstString SymbolFilePDB::GetPluginName() {
171074e08ca0SZachary Turner   static ConstString g_name("pdb");
171174e08ca0SZachary Turner   return g_name;
171274e08ca0SZachary Turner }
171374e08ca0SZachary Turner 
1714b9c1b51eSKate Stone uint32_t SymbolFilePDB::GetPluginVersion() { return 1; }
171574e08ca0SZachary Turner 
1716b9c1b51eSKate Stone IPDBSession &SymbolFilePDB::GetPDBSession() { return *m_session_up; }
1717b9c1b51eSKate Stone 
1718b9c1b51eSKate Stone const IPDBSession &SymbolFilePDB::GetPDBSession() const {
171942dff790SZachary Turner   return *m_session_up;
172042dff790SZachary Turner }
172142dff790SZachary Turner 
1722c8316ed2SAaron Smith lldb::CompUnitSP SymbolFilePDB::ParseCompileUnitForUID(uint32_t id,
1723c8316ed2SAaron Smith                                                        uint32_t index) {
172474e08ca0SZachary Turner   auto found_cu = m_comp_units.find(id);
172574e08ca0SZachary Turner   if (found_cu != m_comp_units.end())
172674e08ca0SZachary Turner     return found_cu->second;
172774e08ca0SZachary Turner 
172810a02577SAaron Smith   auto compiland_up = GetPDBCompilandByUID(id);
172910a02577SAaron Smith   if (!compiland_up)
173010a02577SAaron Smith     return CompUnitSP();
173174e08ca0SZachary Turner 
173274e08ca0SZachary Turner   lldb::LanguageType lang;
173310a02577SAaron Smith   auto details = compiland_up->findOneChild<PDBSymbolCompilandDetails>();
173474e08ca0SZachary Turner   if (!details)
173574e08ca0SZachary Turner     lang = lldb::eLanguageTypeC_plus_plus;
173674e08ca0SZachary Turner   else
173774e08ca0SZachary Turner     lang = TranslateLanguage(details->getLanguage());
173874e08ca0SZachary Turner 
1739f76fe682SAaron Smith   if (lang == lldb::LanguageType::eLanguageTypeUnknown)
1740f76fe682SAaron Smith     return CompUnitSP();
1741f76fe682SAaron Smith 
1742487b0c6bSAaron Smith   std::string path = compiland_up->getSourceFileFullPath();
1743f76fe682SAaron Smith   if (path.empty())
1744f76fe682SAaron Smith     return CompUnitSP();
1745f76fe682SAaron Smith 
1746b9c1b51eSKate Stone   // Don't support optimized code for now, DebugInfoPDB does not return this
1747b9c1b51eSKate Stone   // information.
1748ad2b63cbSGreg Clayton   LazyBool optimized = eLazyBoolNo;
1749d2deeb44SPavel Labath   auto cu_sp = std::make_shared<CompileUnit>(m_objfile_sp->GetModule(), nullptr,
1750c8316ed2SAaron Smith                                              path.c_str(), id, lang, optimized);
175110a02577SAaron Smith 
175210a02577SAaron Smith   if (!cu_sp)
175310a02577SAaron Smith     return CompUnitSP();
175410a02577SAaron Smith 
175510a02577SAaron Smith   m_comp_units.insert(std::make_pair(id, cu_sp));
175610a02577SAaron Smith   if (index == UINT32_MAX)
1757e664b5dcSAaron Smith     GetCompileUnitIndex(*compiland_up, index);
175810a02577SAaron Smith   lldbassert(index != UINT32_MAX);
1759e0119909SPavel Labath   SetCompileUnitAtIndex(index, cu_sp);
176010a02577SAaron Smith   return cu_sp;
176174e08ca0SZachary Turner }
176274e08ca0SZachary Turner 
1763863f8c18SZachary Turner bool SymbolFilePDB::ParseCompileUnitLineTable(CompileUnit &comp_unit,
1764863f8c18SZachary Turner                                               uint32_t match_line) {
1765863f8c18SZachary Turner   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
176610a02577SAaron Smith   if (!compiland_up)
176710a02577SAaron Smith     return false;
176874e08ca0SZachary Turner 
1769b9c1b51eSKate Stone   // LineEntry needs the *index* of the file into the list of support files
17709d0eb996SAdrian McCarthy   // returned by ParseCompileUnitSupportFiles.  But the underlying SDK gives us
177105097246SAdrian Prantl   // a globally unique idenfitifier in the namespace of the PDB.  So, we have
177205097246SAdrian Prantl   // to do a mapping so that we can hand out indices.
177342dff790SZachary Turner   llvm::DenseMap<uint32_t, uint32_t> index_map;
177410a02577SAaron Smith   BuildSupportFileIdToSupportFileIndexMap(*compiland_up, index_map);
1775c4267b7bSPavel Labath   auto line_table = std::make_unique<LineTable>(&comp_unit);
177674e08ca0SZachary Turner 
177710a02577SAaron Smith   // Find contributions to `compiland` from all source and header files.
177810a02577SAaron Smith   auto files = m_session_up->getSourceFilesForCompiland(*compiland_up);
177910a02577SAaron Smith   if (!files)
178010a02577SAaron Smith     return false;
178174e08ca0SZachary Turner 
178205097246SAdrian Prantl   // For each source and header file, create a LineSequence for contributions
178305097246SAdrian Prantl   // to the compiland from that file, and add the sequence.
1784b9c1b51eSKate Stone   while (auto file = files->getNext()) {
1785c4267b7bSPavel Labath     std::unique_ptr<LineSequence> sequence(
1786c4267b7bSPavel Labath         line_table->CreateLineSequenceContainer());
178710a02577SAaron Smith     auto lines = m_session_up->findLineNumbers(*compiland_up, *file);
178810a02577SAaron Smith     if (!lines)
178910a02577SAaron Smith       continue;
179074e08ca0SZachary Turner     int entry_count = lines->getChildCount();
179174e08ca0SZachary Turner 
17927e8c7beaSZachary Turner     uint64_t prev_addr;
17937e8c7beaSZachary Turner     uint32_t prev_length;
17947e8c7beaSZachary Turner     uint32_t prev_line;
17957e8c7beaSZachary Turner     uint32_t prev_source_idx;
17967e8c7beaSZachary Turner 
1797b9c1b51eSKate Stone     for (int i = 0; i < entry_count; ++i) {
179874e08ca0SZachary Turner       auto line = lines->getChildAtIndex(i);
179974e08ca0SZachary Turner 
18007e8c7beaSZachary Turner       uint64_t lno = line->getLineNumber();
18017e8c7beaSZachary Turner       uint64_t addr = line->getVirtualAddress();
18027e8c7beaSZachary Turner       uint32_t length = line->getLength();
180374e08ca0SZachary Turner       uint32_t source_id = line->getSourceFileId();
18047e8c7beaSZachary Turner       uint32_t col = line->getColumnNumber();
180574e08ca0SZachary Turner       uint32_t source_idx = index_map[source_id];
180674e08ca0SZachary Turner 
180705097246SAdrian Prantl       // There was a gap between the current entry and the previous entry if
180805097246SAdrian Prantl       // the addresses don't perfectly line up.
18097e8c7beaSZachary Turner       bool is_gap = (i > 0) && (prev_addr + prev_length < addr);
18107e8c7beaSZachary Turner 
1811b9c1b51eSKate Stone       // Before inserting the current entry, insert a terminal entry at the end
18129d0eb996SAdrian McCarthy       // of the previous entry's address range if the current entry resulted in
18139d0eb996SAdrian McCarthy       // a gap from the previous entry.
1814b9c1b51eSKate Stone       if (is_gap && ShouldAddLine(match_line, prev_line, prev_length)) {
1815c4267b7bSPavel Labath         line_table->AppendLineEntryToSequence(
1816b9c1b51eSKate Stone             sequence.get(), prev_addr + prev_length, prev_line, 0,
18177e8c7beaSZachary Turner             prev_source_idx, false, false, false, false, true);
1818010edd37SAaron Smith 
1819c4267b7bSPavel Labath         line_table->InsertSequence(sequence.release());
1820c4267b7bSPavel Labath         sequence = line_table->CreateLineSequenceContainer();
18217e8c7beaSZachary Turner       }
18227e8c7beaSZachary Turner 
1823b9c1b51eSKate Stone       if (ShouldAddLine(match_line, lno, length)) {
18247e8c7beaSZachary Turner         bool is_statement = line->isStatement();
182574e08ca0SZachary Turner         bool is_prologue = false;
182674e08ca0SZachary Turner         bool is_epilogue = false;
1827b9c1b51eSKate Stone         auto func =
1828b9c1b51eSKate Stone             m_session_up->findSymbolByAddress(addr, PDB_SymType::Function);
1829b9c1b51eSKate Stone         if (func) {
183054fd7ff6SZachary Turner           auto prologue = func->findOneChild<PDBSymbolFuncDebugStart>();
183110a02577SAaron Smith           if (prologue)
18327e8c7beaSZachary Turner             is_prologue = (addr == prologue->getVirtualAddress());
183374e08ca0SZachary Turner 
183454fd7ff6SZachary Turner           auto epilogue = func->findOneChild<PDBSymbolFuncDebugEnd>();
183510a02577SAaron Smith           if (epilogue)
18367e8c7beaSZachary Turner             is_epilogue = (addr == epilogue->getVirtualAddress());
18377e8c7beaSZachary Turner         }
183874e08ca0SZachary Turner 
1839c4267b7bSPavel Labath         line_table->AppendLineEntryToSequence(sequence.get(), addr, lno, col,
1840b9c1b51eSKate Stone                                               source_idx, is_statement, false,
18417e8c7beaSZachary Turner                                               is_prologue, is_epilogue, false);
18427e8c7beaSZachary Turner       }
18437e8c7beaSZachary Turner 
18447e8c7beaSZachary Turner       prev_addr = addr;
18457e8c7beaSZachary Turner       prev_length = length;
18467e8c7beaSZachary Turner       prev_line = lno;
18477e8c7beaSZachary Turner       prev_source_idx = source_idx;
18487e8c7beaSZachary Turner     }
18497e8c7beaSZachary Turner 
1850b9c1b51eSKate Stone     if (entry_count > 0 && ShouldAddLine(match_line, prev_line, prev_length)) {
18517e8c7beaSZachary Turner       // The end is always a terminal entry, so insert it regardless.
1852c4267b7bSPavel Labath       line_table->AppendLineEntryToSequence(
1853b9c1b51eSKate Stone           sequence.get(), prev_addr + prev_length, prev_line, 0,
18547e8c7beaSZachary Turner           prev_source_idx, false, false, false, false, true);
185574e08ca0SZachary Turner     }
185674e08ca0SZachary Turner 
1857c4267b7bSPavel Labath     line_table->InsertSequence(sequence.get());
185874e08ca0SZachary Turner   }
185974e08ca0SZachary Turner 
186010a02577SAaron Smith   if (line_table->GetSize()) {
1861863f8c18SZachary Turner     comp_unit.SetLineTable(line_table.release());
186274e08ca0SZachary Turner     return true;
186374e08ca0SZachary Turner   }
186410a02577SAaron Smith   return false;
186510a02577SAaron Smith }
186674e08ca0SZachary Turner 
1867b9c1b51eSKate Stone void SymbolFilePDB::BuildSupportFileIdToSupportFileIndexMap(
186810a02577SAaron Smith     const PDBSymbolCompiland &compiland,
1869b9c1b51eSKate Stone     llvm::DenseMap<uint32_t, uint32_t> &index_map) const {
187005097246SAdrian Prantl   // This is a hack, but we need to convert the source id into an index into
187105097246SAdrian Prantl   // the support files array.  We don't want to do path comparisons to avoid
18729d0eb996SAdrian McCarthy   // basename / full path issues that may or may not even be a problem, so we
18739d0eb996SAdrian McCarthy   // use the globally unique source file identifiers.  Ideally we could use the
18749d0eb996SAdrian McCarthy   // global identifiers everywhere, but LineEntry currently assumes indices.
187510a02577SAaron Smith   auto source_files = m_session_up->getSourceFilesForCompiland(compiland);
187610a02577SAaron Smith   if (!source_files)
187710a02577SAaron Smith     return;
18789ea80d25SPavel Labath 
187957f8a998SPavel Labath   int index = 0;
1880b9c1b51eSKate Stone   while (auto file = source_files->getNext()) {
188174e08ca0SZachary Turner     uint32_t source_id = file->getUniqueId();
188274e08ca0SZachary Turner     index_map[source_id] = index++;
188374e08ca0SZachary Turner   }
188474e08ca0SZachary Turner }
18857ac1c780SAaron Smith 
18867ac1c780SAaron Smith lldb::CompUnitSP SymbolFilePDB::GetCompileUnitContainsAddress(
18877ac1c780SAaron Smith     const lldb_private::Address &so_addr) {
18887ac1c780SAaron Smith   lldb::addr_t file_vm_addr = so_addr.GetFileAddress();
1889308e39caSAaron Smith   if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
18907ac1c780SAaron Smith     return nullptr;
18917ac1c780SAaron Smith 
1892308e39caSAaron Smith   // If it is a PDB function's vm addr, this is the first sure bet.
1893308e39caSAaron Smith   if (auto lines =
1894308e39caSAaron Smith           m_session_up->findLineNumbersByAddress(file_vm_addr, /*Length=*/1)) {
1895308e39caSAaron Smith     if (auto first_line = lines->getNext())
1896308e39caSAaron Smith       return ParseCompileUnitForUID(first_line->getCompilandId());
18977ac1c780SAaron Smith   }
18987ac1c780SAaron Smith 
1899308e39caSAaron Smith   // Otherwise we resort to section contributions.
1900308e39caSAaron Smith   if (auto sec_contribs = m_session_up->getSectionContribs()) {
1901308e39caSAaron Smith     while (auto section = sec_contribs->getNext()) {
1902308e39caSAaron Smith       auto va = section->getVirtualAddress();
1903308e39caSAaron Smith       if (file_vm_addr >= va && file_vm_addr < va + section->getLength())
1904308e39caSAaron Smith         return ParseCompileUnitForUID(section->getCompilandId());
1905308e39caSAaron Smith     }
1906308e39caSAaron Smith   }
19077ac1c780SAaron Smith   return nullptr;
19087ac1c780SAaron Smith }
19097ac1c780SAaron Smith 
19107ac1c780SAaron Smith Mangled
1911e664b5dcSAaron Smith SymbolFilePDB::GetMangledForPDBFunc(const llvm::pdb::PDBSymbolFunc &pdb_func) {
19127ac1c780SAaron Smith   Mangled mangled;
1913e664b5dcSAaron Smith   auto func_name = pdb_func.getName();
1914e664b5dcSAaron Smith   auto func_undecorated_name = pdb_func.getUndecoratedName();
19157ac1c780SAaron Smith   std::string func_decorated_name;
19167ac1c780SAaron Smith 
19177ac1c780SAaron Smith   // Seek from public symbols for non-static function's decorated name if any.
19187ac1c780SAaron Smith   // For static functions, they don't have undecorated names and aren't exposed
19197ac1c780SAaron Smith   // in Public Symbols either.
19207ac1c780SAaron Smith   if (!func_undecorated_name.empty()) {
1921c8316ed2SAaron Smith     auto result_up = m_global_scope_up->findChildren(
1922c8316ed2SAaron Smith         PDB_SymType::PublicSymbol, func_undecorated_name,
19237ac1c780SAaron Smith         PDB_NameSearchFlags::NS_UndecoratedName);
19247ac1c780SAaron Smith     if (result_up) {
19257ac1c780SAaron Smith       while (auto symbol_up = result_up->getNext()) {
19267ac1c780SAaron Smith         // For a public symbol, it is unique.
19277ac1c780SAaron Smith         lldbassert(result_up->getChildCount() == 1);
19287ac1c780SAaron Smith         if (auto *pdb_public_sym =
1929c8316ed2SAaron Smith                 llvm::dyn_cast_or_null<PDBSymbolPublicSymbol>(
1930c8316ed2SAaron Smith                     symbol_up.get())) {
19317ac1c780SAaron Smith           if (pdb_public_sym->isFunction()) {
19327ac1c780SAaron Smith             func_decorated_name = pdb_public_sym->getName();
1933f76fe682SAaron Smith             break;
19347ac1c780SAaron Smith           }
19357ac1c780SAaron Smith         }
19367ac1c780SAaron Smith       }
19377ac1c780SAaron Smith     }
19387ac1c780SAaron Smith   }
19397ac1c780SAaron Smith   if (!func_decorated_name.empty()) {
19407ac1c780SAaron Smith     mangled.SetMangledName(ConstString(func_decorated_name));
19417ac1c780SAaron Smith 
19427ac1c780SAaron Smith     // For MSVC, format of C funciton's decorated name depends on calling
19437ac1c780SAaron Smith     // conventon. Unfortunately none of the format is recognized by current
19447ac1c780SAaron Smith     // LLDB. For example, `_purecall` is a __cdecl C function. From PDB,
194505097246SAdrian Prantl     // `__purecall` is retrieved as both its decorated and undecorated name
194605097246SAdrian Prantl     // (using PDBSymbolFunc::getUndecoratedName method). However `__purecall`
194705097246SAdrian Prantl     // string is not treated as mangled in LLDB (neither `?` nor `_Z` prefix).
194805097246SAdrian Prantl     // Mangled::GetDemangledName method will fail internally and caches an
194905097246SAdrian Prantl     // empty string as its undecorated name. So we will face a contradition
195005097246SAdrian Prantl     // here for the same symbol:
19517ac1c780SAaron Smith     //   non-empty undecorated name from PDB
19527ac1c780SAaron Smith     //   empty undecorated name from LLDB
1953*22b04487SAlex Langford     if (!func_undecorated_name.empty() && mangled.GetDemangledName().IsEmpty())
19547ac1c780SAaron Smith       mangled.SetDemangledName(ConstString(func_undecorated_name));
19557ac1c780SAaron Smith 
19567ac1c780SAaron Smith     // LLDB uses several flags to control how a C++ decorated name is
195705097246SAdrian Prantl     // undecorated for MSVC. See `safeUndecorateName` in Class Mangled. So the
195805097246SAdrian Prantl     // yielded name could be different from what we retrieve from
19597ac1c780SAaron Smith     // PDB source unless we also apply same flags in getting undecorated
19607ac1c780SAaron Smith     // name through PDBSymbolFunc::getUndecoratedNameEx method.
19617ac1c780SAaron Smith     if (!func_undecorated_name.empty() &&
1962*22b04487SAlex Langford         mangled.GetDemangledName() != ConstString(func_undecorated_name))
19637ac1c780SAaron Smith       mangled.SetDemangledName(ConstString(func_undecorated_name));
19647ac1c780SAaron Smith   } else if (!func_undecorated_name.empty()) {
19657ac1c780SAaron Smith     mangled.SetDemangledName(ConstString(func_undecorated_name));
19667ac1c780SAaron Smith   } else if (!func_name.empty())
19677ac1c780SAaron Smith     mangled.SetValue(ConstString(func_name), false);
19687ac1c780SAaron Smith 
19697ac1c780SAaron Smith   return mangled;
19707ac1c780SAaron Smith }
19717ac1c780SAaron Smith 
19727ac1c780SAaron Smith bool SymbolFilePDB::DeclContextMatchesThisSymbolFile(
19737ac1c780SAaron Smith     const lldb_private::CompilerDeclContext *decl_ctx) {
19747ac1c780SAaron Smith   if (decl_ctx == nullptr || !decl_ctx->IsValid())
19757ac1c780SAaron Smith     return true;
19767ac1c780SAaron Smith 
19777ac1c780SAaron Smith   TypeSystem *decl_ctx_type_system = decl_ctx->GetTypeSystem();
19787ac1c780SAaron Smith   if (!decl_ctx_type_system)
19797ac1c780SAaron Smith     return false;
19800e252e38SAlex Langford   auto type_system_or_err = GetTypeSystemForLanguage(
19817ac1c780SAaron Smith       decl_ctx_type_system->GetMinimumLanguage(nullptr));
19820e252e38SAlex Langford   if (auto err = type_system_or_err.takeError()) {
19830e252e38SAlex Langford     LLDB_LOG_ERROR(
19840e252e38SAlex Langford         lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_SYMBOLS),
19850e252e38SAlex Langford         std::move(err),
19860e252e38SAlex Langford         "Unable to determine if DeclContext matches this symbol file");
19870e252e38SAlex Langford     return false;
19880e252e38SAlex Langford   }
19890e252e38SAlex Langford 
19900e252e38SAlex Langford   if (decl_ctx_type_system == &type_system_or_err.get())
19917ac1c780SAaron Smith     return true; // The type systems match, return true
19927ac1c780SAaron Smith 
19937ac1c780SAaron Smith   return false;
19947ac1c780SAaron Smith }
1995356aa4a9SAleksandr Urakov 
1996356aa4a9SAleksandr Urakov uint32_t SymbolFilePDB::GetCompilandId(const llvm::pdb::PDBSymbolData &data) {
1997356aa4a9SAleksandr Urakov   static const auto pred_upper = [](uint32_t lhs, SecContribInfo rhs) {
1998356aa4a9SAleksandr Urakov     return lhs < rhs.Offset;
1999356aa4a9SAleksandr Urakov   };
2000356aa4a9SAleksandr Urakov 
2001356aa4a9SAleksandr Urakov   // Cache section contributions
2002356aa4a9SAleksandr Urakov   if (m_sec_contribs.empty()) {
2003356aa4a9SAleksandr Urakov     if (auto SecContribs = m_session_up->getSectionContribs()) {
2004356aa4a9SAleksandr Urakov       while (auto SectionContrib = SecContribs->getNext()) {
2005356aa4a9SAleksandr Urakov         auto comp_id = SectionContrib->getCompilandId();
2006356aa4a9SAleksandr Urakov         if (!comp_id)
2007356aa4a9SAleksandr Urakov           continue;
2008356aa4a9SAleksandr Urakov 
2009356aa4a9SAleksandr Urakov         auto sec = SectionContrib->getAddressSection();
2010356aa4a9SAleksandr Urakov         auto &sec_cs = m_sec_contribs[sec];
2011356aa4a9SAleksandr Urakov 
2012356aa4a9SAleksandr Urakov         auto offset = SectionContrib->getAddressOffset();
2013356aa4a9SAleksandr Urakov         auto it =
2014356aa4a9SAleksandr Urakov             std::upper_bound(sec_cs.begin(), sec_cs.end(), offset, pred_upper);
2015356aa4a9SAleksandr Urakov 
2016356aa4a9SAleksandr Urakov         auto size = SectionContrib->getLength();
2017356aa4a9SAleksandr Urakov         sec_cs.insert(it, {offset, size, comp_id});
2018356aa4a9SAleksandr Urakov       }
2019356aa4a9SAleksandr Urakov     }
2020356aa4a9SAleksandr Urakov   }
2021356aa4a9SAleksandr Urakov 
2022356aa4a9SAleksandr Urakov   // Check by line number
2023356aa4a9SAleksandr Urakov   if (auto Lines = data.getLineNumbers()) {
2024356aa4a9SAleksandr Urakov     if (auto FirstLine = Lines->getNext())
2025356aa4a9SAleksandr Urakov       return FirstLine->getCompilandId();
2026356aa4a9SAleksandr Urakov   }
2027356aa4a9SAleksandr Urakov 
2028356aa4a9SAleksandr Urakov   // Retrieve section + offset
2029356aa4a9SAleksandr Urakov   uint32_t DataSection = data.getAddressSection();
2030356aa4a9SAleksandr Urakov   uint32_t DataOffset = data.getAddressOffset();
2031356aa4a9SAleksandr Urakov   if (DataSection == 0) {
2032356aa4a9SAleksandr Urakov     if (auto RVA = data.getRelativeVirtualAddress())
2033356aa4a9SAleksandr Urakov       m_session_up->addressForRVA(RVA, DataSection, DataOffset);
2034356aa4a9SAleksandr Urakov   }
2035356aa4a9SAleksandr Urakov 
2036356aa4a9SAleksandr Urakov   if (DataSection) {
2037356aa4a9SAleksandr Urakov     // Search by section contributions
2038356aa4a9SAleksandr Urakov     auto &sec_cs = m_sec_contribs[DataSection];
2039356aa4a9SAleksandr Urakov     auto it =
2040356aa4a9SAleksandr Urakov         std::upper_bound(sec_cs.begin(), sec_cs.end(), DataOffset, pred_upper);
2041356aa4a9SAleksandr Urakov     if (it != sec_cs.begin()) {
2042356aa4a9SAleksandr Urakov       --it;
2043356aa4a9SAleksandr Urakov       if (DataOffset < it->Offset + it->Size)
2044356aa4a9SAleksandr Urakov         return it->CompilandId;
2045356aa4a9SAleksandr Urakov     }
2046356aa4a9SAleksandr Urakov   } else {
2047356aa4a9SAleksandr Urakov     // Search in lexical tree
2048356aa4a9SAleksandr Urakov     auto LexParentId = data.getLexicalParentId();
2049356aa4a9SAleksandr Urakov     while (auto LexParent = m_session_up->getSymbolById(LexParentId)) {
2050356aa4a9SAleksandr Urakov       if (LexParent->getSymTag() == PDB_SymType::Exe)
2051356aa4a9SAleksandr Urakov         break;
2052356aa4a9SAleksandr Urakov       if (LexParent->getSymTag() == PDB_SymType::Compiland)
2053356aa4a9SAleksandr Urakov         return LexParentId;
2054356aa4a9SAleksandr Urakov       LexParentId = LexParent->getRawSymbol().getLexicalParentId();
2055356aa4a9SAleksandr Urakov     }
2056356aa4a9SAleksandr Urakov   }
2057356aa4a9SAleksandr Urakov 
2058356aa4a9SAleksandr Urakov   return 0;
2059356aa4a9SAleksandr Urakov }
2060