Make SOCKS use TCPIPAcceptor

This commit is contained in:
Francisco Blas (klondike) Izquierdo Riera 2015-01-08 01:45:49 +01:00
parent 8a6bea64bc
commit 7d9c0b76fc
2 changed files with 9 additions and 53 deletions

View File

@ -514,41 +514,9 @@ namespace proxy
}
}
void SOCKSServer::Start ()
std::shared_ptr<i2p::client::I2PServiceHandler> SOCKSServer::CreateHandler(boost::asio::ip::tcp::socket * socket)
{
m_Acceptor.listen ();
Accept ();
}
void SOCKSServer::Stop ()
{
m_Acceptor.close();
m_Timer.cancel ();
ClearHandlers();
}
void SOCKSServer::Accept ()
{
auto newSocket = new boost::asio::ip::tcp::socket (GetService ());
m_Acceptor.async_accept (*newSocket, std::bind (&SOCKSServer::HandleAccept, this,
std::placeholders::_1, newSocket));
}
void SOCKSServer::HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket)
{
if (!ecode)
{
LogPrint(eLogDebug,"--- SOCKS accepted");
auto handle = std::make_shared<SOCKSHandler> (this, socket);
AddHandler(handle);
handle->Handle();
Accept();
}
else
{
LogPrint (eLogError,"--- SOCKS Closing socket on accept because: ", ecode.message ());
delete socket;
}
return std::make_shared<SOCKSHandler> (this, socket);
}
}

26
SOCKS.h
View File

@ -11,28 +11,16 @@ namespace i2p
{
namespace proxy
{
class SOCKSHandler;
class SOCKSServer: public i2p::client::I2PService
class SOCKSServer: public i2p::client::TCPIPAcceptor
{
private:
std::set<std::shared_ptr<SOCKSHandler> > m_Handlers;
boost::asio::ip::tcp::acceptor m_Acceptor;
boost::asio::deadline_timer m_Timer;
std::mutex m_HandlersMutex;
private:
void Accept();
void HandleAccept(const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket);
protected:
// Implements TCPIPAcceptor
std::shared_ptr<i2p::client::I2PServiceHandler> CreateHandler(boost::asio::ip::tcp::socket * socket);
const char* GetName() { return "SOCKS"; }
public:
SOCKSServer(int port) : I2PService(nullptr),
m_Acceptor (GetService (), boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), port)),
m_Timer (GetService ()) {};
~SOCKSServer() { Stop(); }
void Start ();
void Stop ();
SOCKSServer(int port) : TCPIPAcceptor(port) {}
~SOCKSServer() {}
};
typedef SOCKSServer SOCKSProxy;