OLD | NEW |
| (Empty) |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "platform/heap/HeapCompact.h" | |
6 | |
7 #include "platform/heap/Handle.h" | |
8 #include "platform/heap/SparseHeapBitmap.h" | |
9 #include "testing/gtest/include/gtest/gtest.h" | |
10 #include "wtf/Deque.h" | |
11 #include "wtf/HashMap.h" | |
12 #include "wtf/LinkedHashSet.h" | |
13 #include "wtf/Vector.h" | |
14 | |
15 #include <memory> | |
16 | |
17 namespace blink { | |
18 class IntWrapper; | |
19 } | |
20 | |
21 using IntVector = blink::HeapVector<blink::Member<blink::IntWrapper>>; | |
22 using IntDeque = blink::HeapDeque<blink::Member<blink::IntWrapper>>; | |
23 using IntMap = blink::HeapHashMap<blink::Member<blink::IntWrapper>, int>; | |
24 // TODO(sof): decide if this ought to be a global trait specialization. | |
25 // (i.e., for HeapHash*<T>.) | |
26 WTF_ALLOW_CLEAR_UNUSED_SLOTS_WITH_MEM_FUNCTIONS(IntMap); | |
27 | |
28 namespace blink { | |
29 | |
30 static const size_t chunkRange = SparseHeapBitmap::s_bitmapChunkRange; | |
31 static const size_t unitPointer = 0x1u | |
32 << SparseHeapBitmap::s_pointerAlignmentInBits; | |
33 | |
34 TEST(HeapCompactTest, SparseBitmapBasic) { | |
35 Address base = reinterpret_cast<Address>(0x10000u); | |
36 std::unique_ptr<SparseHeapBitmap> bitmap = SparseHeapBitmap::create(base); | |
37 | |
38 size_t doubleChunk = 2 * chunkRange; | |
39 | |
40 // 101010... starting at |base|. | |
41 for (size_t i = 0; i < doubleChunk; i += 2 * unitPointer) | |
42 bitmap->add(base + i); | |
43 | |
44 // Check that hasRange() returns a bitmap subtree, if any, for a given | |
45 // address. | |
46 EXPECT_TRUE(!!bitmap->hasRange(base, 1)); | |
47 EXPECT_TRUE(!!bitmap->hasRange(base + unitPointer, 1)); | |
48 EXPECT_FALSE(!!bitmap->hasRange(base - unitPointer, 1)); | |
49 | |
50 // Test implementation details.. that each SparseHeapBitmap node maps | |
51 // |s_bitmapChunkRange| ranges only. | |
52 EXPECT_EQ(bitmap->hasRange(base + unitPointer, 1), | |
53 bitmap->hasRange(base + 2 * unitPointer, 1)); | |
54 // Second range will be just past the first. | |
55 EXPECT_NE(bitmap->hasRange(base, 1), bitmap->hasRange(base + chunkRange, 1)); | |
56 | |
57 // Iterate a range that will encompass more than one 'chunk'. | |
58 SparseHeapBitmap* start = | |
59 bitmap->hasRange(base + 2 * unitPointer, doubleChunk); | |
60 EXPECT_TRUE(!!start); | |
61 for (size_t i = 2 * unitPointer; i < doubleChunk; i += 2 * unitPointer) { | |
62 EXPECT_TRUE(start->isSet(base + i)); | |
63 EXPECT_FALSE(start->isSet(base + i + unitPointer)); | |
64 } | |
65 } | |
66 | |
67 TEST(HeapCompactTest, SparseBitmapBuild) { | |
68 Address base = reinterpret_cast<Address>(0x10000u); | |
69 std::unique_ptr<SparseHeapBitmap> bitmap = SparseHeapBitmap::create(base); | |
70 | |
71 size_t doubleChunk = 2 * chunkRange; | |
72 | |
73 // Create a sparse bitmap containing at least three chunks. | |
74 bitmap->add(base - doubleChunk); | |
75 bitmap->add(base + doubleChunk); | |
76 | |
77 // This is sanity testing internal implementation details of | |
78 // SparseHeapBitmap; probing |isSet()| outside the bitmap | |
79 // of the range used in |hasRange()|, is not defined. | |
80 // | |
81 // Regardless, the testing here verifies that a |hasRange()| that | |
82 // straddles multiple internal nodes, returns a bitmap that is | |
83 // capable of returning correct |isSet()| results. | |
84 SparseHeapBitmap* start = | |
85 bitmap->hasRange(base - doubleChunk - 2 * unitPointer, 4 * unitPointer); | |
86 EXPECT_TRUE(!!start); | |
87 EXPECT_TRUE(start->isSet(base - doubleChunk)); | |
88 EXPECT_FALSE(start->isSet(base - doubleChunk + unitPointer)); | |
89 EXPECT_FALSE(start->isSet(base)); | |
90 EXPECT_FALSE(start->isSet(base + unitPointer)); | |
91 EXPECT_FALSE(start->isSet(base + doubleChunk)); | |
92 EXPECT_FALSE(start->isSet(base + doubleChunk + unitPointer)); | |
93 | |
94 start = bitmap->hasRange(base - doubleChunk - 2 * unitPointer, | |
95 2 * doubleChunk + 2 * unitPointer); | |
96 EXPECT_TRUE(!!start); | |
97 EXPECT_TRUE(start->isSet(base - doubleChunk)); | |
98 EXPECT_FALSE(start->isSet(base - doubleChunk + unitPointer)); | |
99 EXPECT_TRUE(start->isSet(base)); | |
100 EXPECT_FALSE(start->isSet(base + unitPointer)); | |
101 EXPECT_TRUE(start->isSet(base + doubleChunk)); | |
102 EXPECT_FALSE(start->isSet(base + doubleChunk + unitPointer)); | |
103 | |
104 start = bitmap->hasRange(base, 20); | |
105 EXPECT_TRUE(!!start); | |
106 // Probing for values outside of hasRange() should be considered | |
107 // undefined, but do it to exercise the (left) tree traversal. | |
108 EXPECT_TRUE(start->isSet(base - doubleChunk)); | |
109 EXPECT_FALSE(start->isSet(base - doubleChunk + unitPointer)); | |
110 EXPECT_TRUE(start->isSet(base)); | |
111 EXPECT_FALSE(start->isSet(base + unitPointer)); | |
112 EXPECT_TRUE(start->isSet(base + doubleChunk)); | |
113 EXPECT_FALSE(start->isSet(base + doubleChunk + unitPointer)); | |
114 | |
115 start = bitmap->hasRange(base + chunkRange + 2 * unitPointer, 2048); | |
116 EXPECT_TRUE(!!start); | |
117 // Probing for values outside of hasRange() should be considered | |
118 // undefined, but do it to exercise node traversal. | |
119 EXPECT_FALSE(start->isSet(base - doubleChunk)); | |
120 EXPECT_FALSE(start->isSet(base - doubleChunk + unitPointer)); | |
121 EXPECT_FALSE(start->isSet(base)); | |
122 EXPECT_FALSE(start->isSet(base + unitPointer)); | |
123 EXPECT_FALSE(start->isSet(base + chunkRange)); | |
124 EXPECT_TRUE(start->isSet(base + doubleChunk)); | |
125 EXPECT_FALSE(start->isSet(base + doubleChunk + unitPointer)); | |
126 } | |
127 | |
128 TEST(HeapCompactTest, SparseBitmapLeftExtension) { | |
129 Address base = reinterpret_cast<Address>(0x10000u); | |
130 std::unique_ptr<SparseHeapBitmap> bitmap = SparseHeapBitmap::create(base); | |
131 | |
132 SparseHeapBitmap* start = bitmap->hasRange(base, 1); | |
133 EXPECT_TRUE(start); | |
134 | |
135 // Verify that re-adding is a no-op. | |
136 bitmap->add(base); | |
137 EXPECT_EQ(start, bitmap->hasRange(base, 1)); | |
138 | |
139 // Adding an Address |A| before a single-address SparseHeapBitmap node should | |
140 // cause that node to be "left extended" to use |A| as its new base. | |
141 bitmap->add(base - 2 * unitPointer); | |
142 EXPECT_EQ(bitmap->hasRange(base, 1), | |
143 bitmap->hasRange(base - 2 * unitPointer, 1)); | |
144 | |
145 // Reset. | |
146 bitmap = SparseHeapBitmap::create(base); | |
147 | |
148 // If attempting same as above, but the Address |A| is outside the | |
149 // chunk size of a node, a new SparseHeapBitmap node needs to be | |
150 // created to the left of |bitmap|. | |
151 bitmap->add(base - chunkRange); | |
152 EXPECT_NE(bitmap->hasRange(base, 1), | |
153 bitmap->hasRange(base - 2 * unitPointer, 1)); | |
154 | |
155 bitmap = SparseHeapBitmap::create(base); | |
156 bitmap->add(base - chunkRange + unitPointer); | |
157 // This address is just inside the horizon and shouldn't create a new chunk. | |
158 EXPECT_EQ(bitmap->hasRange(base, 1), | |
159 bitmap->hasRange(base - 2 * unitPointer, 1)); | |
160 // ..but this one should, like for the sub-test above. | |
161 bitmap->add(base - chunkRange); | |
162 EXPECT_EQ(bitmap->hasRange(base, 1), | |
163 bitmap->hasRange(base - 2 * unitPointer, 1)); | |
164 EXPECT_NE(bitmap->hasRange(base, 1), bitmap->hasRange(base - chunkRange, 1)); | |
165 } | |
166 | |
167 static void preciselyCollectGarbage() { | |
168 ThreadState::current()->collectGarbage( | |
169 BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC); | |
170 } | |
171 | |
172 static void performHeapCompaction() { | |
173 EXPECT_FALSE(HeapCompact::scheduleCompactionGCForTesting(true)); | |
174 preciselyCollectGarbage(); | |
175 EXPECT_FALSE(HeapCompact::scheduleCompactionGCForTesting(false)); | |
176 } | |
177 | |
178 // Do several GCs to make sure that later GCs don't free up old memory from | |
179 // previously run tests in this process. | |
180 static void clearOutOldGarbage() { | |
181 ThreadHeap& heap = ThreadState::current()->heap(); | |
182 while (true) { | |
183 size_t used = heap.objectPayloadSizeForTesting(); | |
184 preciselyCollectGarbage(); | |
185 if (heap.objectPayloadSizeForTesting() >= used) | |
186 break; | |
187 } | |
188 } | |
189 | |
190 class IntWrapper : public GarbageCollectedFinalized<IntWrapper> { | |
191 public: | |
192 static IntWrapper* create(int x) { return new IntWrapper(x); } | |
193 | |
194 virtual ~IntWrapper() { ++s_destructorCalls; } | |
195 | |
196 static int s_destructorCalls; | |
197 DEFINE_INLINE_TRACE() {} | |
198 | |
199 int value() const { return m_x; } | |
200 | |
201 bool operator==(const IntWrapper& other) const { | |
202 return other.value() == value(); | |
203 } | |
204 | |
205 unsigned hash() { return IntHash<int>::hash(m_x); } | |
206 | |
207 IntWrapper(int x) : m_x(x) {} | |
208 | |
209 private: | |
210 IntWrapper(); | |
211 int m_x; | |
212 }; | |
213 static_assert(WTF::IsTraceable<IntWrapper>::value, | |
214 "IsTraceable<> template failed to recognize trace method."); | |
215 | |
216 TEST(HeapCompactTest, CompactVector) { | |
217 clearOutOldGarbage(); | |
218 | |
219 IntWrapper* val = IntWrapper::create(1); | |
220 Persistent<IntVector> vector = new IntVector(10, val); | |
221 EXPECT_EQ(10u, vector->size()); | |
222 | |
223 for (size_t i = 0; i < vector->size(); ++i) | |
224 EXPECT_EQ(val, (*vector)[i]); | |
225 | |
226 performHeapCompaction(); | |
227 | |
228 for (size_t i = 0; i < vector->size(); ++i) | |
229 EXPECT_EQ(val, (*vector)[i]); | |
230 } | |
231 | |
232 TEST(HeapCompactTest, CompactHashMap) { | |
233 clearOutOldGarbage(); | |
234 | |
235 Persistent<IntMap> intMap = new IntMap(); | |
236 for (size_t i = 0; i < 100; ++i) { | |
237 IntWrapper* val = IntWrapper::create(i); | |
238 intMap->add(val, 100 - i); | |
239 } | |
240 | |
241 EXPECT_EQ(100u, intMap->size()); | |
242 for (auto k : *intMap) | |
243 EXPECT_EQ(k.key->value(), 100 - k.value); | |
244 | |
245 performHeapCompaction(); | |
246 | |
247 for (auto k : *intMap) | |
248 EXPECT_EQ(k.key->value(), 100 - k.value); | |
249 } | |
250 | |
251 TEST(HeapCompactTest, CompactVectorPartHashMap) { | |
252 clearOutOldGarbage(); | |
253 | |
254 using IntMapVector = HeapVector<IntMap>; | |
255 | |
256 Persistent<IntMapVector> intMapVector = new IntMapVector(); | |
257 for (size_t i = 0; i < 10; ++i) { | |
258 IntMap map; | |
259 for (size_t j = 0; j < 10; ++j) { | |
260 IntWrapper* val = IntWrapper::create(j); | |
261 map.add(val, 10 - j); | |
262 } | |
263 intMapVector->append(map); | |
264 } | |
265 | |
266 EXPECT_EQ(10u, intMapVector->size()); | |
267 for (auto map : *intMapVector) { | |
268 EXPECT_EQ(10u, map.size()); | |
269 for (auto k : map) { | |
270 EXPECT_EQ(k.key->value(), 10 - k.value); | |
271 } | |
272 } | |
273 | |
274 performHeapCompaction(); | |
275 | |
276 EXPECT_EQ(10u, intMapVector->size()); | |
277 for (auto map : *intMapVector) { | |
278 EXPECT_EQ(10u, map.size()); | |
279 for (auto k : map) { | |
280 EXPECT_EQ(k.key->value(), 10 - k.value); | |
281 } | |
282 } | |
283 } | |
284 | |
285 TEST(HeapCompactTest, CompactHashPartVector) { | |
286 clearOutOldGarbage(); | |
287 | |
288 using IntVectorMap = HeapHashMap<int, IntVector>; | |
289 | |
290 Persistent<IntVectorMap> intVectorMap = new IntVectorMap(); | |
291 for (size_t i = 0; i < 10; ++i) { | |
292 IntVector vector; | |
293 for (size_t j = 0; j < 10; ++j) { | |
294 vector.append(IntWrapper::create(j)); | |
295 } | |
296 intVectorMap->add(1 + i, vector); | |
297 } | |
298 | |
299 EXPECT_EQ(10u, intVectorMap->size()); | |
300 for (const IntVector& intVector : intVectorMap->values()) { | |
301 EXPECT_EQ(10u, intVector.size()); | |
302 for (size_t i = 0; i < intVector.size(); ++i) { | |
303 EXPECT_EQ(static_cast<int>(i), intVector[i]->value()); | |
304 } | |
305 } | |
306 | |
307 performHeapCompaction(); | |
308 | |
309 EXPECT_EQ(10u, intVectorMap->size()); | |
310 for (const IntVector& intVector : intVectorMap->values()) { | |
311 EXPECT_EQ(10u, intVector.size()); | |
312 for (size_t i = 0; i < intVector.size(); ++i) { | |
313 EXPECT_EQ(static_cast<int>(i), intVector[i]->value()); | |
314 } | |
315 } | |
316 } | |
317 | |
318 TEST(HeapCompactTest, CompactDeques) { | |
319 Persistent<IntDeque> deque = new IntDeque; | |
320 for (int i = 0; i < 8; ++i) { | |
321 deque->prepend(IntWrapper::create(i)); | |
322 } | |
323 EXPECT_EQ(8u, deque->size()); | |
324 | |
325 for (size_t i = 0; i < deque->size(); ++i) | |
326 EXPECT_EQ(static_cast<int>(7 - i), deque->at(i)->value()); | |
327 | |
328 performHeapCompaction(); | |
329 | |
330 for (size_t i = 0; i < deque->size(); ++i) | |
331 EXPECT_EQ(static_cast<int>(7 - i), deque->at(i)->value()); | |
332 } | |
333 | |
334 TEST(HeapCompactTest, CompactDequeVectors) { | |
335 Persistent<HeapDeque<IntVector>> deque = new HeapDeque<IntVector>; | |
336 for (int i = 0; i < 8; ++i) { | |
337 IntWrapper* value = IntWrapper::create(i); | |
338 IntVector vector = IntVector(8, value); | |
339 deque->prepend(vector); | |
340 } | |
341 EXPECT_EQ(8u, deque->size()); | |
342 | |
343 for (size_t i = 0; i < deque->size(); ++i) | |
344 EXPECT_EQ(static_cast<int>(7 - i), deque->at(i).at(i)->value()); | |
345 | |
346 performHeapCompaction(); | |
347 | |
348 for (size_t i = 0; i < deque->size(); ++i) | |
349 EXPECT_EQ(static_cast<int>(7 - i), deque->at(i).at(i)->value()); | |
350 } | |
351 | |
352 TEST(HeapCompactTest, CompactLinkedHashSet) { | |
353 using OrderedHashSet = HeapLinkedHashSet<Member<IntWrapper>>; | |
354 Persistent<OrderedHashSet> set = new OrderedHashSet; | |
355 for (int i = 0; i < 13; ++i) { | |
356 IntWrapper* value = IntWrapper::create(i); | |
357 set->add(value); | |
358 } | |
359 EXPECT_EQ(13u, set->size()); | |
360 | |
361 int expected = 0; | |
362 for (IntWrapper* v : *set) { | |
363 EXPECT_EQ(expected, v->value()); | |
364 expected++; | |
365 } | |
366 | |
367 performHeapCompaction(); | |
368 | |
369 expected = 0; | |
370 for (IntWrapper* v : *set) { | |
371 EXPECT_EQ(expected, v->value()); | |
372 expected++; | |
373 } | |
374 } | |
375 | |
376 TEST(HeapCompactTest, CompactLinkedHashSetVector) { | |
377 using OrderedHashSet = HeapLinkedHashSet<Member<IntVector>>; | |
378 Persistent<OrderedHashSet> set = new OrderedHashSet; | |
379 for (int i = 0; i < 13; ++i) { | |
380 IntWrapper* value = IntWrapper::create(i); | |
381 IntVector* vector = new IntVector(19, value); | |
382 set->add(vector); | |
383 } | |
384 EXPECT_EQ(13u, set->size()); | |
385 | |
386 int expected = 0; | |
387 for (IntVector* v : *set) { | |
388 EXPECT_EQ(expected, (*v)[0]->value()); | |
389 expected++; | |
390 } | |
391 | |
392 performHeapCompaction(); | |
393 | |
394 expected = 0; | |
395 for (IntVector* v : *set) { | |
396 EXPECT_EQ(expected, (*v)[0]->value()); | |
397 expected++; | |
398 } | |
399 } | |
400 | |
401 TEST(HeapCompactTest, CompactLinkedHashSetMap) { | |
402 using Inner = HeapHashSet<Member<IntWrapper>>; | |
403 using OrderedHashSet = HeapLinkedHashSet<Member<Inner>>; | |
404 | |
405 Persistent<OrderedHashSet> set = new OrderedHashSet; | |
406 for (int i = 0; i < 13; ++i) { | |
407 IntWrapper* value = IntWrapper::create(i); | |
408 Inner* inner = new Inner; | |
409 inner->add(value); | |
410 set->add(inner); | |
411 } | |
412 EXPECT_EQ(13u, set->size()); | |
413 | |
414 int expected = 0; | |
415 for (const Inner* v : *set) { | |
416 EXPECT_EQ(1u, v->size()); | |
417 EXPECT_EQ(expected, (*v->begin())->value()); | |
418 expected++; | |
419 } | |
420 | |
421 performHeapCompaction(); | |
422 | |
423 expected = 0; | |
424 for (const Inner* v : *set) { | |
425 EXPECT_EQ(1u, v->size()); | |
426 EXPECT_EQ(expected, (*v->begin())->value()); | |
427 expected++; | |
428 } | |
429 } | |
430 | |
431 TEST(HeapCompactTest, CompactLinkedHashSetNested) { | |
432 using Inner = HeapLinkedHashSet<Member<IntWrapper>>; | |
433 using OrderedHashSet = HeapLinkedHashSet<Member<Inner>>; | |
434 | |
435 Persistent<OrderedHashSet> set = new OrderedHashSet; | |
436 for (int i = 0; i < 13; ++i) { | |
437 IntWrapper* value = IntWrapper::create(i); | |
438 Inner* inner = new Inner; | |
439 inner->add(value); | |
440 set->add(inner); | |
441 } | |
442 EXPECT_EQ(13u, set->size()); | |
443 | |
444 int expected = 0; | |
445 for (const Inner* v : *set) { | |
446 EXPECT_EQ(1u, v->size()); | |
447 EXPECT_EQ(expected, (*v->begin())->value()); | |
448 expected++; | |
449 } | |
450 | |
451 performHeapCompaction(); | |
452 | |
453 expected = 0; | |
454 for (const Inner* v : *set) { | |
455 EXPECT_EQ(1u, v->size()); | |
456 EXPECT_EQ(expected, (*v->begin())->value()); | |
457 expected++; | |
458 } | |
459 } | |
460 | |
461 } // namespace blink | |
OLD | NEW |