diff options
-rw-r--r-- | src/bin/pg_dump/pg_backup_archiver.c | 10 | ||||
-rw-r--r-- | src/bin/pg_dump/pg_dump.c | 77 |
2 files changed, 51 insertions, 36 deletions
diff --git a/src/bin/pg_dump/pg_backup_archiver.c b/src/bin/pg_dump/pg_backup_archiver.c index 0b8b3e520f7..0829f6c74e9 100644 --- a/src/bin/pg_dump/pg_backup_archiver.c +++ b/src/bin/pg_dump/pg_backup_archiver.c @@ -2841,8 +2841,14 @@ _tocEntryRequired(TocEntry *te, teSection curSection, RestoreOptions *ropt) * other information should be generated in binary-upgrade mode (not * the actual data). */ - if (!(ropt->binary_upgrade && strcmp(te->desc,"BLOB") == 0) && - !(ropt->binary_upgrade && strncmp(te->tag,"LARGE OBJECT ", 13) == 0)) + if (!(ropt->binary_upgrade && + (strcmp(te->desc, "BLOB") == 0 || + (strcmp(te->desc, "ACL") == 0 && + strncmp(te->tag, "LARGE OBJECT ", 13) == 0) || + (strcmp(te->desc, "COMMENT") == 0 && + strncmp(te->tag, "LARGE OBJECT ", 13) == 0) || + (strcmp(te->desc, "SECURITY LABEL") == 0 && + strncmp(te->tag, "LARGE OBJECT ", 13) == 0)))) res = res & REQ_SCHEMA; } diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c index a0206274a45..ea5c700f3ee 100644 --- a/src/bin/pg_dump/pg_dump.c +++ b/src/bin/pg_dump/pg_dump.c @@ -2310,6 +2310,7 @@ dumpDatabase(Archive *fout) PQExpBuffer dbQry = createPQExpBuffer(); PQExpBuffer delQry = createPQExpBuffer(); PQExpBuffer creaQry = createPQExpBuffer(); + PQExpBuffer labelq = createPQExpBuffer(); PGconn *conn = GetConnection(fout); PGresult *res; int i_tableoid, @@ -2596,16 +2597,20 @@ dumpDatabase(Archive *fout) destroyPQExpBuffer(loOutQry); } + /* Compute correct tag for comments etc */ + appendPQExpBuffer(labelq, "DATABASE %s", fmtId(datname)); + /* Dump DB comment if any */ if (fout->remoteVersion >= 80200) { /* - * 8.2 keeps comments on shared objects in a shared table, so we - * cannot use the dumpComment used for other database objects. + * 8.2 and up keep comments on shared objects in a shared table, so we + * cannot use the dumpComment() code used for other database objects. + * Be careful that the ArchiveEntry parameters match that function. */ char *comment = PQgetvalue(res, 0, PQfnumber(res, "description")); - if (comment && strlen(comment)) + if (comment && *comment) { resetPQExpBuffer(dbQry); @@ -2617,17 +2622,17 @@ dumpDatabase(Archive *fout) appendStringLiteralAH(dbQry, comment, fout); appendPQExpBufferStr(dbQry, ";\n"); - ArchiveEntry(fout, dbCatId, createDumpId(), datname, NULL, NULL, - dba, false, "COMMENT", SECTION_NONE, + ArchiveEntry(fout, nilCatalogId, createDumpId(), + labelq->data, NULL, NULL, dba, + false, "COMMENT", SECTION_NONE, dbQry->data, "", NULL, - &dbDumpId, 1, NULL, NULL); + &(dbDumpId), 1, + NULL, NULL); } } else { - resetPQExpBuffer(dbQry); - appendPQExpBuffer(dbQry, "DATABASE %s", fmtId(datname)); - dumpComment(fout, dbQry->data, NULL, "", + dumpComment(fout, labelq->data, NULL, dba, dbCatId, 0, dbDumpId); } @@ -2643,11 +2648,13 @@ dumpDatabase(Archive *fout) shres = ExecuteSqlQuery(fout, seclabelQry->data, PGRES_TUPLES_OK); resetPQExpBuffer(seclabelQry); emitShSecLabels(conn, shres, seclabelQry, "DATABASE", datname); - if (strlen(seclabelQry->data)) - ArchiveEntry(fout, dbCatId, createDumpId(), datname, NULL, NULL, - dba, false, "SECURITY LABEL", SECTION_NONE, + if (seclabelQry->len > 0) + ArchiveEntry(fout, nilCatalogId, createDumpId(), + labelq->data, NULL, NULL, dba, + false, "SECURITY LABEL", SECTION_NONE, seclabelQry->data, "", NULL, - &dbDumpId, 1, NULL, NULL); + &(dbDumpId), 1, + NULL, NULL); destroyPQExpBuffer(seclabelQry); PQclear(shres); } @@ -2657,6 +2664,7 @@ dumpDatabase(Archive *fout) destroyPQExpBuffer(dbQry); destroyPQExpBuffer(delQry); destroyPQExpBuffer(creaQry); + destroyPQExpBuffer(labelq); } /* @@ -8618,7 +8626,7 @@ dumpNamespace(Archive *fout, NamespaceInfo *nspinfo) nspinfo->dobj.catId, 0, nspinfo->dobj.dumpId); dumpACL(fout, nspinfo->dobj.catId, nspinfo->dobj.dumpId, "SCHEMA", - qnspname, NULL, nspinfo->dobj.name, NULL, + qnspname, NULL, labelq->data, NULL, nspinfo->rolname, nspinfo->nspacl); free(qnspname); @@ -8903,7 +8911,7 @@ dumpEnumType(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -9036,7 +9044,7 @@ dumpRangeType(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -9106,7 +9114,7 @@ dumpUndefinedType(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -9495,7 +9503,7 @@ dumpBaseType(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -9658,7 +9666,7 @@ dumpDomain(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -9885,7 +9893,7 @@ dumpCompositeType(Archive *fout, TypeInfo *tyinfo) tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId); dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE", - qtypname, NULL, tyinfo->dobj.name, + qtypname, NULL, labelq->data, tyinfo->dobj.namespace->dobj.name, tyinfo->rolname, tyinfo->typacl); @@ -10197,7 +10205,7 @@ dumpProcLang(Archive *fout, ProcLangInfo *plang) if (plang->lanpltrusted) dumpACL(fout, plang->dobj.catId, plang->dobj.dumpId, "LANGUAGE", - qlanname, NULL, plang->dobj.name, + qlanname, NULL, labelq->data, lanschema, plang->lanowner, plang->lanacl); @@ -10826,7 +10834,7 @@ dumpFunc(Archive *fout, FuncInfo *finfo) finfo->dobj.catId, 0, finfo->dobj.dumpId); dumpACL(fout, finfo->dobj.catId, finfo->dobj.dumpId, "FUNCTION", - funcsig, NULL, funcsig_tag, + funcsig, NULL, labelq->data, finfo->dobj.namespace->dobj.name, finfo->rolname, finfo->proacl); @@ -12711,14 +12719,12 @@ dumpAgg(Archive *fout, AggInfo *agginfo) * syntax for zero-argument aggregates and ordered-set aggregates. */ free(aggsig); - free(aggsig_tag); aggsig = format_function_signature(fout, &agginfo->aggfn, true); - aggsig_tag = format_function_signature(fout, &agginfo->aggfn, false); dumpACL(fout, agginfo->aggfn.dobj.catId, agginfo->aggfn.dobj.dumpId, "FUNCTION", - aggsig, NULL, aggsig_tag, + aggsig, NULL, labelq->data, agginfo->aggfn.dobj.namespace->dobj.name, agginfo->aggfn.rolname, agginfo->aggfn.proacl); @@ -13152,7 +13158,7 @@ dumpForeignDataWrapper(Archive *fout, FdwInfo *fdwinfo) /* Handle the ACL */ dumpACL(fout, fdwinfo->dobj.catId, fdwinfo->dobj.dumpId, "FOREIGN DATA WRAPPER", - qfdwname, NULL, fdwinfo->dobj.name, + qfdwname, NULL, labelq->data, NULL, fdwinfo->rolname, fdwinfo->fdwacl); @@ -13245,7 +13251,7 @@ dumpForeignServer(Archive *fout, ForeignServerInfo *srvinfo) /* Handle the ACL */ dumpACL(fout, srvinfo->dobj.catId, srvinfo->dobj.dumpId, "FOREIGN SERVER", - qsrvname, NULL, srvinfo->dobj.name, + qsrvname, NULL, labelq->data, NULL, srvinfo->rolname, srvinfo->srvacl); @@ -13445,7 +13451,8 @@ dumpDefaultACL(Archive *fout, DefaultACLInfo *daclinfo) * FOREIGN DATA WRAPPER, SERVER, or LARGE OBJECT. * 'name' is the formatted name of the object. Must be quoted etc. already. * 'subname' is the formatted name of the sub-object, if any. Must be quoted. - * 'tag' is the tag for the archive entry (typ. unquoted name of object). + * 'tag' is the tag for the archive entry (should be the same tag as would be + * used for comments etc; for example "TABLE foo"). * 'nspname' is the namespace the object is in (NULL if none). * 'owner' is the owner, NULL if there is no owner (for languages). * 'acls' is the string read out of the fooacl system catalog field; @@ -13523,7 +13530,7 @@ dumpSecLabel(Archive *fout, const char *target, if (dopt->no_security_labels) return; - /* Comments are schema not data ... except blob comments are data */ + /* Security labels are schema not data ... except blob labels are data */ if (strncmp(target, "LARGE OBJECT ", 13) != 0) { if (dopt->dataOnly) @@ -13804,6 +13811,8 @@ dumpTable(Archive *fout, TableInfo *tbinfo) if (tbinfo->dobj.dump && !dopt->dataOnly) { char *namecopy; + const char *objtype; + char *acltag; if (tbinfo->relkind == RELKIND_SEQUENCE) dumpSequence(fout, tbinfo); @@ -13812,12 +13821,13 @@ dumpTable(Archive *fout, TableInfo *tbinfo) /* Handle the ACL here */ namecopy = pg_strdup(fmtId(tbinfo->dobj.name)); + objtype = (tbinfo->relkind == RELKIND_SEQUENCE) ? "SEQUENCE" : "TABLE"; + acltag = psprintf("%s %s", objtype, namecopy); dumpACL(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId, - (tbinfo->relkind == RELKIND_SEQUENCE) ? "SEQUENCE" : - "TABLE", - namecopy, NULL, tbinfo->dobj.name, + objtype, namecopy, NULL, acltag, tbinfo->dobj.namespace->dobj.name, tbinfo->rolname, tbinfo->relacl); + free(acltag); /* * Handle column ACLs, if any. Note: we pull these with a separate @@ -13842,10 +13852,9 @@ dumpTable(Archive *fout, TableInfo *tbinfo) char *attname = PQgetvalue(res, i, 0); char *attacl = PQgetvalue(res, i, 1); char *attnamecopy; - char *acltag; attnamecopy = pg_strdup(fmtId(attname)); - acltag = psprintf("%s.%s", tbinfo->dobj.name, attname); + acltag = psprintf("COLUMN %s.%s", namecopy, attnamecopy); /* Column's GRANT type is always TABLE */ dumpACL(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId, "TABLE", namecopy, attnamecopy, acltag, |