extensions/renderer/script_context_set.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef EXTENSIONS_RENDERER_SCRIPT_CONTEXT_SET_H_
   6 #define EXTENSIONS_RENDERER_SCRIPT_CONTEXT_SET_H_
   7
   8 #include <set>
   9 #include <string>
  10
  11 #include "base/macros.h"
  12 #include "base/memory/weak_ptr.h"
  13 #include "extensions/common/extension.h"
  14 #include "extensions/common/extension_set.h"
  15 #include "extensions/common/features/feature.h"
  16 #include "url/gurl.h"
  17 #include "v8/include/v8.h"
  18
  19 class GURL;
  20
  21 namespace base {
  22 class ListValue;
  23 }
  24
  25 namespace blink {
  26 class WebLocalFrame;
  27 class WebSecurityOrigin;
  28 }
  29
  30 namespace content {
  31 class RenderFrame;
  32 }
  33
  34 namespace extensions {
  35 class ScriptContext;
  36
  37 // A container of ScriptContexts, responsible for both creating and managing
  38 // them.
  39 //
  40 // Since calling JavaScript within a context can cause any number of contexts
  41 // to be created or destroyed, this has additional smarts to help with the set
  42 // changing underneath callers.
  43 class ScriptContextSet {
  44  public:
  45   ScriptContextSet(
  46       ExtensionSet* extensions,
  47       // Set of the IDs of extensions that are active in this process.
  48       // Must outlive this. TODO(kalman): Combine this and |extensions|.
  49       ExtensionIdSet* active_extension_ids);
  50
  51   ~ScriptContextSet();
  52
  53   // Returns the number of contexts being tracked by this set.
  54   // This may also include invalid contexts. TODO(kalman): Useful?
  55   size_t size() const { return contexts_.size(); }
  56
  57   // Creates and starts managing a new ScriptContext. Ownership is held.
  58   // Returns a weak reference to the new ScriptContext.
  59   ScriptContext* Register(blink::WebLocalFrame* frame,
  60                           const v8::Local<v8::Context>& v8_context,
  61                           int extension_group,
  62                           int world_id);
  63
  64   // If the specified context is contained in this set, remove it, then delete
  65   // it asynchronously. After this call returns the context object will still
  66   // be valid, but its frame() pointer will be cleared.
  67   void Remove(ScriptContext* context);
  68
  69   // Gets the ScriptContext corresponding to v8::Context::GetCurrent(), or
  70   // NULL if no such context exists.
  71   ScriptContext* GetCurrent() const;
  72
  73   // Gets the ScriptContext corresponding to v8::Context::GetCalling(), or
  74   // NULL if no such context exists.
  75   ScriptContext* GetCalling() const;
  76
  77   // Gets the ScriptContext corresponding to the specified
  78   // v8::Context or NULL if no such context exists.
  79   ScriptContext* GetByV8Context(const v8::Local<v8::Context>& context) const;
  80
  81   // Synchronously runs |callback| with each ScriptContext that belongs to
  82   // |extension_id| in |render_frame|.
  83   //
  84   // An empty |extension_id| will match all extensions, and a null
  85   // |render_frame| will match all render views, but try to use the inline
  86   // variants of these methods instead.
  87   void ForEach(const std::string& extension_id,
  88                content::RenderFrame* render_frame,
  89                const base::Callback<void(ScriptContext*)>& callback) const;
  90   // ForEach which matches all extensions.
  91   void ForEach(content::RenderFrame* render_frame,
  92                const base::Callback<void(ScriptContext*)>& callback) const {
  93     ForEach(std::string(), render_frame, callback);
  94   }
  95   // ForEach which matches all render views.
  96   void ForEach(const std::string& extension_id,
  97                const base::Callback<void(ScriptContext*)>& callback) const {
  98     ForEach(extension_id, nullptr, callback);
  99   }
 100
 101   // Cleans up contexts belonging to an unloaded extension.
 102   //
 103   // Returns the set of ScriptContexts that were removed as a result. These
 104   // are safe to interact with until the end of the current event loop, since
 105   // they're deleted asynchronously.
 106   std::set<ScriptContext*> OnExtensionUnloaded(const std::string& extension_id);
 107
 108  private:
 109   // Finds the extension for the JavaScript context associated with the
 110   // specified |frame| and isolated world. If |world_id| is zero, finds the
 111   // extension ID associated with the main world's JavaScript context. If the
 112   // JavaScript context isn't from an extension, returns empty string.
 113   const Extension* GetExtensionFromFrameAndWorld(
 114       const blink::WebLocalFrame* frame,
 115       int world_id,
 116       bool use_effective_url);
 117
 118   // Returns the Feature::Context type of context for a JavaScript context.
 119   Feature::Context ClassifyJavaScriptContext(
 120       const Extension* extension,
 121       int extension_group,
 122       const GURL& url,
 123       const blink::WebSecurityOrigin& origin);
 124
 125   // Calls Remove on |context| then appends |context| to |out|.
 126   // This is a helper designed to be used by OnExtensionUnloaded with ForEach.
 127   void DispatchOnUnloadEventAndRemove(std::set<ScriptContext*>* out,
 128                                       ScriptContext* context);
 129
 130   // Weak reference to all installed Extensions.
 131   ExtensionSet* extensions_;
 132
 133   // Weak reference to all installed Extensions that are also active in this
 134   // process.
 135   ExtensionIdSet* active_extension_ids_;
 136
 137   // The set of all ScriptContexts we own.
 138   std::set<ScriptContext*> contexts_;
 139
 140   DISALLOW_COPY_AND_ASSIGN(ScriptContextSet);
 141 };
 142
 143 }  // namespace extensions
 144
 145 #endif  // EXTENSIONS_RENDERER_SCRIPT_CONTEXT_SET_H_