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