Wireup SafeBrowsingNavigationObserverManager to help PPAPI download attribution

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_
 
@@ skipping 111 matching lines @@
                                 const CheckDownloadCallback& callback);
 
   // Returns true iff the download specified by |info| should be scanned by
   // CheckClientDownload() for malicious content.
   virtual bool IsSupportedDownload(
       const content::DownloadItem& item,
       const base::FilePath& target_path) const;
 
   virtual void CheckPPAPIDownloadRequest(
       const GURL& requestor_url,
+      const GURL& initiating_frame_url,
+      content::WebContents* web_contents,
       const base::FilePath& default_file_path,
       const std::vector<base::FilePath::StringType>& alternate_extensions,
       Profile* profile,
       const CheckDownloadCallback& callback);
 
   // Display more information to the user regarding the download specified by
   // |info|. This method is invoked when the user requests more information
   // about a download that was marked as malicious.
   void ShowDetailsForDownload(const content::DownloadItem& item,
                               content::PageNavigator* navigator);
@@ skipping 24 matching lines @@
 
   // Registers a callback that will be run when a PPAPI ClientDownloadRequest
   // has been formed.
   PPAPIDownloadRequestSubscription RegisterPPAPIDownloadRequestCallback(
       const PPAPIDownloadRequestCallback& callback);
 
   double whitelist_sample_rate() const {
     return whitelist_sample_rate_;
   }
 
+  scoped_refptr<SafeBrowsingNavigationObserverManager>
+  navigation_observer_manager() {
+    return navigation_observer_manager_;
+  }
+
   static void SetDownloadPingToken(content::DownloadItem* item,
                                    const std::string& token);
 
   static std::string GetDownloadPingToken(const content::DownloadItem* item);
 
  protected:
   // Enum to keep track why a particular download verdict was chosen.
   // Used for UMA metrics. Do not reorder.
   enum DownloadCheckResultReason {
     REASON_INVALID_URL = 0,
@@ skipping 103 matching lines @@
   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);
 
+  // If kDownloadAttribution feature is enabled, identify referrer chain of the
+  // PPAPI download based on the frame URL where the download is initiated.
+  // Then add referrer chain info to ClientDownloadRequest proto. This function
+  // also records UMA stats of download attribution result.
+  void AddReferrerChainToPPAPIClientDownloadRequest(
+    const GURL& initiating_frame_url,
+    int tab_id,
+    bool has_user_gesture,
+    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.
@@ skipping 28 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_