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

main #8

Merged
sangge merged 4 commits from ccyj/mimajingsai:main into main 2023-10-18 10:21:01 +08:00
Conversation 0 Commits 4 Files Changed 1 +72 -76
1 changed files with 72 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 523bae52a9 - Show all commits

62
src/tpre.py
View File

@ -23,6 +23,12 @@ sm2p256v1 = CurveFp(
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
@ -117,29 +123,29 @@ def jacobianMultiply(
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], Callable, U = multiply(g, random.randint(0, sm2p256v1.P))
Callable, Callable, Callable]:
'''
params:
sec: an init safety param
return: # def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
G: sm2 curve # Tuple[int, int]]:
g: generator # '''
U: another generator # params:
use sm3 as hash function # sec: an init safety param
hash2: G^2 -> Zq
hash3: G^3 -> Zq
hash4: G^3 * Zq -> Zq
'''
G = sm2p256v1 # return:
# G: sm2 curve
# g: generator
# U: another generator
# '''
g = (sm2p256v1.Gx, sm2p256v1.Gy) # G = sm2p256v1
tmp_u = random.randint(0, sm2p256v1.P) # g = (sm2p256v1.Gx, sm2p256v1.Gy)
U = multiply(g, tmp_u)
# tmp_u = random.randint(0, sm2p256v1.P)
# U = multiply(g, tmp_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
@ -183,9 +189,6 @@ def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
digest = int.from_bytes(digest, 'big') % sm2p256v1.P digest = int.from_bytes(digest, 'big') % sm2p256v1.P
return digest return digest
return G, g, U, hash2, hash3, hash4, KDF
def GenerateKeyPair( def GenerateKeyPair(
lamda_parma: int, lamda_parma: int,
public_params: tuple public_params: tuple
@ -210,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)
@ -273,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, hash2, hash3, hash4, KDF = 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:
@ -347,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