hcq/openGauss-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

validate dblink conn: prevent acess to unknown addresses

Browse Source
This commit is contained in:
wuyuechuan
2023-09-16 15:04:47 +08:00
parent 52794627de
commit c4bd5e9098
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
contrib/dblink/dblink.cpp
View File

@ -974,10 +974,10 @@ Datum dblink_open(PG_FUNCTION_ARGS)
rconn = getConnectionByName(conname);
}
linker = rconn->linker;
if (linker == NULL) {
if (rconn == NULL || rconn->linker == NULL) {
DBLINK_CONN_NOT_AVAIL;
}
linker = rconn->linker;
/* Assemble sql */
appendStringInfo(&buf, "DECLARE %s CURSOR FOR %s", curname, sql);
@ -1028,10 +1028,10 @@ Datum dblink_close(PG_FUNCTION_ARGS)
rconn = getConnectionByName(conname);
}
linker = rconn->linker;
if (linker == NULL) {
if (rconn == NULL || rconn->linker == NULL) {
DBLINK_CONN_NOT_AVAIL;
}
linker = rconn->linker;
appendStringInfo(&buf, "CLOSE %s", curname);
@ -3113,4 +3113,4 @@ static void storeRowInit(storeInfo* sinfo, int nfields, bool first)
ALLOCSET_DEFAULT_INITSIZE,
ALLOCSET_DEFAULT_MAXSIZE);
}
}
}