1
0
mirror of https://github.com/postgres/postgres.git synced 2025-11-06 07:49:08 +03:00

Add test scaffolding for soft error reporting from input functions.

pg_input_is_valid() returns boolean, while pg_input_error_message()
returns the primary error message if the input is bad, or NULL
if the input is OK.  The main reason for having two functions is
so that we can test both the details-wanted and the no-details-wanted
code paths.

Although these are primarily designed with testing in mind,
it could well be that they'll be useful to end users as well.

This patch is mostly by me, but it owes very substantial debt to
earlier work by Nikita Glukhov, Andrew Dunstan, and Amul Sul.
Thanks to Andres Freund for review.

Discussion: https://postgr.es/m/3bbbb0df-7382-bf87-9737-340ba096e034@postgrespro.ru
This commit is contained in:
Tom Lane
2022-12-09 10:08:44 -05:00
parent d9f7f5d32f
commit 1939d26282
7 changed files with 277 additions and 1 deletions

View File

@@ -32,6 +32,8 @@
#include "common/keywords.h"
#include "funcapi.h"
#include "miscadmin.h"
#include "nodes/miscnodes.h"
#include "parser/parse_type.h"
#include "parser/scansup.h"
#include "pgstat.h"
#include "postmaster/syslogger.h"
@@ -45,6 +47,25 @@
#include "utils/ruleutils.h"
#include "utils/timestamp.h"
/*
* structure to cache metadata needed in pg_input_is_valid_common
*/
typedef struct ValidIOData
{
Oid typoid;
int32 typmod;
bool typname_constant;
Oid typiofunc;
Oid typioparam;
FmgrInfo inputproc;
} ValidIOData;
static bool pg_input_is_valid_common(FunctionCallInfo fcinfo,
text *txt, text *typname,
ErrorSaveContext *escontext);
/*
* Common subroutine for num_nulls() and num_nonnulls().
* Returns true if successful, false if function should return NULL.
@@ -640,6 +661,114 @@ pg_column_is_updatable(PG_FUNCTION_ARGS)
}
/*
* pg_input_is_valid - test whether string is valid input for datatype.
*
* Returns true if OK, false if not.
*
* This will only work usefully if the datatype's input function has been
* updated to return "soft" errors via errsave/ereturn.
*/
Datum
pg_input_is_valid(PG_FUNCTION_ARGS)
{
text *txt = PG_GETARG_TEXT_PP(0);
text *typname = PG_GETARG_TEXT_PP(1);
ErrorSaveContext escontext = {T_ErrorSaveContext};
PG_RETURN_BOOL(pg_input_is_valid_common(fcinfo, txt, typname,
&escontext));
}
/*
* pg_input_error_message - test whether string is valid input for datatype.
*
* Returns NULL if OK, else the primary message string from the error.
*
* This will only work usefully if the datatype's input function has been
* updated to return "soft" errors via errsave/ereturn.
*/
Datum
pg_input_error_message(PG_FUNCTION_ARGS)
{
text *txt = PG_GETARG_TEXT_PP(0);
text *typname = PG_GETARG_TEXT_PP(1);
ErrorSaveContext escontext = {T_ErrorSaveContext};
/* Enable details_wanted */
escontext.details_wanted = true;
if (pg_input_is_valid_common(fcinfo, txt, typname,
&escontext))
PG_RETURN_NULL();
Assert(escontext.error_occurred);
Assert(escontext.error_data != NULL);
Assert(escontext.error_data->message != NULL);
PG_RETURN_TEXT_P(cstring_to_text(escontext.error_data->message));
}
/* Common subroutine for the above */
static bool
pg_input_is_valid_common(FunctionCallInfo fcinfo,
text *txt, text *typname,
ErrorSaveContext *escontext)
{
char *str = text_to_cstring(txt);
ValidIOData *my_extra;
Datum converted;
/*
* We arrange to look up the needed I/O info just once per series of
* calls, assuming the data type doesn't change underneath us.
*/
my_extra = (ValidIOData *) fcinfo->flinfo->fn_extra;
if (my_extra == NULL)
{
fcinfo->flinfo->fn_extra =
MemoryContextAlloc(fcinfo->flinfo->fn_mcxt,
sizeof(ValidIOData));
my_extra = (ValidIOData *) fcinfo->flinfo->fn_extra;
my_extra->typoid = InvalidOid;
/* Detect whether typname argument is constant. */
my_extra->typname_constant = get_fn_expr_arg_stable(fcinfo->flinfo, 1);
}
/*
* If the typname argument is constant, we only need to parse it the first
* time through.
*/
if (my_extra->typoid == InvalidOid || !my_extra->typname_constant)
{
char *typnamestr = text_to_cstring(typname);
Oid typoid;
/* Parse type-name argument to obtain type OID and encoded typmod. */
parseTypeString(typnamestr, &typoid, &my_extra->typmod, false);
/* Update type-specific info if typoid changed. */
if (my_extra->typoid != typoid)
{
getTypeInputInfo(typoid,
&my_extra->typiofunc,
&my_extra->typioparam);
fmgr_info_cxt(my_extra->typiofunc, &my_extra->inputproc,
fcinfo->flinfo->fn_mcxt);
my_extra->typoid = typoid;
}
}
/* Now we can try to perform the conversion. */
return InputFunctionCallSafe(&my_extra->inputproc,
str,
my_extra->typioparam,
my_extra->typmod,
(Node *) escontext,
&converted);
}
/*
* Is character a valid identifier start?
* Must match scan.l's {ident_start} character class.

View File

@@ -57,6 +57,6 @@
*/
/* yyyymmddN */
#define CATALOG_VERSION_NO 202212091
#define CATALOG_VERSION_NO 202212092
#endif

