From fd0d3d5f129c5da5392684eb45c08cf3334a7ef8 Mon Sep 17 00:00:00 2001
From: Sascha Tommasone <sascha@tommasone.me>
Date: Mon, 27 May 2024 12:54:09 +0200
Subject: [PATCH] [Assignment-4] added comments in mitm and slke attack
 implementations

---
 .../proxy/proxymodules/mitm.py                           | 9 ++++++++-
 .../proxy/proxymodules/slke.py                           | 8 +++++++-
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/mitm.py b/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/mitm.py
index 61a389b..979f47b 100644
--- a/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/mitm.py	
+++ b/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/mitm.py	
@@ -4,7 +4,6 @@ import base64
 import os.path as path
 
 
-# protocol 1: flag{m4n_1n_th3_m1ddl3_w0w}
 class Module:
     def __init__(self, incoming=False, verbose=False, options=None):
         # extract the file name from __file__. __file__ is proxymodules/name.py
@@ -13,24 +12,32 @@ class Module:
         self.incoming = incoming  # incoming means module is on -im chain
         self.find = None  # if find is not None, this text will be highlighted
 
+    # protocol 1: flag{m4n_1n_th3_m1ddl3_w0w}
     def execute(self, data):
         print(f"Incoming data: {data}")
         data_json = json.loads(data)
 
+        # return handshake messages without modification
         if data_json.get("type") == "HANDSHAKE":
             return data
         
+        # replace "Bob" with "Eve" in the first protocol message
         elif data_json.get("id") == 1:
             data_json["content"] = data_json["content"].replace("Bob","Eve")
         
+        # if message 3 received, build valid message 4
+        # instead of {Hello Alice!}_K use arbitary data
         elif data_json.get("id") == 3:
             data_json["id"] = 4
             data_json["sender"] = "Bob"
             data_json["receiver"] = "Alice"
             data_json["content"] = str(base64.b64encode(os.urandom(16)))
         
+        # dump message
         data = json.dumps(data_json) + "\n"
         print(f"Outgoing data: {data}")
+        
+        # return crafted message to proxy
         return data
 
 
diff --git a/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/slke.py b/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/slke.py
index 021b99f..4367a5c 100644
--- a/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/slke.py	
+++ b/Assignment 4 - Protokollsicherheit (Praxis)/proxy/proxymodules/slke.py	
@@ -3,7 +3,6 @@ import time
 import os.path as path
 
 
-# protocol 2: flag{n3v3r_tru5t_b0b}
 class Module:
     def __init__(self, incoming=False, verbose=False, options=None):
         # extract the file name from __file__. __file__ is proxymodules/name.py
@@ -12,22 +11,29 @@ class Module:
         self.incoming = incoming  # incoming means module is on -im chain
         self.find = None  # if find is not None, this text will be highlighted
 
+    # protocol 2: flag{n3v3r_tru5t_b0b}
     def execute(self, data):
         print(f"Incoming data: {data}")
         data_json = json.loads(data)
         
+        # return handshake messages without modification
         if data_json.get("type") == "HANDSHAKE":
             return data
         
+        # swap receiver and sender of protocol messages with id 1,2 or 3
         elif data_json.get("id") in {1, 2, 3}:
             data_json["sender"], data_json["receiver"] = data_json["receiver"], data_json["sender"]
 
+        # if message 3 received from Alice, build valid message 4
         if data_json.get("id") == 3:
             data_json["id"] = 4
             data_json["content"] = str(int(time.time()))
 
+        # dump message
         data = json.dumps(data_json) + "\n"
         print(f"Outgoing data: {data}")
+
+        # return crafted message to proxy
         return data