From e4ce861ec71b46dd1af1db81123a44f74b3e4bdd Mon Sep 17 00:00:00 2001
From: dqy <1016751306@qq.com>
Date: Wed, 18 Oct 2023 16:49:40 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=B7=BB=E5=8A=A0f=E5=87=BD?=
 =?UTF-8?q?=E6=95=B0=E6=B3=A8=E9=87=8A=EF=BC=9Bf=E3=80=81Encapsulate?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=A8=A1P?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/tpre.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/tpre.py b/src/tpre.py
index a4eacd5..f425f9c 100644
--- a/src/tpre.py
+++ b/src/tpre.py
@@ -275,9 +275,18 @@ def H6(triple_G: Tuple[Tuple[int, int],
         return hash
 
 def f(x: int, f_modulus: list, T: int) -> int:
+    '''
+    功能: 通过多项式插值来实现信息的分散和重构
+    例如: 随机生成一个多项式f(x)=4x+5,质数P=11,其中f(0)=5,将多项式的系数分别分配给两个人,例如第一个人得到(1, 9),第二个人得到(2, 2).如果两个人都收集到了这两个点,那么可以使用拉格朗日插值法恢复原始的多项式,进而得到秘密信息"5"
+    param:
+    x, f_modulus(多项式系数列表), T(门限)
+    return:
+    res
+    '''
     res = 0
     for i in range(T):
         res += f_modulus[i] * pow(x, i)
+    res = res % sm2p256v1.P
     return res
 
 def GenerateReKey(sk_A, pk_B, N: int, T: int) -> list:
@@ -325,6 +334,7 @@ def Encapsulate(pk_A: Tuple[int, int]) -> Tuple[int, Tuple[Tuple[int, int], Tupl
     E = multiply(g, r)
     V = multiply(g, u)
     s = u + r * hash2((E, V))
+    s = s % sm2p256v1.P
     pk_A_ru = multiply(pk_A, r + u)
     K = KDF(pk_A_ru)
     capsule = (E, V, s)