Use fast unpadder by default

xymy · Mar 23, 2020 · f1ffa2d · f1ffa2d
1 parent 0487afb
commit f1ffa2d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/xycrypto/padding.py b/src/xycrypto/padding.py
@@ -150,8 +150,12 @@ def finalize(self):
     def wrap_encryptor(self, encryptor_ctx):
         return self._PaddingWrapper(encryptor_ctx, self.padder())
 
-    def wrap_decryptor(self, decryptor_ctx):
-        return self._PaddingWrapper(self.unpadder(), decryptor_ctx)
+    def wrap_decryptor(self, decryptor_ctx, *, fast=True):
+        if fast:
+            unpadder = self.fast_unpadder()
+        else:
+            unpadder = self.unpadder()
+        return self._PaddingWrapper(unpadder, decryptor_ctx)
 
 
 # ==============