MetzNet
/
dfhack
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'cvuchener/server-accept-fix' into myk_remote_server

develop
Myk Taylor 2023-11-20 00:07:00 -08:00
parent 66120a8c6a 77ce84ca0a
commit ba872902c8
No known key found for this signature in database
1 changed files with 16 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
library/RemoteServer.cpp
Unescape Escape View File

@ -475,38 +475,28 @@ void ServerMainImpl::threadFn(std::promise<bool> promise, int port)
{
ServerMainImpl server{std::move(promise), port};
CActiveSocket *client = nullptr;
server.socket.SetBlocking();
try {
for (int acceptFail = 0; server.socket.IsSocketValid() && acceptFail < 5; acceptFail++)
{
if ((client = server.socket.Accept()) != NULL)
{
while (server.socket.IsSocketValid()) {
if (std::unique_ptr<CActiveSocket> client{server.socket.Accept()}) {
BlockGuard lock;
ServerConnection::Accepted(client);
client = nullptr;
acceptFail = 0;
ServerConnection::Accepted(client.release());
}
else
{
WARN(socket).print("Connection failure: %s (%d of %d)\n", server.socket.DescribeError(), acceptFail + 1, 5);
else switch (server.socket.GetSocketError()) {
case CSimpleSocket::SocketInvalidSocket:
WARN(socket).print("Listening socket invalid, shutting down RemoteServer\n");
server.socket.Close();
break;
case CSimpleSocket::SocketFirewallError:
case CSimpleSocket::SocketProtocolError:
WARN(socket).print("Connection failed: %s\n", server.socket.DescribeError());
break;
default:
break;
}
}
}
catch (BlockedException&) {
if (client)
client->Close();
delete client;
WARN(socket).print("Connection failure: unexpectedly blocked");
}
if (server.socket.IsSocketValid())
{
WARN(socket).print("Too many failed accepts, shutting down RemoteServer\n");
}
else
{
WARN(socket).print("Listening socket invalid, shutting down RemoteServer\n");
catch(BlockedException &) {
}
}