Merge pull request 'test: test' (#30) from sangge/mimajingsai:main into main

Reviewed-on: dqy/mimajingsai#30
This commit is contained in:
dqy 2023-10-24 22:53:03 +08:00
commit af8c822c06
2 changed files with 4 additions and 3 deletions

View File

@ -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")

View File

@ -98,7 +98,8 @@ async def user_src(message: Req):
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")