[Tracing] Add metadata filter

content/browser/tracing/tracing_controller_impl_data_sinks.cc

 // Copyright 2015 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 "content/browser/tracing/tracing_controller_impl.h"
 
 #include "base/bind.h"
 #include "base/files/file_util.h"
 #include "base/json/json_writer.h"
+#include "base/strings/pattern.h"
 #include "content/public/browser/browser_thread.h"
 #include "third_party/zlib/zlib.h"
 
 namespace content {
 
 namespace {
 
 class StringTraceDataEndpoint : public TracingController::TraceDataEndpoint {
  public:
   typedef base::Callback<void(scoped_ptr<const base::DictionaryValue>,
@@ skipping 117 matching lines @@
   }
 
   void SetPowerTrace(const std::string& data) override { power_trace_ = data; }
 
   void Close() override {
     AddTraceChunkAndPassToEndpoint("]");
     if (!system_trace_.empty())
       AddTraceChunkAndPassToEndpoint(",\"systemTraceEvents\": " +
                                      system_trace_);
     std::string metadataJSON;
-    if (base::JSONWriter::Write(GetMetadata(), &metadataJSON) &&
+    if (base::JSONWriter::Write(*GetMetadataCopy(), &metadataJSON) &&
         !metadataJSON.empty())
       AddTraceChunkAndPassToEndpoint(",\"metadata\": " + metadataJSON);
     if (!power_trace_.empty()) {
       AddTraceChunkAndPassToEndpoint(",\"powerTraceAsString\": " +
                                      power_trace_);
     }
 
     AddTraceChunkAndPassToEndpoint("}");
 
-    scoped_ptr<const base::DictionaryValue> metadata(GetMetadata().DeepCopy());
-    endpoint_->ReceiveTraceFinalContents(metadata.Pass(), trace_);
+    endpoint_->ReceiveTraceFinalContents(GetMetadataCopy(), trace_);
   }
 
  private:
   ~StringTraceDataSink() override {}
 
   scoped_refptr<TracingController::TraceDataEndpoint> endpoint_;
   std::string trace_;
   std::string system_trace_;
   std::string power_trace_;
 
@@ skipping 103 matching lines @@
 
     if (compressed_trace_data_.empty())
       AddTraceChunkAndCompressOnFileThread("{\"traceEvents\":[", false);
 
     AddTraceChunkAndCompressOnFileThread("]", false);
     if (!system_trace_.empty()) {
       AddTraceChunkAndCompressOnFileThread(
           ",\"systemTraceEvents\": " + system_trace_, false);
     }
     std::string metadataJSON;
-    if (base::JSONWriter::Write(GetMetadata(), &metadataJSON) &&
+    if (base::JSONWriter::Write(*GetMetadataCopy(), &metadataJSON) &&
         !metadataJSON.empty()) {
       AddTraceChunkAndCompressOnFileThread(",\"metadata\": " + metadataJSON,
                                            false);
     }
     if (!power_trace_.empty()) {
       AddTraceChunkAndCompressOnFileThread(
           ",\"powerTraceAsString\": " + power_trace_, false);
     }
     AddTraceChunkAndCompressOnFileThread("}", true);
 
     deflateEnd(stream_.get());
     stream_.reset();
 
-    scoped_ptr<const base::DictionaryValue> metadata(GetMetadata().DeepCopy());
-    endpoint_->ReceiveTraceFinalContents(metadata.Pass(),
+    endpoint_->ReceiveTraceFinalContents(GetMetadataCopy(),
                                          compressed_trace_data_);
   }
 
   scoped_refptr<TracingController::TraceDataEndpoint> endpoint_;
   scoped_ptr<z_stream> stream_;
   bool already_tried_open_;
   std::string compressed_trace_data_;
   std::string system_trace_;
   std::string power_trace_;
 
   DISALLOW_COPY_AND_ASSIGN(CompressedStringTraceDataSink);
 };
 
 }  // namespace
 
+TracingController::TraceDataSink::TraceDataSink() {}
+
+TracingController::TraceDataSink::~TraceDataSink() {}
+
 void TracingController::TraceDataSink::AddMetadata(
     const base::DictionaryValue& data) {
   metadata_.MergeDictionary(&data);
 }
 
-const base::DictionaryValue&
-    TracingController::TraceDataSink::GetMetadata() const {
-  return metadata_;
+void TracingController::TraceDataSink::SetMetadataFilterPredicate(
+    const MetadataFilterPredicate& metadata_filter_predicate) {
+  metadata_filter_predicate_ = metadata_filter_predicate;
+}
+
+scoped_ptr<const base::DictionaryValue>
+    TracingController::TraceDataSink::GetMetadataCopy() const {
+  if (metadata_filter_predicate_.is_null())
+    return scoped_ptr<const base::DictionaryValue>(metadata_.DeepCopy()).Pass();
+
+  scoped_ptr<base::DictionaryValue> metadata_copy(new base::DictionaryValue);
+  for (base::DictionaryValue::Iterator it(metadata_); !it.IsAtEnd();
+       it.Advance()) {
+    if (metadata_filter_predicate_.Run(it.key()))
+      metadata_copy->Set(it.key(), it.value().DeepCopy());
+    else
+      metadata_copy->SetString(it.key(), "__stripped__");
+  }
+  return metadata_copy.Pass();
 }
 
 scoped_refptr<TracingController::TraceDataSink>
 TracingController::CreateStringSink(
     const base::Callback<void(scoped_ptr<const base::DictionaryValue>,
                               base::RefCountedString*)>& callback) {
   return new StringTraceDataSink(new StringTraceDataEndpoint(callback));
 }
 
 scoped_refptr<TracingController::TraceDataSink>
@@ skipping 16 matching lines @@
   return new StringTraceDataEndpoint(callback);
 }
 
 scoped_refptr<TracingController::TraceDataEndpoint>
 TracingController::CreateFileEndpoint(const base::FilePath& file_path,
                                       const base::Closure& callback) {
   return new FileTraceDataEndpoint(file_path, callback);
 }
 
 }  // namespace content