Add PluginServiceFilter interface.

chrome/browser/chrome_plugin_service_filter.h

Index: chrome/browser/chrome_plugin_service_filter.h
diff --git a/chrome/browser/chrome_plugin_service_filter.h b/chrome/browser/chrome_plugin_service_filter.h
new file mode 100644
index 0000000000000000000000000000000000000000..41be795a8a694567600ca2be0247a4c35546ccd3
--- /dev/null
+++ b/chrome/browser/chrome_plugin_service_filter.h
@@ -0,0 +1,90 @@
+// Copyright (c) 2011 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_CHROME_PLUGIN_SERVICE_FILTER_H_
+#define CHROME_BROWSER_CHROME_PLUGIN_SERVICE_FILTER_H_
+#pragma once
+
+#include <map>
+#include <vector>
+
+#include "base/hash_tables.h"
+#include "base/file_path.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/singleton.h"
+#include "base/synchronization/lock.h"
+#include "content/browser/plugin_service_filter.h"
+#include "content/common/notification_observer.h"
+#include "content/common/notification_registrar.h"
+#include "googleurl/src/gurl.h"
+#include "webkit/plugins/npapi/webplugininfo.h"
+
+class PluginPrefs;
+class Profile;
+
+// This class must be created (by calling the |GetInstance| method) on the UI
+// thread, but is safe to use on any thread after that.
+class ChromePluginServiceFilter : public content::PluginServiceFilter,
+                                  public NotificationObserver {
+ public:
+  static ChromePluginServiceFilter* GetInstance();
+
+  // This method should be called on the UI thread.
+  void RegisterResourceContext(Profile* profile,
+                               const content::ResourceContext& context);
+
+  void UnregisterResourceContext(const content::ResourceContext& context);
+
+  // Overrides the plugin lookup mechanism for a given tab and object URL to use
+  // a specifc plugin.
+  void OverridePluginForTab(int render_process_id,
+                            int render_view_id,
+                            const GURL& url,
+                            const webkit::npapi::WebPluginInfo& plugin);
+
+  // Restricts the given plugin to the the scheme and host of the given url.
+  // Call with an empty url to reset this.
+  void RestrictPluginToUrl(const FilePath& plugin_path, const GURL& url);
+
+  // PluginServiceFilter implementation:
+  virtual bool ShouldUsePlugin(
+      int render_process_id,
+      int render_view_id,
+      const content::ResourceContext& resource_context,
+      const GURL& url,
+      const GURL& policy_url,
+      webkit::npapi::WebPluginInfo* plugin) const OVERRIDE;
+
+ private:
+  friend struct DefaultSingletonTraits<ChromePluginServiceFilter>;
+
+  struct OverriddenPlugin {
+    int render_process_id;
+    int render_view_id;
+    GURL url;
+    webkit::npapi::WebPluginInfo plugin;
+  };
+
+  ChromePluginServiceFilter();
+  virtual ~ChromePluginServiceFilter();
+
+  // NotificationObserver implementation:
+  virtual void Observe(int type,
+                       const NotificationSource& source,
+                       const NotificationDetails& details);
+
+  NotificationRegistrar registrar_;
+
+  mutable base::Lock lock_;  // Guards access to member variables.

[jam, 2011/08/10 06:31:51]

nit: doesn't look like this needs to be mutable

[Bernhard Bauer, 2011/08/10 09:18:36]

I used it in ShouldUsePlugin which was const, but I made it non-const now.

+  // Map of plugin paths to the origin they are restricted to.
+  typedef base::hash_map<FilePath, GURL> RestrictedPluginMap;
+  RestrictedPluginMap restricted_plugins_;
+  typedef std::map<const content::ResourceContext*, scoped_refptr<PluginPrefs> >
+      ResourceContextMap;
+  ResourceContextMap resource_context_map_;
+
+  std::vector<OverriddenPlugin> overridden_plugins_;
+};
+
+#endif  // CHROME_BROWSER_CHROME_PLUGIN_SERVICE_FILTER_H_