cryptopals

commit 2962426c6a60462489cfcfd9b52d021ad02c664d
parent 5b428137a6bf2c2278e8a6b9494d1a4c19b65d0c
Author: mpizzzle <michael.770211@gmail.com>
Date:   Tue, 31 Oct 2017 19:34:17 +0000

refactoring padding validation

Diffstat:
M
set2/pkcs7_padding_validation.py
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/set2/pkcs7_padding_validation.py b/set2/pkcs7_padding_validation.py
@@ -7,7 +7,7 @@ def pkcs7_padding(msg):
 def pkcs7_padding_stripper(msg):
     if ord(msg[len(msg) - 1]) > AES.block_size or ord(msg[len(msg) - 1]) == 0:
         raise Exception("invalid pkcs7 padding")
-    for c in msg[:len(msg) - ord(msg[len(msg) - 1]) - 1 : -1]:
+    for c in msg[len(msg) - ord(msg[len(msg) - 1]):]:
         if c != msg[len(msg) - 1]:
             raise Exception("invalid pkcs7 padding")
     return msg[:len(msg) - ord(msg[len(msg) - 1])]