Handle PSS in keys.PrivateKeyInfo.wrap() / keys.PublicKeyInfo.wrap()

wbond · Mar 15, 2022 · d0040c2 · d0040c2
1 parent 0bb7066
commit d0040c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/asn1crypto/keys.py b/asn1crypto/keys.py
@@ -752,7 +752,7 @@ def wrap(cls, private_key, algorithm):
                 type_name(private_key)
             ))
 
-        if algorithm == 'rsa':
+        if algorithm == 'rsa' or algorithm == 'rsassa_pss':
             if not isinstance(private_key, RSAPrivateKey):
                 private_key = RSAPrivateKey.load(private_key)
             params = Null()
@@ -1120,7 +1120,7 @@ def wrap(cls, public_key, algorithm):
                 type_name(public_key)
             ))
 
-        if algorithm != 'rsa':
+        if algorithm != 'rsa' and algorithm != 'rsassa_pss':
             raise ValueError(unwrap(
                 '''
                 algorithm must "rsa", not %s