/*
* Copyright (c) 2013-2026, The PurpleI2P Project
*
* This file is part of Purple i2pd project and licensed under BSD3
*
* See full license text in LICENSE file at top of project tree
*/

#include <string.h>
#include <inttypes.h>
#include <string>
#include <unordered_map>
#include <fstream>
#include <chrono>
#include <future>
#include <openssl/rand.h>
#include <boost/algorithm/string.hpp>
#include "Base.h"
#include "util.h"
#include "Timestamp.h"
#include "Identity.h"
#include "FS.h"
#include "Log.h"
#include "Config.h"
#include "HTTP.h"
#include "NetDb.hpp"
#include "ClientContext.h"
#include "BoostStream.h"
#include "AddressBook.h"

#if STD_FILESYSTEM
#include <filesystem>
namespace fs_lib = std::filesystem;
#else
#include <boost/filesystem.hpp>
namespace fs_lib = boost::filesystem;
#endif

namespace i2p
{
namespace client
{
	// TODO: this is actually proxy class
	class AddressBookFilesystemStorage: public AddressBookStorage
	{
		public:

			AddressBookFilesystemStorage (): storage("addressbook", "b", "", "b32")
			{
				i2p::config::GetOption("persist.addressbook", m_IsPersist);
				if (m_IsPersist)
					i2p::config::GetOption("addressbook.hostsfile", m_HostsFile);
			}
			std::shared_ptr<const i2p::data::IdentityEx> GetAddress (const i2p::data::IdentHash& ident) override;
			void AddAddress (std::shared_ptr<const i2p::data::IdentityEx> address) override;
			void RemoveAddress (const i2p::data::IdentHash& ident) override;
			void CleanUpCache () override;

			bool Init () override;
			int Load (Addresses& addresses) override;
			int LoadLocal (Addresses& addresses) override;
			int Save (const Addresses& addresses) override;

			void SaveEtag (const i2p::data::IdentHash& subscription, const std::string& etag, const std::string& lastModified) override;
			bool GetEtag (const i2p::data::IdentHash& subscription, std::string& etag, std::string& lastModified) override;
			void ResetEtags () override;

		private:

			int LoadFromFile (const std::string& filename, Addresses& addresses); // returns -1 if can't open file, otherwise number of records

		private:

