Exposed Listen and Accept methods to plugin.

content/renderer/pepper_plugin_delegate_impl.cc

 // Copyright (c) 2012 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/renderer/pepper_plugin_delegate_impl.h"
 
 #include <cmath>
 #include <queue>
 
 #include "base/bind.h"
@@ skipping 60 matching lines @@
 #include "webkit/fileapi/file_system_callback_dispatcher.h"
 #include "webkit/glue/context_menu.h"
 #include "webkit/plugins/npapi/webplugin.h"
 #include "webkit/plugins/ppapi/file_path.h"
 #include "webkit/plugins/ppapi/ppb_file_io_impl.h"
 #include "webkit/plugins/ppapi/plugin_module.h"
 #include "webkit/plugins/ppapi/ppapi_plugin_instance.h"
 #include "webkit/plugins/ppapi/ppb_broker_impl.h"
 #include "webkit/plugins/ppapi/ppb_flash_impl.h"
 #include "webkit/plugins/ppapi/ppb_flash_net_connector_impl.h"
+#include "webkit/plugins/ppapi/ppb_tcp_server_socket_private_impl.h"
 #include "webkit/plugins/ppapi/ppb_tcp_socket_private_impl.h"
 #include "webkit/plugins/ppapi/ppb_udp_socket_private_impl.h"
 #include "webkit/plugins/ppapi/resource_helper.h"
 #include "webkit/plugins/webplugininfo.h"
 
 using WebKit::WebView;
 using WebKit::WebFrame;
 
 namespace {
 
@@ skipping 1611 matching lines @@
   render_view_->Send(new PpapiHostMsg_PPBTCPSocket_Create(
       render_view_->routing_id(), 0, &socket_id));
   return socket_id;
 }
 
 void PepperPluginDelegateImpl::TCPSocketConnect(
     webkit::ppapi::PPB_TCPSocket_Private_Impl* socket,
     uint32 socket_id,
     const std::string& host,
     uint16_t port) {
-  tcp_sockets_.AddWithID(socket, socket_id);
+  TCPSocketRegister(socket, socket_id);
   render_view_->Send(
       new PpapiHostMsg_PPBTCPSocket_Connect(socket_id, host, port));
 }
 
 void PepperPluginDelegateImpl::TCPSocketConnectWithNetAddress(
       webkit::ppapi::PPB_TCPSocket_Private_Impl* socket,
       uint32 socket_id,
       const PP_NetAddress_Private& addr) {
   tcp_sockets_.AddWithID(socket, socket_id);
   render_view_->Send(
@@ skipping 23 matching lines @@
 }
 
 void PepperPluginDelegateImpl::TCPSocketDisconnect(uint32 socket_id) {
   // There are no DCHECK(tcp_sockets_.Lookup(socket_id)) because it
   // can be called before
   // TCPSocketConnect/TCPSocketConnectWithNetAddress is called.
   render_view_->Send(new PpapiHostMsg_PPBTCPSocket_Disconnect(socket_id));
   tcp_sockets_.Remove(socket_id);
 }
 
+void PepperPluginDelegateImpl::TCPSocketRegister(
+    webkit::ppapi::PPB_TCPSocket_Private_Impl* socket,
+    uint32 socket_id) {
+  tcp_sockets_.AddWithID(socket, socket_id);
+}
+
 uint32 PepperPluginDelegateImpl::UDPSocketCreate() {
   if (!CanUseSocketAPIs())
     return 0;
 
   uint32 socket_id = 0;
   render_view_->Send(new PpapiHostMsg_PPBUDPSocket_Create(
       render_view_->routing_id(), 0, &socket_id));
   return socket_id;
 }
 
@@ skipping 21 matching lines @@
       new PpapiHostMsg_PPBUDPSocket_SendTo(socket_id, buffer, net_addr));
 }
 
 void PepperPluginDelegateImpl::UDPSocketClose(uint32 socket_id) {
   // There are no DCHECK(udp_sockets_.Lookup(socket_id)) because it
   // can be called before UDPSocketBind is called.
   render_view_->Send(new PpapiHostMsg_PPBUDPSocket_Close(socket_id));
   udp_sockets_.Remove(socket_id);
 }
 
