Zachary Turner | 018338e | 2017-08-02 22:26:09 +0000 | [diff] [blame^] | 1 | //===- llvm/unittest/DebugInfo/MSF/MappedBlockStreamTest.cpp --------------===// |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 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 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 10 | #include "llvm/DebugInfo/MSF/MappedBlockStream.h" |
Zachary Turner | a3225b0 | 2016-07-29 20:56:36 +0000 | [diff] [blame] | 11 | #include "llvm/DebugInfo/MSF/IMSFFile.h" |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 12 | #include "llvm/DebugInfo/MSF/MSFError.h" |
Zachary Turner | d9dc282 | 2017-03-02 20:52:51 +0000 | [diff] [blame] | 13 | #include "llvm/Support/BinaryByteStream.h" |
| 14 | #include "llvm/Support/BinaryStreamReader.h" |
| 15 | #include "llvm/Support/BinaryStreamRef.h" |
| 16 | #include "llvm/Support/BinaryStreamWriter.h" |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 17 | #include "llvm/Testing/Support/Error.h" |
| 18 | |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 19 | #include "gtest/gtest.h" |
| 20 | |
Zachary Turner | f52a899 | 2016-07-15 20:43:38 +0000 | [diff] [blame] | 21 | #include <unordered_map> |
| 22 | |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 23 | using namespace llvm; |
Zachary Turner | bac69d3 | 2016-07-22 19:56:05 +0000 | [diff] [blame] | 24 | using namespace llvm::msf; |
Zachary Turner | 695ed56 | 2017-02-28 00:04:07 +0000 | [diff] [blame] | 25 | using namespace llvm::support; |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 26 | |
| 27 | namespace { |
| 28 | |
NAKAMURA Takumi | 47d1e37 | 2016-06-01 14:26:54 +0000 | [diff] [blame] | 29 | static const uint32_t BlocksAry[] = {0, 1, 2, 5, 4, 3, 6, 7, 8, 9}; |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 30 | static uint8_t DataAry[] = {'A', 'B', 'C', 'F', 'E', 'D', 'G', 'H', 'I', 'J'}; |
NAKAMURA Takumi | 47d1e37 | 2016-06-01 14:26:54 +0000 | [diff] [blame] | 31 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 32 | class DiscontiguousStream : public WritableBinaryStream { |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 33 | public: |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 34 | DiscontiguousStream(ArrayRef<uint32_t> Blocks, MutableArrayRef<uint8_t> Data) |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 35 | : Blocks(Blocks.begin(), Blocks.end()), Data(Data.begin(), Data.end()) {} |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 36 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 37 | uint32_t block_size() const { return 1; } |
| 38 | uint32_t block_count() const { return Blocks.size(); } |
| 39 | |
Zachary Turner | 695ed56 | 2017-02-28 00:04:07 +0000 | [diff] [blame] | 40 | endianness getEndian() const override { return little; } |
| 41 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 42 | Error readBytes(uint32_t Offset, uint32_t Size, |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 43 | ArrayRef<uint8_t> &Buffer) override { |
Zachary Turner | d0b44fa | 2017-02-28 17:49:34 +0000 | [diff] [blame] | 44 | if (auto EC = checkOffset(Offset, Size)) |
| 45 | return EC; |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 46 | Buffer = Data.slice(Offset, Size); |
| 47 | return Error::success(); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 48 | } |
| 49 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 50 | Error readLongestContiguousChunk(uint32_t Offset, |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 51 | ArrayRef<uint8_t> &Buffer) override { |
Zachary Turner | d0b44fa | 2017-02-28 17:49:34 +0000 | [diff] [blame] | 52 | if (auto EC = checkOffset(Offset, 1)) |
| 53 | return EC; |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 54 | Buffer = Data.drop_front(Offset); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 55 | return Error::success(); |
| 56 | } |
| 57 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 58 | uint32_t getLength() override { return Data.size(); } |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 59 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 60 | Error writeBytes(uint32_t Offset, ArrayRef<uint8_t> SrcData) override { |
Zachary Turner | d0b44fa | 2017-02-28 17:49:34 +0000 | [diff] [blame] | 61 | if (auto EC = checkOffset(Offset, SrcData.size())) |
| 62 | return EC; |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 63 | ::memcpy(&Data[Offset], SrcData.data(), SrcData.size()); |
| 64 | return Error::success(); |
| 65 | } |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 66 | Error commit() override { return Error::success(); } |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 67 | |
Zachary Turner | a3225b0 | 2016-07-29 20:56:36 +0000 | [diff] [blame] | 68 | MSFStreamLayout layout() const { |
| 69 | return MSFStreamLayout{static_cast<uint32_t>(Data.size()), Blocks}; |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 70 | } |
| 71 | |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 72 | BumpPtrAllocator Allocator; |
| 73 | |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 74 | private: |
Zachary Turner | 92d9e97 | 2016-06-07 05:32:48 +0000 | [diff] [blame] | 75 | std::vector<support::ulittle32_t> Blocks; |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 76 | MutableArrayRef<uint8_t> Data; |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 77 | }; |
| 78 | |
Zachary Turner | c4e4b7e | 2017-05-25 21:12:00 +0000 | [diff] [blame] | 79 | TEST(MappedBlockStreamTest, NumBlocks) { |
| 80 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 81 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 82 | F.Allocator); |
Zachary Turner | c4e4b7e | 2017-05-25 21:12:00 +0000 | [diff] [blame] | 83 | EXPECT_EQ(F.block_size(), S->getBlockSize()); |
| 84 | EXPECT_EQ(F.layout().Blocks.size(), S->getNumBlocks()); |
Zachary Turner | c4e4b7e | 2017-05-25 21:12:00 +0000 | [diff] [blame] | 85 | } |
| 86 | |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 87 | // Tests that a read which is entirely contained within a single block works |
| 88 | // and does not allocate. |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 89 | TEST(MappedBlockStreamTest, ReadBeyondEndOfStreamRef) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 90 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 91 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 92 | F.Allocator); |
Zachary Turner | 199f48a | 2016-07-28 19:11:09 +0000 | [diff] [blame] | 93 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 94 | BinaryStreamReader R(*S); |
| 95 | BinaryStreamRef SR; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 96 | EXPECT_THAT_ERROR(R.readStreamRef(SR, 0U), Succeeded()); |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 97 | ArrayRef<uint8_t> Buffer; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 98 | EXPECT_THAT_ERROR(SR.readBytes(0U, 1U, Buffer), Failed()); |
| 99 | EXPECT_THAT_ERROR(R.readStreamRef(SR, 1U), Succeeded()); |
| 100 | EXPECT_THAT_ERROR(SR.readBytes(1U, 1U, Buffer), Failed()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 101 | } |
| 102 | |
| 103 | // Tests that a read which outputs into a full destination buffer works and |
| 104 | // does not fail due to the length of the output buffer. |
| 105 | TEST(MappedBlockStreamTest, ReadOntoNonEmptyBuffer) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 106 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 107 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 108 | F.Allocator); |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 109 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 110 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 111 | StringRef Str = "ZYXWVUTSRQPONMLKJIHGFEDCBA"; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 112 | EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Succeeded()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 113 | EXPECT_EQ(Str, StringRef("A")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 114 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 115 | } |
| 116 | |
| 117 | // Tests that a read which crosses a block boundary, but where the subsequent |
| 118 | // blocks are still contiguous in memory to the previous block works and does |
| 119 | // not allocate memory. |
| 120 | TEST(MappedBlockStreamTest, ZeroCopyReadContiguousBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 121 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 122 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 123 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 124 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 125 | StringRef Str; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 126 | EXPECT_THAT_ERROR(R.readFixedString(Str, 2), Succeeded()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 127 | EXPECT_EQ(Str, StringRef("AB")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 128 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 129 | |
| 130 | R.setOffset(6); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 131 | EXPECT_THAT_ERROR(R.readFixedString(Str, 4), Succeeded()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 132 | EXPECT_EQ(Str, StringRef("GHIJ")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 133 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 134 | } |
| 135 | |
| 136 | // Tests that a read which crosses a block boundary and cannot be referenced |
| 137 | // contiguously works and allocates only the precise amount of bytes |
| 138 | // requested. |
| 139 | TEST(MappedBlockStreamTest, CopyReadNonContiguousBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 140 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 141 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 142 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 143 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 144 | StringRef Str; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 145 | EXPECT_THAT_ERROR(R.readFixedString(Str, 10), Succeeded()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 146 | EXPECT_EQ(Str, StringRef("ABCDEFGHIJ")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 147 | EXPECT_EQ(10U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | // Test that an out of bounds read which doesn't cross a block boundary |
| 151 | // fails and allocates no memory. |
| 152 | TEST(MappedBlockStreamTest, InvalidReadSizeNoBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 153 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 154 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 155 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 156 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 157 | StringRef Str; |
| 158 | |
| 159 | R.setOffset(10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 160 | EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Failed()); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 161 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 162 | } |
| 163 | |
| 164 | // Test that an out of bounds read which crosses a contiguous block boundary |
| 165 | // fails and allocates no memory. |
| 166 | TEST(MappedBlockStreamTest, InvalidReadSizeContiguousBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 167 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 168 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 169 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 170 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 171 | StringRef Str; |
| 172 | |
| 173 | R.setOffset(6); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 174 | EXPECT_THAT_ERROR(R.readFixedString(Str, 5), Failed()); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 175 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | // Test that an out of bounds read which crosses a discontiguous block |
| 179 | // boundary fails and allocates no memory. |
| 180 | TEST(MappedBlockStreamTest, InvalidReadSizeNonContiguousBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 181 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 182 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 183 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 184 | BinaryStreamReader R(*S); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 185 | StringRef Str; |
| 186 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 187 | EXPECT_THAT_ERROR(R.readFixedString(Str, 11), Failed()); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 188 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 189 | } |
| 190 | |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 191 | // Tests that a read which is entirely contained within a single block but |
| 192 | // beyond the end of a StreamRef fails. |
| 193 | TEST(MappedBlockStreamTest, ZeroCopyReadNoBreak) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 194 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 195 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 196 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 197 | BinaryStreamReader R(*S); |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 198 | StringRef Str; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 199 | EXPECT_THAT_ERROR(R.readFixedString(Str, 1), Succeeded()); |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 200 | EXPECT_EQ(Str, StringRef("A")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 201 | EXPECT_EQ(0U, F.Allocator.getBytesAllocated()); |
David Majnemer | b6aa875 | 2016-06-01 18:13:06 +0000 | [diff] [blame] | 202 | } |
| 203 | |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 204 | // Tests that a read which is not aligned on the same boundary as a previous |
| 205 | // cached request, but which is known to overlap that request, shares the |
| 206 | // previous allocation. |
| 207 | TEST(MappedBlockStreamTest, UnalignedOverlappingRead) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 208 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 209 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 210 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 211 | BinaryStreamReader R(*S); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 212 | StringRef Str1; |
| 213 | StringRef Str2; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 214 | EXPECT_THAT_ERROR(R.readFixedString(Str1, 7), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 215 | EXPECT_EQ(Str1, StringRef("ABCDEFG")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 216 | EXPECT_EQ(7U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 217 | |
| 218 | R.setOffset(2); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 219 | EXPECT_THAT_ERROR(R.readFixedString(Str2, 3), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 220 | EXPECT_EQ(Str2, StringRef("CDE")); |
| 221 | EXPECT_EQ(Str1.data() + 2, Str2.data()); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 222 | EXPECT_EQ(7U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 223 | } |
| 224 | |
| 225 | // Tests that a read which is not aligned on the same boundary as a previous |
| 226 | // cached request, but which only partially overlaps a previous cached request, |
| 227 | // still works correctly and allocates again from the shared pool. |
| 228 | TEST(MappedBlockStreamTest, UnalignedOverlappingReadFail) { |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 229 | DiscontiguousStream F(BlocksAry, DataAry); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 230 | auto S = MappedBlockStream::createStream(F.block_size(), F.layout(), F, |
| 231 | F.Allocator); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 232 | BinaryStreamReader R(*S); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 233 | StringRef Str1; |
| 234 | StringRef Str2; |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 235 | EXPECT_THAT_ERROR(R.readFixedString(Str1, 6), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 236 | EXPECT_EQ(Str1, StringRef("ABCDEF")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 237 | EXPECT_EQ(6U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 238 | |
| 239 | R.setOffset(4); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 240 | EXPECT_THAT_ERROR(R.readFixedString(Str2, 4), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 241 | EXPECT_EQ(Str2, StringRef("EFGH")); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 242 | EXPECT_EQ(10U, F.Allocator.getBytesAllocated()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 243 | } |
| 244 | |
| 245 | TEST(MappedBlockStreamTest, WriteBeyondEndOfStream) { |
| 246 | static uint8_t Data[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J'}; |
| 247 | static uint8_t LargeBuffer[] = {'0', '1', '2', '3', '4', '5', |
| 248 | '6', '7', '8', '9', 'A'}; |
| 249 | static uint8_t SmallBuffer[] = {'0', '1', '2'}; |
| 250 | static_assert(sizeof(LargeBuffer) > sizeof(Data), |
| 251 | "LargeBuffer is not big enough"); |
| 252 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 253 | DiscontiguousStream F(BlocksAry, Data); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 254 | auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(), |
| 255 | F, F.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 256 | ArrayRef<uint8_t> Buffer; |
| 257 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 258 | EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>(LargeBuffer)), Failed()); |
| 259 | EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>(SmallBuffer)), |
| 260 | Succeeded()); |
| 261 | EXPECT_THAT_ERROR(S->writeBytes(7, ArrayRef<uint8_t>(SmallBuffer)), |
| 262 | Succeeded()); |
| 263 | EXPECT_THAT_ERROR(S->writeBytes(8, ArrayRef<uint8_t>(SmallBuffer)), Failed()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 264 | } |
| 265 | |
| 266 | TEST(MappedBlockStreamTest, TestWriteBytesNoBreakBoundary) { |
| 267 | static uint8_t Data[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J'}; |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 268 | DiscontiguousStream F(BlocksAry, Data); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 269 | auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(), |
| 270 | F, F.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 271 | ArrayRef<uint8_t> Buffer; |
| 272 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 273 | EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 274 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A')); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 275 | EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 276 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J')); |
| 277 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 278 | EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>('J')), Succeeded()); |
| 279 | EXPECT_THAT_ERROR(S->writeBytes(9, ArrayRef<uint8_t>('A')), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 280 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 281 | EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 282 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J')); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 283 | EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 284 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A')); |
| 285 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 286 | EXPECT_THAT_ERROR(S->writeBytes(0, ArrayRef<uint8_t>('A')), Succeeded()); |
| 287 | EXPECT_THAT_ERROR(S->writeBytes(9, ArrayRef<uint8_t>('J')), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 288 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 289 | EXPECT_THAT_ERROR(S->readBytes(0, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 290 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('A')); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 291 | EXPECT_THAT_ERROR(S->readBytes(9, 1, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 292 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>('J')); |
| 293 | } |
| 294 | |
| 295 | TEST(MappedBlockStreamTest, TestWriteBytesBreakBoundary) { |
| 296 | static uint8_t Data[] = {'0', '0', '0', '0', '0', '0', '0', '0', '0', '0'}; |
| 297 | static uint8_t TestData[] = {'T', 'E', 'S', 'T', 'I', 'N', 'G', '.'}; |
| 298 | static uint8_t Expected[] = {'T', 'E', 'S', 'N', 'I', |
| 299 | 'T', 'G', '.', '0', '0'}; |
| 300 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 301 | DiscontiguousStream F(BlocksAry, Data); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 302 | auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(), |
| 303 | F, F.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 304 | ArrayRef<uint8_t> Buffer; |
| 305 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 306 | EXPECT_THAT_ERROR(S->writeBytes(0, TestData), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 307 | // First just compare the memory, then compare the result of reading the |
| 308 | // string out. |
| 309 | EXPECT_EQ(ArrayRef<uint8_t>(Data), ArrayRef<uint8_t>(Expected)); |
| 310 | |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 311 | EXPECT_THAT_ERROR(S->readBytes(0, 8, Buffer), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 312 | EXPECT_EQ(Buffer, ArrayRef<uint8_t>(TestData)); |
| 313 | } |
| 314 | |
| 315 | TEST(MappedBlockStreamTest, TestWriteThenRead) { |
| 316 | std::vector<uint8_t> DataBytes(10); |
| 317 | MutableArrayRef<uint8_t> Data(DataBytes); |
| 318 | const uint32_t Blocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8}; |
| 319 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 320 | DiscontiguousStream F(Blocks, Data); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 321 | auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(), |
| 322 | F, F.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 323 | |
| 324 | enum class MyEnum : uint32_t { Val1 = 2908234, Val2 = 120891234 }; |
Zachary Turner | 97609bb | 2016-06-10 21:47:26 +0000 | [diff] [blame] | 325 | using support::ulittle32_t; |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 326 | |
| 327 | uint16_t u16[] = {31468, 0}; |
| 328 | uint32_t u32[] = {890723408, 0}; |
| 329 | MyEnum Enum[] = {MyEnum::Val1, MyEnum::Val2}; |
| 330 | StringRef ZStr[] = {"Zero Str", ""}; |
| 331 | StringRef FStr[] = {"Fixed Str", ""}; |
NAKAMURA Takumi | f1b183c | 2016-06-11 06:37:28 +0000 | [diff] [blame] | 332 | uint8_t byteArray0[] = {'1', '2'}; |
| 333 | uint8_t byteArray1[] = {'0', '0'}; |
Zachary Turner | 352cfe4 | 2016-07-08 16:57:14 +0000 | [diff] [blame] | 334 | ArrayRef<uint8_t> byteArrayRef0(byteArray0); |
| 335 | ArrayRef<uint8_t> byteArrayRef1(byteArray1); |
Zachary Turner | 018338e | 2017-08-02 22:26:09 +0000 | [diff] [blame^] | 336 | ArrayRef<uint8_t> byteArray[] = {byteArrayRef0, byteArrayRef1}; |
Reid Kleckner | 84149c4 | 2016-07-28 18:17:01 +0000 | [diff] [blame] | 337 | uint32_t intArr0[] = {890723408, 29082234}; |
| 338 | uint32_t intArr1[] = {890723408, 29082234}; |
| 339 | ArrayRef<uint32_t> intArray[] = {intArr0, intArr1}; |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 340 | |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 341 | BinaryStreamReader Reader(*S); |
| 342 | BinaryStreamWriter Writer(*S); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 343 | EXPECT_THAT_ERROR(Writer.writeInteger(u16[0]), Succeeded()); |
| 344 | EXPECT_THAT_ERROR(Reader.readInteger(u16[1]), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 345 | EXPECT_EQ(u16[0], u16[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 346 | EXPECT_EQ(std::vector<uint8_t>({0, 0x7A, 0xEC, 0, 0, 0, 0, 0, 0, 0}), |
| 347 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 348 | |
| 349 | Reader.setOffset(0); |
| 350 | Writer.setOffset(0); |
| 351 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 352 | EXPECT_THAT_ERROR(Writer.writeInteger(u32[0]), Succeeded()); |
| 353 | EXPECT_THAT_ERROR(Reader.readInteger(u32[1]), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 354 | EXPECT_EQ(u32[0], u32[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 355 | EXPECT_EQ(std::vector<uint8_t>({0x17, 0x5C, 0x50, 0, 0, 0, 0x35, 0, 0, 0}), |
| 356 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 357 | |
| 358 | Reader.setOffset(0); |
| 359 | Writer.setOffset(0); |
| 360 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 361 | EXPECT_THAT_ERROR(Writer.writeEnum(Enum[0]), Succeeded()); |
| 362 | EXPECT_THAT_ERROR(Reader.readEnum(Enum[1]), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 363 | EXPECT_EQ(Enum[0], Enum[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 364 | EXPECT_EQ(std::vector<uint8_t>({0x2C, 0x60, 0x4A, 0, 0, 0, 0, 0, 0, 0}), |
| 365 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 366 | |
| 367 | Reader.setOffset(0); |
| 368 | Writer.setOffset(0); |
| 369 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 370 | EXPECT_THAT_ERROR(Writer.writeCString(ZStr[0]), Succeeded()); |
| 371 | EXPECT_THAT_ERROR(Reader.readCString(ZStr[1]), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 372 | EXPECT_EQ(ZStr[0], ZStr[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 373 | EXPECT_EQ( |
| 374 | std::vector<uint8_t>({'r', 'e', 'Z', ' ', 'S', 't', 'o', 'r', 0, 0}), |
| 375 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 376 | |
| 377 | Reader.setOffset(0); |
| 378 | Writer.setOffset(0); |
| 379 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 380 | EXPECT_THAT_ERROR(Writer.writeFixedString(FStr[0]), Succeeded()); |
| 381 | EXPECT_THAT_ERROR(Reader.readFixedString(FStr[1], FStr[0].size()), |
| 382 | Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 383 | EXPECT_EQ(FStr[0], FStr[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 384 | EXPECT_EQ( |
| 385 | std::vector<uint8_t>({'x', 'i', 'F', 'd', ' ', 'S', 'e', 't', 0, 'r'}), |
| 386 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 387 | |
| 388 | Reader.setOffset(0); |
| 389 | Writer.setOffset(0); |
| 390 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 391 | EXPECT_THAT_ERROR(Writer.writeArray(byteArray[0]), Succeeded()); |
| 392 | EXPECT_THAT_ERROR(Reader.readArray(byteArray[1], byteArray[0].size()), |
| 393 | Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 394 | EXPECT_EQ(byteArray[0], byteArray[1]); |
Zachary Turner | d4c1efc | 2016-06-10 22:12:18 +0000 | [diff] [blame] | 395 | EXPECT_EQ(std::vector<uint8_t>({0, 0x32, 0x31, 0, 0, 0, 0, 0, 0, 0}), |
| 396 | DataBytes); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 397 | |
| 398 | Reader.setOffset(0); |
| 399 | Writer.setOffset(0); |
| 400 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 401 | EXPECT_THAT_ERROR(Writer.writeArray(intArray[0]), Succeeded()); |
| 402 | EXPECT_THAT_ERROR(Reader.readArray(intArray[1], intArray[0].size()), |
| 403 | Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 404 | EXPECT_EQ(intArray[0], intArray[1]); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 405 | } |
| 406 | |
| 407 | TEST(MappedBlockStreamTest, TestWriteContiguousStreamRef) { |
| 408 | std::vector<uint8_t> DestDataBytes(10); |
| 409 | MutableArrayRef<uint8_t> DestData(DestDataBytes); |
| 410 | const uint32_t DestBlocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8}; |
| 411 | |
| 412 | std::vector<uint8_t> SrcDataBytes(10); |
| 413 | MutableArrayRef<uint8_t> SrcData(SrcDataBytes); |
| 414 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 415 | DiscontiguousStream F(DestBlocks, DestData); |
| 416 | auto DestStream = WritableMappedBlockStream::createStream( |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 417 | F.block_size(), F.layout(), F, F.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 418 | |
| 419 | // First write "Test Str" into the source stream. |
Zachary Turner | 695ed56 | 2017-02-28 00:04:07 +0000 | [diff] [blame] | 420 | MutableBinaryByteStream SourceStream(SrcData, little); |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 421 | BinaryStreamWriter SourceWriter(SourceStream); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 422 | EXPECT_THAT_ERROR(SourceWriter.writeCString("Test Str"), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 423 | EXPECT_EQ(SrcDataBytes, std::vector<uint8_t>( |
| 424 | {'T', 'e', 's', 't', ' ', 'S', 't', 'r', 0, 0})); |
| 425 | |
| 426 | // Then write the source stream into the dest stream. |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 427 | BinaryStreamWriter DestWriter(*DestStream); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 428 | EXPECT_THAT_ERROR(DestWriter.writeStreamRef(SourceStream), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 429 | EXPECT_EQ(DestDataBytes, std::vector<uint8_t>( |
| 430 | {'s', 'e', 'T', ' ', 'S', 't', 't', 'r', 0, 0})); |
| 431 | |
| 432 | // Then read the string back out of the dest stream. |
| 433 | StringRef Result; |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 434 | BinaryStreamReader DestReader(*DestStream); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 435 | EXPECT_THAT_ERROR(DestReader.readCString(Result), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 436 | EXPECT_EQ(Result, "Test Str"); |
| 437 | } |
| 438 | |
| 439 | TEST(MappedBlockStreamTest, TestWriteDiscontiguousStreamRef) { |
| 440 | std::vector<uint8_t> DestDataBytes(10); |
| 441 | MutableArrayRef<uint8_t> DestData(DestDataBytes); |
| 442 | const uint32_t DestBlocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8}; |
| 443 | |
| 444 | std::vector<uint8_t> SrcDataBytes(10); |
| 445 | MutableArrayRef<uint8_t> SrcData(SrcDataBytes); |
| 446 | const uint32_t SrcBlocks[] = {1, 0, 6, 3, 4, 5, 2, 7, 8, 9}; |
| 447 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 448 | DiscontiguousStream DestF(DestBlocks, DestData); |
| 449 | DiscontiguousStream SrcF(SrcBlocks, SrcData); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 450 | |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 451 | auto Dest = WritableMappedBlockStream::createStream( |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 452 | DestF.block_size(), DestF.layout(), DestF, DestF.Allocator); |
Zachary Turner | d66889c | 2016-07-28 19:12:28 +0000 | [diff] [blame] | 453 | auto Src = WritableMappedBlockStream::createStream( |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 454 | SrcF.block_size(), SrcF.layout(), SrcF, SrcF.Allocator); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 455 | |
| 456 | // First write "Test Str" into the source stream. |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 457 | BinaryStreamWriter SourceWriter(*Src); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 458 | EXPECT_THAT_ERROR(SourceWriter.writeCString("Test Str"), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 459 | EXPECT_EQ(SrcDataBytes, std::vector<uint8_t>( |
| 460 | {'e', 'T', 't', 't', ' ', 'S', 's', 'r', 0, 0})); |
| 461 | |
| 462 | // Then write the source stream into the dest stream. |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 463 | BinaryStreamWriter DestWriter(*Dest); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 464 | EXPECT_THAT_ERROR(DestWriter.writeStreamRef(*Src), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 465 | EXPECT_EQ(DestDataBytes, std::vector<uint8_t>( |
| 466 | {'s', 'e', 'T', ' ', 'S', 't', 't', 'r', 0, 0})); |
| 467 | |
| 468 | // Then read the string back out of the dest stream. |
| 469 | StringRef Result; |
Zachary Turner | 120faca | 2017-02-27 22:11:43 +0000 | [diff] [blame] | 470 | BinaryStreamReader DestReader(*Dest); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 471 | EXPECT_THAT_ERROR(DestReader.readCString(Result), Succeeded()); |
Zachary Turner | 5acb4ac | 2016-06-10 05:09:12 +0000 | [diff] [blame] | 472 | EXPECT_EQ(Result, "Test Str"); |
| 473 | } |
| 474 | |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 475 | TEST(MappedBlockStreamTest, DataLivesAfterStreamDestruction) { |
| 476 | std::vector<uint8_t> DataBytes(10); |
| 477 | MutableArrayRef<uint8_t> Data(DataBytes); |
| 478 | const uint32_t Blocks[] = {2, 1, 0, 6, 3, 4, 5, 7, 9, 8}; |
| 479 | |
| 480 | StringRef Str[] = {"Zero Str", ""}; |
| 481 | |
| 482 | DiscontiguousStream F(Blocks, Data); |
| 483 | { |
| 484 | auto S = WritableMappedBlockStream::createStream(F.block_size(), F.layout(), |
| 485 | F, F.Allocator); |
| 486 | |
| 487 | BinaryStreamReader Reader(*S); |
| 488 | BinaryStreamWriter Writer(*S); |
| 489 | ::memset(DataBytes.data(), 0, 10); |
Zachary Turner | cb30e70 | 2017-06-14 16:41:50 +0000 | [diff] [blame] | 490 | EXPECT_THAT_ERROR(Writer.writeCString(Str[0]), Succeeded()); |
| 491 | EXPECT_THAT_ERROR(Reader.readCString(Str[1]), Succeeded()); |
Zachary Turner | 5b74ff3 | 2017-06-03 00:33:35 +0000 | [diff] [blame] | 492 | EXPECT_EQ(Str[0], Str[1]); |
| 493 | } |
| 494 | |
| 495 | EXPECT_EQ(Str[0], Str[1]); |
| 496 | } |
| 497 | |
Zachary Turner | 90b8b8d | 2016-05-31 22:41:52 +0000 | [diff] [blame] | 498 | } // end anonymous namespace |