• Home
  • History
  • Annotate
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2011, 2012 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
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
13  * distribution.
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.
17  *
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.
29  */
30 
31 #include "config.h"
32 #include "web/WebSocketImpl.h"
33 
34 #include "core/dom/Document.h"
35 #include "core/frame/ConsoleTypes.h"
36 #include "modules/websockets/MainThreadWebSocketChannel.h"
37 #include "modules/websockets/NewWebSocketChannelImpl.h"
38 #include "modules/websockets/WebSocketChannel.h"
39 #include "platform/RuntimeEnabledFeatures.h"
40 #include "public/platform/WebArrayBuffer.h"
41 #include "public/platform/WebString.h"
42 #include "public/platform/WebURL.h"
43 #include "public/web/WebDocument.h"
44 #include "web/WebSocketChannelClientProxy.h"
45 #include "wtf/ArrayBuffer.h"
46 #include "wtf/text/CString.h"
47 #include "wtf/text/WTFString.h"
48 
49 namespace blink {
50 
WebSocketImpl(const WebDocument & document,WebSocketClient * client)51 WebSocketImpl::WebSocketImpl(const WebDocument& document, WebSocketClient* client)
52     : m_client(client)
53     , m_channelProxy(WebSocketChannelClientProxy::create(this))
54     , m_binaryType(BinaryTypeBlob)
55     , m_isClosingOrClosed(false)
56     , m_bufferedAmount(0)
57     , m_bufferedAmountAfterClose(0)
58 {
59     RefPtrWillBeRawPtr<Document> coreDocument = PassRefPtrWillBeRawPtr<Document>(document);
60     if (RuntimeEnabledFeatures::experimentalWebSocketEnabled()) {
61         m_private = NewWebSocketChannelImpl::create(coreDocument.get(), m_channelProxy.get());
62     } else {
63         m_private = MainThreadWebSocketChannel::create(coreDocument.get(), m_channelProxy.get());
64     }
65 }
66 
~WebSocketImpl()67 WebSocketImpl::~WebSocketImpl()
68 {
69     m_private->disconnect();
70 }
71 
binaryType() const72 WebSocket::BinaryType WebSocketImpl::binaryType() const
73 {
74     return m_binaryType;
75 }
76 
setBinaryType(BinaryType binaryType)77 bool WebSocketImpl::setBinaryType(BinaryType binaryType)
78 {
79     if (binaryType > BinaryTypeArrayBuffer)
80         return false;
81     m_binaryType = binaryType;
82     return true;
83 }
84 
connect(const WebURL & url,const WebString & protocol)85 void WebSocketImpl::connect(const WebURL& url, const WebString& protocol)
86 {
87     m_private->connect(url, protocol);
88 }
89 
subprotocol()90 WebString WebSocketImpl::subprotocol()
91 {
92     return m_subprotocol;
93 }
94 
extensions()95 WebString WebSocketImpl::extensions()
96 {
97     return m_extensions;
98 }
99 
sendText(const WebString & message)100 bool WebSocketImpl::sendText(const WebString& message)
101 {
102     size_t size = message.utf8().length();
103     m_bufferedAmount += size;
104     if (m_isClosingOrClosed)
105         m_bufferedAmountAfterClose += size;
106 
107     // FIXME: Deprecate this call.
108     m_client->didUpdateBufferedAmount(m_bufferedAmount);
109 
110     if (m_isClosingOrClosed)
111         return true;
112 
113     m_private->send(message);
114     return true;
115 }
116 
sendArrayBuffer(const WebArrayBuffer & webArrayBuffer)117 bool WebSocketImpl::sendArrayBuffer(const WebArrayBuffer& webArrayBuffer)
118 {
119     size_t size = webArrayBuffer.byteLength();
120     m_bufferedAmount += size;
121     if (m_isClosingOrClosed)
122         m_bufferedAmountAfterClose += size;
123 
124     // FIXME: Deprecate this call.
125     m_client->didUpdateBufferedAmount(m_bufferedAmount);
126 
127     if (m_isClosingOrClosed)
128         return true;
129 
130     m_private->send(*PassRefPtr<ArrayBuffer>(webArrayBuffer), 0, webArrayBuffer.byteLength());
131     return true;
132 }
133 
bufferedAmount() const134 unsigned long WebSocketImpl::bufferedAmount() const
135 {
136     return m_bufferedAmount;
137 }
138 
close(int code,const WebString & reason)139 void WebSocketImpl::close(int code, const WebString& reason)
140 {
141     m_isClosingOrClosed = true;
142     m_private->close(code, reason);
143 }
144 
fail(const WebString & reason)145 void WebSocketImpl::fail(const WebString& reason)
146 {
147     m_private->fail(reason, ErrorMessageLevel, String(), 0);
148 }
149 
disconnect()150 void WebSocketImpl::disconnect()
151 {
152     m_private->disconnect();
153     m_client = 0;
154 }
155 
didConnect(const String & subprotocol,const String & extensions)156 void WebSocketImpl::didConnect(const String& subprotocol, const String& extensions)
157 {
158     m_client->didConnect(subprotocol, extensions);
159 
160     // FIXME: Deprecate these statements.
161     m_subprotocol = subprotocol;
162     m_extensions = extensions;
163     m_client->didConnect();
164 }
165 
didReceiveMessage(const String & message)166 void WebSocketImpl::didReceiveMessage(const String& message)
167 {
168     m_client->didReceiveMessage(WebString(message));
169 }
170 
didReceiveBinaryData(PassOwnPtr<Vector<char>> binaryData)171 void WebSocketImpl::didReceiveBinaryData(PassOwnPtr<Vector<char> > binaryData)
172 {
173     switch (m_binaryType) {
174     case BinaryTypeBlob:
175         // FIXME: Handle Blob after supporting WebBlob.
176         break;
177     case BinaryTypeArrayBuffer:
178         m_client->didReceiveArrayBuffer(WebArrayBuffer(ArrayBuffer::create(binaryData->data(), binaryData->size())));
179         break;
180     }
181 }
182 
didReceiveMessageError()183 void WebSocketImpl::didReceiveMessageError()
184 {
185     m_client->didReceiveMessageError();
186 }
187 
didConsumeBufferedAmount(unsigned long consumed)188 void WebSocketImpl::didConsumeBufferedAmount(unsigned long consumed)
189 {
190     m_client->didConsumeBufferedAmount(consumed);
191 
192     // FIXME: Deprecate the following statements.
193     m_bufferedAmount -= consumed;
194     m_client->didUpdateBufferedAmount(m_bufferedAmount);
195 }
196 
didStartClosingHandshake()197 void WebSocketImpl::didStartClosingHandshake()
198 {
199     m_client->didStartClosingHandshake();
200 }
201 
didClose(WebSocketChannelClient::ClosingHandshakeCompletionStatus status,unsigned short code,const String & reason)202 void WebSocketImpl::didClose(WebSocketChannelClient::ClosingHandshakeCompletionStatus status, unsigned short code, const String& reason)
203 {
204     m_isClosingOrClosed = true;
205     m_client->didClose(static_cast<WebSocketClient::ClosingHandshakeCompletionStatus>(status), code, WebString(reason));
206 
207     // FIXME: Deprecate this call.
208     m_client->didClose(m_bufferedAmount - m_bufferedAmountAfterClose, static_cast<WebSocketClient::ClosingHandshakeCompletionStatus>(status), code, WebString(reason));
209 }
210 
211 } // namespace blink
212