fix vole mpfss index overflow

emp-toolkit · Feb 2, 2024 · 4a0d717 · 4a0d717
1 parent c2d1a16
commit 4a0d717
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/emp-zk/emp-vole-f2k/mpfss.h b/emp-zk/emp-vole-f2k/mpfss.h
@@ -67,7 +67,7 @@ template <typename IO> class MpfssRegF2k {
     if (triple_x == nullptr or party == ALICE)
       error("mpfss: set value vector error");
     for (int i = 0; i < tree_n; ++i) {
-      int pt = i * leave_n + item_pos_recver[i];
+      int64_t pt = (int64_t)i * leave_n + (int64_t)item_pos_recver[i];
       out[pt] = out[pt] ^ triple_x[i];
     }
   }

diff --git a/emp-zk/emp-vole/mpfss_reg.h b/emp-zk/emp-vole/mpfss_reg.h
@@ -67,7 +67,7 @@ template <typename IO> class MpfssRegFp {
 
   void set_vec_x(__uint128_t *out, __uint128_t *in) {
     for (int i = 0; i < tree_n; ++i) {
-      int pt = i * leave_n + (item_pos_recver[i] % leave_n);
+      int64_t pt = (int64_t)i * leave_n + ((int64_t)item_pos_recver[i] % leave_n);
       out[pt] = out[pt] ^ (__uint128_t)makeBlock(in[i], 0x0LL);
     }
   }