Add a hook in ExecCheckRTPerms().

robertmhaas · robertmhaas · commit f4122a8d50a5 · 2010-07-09T14:06:01.000Z
This hook allows a loadable module to gain control when table permissions are checked. It is expected to be used by an eventual SE-PostgreSQL implementation, but there are other possible applications as well. A sample contrib module can be found in the archives at: http://archives.postgresql.org/pgsql-hackers/2010-05/msg01095.php Robert Haas and Stephen Frost
diff --git a/src/backend/executor/execMain.c b/src/backend/executor/execMain.c
@@ -26,7 +26,7 @@
  *
  *
  * IDENTIFICATION
- *	  $PostgreSQL: pgsql/src/backend/executor/execMain.c,v 1.349 2010/04/28 16:10:42 heikki Exp $
+ *	  $PostgreSQL: pgsql/src/backend/executor/execMain.c,v 1.350 2010/07/09 14:06:01 rhaas Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -63,6 +63,9 @@ ExecutorStart_hook_type ExecutorStart_hook = NULL;
 ExecutorRun_hook_type ExecutorRun_hook = NULL;
 ExecutorEnd_hook_type ExecutorEnd_hook = NULL;
 
+/* Hook for plugin to get control in ExecCheckRTPerms() */
+ExecutorCheckPerms_hook_type ExecutorCheckPerms_hook = NULL;
+
 /* decls for local routines only used within this module */
 static void InitPlan(QueryDesc *queryDesc, int eflags);
 static void ExecEndPlan(PlanState *planstate, EState *estate);
@@ -416,6 +419,9 @@ ExecCheckRTPerms(List *rangeTable)
 	{
 		ExecCheckRTEPerms((RangeTblEntry *) lfirst(l));
 	}
+
+	if (ExecutorCheckPerms_hook)
+		(*ExecutorCheckPerms_hook)(rangeTable);
 }
 
 /*
diff --git a/src/include/executor/executor.h b/src/include/executor/executor.h
@@ -7,7 +7,7 @@
  * Portions Copyright (c) 1996-2010, PostgreSQL Global Development Group
  * Portions Copyright (c) 1994, Regents of the University of California
  *
- * $PostgreSQL: pgsql/src/include/executor/executor.h,v 1.168 2010/02/26 02:01:24 momjian Exp $
+ * $PostgreSQL: pgsql/src/include/executor/executor.h,v 1.169 2010/07/09 14:06:01 rhaas Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -74,6 +74,10 @@ extern PGDLLIMPORT ExecutorRun_hook_type ExecutorRun_hook;
 typedef void (*ExecutorEnd_hook_type) (QueryDesc *queryDesc);
 extern PGDLLIMPORT ExecutorEnd_hook_type ExecutorEnd_hook;
 
+/* Hook for plugins to get control in ExecCheckRTPerms() */
+typedef void (*ExecutorCheckPerms_hook_type) (List *);
+extern PGDLLIMPORT ExecutorCheckPerms_hook_type ExecutorCheckPerms_hook;
+
 
 /*
  * prototypes from functions in execAmi.c

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`*`
`27`	`27`	`*`
`28`	`28`	`* IDENTIFICATION`
`29`		`- * $PostgreSQL: pgsql/src/backend/executor/execMain.c,v 1.349 2010/04/28 16:10:42 heikki Exp $`
	`29`	`+ * $PostgreSQL: pgsql/src/backend/executor/execMain.c,v 1.350 2010/07/09 14:06:01 rhaas Exp $`
`30`	`30`	`*`
`31`	`31`	`*-------------------------------------------------------------------------`
`32`	`32`	`*/`
`@@ -63,6 +63,9 @@ ExecutorStart_hook_type ExecutorStart_hook = NULL;`
`63`	`63`	`ExecutorRun_hook_type ExecutorRun_hook = NULL;`
`64`	`64`	`ExecutorEnd_hook_type ExecutorEnd_hook = NULL;`
`65`	`65`
	`66`	`+/* Hook for plugin to get control in ExecCheckRTPerms() */`
	`67`	`+ExecutorCheckPerms_hook_type ExecutorCheckPerms_hook = NULL;`
	`68`	`+`
`66`	`69`	`/* decls for local routines only used within this module */`
`67`	`70`	`static void InitPlan(QueryDesc *queryDesc, int eflags);`
`68`	`71`	`static void ExecEndPlan(PlanState planstate, EState estate);`
`@@ -416,6 +419,9 @@ ExecCheckRTPerms(List *rangeTable)`
`416`	`419`	`{`
`417`	`420`	`ExecCheckRTEPerms((RangeTblEntry *) lfirst(l));`
`418`	`421`	`}`
	`422`	`+`
	`423`	`+ if (ExecutorCheckPerms_hook)`
	`424`	`+ (*ExecutorCheckPerms_hook)(rangeTable);`
`419`	`425`	`}`
`420`	`426`
`421`	`427`	`/*`