Expose malware/phishing/etc. distinction from SafeBrowsingUIManager

chrome/browser/safe_browsing/ui_manager.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.
 //
 // The Safe Browsing service is responsible for downloading anti-phishing and
 // anti-malware tables and checking urls against them.
 
 #ifndef CHROME_BROWSER_SAFE_BROWSING_UI_MANAGER_H_
 #define CHROME_BROWSER_SAFE_BROWSING_UI_MANAGER_H_
 
@@ skipping 117 matching lines @@
   // A convenience wrapper method for IsUrlWhitelistedOrPendingForWebContents.
   bool IsWhitelisted(const UnsafeResource& resource);
 
   // Checks if we already displayed or are displaying an interstitial
   // for the top-level site |url| in a given WebContents. If
   // |whitelist_only|, it returns true only if the user chose to ignore
   // the interstitial. Otherwise, it returns true if an interstitial for
   // |url| is already displaying *or* if the user has seen an
   // interstitial for |url| before in this WebContents and proceeded
   // through it. Called on the UI thread.
+  //
+  // If the resource was found in the whitelist or pending for the
+  // whitelist, |threat_type| will be set to the SBThreatType for which
+  // the URL was first whitelisted.
   bool IsUrlWhitelistedOrPendingForWebContents(
       const GURL& url,
       bool is_subresource,
       content::NavigationEntry* entry,
       content::WebContents* web_contents,
-      bool whitelist_only);
+      bool whitelist_only,
+      SBThreatType* threat_type);
 
   // The blocking page for |web_contents| on the UI thread has
   // completed, with |proceed| set to true if the user has chosen to
   // proceed through the blocking page and false
   // otherwise. |web_contents| is the WebContents that was displaying
   // the blocking page. |main_frame_url| is the top-level URL on which
   // the blocking page was displayed. If |proceed| is true,
   // |main_frame_url| is whitelisted so that the user will not see
   // another warning for that URL in this WebContents.
   void OnBlockingPageDone(const std::vector<UnsafeResource>& resources,
@@ skipping 53 matching lines @@
   void ReportInvalidCertificateChainOnIOThread(
       const std::string& serialized_report);
 
   // Report permission action to SafeBrowsing servers.
   void ReportPermissionActionOnIOThread(
       const PermissionReportInfo& report_info);
 
   // Updates the whitelist URL set for |web_contents|. Called on the UI thread.
   void AddToWhitelistUrlSet(const GURL& whitelist_url,
                             content::WebContents* web_contents,
-                            bool is_pending);
+                            bool is_pending,
+                            SBThreatType threat_type);
 
   // Removes |whitelist_url| from the pending whitelist for
   // |web_contents|. Called on the UI thread.
   void RemoveFromPendingWhitelistUrlSet(const GURL& whitelist_url,
                                         content::WebContents* web_contents);
 
   static GURL GetMainFrameWhitelistUrlForResourceForTesting(
       const safe_browsing::SafeBrowsingUIManager::UnsafeResource& resource);
 
   // Safebrowsing service.
   scoped_refptr<SafeBrowsingService> sb_service_;
 
   base::ObserverList<Observer> observer_list_;
 
   DISALLOW_COPY_AND_ASSIGN(SafeBrowsingUIManager);
 };
 
 }  // namespace safe_browsing
 
 #endif  // CHROME_BROWSER_SAFE_BROWSING_UI_MANAGER_H_