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

add rust implementation
All checks were successful
Test CI / test speed (push) Successful in 15s

Browse Source
This commit is contained in:
sangge-redmi 2024-09-05 10:32:43 +08:00
parent 9654d8504b
commit 2d12e8c99c
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/tpre.py
View File

@ -1,6 +1,7 @@
from gmssl import Sm3, Sm2Key, Sm4Cbc, DO_ENCRYPT, DO_DECRYPT from gmssl import Sm3, Sm2Key, Sm4Cbc, DO_ENCRYPT, DO_DECRYPT
from typing import Tuple from typing import Tuple
import random import random
import ecc_rs
point = Tuple[int, int] point = Tuple[int, int]
capsule = Tuple[point, point, int] capsule = Tuple[point, point, int]
@ -33,17 +34,26 @@ sm2p256v1 = CurveFp(
g = (sm2p256v1.Gx, sm2p256v1.Gy) g = (sm2p256v1.Gx, sm2p256v1.Gy)
def multiply(a: point, n: int) -> point: def multiply(a: point, n: int, flag: int = 0) -> point:
N = sm2p256v1.N if flag == 1:
A = sm2p256v1.A
P = sm2p256v1.P result = ecc_rs.multiply(a, n)
return fromJacobian(jacobianMultiply(toJacobian(a), n, N, A, P), P) return result
else:
N = sm2p256v1.N
A = sm2p256v1.A
P = sm2p256v1.P
return fromJacobian(jacobianMultiply(toJacobian(a), n, N, A, P), P)
def add(a: point, b: point) -> point: def add(a: point, b: point, flag: int = 0) -> point:
A = sm2p256v1.A if flag == 1:
P = sm2p256v1.P result = ecc_rs.add(a, b)
return fromJacobian(jacobianAdd(toJacobian(a), toJacobian(b), A, P), P) return result
else:
A = sm2p256v1.A
P = sm2p256v1.P
return fromJacobian(jacobianAdd(toJacobian(a), toJacobian(b), A, P), P)
def inv(a: int, n: int) -> int: def inv(a: int, n: int) -> int: