| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ppapi/proxy/ppb_flash_tcp_socket_proxy.h" | 5 #include "ppapi/proxy/ppb_tcp_socket_private_proxy.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <cstring> | 8 #include <cstring> |
| 9 #include <map> | 9 #include <map> |
| 10 | 10 |
| 11 #include "base/logging.h" | 11 #include "base/logging.h" |
| 12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/scoped_ptr.h" |
| 13 #include "base/message_loop.h" | 13 #include "base/message_loop.h" |
| 14 #include "base/task.h" | 14 #include "base/task.h" |
| 15 #include "ppapi/c/pp_errors.h" | 15 #include "ppapi/c/pp_errors.h" |
| 16 #include "ppapi/proxy/plugin_dispatcher.h" | 16 #include "ppapi/proxy/plugin_dispatcher.h" |
| 17 #include "ppapi/proxy/plugin_resource_tracker.h" | 17 #include "ppapi/proxy/plugin_resource_tracker.h" |
| 18 #include "ppapi/proxy/ppapi_messages.h" | 18 #include "ppapi/proxy/ppapi_messages.h" |
| 19 #include "ppapi/shared_impl/resource.h" | 19 #include "ppapi/shared_impl/resource.h" |
| 20 #include "ppapi/thunk/ppb_flash_tcp_socket_api.h" | 20 #include "ppapi/thunk/ppb_tcp_socket_private_api.h" |
| 21 #include "ppapi/thunk/thunk.h" | 21 #include "ppapi/thunk/thunk.h" |
| 22 | 22 |
| 23 using ppapi::thunk::PPB_Flash_TCPSocket_API; | 23 using ppapi::thunk::PPB_TCPSocket_Private_API; |
| 24 | 24 |
| 25 namespace ppapi { | 25 namespace ppapi { |
| 26 namespace proxy { | 26 namespace proxy { |
| 27 | 27 |
| 28 const int32_t kFlashTCPSocketMaxReadSize = 1024 * 1024; | 28 const int32_t kTCPSocketMaxReadSize = 1024 * 1024; |
| 29 const int32_t kFlashTCPSocketMaxWriteSize = 1024 * 1024; | 29 const int32_t kTCPSocketMaxWriteSize = 1024 * 1024; |
| 30 | 30 |
| 31 class FlashTCPSocket; | 31 class TCPSocket; |
| 32 | 32 |
| 33 namespace { | 33 namespace { |
| 34 | 34 |
| 35 typedef std::map<uint32, FlashTCPSocket*> IDToSocketMap; | 35 typedef std::map<uint32, TCPSocket*> IDToSocketMap; |
| 36 IDToSocketMap* g_id_to_socket = NULL; | 36 IDToSocketMap* g_id_to_socket = NULL; |
| 37 | 37 |
| 38 class AbortCallbackTask : public Task { | 38 class AbortCallbackTask : public Task { |
| 39 public: | 39 public: |
| 40 explicit AbortCallbackTask(PP_CompletionCallback callback) | 40 explicit AbortCallbackTask(PP_CompletionCallback callback) |
| 41 : callback_(callback) {} | 41 : callback_(callback) {} |
| 42 virtual ~AbortCallbackTask() {} | 42 virtual ~AbortCallbackTask() {} |
| 43 virtual void Run() { | 43 virtual void Run() { |
| 44 if (callback_.func) | 44 if (callback_.func) |
| 45 PP_RunCompletionCallback(&callback_, PP_ERROR_ABORTED); | 45 PP_RunCompletionCallback(&callback_, PP_ERROR_ABORTED); |
| 46 } | 46 } |
| 47 | 47 |
| 48 private: | 48 private: |
| 49 PP_CompletionCallback callback_; | 49 PP_CompletionCallback callback_; |
| 50 }; | 50 }; |
| 51 | 51 |
| 52 } // namespace | 52 } // namespace |
| 53 | 53 |
| 54 class FlashTCPSocket : public PPB_Flash_TCPSocket_API, | 54 class TCPSocket : public PPB_TCPSocket_Private_API, |
| 55 public Resource { | 55 public Resource { |
| 56 public: | 56 public: |
| 57 FlashTCPSocket(const HostResource& resource, uint32 socket_id); | 57 TCPSocket(const HostResource& resource, uint32 socket_id); |
| 58 virtual ~FlashTCPSocket(); | 58 virtual ~TCPSocket(); |
| 59 | 59 |
| 60 // Resource overrides. | 60 // Resource overrides. |
| 61 virtual PPB_Flash_TCPSocket_API* AsPPB_Flash_TCPSocket_API() OVERRIDE; | 61 virtual PPB_TCPSocket_Private_API* AsPPB_TCPSocket_Private_API() OVERRIDE; |
| 62 | 62 |
| 63 // PPB_Flash_TCPSocket_API implementation. | 63 // PPB_TCPSocket_Private_API implementation. |
| 64 virtual int32_t Connect(const char* host, | 64 virtual int32_t Connect(const char* host, |
| 65 uint16_t port, | 65 uint16_t port, |
| 66 PP_CompletionCallback callback) OVERRIDE; | 66 PP_CompletionCallback callback) OVERRIDE; |
| 67 virtual int32_t ConnectWithNetAddress( | 67 virtual int32_t ConnectWithNetAddress( |
| 68 const PP_NetAddress_Private* addr, | 68 const PP_NetAddress_Private* addr, |
| 69 PP_CompletionCallback callback) OVERRIDE; | 69 PP_CompletionCallback callback) OVERRIDE; |
| 70 virtual PP_Bool GetLocalAddress(PP_NetAddress_Private* local_addr) OVERRIDE; | 70 virtual PP_Bool GetLocalAddress(PP_NetAddress_Private* local_addr) OVERRIDE; |
| 71 virtual PP_Bool GetRemoteAddress(PP_NetAddress_Private* remote_addr) OVERRIDE; | 71 virtual PP_Bool GetRemoteAddress(PP_NetAddress_Private* remote_addr) OVERRIDE; |
| 72 virtual int32_t SSLHandshake(const char* server_name, | 72 virtual int32_t SSLHandshake(const char* server_name, |
| 73 uint16_t server_port, | 73 uint16_t server_port, |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 122 PP_CompletionCallback ssl_handshake_callback_; | 122 PP_CompletionCallback ssl_handshake_callback_; |
| 123 PP_CompletionCallback read_callback_; | 123 PP_CompletionCallback read_callback_; |
| 124 PP_CompletionCallback write_callback_; | 124 PP_CompletionCallback write_callback_; |
| 125 | 125 |
| 126 char* read_buffer_; | 126 char* read_buffer_; |
| 127 int32_t bytes_to_read_; | 127 int32_t bytes_to_read_; |
| 128 | 128 |
| 129 PP_NetAddress_Private local_addr_; | 129 PP_NetAddress_Private local_addr_; |
| 130 PP_NetAddress_Private remote_addr_; | 130 PP_NetAddress_Private remote_addr_; |
| 131 | 131 |
| 132 DISALLOW_COPY_AND_ASSIGN(FlashTCPSocket); | 132 DISALLOW_COPY_AND_ASSIGN(TCPSocket); |
| 133 }; | 133 }; |
| 134 | 134 |
| 135 FlashTCPSocket::FlashTCPSocket(const HostResource& resource, uint32 socket_id) | 135 TCPSocket::TCPSocket(const HostResource& resource, uint32 socket_id) |
| 136 : Resource(resource), | 136 : Resource(resource), |
| 137 socket_id_(socket_id), | 137 socket_id_(socket_id), |
| 138 connection_state_(BEFORE_CONNECT), | 138 connection_state_(BEFORE_CONNECT), |
| 139 connect_callback_(PP_BlockUntilComplete()), | 139 connect_callback_(PP_BlockUntilComplete()), |
| 140 ssl_handshake_callback_(PP_BlockUntilComplete()), | 140 ssl_handshake_callback_(PP_BlockUntilComplete()), |
| 141 read_callback_(PP_BlockUntilComplete()), | 141 read_callback_(PP_BlockUntilComplete()), |
| 142 write_callback_(PP_BlockUntilComplete()), | 142 write_callback_(PP_BlockUntilComplete()), |
| 143 read_buffer_(NULL), | 143 read_buffer_(NULL), |
| 144 bytes_to_read_(-1) { | 144 bytes_to_read_(-1) { |
| 145 DCHECK(socket_id != 0); | 145 DCHECK(socket_id != 0); |
| 146 | 146 |
| 147 local_addr_.size = 0; | 147 local_addr_.size = 0; |
| 148 memset(local_addr_.data, 0, sizeof(local_addr_.data)); | 148 memset(local_addr_.data, 0, sizeof(local_addr_.data)); |
| 149 remote_addr_.size = 0; | 149 remote_addr_.size = 0; |
| 150 memset(remote_addr_.data, 0, sizeof(remote_addr_.data)); | 150 memset(remote_addr_.data, 0, sizeof(remote_addr_.data)); |
| 151 | 151 |
| 152 if (!g_id_to_socket) | 152 if (!g_id_to_socket) |
| 153 g_id_to_socket = new IDToSocketMap(); | 153 g_id_to_socket = new IDToSocketMap(); |
| 154 DCHECK(g_id_to_socket->find(socket_id) == g_id_to_socket->end()); | 154 DCHECK(g_id_to_socket->find(socket_id) == g_id_to_socket->end()); |
| 155 (*g_id_to_socket)[socket_id] = this; | 155 (*g_id_to_socket)[socket_id] = this; |
| 156 } | 156 } |
| 157 | 157 |
| 158 FlashTCPSocket::~FlashTCPSocket() { | 158 TCPSocket::~TCPSocket() { |
| 159 Disconnect(); | 159 Disconnect(); |
| 160 } | 160 } |
| 161 | 161 |
| 162 PPB_Flash_TCPSocket_API* FlashTCPSocket::AsPPB_Flash_TCPSocket_API() { | 162 PPB_TCPSocket_Private_API* TCPSocket::AsPPB_TCPSocket_Private_API() { |
| 163 return this; | 163 return this; |
| 164 } | 164 } |
| 165 | 165 |
| 166 int32_t FlashTCPSocket::Connect(const char* host, | 166 int32_t TCPSocket::Connect(const char* host, |
| 167 uint16_t port, | 167 uint16_t port, |
| 168 PP_CompletionCallback callback) { | 168 PP_CompletionCallback callback) { |
| 169 if (!host) | 169 if (!host) |
| 170 return PP_ERROR_BADARGUMENT; | 170 return PP_ERROR_BADARGUMENT; |
| 171 | 171 |
| 172 return ConnectWithMessage( | 172 return ConnectWithMessage( |
| 173 new PpapiHostMsg_PPBFlashTCPSocket_Connect(socket_id_, host, port), | 173 new PpapiHostMsg_PPBTCPSocket_Connect(socket_id_, host, port), |
| 174 callback); | 174 callback); |
| 175 } | 175 } |
| 176 | 176 |
| 177 int32_t FlashTCPSocket::ConnectWithNetAddress( | 177 int32_t TCPSocket::ConnectWithNetAddress( |
| 178 const PP_NetAddress_Private* addr, | 178 const PP_NetAddress_Private* addr, |
| 179 PP_CompletionCallback callback) { | 179 PP_CompletionCallback callback) { |
| 180 if (!addr) | 180 if (!addr) |
| 181 return PP_ERROR_BADARGUMENT; | 181 return PP_ERROR_BADARGUMENT; |
| 182 | 182 |
| 183 return ConnectWithMessage( | 183 return ConnectWithMessage( |
| 184 new PpapiHostMsg_PPBFlashTCPSocket_ConnectWithNetAddress( | 184 new PpapiHostMsg_PPBTCPSocket_ConnectWithNetAddress( |
| 185 socket_id_, *addr), | 185 socket_id_, *addr), |
| 186 callback); | 186 callback); |
| 187 } | 187 } |
| 188 | 188 |
| 189 PP_Bool FlashTCPSocket::GetLocalAddress(PP_NetAddress_Private* local_addr) { | 189 PP_Bool TCPSocket::GetLocalAddress(PP_NetAddress_Private* local_addr) { |
| 190 if (!IsConnected() || !local_addr) | 190 if (!IsConnected() || !local_addr) |
| 191 return PP_FALSE; | 191 return PP_FALSE; |
| 192 | 192 |
| 193 *local_addr = local_addr_; | 193 *local_addr = local_addr_; |
| 194 return PP_TRUE; | 194 return PP_TRUE; |
| 195 } | 195 } |
| 196 | 196 |
| 197 PP_Bool FlashTCPSocket::GetRemoteAddress(PP_NetAddress_Private* remote_addr) { | 197 PP_Bool TCPSocket::GetRemoteAddress(PP_NetAddress_Private* remote_addr) { |
| 198 if (!IsConnected() || !remote_addr) | 198 if (!IsConnected() || !remote_addr) |
| 199 return PP_FALSE; | 199 return PP_FALSE; |
| 200 | 200 |
| 201 *remote_addr = remote_addr_; | 201 *remote_addr = remote_addr_; |
| 202 return PP_TRUE; | 202 return PP_TRUE; |
| 203 } | 203 } |
| 204 | 204 |
| 205 int32_t FlashTCPSocket::SSLHandshake(const char* server_name, | 205 int32_t TCPSocket::SSLHandshake(const char* server_name, |
| 206 uint16_t server_port, | 206 uint16_t server_port, |
| 207 PP_CompletionCallback callback) { | 207 PP_CompletionCallback callback) { |
| 208 if (!server_name) | 208 if (!server_name) |
| 209 return PP_ERROR_BADARGUMENT; | 209 return PP_ERROR_BADARGUMENT; |
| 210 if (!callback.func) | 210 if (!callback.func) |
| 211 return PP_ERROR_BLOCKS_MAIN_THREAD; | 211 return PP_ERROR_BLOCKS_MAIN_THREAD; |
| 212 | 212 |
| 213 if (connection_state_ != CONNECTED) | 213 if (connection_state_ != CONNECTED) |
| 214 return PP_ERROR_FAILED; | 214 return PP_ERROR_FAILED; |
| 215 if (ssl_handshake_callback_.func || read_callback_.func || | 215 if (ssl_handshake_callback_.func || read_callback_.func || |
| 216 write_callback_.func) | 216 write_callback_.func) |
| 217 return PP_ERROR_INPROGRESS; | 217 return PP_ERROR_INPROGRESS; |
| 218 | 218 |
| 219 ssl_handshake_callback_ = callback; | 219 ssl_handshake_callback_ = callback; |
| 220 | 220 |
| 221 // Send the request, the browser will call us back via SSLHandshakeACK. | 221 // Send the request, the browser will call us back via SSLHandshakeACK. |
| 222 GetDispatcher()->SendToBrowser( | 222 GetDispatcher()->SendToBrowser( |
| 223 new PpapiHostMsg_PPBFlashTCPSocket_SSLHandshake( | 223 new PpapiHostMsg_PPBTCPSocket_SSLHandshake( |
| 224 socket_id_, std::string(server_name), server_port)); | 224 socket_id_, std::string(server_name), server_port)); |
| 225 return PP_OK_COMPLETIONPENDING; | 225 return PP_OK_COMPLETIONPENDING; |
| 226 } | 226 } |
| 227 | 227 |
| 228 int32_t FlashTCPSocket::Read(char* buffer, | 228 int32_t TCPSocket::Read(char* buffer, |
| 229 int32_t bytes_to_read, | 229 int32_t bytes_to_read, |
| 230 PP_CompletionCallback callback) { | 230 PP_CompletionCallback callback) { |
| 231 if (!buffer || bytes_to_read <= 0) | 231 if (!buffer || bytes_to_read <= 0) |
| 232 return PP_ERROR_BADARGUMENT; | 232 return PP_ERROR_BADARGUMENT; |
| 233 if (!callback.func) | 233 if (!callback.func) |
| 234 return PP_ERROR_BLOCKS_MAIN_THREAD; | 234 return PP_ERROR_BLOCKS_MAIN_THREAD; |
| 235 | 235 |
| 236 if (!IsConnected()) | 236 if (!IsConnected()) |
| 237 return PP_ERROR_FAILED; | 237 return PP_ERROR_FAILED; |
| 238 if (read_callback_.func || ssl_handshake_callback_.func) | 238 if (read_callback_.func || ssl_handshake_callback_.func) |
| 239 return PP_ERROR_INPROGRESS; | 239 return PP_ERROR_INPROGRESS; |
| 240 | 240 |
| 241 read_buffer_ = buffer; | 241 read_buffer_ = buffer; |
| 242 bytes_to_read_ = std::min(bytes_to_read, kFlashTCPSocketMaxReadSize); | 242 bytes_to_read_ = std::min(bytes_to_read, kTCPSocketMaxReadSize); |
| 243 read_callback_ = callback; | 243 read_callback_ = callback; |
| 244 | 244 |
| 245 // Send the request, the browser will call us back via ReadACK. | 245 // Send the request, the browser will call us back via ReadACK. |
| 246 GetDispatcher()->SendToBrowser( | 246 GetDispatcher()->SendToBrowser( |
| 247 new PpapiHostMsg_PPBFlashTCPSocket_Read(socket_id_, bytes_to_read_)); | 247 new PpapiHostMsg_PPBTCPSocket_Read(socket_id_, bytes_to_read_)); |
| 248 return PP_OK_COMPLETIONPENDING; | 248 return PP_OK_COMPLETIONPENDING; |
| 249 } | 249 } |
| 250 | 250 |
| 251 int32_t FlashTCPSocket::Write(const char* buffer, | 251 int32_t TCPSocket::Write(const char* buffer, |
| 252 int32_t bytes_to_write, | 252 int32_t bytes_to_write, |
| 253 PP_CompletionCallback callback) { | 253 PP_CompletionCallback callback) { |
| 254 if (!buffer || bytes_to_write <= 0) | 254 if (!buffer || bytes_to_write <= 0) |
| 255 return PP_ERROR_BADARGUMENT; | 255 return PP_ERROR_BADARGUMENT; |
| 256 if (!callback.func) | 256 if (!callback.func) |
| 257 return PP_ERROR_BLOCKS_MAIN_THREAD; | 257 return PP_ERROR_BLOCKS_MAIN_THREAD; |
| 258 | 258 |
| 259 if (!IsConnected()) | 259 if (!IsConnected()) |
| 260 return PP_ERROR_FAILED; | 260 return PP_ERROR_FAILED; |
| 261 if (write_callback_.func || ssl_handshake_callback_.func) | 261 if (write_callback_.func || ssl_handshake_callback_.func) |
| 262 return PP_ERROR_INPROGRESS; | 262 return PP_ERROR_INPROGRESS; |
| 263 | 263 |
| 264 if (bytes_to_write > kFlashTCPSocketMaxWriteSize) | 264 if (bytes_to_write > kTCPSocketMaxWriteSize) |
| 265 bytes_to_write = kFlashTCPSocketMaxWriteSize; | 265 bytes_to_write = kTCPSocketMaxWriteSize; |
| 266 | 266 |
| 267 write_callback_ = callback; | 267 write_callback_ = callback; |
| 268 | 268 |
| 269 // Send the request, the browser will call us back via WriteACK. | 269 // Send the request, the browser will call us back via WriteACK. |
| 270 GetDispatcher()->SendToBrowser( | 270 GetDispatcher()->SendToBrowser( |
| 271 new PpapiHostMsg_PPBFlashTCPSocket_Write( | 271 new PpapiHostMsg_PPBTCPSocket_Write( |
| 272 socket_id_, std::string(buffer, bytes_to_write))); | 272 socket_id_, std::string(buffer, bytes_to_write))); |
| 273 return PP_OK_COMPLETIONPENDING; | 273 return PP_OK_COMPLETIONPENDING; |
| 274 } | 274 } |
| 275 | 275 |
| 276 void FlashTCPSocket::Disconnect() { | 276 void TCPSocket::Disconnect() { |
| 277 if (connection_state_ == DISCONNECTED) | 277 if (connection_state_ == DISCONNECTED) |
| 278 return; | 278 return; |
| 279 | 279 |
| 280 connection_state_ = DISCONNECTED; | 280 connection_state_ = DISCONNECTED; |
| 281 // After removed from the mapping, this object won't receive any notifications | 281 // After removed from the mapping, this object won't receive any notifications |
| 282 // from the proxy. | 282 // from the proxy. |
| 283 DCHECK(g_id_to_socket->find(socket_id_) != g_id_to_socket->end()); | 283 DCHECK(g_id_to_socket->find(socket_id_) != g_id_to_socket->end()); |
| 284 g_id_to_socket->erase(socket_id_); | 284 g_id_to_socket->erase(socket_id_); |
| 285 | 285 |
| 286 GetDispatcher()->SendToBrowser( | 286 GetDispatcher()->SendToBrowser( |
| 287 new PpapiHostMsg_PPBFlashTCPSocket_Disconnect(socket_id_)); | 287 new PpapiHostMsg_PPBTCPSocket_Disconnect(socket_id_)); |
| 288 socket_id_ = 0; | 288 socket_id_ = 0; |
| 289 | 289 |
| 290 PostAbortAndClearIfNecessary(&connect_callback_); | 290 PostAbortAndClearIfNecessary(&connect_callback_); |
| 291 PostAbortAndClearIfNecessary(&ssl_handshake_callback_); | 291 PostAbortAndClearIfNecessary(&ssl_handshake_callback_); |
| 292 PostAbortAndClearIfNecessary(&read_callback_); | 292 PostAbortAndClearIfNecessary(&read_callback_); |
| 293 PostAbortAndClearIfNecessary(&write_callback_); | 293 PostAbortAndClearIfNecessary(&write_callback_); |
| 294 read_buffer_ = NULL; | 294 read_buffer_ = NULL; |
| 295 bytes_to_read_ = -1; | 295 bytes_to_read_ = -1; |
| 296 } | 296 } |
| 297 | 297 |
| 298 void FlashTCPSocket::OnConnectCompleted( | 298 void TCPSocket::OnConnectCompleted( |
| 299 bool succeeded, | 299 bool succeeded, |
| 300 const PP_NetAddress_Private& local_addr, | 300 const PP_NetAddress_Private& local_addr, |
| 301 const PP_NetAddress_Private& remote_addr) { | 301 const PP_NetAddress_Private& remote_addr) { |
| 302 if (connection_state_ != BEFORE_CONNECT || !connect_callback_.func) { | 302 if (connection_state_ != BEFORE_CONNECT || !connect_callback_.func) { |
| 303 NOTREACHED(); | 303 NOTREACHED(); |
| 304 return; | 304 return; |
| 305 } | 305 } |
| 306 | 306 |
| 307 if (succeeded) { | 307 if (succeeded) { |
| 308 local_addr_ = local_addr; | 308 local_addr_ = local_addr; |
| 309 remote_addr_ = remote_addr; | 309 remote_addr_ = remote_addr; |
| 310 connection_state_ = CONNECTED; | 310 connection_state_ = CONNECTED; |
| 311 } | 311 } |
| 312 PP_RunAndClearCompletionCallback(&connect_callback_, | 312 PP_RunAndClearCompletionCallback(&connect_callback_, |
| 313 succeeded ? PP_OK : PP_ERROR_FAILED); | 313 succeeded ? PP_OK : PP_ERROR_FAILED); |
| 314 } | 314 } |
| 315 | 315 |
| 316 void FlashTCPSocket::OnSSLHandshakeCompleted(bool succeeded) { | 316 void TCPSocket::OnSSLHandshakeCompleted(bool succeeded) { |
| 317 if (connection_state_ != CONNECTED || !ssl_handshake_callback_.func) { | 317 if (connection_state_ != CONNECTED || !ssl_handshake_callback_.func) { |
| 318 NOTREACHED(); | 318 NOTREACHED(); |
| 319 return; | 319 return; |
| 320 } | 320 } |
| 321 | 321 |
| 322 if (succeeded) { | 322 if (succeeded) { |
| 323 connection_state_ = SSL_CONNECTED; | 323 connection_state_ = SSL_CONNECTED; |
| 324 PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_OK); | 324 PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_OK); |
| 325 } else { | 325 } else { |
| 326 PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_ERROR_FAILED); | 326 PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_ERROR_FAILED); |
| 327 Disconnect(); | 327 Disconnect(); |
| 328 } | 328 } |
| 329 } | 329 } |
| 330 | 330 |
| 331 void FlashTCPSocket::OnReadCompleted(bool succeeded, const std::string& data) { | 331 void TCPSocket::OnReadCompleted(bool succeeded, const std::string& data) { |
| 332 if (!read_callback_.func || !read_buffer_) { | 332 if (!read_callback_.func || !read_buffer_) { |
| 333 NOTREACHED(); | 333 NOTREACHED(); |
| 334 return; | 334 return; |
| 335 } | 335 } |
| 336 | 336 |
| 337 if (succeeded) { | 337 if (succeeded) { |
| 338 CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_); | 338 CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_); |
| 339 if (!data.empty()) | 339 if (!data.empty()) |
| 340 memcpy(read_buffer_, data.c_str(), data.size()); | 340 memcpy(read_buffer_, data.c_str(), data.size()); |
| 341 } | 341 } |
| 342 read_buffer_ = NULL; | 342 read_buffer_ = NULL; |
| 343 bytes_to_read_ = -1; | 343 bytes_to_read_ = -1; |
| 344 | 344 |
| 345 PP_RunAndClearCompletionCallback( | 345 PP_RunAndClearCompletionCallback( |
| 346 &read_callback_, | 346 &read_callback_, |
| 347 succeeded ? static_cast<int32_t>(data.size()) : | 347 succeeded ? static_cast<int32_t>(data.size()) : |
| 348 static_cast<int32_t>(PP_ERROR_FAILED)); | 348 static_cast<int32_t>(PP_ERROR_FAILED)); |
| 349 } | 349 } |
| 350 | 350 |
| 351 void FlashTCPSocket::OnWriteCompleted(bool succeeded, int32_t bytes_written) { | 351 void TCPSocket::OnWriteCompleted(bool succeeded, int32_t bytes_written) { |
| 352 if (!write_callback_.func || (succeeded && bytes_written < 0)) { | 352 if (!write_callback_.func || (succeeded && bytes_written < 0)) { |
| 353 NOTREACHED(); | 353 NOTREACHED(); |
| 354 return; | 354 return; |
| 355 } | 355 } |
| 356 | 356 |
| 357 PP_RunAndClearCompletionCallback( | 357 PP_RunAndClearCompletionCallback( |
| 358 &write_callback_, | 358 &write_callback_, |
| 359 succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED)); | 359 succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED)); |
| 360 } | 360 } |
| 361 | 361 |
| 362 bool FlashTCPSocket::IsConnected() const { | 362 bool TCPSocket::IsConnected() const { |
| 363 return connection_state_ == CONNECTED || connection_state_ == SSL_CONNECTED; | 363 return connection_state_ == CONNECTED || connection_state_ == SSL_CONNECTED; |
| 364 } | 364 } |
| 365 | 365 |
| 366 int32_t FlashTCPSocket::ConnectWithMessage(IPC::Message* msg, | 366 int32_t TCPSocket::ConnectWithMessage(IPC::Message* msg, |
| 367 PP_CompletionCallback callback) { | 367 PP_CompletionCallback callback) { |
| 368 scoped_ptr<IPC::Message> msg_deletor(msg); | 368 scoped_ptr<IPC::Message> msg_deletor(msg); |
| 369 if (!callback.func) | 369 if (!callback.func) |
| 370 return PP_ERROR_BLOCKS_MAIN_THREAD; | 370 return PP_ERROR_BLOCKS_MAIN_THREAD; |
| 371 if (connection_state_ != BEFORE_CONNECT) | 371 if (connection_state_ != BEFORE_CONNECT) |
| 372 return PP_ERROR_FAILED; | 372 return PP_ERROR_FAILED; |
| 373 if (connect_callback_.func) | 373 if (connect_callback_.func) |
| 374 return PP_ERROR_INPROGRESS; // Can only have one pending request. | 374 return PP_ERROR_INPROGRESS; // Can only have one pending request. |
| 375 | 375 |
| 376 connect_callback_ = callback; | 376 connect_callback_ = callback; |
| 377 // Send the request, the browser will call us back via ConnectACK. | 377 // Send the request, the browser will call us back via ConnectACK. |
| 378 GetDispatcher()->SendToBrowser(msg_deletor.release()); | 378 GetDispatcher()->SendToBrowser(msg_deletor.release()); |
| 379 return PP_OK_COMPLETIONPENDING; | 379 return PP_OK_COMPLETIONPENDING; |
| 380 } | 380 } |
| 381 | 381 |
| 382 void FlashTCPSocket::PostAbortAndClearIfNecessary( | 382 void TCPSocket::PostAbortAndClearIfNecessary( |
| 383 PP_CompletionCallback* callback) { | 383 PP_CompletionCallback* callback) { |
| 384 DCHECK(callback); | 384 DCHECK(callback); |
| 385 | 385 |
| 386 if (callback->func) { | 386 if (callback->func) { |
| 387 MessageLoop::current()->PostTask(FROM_HERE, | 387 MessageLoop::current()->PostTask(FROM_HERE, |
| 388 new AbortCallbackTask(*callback)); | 388 new AbortCallbackTask(*callback)); |
| 389 *callback = PP_BlockUntilComplete(); | 389 *callback = PP_BlockUntilComplete(); |
| 390 } | 390 } |
| 391 } | 391 } |
| 392 | 392 |
| 393 PPB_Flash_TCPSocket_Proxy::PPB_Flash_TCPSocket_Proxy(Dispatcher* dispatcher) | 393 PPB_TCPSocket_Private_Proxy::PPB_TCPSocket_Private_Proxy(Dispatcher* dispatcher) |
| 394 : InterfaceProxy(dispatcher) { | 394 : InterfaceProxy(dispatcher) { |
| 395 } | 395 } |
| 396 | 396 |
| 397 PPB_Flash_TCPSocket_Proxy::~PPB_Flash_TCPSocket_Proxy() { | 397 PPB_TCPSocket_Private_Proxy::~PPB_TCPSocket_Private_Proxy() { |
| 398 } | 398 } |
| 399 | 399 |
| 400 // static | 400 // static |
| 401 PP_Resource PPB_Flash_TCPSocket_Proxy::CreateProxyResource( | 401 PP_Resource PPB_TCPSocket_Private_Proxy::CreateProxyResource( |
| 402 PP_Instance instance) { | 402 PP_Instance instance) { |
| 403 PluginDispatcher* dispatcher = PluginDispatcher::GetForInstance(instance); | 403 PluginDispatcher* dispatcher = PluginDispatcher::GetForInstance(instance); |
| 404 if (!dispatcher) | 404 if (!dispatcher) |
| 405 return 0; | 405 return 0; |
| 406 | 406 |
| 407 uint32 socket_id = 0; | 407 uint32 socket_id = 0; |
| 408 dispatcher->SendToBrowser(new PpapiHostMsg_PPBFlashTCPSocket_Create( | 408 dispatcher->SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Create( |
| 409 API_ID_PPB_FLASH_TCPSOCKET, dispatcher->plugin_dispatcher_id(), | 409 API_ID_PPB_TCPSOCKET_PRIVATE, dispatcher->plugin_dispatcher_id(), |
| 410 &socket_id)); | 410 &socket_id)); |
| 411 if (socket_id == 0) | 411 if (socket_id == 0) |
| 412 return 0; | 412 return 0; |
| 413 return (new FlashTCPSocket(HostResource::MakeInstanceOnly(instance), | 413 return (new TCPSocket(HostResource::MakeInstanceOnly(instance), |
| 414 socket_id))->GetReference(); | 414 socket_id))->GetReference(); |
| 415 } | 415 } |
| 416 | 416 |
| 417 bool PPB_Flash_TCPSocket_Proxy::OnMessageReceived(const IPC::Message& msg) { | 417 bool PPB_TCPSocket_Private_Proxy::OnMessageReceived(const IPC::Message& msg) { |
| 418 bool handled = true; | 418 bool handled = true; |
| 419 IPC_BEGIN_MESSAGE_MAP(PPB_Flash_TCPSocket_Proxy, msg) | 419 IPC_BEGIN_MESSAGE_MAP(PPB_TCPSocket_Private_Proxy, msg) |
| 420 IPC_MESSAGE_HANDLER(PpapiMsg_PPBFlashTCPSocket_ConnectACK, OnMsgConnectACK) | 420 IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ConnectACK, |
| 421 IPC_MESSAGE_HANDLER(PpapiMsg_PPBFlashTCPSocket_SSLHandshakeACK, | 421 OnMsgConnectACK) |
| 422 IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_SSLHandshakeACK, |
| 422 OnMsgSSLHandshakeACK) | 423 OnMsgSSLHandshakeACK) |
| 423 IPC_MESSAGE_HANDLER(PpapiMsg_PPBFlashTCPSocket_ReadACK, OnMsgReadACK) | 424 IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_ReadACK, OnMsgReadACK) |
| 424 IPC_MESSAGE_HANDLER(PpapiMsg_PPBFlashTCPSocket_WriteACK, OnMsgWriteACK) | 425 IPC_MESSAGE_HANDLER(PpapiMsg_PPBTCPSocket_WriteACK, OnMsgWriteACK) |
| 425 IPC_MESSAGE_UNHANDLED(handled = false) | 426 IPC_MESSAGE_UNHANDLED(handled = false) |
| 426 IPC_END_MESSAGE_MAP() | 427 IPC_END_MESSAGE_MAP() |
| 427 return handled; | 428 return handled; |
| 428 } | 429 } |
| 429 | 430 |
| 430 void PPB_Flash_TCPSocket_Proxy::OnMsgConnectACK( | 431 void PPB_TCPSocket_Private_Proxy::OnMsgConnectACK( |
| 431 uint32 /* plugin_dispatcher_id */, | 432 uint32 /* plugin_dispatcher_id */, |
| 432 uint32 socket_id, | 433 uint32 socket_id, |
| 433 bool succeeded, | 434 bool succeeded, |
| 434 const PP_NetAddress_Private& local_addr, | 435 const PP_NetAddress_Private& local_addr, |
| 435 const PP_NetAddress_Private& remote_addr) { | 436 const PP_NetAddress_Private& remote_addr) { |
| 436 if (!g_id_to_socket) { | 437 if (!g_id_to_socket) { |
| 437 NOTREACHED(); | 438 NOTREACHED(); |
| 438 return; | 439 return; |
| 439 } | 440 } |
| 440 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); | 441 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); |
| 441 if (iter == g_id_to_socket->end()) | 442 if (iter == g_id_to_socket->end()) |
| 442 return; | 443 return; |
| 443 iter->second->OnConnectCompleted(succeeded, local_addr, remote_addr); | 444 iter->second->OnConnectCompleted(succeeded, local_addr, remote_addr); |
| 444 } | 445 } |
| 445 | 446 |
| 446 void PPB_Flash_TCPSocket_Proxy::OnMsgSSLHandshakeACK( | 447 void PPB_TCPSocket_Private_Proxy::OnMsgSSLHandshakeACK( |
| 447 uint32 /* plugin_dispatcher_id */, | 448 uint32 /* plugin_dispatcher_id */, |
| 448 uint32 socket_id, | 449 uint32 socket_id, |
| 449 bool succeeded) { | 450 bool succeeded) { |
| 450 if (!g_id_to_socket) { | 451 if (!g_id_to_socket) { |
| 451 NOTREACHED(); | 452 NOTREACHED(); |
| 452 return; | 453 return; |
| 453 } | 454 } |
| 454 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); | 455 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); |
| 455 if (iter == g_id_to_socket->end()) | 456 if (iter == g_id_to_socket->end()) |
| 456 return; | 457 return; |
| 457 iter->second->OnSSLHandshakeCompleted(succeeded); | 458 iter->second->OnSSLHandshakeCompleted(succeeded); |
| 458 } | 459 } |
| 459 | 460 |
| 460 void PPB_Flash_TCPSocket_Proxy::OnMsgReadACK(uint32 /* plugin_dispatcher_id */, | 461 void PPB_TCPSocket_Private_Proxy::OnMsgReadACK( |
| 461 uint32 socket_id, | 462 uint32 /* plugin_dispatcher_id */, |
| 462 bool succeeded, | 463 uint32 socket_id, |
| 463 const std::string& data) { | 464 bool succeeded, |
| 465 const std::string& data) { |
| 464 if (!g_id_to_socket) { | 466 if (!g_id_to_socket) { |
| 465 NOTREACHED(); | 467 NOTREACHED(); |
| 466 return; | 468 return; |
| 467 } | 469 } |
| 468 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); | 470 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); |
| 469 if (iter == g_id_to_socket->end()) | 471 if (iter == g_id_to_socket->end()) |
| 470 return; | 472 return; |
| 471 iter->second->OnReadCompleted(succeeded, data); | 473 iter->second->OnReadCompleted(succeeded, data); |
| 472 } | 474 } |
| 473 | 475 |
| 474 void PPB_Flash_TCPSocket_Proxy::OnMsgWriteACK(uint32 /* plugin_dispatcher_id */, | 476 void PPB_TCPSocket_Private_Proxy::OnMsgWriteACK( |
| 475 uint32 socket_id, | 477 uint32 /* plugin_dispatcher_id */, |
| 476 bool succeeded, | 478 uint32 socket_id, |
| 477 int32_t bytes_written) { | 479 bool succeeded, |
| 480 int32_t bytes_written) { |
| 478 if (!g_id_to_socket) { | 481 if (!g_id_to_socket) { |
| 479 NOTREACHED(); | 482 NOTREACHED(); |
| 480 return; | 483 return; |
| 481 } | 484 } |
| 482 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); | 485 IDToSocketMap::iterator iter = g_id_to_socket->find(socket_id); |
| 483 if (iter == g_id_to_socket->end()) | 486 if (iter == g_id_to_socket->end()) |
| 484 return; | 487 return; |
| 485 iter->second->OnWriteCompleted(succeeded, bytes_written); | 488 iter->second->OnWriteCompleted(succeeded, bytes_written); |
| 486 } | 489 } |
| 487 | 490 |
| 488 } // namespace proxy | 491 } // namespace proxy |
| 489 } // namespace ppapi | 492 } // namespace ppapi |
| OLD | NEW |