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.
109 lines
4.3 KiB
Python
109 lines
4.3 KiB
Python
#!/usr/bin/env python
|
|
"""Create signatures and export public keys for GPG using TREZOR."""
|
|
import argparse
|
|
import contextlib
|
|
import io
|
|
import logging
|
|
import sys
|
|
import time
|
|
|
|
import semver
|
|
|
|
from . import agent, decode, device, encode, keyring, protocol
|
|
from .. import formats, server, util
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def key_exists(user_id):
|
|
"""Return True iff there is a GPG key with specified user ID."""
|
|
for p in decode.parse_packets(io.BytesIO(keyring.export_public_keys())):
|
|
if p['type'] == 'user_id' and p['value'] == user_id:
|
|
return True
|
|
return False
|
|
|
|
|
|
def run_create(args):
|
|
"""Generate a new pubkey for a new/existing GPG identity."""
|
|
log.warning('NOTE: in order to re-generate the exact same GPG key later, '
|
|
'run this command with "--time=%d" commandline flag (to set '
|
|
'the timestamp of the GPG key manually).', args.time)
|
|
conn = device.HardwareSigner(user_id=args.user_id,
|
|
curve_name=args.ecdsa_curve)
|
|
verifying_key = conn.pubkey(ecdh=False)
|
|
decryption_key = conn.pubkey(ecdh=True)
|
|
|
|
if key_exists(args.user_id): # add as subkey
|
|
log.info('adding %s GPG subkey for "%s" to existing key',
|
|
args.ecdsa_curve, args.user_id)
|
|
# subkey for signing
|
|
signing_key = protocol.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=False)
|
|
# subkey for encryption
|
|
encryption_key = protocol.PublicKey(
|
|
curve_name=formats.get_ecdh_curve_name(args.ecdsa_curve),
|
|
created=args.time, verifying_key=decryption_key, ecdh=True)
|
|
primary_bytes = keyring.export_public_key(args.user_id)
|
|
result = encode.create_subkey(primary_bytes=primary_bytes,
|
|
subkey=signing_key,
|
|
signer_func=conn.sign)
|
|
result = encode.create_subkey(primary_bytes=result,
|
|
subkey=encryption_key,
|
|
signer_func=conn.sign)
|
|
else: # add as primary
|
|
log.info('creating new %s GPG primary key for "%s"',
|
|
args.ecdsa_curve, args.user_id)
|
|
# primary key for signing
|
|
primary = protocol.PublicKey(
|
|
curve_name=args.ecdsa_curve, created=args.time,
|
|
verifying_key=verifying_key, ecdh=False)
|
|
# subkey for encryption
|
|
subkey = protocol.PublicKey(
|
|
curve_name=formats.get_ecdh_curve_name(args.ecdsa_curve),
|
|
created=args.time, verifying_key=decryption_key, ecdh=True)
|
|
|
|
result = encode.create_primary(user_id=args.user_id,
|
|
pubkey=primary,
|
|
signer_func=conn.sign)
|
|
result = encode.create_subkey(primary_bytes=result,
|
|
subkey=subkey,
|
|
signer_func=conn.sign)
|
|
|
|
sys.stdout.write(protocol.armor(result, 'PUBLIC KEY BLOCK'))
|
|
|
|
|
|
def main_create():
|
|
"""Main function for GPG identity creation."""
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument('user_id')
|
|
p.add_argument('-e', '--ecdsa-curve', default='nist256p1')
|
|
p.add_argument('-t', '--time', type=int, default=int(time.time()))
|
|
p.add_argument('-v', '--verbose', default=0, action='count')
|
|
|
|
args = p.parse_args()
|
|
util.setup_logging(verbosity=args.verbose)
|
|
log.warning('This GPG tool is still in EXPERIMENTAL mode, '
|
|
'so please note that the API and features may '
|
|
'change without backwards compatibility!')
|
|
|
|
existing_gpg = keyring.gpg_version().decode('ascii')
|
|
required_gpg = '>=2.1.15'
|
|
if semver.match(existing_gpg, required_gpg):
|
|
run_create(args)
|
|
else:
|
|
log.error('Existing gpg2 has version "%s" (%s required)',
|
|
existing_gpg, required_gpg)
|
|
|
|
|
|
def main_agent():
|
|
"""Run a simple GPG-agent server."""
|
|
sock_path = keyring.get_agent_sock_path()
|
|
with server.unix_domain_socket_server(sock_path) as sock:
|
|
for conn in agent.yield_connections(sock):
|
|
with contextlib.closing(conn):
|
|
try:
|
|
agent.handle_connection(conn)
|
|
except Exception as e: # pylint: disable=broad-except
|
|
log.exception('gpg-agent failed: %s', e)
|