| // 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 <list> |
| |
| #include "base/memory/ptr_util.h" |
| #include "base/run_loop.h" |
| #include "base/test/test_io_thread.h" |
| #include "device/base/mock_device_client.h" |
| #include "device/hid/mock_hid_service.h" |
| #include "device/test/test_device_client.h" |
| #include "device/test/usb_test_gadget.h" |
| #include "device/u2f/u2f_hid_device.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "u2f_request.h" |
| |
| namespace device { |
| namespace { |
| #if defined(OS_MACOSX) |
| const uint64_t kTestDeviceId0 = 42; |
| const uint64_t kTestDeviceId1 = 43; |
| const uint64_t kTestDeviceId2 = 44; |
| #else |
| const char* kTestDeviceId0 = "device0"; |
| const char* kTestDeviceId1 = "device1"; |
| const char* kTestDeviceId2 = "device2"; |
| #endif |
| |
| class FakeU2fRequest : public U2fRequest { |
| public: |
| FakeU2fRequest(const ResponseCallback& cb) : U2fRequest(cb) {} |
| ~FakeU2fRequest() override {} |
| |
| void TryDevice() override { |
| cb_.Run(U2fReturnCode::SUCCESS, std::vector<uint8_t>()); |
| } |
| }; |
| } // namespace |
| |
| class TestResponseCallback { |
| public: |
| TestResponseCallback() |
| : callback_(base::Bind(&TestResponseCallback::ReceivedCallback, |
| base::Unretained(this))) {} |
| ~TestResponseCallback() {} |
| |
| void ReceivedCallback(U2fReturnCode status, std::vector<uint8_t> data) { |
| closure_.Run(); |
| } |
| |
| void WaitForCallback() { |
| closure_ = run_loop_.QuitClosure(); |
| run_loop_.Run(); |
| } |
| |
| U2fRequest::ResponseCallback& callback() { return callback_; } |
| |
| private: |
| base::Closure closure_; |
| U2fRequest::ResponseCallback callback_; |
| base::RunLoop run_loop_; |
| }; |
| |
| class U2fRequestTest : public testing::Test { |
| public: |
| U2fRequestTest() : io_thread_(base::TestIOThread::kAutoStart) {} |
| |
| protected: |
| base::MessageLoopForUI message_loop_; |
| base::TestIOThread io_thread_; |
| device::MockDeviceClient device_client_; |
| }; |
| |
| TEST_F(U2fRequestTest, TestAddRemoveDevice) { |
| MockHidService* hid_service = device_client_.hid_service(); |
| HidCollectionInfo c_info; |
| hid_service->FirstEnumerationComplete(); |
| |
| TestResponseCallback cb; |
| FakeU2fRequest request(cb.callback()); |
| request.Enumerate(); |
| EXPECT_EQ(static_cast<size_t>(0), request.devices_.size()); |
| |
| // Add one U2F device |
| c_info.usage = HidUsageAndPage(1, static_cast<HidUsageAndPage::Page>(0xf1d0)); |
| scoped_refptr<HidDeviceInfo> u2f_device = make_scoped_refptr( |
| new HidDeviceInfo(kTestDeviceId0, 0, 0, "Test Fido Device", "123FIDO", |
| kHIDBusTypeUSB, c_info, 64, 64, 0)); |
| hid_service->AddDevice(u2f_device); |
| EXPECT_EQ(static_cast<size_t>(1), request.devices_.size()); |
| |
| // Add one non-U2F device. Verify that it is not added to our device list. |
| scoped_refptr<HidDeviceInfo> other_device = make_scoped_refptr( |
| new HidDeviceInfo(kTestDeviceId2, 0, 0, "Other Device", "OtherDevice", |
| kHIDBusTypeUSB, std::vector<uint8_t>())); |
| hid_service->AddDevice(other_device); |
| EXPECT_EQ(static_cast<size_t>(1), request.devices_.size()); |
| |
| // Remove the non-U2F device and verify that device list was unchanged. |
| hid_service->RemoveDevice(kTestDeviceId2); |
| EXPECT_EQ(static_cast<size_t>(1), request.devices_.size()); |
| |
| // Remove the U2F device and verify that device list is empty. |
| hid_service->RemoveDevice(kTestDeviceId0); |
| EXPECT_EQ(static_cast<size_t>(0), request.devices_.size()); |
| } |
| |
| TEST_F(U2fRequestTest, TestIterateDevice) { |
| TestResponseCallback cb; |
| FakeU2fRequest request(cb.callback()); |
| HidCollectionInfo c_info; |
| // Add one U2F device and one non-U2f device |
| c_info.usage = HidUsageAndPage(1, static_cast<HidUsageAndPage::Page>(0xf1d0)); |
| scoped_refptr<HidDeviceInfo> device0 = make_scoped_refptr( |
| new HidDeviceInfo(kTestDeviceId0, 0, 0, "Test Fido Device", "123FIDO", |
| kHIDBusTypeUSB, c_info, 64, 64, 0)); |
| request.devices_.push_back(base::MakeUnique<U2fHidDevice>(device0)); |
| scoped_refptr<HidDeviceInfo> device1 = make_scoped_refptr( |
| new HidDeviceInfo(kTestDeviceId1, 0, 0, "Test Fido Device", "123FIDO", |
| kHIDBusTypeUSB, c_info, 64, 64, 0)); |
| request.devices_.push_back(base::MakeUnique<U2fHidDevice>(device1)); |
| |
| // Move first device to current |
| request.IterateDevice(); |
| ASSERT_NE(nullptr, request.current_device_); |
| EXPECT_EQ(static_cast<size_t>(1), request.devices_.size()); |
| |
| // Move second device to current, first to attempted |
| request.IterateDevice(); |
| ASSERT_NE(nullptr, request.current_device_); |
| EXPECT_EQ(static_cast<size_t>(1), request.attempted_devices_.size()); |
| |
| // Move second device from current to attempted, move attempted to devices as |
| // all devices have been attempted |
| request.IterateDevice(); |
| ASSERT_EQ(nullptr, request.current_device_); |
| EXPECT_EQ(static_cast<size_t>(2), request.devices_.size()); |
| EXPECT_EQ(static_cast<size_t>(0), request.attempted_devices_.size()); |
| } |
| |
| TEST_F(U2fRequestTest, TestBasicMachine) { |
| MockHidService* hid_service = device_client_.hid_service(); |
| hid_service->FirstEnumerationComplete(); |
| TestResponseCallback cb; |
| FakeU2fRequest request(cb.callback()); |
| request.Start(); |
| // Add one U2F device |
| HidCollectionInfo c_info; |
| c_info.usage = HidUsageAndPage(1, static_cast<HidUsageAndPage::Page>(0xf1d0)); |
| scoped_refptr<HidDeviceInfo> u2f_device = make_scoped_refptr( |
| new HidDeviceInfo(kTestDeviceId0, 0, 0, "Test Fido Device", "123FIDO", |
| kHIDBusTypeUSB, c_info, 64, 64, 0)); |
| hid_service->AddDevice(u2f_device); |
| cb.WaitForCallback(); |
| EXPECT_EQ(U2fRequest::State::BUSY, request.state_); |
| } |
| |
| } // namespace device |