ppapi/proxy/ppb_tcp_socket_proxy.cc

   1 // Copyright 2013 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 "ppapi/proxy/ppb_tcp_socket_proxy.h"
   6
   7 #include <map>
   8
   9 #include "base/logging.h"
  10 #include "ppapi/proxy/plugin_dispatcher.h"
  11 #include "ppapi/proxy/plugin_globals.h"
  12 #include "ppapi/proxy/ppapi_messages.h"
  13 #include "ppapi/shared_impl/resource.h"
  14 #include "ppapi/shared_impl/socket_option_data.h"
  15 #include "ppapi/shared_impl/tcp_socket_shared.h"
  16 #include "ppapi/thunk/enter.h"
  17 #include "ppapi/thunk/ppb_net_address_api.h"
  18 #include "ppapi/thunk/ppb_tcp_socket_api.h"
  19 #include "ppapi/thunk/thunk.h"
  20
  21 namespace ppapi {
  22 namespace proxy {
  23
  24 namespace {
  25
  26 typedef thunk::EnterResourceNoLock<thunk::PPB_NetAddress_API>
  27     EnterNetAddressNoLock;
  28
  29 typedef std::map<uint32, TCPSocketShared*> IDToSocketMap;
  30 IDToSocketMap* g_id_to_socket = NULL;
  31
  32 class TCPSocket : public thunk::PPB_TCPSocket_API,
  33                   public Resource,
  34                   public TCPSocketShared {
  35  public:
  36   TCPSocket(const HostResource& resource, uint32 socket_id);
  37   virtual ~TCPSocket();
  38
  39   // Resource overrides.
  40   virtual thunk::PPB_TCPSocket_API* AsPPB_TCPSocket_API() OVERRIDE;
  41
  42   // thunk::PPB_TCPSocket_API implementation.
  43   virtual int32_t Connect(PP_Resource addr,
  44                           scoped_refptr<TrackedCallback> callback) OVERRIDE;
  45   virtual PP_Resource GetLocalAddress() OVERRIDE;
  46   virtual PP_Resource GetRemoteAddress() OVERRIDE;
  47   virtual int32_t Read(char* buffer,
  48                        int32_t bytes_to_read,
  49                        scoped_refptr<TrackedCallback> callback) OVERRIDE;
  50   virtual int32_t Write(const char* buffer,
  51                         int32_t bytes_to_write,
  52                         scoped_refptr<TrackedCallback> callback) OVERRIDE;
  53   virtual void Close() OVERRIDE;
  54   virtual int32_t SetOption(PP_TCPSocket_Option name,
  55                             const PP_Var& value,
  56                             scoped_refptr<TrackedCallback> callback) OVERRIDE;
  57
  58   // TCPSocketShared implementation.
  59   virtual void SendConnect(const std::string& host, uint16_t port) OVERRIDE;
  60   virtual void SendConnectWithNetAddress(
  61       const PP_NetAddress_Private& addr) OVERRIDE;
  62   virtual void SendSSLHandshake(
  63       const std::string& server_name,
  64       uint16_t server_port,
  65       const std::vector<std::vector<char> >& trusted_certs,
  66       const std::vector<std::vector<char> >& untrusted_certs) OVERRIDE;
  67   virtual void SendRead(int32_t bytes_to_read) OVERRIDE;
  68   virtual void SendWrite(const std::string& buffer) OVERRIDE;
  69   virtual void SendDisconnect() OVERRIDE;
  70   virtual void SendSetOption(PP_TCPSocket_Option name,
  71                              const SocketOptionData& value) OVERRIDE;
  72   virtual Resource* GetOwnerResource() OVERRIDE;
  73
  74  private:
  75   void SendToBrowser(IPC::Message* msg);
  76
  77   DISALLOW_COPY_AND_ASSIGN(TCPSocket);
  78 };
  79
  80 TCPSocket::TCPSocket(const HostResource& resource, uint32 socket_id)
  81     : Resource(OBJECT_IS_PROXY, resource),
  82       TCPSocketShared(OBJECT_IS_PROXY, socket_id) {
  83   if (!g_id_to_socket)
  84     g_id_to_socket = new IDToSocketMap();
  85   DCHECK(g_id_to_socket->find(socket_id) == g_id_to_socket->end());
  86   (*g_id_to_socket)[socket_id] = this;
  87 }
  88
  89 TCPSocket::~TCPSocket() {
  90   DisconnectImpl();
  91 }
  92
  93 thunk::PPB_TCPSocket_API* TCPSocket::AsPPB_TCPSocket_API() {
  94   return this;
  95 }
  96
  97 int32_t TCPSocket::Connect(PP_Resource addr,
  98                            scoped_refptr<TrackedCallback> callback) {
  99   EnterNetAddressNoLock enter(addr, true);
 100   if (enter.failed())
 101     return PP_ERROR_BADARGUMENT;
 102
 103   return ConnectWithNetAddressImpl(&enter.object()->GetNetAddressPrivate(),
 104                                    callback);
 105 }
 106
 107 PP_Resource TCPSocket::GetLocalAddress() {
 108   PP_NetAddress_Private addr_private;
 109   if (!GetLocalAddressImpl(&addr_private))
 110     return 0;
 111
 112   thunk::EnterResourceCreationNoLock enter(pp_instance());
 113   if (enter.failed())
 114     return 0;
 115   return enter.functions()->CreateNetAddressFromNetAddressPrivate(
 116       pp_instance(), addr_private);
 117 }
 118
 119 PP_Resource TCPSocket::GetRemoteAddress() {
 120   PP_NetAddress_Private addr_private;
 121   if (!GetRemoteAddressImpl(&addr_private))
 122     return 0;
 123
 124   thunk::EnterResourceCreationNoLock enter(pp_instance());
 125   if (enter.failed())
 126     return 0;
 127   return enter.functions()->CreateNetAddressFromNetAddressPrivate(
 128       pp_instance(), addr_private);
 129 }
 130
 131 int32_t TCPSocket::Read(char* buffer,
 132                         int32_t bytes_to_read,
 133                         scoped_refptr<TrackedCallback> callback) {
 134   return ReadImpl(buffer, bytes_to_read, callback);
 135 }
 136
 137 int32_t TCPSocket::Write(const char* buffer,
 138                          int32_t bytes_to_write,
 139                          scoped_refptr<TrackedCallback> callback) {
 140   return WriteImpl(buffer, bytes_to_write, callback);
 141 }
 142
 143 void TCPSocket::Close() {
 144   DisconnectImpl();
 145 }
 146
 147 int32_t TCPSocket::SetOption(PP_TCPSocket_Option name,
 148                              const PP_Var& value,
 149                              scoped_refptr<TrackedCallback> callback) {
 150   return SetOptionImpl(name, value, callback);
 151 }
 152
 153 void TCPSocket::SendConnect(const std::string& host, uint16_t port) {
 154   NOTREACHED();
 155 }
 156
 157 void TCPSocket::SendConnectWithNetAddress(const PP_NetAddress_Private& addr) {
 158   SendToBrowser(new PpapiHostMsg_PPBTCPSocket_ConnectWithNetAddress(
 159       API_ID_PPB_TCPSOCKET, socket_id_, addr));
 160 }
 161
 162 void TCPSocket::SendSSLHandshake(
 163     const std::string& server_name,
 164     uint16_t server_port,
 165     const std::vector<std::vector<char> >& trusted_certs,
 166     const std::vector<std::vector<char> >& untrusted_certs) {
 167   NOTREACHED();
 168 }
 169
 170 void TCPSocket::SendRead(int32_t bytes_to_read) {
 171   SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Read(socket_id_, bytes_to_read));
 172 }
 173
 174 void TCPSocket::SendWrite(const std::string& buffer) {
 175   SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Write(socket_id_, buffer));
 176 }
 177
 178 void TCPSocket::SendDisconnect() {
 179   // After removed from the mapping, this object won't receive any notifications
 180   // from the proxy.
 181   DCHECK(g_id_to_socket->find(socket_id_) != g_id_to_socket->end());
 182   g_id_to_socket->erase(socket_id_);
 183   SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Disconnect(socket_id_));
 184 }
 185
 186 void TCPSocket::SendSetOption(PP_TCPSocket_Option name,
 187                               const SocketOptionData& value) {
 188   SendToBrowser(
 189       new PpapiHostMsg_PPBTCPSocket_SetOption(socket_id_, name, value));
 190 }
 191
 192 Resource* TCPSocket::GetOwnerResource() {
 193   return this;
 194 }
 195
 196 void TCPSocket::SendToBrowser(IPC::Message* msg) {
 197   PluginGlobals::Get()->GetBrowserSender()->Send(msg);
 198 }
 199
 200 }  // namespace
 201
 202 //------------------------------------------------------------------------------
 203
 204 PPB_TCPSocket_Proxy::PPB_TCPSocket_Proxy(Dispatcher* dispatcher)
 205     : InterfaceProxy(dispatcher) {
 206 }
 207
 208 PPB_TCPSocket_Proxy::~PPB_TCPSocket_Proxy() {
 209 }
 210
 211 // static
 212 PP_Resource PPB_TCPSocket_Proxy::CreateProxyResource(PP_Instance instance) {
 213   PluginDispatcher* dispatcher = PluginDispatcher::GetForInstance(instance);
 214   if (!dispatcher)
 215     return 0;
 216
 217   uint32 socket_id = 0;
 218   PluginGlobals::Get()->GetBrowserSender()->Send(
 219       new PpapiHostMsg_PPBTCPSocket_Create(
 220           API_ID_PPB_TCPSOCKET, dispatcher->plugin_dispatcher_id(),
 221           &socket_id));
 222   if (socket_id == 0)
 223     return 0;
 224   return (new TCPSocket(HostResource::MakeInstanceOnly(instance),
 225                         socket_id))->GetReference();
 226 }
 227
 228 bool PPB_TCPSocket_Proxy::OnMessageReceived(const IPC::Message& msg) {
 229   bool handled = true;
 230   IPC_BEGIN_MESSAGE_MAP(PPB_TCPSocket_Proxy, msg)
 231     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ConnectACK,
 232                         OnMsgConnectACK)
 233     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ReadACK, OnMsgReadACK)
 234     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_WriteACK, OnMsgWriteACK)
 235     IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_SetOptionACK,
 236                         OnMsgSetOptionACK)
 237     IPC_MESSAGE_UNHANDLED(handled = false)
 238   IPC_END_MESSAGE_MAP()
 239   return handled;
 240 }
 241
 242 void PPB_TCPSocket_Proxy::OnMsgConnectACK(
 243     uint32 /* plugin_dispatcher_id */,
 244     uint32 socket_id,
 245     int32_t result,
 246     const PP_NetAddress_Private& local_addr,
 247     const PP_NetAddress_Private& remote_addr) {
 248   if (!g_id_to_socket) {
 249     NOTREACHED();
 250     return;
 251   }
 252   IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id);
 253   if (iter == g_id_to_socket->end())
 254     return;
 255   iter->second->OnConnectCompleted(result, local_addr, remote_addr);
 256 }
 257
 258 void PPB_TCPSocket_Proxy::OnMsgReadACK(uint32 /* plugin_dispatcher_id */,
 259                                        uint32 socket_id,
 260                                        int32_t result,
 261                                        const std::string& data) {
 262   if (!g_id_to_socket) {
 263     NOTREACHED();
 264     return;
 265   }
 266   IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id);
 267   if (iter == g_id_to_socket->end())
 268     return;
 269   iter->second->OnReadCompleted(result, data);
 270 }
 271
 272 void PPB_TCPSocket_Proxy::OnMsgWriteACK(uint32 /* plugin_dispatcher_id */,
 273                                         uint32 socket_id,
 274                                         int32_t result) {
 275   if (!g_id_to_socket) {
 276     NOTREACHED();
 277     return;
 278   }
 279   IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id);
 280   if (iter == g_id_to_socket->end())
 281     return;
 282   iter->second->OnWriteCompleted(result);
 283 }
 284
 285 void PPB_TCPSocket_Proxy::OnMsgSetOptionACK(uint32 /* plugin_dispatcher_id */,
 286                                             uint32 socket_id,
 287                                             int32_t result) {
 288   if (!g_id_to_socket) {
 289     NOTREACHED();
 290     return;
 291   }
 292   IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id);
 293   if (iter == g_id_to_socket->end())
 294     return;
 295   iter->second->OnSetOptionCompleted(result);
 296 }
 297
 298 }  // namespace proxy
 299 }  // namespace ppapi