#include "device/bluetooth/bluetooth_socket_win.h"
+#include <objbase.h>
+
#include <string>
#include "base/logging.h"
#include "base/memory/ref_counted.h"
+#include "base/sequenced_task_runner.h"
#include "base/strings/sys_string_conversions.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/threading/thread_restrictions.h"
#include "device/bluetooth/bluetooth_init_win.h"
#include "device/bluetooth/bluetooth_service_record_win.h"
+#include "device/bluetooth/bluetooth_socket_thread.h"
#include "net/base/io_buffer.h"
+#include "net/base/ip_endpoint.h"
+#include "net/base/net_errors.h"
#include "net/base/winsock_init.h"
namespace {
-std::string FormatErrorMessage(DWORD error_code) {
- TCHAR error_msg[1024];
- FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
- 0,
- error_code,
- 0,
- error_msg,
- 1024,
- NULL);
- return base::SysWideToUTF8(error_msg);
-}
+const char kL2CAPNotSupported[] = "Bluetooth L2CAP protocal is not supported";
+const char kSocketAlreadyConnected[] = "Socket is already connected.";
+const char kInvalidRfcommPort[] = "Invalid RFCCOMM port.";
+const char kFailedToCreateSocket[] = "Failed to create socket.";
+const char kFailedToBindSocket[] = "Failed to bind socket.";
+const char kFailedToListenOnSocket[] = "Failed to listen on socket.";
+const char kFailedToGetSockNameForSocket[] = "Failed to getsockname.";
+const char kBadUuid[] = "Bad uuid.";
+const char kWsaSetServiceError[] = "WSASetService error.";
} // namespace
namespace device {
-BluetoothSocketWin::BluetoothSocketWin(SOCKET fd) : fd_(fd) {
+struct BluetoothSocketWin::ServiceRegData {
+ ServiceRegData() {
+ ZeroMemory(&address, sizeof(address));
+ ZeroMemory(&address_info, sizeof(address_info));
+ ZeroMemory(&uuid, sizeof(uuid));
+ ZeroMemory(&service, sizeof(service));
+ }
+
+ SOCKADDR_BTH address;
+ CSADDR_INFO address_info;
+ GUID uuid;
+ base::string16 name;
+ WSAQUERYSET service;
+};
+
+// static
+scoped_refptr<BluetoothSocketWin>
+BluetoothSocketWin::CreateBluetoothSocket(
+ scoped_refptr<base::SequencedTaskRunner> ui_task_runner,
+ scoped_refptr<device::BluetoothSocketThread> socket_thread,
+ net::NetLog* net_log,
+ const net::NetLog::Source& source) {
+ DCHECK(ui_task_runner->RunsTasksOnCurrentThread());
+
+ return make_scoped_refptr(
+ new BluetoothSocketWin(ui_task_runner, socket_thread, net_log, source));
+}
+
+BluetoothSocketWin::BluetoothSocketWin(
+ scoped_refptr<base::SequencedTaskRunner> ui_task_runner,
+ scoped_refptr<BluetoothSocketThread> socket_thread,
+ net::NetLog* net_log,
+ const net::NetLog::Source& source)
+ : BluetoothSocketNet(ui_task_runner, socket_thread, net_log, source),
+ supports_rfcomm_(false),
+ rfcomm_channel_(-1),
+ bth_addr_(BTH_ADDR_NULL) {
}
BluetoothSocketWin::~BluetoothSocketWin() {
- closesocket(fd_);
}
-// static
-scoped_refptr<BluetoothSocket> BluetoothSocketWin::CreateBluetoothSocket(
- const BluetoothServiceRecord& service_record) {
- BluetoothSocketWin* bluetooth_socket = NULL;
+void BluetoothSocketWin::StartService(
+ const BluetoothUUID& uuid,
+ const std::string& name,
+ int rfcomm_channel,
+ const base::Closure& success_callback,
+ const ErrorCompletionCallback& error_callback,
+ const OnNewConnectionCallback& new_connection_callback) {
+ DCHECK(ui_task_runner()->RunsTasksOnCurrentThread());
+
+ socket_thread()->task_runner()->PostTask(
+ FROM_HERE,
+ base::Bind(&BluetoothSocketWin::DoStartService,
+ this,
+ uuid,
+ name,
+ rfcomm_channel,
+ success_callback,
+ error_callback,
+ new_connection_callback));
+}
+
+void BluetoothSocketWin::Connect(
+ const BluetoothServiceRecord& service_record,
+ const base::Closure& success_callback,
+ const ErrorCompletionCallback& error_callback) {
+ DCHECK(ui_task_runner()->RunsTasksOnCurrentThread());
+
+ const BluetoothServiceRecordWin* service_record_win =
+ static_cast<const BluetoothServiceRecordWin*>(&service_record);
+ device_address_ = service_record_win->address();
if (service_record.SupportsRfcomm()) {
- net::EnsureWinsockInit();
- SOCKET socket_fd = socket(AF_BTH, SOCK_STREAM, BTHPROTO_RFCOMM);
- SOCKADDR_BTH sa;
- ZeroMemory(&sa, sizeof(sa));
- sa.addressFamily = AF_BTH;
- sa.port = service_record.rfcomm_channel();
- const BluetoothServiceRecordWin* service_record_win =
- static_cast<const BluetoothServiceRecordWin*>(&service_record);
- sa.btAddr = service_record_win->bth_addr();
-
- int status = connect(socket_fd,
- reinterpret_cast<SOCKADDR *>(&sa),
- sizeof(sa));
- DWORD error_code = WSAGetLastError();
- if (status == 0 || error_code == WSAEINPROGRESS) {
- bluetooth_socket =
- new BluetoothSocketWin(socket_fd);
- } else {
- LOG(ERROR) << "Failed to connect bluetooth socket "
- << "(" << service_record.address() << "): "
- << "(" << error_code << ")" << FormatErrorMessage(error_code);
- closesocket(socket_fd);
- }
+ supports_rfcomm_ = true;
+ rfcomm_channel_ = service_record_win->rfcomm_channel();
+ bth_addr_ = service_record_win->bth_addr();
}
- // TODO(youngki) add support for L2CAP sockets as well.
- return scoped_refptr<BluetoothSocketWin>(bluetooth_socket);
+ socket_thread()->task_runner()->PostTask(
+ FROM_HERE,
+ base::Bind(
+ &BluetoothSocketWin::DoConnect,
+ this,
+ base::Bind(&BluetoothSocketWin::PostSuccess, this, success_callback),
+ base::Bind(
+ &BluetoothSocketWin::PostErrorCompletion, this, error_callback)));
}
-bool BluetoothSocketWin::Receive(net::GrowableIOBuffer* buffer) {
- buffer->SetCapacity(1024);
- int bytes_read;
- do {
- if (buffer->RemainingCapacity() == 0)
- buffer->SetCapacity(buffer->capacity() * 2);
- bytes_read = recv(fd_, buffer->data(), buffer->RemainingCapacity(), 0);
- if (bytes_read > 0)
- buffer->set_offset(buffer->offset() + bytes_read);
- } while (bytes_read > 0);
- DWORD error_code = WSAGetLastError();
- if (bytes_read < 0 && error_code != WSAEWOULDBLOCK) {
- error_message_ = FormatErrorMessage(error_code);
- return false;
+void BluetoothSocketWin::ResetData() {
+ if (service_reg_data_) {
+ if (WSASetService(&service_reg_data_->service,RNRSERVICE_DELETE, 0) ==
+ SOCKET_ERROR) {
+ LOG(WARNING) << "Failed to unregister service.";
+ }
+ service_reg_data_.reset();
}
- return true;
}
-bool BluetoothSocketWin::Send(net::DrainableIOBuffer* buffer) {
- int bytes_written;
- do {
- bytes_written = send(fd_, buffer->data(), buffer->BytesRemaining(), 0);
- if (bytes_written > 0)
- buffer->DidConsume(bytes_written);
- } while (buffer->BytesRemaining() > 0 && bytes_written > 0);
+void BluetoothSocketWin::DoConnect(
+ const base::Closure& success_callback,
+ const ErrorCompletionCallback& error_callback) {
+ DCHECK(socket_thread()->task_runner()->RunsTasksOnCurrentThread());
+ base::ThreadRestrictions::AssertIOAllowed();
+
+ if (tcp_socket()) {
+ error_callback.Run(kSocketAlreadyConnected);
+ return;
+ }
+
+ if (!supports_rfcomm_) {
+ // TODO(youngki) add support for L2CAP sockets as well.
+ error_callback.Run(kL2CAPNotSupported);
+ return;
+ }
+
+ ResetTCPSocket();
+ net::EnsureWinsockInit();
+ SOCKET socket_fd = socket(AF_BTH, SOCK_STREAM, BTHPROTO_RFCOMM);
+ SOCKADDR_BTH sa;
+ ZeroMemory(&sa, sizeof(sa));
+ sa.addressFamily = AF_BTH;
+ sa.port = rfcomm_channel_;
+ sa.btAddr = bth_addr_;
+ // TODO(rpaquay): Condider making this call non-blocking.
+ int status = connect(socket_fd, reinterpret_cast<SOCKADDR*>(&sa), sizeof(sa));
DWORD error_code = WSAGetLastError();
- if (bytes_written < 0 && error_code != WSAEWOULDBLOCK) {
- error_message_ = FormatErrorMessage(error_code);
- return false;
+ if (!(status == 0 || error_code == WSAEINPROGRESS)) {
+ LOG(ERROR) << "Failed to connect bluetooth socket "
+ << "(" << device_address_ << "): "
+ << logging::SystemErrorCodeToString(error_code);
+ error_callback.Run("Error connecting to socket: " +
+ logging::SystemErrorCodeToString(error_code));
+ closesocket(socket_fd);
+ return;
+ }
+
+ // Note: We don't have a meaningful |IPEndPoint|, but that is ok since the
+ // TCPSocket implementation does not actually require one.
+ int net_result =
+ tcp_socket()->AdoptConnectedSocket(socket_fd, net::IPEndPoint());
+ if (net_result != net::OK) {
+ error_callback.Run("Error connecting to socket: " +
+ std::string(net::ErrorToString(net_result)));
+ closesocket(socket_fd);
+ return;
}
- return true;
+
+ success_callback.Run();
}
-std::string BluetoothSocketWin::GetLastErrorMessage() const {
- return error_message_;
+void BluetoothSocketWin::DoStartService(
+ const BluetoothUUID& uuid,
+ const std::string& name,
+ int rfcomm_channel,
+ const base::Closure& success_callback,
+ const ErrorCompletionCallback& error_callback,
+ const OnNewConnectionCallback& new_connection_callback) {
+ DCHECK(socket_thread()->task_runner()->RunsTasksOnCurrentThread());
+ DCHECK(!tcp_socket() &&
+ !service_reg_data_ &&
+ on_new_connection_callback_.is_null());
+
+ // The valid range is 0-30. 0 means BT_PORT_ANY and 1-30 are the
+ // valid RFCOMM port numbers of SOCKADDR_BTH.
+ if (rfcomm_channel < 0 || rfcomm_channel > 30) {
+ LOG(WARNING) << "Failed to start service: "
+ << "Invalid RFCCOMM port " << rfcomm_channel
+ << ", uuid=" << uuid.value();
+ PostErrorCompletion(error_callback, kInvalidRfcommPort);
+ return;
+ }
+
+ SOCKET socket_fd = socket(AF_BTH, SOCK_STREAM, BTHPROTO_RFCOMM);
+ if (socket_fd == INVALID_SOCKET) {
+ LOG(WARNING) << "Failed to start service: create socket, "
+ << "winsock err=" << WSAGetLastError();
+ PostErrorCompletion(error_callback, kFailedToCreateSocket);
+ return;
+ }
+
+ // Note that |socket_fd| belongs to a non-TCP address family (i.e. AF_BTH),
+ // TCPSocket methods that involve address could not be called. So bind()
+ // is called on |socket_fd| directly.
+ scoped_ptr<net::TCPSocket> scoped_socket(
+ new net::TCPSocket(NULL, net::NetLog::Source()));
+ scoped_socket->AdoptListenSocket(socket_fd);
+
+ SOCKADDR_BTH sa;
+ struct sockaddr* sock_addr = reinterpret_cast<struct sockaddr*>(&sa);
+ int sock_addr_len = sizeof(sa);
+ ZeroMemory(&sa, sock_addr_len);
+ sa.addressFamily = AF_BTH;
+ sa.port = rfcomm_channel ? rfcomm_channel : BT_PORT_ANY;
+ if (bind(socket_fd, sock_addr, sock_addr_len) < 0) {
+ LOG(WARNING) << "Failed to start service: create socket, "
+ << "winsock err=" << WSAGetLastError();
+ PostErrorCompletion(error_callback, kFailedToBindSocket);
+ return;
+ }
+
+ const int kListenBacklog = 5;
+ if (scoped_socket->Listen(kListenBacklog) < 0) {
+ LOG(WARNING) << "Failed to start service: Listen"
+ << "winsock err=" << WSAGetLastError();
+ PostErrorCompletion(error_callback, kFailedToListenOnSocket);
+ return;
+ }
+
+ scoped_ptr<ServiceRegData> reg_data(new ServiceRegData);
+ reg_data->name = base::UTF8ToUTF16(name);
+
+ if (getsockname(socket_fd, sock_addr, &sock_addr_len)) {
+ LOG(WARNING) << "Failed to start service: getsockname, "
+ << "winsock err=" << WSAGetLastError();
+ PostErrorCompletion(error_callback, kFailedToGetSockNameForSocket);
+ return;
+ }
+ reg_data->address = sa;
+
+ reg_data->address_info.LocalAddr.iSockaddrLength = sizeof(sa);
+ reg_data->address_info.LocalAddr.lpSockaddr =
+ reinterpret_cast<struct sockaddr*>(®_data->address);
+ reg_data->address_info.iSocketType = SOCK_STREAM;
+ reg_data->address_info.iProtocol = BTHPROTO_RFCOMM;
+
+ base::string16 cannonical_uuid = L"{" + base::ASCIIToUTF16(
+ uuid.canonical_value()) + L"}";
+ if (!SUCCEEDED(CLSIDFromString(cannonical_uuid.c_str(), ®_data->uuid))) {
+ LOG(WARNING) << "Failed to start service: "
+ << ", bad uuid=" << cannonical_uuid;
+ PostErrorCompletion(error_callback, kBadUuid);
+ return;
+ }
+
+ reg_data->service.dwSize = sizeof(WSAQUERYSET);
+ reg_data->service.lpszServiceInstanceName =
+ const_cast<LPWSTR>(reg_data->name.c_str());
+ reg_data->service.lpServiceClassId = ®_data->uuid;
+ reg_data->service.dwNameSpace = NS_BTH;
+ reg_data->service.dwNumberOfCsAddrs = 1;
+ reg_data->service.lpcsaBuffer = ®_data->address_info;
+
+ if (WSASetService(®_data->service,
+ RNRSERVICE_REGISTER, 0) == SOCKET_ERROR) {
+ LOG(WARNING) << "Failed to register profile: WSASetService"
+ << "winsock err=" << WSAGetLastError();
+ PostErrorCompletion(error_callback, kWsaSetServiceError);
+ return;
+ }
+
+ SetTCPSocket(scoped_socket.Pass());
+ service_reg_data_ = reg_data.Pass();
+ on_new_connection_callback_ = new_connection_callback;
+ DoAccept();
+
+ PostSuccess(success_callback);
+}
+
+void BluetoothSocketWin::DoAccept() {
+ DCHECK(socket_thread()->task_runner()->RunsTasksOnCurrentThread());
+ int result = tcp_socket()->Accept(
+ &accept_socket_,
+ &accept_address_,
+ base::Bind(&BluetoothSocketWin::OnAcceptOnSocketThread, this));
+ if (result != net::OK && result != net::ERR_IO_PENDING)
+ LOG(WARNING) << "Failed to accept, net err=" << result;
+}
+
+void BluetoothSocketWin::OnAcceptOnSocketThread(int accept_result) {
+ DCHECK(socket_thread()->task_runner()->RunsTasksOnCurrentThread());
+ if (accept_result != net::OK) {
+ LOG(WARNING) << "OnAccept error, net err=" << accept_result;
+ return;
+ }
+
+ ui_task_runner()->PostTask(
+ FROM_HERE,
+ base::Bind(&BluetoothSocketWin::OnAcceptOnUI,
+ this,
+ base::Passed(&accept_socket_),
+ accept_address_));
+ DoAccept();
+}
+
+void BluetoothSocketWin::OnAcceptOnUI(
+ scoped_ptr<net::TCPSocket> accept_socket,
+ const net::IPEndPoint& peer_address) {
+ DCHECK(ui_task_runner()->RunsTasksOnCurrentThread());
+
+ scoped_refptr<BluetoothSocketWin> peer = CreateBluetoothSocket(
+ ui_task_runner(),
+ socket_thread(),
+ net_log(),
+ source());
+ peer->SetTCPSocket(accept_socket.Pass());
+
+ on_new_connection_callback_.Run(peer, peer_address);
}
} // namespace device