Add U2F request state machines

device/u2f/u2f_request.h

+// 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 DEVICE_U2F_U2F_REQUEST_H_
+#define DEVICE_U2F_U2F_REQUEST_H_
+
+#include "base/cancelable_callback.h"
+#include "base/scoped_observer.h"
+#include "device/hid/hid_device_filter.h"
+#include "device/hid/hid_service.h"
+#include "u2f_device.h"
+
+namespace device {
+class U2fRequest : HidService::Observer {
+ public:
+  using ResponseCallback =
+      base::Callback<void(uint8_t status_code, std::vector<uint8_t> response)>;
+
+  U2fRequest(const ResponseCallback& callback);
+  virtual ~U2fRequest();
+
+  void Start();
+
+ protected:
+  enum class State {
+    INIT,
+    BUSY,
+    WINK,
+    IDLE,
+    OFF,
+    COMPLETE,
+  };
+
+  void Transition();
+  virtual void TryDevice() = 0;
+
+  std::unique_ptr<U2fDevice> current_device_;
+  State state_;
+  const ResponseCallback& cb_;

[Reilly Grant (use Gerrit), 2017/04/21 17:19:16]

This should not be a reference as there is no guarantee that it will be valid
after it is set.

[Casey Piper, 2017/04/21 22:55:16]

Done.

+
+ private:
+  FRIEND_TEST_ALL_PREFIXES(U2fRequestTest, TestAddRemoveDevice);
+  FRIEND_TEST_ALL_PREFIXES(U2fRequestTest, TestIterateDevice);
+  FRIEND_TEST_ALL_PREFIXES(U2fRequestTest, TestBasicMachine);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestSimpleSign);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestMultipleHandles);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestSignSuccess);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestDelayedSuccess);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestMultipleDevices);
+  FRIEND_TEST_ALL_PREFIXES(U2fSignTest, TestFakeEnroll);
+  FRIEND_TEST_ALL_PREFIXES(U2fRegisterTest, TestRegisterSuccess);
+  FRIEND_TEST_ALL_PREFIXES(U2fRegisterTest, TestDelayedSuccess);
+  FRIEND_TEST_ALL_PREFIXES(U2fRegisterTest, TestMultipleDevices);
+
+  void Enumerate();
+  void IterateDevice();
+  void OnWaitComplete();
+  void OnDeviceAdded(scoped_refptr<HidDeviceInfo> device_info) override;
+  void OnDeviceRemoved(scoped_refptr<HidDeviceInfo> device_info) override;
+  void OnEnumerate(const std::vector<scoped_refptr<HidDeviceInfo>>& devices);
+
+  std::list<std::unique_ptr<U2fDevice>> devices_;
+  std::list<std::unique_ptr<U2fDevice>> attempted_devices_;
+  base::CancelableClosure delay_callback_;
+  HidDeviceFilter filter_;
+  ScopedObserver<HidService, HidService::Observer> hid_service_observer_;
+  base::WeakPtrFactory<U2fRequest> weak_factory_;
+};
+}  // namespace device
+
+#endif  // DEVICE_U2F_U2F_REQUEST_H_