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

Merge pull request 'test: test' (#28) from sangge/mimajingsai:main into main

Browse Source 
Reviewed-on: dqy/mimajingsai#28
This commit is contained in:
dqy 2023-10-24 22:32:51 +08:00
commit f6b074fd62
3 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
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
@ -199,14 +200,14 @@ 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)
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)
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,

2
src/client_cli.py
View File

@ -2,7 +2,7 @@ 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, "dest_ip": ip_addr,
"message_name": message_name "message_name": message_name

9
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)
@ -107,7 +108,7 @@ async def send_user_des_message(source_ip: str, dest_ip: str, re_message): #
# 发送 HTTP POST 请求 # 发送 HTTP POST 请求
response = requests.post( response = requests.post(
"http://" + dest_ip + "/receive_messages?message", json=data "http://" + dest_ip + "/receive_messages", json=data
) )
print(response) print(response)