mirror of
https://git.postgresql.org/git/postgresql.git
synced 2025-01-06 15:24:56 +08:00
f763b77193
When starting either an old or new postmaster, force it to place its Unix socket in the current directory. This makes it even harder for accidental connections to occur during pg_upgrade, and also works around some scenarios where the default socket location isn't usable. (For example, if the default location is something other than "/tmp", it might not exist during "make check".) When checking an already-running old postmaster, find out its actual socket directory location from postmaster.pid, if possible. This dodges problems with an old postmaster having a configured location different from the default built into pg_upgrade's libpq. We can't find that out if the old postmaster is pre-9.1, so also document how to cope with such scenarios manually. In support of this, centralize handling of the connection-related command line options passed to pg_upgrade's subsidiary programs, such as pg_dump. This should make future changes easier. Bruce Momjian and Tom Lane
105 lines
3.1 KiB
C
105 lines
3.1 KiB
C
/*
|
|
* dump.c
|
|
*
|
|
* dump functions
|
|
*
|
|
* Copyright (c) 2010-2012, PostgreSQL Global Development Group
|
|
* contrib/pg_upgrade/dump.c
|
|
*/
|
|
|
|
#include "postgres.h"
|
|
|
|
#include "pg_upgrade.h"
|
|
|
|
#include <sys/types.h>
|
|
|
|
void
|
|
generate_old_dump(void)
|
|
{
|
|
/* run new pg_dumpall binary */
|
|
prep_status("Creating catalog dump");
|
|
|
|
/*
|
|
* --binary-upgrade records the width of dropped columns in pg_class, and
|
|
* restores the frozenid's for databases and relations.
|
|
*/
|
|
exec_prog(UTILITY_LOG_FILE, NULL, true,
|
|
"\"%s/pg_dumpall\" %s --schema-only --binary-upgrade %s -f %s",
|
|
new_cluster.bindir, cluster_conn_opts(&old_cluster),
|
|
log_opts.verbose ? "--verbose" : "",
|
|
ALL_DUMP_FILE);
|
|
check_ok();
|
|
}
|
|
|
|
|
|
/*
|
|
* split_old_dump
|
|
*
|
|
* This function splits pg_dumpall output into global values and
|
|
* database creation, and per-db schemas. This allows us to create
|
|
* the support functions between restoring these two parts of the
|
|
* dump. We split on the first "\connect " after a CREATE ROLE
|
|
* username match; this is where the per-db restore starts.
|
|
*
|
|
* We suppress recreation of our own username so we don't generate
|
|
* an error during restore
|
|
*/
|
|
void
|
|
split_old_dump(void)
|
|
{
|
|
FILE *all_dump,
|
|
*globals_dump,
|
|
*db_dump;
|
|
FILE *current_output;
|
|
char line[LINE_ALLOC];
|
|
bool start_of_line = true;
|
|
char create_role_str[MAX_STRING];
|
|
char create_role_str_quote[MAX_STRING];
|
|
char filename[MAXPGPATH];
|
|
bool suppressed_username = false;
|
|
|
|
snprintf(filename, sizeof(filename), "%s", ALL_DUMP_FILE);
|
|
if ((all_dump = fopen(filename, "r")) == NULL)
|
|
pg_log(PG_FATAL, "Could not open dump file \"%s\": %s\n", filename, getErrorText(errno));
|
|
snprintf(filename, sizeof(filename), "%s", GLOBALS_DUMP_FILE);
|
|
if ((globals_dump = fopen_priv(filename, "w")) == NULL)
|
|
pg_log(PG_FATAL, "Could not write to dump file \"%s\": %s\n", filename, getErrorText(errno));
|
|
snprintf(filename, sizeof(filename), "%s", DB_DUMP_FILE);
|
|
if ((db_dump = fopen_priv(filename, "w")) == NULL)
|
|
pg_log(PG_FATAL, "Could not write to dump file \"%s\": %s\n", filename, getErrorText(errno));
|
|
|
|
current_output = globals_dump;
|
|
|
|
/* patterns used to prevent our own username from being recreated */
|
|
snprintf(create_role_str, sizeof(create_role_str),
|
|
"CREATE ROLE %s;", os_info.user);
|
|
snprintf(create_role_str_quote, sizeof(create_role_str_quote),
|
|
"CREATE ROLE %s;", quote_identifier(os_info.user));
|
|
|
|
while (fgets(line, sizeof(line), all_dump) != NULL)
|
|
{
|
|
/* switch to db_dump file output? */
|
|
if (current_output == globals_dump && start_of_line &&
|
|
suppressed_username &&
|
|
strncmp(line, "\\connect ", strlen("\\connect ")) == 0)
|
|
current_output = db_dump;
|
|
|
|
/* output unless we are recreating our own username */
|
|
if (current_output != globals_dump || !start_of_line ||
|
|
(strncmp(line, create_role_str, strlen(create_role_str)) != 0 &&
|
|
strncmp(line, create_role_str_quote, strlen(create_role_str_quote)) != 0))
|
|
fputs(line, current_output);
|
|
else
|
|
suppressed_username = true;
|
|
|
|
if (strlen(line) > 0 && line[strlen(line) - 1] == '\n')
|
|
start_of_line = true;
|
|
else
|
|
start_of_line = false;
|
|
}
|
|
|
|
fclose(all_dump);
|
|
fclose(globals_dump);
|
|
fclose(db_dump);
|
|
}
|