mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 14:14:22 +01:00
83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
import os.path
|
|
from pathlib import Path
|
|
from coincurve import PrivateKey
|
|
|
|
from common.exceptions import InvalidKey
|
|
from common.cryptographer import Cryptographer
|
|
|
|
|
|
def save_key(sk, filename):
|
|
"""
|
|
Saves the secret key on disk.
|
|
|
|
Args:
|
|
sk (:obj:`EllipticCurvePrivateKey`): a private key file to be saved on disk.
|
|
filename (:obj:`str`): the name that will be given to the key file.
|
|
"""
|
|
|
|
with open(filename, "wb") as der_out:
|
|
der_out.write(sk.to_der())
|
|
|
|
|
|
def generate_keys(data_dir):
|
|
"""
|
|
Generates a key pair for the client.
|
|
|
|
Args:
|
|
data_dir (:obj:`str`): path to data directory where the keys will be stored.
|
|
|
|
Returns:
|
|
:obj:`tuple`: a tuple containing a ``PrivateKey`` and a ``str`` representing the client sk and compressed pk
|
|
respectively.
|
|
|
|
Raises:
|
|
:obj:`FileExistsError`: if the key pair already exists in the given directory.
|
|
"""
|
|
|
|
# Create the output folder it it does not exist (and all the parents if they don't either)
|
|
Path(data_dir).mkdir(parents=True, exist_ok=True)
|
|
sk_file_name = os.path.join(data_dir, "sk.der")
|
|
|
|
if os.path.exists(sk_file_name):
|
|
raise FileExistsError("The client key pair already exists")
|
|
|
|
sk = PrivateKey()
|
|
pk = sk.public_key
|
|
save_key(sk, sk_file_name)
|
|
|
|
return sk, Cryptographer.get_compressed_pk(pk)
|
|
|
|
|
|
def load_keys(data_dir):
|
|
"""
|
|
Loads a the client key pair.
|
|
|
|
Args:
|
|
data_dir (:obj:`str`): path to data directory where the keys are stored.
|
|
|
|
Returns:
|
|
:obj:`tuple`: a tuple containing a ``PrivateKey`` and a ``str`` representing the client sk and compressed pk
|
|
respectively.
|
|
|
|
Raises:
|
|
:obj:`InvalidKey <cli.exceptions.InvalidKey>`: if any of the keys is invalid or cannot be loaded.
|
|
"""
|
|
|
|
if not isinstance(data_dir, str):
|
|
raise ValueError("Invalid data_dir. Please check your settings")
|
|
|
|
sk_file_path = os.path.join(data_dir, "sk.der")
|
|
|
|
cli_sk_der = Cryptographer.load_key_file(sk_file_path)
|
|
cli_sk = Cryptographer.load_private_key_der(cli_sk_der)
|
|
|
|
if cli_sk is None:
|
|
raise InvalidKey("Client private key is invalid or cannot be parsed")
|
|
|
|
compressed_cli_pk = Cryptographer.get_compressed_pk(cli_sk.public_key)
|
|
|
|
if compressed_cli_pk is None:
|
|
raise InvalidKey("Client public key cannot be loaded")
|
|
|
|
return cli_sk, compressed_cli_pk
|