diff --git a/sshagent/tests/test_trezor.py b/sshagent/tests/test_trezor.py index 202a7b5..e9a191e 100644 --- a/sshagent/tests/test_trezor.py +++ b/sshagent/tests/test_trezor.py @@ -1,4 +1,5 @@ from ..trezor import client +from .. import formats import mock @@ -59,6 +60,20 @@ class FactoryMock(object): return result +BLOB = (b'\x00\x00\x00 \xce\xe0\xc9\xd5\xceu/\xe8\xc5\xf2\xbfR+x\xa1\xcf\xb0' + b'\x8e;R\xd3)m\x96\x1b\xb4\xd8s\xf1\x99\x16\xaa2\x00\x00\x00\x05roman' + b'\x00\x00\x00\x0essh-connection\x00\x00\x00\tpublickey' + b'\x01\x00\x00\x00\x13ecdsa-sha2-nistp256\x00\x00\x00h\x00\x00\x00' + b'\x13ecdsa-sha2-nistp256\x00\x00\x00\x08nistp256\x00\x00\x00A' + b'\x04\xd8(\xb5\xa6`\xbet0\x95\xac:[;]\xdc,\xbd\xdc?\xd7\xc0\xec' + b'\xdd\xbc+\xfar~\x9dAis4\xc1\x10yeT~\x1b\xeb\x1aX\xd1\xd9\x9f\xc21' + b'\x13\x8dc\xa7\xa3\x07\xefO\x9e\x95\x0e>\xec\xd8\xaa/') + +SIG = (b'\x00R\x19T\xf2\x84$\xef#\x0e\xee\x04X\xc6\xc3\x99T`\xd1\xd8\xf7!' + b'\x862@cx\xb8\xb9i@1\x1b3#\x938\x86]\x97*Y\xb2\x02Xa\xdf@\xecK' + b'\xdc\xf0H\xab\xa8\xac\xa7? \x8f=C\x88N\xe2') + + def test_client(): c = client.Client(factory=FactoryMock) ident = c.get_identity(label='localhost:22', protocol='ssh') @@ -70,3 +85,22 @@ def test_client(): with c: assert c.get_public_key(ident) == PUBKEY_TEXT + + def _sign_identity(identity, challenge_hidden, + challenge_visual, ecdsa_curve_name): + assert identity is ident + assert challenge_hidden == BLOB + assert challenge_visual == identity.path + assert ecdsa_curve_name == 'nist256p1' + + result = mock.Mock(spec=[]) + result.public_key = PUBKEY + result.signature = SIG + return result + + c.client.sign_identity = _sign_identity + signature = c.sign_ssh_challenge(identity=ident, blob=BLOB) + + key = formats.import_public_key(PUBKEY_TEXT) + assert key['verifying_key'].verify(signature=signature, data=BLOB, + sigdecode=lambda sig, _: sig)