faa554b2fd
Previously we would read a PDB, then write some of it back out, but write the directory, super block, and other pertinent metadata back out unchanged. This generates incorrect PDBs since the amount of data written was not always the same as the amount of data read. This patch changes things to use the newly introduced `MsfBuilder` class to write out a correct and accurate set of Msf metadata for the data *actually* written, which opens up the door for adding and removing type records, symbol records, and other types of data to an existing PDB. llvm-svn: 275627
91 lines
2.4 KiB
C++
91 lines
2.4 KiB
C++
//===- InfoStream.cpp - PDB Info Stream (Stream 1) Access -------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/DebugInfo/PDB/Raw/InfoStream.h"
|
|
#include "llvm/ADT/BitVector.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/DebugInfo/CodeView/StreamReader.h"
|
|
#include "llvm/DebugInfo/CodeView/StreamWriter.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/IndexedStreamData.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/PDBFile.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/RawConstants.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/RawError.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::codeview;
|
|
using namespace llvm::pdb;
|
|
|
|
InfoStream::InfoStream(std::unique_ptr<MappedBlockStream> Stream)
|
|
: Stream(std::move(Stream)) {}
|
|
|
|
Error InfoStream::reload() {
|
|
codeview::StreamReader Reader(*Stream);
|
|
|
|
const HeaderInfo *H;
|
|
if (auto EC = Reader.readObject(H))
|
|
return joinErrors(
|
|
std::move(EC),
|
|
make_error<RawError>(raw_error_code::corrupt_file,
|
|
"PDB Stream does not contain a header."));
|
|
|
|
switch (H->Version) {
|
|
case PdbImplVC70:
|
|
case PdbImplVC80:
|
|
case PdbImplVC110:
|
|
case PdbImplVC140:
|
|
break;
|
|
default:
|
|
return make_error<RawError>(raw_error_code::corrupt_file,
|
|
"Unsupported PDB stream version.");
|
|
}
|
|
|
|
Version = H->Version;
|
|
Signature = H->Signature;
|
|
Age = H->Age;
|
|
Guid = H->Guid;
|
|
|
|
return NamedStreams.load(Reader);
|
|
}
|
|
|
|
uint32_t InfoStream::getNamedStreamIndex(llvm::StringRef Name) const {
|
|
uint32_t Result;
|
|
if (!NamedStreams.tryGetValue(Name, Result))
|
|
return 0;
|
|
return Result;
|
|
}
|
|
|
|
iterator_range<StringMapConstIterator<uint32_t>>
|
|
InfoStream::named_streams() const {
|
|
return NamedStreams.entries();
|
|
}
|
|
|
|
PdbRaw_ImplVer InfoStream::getVersion() const {
|
|
return static_cast<PdbRaw_ImplVer>(Version);
|
|
}
|
|
|
|
uint32_t InfoStream::getSignature() const { return Signature; }
|
|
|
|
uint32_t InfoStream::getAge() const { return Age; }
|
|
|
|
PDB_UniqueId InfoStream::getGuid() const { return Guid; }
|
|
|
|
Error InfoStream::commit() {
|
|
StreamWriter Writer(*Stream);
|
|
|
|
HeaderInfo H;
|
|
H.Age = Age;
|
|
H.Signature = Signature;
|
|
H.Version = Version;
|
|
H.Guid = Guid;
|
|
if (auto EC = Writer.writeObject(H))
|
|
return EC;
|
|
|
|
return NamedStreams.commit(Writer);
|
|
}
|