| /* |
| * Copyright (C) 2011, 2012 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. |
| */ |
| |
| #include "web/WebPepperSocketImpl.h" |
| |
| #include "core/dom/DOMArrayBuffer.h" |
| #include "core/dom/Document.h" |
| #include "core/inspector/ConsoleTypes.h" |
| #include "modules/websockets/DocumentWebSocketChannel.h" |
| #include "modules/websockets/WebSocketChannel.h" |
| #include "public/platform/WebString.h" |
| #include "public/platform/WebURL.h" |
| #include "public/web/WebArrayBuffer.h" |
| #include "public/web/WebDocument.h" |
| #include "web/WebPepperSocketChannelClientProxy.h" |
| #include "wtf/text/CString.h" |
| #include "wtf/text/WTFString.h" |
| #include <memory> |
| |
| namespace blink { |
| |
| WebPepperSocketImpl::WebPepperSocketImpl(const WebDocument& document, WebPepperSocketClient* client) |
| : m_client(client) |
| , m_channelProxy(WebPepperSocketChannelClientProxy::create(this)) |
| , m_binaryType(BinaryTypeBlob) |
| , m_isClosingOrClosed(false) |
| , m_bufferedAmount(0) |
| , m_bufferedAmountAfterClose(0) |
| { |
| Document* coreDocument = document; |
| m_private = DocumentWebSocketChannel::create(coreDocument, m_channelProxy.get(), SourceLocation::capture()); |
| } |
| |
| WebPepperSocketImpl::~WebPepperSocketImpl() |
| { |
| m_private->disconnect(); |
| } |
| |
| WebPepperSocket::BinaryType WebPepperSocketImpl::binaryType() const |
| { |
| return m_binaryType; |
| } |
| |
| bool WebPepperSocketImpl::setBinaryType(BinaryType binaryType) |
| { |
| if (binaryType > BinaryTypeArrayBuffer) |
| return false; |
| m_binaryType = binaryType; |
| return true; |
| } |
| |
| void WebPepperSocketImpl::connect(const WebURL& url, const WebString& protocol) |
| { |
| m_private->connect(url, protocol); |
| } |
| |
| WebString WebPepperSocketImpl::subprotocol() |
| { |
| return m_subprotocol; |
| } |
| |
| WebString WebPepperSocketImpl::extensions() |
| { |
| return m_extensions; |
| } |
| |
| bool WebPepperSocketImpl::sendText(const WebString& message) |
| { |
| String coreMessage = message; |
| CString encodedMessage = coreMessage.utf8(); |
| size_t size = encodedMessage.length(); |
| m_bufferedAmount += size; |
| if (m_isClosingOrClosed) |
| m_bufferedAmountAfterClose += size; |
| |
| // FIXME: Deprecate this call. |
| m_client->didUpdateBufferedAmount(m_bufferedAmount); |
| |
| if (m_isClosingOrClosed) |
| return true; |
| |
| m_private->send(encodedMessage); |
| return true; |
| } |
| |
| bool WebPepperSocketImpl::sendArrayBuffer(const WebArrayBuffer& webArrayBuffer) |
| { |
| size_t size = webArrayBuffer.byteLength(); |
| m_bufferedAmount += size; |
| if (m_isClosingOrClosed) |
| m_bufferedAmountAfterClose += size; |
| |
| // FIXME: Deprecate this call. |
| m_client->didUpdateBufferedAmount(m_bufferedAmount); |
| |
| if (m_isClosingOrClosed) |
| return true; |
| |
| DOMArrayBuffer* arrayBuffer = webArrayBuffer; |
| m_private->send(*arrayBuffer, 0, arrayBuffer->byteLength()); |
| return true; |
| } |
| |
| unsigned long WebPepperSocketImpl::bufferedAmount() const |
| { |
| return m_bufferedAmount; |
| } |
| |
| void WebPepperSocketImpl::close(int code, const WebString& reason) |
| { |
| m_isClosingOrClosed = true; |
| m_private->close(code, reason); |
| } |
| |
| void WebPepperSocketImpl::fail(const WebString& reason) |
| { |
| m_private->fail(reason, ErrorMessageLevel, SourceLocation::create(String(), 0, 0, nullptr)); |
| } |
| |
| void WebPepperSocketImpl::disconnect() |
| { |
| m_private->disconnect(); |
| m_client = nullptr; |
| } |
| |
| void WebPepperSocketImpl::didConnect(const String& subprotocol, const String& extensions) |
| { |
| m_client->didConnect(subprotocol, extensions); |
| |
| // FIXME: Deprecate these statements. |
| m_subprotocol = subprotocol; |
| m_extensions = extensions; |
| m_client->didConnect(); |
| } |
| |
| void WebPepperSocketImpl::didReceiveTextMessage(const String& payload) |
| { |
| m_client->didReceiveMessage(WebString(payload)); |
| } |
| |
| void WebPepperSocketImpl::didReceiveBinaryMessage(std::unique_ptr<Vector<char>> payload) |
| { |
| switch (m_binaryType) { |
| case BinaryTypeBlob: |
| // FIXME: Handle Blob after supporting WebBlob. |
| break; |
| case BinaryTypeArrayBuffer: |
| m_client->didReceiveArrayBuffer(WebArrayBuffer(DOMArrayBuffer::create(payload->data(), payload->size()))); |
| break; |
| } |
| } |
| |
| void WebPepperSocketImpl::didError() |
| { |
| m_client->didReceiveMessageError(); |
| } |
| |
| void WebPepperSocketImpl::didConsumeBufferedAmount(unsigned long consumed) |
| { |
| m_client->didConsumeBufferedAmount(consumed); |
| |
| // FIXME: Deprecate the following statements. |
| m_bufferedAmount -= consumed; |
| m_client->didUpdateBufferedAmount(m_bufferedAmount); |
| } |
| |
| void WebPepperSocketImpl::didStartClosingHandshake() |
| { |
| m_client->didStartClosingHandshake(); |
| } |
| |
| void WebPepperSocketImpl::didClose(WebSocketChannelClient::ClosingHandshakeCompletionStatus status, unsigned short code, const String& reason) |
| { |
| m_isClosingOrClosed = true; |
| m_client->didClose(static_cast<WebPepperSocketClient::ClosingHandshakeCompletionStatus>(status), code, WebString(reason)); |
| |
| // FIXME: Deprecate this call. |
| m_client->didClose(m_bufferedAmount - m_bufferedAmountAfterClose, static_cast<WebPepperSocketClient::ClosingHandshakeCompletionStatus>(status), code, WebString(reason)); |
| } |
| |
| } // namespace blink |