Commit 001bb9f3 authored by Tom Lane's avatar Tom Lane

Add stack depth checks to key recursive functions in backend/nodes/*.c.

Although copyfuncs.c has a check_stack_depth call in its recursion,
equalfuncs.c, outfuncs.c, and readfuncs.c lacked one.  This seems
unwise.

Likewise fix planstate_tree_walker(), in branches where that exists.

Discussion: https://postgr.es/m/30253.1544286631@sss.pgh.pa.us
parent e28649a6
...@@ -29,6 +29,7 @@ ...@@ -29,6 +29,7 @@
#include "postgres.h" #include "postgres.h"
#include "miscadmin.h"
#include "nodes/extensible.h" #include "nodes/extensible.h"
#include "nodes/relation.h" #include "nodes/relation.h"
#include "utils/datum.h" #include "utils/datum.h"
...@@ -3002,6 +3003,9 @@ equal(const void *a, const void *b) ...@@ -3002,6 +3003,9 @@ equal(const void *a, const void *b)
if (nodeTag(a) != nodeTag(b)) if (nodeTag(a) != nodeTag(b))
return false; return false;
/* Guard against stack overflow due to overly complex expressions */
check_stack_depth();
switch (nodeTag(a)) switch (nodeTag(a))
{ {
/* /*
......
...@@ -3726,6 +3726,9 @@ planstate_tree_walker(PlanState *planstate, ...@@ -3726,6 +3726,9 @@ planstate_tree_walker(PlanState *planstate,
Plan *plan = planstate->plan; Plan *plan = planstate->plan;
ListCell *lc; ListCell *lc;
/* Guard against stack overflow due to overly complex plan trees */
check_stack_depth();
/* initPlan-s */ /* initPlan-s */
if (planstate_walk_subplans(planstate->initPlan, walker, context)) if (planstate_walk_subplans(planstate->initPlan, walker, context))
return true; return true;
......
...@@ -30,6 +30,7 @@ ...@@ -30,6 +30,7 @@
#include <ctype.h> #include <ctype.h>
#include "lib/stringinfo.h" #include "lib/stringinfo.h"
#include "miscadmin.h"
#include "nodes/extensible.h" #include "nodes/extensible.h"
#include "nodes/plannodes.h" #include "nodes/plannodes.h"
#include "nodes/relation.h" #include "nodes/relation.h"
...@@ -3701,6 +3702,9 @@ _outPartitionRangeDatum(StringInfo str, const PartitionRangeDatum *node) ...@@ -3701,6 +3702,9 @@ _outPartitionRangeDatum(StringInfo str, const PartitionRangeDatum *node)
void void
outNode(StringInfo str, const void *obj) outNode(StringInfo str, const void *obj)
{ {
/* Guard against stack overflow due to overly complex expressions */
check_stack_depth();
if (obj == NULL) if (obj == NULL)
appendStringInfoString(str, "<>"); appendStringInfoString(str, "<>");
else if (IsA(obj, List) ||IsA(obj, IntList) || IsA(obj, OidList)) else if (IsA(obj, List) ||IsA(obj, IntList) || IsA(obj, OidList))
......
...@@ -33,6 +33,7 @@ ...@@ -33,6 +33,7 @@
#include <math.h> #include <math.h>
#include "fmgr.h" #include "fmgr.h"
#include "miscadmin.h"
#include "nodes/extensible.h" #include "nodes/extensible.h"
#include "nodes/parsenodes.h" #include "nodes/parsenodes.h"
#include "nodes/plannodes.h" #include "nodes/plannodes.h"
...@@ -2549,6 +2550,9 @@ parseNodeString(void) ...@@ -2549,6 +2550,9 @@ parseNodeString(void)
READ_TEMP_LOCALS(); READ_TEMP_LOCALS();
/* Guard against stack overflow due to overly complex expressions */
check_stack_depth();
token = pg_strtok(&length); token = pg_strtok(&length);
#define MATCH(tokname, namelen) \ #define MATCH(tokname, namelen) \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment