| // Copyright 2017 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 CHROMEOS_COMPONENTS_TETHER_BLE_ADVERTISER_IMPL_H_ |
| #define CHROMEOS_COMPONENTS_TETHER_BLE_ADVERTISER_IMPL_H_ |
| |
| #include <array> |
| |
| #include "base/macros.h" |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/weak_ptr.h" |
| #include "chromeos/components/tether/ble_advertiser.h" |
| #include "chromeos/services/secure_channel/ble_constants.h" |
| #include "components/cryptauth/data_with_timestamp.h" |
| |
| namespace base { |
| class TaskRunner; |
| } // namespace base |
| |
| namespace chromeos { |
| |
| namespace secure_channel { |
| class BleServiceDataHelper; |
| class BleSynchronizerBase; |
| class ErrorTolerantBleAdvertisement; |
| } // namespace secure_channel |
| |
| namespace tether { |
| |
| // Concrete BleAdvertiser implementation. |
| class BleAdvertiserImpl : public BleAdvertiser { |
| public: |
| class Factory { |
| public: |
| static std::unique_ptr<BleAdvertiser> NewInstance( |
| secure_channel::BleServiceDataHelper* ble_service_data_helper, |
| secure_channel::BleSynchronizerBase* ble_synchronizer); |
| static void SetInstanceForTesting(Factory* factory); |
| |
| protected: |
| virtual std::unique_ptr<BleAdvertiser> BuildInstance( |
| secure_channel::BleServiceDataHelper* ble_service_data_helper, |
| secure_channel::BleSynchronizerBase* ble_synchronizer); |
| |
| private: |
| static Factory* factory_instance_; |
| }; |
| |
| ~BleAdvertiserImpl() override; |
| |
| // BleAdvertiser: |
| bool StartAdvertisingToDevice(const std::string& device_id) override; |
| bool StopAdvertisingToDevice(const std::string& device_id) override; |
| bool AreAdvertisementsRegistered() override; |
| |
| protected: |
| BleAdvertiserImpl( |
| secure_channel::BleServiceDataHelper* ble_service_data_helper, |
| secure_channel::BleSynchronizerBase* ble_synchronizer); |
| |
| private: |
| friend class BleAdvertiserImplTest; |
| |
| // Data needed to generate an advertisement. |
| struct AdvertisementMetadata { |
| AdvertisementMetadata( |
| const std::string& device_id, |
| std::unique_ptr<cryptauth::DataWithTimestamp> service_data); |
| ~AdvertisementMetadata(); |
| |
| std::string device_id; |
| std::unique_ptr<cryptauth::DataWithTimestamp> service_data; |
| }; |
| |
| void SetTaskRunnerForTesting( |
| scoped_refptr<base::TaskRunner> test_task_runner); |
| void UpdateAdvertisements(); |
| void OnAdvertisementStopped(size_t index); |
| |
| secure_channel::BleServiceDataHelper* ble_service_data_helper_; |
| secure_channel::BleSynchronizerBase* ble_synchronizer_; |
| |
| scoped_refptr<base::TaskRunner> task_runner_; |
| |
| // |registered_device_ids_| holds the device IDs that are currently |
| // registered and is always up-to-date. |advertisements_| contains the active |
| // advertisements, which may not correspond exactly to |
| // |registered_device_ids_| in the case that a previous advertisement failed |
| // to unregister. |
| std::array<std::unique_ptr<AdvertisementMetadata>, |
| secure_channel::kMaxConcurrentAdvertisements> |
| registered_device_metadata_; |
| std::array<std::unique_ptr<secure_channel::ErrorTolerantBleAdvertisement>, |
| secure_channel::kMaxConcurrentAdvertisements> |
| advertisements_; |
| |
| base::WeakPtrFactory<BleAdvertiserImpl> weak_ptr_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(BleAdvertiserImpl); |
| }; |
| |
| } // namespace tether |
| |
| } // namespace chromeos |
| |
| #endif // CHROMEOS_COMPONENTS_TETHER_BLE_ADVERTISER_IMPL_H_ |