| // Copyright 2015 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_CRYPTAUTH_BLUETOOTH_THROTTLER_IMPL_H_ |
| #define COMPONENTS_CRYPTAUTH_BLUETOOTH_THROTTLER_IMPL_H_ |
| |
| #include <memory> |
| #include <set> |
| |
| #include "base/macros.h" |
| #include "base/memory/singleton.h" |
| #include "base/time/time.h" |
| #include "components/cryptauth/bluetooth_throttler.h" |
| #include "components/cryptauth/connection_observer.h" |
| |
| namespace base { |
| class TickClock; |
| } |
| |
| namespace cryptauth { |
| |
| class Connection; |
| |
| // This class throttles repeated connection attempts to the same device. This |
| // throttling is necessary to prevent a kernel race condition when connecting |
| // before the previous connection fully closes, putting the connection in a |
| // corrupted, and unrecoverable state. http://crbug.com/345232 |
| class BluetoothThrottlerImpl : public BluetoothThrottler, |
| public ConnectionObserver { |
| public: |
| static BluetoothThrottlerImpl* GetInstance(); |
| ~BluetoothThrottlerImpl() override; |
| |
| // BluetoothThrottler: |
| base::TimeDelta GetDelay() const override; |
| void OnConnection(Connection* connection) override; |
| |
| protected: |
| // Creates a throttler for connections to a remote device, using the |clock| |
| // as a time source. |
| explicit BluetoothThrottlerImpl(std::unique_ptr<base::TickClock> clock); |
| |
| // Returns the duration to wait, after disconnecting, before reattempting a |
| // connection to the remote device. Exposed for testing. |
| base::TimeDelta GetCooldownTimeDelta() const; |
| |
| private: |
| friend struct base::DefaultSingletonTraits<BluetoothThrottlerImpl>; |
| |
| BluetoothThrottlerImpl(); |
| |
| // ConnectionObserver: |
| void OnConnectionStatusChanged(Connection* connection, |
| Connection::Status old_status, |
| Connection::Status new_status) override; |
| |
| // Tracks the last seen disconnect time for the |remote_device_|. |
| base::TimeTicks last_disconnect_time_; |
| |
| // The time source. |
| std::unique_ptr<base::TickClock> clock_; |
| |
| // The currently connected connections. |
| // Each connection is stored as a weak reference, which is safe because |this| |
| // instance is registered as an observer, and will unregister when the |
| // connection disconnects, which is guaranteed to occur before the connection |
| // is destroyed. |
| std::set<Connection*> connections_; |
| |
| DISALLOW_COPY_AND_ASSIGN(BluetoothThrottlerImpl); |
| }; |
| |
| } // namespace cryptauth |
| |
| #endif // COMPONENTS_CRYPTAUTH_BLUETOOTH_THROTTLER_IMPL_H_ |