mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 08:00:38 +03:00
108 lines
3.0 KiB
C++
108 lines
3.0 KiB
C++
#include <boost/filesystem.hpp>
|
|
#include <boost/property_tree/ptree.hpp>
|
|
#include <boost/property_tree/ini_parser.hpp>
|
|
#include "base64.h"
|
|
#include "util.h"
|
|
#include "Profiling.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace data
|
|
{
|
|
RouterProfile::RouterProfile (const IdentHash& identHash):
|
|
m_IdentHash (identHash), m_NumTunnelsAgreed (0), m_NumTunnelsDeclined (0)
|
|
{
|
|
}
|
|
|
|
void RouterProfile::Save ()
|
|
{
|
|
// fill sections
|
|
boost::property_tree::ptree participation;
|
|
participation.put (PEER_PROFILE_PARTICIPATION_AGREED, m_NumTunnelsAgreed);
|
|
participation.put (PEER_PROFILE_PARTICIPATION_DECLINED, m_NumTunnelsDeclined);
|
|
// fill property tree
|
|
boost::property_tree::ptree pt;
|
|
pt.put_child (PEER_PROFILE_SECTION_PARTICIPATION, participation);
|
|
|
|
// save to file
|
|
auto path = i2p::util::filesystem::GetDefaultDataDir() / PEER_PROFILES_DIRECTORY;
|
|
if (!boost::filesystem::exists (path))
|
|
{
|
|
// Create directory is necessary
|
|
if (!boost::filesystem::create_directory (path))
|
|
{
|
|
LogPrint (eLogError, "Failed to create directory ", path);
|
|
return;
|
|
}
|
|
const char * chars = GetBase64SubstitutionTable (); // 64 bytes
|
|
for (int i = 0; i < 64; i++)
|
|
{
|
|
auto path1 = path / (std::string ("p") + chars[i]);
|
|
if (!boost::filesystem::create_directory (path1))
|
|
{
|
|
LogPrint (eLogError, "Failed to create directory ", path1);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
std::string base64 = m_IdentHash.ToBase64 ();
|
|
path = path / (std::string ("p") + base64[0]);
|
|
auto filename = path / (std::string (PEER_PROFILE_PREFIX) + base64 + ".txt");
|
|
try
|
|
{
|
|
boost::property_tree::write_ini (filename.string (), pt);
|
|
}
|
|
catch (std::exception& ex)
|
|
{
|
|
LogPrint (eLogError, "Can't write ", filename, ": ", ex.what ());
|
|
}
|
|
}
|
|
|
|
void RouterProfile::Load ()
|
|
{
|
|
std::string base64 = m_IdentHash.ToBase64 ();
|
|
auto path = i2p::util::filesystem::GetDefaultDataDir() / PEER_PROFILES_DIRECTORY;
|
|
path /= std::string ("p") + base64[0];
|
|
auto filename = path / (std::string (PEER_PROFILE_PREFIX) + base64 + ".txt");
|
|
if (boost::filesystem::exists (filename))
|
|
{
|
|
boost::property_tree::ptree pt;
|
|
try
|
|
{
|
|
boost::property_tree::read_ini (filename.string (), pt);
|
|
}
|
|
catch (std::exception& ex)
|
|
{
|
|
LogPrint (eLogError, "Can't read ", filename, ": ", ex.what ());
|
|
return;
|
|
}
|
|
try
|
|
{
|
|
// read participations
|
|
auto participations = pt.get_child (PEER_PROFILE_SECTION_PARTICIPATION);
|
|
m_NumTunnelsAgreed = participations.get (PEER_PROFILE_PARTICIPATION_AGREED, 0);
|
|
m_NumTunnelsDeclined = participations.get (PEER_PROFILE_PARTICIPATION_DECLINED, 0);
|
|
}
|
|
catch (std::exception& ex)
|
|
{
|
|
LogPrint (eLogError, "Can't read profile ", base64, " :", ex.what ());
|
|
}
|
|
}
|
|
}
|
|
|
|
void RouterProfile::TunnelBuildResponse (uint8_t ret)
|
|
{
|
|
if (ret > 0)
|
|
m_NumTunnelsDeclined++;
|
|
else
|
|
m_NumTunnelsAgreed++;
|
|
}
|
|
|
|
std::shared_ptr<RouterProfile> GetRouterProfile (const IdentHash& identHash)
|
|
{
|
|
auto profile = std::make_shared<RouterProfile> (identHash);
|
|
profile->Load (); // if possible
|
|
return profile;
|
|
}
|
|
}
|
|
}
|