diff --git a/BOB.cpp b/BOB.cpp index af1fb19a..c074b53b 100644 --- a/BOB.cpp +++ b/BOB.cpp @@ -202,9 +202,9 @@ namespace client } BOBCommandSession::BOBCommandSession (BOBCommandChannel& owner): - m_Owner (owner), m_Socket (m_Owner.GetService ()), m_ReceiveBufferOffset (0), - m_IsOpen (true), m_IsQuiet (false), m_InPort (0), m_OutPort (0), - m_CurrentDestination (nullptr) + m_Owner (owner), m_Socket (m_Owner.GetService ()), m_Timer (m_Owner.GetService ()), + m_ReceiveBufferOffset (0), m_IsOpen (true), m_IsQuiet (false), + m_InPort (0), m_OutPort (0), m_CurrentDestination (nullptr) { } @@ -364,9 +364,31 @@ namespace client if (m_OutPort && !m_Address.empty ()) m_CurrentDestination->CreateOutboundTunnel (m_Address, m_OutPort, m_IsQuiet); m_CurrentDestination->Start (); - SendReplyOK ("tunnel starting"); + if (m_CurrentDestination->GetLocalDestination ()->IsReady ()) + SendReplyOK ("tunnel starting"); + else + { + m_Timer.expires_from_now (boost::posix_time::seconds(BOB_SESSION_READINESS_CHECK_INTERVAL)); + m_Timer.async_wait (std::bind (&BOBCommandSession::HandleSessionReadinessCheckTimer, + shared_from_this (), std::placeholders::_1)); + } } + void BOBCommandSession::HandleSessionReadinessCheckTimer (const boost::system::error_code& ecode) + { + if (ecode != boost::asio::error::operation_aborted) + { + if (m_CurrentDestination->GetLocalDestination ()->IsReady ()) + SendReplyOK ("tunnel starting"); + else + { + m_Timer.expires_from_now (boost::posix_time::seconds(BOB_SESSION_READINESS_CHECK_INTERVAL)); + m_Timer.async_wait (std::bind (&BOBCommandSession::HandleSessionReadinessCheckTimer, + shared_from_this (), std::placeholders::_1)); + } + } + } + void BOBCommandSession::StopCommandHandler (const char * operand, size_t len) { auto dest = m_Owner.FindDestination (m_Nickname); diff --git a/BOB.h b/BOB.h index b73e390e..a6a6fa11 100644 --- a/BOB.h +++ b/BOB.h @@ -42,6 +42,8 @@ namespace client const char BOB_REPLY_ERROR[] = "ERROR %s\n"; const char BOB_DATA[] = "NICKNAME %s\n"; + const int BOB_SESSION_READINESS_CHECK_INTERVAL = 5; // in seconds + class BOBI2PTunnel: public I2PService { public: @@ -173,6 +175,7 @@ namespace client void Receive (); void HandleReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred); + void HandleSessionReadinessCheckTimer (const boost::system::error_code& ecode); void Send (size_t len); void HandleSent (const boost::system::error_code& ecode, std::size_t bytes_transferred); @@ -184,6 +187,7 @@ namespace client BOBCommandChannel& m_Owner; boost::asio::ip::tcp::socket m_Socket; + boost::asio::deadline_timer m_Timer; char m_ReceiveBuffer[BOB_COMMAND_BUFFER_SIZE + 1], m_SendBuffer[BOB_COMMAND_BUFFER_SIZE + 1]; size_t m_ReceiveBufferOffset; bool m_IsOpen, m_IsQuiet;