Cast to long when multiplyExact to avoid integer overflow

apache · Sep 18, 2024 · 4f81956 · 4f81956
1 parent 3acf852
commit 4f81956
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...re/src/java/org/apache/lucene/codecs/lucene912/Lucene912BinaryQuantizedVectorsReader.java b/...re/src/java/org/apache/lucene/codecs/lucene912/Lucene912BinaryQuantizedVectorsReader.java
@@ -140,7 +140,7 @@ static void validateFieldEntry(FieldInfo info, FieldEntry fieldEntry) {
     int correctionsCount =
         fieldEntry.similarityFunction != VectorSimilarityFunction.EUCLIDEAN ? 3 : 2;
     long numQuantizedVectorBytes =
-        Math.multiplyExact(binaryDims + (Float.BYTES * correctionsCount), fieldEntry.size);
+        Math.multiplyExact((binaryDims + (Float.BYTES * correctionsCount)), (long) fieldEntry.size);
     if (numQuantizedVectorBytes != fieldEntry.vectorDataLength) {
       throw new IllegalStateException(
           "Binarized vector data length "