			i2p::fs::HashedStorage storage;
			std::string etagsPath, indexPath, localPath;
			bool m_IsPersist = false;
			std::string m_HostsFile; // file to dump hosts.txt, empty if not used
			std::unordered_map<i2p::data::IdentHash, std::pair<std::vector<uint8_t>, uint64_t> > m_FullAddressCache; // ident hash -> (full ident buffer, last access timestamp)
			std::mutex m_FullAddressCacheMutex;
	};

	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)
	{
		auto ts = i2p::util::GetMonotonicSeconds ();
		{
			std::lock_guard<std::mutex> l(m_FullAddressCacheMutex);
			auto it = m_FullAddressCache.find (ident);
			if (it != m_FullAddressCache.end ())
			{
				it->second.second = ts;
				return std::make_shared<i2p::data::IdentityEx>(it->second.first.data (), it->second.first.size ());
			}
		}

		if (!m_IsPersist)
		{
			LogPrint(eLogDebug, "Addressbook: Persistence is disabled");
			return nullptr;
		}
		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);
		std::vector<uint8_t> buf(len);
		f.read((char *)buf.data (), len);
		if (!f)
		{
			LogPrint (eLogError, "Addressbook: Couldn't read ", filename);
			return nullptr;
		}
		{
			std::lock_guard<std::mutex> l(m_FullAddressCacheMutex);
			m_FullAddressCache.try_emplace (ident, buf, ts);
		}
		return std::make_shared<i2p::data::IdentityEx>(buf.data (), len);
	}

	void AddressBookFilesystemStorage::AddAddress (std::shared_ptr<const i2p::data::IdentityEx> address)
	{
		if (!address) return;
		size_t len = address->GetFullLen ();
		std::vector<uint8_t> buf;
		if (!len) return; // invalid address
		{
			std::lock_guard<std::mutex> l(m_FullAddressCacheMutex);
			auto [it, inserted] = m_FullAddressCache.try_emplace (address->GetIdentHash(), len, i2p::util::GetMonotonicSeconds ());
			if (inserted)
				address->ToBuffer (it->second.first.data (), len);
			if (m_IsPersist)
				buf = it->second.first;
		}
		if (m_IsPersist && !buf.empty ())
		{
			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;
			}
			f.write ((const char *)buf.data (), len);
		}
	}

	void AddressBookFilesystemStorage::RemoveAddress (const i2p::data::IdentHash& ident)
	{
		{
			std::lock_guard<std::mutex> l(m_FullAddressCacheMutex);
			m_FullAddressCache.erase (ident);
		}
		if (!m_IsPersist) return;
		storage.Remove( ident.ToBase32() );
	}

	int AddressBookFilesystemStorage::LoadFromFile (const std::string& filename, Addresses& 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);

				addresses[name] = std::make_shared<Address>(addr);
				num++;
			}
		}
		return num;
	}

	int AddressBookFilesystemStorage::Load (Addresses& 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 (Addresses& addresses)
	{
		int num = LoadFromFile (localPath, addresses);
		if (num < 0) return 0;
		LogPrint (eLogInfo, "Addressbook: ", num, " local addresses loaded");
		return num;
	}

	int AddressBookFilesystemStorage::Save (const Addresses& addresses)
	{
		if (addresses.empty())
		{
			LogPrint(eLogWarning, "Addressbook: Not saving empty addressbook");
			return 0;
		}

		int num = 0;
		{
			// save index file
			std::ofstream f (indexPath, std::ofstream::out); // in text mode
			if (f.is_open ())
			{
				for (const auto& it: addresses)
				{
					if (it.second->IsValid ())
					{
						f << it.first << ",";
						if (it.second->IsIdentHash ())
							f << it.second->identHash.ToBase32 ();
						else
							f << it.second->blindedPublicKey->ToB33 ();
						f << std::endl;
						num++;
					}
					else
						LogPrint (eLogWarning, "Addressbook: Invalid address ", it.first);
				}
				LogPrint (eLogInfo, "Addressbook: ", num, " addresses saved");
			}
			else
				LogPrint (eLogWarning, "Addressbook: Can't open ", indexPath);
		}
		if (!m_HostsFile.empty ())
		{
			// dump full hosts.txt
			std::ofstream f (m_HostsFile, std::ofstream::out); // in text mode
			if (f.is_open ())
			{
				for (const auto& it: addresses)
				{
					std::shared_ptr<const i2p::data::IdentityEx> addr;
					if (it.second->IsIdentHash ())
					{
						addr = GetAddress (it.second->identHash);
						if (addr)
							f << it.first << "=" << addr->ToBase64 () << std::endl;
					}
				}
			}
			else
				LogPrint (eLogWarning, "Addressbook: Can't open ", m_HostsFile);
		}

		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;
	}

	void AddressBookFilesystemStorage::ResetEtags ()
	{
		LogPrint (eLogError, "Addressbook: Resetting eTags");
		for (fs_lib::directory_iterator it (etagsPath); it != fs_lib::directory_iterator (); ++it)
		{
			if (!fs_lib::is_regular_file (it->status ()))
				continue;
			fs_lib::remove (it->path ());
		}
	}

	void AddressBookFilesystemStorage::CleanUpCache ()
	{
		auto ts = i2p::util::GetMonotonicSeconds ();
		std::lock_guard<std::mutex> l(m_FullAddressCacheMutex);
		for (auto it = m_FullAddressCache.begin (); it != m_FullAddressCache.end ();)
		{
			if (ts > it->second.second + ADDRESS_CACHE_EXPIRATION_TIMEOUT)
				it = m_FullAddressCache.erase (it);
			else
				it++;
		}
	}

//---------------------------------------------------------------------

	Address::Address (std::string_view b32):
		addressType (eAddressInvalid)
	{
		if (b32.length () <= B33_ADDRESS_THRESHOLD)
		{
			if (identHash.FromBase32 (b32) > 0)
				addressType = eAddressIndentHash;
		}
		else
		{
			blindedPublicKey = std::make_shared<i2p::data::BlindedPublicKey>(b32);
			if (blindedPublicKey->IsValid ())
				addressType = eAddressBlindedPublicKey;
		}
	}

	Address::Address (const i2p::data::IdentHash& hash)
	{
		addressType = eAddressIndentHash;
		identHash = hash;
	}

	std::string Address::ToBase32 () const
	{
		switch (addressType)
		{
			case eAddressIndentHash:
				return identHash.ToBase32 ();
			break;
			case eAddressBlindedPublicKey:
				if (blindedPublicKey)
					return blindedPublicKey->ToB33 ();
			break;
			default: ;
		}
		return "";
	}

	AddressBook::AddressBook (): m_Storage(nullptr), m_IsLoaded (false),
		m_NumRetries (0), m_DefaultSubscription (nullptr), m_SubscriptionsUpdateTimer (nullptr),
		m_IsEnabled (true)
	{
	}

	AddressBook::~AddressBook ()
	{
		Stop ();
	}

	void AddressBook::Start ()
	{
		i2p::config::GetOption("addressbook.enabled", m_IsEnabled);
		if (m_IsEnabled)
		{
			if (!m_Storage)
				m_Storage = new AddressBookFilesystemStorage;
			m_Storage->Init();
			LoadHosts (); /* try storage, then hosts.txt, then download */
			StartSubscriptions ();
			StartLookups ();
			ScheduleCacheUpdate ();
		}
	}

	void AddressBook::StartResolvers ()
	{
		LoadLocal ();
	}

	void AddressBook::Stop ()
	{
		StopLookups ();
		StopSubscriptions ();
		if (m_SubscriptionsUpdateTimer)
		{
			m_SubscriptionsUpdateTimer->cancel ();
			m_SubscriptionsUpdateTimer = nullptr;
		}
		if (m_AddressCacheUpdateTimer)
		{
			m_AddressCacheUpdateTimer->cancel ();
			m_AddressCacheUpdateTimer = nullptr;
		}
		bool isDownloading = m_Downloading.valid ();
		if (isDownloading)
		{
			if (m_Downloading.wait_for(std::chrono::seconds(0)) == std::future_status::ready)
				isDownloading = false;
			else
			{
				LogPrint (eLogInfo, "Addressbook: Subscriptions are downloading, abort");
				for (int i = 0; i < 30; i++)
				{
					if (m_Downloading.wait_for(std::chrono::seconds(1)) == std::future_status::ready) // wait for 1 seconds
					{
						isDownloading = false;
						LogPrint (eLogInfo, "Addressbook: Subscriptions download complete");
						break;
					}
				}
			}
			if (!isDownloading)
				m_Downloading.get ();
			else
				LogPrint (eLogError, "Addressbook: Subscription download timeout");
		}
		if (m_Storage)
		{
			m_Storage->Save (m_Addresses);
			delete m_Storage;
			m_Storage = nullptr;
		}
		m_DefaultSubscription = nullptr;
		m_Subscriptions.clear ();
	}

	std::shared_ptr<const Address> AddressBook::GetAddress (std::string_view address)
	{
		auto pos = address.find(".b32.i2p");
		if (pos != std::string::npos)
		{
			auto addr = std::make_shared<const Address>(address.substr (0, pos));
			return addr->IsValid () ? addr : nullptr;
		}
		else
#if __cplusplus >= 202002L // C++20
		if (address.ends_with (".i2p"))
#else
		if (address.find (".i2p") != std::string::npos)
#endif
		{
			if (!m_IsEnabled) return nullptr;
			auto addr = FindAddress (address);
			if (!addr)
				LookupAddress (address); // TODO:
			return addr;
		}
		// if not .b32 we assume full base64 address
		i2p::data::IdentityEx dest;
		if (!dest.FromBase64 (address))
			return nullptr;
		return std::make_shared<const Address>(dest.GetIdentHash ());
	}

	std::shared_ptr<const Address> AddressBook::FindAddress (std::string_view address)
	{
		std::lock_guard<std::mutex> l(m_AddressBookMutex);
		auto it = m_Addresses.find (address);
		if (it != m_Addresses.end ())
			return it->second;
		return nullptr;
	}

	bool AddressBook::RecordExists (const std::string& address, const std::string& jump)
	{
		auto addr = FindAddress(address);
		if (!addr)
			return false;

		auto pos = jump.find(".b32.i2p");
		if (pos != std::string::npos)
		{
			i2p::data::IdentHash identHash;
			if (identHash.FromBase32(jump.substr (0, pos)) && identHash == addr->identHash)
				return true;
		}
		else
		{
			i2p::data::IdentityEx ident;
			if (ident.FromBase64 (jump) && ident.GetIdentHash () == addr->identHash)
				return true;
		}

		return false;
	}

	void AddressBook::InsertAddress (const std::string& address, const std::string& jump)
	{
		std::lock_guard<std::mutex> l(m_AddressBookMutex);
		auto pos = jump.find(".b32.i2p");
		if (pos != std::string::npos)
		{
			m_Addresses[address] = std::make_shared<Address>(jump.substr (0, pos));
			LogPrint (eLogInfo, "Addressbook: Added ", address," -> ", jump);
		}
		else
		{
			// assume base64
			auto ident = std::make_shared<i2p::data::IdentityEx>();
			if (ident->FromBase64 (jump))
			{
				if (m_Storage) m_Storage->AddAddress (ident);
				m_Addresses[address] = std::make_shared<Address>(ident->GetIdentHash ());
				LogPrint (eLogInfo, "Addressbook: Added ", address," -> ", ToAddress(ident->GetIdentHash ()));
			}
			else
				LogPrint (eLogError, "Addressbook: Malformed address ", jump);
		}
	}

	void AddressBook::InsertFullAddress (std::shared_ptr<const i2p::data::IdentityEx> address)
	{
		if (m_Storage) m_Storage->AddAddress (address);
	}

	std::shared_ptr<const i2p::data::IdentityEx> AddressBook::GetFullAddress (const std::string& address)
	{
		auto addr = GetAddress (address);
		if (!addr || !addr->IsIdentHash ()) return nullptr;
		return m_Storage ? m_Storage->GetAddress (addr->identHash) : nullptr;
	}

	void AddressBook::LoadHosts ()
	{
		if (!m_Storage) return;
		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 (std::move (f), false);
			m_IsLoaded = true;
		}

		// reset eTags, because we don’t know how old hosts.txt is or can't load addressbook
		m_Storage->ResetEtags ();
	}

	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_view name = std::string_view(s).substr(0, pos++);
				std::string_view addr = std::string_view(s).substr(pos);

				size_t pos = addr.find('#');
				if (pos != addr.npos)
					addr = addr.substr(0, pos); // remove comments
