lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-04-02 09:46:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

agent: fix argument parser

Browse Source
This commit is contained in:
Roman Zeyde
2015-07-04 10:47:32 +03:00
parent 93bac8b011
commit 8273cc89f8
2 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
sshagent/trezor.py
View File

@@ -51,9 +51,11 @@ class Client(object):
self.client.clear_session()
self.client.close()
def get_public_key(self, label):
identity = self.factory.parse_identity(label)
label = _identity_to_string(identity) # update label after parsing
def get_identity(self, label):
return self.factory.parse_identity(label)
def get_public_key(self, identity):
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)
@@ -64,13 +66,12 @@ class Client(object):
def sign_ssh_challenge(self, label, blob):
identity = self.factory.parse_identity(label)
msg = _parse_ssh_blob(blob)
request = 'user: "{user}"'.format(**msg)
log.info('confirm %s connection to %r using Trezor...',
request, label)
log.info('confirm user %s connection to %r using Trezor...',
msg['user'], label)
s = self.client.sign_identity(identity=identity,
challenge_hidden=blob,
challenge_visual=request,
challenge_visual='',
ecdsa_curve_name=self.curve_name)
assert len(s.signature) == 65
assert s.signature[0] == b'\x00'