Wire up download attribution enable finch experiment

chrome/browser/safe_browsing/download_protection_service.h

 // 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.
 //
 // Helper class which handles communication with the SafeBrowsing servers for
 // improved binary download protection.
 
 #ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_
 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_
 
 #include <stdint.h>
 
 #include <memory>
 #include <set>
 #include <string>
 #include <unordered_map>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/callback_list.h"
 #include "base/files/file_path.h"
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/supports_user_data.h"
+#include "chrome/browser/safe_browsing/safe_browsing_navigation_observer_manager.h"
 #include "chrome/browser/safe_browsing/ui_manager.h"
 #include "components/safe_browsing_db/database_manager.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "url/gurl.h"
 
 
 namespace content {
 class DownloadItem;
 class PageNavigator;
 }
@@ skipping 244 matching lines @@
   // list of strings that need to be checked against the download whitelist to
   // determine whether the certificate is whitelisted.
   static void GetCertificateWhitelistStrings(
       const net::X509Certificate& certificate,
       const net::X509Certificate& issuer,
       std::vector<std::string>* whitelist_strings);
 
   // Returns the URL that will be used for download requests.
   static GURL GetDownloadRequestUrl();
 
+  // If kDownloadAttribution feature is enabled, identify and add referrer chain
+  // info of a download to ClientDownloadRequest proto. This function also
+  // records UMA stats of download attribution result.
+  void AddReferrerChainToClientDownloadRequest(
+    const GURL& download_url,
+    content::WebContents* web_contents,
+    ClientDownloadRequest* out_request);
+
   // These pointers may be NULL if SafeBrowsing is disabled.
   scoped_refptr<SafeBrowsingUIManager> ui_manager_;
   scoped_refptr<SafeBrowsingDatabaseManager> database_manager_;
+  scoped_refptr<SafeBrowsingNavigationObserverManager>
+  navigation_observer_manager_;
 
   // The context we use to issue network requests.
   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
 
   // Set of pending server requests for DownloadManager mediated downloads.
   std::set<scoped_refptr<CheckClientDownloadRequest>> download_requests_;
 
   // Set of pending server requests for PPAPI mediated downloads. Using a map
   // because heterogeneous lookups aren't available yet in std::unordered_map.
   std::unordered_map<PPAPIDownloadRequest*,
@@ skipping 23 matching lines @@
   std::set<std::string> manual_blacklist_hashes_;
 
   // Rate of whitelisted downloads we sample to send out download ping.
   double whitelist_sample_rate_;
 
   DISALLOW_COPY_AND_ASSIGN(DownloadProtectionService);
 };
 }  // namespace safe_browsing
 
 #endif  // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_PROTECTION_SERVICE_H_