Implement BluetoothRemoteGattServiceWin and related unit tests

device/bluetooth/bluetooth_low_energy_win_fake.cc

 // Copyright 2016 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_low_energy_win_fake.h"
 
 #include "base/strings/stringprintf.h"
+#include "device/bluetooth/bluetooth_low_energy_defs_win.h"
 
 namespace {
 const char kPlatformNotSupported[] =
     "Bluetooth Low energy is only supported on Windows 8 and later.";
 }  // namespace
 
 namespace device {
 namespace win {
 
 BLEDevice::BLEDevice() {}
 BLEDevice::~BLEDevice() {}
 
 BLEGattService::BLEGattService() {}
 BLEGattService::~BLEGattService() {}
 
 BLEGattCharacteristic::BLEGattCharacteristic() {}
 BLEGattCharacteristic::~BLEGattCharacteristic() {}
 
 BLEGattDescriptor::BLEGattDescriptor() {}
 BLEGattDescriptor::~BLEGattDescriptor() {}
 
 BluetoothLowEnergyWrapperFake::BluetoothLowEnergyWrapperFake() {}
 BluetoothLowEnergyWrapperFake::~BluetoothLowEnergyWrapperFake() {}
 
+bool BluetoothLowEnergyWrapperFake::IsBluetoothLowEnergySupported() {
+  return true;
+}
+
 bool BluetoothLowEnergyWrapperFake::EnumerateKnownBluetoothLowEnergyDevices(
     ScopedVector<BluetoothLowEnergyDeviceInfo>* devices,
     std::string* error) {
   if (!IsBluetoothLowEnergySupported()) {
     *error = kPlatformNotSupported;
     return false;
   }
 
   for (auto& device : simulated_devices_) {
     BluetoothLowEnergyDeviceInfo* device_info =
@@ skipping 32 matching lines @@
     const base::FilePath& device_path,
     ScopedVector<BluetoothLowEnergyServiceInfo>* services,
     std::string* error) {
   if (!IsBluetoothLowEnergySupported()) {
     *error = kPlatformNotSupported;
     return false;
   }
 
   base::string16 device_address =
       ExtractDeviceAddressFromDevicePath(device_path.value());
-  base::string16 service_attribute_handle =
-      ExtractServiceAttributeHandleFromDevicePath(device_path.value());
+  std::vector<std::string> service_attribute_handles =
+      ExtractServiceAttributeHandlesFromDevicePath(device_path.value());
 
   BLEDevicesMap::iterator it_d = simulated_devices_.find(
       std::string(device_address.begin(), device_address.end()));
   CHECK(it_d != simulated_devices_.end());
 
-  // |service_attribute_handle| is empty means |device_path| is a BLE device
+  // |service_attribute_handles| is empty means |device_path| is a BLE device
   // path, otherwise it is a BLE GATT service device path.
-  if (service_attribute_handle.empty()) {
+  if (service_attribute_handles.empty()) {
     // Return all primary services for BLE device.
     for (auto& primary_service : it_d->second->primary_services) {
       BluetoothLowEnergyServiceInfo* service_info =
           new BluetoothLowEnergyServiceInfo();
       service_info->uuid = primary_service.second->service_info->ServiceUuid;
       service_info->attribute_handle =
           primary_service.second->service_info->AttributeHandle;
       services->push_back(service_info);
     }
   } else {
     // Return corresponding GATT service for BLE GATT service device.
-    BLEGattServicesMap::iterator it_s =
-        it_d->second->primary_services.find(std::string(
-            service_attribute_handle.begin(), service_attribute_handle.end()));
-    CHECK(it_s != it_d->second->primary_services.end());
+    BLEGattService* target_service =
+        GetSimulatedGattService(it_d->second.get(), service_attribute_handles);
     BluetoothLowEnergyServiceInfo* service_info =
         new BluetoothLowEnergyServiceInfo();
-    service_info->uuid = it_s->second->service_info->ServiceUuid;
+    service_info->uuid = target_service->service_info->ServiceUuid;
     service_info->attribute_handle =
-        it_s->second->service_info->AttributeHandle;
+        target_service->service_info->AttributeHandle;
     services->push_back(service_info);
   }
 
   return true;
 }
 
+HRESULT BluetoothLowEnergyWrapperFake::ReadIncludedServicesOfAService(
+    base::FilePath& service_path,
+    const PBTH_LE_GATT_SERVICE service,
+    scoped_ptr<BTH_LE_GATT_SERVICE>* out_included_services,
+    USHORT* out_counts) {
+  base::string16 device_address =
+      ExtractDeviceAddressFromDevicePath(service_path.value());
+  std::vector<std::string> service_att_handles =
+      ExtractServiceAttributeHandlesFromDevicePath(service_path.value());
+  BLEGattService* target_service = GetSimulatedGattService(
+      GetSimulatedBLEDevice(
+          std::string(device_address.begin(), device_address.end())),
+      service_att_handles);
+  if (target_service == nullptr)
+    return HRESULT_FROM_WIN32(ERROR_NOT_FOUND);
+
+  std::size_t number_of_included_services =
+      target_service->included_services.size();
+  if (number_of_included_services) {
+    *out_counts = (USHORT)number_of_included_services;
+    out_included_services->reset(
+        new BTH_LE_GATT_SERVICE[number_of_included_services]);
+    std::size_t i = 0;
+    for (const auto& service : target_service->included_services) {
+      out_included_services->get()[i].ServiceUuid =
+          service.second->service_info->ServiceUuid;
+      out_included_services->get()[i].AttributeHandle =
+          service.second->service_info->AttributeHandle;
+      i++;
+    }
+    return S_OK;
+  }
+  return HRESULT_FROM_WIN32(ERROR_NO_MORE_ITEMS);
+}
+
+HRESULT BluetoothLowEnergyWrapperFake::ReadCharacteristicsOfAService(
+    base::FilePath& service_path,
+    const PBTH_LE_GATT_SERVICE service,
+    scoped_ptr<BTH_LE_GATT_CHARACTERISTIC>* out_included_characteristics,
+    USHORT* out_counts) {
+  base::string16 device_address =
+      ExtractDeviceAddressFromDevicePath(service_path.value());
+  const std::vector<std::string> service_att_handles =
+      ExtractServiceAttributeHandlesFromDevicePath(service_path.value());
+  BLEGattService* target_service = GetSimulatedGattService(
+      GetSimulatedBLEDevice(
+          std::string(device_address.begin(), device_address.end())),
+      service_att_handles);
+  if (target_service == nullptr)
+    return HRESULT_FROM_WIN32(ERROR_NOT_FOUND);
+
+  std::size_t number_of_included_characteristic =
+      target_service->included_characteristics.size();
+  if (number_of_included_characteristic) {
+    *out_counts = (USHORT)number_of_included_characteristic;
+    out_included_characteristics->reset(
+        new BTH_LE_GATT_CHARACTERISTIC[number_of_included_characteristic]);
+    std::size_t i = 0;
+    for (const auto& cha : target_service->included_characteristics) {
+      out_included_characteristics->get()[i] =
+          *(cha.second->characteristic_info);
+      i++;
+    }
+    return S_OK;
+  }
+  return HRESULT_FROM_WIN32(ERROR_NO_MORE_ITEMS);
+}
+
 BLEDevice* BluetoothLowEnergyWrapperFake::SimulateBLEDevice(
     std::string device_name,
     BLUETOOTH_ADDRESS device_address) {
   BLEDevice* device = new BLEDevice();
   BluetoothLowEnergyDeviceInfo* device_info =
       new BluetoothLowEnergyDeviceInfo();
   std::string string_device_address =
       BluetoothAddressToCanonicalString(device_address);
   device_info->path =
       base::FilePath(GenerateBLEDevicePath(string_device_address));
   device_info->friendly_name = device_name;
   device_info->address = device_address;
   device->device_info.reset(device_info);
   simulated_devices_[string_device_address] = make_scoped_ptr(device);
   return device;
 }
 
+BLEDevice* BluetoothLowEnergyWrapperFake::GetSimulatedBLEDevice(
+    std::string device_address) {
+  BLEDevicesMap::iterator it_d = simulated_devices_.find(device_address);
+  if (it_d == simulated_devices_.end())
+    return nullptr;
+  return it_d->second.get();
+}
+
 BLEGattService* BluetoothLowEnergyWrapperFake::SimulateBLEGattService(
     BLEDevice* device,
-    std::string uuid) {
+    BLEGattService* parent_service,
+    const BTH_LE_UUID& uuid) {
   CHECK(device);
 
   BLEGattService* service = new BLEGattService();
   PBTH_LE_GATT_SERVICE service_info = new BTH_LE_GATT_SERVICE[1];
   std::string string_device_address =
       BluetoothAddressToCanonicalString(device->device_info->address);
   service_info->AttributeHandle =
       GenerateAUniqueAttributeHandle(string_device_address);
-  service_info->ServiceUuid = CanonicalStringToBTH_LE_UUID(uuid);
+  service_info->ServiceUuid = uuid;
   service->service_info.reset(service_info);
-  device->primary_services[std::to_string(service_info->AttributeHandle)] =
-      make_scoped_ptr(service);
+
+  if (parent_service) {
+    parent_service
+        ->included_services[std::to_string(service_info->AttributeHandle)] =
+        make_scoped_ptr(service);
+  } else {
+    device->primary_services[std::to_string(service_info->AttributeHandle)] =
+        make_scoped_ptr(service);
+  }
   return service;
 }
 
-BLEDevice* BluetoothLowEnergyWrapperFake::GetSimulatedBLEDevice(
-    std::string device_address) {
-  BLEDevicesMap::iterator it_d = simulated_devices_.find(device_address);
-  if (it_d == simulated_devices_.end())
+void BluetoothLowEnergyWrapperFake::SimulateBLEGattServiceRemoved(
+    BLEDevice* device,
+    BLEGattService* parent_service,
+    std::string attribute_handle) {
+  if (parent_service) {
+    parent_service->included_services.erase(attribute_handle);
+  } else {
+    device->primary_services.erase(attribute_handle);
+  }
+}
+
+BLEGattService* BluetoothLowEnergyWrapperFake::GetSimulatedGattService(
+    BLEDevice* device,
+    const std::vector<std::string>& chain_of_att_handle) {
+  // First, find the root primary service.
+  BLEGattServicesMap::iterator it_s =
+      device->primary_services.find(chain_of_att_handle[0]);
+  if (it_s == device->primary_services.end())
     return nullptr;
-  return it_d->second.get();
+
+  // Iteratively follow the chain of included service attribute handles to find
+  // the target service.
+  for (std::size_t i = 1; i < chain_of_att_handle.size(); i++) {
+    std::string included_att_handle = std::string(
+        chain_of_att_handle[i].begin(), chain_of_att_handle[i].end());
+    BLEGattServicesMap::iterator it_i =
+        it_s->second->included_services.find(included_att_handle);
+    if (it_i == it_s->second->included_services.end())
+      return nullptr;
+    it_s = it_i;
+  }
+  return it_s->second.get();
+}
+
+BLEGattCharacteristic*
+BluetoothLowEnergyWrapperFake::SimulateBLEGattCharacterisc(
+    std::string device_address,
+    BLEGattService* parent_service,
+    const BTH_LE_GATT_CHARACTERISTIC& characteristic) {
+  CHECK(parent_service);
+
+  BLEGattCharacteristic* cha = new BLEGattCharacteristic();
+  PBTH_LE_GATT_CHARACTERISTIC win_cha_info = new BTH_LE_GATT_CHARACTERISTIC[1];
+  *win_cha_info = characteristic;
+  (cha->characteristic_info).reset(win_cha_info);
+  cha->characteristic_info->AttributeHandle =
+      GenerateAUniqueAttributeHandle(device_address);
+  parent_service->included_characteristics[std::to_string(
+      cha->characteristic_info->AttributeHandle)] = make_scoped_ptr(cha);
+  return cha;
+}
+
+void BluetoothLowEnergyWrapperFake::SimulateBLEGattCharacteriscRemove(
+    BLEGattService* parent_service,
+    std::string attribute_handle) {
+  CHECK(parent_service);
+  parent_service->included_characteristics.erase(attribute_handle);
 }
 
 USHORT BluetoothLowEnergyWrapperFake::GenerateAUniqueAttributeHandle(
     std::string device_address) {
   scoped_ptr<std::set<USHORT>>& set_of_ushort =
       attribute_handle_table_[device_address];
   if (set_of_ushort) {
     USHORT max_attribute_handle = *set_of_ushort->rbegin();
     if (max_attribute_handle < 0xFFFF) {
       USHORT new_attribute_handle = max_attribute_handle + 1;
@@ skipping 28 matching lines @@
     base::string16 resident_device_path,
     USHORT service_attribute_handle) {
   std::string sub_path = std::to_string(service_attribute_handle);
   return resident_device_path + L"/" +
          base::string16(sub_path.begin(), sub_path.end());
 }
 
 base::string16
 BluetoothLowEnergyWrapperFake::ExtractDeviceAddressFromDevicePath(
     base::string16 path) {
-  std::size_t found = path.find('/');
+  std::size_t found = path.find_first_of('/');
   if (found != base::string16::npos) {
     return path.substr(0, found);
   }
   return path;
 }
 
-base::string16
-BluetoothLowEnergyWrapperFake::ExtractServiceAttributeHandleFromDevicePath(
+std::vector<std::string>
+BluetoothLowEnergyWrapperFake::ExtractServiceAttributeHandlesFromDevicePath(
     base::string16 path) {
   std::size_t found = path.find('/');
   if (found == base::string16::npos)
-    return base::string16();
-  return path.substr(found + 1);
-}
+    return std::vector<std::string>();
 
-BTH_LE_UUID BluetoothLowEnergyWrapperFake::CanonicalStringToBTH_LE_UUID(
-    std::string uuid) {
-  BTH_LE_UUID win_uuid;
-  // Only short UUIDs (4 hex digits) have beened used in BluetoothTest right
-  // now. Need fix after using long UUIDs.
-  win_uuid.IsShortUuid = true;
-  unsigned int data[1];
-  int result = sscanf_s(uuid.c_str(), "%04x", &data[0]);
-  CHECK(result == 1);
-  win_uuid.Value.ShortUuid = data[0];
-  return win_uuid;
+  std::vector<std::string> chain_of_att_handle;
+  while (true) {
+    std::size_t next_found = path.find(path, found + 1);
+    if (next_found == base::string16::npos)
+      break;
+    base::string16 w_att_handle = path.substr(found + 1, next_found);
+    chain_of_att_handle.push_back(
+        std::string(w_att_handle.begin(), w_att_handle.end()));
+    found = next_found;
+  }
+  base::string16 w_att_handle = path.substr(found + 1);
+  chain_of_att_handle.push_back(
+      std::string(w_att_handle.begin(), w_att_handle.end()));
+  return chain_of_att_handle;
 }
 
 std::string BluetoothLowEnergyWrapperFake::BluetoothAddressToCanonicalString(
     const BLUETOOTH_ADDRESS& btha) {
   std::string result = base::StringPrintf(
       "%02X:%02X:%02X:%02X:%02X:%02X", btha.rgBytes[5], btha.rgBytes[4],
       btha.rgBytes[3], btha.rgBytes[2], btha.rgBytes[1], btha.rgBytes[0]);
   return result;
 }
 
 }  // namespace win
 }  // namespace device