sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

feat: add send_messages debug info

Browse Source
This commit is contained in:
Smart-SangGe 2023-10-27 00:52:47 +08:00
parent 995a644603
commit 33ee464466
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/client.py
View File

@ -123,7 +123,7 @@ async def receive_messages(message: C):
status_code status_code
""" """
print(f"Received message: {message}") print(f"Received message: {message}")
if not message.Tuple or not message.ip: if not message.Tuple or not message.ip:
print("Invalid input data received.") print("Invalid input data received.")
raise HTTPException(status_code=400, detail="Invalid input data") raise HTTPException(status_code=400, detail="Invalid input data")
@ -133,7 +133,6 @@ async def receive_messages(message: C):
# Serialization # Serialization
bin_C_capsule = pickle.dumps(C_capsule) bin_C_capsule = pickle.dumps(C_capsule)
# insert record into database # insert record into database
with sqlite3.connect("client.db") as db: with sqlite3.connect("client.db") as db:
@ -235,11 +234,13 @@ async def send_messages(
for i in range(4): for i in range(4):
id += int(ip_parts[i]) << (24 - (8 * i)) id += int(ip_parts[i]) << (24 - (8 * i))
id_list.append(id) id_list.append(id)
print(f"Calculated IDs: {id_list}")
# generate rk # generate rk
rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list)) # type: ignore rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list)) # type: ignore
print(f"Generated ReKey list: {rk_list}")
capsule, ct = Encrypt(pk, message) # type: ignore capsule, ct = Encrypt(pk, message) # type: ignore
# capsule_ct = (capsule, int.from_bytes(ct)) # capsule_ct = (capsule, int.from_bytes(ct))
print(f"Encrypted message to capsule={capsule}, ct={ct}")
for i in range(len(node_ips)): for i in range(len(node_ips)):
url = "http://" + node_ips[i][0] + ":8001" + "/user_src" url = "http://" + node_ips[i][0] + ":8001" + "/user_src"
@ -250,11 +251,15 @@ async def send_messages(
"ct": int.from_bytes(ct), "ct": int.from_bytes(ct),
"rk": rk_list[i], "rk": rk_list[i],
} }
print(json.dumps(payload)) print(f"Sending payload to {url}: {json.dumps(payload)}")
response = requests.post(url, json=payload) response = requests.post(url, json=payload)
if response.status_code == 200: if response.status_code == 200:
print(f"send to {node_ips[i]} successful") print(f"send to {node_ips[i]} successful")
else:
print(
f"Failed to send to {node_ips[i]}. Response code: {response.status_code}, Response text: {response.text}"
)
return 0 return 0