Fix remaining RI permission problems (cascaded update/delete, restrict,

set null/default).

diff --git a/src/backend/utils/adt/ri_triggers.c b/src/backend/utils/adt/ri_triggers.c
index 6294eb086cf2c39f6b2390716b9245fa666c82de..71c1ccd4549a9259fc46408897908df785adc187 100644 (file)
--- a/src/backend/utils/adt/ri_triggers.c
+++ b/src/backend/utils/adt/ri_triggers.c
@@ -18,7 +18,7 @@
  * Portions Copyright (c) 2000-2001, PostgreSQL Global Development Group
  * Copyright 1999 Jan Wieck
  *
- * $Header: /cvsroot/pgsql/src/backend/utils/adt/ri_triggers.c,v 1.23 2001/03/22 06:16:17 momjian Exp $
+ * $Header: /cvsroot/pgsql/src/backend/utils/adt/ri_triggers.c,v 1.24 2001/05/07 19:57:24 petere Exp $
  *
  * ----------
  */
@@ -941,6 +941,8 @@ RI_FKey_cascade_del(PG_FUNCTION_ARGS)
        char            del_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -978,6 +980,7 @@ RI_FKey_cascade_del(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        old_row = trigdata->tg_trigtuple;
 
@@ -1081,9 +1084,14 @@ RI_FKey_cascade_del(PG_FUNCTION_ARGS)
                        /*
                         * Now delete constraint
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, del_values, del_nulls, 0) != SPI_OK_DELETE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_cascade_del()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_cascade_del()");
 
@@ -1128,6 +1136,8 @@ RI_FKey_cascade_upd(PG_FUNCTION_ARGS)
        bool            isnull;
        int                     i;
        int                     j;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -1165,6 +1175,7 @@ RI_FKey_cascade_upd(PG_FUNCTION_ARGS)
         * and old tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        new_row = trigdata->tg_newtuple;
        old_row = trigdata->tg_trigtuple;
@@ -1297,9 +1308,14 @@ RI_FKey_cascade_upd(PG_FUNCTION_ARGS)
                        /*
                         * Now update the existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 0) != SPI_OK_UPDATE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_cascade_upd()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_cascade_upd()");
 
@@ -1349,6 +1365,8 @@ RI_FKey_restrict_del(PG_FUNCTION_ARGS)
        char            del_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -1386,6 +1404,7 @@ RI_FKey_restrict_del(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        old_row = trigdata->tg_trigtuple;
 
@@ -1493,9 +1512,14 @@ RI_FKey_restrict_del(PG_FUNCTION_ARGS)
                        /*
                         * Now check for existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, del_values, del_nulls, 1) != SPI_OK_SELECT)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_restrict_del()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_processed > 0)
                                elog(ERROR, "%s referential integrity violation - "
                                         "key in %s still referenced from %s",
@@ -1554,8 +1578,7 @@ RI_FKey_restrict_upd(PG_FUNCTION_ARGS)
        bool            isnull;
        int                     i;
        Oid                     save_uid;
-
-       save_uid = GetUserId();
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -1593,6 +1616,7 @@ RI_FKey_restrict_upd(PG_FUNCTION_ARGS)
         * and old tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        new_row = trigdata->tg_newtuple;
        old_row = trigdata->tg_trigtuple;
@@ -1708,6 +1732,9 @@ RI_FKey_restrict_upd(PG_FUNCTION_ARGS)
                        /*
                         * Now check for existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        SetUserId(RelationGetForm(pk_rel)->relowner);
 
                        if (SPI_execp(qplan, upd_values, upd_nulls, 1) != SPI_OK_SELECT)
@@ -1764,6 +1791,8 @@ RI_FKey_setnull_del(PG_FUNCTION_ARGS)
        char            upd_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -1801,6 +1830,7 @@ RI_FKey_setnull_del(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        old_row = trigdata->tg_trigtuple;
 
@@ -1915,9 +1945,14 @@ RI_FKey_setnull_del(PG_FUNCTION_ARGS)
                        /*
                         * Now update the existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 0) != SPI_OK_UPDATE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_setnull_del()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_setnull_del()");
 
@@ -1963,6 +1998,8 @@ RI_FKey_setnull_upd(PG_FUNCTION_ARGS)
        int                     i;
        int                     match_type;
        bool            use_cached_query;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -2000,6 +2037,7 @@ RI_FKey_setnull_upd(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        new_row = trigdata->tg_newtuple;
        old_row = trigdata->tg_trigtuple;
@@ -2161,9 +2199,14 @@ RI_FKey_setnull_upd(PG_FUNCTION_ARGS)
                        /*
                         * Now update the existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 0) != SPI_OK_UPDATE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_setnull_upd()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_setnull_upd()");
 
@@ -2206,6 +2249,8 @@ RI_FKey_setdefault_del(PG_FUNCTION_ARGS)
        char            upd_nulls[RI_MAX_NUMKEYS + 1];
        bool            isnull;
        int                     i;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -2243,6 +2288,7 @@ RI_FKey_setdefault_del(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        old_row = trigdata->tg_trigtuple;
 
@@ -2404,9 +2450,14 @@ RI_FKey_setdefault_del(PG_FUNCTION_ARGS)
                        /*
                         * Now update the existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 0) != SPI_OK_UPDATE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_setdefault_del()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_setdefault_del()");
 
@@ -2451,6 +2502,8 @@ RI_FKey_setdefault_upd(PG_FUNCTION_ARGS)
        bool            isnull;
        int                     i;
        int                     match_type;
+       Oid                     save_uid;
+       Oid                     fk_owner;
 
        ReferentialIntegritySnapshotOverride = true;
 
@@ -2488,6 +2541,7 @@ RI_FKey_setdefault_upd(PG_FUNCTION_ARGS)
         * tuple.
         */
        fk_rel = heap_openr(tgargs[RI_FK_RELNAME_ARGNO], NoLock);
+       fk_owner = RelationGetForm(fk_rel)->relowner;
        pk_rel = trigdata->tg_relation;
        new_row = trigdata->tg_newtuple;
        old_row = trigdata->tg_trigtuple;
@@ -2676,9 +2730,14 @@ RI_FKey_setdefault_upd(PG_FUNCTION_ARGS)
                        /*
                         * Now update the existing references
                         */
+                       save_uid = GetUserId();
+                       SetUserId(fk_owner);
+
                        if (SPI_execp(qplan, upd_values, upd_nulls, 0) != SPI_OK_UPDATE)
                                elog(ERROR, "SPI_execp() failed in RI_FKey_setdefault_upd()");
 
+                       SetUserId(save_uid);
+
                        if (SPI_finish() != SPI_OK_FINISH)
                                elog(NOTICE, "SPI_finish() failed in RI_FKey_setdefault_upd()");