| Index: ppapi/shared_impl/private/tcp_socket_private_impl.cc
|
| diff --git a/ppapi/shared_impl/private/tcp_socket_private_impl.cc b/ppapi/shared_impl/private/tcp_socket_private_impl.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..7ec1535907a7f76022dd1d3dd6af8a732b57d84f
|
| --- /dev/null
|
| +++ b/ppapi/shared_impl/private/tcp_socket_private_impl.cc
|
| @@ -0,0 +1,289 @@
|
| +// Copyright (c) 2011 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 "ppapi/shared_impl/private/tcp_socket_private_impl.h"
|
| +
|
| +#include <string.h>
|
| +
|
| +#include <algorithm>
|
| +
|
| +#include "base/basictypes.h"
|
| +#include "base/bind.h"
|
| +#include "base/logging.h"
|
| +#include "base/message_loop.h"
|
| +#include "ppapi/c/pp_completion_callback.h"
|
| +#include "ppapi/c/pp_errors.h"
|
| +
|
| +namespace ppapi {
|
| +
|
| +namespace {
|
| +
|
| +void AbortCallback(PP_CompletionCallback callback) {
|
| + PP_RunCompletionCallback(&callback, PP_ERROR_ABORTED);
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| +const int32_t TCPSocketPrivateImpl::kMaxReadSize = 1024 * 1024;
|
| +const int32_t TCPSocketPrivateImpl::kMaxWriteSize = 1024 * 1024;
|
| +
|
| +TCPSocketPrivateImpl::TCPSocketPrivateImpl(PP_Instance instance,
|
| + uint32 socket_id)
|
| + : Resource(instance) {
|
| + Init(socket_id);
|
| +}
|
| +
|
| +TCPSocketPrivateImpl::TCPSocketPrivateImpl(const HostResource& resource,
|
| + uint32 socket_id)
|
| + : Resource(resource) {
|
| + Init(socket_id);
|
| +}
|
| +
|
| +TCPSocketPrivateImpl::~TCPSocketPrivateImpl() {
|
| +}
|
| +
|
| +thunk::PPB_TCPSocket_Private_API*
|
| +TCPSocketPrivateImpl::AsPPB_TCPSocket_Private_API() {
|
| + return this;
|
| +}
|
| +
|
| +int32_t TCPSocketPrivateImpl::Connect(const char* host,
|
| + uint16_t port,
|
| + PP_CompletionCallback callback) {
|
| + if (!host)
|
| + return PP_ERROR_BADARGUMENT;
|
| + 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.
|
| + SendConnect(host, port);
|
| + return PP_OK_COMPLETIONPENDING;
|
| +}
|
| +
|
| +int32_t TCPSocketPrivateImpl::ConnectWithNetAddress(
|
| + const PP_NetAddress_Private* addr,
|
| + PP_CompletionCallback callback) {
|
| + if (!addr)
|
| + return PP_ERROR_BADARGUMENT;
|
| + 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.
|
| + SendConnectWithNetAddress(*addr);
|
| + return PP_OK_COMPLETIONPENDING;
|
| +}
|
| +
|
| +PP_Bool TCPSocketPrivateImpl::GetLocalAddress(
|
| + PP_NetAddress_Private* local_addr) {
|
| + if (!IsConnected() || !local_addr)
|
| + return PP_FALSE;
|
| +
|
| + *local_addr = local_addr_;
|
| + return PP_TRUE;
|
| +}
|
| +
|
| +PP_Bool TCPSocketPrivateImpl::GetRemoteAddress(
|
| + PP_NetAddress_Private* remote_addr) {
|
| + if (!IsConnected() || !remote_addr)
|
| + return PP_FALSE;
|
| +
|
| + *remote_addr = remote_addr_;
|
| + return PP_TRUE;
|
| +}
|
| +
|
| +int32_t TCPSocketPrivateImpl::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.
|
| + SendSSLHandshake(server_name, server_port);
|
| + return PP_OK_COMPLETIONPENDING;
|
| +}
|
| +
|
| +int32_t TCPSocketPrivateImpl::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;
|
| + // TODO(dmichael): use some other strategy for determining if an
|
| + // operation is in progress
|
| + read_buffer_ = buffer;
|
| + bytes_to_read_ = std::min(bytes_to_read, kMaxReadSize);
|
| + read_callback_ = callback;
|
| +
|
| + // Send the request, the browser will call us back via ReadACK.
|
| + SendRead(bytes_to_read_);
|
| + return PP_OK_COMPLETIONPENDING;
|
| +}
|
| +
|
| +int32_t TCPSocketPrivateImpl::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 > kMaxWriteSize)
|
| + bytes_to_write = kMaxWriteSize;
|
| +
|
| + write_callback_ = callback;
|
| +
|
| + // Send the request, the browser will call us back via WriteACK.
|
| + SendWrite(std::string(buffer, bytes_to_write));
|
| + return PP_OK_COMPLETIONPENDING;
|
| +}
|
| +
|
| +void TCPSocketPrivateImpl::Disconnect() {
|
| + if (connection_state_ == DISCONNECTED)
|
| + return;
|
| +
|
| + connection_state_ = DISCONNECTED;
|
| +
|
| + SendDisconnect();
|
| + socket_id_ = 0;
|
| +
|
| + PostAbortAndClearIfNecessary(&connect_callback_);
|
| + PostAbortAndClearIfNecessary(&ssl_handshake_callback_);
|
| + PostAbortAndClearIfNecessary(&read_callback_);
|
| + PostAbortAndClearIfNecessary(&write_callback_);
|
| + read_buffer_ = NULL;
|
| + bytes_to_read_ = -1;
|
| +}
|
| +
|
| +void TCPSocketPrivateImpl::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 TCPSocketPrivateImpl::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();
|
| + }
|
| +}
|
| +
|
| +void TCPSocketPrivateImpl::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 TCPSocketPrivateImpl::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));
|
| +}
|
| +
|
| +void TCPSocketPrivateImpl::Init(uint32 socket_id) {
|
| + DCHECK(socket_id != 0);
|
| + 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;
|
| +
|
| + local_addr_.size = 0;
|
| + memset(local_addr_.data, 0,
|
| + arraysize(local_addr_.data) * sizeof(*local_addr_.data));
|
| + remote_addr_.size = 0;
|
| + memset(remote_addr_.data, 0,
|
| + arraysize(remote_addr_.data) * sizeof(*remote_addr_.data));
|
| +}
|
| +
|
| +bool TCPSocketPrivateImpl::IsConnected() const {
|
| + return connection_state_ == CONNECTED || connection_state_ == SSL_CONNECTED;
|
| +}
|
| +
|
| +void TCPSocketPrivateImpl::PostAbortAndClearIfNecessary(
|
| + PP_CompletionCallback* callback) {
|
| + DCHECK(callback);
|
| +
|
| + if (callback->func) {
|
| + MessageLoop::current()->PostTask(FROM_HERE,
|
| + base::Bind(&AbortCallback, *callback));
|
| + *callback = PP_BlockUntilComplete();
|
| + }
|
| +}
|
| +
|
| +} // namespace ppapi
|
|
|