mirror of
https://github.com/romanz/amodem.git
synced 2026-03-04 13:49:46 +08:00
93 lines
3.3 KiB
Python
Executable File
93 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""Create signatures and export public keys for GPG using TREZOR."""
|
|
import argparse
|
|
import contextlib
|
|
import logging
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
from . import agent, encode, keyring, proto
|
|
from .. import server
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def run_create(args):
|
|
"""Generate a new pubkey for a new/existing GPG identity."""
|
|
user_id = os.environ['TREZOR_GPG_USER_ID']
|
|
conn = encode.HardwareSigner(user_id=user_id,
|
|
curve_name=args.ecdsa_curve)
|
|
verifying_key = conn.pubkey()
|
|
|
|
if args.subkey:
|
|
primary_bytes = keyring.export_public_key(user_id=user_id)
|
|
# subkey for signing
|
|
signing_key = proto.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=False)
|
|
# subkey for encryption
|
|
encryption_key = proto.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=True)
|
|
result = encode.create_subkey(primary_bytes=primary_bytes,
|
|
pubkey=signing_key,
|
|
signer_func=conn.sign)
|
|
result = encode.create_subkey(primary_bytes=result,
|
|
pubkey=encryption_key,
|
|
signer_func=conn.sign)
|
|
else:
|
|
# primary key for signing
|
|
primary = proto.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=False)
|
|
# subkey for encryption
|
|
subkey = proto.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=True)
|
|
|
|
result = encode.create_primary(user_id=user_id,
|
|
pubkey=primary,
|
|
signer_func=conn.sign)
|
|
result = encode.create_subkey(primary_bytes=result,
|
|
pubkey=subkey,
|
|
signer_func=conn.sign)
|
|
|
|
sys.stdout.write(proto.armor(result, 'PUBLIC KEY BLOCK'))
|
|
|
|
|
|
def run_agent(_):
|
|
"""Run a simple GPG-agent server."""
|
|
sock_path = os.path.expanduser('~/.gnupg/S.gpg-agent')
|
|
with server.unix_domain_socket_server(sock_path) as sock:
|
|
for conn in agent.yield_connections(sock):
|
|
with contextlib.closing(conn):
|
|
agent.handle_connection(conn)
|
|
|
|
|
|
def main():
|
|
"""Main function."""
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument('-v', '--verbose', action='store_true', default=False)
|
|
subparsers = p.add_subparsers()
|
|
subparsers.required = True
|
|
subparsers.dest = 'command'
|
|
|
|
create_cmd = subparsers.add_parser('create')
|
|
create_cmd.add_argument('-s', '--subkey', action='store_true', default=False)
|
|
create_cmd.add_argument('-e', '--ecdsa-curve', default='nist256p1')
|
|
create_cmd.add_argument('-t', '--time', type=int, default=int(time.time()))
|
|
create_cmd.set_defaults(run=run_create)
|
|
|
|
agent_cmd = subparsers.add_parser('agent')
|
|
agent_cmd.set_defaults(run=run_agent)
|
|
|
|
args = p.parse_args()
|
|
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO,
|
|
format='%(asctime)s %(levelname)-10s %(message)s')
|
|
args.run(args)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|