Leak reports collect information about the memory usage

chrome/browser/metrics/leak_detector/leak_detector_controller.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/metrics/leak_detector/leak_detector_controller.h"
 
 #include <algorithm>
 
 #include "base/logging.h"
+#include "base/memory/weak_ptr.h"
 #include "base/rand_util.h"
 #include "base/strings/string_number_conversions.h"
+#include "base/sys_info.h"
 #include "components/metrics/leak_detector/gnu_build_id_reader.h"
 #include "components/variations/variations_associated_data.h"
 #include "content/public/browser/browser_thread.h"
 
 namespace metrics {
 
 namespace {
 
 using ParamsMap = std::map<std::string, std::string>;
 
@@ skipping 116 matching lines @@
   }
   iter = params.find(kMaxRendererProcessesEnabledParam);
   if (iter == params.end() ||
       !base::StringToInt(iter->second, max_renderer_processes_enabled)) {
     *max_renderer_processes_enabled = kDefaultNumProcessesEnabled;
   }
 }
 
 }  // namespace
 
+TotalMemoryGrowthTracker::TotalMemoryGrowthTracker() : total_usage_kb_(0) {
+}
+
+TotalMemoryGrowthTracker::~TotalMemoryGrowthTracker() {
+}
+
+bool TotalMemoryGrowthTracker::UpdateSample(
+    base::ProcessId pid, int sample, int* diff) {
+  total_usage_kb_ += sample;
+  return false;
+}
+
+size_t TotalMemoryGrowthTracker::GetTotalUsageKb() const {

[Simon Que, 2016/10/05 22:12:43]

This can be inlined, and renamed to total_usage_kb():

https://google.github.io/styleguide/cppguide.html#Function_Names

[mwlodar, 2016/10/06 00:07:53]

Done.

+  return total_usage_kb_;
+}
+
+void TotalMemoryGrowthTracker::Reset() {

[Simon Que, 2016/10/05 22:12:43]

Same here.

[mwlodar, 2016/10/06 00:07:53]

Done.

+  total_usage_kb_ = 0;
+}
+
 LeakDetectorController::LeakDetectorController()
     : params_(GetLeakDetectorParams()),
       browser_process_enable_probability_(0),
       renderer_process_enable_probability_(0),
       max_renderer_processes_with_leak_detector_enabled_(0),
-      num_renderer_processes_with_leak_detector_enabled_(0) {
+      num_renderer_processes_with_leak_detector_enabled_(0),
+      waiting_for_collect_memory_usage_(false),
+      weak_ptr_factory_(this) {
   // Read the build ID once and store it.
   leak_detector::gnu_build_id_reader::ReadBuildID(&build_id_);
 
   GetLeakDetectorEnableParams(
       &browser_process_enable_probability_,
       &renderer_process_enable_probability_,
       &max_renderer_processes_with_leak_detector_enabled_);
 
   // Register the LeakDetectorController with the remote controller, so this
   // class can send/receive data to/from remote processes.
@@ skipping 50 matching lines @@
   --num_renderer_processes_with_leak_detector_enabled_;
 }
 
 bool LeakDetectorController::ShouldRandomlyEnableLeakDetectorOnRendererProcess()
     const {
   return base::RandDouble() < renderer_process_enable_probability_ &&
          num_renderer_processes_with_leak_detector_enabled_ <
              max_renderer_processes_with_leak_detector_enabled_;
 }
 
+void LeakDetectorController::OnMemoryDetailCollectionDone(size_t index) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(waiting_for_collect_memory_usage_);
+  waiting_for_collect_memory_usage_ = false;
+
+  // The available physical memory must be translated from bytes
+  // and the total memory usage must be translated from kb.
+  MemoryLeakReportProto::MemoryUsageInfo mem_info;
+  mem_info.set_available_ram_mb(
+      base::SysInfo::AmountOfAvailablePhysicalMemory() / 1024 / 1024);
+  mem_info.set_chrome_ram_usage_mb(
+      total_memory_growth_tracker_.GetTotalUsageKb() / 1024);
+
+  // Update all reports that arrived in the meantime.
+  for (; index < stored_reports_.size(); index++) {
+    stored_reports_[index].mutable_memory_usage_info()->CopyFrom(mem_info);

[Simon Que, 2016/10/05 22:12:43]

You could be overwriting mem info in previously stored reports. You'll need a
different implementation.

Also keep in mind that GetLeakReports() could be called between
StoreLeakReports() and OnMemoryDetailCollectionDone.

[mwlodar, 2016/10/06 00:07:53]

We have talked about this. No fix needed.

+  }
+}
+
 void LeakDetectorController::StoreLeakReports(
     const std::vector<MemoryLeakReportProto>& reports,
     MemoryLeakReportProto::ProcessType process_type) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
+  // Postpone a task of collecting info about the memory usage.
+  // When the task is done, all reports collected after this point
+  // will be updated, unless GetLeakReports() gets called first.
+  if (!waiting_for_collect_memory_usage_) {
+    waiting_for_collect_memory_usage_ = true;
+
+    total_memory_growth_tracker_.Reset();
+
+    // Idea of using MetricsMemoryDetails is the same as in
+    // ChromeMetricsServiceClient. As a side effect, the histogram
+    // data is generated.
+    base::Closure callback = base::Bind(
+        &LeakDetectorController::OnMemoryDetailCollectionDone,
+        weak_ptr_factory_.GetWeakPtr(), stored_reports_.size());
+
+    scoped_refptr<MetricsMemoryDetails> details(
+        new MetricsMemoryDetails(callback, &total_memory_growth_tracker_));
+    details->StartFetch();
+  }
+
   for (const auto& report : reports) {
     // Store the report and insert stored parameters.
     stored_reports_.push_back(report);
     stored_reports_.back().mutable_params()->CopyFrom(params_);
     stored_reports_.back().set_source_process(process_type);
     stored_reports_.back().mutable_build_id()->assign(build_id_.begin(),
                                                       build_id_.end());
   }
 }
 
 }  // namespace metrics