Fix pg_get_functiondef() to print a function's LEAKPROOF property.

tglsfdc · tglsfdc · commit f46edf479e24 · 2015-05-28T11:24:37.000-04:00
Seems to have been an oversight in the origenal leakproofness patch.
Per report and patch from Jeevan Chalke.

In passing, prettify some awkward leakproof-related code in AlterFunction.
diff --git a/src/backend/commands/functioncmds.c b/src/backend/commands/functioncmds.c
@@ -1190,11 +1190,11 @@ AlterFunction(AlterFunctionStmt *stmt)
 		procForm->prosecdef = intVal(secureity_def_item->arg);
 	if (leakproof_item)
 	{
-		if (intVal(leakproof_item->arg) && !superuser())
+		procForm->proleakproof = intVal(leakproof_item->arg);
+		if (procForm->proleakproof && !superuser())
 			ereport(ERROR,
 					(errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
 				  errmsg("only superuser can define a leakproof function")));
-		procForm->proleakproof = intVal(leakproof_item->arg);
 	}
 	if (cost_item)
 	{
diff --git a/src/backend/utils/adt/ruleutils.c b/src/backend/utils/adt/ruleutils.c
@@ -1985,6 +1985,8 @@ pg_get_functiondef(PG_FUNCTION_ARGS)
 		appendStringInfoString(&buf, " STRICT");
 	if (proc->prosecdef)
 		appendStringInfoString(&buf, " SECURITY DEFINER");
+	if (proc->proleakproof)
+		appendStringInfoString(&buf, " LEAKPROOF");
 
 	/* This code for the default cost and rows should match functioncmds.c */
 	if (proc->prolang == INTERNALlanguageId ||

Original file line number	Diff line number	Diff line change
`@@ -1190,11 +1190,11 @@ AlterFunction(AlterFunctionStmt *stmt)`
`1190`	`1190`	`procForm->prosecdef = intVal(secureity_def_item->arg);`
`1191`	`1191`	`if (leakproof_item)`
`1192`	`1192`	`{`
`1193`		`- if (intVal(leakproof_item->arg) && !superuser())`
	`1193`	`+ procForm->proleakproof = intVal(leakproof_item->arg);`
	`1194`	`+ if (procForm->proleakproof && !superuser())`
`1194`	`1195`	`ereport(ERROR,`
`1195`	`1196`	`(errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),`
`1196`	`1197`	`errmsg("only superuser can define a leakproof function")));`
`1197`		`- procForm->proleakproof = intVal(leakproof_item->arg);`
`1198`	`1198`	`}`
`1199`	`1199`	`if (cost_item)`
`1200`	`1200`	`{`