| 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 24 matching lines...) Expand all Loading... |
| 35 #include "core/dom/Document.h" | 35 #include "core/dom/Document.h" |
| 36 #include "core/frame/ConsoleTypes.h" | 36 #include "core/frame/ConsoleTypes.h" |
| 37 #include "modules/websockets/MainThreadWebSocketChannel.h" | 37 #include "modules/websockets/MainThreadWebSocketChannel.h" |
| 38 #include "modules/websockets/NewWebSocketChannelImpl.h" | 38 #include "modules/websockets/NewWebSocketChannelImpl.h" |
| 39 #include "modules/websockets/WebSocketChannel.h" | 39 #include "modules/websockets/WebSocketChannel.h" |
| 40 #include "public/platform/WebArrayBuffer.h" | 40 #include "public/platform/WebArrayBuffer.h" |
| 41 #include "public/platform/WebString.h" | 41 #include "public/platform/WebString.h" |
| 42 #include "public/platform/WebURL.h" | 42 #include "public/platform/WebURL.h" |
| 43 #include "public/web/WebDocument.h" | 43 #include "public/web/WebDocument.h" |
| 44 #include "wtf/ArrayBuffer.h" | 44 #include "wtf/ArrayBuffer.h" |
| 45 #include "wtf/text/CString.h" | |
| 46 #include "wtf/text/WTFString.h" | 45 #include "wtf/text/WTFString.h" |
| 47 | 46 |
| 48 using namespace WebCore; | 47 using namespace WebCore; |
| 49 | 48 |
| 50 namespace blink { | 49 namespace blink { |
| 51 | 50 |
| 52 WebSocketImpl::WebSocketImpl(const WebDocument& document, WebSocketClient* clien
t) | 51 WebSocketImpl::WebSocketImpl(const WebDocument& document, WebSocketClient* clien
t) |
| 53 : m_client(client) | 52 : m_client(client) |
| 54 , m_binaryType(BinaryTypeBlob) | 53 , m_binaryType(BinaryTypeBlob) |
| 55 , m_bufferedAmount(0) | |
| 56 { | 54 { |
| 57 RefPtrWillBeRawPtr<Document> coreDocument = PassRefPtrWillBeRawPtr<Document>
(document); | 55 RefPtrWillBeRawPtr<Document> coreDocument = PassRefPtrWillBeRawPtr<Document>
(document); |
| 58 if (RuntimeEnabledFeatures::experimentalWebSocketEnabled()) { | 56 if (RuntimeEnabledFeatures::experimentalWebSocketEnabled()) { |
| 59 m_private = NewWebSocketChannelImpl::create(coreDocument.get(), this); | 57 m_private = NewWebSocketChannelImpl::create(coreDocument.get(), this); |
| 60 } else { | 58 } else { |
| 61 m_private = MainThreadWebSocketChannel::create(coreDocument.get(), this)
; | 59 m_private = MainThreadWebSocketChannel::create(coreDocument.get(), this)
; |
| 62 } | 60 } |
| 63 } | 61 } |
| 64 | 62 |
| 65 WebSocketImpl::~WebSocketImpl() | 63 WebSocketImpl::~WebSocketImpl() |
| (...skipping 24 matching lines...) Expand all Loading... |
| 90 return m_subprotocol; | 88 return m_subprotocol; |
| 91 } | 89 } |
| 92 | 90 |
| 93 WebString WebSocketImpl::extensions() | 91 WebString WebSocketImpl::extensions() |
| 94 { | 92 { |
| 95 return m_extensions; | 93 return m_extensions; |
| 96 } | 94 } |
| 97 | 95 |
| 98 bool WebSocketImpl::sendText(const WebString& message) | 96 bool WebSocketImpl::sendText(const WebString& message) |
| 99 { | 97 { |
| 100 m_bufferedAmount += message.utf8().length(); | |
| 101 return m_private->send(message) == WebSocketChannel::SendSuccess; | 98 return m_private->send(message) == WebSocketChannel::SendSuccess; |
| 102 } | 99 } |
| 103 | 100 |
| 104 bool WebSocketImpl::sendArrayBuffer(const WebArrayBuffer& webArrayBuffer) | 101 bool WebSocketImpl::sendArrayBuffer(const WebArrayBuffer& webArrayBuffer) |
| 105 { | 102 { |
| 106 m_bufferedAmount += webArrayBuffer.byteLength(); | |
| 107 return m_private->send(*PassRefPtr<ArrayBuffer>(webArrayBuffer), 0, webArray
Buffer.byteLength()) == WebSocketChannel::SendSuccess; | 103 return m_private->send(*PassRefPtr<ArrayBuffer>(webArrayBuffer), 0, webArray
Buffer.byteLength()) == WebSocketChannel::SendSuccess; |
| 108 } | 104 } |
| 109 | 105 |
| 110 unsigned long WebSocketImpl::bufferedAmount() const | 106 unsigned long WebSocketImpl::bufferedAmount() const |
| 111 { | 107 { |
| 112 return m_bufferedAmount; | 108 return m_private->bufferedAmount(); |
| 113 } | 109 } |
| 114 | 110 |
| 115 void WebSocketImpl::close(int code, const WebString& reason) | 111 void WebSocketImpl::close(int code, const WebString& reason) |
| 116 { | 112 { |
| 117 m_private->close(code, reason); | 113 m_private->close(code, reason); |
| 118 } | 114 } |
| 119 | 115 |
| 120 void WebSocketImpl::fail(const WebString& reason) | 116 void WebSocketImpl::fail(const WebString& reason) |
| 121 { | 117 { |
| 122 m_private->fail(reason, ErrorMessageLevel, String(), 0); | 118 m_private->fail(reason, ErrorMessageLevel, String(), 0); |
| (...skipping 27 matching lines...) Expand all Loading... |
| 150 m_client->didReceiveArrayBuffer(WebArrayBuffer(ArrayBuffer::create(binar
yData->data(), binaryData->size()))); | 146 m_client->didReceiveArrayBuffer(WebArrayBuffer(ArrayBuffer::create(binar
yData->data(), binaryData->size()))); |
| 151 break; | 147 break; |
| 152 } | 148 } |
| 153 } | 149 } |
| 154 | 150 |
| 155 void WebSocketImpl::didReceiveMessageError() | 151 void WebSocketImpl::didReceiveMessageError() |
| 156 { | 152 { |
| 157 m_client->didReceiveMessageError(); | 153 m_client->didReceiveMessageError(); |
| 158 } | 154 } |
| 159 | 155 |
| 160 void WebSocketImpl::didConsumeBufferedAmount(unsigned long consumed) | 156 void WebSocketImpl::didUpdateBufferedAmount(unsigned long bufferedAmount) |
| 161 { | 157 { |
| 162 m_bufferedAmount -= consumed; | 158 m_client->didUpdateBufferedAmount(bufferedAmount); |
| 163 } | 159 } |
| 164 | 160 |
| 165 void WebSocketImpl::didStartClosingHandshake() | 161 void WebSocketImpl::didStartClosingHandshake() |
| 166 { | 162 { |
| 167 m_client->didStartClosingHandshake(); | 163 m_client->didStartClosingHandshake(); |
| 168 } | 164 } |
| 169 | 165 |
| 170 void WebSocketImpl::didClose(ClosingHandshakeCompletionStatus status, unsigned s
hort code, const String& reason) | 166 void WebSocketImpl::didClose(unsigned long bufferedAmount, ClosingHandshakeCompl
etionStatus status, unsigned short code, const String& reason) |
| 171 { | 167 { |
| 172 m_client->didClose(m_bufferedAmount, static_cast<WebSocketClient::ClosingHan
dshakeCompletionStatus>(status), code, WebString(reason)); | 168 m_client->didClose(bufferedAmount, static_cast<WebSocketClient::ClosingHands
hakeCompletionStatus>(status), code, WebString(reason)); |
| 173 } | 169 } |
| 174 | 170 |
| 175 } // namespace blink | 171 } // namespace blink |
| OLD | NEW |