From ab90b49f6bd2ca40cf9bf8077760104a951674dc Mon Sep 17 00:00:00 2001 From: Tom Lane Date: Fri, 16 May 2003 06:07:58 +0000 Subject: [PATCH] Check calling context for connectby_text(), per Joe Conway. --- contrib/tablefunc/tablefunc.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/contrib/tablefunc/tablefunc.c b/contrib/tablefunc/tablefunc.c index d5db3ceed73..4fd3846820b 100644 --- a/contrib/tablefunc/tablefunc.c +++ b/contrib/tablefunc/tablefunc.c @@ -646,6 +646,11 @@ connectby_text(PG_FUNCTION_ARGS) MemoryContext per_query_ctx; MemoryContext oldcontext; + /* check to see if caller supports us returning a tuplestore */ + if (!rsinfo || !(rsinfo->allowedModes & SFRM_Materialize)) + elog(ERROR, "connectby: materialize mode required, but it is not " + "allowed in this context"); + if (fcinfo->nargs == 6) { branch_delim = GET_STR(PG_GETARG_TEXT_P(5));