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/process_memory_totals_dump_provider.h" | |
6 | |
7 #include <stddef.h> | |
8 | |
9 #include "base/process/process_metrics.h" | |
10 #include "base/trace_event/process_memory_dump.h" | |
11 #include "base/trace_event/process_memory_totals.h" | |
12 #include "build/build_config.h" | |
13 | |
14 #if defined(OS_LINUX) || defined(OS_ANDROID) | |
15 #include <fcntl.h> | |
16 | |
17 #include "base/files/file_util.h" | |
18 | |
19 namespace { | |
20 bool kernel_supports_rss_peak_reset = true; | |
21 const char kClearPeakRssCommand[] = "5"; | |
22 } | |
23 #endif | |
24 | |
25 namespace base { | |
26 namespace trace_event { | |
27 | |
28 // static | |
29 uint64_t ProcessMemoryTotalsDumpProvider::rss_bytes_for_testing = 0; | |
30 | |
31 // static | |
32 ProcessMemoryTotalsDumpProvider* | |
33 ProcessMemoryTotalsDumpProvider::GetInstance() { | |
34 return Singleton< | |
35 ProcessMemoryTotalsDumpProvider, | |
36 LeakySingletonTraits<ProcessMemoryTotalsDumpProvider>>::get(); | |
37 } | |
38 | |
39 ProcessMemoryTotalsDumpProvider::ProcessMemoryTotalsDumpProvider() | |
40 : process_metrics_(ProcessMetrics::CreateCurrentProcessMetrics()) {} | |
41 | |
42 ProcessMemoryTotalsDumpProvider::~ProcessMemoryTotalsDumpProvider() { | |
43 } | |
44 | |
45 // Called at trace dump point time. Creates a snapshot the memory counters for | |
46 // the current process. | |
47 bool ProcessMemoryTotalsDumpProvider::OnMemoryDump(const MemoryDumpArgs& args, | |
48 ProcessMemoryDump* pmd) { | |
49 const uint64_t rss_bytes = rss_bytes_for_testing | |
50 ? rss_bytes_for_testing | |
51 : process_metrics_->GetWorkingSetSize(); | |
52 | |
53 uint64_t peak_rss_bytes = 0; | |
54 | |
55 #if !defined(OS_IOS) | |
56 peak_rss_bytes = process_metrics_->GetPeakWorkingSetSize(); | |
57 #if defined(OS_LINUX) || defined(OS_ANDROID) | |
58 if (kernel_supports_rss_peak_reset) { | |
59 // TODO(ssid): Fix crbug.com/461788 to write to the file from sandboxed | |
60 // processes. | |
61 int clear_refs_fd = open("/proc/self/clear_refs", O_WRONLY); | |
62 if (clear_refs_fd > 0 && | |
63 WriteFileDescriptor(clear_refs_fd, kClearPeakRssCommand, | |
64 sizeof(kClearPeakRssCommand))) { | |
65 pmd->process_totals()->set_is_peak_rss_resetable(true); | |
66 } else { | |
67 kernel_supports_rss_peak_reset = false; | |
68 } | |
69 close(clear_refs_fd); | |
70 } | |
71 #elif defined(OS_MACOSX) | |
72 size_t private_bytes; | |
73 bool res = process_metrics_->GetMemoryBytes(&private_bytes, | |
74 nullptr /* shared_bytes */); | |
75 if (res) { | |
76 pmd->process_totals()->SetExtraFieldInBytes("private_bytes", private_bytes); | |
77 } | |
78 #endif // defined(OS_LINUX) || defined(OS_ANDROID) | |
79 #endif // !defined(OS_IOS) | |
80 | |
81 if (rss_bytes > 0) { | |
82 pmd->process_totals()->set_resident_set_bytes(rss_bytes); | |
83 pmd->process_totals()->set_peak_resident_set_bytes(peak_rss_bytes); | |
84 pmd->set_has_process_totals(); | |
85 return true; | |
86 } | |
87 | |
88 return false; | |
89 } | |
90 | |
91 } // namespace trace_event | |
92 } // namespace base | |
OLD | NEW |