LCOV - code coverage report
Current view: top level - src/backend/catalog - pg_shdepend.c (source / functions) Coverage Total Hit
Test: Code coverage Lines: 76.6 % 638 489
Test Date: 2026-01-26 10:56:24 Functions: 95.8 % 24 23
Legend: Lines:     hit not hit
Branches: + taken - not taken # not executed
Branches: 54.4 % 309 168

             Branch data     Line data    Source code
       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * pg_shdepend.c
       4                 :             :  *        routines to support manipulation of the pg_shdepend relation
       5                 :             :  *
       6                 :             :  * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
       7                 :             :  * Portions Copyright (c) 1994, Regents of the University of California
       8                 :             :  *
       9                 :             :  *
      10                 :             :  * IDENTIFICATION
      11                 :             :  *        src/backend/catalog/pg_shdepend.c
      12                 :             :  *
      13                 :             :  *-------------------------------------------------------------------------
      14                 :             :  */
      15                 :             : #include "postgres.h"
      16                 :             : 
      17                 :             : #include "access/genam.h"
      18                 :             : #include "access/htup_details.h"
      19                 :             : #include "access/table.h"
      20                 :             : #include "access/xact.h"
      21                 :             : #include "catalog/catalog.h"
      22                 :             : #include "catalog/dependency.h"
      23                 :             : #include "catalog/indexing.h"
      24                 :             : #include "catalog/pg_authid.h"
      25                 :             : #include "catalog/pg_auth_members.h"
      26                 :             : #include "catalog/pg_collation.h"
      27                 :             : #include "catalog/pg_conversion.h"
      28                 :             : #include "catalog/pg_database.h"
      29                 :             : #include "catalog/pg_default_acl.h"
      30                 :             : #include "catalog/pg_event_trigger.h"
      31                 :             : #include "catalog/pg_extension.h"
      32                 :             : #include "catalog/pg_foreign_data_wrapper.h"
      33                 :             : #include "catalog/pg_foreign_server.h"
      34                 :             : #include "catalog/pg_language.h"
      35                 :             : #include "catalog/pg_largeobject.h"
      36                 :             : #include "catalog/pg_namespace.h"
      37                 :             : #include "catalog/pg_opclass.h"
      38                 :             : #include "catalog/pg_operator.h"
      39                 :             : #include "catalog/pg_opfamily.h"
      40                 :             : #include "catalog/pg_proc.h"
      41                 :             : #include "catalog/pg_shdepend.h"
      42                 :             : #include "catalog/pg_statistic_ext.h"
      43                 :             : #include "catalog/pg_subscription.h"
      44                 :             : #include "catalog/pg_tablespace.h"
      45                 :             : #include "catalog/pg_ts_config.h"
      46                 :             : #include "catalog/pg_ts_dict.h"
      47                 :             : #include "catalog/pg_type.h"
      48                 :             : #include "catalog/pg_user_mapping.h"
      49                 :             : #include "commands/alter.h"
      50                 :             : #include "commands/defrem.h"
      51                 :             : #include "commands/event_trigger.h"
      52                 :             : #include "commands/policy.h"
      53                 :             : #include "commands/publicationcmds.h"
      54                 :             : #include "commands/schemacmds.h"
      55                 :             : #include "commands/subscriptioncmds.h"
      56                 :             : #include "commands/tablecmds.h"
      57                 :             : #include "commands/tablespace.h"
      58                 :             : #include "commands/typecmds.h"
      59                 :             : #include "miscadmin.h"
      60                 :             : #include "storage/lmgr.h"
      61                 :             : #include "utils/acl.h"
      62                 :             : #include "utils/fmgroids.h"
      63                 :             : #include "utils/lsyscache.h"
      64                 :             : #include "utils/memutils.h"
      65                 :             : #include "utils/syscache.h"
      66                 :             : 
      67                 :             : typedef enum
      68                 :             : {
      69                 :             :         LOCAL_OBJECT,
      70                 :             :         SHARED_OBJECT,
      71                 :             :         REMOTE_OBJECT,
      72                 :             : } SharedDependencyObjectType;
      73                 :             : 
      74                 :             : typedef struct
      75                 :             : {
      76                 :             :         ObjectAddress object;
      77                 :             :         char            deptype;
      78                 :             :         SharedDependencyObjectType objtype;
      79                 :             : } ShDependObjectInfo;
      80                 :             : 
      81                 :             : static void getOidListDiff(Oid *list1, int *nlist1, Oid *list2, int *nlist2);
      82                 :             : static Oid      classIdGetDbId(Oid classId);
      83                 :             : static void shdepChangeDep(Relation sdepRel,
      84                 :             :                                                    Oid classid, Oid objid, int32 objsubid,
      85                 :             :                                                    Oid refclassid, Oid refobjid,
      86                 :             :                                                    SharedDependencyType deptype);
      87                 :             : static void updateAclDependenciesWorker(Oid classId, Oid objectId,
      88                 :             :                                                                                 int32 objsubId, Oid ownerId,
      89                 :             :                                                                                 SharedDependencyType deptype,
      90                 :             :                                                                                 int noldmembers, Oid *oldmembers,
      91                 :             :                                                                                 int nnewmembers, Oid *newmembers);
      92                 :             : static void shdepAddDependency(Relation sdepRel,
      93                 :             :                                                            Oid classId, Oid objectId, int32 objsubId,
      94                 :             :                                                            Oid refclassId, Oid refobjId,
      95                 :             :                                                            SharedDependencyType deptype);
      96                 :             : static void shdepDropDependency(Relation sdepRel,
      97                 :             :                                                                 Oid classId, Oid objectId, int32 objsubId,
      98                 :             :                                                                 bool drop_subobjects,
      99                 :             :                                                                 Oid refclassId, Oid refobjId,
     100                 :             :                                                                 SharedDependencyType deptype);
     101                 :             : static void storeObjectDescription(StringInfo descs,
     102                 :             :                                                                    SharedDependencyObjectType type,
     103                 :             :                                                                    ObjectAddress *object,
     104                 :             :                                                                    SharedDependencyType deptype,
     105                 :             :                                                                    int count);
     106                 :             : static void shdepReassignOwned_Owner(Form_pg_shdepend sdepForm, Oid newrole);
     107                 :             : static void shdepReassignOwned_InitAcl(Form_pg_shdepend sdepForm,
     108                 :             :                                                                            Oid oldrole, Oid newrole);
     109                 :             : 
     110                 :             : 
     111                 :             : /*
     112                 :             :  * recordSharedDependencyOn
     113                 :             :  *
     114                 :             :  * Record a dependency between 2 objects via their respective ObjectAddresses.
     115                 :             :  * The first argument is the dependent object, the second the one it
     116                 :             :  * references (which must be a shared object).
     117                 :             :  *
     118                 :             :  * This locks the referenced object and makes sure it still exists.
     119                 :             :  * Then it creates an entry in pg_shdepend.  The lock is kept until
     120                 :             :  * the end of the transaction.
     121                 :             :  *
     122                 :             :  * Dependencies on pinned objects are not recorded.
     123                 :             :  */
     124                 :             : void
     125                 :        9221 : recordSharedDependencyOn(ObjectAddress *depender,
     126                 :             :                                                  ObjectAddress *referenced,
     127                 :             :                                                  SharedDependencyType deptype)
     128                 :             : {
     129                 :        9221 :         Relation        sdepRel;
     130                 :             : 
     131                 :             :         /*
     132                 :             :          * Objects in pg_shdepend can't have SubIds.
     133                 :             :          */
     134         [ +  - ]:        9221 :         Assert(depender->objectSubId == 0);
     135         [ +  - ]:        9221 :         Assert(referenced->objectSubId == 0);
     136                 :             : 
     137                 :             :         /*
     138                 :             :          * During bootstrap, do nothing since pg_shdepend may not exist yet.
     139                 :             :          * initdb will fill in appropriate pg_shdepend entries after bootstrap.
     140                 :             :          */
     141         [ +  - ]:        9221 :         if (IsBootstrapProcessingMode())
     142                 :           0 :                 return;
     143                 :             : 
     144                 :        9221 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
     145                 :             : 
     146                 :             :         /* If the referenced object is pinned, do nothing. */
     147         [ +  + ]:        9221 :         if (!IsPinnedObject(referenced->classId, referenced->objectId))
     148                 :             :         {
     149                 :        1464 :                 shdepAddDependency(sdepRel, depender->classId, depender->objectId,
     150                 :         732 :                                                    depender->objectSubId,
     151                 :         732 :                                                    referenced->classId, referenced->objectId,
     152                 :         732 :                                                    deptype);
     153                 :         732 :         }
     154                 :             : 
     155                 :        9221 :         table_close(sdepRel, RowExclusiveLock);
     156         [ -  + ]:        9221 : }
     157                 :             : 
     158                 :             : /*
     159                 :             :  * recordDependencyOnOwner
     160                 :             :  *
     161                 :             :  * A convenient wrapper of recordSharedDependencyOn -- register the specified
     162                 :             :  * user as owner of the given object.
     163                 :             :  *
     164                 :             :  * Note: it's the caller's responsibility to ensure that there isn't an owner
     165                 :             :  * entry for the object already.
     166                 :             :  */
     167                 :             : void
     168                 :        9168 : recordDependencyOnOwner(Oid classId, Oid objectId, Oid owner)
     169                 :             : {
     170                 :        9168 :         ObjectAddress myself,
     171                 :             :                                 referenced;
     172                 :             : 
     173                 :        9168 :         myself.classId = classId;
     174                 :        9168 :         myself.objectId = objectId;
     175                 :        9168 :         myself.objectSubId = 0;
     176                 :             : 
     177                 :        9168 :         referenced.classId = AuthIdRelationId;
     178                 :        9168 :         referenced.objectId = owner;
     179                 :        9168 :         referenced.objectSubId = 0;
     180                 :             : 
     181                 :        9168 :         recordSharedDependencyOn(&myself, &referenced, SHARED_DEPENDENCY_OWNER);
     182                 :        9168 : }
     183                 :             : 
     184                 :             : /*
     185                 :             :  * shdepChangeDep
     186                 :             :  *
     187                 :             :  * Update shared dependency records to account for an updated referenced
     188                 :             :  * object.  This is an internal workhorse for operations such as changing
     189                 :             :  * an object's owner.
     190                 :             :  *
     191                 :             :  * There must be no more than one existing entry for the given dependent
     192                 :             :  * object and dependency type!  So in practice this can only be used for
     193                 :             :  * updating SHARED_DEPENDENCY_OWNER and SHARED_DEPENDENCY_TABLESPACE
     194                 :             :  * entries, which should have that property.
     195                 :             :  *
     196                 :             :  * If there is no previous entry, we assume it was referencing a PINned
     197                 :             :  * object, so we create a new entry.  If the new referenced object is
     198                 :             :  * PINned, we don't create an entry (and drop the old one, if any).
     199                 :             :  * (For tablespaces, we don't record dependencies in certain cases, so
     200                 :             :  * there are other possible reasons for entries to be missing.)
     201                 :             :  *
     202                 :             :  * sdepRel must be the pg_shdepend relation, already opened and suitably
     203                 :             :  * locked.
     204                 :             :  */
     205                 :             : static void
     206                 :         104 : shdepChangeDep(Relation sdepRel,
     207                 :             :                            Oid classid, Oid objid, int32 objsubid,
     208                 :             :                            Oid refclassid, Oid refobjid,
     209                 :             :                            SharedDependencyType deptype)
     210                 :             : {
     211                 :         104 :         Oid                     dbid = classIdGetDbId(classid);
     212                 :         104 :         HeapTuple       oldtup = NULL;
     213                 :         104 :         HeapTuple       scantup;
     214                 :         104 :         ScanKeyData key[4];
     215                 :         104 :         SysScanDesc scan;
     216                 :             : 
     217                 :             :         /*
     218                 :             :          * Make sure the new referenced object doesn't go away while we record the
     219                 :             :          * dependency.
     220                 :             :          */
     221                 :         104 :         shdepLockAndCheckObject(refclassid, refobjid);
     222                 :             : 
     223                 :             :         /*
     224                 :             :          * Look for a previous entry
     225                 :             :          */
     226                 :         208 :         ScanKeyInit(&key[0],
     227                 :             :                                 Anum_pg_shdepend_dbid,
     228                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     229                 :         104 :                                 ObjectIdGetDatum(dbid));
     230                 :         208 :         ScanKeyInit(&key[1],
     231                 :             :                                 Anum_pg_shdepend_classid,
     232                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     233                 :         104 :                                 ObjectIdGetDatum(classid));
     234                 :         208 :         ScanKeyInit(&key[2],
     235                 :             :                                 Anum_pg_shdepend_objid,
     236                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     237                 :         104 :                                 ObjectIdGetDatum(objid));
     238                 :         208 :         ScanKeyInit(&key[3],
     239                 :             :                                 Anum_pg_shdepend_objsubid,
     240                 :             :                                 BTEqualStrategyNumber, F_INT4EQ,
     241                 :         104 :                                 Int32GetDatum(objsubid));
     242                 :             : 
     243                 :         208 :         scan = systable_beginscan(sdepRel, SharedDependDependerIndexId, true,
     244                 :         104 :                                                           NULL, 4, key);
     245                 :             : 
     246         [ +  + ]:         156 :         while ((scantup = systable_getnext(scan)) != NULL)
     247                 :             :         {
     248                 :             :                 /* Ignore if not of the target dependency type */
     249         [ +  + ]:          52 :                 if (((Form_pg_shdepend) GETSTRUCT(scantup))->deptype != deptype)
     250                 :           3 :                         continue;
     251                 :             :                 /* Caller screwed up if multiple matches */
     252         [ +  - ]:          49 :                 if (oldtup)
     253   [ #  #  #  # ]:           0 :                         elog(ERROR,
     254                 :             :                                  "multiple pg_shdepend entries for object %u/%u/%d deptype %c",
     255                 :             :                                  classid, objid, objsubid, deptype);
     256                 :          49 :                 oldtup = heap_copytuple(scantup);
     257                 :             :         }
     258                 :             : 
     259                 :         104 :         systable_endscan(scan);
     260                 :             : 
     261         [ +  + ]:         104 :         if (IsPinnedObject(refclassid, refobjid))
     262                 :             :         {
     263                 :             :                 /* No new entry needed, so just delete existing entry if any */
     264         [ +  + ]:           4 :                 if (oldtup)
     265                 :           2 :                         CatalogTupleDelete(sdepRel, &oldtup->t_self);
     266                 :           4 :         }
     267         [ +  + ]:         100 :         else if (oldtup)
     268                 :             :         {
     269                 :             :                 /* Need to update existing entry */
     270                 :          47 :                 Form_pg_shdepend shForm = (Form_pg_shdepend) GETSTRUCT(oldtup);
     271                 :             : 
     272                 :             :                 /* Since oldtup is a copy, we can just modify it in-memory */
     273                 :          47 :                 shForm->refclassid = refclassid;
     274                 :          47 :                 shForm->refobjid = refobjid;
     275                 :             : 
     276                 :          47 :                 CatalogTupleUpdate(sdepRel, &oldtup->t_self, oldtup);
     277                 :          47 :         }
     278                 :             :         else
     279                 :             :         {
     280                 :             :                 /* Need to insert new entry */
     281                 :          53 :                 Datum           values[Natts_pg_shdepend];
     282                 :          53 :                 bool            nulls[Natts_pg_shdepend];
     283                 :             : 
     284                 :          53 :                 memset(nulls, false, sizeof(nulls));
     285                 :             : 
     286                 :          53 :                 values[Anum_pg_shdepend_dbid - 1] = ObjectIdGetDatum(dbid);
     287                 :          53 :                 values[Anum_pg_shdepend_classid - 1] = ObjectIdGetDatum(classid);
     288                 :          53 :                 values[Anum_pg_shdepend_objid - 1] = ObjectIdGetDatum(objid);
     289                 :          53 :                 values[Anum_pg_shdepend_objsubid - 1] = Int32GetDatum(objsubid);
     290                 :             : 
     291                 :          53 :                 values[Anum_pg_shdepend_refclassid - 1] = ObjectIdGetDatum(refclassid);
     292                 :          53 :                 values[Anum_pg_shdepend_refobjid - 1] = ObjectIdGetDatum(refobjid);
     293                 :          53 :                 values[Anum_pg_shdepend_deptype - 1] = CharGetDatum(deptype);
     294                 :             : 
     295                 :             :                 /*
     296                 :             :                  * we are reusing oldtup just to avoid declaring a new variable, but
     297                 :             :                  * it's certainly a new tuple
     298                 :             :                  */
     299                 :          53 :                 oldtup = heap_form_tuple(RelationGetDescr(sdepRel), values, nulls);
     300                 :          53 :                 CatalogTupleInsert(sdepRel, oldtup);
     301                 :          53 :         }
     302                 :             : 
     303         [ +  + ]:         104 :         if (oldtup)
     304                 :         102 :                 heap_freetuple(oldtup);
     305                 :         104 : }
     306                 :             : 
     307                 :             : /*
     308                 :             :  * changeDependencyOnOwner
     309                 :             :  *
     310                 :             :  * Update the shared dependencies to account for the new owner.
     311                 :             :  *
     312                 :             :  * Note: we don't need an objsubid argument because only whole objects
     313                 :             :  * have owners.
     314                 :             :  */
     315                 :             : void
     316                 :         102 : changeDependencyOnOwner(Oid classId, Oid objectId, Oid newOwnerId)
     317                 :             : {
     318                 :         102 :         Relation        sdepRel;
     319                 :             : 
     320                 :         102 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
     321                 :             : 
     322                 :             :         /* Adjust the SHARED_DEPENDENCY_OWNER entry */
     323                 :         204 :         shdepChangeDep(sdepRel,
     324                 :         102 :                                    classId, objectId, 0,
     325                 :         102 :                                    AuthIdRelationId, newOwnerId,
     326                 :             :                                    SHARED_DEPENDENCY_OWNER);
     327                 :             : 
     328                 :             :         /*----------
     329                 :             :          * There should never be a SHARED_DEPENDENCY_ACL entry for the owner,
     330                 :             :          * so get rid of it if there is one.  This can happen if the new owner
     331                 :             :          * was previously granted some rights to the object.
     332                 :             :          *
     333                 :             :          * This step is analogous to aclnewowner's removal of duplicate entries
     334                 :             :          * in the ACL.  We have to do it to handle this scenario:
     335                 :             :          *              A grants some rights on an object to B
     336                 :             :          *              ALTER OWNER changes the object's owner to B
     337                 :             :          *              ALTER OWNER changes the object's owner to C
     338                 :             :          * The third step would remove all mention of B from the object's ACL,
     339                 :             :          * but we'd still have a SHARED_DEPENDENCY_ACL for B if we did not do
     340                 :             :          * things this way.
     341                 :             :          *
     342                 :             :          * The rule against having a SHARED_DEPENDENCY_ACL entry for the owner
     343                 :             :          * allows us to fix things up in just this one place, without having
     344                 :             :          * to make the various ALTER OWNER routines each know about it.
     345                 :             :          *----------
     346                 :             :          */
     347                 :         204 :         shdepDropDependency(sdepRel, classId, objectId, 0, true,
     348                 :         102 :                                                 AuthIdRelationId, newOwnerId,
     349                 :             :                                                 SHARED_DEPENDENCY_ACL);
     350                 :             : 
     351                 :             :         /*
     352                 :             :          * However, nothing need be done about SHARED_DEPENDENCY_INITACL entries,
     353                 :             :          * since those exist whether or not the role is the object's owner, and
     354                 :             :          * ALTER OWNER does not modify the underlying pg_init_privs entry.
     355                 :             :          */
     356                 :             : 
     357                 :         102 :         table_close(sdepRel, RowExclusiveLock);
     358                 :         102 : }
     359                 :             : 
     360                 :             : /*
     361                 :             :  * recordDependencyOnTablespace
     362                 :             :  *
     363                 :             :  * A convenient wrapper of recordSharedDependencyOn -- register the specified
     364                 :             :  * tablespace as default for the given object.
     365                 :             :  *
     366                 :             :  * Note: it's the caller's responsibility to ensure that there isn't a
     367                 :             :  * tablespace entry for the object already.
     368                 :             :  */
     369                 :             : void
     370                 :          21 : recordDependencyOnTablespace(Oid classId, Oid objectId, Oid tablespace)
     371                 :             : {
     372                 :          21 :         ObjectAddress myself,
     373                 :             :                                 referenced;
     374                 :             : 
     375                 :          21 :         ObjectAddressSet(myself, classId, objectId);
     376                 :          21 :         ObjectAddressSet(referenced, TableSpaceRelationId, tablespace);
     377                 :             : 
     378                 :          21 :         recordSharedDependencyOn(&myself, &referenced,
     379                 :             :                                                          SHARED_DEPENDENCY_TABLESPACE);
     380                 :          21 : }
     381                 :             : 
     382                 :             : /*
     383                 :             :  * changeDependencyOnTablespace
     384                 :             :  *
     385                 :             :  * Update the shared dependencies to account for the new tablespace.
     386                 :             :  *
     387                 :             :  * Note: we don't need an objsubid argument because only whole objects
     388                 :             :  * have tablespaces.
     389                 :             :  */
     390                 :             : void
     391                 :           5 : changeDependencyOnTablespace(Oid classId, Oid objectId, Oid newTablespaceId)
     392                 :             : {
     393                 :           5 :         Relation        sdepRel;
     394                 :             : 
     395                 :           5 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
     396                 :             : 
     397   [ +  -  +  + ]:           5 :         if (newTablespaceId != DEFAULTTABLESPACE_OID &&
     398                 :           5 :                 newTablespaceId != InvalidOid)
     399                 :           4 :                 shdepChangeDep(sdepRel,
     400                 :           2 :                                            classId, objectId, 0,
     401                 :           2 :                                            TableSpaceRelationId, newTablespaceId,
     402                 :             :                                            SHARED_DEPENDENCY_TABLESPACE);
     403                 :             :         else
     404                 :           6 :                 shdepDropDependency(sdepRel,
     405                 :           3 :                                                         classId, objectId, 0, true,
     406                 :             :                                                         InvalidOid, InvalidOid,
     407                 :             :                                                         SHARED_DEPENDENCY_INVALID);
     408                 :             : 
     409                 :           5 :         table_close(sdepRel, RowExclusiveLock);
     410                 :           5 : }
     411                 :             : 
     412                 :             : /*
     413                 :             :  * getOidListDiff
     414                 :             :  *              Helper for updateAclDependencies.
     415                 :             :  *
     416                 :             :  * Takes two Oid arrays and removes elements that are common to both arrays,
     417                 :             :  * leaving just those that are in one input but not the other.
     418                 :             :  * We assume both arrays have been sorted and de-duped.
     419                 :             :  */
     420                 :             : static void
     421                 :         850 : getOidListDiff(Oid *list1, int *nlist1, Oid *list2, int *nlist2)
     422                 :             : {
     423                 :         850 :         int                     in1,
     424                 :             :                                 in2,
     425                 :             :                                 out1,
     426                 :             :                                 out2;
     427                 :             : 
     428                 :         850 :         in1 = in2 = out1 = out2 = 0;
     429   [ +  +  +  + ]:        1266 :         while (in1 < *nlist1 && in2 < *nlist2)
     430                 :             :         {
     431         [ +  + ]:         416 :                 if (list1[in1] == list2[in2])
     432                 :             :                 {
     433                 :             :                         /* skip over duplicates */
     434                 :         392 :                         in1++;
     435                 :         392 :                         in2++;
     436                 :         392 :                 }
     437         [ +  + ]:          24 :                 else if (list1[in1] < list2[in2])
     438                 :             :                 {
     439                 :             :                         /* list1[in1] is not in list2 */
     440                 :          18 :                         list1[out1++] = list1[in1++];
     441                 :          18 :                 }
     442                 :             :                 else
     443                 :             :                 {
     444                 :             :                         /* list2[in2] is not in list1 */
     445                 :           6 :                         list2[out2++] = list2[in2++];
     446                 :             :                 }
     447                 :             :         }
     448                 :             : 
     449                 :             :         /* any remaining list1 entries are not in list2 */
     450         [ +  + ]:         941 :         while (in1 < *nlist1)
     451                 :             :         {
     452                 :          91 :                 list1[out1++] = list1[in1++];
     453                 :             :         }
     454                 :             : 
     455                 :             :         /* any remaining list2 entries are not in list1 */
     456         [ +  + ]:        1704 :         while (in2 < *nlist2)
     457                 :             :         {
     458                 :         854 :                 list2[out2++] = list2[in2++];
     459                 :             :         }
     460                 :             : 
     461                 :         850 :         *nlist1 = out1;
     462                 :         850 :         *nlist2 = out2;
     463                 :         850 : }
     464                 :             : 
     465                 :             : /*
     466                 :             :  * updateAclDependencies
     467                 :             :  *              Update the pg_shdepend info for an object's ACL during GRANT/REVOKE.
     468                 :             :  *
     469                 :             :  * classId, objectId, objsubId: identify the object whose ACL this is
     470                 :             :  * ownerId: role owning the object
     471                 :             :  * noldmembers, oldmembers: array of roleids appearing in old ACL
     472                 :             :  * nnewmembers, newmembers: array of roleids appearing in new ACL
     473                 :             :  *
     474                 :             :  * We calculate the differences between the new and old lists of roles,
     475                 :             :  * and then insert or delete from pg_shdepend as appropriate.
     476                 :             :  *
     477                 :             :  * Note that we can't just insert all referenced roles blindly during GRANT,
     478                 :             :  * because we would end up with duplicate registered dependencies.  We could
     479                 :             :  * check for existence of the tuples before inserting, but that seems to be
     480                 :             :  * more expensive than what we are doing here.  Likewise we can't just delete
     481                 :             :  * blindly during REVOKE, because the user may still have other privileges.
     482                 :             :  * It is also possible that REVOKE actually adds dependencies, due to
     483                 :             :  * instantiation of a formerly implicit default ACL (although at present,
     484                 :             :  * all such dependencies should be for the owning role, which we ignore here).
     485                 :             :  *
     486                 :             :  * NOTE: Both input arrays must be sorted and de-duped.  (Typically they
     487                 :             :  * are extracted from an ACL array by aclmembers(), which takes care of
     488                 :             :  * both requirements.)  The arrays are pfreed before return.
     489                 :             :  */
     490                 :             : void
     491                 :         846 : updateAclDependencies(Oid classId, Oid objectId, int32 objsubId,
     492                 :             :                                           Oid ownerId,
     493                 :             :                                           int noldmembers, Oid *oldmembers,
     494                 :             :                                           int nnewmembers, Oid *newmembers)
     495                 :             : {
     496                 :        1692 :         updateAclDependenciesWorker(classId, objectId, objsubId,
     497                 :         846 :                                                                 ownerId, SHARED_DEPENDENCY_ACL,
     498                 :         846 :                                                                 noldmembers, oldmembers,
     499                 :         846 :                                                                 nnewmembers, newmembers);
     500                 :         846 : }
     501                 :             : 
     502                 :             : /*
     503                 :             :  * updateInitAclDependencies
     504                 :             :  *              Update the pg_shdepend info for a pg_init_privs entry.
     505                 :             :  *
     506                 :             :  * Exactly like updateAclDependencies, except we are considering a
     507                 :             :  * pg_init_privs ACL for the specified object.  Since recording of
     508                 :             :  * pg_init_privs role dependencies is the same for owners and non-owners,
     509                 :             :  * we do not need an ownerId argument.
     510                 :             :  */
     511                 :             : void
     512                 :           4 : updateInitAclDependencies(Oid classId, Oid objectId, int32 objsubId,
     513                 :             :                                                   int noldmembers, Oid *oldmembers,
     514                 :             :                                                   int nnewmembers, Oid *newmembers)
     515                 :             : {
     516                 :           8 :         updateAclDependenciesWorker(classId, objectId, objsubId,
     517                 :             :                                                                 InvalidOid, /* ownerId will not be consulted */
     518                 :             :                                                                 SHARED_DEPENDENCY_INITACL,
     519                 :           4 :                                                                 noldmembers, oldmembers,
     520                 :           4 :                                                                 nnewmembers, newmembers);
     521                 :           4 : }
     522                 :             : 
     523                 :             : /* Common code for the above two functions */
     524                 :             : static void
     525                 :         850 : updateAclDependenciesWorker(Oid classId, Oid objectId, int32 objsubId,
     526                 :             :                                                         Oid ownerId, SharedDependencyType deptype,
     527                 :             :                                                         int noldmembers, Oid *oldmembers,
     528                 :             :                                                         int nnewmembers, Oid *newmembers)
     529                 :             : {
     530                 :         850 :         Relation        sdepRel;
     531                 :         850 :         int                     i;
     532                 :             : 
     533                 :             :         /*
     534                 :             :          * Remove entries that are common to both lists; those represent existing
     535                 :             :          * dependencies we don't need to change.
     536                 :             :          *
     537                 :             :          * OK to overwrite the inputs since we'll pfree them anyway.
     538                 :             :          */
     539                 :         850 :         getOidListDiff(oldmembers, &noldmembers, newmembers, &nnewmembers);
     540                 :             : 
     541   [ +  +  +  + ]:         850 :         if (noldmembers > 0 || nnewmembers > 0)
     542                 :             :         {
     543                 :         754 :                 sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
     544                 :             : 
     545                 :             :                 /* Add new dependencies that weren't already present */
     546         [ +  + ]:        1614 :                 for (i = 0; i < nnewmembers; i++)
     547                 :             :                 {
     548                 :         860 :                         Oid                     roleid = newmembers[i];
     549                 :             : 
     550                 :             :                         /*
     551                 :             :                          * For SHARED_DEPENDENCY_ACL entries, skip the owner: she has an
     552                 :             :                          * OWNER shdep entry instead.  (This is not just a space
     553                 :             :                          * optimization; it makes ALTER OWNER easier.  See notes in
     554                 :             :                          * changeDependencyOnOwner.)  But for INITACL entries, we record
     555                 :             :                          * the owner too.
     556                 :             :                          */
     557   [ +  +  +  + ]:         860 :                         if (deptype == SHARED_DEPENDENCY_ACL && roleid == ownerId)
     558                 :         481 :                                 continue;
     559                 :             : 
     560                 :             :                         /* Skip pinned roles; they don't need dependency entries */
     561         [ +  + ]:         379 :                         if (IsPinnedObject(AuthIdRelationId, roleid))
     562                 :          88 :                                 continue;
     563                 :             : 
     564                 :         582 :                         shdepAddDependency(sdepRel, classId, objectId, objsubId,
     565                 :         291 :                                                            AuthIdRelationId, roleid,
     566                 :         291 :                                                            deptype);
     567         [ +  + ]:         860 :                 }
     568                 :             : 
     569                 :             :                 /* Drop no-longer-used old dependencies */
     570         [ +  + ]:         863 :                 for (i = 0; i < noldmembers; i++)
     571                 :             :                 {
     572                 :         109 :                         Oid                     roleid = oldmembers[i];
     573                 :             : 
     574                 :             :                         /* Skip the owner for ACL entries, same as above */
     575   [ +  -  +  + ]:         109 :                         if (deptype == SHARED_DEPENDENCY_ACL && roleid == ownerId)
     576                 :          15 :                                 continue;
     577                 :             : 
     578                 :             :                         /* Skip pinned roles */
     579         [ -  + ]:          94 :                         if (IsPinnedObject(AuthIdRelationId, roleid))
     580                 :           0 :                                 continue;
     581                 :             : 
     582                 :         188 :                         shdepDropDependency(sdepRel, classId, objectId, objsubId,
     583                 :             :                                                                 false,  /* exact match on objsubId */
     584                 :          94 :                                                                 AuthIdRelationId, roleid,
     585                 :          94 :                                                                 deptype);
     586         [ +  + ]:         109 :                 }
     587                 :             : 
     588                 :         754 :                 table_close(sdepRel, RowExclusiveLock);
     589                 :         754 :         }
     590                 :             : 
     591         [ +  + ]:         850 :         if (oldmembers)
     592                 :         253 :                 pfree(oldmembers);
     593         [ +  + ]:         850 :         if (newmembers)
     594                 :         821 :                 pfree(newmembers);
     595                 :         850 : }
     596                 :             : 
     597                 :             : /*
     598                 :             :  * A struct to keep track of dependencies found in other databases.
     599                 :             :  */
     600                 :             : typedef struct
     601                 :             : {
     602                 :             :         Oid                     dbOid;
     603                 :             :         int                     count;
     604                 :             : } remoteDep;
     605                 :             : 
     606                 :             : /*
     607                 :             :  * qsort comparator for ShDependObjectInfo items
     608                 :             :  */
     609                 :             : static int
     610                 :          20 : shared_dependency_comparator(const void *a, const void *b)
     611                 :             : {
     612                 :          20 :         const ShDependObjectInfo *obja = (const ShDependObjectInfo *) a;
     613                 :          20 :         const ShDependObjectInfo *objb = (const ShDependObjectInfo *) b;
     614                 :             : 
     615                 :             :         /*
     616                 :             :          * Primary sort key is OID ascending.
     617                 :             :          */
     618         [ +  + ]:          20 :         if (obja->object.objectId < objb->object.objectId)
     619                 :          14 :                 return -1;
     620         [ +  - ]:           6 :         if (obja->object.objectId > objb->object.objectId)
     621                 :           6 :                 return 1;
     622                 :             : 
     623                 :             :         /*
     624                 :             :          * Next sort on catalog ID, in case identical OIDs appear in different
     625                 :             :          * catalogs.  Sort direction is pretty arbitrary here.
     626                 :             :          */
     627         [ #  # ]:           0 :         if (obja->object.classId < objb->object.classId)
     628                 :           0 :                 return -1;
     629         [ #  # ]:           0 :         if (obja->object.classId > objb->object.classId)
     630                 :           0 :                 return 1;
     631                 :             : 
     632                 :             :         /*
     633                 :             :          * Sort on object subId.
     634                 :             :          *
     635                 :             :          * We sort the subId as an unsigned int so that 0 (the whole object) will
     636                 :             :          * come first.
     637                 :             :          */
     638         [ #  # ]:           0 :         if ((unsigned int) obja->object.objectSubId < (unsigned int) objb->object.objectSubId)
     639                 :           0 :                 return -1;
     640         [ #  # ]:           0 :         if ((unsigned int) obja->object.objectSubId > (unsigned int) objb->object.objectSubId)
     641                 :           0 :                 return 1;
     642                 :             : 
     643                 :             :         /*
     644                 :             :          * Last, sort on deptype, in case the same object has multiple dependency
     645                 :             :          * types.  (Note that there's no need to consider objtype, as that's
     646                 :             :          * determined by the catalog OID.)
     647                 :             :          */
     648         [ #  # ]:           0 :         if (obja->deptype < objb->deptype)
     649                 :           0 :                 return -1;
     650         [ #  # ]:           0 :         if (obja->deptype > objb->deptype)
     651                 :           0 :                 return 1;
     652                 :             : 
     653                 :           0 :         return 0;
     654                 :          20 : }
     655                 :             : 
     656                 :             : /*
     657                 :             :  * checkSharedDependencies
     658                 :             :  *
     659                 :             :  * Check whether there are shared dependency entries for a given shared
     660                 :             :  * object; return true if so.
     661                 :             :  *
     662                 :             :  * In addition, return a string containing a newline-separated list of object
     663                 :             :  * descriptions that depend on the shared object, or NULL if none is found.
     664                 :             :  * We actually return two such strings; the "detail" result is suitable for
     665                 :             :  * returning to the client as an errdetail() string, and is limited in size.
     666                 :             :  * The "detail_log" string is potentially much longer, and should be emitted
     667                 :             :  * to the server log only.
     668                 :             :  *
     669                 :             :  * We can find three different kinds of dependencies: dependencies on objects
     670                 :             :  * of the current database; dependencies on shared objects; and dependencies
     671                 :             :  * on objects local to other databases.  We can (and do) provide descriptions
     672                 :             :  * of the two former kinds of objects, but we can't do that for "remote"
     673                 :             :  * objects, so we just provide a count of them.
     674                 :             :  */
     675                 :             : bool
     676                 :         229 : checkSharedDependencies(Oid classId, Oid objectId,
     677                 :             :                                                 char **detail_msg, char **detail_log_msg)
     678                 :             : {
     679                 :         229 :         Relation        sdepRel;
     680                 :         229 :         ScanKeyData key[2];
     681                 :         229 :         SysScanDesc scan;
     682                 :         229 :         HeapTuple       tup;
     683                 :         229 :         int                     numReportedDeps = 0;
     684                 :         229 :         int                     numNotReportedDeps = 0;
     685                 :         229 :         int                     numNotReportedDbs = 0;
     686                 :         229 :         List       *remDeps = NIL;
     687                 :         229 :         ListCell   *cell;
     688                 :         229 :         ObjectAddress object;
     689                 :         229 :         ShDependObjectInfo *objects;
     690                 :         229 :         int                     numobjects;
     691                 :         229 :         int                     allocedobjects;
     692                 :         229 :         StringInfoData descs;
     693                 :         229 :         StringInfoData alldescs;
     694                 :             : 
     695                 :             :         /* This case can be dispatched quickly */
     696         [ +  - ]:         229 :         if (IsPinnedObject(classId, objectId))
     697                 :             :         {
     698                 :           0 :                 object.classId = classId;
     699                 :           0 :                 object.objectId = objectId;
     700                 :           0 :                 object.objectSubId = 0;
     701   [ #  #  #  # ]:           0 :                 ereport(ERROR,
     702                 :             :                                 (errcode(ERRCODE_DEPENDENT_OBJECTS_STILL_EXIST),
     703                 :             :                                  errmsg("cannot drop %s because it is required by the database system",
     704                 :             :                                                 getObjectDescription(&object, false))));
     705                 :           0 :         }
     706                 :             : 
     707                 :             :         /*
     708                 :             :          * We limit the number of dependencies reported to the client to
     709                 :             :          * MAX_REPORTED_DEPS, since client software may not deal well with
     710                 :             :          * enormous error strings.  The server log always gets a full report.
     711                 :             :          *
     712                 :             :          * For stability of regression test results, we sort local and shared
     713                 :             :          * objects by OID before reporting them.  We don't worry about the order
     714                 :             :          * in which other databases are reported, though.
     715                 :             :          */
     716                 :             : #define MAX_REPORTED_DEPS 100
     717                 :             : 
     718                 :         229 :         allocedobjects = 128;           /* arbitrary initial array size */
     719                 :         229 :         objects = (ShDependObjectInfo *)
     720                 :         229 :                 palloc(allocedobjects * sizeof(ShDependObjectInfo));
     721                 :         229 :         numobjects = 0;
     722                 :         229 :         initStringInfo(&descs);
     723                 :         229 :         initStringInfo(&alldescs);
     724                 :             : 
     725                 :         229 :         sdepRel = table_open(SharedDependRelationId, AccessShareLock);
     726                 :             : 
     727                 :         458 :         ScanKeyInit(&key[0],
     728                 :             :                                 Anum_pg_shdepend_refclassid,
     729                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     730                 :         229 :                                 ObjectIdGetDatum(classId));
     731                 :         458 :         ScanKeyInit(&key[1],
     732                 :             :                                 Anum_pg_shdepend_refobjid,
     733                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     734                 :         229 :                                 ObjectIdGetDatum(objectId));
     735                 :             : 
     736                 :         458 :         scan = systable_beginscan(sdepRel, SharedDependReferenceIndexId, true,
     737                 :         229 :                                                           NULL, 2, key);
     738                 :             : 
     739         [ +  + ]:         267 :         while (HeapTupleIsValid(tup = systable_getnext(scan)))
     740                 :             :         {
     741                 :          38 :                 Form_pg_shdepend sdepForm = (Form_pg_shdepend) GETSTRUCT(tup);
     742                 :             : 
     743                 :          38 :                 object.classId = sdepForm->classid;
     744                 :          38 :                 object.objectId = sdepForm->objid;
     745                 :          38 :                 object.objectSubId = sdepForm->objsubid;
     746                 :             : 
     747                 :             :                 /*
     748                 :             :                  * If it's a dependency local to this database or it's a shared
     749                 :             :                  * object, add it to the objects array.
     750                 :             :                  *
     751                 :             :                  * If it's a remote dependency, keep track of it so we can report the
     752                 :             :                  * number of them later.
     753                 :             :                  */
     754   [ +  +  +  - ]:          38 :                 if (sdepForm->dbid == MyDatabaseId ||
     755                 :           5 :                         sdepForm->dbid == InvalidOid)
     756                 :             :                 {
     757         [ +  - ]:          38 :                         if (numobjects >= allocedobjects)
     758                 :             :                         {
     759                 :           0 :                                 allocedobjects *= 2;
     760                 :           0 :                                 objects = (ShDependObjectInfo *)
     761                 :           0 :                                         repalloc(objects,
     762                 :           0 :                                                          allocedobjects * sizeof(ShDependObjectInfo));
     763                 :           0 :                         }
     764                 :          38 :                         objects[numobjects].object = object;
     765                 :          38 :                         objects[numobjects].deptype = sdepForm->deptype;
     766                 :          38 :                         objects[numobjects].objtype = (sdepForm->dbid == MyDatabaseId) ?
     767                 :             :                                 LOCAL_OBJECT : SHARED_OBJECT;
     768                 :          38 :                         numobjects++;
     769                 :          38 :                 }
     770                 :             :                 else
     771                 :             :                 {
     772                 :             :                         /* It's not local nor shared, so it must be remote. */
     773                 :           0 :                         remoteDep  *dep;
     774                 :           0 :                         bool            stored = false;
     775                 :             : 
     776                 :             :                         /*
     777                 :             :                          * XXX this info is kept on a simple List.  Maybe it's not good
     778                 :             :                          * for performance, but using a hash table seems needlessly
     779                 :             :                          * complex.  The expected number of databases is not high anyway,
     780                 :             :                          * I suppose.
     781                 :             :                          */
     782   [ #  #  #  #  :           0 :                         foreach(cell, remDeps)
                   #  # ]
     783                 :             :                         {
     784                 :           0 :                                 dep = lfirst(cell);
     785         [ #  # ]:           0 :                                 if (dep->dbOid == sdepForm->dbid)
     786                 :             :                                 {
     787                 :           0 :                                         dep->count++;
     788                 :           0 :                                         stored = true;
     789                 :           0 :                                         break;
     790                 :             :                                 }
     791                 :           0 :                         }
     792         [ #  # ]:           0 :                         if (!stored)
     793                 :             :                         {
     794                 :           0 :                                 dep = palloc_object(remoteDep);
     795                 :           0 :                                 dep->dbOid = sdepForm->dbid;
     796                 :           0 :                                 dep->count = 1;
     797                 :           0 :                                 remDeps = lappend(remDeps, dep);
     798                 :           0 :                         }
     799                 :           0 :                 }
     800                 :          38 :         }
     801                 :             : 
     802                 :         229 :         systable_endscan(scan);
     803                 :             : 
     804                 :         229 :         table_close(sdepRel, AccessShareLock);
     805                 :             : 
     806                 :             :         /*
     807                 :             :          * Sort and report local and shared objects.
     808                 :             :          */
     809         [ +  + ]:         229 :         if (numobjects > 1)
     810                 :           8 :                 qsort(objects, numobjects,
     811                 :             :                           sizeof(ShDependObjectInfo), shared_dependency_comparator);
     812                 :             : 
     813         [ +  + ]:         267 :         for (int i = 0; i < numobjects; i++)
     814                 :             :         {
     815         [ +  - ]:          38 :                 if (numReportedDeps < MAX_REPORTED_DEPS)
     816                 :             :                 {
     817                 :          38 :                         numReportedDeps++;
     818                 :          38 :                         storeObjectDescription(&descs,
     819                 :          38 :                                                                    objects[i].objtype,
     820                 :          38 :                                                                    &objects[i].object,
     821                 :          38 :                                                                    objects[i].deptype,
     822                 :             :                                                                    0);
     823                 :          38 :                 }
     824                 :             :                 else
     825                 :           0 :                         numNotReportedDeps++;
     826                 :          38 :                 storeObjectDescription(&alldescs,
     827                 :          38 :                                                            objects[i].objtype,
     828                 :          38 :                                                            &objects[i].object,
     829                 :          38 :                                                            objects[i].deptype,
     830                 :             :                                                            0);
     831                 :          38 :         }
     832                 :             : 
     833                 :             :         /*
     834                 :             :          * Summarize dependencies in remote databases.
     835                 :             :          */
     836   [ -  +  #  #  :         229 :         foreach(cell, remDeps)
                   -  + ]
     837                 :             :         {
     838                 :           0 :                 remoteDep  *dep = lfirst(cell);
     839                 :             : 
     840                 :           0 :                 object.classId = DatabaseRelationId;
     841                 :           0 :                 object.objectId = dep->dbOid;
     842                 :           0 :                 object.objectSubId = 0;
     843                 :             : 
     844         [ #  # ]:           0 :                 if (numReportedDeps < MAX_REPORTED_DEPS)
     845                 :             :                 {
     846                 :           0 :                         numReportedDeps++;
     847                 :           0 :                         storeObjectDescription(&descs, REMOTE_OBJECT, &object,
     848                 :           0 :                                                                    SHARED_DEPENDENCY_INVALID, dep->count);
     849                 :           0 :                 }
     850                 :             :                 else
     851                 :           0 :                         numNotReportedDbs++;
     852                 :           0 :                 storeObjectDescription(&alldescs, REMOTE_OBJECT, &object,
     853                 :           0 :                                                            SHARED_DEPENDENCY_INVALID, dep->count);
     854                 :           0 :         }
     855                 :             : 
     856                 :         229 :         pfree(objects);
     857                 :         229 :         list_free_deep(remDeps);
     858                 :             : 
     859         [ +  + ]:         229 :         if (descs.len == 0)
     860                 :             :         {
     861                 :         209 :                 pfree(descs.data);
     862                 :         209 :                 pfree(alldescs.data);
     863                 :         209 :                 *detail_msg = *detail_log_msg = NULL;
     864                 :         209 :                 return false;
     865                 :             :         }
     866                 :             : 
     867         [ +  - ]:          20 :         if (numNotReportedDeps > 0)
     868                 :           0 :                 appendStringInfo(&descs, ngettext("\nand %d other object "
     869                 :             :                                                                                   "(see server log for list)",
     870                 :             :                                                                                   "\nand %d other objects "
     871                 :             :                                                                                   "(see server log for list)",
     872                 :           0 :                                                                                   numNotReportedDeps),
     873                 :           0 :                                                  numNotReportedDeps);
     874         [ +  - ]:          20 :         if (numNotReportedDbs > 0)
     875                 :           0 :                 appendStringInfo(&descs, ngettext("\nand objects in %d other database "
     876                 :             :                                                                                   "(see server log for list)",
     877                 :             :                                                                                   "\nand objects in %d other databases "
     878                 :             :                                                                                   "(see server log for list)",
     879                 :           0 :                                                                                   numNotReportedDbs),
     880                 :           0 :                                                  numNotReportedDbs);
     881                 :             : 
     882                 :          20 :         *detail_msg = descs.data;
     883                 :          20 :         *detail_log_msg = alldescs.data;
     884                 :          20 :         return true;
     885                 :         229 : }
     886                 :             : 
     887                 :             : 
     888                 :             : /*
     889                 :             :  * copyTemplateDependencies
     890                 :             :  *
     891                 :             :  * Routine to create the initial shared dependencies of a new database.
     892                 :             :  * We simply copy the dependencies from the template database.
     893                 :             :  */
     894                 :             : void
     895                 :           5 : copyTemplateDependencies(Oid templateDbId, Oid newDbId)
     896                 :             : {
     897                 :           5 :         Relation        sdepRel;
     898                 :           5 :         TupleDesc       sdepDesc;
     899                 :           5 :         ScanKeyData key[1];
     900                 :           5 :         SysScanDesc scan;
     901                 :           5 :         HeapTuple       tup;
     902                 :           5 :         CatalogIndexState indstate;
     903                 :           5 :         TupleTableSlot **slot;
     904                 :           5 :         int                     max_slots,
     905                 :             :                                 slot_init_count,
     906                 :             :                                 slot_stored_count;
     907                 :             : 
     908                 :           5 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
     909                 :           5 :         sdepDesc = RelationGetDescr(sdepRel);
     910                 :             : 
     911                 :             :         /*
     912                 :             :          * Allocate the slots to use, but delay costly initialization until we
     913                 :             :          * know that they will be used.
     914                 :             :          */
     915                 :           5 :         max_slots = MAX_CATALOG_MULTI_INSERT_BYTES / sizeof(FormData_pg_shdepend);
     916                 :           5 :         slot = palloc_array(TupleTableSlot *, max_slots);
     917                 :             : 
     918                 :           5 :         indstate = CatalogOpenIndexes(sdepRel);
     919                 :             : 
     920                 :             :         /* Scan all entries with dbid = templateDbId */
     921                 :          10 :         ScanKeyInit(&key[0],
     922                 :             :                                 Anum_pg_shdepend_dbid,
     923                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
     924                 :           5 :                                 ObjectIdGetDatum(templateDbId));
     925                 :             : 
     926                 :          10 :         scan = systable_beginscan(sdepRel, SharedDependDependerIndexId, true,
     927                 :           5 :                                                           NULL, 1, key);
     928                 :             : 
     929                 :             :         /* number of slots currently storing tuples */
     930                 :           5 :         slot_stored_count = 0;
     931                 :             :         /* number of slots currently initialized */
     932                 :           5 :         slot_init_count = 0;
     933                 :             : 
     934                 :             :         /*
     935                 :             :          * Copy the entries of the original database, changing the database Id to
     936                 :             :          * that of the new database.  Note that because we are not copying rows
     937                 :             :          * with dbId == 0 (ie, rows describing dependent shared objects) we won't
     938                 :             :          * copy the ownership dependency of the template database itself; this is
     939                 :             :          * what we want.
     940                 :             :          */
     941         [ -  + ]:           5 :         while (HeapTupleIsValid(tup = systable_getnext(scan)))
     942                 :             :         {
     943                 :           0 :                 Form_pg_shdepend shdep;
     944                 :             : 
     945         [ #  # ]:           0 :                 if (slot_init_count < max_slots)
     946                 :             :                 {
     947                 :           0 :                         slot[slot_stored_count] = MakeSingleTupleTableSlot(sdepDesc, &TTSOpsHeapTuple);
     948                 :           0 :                         slot_init_count++;
     949                 :           0 :                 }
     950                 :             : 
     951                 :           0 :                 ExecClearTuple(slot[slot_stored_count]);
     952                 :             : 
     953                 :           0 :                 memset(slot[slot_stored_count]->tts_isnull, false,
     954                 :             :                            slot[slot_stored_count]->tts_tupleDescriptor->natts * sizeof(bool));
     955                 :             : 
     956                 :           0 :                 shdep = (Form_pg_shdepend) GETSTRUCT(tup);
     957                 :             : 
     958                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_dbid - 1] = ObjectIdGetDatum(newDbId);
     959                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_classid - 1] = ObjectIdGetDatum(shdep->classid);
     960                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_objid - 1] = ObjectIdGetDatum(shdep->objid);
     961                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_objsubid - 1] = Int32GetDatum(shdep->objsubid);
     962                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_refclassid - 1] = ObjectIdGetDatum(shdep->refclassid);
     963                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_refobjid - 1] = ObjectIdGetDatum(shdep->refobjid);
     964                 :           0 :                 slot[slot_stored_count]->tts_values[Anum_pg_shdepend_deptype - 1] = CharGetDatum(shdep->deptype);
     965                 :             : 
     966                 :           0 :                 ExecStoreVirtualTuple(slot[slot_stored_count]);
     967                 :           0 :                 slot_stored_count++;
     968                 :             : 
     969                 :             :                 /* If slots are full, insert a batch of tuples */
     970         [ #  # ]:           0 :                 if (slot_stored_count == max_slots)
     971                 :             :                 {
     972                 :           0 :                         CatalogTuplesMultiInsertWithInfo(sdepRel, slot, slot_stored_count, indstate);
     973                 :           0 :                         slot_stored_count = 0;
     974                 :           0 :                 }
     975                 :           0 :         }
     976                 :             : 
     977                 :             :         /* Insert any tuples left in the buffer */
     978         [ +  - ]:           5 :         if (slot_stored_count > 0)
     979                 :           0 :                 CatalogTuplesMultiInsertWithInfo(sdepRel, slot, slot_stored_count, indstate);
     980                 :             : 
     981                 :           5 :         systable_endscan(scan);
     982                 :             : 
     983                 :           5 :         CatalogCloseIndexes(indstate);
     984                 :           5 :         table_close(sdepRel, RowExclusiveLock);
     985                 :             : 
     986                 :             :         /* Drop only the number of slots used */
     987         [ +  - ]:           5 :         for (int i = 0; i < slot_init_count; i++)
     988                 :           0 :                 ExecDropSingleTupleTableSlot(slot[i]);
     989                 :           5 :         pfree(slot);
     990                 :           5 : }
     991                 :             : 
     992                 :             : /*
     993                 :             :  * dropDatabaseDependencies
     994                 :             :  *
     995                 :             :  * Delete pg_shdepend entries corresponding to a database that's being
     996                 :             :  * dropped.
     997                 :             :  */
     998                 :             : void
     999                 :           1 : dropDatabaseDependencies(Oid databaseId)
    1000                 :             : {
    1001                 :           1 :         Relation        sdepRel;
    1002                 :           1 :         ScanKeyData key[1];
    1003                 :           1 :         SysScanDesc scan;
    1004                 :           1 :         HeapTuple       tup;
    1005                 :             : 
    1006                 :           1 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
    1007                 :             : 
    1008                 :             :         /*
    1009                 :             :          * First, delete all the entries that have the database Oid in the dbid
    1010                 :             :          * field.
    1011                 :             :          */
    1012                 :           2 :         ScanKeyInit(&key[0],
    1013                 :             :                                 Anum_pg_shdepend_dbid,
    1014                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
    1015                 :           1 :                                 ObjectIdGetDatum(databaseId));
    1016                 :             :         /* We leave the other index fields unspecified */
    1017                 :             : 
    1018                 :           2 :         scan = systable_beginscan(sdepRel, SharedDependDependerIndexId, true,
    1019                 :           1 :                                                           NULL, 1, key);
    1020                 :             : 
    1021         [ -  + ]:           1 :         while (HeapTupleIsValid(tup = systable_getnext(scan)))
    1022                 :             :         {
    1023                 :           0 :                 CatalogTupleDelete(sdepRel, &tup->t_self);
    1024                 :             :         }
    1025                 :             : 
    1026                 :           1 :         systable_endscan(scan);
    1027                 :             : 
    1028                 :             :         /* Now delete all entries corresponding to the database itself */
    1029                 :           1 :         shdepDropDependency(sdepRel, DatabaseRelationId, databaseId, 0, true,
    1030                 :             :                                                 InvalidOid, InvalidOid,
    1031                 :             :                                                 SHARED_DEPENDENCY_INVALID);
    1032                 :             : 
    1033                 :           1 :         table_close(sdepRel, RowExclusiveLock);
    1034                 :           1 : }
    1035                 :             : 
    1036                 :             : /*
    1037                 :             :  * deleteSharedDependencyRecordsFor
    1038                 :             :  *
    1039                 :             :  * Delete all pg_shdepend entries corresponding to an object that's being
    1040                 :             :  * dropped or modified.  The object is assumed to be either a shared object
    1041                 :             :  * or local to the current database (the classId tells us which).
    1042                 :             :  *
    1043                 :             :  * If objectSubId is zero, we are deleting a whole object, so get rid of
    1044                 :             :  * pg_shdepend entries for subobjects as well.
    1045                 :             :  */
    1046                 :             : void
    1047                 :       24980 : deleteSharedDependencyRecordsFor(Oid classId, Oid objectId, int32 objectSubId)
    1048                 :             : {
    1049                 :       24980 :         Relation        sdepRel;
    1050                 :             : 
    1051                 :       24980 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
    1052                 :             : 
    1053                 :       49960 :         shdepDropDependency(sdepRel, classId, objectId, objectSubId,
    1054                 :       24980 :                                                 (objectSubId == 0),
    1055                 :             :                                                 InvalidOid, InvalidOid,
    1056                 :             :                                                 SHARED_DEPENDENCY_INVALID);
    1057                 :             : 
    1058                 :       24980 :         table_close(sdepRel, RowExclusiveLock);
    1059                 :       24980 : }
    1060                 :             : 
    1061                 :             : /*
    1062                 :             :  * shdepAddDependency
    1063                 :             :  *              Internal workhorse for inserting into pg_shdepend
    1064                 :             :  *
    1065                 :             :  * sdepRel must be the pg_shdepend relation, already opened and suitably
    1066                 :             :  * locked.
    1067                 :             :  */
    1068                 :             : static void
    1069                 :        1023 : shdepAddDependency(Relation sdepRel,
    1070                 :             :                                    Oid classId, Oid objectId, int32 objsubId,
    1071                 :             :                                    Oid refclassId, Oid refobjId,
    1072                 :             :                                    SharedDependencyType deptype)
    1073                 :             : {
    1074                 :        1023 :         HeapTuple       tup;
    1075                 :        1023 :         Datum           values[Natts_pg_shdepend];
    1076                 :        1023 :         bool            nulls[Natts_pg_shdepend];
    1077                 :             : 
    1078                 :             :         /*
    1079                 :             :          * Make sure the object doesn't go away while we record the dependency on
    1080                 :             :          * it.  DROP routines should lock the object exclusively before they check
    1081                 :             :          * shared dependencies.
    1082                 :             :          */
    1083                 :        1023 :         shdepLockAndCheckObject(refclassId, refobjId);
    1084                 :             : 
    1085                 :        1023 :         memset(nulls, false, sizeof(nulls));
    1086                 :             : 
    1087                 :             :         /*
    1088                 :             :          * Form the new tuple and record the dependency.
    1089                 :             :          */
    1090                 :        1023 :         values[Anum_pg_shdepend_dbid - 1] = ObjectIdGetDatum(classIdGetDbId(classId));
    1091                 :        1023 :         values[Anum_pg_shdepend_classid - 1] = ObjectIdGetDatum(classId);
    1092                 :        1023 :         values[Anum_pg_shdepend_objid - 1] = ObjectIdGetDatum(objectId);
    1093                 :        1023 :         values[Anum_pg_shdepend_objsubid - 1] = Int32GetDatum(objsubId);
    1094                 :             : 
    1095                 :        1023 :         values[Anum_pg_shdepend_refclassid - 1] = ObjectIdGetDatum(refclassId);
    1096                 :        1023 :         values[Anum_pg_shdepend_refobjid - 1] = ObjectIdGetDatum(refobjId);
    1097                 :        1023 :         values[Anum_pg_shdepend_deptype - 1] = CharGetDatum(deptype);
    1098                 :             : 
    1099                 :        1023 :         tup = heap_form_tuple(sdepRel->rd_att, values, nulls);
    1100                 :             : 
    1101                 :        1023 :         CatalogTupleInsert(sdepRel, tup);
    1102                 :             : 
    1103                 :             :         /* clean up */
    1104                 :        1023 :         heap_freetuple(tup);
    1105                 :        1023 : }
    1106                 :             : 
    1107                 :             : /*
    1108                 :             :  * shdepDropDependency
    1109                 :             :  *              Internal workhorse for deleting entries from pg_shdepend.
    1110                 :             :  *
    1111                 :             :  * We drop entries having the following properties:
    1112                 :             :  *      dependent object is the one identified by classId/objectId/objsubId
    1113                 :             :  *      if refclassId isn't InvalidOid, it must match the entry's refclassid
    1114                 :             :  *      if refobjId isn't InvalidOid, it must match the entry's refobjid
    1115                 :             :  *      if deptype isn't SHARED_DEPENDENCY_INVALID, it must match entry's deptype
    1116                 :             :  *
    1117                 :             :  * If drop_subobjects is true, we ignore objsubId and consider all entries
    1118                 :             :  * matching classId/objectId.
    1119                 :             :  *
    1120                 :             :  * sdepRel must be the pg_shdepend relation, already opened and suitably
    1121                 :             :  * locked.
    1122                 :             :  */
    1123                 :             : static void
    1124                 :       25180 : shdepDropDependency(Relation sdepRel,
    1125                 :             :                                         Oid classId, Oid objectId, int32 objsubId,
    1126                 :             :                                         bool drop_subobjects,
    1127                 :             :                                         Oid refclassId, Oid refobjId,
    1128                 :             :                                         SharedDependencyType deptype)
    1129                 :             : {
    1130                 :       25180 :         ScanKeyData key[4];
    1131                 :       25180 :         int                     nkeys;
    1132                 :       25180 :         SysScanDesc scan;
    1133                 :       25180 :         HeapTuple       tup;
    1134                 :             : 
    1135                 :             :         /* Scan for entries matching the dependent object */
    1136                 :       50360 :         ScanKeyInit(&key[0],
    1137                 :             :                                 Anum_pg_shdepend_dbid,
    1138                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
    1139                 :       25180 :                                 ObjectIdGetDatum(classIdGetDbId(classId)));
    1140                 :       50360 :         ScanKeyInit(&key[1],
    1141                 :             :                                 Anum_pg_shdepend_classid,
    1142                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
    1143                 :       25180 :                                 ObjectIdGetDatum(classId));
    1144                 :       50360 :         ScanKeyInit(&key[2],
    1145                 :             :                                 Anum_pg_shdepend_objid,
    1146                 :             :                                 BTEqualStrategyNumber, F_OIDEQ,
    1147                 :       25180 :                                 ObjectIdGetDatum(objectId));
    1148         [ +  + ]:       25180 :         if (drop_subobjects)
    1149                 :       24780 :                 nkeys = 3;
    1150                 :             :         else
    1151                 :             :         {
    1152                 :         800 :                 ScanKeyInit(&key[3],
    1153                 :             :                                         Anum_pg_shdepend_objsubid,
    1154                 :             :                                         BTEqualStrategyNumber, F_INT4EQ,
    1155                 :         400 :                                         Int32GetDatum(objsubId));
    1156                 :         400 :                 nkeys = 4;
    1157                 :             :         }
    1158                 :             : 
    1159                 :       50360 :         scan = systable_beginscan(sdepRel, SharedDependDependerIndexId, true,
    1160                 :       25180 :                                                           NULL, nkeys, key);
    1161                 :             : 
    1162         [ +  + ]:       26337 :         while (HeapTupleIsValid(tup = systable_getnext(scan)))
    1163                 :             :         {
    1164                 :        1157 :                 Form_pg_shdepend shdepForm = (Form_pg_shdepend) GETSTRUCT(tup);
    1165                 :             : 
    1166                 :             :                 /* Filter entries according to additional parameters */
    1167   [ +  +  +  - ]:        1157 :                 if (OidIsValid(refclassId) && shdepForm->refclassid != refclassId)
    1168                 :           0 :                         continue;
    1169   [ +  +  +  + ]:        1157 :                 if (OidIsValid(refobjId) && shdepForm->refobjid != refobjId)
    1170                 :         118 :                         continue;
    1171   [ +  +  +  - ]:        1039 :                 if (deptype != SHARED_DEPENDENCY_INVALID &&
    1172                 :          95 :                         shdepForm->deptype != deptype)
    1173                 :           0 :                         continue;
    1174                 :             : 
    1175                 :             :                 /* OK, delete it */
    1176                 :        1039 :                 CatalogTupleDelete(sdepRel, &tup->t_self);
    1177      [ -  +  + ]:        1157 :         }
    1178                 :             : 
    1179                 :       25180 :         systable_endscan(scan);
    1180                 :       25180 : }
    1181                 :             : 
    1182                 :             : /*
    1183                 :             :  * classIdGetDbId
    1184                 :             :  *
    1185                 :             :  * Get the database Id that should be used in pg_shdepend, given the OID
    1186                 :             :  * of the catalog containing the object.  For shared objects, it's 0
    1187                 :             :  * (InvalidOid); for all other objects, it's the current database Id.
    1188                 :             :  */
    1189                 :             : static Oid
    1190                 :       26307 : classIdGetDbId(Oid classId)
    1191                 :             : {
    1192                 :       26307 :         Oid                     dbId;
    1193                 :             : 
    1194         [ +  + ]:       26307 :         if (IsSharedRelation(classId))
    1195                 :         189 :                 dbId = InvalidOid;
    1196                 :             :         else
    1197                 :       26118 :                 dbId = MyDatabaseId;
    1198                 :             : 
    1199                 :       52614 :         return dbId;
    1200                 :       26307 : }
    1201                 :             : 
    1202                 :             : /*
    1203                 :             :  * shdepLockAndCheckObject
    1204                 :             :  *
    1205                 :             :  * Lock the object that we are about to record a dependency on.
    1206                 :             :  * After it's locked, verify that it hasn't been dropped while we
    1207                 :             :  * weren't looking.  If the object has been dropped, this function
    1208                 :             :  * does not return!
    1209                 :             :  */
    1210                 :             : void
    1211                 :        1140 : shdepLockAndCheckObject(Oid classId, Oid objectId)
    1212                 :             : {
    1213                 :             :         /* AccessShareLock should be OK, since we are not modifying the object */
    1214                 :        1140 :         LockSharedObject(classId, objectId, 0, AccessShareLock);
    1215                 :             : 
    1216   [ +  +  +  - ]:        1140 :         switch (classId)
    1217                 :             :         {
    1218                 :             :                 case AuthIdRelationId:
    1219         [ +  - ]:        1105 :                         if (!SearchSysCacheExists1(AUTHOID, ObjectIdGetDatum(objectId)))
    1220   [ #  #  #  # ]:           0 :                                 ereport(ERROR,
    1221                 :             :                                                 (errcode(ERRCODE_UNDEFINED_OBJECT),
    1222                 :             :                                                  errmsg("role %u was concurrently dropped",
    1223                 :             :                                                                 objectId)));
    1224                 :        1105 :                         break;
    1225                 :             : 
    1226                 :             :                 case TableSpaceRelationId:
    1227                 :             :                         {
    1228                 :             :                                 /* For lack of a syscache on pg_tablespace, do this: */
    1229                 :          23 :                                 char       *tablespace = get_tablespace_name(objectId);
    1230                 :             : 
    1231         [ +  - ]:          23 :                                 if (tablespace == NULL)
    1232   [ #  #  #  # ]:           0 :                                         ereport(ERROR,
    1233                 :             :                                                         (errcode(ERRCODE_UNDEFINED_OBJECT),
    1234                 :             :                                                          errmsg("tablespace %u was concurrently dropped",
    1235                 :             :                                                                         objectId)));
    1236                 :          23 :                                 pfree(tablespace);
    1237                 :             :                                 break;
    1238                 :          23 :                         }
    1239                 :             : 
    1240                 :             :                 case DatabaseRelationId:
    1241                 :             :                         {
    1242                 :             :                                 /* For lack of a syscache on pg_database, do this: */
    1243                 :          12 :                                 char       *database = get_database_name(objectId);
    1244                 :             : 
    1245         [ +  - ]:          12 :                                 if (database == NULL)
    1246   [ #  #  #  # ]:           0 :                                         ereport(ERROR,
    1247                 :             :                                                         (errcode(ERRCODE_UNDEFINED_OBJECT),
    1248                 :             :                                                          errmsg("database %u was concurrently dropped",
    1249                 :             :                                                                         objectId)));
    1250                 :          12 :                                 pfree(database);
    1251                 :             :                                 break;
    1252                 :          12 :                         }
    1253                 :             : 
    1254                 :             : 
    1255                 :             :                 default:
    1256   [ #  #  #  # ]:           0 :                         elog(ERROR, "unrecognized shared classId: %u", classId);
    1257                 :           0 :         }
    1258                 :        1140 : }
    1259                 :             : 
    1260                 :             : 
    1261                 :             : /*
    1262                 :             :  * storeObjectDescription
    1263                 :             :  *              Append the description of a dependent object to "descs"
    1264                 :             :  *
    1265                 :             :  * While searching for dependencies of a shared object, we stash the
    1266                 :             :  * descriptions of dependent objects we find in a single string, which we
    1267                 :             :  * later pass to ereport() in the DETAIL field when somebody attempts to
    1268                 :             :  * drop a referenced shared object.
    1269                 :             :  *
    1270                 :             :  * When type is LOCAL_OBJECT or SHARED_OBJECT, we expect object to be the
    1271                 :             :  * dependent object, deptype is the dependency type, and count is not used.
    1272                 :             :  * When type is REMOTE_OBJECT, we expect object to be the database object,
    1273                 :             :  * and count to be nonzero; deptype is not used in this case.
    1274                 :             :  */
    1275                 :             : static void
    1276                 :          76 : storeObjectDescription(StringInfo descs,
    1277                 :             :                                            SharedDependencyObjectType type,
    1278                 :             :                                            ObjectAddress *object,
    1279                 :             :                                            SharedDependencyType deptype,
    1280                 :             :                                            int count)
    1281                 :             : {
    1282                 :          76 :         char       *objdesc = getObjectDescription(object, false);
    1283                 :             : 
    1284                 :             :         /*
    1285                 :             :          * An object being dropped concurrently doesn't need to be reported.
    1286                 :             :          */
    1287         [ +  - ]:          76 :         if (objdesc == NULL)
    1288                 :           0 :                 return;
    1289                 :             : 
    1290                 :             :         /* separate entries with a newline */
    1291         [ +  + ]:          76 :         if (descs->len != 0)
    1292                 :          36 :                 appendStringInfoChar(descs, '\n');
    1293                 :             : 
    1294      [ +  -  - ]:          76 :         switch (type)
    1295                 :             :         {
    1296                 :             :                 case LOCAL_OBJECT:
    1297                 :             :                 case SHARED_OBJECT:
    1298         [ +  + ]:          76 :                         if (deptype == SHARED_DEPENDENCY_OWNER)
    1299                 :          40 :                                 appendStringInfo(descs, _("owner of %s"), objdesc);
    1300         [ +  + ]:          36 :                         else if (deptype == SHARED_DEPENDENCY_ACL)
    1301                 :          30 :                                 appendStringInfo(descs, _("privileges for %s"), objdesc);
    1302         [ -  + ]:           6 :                         else if (deptype == SHARED_DEPENDENCY_INITACL)
    1303                 :           0 :                                 appendStringInfo(descs, _("initial privileges for %s"), objdesc);
    1304         [ +  + ]:           6 :                         else if (deptype == SHARED_DEPENDENCY_POLICY)
    1305                 :           4 :                                 appendStringInfo(descs, _("target of %s"), objdesc);
    1306         [ +  - ]:           2 :                         else if (deptype == SHARED_DEPENDENCY_TABLESPACE)
    1307                 :           2 :                                 appendStringInfo(descs, _("tablespace for %s"), objdesc);
    1308                 :             :                         else
    1309   [ #  #  #  # ]:           0 :                                 elog(ERROR, "unrecognized dependency type: %d",
    1310                 :             :                                          (int) deptype);
    1311                 :          76 :                         break;
    1312                 :             : 
    1313                 :             :                 case REMOTE_OBJECT:
    1314                 :             :                         /* translator: %s will always be "database %s" */
    1315                 :           0 :                         appendStringInfo(descs, ngettext("%d object in %s",
    1316                 :             :                                                                                          "%d objects in %s",
    1317                 :           0 :                                                                                          count),
    1318                 :           0 :                                                          count, objdesc);
    1319                 :           0 :                         break;
    1320                 :             : 
    1321                 :             :                 default:
    1322   [ #  #  #  # ]:           0 :                         elog(ERROR, "unrecognized object type: %d", type);
    1323                 :           0 :         }
    1324                 :             : 
    1325                 :          76 :         pfree(objdesc);
    1326         [ -  + ]:          76 : }
    1327                 :             : 
    1328                 :             : 
    1329                 :             : /*
    1330                 :             :  * shdepDropOwned
    1331                 :             :  *
    1332                 :             :  * Drop the objects owned by any one of the given RoleIds.  If a role has
    1333                 :             :  * access to an object, the grant will be removed as well (but the object
    1334                 :             :  * will not, of course).
    1335                 :             :  *
    1336                 :             :  * We can revoke grants immediately while doing the scan, but drops are
    1337                 :             :  * saved up and done all at once with performMultipleDeletions.  This
    1338                 :             :  * is necessary so that we don't get failures from trying to delete
    1339                 :             :  * interdependent objects in the wrong order.
    1340                 :             :  */
    1341                 :             : void
    1342                 :          18 : shdepDropOwned(List *roleids, DropBehavior behavior)
    1343                 :             : {
    1344                 :          18 :         Relation        sdepRel;
    1345                 :          18 :         ListCell   *cell;
    1346                 :          18 :         ObjectAddresses *deleteobjs;
    1347                 :             : 
    1348                 :          18 :         deleteobjs = new_object_addresses();
    1349                 :             : 
    1350                 :             :         /*
    1351                 :             :          * We don't need this strong a lock here, but we'll call routines that
    1352                 :             :          * acquire RowExclusiveLock.  Better get that right now to avoid potential
    1353                 :             :          * deadlock failures.
    1354                 :             :          */
    1355                 :          18 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
    1356                 :             : 
    1357                 :             :         /*
    1358                 :             :          * For each role, find the dependent objects and drop them using the
    1359                 :             :          * regular (non-shared) dependency management.
    1360                 :             :          */
    1361   [ +  -  +  +  :          38 :         foreach(cell, roleids)
                   +  + ]
    1362                 :             :         {
    1363                 :          20 :                 Oid                     roleid = lfirst_oid(cell);
    1364                 :          20 :                 ScanKeyData key[2];
    1365                 :          20 :                 SysScanDesc scan;
    1366                 :          20 :                 HeapTuple       tuple;
    1367                 :             : 
    1368                 :             :                 /* Doesn't work for pinned objects */
    1369         [ +  - ]:          20 :                 if (IsPinnedObject(AuthIdRelationId, roleid))
    1370                 :             :                 {
    1371                 :           0 :                         ObjectAddress obj;
    1372                 :             : 
    1373                 :           0 :                         obj.classId = AuthIdRelationId;
    1374                 :           0 :                         obj.objectId = roleid;
    1375                 :           0 :                         obj.objectSubId = 0;
    1376                 :             : 
    1377   [ #  #  #  # ]:           0 :                         ereport(ERROR,
    1378                 :             :                                         (errcode(ERRCODE_DEPENDENT_OBJECTS_STILL_EXIST),
    1379                 :             :                                          errmsg("cannot drop objects owned by %s because they are "
    1380                 :             :                                                         "required by the database system",
    1381                 :             :                                                         getObjectDescription(&obj, false))));
    1382                 :           0 :                 }
    1383                 :             : 
    1384                 :          40 :                 ScanKeyInit(&key[0],
    1385                 :             :                                         Anum_pg_shdepend_refclassid,
    1386                 :             :                                         BTEqualStrategyNumber, F_OIDEQ,
    1387                 :          20 :                                         ObjectIdGetDatum(AuthIdRelationId));
    1388                 :          40 :                 ScanKeyInit(&key[1],
    1389                 :             :                                         Anum_pg_shdepend_refobjid,
    1390                 :             :                                         BTEqualStrategyNumber, F_OIDEQ,
    1391                 :          20 :                                         ObjectIdGetDatum(roleid));
    1392                 :             : 
    1393                 :          40 :                 scan = systable_beginscan(sdepRel, SharedDependReferenceIndexId, true,
    1394                 :          20 :                                                                   NULL, 2, key);
    1395                 :             : 
    1396         [ +  + ]:         104 :                 while ((tuple = systable_getnext(scan)) != NULL)
    1397                 :             :                 {
    1398                 :          84 :                         Form_pg_shdepend sdepForm = (Form_pg_shdepend) GETSTRUCT(tuple);
    1399                 :          84 :                         ObjectAddress obj;
    1400                 :             : 
    1401                 :             :                         /*
    1402                 :             :                          * We only operate on shared objects and objects in the current
    1403                 :             :                          * database
    1404                 :             :                          */
    1405   [ +  +  +  - ]:          84 :                         if (sdepForm->dbid != MyDatabaseId &&
    1406                 :           2 :                                 sdepForm->dbid != InvalidOid)
    1407                 :           0 :                                 continue;
    1408                 :             : 
    1409   [ -  +  +  +  :          84 :                         switch (sdepForm->deptype)
                   -  - ]
    1410                 :             :                         {
    1411                 :             :                                         /* Shouldn't happen */
    1412                 :             :                                 case SHARED_DEPENDENCY_INVALID:
    1413   [ #  #  #  # ]:           0 :                                         elog(ERROR, "unexpected dependency type");
    1414                 :           0 :                                         break;
    1415                 :             :                                 case SHARED_DEPENDENCY_POLICY:
    1416                 :             : 
    1417                 :             :                                         /*
    1418                 :             :                                          * Try to remove role from policy; if unable to, remove
    1419                 :             :                                          * policy.
    1420                 :             :                                          */
    1421   [ +  +  +  + ]:          14 :                                         if (!RemoveRoleFromObjectPolicy(roleid,
    1422                 :           7 :                                                                                                         sdepForm->classid,
    1423                 :           7 :                                                                                                         sdepForm->objid))
    1424                 :             :                                         {
    1425                 :           3 :                                                 obj.classId = sdepForm->classid;
    1426                 :           3 :                                                 obj.objectId = sdepForm->objid;
    1427                 :           3 :                                                 obj.objectSubId = sdepForm->objsubid;
    1428                 :             : 
    1429                 :             :                                                 /*
    1430                 :             :                                                  * Acquire lock on object, then verify this dependency
    1431                 :             :                                                  * is still relevant.  If not, the object might have
    1432                 :             :                                                  * been dropped or the policy modified.  Ignore the
    1433                 :             :                                                  * object in that case.
    1434                 :             :                                                  */
    1435                 :           3 :                                                 AcquireDeletionLock(&obj, 0);
    1436         [ +  - ]:           3 :                                                 if (!systable_recheck_tuple(scan, tuple))
    1437                 :             :                                                 {
    1438                 :           0 :                                                         ReleaseDeletionLock(&obj);
    1439                 :           0 :                                                         break;
    1440                 :             :                                                 }
    1441                 :           3 :                                                 add_exact_object_address(&obj, deleteobjs);
    1442                 :           3 :                                         }
    1443                 :           7 :                                         break;
    1444                 :             :                                 case SHARED_DEPENDENCY_ACL:
    1445                 :             : 
    1446                 :             :                                         /*
    1447                 :             :                                          * Dependencies on role grants are recorded using
    1448                 :             :                                          * SHARED_DEPENDENCY_ACL, but unlike a regular ACL list
    1449                 :             :                                          * which stores all permissions for a particular object in
    1450                 :             :                                          * a single ACL array, there's a separate catalog row for
    1451                 :             :                                          * each grant - so removing the grant just means removing
    1452                 :             :                                          * the entire row.
    1453                 :             :                                          */
    1454         [ +  + ]:          30 :                                         if (sdepForm->classid != AuthMemRelationId)
    1455                 :             :                                         {
    1456                 :          58 :                                                 RemoveRoleFromObjectACL(roleid,
    1457                 :          29 :                                                                                                 sdepForm->classid,
    1458                 :          29 :                                                                                                 sdepForm->objid);
    1459                 :          29 :                                                 break;
    1460                 :             :                                         }
    1461                 :             :                                         /* FALLTHROUGH */
    1462                 :             : 
    1463                 :             :                                 case SHARED_DEPENDENCY_OWNER:
    1464                 :             : 
    1465                 :             :                                         /*
    1466                 :             :                                          * Save it for deletion below, if it's a local object or a
    1467                 :             :                                          * role grant. Other shared objects, such as databases,
    1468                 :             :                                          * should not be removed here.
    1469                 :             :                                          */
    1470   [ +  +  +  - ]:          48 :                                         if (sdepForm->dbid == MyDatabaseId ||
    1471                 :           1 :                                                 sdepForm->classid == AuthMemRelationId)
    1472                 :             :                                         {
    1473                 :          48 :                                                 obj.classId = sdepForm->classid;
    1474                 :          48 :                                                 obj.objectId = sdepForm->objid;
    1475                 :          48 :                                                 obj.objectSubId = sdepForm->objsubid;
    1476                 :             :                                                 /* as above */
    1477                 :          48 :                                                 AcquireDeletionLock(&obj, 0);
    1478         [ +  - ]:          48 :                                                 if (!systable_recheck_tuple(scan, tuple))
    1479                 :             :                                                 {
    1480                 :           0 :                                                         ReleaseDeletionLock(&obj);
    1481                 :           0 :                                                         break;
    1482                 :             :                                                 }
    1483                 :          48 :                                                 add_exact_object_address(&obj, deleteobjs);
    1484                 :          48 :                                         }
    1485                 :          48 :                                         break;
    1486                 :             :                                 case SHARED_DEPENDENCY_INITACL:
    1487                 :             : 
    1488                 :             :                                         /*
    1489                 :             :                                          * Any mentions of the role that remain in pg_init_privs
    1490                 :             :                                          * entries are just dropped.  This is the same policy as
    1491                 :             :                                          * we apply to regular ACLs.
    1492                 :             :                                          */
    1493                 :             : 
    1494                 :             :                                         /* Shouldn't see a role grant here */
    1495         [ #  # ]:           0 :                                         Assert(sdepForm->classid != AuthMemRelationId);
    1496                 :           0 :                                         RemoveRoleFromInitPriv(roleid,
    1497                 :           0 :                                                                                    sdepForm->classid,
    1498                 :           0 :                                                                                    sdepForm->objid,
    1499                 :           0 :                                                                                    sdepForm->objsubid);
    1500                 :           0 :                                         break;
    1501                 :             :                         }
    1502      [ -  -  + ]:          84 :                 }
    1503                 :             : 
    1504                 :          20 :                 systable_endscan(scan);
    1505                 :          20 :         }
    1506                 :             : 
    1507                 :             :         /*
    1508                 :             :          * For stability of deletion-report ordering, sort the objects into
    1509                 :             :          * approximate reverse creation order before deletion.  (This might also
    1510                 :             :          * make the deletion go a bit faster, since there's less chance of having
    1511                 :             :          * to rearrange the objects due to dependencies.)
    1512                 :             :          */
    1513                 :          18 :         sort_object_addresses(deleteobjs);
    1514                 :             : 
    1515                 :             :         /* the dependency mechanism does the actual work */
    1516                 :          18 :         performMultipleDeletions(deleteobjs, behavior, 0);
    1517                 :             : 
    1518                 :          18 :         table_close(sdepRel, RowExclusiveLock);
    1519                 :             : 
    1520                 :          18 :         free_object_addresses(deleteobjs);
    1521                 :          18 : }
    1522                 :             : 
    1523                 :             : /*
    1524                 :             :  * shdepReassignOwned
    1525                 :             :  *
    1526                 :             :  * Change the owner of objects owned by any of the roles in roleids to
    1527                 :             :  * newrole.  Grants are not touched.
    1528                 :             :  */
    1529                 :             : void
    1530                 :           4 : shdepReassignOwned(List *roleids, Oid newrole)
    1531                 :             : {
    1532                 :           4 :         Relation        sdepRel;
    1533                 :           4 :         ListCell   *cell;
    1534                 :             : 
    1535                 :             :         /*
    1536                 :             :          * We don't need this strong a lock here, but we'll call routines that
    1537                 :             :          * acquire RowExclusiveLock.  Better get that right now to avoid potential
    1538                 :             :          * deadlock problems.
    1539                 :             :          */
    1540                 :           4 :         sdepRel = table_open(SharedDependRelationId, RowExclusiveLock);
    1541                 :             : 
    1542   [ +  -  +  +  :           8 :         foreach(cell, roleids)
                   +  + ]
    1543                 :             :         {
    1544                 :           4 :                 SysScanDesc scan;
    1545                 :           4 :                 ScanKeyData key[2];
    1546                 :           4 :                 HeapTuple       tuple;
    1547                 :           4 :                 Oid                     roleid = lfirst_oid(cell);
    1548                 :             : 
    1549                 :             :                 /* Refuse to work on pinned roles */
    1550         [ +  - ]:           4 :                 if (IsPinnedObject(AuthIdRelationId, roleid))
    1551                 :             :                 {
    1552                 :           0 :                         ObjectAddress obj;
    1553                 :             : 
    1554                 :           0 :                         obj.classId = AuthIdRelationId;
    1555                 :           0 :                         obj.objectId = roleid;
    1556                 :           0 :                         obj.objectSubId = 0;
    1557                 :             : 
    1558   [ #  #  #  # ]:           0 :                         ereport(ERROR,
    1559                 :             :                                         (errcode(ERRCODE_DEPENDENT_OBJECTS_STILL_EXIST),
    1560                 :             :                                          errmsg("cannot reassign ownership of objects owned by %s because they are required by the database system",
    1561                 :             :                                                         getObjectDescription(&obj, false))));
    1562                 :             : 
    1563                 :             :                         /*
    1564                 :             :                          * There's no need to tell the whole truth, which is that we
    1565                 :             :                          * didn't track these dependencies at all ...
    1566                 :             :                          */
    1567                 :           0 :                 }
    1568                 :             : 
    1569                 :           8 :                 ScanKeyInit(&key[0],
    1570                 :             :                                         Anum_pg_shdepend_refclassid,
    1571                 :             :                                         BTEqualStrategyNumber, F_OIDEQ,
    1572                 :           4 :                                         ObjectIdGetDatum(AuthIdRelationId));
    1573                 :           8 :                 ScanKeyInit(&key[1],
    1574                 :             :                                         Anum_pg_shdepend_refobjid,
    1575                 :             :                                         BTEqualStrategyNumber, F_OIDEQ,
    1576                 :           4 :                                         ObjectIdGetDatum(roleid));
    1577                 :             : 
    1578                 :           8 :                 scan = systable_beginscan(sdepRel, SharedDependReferenceIndexId, true,
    1579                 :           4 :                                                                   NULL, 2, key);
    1580                 :             : 
    1581         [ +  + ]:          24 :                 while ((tuple = systable_getnext(scan)) != NULL)
    1582                 :             :                 {
    1583                 :          20 :                         Form_pg_shdepend sdepForm = (Form_pg_shdepend) GETSTRUCT(tuple);
    1584                 :          20 :                         MemoryContext cxt,
    1585                 :             :                                                 oldcxt;
    1586                 :             : 
    1587                 :             :                         /*
    1588                 :             :                          * We only operate on shared objects and objects in the current
    1589                 :             :                          * database
    1590                 :             :                          */
    1591   [ +  +  +  - ]:          20 :                         if (sdepForm->dbid != MyDatabaseId &&
    1592                 :           3 :                                 sdepForm->dbid != InvalidOid)
    1593                 :           0 :                                 continue;
    1594                 :             : 
    1595                 :             :                         /*
    1596                 :             :                          * The various DDL routines called here tend to leak memory in
    1597                 :             :                          * CurrentMemoryContext.  That's not a problem when they're only
    1598                 :             :                          * called once per command; but in this usage where we might be
    1599                 :             :                          * touching many objects, it can amount to a serious memory leak.
    1600                 :             :                          * Fix that by running each call in a short-lived context.
    1601                 :             :                          */
    1602                 :          20 :                         cxt = AllocSetContextCreate(CurrentMemoryContext,
    1603                 :             :                                                                                 "shdepReassignOwned",
    1604                 :             :                                                                                 ALLOCSET_DEFAULT_SIZES);
    1605                 :          20 :                         oldcxt = MemoryContextSwitchTo(cxt);
    1606                 :             : 
    1607                 :             :                         /* Perform the appropriate processing */
    1608   [ +  +  -  - ]:          20 :                         switch (sdepForm->deptype)
    1609                 :             :                         {
    1610                 :             :                                 case SHARED_DEPENDENCY_OWNER:
    1611                 :          15 :                                         shdepReassignOwned_Owner(sdepForm, newrole);
    1612                 :          15 :                                         break;
    1613                 :             :                                 case SHARED_DEPENDENCY_INITACL:
    1614                 :           0 :                                         shdepReassignOwned_InitAcl(sdepForm, roleid, newrole);
    1615                 :           0 :                                         break;
    1616                 :             :                                 case SHARED_DEPENDENCY_ACL:
    1617                 :             :                                 case SHARED_DEPENDENCY_POLICY:
    1618                 :             :                                 case SHARED_DEPENDENCY_TABLESPACE:
    1619                 :             :                                         /* Nothing to do for these entry types */
    1620                 :           5 :                                         break;
    1621                 :             :                                 default:
    1622   [ #  #  #  # ]:           0 :                                         elog(ERROR, "unrecognized dependency type: %d",
    1623                 :             :                                                  (int) sdepForm->deptype);
    1624                 :           0 :                                         break;
    1625                 :             :                         }
    1626                 :             : 
    1627                 :             :                         /* Clean up */
    1628                 :          20 :                         MemoryContextSwitchTo(oldcxt);
    1629                 :          20 :                         MemoryContextDelete(cxt);
    1630                 :             : 
    1631                 :             :                         /* Make sure the next iteration will see my changes */
    1632                 :          20 :                         CommandCounterIncrement();
    1633      [ -  -  + ]:          20 :                 }
    1634                 :             : 
    1635                 :           4 :                 systable_endscan(scan);
    1636                 :           4 :         }
    1637                 :             : 
    1638                 :           4 :         table_close(sdepRel, RowExclusiveLock);
    1639                 :           4 : }
    1640                 :             : 
    1641                 :             : /*
    1642                 :             :  * shdepReassignOwned_Owner
    1643                 :             :  *
    1644                 :             :  * shdepReassignOwned's processing of SHARED_DEPENDENCY_OWNER entries
    1645                 :             :  */
    1646                 :             : static void
    1647                 :          15 : shdepReassignOwned_Owner(Form_pg_shdepend sdepForm, Oid newrole)
    1648                 :             : {
    1649                 :             :         /* Issue the appropriate ALTER OWNER call */
    1650   [ +  +  +  +  :          15 :         switch (sdepForm->classid)
          +  +  -  -  -  
                   -  - ]
    1651                 :             :         {
    1652                 :             :                 case TypeRelationId:
    1653                 :           3 :                         AlterTypeOwner_oid(sdepForm->objid, newrole, true);
    1654                 :           3 :                         break;
    1655                 :             : 
    1656                 :             :                 case NamespaceRelationId:
    1657                 :           1 :                         AlterSchemaOwner_oid(sdepForm->objid, newrole);
    1658                 :           1 :                         break;
    1659                 :             : 
    1660                 :             :                 case RelationRelationId:
    1661                 :             : 
    1662                 :             :                         /*
    1663                 :             :                          * Pass recursing = true so that we don't fail on indexes, owned
    1664                 :             :                          * sequences, etc when we happen to visit them before their parent
    1665                 :             :                          * table.
    1666                 :             :                          */
    1667                 :           4 :                         ATExecChangeOwner(sdepForm->objid, newrole, true, AccessExclusiveLock);
    1668                 :           4 :                         break;
    1669                 :             : 
    1670                 :             :                 case DefaultAclRelationId:
    1671                 :             : 
    1672                 :             :                         /*
    1673                 :             :                          * Ignore default ACLs; they should be handled by DROP OWNED, not
    1674                 :             :                          * REASSIGN OWNED.
    1675                 :             :                          */
    1676                 :             :                         break;
    1677                 :             : 
    1678                 :             :                 case UserMappingRelationId:
    1679                 :             :                         /* ditto */
    1680                 :             :                         break;
    1681                 :             : 
    1682                 :             :                 case ForeignServerRelationId:
    1683                 :           2 :                         AlterForeignServerOwner_oid(sdepForm->objid, newrole);
    1684                 :           2 :                         break;
    1685                 :             : 
    1686                 :             :                 case ForeignDataWrapperRelationId:
    1687                 :           0 :                         AlterForeignDataWrapperOwner_oid(sdepForm->objid, newrole);
    1688                 :           0 :                         break;
    1689                 :             : 
    1690                 :             :                 case EventTriggerRelationId:
    1691                 :           0 :                         AlterEventTriggerOwner_oid(sdepForm->objid, newrole);
    1692                 :           0 :                         break;
    1693                 :             : 
    1694                 :             :                 case PublicationRelationId:
    1695                 :           0 :                         AlterPublicationOwner_oid(sdepForm->objid, newrole);
    1696                 :           0 :                         break;
    1697                 :             : 
    1698                 :             :                 case SubscriptionRelationId:
    1699                 :           0 :                         AlterSubscriptionOwner_oid(sdepForm->objid, newrole);
    1700                 :           0 :                         break;
    1701                 :             : 
    1702                 :             :                         /* Generic alter owner cases */
    1703                 :             :                 case CollationRelationId:
    1704                 :             :                 case ConversionRelationId:
    1705                 :             :                 case OperatorRelationId:
    1706                 :             :                 case ProcedureRelationId:
    1707                 :             :                 case LanguageRelationId:
    1708                 :             :                 case LargeObjectRelationId:
    1709                 :             :                 case OperatorFamilyRelationId:
    1710                 :             :                 case OperatorClassRelationId:
    1711                 :             :                 case ExtensionRelationId:
    1712                 :             :                 case StatisticExtRelationId:
    1713                 :             :                 case TableSpaceRelationId:
    1714                 :             :                 case DatabaseRelationId:
    1715                 :             :                 case TSConfigRelationId:
    1716                 :             :                 case TSDictionaryRelationId:
    1717                 :           4 :                         AlterObjectOwner_internal(sdepForm->classid,
    1718                 :           2 :                                                                           sdepForm->objid,
    1719                 :           2 :                                                                           newrole);
    1720                 :           2 :                         break;
    1721                 :             : 
    1722                 :             :                 default:
    1723   [ #  #  #  # ]:           0 :                         elog(ERROR, "unexpected classid %u", sdepForm->classid);
    1724                 :           0 :                         break;
    1725                 :             :         }
    1726                 :          15 : }
    1727                 :             : 
    1728                 :             : /*
    1729                 :             :  * shdepReassignOwned_InitAcl
    1730                 :             :  *
    1731                 :             :  * shdepReassignOwned's processing of SHARED_DEPENDENCY_INITACL entries
    1732                 :             :  */
    1733                 :             : static void
    1734                 :           0 : shdepReassignOwned_InitAcl(Form_pg_shdepend sdepForm, Oid oldrole, Oid newrole)
    1735                 :             : {
    1736                 :             :         /*
    1737                 :             :          * Currently, REASSIGN OWNED replaces mentions of oldrole with newrole in
    1738                 :             :          * pg_init_privs entries, just as it does in the object's regular ACL.
    1739                 :             :          * This is less than ideal, since pg_init_privs ought to retain a
    1740                 :             :          * historical record of the situation at the end of CREATE EXTENSION.
    1741                 :             :          * However, there are two big stumbling blocks to doing something
    1742                 :             :          * different:
    1743                 :             :          *
    1744                 :             :          * 1. If we don't replace the references, what is to happen if the old
    1745                 :             :          * role gets dropped?  (DROP OWNED's current answer is to just delete the
    1746                 :             :          * pg_init_privs entry, which is surely ahistorical.)
    1747                 :             :          *
    1748                 :             :          * 2. It's unlikely that pg_dump will cope nicely with pg_init_privs
    1749                 :             :          * entries that are based on a different owner than the object now has ---
    1750                 :             :          * the more so given that pg_init_privs doesn't record the original owner
    1751                 :             :          * explicitly.  (This problem actually exists anyway given that a bare
    1752                 :             :          * ALTER OWNER won't update pg_init_privs, but we don't need REASSIGN
    1753                 :             :          * OWNED making it worse.)
    1754                 :             :          */
    1755                 :           0 :         ReplaceRoleInInitPriv(oldrole, newrole,
    1756                 :           0 :                                                   sdepForm->classid,
    1757                 :           0 :                                                   sdepForm->objid,
    1758                 :           0 :                                                   sdepForm->objsubid);
    1759                 :           0 : }
        

Generated by: LCOV version 2.3.2-1