ccyj/mimajingsai
1
0
Fork 0
forked from sangge/tpre-python
Code Pull Requests Actions Activity

fix: extract hash and kdf from setup

Browse Source
This commit is contained in:
sangge 2023-10-16 22:30:15 +08:00
parent 96e04c155e
commit cd1e884713
1 changed files with 45 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
src/tpre.py
View File

@ -118,8 +118,7 @@ def jacobianMultiply(
raise ValueError("jacobian Multiply error") raise ValueError("jacobian Multiply error")
def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int], def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
Tuple[int, int], Callable, Tuple[int, int]]:
Callable, Callable, Callable]:
''' '''
params: params:
sec: an init safety param sec: an init safety param
@ -128,10 +127,6 @@ def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
G: sm2 curve G: sm2 curve
g: generator g: generator
U: another generator U: another generator
use sm3 as hash function
hash2: G^2 -> Zq
hash3: G^3 -> Zq
hash4: G^3 * Zq -> Zq
''' '''
G = sm2p256v1 G = sm2p256v1
@ -141,50 +136,49 @@ def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
tmp_u = random.randint(0, sm2p256v1.P) tmp_u = random.randint(0, sm2p256v1.P)
U = multiply(g, tmp_u) U = multiply(g, tmp_u)
def hash2(double_G: Tuple[Tuple[int, int], Tuple[int, int]]) -> int: return G, g, U
sm3 = Sm3() #pylint: disable=e0602
for i in double_G: def hash2(double_G: Tuple[Tuple[int, int], Tuple[int, int]]) -> int:
for j in i: sm3 = Sm3() #pylint: disable=e0602
sm3.update(j.to_bytes(32)) for i in double_G:
digest = sm3.digest() for j in i:
digest = int.from_bytes(digest,'big') % sm2p256v1.P sm3.update(j.to_bytes(32))
return digest digest = sm3.digest()
digest = int.from_bytes(digest,'big') % sm2p256v1.P
def hash3(triple_G: Tuple[Tuple[int, int], return digest
Tuple[int, int],
Tuple[int, int]]) -> int: def hash3(triple_G: Tuple[Tuple[int, int],
sm3 = Sm3() #pylint: disable=e0602 Tuple[int, int],
for i in triple_G: Tuple[int, int]]) -> int:
for j in i: sm3 = Sm3() #pylint: disable=e0602
sm3.update(j.to_bytes(32)) for i in triple_G:
digest = sm3.digest() for j in i:
digest = int.from_bytes(digest, 'big') % sm2p256v1.P sm3.update(j.to_bytes(32))
return digest digest = sm3.digest()
digest = int.from_bytes(digest, 'big') % sm2p256v1.P
def hash4(triple_G: Tuple[Tuple[int, int], return digest
Tuple[int, int],
Tuple[int, int]], def hash4(triple_G: Tuple[Tuple[int, int],
Zp: int) -> int: Tuple[int, int],
sm3 = Sm3() #pylint: disable=e0602 Tuple[int, int]],
for i in triple_G: Zp: int) -> int:
for j in i: sm3 = Sm3() #pylint: disable=e0602
sm3.update(j.to_bytes(32)) for i in triple_G:
sm3.update(Zp.to_bytes(32)) for j in i:
digest = sm3.digest() sm3.update(j.to_bytes(32))
digest = int.from_bytes(digest, 'big') % sm2p256v1.P sm3.update(Zp.to_bytes(32))
return digest digest = sm3.digest()
digest = int.from_bytes(digest, 'big') % sm2p256v1.P
def KDF(G: Tuple[int, int]) -> int: return digest
sm3 = Sm3() #pylint: disable=e0602
for i in G: def KDF(G: Tuple[int, int]) -> int:
sm3.update(i.to_bytes(32)) sm3 = Sm3() #pylint: disable=e0602
digest = sm3.digest(32) for i in G:
digest = digest sm3.update(i.to_bytes(32))
digest = int.from_bytes(digest, 'big') % sm2p256v1.P digest = sm3.digest(32)
return digest digest = digest
digest = int.from_bytes(digest, 'big') % sm2p256v1.P
return digest
return G, g, U, hash2, hash3, hash4, KDF
def GenerateKeyPair( def GenerateKeyPair(
lamda_parma: int, lamda_parma: int,
@ -281,7 +275,7 @@ pk_B, sk_B = GenerateKeyPair(0, ())
sec = 256 sec = 256
# 调用Setup函数 # 调用Setup函数
G, g, U, hash2, hash3, hash4, KDF = Setup(sec) G, g, U= Setup(sec)
def GenerateReKey(sk_A, pk_B, N: int, T: int) -> list: def GenerateReKey(sk_A, pk_B, N: int, T: int) -> list:
''' '''