jacobpascual/trezor-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

trezor: explicitly specify SSH curve

Browse Source
This commit is contained in:
Roman Zeyde
2015-06-23 17:53:59 +03:00
parent 3c1c1d0e0c
commit ef56779461
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sshagent/trezor.py
View File

@@ -29,6 +29,8 @@ class TrezorLibrary(object):
class Client(object):
curve_name = 'nist256p1'
def __init__(self, factory=TrezorLibrary):
self.factory = factory
self.client = self.factory.client()
@@ -51,7 +53,7 @@ class Client(object):
def get_public_key(self, label):
addr = _get_address(self.factory.identity(label))
log.info('getting %r SSH public key from Trezor...', label)
node = self.client.get_public_node(addr)
node = self.client.get_public_node(addr, self.curve_name)
return node.node.public_key
def sign_ssh_challenge(self, label, blob):
@@ -63,7 +65,8 @@ class Client(object):
request, label)
s = self.client.sign_identity(identity=ident,
challenge_hidden=blob,
challenge_visual=request)
challenge_visual=request,
ecdsa_curve_name=self.curve_name)
assert len(s.signature) == 65
assert s.signature[0] == b'\x00'