remove dependency on M2Crypto.util

mwielgoszewski · mwielgoszewski · commit 3209a99f261c · 2014-11-25T08:57:50.000-05:00
diff --git a/paddingoracle.py b/paddingoracle.py
@@ -259,11 +259,16 @@ def xor(data, key):
 
 def test():
     import os
-    from M2Crypto.util import pkcs7_pad
     from Crypto.Cipher import AES
 
     teststring = 'The quick brown fox jumped over the lazy dog'
 
+    def pkcs7_pad(data, blklen=16):
+        if blklen > 255:
+            raise ValueError('Illegal block size %d' % (blklen, ))
+        pad = (blklen - (len(data) % blklen))
+        return data + chr(pad) * pad
+
     class PadBuster(PaddingOracle):
         def oracle(self, data):
             _cipher = AES.new(key, AES.MODE_CBC, str(iv))
