i2pd/RouterContext.cpp

99 lines
2.8 KiB
C++
Raw Normal View History

2013-10-23 02:45:40 +00:00
#include <fstream>
#include <cryptopp/dh.h>
#include <cryptopp/dsa.h>
#include "CryptoConst.h"
#include "RouterContext.h"
2014-01-30 00:56:48 +00:00
#include "util.h"
2013-10-23 02:45:40 +00:00
namespace i2p
{
RouterContext context;
RouterContext::RouterContext ()
{
if (!Load ())
CreateNewRouter ();
Save ();
}
void RouterContext::CreateNewRouter ()
{
2013-12-21 01:22:55 +00:00
m_Keys = i2p::data::CreateRandomKeys ();
m_SigningPrivateKey.Initialize (i2p::crypto::dsap, i2p::crypto::dsaq, i2p::crypto::dsag,
CryptoPP::Integer (m_Keys.signingPrivateKey, 20));
2014-02-23 16:48:09 +00:00
UpdateRouterInfo ();
}
void RouterContext::UpdateRouterInfo ()
{
2013-12-21 01:22:55 +00:00
i2p::data::Identity ident;
2013-12-22 16:29:57 +00:00
ident = m_Keys;
2013-10-23 02:45:40 +00:00
2014-02-24 01:48:28 +00:00
i2p::data::RouterInfo routerInfo;
routerInfo.SetRouterIdentity (ident);
2014-02-25 00:25:26 +00:00
routerInfo.AddSSUAddress ("127.0.0.1", 17007, routerInfo.GetIdentHash ());
2014-02-24 01:48:28 +00:00
routerInfo.AddNTCPAddress ("127.0.0.1", 17007); // TODO:
routerInfo.SetProperty ("caps", "LR");
2014-07-08 11:45:16 +00:00
routerInfo.SetProperty ("coreVersion", "0.9.11");
2014-02-24 01:48:28 +00:00
routerInfo.SetProperty ("netId", "2");
2014-07-08 11:45:16 +00:00
routerInfo.SetProperty ("router.version", "0.9.11");
2014-02-24 01:48:28 +00:00
routerInfo.SetProperty ("start_uptime", "90m");
routerInfo.CreateBuffer ();
2013-10-23 02:45:40 +00:00
2014-07-22 12:03:02 +00:00
m_RouterInfo.Update (routerInfo.GetBuffer (), routerInfo.GetBufferLen ());
2014-02-23 16:48:09 +00:00
}
2013-12-10 13:10:49 +00:00
void RouterContext::OverrideNTCPAddress (const char * host, int port)
{
m_RouterInfo.CreateBuffer ();
2014-02-09 13:52:56 +00:00
auto address = const_cast<i2p::data::RouterInfo::Address *>(m_RouterInfo.GetNTCPAddress ());
2013-12-10 13:10:49 +00:00
if (address)
{
2014-01-21 21:07:16 +00:00
address->host = boost::asio::ip::address::from_string (host);
2013-12-10 13:10:49 +00:00
address->port = port;
}
m_RouterInfo.CreateBuffer ();
2014-02-23 16:48:09 +00:00
Save (true);
2013-12-10 13:10:49 +00:00
}
2014-02-09 02:06:40 +00:00
void RouterContext::UpdateAddress (const char * host)
{
for (auto& address : m_RouterInfo.GetAddresses ())
address.host = boost::asio::ip::address::from_string (host);
m_RouterInfo.CreateBuffer ();
}
2013-12-10 13:10:49 +00:00
2014-07-29 17:44:54 +00:00
void RouterContext::Sign (const uint8_t * buf, int len, uint8_t * signature) const
2013-10-23 02:45:40 +00:00
{
CryptoPP::DSA::Signer signer (m_SigningPrivateKey);
2014-07-29 17:44:54 +00:00
signer.SignMessage (i2p::context.GetRandomNumberGenerator (), buf, len, signature);
2013-10-23 02:45:40 +00:00
}
bool RouterContext::Load ()
{
2014-03-14 11:32:11 +00:00
std::ifstream fk (i2p::util::filesystem::GetFullPath (ROUTER_KEYS).c_str (), std::ifstream::binary | std::ofstream::in);
2013-10-23 02:45:40 +00:00
if (!fk.is_open ()) return false;
2013-12-21 01:22:55 +00:00
fk.read ((char *)&m_Keys, sizeof (m_Keys));
2013-10-23 02:45:40 +00:00
m_SigningPrivateKey.Initialize (i2p::crypto::dsap, i2p::crypto::dsaq, i2p::crypto::dsag,
2013-12-21 01:22:55 +00:00
CryptoPP::Integer (m_Keys.signingPrivateKey, 20));
2013-10-23 02:45:40 +00:00
2014-07-24 01:45:45 +00:00
i2p::data::RouterInfo routerInfo(i2p::util::filesystem::GetFullPath (ROUTER_INFO)); // TODO
m_RouterInfo.Update (routerInfo.GetBuffer (), routerInfo.GetBufferLen ());
2013-10-23 02:45:40 +00:00
return true;
}
2014-02-23 16:48:09 +00:00
void RouterContext::Save (bool infoOnly)
2013-10-23 02:45:40 +00:00
{
2014-02-23 16:48:09 +00:00
if (!infoOnly)
{
2014-03-14 11:32:11 +00:00
std::ofstream fk (i2p::util::filesystem::GetFullPath (ROUTER_KEYS).c_str (), std::ofstream::binary | std::ofstream::out);
2014-02-23 16:48:09 +00:00
fk.write ((char *)&m_Keys, sizeof (m_Keys));
}
2014-07-10 19:33:42 +00:00
m_RouterInfo.SaveToFile (i2p::util::filesystem::GetFullPath (ROUTER_INFO));
2013-10-23 02:45:40 +00:00
}
2014-01-21 21:07:16 +00:00
}