244 lines
8.8 KiB
Plaintext
244 lines
8.8 KiB
Plaintext
--
|
|
-- CREATE FUNCTION
|
|
--
|
|
-- sanity check of pg_proc catalog to the given parameters
|
|
--
|
|
CREATE USER regtest_unpriv_user;
|
|
CREATE SCHEMA temp_func_test;
|
|
GRANT ALL ON SCHEMA temp_func_test TO public;
|
|
SET search_path TO temp_func_test, public;
|
|
--
|
|
-- ARGUMENT and RETURN TYPES
|
|
--
|
|
CREATE FUNCTION functest_A_1(text, date) RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT $1 = ''abcd'' AND $2 > ''2001-01-01''';
|
|
CREATE FUNCTION functest_A_2(text[]) RETURNS int LANGUAGE 'sql'
|
|
AS 'SELECT $1[0]::int';
|
|
CREATE FUNCTION functest_A_3() RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT false';
|
|
SELECT proname, prorettype::regtype, proargtypes::regtype[] FROM pg_proc
|
|
WHERE oid in ('functest_A_1'::regproc,
|
|
'functest_A_2'::regproc,
|
|
'functest_A_3'::regproc) ORDER BY proname;
|
|
proname | prorettype | proargtypes
|
|
--------------+------------+-------------------
|
|
functest_a_1 | boolean | [0:1]={text,date}
|
|
functest_a_2 | integer | [0:0]={text[]}
|
|
functest_a_3 | boolean | {}
|
|
(3 rows)
|
|
|
|
--
|
|
-- IMMUTABLE | STABLE | VOLATILE
|
|
--
|
|
CREATE FUNCTION functest_B_1(int) RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT $1 > 0';
|
|
CREATE FUNCTION functest_B_2(int) RETURNS bool LANGUAGE 'sql'
|
|
IMMUTABLE AS 'SELECT $1 > 0';
|
|
CREATE FUNCTION functest_B_3(int) RETURNS bool LANGUAGE 'sql'
|
|
STABLE AS 'SELECT $1 = 0';
|
|
CREATE FUNCTION functest_B_4(int) RETURNS bool LANGUAGE 'sql'
|
|
VOLATILE AS 'SELECT $1 < 0';
|
|
SELECT proname, provolatile FROM pg_proc
|
|
WHERE oid in ('functest_B_1'::regproc,
|
|
'functest_B_2'::regproc,
|
|
'functest_B_3'::regproc,
|
|
'functest_B_4'::regproc) ORDER BY proname;
|
|
proname | provolatile
|
|
--------------+-------------
|
|
functest_b_1 | v
|
|
functest_b_2 | i
|
|
functest_b_3 | s
|
|
functest_b_4 | v
|
|
(4 rows)
|
|
|
|
ALTER FUNCTION functest_B_2(int) VOLATILE;
|
|
ALTER FUNCTION functest_B_3(int) COST 100; -- unrelated change, no effect
|
|
SELECT proname, provolatile FROM pg_proc
|
|
WHERE oid in ('functest_B_1'::regproc,
|
|
'functest_B_2'::regproc,
|
|
'functest_B_3'::regproc,
|
|
'functest_B_4'::regproc) ORDER BY proname;
|
|
proname | provolatile
|
|
--------------+-------------
|
|
functest_b_1 | v
|
|
functest_b_2 | v
|
|
functest_b_3 | s
|
|
functest_b_4 | v
|
|
(4 rows)
|
|
|
|
--
|
|
-- SECURITY DEFINER | INVOKER
|
|
--
|
|
CREATE FUNCTION functext_C_1(int) RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT $1 > 0';
|
|
CREATE FUNCTION functext_C_2(int) RETURNS bool LANGUAGE 'sql'
|
|
SECURITY DEFINER AS 'SELECT $1 = 0';
|
|
CREATE FUNCTION functext_C_3(int) RETURNS bool LANGUAGE 'sql'
|
|
SECURITY INVOKER AS 'SELECT $1 < 0';
|
|
SELECT proname, prosecdef FROM pg_proc
|
|
WHERE oid in ('functext_C_1'::regproc,
|
|
'functext_C_2'::regproc,
|
|
'functext_C_3'::regproc) ORDER BY proname;
|
|
proname | prosecdef
|
|
--------------+-----------
|
|
functext_c_1 | f
|
|
functext_c_2 | t
|
|
functext_c_3 | f
|
|
(3 rows)
|
|
|
|
ALTER FUNCTION functext_C_1(int) IMMUTABLE; -- unrelated change, no effect
|
|
ALTER FUNCTION functext_C_2(int) SECURITY INVOKER;
|
|
ALTER FUNCTION functext_C_3(int) SECURITY DEFINER;
|
|
SELECT proname, prosecdef FROM pg_proc
|
|
WHERE oid in ('functext_C_1'::regproc,
|
|
'functext_C_2'::regproc,
|
|
'functext_C_3'::regproc) ORDER BY proname;
|
|
proname | prosecdef
|
|
--------------+-----------
|
|
functext_c_1 | f
|
|
functext_c_2 | f
|
|
functext_c_3 | t
|
|
(3 rows)
|
|
|
|
--
|
|
-- LEAKPROOF
|
|
--
|
|
CREATE FUNCTION functext_E_1(int) RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT $1 > 100';
|
|
CREATE FUNCTION functext_E_2(int) RETURNS bool LANGUAGE 'sql'
|
|
LEAKPROOF AS 'SELECT $1 > 100';
|
|
SELECT proname, proleakproof FROM pg_proc
|
|
WHERE oid in ('functext_E_1'::regproc,
|
|
'functext_E_2'::regproc) ORDER BY proname;
|
|
proname | proleakproof
|
|
--------------+--------------
|
|
functext_e_1 | f
|
|
functext_e_2 | t
|
|
(2 rows)
|
|
|
|
ALTER FUNCTION functext_E_1(int) LEAKPROOF;
|
|
ALTER FUNCTION functext_E_2(int) STABLE; -- unrelated change, no effect
|
|
SELECT proname, proleakproof FROM pg_proc
|
|
WHERE oid in ('functext_E_1'::regproc,
|
|
'functext_E_2'::regproc) ORDER BY proname;
|
|
proname | proleakproof
|
|
--------------+--------------
|
|
functext_e_1 | t
|
|
functext_e_2 | t
|
|
(2 rows)
|
|
|
|
ALTER FUNCTION functext_E_2(int) NOT LEAKPROOF; -- remove leakproog attribute
|
|
SELECT proname, proleakproof FROM pg_proc
|
|
WHERE oid in ('functext_E_1'::regproc,
|
|
'functext_E_2'::regproc) ORDER BY proname;
|
|
proname | proleakproof
|
|
--------------+--------------
|
|
functext_e_1 | t
|
|
functext_e_2 | f
|
|
(2 rows)
|
|
|
|
-- it takes superuser privilege to turn on leakproof, but not for turn off
|
|
ALTER FUNCTION functext_E_1(int) OWNER TO regtest_unpriv_user;
|
|
ALTER FUNCTION functext_E_2(int) OWNER TO regtest_unpriv_user;
|
|
SET SESSION AUTHORIZATION regtest_unpriv_user;
|
|
SET search_path TO temp_func_test, public;
|
|
ALTER FUNCTION functext_E_1(int) NOT LEAKPROOF;
|
|
ALTER FUNCTION functext_E_2(int) LEAKPROOF;
|
|
ERROR: only superuser can define a leakproof function
|
|
CREATE FUNCTION functext_E_3(int) RETURNS bool LANGUAGE 'sql'
|
|
LEAKPROOF AS 'SELECT $1 < 200'; -- failed
|
|
ERROR: only superuser can define a leakproof function
|
|
RESET SESSION AUTHORIZATION;
|
|
--
|
|
-- CALLED ON NULL INPUT | RETURNS NULL ON NULL INPUT | STRICT
|
|
--
|
|
CREATE FUNCTION functext_F_1(int) RETURNS bool LANGUAGE 'sql'
|
|
AS 'SELECT $1 > 50';
|
|
CREATE FUNCTION functext_F_2(int) RETURNS bool LANGUAGE 'sql'
|
|
CALLED ON NULL INPUT AS 'SELECT $1 = 50';
|
|
CREATE FUNCTION functext_F_3(int) RETURNS bool LANGUAGE 'sql'
|
|
RETURNS NULL ON NULL INPUT AS 'SELECT $1 < 50';
|
|
CREATE FUNCTION functext_F_4(int) RETURNS bool LANGUAGE 'sql'
|
|
STRICT AS 'SELECT $1 = 50';
|
|
SELECT proname, proisstrict FROM pg_proc
|
|
WHERE oid in ('functext_F_1'::regproc,
|
|
'functext_F_2'::regproc,
|
|
'functext_F_3'::regproc,
|
|
'functext_F_4'::regproc) ORDER BY proname;
|
|
proname | proisstrict
|
|
--------------+-------------
|
|
functext_f_1 | f
|
|
functext_f_2 | f
|
|
functext_f_3 | t
|
|
functext_f_4 | t
|
|
(4 rows)
|
|
|
|
ALTER FUNCTION functext_F_1(int) IMMUTABLE; -- unrelated change, no effect
|
|
ALTER FUNCTION functext_F_2(int) STRICT;
|
|
ALTER FUNCTION functext_F_3(int) CALLED ON NULL INPUT;
|
|
SELECT proname, proisstrict FROM pg_proc
|
|
WHERE oid in ('functext_F_1'::regproc,
|
|
'functext_F_2'::regproc,
|
|
'functext_F_3'::regproc,
|
|
'functext_F_4'::regproc) ORDER BY proname;
|
|
proname | proisstrict
|
|
--------------+-------------
|
|
functext_f_1 | f
|
|
functext_f_2 | t
|
|
functext_f_3 | f
|
|
functext_f_4 | t
|
|
(4 rows)
|
|
|
|
-- information_schema tests
|
|
CREATE FUNCTION functest_IS_1(a int, b int default 1, c text default 'foo')
|
|
RETURNS int
|
|
LANGUAGE SQL
|
|
AS 'SELECT $1 + $2';
|
|
CREATE FUNCTION functest_IS_2(out a int, b int default 1)
|
|
RETURNS int
|
|
LANGUAGE SQL
|
|
AS 'SELECT $1';
|
|
CREATE FUNCTION functest_IS_3(a int default 1, out b int)
|
|
RETURNS int
|
|
LANGUAGE SQL
|
|
AS 'SELECT $1';
|
|
SELECT routine_name, ordinal_position, parameter_name, parameter_default
|
|
FROM information_schema.parameters JOIN information_schema.routines USING (specific_schema, specific_name)
|
|
WHERE routine_schema = 'temp_func_test' AND routine_name ~ '^functest_is_'
|
|
ORDER BY 1, 2;
|
|
routine_name | ordinal_position | parameter_name | parameter_default
|
|
---------------+------------------+----------------+-------------------
|
|
functest_is_1 | 1 | a |
|
|
functest_is_1 | 2 | b | 1
|
|
functest_is_1 | 3 | c | 'foo'::text
|
|
functest_is_2 | 1 | a |
|
|
functest_is_2 | 2 | b | 1
|
|
functest_is_3 | 1 | a | 1
|
|
functest_is_3 | 2 | b |
|
|
(7 rows)
|
|
|
|
-- Cleanups
|
|
DROP SCHEMA temp_func_test CASCADE;
|
|
NOTICE: drop cascades to 19 other objects
|
|
DETAIL: drop cascades to function functest_a_1(text,date)
|
|
drop cascades to function functest_a_2(text[])
|
|
drop cascades to function functest_a_3()
|
|
drop cascades to function functest_b_1(integer)
|
|
drop cascades to function functest_b_2(integer)
|
|
drop cascades to function functest_b_3(integer)
|
|
drop cascades to function functest_b_4(integer)
|
|
drop cascades to function functext_c_1(integer)
|
|
drop cascades to function functext_c_2(integer)
|
|
drop cascades to function functext_c_3(integer)
|
|
drop cascades to function functext_e_1(integer)
|
|
drop cascades to function functext_e_2(integer)
|
|
drop cascades to function functext_f_1(integer)
|
|
drop cascades to function functext_f_2(integer)
|
|
drop cascades to function functext_f_3(integer)
|
|
drop cascades to function functext_f_4(integer)
|
|
drop cascades to function functest_is_1(integer,integer,text)
|
|
drop cascades to function functest_is_2(integer)
|
|
drop cascades to function functest_is_3(integer)
|
|
DROP USER regtest_unpriv_user;
|
|
RESET search_path;
|