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

Merge branch 'main' of https://git.mamahaha.work/dqy/mimajingsai

Browse Source
This commit is contained in:
dqy 2023-10-24 22:34:11 +08:00
commit 6702a04d27
4 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
doc/README_app_en.md
View File

@ -7,10 +7,10 @@ get method
pr pr
docker run --rm -it -p 8000:8000 -p 8001:8001 -p 8002:8002 -v ~/mimajingsai:/app -e HOST_IP=110.41.130.197 git.mamahaha.work/sangge/tpre:base bash docker run -it -p 8000:8000 -p 8001:8001 -p 8002:8002 -v ~/mimajingsai:/app -e HOST_IP=110.41.130.197 git.mamahaha.work/sangge/tpre:base bash
tpre3: docker run --rm -it -p 8000:8000 -p 8001:8001 -p 8002:8002 -v ~/mimajingsai:/app -e HOST_IP=110.41.21.35 git.mamahaha.work/sangge/tpre:base bash tpre3: docker run -it -p 8000:8000 -p 8001:8001 -p 8002:8002 -v ~/mimajingsai:/app -e HOST_IP=110.41.21.35 git.mamahaha.work/sangge/tpre:base bash
110.41.155.96 tpre1 110.41.155.96 tpre1

5
src/client.py
View File

@ -9,6 +9,7 @@ from pydantic import BaseModel
import socket import socket
import random import random
import time import time
import base64
@asynccontextmanager @asynccontextmanager
@ -200,11 +201,10 @@ async def send_messages(
for i in range(len(node_ips)): for i in range(len(node_ips)):
url = "http://" + node_ips[i][0] + ":8001" + "/user_src" url = "http://" + node_ips[i][0] + ":8001" + "/user_src"
print(url)
payload = { payload = {
"source_ip": local_ip, "source_ip": local_ip,
"dest_ip": dest_ip, "dest_ip": dest_ip,
"capsule_ct": capsule_ct, "capsule_ct": base64.b64encode(str(capsule_ct)),
"rk": rk_list[i], "rk": rk_list[i],
} }
print(payload) print(payload)
@ -245,6 +245,7 @@ async def request_message(i_m: Request_Message):
} }
try: try:
response = requests.post(url, json=payload, timeout=3) response = requests.post(url, json=payload, timeout=3)
print(response.text)
except: except:
print("can't post") print("can't post")

2
src/client_cli.py
View File

@ -3,7 +3,7 @@ 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/" url = f"http://localhost:8002/request_message"
data = {"dest_ip": ip_addr, "message_name": message_name} data = {"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

7
src/node.py
View File

@ -7,6 +7,7 @@ from pydantic import BaseModel
from tpre import * from tpre import *
import os import os
from typing import Any, Tuple from typing import Any, Tuple
import base64
@asynccontextmanager @asynccontextmanager
@ -80,7 +81,7 @@ class Req(BaseModel):
@app.post("/user_src") # 接收用户1发送的信息 @app.post("/user_src") # 接收用户1发送的信息
async def receive_user_src_message(message: Req): async def user_src(message: Req):
global client_ip_src, client_ip_des global client_ip_src, client_ip_des
# kfrag , capsule_ct ,client_ip_src , client_ip_des = json_data[] # 看梁俊勇 # kfrag , capsule_ct ,client_ip_src , client_ip_des = json_data[] # 看梁俊勇
""" """
@ -91,10 +92,10 @@ async def receive_user_src_message(message: Req):
"rk": rk_list[i], "rk": rk_list[i],
} }
""" """
print(type(message))
source_ip = message.source_ip source_ip = message.source_ip
dest_ip = message.dest_ip dest_ip = message.dest_ip
capsule_ct = message.capsule_ct b64capsule_ct = message.capsule_ct
capsule_ct = base64.b64decode(b64capsule_ct)
rk = message.rk rk = message.rk
processed_message = ReEncrypt(rk, capsule_ct) processed_message = ReEncrypt(rk, capsule_ct)