Network service: SafeBrowsing check for frame-resources from browser.

chrome/browser/safe_browsing/safe_browsing_browser_throttle.h

+// 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.
+
+#ifndef CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_BROWSER_THROTTLE_H_

[vakh (use Gerrit instead), 2017/06/15 21:52:59]

Since this file is already under safe_browsing directory, do you mind
calling it browser_throttle.h ?

[yzshen1, 2017/06/16 21:27:09]

Done. I changed it to browser_url_loader_throttle.

+#define CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_BROWSER_THROTTLE_H_
+
+#include <memory>
+
+#include "base/callback.h"
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "content/public/common/url_loader_throttle.h"
+
+namespace content {
+class WebContents;
+}
+
+namespace safe_browsing {
+
+class SafeBrowsingDatabaseManager;
+class SafeBrowsingUIManager;
+class SafeBrowsingUrlCheckerImpl;
+
+// SafeBrowsingBrowserThrottle is used in the browser process to query
+// SafeBrowsing to determine whether a URL and also its redirect URLs are safe
+// to load. It defers response processing until all URL checks are completed;
+// cancels the load if any URLs turn out to be bad.
+// Used when --enable-network-service is in effect.
+class SafeBrowsingBrowserThrottle : public content::URLLoaderThrottle {
+ public:
+  // |web_contents_getter| is used for displaying SafeBrowsing UI when
+  // necessary.
+  SafeBrowsingBrowserThrottle(
+      scoped_refptr<SafeBrowsingDatabaseManager> database_manager,
+      scoped_refptr<SafeBrowsingUIManager> ui_manager,
+      const base::Callback<content::WebContents*()>& web_contents_getter);
+  ~SafeBrowsingBrowserThrottle() override;
+
+  // content::URLLoaderThrottle implementation.
+  void WillStartRequest(const GURL& url,
+                        int load_flags,
+                        content::ResourceType resource_type,
+                        bool* defer) override;
+  void WillRedirectRequest(const net::RedirectInfo& redirect_info,
+                           bool* defer) override;
+  void WillProcessResponse(bool* defer) override;
+
+ private:
+  void OnCheckUrlResult(bool safe);
+
+  // The following two members stay valid until |url_checker_| is created.
+  scoped_refptr<SafeBrowsingDatabaseManager> database_manager_;
+  scoped_refptr<SafeBrowsingUIManager> ui_manager_;
+
+  base::Callback<content::WebContents*()> web_contents_getter_;
+
+  std::unique_ptr<SafeBrowsingUrlCheckerImpl> url_checker_;
+
+  size_t pending_checks_ = 0;
+  bool blocked_ = false;
+  bool during_destruction_ = false;
+
+  DISALLOW_COPY_AND_ASSIGN(SafeBrowsingBrowserThrottle);
+};
+
+}  // namespace safe_browsing
+
+#endif  // CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_BROWSER_THROTTLE_H_