Instrument SourceStream using MemoryDumpProvider

net/filter/brotli_source_stream.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 "net/filter/brotli_source_stream.h"
 
 #include "base/bind.h"
 #include "base/bit_cast.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
+#include "base/strings/stringprintf.h"
+#include "base/trace_event/memory_allocator_dump.h"
+#include "base/trace_event/process_memory_dump.h"
 #include "net/base/io_buffer.h"
 #include "third_party/brotli/include/brotli/decode.h"
 
 namespace net {
 
 namespace {
 
 const char kBrotli[] = "BROTLI";
 const uint8_t kGzipHeader[] = {0x1f, 0x8b, 0x08};
 
@@ skipping 45 matching lines @@
 
     // All code here is for gathering stats, and can be removed when
     // BrotliSourceStream is considered stable.
     const int kBuckets = 48;
     const int64_t kMaxKb = 1 << (kBuckets / 3);  // 64MiB in KiB
     UMA_HISTOGRAM_CUSTOM_COUNTS("BrotliFilter.UsedMemoryKB",
                                 used_memory_maximum_ / 1024, 1, kMaxKb,
                                 kBuckets);
   }
 
+  void DumpMemoryStatsImpl(
+      base::trace_event::ProcessMemoryDump* pmd,
+      const std::string& parent_dump_absolute_name) const override {
+    // Log pointer address to avoid duplication though in practice there is
+    // often only one BrotliSourceStream per URLRequest.
+    base::trace_event::MemoryAllocatorDump* brotli_dump =
+        pmd->CreateAllocatorDump(base::StringPrintf(
+            "%s/brotli_%p", parent_dump_absolute_name.c_str(), this));
+    brotli_dump->AddScalar(base::trace_event::MemoryAllocatorDump::kNameSize,
+                           base::trace_event::MemoryAllocatorDump::kUnitsBytes,
+                           used_memory_);
+  }
+
  private:
   // Reported in UMA and must be kept in sync with the histograms.xml file.
   enum class DecodingStatus : int {
     DECODING_IN_PROGRESS = 0,
     DECODING_DONE,
     DECODING_ERROR,
 
     DECODING_STATUS_COUNT
     // DECODING_STATUS_COUNT must always be the last element in this enum.
   };
@@ skipping 106 matching lines @@
 };
 
 }  // namespace
 
 std::unique_ptr<FilterSourceStream> CreateBrotliSourceStream(
     std::unique_ptr<SourceStream> previous) {
   return base::WrapUnique(new BrotliSourceStream(std::move(previous)));
 }
 
 }  // namespace net