+uint32 PepperPluginDelegateImpl::TCPServerSocketCreate() {
+  if (!CanUseSocketAPIs())
+    return 0;
+
+  uint32 socket_id = 0;
+  render_view_->Send(new PpapiHostMsg_PPBTCPServerSocket_Create(

[brettw, 2012/02/07 00:15:22]

Supporting these in the renderer is very problematic. Justin will follow up, we
also need to figure out what to do about the existing socket sending stuff.

[Dmitry Polukhin, 2012/02/07 06:45:50]

I think we need to transform "create" methods to async methods with callback. It
will also simplify check what pages/renders can use socket APIs. Now this check
happens in render but it needs to be moved to browser process. Inside browser
process it is easy to do on UI thread but create messages are processed to IO
thread. With async methods it will be easy to send message from IO to UI thread.

[ygorshenin1, 2012/02/07 13:42:59]

Fixed: PPB_TCPServerSocke_Private::Create method is asynchronous now.

On 2012/02/07 06:45:50, Dmitry Polukhin wrote:

+      render_view_->routing_id(), 0, &socket_id));
+  return socket_id;
+}
+
+void PepperPluginDelegateImpl::TCPServerSocketListen(
+    webkit::ppapi::PPB_TCPServerSocket_Private_Impl* socket,
+    uint32 socket_id,
+    const PP_NetAddress_Private& addr,
+    int32_t backlog) {
+  tcp_server_sockets_.AddWithID(socket, socket_id);
+  render_view_->Send(
+      new PpapiHostMsg_PPBTCPServerSocket_Listen(socket_id, addr, backlog));
+}
+
+void PepperPluginDelegateImpl::TCPServerSocketAccept(uint32 socket_id) {
+  DCHECK(tcp_server_sockets_.Lookup(socket_id));
+  render_view_->Send(new PpapiHostMsg_PPBTCPServerSocket_Accept(socket_id));
+}
+
+void PepperPluginDelegateImpl::TCPServerSocketStopListening(uint32 socket_id) {
+  // There are no DCHECK(tcp_server_sockets_.Lookup(socket_id))
+  // because it can be called before TCPServerSocketListen is called.
+  render_view_->Send(
+      new PpapiHostMsg_PPBTCPServerSocket_StopListening(socket_id));
+  tcp_server_sockets_.Remove(socket_id);
+}
+
 int32_t PepperPluginDelegateImpl::ShowContextMenu(
     webkit::ppapi::PluginInstance* instance,
     webkit::ppapi::PPB_Flash_Menu_Impl* menu,
     const gfx::Point& position) {
   int32 render_widget_id = render_view_->routing_id();
   if (instance->FlashIsFullscreen(instance->pp_instance())) {
     webkit::ppapi::FullscreenContainer* container =
         instance->fullscreen_container();
     DCHECK(container);
     render_widget_id =
@@ skipping 243 matching lines @@
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ConnectACK,
                         OnTCPSocketConnectACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_SSLHandshakeACK,
                         OnTCPSocketSSLHandshakeACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ReadACK, OnTCPSocketReadACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_WriteACK, OnTCPSocketWriteACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBUDPSocket_BindACK, OnUDPSocketBindACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBUDPSocket_RecvFromACK,
                         OnUDPSocketRecvFromACK)
     IPC_MESSAGE_HANDLER(PpapiMsg_PPBUDPSocket_SendToACK, OnUDPSocketSendToACK)
+    IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPServerSocket_ListenACK,
+                        OnTCPServerSocketListenACK)
+    IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPServerSocket_AcceptACK,
+                        OnTCPServerSocketAcceptACK)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void PepperPluginDelegateImpl::OnDestruct() {
   // Nothing to do here. Default implementation in RenderViewObserver does
   // 'delete this' but it's not suitable for PepperPluginDelegateImpl because
   // it's non-pointer member in RenderViewImpl.
 }
 
 void PepperPluginDelegateImpl::OnTCPSocketConnectACK(
     uint32 plugin_dispatcher_id,
     uint32 socket_id,
     bool succeeded,
     const PP_NetAddress_Private& local_addr,
     const PP_NetAddress_Private& remote_addr) {
   webkit::ppapi::PPB_TCPSocket_Private_Impl* socket =
       tcp_sockets_.Lookup(socket_id);
   if (socket)
     socket->OnConnectCompleted(succeeded, local_addr, remote_addr);
+  if (!succeeded)
+    tcp_sockets_.Remove(socket_id);
 }
 
 void PepperPluginDelegateImpl::OnTCPSocketSSLHandshakeACK(
     uint32 plugin_dispatcher_id,
     uint32 socket_id,
     bool succeeded) {
   webkit::ppapi::PPB_TCPSocket_Private_Impl* socket =
       tcp_sockets_.Lookup(socket_id);
   if (socket)
     socket->OnSSLHandshakeCompleted(succeeded);
@@ skipping 19 matching lines @@
     socket->OnWriteCompleted(succeeded, bytes_written);
 }
 
 void PepperPluginDelegateImpl::OnUDPSocketBindACK(uint32 plugin_dispatcher_id,
                                                   uint32 socket_id,
                                                   bool succeeded) {
   webkit::ppapi::PPB_UDPSocket_Private_Impl* socket =
       udp_sockets_.Lookup(socket_id);
   if (socket)
     socket->OnBindCompleted(succeeded);
+  if (!succeeded)
+    udp_sockets_.Remove(socket_id);
 }
 
 void PepperPluginDelegateImpl::OnUDPSocketRecvFromACK(
     uint32 plugin_dispatcher_id,
     uint32 socket_id,
     bool succeeded,
     const std::string& data,
     const PP_NetAddress_Private& remote_addr) {
   webkit::ppapi::PPB_UDPSocket_Private_Impl* socket =
       udp_sockets_.Lookup(socket_id);
   if (socket)
     socket->OnRecvFromCompleted(succeeded, data, remote_addr);
 }
 
 void PepperPluginDelegateImpl::OnUDPSocketSendToACK(uint32 plugin_dispatcher_id,
                                                     uint32 socket_id,
                                                     bool succeeded,
                                                     int32_t bytes_written) {
   webkit::ppapi::PPB_UDPSocket_Private_Impl* socket =
       udp_sockets_.Lookup(socket_id);
   if (socket)
     socket->OnSendToCompleted(succeeded, bytes_written);
 }
 
