test: update files

crypto/easy_dhke/easy_dhke.py

@@ -6,6 +6,7 @@ import signal
 import string
 import random
 import os
+from flag import flag
 
 
 class Task(socketserver.BaseRequestHandler):
@@ -32,17 +33,28 @@ class Task(socketserver.BaseRequestHandler):
         return self._recvall()
 
     def dhke(self):
+        # p is a large prime number used for modulo operations in the Diffie-Hellman key exchange
         p = 327824197795087630552811243153730025469
+        # g is the base used for generating public keys in the Diffie-Hellman key exchange
         g = 5
+        # alice is Alice's private key, an integer chosen by Alice
         alice = 22751
+        # bob is Bob's private key, an integer chosen by Bob
         bob = 39494
+        # Bob calculates his public key as g^bob mod p and assigns it to Bob (uppercase to distinguish from private key)
         Bob = pow(g, bob, p)
+        # The shared secret key is calculated by Alice using Bob's public key, raised to the power of Alice's private key mod p
         key = long_to_bytes(pow(Bob, alice, p))
+        # The random seed is initialized with 8 bytes of random data from the OS's cryptographically secure random generator
         random.seed(os.urandom(8))
+        # secret is a random string consisting of 20 alphanumeric characters, used as the secret message
         secret = ''.join(
             [random.choice(string.ascii_letters+string.digits) for _ in range(20)])
+        
         self.send(b"[+] Alice said :")
-        self.send(self.encrypt(secret.encode(),key))
+        
+        self.send(self.encrypt(secret.encode(), key))
+        
         message = self.recv(b"[+] Now tell me what are they talking about: ")
         if message != secret.encode():
             return False
@@ -51,8 +63,10 @@ class Task(socketserver.BaseRequestHandler):
         hacked = self.recv(b"[+] Tell me the cipher:")
         if self.decrypt(hacked, key) != b"HackedBy0xfa":
             return False
+        # If all checks pass, return True indicating the exchange was successful and not intercepted
         return True
 
+
     def encrypt(self, plain_text:bytes, key:bytes)->bytes:
         cipher = AES.new(key, AES.MODE_ECB)
         cipher_text = cipher.encrypt(pad(plain_text, AES.block_size))
@@ -82,8 +96,6 @@ class ForkedServer(socketserver.ForkingMixIn, socketserver.TCPServer):
 
 
 if __name__ == "__main__":
-    # flag = bytes(os.getenv("FLAG"),"utf-8")
-    flag = b"flag{coooloooool}"
     HOST, PORT = '0.0.0.0', 10001
     server = ForkedServer((HOST, PORT), Task)
     server.allow_reuse_address = True