database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-05-17 06:41:24 +03:00
Code

Change tsearch2 to not use the unsafe practice of creating functions

Browse Source 
that return INTERNAL without also having INTERNAL arguments.  Since the
functions in question aren't meant to be called by hand anyway, I just
redeclared them to take 'internal' instead of 'text'.  Also add code
to ProcedureCreate() to enforce the restriction, as I should have done
to start with :-(
This commit is contained in:
Tom Lane 2005-05-03 16:51:45 +00:00
parent a935e36ae9
commit 308f01c304
4 changed files with 36 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
contrib/tsearch2/gendict/sql.IN
View File

@ -1,7 +1,7 @@
SET search_path = public; SET search_path = public;
BEGIN; BEGIN;
HASINIT create function dinit_CFG_MODNAME(text) HASINIT create function dinit_CFG_MODNAME(internal)
HASINIT returns internal HASINIT returns internal
HASINIT as 'MODULE_PATHNAME' HASINIT as 'MODULE_PATHNAME'
HASINIT language 'C'; HASINIT language 'C';

10
contrib/tsearch2/tsearch.sql.in
View File

@ -44,7 +44,7 @@ CREATE FUNCTION set_curdict(text)
with (isstrict); with (isstrict);
--built-in dictionaries --built-in dictionaries
CREATE FUNCTION dex_init(text) CREATE FUNCTION dex_init(internal)
returns internal returns internal
as 'MODULE_PATHNAME' as 'MODULE_PATHNAME'
language 'C'; language 'C';
@ -63,7 +63,7 @@ insert into pg_ts_dict select
'Simple example of dictionary.' 'Simple example of dictionary.'
; ;
CREATE FUNCTION snb_en_init(text) CREATE FUNCTION snb_en_init(internal)
returns internal returns internal
as 'MODULE_PATHNAME' as 'MODULE_PATHNAME'
language 'C'; language 'C';
@ -82,7 +82,7 @@ insert into pg_ts_dict select
'English Stemmer. Snowball.' 'English Stemmer. Snowball.'
; ;
CREATE FUNCTION snb_ru_init(text) CREATE FUNCTION snb_ru_init(internal)
returns internal returns internal
as 'MODULE_PATHNAME' as 'MODULE_PATHNAME'
language 'C'; language 'C';
@ -95,7 +95,7 @@ insert into pg_ts_dict select
'Russian Stemmer. Snowball.' 'Russian Stemmer. Snowball.'
; ;
CREATE FUNCTION spell_init(text) CREATE FUNCTION spell_init(internal)
returns internal returns internal
as 'MODULE_PATHNAME' as 'MODULE_PATHNAME'
language 'C'; language 'C';
@ -114,7 +114,7 @@ insert into pg_ts_dict select
'ISpell interface. Must have .dict and .aff files' 'ISpell interface. Must have .dict and .aff files'
; ;
CREATE FUNCTION syn_init(text) CREATE FUNCTION syn_init(internal)
returns internal returns internal
as 'MODULE_PATHNAME' as 'MODULE_PATHNAME'
language 'C'; language 'C';

10
contrib/tsearch2/untsearch.sql.in
View File

@ -34,14 +34,14 @@ DROP FUNCTION lexize(text, text);
DROP FUNCTION lexize(text); DROP FUNCTION lexize(text);
DROP FUNCTION set_curdict(int); DROP FUNCTION set_curdict(int);
DROP FUNCTION set_curdict(text); DROP FUNCTION set_curdict(text);
DROP FUNCTION dex_init(text); DROP FUNCTION dex_init(internal);
DROP FUNCTION dex_lexize(internal,internal,int4); DROP FUNCTION dex_lexize(internal,internal,int4);
DROP FUNCTION snb_en_init(text); DROP FUNCTION snb_en_init(internal);
DROP FUNCTION snb_lexize(internal,internal,int4); DROP FUNCTION snb_lexize(internal,internal,int4);
DROP FUNCTION snb_ru_init(text); DROP FUNCTION snb_ru_init(internal);
DROP FUNCTION spell_init(text); DROP FUNCTION spell_init(internal);
DROP FUNCTION spell_lexize(internal,internal,int4); DROP FUNCTION spell_lexize(internal,internal,int4);
DROP FUNCTION syn_init(text); DROP FUNCTION syn_init(internal);
DROP FUNCTION syn_lexize(internal,internal,int4); DROP FUNCTION syn_lexize(internal,internal,int4);
DROP FUNCTION set_curprs(int); DROP FUNCTION set_curprs(int);
DROP FUNCTION set_curprs(text); DROP FUNCTION set_curprs(text);

29
src/backend/catalog/pg_proc.c
View File

@ -8,7 +8,7 @@
* *
* *
* IDENTIFICATION * IDENTIFICATION
* $Header: /cvsroot/pgsql/src/backend/catalog/pg_proc.c,v 1.109 2003/10/03 19:26:49 tgl Exp $ * $Header: /cvsroot/pgsql/src/backend/catalog/pg_proc.c,v 1.109.2.1 2005/05/03 16:51:45 tgl Exp $
* *
*------------------------------------------------------------------------- *-------------------------------------------------------------------------
*/ */
@ -65,6 +65,8 @@ ProcedureCreate(const char *procedureName,
const Oid *parameterTypes) const Oid *parameterTypes)
{ {
int i; int i;
bool genericParam = false;
bool internalParam = false;
Relation rel; Relation rel;
HeapTuple tup; HeapTuple tup;
HeapTuple oldtup; HeapTuple oldtup;
@ -94,28 +96,35 @@ ProcedureCreate(const char *procedureName,
/* /*
* Do not allow return type ANYARRAY or ANYELEMENT unless at least one * Do not allow return type ANYARRAY or ANYELEMENT unless at least one
* argument is also ANYARRAY or ANYELEMENT * input argument is ANYARRAY or ANYELEMENT. Also, do not allow
* return type INTERNAL unless at least one input argument is INTERNAL.
*/ */
if (returnType == ANYARRAYOID || returnType == ANYELEMENTOID)
{
bool genericParam = false;
for (i = 0; i < parameterCount; i++) for (i = 0; i < parameterCount; i++)
{ {
if (parameterTypes[i] == ANYARRAYOID || switch (parameterTypes[i])
parameterTypes[i] == ANYELEMENTOID)
{ {
case ANYARRAYOID:
case ANYELEMENTOID:
genericParam = true; genericParam = true;
break; break;
case INTERNALOID:
internalParam = true;
break;
} }
} }
if (!genericParam) if ((returnType == ANYARRAYOID || returnType == ANYELEMENTOID)
&& !genericParam)
ereport(ERROR, ereport(ERROR,
(errcode(ERRCODE_INVALID_FUNCTION_DEFINITION), (errcode(ERRCODE_INVALID_FUNCTION_DEFINITION),
errmsg("cannot determine result data type"), errmsg("cannot determine result data type"),
errdetail("A function returning \"anyarray\" or \"anyelement\" must have at least one argument of either type."))); errdetail("A function returning \"anyarray\" or \"anyelement\" must have at least one argument of either type.")));
}
if (returnType == INTERNALOID && !internalParam)
ereport(ERROR,
(errcode(ERRCODE_INVALID_FUNCTION_DEFINITION),
errmsg("unsafe use of INTERNAL pseudo-type"),
errdetail("A function returning \"internal\" must have at least one \"internal\" argument.")));
/* Make sure we have a zero-padded param type array */ /* Make sure we have a zero-padded param type array */
MemSet(typev, 0, FUNC_MAX_ARGS * sizeof(Oid)); MemSet(typev, 0, FUNC_MAX_ARGS * sizeof(Oid));