Index: components/subresource_filter/content/browser/subframe_filtering_navigation_throttle.cc |
diff --git a/components/subresource_filter/content/browser/subframe_filtering_navigation_throttle.cc b/components/subresource_filter/content/browser/subframe_filtering_navigation_throttle.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..14dcf618fa67588dff597cf4e3a3f907d9e9fe72 |
--- /dev/null |
+++ b/components/subresource_filter/content/browser/subframe_filtering_navigation_throttle.cc |
@@ -0,0 +1,56 @@ |
+// Copyright 2017 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 "components/subresource_filter/content/browser/subframe_filtering_navigation_throttle.h" |
+ |
+#include "base/bind.h" |
+#include "components/subresource_filter/content/browser/async_document_subresource_filter.h" |
+#include "content/public/browser/navigation_handle.h" |
+ |
+namespace subresource_filter { |
+ |
+SubframeFilteringNavigationThrottle::SubframeFilteringNavigationThrottle( |
+ content::NavigationHandle* handle, |
+ AsyncDocumentSubresourceFilter* async_filter) |
pkalinnikov
2017/02/14 12:11:07
nit: Rename to |parent_filter| too (see comment in
Charlie Harrison
2017/02/14 14:09:38
Done.
|
+ : content::NavigationThrottle(handle), |
+ parent_filter_(async_filter), |
+ weak_ptr_factory_(this) { |
+ DCHECK(!handle->IsInMainFrame()); |
+ DCHECK(parent_filter_); |
+} |
+ |
+SubframeFilteringNavigationThrottle::~SubframeFilteringNavigationThrottle() {} |
+ |
+content::NavigationThrottle::ThrottleCheckResult |
+SubframeFilteringNavigationThrottle::WillStartRequest() { |
+ return DeferToCalculateLoadPolicy(); |
+} |
+ |
+content::NavigationThrottle::ThrottleCheckResult |
+SubframeFilteringNavigationThrottle::WillRedirectRequest() { |
+ return DeferToCalculateLoadPolicy(); |
+} |
+ |
+content::NavigationThrottle::ThrottleCheckResult |
+SubframeFilteringNavigationThrottle::DeferToCalculateLoadPolicy() { |
+ parent_filter_->GetLoadPolicyForSubdocument( |
+ navigation_handle()->GetURL(), |
+ base::Bind(&SubframeFilteringNavigationThrottle::OnCalculatedLoadPolicy, |
+ weak_ptr_factory_.GetWeakPtr())); |
+ return content::NavigationThrottle::ThrottleCheckResult::DEFER; |
+} |
+ |
+void SubframeFilteringNavigationThrottle::OnCalculatedLoadPolicy( |
+ blink::WebDocumentSubresourceFilter::LoadPolicy policy) { |
+ // TODO(csharrison): Support WouldDisallow pattern and expose the policy for |
+ // metrics. |
+ if (policy == blink::WebDocumentSubresourceFilter::Disallow) { |
+ navigation_handle()->CancelDeferredNavigation( |
+ content::NavigationThrottle::CANCEL); |
pkalinnikov
2017/02/14 12:11:07
We should also to call parent_filter_->ReportDisal
Charlie Harrison
2017/02/14 14:09:38
Done.
|
+ } else { |
+ navigation_handle()->Resume(); |
+ } |
+} |
+ |
+} // namespace subresource_filter |