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

Merge pull request 'main' (#5) from dqy/mimajingsai:main into main

Browse Source 
Reviewed-on: sangge/mimajingsai#5
This commit is contained in:
sangge 2023-10-17 21:10:14 +08:00
commit 1de0e1c4f9
1 changed files with 30 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/tpre.py
View File

@ -22,7 +22,13 @@ sm2p256v1 = CurveFp(
Gx=0x32C4AE2C1F1981195F9904466A39C9948FE30BBFF2660BE1715A4589334C74C7, Gx=0x32C4AE2C1F1981195F9904466A39C9948FE30BBFF2660BE1715A4589334C74C7,
Gy=0xBC3736A2F4F6779C59BDCEE36B692153D0A9877CC62A474002DF32E52139F0A0 Gy=0xBC3736A2F4F6779C59BDCEE36B692153D0A9877CC62A474002DF32E52139F0A0
) )
# 椭圆曲线
G = sm2p256v1
# 生成元
g = (sm2p256v1.Gx, sm2p256v1.Gy)
def multiply(a: Tuple[int, int], n: int) -> Tuple[int, int]: def multiply(a: Tuple[int, int], n: int) -> Tuple[int, int]:
N = sm2p256v1.N N = sm2p256v1.N
A = sm2p256v1.A A = sm2p256v1.A
@ -116,27 +122,30 @@ def jacobianMultiply(
if (n % 2) == 1: if (n % 2) == 1:
return jacobianAdd(jacobianDouble(jacobianMultiply((Xp, Yp, Zp), n // 2, N, A, P), A, P), (Xp, Yp, Zp), A, P) return jacobianAdd(jacobianDouble(jacobianMultiply((Xp, Yp, Zp), n // 2, N, A, P), A, P), (Xp, Yp, Zp), A, P)
raise ValueError("jacobian Multiply error") raise ValueError("jacobian Multiply error")
def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int], # 生成元
Tuple[int, int]]: U = multiply(g, random.randint(0, sm2p256v1.P))
'''
params: # def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
sec: an init safety param # Tuple[int, int]]:
# '''
# params:
# sec: an init safety param
return: # return:
G: sm2 curve # G: sm2 curve
g: generator # g: generator
U: another generator # U: another generator
''' # '''
G = sm2p256v1 # G = sm2p256v1
g = (sm2p256v1.Gx, sm2p256v1.Gy) # g = (sm2p256v1.Gx, sm2p256v1.Gy)
tmp_u = random.randint(0, sm2p256v1.P) # tmp_u = random.randint(0, sm2p256v1.P)
U = multiply(g, tmp_u) # U = multiply(g, tmp_u)
return G, g, U # return G, g, U
def hash2(double_G: Tuple[Tuple[int, int], Tuple[int, int]]) -> int: def hash2(double_G: Tuple[Tuple[int, int], Tuple[int, int]]) -> int:
sm3 = Sm3() #pylint: disable=e0602 sm3 = Sm3() #pylint: disable=e0602
@ -204,6 +213,10 @@ def GenerateKeyPair(
return public_key, secret_key return public_key, secret_key
# 生成A和B的公钥和私钥
pk_A, sk_A = GenerateKeyPair(0, ())
pk_B, sk_B = GenerateKeyPair(0, ())
def Encrypt(pk: Tuple[int, int], m: int) -> Tuple[Tuple[ def Encrypt(pk: Tuple[int, int], m: int) -> Tuple[Tuple[
Tuple[int, int],Tuple[int, int], int], int]: Tuple[int, int],Tuple[int, int], int], int]:
enca = Encapsulate(pk) enca = Encapsulate(pk)
@ -267,16 +280,6 @@ def f(x: int, f_modulus: list, T: int) -> int:
res += f_modulus[i] * pow(x, i) res += f_modulus[i] * pow(x, i)
return res return res
# 生成A和B的公钥和私钥
pk_A, sk_A = GenerateKeyPair(0, ())
pk_B, sk_B = GenerateKeyPair(0, ())
# sec需要重新设置
sec = 256
# 调用Setup函数
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:
''' '''
param: param:
@ -341,7 +344,6 @@ def Checkcapsule(capsule:Tuple[Tuple[int,int],Tuple[int,int],int]) -> bool: #
return flag return flag
def ReEncapsulate(kFrag:list,capsule:Tuple[Tuple[int,int],Tuple[int,int],int]) -> Tuple[Tuple[int,int],Tuple[int,int],int,Tuple[int,int]] : def ReEncapsulate(kFrag:list,capsule:Tuple[Tuple[int,int],Tuple[int,int],int]) -> Tuple[Tuple[int,int],Tuple[int,int],int,Tuple[int,int]] :
id,rk,Xa,U1 = kFrag id,rk,Xa,U1 = kFrag
E,V,s = capsule E,V,s = capsule