#if __cplusplus >= 202002L // C++20
				if (name.ends_with (".b32.i2p"))
#else
				if (name.find(".b32.i2p") != name.npos)
#endif
				{
					LogPrint (eLogError, "Addressbook: Skipped adding of b32 address: ", name);
					continue;
				}

#if __cplusplus >= 202002L // C++20
				if (!name.ends_with (".i2p"))
#else
				if (name.find(".i2p") == name.npos)
#endif
				{
					LogPrint (eLogError, "Addressbook: Malformed domain: ", name);
					continue;
				}

				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->IsIdentHash () && it->second->identHash != ident->GetIdentHash () && // address changed?
						ident->GetSigningKeyType () != i2p::data::SIGNING_KEY_TYPE_DSA_SHA1) // don't replace by DSA
					{
						it->second->identHash = ident->GetIdentHash ();
						if (m_Storage)
						{
							m_Storage->AddAddress (ident);
							m_Storage->RemoveAddress (it->second->identHash);
						}
						LogPrint (eLogInfo, "Addressbook: Updated host: ", name);
					}
				}
				else
				{
					m_Addresses.emplace (name, std::make_shared<Address>(ident->GetIdentHash ()));
					if (m_Storage) 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;
			if (m_Storage) 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.empty () || s[0] == '#') continue; // skip empty line or comment
					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
			{
				LogPrint (eLogInfo, "Addressbook: Loading subscriptions from config");
				// 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 (const auto& s: subsList)
					if (!s.empty ())
						m_Subscriptions.push_back (std::make_shared<AddressBookSubscription> (*this, s));
				LogPrint (eLogInfo, "Addressbook: ", m_Subscriptions.size (), " subscriptions urls loaded");
			}
		}
		else
			LogPrint (eLogError, "Addressbook: Subscriptions already loaded");
	}

	void AddressBook::LoadLocal ()
	{
		if (!m_Storage) return;
		AddressBookStorage::Addresses localAddresses;
		m_Storage->LoadLocal (localAddresses);
		for (const auto& it: localAddresses)
		{
			if (!it.second->IsIdentHash ()) continue; // skip blinded for now
			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 () && it1->second->IsIdentHash ())
				{
					auto dest = context.FindLocalDestination (it1->second->identHash);
					if (dest)
					{
						// address is ours
						std::shared_ptr<AddressResolver> resolver;
						auto it2 = m_Resolvers.find (it1->second->identHash);
						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->identHash, resolver));
						}
						resolver->AddAddress (it.first, it.second->identHash);
					}
				}
			}
		}
	}

	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_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_after (std::chrono::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 = std::make_unique<boost::asio::steady_timer>(dest->GetService ());
			m_SubscriptionsUpdateTimer->expires_after (std::chrono::minutes(INITIAL_SUBSCRIPTION_UPDATE_TIMEOUT));
			m_SubscriptionsUpdateTimer->async_wait (std::bind (&AddressBook::HandleSubscriptionsUpdateTimer,
				this, std::placeholders::_1));
		}
		else
			LogPrint (eLogCritical, "Addressbook: Can't start subscriptions: missing shared local destination");
	}

	void AddressBook::StopSubscriptions ()
	{
		if (m_SubscriptionsUpdateTimer)
			m_SubscriptionsUpdateTimer->cancel ();
	}

	void AddressBook::ResetTimers ()
	{
		// both timers run on the shared local destination's service, so they must be
		// created anew when that destination is replaced
		auto dest = i2p::client::context.GetSharedLocalDestination ();
		if (m_SubscriptionsUpdateTimer)
		{
			m_SubscriptionsUpdateTimer->cancel ();
			m_SubscriptionsUpdateTimer = nullptr;
			if (dest)
			{
				m_SubscriptionsUpdateTimer = std::make_unique<boost::asio::steady_timer>(dest->GetService ());
				m_SubscriptionsUpdateTimer->expires_after (std::chrono::minutes(INITIAL_SUBSCRIPTION_UPDATE_TIMEOUT));
				m_SubscriptionsUpdateTimer->async_wait (std::bind (&AddressBook::HandleSubscriptionsUpdateTimer,
					this, std::placeholders::_1));
			}
		}
		if (m_AddressCacheUpdateTimer)
		{
			m_AddressCacheUpdateTimer->cancel ();
			m_AddressCacheUpdateTimer = nullptr;
			ScheduleCacheUpdate (); // creates it again on the current destination
		}
	}

	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;
			}
			bool isDownloading = m_Downloading.valid ();
			if (isDownloading && m_Downloading.wait_for(std::chrono::seconds(0)) == std::future_status::ready) // still active?
			{
				m_Downloading.get ();
				isDownloading = false;
			}
			if (!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_Downloading = std::async (std::launch::async,
						std::bind (&AddressBookSubscription::CheckUpdates, m_DefaultSubscription));
				}
				else if (!m_Subscriptions.empty ())
				{
					// pick random subscription
					auto ind = rand () % m_Subscriptions.size();
					m_Downloading = std::async (std::launch::async,
						std::bind (&AddressBookSubscription::CheckUpdates, m_Subscriptions[ind]));
				}
			}
			else
			{
				// try it again later
				m_SubscriptionsUpdateTimer->expires_after (std::chrono::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 (std::string_view address)
	{
		std::shared_ptr<const Address> addr;
		auto dot = address.find ('.');
		if (dot != std::string::npos)
			addr = FindAddress (address.substr (dot + 1));
		if (!addr || !addr->IsIdentHash ()) // TODO:
		{
			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 ", addr->identHash.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.data (), address.length ());
				datagram->SendDatagramTo (buf, len, addr->identHash, 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 ())
			{
				std::lock_guard<std::mutex> l(m_AddressBookMutex);
				m_Addresses[address] = std::make_shared<Address>(hash);
			}
			else
				LogPrint (eLogInfo, "AddressBook: Lookup response: ", address, " not found");
		}
	}

	void AddressBook::ScheduleCacheUpdate ()
	{
		if (!m_AddressCacheUpdateTimer)
		{
			auto dest = i2p::client::context.GetSharedLocalDestination ();
			if(dest)
				m_AddressCacheUpdateTimer = std::make_unique<boost::asio::steady_timer>(dest->GetService ());
		}
		if (m_AddressCacheUpdateTimer)
		{
			m_AddressCacheUpdateTimer->expires_after (std::chrono::seconds(ADDRESS_CACHE_UPDATE_INTERVAL ));
			m_AddressCacheUpdateTimer->async_wait (
				[this](const boost::system::error_code& ecode)
				{
					if (ecode != boost::asio::error::operation_aborted)
					{
						if (m_Storage) m_Storage->CleanUpCache ();
						ScheduleCacheUpdate ();
					}
				});
		}
	}

	AddressBookSubscription::AddressBookSubscription (AddressBook& book, std::string_view link):
		m_Book (book), m_Link (link)
	{
	}

	void AddressBookSubscription::CheckUpdates ()
	{
		i2p::util::SetThreadName("Addressbook");

		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;
		}
		auto addr = m_Book.GetAddress (url.host);
		if (!addr || !addr->IsIdentHash ())
		{
			LogPrint (eLogError, "Addressbook: Can't resolve ", url.host);
			return false;
		}
		else
			m_Ident = addr->identHash;
		// save url parts for later use
		std::string dest_host = url.host;
		int         dest_port = url.port ? url.port : 80;
		// try to create stream to addressbook site
		auto stream = i2p::client::context.GetSharedLocalDestination ()->CreateStream (m_Ident, dest_port);
		if (!stream)
		{
			LogPrint (eLogWarning, "Addressbook: LeaseSet for address ", url.host, " not found");
			return false;
		}
		i2p::util::Defer d ([stream]()
			{
				// destroy stream in destination's thread on exit
				boost::asio::post (i2p::client::context.GetSharedLocalDestination ()->GetService (), [s = std::move (stream)](){});
			});
		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);
		}
		// create http request & send it
		boost::beast::http::request<boost::beast::http::string_body> req (boost::beast::http::verb::get, url.to_string (true), 11); // HTTP 1.1
		req.set (boost::beast::http::field::host, dest_host);
		req.set (boost::beast::http::field::user_agent, "Wget/1.11.4");
		req.set (boost::beast::http::field::accept_encoding, "gzip");
		req.set ("X-Accept-Encoding", "x-i2p-gzip;q=1.0, identity;q=0.5, deflate;q=0, gzip;q=0, *;q=0");
		if (!m_Etag.empty())
			req.set (boost::beast::http::field::if_none_match, m_Etag); // If-None-Match
		if (!m_LastModified.empty())
			req.set (boost::beast::http::field::if_modified_since, m_LastModified); // If-Modified-Since
		req.keep_alive (false); // Connection: close

		i2p::client::BoostAsyncStream httpStream (stream, SUBSCRIPTION_REQUEST_TIMEOUT);
		std::future<size_t> sending = boost::beast::http::async_write (httpStream, req, boost::asio::use_future);
		try
		{
			size_t bytes_transferred = sending.get();
			if (!bytes_transferred)
			{
				LogPrint(eLogWarning, "Addressbook: HTTP request to ", dest_host, " was not sent");
				return false;
			}
		}
		catch (std::exception& ex)
		{
			LogPrint(eLogWarning, "Addressbook: Send HTTP request exception: ", ex.what ());
			return false;
		}
		boost::beast::flat_buffer receiveBuffer;
		boost::beast::http::response<boost::beast::http::string_body> res;
		std::future<size_t> receiving = boost::beast::http::async_read (httpStream, receiveBuffer, res, boost::asio::use_future);
		try
		{
			size_t bytes_transferred = receiving.get();
			if (!bytes_transferred)
			{
				LogPrint(eLogWarning, "Addressbook: HTTP response from ", dest_host, " was not received");
				return false;
			}
		}
		catch (std::exception& ex)
		{
			LogPrint(eLogWarning, "Addressbook: Receive HTTP response exception: ", ex.what ());
			return false;
		}
		stream->AsyncClose ();
		// check result
		if (res.result () != boost::beast::http::status::ok)
		{
			if (res.result_int() == 304)
				LogPrint (eLogInfo, "Addressbook: No updates from ", dest_host, ", code 304");
			else
				LogPrint (eLogWarning, "Addressbook: Response code ", res.result_int(), " from ", dest_host);
			return false;
		}
		// update Etag and Last-Modified
		auto eTag = res[boost::beast::http::field::etag];
		if (!eTag.empty ()) m_Etag = std::string (eTag);
		auto lastModified = res[boost::beast::http::field::last_modified];
		if (!lastModified.empty ()) m_LastModified = std::string (lastModified);
		// parse response
		LogPrint (eLogInfo, "Addressbook: Got update from ", dest_host);
		if (res[boost::beast::http::field::content_encoding] == "gzip" ||
			res[boost::beast::http::field::content_encoding] =="x-i2p-gzip")
		{
			std::stringstream unzipped;
			i2p::data::GzipInflator inflator;
			inflator.Inflate ((const uint8_t *)res.body().data(), res.body().size(), unzipped);
			if (unzipped.fail())
			{
				LogPrint(eLogError, "Addressbook: Can't gunzip http response");
				return false;
			}
			m_Book.LoadHostsFromStream (std::move (unzipped), true);
		}
		else
			m_Book.LoadHostsFromStream (std::stringstream(res.body ()), 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[256];
		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;
	}

}
}
