chrome.bluetooth API improvements.

device/bluetooth/bluetooth_task_manager_win.cc

 // Copyright (c) 2012 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_task_manager_win.h"
 
 #include <winsock2.h>
 
 #include <string>
 
@@ skipping 12 matching lines @@
 #include "net/base/winsock_init.h"
 
 namespace {
 
 const int kNumThreadsInWorkerPool = 3;
 const char kBluetoothThreadName[] = "BluetoothPollingThreadWin";
 const int kMaxNumDeviceAddressChar = 127;
 const int kServiceDiscoveryResultBufferSize = 5000;
 const int kMaxDeviceDiscoveryTimeout = 48;
 
+typedef device::BluetoothTaskManagerWin::ServiceRecordState ServiceRecordState;
+
 // Populates bluetooth adapter state using adapter_handle.
 void GetAdapterState(HANDLE adapter_handle,
                      device::BluetoothTaskManagerWin::AdapterState* state) {
   std::string name;
   std::string address;
   bool powered = false;
   BLUETOOTH_RADIO_INFO adapter_info = { sizeof(BLUETOOTH_RADIO_INFO), 0 };
   if (adapter_handle &&
       ERROR_SUCCESS == BluetoothGetRadioInfo(adapter_handle,
                                              &adapter_info)) {
@@ skipping 21 matching lines @@
       device_info.Address.rgBytes[3],
       device_info.Address.rgBytes[2],
       device_info.Address.rgBytes[1],
       device_info.Address.rgBytes[0]);
   state->bluetooth_class = device_info.ulClassofDevice;
   state->visible = true;
   state->connected = !!device_info.fConnected;
   state->authenticated = !!device_info.fAuthenticated;
 }
 
+void DiscoverDeviceServices(
+    const std::string& device_address,
+    const GUID& protocol_uuid,
+    ScopedVector<ServiceRecordState>* service_record_states) {
+  // Bluetooth and WSAQUERYSET for Service Inquiry. See http://goo.gl/2v9pyt.
+  WSAQUERYSET sdp_query;
+  ZeroMemory(&sdp_query, sizeof(sdp_query));
+  sdp_query.dwSize = sizeof(sdp_query);
+  GUID protocol = protocol_uuid;
+  sdp_query.lpServiceClassId = &protocol;
+  sdp_query.dwNameSpace = NS_BTH;
+  wchar_t device_address_context[kMaxNumDeviceAddressChar];
+  std::size_t length = base::SysUTF8ToWide("(" + device_address + ")").copy(
+      device_address_context, kMaxNumDeviceAddressChar);
+  device_address_context[length] = NULL;
+  sdp_query.lpszContext = device_address_context;
+  HANDLE sdp_handle;
+  if (ERROR_SUCCESS !=
+      WSALookupServiceBegin(&sdp_query, LUP_RETURN_ALL, &sdp_handle)) {
+    return;
+  }
+  char sdp_buffer[kServiceDiscoveryResultBufferSize];
+  LPWSAQUERYSET sdp_result_data = reinterpret_cast<LPWSAQUERYSET>(sdp_buffer);
+  DWORD sdp_buffer_size = sizeof(sdp_buffer);
+  while (ERROR_SUCCESS ==
+         WSALookupServiceNext(
+             sdp_handle, LUP_RETURN_ALL, &sdp_buffer_size, sdp_result_data)) {
+    ServiceRecordState* service_record_state = new ServiceRecordState();
+    service_record_state->name =
+        base::SysWideToUTF8(sdp_result_data->lpszServiceInstanceName);
+    service_record_state->address = device_address;
+    for (uint64 i = 0; i < sdp_result_data->lpBlob->cbSize; i++) {
+      service_record_state->sdp_bytes.push_back(
+          sdp_result_data->lpBlob->pBlobData[i]);
+    }
+    service_record_states->push_back(service_record_state);
+  }
+  WSALookupServiceEnd(sdp_handle);
+}
+
 }  // namespace
 
 namespace device {
 
 // static
 const int BluetoothTaskManagerWin::kPollIntervalMs = 500;
 
 BluetoothTaskManagerWin::BluetoothTaskManagerWin(
     scoped_refptr<base::SequencedTaskRunner> ui_task_runner)
     : ui_task_runner_(ui_task_runner),
@@ skipping 284 matching lines @@
 void BluetoothTaskManagerWin::DiscoverServices(
     ScopedVector<DeviceState>* device_list) {
   DCHECK(bluetooth_task_runner_->RunsTasksOnCurrentThread());
   net::EnsureWinsockInit();
   for (ScopedVector<DeviceState>::iterator iter = device_list->begin();
       iter != device_list->end();
       ++iter) {
     const std::string device_address = (*iter)->address;
     ScopedVector<ServiceRecordState>* service_record_states =
         &(*iter)->service_record_states;
-    WSAQUERYSET sdp_query;
-    ZeroMemory(&sdp_query, sizeof(sdp_query));
-    sdp_query.dwSize = sizeof(sdp_query);
-    GUID protocol = L2CAP_PROTOCOL_UUID;
-    sdp_query.lpServiceClassId = &protocol;
-    sdp_query.dwNameSpace = NS_BTH;
-    wchar_t device_address_context[kMaxNumDeviceAddressChar];
-    std::size_t length =
-        base::SysUTF8ToWide("(" + device_address + ")").copy(
-            device_address_context, kMaxNumDeviceAddressChar);
-    device_address_context[length] = NULL;
-    sdp_query.lpszContext = device_address_context;
-    HANDLE sdp_handle;
-    if (ERROR_SUCCESS !=
-        WSALookupServiceBegin(&sdp_query, LUP_RETURN_ALL, &sdp_handle)) {
-      return;
-    }
-    char sdp_buffer[kServiceDiscoveryResultBufferSize];
-    LPWSAQUERYSET sdp_result_data = reinterpret_cast<LPWSAQUERYSET>(sdp_buffer);
-    DWORD sdp_buffer_size = sizeof(sdp_buffer);
-    while (ERROR_SUCCESS == WSALookupServiceNext(sdp_handle,
-                                                 LUP_RETURN_ALL,
-                                                 &sdp_buffer_size,
-                                                 sdp_result_data)) {
-      ServiceRecordState* service_record_state = new ServiceRecordState();
-      service_record_state->name =
-          base::SysWideToUTF8(sdp_result_data->lpszServiceInstanceName);
-      service_record_state->address = device_address;
-      for (uint64 i = 0; i < sdp_result_data->lpBlob->cbSize; i++) {
-        service_record_state->sdp_bytes.push_back(
-            sdp_result_data->lpBlob->pBlobData[i]);
-      }
-      service_record_states->push_back(service_record_state);
-    }
-    WSALookupServiceEnd(sdp_handle);
+
+    // TODO(rpaquay): Find a better way to enumerate registered protocols.
+    DiscoverDeviceServices(
+        device_address, SDP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, UDP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, RFCOMM_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, TCP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, TCSBIN_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, TCSAT_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, OBEX_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, IP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, FTP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, HTTP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, WSP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, BNEP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, UPNP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, HID_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, HCCC_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, HCDC_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, HN_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, AVCTP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, AVDTP_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, CMPT_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, UDI_C_PLANE_PROTOCOL_UUID, service_record_states);
+    DiscoverDeviceServices(
+        device_address, L2CAP_PROTOCOL_UUID, service_record_states);

[xiyuan, 2014/03/17 22:55:11]

It seems we only need enumerate L2CAP_PROTOCOL_UUID as in previous code.

Previous code has a bug that |sdp_buffer_size| might be changed after returning
from WSALookupServiceNext. Sometime |sdp_buffer_size| becomes too small and
causes service enumeration to fail with WSAEFAULT.

With that fixed (i.e. "sdp_buffer_size = sizeof(sdp_buffer)" at the end of the
WSALookupServiceNext loop and before calling it again), the loop gets all the
services, including the OBEXObjectPushServiceClass_UUID and I don't need to run
the loop for all other protocol UUIDs.

[rpaquay, 2014/03/20 00:48:18]

Done.

+
+    // TODO(rpaquay): Workaround issue where this profile is not discovered.
+    DiscoverDeviceServices(
+        device_address, OBEXObjectPushServiceClass_UUID, service_record_states);
   }
 }
 
 }  // namespace device