i2pd/AddressBook.h

107 lines
3.3 KiB
C
Raw Normal View History

2014-03-13 22:22:32 +04:00
#ifndef ADDRESS_BOOK_H__
#define ADDRESS_BOOK_H__
#include <string.h>
#include <string>
#include <map>
#include <vector>
2014-12-20 06:03:34 +03:00
#include <iostream>
2014-12-21 17:33:02 +03:00
#include <mutex>
2015-11-03 17:15:49 +03:00
#include <memory>
#include <boost/asio.hpp>
2015-11-03 17:15:49 +03:00
#include "Base.h"
2014-03-13 22:22:32 +04:00
#include "util.h"
#include "Identity.h"
#include "Log.h"
namespace i2p
{
2014-10-24 23:22:36 +04:00
namespace client
2014-03-13 22:22:32 +04:00
{
2014-12-21 17:33:02 +03:00
const char DEFAULT_SUBSCRIPTION_ADDRESS[] = "http://udhdrtrcetjm5sxzskjyr5ztpeszydbh4dpl3pl4utgqqw2v4jna.b32.i2p/hosts.txt";
const int INITIAL_SUBSCRIPTION_UPDATE_TIMEOUT = 3; // in minutes
const int INITIAL_SUBSCRIPTION_RETRY_TIMEOUT = 1; // in minutes
const int CONTINIOUS_SUBSCRIPTION_UPDATE_TIMEOUT = 720; // in minutes (12 hours)
const int CONTINIOUS_SUBSCRIPTION_RETRY_TIMEOUT = 5; // in minutes
const int SUBSCRIPTION_REQUEST_TIMEOUT = 60; //in second
2015-01-07 23:50:12 +03:00
inline std::string GetB32Address(const i2p::data::IdentHash& ident) { return ident.ToBase32().append(".b32.i2p"); }
2014-11-27 00:19:36 +03:00
class AddressBookStorage // interface for storage
{
public:
virtual ~AddressBookStorage () {};
2015-11-03 17:15:49 +03:00
virtual std::shared_ptr<const i2p::data::IdentityEx> GetAddress (const i2p::data::IdentHash& ident) const = 0;
virtual void AddAddress (std::shared_ptr<const i2p::data::IdentityEx> address) = 0;
2014-11-27 00:19:36 +03:00
virtual void RemoveAddress (const i2p::data::IdentHash& ident) = 0;
2014-11-28 00:26:55 +03:00
2014-11-28 17:40:27 +03:00
virtual int Load (std::map<std::string, i2p::data::IdentHash>& addresses) = 0;
virtual int Save (const std::map<std::string, i2p::data::IdentHash>& addresses) = 0;
2014-11-27 00:19:36 +03:00
};
2014-12-20 06:03:34 +03:00
class AddressBookSubscription;
2014-03-13 22:22:32 +04:00
class AddressBook
{
public:
2014-04-01 23:18:14 +04:00
AddressBook ();
2014-11-27 00:19:36 +03:00
~AddressBook ();
2015-03-30 17:21:52 +03:00
void Start ();
void Stop ();
2014-10-24 23:22:36 +04:00
bool GetIdentHash (const std::string& address, i2p::data::IdentHash& ident);
2015-11-03 17:15:49 +03:00
std::shared_ptr<const i2p::data::IdentityEx> GetAddress (const std::string& address);
2014-10-24 23:22:36 +04:00
const i2p::data::IdentHash * FindAddress (const std::string& address);
2014-09-23 23:38:56 +04:00
void InsertAddress (const std::string& address, const std::string& base64); // for jump service
2015-11-03 17:15:49 +03:00
void InsertAddress (std::shared_ptr<const i2p::data::IdentityEx> address);
2014-12-20 06:03:34 +03:00
2014-12-23 05:20:39 +03:00
void LoadHostsFromStream (std::istream& f);
void DownloadComplete (bool success);
//This method returns the ".b32.i2p" address
2015-01-07 23:50:12 +03:00
std::string ToAddress(const i2p::data::IdentHash& ident) { return GetB32Address(ident); }
2015-11-03 17:15:49 +03:00
std::string ToAddress(std::shared_ptr<const i2p::data::IdentityEx> ident) { return ToAddress(ident->GetIdentHash ()); }
2014-03-13 22:22:32 +04:00
private:
2014-11-27 00:51:36 +03:00
2015-03-30 17:21:52 +03:00
void StartSubscriptions ();
void StopSubscriptions ();
2014-11-27 00:51:36 +03:00
AddressBookStorage * CreateStorage ();
2014-04-01 23:18:14 +04:00
void LoadHosts ();
2014-12-22 23:06:54 +03:00
void LoadSubscriptions ();
2014-03-13 22:22:32 +04:00
void HandleSubscriptionsUpdateTimer (const boost::system::error_code& ecode);
2014-12-20 06:03:34 +03:00
private:
2014-12-21 17:33:02 +03:00
std::mutex m_AddressBookMutex;
2014-10-24 23:22:36 +04:00
std::map<std::string, i2p::data::IdentHash> m_Addresses;
2014-11-27 00:19:36 +03:00
AddressBookStorage * m_Storage;
2014-12-21 17:33:02 +03:00
volatile bool m_IsLoaded, m_IsDownloading;
std::vector<AddressBookSubscription *> m_Subscriptions;
2014-12-21 17:33:02 +03:00
AddressBookSubscription * m_DefaultSubscription; // in case if we don't know any addresses yet
boost::asio::deadline_timer * m_SubscriptionsUpdateTimer;
2014-03-13 22:22:32 +04:00
};
class AddressBookSubscription
{
public:
AddressBookSubscription (AddressBook& book, const std::string& link);
void CheckSubscription ();
private:
void Request ();
private:
AddressBook& m_Book;
2014-12-22 23:06:54 +03:00
std::string m_Link, m_Etag, m_LastModified;
};
2014-03-13 22:22:32 +04:00
}
}
#endif