aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/bin/pg_dump/pg_backup_archiver.c3
-rw-r--r--src/bin/pg_dump/pg_dump.c97
-rw-r--r--src/bin/pg_dump/pg_restore.c57
3 files changed, 112 insertions, 45 deletions
diff --git a/src/bin/pg_dump/pg_backup_archiver.c b/src/bin/pg_dump/pg_backup_archiver.c
index e6e9bddd99d..fdd0bdbb96b 100644
--- a/src/bin/pg_dump/pg_backup_archiver.c
+++ b/src/bin/pg_dump/pg_backup_archiver.c
@@ -15,7 +15,7 @@
*
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_archiver.c,v 1.50 2002/07/02 05:49:51 momjian Exp $
+ * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_archiver.c,v 1.51 2002/07/04 03:04:54 momjian Exp $
*
* Modifications - 28-Jun-2000 - pjw@rhyme.com.au
*
@@ -191,6 +191,7 @@ RestoreArchive(Archive *AHX, RestoreOptions *ropt)
* initially connected to, not the one we will create, which is very
* bad...
*/
+
if (ropt->create && ropt->noReconnect)
die_horribly(AH, modulename, "-C and -R are incompatible options\n");
diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c
index a12e095b1fb..77adb3897f3 100644
--- a/src/bin/pg_dump/pg_dump.c
+++ b/src/bin/pg_dump/pg_dump.c
@@ -22,7 +22,7 @@
*
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.268 2002/07/02 05:49:51 momjian Exp $
+ * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.269 2002/07/04 03:04:54 momjian Exp $
*
*-------------------------------------------------------------------------
*/
@@ -104,14 +104,14 @@ static void dumpTableACL(Archive *fout, TableInfo *tbinfo);
static void dumpFuncACL(Archive *fout, FuncInfo *finfo);
static void dumpAggACL(Archive *fout, AggInfo *finfo);
static void dumpACL(Archive *fout, const char *type, const char *name,
- const char *nspname, const char *usename,
- const char *acl, const char *objoid);
+ const char *name_noquotes, const char *nspname,
+ const char *usename, const char *acl, const char *objoid);
static void dumpTriggers(Archive *fout, TableInfo *tblinfo, int numTables);
static void dumpRules(Archive *fout, TableInfo *tblinfo, int numTables);
static void formatStringLiteral(PQExpBuffer buf, const char *str,
const formatLiteralOptions opts);
-static char *format_function_signature(FuncInfo *finfo);
+static char *format_function_signature(FuncInfo *finfo, bool honor_quotes);
static void dumpOneFunc(Archive *fout, FuncInfo *finfo);
static void dumpOneOpr(Archive *fout, OprInfo *oprinfo,
OprInfo *g_oprinfo, int numOperators);
@@ -2721,7 +2721,7 @@ dumpNamespaces(Archive *fout, NamespaceInfo *nsinfo, int numNamespaces)
if (strcmp(nspinfo->nspname, "public") == 0)
{
if (!aclsSkip && strcmp(nspinfo->nspacl, "{=UC}") != 0)
- dumpACL(fout, "SCHEMA", qnspname, NULL,
+ dumpACL(fout, "SCHEMA", qnspname, nspinfo->nspname, NULL,
nspinfo->usename, nspinfo->nspacl,
nspinfo->oid);
}
@@ -2747,7 +2747,7 @@ dumpNamespaces(Archive *fout, NamespaceInfo *nsinfo, int numNamespaces)
nspinfo->oid, "pg_namespace", 0, NULL);
if (!aclsSkip)
- dumpACL(fout, "SCHEMA", qnspname, NULL,
+ dumpACL(fout, "SCHEMA", qnspname, nspinfo->nspname, NULL,
nspinfo->usename, nspinfo->nspacl,
nspinfo->oid);
}
@@ -3291,8 +3291,9 @@ dumpProcLangs(Archive *fout, FuncInfo finfo[], int numFuncs)
if (!aclsSkip)
{
- char * tmp = strdup(fmtId(lanname, force_quotes));
- dumpACL(fout, "LANGUAGE", tmp, finfo[fidx].pronamespace->nspname,
+ char *tmp = strdup(fmtId(lanname, force_quotes));
+ dumpACL(fout, "LANGUAGE", tmp, lanname,
+ finfo[fidx].pronamespace->nspname,
NULL, lanacl, lanoid);
free(tmp);
}
@@ -3333,13 +3334,16 @@ dumpFuncs(Archive *fout, FuncInfo finfo[], int numFuncs)
* is never qualified.
*/
static char *
-format_function_signature(FuncInfo *finfo)
+format_function_signature(FuncInfo *finfo, bool honor_quotes)
{
PQExpBufferData fn;
int j;
initPQExpBuffer(&fn);
- appendPQExpBuffer(&fn, "%s (", fmtId(finfo->proname, force_quotes));
+ if (honor_quotes)
+ appendPQExpBuffer(&fn, "%s(", fmtId(finfo->proname, force_quotes));
+ else
+ appendPQExpBuffer(&fn, "%s(", finfo->proname);
for (j = 0; j < finfo->nargs; j++)
{
char *typname;
@@ -3358,12 +3362,15 @@ format_function_signature(FuncInfo *finfo)
static void
dumpFuncACL(Archive *fout, FuncInfo *finfo)
{
- char *funcsig;
+ char *funcsig, *funcsig_noquotes;
- funcsig = format_function_signature(finfo);
- dumpACL(fout, "FUNCTION", funcsig, finfo->pronamespace->nspname,
+ funcsig = format_function_signature(finfo, true);
+ funcsig_noquotes = format_function_signature(finfo, false);
+ dumpACL(fout, "FUNCTION", funcsig, funcsig_noquotes,
+ finfo->pronamespace->nspname,
finfo->usename, finfo->proacl, finfo->oid);
free(funcsig);
+ free(funcsig_noquotes);
}
@@ -3380,6 +3387,7 @@ dumpOneFunc(Archive *fout, FuncInfo *finfo)
PQExpBuffer asPart = createPQExpBuffer();
PGresult *res = NULL;
char *funcsig = NULL;
+ char *funcsig_noquotes = NULL;
int ntups;
char *proretset;
char *prosrc;
@@ -3487,7 +3495,8 @@ dumpOneFunc(Archive *fout, FuncInfo *finfo)
}
}
- funcsig = format_function_signature(finfo);
+ funcsig = format_function_signature(finfo, true);
+ funcsig_noquotes = format_function_signature(finfo, false);
/* DROP must be fully qualified in case same name appears in pg_catalog */
appendPQExpBuffer(delqry, "DROP FUNCTION %s.%s;\n",
@@ -3517,7 +3526,7 @@ dumpOneFunc(Archive *fout, FuncInfo *finfo)
finfo->proname);
exit_nicely();
}
- }
+ }
if (proimplicit[0] == 't')
appendPQExpBuffer(q, " IMPLICIT CAST");
@@ -3530,7 +3539,8 @@ dumpOneFunc(Archive *fout, FuncInfo *finfo)
appendPQExpBuffer(q, ";\n");
- ArchiveEntry(fout, finfo->oid, funcsig, finfo->pronamespace->nspname,
+ ArchiveEntry(fout, finfo->oid, funcsig_noquotes,
+ finfo->pronamespace->nspname,
finfo->usename, "FUNCTION", NULL,
q->data, delqry->data,
NULL, NULL, NULL);
@@ -3551,6 +3561,7 @@ done:
destroyPQExpBuffer(delqry);
destroyPQExpBuffer(asPart);
free(funcsig);
+ free(funcsig_noquotes);
}
/*
@@ -3939,14 +3950,17 @@ dumpAggs(Archive *fout, AggInfo agginfo[], int numAggs)
* is never qualified.
*/
static char *
-format_aggregate_signature(AggInfo *agginfo, Archive *fout)
+format_aggregate_signature(AggInfo *agginfo, Archive *fout, bool honor_quotes)
{
PQExpBufferData buf;
bool anybasetype;
initPQExpBuffer(&buf);
- appendPQExpBuffer(&buf, "%s",
+ if (honor_quotes)
+ appendPQExpBuffer(&buf, "%s",
fmtId(agginfo->aggname, force_quotes));
+ else
+ appendPQExpBuffer(&buf, "%s", agginfo->aggname);
anybasetype = (strcmp(agginfo->aggbasetype, "0") == 0);
@@ -3974,12 +3988,15 @@ format_aggregate_signature(AggInfo *agginfo, Archive *fout)
static void
dumpAggACL(Archive *fout, AggInfo *finfo)
{
- char *aggsig;
+ char *aggsig, *aggsig_noquotes;
- aggsig = format_aggregate_signature(finfo, fout);
- dumpACL(fout, "FUNCTION", aggsig, finfo->aggnamespace->nspname,
+ aggsig = format_aggregate_signature(finfo, fout, true);
+ aggsig_noquotes = format_aggregate_signature(finfo, fout, false);
+ dumpACL(fout, "FUNCTION", aggsig, aggsig_noquotes,
+ finfo->aggnamespace->nspname,
finfo->usename, finfo->aggacl, finfo->oid);
free(aggsig);
+ free(aggsig_noquotes);
}
@@ -3994,7 +4011,8 @@ dumpOneAgg(Archive *fout, AggInfo *agginfo)
PQExpBuffer q = createPQExpBuffer();
PQExpBuffer delq = createPQExpBuffer();
PQExpBuffer details = createPQExpBuffer();
- char *aggSig;
+ char *aggsig;
+ char *aggsig_noquotes;
PGresult *res;
int ntups;
int i_aggtransfn;
@@ -4084,16 +4102,17 @@ dumpOneAgg(Archive *fout, AggInfo *agginfo)
agginfo->fmtbasetype = strdup(PQgetvalue(res, 0, i_fmtbasetype));
convertok = (PQgetvalue(res, 0, i_convertok)[0] == 't');
- aggSig = format_aggregate_signature(agginfo, g_fout);
+ aggsig = format_aggregate_signature(agginfo, g_fout, true);
+ aggsig_noquotes = format_aggregate_signature(agginfo, g_fout, false);
if (!convertok)
{
write_msg(NULL, "WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
- aggSig);
+ aggsig);
appendPQExpBuffer(q, "-- WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
- aggSig);
- ArchiveEntry(fout, agginfo->oid, aggSig,
+ aggsig);
+ ArchiveEntry(fout, agginfo->oid, aggsig_noquotes,
agginfo->aggnamespace->nspname, agginfo->usename,
"WARNING", NULL,
q->data, "" /* Del */ ,
@@ -4146,13 +4165,13 @@ dumpOneAgg(Archive *fout, AggInfo *agginfo)
/* DROP must be fully qualified in case same name appears in pg_catalog */
appendPQExpBuffer(delq, "DROP AGGREGATE %s.%s;\n",
fmtId(agginfo->aggnamespace->nspname, force_quotes),
- aggSig);
+ aggsig);
appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
fmtId(agginfo->aggname, force_quotes),
details->data);
- ArchiveEntry(fout, agginfo->oid, aggSig,
+ ArchiveEntry(fout, agginfo->oid, aggsig_noquotes,
agginfo->aggnamespace->nspname, agginfo->usename,
"AGGREGATE", NULL,
q->data, delq->data,
@@ -4161,7 +4180,7 @@ dumpOneAgg(Archive *fout, AggInfo *agginfo)
/*** Dump Aggregate Comments ***/
resetPQExpBuffer(q);
- appendPQExpBuffer(q, "AGGREGATE %s", aggSig);
+ appendPQExpBuffer(q, "AGGREGATE %s", aggsig);
if (g_fout->remoteVersion >= 70300)
dumpComment(fout, q->data,
agginfo->aggnamespace->nspname, agginfo->usename,
@@ -4177,7 +4196,8 @@ dumpOneAgg(Archive *fout, AggInfo *agginfo)
destroyPQExpBuffer(q);
destroyPQExpBuffer(delq);
destroyPQExpBuffer(details);
- free(aggSig);
+ free(aggsig);
+ free(aggsig_noquotes);
}
@@ -4276,7 +4296,7 @@ GetPrivileges(Archive *AH, const char *s, const char *type)
*/
static void
dumpACL(Archive *fout, const char *type, const char *name,
- const char *nspname, const char *usename,
+ const char *name_noquotes, const char *nspname, const char *usename,
const char *acls, const char *objoid)
{
char *aclbuf,
@@ -4390,7 +4410,7 @@ dumpACL(Archive *fout, const char *type, const char *name,
appendPQExpBuffer(sql, "%s;\n", fmtId(usename, force_quotes));
}
- ArchiveEntry(fout, objoid, name, nspname, usename ? usename : "",
+ ArchiveEntry(fout, objoid, name_noquotes, nspname, usename ? usename : "",
"ACL", NULL, sql->data, "", NULL, NULL, NULL);
free(aclbuf);
@@ -4401,9 +4421,9 @@ dumpACL(Archive *fout, const char *type, const char *name,
static void
dumpTableACL(Archive *fout, TableInfo *tbinfo)
{
- char * tmp = strdup( fmtId(tbinfo->relname, force_quotes) );
- dumpACL(fout, "TABLE", tmp, tbinfo->relnamespace->nspname,
- tbinfo->usename, tbinfo->relacl,
+ char *tmp = strdup(fmtId(tbinfo->relname, force_quotes));
+ dumpACL(fout, "TABLE", tmp, tbinfo->relname,
+ tbinfo->relnamespace->nspname, tbinfo->usename, tbinfo->relacl,
tbinfo->viewoid != NULL ? tbinfo->viewoid : tbinfo->oid);
free(tmp);
}
@@ -5793,20 +5813,17 @@ myFormatType(const char *typname, int32 typmod)
precision, scale);
}
}
-
/*
* char is an internal single-byte data type; Let's make sure we force
* it through with quotes. - thomas 1998-12-13
*/
else if (!strcmp(typname, "char"))
{
- appendPQExpBuffer(buf, "%s",
- fmtId(typname, true));
+ appendPQExpBuffer(buf, "%s", fmtId(typname, true));
}
else
{
- appendPQExpBuffer(buf, "%s",
- fmtId(typname, false));
+ appendPQExpBuffer(buf, "%s", fmtId(typname, false));
}
result = strdup(buf->data);
diff --git a/src/bin/pg_dump/pg_restore.c b/src/bin/pg_dump/pg_restore.c
index 9d2af919aca..db417d41f60 100644
--- a/src/bin/pg_dump/pg_restore.c
+++ b/src/bin/pg_dump/pg_restore.c
@@ -34,7 +34,7 @@
*
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_restore.c,v 1.34 2002/05/10 22:36:27 tgl Exp $
+ * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_restore.c,v 1.35 2002/07/04 03:04:55 momjian Exp $
*
*-------------------------------------------------------------------------
*/
@@ -65,6 +65,7 @@
/* Forward decls */
static void usage(const char *progname);
static char *_cleanupName(char *name);
+static char *_cleanupFuncName(char *name);
typedef struct option optType;
@@ -220,7 +221,7 @@ main(int argc, char **argv)
case 'P': /* Function */
opts->selTypes = 1;
opts->selFunction = 1;
- opts->functionNames = optarg ? strdup(optarg) : NULL;
+ opts->functionNames = _cleanupFuncName(optarg);
break;
case 'I': /* Index */
opts->selTypes = 1;
@@ -392,7 +393,7 @@ usage(const char *progname)
" -O, --no-owner do not reconnect to database to match\n"
" object owner\n"
" -p, --port=PORT server port number\n"
- " -P, --function=NAME restore named function\n"
+ " -P, --function=NAME(args)restore named function\n"
" -r, --rearrange rearrange output to put indexes etc. at end\n"
" -R, --no-reconnect disallow ALL reconnections to the database\n"
" -s, --schema-only restore only the schema, no data\n"
@@ -430,7 +431,7 @@ usage(const char *progname)
" -O do not reconnect to database to match\n"
" object owner\n"
" -p PORT server port number\n"
- " -P NAME restore named function\n"
+ " -P NAME(args) restore named function\n"
" -r rearrange output to put indexes etc. at end\n"
" -R disallow ALL reconnections to the database\n"
" -s restore only the schema, no data\n"
@@ -478,3 +479,51 @@ _cleanupName(char *name)
}
return name;
}
+
+
+static char *
+_cleanupFuncName(char *name)
+{
+ int i;
+ char *ch;
+
+ if (!name || !name[0])
+ return NULL;
+
+ name = strdup(name);
+
+ if (name[0] == '"')
+ {
+ strcpy(name, &name[1]);
+ if (strchr(name, '"') != NULL)
+ strcpy(strchr(name, '"'), strchr(name, '"')+1);
+ }
+ /* otherwise, convert function name to lowercase... */
+ else
+ {
+ for (i = 0; name[i]; i++)
+ if (isupper((unsigned char) name[i]))
+ name[i] = tolower((unsigned char) name[i]);
+ }
+
+ /* strip out any space before paren */
+ ch = strchr(name,'(');
+ while (ch && ch > name && *(ch-1) == ' ')
+ {
+ strcpy(ch - 1, ch);
+ ch--;
+ }
+
+ /*
+ * Strip out spaces after commas in parameter list.
+ * We can't remove all spaces because some types, like
+ * 'double precision' have spaces.
+ */
+ if ((ch = strchr(name,'(')) != NULL)
+ {
+ while ((ch = strstr(ch,", ")) != NULL)
+ strcpy(ch + 1, ch + 2);
+ }
+
+ return name;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2025-05-31 13:13:54 +0000