Chromium Code Reviews| Index: chrome/browser/loader/safe_browsing_resource_throttle.cc |
| diff --git a/chrome/browser/loader/safe_browsing_resource_throttle.cc b/chrome/browser/loader/safe_browsing_resource_throttle.cc |
| index b8a7c538393a10609b71cf8c9f1b59991f6de7f4..936fed6dd405e277abd495533ab7b4b4201d9bc2 100644 |
| --- a/chrome/browser/loader/safe_browsing_resource_throttle.cc |
| +++ b/chrome/browser/loader/safe_browsing_resource_throttle.cc |
| @@ -15,6 +15,7 @@ |
| #include "chrome/browser/prerender/prerender_contents.h" |
| #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| #include "components/safe_browsing_db/util.h" |
| +#include "components/safe_browsing_db/v4_local_database_manager.h" |
| #include "components/subresource_filter/content/browser/content_subresource_filter_driver_factory.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/render_frame_host.h" |
| @@ -96,7 +97,8 @@ SafeBrowsingResourceThrottle::SafeBrowsingResourceThrottle( |
| resource_type_(resource_type), |
| net_log_with_source_( |
| net::NetLogWithSource::Make(request->net_log().net_log(), |
| - NetLogSourceType::SAFE_BROWSING)) {} |
| + NetLogSourceType::SAFE_BROWSING)), |
| + v4_local_database_manager_(sb_service->v4_local_database_manager()) {} |
| SafeBrowsingResourceThrottle::~SafeBrowsingResourceThrottle() { |
| if (defer_state_ != DEFERRED_NONE) { |
| @@ -107,6 +109,10 @@ SafeBrowsingResourceThrottle::~SafeBrowsingResourceThrottle() { |
| database_manager_->CancelCheck(this); |
| EndNetLogEvent(NetLogEventType::SAFE_BROWSING_CHECKING_URL, "result", |
| "request_canceled"); |
| + |
| + if (v4_local_database_manager_) { |
| + v4_local_database_manager_->CancelCheck(this); |
| + } |
|
Scott Hess - ex-Googler
2016/09/27 22:24:53
Move this above the log event.
vakh (use Gerrit instead)
2016/09/28 01:24:33
Done.
|
| } |
| } |
| @@ -373,6 +379,11 @@ bool SafeBrowsingResourceThrottle::CheckUrl(const GURL& url) { |
| UMA_HISTOGRAM_ENUMERATION("SB2.ResourceTypes2.Checked", resource_type_, |
| content::RESOURCE_TYPE_LAST_TYPE); |
| + if (v4_local_database_manager_ && |
| + v4_local_database_manager_->CanCheckResourceType(resource_type_)) { |
| + v4_local_database_manager_->CheckBrowseUrl(url, this); |
| + } |
| + |
| if (succeeded_synchronously) { |
| threat_type_ = safe_browsing::SB_THREAT_TYPE_SAFE; |
| ui_manager_->LogPauseDelay(base::TimeDelta()); // No delay. |
| @@ -400,6 +411,9 @@ void SafeBrowsingResourceThrottle::OnCheckUrlTimeout() { |
| database_manager_->CancelCheck(this); |
| OnCheckBrowseUrlResult(url_being_checked_, safe_browsing::SB_THREAT_TYPE_SAFE, |
| safe_browsing::ThreatMetadata()); |
| + if (v4_local_database_manager_) { |
| + v4_local_database_manager_->CancelCheck(this); |
| + } |
|
Scott Hess - ex-Googler
2016/09/27 22:24:53
Probably move this above the response, so that bot
vakh (use Gerrit instead)
2016/09/28 01:24:33
Done.
|
| } |
| void SafeBrowsingResourceThrottle::ResumeRequest() { |