OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/loader/safe_browsing_resource_throttle.h" | 5 #include "chrome/browser/loader/safe_browsing_resource_throttle.h" |
6 | 6 |
7 #include <iterator> | 7 #include <iterator> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/metrics/histogram_macros.h" | 11 #include "base/metrics/histogram_macros.h" |
12 #include "base/trace_event/trace_event.h" | 12 #include "base/trace_event/trace_event.h" |
13 #include "base/values.h" | 13 #include "base/values.h" |
14 #include "chrome/browser/browser_process.h" | 14 #include "chrome/browser/browser_process.h" |
15 #include "chrome/browser/prerender/prerender_contents.h" | 15 #include "chrome/browser/prerender/prerender_contents.h" |
16 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 16 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
17 #include "components/safe_browsing_db/util.h" | 17 #include "components/safe_browsing_db/util.h" |
18 #include "components/safe_browsing_db/v4_feature_list.h" | |
19 #include "components/safe_browsing_db/v4_local_database_manager.h" | |
20 #include "components/subresource_filter/content/browser/content_subresource_filt
er_driver_factory.h" | 18 #include "components/subresource_filter/content/browser/content_subresource_filt
er_driver_factory.h" |
21 #include "content/public/browser/browser_thread.h" | 19 #include "content/public/browser/browser_thread.h" |
22 #include "content/public/browser/render_frame_host.h" | 20 #include "content/public/browser/render_frame_host.h" |
23 #include "content/public/browser/resource_controller.h" | 21 #include "content/public/browser/resource_controller.h" |
24 #include "content/public/browser/resource_request_info.h" | 22 #include "content/public/browser/resource_request_info.h" |
25 #include "content/public/browser/web_contents.h" | 23 #include "content/public/browser/web_contents.h" |
26 #include "net/base/load_flags.h" | 24 #include "net/base/load_flags.h" |
27 #include "net/log/net_log.h" | 25 #include "net/log/net_log.h" |
28 #include "net/log/net_log_source_type.h" | 26 #include "net/log/net_log_source_type.h" |
29 #include "net/url_request/redirect_info.h" | 27 #include "net/url_request/redirect_info.h" |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 safe_browsing::SafeBrowsingService* sb_service) | 89 safe_browsing::SafeBrowsingService* sb_service) |
92 : state_(STATE_NONE), | 90 : state_(STATE_NONE), |
93 defer_state_(DEFERRED_NONE), | 91 defer_state_(DEFERRED_NONE), |
94 threat_type_(safe_browsing::SB_THREAT_TYPE_SAFE), | 92 threat_type_(safe_browsing::SB_THREAT_TYPE_SAFE), |
95 database_manager_(sb_service->database_manager()), | 93 database_manager_(sb_service->database_manager()), |
96 ui_manager_(sb_service->ui_manager()), | 94 ui_manager_(sb_service->ui_manager()), |
97 request_(request), | 95 request_(request), |
98 resource_type_(resource_type), | 96 resource_type_(resource_type), |
99 net_log_with_source_( | 97 net_log_with_source_( |
100 net::NetLogWithSource::Make(request->net_log().net_log(), | 98 net::NetLogWithSource::Make(request->net_log().net_log(), |
101 NetLogSourceType::SAFE_BROWSING)), | 99 NetLogSourceType::SAFE_BROWSING)) {} |
102 v4_local_database_manager_(sb_service->v4_local_database_manager()) {} | |
103 | 100 |
104 SafeBrowsingResourceThrottle::~SafeBrowsingResourceThrottle() { | 101 SafeBrowsingResourceThrottle::~SafeBrowsingResourceThrottle() { |
105 if (defer_state_ != DEFERRED_NONE) { | 102 if (defer_state_ != DEFERRED_NONE) { |
106 EndNetLogEvent(NetLogEventType::SAFE_BROWSING_DEFERRED, nullptr, nullptr); | 103 EndNetLogEvent(NetLogEventType::SAFE_BROWSING_DEFERRED, nullptr, nullptr); |
107 } | 104 } |
108 | 105 |
109 if (state_ == STATE_CHECKING_URL) { | 106 if (state_ == STATE_CHECKING_URL) { |
110 database_manager_->CancelCheck(this); | 107 database_manager_->CancelCheck(this); |
111 if (safe_browsing::V4FeatureList::IsParallelCheckEnabled()) { | |
112 v4_local_database_manager_->CancelCheck(this); | |
113 } | |
114 | |
115 EndNetLogEvent(NetLogEventType::SAFE_BROWSING_CHECKING_URL, "result", | 108 EndNetLogEvent(NetLogEventType::SAFE_BROWSING_CHECKING_URL, "result", |
116 "request_canceled"); | 109 "request_canceled"); |
117 } | 110 } |
118 } | 111 } |
119 | 112 |
120 // Note on net_log calls: SAFE_BROWSING_DEFERRED events must be wholly | 113 // Note on net_log calls: SAFE_BROWSING_DEFERRED events must be wholly |
121 // nested within SAFE_BROWSING_CHECKING_URL events. Synchronous checks | 114 // nested within SAFE_BROWSING_CHECKING_URL events. Synchronous checks |
122 // are not logged at all. | 115 // are not logged at all. |
123 void SafeBrowsingResourceThrottle::BeginNetLogEvent(NetLogEventType type, | 116 void SafeBrowsingResourceThrottle::BeginNetLogEvent(NetLogEventType type, |
124 const GURL& url, | 117 const GURL& url, |
(...skipping 248 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
373 if (!database_manager_->CanCheckResourceType(resource_type_)) { | 366 if (!database_manager_->CanCheckResourceType(resource_type_)) { |
374 UMA_HISTOGRAM_ENUMERATION("SB2.ResourceTypes2.Skipped", resource_type_, | 367 UMA_HISTOGRAM_ENUMERATION("SB2.ResourceTypes2.Skipped", resource_type_, |
375 content::RESOURCE_TYPE_LAST_TYPE); | 368 content::RESOURCE_TYPE_LAST_TYPE); |
376 return true; | 369 return true; |
377 } | 370 } |
378 | 371 |
379 bool succeeded_synchronously = database_manager_->CheckBrowseUrl(url, this); | 372 bool succeeded_synchronously = database_manager_->CheckBrowseUrl(url, this); |
380 UMA_HISTOGRAM_ENUMERATION("SB2.ResourceTypes2.Checked", resource_type_, | 373 UMA_HISTOGRAM_ENUMERATION("SB2.ResourceTypes2.Checked", resource_type_, |
381 content::RESOURCE_TYPE_LAST_TYPE); | 374 content::RESOURCE_TYPE_LAST_TYPE); |
382 | 375 |
383 if (safe_browsing::V4FeatureList::IsParallelCheckEnabled() && | |
384 v4_local_database_manager_->CanCheckResourceType(resource_type_)) { | |
385 v4_local_database_manager_->CheckBrowseUrl(url, this); | |
386 } | |
387 | |
388 if (succeeded_synchronously) { | 376 if (succeeded_synchronously) { |
389 threat_type_ = safe_browsing::SB_THREAT_TYPE_SAFE; | 377 threat_type_ = safe_browsing::SB_THREAT_TYPE_SAFE; |
390 ui_manager_->LogPauseDelay(base::TimeDelta()); // No delay. | 378 ui_manager_->LogPauseDelay(base::TimeDelta()); // No delay. |
391 return true; | 379 return true; |
392 } | 380 } |
393 | 381 |
394 state_ = STATE_CHECKING_URL; | 382 state_ = STATE_CHECKING_URL; |
395 url_being_checked_ = url; | 383 url_being_checked_ = url; |
396 BeginNetLogEvent(NetLogEventType::SAFE_BROWSING_CHECKING_URL, url, nullptr, | 384 BeginNetLogEvent(NetLogEventType::SAFE_BROWSING_CHECKING_URL, url, nullptr, |
397 nullptr); | 385 nullptr); |
398 | 386 |
399 // Start a timer to abort the check if it takes too long. | 387 // Start a timer to abort the check if it takes too long. |
400 // TODO(nparker): Set this only when we defer, based on remaining time, | 388 // TODO(nparker): Set this only when we defer, based on remaining time, |
401 // so we don't cancel earlier than necessary. | 389 // so we don't cancel earlier than necessary. |
402 timer_.Start(FROM_HERE, | 390 timer_.Start(FROM_HERE, |
403 base::TimeDelta::FromMilliseconds(kCheckUrlTimeoutMs), | 391 base::TimeDelta::FromMilliseconds(kCheckUrlTimeoutMs), |
404 this, &SafeBrowsingResourceThrottle::OnCheckUrlTimeout); | 392 this, &SafeBrowsingResourceThrottle::OnCheckUrlTimeout); |
405 | 393 |
406 return false; | 394 return false; |
407 } | 395 } |
408 | 396 |
409 void SafeBrowsingResourceThrottle::OnCheckUrlTimeout() { | 397 void SafeBrowsingResourceThrottle::OnCheckUrlTimeout() { |
410 CHECK_EQ(state_, STATE_CHECKING_URL); | 398 CHECK_EQ(state_, STATE_CHECKING_URL); |
411 | 399 |
412 database_manager_->CancelCheck(this); | 400 database_manager_->CancelCheck(this); |
413 if (safe_browsing::V4FeatureList::IsParallelCheckEnabled()) { | |
414 v4_local_database_manager_->CancelCheck(this); | |
415 } | |
416 OnCheckBrowseUrlResult(url_being_checked_, safe_browsing::SB_THREAT_TYPE_SAFE, | 401 OnCheckBrowseUrlResult(url_being_checked_, safe_browsing::SB_THREAT_TYPE_SAFE, |
417 safe_browsing::ThreatMetadata()); | 402 safe_browsing::ThreatMetadata()); |
418 } | 403 } |
419 | 404 |
420 void SafeBrowsingResourceThrottle::ResumeRequest() { | 405 void SafeBrowsingResourceThrottle::ResumeRequest() { |
421 CHECK_EQ(state_, STATE_NONE); | 406 CHECK_EQ(state_, STATE_NONE); |
422 CHECK_NE(defer_state_, DEFERRED_NONE); | 407 CHECK_NE(defer_state_, DEFERRED_NONE); |
423 | 408 |
424 bool resume = true; | 409 bool resume = true; |
425 if (defer_state_ == DEFERRED_UNCHECKED_REDIRECT) { | 410 if (defer_state_ == DEFERRED_UNCHECKED_REDIRECT) { |
426 // Save the redirect urls for possible malware detail reporting later. | 411 // Save the redirect urls for possible malware detail reporting later. |
427 redirect_urls_.push_back(unchecked_redirect_url_); | 412 redirect_urls_.push_back(unchecked_redirect_url_); |
428 if (!CheckUrl(unchecked_redirect_url_)) { | 413 if (!CheckUrl(unchecked_redirect_url_)) { |
429 // We're now waiting for the unchecked_redirect_url_. | 414 // We're now waiting for the unchecked_redirect_url_. |
430 defer_state_ = DEFERRED_REDIRECT; | 415 defer_state_ = DEFERRED_REDIRECT; |
431 resume = false; | 416 resume = false; |
432 BeginNetLogEvent(NetLogEventType::SAFE_BROWSING_DEFERRED, | 417 BeginNetLogEvent(NetLogEventType::SAFE_BROWSING_DEFERRED, |
433 unchecked_redirect_url_, "defer_reason", | 418 unchecked_redirect_url_, "defer_reason", |
434 "resumed_redirect"); | 419 "resumed_redirect"); |
435 } | 420 } |
436 } | 421 } |
437 | 422 |
438 if (resume) { | 423 if (resume) { |
439 defer_state_ = DEFERRED_NONE; | 424 defer_state_ = DEFERRED_NONE; |
440 controller()->Resume(); | 425 controller()->Resume(); |
441 } | 426 } |
442 } | 427 } |
OLD | NEW |