View File

@@ -7060,6 +7060,14 @@
prorettype => 'regnamespace', proargtypes => 'text',
prosrc => 'to_regnamespace' },
{ oid => '8050', descr => 'test whether string is valid input for data type',
proname => 'pg_input_is_valid', provolatile => 's', prorettype => 'bool',
proargtypes => 'text text', prosrc => 'pg_input_is_valid' },
{ oid => '8051',
descr => 'get error message if string is not valid input for data type',
proname => 'pg_input_error_message', provolatile => 's', prorettype => 'text',
proargtypes => 'text text', prosrc => 'pg_input_error_message' },
{ oid => '1268',
descr => 'parse qualified identifier to array of identifiers',
proname => 'parse_ident', prorettype => '_text', proargtypes => 'text bool',

View File

@@ -249,6 +249,31 @@ select format_type('bpchar'::regtype, -1);
bpchar
(1 row)
-- Test non-error-throwing APIs using widget, which still throws errors
SELECT pg_input_is_valid('(1,2,3)', 'widget');
pg_input_is_valid
-------------------
t
(1 row)
SELECT pg_input_is_valid('(1,2)', 'widget'); -- hard error expected
ERROR: invalid input syntax for type widget: "(1,2)"
SELECT pg_input_is_valid('{"(1,2,3)"}', 'widget[]');
pg_input_is_valid
-------------------
t
(1 row)
SELECT pg_input_is_valid('{"(1,2)"}', 'widget[]'); -- hard error expected
ERROR: invalid input syntax for type widget: "(1,2)"
SELECT pg_input_is_valid('("(1,2,3)")', 'mytab');
pg_input_is_valid
-------------------
t
(1 row)
SELECT pg_input_is_valid('("(1,2)")', 'mytab'); -- hard error expected
ERROR: invalid input syntax for type widget: "(1,2)"
-- Test creation of an operator over a user-defined type
CREATE FUNCTION pt_in_widget(point, widget)
RETURNS bool

View File

@@ -183,6 +183,11 @@ widget_in(PG_FUNCTION_ARGS)
coord[i++] = p + 1;
}
/*
* Note: DON'T convert this error to "soft" style (errsave/ereturn). We
* want this data type to stay permanently in the hard-error world so that
* it can be used for testing that such cases still work reasonably.
*/
if (i < NARGS)
ereport(ERROR,
(errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),

View File

@@ -192,6 +192,14 @@ select format_type('bpchar'::regtype, null);
-- this behavior difference is intentional
select format_type('bpchar'::regtype, -1);
-- Test non-error-throwing APIs using widget, which still throws errors
SELECT pg_input_is_valid('(1,2,3)', 'widget');
SELECT pg_input_is_valid('(1,2)', 'widget'); -- hard error expected
SELECT pg_input_is_valid('{"(1,2,3)"}', 'widget[]');
SELECT pg_input_is_valid('{"(1,2)"}', 'widget[]'); -- hard error expected
SELECT pg_input_is_valid('("(1,2,3)")', 'mytab');
SELECT pg_input_is_valid('("(1,2)")', 'mytab'); -- hard error expected
-- Test creation of an operator over a user-defined type
CREATE FUNCTION pt_in_widget(point, widget)