ecc_rs/benchmark.py

import ecc_rs
import time
import ecc_py
import random

point = tuple[int, int]
# Example point coordinates for P1 and P2 as tuples (x1, y1) and (x2, y2)
p1 = (
    1234567890123456789012345678901234567890123456789012345678901234,
    9876543210987654321098765432109876543210987654321098765432109876,
)
p2 = (
    2234567890123456789012345678901234567890123456789012345678901234,
    2876543210987654321098765432109876543210987654321098765432109876,
)

# Add the two points
result_x, result_y = ecc_rs.add(p1, p2)
print(f"Resulting Point: x = {result_x}, y = {result_y}")


result = ecc_rs.multiply(p1, 2)
print(f"Resulting Point: x = {result[0]}, y = {result[1]}")


# 生成密钥对模块
class CurveFp:
    def __init__(self, A, B, P, N, Gx, Gy, name):
        self.A = A
        self.B = B
        self.P = P
        self.N = N
        self.Gx = Gx
        self.Gy = Gy
        self.name = name


sm2p256v1 = CurveFp(
    name="sm2p256v1",
    A=0xFFFFFFFEFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF00000000FFFFFFFFFFFFFFFC,
    B=0x28E9FA9E9D9F5E344D5A9E4BCF6509A7F39789F515AB8F92DDBCBD414D940E93,
    P=0xFFFFFFFEFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF00000000FFFFFFFFFFFFFFFF,
    N=0xFFFFFFFEFFFFFFFFFFFFFFFFFFFFFFFF7203DF6B21C6052B53BBF40939D54123,
    Gx=0x32C4AE2C1F1981195F9904466A39C9948FE30BBFF2660BE1715A4589334C74C7,
    Gy=0xBC3736A2F4F6779C59BDCEE36B692153D0A9877CC62A474002DF32E52139F0A0,
)

# 生成元
g = (sm2p256v1.Gx, sm2p256v1.Gy)


def multiply(a: point, n: int) -> point:

    result = ecc_rs.multiply(a, n)
    return result


def add(a: point, b: point) -> point:
    result = ecc_rs.add(a, b)
    return result


random_mul_times = random.randint(0, sm2p256v1.N)

start_time = time.time()  # 获取开始时间
for i in range(10):
    result = multiply(g, random_mul_times)  # 执行函数
end_time = time.time()  # 获取结束时间
rs_mul_elapsed_time = end_time - start_time  # 计算执行时间
print(f"rust multiply 执行时间: {rs_mul_elapsed_time:.6f} 秒")
# rust multiply 执行时间: 0.000849 秒

start_time = time.time()  # 获取开始时间
for i in range(10):
    result = ecc_py.multiply(g, random_mul_times)  # 执行函数
end_time = time.time()  # 获取结束时间
py_mul_elapsed_time = end_time - start_time  # 计算执行时间
print(f"ecc_py multiply 执行时间: {py_mul_elapsed_time:.6f} 秒")
# ecc_py multiply 执行时间: 0.001374 秒

mul_improve = (py_mul_elapsed_time - rs_mul_elapsed_time) / py_mul_elapsed_time * 100

print(f"rust 实现 mul 提速 {mul_improve:.3f}%")

start_time = time.time()  # 获取开始时间
for i in range(10):
    result = add(g, g)  # 执行函数
end_time = time.time()  # 获取结束时间
rs_add_elapsed_time = end_time - start_time  # 计算执行时间
print(f"rust add 执行时间: {rs_add_elapsed_time:.6f} 秒")
# rust add 执行时间: 0.000378 秒

start_time = time.time()  # 获取开始时间
for i in range(10):
    result = ecc_py.add(g, g)  # 执行函数
end_time = time.time()  # 获取结束时间
py_add_elapsed_time = end_time - start_time  # 计算执行时间
print(f"ecc_py add 执行时间: {py_add_elapsed_time:.6f} 秒")
# ecc_py add 执行时间: 0.000510 秒

add_improve = (py_add_elapsed_time - rs_add_elapsed_time) / py_add_elapsed_time * 100

print(f"rust 实现 add 提速 {add_improve:.3f}%")