Properly retain non-node event listeners.

Source/WebCore/bindings/dart/DartGCController.cpp

 /*
  * 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:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 30 matching lines @@
 
 #include <wtf/Vector.h>
 
 namespace WebCore {
 
 struct WeakReferenceSet {
     Vector<Dart_Handle> keys;
     Vector<Dart_Handle> values;
 };
 
-typedef HashMap<Node*, WeakReferenceSet*> weakReferenceSetForRootMap;
+typedef Vector<WeakReferenceSet*> WeakReferenceSets;
 
-static weakReferenceSetForRootMap* weakReferenceSetForRoot()
+static WeakReferenceSets* weakReferenceSets()
 {
-    DEFINE_STATIC_LOCAL(weakReferenceSetForRootMap, map, ());
-    return &map;
+    DEFINE_STATIC_LOCAL(WeakReferenceSets, sets, ());
+    return &sets;
 }
 
 static Node* calculateRootNode(Node*);
-static void addEventListenersToReferenceSet(EventTarget*, WeakReferenceSet*);
+static void collectEventListenerWrappers(EventTarget*, Vector<Dart_Handle>& wrappers);
 
-void DartGCController::prologueCallback()
+static void createWeakReferenceSetsForNodes(DartDOMData* domData, WeakReferenceSet* documentWeakReferenceSet, WeakReferenceSets* weakReferenceSets)
 {
-    ASSERT(DartIsolate::isDOMIsolate(Dart_CurrentIsolate()));
-
-    // FIXME: pass the state to epilogue callback instead of using statics.
-    ASSERT(weakReferenceSetForRoot()->isEmpty());
-
-    DartDOMData* domData = DartIsolate::currentDOMData();
-
-    ASSERT(domData->scriptExecutionContext()->isDocument());
-    Document* document = static_cast<Document*>(domData->scriptExecutionContext());
-
-    WeakReferenceSet* documentWeakReferenceSet = new WeakReferenceSet;
-    weakReferenceSetForRoot()->set(document, documentWeakReferenceSet);
+    typedef HashMap<Node*, WeakReferenceSet*> WeakReferenceSetForRootMap;
+    WeakReferenceSetForRootMap weakReferenceSetForRootMap;
 
     DartDOMNodeMap* nodeMap = domData->nodeMap();
     for (DartDOMNodeMap::iterator it = nodeMap->begin(); it != nodeMap->end(); ++it) {
         Node* node = it->first;
 
         WeakReferenceSet* weakReferenceSet = 0;
         if (node->inDocument() || (node->hasTagName(HTMLNames::imgTag) && static_cast<HTMLImageElement*>(node)->hasPendingLoadEvent())) {
             ASSERT(node->document() == document);
             weakReferenceSet = documentWeakReferenceSet;
         } else {
             Node* root = calculateRootNode(node);
             if (!root)
                 continue;
-            weakReferenceSetForRootMap::iterator weakReferenceSetIterator = weakReferenceSetForRoot()->find(root);
-            if (weakReferenceSetIterator != weakReferenceSetForRoot()->end())
+            WeakReferenceSetForRootMap::iterator weakReferenceSetIterator = weakReferenceSetForRootMap.find(root);
+            if (weakReferenceSetIterator != weakReferenceSetForRootMap.end())
                 weakReferenceSet = weakReferenceSetIterator->second;
             else {
                 weakReferenceSet = new WeakReferenceSet;
-                weakReferenceSetForRoot()->set(root, weakReferenceSet);
+                weakReferenceSets->append(weakReferenceSet);
+                weakReferenceSetForRootMap.set(root, weakReferenceSet);
             }
         }
         ASSERT(weakReferenceSet);
 
         Dart_Handle wrapper = it->second;
         weakReferenceSet->keys.append(wrapper);
         weakReferenceSet->values.append(wrapper);
-        addEventListenersToReferenceSet(node, weakReferenceSet);
+        collectEventListenerWrappers(node, weakReferenceSet->values);
     }
+}
 
+static void createWeakReferenceSetsForEventTargets(DartDOMData* domData, WeakReferenceSets* weakReferenceSets)
+{
+    DartEventTargetMap* eventTargetMap = domData->eventTargetMap();
+    for (DartEventTargetMap::iterator it = eventTargetMap->begin(); it != eventTargetMap->end(); ++it) {
+        WeakReferenceSet* weakReferenceSet = new WeakReferenceSet;
+        weakReferenceSets->append(weakReferenceSet);
+        weakReferenceSet->keys.append(it->second);
+        collectEventListenerWrappers(it->first, weakReferenceSet->values);
+    }
+}
+
+static void collectRetainedActiveDOMObjectWrappers(DartDOMData* domData, Vector<Dart_Handle>& wrappers)
+{
     DartActiveDOMObjectMap* activeObjectMap = domData->activeObjectMap();
     for (DartActiveDOMObjectMap::iterator it = activeObjectMap->begin(); it != activeObjectMap->end(); ++it) {
         ActiveDOMObject* activeDOMObject = it->first;
         if (activeDOMObject->hasPendingActivity()) {
             Dart_Handle wrapper = it->second;
-            documentWeakReferenceSet->values.append(wrapper);
+            wrappers.append(wrapper);
         }
     }
 
     DartMessagePortMap* messagePortMap = domData->messagePortMap();
     for (DartMessagePortMap::iterator it = messagePortMap->begin(); it != messagePortMap->end(); ++it) {
         MessagePort* messagePort = it->first;
         if (messagePort->isEntangled() || messagePort->hasPendingActivity()) {
             Dart_Handle wrapper = it->second;
-            documentWeakReferenceSet->values.append(wrapper);
+            wrappers.append(wrapper);
         }
     }
-
-    addEventListenersToReferenceSet(document->domWindow(), documentWeakReferenceSet);
-
-    for (weakReferenceSetForRootMap::iterator it = weakReferenceSetForRoot()->begin(); it != weakReferenceSetForRoot()->end(); ++it) {
-        WeakReferenceSet* set = it->second;
-        Dart_NewWeakReferenceSet(set->keys.data(), set->keys.size(), set->values.data(), set->values.size());
-    }
 }
 
-void DartGCController::epilogueCallback()
-{
-    WTF::deleteAllValues(*weakReferenceSetForRoot());
-    weakReferenceSetForRoot()->clear();
-}
-
 static Node* calculateRootNode(Node* node)
 {
     Node* root = node;
     if (node->isAttributeNode()) {
         root = static_cast<Attr*>(node)->ownerElement();
         // If the attribute has no element, no need to put it in the group,
         // because it'll always be a group of 1.
         if (!root)
             return 0;
     } else {
         while (Node* parent = root->parentOrHostNode())
             root = parent;
     }
     return root;
 }
 
-static void addEventListenersToReferenceSet(EventTarget* eventTarget, WeakReferenceSet* weakReferenceSet)
+static void collectEventListenerWrappers(EventTarget* eventTarget, Vector<Dart_Handle>& wrappers)
 {
     if (!eventTarget->hasEventListeners())
         return;
 
     EventListenerIterator iterator(eventTarget);
     while (EventListener* listener = iterator.nextListener()) {
         if (listener->type() != static_cast<int>(DartEventListenerType))
             continue;
         DartEventListener* dartListener = static_cast<DartEventListener*>(listener);
         if (dartListener->isolate() != DartIsolate::current())
             continue;
-        weakReferenceSet->values.append(dartListener->listenerObject());
+        wrappers.append(dartListener->listenerObject());
     }
 }
 
+void DartGCController::prologueCallback()
+{
+    ASSERT(DartIsolate::isDOMIsolate(Dart_CurrentIsolate()));
+    // FIXME: pass the state to epilogue callback instead of using statics.
+    ASSERT(weakReferenceSets()->isEmpty());
+
+    DartDOMData* domData = DartIsolate::currentDOMData();
+
+    WeakReferenceSet* documentWeakReferenceSet = new WeakReferenceSet;
+    weakReferenceSets()->append(documentWeakReferenceSet);
+
+    createWeakReferenceSetsForNodes(domData, documentWeakReferenceSet, weakReferenceSets());
+    createWeakReferenceSetsForEventTargets(domData, weakReferenceSets());
+    collectRetainedActiveDOMObjectWrappers(domData, documentWeakReferenceSet->values);
+
+    for (size_t i = 0; i < weakReferenceSets()->size(); ++i) {
+        WeakReferenceSet* set = (*weakReferenceSets())[i];
+        Dart_NewWeakReferenceSet(set->keys.data(), set->keys.size(), set->values.data(), set->values.size());
+    }
 }
+
+void DartGCController::epilogueCallback()
+{
+    WTF::deleteAllValues(*weakReferenceSets());
+    weakReferenceSets()->clear();
+}
+
+}