| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "content/child/websocket_bridge.h" |
| |
| #include <stdint.h> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/logging.h" |
| #include "base/strings/string_util.h" |
| #include "content/child/child_thread.h" |
| #include "content/child/websocket_dispatcher.h" |
| #include "content/common/websocket.h" |
| #include "content/common/websocket_messages.h" |
| #include "ipc/ipc_message.h" |
| #include "ipc/ipc_message_macros.h" |
| #include "url/gurl.h" |
| #include "third_party/WebKit/public/platform/WebSocketHandle.h" |
| #include "third_party/WebKit/public/platform/WebSocketHandleClient.h" |
| #include "third_party/WebKit/public/platform/WebSocketHandshakeRequestInfo.h" |
| #include "third_party/WebKit/public/platform/WebSocketHandshakeResponseInfo.h" |
| #include "third_party/WebKit/public/platform/WebString.h" |
| #include "third_party/WebKit/public/platform/WebURL.h" |
| #include "third_party/WebKit/public/platform/WebVector.h" |
| |
| using blink::WebSocketHandle; |
| using blink::WebSocketHandleClient; |
| using blink::WebString; |
| using blink::WebURL; |
| using blink::WebVector; |
| |
| namespace content { |
| |
| namespace { |
| |
| const unsigned short kAbnormalShutdownOpCode = 1006; |
| |
| } // namespace |
| |
| WebSocketBridge::WebSocketBridge() |
| : channel_id_(kInvalidChannelId), client_(NULL) {} |
| |
| WebSocketBridge::~WebSocketBridge() { |
| if (channel_id_ != kInvalidChannelId) { |
| // The connection is abruptly disconnected by the renderer without |
| // closing handshake. |
| ChildThread::current()->Send( |
| new WebSocketMsg_DropChannel(channel_id_, |
| false, |
| kAbnormalShutdownOpCode, |
| std::string())); |
| } |
| Disconnect(); |
| } |
| |
| bool WebSocketBridge::OnMessageReceived(const IPC::Message& msg) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(WebSocketBridge, msg) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_AddChannelResponse, DidConnect) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_NotifyStartOpeningHandshake, |
| DidStartOpeningHandshake) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_NotifyFinishOpeningHandshake, |
| DidFinishOpeningHandshake) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_NotifyFailure, DidFail) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_SendFrame, DidReceiveData) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_FlowControl, DidReceiveFlowControl) |
| IPC_MESSAGE_HANDLER(WebSocketMsg_DropChannel, DidClose) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| void WebSocketBridge::DidConnect(bool fail, |
| const std::string& selected_protocol, |
| const std::string& extensions) { |
| WebSocketHandleClient* client = client_; |
| DVLOG(1) << "WebSocketBridge::DidConnect(" |
| << fail << ", " |
| << selected_protocol << ", " |
| << extensions << ")"; |
| if (fail) |
| Disconnect(); |
| if (!client) |
| return; |
| |
| WebString protocol_to_pass = WebString::fromUTF8(selected_protocol); |
| WebString extensions_to_pass = WebString::fromUTF8(extensions); |
| client->didConnect(this, fail, protocol_to_pass, extensions_to_pass); |
| // |this| can be deleted here. |
| } |
| |
| void WebSocketBridge::DidStartOpeningHandshake( |
| const WebSocketHandshakeRequest& request) { |
| DVLOG(1) << "WebSocketBridge::DidStartOpeningHandshake(" |
| << request.url << ")"; |
| // All strings are already encoded to ASCII in the browser. |
| blink::WebSocketHandshakeRequestInfo request_to_pass; |
| request_to_pass.setURL(WebURL(request.url)); |
| for (size_t i = 0; i < request.headers.size(); ++i) { |
| const std::pair<std::string, std::string>& header = request.headers[i]; |
| request_to_pass.addHeaderField(WebString::fromLatin1(header.first), |
| WebString::fromLatin1(header.second)); |
| } |
| client_->didStartOpeningHandshake(this, request_to_pass); |
| } |
| |
| void WebSocketBridge::DidFinishOpeningHandshake( |
| const WebSocketHandshakeResponse& response) { |
| DVLOG(1) << "WebSocketBridge::DidFinishOpeningHandshake(" |
| << response.url << ")"; |
| // All strings are already encoded to ASCII in the browser. |
| blink::WebSocketHandshakeResponseInfo response_to_pass; |
| response_to_pass.setStatusCode(response.status_code); |
| response_to_pass.setStatusText(WebString::fromLatin1(response.status_text)); |
| for (size_t i = 0; i < response.headers.size(); ++i) { |
| const std::pair<std::string, std::string>& header = response.headers[i]; |
| response_to_pass.addHeaderField(WebString::fromLatin1(header.first), |
| WebString::fromLatin1(header.second)); |
| } |
| client_->didFinishOpeningHandshake(this, response_to_pass); |
| } |
| |
| void WebSocketBridge::DidFail(const std::string& message) { |
| DVLOG(1) << "WebSocketBridge::DidFail(" << message << ")"; |
| WebSocketHandleClient* client = client_; |
| Disconnect(); |
| if (!client) |
| return; |
| |
| WebString message_to_pass = WebString::fromUTF8(message); |
| client->didFail(this, message_to_pass); |
| // |this| can be deleted here. |
| } |
| |
| void WebSocketBridge::DidReceiveData(bool fin, |
| WebSocketMessageType type, |
| const std::vector<char>& data) { |
| DVLOG(1) << "WebSocketBridge::DidReceiveData(" |
| << fin << ", " |
| << type << ", " |
| << "(data size = " << data.size() << "))"; |
| if (!client_) |
| return; |
| |
| WebSocketHandle::MessageType type_to_pass = |
| WebSocketHandle::MessageTypeContinuation; |
| switch (type) { |
| case WEB_SOCKET_MESSAGE_TYPE_CONTINUATION: |
| type_to_pass = WebSocketHandle::MessageTypeContinuation; |
| break; |
| case WEB_SOCKET_MESSAGE_TYPE_TEXT: |
| type_to_pass = WebSocketHandle::MessageTypeText; |
| break; |
| case WEB_SOCKET_MESSAGE_TYPE_BINARY: |
| type_to_pass = WebSocketHandle::MessageTypeBinary; |
| break; |
| } |
| const char* data_to_pass = data.empty() ? NULL : &data[0]; |
| client_->didReceiveData(this, fin, type_to_pass, data_to_pass, data.size()); |
| // |this| can be deleted here. |
| } |
| |
| void WebSocketBridge::DidReceiveFlowControl(int64_t quota) { |
| DVLOG(1) << "WebSocketBridge::DidReceiveFlowControl(" << quota << ")"; |
| if (!client_) |
| return; |
| |
| client_->didReceiveFlowControl(this, quota); |
| // |this| can be deleted here. |
| } |
| |
| void WebSocketBridge::DidClose(bool was_clean, |
| unsigned short code, |
| const std::string& reason) { |
| DVLOG(1) << "WebSocketBridge::DidClose(" |
| << was_clean << ", " |
| << code << ", " |
| << reason << ")"; |
| WebSocketHandleClient* client = client_; |
| Disconnect(); |
| if (!client) |
| return; |
| |
| WebString reason_to_pass = WebString::fromUTF8(reason); |
| client->didClose(this, was_clean, code, reason_to_pass); |
| // |this| can be deleted here. |
| } |
| |
| void WebSocketBridge::connect( |
| const WebURL& url, |
| const WebVector<WebString>& protocols, |
| const WebString& origin, |
| WebSocketHandleClient* client) { |
| DCHECK_EQ(kInvalidChannelId, channel_id_); |
| WebSocketDispatcher* dispatcher = |
| ChildThread::current()->websocket_dispatcher(); |
| channel_id_ = dispatcher->AddBridge(this); |
| client_ = client; |
| |
| std::vector<std::string> protocols_to_pass; |
| for (size_t i = 0; i < protocols.size(); ++i) |
| protocols_to_pass.push_back(protocols[i].utf8()); |
| GURL origin_to_pass(origin.utf8()); |
| |
| DVLOG(1) << "Bridge#" << channel_id_ << " Connect(" |
| << url << ", (" << JoinString(protocols_to_pass, ", ") << "), " |
| << origin_to_pass << ")"; |
| |
| ChildThread::current()->Send( |
| new WebSocketHostMsg_AddChannelRequest(channel_id_, |
| url, |
| protocols_to_pass, |
| origin_to_pass)); |
| } |
| |
| void WebSocketBridge::send(bool fin, |
| WebSocketHandle::MessageType type, |
| const char* data, |
| size_t size) { |
| if (channel_id_ == kInvalidChannelId) |
| return; |
| |
| WebSocketMessageType type_to_pass = WEB_SOCKET_MESSAGE_TYPE_CONTINUATION; |
| switch (type) { |
| case WebSocketHandle::MessageTypeContinuation: |
| type_to_pass = WEB_SOCKET_MESSAGE_TYPE_CONTINUATION; |
| break; |
| case WebSocketHandle::MessageTypeText: |
| type_to_pass = WEB_SOCKET_MESSAGE_TYPE_TEXT; |
| break; |
| case WebSocketHandle::MessageTypeBinary: |
| type_to_pass = WEB_SOCKET_MESSAGE_TYPE_BINARY; |
| break; |
| } |
| |
| DVLOG(1) << "Bridge #" << channel_id_ << " Send(" |
| << fin << ", " << type_to_pass << ", " |
| << "(data size = " << size << "))"; |
| |
| ChildThread::current()->Send( |
| new WebSocketMsg_SendFrame(channel_id_, |
| fin, |
| type_to_pass, |
| std::vector<char>(data, data + size))); |
| } |
| |
| void WebSocketBridge::flowControl(int64_t quota) { |
| if (channel_id_ == kInvalidChannelId) |
| return; |
| |
| DVLOG(1) << "Bridge #" << channel_id_ << " FlowControl(" << quota << ")"; |
| |
| ChildThread::current()->Send( |
| new WebSocketMsg_FlowControl(channel_id_, quota)); |
| } |
| |
| void WebSocketBridge::close(unsigned short code, |
| const WebString& reason) { |
| if (channel_id_ == kInvalidChannelId) |
| return; |
| |
| std::string reason_to_pass = reason.utf8(); |
| DVLOG(1) << "Bridge #" << channel_id_ << " Close(" |
| << code << ", " << reason_to_pass << ")"; |
| // This method is for closing handshake and hence |was_clean| shall be true. |
| ChildThread::current()->Send( |
| new WebSocketMsg_DropChannel(channel_id_, true, code, reason_to_pass)); |
| } |
| |
| void WebSocketBridge::Disconnect() { |
| if (channel_id_ == kInvalidChannelId) |
| return; |
| WebSocketDispatcher* dispatcher = |
| ChildThread::current()->websocket_dispatcher(); |
| dispatcher->RemoveBridge(channel_id_); |
| |
| channel_id_ = kInvalidChannelId; |
| client_ = NULL; |
| } |
| |
| } // namespace content |