mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 00:00:29 +03:00
174 lines
4.3 KiB
C++
174 lines
4.3 KiB
C++
#include <boost/bind.hpp>
|
|
#include <boost/lexical_cast.hpp>
|
|
#include "Tunnel.h"
|
|
#include "TransitTunnel.h"
|
|
#include "Transports.h"
|
|
#include "HTTPServer.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace util
|
|
{
|
|
namespace misc_strings
|
|
{
|
|
|
|
const char name_value_separator[] = { ':', ' ' };
|
|
const char crlf[] = { '\r', '\n' };
|
|
|
|
} // namespace misc_strings
|
|
|
|
std::vector<boost::asio::const_buffer> HTTPConnection::reply::to_buffers()
|
|
{
|
|
std::vector<boost::asio::const_buffer> buffers;
|
|
buffers.push_back (boost::asio::buffer ("HTTP/1.0 200 OK\r\n")); // always OK
|
|
for (std::size_t i = 0; i < headers.size(); ++i)
|
|
{
|
|
header& h = headers[i];
|
|
buffers.push_back(boost::asio::buffer(h.name));
|
|
buffers.push_back(boost::asio::buffer(misc_strings::name_value_separator));
|
|
buffers.push_back(boost::asio::buffer(h.value));
|
|
buffers.push_back(boost::asio::buffer(misc_strings::crlf));
|
|
}
|
|
buffers.push_back(boost::asio::buffer(misc_strings::crlf));
|
|
buffers.push_back(boost::asio::buffer(content));
|
|
return buffers;
|
|
}
|
|
|
|
void HTTPConnection::Terminate ()
|
|
{
|
|
m_Socket->close ();
|
|
delete this;
|
|
}
|
|
|
|
void HTTPConnection::Receive ()
|
|
{
|
|
m_Socket->async_read_some (boost::asio::buffer (m_Buffer),
|
|
boost::bind(&HTTPConnection::HandleReceive, this,
|
|
boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
|
|
}
|
|
|
|
void HTTPConnection::HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred)
|
|
{
|
|
if (!ecode)
|
|
{
|
|
HandleRequest ();
|
|
boost::asio::async_write (*m_Socket, m_Reply.to_buffers(),
|
|
boost::bind (&HTTPConnection::HandleWrite, this,
|
|
boost::asio::placeholders::error));
|
|
//Receive ();
|
|
}
|
|
else if (ecode != boost::asio::error::operation_aborted)
|
|
Terminate ();
|
|
}
|
|
|
|
void HTTPConnection::HandleWrite (const boost::system::error_code& ecode)
|
|
{
|
|
Terminate ();
|
|
}
|
|
|
|
void HTTPConnection::HandleRequest ()
|
|
{
|
|
std::stringstream s;
|
|
s << "<html>";
|
|
FillContent (s);
|
|
s << "</html>";
|
|
m_Reply.content = s.str ();
|
|
m_Reply.headers.resize(2);
|
|
m_Reply.headers[0].name = "Content-Length";
|
|
m_Reply.headers[0].value = boost::lexical_cast<std::string>(m_Reply.content.size());
|
|
m_Reply.headers[1].name = "Content-Type";
|
|
m_Reply.headers[1].value = "text/html";
|
|
}
|
|
|
|
void HTTPConnection::FillContent (std::stringstream& s)
|
|
{
|
|
s << "<P>Tunnels</P>";
|
|
for (auto it: i2p::tunnel::tunnels.GetOutboundTunnels ())
|
|
{
|
|
it->GetTunnelConfig ()->Print (s);
|
|
s << " " << (int)it->GetNumSentBytes () << "<BR>";
|
|
}
|
|
|
|
for (auto it: i2p::tunnel::tunnels.GetInboundTunnels ())
|
|
{
|
|
it.second->GetTunnelConfig ()->Print (s);
|
|
s << " " << (int)it.second->GetNumReceivedBytes () << "<BR>";
|
|
}
|
|
|
|
s << "<P>Transit tunnels</P>";
|
|
for (auto it: i2p::tunnel::tunnels.GetTransitTunnels ())
|
|
{
|
|
if (dynamic_cast<i2p::tunnel::TransitTunnelGateway *>(it.second))
|
|
s << it.second->GetTunnelID () << "-->";
|
|
else if (dynamic_cast<i2p::tunnel::TransitTunnelEndpoint *>(it.second))
|
|
s << "-->" << it.second->GetTunnelID ();
|
|
else
|
|
s << "-->" << it.second->GetTunnelID () << "-->";
|
|
s << "<BR>";
|
|
}
|
|
|
|
s << "<P>Transports</P>";
|
|
for (auto it: i2p::transports.GetNTCPSessions ())
|
|
{
|
|
if (it.second->IsEstablished ())
|
|
s << it.second->GetRemoteRouterInfo ().GetIdentHashAbbreviation () << ": "
|
|
<< it.second->GetSocket ().remote_endpoint().address ().to_string () << "<BR>";
|
|
}
|
|
}
|
|
|
|
|
|
HTTPServer::HTTPServer (int port):
|
|
m_Thread (nullptr), m_Work (m_Service),
|
|
m_Acceptor (m_Service, boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), port)),
|
|
m_NewSocket (nullptr)
|
|
{
|
|
}
|
|
|
|
HTTPServer::~HTTPServer ()
|
|
{
|
|
Stop ();
|
|
}
|
|
|
|
void HTTPServer::Start ()
|
|
{
|
|
m_Thread = new std::thread (std::bind (&HTTPServer::Run, this));
|
|
m_Acceptor.listen ();
|
|
Accept ();
|
|
}
|
|
|
|
void HTTPServer::Stop ()
|
|
{
|
|
m_Acceptor.close();
|
|
m_Service.stop ();
|
|
if (m_Thread)
|
|
{
|
|
m_Thread->join ();
|
|
delete m_Thread;
|
|
m_Thread = nullptr;
|
|
}
|
|
}
|
|
|
|
void HTTPServer::Run ()
|
|
{
|
|
m_Service.run ();
|
|
}
|
|
|
|
void HTTPServer::Accept ()
|
|
{
|
|
m_NewSocket = new boost::asio::ip::tcp::socket (m_Service);
|
|
m_Acceptor.async_accept (*m_NewSocket, boost::bind (&HTTPServer::HandleAccept, this,
|
|
boost::asio::placeholders::error));
|
|
}
|
|
|
|
void HTTPServer::HandleAccept(const boost::system::error_code& ecode)
|
|
{
|
|
if (!ecode)
|
|
{
|
|
new HTTPConnection (m_NewSocket);
|
|
Accept ();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|