Manipulations of remote_endpoint must be done in a try/catch

block to avoid race conditions.
This commit is contained in:
JoelKatz
2013-06-07 05:07:56 -07:00
parent 4492f0a1d2
commit 7643f3673c
3 changed files with 31 additions and 5 deletions

View File

@@ -278,9 +278,21 @@ void Peer::handleConnect(const boost::system::error_code& error, boost::asio::ip
// - We don't bother remembering the inbound IP or port. Only useful for debugging.
void Peer::connected(const boost::system::error_code& error)
{
boost::asio::ip::tcp::endpoint ep = getSocket().remote_endpoint();
int iPort = ep.port();
std::string strIp = ep.address().to_string();
boost::asio::ip::tcp::endpoint ep;
int iPort;
std::string strIp;
try
{
ep = getSocket().remote_endpoint();
iPort = ep.port();
strIp = ep.address().to_string();
}
catch (...)
{
detach("edc", false);
return;
}
mClientConnect = false;
mIpPortConnect = make_pair(strIp, iPort);

View File

@@ -50,8 +50,16 @@ void RPCDoor::handleConnect(RPCServer::pointer new_connection,
if (!error)
{
// Restrict callers by IP
if (!isClientAllowed(new_connection->getSocket().remote_endpoint().address().to_string()))
try
{
if (!isClientAllowed(new_connection->getSocket().remote_endpoint().address().to_string()))
{
startListening();
return;
}
}
catch (...)
{ // client may have disconnected
startListening();
return;
}

View File

@@ -142,7 +142,13 @@ public:
{
boost::mutex::scoped_lock sl(mMapLock);
mMap[cpClient] = boost::make_shared< WSConnection<endpoint_type> >(this, cpClient);
try
{
mMap[cpClient] = boost::make_shared< WSConnection<endpoint_type> >(this, cpClient);
}
catch (...)
{
}
}
void on_pong(connection_ptr cpClient, std::string data)