When the RI triggers lock the PK table, temporarily switch the current user

id to the owner of the PK table, to avoid permission problems.

diff --git a/src/backend/utils/adt/ri_triggers.c b/src/backend/utils/adt/ri_triggers.c
index df2447d66f8e58544290d82241020159f3078d2b..723adcda35cc2b583e4fd652d89c8c240095dc82 100644 (file)
--- a/src/backend/utils/adt/ri_triggers.c
+++ b/src/backend/utils/adt/ri_triggers.c
@@ -6,7 +6,7 @@
  *
  *     1999 Jan Wieck
  *
- * $Header: /cvsroot/pgsql/src/backend/utils/adt/ri_triggers.c,v 1.16 2000/05/30 00:49:53 momjian Exp $
+ * $Header: /cvsroot/pgsql/src/backend/utils/adt/ri_triggers.c,v 1.17 2000/09/25 22:34:20 petere Exp $
  *
  * ----------
  */
@@ -24,6 +24,7 @@
 #include "catalog/pg_operator.h"
 #include "commands/trigger.h"
 #include "executor/spi_priv.h"
+#include "miscadmin.h"
 
 
 /* ----------
@@ -158,6 +159,9 @@ RI_FKey_check(PG_FUNCTION_ARGS)
        bool            isnull;
        int                     i;
        int                     match_type;
+       Oid                     save_uid;
+
+       save_uid = GetUserId();
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -252,9 +256,13 @@ RI_FKey_check(PG_FUNCTION_ARGS)
                if (SPI_connect() != SPI_OK_CONNECT)
                        elog(NOTICE, "SPI_connect() failed in RI_FKey_check()");
 
+               SetUserId(RelationGetForm(pk_rel)->relowner);
+
                if (SPI_execp(qplan, check_values, check_nulls, 1) != SPI_OK_SELECT)
                        elog(ERROR, "SPI_execp() failed in RI_FKey_check()");
 
+               SetUserId(save_uid);
+
                if (SPI_processed == 0)
                        elog(ERROR, "%s referential integrity violation - "
                                 "no rows found in %s",
@@ -435,9 +443,14 @@ RI_FKey_check(PG_FUNCTION_ARGS)
         * Now check that foreign key exists in PK table
         * ----------
         */
+
+       SetUserId(RelationGetForm(pk_rel)->relowner);
+
        if (SPI_execp(qplan, check_values, check_nulls, 1) != SPI_OK_SELECT)
                elog(ERROR, "SPI_execp() failed in RI_FKey_check()");
 
+       SetUserId(save_uid);
+
        if (SPI_processed == 0)
                elog(ERROR, "%s referential integrity violation - "
                         "key referenced from %s not found in %s",
@@ -508,6 +521,9 @@ RI_FKey_noaction_del(PG_FUNCTION_ARGS)
        char            del_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid         save_uid;
+
+       save_uid = GetUserId();
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -659,9 +675,13 @@ RI_FKey_noaction_del(PG_FUNCTION_ARGS)
                         * Now check for existing references
                         * ----------
                         */
+                       SetUserId(RelationGetForm(pk_rel)->relowner);
+
                        if (SPI_execp(qplan, del_values, del_nulls, 1) != SPI_OK_SELECT)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_noaction_del()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_processed > 0)
                                elog(ERROR, "%s referential integrity violation - "
                                         "key in %s still referenced from %s",
@@ -716,6 +736,9 @@ RI_FKey_noaction_upd(PG_FUNCTION_ARGS)
        char            upd_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid         save_uid;
+
+       save_uid = GetUserId();
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -876,9 +899,13 @@ RI_FKey_noaction_upd(PG_FUNCTION_ARGS)
                         * Now check for existing references
                         * ----------
                         */
+                       SetUserId(RelationGetForm(pk_rel)->relowner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 1) != SPI_OK_SELECT)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_noaction_upd()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_processed > 0)
                                elog(ERROR, "%s referential integrity violation - "
                                         "key in %s still referenced from %s",
@@ -1570,6 +1597,9 @@ RI_FKey_restrict_upd(PG_FUNCTION_ARGS)
        char            upd_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid         save_uid;
+
+       save_uid = GetUserId();
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -1730,9 +1760,13 @@ RI_FKey_restrict_upd(PG_FUNCTION_ARGS)
                         * Now check for existing references
                         * ----------
                         */
+                       SetUserId(RelationGetForm(pk_rel)->relowner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 1) != SPI_OK_SELECT)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_restrict_upd()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_processed > 0)
                                elog(ERROR, "%s referential integrity violation - "
                                         "key in %s still referenced from %s",