lilliput-ae-reference-implementation

commit 5949f01e728c11990280f6b1d1a35c2153db4578
parent f468f2ca79e480fc058d37c338eb756ef88779f6
Author: Kévin Le Gouguec <kevin.legouguec@airbus.com>
Date:   Mon, 25 Mar 2019 10:41:02 +0100

[implem-python] Retrait de range()s et variables intermédiaires

Diffstat:
M
src/add_python/lilliput/ae_mode_1.py
 | 
14
++++++--------

1 file changed, 6 insertions(+), 8 deletions(-)
diff --git a/src/add_python/lilliput/ae_mode_1.py b/src/add_python/lilliput/ae_mode_1.py
@@ -74,7 +74,7 @@ def _tweak_message(N, j, padding):
 
 
 def _treat_message_enc(M, N, key):
-    checksum = [0 for byte in range(0, BLOCK_BYTES)]
+    checksum = [0]*BLOCK_BYTES
 
     l = len(M)//BLOCK_BYTES
     padding_bytes = len(M)%BLOCK_BYTES
@@ -95,10 +95,9 @@ def _treat_message_enc(M, N, key):
         m_padded = pad10(M[l])
         checksum = xor(checksum, m_padded)
         tweak = _tweak_message(N, l, _MessageTweak.PAD)
-        pad = tbc.encrypt(tweak, key, [0 for byte in range(0, BLOCK_BYTES)])
+        pad = tbc.encrypt(tweak, key, [0]*BLOCK_BYTES)
 
-        lower_part = pad[:padding_bytes]
-        C.append(xor(M[l], lower_part))
+        C.append(xor(M[l], pad[:padding_bytes]))
         tweak_final = _tweak_message(N, l+1, _MessageTweak.FINAL)
         Final = tbc.encrypt(tweak_final, key, checksum)
 
@@ -106,7 +105,7 @@ def _treat_message_enc(M, N, key):
 
 
 def _treat_message_dec(C, N, key):
-    checksum = [0 for byte in range(0, BLOCK_BYTES)]
+    checksum = [0]*BLOCK_BYTES
 
     l = len(C)//BLOCK_BYTES
     padding_bytes = len(C)%BLOCK_BYTES
@@ -125,9 +124,8 @@ def _treat_message_dec(C, N, key):
 
     else:
         tweak = _tweak_message(N, l, _MessageTweak.PAD)
-        pad = tbc.encrypt(tweak, key, [0 for byte in range(0, BLOCK_BYTES)])
-        lower_part = pad[:padding_bytes]
-        M.append(xor(C[l], lower_part))
+        pad = tbc.encrypt(tweak, key, [0]*BLOCK_BYTES)
+        M.append(xor(C[l], pad[:padding_bytes]))
 
         m_padded = pad10(M[l])
         checksum = xor(checksum, m_padded)