Browse Source

save new RouterInfos to ~/.i2pd

pull/24/merge
orignal 11 years ago
parent
commit
8963c0b309
  1. 40
      NetDb.cpp
  2. 3
      NetDb.h

40
NetDb.cpp

@ -2,7 +2,6 @@
#include <fstream> #include <fstream>
#include <vector> #include <vector>
#include <boost/asio.hpp> #include <boost/asio.hpp>
#include <boost/filesystem.hpp>
#include <cryptopp/gzip.h> #include <cryptopp/gzip.h>
#include "base64.h" #include "base64.h"
#include "Log.h" #include "Log.h"
@ -20,7 +19,6 @@ namespace i2p
{ {
namespace data namespace data
{ {
I2NPMessage * RequestedDestination::CreateRequestMessage (const RouterInfo * router, I2NPMessage * RequestedDestination::CreateRequestMessage (const RouterInfo * router,
const i2p::tunnel::InboundTunnel * replyTunnel) const i2p::tunnel::InboundTunnel * replyTunnel)
{ {
@ -43,7 +41,12 @@ namespace data
m_LastReplyTunnel = nullptr; m_LastReplyTunnel = nullptr;
return msg; return msg;
} }
#ifndef _WIN32
const char NetDb::m_NetDbPath[] = "/netDb";
#else
const char NetDb::m_NetDbPath[] = "\\netDb";
#endif
NetDb netdb; NetDb netdb;
NetDb::NetDb (): m_IsRunning (false), m_ReseedRetries (0), m_Thread (0) NetDb::NetDb (): m_IsRunning (false), m_ReseedRetries (0), m_Thread (0)
@ -62,22 +65,14 @@ namespace data
} }
void NetDb::Start () void NetDb::Start ()
{ {
#ifndef _WIN32 Load (m_NetDbPath);
Load ("/netDb");
#else
Load ("\\netDb");
#endif
while (m_RouterInfos.size () < 100 && m_ReseedRetries < 10) while (m_RouterInfos.size () < 100 && m_ReseedRetries < 10)
{ {
Reseeder reseeder; Reseeder reseeder;
reseeder.reseedNow(); reseeder.reseedNow();
m_ReseedRetries++; m_ReseedRetries++;
#ifndef _WIN32 Load (m_NetDbPath);
Load ("/netDb");
#else
Load ("\\netDb");
#endif
} }
m_Thread = new std::thread (std::bind (&NetDb::Run, this)); m_Thread = new std::thread (std::bind (&NetDb::Run, this));
} }
@ -128,7 +123,7 @@ namespace data
if (ts - lastTs >= 60) // save routers every minute if (ts - lastTs >= 60) // save routers every minute
{ {
if (lastTs) if (lastTs)
SaveUpdated ("netDb"); SaveUpdated (m_NetDbPath);
lastTs = ts; lastTs = ts;
} }
} }
@ -270,7 +265,14 @@ namespace data
#endif #endif
routerInfo->GetIdentHashBase64 () + ".dat"; routerInfo->GetIdentHashBase64 () + ".dat";
}; };
boost::filesystem::path p (i2p::util::filesystem::GetDataDir());
p /= (directory);
#if BOOST_VERSION > 10500
const char * fullDirectory = p.string().c_str ();
#else
const char * fullDirectory = p.c_str ();
#endif
int count = 0, deletedCount = 0; int count = 0, deletedCount = 0;
auto total = m_RouterInfos.size (); auto total = m_RouterInfos.size ();
uint64_t ts = i2p::util::GetMillisecondsSinceEpoch (); uint64_t ts = i2p::util::GetMillisecondsSinceEpoch ();
@ -278,7 +280,7 @@ namespace data
{ {
if (it.second->IsUpdated ()) if (it.second->IsUpdated ())
{ {
std::ofstream r (GetFilePath(directory, it.second), std::ofstream::binary); std::ofstream r (GetFilePath(fullDirectory, it.second), std::ofstream::binary);
r.write ((char *)it.second->GetBuffer (), it.second->GetBufferLen ()); r.write ((char *)it.second->GetBuffer (), it.second->GetBufferLen ());
it.second->SetUpdated (false); it.second->SetUpdated (false);
count++; count++;
@ -294,9 +296,9 @@ namespace data
if (it.second->IsUnreachable ()) if (it.second->IsUnreachable ())
{ {
if (boost::filesystem::exists (GetFilePath (directory, it.second))) if (boost::filesystem::exists (GetFilePath (fullDirectory, it.second)))
{ {
boost::filesystem::remove (GetFilePath (directory, it.second)); boost::filesystem::remove (GetFilePath (fullDirectory, it.second));
deletedCount++; deletedCount++;
} }
} }

3
NetDb.h

@ -6,6 +6,7 @@
#include <map> #include <map>
#include <string> #include <string>
#include <thread> #include <thread>
#include <boost/filesystem.hpp>
#include "Queue.h" #include "Queue.h"
#include "I2NPProtocol.h" #include "I2NPProtocol.h"
#include "RouterInfo.h" #include "RouterInfo.h"
@ -97,6 +98,8 @@ namespace data
int m_ReseedRetries; int m_ReseedRetries;
std::thread * m_Thread; std::thread * m_Thread;
i2p::util::Queue<I2NPMessage> m_Queue; // of I2NPDatabaseStoreMsg i2p::util::Queue<I2NPMessage> m_Queue; // of I2NPDatabaseStoreMsg
static const char m_NetDbPath[];
}; };
extern NetDb netdb; extern NetDb netdb;

Loading…
Cancel
Save