On Windows, close the client socket explicitly during backend shutdown.

It turns out that this is necessary to keep Winsock from dropping any
not-yet-sent data, such as an error message explaining the reason for
process termination.  It's pretty weird that the implicit close done
by the kernel acts differently from an explicit close, but it's hard
to argue with experimental results.

Independently submitted by Alexander Lakhin and Lars Kanis (comments
by me, though).  Back-patch to all supported branches.

Discussion: https://postgr.es/m/90b34057-4176-7bb0-0dbb-9822a5f6425b@greiz-reinsdorf.de
Discussion: https://postgr.es/m/16678-253e48d34dc0c376@postgresql.org
This commit is contained in:
Tom Lane 2021-12-02 17:14:43 -05:00
parent a7da419810
commit 6051857fc9
1 changed files with 20 additions and 7 deletions

View File

@ -277,15 +277,28 @@ socket_close(int code, Datum arg)
secure_close(MyProcPort);
/*
* Formerly we did an explicit close() here, but it seems better to
* leave the socket open until the process dies. This allows clients
* to perform a "synchronous close" if they care --- wait till the
* transport layer reports connection closure, and you can be sure the
* backend has exited.
* On most platforms, we leave the socket open until the process dies.
* This allows clients to perform a "synchronous close" if they care
* --- wait till the transport layer reports connection closure, and
* you can be sure the backend has exited. Saves a kernel call, too.
*
* We do set sock to PGINVALID_SOCKET to prevent any further I/O,
* though.
* However, that does not work on Windows: if the kernel closes the
* socket it will invoke an "abortive shutdown" that discards any data
* not yet sent to the client. (This is a flat-out violation of the
* TCP RFCs, but count on Microsoft not to care about that.) To get
* the spec-compliant "graceful shutdown" behavior, we must invoke
* closesocket() explicitly.
*
* This code runs late enough during process shutdown that we should
* have finished all externally-visible shutdown activities, so that
* in principle it's good enough to act as a synchronous close on
* Windows too. But it's a lot more fragile than the other way.
*/
#ifdef WIN32
closesocket(MyProcPort->sock);
#endif
/* In any case, set sock to PGINVALID_SOCKET to prevent further I/O */
MyProcPort->sock = PGINVALID_SOCKET;
}
}