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/RawConstants.h" 14 #include "llvm/DebugInfo/PDB/Native/RawError.h" 15 #include "llvm/DebugInfo/PDB/Native/RawTypes.h" 16 #include "llvm/Support/BinaryStreamReader.h" 17 18 using namespace llvm; 19 using namespace llvm::codeview; 20 using namespace llvm::msf; 21 using namespace llvm::pdb; 22 23 InfoStream::InfoStream(std::unique_ptr<BinaryStream> Stream) 24 : Stream(std::move(Stream)), Header(nullptr) {} 25 26 Error InfoStream::reload() { 27 BinaryStreamReader Reader(*Stream); 28 29 if (auto EC = Reader.readObject(Header)) 30 return joinErrors( 31 std::move(EC), 32 make_error<RawError>(raw_error_code::corrupt_file, 33 "PDB Stream does not contain a header.")); 34 35 switch (Header->Version) { 36 case PdbImplVC70: 37 case PdbImplVC80: 38 case PdbImplVC110: 39 case PdbImplVC140: 40 break; 41 default: 42 return make_error<RawError>(raw_error_code::corrupt_file, 43 "Unsupported PDB stream version."); 44 } 45 46 uint32_t Offset = Reader.getOffset(); 47 if (auto EC = NamedStreams.load(Reader)) 48 return EC; 49 uint32_t NewOffset = Reader.getOffset(); 50 NamedStreamMapByteSize = NewOffset - Offset; 51 52 Reader.setOffset(Offset); 53 if (auto EC = Reader.readSubstream(SubNamedStreams, NamedStreamMapByteSize)) 54 return EC; 55 56 bool Stop = false; 57 while (!Stop && !Reader.empty()) { 58 PdbRaw_FeatureSig Sig; 59 if (auto EC = Reader.readEnum(Sig)) 60 return EC; 61 // Since this value comes from a file, it's possible we have some strange 62 // value which doesn't correspond to any value. We don't want to warn on 63 // -Wcovered-switch-default in this case, so switch on the integral value 64 // instead of the enumeration value. 65 switch (uint32_t(Sig)) { 66 case uint32_t(PdbRaw_FeatureSig::VC110): 67 // No other flags for VC110 PDB. 68 Stop = true; 69 LLVM_FALLTHROUGH; 70 case uint32_t(PdbRaw_FeatureSig::VC140): 71 Features |= PdbFeatureContainsIdStream; 72 break; 73 case uint32_t(PdbRaw_FeatureSig::NoTypeMerge): 74 Features |= PdbFeatureNoTypeMerging; 75 break; 76 case uint32_t(PdbRaw_FeatureSig::MinimalDebugInfo): 77 Features |= PdbFeatureMinimalDebugInfo; 78 break; 79 default: 80 continue; 81 } 82 FeatureSignatures.push_back(Sig); 83 } 84 return Error::success(); 85 } 86 87 uint32_t InfoStream::getStreamSize() const { return Stream->getLength(); } 88 89 Expected<uint32_t> InfoStream::getNamedStreamIndex(llvm::StringRef Name) const { 90 uint32_t Result; 91 if (!NamedStreams.get(Name, Result)) 92 return make_error<RawError>(raw_error_code::no_stream); 93 return Result; 94 } 95 96 StringMap<uint32_t> InfoStream::named_streams() const { 97 return NamedStreams.entries(); 98 } 99 100 bool InfoStream::containsIdStream() const { 101 return !!(Features & PdbFeatureContainsIdStream); 102 } 103 104 PdbRaw_ImplVer InfoStream::getVersion() const { 105 return static_cast<PdbRaw_ImplVer>(uint32_t(Header->Version)); 106 } 107 108 uint32_t InfoStream::getSignature() const { 109 return uint32_t(Header->Signature); 110 } 111 112 uint32_t InfoStream::getAge() const { return uint32_t(Header->Age); } 113 114 GUID InfoStream::getGuid() const { return Header->Guid; } 115 116 uint32_t InfoStream::getNamedStreamMapByteSize() const { 117 return NamedStreamMapByteSize; 118 } 119 120 PdbRaw_Features InfoStream::getFeatures() const { return Features; } 121 122 ArrayRef<PdbRaw_FeatureSig> InfoStream::getFeatureSignatures() const { 123 return FeatureSignatures; 124 } 125 126 const NamedStreamMap &InfoStream::getNamedStreams() const { 127 return NamedStreams; 128 } 129 130 BinarySubstreamRef InfoStream::getNamedStreamsBuffer() const { 131 return SubNamedStreams; 132 } 133