|
|
|
@ -15,7 +15,8 @@ namespace client
@@ -15,7 +15,8 @@ namespace client
|
|
|
|
|
LeaseSetDestination::LeaseSetDestination (bool isPublic, const std::map<std::string, std::string> * params): |
|
|
|
|
m_IsRunning (false), m_Thread (nullptr), m_IsPublic (isPublic), |
|
|
|
|
m_PublishReplyToken (0), m_LastSubmissionTime (0), m_PublishConfirmationTimer (m_Service), |
|
|
|
|
m_PublishVerificationTimer (m_Service), m_PublishDelayTimer (m_Service), m_CleanupTimer (m_Service) |
|
|
|
|
m_PublishVerificationTimer (m_Service), m_PublishDelayTimer (m_Service), m_CleanupTimer (m_Service), |
|
|
|
|
m_ReadyCheckTimer(m_Service) |
|
|
|
|
{ |
|
|
|
|
int inLen = DEFAULT_INBOUND_TUNNEL_LENGTH; |
|
|
|
|
int inQty = DEFAULT_INBOUND_TUNNELS_QUANTITY; |
|
|
|
@ -118,7 +119,9 @@ namespace client
@@ -118,7 +119,9 @@ namespace client
|
|
|
|
|
m_CleanupTimer.async_wait (std::bind (&LeaseSetDestination::HandleCleanupTimer, |
|
|
|
|
shared_from_this (), std::placeholders::_1)); |
|
|
|
|
m_Thread = new std::thread (std::bind (&LeaseSetDestination::Run, shared_from_this ())); |
|
|
|
|
|
|
|
|
|
m_ReadyCheckTimer.expires_from_now(boost::posix_time::seconds (1)); |
|
|
|
|
m_ReadyCheckTimer.async_wait(std::bind(&LeaseSetDestination::HandleReadyCheckTimer, |
|
|
|
|
shared_from_this (), std::placeholders::_1)); |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
@ -132,6 +135,7 @@ namespace client
@@ -132,6 +135,7 @@ namespace client
|
|
|
|
|
m_CleanupTimer.cancel (); |
|
|
|
|
m_PublishConfirmationTimer.cancel (); |
|
|
|
|
m_PublishVerificationTimer.cancel (); |
|
|
|
|
m_ReadyCheckTimer.cancel (); |
|
|
|
|
|
|
|
|
|
m_IsRunning = false; |
|
|
|
|
if (m_Pool) |
|
|
|
@ -231,6 +235,11 @@ namespace client
@@ -231,6 +235,11 @@ namespace client
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void LeaseSetDestination::AddReadyCallback(ReadyCallback cb) |
|
|
|
|
{ |
|
|
|
|
m_ReadyCallbacks.push_back(cb); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void LeaseSetDestination::UpdateLeaseSet () |
|
|
|
|
{ |
|
|
|
|
int numTunnels = m_Pool->GetNumInboundTunnels () + 2; // 2 backup tunnels
|
|
|
|
@ -676,6 +685,30 @@ namespace client
@@ -676,6 +685,30 @@ namespace client
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void LeaseSetDestination::HandleReadyCheckTimer(const boost::system::error_code & ec) |
|
|
|
|
{ |
|
|
|
|
if (ec != boost::asio::error::operation_aborted) |
|
|
|
|
{ |
|
|
|
|
// TODO: locking ?
|
|
|
|
|
if(IsReady()) |
|
|
|
|
{ |
|
|
|
|
for (auto & itr : m_ReadyCallbacks) |
|
|
|
|
{ |
|
|
|
|
itr(ec); |
|
|
|
|
} |
|
|
|
|
m_ReadyCallbacks.clear(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
for (auto & itr : m_ReadyCallbacks) |
|
|
|
|
{ |
|
|
|
|
itr(ec); |
|
|
|
|
} |
|
|
|
|
m_ReadyCallbacks.clear(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void LeaseSetDestination::CleanupRemoteLeaseSets () |
|
|
|
|
{ |
|
|
|
|
auto ts = i2p::util::GetMillisecondsSinceEpoch (); |
|
|
|
@ -812,6 +845,8 @@ namespace client
@@ -812,6 +845,8 @@ namespace client
|
|
|
|
|
LogPrint (eLogError, "Destination: request callback is not specified in CreateStream"); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
if(IsReady()) |
|
|
|
|
{ |
|
|
|
|
auto leaseSet = FindLeaseSet (dest); |
|
|
|
|
if (leaseSet) |
|
|
|
|
streamRequestComplete(CreateStream (leaseSet, port)); |
|
|
|
@ -828,6 +863,17 @@ namespace client
@@ -828,6 +863,17 @@ namespace client
|
|
|
|
|
}); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
{ |
|
|
|
|
// call if tunnel is not ready
|
|
|
|
|
AddReadyCallback([&](const boost::system::error_code & ec) { |
|
|
|
|
if(ec) |
|
|
|
|
streamRequestComplete(nullptr); |
|
|
|
|
else |
|
|
|
|
CreateStream(streamRequestComplete, dest, port); |
|
|
|
|
}); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
std::shared_ptr<i2p::stream::Stream> ClientDestination::CreateStream (std::shared_ptr<const i2p::data::LeaseSet> remote, int port) |
|
|
|
|
{ |
|
|
|
|