mirror of https://github.com/PurpleI2P/i2pd.git
orignal
11 years ago
2 changed files with 263 additions and 0 deletions
@ -0,0 +1,173 @@ |
|||||||
|
#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 (); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
@ -0,0 +1,90 @@ |
|||||||
|
#ifndef HTTP_SERVER_H__ |
||||||
|
#define HTTP_SERVER_H__ |
||||||
|
|
||||||
|
#include <sstream> |
||||||
|
#include <thread> |
||||||
|
#include <boost/asio.hpp> |
||||||
|
#include <boost/array.hpp> |
||||||
|
|
||||||
|
namespace i2p |
||||||
|
{ |
||||||
|
namespace util |
||||||
|
{ |
||||||
|
class HTTPConnection |
||||||
|
{ |
||||||
|
struct header |
||||||
|
{ |
||||||
|
std::string name; |
||||||
|
std::string value; |
||||||
|
}; |
||||||
|
|
||||||
|
struct request |
||||||
|
{ |
||||||
|
std::string method; |
||||||
|
std::string uri; |
||||||
|
int http_version_major; |
||||||
|
int http_version_minor; |
||||||
|
std::vector<header> headers; |
||||||
|
}; |
||||||
|
|
||||||
|
struct reply |
||||||
|
{ |
||||||
|
std::vector<header> headers; |
||||||
|
std::string content; |
||||||
|
|
||||||
|
std::vector<boost::asio::const_buffer> to_buffers(); |
||||||
|
}; |
||||||
|
|
||||||
|
public: |
||||||
|
|
||||||
|
HTTPConnection (boost::asio::ip::tcp::socket * socket): m_Socket (socket) { Receive (); }; |
||||||
|
~HTTPConnection () { delete m_Socket; } |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
void Terminate (); |
||||||
|
void Receive (); |
||||||
|
void HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred); |
||||||
|
void HandleWrite(const boost::system::error_code& ecode); |
||||||
|
|
||||||
|
void HandleRequest (); |
||||||
|
void FillContent (std::stringstream& s); |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
boost::asio::ip::tcp::socket * m_Socket; |
||||||
|
boost::array<char, 8192> m_Buffer; |
||||||
|
request m_Request; |
||||||
|
reply m_Reply; |
||||||
|
}; |
||||||
|
|
||||||
|
class HTTPServer |
||||||
|
{ |
||||||
|
public: |
||||||
|
|
||||||
|
HTTPServer (int port); |
||||||
|
~HTTPServer (); |
||||||
|
|
||||||
|
void Start (); |
||||||
|
void Stop (); |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
void Run (); |
||||||
|
void Accept (); |
||||||
|
void HandleAccept(const boost::system::error_code& ecode); |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
std::thread * m_Thread; |
||||||
|
boost::asio::io_service m_Service; |
||||||
|
boost::asio::io_service::work m_Work; |
||||||
|
boost::asio::ip::tcp::acceptor m_Acceptor; |
||||||
|
boost::asio::ip::tcp::socket * m_NewSocket; |
||||||
|
}; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#endif |
||||||
|
|
||||||
|
|
Loading…
Reference in new issue