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

test:test

Browse Source
This commit is contained in:
dqy 2023-10-24 22:33:07 +08:00
parent 3f5fbbcc91
commit 97f5b4eac7
3 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/client.py
View File

@ -199,7 +199,7 @@ async def send_messages(
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)):
url = "http://" + node_ips[i][0] + ":8001" + "/user_src?message" url = "http://" + node_ips[i][0] + ":8001" + "/user_src"
print(url) print(url)
payload = { payload = {
"source_ip": local_ip, "source_ip": local_ip,
@ -207,6 +207,7 @@ async def send_messages(
"capsule_ct": capsule_ct, "capsule_ct": capsule_ct,
"rk": rk_list[i], "rk": rk_list[i],
} }
print(payload)
response = requests.post(url, json=payload) response = requests.post(url, json=payload)
if response.status_code == 200: if response.status_code == 200:
@ -235,7 +236,7 @@ async def request_message(i_m: Request_Message):
message_name = i_m.message_name message_name = i_m.message_name
source_ip = get_own_ip() source_ip = get_own_ip()
dest_port = "8002" dest_port = "8002"
url = "http://" + dest_ip + ":" + dest_port + "/recieve_request?i_m" url = "http://" + dest_ip + ":" + dest_port + "/recieve_request"
payload = { payload = {
"dest_ip": dest_ip, "dest_ip": dest_ip,
"message_name": message_name, "message_name": message_name,

10
src/client_cli.py
View File

@ -1,15 +1,14 @@
import argparse import argparse
import requests import requests
def send_post_request(ip_addr, message_name): def send_post_request(ip_addr, message_name):
url = f"http://localhost:8002/request_message/?i_m" url = f"http://localhost:8002/request_message/"
data = { data = {"dest_ip": ip_addr, "message_name": message_name}
"dest_ip": ip_addr,
"message_name": message_name
}
response = requests.post(url, json=data) response = requests.post(url, json=data)
return response.text return response.text
def main(): def main():
parser = argparse.ArgumentParser(description="Send POST request to a specified IP.") parser = argparse.ArgumentParser(description="Send POST request to a specified IP.")
parser.add_argument("ip_addr", help="IP address to send request to.") parser.add_argument("ip_addr", help="IP address to send request to.")
@ -19,5 +18,6 @@ def main():
response = send_post_request(args.ip_addr, args.message_name) response = send_post_request(args.ip_addr, args.message_name)
print(response) print(response)
if __name__ == "__main__": if __name__ == "__main__":
main() main()

4
src/node.py
View File

@ -106,9 +106,7 @@ async def send_user_des_message(source_ip: str, dest_ip: str, re_message): #
data = {"Tuple": re_message, "ip": source_ip} # 类型不匹配 data = {"Tuple": re_message, "ip": source_ip} # 类型不匹配
# 发送 HTTP POST 请求 # 发送 HTTP POST 请求
response = requests.post( response = requests.post("http://" + dest_ip + "/receive_messages", json=data)
"http://" + dest_ip + "/receive_messages?message", json=data
)
print(response) print(response)