summaryrefslogtreecommitdiff
path: root/unittests/DebugInfo/PDB
diff options
context:
space:
mode:
authorDimitry Andric <dim@FreeBSD.org>2017-12-18 20:10:56 +0000
committerDimitry Andric <dim@FreeBSD.org>2017-12-18 20:10:56 +0000
commit044eb2f6afba375a914ac9d8024f8f5142bb912e (patch)
tree1475247dc9f9fe5be155ebd4c9069c75aadf8c20 /unittests/DebugInfo/PDB
parenteb70dddbd77e120e5d490bd8fbe7ff3f8fa81c6b (diff)
Notes
Diffstat (limited to 'unittests/DebugInfo/PDB')
-rw-r--r--unittests/DebugInfo/PDB/CMakeLists.txt4
-rw-r--r--unittests/DebugInfo/PDB/MSFBuilderTest.cpp361
-rw-r--r--unittests/DebugInfo/PDB/MappedBlockStreamTest.cpp500
-rw-r--r--unittests/DebugInfo/PDB/PDBApiTest.cpp9
4 files changed, 10 insertions, 864 deletions
diff --git a/unittests/DebugInfo/PDB/CMakeLists.txt b/unittests/DebugInfo/PDB/CMakeLists.txt
index 583b065f464c..b19ee2cf43a5 100644
--- a/unittests/DebugInfo/PDB/CMakeLists.txt
+++ b/unittests/DebugInfo/PDB/CMakeLists.txt
@@ -6,9 +6,7 @@ set(LLVM_LINK_COMPONENTS
set(DebugInfoPDBSources
HashTableTest.cpp
- MappedBlockStreamTest.cpp
StringTableBuilderTest.cpp
- MSFBuilderTest.cpp
PDBApiTest.cpp
)
@@ -16,4 +14,4 @@ add_llvm_unittest(DebugInfoPDBTests
${DebugInfoPDBSources}
)
-target_link_libraries(DebugInfoPDBTests LLVMTestingSupport)
+target_link_libraries(DebugInfoPDBTests PRIVATE LLVMTestingSupport)
diff --git a/unittests/DebugInfo/PDB/MSFBuilderTest.cpp b/unittests/DebugInfo/PDB/MSFBuilderTest.cpp
deleted file mode 100644
index 23a15d14f756..000000000000
--- a/unittests/DebugInfo/PDB/MSFBuilderTest.cpp
+++ /dev/null
@@ -1,361 +0,0 @@
-//===- MSFBuilderTest.cpp Tests manipulation of MSF stream metadata ------===//
-//
-// The LLVM Compiler Infrastructure
-//
-// This file is distributed under the University of Illinois Open Source
-// License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
-
-#include "llvm/DebugInfo/MSF/MSFBuilder.h"
-#include "llvm/DebugInfo/MSF/MSFCommon.h"
-#include "llvm/Testing/Support/Error.h"
-
-#include "gtest/gtest.h"
-
-using namespace llvm;
-using namespace llvm::msf;
-
-namespace {
-class MSFBuilderTest : public testing::Test {
-protected:
- void initializeSimpleSuperBlock(msf::SuperBlock &SB) {
- initializeSuperBlock(SB);
- SB.NumBlocks = 1000;
- SB.NumDirectoryBytes = 8192;
- }
-
- void initializeSuperBlock(msf::SuperBlock &SB) {
- ::memset(&SB, 0, sizeof(SB));
-
- ::memcpy(SB.MagicBytes, msf::Magic, sizeof(msf::Magic));
- SB.FreeBlockMapBlock = 1;
- SB.BlockMapAddr = 1;
- SB.BlockSize = 4096;
- SB.NumDirectoryBytes = 0;
- SB.NumBlocks = 2; // one for the Super Block, one for the directory
- }
-
- BumpPtrAllocator Allocator;
-};
-}
-
-TEST_F(MSFBuilderTest, ValidateSuperBlockAccept) {
- // Test that a known good super block passes validation.
- SuperBlock SB;
- initializeSuperBlock(SB);
-
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Succeeded());
-}
-
-TEST_F(MSFBuilderTest, ValidateSuperBlockReject) {
- // Test that various known problems cause a super block to be rejected.
- SuperBlock SB;
- initializeSimpleSuperBlock(SB);
-
- // Mismatched magic
- SB.MagicBytes[0] = 8;
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Failed());
- initializeSimpleSuperBlock(SB);
-
- // Block 0 is reserved for super block, can't be occupied by the block map
- SB.BlockMapAddr = 0;
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Failed());
- initializeSimpleSuperBlock(SB);
-
- // Block sizes have to be powers of 2.
- SB.BlockSize = 3120;
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Failed());
- initializeSimpleSuperBlock(SB);
-
- // The directory itself has a maximum size.
- SB.NumDirectoryBytes = SB.BlockSize * SB.BlockSize / 4;
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Succeeded());
- SB.NumDirectoryBytes = SB.NumDirectoryBytes + 4;
- EXPECT_THAT_ERROR(msf::validateSuperBlock(SB), Failed());
-}
-
-TEST_F(MSFBuilderTest, TestUsedBlocksMarkedAsUsed) {
- // Test that when assigning a stream to a known list of blocks, the blocks
- // are correctly marked as used after adding, but no other incorrect blocks
- // are accidentally marked as used.
-
- std::vector<uint32_t> Blocks = {4, 5, 6, 7, 8, 9, 10, 11, 12};
- // Allocate some extra blocks at the end so we can verify that they're free
- // after the initialization.
- uint32_t NumBlocks = msf::getMinimumBlockCount() + Blocks.size() + 10;
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096, NumBlocks);
- ASSERT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(Blocks.size() * 4096, Blocks),
- Succeeded());
-
- for (auto B : Blocks) {
- EXPECT_FALSE(Msf.isBlockFree(B));
- }
-
- uint32_t FreeBlockStart = Blocks.back() + 1;
- for (uint32_t I = FreeBlockStart; I < NumBlocks; ++I) {
- EXPECT_TRUE(Msf.isBlockFree(I));
- }
-}
-
-TEST_F(MSFBuilderTest, TestAddStreamNoDirectoryBlockIncrease) {
- // Test that adding a new stream correctly updates the directory. This only
- // tests the case where the directory *DOES NOT* grow large enough that it
- // crosses a Block boundary.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- auto ExpectedL1 = Msf.build();
- EXPECT_THAT_EXPECTED(ExpectedL1, Succeeded());
- MSFLayout &L1 = *ExpectedL1;
-
- auto OldDirBlocks = L1.DirectoryBlocks;
- EXPECT_EQ(1U, OldDirBlocks.size());
-
- auto ExpectedMsf2 = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf2, Succeeded());
- auto &Msf2 = *ExpectedMsf2;
-
- EXPECT_THAT_EXPECTED(Msf2.addStream(4000), Succeeded());
- EXPECT_EQ(1U, Msf2.getNumStreams());
- EXPECT_EQ(4000U, Msf2.getStreamSize(0));
- auto Blocks = Msf2.getStreamBlocks(0);
- EXPECT_EQ(1U, Blocks.size());
-
- auto ExpectedL2 = Msf2.build();
- EXPECT_THAT_EXPECTED(ExpectedL2, Succeeded());
- MSFLayout &L2 = *ExpectedL2;
- auto NewDirBlocks = L2.DirectoryBlocks;
- EXPECT_EQ(1U, NewDirBlocks.size());
-}
-
-TEST_F(MSFBuilderTest, TestAddStreamWithDirectoryBlockIncrease) {
- // Test that adding a new stream correctly updates the directory. This only
- // tests the case where the directory *DOES* grow large enough that it
- // crosses a Block boundary. This is because the newly added stream occupies
- // so many Blocks that need to be indexed in the directory that the directory
- // crosses a Block boundary.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(4096 * 4096 / sizeof(uint32_t)),
- Succeeded());
-
- auto ExpectedL1 = Msf.build();
- EXPECT_THAT_EXPECTED(ExpectedL1, Succeeded());
- MSFLayout &L1 = *ExpectedL1;
- auto DirBlocks = L1.DirectoryBlocks;
- EXPECT_EQ(2U, DirBlocks.size());
-}
-
-TEST_F(MSFBuilderTest, TestGrowStreamNoBlockIncrease) {
- // Test growing an existing stream by a value that does not affect the number
- // of blocks it occupies.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(1024), Succeeded());
- EXPECT_EQ(1024U, Msf.getStreamSize(0));
- auto OldStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, OldStreamBlocks.size());
-
- EXPECT_THAT_ERROR(Msf.setStreamSize(0, 2048), Succeeded());
- EXPECT_EQ(2048U, Msf.getStreamSize(0));
- auto NewStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, NewStreamBlocks.size());
-
- EXPECT_EQ(OldStreamBlocks, NewStreamBlocks);
-}
-
-TEST_F(MSFBuilderTest, TestGrowStreamWithBlockIncrease) {
- // Test that growing an existing stream to a value large enough that it causes
- // the need to allocate new Blocks to the stream correctly updates the
- // stream's
- // block list.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(2048), Succeeded());
- EXPECT_EQ(2048U, Msf.getStreamSize(0));
- std::vector<uint32_t> OldStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, OldStreamBlocks.size());
-
- EXPECT_THAT_ERROR(Msf.setStreamSize(0, 6144), Succeeded());
- EXPECT_EQ(6144U, Msf.getStreamSize(0));
- std::vector<uint32_t> NewStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(2U, NewStreamBlocks.size());
-
- EXPECT_EQ(OldStreamBlocks[0], NewStreamBlocks[0]);
- EXPECT_NE(NewStreamBlocks[0], NewStreamBlocks[1]);
-}
-
-TEST_F(MSFBuilderTest, TestShrinkStreamNoBlockDecrease) {
- // Test that shrinking an existing stream by a value that does not affect the
- // number of Blocks it occupies makes no changes to stream's block list.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(2048), Succeeded());
- EXPECT_EQ(2048U, Msf.getStreamSize(0));
- std::vector<uint32_t> OldStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, OldStreamBlocks.size());
-
- EXPECT_THAT_ERROR(Msf.setStreamSize(0, 1024), Succeeded());
- EXPECT_EQ(1024U, Msf.getStreamSize(0));
- std::vector<uint32_t> NewStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, NewStreamBlocks.size());
-
- EXPECT_EQ(OldStreamBlocks, NewStreamBlocks);
-}
-
-TEST_F(MSFBuilderTest, TestShrinkStreamWithBlockDecrease) {
- // Test that shrinking an existing stream to a value large enough that it
- // causes the need to deallocate new Blocks to the stream correctly updates
- // the stream's block list.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(6144), Succeeded());
- EXPECT_EQ(6144U, Msf.getStreamSize(0));
- std::vector<uint32_t> OldStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(2U, OldStreamBlocks.size());
-
- EXPECT_THAT_ERROR(Msf.setStreamSize(0, 2048), Succeeded());
- EXPECT_EQ(2048U, Msf.getStreamSize(0));
- std::vector<uint32_t> NewStreamBlocks = Msf.getStreamBlocks(0);
- EXPECT_EQ(1U, NewStreamBlocks.size());
-
- EXPECT_EQ(OldStreamBlocks[0], NewStreamBlocks[0]);
-}
-
-TEST_F(MSFBuilderTest, TestRejectReusedStreamBlock) {
- // Test that attempting to add a stream and assigning a block that is already
- // in use by another stream fails.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- EXPECT_THAT_EXPECTED(Msf.addStream(6144), Succeeded());
-
- std::vector<uint32_t> Blocks = {2, 3};
- EXPECT_THAT_EXPECTED(Msf.addStream(6144, Blocks), Failed());
-}
-
-TEST_F(MSFBuilderTest, TestBlockCountsWhenAddingStreams) {
- // Test that when adding multiple streams, the number of used and free Blocks
- // allocated to the MSF file are as expected.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- // one for the super block, one for the directory block map
- uint32_t NumUsedBlocks = Msf.getNumUsedBlocks();
- EXPECT_EQ(msf::getMinimumBlockCount(), NumUsedBlocks);
- EXPECT_EQ(0U, Msf.getNumFreeBlocks());
-
- const uint32_t StreamSizes[] = {4000, 6193, 189723};
- for (int I = 0; I < 3; ++I) {
- EXPECT_THAT_EXPECTED(Msf.addStream(StreamSizes[I]), Succeeded());
- NumUsedBlocks += bytesToBlocks(StreamSizes[I], 4096);
- EXPECT_EQ(NumUsedBlocks, Msf.getNumUsedBlocks());
- EXPECT_EQ(0U, Msf.getNumFreeBlocks());
- }
-}
-
-TEST_F(MSFBuilderTest, BuildMsfLayout) {
- // Test that we can generate an MSFLayout structure from a valid layout
- // specification.
- auto ExpectedMsf = MSFBuilder::create(Allocator, 4096);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- const uint32_t StreamSizes[] = {4000, 6193, 189723};
- uint32_t ExpectedNumBlocks = msf::getMinimumBlockCount();
- for (int I = 0; I < 3; ++I) {
- EXPECT_THAT_EXPECTED(Msf.addStream(StreamSizes[I]), Succeeded());
- ExpectedNumBlocks += bytesToBlocks(StreamSizes[I], 4096);
- }
- ++ExpectedNumBlocks; // The directory itself should use 1 block
-
- auto ExpectedLayout = Msf.build();
- EXPECT_THAT_EXPECTED(ExpectedLayout, Succeeded());
- MSFLayout &L = *ExpectedLayout;
- EXPECT_EQ(4096U, L.SB->BlockSize);
- EXPECT_EQ(ExpectedNumBlocks, L.SB->NumBlocks);
-
- EXPECT_EQ(1U, L.DirectoryBlocks.size());
-
- EXPECT_EQ(3U, L.StreamMap.size());
- EXPECT_EQ(3U, L.StreamSizes.size());
- for (int I = 0; I < 3; ++I) {
- EXPECT_EQ(StreamSizes[I], L.StreamSizes[I]);
- uint32_t ExpectedNumBlocks = bytesToBlocks(StreamSizes[I], 4096);
- EXPECT_EQ(ExpectedNumBlocks, L.StreamMap[I].size());
- }
-}
-
-TEST_F(MSFBuilderTest, UseDirectoryBlockHint) {
- Expected<MSFBuilder> ExpectedMsf = MSFBuilder::create(
- Allocator, 4096, msf::getMinimumBlockCount() + 1, false);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- uint32_t B = msf::getFirstUnreservedBlock();
- EXPECT_THAT_ERROR(Msf.setDirectoryBlocksHint({B + 1}), Succeeded());
- EXPECT_THAT_EXPECTED(Msf.addStream(2048, {B + 2}), Succeeded());
-
- auto ExpectedLayout = Msf.build();
- EXPECT_THAT_EXPECTED(ExpectedLayout, Succeeded());
- MSFLayout &L = *ExpectedLayout;
- EXPECT_EQ(msf::getMinimumBlockCount() + 2, L.SB->NumBlocks);
- EXPECT_EQ(1U, L.DirectoryBlocks.size());
- EXPECT_EQ(1U, L.StreamMap[0].size());
-
- EXPECT_EQ(B + 1, L.DirectoryBlocks[0]);
- EXPECT_EQ(B + 2, L.StreamMap[0].front());
-}
-
-TEST_F(MSFBuilderTest, DirectoryBlockHintInsufficient) {
- Expected<MSFBuilder> ExpectedMsf =
- MSFBuilder::create(Allocator, 4096, msf::getMinimumBlockCount() + 2);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
- uint32_t B = msf::getFirstUnreservedBlock();
- EXPECT_THAT_ERROR(Msf.setDirectoryBlocksHint({B + 1}), Succeeded());
-
- uint32_t Size = 4096 * 4096 / 4;
- EXPECT_THAT_EXPECTED(Msf.addStream(Size), Succeeded());
-
- auto ExpectedLayout = Msf.build();
- EXPECT_THAT_EXPECTED(ExpectedLayout, Succeeded());
- MSFLayout &L = *ExpectedLayout;
- EXPECT_EQ(2U, L.DirectoryBlocks.size());
- EXPECT_EQ(B + 1, L.DirectoryBlocks[0]);
-}
-
-TEST_F(MSFBuilderTest, DirectoryBlockHintOverestimated) {
- Expected<MSFBuilder> ExpectedMsf =
- MSFBuilder::create(Allocator, 4096, msf::getMinimumBlockCount() + 2);
- EXPECT_THAT_EXPECTED(ExpectedMsf, Succeeded());
- auto &Msf = *ExpectedMsf;
-
- uint32_t B = msf::getFirstUnreservedBlock();
- EXPECT_THAT_ERROR(Msf.setDirectoryBlocksHint({B + 1, B + 2}), Succeeded());
-
- ASSERT_THAT_EXPECTED(Msf.addStream(2048), Succeeded());
-
- auto ExpectedLayout = Msf.build();
- ASSERT_THAT_EXPECTED(ExpectedLayout, Succeeded());
- MSFLayout &L = *ExpectedLayout;
- EXPECT_EQ(1U, L.DirectoryBlocks.size());
- EXPECT_EQ(B + 1, L.DirectoryBlocks[0]);
-}
diff --git a/unittests/DebugInfo/PDB/MappedBlockStreamTest.cpp b/unittests/DebugInfo/PDB/MappedBlockStreamTest.cpp
deleted file mode 100644
index a9a1ee4d65b9..000000000000
--- a/unittests/DebugInfo/PDB/MappedBlockStreamTest.cpp
+++ /dev/null
@@ -1,500 +0,0 @@
-//===- llvm/unittest/DebugInfo/PDB/MappedBlockStreamTest.cpp --------------===//
-//
-// The LLVM Compiler Infrastructure
-//
-// This file is distributed under the University of Illinois Open Source
-// License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
-
-#include "llvm/DebugInfo/MSF/MappedBlockStream.h"
-#include "llvm/DebugInfo/MSF/IMSFFile.h"
-#include "llvm/DebugInfo/MSF/MSFError.h"
-#include "llvm/DebugInfo/MSF/MSFStreamLayout.h"
-#include "llvm/Support/BinaryByteStream.h"
-#include "llvm/Support/BinaryStreamReader.h"
-#include "llvm/Support/BinaryStreamRef.h"
-#include "llvm/Support/BinaryStreamWriter.h"
-#include "llvm/Testing/Support/Error.h"
-
-#include "gtest/gtest.h"
-
-#include <unordered_map>
-
-using namespace llvm;
-using namespace llvm::msf;
-using namespace llvm::support;
-
-namespace {
-
-static const uint32_t BlocksAry[] = {0, 1, 2, 5, 4, 3, 6, 7, 8, 9};
-static uint8_t DataAry[] = {'A', 'B', 'C', 'F', 'E', 'D', 'G', 'H', 'I', 'J'};
-
-class DiscontiguousStream : public WritableBinaryStream {
-public:
- DiscontiguousStream(ArrayRef<uint32_t> Blocks, MutableArrayRef<uint8_t> Data)
- : Blocks(Blocks.begin(), Blocks.end()), Data(Data.begin(), Data.end()) {}
-
- uint32_t block_size() const { return 1; }
- uint32_t block_count() const { return Blocks.size(); }
-
- endianness getEndian() const override { return little; }
-
- Error readBytes(uint32_t Offset, uint32_t Size,
- ArrayRef<uint8_t> &Buffer) override {
- if (auto EC = checkOffset(Offset, Size))
- return EC;
- Buffer = Data.slice(Offset, Size);
- return Error::success();
- }
-
- Error readLongestContiguousChunk(uint32_t Offset,
- ArrayRef<uint8_t> &Buffer) override {
- if (auto EC = checkOffset(Offset, 1))
- return EC;
- Buffer = Data.drop_front(Offset);
- return Error::success();
- }
-
- uint32_t getLength() override { return Data.size(); }
-
- Error writeBytes(uint32_t Offset, ArrayRef<uint8_t> SrcData) override {
- if (auto EC = checkOffset(Offset, SrcData.size()))
- return EC;
- ::memcpy(&Data[Offset], SrcData.data(), SrcData.size());
- return Error::success();
- }
- Error commit() override { return Error::success(); }
-
- MSFStreamLayout layout() const {
- return MSFStreamLayout{static_cast<uint32_t>(Data.size()), Blocks};
- }
-
- BumpPtrAllocator Allocator;
-
-private:
- std::vector<support::ulittle32_t> Blocks;
- MutableArrayRef<uint8_t> Data;
-};
-
-TEST(MappedBlockStreamTest, NumBlocks) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- EXPECT_EQ(F.block_size(), S->getBlockSize());
- EXPECT_EQ(F.layout().Blocks.size(), S->getNumBlocks());
-
-}
-
-// Tests that a read which is entirely contained within a single block works
-// and does not allocate.
-TEST(MappedBlockStreamTest, ReadBeyondEndOfStreamRef) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
-
- BinaryStreamReader R(*S);
- BinaryStreamRef SR;
- EXPECT_THAT_ERROR(R.readStreamRef(SR, 0U), Succeeded());
- ArrayRef<uint8_t> Buffer;
- EXPECT_THAT_ERROR(SR.readBytes(0U, 1U, Buffer), Failed());
- EXPECT_THAT_ERROR(R.readStreamRef(SR, 1U), Succeeded());
- EXPECT_THAT_ERROR(SR.readBytes(1U, 1U, Buffer), Failed());
-}
-
-// Tests that a read which outputs into a full destination buffer works and
-// does not fail due to the length of the output buffer.
-TEST(MappedBlockStreamTest, ReadOntoNonEmptyBuffer) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
-
- BinaryStreamReader R(*S);
- StringRef Str = "ZYXWVUTSRQPONMLKJIHGFEDCBA";
- EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Succeeded());
- EXPECT_EQ(Str, StringRef("A"));
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Tests that a read which crosses a block boundary, but where the subsequent
-// blocks are still contiguous in memory to the previous block works and does
-// not allocate memory.
-TEST(MappedBlockStreamTest, ZeroCopyReadContiguousBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
- EXPECT_THAT_ERROR(R.readFixedString(Str, 2), Succeeded());
- EXPECT_EQ(Str, StringRef("AB"));
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-
- R.setOffset(6);
- EXPECT_THAT_ERROR(R.readFixedString(Str, 4), Succeeded());
- EXPECT_EQ(Str, StringRef("GHIJ"));
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Tests that a read which crosses a block boundary and cannot be referenced
-// contiguously works and allocates only the precise amount of bytes
-// requested.
-TEST(MappedBlockStreamTest, CopyReadNonContiguousBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
- EXPECT_THAT_ERROR(R.readFixedString(Str, 10), Succeeded());
- EXPECT_EQ(Str, StringRef("ABCDEFGHIJ"));
- EXPECT_EQ(10U, F.Allocator.getBytesAllocated());
-}
-
-// Test that an out of bounds read which doesn't cross a block boundary
-// fails and allocates no memory.
-TEST(MappedBlockStreamTest, InvalidReadSizeNoBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
-
- R.setOffset(10);
- EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Failed());
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Test that an out of bounds read which crosses a contiguous block boundary
-// fails and allocates no memory.
-TEST(MappedBlockStreamTest, InvalidReadSizeContiguousBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
-
- R.setOffset(6);
- EXPECT_THAT_ERROR(R.readFixedString(Str, 5), Failed());
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Test that an out of bounds read which crosses a discontiguous block
-// boundary fails and allocates no memory.
-TEST(MappedBlockStreamTest, InvalidReadSizeNonContiguousBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
-
- EXPECT_THAT_ERROR(R.readFixedString(Str, 11), Failed());
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Tests that a read which is entirely contained within a single block but
-// beyond the end of a StreamRef fails.
-TEST(MappedBlockStreamTest, ZeroCopyReadNoBreak) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str;
- EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Succeeded());
- EXPECT_EQ(Str, StringRef("A"));
- EXPECT_EQ(0U, F.Allocator.getBytesAllocated());
-}
-
-// Tests that a read which is not aligned on the same boundary as a previous
-// cached request, but which is known to overlap that request, shares the
-// previous allocation.
-TEST(MappedBlockStreamTest, UnalignedOverlappingRead) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str1;
- StringRef Str2;
- EXPECT_THAT_ERROR(R.readFixedString(Str1, 7), Succeeded());
- EXPECT_EQ(Str1, StringRef("ABCDEFG"));
- EXPECT_EQ(7U, F.Allocator.getBytesAllocated());
-
- R.setOffset(2);
- EXPECT_THAT_ERROR(R.readFixedString(Str2, 3), Succeeded());
- EXPECT_EQ(Str2, StringRef("CDE"));
- EXPECT_EQ(Str1.data() + 2, Str2.data());
- EXPECT_EQ(7U, F.Allocator.getBytesAllocated());
-}
-
-// Tests that a read which is not aligned on the same boundary as a previous
-// cached request, but which only partially overlaps a previous cached request,
-// still works correctly and allocates again from the shared pool.
-TEST(MappedBlockStreamTest, UnalignedOverlappingReadFail) {
- DiscontiguousStream F(BlocksAry, DataAry);
- auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F,
- F.Allocator);
- BinaryStreamReader R(*S);
- StringRef Str1;
- StringRef Str2;
- EXPECT_THAT_ERROR(R.readFixedString(Str1, 6), Succeeded());
- EXPECT_EQ(Str1, StringRef("ABCDEF"));
- EXPECT_EQ(6U, F.Allocator.getBytesAllocated());
-
- R.setOffset(4);
- EXPECT_THAT_ERROR(R.readFixedString(Str2, 4), Succeeded());
- EXPECT_EQ(Str2, StringRef("EFGH"));
- EXPECT_EQ(10U, F.Allocator.getBytesAllocated());
-}
-
-TEST(MappedBlockStreamTest, WriteBeyondEndOfStream) {
- static uint8_t Data[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J'};
- static uint8_t LargeBuffer[] = {'0', '1', '2', '3', '4', '5',
- '6', '7', '8', '9', 'A'};
- static uint8_t SmallBuffer[] = {'0', '1', '2'};
- static_assert(sizeof(LargeBuffer) > sizeof(Data),
- "LargeBuffer is not big enough");
-
- DiscontiguousStream F(BlocksAry, Data);
- auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(),
- F, F.Allocator);
- ArrayRef<uint8_t> Buffer;
-
- EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>(LargeBuffer)), Failed());
- EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>(SmallBuffer)),
- Succeeded());
- EXPECT_THAT_ERROR(S->writeBytes(7, ArrayRef<uint8_t>(SmallBuffer)),
- Succeeded());
- EXPECT_THAT_ERROR(S->writeBytes(8, ArrayRef<uint8_t>(SmallBuffer)), Failed());
-}
-
-TEST(MappedBlockStreamTest, TestWriteBytesNoBreakBoundary) {
- static uint8_t Data[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J'};
- DiscontiguousStream F(BlocksAry, Data);
- auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(),
- F, F.Allocator);
- ArrayRef<uint8_t> Buffer;
-
- EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A'));
- EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J'));
-
- EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>('J')), Succeeded());
- EXPECT_THAT_ERROR(S->writeBytes(9, ArrayRef<uint8_t>('A')), Succeeded());
-
- EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J'));
- EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A'));
-
- EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>('A')), Succeeded());
- EXPECT_THAT_ERROR(S->writeBytes(9, ArrayRef<uint8_t>('J')), Succeeded());
-
- EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A'));
- EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J'));
-}
-
-TEST(MappedBlockStreamTest, TestWriteBytesBreakBoundary) {
- static uint8_t Data[] = {'0', '0', '0', '0', '0', '0', '0', '0', '0', '0'};
- static uint8_t TestData[] = {'T', 'E', 'S', 'T', 'I', 'N', 'G', '.'};
- static uint8_t Expected[] = {'T', 'E', 'S', 'N', 'I',
- 'T', 'G', '.', '0', '0'};
-
- DiscontiguousStream F(BlocksAry, Data);
- auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(),
- F, F.Allocator);
- ArrayRef<uint8_t> Buffer;
-
- EXPECT_THAT_ERROR(S->writeBytes(0, TestData), Succeeded());
- // First just compare the memory, then compare the result of reading the
- // string out.
- EXPECT_EQ(ArrayRef<uint8_t>(Data), ArrayRef<uint8_t>(Expected));
-
- EXPECT_THAT_ERROR(S->readBytes(0, 8, Buffer), Succeeded());
- EXPECT_EQ(Buffer, ArrayRef<uint8_t>(TestData));
-}
-
-TEST(MappedBlockStreamTest, TestWriteThenRead) {
- std::vector<uint8_t> DataBytes(10);
- MutableArrayRef<uint8_t> Data(DataBytes);
- const uint32_t Blocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8};
-
- DiscontiguousStream F(Blocks, Data);
- auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(),
- F, F.Allocator);
-
- enum class MyEnum : uint32_t { Val1 = 2908234, Val2 = 120891234 };
- using support::ulittle32_t;
-
- uint16_t u16[] = {31468, 0};
- uint32_t u32[] = {890723408, 0};
- MyEnum Enum[] = {MyEnum::Val1, MyEnum::Val2};
- StringRef ZStr[] = {"Zero Str", ""};
- StringRef FStr[] = {"Fixed Str", ""};
- uint8_t byteArray0[] = {'1', '2'};
- uint8_t byteArray1[] = {'0', '0'};
- ArrayRef<uint8_t> byteArrayRef0(byteArray0);
- ArrayRef<uint8_t> byteArrayRef1(byteArray1);
- ArrayRef<uint8_t> byteArray[] = { byteArrayRef0, byteArrayRef1 };
- uint32_t intArr0[] = {890723408, 29082234};
- uint32_t intArr1[] = {890723408, 29082234};
- ArrayRef<uint32_t> intArray[] = {intArr0, intArr1};
-
- BinaryStreamReader Reader(*S);
- BinaryStreamWriter Writer(*S);
- EXPECT_THAT_ERROR(Writer.writeInteger(u16[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readInteger(u16[1]), Succeeded());
- EXPECT_EQ(u16[0], u16[1]);
- EXPECT_EQ(std::vector<uint8_t>({0, 0x7A, 0xEC, 0, 0, 0, 0, 0, 0, 0}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeInteger(u32[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readInteger(u32[1]), Succeeded());
- EXPECT_EQ(u32[0], u32[1]);
- EXPECT_EQ(std::vector<uint8_t>({0x17, 0x5C, 0x50, 0, 0, 0, 0x35, 0, 0, 0}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeEnum(Enum[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readEnum(Enum[1]), Succeeded());
- EXPECT_EQ(Enum[0], Enum[1]);
- EXPECT_EQ(std::vector<uint8_t>({0x2C, 0x60, 0x4A, 0, 0, 0, 0, 0, 0, 0}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeCString(ZStr[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readCString(ZStr[1]), Succeeded());
- EXPECT_EQ(ZStr[0], ZStr[1]);
- EXPECT_EQ(
- std::vector<uint8_t>({'r', 'e', 'Z', ' ', 'S', 't', 'o', 'r', 0, 0}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeFixedString(FStr[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readFixedString(FStr[1], FStr[0].size()),
- Succeeded());
- EXPECT_EQ(FStr[0], FStr[1]);
- EXPECT_EQ(
- std::vector<uint8_t>({'x', 'i', 'F', 'd', ' ', 'S', 'e', 't', 0, 'r'}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeArray(byteArray[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readArray(byteArray[1], byteArray[0].size()),
- Succeeded());
- EXPECT_EQ(byteArray[0], byteArray[1]);
- EXPECT_EQ(std::vector<uint8_t>({0, 0x32, 0x31, 0, 0, 0, 0, 0, 0, 0}),
- DataBytes);
-
- Reader.setOffset(0);
- Writer.setOffset(0);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeArray(intArray[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readArray(intArray[1], intArray[0].size()),
- Succeeded());
- EXPECT_EQ(intArray[0], intArray[1]);
-}
-
-TEST(MappedBlockStreamTest, TestWriteContiguousStreamRef) {
- std::vector<uint8_t> DestDataBytes(10);
- MutableArrayRef<uint8_t> DestData(DestDataBytes);
- const uint32_t DestBlocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8};
-
- std::vector<uint8_t> SrcDataBytes(10);
- MutableArrayRef<uint8_t> SrcData(SrcDataBytes);
-
- DiscontiguousStream F(DestBlocks, DestData);
- auto DestStream = WritableMappedBlockStream::createStream(
- F.block_size(), F.layout(), F, F.Allocator);
-
- // First write "Test Str" into the source stream.
- MutableBinaryByteStream SourceStream(SrcData, little);
- BinaryStreamWriter SourceWriter(SourceStream);
- EXPECT_THAT_ERROR(SourceWriter.writeCString("Test Str"), Succeeded());
- EXPECT_EQ(SrcDataBytes, std::vector<uint8_t>(
- {'T', 'e', 's', 't', ' ', 'S', 't', 'r', 0, 0}));
-
- // Then write the source stream into the dest stream.
- BinaryStreamWriter DestWriter(*DestStream);
- EXPECT_THAT_ERROR(DestWriter.writeStreamRef(SourceStream), Succeeded());
- EXPECT_EQ(DestDataBytes, std::vector<uint8_t>(
- {'s', 'e', 'T', ' ', 'S', 't', 't', 'r', 0, 0}));
-
- // Then read the string back out of the dest stream.
- StringRef Result;
- BinaryStreamReader DestReader(*DestStream);
- EXPECT_THAT_ERROR(DestReader.readCString(Result), Succeeded());
- EXPECT_EQ(Result, "Test Str");
-}
-
-TEST(MappedBlockStreamTest, TestWriteDiscontiguousStreamRef) {
- std::vector<uint8_t> DestDataBytes(10);
- MutableArrayRef<uint8_t> DestData(DestDataBytes);
- const uint32_t DestBlocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8};
-
- std::vector<uint8_t> SrcDataBytes(10);
- MutableArrayRef<uint8_t> SrcData(SrcDataBytes);
- const uint32_t SrcBlocks[] = {1, 0, 6, 3, 4, 5, 2, 7, 8, 9};
-
- DiscontiguousStream DestF(DestBlocks, DestData);
- DiscontiguousStream SrcF(SrcBlocks, SrcData);
-
- auto Dest = WritableMappedBlockStream::createStream(
- DestF.block_size(), DestF.layout(), DestF, DestF.Allocator);
- auto Src = WritableMappedBlockStream::createStream(
- SrcF.block_size(), SrcF.layout(), SrcF, SrcF.Allocator);
-
- // First write "Test Str" into the source stream.
- BinaryStreamWriter SourceWriter(*Src);
- EXPECT_THAT_ERROR(SourceWriter.writeCString("Test Str"), Succeeded());
- EXPECT_EQ(SrcDataBytes, std::vector<uint8_t>(
- {'e', 'T', 't', 't', ' ', 'S', 's', 'r', 0, 0}));
-
- // Then write the source stream into the dest stream.
- BinaryStreamWriter DestWriter(*Dest);
- EXPECT_THAT_ERROR(DestWriter.writeStreamRef(*Src), Succeeded());
- EXPECT_EQ(DestDataBytes, std::vector<uint8_t>(
- {'s', 'e', 'T', ' ', 'S', 't', 't', 'r', 0, 0}));
-
- // Then read the string back out of the dest stream.
- StringRef Result;
- BinaryStreamReader DestReader(*Dest);
- EXPECT_THAT_ERROR(DestReader.readCString(Result), Succeeded());
- EXPECT_EQ(Result, "Test Str");
-}
-
-TEST(MappedBlockStreamTest, DataLivesAfterStreamDestruction) {
- std::vector<uint8_t> DataBytes(10);
- MutableArrayRef<uint8_t> Data(DataBytes);
- const uint32_t Blocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8};
-
- StringRef Str[] = {"Zero Str", ""};
-
- DiscontiguousStream F(Blocks, Data);
- {
- auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(),
- F, F.Allocator);
-
- BinaryStreamReader Reader(*S);
- BinaryStreamWriter Writer(*S);
- ::memset(DataBytes.data(), 0, 10);
- EXPECT_THAT_ERROR(Writer.writeCString(Str[0]), Succeeded());
- EXPECT_THAT_ERROR(Reader.readCString(Str[1]), Succeeded());
- EXPECT_EQ(Str[0], Str[1]);
- }
-
- EXPECT_EQ(Str[0], Str[1]);
-}
-
-} // end anonymous namespace
diff --git a/unittests/DebugInfo/PDB/PDBApiTest.cpp b/unittests/DebugInfo/PDB/PDBApiTest.cpp
index 257a8879e439..e998acf009ec 100644
--- a/unittests/DebugInfo/PDB/PDBApiTest.cpp
+++ b/unittests/DebugInfo/PDB/PDBApiTest.cpp
@@ -14,6 +14,7 @@
#include "llvm/DebugInfo/PDB/IPDBRawSymbol.h"
#include "llvm/DebugInfo/PDB/IPDBSession.h"
#include "llvm/DebugInfo/PDB/IPDBSourceFile.h"
+#include "llvm/DebugInfo/PDB/IPDBTable.h"
#include "llvm/DebugInfo/PDB/PDBSymbol.h"
#include "llvm/DebugInfo/PDB/PDBSymbolAnnotation.h"
@@ -118,6 +119,10 @@ class MockSession : public IPDBSession {
std::unique_ptr<IPDBEnumDataStreams> getDebugStreams() const override {
return nullptr;
}
+
+ std::unique_ptr<IPDBEnumTables> getEnumTables() const override {
+ return nullptr;
+ }
};
class MockRawSymbol : public IPDBRawSymbol {
@@ -152,6 +157,10 @@ public:
PDB_SymType getSymTag() const override { return Type; }
+ std::string getUndecoratedNameEx(PDB_UndnameFlags Flags) const override {
+ return {};
+ }
+
MOCK_SYMBOL_ACCESSOR(getAccess)
MOCK_SYMBOL_ACCESSOR(getAddressOffset)
MOCK_SYMBOL_ACCESSOR(getAddressSection)