Bluetooth: mac: add connected BLE devices at startup

device/bluetooth/bluetooth_adapter_mac.mm

 // Copyright 2013 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_adapter_mac.h"
 
 #import <IOBluetooth/objc/IOBluetoothDevice.h>
 #import <IOBluetooth/objc/IOBluetoothHostController.h>
 #include <stddef.h>
 
@@ skipping 39 matching lines @@
 base::WeakPtr<BluetoothAdapterMac> BluetoothAdapterMac::CreateAdapter() {
   BluetoothAdapterMac* adapter = new BluetoothAdapterMac();
   adapter->Init();
   return adapter->weak_ptr_factory_.GetWeakPtr();
 }
 
 // static
 base::WeakPtr<BluetoothAdapterMac> BluetoothAdapterMac::CreateAdapterForTest(
     std::string name,
     std::string address,
-    scoped_refptr<base::SequencedTaskRunner> ui_task_runner) {
+    scoped_refptr<base::SequencedTaskRunner> ui_task_runner,
+    CBCentralManager* central_manager) {
   BluetoothAdapterMac* adapter = new BluetoothAdapterMac();
+  if (central_manager) {
+    adapter->SetCentralManagerForTesting(central_manager);
+  }
   adapter->InitForTest(ui_task_runner);
   adapter->name_ = name;
   adapter->address_ = address;
   return adapter->weak_ptr_factory_.GetWeakPtr();
 }
 
 // static
 BluetoothUUID BluetoothAdapterMac::BluetoothUUIDWithCBUUID(CBUUID* uuid) {
   // UUIDString only available OS X >= 10.10.
   DCHECK(base::mac::IsAtLeastOS10_10());
@@ skipping 300 matching lines @@
     // Begin a low energy discovery session or update it if one is already
     // running.
     if (IsLowEnergyAvailable())
       low_energy_discovery_manager_->StartDiscovery(
           BluetoothDevice::UUIDList());
   }
   return true;
 }
 
 void BluetoothAdapterMac::Init() {
+  if (IsLowEnergyAvailable())
+    AddConnectedLowEnergyDevices();
   ui_task_runner_ = base::ThreadTaskRunnerHandle::Get();
   PollAdapter();
 }
 
 void BluetoothAdapterMac::InitForTest(
     scoped_refptr<base::SequencedTaskRunner> ui_task_runner) {
+  if (IsLowEnergyAvailable())
+    AddConnectedLowEnergyDevices();
   ui_task_runner_ = ui_task_runner;
 }
 
 void BluetoothAdapterMac::PollAdapter() {
   // TODO(erikchen): Remove ScopedTracker below once http://crbug.com/461181
   // is fixed.
   tracked_objects::ScopedTracker tracking_profile1(
       FROM_HERE_WITH_EXPLICIT_FUNCTION(
           "461181 BluetoothAdapterMac::PollAdapter::Start"));
   bool was_present = IsPresent();
@@ skipping 50 matching lines @@
   // is fixed.
   tracked_objects::ScopedTracker tracking_profile5(
       FROM_HERE_WITH_EXPLICIT_FUNCTION(
           "461181 BluetoothAdapterMac::PollAdapter::RemoveTimedOutDevices"));
   RemoveTimedOutDevices();
 
   // TODO(erikchen): Remove ScopedTracker below once http://crbug.com/461181
   // is fixed.
   tracked_objects::ScopedTracker tracking_profile6(
       FROM_HERE_WITH_EXPLICIT_FUNCTION(
-          "461181 BluetoothAdapterMac::PollAdapter::AddPairedDevices"));
-  AddPairedDevices();
+          "461181 BluetoothAdapterMac::PollAdapter::AddPairedClassicDevices"));
+  AddPairedClassicDevices();
 
   ui_task_runner_->PostDelayedTask(
       FROM_HERE,
       base::Bind(&BluetoothAdapterMac::PollAdapter,
                  weak_ptr_factory_.GetWeakPtr()),
       base::TimeDelta::FromMilliseconds(kPollIntervalMs));
 }
 
 void BluetoothAdapterMac::ClassicDeviceAdded(IOBluetoothDevice* device) {
   std::string device_address =
@@ skipping 87 matching lines @@
                       DeviceAdded(this, device_mac));
   } else {
     FOR_EACH_OBSERVER(BluetoothAdapter::Observer, observers_,
                       DeviceChanged(this, device_mac));
   }
 }
 
 // TODO(krstnmnlsn): Implement. crbug.com/511025
 void BluetoothAdapterMac::LowEnergyCentralManagerUpdatedState() {}
 
-void BluetoothAdapterMac::AddPairedDevices() {
-  // Add any new paired devices.
+void BluetoothAdapterMac::AddPairedClassicDevices() {
   for (IOBluetoothDevice* device in [IOBluetoothDevice pairedDevices]) {
-    ClassicDeviceAdded(device);
+    IOBluetoothSDPServiceRecord* pnp_information =
+        [device getServiceRecordForUUID:
+                    [IOBluetoothSDPUUID
+                        uuid16:kBluetoothSDPUUID16ServiceClassPnPInformation]];
+
+    // IOBluetoothDevice instance is a Bluetooth Classic device if it has a PNP
+    // information service SDP entry.
+    if (pnp_information) {
+      ClassicDeviceAdded(device);
+    }
   }
 }
 
+void BluetoothAdapterMac::AddConnectedLowEnergyDevices() {
+  // Look for Device Information Service. BLE devices should implement it.
+  CBUUID* device_information_uuid = [CBUUID UUIDWithString:@"180A"];
+
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Wpartial-availability"
+  // This list of devices contain both devices that are connected by other apps
+  // or by system. In that case, user will need to connect peripheral using
+  // CBCentralManager's connectPeripheral:options: method.
+  NSArray* connected_peripherals = [low_energy_central_manager_
+      retrieveConnectedPeripheralsWithServices:@[ device_information_uuid ]];
+#pragma clang diagnostic pop
+
+  for (CBPeripheral* peripheral in connected_peripherals) {
+    BluetoothLowEnergyDeviceMac* device_mac =
+        GetBluetoothLowEnergyDeviceMac(peripheral);
+
+    if (!device_mac) {
+      device_mac = new BluetoothLowEnergyDeviceMac(this, peripheral);
+      std::string device_address =
+          BluetoothLowEnergyDeviceMac::GetPeripheralHashAddress(peripheral);
+      devices_.add(device_address,
+                   std::unique_ptr<BluetoothDevice>(device_mac));
+      FOR_EACH_OBSERVER(BluetoothAdapter::Observer, observers_,
+                        DeviceAdded(this, device_mac));
+    }
+  }
+}
+
 void BluetoothAdapterMac::CreateGattConnection(
     BluetoothLowEnergyDeviceMac* device_mac) {
   [low_energy_central_manager_ connectPeripheral:device_mac->peripheral_
                                          options:nil];
 }
 
 void BluetoothAdapterMac::DisconnectGatt(
     BluetoothLowEnergyDeviceMac* device_mac) {
   [low_energy_central_manager_
       cancelPeripheralConnection:device_mac->peripheral_];
@@ skipping 51 matching lines @@
   std::string device_address =
       BluetoothLowEnergyDeviceMac::GetPeripheralHashAddress(peripheral);
   DevicesMap::const_iterator iter = devices_.find(device_address);
   if (iter == devices_.end()) {
     return nil;
   }
   return static_cast<BluetoothLowEnergyDeviceMac*>(iter->second);
 }
 
 }  // namespace device