| 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/renderer/safe_browsing/phishing_classifier_delegate.h" | 5 #include "chrome/renderer/safe_browsing/phishing_classifier_delegate.h" |
| 6 | 6 |
| 7 #include <set> | 7 #include <set> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/callback.h" | 10 #include "base/callback.h" |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 } | 136 } |
| 137 | 137 |
| 138 void PhishingClassifierDelegate::OnStartPhishingDetection(const GURL& url) { | 138 void PhishingClassifierDelegate::OnStartPhishingDetection(const GURL& url) { |
| 139 last_url_received_from_browser_ = StripRef(url); | 139 last_url_received_from_browser_ = StripRef(url); |
| 140 // Start classifying the current page if all conditions are met. | 140 // Start classifying the current page if all conditions are met. |
| 141 // See MaybeStartClassification() for details. | 141 // See MaybeStartClassification() for details. |
| 142 MaybeStartClassification(); | 142 MaybeStartClassification(); |
| 143 } | 143 } |
| 144 | 144 |
| 145 void PhishingClassifierDelegate::DidCommitProvisionalLoad( | 145 void PhishingClassifierDelegate::DidCommitProvisionalLoad( |
| 146 WebKit::WebFrame* frame, bool is_new_navigation) { | 146 blink::WebFrame* frame, bool is_new_navigation) { |
| 147 // A new page is starting to load, so cancel classificaiton. | 147 // A new page is starting to load, so cancel classificaiton. |
| 148 // | 148 // |
| 149 // TODO(bryner): We shouldn't need to cancel classification if the navigation | 149 // TODO(bryner): We shouldn't need to cancel classification if the navigation |
| 150 // is within the same page. However, if we let classification continue in | 150 // is within the same page. However, if we let classification continue in |
| 151 // this case, we need to properly deal with the fact that PageCaptured will | 151 // this case, we need to properly deal with the fact that PageCaptured will |
| 152 // be called again for the in-page navigation. We need to be sure not to | 152 // be called again for the in-page navigation. We need to be sure not to |
| 153 // swap out the page text while the term feature extractor is still running. | 153 // swap out the page text while the term feature extractor is still running. |
| 154 DocumentState* document_state = DocumentState::FromDataSource( | 154 DocumentState* document_state = DocumentState::FromDataSource( |
| 155 frame->dataSource()); | 155 frame->dataSource()); |
| 156 NavigationState* navigation_state = document_state->navigation_state(); | 156 NavigationState* navigation_state = document_state->navigation_state(); |
| (...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 283 VLOG(2) << "Starting classification for " << last_finished_load_url_; | 283 VLOG(2) << "Starting classification for " << last_finished_load_url_; |
| 284 last_url_sent_to_classifier_ = last_finished_load_url_; | 284 last_url_sent_to_classifier_ = last_finished_load_url_; |
| 285 is_classifying_ = true; | 285 is_classifying_ = true; |
| 286 classifier_->BeginClassification( | 286 classifier_->BeginClassification( |
| 287 &classifier_page_text_, | 287 &classifier_page_text_, |
| 288 base::Bind(&PhishingClassifierDelegate::ClassificationDone, | 288 base::Bind(&PhishingClassifierDelegate::ClassificationDone, |
| 289 base::Unretained(this))); | 289 base::Unretained(this))); |
| 290 } | 290 } |
| 291 | 291 |
| 292 } // namespace safe_browsing | 292 } // namespace safe_browsing |
| OLD | NEW |