diff options
Diffstat (limited to 'src/add_python/lilliput/ae_mode_1.py')
| -rw-r--r-- | src/add_python/lilliput/ae_mode_1.py | 14 |
1 files 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 index 1a3c39e..23f4c7b 100644 --- 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) |