| Index: ppapi/proxy/ppb_tcp_socket_private_proxy.cc
|
| diff --git a/ppapi/proxy/ppb_tcp_socket_private_proxy.cc b/ppapi/proxy/ppb_tcp_socket_private_proxy.cc
|
| index 46137841bf3094696d7e4e7f9f112dace49e0ef7..b943889a107f0ecae4523ac28939cdbf2256de7a 100644
|
| --- a/ppapi/proxy/ppb_tcp_socket_private_proxy.cc
|
| +++ b/ppapi/proxy/ppb_tcp_socket_private_proxy.cc
|
| @@ -4,143 +4,46 @@
|
|
|
| #include "ppapi/proxy/ppb_tcp_socket_private_proxy.h"
|
|
|
| -#include <algorithm>
|
| -#include <cstring>
|
| #include <map>
|
|
|
| -#include "base/bind.h"
|
| #include "base/logging.h"
|
| -#include "base/memory/scoped_ptr.h"
|
| -#include "base/message_loop.h"
|
| -#include "base/task.h"
|
| -#include "ppapi/c/pp_errors.h"
|
| #include "ppapi/proxy/plugin_dispatcher.h"
|
| #include "ppapi/proxy/plugin_resource_tracker.h"
|
| #include "ppapi/proxy/ppapi_messages.h"
|
| +#include "ppapi/shared_impl/private/tcp_socket_private_impl.h"
|
| #include "ppapi/shared_impl/resource.h"
|
| -#include "ppapi/thunk/ppb_tcp_socket_private_api.h"
|
| #include "ppapi/thunk/thunk.h"
|
|
|
| -using ppapi::thunk::PPB_TCPSocket_Private_API;
|
| -
|
| namespace ppapi {
|
| namespace proxy {
|
|
|
| -const int32_t kTCPSocketMaxReadSize = 1024 * 1024;
|
| -const int32_t kTCPSocketMaxWriteSize = 1024 * 1024;
|
| -
|
| -class TCPSocket;
|
| -
|
| namespace {
|
|
|
| -typedef std::map<uint32, TCPSocket*> IDToSocketMap;
|
| +typedef std::map<uint32, TCPSocketPrivateImpl*> IDToSocketMap;
|
| IDToSocketMap* g_id_to_socket = NULL;
|
|
|
| -void AbortCallback(PP_CompletionCallback callback) {
|
| - if (callback.func)
|
| - PP_RunCompletionCallback(&callback, PP_ERROR_ABORTED);
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -class TCPSocket : public PPB_TCPSocket_Private_API,
|
| - public Resource {
|
| +class TCPSocket : public TCPSocketPrivateImpl {
|
| public:
|
| TCPSocket(const HostResource& resource, uint32 socket_id);
|
| virtual ~TCPSocket();
|
|
|
| - // Resource overrides.
|
| - virtual PPB_TCPSocket_Private_API* AsPPB_TCPSocket_Private_API() OVERRIDE;
|
| -
|
| - // PPB_TCPSocket_Private_API implementation.
|
| - virtual int32_t Connect(const char* host,
|
| - uint16_t port,
|
| - PP_CompletionCallback callback) OVERRIDE;
|
| - virtual int32_t ConnectWithNetAddress(
|
| - const PP_NetAddress_Private* addr,
|
| - PP_CompletionCallback callback) OVERRIDE;
|
| - virtual PP_Bool GetLocalAddress(PP_NetAddress_Private* local_addr) OVERRIDE;
|
| - virtual PP_Bool GetRemoteAddress(PP_NetAddress_Private* remote_addr) OVERRIDE;
|
| - virtual int32_t SSLHandshake(const char* server_name,
|
| - uint16_t server_port,
|
| - PP_CompletionCallback callback) OVERRIDE;
|
| - virtual int32_t Read(char* buffer,
|
| - int32_t bytes_to_read,
|
| - PP_CompletionCallback callback) OVERRIDE;
|
| - virtual int32_t Write(const char* buffer,
|
| - int32_t bytes_to_write,
|
| - PP_CompletionCallback callback) OVERRIDE;
|
| - virtual void Disconnect() OVERRIDE;
|
| -
|
| - // Notifications from the proxy.
|
| - void OnConnectCompleted(bool succeeded,
|
| - const PP_NetAddress_Private& local_addr,
|
| - const PP_NetAddress_Private& remote_addr);
|
| - void OnSSLHandshakeCompleted(bool succeeded);
|
| - void OnReadCompleted(bool succeeded, const std::string& data);
|
| - void OnWriteCompleted(bool succeeded, int32_t bytes_written);
|
| + virtual void SendConnect(const std::string& host, uint16_t port) OVERRIDE;
|
| + virtual void SendConnectWithNetAddress(
|
| + const PP_NetAddress_Private& addr) OVERRIDE;
|
| + virtual void SendSSLHandshake(const std::string& server_name,
|
| + uint16_t server_port) OVERRIDE;
|
| + virtual void SendRead(int32_t bytes_to_read) OVERRIDE;
|
| + virtual void SendWrite(const std::string& buffer) OVERRIDE;
|
| + virtual void SendDisconnect() OVERRIDE;
|
|
|
| private:
|
| - enum ConnectionState {
|
| - // Before a connection is successfully established (including a connect
|
| - // request is pending or a previous connect request failed).
|
| - BEFORE_CONNECT,
|
| - // A connection has been successfully established (including a request of
|
| - // initiating SSL is pending).
|
| - CONNECTED,
|
| - // An SSL connection has been successfully established.
|
| - SSL_CONNECTED,
|
| - // The connection has been ended.
|
| - DISCONNECTED
|
| - };
|
| -
|
| - bool IsConnected() const;
|
| -
|
| - PluginDispatcher* GetDispatcher() const {
|
| - return PluginDispatcher::GetForResource(this);
|
| - }
|
| -
|
| - // Backend for both Connect() and ConnectWithNetAddress(). To keep things
|
| - // generic, the message is passed in (on error, it's deleted).
|
| - int32_t ConnectWithMessage(IPC::Message* msg,
|
| - PP_CompletionCallback callback);
|
| -
|
| - void PostAbortAndClearIfNecessary(PP_CompletionCallback* callback);
|
| -
|
| - uint32 socket_id_;
|
| - ConnectionState connection_state_;
|
| -
|
| - PP_CompletionCallback connect_callback_;
|
| - PP_CompletionCallback ssl_handshake_callback_;
|
| - PP_CompletionCallback read_callback_;
|
| - PP_CompletionCallback write_callback_;
|
| -
|
| - char* read_buffer_;
|
| - int32_t bytes_to_read_;
|
| -
|
| - PP_NetAddress_Private local_addr_;
|
| - PP_NetAddress_Private remote_addr_;
|
| + void SendToBrowser(IPC::Message* msg);
|
|
|
| DISALLOW_COPY_AND_ASSIGN(TCPSocket);
|
| };
|
|
|
| TCPSocket::TCPSocket(const HostResource& resource, uint32 socket_id)
|
| - : Resource(resource),
|
| - socket_id_(socket_id),
|
| - connection_state_(BEFORE_CONNECT),
|
| - connect_callback_(PP_BlockUntilComplete()),
|
| - ssl_handshake_callback_(PP_BlockUntilComplete()),
|
| - read_callback_(PP_BlockUntilComplete()),
|
| - write_callback_(PP_BlockUntilComplete()),
|
| - read_buffer_(NULL),
|
| - bytes_to_read_(-1) {
|
| - DCHECK(socket_id != 0);
|
| -
|
| - local_addr_.size = 0;
|
| - memset(local_addr_.data, 0, sizeof(local_addr_.data));
|
| - remote_addr_.size = 0;
|
| - memset(remote_addr_.data, 0, sizeof(remote_addr_.data));
|
| -
|
| + : TCPSocketPrivateImpl(resource, socket_id) {
|
| if (!g_id_to_socket)
|
| g_id_to_socket = new IDToSocketMap();
|
| DCHECK(g_id_to_socket->find(socket_id) == g_id_to_socket->end());
|
| @@ -151,236 +54,44 @@ TCPSocket::~TCPSocket() {
|
| Disconnect();
|
| }
|
|
|
| -PPB_TCPSocket_Private_API* TCPSocket::AsPPB_TCPSocket_Private_API() {
|
| - return this;
|
| +void TCPSocket::SendConnect(const std::string& host, uint16_t port) {
|
| + SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Connect(socket_id_, host, port));
|
| }
|
|
|
| -int32_t TCPSocket::Connect(const char* host,
|
| - uint16_t port,
|
| - PP_CompletionCallback callback) {
|
| - if (!host)
|
| - return PP_ERROR_BADARGUMENT;
|
| -
|
| - return ConnectWithMessage(
|
| - new PpapiHostMsg_PPBTCPSocket_Connect(socket_id_, host, port),
|
| - callback);
|
| +void TCPSocket::SendConnectWithNetAddress(const PP_NetAddress_Private& addr) {
|
| + SendToBrowser(
|
| + new PpapiHostMsg_PPBTCPSocket_ConnectWithNetAddress(socket_id_, addr));
|
| }
|
|
|
| -int32_t TCPSocket::ConnectWithNetAddress(
|
| - const PP_NetAddress_Private* addr,
|
| - PP_CompletionCallback callback) {
|
| - if (!addr)
|
| - return PP_ERROR_BADARGUMENT;
|
| -
|
| - return ConnectWithMessage(
|
| - new PpapiHostMsg_PPBTCPSocket_ConnectWithNetAddress(
|
| - socket_id_, *addr),
|
| - callback);
|
| -}
|
| -
|
| -PP_Bool TCPSocket::GetLocalAddress(PP_NetAddress_Private* local_addr) {
|
| - if (!IsConnected() || !local_addr)
|
| - return PP_FALSE;
|
| -
|
| - *local_addr = local_addr_;
|
| - return PP_TRUE;
|
| -}
|
| -
|
| -PP_Bool TCPSocket::GetRemoteAddress(PP_NetAddress_Private* remote_addr) {
|
| - if (!IsConnected() || !remote_addr)
|
| - return PP_FALSE;
|
| -
|
| - *remote_addr = remote_addr_;
|
| - return PP_TRUE;
|
| -}
|
| -
|
| -int32_t TCPSocket::SSLHandshake(const char* server_name,
|
| - uint16_t server_port,
|
| - PP_CompletionCallback callback) {
|
| - if (!server_name)
|
| - return PP_ERROR_BADARGUMENT;
|
| - if (!callback.func)
|
| - return PP_ERROR_BLOCKS_MAIN_THREAD;
|
| -
|
| - if (connection_state_ != CONNECTED)
|
| - return PP_ERROR_FAILED;
|
| - if (ssl_handshake_callback_.func || read_callback_.func ||
|
| - write_callback_.func)
|
| - return PP_ERROR_INPROGRESS;
|
| -
|
| - ssl_handshake_callback_ = callback;
|
| -
|
| - // Send the request, the browser will call us back via SSLHandshakeACK.
|
| - GetDispatcher()->SendToBrowser(
|
| - new PpapiHostMsg_PPBTCPSocket_SSLHandshake(
|
| - socket_id_, std::string(server_name), server_port));
|
| - return PP_OK_COMPLETIONPENDING;
|
| +void TCPSocket::SendSSLHandshake(const std::string& server_name,
|
| + uint16_t server_port) {
|
| + SendToBrowser(new PpapiHostMsg_PPBTCPSocket_SSLHandshake(
|
| + socket_id_, server_name, server_port));
|
| }
|
|
|
| -int32_t TCPSocket::Read(char* buffer,
|
| - int32_t bytes_to_read,
|
| - PP_CompletionCallback callback) {
|
| - if (!buffer || bytes_to_read <= 0)
|
| - return PP_ERROR_BADARGUMENT;
|
| - if (!callback.func)
|
| - return PP_ERROR_BLOCKS_MAIN_THREAD;
|
| -
|
| - if (!IsConnected())
|
| - return PP_ERROR_FAILED;
|
| - if (read_callback_.func || ssl_handshake_callback_.func)
|
| - return PP_ERROR_INPROGRESS;
|
| -
|
| - read_buffer_ = buffer;
|
| - bytes_to_read_ = std::min(bytes_to_read, kTCPSocketMaxReadSize);
|
| - read_callback_ = callback;
|
| -
|
| - // Send the request, the browser will call us back via ReadACK.
|
| - GetDispatcher()->SendToBrowser(
|
| - new PpapiHostMsg_PPBTCPSocket_Read(socket_id_, bytes_to_read_));
|
| - return PP_OK_COMPLETIONPENDING;
|
| +void TCPSocket::SendRead(int32_t bytes_to_read) {
|
| + SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Read(socket_id_, bytes_to_read));
|
| }
|
|
|
| -int32_t TCPSocket::Write(const char* buffer,
|
| - int32_t bytes_to_write,
|
| - PP_CompletionCallback callback) {
|
| - if (!buffer || bytes_to_write <= 0)
|
| - return PP_ERROR_BADARGUMENT;
|
| - if (!callback.func)
|
| - return PP_ERROR_BLOCKS_MAIN_THREAD;
|
| -
|
| - if (!IsConnected())
|
| - return PP_ERROR_FAILED;
|
| - if (write_callback_.func || ssl_handshake_callback_.func)
|
| - return PP_ERROR_INPROGRESS;
|
| -
|
| - if (bytes_to_write > kTCPSocketMaxWriteSize)
|
| - bytes_to_write = kTCPSocketMaxWriteSize;
|
| -
|
| - write_callback_ = callback;
|
| -
|
| - // Send the request, the browser will call us back via WriteACK.
|
| - GetDispatcher()->SendToBrowser(
|
| - new PpapiHostMsg_PPBTCPSocket_Write(
|
| - socket_id_, std::string(buffer, bytes_to_write)));
|
| - return PP_OK_COMPLETIONPENDING;
|
| +void TCPSocket::SendWrite(const std::string& buffer) {
|
| + SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Write(socket_id_, buffer));
|
| }
|
|
|
| -void TCPSocket::Disconnect() {
|
| - if (connection_state_ == DISCONNECTED)
|
| - return;
|
| -
|
| - connection_state_ = DISCONNECTED;
|
| +void TCPSocket::SendDisconnect() {
|
| // After removed from the mapping, this object won't receive any notifications
|
| // from the proxy.
|
| DCHECK(g_id_to_socket->find(socket_id_) != g_id_to_socket->end());
|
| g_id_to_socket->erase(socket_id_);
|
| -
|
| - GetDispatcher()->SendToBrowser(
|
| - new PpapiHostMsg_PPBTCPSocket_Disconnect(socket_id_));
|
| - socket_id_ = 0;
|
| -
|
| - PostAbortAndClearIfNecessary(&connect_callback_);
|
| - PostAbortAndClearIfNecessary(&ssl_handshake_callback_);
|
| - PostAbortAndClearIfNecessary(&read_callback_);
|
| - PostAbortAndClearIfNecessary(&write_callback_);
|
| - read_buffer_ = NULL;
|
| - bytes_to_read_ = -1;
|
| -}
|
| -
|
| -void TCPSocket::OnConnectCompleted(
|
| - bool succeeded,
|
| - const PP_NetAddress_Private& local_addr,
|
| - const PP_NetAddress_Private& remote_addr) {
|
| - if (connection_state_ != BEFORE_CONNECT || !connect_callback_.func) {
|
| - NOTREACHED();
|
| - return;
|
| - }
|
| -
|
| - if (succeeded) {
|
| - local_addr_ = local_addr;
|
| - remote_addr_ = remote_addr;
|
| - connection_state_ = CONNECTED;
|
| - }
|
| - PP_RunAndClearCompletionCallback(&connect_callback_,
|
| - succeeded ? PP_OK : PP_ERROR_FAILED);
|
| -}
|
| -
|
| -void TCPSocket::OnSSLHandshakeCompleted(bool succeeded) {
|
| - if (connection_state_ != CONNECTED || !ssl_handshake_callback_.func) {
|
| - NOTREACHED();
|
| - return;
|
| - }
|
| -
|
| - if (succeeded) {
|
| - connection_state_ = SSL_CONNECTED;
|
| - PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_OK);
|
| - } else {
|
| - PP_RunAndClearCompletionCallback(&ssl_handshake_callback_, PP_ERROR_FAILED);
|
| - Disconnect();
|
| - }
|
| + SendToBrowser(new PpapiHostMsg_PPBTCPSocket_Disconnect(socket_id_));
|
| }
|
|
|
| -void TCPSocket::OnReadCompleted(bool succeeded, const std::string& data) {
|
| - if (!read_callback_.func || !read_buffer_) {
|
| - NOTREACHED();
|
| - return;
|
| - }
|
| -
|
| - if (succeeded) {
|
| - CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_);
|
| - if (!data.empty())
|
| - memcpy(read_buffer_, data.c_str(), data.size());
|
| - }
|
| - read_buffer_ = NULL;
|
| - bytes_to_read_ = -1;
|
| -
|
| - PP_RunAndClearCompletionCallback(
|
| - &read_callback_,
|
| - succeeded ? static_cast<int32_t>(data.size()) :
|
| - static_cast<int32_t>(PP_ERROR_FAILED));
|
| -}
|
| -
|
| -void TCPSocket::OnWriteCompleted(bool succeeded, int32_t bytes_written) {
|
| - if (!write_callback_.func || (succeeded && bytes_written < 0)) {
|
| - NOTREACHED();
|
| - return;
|
| - }
|
| -
|
| - PP_RunAndClearCompletionCallback(
|
| - &write_callback_,
|
| - succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED));
|
| -}
|
| -
|
| -bool TCPSocket::IsConnected() const {
|
| - return connection_state_ == CONNECTED || connection_state_ == SSL_CONNECTED;
|
| -}
|
| -
|
| -int32_t TCPSocket::ConnectWithMessage(IPC::Message* msg,
|
| - PP_CompletionCallback callback) {
|
| - scoped_ptr<IPC::Message> msg_deletor(msg);
|
| - if (!callback.func)
|
| - return PP_ERROR_BLOCKS_MAIN_THREAD;
|
| - if (connection_state_ != BEFORE_CONNECT)
|
| - return PP_ERROR_FAILED;
|
| - if (connect_callback_.func)
|
| - return PP_ERROR_INPROGRESS; // Can only have one pending request.
|
| -
|
| - connect_callback_ = callback;
|
| - // Send the request, the browser will call us back via ConnectACK.
|
| - GetDispatcher()->SendToBrowser(msg_deletor.release());
|
| - return PP_OK_COMPLETIONPENDING;
|
| +void TCPSocket::SendToBrowser(IPC::Message* msg) {
|
| + PluginDispatcher::GetForResource(this)->SendToBrowser(msg);
|
| }
|
|
|
| -void TCPSocket::PostAbortAndClearIfNecessary(
|
| - PP_CompletionCallback* callback) {
|
| - DCHECK(callback);
|
| +} // namespace
|
|
|
| - if (callback->func) {
|
| - MessageLoop::current()->PostTask(
|
| - FROM_HERE, base::Bind(&AbortCallback, *callback));
|
| - *callback = PP_BlockUntilComplete();
|
| - }
|
| -}
|
| +//------------------------------------------------------------------------------
|
|
|
| PPB_TCPSocket_Private_Proxy::PPB_TCPSocket_Private_Proxy(Dispatcher* dispatcher)
|
| : InterfaceProxy(dispatcher) {
|
|
|