mirror of https://github.com/PurpleI2P/i2pd.git
I2P: End-to-End encrypted and anonymous Internet
https://i2pd.website/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.6 KiB
111 lines
3.6 KiB
#ifndef NETDB_H__ |
|
#define NETDB_H__ |
|
|
|
#include <inttypes.h> |
|
#include <set> |
|
#include <map> |
|
#include <list> |
|
#include <string> |
|
#include <thread> |
|
#include <mutex> |
|
#include "Base.h" |
|
#include "FS.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: |
|
|
|
void Load (); |
|
bool LoadRouterInfo (const std::string & path); |
|
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; |
|
uint64_t m_LastLoad; |
|
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; |
|
i2p::fs::HashedStorage m_Storage; |
|
|
|
friend class NetDbRequests; |
|
NetDbRequests m_Requests; |
|
}; |
|
|
|
extern NetDb netdb; |
|
} |
|
} |
|
|
|
#endif
|
|
|