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

Hardened VectorArrayGet and VectorArraySet [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2026-06-18 12:56:12 -07:00
parent 6731c49811
commit 08c4e7ff10
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ivfflat.h
View File

@@ -305,6 +305,9 @@ typedef IvfflatScanOpaqueData * IvfflatScanOpaque;
static inline Pointer static inline Pointer
VectorArrayGet(VectorArray arr, int offset) VectorArrayGet(VectorArray arr, int offset)
{ {
if (offset >= arr->maxlen)
elog(ERROR, "safety check failed");
return ((char *) arr->items) + (offset * arr->itemsize); return ((char *) arr->items) + (offset * arr->itemsize);
} }
@@ -313,7 +316,7 @@ VectorArraySet(VectorArray arr, int offset, Pointer val)
{ {
Size size = VARSIZE_ANY(val); Size size = VARSIZE_ANY(val);
if (size > arr->itemsize) if (offset >= arr->maxlen || size > arr->itemsize)
elog(ERROR, "safety check failed"); elog(ERROR, "safety check failed");
memcpy(VectorArrayGet(arr, offset), val, size); memcpy(VectorArrayGet(arr, offset), val, size);