Browse Source

fixed crash on termination

pull/100/head
orignal 10 years ago
parent
commit
81a7253e0f
  1. 19
      NTCPSession.cpp
  2. 8
      Transports.cpp

19
NTCPSession.cpp

@ -35,6 +35,9 @@ namespace ntcp @@ -35,6 +35,9 @@ namespace ntcp
delete m_DHKeysPair;
if (m_NextMessage)
i2p::DeleteI2NPMessage (m_NextMessage);
for (auto it :m_DelayedMessages)
i2p::DeleteI2NPMessage (it);
m_DelayedMessages.clear ();
}
void NTCPSession::CreateAESKey (uint8_t * pubKey, uint8_t * aesKey)
@ -144,6 +147,7 @@ namespace ntcp @@ -144,6 +147,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Couldn't send Phase 1 message: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -160,6 +164,7 @@ namespace ntcp @@ -160,6 +164,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Phase 1 read error: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -215,6 +220,7 @@ namespace ntcp @@ -215,6 +220,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Couldn't send Phase 2 message: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -231,11 +237,14 @@ namespace ntcp @@ -231,11 +237,14 @@ namespace ntcp
if (ecode)
{
LogPrint ("Phase 2 read error: ", ecode.message (), ". Wrong ident assumed");
if (ecode != boost::asio::error::operation_aborted)
{
GetRemoteRouterInfo ().SetUnreachable (true); // this RouterInfo is not valid
i2p::transports.ReuseDHKeysPair (m_DHKeysPair);
m_DHKeysPair = nullptr;
Terminate ();
}
}
else
{
LogPrint ("Phase 2 received: ", bytes_transferred);
@ -291,6 +300,7 @@ namespace ntcp @@ -291,6 +300,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Couldn't send Phase 3 message: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -307,6 +317,7 @@ namespace ntcp @@ -307,6 +317,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Phase 3 read error: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -356,6 +367,7 @@ namespace ntcp @@ -356,6 +367,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Couldn't send Phase 4 message: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -373,9 +385,12 @@ namespace ntcp @@ -373,9 +385,12 @@ namespace ntcp
if (ecode)
{
LogPrint ("Phase 4 read error: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
{
GetRemoteRouterInfo ().SetUnreachable (true); // this router doesn't like us
Terminate ();
}
}
else
{
LogPrint ("Phase 4 received: ", bytes_transferred);
@ -418,6 +433,7 @@ namespace ntcp @@ -418,6 +433,7 @@ namespace ntcp
if (ecode)
{
LogPrint ("Read error: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
Terminate ();
}
else
@ -604,9 +620,12 @@ namespace ntcp @@ -604,9 +620,12 @@ namespace ntcp
if (ecode)
{
LogPrint ("Connect error: ", ecode.message ());
if (ecode != boost::asio::error::operation_aborted)
{
GetRemoteRouterInfo ().SetUnreachable (true);
Terminate ();
}
}
else
{
LogPrint ("Connected");

8
Transports.cpp

@ -135,6 +135,7 @@ namespace i2p @@ -135,6 +135,7 @@ namespace i2p
{
m_SSUServer->Stop ();
delete m_SSUServer;
m_SSUServer = nullptr;
}
for (auto session: m_NTCPSessions)
@ -149,7 +150,7 @@ namespace i2p @@ -149,7 +150,7 @@ namespace i2p
{
m_Thread->join ();
delete m_Thread;
m_Thread = 0;
m_Thread = nullptr;
}
}
@ -188,14 +189,15 @@ namespace i2p @@ -188,14 +189,15 @@ namespace i2p
conn->ServerLogin ();
}
else
{
delete conn;
}
if (error != boost::asio::error::operation_aborted)
{
conn = new i2p::ntcp::NTCPServerConnection (m_Service);
m_NTCPAcceptor->async_accept(conn->GetSocket (), boost::bind (&Transports::HandleAccept, this,
conn, boost::asio::placeholders::error));
}
}
i2p::ntcp::NTCPSession * Transports::GetNextNTCPSession ()
{

Loading…
Cancel
Save