#ifndef NETDB_H__ #define NETDB_H__ #include #include #include #include #include #include #include #include #include "Queue.h" #include "I2NPProtocol.h" #include "RouterInfo.h" #include "LeaseSet.h" #include "Tunnel.h" #include "TunnelPool.h" #include "AddressBook.h" namespace i2p { namespace data { class RequestedDestination { public: RequestedDestination (const IdentHash& destination, bool isLeaseSet, bool isExploratory = false, i2p::tunnel::TunnelPool * pool = nullptr): m_Destination (destination), m_IsLeaseSet (isLeaseSet), m_IsExploratory (isExploratory), m_Pool (pool), m_LastRouter (nullptr), m_CreationTime (0) {}; const IdentHash& GetDestination () const { return m_Destination; }; int GetNumExcludedPeers () const { return m_ExcludedPeers.size (); }; const std::set& GetExcludedPeers () { return m_ExcludedPeers; }; void ClearExcludedPeers (); const RouterInfo * GetLastRouter () const { return m_LastRouter; }; i2p::tunnel::TunnelPool * GetTunnelPool () { return m_Pool; }; bool IsExploratory () const { return m_IsExploratory; }; bool IsLeaseSet () const { return m_IsLeaseSet; }; bool IsExcluded (const IdentHash& ident) const { return m_ExcludedPeers.count (ident); }; uint64_t GetCreationTime () const { return m_CreationTime; }; I2NPMessage * CreateRequestMessage (const RouterInfo * router, const i2p::tunnel::InboundTunnel * replyTunnel); I2NPMessage * CreateRequestMessage (const IdentHash& floodfill); private: IdentHash m_Destination; bool m_IsLeaseSet, m_IsExploratory; i2p::tunnel::TunnelPool * m_Pool; std::set m_ExcludedPeers; const RouterInfo * m_LastRouter; uint64_t m_CreationTime; }; class NetDb { public: NetDb (); ~NetDb (); void Start (); void Stop (); void AddRouterInfo (const IdentHash& ident, const uint8_t * buf, int len); void AddLeaseSet (const IdentHash& ident, const uint8_t * buf, int len, i2p::tunnel::InboundTunnel * from); RouterInfo * FindRouter (const IdentHash& ident) const; LeaseSet * FindLeaseSet (const IdentHash& destination) const; AddressBook& GetAddressBook () { return m_AddressBook; };// TODO: move AddressBook away from NetDb void Subscribe (const IdentHash& ident, i2p::tunnel::TunnelPool * pool = nullptr); // keep LeaseSets upto date void Unsubscribe (const IdentHash& ident); void PublishLeaseSet (const LeaseSet * leaseSet, i2p::tunnel::TunnelPool * pool); void RequestDestination (const IdentHash& destination, bool isLeaseSet = false, i2p::tunnel::TunnelPool * pool = nullptr); void HandleDatabaseStoreMsg (I2NPMessage * msg); void HandleDatabaseSearchReplyMsg (I2NPMessage * msg); void HandleDatabaseLookupMsg (I2NPMessage * msg); const RouterInfo * GetRandomRouter () const; const RouterInfo * GetRandomRouter (const RouterInfo * compatibleWith) const; const RouterInfo * GetHighBandwidthRandomRouter (const RouterInfo * compatibleWith) const; void PostI2NPMsg (I2NPMessage * msg); // 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 (const char * directory); void SaveUpdated (const char * directory); void Run (); // exploratory thread void Explore (int numDestinations); void Publish (); void ValidateSubscriptions (); const RouterInfo * GetClosestFloodfill (const IdentHash& destination, const std::set& excluded) const; void KeyspaceRotation (); void ManageLeaseSets (); RequestedDestination * CreateRequestedDestination (const IdentHash& dest, bool isLeaseSet, bool isExploratory = false, i2p::tunnel::TunnelPool * pool = nullptr); bool DeleteRequestedDestination (const IdentHash& dest); // returns true if found void DeleteRequestedDestination (RequestedDestination * dest); template const RouterInfo * GetRandomRouter (Filter filter) const; private: std::map m_LeaseSets; std::map m_RouterInfos; mutable std::mutex m_FloodfillsMutex; std::vector m_Floodfills; std::mutex m_RequestedDestinationsMutex; std::map m_RequestedDestinations; std::map m_Subscriptions; bool m_IsRunning; int m_ReseedRetries; std::thread * m_Thread; i2p::util::Queue m_Queue; // of I2NPDatabaseStoreMsg AddressBook m_AddressBook; static const char m_NetDbPath[]; }; extern NetDb netdb; } } #endif