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

perf: rewrite DecapsulateFrags

Browse Source
This commit is contained in:
sangge 2023-10-18 16:02:21 +08:00
parent a68fcf1eec
commit a196530fde
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/tpre.py
View File

@ -213,7 +213,7 @@ def GenerateKeyPair(lamda_parma: int, public_params: tuple) -> Tuple[point, int]
# pk_B, sk_B = GenerateKeyPair(0, ()) # pk_B, sk_B = GenerateKeyPair(0, ())
def Encrypt(pk: point, m: bytes) -> Tuple[Tuple[point, point, int], bytes]: def Encrypt(pk: point, m: bytes) -> Tuple[capsule, bytes]:
enca = Encapsulate(pk) enca = Encapsulate(pk)
K = enca[0].to_bytes(16) K = enca[0].to_bytes(16)
capsule = enca[1] capsule = enca[1]
@ -361,8 +361,8 @@ def ReEncapsulate(kFrag: list, capsule: capsule) -> Tuple[point, point, int, poi
def ReEncrypt( def ReEncrypt(
kFrag: list, C: Tuple[capsule, int] kFrag: list, C: Tuple[capsule, bytes]
) -> Tuple[Tuple[point, point, int, point], int]: ) -> Tuple[Tuple[point, point, int, point], bytes]:
capsule, enc_Data = C capsule, enc_Data = C
cFrag = ReEncapsulate(kFrag, capsule) cFrag = ReEncapsulate(kFrag, capsule)
@ -395,13 +395,11 @@ def DecapsulateFrags(sk_B: int, pk_B: point, pk_A: point, cFrags: list) -> int:
Vlist = [] Vlist = []
idlist = [] idlist = []
X_Alist = [] X_Alist = []
t = 0
for cfrag in cFrags: # Ei,Vi,id,Xa = cFrag for cfrag in cFrags: # Ei,Vi,id,Xa = cFrag
Elist.append(cfrag[0]) Elist.append(cfrag[0])
Vlist.append(cfrag[1]) Vlist.append(cfrag[1])
idlist.append(cfrag[2]) idlist.append(cfrag[2])
X_Alist.append(cfrag[3]) X_Alist.append(cfrag[3])
t = t + 1 # 总共有t个片段,t为阈值
pkab = multiply(pk_A, sk_B) # pka^b pkab = multiply(pk_A, sk_B) # pka^b
D = hash6((pk_A, pk_B, pkab)) D = hash6((pk_A, pk_B, pkab))
@ -413,18 +411,18 @@ def DecapsulateFrags(sk_B: int, pk_B: point, pk_A: point, cFrags: list) -> int:
j = 1 j = 1
i = 1 i = 1
bi = 1 bi = 1
for i in range(t): for i in range(len(cFrags)):
for j in range(t): for j in range(len(cFrags)):
if j == i: if j != i:
# j=j+1 # bi = bi * (Sx[j] // (Sx[j] - Sx[i])) # 暂定整除
continue Sxj_sub_Sxi = (Sx[j] - Sx[i]) % sm2p256v1.P
else: Sxj_sub_Sxi_inv = inv(Sxj_sub_Sxi, sm2p256v1.P)
bi = bi * (Sx[j] // (Sx[j] - Sx[i])) # 暂定整除 bi = (bi * Sx[j] * Sxj_sub_Sxi_inv) % sm2p256v1.P
bis.append(bi) bis.append(bi)
E2 = multiply(Elist[0], bis[0]) # E^ 便于计算 E2 = multiply(Elist[0], bis[0]) # E^ 便于计算
V2 = multiply(Vlist[0], bis[0]) # V^ V2 = multiply(Vlist[0], bis[0]) # V^
for k in range(1, t): for k in range(1, len(cFrags)):
Ek = multiply(Elist[k], bis[k]) # EK/Vk 是个列表 Ek = multiply(Elist[k], bis[k]) # EK/Vk 是个列表
Vk = multiply(Vlist[k], bis[k]) Vk = multiply(Vlist[k], bis[k])
E2 = add(Ek, E2) E2 = add(Ek, E2)