xref: /src/contrib/llvm-project/lldb/source/Plugins/SymbolFile/PDB/SymbolFilePDB.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1cfca06d7SDimitry Andric //===-- SymbolFilePDB.cpp -------------------------------------------------===//
2f3fbd1c0SDimitry Andric //
35f29bb8aSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
45f29bb8aSDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
55f29bb8aSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6f3fbd1c0SDimitry Andric //
7f3fbd1c0SDimitry Andric //===----------------------------------------------------------------------===//
8f3fbd1c0SDimitry Andric 
9f3fbd1c0SDimitry Andric #include "SymbolFilePDB.h"
10f3fbd1c0SDimitry Andric 
1194994d37SDimitry Andric #include "PDBASTParser.h"
1294994d37SDimitry Andric #include "PDBLocationToDWARFExpression.h"
1394994d37SDimitry Andric 
14f3fbd1c0SDimitry Andric #include "clang/Lex/Lexer.h"
15f3fbd1c0SDimitry Andric 
16cfca06d7SDimitry Andric #include "Plugins/TypeSystem/Clang/TypeSystemClang.h"
17f3fbd1c0SDimitry Andric #include "lldb/Core/Module.h"
18f3fbd1c0SDimitry Andric #include "lldb/Core/PluginManager.h"
19f3fbd1c0SDimitry Andric #include "lldb/Symbol/CompileUnit.h"
20f3fbd1c0SDimitry Andric #include "lldb/Symbol/LineTable.h"
21f3fbd1c0SDimitry Andric #include "lldb/Symbol/ObjectFile.h"
22f3fbd1c0SDimitry Andric #include "lldb/Symbol/SymbolContext.h"
23f73363f1SDimitry Andric #include "lldb/Symbol/SymbolVendor.h"
24f73363f1SDimitry Andric #include "lldb/Symbol/TypeList.h"
25f3fbd1c0SDimitry Andric #include "lldb/Symbol/TypeMap.h"
26f73363f1SDimitry Andric #include "lldb/Symbol/Variable.h"
27145449b1SDimitry Andric #include "lldb/Utility/LLDBLog.h"
28ead24645SDimitry Andric #include "lldb/Utility/Log.h"
2923629167SDimitry Andric #include "lldb/Utility/RegularExpression.h"
30f3fbd1c0SDimitry Andric 
31145449b1SDimitry Andric #include "llvm/DebugInfo/PDB/ConcreteSymbolEnumerator.h"
32f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/GenericError.h"
3323629167SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBDataStream.h"
34f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBEnumChildren.h"
35f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBLineNumber.h"
36f73363f1SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBSectionContrib.h"
37f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBSourceFile.h"
3823629167SDimitry Andric #include "llvm/DebugInfo/PDB/IPDBTable.h"
39f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbol.h"
40f73363f1SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolBlock.h"
41f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
42f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolCompilandDetails.h"
4323629167SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolData.h"
44f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
45f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolFunc.h"
46f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugEnd.h"
47f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugStart.h"
48f73363f1SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolPublicSymbol.h"
49145449b1SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolTypeBuiltin.h"
50f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolTypeEnum.h"
51145449b1SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolTypeFunctionSig.h"
52f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolTypeTypedef.h"
53f3fbd1c0SDimitry Andric #include "llvm/DebugInfo/PDB/PDBSymbolTypeUDT.h"
54f3fbd1c0SDimitry Andric 
55f73363f1SDimitry Andric #include "Plugins/Language/CPlusPlus/CPlusPlusLanguage.h"
5694994d37SDimitry Andric #include "Plugins/Language/CPlusPlus/MSVCUndecoratedNameParser.h"
5794994d37SDimitry Andric #include "Plugins/SymbolFile/NativePDB/SymbolFileNativePDB.h"
58f3fbd1c0SDimitry Andric 
59c0981da4SDimitry Andric #if defined(_WIN32)
60145449b1SDimitry Andric #include "llvm/Config/llvm-config.h"
61e3b55780SDimitry Andric #include <optional>
62c0981da4SDimitry Andric #endif
63c0981da4SDimitry Andric 
64f73363f1SDimitry Andric using namespace lldb;
65f3fbd1c0SDimitry Andric using namespace lldb_private;
66f3fbd1c0SDimitry Andric using namespace llvm::pdb;
67f3fbd1c0SDimitry Andric 
68cfca06d7SDimitry Andric LLDB_PLUGIN_DEFINE(SymbolFilePDB)
69cfca06d7SDimitry Andric 
70706b4fc4SDimitry Andric char SymbolFilePDB::ID;
71706b4fc4SDimitry Andric 
7214f1b3e8SDimitry Andric namespace {
TranslateLanguage(PDB_Lang lang)7314f1b3e8SDimitry Andric lldb::LanguageType TranslateLanguage(PDB_Lang lang) {
7414f1b3e8SDimitry Andric   switch (lang) {
75f3fbd1c0SDimitry Andric   case PDB_Lang::Cpp:
76f3fbd1c0SDimitry Andric     return lldb::LanguageType::eLanguageTypeC_plus_plus;
77f3fbd1c0SDimitry Andric   case PDB_Lang::C:
78f3fbd1c0SDimitry Andric     return lldb::LanguageType::eLanguageTypeC;
795f29bb8aSDimitry Andric   case PDB_Lang::Swift:
805f29bb8aSDimitry Andric     return lldb::LanguageType::eLanguageTypeSwift;
81145449b1SDimitry Andric   case PDB_Lang::Rust:
82145449b1SDimitry Andric     return lldb::LanguageType::eLanguageTypeRust;
837fa27ce4SDimitry Andric   case PDB_Lang::ObjC:
847fa27ce4SDimitry Andric     return lldb::LanguageType::eLanguageTypeObjC;
857fa27ce4SDimitry Andric   case PDB_Lang::ObjCpp:
867fa27ce4SDimitry Andric     return lldb::LanguageType::eLanguageTypeObjC_plus_plus;
87f3fbd1c0SDimitry Andric   default:
88f3fbd1c0SDimitry Andric     return lldb::LanguageType::eLanguageTypeUnknown;
89f3fbd1c0SDimitry Andric   }
90f3fbd1c0SDimitry Andric }
91f3fbd1c0SDimitry Andric 
ShouldAddLine(uint32_t requested_line,uint32_t actual_line,uint32_t addr_length)9214f1b3e8SDimitry Andric bool ShouldAddLine(uint32_t requested_line, uint32_t actual_line,
9314f1b3e8SDimitry Andric                    uint32_t addr_length) {
9414f1b3e8SDimitry Andric   return ((requested_line == 0 || actual_line == requested_line) &&
9514f1b3e8SDimitry Andric           addr_length > 0);
96f3fbd1c0SDimitry Andric }
97f73363f1SDimitry Andric } // namespace
98f3fbd1c0SDimitry Andric 
ShouldUseNativeReader()9994994d37SDimitry Andric static bool ShouldUseNativeReader() {
10094994d37SDimitry Andric #if defined(_WIN32)
101c0981da4SDimitry Andric #if LLVM_ENABLE_DIA_SDK
10294994d37SDimitry Andric   llvm::StringRef use_native = ::getenv("LLDB_USE_NATIVE_PDB_READER");
103c0981da4SDimitry Andric   if (!use_native.equals_insensitive("on") &&
104c0981da4SDimitry Andric       !use_native.equals_insensitive("yes") &&
105c0981da4SDimitry Andric       !use_native.equals_insensitive("1") &&
106c0981da4SDimitry Andric       !use_native.equals_insensitive("true"))
107c0981da4SDimitry Andric     return false;
10894994d37SDimitry Andric #endif
109c0981da4SDimitry Andric #endif
110c0981da4SDimitry Andric   return true;
11194994d37SDimitry Andric }
11294994d37SDimitry Andric 
Initialize()11314f1b3e8SDimitry Andric void SymbolFilePDB::Initialize() {
11494994d37SDimitry Andric   if (ShouldUseNativeReader()) {
11594994d37SDimitry Andric     npdb::SymbolFileNativePDB::Initialize();
11694994d37SDimitry Andric   } else {
11714f1b3e8SDimitry Andric     PluginManager::RegisterPlugin(GetPluginNameStatic(),
11814f1b3e8SDimitry Andric                                   GetPluginDescriptionStatic(), CreateInstance,
119f3fbd1c0SDimitry Andric                                   DebuggerInitialize);
120f3fbd1c0SDimitry Andric   }
12194994d37SDimitry Andric }
122f3fbd1c0SDimitry Andric 
Terminate()12314f1b3e8SDimitry Andric void SymbolFilePDB::Terminate() {
12494994d37SDimitry Andric   if (ShouldUseNativeReader()) {
12594994d37SDimitry Andric     npdb::SymbolFileNativePDB::Terminate();
12694994d37SDimitry Andric   } else {
127f3fbd1c0SDimitry Andric     PluginManager::UnregisterPlugin(CreateInstance);
128f3fbd1c0SDimitry Andric   }
12994994d37SDimitry Andric }
130f3fbd1c0SDimitry Andric 
DebuggerInitialize(lldb_private::Debugger & debugger)13114f1b3e8SDimitry Andric void SymbolFilePDB::DebuggerInitialize(lldb_private::Debugger &debugger) {}
132f3fbd1c0SDimitry Andric 
GetPluginDescriptionStatic()133c0981da4SDimitry Andric llvm::StringRef SymbolFilePDB::GetPluginDescriptionStatic() {
134f3fbd1c0SDimitry Andric   return "Microsoft PDB debug symbol file reader.";
135f3fbd1c0SDimitry Andric }
136f3fbd1c0SDimitry Andric 
137f3fbd1c0SDimitry Andric lldb_private::SymbolFile *
CreateInstance(ObjectFileSP objfile_sp)138ead24645SDimitry Andric SymbolFilePDB::CreateInstance(ObjectFileSP objfile_sp) {
139ead24645SDimitry Andric   return new SymbolFilePDB(std::move(objfile_sp));
140f3fbd1c0SDimitry Andric }
141f3fbd1c0SDimitry Andric 
SymbolFilePDB(lldb::ObjectFileSP objfile_sp)142ead24645SDimitry Andric SymbolFilePDB::SymbolFilePDB(lldb::ObjectFileSP objfile_sp)
143145449b1SDimitry Andric     : SymbolFileCommon(std::move(objfile_sp)), m_session_up(), m_global_scope_up() {}
144f3fbd1c0SDimitry Andric 
145344a3780SDimitry Andric SymbolFilePDB::~SymbolFilePDB() = default;
146f3fbd1c0SDimitry Andric 
CalculateAbilities()14714f1b3e8SDimitry Andric uint32_t SymbolFilePDB::CalculateAbilities() {
14823629167SDimitry Andric   uint32_t abilities = 0;
149ead24645SDimitry Andric   if (!m_objfile_sp)
15023629167SDimitry Andric     return 0;
15123629167SDimitry Andric 
15214f1b3e8SDimitry Andric   if (!m_session_up) {
153f3fbd1c0SDimitry Andric     // Lazily load and match the PDB file, but only do this once.
154ead24645SDimitry Andric     std::string exePath = m_objfile_sp->GetFileSpec().GetPath();
15514f1b3e8SDimitry Andric     auto error = loadDataForEXE(PDB_ReaderType::DIA, llvm::StringRef(exePath),
15614f1b3e8SDimitry Andric                                 m_session_up);
15714f1b3e8SDimitry Andric     if (error) {
158f3fbd1c0SDimitry Andric       llvm::consumeError(std::move(error));
159ead24645SDimitry Andric       auto module_sp = m_objfile_sp->GetModule();
16023629167SDimitry Andric       if (!module_sp)
16123629167SDimitry Andric         return 0;
16223629167SDimitry Andric       // See if any symbol file is specified through `--symfile` option.
16323629167SDimitry Andric       FileSpec symfile = module_sp->GetSymbolFileFileSpec();
16423629167SDimitry Andric       if (!symfile)
16523629167SDimitry Andric         return 0;
16623629167SDimitry Andric       error = loadDataForPDB(PDB_ReaderType::DIA,
167f73363f1SDimitry Andric                              llvm::StringRef(symfile.GetPath()), m_session_up);
16823629167SDimitry Andric       if (error) {
16923629167SDimitry Andric         llvm::consumeError(std::move(error));
170f3fbd1c0SDimitry Andric         return 0;
171f3fbd1c0SDimitry Andric       }
172f3fbd1c0SDimitry Andric     }
17323629167SDimitry Andric   }
174f73363f1SDimitry Andric   if (!m_session_up)
17523629167SDimitry Andric     return 0;
17623629167SDimitry Andric 
17723629167SDimitry Andric   auto enum_tables_up = m_session_up->getEnumTables();
17823629167SDimitry Andric   if (!enum_tables_up)
17923629167SDimitry Andric     return 0;
18023629167SDimitry Andric   while (auto table_up = enum_tables_up->getNext()) {
18123629167SDimitry Andric     if (table_up->getItemCount() == 0)
18223629167SDimitry Andric       continue;
18323629167SDimitry Andric     auto type = table_up->getTableType();
18423629167SDimitry Andric     switch (type) {
18523629167SDimitry Andric     case PDB_TableType::Symbols:
18623629167SDimitry Andric       // This table represents a store of symbols with types listed in
18723629167SDimitry Andric       // PDBSym_Type
188f73363f1SDimitry Andric       abilities |= (CompileUnits | Functions | Blocks | GlobalVariables |
189f73363f1SDimitry Andric                     LocalVariables | VariableTypes);
19023629167SDimitry Andric       break;
19123629167SDimitry Andric     case PDB_TableType::LineNumbers:
19223629167SDimitry Andric       abilities |= LineTables;
19323629167SDimitry Andric       break;
194f73363f1SDimitry Andric     default:
195f73363f1SDimitry Andric       break;
19623629167SDimitry Andric     }
19723629167SDimitry Andric   }
19823629167SDimitry Andric   return abilities;
199f3fbd1c0SDimitry Andric }
200f3fbd1c0SDimitry Andric 
InitializeObject()20114f1b3e8SDimitry Andric void SymbolFilePDB::InitializeObject() {
202ead24645SDimitry Andric   lldb::addr_t obj_load_address =
203ead24645SDimitry Andric       m_objfile_sp->GetBaseAddress().GetFileAddress();
204f73363f1SDimitry Andric   lldbassert(obj_load_address && obj_load_address != LLDB_INVALID_ADDRESS);
205f3fbd1c0SDimitry Andric   m_session_up->setLoadAddress(obj_load_address);
206f73363f1SDimitry Andric   if (!m_global_scope_up)
207f73363f1SDimitry Andric     m_global_scope_up = m_session_up->getGlobalScope();
208f73363f1SDimitry Andric   lldbassert(m_global_scope_up.get());
209f3fbd1c0SDimitry Andric }
210f3fbd1c0SDimitry Andric 
CalculateNumCompileUnits()211ead24645SDimitry Andric uint32_t SymbolFilePDB::CalculateNumCompileUnits() {
212f73363f1SDimitry Andric   auto compilands = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
213f73363f1SDimitry Andric   if (!compilands)
214f73363f1SDimitry Andric     return 0;
215f73363f1SDimitry Andric 
216f73363f1SDimitry Andric   // The linker could link *.dll (compiland language = LINK), or import
217f73363f1SDimitry Andric   // *.dll. For example, a compiland with name `Import:KERNEL32.dll` could be
218f73363f1SDimitry Andric   // found as a child of the global scope (PDB executable). Usually, such
219f73363f1SDimitry Andric   // compilands contain `thunk` symbols in which we are not interested for
220f73363f1SDimitry Andric   // now. However we still count them in the compiland list. If we perform
221f73363f1SDimitry Andric   // any compiland related activity, like finding symbols through
222f73363f1SDimitry Andric   // llvm::pdb::IPDBSession methods, such compilands will all be searched
223f73363f1SDimitry Andric   // automatically no matter whether we include them or not.
224ead24645SDimitry Andric   uint32_t compile_unit_count = compilands->getChildCount();
225f3fbd1c0SDimitry Andric 
22614f1b3e8SDimitry Andric   // The linker can inject an additional "dummy" compilation unit into the
22774a628f7SDimitry Andric   // PDB. Ignore this special compile unit for our purposes, if it is there.
22874a628f7SDimitry Andric   // It is always the last one.
229ead24645SDimitry Andric   auto last_compiland_up = compilands->getChildAtIndex(compile_unit_count - 1);
230f73363f1SDimitry Andric   lldbassert(last_compiland_up.get());
231f73363f1SDimitry Andric   std::string name = last_compiland_up->getName();
232f3fbd1c0SDimitry Andric   if (name == "* Linker *")
233ead24645SDimitry Andric     --compile_unit_count;
234ead24645SDimitry Andric   return compile_unit_count;
235f3fbd1c0SDimitry Andric }
236f3fbd1c0SDimitry Andric 
GetCompileUnitIndex(const llvm::pdb::PDBSymbolCompiland & pdb_compiland,uint32_t & index)237f73363f1SDimitry Andric void SymbolFilePDB::GetCompileUnitIndex(
238f73363f1SDimitry Andric     const llvm::pdb::PDBSymbolCompiland &pdb_compiland, uint32_t &index) {
239f73363f1SDimitry Andric   auto results_up = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
240f73363f1SDimitry Andric   if (!results_up)
241f73363f1SDimitry Andric     return;
242f73363f1SDimitry Andric   auto uid = pdb_compiland.getSymIndexId();
243f73363f1SDimitry Andric   for (uint32_t cu_idx = 0; cu_idx < GetNumCompileUnits(); ++cu_idx) {
244f73363f1SDimitry Andric     auto compiland_up = results_up->getChildAtIndex(cu_idx);
245f73363f1SDimitry Andric     if (!compiland_up)
246f73363f1SDimitry Andric       continue;
247f73363f1SDimitry Andric     if (compiland_up->getSymIndexId() == uid) {
248f73363f1SDimitry Andric       index = cu_idx;
249f73363f1SDimitry Andric       return;
250f73363f1SDimitry Andric     }
251f73363f1SDimitry Andric   }
252f73363f1SDimitry Andric   index = UINT32_MAX;
253f73363f1SDimitry Andric }
254f73363f1SDimitry Andric 
255f73363f1SDimitry Andric std::unique_ptr<llvm::pdb::PDBSymbolCompiland>
GetPDBCompilandByUID(uint32_t uid)256f73363f1SDimitry Andric SymbolFilePDB::GetPDBCompilandByUID(uint32_t uid) {
257f73363f1SDimitry Andric   return m_session_up->getConcreteSymbolById<PDBSymbolCompiland>(uid);
258f73363f1SDimitry Andric }
259f73363f1SDimitry Andric 
ParseCompileUnitAtIndex(uint32_t index)26014f1b3e8SDimitry Andric lldb::CompUnitSP SymbolFilePDB::ParseCompileUnitAtIndex(uint32_t index) {
261f73363f1SDimitry Andric   if (index >= GetNumCompileUnits())
262f73363f1SDimitry Andric     return CompUnitSP();
263f3fbd1c0SDimitry Andric 
264f73363f1SDimitry Andric   // Assuming we always retrieve same compilands listed in same order through
265f73363f1SDimitry Andric   // `PDBSymbolExe::findAllChildren` method, otherwise using `index` to get a
266f73363f1SDimitry Andric   // compile unit makes no sense.
267f73363f1SDimitry Andric   auto results = m_global_scope_up->findAllChildren<PDBSymbolCompiland>();
268f73363f1SDimitry Andric   if (!results)
269f73363f1SDimitry Andric     return CompUnitSP();
270f73363f1SDimitry Andric   auto compiland_up = results->getChildAtIndex(index);
271f73363f1SDimitry Andric   if (!compiland_up)
272f73363f1SDimitry Andric     return CompUnitSP();
273f73363f1SDimitry Andric   return ParseCompileUnitForUID(compiland_up->getSymIndexId(), index);
274f3fbd1c0SDimitry Andric }
275f3fbd1c0SDimitry Andric 
ParseLanguage(CompileUnit & comp_unit)27694994d37SDimitry Andric lldb::LanguageType SymbolFilePDB::ParseLanguage(CompileUnit &comp_unit) {
277ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
27894994d37SDimitry Andric   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
279f73363f1SDimitry Andric   if (!compiland_up)
280f3fbd1c0SDimitry Andric     return lldb::eLanguageTypeUnknown;
281f73363f1SDimitry Andric   auto details = compiland_up->findOneChild<PDBSymbolCompilandDetails>();
282f3fbd1c0SDimitry Andric   if (!details)
283f3fbd1c0SDimitry Andric     return lldb::eLanguageTypeUnknown;
284f3fbd1c0SDimitry Andric   return TranslateLanguage(details->getLanguage());
285f3fbd1c0SDimitry Andric }
286f3fbd1c0SDimitry Andric 
28794994d37SDimitry Andric lldb_private::Function *
ParseCompileUnitFunctionForPDBFunc(const PDBSymbolFunc & pdb_func,CompileUnit & comp_unit)28894994d37SDimitry Andric SymbolFilePDB::ParseCompileUnitFunctionForPDBFunc(const PDBSymbolFunc &pdb_func,
28994994d37SDimitry Andric                                                   CompileUnit &comp_unit) {
29094994d37SDimitry Andric   if (FunctionSP result = comp_unit.FindFunctionByUID(pdb_func.getSymIndexId()))
29194994d37SDimitry Andric     return result.get();
292f73363f1SDimitry Andric 
293f73363f1SDimitry Andric   auto file_vm_addr = pdb_func.getVirtualAddress();
294f73363f1SDimitry Andric   if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
295f73363f1SDimitry Andric     return nullptr;
296f73363f1SDimitry Andric 
297f73363f1SDimitry Andric   auto func_length = pdb_func.getLength();
298f73363f1SDimitry Andric   AddressRange func_range =
29994994d37SDimitry Andric       AddressRange(file_vm_addr, func_length,
30094994d37SDimitry Andric                    GetObjectFile()->GetModule()->GetSectionList());
301f73363f1SDimitry Andric   if (!func_range.GetBaseAddress().IsValid())
302f73363f1SDimitry Andric     return nullptr;
303f73363f1SDimitry Andric 
304f73363f1SDimitry Andric   lldb_private::Type *func_type = ResolveTypeUID(pdb_func.getSymIndexId());
305f73363f1SDimitry Andric   if (!func_type)
306f73363f1SDimitry Andric     return nullptr;
307f73363f1SDimitry Andric 
308f73363f1SDimitry Andric   user_id_t func_type_uid = pdb_func.getSignatureId();
309f73363f1SDimitry Andric 
310f73363f1SDimitry Andric   Mangled mangled = GetMangledForPDBFunc(pdb_func);
311f73363f1SDimitry Andric 
312f73363f1SDimitry Andric   FunctionSP func_sp =
31394994d37SDimitry Andric       std::make_shared<Function>(&comp_unit, pdb_func.getSymIndexId(),
314f73363f1SDimitry Andric                                  func_type_uid, mangled, func_type, func_range);
315f73363f1SDimitry Andric 
31694994d37SDimitry Andric   comp_unit.AddFunction(func_sp);
31794994d37SDimitry Andric 
3185f29bb8aSDimitry Andric   LanguageType lang = ParseLanguage(comp_unit);
319ead24645SDimitry Andric   auto type_system_or_err = GetTypeSystemForLanguage(lang);
320ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
321145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
3227fa27ce4SDimitry Andric                    "Unable to parse PDBFunc: {0}");
32394994d37SDimitry Andric     return nullptr;
324ead24645SDimitry Andric   }
325ead24645SDimitry Andric 
326e3b55780SDimitry Andric   auto ts = *type_system_or_err;
327cfca06d7SDimitry Andric   TypeSystemClang *clang_type_system =
328e3b55780SDimitry Andric     llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
32994994d37SDimitry Andric   if (!clang_type_system)
33094994d37SDimitry Andric     return nullptr;
33194994d37SDimitry Andric   clang_type_system->GetPDBParser()->GetDeclForSymbol(pdb_func);
33294994d37SDimitry Andric 
333f73363f1SDimitry Andric   return func_sp.get();
334f73363f1SDimitry Andric }
335f73363f1SDimitry Andric 
ParseFunctions(CompileUnit & comp_unit)33694994d37SDimitry Andric size_t SymbolFilePDB::ParseFunctions(CompileUnit &comp_unit) {
337ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
338f73363f1SDimitry Andric   size_t func_added = 0;
33994994d37SDimitry Andric   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
340f73363f1SDimitry Andric   if (!compiland_up)
341f73363f1SDimitry Andric     return 0;
342f73363f1SDimitry Andric   auto results_up = compiland_up->findAllChildren<PDBSymbolFunc>();
343f73363f1SDimitry Andric   if (!results_up)
344f73363f1SDimitry Andric     return 0;
345f73363f1SDimitry Andric   while (auto pdb_func_up = results_up->getNext()) {
34694994d37SDimitry Andric     auto func_sp = comp_unit.FindFunctionByUID(pdb_func_up->getSymIndexId());
347f73363f1SDimitry Andric     if (!func_sp) {
34894994d37SDimitry Andric       if (ParseCompileUnitFunctionForPDBFunc(*pdb_func_up, comp_unit))
349f73363f1SDimitry Andric         ++func_added;
350f73363f1SDimitry Andric     }
351f73363f1SDimitry Andric   }
352f73363f1SDimitry Andric   return func_added;
353f3fbd1c0SDimitry Andric }
354f3fbd1c0SDimitry Andric 
ParseLineTable(CompileUnit & comp_unit)35594994d37SDimitry Andric bool SymbolFilePDB::ParseLineTable(CompileUnit &comp_unit) {
356ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
35794994d37SDimitry Andric   if (comp_unit.GetLineTable())
358f73363f1SDimitry Andric     return true;
35994994d37SDimitry Andric   return ParseCompileUnitLineTable(comp_unit, 0);
360f3fbd1c0SDimitry Andric }
361f3fbd1c0SDimitry Andric 
ParseDebugMacros(CompileUnit & comp_unit)36294994d37SDimitry Andric bool SymbolFilePDB::ParseDebugMacros(CompileUnit &comp_unit) {
363f3fbd1c0SDimitry Andric   // PDB doesn't contain information about macros
364f3fbd1c0SDimitry Andric   return false;
365f3fbd1c0SDimitry Andric }
366f3fbd1c0SDimitry Andric 
ParseSupportFiles(CompileUnit & comp_unit,lldb_private::SupportFileList & support_files)36794994d37SDimitry Andric bool SymbolFilePDB::ParseSupportFiles(
368aca2e42cSDimitry Andric     CompileUnit &comp_unit, lldb_private::SupportFileList &support_files) {
369f3fbd1c0SDimitry Andric 
37014f1b3e8SDimitry Andric   // In theory this is unnecessary work for us, because all of this information
37174a628f7SDimitry Andric   // is easily (and quickly) accessible from DebugInfoPDB, so caching it a
37274a628f7SDimitry Andric   // second time seems like a waste.  Unfortunately, there's no good way around
37374a628f7SDimitry Andric   // this short of a moderate refactor since SymbolVendor depends on being able
37474a628f7SDimitry Andric   // to cache this list.
375ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
37694994d37SDimitry Andric   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
377f73363f1SDimitry Andric   if (!compiland_up)
378f3fbd1c0SDimitry Andric     return false;
379f73363f1SDimitry Andric   auto files = m_session_up->getSourceFilesForCompiland(*compiland_up);
380f3fbd1c0SDimitry Andric   if (!files || files->getChildCount() == 0)
381f3fbd1c0SDimitry Andric     return false;
382f3fbd1c0SDimitry Andric 
38314f1b3e8SDimitry Andric   while (auto file = files->getNext()) {
38494994d37SDimitry Andric     FileSpec spec(file->getFileName(), FileSpec::Style::windows);
385f73363f1SDimitry Andric     support_files.AppendIfUnique(spec);
386f3fbd1c0SDimitry Andric   }
387f73363f1SDimitry Andric 
388f3fbd1c0SDimitry Andric   return true;
389f3fbd1c0SDimitry Andric }
390f3fbd1c0SDimitry Andric 
ParseImportedModules(const lldb_private::SymbolContext & sc,std::vector<SourceModule> & imported_modules)39114f1b3e8SDimitry Andric bool SymbolFilePDB::ParseImportedModules(
39214f1b3e8SDimitry Andric     const lldb_private::SymbolContext &sc,
3935f29bb8aSDimitry Andric     std::vector<SourceModule> &imported_modules) {
394f3fbd1c0SDimitry Andric   // PDB does not yet support module debug info
395f3fbd1c0SDimitry Andric   return false;
396f3fbd1c0SDimitry Andric }
397f3fbd1c0SDimitry Andric 
ParseFunctionBlocksForPDBSymbol(uint64_t func_file_vm_addr,const llvm::pdb::PDBSymbol * pdb_symbol,lldb_private::Block * parent_block,bool is_top_parent)398f73363f1SDimitry Andric static size_t ParseFunctionBlocksForPDBSymbol(
39994994d37SDimitry Andric     uint64_t func_file_vm_addr, const llvm::pdb::PDBSymbol *pdb_symbol,
40094994d37SDimitry Andric     lldb_private::Block *parent_block, bool is_top_parent) {
401f73363f1SDimitry Andric   assert(pdb_symbol && parent_block);
402f73363f1SDimitry Andric 
403f73363f1SDimitry Andric   size_t num_added = 0;
404f73363f1SDimitry Andric   switch (pdb_symbol->getSymTag()) {
405f73363f1SDimitry Andric   case PDB_SymType::Block:
406f73363f1SDimitry Andric   case PDB_SymType::Function: {
407f73363f1SDimitry Andric     Block *block = nullptr;
408f73363f1SDimitry Andric     auto &raw_sym = pdb_symbol->getRawSymbol();
409f73363f1SDimitry Andric     if (auto *pdb_func = llvm::dyn_cast<PDBSymbolFunc>(pdb_symbol)) {
410f73363f1SDimitry Andric       if (pdb_func->hasNoInlineAttribute())
411f73363f1SDimitry Andric         break;
412f73363f1SDimitry Andric       if (is_top_parent)
413f73363f1SDimitry Andric         block = parent_block;
414f73363f1SDimitry Andric       else
415f73363f1SDimitry Andric         break;
41677fc4c14SDimitry Andric     } else if (llvm::isa<PDBSymbolBlock>(pdb_symbol)) {
417f73363f1SDimitry Andric       auto uid = pdb_symbol->getSymIndexId();
418f73363f1SDimitry Andric       if (parent_block->FindBlockByID(uid))
419f73363f1SDimitry Andric         break;
420f73363f1SDimitry Andric       if (raw_sym.getVirtualAddress() < func_file_vm_addr)
421f73363f1SDimitry Andric         break;
422f73363f1SDimitry Andric 
423f73363f1SDimitry Andric       auto block_sp = std::make_shared<Block>(pdb_symbol->getSymIndexId());
424f73363f1SDimitry Andric       parent_block->AddChild(block_sp);
425f73363f1SDimitry Andric       block = block_sp.get();
426f73363f1SDimitry Andric     } else
427f73363f1SDimitry Andric       llvm_unreachable("Unexpected PDB symbol!");
428f73363f1SDimitry Andric 
429f73363f1SDimitry Andric     block->AddRange(Block::Range(
430f73363f1SDimitry Andric         raw_sym.getVirtualAddress() - func_file_vm_addr, raw_sym.getLength()));
431f73363f1SDimitry Andric     block->FinalizeRanges();
432f73363f1SDimitry Andric     ++num_added;
433f73363f1SDimitry Andric 
434f73363f1SDimitry Andric     auto results_up = pdb_symbol->findAllChildren();
435f73363f1SDimitry Andric     if (!results_up)
436f73363f1SDimitry Andric       break;
437f73363f1SDimitry Andric     while (auto symbol_up = results_up->getNext()) {
438f73363f1SDimitry Andric       num_added += ParseFunctionBlocksForPDBSymbol(
43994994d37SDimitry Andric           func_file_vm_addr, symbol_up.get(), block, false);
440f73363f1SDimitry Andric     }
441f73363f1SDimitry Andric   } break;
442f73363f1SDimitry Andric   default:
443f73363f1SDimitry Andric     break;
444f73363f1SDimitry Andric   }
445f73363f1SDimitry Andric   return num_added;
446f73363f1SDimitry Andric }
447f73363f1SDimitry Andric 
ParseBlocksRecursive(Function & func)44894994d37SDimitry Andric size_t SymbolFilePDB::ParseBlocksRecursive(Function &func) {
449ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
450f73363f1SDimitry Andric   size_t num_added = 0;
45194994d37SDimitry Andric   auto uid = func.GetID();
452f73363f1SDimitry Andric   auto pdb_func_up = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(uid);
453f73363f1SDimitry Andric   if (!pdb_func_up)
454f73363f1SDimitry Andric     return 0;
45594994d37SDimitry Andric   Block &parent_block = func.GetBlock(false);
45694994d37SDimitry Andric   num_added = ParseFunctionBlocksForPDBSymbol(
45794994d37SDimitry Andric       pdb_func_up->getVirtualAddress(), pdb_func_up.get(), &parent_block, true);
458f73363f1SDimitry Andric   return num_added;
45914f1b3e8SDimitry Andric }
46014f1b3e8SDimitry Andric 
ParseTypes(CompileUnit & comp_unit)46194994d37SDimitry Andric size_t SymbolFilePDB::ParseTypes(CompileUnit &comp_unit) {
462ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
463f73363f1SDimitry Andric 
464f73363f1SDimitry Andric   size_t num_added = 0;
46594994d37SDimitry Andric   auto compiland = GetPDBCompilandByUID(comp_unit.GetID());
466f73363f1SDimitry Andric   if (!compiland)
467f73363f1SDimitry Andric     return 0;
468f73363f1SDimitry Andric 
469f73363f1SDimitry Andric   auto ParseTypesByTagFn = [&num_added, this](const PDBSymbol &raw_sym) {
470f73363f1SDimitry Andric     std::unique_ptr<IPDBEnumSymbols> results;
471f73363f1SDimitry Andric     PDB_SymType tags_to_search[] = {PDB_SymType::Enum, PDB_SymType::Typedef,
472f73363f1SDimitry Andric                                     PDB_SymType::UDT};
473f73363f1SDimitry Andric     for (auto tag : tags_to_search) {
474f73363f1SDimitry Andric       results = raw_sym.findAllChildren(tag);
475f73363f1SDimitry Andric       if (!results || results->getChildCount() == 0)
476f73363f1SDimitry Andric         continue;
477f73363f1SDimitry Andric       while (auto symbol = results->getNext()) {
478f73363f1SDimitry Andric         switch (symbol->getSymTag()) {
479f73363f1SDimitry Andric         case PDB_SymType::Enum:
480f73363f1SDimitry Andric         case PDB_SymType::UDT:
481f73363f1SDimitry Andric         case PDB_SymType::Typedef:
482f73363f1SDimitry Andric           break;
483f73363f1SDimitry Andric         default:
484f73363f1SDimitry Andric           continue;
485f73363f1SDimitry Andric         }
486f73363f1SDimitry Andric 
487f73363f1SDimitry Andric         // This should cause the type to get cached and stored in the `m_types`
488f73363f1SDimitry Andric         // lookup.
48994994d37SDimitry Andric         if (auto type = ResolveTypeUID(symbol->getSymIndexId())) {
49094994d37SDimitry Andric           // Resolve the type completely to avoid a completion
49194994d37SDimitry Andric           // (and so a list change, which causes an iterators invalidation)
49294994d37SDimitry Andric           // during a TypeList dumping
49394994d37SDimitry Andric           type->GetFullCompilerType();
494f73363f1SDimitry Andric           ++num_added;
495f73363f1SDimitry Andric         }
496f73363f1SDimitry Andric       }
49794994d37SDimitry Andric     }
498f73363f1SDimitry Andric   };
499f73363f1SDimitry Andric 
500f73363f1SDimitry Andric   ParseTypesByTagFn(*compiland);
501f73363f1SDimitry Andric 
502f73363f1SDimitry Andric   // Also parse global types particularly coming from this compiland.
503f73363f1SDimitry Andric   // Unfortunately, PDB has no compiland information for each global type. We
504f73363f1SDimitry Andric   // have to parse them all. But ensure we only do this once.
505f73363f1SDimitry Andric   static bool parse_all_global_types = false;
506f73363f1SDimitry Andric   if (!parse_all_global_types) {
507f73363f1SDimitry Andric     ParseTypesByTagFn(*m_global_scope_up);
508f73363f1SDimitry Andric     parse_all_global_types = true;
509f73363f1SDimitry Andric   }
510f73363f1SDimitry Andric   return num_added;
511f3fbd1c0SDimitry Andric }
512f3fbd1c0SDimitry Andric 
513f3fbd1c0SDimitry Andric size_t
ParseVariablesForContext(const lldb_private::SymbolContext & sc)51414f1b3e8SDimitry Andric SymbolFilePDB::ParseVariablesForContext(const lldb_private::SymbolContext &sc) {
515ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
516f73363f1SDimitry Andric   if (!sc.comp_unit)
517f73363f1SDimitry Andric     return 0;
518f73363f1SDimitry Andric 
519f73363f1SDimitry Andric   size_t num_added = 0;
520f73363f1SDimitry Andric   if (sc.function) {
521f73363f1SDimitry Andric     auto pdb_func = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(
522f73363f1SDimitry Andric         sc.function->GetID());
523f73363f1SDimitry Andric     if (!pdb_func)
524f73363f1SDimitry Andric       return 0;
525f73363f1SDimitry Andric 
526f73363f1SDimitry Andric     num_added += ParseVariables(sc, *pdb_func);
527f73363f1SDimitry Andric     sc.function->GetBlock(false).SetDidParseVariables(true, true);
528f73363f1SDimitry Andric   } else if (sc.comp_unit) {
529f73363f1SDimitry Andric     auto compiland = GetPDBCompilandByUID(sc.comp_unit->GetID());
530f73363f1SDimitry Andric     if (!compiland)
531f73363f1SDimitry Andric       return 0;
532f73363f1SDimitry Andric 
533f73363f1SDimitry Andric     if (sc.comp_unit->GetVariableList(false))
534f73363f1SDimitry Andric       return 0;
535f73363f1SDimitry Andric 
536f73363f1SDimitry Andric     auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
537f73363f1SDimitry Andric     if (results && results->getChildCount()) {
538f73363f1SDimitry Andric       while (auto result = results->getNext()) {
53994994d37SDimitry Andric         auto cu_id = GetCompilandId(*result);
540f73363f1SDimitry Andric         // FIXME: We are not able to determine variable's compile unit.
541f73363f1SDimitry Andric         if (cu_id == 0)
542f73363f1SDimitry Andric           continue;
543f73363f1SDimitry Andric 
544f73363f1SDimitry Andric         if (cu_id == sc.comp_unit->GetID())
545f73363f1SDimitry Andric           num_added += ParseVariables(sc, *result);
546f73363f1SDimitry Andric       }
547f73363f1SDimitry Andric     }
548f73363f1SDimitry Andric 
549f73363f1SDimitry Andric     // FIXME: A `file static` or `global constant` variable appears both in
550f73363f1SDimitry Andric     // compiland's children and global scope's children with unexpectedly
551f73363f1SDimitry Andric     // different symbol's Id making it ambiguous.
552f73363f1SDimitry Andric 
553f73363f1SDimitry Andric     // FIXME: 'local constant', for example, const char var[] = "abc", declared
554f73363f1SDimitry Andric     // in a function scope, can't be found in PDB.
555f73363f1SDimitry Andric 
556f73363f1SDimitry Andric     // Parse variables in this compiland.
557f73363f1SDimitry Andric     num_added += ParseVariables(sc, *compiland);
558f73363f1SDimitry Andric   }
559f73363f1SDimitry Andric 
560f73363f1SDimitry Andric   return num_added;
561f3fbd1c0SDimitry Andric }
562f3fbd1c0SDimitry Andric 
ResolveTypeUID(lldb::user_id_t type_uid)56314f1b3e8SDimitry Andric lldb_private::Type *SymbolFilePDB::ResolveTypeUID(lldb::user_id_t type_uid) {
564ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
565f3fbd1c0SDimitry Andric   auto find_result = m_types.find(type_uid);
566f3fbd1c0SDimitry Andric   if (find_result != m_types.end())
567f3fbd1c0SDimitry Andric     return find_result->second.get();
568f3fbd1c0SDimitry Andric 
569ead24645SDimitry Andric   auto type_system_or_err =
57014f1b3e8SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
571ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
572145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
5737fa27ce4SDimitry Andric                    "Unable to ResolveTypeUID: {0}");
574ead24645SDimitry Andric     return nullptr;
575ead24645SDimitry Andric   }
576ead24645SDimitry Andric 
577e3b55780SDimitry Andric   auto ts = *type_system_or_err;
578cfca06d7SDimitry Andric   TypeSystemClang *clang_type_system =
579e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
580f3fbd1c0SDimitry Andric   if (!clang_type_system)
581f3fbd1c0SDimitry Andric     return nullptr;
58294994d37SDimitry Andric   PDBASTParser *pdb = clang_type_system->GetPDBParser();
583f3fbd1c0SDimitry Andric   if (!pdb)
584f3fbd1c0SDimitry Andric     return nullptr;
585f3fbd1c0SDimitry Andric 
586f3fbd1c0SDimitry Andric   auto pdb_type = m_session_up->getSymbolById(type_uid);
587f3fbd1c0SDimitry Andric   if (pdb_type == nullptr)
588f3fbd1c0SDimitry Andric     return nullptr;
589f3fbd1c0SDimitry Andric 
590f3fbd1c0SDimitry Andric   lldb::TypeSP result = pdb->CreateLLDBTypeFromPDBType(*pdb_type);
591f73363f1SDimitry Andric   if (result) {
592f3fbd1c0SDimitry Andric     m_types.insert(std::make_pair(type_uid, result));
593f73363f1SDimitry Andric   }
594f3fbd1c0SDimitry Andric   return result.get();
595f3fbd1c0SDimitry Andric }
596f3fbd1c0SDimitry Andric 
GetDynamicArrayInfoForUID(lldb::user_id_t type_uid,const lldb_private::ExecutionContext * exe_ctx)597e3b55780SDimitry Andric std::optional<SymbolFile::ArrayInfo> SymbolFilePDB::GetDynamicArrayInfoForUID(
59894994d37SDimitry Andric     lldb::user_id_t type_uid, const lldb_private::ExecutionContext *exe_ctx) {
599e3b55780SDimitry Andric   return std::nullopt;
60094994d37SDimitry Andric }
60194994d37SDimitry Andric 
CompleteType(lldb_private::CompilerType & compiler_type)60214f1b3e8SDimitry Andric bool SymbolFilePDB::CompleteType(lldb_private::CompilerType &compiler_type) {
60394994d37SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(
60494994d37SDimitry Andric       GetObjectFile()->GetModule()->GetMutex());
60594994d37SDimitry Andric 
606ead24645SDimitry Andric   auto type_system_or_err =
607ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
608ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
609145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
6107fa27ce4SDimitry Andric                    "Unable to get dynamic array info for UID: {0}");
611ead24645SDimitry Andric     return false;
612ead24645SDimitry Andric   }
613e3b55780SDimitry Andric   auto ts = *type_system_or_err;
614cfca06d7SDimitry Andric   TypeSystemClang *clang_ast_ctx =
615e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
616ead24645SDimitry Andric 
61794994d37SDimitry Andric   if (!clang_ast_ctx)
618f3fbd1c0SDimitry Andric     return false;
61994994d37SDimitry Andric 
62094994d37SDimitry Andric   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
62194994d37SDimitry Andric   if (!pdb)
62294994d37SDimitry Andric     return false;
62394994d37SDimitry Andric 
62494994d37SDimitry Andric   return pdb->CompleteTypeFromPDB(compiler_type);
625f3fbd1c0SDimitry Andric }
626f3fbd1c0SDimitry Andric 
GetDeclForUID(lldb::user_id_t uid)62714f1b3e8SDimitry Andric lldb_private::CompilerDecl SymbolFilePDB::GetDeclForUID(lldb::user_id_t uid) {
628ead24645SDimitry Andric   auto type_system_or_err =
629ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
630ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
631145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
6327fa27ce4SDimitry Andric                    "Unable to get decl for UID: {0}");
633ead24645SDimitry Andric     return CompilerDecl();
634ead24645SDimitry Andric   }
635e3b55780SDimitry Andric   auto ts = *type_system_or_err;
636cfca06d7SDimitry Andric   TypeSystemClang *clang_ast_ctx =
637e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
63894994d37SDimitry Andric   if (!clang_ast_ctx)
63994994d37SDimitry Andric     return CompilerDecl();
64094994d37SDimitry Andric 
64194994d37SDimitry Andric   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
64294994d37SDimitry Andric   if (!pdb)
64394994d37SDimitry Andric     return CompilerDecl();
64494994d37SDimitry Andric 
64594994d37SDimitry Andric   auto symbol = m_session_up->getSymbolById(uid);
64694994d37SDimitry Andric   if (!symbol)
64794994d37SDimitry Andric     return CompilerDecl();
64894994d37SDimitry Andric 
64994994d37SDimitry Andric   auto decl = pdb->GetDeclForSymbol(*symbol);
65094994d37SDimitry Andric   if (!decl)
65194994d37SDimitry Andric     return CompilerDecl();
65294994d37SDimitry Andric 
653cfca06d7SDimitry Andric   return clang_ast_ctx->GetCompilerDecl(decl);
654f3fbd1c0SDimitry Andric }
655f3fbd1c0SDimitry Andric 
656f3fbd1c0SDimitry Andric lldb_private::CompilerDeclContext
GetDeclContextForUID(lldb::user_id_t uid)65714f1b3e8SDimitry Andric SymbolFilePDB::GetDeclContextForUID(lldb::user_id_t uid) {
658ead24645SDimitry Andric   auto type_system_or_err =
659ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
660ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
661145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
6627fa27ce4SDimitry Andric                    "Unable to get DeclContext for UID: {0}");
663ead24645SDimitry Andric     return CompilerDeclContext();
664ead24645SDimitry Andric   }
665ead24645SDimitry Andric 
666e3b55780SDimitry Andric   auto ts = *type_system_or_err;
667cfca06d7SDimitry Andric   TypeSystemClang *clang_ast_ctx =
668e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
66994994d37SDimitry Andric   if (!clang_ast_ctx)
67094994d37SDimitry Andric     return CompilerDeclContext();
67194994d37SDimitry Andric 
67294994d37SDimitry Andric   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
67394994d37SDimitry Andric   if (!pdb)
67494994d37SDimitry Andric     return CompilerDeclContext();
67594994d37SDimitry Andric 
67694994d37SDimitry Andric   auto symbol = m_session_up->getSymbolById(uid);
67794994d37SDimitry Andric   if (!symbol)
67894994d37SDimitry Andric     return CompilerDeclContext();
67994994d37SDimitry Andric 
68094994d37SDimitry Andric   auto decl_context = pdb->GetDeclContextForSymbol(*symbol);
68194994d37SDimitry Andric   if (!decl_context)
68294994d37SDimitry Andric     return GetDeclContextContainingUID(uid);
68394994d37SDimitry Andric 
684706b4fc4SDimitry Andric   return clang_ast_ctx->CreateDeclContext(decl_context);
685f3fbd1c0SDimitry Andric }
686f3fbd1c0SDimitry Andric 
687f3fbd1c0SDimitry Andric lldb_private::CompilerDeclContext
GetDeclContextContainingUID(lldb::user_id_t uid)68814f1b3e8SDimitry Andric SymbolFilePDB::GetDeclContextContainingUID(lldb::user_id_t uid) {
689ead24645SDimitry Andric   auto type_system_or_err =
690ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
691ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
692145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
6937fa27ce4SDimitry Andric                    "Unable to get DeclContext containing UID: {0}");
694ead24645SDimitry Andric     return CompilerDeclContext();
695ead24645SDimitry Andric   }
696ead24645SDimitry Andric 
697e3b55780SDimitry Andric   auto ts = *type_system_or_err;
698cfca06d7SDimitry Andric   TypeSystemClang *clang_ast_ctx =
699e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
70094994d37SDimitry Andric   if (!clang_ast_ctx)
70194994d37SDimitry Andric     return CompilerDeclContext();
70294994d37SDimitry Andric 
70394994d37SDimitry Andric   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
70494994d37SDimitry Andric   if (!pdb)
70594994d37SDimitry Andric     return CompilerDeclContext();
70694994d37SDimitry Andric 
70794994d37SDimitry Andric   auto symbol = m_session_up->getSymbolById(uid);
70894994d37SDimitry Andric   if (!symbol)
70994994d37SDimitry Andric     return CompilerDeclContext();
71094994d37SDimitry Andric 
71194994d37SDimitry Andric   auto decl_context = pdb->GetDeclContextContainingSymbol(*symbol);
71294994d37SDimitry Andric   assert(decl_context);
71394994d37SDimitry Andric 
714706b4fc4SDimitry Andric   return clang_ast_ctx->CreateDeclContext(decl_context);
715f3fbd1c0SDimitry Andric }
716f3fbd1c0SDimitry Andric 
ParseDeclsForContext(lldb_private::CompilerDeclContext decl_ctx)71714f1b3e8SDimitry Andric void SymbolFilePDB::ParseDeclsForContext(
71894994d37SDimitry Andric     lldb_private::CompilerDeclContext decl_ctx) {
719ead24645SDimitry Andric   auto type_system_or_err =
720ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
721ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
722145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
7237fa27ce4SDimitry Andric                    "Unable to parse decls for context: {0}");
724ead24645SDimitry Andric     return;
725ead24645SDimitry Andric   }
726ead24645SDimitry Andric 
727e3b55780SDimitry Andric   auto ts = *type_system_or_err;
728cfca06d7SDimitry Andric   TypeSystemClang *clang_ast_ctx =
729e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
73094994d37SDimitry Andric   if (!clang_ast_ctx)
73194994d37SDimitry Andric     return;
73294994d37SDimitry Andric 
73394994d37SDimitry Andric   PDBASTParser *pdb = clang_ast_ctx->GetPDBParser();
73494994d37SDimitry Andric   if (!pdb)
73594994d37SDimitry Andric     return;
73694994d37SDimitry Andric 
73794994d37SDimitry Andric   pdb->ParseDeclsForDeclContext(
73894994d37SDimitry Andric       static_cast<clang::DeclContext *>(decl_ctx.GetOpaqueDeclContext()));
73994994d37SDimitry Andric }
740f3fbd1c0SDimitry Andric 
741f3fbd1c0SDimitry Andric uint32_t
ResolveSymbolContext(const lldb_private::Address & so_addr,SymbolContextItem resolve_scope,lldb_private::SymbolContext & sc)74214f1b3e8SDimitry Andric SymbolFilePDB::ResolveSymbolContext(const lldb_private::Address &so_addr,
74394994d37SDimitry Andric                                     SymbolContextItem resolve_scope,
74414f1b3e8SDimitry Andric                                     lldb_private::SymbolContext &sc) {
745ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
746f73363f1SDimitry Andric   uint32_t resolved_flags = 0;
747f73363f1SDimitry Andric   if (resolve_scope & eSymbolContextCompUnit ||
748f73363f1SDimitry Andric       resolve_scope & eSymbolContextVariable ||
749f73363f1SDimitry Andric       resolve_scope & eSymbolContextFunction ||
750f73363f1SDimitry Andric       resolve_scope & eSymbolContextBlock ||
751f73363f1SDimitry Andric       resolve_scope & eSymbolContextLineEntry) {
752f73363f1SDimitry Andric     auto cu_sp = GetCompileUnitContainsAddress(so_addr);
753f73363f1SDimitry Andric     if (!cu_sp) {
754ead24645SDimitry Andric       if (resolved_flags & eSymbolContextVariable) {
755f73363f1SDimitry Andric         // TODO: Resolve variables
756f73363f1SDimitry Andric       }
757f73363f1SDimitry Andric       return 0;
758f73363f1SDimitry Andric     }
759f73363f1SDimitry Andric     sc.comp_unit = cu_sp.get();
760f73363f1SDimitry Andric     resolved_flags |= eSymbolContextCompUnit;
761f73363f1SDimitry Andric     lldbassert(sc.module_sp == cu_sp->GetModule());
762f73363f1SDimitry Andric   }
763f73363f1SDimitry Andric 
76494994d37SDimitry Andric   if (resolve_scope & eSymbolContextFunction ||
76594994d37SDimitry Andric       resolve_scope & eSymbolContextBlock) {
766f73363f1SDimitry Andric     addr_t file_vm_addr = so_addr.GetFileAddress();
767f73363f1SDimitry Andric     auto symbol_up =
768f73363f1SDimitry Andric         m_session_up->findSymbolByAddress(file_vm_addr, PDB_SymType::Function);
769f73363f1SDimitry Andric     if (symbol_up) {
770f73363f1SDimitry Andric       auto *pdb_func = llvm::dyn_cast<PDBSymbolFunc>(symbol_up.get());
771f73363f1SDimitry Andric       assert(pdb_func);
772f73363f1SDimitry Andric       auto func_uid = pdb_func->getSymIndexId();
773f73363f1SDimitry Andric       sc.function = sc.comp_unit->FindFunctionByUID(func_uid).get();
774f73363f1SDimitry Andric       if (sc.function == nullptr)
77594994d37SDimitry Andric         sc.function =
77694994d37SDimitry Andric             ParseCompileUnitFunctionForPDBFunc(*pdb_func, *sc.comp_unit);
777f73363f1SDimitry Andric       if (sc.function) {
778f73363f1SDimitry Andric         resolved_flags |= eSymbolContextFunction;
779f73363f1SDimitry Andric         if (resolve_scope & eSymbolContextBlock) {
78094994d37SDimitry Andric           auto block_symbol = m_session_up->findSymbolByAddress(
78194994d37SDimitry Andric               file_vm_addr, PDB_SymType::Block);
78294994d37SDimitry Andric           auto block_id = block_symbol ? block_symbol->getSymIndexId()
78394994d37SDimitry Andric                                        : sc.function->GetID();
78494994d37SDimitry Andric           sc.block = sc.function->GetBlock(true).FindBlockByID(block_id);
785f73363f1SDimitry Andric           if (sc.block)
786f73363f1SDimitry Andric             resolved_flags |= eSymbolContextBlock;
787f73363f1SDimitry Andric         }
788f73363f1SDimitry Andric       }
789f73363f1SDimitry Andric     }
790f73363f1SDimitry Andric   }
791f73363f1SDimitry Andric 
792f73363f1SDimitry Andric   if (resolve_scope & eSymbolContextLineEntry) {
793f73363f1SDimitry Andric     if (auto *line_table = sc.comp_unit->GetLineTable()) {
794f73363f1SDimitry Andric       Address addr(so_addr);
795f73363f1SDimitry Andric       if (line_table->FindLineEntryByAddress(addr, sc.line_entry))
796f73363f1SDimitry Andric         resolved_flags |= eSymbolContextLineEntry;
797f73363f1SDimitry Andric     }
798f73363f1SDimitry Andric   }
799f73363f1SDimitry Andric 
800f73363f1SDimitry Andric   return resolved_flags;
801f3fbd1c0SDimitry Andric }
802f3fbd1c0SDimitry Andric 
ResolveSymbolContext(const lldb_private::SourceLocationSpec & src_location_spec,SymbolContextItem resolve_scope,lldb_private::SymbolContextList & sc_list)80314f1b3e8SDimitry Andric uint32_t SymbolFilePDB::ResolveSymbolContext(
804344a3780SDimitry Andric     const lldb_private::SourceLocationSpec &src_location_spec,
80594994d37SDimitry Andric     SymbolContextItem resolve_scope, lldb_private::SymbolContextList &sc_list) {
806ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
807f73363f1SDimitry Andric   const size_t old_size = sc_list.GetSize();
808344a3780SDimitry Andric   const FileSpec &file_spec = src_location_spec.GetFileSpec();
809145449b1SDimitry Andric   const uint32_t line = src_location_spec.GetLine().value_or(0);
81014f1b3e8SDimitry Andric   if (resolve_scope & lldb::eSymbolContextCompUnit) {
81114f1b3e8SDimitry Andric     // Locate all compilation units with line numbers referencing the specified
81274a628f7SDimitry Andric     // file.  For example, if `file_spec` is <vector>, then this should return
81374a628f7SDimitry Andric     // all source files and header files that reference <vector>, either
81474a628f7SDimitry Andric     // directly or indirectly.
81514f1b3e8SDimitry Andric     auto compilands = m_session_up->findCompilandsForSourceFile(
81614f1b3e8SDimitry Andric         file_spec.GetPath(), PDB_NameSearchFlags::NS_CaseInsensitive);
817f3fbd1c0SDimitry Andric 
818f73363f1SDimitry Andric     if (!compilands)
819f73363f1SDimitry Andric       return 0;
820f73363f1SDimitry Andric 
82174a628f7SDimitry Andric     // For each one, either find its previously parsed data or parse it afresh
82274a628f7SDimitry Andric     // and add it to the symbol context list.
82314f1b3e8SDimitry Andric     while (auto compiland = compilands->getNext()) {
824f73363f1SDimitry Andric       // If we're not checking inlines, then don't add line information for
825f73363f1SDimitry Andric       // this file unless the FileSpec matches. For inline functions, we don't
826f73363f1SDimitry Andric       // have to match the FileSpec since they could be defined in headers
827f73363f1SDimitry Andric       // other than file specified in FileSpec.
828344a3780SDimitry Andric       if (!src_location_spec.GetCheckInlines()) {
829f73363f1SDimitry Andric         std::string source_file = compiland->getSourceFileFullPath();
830f73363f1SDimitry Andric         if (source_file.empty())
831f73363f1SDimitry Andric           continue;
83294994d37SDimitry Andric         FileSpec this_spec(source_file, FileSpec::Style::windows);
833f73363f1SDimitry Andric         bool need_full_match = !file_spec.GetDirectory().IsEmpty();
834f73363f1SDimitry Andric         if (FileSpec::Compare(file_spec, this_spec, need_full_match) != 0)
835f3fbd1c0SDimitry Andric           continue;
836f3fbd1c0SDimitry Andric       }
837f3fbd1c0SDimitry Andric 
838f3fbd1c0SDimitry Andric       SymbolContext sc;
839f73363f1SDimitry Andric       auto cu = ParseCompileUnitForUID(compiland->getSymIndexId());
840f73363f1SDimitry Andric       if (!cu)
841f73363f1SDimitry Andric         continue;
842f3fbd1c0SDimitry Andric       sc.comp_unit = cu.get();
843f3fbd1c0SDimitry Andric       sc.module_sp = cu->GetModule();
844f3fbd1c0SDimitry Andric 
84514f1b3e8SDimitry Andric       // If we were asked to resolve line entries, add all entries to the line
84674a628f7SDimitry Andric       // table that match the requested line (or all lines if `line` == 0).
847f73363f1SDimitry Andric       if (resolve_scope & (eSymbolContextFunction | eSymbolContextBlock |
848f73363f1SDimitry Andric                            eSymbolContextLineEntry)) {
84994994d37SDimitry Andric         bool has_line_table = ParseCompileUnitLineTable(*sc.comp_unit, line);
850f73363f1SDimitry Andric 
851f73363f1SDimitry Andric         if ((resolve_scope & eSymbolContextLineEntry) && !has_line_table) {
852f73363f1SDimitry Andric           // The query asks for line entries, but we can't get them for the
853f73363f1SDimitry Andric           // compile unit. This is not normal for `line` = 0. So just assert
854f73363f1SDimitry Andric           // it.
855f73363f1SDimitry Andric           assert(line && "Couldn't get all line entries!\n");
856f73363f1SDimitry Andric 
857f73363f1SDimitry Andric           // Current compiland does not have the requested line. Search next.
858f73363f1SDimitry Andric           continue;
859f73363f1SDimitry Andric         }
860f73363f1SDimitry Andric 
861f73363f1SDimitry Andric         if (resolve_scope & (eSymbolContextFunction | eSymbolContextBlock)) {
862f73363f1SDimitry Andric           if (!has_line_table)
863f73363f1SDimitry Andric             continue;
864f73363f1SDimitry Andric 
865f73363f1SDimitry Andric           auto *line_table = sc.comp_unit->GetLineTable();
866f73363f1SDimitry Andric           lldbassert(line_table);
867f73363f1SDimitry Andric 
868f73363f1SDimitry Andric           uint32_t num_line_entries = line_table->GetSize();
869f73363f1SDimitry Andric           // Skip the terminal line entry.
870f73363f1SDimitry Andric           --num_line_entries;
871f73363f1SDimitry Andric 
872f73363f1SDimitry Andric           // If `line `!= 0, see if we can resolve function for each line entry
873f73363f1SDimitry Andric           // in the line table.
874f73363f1SDimitry Andric           for (uint32_t line_idx = 0; line && line_idx < num_line_entries;
875f73363f1SDimitry Andric                ++line_idx) {
876f73363f1SDimitry Andric             if (!line_table->GetLineEntryAtIndex(line_idx, sc.line_entry))
877f73363f1SDimitry Andric               continue;
878f73363f1SDimitry Andric 
879f73363f1SDimitry Andric             auto file_vm_addr =
880f73363f1SDimitry Andric                 sc.line_entry.range.GetBaseAddress().GetFileAddress();
881f73363f1SDimitry Andric             if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
882f73363f1SDimitry Andric               continue;
883f73363f1SDimitry Andric 
884f73363f1SDimitry Andric             auto symbol_up = m_session_up->findSymbolByAddress(
885f73363f1SDimitry Andric                 file_vm_addr, PDB_SymType::Function);
886f73363f1SDimitry Andric             if (symbol_up) {
887f73363f1SDimitry Andric               auto func_uid = symbol_up->getSymIndexId();
888f73363f1SDimitry Andric               sc.function = sc.comp_unit->FindFunctionByUID(func_uid).get();
889f73363f1SDimitry Andric               if (sc.function == nullptr) {
890f73363f1SDimitry Andric                 auto pdb_func = llvm::dyn_cast<PDBSymbolFunc>(symbol_up.get());
891f73363f1SDimitry Andric                 assert(pdb_func);
89294994d37SDimitry Andric                 sc.function = ParseCompileUnitFunctionForPDBFunc(*pdb_func,
89394994d37SDimitry Andric                                                                  *sc.comp_unit);
894f73363f1SDimitry Andric               }
895f73363f1SDimitry Andric               if (sc.function && (resolve_scope & eSymbolContextBlock)) {
896f73363f1SDimitry Andric                 Block &block = sc.function->GetBlock(true);
897f73363f1SDimitry Andric                 sc.block = block.FindBlockByID(sc.function->GetID());
898f3fbd1c0SDimitry Andric               }
899f3fbd1c0SDimitry Andric             }
900f73363f1SDimitry Andric             sc_list.Append(sc);
901f73363f1SDimitry Andric           }
902f73363f1SDimitry Andric         } else if (has_line_table) {
903f73363f1SDimitry Andric           // We can parse line table for the compile unit. But no query to
904f73363f1SDimitry Andric           // resolve function or block. We append `sc` to the list anyway.
905f73363f1SDimitry Andric           sc_list.Append(sc);
906f73363f1SDimitry Andric         }
907f73363f1SDimitry Andric       } else {
908f73363f1SDimitry Andric         // No query for line entry, function or block. But we have a valid
909f73363f1SDimitry Andric         // compile unit, append `sc` to the list.
910f73363f1SDimitry Andric         sc_list.Append(sc);
911f73363f1SDimitry Andric       }
912f73363f1SDimitry Andric     }
913f73363f1SDimitry Andric   }
914f73363f1SDimitry Andric   return sc_list.GetSize() - old_size;
915f73363f1SDimitry Andric }
916f73363f1SDimitry Andric 
GetMangledForPDBData(const PDBSymbolData & pdb_data)917f73363f1SDimitry Andric std::string SymbolFilePDB::GetMangledForPDBData(const PDBSymbolData &pdb_data) {
91894994d37SDimitry Andric   // Cache public names at first
91994994d37SDimitry Andric   if (m_public_names.empty())
92094994d37SDimitry Andric     if (auto result_up =
92194994d37SDimitry Andric             m_global_scope_up->findAllChildren(PDB_SymType::PublicSymbol))
92294994d37SDimitry Andric       while (auto symbol_up = result_up->getNext())
92394994d37SDimitry Andric         if (auto addr = symbol_up->getRawSymbol().getVirtualAddress())
92494994d37SDimitry Andric           m_public_names[addr] = symbol_up->getRawSymbol().getName();
925f73363f1SDimitry Andric 
92694994d37SDimitry Andric   // Look up the name in the cache
92794994d37SDimitry Andric   return m_public_names.lookup(pdb_data.getVirtualAddress());
928f73363f1SDimitry Andric }
929f73363f1SDimitry Andric 
ParseVariableForPDBData(const lldb_private::SymbolContext & sc,const llvm::pdb::PDBSymbolData & pdb_data)930f73363f1SDimitry Andric VariableSP SymbolFilePDB::ParseVariableForPDBData(
931f73363f1SDimitry Andric     const lldb_private::SymbolContext &sc,
932f73363f1SDimitry Andric     const llvm::pdb::PDBSymbolData &pdb_data) {
933f73363f1SDimitry Andric   VariableSP var_sp;
934f73363f1SDimitry Andric   uint32_t var_uid = pdb_data.getSymIndexId();
935f73363f1SDimitry Andric   auto result = m_variables.find(var_uid);
936f73363f1SDimitry Andric   if (result != m_variables.end())
937f73363f1SDimitry Andric     return result->second;
938f73363f1SDimitry Andric 
939f73363f1SDimitry Andric   ValueType scope = eValueTypeInvalid;
940f73363f1SDimitry Andric   bool is_static_member = false;
941f73363f1SDimitry Andric   bool is_external = false;
942f73363f1SDimitry Andric   bool is_artificial = false;
943f73363f1SDimitry Andric 
944f73363f1SDimitry Andric   switch (pdb_data.getDataKind()) {
945f73363f1SDimitry Andric   case PDB_DataKind::Global:
946f73363f1SDimitry Andric     scope = eValueTypeVariableGlobal;
947f73363f1SDimitry Andric     is_external = true;
948f73363f1SDimitry Andric     break;
949f73363f1SDimitry Andric   case PDB_DataKind::Local:
950f73363f1SDimitry Andric     scope = eValueTypeVariableLocal;
951f73363f1SDimitry Andric     break;
952f73363f1SDimitry Andric   case PDB_DataKind::FileStatic:
953f73363f1SDimitry Andric     scope = eValueTypeVariableStatic;
954f73363f1SDimitry Andric     break;
955f73363f1SDimitry Andric   case PDB_DataKind::StaticMember:
956f73363f1SDimitry Andric     is_static_member = true;
957f73363f1SDimitry Andric     scope = eValueTypeVariableStatic;
958f73363f1SDimitry Andric     break;
959f73363f1SDimitry Andric   case PDB_DataKind::Member:
960f73363f1SDimitry Andric     scope = eValueTypeVariableStatic;
961f73363f1SDimitry Andric     break;
962f73363f1SDimitry Andric   case PDB_DataKind::Param:
963f73363f1SDimitry Andric     scope = eValueTypeVariableArgument;
964f73363f1SDimitry Andric     break;
965f73363f1SDimitry Andric   case PDB_DataKind::Constant:
966f73363f1SDimitry Andric     scope = eValueTypeConstResult;
967f73363f1SDimitry Andric     break;
968f73363f1SDimitry Andric   default:
969f73363f1SDimitry Andric     break;
970f73363f1SDimitry Andric   }
971f73363f1SDimitry Andric 
972f73363f1SDimitry Andric   switch (pdb_data.getLocationType()) {
973f73363f1SDimitry Andric   case PDB_LocType::TLS:
974f73363f1SDimitry Andric     scope = eValueTypeVariableThreadLocal;
975f73363f1SDimitry Andric     break;
976f73363f1SDimitry Andric   case PDB_LocType::RegRel: {
977f73363f1SDimitry Andric     // It is a `this` pointer.
978f73363f1SDimitry Andric     if (pdb_data.getDataKind() == PDB_DataKind::ObjectPtr) {
979f73363f1SDimitry Andric       scope = eValueTypeVariableArgument;
980f73363f1SDimitry Andric       is_artificial = true;
981f73363f1SDimitry Andric     }
982f73363f1SDimitry Andric   } break;
983f73363f1SDimitry Andric   default:
984f73363f1SDimitry Andric     break;
985f73363f1SDimitry Andric   }
986f73363f1SDimitry Andric 
987f73363f1SDimitry Andric   Declaration decl;
988f73363f1SDimitry Andric   if (!is_artificial && !pdb_data.isCompilerGenerated()) {
989f73363f1SDimitry Andric     if (auto lines = pdb_data.getLineNumbers()) {
990f73363f1SDimitry Andric       if (auto first_line = lines->getNext()) {
991f73363f1SDimitry Andric         uint32_t src_file_id = first_line->getSourceFileId();
992f73363f1SDimitry Andric         auto src_file = m_session_up->getSourceFileById(src_file_id);
993f73363f1SDimitry Andric         if (src_file) {
99494994d37SDimitry Andric           FileSpec spec(src_file->getFileName());
995f73363f1SDimitry Andric           decl.SetFile(spec);
996f73363f1SDimitry Andric           decl.SetColumn(first_line->getColumnNumber());
997f73363f1SDimitry Andric           decl.SetLine(first_line->getLineNumber());
998f73363f1SDimitry Andric         }
999f73363f1SDimitry Andric       }
1000f73363f1SDimitry Andric     }
1001f73363f1SDimitry Andric   }
1002f73363f1SDimitry Andric 
1003f73363f1SDimitry Andric   Variable::RangeList ranges;
1004f73363f1SDimitry Andric   SymbolContextScope *context_scope = sc.comp_unit;
10055f29bb8aSDimitry Andric   if (scope == eValueTypeVariableLocal || scope == eValueTypeVariableArgument) {
1006f73363f1SDimitry Andric     if (sc.function) {
10075f29bb8aSDimitry Andric       Block &function_block = sc.function->GetBlock(true);
10085f29bb8aSDimitry Andric       Block *block =
10095f29bb8aSDimitry Andric           function_block.FindBlockByID(pdb_data.getLexicalParentId());
10105f29bb8aSDimitry Andric       if (!block)
10115f29bb8aSDimitry Andric         block = &function_block;
10125f29bb8aSDimitry Andric 
10135f29bb8aSDimitry Andric       context_scope = block;
10145f29bb8aSDimitry Andric 
10155f29bb8aSDimitry Andric       for (size_t i = 0, num_ranges = block->GetNumRanges(); i < num_ranges;
10165f29bb8aSDimitry Andric            ++i) {
10175f29bb8aSDimitry Andric         AddressRange range;
10185f29bb8aSDimitry Andric         if (!block->GetRangeAtIndex(i, range))
10195f29bb8aSDimitry Andric           continue;
10205f29bb8aSDimitry Andric 
10215f29bb8aSDimitry Andric         ranges.Append(range.GetBaseAddress().GetFileAddress(),
10225f29bb8aSDimitry Andric                       range.GetByteSize());
10235f29bb8aSDimitry Andric       }
1024f73363f1SDimitry Andric     }
1025f73363f1SDimitry Andric   }
1026f73363f1SDimitry Andric 
1027f73363f1SDimitry Andric   SymbolFileTypeSP type_sp =
1028f73363f1SDimitry Andric       std::make_shared<SymbolFileType>(*this, pdb_data.getTypeId());
1029f73363f1SDimitry Andric 
1030f73363f1SDimitry Andric   auto var_name = pdb_data.getName();
1031f73363f1SDimitry Andric   auto mangled = GetMangledForPDBData(pdb_data);
1032f73363f1SDimitry Andric   auto mangled_cstr = mangled.empty() ? nullptr : mangled.c_str();
1033f73363f1SDimitry Andric 
1034f73363f1SDimitry Andric   bool is_constant;
10351f917f69SDimitry Andric   ModuleSP module_sp = GetObjectFile()->GetModule();
10361f917f69SDimitry Andric   DWARFExpressionList location(module_sp,
10371f917f69SDimitry Andric                                ConvertPDBLocationToDWARFExpression(
10381f917f69SDimitry Andric                                    module_sp, pdb_data, ranges, is_constant),
10391f917f69SDimitry Andric                                nullptr);
1040f73363f1SDimitry Andric 
1041f73363f1SDimitry Andric   var_sp = std::make_shared<Variable>(
1042f73363f1SDimitry Andric       var_uid, var_name.c_str(), mangled_cstr, type_sp, scope, context_scope,
1043b60736ecSDimitry Andric       ranges, &decl, location, is_external, is_artificial, is_constant,
1044b60736ecSDimitry Andric       is_static_member);
1045f73363f1SDimitry Andric 
1046f73363f1SDimitry Andric   m_variables.insert(std::make_pair(var_uid, var_sp));
1047f73363f1SDimitry Andric   return var_sp;
1048f73363f1SDimitry Andric }
1049f73363f1SDimitry Andric 
1050f73363f1SDimitry Andric size_t
ParseVariables(const lldb_private::SymbolContext & sc,const llvm::pdb::PDBSymbol & pdb_symbol,lldb_private::VariableList * variable_list)1051f73363f1SDimitry Andric SymbolFilePDB::ParseVariables(const lldb_private::SymbolContext &sc,
1052f73363f1SDimitry Andric                               const llvm::pdb::PDBSymbol &pdb_symbol,
1053f73363f1SDimitry Andric                               lldb_private::VariableList *variable_list) {
1054f73363f1SDimitry Andric   size_t num_added = 0;
1055f73363f1SDimitry Andric 
1056f73363f1SDimitry Andric   if (auto pdb_data = llvm::dyn_cast<PDBSymbolData>(&pdb_symbol)) {
1057f73363f1SDimitry Andric     VariableListSP local_variable_list_sp;
1058f73363f1SDimitry Andric 
1059f73363f1SDimitry Andric     auto result = m_variables.find(pdb_data->getSymIndexId());
1060f73363f1SDimitry Andric     if (result != m_variables.end()) {
1061f73363f1SDimitry Andric       if (variable_list)
1062f73363f1SDimitry Andric         variable_list->AddVariableIfUnique(result->second);
1063f73363f1SDimitry Andric     } else {
1064f73363f1SDimitry Andric       // Prepare right VariableList for this variable.
1065f73363f1SDimitry Andric       if (auto lexical_parent = pdb_data->getLexicalParent()) {
1066f73363f1SDimitry Andric         switch (lexical_parent->getSymTag()) {
1067f73363f1SDimitry Andric         case PDB_SymType::Exe:
1068f73363f1SDimitry Andric           assert(sc.comp_unit);
1069e3b55780SDimitry Andric           [[fallthrough]];
1070f73363f1SDimitry Andric         case PDB_SymType::Compiland: {
1071f73363f1SDimitry Andric           if (sc.comp_unit) {
1072f73363f1SDimitry Andric             local_variable_list_sp = sc.comp_unit->GetVariableList(false);
1073f73363f1SDimitry Andric             if (!local_variable_list_sp) {
1074f73363f1SDimitry Andric               local_variable_list_sp = std::make_shared<VariableList>();
1075f73363f1SDimitry Andric               sc.comp_unit->SetVariableList(local_variable_list_sp);
1076f73363f1SDimitry Andric             }
1077f73363f1SDimitry Andric           }
1078f73363f1SDimitry Andric         } break;
1079f73363f1SDimitry Andric         case PDB_SymType::Block:
1080f73363f1SDimitry Andric         case PDB_SymType::Function: {
1081f73363f1SDimitry Andric           if (sc.function) {
1082f73363f1SDimitry Andric             Block *block = sc.function->GetBlock(true).FindBlockByID(
1083f73363f1SDimitry Andric                 lexical_parent->getSymIndexId());
1084f73363f1SDimitry Andric             if (block) {
1085f73363f1SDimitry Andric               local_variable_list_sp = block->GetBlockVariableList(false);
1086f73363f1SDimitry Andric               if (!local_variable_list_sp) {
1087f73363f1SDimitry Andric                 local_variable_list_sp = std::make_shared<VariableList>();
1088f73363f1SDimitry Andric                 block->SetVariableList(local_variable_list_sp);
1089f73363f1SDimitry Andric               }
1090f73363f1SDimitry Andric             }
1091f73363f1SDimitry Andric           }
1092f73363f1SDimitry Andric         } break;
1093f73363f1SDimitry Andric         default:
1094f73363f1SDimitry Andric           break;
1095f73363f1SDimitry Andric         }
1096f73363f1SDimitry Andric       }
1097f73363f1SDimitry Andric 
1098f73363f1SDimitry Andric       if (local_variable_list_sp) {
1099f73363f1SDimitry Andric         if (auto var_sp = ParseVariableForPDBData(sc, *pdb_data)) {
1100f73363f1SDimitry Andric           local_variable_list_sp->AddVariableIfUnique(var_sp);
1101f73363f1SDimitry Andric           if (variable_list)
1102f73363f1SDimitry Andric             variable_list->AddVariableIfUnique(var_sp);
1103f73363f1SDimitry Andric           ++num_added;
110494994d37SDimitry Andric           PDBASTParser *ast = GetPDBAstParser();
110594994d37SDimitry Andric           if (ast)
110694994d37SDimitry Andric             ast->GetDeclForSymbol(*pdb_data);
1107f73363f1SDimitry Andric         }
1108f73363f1SDimitry Andric       }
1109f73363f1SDimitry Andric     }
1110f73363f1SDimitry Andric   }
1111f73363f1SDimitry Andric 
1112f73363f1SDimitry Andric   if (auto results = pdb_symbol.findAllChildren()) {
1113f73363f1SDimitry Andric     while (auto result = results->getNext())
1114f73363f1SDimitry Andric       num_added += ParseVariables(sc, *result, variable_list);
1115f73363f1SDimitry Andric   }
1116f73363f1SDimitry Andric 
1117f73363f1SDimitry Andric   return num_added;
1118f3fbd1c0SDimitry Andric }
1119f3fbd1c0SDimitry Andric 
FindGlobalVariables(lldb_private::ConstString name,const CompilerDeclContext & parent_decl_ctx,uint32_t max_matches,lldb_private::VariableList & variables)1120ead24645SDimitry Andric void SymbolFilePDB::FindGlobalVariables(
1121cfca06d7SDimitry Andric     lldb_private::ConstString name, const CompilerDeclContext &parent_decl_ctx,
112214f1b3e8SDimitry Andric     uint32_t max_matches, lldb_private::VariableList &variables) {
1123ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1124f73363f1SDimitry Andric   if (!DeclContextMatchesThisSymbolFile(parent_decl_ctx))
1125ead24645SDimitry Andric     return;
1126f73363f1SDimitry Andric   if (name.IsEmpty())
1127ead24645SDimitry Andric     return;
1128f73363f1SDimitry Andric 
112994994d37SDimitry Andric   auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
1130f73363f1SDimitry Andric   if (!results)
1131ead24645SDimitry Andric     return;
1132f73363f1SDimitry Andric 
1133f73363f1SDimitry Andric   uint32_t matches = 0;
1134f73363f1SDimitry Andric   size_t old_size = variables.GetSize();
1135f73363f1SDimitry Andric   while (auto result = results->getNext()) {
1136f73363f1SDimitry Andric     auto pdb_data = llvm::dyn_cast<PDBSymbolData>(result.get());
1137f73363f1SDimitry Andric     if (max_matches > 0 && matches >= max_matches)
1138f73363f1SDimitry Andric       break;
1139f73363f1SDimitry Andric 
1140f73363f1SDimitry Andric     SymbolContext sc;
1141ead24645SDimitry Andric     sc.module_sp = m_objfile_sp->GetModule();
1142f73363f1SDimitry Andric     lldbassert(sc.module_sp.get());
1143f73363f1SDimitry Andric 
1144ac9a064cSDimitry Andric     if (name.GetStringRef() !=
1145ac9a064cSDimitry Andric         MSVCUndecoratedNameParser::DropScope(pdb_data->getName()))
114694994d37SDimitry Andric       continue;
114794994d37SDimitry Andric 
114894994d37SDimitry Andric     sc.comp_unit = ParseCompileUnitForUID(GetCompilandId(*pdb_data)).get();
1149f73363f1SDimitry Andric     // FIXME: We are not able to determine the compile unit.
1150f73363f1SDimitry Andric     if (sc.comp_unit == nullptr)
1151f73363f1SDimitry Andric       continue;
1152f73363f1SDimitry Andric 
1153cfca06d7SDimitry Andric     if (parent_decl_ctx.IsValid() &&
1154cfca06d7SDimitry Andric         GetDeclContextContainingUID(result->getSymIndexId()) != parent_decl_ctx)
115594994d37SDimitry Andric       continue;
115694994d37SDimitry Andric 
1157f73363f1SDimitry Andric     ParseVariables(sc, *pdb_data, &variables);
1158f73363f1SDimitry Andric     matches = variables.GetSize() - old_size;
1159f73363f1SDimitry Andric   }
1160f3fbd1c0SDimitry Andric }
1161f3fbd1c0SDimitry Andric 
FindGlobalVariables(const lldb_private::RegularExpression & regex,uint32_t max_matches,lldb_private::VariableList & variables)1162ead24645SDimitry Andric void SymbolFilePDB::FindGlobalVariables(
1163ead24645SDimitry Andric     const lldb_private::RegularExpression &regex, uint32_t max_matches,
116414f1b3e8SDimitry Andric     lldb_private::VariableList &variables) {
1165ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1166f73363f1SDimitry Andric   if (!regex.IsValid())
1167ead24645SDimitry Andric     return;
1168f73363f1SDimitry Andric   auto results = m_global_scope_up->findAllChildren<PDBSymbolData>();
1169f73363f1SDimitry Andric   if (!results)
1170ead24645SDimitry Andric     return;
1171f73363f1SDimitry Andric 
1172f73363f1SDimitry Andric   uint32_t matches = 0;
1173f73363f1SDimitry Andric   size_t old_size = variables.GetSize();
1174f73363f1SDimitry Andric   while (auto pdb_data = results->getNext()) {
1175f73363f1SDimitry Andric     if (max_matches > 0 && matches >= max_matches)
1176f73363f1SDimitry Andric       break;
1177f73363f1SDimitry Andric 
1178f73363f1SDimitry Andric     auto var_name = pdb_data->getName();
1179f73363f1SDimitry Andric     if (var_name.empty())
1180f73363f1SDimitry Andric       continue;
1181f73363f1SDimitry Andric     if (!regex.Execute(var_name))
1182f73363f1SDimitry Andric       continue;
1183f73363f1SDimitry Andric     SymbolContext sc;
1184ead24645SDimitry Andric     sc.module_sp = m_objfile_sp->GetModule();
1185f73363f1SDimitry Andric     lldbassert(sc.module_sp.get());
1186f73363f1SDimitry Andric 
118794994d37SDimitry Andric     sc.comp_unit = ParseCompileUnitForUID(GetCompilandId(*pdb_data)).get();
1188f73363f1SDimitry Andric     // FIXME: We are not able to determine the compile unit.
1189f73363f1SDimitry Andric     if (sc.comp_unit == nullptr)
1190f73363f1SDimitry Andric       continue;
1191f73363f1SDimitry Andric 
1192f73363f1SDimitry Andric     ParseVariables(sc, *pdb_data, &variables);
1193f73363f1SDimitry Andric     matches = variables.GetSize() - old_size;
1194f73363f1SDimitry Andric   }
1195f73363f1SDimitry Andric }
1196f73363f1SDimitry Andric 
ResolveFunction(const llvm::pdb::PDBSymbolFunc & pdb_func,bool include_inlines,lldb_private::SymbolContextList & sc_list)1197f73363f1SDimitry Andric bool SymbolFilePDB::ResolveFunction(const llvm::pdb::PDBSymbolFunc &pdb_func,
1198f73363f1SDimitry Andric                                     bool include_inlines,
1199f73363f1SDimitry Andric                                     lldb_private::SymbolContextList &sc_list) {
1200f73363f1SDimitry Andric   lldb_private::SymbolContext sc;
1201f73363f1SDimitry Andric   sc.comp_unit = ParseCompileUnitForUID(pdb_func.getCompilandId()).get();
1202f73363f1SDimitry Andric   if (!sc.comp_unit)
1203f73363f1SDimitry Andric     return false;
1204f73363f1SDimitry Andric   sc.module_sp = sc.comp_unit->GetModule();
120594994d37SDimitry Andric   sc.function = ParseCompileUnitFunctionForPDBFunc(pdb_func, *sc.comp_unit);
1206f73363f1SDimitry Andric   if (!sc.function)
1207f73363f1SDimitry Andric     return false;
1208f73363f1SDimitry Andric 
1209f73363f1SDimitry Andric   sc_list.Append(sc);
1210f73363f1SDimitry Andric   return true;
1211f73363f1SDimitry Andric }
1212f73363f1SDimitry Andric 
ResolveFunction(uint32_t uid,bool include_inlines,lldb_private::SymbolContextList & sc_list)1213f73363f1SDimitry Andric bool SymbolFilePDB::ResolveFunction(uint32_t uid, bool include_inlines,
1214f73363f1SDimitry Andric                                     lldb_private::SymbolContextList &sc_list) {
1215f73363f1SDimitry Andric   auto pdb_func_up = m_session_up->getConcreteSymbolById<PDBSymbolFunc>(uid);
1216f73363f1SDimitry Andric   if (!pdb_func_up && !(include_inlines && pdb_func_up->hasInlineAttribute()))
1217f73363f1SDimitry Andric     return false;
1218f73363f1SDimitry Andric   return ResolveFunction(*pdb_func_up, include_inlines, sc_list);
1219f73363f1SDimitry Andric }
1220f73363f1SDimitry Andric 
CacheFunctionNames()1221f73363f1SDimitry Andric void SymbolFilePDB::CacheFunctionNames() {
1222f73363f1SDimitry Andric   if (!m_func_full_names.IsEmpty())
1223f73363f1SDimitry Andric     return;
1224f73363f1SDimitry Andric 
1225f73363f1SDimitry Andric   std::map<uint64_t, uint32_t> addr_ids;
1226f73363f1SDimitry Andric 
1227f73363f1SDimitry Andric   if (auto results_up = m_global_scope_up->findAllChildren<PDBSymbolFunc>()) {
1228f73363f1SDimitry Andric     while (auto pdb_func_up = results_up->getNext()) {
1229f73363f1SDimitry Andric       if (pdb_func_up->isCompilerGenerated())
1230f73363f1SDimitry Andric         continue;
1231f73363f1SDimitry Andric 
1232f73363f1SDimitry Andric       auto name = pdb_func_up->getName();
1233f73363f1SDimitry Andric       auto demangled_name = pdb_func_up->getUndecoratedName();
1234f73363f1SDimitry Andric       if (name.empty() && demangled_name.empty())
1235f73363f1SDimitry Andric         continue;
1236f73363f1SDimitry Andric 
1237f73363f1SDimitry Andric       auto uid = pdb_func_up->getSymIndexId();
1238f73363f1SDimitry Andric       if (!demangled_name.empty() && pdb_func_up->getVirtualAddress())
1239f73363f1SDimitry Andric         addr_ids.insert(std::make_pair(pdb_func_up->getVirtualAddress(), uid));
1240f73363f1SDimitry Andric 
1241f73363f1SDimitry Andric       if (auto parent = pdb_func_up->getClassParent()) {
1242f73363f1SDimitry Andric 
1243f73363f1SDimitry Andric         // PDB have symbols for class/struct methods or static methods in Enum
1244f73363f1SDimitry Andric         // Class. We won't bother to check if the parent is UDT or Enum here.
1245f73363f1SDimitry Andric         m_func_method_names.Append(ConstString(name), uid);
1246f73363f1SDimitry Andric 
1247f73363f1SDimitry Andric         // To search a method name, like NS::Class:MemberFunc, LLDB searches
1248f73363f1SDimitry Andric         // its base name, i.e. MemberFunc by default. Since PDBSymbolFunc does
1249cfca06d7SDimitry Andric         // not have information of this, we extract base names and cache them
1250f73363f1SDimitry Andric         // by our own effort.
125194994d37SDimitry Andric         llvm::StringRef basename = MSVCUndecoratedNameParser::DropScope(name);
1252f73363f1SDimitry Andric         if (!basename.empty())
1253f73363f1SDimitry Andric           m_func_base_names.Append(ConstString(basename), uid);
1254f73363f1SDimitry Andric         else {
1255f73363f1SDimitry Andric           m_func_base_names.Append(ConstString(name), uid);
1256f73363f1SDimitry Andric         }
1257f73363f1SDimitry Andric 
1258f73363f1SDimitry Andric         if (!demangled_name.empty())
1259f73363f1SDimitry Andric           m_func_full_names.Append(ConstString(demangled_name), uid);
1260f73363f1SDimitry Andric 
1261f73363f1SDimitry Andric       } else {
1262f73363f1SDimitry Andric         // Handle not-method symbols.
1263f73363f1SDimitry Andric 
126494994d37SDimitry Andric         // The function name might contain namespace, or its lexical scope.
126594994d37SDimitry Andric         llvm::StringRef basename = MSVCUndecoratedNameParser::DropScope(name);
126694994d37SDimitry Andric         if (!basename.empty())
126794994d37SDimitry Andric           m_func_base_names.Append(ConstString(basename), uid);
126894994d37SDimitry Andric         else
1269f73363f1SDimitry Andric           m_func_base_names.Append(ConstString(name), uid);
1270f73363f1SDimitry Andric 
1271f73363f1SDimitry Andric         if (name == "main") {
1272f73363f1SDimitry Andric           m_func_full_names.Append(ConstString(name), uid);
1273f73363f1SDimitry Andric 
1274f73363f1SDimitry Andric           if (!demangled_name.empty() && name != demangled_name) {
1275f73363f1SDimitry Andric             m_func_full_names.Append(ConstString(demangled_name), uid);
1276f73363f1SDimitry Andric             m_func_base_names.Append(ConstString(demangled_name), uid);
1277f73363f1SDimitry Andric           }
1278f73363f1SDimitry Andric         } else if (!demangled_name.empty()) {
1279f73363f1SDimitry Andric           m_func_full_names.Append(ConstString(demangled_name), uid);
1280f73363f1SDimitry Andric         } else {
1281f73363f1SDimitry Andric           m_func_full_names.Append(ConstString(name), uid);
1282f73363f1SDimitry Andric         }
1283f73363f1SDimitry Andric       }
1284f73363f1SDimitry Andric     }
1285f73363f1SDimitry Andric   }
1286f73363f1SDimitry Andric 
1287f73363f1SDimitry Andric   if (auto results_up =
1288f73363f1SDimitry Andric           m_global_scope_up->findAllChildren<PDBSymbolPublicSymbol>()) {
1289f73363f1SDimitry Andric     while (auto pub_sym_up = results_up->getNext()) {
1290f73363f1SDimitry Andric       if (!pub_sym_up->isFunction())
1291f73363f1SDimitry Andric         continue;
1292f73363f1SDimitry Andric       auto name = pub_sym_up->getName();
1293f73363f1SDimitry Andric       if (name.empty())
1294f73363f1SDimitry Andric         continue;
1295f73363f1SDimitry Andric 
1296f73363f1SDimitry Andric       if (CPlusPlusLanguage::IsCPPMangledName(name.c_str())) {
1297f73363f1SDimitry Andric         auto vm_addr = pub_sym_up->getVirtualAddress();
1298f73363f1SDimitry Andric 
1299f73363f1SDimitry Andric         // PDB public symbol has mangled name for its associated function.
1300f73363f1SDimitry Andric         if (vm_addr && addr_ids.find(vm_addr) != addr_ids.end()) {
1301f73363f1SDimitry Andric           // Cache mangled name.
1302f73363f1SDimitry Andric           m_func_full_names.Append(ConstString(name), addr_ids[vm_addr]);
1303f73363f1SDimitry Andric         }
1304f73363f1SDimitry Andric       }
1305f73363f1SDimitry Andric     }
1306f73363f1SDimitry Andric   }
1307f73363f1SDimitry Andric   // Sort them before value searching is working properly
1308f73363f1SDimitry Andric   m_func_full_names.Sort();
1309f73363f1SDimitry Andric   m_func_full_names.SizeToFit();
1310f73363f1SDimitry Andric   m_func_method_names.Sort();
1311f73363f1SDimitry Andric   m_func_method_names.SizeToFit();
1312f73363f1SDimitry Andric   m_func_base_names.Sort();
1313f73363f1SDimitry Andric   m_func_base_names.SizeToFit();
131414f1b3e8SDimitry Andric }
131514f1b3e8SDimitry Andric 
FindFunctions(const lldb_private::Module::LookupInfo & lookup_info,const lldb_private::CompilerDeclContext & parent_decl_ctx,bool include_inlines,lldb_private::SymbolContextList & sc_list)1316ead24645SDimitry Andric void SymbolFilePDB::FindFunctions(
1317e3b55780SDimitry Andric     const lldb_private::Module::LookupInfo &lookup_info,
1318cfca06d7SDimitry Andric     const lldb_private::CompilerDeclContext &parent_decl_ctx,
1319e3b55780SDimitry Andric     bool include_inlines,
132014f1b3e8SDimitry Andric     lldb_private::SymbolContextList &sc_list) {
1321ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1322e3b55780SDimitry Andric   ConstString name = lookup_info.GetLookupName();
1323e3b55780SDimitry Andric   FunctionNameType name_type_mask = lookup_info.GetNameTypeMask();
1324f73363f1SDimitry Andric   lldbassert((name_type_mask & eFunctionNameTypeAuto) == 0);
1325f73363f1SDimitry Andric 
1326e3b55780SDimitry Andric   if (name_type_mask & eFunctionNameTypeFull)
1327e3b55780SDimitry Andric     name = lookup_info.GetName();
1328e3b55780SDimitry Andric 
1329f73363f1SDimitry Andric   if (name_type_mask == eFunctionNameTypeNone)
1330ead24645SDimitry Andric     return;
1331f73363f1SDimitry Andric   if (!DeclContextMatchesThisSymbolFile(parent_decl_ctx))
1332ead24645SDimitry Andric     return;
1333f73363f1SDimitry Andric   if (name.IsEmpty())
1334ead24645SDimitry Andric     return;
1335f73363f1SDimitry Andric 
1336f73363f1SDimitry Andric   if (name_type_mask & eFunctionNameTypeFull ||
1337f73363f1SDimitry Andric       name_type_mask & eFunctionNameTypeBase ||
1338f73363f1SDimitry Andric       name_type_mask & eFunctionNameTypeMethod) {
1339f73363f1SDimitry Andric     CacheFunctionNames();
1340f73363f1SDimitry Andric 
1341f73363f1SDimitry Andric     std::set<uint32_t> resolved_ids;
134294994d37SDimitry Andric     auto ResolveFn = [this, &name, parent_decl_ctx, include_inlines, &sc_list,
134394994d37SDimitry Andric                       &resolved_ids](UniqueCStringMap<uint32_t> &Names) {
1344f73363f1SDimitry Andric       std::vector<uint32_t> ids;
134594994d37SDimitry Andric       if (!Names.GetValues(name, ids))
134694994d37SDimitry Andric         return;
134794994d37SDimitry Andric 
134894994d37SDimitry Andric       for (uint32_t id : ids) {
134994994d37SDimitry Andric         if (resolved_ids.find(id) != resolved_ids.end())
135094994d37SDimitry Andric           continue;
135194994d37SDimitry Andric 
1352cfca06d7SDimitry Andric         if (parent_decl_ctx.IsValid() &&
1353cfca06d7SDimitry Andric             GetDeclContextContainingUID(id) != parent_decl_ctx)
135494994d37SDimitry Andric           continue;
135594994d37SDimitry Andric 
1356f73363f1SDimitry Andric         if (ResolveFunction(id, include_inlines, sc_list))
1357f73363f1SDimitry Andric           resolved_ids.insert(id);
1358f73363f1SDimitry Andric       }
1359f73363f1SDimitry Andric     };
1360f73363f1SDimitry Andric     if (name_type_mask & eFunctionNameTypeFull) {
1361f73363f1SDimitry Andric       ResolveFn(m_func_full_names);
136294994d37SDimitry Andric       ResolveFn(m_func_base_names);
136394994d37SDimitry Andric       ResolveFn(m_func_method_names);
1364f73363f1SDimitry Andric     }
1365ead24645SDimitry Andric     if (name_type_mask & eFunctionNameTypeBase)
1366f73363f1SDimitry Andric       ResolveFn(m_func_base_names);
1367ead24645SDimitry Andric     if (name_type_mask & eFunctionNameTypeMethod)
1368f73363f1SDimitry Andric       ResolveFn(m_func_method_names);
1369f73363f1SDimitry Andric   }
1370f73363f1SDimitry Andric }
1371f3fbd1c0SDimitry Andric 
FindFunctions(const lldb_private::RegularExpression & regex,bool include_inlines,lldb_private::SymbolContextList & sc_list)1372ead24645SDimitry Andric void SymbolFilePDB::FindFunctions(const lldb_private::RegularExpression &regex,
1373ead24645SDimitry Andric                                   bool include_inlines,
137414f1b3e8SDimitry Andric                                   lldb_private::SymbolContextList &sc_list) {
1375ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1376f73363f1SDimitry Andric   if (!regex.IsValid())
1377ead24645SDimitry Andric     return;
1378f73363f1SDimitry Andric 
1379f73363f1SDimitry Andric   CacheFunctionNames();
1380f73363f1SDimitry Andric 
1381f73363f1SDimitry Andric   std::set<uint32_t> resolved_ids;
1382f73363f1SDimitry Andric   auto ResolveFn = [&regex, include_inlines, &sc_list, &resolved_ids,
1383f73363f1SDimitry Andric                     this](UniqueCStringMap<uint32_t> &Names) {
1384f73363f1SDimitry Andric     std::vector<uint32_t> ids;
1385f73363f1SDimitry Andric     if (Names.GetValues(regex, ids)) {
1386f73363f1SDimitry Andric       for (auto id : ids) {
1387f73363f1SDimitry Andric         if (resolved_ids.find(id) == resolved_ids.end())
1388f73363f1SDimitry Andric           if (ResolveFunction(id, include_inlines, sc_list))
1389f73363f1SDimitry Andric             resolved_ids.insert(id);
1390f73363f1SDimitry Andric       }
1391f73363f1SDimitry Andric     }
1392f73363f1SDimitry Andric   };
1393f73363f1SDimitry Andric   ResolveFn(m_func_full_names);
1394f73363f1SDimitry Andric   ResolveFn(m_func_base_names);
1395f3fbd1c0SDimitry Andric }
1396f3fbd1c0SDimitry Andric 
GetMangledNamesForFunction(const std::string & scope_qualified_name,std::vector<lldb_private::ConstString> & mangled_names)139714f1b3e8SDimitry Andric void SymbolFilePDB::GetMangledNamesForFunction(
139814f1b3e8SDimitry Andric     const std::string &scope_qualified_name,
139914f1b3e8SDimitry Andric     std::vector<lldb_private::ConstString> &mangled_names) {}
1400f3fbd1c0SDimitry Andric 
AddSymbols(lldb_private::Symtab & symtab)140194994d37SDimitry Andric void SymbolFilePDB::AddSymbols(lldb_private::Symtab &symtab) {
140294994d37SDimitry Andric   std::set<lldb::addr_t> sym_addresses;
140394994d37SDimitry Andric   for (size_t i = 0; i < symtab.GetNumSymbols(); i++)
140494994d37SDimitry Andric     sym_addresses.insert(symtab.SymbolAtIndex(i)->GetFileAddress());
140594994d37SDimitry Andric 
140694994d37SDimitry Andric   auto results = m_global_scope_up->findAllChildren<PDBSymbolPublicSymbol>();
140794994d37SDimitry Andric   if (!results)
140894994d37SDimitry Andric     return;
140994994d37SDimitry Andric 
1410ead24645SDimitry Andric   auto section_list = m_objfile_sp->GetSectionList();
141194994d37SDimitry Andric   if (!section_list)
141294994d37SDimitry Andric     return;
141394994d37SDimitry Andric 
141494994d37SDimitry Andric   while (auto pub_symbol = results->getNext()) {
14155f29bb8aSDimitry Andric     auto section_id = pub_symbol->getAddressSection();
141694994d37SDimitry Andric 
14175f29bb8aSDimitry Andric     auto section = section_list->FindSectionByID(section_id);
141894994d37SDimitry Andric     if (!section)
141994994d37SDimitry Andric       continue;
142094994d37SDimitry Andric 
142194994d37SDimitry Andric     auto offset = pub_symbol->getAddressOffset();
142294994d37SDimitry Andric 
142394994d37SDimitry Andric     auto file_addr = section->GetFileAddress() + offset;
142494994d37SDimitry Andric     if (sym_addresses.find(file_addr) != sym_addresses.end())
142594994d37SDimitry Andric       continue;
142694994d37SDimitry Andric     sym_addresses.insert(file_addr);
142794994d37SDimitry Andric 
142894994d37SDimitry Andric     auto size = pub_symbol->getLength();
142994994d37SDimitry Andric     symtab.AddSymbol(
143094994d37SDimitry Andric         Symbol(pub_symbol->getSymIndexId(),   // symID
143194994d37SDimitry Andric                pub_symbol->getName().c_str(), // name
143294994d37SDimitry Andric                pub_symbol->isCode() ? eSymbolTypeCode : eSymbolTypeData, // type
143394994d37SDimitry Andric                true,      // external
143494994d37SDimitry Andric                false,     // is_debug
143594994d37SDimitry Andric                false,     // is_trampoline
143694994d37SDimitry Andric                false,     // is_artificial
143794994d37SDimitry Andric                section,   // section_sp
143894994d37SDimitry Andric                offset,    // value
143994994d37SDimitry Andric                size,      // size
144094994d37SDimitry Andric                size != 0, // size_is_valid
144194994d37SDimitry Andric                false,     // contains_linker_annotations
144294994d37SDimitry Andric                0          // flags
144394994d37SDimitry Andric                ));
144494994d37SDimitry Andric   }
144594994d37SDimitry Andric 
144694994d37SDimitry Andric   symtab.Finalize();
144794994d37SDimitry Andric }
144894994d37SDimitry Andric 
DumpClangAST(Stream & s)144994994d37SDimitry Andric void SymbolFilePDB::DumpClangAST(Stream &s) {
1450ead24645SDimitry Andric   auto type_system_or_err =
1451ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
1452ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
1453145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
14547fa27ce4SDimitry Andric                    "Unable to dump ClangAST: {0}");
145594994d37SDimitry Andric     return;
1456ead24645SDimitry Andric   }
1457ead24645SDimitry Andric 
1458e3b55780SDimitry Andric   auto ts = *type_system_or_err;
1459e3b55780SDimitry Andric   TypeSystemClang *clang_type_system =
1460e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
1461ead24645SDimitry Andric   if (!clang_type_system)
1462ead24645SDimitry Andric     return;
1463c0981da4SDimitry Andric   clang_type_system->Dump(s.AsRawOstream());
146494994d37SDimitry Andric }
146594994d37SDimitry Andric 
FindTypesByRegex(const lldb_private::RegularExpression & regex,uint32_t max_matches,lldb_private::TypeMap & types)1466f73363f1SDimitry Andric void SymbolFilePDB::FindTypesByRegex(
1467f73363f1SDimitry Andric     const lldb_private::RegularExpression &regex, uint32_t max_matches,
146814f1b3e8SDimitry Andric     lldb_private::TypeMap &types) {
146914f1b3e8SDimitry Andric   // When searching by regex, we need to go out of our way to limit the search
147074a628f7SDimitry Andric   // space as much as possible since this searches EVERYTHING in the PDB,
147174a628f7SDimitry Andric   // manually doing regex comparisons.  PDB library isn't optimized for regex
147274a628f7SDimitry Andric   // searches or searches across multiple symbol types at the same time, so the
147314f1b3e8SDimitry Andric   // best we can do is to search enums, then typedefs, then classes one by one,
147474a628f7SDimitry Andric   // and do a regex comparison against each of them.
147514f1b3e8SDimitry Andric   PDB_SymType tags_to_search[] = {PDB_SymType::Enum, PDB_SymType::Typedef,
147614f1b3e8SDimitry Andric                                   PDB_SymType::UDT};
1477f3fbd1c0SDimitry Andric   std::unique_ptr<IPDBEnumSymbols> results;
1478f3fbd1c0SDimitry Andric 
1479f3fbd1c0SDimitry Andric   uint32_t matches = 0;
1480f3fbd1c0SDimitry Andric 
148114f1b3e8SDimitry Andric   for (auto tag : tags_to_search) {
1482f73363f1SDimitry Andric     results = m_global_scope_up->findAllChildren(tag);
1483f73363f1SDimitry Andric     if (!results)
1484f73363f1SDimitry Andric       continue;
1485f73363f1SDimitry Andric 
148614f1b3e8SDimitry Andric     while (auto result = results->getNext()) {
1487f3fbd1c0SDimitry Andric       if (max_matches > 0 && matches >= max_matches)
1488f3fbd1c0SDimitry Andric         break;
1489f3fbd1c0SDimitry Andric 
1490f3fbd1c0SDimitry Andric       std::string type_name;
1491f3fbd1c0SDimitry Andric       if (auto enum_type = llvm::dyn_cast<PDBSymbolTypeEnum>(result.get()))
1492f3fbd1c0SDimitry Andric         type_name = enum_type->getName();
149314f1b3e8SDimitry Andric       else if (auto typedef_type =
149414f1b3e8SDimitry Andric                    llvm::dyn_cast<PDBSymbolTypeTypedef>(result.get()))
1495f3fbd1c0SDimitry Andric         type_name = typedef_type->getName();
1496f3fbd1c0SDimitry Andric       else if (auto class_type = llvm::dyn_cast<PDBSymbolTypeUDT>(result.get()))
1497f3fbd1c0SDimitry Andric         type_name = class_type->getName();
149814f1b3e8SDimitry Andric       else {
149974a628f7SDimitry Andric         // We're looking only for types that have names.  Skip symbols, as well
150074a628f7SDimitry Andric         // as unnamed types such as arrays, pointers, etc.
1501f3fbd1c0SDimitry Andric         continue;
1502f3fbd1c0SDimitry Andric       }
1503f3fbd1c0SDimitry Andric 
150423629167SDimitry Andric       if (!regex.Execute(type_name))
1505f3fbd1c0SDimitry Andric         continue;
1506f3fbd1c0SDimitry Andric 
150714f1b3e8SDimitry Andric       // This should cause the type to get cached and stored in the `m_types`
150814f1b3e8SDimitry Andric       // lookup.
1509f3fbd1c0SDimitry Andric       if (!ResolveTypeUID(result->getSymIndexId()))
1510f3fbd1c0SDimitry Andric         continue;
1511f3fbd1c0SDimitry Andric 
1512f3fbd1c0SDimitry Andric       auto iter = m_types.find(result->getSymIndexId());
1513f3fbd1c0SDimitry Andric       if (iter == m_types.end())
1514f3fbd1c0SDimitry Andric         continue;
1515f3fbd1c0SDimitry Andric       types.Insert(iter->second);
1516f3fbd1c0SDimitry Andric       ++matches;
1517f3fbd1c0SDimitry Andric     }
1518f3fbd1c0SDimitry Andric   }
1519f3fbd1c0SDimitry Andric }
1520f3fbd1c0SDimitry Andric 
FindTypes(const lldb_private::TypeQuery & query,lldb_private::TypeResults & type_results)1521312c0ed1SDimitry Andric void SymbolFilePDB::FindTypes(const lldb_private::TypeQuery &query,
1522312c0ed1SDimitry Andric                               lldb_private::TypeResults &type_results) {
1523312c0ed1SDimitry Andric 
1524312c0ed1SDimitry Andric   // Make sure we haven't already searched this SymbolFile before.
1525312c0ed1SDimitry Andric   if (type_results.AlreadySearched(this))
1526312c0ed1SDimitry Andric     return;
1527312c0ed1SDimitry Andric 
1528312c0ed1SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1529312c0ed1SDimitry Andric 
1530f3fbd1c0SDimitry Andric   std::unique_ptr<IPDBEnumSymbols> results;
1531312c0ed1SDimitry Andric   llvm::StringRef basename = query.GetTypeBasename().GetStringRef();
1532312c0ed1SDimitry Andric   if (basename.empty())
1533f73363f1SDimitry Andric     return;
153494994d37SDimitry Andric   results = m_global_scope_up->findAllChildren(PDB_SymType::None);
1535f73363f1SDimitry Andric   if (!results)
1536f73363f1SDimitry Andric     return;
1537f3fbd1c0SDimitry Andric 
153814f1b3e8SDimitry Andric   while (auto result = results->getNext()) {
153994994d37SDimitry Andric 
154014f1b3e8SDimitry Andric     switch (result->getSymTag()) {
1541f3fbd1c0SDimitry Andric     case PDB_SymType::Enum:
1542f3fbd1c0SDimitry Andric     case PDB_SymType::UDT:
1543f3fbd1c0SDimitry Andric     case PDB_SymType::Typedef:
1544f3fbd1c0SDimitry Andric       break;
1545f3fbd1c0SDimitry Andric     default:
1546f73363f1SDimitry Andric       // We're looking only for types that have names.  Skip symbols, as well
1547f73363f1SDimitry Andric       // as unnamed types such as arrays, pointers, etc.
1548f3fbd1c0SDimitry Andric       continue;
1549f3fbd1c0SDimitry Andric     }
1550f3fbd1c0SDimitry Andric 
1551312c0ed1SDimitry Andric     if (MSVCUndecoratedNameParser::DropScope(
1552312c0ed1SDimitry Andric             result->getRawSymbol().getName()) != basename)
1553312c0ed1SDimitry Andric       continue;
1554312c0ed1SDimitry Andric 
155514f1b3e8SDimitry Andric     // This should cause the type to get cached and stored in the `m_types`
155614f1b3e8SDimitry Andric     // lookup.
1557f3fbd1c0SDimitry Andric     if (!ResolveTypeUID(result->getSymIndexId()))
1558f3fbd1c0SDimitry Andric       continue;
1559f3fbd1c0SDimitry Andric 
1560f3fbd1c0SDimitry Andric     auto iter = m_types.find(result->getSymIndexId());
1561f3fbd1c0SDimitry Andric     if (iter == m_types.end())
1562f3fbd1c0SDimitry Andric       continue;
1563312c0ed1SDimitry Andric     // We resolved a type. Get the fully qualified name to ensure it matches.
1564312c0ed1SDimitry Andric     ConstString name = iter->second->GetQualifiedName();
1565312c0ed1SDimitry Andric     TypeQuery type_match(name.GetStringRef(), TypeQueryOptions::e_exact_match);
1566312c0ed1SDimitry Andric     if (query.ContextMatches(type_match.GetContextRef())) {
1567312c0ed1SDimitry Andric       type_results.InsertUnique(iter->second);
1568312c0ed1SDimitry Andric       if (type_results.Done(query))
1569312c0ed1SDimitry Andric         return;
1570f3fbd1c0SDimitry Andric     }
1571f3fbd1c0SDimitry Andric   }
1572312c0ed1SDimitry Andric }
1573f73363f1SDimitry Andric 
GetTypesForPDBSymbol(const llvm::pdb::PDBSymbol & pdb_symbol,uint32_t type_mask,TypeCollection & type_collection)1574f73363f1SDimitry Andric void SymbolFilePDB::GetTypesForPDBSymbol(const llvm::pdb::PDBSymbol &pdb_symbol,
1575f73363f1SDimitry Andric                                          uint32_t type_mask,
1576f73363f1SDimitry Andric                                          TypeCollection &type_collection) {
1577f73363f1SDimitry Andric   bool can_parse = false;
1578f73363f1SDimitry Andric   switch (pdb_symbol.getSymTag()) {
1579f73363f1SDimitry Andric   case PDB_SymType::ArrayType:
1580f73363f1SDimitry Andric     can_parse = ((type_mask & eTypeClassArray) != 0);
1581f73363f1SDimitry Andric     break;
1582f73363f1SDimitry Andric   case PDB_SymType::BuiltinType:
1583f73363f1SDimitry Andric     can_parse = ((type_mask & eTypeClassBuiltin) != 0);
1584f73363f1SDimitry Andric     break;
1585f73363f1SDimitry Andric   case PDB_SymType::Enum:
1586f73363f1SDimitry Andric     can_parse = ((type_mask & eTypeClassEnumeration) != 0);
1587f73363f1SDimitry Andric     break;
1588f73363f1SDimitry Andric   case PDB_SymType::Function:
1589f73363f1SDimitry Andric   case PDB_SymType::FunctionSig:
1590f73363f1SDimitry Andric     can_parse = ((type_mask & eTypeClassFunction) != 0);
1591f73363f1SDimitry Andric     break;
1592f73363f1SDimitry Andric   case PDB_SymType::PointerType:
1593f73363f1SDimitry Andric     can_parse = ((type_mask & (eTypeClassPointer | eTypeClassBlockPointer |
1594f73363f1SDimitry Andric                                eTypeClassMemberPointer)) != 0);
1595f73363f1SDimitry Andric     break;
1596f73363f1SDimitry Andric   case PDB_SymType::Typedef:
1597f73363f1SDimitry Andric     can_parse = ((type_mask & eTypeClassTypedef) != 0);
1598f73363f1SDimitry Andric     break;
1599f73363f1SDimitry Andric   case PDB_SymType::UDT: {
1600f73363f1SDimitry Andric     auto *udt = llvm::dyn_cast<PDBSymbolTypeUDT>(&pdb_symbol);
1601f73363f1SDimitry Andric     assert(udt);
1602f73363f1SDimitry Andric     can_parse = (udt->getUdtKind() != PDB_UdtType::Interface &&
1603f73363f1SDimitry Andric                  ((type_mask & (eTypeClassClass | eTypeClassStruct |
1604f73363f1SDimitry Andric                                 eTypeClassUnion)) != 0));
1605f73363f1SDimitry Andric   } break;
1606f73363f1SDimitry Andric   default:
1607f73363f1SDimitry Andric     break;
1608f73363f1SDimitry Andric   }
1609f73363f1SDimitry Andric 
1610f73363f1SDimitry Andric   if (can_parse) {
1611f73363f1SDimitry Andric     if (auto *type = ResolveTypeUID(pdb_symbol.getSymIndexId())) {
1612e3b55780SDimitry Andric       if (!llvm::is_contained(type_collection, type))
1613f73363f1SDimitry Andric         type_collection.push_back(type);
1614f73363f1SDimitry Andric     }
1615f73363f1SDimitry Andric   }
1616f73363f1SDimitry Andric 
1617f73363f1SDimitry Andric   auto results_up = pdb_symbol.findAllChildren();
1618f73363f1SDimitry Andric   while (auto symbol_up = results_up->getNext())
1619f73363f1SDimitry Andric     GetTypesForPDBSymbol(*symbol_up, type_mask, type_collection);
1620f73363f1SDimitry Andric }
1621f3fbd1c0SDimitry Andric 
GetTypes(lldb_private::SymbolContextScope * sc_scope,TypeClass type_mask,lldb_private::TypeList & type_list)1622ead24645SDimitry Andric void SymbolFilePDB::GetTypes(lldb_private::SymbolContextScope *sc_scope,
162394994d37SDimitry Andric                              TypeClass type_mask,
162414f1b3e8SDimitry Andric                              lldb_private::TypeList &type_list) {
1625ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1626f73363f1SDimitry Andric   TypeCollection type_collection;
1627f73363f1SDimitry Andric   CompileUnit *cu =
1628f73363f1SDimitry Andric       sc_scope ? sc_scope->CalculateSymbolContextCompileUnit() : nullptr;
1629f73363f1SDimitry Andric   if (cu) {
1630f73363f1SDimitry Andric     auto compiland_up = GetPDBCompilandByUID(cu->GetID());
1631f73363f1SDimitry Andric     if (!compiland_up)
1632ead24645SDimitry Andric       return;
1633f73363f1SDimitry Andric     GetTypesForPDBSymbol(*compiland_up, type_mask, type_collection);
1634f73363f1SDimitry Andric   } else {
1635f73363f1SDimitry Andric     for (uint32_t cu_idx = 0; cu_idx < GetNumCompileUnits(); ++cu_idx) {
1636f73363f1SDimitry Andric       auto cu_sp = ParseCompileUnitAtIndex(cu_idx);
1637f73363f1SDimitry Andric       if (cu_sp) {
1638f73363f1SDimitry Andric         if (auto compiland_up = GetPDBCompilandByUID(cu_sp->GetID()))
1639f73363f1SDimitry Andric           GetTypesForPDBSymbol(*compiland_up, type_mask, type_collection);
1640f73363f1SDimitry Andric       }
1641f73363f1SDimitry Andric     }
1642f73363f1SDimitry Andric   }
1643f73363f1SDimitry Andric 
1644f73363f1SDimitry Andric   for (auto type : type_collection) {
1645f73363f1SDimitry Andric     type->GetForwardCompilerType();
1646f73363f1SDimitry Andric     type_list.Insert(type->shared_from_this());
1647f73363f1SDimitry Andric   }
1648f3fbd1c0SDimitry Andric }
1649f3fbd1c0SDimitry Andric 
1650e3b55780SDimitry Andric llvm::Expected<lldb::TypeSystemSP>
GetTypeSystemForLanguage(lldb::LanguageType language)165114f1b3e8SDimitry Andric SymbolFilePDB::GetTypeSystemForLanguage(lldb::LanguageType language) {
1652ead24645SDimitry Andric   auto type_system_or_err =
1653ead24645SDimitry Andric       m_objfile_sp->GetModule()->GetTypeSystemForLanguage(language);
1654ead24645SDimitry Andric   if (type_system_or_err) {
1655e3b55780SDimitry Andric     if (auto ts = *type_system_or_err)
1656e3b55780SDimitry Andric       ts->SetSymbolFile(this);
1657ead24645SDimitry Andric   }
1658ead24645SDimitry Andric   return type_system_or_err;
1659f3fbd1c0SDimitry Andric }
1660f3fbd1c0SDimitry Andric 
GetPDBAstParser()166194994d37SDimitry Andric PDBASTParser *SymbolFilePDB::GetPDBAstParser() {
1662ead24645SDimitry Andric   auto type_system_or_err =
1663ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
1664ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
1665145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
16667fa27ce4SDimitry Andric                    "Unable to get PDB AST parser: {0}");
1667ead24645SDimitry Andric     return nullptr;
1668ead24645SDimitry Andric   }
1669ead24645SDimitry Andric 
1670e3b55780SDimitry Andric   auto ts = *type_system_or_err;
1671ead24645SDimitry Andric   auto *clang_type_system =
1672e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
167394994d37SDimitry Andric   if (!clang_type_system)
167494994d37SDimitry Andric     return nullptr;
167594994d37SDimitry Andric 
167694994d37SDimitry Andric   return clang_type_system->GetPDBParser();
167794994d37SDimitry Andric }
167894994d37SDimitry Andric 
1679cfca06d7SDimitry Andric lldb_private::CompilerDeclContext
FindNamespace(lldb_private::ConstString name,const CompilerDeclContext & parent_decl_ctx,bool)1680cfca06d7SDimitry Andric SymbolFilePDB::FindNamespace(lldb_private::ConstString name,
16817fa27ce4SDimitry Andric                              const CompilerDeclContext &parent_decl_ctx, bool) {
1682ead24645SDimitry Andric   std::lock_guard<std::recursive_mutex> guard(GetModuleMutex());
1683ead24645SDimitry Andric   auto type_system_or_err =
1684ead24645SDimitry Andric       GetTypeSystemForLanguage(lldb::eLanguageTypeC_plus_plus);
1685ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
1686145449b1SDimitry Andric     LLDB_LOG_ERROR(GetLog(LLDBLog::Symbols), std::move(err),
16877fa27ce4SDimitry Andric                    "Unable to find namespace {1}: {0}", name.AsCString());
1688ead24645SDimitry Andric     return CompilerDeclContext();
1689ead24645SDimitry Andric   }
1690e3b55780SDimitry Andric   auto ts = *type_system_or_err;
1691ead24645SDimitry Andric   auto *clang_type_system =
1692e3b55780SDimitry Andric       llvm::dyn_cast_or_null<TypeSystemClang>(ts.get());
169394994d37SDimitry Andric   if (!clang_type_system)
169494994d37SDimitry Andric     return CompilerDeclContext();
169594994d37SDimitry Andric 
169694994d37SDimitry Andric   PDBASTParser *pdb = clang_type_system->GetPDBParser();
169794994d37SDimitry Andric   if (!pdb)
169894994d37SDimitry Andric     return CompilerDeclContext();
169994994d37SDimitry Andric 
170094994d37SDimitry Andric   clang::DeclContext *decl_context = nullptr;
170194994d37SDimitry Andric   if (parent_decl_ctx)
170294994d37SDimitry Andric     decl_context = static_cast<clang::DeclContext *>(
1703cfca06d7SDimitry Andric         parent_decl_ctx.GetOpaqueDeclContext());
170494994d37SDimitry Andric 
170594994d37SDimitry Andric   auto namespace_decl =
170694994d37SDimitry Andric       pdb->FindNamespaceDecl(decl_context, name.GetStringRef());
170794994d37SDimitry Andric   if (!namespace_decl)
170894994d37SDimitry Andric     return CompilerDeclContext();
170994994d37SDimitry Andric 
1710706b4fc4SDimitry Andric   return clang_type_system->CreateDeclContext(namespace_decl);
1711f3fbd1c0SDimitry Andric }
1712f3fbd1c0SDimitry Andric 
GetPDBSession()171314f1b3e8SDimitry Andric IPDBSession &SymbolFilePDB::GetPDBSession() { return *m_session_up; }
171414f1b3e8SDimitry Andric 
GetPDBSession() const171514f1b3e8SDimitry Andric const IPDBSession &SymbolFilePDB::GetPDBSession() const {
1716f3fbd1c0SDimitry Andric   return *m_session_up;
1717f3fbd1c0SDimitry Andric }
1718f3fbd1c0SDimitry Andric 
ParseCompileUnitForUID(uint32_t id,uint32_t index)1719f73363f1SDimitry Andric lldb::CompUnitSP SymbolFilePDB::ParseCompileUnitForUID(uint32_t id,
1720f73363f1SDimitry Andric                                                        uint32_t index) {
1721f3fbd1c0SDimitry Andric   auto found_cu = m_comp_units.find(id);
1722f3fbd1c0SDimitry Andric   if (found_cu != m_comp_units.end())
1723f3fbd1c0SDimitry Andric     return found_cu->second;
1724f3fbd1c0SDimitry Andric 
1725f73363f1SDimitry Andric   auto compiland_up = GetPDBCompilandByUID(id);
1726f73363f1SDimitry Andric   if (!compiland_up)
1727f73363f1SDimitry Andric     return CompUnitSP();
1728f3fbd1c0SDimitry Andric 
1729f3fbd1c0SDimitry Andric   lldb::LanguageType lang;
1730f73363f1SDimitry Andric   auto details = compiland_up->findOneChild<PDBSymbolCompilandDetails>();
1731f3fbd1c0SDimitry Andric   if (!details)
1732f3fbd1c0SDimitry Andric     lang = lldb::eLanguageTypeC_plus_plus;
1733f3fbd1c0SDimitry Andric   else
1734f3fbd1c0SDimitry Andric     lang = TranslateLanguage(details->getLanguage());
1735f3fbd1c0SDimitry Andric 
1736f73363f1SDimitry Andric   if (lang == lldb::LanguageType::eLanguageTypeUnknown)
1737f73363f1SDimitry Andric     return CompUnitSP();
1738f73363f1SDimitry Andric 
1739f73363f1SDimitry Andric   std::string path = compiland_up->getSourceFileFullPath();
1740f73363f1SDimitry Andric   if (path.empty())
1741f73363f1SDimitry Andric     return CompUnitSP();
1742f73363f1SDimitry Andric 
174314f1b3e8SDimitry Andric   // Don't support optimized code for now, DebugInfoPDB does not return this
174414f1b3e8SDimitry Andric   // information.
1745f3fbd1c0SDimitry Andric   LazyBool optimized = eLazyBoolNo;
1746ead24645SDimitry Andric   auto cu_sp = std::make_shared<CompileUnit>(m_objfile_sp->GetModule(), nullptr,
1747f73363f1SDimitry Andric                                              path.c_str(), id, lang, optimized);
1748f73363f1SDimitry Andric 
1749f73363f1SDimitry Andric   if (!cu_sp)
1750f73363f1SDimitry Andric     return CompUnitSP();
1751f73363f1SDimitry Andric 
1752f73363f1SDimitry Andric   m_comp_units.insert(std::make_pair(id, cu_sp));
1753f73363f1SDimitry Andric   if (index == UINT32_MAX)
1754f73363f1SDimitry Andric     GetCompileUnitIndex(*compiland_up, index);
1755f73363f1SDimitry Andric   lldbassert(index != UINT32_MAX);
1756ead24645SDimitry Andric   SetCompileUnitAtIndex(index, cu_sp);
1757f73363f1SDimitry Andric   return cu_sp;
1758f3fbd1c0SDimitry Andric }
1759f3fbd1c0SDimitry Andric 
ParseCompileUnitLineTable(CompileUnit & comp_unit,uint32_t match_line)176094994d37SDimitry Andric bool SymbolFilePDB::ParseCompileUnitLineTable(CompileUnit &comp_unit,
176194994d37SDimitry Andric                                               uint32_t match_line) {
176294994d37SDimitry Andric   auto compiland_up = GetPDBCompilandByUID(comp_unit.GetID());
1763f73363f1SDimitry Andric   if (!compiland_up)
1764f73363f1SDimitry Andric     return false;
1765f3fbd1c0SDimitry Andric 
176614f1b3e8SDimitry Andric   // LineEntry needs the *index* of the file into the list of support files
176774a628f7SDimitry Andric   // returned by ParseCompileUnitSupportFiles.  But the underlying SDK gives us
1768f73363f1SDimitry Andric   // a globally unique idenfitifier in the namespace of the PDB.  So, we have
1769f73363f1SDimitry Andric   // to do a mapping so that we can hand out indices.
1770f3fbd1c0SDimitry Andric   llvm::DenseMap<uint32_t, uint32_t> index_map;
1771f73363f1SDimitry Andric   BuildSupportFileIdToSupportFileIndexMap(*compiland_up, index_map);
1772ead24645SDimitry Andric   auto line_table = std::make_unique<LineTable>(&comp_unit);
1773f3fbd1c0SDimitry Andric 
1774f73363f1SDimitry Andric   // Find contributions to `compiland` from all source and header files.
1775f73363f1SDimitry Andric   auto files = m_session_up->getSourceFilesForCompiland(*compiland_up);
1776f73363f1SDimitry Andric   if (!files)
1777f73363f1SDimitry Andric     return false;
1778f3fbd1c0SDimitry Andric 
1779f73363f1SDimitry Andric   // For each source and header file, create a LineSequence for contributions
1780f73363f1SDimitry Andric   // to the compiland from that file, and add the sequence.
178114f1b3e8SDimitry Andric   while (auto file = files->getNext()) {
178214f1b3e8SDimitry Andric     std::unique_ptr<LineSequence> sequence(
178314f1b3e8SDimitry Andric         line_table->CreateLineSequenceContainer());
1784f73363f1SDimitry Andric     auto lines = m_session_up->findLineNumbers(*compiland_up, *file);
1785f73363f1SDimitry Andric     if (!lines)
1786f73363f1SDimitry Andric       continue;
1787f3fbd1c0SDimitry Andric     int entry_count = lines->getChildCount();
1788f3fbd1c0SDimitry Andric 
1789f3fbd1c0SDimitry Andric     uint64_t prev_addr;
1790f3fbd1c0SDimitry Andric     uint32_t prev_length;
1791f3fbd1c0SDimitry Andric     uint32_t prev_line;
1792f3fbd1c0SDimitry Andric     uint32_t prev_source_idx;
1793f3fbd1c0SDimitry Andric 
179414f1b3e8SDimitry Andric     for (int i = 0; i < entry_count; ++i) {
1795f3fbd1c0SDimitry Andric       auto line = lines->getChildAtIndex(i);
1796f3fbd1c0SDimitry Andric 
1797f3fbd1c0SDimitry Andric       uint64_t lno = line->getLineNumber();
1798f3fbd1c0SDimitry Andric       uint64_t addr = line->getVirtualAddress();
1799f3fbd1c0SDimitry Andric       uint32_t length = line->getLength();
1800f3fbd1c0SDimitry Andric       uint32_t source_id = line->getSourceFileId();
1801f3fbd1c0SDimitry Andric       uint32_t col = line->getColumnNumber();
1802f3fbd1c0SDimitry Andric       uint32_t source_idx = index_map[source_id];
1803f3fbd1c0SDimitry Andric 
1804f73363f1SDimitry Andric       // There was a gap between the current entry and the previous entry if
1805f73363f1SDimitry Andric       // the addresses don't perfectly line up.
1806f3fbd1c0SDimitry Andric       bool is_gap = (i > 0) && (prev_addr + prev_length < addr);
1807f3fbd1c0SDimitry Andric 
180814f1b3e8SDimitry Andric       // Before inserting the current entry, insert a terminal entry at the end
180974a628f7SDimitry Andric       // of the previous entry's address range if the current entry resulted in
181074a628f7SDimitry Andric       // a gap from the previous entry.
181114f1b3e8SDimitry Andric       if (is_gap && ShouldAddLine(match_line, prev_line, prev_length)) {
181214f1b3e8SDimitry Andric         line_table->AppendLineEntryToSequence(
181314f1b3e8SDimitry Andric             sequence.get(), prev_addr + prev_length, prev_line, 0,
1814f3fbd1c0SDimitry Andric             prev_source_idx, false, false, false, false, true);
1815f73363f1SDimitry Andric 
1816344a3780SDimitry Andric         line_table->InsertSequence(sequence.get());
1817cfca06d7SDimitry Andric         sequence = line_table->CreateLineSequenceContainer();
1818f3fbd1c0SDimitry Andric       }
1819f3fbd1c0SDimitry Andric 
182014f1b3e8SDimitry Andric       if (ShouldAddLine(match_line, lno, length)) {
1821f3fbd1c0SDimitry Andric         bool is_statement = line->isStatement();
1822f3fbd1c0SDimitry Andric         bool is_prologue = false;
1823f3fbd1c0SDimitry Andric         bool is_epilogue = false;
182414f1b3e8SDimitry Andric         auto func =
182514f1b3e8SDimitry Andric             m_session_up->findSymbolByAddress(addr, PDB_SymType::Function);
182614f1b3e8SDimitry Andric         if (func) {
1827f3fbd1c0SDimitry Andric           auto prologue = func->findOneChild<PDBSymbolFuncDebugStart>();
1828f73363f1SDimitry Andric           if (prologue)
1829f3fbd1c0SDimitry Andric             is_prologue = (addr == prologue->getVirtualAddress());
1830f3fbd1c0SDimitry Andric 
1831f3fbd1c0SDimitry Andric           auto epilogue = func->findOneChild<PDBSymbolFuncDebugEnd>();
1832f73363f1SDimitry Andric           if (epilogue)
1833f3fbd1c0SDimitry Andric             is_epilogue = (addr == epilogue->getVirtualAddress());
1834f3fbd1c0SDimitry Andric         }
1835f3fbd1c0SDimitry Andric 
183614f1b3e8SDimitry Andric         line_table->AppendLineEntryToSequence(sequence.get(), addr, lno, col,
183714f1b3e8SDimitry Andric                                               source_idx, is_statement, false,
1838f3fbd1c0SDimitry Andric                                               is_prologue, is_epilogue, false);
1839f3fbd1c0SDimitry Andric       }
1840f3fbd1c0SDimitry Andric 
1841f3fbd1c0SDimitry Andric       prev_addr = addr;
1842f3fbd1c0SDimitry Andric       prev_length = length;
1843f3fbd1c0SDimitry Andric       prev_line = lno;
1844f3fbd1c0SDimitry Andric       prev_source_idx = source_idx;
1845f3fbd1c0SDimitry Andric     }
1846f3fbd1c0SDimitry Andric 
184714f1b3e8SDimitry Andric     if (entry_count > 0 && ShouldAddLine(match_line, prev_line, prev_length)) {
1848f3fbd1c0SDimitry Andric       // The end is always a terminal entry, so insert it regardless.
184914f1b3e8SDimitry Andric       line_table->AppendLineEntryToSequence(
185014f1b3e8SDimitry Andric           sequence.get(), prev_addr + prev_length, prev_line, 0,
1851f3fbd1c0SDimitry Andric           prev_source_idx, false, false, false, false, true);
1852f3fbd1c0SDimitry Andric     }
1853f3fbd1c0SDimitry Andric 
1854cfca06d7SDimitry Andric     line_table->InsertSequence(sequence.get());
1855f3fbd1c0SDimitry Andric   }
1856f3fbd1c0SDimitry Andric 
1857f73363f1SDimitry Andric   if (line_table->GetSize()) {
185894994d37SDimitry Andric     comp_unit.SetLineTable(line_table.release());
1859f3fbd1c0SDimitry Andric     return true;
1860f3fbd1c0SDimitry Andric   }
1861f73363f1SDimitry Andric   return false;
1862f73363f1SDimitry Andric }
1863f3fbd1c0SDimitry Andric 
BuildSupportFileIdToSupportFileIndexMap(const PDBSymbolCompiland & compiland,llvm::DenseMap<uint32_t,uint32_t> & index_map) const186414f1b3e8SDimitry Andric void SymbolFilePDB::BuildSupportFileIdToSupportFileIndexMap(
1865f73363f1SDimitry Andric     const PDBSymbolCompiland &compiland,
186614f1b3e8SDimitry Andric     llvm::DenseMap<uint32_t, uint32_t> &index_map) const {
1867f73363f1SDimitry Andric   // This is a hack, but we need to convert the source id into an index into
1868f73363f1SDimitry Andric   // the support files array.  We don't want to do path comparisons to avoid
186974a628f7SDimitry Andric   // basename / full path issues that may or may not even be a problem, so we
187074a628f7SDimitry Andric   // use the globally unique source file identifiers.  Ideally we could use the
187174a628f7SDimitry Andric   // global identifiers everywhere, but LineEntry currently assumes indices.
1872f73363f1SDimitry Andric   auto source_files = m_session_up->getSourceFilesForCompiland(compiland);
1873f73363f1SDimitry Andric   if (!source_files)
1874f73363f1SDimitry Andric     return;
1875f73363f1SDimitry Andric 
1876706b4fc4SDimitry Andric   int index = 0;
187714f1b3e8SDimitry Andric   while (auto file = source_files->getNext()) {
1878f3fbd1c0SDimitry Andric     uint32_t source_id = file->getUniqueId();
1879f3fbd1c0SDimitry Andric     index_map[source_id] = index++;
1880f3fbd1c0SDimitry Andric   }
1881f3fbd1c0SDimitry Andric }
1882f73363f1SDimitry Andric 
GetCompileUnitContainsAddress(const lldb_private::Address & so_addr)1883f73363f1SDimitry Andric lldb::CompUnitSP SymbolFilePDB::GetCompileUnitContainsAddress(
1884f73363f1SDimitry Andric     const lldb_private::Address &so_addr) {
1885f73363f1SDimitry Andric   lldb::addr_t file_vm_addr = so_addr.GetFileAddress();
1886f73363f1SDimitry Andric   if (file_vm_addr == LLDB_INVALID_ADDRESS || file_vm_addr == 0)
1887f73363f1SDimitry Andric     return nullptr;
1888f73363f1SDimitry Andric 
1889f73363f1SDimitry Andric   // If it is a PDB function's vm addr, this is the first sure bet.
1890f73363f1SDimitry Andric   if (auto lines =
1891f73363f1SDimitry Andric           m_session_up->findLineNumbersByAddress(file_vm_addr, /*Length=*/1)) {
1892f73363f1SDimitry Andric     if (auto first_line = lines->getNext())
1893f73363f1SDimitry Andric       return ParseCompileUnitForUID(first_line->getCompilandId());
1894f73363f1SDimitry Andric   }
1895f73363f1SDimitry Andric 
1896f73363f1SDimitry Andric   // Otherwise we resort to section contributions.
1897f73363f1SDimitry Andric   if (auto sec_contribs = m_session_up->getSectionContribs()) {
1898f73363f1SDimitry Andric     while (auto section = sec_contribs->getNext()) {
1899f73363f1SDimitry Andric       auto va = section->getVirtualAddress();
1900f73363f1SDimitry Andric       if (file_vm_addr >= va && file_vm_addr < va + section->getLength())
1901f73363f1SDimitry Andric         return ParseCompileUnitForUID(section->getCompilandId());
1902f73363f1SDimitry Andric     }
1903f73363f1SDimitry Andric   }
1904f73363f1SDimitry Andric   return nullptr;
1905f73363f1SDimitry Andric }
1906f73363f1SDimitry Andric 
1907f73363f1SDimitry Andric Mangled
GetMangledForPDBFunc(const llvm::pdb::PDBSymbolFunc & pdb_func)1908f73363f1SDimitry Andric SymbolFilePDB::GetMangledForPDBFunc(const llvm::pdb::PDBSymbolFunc &pdb_func) {
1909f73363f1SDimitry Andric   Mangled mangled;
1910f73363f1SDimitry Andric   auto func_name = pdb_func.getName();
1911f73363f1SDimitry Andric   auto func_undecorated_name = pdb_func.getUndecoratedName();
1912f73363f1SDimitry Andric   std::string func_decorated_name;
1913f73363f1SDimitry Andric 
1914f73363f1SDimitry Andric   // Seek from public symbols for non-static function's decorated name if any.
1915f73363f1SDimitry Andric   // For static functions, they don't have undecorated names and aren't exposed
1916f73363f1SDimitry Andric   // in Public Symbols either.
1917f73363f1SDimitry Andric   if (!func_undecorated_name.empty()) {
1918f73363f1SDimitry Andric     auto result_up = m_global_scope_up->findChildren(
1919f73363f1SDimitry Andric         PDB_SymType::PublicSymbol, func_undecorated_name,
1920f73363f1SDimitry Andric         PDB_NameSearchFlags::NS_UndecoratedName);
1921f73363f1SDimitry Andric     if (result_up) {
1922f73363f1SDimitry Andric       while (auto symbol_up = result_up->getNext()) {
1923f73363f1SDimitry Andric         // For a public symbol, it is unique.
1924f73363f1SDimitry Andric         lldbassert(result_up->getChildCount() == 1);
1925f73363f1SDimitry Andric         if (auto *pdb_public_sym =
1926f73363f1SDimitry Andric                 llvm::dyn_cast_or_null<PDBSymbolPublicSymbol>(
1927f73363f1SDimitry Andric                     symbol_up.get())) {
1928f73363f1SDimitry Andric           if (pdb_public_sym->isFunction()) {
1929f73363f1SDimitry Andric             func_decorated_name = pdb_public_sym->getName();
1930f73363f1SDimitry Andric             break;
1931f73363f1SDimitry Andric           }
1932f73363f1SDimitry Andric         }
1933f73363f1SDimitry Andric       }
1934f73363f1SDimitry Andric     }
1935f73363f1SDimitry Andric   }
1936f73363f1SDimitry Andric   if (!func_decorated_name.empty()) {
1937f73363f1SDimitry Andric     mangled.SetMangledName(ConstString(func_decorated_name));
1938f73363f1SDimitry Andric 
1939e3b55780SDimitry Andric     // For MSVC, format of C function's decorated name depends on calling
1940cfca06d7SDimitry Andric     // convention. Unfortunately none of the format is recognized by current
1941f73363f1SDimitry Andric     // LLDB. For example, `_purecall` is a __cdecl C function. From PDB,
1942f73363f1SDimitry Andric     // `__purecall` is retrieved as both its decorated and undecorated name
1943f73363f1SDimitry Andric     // (using PDBSymbolFunc::getUndecoratedName method). However `__purecall`
1944f73363f1SDimitry Andric     // string is not treated as mangled in LLDB (neither `?` nor `_Z` prefix).
1945f73363f1SDimitry Andric     // Mangled::GetDemangledName method will fail internally and caches an
1946cfca06d7SDimitry Andric     // empty string as its undecorated name. So we will face a contradiction
1947f73363f1SDimitry Andric     // here for the same symbol:
1948f73363f1SDimitry Andric     //   non-empty undecorated name from PDB
1949f73363f1SDimitry Andric     //   empty undecorated name from LLDB
1950cfca06d7SDimitry Andric     if (!func_undecorated_name.empty() && mangled.GetDemangledName().IsEmpty())
1951f73363f1SDimitry Andric       mangled.SetDemangledName(ConstString(func_undecorated_name));
1952f73363f1SDimitry Andric 
1953f73363f1SDimitry Andric     // LLDB uses several flags to control how a C++ decorated name is
1954f73363f1SDimitry Andric     // undecorated for MSVC. See `safeUndecorateName` in Class Mangled. So the
1955f73363f1SDimitry Andric     // yielded name could be different from what we retrieve from
1956f73363f1SDimitry Andric     // PDB source unless we also apply same flags in getting undecorated
1957f73363f1SDimitry Andric     // name through PDBSymbolFunc::getUndecoratedNameEx method.
1958f73363f1SDimitry Andric     if (!func_undecorated_name.empty() &&
1959cfca06d7SDimitry Andric         mangled.GetDemangledName() != ConstString(func_undecorated_name))
1960f73363f1SDimitry Andric       mangled.SetDemangledName(ConstString(func_undecorated_name));
1961f73363f1SDimitry Andric   } else if (!func_undecorated_name.empty()) {
1962f73363f1SDimitry Andric     mangled.SetDemangledName(ConstString(func_undecorated_name));
1963f73363f1SDimitry Andric   } else if (!func_name.empty())
19647fa27ce4SDimitry Andric     mangled.SetValue(ConstString(func_name));
1965f73363f1SDimitry Andric 
1966f73363f1SDimitry Andric   return mangled;
1967f73363f1SDimitry Andric }
1968f73363f1SDimitry Andric 
DeclContextMatchesThisSymbolFile(const lldb_private::CompilerDeclContext & decl_ctx)1969f73363f1SDimitry Andric bool SymbolFilePDB::DeclContextMatchesThisSymbolFile(
1970cfca06d7SDimitry Andric     const lldb_private::CompilerDeclContext &decl_ctx) {
1971cfca06d7SDimitry Andric   if (!decl_ctx.IsValid())
1972f73363f1SDimitry Andric     return true;
1973f73363f1SDimitry Andric 
1974cfca06d7SDimitry Andric   TypeSystem *decl_ctx_type_system = decl_ctx.GetTypeSystem();
1975f73363f1SDimitry Andric   if (!decl_ctx_type_system)
1976f73363f1SDimitry Andric     return false;
1977ead24645SDimitry Andric   auto type_system_or_err = GetTypeSystemForLanguage(
1978f73363f1SDimitry Andric       decl_ctx_type_system->GetMinimumLanguage(nullptr));
1979ead24645SDimitry Andric   if (auto err = type_system_or_err.takeError()) {
1980ead24645SDimitry Andric     LLDB_LOG_ERROR(
1981145449b1SDimitry Andric         GetLog(LLDBLog::Symbols), std::move(err),
19827fa27ce4SDimitry Andric         "Unable to determine if DeclContext matches this symbol file: {0}");
1983ead24645SDimitry Andric     return false;
1984ead24645SDimitry Andric   }
1985ead24645SDimitry Andric 
1986e3b55780SDimitry Andric   if (decl_ctx_type_system == type_system_or_err->get())
1987f73363f1SDimitry Andric     return true; // The type systems match, return true
1988f73363f1SDimitry Andric 
1989f73363f1SDimitry Andric   return false;
1990f73363f1SDimitry Andric }
199194994d37SDimitry Andric 
GetCompilandId(const llvm::pdb::PDBSymbolData & data)199294994d37SDimitry Andric uint32_t SymbolFilePDB::GetCompilandId(const llvm::pdb::PDBSymbolData &data) {
199394994d37SDimitry Andric   static const auto pred_upper = [](uint32_t lhs, SecContribInfo rhs) {
199494994d37SDimitry Andric     return lhs < rhs.Offset;
199594994d37SDimitry Andric   };
199694994d37SDimitry Andric 
199794994d37SDimitry Andric   // Cache section contributions
199894994d37SDimitry Andric   if (m_sec_contribs.empty()) {
199994994d37SDimitry Andric     if (auto SecContribs = m_session_up->getSectionContribs()) {
200094994d37SDimitry Andric       while (auto SectionContrib = SecContribs->getNext()) {
200194994d37SDimitry Andric         auto comp_id = SectionContrib->getCompilandId();
200294994d37SDimitry Andric         if (!comp_id)
200394994d37SDimitry Andric           continue;
200494994d37SDimitry Andric 
200594994d37SDimitry Andric         auto sec = SectionContrib->getAddressSection();
200694994d37SDimitry Andric         auto &sec_cs = m_sec_contribs[sec];
200794994d37SDimitry Andric 
200894994d37SDimitry Andric         auto offset = SectionContrib->getAddressOffset();
2009e3b55780SDimitry Andric         auto it = llvm::upper_bound(sec_cs, offset, pred_upper);
201094994d37SDimitry Andric 
201194994d37SDimitry Andric         auto size = SectionContrib->getLength();
201294994d37SDimitry Andric         sec_cs.insert(it, {offset, size, comp_id});
201394994d37SDimitry Andric       }
201494994d37SDimitry Andric     }
201594994d37SDimitry Andric   }
201694994d37SDimitry Andric 
201794994d37SDimitry Andric   // Check by line number
201894994d37SDimitry Andric   if (auto Lines = data.getLineNumbers()) {
201994994d37SDimitry Andric     if (auto FirstLine = Lines->getNext())
202094994d37SDimitry Andric       return FirstLine->getCompilandId();
202194994d37SDimitry Andric   }
202294994d37SDimitry Andric 
202394994d37SDimitry Andric   // Retrieve section + offset
202494994d37SDimitry Andric   uint32_t DataSection = data.getAddressSection();
202594994d37SDimitry Andric   uint32_t DataOffset = data.getAddressOffset();
202694994d37SDimitry Andric   if (DataSection == 0) {
202794994d37SDimitry Andric     if (auto RVA = data.getRelativeVirtualAddress())
202894994d37SDimitry Andric       m_session_up->addressForRVA(RVA, DataSection, DataOffset);
202994994d37SDimitry Andric   }
203094994d37SDimitry Andric 
203194994d37SDimitry Andric   if (DataSection) {
203294994d37SDimitry Andric     // Search by section contributions
203394994d37SDimitry Andric     auto &sec_cs = m_sec_contribs[DataSection];
2034e3b55780SDimitry Andric     auto it = llvm::upper_bound(sec_cs, DataOffset, pred_upper);
203594994d37SDimitry Andric     if (it != sec_cs.begin()) {
203694994d37SDimitry Andric       --it;
203794994d37SDimitry Andric       if (DataOffset < it->Offset + it->Size)
203894994d37SDimitry Andric         return it->CompilandId;
203994994d37SDimitry Andric     }
204094994d37SDimitry Andric   } else {
204194994d37SDimitry Andric     // Search in lexical tree
204294994d37SDimitry Andric     auto LexParentId = data.getLexicalParentId();
204394994d37SDimitry Andric     while (auto LexParent = m_session_up->getSymbolById(LexParentId)) {
204494994d37SDimitry Andric       if (LexParent->getSymTag() == PDB_SymType::Exe)
204594994d37SDimitry Andric         break;
204694994d37SDimitry Andric       if (LexParent->getSymTag() == PDB_SymType::Compiland)
204794994d37SDimitry Andric         return LexParentId;
204894994d37SDimitry Andric       LexParentId = LexParent->getRawSymbol().getLexicalParentId();
204994994d37SDimitry Andric     }
205094994d37SDimitry Andric   }
205194994d37SDimitry Andric 
205294994d37SDimitry Andric   return 0;
205394994d37SDimitry Andric }
2054