Navigation throttle for the Safe Browsing Subresource Filter.

components/subresource_filter/content/browser/subresource_filter_navigation_throttle.cc

Index: components/subresource_filter/content/browser/subresource_filter_navigation_throttle.cc
diff --git a/components/subresource_filter/content/browser/subresource_filter_navigation_throttle.cc b/components/subresource_filter/content/browser/subresource_filter_navigation_throttle.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4c57263624fcdf58243750bf6f87b475097f854d
--- /dev/null
+++ b/components/subresource_filter/content/browser/subresource_filter_navigation_throttle.cc
@@ -0,0 +1,60 @@
+// Copyright 2016 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/subresource_filter_navigation_throttle.h"
+
+#include "components/subresource_filter/content/browser/content_subresource_filter_driver.h"
+#include "components/subresource_filter/content/browser/content_subresource_filter_driver_factory.h"
+#include "components/subresource_filter/core/browser/subresource_filter_features.h"
+#include "components/subresource_filter/core/common/activation_state.h"
+#include "content/public/browser/navigation_handle.h"
+
+namespace subresource_filter {
+
+// static
+std::unique_ptr<content::NavigationThrottle>
+SubresourceFilterNavigationThrottle::Create(content::NavigationHandle* handle) {
+  return std::unique_ptr<content::NavigationThrottle>(
+      new SubresourceFilterNavigationThrottle(handle));
+}
+
+SubresourceFilterNavigationThrottle::SubresourceFilterNavigationThrottle(
+    content::NavigationHandle* handle)
+    : content::NavigationThrottle(handle) {
+  initial_url_ = navigation_handle()->GetURL();
+}
+
+SubresourceFilterNavigationThrottle::~SubresourceFilterNavigationThrottle() {}
+
+content::NavigationThrottle::ThrottleCheckResult
+SubresourceFilterNavigationThrottle::WillRedirectRequest() {
+  subresource_filter::ContentSubresourceFilterDriverFactory* driver_factory =
+      subresource_filter::ContentSubresourceFilterDriverFactory::
+          FromWebContents(navigation_handle()->GetWebContents());
+  if (driver_factory->IsURLInSocEngList(initial_url_)) {
+    driver_factory->AddToSocEngList(navigation_handle()->GetURL());
+  }
+
+  return NavigationThrottle::PROCEED;
+}
+
+content::NavigationThrottle::ThrottleCheckResult
+SubresourceFilterNavigationThrottle::WillProcessResponse() {
+  if (!navigation_handle()->GetURL().SchemeIsHTTPOrHTTPS())
+    return NavigationThrottle::PROCEED;
+
+  subresource_filter::ContentSubresourceFilterDriverFactory* driver_factory =
+      subresource_filter::ContentSubresourceFilterDriverFactory::
+          FromWebContents(navigation_handle()->GetWebContents());
+  if (driver_factory->IsURLInSocEngList(initial_url_)) {
+    ContentSubresourceFilterDriver* driver =
+        driver_factory->DriverFromFrameHost(
+            navigation_handle()->GetRenderFrameHost());
+    driver->ActivateForProvisionalLoad(GetMaximumActivationState());
+  }
+
+  return NavigationThrottle::PROCEED;
+}
+
+}  // namespace subresource_filter