i2pd/NetDb.h

120 lines
4.0 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-03-19 22:08:09 +04:00
#include <vector>
2013-11-13 16:59:21 +04:00
#include <string>
2013-11-19 05:37:38 +04:00
#include <thread>
2014-02-02 00:57:46 +04:00
#include <boost/filesystem.hpp>
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-03-13 22:22:32 +04:00
#include "AddressBook.h"
2013-11-13 16:59:21 +04:00
namespace i2p
{
namespace data
{
2014-01-05 18:53:44 +04:00
class RequestedDestination
{
public:
RequestedDestination (const IdentHash& destination, bool isLeaseSet, bool isExploratory = false):
m_Destination (destination), m_IsLeaseSet (isLeaseSet), m_IsExploratory (isExploratory),
m_LastRouter (nullptr), m_LastReplyTunnel (nullptr), m_LastOutboundTunnel (nullptr) {};
const IdentHash& GetDestination () const { return m_Destination; };
int GetNumExcludedPeers () const { return m_ExcludedPeers.size (); };
const std::set<IdentHash>& GetExcludedPeers () { return m_ExcludedPeers; };
void ClearExcludedPeers ();
const RouterInfo * GetLastRouter () const { return m_LastRouter; };
2014-01-05 18:53:44 +04:00
const i2p::tunnel::InboundTunnel * GetLastReplyTunnel () const { return m_LastReplyTunnel; };
bool IsExploratory () const { return m_IsExploratory; };
bool IsLeaseSet () const { return m_IsLeaseSet; };
2014-01-05 18:53:44 +04:00
bool IsExcluded (const IdentHash& ident) const { return m_ExcludedPeers.count (ident); };
I2NPMessage * CreateRequestMessage (const RouterInfo * router, const i2p::tunnel::InboundTunnel * replyTunnel);
I2NPMessage * CreateRequestMessage (const IdentHash& floodfill);
2014-01-05 18:53:44 +04:00
i2p::tunnel::OutboundTunnel * GetLastOutboundTunnel () const { return m_LastOutboundTunnel; };
void SetLastOutboundTunnel (i2p::tunnel::OutboundTunnel * tunnel) { m_LastOutboundTunnel = tunnel; };
private:
IdentHash m_Destination;
bool m_IsLeaseSet, m_IsExploratory;
std::set<IdentHash> m_ExcludedPeers;
const RouterInfo * m_LastRouter;
const i2p::tunnel::InboundTunnel * m_LastReplyTunnel;
i2p::tunnel::OutboundTunnel * m_LastOutboundTunnel;
};
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
void AddRouterInfo (uint8_t * buf, int len);
void AddLeaseSet (uint8_t * buf, int len);
2013-11-29 16:52:09 +04:00
RouterInfo * FindRouter (const IdentHash& ident) const;
2013-12-22 20:29:57 +04:00
LeaseSet * FindLeaseSet (const IdentHash& destination) const;
2014-03-14 00:26:04 +04:00
const IdentHash * FindAddress (const std::string& address) { return m_AddressBook.FindAddress (address); }; // TODO: move AddressBook away from NetDb
2014-02-15 01:10:25 +04:00
void Subscribe (const IdentHash& ident); // keep LeaseSets upto date
void Unsubscribe (const IdentHash& ident);
2014-01-03 06:22:48 +04:00
void RequestDestination (const IdentHash& destination, bool isLeaseSet = false);
void HandleDatabaseStoreMsg (uint8_t * buf, size_t len);
void HandleDatabaseSearchReplyMsg (I2NPMessage * msg);
2013-11-13 16:59:21 +04:00
2014-03-19 23:58:57 +04:00
const RouterInfo * GetRandomRouter (const RouterInfo * compatibleWith = nullptr, uint8_t caps = 0) const;
2013-11-20 16:46:09 +04:00
void PostI2NPMsg (I2NPMessage * msg);
2013-11-13 16:59:21 +04:00
private:
2014-02-01 19:10:15 +04:00
bool CreateNetDb(boost::filesystem::path directory);
2013-11-13 16:59:21 +04:00
void Load (const char * directory);
2013-11-20 16:46:09 +04:00
void SaveUpdated (const char * directory);
2013-11-19 05:37:38 +04:00
void Run (); // exploratory thread
void Explore ();
2014-02-13 07:02:39 +04:00
void Publish ();
2014-02-15 01:10:25 +04:00
void ValidateSubscriptions ();
const RouterInfo * GetClosestFloodfill (const IdentHash& destination, const std::set<IdentHash>& excluded) const;
2014-01-05 18:53:44 +04:00
RequestedDestination * CreateRequestedDestination (const IdentHash& dest,
bool isLeaseSet, bool isExploratory = false);
void DeleteRequestedDestination (const IdentHash& dest);
2014-02-12 07:19:51 +04:00
void DeleteRequestedDestination (RequestedDestination * dest);
2013-11-19 05:37:38 +04:00
2013-11-13 16:59:21 +04:00
private:
2013-11-29 16:52:09 +04:00
std::map<IdentHash, LeaseSet *> m_LeaseSets;
std::map<IdentHash, RouterInfo *> m_RouterInfos;
2014-03-19 22:08:09 +04:00
std::vector<RouterInfo *> m_Floodfills;
2014-01-05 18:53:44 +04:00
std::map<IdentHash, RequestedDestination *> m_RequestedDestinations;
2014-02-15 01:10:25 +04:00
std::set<IdentHash> m_Subscriptions;
2014-01-05 18:53:44 +04:00
2013-11-19 05:37:38 +04:00
bool m_IsRunning;
2014-01-31 16:32:34 +04:00
int m_ReseedRetries;
2013-11-19 05:37:38 +04:00
std::thread * m_Thread;
2013-11-20 16:46:09 +04:00
i2p::util::Queue<I2NPMessage> m_Queue; // of I2NPDatabaseStoreMsg
2014-03-13 22:22:32 +04:00
AddressBook m_AddressBook;
2014-02-02 00:57:46 +04:00
static const char m_NetDbPath[];
2013-11-13 16:59:21 +04:00
};
extern NetDb netdb;
}
}
#endif