1 //===-- bytemap_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 "bytemap.h"
12 
13 #include <pthread.h>
14 #include <string.h>
15 
16 template <typename T> void testMap(T &Map, scudo::uptr Size) {
17   Map.init();
18   for (scudo::uptr I = 0; I < Size; I += 7)
19     Map.set(I, (I % 100) + 1);
20   for (scudo::uptr J = 0; J < Size; J++) {
21     if (J % 7)
22       EXPECT_EQ(Map[J], 0);
23     else
24       EXPECT_EQ(Map[J], (J % 100) + 1);
25   }
26 }
27 
28 TEST(ScudoByteMapTest, FlatByteMap) {
29   const scudo::uptr Size = 1U << 10;
30   scudo::FlatByteMap<Size> Map;
31   testMap(Map, Size);
32   Map.unmapTestOnly();
33 }
34 
35 TEST(ScudoByteMapTest, TwoLevelByteMap) {
36   const scudo::uptr Size1 = 1U << 6, Size2 = 1U << 12;
37   scudo::TwoLevelByteMap<Size1, Size2> Map;
38   testMap(Map, Size1 * Size2);
39   Map.unmapTestOnly();
40 }
41 
42 using TestByteMap = scudo::TwoLevelByteMap<1U << 12, 1U << 13>;
43 
44 struct TestByteMapParam {
45   TestByteMap *Map;
46   scudo::uptr Shard;
47   scudo::uptr NumberOfShards;
48 };
49 
50 void *populateByteMap(void *Param) {
51   TestByteMapParam *P = reinterpret_cast<TestByteMapParam *>(Param);
52   for (scudo::uptr I = P->Shard; I < P->Map->size(); I += P->NumberOfShards) {
53     scudo::u8 V = static_cast<scudo::u8>((I % 100) + 1);
54     P->Map->set(I, V);
55     EXPECT_EQ((*P->Map)[I], V);
56   }
57   return 0;
58 }
59 
60 TEST(ScudoByteMapTest, ThreadedTwoLevelByteMap) {
61   TestByteMap Map;
62   Map.init();
63   static const scudo::uptr NumberOfThreads = 16U;
64   pthread_t T[NumberOfThreads];
65   TestByteMapParam P[NumberOfThreads];
66   for (scudo::uptr I = 0; I < NumberOfThreads; I++) {
67     P[I].Map = &Map;
68     P[I].Shard = I;
69     P[I].NumberOfShards = NumberOfThreads;
70     pthread_create(&T[I], 0, populateByteMap, &P[I]);
71   }
72   for (scudo::uptr I = 0; I < NumberOfThreads; I++)
73     pthread_join(T[I], 0);
74   Map.unmapTestOnly();
75 }
76