From dc4a79a7e7031f951d403af5b8a39a759cc60894 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Sun, 30 Oct 2022 12:01:24 -0700 Subject: [PATCH] Added check for indexing overflow --- src/ivfkmeans.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/ivfkmeans.c b/src/ivfkmeans.c index 9987983..7f205a7 100644 --- a/src/ivfkmeans.c +++ b/src/ivfkmeans.c @@ -204,6 +204,9 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers) double dxcx; double dxc; + /* Ensure indexing does not overflow */ + Assert(numCenters * numCenters <= INT_MAX); + /* Calculate allocation sizes */ Size samplesSize = VECTOR_ARRAY_SIZE(samples->maxlen, samples->dim); Size centersSize = VECTOR_ARRAY_SIZE(centers->maxlen, centers->dim);