liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-10 22:56:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added check for indexing overflow

Browse Source
This commit is contained in:
Andrew Kane
2022-10-30 12:01:24 -07:00
parent 1652a130c5
commit dc4a79a7e7
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/ivfkmeans.c
View File

@@ -204,6 +204,9 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers)
double dxcx; double dxcx;
double dxc; double dxc;
/* Ensure indexing does not overflow */
Assert(numCenters * numCenters <= INT_MAX);
/* Calculate allocation sizes */ /* Calculate allocation sizes */
Size samplesSize = VECTOR_ARRAY_SIZE(samples->maxlen, samples->dim); Size samplesSize = VECTOR_ARRAY_SIZE(samples->maxlen, samples->dim);
Size centersSize = VECTOR_ARRAY_SIZE(centers->maxlen, centers->dim); Size centersSize = VECTOR_ARRAY_SIZE(centers->maxlen, centers->dim);