Browse Source

global peer tests

pull/163/head
orignal 9 years ago
parent
commit
e250628174
  1. 10
      SSU.cpp
  2. 6
      SSU.h
  3. 5
      SSUSession.cpp

10
SSU.cpp

@ -461,6 +461,16 @@ namespace transport
m_Introducers = newList; m_Introducers = newList;
ScheduleIntroducersUpdateTimer (); ScheduleIntroducersUpdateTimer ();
} }
}
void SSUServer::NewPeerTest (uint32_t nonce)
{
m_PeerTests[nonce] = i2p::util::GetMillisecondsSinceEpoch ();
}
void SSUServer::PeerTestComplete (uint32_t nonce)
{
m_PeerTests.erase (nonce);
} }
} }
} }

6
SSU.h

@ -53,6 +53,9 @@ namespace transport
void AddRelay (uint32_t tag, const boost::asio::ip::udp::endpoint& relay); void AddRelay (uint32_t tag, const boost::asio::ip::udp::endpoint& relay);
std::shared_ptr<SSUSession> FindRelaySession (uint32_t tag); std::shared_ptr<SSUSession> FindRelaySession (uint32_t tag);
void NewPeerTest (uint32_t nonce);
void PeerTestComplete (uint32_t nonce);
private: private:
void Run (); void Run ();
@ -70,7 +73,7 @@ namespace transport
std::set<SSUSession *> FindIntroducers (int maxNumIntroducers); std::set<SSUSession *> FindIntroducers (int maxNumIntroducers);
void ScheduleIntroducersUpdateTimer (); void ScheduleIntroducersUpdateTimer ();
void HandleIntroducersUpdateTimer (const boost::system::error_code& ecode); void HandleIntroducersUpdateTimer (const boost::system::error_code& ecode);
private: private:
bool m_IsRunning; bool m_IsRunning;
@ -84,6 +87,7 @@ namespace transport
std::mutex m_SessionsMutex; std::mutex m_SessionsMutex;
std::map<boost::asio::ip::udp::endpoint, std::shared_ptr<SSUSession> > m_Sessions; std::map<boost::asio::ip::udp::endpoint, std::shared_ptr<SSUSession> > m_Sessions;
std::map<uint32_t, boost::asio::ip::udp::endpoint> m_Relays; // we are introducer std::map<uint32_t, boost::asio::ip::udp::endpoint> m_Relays; // we are introducer
std::map<uint32_t, uint64_t> m_PeerTests; // nonce -> creation time in milliseconds
public: public:
// for HTTP only // for HTTP only

5
SSUSession.cpp

@ -942,7 +942,10 @@ namespace transport
} }
} }
else else
{
LogPrint (eLogDebug, "SSU peer test from Charlie. We are Alice"); LogPrint (eLogDebug, "SSU peer test from Charlie. We are Alice");
m_Server.PeerTestComplete (nonce);
}
} }
} }
@ -989,6 +992,7 @@ namespace transport
void SSUSession::SendPeerTest () void SSUSession::SendPeerTest ()
{ {
// we are Alice
LogPrint (eLogDebug, "SSU sending peer test"); LogPrint (eLogDebug, "SSU sending peer test");
auto address = i2p::context.GetRouterInfo ().GetSSUAddress (); auto address = i2p::context.GetRouterInfo ().GetSSUAddress ();
if (!address) if (!address)
@ -999,6 +1003,7 @@ namespace transport
uint32_t nonce = i2p::context.GetRandomNumberGenerator ().GenerateWord32 (); uint32_t nonce = i2p::context.GetRandomNumberGenerator ().GenerateWord32 ();
if (!nonce) nonce = 1; if (!nonce) nonce = 1;
m_PeerTestNonces.insert (nonce); m_PeerTestNonces.insert (nonce);
m_Server.NewPeerTest (nonce);
SendPeerTest (nonce, 0, 0, address->key, false); // address and port always zero for Alice SendPeerTest (nonce, 0, 0, address->key, false); // address and port always zero for Alice
} }

Loading…
Cancel
Save