Skeleton of the Safe Browsing Subresource Filter.

components/subresource_filter/content/browser/content_subresource_filter_driver_factory.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 COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER_DRIVER_FACTORY_H_
+#define COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER_DRIVER_FACTORY_H_
+
+#include <map>
+#include <utility>
+
+#include "base/macros.h"
+#include "base/supports_user_data.h"
+#include "content/public/browser/web_contents_observer.h"
+
+namespace content {
+class WebContents;
+class RenderFrameHost;
+}  // namespace content
+
+namespace subresource_filter {
+
+class ContentSubresourceFilterDriver;
+
+// Controls the activation of subresource filtering for each page load in a
+// WebContents and manufactures the per-frame ContentSubresourceFilterDrivers.
+class ContentSubresourceFilterDriverFactory
+    : public base::SupportsUserData::Data,
+      public content::WebContentsObserver {
+ public:
+  static void CreateForWebContents(content::WebContents* web_contents);
+  static ContentSubresourceFilterDriverFactory* FromWebContents(
+      content::WebContents* web_contents);
+
+  explicit ContentSubresourceFilterDriverFactory(
+      content::WebContents* web_contents);
+  ~ContentSubresourceFilterDriverFactory() override;
+
+ private:
+  typedef std::map<content::RenderFrameHost*,
+                   std::unique_ptr<ContentSubresourceFilterDriver>>
+      FrameHostToOwnedDriverMap;
+
+  void CreateDriverForFrameHostIfNeeded(
+      content::RenderFrameHost* render_frame_host);
+  ContentSubresourceFilterDriver* DriverFromFrameHost(
+      content::RenderFrameHost* render_frame_host);
+
+  // content::WebContentsObserver:
+  void RenderFrameCreated(content::RenderFrameHost* render_frame_host) override;
+  void RenderFrameDeleted(content::RenderFrameHost* render_frame_host) override;
+  void DidStartProvisionalLoadForFrame(
+      content::RenderFrameHost* render_frame_host,
+      const GURL& validated_url,
+      bool is_error_page,
+      bool is_iframe_srcdoc) override;
+
+  static const char kWebContentsUserDataKey[];
+
+  FrameHostToOwnedDriverMap frame_drivers_;
+
+  DISALLOW_COPY_AND_ASSIGN(ContentSubresourceFilterDriverFactory);
+};
+
+}  // namespace subresource_filter
+
+#endif  // COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER_DRIVER_FACTORY_H_