blob: 6afe83cd90dd51339decda4634f3034455b0ca4e [file] [log] [blame]
Zachary Turner94269a42015-02-07 01:47:14 +00001//===- llvm/unittest/DebugInfo/PDB/PDBApiTest.cpp -------------------------===//
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 <unordered_map>
11
Zachary Turnerbe6d1e42015-02-10 23:46:48 +000012#include "llvm/ADT/STLExtras.h"
Zachary Turner94269a42015-02-07 01:47:14 +000013#include "llvm/DebugInfo/PDB/IPDBEnumChildren.h"
14#include "llvm/DebugInfo/PDB/IPDBRawSymbol.h"
Zachary Turnerbae16b32015-02-08 20:58:09 +000015#include "llvm/DebugInfo/PDB/IPDBSession.h"
16#include "llvm/DebugInfo/PDB/IPDBSourceFile.h"
17
Zachary Turner94269a42015-02-07 01:47:14 +000018#include "llvm/DebugInfo/PDB/PDBSymbol.h"
19#include "llvm/DebugInfo/PDB/PDBSymbolAnnotation.h"
20#include "llvm/DebugInfo/PDB/PDBSymbolBlock.h"
21#include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
22#include "llvm/DebugInfo/PDB/PDBSymbolCompilandDetails.h"
23#include "llvm/DebugInfo/PDB/PDBSymbolCompilandEnv.h"
24#include "llvm/DebugInfo/PDB/PDBSymbolCustom.h"
25#include "llvm/DebugInfo/PDB/PDBSymbolData.h"
26#include "llvm/DebugInfo/PDB/PDBSymbolExe.h"
27#include "llvm/DebugInfo/PDB/PDBSymbolFunc.h"
28#include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugEnd.h"
29#include "llvm/DebugInfo/PDB/PDBSymbolFuncDebugStart.h"
30#include "llvm/DebugInfo/PDB/PDBSymbolLabel.h"
31#include "llvm/DebugInfo/PDB/PDBSymbolPublicSymbol.h"
32#include "llvm/DebugInfo/PDB/PDBSymbolThunk.h"
33#include "llvm/DebugInfo/PDB/PDBSymbolTypeArray.h"
34#include "llvm/DebugInfo/PDB/PDBSymbolTypeBaseClass.h"
35#include "llvm/DebugInfo/PDB/PDBSymbolTypeBuiltin.h"
36#include "llvm/DebugInfo/PDB/PDBSymbolTypeCustom.h"
37#include "llvm/DebugInfo/PDB/PDBSymbolTypeDimension.h"
38#include "llvm/DebugInfo/PDB/PDBSymbolTypeEnum.h"
39#include "llvm/DebugInfo/PDB/PDBSymbolTypeFriend.h"
40#include "llvm/DebugInfo/PDB/PDBSymbolTypeFunctionArg.h"
41#include "llvm/DebugInfo/PDB/PDBSymbolTypeFunctionSig.h"
42#include "llvm/DebugInfo/PDB/PDBSymbolTypeManaged.h"
43#include "llvm/DebugInfo/PDB/PDBSymbolTypePointer.h"
44#include "llvm/DebugInfo/PDB/PDBSymbolTypeTypedef.h"
45#include "llvm/DebugInfo/PDB/PDBSymbolTypeUDT.h"
46#include "llvm/DebugInfo/PDB/PDBSymbolTypeVTable.h"
47#include "llvm/DebugInfo/PDB/PDBSymbolTypeVTableShape.h"
48#include "llvm/DebugInfo/PDB/PDBSymbolUnknown.h"
49#include "llvm/DebugInfo/PDB/PDBSymbolUsingNamespace.h"
50#include "llvm/DebugInfo/PDB/PDBTypes.h"
51#include "gtest/gtest.h"
52using namespace llvm;
Zachary Turnere13787e2016-05-04 21:06:12 +000053using namespace llvm::pdb;
Zachary Turner94269a42015-02-07 01:47:14 +000054
Zachary Turner94269a42015-02-07 01:47:14 +000055namespace {
56
57#define MOCK_SYMBOL_ACCESSOR(Func) \
Zachary Turner635c2c42015-02-07 02:02:23 +000058 decltype(std::declval<IPDBRawSymbol>().Func()) Func() const override { \
Zachary Turner94269a42015-02-07 01:47:14 +000059 typedef decltype(IPDBRawSymbol::Func()) ReturnType; \
60 return ReturnType(); \
61 }
62
Zachary Turnerbae16b32015-02-08 20:58:09 +000063class MockSession : public IPDBSession {
64 uint64_t getLoadAddress() const override { return 0; }
65 void setLoadAddress(uint64_t Address) override {}
Zachary Turnerc883a8c2017-04-12 23:18:21 +000066 std::unique_ptr<PDBSymbolExe> getGlobalScope() const override {
Zachary Turnerbae16b32015-02-08 20:58:09 +000067 return nullptr;
68 }
Zachary Turner0ca2bde2015-02-10 21:40:29 +000069 std::unique_ptr<PDBSymbol> getSymbolById(uint32_t SymbolId) const override {
70 return nullptr;
71 }
72 std::unique_ptr<IPDBSourceFile>
73 getSourceFileById(uint32_t SymbolId) const override {
Zachary Turnerbae16b32015-02-08 20:58:09 +000074 return nullptr;
75 }
Zachary Turner4b083542015-04-17 22:40:36 +000076
77 std::unique_ptr<PDBSymbol>
Zachary Turner292e92d2015-05-01 20:51:49 +000078 findSymbolByAddress(uint64_t Address, PDB_SymType Type) const override {
Zachary Turner4b083542015-04-17 22:40:36 +000079 return nullptr;
80 }
81 std::unique_ptr<IPDBEnumLineNumbers>
Zachary Turner43ec3af2016-02-18 18:47:29 +000082 findLineNumbers(const PDBSymbolCompiland &Compiland,
83 const IPDBSourceFile &File) const override {
84 return nullptr;
85 }
86 std::unique_ptr<IPDBEnumLineNumbers>
Zachary Turner4b083542015-04-17 22:40:36 +000087 findLineNumbersByAddress(uint64_t Address, uint32_t Length) const override {
88 return nullptr;
89 }
Zachary Turner43ec3af2016-02-18 18:47:29 +000090 std::unique_ptr<IPDBEnumSourceFiles>
91 findSourceFiles(const PDBSymbolCompiland *Compiland, llvm::StringRef Pattern,
92 PDB_NameSearchFlags Flags) const override {
93 return nullptr;
94 }
95 std::unique_ptr<IPDBSourceFile>
96 findOneSourceFile(const PDBSymbolCompiland *Compiland,
97 llvm::StringRef Pattern,
98 PDB_NameSearchFlags Flags) const override {
99 return nullptr;
100 }
101 std::unique_ptr<IPDBEnumChildren<PDBSymbolCompiland>>
102 findCompilandsForSourceFile(llvm::StringRef Pattern,
103 PDB_NameSearchFlags Flags) const override {
104 return nullptr;
105 }
106 std::unique_ptr<PDBSymbolCompiland>
107 findOneCompilandForSourceFile(llvm::StringRef Pattern,
108 PDB_NameSearchFlags Flags) const override {
109 return nullptr;
110 }
Zachary Turner4b083542015-04-17 22:40:36 +0000111
Zachary Turnera5549172015-02-10 22:43:25 +0000112 std::unique_ptr<IPDBEnumSourceFiles> getAllSourceFiles() const override {
113 return nullptr;
114 }
115 std::unique_ptr<IPDBEnumSourceFiles> getSourceFilesForCompiland(
116 const PDBSymbolCompiland &Compiland) const override {
Zachary Turnerbae16b32015-02-08 20:58:09 +0000117 return nullptr;
118 }
Zachary Turnerc9c9c342015-02-10 22:56:21 +0000119
120 std::unique_ptr<IPDBEnumDataStreams> getDebugStreams() const override {
121 return nullptr;
122 }
Zachary Turnerbae16b32015-02-08 20:58:09 +0000123};
124
Zachary Turner94269a42015-02-07 01:47:14 +0000125class MockRawSymbol : public IPDBRawSymbol {
126public:
Zachary Turnerf07ef222015-02-11 00:33:00 +0000127 MockRawSymbol(PDB_SymType SymType)
128 : Type(SymType) {}
Zachary Turner94269a42015-02-07 01:47:14 +0000129
Zachary Turnerbc42da02015-02-23 05:59:14 +0000130 void dump(raw_ostream &OS, int Indent) const override {}
Zachary Turner94269a42015-02-07 01:47:14 +0000131
Chandler Carruthc94234c2015-02-07 08:12:22 +0000132 std::unique_ptr<IPDBEnumSymbols>
Zachary Turnera5549172015-02-10 22:43:25 +0000133 findChildren(PDB_SymType Type) const override {
134 return nullptr;
135 }
136 std::unique_ptr<IPDBEnumSymbols>
Zachary Turner94269a42015-02-07 01:47:14 +0000137 findChildren(PDB_SymType Type, StringRef Name,
Chandler Carruthc94234c2015-02-07 08:12:22 +0000138 PDB_NameSearchFlags Flags) const override {
Zachary Turner94269a42015-02-07 01:47:14 +0000139 return nullptr;
140 }
Chandler Carruthc94234c2015-02-07 08:12:22 +0000141 std::unique_ptr<IPDBEnumSymbols>
Zachary Turner94269a42015-02-07 01:47:14 +0000142 findChildrenByRVA(PDB_SymType Type, StringRef Name, PDB_NameSearchFlags Flags,
Chandler Carruthc94234c2015-02-07 08:12:22 +0000143 uint32_t RVA) const override {
Zachary Turner94269a42015-02-07 01:47:14 +0000144 return nullptr;
145 }
Chandler Carruthc94234c2015-02-07 08:12:22 +0000146 std::unique_ptr<IPDBEnumSymbols>
147 findInlineFramesByRVA(uint32_t RVA) const override {
Zachary Turner94269a42015-02-07 01:47:14 +0000148 return nullptr;
149 }
150
Chandler Carruthc94234c2015-02-07 08:12:22 +0000151 void getDataBytes(llvm::SmallVector<uint8_t, 32> &bytes) const override {}
Zachary Turner21473f72015-02-08 00:29:29 +0000152 void getFrontEndVersion(VersionInfo &Version) const override {}
153 void getBackEndVersion(VersionInfo &Version) const override {}
Zachary Turner94269a42015-02-07 01:47:14 +0000154
155 PDB_SymType getSymTag() const override { return Type; }
156
157 MOCK_SYMBOL_ACCESSOR(getAccess)
158 MOCK_SYMBOL_ACCESSOR(getAddressOffset)
159 MOCK_SYMBOL_ACCESSOR(getAddressSection)
160 MOCK_SYMBOL_ACCESSOR(getAge)
161 MOCK_SYMBOL_ACCESSOR(getArrayIndexTypeId)
Zachary Turner94269a42015-02-07 01:47:14 +0000162 MOCK_SYMBOL_ACCESSOR(getBaseDataOffset)
163 MOCK_SYMBOL_ACCESSOR(getBaseDataSlot)
164 MOCK_SYMBOL_ACCESSOR(getBaseSymbolId)
165 MOCK_SYMBOL_ACCESSOR(getBuiltinType)
166 MOCK_SYMBOL_ACCESSOR(getBitPosition)
167 MOCK_SYMBOL_ACCESSOR(getCallingConvention)
168 MOCK_SYMBOL_ACCESSOR(getClassParentId)
169 MOCK_SYMBOL_ACCESSOR(getCompilerName)
170 MOCK_SYMBOL_ACCESSOR(getCount)
171 MOCK_SYMBOL_ACCESSOR(getCountLiveRanges)
Zachary Turner94269a42015-02-07 01:47:14 +0000172 MOCK_SYMBOL_ACCESSOR(getLanguage)
173 MOCK_SYMBOL_ACCESSOR(getLexicalParentId)
174 MOCK_SYMBOL_ACCESSOR(getLibraryName)
175 MOCK_SYMBOL_ACCESSOR(getLiveRangeStartAddressOffset)
176 MOCK_SYMBOL_ACCESSOR(getLiveRangeStartAddressSection)
177 MOCK_SYMBOL_ACCESSOR(getLiveRangeStartRelativeVirtualAddress)
178 MOCK_SYMBOL_ACCESSOR(getLocalBasePointerRegisterId)
179 MOCK_SYMBOL_ACCESSOR(getLowerBoundId)
180 MOCK_SYMBOL_ACCESSOR(getMemorySpaceKind)
181 MOCK_SYMBOL_ACCESSOR(getName)
182 MOCK_SYMBOL_ACCESSOR(getNumberOfAcceleratorPointerTags)
183 MOCK_SYMBOL_ACCESSOR(getNumberOfColumns)
184 MOCK_SYMBOL_ACCESSOR(getNumberOfModifiers)
185 MOCK_SYMBOL_ACCESSOR(getNumberOfRegisterIndices)
186 MOCK_SYMBOL_ACCESSOR(getNumberOfRows)
187 MOCK_SYMBOL_ACCESSOR(getObjectFileName)
188 MOCK_SYMBOL_ACCESSOR(getOemId)
189 MOCK_SYMBOL_ACCESSOR(getOemSymbolId)
190 MOCK_SYMBOL_ACCESSOR(getOffsetInUdt)
191 MOCK_SYMBOL_ACCESSOR(getPlatform)
192 MOCK_SYMBOL_ACCESSOR(getRank)
193 MOCK_SYMBOL_ACCESSOR(getRegisterId)
194 MOCK_SYMBOL_ACCESSOR(getRegisterType)
195 MOCK_SYMBOL_ACCESSOR(getRelativeVirtualAddress)
196 MOCK_SYMBOL_ACCESSOR(getSamplerSlot)
197 MOCK_SYMBOL_ACCESSOR(getSignature)
198 MOCK_SYMBOL_ACCESSOR(getSizeInUdt)
199 MOCK_SYMBOL_ACCESSOR(getSlot)
200 MOCK_SYMBOL_ACCESSOR(getSourceFileName)
201 MOCK_SYMBOL_ACCESSOR(getStride)
202 MOCK_SYMBOL_ACCESSOR(getSubTypeId)
203 MOCK_SYMBOL_ACCESSOR(getSymbolsFileName)
204 MOCK_SYMBOL_ACCESSOR(getSymIndexId)
205 MOCK_SYMBOL_ACCESSOR(getTargetOffset)
206 MOCK_SYMBOL_ACCESSOR(getTargetRelativeVirtualAddress)
207 MOCK_SYMBOL_ACCESSOR(getTargetVirtualAddress)
208 MOCK_SYMBOL_ACCESSOR(getTargetSection)
209 MOCK_SYMBOL_ACCESSOR(getTextureSlot)
210 MOCK_SYMBOL_ACCESSOR(getTimeStamp)
211 MOCK_SYMBOL_ACCESSOR(getToken)
212 MOCK_SYMBOL_ACCESSOR(getTypeId)
213 MOCK_SYMBOL_ACCESSOR(getUavSlot)
214 MOCK_SYMBOL_ACCESSOR(getUndecoratedName)
215 MOCK_SYMBOL_ACCESSOR(getUnmodifiedTypeId)
216 MOCK_SYMBOL_ACCESSOR(getUpperBoundId)
217 MOCK_SYMBOL_ACCESSOR(getVirtualBaseDispIndex)
218 MOCK_SYMBOL_ACCESSOR(getVirtualBaseOffset)
219 MOCK_SYMBOL_ACCESSOR(getVirtualTableShapeId)
220 MOCK_SYMBOL_ACCESSOR(getDataKind)
221 MOCK_SYMBOL_ACCESSOR(getGuid)
222 MOCK_SYMBOL_ACCESSOR(getOffset)
223 MOCK_SYMBOL_ACCESSOR(getThisAdjust)
224 MOCK_SYMBOL_ACCESSOR(getVirtualBasePointerOffset)
225 MOCK_SYMBOL_ACCESSOR(getLocationType)
226 MOCK_SYMBOL_ACCESSOR(getMachineType)
227 MOCK_SYMBOL_ACCESSOR(getThunkOrdinal)
228 MOCK_SYMBOL_ACCESSOR(getLength)
Zachary Turnerc883a8c2017-04-12 23:18:21 +0000229 MOCK_SYMBOL_ACCESSOR(getVirtualBaseTableType)
Zachary Turner94269a42015-02-07 01:47:14 +0000230 MOCK_SYMBOL_ACCESSOR(getLiveRangeLength)
231 MOCK_SYMBOL_ACCESSOR(getVirtualAddress)
232 MOCK_SYMBOL_ACCESSOR(getUdtKind)
233 MOCK_SYMBOL_ACCESSOR(hasConstructor)
234 MOCK_SYMBOL_ACCESSOR(hasCustomCallingConvention)
235 MOCK_SYMBOL_ACCESSOR(hasFarReturn)
236 MOCK_SYMBOL_ACCESSOR(isCode)
237 MOCK_SYMBOL_ACCESSOR(isCompilerGenerated)
238 MOCK_SYMBOL_ACCESSOR(isConstType)
239 MOCK_SYMBOL_ACCESSOR(isEditAndContinueEnabled)
240 MOCK_SYMBOL_ACCESSOR(isFunction)
241 MOCK_SYMBOL_ACCESSOR(getAddressTaken)
242 MOCK_SYMBOL_ACCESSOR(getNoStackOrdering)
243 MOCK_SYMBOL_ACCESSOR(hasAlloca)
244 MOCK_SYMBOL_ACCESSOR(hasAssignmentOperator)
245 MOCK_SYMBOL_ACCESSOR(hasCTypes)
246 MOCK_SYMBOL_ACCESSOR(hasCastOperator)
247 MOCK_SYMBOL_ACCESSOR(hasDebugInfo)
248 MOCK_SYMBOL_ACCESSOR(hasEH)
249 MOCK_SYMBOL_ACCESSOR(hasEHa)
Zachary Turnera5549172015-02-10 22:43:25 +0000250 MOCK_SYMBOL_ACCESSOR(hasFramePointer)
Zachary Turner94269a42015-02-07 01:47:14 +0000251 MOCK_SYMBOL_ACCESSOR(hasInlAsm)
252 MOCK_SYMBOL_ACCESSOR(hasInlineAttribute)
253 MOCK_SYMBOL_ACCESSOR(hasInterruptReturn)
254 MOCK_SYMBOL_ACCESSOR(hasLongJump)
255 MOCK_SYMBOL_ACCESSOR(hasManagedCode)
256 MOCK_SYMBOL_ACCESSOR(hasNestedTypes)
257 MOCK_SYMBOL_ACCESSOR(hasNoInlineAttribute)
258 MOCK_SYMBOL_ACCESSOR(hasNoReturnAttribute)
259 MOCK_SYMBOL_ACCESSOR(hasOptimizedCodeDebugInfo)
260 MOCK_SYMBOL_ACCESSOR(hasOverloadedOperator)
261 MOCK_SYMBOL_ACCESSOR(hasSEH)
262 MOCK_SYMBOL_ACCESSOR(hasSecurityChecks)
263 MOCK_SYMBOL_ACCESSOR(hasSetJump)
264 MOCK_SYMBOL_ACCESSOR(hasStrictGSCheck)
265 MOCK_SYMBOL_ACCESSOR(isAcceleratorGroupSharedLocal)
266 MOCK_SYMBOL_ACCESSOR(isAcceleratorPointerTagLiveRange)
267 MOCK_SYMBOL_ACCESSOR(isAcceleratorStubFunction)
268 MOCK_SYMBOL_ACCESSOR(isAggregated)
269 MOCK_SYMBOL_ACCESSOR(isIntroVirtualFunction)
270 MOCK_SYMBOL_ACCESSOR(isCVTCIL)
271 MOCK_SYMBOL_ACCESSOR(isConstructorVirtualBase)
272 MOCK_SYMBOL_ACCESSOR(isCxxReturnUdt)
273 MOCK_SYMBOL_ACCESSOR(isDataAligned)
274 MOCK_SYMBOL_ACCESSOR(isHLSLData)
275 MOCK_SYMBOL_ACCESSOR(isHotpatchable)
276 MOCK_SYMBOL_ACCESSOR(isIndirectVirtualBaseClass)
277 MOCK_SYMBOL_ACCESSOR(isInterfaceUdt)
278 MOCK_SYMBOL_ACCESSOR(isIntrinsic)
279 MOCK_SYMBOL_ACCESSOR(isLTCG)
280 MOCK_SYMBOL_ACCESSOR(isLocationControlFlowDependent)
281 MOCK_SYMBOL_ACCESSOR(isMSILNetmodule)
282 MOCK_SYMBOL_ACCESSOR(isMatrixRowMajor)
283 MOCK_SYMBOL_ACCESSOR(isManagedCode)
284 MOCK_SYMBOL_ACCESSOR(isMSILCode)
285 MOCK_SYMBOL_ACCESSOR(isMultipleInheritance)
286 MOCK_SYMBOL_ACCESSOR(isNaked)
287 MOCK_SYMBOL_ACCESSOR(isNested)
288 MOCK_SYMBOL_ACCESSOR(isOptimizedAway)
289 MOCK_SYMBOL_ACCESSOR(isPacked)
290 MOCK_SYMBOL_ACCESSOR(isPointerBasedOnSymbolValue)
291 MOCK_SYMBOL_ACCESSOR(isPointerToDataMember)
292 MOCK_SYMBOL_ACCESSOR(isPointerToMemberFunction)
293 MOCK_SYMBOL_ACCESSOR(isPureVirtual)
294 MOCK_SYMBOL_ACCESSOR(isRValueReference)
295 MOCK_SYMBOL_ACCESSOR(isRefUdt)
296 MOCK_SYMBOL_ACCESSOR(isReference)
297 MOCK_SYMBOL_ACCESSOR(isRestrictedType)
298 MOCK_SYMBOL_ACCESSOR(isReturnValue)
299 MOCK_SYMBOL_ACCESSOR(isSafeBuffers)
300 MOCK_SYMBOL_ACCESSOR(isScoped)
301 MOCK_SYMBOL_ACCESSOR(isSdl)
302 MOCK_SYMBOL_ACCESSOR(isSingleInheritance)
303 MOCK_SYMBOL_ACCESSOR(isSplitted)
304 MOCK_SYMBOL_ACCESSOR(isStatic)
305 MOCK_SYMBOL_ACCESSOR(hasPrivateSymbols)
306 MOCK_SYMBOL_ACCESSOR(isUnalignedType)
307 MOCK_SYMBOL_ACCESSOR(isUnreached)
308 MOCK_SYMBOL_ACCESSOR(isValueUdt)
309 MOCK_SYMBOL_ACCESSOR(isVirtual)
310 MOCK_SYMBOL_ACCESSOR(isVirtualBaseClass)
311 MOCK_SYMBOL_ACCESSOR(isVirtualInheritance)
312 MOCK_SYMBOL_ACCESSOR(isVolatileType)
Zachary Turner00dbc752015-02-13 01:39:22 +0000313 MOCK_SYMBOL_ACCESSOR(getValue)
314 MOCK_SYMBOL_ACCESSOR(wasInlined)
315 MOCK_SYMBOL_ACCESSOR(getUnused)
Zachary Turner94269a42015-02-07 01:47:14 +0000316
317private:
318 PDB_SymType Type;
319};
320
321class PDBApiTest : public testing::Test {
322public:
323 std::unordered_map<PDB_SymType, std::unique_ptr<PDBSymbol>> SymbolMap;
324
325 void SetUp() override {
Zachary Turnerbae16b32015-02-08 20:58:09 +0000326 Session.reset(new MockSession());
327
Zachary Turner94269a42015-02-07 01:47:14 +0000328 InsertItemWithTag(PDB_SymType::None);
329 InsertItemWithTag(PDB_SymType::Exe);
330 InsertItemWithTag(PDB_SymType::Compiland);
331 InsertItemWithTag(PDB_SymType::CompilandDetails);
332 InsertItemWithTag(PDB_SymType::CompilandEnv);
333 InsertItemWithTag(PDB_SymType::Function);
334 InsertItemWithTag(PDB_SymType::Block);
335 InsertItemWithTag(PDB_SymType::Data);
336 InsertItemWithTag(PDB_SymType::Annotation);
337 InsertItemWithTag(PDB_SymType::Label);
338 InsertItemWithTag(PDB_SymType::PublicSymbol);
339 InsertItemWithTag(PDB_SymType::UDT);
340 InsertItemWithTag(PDB_SymType::Enum);
341 InsertItemWithTag(PDB_SymType::FunctionSig);
342 InsertItemWithTag(PDB_SymType::PointerType);
343 InsertItemWithTag(PDB_SymType::ArrayType);
344 InsertItemWithTag(PDB_SymType::BuiltinType);
345 InsertItemWithTag(PDB_SymType::Typedef);
346 InsertItemWithTag(PDB_SymType::BaseClass);
347 InsertItemWithTag(PDB_SymType::Friend);
348 InsertItemWithTag(PDB_SymType::FunctionArg);
349 InsertItemWithTag(PDB_SymType::FuncDebugStart);
350 InsertItemWithTag(PDB_SymType::FuncDebugEnd);
351 InsertItemWithTag(PDB_SymType::UsingNamespace);
352 InsertItemWithTag(PDB_SymType::VTableShape);
353 InsertItemWithTag(PDB_SymType::VTable);
354 InsertItemWithTag(PDB_SymType::Custom);
355 InsertItemWithTag(PDB_SymType::Thunk);
356 InsertItemWithTag(PDB_SymType::CustomType);
357 InsertItemWithTag(PDB_SymType::ManagedType);
358 InsertItemWithTag(PDB_SymType::Dimension);
359 InsertItemWithTag(PDB_SymType::Max);
360 }
361
Zachary Turner94269a42015-02-07 01:47:14 +0000362 template <class ExpectedType> void VerifyDyncast(PDB_SymType Tag) {
363 for (auto item = SymbolMap.begin(); item != SymbolMap.end(); ++item) {
364 EXPECT_EQ(item->first == Tag, llvm::isa<ExpectedType>(*item->second));
365 }
366 }
367
368 void VerifyUnknownDyncasts() {
369 for (auto item = SymbolMap.begin(); item != SymbolMap.end(); ++item) {
370 bool should_match = false;
371 if (item->first == PDB_SymType::None || item->first >= PDB_SymType::Max)
372 should_match = true;
373
374 EXPECT_EQ(should_match, llvm::isa<PDBSymbolUnknown>(*item->second));
375 }
376 }
Zachary Turnerbae16b32015-02-08 20:58:09 +0000377
378private:
379 std::unique_ptr<IPDBSession> Session;
380
381 void InsertItemWithTag(PDB_SymType Tag) {
Zachary Turnerf07ef222015-02-11 00:33:00 +0000382 auto RawSymbol = llvm::make_unique<MockRawSymbol>(Tag);
Zachary Turnerbae16b32015-02-08 20:58:09 +0000383 auto Symbol = PDBSymbol::create(*Session, std::move(RawSymbol));
384 SymbolMap.insert(std::make_pair(Tag, std::move(Symbol)));
385 }
Zachary Turner94269a42015-02-07 01:47:14 +0000386};
387
388TEST_F(PDBApiTest, Dyncast) {
389
390 // Most of the types have a one-to-one mapping between Tag and concrete type.
391 VerifyDyncast<PDBSymbolExe>(PDB_SymType::Exe);
392 VerifyDyncast<PDBSymbolCompiland>(PDB_SymType::Compiland);
393 VerifyDyncast<PDBSymbolCompilandDetails>(PDB_SymType::CompilandDetails);
394 VerifyDyncast<PDBSymbolCompilandEnv>(PDB_SymType::CompilandEnv);
395 VerifyDyncast<PDBSymbolFunc>(PDB_SymType::Function);
396 VerifyDyncast<PDBSymbolBlock>(PDB_SymType::Block);
397 VerifyDyncast<PDBSymbolData>(PDB_SymType::Data);
398 VerifyDyncast<PDBSymbolAnnotation>(PDB_SymType::Annotation);
399 VerifyDyncast<PDBSymbolLabel>(PDB_SymType::Label);
400 VerifyDyncast<PDBSymbolPublicSymbol>(PDB_SymType::PublicSymbol);
401 VerifyDyncast<PDBSymbolTypeUDT>(PDB_SymType::UDT);
402 VerifyDyncast<PDBSymbolTypeEnum>(PDB_SymType::Enum);
403 VerifyDyncast<PDBSymbolTypeFunctionSig>(PDB_SymType::FunctionSig);
404 VerifyDyncast<PDBSymbolTypePointer>(PDB_SymType::PointerType);
405 VerifyDyncast<PDBSymbolTypeArray>(PDB_SymType::ArrayType);
406 VerifyDyncast<PDBSymbolTypeBuiltin>(PDB_SymType::BuiltinType);
407 VerifyDyncast<PDBSymbolTypeTypedef>(PDB_SymType::Typedef);
408 VerifyDyncast<PDBSymbolTypeBaseClass>(PDB_SymType::BaseClass);
409 VerifyDyncast<PDBSymbolTypeFriend>(PDB_SymType::Friend);
410 VerifyDyncast<PDBSymbolTypeFunctionArg>(PDB_SymType::FunctionArg);
411 VerifyDyncast<PDBSymbolFuncDebugStart>(PDB_SymType::FuncDebugStart);
412 VerifyDyncast<PDBSymbolFuncDebugEnd>(PDB_SymType::FuncDebugEnd);
413 VerifyDyncast<PDBSymbolUsingNamespace>(PDB_SymType::UsingNamespace);
414 VerifyDyncast<PDBSymbolTypeVTableShape>(PDB_SymType::VTableShape);
415 VerifyDyncast<PDBSymbolTypeVTable>(PDB_SymType::VTable);
416 VerifyDyncast<PDBSymbolCustom>(PDB_SymType::Custom);
417 VerifyDyncast<PDBSymbolThunk>(PDB_SymType::Thunk);
418 VerifyDyncast<PDBSymbolTypeCustom>(PDB_SymType::CustomType);
419 VerifyDyncast<PDBSymbolTypeManaged>(PDB_SymType::ManagedType);
420 VerifyDyncast<PDBSymbolTypeDimension>(PDB_SymType::Dimension);
421
422 VerifyUnknownDyncasts();
423}
424
425} // end anonymous namespace