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