Add data usage tracking for safe browsing

chrome/browser/safe_browsing/protocol_manager.cc

Index: chrome/browser/safe_browsing/protocol_manager.cc
diff --git a/chrome/browser/safe_browsing/protocol_manager.cc b/chrome/browser/safe_browsing/protocol_manager.cc
index e8c600d0913ac96cbd364096bda05c6e8165a56c..85279062dc54eb16b89b879cb41870a5654a57fa 100644
--- a/chrome/browser/safe_browsing/protocol_manager.cc
+++ b/chrome/browser/safe_browsing/protocol_manager.cc
@@ -221,6 +221,8 @@ void SafeBrowsingProtocolManager::GetFullHash(
       net::URLFetcher::Create(url_fetcher_id_++, gethash_url,
                               net::URLFetcher::POST, this)
           .release();
+  data_use_measurement::DataUseUserData::AttachToFetcher(

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

We are going to deprecate this protocol manager soon. New protocol manager (pv4)
will be in components/safe_browsing_db/* code path. 

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

Thanks for the heads up.
Added to that code as well.

+      fetcher, data_use_measurement::DataUseUserData::SAFE_BROWSING);
   hash_requests_[fetcher] = FullHashDetails(callback, is_download);
 
   const std::string get_hash = FormatGetHash(prefixes);
@@ -589,6 +591,8 @@ bool SafeBrowsingProtocolManager::IssueBackupUpdateRequest(
   GURL backup_update_url = BackupUpdateUrl(backup_update_reason);
   request_ = net::URLFetcher::Create(url_fetcher_id_++, backup_update_url,
                                      net::URLFetcher::POST, this);
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      request_.get(), data_use_measurement::DataUseUserData::SAFE_BROWSING);
   request_->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   request_->SetRequestContext(request_context_getter_.get());
   request_->SetUploadData("text/plain", update_list_data_);
@@ -616,6 +620,8 @@ void SafeBrowsingProtocolManager::IssueChunkRequest() {
   request_type_ = CHUNK_REQUEST;
   request_ = net::URLFetcher::Create(url_fetcher_id_++, chunk_url,
                                      net::URLFetcher::GET, this);
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      request_.get(), data_use_measurement::DataUseUserData::SAFE_BROWSING);
   request_->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   request_->SetRequestContext(request_context_getter_.get());
   chunk_request_start_ = base::Time::Now();
@@ -667,6 +673,8 @@ void SafeBrowsingProtocolManager::OnGetChunksComplete(
   GURL update_url = UpdateUrl(is_extended_reporting);
   request_ = net::URLFetcher::Create(url_fetcher_id_++, update_url,
                                      net::URLFetcher::POST, this);
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      request_.get(), data_use_measurement::DataUseUserData::SAFE_BROWSING);
   request_->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   request_->SetRequestContext(request_context_getter_.get());
   request_->SetUploadData("text/plain", update_list_data_);