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

feat: add hash234 function

Browse Source
This commit is contained in:
sangge 2023-10-14 13:03:43 +08:00
parent 6349576d48
commit 4bba049047
1 changed files with 44 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/tpre.py
View File

@ -12,7 +12,7 @@ class CurveFp:
self.Gx = Gx self.Gx = Gx
self.Gy = Gy self.Gy = Gy
self.name = name self.name = name
sm2p256v1 = CurveFp( sm2p256v1 = CurveFp(
name="sm2p256v1", name="sm2p256v1",
A=0xFFFFFFFEFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF00000000FFFFFFFFFFFFFFFC, A=0xFFFFFFFEFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF00000000FFFFFFFFFFFFFFFC,
@ -128,7 +128,7 @@ 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
sm3 use sm3 as hash function
hash2: G^2 -> Zq hash2: G^2 -> Zq
hash3: G^3 -> Zq hash3: G^3 -> Zq
hash4: G^3 * Zq -> Zq hash4: G^3 * Zq -> Zq
@ -141,11 +141,38 @@ 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)
hash2 = Sm3() #pylint: disable=e0602 def hash2(double_G: Tuple[Tuple[int, int], Tuple[int, int]]) -> int:
sm3 = Sm3() #pylint: disable=e0602
for i in double_G:
for j in i:
sm3.update(j.to_bytes())
digest = sm3.digest()
digest = int.from_bytes(digest,'big') % sm2p256v1.P
return digest
hash3 = Sm3() #pylint: disable=e0602 def hash3(triple_G: Tuple[Tuple[int, int],
Tuple[int, int],
Tuple[int, int]]) -> int:
sm3 = Sm3() #pylint: disable=e0602
for i in triple_G:
for j in i:
sm3.update(j.to_bytes())
digest = sm3.digest()
digest = int.from_bytes(digest,'big') % sm2p256v1.P
return digest
hash4 = Sm3() #pylint: disable=e0602 def hash4(triple_G: Tuple[Tuple[int, int],
Tuple[int, int],
Tuple[int, int]],
Zp: int) -> int:
sm3 = Sm3() #pylint: disable=e0602
for i in triple_G:
for j in i:
sm3.update(j.to_bytes())
sm3.update(Zp.to_bytes())
digest = sm3.digest()
digest = int.from_bytes(digest,'big') % sm2p256v1.P
return digest
KDF = Sm3() #pylint: disable=e0602 KDF = Sm3() #pylint: disable=e0602
@ -175,5 +202,15 @@ def GenerateKeyPair(
return public_key, secret_key return public_key, secret_key
def Enc(pk, m): def Enc(pk: Tuple[int, int], m: int) -> Tuple[Tuple[
pass Tuple[int, int],Tuple[int, int], int], int]:
enca = Encapsulate(pk)
K = enca[0]
capsule = enca[1]
sm4_enc = Sm4Cbc(key, iv, DO_ENCRYPT) #pylint: disable=e0602
plain_Data = m.to_bytes()
enc_Data = sm4_enc.update(plain_Data)
enc_Data += sm4_enc.finish()
enc_message = (capsule, enc_Data)
return enc_message