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 ®ex, 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 ®ex, 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 = [®ex, 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 ®ex, 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