xref: /openbsd-src/gnu/llvm/compiler-rt/lib/scudo/standalone/tests/map_test.cpp (revision 810390e339a5425391477d5d41c78d7cab2424ac)
1 //===-- map_test.cpp --------------------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "tests/scudo_unit_test.h"
10 
11 #include "common.h"
12 
13 #include <string.h>
14 #include <unistd.h>
15 
16 static const char *MappingName = "scudo:test";
17 
TEST(ScudoMapTest,PageSize)18 TEST(ScudoMapTest, PageSize) {
19   EXPECT_EQ(scudo::getPageSizeCached(),
20             static_cast<scudo::uptr>(sysconf(_SC_PAGESIZE)));
21 }
22 
TEST(ScudoMapDeathTest,MapNoAccessUnmap)23 TEST(ScudoMapDeathTest, MapNoAccessUnmap) {
24   const scudo::uptr Size = 4 * scudo::getPageSizeCached();
25   scudo::MapPlatformData Data = {};
26   void *P = scudo::map(nullptr, Size, MappingName, MAP_NOACCESS, &Data);
27   EXPECT_NE(P, nullptr);
28   EXPECT_DEATH(memset(P, 0xaa, Size), "");
29   scudo::unmap(P, Size, UNMAP_ALL, &Data);
30 }
31 
TEST(ScudoMapDeathTest,MapUnmap)32 TEST(ScudoMapDeathTest, MapUnmap) {
33   const scudo::uptr Size = 4 * scudo::getPageSizeCached();
34   EXPECT_DEATH(
35       {
36         // Repeat few time to avoid missing crash if it's mmaped by unrelated
37         // code.
38         for (int i = 0; i < 10; ++i) {
39           void *P = scudo::map(nullptr, Size, MappingName, 0, nullptr);
40           if (!P)
41             continue;
42           scudo::unmap(P, Size, 0, nullptr);
43           memset(P, 0xbb, Size);
44         }
45       },
46       "");
47 }
48 
TEST(ScudoMapDeathTest,MapWithGuardUnmap)49 TEST(ScudoMapDeathTest, MapWithGuardUnmap) {
50   const scudo::uptr PageSize = scudo::getPageSizeCached();
51   const scudo::uptr Size = 4 * PageSize;
52   scudo::MapPlatformData Data = {};
53   void *P = scudo::map(nullptr, Size + 2 * PageSize, MappingName, MAP_NOACCESS,
54                        &Data);
55   EXPECT_NE(P, nullptr);
56   void *Q =
57       reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(P) + PageSize);
58   EXPECT_EQ(scudo::map(Q, Size, MappingName, 0, &Data), Q);
59   memset(Q, 0xaa, Size);
60   EXPECT_DEATH(memset(Q, 0xaa, Size + 1), "");
61   scudo::unmap(P, Size + 2 * PageSize, UNMAP_ALL, &Data);
62 }
63 
TEST(ScudoMapTest,MapGrowUnmap)64 TEST(ScudoMapTest, MapGrowUnmap) {
65   const scudo::uptr PageSize = scudo::getPageSizeCached();
66   const scudo::uptr Size = 4 * PageSize;
67   scudo::MapPlatformData Data = {};
68   void *P = scudo::map(nullptr, Size, MappingName, MAP_NOACCESS, &Data);
69   EXPECT_NE(P, nullptr);
70   void *Q =
71       reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(P) + PageSize);
72   EXPECT_EQ(scudo::map(Q, PageSize, MappingName, 0, &Data), Q);
73   memset(Q, 0xaa, PageSize);
74   Q = reinterpret_cast<void *>(reinterpret_cast<scudo::uptr>(Q) + PageSize);
75   EXPECT_EQ(scudo::map(Q, PageSize, MappingName, 0, &Data), Q);
76   memset(Q, 0xbb, PageSize);
77   scudo::unmap(P, Size, UNMAP_ALL, &Data);
78 }
79