diff --git a/src/client.py b/src/client.py index c9d270a..d9f9c58 100644 --- a/src/client.py +++ b/src/client.py @@ -112,8 +112,8 @@ async def receive_messages(message: C): return: status_code """ - a,b = message.Tuple - C_tuple = (a, b.to_bytes(32)) + a, b = message.Tuple + C_tuple = (a, b) ip = message.ip if not C_tuple or not ip: raise HTTPException(status_code=400, detail="Invalid input data") @@ -198,14 +198,15 @@ async def send_messages( rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list)) # type: ignore capsule, ct = Encrypt(pk, message) # type: ignore - capsule_ct = (capsule, int.from_bytes(ct)) + #capsule_ct = (capsule, int.from_bytes(ct)) for i in range(len(node_ips)): url = "http://" + node_ips[i][0] + ":8001" + "/user_src" payload = { "source_ip": local_ip, "dest_ip": dest_ip, - "capsule_ct": capsule_ct, + "capsule": capsule, + "ct": int.from_bytes(ct), "rk": rk_list[i], } print(payload) diff --git a/src/node.py b/src/node.py index 0644121..76eff38 100644 --- a/src/node.py +++ b/src/node.py @@ -76,7 +76,8 @@ async def send_heartbeat_internal() -> None: class Req(BaseModel): source_ip: str dest_ip: str - capsule_ct: Tuple[capsule, int] + capsule: capsule + ct: int rk: Any @@ -94,7 +95,8 @@ async def user_src(message: Req): """ source_ip = message.source_ip dest_ip = message.dest_ip - capsule, ct = message.capsule_ct + capsule = message.capsule + ct = message.ct capsule_ct = (capsule, ct.to_bytes(32)) rk = message.rk