blob: 5f0e37d0cc0a2c8d5630297b4a105c39168e1cde [file] [log] [blame]
// 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 <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/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_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::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