fix issue peihanmiao#6

Ni-kaFruit · Aug 5, 2022 · 76faaef · 76faaef
1 parent 694921c
commit 76faaef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/PSI/src/PsiReceiver.cpp b/PSI/src/PsiReceiver.cpp
@@ -141,7 +141,7 @@ namespace PSI {
 
 			for (auto i = 0; i < w; ++i) {
 				for (auto j = 0; j < receiverSize; ++j) {
-					auto location = (*(u32*)(transLocations[i] + j * locationInBytes)) & shift;
+					auto location = ((*(u32*)(transLocations[i] + j * locationInBytes)) & shift) % height;
 
 					matrixDelta[i][location >> 3] &= ~(1 << (location & 7));
 				}
@@ -174,7 +174,7 @@ namespace PSI {
 
 			for (auto i = 0; i < w; ++i) {
 				for (auto j = 0; j < receiverSize; ++j) {
-					auto location = (*(u32*)(transLocations[i] + j * locationInBytes)) & shift;
+					auto location = ((*(u32*)(transLocations[i] + j * locationInBytes)) & shift) % height;
 
 					transHashInputs[i + wLeft][j >> 3] |= (u8)((bool)(matrixA[i][location >> 3] & (1 << (location & 7)))) << (j & 7);
 				}		

diff --git a/PSI/src/PsiSender.cpp b/PSI/src/PsiSender.cpp
@@ -142,7 +142,7 @@ namespace PSI {
 
 			for (auto i = 0; i < w; ++i) {
 				for (auto j = 0; j < senderSize; ++j) {
-					auto location = (*(u32*)(transLocations[i] + j * locationInBytes)) & shift;
+					auto location = ((*(u32*)(transLocations[i] + j * locationInBytes)) & shift) % height;
 
 					transHashInputs[i + wLeft][j >> 3] |= (u8)((bool)(matrixC[i][location >> 3] & (1 << (location & 7)))) << (j & 7);
 				}