| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 "base/trace_event/winheap_dump_provider_win.h" | |
| 6 | |
| 7 #include <windows.h> | |
| 8 | |
| 9 #include "base/trace_event/memory_dump_session_state.h" | |
| 10 #include "base/trace_event/process_memory_dump.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 | |
| 13 namespace base { | |
| 14 namespace trace_event { | |
| 15 | |
| 16 class WinHeapDumpProviderTest : public testing::Test { | |
| 17 public: | |
| 18 bool GetHeapInformation(WinHeapDumpProvider* provider, | |
| 19 WinHeapInfo* heap_info, | |
| 20 const std::set<void*>& block_to_skip) { | |
| 21 return provider->GetHeapInformation(heap_info, block_to_skip); | |
| 22 } | |
| 23 }; | |
| 24 | |
| 25 TEST_F(WinHeapDumpProviderTest, DumpInto) { | |
| 26 ProcessMemoryDump pmd(make_scoped_refptr(new MemoryDumpSessionState())); | |
| 27 | |
| 28 WinHeapDumpProvider* winheap_dump_provider = | |
| 29 WinHeapDumpProvider::GetInstance(); | |
| 30 ASSERT_NE(reinterpret_cast<WinHeapDumpProvider*>(nullptr), | |
| 31 winheap_dump_provider); | |
| 32 | |
| 33 ASSERT_TRUE(winheap_dump_provider->DumpInto(&pmd)); | |
| 34 } | |
| 35 | |
| 36 TEST_F(WinHeapDumpProviderTest, GetHeapInformation) { | |
| 37 ProcessMemoryDump pmd(make_scoped_refptr(new MemoryDumpSessionState())); | |
| 38 | |
| 39 WinHeapDumpProvider* winheap_dump_provider = | |
| 40 WinHeapDumpProvider::GetInstance(); | |
| 41 ASSERT_NE(reinterpret_cast<WinHeapDumpProvider*>(nullptr), | |
| 42 winheap_dump_provider); | |
| 43 | |
| 44 HANDLE heap = ::HeapCreate(NULL, 0, 0); | |
| 45 ASSERT_NE(nullptr, heap); | |
| 46 | |
| 47 const size_t kAllocSize = 42; | |
| 48 void* alloc = ::HeapAlloc(heap, 0, kAllocSize); | |
| 49 ASSERT_NE(nullptr, alloc); | |
| 50 | |
| 51 WinHeapInfo heap_info = {0}; | |
| 52 heap_info.heap_id = heap; | |
| 53 std::set<void*> block_to_skip; | |
| 54 | |
| 55 // Put the allocation into the skip list and make sure that the provider | |
| 56 // ignores it. | |
| 57 block_to_skip.insert(alloc); | |
| 58 ASSERT_TRUE( | |
| 59 GetHeapInformation(winheap_dump_provider, &heap_info, block_to_skip)); | |
| 60 EXPECT_EQ(0U, heap_info.block_count); | |
| 61 EXPECT_EQ(0U, heap_info.allocated_size); | |
| 62 // We can't check the committed size here, as it can depend on the version of | |
| 63 // kernel32.dll. | |
| 64 | |
| 65 // Remove the allocation from the skip list and check if it's analysed | |
| 66 // properlyly. | |
| 67 block_to_skip.erase(alloc); | |
| 68 ASSERT_TRUE( | |
| 69 GetHeapInformation(winheap_dump_provider, &heap_info, block_to_skip)); | |
| 70 EXPECT_EQ(1, heap_info.block_count); | |
| 71 EXPECT_EQ(kAllocSize, heap_info.allocated_size); | |
| 72 EXPECT_LT(kAllocSize, heap_info.committed_size); | |
| 73 | |
| 74 EXPECT_TRUE(::HeapFree(heap, 0, alloc)); | |
| 75 EXPECT_EQ(TRUE, ::HeapDestroy(heap)); | |
| 76 } | |
| 77 | |
| 78 } // namespace trace_event | |
| 79 } // namespace base | |
| OLD | NEW |