mirror of https://github.com/PurpleI2P/i2pd.git
orignal
11 years ago
4 changed files with 88 additions and 0 deletions
@ -0,0 +1,73 @@ |
|||||||
|
#ifndef ADDRESS_BOOK_H__ |
||||||
|
#define ADDRESS_BOOK_H__ |
||||||
|
|
||||||
|
#include <string.h> |
||||||
|
#include <string> |
||||||
|
#include <map> |
||||||
|
#include "base64.h" |
||||||
|
#include "util.h" |
||||||
|
#include "Identity.h" |
||||||
|
#include "Log.h" |
||||||
|
|
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace data |
||||||
|
{ |
||||||
|
class AddressBook |
||||||
|
{ |
||||||
|
public: |
||||||
|
|
||||||
|
AddressBook (): m_IsLoaded (false) {}; |
||||||
|
|
||||||
|
const IdentHash * 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; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
void 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; |
||||||
|
char str[1024]; |
||||||
|
while (!f.eof ()) |
||||||
|
{ |
||||||
|
f.getline (str, 1024); |
||||||
|
char * key = strchr (str, '='); |
||||||
|
if (key) |
||||||
|
{ |
||||||
|
*key = 0; |
||||||
|
key++; |
||||||
|
Identity ident; |
||||||
|
Base64ToByteStream (key, strlen(key), (uint8_t *)&ident, sizeof (ident)); |
||||||
|
m_Addresses[str] = CalculateIdentHash (ident); |
||||||
|
numAddresses++; |
||||||
|
} |
||||||
|
} |
||||||
|
LogPrint (numAddresses, " addresses loaded"); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
std::map<std::string, IdentHash> m_Addresses; |
||||||
|
bool m_IsLoaded; |
||||||
|
}; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#endif |
||||||
|
|
||||||
|
|
Loading…
Reference in new issue