Chromium Code Reviews

Side by Side Diff: base/trace_event/winheap_dump_provider_win.cc

Issue 1128733002: Update from https://crrev.com/328418 (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff |
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 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 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 "base/trace_event/winheap_dump_provider_win.h" 5 #include "base/trace_event/winheap_dump_provider_win.h"
6 6
7 #include <windows.h> 7 #include <windows.h>
8 8
9 #include "base/trace_event/process_memory_dump.h" 9 #include "base/trace_event/process_memory_dump.h"
10 10
11 namespace base { 11 namespace base {
12 namespace trace_event { 12 namespace trace_event {
13 13
14 namespace { 14 namespace {
15 15
16 const char kDumperFriendlyName[] = "winheap";
17
18 // Report a heap dump to a process memory dump. The |heap_info| structure 16 // Report a heap dump to a process memory dump. The |heap_info| structure
19 // contains the information about this heap, and |heap_name| will be used to 17 // contains the information about this heap, and |dump_absolute_name| will be
20 // represent it in the report. 18 // used to represent it in the report.
21 bool ReportHeapDump(ProcessMemoryDump* pmd, 19 bool ReportHeapDump(ProcessMemoryDump* pmd,
22 const WinHeapInfo& heap_info, 20 const WinHeapInfo& heap_info,
23 const std::string& heap_name) { 21 const std::string& dump_absolute_name) {
24 MemoryAllocatorDump* dump = 22 MemoryAllocatorDump* dump = pmd->CreateAllocatorDump(dump_absolute_name);
25 pmd->CreateAllocatorDump(kDumperFriendlyName, heap_name);
26 if (!dump) 23 if (!dump)
27 return false; 24 return false;
28 dump->set_physical_size_in_bytes(heap_info.committed_size); 25 dump->AddScalar(MemoryAllocatorDump::kNameOuterSize,
29 dump->set_allocated_objects_count(heap_info.block_count); 26 MemoryAllocatorDump::kUnitsBytes, heap_info.committed_size);
30 dump->set_allocated_objects_size_in_bytes(heap_info.allocated_size); 27 dump->AddScalar(MemoryAllocatorDump::kNameInnerSize,
28 MemoryAllocatorDump::kUnitsBytes, heap_info.allocated_size);
29 dump->AddScalar(MemoryAllocatorDump::kNameObjectsCount,
30 MemoryAllocatorDump::kUnitsObjects, heap_info.block_count);
31 return true; 31 return true;
32 } 32 }
33 33
34 } // namespace 34 } // namespace
35 35
36 WinHeapDumpProvider* WinHeapDumpProvider::GetInstance() { 36 WinHeapDumpProvider* WinHeapDumpProvider::GetInstance() {
37 return Singleton<WinHeapDumpProvider, 37 return Singleton<WinHeapDumpProvider,
38 LeakySingletonTraits<WinHeapDumpProvider>>::get(); 38 LeakySingletonTraits<WinHeapDumpProvider>>::get();
39 } 39 }
40 40
41 bool WinHeapDumpProvider::DumpInto(ProcessMemoryDump* pmd) { 41 bool WinHeapDumpProvider::OnMemoryDump(ProcessMemoryDump* pmd) {
42 // Retrieves the number of heaps in the current process. 42 // Retrieves the number of heaps in the current process.
43 DWORD number_of_heaps = ::GetProcessHeaps(0, NULL); 43 DWORD number_of_heaps = ::GetProcessHeaps(0, NULL);
44 WinHeapInfo all_heap_info = {0}; 44 WinHeapInfo all_heap_info = {0};
45 45
46 // Try to retrieve a handle to all the heaps owned by this process. Returns 46 // Try to retrieve a handle to all the heaps owned by this process. Returns
47 // false if the number of heaps has changed. 47 // false if the number of heaps has changed.
48 // 48 //
49 // This is inherently racy as is, but it's not something that we observe a lot 49 // This is inherently racy as is, but it's not something that we observe a lot
50 // in Chrome, the heaps tend to be created at startup only. 50 // in Chrome, the heaps tend to be created at startup only.
51 scoped_ptr<HANDLE[]> all_heaps(new HANDLE[number_of_heaps]); 51 scoped_ptr<HANDLE[]> all_heaps(new HANDLE[number_of_heaps]);
52 if (::GetProcessHeaps(number_of_heaps, all_heaps.get()) != number_of_heaps) 52 if (::GetProcessHeaps(number_of_heaps, all_heaps.get()) != number_of_heaps)
53 return false; 53 return false;
54 54
55 // Skip the pointer to the heap array to avoid accounting the memory used by 55 // Skip the pointer to the heap array to avoid accounting the memory used by
56 // this dump provider. 56 // this dump provider.
57 std::set<void*> block_to_skip; 57 std::set<void*> block_to_skip;
58 block_to_skip.insert(all_heaps.get()); 58 block_to_skip.insert(all_heaps.get());
59 59
60 // Retrieves some metrics about each heap. 60 // Retrieves some metrics about each heap.
61 for (size_t i = 0; i < number_of_heaps; ++i) { 61 for (size_t i = 0; i < number_of_heaps; ++i) {
62 WinHeapInfo heap_info = {0}; 62 WinHeapInfo heap_info = {0};
63 heap_info.heap_id = all_heaps[i]; 63 heap_info.heap_id = all_heaps[i];
64 GetHeapInformation(&heap_info, block_to_skip); 64 GetHeapInformation(&heap_info, block_to_skip);
65 65
66 all_heap_info.allocated_size += heap_info.allocated_size; 66 all_heap_info.allocated_size += heap_info.allocated_size;
67 all_heap_info.committed_size += heap_info.committed_size; 67 all_heap_info.committed_size += heap_info.committed_size;
68 all_heap_info.block_count += heap_info.block_count; 68 all_heap_info.block_count += heap_info.block_count;
69 } 69 }
70 // Report the heap dump. 70 // Report the heap dump.
71 if (!ReportHeapDump(pmd, all_heap_info, MemoryAllocatorDump::kRootHeap)) 71 if (!ReportHeapDump(pmd, all_heap_info, "winheap"))
72 return false; 72 return false;
73 73
74 return true; 74 return true;
75 } 75 }
76 76
77 const char* WinHeapDumpProvider::GetFriendlyName() const {
78 return kDumperFriendlyName;
79 }
80
81 bool WinHeapDumpProvider::GetHeapInformation( 77 bool WinHeapDumpProvider::GetHeapInformation(
82 WinHeapInfo* heap_info, 78 WinHeapInfo* heap_info,
83 const std::set<void*>& block_to_skip) { 79 const std::set<void*>& block_to_skip) {
84 CHECK(::HeapLock(heap_info->heap_id) == TRUE); 80 CHECK(::HeapLock(heap_info->heap_id) == TRUE);
85 PROCESS_HEAP_ENTRY heap_entry; 81 PROCESS_HEAP_ENTRY heap_entry;
86 heap_entry.lpData = nullptr; 82 heap_entry.lpData = nullptr;
87 // Walk over all the entries in this heap. 83 // Walk over all the entries in this heap.
88 while (::HeapWalk(heap_info->heap_id, &heap_entry) != FALSE) { 84 while (::HeapWalk(heap_info->heap_id, &heap_entry) != FALSE) {
89 if (block_to_skip.count(heap_entry.lpData) == 1) 85 if (block_to_skip.count(heap_entry.lpData) == 1)
90 continue; 86 continue;
91 if ((heap_entry.wFlags & PROCESS_HEAP_ENTRY_BUSY) != 0) { 87 if ((heap_entry.wFlags & PROCESS_HEAP_ENTRY_BUSY) != 0) {
92 heap_info->allocated_size += heap_entry.cbData; 88 heap_info->allocated_size += heap_entry.cbData;
93 heap_info->block_count++; 89 heap_info->block_count++;
94 } else if ((heap_entry.wFlags & PROCESS_HEAP_REGION) != 0) { 90 } else if ((heap_entry.wFlags & PROCESS_HEAP_REGION) != 0) {
95 heap_info->committed_size += heap_entry.Region.dwCommittedSize; 91 heap_info->committed_size += heap_entry.Region.dwCommittedSize;
96 } 92 }
97 } 93 }
98 CHECK(::HeapUnlock(heap_info->heap_id) == TRUE); 94 CHECK(::HeapUnlock(heap_info->heap_id) == TRUE);
99 return true; 95 return true;
100 } 96 }
101 97
102 } // namespace trace_event 98 } // namespace trace_event
103 } // namespace base 99 } // namespace base
OLDNEW
« no previous file with comments | « base/trace_event/winheap_dump_provider_win.h ('k') | base/trace_event/winheap_dump_provider_win_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine