| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/common/discardable_shared_memory_heap.h" | 5 #include "content/common/discardable_shared_memory_heap.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/discardable_shared_memory.h" | 8 #include "base/memory/discardable_shared_memory.h" |
| 9 #include "base/process/process_metrics.h" | 9 #include "base/process/process_metrics.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 EXPECT_EQ(0u, heap.GetSize()); | 23 EXPECT_EQ(0u, heap.GetSize()); |
| 24 | 24 |
| 25 // Initial size of free lists should be 0. | 25 // Initial size of free lists should be 0. |
| 26 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); | 26 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); |
| 27 | 27 |
| 28 // Free lists are initially empty. | 28 // Free lists are initially empty. |
| 29 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); | 29 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); |
| 30 | 30 |
| 31 const size_t kBlocks = 10; | 31 const size_t kBlocks = 10; |
| 32 size_t memory_size = block_size * kBlocks; | 32 size_t memory_size = block_size * kBlocks; |
| 33 int next_discardable_shared_memory_id = 0; |
| 33 | 34 |
| 34 scoped_ptr<base::DiscardableSharedMemory> memory( | 35 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 35 new base::DiscardableSharedMemory); | 36 new base::DiscardableSharedMemory); |
| 36 ASSERT_TRUE(memory->CreateAndMap(memory_size)); | 37 ASSERT_TRUE(memory->CreateAndMap(memory_size)); |
| 37 | 38 |
| 38 // Create new span for memory. | 39 // Create new span for memory. |
| 39 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( | 40 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( |
| 40 heap.Grow(memory.Pass(), memory_size, base::Bind(NullTask))); | 41 heap.Grow(memory.Pass(), memory_size, next_discardable_shared_memory_id++, |
| 42 base::Bind(NullTask))); |
| 41 | 43 |
| 42 // Size should match |memory_size|. | 44 // Size should match |memory_size|. |
| 43 EXPECT_EQ(memory_size, heap.GetSize()); | 45 EXPECT_EQ(memory_size, heap.GetSize()); |
| 44 | 46 |
| 45 // Size of free lists should still be 0. | 47 // Size of free lists should still be 0. |
| 46 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); | 48 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); |
| 47 | 49 |
| 48 // Free list should still be empty as |new_span| is currently in use. | 50 // Free list should still be empty as |new_span| is currently in use. |
| 49 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); | 51 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); |
| 50 | 52 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 68 // Merge it into the free lists again. | 70 // Merge it into the free lists again. |
| 69 heap.MergeIntoFreeLists(span.Pass()); | 71 heap.MergeIntoFreeLists(span.Pass()); |
| 70 } | 72 } |
| 71 | 73 |
| 72 TEST(DiscardableSharedMemoryHeapTest, SplitAndMerge) { | 74 TEST(DiscardableSharedMemoryHeapTest, SplitAndMerge) { |
| 73 size_t block_size = base::GetPageSize(); | 75 size_t block_size = base::GetPageSize(); |
| 74 DiscardableSharedMemoryHeap heap(block_size); | 76 DiscardableSharedMemoryHeap heap(block_size); |
| 75 | 77 |
| 76 const size_t kBlocks = 6; | 78 const size_t kBlocks = 6; |
| 77 size_t memory_size = block_size * kBlocks; | 79 size_t memory_size = block_size * kBlocks; |
| 80 int next_discardable_shared_memory_id = 0; |
| 78 | 81 |
| 79 scoped_ptr<base::DiscardableSharedMemory> memory( | 82 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 80 new base::DiscardableSharedMemory); | 83 new base::DiscardableSharedMemory); |
| 81 ASSERT_TRUE(memory->CreateAndMap(memory_size)); | 84 ASSERT_TRUE(memory->CreateAndMap(memory_size)); |
| 82 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( | 85 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( |
| 83 heap.Grow(memory.Pass(), memory_size, base::Bind(NullTask))); | 86 heap.Grow(memory.Pass(), memory_size, next_discardable_shared_memory_id++, |
| 87 base::Bind(NullTask))); |
| 84 | 88 |
| 85 // Split span into two. | 89 // Split span into two. |
| 86 scoped_ptr<DiscardableSharedMemoryHeap::Span> leftover = | 90 scoped_ptr<DiscardableSharedMemoryHeap::Span> leftover = |
| 87 heap.Split(new_span.get(), 3); | 91 heap.Split(new_span.get(), 3); |
| 88 ASSERT_TRUE(leftover); | 92 ASSERT_TRUE(leftover); |
| 89 | 93 |
| 90 // Merge |leftover| into free lists. | 94 // Merge |leftover| into free lists. |
| 91 heap.MergeIntoFreeLists(leftover.Pass()); | 95 heap.MergeIntoFreeLists(leftover.Pass()); |
| 92 | 96 |
| 93 // Some of the memory is still in use. | 97 // Some of the memory is still in use. |
| (...skipping 29 matching lines...) Expand all Loading... |
| 123 // Merge it into the free lists again. | 127 // Merge it into the free lists again. |
| 124 heap.MergeIntoFreeLists(large_span.Pass()); | 128 heap.MergeIntoFreeLists(large_span.Pass()); |
| 125 } | 129 } |
| 126 | 130 |
| 127 TEST(DiscardableSharedMemoryHeapTest, MergeSingleBlockSpan) { | 131 TEST(DiscardableSharedMemoryHeapTest, MergeSingleBlockSpan) { |
| 128 size_t block_size = base::GetPageSize(); | 132 size_t block_size = base::GetPageSize(); |
| 129 DiscardableSharedMemoryHeap heap(block_size); | 133 DiscardableSharedMemoryHeap heap(block_size); |
| 130 | 134 |
| 131 const size_t kBlocks = 6; | 135 const size_t kBlocks = 6; |
| 132 size_t memory_size = block_size * kBlocks; | 136 size_t memory_size = block_size * kBlocks; |
| 137 int next_discardable_shared_memory_id = 0; |
| 133 | 138 |
| 134 scoped_ptr<base::DiscardableSharedMemory> memory( | 139 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 135 new base::DiscardableSharedMemory); | 140 new base::DiscardableSharedMemory); |
| 136 ASSERT_TRUE(memory->CreateAndMap(memory_size)); | 141 ASSERT_TRUE(memory->CreateAndMap(memory_size)); |
| 137 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( | 142 scoped_ptr<DiscardableSharedMemoryHeap::Span> new_span( |
| 138 heap.Grow(memory.Pass(), memory_size, base::Bind(NullTask))); | 143 heap.Grow(memory.Pass(), memory_size, next_discardable_shared_memory_id++, |
| 144 base::Bind(NullTask))); |
| 139 | 145 |
| 140 // Split span into two. | 146 // Split span into two. |
| 141 scoped_ptr<DiscardableSharedMemoryHeap::Span> leftover = | 147 scoped_ptr<DiscardableSharedMemoryHeap::Span> leftover = |
| 142 heap.Split(new_span.get(), 5); | 148 heap.Split(new_span.get(), 5); |
| 143 ASSERT_TRUE(leftover); | 149 ASSERT_TRUE(leftover); |
| 144 | 150 |
| 145 // Merge |new_span| into free lists. | 151 // Merge |new_span| into free lists. |
| 146 heap.MergeIntoFreeLists(new_span.Pass()); | 152 heap.MergeIntoFreeLists(new_span.Pass()); |
| 147 | 153 |
| 148 // Merge |leftover| into free lists. | 154 // Merge |leftover| into free lists. |
| 149 heap.MergeIntoFreeLists(leftover.Pass()); | 155 heap.MergeIntoFreeLists(leftover.Pass()); |
| 150 } | 156 } |
| 151 | 157 |
| 152 TEST(DiscardableSharedMemoryHeapTest, Grow) { | 158 TEST(DiscardableSharedMemoryHeapTest, Grow) { |
| 153 size_t block_size = base::GetPageSize(); | 159 size_t block_size = base::GetPageSize(); |
| 154 DiscardableSharedMemoryHeap heap(block_size); | 160 DiscardableSharedMemoryHeap heap(block_size); |
| 161 int next_discardable_shared_memory_id = 0; |
| 155 | 162 |
| 156 scoped_ptr<base::DiscardableSharedMemory> memory1( | 163 scoped_ptr<base::DiscardableSharedMemory> memory1( |
| 157 new base::DiscardableSharedMemory); | 164 new base::DiscardableSharedMemory); |
| 158 ASSERT_TRUE(memory1->CreateAndMap(block_size)); | 165 ASSERT_TRUE(memory1->CreateAndMap(block_size)); |
| 159 heap.MergeIntoFreeLists( | 166 heap.MergeIntoFreeLists(heap.Grow(memory1.Pass(), block_size, |
| 160 heap.Grow(memory1.Pass(), block_size, base::Bind(NullTask)).Pass()); | 167 next_discardable_shared_memory_id++, |
| 168 base::Bind(NullTask)).Pass()); |
| 161 | 169 |
| 162 // Remove a span from free lists. | 170 // Remove a span from free lists. |
| 163 scoped_ptr<DiscardableSharedMemoryHeap::Span> span1 = | 171 scoped_ptr<DiscardableSharedMemoryHeap::Span> span1 = |
| 164 heap.SearchFreeLists(1, 0); | 172 heap.SearchFreeLists(1, 0); |
| 165 EXPECT_TRUE(span1); | 173 EXPECT_TRUE(span1); |
| 166 | 174 |
| 167 // No more memory available. | 175 // No more memory available. |
| 168 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); | 176 EXPECT_FALSE(heap.SearchFreeLists(1, 0)); |
| 169 | 177 |
| 170 // Grow free lists using new memory. | 178 // Grow free lists using new memory. |
| 171 scoped_ptr<base::DiscardableSharedMemory> memory2( | 179 scoped_ptr<base::DiscardableSharedMemory> memory2( |
| 172 new base::DiscardableSharedMemory); | 180 new base::DiscardableSharedMemory); |
| 173 ASSERT_TRUE(memory2->CreateAndMap(block_size)); | 181 ASSERT_TRUE(memory2->CreateAndMap(block_size)); |
| 174 heap.MergeIntoFreeLists( | 182 heap.MergeIntoFreeLists(heap.Grow(memory2.Pass(), block_size, |
| 175 heap.Grow(memory2.Pass(), block_size, base::Bind(NullTask)).Pass()); | 183 next_discardable_shared_memory_id++, |
| 184 base::Bind(NullTask)).Pass()); |
| 176 | 185 |
| 177 // Memory should now be available. | 186 // Memory should now be available. |
| 178 scoped_ptr<DiscardableSharedMemoryHeap::Span> span2 = | 187 scoped_ptr<DiscardableSharedMemoryHeap::Span> span2 = |
| 179 heap.SearchFreeLists(1, 0); | 188 heap.SearchFreeLists(1, 0); |
| 180 EXPECT_TRUE(span2); | 189 EXPECT_TRUE(span2); |
| 181 | 190 |
| 182 // Merge spans into the free lists again. | 191 // Merge spans into the free lists again. |
| 183 heap.MergeIntoFreeLists(span1.Pass()); | 192 heap.MergeIntoFreeLists(span1.Pass()); |
| 184 heap.MergeIntoFreeLists(span2.Pass()); | 193 heap.MergeIntoFreeLists(span2.Pass()); |
| 185 } | 194 } |
| 186 | 195 |
| 187 TEST(DiscardableSharedMemoryHeapTest, ReleaseFreeMemory) { | 196 TEST(DiscardableSharedMemoryHeapTest, ReleaseFreeMemory) { |
| 188 size_t block_size = base::GetPageSize(); | 197 size_t block_size = base::GetPageSize(); |
| 189 DiscardableSharedMemoryHeap heap(block_size); | 198 DiscardableSharedMemoryHeap heap(block_size); |
| 199 int next_discardable_shared_memory_id = 0; |
| 190 | 200 |
| 191 scoped_ptr<base::DiscardableSharedMemory> memory( | 201 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 192 new base::DiscardableSharedMemory); | 202 new base::DiscardableSharedMemory); |
| 193 ASSERT_TRUE(memory->CreateAndMap(block_size)); | 203 ASSERT_TRUE(memory->CreateAndMap(block_size)); |
| 194 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = | 204 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = |
| 195 heap.Grow(memory.Pass(), block_size, base::Bind(NullTask)); | 205 heap.Grow(memory.Pass(), block_size, next_discardable_shared_memory_id++, |
| 206 base::Bind(NullTask)); |
| 196 | 207 |
| 197 // Free lists should be empty. | 208 // Free lists should be empty. |
| 198 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); | 209 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); |
| 199 | 210 |
| 200 heap.ReleaseFreeMemory(); | 211 heap.ReleaseFreeMemory(); |
| 201 | 212 |
| 202 // Size should still match |block_size|. | 213 // Size should still match |block_size|. |
| 203 EXPECT_EQ(block_size, heap.GetSize()); | 214 EXPECT_EQ(block_size, heap.GetSize()); |
| 204 | 215 |
| 205 heap.MergeIntoFreeLists(span.Pass()); | 216 heap.MergeIntoFreeLists(span.Pass()); |
| 206 heap.ReleaseFreeMemory(); | 217 heap.ReleaseFreeMemory(); |
| 207 | 218 |
| 208 // Memory should have been released. | 219 // Memory should have been released. |
| 209 EXPECT_EQ(0u, heap.GetSize()); | 220 EXPECT_EQ(0u, heap.GetSize()); |
| 210 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); | 221 EXPECT_EQ(0u, heap.GetSizeOfFreeLists()); |
| 211 } | 222 } |
| 212 | 223 |
| 213 TEST(DiscardableSharedMemoryHeapTest, ReleasePurgedMemory) { | 224 TEST(DiscardableSharedMemoryHeapTest, ReleasePurgedMemory) { |
| 214 size_t block_size = base::GetPageSize(); | 225 size_t block_size = base::GetPageSize(); |
| 215 DiscardableSharedMemoryHeap heap(block_size); | 226 DiscardableSharedMemoryHeap heap(block_size); |
| 227 int next_discardable_shared_memory_id = 0; |
| 216 | 228 |
| 217 scoped_ptr<base::DiscardableSharedMemory> memory( | 229 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 218 new base::DiscardableSharedMemory); | 230 new base::DiscardableSharedMemory); |
| 219 ASSERT_TRUE(memory->CreateAndMap(block_size)); | 231 ASSERT_TRUE(memory->CreateAndMap(block_size)); |
| 220 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = | 232 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = |
| 221 heap.Grow(memory.Pass(), block_size, base::Bind(NullTask)); | 233 heap.Grow(memory.Pass(), block_size, next_discardable_shared_memory_id++, |
| 234 base::Bind(NullTask)); |
| 222 | 235 |
| 223 // Unlock memory so it can be purged. | 236 // Unlock memory so it can be purged. |
| 224 span->shared_memory()->Unlock(0, 0); | 237 span->shared_memory()->Unlock(0, 0); |
| 225 | 238 |
| 226 // Purge and release shared memory. | 239 // Purge and release shared memory. |
| 227 bool rv = span->shared_memory()->Purge(base::Time::Now()); | 240 bool rv = span->shared_memory()->Purge(base::Time::Now()); |
| 228 EXPECT_TRUE(rv); | 241 EXPECT_TRUE(rv); |
| 229 heap.ReleasePurgedMemory(); | 242 heap.ReleasePurgedMemory(); |
| 230 | 243 |
| 231 // Shared memory backing for |span| should be gone. | 244 // Shared memory backing for |span| should be gone. |
| 232 EXPECT_FALSE(span->shared_memory()); | 245 EXPECT_FALSE(span->shared_memory()); |
| 233 | 246 |
| 234 // Size should be 0. | 247 // Size should be 0. |
| 235 EXPECT_EQ(0u, heap.GetSize()); | 248 EXPECT_EQ(0u, heap.GetSize()); |
| 236 } | 249 } |
| 237 | 250 |
| 238 TEST(DiscardableSharedMemoryHeapTest, Slack) { | 251 TEST(DiscardableSharedMemoryHeapTest, Slack) { |
| 239 size_t block_size = base::GetPageSize(); | 252 size_t block_size = base::GetPageSize(); |
| 240 DiscardableSharedMemoryHeap heap(block_size); | 253 DiscardableSharedMemoryHeap heap(block_size); |
| 241 | 254 |
| 242 const size_t kBlocks = 6; | 255 const size_t kBlocks = 6; |
| 243 size_t memory_size = block_size * kBlocks; | 256 size_t memory_size = block_size * kBlocks; |
| 257 int next_discardable_shared_memory_id = 0; |
| 244 | 258 |
| 245 scoped_ptr<base::DiscardableSharedMemory> memory( | 259 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 246 new base::DiscardableSharedMemory); | 260 new base::DiscardableSharedMemory); |
| 247 ASSERT_TRUE(memory->CreateAndMap(memory_size)); | 261 ASSERT_TRUE(memory->CreateAndMap(memory_size)); |
| 248 heap.MergeIntoFreeLists( | 262 heap.MergeIntoFreeLists(heap.Grow(memory.Pass(), memory_size, |
| 249 heap.Grow(memory.Pass(), memory_size, base::Bind(NullTask)).Pass()); | 263 next_discardable_shared_memory_id++, |
| 264 base::Bind(NullTask)).Pass()); |
| 250 | 265 |
| 251 // No free span that is less or equal to 3 + 1. | 266 // No free span that is less or equal to 3 + 1. |
| 252 EXPECT_FALSE(heap.SearchFreeLists(3, 1)); | 267 EXPECT_FALSE(heap.SearchFreeLists(3, 1)); |
| 253 | 268 |
| 254 // No free span that is less or equal to 3 + 2. | 269 // No free span that is less or equal to 3 + 2. |
| 255 EXPECT_FALSE(heap.SearchFreeLists(3, 2)); | 270 EXPECT_FALSE(heap.SearchFreeLists(3, 2)); |
| 256 | 271 |
| 257 // No free span that is less or equal to 1 + 4. | 272 // No free span that is less or equal to 1 + 4. |
| 258 EXPECT_FALSE(heap.SearchFreeLists(1, 4)); | 273 EXPECT_FALSE(heap.SearchFreeLists(1, 4)); |
| 259 | 274 |
| 260 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = | 275 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = |
| 261 heap.SearchFreeLists(1, 5); | 276 heap.SearchFreeLists(1, 5); |
| 262 EXPECT_TRUE(span); | 277 EXPECT_TRUE(span); |
| 263 | 278 |
| 264 heap.MergeIntoFreeLists(span.Pass()); | 279 heap.MergeIntoFreeLists(span.Pass()); |
| 265 } | 280 } |
| 266 | 281 |
| 267 void OnDeleted(bool* deleted) { | 282 void OnDeleted(bool* deleted) { |
| 268 *deleted = true; | 283 *deleted = true; |
| 269 } | 284 } |
| 270 | 285 |
| 271 TEST(DiscardableSharedMemoryHeapTest, DeletedCallback) { | 286 TEST(DiscardableSharedMemoryHeapTest, DeletedCallback) { |
| 272 size_t block_size = base::GetPageSize(); | 287 size_t block_size = base::GetPageSize(); |
| 273 DiscardableSharedMemoryHeap heap(block_size); | 288 DiscardableSharedMemoryHeap heap(block_size); |
| 289 int next_discardable_shared_memory_id = 0; |
| 274 | 290 |
| 275 scoped_ptr<base::DiscardableSharedMemory> memory( | 291 scoped_ptr<base::DiscardableSharedMemory> memory( |
| 276 new base::DiscardableSharedMemory); | 292 new base::DiscardableSharedMemory); |
| 277 ASSERT_TRUE(memory->CreateAndMap(block_size)); | 293 ASSERT_TRUE(memory->CreateAndMap(block_size)); |
| 278 bool deleted = false; | 294 bool deleted = false; |
| 279 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = | 295 scoped_ptr<DiscardableSharedMemoryHeap::Span> span = |
| 280 heap.Grow(memory.Pass(), block_size, | 296 heap.Grow(memory.Pass(), block_size, next_discardable_shared_memory_id++, |
| 281 base::Bind(OnDeleted, base::Unretained(&deleted))); | 297 base::Bind(OnDeleted, base::Unretained(&deleted))); |
| 282 | 298 |
| 283 heap.MergeIntoFreeLists(span.Pass()); | 299 heap.MergeIntoFreeLists(span.Pass()); |
| 284 heap.ReleaseFreeMemory(); | 300 heap.ReleaseFreeMemory(); |
| 285 | 301 |
| 286 EXPECT_TRUE(deleted); | 302 EXPECT_TRUE(deleted); |
| 287 } | 303 } |
| 288 | 304 |
| 289 } // namespace | 305 } // namespace |
| 290 } // namespace content | 306 } // namespace content |
| OLD | NEW |