2 * Copyright (C) 2011, 2012 Google Inc. All rights reserved.
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
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
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.
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.
31 #ifndef MainThreadWebSocketChannel_h
32 #define MainThreadWebSocketChannel_h
34 #include "core/fileapi/FileError.h"
35 #include "core/fileapi/FileReaderLoaderClient.h"
36 #include "core/frame/ConsoleTypes.h"
37 #include "modules/websockets/WebSocketChannel.h"
38 #include "modules/websockets/WebSocketDeflateFramer.h"
39 #include "modules/websockets/WebSocketFrame.h"
40 #include "modules/websockets/WebSocketHandshake.h"
41 #include "modules/websockets/WebSocketPerMessageDeflate.h"
42 #include "platform/Timer.h"
43 #include "platform/network/SocketStreamHandleClient.h"
44 #include "wtf/Deque.h"
45 #include "wtf/Forward.h"
46 #include "wtf/PassOwnPtr.h"
47 #include "wtf/RefCounted.h"
48 #include "wtf/Vector.h"
49 #include "wtf/text/CString.h"
55 class FileReaderLoader;
56 class SocketStreamHandle;
57 class SocketStreamError;
58 class WebSocketChannelClient;
60 class MainThreadWebSocketChannel FINAL : public RefCounted<MainThreadWebSocketChannel>, public SocketStreamHandleClient, public WebSocketChannel, public FileReaderLoaderClient {
61 WTF_MAKE_FAST_ALLOCATED;
63 // You can specify the source file and the line number information
64 // explicitly by passing the last parameter.
65 // In the usual case, they are set automatically and you don't have to
67 static PassRefPtr<MainThreadWebSocketChannel> create(Document* document, WebSocketChannelClient* client, const String& sourceURL = String(), unsigned lineNumber = 0) { return adoptRef(new MainThreadWebSocketChannel(document, client, sourceURL, lineNumber)); }
68 virtual ~MainThreadWebSocketChannel();
70 bool send(const char* data, int length);
72 // WebSocketChannel functions.
73 virtual void connect(const KURL&, const String& protocol) OVERRIDE;
74 virtual String subprotocol() OVERRIDE;
75 virtual String extensions() OVERRIDE;
76 virtual WebSocketChannel::SendResult send(const String& message) OVERRIDE;
77 virtual WebSocketChannel::SendResult send(const ArrayBuffer&, unsigned byteOffset, unsigned byteLength) OVERRIDE;
78 virtual WebSocketChannel::SendResult send(PassRefPtr<BlobDataHandle>) OVERRIDE;
79 virtual unsigned long bufferedAmount() const OVERRIDE;
80 // Start closing handshake. Use the CloseEventCodeNotSpecified for the code
81 // argument to omit payload.
82 virtual void close(int code, const String& reason) OVERRIDE;
83 virtual void fail(const String& reason, MessageLevel, const String&, unsigned lineNumber) OVERRIDE;
84 using WebSocketChannel::fail;
85 virtual void disconnect() OVERRIDE;
87 virtual void suspend() OVERRIDE;
88 virtual void resume() OVERRIDE;
90 // SocketStreamHandleClient functions.
91 virtual void willOpenSocketStream(SocketStreamHandle*) OVERRIDE;
92 virtual void didOpenSocketStream(SocketStreamHandle*) OVERRIDE;
93 virtual void didCloseSocketStream(SocketStreamHandle*) OVERRIDE;
94 virtual void didReceiveSocketStreamData(SocketStreamHandle*, const char*, int) OVERRIDE;
95 virtual void didUpdateBufferedAmount(SocketStreamHandle*, size_t bufferedAmount) OVERRIDE;
96 virtual void didFailSocketStream(SocketStreamHandle*, const SocketStreamError&) OVERRIDE;
98 // FileReaderLoaderClient functions.
99 virtual void didStartLoading() OVERRIDE;
100 virtual void didReceiveData() OVERRIDE;
101 virtual void didFinishLoading() OVERRIDE;
102 virtual void didFail(FileError::ErrorCode) OVERRIDE;
104 using RefCounted<MainThreadWebSocketChannel>::ref;
105 using RefCounted<MainThreadWebSocketChannel>::deref;
108 // WebSocketChannel functions.
109 virtual void refWebSocketChannel() OVERRIDE { ref(); }
110 virtual void derefWebSocketChannel() OVERRIDE { deref(); }
113 MainThreadWebSocketChannel(Document*, WebSocketChannelClient*, const String&, unsigned);
115 void clearDocument();
117 void disconnectHandle();
119 // Calls didReceiveMessageError() on m_client if we haven't yet.
120 void callDidReceiveMessageError();
122 bool appendToBuffer(const char* data, size_t len);
123 void skipBuffer(size_t len);
124 // Repeats parsing data from m_buffer until instructed to stop.
125 void processBuffer();
126 // Parses a handshake response or one frame from m_buffer and processes it.
127 bool processOneItemFromBuffer();
128 void resumeTimerFired(Timer<MainThreadWebSocketChannel>*);
129 void startClosingHandshake(int code, const String& reason);
130 void closingTimerFired(Timer<MainThreadWebSocketChannel>*);
132 // Parses one frame from m_buffer and processes it.
135 // It is allowed to send a Blob as a binary frame if hybi-10 protocol is in use. Sending a Blob
136 // can be delayed because it must be read asynchronously. Other types of data (String or
137 // ArrayBuffer) may also be blocked by preceding sending request of a Blob.
139 // To address this situation, messages to be sent need to be stored in a queue. Whenever a new
140 // data frame is going to be sent, it first must go to the queue. Items in the queue are processed
141 // in the order they were put into the queue. Sending request of a Blob blocks further processing
142 // until the Blob is completely read and sent to the socket stream.
143 enum QueuedFrameType {
144 QueuedFrameTypeString,
145 QueuedFrameTypeVector,
149 WebSocketFrame::OpCode opCode;
150 QueuedFrameType frameType;
151 // Only one of the following items is used, according to the value of frameType.
153 Vector<char> vectorData;
154 RefPtr<BlobDataHandle> blobData;
156 void enqueueTextFrame(const CString&);
157 void enqueueRawFrame(WebSocketFrame::OpCode, const char* data, size_t dataLength);
158 void enqueueBlobFrame(WebSocketFrame::OpCode, PassRefPtr<BlobDataHandle>);
160 void failAsError(const String& reason) { fail(reason, ErrorMessageLevel, m_sourceURLAtConstruction, m_lineNumberAtConstruction); }
161 void processOutgoingFrameQueue();
162 void abortOutgoingFrameQueue();
164 enum OutgoingFrameQueueStatus {
165 // It is allowed to put a new item into the queue.
166 OutgoingFrameQueueOpen,
167 // Close frame has already been put into the queue but may not have been sent yet;
168 // m_handle->close() will be called as soon as the queue is cleared. It is not
169 // allowed to put a new item into the queue.
170 OutgoingFrameQueueClosing,
171 // Close frame has been sent or the queue was aborted. It is not allowed to put
172 // a new item to the queue.
173 OutgoingFrameQueueClosed
176 // In principle, this method is called only by processOutgoingFrameQueue().
177 // It does work necessary to build frames including Blob loading for queued
178 // data in order. Calling this method directly jumps in the process.
179 bool sendFrame(WebSocketFrame::OpCode, const char* data, size_t dataLength);
181 enum BlobLoaderStatus {
182 BlobLoaderNotStarted,
194 Document* m_document;
195 WebSocketChannelClient* m_client;
196 OwnPtr<WebSocketHandshake> m_handshake;
197 RefPtr<SocketStreamHandle> m_handle;
198 Vector<char> m_buffer;
200 Timer<MainThreadWebSocketChannel> m_resumeTimer;
202 bool m_didFailOfClientAlreadyRun;
203 // Set to true iff this instance called disconnect() on m_handle.
204 bool m_hasCalledDisconnectOnHandle;
205 bool m_receivedClosingHandshake;
206 Timer<MainThreadWebSocketChannel> m_closingTimer;
207 ChannelState m_state;
208 bool m_shouldDiscardReceivedData;
209 unsigned long m_unhandledBufferedAmount;
211 unsigned long m_identifier; // m_identifier == 0 means that we could not obtain a valid identifier.
213 // Private members only for hybi-10 protocol.
214 bool m_hasContinuousFrame;
215 WebSocketFrame::OpCode m_continuousFrameOpCode;
216 Vector<char> m_continuousFrameData;
217 unsigned short m_closeEventCode;
218 String m_closeEventReason;
220 Deque<OwnPtr<QueuedFrame> > m_outgoingFrameQueue;
221 OutgoingFrameQueueStatus m_outgoingFrameQueueStatus;
223 // FIXME: Load two or more Blobs simultaneously for better performance.
224 OwnPtr<FileReaderLoader> m_blobLoader;
225 BlobLoaderStatus m_blobLoaderStatus;
227 // Source code position where construction happened. To be used to show a
228 // console message where no JS callstack info available.
229 String m_sourceURLAtConstruction;
230 unsigned m_lineNumberAtConstruction;
232 WebSocketPerMessageDeflate m_perMessageDeflate;
234 WebSocketDeflateFramer m_deflateFramer;
237 } // namespace WebCore
239 #endif // MainThreadWebSocketChannel_h