Browse Source

Create netDb directory structure

pull/15/head
Meeh 11 years ago
parent
commit
ee77bba5e0
  1. 43
      NetDb.cpp
  2. 1
      NetDb.h

43
NetDb.cpp

@ -179,11 +179,41 @@ namespace data
return nullptr; return nullptr;
} }
// TODO: Move to reseed and/or scheduled tasks. (In java version, scheduler fix this as well as sort RIs.)
bool NetDb::CreateNetDb(const char * directory)
{
boost::filesystem::path p (directory);
LogPrint (directory, " doesn't exist, trying to create it.");
if (!boost::filesystem::create_directory (p))
{
LogPrint("Failed to create directory ", directory);
return false;
}
// Random chars
std::string chars = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-~";
boost::filesystem::path suffix;
for (auto c : chars)
{
#ifndef _WIN32
suffix = "/r";
#else
suffix = "\\r";
#endif
suffix += c;
if (!boost::filesystem::create_directory( boost::filesystem::path (p / suffix) )) return false;
}
return true;
}
void NetDb::Load (const char * directory) void NetDb::Load (const char * directory)
{ {
boost::filesystem::path p (directory); boost::filesystem::path p (directory);
if (boost::filesystem::exists (p)) if (!boost::filesystem::exists (p))
{ {
if (!CreateNetDb(directory)) return;
}
// TODO: Reseed if needed.
int numRouters = 0; int numRouters = 0;
boost::filesystem::directory_iterator end; boost::filesystem::directory_iterator end;
for (boost::filesystem::directory_iterator it (p); it != end; ++it) for (boost::filesystem::directory_iterator it (p); it != end; ++it)
@ -204,16 +234,18 @@ namespace data
} }
LogPrint (numRouters, " routers loaded"); LogPrint (numRouters, " routers loaded");
} }
else
LogPrint (directory, " doesn't exist");
}
void NetDb::SaveUpdated (const char * directory) void NetDb::SaveUpdated (const char * directory)
{ {
auto GetFilePath = [](const char * directory, const RouterInfo * routerInfo) auto GetFilePath = [](const char * directory, const RouterInfo * routerInfo)
{ {
#ifndef _WIN32
return std::string (directory) + "/r" + return std::string (directory) + "/r" +
routerInfo->GetIdentHashBase64 ()[0] + "/routerInfo-" + routerInfo->GetIdentHashBase64 ()[0] + "/routerInfo-" +
#else
return std::string (directory) + "\\r" +
routerInfo->GetIdentHashBase64 ()[0] + "\\routerInfo-" +
#endif
routerInfo->GetIdentHashBase64 () + ".dat"; routerInfo->GetIdentHashBase64 () + ".dat";
}; };
@ -570,6 +602,9 @@ namespace data
return r; return r;
} }
//TODO: Move to reseed.
//TODO: Implement v1 & v2 reseeding. Lightweight zip library is needed for v2.
//TODO: Implement SU3, utils.
void NetDb::DownloadRouterInfo (const std::string& address, const std::string& filename) void NetDb::DownloadRouterInfo (const std::string& address, const std::string& filename)
{ {
try try

1
NetDb.h

@ -76,6 +76,7 @@ namespace data
private: private:
bool CreateNetDb(const char * directory);
void Load (const char * directory); void Load (const char * directory);
void SaveUpdated (const char * directory); void SaveUpdated (const char * directory);
void DownloadRouterInfo (const std::string& address, const std::string& filename); // for reseed void DownloadRouterInfo (const std::string& address, const std::string& filename); // for reseed

Loading…
Cancel
Save