| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/safe_browsing/client_side_detection_host.h" | 5 #include "chrome/browser/safe_browsing/client_side_detection_host.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| 11 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/scoped_ptr.h" |
| 12 #include "base/metrics/histogram.h" | 12 #include "base/metrics/histogram.h" |
| 13 #include "base/task.h" | 13 #include "base/task.h" |
| 14 #include "chrome/browser/browser_process.h" | 14 #include "chrome/browser/browser_process.h" |
| 15 #include "chrome/browser/prefs/pref_service.h" | 15 #include "chrome/browser/prefs/pref_service.h" |
| 16 #include "chrome/browser/profiles/profile.h" | 16 #include "chrome/browser/profiles/profile.h" |
| 17 #include "chrome/browser/safe_browsing/browser_feature_extractor.h" | 17 #include "chrome/browser/safe_browsing/browser_feature_extractor.h" |
| 18 #include "chrome/browser/safe_browsing/client_side_detection_service.h" | 18 #include "chrome/browser/safe_browsing/client_side_detection_service.h" |
| 19 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 19 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 20 #include "chrome/common/chrome_switches.h" | 20 #include "chrome/common/chrome_switches.h" |
| 21 #include "chrome/common/pref_names.h" | 21 #include "chrome/common/pref_names.h" |
| 22 #include "chrome/common/safe_browsing/csd.pb.h" | 22 #include "chrome/common/safe_browsing/csd.pb.h" |
| 23 #include "chrome/common/safe_browsing/safebrowsing_messages.h" | 23 #include "chrome/common/safe_browsing/safebrowsing_messages.h" |
| 24 #include "content/browser/renderer_host/render_process_host.h" | |
| 25 #include "content/browser/renderer_host/render_view_host.h" | 24 #include "content/browser/renderer_host/render_view_host.h" |
| 26 #include "content/browser/renderer_host/render_view_host_delegate.h" | 25 #include "content/browser/renderer_host/render_view_host_delegate.h" |
| 27 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 26 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 28 #include "content/browser/renderer_host/resource_request_details.h" | 27 #include "content/browser/renderer_host/resource_request_details.h" |
| 29 #include "content/browser/tab_contents/navigation_details.h" | 28 #include "content/browser/tab_contents/navigation_details.h" |
| 30 #include "content/browser/tab_contents/tab_contents.h" | 29 #include "content/browser/tab_contents/tab_contents.h" |
| 31 #include "content/common/view_messages.h" | 30 #include "content/common/view_messages.h" |
| 32 #include "content/public/browser/browser_thread.h" | 31 #include "content/public/browser/browser_thread.h" |
| 33 #include "content/public/browser/notification_details.h" | 32 #include "content/public/browser/notification_details.h" |
| 34 #include "content/public/browser/notification_source.h" | 33 #include "content/public/browser/notification_source.h" |
| 35 #include "content/public/browser/notification_types.h" | 34 #include "content/public/browser/notification_types.h" |
| 35 #include "content/public/browser/render_process_host.h" |
| 36 #include "googleurl/src/gurl.h" | 36 #include "googleurl/src/gurl.h" |
| 37 | 37 |
| 38 using content::BrowserThread; | 38 using content::BrowserThread; |
| 39 | 39 |
| 40 namespace safe_browsing { | 40 namespace safe_browsing { |
| 41 | 41 |
| 42 // This class is instantiated each time a new toplevel URL loads, and | 42 // This class is instantiated each time a new toplevel URL loads, and |
| 43 // asynchronously checks whether the phishing classifier should run for this | 43 // asynchronously checks whether the phishing classifier should run for this |
| 44 // URL. If so, it notifies the renderer with a StartPhishingDetection IPC. | 44 // URL. If so, it notifies the renderer with a StartPhishingDetection IPC. |
| 45 // Objects of this class are ref-counted and will be destroyed once nobody | 45 // Objects of this class are ref-counted and will be destroyed once nobody |
| (...skipping 290 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 336 this); | 336 this); |
| 337 classification_request_->Start(); | 337 classification_request_->Start(); |
| 338 } | 338 } |
| 339 | 339 |
| 340 void ClientSideDetectionHost::OnSafeBrowsingHit( | 340 void ClientSideDetectionHost::OnSafeBrowsingHit( |
| 341 const SafeBrowsingService::UnsafeResource& resource) { | 341 const SafeBrowsingService::UnsafeResource& resource) { |
| 342 // Check that this notification is really for us and that it corresponds to | 342 // Check that this notification is really for us and that it corresponds to |
| 343 // either a malware or phishing hit. In this case we store the unique page | 343 // either a malware or phishing hit. In this case we store the unique page |
| 344 // ID for later. | 344 // ID for later. |
| 345 if (tab_contents() && | 345 if (tab_contents() && |
| 346 tab_contents()->GetRenderProcessHost()->id() == | 346 tab_contents()->GetRenderProcessHost()->GetID() == |
| 347 resource.render_process_host_id && | 347 resource.render_process_host_id && |
| 348 tab_contents()->render_view_host()->routing_id() == | 348 tab_contents()->render_view_host()->routing_id() == |
| 349 resource.render_view_id && | 349 resource.render_view_id && |
| 350 (resource.threat_type == SafeBrowsingService::URL_PHISHING || | 350 (resource.threat_type == SafeBrowsingService::URL_PHISHING || |
| 351 resource.threat_type == SafeBrowsingService::URL_MALWARE) && | 351 resource.threat_type == SafeBrowsingService::URL_MALWARE) && |
| 352 tab_contents()->controller().GetActiveEntry()) { | 352 tab_contents()->controller().GetActiveEntry()) { |
| 353 unsafe_unique_page_id_ = | 353 unsafe_unique_page_id_ = |
| 354 tab_contents()->controller().GetActiveEntry()->unique_id(); | 354 tab_contents()->controller().GetActiveEntry()->unique_id(); |
| 355 // We also keep the resource around in order to be able to send the | 355 // We also keep the resource around in order to be able to send the |
| 356 // malicious URL to the server. | 356 // malicious URL to the server. |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 414 << " is_phishing:" << is_phishing; | 414 << " is_phishing:" << is_phishing; |
| 415 if (is_phishing) { | 415 if (is_phishing) { |
| 416 DCHECK(tab_contents()); | 416 DCHECK(tab_contents()); |
| 417 if (sb_service_) { | 417 if (sb_service_) { |
| 418 SafeBrowsingService::UnsafeResource resource; | 418 SafeBrowsingService::UnsafeResource resource; |
| 419 resource.url = phishing_url; | 419 resource.url = phishing_url; |
| 420 resource.original_url = phishing_url; | 420 resource.original_url = phishing_url; |
| 421 resource.is_subresource = false; | 421 resource.is_subresource = false; |
| 422 resource.threat_type = SafeBrowsingService::CLIENT_SIDE_PHISHING_URL; | 422 resource.threat_type = SafeBrowsingService::CLIENT_SIDE_PHISHING_URL; |
| 423 resource.render_process_host_id = | 423 resource.render_process_host_id = |
| 424 tab_contents()->GetRenderProcessHost()->id(); | 424 tab_contents()->GetRenderProcessHost()->GetID(); |
| 425 resource.render_view_id = | 425 resource.render_view_id = |
| 426 tab_contents()->render_view_host()->routing_id(); | 426 tab_contents()->render_view_host()->routing_id(); |
| 427 if (!sb_service_->IsWhitelisted(resource)) { | 427 if (!sb_service_->IsWhitelisted(resource)) { |
| 428 // We need to stop any pending navigations, otherwise the interstital | 428 // We need to stop any pending navigations, otherwise the interstital |
| 429 // might not get created properly. | 429 // might not get created properly. |
| 430 tab_contents()->controller().DiscardNonCommittedEntries(); | 430 tab_contents()->controller().DiscardNonCommittedEntries(); |
| 431 resource.client = new CsdClient(); // Will delete itself | 431 resource.client = new CsdClient(); // Will delete itself |
| 432 sb_service_->DoDisplayBlockingPage(resource); | 432 sb_service_->DoDisplayBlockingPage(resource); |
| 433 } | 433 } |
| 434 } | 434 } |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 489 if (sb_service_) { | 489 if (sb_service_) { |
| 490 sb_service_->RemoveObserver(this); | 490 sb_service_->RemoveObserver(this); |
| 491 } | 491 } |
| 492 sb_service_ = service; | 492 sb_service_ = service; |
| 493 if (sb_service_) { | 493 if (sb_service_) { |
| 494 sb_service_->AddObserver(this); | 494 sb_service_->AddObserver(this); |
| 495 } | 495 } |
| 496 } | 496 } |
| 497 | 497 |
| 498 } // namespace safe_browsing | 498 } // namespace safe_browsing |
| OLD | NEW |