liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-06-29 17:21:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Switched to strtoint for sparsevec input

Browse Source
This commit is contained in:
Andrew Kane
2024-04-24 09:56:09 -07:00
parent c198fd58ee
commit b2a5259607
3 changed files with 32 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/sparsevec.c
View File

@@ -3,6 +3,7 @@
#include <limits.h>
#include <math.h>
#include "common/string.h"
#include "fmgr.h"
#include "halfutils.h"
#include "halfvec.h"
@@ -243,7 +244,7 @@ sparsevec_in(PG_FUNCTION_ARGS)
{
for (;;)
{
long index;
int index;
float value;
/* TODO Better error */
@@ -263,17 +264,17 @@ sparsevec_in(PG_FUNCTION_ARGS)
/* Use similar logic as int2vectorin */
errno = 0;
index = strtol(pt, &stringEnd, 10);
index = strtoint(pt, &stringEnd, 10);
if (stringEnd == pt)
ereport(ERROR,
(errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),
errmsg("invalid input syntax for type sparsevec: \"%s\"", lit)));
if (errno == ERANGE || index < 1 || index > INT_MAX)
if (errno == ERANGE)
ereport(ERROR,
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE),
errmsg("index \"%ld\" is out of range for type sparsevec", index)));
errmsg("index \"%s\" is out of range for type sparsevec", pnstrdup(pt, stringEnd - pt))));
pt = stringEnd;
@@ -352,13 +353,18 @@ sparsevec_in(PG_FUNCTION_ARGS)
/* Use similar logic as int2vectorin */
errno = 0;
dim = strtol(pt, &stringEnd, 10);
dim = strtoint(pt, &stringEnd, 10);
if (stringEnd == pt)
ereport(ERROR,
(errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),
errmsg("invalid input syntax for type sparsevec: \"%s\"", lit)));
if (errno == ERANGE)
ereport(ERROR,
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE),
errmsg("dimensions \"%s\" is out of range for type sparsevec", pnstrdup(pt, stringEnd - pt))));
pt = stringEnd;
/* Only whitespace is allowed after the closing brace */