Added Dartium changes onto 45.0.2454.104

Source/bindings/core/dart/DartInjectedScriptManager.cpp

 /*
  * Copyright (C) 2007, 2008 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Matt Lilek <webkit@mattlilek.com>
  * Copyright (C) 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
@@ skipping 11 matching lines @@
  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
+#include "bindings/core/dart/DartInjectedScriptManager.h"
+
+#include "bindings/common/ScriptValue.h"
+#include "bindings/core/dart/DartInjectedScript.h"
+#include "bindings/core/dart/DartScriptState.h"
+#include "core/inspector/InjectedScriptHost.h"
 #include "core/inspector/InjectedScriptManager.h"
-
-#include "bindings/core/v8/ScriptValue.h"
-#include "core/inspector/InjectedScript.h"
-#include "core/inspector/InjectedScriptHost.h"
-#include "core/inspector/InjectedScriptNative.h"
 #include "core/inspector/JSONParser.h"
 #include "platform/JSONValues.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebData.h"
 #include "wtf/PassOwnPtr.h"
 
 namespace blink {
 
-PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
-{
-    return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
-}
-
-PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
-{
-    return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope));
-}
-
-InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck)
-    : m_nextInjectedScriptId(1)
-    , m_injectedScriptHost(InjectedScriptHost::create())
+DartInjectedScriptManager::DartInjectedScriptManager(InspectedStateAccessCheck accessCheck, InjectedScriptManager* javaScriptInjectedScriptManager)
+    : m_nextInjectedScriptId(1000000000) // Elegant design so that Dart and JavaScript ids don't overlap.
     , m_inspectedStateAccessCheck(accessCheck)
-    , m_customObjectFormatterEnabled(false)
+    , m_javaScriptInjectedScriptManager(javaScriptInjectedScriptManager)
 {
 }
 
-InjectedScriptManager::~InjectedScriptManager()
+DartInjectedScriptManager::~DartInjectedScriptManager()
 {
 }
 
-DEFINE_TRACE(InjectedScriptManager)
+InjectedScriptHost* DartInjectedScriptManager::injectedScriptHost()
 {
-    visitor->trace(m_injectedScriptHost);
+    return m_javaScriptInjectedScriptManager->injectedScriptHost();
 }
 
-void InjectedScriptManager::disconnect()
-{
-    m_injectedScriptHost->disconnect();
-    m_injectedScriptHost.clear();
-}
-
-InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
-{
-    return m_injectedScriptHost.get();
-}
-
-InjectedScript InjectedScriptManager::injectedScriptForId(int id)
+DartInjectedScript* DartInjectedScriptManager::injectedScriptForId(int id)
 {
     IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id);
     if (it != m_idToInjectedScript.end())
         return it->value;
-    for (auto& state : m_scriptStateToId) {
-        if (state.value == id)
-            return injectedScriptFor(state.key.get());
+    for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
+        if (it->value == id)
+            return injectedScriptFor(it->key.get());
     }
-    return InjectedScript();
+    return 0;
 }
 
-int InjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState)
+int DartInjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState)
 {
     ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState);
     if (it != m_scriptStateToId.end())
         return it->value;
     int id = m_nextInjectedScriptId++;
     m_scriptStateToId.set(scriptState, id);
     return id;
 }
 
-InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
+DartInjectedScript* DartInjectedScriptManager::injectedScriptForObjectId(const String& objectId)
 {
     RefPtr<JSONValue> parsedObjectId = parseJSON(objectId);
     if (parsedObjectId && parsedObjectId->type() == JSONValue::TypeObject) {
         long injectedScriptId = 0;
         bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId);
-        if (success)
-            return m_idToInjectedScript.get(injectedScriptId);
+        if (success) {
+            IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(injectedScriptId);
+            if (s != m_idToInjectedScript.end())
+                return s->value;
+        }
     }
-    return InjectedScript();
+    return 0;
 }
 
-void InjectedScriptManager::discardInjectedScripts()
+void DartInjectedScriptManager::discardInjectedScripts()
 {
     m_idToInjectedScript.clear();
     m_scriptStateToId.clear();
 }
 
-void InjectedScriptManager::discardInjectedScriptFor(ScriptState* scriptState)
+void DartInjectedScriptManager::discardInjectedScriptsFor(LocalDOMWindow* window)
 {
-    ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState);
-    if (it == m_scriptStateToId.end())
+    if (m_scriptStateToId.isEmpty())
         return;
 
-    m_idToInjectedScript.remove(it->value);
-    m_scriptStateToId.remove(it);
+    Vector<long> idsToRemove;
+    IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
+    for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
+        ScriptState* scriptState = it->value->scriptState();
+        if (window != scriptState->domWindow())
+            continue;
+        m_scriptStateToId.remove(scriptState);
+        idsToRemove.append(it->key);
+    }
+    for (size_t i = 0; i < idsToRemove.size(); i++)
+        delete m_idToInjectedScript.get(idsToRemove[i]);
+    m_idToInjectedScript.removeAll(idsToRemove);
+
+
+    // Now remove script states that have id but no injected script.
+    Vector<ScriptState*> scriptStatesToRemove;
+    for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
+        ScriptState* scriptState = it->key.get();
+        if (window == scriptState->domWindow())
+            scriptStatesToRemove.append(scriptState);
+    }
+    m_scriptStateToId.removeAll(scriptStatesToRemove);
 }
 
-bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*)
+bool DartInjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*)
 {
     return true;
 }
 
-void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
+void DartInjectedScriptManager::releaseObjectGroup(const String& objectGroup)
 {
     Vector<int> keys;
     keys.appendRange(m_idToInjectedScript.keys().begin(), m_idToInjectedScript.keys().end());
-    for (auto& key : keys) {
-        IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(key);
+    for (Vector<int>::iterator k = keys.begin(); k != keys.end(); ++k) {
+        IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(*k);
         if (s != m_idToInjectedScript.end())
-            s->value.releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here.
+            s->value->releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here.
     }
 }
 
-void InjectedScriptManager::setCustomObjectFormatterEnabled(bool enabled)
+DartInjectedScript* DartInjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState)
 {
-    m_customObjectFormatterEnabled = enabled;
-    IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
-    for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
-        if (!it->value.isEmpty())
-            it->value.setCustomObjectFormatterEnabled(enabled);
-    }
-}
-
-String InjectedScriptManager::injectedScriptSource()
-{
-    const WebData& injectedScriptSourceResource = Platform::current()->loadResource("InjectedScriptSource.js");
-    return String(injectedScriptSourceResource.data(), injectedScriptSourceResource.size());
-}
-
-InjectedScript InjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState)
-{
+    if (!inspectedScriptState)
+        return 0;
     ScriptStateToId::iterator it = m_scriptStateToId.find(inspectedScriptState);
     if (it != m_scriptStateToId.end()) {
         IdToInjectedScriptMap::iterator it1 = m_idToInjectedScript.find(it->value);
         if (it1 != m_idToInjectedScript.end())
             return it1->value;
     }
 
     if (!m_inspectedStateAccessCheck(inspectedScriptState))
-        return InjectedScript();
+        return 0;
 
     int id = injectedScriptIdFor(inspectedScriptState);
-    RefPtr<InjectedScriptNative> injectedScriptNative = adoptRef(new InjectedScriptNative(inspectedScriptState->isolate()));
-    ScriptValue injectedScriptValue = createInjectedScript(injectedScriptSource(), inspectedScriptState, id, injectedScriptNative.get());
-    InjectedScript result(injectedScriptValue, m_inspectedStateAccessCheck, injectedScriptNative.release());
-    if (m_customObjectFormatterEnabled)
-        result.setCustomObjectFormatterEnabled(m_customObjectFormatterEnabled);
+
+    DartInjectedScript* result;
+    ASSERT(!inspectedScriptState->isJavaScript());
+    result = new DartInjectedScript(static_cast<DartScriptState*>(inspectedScriptState), m_inspectedStateAccessCheck, id, injectedScriptHost(), m_javaScriptInjectedScriptManager);
     m_idToInjectedScript.set(id, result);
     return result;
 }
 
 } // namespace blink
+