liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-12 15:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added checks for invalid indexes - #591

Browse Source
This commit is contained in:
Andrew Kane
2024-06-10 21:20:54 -07:00
parent 9dcf1bdc80
commit 9e91af5989
3 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/hnswutils.c
View File

@@ -300,6 +300,9 @@ HnswGetMetaPageInfo(Relation index, int *m, HnswElement * entryPoint)
page = BufferGetPage(buf); page = BufferGetPage(buf);
metap = HnswPageGetMeta(page); metap = HnswPageGetMeta(page);
if (unlikely(metap->magicNumber != HNSW_MAGIC_NUMBER))
elog(ERROR, "hnsw index is not valid");
if (m != NULL) if (m != NULL)
*m = metap->m; *m = metap->m;

3
src/ivfinsert.c
View File

@@ -19,6 +19,9 @@ FindInsertPage(Relation index, Datum *values, BlockNumber *insertPage, ListInfo
FmgrInfo *procinfo; FmgrInfo *procinfo;
Oid collation; Oid collation;
/* Ensure index is valid */
IvfflatGetMetaPageInfo(index, NULL, NULL);
/* Avoid compiler warning */ /* Avoid compiler warning */
listInfo->blkno = nextblkno; listInfo->blkno = nextblkno;
listInfo->offno = FirstOffsetNumber; listInfo->offno = FirstOffsetNumber;

6
src/ivfutils.c
View File

@@ -170,7 +170,11 @@ IvfflatGetMetaPageInfo(Relation index, int *lists, int *dimensions)
page = BufferGetPage(buf); page = BufferGetPage(buf);
metap = IvfflatPageGetMeta(page); metap = IvfflatPageGetMeta(page);
*lists = metap->lists; if (unlikely(metap->magicNumber != IVFFLAT_MAGIC_NUMBER))
elog(ERROR, "ivfflat index is not valid");
if (lists != NULL)
*lists = metap->lists;
if (dimensions != NULL) if (dimensions != NULL)
*dimensions = metap->dimensions; *dimensions = metap->dimensions;