HTTP Server added

chadf · Dec 10, 2013 · bf8a612 · bf8a612
1 parent 465075d
commit bf8a612
Show file tree

Hide file tree

Showing 2 changed files with 263 additions and 0 deletions.
diff --git a/HTTPServer.cpp b/HTTPServer.cpp
@@ -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 ();
+		}
+	}	
+}
+}
+
diff --git a/HTTPServer.h b/HTTPServer.h
@@ -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
+
+