Browse Source

fixed race condtion #350

pull/352/head
orignal 9 years ago
parent
commit
85830d5076
  1. 22
      Transports.cpp

22
Transports.cpp

@ -69,20 +69,20 @@ namespace transport
std::shared_ptr<i2p::crypto::DHKeys> DHKeysPairSupplier::Acquire () std::shared_ptr<i2p::crypto::DHKeys> DHKeysPairSupplier::Acquire ()
{ {
if (!m_Queue.empty ())
{ {
std::unique_lock<std::mutex> l(m_AcquiredMutex); std::unique_lock<std::mutex> l(m_AcquiredMutex);
auto pair = m_Queue.front (); if (!m_Queue.empty ())
m_Queue.pop (); {
m_Acquired.notify_one (); auto pair = m_Queue.front ();
return pair; m_Queue.pop ();
} m_Acquired.notify_one ();
else // queue is empty, create new return pair;
{ }
auto pair = std::make_shared<i2p::crypto::DHKeys> ();
pair->GenerateKeys ();
return pair;
} }
// queue is empty, create new
auto pair = std::make_shared<i2p::crypto::DHKeys> ();
pair->GenerateKeys ();
return pair;
} }
void DHKeysPairSupplier::Return (std::shared_ptr<i2p::crypto::DHKeys> pair) void DHKeysPairSupplier::Return (std::shared_ptr<i2p::crypto::DHKeys> pair)

Loading…
Cancel
Save