+void PepperPluginDelegateImpl::OnTCPServerSocketListenACK(
+    uint32 plugin_dispatcher_id,
+    uint32 socket_id,
+    bool succeeded) {
+  webkit::ppapi::PPB_TCPServerSocket_Private_Impl* socket =
+      tcp_server_sockets_.Lookup(socket_id);
+  if (socket)
+    socket->OnListenCompleted(succeeded);
+  if (!succeeded)
+    tcp_server_sockets_.Remove(socket_id);
+}
+
+void PepperPluginDelegateImpl::OnTCPServerSocketAcceptACK(
+    uint32 plugin_dispatcher_id,
+    uint32 tcp_server_socket_id,
+    uint32 tcp_socket_id,
+    const PP_NetAddress_Private& local_addr,
+    const PP_NetAddress_Private& remote_addr) {
+  webkit::ppapi::PPB_TCPServerSocket_Private_Impl* socket =
+      tcp_server_sockets_.Lookup(tcp_server_socket_id);
+  if (socket) {
+    bool succeeded = tcp_socket_id != 0;
+    socket->OnAcceptCompleted(succeeded,
+                              tcp_socket_id,
+                              local_addr,
+                              remote_addr);
+  }
+}
+
 int PepperPluginDelegateImpl::GetRoutingId() const {
   return render_view_->routing_id();
 }
 
 RendererGLContext*
 PepperPluginDelegateImpl::GetParentContextForPlatformContext3D() {
   WebGraphicsContext3DCommandBufferImpl* context =
       static_cast<WebGraphicsContext3DCommandBufferImpl*>(
           render_view_->webview()->graphicsContext3D());
   if (!context)
     return NULL;
   if (!context->makeContextCurrent() || context->isContextLost())
     return NULL;
 
   RendererGLContext* parent_context = context->context();
   if (!parent_context)
     return NULL;
   return parent_context;
 }
 
 bool PepperPluginDelegateImpl::CanUseSocketAPIs() {
   WebView* webview = render_view_->webview();
   WebFrame* main_frame = webview ? webview->mainFrame() : NULL;
   GURL url(main_frame ? GURL(main_frame->document().url()) : GURL());
   if (!url.is_valid())
     return false;
 
   return content::GetContentClient()->renderer()->AllowSocketAPI(url);
 }