Browse Source

SSU timeout

pull/60/head
orignal 11 years ago
parent
commit
c30dd2639a
  1. 4
      NTCPSession.cpp
  2. 2
      NTCPSession.h
  3. 19
      SSU.cpp
  4. 4
      SSU.h

4
NTCPSession.cpp

@ -519,7 +519,7 @@ namespace ntcp @@ -519,7 +519,7 @@ namespace ntcp
void NTCPSession::ScheduleTermination ()
{
m_TerminationTimer.cancel ();
m_TerminationTimer.expires_from_now (boost::posix_time::seconds(TERMINATION_TIMEOUT));
m_TerminationTimer.expires_from_now (boost::posix_time::seconds(NTCP_TERMINATION_TIMEOUT));
m_TerminationTimer.async_wait (boost::bind (&NTCPSession::HandleTerminationTimer,
this, boost::asio::placeholders::error));
}
@ -528,7 +528,7 @@ namespace ntcp @@ -528,7 +528,7 @@ namespace ntcp
{
if (ecode != boost::asio::error::operation_aborted)
{
LogPrint ("No activity fo ", TERMINATION_TIMEOUT, " seconds");
LogPrint ("No activity fo ", NTCP_TERMINATION_TIMEOUT, " seconds");
m_Socket.close ();
}
}

2
NTCPSession.h

@ -61,7 +61,7 @@ namespace ntcp @@ -61,7 +61,7 @@ namespace ntcp
#pragma pack()
const int TERMINATION_TIMEOUT = 120; // 2 minutes
const int NTCP_TERMINATION_TIMEOUT = 120; // 2 minutes
class NTCPSession
{
public:

19
SSU.cpp

@ -546,6 +546,7 @@ namespace ssu @@ -546,6 +546,7 @@ namespace ssu
}
if (m_PeerTest)
SendPeerTest ();
ScheduleTermination ();
}
void SSUSession::Failed ()
@ -558,6 +559,23 @@ namespace ssu @@ -558,6 +559,23 @@ namespace ssu
}
}
void SSUSession::ScheduleTermination ()
{
m_Timer.cancel ();
m_Timer.expires_from_now (boost::posix_time::seconds(SSU_TERMINATION_TIMEOUT));
m_Timer.async_wait (boost::bind (&SSUSession::HandleTerminationTimer,
this, boost::asio::placeholders::error));
}
void SSUSession::HandleTerminationTimer (const boost::system::error_code& ecode)
{
if (ecode != boost::asio::error::operation_aborted)
{
LogPrint ("SSU no activity fo ", SSU_TERMINATION_TIMEOUT, " seconds");
Close ();
}
}
const uint8_t * SSUSession::GetIntroKey () const
{
if (m_RemoteRouter)
@ -587,6 +605,7 @@ namespace ssu @@ -587,6 +605,7 @@ namespace ssu
void SSUSession::ProcessData (uint8_t * buf, size_t len)
{
ScheduleTermination ();
//uint8_t * start = buf;
uint8_t flag = *buf;
buf++;

4
SSU.h

@ -30,6 +30,7 @@ namespace ssu @@ -30,6 +30,7 @@ namespace ssu
const size_t SSU_MTU = 1484;
const int SSU_CONNECT_TIMEOUT = 5; // 5 seconds
const int SSU_TERMINATION_TIMEOUT = 270; // 4.5 minutes
// payload types (4 bits)
const uint8_t PAYLOAD_TYPE_SESSION_REQUEST = 0;
@ -115,6 +116,9 @@ namespace ssu @@ -115,6 +116,9 @@ namespace ssu
bool Validate (uint8_t * buf, size_t len, const uint8_t * macKey);
const uint8_t * GetIntroKey () const;
void ScheduleTermination ();
void HandleTerminationTimer (const boost::system::error_code& ecode);
private:
SSUServer& m_Server;

Loading…
Cancel
Save