Add data usage tracking for safe browsing

chrome/browser/safe_browsing/ping_manager.cc

 // Copyright (c) 2012 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/safe_browsing/ping_manager.h"
 
 #include <utility>
 
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/stl_util.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/safe_browsing/permission_reporter.h"
 #include "components/certificate_reporting/error_reporter.h"
+#include "components/data_use_measurement/core/data_use_user_data.h"
 #include "content/public/browser/browser_thread.h"
 #include "google_apis/google_api_keys.h"
 #include "net/base/escape.h"
 #include "net/base/load_flags.h"
 #include "net/ssl/ssl_info.h"
 #include "net/url_request/report_sender.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_status.h"
 #include "url/gurl.h"
@@ skipping 67 matching lines @@
 // Sends a SafeBrowsing "hit" report.
 void SafeBrowsingPingManager::ReportSafeBrowsingHit(
     const safe_browsing::HitReport& hit_report) {
   GURL report_url = SafeBrowsingHitUrl(hit_report);
   net::URLFetcher* report =
       net::URLFetcher::Create(report_url, hit_report.post_data.empty()
                                               ? net::URLFetcher::GET
                                               : net::URLFetcher::POST,
                               this)
           .release();
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      report, data_use_measurement::DataUseUserData::SAFE_BROWSING);
   report->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   report->SetRequestContext(request_context_getter_.get());
   if (!hit_report.post_data.empty())
     report->SetUploadData("text/plain", hit_report.post_data);
   safebrowsing_reports_.insert(report);
   report->Start();
 }
 
 // Sends threat details for users who opt-in.
 void SafeBrowsingPingManager::ReportThreatDetails(const std::string& report) {
   GURL report_url = ThreatDetailsUrl();
   net::URLFetcher* fetcher =
       net::URLFetcher::Create(report_url, net::URLFetcher::POST, this)
           .release();
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      fetcher, data_use_measurement::DataUseUserData::SAFE_BROWSING);

[Jialiu Lin, 2016/08/24 19:10:45]

In addition to these two pings, download pings are handled in
chrome/browser/safe_browsing/download_protection_service.cc, more specifically,

DownloadProtectionService::CheckClientDownloadRequest::SendRequest() and 
DownloadProtectionService::PPAPIDownloadRequest::SendRequest()

You might want to take a look.

[Raj, 2016/08/24 20:00:01]

Thanks. Added to that file as well.

   fetcher->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   fetcher->SetRequestContext(request_context_getter_.get());
   fetcher->SetUploadData("application/octet-stream", report);
   // Don't try too hard to send reports on failures.
   fetcher->SetAutomaticallyRetryOn5xx(false);
   fetcher->Start();
   safebrowsing_reports_.insert(fetcher);
 }
 
 void SafeBrowsingPingManager::ReportInvalidCertificateChain(
@@ skipping 108 matching lines @@
           version_.c_str());
   std::string api_key = google_apis::GetAPIKey();
   if (!api_key.empty()) {
     base::StringAppendF(&url, "&key=%s",
                         net::EscapeQueryParamValue(api_key, true).c_str());
   }
   return GURL(url);
 }
 
 }  // namespace safe_browsing