687 lines
31 KiB
Python
Executable File
687 lines
31 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# link: https://git.p2p.legal/aya/dpgpid/
|
|
# desc: generate ed25519 keys suitables for duniter or ipfs
|
|
|
|
# Copyleft 2022 Yann Autissier <aya@asycn.io>
|
|
# all crypto science belongs to Pascal Engélibert <tuxmain@zettascript.org>
|
|
# coming from files available at https://git.p2p.legal/qo-op/Astroport.ONE/tools
|
|
# gpgme stuff has been provided by Ben McGinnes
|
|
# and comes from http://files.au.adversary.org/crypto/gpgme-python-howto.html
|
|
# gpg key extraction is taken from work of Simon Vareille available at
|
|
# https://gist.github.com/SimonVareille/fda49baf5f3e15b5c88e25560aeb2822
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import argparse
|
|
import base58
|
|
import base64
|
|
import configparser
|
|
from cryptography.hazmat.primitives.asymmetric import ed25519
|
|
from cryptography.hazmat.primitives import serialization
|
|
import duniterpy.key
|
|
import gpg
|
|
import logging as log
|
|
import nacl.bindings
|
|
import nacl.encoding
|
|
import pgpy
|
|
import pynentry
|
|
from SecureBytes import clearmem
|
|
import os
|
|
import re
|
|
import struct
|
|
import sys
|
|
import time
|
|
import warnings
|
|
|
|
__version__='0.0.4'
|
|
|
|
class keygen:
|
|
def __init__(self):
|
|
# desc: generate ed25519 keys suitables for duniter or ipfs
|
|
self.parser = argparse.ArgumentParser(description="""
|
|
Generate ed25519 keys suitables for duniter and ipfs.
|
|
It converts an ed25519 key, duniter username/password, or a GPG key, to
|
|
a duniter wallet or an IPFS PeerID/PrivateKEY.""")
|
|
self.parser.add_argument(
|
|
"-d",
|
|
"--debug",
|
|
action="store_true",
|
|
help="show debug informations (WARNING: including SECRET KEY value)",
|
|
)
|
|
self.parser.add_argument(
|
|
"-f",
|
|
"--format",
|
|
dest="format",
|
|
default=None,
|
|
help="output file format: [ewif|nacl|pb2|pem|pubsec|seed|wif], default: pem (pkcs8)",
|
|
)
|
|
self.parser.add_argument(
|
|
"-g",
|
|
"--gpg",
|
|
action="store_true",
|
|
help="use gpg key with uid matched by username",
|
|
)
|
|
self.parser.add_argument(
|
|
"-i",
|
|
"--input",
|
|
dest="input",
|
|
default=None,
|
|
help="read ed25519 key from file INPUT, autodetect format: [credentials|ewif|libnacl|mnemonic|pubsec|seedhex|wif]",
|
|
)
|
|
self.parser.add_argument(
|
|
"-k",
|
|
"--keys",
|
|
action="store_true",
|
|
help="show public and secret keys",
|
|
)
|
|
self.parser.add_argument(
|
|
"-m",
|
|
"--mnemonic",
|
|
action="store_true",
|
|
help="use username as a DUBP mnemonic passphrase",
|
|
)
|
|
self.parser.add_argument(
|
|
"-o",
|
|
"--output",
|
|
dest="output",
|
|
default=None,
|
|
help="write ed25519 key to file OUTPUT",
|
|
)
|
|
self.parser.add_argument(
|
|
"-p",
|
|
"--prefix",
|
|
action="store_true",
|
|
help="prefix output text with key type",
|
|
)
|
|
self.parser.add_argument(
|
|
"-q",
|
|
"--quiet",
|
|
action="store_true",
|
|
help="show only errors",
|
|
)
|
|
self.parser.add_argument(
|
|
"-s",
|
|
"--secret",
|
|
action="store_true",
|
|
help="show secret key",
|
|
)
|
|
self.parser.add_argument(
|
|
"-t",
|
|
"--type",
|
|
dest="type",
|
|
default="base58",
|
|
help="output text format: [base58|base64|duniter|ipfs], default: base58",
|
|
)
|
|
self.parser.add_argument(
|
|
"-v",
|
|
"--verbose",
|
|
action="store_true",
|
|
help="show more informations",
|
|
)
|
|
self.parser.add_argument(
|
|
"--version",
|
|
action="store_true",
|
|
help="show version and exit",
|
|
)
|
|
self.parser.add_argument(
|
|
'username',
|
|
nargs="?",
|
|
)
|
|
self.parser.add_argument(
|
|
'password',
|
|
nargs="?",
|
|
)
|
|
|
|
def _check_args(self, args):
|
|
log.debug("keygen._check_args(%s)" % args)
|
|
if self.input is None:
|
|
if self.password is None:
|
|
if self.username is None:
|
|
self.parser.error(f"keygen requires an input file or username args")
|
|
if self.format not in [None, 'ewif', 'nacl', 'pb2', 'pem', 'pubsec', 'seed', 'wif']:
|
|
self.parser.error(f"format not valid")
|
|
|
|
def _cleanup(self):
|
|
log.debug("keygen._cleanup()")
|
|
if hasattr(self, 'duniterpy'):
|
|
if hasattr(self.duniterpy, 'seed') and self.duniterpy.seed:
|
|
clearmem(self.duniterpy.seed)
|
|
log.debug("cleared: keygen.duniterpy.seed")
|
|
if hasattr(self.duniterpy, 'sk') and self.duniterpy.sk:
|
|
clearmem(self.duniterpy.sk)
|
|
log.debug("cleared: keygen.duniterpy.sk")
|
|
if hasattr(self, 'ed25519_secret_base58') and self.ed25519_secret_base58:
|
|
clearmem(self.ed25519_secret_base58)
|
|
log.debug("cleared: keygen.ed25519_secret_base58")
|
|
if hasattr(self, 'ed25519_secret_base64') and self.ed25519_secret_base64:
|
|
clearmem(self.ed25519_secret_base64)
|
|
log.debug("cleared: keygen.ed25519_secret_base64")
|
|
if hasattr(self, 'ed25519_secret_bytes') and self.ed25519_secret_bytes:
|
|
clearmem(self.ed25519_secret_bytes)
|
|
log.debug("cleared: keygen.ed25519_secret_bytes")
|
|
if hasattr(self, 'ed25519_secret_pem_pkcs8') and self.ed25519_secret_pem_pkcs8:
|
|
clearmem(self.ed25519_secret_pem_pkcs8)
|
|
log.debug("cleared: keygen.ed25515_secret_pem_pkcs8")
|
|
if hasattr(self, 'ed25519_secret_protobuf2') and self.ed25519_secret_protobuf2:
|
|
clearmem(self.ed25519_secret_protobuf2)
|
|
log.debug("cleared: keygen.ed25515_secret_protobuf2")
|
|
if hasattr(self, 'ed25519_seed_bytes') and self.ed25519_seed_bytes:
|
|
clearmem(self.ed25519_seed_bytes)
|
|
log.debug("cleared: keygen.ed25519_seed_bytes")
|
|
if hasattr(self, 'ipfs_privkey') and self.ipfs_privkey:
|
|
clearmem(self.ipfs_privkey)
|
|
log.debug("cleared: keygen.ipfs_privkey")
|
|
if hasattr(self, 'password') and self.password:
|
|
clearmem(self.password)
|
|
log.debug("cleared: keygen.password")
|
|
if hasattr(self, 'pgp_secret_armored') and self.pgp_secret_armored:
|
|
clearmem(self.pgp_secret_armored)
|
|
log.debug("cleared: keygen.pgp_secret_armored")
|
|
if hasattr(self, 'pgpy'):
|
|
if hasattr(self.pgpy._key.keymaterial, 'p') and self.pgpy._key.keymaterial.p and not isinstance(self.pgpy._key.keymaterial.p, pgpy.packet.fields.ECPoint):
|
|
clearmem(self.pgpy._key.keymaterial.p)
|
|
log.debug("cleared: keygen.pgpy._key.material.p")
|
|
if hasattr(self.pgpy._key.keymaterial, 'q') and self.pgpy._key.keymaterial.q:
|
|
clearmem(self.pgpy._key.keymaterial.q)
|
|
log.debug("cleared: keygen.pgpy._key.material.q")
|
|
if hasattr(self.pgpy._key.keymaterial, 's') and self.pgpy._key.keymaterial.s:
|
|
clearmem(self.pgpy._key.keymaterial.s)
|
|
log.debug("cleared: keygen.pgpy._key.material.s")
|
|
if hasattr(self, 'username') and self.username:
|
|
clearmem(self.username)
|
|
log.debug("cleared: keygen.username")
|
|
|
|
def _invalid_type(self):
|
|
log.debug("keygen._invalid_type()")
|
|
self.parser.error(f"type: {self.type} is not valid.")
|
|
|
|
def _load_config(self):
|
|
log.debug("keygen._load_config()")
|
|
self.config = configparser.RawConfigParser()
|
|
config_dir = os.path.join(os.environ.get('XDG_CONFIG_HOME', os.path.expanduser('~/.config')), 'dpgpid')
|
|
log.debug("config_dir=%s" % config_dir)
|
|
self.config.read( [config_dir + '/keygen.conf'] )
|
|
|
|
def _output(self, public_key, secret_key, public_key_prefix, secret_key_prefix):
|
|
log.debug("keygen._output()")
|
|
if self.output is None:
|
|
self._output_text(public_key, secret_key, public_key_prefix, secret_key_prefix)
|
|
else:
|
|
self._output_file()
|
|
os.chmod(self.output, 0o600)
|
|
self._cleanup()
|
|
|
|
def _output_file(self):
|
|
log.debug("keygen._output_file()")
|
|
if self.format == 'ewif':
|
|
if not hasattr(self, 'duniterpy'):
|
|
self.duniterpy_from_ed25519()
|
|
if not self.password:
|
|
with pynentry.PynEntry() as p:
|
|
p.description = f"""Data in EWIF file needs to be encrypted.
|
|
Please enter a password to encrypt seed.
|
|
"""
|
|
p.prompt = 'Passphrase:'
|
|
try:
|
|
self.password = p.get_pin()
|
|
except pynentry.PinEntryCancelled:
|
|
log.warning('Cancelled! Goodbye.')
|
|
self._cleanup()
|
|
exit(2)
|
|
self.duniterpy.save_ewif_file(self.output, self.password)
|
|
elif self.format == 'nacl':
|
|
if not hasattr(self, 'duniterpy'):
|
|
self.duniterpy_from_ed25519()
|
|
self.duniterpy.save_private_key(self.output)
|
|
elif self.format == 'pb2':
|
|
if not hasattr(self, 'ed25519_secret_protobuf2'):
|
|
self.protobuf2_from_ed25519()
|
|
with open(self.output, "wb") as fh:
|
|
fh.write(self.ed25519_secret_protobuf2)
|
|
elif self.format == 'pubsec':
|
|
if not hasattr(self, 'duniterpy'):
|
|
self.duniterpy_from_ed25519()
|
|
self.duniterpy.save_pubsec_file(self.output)
|
|
elif self.format == 'seed':
|
|
if not hasattr(self, 'duniterpy'):
|
|
self.duniterpy_from_ed25519()
|
|
self.duniterpy.save_seedhex_file(self.output)
|
|
elif self.format == 'wif':
|
|
if not hasattr(self, 'duniterpy'):
|
|
self.duniterpy_from_ed25519()
|
|
self.duniterpy.save_wif_file(self.output)
|
|
else:
|
|
if not hasattr(self, 'ed25519_secret_pem_pkcs8'):
|
|
self.pem_pkcs8_from_ed25519()
|
|
with open(self.output, "w") as fh:
|
|
fh.write(self.ed25519_secret_pem_pkcs8)
|
|
|
|
def _output_text(self, public_key, secret_key, public_key_prefix, secret_key_prefix):
|
|
log.debug("keygen._output_text()")
|
|
if self.keys or not self.secret:
|
|
print("%s" % ''.join([self.prefix * public_key_prefix, public_key]))
|
|
if self.keys or self.secret:
|
|
print("%s" % ''.join([self.prefix * secret_key_prefix, secret_key]))
|
|
|
|
def _run(self, argv):
|
|
args = self.parser.parse_args(argv)
|
|
vars(self).update(vars(args))
|
|
|
|
# display version
|
|
if args.version:
|
|
version()
|
|
sys.exit()
|
|
|
|
# define log format
|
|
log_format='%(asctime)s %(levelname)s: %(message)s'
|
|
log_datefmt='%Y/%m/%d %H:%M:%S'
|
|
if args.debug:
|
|
log_level='DEBUG'
|
|
elif args.quiet:
|
|
log_level='ERROR'
|
|
elif args.verbose:
|
|
log_level='INFO'
|
|
else:
|
|
log_level='WARNING'
|
|
log.basicConfig(format=log_format, datefmt=log_datefmt, level=log_level)
|
|
log.debug("keygen.run(%s)" % argv)
|
|
|
|
self._check_args(args)
|
|
self._load_config()
|
|
self.gpg = gpg.Context(armor=True, offline=True)
|
|
self.gpg.set_passphrase_cb(self.gpg_passphrase_cb)
|
|
self.ed25519(args)
|
|
method = getattr(self, f'do_{self.type}', self._invalid_type)
|
|
return method()
|
|
|
|
def base58_from_ed25519(self):
|
|
log.debug("keygen.base58_from_ed25519()")
|
|
self.ed25519_public_base58 = base58.b58encode(self.ed25519_public_bytes).decode('ascii')
|
|
log.debug("keygen.ed25519_public_base58=%s" % self.ed25519_public_base58)
|
|
self.ed25519_secret_base58 = base58.b58encode(self.ed25519_secret_bytes).decode('ascii')
|
|
log.debug("keygen.ed25519_secret_base58=%s" % self.ed25519_secret_base58)
|
|
|
|
def base64_from_ed25519(self):
|
|
log.debug("keygen.base64_from_ed25519()")
|
|
self.ed25519_public_base64 = base64.b64encode(self.ed25519_public_bytes).decode('ascii')
|
|
log.debug("keygen.ed25519_public_base64=%s" % self.ed25519_public_base64)
|
|
self.ed25519_secret_base64 = base64.b64encode(self.ed25519_secret_bytes).decode('ascii')
|
|
log.debug("keygen.ed25519_secret_base64=%s" % self.ed25519_secret_base64)
|
|
|
|
def base58_from_pubsec(self):
|
|
log.debug("keygen.base58_from_pubsec()")
|
|
for line in open(self.input, "r"):
|
|
if re.search("pub", line):
|
|
self.ed25519_public_base58 = line.replace('\n','').split(': ')[1]
|
|
elif re.search("sec", line):
|
|
self.ed25519_secret_base58 = line.replace('\n','').split(': ')[1]
|
|
|
|
def do_base58(self):
|
|
log.debug("keygen.do_base58()")
|
|
self.base58_from_ed25519()
|
|
self._output(self.ed25519_public_base58, self.ed25519_secret_base58, 'pub: ', 'sec: ')
|
|
|
|
def do_base64(self):
|
|
log.debug("keygen.do_base64()")
|
|
self.base64_from_ed25519()
|
|
self._output(self.ed25519_public_base64, self.ed25519_secret_base64, 'pub: ', 'sec: ')
|
|
|
|
def do_duniter(self):
|
|
if not self.format:
|
|
self.format = 'pubsec'
|
|
self.do_base58()
|
|
|
|
def do_ipfs(self):
|
|
log.debug("keygen.do_ipfs()")
|
|
self.protobuf2_from_ed25519()
|
|
self.ipfs_from_protobuf2()
|
|
self._output(self.ipfs_peerid, self.ipfs_privkey, 'PeerID: ', 'PrivKEY: ')
|
|
|
|
def duniterpy_from_credentials(self):
|
|
log.debug("keygen.duniterpy_from_credentials()")
|
|
scrypt_params = duniterpy.key.scrypt_params.ScryptParams(
|
|
int(self.config.get('scrypt', 'n')) if self.config.has_option('scrypt', 'n') else 4096,
|
|
int(self.config.get('scrypt', 'r')) if self.config.has_option('scrypt', 'r') else 16,
|
|
int(self.config.get('scrypt', 'p')) if self.config.has_option('scrypt', 'p') else 1,
|
|
int(self.config.get('scrypt', 'sl')) if self.config.has_option('scrypt', 'sl') else 32,
|
|
)
|
|
if not self.password:
|
|
with pynentry.PynEntry() as p:
|
|
p.description = f"""Please enter the passord for username "{self.username}"."""
|
|
p.prompt = 'Passsord:'
|
|
try:
|
|
self.password = p.get_pin()
|
|
except pynentry.PinEntryCancelled:
|
|
log.warning('Cancelled! Goodbye.')
|
|
self._cleanup()
|
|
exit(2)
|
|
self.duniterpy = duniterpy.key.SigningKey.from_credentials(
|
|
self.username,
|
|
self.password,
|
|
scrypt_params
|
|
)
|
|
log.debug("keygen.duniterpy.seed: %s" % self.duniterpy.seed)
|
|
|
|
def duniterpy_from_ed25519(self):
|
|
log.debug("keygen.duniterpy_from_ed25519()")
|
|
self.duniterpy = duniterpy.key.SigningKey(self.ed25519_secret_bytes[:32])
|
|
log.debug("keygen.duniterpy.seed: %s" % self.duniterpy.seed)
|
|
|
|
def duniterpy_from_file(self):
|
|
log.debug("keygen.duniterpy_from_file()")
|
|
try:
|
|
with open(self.input, 'r') as file:
|
|
lines = file.readlines()
|
|
if len(lines) > 0:
|
|
line = lines[0].strip()
|
|
regex_ewif = re.compile('^Type: EWIF$')
|
|
regex_nacl = re.compile('^\\s*{\\s*"priv":\\s*"[0-9a-fA-F]+",\\s*"verify":\\s*"[0-9a-fA-F]+",\\s*"sign":\\s*"[0-9a-fA-F]+"\\s*}$')
|
|
regex_pem = re.compile('^-----BEGIN PRIVATE KEY-----$')
|
|
regex_pubsec = re.compile('^Type: PubSec$')
|
|
regex_seed = re.compile('^[0-9a-fA-F]{64}$')
|
|
regex_ssb = re.compile('\\s*{\\s*"curve": "ed25519",\\s*"public": "(.+)\\.ed25519",\\s*"private":\\s*"(.+)\\.ed25519",\\s*"id":\\s*"@(.+).ed25519"\\s*}')
|
|
regex_wif = re.compile('^Type: WIF$')
|
|
if re.search(regex_ewif, line):
|
|
log.info("input file format detected: ewif")
|
|
if not self.password:
|
|
with pynentry.PynEntry() as p:
|
|
p.description = f"""Data in EWIF file needs to be decrypted.
|
|
Please enter a password to decrypt seed.
|
|
"""
|
|
p.prompt = 'Passphrase:'
|
|
try:
|
|
self.password = p.get_pin()
|
|
except pynentry.PinEntryCancelled:
|
|
log.warning('Cancelled! Goodbye.')
|
|
self._cleanup()
|
|
exit(2)
|
|
self.duniterpy = duniterpy.key.SigningKey.from_ewif_file(self.input, self.password)
|
|
elif re.search(regex_nacl, line):
|
|
log.info("input file format detected: nacl")
|
|
self.duniterpy = duniterpy.key.SigningKey.from_private_key(self.input)
|
|
elif re.search(regex_pem, line):
|
|
log.info("input file format detected: pem")
|
|
self.ed25519_secret_bytes = serialization.load_pem_private_key(''.join(lines).encode(), password=None).private_bytes(encoding=serialization.Encoding.Raw, format=serialization.PrivateFormat.Raw, encryption_algorithm=serialization.NoEncryption())
|
|
self.duniterpy_from_ed25519()
|
|
## at this stage, self.ed25519_secret_bytes contains only the 32 seed bytes and not the 32 seed bytes + 32 public bytes as it should
|
|
# we need to call self.ed25519_from_duniterpy() later to correctly build the self.ed25519_secret_bytes
|
|
elif re.search(regex_pubsec, line):
|
|
log.info("input file format detected: pubsec")
|
|
self.duniterpy = duniterpy.key.SigningKey.from_pubsec_file(self.input)
|
|
elif re.search(regex_seed, line):
|
|
log.info("input file format detected: seed")
|
|
self.duniterpy = duniterpy.key.SigningKey.from_seedhex_file(self.input)
|
|
elif re.search(regex_ssb, line):
|
|
log.info("input file format detected: ssb")
|
|
self.duniterpy = duniterpy.key.SigningKey.from_ssb_file(self.input)
|
|
elif re.search(regex_wif, line):
|
|
log.info("input file format detected: wif")
|
|
self.duniterpy = duniterpy.key.SigningKey.from_wif_file(self.input)
|
|
elif len(line.split(' ')) == 12:
|
|
log.info("input file format detected: mnemonic")
|
|
self.username = line
|
|
self.duniterpy_from_mnemonic()
|
|
elif len(lines) > 1:
|
|
log.info("input file format detected: credentials")
|
|
self.username = line
|
|
self.password = lines[1].strip()
|
|
self.duniterpy_from_credentials()
|
|
else:
|
|
raise NotImplementedError(f"""unable to detect input file format.""")
|
|
except Exception as e:
|
|
log.error(f"""Unable to open file {self.input}: {e}""")
|
|
self._cleanup()
|
|
exit(1)
|
|
log.debug("keygen.duniterpy.seed: %s" % self.duniterpy.seed)
|
|
|
|
def duniterpy_from_mnemonic(self):
|
|
log.debug("keygen.duniterpy_from_mnemonic()")
|
|
scrypt_params = duniterpy.key.scrypt_params.ScryptParams(
|
|
int(self.config.get('scrypt', 'n')) if self.config.has_option('scrypt', 'n') else 4096,
|
|
int(self.config.get('scrypt', 'r')) if self.config.has_option('scrypt', 'r') else 16,
|
|
int(self.config.get('scrypt', 'p')) if self.config.has_option('scrypt', 'p') else 1,
|
|
int(self.config.get('scrypt', 'sl')) if self.config.has_option('scrypt', 'sl') else 32,
|
|
)
|
|
self.duniterpy = duniterpy.key.SigningKey.from_dubp_mnemonic(
|
|
self.username,
|
|
scrypt_params
|
|
)
|
|
log.debug("keygen.duniterpy.seed: %s" % self.duniterpy.seed)
|
|
|
|
def ed25519(self, args):
|
|
log.debug("keygen.ed25519(%s)" % args)
|
|
if args.gpg:
|
|
self.ed25519_from_gpg()
|
|
else:
|
|
if self.input:
|
|
self.duniterpy_from_file()
|
|
else:
|
|
if self.mnemonic:
|
|
self.duniterpy_from_mnemonic()
|
|
else:
|
|
self.duniterpy_from_credentials()
|
|
self.ed25519_from_duniterpy()
|
|
|
|
def ed25519_from_base58(self):
|
|
log.debug("keygen.ed25519_from_base58()")
|
|
self.ed25519_public_bytes = base58.b58decode(self.ed25519_public_base58)
|
|
self.ed25519_secret_bytes = base58.b58decode(self.ed25519_secret_base58)
|
|
log.debug("keygen.ed25519_public_bytes=%s" % self.ed25519_public_bytes)
|
|
log.debug("keygen.ed25519_secret_bytes=%s" % self.ed25519_secret_bytes)
|
|
|
|
def ed25519_from_duniterpy(self):
|
|
log.debug("keygen.ed25519_from_duniterpy()")
|
|
self.ed25519_public_bytes = base58.b58decode(self.duniterpy.pubkey)
|
|
self.ed25519_secret_bytes = self.duniterpy.sk
|
|
log.debug("keygen.ed25519_public_bytes=%s" % self.ed25519_public_bytes)
|
|
log.debug("keygen.ed25519_secret_bytes=%s" % self.ed25519_secret_bytes)
|
|
|
|
def ed25519_from_gpg(self):
|
|
log.debug("keygen.ed25519_from_gpg()")
|
|
self.pgpy_from_gpg()
|
|
self.ed25519_from_pgpy()
|
|
|
|
def pgpy_from_gpg(self):
|
|
log.debug("keygen.pgpy_from_gpg()")
|
|
self.gpg_seckeys = list(self.gpg.keylist(pattern=self.username, secret=True))
|
|
log.debug("keygen.gpg_seckeys=%s" % self.gpg_seckeys)
|
|
if not self.gpg_seckeys:
|
|
log.error(f"""Unable to find any key matching username "{self.username}".""")
|
|
self._cleanup()
|
|
exit(1)
|
|
else:
|
|
self.gpg_seckey = self.gpg_seckeys[0]
|
|
log.info(f"""Found key id "{self.gpg_seckey.fpr}" matching username "{self.username}".""")
|
|
log.debug("keygen.gpg_seckey.expired=%s" % self.gpg_seckey.expired)
|
|
log.debug("keygen.gpg_seckey.fpr=%s" % self.gpg_seckey.fpr)
|
|
log.debug("keygen.gpg_seckey.revoked=%s" % self.gpg_seckey.revoked)
|
|
log.debug("keygen.gpg_seckey.uids=%s" % self.gpg_seckey.uids)
|
|
log.debug("keygen.gpg_seckey.owner_trust=%s" % self.gpg_seckey.owner_trust)
|
|
log.debug("keygen.gpg_seckey.last_update=%s" % self.gpg_seckey.last_update)
|
|
if self.password:
|
|
self.gpg.set_pinentry_mode(gpg.constants.PINENTRY_MODE_LOOPBACK)
|
|
self.pgp_secret_armored = self.gpg.key_export_secret(self.gpg_seckey.fpr)
|
|
log.debug("keygen.pgp_secret_armored=%s" % self.pgp_secret_armored)
|
|
if not self.pgp_secret_armored:
|
|
log.error(f"""Unable to export gpg secret key id "{self.gpg_seckey.fpr}" of user "{self.username}". Please check your password!""")
|
|
self._cleanup()
|
|
exit(2)
|
|
with warnings.catch_warnings():
|
|
# remove CryptographyDeprecationWarning about deprecated
|
|
# SymmetricKeyAlgorithm IDEA, CAST5 and Blowfish (PGPy v0.5.4)
|
|
warnings.simplefilter('ignore')
|
|
self.pgpy, _ = pgpy.PGPKey.from_blob(self.pgp_secret_armored)
|
|
|
|
def ed25519_from_pgpy(self):
|
|
log.debug("keygen.ed25519_from_pgpy()")
|
|
log.debug("keygen.pgpy.fingerprint.keyid=%s" % self.pgpy.fingerprint.keyid)
|
|
log.debug("keygen.pgpy.is_protected=%s" % self.pgpy.is_protected)
|
|
if self.pgpy.is_protected:
|
|
if not self.password:
|
|
with pynentry.PynEntry() as p:
|
|
p.description = f"""The exported pgp key id "{self.pgpy.fingerprint.keyid}" of user "{self.username}" is password protected.
|
|
Please enter the passphrase again to unlock it.
|
|
"""
|
|
p.prompt = 'Passphrase:'
|
|
try:
|
|
self.password = p.get_pin()
|
|
except pynentry.PinEntryCancelled:
|
|
log.warning('Cancelled! Goodbye.')
|
|
self._cleanup()
|
|
exit(2)
|
|
try:
|
|
with warnings.catch_warnings():
|
|
# remove CryptographyDeprecationWarning about deprecated
|
|
# SymmetricKeyAlgorithm IDEA, CAST5 and Blowfish (PGPy v0.5.4)
|
|
warnings.simplefilter('ignore')
|
|
with self.pgpy.unlock(self.password):
|
|
assert self.pgpy.is_unlocked
|
|
log.debug("keygen.pgpy.is_unlocked=%s" % self.pgpy.is_unlocked)
|
|
self.ed25519_seed_bytes_from_pgpy()
|
|
except Exception as e:
|
|
log.error(f"""Unable to unlock pgp secret key id "{self.pgpy.fingerprint.keyid}" of user "{self.username}". Please check your password!""")
|
|
self._cleanup()
|
|
exit(2)
|
|
else:
|
|
self.ed25519_seed_bytes_from_pgpy()
|
|
self.ed25519_public_bytes, self.ed25519_secret_bytes = nacl.bindings.crypto_sign_seed_keypair(self.ed25519_seed_bytes)
|
|
log.debug("keygen.ed25519_public_bytes=%s" % self.ed25519_public_bytes)
|
|
log.debug("keygen.ed25519_secret_bytes=%s" % self.ed25519_secret_bytes)
|
|
|
|
def ed25519_seed_bytes_from_pgpy(self):
|
|
log.debug("keygen.ed25519_seed_bytes_from_pgpy()")
|
|
self.pgpy_key_type()
|
|
if self.pgpy_key_type == 'RSA':
|
|
log.debug("keygen.pgpy._key.keymaterial.p=%s" % self.pgpy._key.keymaterial.p)
|
|
log.debug("keygen.pgpy._key.keymaterial.q=%s" % self.pgpy._key.keymaterial.q)
|
|
# custom seed: use sha256 hash of (p + q)
|
|
self.ed25519_seed_bytes = nacl.bindings.crypto_hash_sha256(long_to_bytes(self.pgpy._key.keymaterial.p + self.pgpy._key.keymaterial.q))
|
|
p = long_to_bytes(self.pgpy._key.keymaterial.p)
|
|
q = long_to_bytes(self.pgpy._key.keymaterial.q)
|
|
log.debug("keygen.ed25519_seed_bytes=%s" % self.ed25519_seed_bytes)
|
|
elif self.pgpy_key_type in ('ECDSA', 'EdDSA', 'ECDH'):
|
|
log.debug("keygen.pgpy._key.keymaterial.s=%s" % self.pgpy._key.keymaterial.s)
|
|
self.ed25519_seed_bytes = long_to_bytes(self.pgpy._key.keymaterial.s)
|
|
log.debug("keygen.ed25519_seed_bytes=%s" % self.ed25519_seed_bytes)
|
|
else:
|
|
raise NotImplementedError(f"Getting seed from {self.pgpy_key_type} key is not implemented")
|
|
|
|
def gpg_passphrase_cb(self, uid_hint, passphrase_info, prev_was_bad):
|
|
log.debug("keygen.gpg_passphrase_cb(%s, %s, %s)" % (uid_hint, passphrase_info, prev_was_bad))
|
|
return self.password
|
|
|
|
def ipfs_from_protobuf2(self):
|
|
log.debug("keygen.ipfs_from_protobuf2()")
|
|
|
|
# PeerID
|
|
self.ipfs_peerid = base58.b58encode(self.ed25519_public_protobuf2).decode('ascii')
|
|
log.debug("keygen.ipfs_peerid=%s" % self.ipfs_peerid)
|
|
|
|
# PrivKey
|
|
self.ipfs_privkey = base64.b64encode(self.ed25519_secret_protobuf2).decode('ascii')
|
|
log.debug("keygen.ipfs_privkey=%s" % self.ipfs_privkey)
|
|
|
|
def pem_pkcs8_from_ed25519(self):
|
|
log.debug("keygen.pem_pkcs8_from_ed25519()")
|
|
|
|
self.ed25519_secret_pem_pkcs8 = ed25519.Ed25519PrivateKey.from_private_bytes(self.ed25519_secret_bytes[:32]).private_bytes(encoding=serialization.Encoding.PEM, format=serialization.PrivateFormat.PKCS8, encryption_algorithm=serialization.NoEncryption()).decode('ascii')
|
|
log.debug("keygen.ed25519_secret_pem_pkcs8=%s" % self.ed25519_secret_pem_pkcs8)
|
|
|
|
def pgpy_key_type(self):
|
|
log.debug("keygen.pgpy_key_type()")
|
|
if isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.RSAPriv):
|
|
self.pgpy_key_type = 'RSA'
|
|
elif isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.DSAPriv):
|
|
self.pgpy_key_type = 'DSA'
|
|
elif isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.ElGPriv):
|
|
self.pgpy_key_type = 'ElGamal'
|
|
elif isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.ECDSAPriv):
|
|
self.pgpy_key_type = 'ECDSA'
|
|
elif isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.EdDSAPriv):
|
|
self.pgpy_key_type = 'EdDSA'
|
|
elif isinstance(self.pgpy._key.keymaterial, pgpy.packet.fields.ECDHPriv):
|
|
self.pgpy_key_type = 'ECDH'
|
|
else:
|
|
self.pgpy_key_type = 'undefined'
|
|
log.debug("keygen.pgpy_key_type=%s" % self.pgpy_key_type)
|
|
|
|
def protobuf2_from_ed25519(self):
|
|
# libp2p protobuf version 2
|
|
log.debug("keygen.protobuf2_from_ed25519()")
|
|
self.ed25519_public_protobuf2 = b'\x00$\x08\x01\x12 ' + self.ed25519_public_bytes
|
|
self.ed25519_secret_protobuf2 = b'\x08\x01\x12@' + self.ed25519_secret_bytes
|
|
log.debug("keygen.ed25519_public_protobuf2=%s" % self.ed25519_public_protobuf2)
|
|
log.debug("keygen.ed25519_secret_protobuf2=%s" % self.ed25519_secret_protobuf2)
|
|
|
|
##
|
|
# long_to_bytes comes from PyCrypto, which is released into Public Domain
|
|
# https://github.com/dlitz/pycrypto/blob/master/lib/Crypto/Util/number.py
|
|
def bytes_to_long(s):
|
|
"""bytes_to_long(string) : long
|
|
Convert a byte string to a long integer.
|
|
This is (essentially) the inverse of long_to_bytes().
|
|
"""
|
|
acc = 0
|
|
unpack = struct.unpack
|
|
length = len(s)
|
|
if length % 4:
|
|
extra = (4 - length % 4)
|
|
s = b'\000' * extra + s
|
|
length = length + extra
|
|
for i in range(0, length, 4):
|
|
acc = (acc << 32) + unpack('>I', s[i:i+4])[0]
|
|
return acc
|
|
|
|
def long_to_bytes(n, blocksize=0):
|
|
"""long_to_bytes(n:long, blocksize:int) : string
|
|
Convert a long integer to a byte string.
|
|
If optional blocksize is given and greater than zero, pad the front of the
|
|
byte string with binary zeros so that the length is a multiple of
|
|
blocksize.
|
|
"""
|
|
# after much testing, this algorithm was deemed to be the fastest
|
|
s = b''
|
|
n = int(n)
|
|
pack = struct.pack
|
|
while n > 0:
|
|
s = pack('>I', n & 0xffffffff) + s
|
|
n = n >> 32
|
|
# strip off leading zeros
|
|
for i in range(len(s)):
|
|
if s[i] != b'\000'[0]:
|
|
break
|
|
else:
|
|
# only happens when n == 0
|
|
s = b'\000'
|
|
i = 0
|
|
s = s[i:]
|
|
# add back some pad bytes. this could be done more efficiently w.r.t. the
|
|
# de-padding being done above, but sigh...
|
|
if blocksize > 0 and len(s) % blocksize:
|
|
s = (blocksize - len(s) % blocksize) * b'\000' + s
|
|
return s
|
|
|
|
def main(argv=None):
|
|
if argv is None:
|
|
argv = sys.argv[1:]
|
|
|
|
cli = keygen()
|
|
return cli._run(argv)
|
|
|
|
def version(version=__version__):
|
|
print("%s v%s" % (sys.argv[0],version))
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|