mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 08:00:38 +03:00
104 lines
2.1 KiB
C++
104 lines
2.1 KiB
C++
#include <string.h>
|
|
#include <string>
|
|
#include <map>
|
|
#include "base64.h"
|
|
#include "util.h"
|
|
#include "Identity.h"
|
|
#include "Log.h"
|
|
#include "AddressBook.h"
|
|
|
|
#include <boost/algorithm/string.hpp>
|
|
|
|
namespace i2p
|
|
{
|
|
namespace data
|
|
{
|
|
|
|
AddressBook::AddressBook (): m_IsLoaded (false)
|
|
{
|
|
}
|
|
|
|
|
|
const IdentHash * AddressBook::FindAddress (const std::string& address)
|
|
{
|
|
if (!m_IsLoaded)
|
|
LoadHosts ();
|
|
auto it = m_Addresses.find (address);
|
|
if (it != m_Addresses.end ())
|
|
return &it->second;
|
|
else
|
|
return nullptr;
|
|
}
|
|
|
|
void AddressBook::LoadHostsFromI2P ()
|
|
{
|
|
std::string content;
|
|
|
|
std::stringstream url_ss;
|
|
// TODO: hosts link in config
|
|
// TODO: url download via HTTPProxy
|
|
url_ss << "http://127.0.0.1:" << i2p::util::config::GetArg("-httpport", 7070) << "/udhdrtrcetjm5sxzskjyr5ztpeszydbh4dpl3pl4utgqqw2v4jna/hosts.txt";
|
|
while (true)
|
|
{
|
|
content = i2p::util::http::httpRequest(url_ss.str());
|
|
|
|
// TODO: check http errors
|
|
if (! boost::starts_with(content, "<html>"))
|
|
break;
|
|
std::this_thread::sleep_for(std::chrono::seconds(5));
|
|
}
|
|
|
|
std::ofstream f_save(i2p::util::filesystem::GetFullPath("hosts.txt").c_str(), std::ofstream::out);
|
|
if (f_save.is_open())
|
|
{
|
|
f_save << content;
|
|
f_save.close();
|
|
}
|
|
else
|
|
LogPrint("Can't write hosts.txt");
|
|
m_IsLoaded = false;
|
|
return;
|
|
}
|
|
|
|
void AddressBook::LoadHosts ()
|
|
{
|
|
m_IsLoaded = true;
|
|
std::ifstream f (i2p::util::filesystem::GetFullPath ("hosts.txt").c_str (), std::ofstream::in); // in text mode
|
|
if (!f.is_open ())
|
|
{
|
|
LogPrint ("hosts.txt not found. Try to load...");
|
|
std::thread load_hosts(&AddressBook::LoadHostsFromI2P, this);
|
|
load_hosts.detach();
|
|
return;
|
|
}
|
|
int numAddresses = 0;
|
|
|
|
std::string s;
|
|
|
|
while (!f.eof ())
|
|
{
|
|
getline(f, s);
|
|
|
|
if (!s.length())
|
|
break;
|
|
|
|
size_t pos = s.find('=');
|
|
|
|
if (pos != std::string::npos)
|
|
{
|
|
std::string name = s.substr(0, pos++);
|
|
std::string addr = s.substr(pos);
|
|
|
|
Identity ident;
|
|
Base64ToByteStream (addr.c_str(), addr.length(), (uint8_t *)&ident, sizeof (ident));
|
|
m_Addresses[name] = CalculateIdentHash (ident);
|
|
numAddresses++;
|
|
}
|
|
}
|
|
LogPrint (numAddresses, " addresses loaded");
|
|
}
|
|
|
|
}
|
|
}
|
|
|