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

fix: update

Browse Source
This commit is contained in:
Smart-SangGe 2023-10-24 22:40:12 +08:00
parent e9f523014f
commit 1c41ab94b2
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/client.py
View File

@ -197,14 +197,15 @@ async def send_messages(
# 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
capsule_ct = (capsule, int.from_bytes(ct))
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"
payload = { payload = {
"source_ip": local_ip, "source_ip": local_ip,
"dest_ip": dest_ip, "dest_ip": dest_ip,
"capsule_ct": base64.b64encode(str(capsule_ct)), "capsule_ct": capsule_ct,
"rk": rk_list[i], "rk": rk_list[i],
} }
print(payload) print(payload)

6
src/node.py
View File

@ -76,7 +76,7 @@ async def send_heartbeat_internal() -> None:
class Req(BaseModel): class Req(BaseModel):
source_ip: str source_ip: str
dest_ip: str dest_ip: str
capsule_ct: Tuple[capsule, bytes] capsule_ct: Tuple[capsule, int]
rk: Any rk: Any
@ -94,8 +94,8 @@ async def user_src(message: Req):
""" """
source_ip = message.source_ip source_ip = message.source_ip
dest_ip = message.dest_ip dest_ip = message.dest_ip
b64capsule_ct = message.capsule_ct capsule, ct = message.capsule_ct
capsule_ct = base64.b64decode(b64capsule_ct) capsule_ct = (capsule, ct.to_bytes())
rk = message.rk rk = message.rk
processed_message = ReEncrypt(rk, capsule_ct) processed_message = ReEncrypt(rk, capsule_ct)