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

sytle: modify code

Browse Source
This commit is contained in:
sangge 2023-10-23 17:30:17 +08:00
parent 989300a841
commit 4e67b03565
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/client.py
View File

@ -97,7 +97,7 @@ class C(BaseModel):
Tuple: Tuple[capsule, int] Tuple: Tuple[capsule, int]
ip: str ip: str
# receive messages from node # receive messages from nodes
@app.post("/receive_messages") @app.post("/receive_messages")
async def receive_messages(message: C): async def receive_messages(message: C):
""" """
@ -181,13 +181,17 @@ async def send_messages(
): ):
global pk, sk global pk, sk
id_list = [] id_list = []
# 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(".")
id = 0 id = 0
for i in range(4): for i in range(4):
id += int(ip_parts[i]) << (24 - (8 * i)) id += int(ip_parts[i]) << (24 - (8 * i))
id_list.append(id) id_list.append(id)
# generate rk
rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list)) # type: ignore rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list)) # type: ignore
capsule_ct = Encrypt(pk, message) # type: ignore capsule_ct = Encrypt(pk, message) # type: ignore
for i in range(len(node_ips)): for i in range(len(node_ips)):
@ -315,8 +319,6 @@ def get_own_ip() -> str:
# get node list from central server # get node list from central server
def get_node_list(count: int, server_addr: str): def get_node_list(count: int, server_addr: str):
url = "http://" + server_addr + "/server/send_nodes_list?count=" + str(count) url = "http://" + server_addr + "/server/send_nodes_list?count=" + str(count)
# payload = {"count": count}
# response = requests.post(url, json=payload)
response = requests.get(url) response = requests.get(url)
# Checking the response # Checking the response
if response.status_code == 200: if response.status_code == 200: