ccyj/mimajingsai
1
0
Fork 0
forked from sangge/tpre-python
Code Pull Requests Actions Activity

test: test

Browse Source
This commit is contained in:
Smart-SangGe 2023-10-24 22:52:21 +08:00
parent 1c41ab94b2
commit 558552a441
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/client.py
View File

@ -112,8 +112,8 @@ async def receive_messages(message: C):
return: return:
status_code status_code
""" """
C_tuple = message.Tuple a,b = message.Tuple
C_tuple = (a, b.to_bytes(32))
ip = message.ip ip = message.ip
if not C_tuple or not ip: if not C_tuple or not ip:
raise HTTPException(status_code=400, detail="Invalid input data") raise HTTPException(status_code=400, detail="Invalid input data")

5
src/node.py
View File

@ -95,10 +95,11 @@ async def user_src(message: Req):
source_ip = message.source_ip source_ip = message.source_ip
dest_ip = message.dest_ip dest_ip = message.dest_ip
capsule, ct = message.capsule_ct capsule, ct = message.capsule_ct
capsule_ct = (capsule, ct.to_bytes()) capsule_ct = (capsule, ct.to_bytes(32))
rk = message.rk 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) await send_user_des_message(source_ip, dest_ip, processed_message)
return HTTPException(status_code=200, detail="message recieved") return HTTPException(status_code=200, detail="message recieved")