From ee5ef9d8117d9662267540c8adf8ca0b61994965 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Fri, 13 Oct 2023 21:54:59 +0800
Subject: [PATCH] fix: use randint to generate U

---
 src/tpre.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/tpre.py b/src/tpre.py
index 584e59e..f75c1df 100644
--- a/src/tpre.py
+++ b/src/tpre.py
@@ -1,5 +1,6 @@
 from gmssl import * #pylint: disable = e0401 
 from typing import Tuple, Callable
+import random
 
 # 生成密钥对模块
 class CurveFp:
@@ -137,7 +138,7 @@ def Setup(sec: int) -> Tuple[CurveFp, Tuple[int, int],
     
     g = (sm2p256v1.Gx, sm2p256v1.Gy)
     
-    tmp_u = sec
+    tmp_u = random.randint(0, sm2p256v1.P)
     U = multiply(g, tmp_u)
     
     hash2 = Sm3() #pylint: disable=e0602