sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

test: threshold set to 2

Browse Source
This commit is contained in:
dqy 2023-10-24 21:00:19 +08:00
parent 4e577e41c7
commit 63933a1d19
1 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/client.py
View File

@ -96,6 +96,7 @@ class C(BaseModel):
Tuple: Tuple[capsule, int] Tuple: Tuple[capsule, int]
ip: str ip: str
# receive messages from nodes # receive messages from nodes
@app.post("/receive_messages") @app.post("/receive_messages")
async def receive_messages(message: C): async def receive_messages(message: C):
@ -180,6 +181,7 @@ async def send_messages(
): ):
global pk, sk global pk, sk
id_list = [] id_list = []
print(node_ips)
# calculate id of nodes # calculate id of nodes
for node_ip in node_ips: for node_ip in node_ips:
ip_parts = node_ip.split(".") ip_parts = node_ip.split(".")
@ -256,9 +258,6 @@ async def request_message(i_m: Request_Message):
(public_key, threshold), (public_key, threshold),
) )
try: try:
if response.status_code == 200: if response.status_code == 200:
data = response.json() data = response.json()
@ -301,7 +300,8 @@ async def recieve_request(i_m: IP_Message):
if source_ip != i_m.dest_ip: if source_ip != i_m.dest_ip:
return HTTPException(status_code=400, detail="Wrong ip") return HTTPException(status_code=400, detail="Wrong ip")
dest_ip = i_m.source_ip dest_ip = i_m.source_ip
threshold = random.randrange(1, 2) # threshold = random.randrange(1, 2)
threshold = 2
own_public_key = pk own_public_key = pk
pk_B = i_m.pk pk_B = i_m.pk
@ -326,7 +326,6 @@ async def recieve_request(i_m: IP_Message):
def get_own_ip() -> str: def get_own_ip() -> str:
ip = os.environ.get("HOST_IP", "IP not set") ip = os.environ.get("HOST_IP", "IP not set")
return ip return ip