| Server IP : 153.92.12.84 / Your IP : 216.73.217.47 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/./self/root/opt/gsutil/third_party/pyu2f/pyu2f/tests/ |
Upload File : |
# Copyright 2016 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for pyu2f.u2f."""
import sys
import mock
from pyu2f import errors
from pyu2f import model
from pyu2f import u2f
if sys.version_info[:2] < (2, 7):
import unittest2 as unittest # pylint: disable=g-import-not-at-top
else:
import unittest # pylint: disable=g-import-not-at-top
class U2fTest(unittest.TestCase):
def testRegisterSuccessWithTUP(self):
mock_sk = mock.MagicMock()
mock_sk.CmdRegister.side_effect = [errors.TUPRequiredError, 'regdata']
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
resp = u2f_api.Register('testapp', b'ABCD', [])
self.assertEqual(mock_sk.CmdRegister.call_count, 2)
self.assertEqual(mock_sk.CmdWink.call_count, 1)
self.assertEqual(resp.client_data.raw_server_challenge, b'ABCD')
self.assertEqual(resp.client_data.typ, 'navigator.id.finishEnrollment')
self.assertEqual(resp.registration_data, 'regdata')
def testRegisterSuccessWithPreviousKeys(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = errors.InvalidKeyHandleError
mock_sk.CmdRegister.side_effect = [errors.TUPRequiredError, 'regdata']
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
resp = u2f_api.Register('testapp', b'ABCD', [model.RegisteredKey('khA')])
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 1)
# Should be "Check only"
self.assertTrue(mock_sk.CmdAuthenticate.call_args[0][3])
self.assertEqual(mock_sk.CmdRegister.call_count, 2)
self.assertEqual(mock_sk.CmdWink.call_count, 1)
self.assertEqual(resp.client_data.raw_server_challenge, b'ABCD')
self.assertEqual(resp.client_data.typ, 'navigator.id.finishEnrollment')
self.assertEqual(resp.registration_data, 'regdata')
def testRegisterFailAlreadyRegistered(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = errors.TUPRequiredError
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Register('testapp', b'ABCD', [model.RegisteredKey('khA')])
self.assertEqual(cm.exception.code, errors.U2FError.DEVICE_INELIGIBLE)
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 1)
# Should be "Check only"
self.assertTrue(mock_sk.CmdAuthenticate.call_args[0][3])
self.assertEqual(mock_sk.CmdRegister.call_count, 0)
self.assertEqual(mock_sk.CmdWink.call_count, 0)
def testRegisterTimeout(self):
mock_sk = mock.MagicMock()
mock_sk.CmdRegister.side_effect = errors.TUPRequiredError
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
# Speed up the test by mocking out sleep to do nothing
with mock.patch.object(u2f, 'time') as _:
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Register('testapp', b'ABCD', [])
self.assertEqual(cm.exception.code, errors.U2FError.TIMEOUT)
self.assertEqual(mock_sk.CmdRegister.call_count, 30)
self.assertEqual(mock_sk.CmdWink.call_count, 30)
def testRegisterError(self):
mock_sk = mock.MagicMock()
mock_sk.CmdRegister.side_effect = errors.ApduError(0xff, 0xff)
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Register('testapp', b'ABCD', [])
self.assertEqual(cm.exception.code, errors.U2FError.BAD_REQUEST)
self.assertEqual(cm.exception.cause.sw1, 0xff)
self.assertEqual(cm.exception.cause.sw2, 0xff)
self.assertEqual(mock_sk.CmdRegister.call_count, 1)
self.assertEqual(mock_sk.CmdWink.call_count, 0)
def testAuthenticateSuccessWithTUP(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = [errors.TUPRequiredError, 'signature']
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
resp = u2f_api.Authenticate('testapp', b'ABCD',
[model.RegisteredKey('khA')])
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 2)
self.assertEqual(mock_sk.CmdWink.call_count, 1)
self.assertEqual(resp.key_handle, 'khA')
self.assertEqual(resp.client_data.raw_server_challenge, b'ABCD')
self.assertEqual(resp.client_data.typ, 'navigator.id.getAssertion')
self.assertEqual(resp.signature_data, 'signature')
def testAuthenticateSuccessSkipInvalidKey(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = [errors.InvalidKeyHandleError,
'signature']
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
resp = u2f_api.Authenticate(
'testapp', b'ABCD',
[model.RegisteredKey('khA'), model.RegisteredKey('khB')])
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 2)
self.assertEqual(mock_sk.CmdWink.call_count, 0)
self.assertEqual(resp.key_handle, 'khB')
self.assertEqual(resp.client_data.raw_server_challenge, b'ABCD')
self.assertEqual(resp.client_data.typ, 'navigator.id.getAssertion')
self.assertEqual(resp.signature_data, 'signature')
def testAuthenticateSuccessSkipInvalidVersion(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.return_value = 'signature'
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
resp = u2f_api.Authenticate('testapp',
b'ABCD',
[model.RegisteredKey('khA',
version='U2F_V3'),
model.RegisteredKey('khB')])
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 1)
self.assertEqual(mock_sk.CmdWink.call_count, 0)
self.assertEqual(resp.key_handle, 'khB')
self.assertEqual(resp.client_data.raw_server_challenge, b'ABCD')
self.assertEqual(resp.client_data.typ, 'navigator.id.getAssertion')
self.assertEqual(resp.signature_data, 'signature')
def testAuthenticateTimeout(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = errors.TUPRequiredError
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
# Speed up the test by mocking out sleep to do nothing
with mock.patch.object(u2f, 'time') as _:
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Authenticate('testapp', b'ABCD', [model.RegisteredKey('khA')])
self.assertEqual(cm.exception.code, errors.U2FError.TIMEOUT)
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 30)
self.assertEqual(mock_sk.CmdWink.call_count, 30)
def testAuthenticateAllKeysInvalid(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = errors.InvalidKeyHandleError
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Authenticate('testapp', b'ABCD',
[model.RegisteredKey('khA'),
model.RegisteredKey('khB')])
self.assertEqual(cm.exception.code, errors.U2FError.DEVICE_INELIGIBLE)
u2f_api = u2f.U2FInterface(mock_sk)
def testAuthenticateError(self):
mock_sk = mock.MagicMock()
mock_sk.CmdAuthenticate.side_effect = errors.ApduError(0xff, 0xff)
mock_sk.CmdVersion.return_value = b'U2F_V2'
u2f_api = u2f.U2FInterface(mock_sk)
with self.assertRaises(errors.U2FError) as cm:
u2f_api.Authenticate('testapp', b'ABCD', [model.RegisteredKey('khA')])
self.assertEqual(cm.exception.code, errors.U2FError.BAD_REQUEST)
self.assertEqual(cm.exception.cause.sw1, 0xff)
self.assertEqual(cm.exception.cause.sw2, 0xff)
self.assertEqual(mock_sk.CmdAuthenticate.call_count, 1)
self.assertEqual(mock_sk.CmdWink.call_count, 0)
if __name__ == '__main__':
unittest.main()