OFB implemented partially

Author: Glindeb

Documentation/Rapport/Introduction.tex

@@ -1,6 +1,5 @@
 \mychapter{Introduktion}
 
-\section{Inledning}
 Kryptering, något vi möter varje dag i dagens samhälle men...
 
 När de vi idag kallar internet uppkom och öppnades för allmänheten så var tankarna på hur

Documentation/Rapport/Main.pdf

@@ -5,4 +5,5 @@ @misc{wikipedia_2022
     publisher = {Wikimedia Foundation},
     year = {2022},
     month = {Jun}
-}
+}
+

Documentation/Rapport/Main.synctex.gz

@@ -585,3 +585,84 @@ def pcbc_dec(key, file_path, iv, terminal_width=80):
         progress = progress_bar(progress, file_size, terminal_width)
     progress = progress_bar(progress, file_size, terminal_width)
     remove(file_path)
+
+
+# OFB encryption function (Tested but no automated test implemented and no test with different file sizes)
+def ofb_enc(key, file_path, iv, terminal_width=80):
+    file_size = getsize(file_path)
+    progress = 0
+    progress = progress_bar(progress, file_size, terminal_width)
+    iv = [int(iv[i:i+2], 16) for i in range(0, len(iv), 2)]
+
+    with open(f"{file_path}.enc", 'wb') as output, open(file_path, 'rb') as data:
+        for i in range(int(file_size/16)):
+            raw = [i for i in data.read(16)]
+            mix = encryption_rounds(iv, key)
+            result = xor(raw, mix)
+            output.write(bytes(result))
+            progress = progress_bar(progress, file_size, terminal_width)
+
+        if file_size % 16 != 0:
+            raw = [i for i in data.read()]
+            raw, length = add_padding(raw)
+
+            if file_size < 16:
+                mix = encryption_rounds(iv, key)
+            else:
+                mix = encryption_rounds(mix, key)
+            result = xor(mix, raw)
+
+            mix = encryption_rounds(mix, key)
+            identifier = xor(([0 for i in range(15)] + [length]), mix)
+
+            output.write(bytes(result + identifier))
+            progress = progress_bar(progress, file_size, terminal_width)
+        else:
+            mix = bytes(encryption_rounds(mix, key))
+            identifier = xor([0 for i in range(16)], mix)
+            output.write(identifier)
+            progress = progress_bar(progress, file_size, terminal_width)
+    progress = progress_bar(progress, file_size, terminal_width)
+    remove(file_path)
+
+
+# OFB decryption function (Tested but no automated test implemented and no test with different file sizes)
+def ofb_dec(key, file_path, iv, terminal_width=80):
+    iv = [int(iv[i:i+2], 16) for i in range(0, len(iv), 2)]
+    file_size = getsize(file_path)
+    progress = 0
+    progress = progress_bar(progress, file_size, terminal_width)
+    file_name = file_path[:-4]
+
+    with open(f"{file_name}", 'wb') as output, open(file_path, 'rb') as data:
+        if int(file_size/16) - 3 >= 0:
+            raw = [i for i in data.read(16)]
+            mix = encryption_rounds(iv, key)
+            result = xor(raw, mix)
+            output.write(bytes(result))
+            progress = progress_bar(progress, file_size, terminal_width)
+
+            for i in range(int(file_size/16) - 3):
+                raw = [i for i in data.read(16)]
+                mix = encryption_rounds(mix, key)
+                result = xor(raw, mix)
+                output.write(bytes(result))
+                progress = progress_bar(progress, file_size, terminal_width)
+        else:
+            mix = iv
+
+        data_pice = [i for i in data.read(16)]
+        identifier = [i for i in data.read()]
+
+        mix = encryption_rounds(mix, key)
+        data_pice = xor(data_pice, mix)
+
+        mix = encryption_rounds(mix, key)
+        identifier = xor(identifier, mix)
+
+        result = bytes(remove_padding(data_pice, identifier))
+
+        output.write(result)
+        progress = progress_bar(progress, file_size, terminal_width)
+    progress = progress_bar(progress, file_size, terminal_width)
+    remove(file_path)

Documentation/Rapport/Ref.bib

@@ -1,6 +1,15 @@
-with open(r".\tmp\test_files\data.txt.enc", "rb") as p:
-    t = p.read()
-    print(t)
+from PyAES.AES import ofb_dec, ofb_enc
 
-#with open(r"C:\Users\Gabriel\Documents\GitHub\AES-Python\tmp\test_files\data.txt.enc", "wb") as p:
-#    p.write(b'2 ?\xebm\xf5o\xc2\x8b\x90\x80\x84 D\xc4\x95\x89\x18\n\xeb\xac\xde\xa7P>Ei\xbc|\x9c\xfa\xf2')
+def run(val):
+    key = "2b7e151628aed2a6abf7158809cf4f3c"
+    iv = "000102030405060708090a0b0c0d0e0f"
+    file_path = r"/Users/gabriellindeblad/Documents/GitHub/AES-Python/tmp/test_files/data.txt"
+
+    if val == 1:
+        ofb_enc(key, file_path, iv)
+    elif val == 2:
+        ofb_dec(key, f"{file_path}.enc", iv)
+
+if __name__ == "__main__":
+    run(2)
+    print("Done!")