i2pd/HTTPServer.h

137 lines
3.0 KiB
C
Raw Normal View History

2013-12-10 17:03:22 +04:00
#ifndef HTTP_SERVER_H__
#define HTTP_SERVER_H__
#include <sstream>
#include <thread>
#include <boost/asio.hpp>
#include <boost/array.hpp>
2014-03-27 23:42:23 +04:00
#include "Streaming.h"
2013-12-10 17:03:22 +04:00
namespace i2p
{
namespace util
{
class HTTPConnection
{
protected:
2014-07-16 20:41:40 +04:00
struct header
{
std::string name;
std::string value;
};
2014-07-16 20:41:40 +04:00
struct request
{
std::string method;
std::string uri;
std::string host;
int http_version_major;
int http_version_minor;
std::vector<header> headers;
};
struct reply
{
std::vector<header> headers;
std::string content;
2014-04-18 01:53:48 +04:00
std::vector<boost::asio::const_buffer> to_buffers (int status);
};
2014-07-16 20:41:40 +04:00
2013-12-10 17:03:22 +04:00
public:
2014-03-27 23:42:23 +04:00
HTTPConnection (boost::asio::ip::tcp::socket * socket): m_Socket (socket), m_Stream (nullptr) { Receive (); };
virtual ~HTTPConnection() { delete m_Socket; }
2013-12-10 17:03:22 +04:00
private:
void Terminate ();
void Receive ();
2014-07-16 20:41:40 +04:00
void HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-03-27 23:42:23 +04:00
void AsyncStreamReceive ();
2014-07-16 20:41:40 +04:00
void HandleStreamReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2014-03-27 23:42:23 +04:00
void HandleWriteReply(const boost::system::error_code& ecode);
void HandleWrite (const boost::system::error_code& ecode);
2014-04-18 01:53:48 +04:00
void SendReply (const std::string& content, int status = 200);
2013-12-10 17:03:22 +04:00
void HandleRequest ();
void FillContent (std::stringstream& s);
2014-01-15 04:56:34 +04:00
std::string ExtractAddress ();
2014-07-16 20:41:40 +04:00
2014-08-05 00:30:37 +04:00
// for eepsite
void EepAccept (i2p::stream::StreamingDestination * destination);
void HandleEepAccept (i2p::stream::Stream * stream);
protected:
2014-07-16 20:41:40 +04:00
2013-12-10 17:03:22 +04:00
boost::asio::ip::tcp::socket * m_Socket;
2014-03-27 23:42:23 +04:00
i2p::stream::Stream * m_Stream;
char m_Buffer[8192], m_StreamBuffer[8192];
2013-12-10 17:03:22 +04:00
request m_Request;
reply m_Reply;
protected:
2014-07-16 20:41:40 +04:00
virtual void HandleDestinationRequest(const std::string& address, const std::string& uri);
2014-07-16 20:41:40 +04:00
virtual void HandleDestinationRequest(const std::string& address, const std::string& method, const std::string& data, const std::string& uri);
virtual void RunRequest ();
2014-08-11 20:10:41 +04:00
public:
static const std::string itoopieImage;
2014-07-16 20:41:40 +04:00
};
2013-12-10 17:03:22 +04:00
class HTTPServer
{
public:
HTTPServer (int port);
virtual ~HTTPServer ();
2013-12-10 17:03:22 +04:00
void Start ();
void Stop ();
private:
2014-07-16 20:41:40 +04:00
void Run ();
2013-12-10 17:03:22 +04:00
void Accept ();
2014-07-16 20:41:40 +04:00
void HandleAccept(const boost::system::error_code& ecode);
2014-08-05 00:30:37 +04:00
2013-12-10 17:03:22 +04:00
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;
protected:
virtual void CreateConnection(boost::asio::ip::tcp::socket * m_NewSocket);
2014-07-16 20:41:40 +04:00
};
2014-08-05 00:30:37 +04:00
// TODO: move away
class EepSiteDummyConnection
{
public:
EepSiteDummyConnection (i2p::stream::Stream * stream): m_Stream (stream) {};
void AsyncStreamReceive ();
private:
void HandleStreamReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
private:
i2p::stream::Stream * m_Stream;
char m_StreamBuffer[8192];
};
2013-12-10 17:03:22 +04:00
}
}
#endif