fix problems

secretflow · Feb 7, 2025 · 9469172 · 9469172
1 parent f5a8fb2
commit 9469172
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/yacl/examples/sse/sse.cc b/yacl/examples/sse/sse.cc
@@ -24,7 +24,7 @@ Sse::Sse(int bucket_size, int slot_size, int lambda, int n_lambda)
 
 // EDBSetup
 std::pair<std::vector<std::vector<TSet::Record>>, std::string> Sse::EDBSetup() {
-  ProcessAndUpdateTAndXSet(iv_);
+  ProcessAndUpdateTAndXSet();
   auto [TSet, Kt] = tset_.TSetSetup(T_, keywords_);
   TSet_ = TSet;
   k_map_["Kt"] = Kt;
@@ -167,8 +167,6 @@ void Sse::Initialize() {
   keyValuePairs_ = keyValuePairs;
   reverseIndex_ = reverseIndex;
 
-  iv_ = yacl::crypto::RandU32();
-
   auto rand_bytes_Ks = yacl::crypto::RandU32();
   auto rand_bytes_Kx = yacl::crypto::RandU32();
   auto rand_bytes_Ki = yacl::crypto::RandU32();
@@ -183,7 +181,7 @@ void Sse::Initialize() {
 }
 
 // 主功能函数：计算并更新 T 和 XSet
-void Sse::ProcessAndUpdateTAndXSet(const uint128_t& iv_) {
+void Sse::ProcessAndUpdateTAndXSet() {
   yacl::crypto::HmacSha256 hmac_F_SSE_Ks(k_map_["Ks"]);
   yacl::crypto::HmacSha256 hmac_F_SSE_Kx(k_map_["Kx"]);
   yacl::crypto::HmacSha256 hmac_F_SSE_Ki(k_map_["Ki"]);

diff --git a/yacl/examples/sse/sse.h b/yacl/examples/sse/sse.h
@@ -65,7 +65,7 @@ class Sse {
                 const yacl::crypto::EcPoint& xtag,
                 const std::vector<yacl::crypto::EcPoint>& XSet);
   void Initialize();
-  void ProcessAndUpdateTAndXSet(const uint128_t& iv);
+  void ProcessAndUpdateTAndXSet();
 
   std::tuple<std::vector<std::string>,
              std::vector<std::pair<std::string, std::string>>,