Add data usage tracking for safe browsing

chrome/browser/safe_browsing/client_side_detection_service.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/client_side_detection_service.h"
 
 #include <algorithm>
 #include <memory>
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/metrics/sparse_histogram.h"
 #include "base/single_thread_task_runner.h"
 #include "base/stl_util.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/safe_browsing/client_model.pb.h"
 #include "chrome/common/safe_browsing/csd.pb.h"
 #include "chrome/common/safe_browsing/safebrowsing_messages.h"
+#include "components/data_use_measurement/core/data_use_user_data.h"
 #include "components/prefs/pref_service.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_process_host.h"
 #include "crypto/sha2.h"
 #include "google_apis/google_api_keys.h"
 #include "net/base/escape.h"
 #include "net/base/ip_address.h"
 #include "net/base/load_flags.h"
@@ skipping 247 matching lines @@
     if (!callback.is_null())
       callback.Run(GURL(request->url()), false);
     return;
   }
 
   std::unique_ptr<net::URLFetcher> fetcher(
       net::URLFetcher::Create(0 /* ID used for testing */,
                               GetClientReportUrl(kClientReportPhishingUrl),
                               net::URLFetcher::POST, this));
   net::URLFetcher* fetcher_ptr = fetcher.get();
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      fetcher_ptr, data_use_measurement::DataUseUserData::SAFE_BROWSING);
 
   // Remember which callback and URL correspond to the current fetcher object.
   std::unique_ptr<ClientPhishingReportInfo> info(new ClientPhishingReportInfo);
   info->fetcher = std::move(fetcher);
   info->callback = callback;
   info->phishing_url = GURL(request->url());
   client_phishing_reports_[fetcher_ptr] = std::move(info);
 
   fetcher_ptr->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   fetcher_ptr->SetRequestContext(request_context_getter_.get());
@@ skipping 23 matching lines @@
     if (!callback.is_null())
       callback.Run(GURL(request->url()), GURL(request->url()), false);
     return;
   }
 
   std::unique_ptr<net::URLFetcher> fetcher(
       net::URLFetcher::Create(0 /* ID used for testing */,
                               GetClientReportUrl(kClientReportMalwareUrl),
                               net::URLFetcher::POST, this));
   net::URLFetcher* fetcher_ptr = fetcher.get();
+  data_use_measurement::DataUseUserData::AttachToFetcher(
+      fetcher_ptr, data_use_measurement::DataUseUserData::SAFE_BROWSING);
 
   // Remember which callback and URL correspond to the current fetcher object.
   std::unique_ptr<ClientMalwareReportInfo> info(new ClientMalwareReportInfo);
   info->fetcher = std::move(fetcher);
   info->callback = callback;
   info->original_url = GURL(request->url());
   client_malware_reports_[fetcher_ptr] = std::move(info);
 
   fetcher_ptr->SetLoadFlags(net::LOAD_DISABLE_CACHE);
   fetcher_ptr->SetRequestContext(request_context_getter_.get());
@@ skipping 166 matching lines @@
     const std::string& report_url) {
   GURL url(report_url);
   std::string api_key = google_apis::GetAPIKey();
   if (!api_key.empty())
     url = url.Resolve("?key=" + net::EscapeQueryParamValue(api_key, true));
 
   return url;
 }
 
 }  // namespace safe_browsing