diff --git a/src/client.py b/src/client.py index c718cb0..c9d270a 100644 --- a/src/client.py +++ b/src/client.py @@ -112,8 +112,8 @@ async def receive_messages(message: C): return: status_code """ - C_tuple = message.Tuple - + a,b = message.Tuple + C_tuple = (a, b.to_bytes(32)) ip = message.ip if not C_tuple or not ip: raise HTTPException(status_code=400, detail="Invalid input data") diff --git a/src/node.py b/src/node.py index 228e4a9..3998ada 100644 --- a/src/node.py +++ b/src/node.py @@ -95,10 +95,11 @@ async def user_src(message: Req): source_ip = message.source_ip dest_ip = message.dest_ip capsule, ct = message.capsule_ct - capsule_ct = (capsule, ct.to_bytes()) + capsule_ct = (capsule, ct.to_bytes(32)) rk = message.rk - processed_message = ReEncrypt(rk, capsule_ct) + a, b = ReEncrypt(rk, capsule_ct) + processed_message = (a, int.from_bytes(b)) await send_user_des_message(source_ip, dest_ip, processed_message) return HTTPException(status_code=200, detail="message recieved")