diff --git a/utils/crypto/elliptic_curve.py b/utils/crypto/elliptic_curve.py index 5cdbef3..dbf396c 100644 --- a/utils/crypto/elliptic_curve.py +++ b/utils/crypto/elliptic_curve.py @@ -1,7 +1,9 @@ import hashlib + +from cryptography.hazmat.backends import default_backend from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives.asymmetric import ec -from cryptography.hazmat.backends import default_backend + class EllipticCurve: def __init__(self): @@ -14,12 +16,19 @@ def generate_key_pair(self): return private_key, public_key def sign(self, private_key, message): - signer = private_key.signer(padding=ec.ECDSA(max_size=32).padding, algorithm=ec.ECDSA(hash_func=hashlib.sha256)) + signer = private_key.signer( + padding=ec.ECDSA(max_size=32).padding, + algorithm=ec.ECDSA(hash_func=hashlib.sha256), + ) signature = signer.sign(message.encode()) return signature def verify(self, public_key, message, signature): - verifier = public_key.verifier(signature, padding=ec.ECDSA(max_size=32).padding, algorithm=ec.ECDSA(hash_func=hashlib.sha256)) + verifier = public_key.verifier( + signature, + padding=ec.ECDSA(max_size=32).padding, + algorithm=ec.ECDSA(hash_func=hashlib.sha256), + ) try: verifier.verify(message.encode()) return True