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

feat: finish setup() and GenKeyPair()

Browse Source
This commit is contained in:
sangge 2023-10-13 21:50:10 +08:00
parent 5d7fda5705
commit b3dcf06fd5
1 changed files with 32 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/tpre.py
View File

@ -28,7 +28,7 @@ def multiply(a: Tuple[int, int], n: int) -> Tuple[int, int]:
P = sm2p256v1.P P = sm2p256v1.P
return fromJacobian(jacobianMultiply(toJacobian(a), n, N, A, P), P) return fromJacobian(jacobianMultiply(toJacobian(a), n, N, A, P), P)
def add(a: Tuple[int, int], b: Tuple[int, int], A: int, P: int) -> Tuple[int, int]: def add(a: Tuple[int, int], b: Tuple[int, int]) -> Tuple[int, int]:
A = sm2p256v1.A A = sm2p256v1.A
P = sm2p256v1.P P = sm2p256v1.P
return fromJacobian(jacobianAdd(toJacobian(a), toJacobian(b), A, P), P) return fromJacobian(jacobianAdd(toJacobian(a), toJacobian(b), A, P), P)
@ -116,22 +116,44 @@ 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[int, int, int, Callable, Callable, Callable, Callable]: def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
Tuple[int, int], Callable,
Callable, Callable, Callable]:
''' '''
params: params:
sec: an init safety param sec: an init safety param
return: return:
G: G: sm2 curve
g: generator
U: another generator
sm3
hash2: G^2 -> Zq
hash3: G^3 -> Zq
hash4: G^3 * Zq -> Zq
''' '''
G = sm2p256v1
g = (sm2p256v1.Gx, sm2p256v1.Gy)
tmp_u = sec
U = multiply(g, tmp_u)
hash2 = Sm3() #pylint: disable=e0602
hash3 = Sm3() #pylint: disable=e0602
hash4 = Sm3() #pylint: disable=e0602
KDF = Sm3() #pylint: disable=e0602
return G, g, U, hash2, hash3, hash4, KDF return G, g, U, hash2, hash3, hash4, KDF
def GenerateKeyPair( def GenerateKeyPair(
lamda_parma: int, lamda_parma: int,
public_params: tuple public_params: tuple
) -> Tuple[Tuple[bytes, bytes], bytes]: ) -> Tuple[Tuple[int, int], int]:
''' '''
params: params:
lamda_param: an init safety param lamda_param: an init safety param
@ -140,16 +162,14 @@ def GenerateKeyPair(
return: return:
public_key, secret_key public_key, secret_key
''' '''
sm2 = Sm2Key() sm2 = Sm2Key() #pylint: disable=e0602
sm2.generate_key() sm2.generate_key()
public_key_x = bytes(sm2.public_key.x)
public_key_y = bytes(sm2.public_key.y) public_key_x = int.from_bytes(bytes(sm2.public_key.x),"big")
public_key_y = int.from_bytes(bytes(sm2.public_key.y),"big")
public_key = (public_key_x, public_key_y) public_key = (public_key_x, public_key_y)
secret_key = bytes(sm2.private_key)
print(private_key)
secret_key = int.from_bytes(bytes(sm2.private_key),"big")
return public_key, secret_key return public_key, secret_key