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 @@
@@ -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