i2pd/AddressBook.cpp

71 lines
1.2 KiB
C++
Raw Normal View History

2014-04-01 19:18:14 +00:00
#include <string.h>
#include <string>
#include <map>
#include "base64.h"
#include "util.h"
#include "Identity.h"
#include "Log.h"
#include "AddressBook.h"
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::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");
return;
}
int numAddresses = 0;
2014-04-01 20:27:40 +00:00
std::string s;
2014-04-01 19:18:14 +00:00
while (!f.eof ())
{
2014-04-01 20:27:40 +00:00
getline(f, s);
if (!s.length())
break;
size_t pos = s.find('=');
if (pos != std::string::npos)
2014-04-01 19:18:14 +00:00
{
2014-04-01 20:27:40 +00:00
std::string name = s.substr(0, pos++);
std::string addr = s.substr(pos);
2014-04-01 19:18:14 +00:00
Identity ident;
2014-04-01 20:27:40 +00:00
Base64ToByteStream (addr.c_str(), addr.length(), (uint8_t *)&ident, sizeof (ident));
m_Addresses[name] = CalculateIdentHash (ident);
2014-04-01 19:18:14 +00:00
numAddresses++;
}
}
LogPrint (numAddresses, " addresses loaded");
}
}
}