mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 08:00:38 +03:00
336 lines
11 KiB
C++
336 lines
11 KiB
C++
#include <fstream>
|
|
#include <iostream>
|
|
#include <boost/program_options/parsers.hpp>
|
|
#include <boost/program_options/variables_map.hpp>
|
|
#include "util.h"
|
|
#include "Log.h"
|
|
#include "Identity.h"
|
|
#include "ClientContext.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace client
|
|
{
|
|
ClientContext context;
|
|
|
|
ClientContext::ClientContext (): m_SharedLocalDestination (nullptr),
|
|
m_HttpProxy (nullptr), m_SocksProxy (nullptr), m_SamBridge (nullptr),
|
|
m_BOBCommandChannel (nullptr), m_I2PControlService (nullptr)
|
|
{
|
|
}
|
|
|
|
ClientContext::~ClientContext ()
|
|
{
|
|
delete m_HttpProxy;
|
|
delete m_SocksProxy;
|
|
delete m_SamBridge;
|
|
delete m_BOBCommandChannel;
|
|
delete m_I2PControlService;
|
|
}
|
|
|
|
void ClientContext::Start ()
|
|
{
|
|
if (!m_SharedLocalDestination)
|
|
{
|
|
m_SharedLocalDestination = CreateNewLocalDestination (); // non-public, DSA
|
|
m_Destinations[m_SharedLocalDestination->GetIdentity ().GetIdentHash ()] = m_SharedLocalDestination;
|
|
m_SharedLocalDestination->Start ();
|
|
}
|
|
|
|
// proxies
|
|
m_HttpProxy = new i2p::proxy::HTTPProxy(i2p::util::config::GetArg("-httpproxyport", 4446));
|
|
m_HttpProxy->Start();
|
|
LogPrint("HTTP Proxy started");
|
|
m_SocksProxy = new i2p::proxy::SOCKSProxy(i2p::util::config::GetArg("-socksproxyport", 4447));
|
|
m_SocksProxy->Start();
|
|
LogPrint("SOCKS Proxy Started");
|
|
|
|
// I2P tunnels
|
|
std::string ircDestination = i2p::util::config::GetArg("-ircdest", "");
|
|
if (ircDestination.length () > 0) // ircdest is presented
|
|
{
|
|
ClientDestination * localDestination = nullptr;
|
|
std::string ircKeys = i2p::util::config::GetArg("-irckeys", "");
|
|
if (ircKeys.length () > 0)
|
|
localDestination = LoadLocalDestination (ircKeys, false);
|
|
auto ircPort = i2p::util::config::GetArg("-ircport", 6668);
|
|
auto ircTunnel = new I2PClientTunnel (ircDestination, ircPort, localDestination);
|
|
ircTunnel->Start ();
|
|
m_ClientTunnels.insert (std::make_pair(ircPort, std::unique_ptr<I2PClientTunnel>(ircTunnel)));
|
|
LogPrint("IRC tunnel started");
|
|
}
|
|
std::string eepKeys = i2p::util::config::GetArg("-eepkeys", "");
|
|
if (eepKeys.length () > 0) // eepkeys file is presented
|
|
{
|
|
auto localDestination = LoadLocalDestination (eepKeys, true);
|
|
auto serverTunnel = new I2PServerTunnel (i2p::util::config::GetArg("-eephost", "127.0.0.1"),
|
|
i2p::util::config::GetArg("-eepport", 80), localDestination);
|
|
serverTunnel->Start ();
|
|
m_ServerTunnels.insert (std::make_pair(localDestination->GetIdentHash (), std::unique_ptr<I2PServerTunnel>(serverTunnel)));
|
|
LogPrint("Server tunnel started");
|
|
}
|
|
ReadTunnels ();
|
|
|
|
// SAM
|
|
int samPort = i2p::util::config::GetArg("-samport", 0);
|
|
if (samPort)
|
|
{
|
|
m_SamBridge = new SAMBridge (samPort);
|
|
m_SamBridge->Start ();
|
|
LogPrint("SAM bridge started");
|
|
}
|
|
|
|
// BOB
|
|
int bobPort = i2p::util::config::GetArg("-bobport", 0);
|
|
if (bobPort)
|
|
{
|
|
m_BOBCommandChannel = new BOBCommandChannel (bobPort);
|
|
m_BOBCommandChannel->Start ();
|
|
LogPrint("BOB command channel started");
|
|
}
|
|
|
|
// I2P Control
|
|
int i2pcontrolPort = i2p::util::config::GetArg("-i2pcontrolport", 0);
|
|
if (i2pcontrolPort)
|
|
{
|
|
m_I2PControlService = new I2PControlService (i2pcontrolPort);
|
|
m_I2PControlService->Start ();
|
|
LogPrint("I2PControl started");
|
|
}
|
|
m_AddressBook.StartSubscriptions ();
|
|
}
|
|
|
|
void ClientContext::Stop ()
|
|
{
|
|
m_AddressBook.StopSubscriptions ();
|
|
m_HttpProxy->Stop();
|
|
delete m_HttpProxy;
|
|
m_HttpProxy = nullptr;
|
|
LogPrint("HTTP Proxy stopped");
|
|
m_SocksProxy->Stop();
|
|
delete m_SocksProxy;
|
|
m_SocksProxy = nullptr;
|
|
LogPrint("SOCKS Proxy stopped");
|
|
for (auto& it: m_ClientTunnels)
|
|
{
|
|
it.second->Stop ();
|
|
LogPrint("I2P client tunnel on port ", it.first, " stopped");
|
|
}
|
|
m_ClientTunnels.clear ();
|
|
for (auto& it: m_ServerTunnels)
|
|
{
|
|
it.second->Stop ();
|
|
LogPrint("I2P server tunnel stopped");
|
|
}
|
|
m_ServerTunnels.clear ();
|
|
if (m_SamBridge)
|
|
{
|
|
m_SamBridge->Stop ();
|
|
delete m_SamBridge;
|
|
m_SamBridge = nullptr;
|
|
LogPrint("SAM brdige stopped");
|
|
}
|
|
if (m_BOBCommandChannel)
|
|
{
|
|
m_BOBCommandChannel->Stop ();
|
|
delete m_BOBCommandChannel;
|
|
m_BOBCommandChannel = nullptr;
|
|
LogPrint("BOB command channel stopped");
|
|
}
|
|
if (m_I2PControlService)
|
|
{
|
|
m_I2PControlService->Stop ();
|
|
delete m_I2PControlService;
|
|
m_I2PControlService = nullptr;
|
|
LogPrint("I2PControl stopped");
|
|
}
|
|
|
|
for (auto it: m_Destinations)
|
|
{
|
|
it.second->Stop ();
|
|
delete it.second;
|
|
}
|
|
m_Destinations.clear ();
|
|
m_SharedLocalDestination = 0; // deleted through m_Destination
|
|
}
|
|
|
|
ClientDestination * ClientContext::LoadLocalDestination (const std::string& filename, bool isPublic)
|
|
{
|
|
i2p::data::PrivateKeys keys;
|
|
std::string fullPath = i2p::util::filesystem::GetFullPath (filename);
|
|
std::ifstream s(fullPath.c_str (), std::ifstream::binary);
|
|
if (s.is_open ())
|
|
{
|
|
s.seekg (0, std::ios::end);
|
|
size_t len = s.tellg();
|
|
s.seekg (0, std::ios::beg);
|
|
uint8_t * buf = new uint8_t[len];
|
|
s.read ((char *)buf, len);
|
|
keys.FromBuffer (buf, len);
|
|
delete[] buf;
|
|
LogPrint ("Local address ", m_AddressBook.ToAddress(keys.GetPublic ().GetIdentHash ()), " loaded");
|
|
}
|
|
else
|
|
{
|
|
LogPrint ("Can't open file ", fullPath, " Creating new one");
|
|
keys = i2p::data::PrivateKeys::CreateRandomKeys (i2p::data::SIGNING_KEY_TYPE_DSA_SHA1);
|
|
std::ofstream f (fullPath, std::ofstream::binary | std::ofstream::out);
|
|
size_t len = keys.GetFullLen ();
|
|
uint8_t * buf = new uint8_t[len];
|
|
len = keys.ToBuffer (buf, len);
|
|
f.write ((char *)buf, len);
|
|
delete[] buf;
|
|
|
|
LogPrint ("New private keys file ", fullPath, " for ", m_AddressBook.ToAddress(keys.GetPublic ().GetIdentHash ()), " created");
|
|
}
|
|
|
|
ClientDestination * localDestination = nullptr;
|
|
std::unique_lock<std::mutex> l(m_DestinationsMutex);
|
|
auto it = m_Destinations.find (keys.GetPublic ().GetIdentHash ());
|
|
if (it != m_Destinations.end ())
|
|
{
|
|
LogPrint (eLogWarning, "Local destination ", m_AddressBook.ToAddress(keys.GetPublic ().GetIdentHash ()), " alreday exists");
|
|
localDestination = it->second;
|
|
}
|
|
else
|
|
{
|
|
localDestination = new ClientDestination (keys, isPublic);
|
|
m_Destinations[localDestination->GetIdentHash ()] = localDestination;
|
|
localDestination->Start ();
|
|
}
|
|
return localDestination;
|
|
}
|
|
|
|
ClientDestination * ClientContext::CreateNewLocalDestination (bool isPublic, i2p::data::SigningKeyType sigType,
|
|
const std::map<std::string, std::string> * params)
|
|
{
|
|
i2p::data::PrivateKeys keys = i2p::data::PrivateKeys::CreateRandomKeys (sigType);
|
|
auto localDestination = new ClientDestination (keys, isPublic, params);
|
|
std::unique_lock<std::mutex> l(m_DestinationsMutex);
|
|
m_Destinations[localDestination->GetIdentHash ()] = localDestination;
|
|
localDestination->Start ();
|
|
return localDestination;
|
|
}
|
|
|
|
void ClientContext::DeleteLocalDestination (ClientDestination * destination)
|
|
{
|
|
if (!destination) return;
|
|
auto it = m_Destinations.find (destination->GetIdentHash ());
|
|
if (it != m_Destinations.end ())
|
|
{
|
|
auto d = it->second;
|
|
{
|
|
std::unique_lock<std::mutex> l(m_DestinationsMutex);
|
|
m_Destinations.erase (it);
|
|
}
|
|
d->Stop ();
|
|
delete d;
|
|
}
|
|
}
|
|
|
|
ClientDestination * ClientContext::CreateNewLocalDestination (const i2p::data::PrivateKeys& keys, bool isPublic,
|
|
const std::map<std::string, std::string> * params)
|
|
{
|
|
auto it = m_Destinations.find (keys.GetPublic ().GetIdentHash ());
|
|
if (it != m_Destinations.end ())
|
|
{
|
|
LogPrint ("Local destination ", m_AddressBook.ToAddress(keys.GetPublic ().GetIdentHash ()), " exists");
|
|
if (!it->second->IsRunning ())
|
|
{
|
|
it->second->Start ();
|
|
return it->second;
|
|
}
|
|
return nullptr;
|
|
}
|
|
auto localDestination = new ClientDestination (keys, isPublic, params);
|
|
std::unique_lock<std::mutex> l(m_DestinationsMutex);
|
|
m_Destinations[keys.GetPublic ().GetIdentHash ()] = localDestination;
|
|
localDestination->Start ();
|
|
return localDestination;
|
|
}
|
|
|
|
ClientDestination * ClientContext::FindLocalDestination (const i2p::data::IdentHash& destination) const
|
|
{
|
|
auto it = m_Destinations.find (destination);
|
|
if (it != m_Destinations.end ())
|
|
return it->second;
|
|
return nullptr;
|
|
}
|
|
|
|
void ClientContext::ReadTunnels ()
|
|
{
|
|
std::ifstream ifs (i2p::util::filesystem::GetFullPath (TUNNELS_CONFIG_FILENAME));
|
|
if (ifs.good ())
|
|
{
|
|
boost::program_options::options_description params ("I2P tunnels parameters");
|
|
params.add_options ()
|
|
// client
|
|
(I2P_CLIENT_TUNNEL_NAME, boost::program_options::value<std::vector<std::string> >(), "tunnel name")
|
|
(I2P_CLIENT_TUNNEL_PORT, boost::program_options::value<std::vector<int> >(), "Local port")
|
|
(I2P_CLIENT_TUNNEL_DESTINATION, boost::program_options::value<std::vector<std::string> >(), "destination")
|
|
(I2P_CLIENT_TUNNEL_KEYS, boost::program_options::value<std::vector<std::string> >(), "keys")
|
|
// server
|
|
(I2P_SERVER_TUNNEL_NAME, boost::program_options::value<std::vector<std::string> >(), "tunnel name")
|
|
(I2P_SERVER_TUNNEL_HOST, boost::program_options::value<std::vector<std::string> >(), "host")
|
|
(I2P_SERVER_TUNNEL_PORT, boost::program_options::value<std::vector<int> >(), "port")
|
|
(I2P_SERVER_TUNNEL_KEYS, boost::program_options::value<std::vector<std::string> >(), "keys")
|
|
;
|
|
|
|
|
|
boost::program_options::variables_map vm;
|
|
try
|
|
{
|
|
boost::program_options::store (boost::program_options::parse_config_file (ifs, params), vm);
|
|
boost::program_options::notify (vm);
|
|
}
|
|
catch (boost::program_options::error& ex)
|
|
{
|
|
LogPrint (eLogError, "Can't parse ", TUNNELS_CONFIG_FILENAME,": ", ex.what ());
|
|
return;
|
|
}
|
|
|
|
if (vm.count (I2P_CLIENT_TUNNEL_NAME) > 0)
|
|
{
|
|
auto names = vm[I2P_CLIENT_TUNNEL_NAME].as<std::vector<std::string> >();
|
|
int numClientTunnels = names.size ();
|
|
auto ports = vm[I2P_CLIENT_TUNNEL_PORT].as<std::vector<int> >();
|
|
auto destinations = vm[I2P_CLIENT_TUNNEL_DESTINATION].as<std::vector<std::string> >();
|
|
auto keys = vm[I2P_CLIENT_TUNNEL_KEYS].as<std::vector<std::string> >();
|
|
|
|
for (int i = 0; i < numClientTunnels; i++)
|
|
{
|
|
ClientDestination * localDestination = nullptr;
|
|
if (keys[i].length () > 0)
|
|
localDestination = LoadLocalDestination (keys[i], false);
|
|
auto clientTunnel = new I2PClientTunnel (destinations[i], ports[i], localDestination);
|
|
if (m_ClientTunnels.insert (std::make_pair (ports[i], std::unique_ptr<I2PClientTunnel>(clientTunnel))).second)
|
|
clientTunnel->Start ();
|
|
else
|
|
LogPrint (eLogError, "I2P client tunnel with port ", ports[i], " already exists");
|
|
}
|
|
LogPrint (eLogInfo, numClientTunnels, " I2P client tunnels created");
|
|
}
|
|
|
|
if (vm.count (I2P_SERVER_TUNNEL_NAME) > 0)
|
|
{
|
|
auto names = vm[I2P_SERVER_TUNNEL_NAME].as<std::vector<std::string> >();
|
|
int numServerTunnels = names.size ();
|
|
auto hosts = vm[I2P_SERVER_TUNNEL_HOST].as<std::vector<std::string> >();
|
|
auto ports = vm[I2P_SERVER_TUNNEL_PORT].as<std::vector<int> >();
|
|
auto keys = vm[I2P_SERVER_TUNNEL_KEYS].as<std::vector<std::string> >();
|
|
for (int i = 0; i < numServerTunnels; i++)
|
|
{
|
|
auto localDestination = LoadLocalDestination (keys[i], true);
|
|
auto serverTunnel = new I2PServerTunnel (hosts[i], ports[i], localDestination);
|
|
if (m_ServerTunnels.insert (std::make_pair (localDestination->GetIdentHash (), std::unique_ptr<I2PServerTunnel>(serverTunnel))).second)
|
|
serverTunnel->Start ();
|
|
else
|
|
LogPrint (eLogError, "I2P server tunnel for destination ", m_AddressBook.ToAddress(localDestination->GetIdentHash ()), " already exists");
|
|
}
|
|
LogPrint (eLogInfo, numServerTunnels, " I2P server tunnels created");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|