chrome/browser/safe_browsing/ping_manager.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_SAFE_BROWSING_PING_MANAGER_H_
   6 #define CHROME_BROWSER_SAFE_BROWSING_PING_MANAGER_H_
   7
   8 // A class that reports safebrowsing statistics to Google's SafeBrowsing
   9 // servers.
  10 #include <set>
  11 #include <string>
  12 #include <vector>
  13
  14 #include "base/gtest_prod_util.h"
  15 #include "base/memory/scoped_ptr.h"
  16 #include "chrome/browser/safe_browsing/protocol_manager_helper.h"
  17 #include "chrome/browser/safe_browsing/safe_browsing_util.h"
  18 #include "net/url_request/url_fetcher_delegate.h"
  19 #include "url/gurl.h"
  20
  21 namespace chrome_browser_net {
  22 class CertificateErrorReporter;
  23 }
  24
  25 namespace net {
  26 class SSLInfo;
  27 class URLRequestContextGetter;
  28 }  // namespace net
  29
  30
  31 class SafeBrowsingPingManager : public net::URLFetcherDelegate {
  32  public:
  33   ~SafeBrowsingPingManager() override;
  34
  35   // Create an instance of the safe browsing ping manager.
  36   static SafeBrowsingPingManager* Create(
  37       net::URLRequestContextGetter* request_context_getter,
  38       const SafeBrowsingProtocolConfig& config);
  39
  40   // net::URLFetcherDelegate interface.
  41   void OnURLFetchComplete(const net::URLFetcher* source) override;
  42
  43   // For UMA users we report to Google when a SafeBrowsing interstitial is shown
  44   // to the user.  |threat_type| should be one of the types known by
  45   // SafeBrowsingHitUrl.
  46   void ReportSafeBrowsingHit(const GURL& malicious_url,
  47                              const GURL& page_url,
  48                              const GURL& referrer_url,
  49                              bool is_subresource,
  50                              SBThreatType threat_type,
  51                              const std::string& post_data);
  52
  53   // Users can opt-in on the SafeBrowsing interstitial to send detailed
  54   // malware reports. |report| is the serialized report.
  55   void ReportMalwareDetails(const std::string& report);
  56
  57   // Users can opt-in on the SSL interstitial to send reports of invalid
  58   // certificate chains.
  59   void ReportInvalidCertificateChain(const std::string& hostname,
  60                                      const net::SSLInfo& ssl_info);
  61
  62   void SetCertificateErrorReporterForTesting(scoped_ptr<
  63       chrome_browser_net::CertificateErrorReporter> certificate_error_reporter);
  64
  65  private:
  66   FRIEND_TEST_ALL_PREFIXES(SafeBrowsingPingManagerTest,
  67                            TestSafeBrowsingHitUrl);
  68   FRIEND_TEST_ALL_PREFIXES(SafeBrowsingPingManagerTest,
  69                            TestMalwareDetailsUrl);
  70
  71   typedef std::set<const net::URLFetcher*> Reports;
  72
  73   // Constructs a SafeBrowsingPingManager that issues network requests
  74   // using |request_context_getter|.
  75   SafeBrowsingPingManager(
  76       net::URLRequestContextGetter* request_context_getter,
  77       const SafeBrowsingProtocolConfig& config);
  78
  79   // Generates URL for reporting safe browsing hits for UMA users.
  80   GURL SafeBrowsingHitUrl(
  81       const GURL& malicious_url, const GURL& page_url, const GURL& referrer_url,
  82       bool is_subresource,
  83       SBThreatType threat_type) const;
  84   // Generates URL for reporting malware details for users who opt-in.
  85   GURL MalwareDetailsUrl() const;
  86
  87   // Current product version sent in each request.
  88   std::string version_;
  89
  90   // The safe browsing client name sent in each request.
  91   std::string client_name_;
  92
  93   // The context we use to issue network requests.
  94   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
  95
  96   // URL prefix where browser reports hits to the safebrowsing list and
  97   // sends detaild malware reports for UMA users.
  98   std::string url_prefix_;
  99
 100   // Track outstanding SafeBrowsing report fetchers for clean up.
 101   // We add both "hit" and "detail" fetchers in this set.
 102   Reports safebrowsing_reports_;
 103
 104   // Sends reports of invalid SSL certificate chains.
 105   scoped_ptr<chrome_browser_net::CertificateErrorReporter>
 106       certificate_error_reporter_;
 107
 108   DISALLOW_COPY_AND_ASSIGN(SafeBrowsingPingManager);
 109 };
 110
 111 #endif  // CHROME_BROWSER_SAFE_BROWSING_PING_MANAGER_H_