Chromium Code Reviews| Index: Source/modules/websockets/NewWebSocketChannelImpl.h |
| diff --git a/Source/modules/websockets/NewWebSocketChannelImpl.h b/Source/modules/websockets/NewWebSocketChannelImpl.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..29e335594557f2f2e57d55bef770c8ab31949e40 |
| --- /dev/null |
| +++ b/Source/modules/websockets/NewWebSocketChannelImpl.h |
| @@ -0,0 +1,182 @@ |
| +/* |
| + * Copyright (C) 2013 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 |
| + * copyright notice, this list of conditions and the following disclaimer |
| + * in the documentation and/or other materials provided with the |
| + * distribution. |
| + * * Neither the name of Google Inc. nor the names of its |
| + * contributors may be used to endorse or promote products derived from |
| + * this software without specific prior written permission. |
| + * |
| + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| + */ |
| + |
| +#ifndef NewWebSocketChannelImpl_h |
| +#define NewWebSocketChannelImpl_h |
| + |
| +#include "core/fileapi/Blob.h" |
| +#include "core/fileapi/FileReaderLoader.h" |
| +#include "core/fileapi/FileReaderLoaderClient.h" |
| +#include "core/platform/SharedBuffer.h" |
| +#include "modules/websockets/WebSocketChannel.h" |
| +#include "public/platform/WebSocketChannelHandle.h" |
| +#include "public/platform/WebSocketChannelHandleClient.h" |
| +#include "wtf/ArrayBuffer.h" |
| +#include "wtf/PassRefPtr.h" |
| +#include "wtf/RefCounted.h" |
| +#include "wtf/RefPtr.h" |
| +#include "wtf/Vector.h" |
| +#include "wtf/text/WTFString.h" |
| + |
| +namespace WebCore { |
| + |
| +// This class may replace MainThreadWebSocketChannel. |
| +class NewWebSocketChannelImpl : public WebSocketChannel, public RefCounted<NewWebSocketChannelImpl>, public FileReaderLoaderClient, public WebKit::WebSocketChannelHandleClient { |
| + WTF_MAKE_FAST_ALLOCATED; |
| +public: |
| + // You can specify the source file and the line number information |
| + // explicitly by passing the last parameter. |
| + // In the usual case, they are set automatically and you don't have to |
| + // pass it. |
| + static PassRefPtr<NewWebSocketChannelImpl> create(ScriptExecutionContext* context, WebSocketChannelClient* client, const String& sourceURL = String(), unsigned lineNumber = 0) { return adoptRef(new NewWebSocketChannelImpl(context, client, sourceURL, lineNumber)); } |
|
tyoshino (SeeGerritForStatus)
2013/08/22 08:25:22
let's break at {
yhirano
2013/08/22 08:40:31
Done.
|
| + virtual ~NewWebSocketChannelImpl() { } |
| + |
| + // WebSocketChannel functions. |
| + virtual void connect(const KURL&, const String& protocol) OVERRIDE; |
| + virtual String subprotocol() OVERRIDE; |
| + virtual String extensions() OVERRIDE; |
| + virtual WebSocketChannel::SendResult send(const String& message) OVERRIDE; |
| + virtual WebSocketChannel::SendResult send(const ArrayBuffer&, unsigned byteOffset, unsigned byteLength) OVERRIDE; |
| + virtual WebSocketChannel::SendResult send(const Blob&) OVERRIDE; |
| + virtual unsigned long bufferedAmount() const OVERRIDE; |
| + // Start closing handshake. Use the CloseEventCodeNotSpecified for the code |
| + // argument to omit payload. |
| + virtual void close(int code, const String& reason) OVERRIDE; |
| + virtual void fail(const String& reason, MessageLevel, const String&, unsigned lineNumber) OVERRIDE; |
| + using WebSocketChannel::fail; |
| + virtual void disconnect() OVERRIDE; |
| + |
| + using RefCounted<NewWebSocketChannelImpl>::ref; |
| + using RefCounted<NewWebSocketChannelImpl>::deref; |
| + |
| + virtual void suspend() OVERRIDE; |
| + virtual void resume() OVERRIDE; |
| + |
| +private: |
| + enum MessageType { |
| + MessageTypeText, |
| + MessageTypeBlob, |
| + MessageTypeArrayBuffer, |
| + }; |
| + struct Message { |
| + Message(const String&); |
| + Message(const Blob&); |
| + Message(PassRefPtr<ArrayBuffer>); |
| + MessageType type; |
| + CString text; |
| + RefPtr<Blob> blob; |
| + RefPtr<ArrayBuffer> arrayBuffer; |
| + }; |
| + |
| + struct ReceivedMessage { |
| + bool isMessageText; |
| + Vector<char> data; |
| + }; |
| + |
| + enum State { |
| + NotConnected, |
| + Connecting, |
| + Open, |
| + Closing, |
| + Closed, |
| + }; |
| + |
| + struct PendingEvent { |
| + enum Type { |
| + DidConnectComplete, |
| + DidReceiveTextMessage, |
| + DidReceiveBinaryMessage, |
| + DidReceiveError, |
| + DidClose, |
| + }; |
| + Type type; |
| + Vector<char> message; // for DidReceive*Message |
| + int closingCode; // for DidClose |
| + String closingReason; // for DidClose |
| + |
| + PendingEvent(Type type) : type(type) { } |
| + PendingEvent(int code, const String& reason) : type(DidClose), closingCode(code), closingReason(reason) { } |
| + }; |
| + |
| + NewWebSocketChannelImpl(ScriptExecutionContext*, WebSocketChannelClient*, const String&, unsigned); |
| + void sendInternal(); |
| + void flowControlIfNecessary(); |
| + void failAsError(const String& reason) { fail(reason, ErrorMessageLevel, m_sourceURLAtConnection, m_lineNumberAtConnection); } |
| + |
| + // WebSocketChannelHandleClient functions. |
| + virtual void didConnect(WebKit::WebSocketChannelHandle*, bool succeed, const WebKit::WebString& selectedProtocol, const WebKit::WebString& extensions) OVERRIDE; |
| + virtual void didReceiveData(WebKit::WebSocketChannelHandle*, const WebKit::WebData&, WebKit::WebSocketChannelHandle::MessageType, bool fin) OVERRIDE; |
| + virtual void didClose(WebKit::WebSocketChannelHandle*, unsigned short code, const WebKit::WebString& reason) OVERRIDE; |
| + virtual void didReceiveFlowControl(WebKit::WebSocketChannelHandle*, int64_t quota) OVERRIDE { m_sendingQuota += quota; } |
| + |
| + // FileReaderLoaderClient functions. |
| + virtual void didStartLoading() OVERRIDE { } |
| + virtual void didReceiveData() OVERRIDE { } |
| + virtual void didFinishLoading() OVERRIDE; |
| + virtual void didFail(FileError::ErrorCode) OVERRIDE; |
| + |
| + void startLoadingBlob(const Blob&); |
| + // Calling this function may delete this object. |
| + void processPendingEvents(); |
| + |
|
tyoshino (SeeGerritForStatus)
2013/08/22 08:25:22
// WebSocketChannel functions
yhirano
2013/08/22 08:40:31
Done.
|
| + virtual void refWebSocketChannel() OVERRIDE { ref(); } |
| + virtual void derefWebSocketChannel() OVERRIDE { deref(); } |
| + |
| + ScriptExecutionContext* m_context; |
| + WebSocketChannel* m_channel; |
| + OwnPtr<WebKit::WebSocketChannelHandle> m_handle; |
| + WebSocketChannelClient* m_client; |
| + unsigned long m_identifier; // m_identifier == 0 means that we could not obtain a valid identifier. |
| + KURL m_url; |
| + OwnPtr<FileReaderLoader> m_blobLoader; |
| + bool m_isLoadingBlob; |
| + Vector<Message> m_messages; |
| + Vector<PendingEvent> m_pendingEvents; |
| + Vector<char> m_receivingMessageData; |
| + |
| + State m_state; |
| + bool m_receivingMessageTypeIsText; |
| + int64_t m_sendingQuota; |
| + int64_t m_receivedDataSizeForFlowControl; |
| + unsigned long m_bufferedAmount; |
| + size_t m_sentSizeOfTopMessage; |
| + String m_subprotocol; |
| + String m_extensions; |
| + bool m_hasAlreadyFailed; |
| + bool m_isSuspended; |
| + String m_sourceURLAtConnection; |
| + int m_lineNumberAtConnection; |
| + |
| + static const int64_t receivedDataSizeForFlowControlHighWaterMark = 1 << 13; |
| +}; |
| + |
| +} // namespace WebCore |
| + |
| +#endif // WebSocketMessagePool_h |
|
tyoshino (SeeGerritForStatus)
2013/08/22 08:25:22
correct comment
yhirano
2013/08/22 08:40:31
Done.
|