mirror of https://github.com/PurpleI2P/i2pd
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
890 lines
28 KiB
C++
890 lines
28 KiB
C++
#include <string.h>
|
|
#include <inttypes.h>
|
|
#include <string>
|
|
#include <map>
|
|
#include <fstream>
|
|
#include <chrono>
|
|
#include <condition_variable>
|
|
#include <openssl/rand.h>
|
|
#include <boost/algorithm/string.hpp>
|
|
#include "Base.h"
|
|
#include "util.h"
|
|
#include "Identity.h"
|
|
#include "FS.h"
|
|
#include "Log.h"
|
|
#include "HTTP.h"
|
|
#include "NetDb.hpp"
|
|
#include "ClientContext.h"
|
|
#include "AddressBook.h"
|
|
#include "Config.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace client
|
|
{
|
|
// TODO: this is actually proxy class
|
|
class AddressBookFilesystemStorage: public AddressBookStorage
|
|
{
|
|
private:
|
|
i2p::fs::HashedStorage storage;
|
|
std::string etagsPath, indexPath, localPath;
|
|
|
|
public:
|
|
AddressBookFilesystemStorage (): storage("addressbook", "b", "", "b32") {};
|
|
std::shared_ptr<const i2p::data::IdentityEx> GetAddress (const i2p::data::IdentHash& ident) const;
|
|
void AddAddress (std::shared_ptr<const i2p::data::IdentityEx> address);
|
|
void RemoveAddress (const i2p::data::IdentHash& ident);
|
|
|
|
bool Init ();
|
|
int Load (std::map<std::string, i2p::data::IdentHash>& addresses);
|
|
int LoadLocal (std::map<std::string, i2p::data::IdentHash>& addresses);
|
|
int Save (const std::map<std::string, i2p::data::IdentHash>& addresses);
|
|
|
|
void SaveEtag (const i2p::data::IdentHash& subsciption, const std::string& etag, const std::string& lastModified);
|
|
bool GetEtag (const i2p::data::IdentHash& subscription, std::string& etag, std::string& lastModified);
|
|
|
|
private:
|
|
|
|
int LoadFromFile (const std::string& filename, std::map<std::string, i2p::data::IdentHash>& addresses); // returns -1 if can't open file, otherwise number of records
|
|
|
|
};
|
|
|
|
bool AddressBookFilesystemStorage::Init()
|
|
{
|
|
storage.SetPlace(i2p::fs::GetDataDir());
|
|
// init storage
|
|
if (storage.Init(i2p::data::GetBase32SubstitutionTable(), 32))
|
|
{
|
|
// init ETags
|
|
etagsPath = i2p::fs::StorageRootPath (storage, "etags");
|
|
if (!i2p::fs::Exists (etagsPath))
|
|
i2p::fs::CreateDirectory (etagsPath);
|
|
// init address files
|
|
indexPath = i2p::fs::StorageRootPath (storage, "addresses.csv");
|
|
localPath = i2p::fs::StorageRootPath (storage, "local.csv");
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::shared_ptr<const i2p::data::IdentityEx> AddressBookFilesystemStorage::GetAddress (const i2p::data::IdentHash& ident) const
|
|
{
|
|
std::string filename = storage.Path(ident.ToBase32());
|
|
std::ifstream f(filename, std::ifstream::binary);
|
|
if (!f.is_open ()) {
|
|
LogPrint(eLogDebug, "Addressbook: Requested, but not found: ", filename);
|
|
return nullptr;
|
|
}
|
|
|
|
f.seekg (0,std::ios::end);
|
|
size_t len = f.tellg ();
|
|
if (len < i2p::data::DEFAULT_IDENTITY_SIZE) {
|
|
LogPrint (eLogError, "Addressbook: File ", filename, " is too short: ", len);
|
|
return nullptr;
|
|
}
|
|
f.seekg(0, std::ios::beg);
|
|
uint8_t * buf = new uint8_t[len];
|
|
f.read((char *)buf, len);
|
|
auto address = std::make_shared<i2p::data::IdentityEx>(buf, len);
|
|
delete[] buf;
|
|
return address;
|
|
}
|
|
|
|
void AddressBookFilesystemStorage::AddAddress (std::shared_ptr<const i2p::data::IdentityEx> address)
|
|
{
|
|
std::string path = storage.Path( address->GetIdentHash().ToBase32() );
|
|
std::ofstream f (path, std::ofstream::binary | std::ofstream::out);
|
|
if (!f.is_open ()) {
|
|
LogPrint (eLogError, "Addressbook: can't open file ", path);
|
|
return;
|
|
}
|
|
size_t len = address->GetFullLen ();
|
|
uint8_t * buf = new uint8_t[len];
|
|
address->ToBuffer (buf, len);
|
|
f.write ((char *)buf, len);
|
|
delete[] buf;
|
|
}
|
|
|
|
void AddressBookFilesystemStorage::RemoveAddress (const i2p::data::IdentHash& ident)
|
|
{
|
|
storage.Remove( ident.ToBase32() );
|
|
}
|
|
|
|
int AddressBookFilesystemStorage::LoadFromFile (const std::string& filename, std::map<std::string, i2p::data::IdentHash>& addresses)
|
|
{
|
|
int num = 0;
|
|
std::ifstream f (filename, std::ifstream::in); // in text mode
|
|
if (!f) return -1;
|
|
|
|
addresses.clear ();
|
|
while (!f.eof ())
|
|
{
|
|
std::string s;
|
|
getline(f, s);
|
|
if (!s.length()) continue; // skip empty line
|
|
|
|
std::size_t pos = s.find(',');
|
|
if (pos != std::string::npos)
|
|
{
|
|
std::string name = s.substr(0, pos++);
|
|
std::string addr = s.substr(pos);
|
|
|
|
i2p::data::IdentHash ident;
|
|
ident.FromBase32 (addr);
|
|
addresses[name] = ident;
|
|
num++;
|
|
}
|
|
}
|
|
return num;
|
|
}
|
|
|
|
int AddressBookFilesystemStorage::Load (std::map<std::string, i2p::data::IdentHash>& addresses)
|
|
{
|
|
int num = LoadFromFile (indexPath, addresses);
|
|
if (num < 0)
|
|
{
|
|
LogPrint(eLogWarning, "Addressbook: Can't open ", indexPath);
|
|
return 0;
|
|
}
|
|
LogPrint(eLogInfo, "Addressbook: using index file ", indexPath);
|
|
LogPrint (eLogInfo, "Addressbook: ", num, " addresses loaded from storage");
|
|
|
|
return num;
|
|
}
|
|
|
|
int AddressBookFilesystemStorage::LoadLocal (std::map<std::string, i2p::data::IdentHash>& addresses)
|
|
{
|
|
int num = LoadFromFile (localPath, addresses);
|
|
if (num < 0) return 0;
|
|
LogPrint (eLogInfo, "Addressbook: ", num, " local addresses loaded");
|
|
return num;
|
|
}
|
|
|
|
int AddressBookFilesystemStorage::Save (const std::map<std::string, i2p::data::IdentHash>& addresses)
|
|
{
|
|
if (addresses.empty()) {
|
|
LogPrint(eLogWarning, "Addressbook: not saving empty addressbook");
|
|
return 0;
|
|
}
|
|
|
|
int num = 0;
|
|
std::ofstream f (indexPath, std::ofstream::out); // in text mode
|
|
|
|
if (!f.is_open ()) {
|
|
LogPrint (eLogWarning, "Addressbook: Can't open ", indexPath);
|
|
return 0;
|
|
}
|
|
|
|
for (const auto& it: addresses) {
|
|
f << it.first << "," << it.second.ToBase32 () << std::endl;
|
|
num++;
|
|
}
|
|
LogPrint (eLogInfo, "Addressbook: ", num, " addresses saved");
|
|
return num;
|
|
}
|
|
|
|
void AddressBookFilesystemStorage::SaveEtag (const i2p::data::IdentHash& subscription, const std::string& etag, const std::string& lastModified)
|
|
{
|
|
std::string fname = etagsPath + i2p::fs::dirSep + subscription.ToBase32 () + ".txt";
|
|
std::ofstream f (fname, std::ofstream::out | std::ofstream::trunc);
|
|
if (f)
|
|
{
|
|
f << etag << std::endl;
|
|
f<< lastModified << std::endl;
|
|
}
|
|
}
|
|
|
|
bool AddressBookFilesystemStorage::GetEtag (const i2p::data::IdentHash& subscription, std::string& etag, std::string& lastModified)
|
|
{
|
|
std::string fname = etagsPath + i2p::fs::dirSep + subscription.ToBase32 () + ".txt";
|
|
std::ifstream f (fname, std::ofstream::in);
|
|
if (!f || f.eof ()) return false;
|
|
std::getline (f, etag);
|
|
if (f.eof ()) return false;
|
|
std::getline (f, lastModified);
|
|
return true;
|
|
}
|
|
|
|
//---------------------------------------------------------------------
|
|
AddressBook::AddressBook (): m_Storage(nullptr), m_IsLoaded (false), m_IsDownloading (false),
|
|
m_NumRetries (0), m_DefaultSubscription (nullptr), m_SubscriptionsUpdateTimer (nullptr)
|
|
{
|
|
}
|
|
|
|
AddressBook::~AddressBook ()
|
|
{
|
|
Stop ();
|
|
}
|
|
|
|
void AddressBook::Start ()
|
|
{
|
|
if (!m_Storage)
|
|
m_Storage = new AddressBookFilesystemStorage;
|
|
m_Storage->Init();
|
|
LoadHosts (); /* try storage, then hosts.txt, then download */
|
|
StartSubscriptions ();
|
|
StartLookups ();
|
|
}
|
|
|
|
void AddressBook::StartResolvers ()
|
|
{
|
|
LoadLocal ();
|
|
}
|
|
|
|
void AddressBook::Stop ()
|
|
{
|
|
StopLookups ();
|
|
StopSubscriptions ();
|
|
if (m_SubscriptionsUpdateTimer)
|
|
{
|
|
delete m_SubscriptionsUpdateTimer;
|
|
m_SubscriptionsUpdateTimer = nullptr;
|
|
}
|
|
if (m_IsDownloading)
|
|
{
|
|
LogPrint (eLogInfo, "Addressbook: subscriptions is downloading, abort");
|
|
for (int i = 0; i < 30; i++)
|
|
{
|
|
if (!m_IsDownloading)
|
|
{
|
|
LogPrint (eLogInfo, "Addressbook: subscriptions download complete");
|
|
break;
|
|
}
|
|
std::this_thread::sleep_for (std::chrono::seconds (1)); // wait for 1 seconds
|
|
}
|
|
LogPrint (eLogError, "Addressbook: subscription download timeout");
|
|
m_IsDownloading = false;
|
|
}
|
|
if (m_Storage)
|
|
{
|
|
m_Storage->Save (m_Addresses);
|
|
delete m_Storage;
|
|
m_Storage = nullptr;
|
|
}
|
|
m_DefaultSubscription = nullptr;
|
|
m_Subscriptions.clear ();
|
|
}
|
|
|
|
bool AddressBook::GetIdentHash (const std::string& address, i2p::data::IdentHash& ident)
|
|
{
|
|
auto pos = address.find(".b32.i2p");
|
|
if (pos != std::string::npos)
|
|
{
|
|
Base32ToByteStream (address.c_str(), pos, ident, 32);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
pos = address.find (".i2p");
|
|
if (pos != std::string::npos)
|
|
{
|
|
auto identHash = FindAddress (address);
|
|
if (identHash)
|
|
{
|
|
ident = *identHash;
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
LookupAddress (address); // TODO:
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
// if not .b32 we assume full base64 address
|
|
i2p::data::IdentityEx dest;
|
|
if (!dest.FromBase64 (address))
|
|
return false;
|
|
ident = dest.GetIdentHash ();
|
|
return true;
|
|
}
|
|
|
|
const i2p::data::IdentHash * AddressBook::FindAddress (const std::string& address)
|
|
{
|
|
auto it = m_Addresses.find (address);
|
|
if (it != m_Addresses.end ())
|
|
return &it->second;
|
|
return nullptr;
|
|
}
|
|
|
|
void AddressBook::InsertAddress (const std::string& address, const std::string& base64)
|
|
{
|
|
auto ident = std::make_shared<i2p::data::IdentityEx>();
|
|
ident->FromBase64 (base64);
|
|
m_Storage->AddAddress (ident);
|
|
m_Addresses[address] = ident->GetIdentHash ();
|
|
LogPrint (eLogInfo, "Addressbook: added ", address," -> ", ToAddress(ident->GetIdentHash ()));
|
|
}
|
|
|
|
void AddressBook::InsertAddress (std::shared_ptr<const i2p::data::IdentityEx> address)
|
|
{
|
|
m_Storage->AddAddress (address);
|
|
}
|
|
|
|
std::shared_ptr<const i2p::data::IdentityEx> AddressBook::GetAddress (const std::string& address)
|
|
{
|
|
i2p::data::IdentHash ident;
|
|
if (!GetIdentHash (address, ident)) return nullptr;
|
|
return m_Storage->GetAddress (ident);
|
|
}
|
|
|
|
void AddressBook::LoadHosts ()
|
|
{
|
|
if (m_Storage->Load (m_Addresses) > 0)
|
|
{
|
|
m_IsLoaded = true;
|
|
return;
|
|
}
|
|
|
|
// then try hosts.txt
|
|
std::ifstream f (i2p::fs::DataDirPath("hosts.txt"), std::ifstream::in); // in text mode
|
|
if (f.is_open ())
|
|
{
|
|
LoadHostsFromStream (f, false);
|
|
m_IsLoaded = true;
|
|
}
|
|
}
|
|
|
|
bool AddressBook::LoadHostsFromStream (std::istream& f, bool is_update)
|
|
{
|
|
std::unique_lock<std::mutex> l(m_AddressBookMutex);
|
|
int numAddresses = 0;
|
|
bool incomplete = false;
|
|
std::string s;
|
|
while (!f.eof ())
|
|
{
|
|
getline(f, s);
|
|
|
|
if (!s.length() || s[0] == '#')
|
|
continue; // skip empty or comment line
|
|
|
|
size_t pos = s.find('=');
|
|
|
|
if (pos != std::string::npos)
|
|
{
|
|
std::string name = s.substr(0, pos++);
|
|
std::string addr = s.substr(pos);
|
|
|
|
size_t pos = s.find('#');
|
|
if (pos != std::string::npos)
|
|
addr = addr.substr(pos); // remove comments
|
|
|
|
auto ident = std::make_shared<i2p::data::IdentityEx> ();
|
|
if (!ident->FromBase64(addr)) {
|
|
LogPrint (eLogError, "Addressbook: malformed address ", addr, " for ", name);
|
|
incomplete = f.eof ();
|
|
continue;
|
|
}
|
|
numAddresses++;
|
|
auto it = m_Addresses.find (name);
|
|
if (it != m_Addresses.end ()) // already exists ?
|
|
{
|
|
if (it->second != ident->GetIdentHash ()) // address changed?
|
|
{
|
|
it->second = ident->GetIdentHash ();
|
|
m_Storage->AddAddress (ident);
|
|
LogPrint (eLogInfo, "Addressbook: updated host: ", name);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_Addresses.insert (std::make_pair (name, ident->GetIdentHash ()));
|
|
m_Storage->AddAddress (ident);
|
|
if (is_update)
|
|
LogPrint (eLogInfo, "Addressbook: added new host: ", name);
|
|
}
|
|
}
|
|
else
|
|
incomplete = f.eof ();
|
|
}
|
|
LogPrint (eLogInfo, "Addressbook: ", numAddresses, " addresses processed");
|
|
if (numAddresses > 0)
|
|
{
|
|
if (!incomplete) m_IsLoaded = true;
|
|
m_Storage->Save (m_Addresses);
|
|
}
|
|
return !incomplete;
|
|
}
|
|
|
|
void AddressBook::LoadSubscriptions ()
|
|
{
|
|
if (!m_Subscriptions.size ())
|
|
{
|
|
std::ifstream f (i2p::fs::DataDirPath ("subscriptions.txt"), std::ifstream::in); // in text mode
|
|
if (f.is_open ())
|
|
{
|
|
std::string s;
|
|
while (!f.eof ())
|
|
{
|
|
getline(f, s);
|
|
if (!s.length()) continue; // skip empty line
|
|
m_Subscriptions.push_back (std::make_shared<AddressBookSubscription> (*this, s));
|
|
}
|
|
LogPrint (eLogInfo, "Addressbook: ", m_Subscriptions.size (), " subscriptions urls loaded");
|
|
LogPrint (eLogWarning, "Addressbook: subscriptions.txt usage is deprecated, use config file instead");
|
|
}
|
|
else if (!i2p::config::IsDefault("addressbook.subscriptions"))
|
|
{
|
|
// using config file items
|
|
std::string subscriptionURLs; i2p::config::GetOption("addressbook.subscriptions", subscriptionURLs);
|
|
std::vector<std::string> subsList;
|
|
boost::split(subsList, subscriptionURLs, boost::is_any_of(","), boost::token_compress_on);
|
|
|
|
for (size_t i = 0; i < subsList.size (); i++)
|
|
{
|
|
m_Subscriptions.push_back (std::make_shared<AddressBookSubscription> (*this, subsList[i]));
|
|
}
|
|
LogPrint (eLogInfo, "Addressbook: ", m_Subscriptions.size (), " subscriptions urls loaded");
|
|
}
|
|
}
|
|
else
|
|
LogPrint (eLogError, "Addressbook: subscriptions already loaded");
|
|
}
|
|
|
|
void AddressBook::LoadLocal ()
|
|
{
|
|
std::map<std::string, i2p::data::IdentHash> localAddresses;
|
|
m_Storage->LoadLocal (localAddresses);
|
|
for (const auto& it: localAddresses)
|
|
{
|
|
auto dot = it.first.find ('.');
|
|
if (dot != std::string::npos)
|
|
{
|
|
auto domain = it.first.substr (dot + 1);
|
|
auto it1 = m_Addresses.find (domain); // find domain in our addressbook
|
|
if (it1 != m_Addresses.end ())
|
|
{
|
|
auto dest = context.FindLocalDestination (it1->second);
|
|
if (dest)
|
|
{
|
|
// address is ours
|
|
std::shared_ptr<AddressResolver> resolver;
|
|
auto it2 = m_Resolvers.find (it1->second);
|
|
if (it2 != m_Resolvers.end ())
|
|
resolver = it2->second; // resolver exists
|
|
else
|
|
{
|
|
// create new resolver
|
|
resolver = std::make_shared<AddressResolver>(dest);
|
|
m_Resolvers.insert (std::make_pair(it1->second, resolver));
|
|
}
|
|
resolver->AddAddress (it.first, it.second);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool AddressBook::GetEtag (const i2p::data::IdentHash& subscription, std::string& etag, std::string& lastModified)
|
|
{
|
|
if (m_Storage)
|
|
return m_Storage->GetEtag (subscription, etag, lastModified);
|
|
else
|
|
return false;
|
|
}
|
|
|
|
void AddressBook::DownloadComplete (bool success, const i2p::data::IdentHash& subscription, const std::string& etag, const std::string& lastModified)
|
|
{
|
|
m_IsDownloading = false;
|
|
m_NumRetries++;
|
|
int nextUpdateTimeout = m_NumRetries*CONTINIOUS_SUBSCRIPTION_RETRY_TIMEOUT;
|
|
if (m_NumRetries > CONTINIOUS_SUBSCRIPTION_MAX_NUM_RETRIES || nextUpdateTimeout > CONTINIOUS_SUBSCRIPTION_UPDATE_TIMEOUT)
|
|
nextUpdateTimeout = CONTINIOUS_SUBSCRIPTION_UPDATE_TIMEOUT;
|
|
if (success)
|
|
{
|
|
m_NumRetries = 0;
|
|
if (m_DefaultSubscription) m_DefaultSubscription = nullptr;
|
|
if (m_IsLoaded)
|
|
nextUpdateTimeout = CONTINIOUS_SUBSCRIPTION_UPDATE_TIMEOUT;
|
|
else
|
|
m_IsLoaded = true;
|
|
if (m_Storage) m_Storage->SaveEtag (subscription, etag, lastModified);
|
|
}
|
|
if (m_SubscriptionsUpdateTimer)
|
|
{
|
|
m_SubscriptionsUpdateTimer->expires_from_now (boost::posix_time::minutes(nextUpdateTimeout));
|
|
m_SubscriptionsUpdateTimer->async_wait (std::bind (&AddressBook::HandleSubscriptionsUpdateTimer,
|
|
this, std::placeholders::_1));
|
|
}
|
|
}
|
|
|
|
void AddressBook::StartSubscriptions ()
|
|
{
|
|
LoadSubscriptions ();
|
|
if (m_IsLoaded && m_Subscriptions.empty ()) return;
|
|
|
|
auto dest = i2p::client::context.GetSharedLocalDestination ();
|
|
if (dest)
|
|
{
|
|
m_SubscriptionsUpdateTimer = new boost::asio::deadline_timer (dest->GetService ());
|
|
m_SubscriptionsUpdateTimer->expires_from_now (boost::posix_time::minutes(INITIAL_SUBSCRIPTION_UPDATE_TIMEOUT));
|
|
m_SubscriptionsUpdateTimer->async_wait (std::bind (&AddressBook::HandleSubscriptionsUpdateTimer,
|
|
this, std::placeholders::_1));
|
|
}
|
|
else
|
|
LogPrint (eLogError, "Addressbook: can't start subscriptions: missing shared local destination");
|
|
}
|
|
|
|
void AddressBook::StopSubscriptions ()
|
|
{
|
|
if (m_SubscriptionsUpdateTimer)
|
|
m_SubscriptionsUpdateTimer->cancel ();
|
|
}
|
|
|
|
void AddressBook::HandleSubscriptionsUpdateTimer (const boost::system::error_code& ecode)
|
|
{
|
|
if (ecode != boost::asio::error::operation_aborted)
|
|
{
|
|
auto dest = i2p::client::context.GetSharedLocalDestination ();
|
|
if (!dest) {
|
|
LogPrint(eLogWarning, "Addressbook: missing local destination, skip subscription update");
|
|
return;
|
|
}
|
|
if (!m_IsDownloading && dest->IsReady ())
|
|
{
|
|
if (!m_IsLoaded)
|
|
{
|
|
// download it from default subscription
|
|
LogPrint (eLogInfo, "Addressbook: trying to download it from default subscription.");
|
|
std::string defaultSubURL; i2p::config::GetOption("addressbook.defaulturl", defaultSubURL);
|
|
if (!m_DefaultSubscription)
|
|
m_DefaultSubscription = std::make_shared<AddressBookSubscription>(*this, defaultSubURL);
|
|
m_IsDownloading = true;
|
|
std::thread load_hosts(std::bind (&AddressBookSubscription::CheckUpdates, m_DefaultSubscription));
|
|
load_hosts.detach(); // TODO: use join
|
|
}
|
|
else if (!m_Subscriptions.empty ())
|
|
{
|
|
// pick random subscription
|
|
auto ind = rand () % m_Subscriptions.size();
|
|
m_IsDownloading = true;
|
|
std::thread load_hosts(std::bind (&AddressBookSubscription::CheckUpdates, m_Subscriptions[ind]));
|
|
load_hosts.detach(); // TODO: use join
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// try it again later
|
|
m_SubscriptionsUpdateTimer->expires_from_now (boost::posix_time::minutes(INITIAL_SUBSCRIPTION_RETRY_TIMEOUT));
|
|
m_SubscriptionsUpdateTimer->async_wait (std::bind (&AddressBook::HandleSubscriptionsUpdateTimer,
|
|
this, std::placeholders::_1));
|
|
}
|
|
}
|
|
}
|
|
|
|
void AddressBook::StartLookups ()
|
|
{
|
|
auto dest = i2p::client::context.GetSharedLocalDestination ();
|
|
if (dest)
|
|
{
|
|
auto datagram = dest->GetDatagramDestination ();
|
|
if (!datagram)
|
|
datagram = dest->CreateDatagramDestination ();
|
|
datagram->SetReceiver (std::bind (&AddressBook::HandleLookupResponse, this,
|
|
std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5),
|
|
ADDRESS_RESPONSE_DATAGRAM_PORT);
|
|
}
|
|
}
|
|
|
|
void AddressBook::StopLookups ()
|
|
{
|
|
auto dest = i2p::client::context.GetSharedLocalDestination ();
|
|
if (dest)
|
|
{
|
|
auto datagram = dest->GetDatagramDestination ();
|
|
if (datagram) datagram->ResetReceiver (ADDRESS_RESPONSE_DATAGRAM_PORT);
|
|
}
|
|
}
|
|
|
|
void AddressBook::LookupAddress (const std::string& address)
|
|
{
|
|
const i2p::data::IdentHash * ident = nullptr;
|
|
auto dot = address.find ('.');
|
|
if (dot != std::string::npos)
|
|
ident = FindAddress (address.substr (dot + 1));
|
|
if (!ident)
|
|
{
|
|
LogPrint (eLogError, "Addressbook: Can't find domain for ", address);
|
|
return;
|
|
}
|
|
|
|
auto dest = i2p::client::context.GetSharedLocalDestination ();
|
|
if (dest)
|
|
{
|
|
auto datagram = dest->GetDatagramDestination ();
|
|
if (datagram)
|
|
{
|
|
uint32_t nonce;
|
|
RAND_bytes ((uint8_t *)&nonce, 4);
|
|
{
|
|
std::unique_lock<std::mutex> l(m_LookupsMutex);
|
|
m_Lookups[nonce] = address;
|
|
}
|
|
LogPrint (eLogDebug, "Addressbook: Lookup of ", address, " to ", ident->ToBase32 (), " nonce=", nonce);
|
|
size_t len = address.length () + 9;
|
|
uint8_t * buf = new uint8_t[len];
|
|
memset (buf, 0, 4);
|
|
htobe32buf (buf + 4, nonce);
|
|
buf[8] = address.length ();
|
|
memcpy (buf + 9, address.c_str (), address.length ());
|
|
datagram->SendDatagramTo (buf, len, *ident, ADDRESS_RESPONSE_DATAGRAM_PORT, ADDRESS_RESOLVER_DATAGRAM_PORT);
|
|
delete[] buf;
|
|
}
|
|
}
|
|
}
|
|
|
|
void AddressBook::HandleLookupResponse (const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len)
|
|
{
|
|
if (len < 44)
|
|
{
|
|
LogPrint (eLogError, "Addressbook: Lookup response is too short ", len);
|
|
return;
|
|
}
|
|
uint32_t nonce = bufbe32toh (buf + 4);
|
|
LogPrint (eLogDebug, "Addressbook: Lookup response received from ", from.GetIdentHash ().ToBase32 (), " nonce=", nonce);
|
|
std::string address;
|
|
{
|
|
std::unique_lock<std::mutex> l(m_LookupsMutex);
|
|
auto it = m_Lookups.find (nonce);
|
|
if (it != m_Lookups.end ())
|
|
{
|
|
address = it->second;
|
|
m_Lookups.erase (it);
|
|
}
|
|
}
|
|
if (address.length () > 0)
|
|
{
|
|
// TODO: verify from
|
|
i2p::data::IdentHash hash(buf + 8);
|
|
if (!hash.IsZero ())
|
|
m_Addresses[address] = hash;
|
|
else
|
|
LogPrint (eLogInfo, "AddressBook: Lookup response: ", address, " not found");
|
|
}
|
|
}
|
|
|
|
AddressBookSubscription::AddressBookSubscription (AddressBook& book, const std::string& link):
|
|
m_Book (book), m_Link (link)
|
|
{
|
|
}
|
|
|
|
void AddressBookSubscription::CheckUpdates ()
|
|
{
|
|
bool result = MakeRequest ();
|
|
m_Book.DownloadComplete (result, m_Ident, m_Etag, m_LastModified);
|
|
}
|
|
|
|
bool AddressBookSubscription::MakeRequest ()
|
|
{
|
|
i2p::http::URL url;
|
|
// must be run in separate thread
|
|
LogPrint (eLogInfo, "Addressbook: Downloading hosts database from ", m_Link);
|
|
if (!url.parse(m_Link)) {
|
|
LogPrint(eLogError, "Addressbook: failed to parse url: ", m_Link);
|
|
return false;
|
|
}
|
|
if (!m_Book.GetIdentHash (url.host, m_Ident)) {
|
|
LogPrint (eLogError, "Addressbook: Can't resolve ", url.host);
|
|
return false;
|
|
}
|
|
/* this code block still needs some love */
|
|
std::condition_variable newDataReceived;
|
|
std::mutex newDataReceivedMutex;
|
|
auto leaseSet = i2p::client::context.GetSharedLocalDestination ()->FindLeaseSet (m_Ident);
|
|
if (!leaseSet)
|
|
{
|
|
std::unique_lock<std::mutex> l(newDataReceivedMutex);
|
|
i2p::client::context.GetSharedLocalDestination ()->RequestDestination (m_Ident,
|
|
[&newDataReceived, &leaseSet, &newDataReceivedMutex](std::shared_ptr<i2p::data::LeaseSet> ls)
|
|
{
|
|
leaseSet = ls;
|
|
std::unique_lock<std::mutex> l1(newDataReceivedMutex);
|
|
newDataReceived.notify_all ();
|
|
});
|
|
if (newDataReceived.wait_for (l, std::chrono::seconds (SUBSCRIPTION_REQUEST_TIMEOUT)) == std::cv_status::timeout)
|
|
{
|
|
LogPrint (eLogError, "Addressbook: Subscription LeaseSet request timeout expired");
|
|
i2p::client::context.GetSharedLocalDestination ()->CancelDestinationRequest (m_Ident, false); // don't notify, because we know it already
|
|
return false;
|
|
}
|
|
}
|
|
if (!leaseSet) {
|
|
/* still no leaseset found */
|
|
LogPrint (eLogError, "Addressbook: LeaseSet for address ", url.host, " not found");
|
|
return false;
|
|
}
|
|
if (m_Etag.empty() && m_LastModified.empty()) {
|
|
m_Book.GetEtag (m_Ident, m_Etag, m_LastModified);
|
|
LogPrint (eLogDebug, "Addressbook: loaded for ", url.host, ": ETag: ", m_Etag, ", Last-Modified: ", m_LastModified);
|
|
}
|
|
/* save url parts for later use */
|
|
std::string dest_host = url.host;
|
|
int dest_port = url.port ? url.port : 80;
|
|
/* create http request & send it */
|
|
i2p::http::HTTPReq req;
|
|
req.AddHeader("Host", dest_host);
|
|
req.AddHeader("User-Agent", "Wget/1.11.4");
|
|
req.AddHeader("X-Accept-Encoding", "x-i2p-gzip;q=1.0, identity;q=0.5, deflate;q=0, gzip;q=0, *;q=0");
|
|
req.AddHeader("Connection", "close");
|
|
if (!m_Etag.empty())
|
|
req.AddHeader("If-None-Match", m_Etag);
|
|
if (!m_LastModified.empty())
|
|
req.AddHeader("If-Modified-Since", m_LastModified);
|
|
/* convert url to relative */
|
|
url.schema = "";
|
|
url.host = "";
|
|
req.uri = url.to_string();
|
|
auto stream = i2p::client::context.GetSharedLocalDestination ()->CreateStream (leaseSet, dest_port);
|
|
std::string request = req.to_string();
|
|
stream->Send ((const uint8_t *) request.data(), request.length());
|
|
/* read response */
|
|
std::string response;
|
|
uint8_t recv_buf[4096];
|
|
bool end = false;
|
|
int numAttempts = 5;
|
|
while (!end)
|
|
{
|
|
stream->AsyncReceive (boost::asio::buffer (recv_buf, 4096),
|
|
[&](const boost::system::error_code& ecode, std::size_t bytes_transferred)
|
|
{
|
|
if (bytes_transferred)
|
|
response.append ((char *)recv_buf, bytes_transferred);
|
|
if (ecode == boost::asio::error::timed_out || !stream->IsOpen ())
|
|
end = true;
|
|
newDataReceived.notify_all ();
|
|
},
|
|
SUBSCRIPTION_REQUEST_TIMEOUT);
|
|
std::unique_lock<std::mutex> l(newDataReceivedMutex);
|
|
if (newDataReceived.wait_for (l, std::chrono::seconds (SUBSCRIPTION_REQUEST_TIMEOUT)) == std::cv_status::timeout)
|
|
{
|
|
LogPrint (eLogError, "Addressbook: subscriptions request timeout expired");
|
|
numAttempts++;
|
|
if (numAttempts > 5) end = true;
|
|
}
|
|
}
|
|
// process remaining buffer
|
|
while (size_t len = stream->ReadSome (recv_buf, sizeof(recv_buf)))
|
|
response.append ((char *)recv_buf, len);
|
|
/* parse response */
|
|
i2p::http::HTTPRes res;
|
|
int res_head_len = res.parse(response);
|
|
if (res_head_len < 0)
|
|
{
|
|
LogPrint(eLogError, "Addressbook: can't parse http response from ", dest_host);
|
|
return false;
|
|
}
|
|
if (res_head_len == 0)
|
|
{
|
|
LogPrint(eLogError, "Addressbook: incomplete http response from ", dest_host, ", interrupted by timeout");
|
|
return false;
|
|
}
|
|
/* assert: res_head_len > 0 */
|
|
response.erase(0, res_head_len);
|
|
if (res.code == 304)
|
|
{
|
|
LogPrint (eLogInfo, "Addressbook: no updates from ", dest_host, ", code 304");
|
|
return false;
|
|
}
|
|
if (res.code != 200)
|
|
{
|
|
LogPrint (eLogWarning, "Adressbook: can't get updates from ", dest_host, ", response code ", res.code);
|
|
return false;
|
|
}
|
|
int len = res.content_length();
|
|
if (response.empty())
|
|
{
|
|
LogPrint(eLogError, "Addressbook: empty response from ", dest_host, ", expected ", len, " bytes");
|
|
return false;
|
|
}
|
|
if (!res.is_gzipped () && len > 0 && len != (int) response.length())
|
|
{
|
|
LogPrint(eLogError, "Addressbook: response size mismatch, expected: ", len, ", got: ", response.length(), "bytes");
|
|
return false;
|
|
}
|
|
/* assert: res.code == 200 */
|
|
auto it = res.headers.find("ETag");
|
|
if (it != res.headers.end()) m_Etag = it->second;
|
|
it = res.headers.find("If-Modified-Since");
|
|
if (it != res.headers.end()) m_LastModified = it->second;
|
|
if (res.is_chunked())
|
|
{
|
|
std::stringstream in(response), out;
|
|
i2p::http::MergeChunkedResponse (in, out);
|
|
response = out.str();
|
|
}
|
|
else if (res.is_gzipped())
|
|
{
|
|
std::stringstream out;
|
|
i2p::data::GzipInflator inflator;
|
|
inflator.Inflate ((const uint8_t *) response.data(), response.length(), out);
|
|
if (out.fail())
|
|
{
|
|
LogPrint(eLogError, "Addressbook: can't gunzip http response");
|
|
return false;
|
|
}
|
|
response = out.str();
|
|
}
|
|
std::stringstream ss(response);
|
|
LogPrint (eLogInfo, "Addressbook: got update from ", dest_host);
|
|
m_Book.LoadHostsFromStream (ss, true);
|
|
return true;
|
|
}
|
|
|
|
AddressResolver::AddressResolver (std::shared_ptr<ClientDestination> destination):
|
|
m_LocalDestination (destination)
|
|
{
|
|
if (m_LocalDestination)
|
|
{
|
|
auto datagram = m_LocalDestination->GetDatagramDestination ();
|
|
if (!datagram)
|
|
datagram = m_LocalDestination->CreateDatagramDestination ();
|
|
datagram->SetReceiver (std::bind (&AddressResolver::HandleRequest, this,
|
|
std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5),
|
|
ADDRESS_RESOLVER_DATAGRAM_PORT);
|
|
}
|
|
}
|
|
|
|
AddressResolver::~AddressResolver ()
|
|
{
|
|
if (m_LocalDestination)
|
|
{
|
|
auto datagram = m_LocalDestination->GetDatagramDestination ();
|
|
if (datagram)
|
|
datagram->ResetReceiver (ADDRESS_RESOLVER_DATAGRAM_PORT);
|
|
}
|
|
}
|
|
|
|
void AddressResolver::HandleRequest (const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len)
|
|
{
|
|
if (len < 9 || len < buf[8] + 9U)
|
|
{
|
|
LogPrint (eLogError, "Addressbook: Address request is too short ", len);
|
|
return;
|
|
}
|
|
// read requested address
|
|
uint8_t l = buf[8];
|
|
char address[255];
|
|
memcpy (address, buf + 9, l);
|
|
address[l] = 0;
|
|
LogPrint (eLogDebug, "Addressbook: Address request ", address);
|
|
// send response
|
|
uint8_t response[44];
|
|
memset (response, 0, 4); // reserved
|
|
memcpy (response + 4, buf + 4, 4); // nonce
|
|
auto it = m_LocalAddresses.find (address); // address lookup
|
|
if (it != m_LocalAddresses.end ())
|
|
memcpy (response + 8, it->second, 32); // ident
|
|
else
|
|
memset (response + 8, 0, 32); // not found
|
|
memset (response + 40, 0, 4); // set expiration time to zero
|
|
m_LocalDestination->GetDatagramDestination ()->SendDatagramTo (response, 44, from.GetIdentHash(), toPort, fromPort);
|
|
}
|
|
|
|
void AddressResolver::AddAddress (const std::string& name, const i2p::data::IdentHash& ident)
|
|
{
|
|
m_LocalAddresses[name] = ident;
|
|
}
|
|
|
|
}
|
|
}
|