Chromium Code Reviews| Index: device/bluetooth/bluetooth_socket_net.cc |
| diff --git a/device/bluetooth/bluetooth_socket_net.cc b/device/bluetooth/bluetooth_socket_net.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..d1cc933e89eb36f5545342400696dadab6b6a845 |
| --- /dev/null |
| +++ b/device/bluetooth/bluetooth_socket_net.cc |
| @@ -0,0 +1,331 @@ |
| +// Copyright 2014 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 "device/bluetooth/bluetooth_socket_net.h" |
| + |
| +#include <string> |
| + |
| +#include "base/logging.h" |
| +#include "base/memory/ref_counted.h" |
| +#include "base/sequenced_task_runner.h" |
| +#include "base/threading/thread_restrictions.h" |
| +#include "device/bluetooth/bluetooth_socket.h" |
| +#include "device/bluetooth/bluetooth_socket_thread.h" |
| +#include "net/base/io_buffer.h" |
| +#include "net/base/net_errors.h" |
| + |
| +namespace { |
| + |
| +const char kSocketNotConnected[] = "Socket is not connected."; |
| + |
| +static void DeactivateSocket( |
| + const scoped_refptr<device::BluetoothSocketThread>& socket_thread) { |
| + socket_thread->OnSocketDeactivate(); |
| +} |
| + |
| +} // namespace |
| + |
| +namespace device { |
| + |
| +// static |
| +scoped_refptr<BluetoothSocketNet> |
| +BluetoothSocketNet::CreateBluetoothSocket( |
| + scoped_refptr<base::SequencedTaskRunner> ui_task_runner, |
| + scoped_refptr<BluetoothSocketThread> socket_thread, |
| + net::NetLog* net_log, |
| + const net::NetLog::Source& source) { |
| + DCHECK(ui_task_runner->RunsTasksOnCurrentThread()); |
| + |
| + return make_scoped_refptr( |
| + new BluetoothSocketNet(ui_task_runner, socket_thread, net_log, source)); |
| +} |
| + |
| +BluetoothSocketNet::WriteRequest::WriteRequest() |
| + : buffer_size(0) {} |
| + |
| +BluetoothSocketNet::WriteRequest::~WriteRequest() {} |
| + |
| +BluetoothSocketNet::BluetoothSocketNet( |
| + scoped_refptr<base::SequencedTaskRunner> ui_task_runner, |
| + scoped_refptr<BluetoothSocketThread> socket_thread, |
| + net::NetLog* net_log, |
| + const net::NetLog::Source& source) |
| + : ui_task_runner_(ui_task_runner), |
| + socket_thread_(socket_thread), |
| + net_log_(net_log), |
| + source_(source) { |
| + DCHECK(ui_task_runner->RunsTasksOnCurrentThread()); |
| + socket_thread_->OnSocketActivate(); |
| +} |
| + |
| +BluetoothSocketNet::~BluetoothSocketNet() { |
| + DCHECK(tcp_socket_.get() == NULL); |
| + ui_task_runner_->PostTask(FROM_HERE, |
| + base::Bind(&DeactivateSocket, socket_thread_)); |
| +} |
| + |
| +void BluetoothSocketNet::Close() { |
| + DCHECK(ui_task_runner_->RunsTasksOnCurrentThread()); |
| + socket_thread_->task_runner()->PostTask( |
| + FROM_HERE, base::Bind(&BluetoothSocketNet::DoClose, this)); |
| +} |
| + |
| +void BluetoothSocketNet::Disconnect( |
| + const base::Closure& success_callback) { |
| + DCHECK(ui_task_runner_->RunsTasksOnCurrentThread()); |
| + socket_thread_->task_runner()->PostTask( |
| + FROM_HERE, |
| + base::Bind( |
| + &BluetoothSocketNet::DoDisconnect, |
| + this, |
| + base::Bind(&BluetoothSocketNet::PostSuccess, |
| + this, |
| + success_callback))); |
| +} |
| + |
| +void BluetoothSocketNet::Receive( |
| + int buffer_size, |
| + const ReceiveCompletionCallback& success_callback, |
| + const ReceiveErrorCompletionCallback& error_callback) { |
| + DCHECK(ui_task_runner_->RunsTasksOnCurrentThread()); |
| + socket_thread_->task_runner()->PostTask( |
| + FROM_HERE, |
| + base::Bind( |
| + &BluetoothSocketNet::DoReceive, |
| + this, |
| + buffer_size, |
| + base::Bind(&BluetoothSocketNet::PostReceiveCompletion, |
| + this, |
| + success_callback), |
| + base::Bind(&BluetoothSocketNet::PostReceiveErrorCompletion, |
| + this, |
| + error_callback))); |
| +} |
| + |
| +void BluetoothSocketNet::Send( |
| + scoped_refptr<net::IOBuffer> buffer, |
| + int buffer_size, |
| + const SendCompletionCallback& success_callback, |
| + const ErrorCompletionCallback& error_callback) { |
| + DCHECK(ui_task_runner_->RunsTasksOnCurrentThread()); |
| + socket_thread_->task_runner()->PostTask( |
| + FROM_HERE, |
| + base::Bind( |
| + &BluetoothSocketNet::DoSend, |
| + this, |
| + buffer, |
| + buffer_size, |
| + base::Bind(&BluetoothSocketNet::PostSendCompletion, |
| + this, |
| + success_callback), |
| + base::Bind(&BluetoothSocketNet::PostErrorCompletion, |
| + this, |
| + error_callback))); |
| +} |
| + |
| +void BluetoothSocketNet::ResetData() { |
|
armansito
2014/05/01 00:14:25
I would at least log something here, maybe VLOG(1)
keybuk
2014/05/01 00:24:09
NAK, this is just for subclasses - Win happens to
armansito
2014/05/01 00:54:17
sgtm
|
| +} |
| + |
| +void BluetoothSocketNet::ResetTCPSocket() { |
| + tcp_socket_.reset(new net::TCPSocket(net_log_, source_)); |
| +} |
| + |
| +void BluetoothSocketNet::SetTCPSocket(scoped_ptr<net::TCPSocket> tcp_socket) { |
| + tcp_socket_ = tcp_socket.Pass(); |
| +} |
| + |
| +void BluetoothSocketNet::PostSuccess(const base::Closure& callback) { |
| + ui_task_runner_->PostTask(FROM_HERE, callback); |
| +} |
| + |
| +void BluetoothSocketNet::PostErrorCompletion( |
| + const ErrorCompletionCallback& callback, |
| + const std::string& error) { |
| + ui_task_runner_->PostTask(FROM_HERE, base::Bind(callback, error)); |
| +} |
| + |
| +void BluetoothSocketNet::DoClose() { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + if (tcp_socket_) { |
| + tcp_socket_->Close(); |
| + tcp_socket_.reset(NULL); |
| + } |
| + |
| + // Note: Closing |tcp_socket_| above released all potential pending |
| + // Send/Receive operations, so we can no safely release the state associated |
| + // to those pending operations. |
| + read_buffer_ = NULL; |
| + std::queue<linked_ptr<WriteRequest> > empty; |
| + write_queue_.swap(empty); |
| + |
| + ResetData(); |
| +} |
| + |
| +void BluetoothSocketNet::DoDisconnect( |
| + const base::Closure& success_callback) { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + DoClose(); |
| + success_callback.Run(); |
| +} |
| + |
| +void BluetoothSocketNet::DoReceive( |
| + int buffer_size, |
| + const ReceiveCompletionCallback& success_callback, |
| + const ReceiveErrorCompletionCallback& error_callback) { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + if (!tcp_socket_) { |
| + error_callback.Run(BluetoothSocket::kDisconnected, kSocketNotConnected); |
| + return; |
| + } |
| + |
| + // Only one pending read at a time |
| + if (read_buffer_.get()) { |
| + error_callback.Run(BluetoothSocket::kIOPending, |
| + net::ErrorToString(net::ERR_IO_PENDING)); |
| + return; |
| + } |
| + |
| + scoped_refptr<net::IOBufferWithSize> buffer( |
| + new net::IOBufferWithSize(buffer_size)); |
| + int read_result = |
| + tcp_socket_->Read(buffer.get(), |
| + buffer->size(), |
| + base::Bind(&BluetoothSocketNet::OnSocketReadComplete, |
| + this, |
| + success_callback, |
| + error_callback)); |
| + |
| + if (read_result > 0) { |
| + success_callback.Run(read_result, buffer); |
| + } else if (read_result == net::OK || |
| + read_result == net::ERR_CONNECTION_CLOSED) { |
| + error_callback.Run(BluetoothSocket::kDisconnected, |
| + net::ErrorToString(net::ERR_CONNECTION_CLOSED)); |
| + } else if (read_result == net::ERR_IO_PENDING) { |
| + read_buffer_ = buffer; |
| + } else { |
| + error_callback.Run(BluetoothSocket::kSystemError, |
| + net::ErrorToString(read_result)); |
| + } |
| +} |
| + |
| +void BluetoothSocketNet::OnSocketReadComplete( |
| + const ReceiveCompletionCallback& success_callback, |
| + const ReceiveErrorCompletionCallback& error_callback, |
| + int read_result) { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + scoped_refptr<net::IOBufferWithSize> buffer; |
| + buffer.swap(read_buffer_); |
| + if (read_result > 0) { |
| + success_callback.Run(read_result, buffer); |
| + } else if (read_result == net::OK || |
| + read_result == net::ERR_CONNECTION_CLOSED) { |
| + error_callback.Run(BluetoothSocket::kDisconnected, |
| + net::ErrorToString(net::ERR_CONNECTION_CLOSED)); |
| + } else { |
| + error_callback.Run(BluetoothSocket::kSystemError, |
| + net::ErrorToString(read_result)); |
| + } |
| +} |
| + |
| +void BluetoothSocketNet::DoSend( |
| + scoped_refptr<net::IOBuffer> buffer, |
| + int buffer_size, |
| + const SendCompletionCallback& success_callback, |
| + const ErrorCompletionCallback& error_callback) { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + if (!tcp_socket_) { |
| + error_callback.Run(kSocketNotConnected); |
| + return; |
| + } |
| + |
| + linked_ptr<WriteRequest> request(new WriteRequest()); |
| + request->buffer = buffer; |
| + request->buffer_size = buffer_size; |
| + request->success_callback = success_callback; |
| + request->error_callback = error_callback; |
| + |
| + write_queue_.push(request); |
| + if (write_queue_.size() == 1) { |
| + SendFrontWriteRequest(); |
| + } |
| +} |
| + |
| +void BluetoothSocketNet::SendFrontWriteRequest() { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + if (!tcp_socket_) |
| + return; |
| + |
| + if (write_queue_.size() == 0) |
| + return; |
| + |
| + linked_ptr<WriteRequest> request = write_queue_.front(); |
| + net::CompletionCallback callback = |
| + base::Bind(&BluetoothSocketNet::OnSocketWriteComplete, |
| + this, |
| + request->success_callback, |
| + request->error_callback); |
| + int send_result = |
| + tcp_socket_->Write(request->buffer, request->buffer_size, callback); |
| + if (send_result != net::ERR_IO_PENDING) { |
| + callback.Run(send_result); |
| + } |
| +} |
| + |
| +void BluetoothSocketNet::OnSocketWriteComplete( |
| + const SendCompletionCallback& success_callback, |
| + const ErrorCompletionCallback& error_callback, |
| + int send_result) { |
| + DCHECK(socket_thread_->task_runner()->RunsTasksOnCurrentThread()); |
| + base::ThreadRestrictions::AssertIOAllowed(); |
| + |
| + write_queue_.pop(); |
| + |
| + if (send_result >= net::OK) { |
| + success_callback.Run(send_result); |
| + } else { |
| + error_callback.Run(net::ErrorToString(send_result)); |
| + } |
| + |
| + // Don't call directly to avoid potentail large recursion. |
| + socket_thread_->task_runner()->PostNonNestableTask( |
| + FROM_HERE, |
| + base::Bind(&BluetoothSocketNet::SendFrontWriteRequest, this)); |
| +} |
| + |
| +void BluetoothSocketNet::PostReceiveCompletion( |
| + const ReceiveCompletionCallback& callback, |
| + int io_buffer_size, |
| + scoped_refptr<net::IOBuffer> io_buffer) { |
| + ui_task_runner_->PostTask(FROM_HERE, |
| + base::Bind(callback, io_buffer_size, io_buffer)); |
| +} |
| + |
| +void BluetoothSocketNet::PostReceiveErrorCompletion( |
| + const ReceiveErrorCompletionCallback& callback, |
| + ErrorReason reason, |
| + const std::string& error_message) { |
| + ui_task_runner_->PostTask(FROM_HERE, |
| + base::Bind(callback, reason, error_message)); |
| +} |
| + |
| +void BluetoothSocketNet::PostSendCompletion( |
| + const SendCompletionCallback& callback, |
| + int bytes_written) { |
| + ui_task_runner_->PostTask(FROM_HERE, base::Bind(callback, bytes_written)); |
| +} |
| + |
| +} // namespace device |