MessagePort must set m_closed to be true at the end of MessagePort::close function
[WebKit-https.git] / Source / WebCore / dom / MessagePort.cpp
1 /*
2  * Copyright (C) 2008 Apple 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
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  *
25  */
26
27 #include "config.h"
28 #include "MessagePort.h"
29
30 #include "DOMWindow.h"
31 #include "Document.h"
32 #include "EventException.h"
33 #include "EventNames.h"
34 #include "ExceptionCode.h"
35 #include "MessageEvent.h"
36 #include "SecurityOrigin.h"
37 #include "Timer.h"
38 #include "WorkerContext.h"
39 #include <wtf/text/AtomicString.h>
40
41 namespace WebCore {
42
43 MessagePort::MessagePort(ScriptExecutionContext& scriptExecutionContext)
44     : m_started(false)
45     , m_closed(false)
46     , m_scriptExecutionContext(&scriptExecutionContext)
47 {
48     m_scriptExecutionContext->createdMessagePort(this);
49
50     // Don't need to call processMessagePortMessagesSoon() here, because the port will not be opened until start() is invoked.
51 }
52
53 MessagePort::~MessagePort()
54 {
55     close();
56     if (m_scriptExecutionContext)
57         m_scriptExecutionContext->destroyedMessagePort(this);
58 }
59
60 // FIXME: remove this when we update the ObjC bindings (bug #28774).
61 void MessagePort::postMessage(PassRefPtr<SerializedScriptValue> message, MessagePort* port, ExceptionCode& ec)
62 {
63     MessagePortArray ports;
64     if (port)
65         ports.append(port);
66     postMessage(message, &ports, ec);
67 }
68
69 void MessagePort::postMessage(PassRefPtr<SerializedScriptValue> message, ExceptionCode& ec)
70 {
71     postMessage(message, static_cast<MessagePortArray*>(0), ec);
72 }
73
74 void MessagePort::postMessage(PassRefPtr<SerializedScriptValue> message, const MessagePortArray* ports, ExceptionCode& ec)
75 {
76     if (!isEntangled())
77         return;
78     ASSERT(m_scriptExecutionContext);
79
80     OwnPtr<MessagePortChannelArray> channels;
81     // Make sure we aren't connected to any of the passed-in ports.
82     if (ports) {
83         for (unsigned int i = 0; i < ports->size(); ++i) {
84             MessagePort* dataPort = (*ports)[i].get();
85             if (dataPort == this || m_entangledChannel->isConnectedTo(dataPort)) {
86                 ec = DATA_CLONE_ERR;
87                 return;
88             }
89         }
90         channels = MessagePort::disentanglePorts(ports, ec);
91         if (ec)
92             return;
93     }
94     m_entangledChannel->postMessageToRemote(MessagePortChannel::EventData::create(message, channels.release()));
95 }
96
97 PassOwnPtr<MessagePortChannel> MessagePort::disentangle(ExceptionCode& ec)
98 {
99     if (!m_entangledChannel)
100         ec = INVALID_STATE_ERR;
101     else {
102         m_entangledChannel->disentangle();
103
104         // We can't receive any messages or generate any events, so remove ourselves from the list of active ports.
105         ASSERT(m_scriptExecutionContext);
106         m_scriptExecutionContext->destroyedMessagePort(this);
107         m_scriptExecutionContext = 0;
108     }
109     return m_entangledChannel.release();
110 }
111
112 // Invoked to notify us that there are messages available for this port.
113 // This code may be called from another thread, and so should not call any non-threadsafe APIs (i.e. should not call into the entangled channel or access mutable variables).
114 void MessagePort::messageAvailable()
115 {
116     ASSERT(m_scriptExecutionContext);
117     m_scriptExecutionContext->processMessagePortMessagesSoon();
118 }
119
120 void MessagePort::start()
121 {
122     // Do nothing if we've been cloned or closed.
123     if (!isEntangled())
124         return;
125
126     ASSERT(m_scriptExecutionContext);
127     if (m_started)
128         return;
129
130     m_started = true;
131     m_scriptExecutionContext->processMessagePortMessagesSoon();
132 }
133
134 void MessagePort::close()
135 {
136     if (!isEntangled())
137         return;
138     m_entangledChannel->close();
139     m_closed = true;
140 }
141
142 void MessagePort::entangle(PassOwnPtr<MessagePortChannel> remote)
143 {
144     // Only invoked to set our initial entanglement.
145     ASSERT(!m_entangledChannel);
146     ASSERT(m_scriptExecutionContext);
147
148     // Don't entangle the ports if the channel is closed.
149     if (remote->entangleIfOpen(this))
150         m_entangledChannel = remote;
151 }
152
153 void MessagePort::contextDestroyed()
154 {
155     ASSERT(m_scriptExecutionContext);
156     // Must be closed before blowing away the cached context, to ensure that we get no more calls to messageAvailable().
157     // ScriptExecutionContext::closeMessagePorts() takes care of that.
158     ASSERT(m_closed);
159     m_scriptExecutionContext = 0;
160 }
161
162 const AtomicString& MessagePort::interfaceName() const
163 {
164     return eventNames().interfaceForMessagePort;
165 }
166
167 ScriptExecutionContext* MessagePort::scriptExecutionContext() const
168 {
169     return m_scriptExecutionContext;
170 }
171
172 void MessagePort::dispatchMessages()
173 {
174     // Messages for contexts that are not fully active get dispatched too, but JSAbstractEventListener::handleEvent() doesn't call handlers for these.
175     // The HTML5 spec specifies that any messages sent to a document that is not fully active should be dropped, so this behavior is OK.
176     ASSERT(started());
177
178     OwnPtr<MessagePortChannel::EventData> eventData;
179     while (m_entangledChannel && m_entangledChannel->tryGetMessageFromRemote(eventData)) {
180
181 #if ENABLE(WORKERS)
182         // close() in Worker onmessage handler should prevent next message from dispatching.
183         if (m_scriptExecutionContext->isWorkerContext() && static_cast<WorkerContext*>(m_scriptExecutionContext)->isClosing())
184             return;
185 #endif
186
187         OwnPtr<MessagePortArray> ports = MessagePort::entanglePorts(*m_scriptExecutionContext, eventData->channels());
188         RefPtr<Event> evt = MessageEvent::create(ports.release(), eventData->message());
189
190         ExceptionCode ec = 0;
191         dispatchEvent(evt.release(), ec);
192         ASSERT(!ec);
193     }
194 }
195
196 bool MessagePort::hasPendingActivity()
197 {
198     // The spec says that entangled message ports should always be treated as if they have a strong reference.
199     // We'll also stipulate that the queue needs to be open (if the app drops its reference to the port before start()-ing it, then it's not really entangled as it's unreachable).
200     if (m_started && m_entangledChannel && m_entangledChannel->hasPendingActivity())
201         return true;
202     if (isEntangled() && !locallyEntangledPort())
203         return true;
204     return false;
205 }
206
207 MessagePort* MessagePort::locallyEntangledPort()
208 {
209     return m_entangledChannel ? m_entangledChannel->locallyEntangledPort(m_scriptExecutionContext) : 0;
210 }
211
212 PassOwnPtr<MessagePortChannelArray> MessagePort::disentanglePorts(const MessagePortArray* ports, ExceptionCode& ec)
213 {
214     if (!ports || !ports->size())
215         return nullptr;
216
217     // HashSet used to efficiently check for duplicates in the passed-in array.
218     HashSet<MessagePort*> portSet;
219
220     // Walk the incoming array - if there are any duplicate ports, or null ports or cloned ports, throw an error (per section 8.3.3 of the HTML5 spec).
221     for (unsigned int i = 0; i < ports->size(); ++i) {
222         MessagePort* port = (*ports)[i].get();
223         if (!port || port->isCloned() || portSet.contains(port)) {
224             ec = DATA_CLONE_ERR;
225             return nullptr;
226         }
227         portSet.add(port);
228     }
229
230     // Passed-in ports passed validity checks, so we can disentangle them.
231     OwnPtr<MessagePortChannelArray> portArray = adoptPtr(new MessagePortChannelArray(ports->size()));
232     for (unsigned int i = 0 ; i < ports->size() ; ++i) {
233         OwnPtr<MessagePortChannel> channel = (*ports)[i]->disentangle(ec);
234         ASSERT(!ec); // Can't generate exception here if passed above checks.
235         (*portArray)[i] = channel.release();
236     }
237     return portArray.release();
238 }
239
240 PassOwnPtr<MessagePortArray> MessagePort::entanglePorts(ScriptExecutionContext& context, PassOwnPtr<MessagePortChannelArray> channels)
241 {
242     if (!channels || !channels->size())
243         return nullptr;
244
245     OwnPtr<MessagePortArray> portArray = adoptPtr(new MessagePortArray(channels->size()));
246     for (unsigned int i = 0; i < channels->size(); ++i) {
247         RefPtr<MessagePort> port = MessagePort::create(context);
248         port->entangle((*channels)[i].release());
249         (*portArray)[i] = port.release();
250     }
251     return portArray.release();
252 }
253
254 EventTargetData* MessagePort::eventTargetData()
255 {
256     return &m_eventTargetData;
257 }
258
259 EventTargetData* MessagePort::ensureEventTargetData()
260 {
261     return &m_eventTargetData;
262 }
263
264 } // namespace WebCore