#include "tests/scudo_unit_test.h"
#include "common.h"
-#include "mem_map.h"
#include <algorithm>
#include <fstream>
const uptr Size = 1ull << 30;
const uptr Threshold = Size >> 3;
- MemMapT MemMap;
- ASSERT_TRUE(MemMap.map(/*Addr=*/0U, Size, "ResidentMemorySize"));
- void *P = reinterpret_cast<void *>(MemMap.getBase());
+ MapPlatformData Data = {};
+ void *P = map(nullptr, Size, "ResidentMemorySize", 0, &Data);
+ ASSERT_NE(nullptr, P);
EXPECT_LT(getResidentMemorySize(), OnStart + Threshold);
memset(P, 1, Size);
EXPECT_GT(getResidentMemorySize(), OnStart + Size - Threshold);
- MemMap.releasePagesToOS(MemMap.getBase(), Size);
+ releasePagesToOS((uptr)P, 0, Size, &Data);
EXPECT_LT(getResidentMemorySize(), OnStart + Threshold);
memset(P, 1, Size);
EXPECT_GT(getResidentMemorySize(), OnStart + Size - Threshold);
- MemMap.unmap(MemMap.getBase(), Size);
+ unmap(P, Size, 0, &Data);
}
TEST(ScudoCommonTest, Zeros) {
const uptr Size = 1ull << 20;
- MemMapT MemMap;
- ASSERT_TRUE(MemMap.map(/*Addr=*/0U, Size, "Zeros"));
- uptr *P = reinterpret_cast<uptr *>(MemMap.getBase());
- const ptrdiff_t N = Size / sizeof(uptr);
+ MapPlatformData Data = {};
+ uptr *P = reinterpret_cast<uptr *>(map(nullptr, Size, "Zeros", 0, &Data));
+ const ptrdiff_t N = Size / sizeof(*P);
+ ASSERT_NE(nullptr, P);
EXPECT_EQ(std::count(P, P + N, 0), N);
memset(P, 1, Size);
EXPECT_EQ(std::count(P, P + N, 0), 0);
- MemMap.releasePagesToOS(MemMap.getBase(), Size);
+ releasePagesToOS((uptr)P, 0, Size, &Data);
EXPECT_EQ(std::count(P, P + N, 0), N);
- MemMap.unmap(MemMap.getBase(), Size);
+ unmap(P, Size, 0, &Data);
}
#if 0
#include "tests/scudo_unit_test.h"
#include "common.h"
-#include "mem_map.h"
#include <string.h>
#include <unistd.h>
TEST(ScudoMapDeathTest, MapNoAccessUnmap) {
const scudo::uptr Size = 4 * scudo::getPageSizeCached();
- scudo::ReservedMemoryT ReservedMemory;
-
- ASSERT_TRUE(ReservedMemory.create(/*Addr=*/0U, Size, MappingName));
- EXPECT_NE(ReservedMemory.getBase(), 0U);
- EXPECT_DEATH(
- memset(reinterpret_cast<void *>(ReservedMemory.getBase()), 0xaa, Size),
- "");
-
- ReservedMemory.release();
+ scudo::MapPlatformData Data = {};
+ void *P = scudo::map(nullptr, Size, MappingName, MAP_NOACCESS, &Data);
+ EXPECT_NE(P, nullptr);
+ EXPECT_DEATH(memset(P, 0xaa, Size), "");
+ scudo::unmap(P, Size, UNMAP_ALL, &Data);
}
TEST(ScudoMapDeathTest, MapUnmap) {
// Repeat few time to avoid missing crash if it's mmaped by unrelated
// code.
for (int i = 0; i < 10; ++i) {
- scudo::MemMapT MemMap;
- MemMap.map(/*Addr=*/0U, Size, MappingName);
- scudo::uptr P = MemMap.getBase();
- if (P == 0U)
+ void *P = scudo::map(nullptr, Size, MappingName, 0, nullptr);
+ if (!P)
continue;
- MemMap.unmap(MemMap.getBase(), Size);
- memset(reinterpret_cast<void *>(P), 0xbb, Size);
+ scudo::unmap(P, Size, 0, nullptr);
+ memset(P, 0xbb, Size);
}
},
"");
TEST(ScudoMapDeathTest, MapWithGuardUnmap) {
const scudo::uptr PageSize = scudo::getPageSizeCached();
const scudo::uptr Size = 4 * PageSize;
- scudo::ReservedMemoryT ReservedMemory;
- ASSERT_TRUE(
- ReservedMemory.create(/*Addr=*/0U, Size + 2 * PageSize, MappingName));
- ASSERT_NE(ReservedMemory.getBase(), 0U);
-
- scudo::MemMapT MemMap =
- ReservedMemory.dispatch(ReservedMemory.getBase(), Size + 2 * PageSize);
- ASSERT_TRUE(MemMap.isAllocated());
- scudo::uptr Q = MemMap.getBase() + PageSize;
- ASSERT_TRUE(MemMap.remap(Q, Size, MappingName));
- memset(reinterpret_cast<void *>(Q), 0xaa, Size);
- EXPECT_DEATH(memset(reinterpret_cast<void *>(Q), 0xaa, Size + 1), "");
- MemMap.unmap(MemMap.getBase(), MemMap.getCapacity());
+ scudo::MapPlatformData Data = {};
+ void *P = scudo::map(nullptr, Size + 2 * PageSize, MappingName, MAP_NOACCESS,
+ &Data);
+ EXPECT_NE(P, nullptr);
+ void *Q =
+ reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(P) + PageSize);
+ EXPECT_EQ(scudo::map(Q, Size, MappingName, 0, &Data), Q);
+ memset(Q, 0xaa, Size);
+ EXPECT_DEATH(memset(Q, 0xaa, Size + 1), "");
+ scudo::unmap(P, Size + 2 * PageSize, UNMAP_ALL, &Data);
}
TEST(ScudoMapTest, MapGrowUnmap) {
const scudo::uptr PageSize = scudo::getPageSizeCached();
const scudo::uptr Size = 4 * PageSize;
- scudo::ReservedMemoryT ReservedMemory;
- ReservedMemory.create(/*Addr=*/0U, Size, MappingName);
- ASSERT_TRUE(ReservedMemory.isCreated());
-
- scudo::MemMapT MemMap =
- ReservedMemory.dispatch(ReservedMemory.getBase(), Size);
- ASSERT_TRUE(MemMap.isAllocated());
- scudo::uptr Q = MemMap.getBase() + PageSize;
- ASSERT_TRUE(MemMap.remap(Q, PageSize, MappingName));
- memset(reinterpret_cast<void *>(Q), 0xaa, PageSize);
- Q += PageSize;
- ASSERT_TRUE(MemMap.remap(Q, PageSize, MappingName));
- memset(reinterpret_cast<void *>(Q), 0xbb, PageSize);
- MemMap.unmap(MemMap.getBase(), MemMap.getCapacity());
+ scudo::MapPlatformData Data = {};
+ void *P = scudo::map(nullptr, Size, MappingName, MAP_NOACCESS, &Data);
+ EXPECT_NE(P, nullptr);
+ void *Q =
+ reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(P) + PageSize);
+ EXPECT_EQ(scudo::map(Q, PageSize, MappingName, 0, &Data), Q);
+ memset(Q, 0xaa, PageSize);
+ Q = reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(Q) + PageSize);
+ EXPECT_EQ(scudo::map(Q, PageSize, MappingName, 0, &Data), Q);
+ memset(Q, 0xbb, PageSize);
+ scudo::unmap(P, Size, UNMAP_ALL, &Data);
}
//===----------------------------------------------------------------------===//
#include "common.h"
-#include "mem_map.h"
#include "memtag.h"
#include "platform.h"
#include "tests/scudo_unit_test.h"
GTEST_SKIP() << "Memory tagging is not supported";
BufferSize = getPageSizeCached();
- scudo::MemMapT MemMap;
- ASSERT_TRUE(MemMap.map(/*Addr=*/0U, BufferSize, "MemtagTest", MAP_MEMTAG));
- Addr = MemMap.getBase();
- Buffer = reinterpret_cast<u8 *>(Addr);
+ Buffer = reinterpret_cast<u8 *>(
+ map(nullptr, BufferSize, "MemtagTest", MAP_MEMTAG, &Data));
+ Addr = reinterpret_cast<uptr>(Buffer);
EXPECT_TRUE(isAligned(Addr, archMemoryTagGranuleSize()));
EXPECT_EQ(Addr, untagPointer(Addr));
}
void TearDown() override {
- if (Buffer) {
- ASSERT_TRUE(MemMap.isAllocated());
- MemMap.unmap(MemMap.getBase(), MemMap.getCapacity());
- }
+ if (Buffer)
+ unmap(Buffer, BufferSize, 0, &Data);
}
uptr BufferSize = 0;
- scudo::MemMapT MemMap;
+ MapPlatformData Data = {};
u8 *Buffer = nullptr;
uptr Addr = 0;
};
EXPECT_EQ(LoadPtr, loadTag(LoadPtr));
// Reset tags without using StoreTags.
- MemMap.releasePagesToOS(Addr, BufferSize);
+ releasePagesToOS(Addr, 0, BufferSize, &Data);
}
}