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.
134 lines
2.8 KiB
C++
134 lines
2.8 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 client
|
|
{
|
|
|
|
AddressBook::AddressBook (): m_IsLoaded (false), m_IsDowloading (false)
|
|
{
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
const i2p::data::IdentHash * AddressBook::FindAddress (const std::string& address)
|
|
{
|
|
if (!m_IsLoaded)
|
|
LoadHosts ();
|
|
if (m_IsLoaded)
|
|
{
|
|
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)
|
|
{
|
|
i2p::data::IdentityEx ident;
|
|
ident.FromBase64 (base64);
|
|
m_Addresses[address] = ident.GetIdentHash ();
|
|
LogPrint (address,"->",ident.GetIdentHash ().ToBase32 (), ".b32.i2p added");
|
|
}
|
|
|
|
void AddressBook::LoadHostsFromI2P ()
|
|
{
|
|
std::string content;
|
|
int http_code = i2p::util::http::httpRequestViaI2pProxy("http://udhdrtrcetjm5sxzskjyr5ztpeszydbh4dpl3pl4utgqqw2v4jna.b32.i2p/hosts.txt", content);
|
|
if (http_code == 200)
|
|
{
|
|
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;
|
|
}
|
|
else
|
|
LogPrint ("Failed to download hosts.txt");
|
|
m_IsDowloading = false;
|
|
|
|
return;
|
|
}
|
|
|
|
void AddressBook::LoadHosts ()
|
|
{
|
|
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...");
|
|
if (!m_IsDowloading)
|
|
{
|
|
m_IsDowloading = true;
|
|
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())
|
|
continue; // skip empty line
|
|
|
|
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::IdentityEx ident;
|
|
ident.FromBase64(addr);
|
|
m_Addresses[name] = ident.GetIdentHash ();
|
|
numAddresses++;
|
|
}
|
|
}
|
|
LogPrint (numAddresses, " addresses loaded");
|
|
m_IsLoaded = true;
|
|
}
|
|
|
|
}
|
|
}
|
|
|