[ABANDONED] Introduce blink-side bridges for the new WebSocket implementation.

Source/modules/websockets/NewWebSocketChannelImpl.h

+/*
+ * 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 "core/platform/Timer.h"
+#include "modules/websockets/WebSocketChannel.h"
+#include "public/platform/WebSocketHandle.h"
+#include "public/platform/WebSocketHandleClient.h"
+#include "wtf/ArrayBuffer.h"
+#include "wtf/Deque.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::WebSocketHandleClient {
+    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));
+    }
+    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,
+    };
+
+    enum SuspendState {
+        Suspended,
+        Resuming,
+        Active,
+    };
+
+    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); }
+
+    // WebSocketHandleClient functions.
+    virtual void didConnect(WebKit::WebSocketHandle*, bool succeed, const WebKit::WebString& selectedProtocol, const WebKit::WebString& extensions) OVERRIDE;
+    virtual void didReceiveData(WebKit::WebSocketHandle*, WebKit::WebSocketHandle::MessageType, const char* data, size_t /* size */, bool fin) OVERRIDE;
+    virtual void didClose(WebKit::WebSocketHandle*, unsigned short code, const WebKit::WebString& reason) OVERRIDE;
+    virtual void didReceiveFlowControl(WebKit::WebSocketHandle*, 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 resumeTimerFired(Timer<NewWebSocketChannelImpl>*);
+    void handleTextMessage(Vector<char>*);
+    void handleBinaryMessage(Vector<char>*);
+    void handleDidClose(unsigned short code, const String& reason);
+    void startLoadingBlob(const Blob&);
+    // Calling this function may delete this object.
+    void processPendingEvents();
+
+    // WebSocketChannel functions.
+    virtual void refWebSocketChannel() OVERRIDE { ref(); }
+    virtual void derefWebSocketChannel() OVERRIDE { deref(); }
+
+    // m_context can be accessed when m_state != Closed.
+    ScriptExecutionContext* m_context;
+    WebSocketChannel* m_channel;
+    OwnPtr<WebKit::WebSocketHandle> m_handle;
+    WebSocketChannelClient* m_client;
+    // An identifier which is used for identifying this WebSocketChannel
+    // in devtools.
+    // m_identifier == 0 means that we could not obtain a valid identifier.
+    // m_identifier will be valid only if m_context is a Document.
+    unsigned long m_identifier;
+    KURL m_url;
+    OwnPtr<FileReaderLoader> m_blobLoader;
+    Deque<Message> m_messages;
+    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;
+
+    // Other state variables (m_state, m_hasAlreadyFailed, ...) don't care
+    // about the suspend state. e.g. m_state will become Closed
+    // when it receives didClose message from m_handle even if
+    // it is suspended. The code and the reason will be stored
+    // until it resumes.
+    SuspendState m_suspendState;
+    Vector<PendingEvent> m_pendingEvents;
+    Timer<NewWebSocketChannelImpl> m_resumeTimer;
+
+    String m_sourceURLAtConnection;
+    int m_lineNumberAtConnection;
+
+    static const int64_t receivedDataSizeForFlowControlHighWaterMark = 1 << 13;
+};
+
+} // namespace WebCore
+
+#endif // NewWebSocketChannelImpl_h