diff --git a/requests_http_signature/__init__.py b/requests_http_signature/__init__.py index a61e67c..4786131 100644 --- a/requests_http_signature/__init__.py +++ b/requests_http_signature/__init__.py @@ -13,7 +13,7 @@ class Crypto: from cryptography.hazmat.backends import default_backend from cryptography.hazmat.primitives.asymmetric import rsa, ec from cryptography.hazmat.primitives.asymmetric.padding import PKCS1v15 - from cryptography.hazmat.primitives.hashes import SHA1, SHA256 + from cryptography.hazmat.primitives.hashes import SHA1, SHA256, SHA512 self.__dict__.update(locals()) def sign(self, string_to_sign, key, passphrase=None): @@ -23,6 +23,9 @@ class Crypto: if self.algorithm in {"rsa-sha1", "rsa-sha256"}: hasher = self.SHA1() if self.algorithm.endswith("sha1") else self.SHA256() signer = key.signer(padding=self.PKCS1v15(), algorithm=hasher) + elif self.algorithm in {"rsa-sha512"}: + hasher = self.SHA512() + signer = key.signer(padding=self.PKCS1v15(), algorithm=hasher) elif self.algorithm == "ecdsa-sha256": signer = key.signer(signature_algorithm=self.ec.ECDSA(algorithm=self.SHA256())) signer.update(string_to_sign) @@ -44,6 +47,7 @@ class HTTPSignatureAuth(requests.auth.AuthBase): known_algorithms = { "rsa-sha1", "rsa-sha256", + "rsa-sha512", "hmac-sha256", "ecdsa-sha256", }