Fix crash due to unexpected Node deletion during MutationObserver registration book...
[WebKit-https.git] / Source / WebCore / dom / MutationObserverRegistration.cpp
1 /*
2  * Copyright (C) 2011 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32
33 #include "MutationObserverRegistration.h"
34
35 #include "Document.h"
36 #include "Node.h"
37 #include "QualifiedName.h"
38
39 namespace WebCore {
40
41 PassOwnPtr<MutationObserverRegistration> MutationObserverRegistration::create(PassRefPtr<MutationObserver> observer, Node* registrationNode, MutationObserverOptions options, const HashSet<AtomicString>& attributeFilter)
42 {
43     return adoptPtr(new MutationObserverRegistration(observer, registrationNode, options, attributeFilter));
44 }
45
46 MutationObserverRegistration::MutationObserverRegistration(PassRefPtr<MutationObserver> observer, Node* registrationNode, MutationObserverOptions options, const HashSet<AtomicString>& attributeFilter)
47     : m_observer(observer)
48     , m_registrationNode(registrationNode)
49     , m_options(options)
50     , m_attributeFilter(attributeFilter)
51 {
52     m_observer->observationStarted(this);
53 }
54
55 MutationObserverRegistration::~MutationObserverRegistration()
56 {
57     clearTransientRegistrations();
58     m_observer->observationEnded(this);
59 }
60
61 void MutationObserverRegistration::resetObservation(MutationObserverOptions options, const HashSet<AtomicString>& attributeFilter)
62 {
63     clearTransientRegistrations();
64     m_options = options;
65     m_attributeFilter = attributeFilter;
66 }
67
68 void MutationObserverRegistration::observedSubtreeNodeWillDetach(Node* node)
69 {
70     if (!isSubtree())
71         return;
72
73     node->registerTransientMutationObserver(this);
74     m_observer->setHasTransientRegistration();
75
76     if (!m_transientRegistrationNodes) {
77         m_transientRegistrationNodes = adoptPtr(new NodeHashSet);
78
79         ASSERT(!m_registrationNodeKeepAlive);
80         m_registrationNodeKeepAlive = m_registrationNode; // Balanced in clearTransientRegistrations.
81     }
82     m_transientRegistrationNodes->add(node);
83 }
84
85 void MutationObserverRegistration::clearTransientRegistrations()
86 {
87     if (!m_transientRegistrationNodes) {
88         ASSERT(!m_registrationNodeKeepAlive);
89         return;
90     }
91
92     for (NodeHashSet::iterator iter = m_transientRegistrationNodes->begin(); iter != m_transientRegistrationNodes->end(); ++iter)
93         (*iter)->unregisterTransientMutationObserver(this);
94
95     m_transientRegistrationNodes.clear();
96
97     ASSERT(m_registrationNodeKeepAlive);
98     m_registrationNodeKeepAlive = 0; // Balanced in observeSubtreeNodeWillDetach.
99 }
100
101 void MutationObserverRegistration::unregisterAndDelete(MutationObserverRegistration* registry)
102 {
103     RefPtr<Node> registrationNode(registry->m_registrationNode);
104     registrationNode->unregisterMutationObserver(registry);
105     // The above line will cause registry to be deleted, so don't do any more in this function.
106 }
107
108 bool MutationObserverRegistration::shouldReceiveMutationFrom(Node* node, MutationObserver::MutationType type, const QualifiedName* attributeName) const
109 {
110     ASSERT((type == MutationObserver::Attributes && attributeName) || !attributeName);
111     if (!(m_options & type))
112         return false;
113
114     if (m_registrationNode != node && !isSubtree())
115         return false;
116
117     if (type != MutationObserver::Attributes || !(m_options & MutationObserver::AttributeFilter))
118         return true;
119
120     if (!attributeName->namespaceURI().isNull())
121         return false;
122
123     return m_attributeFilter.contains(attributeName->localName());
124 }
125
126 void MutationObserverRegistration::addRegistrationNodesToSet(HashSet<Node*>& nodes) const
127 {
128     nodes.add(m_registrationNode);
129     if (!m_transientRegistrationNodes)
130         return;
131     for (NodeHashSet::const_iterator iter = m_transientRegistrationNodes->begin(); iter != m_transientRegistrationNodes->end(); ++iter)
132         nodes.add(iter->get());
133 }
134
135 } // namespace WebCore