funcionando trabalho 4

This commit is contained in:
2023-10-22 17:15:12 -03:00
parent c8799ed37a
commit 83f806b815
6 changed files with 45 additions and 43 deletions
+15 -15
View File
@@ -13,7 +13,7 @@ def register_user(username, password):
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, AS_PORT))
key = os.urandom(32)
message = f"register||{username},{password},{key}"
message = f"register||{username}<>{password}<>{key}"
s.sendall(message.encode())
data = s.recv(4096)
@@ -27,12 +27,12 @@ def login(username, password):
global CLIENT_ID, KEY
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, AS_PORT))
message = f"login||{username},{password}"
message = f"login||{username}<>{password}"
s.sendall(message.encode())
data = s.recv(4096)
data = data.decode()
data = data.split(",")
data = data.split("<>")
if data[0] == "success":
print(f"Login succeeded!")
CLIENT_ID = username
@@ -57,9 +57,9 @@ def print_something():
M4 = None
M6 = None
M1_inner = f"{ID_S},{T_R},{N1}"
M1_inner = f"{ID_S}<>{T_R}<>{N1}"
M1_inner = AES.encrypt(M1_inner)
M1 = f"{ID_C},{M1_inner}"
M1 = f"{ID_C}<>{M1_inner}"
message = f"request||{M1}"
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, AS_PORT))
@@ -75,9 +75,9 @@ def print_something():
M2 = [{K_c_tgs + N_1}Kc + T_c_tgs]
M3 = [{ID_C + ID_S + T_R + N2}K_c_tgs + T_c_tgs]
'''
M2 = M2.split(",")
M2 = M2.split("<>")
M2_inner = AES.decrypt(M2[0])
M2_inner = M2_inner.split(",")
M2_inner = M2_inner.split("<>")
K_c_tgs = eval(M2_inner[0])
N1 = M2_inner[1]
@@ -86,9 +86,9 @@ def print_something():
M2_AES = AESCipher(K_c_tgs)
N2 = randint(0, 1000000)
M3_inner = f"{ID_C},{ID_S},{T_R},{N2}"
M3_inner = f"{ID_C}<>{ID_S}<>{T_R}<>{N2}"
M3_inner = M2_AES.encrypt(M3_inner)
M3 = f"{M3_inner},{T_c_tgs}"
M3 = f"{M3_inner}<>{T_c_tgs}"
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, TGS_PORT))
message = f"request||{M3}"
@@ -105,10 +105,10 @@ def print_something():
T_c_s = {ID_C + T_A + K_c_s}K_s
M5 = [{ID_C + T_A + S_R + N3}K_c_s + T_c_s]
'''
M4 = M4.split(",")
M4 = M4.split("<>")
M4_AES = AESCipher(K_c_tgs)
M4_inner = M4_AES.decrypt(M4[0])
M4_inner = M4_inner.split(",")
M4_inner = M4_inner.split("<>")
print('M4_inner', M4_inner)
K_c_s = eval(M4_inner[0])
@@ -120,9 +120,9 @@ def print_something():
M5_AES = AESCipher(K_c_s)
N3 = randint(0, 1000000)
M5_inner = f"{ID_C},{T_A},{ID_S},{N3}"
M5_inner = f"{ID_C}<>{T_A}<>{ID_S}<>{N3}"
M5_inner = M5_AES.encrypt(M5_inner)
M5 = f"{M5_inner},{T_c_s}"
M5 = f"{M5_inner}<>{T_c_s}"
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, SERVICE_PORT))
@@ -138,10 +138,10 @@ def print_something():
'''
M6 = [{Resposta, N3}K_c_s]
'''
M6 = M6.split(",")
M6 = M6.split("<>")
M6_AES = AESCipher(K_c_s)
M6_inner = M6_AES.decrypt(M6[0])
M6_inner = M6_inner.split(",")
M6_inner = M6_inner.split("<>")
resposta = M6_inner[0]
N3 = M6_inner[1]