i2pd/NetDb.h

134 lines
4.6 KiB
C
Raw Normal View History

2013-11-13 12:59:21 +00:00
#ifndef NETDB_H__
#define NETDB_H__
#include <inttypes.h>
2014-01-05 14:53:44 +00:00
#include <set>
2013-11-13 12:59:21 +00:00
#include <map>
2014-11-21 18:02:46 +00:00
#include <list>
2013-11-13 12:59:21 +00:00
#include <string>
2013-11-19 01:37:38 +00:00
#include <thread>
2014-10-06 01:59:05 +00:00
#include <mutex>
2014-02-01 20:57:46 +00:00
#include <boost/filesystem.hpp>
2013-11-20 12:46:09 +00:00
#include "Queue.h"
#include "I2NPProtocol.h"
2013-11-13 12:59:21 +00:00
#include "RouterInfo.h"
#include "LeaseSet.h"
2013-12-25 17:19:46 +00:00
#include "Tunnel.h"
2014-08-20 15:12:53 +00:00
#include "TunnelPool.h"
2015-01-19 18:57:37 +00:00
#include "Reseed.h"
2013-11-13 12:59:21 +00:00
namespace i2p
{
namespace data
{
2014-01-05 14:53:44 +00:00
class RequestedDestination
2015-01-14 21:11:09 +00:00
{
2014-01-05 14:53:44 +00:00
public:
2015-01-14 21:11:09 +00:00
typedef std::function<void (std::shared_ptr<RouterInfo>)> RequestComplete;
RequestedDestination (const IdentHash& destination, bool isExploratory = false):
m_Destination (destination), m_IsExploratory (isExploratory), m_CreationTime (0) {};
2015-01-14 21:11:09 +00:00
~RequestedDestination () { if (m_RequestComplete) m_RequestComplete (nullptr); };
2014-01-05 14:53:44 +00:00
const IdentHash& GetDestination () const { return m_Destination; };
int GetNumExcludedPeers () const { return m_ExcludedPeers.size (); };
const std::set<IdentHash>& GetExcludedPeers () { return m_ExcludedPeers; };
void ClearExcludedPeers ();
2014-01-05 14:53:44 +00:00
bool IsExploratory () const { return m_IsExploratory; };
bool IsExcluded (const IdentHash& ident) const { return m_ExcludedPeers.count (ident); };
2014-07-07 19:35:42 +00:00
uint64_t GetCreationTime () const { return m_CreationTime; };
2015-01-27 19:55:46 +00:00
I2NPMessage * CreateRequestMessage (std::shared_ptr<const RouterInfo>, std::shared_ptr<const i2p::tunnel::InboundTunnel> replyTunnel);
I2NPMessage * CreateRequestMessage (const IdentHash& floodfill);
2015-01-14 21:11:09 +00:00
void SetRequestComplete (const RequestComplete& requestComplete) { m_RequestComplete = requestComplete; };
2015-02-03 21:14:33 +00:00
bool IsRequestComplete () const { return m_RequestComplete != nullptr; };
2015-01-14 21:11:09 +00:00
void Success (std::shared_ptr<RouterInfo> r);
void Fail ();
2014-01-05 14:53:44 +00:00
private:
IdentHash m_Destination;
bool m_IsExploratory;
2014-01-05 14:53:44 +00:00
std::set<IdentHash> m_ExcludedPeers;
2014-07-07 19:35:42 +00:00
uint64_t m_CreationTime;
2015-01-14 21:11:09 +00:00
RequestComplete m_RequestComplete;
2014-01-05 14:53:44 +00:00
};
2013-11-13 12:59:21 +00:00
class NetDb
{
public:
NetDb ();
~NetDb ();
2013-11-19 01:37:38 +00:00
void Start ();
void Stop ();
2013-11-13 12:59:21 +00:00
2014-12-11 20:41:04 +00:00
void AddRouterInfo (const uint8_t * buf, int len);
void AddRouterInfo (const IdentHash& ident, const uint8_t * buf, int len);
2015-02-05 23:53:43 +00:00
void AddLeaseSet (const IdentHash& ident, const uint8_t * buf, int len, std::shared_ptr<i2p::tunnel::InboundTunnel> from);
std::shared_ptr<RouterInfo> FindRouter (const IdentHash& ident) const;
2015-01-27 16:27:58 +00:00
std::shared_ptr<LeaseSet> FindLeaseSet (const IdentHash& destination) const;
2014-03-13 20:26:04 +00:00
2015-01-14 21:11:09 +00:00
void RequestDestination (const IdentHash& destination, RequestedDestination::RequestComplete requestComplete = nullptr);
2014-08-20 15:12:53 +00:00
void HandleDatabaseStoreMsg (I2NPMessage * msg);
void HandleDatabaseSearchReplyMsg (I2NPMessage * msg);
void HandleDatabaseLookupMsg (I2NPMessage * msg);
2014-11-20 21:20:02 +00:00
std::shared_ptr<const RouterInfo> GetRandomRouter () const;
std::shared_ptr<const RouterInfo> GetRandomRouter (std::shared_ptr<const RouterInfo> compatibleWith) const;
std::shared_ptr<const RouterInfo> GetHighBandwidthRandomRouter (std::shared_ptr<const RouterInfo> compatibleWith) const;
2014-11-28 18:01:35 +00:00
std::shared_ptr<const RouterInfo> GetClosestFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded) const;
2015-02-03 03:34:55 +00:00
std::shared_ptr<const RouterInfo> GetClosestNonFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded) const;
2014-10-24 19:39:53 +00:00
void SetUnreachable (const IdentHash& ident, bool unreachable);
void PostI2NPMsg (I2NPMessage * msg);
2015-01-19 18:57:37 +00:00
void Reseed ();
// for web interface
int GetNumRouters () const { return m_RouterInfos.size (); };
int GetNumFloodfills () const { return m_Floodfills.size (); };
int GetNumLeaseSets () const { return m_LeaseSets.size (); };
2013-11-13 12:59:21 +00:00
private:
2014-02-01 15:10:15 +00:00
bool CreateNetDb(boost::filesystem::path directory);
2013-11-13 12:59:21 +00:00
void Load (const char * directory);
2013-11-20 12:46:09 +00:00
void SaveUpdated (const char * directory);
2013-11-19 01:37:38 +00:00
void Run (); // exploratory thread
2014-12-24 16:20:38 +00:00
void Explore (int numDestinations);
2014-02-13 03:02:39 +00:00
void Publish ();
2014-07-31 16:59:43 +00:00
void ManageLeaseSets ();
2014-12-24 16:20:38 +00:00
void ManageRequests ();
2014-01-05 14:53:44 +00:00
2014-09-25 01:45:19 +00:00
template<typename Filter>
2014-11-20 21:20:02 +00:00
std::shared_ptr<const RouterInfo> GetRandomRouter (Filter filter) const;
2014-07-03 18:47:12 +00:00
2013-11-13 12:59:21 +00:00
private:
2015-01-27 16:27:58 +00:00
std::map<IdentHash, std::shared_ptr<LeaseSet> > m_LeaseSets;
2014-11-21 18:29:19 +00:00
mutable std::mutex m_RouterInfosMutex;
2014-11-18 16:08:10 +00:00
std::map<IdentHash, std::shared_ptr<RouterInfo> > m_RouterInfos;
2014-10-06 01:59:05 +00:00
mutable std::mutex m_FloodfillsMutex;
2014-11-21 18:02:46 +00:00
std::list<std::shared_ptr<RouterInfo> > m_Floodfills;
2014-09-13 16:39:02 +00:00
std::mutex m_RequestedDestinationsMutex;
std::map<IdentHash, std::unique_ptr<RequestedDestination> > m_RequestedDestinations;
2014-01-05 14:53:44 +00:00
2013-11-19 01:37:38 +00:00
bool m_IsRunning;
std::thread * m_Thread;
2013-11-20 12:46:09 +00:00
i2p::util::Queue<I2NPMessage> m_Queue; // of I2NPDatabaseStoreMsg
2014-02-01 20:57:46 +00:00
2015-01-19 18:57:37 +00:00
Reseeder * m_Reseeder;
2014-02-01 20:57:46 +00:00
static const char m_NetDbPath[];
2013-11-13 12:59:21 +00:00
};
extern NetDb netdb;
}
}
#endif