finalizando trabalho 4

trabalho4/tgs.py

@@ -1,9 +1,9 @@
-import socket
 from common import *
-import socket
 from AES import AESCipher
+import socket
 import os
 import base64
+import time
 
 HOST = "127.0.0.1"
 PORT = TGS_PORT
@@ -15,6 +15,7 @@ def process_message(message):
     message = message.split("||")
 
     if message[0] == "getkey":
+        print('ktgs', TGS_KEY)
         return f"{base64.b64encode(TGS_KEY)}"
     if message[0] == "request":
         """
@@ -48,6 +49,7 @@ def process_message(message):
         # generate M4
         K_c_s = os.urandom(16)
         T_A = 600 if T_R > 600 else T_R
+        T_A = int(time.time()) + T_A
         T_c_s = f"{ID_C}<>{T_A}<>{K_c_s}"
 
         K_s = get_ks()
@@ -58,6 +60,8 @@ def process_message(message):
         M4_inner_AES = AESCipher(K_c_tgs)
         M4_inner = M4_inner_AES.encrypt(M4_inner)
 
+        print("Sending M4", f"[{K_c_s}<>{T_A}<>{N2}]<>[T_c_s] | T_c_s = [{ID_C}<>{T_A}<>K_c_s]")
+
         M4 = f"{M4_inner}<>{T_c_s}"
         return M4
 
@@ -80,7 +84,6 @@ def main():
         while True:
             conn, addr = s.accept()
             with conn:
-                print(f"Connected by {addr}")
                 all_data = ""
                 data = conn.recv(4096)
                 all_data += data.decode()
@@ -90,8 +93,4 @@ def main():
                 conn.sendall(response.encode())
 
 if __name__ == "__main__":
-    if not os.path.exists("tgs_db.data"):
-        with open("tgs_db.data", "w") as f:
-            pass
-
     main()