mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-27 08:39:28 +08:00
Fix bugs in "restore.sql" script emitted in pg_dump tar output.
The tar output module did some very ugly and ultimately incorrect hacking on COPY commands to try to get them to work in the context of restoring a deconstructed tar archive. In particular, it would fail altogether for table names containing any upper-case characters, since it smashed the command string to lower-case before modifying it (and, just to add insult to injury, did that in a way that would fail in multibyte encodings). I don't see any particular value in being flexible about the case of the command keywords, since the string will just have been created by dumpTableData, so let's get rid of the whole case-folding thing. Also, it doesn't seem to meet the POLA for the script to restore data only in COPY mode, so add \i commands to make it have comparable behavior in --inserts mode. Noted while looking at the tar-output code in connection with Brian Weaver's patch.
This commit is contained in:
parent
997fa75d6b
commit
ff75219e9f
@ -647,56 +647,46 @@ _PrintTocData(ArchiveHandle *AH, TocEntry *te, RestoreOptions *ropt)
|
|||||||
{
|
{
|
||||||
lclContext *ctx = (lclContext *) AH->formatData;
|
lclContext *ctx = (lclContext *) AH->formatData;
|
||||||
lclTocEntry *tctx = (lclTocEntry *) te->formatData;
|
lclTocEntry *tctx = (lclTocEntry *) te->formatData;
|
||||||
char *tmpCopy;
|
int pos1;
|
||||||
size_t i,
|
|
||||||
pos1,
|
|
||||||
pos2;
|
|
||||||
|
|
||||||
if (!tctx->filename)
|
if (!tctx->filename)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* If we're writing the special restore.sql script, emit a suitable
|
||||||
|
* command to include each table's data from the corresponding file.
|
||||||
|
*
|
||||||
|
* In the COPY case this is a bit klugy because the regular COPY command
|
||||||
|
* was already printed before we get control.
|
||||||
|
*/
|
||||||
if (ctx->isSpecialScript)
|
if (ctx->isSpecialScript)
|
||||||
{
|
{
|
||||||
if (!te->copyStmt)
|
if (te->copyStmt)
|
||||||
return;
|
{
|
||||||
|
/* Abort the COPY FROM stdin */
|
||||||
|
ahprintf(AH, "\\.\n");
|
||||||
|
|
||||||
/* Abort the default COPY */
|
/*
|
||||||
ahprintf(AH, "\\.\n");
|
* The COPY statement should look like "COPY ... FROM stdin;\n",
|
||||||
|
* see dumpTableData().
|
||||||
|
*/
|
||||||
|
pos1 = (int) strlen(te->copyStmt) - 13;
|
||||||
|
if (pos1 < 6 || strncmp(te->copyStmt, "COPY ", 5) != 0 ||
|
||||||
|
strcmp(te->copyStmt + pos1, " FROM stdin;\n") != 0)
|
||||||
|
exit_horribly(modulename,
|
||||||
|
"unexpected COPY statement syntax: \"%s\"\n",
|
||||||
|
te->copyStmt);
|
||||||
|
|
||||||
/* Get a copy of the COPY statement and clean it up */
|
/* Emit all but the FROM part ... */
|
||||||
tmpCopy = pg_strdup(te->copyStmt);
|
ahwrite(te->copyStmt, 1, pos1, AH);
|
||||||
for (i = 0; i < strlen(tmpCopy); i++)
|
/* ... and insert modified FROM */
|
||||||
tmpCopy[i] = pg_tolower((unsigned char) tmpCopy[i]);
|
ahprintf(AH, " FROM '$$PATH$$/%s';\n\n", tctx->filename);
|
||||||
|
}
|
||||||
/*
|
else
|
||||||
* This is very nasty; we don't know if the archive used WITH OIDS, so
|
{
|
||||||
* we search the string for it in a paranoid sort of way.
|
/* --inserts mode, no worries, just include the data file */
|
||||||
*/
|
ahprintf(AH, "\\i $$PATH$$/%s\n\n", tctx->filename);
|
||||||
if (strncmp(tmpCopy, "copy ", 5) != 0)
|
}
|
||||||
exit_horribly(modulename,
|
|
||||||
"invalid COPY statement -- could not find \"copy\" in string \"%s\"\n", tmpCopy);
|
|
||||||
|
|
||||||
pos1 = 5;
|
|
||||||
for (pos1 = 5; pos1 < strlen(tmpCopy); pos1++)
|
|
||||||
if (tmpCopy[pos1] != ' ')
|
|
||||||
break;
|
|
||||||
|
|
||||||
if (tmpCopy[pos1] == '"')
|
|
||||||
pos1 += 2;
|
|
||||||
|
|
||||||
pos1 += strlen(te->tag);
|
|
||||||
|
|
||||||
for (pos2 = pos1; pos2 < strlen(tmpCopy); pos2++)
|
|
||||||
if (strncmp(&tmpCopy[pos2], "from stdin", 10) == 0)
|
|
||||||
break;
|
|
||||||
|
|
||||||
if (pos2 >= strlen(tmpCopy))
|
|
||||||
exit_horribly(modulename,
|
|
||||||
"invalid COPY statement -- could not find \"from stdin\" in string \"%s\" starting at position %lu\n",
|
|
||||||
tmpCopy, (unsigned long) pos1);
|
|
||||||
|
|
||||||
ahwrite(tmpCopy, 1, pos2, AH); /* 'copy "table" [with oids]' */
|
|
||||||
ahprintf(AH, " from '$$PATH$$/%s' %s", tctx->filename, &tmpCopy[pos2 + 10]);
|
|
||||||
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -843,18 +833,14 @@ _CloseArchive(ArchiveHandle *AH)
|
|||||||
* if the files have been extracted.
|
* if the files have been extracted.
|
||||||
*/
|
*/
|
||||||
th = tarOpen(AH, "restore.sql", 'w');
|
th = tarOpen(AH, "restore.sql", 'w');
|
||||||
tarPrintf(AH, th, "create temporary table pgdump_restore_path(p text);\n");
|
|
||||||
tarPrintf(AH, th, "--\n"
|
tarPrintf(AH, th, "--\n"
|
||||||
"-- NOTE:\n"
|
"-- NOTE:\n"
|
||||||
"--\n"
|
"--\n"
|
||||||
"-- File paths need to be edited. Search for $$PATH$$ and\n"
|
"-- File paths need to be edited. Search for $$PATH$$ and\n"
|
||||||
"-- replace it with the path to the directory containing\n"
|
"-- replace it with the path to the directory containing\n"
|
||||||
"-- the extracted data files.\n"
|
"-- the extracted data files.\n"
|
||||||
"--\n"
|
|
||||||
"-- Edit the following to match the path where the\n"
|
|
||||||
"-- tar archive has been extracted.\n"
|
|
||||||
"--\n");
|
"--\n");
|
||||||
tarPrintf(AH, th, "insert into pgdump_restore_path values('/tmp');\n\n");
|
|
||||||
|
|
||||||
AH->CustomOutPtr = _scriptOut;
|
AH->CustomOutPtr = _scriptOut;
|
||||||
|
|
||||||
@ -882,6 +868,8 @@ _CloseArchive(ArchiveHandle *AH)
|
|||||||
|
|
||||||
tarClose(AH, th);
|
tarClose(AH, th);
|
||||||
|
|
||||||
|
ctx->isSpecialScript = 0;
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* EOF marker for tar files is two blocks of NULLs.
|
* EOF marker for tar files is two blocks of NULLs.
|
||||||
*/
|
*/
|
||||||
|
Loading…
Reference in New Issue
Block a user