From 80c9bb8c14c6b7a790d581148b927b1a1b20ec7c Mon Sep 17 00:00:00 2001 From: Kevin Gill Date: Fri, 19 Oct 2018 16:38:40 +0100 Subject: [PATCH] Added RSA512 algorithm. --- requests_http_signature/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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", }