Check calling context for connectby_text(), per Joe Conway.

tglsfdc · tglsfdc · commit bcfeca5856a9 · 2003-05-16T06:07:51.000Z
diff --git a/contrib/tablefunc/tablefunc.c b/contrib/tablefunc/tablefunc.c
@@ -1048,6 +1048,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));
