mirror/postgresql
2
0
Fork 0
mirror of https://git.postgresql.org/git/postgresql.git synced 2024-12-27 08:39:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

pg_receivewal, pg_recvlogical: allow canceling initial password prompt.

Browse Source 
Previously it was impossible to terminate these programs via control-C
while they were prompting for a password.  We can fix that trivially
for their initial password prompts, by moving setup of the SIGINT
handler from just before to just after their initial GetConnection()
calls.

This fix doesn't permit escaping out of later re-prompts, but those
should be exceedingly rare, since the user's password or the server's
authentication setup would have to have changed meanwhile.  We
considered applying a fix similar to commit 46d665bc2, but that
seemed more complicated than it'd be worth.  Moreover, this way is
back-patchable, which that wasn't.

The misbehavior exists in all supported versions, so back-patch to all.

Tom Lane and Nathan Bossart

Discussion: https://postgr.es/m/747443.1635536754@sss.pgh.pa.us
This commit is contained in:
Tom Lane 2021-11-21 14:13:35 -05:00
parent 92e70796e9
commit 282b6d00ab
2 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/bin/pg_basebackup/pg_receivewal.c
View File

@ -917,10 +917,6 @@ main(int argc, char **argv)
close_destination_dir(dir, basedir);
}
#ifndef WIN32
pqsignal(SIGINT, sigint_handler);
#endif
/*
* Obtain a connection before doing anything.
*/
@ -930,6 +926,14 @@ main(int argc, char **argv)
exit(1);
atexit(disconnect_atexit);
/*
* Trap signals. (Don't do this until after the initial password prompt,
* if one is needed, in GetConnection.)
*/
#ifndef WIN32
pqsignal(SIGINT, sigint_handler);
#endif
/*
* Run IDENTIFY_SYSTEM to make sure we've successfully have established a
* replication connection and haven't connected using a database specific

23
src/bin/pg_basebackup/pg_recvlogical.c
View File

@ -216,8 +216,6 @@ StreamLogicalLog(void)
output_written_lsn = InvalidXLogRecPtr;
output_fsync_lsn = InvalidXLogRecPtr;
query = createPQExpBuffer();
/*
* Connect in replication mode to the server
*/
@ -236,6 +234,7 @@ StreamLogicalLog(void)
replication_slot);
/* Initiate the replication stream at specified location */
query = createPQExpBuffer();
appendPQExpBuffer(query, "START_REPLICATION SLOT \"%s\" LOGICAL %X/%X",
replication_slot, LSN_FORMAT_ARGS(startpos));
@ -932,16 +931,9 @@ main(int argc, char **argv)
exit(1);
}
#ifndef WIN32
pqsignal(SIGINT, sigint_handler);
pqsignal(SIGHUP, sighup_handler);
#endif
/*
* Obtain a connection to server. This is not really necessary but it
* helps to get more precise error messages about authentication, required
* GUC parameters and such.
* Obtain a connection to server. Notably, if we need a password, we want
* to collect it from the user immediately.
*/
conn = GetConnection();
if (!conn)
@ -949,6 +941,15 @@ main(int argc, char **argv)
exit(1);
atexit(disconnect_atexit);
/*
* Trap signals. (Don't do this until after the initial password prompt,
* if one is needed, in GetConnection.)
*/
#ifndef WIN32
pqsignal(SIGINT, sigint_handler);
pqsignal(SIGHUP, sighup_handler);
#endif
/*
* Run IDENTIFY_SYSTEM to make sure we connected using a database specific
* replication connection.