i2pd/HTTPServer.h

140 lines
4.0 KiB
C
Raw Normal View History

2013-12-10 13:03:22 +00:00
#ifndef HTTP_SERVER_H__
#define HTTP_SERVER_H__
#include <sstream>
#include <thread>
2014-11-23 16:33:58 +00:00
#include <memory>
2013-12-10 13:03:22 +00:00
#include <boost/asio.hpp>
#include <boost/array.hpp>
#include "LeaseSet.h"
2014-03-27 19:42:23 +00:00
#include "Streaming.h"
2013-12-10 13:03:22 +00:00
namespace i2p
{
namespace util
{
2014-09-16 14:28:45 +00:00
const size_t HTTP_CONNECTION_BUFFER_SIZE = 8192;
const int HTTP_DESTINATION_REQUEST_TIMEOUT = 10; // in seconds
2015-04-04 19:44:29 +00:00
class HTTPConnection: public std::enable_shared_from_this<HTTPConnection>
2013-12-10 13:03:22 +00:00
{
protected:
2014-07-16 16:41:40 +00:00
struct header
{
std::string name;
std::string value;
};
2014-07-16 16:41:40 +00:00
struct request
{
std::string method;
std::string uri;
std::string host;
2014-10-23 01:36:11 +00:00
int port;
int http_version_major;
int http_version_minor;
std::vector<header> headers;
};
struct reply
{
std::vector<header> headers;
2015-11-22 15:58:57 +00:00
std::string status_string, content;
2014-04-17 21:53:48 +00:00
std::vector<boost::asio::const_buffer> to_buffers (int status);
};
2014-07-16 16:41:40 +00:00
2013-12-10 13:03:22 +00:00
public:
2015-11-22 22:01:37 +00:00
HTTPConnection (std::shared_ptr<boost::asio::ip::tcp::socket> socket):
m_Socket (socket), m_Timer (socket->get_io_service ()),
2015-04-04 19:44:29 +00:00
m_Stream (nullptr), m_BufferLen (0) {};
void Receive ();
2013-12-10 13:03:22 +00:00
private:
void Terminate ();
2014-07-16 16:41:40 +00:00
void HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-03-27 19:42:23 +00:00
void AsyncStreamReceive ();
2014-07-16 16:41:40 +00:00
void HandleStreamReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-03-27 19:42:23 +00:00
void HandleWriteReply(const boost::system::error_code& ecode);
void HandleWrite (const boost::system::error_code& ecode);
2014-04-17 21:53:48 +00:00
void SendReply (const std::string& content, int status = 200);
2013-12-10 13:03:22 +00:00
2014-09-28 20:12:25 +00:00
void HandleRequest (const std::string& address);
void HandleCommand (const std::string& command, std::stringstream& s);
2016-03-11 10:30:50 +00:00
void ShowJumpServices (const std::string& address, std::stringstream& s);
2014-09-28 20:12:25 +00:00
void ShowTransports (std::stringstream& s);
void ShowTunnels (std::stringstream& s);
void ShowTransitTunnels (std::stringstream& s);
2014-09-28 23:15:04 +00:00
void ShowLocalDestinations (std::stringstream& s);
2014-09-30 02:18:32 +00:00
void ShowLocalDestination (const std::string& b32, std::stringstream& s);
void ShowSAMSessions (std::stringstream& s);
void ShowSAMSession (const std::string& id, std::stringstream& s);
2016-01-14 01:21:53 +00:00
void ShowI2PTunnels (std::stringstream& s);
2014-09-30 17:34:29 +00:00
void StartAcceptingTunnels (std::stringstream& s);
void StopAcceptingTunnels (std::stringstream& s);
2015-11-03 14:15:49 +00:00
void RunPeerTest (std::stringstream& s);
2013-12-10 13:03:22 +00:00
void FillContent (std::stringstream& s);
2014-01-15 00:56:34 +00:00
std::string ExtractAddress ();
2014-09-30 02:18:32 +00:00
void ExtractParams (const std::string& str, std::map<std::string, std::string>& params);
2014-08-04 20:30:37 +00:00
protected:
2014-07-16 16:41:40 +00:00
2015-11-22 22:01:37 +00:00
std::shared_ptr<boost::asio::ip::tcp::socket> m_Socket;
boost::asio::deadline_timer m_Timer;
2014-11-23 16:33:58 +00:00
std::shared_ptr<i2p::stream::Stream> m_Stream;
2014-09-16 14:28:45 +00:00
char m_Buffer[HTTP_CONNECTION_BUFFER_SIZE + 1], m_StreamBuffer[HTTP_CONNECTION_BUFFER_SIZE + 1];
size_t m_BufferLen;
2013-12-10 13:03:22 +00:00
request m_Request;
reply m_Reply;
protected:
virtual void RunRequest ();
void HandleDestinationRequest(const std::string& address, const std::string& uri);
2014-10-23 01:36:11 +00:00
void SendToAddress (const std::string& address, int port, const char * buf, size_t len);
void HandleDestinationRequestTimeout (const boost::system::error_code& ecode,
i2p::data::IdentHash destination, int port, const char * buf, size_t len);
2015-01-27 16:27:58 +00:00
void SendToDestination (std::shared_ptr<const i2p::data::LeaseSet> remote, int port, const char * buf, size_t len);
2014-08-11 16:10:41 +00:00
public:
static const std::string itoopieImage;
static const std::string itoopieFavicon;
2014-07-16 16:41:40 +00:00
};
2013-12-10 13:03:22 +00:00
class HTTPServer
{
public:
2015-11-30 14:44:32 +00:00
HTTPServer (const std::string& address, int port);
virtual ~HTTPServer ();
2013-12-10 13:03:22 +00:00
void Start ();
void Stop ();
private:
2014-07-16 16:41:40 +00:00
void Run ();
2013-12-10 13:03:22 +00:00
void Accept ();
2015-11-22 22:01:37 +00:00
void HandleAccept(const boost::system::error_code& ecode,
std::shared_ptr<boost::asio::ip::tcp::socket> newSocket);
2014-08-04 20:30:37 +00:00
2013-12-10 13:03:22 +00:00
private:
2015-11-22 22:01:37 +00:00
std::unique_ptr<std::thread> m_Thread;
2013-12-10 13:03:22 +00:00
boost::asio::io_service m_Service;
boost::asio::io_service::work m_Work;
boost::asio::ip::tcp::acceptor m_Acceptor;
protected:
2015-11-22 22:01:37 +00:00
virtual void CreateConnection(std::shared_ptr<boost::asio::ip::tcp::socket> newSocket);
2014-07-16 16:41:40 +00:00
};
2013-12-10 13:03:22 +00:00
}
}
#endif