diff options
Diffstat (limited to 'src/backend')
-rw-r--r-- | src/backend/catalog/pg_proc.c | 215 | ||||
-rw-r--r-- | src/backend/commands/portalcmds.c | 6 | ||||
-rw-r--r-- | src/backend/executor/functions.c | 41 | ||||
-rw-r--r-- | src/backend/executor/spi.c | 167 | ||||
-rw-r--r-- | src/backend/optimizer/util/clauses.c | 26 | ||||
-rw-r--r-- | src/backend/parser/parse_type.c | 8 | ||||
-rw-r--r-- | src/backend/tcop/postgres.c | 3 | ||||
-rw-r--r-- | src/backend/tcop/pquery.c | 17 | ||||
-rw-r--r-- | src/backend/utils/error/elog.c | 112 |
9 files changed, 517 insertions, 78 deletions
diff --git a/src/backend/catalog/pg_proc.c b/src/backend/catalog/pg_proc.c index 3ea2af44d3e..6fe64eadd0d 100644 --- a/src/backend/catalog/pg_proc.c +++ b/src/backend/catalog/pg_proc.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/catalog/pg_proc.c,v 1.112 2004/03/14 01:58:41 tgl Exp $ + * $PostgreSQL: pgsql/src/backend/catalog/pg_proc.c,v 1.113 2004/03/21 22:29:10 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -23,9 +23,11 @@ #include "executor/executor.h" #include "fmgr.h" #include "miscadmin.h" +#include "mb/pg_wchar.h" #include "parser/parse_coerce.h" #include "parser/parse_expr.h" #include "parser/parse_type.h" +#include "tcop/pquery.h" #include "tcop/tcopprot.h" #include "utils/acl.h" #include "utils/builtins.h" @@ -45,6 +47,10 @@ Datum fmgr_sql_validator(PG_FUNCTION_ARGS); static Datum create_parameternames_array(int parameterCount, const char *parameterNames[]); static void sql_function_parse_error_callback(void *arg); +static int match_prosrc_to_query(const char *prosrc, const char *queryText, + int cursorpos); +static bool match_prosrc_to_literal(const char *prosrc, const char *literal, + int cursorpos, int *newcursorpos); /* ---------------------------------------------------------------- @@ -763,12 +769,10 @@ fmgr_sql_validator(PG_FUNCTION_ARGS) prosrc = DatumGetCString(DirectFunctionCall1(textout, tmp)); /* - * Setup error traceback support for ereport(). This is mostly - * so we can add context info that shows that a syntax-error - * location is inside the function body, not out in CREATE FUNCTION. + * Setup error traceback support for ereport(). */ sqlerrcontext.callback = sql_function_parse_error_callback; - sqlerrcontext.arg = proc; + sqlerrcontext.arg = tuple; sqlerrcontext.previous = error_context_stack; error_context_stack = &sqlerrcontext; @@ -800,22 +804,203 @@ fmgr_sql_validator(PG_FUNCTION_ARGS) } /* - * error context callback to let us supply a context marker + * Error context callback for handling errors in SQL function definitions */ static void sql_function_parse_error_callback(void *arg) { - Form_pg_proc proc = (Form_pg_proc) arg; + HeapTuple tuple = (HeapTuple) arg; + Form_pg_proc proc = (Form_pg_proc) GETSTRUCT(tuple); + bool isnull; + Datum tmp; + char *prosrc; + + /* See if it's a syntax error; if so, transpose to CREATE FUNCTION */ + tmp = SysCacheGetAttr(PROCOID, tuple, Anum_pg_proc_prosrc, &isnull); + if (isnull) + elog(ERROR, "null prosrc"); + prosrc = DatumGetCString(DirectFunctionCall1(textout, tmp)); + + if (!function_parse_error_transpose(prosrc)) + { + /* If it's not a syntax error, push info onto context stack */ + errcontext("SQL function \"%s\"", NameStr(proc->proname)); + } + + pfree(prosrc); +} + +/* + * Adjust a syntax error occurring inside the function body of a CREATE + * FUNCTION command. This can be used by any function validator, not only + * for SQL-language functions. It is assumed that the syntax error position + * is initially relative to the function body string (as passed in). If + * possible, we adjust the position to reference the original CREATE command; + * if we can't manage that, we set up an "internal query" syntax error instead. + * + * Returns true if a syntax error was processed, false if not. + */ +bool +function_parse_error_transpose(const char *prosrc) +{ + int origerrposition; + int newerrposition; + const char *queryText; + + /* + * Nothing to do unless we are dealing with a syntax error that has + * a cursor position. + * + * Some PLs may prefer to report the error position as an internal + * error to begin with, so check that too. + */ + origerrposition = geterrposition(); + if (origerrposition <= 0) + { + origerrposition = getinternalerrposition(); + if (origerrposition <= 0) + return false; + } + + /* We can get the original query text from the active portal (hack...) */ + Assert(ActivePortal && ActivePortal->portalActive); + queryText = ActivePortal->sourceText; + + /* Try to locate the prosrc in the original text */ + newerrposition = match_prosrc_to_query(prosrc, queryText, origerrposition); + + if (newerrposition > 0) + { + /* Successful, so fix error position to reference original query */ + errposition(newerrposition); + /* Get rid of any report of the error as an "internal query" */ + internalerrposition(0); + internalerrquery(NULL); + } + else + { + /* + * If unsuccessful, convert the position to an internal position + * marker and give the function text as the internal query. + */ + errposition(0); + internalerrposition(origerrposition); + internalerrquery(prosrc); + } + + return true; +} +/* + * Try to locate the string literal containing the function body in the + * given text of the CREATE FUNCTION command. If successful, return the + * character (not byte) index within the command corresponding to the + * given character index within the literal. If not successful, return 0. + */ +static int +match_prosrc_to_query(const char *prosrc, const char *queryText, + int cursorpos) +{ /* - * XXX it'd be really nice to adjust the syntax error position to - * account for the offset from the start of the statement to the - * function body string, not to mention any quoting characters in - * the string, but I can't see any decent way to do that... + * Rather than fully parsing the CREATE FUNCTION command, we just scan + * the command looking for $prosrc$ or 'prosrc'. This could be fooled + * (though not in any very probable scenarios), so fail if we find + * more than one match. + */ + int prosrclen = strlen(prosrc); + int querylen = strlen(queryText); + int matchpos = 0; + int curpos; + int newcursorpos; + + for (curpos = 0; curpos < querylen-prosrclen; curpos++) + { + if (queryText[curpos] == '$' && + strncmp(prosrc, &queryText[curpos+1], prosrclen) == 0 && + queryText[curpos+1+prosrclen] == '$') + { + /* + * Found a $foo$ match. Since there are no embedded quoting + * characters in a dollar-quoted literal, we don't have to do + * any fancy arithmetic; just offset by the starting position. + */ + if (matchpos) + return 0; /* multiple matches, fail */ + matchpos = pg_mbstrlen_with_len(queryText, curpos+1) + + cursorpos; + } + else if (queryText[curpos] == '\'' && + match_prosrc_to_literal(prosrc, &queryText[curpos+1], + cursorpos, &newcursorpos)) + { + /* + * Found a 'foo' match. match_prosrc_to_literal() has adjusted + * for any quotes or backslashes embedded in the literal. + */ + if (matchpos) + return 0; /* multiple matches, fail */ + matchpos = pg_mbstrlen_with_len(queryText, curpos+1) + + newcursorpos; + } + } + + return matchpos; +} + +/* + * Try to match the given source text to a single-quoted literal. + * If successful, adjust newcursorpos to correspond to the character + * (not byte) index corresponding to cursorpos in the source text. + * + * At entry, literal points just past a ' character. We must check for the + * trailing quote. + */ +static bool +match_prosrc_to_literal(const char *prosrc, const char *literal, + int cursorpos, int *newcursorpos) +{ + int newcp = cursorpos; + int chlen; + + /* + * This implementation handles backslashes and doubled quotes in the + * string literal. It does not handle the SQL syntax for literals + * continued across line boundaries. * - * In the meantime, put in a CONTEXT entry that can cue clients - * not to trust the syntax error position completely. + * We do the comparison a character at a time, not a byte at a time, + * so that we can do the correct cursorpos math. */ - errcontext("SQL function \"%s\"", - NameStr(proc->proname)); + while (*prosrc) + { + cursorpos--; /* characters left before cursor */ + /* + * Check for backslashes and doubled quotes in the literal; adjust + * newcp when one is found before the cursor. + */ + if (*literal == '\\') + { + literal++; + if (cursorpos > 0) + newcp++; + } + else if (*literal == '\'') + { + if (literal[1] != '\'') + return false; + literal++; + if (cursorpos > 0) + newcp++; + } + chlen = pg_mblen(prosrc); + if (strncmp(prosrc, literal, chlen) != 0) + return false; + prosrc += chlen; + literal += chlen; + } + + *newcursorpos = newcp; + + if (*literal == '\'' && literal[1] != '\'') + return true; + return false; } diff --git a/src/backend/commands/portalcmds.c b/src/backend/commands/portalcmds.c index 34748fa77e1..855c9391c18 100644 --- a/src/backend/commands/portalcmds.c +++ b/src/backend/commands/portalcmds.c @@ -14,7 +14,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/commands/portalcmds.c,v 1.25 2003/11/29 19:51:47 pgsql Exp $ + * $PostgreSQL: pgsql/src/backend/commands/portalcmds.c,v 1.26 2004/03/21 22:29:10 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -270,6 +270,7 @@ void PersistHoldablePortal(Portal portal) { QueryDesc *queryDesc = PortalGetQueryDesc(portal); + Portal saveActivePortal; MemoryContext savePortalContext; MemoryContext saveQueryContext; MemoryContext oldcxt; @@ -311,6 +312,8 @@ PersistHoldablePortal(Portal portal) /* * Set global portal context pointers. */ + saveActivePortal = ActivePortal; + ActivePortal = portal; savePortalContext = PortalContext; PortalContext = PortalGetHeapMemory(portal); saveQueryContext = QueryContext; @@ -342,6 +345,7 @@ PersistHoldablePortal(Portal portal) /* Mark portal not active */ portal->portalActive = false; + ActivePortal = saveActivePortal; PortalContext = savePortalContext; QueryContext = saveQueryContext; diff --git a/src/backend/executor/functions.c b/src/backend/executor/functions.c index 6ffc8875f19..8dec6131fb4 100644 --- a/src/backend/executor/functions.c +++ b/src/backend/executor/functions.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/executor/functions.c,v 1.77 2004/01/07 18:56:26 neilc Exp $ + * $PostgreSQL: pgsql/src/backend/executor/functions.c,v 1.78 2004/03/21 22:29:11 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -645,12 +645,40 @@ sql_exec_error_callback(void *arg) { FmgrInfo *flinfo = (FmgrInfo *) arg; SQLFunctionCachePtr fcache = (SQLFunctionCachePtr) flinfo->fn_extra; + HeapTuple func_tuple; + Form_pg_proc functup; char *fn_name; + int syntaxerrposition; - fn_name = get_func_name(flinfo->fn_oid); - /* safety check, shouldn't happen */ - if (fn_name == NULL) - return; + /* Need access to function's pg_proc tuple */ + func_tuple = SearchSysCache(PROCOID, + ObjectIdGetDatum(flinfo->fn_oid), + 0, 0, 0); + if (!HeapTupleIsValid(func_tuple)) + return; /* shouldn't happen */ + functup = (Form_pg_proc) GETSTRUCT(func_tuple); + fn_name = NameStr(functup->proname); + + /* + * If there is a syntax error position, convert to internal syntax error + */ + syntaxerrposition = geterrposition(); + if (syntaxerrposition > 0) + { + bool isnull; + Datum tmp; + char *prosrc; + + tmp = SysCacheGetAttr(PROCOID, func_tuple, Anum_pg_proc_prosrc, + &isnull); + if (isnull) + elog(ERROR, "null prosrc"); + prosrc = DatumGetCString(DirectFunctionCall1(textout, tmp)); + errposition(0); + internalerrposition(syntaxerrposition); + internalerrquery(prosrc); + pfree(prosrc); + } /* * Try to determine where in the function we failed. If there is a @@ -692,8 +720,7 @@ sql_exec_error_callback(void *arg) errcontext("SQL function \"%s\" during startup", fn_name); } - /* free result of get_func_name (in case this is only a notice) */ - pfree(fn_name); + ReleaseSysCache(func_tuple); } diff --git a/src/backend/executor/spi.c b/src/backend/executor/spi.c index fb44d9d56fd..1fb60fff02d 100644 --- a/src/backend/executor/spi.c +++ b/src/backend/executor/spi.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/executor/spi.c,v 1.111 2004/03/17 01:05:10 momjian Exp $ + * $PostgreSQL: pgsql/src/backend/executor/spi.c,v 1.112 2004/03/21 22:29:11 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -39,6 +39,8 @@ static int _SPI_execute_plan(_SPI_plan *plan, Datum *Values, const char *Nulls, bool useCurrentSnapshot, int tcount); +static void _SPI_error_callback(void *arg); + static void _SPI_cursor_operation(Portal portal, bool forward, int count, DestReceiver *dest); @@ -286,7 +288,8 @@ SPI_execp_current(void *plan, Datum *Values, const char *Nulls, void * SPI_prepare(const char *src, int nargs, Oid *argtypes) { - _SPI_plan *plan; + _SPI_plan plan; + _SPI_plan *result; if (src == NULL || nargs < 0 || (nargs > 0 && argtypes == NULL)) { @@ -298,20 +301,21 @@ SPI_prepare(const char *src, int nargs, Oid *argtypes) if (SPI_result < 0) return NULL; - plan = (_SPI_plan *) palloc(sizeof(_SPI_plan)); /* Executor context */ - plan->argtypes = argtypes; - plan->nargs = nargs; + plan.plancxt = NULL; /* doesn't have own context */ + plan.query = src; + plan.nargs = nargs; + plan.argtypes = argtypes; - SPI_result = _SPI_execute(src, 0, plan); + SPI_result = _SPI_execute(src, 0, &plan); if (SPI_result >= 0) /* copy plan to procedure context */ - plan = _SPI_copy_plan(plan, _SPI_CPLAN_PROCXT); + result = _SPI_copy_plan(&plan, _SPI_CPLAN_PROCXT); else - plan = NULL; + result = NULL; _SPI_end_call(true); - return (void *) plan; + return (void *) result; } void * @@ -335,7 +339,6 @@ SPI_saveplan(void *plan) SPI_result = 0; return (void *) newplan; - } int @@ -927,12 +930,12 @@ SPI_cursor_close(Portal portal) Oid SPI_getargtypeid(void *plan, int argIndex) { - if (plan == NULL || argIndex < 0 || argIndex >= ((_SPI_plan*)plan)->nargs) - { - SPI_result = SPI_ERROR_ARGUMENT; - return InvalidOid; - } - return ((_SPI_plan *) plan)->argtypes[argIndex]; + if (plan == NULL || argIndex < 0 || argIndex >= ((_SPI_plan*)plan)->nargs) + { + SPI_result = SPI_ERROR_ARGUMENT; + return InvalidOid; + } + return ((_SPI_plan *) plan)->argtypes[argIndex]; } /* @@ -941,12 +944,12 @@ SPI_getargtypeid(void *plan, int argIndex) int SPI_getargcount(void *plan) { - if (plan == NULL) - { - SPI_result = SPI_ERROR_ARGUMENT; - return -1; - } - return ((_SPI_plan *) plan)->nargs; + if (plan == NULL) + { + SPI_result = SPI_ERROR_ARGUMENT; + return -1; + } + return ((_SPI_plan *) plan)->nargs; } /* @@ -961,22 +964,24 @@ SPI_getargcount(void *plan) bool SPI_is_cursor_plan(void *plan) { - List *qtlist; - _SPI_plan *spiplan = (_SPI_plan *) plan; - if (spiplan == NULL) - { - SPI_result = SPI_ERROR_ARGUMENT; - return false; - } - - qtlist = spiplan->qtlist; - if(length(spiplan->ptlist) == 1 && length(qtlist) == 1) - { - Query *queryTree = (Query *) lfirst((List *) lfirst(qtlist)); - if(queryTree->commandType == CMD_SELECT && queryTree->into == NULL) - return true; - } - return false; + _SPI_plan *spiplan = (_SPI_plan *) plan; + List *qtlist; + + if (spiplan == NULL) + { + SPI_result = SPI_ERROR_ARGUMENT; + return false; + } + + qtlist = spiplan->qtlist; + if (length(spiplan->ptlist) == 1 && length(qtlist) == 1) + { + Query *queryTree = (Query *) lfirst((List *) lfirst(qtlist)); + + if (queryTree->commandType == CMD_SELECT && queryTree->into == NULL) + return true; + } + return false; } /* =================== private functions =================== */ @@ -1071,7 +1076,8 @@ spi_printtup(HeapTuple tuple, TupleDesc tupdesc, DestReceiver *self) /* * Plan and optionally execute a querystring. * - * If plan != NULL, just prepare plan tree, else execute immediately. + * If plan != NULL, just prepare plan trees and save them in *plan; + * else execute immediately. */ static int _SPI_execute(const char *src, int tcount, _SPI_plan *plan) @@ -1080,6 +1086,7 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) List *query_list_list; List *plan_list; List *list_item; + ErrorContextCallback spierrcontext; int nargs = 0; Oid *argtypes = NULL; int res = 0; @@ -1100,6 +1107,14 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) _SPI_current->tuptable = NULL; /* + * Setup error traceback support for ereport() + */ + spierrcontext.callback = _SPI_error_callback; + spierrcontext.arg = (void *) src; + spierrcontext.previous = error_context_stack; + error_context_stack = &spierrcontext; + + /* * Parse the request string into a list of raw parse trees. */ raw_parsetree_list = pg_parse_query(src); @@ -1149,14 +1164,23 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) CopyStmt *stmt = (CopyStmt *) queryTree->utilityStmt; if (stmt->filename == NULL) - return SPI_ERROR_COPY; + { + res = SPI_ERROR_COPY; + goto fail; + } } else if (IsA(queryTree->utilityStmt, DeclareCursorStmt) || IsA(queryTree->utilityStmt, ClosePortalStmt) || IsA(queryTree->utilityStmt, FetchStmt)) - return SPI_ERROR_CURSOR; + { + res = SPI_ERROR_CURSOR; + goto fail; + } else if (IsA(queryTree->utilityStmt, TransactionStmt)) - return SPI_ERROR_TRANSACTION; + { + res = SPI_ERROR_TRANSACTION; + goto fail; + } res = SPI_OK_UTILITY; if (plan == NULL) { @@ -1171,7 +1195,7 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) res = _SPI_pquery(qdesc, true, false, queryTree->canSetTag ? tcount : 0); if (res < 0) - return res; + goto fail; CommandCounterIncrement(); } else @@ -1180,7 +1204,7 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) NULL, false); res = _SPI_pquery(qdesc, false, false, 0); if (res < 0) - return res; + goto fail; } } } @@ -1191,6 +1215,13 @@ _SPI_execute(const char *src, int tcount, _SPI_plan *plan) plan->ptlist = plan_list; } +fail: + + /* + * Pop the error context stack + */ + error_context_stack = spierrcontext.previous; + return res; } @@ -1201,6 +1232,7 @@ _SPI_execute_plan(_SPI_plan *plan, Datum *Values, const char *Nulls, List *query_list_list = plan->qtlist; List *plan_list = plan->ptlist; List *query_list_list_item; + ErrorContextCallback spierrcontext; int nargs = plan->nargs; int res = 0; ParamListInfo paramLI; @@ -1234,6 +1266,14 @@ _SPI_execute_plan(_SPI_plan *plan, Datum *Values, const char *Nulls, SPI_tuptable = NULL; _SPI_current->tuptable = NULL; + /* + * Setup error traceback support for ereport() + */ + spierrcontext.callback = _SPI_error_callback; + spierrcontext.arg = (void *) plan->query; + spierrcontext.previous = error_context_stack; + error_context_stack = &spierrcontext; + foreach(query_list_list_item, query_list_list) { List *query_list = lfirst(query_list_list_item); @@ -1270,12 +1310,19 @@ _SPI_execute_plan(_SPI_plan *plan, Datum *Values, const char *Nulls, res = _SPI_pquery(qdesc, true, useCurrentSnapshot, queryTree->canSetTag ? tcount : 0); if (res < 0) - return res; + goto fail; CommandCounterIncrement(); } } } +fail: + + /* + * Pop the error context stack + */ + error_context_stack = spierrcontext.previous; + return res; } @@ -1356,6 +1403,32 @@ _SPI_pquery(QueryDesc *queryDesc, bool runit, } /* + * _SPI_error_callback + * + * Add context information when a query invoked via SPI fails + */ +static void +_SPI_error_callback(void *arg) +{ + const char *query = (const char *) arg; + int syntaxerrposition; + + /* + * If there is a syntax error position, convert to internal syntax error; + * otherwise treat the query as an item of context stack + */ + syntaxerrposition = geterrposition(); + if (syntaxerrposition > 0) + { + errposition(0); + internalerrposition(syntaxerrposition); + internalerrquery(query); + } + else + errcontext("SQL query \"%s\"", query); +} + +/* * _SPI_cursor_operation() * * Do a FETCH or MOVE in a cursor @@ -1490,8 +1563,7 @@ _SPI_copy_plan(_SPI_plan *plan, int location) parentcxt = _SPI_current->procCxt; else if (location == _SPI_CPLAN_TOPCXT) parentcxt = TopMemoryContext; - else -/* (this case not currently used) */ + else /* (this case not currently used) */ parentcxt = CurrentMemoryContext; /* @@ -1508,6 +1580,7 @@ _SPI_copy_plan(_SPI_plan *plan, int location) /* Copy the SPI plan into its own context */ newplan = (_SPI_plan *) palloc(sizeof(_SPI_plan)); newplan->plancxt = plancxt; + newplan->query = pstrdup(plan->query); newplan->qtlist = (List *) copyObject(plan->qtlist); newplan->ptlist = (List *) copyObject(plan->ptlist); newplan->nargs = plan->nargs; diff --git a/src/backend/optimizer/util/clauses.c b/src/backend/optimizer/util/clauses.c index 1487aec453d..c006cd49a1f 100644 --- a/src/backend/optimizer/util/clauses.c +++ b/src/backend/optimizer/util/clauses.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/optimizer/util/clauses.c,v 1.165 2004/03/17 20:48:42 tgl Exp $ + * $PostgreSQL: pgsql/src/backend/optimizer/util/clauses.c,v 1.166 2004/03/21 22:29:11 tgl Exp $ * * HISTORY * AUTHOR DATE MAJOR EVENT @@ -1917,7 +1917,7 @@ inline_function(Oid funcid, Oid result_type, List *args, * can finger the function that bad information came from. */ sqlerrcontext.callback = sql_inline_error_callback; - sqlerrcontext.arg = funcform; + sqlerrcontext.arg = func_tuple; sqlerrcontext.previous = error_context_stack; error_context_stack = &sqlerrcontext; @@ -2146,7 +2146,27 @@ substitute_actual_parameters_mutator(Node *node, static void sql_inline_error_callback(void *arg) { - Form_pg_proc funcform = (Form_pg_proc) arg; + HeapTuple func_tuple = (HeapTuple) arg; + Form_pg_proc funcform = (Form_pg_proc) GETSTRUCT(func_tuple); + int syntaxerrposition; + + /* If it's a syntax error, convert to internal syntax error report */ + syntaxerrposition = geterrposition(); + if (syntaxerrposition > 0) + { + bool isnull; + Datum tmp; + char *prosrc; + + tmp = SysCacheGetAttr(PROCOID, func_tuple, Anum_pg_proc_prosrc, + &isnull); + if (isnull) + elog(ERROR, "null prosrc"); + prosrc = DatumGetCString(DirectFunctionCall1(textout, tmp)); + errposition(0); + internalerrposition(syntaxerrposition); + internalerrquery(prosrc); + } errcontext("SQL function \"%s\" during inlining", NameStr(funcform->proname)); diff --git a/src/backend/parser/parse_type.c b/src/backend/parser/parse_type.c index 7695bc49e51..e6e8b00cda4 100644 --- a/src/backend/parser/parse_type.c +++ b/src/backend/parser/parse_type.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/parser/parse_type.c,v 1.64 2003/11/29 19:51:52 pgsql Exp $ + * $PostgreSQL: pgsql/src/backend/parser/parse_type.c,v 1.65 2004/03/21 22:29:11 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -439,6 +439,12 @@ pts_error_callback(void *arg) const char *str = (const char *) arg; errcontext("invalid type name \"%s\"", str); + /* + * Currently we just suppress any syntax error position report, + * rather than transforming to an "internal query" error. It's + * unlikely that a type name is complex enough to need positioning. + */ + errposition(0); } /* diff --git a/src/backend/tcop/postgres.c b/src/backend/tcop/postgres.c index 659eda80c85..91442d49e6e 100644 --- a/src/backend/tcop/postgres.c +++ b/src/backend/tcop/postgres.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/tcop/postgres.c,v 1.395 2004/03/15 15:56:22 momjian Exp $ + * $PostgreSQL: pgsql/src/backend/tcop/postgres.c,v 1.396 2004/03/21 22:29:11 tgl Exp $ * * NOTES * this is the "main" module of the postgres backend and @@ -2710,6 +2710,7 @@ PostgresMain(int argc, char *argv[], const char *username) */ MemoryContextSwitchTo(TopMemoryContext); MemoryContextResetAndDeleteChildren(ErrorContext); + ActivePortal = NULL; PortalContext = NULL; QueryContext = NULL; diff --git a/src/backend/tcop/pquery.c b/src/backend/tcop/pquery.c index 132afd3c6eb..c213182ad1f 100644 --- a/src/backend/tcop/pquery.c +++ b/src/backend/tcop/pquery.c @@ -8,7 +8,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/tcop/pquery.c,v 1.76 2004/03/18 23:26:17 momjian Exp $ + * $PostgreSQL: pgsql/src/backend/tcop/pquery.c,v 1.77 2004/03/21 22:29:11 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -23,6 +23,13 @@ #include "utils/memutils.h" +/* + * ActivePortal is the currently executing Portal (the most closely nested, + * if there are several). + */ +Portal ActivePortal = NULL; + + static uint32 RunFromStore(Portal portal, ScanDirection direction, long count, DestReceiver *dest); static long PortalRunSelect(Portal portal, bool forward, long count, @@ -395,6 +402,7 @@ PortalRun(Portal portal, long count, char *completionTag) { bool result; + Portal saveActivePortal; MemoryContext savePortalContext; MemoryContext saveQueryContext; MemoryContext oldContext; @@ -430,6 +438,8 @@ PortalRun(Portal portal, long count, /* * Set global portal context pointers. */ + saveActivePortal = ActivePortal; + ActivePortal = portal; savePortalContext = PortalContext; PortalContext = PortalGetHeapMemory(portal); saveQueryContext = QueryContext; @@ -505,6 +515,7 @@ PortalRun(Portal portal, long count, /* Mark portal not active */ portal->portalActive = false; + ActivePortal = saveActivePortal; PortalContext = savePortalContext; QueryContext = saveQueryContext; @@ -922,6 +933,7 @@ PortalRunFetch(Portal portal, DestReceiver *dest) { long result; + Portal saveActivePortal; MemoryContext savePortalContext; MemoryContext saveQueryContext; MemoryContext oldContext; @@ -945,6 +957,8 @@ PortalRunFetch(Portal portal, /* * Set global portal context pointers. */ + saveActivePortal = ActivePortal; + ActivePortal = portal; savePortalContext = PortalContext; PortalContext = PortalGetHeapMemory(portal); saveQueryContext = QueryContext; @@ -969,6 +983,7 @@ PortalRunFetch(Portal portal, /* Mark portal not active */ portal->portalActive = false; + ActivePortal = saveActivePortal; PortalContext = savePortalContext; QueryContext = saveQueryContext; diff --git a/src/backend/utils/error/elog.c b/src/backend/utils/error/elog.c index e826e6d99ee..8569b6ce5cf 100644 --- a/src/backend/utils/error/elog.c +++ b/src/backend/utils/error/elog.c @@ -37,7 +37,7 @@ * * * IDENTIFICATION - * $PostgreSQL: pgsql/src/backend/utils/error/elog.c,v 1.129 2004/03/19 02:23:59 tgl Exp $ + * $PostgreSQL: pgsql/src/backend/utils/error/elog.c,v 1.130 2004/03/21 22:29:11 tgl Exp $ * *------------------------------------------------------------------------- */ @@ -112,6 +112,8 @@ typedef struct ErrorData char *hint; /* hint message */ char *context; /* context message */ int cursorpos; /* cursor index into query string */ + int internalpos; /* cursor index into internalquery */ + char *internalquery; /* text of internally-generated query */ int saved_errno; /* errno at entry */ } ErrorData; @@ -364,6 +366,8 @@ errfinish(int dummy,...) pfree(edata->hint); if (edata->context) pfree(edata->context); + if (edata->internalquery) + pfree(edata->internalquery); MemoryContextSwitchTo(oldcontext); @@ -809,6 +813,83 @@ errposition(int cursorpos) return 0; /* return value does not matter */ } +/* + * internalerrposition --- add internal cursor position to the current error + */ +int +internalerrposition(int cursorpos) +{ + ErrorData *edata = &errordata[errordata_stack_depth]; + + /* we don't bother incrementing recursion_depth */ + CHECK_STACK_DEPTH(); + + edata->internalpos = cursorpos; + + return 0; /* return value does not matter */ +} + +/* + * internalerrquery --- add internal query text to the current error + * + * Can also pass NULL to drop the internal query text entry. This case + * is intended for use in error callback subroutines that are editorializing + * on the layout of the error report. + */ +int +internalerrquery(const char *query) +{ + ErrorData *edata = &errordata[errordata_stack_depth]; + + /* we don't bother incrementing recursion_depth */ + CHECK_STACK_DEPTH(); + + if (edata->internalquery) + { + pfree(edata->internalquery); + edata->internalquery = NULL; + } + + if (query) + edata->internalquery = MemoryContextStrdup(ErrorContext, query); + + return 0; /* return value does not matter */ +} + +/* + * geterrposition --- return the currently set error position (0 if none) + * + * This is only intended for use in error callback subroutines, since there + * is no other place outside elog.c where the concept is meaningful. + */ +int +geterrposition(void) +{ + ErrorData *edata = &errordata[errordata_stack_depth]; + + /* we don't bother incrementing recursion_depth */ + CHECK_STACK_DEPTH(); + + return edata->cursorpos; +} + +/* + * getinternalerrposition --- same for internal error position + * + * This is only intended for use in error callback subroutines, since there + * is no other place outside elog.c where the concept is meaningful. + */ +int +getinternalerrposition(void) +{ + ErrorData *edata = &errordata[errordata_stack_depth]; + + /* we don't bother incrementing recursion_depth */ + CHECK_STACK_DEPTH(); + + return edata->internalpos; +} + /* * elog_finish --- finish up for old-style API @@ -1192,7 +1273,11 @@ send_message_to_server_log(ErrorData *edata) append_with_tabs(&buf, gettext("missing error text")); if (edata->cursorpos > 0) - appendStringInfo(&buf, gettext(" at character %d"), edata->cursorpos); + appendStringInfo(&buf, gettext(" at character %d"), + edata->cursorpos); + else if (edata->internalpos > 0) + appendStringInfo(&buf, gettext(" at character %d"), + edata->internalpos); appendStringInfoChar(&buf, '\n'); @@ -1212,6 +1297,13 @@ send_message_to_server_log(ErrorData *edata) append_with_tabs(&buf, edata->hint); appendStringInfoChar(&buf, '\n'); } + if (edata->internalquery) + { + log_line_prefix(&buf); + appendStringInfoString(&buf, gettext("QUERY: ")); + append_with_tabs(&buf, edata->internalquery); + appendStringInfoChar(&buf, '\n'); + } if (edata->context) { log_line_prefix(&buf); @@ -1365,6 +1457,19 @@ send_message_to_frontend(ErrorData *edata) pq_sendstring(&msgbuf, tbuf); } + if (edata->internalpos > 0) + { + snprintf(tbuf, sizeof(tbuf), "%d", edata->internalpos); + pq_sendbyte(&msgbuf, PG_DIAG_INTERNAL_POSITION); + pq_sendstring(&msgbuf, tbuf); + } + + if (edata->internalquery) + { + pq_sendbyte(&msgbuf, PG_DIAG_INTERNAL_QUERY); + pq_sendstring(&msgbuf, edata->internalquery); + } + if (edata->filename) { pq_sendbyte(&msgbuf, PG_DIAG_SOURCE_FILE); @@ -1406,6 +1511,9 @@ send_message_to_frontend(ErrorData *edata) if (edata->cursorpos > 0) appendStringInfo(&buf, gettext(" at character %d"), edata->cursorpos); + else if (edata->internalpos > 0) + appendStringInfo(&buf, gettext(" at character %d"), + edata->internalpos); appendStringInfoChar(&buf, '\n'); |