components/arc: implement multi advertising

components/arc/bluetooth/arc_bluetooth_bridge.h

 // 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.
 
 #ifndef COMPONENTS_ARC_BLUETOOTH_ARC_BLUETOOTH_BRIDGE_H_
 #define COMPONENTS_ARC_BLUETOOTH_ARC_BLUETOOTH_BRIDGE_H_
 
 #include <stdint.h>
 
 #include <memory>
 #include <string>
 #include <unordered_map>
 #include <unordered_set>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/timer/timer.h"
 #include "components/arc/arc_bridge_service.h"
 #include "components/arc/arc_service.h"
 #include "components/arc/common/bluetooth.mojom.h"
 #include "components/arc/instance_holder.h"
 #include "device/bluetooth/bluetooth_adapter.h"
 #include "device/bluetooth/bluetooth_adapter_factory.h"
+#include "device/bluetooth/bluetooth_advertisement.h"
 #include "device/bluetooth/bluetooth_device.h"
 #include "device/bluetooth/bluetooth_discovery_session.h"
 #include "device/bluetooth/bluetooth_local_gatt_service.h"
 #include "device/bluetooth/bluetooth_remote_gatt_characteristic.h"
 #include "device/bluetooth/bluetooth_remote_gatt_descriptor.h"
 #include "device/bluetooth/bluetooth_remote_gatt_service.h"
 #include "device/bluetooth/bluez/bluetooth_adapter_bluez.h"
 #include "mojo/public/cpp/bindings/binding.h"
 
 namespace arc {
@@ skipping 232 matching lines @@
                       const SendIndicationCallback& callback) override;
 
   // Bluetooth Mojo host interface - Bluetooth SDP functions
   void GetSdpRecords(mojom::BluetoothAddressPtr remote_addr,
                      const device::BluetoothUUID& target_uuid) override;
   void CreateSdpRecord(mojom::BluetoothSdpRecordPtr record_mojo,
                        const CreateSdpRecordCallback& callback) override;
   void RemoveSdpRecord(uint32_t service_handle,
                        const RemoveSdpRecordCallback& callback) override;
 
+  // Set up or disable multiple advertising.
+  void ReserveAdvertisementHandle(
+      const ReserveAdvertisementHandleCallback& callback) override;
+  void BroadcastAdvertisement(
+      int32_t adv_handle,
+      std::unique_ptr<device::BluetoothAdvertisement::Data> advertisement,
+      const BroadcastAdvertisementCallback& callback) override;
+  void ReleaseAdvertisementHandle(
+      int32_t adv_handle,
+      const ReleaseAdvertisementHandleCallback& callback) override;
+
   // Chrome observer callbacks
   void OnPoweredOn(
       const base::Callback<void(mojom::BluetoothAdapterState)>& callback) const;
   void OnPoweredOff(
       const base::Callback<void(mojom::BluetoothAdapterState)>& callback) const;
   void OnPoweredError(
       const base::Callback<void(mojom::BluetoothAdapterState)>& callback) const;
   void OnDiscoveryStarted(
       std::unique_ptr<device::BluetoothDiscoverySession> session);
   void OnDiscoveryStopped();
@@ skipping 84 matching lines @@
 
   void OnGetServiceRecordsDone(
       mojom::BluetoothAddressPtr remote_addr,
       const device::BluetoothUUID& target_uuid,
       const std::vector<bluez::BluetoothServiceRecordBlueZ>& records_bluez);
   void OnGetServiceRecordsError(
       mojom::BluetoothAddressPtr remote_addr,
       const device::BluetoothUUID& target_uuid,
       bluez::BluetoothServiceRecordBlueZ::ErrorCode error_code);
 
+  // Callbacks for managing advertisements registered from the instance.
+
+  // Called when we have an open slot in the advertisement map and want to
+  // register the advertisement given by |data| for handle |adv_handle|.
+  void OnReadyToRegisterAdvertisement(
+      const BroadcastAdvertisementCallback& callback,
+      int32_t adv_handle,
+      std::unique_ptr<device::BluetoothAdvertisement::Data> data);
+  // Called when we've successfully registered a new advertisement for
+  // handle |adv_handle|.
+  void OnRegisterAdvertisementDone(
+      const BroadcastAdvertisementCallback& callback,
+      int32_t adv_handle,
+      scoped_refptr<device::BluetoothAdvertisement> advertisement);
+  // Called when the attempt to register an advertisement for handle
+  // |adv_handle| has failed. |adv_handle| remains reserved, but no
+  // advertisement is associated with it.
+  void OnRegisterAdvertisementError(
+      const BroadcastAdvertisementCallback& callback,
+      int32_t adv_handle,
+      device::BluetoothAdvertisement::ErrorCode error_code);
+  // Both of the following are called after we've tried to unregister
+  // the advertisement for |adv_handle|. Either way, we will no
+  // longer be broadcasting this advertisement, so in either case, the
+  // handle can be released.
+  void OnUnregisterAdvertisementDone(
+      const ReleaseAdvertisementHandleCallback& callback,
+      int32_t adv_handle);
+  void OnUnregisterAdvertisementError(
+      const ReleaseAdvertisementHandleCallback& callback,
+      int32_t adv_handle,
+      device::BluetoothAdvertisement::ErrorCode error_code);
+  // Find the next free advertisement handle and put it in *adv_handle,
+  // or return false if the advertisement map is full.
+  bool GetAdvertisementHandle(int32_t* adv_handle);
+
   bool CalledOnValidThread();
 
   mojo::Binding<mojom::BluetoothHost> binding_;
 
   scoped_refptr<bluez::BluetoothAdapterBlueZ> bluetooth_adapter_;
   scoped_refptr<device::BluetoothAdvertisement> advertisment_;
   std::unique_ptr<device::BluetoothDiscoverySession> discovery_session_;
   std::unordered_map<std::string,
                      std::unique_ptr<device::BluetoothGattNotifySession>>
       notification_session_;
@@ skipping 11 matching lines @@
   // have made. We need to hang on to these as long as the connection is
   // active since their destructors will drop the connections otherwise.
   std::unordered_map<std::string,
                      std::unique_ptr<device::BluetoothGattConnection>>
       gatt_connections_;
   // Timer to turn discovery off.
   base::OneShotTimer discovery_off_timer_;
   // Timer to turn adapter discoverable off.
   base::OneShotTimer discoverable_off_timer_;
 
+  // Holds advertising data registered by the instance.
+  //
+  // When a handle is reserved, an entry is placed into the advertisements_
+  // map. This entry is not yet associated with a device::BluetoothAdvertisement
+  // because the instance hasn't sent us any advertising data yet, so its
+  // mapped value is nullptr until that happens. Thus we have three states for a
+  // handle:
+  // * unmapped -> free
+  // * mapped to nullptr -> reserved, awaiting data
+  // * mapped to a device::BluetoothAdvertisement -> in use, and the mapped
+  //   BluetoothAdvertisement is currently registered with the adapter.
+  enum { kMaxAdvertisements = 5 };
+  std::map<int32_t, scoped_refptr<device::BluetoothAdvertisement>>
+      advertisements_;
+
   base::ThreadChecker thread_checker_;
 
   // WeakPtrFactory to use for callbacks.
   base::WeakPtrFactory<ArcBluetoothBridge> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ArcBluetoothBridge);
 };
 
 }  // namespace arc
 
 #endif  // COMPONENTS_ARC_BLUETOOTH_ARC_BLUETOOTH_BRIDGE_H_