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::Enum:
45     return Session.getSymbolCache().createTypeEnumerator(codeview::LF_ENUM);
46   case PDB_SymType::PointerType:
47     return Session.getSymbolCache().createTypeEnumerator(codeview::LF_POINTER);
48   case PDB_SymType::UDT:
49     return Session.getSymbolCache().createTypeEnumerator(
50         {codeview::LF_STRUCTURE, codeview::LF_CLASS, codeview::LF_UNION,
51          codeview::LF_INTERFACE});
52   default:
53     break;
54   }
55   return nullptr;
56 }
57 
58 uint32_t NativeExeSymbol::getAge() const {
59   auto IS = Session.getPDBFile().getPDBInfoStream();
60   if (IS)
61     return IS->getAge();
62   consumeError(IS.takeError());
63   return 0;
64 }
65 
66 std::string NativeExeSymbol::getSymbolsFileName() const {
67   return Session.getPDBFile().getFilePath();
68 }
69 
70 codeview::GUID NativeExeSymbol::getGuid() const {
71   auto IS = Session.getPDBFile().getPDBInfoStream();
72   if (IS)
73     return IS->getGuid();
74   consumeError(IS.takeError());
75   return codeview::GUID{{0}};
76 }
77 
78 bool NativeExeSymbol::hasCTypes() const {
79   auto Dbi = Session.getPDBFile().getPDBDbiStream();
80   if (Dbi)
81     return Dbi->hasCTypes();
82   consumeError(Dbi.takeError());
83   return false;
84 }
85 
86 bool NativeExeSymbol::hasPrivateSymbols() const {
87   auto Dbi = Session.getPDBFile().getPDBDbiStream();
88   if (Dbi)
89     return !Dbi->isStripped();
90   consumeError(Dbi.takeError());
91   return false;
92 }
93