1 //===- NativeExeSymbol.cpp - native impl for PDBSymbolExe -------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/DebugInfo/PDB/Native/NativeExeSymbol.h"
11 
12 #include "llvm/ADT/STLExtras.h"
13 #include "llvm/DebugInfo/PDB/Native/DbiStream.h"
14 #include "llvm/DebugInfo/PDB/Native/InfoStream.h"
15 #include "llvm/DebugInfo/PDB/Native/NativeCompilandSymbol.h"
16 #include "llvm/DebugInfo/PDB/Native/NativeEnumModules.h"
17 #include "llvm/DebugInfo/PDB/Native/PDBFile.h"
18 #include "llvm/DebugInfo/PDB/Native/SymbolCache.h"
19 #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
20 
21 using namespace llvm;
22 using namespace llvm::pdb;
23 
24 static DbiStream *getDbiStreamPtr(NativeSession &Session) {
25   Expected<DbiStream &> DbiS = Session.getPDBFile().getPDBDbiStream();
26   if (DbiS)
27     return &DbiS.get();
28 
29   consumeError(DbiS.takeError());
30   return nullptr;
31 }
32 
33 NativeExeSymbol::NativeExeSymbol(NativeSession &Session, SymIndexId SymbolId)
34     : NativeRawSymbol(Session, PDB_SymType::Exe, SymbolId),
35       Dbi(getDbiStreamPtr(Session)) {}
36 
37 std::unique_ptr<IPDBEnumSymbols>
38 NativeExeSymbol::findChildren(PDB_SymType Type) const {
39   switch (Type) {
40   case PDB_SymType::Compiland: {
41     return std::unique_ptr<IPDBEnumSymbols>(new NativeEnumModules(Session));
42     break;
43   }
44   case PDB_SymType::ArrayType:
45     return Session.getSymbolCache().createTypeEnumerator(codeview::LF_ARRAY);
46   case PDB_SymType::Enum:
47     return Session.getSymbolCache().createTypeEnumerator(codeview::LF_ENUM);
48   case PDB_SymType::PointerType:
49     return Session.getSymbolCache().createTypeEnumerator(codeview::LF_POINTER);
50   case PDB_SymType::UDT:
51     return Session.getSymbolCache().createTypeEnumerator(
52         {codeview::LF_STRUCTURE, codeview::LF_CLASS, codeview::LF_UNION,
53          codeview::LF_INTERFACE});
54   case PDB_SymType::FunctionSig:
55     return Session.getSymbolCache().createTypeEnumerator(
56         {codeview::LF_PROCEDURE, codeview::LF_MFUNCTION});
57 
58   default:
59     break;
60   }
61   return nullptr;
62 }
63 
64 uint32_t NativeExeSymbol::getAge() const {
65   auto IS = Session.getPDBFile().getPDBInfoStream();
66   if (IS)
67     return IS->getAge();
68   consumeError(IS.takeError());
69   return 0;
70 }
71 
72 std::string NativeExeSymbol::getSymbolsFileName() const {
73   return Session.getPDBFile().getFilePath();
74 }
75 
76 codeview::GUID NativeExeSymbol::getGuid() const {
77   auto IS = Session.getPDBFile().getPDBInfoStream();
78   if (IS)
79     return IS->getGuid();
80   consumeError(IS.takeError());
81   return codeview::GUID{{0}};
82 }
83 
84 bool NativeExeSymbol::hasCTypes() const {
85   auto Dbi = Session.getPDBFile().getPDBDbiStream();
86   if (Dbi)
87     return Dbi->hasCTypes();
88   consumeError(Dbi.takeError());
89   return false;
90 }
91 
92 bool NativeExeSymbol::hasPrivateSymbols() const {
93   auto Dbi = Session.getPDBFile().getPDBDbiStream();
94   if (Dbi)
95     return !Dbi->isStripped();
96   consumeError(Dbi.takeError());
97   return false;
98 }
99