Line data    Source code

       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          86 : InfoStream::InfoStream(std::unique_ptr<BinaryStream> Stream)
      24         172 :     : Stream(std::move(Stream)), Header(nullptr) {}
      25             : 
      26          86 : Error InfoStream::reload() {
      27          86 :   BinaryStreamReader Reader(*Stream);
      28             : 
      29         172 :   if (auto EC = Reader.readObject(Header))
      30             :     return joinErrors(
      31             :         std::move(EC),
      32           0 :         make_error<RawError>(raw_error_code::corrupt_file,
      33           0 :                              "PDB Stream does not contain a header."));
      34             : 
      35         172 :   switch (Header->Version) {
      36             :   case PdbImplVC70:
      37             :   case PdbImplVC80:
      38             :   case PdbImplVC110:
      39             :   case PdbImplVC140:
      40             :     break;
      41           0 :   default:
      42             :     return make_error<RawError>(raw_error_code::corrupt_file,
      43             :                                 "Unsupported PDB stream version.");
      44             :   }
      45             : 
      46          86 :   uint32_t Offset = Reader.getOffset();
      47         172 :   if (auto EC = NamedStreams.load(Reader))
      48             :     return EC;
      49          86 :   uint32_t NewOffset = Reader.getOffset();
      50          86 :   NamedStreamMapByteSize = NewOffset - Offset;
      51             : 
      52             :   Reader.setOffset(Offset);
      53         172 :   if (auto EC = Reader.readSubstream(SubNamedStreams, NamedStreamMapByteSize))
      54             :     return EC;
      55             : 
      56             :   bool Stop = false;
      57         471 :   while (!Stop && !Reader.empty()) {
      58             :     PdbRaw_FeatureSig Sig;
      59         324 :     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         162 :     switch (uint32_t(Sig)) {
      66          25 :     case uint32_t(PdbRaw_FeatureSig::VC110):
      67             :       // No other flags for VC110 PDB.
      68             :       Stop = true;
      69             :       LLVM_FALLTHROUGH;
      70          76 :     case uint32_t(PdbRaw_FeatureSig::VC140):
      71             :       Features |= PdbFeatureContainsIdStream;
      72             :       break;
      73           0 :     case uint32_t(PdbRaw_FeatureSig::NoTypeMerge):
      74             :       Features |= PdbFeatureNoTypeMerging;
      75             :       break;
      76           0 :     case uint32_t(PdbRaw_FeatureSig::MinimalDebugInfo):
      77             :       Features |= PdbFeatureMinimalDebugInfo;
      78             :       break;
      79          86 :     default:
      80          86 :       continue;
      81             :     }
      82          76 :     FeatureSignatures.push_back(Sig);
      83             :   }
      84             :   return Error::success();
      85             : }
      86             : 
      87           0 : uint32_t InfoStream::getStreamSize() const { return Stream->getLength(); }
      88             : 
      89          41 : Expected<uint32_t> InfoStream::getNamedStreamIndex(llvm::StringRef Name) const {
      90             :   uint32_t Result;
      91          41 :   if (!NamedStreams.get(Name, Result))
      92             :     return make_error<RawError>(raw_error_code::no_stream);
      93             :   return Result;
      94             : }
      95             : 
      96           9 : StringMap<uint32_t> InfoStream::named_streams() const {
      97           9 :   return NamedStreams.entries();
      98             : }
      99             : 
     100          99 : bool InfoStream::containsIdStream() const {
     101         198 :   return !!(Features & PdbFeatureContainsIdStream);
     102             : }
     103             : 
     104           9 : PdbRaw_ImplVer InfoStream::getVersion() const {
     105          18 :   return static_cast<PdbRaw_ImplVer>(uint32_t(Header->Version));
     106             : }
     107             : 
     108          13 : uint32_t InfoStream::getSignature() const {
     109          26 :   return uint32_t(Header->Signature);
     110             : }
     111             : 
     112          42 : uint32_t InfoStream::getAge() const { return uint32_t(Header->Age); }
     113             : 
     114          25 : GUID InfoStream::getGuid() const { return Header->Guid; }
     115             : 
     116           0 : uint32_t InfoStream::getNamedStreamMapByteSize() const {
     117           0 :   return NamedStreamMapByteSize;
     118             : }
     119             : 
     120           4 : PdbRaw_Features InfoStream::getFeatures() const { return Features; }
     121             : 
     122           9 : ArrayRef<PdbRaw_FeatureSig> InfoStream::getFeatureSignatures() const {
     123           9 :   return FeatureSignatures;
     124             : }
     125             : 
     126           0 : const NamedStreamMap &InfoStream::getNamedStreams() const {
     127           0 :   return NamedStreams;
     128             : }
     129             : 
     130           1 : BinarySubstreamRef InfoStream::getNamedStreamsBuffer() const {
     131           1 :   return SubNamedStreams;
     132             : }