| /* |
| * Copyright (C) 2013 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "mem_map.h" |
| |
| #include <sys/mman.h> |
| |
| #include <memory> |
| #include <random> |
| |
| #include "base/common_art_test.h" |
| #include "common_runtime_test.h" // For TEST_DISABLED_FOR_MIPS |
| #include "memory_tool.h" |
| #include "unix_file/fd_file.h" |
| |
| namespace art { |
| |
| class MemMapTest : public CommonArtTest { |
| public: |
| static bool IsAddressMapped(void* addr) { |
| bool res = msync(addr, 1, MS_SYNC) == 0; |
| if (!res && errno != ENOMEM) { |
| PLOG(FATAL) << "Unexpected error occurred on msync"; |
| } |
| return res; |
| } |
| |
| static std::vector<uint8_t> RandomData(size_t size) { |
| std::random_device rd; |
| std::uniform_int_distribution<uint8_t> dist; |
| std::vector<uint8_t> res; |
| res.resize(size); |
| for (size_t i = 0; i < size; i++) { |
| res[i] = dist(rd); |
| } |
| return res; |
| } |
| |
| static uint8_t* GetValidMapAddress(size_t size, bool low_4gb) { |
| // Find a valid map address and unmap it before returning. |
| std::string error_msg; |
| MemMap map = MemMap::MapAnonymous("temp", |
| /* addr */ nullptr, |
| size, |
| PROT_READ, |
| low_4gb, |
| &error_msg); |
| CHECK(map.IsValid()); |
| return map.Begin(); |
| } |
| |
| static void RemapAtEndTest(bool low_4gb) { |
| std::string error_msg; |
| // Cast the page size to size_t. |
| const size_t page_size = static_cast<size_t>(kPageSize); |
| // Map a two-page memory region. |
| MemMap m0 = MemMap::MapAnonymous("MemMapTest_RemapAtEndTest_map0", |
| /* addr */ nullptr, |
| 2 * page_size, |
| PROT_READ | PROT_WRITE, |
| low_4gb, |
| &error_msg); |
| // Check its state and write to it. |
| ASSERT_TRUE(m0.IsValid()); |
| uint8_t* base0 = m0.Begin(); |
| ASSERT_TRUE(base0 != nullptr) << error_msg; |
| size_t size0 = m0.Size(); |
| EXPECT_EQ(m0.Size(), 2 * page_size); |
| EXPECT_EQ(m0.BaseBegin(), base0); |
| EXPECT_EQ(m0.BaseSize(), size0); |
| memset(base0, 42, 2 * page_size); |
| // Remap the latter half into a second MemMap. |
| MemMap m1 = m0.RemapAtEnd(base0 + page_size, |
| "MemMapTest_RemapAtEndTest_map1", |
| PROT_READ | PROT_WRITE, |
| &error_msg); |
| // Check the states of the two maps. |
| EXPECT_EQ(m0.Begin(), base0) << error_msg; |
| EXPECT_EQ(m0.Size(), page_size); |
| EXPECT_EQ(m0.BaseBegin(), base0); |
| EXPECT_EQ(m0.BaseSize(), page_size); |
| uint8_t* base1 = m1.Begin(); |
| size_t size1 = m1.Size(); |
| EXPECT_EQ(base1, base0 + page_size); |
| EXPECT_EQ(size1, page_size); |
| EXPECT_EQ(m1.BaseBegin(), base1); |
| EXPECT_EQ(m1.BaseSize(), size1); |
| // Write to the second region. |
| memset(base1, 43, page_size); |
| // Check the contents of the two regions. |
| for (size_t i = 0; i < page_size; ++i) { |
| EXPECT_EQ(base0[i], 42); |
| } |
| for (size_t i = 0; i < page_size; ++i) { |
| EXPECT_EQ(base1[i], 43); |
| } |
| // Unmap the first region. |
| m0.Reset(); |
| // Make sure the second region is still accessible after the first |
| // region is unmapped. |
| for (size_t i = 0; i < page_size; ++i) { |
| EXPECT_EQ(base1[i], 43); |
| } |
| MemMap m2 = m1.RemapAtEnd(m1.Begin(), |
| "MemMapTest_RemapAtEndTest_map1", |
| PROT_READ | PROT_WRITE, |
| &error_msg); |
| ASSERT_TRUE(m2.IsValid()) << error_msg; |
| ASSERT_FALSE(m1.IsValid()); |
| } |
| |
| void CommonInit() { |
| MemMap::Init(); |
| } |
| |
| #if defined(__LP64__) && !defined(__x86_64__) |
| static uintptr_t GetLinearScanPos() { |
| return MemMap::next_mem_pos_; |
| } |
| #endif |
| }; |
| |
| #if defined(__LP64__) && !defined(__x86_64__) |
| |
| #ifdef __BIONIC__ |
| extern uintptr_t CreateStartPos(uint64_t input); |
| #endif |
| |
| TEST_F(MemMapTest, Start) { |
| CommonInit(); |
| uintptr_t start = GetLinearScanPos(); |
| EXPECT_LE(64 * KB, start); |
| EXPECT_LT(start, static_cast<uintptr_t>(ART_BASE_ADDRESS)); |
| #ifdef __BIONIC__ |
| // Test a couple of values. Make sure they are different. |
| uintptr_t last = 0; |
| for (size_t i = 0; i < 100; ++i) { |
| uintptr_t random_start = CreateStartPos(i * kPageSize); |
| EXPECT_NE(last, random_start); |
| last = random_start; |
| } |
| |
| // Even on max, should be below ART_BASE_ADDRESS. |
| EXPECT_LT(CreateStartPos(~0), static_cast<uintptr_t>(ART_BASE_ADDRESS)); |
| #endif |
| // End of test. |
| } |
| #endif |
| |
| // We need mremap to be able to test ReplaceMapping at all |
| #if HAVE_MREMAP_SYSCALL |
| TEST_F(MemMapTest, ReplaceMapping_SameSize) { |
| std::string error_msg; |
| MemMap dest = MemMap::MapAnonymous("MapAnonymousEmpty-atomic-replace-dest", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(dest.IsValid()); |
| MemMap source = MemMap::MapAnonymous("MapAnonymous-atomic-replace-source", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_WRITE | PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(source.IsValid()); |
| void* source_addr = source.Begin(); |
| void* dest_addr = dest.Begin(); |
| ASSERT_TRUE(IsAddressMapped(source_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| |
| std::vector<uint8_t> data = RandomData(kPageSize); |
| memcpy(source.Begin(), data.data(), data.size()); |
| |
| ASSERT_TRUE(dest.ReplaceWith(&source, &error_msg)) << error_msg; |
| |
| ASSERT_FALSE(IsAddressMapped(source_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_FALSE(source.IsValid()); |
| |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(kPageSize)); |
| |
| ASSERT_EQ(memcmp(dest.Begin(), data.data(), dest.Size()), 0); |
| } |
| |
| TEST_F(MemMapTest, ReplaceMapping_MakeLarger) { |
| std::string error_msg; |
| MemMap dest = MemMap::MapAnonymous("MapAnonymousEmpty-atomic-replace-dest", |
| /* addr */ nullptr, |
| 5 * kPageSize, // Need to make it larger |
| // initially so we know |
| // there won't be mappings |
| // in the way we we move |
| // source. |
| PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(dest.IsValid()); |
| MemMap source = MemMap::MapAnonymous("MapAnonymous-atomic-replace-source", |
| /* addr */ nullptr, |
| 3 * kPageSize, |
| PROT_WRITE | PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(source.IsValid()); |
| uint8_t* source_addr = source.Begin(); |
| uint8_t* dest_addr = dest.Begin(); |
| ASSERT_TRUE(IsAddressMapped(source_addr)); |
| |
| // Fill the source with random data. |
| std::vector<uint8_t> data = RandomData(3 * kPageSize); |
| memcpy(source.Begin(), data.data(), data.size()); |
| |
| // Make the dest smaller so that we know we'll have space. |
| dest.SetSize(kPageSize); |
| |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_FALSE(IsAddressMapped(dest_addr + 2 * kPageSize)); |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(kPageSize)); |
| |
| ASSERT_TRUE(dest.ReplaceWith(&source, &error_msg)) << error_msg; |
| |
| ASSERT_FALSE(IsAddressMapped(source_addr)); |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(3 * kPageSize)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr + 2 * kPageSize)); |
| ASSERT_FALSE(source.IsValid()); |
| |
| ASSERT_EQ(memcmp(dest.Begin(), data.data(), dest.Size()), 0); |
| } |
| |
| TEST_F(MemMapTest, ReplaceMapping_MakeSmaller) { |
| std::string error_msg; |
| MemMap dest = MemMap::MapAnonymous("MapAnonymousEmpty-atomic-replace-dest", |
| /* addr */ nullptr, |
| 3 * kPageSize, |
| PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(dest.IsValid()); |
| MemMap source = MemMap::MapAnonymous("MapAnonymous-atomic-replace-source", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_WRITE | PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(source.IsValid()); |
| uint8_t* source_addr = source.Begin(); |
| uint8_t* dest_addr = dest.Begin(); |
| ASSERT_TRUE(IsAddressMapped(source_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr + 2 * kPageSize)); |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(3 * kPageSize)); |
| |
| std::vector<uint8_t> data = RandomData(kPageSize); |
| memcpy(source.Begin(), data.data(), kPageSize); |
| |
| ASSERT_TRUE(dest.ReplaceWith(&source, &error_msg)) << error_msg; |
| |
| ASSERT_FALSE(IsAddressMapped(source_addr)); |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(kPageSize)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_FALSE(IsAddressMapped(dest_addr + 2 * kPageSize)); |
| ASSERT_FALSE(source.IsValid()); |
| |
| ASSERT_EQ(memcmp(dest.Begin(), data.data(), dest.Size()), 0); |
| } |
| |
| TEST_F(MemMapTest, ReplaceMapping_FailureOverlap) { |
| std::string error_msg; |
| MemMap dest = |
| MemMap::MapAnonymous( |
| "MapAnonymousEmpty-atomic-replace-dest", |
| /* addr */ nullptr, |
| 3 * kPageSize, // Need to make it larger initially so we know there won't be mappings in |
| // the way we we move source. |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(dest.IsValid()); |
| // Resize down to 1 page so we can remap the rest. |
| dest.SetSize(kPageSize); |
| // Create source from the last 2 pages |
| MemMap source = MemMap::MapAnonymous("MapAnonymous-atomic-replace-source", |
| dest.Begin() + kPageSize, |
| 2 * kPageSize, |
| PROT_WRITE | PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(source.IsValid()); |
| ASSERT_EQ(dest.Begin() + kPageSize, source.Begin()); |
| uint8_t* source_addr = source.Begin(); |
| uint8_t* dest_addr = dest.Begin(); |
| ASSERT_TRUE(IsAddressMapped(source_addr)); |
| |
| // Fill the source and dest with random data. |
| std::vector<uint8_t> data = RandomData(2 * kPageSize); |
| memcpy(source.Begin(), data.data(), data.size()); |
| std::vector<uint8_t> dest_data = RandomData(kPageSize); |
| memcpy(dest.Begin(), dest_data.data(), dest_data.size()); |
| |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_EQ(dest.Size(), static_cast<size_t>(kPageSize)); |
| |
| ASSERT_FALSE(dest.ReplaceWith(&source, &error_msg)) << error_msg; |
| |
| ASSERT_TRUE(IsAddressMapped(source_addr)); |
| ASSERT_TRUE(IsAddressMapped(dest_addr)); |
| ASSERT_EQ(source.Size(), data.size()); |
| ASSERT_EQ(dest.Size(), dest_data.size()); |
| |
| ASSERT_EQ(memcmp(source.Begin(), data.data(), data.size()), 0); |
| ASSERT_EQ(memcmp(dest.Begin(), dest_data.data(), dest_data.size()), 0); |
| } |
| #endif // HAVE_MREMAP_SYSCALL |
| |
| TEST_F(MemMapTest, MapAnonymousEmpty) { |
| CommonInit(); |
| std::string error_msg; |
| MemMap map = MemMap::MapAnonymous("MapAnonymousEmpty", |
| /* addr */ nullptr, |
| 0, |
| PROT_READ, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_FALSE(map.IsValid()) << error_msg; |
| ASSERT_FALSE(error_msg.empty()); |
| |
| error_msg.clear(); |
| map = MemMap::MapAnonymous("MapAnonymousNonEmpty", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| } |
| |
| TEST_F(MemMapTest, MapAnonymousFailNullError) { |
| CommonInit(); |
| // Test that we don't crash with a null error_str when mapping at an invalid location. |
| MemMap map = MemMap::MapAnonymous("MapAnonymousInvalid", |
| reinterpret_cast<uint8_t*>(kPageSize), |
| 0x20000, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| nullptr); |
| ASSERT_FALSE(map.IsValid()); |
| } |
| |
| #ifdef __LP64__ |
| TEST_F(MemMapTest, MapAnonymousEmpty32bit) { |
| CommonInit(); |
| std::string error_msg; |
| MemMap map = MemMap::MapAnonymous("MapAnonymousEmpty", |
| /* addr */ nullptr, |
| 0, |
| PROT_READ, |
| /* low_4gb */ true, |
| &error_msg); |
| ASSERT_FALSE(map.IsValid()) << error_msg; |
| ASSERT_FALSE(error_msg.empty()); |
| |
| error_msg.clear(); |
| map = MemMap::MapAnonymous("MapAnonymousNonEmpty", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ true, |
| &error_msg); |
| ASSERT_TRUE(map.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_LT(reinterpret_cast<uintptr_t>(map.BaseBegin()), 1ULL << 32); |
| } |
| TEST_F(MemMapTest, MapFile32Bit) { |
| CommonInit(); |
| std::string error_msg; |
| ScratchFile scratch_file; |
| constexpr size_t kMapSize = kPageSize; |
| std::unique_ptr<uint8_t[]> data(new uint8_t[kMapSize]()); |
| ASSERT_TRUE(scratch_file.GetFile()->WriteFully(&data[0], kMapSize)); |
| MemMap map = MemMap::MapFile(/*byte_count*/kMapSize, |
| PROT_READ, |
| MAP_PRIVATE, |
| scratch_file.GetFd(), |
| /*start*/0, |
| /*low_4gb*/true, |
| scratch_file.GetFilename().c_str(), |
| &error_msg); |
| ASSERT_TRUE(map.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map.Size(), kMapSize); |
| ASSERT_LT(reinterpret_cast<uintptr_t>(map.BaseBegin()), 1ULL << 32); |
| } |
| #endif |
| |
| TEST_F(MemMapTest, MapAnonymousExactAddr) { |
| CommonInit(); |
| std::string error_msg; |
| // Find a valid address. |
| uint8_t* valid_address = GetValidMapAddress(kPageSize, /*low_4gb*/false); |
| // Map at an address that should work, which should succeed. |
| MemMap map0 = MemMap::MapAnonymous("MapAnonymous0", |
| valid_address, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map0.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_TRUE(map0.BaseBegin() == valid_address); |
| // Map at an unspecified address, which should succeed. |
| MemMap map1 = MemMap::MapAnonymous("MapAnonymous1", |
| /* addr */ nullptr, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map1.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_TRUE(map1.BaseBegin() != nullptr); |
| // Attempt to map at the same address, which should fail. |
| MemMap map2 = MemMap::MapAnonymous("MapAnonymous2", |
| reinterpret_cast<uint8_t*>(map1.BaseBegin()), |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_FALSE(map2.IsValid()) << error_msg; |
| ASSERT_TRUE(!error_msg.empty()); |
| } |
| |
| TEST_F(MemMapTest, RemapAtEnd) { |
| RemapAtEndTest(false); |
| } |
| |
| #ifdef __LP64__ |
| TEST_F(MemMapTest, RemapAtEnd32bit) { |
| RemapAtEndTest(true); |
| } |
| #endif |
| |
| TEST_F(MemMapTest, MapAnonymousExactAddr32bitHighAddr) { |
| // Some MIPS32 hardware (namely the Creator Ci20 development board) |
| // cannot allocate in the 2GB-4GB region. |
| TEST_DISABLED_FOR_MIPS(); |
| |
| // This test does not work under AddressSanitizer. |
| // Historical note: This test did not work under Valgrind either. |
| TEST_DISABLED_FOR_MEMORY_TOOL(); |
| |
| CommonInit(); |
| constexpr size_t size = 0x100000; |
| // Try all addresses starting from 2GB to 4GB. |
| size_t start_addr = 2 * GB; |
| std::string error_msg; |
| MemMap map; |
| for (; start_addr <= std::numeric_limits<uint32_t>::max() - size; start_addr += size) { |
| map = MemMap::MapAnonymous("MapAnonymousExactAddr32bitHighAddr", |
| reinterpret_cast<uint8_t*>(start_addr), |
| size, |
| PROT_READ | PROT_WRITE, |
| /*low_4gb*/ true, |
| &error_msg); |
| if (map.IsValid()) { |
| break; |
| } |
| } |
| ASSERT_TRUE(map.IsValid()) << error_msg; |
| ASSERT_GE(reinterpret_cast<uintptr_t>(map.End()), 2u * GB); |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map.BaseBegin(), reinterpret_cast<void*>(start_addr)); |
| } |
| |
| TEST_F(MemMapTest, MapAnonymousOverflow) { |
| CommonInit(); |
| std::string error_msg; |
| uintptr_t ptr = 0; |
| ptr -= kPageSize; // Now it's close to the top. |
| MemMap map = MemMap::MapAnonymous("MapAnonymousOverflow", |
| reinterpret_cast<uint8_t*>(ptr), |
| 2 * kPageSize, // brings it over the top. |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_FALSE(map.IsValid()); |
| ASSERT_FALSE(error_msg.empty()); |
| } |
| |
| #ifdef __LP64__ |
| TEST_F(MemMapTest, MapAnonymousLow4GBExpectedTooHigh) { |
| CommonInit(); |
| std::string error_msg; |
| MemMap map = |
| MemMap::MapAnonymous("MapAnonymousLow4GBExpectedTooHigh", |
| reinterpret_cast<uint8_t*>(UINT64_C(0x100000000)), |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ true, |
| &error_msg); |
| ASSERT_FALSE(map.IsValid()); |
| ASSERT_FALSE(error_msg.empty()); |
| } |
| |
| TEST_F(MemMapTest, MapAnonymousLow4GBRangeTooHigh) { |
| CommonInit(); |
| std::string error_msg; |
| MemMap map = MemMap::MapAnonymous("MapAnonymousLow4GBRangeTooHigh", |
| reinterpret_cast<uint8_t*>(0xF0000000), |
| 0x20000000, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ true, |
| &error_msg); |
| ASSERT_FALSE(map.IsValid()); |
| ASSERT_FALSE(error_msg.empty()); |
| } |
| #endif |
| |
| TEST_F(MemMapTest, MapAnonymousReuse) { |
| CommonInit(); |
| std::string error_msg; |
| MemMap map = MemMap::MapAnonymous("MapAnonymousReserve", |
| nullptr, |
| 0x20000, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| /* reuse */ false, |
| /* reservation */ nullptr, |
| &error_msg); |
| ASSERT_TRUE(map.IsValid()); |
| ASSERT_TRUE(error_msg.empty()); |
| MemMap map2 = MemMap::MapAnonymous("MapAnonymousReused", |
| reinterpret_cast<uint8_t*>(map.BaseBegin()), |
| 0x10000, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| /* reuse */ true, |
| /* reservation */ nullptr, |
| &error_msg); |
| ASSERT_TRUE(map2.IsValid()); |
| ASSERT_TRUE(error_msg.empty()); |
| } |
| |
| TEST_F(MemMapTest, CheckNoGaps) { |
| CommonInit(); |
| std::string error_msg; |
| constexpr size_t kNumPages = 3; |
| // Map a 3-page mem map. |
| MemMap map = MemMap::MapAnonymous("MapAnonymous0", |
| /* addr */ nullptr, |
| kPageSize * kNumPages, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| // Record the base address. |
| uint8_t* map_base = reinterpret_cast<uint8_t*>(map.BaseBegin()); |
| // Unmap it. |
| map.Reset(); |
| |
| // Map at the same address, but in page-sized separate mem maps, |
| // assuming the space at the address is still available. |
| MemMap map0 = MemMap::MapAnonymous("MapAnonymous0", |
| map_base, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map0.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| MemMap map1 = MemMap::MapAnonymous("MapAnonymous1", |
| map_base + kPageSize, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map1.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| MemMap map2 = MemMap::MapAnonymous("MapAnonymous2", |
| map_base + kPageSize * 2, |
| kPageSize, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(map2.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| |
| // One-map cases. |
| ASSERT_TRUE(MemMap::CheckNoGaps(map0, map0)); |
| ASSERT_TRUE(MemMap::CheckNoGaps(map1, map1)); |
| ASSERT_TRUE(MemMap::CheckNoGaps(map2, map2)); |
| |
| // Two or three-map cases. |
| ASSERT_TRUE(MemMap::CheckNoGaps(map0, map1)); |
| ASSERT_TRUE(MemMap::CheckNoGaps(map1, map2)); |
| ASSERT_TRUE(MemMap::CheckNoGaps(map0, map2)); |
| |
| // Unmap the middle one. |
| map1.Reset(); |
| |
| // Should return false now that there's a gap in the middle. |
| ASSERT_FALSE(MemMap::CheckNoGaps(map0, map2)); |
| } |
| |
| TEST_F(MemMapTest, AlignBy) { |
| CommonInit(); |
| std::string error_msg; |
| // Cast the page size to size_t. |
| const size_t page_size = static_cast<size_t>(kPageSize); |
| // Map a region. |
| MemMap m0 = MemMap::MapAnonymous("MemMapTest_AlignByTest_map0", |
| /* addr */ nullptr, |
| 14 * page_size, |
| PROT_READ | PROT_WRITE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(m0.IsValid()); |
| uint8_t* base0 = m0.Begin(); |
| ASSERT_TRUE(base0 != nullptr) << error_msg; |
| ASSERT_EQ(m0.Size(), 14 * page_size); |
| ASSERT_EQ(m0.BaseBegin(), base0); |
| ASSERT_EQ(m0.BaseSize(), m0.Size()); |
| |
| // Break it into several regions by using RemapAtEnd. |
| MemMap m1 = m0.RemapAtEnd(base0 + 3 * page_size, |
| "MemMapTest_AlignByTest_map1", |
| PROT_READ | PROT_WRITE, |
| &error_msg); |
| uint8_t* base1 = m1.Begin(); |
| ASSERT_TRUE(base1 != nullptr) << error_msg; |
| ASSERT_EQ(base1, base0 + 3 * page_size); |
| ASSERT_EQ(m0.Size(), 3 * page_size); |
| |
| MemMap m2 = m1.RemapAtEnd(base1 + 4 * page_size, |
| "MemMapTest_AlignByTest_map2", |
| PROT_READ | PROT_WRITE, |
| &error_msg); |
| uint8_t* base2 = m2.Begin(); |
| ASSERT_TRUE(base2 != nullptr) << error_msg; |
| ASSERT_EQ(base2, base1 + 4 * page_size); |
| ASSERT_EQ(m1.Size(), 4 * page_size); |
| |
| MemMap m3 = m2.RemapAtEnd(base2 + 3 * page_size, |
| "MemMapTest_AlignByTest_map1", |
| PROT_READ | PROT_WRITE, |
| &error_msg); |
| uint8_t* base3 = m3.Begin(); |
| ASSERT_TRUE(base3 != nullptr) << error_msg; |
| ASSERT_EQ(base3, base2 + 3 * page_size); |
| ASSERT_EQ(m2.Size(), 3 * page_size); |
| ASSERT_EQ(m3.Size(), 4 * page_size); |
| |
| uint8_t* end0 = base0 + m0.Size(); |
| uint8_t* end1 = base1 + m1.Size(); |
| uint8_t* end2 = base2 + m2.Size(); |
| uint8_t* end3 = base3 + m3.Size(); |
| |
| ASSERT_EQ(static_cast<size_t>(end3 - base0), 14 * page_size); |
| |
| if (IsAlignedParam(base0, 2 * page_size)) { |
| ASSERT_FALSE(IsAlignedParam(base1, 2 * page_size)); |
| ASSERT_FALSE(IsAlignedParam(base2, 2 * page_size)); |
| ASSERT_TRUE(IsAlignedParam(base3, 2 * page_size)); |
| ASSERT_TRUE(IsAlignedParam(end3, 2 * page_size)); |
| } else { |
| ASSERT_TRUE(IsAlignedParam(base1, 2 * page_size)); |
| ASSERT_TRUE(IsAlignedParam(base2, 2 * page_size)); |
| ASSERT_FALSE(IsAlignedParam(base3, 2 * page_size)); |
| ASSERT_FALSE(IsAlignedParam(end3, 2 * page_size)); |
| } |
| |
| // Align by 2 * page_size; |
| m0.AlignBy(2 * page_size); |
| m1.AlignBy(2 * page_size); |
| m2.AlignBy(2 * page_size); |
| m3.AlignBy(2 * page_size); |
| |
| EXPECT_TRUE(IsAlignedParam(m0.Begin(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m1.Begin(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m2.Begin(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m3.Begin(), 2 * page_size)); |
| |
| EXPECT_TRUE(IsAlignedParam(m0.Begin() + m0.Size(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m1.Begin() + m1.Size(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m2.Begin() + m2.Size(), 2 * page_size)); |
| EXPECT_TRUE(IsAlignedParam(m3.Begin() + m3.Size(), 2 * page_size)); |
| |
| if (IsAlignedParam(base0, 2 * page_size)) { |
| EXPECT_EQ(m0.Begin(), base0); |
| EXPECT_EQ(m0.Begin() + m0.Size(), end0 - page_size); |
| EXPECT_EQ(m1.Begin(), base1 + page_size); |
| EXPECT_EQ(m1.Begin() + m1.Size(), end1 - page_size); |
| EXPECT_EQ(m2.Begin(), base2 + page_size); |
| EXPECT_EQ(m2.Begin() + m2.Size(), end2); |
| EXPECT_EQ(m3.Begin(), base3); |
| EXPECT_EQ(m3.Begin() + m3.Size(), end3); |
| } else { |
| EXPECT_EQ(m0.Begin(), base0 + page_size); |
| EXPECT_EQ(m0.Begin() + m0.Size(), end0); |
| EXPECT_EQ(m1.Begin(), base1); |
| EXPECT_EQ(m1.Begin() + m1.Size(), end1); |
| EXPECT_EQ(m2.Begin(), base2); |
| EXPECT_EQ(m2.Begin() + m2.Size(), end2 - page_size); |
| EXPECT_EQ(m3.Begin(), base3 + page_size); |
| EXPECT_EQ(m3.Begin() + m3.Size(), end3 - page_size); |
| } |
| } |
| |
| TEST_F(MemMapTest, Reservation) { |
| CommonInit(); |
| std::string error_msg; |
| ScratchFile scratch_file; |
| constexpr size_t kMapSize = 5 * kPageSize; |
| std::unique_ptr<uint8_t[]> data(new uint8_t[kMapSize]()); |
| ASSERT_TRUE(scratch_file.GetFile()->WriteFully(&data[0], kMapSize)); |
| |
| MemMap reservation = MemMap::MapAnonymous("Test reservation", |
| /* addr */ nullptr, |
| kMapSize, |
| PROT_NONE, |
| /* low_4gb */ false, |
| &error_msg); |
| ASSERT_TRUE(reservation.IsValid()); |
| ASSERT_TRUE(error_msg.empty()); |
| |
| // Map first part of the reservation. |
| constexpr size_t kChunk1Size = kPageSize - 1u; |
| static_assert(kChunk1Size < kMapSize, "We want to split the reservation."); |
| uint8_t* addr1 = reservation.Begin(); |
| MemMap map1 = MemMap::MapFileAtAddress(addr1, |
| /* byte_count */ kChunk1Size, |
| PROT_READ, |
| MAP_PRIVATE, |
| scratch_file.GetFd(), |
| /* start */ 0, |
| /* low_4gb */ false, |
| scratch_file.GetFilename().c_str(), |
| /* reuse */ false, |
| &reservation, |
| &error_msg); |
| ASSERT_TRUE(map1.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map1.Size(), kChunk1Size); |
| ASSERT_EQ(addr1, map1.Begin()); |
| ASSERT_TRUE(reservation.IsValid()); |
| // Entire pages are taken from the `reservation`. |
| ASSERT_LT(map1.End(), map1.BaseEnd()); |
| ASSERT_EQ(map1.BaseEnd(), reservation.Begin()); |
| |
| // Map second part as an anonymous mapping. |
| constexpr size_t kChunk2Size = 2 * kPageSize; |
| DCHECK_LT(kChunk2Size, reservation.Size()); // We want to split the reservation. |
| uint8_t* addr2 = reservation.Begin(); |
| MemMap map2 = MemMap::MapAnonymous("MiddleReservation", |
| addr2, |
| /* byte_count */ kChunk2Size, |
| PROT_READ, |
| /* low_4gb */ false, |
| /* reuse */ false, |
| &reservation, |
| &error_msg); |
| ASSERT_TRUE(map2.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map2.Size(), kChunk2Size); |
| ASSERT_EQ(addr2, map2.Begin()); |
| ASSERT_EQ(map2.End(), map2.BaseEnd()); // kChunk2Size is page aligned. |
| ASSERT_EQ(map2.BaseEnd(), reservation.Begin()); |
| |
| // Map the rest of the reservation except the last byte. |
| const size_t kChunk3Size = reservation.Size() - 1u; |
| uint8_t* addr3 = reservation.Begin(); |
| MemMap map3 = MemMap::MapFileAtAddress(addr3, |
| /* byte_count */ kChunk3Size, |
| PROT_READ, |
| MAP_PRIVATE, |
| scratch_file.GetFd(), |
| /* start */ dchecked_integral_cast<size_t>(addr3 - addr1), |
| /* low_4gb */ false, |
| scratch_file.GetFilename().c_str(), |
| /* reuse */ false, |
| &reservation, |
| &error_msg); |
| ASSERT_TRUE(map3.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map3.Size(), kChunk3Size); |
| ASSERT_EQ(addr3, map3.Begin()); |
| // Entire pages are taken from the `reservation`, so it's now exhausted. |
| ASSERT_FALSE(reservation.IsValid()); |
| |
| // Now split the MiddleReservation. |
| constexpr size_t kChunk2ASize = kPageSize - 1u; |
| DCHECK_LT(kChunk2ASize, map2.Size()); // We want to split the reservation. |
| MemMap map2a = map2.TakeReservedMemory(kChunk2ASize); |
| ASSERT_TRUE(map2a.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map2a.Size(), kChunk2ASize); |
| ASSERT_EQ(addr2, map2a.Begin()); |
| ASSERT_TRUE(map2.IsValid()); |
| ASSERT_LT(map2a.End(), map2a.BaseEnd()); |
| ASSERT_EQ(map2a.BaseEnd(), map2.Begin()); |
| |
| // And take the rest of the middle reservation. |
| const size_t kChunk2BSize = map2.Size() - 1u; |
| uint8_t* addr2b = map2.Begin(); |
| MemMap map2b = map2.TakeReservedMemory(kChunk2BSize); |
| ASSERT_TRUE(map2b.IsValid()) << error_msg; |
| ASSERT_TRUE(error_msg.empty()); |
| ASSERT_EQ(map2b.Size(), kChunk2ASize); |
| ASSERT_EQ(addr2b, map2b.Begin()); |
| ASSERT_FALSE(map2.IsValid()); |
| } |
| |
| } // namespace art |