Slow Reports - Embed Metadata in Traces

chrome/browser/tracing/crash_service_uploader.cc

 // Copyright 2014 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/tracing/crash_service_uploader.h"
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/format_macros.h"
@@ skipping 32 matching lines @@
     net::URLRequestContextGetter* request_context)
     : request_context_(request_context) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   const base::CommandLine& command_line =
       *base::CommandLine::ForCurrentProcess();
   std::string upload_url = kUploadURL;
   if (command_line.HasSwitch(switches::kTraceUploadURL)) {
     upload_url = command_line.GetSwitchValueASCII(switches::kTraceUploadURL);
   }
   SetUploadURL(upload_url);
+
+#if defined(OS_WIN)
+  const char product[] = "Chrome";
+#elif defined(OS_MACOSX)
+  const char product[] = "Chrome_Mac";
+#elif defined(OS_LINUX)
+  const char product[] = "Chrome_Linux";
+#elif defined(OS_ANDROID)
+  const char product[] = "Chrome_Android";
+#elif defined(OS_CHROMEOS)
+  const char product[] = "Chrome_ChromeOS";
+#else
+#error Platform not supported.
+#endif
+
+  // VersionInfo::ProductNameAndVersionForUserAgent() returns a string like
+  // "Chrome/aa.bb.cc.dd", split out the part before the "/".
+  chrome::VersionInfo version_info;
+  std::vector<std::string> product_components;
+  base::SplitString(version_info.ProductNameAndVersionForUserAgent(), '/',
+                    &product_components);
+  DCHECK_EQ(2U, product_components.size());
+  std::string version;
+  if (product_components.size() == 2U) {
+    version = product_components[1];
+  } else {
+    version = "unknown";
+  }
+
+  metadata_.insert(std::pair<std::string, std::string>("prod", product));
+  metadata_.insert(
+      std::pair<std::string, std::string>("ver", version + "-trace"));
+  metadata_.insert(std::pair<std::string, std::string>("guid", "0"));
+  metadata_.insert(std::pair<std::string, std::string>("type", "trace"));
+  // No minidump means no need for crash to process the report.
+  metadata_.insert(
+      std::pair<std::string, std::string>("should_process", "false"));
 }
 
 TraceCrashServiceUploader::~TraceCrashServiceUploader() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 }
 
 void TraceCrashServiceUploader::SetUploadURL(const std::string& url) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   upload_url_ = url;
 
   if (!GURL(upload_url_).is_valid())
     upload_url_.clear();
 }
 
+void TraceCrashServiceUploader::SetMetadata(
+    const std::map<std::string, std::string>& metadata) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  metadata_.insert(metadata.begin(), metadata.end());
+}
+
 void TraceCrashServiceUploader::OnURLFetchComplete(
     const net::URLFetcher* source) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   DCHECK_EQ(source, url_fetcher_.get());
   int response_code = source->GetResponseCode();
   string feedback;
   bool success = (response_code == kHttpResponseOk);
   if (success) {
     source->GetResponseAsString(&feedback);
   } else {
@@ skipping 20 matching lines @@
 }
 
 void TraceCrashServiceUploader::DoUpload(
     const std::string& file_contents,
     const UploadProgressCallback& progress_callback,
     const UploadDoneCallback& done_callback) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   content::BrowserThread::PostTask(
       content::BrowserThread::FILE, FROM_HERE,
       base::Bind(&TraceCrashServiceUploader::DoUploadOnFileThread,
-                 base::Unretained(this), file_contents, upload_url_,
+                 base::Unretained(this), file_contents, upload_url_, metadata_,
                  progress_callback, done_callback));
 }
 
 void TraceCrashServiceUploader::DoUploadOnFileThread(
     const std::string& file_contents,
     const std::string& upload_url,
+    const std::map<std::string, std::string>& metadata,
     const UploadProgressCallback& progress_callback,
     const UploadDoneCallback& done_callback) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::FILE);
   DCHECK(!url_fetcher_.get());
 
   progress_callback_ = progress_callback;
   done_callback_ = done_callback;
 
   if (upload_url.empty()) {
     OnUploadError("Upload URL empty or invalid");
     return;
   }
 
