Browse Source

fixed race consition

pull/93/head
orignal 10 years ago
parent
commit
168e20053a
  1. 5
      Garlic.cpp
  2. 1
      Garlic.h
  3. 8
      Tunnel.cpp
  4. 2
      Tunnel.h

5
Garlic.cpp

@ -323,7 +323,7 @@ namespace garlic
void GarlicRouting::DeliveryStatusSent (GarlicRoutingSession * session, uint32_t msgID) void GarlicRouting::DeliveryStatusSent (GarlicRoutingSession * session, uint32_t msgID)
{ {
std::unique_lock<std::mutex> l(m_SessionsMutex); std::unique_lock<std::mutex> l(m_CreatedSessionsMutex);
m_CreatedSessions[msgID] = session; m_CreatedSessions[msgID] = session;
} }
@ -480,6 +480,8 @@ namespace garlic
{ {
I2NPDeliveryStatusMsg * deliveryStatus = (I2NPDeliveryStatusMsg *)msg->GetPayload (); I2NPDeliveryStatusMsg * deliveryStatus = (I2NPDeliveryStatusMsg *)msg->GetPayload ();
uint32_t msgID = be32toh (deliveryStatus->msgID); uint32_t msgID = be32toh (deliveryStatus->msgID);
{
std::unique_lock<std::mutex> l(m_CreatedSessionsMutex);
auto it = m_CreatedSessions.find (msgID); auto it = m_CreatedSessions.find (msgID);
if (it != m_CreatedSessions.end ()) if (it != m_CreatedSessions.end ())
{ {
@ -487,6 +489,7 @@ namespace garlic
m_CreatedSessions.erase (it); m_CreatedSessions.erase (it);
LogPrint ("Garlic message ", msgID, " acknowledged"); LogPrint ("Garlic message ", msgID, " acknowledged");
} }
}
DeleteI2NPMessage (msg); DeleteI2NPMessage (msg);
} }

1
Garlic.h

@ -125,6 +125,7 @@ namespace garlic
// outgoing sessions // outgoing sessions
std::mutex m_SessionsMutex; std::mutex m_SessionsMutex;
std::map<i2p::data::IdentHash, GarlicRoutingSession *> m_Sessions; std::map<i2p::data::IdentHash, GarlicRoutingSession *> m_Sessions;
std::mutex m_CreatedSessionsMutex;
std::map<uint32_t, GarlicRoutingSession *> m_CreatedSessions; // msgID -> session std::map<uint32_t, GarlicRoutingSession *> m_CreatedSessions; // msgID -> session
// incoming session // incoming session
// multiple tags refer to one decyption // multiple tags refer to one decyption

8
Tunnel.cpp

@ -396,6 +396,8 @@ namespace tunnel
void Tunnels::ManageOutboundTunnels () void Tunnels::ManageOutboundTunnels ()
{ {
uint64_t ts = i2p::util::GetSecondsSinceEpoch (); uint64_t ts = i2p::util::GetSecondsSinceEpoch ();
{
std::unique_lock<std::mutex> l(m_OutboundTunnelsMutex);
for (auto it = m_OutboundTunnels.begin (); it != m_OutboundTunnels.end ();) for (auto it = m_OutboundTunnels.begin (); it != m_OutboundTunnels.end ();)
{ {
if (ts > (*it)->GetCreationTime () + TUNNEL_EXPIRATION_TIMEOUT) if (ts > (*it)->GetCreationTime () + TUNNEL_EXPIRATION_TIMEOUT)
@ -414,6 +416,7 @@ namespace tunnel
it++; it++;
} }
} }
}
if (m_OutboundTunnels.size () < 5) if (m_OutboundTunnels.size () < 5)
{ {
@ -433,6 +436,8 @@ namespace tunnel
void Tunnels::ManageInboundTunnels () void Tunnels::ManageInboundTunnels ()
{ {
uint64_t ts = i2p::util::GetSecondsSinceEpoch (); uint64_t ts = i2p::util::GetSecondsSinceEpoch ();
{
std::unique_lock<std::mutex> l(m_InboundTunnelsMutex);
for (auto it = m_InboundTunnels.begin (); it != m_InboundTunnels.end ();) for (auto it = m_InboundTunnels.begin (); it != m_InboundTunnels.end ();)
{ {
if (ts > it->second->GetCreationTime () + TUNNEL_EXPIRATION_TIMEOUT) if (ts > it->second->GetCreationTime () + TUNNEL_EXPIRATION_TIMEOUT)
@ -451,6 +456,7 @@ namespace tunnel
it++; it++;
} }
} }
}
if (m_InboundTunnels.empty ()) if (m_InboundTunnels.empty ())
{ {
@ -516,6 +522,7 @@ namespace tunnel
void Tunnels::AddOutboundTunnel (OutboundTunnel * newTunnel) void Tunnels::AddOutboundTunnel (OutboundTunnel * newTunnel)
{ {
std::unique_lock<std::mutex> l(m_OutboundTunnelsMutex);
m_OutboundTunnels.push_back (newTunnel); m_OutboundTunnels.push_back (newTunnel);
auto pool = newTunnel->GetTunnelPool (); auto pool = newTunnel->GetTunnelPool ();
if (pool) if (pool)
@ -524,6 +531,7 @@ namespace tunnel
void Tunnels::AddInboundTunnel (InboundTunnel * newTunnel) void Tunnels::AddInboundTunnel (InboundTunnel * newTunnel)
{ {
std::unique_lock<std::mutex> l(m_InboundTunnelsMutex);
m_InboundTunnels[newTunnel->GetTunnelID ()] = newTunnel; m_InboundTunnels[newTunnel->GetTunnelID ()] = newTunnel;
auto pool = newTunnel->GetTunnelPool (); auto pool = newTunnel->GetTunnelPool ();
if (!pool) if (!pool)

2
Tunnel.h

@ -148,7 +148,9 @@ namespace tunnel
uint32_t m_NextReplyMsgID; // TODO: make it random later uint32_t m_NextReplyMsgID; // TODO: make it random later
std::thread * m_Thread; std::thread * m_Thread;
std::map<uint32_t, Tunnel *> m_PendingTunnels; // by replyMsgID std::map<uint32_t, Tunnel *> m_PendingTunnels; // by replyMsgID
std::mutex m_InboundTunnelsMutex;
std::map<uint32_t, InboundTunnel *> m_InboundTunnels; std::map<uint32_t, InboundTunnel *> m_InboundTunnels;
std::mutex m_OutboundTunnelsMutex;
std::list<OutboundTunnel *> m_OutboundTunnels; std::list<OutboundTunnel *> m_OutboundTunnels;
std::map<uint32_t, TransitTunnel *> m_TransitTunnels; std::map<uint32_t, TransitTunnel *> m_TransitTunnels;
std::map<i2p::data::IdentHash, TunnelPool *> m_Pools; std::map<i2p::data::IdentHash, TunnelPool *> m_Pools;

Loading…
Cancel
Save