mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2024-11-08 01:10:40 +00:00
113 lines
3.6 KiB
C++
113 lines
3.6 KiB
C++
#ifndef NETDB_H__
|
|
#define NETDB_H__
|
|
|
|
#include <inttypes.h>
|
|
#include <set>
|
|
#include <map>
|
|
#include <list>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <boost/filesystem.hpp>
|
|
#include "Base.h"
|
|
#include "Queue.h"
|
|
#include "I2NPProtocol.h"
|
|
#include "RouterInfo.h"
|
|
#include "LeaseSet.h"
|
|
#include "Tunnel.h"
|
|
#include "TunnelPool.h"
|
|
#include "Reseed.h"
|
|
#include "NetDbRequests.h"
|
|
#include "Family.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace data
|
|
{
|
|
|
|
class NetDb
|
|
{
|
|
public:
|
|
|
|
NetDb ();
|
|
~NetDb ();
|
|
|
|
void Start ();
|
|
void Stop ();
|
|
|
|
bool AddRouterInfo (const uint8_t * buf, int len);
|
|
bool AddRouterInfo (const IdentHash& ident, const uint8_t * buf, int len);
|
|
bool 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;
|
|
std::shared_ptr<LeaseSet> FindLeaseSet (const IdentHash& destination) const;
|
|
std::shared_ptr<RouterProfile> FindRouterProfile (const IdentHash& ident) const;
|
|
|
|
void RequestDestination (const IdentHash& destination, RequestedDestination::RequestComplete requestComplete = nullptr);
|
|
|
|
void HandleDatabaseStoreMsg (std::shared_ptr<const I2NPMessage> msg);
|
|
void HandleDatabaseSearchReplyMsg (std::shared_ptr<const I2NPMessage> msg);
|
|
void HandleDatabaseLookupMsg (std::shared_ptr<const I2NPMessage> msg);
|
|
|
|
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;
|
|
std::shared_ptr<const RouterInfo> GetRandomPeerTestRouter () const;
|
|
std::shared_ptr<const RouterInfo> GetRandomIntroducer () const;
|
|
std::shared_ptr<const RouterInfo> GetClosestFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded, bool closeThanUsOnly = false) const;
|
|
std::vector<IdentHash> GetClosestFloodfills (const IdentHash& destination, size_t num,
|
|
std::set<IdentHash>& excluded) const;
|
|
std::shared_ptr<const RouterInfo> GetClosestNonFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded) const;
|
|
void SetUnreachable (const IdentHash& ident, bool unreachable);
|
|
|
|
void PostI2NPMsg (std::shared_ptr<const I2NPMessage> msg);
|
|
|
|
void Reseed ();
|
|
Families& GetFamilies () { return m_Families; };
|
|
|
|
// for web interface
|
|
int GetNumRouters () const { return m_RouterInfos.size (); };
|
|
int GetNumFloodfills () const { return m_Floodfills.size (); };
|
|
int GetNumLeaseSets () const { return m_LeaseSets.size (); };
|
|
|
|
private:
|
|
|
|
bool CreateNetDb(boost::filesystem::path directory);
|
|
void Load ();
|
|
void SaveUpdated ();
|
|
void Run (); // exploratory thread
|
|
void Explore (int numDestinations);
|
|
void Publish ();
|
|
void ManageLeaseSets ();
|
|
void ManageRequests ();
|
|
|
|
template<typename Filter>
|
|
std::shared_ptr<const RouterInfo> GetRandomRouter (Filter filter) const;
|
|
|
|
private:
|
|
|
|
std::map<IdentHash, std::shared_ptr<LeaseSet> > m_LeaseSets;
|
|
mutable std::mutex m_RouterInfosMutex;
|
|
std::map<IdentHash, std::shared_ptr<RouterInfo> > m_RouterInfos;
|
|
mutable std::mutex m_FloodfillsMutex;
|
|
std::list<std::shared_ptr<RouterInfo> > m_Floodfills;
|
|
|
|
bool m_IsRunning;
|
|
std::thread * m_Thread;
|
|
i2p::util::Queue<std::shared_ptr<const I2NPMessage> > m_Queue; // of I2NPDatabaseStoreMsg
|
|
|
|
GzipInflator m_Inflator;
|
|
Reseeder * m_Reseeder;
|
|
Families m_Families;
|
|
|
|
friend class NetDbRequests;
|
|
NetDbRequests m_Requests;
|
|
|
|
static const char m_NetDbPath[];
|
|
};
|
|
|
|
extern NetDb netdb;
|
|
}
|
|
}
|
|
|
|
#endif
|