| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (C) 2011, 2012 Google Inc. All rights reserved. | 2 * Copyright (C) 2011, 2012 Google Inc. All rights reserved. |
| 3 * | 3 * |
| 4 * Redistribution and use in source and binary forms, with or without | 4 * Redistribution and use in source and binary forms, with or without |
| 5 * modification, are permitted provided that the following conditions are | 5 * modification, are permitted provided that the following conditions are |
| 6 * met: | 6 * met: |
| 7 * | 7 * |
| 8 * * Redistributions of source code must retain the above copyright | 8 * * Redistributions of source code must retain the above copyright |
| 9 * notice, this list of conditions and the following disclaimer. | 9 * notice, this list of conditions and the following disclaimer. |
| 10 * * Redistributions in binary form must reproduce the above | 10 * * Redistributions in binary form must reproduce the above |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 68 return adoptRefWillBeRefCountedGarbageCollected(new MainThreadWebSocketC
hannel(document, client, sourceURL, lineNumber)); | 68 return adoptRefWillBeRefCountedGarbageCollected(new MainThreadWebSocketC
hannel(document, client, sourceURL, lineNumber)); |
| 69 } | 69 } |
| 70 virtual ~MainThreadWebSocketChannel(); | 70 virtual ~MainThreadWebSocketChannel(); |
| 71 | 71 |
| 72 // WebSocketChannel functions. | 72 // WebSocketChannel functions. |
| 73 virtual bool connect(const KURL&, const String& protocol) OVERRIDE; | 73 virtual bool connect(const KURL&, const String& protocol) OVERRIDE; |
| 74 virtual WebSocketChannel::SendResult send(const String& message) OVERRIDE; | 74 virtual WebSocketChannel::SendResult send(const String& message) OVERRIDE; |
| 75 virtual WebSocketChannel::SendResult send(const ArrayBuffer&, unsigned byteO
ffset, unsigned byteLength) OVERRIDE; | 75 virtual WebSocketChannel::SendResult send(const ArrayBuffer&, unsigned byteO
ffset, unsigned byteLength) OVERRIDE; |
| 76 virtual WebSocketChannel::SendResult send(PassRefPtr<BlobDataHandle>) OVERRI
DE; | 76 virtual WebSocketChannel::SendResult send(PassRefPtr<BlobDataHandle>) OVERRI
DE; |
| 77 virtual WebSocketChannel::SendResult send(PassOwnPtr<Vector<char> > data) OV
ERRIDE; | 77 virtual WebSocketChannel::SendResult send(PassOwnPtr<Vector<char> > data) OV
ERRIDE; |
| 78 virtual unsigned long bufferedAmount() const OVERRIDE; |
| 78 // Start closing handshake. Use the CloseEventCodeNotSpecified for the code | 79 // Start closing handshake. Use the CloseEventCodeNotSpecified for the code |
| 79 // argument to omit payload. | 80 // argument to omit payload. |
| 80 virtual void close(int code, const String& reason) OVERRIDE; | 81 virtual void close(int code, const String& reason) OVERRIDE; |
| 81 virtual void fail(const String& reason, MessageLevel, const String&, unsigne
d lineNumber) OVERRIDE; | 82 virtual void fail(const String& reason, MessageLevel, const String&, unsigne
d lineNumber) OVERRIDE; |
| 82 virtual void disconnect() OVERRIDE; | 83 virtual void disconnect() OVERRIDE; |
| 83 | 84 |
| 84 virtual void suspend() OVERRIDE; | 85 virtual void suspend() OVERRIDE; |
| 85 virtual void resume() OVERRIDE; | 86 virtual void resume() OVERRIDE; |
| 86 | 87 |
| 87 // SocketStreamHandleClient functions. | 88 // SocketStreamHandleClient functions. |
| 88 virtual void didOpenSocketStream(SocketStreamHandle*) OVERRIDE; | 89 virtual void didOpenSocketStream(SocketStreamHandle*) OVERRIDE; |
| 89 virtual void didCloseSocketStream(SocketStreamHandle*) OVERRIDE; | 90 virtual void didCloseSocketStream(SocketStreamHandle*) OVERRIDE; |
| 90 virtual void didReceiveSocketStreamData(SocketStreamHandle*, const char*, in
t) OVERRIDE; | 91 virtual void didReceiveSocketStreamData(SocketStreamHandle*, const char*, in
t) OVERRIDE; |
| 91 virtual void didConsumeBufferedAmount(SocketStreamHandle*, size_t consumed)
OVERRIDE; | 92 virtual void didUpdateBufferedAmount(SocketStreamHandle*, size_t bufferedAmo
unt) OVERRIDE; |
| 92 virtual void didFailSocketStream(SocketStreamHandle*, const SocketStreamErro
r&) OVERRIDE; | 93 virtual void didFailSocketStream(SocketStreamHandle*, const SocketStreamErro
r&) OVERRIDE; |
| 93 | 94 |
| 94 // FileReaderLoaderClient functions. | 95 // FileReaderLoaderClient functions. |
| 95 virtual void didStartLoading() OVERRIDE; | 96 virtual void didStartLoading() OVERRIDE; |
| 96 virtual void didReceiveData() OVERRIDE; | 97 virtual void didReceiveData() OVERRIDE; |
| 97 virtual void didFinishLoading() OVERRIDE; | 98 virtual void didFinishLoading() OVERRIDE; |
| 98 virtual void didFail(FileError::ErrorCode) OVERRIDE; | 99 virtual void didFail(FileError::ErrorCode) OVERRIDE; |
| 99 | 100 |
| 100 private: | 101 private: |
| 101 MainThreadWebSocketChannel(Document*, WebSocketChannelClient*, const String&
, unsigned); | 102 MainThreadWebSocketChannel(Document*, WebSocketChannelClient*, const String&
, unsigned); |
| 102 | 103 |
| 103 class FramingOverhead { | |
| 104 public: | |
| 105 FramingOverhead(WebSocketFrame::OpCode opcode, size_t frameDataSize, siz
e_t originalPayloadLength) | |
| 106 : m_opcode(opcode) | |
| 107 , m_frameDataSize(frameDataSize) | |
| 108 , m_originalPayloadLength(originalPayloadLength) | |
| 109 { | |
| 110 } | |
| 111 | |
| 112 WebSocketFrame::OpCode opcode() const { return m_opcode; } | |
| 113 size_t frameDataSize() const { return m_frameDataSize; } | |
| 114 size_t originalPayloadLength() const { return m_originalPayloadLength; } | |
| 115 | |
| 116 private: | |
| 117 WebSocketFrame::OpCode m_opcode; | |
| 118 size_t m_frameDataSize; | |
| 119 size_t m_originalPayloadLength; | |
| 120 }; | |
| 121 | |
| 122 void clearDocument(); | 104 void clearDocument(); |
| 123 | 105 |
| 124 void disconnectHandle(); | 106 void disconnectHandle(); |
| 125 | 107 |
| 126 // Calls didReceiveMessageError() on m_client if we haven't yet. | 108 // Calls didReceiveMessageError() on m_client if we haven't yet. |
| 127 void callDidReceiveMessageError(); | 109 void callDidReceiveMessageError(); |
| 128 | 110 |
| 129 bool appendToBuffer(const char* data, size_t len); | 111 bool appendToBuffer(const char* data, size_t len); |
| 130 void skipBuffer(size_t len); | 112 void skipBuffer(size_t len); |
| 131 // Repeats parsing data from m_buffer until instructed to stop. | 113 // Repeats parsing data from m_buffer until instructed to stop. |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 207 | 189 |
| 208 Timer<MainThreadWebSocketChannel> m_resumeTimer; | 190 Timer<MainThreadWebSocketChannel> m_resumeTimer; |
| 209 bool m_suspended; | 191 bool m_suspended; |
| 210 bool m_didFailOfClientAlreadyRun; | 192 bool m_didFailOfClientAlreadyRun; |
| 211 // Set to true iff this instance called disconnect() on m_handle. | 193 // Set to true iff this instance called disconnect() on m_handle. |
| 212 bool m_hasCalledDisconnectOnHandle; | 194 bool m_hasCalledDisconnectOnHandle; |
| 213 bool m_receivedClosingHandshake; | 195 bool m_receivedClosingHandshake; |
| 214 Timer<MainThreadWebSocketChannel> m_closingTimer; | 196 Timer<MainThreadWebSocketChannel> m_closingTimer; |
| 215 ChannelState m_state; | 197 ChannelState m_state; |
| 216 bool m_shouldDiscardReceivedData; | 198 bool m_shouldDiscardReceivedData; |
| 199 unsigned long m_unhandledBufferedAmount; |
| 217 | 200 |
| 218 unsigned long m_identifier; // m_identifier == 0 means that we could not obt
ain a valid identifier. | 201 unsigned long m_identifier; // m_identifier == 0 means that we could not obt
ain a valid identifier. |
| 219 | 202 |
| 220 // Private members only for hybi-10 protocol. | 203 // Private members only for hybi-10 protocol. |
| 221 bool m_hasContinuousFrame; | 204 bool m_hasContinuousFrame; |
| 222 WebSocketFrame::OpCode m_continuousFrameOpCode; | 205 WebSocketFrame::OpCode m_continuousFrameOpCode; |
| 223 Vector<char> m_continuousFrameData; | 206 Vector<char> m_continuousFrameData; |
| 224 unsigned short m_closeEventCode; | 207 unsigned short m_closeEventCode; |
| 225 String m_closeEventReason; | 208 String m_closeEventReason; |
| 226 | 209 |
| 227 Deque<OwnPtr<QueuedFrame> > m_outgoingFrameQueue; | 210 Deque<OwnPtr<QueuedFrame> > m_outgoingFrameQueue; |
| 228 OutgoingFrameQueueStatus m_outgoingFrameQueueStatus; | 211 OutgoingFrameQueueStatus m_outgoingFrameQueueStatus; |
| 229 Deque<FramingOverhead> m_framingOverheadQueue; | |
| 230 // The number of bytes that are already consumed (i.e. sent) in the | |
| 231 // current frame. | |
| 232 size_t m_numConsumedBytesInCurrentFrame; | |
| 233 | 212 |
| 234 // FIXME: Load two or more Blobs simultaneously for better performance. | 213 // FIXME: Load two or more Blobs simultaneously for better performance. |
| 235 OwnPtr<FileReaderLoader> m_blobLoader; | 214 OwnPtr<FileReaderLoader> m_blobLoader; |
| 236 BlobLoaderStatus m_blobLoaderStatus; | 215 BlobLoaderStatus m_blobLoaderStatus; |
| 237 | 216 |
| 238 // Source code position where construction happened. To be used to show a | 217 // Source code position where construction happened. To be used to show a |
| 239 // console message where no JS callstack info available. | 218 // console message where no JS callstack info available. |
| 240 String m_sourceURLAtConstruction; | 219 String m_sourceURLAtConstruction; |
| 241 unsigned m_lineNumberAtConstruction; | 220 unsigned m_lineNumberAtConstruction; |
| 242 | 221 |
| 243 WebSocketPerMessageDeflate m_perMessageDeflate; | 222 WebSocketPerMessageDeflate m_perMessageDeflate; |
| 244 | 223 |
| 245 WebSocketDeflateFramer m_deflateFramer; | 224 WebSocketDeflateFramer m_deflateFramer; |
| 246 }; | 225 }; |
| 247 | 226 |
| 248 } // namespace WebCore | 227 } // namespace WebCore |
| 249 | 228 |
| 250 #endif // MainThreadWebSocketChannel_h | 229 #endif // MainThreadWebSocketChannel_h |
| OLD | NEW |