0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
1 //===- llvm/unittest/Support/MemoryBufferTest.cpp - MemoryBuffer tests ----===//
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
2 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
3 // The LLVM Compiler Infrastructure
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
4 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
5 // This file is distributed under the University of Illinois Open Source
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
6 // License. See LICENSE.TXT for details.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
7 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
8 //===----------------------------------------------------------------------===//
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
9 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
10 // This file implements unit tests for the MemoryBuffer support class.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
11 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
12 //===----------------------------------------------------------------------===//
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
13
|
121
|
14 #include "llvm/Support/MemoryBuffer.h"
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
15 #include "llvm/Support/FileSystem.h"
|
120
|
16 #include "llvm/Support/FileUtilities.h"
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
17 #include "llvm/Support/raw_ostream.h"
|
134
|
18 #include "llvm/Testing/Support/Error.h"
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
19 #include "gtest/gtest.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
20
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
21 using namespace llvm;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
22
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
23 namespace {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
24
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
25 class MemoryBufferTest : public testing::Test {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
26 protected:
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
27 MemoryBufferTest()
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
28 : data("this is some data")
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
29 { }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
30
|
95
|
31 void SetUp() override {}
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
32
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
33 /// Common testing for different modes of getOpenFileSlice.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
34 /// Creates a temporary file with known contents, and uses
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
35 /// MemoryBuffer::getOpenFileSlice to map it.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
36 /// If \p Reopen is true, the file is closed after creating and reopened
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
37 /// anew before using MemoryBuffer.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
38 void testGetOpenFileSlice(bool Reopen);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
39
|
77
|
40 typedef std::unique_ptr<MemoryBuffer> OwningBuffer;
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
41
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
42 std::string data;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
43 };
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
44
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
45 TEST_F(MemoryBufferTest, get) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
46 // Default name and null-terminator flag
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
47 OwningBuffer MB1(MemoryBuffer::getMemBuffer(data));
|
77
|
48 EXPECT_TRUE(nullptr != MB1.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
49
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
50 // RequiresNullTerminator = false
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
51 OwningBuffer MB2(MemoryBuffer::getMemBuffer(data, "one", false));
|
77
|
52 EXPECT_TRUE(nullptr != MB2.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
53
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
54 // RequiresNullTerminator = true
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
55 OwningBuffer MB3(MemoryBuffer::getMemBuffer(data, "two", true));
|
77
|
56 EXPECT_TRUE(nullptr != MB3.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
57
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
58 // verify all 3 buffers point to the same address
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
59 EXPECT_EQ(MB1->getBufferStart(), MB2->getBufferStart());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
60 EXPECT_EQ(MB2->getBufferStart(), MB3->getBufferStart());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
61
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
62 // verify the original data is unmodified after deleting the buffers
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
63 MB1.reset();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
64 MB2.reset();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
65 MB3.reset();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
66 EXPECT_EQ("this is some data", data);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
67 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
68
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
69 TEST_F(MemoryBufferTest, NullTerminator4K) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
70 // Test that a file with size that is a multiple of the page size can be null
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
71 // terminated correctly by MemoryBuffer.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
72 int TestFD;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
73 SmallString<64> TestPath;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
74 sys::fs::createTemporaryFile("MemoryBufferTest_NullTerminator4K", "temp",
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
75 TestFD, TestPath);
|
120
|
76 FileRemover Cleanup(TestPath);
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
77 raw_fd_ostream OF(TestFD, true, /*unbuffered=*/true);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
78 for (unsigned i = 0; i < 4096 / 16; ++i) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
79 OF << "0123456789abcdef";
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
80 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
81 OF.close();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
82
|
77
|
83 ErrorOr<OwningBuffer> MB = MemoryBuffer::getFile(TestPath.c_str());
|
|
84 std::error_code EC = MB.getError();
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
85 ASSERT_FALSE(EC);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
86
|
77
|
87 const char *BufData = MB.get()->getBufferStart();
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
88 EXPECT_EQ('f', BufData[4095]);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
89 EXPECT_EQ('\0', BufData[4096]);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
90 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
91
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
92 TEST_F(MemoryBufferTest, copy) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
93 // copy with no name
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
94 OwningBuffer MBC1(MemoryBuffer::getMemBufferCopy(data));
|
77
|
95 EXPECT_TRUE(nullptr != MBC1.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
96
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
97 // copy with a name
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
98 OwningBuffer MBC2(MemoryBuffer::getMemBufferCopy(data, "copy"));
|
77
|
99 EXPECT_TRUE(nullptr != MBC2.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
100
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
101 // verify the two copies do not point to the same place
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
102 EXPECT_NE(MBC1->getBufferStart(), MBC2->getBufferStart());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
103 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
104
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
105 TEST_F(MemoryBufferTest, make_new) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
106 // 0-sized buffer
|
134
|
107 OwningBuffer Zero(WritableMemoryBuffer::getNewUninitMemBuffer(0));
|
77
|
108 EXPECT_TRUE(nullptr != Zero.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
109
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
110 // uninitialized buffer with no name
|
134
|
111 OwningBuffer One(WritableMemoryBuffer::getNewUninitMemBuffer(321));
|
77
|
112 EXPECT_TRUE(nullptr != One.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
113
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
114 // uninitialized buffer with name
|
134
|
115 OwningBuffer Two(WritableMemoryBuffer::getNewUninitMemBuffer(123, "bla"));
|
77
|
116 EXPECT_TRUE(nullptr != Two.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
117
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
118 // 0-initialized buffer with no name
|
134
|
119 OwningBuffer Three(WritableMemoryBuffer::getNewMemBuffer(321, data));
|
77
|
120 EXPECT_TRUE(nullptr != Three.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
121 for (size_t i = 0; i < 321; ++i)
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
122 EXPECT_EQ(0, Three->getBufferStart()[0]);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
123
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
124 // 0-initialized buffer with name
|
134
|
125 OwningBuffer Four(WritableMemoryBuffer::getNewMemBuffer(123, "zeros"));
|
77
|
126 EXPECT_TRUE(nullptr != Four.get());
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
127 for (size_t i = 0; i < 123; ++i)
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
128 EXPECT_EQ(0, Four->getBufferStart()[0]);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
129 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
130
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
131 void MemoryBufferTest::testGetOpenFileSlice(bool Reopen) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
132 // Test that MemoryBuffer::getOpenFile works properly when no null
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
133 // terminator is requested and the size is large enough to trigger
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
134 // the usage of memory mapping.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
135 int TestFD;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
136 SmallString<64> TestPath;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
137 // Create a temporary file and write data into it.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
138 sys::fs::createTemporaryFile("prefix", "temp", TestFD, TestPath);
|
120
|
139 FileRemover Cleanup(TestPath);
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
140 // OF is responsible for closing the file; If the file is not
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
141 // reopened, it will be unbuffered so that the results are
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
142 // immediately visible through the fd.
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
143 raw_fd_ostream OF(TestFD, true, !Reopen);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
144 for (int i = 0; i < 60000; ++i) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
145 OF << "0123456789";
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
146 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
147
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
148 if (Reopen) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
149 OF.close();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
150 EXPECT_FALSE(sys::fs::openFileForRead(TestPath.c_str(), TestFD));
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
151 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
152
|
77
|
153 ErrorOr<OwningBuffer> Buf =
|
|
154 MemoryBuffer::getOpenFileSlice(TestFD, TestPath.c_str(),
|
|
155 40000, // Size
|
|
156 80000 // Offset
|
|
157 );
|
|
158
|
|
159 std::error_code EC = Buf.getError();
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
160 EXPECT_FALSE(EC);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
161
|
77
|
162 StringRef BufData = Buf.get()->getBuffer();
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
163 EXPECT_EQ(BufData.size(), 40000U);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
164 EXPECT_EQ(BufData[0], '0');
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
165 EXPECT_EQ(BufData[9], '9');
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
166 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
167
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
168 TEST_F(MemoryBufferTest, getOpenFileNoReopen) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
169 testGetOpenFileSlice(false);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
170 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
171
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
172 TEST_F(MemoryBufferTest, getOpenFileReopened) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
173 testGetOpenFileSlice(true);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
174 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
175
|
100
|
176 TEST_F(MemoryBufferTest, reference) {
|
|
177 OwningBuffer MB(MemoryBuffer::getMemBuffer(data));
|
|
178 MemoryBufferRef MBR(*MB);
|
|
179
|
|
180 EXPECT_EQ(MB->getBufferStart(), MBR.getBufferStart());
|
|
181 EXPECT_EQ(MB->getBufferIdentifier(), MBR.getBufferIdentifier());
|
|
182 }
|
83
|
183
|
|
184 TEST_F(MemoryBufferTest, slice) {
|
|
185 // Create a file that is six pages long with different data on each page.
|
|
186 int FD;
|
|
187 SmallString<64> TestPath;
|
|
188 sys::fs::createTemporaryFile("MemoryBufferTest_Slice", "temp", FD, TestPath);
|
120
|
189 FileRemover Cleanup(TestPath);
|
83
|
190 raw_fd_ostream OF(FD, true, /*unbuffered=*/true);
|
|
191 for (unsigned i = 0; i < 0x2000 / 8; ++i) {
|
|
192 OF << "12345678";
|
|
193 }
|
|
194 for (unsigned i = 0; i < 0x2000 / 8; ++i) {
|
|
195 OF << "abcdefgh";
|
|
196 }
|
|
197 for (unsigned i = 0; i < 0x2000 / 8; ++i) {
|
|
198 OF << "ABCDEFGH";
|
|
199 }
|
|
200 OF.close();
|
|
201
|
|
202 // Try offset of one page.
|
|
203 ErrorOr<OwningBuffer> MB = MemoryBuffer::getFileSlice(TestPath.str(),
|
|
204 0x4000, 0x1000);
|
|
205 std::error_code EC = MB.getError();
|
|
206 ASSERT_FALSE(EC);
|
|
207 EXPECT_EQ(0x4000UL, MB.get()->getBufferSize());
|
|
208
|
|
209 StringRef BufData = MB.get()->getBuffer();
|
|
210 EXPECT_TRUE(BufData.substr(0x0000,8).equals("12345678"));
|
|
211 EXPECT_TRUE(BufData.substr(0x0FF8,8).equals("12345678"));
|
|
212 EXPECT_TRUE(BufData.substr(0x1000,8).equals("abcdefgh"));
|
|
213 EXPECT_TRUE(BufData.substr(0x2FF8,8).equals("abcdefgh"));
|
|
214 EXPECT_TRUE(BufData.substr(0x3000,8).equals("ABCDEFGH"));
|
|
215 EXPECT_TRUE(BufData.substr(0x3FF8,8).equals("ABCDEFGH"));
|
|
216
|
|
217 // Try non-page aligned.
|
|
218 ErrorOr<OwningBuffer> MB2 = MemoryBuffer::getFileSlice(TestPath.str(),
|
|
219 0x3000, 0x0800);
|
|
220 EC = MB2.getError();
|
|
221 ASSERT_FALSE(EC);
|
|
222 EXPECT_EQ(0x3000UL, MB2.get()->getBufferSize());
|
|
223
|
|
224 StringRef BufData2 = MB2.get()->getBuffer();
|
|
225 EXPECT_TRUE(BufData2.substr(0x0000,8).equals("12345678"));
|
|
226 EXPECT_TRUE(BufData2.substr(0x17F8,8).equals("12345678"));
|
|
227 EXPECT_TRUE(BufData2.substr(0x1800,8).equals("abcdefgh"));
|
|
228 EXPECT_TRUE(BufData2.substr(0x2FF8,8).equals("abcdefgh"));
|
0
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
229 }
|
134
|
230
|
|
231 TEST_F(MemoryBufferTest, writableSlice) {
|
|
232 // Create a file initialized with some data
|
|
233 int FD;
|
|
234 SmallString<64> TestPath;
|
|
235 sys::fs::createTemporaryFile("MemoryBufferTest_WritableSlice", "temp", FD,
|
|
236 TestPath);
|
|
237 FileRemover Cleanup(TestPath);
|
|
238 raw_fd_ostream OF(FD, true);
|
|
239 for (unsigned i = 0; i < 0x1000; ++i)
|
|
240 OF << "0123456789abcdef";
|
|
241 OF.close();
|
|
242
|
|
243 {
|
|
244 auto MBOrError =
|
|
245 WritableMemoryBuffer::getFileSlice(TestPath.str(), 0x6000, 0x2000);
|
|
246 ASSERT_FALSE(MBOrError.getError());
|
|
247 // Write some data. It should be mapped private, so that upon completion
|
|
248 // the original file contents are not modified.
|
|
249 WritableMemoryBuffer &MB = **MBOrError;
|
|
250 ASSERT_EQ(0x6000u, MB.getBufferSize());
|
|
251 char *Start = MB.getBufferStart();
|
|
252 ASSERT_EQ(MB.getBufferEnd(), MB.getBufferStart() + MB.getBufferSize());
|
|
253 ::memset(Start, 'x', MB.getBufferSize());
|
|
254 }
|
|
255
|
|
256 auto MBOrError = MemoryBuffer::getFile(TestPath);
|
|
257 ASSERT_FALSE(MBOrError.getError());
|
|
258 auto &MB = **MBOrError;
|
|
259 ASSERT_EQ(0x10000u, MB.getBufferSize());
|
|
260 for (size_t i = 0; i < MB.getBufferSize(); i += 0x10)
|
|
261 EXPECT_EQ("0123456789abcdef", MB.getBuffer().substr(i, 0x10)) << "i: " << i;
|
83
|
262 }
|
134
|
263 }
|