Add SafeBrowsingNavigationThrottle

chrome/browser/renderer_host/safe_browsing_navigation_throttle.h

+// 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.
+
+#ifndef CHROME_BROWSER_RENDERER_HOST_SAFE_BROWSING_NAVIGATION_THROTTLE_H_
+#define CHROME_BROWSER_RENDERER_HOST_SAFE_BROWSING_NAVIGATION_THROTTLE_H_
+
+#include <memory>
+
+#include "base/macros.h"
+#include "chrome/browser/safe_browsing/safe_browsing_service.h"
+#include "chrome/browser/safe_browsing/ui_manager.h"
+#include "components/safe_browsing_db/database_manager.h"
+#include "content/public/browser/navigation_throttle.h"
+
+class SafeBrowsingNavigationThrottle
+    : public content::NavigationThrottle,
+      public safe_browsing::SafeBrowsingDatabaseManager::Client,
+      public base::SupportsWeakPtr<SafeBrowsingNavigationThrottle> {
+ public:
+  static std::unique_ptr<content::NavigationThrottle> MaybeCreate(
+      content::NavigationHandle* handle,
+      safe_browsing::SafeBrowsingService* sb_service);
+  ~SafeBrowsingNavigationThrottle() override;
+
+ private:
+  explicit SafeBrowsingNavigationThrottle(
+      content::NavigationHandle* handle,
+      safe_browsing::SafeBrowsingService* sb_service);
+
+  // content::NavigationThrottle implementation:
+  content::NavigationThrottle::ThrottleCheckResult WillStartRequest() override;
+  content::NavigationThrottle::ThrottleCheckResult WillRedirectRequest()
+      override;
+
+  // SafeBrowsingDatabaseManager::Client implementation:
+  void OnCheckBrowseUrlResult(
+      const GURL& url,
+      safe_browsing::SBThreatType result,
+      const safe_browsing::ThreatMetadata& metadata) override;
+
+  void CheckUrlOnIO(const GURL& url);
+
+  content::WebContents* GetNavigationHandlesWebContents();
+  void OnCheckUrlResultOnUI(const GURL& url,
+                            safe_browsing::SBThreatType result,
+                            const safe_browsing::ThreatMetadata& metadata);
+  void OnBlockingPageComplete(bool proceed);
+
+  scoped_refptr<safe_browsing::SafeBrowsingDatabaseManager> database_manager_;
+  scoped_refptr<safe_browsing::SafeBrowsingUIManager> ui_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(SafeBrowsingNavigationThrottle);
+};
+
+#endif  // CHROME_BROWSER_RENDERER_HOST_SAFE_BROWSING_NAVIGATION_THROTTLE_H_