b927e02e1b
Block 1 and 2 of an MSF file are bit vectors that represent the list of blocks allocated and free in the file. We had been using these blocks to write stream data and other data, so we mark them as the free page map now. We don't yet serialize these pages to the disk, but at least we make a note of what it is, and avoid writing random data to them. Doing this also necessitated cleaning up some of the tests to be more general and hardcode fewer values, which is nice. llvm-svn: 275629
103 lines
3.0 KiB
C++
103 lines
3.0 KiB
C++
//===- PDBFileBuilder.cpp - PDB File Creation -------------------*- 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/PDBFileBuilder.h"
|
|
|
|
#include "llvm/ADT/BitVector.h"
|
|
|
|
#include "llvm/DebugInfo/CodeView/StreamInterface.h"
|
|
#include "llvm/DebugInfo/CodeView/StreamWriter.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/DbiStream.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/DbiStreamBuilder.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/InfoStream.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/InfoStreamBuilder.h"
|
|
#include "llvm/DebugInfo/PDB/Raw/RawError.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::codeview;
|
|
using namespace llvm::pdb;
|
|
using namespace llvm::support;
|
|
|
|
PDBFileBuilder::PDBFileBuilder(
|
|
std::unique_ptr<codeview::StreamInterface> FileBuffer)
|
|
: File(llvm::make_unique<PDBFile>(std::move(FileBuffer))) {}
|
|
|
|
Error PDBFileBuilder::initialize(const msf::SuperBlock &Super) {
|
|
auto ExpectedMsf =
|
|
MsfBuilder::create(File->Allocator, Super.BlockSize, Super.NumBlocks);
|
|
if (!ExpectedMsf)
|
|
return ExpectedMsf.takeError();
|
|
|
|
auto &MsfResult = *ExpectedMsf;
|
|
if (auto EC = MsfResult.setBlockMapAddr(Super.BlockMapAddr))
|
|
return EC;
|
|
Msf = llvm::make_unique<MsfBuilder>(std::move(MsfResult));
|
|
Msf->setFreePageMap(Super.FreeBlockMapBlock);
|
|
Msf->setUnknown1(Super.Unknown1);
|
|
return Error::success();
|
|
}
|
|
|
|
MsfBuilder &PDBFileBuilder::getMsfBuilder() { return *Msf; }
|
|
|
|
InfoStreamBuilder &PDBFileBuilder::getInfoBuilder() {
|
|
if (!Info)
|
|
Info = llvm::make_unique<InfoStreamBuilder>();
|
|
return *Info;
|
|
}
|
|
|
|
DbiStreamBuilder &PDBFileBuilder::getDbiBuilder() {
|
|
if (!Dbi)
|
|
Dbi = llvm::make_unique<DbiStreamBuilder>();
|
|
return *Dbi;
|
|
}
|
|
|
|
Expected<std::unique_ptr<PDBFile>> PDBFileBuilder::build() {
|
|
if (Info) {
|
|
uint32_t Length = Info->calculateSerializedLength();
|
|
if (auto EC = Msf->setStreamSize(StreamPDB, Length))
|
|
return std::move(EC);
|
|
}
|
|
if (Dbi) {
|
|
uint32_t Length = Dbi->calculateSerializedLength();
|
|
if (auto EC = Msf->setStreamSize(StreamDBI, Length))
|
|
return std::move(EC);
|
|
}
|
|
|
|
auto ExpectedLayout = Msf->build();
|
|
if (!ExpectedLayout)
|
|
return ExpectedLayout.takeError();
|
|
|
|
const msf::Layout &L = *ExpectedLayout;
|
|
File->StreamMap = L.StreamMap;
|
|
File->StreamSizes = L.StreamSizes;
|
|
File->DirectoryBlocks = L.DirectoryBlocks;
|
|
File->SB = L.SB;
|
|
|
|
if (Info) {
|
|
auto ExpectedInfo = Info->build(*File);
|
|
if (!ExpectedInfo)
|
|
return ExpectedInfo.takeError();
|
|
File->Info = std::move(*ExpectedInfo);
|
|
}
|
|
|
|
if (Dbi) {
|
|
auto ExpectedDbi = Dbi->build(*File);
|
|
if (!ExpectedDbi)
|
|
return ExpectedDbi.takeError();
|
|
File->Dbi = std::move(*ExpectedDbi);
|
|
}
|
|
|
|
if (File->Info && File->Dbi && File->Info->getAge() != File->Dbi->getAge())
|
|
return llvm::make_error<RawError>(
|
|
raw_error_code::corrupt_file,
|
|
"PDB Stream Age doesn't match Dbi Stream Age!");
|
|
|
|
return std::move(File);
|
|
}
|