src/content/renderer/pepper/pepper_websocket_host.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/renderer/pepper/pepper_websocket_host.h"
   6
   7 #include <string>
   8
   9 #include "content/public/renderer/renderer_ppapi_host.h"
  10 #include "net/base/net_util.h"
  11 #include "ppapi/c/pp_errors.h"
  12 #include "ppapi/c/ppb_websocket.h"
  13 #include "ppapi/host/dispatch_host_message.h"
  14 #include "ppapi/host/host_message_context.h"
  15 #include "ppapi/host/ppapi_host.h"
  16 #include "ppapi/proxy/ppapi_messages.h"
  17 #include "third_party/WebKit/public/platform/WebArrayBuffer.h"
  18 #include "third_party/WebKit/public/platform/WebString.h"
  19 #include "third_party/WebKit/public/platform/WebURL.h"
  20 #include "third_party/WebKit/public/web/WebDocument.h"
  21 #include "third_party/WebKit/public/web/WebElement.h"
  22 #include "third_party/WebKit/public/web/WebPluginContainer.h"
  23 #include "third_party/WebKit/public/web/WebSocket.h"
  24
  25 using blink::WebArrayBuffer;
  26 using blink::WebDocument;
  27 using blink::WebString;
  28 using blink::WebSocket;
  29 using blink::WebURL;
  30
  31 namespace content {
  32
  33 #define COMPILE_ASSERT_MATCHING_ENUM(webkit_name, np_name)                   \
  34   COMPILE_ASSERT(                                                            \
  35       static_cast<int>(WebSocket::webkit_name) == static_cast<int>(np_name), \
  36       mismatching_enums)
  37
  38 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeNormalClosure,
  39                              PP_WEBSOCKETSTATUSCODE_NORMAL_CLOSURE);
  40 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeGoingAway,
  41                              PP_WEBSOCKETSTATUSCODE_GOING_AWAY);
  42 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeProtocolError,
  43                              PP_WEBSOCKETSTATUSCODE_PROTOCOL_ERROR);
  44 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeUnsupportedData,
  45                              PP_WEBSOCKETSTATUSCODE_UNSUPPORTED_DATA);
  46 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeNoStatusRcvd,
  47                              PP_WEBSOCKETSTATUSCODE_NO_STATUS_RECEIVED);
  48 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeAbnormalClosure,
  49                              PP_WEBSOCKETSTATUSCODE_ABNORMAL_CLOSURE);
  50 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeInvalidFramePayloadData,
  51                              PP_WEBSOCKETSTATUSCODE_INVALID_FRAME_PAYLOAD_DATA);
  52 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodePolicyViolation,
  53                              PP_WEBSOCKETSTATUSCODE_POLICY_VIOLATION);
  54 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeMessageTooBig,
  55                              PP_WEBSOCKETSTATUSCODE_MESSAGE_TOO_BIG);
  56 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeMandatoryExt,
  57                              PP_WEBSOCKETSTATUSCODE_MANDATORY_EXTENSION);
  58 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeInternalError,
  59                              PP_WEBSOCKETSTATUSCODE_INTERNAL_SERVER_ERROR);
  60 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeTLSHandshake,
  61                              PP_WEBSOCKETSTATUSCODE_TLS_HANDSHAKE);
  62 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeMinimumUserDefined,
  63                              PP_WEBSOCKETSTATUSCODE_USER_REGISTERED_MIN);
  64 COMPILE_ASSERT_MATCHING_ENUM(CloseEventCodeMaximumUserDefined,
  65                              PP_WEBSOCKETSTATUSCODE_USER_PRIVATE_MAX);
  66
  67 PepperWebSocketHost::PepperWebSocketHost(RendererPpapiHost* host,
  68                                          PP_Instance instance,
  69                                          PP_Resource resource)
  70     : ResourceHost(host->GetPpapiHost(), instance, resource),
  71       renderer_ppapi_host_(host),
  72       connecting_(false),
  73       initiating_close_(false),
  74       accepting_close_(false),
  75       error_was_received_(false) {}
  76
  77 PepperWebSocketHost::~PepperWebSocketHost() {
  78   if (websocket_)
  79     websocket_->disconnect();
  80 }
  81
  82 int32_t PepperWebSocketHost::OnResourceMessageReceived(
  83     const IPC::Message& msg,
  84     ppapi::host::HostMessageContext* context) {
  85   PPAPI_BEGIN_MESSAGE_MAP(PepperWebSocketHost, msg)
  86     PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_WebSocket_Connect,
  87                                       OnHostMsgConnect)
  88     PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_WebSocket_Close,
  89                                       OnHostMsgClose)
  90     PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_WebSocket_SendText,
  91                                       OnHostMsgSendText)
  92     PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_WebSocket_SendBinary,
  93                                       OnHostMsgSendBinary)
  94     PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_WebSocket_Fail,
  95                                       OnHostMsgFail)
  96   PPAPI_END_MESSAGE_MAP()
  97   return PP_ERROR_FAILED;
  98 }
  99
 100 void PepperWebSocketHost::didConnect() {
 101   std::string protocol;
 102   if (websocket_)
 103     protocol = websocket_->subprotocol().utf8();
 104   connecting_ = false;
 105   connect_reply_.params.set_result(PP_OK);
 106   host()->SendReply(connect_reply_,
 107                     PpapiPluginMsg_WebSocket_ConnectReply(url_, protocol));
 108 }
 109
 110 void PepperWebSocketHost::didReceiveMessage(const blink::WebString& message) {
 111   // Dispose packets after receiving an error.
 112   if (error_was_received_)
 113     return;
 114
 115   // Send an IPC to transport received data.
 116   std::string string_message = message.utf8();
 117   host()->SendUnsolicitedReply(
 118       pp_resource(), PpapiPluginMsg_WebSocket_ReceiveTextReply(string_message));
 119 }
 120
 121 void PepperWebSocketHost::didReceiveArrayBuffer(
 122     const blink::WebArrayBuffer& binaryData) {
 123   // Dispose packets after receiving an error.
 124   if (error_was_received_)
 125     return;
 126
 127   // Send an IPC to transport received data.
 128   uint8_t* data = static_cast<uint8_t*>(binaryData.data());
 129   std::vector<uint8_t> array_message(data, data + binaryData.byteLength());
 130   host()->SendUnsolicitedReply(
 131       pp_resource(),
 132       PpapiPluginMsg_WebSocket_ReceiveBinaryReply(array_message));
 133 }
 134
 135 void PepperWebSocketHost::didReceiveMessageError() {
 136   // Records the error, then stops receiving any frames after this error.
 137   // The error must be notified after all queued messages are read.
 138   error_was_received_ = true;
 139
 140   // Send an IPC to report the error. After this IPC, ReceiveTextReply and
 141   // ReceiveBinaryReply IPC are not sent anymore because |error_was_received_|
 142   // blocks.
 143   host()->SendUnsolicitedReply(pp_resource(),
 144                                PpapiPluginMsg_WebSocket_ErrorReply());
 145 }
 146
 147 void PepperWebSocketHost::didUpdateBufferedAmount(
 148     unsigned long buffered_amount) {
 149   // Send an IPC to update buffered amount.
 150   host()->SendUnsolicitedReply(
 151       pp_resource(),
 152       PpapiPluginMsg_WebSocket_BufferedAmountReply(buffered_amount));
 153 }
 154
 155 void PepperWebSocketHost::didStartClosingHandshake() {
 156   accepting_close_ = true;
 157
 158   // Send an IPC to notice that server starts closing handshake.
 159   host()->SendUnsolicitedReply(
 160       pp_resource(),
 161       PpapiPluginMsg_WebSocket_StateReply(PP_WEBSOCKETREADYSTATE_CLOSING));
 162 }
 163
 164 void PepperWebSocketHost::didClose(unsigned long unhandled_buffered_amount,
 165                                    ClosingHandshakeCompletionStatus status,
 166                                    unsigned short code,
 167                                    const blink::WebString& reason) {
 168   if (connecting_) {
 169     connecting_ = false;
 170     connect_reply_.params.set_result(PP_ERROR_FAILED);
 171     host()->SendReply(
 172         connect_reply_,
 173         PpapiPluginMsg_WebSocket_ConnectReply(url_, std::string()));
 174   }
 175
 176   // Set close_was_clean_.
 177   bool was_clean = (initiating_close_ || accepting_close_) &&
 178                    !unhandled_buffered_amount &&
 179                    status == WebSocketClient::ClosingHandshakeComplete;
 180
 181   if (initiating_close_) {
 182     initiating_close_ = false;
 183     close_reply_.params.set_result(PP_OK);
 184     host()->SendReply(
 185         close_reply_,
 186         PpapiPluginMsg_WebSocket_CloseReply(
 187             unhandled_buffered_amount, was_clean, code, reason.utf8()));
 188   } else {
 189     accepting_close_ = false;
 190     host()->SendUnsolicitedReply(
 191         pp_resource(),
 192         PpapiPluginMsg_WebSocket_ClosedReply(
 193             unhandled_buffered_amount, was_clean, code, reason.utf8()));
 194   }
 195
 196   // Disconnect.
 197   if (websocket_) {
 198     websocket_->disconnect();
 199     websocket_.reset();
 200   }
 201 }
 202
 203 int32_t PepperWebSocketHost::OnHostMsgConnect(
 204     ppapi::host::HostMessageContext* context,
 205     const std::string& url,
 206     const std::vector<std::string>& protocols) {
 207   // Validate url and convert it to WebURL.
 208   GURL gurl(url);
 209   url_ = gurl.spec();
 210   if (!gurl.is_valid())
 211     return PP_ERROR_BADARGUMENT;
 212   if (!gurl.SchemeIs("ws") && !gurl.SchemeIs("wss"))
 213     return PP_ERROR_BADARGUMENT;
 214   if (gurl.has_ref())
 215     return PP_ERROR_BADARGUMENT;
 216   if (!net::IsPortAllowedByDefault(gurl.IntPort()))
 217     return PP_ERROR_BADARGUMENT;
 218   WebURL web_url(gurl);
 219
 220   // Validate protocols.
 221   std::string protocol_string;
 222   for (std::vector<std::string>::const_iterator vector_it = protocols.begin();
 223        vector_it != protocols.end();
 224        ++vector_it) {
 225
 226     // Check containing characters.
 227     for (std::string::const_iterator string_it = vector_it->begin();
 228          string_it != vector_it->end();
 229          ++string_it) {
 230       uint8_t character = *string_it;
 231       // WebSocket specification says "(Subprotocol string must consist of)
 232       // characters in the range U+0021 to U+007E not including separator
 233       // characters as defined in [RFC2616]."
 234       const uint8_t minimumProtocolCharacter = '!';  // U+0021.
 235       const uint8_t maximumProtocolCharacter = '~';  // U+007E.
 236       if (character < minimumProtocolCharacter ||
 237           character > maximumProtocolCharacter || character == '"' ||
 238           character == '(' || character == ')' || character == ',' ||
 239           character == '/' ||
 240           (character >= ':' && character <= '@') ||  // U+003A - U+0040
 241           (character >= '[' && character <= ']') ||  // U+005B - u+005D
 242           character == '{' ||
 243           character == '}')
 244         return PP_ERROR_BADARGUMENT;
 245     }
 246     // Join protocols with the comma separator.
 247     if (vector_it != protocols.begin())
 248       protocol_string.append(",");
 249     protocol_string.append(*vector_it);
 250   }
 251
 252   // Convert protocols to WebString.
 253   WebString web_protocols = WebString::fromUTF8(protocol_string);
 254
 255   // Create blink::WebSocket object and connect.
 256   blink::WebPluginContainer* container =
 257       renderer_ppapi_host_->GetContainerForInstance(pp_instance());
 258   if (!container)
 259     return PP_ERROR_BADARGUMENT;
 260   // TODO(toyoshim) Remove following WebDocument object copy.
 261   WebDocument document = container->element().document();
 262   websocket_.reset(WebSocket::create(document, this));
 263   DCHECK(websocket_.get());
 264   if (!websocket_)
 265     return PP_ERROR_NOTSUPPORTED;
 266
 267   // Set receiving binary object type.
 268   websocket_->setBinaryType(WebSocket::BinaryTypeArrayBuffer);
 269   websocket_->connect(web_url, web_protocols);
 270
 271   connect_reply_ = context->MakeReplyMessageContext();
 272   connecting_ = true;
 273   return PP_OK_COMPLETIONPENDING;
 274 }
 275
 276 int32_t PepperWebSocketHost::OnHostMsgClose(
 277     ppapi::host::HostMessageContext* context,
 278     int32_t code,
 279     const std::string& reason) {
 280   if (!websocket_)
 281     return PP_ERROR_FAILED;
 282   close_reply_ = context->MakeReplyMessageContext();
 283   initiating_close_ = true;
 284
 285   blink::WebSocket::CloseEventCode event_code =
 286       static_cast<blink::WebSocket::CloseEventCode>(code);
 287   if (code == PP_WEBSOCKETSTATUSCODE_NOT_SPECIFIED) {
 288     // PP_WEBSOCKETSTATUSCODE_NOT_SPECIFIED and CloseEventCodeNotSpecified are
 289     // assigned to different values. A conversion is needed if
 290     // PP_WEBSOCKETSTATUSCODE_NOT_SPECIFIED is specified.
 291     event_code = blink::WebSocket::CloseEventCodeNotSpecified;
 292   }
 293
 294   WebString web_reason = WebString::fromUTF8(reason);
 295   websocket_->close(event_code, web_reason);
 296   return PP_OK_COMPLETIONPENDING;
 297 }
 298
 299 int32_t PepperWebSocketHost::OnHostMsgSendText(
 300     ppapi::host::HostMessageContext* context,
 301     const std::string& message) {
 302   if (websocket_) {
 303     WebString web_message = WebString::fromUTF8(message);
 304     websocket_->sendText(web_message);
 305   }
 306   return PP_OK;
 307 }
 308
 309 int32_t PepperWebSocketHost::OnHostMsgSendBinary(
 310     ppapi::host::HostMessageContext* context,
 311     const std::vector<uint8_t>& message) {
 312   if (websocket_.get() && !message.empty()) {
 313     WebArrayBuffer web_message = WebArrayBuffer::create(message.size(), 1);
 314     memcpy(web_message.data(), &message.front(), message.size());
 315     websocket_->sendArrayBuffer(web_message);
 316   }
 317   return PP_OK;
 318 }
 319
 320 int32_t PepperWebSocketHost::OnHostMsgFail(
 321     ppapi::host::HostMessageContext* context,
 322     const std::string& message) {
 323   if (websocket_)
 324     websocket_->fail(WebString::fromUTF8(message));
 325   return PP_OK;
 326 }
 327
 328 }  // namespace content