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

29
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):
@ -144,7 +145,7 @@ async def receive_messages(message: C):
async def check_merge(ct: int, ip: str): async def check_merge(ct: int, ip: str):
global sk, pk, node_response, message global sk, pk, node_response, message
with sqlite3.connect("client.db") as db: with sqlite3.connect("client.db") as db:
# Check if the combination of ct_column and ip_column appears more than once. # Check if the combination of ct_column and ip_column appears more than once.
cursor = db.execute( cursor = db.execute(
""" """
SELECT capsule, ct SELECT capsule, ct
@ -166,8 +167,8 @@ async def check_merge(ct: int, ip: str):
(ip), (ip),
) )
result = cursor.fetchall() result = cursor.fetchall()
pk_sender, T = result[0] # result[0] = (pk, threshold) pk_sender, T = result[0] # result[0] = (pk, threshold)
if len(cfrag_cts) >= T: if len(cfrag_cts) >= T:
cfrags = mergecfrag(cfrag_cts) cfrags = mergecfrag(cfrag_cts)
message = DecryptFrags(sk, pk, pk_sender, cfrags) # type: ignore message = DecryptFrags(sk, pk, pk_sender, cfrags) # type: ignore
@ -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(".")
@ -187,10 +189,10 @@ async def send_messages(
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 # 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)):
@ -255,9 +257,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:
@ -282,7 +281,7 @@ async def request_message(i_m: Request_Message):
for _ in range(10): for _ in range(10):
if node_response: if node_response:
data = message data = message
# reset message and node_response # reset message and node_response
message = b"" message = b""
node_response = False node_response = False
@ -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
@ -315,18 +315,17 @@ async def recieve_request(i_m: IP_Message):
(threshold,), (threshold,),
) )
node_ips = cursor.fetchall() node_ips = cursor.fetchall()
# message name # message name
message = b"hello world" + random.randbytes(8) message = b"hello world" + random.randbytes(8)
# send message to nodes # send message to nodes
await send_messages(tuple(node_ips), message, dest_ip, pk_B, threshold) await send_messages(tuple(node_ips), message, dest_ip, pk_B, threshold)
response = {"threshold": threshold, "public_key": own_public_key} response = {"threshold": threshold, "public_key": own_public_key}
return response return response
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
@ -334,7 +333,7 @@ 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)
response = requests.get(url,timeout=3) response = requests.get(url, timeout=3)
# Checking the response # Checking the response
if response.status_code == 200: if response.status_code == 200:
print("Success get node list") print("Success get node list")