make hashfn a keyword argument for ecdh

This commit is contained in:
jeffthibault
2022-10-26 07:32:24 -04:00
parent 8e0efb4f37
commit 751c1a3c7d

View File

@@ -31,7 +31,7 @@ def tweak_add_private_key(private_key: str, scalar: bytes) -> str:
def compute_shared_secret(sender_private_key: str, receiver_public_key: str) -> str: def compute_shared_secret(sender_private_key: str, receiver_public_key: str) -> str:
public_key = PublicKey(bytes.fromhex("02" + receiver_public_key), True) public_key = PublicKey(bytes.fromhex("02" + receiver_public_key), True)
return public_key.ecdh(bytes.fromhex(sender_private_key), copy_x).hex() return public_key.ecdh(bytes.fromhex(sender_private_key), hashfn=copy_x).hex()
def encrypt_message(content: str, shared_secret: str) -> str: def encrypt_message(content: str, shared_secret: str) -> str:
iv = os.urandom(16) iv = os.urandom(16)