50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
import ecdsa
|
|
from hashlib import sha256, sha512
|
|
|
|
|
|
class Verifier:
|
|
def verify_rfc6979(self, public_key: bytes, message: bytes, signature: bytes) -> bool:
|
|
"""
|
|
Verify a signature using the public key.
|
|
|
|
:param public_key: Public key in byte format.
|
|
:param message: Signature verification message in byte format.
|
|
:param signature: Signature in byte format.
|
|
:return: True if the signature is correct, otherwise False.
|
|
:raises: ValueError: If the public_key key is incorrect.
|
|
"""
|
|
if len(public_key) == 0 or public_key is None:
|
|
raise ValueError(f"Incorrect public key: {public_key}")
|
|
|
|
if message is None or signature is None:
|
|
return False
|
|
|
|
vk = ecdsa.VerifyingKey.from_string(public_key, curve=ecdsa.NIST256p, hashfunc=sha256)
|
|
|
|
try:
|
|
return vk.verify(signature, message)
|
|
except ecdsa.BadSignatureError:
|
|
return False
|
|
|
|
def verify(self, public_key: bytes, message: bytes, signature: bytes) -> bool:
|
|
"""
|
|
Verify a signature using the public key.
|
|
|
|
:param public_key: Public key in byte format.
|
|
:param message: Signature verification message in byte format.
|
|
:param signature: Signature in byte format.
|
|
:return: True if the signature is correct, otherwise False.
|
|
:raises: ValueError: If the public_key key is incorrect.
|
|
"""
|
|
if len(public_key) == 0 or public_key is None:
|
|
raise ValueError(f"Incorrect public key: {public_key}")
|
|
|
|
if message is None or signature is None:
|
|
return False
|
|
|
|
vk = ecdsa.VerifyingKey.from_string(public_key, curve=ecdsa.NIST256p, hashfunc=sha512)
|
|
|
|
try:
|
|
return vk.verify(signature, message)
|
|
except ecdsa.BadSignatureError:
|
|
return False
|