src/third_party/WebKit/Source/core/inspector/InjectedScriptManager.h

   1 /*
   2  * Copyright (C) 2007 Apple Inc. All rights reserved.
   3  * Copyright (C) 2012 Google Inc. All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1.  Redistributions of source code must retain the above copyright
  10  *     notice, this list of conditions and the following disclaimer.
  11  * 2.  Redistributions in binary form must reproduce the above copyright
  12  *     notice, this list of conditions and the following disclaimer in the
  13  *     documentation and/or other materials provided with the distribution.
  14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  15  *     its contributors may be used to endorse or promote products derived
  16  *     from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #ifndef InjectedScriptManager_h
  31 #define InjectedScriptManager_h
  32
  33 #include "bindings/core/v8/ScriptState.h"
  34 #include "wtf/Forward.h"
  35 #include "wtf/HashMap.h"
  36 #include "wtf/text/WTFString.h"
  37 #include <v8.h>
  38
  39 namespace blink {
  40
  41 class LocalDOMWindow;
  42 class InjectedScript;
  43 class InjectedScriptHost;
  44 class ScriptValue;
  45
  46 class InjectedScriptManager : public NoBaseWillBeGarbageCollectedFinalized<InjectedScriptManager> {
  47     WTF_MAKE_NONCOPYABLE(InjectedScriptManager);
  48     WTF_MAKE_FAST_ALLOCATED_WILL_BE_REMOVED;
  49 public:
  50     struct CallbackData {
  51         ScopedPersistent<v8::Object> handle;
  52         RefPtrWillBePersistent<InjectedScriptHost> host;
  53         InjectedScriptManager* injectedScriptManager;
  54     };
  55
  56     static PassOwnPtrWillBeRawPtr<InjectedScriptManager> createForPage();
  57     static PassOwnPtrWillBeRawPtr<InjectedScriptManager> createForWorker();
  58     ~InjectedScriptManager();
  59     void trace(Visitor*);
  60
  61     void disconnect();
  62
  63     InjectedScriptHost* injectedScriptHost();
  64
  65     InjectedScript injectedScriptFor(ScriptState*);
  66     InjectedScript injectedScriptForId(int);
  67     int injectedScriptIdFor(ScriptState*);
  68     InjectedScript injectedScriptForObjectId(const String& objectId);
  69     void discardInjectedScripts();
  70     void discardInjectedScriptsFor(LocalDOMWindow*);
  71     void releaseObjectGroup(const String& objectGroup);
  72
  73     typedef bool (*InspectedStateAccessCheck)(ScriptState*);
  74     InspectedStateAccessCheck inspectedStateAccessCheck() const { return m_inspectedStateAccessCheck; }
  75
  76     static void setWeakCallback(const v8::WeakCallbackData<v8::Object, CallbackData>&);
  77     CallbackData* createCallbackData(InjectedScriptManager*);
  78     void removeCallbackData(CallbackData*);
  79
  80 private:
  81     explicit InjectedScriptManager(InspectedStateAccessCheck);
  82
  83     String injectedScriptSource();
  84     ScriptValue createInjectedScript(const String& source, ScriptState*, int id);
  85
  86     static bool canAccessInspectedWindow(ScriptState*);
  87     static bool canAccessInspectedWorkerGlobalScope(ScriptState*);
  88
  89     int m_nextInjectedScriptId;
  90     typedef HashMap<int, InjectedScript> IdToInjectedScriptMap;
  91     IdToInjectedScriptMap m_idToInjectedScript;
  92     RefPtrWillBeMember<InjectedScriptHost> m_injectedScriptHost;
  93     InspectedStateAccessCheck m_inspectedStateAccessCheck;
  94     typedef HashMap<RefPtr<ScriptState>, int> ScriptStateToId;
  95     ScriptStateToId m_scriptStateToId;
  96     HashSet<OwnPtr<CallbackData> > m_callbackDataSet;
  97 };
  98
  99 } // namespace blink
 100
 101 #endif // !defined(InjectedScriptManager_h)