Skip to content

Commit

Permalink
Make the HTTP Proxy use TCPIPAcceptor
Browse files Browse the repository at this point in the history
  • Loading branch information
klondi committed Jan 8, 2015
1 parent 114022d commit 8a6bea6
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 52 deletions.
36 changes: 2 additions & 34 deletions HTTPProxy.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -231,41 +231,9 @@ namespace proxy
}
}

void HTTPProxyServer::Start ()
std::shared_ptr<i2p::client::I2PServiceHandler> HTTPProxyServer::CreateHandler(boost::asio::ip::tcp::socket * socket)
{
m_Acceptor.listen ();
Accept ();
}

void HTTPProxyServer::Stop ()
{
m_Acceptor.close();
m_Timer.cancel ();
ClearHandlers();
}

void HTTPProxyServer::Accept ()
{
auto newSocket = new boost::asio::ip::tcp::socket (GetService ());
m_Acceptor.async_accept (*newSocket, std::bind (&HTTPProxyServer::HandleAccept, this,
std::placeholders::_1, newSocket));
}

void HTTPProxyServer::HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket)
{
if (!ecode)
{
LogPrint(eLogDebug,"--- HTTP Proxy accepted");
auto handler = std::make_shared<HTTPProxyHandler> (this, socket);
AddHandler(handler);
handler->Handle();
Accept();
}
else
{
LogPrint (eLogError,"--- HTTP Proxy Closing socket on accept because: ", ecode.message ());
delete socket;
}
return std::make_shared<HTTPProxyHandler> (this, socket);
}

}
Expand Down
25 changes: 7 additions & 18 deletions HTTPProxy.h
Original file line number Diff line number Diff line change
Expand Up @@ -12,27 +12,16 @@ namespace i2p
namespace proxy
{
class HTTPProxyHandler;
class HTTPProxyServer: public i2p::client::I2PService
class HTTPProxyServer: public i2p::client::TCPIPAcceptor
{
private:
std::set<std::shared_ptr<HTTPProxyHandler> > 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 "HTTP Proxy"; }

public:
HTTPProxyServer(int port) : I2PService(i2p::data::SIGNING_KEY_TYPE_DSA_SHA1),
m_Acceptor (GetService (), boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), port)),
m_Timer (GetService ()) {};
~HTTPProxyServer() { Stop(); }

void Start ();
void Stop ();
HTTPProxyServer(int port) : TCPIPAcceptor(port, i2p::data::SIGNING_KEY_TYPE_DSA_SHA1) {}
~HTTPProxyServer() {}
};

typedef HTTPProxyServer HTTPProxy;
Expand Down

0 comments on commit 8a6bea6

Please sign in to comment.