map_test.cpp
2.45 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
//===-- map_test.cpp --------------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "tests/scudo_unit_test.h"
#include "common.h"
#include <string.h>
#include <unistd.h>
static const char *MappingName = "scudo:test";
TEST(ScudoMapTest, PageSize) {
EXPECT_EQ(scudo::getPageSizeCached(),
static_cast<scudo::uptr>(getpagesize()));
}
TEST(ScudoMapTest, MapNoAccessUnmap) {
const scudo::uptr Size = 4 * scudo::getPageSizeCached();
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(ScudoMapTest, MapUnmap) {
const scudo::uptr Size = 4 * scudo::getPageSizeCached();
void *P = scudo::map(nullptr, Size, MappingName, 0, nullptr);
EXPECT_NE(P, nullptr);
memset(P, 0xaa, Size);
scudo::unmap(P, Size, 0, nullptr);
EXPECT_DEATH(memset(P, 0xbb, Size), "");
}
TEST(ScudoMapTest, MapWithGuardUnmap) {
const scudo::uptr PageSize = scudo::getPageSizeCached();
const scudo::uptr Size = 4 * PageSize;
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::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);
}