added signature tests

This commit is contained in:
Oscar Krause
2025-04-14 21:38:01 +02:00
parent 79f1015a86
commit 445a303955
2 changed files with 27 additions and 1 deletions

View File

@@ -43,7 +43,7 @@ class PrivateKey:
return PublicKey(data=data)
def generate_signature(self, data: bytes) -> bytes:
return self.__key.sign(data, PKCS1v15(), SHA256())
return self.__key.sign(data, padding=PKCS1v15(), algorithm=SHA256())
@staticmethod
def generate(public_exponent: int = 65537, key_size: int = 2048) -> "PrivateKey":
@@ -82,6 +82,9 @@ class PublicKey:
format=serialization.PublicFormat.SubjectPublicKeyInfo
)
def verify_signature(self, signature: bytes, data: bytes) -> bytes:
return self.__key.verify(signature, data, padding=PKCS1v15(), algorithm=SHA256())
class Cert: