Downgrade security state after user clicks through SB interstitial

chrome/browser/safe_browsing/ui_manager.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/ui_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
 #include "base/debug/leak_tracker.h"
@@ skipping 353 matching lines @@
     if (!entry)
       return;
     whitelisted_url = entry->GetURL();
   } else {
     whitelisted_url = resource.url;
   }
 
   site_list->Insert(whitelisted_url);
 }
 
+bool SafeBrowsingUIManager::IsWhitelisted(const UnsafeResource& resource) {
+  NavigationEntry* entry = nullptr;
+  if (resource.is_subresource) {
+    entry = resource.GetNavigationEntryForResource();
+  }
+  return IsUrlWhitelistedForWebContents(resource.url, resource.is_subresource,
+                                        entry,
+                                        resource.web_contents_getter.Run());
+}
+
 // Check if the user has already ignored a SB warning for this WebContents and
 // top-level domain.
-bool SafeBrowsingUIManager::IsWhitelisted(const UnsafeResource& resource) {
+bool SafeBrowsingUIManager::IsUrlWhitelistedForWebContents(
+    const GURL& url,
+    bool is_subresource,
+    NavigationEntry* entry,
+    content::WebContents* web_contents) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   GURL maybe_whitelisted_url;
-  if (resource.is_subresource) {
-    NavigationEntry* entry = resource.GetNavigationEntryForResource();
+  if (is_subresource) {
     if (!entry)
       return false;
     maybe_whitelisted_url = entry->GetURL();
   } else {
-    maybe_whitelisted_url = resource.url;
+    maybe_whitelisted_url = url;
   }
 
-  WhitelistUrlSet* site_list = static_cast<WhitelistUrlSet*>(
-      resource.web_contents_getter.Run()->GetUserData(kWhitelistKey));
+  WhitelistUrlSet* site_list =
+      static_cast<WhitelistUrlSet*>(web_contents->GetUserData(kWhitelistKey));
   if (!site_list)
     return false;
   return site_list->Contains(maybe_whitelisted_url);
 }
 
 }  // namespace safe_browsing