diff options
| author | Dimitry Andric <dim@FreeBSD.org> | 2017-12-18 20:10:56 +0000 |
|---|---|---|
| committer | Dimitry Andric <dim@FreeBSD.org> | 2017-12-18 20:10:56 +0000 |
| commit | 044eb2f6afba375a914ac9d8024f8f5142bb912e (patch) | |
| tree | 1475247dc9f9fe5be155ebd4c9069c75aadf8c20 /unittests/DebugInfo/PDB | |
| parent | eb70dddbd77e120e5d490bd8fbe7ff3f8fa81c6b (diff) | |
Notes
Diffstat (limited to 'unittests/DebugInfo/PDB')
| -rw-r--r-- | unittests/DebugInfo/PDB/CMakeLists.txt | 4 | ||||
| -rw-r--r-- | unittests/DebugInfo/PDB/MSFBuilderTest.cpp | 361 | ||||
| -rw-r--r-- | unittests/DebugInfo/PDB/MappedBlockStreamTest.cpp | 500 | ||||
| -rw-r--r-- | unittests/DebugInfo/PDB/PDBApiTest.cpp | 9 |
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) |
