Added file encryption tests for ECB and CBC

Author: DenseLance

hight_ECB.py

@@ -29,7 +29,7 @@ def ecb_hight_encryption(P, MK):
 def ecb_hight_decryption(C, MK):
     WK, SK = decryption_key_schedule(MK)
     D = decryption_transformation(C, WK, SK)
-    for block in range(8, len(P), 8):
+    for block in range(8, len(C), 8):
         D += decryption_transformation(C[block:block + 8], WK, SK)
     return D
 

test.txt

@@ -0,0 +1,100 @@
+1
+2
+3
+4
+5
+6
+7
+8
+9
+10
+11
+12
+13
+14
+15
+16
+17
+18
+19
+20
+21
+22
+23
+24
+25
+26
+27
+28
+29
+30
+31
+32
+33
+34
+35
+36
+37
+38
+39
+40
+41
+42
+43
+44
+45
+46
+47
+48
+49
+50
+51
+52
+53
+54
+55
+56
+57
+58
+59
+60
+61
+62
+63
+64
+65
+66
+67
+68
+69
+70
+71
+72
+73
+74
+75
+76
+77
+78
+79
+80
+81
+82
+83
+84
+85
+86
+87
+88
+89
+90
+91
+92
+93
+94
+95
+96
+97
+98
+99
+100

test_CBC.py

@@ -0,0 +1,31 @@
+import io
+import sys
+sys.stdout, temp = io.StringIO(), sys.stdout
+from hight_CBC import cbc_hight_encryption, cbc_hight_decryption
+sys.stdout = temp
+
+# Similar to how Crypto.Util.Padding is implemented
+def pad(byte_list_to_pad: list, block_size: int):
+    assert byte_list_to_pad[-1] != 13
+    return byte_list_to_pad + list(b"\r") * (-len(byte_list_to_pad) % block_size)
+
+def unpad(padded_byte_list: list):
+    while padded_byte_list and padded_byte_list[-1] == 13:
+        del padded_byte_list[-1]
+    return padded_byte_list
+
+# TEST CASE
+MK = [0x88, 0xE3, 0x4F, 0x8F, 0x08, 0x17, 0x79, 0xF1, 0xE9, 0xF3, 0x94, 0x37, 0x0A, 0xD4, 0x05, 0x89]
+IV = [0x26, 0x8D, 0x66, 0xA7, 0x35, 0xA8, 0x1A, 0x81]
+P = list(open("test.txt", "rb").read())
+
+# MAIN CODE
+print("Plaintext:", [hex(byte)[2:].upper() for byte in P])
+
+C = cbc_hight_encryption(pad(P, 8), IV, MK)
+print("Encrypted bytes:", [hex(byte)[2:].upper() for byte in C])
+
+D = unpad(cbc_hight_decryption(C, IV, MK))
+print("Decrypted bytes:", [hex(byte)[2:].upper() for byte in D])
+
+assert D == P

test_ECB.py

@@ -0,0 +1,30 @@
+import io
+import sys
+sys.stdout, temp = io.StringIO(), sys.stdout
+from hight_ECB import ecb_hight_encryption, ecb_hight_decryption
+sys.stdout = temp
+
+# Similar to how Crypto.Util.Padding is implemented
+def pad(byte_list_to_pad: list, block_size: int):
+    assert byte_list_to_pad[-1] != 13
+    return byte_list_to_pad + list(b"\r") * (-len(byte_list_to_pad) % block_size)
+
+def unpad(padded_byte_list: list):
+    while padded_byte_list and padded_byte_list[-1] == 13:
+        del padded_byte_list[-1]
+    return padded_byte_list
+
+# TEST CASE
+MK = [0x88, 0xE3, 0x4F, 0x8F, 0x08, 0x17, 0x79, 0xF1, 0xE9, 0xF3, 0x94, 0x37, 0x0A, 0xD4, 0x05, 0x89]
+P = list(open("test.txt", "rb").read())
+
+# MAIN CODE
+print("Plaintext:", [hex(byte)[2:].upper() for byte in P])
+
+C = ecb_hight_encryption(pad(P, 8), MK)
+print("Encrypted bytes:", [hex(byte)[2:].upper() for byte in C])
+
+D = unpad(ecb_hight_decryption(C, MK))
+print("Decrypted bytes:", [hex(byte)[2:].upper() for byte in D])
+
+assert D == P