-#if defined(OS_WIN)
-  const char product[] = "Chrome";
-#elif defined(OS_MACOSX)
-  const char product[] = "Chrome_Mac";
-#elif defined(OS_LINUX)
-  const char product[] = "Chrome_Linux";
-#elif defined(OS_ANDROID)
-  const char product[] = "Chrome_Android";
-#elif defined(OS_CHROMEOS)
-  const char product[] = "Chrome_ChromeOS";
-#else
-#error Platform not supported.
-#endif
-
-  // VersionInfo::ProductNameAndVersionForUserAgent() returns a string like
-  // "Chrome/aa.bb.cc.dd", split out the part before the "/".
-  chrome::VersionInfo version_info;
-  std::vector<std::string> product_components;
-  base::SplitString(version_info.ProductNameAndVersionForUserAgent(), '/',
-                    &product_components);
-  DCHECK_EQ(2U, product_components.size());
-  std::string version;
-  if (product_components.size() == 2U) {
-    version = product_components[1];
-  } else {
-    version = "unknown";
-  }
-
   if (url_fetcher_.get()) {
     OnUploadError("Already uploading.");
     return;
   }
 
   scoped_ptr<char[]> compressed_contents(new char[kMaxUploadBytes]);
   int compressed_bytes;
   if (!Compress(file_contents, kMaxUploadBytes, compressed_contents.get(),
                 &compressed_bytes)) {
     OnUploadError("Compressing file failed.");
     return;
   }
 
   std::string post_data;
-  SetupMultipart(product, version, "trace.json.gz",
+  SetupMultipart(metadata, "trace.json.gz",
                  std::string(compressed_contents.get(), compressed_bytes),
                  &post_data);
 
   content::BrowserThread::PostTask(
       content::BrowserThread::UI, FROM_HERE,
       base::Bind(&TraceCrashServiceUploader::CreateAndStartURLFetcher,
                  base::Unretained(this), upload_url, post_data));
 }
 
 void TraceCrashServiceUploader::OnUploadError(std::string error_message) {
   LOG(ERROR) << error_message;
   content::BrowserThread::PostTask(
       content::BrowserThread::UI, FROM_HERE,
       base::Bind(done_callback_, false, error_message));
 }
 
 void TraceCrashServiceUploader::SetupMultipart(
-    const std::string& product,
-    const std::string& version,
+    const std::map<std::string, std::string>& metadata,
     const std::string& trace_filename,
     const std::string& trace_contents,
     std::string* post_data) {
-  net::AddMultipartValueForUpload("prod", product, kMultipartBoundary, "",
-                                  post_data);
-  net::AddMultipartValueForUpload("ver", version + "-trace", kMultipartBoundary,
-                                  "", post_data);
-  net::AddMultipartValueForUpload("guid", "0", kMultipartBoundary, "",
-                                  post_data);
-  net::AddMultipartValueForUpload("type", "trace", kMultipartBoundary, "",
-                                  post_data);
-  // No minidump means no need for crash to process the report.
-  net::AddMultipartValueForUpload("should_process", "false", kMultipartBoundary,
-                                  "", post_data);
+  for (auto it = metadata.begin(); it != metadata.end(); ++it) {
+    net::AddMultipartValueForUpload(it->first, it->second, kMultipartBoundary,
+                                    "", post_data);
+  }
 
   AddTraceFile(trace_filename, trace_contents, post_data);
 
   net::AddMultipartFinalDelimiterForUpload(kMultipartBoundary, post_data);
 }
 
 void TraceCrashServiceUploader::AddTraceFile(const std::string& trace_filename,
                                              const std::string& trace_contents,
                                              std::string* post_data) {
   post_data->append("--");
@@ skipping 49 matching lines @@
   std::string content_type = kUploadContentType;
   content_type.append("; boundary=");
   content_type.append(kMultipartBoundary);
 
   url_fetcher_ =
       net::URLFetcher::Create(GURL(upload_url), net::URLFetcher::POST, this);
   url_fetcher_->SetRequestContext(request_context_);
   url_fetcher_->SetUploadData(content_type, post_data);
   url_fetcher_->Start();
 }