i2pd/NetDb.h

138 lines
4.7 KiB
C
Raw Normal View History

2013-11-13 16:59:21 +04:00
#ifndef NETDB_H__
#define NETDB_H__
#include <inttypes.h>
2014-01-05 18:53:44 +04:00
#include <set>
2013-11-13 16:59:21 +04:00
#include <map>
2014-11-21 21:02:46 +03:00
#include <list>
2013-11-13 16:59:21 +04:00
#include <string>
2013-11-19 05:37:38 +04:00
#include <thread>
2014-10-06 05:59:05 +04:00
#include <mutex>
2015-11-03 17:15:49 +03:00
#include "Base.h"
#include "Gzip.h"
#include "FS.h"
2013-11-20 16:46:09 +04:00
#include "Queue.h"
#include "I2NPProtocol.h"
2013-11-13 16:59:21 +04:00
#include "RouterInfo.h"
#include "LeaseSet.h"
2013-12-25 21:19:46 +04:00
#include "Tunnel.h"
2014-08-20 19:12:53 +04:00
#include "TunnelPool.h"
2015-01-19 21:57:37 +03:00
#include "Reseed.h"
2015-04-09 19:45:00 +03:00
#include "NetDbRequests.h"
2016-02-18 23:57:43 +03:00
#include "Family.h"
2013-11-13 16:59:21 +04:00
namespace i2p
{
namespace data
{
2016-02-24 19:31:14 +03:00
const int NETDB_MIN_ROUTERS = 90;
const int NETDB_FLOODFILL_EXPIRATION_TIMEOUT = 60*60; // 1 hour, in seconds
const int NETDB_INTRODUCEE_EXPIRATION_TIMEOUT = 65*60;
const int NETDB_MIN_EXPIRATION_TIMEOUT = 90*60; // 1.5 hours
const int NETDB_MAX_EXPIRATION_TIMEOUT = 27*60*60; // 27 hours
2016-06-30 19:27:23 +03:00
const int NETDB_PUBLISH_INTERVAL = 60*40;
2016-07-15 16:38:21 +03:00
/** function for visiting a leaseset stored in a floodfill */
typedef std::function<void(const IdentHash, std::shared_ptr<LeaseSet>)> LeaseSetVisitor;
2013-11-13 16:59:21 +04:00
class NetDb
{
public:
NetDb ();
~NetDb ();
2013-11-19 05:37:38 +04:00
void Start ();
void Stop ();
2013-11-13 16:59:21 +04:00
2016-02-17 23:36:55 +03:00
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;
2015-01-27 19:27:58 +03:00
std::shared_ptr<LeaseSet> FindLeaseSet (const IdentHash& destination) const;
2015-11-03 17:15:49 +03:00
std::shared_ptr<RouterProfile> FindRouterProfile (const IdentHash& ident) const;
2014-03-14 00:26:04 +04:00
2015-01-15 00:11:09 +03:00
void RequestDestination (const IdentHash& destination, RequestedDestination::RequestComplete requestComplete = nullptr);
2014-08-20 19:12:53 +04:00
2015-07-04 04:27:40 +03:00
void HandleDatabaseStoreMsg (std::shared_ptr<const I2NPMessage> msg);
void HandleDatabaseSearchReplyMsg (std::shared_ptr<const I2NPMessage> msg);
void HandleDatabaseLookupMsg (std::shared_ptr<const I2NPMessage> msg);
2014-11-21 00:20:02 +03: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;
2015-02-26 22:17:16 +03:00
std::shared_ptr<const RouterInfo> GetRandomPeerTestRouter () const;
2015-02-27 05:05:35 +03:00
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;
2015-06-11 18:43:35 +03:00
std::vector<IdentHash> GetClosestFloodfills (const IdentHash& destination, size_t num,
std::set<IdentHash>& excluded, bool closeThanUsOnly = false) const;
2015-02-03 06:34:55 +03:00
std::shared_ptr<const RouterInfo> GetClosestNonFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded) const;
2016-06-17 18:03:33 +03:00
std::shared_ptr<const RouterInfo> GetRandomRouterInFamily(const std::string & fam) const;
2014-10-24 23:39:53 +04:00
void SetUnreachable (const IdentHash& ident, bool unreachable);
2015-07-04 04:27:40 +03:00
void PostI2NPMsg (std::shared_ptr<const I2NPMessage> msg);
/** set hidden mode, aka don't publish our RI to netdb and don't explore */
void SetHidden(bool hide);
2015-01-19 21:57:37 +03:00
void Reseed ();
Families& GetFamilies () { return m_Families; };
2015-01-19 21:57:37 +03:00
2015-04-05 20:56:41 +03:00
// for web interface
int GetNumRouters () const { return m_RouterInfos.size (); };
int GetNumFloodfills () const { return m_Floodfills.size (); };
int GetNumLeaseSets () const { return m_LeaseSets.size (); };
2016-07-15 16:38:21 +03:00
/** visit all lease sets we currently store */
void VisitLeaseSets(LeaseSetVisitor v);
2013-11-13 16:59:21 +04:00
private:
void Load ();
2016-02-11 03:00:00 +03:00
bool LoadRouterInfo (const std::string & path);
void SaveUpdated ();
2013-11-19 05:37:38 +04:00
void Run (); // exploratory thread
2014-12-24 19:20:38 +03:00
void Explore (int numDestinations);
2014-02-13 07:02:39 +04:00
void Publish ();
2014-07-31 20:59:43 +04:00
void ManageLeaseSets ();
2014-12-24 19:20:38 +03:00
void ManageRequests ();
2016-03-22 20:10:02 +03:00
void ManageLookupResponses ();
2014-01-05 18:53:44 +04:00
2016-06-17 18:03:33 +03:00
template<typename Filter>
std::shared_ptr<const RouterInfo> GetRandomRouter (Filter filter) const;
2014-07-03 22:47:12 +04:00
2013-11-13 16:59:21 +04:00
private:
2016-07-15 16:38:21 +03:00
mutable std::mutex m_LeaseSetsMutex;
2015-01-27 19:27:58 +03:00
std::map<IdentHash, std::shared_ptr<LeaseSet> > m_LeaseSets;
2014-11-21 21:29:19 +03:00
mutable std::mutex m_RouterInfosMutex;
2014-11-18 19:08:10 +03:00
std::map<IdentHash, std::shared_ptr<RouterInfo> > m_RouterInfos;
2014-10-06 05:59:05 +04:00
mutable std::mutex m_FloodfillsMutex;
2014-11-21 21:02:46 +03:00
std::list<std::shared_ptr<RouterInfo> > m_Floodfills;
2014-01-05 18:53:44 +04:00
2013-11-19 05:37:38 +04:00
bool m_IsRunning;
2016-02-11 03:00:00 +03:00
uint64_t m_LastLoad;
2013-11-19 05:37:38 +04:00
std::thread * m_Thread;
2015-07-04 04:27:40 +03:00
i2p::util::Queue<std::shared_ptr<const I2NPMessage> > m_Queue; // of I2NPDatabaseStoreMsg
2014-02-02 00:57:46 +04:00
2015-11-03 17:15:49 +03:00
GzipInflator m_Inflator;
2015-01-19 21:57:37 +03:00
Reseeder * m_Reseeder;
2016-02-18 23:57:43 +03:00
Families m_Families;
i2p::fs::HashedStorage m_Storage;
2015-01-19 21:57:37 +03:00
2015-05-07 02:18:00 +03:00
friend class NetDbRequests;
2015-04-09 19:45:00 +03:00
NetDbRequests m_Requests;
2016-03-22 20:10:02 +03:00
std::map<IdentHash, std::pair<std::vector<IdentHash>, uint64_t> > m_LookupResponses; // ident->(closest FFs, timestamp)
/** true if in hidden mode */
bool m_HiddenMode;
2013-11-13 16:59:21 +04:00
};
extern NetDb netdb;
}
}
#endif