diff --git a/src/client.py b/src/client.py index 2f0e8c9..433197d 100644 --- a/src/client.py +++ b/src/client.py @@ -15,6 +15,13 @@ import pickle from fastapi.responses import JSONResponse import asyncio +# 测试文本 +test_msessgaes = { + "name": b"proxy re-encryption", + "environment": b"distributed environment", + "usage": b"data sharing", +} + @asynccontextmanager async def lifespan(app: FastAPI): @@ -211,7 +218,7 @@ def check_merge(ct: int, ip: str): print("pk:", type(pk)) print("pk_sender:", type(pk_sender)) print("cfrags:", type(cfrags)) - message = DecryptFrags(sk, pk, pk_sender, cfrags) + message = DecryptFrags(sk, pk, pk_sender, cfrags) # type: ignore print("merge success", message) node_response = True @@ -359,7 +366,11 @@ async def receive_request(i_m: IP_Message): # message name # message_name = i_m.message_name # message = xxxxx - message = b"hello world" + random.randbytes(8) + + # 根据message name到测试文本查找对应值 + message = test_msessgaes[i_m.message_name] + + # message = b"hello world" + random.randbytes(8) print(f"Generated message: {message}") # send message to nodes