You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import sys
|
|
import argparse
|
|
|
|
from . import trezor
|
|
from . import server
|
|
from . import formats
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
fmt = '%(asctime)s %(levelname)-12s %(message)-100s [%(filename)s]'
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument('-k', '--key-label',
|
|
metavar='LABEL', dest='labels', action='append', default=[])
|
|
p.add_argument('-v', '--verbose', action='count', default=0)
|
|
p.add_argument('command', type=str, nargs='*')
|
|
args = p.parse_args()
|
|
|
|
verbosity = [logging.WARNING, logging.INFO, logging.DEBUG]
|
|
level = verbosity[min(args.verbose, len(verbosity) - 1)]
|
|
logging.basicConfig(level=level, format=fmt)
|
|
|
|
client = trezor.Client()
|
|
|
|
key_files = []
|
|
for label in args.labels:
|
|
pubkey = client.get_public_key(label=label)
|
|
key_file = formats.export_public_key(pubkey=pubkey, label=label)
|
|
key_files.append(key_file)
|
|
|
|
if not args.command:
|
|
sys.stdout.write(''.join(key_files))
|
|
return
|
|
|
|
signer = client.sign_ssh_challenge
|
|
|
|
ret = -1
|
|
try:
|
|
ret = server.serve(
|
|
key_files=key_files,
|
|
command=args.command,
|
|
signer=signer)
|
|
log.info('exitcode: %d', ret)
|
|
except KeyboardInterrupt:
|
|
log.info('server stopped')
|
|
sys.exit(ret)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|