Improve Accept error handling in RemoteServer

Also makes sure the socket is blocking to avoid EWOULDBLOCK errors.
develop
Clément Vuchener 2023-11-19 14:46:14 +01:00
parent b078036ac6
commit 77ce84ca0a
1 changed files with 16 additions and 23 deletions

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