1 //===- InfoStream.cpp - PDB Info Stream (Stream 1) Access -------*- 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/InfoStream.h"
11 #include "llvm/ADT/BitVector.h"
12 #include "llvm/ADT/SmallVector.h"
13 #include "llvm/DebugInfo/PDB/Native/PDBFile.h"
14 #include "llvm/DebugInfo/PDB/Native/RawConstants.h"
15 #include "llvm/DebugInfo/PDB/Native/RawError.h"
16 #include "llvm/DebugInfo/PDB/Native/RawTypes.h"
17 #include "llvm/Support/BinaryStreamReader.h"
18 #include "llvm/Support/BinaryStreamWriter.h"
19 
20 using namespace llvm;
21 using namespace llvm::codeview;
22 using namespace llvm::msf;
23 using namespace llvm::pdb;
24 
25 InfoStream::InfoStream(std::unique_ptr<MappedBlockStream> Stream)
26     : Stream(std::move(Stream)) {}
27 
28 Error InfoStream::reload() {
29   BinaryStreamReader Reader(*Stream);
30 
31   const InfoStreamHeader *H;
32   if (auto EC = Reader.readObject(H))
33     return joinErrors(
34         std::move(EC),
35         make_error<RawError>(raw_error_code::corrupt_file,
36                              "PDB Stream does not contain a header."));
37 
38   switch (H->Version) {
39   case PdbImplVC70:
40   case PdbImplVC80:
41   case PdbImplVC110:
42   case PdbImplVC140:
43     break;
44   default:
45     return make_error<RawError>(raw_error_code::corrupt_file,
46                                 "Unsupported PDB stream version.");
47   }
48 
49   Version = H->Version;
50   Signature = H->Signature;
51   Age = H->Age;
52   Guid = H->Guid;
53 
54   uint32_t Offset = Reader.getOffset();
55   if (auto EC = NamedStreams.load(Reader))
56     return EC;
57   uint32_t NewOffset = Reader.getOffset();
58   NamedStreamMapByteSize = NewOffset - Offset;
59 
60   bool Stop = false;
61   while (!Stop && !Reader.empty()) {
62     PdbRaw_FeatureSig Sig;
63     if (auto EC = Reader.readEnum(Sig))
64       return EC;
65     switch (Sig) {
66     case PdbRaw_FeatureSig::VC110:
67       // No other flags for VC110 PDB.
68       Stop = true;
69       LLVM_FALLTHROUGH;
70     case PdbRaw_FeatureSig::VC140:
71       Features |= PdbFeatureContainsIdStream;
72       break;
73     case PdbRaw_FeatureSig::NoTypeMerge:
74       Features |= PdbFeatureNoTypeMerging;
75       break;
76     case PdbRaw_FeatureSig::MinimalDebugInfo:
77       Features |= PdbFeatureMinimalDebugInfo;
78     default:
79       continue;
80     }
81     FeatureSignatures.push_back(Sig);
82   }
83   return Error::success();
84 }
85 
86 uint32_t InfoStream::getStreamSize() const { return Stream->getLength(); }
87 
88 uint32_t InfoStream::getNamedStreamIndex(llvm::StringRef Name) const {
89   uint32_t Result;
90   if (!NamedStreams.get(Name, Result))
91     return 0;
92   return Result;
93 }
94 
95 iterator_range<StringMapConstIterator<uint32_t>>
96 InfoStream::named_streams() const {
97   return NamedStreams.entries();
98 }
99 
100 PdbRaw_ImplVer InfoStream::getVersion() const {
101   return static_cast<PdbRaw_ImplVer>(Version);
102 }
103 
104 uint32_t InfoStream::getSignature() const { return Signature; }
105 
106 uint32_t InfoStream::getAge() const { return Age; }
107 
108 PDB_UniqueId InfoStream::getGuid() const { return Guid; }
109 
110 uint32_t InfoStream::getNamedStreamMapByteSize() const {
111   return NamedStreamMapByteSize;
112 }
113 
114 PdbRaw_Features InfoStream::getFeatures() const { return Features; }
115 
116 ArrayRef<PdbRaw_FeatureSig> InfoStream::getFeatureSignatures() const {
117   return FeatureSignatures;
118 }
119 
120 const NamedStreamMap &InfoStream::getNamedStreams() const {
121   return NamedStreams;
122 }
123