From dd93dea0a975ddfa098f9767466fca9af5514b96 Mon Sep 17 00:00:00 2001 From: Roman Zeyde Date: Tue, 18 Aug 2015 17:03:28 +0300 Subject: [PATCH] trezor: set NIST256P1 curve when needed --- sshagent/trezor/client.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/sshagent/trezor/client.py b/sshagent/trezor/client.py index 3101ee7..a0bc9ed 100644 --- a/sshagent/trezor/client.py +++ b/sshagent/trezor/client.py @@ -15,8 +15,6 @@ log = logging.getLogger(__name__) class Client(object): - curve_name = 'nist256p1' - def __init__(self, factory=trezor_library): self.factory = factory self.client = self.factory.client() @@ -48,7 +46,8 @@ class Client(object): label = _identity_to_string(identity) log.info('getting "%s" public key from Trezor...', label) addr = _get_address(identity) - node = self.client.get_public_node(addr, self.curve_name) + node = self.client.get_public_node(n=addr, + ecdsa_curve_name='nist256p1') pubkey = node.node.public_key return formats.export_public_key(pubkey=pubkey, label=label) @@ -65,7 +64,7 @@ class Client(object): result = self.client.sign_identity(identity=identity, challenge_hidden=blob, challenge_visual=visual, - ecdsa_curve_name=self.curve_name) + ecdsa_curve_name='nist256p1') verifying_key = formats.decompress_pubkey(result.public_key) public_key_blob = formats.serialize_verifying_key(verifying_key) assert public_key_blob == msg['public_key']['blob']