|
|
|
#ifndef HTTP_SERVER_H__
|
|
|
|
#define HTTP_SERVER_H__
|
|
|
|
|
|
|
|
namespace i2p {
|
|
|
|
namespace http {
|
|
|
|
extern const char *itoopieFavicon;
|
|
|
|
const size_t HTTP_CONNECTION_BUFFER_SIZE = 8192;
|
|
|
|
|
|
|
|
class HTTPConnection: public std::enable_shared_from_this<HTTPConnection>
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
HTTPConnection (std::shared_ptr<boost::asio::ip::tcp::socket> socket);
|
|
|
|
void Receive ();
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
void HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
|
|
|
void Terminate (const boost::system::error_code& ecode);
|
|
|
|
|
|
|
|
void RunRequest ();
|
|
|
|
bool CheckAuth (const HTTPReq & req);
|
|
|
|
void HandleRequest (const HTTPReq & req);
|
|
|
|
void HandlePage (const HTTPReq & req, HTTPRes & res, std::stringstream& data);
|
|
|
|
void HandleCommand (const HTTPReq & req, HTTPRes & res, std::stringstream& data);
|
|
|
|
void SendReply (HTTPRes & res, std::string & content);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
std::shared_ptr<boost::asio::ip::tcp::socket> m_Socket;
|
|
|
|
boost::asio::deadline_timer m_Timer;
|
|
|
|
char m_Buffer[HTTP_CONNECTION_BUFFER_SIZE + 1];
|
|
|
|
size_t m_BufferLen;
|
|
|
|
std::string m_SendBuffer;
|
|
|
|
bool needAuth;
|
|
|
|
std::string user;
|
|
|
|
std::string pass;
|
|
|
|
};
|
|
|
|
|
|
|
|
class HTTPServer
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
HTTPServer (const std::string& address, int port);
|
|
|
|
~HTTPServer ();
|
|
|
|
|
|
|
|
void Start ();
|
|
|
|
void Stop ();
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
void Run ();
|
|
|
|
void Accept ();
|
|
|
|
void HandleAccept(const boost::system::error_code& ecode,
|
|
|
|
std::shared_ptr<boost::asio::ip::tcp::socket> newSocket);
|
|
|
|
void CreateConnection(std::shared_ptr<boost::asio::ip::tcp::socket> newSocket);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
std::unique_ptr<std::thread> m_Thread;
|
|
|
|
boost::asio::io_service m_Service;
|
|
|
|
boost::asio::io_service::work m_Work;
|
|
|
|
boost::asio::ip::tcp::acceptor m_Acceptor;
|
|
|
|
};
|
|
|
|
} // http
|
|
|
|
} // i2p
|
|
|
|
|
|
|
|
#endif /* HTTP_SERVER_H__ */
|