Add support for U2fHidDevice interaction

device/u2f/u2f_device.cc

 // 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.
 
+#include "u2f_device.h"
+
 #include "base/bind.h"
 #include "u2f_apdu_command.h"
-#include "u2f_device.h"
+#include "u2f_apdu_response.h"
 
 namespace device {
 
 U2fDevice::U2fDevice() : weak_factory_(this) {}
 
 U2fDevice::~U2fDevice() {}
 
 void U2fDevice::Register(const std::vector<uint8_t>& app_param,
-                         U2fDevice::ProtocolVersion version,
                          const std::vector<uint8_t>& challenge_param,
                          const MessageCallback& callback) {
   scoped_refptr<U2fApduCommand> register_cmd =
       U2fApduCommand::CreateRegister(app_param, challenge_param);
   if (!register_cmd) {
     callback.Run(ReturnCode::INVALID_PARAMS, std::vector<uint8_t>());
     return;
   }
   DeviceTransact(std::move(register_cmd),
                  base::Bind(&U2fDevice::OnRegisterComplete,
@@ skipping 23 matching lines @@
   }
   DeviceTransact(std::move(version_cmd),
                  base::Bind(&U2fDevice::OnVersionComplete,
                             weak_factory_.GetWeakPtr(), callback));
 }
 
 void U2fDevice::OnRegisterComplete(
     const MessageCallback& callback,
     bool success,
     scoped_refptr<U2fApduResponse> register_response) {
-  NOTIMPLEMENTED();
+  if (!success || !register_response) {
+    callback.Run(ReturnCode::FAILURE, std::vector<uint8_t>());
+    return;
+  }
+  switch (register_response->status()) {
+    case U2fApduResponse::Status::SW_CONDITIONS_NOT_SATISFIED:
+      callback.Run(ReturnCode::CONDITIONS_NOT_SATISFIED,
+                   std::vector<uint8_t>());
+      break;
+    case U2fApduResponse::Status::SW_NO_ERROR:
+      callback.Run(ReturnCode::SUCCESS, register_response->data());
+      break;
+    case U2fApduResponse::Status::SW_WRONG_DATA:
+      callback.Run(ReturnCode::INVALID_PARAMS, std::vector<uint8_t>());
+      break;
+    default:
+      callback.Run(ReturnCode::FAILURE, std::vector<uint8_t>());
+      break;
+  }
 }
 
 void U2fDevice::OnSignComplete(const MessageCallback& callback,
                                bool success,
                                scoped_refptr<U2fApduResponse> sign_response) {
-  NOTIMPLEMENTED();
+  if (!success || !sign_response) {
+    callback.Run(ReturnCode::FAILURE, std::vector<uint8_t>());
+    return;
+  }
+  switch (sign_response->status()) {
+    case U2fApduResponse::Status::SW_CONDITIONS_NOT_SATISFIED:
+      callback.Run(ReturnCode::CONDITIONS_NOT_SATISFIED,
+                   std::vector<uint8_t>());
+      break;
+    case U2fApduResponse::Status::SW_NO_ERROR:
+      callback.Run(ReturnCode::SUCCESS, sign_response->data());
+      break;
+    case U2fApduResponse::Status::SW_WRONG_DATA:
+      callback.Run(ReturnCode::INVALID_PARAMS, std::vector<uint8_t>());
+      break;
+    default:
+      callback.Run(ReturnCode::FAILURE, std::vector<uint8_t>());
+      break;
+  }
 }
 
 void U2fDevice::OnVersionComplete(
     const VersionCallback& callback,
     bool success,
     scoped_refptr<U2fApduResponse> version_response) {
   if (success && version_response &&
       version_response->status() == U2fApduResponse::Status::SW_NO_ERROR &&
       version_response->data() ==
           std::vector<uint8_t>({'U', '2', 'F', '_', 'V', '2'})) {
     callback.Run(success, ProtocolVersion::U2F_V2);
     return;
   }
   callback.Run(success, ProtocolVersion::UNKNOWN);
 }
 
 void U2fDevice::OnLegacyVersionComplete(
     const VersionCallback& callback,
     bool success,
     scoped_refptr<U2fApduResponse> legacy_version_response) {
-  NOTIMPLEMENTED();
+  if (success && legacy_version_response &&
+      legacy_version_response->status() ==
+          U2fApduResponse::Status::SW_NO_ERROR &&
+      legacy_version_response->data() ==
+          std::vector<uint8_t>({'U', '2', 'F', '_', 'V', '2'})) {
+    callback.Run(success, ProtocolVersion::U2F_V2);
+    return;
+  }
+  callback.Run(success, ProtocolVersion::UNKNOWN);
 }
 
 }  // namespace device