1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
| //===- TypeReferenceTracker.h --------------------------------- *- C++ --*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_TOOLS_LLVMPDBDUMP_TYPEREFERENCETRACKER_H
#define LLVM_TOOLS_LLVMPDBDUMP_TYPEREFERENCETRACKER_H
#include "InputFile.h"
#include "llvm/ADT/BitVector.h"
#include "llvm/ADT/Optional.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/DebugInfo/CodeView/CVRecord.h"
#include "llvm/DebugInfo/CodeView/TypeIndex.h"
#include "llvm/DebugInfo/CodeView/TypeIndexDiscovery.h"
#include "llvm/Support/Error.h"
namespace llvm {
namespace pdb {
class TpiStream;
/// Maintains bitvector to track whether a type was referenced by a symbol
/// record.
class TypeReferenceTracker {
public:
TypeReferenceTracker(InputFile &File);
// Do the work of marking referenced types.
void mark();
// Return true if a symbol record transitively references this type.
bool isTypeReferenced(codeview::TypeIndex TI) {
return TI.toArrayIndex() <= NumTypeRecords &&
TypeReferenced.test(TI.toArrayIndex());
}
private:
void addTypeRefsFromSymbol(const codeview::CVSymbol &Sym);
// Mark types on this list as referenced.
void addReferencedTypes(ArrayRef<uint8_t> RecData,
ArrayRef<codeview::TiReference> Refs);
// Consume all types on the worklist.
void markReferencedTypes();
void addOneTypeRef(codeview::TiRefKind RefKind, codeview::TypeIndex RefTI);
InputFile &File;
codeview::LazyRandomTypeCollection &Types;
codeview::LazyRandomTypeCollection *Ids = nullptr;
TpiStream *Tpi = nullptr;
BitVector TypeReferenced;
BitVector IdReferenced;
SmallVector<std::pair<codeview::TiRefKind, codeview::TypeIndex>, 10>
RefWorklist;
uint32_t NumTypeRecords = 0;
uint32_t NumIdRecords = 0;
};
} // namespace pdb
} // namespace llvm
#endif // LLVM_TOOLS_LLVMPDBDUMP_TYPEREFERENCETRACKER_H
|