i2pd/daemon/HTTPServer.h

82 lines
2.0 KiB
C
Raw Normal View History

2013-12-10 17:03:22 +04:00
#ifndef HTTP_SERVER_H__
#define HTTP_SERVER_H__
2016-12-15 21:10:12 +03:00
#include <inttypes.h>
#include <string>
#include <memory>
#include <map>
#include <thread>
#include <boost/asio.hpp>
#include "HTTP.h"
namespace i2p
{
namespace http
{
const size_t HTTP_CONNECTION_BUFFER_SIZE = 8192;
const int TOKEN_EXPIRATION_TIMEOUT = 30; // in seconds
2015-04-04 22:44:29 +03:00
class HTTPConnection: public std::enable_shared_from_this<HTTPConnection>
2013-12-10 17:03:22 +04:00
{
public:
2016-04-27 03:00:00 +03:00
HTTPConnection (std::shared_ptr<boost::asio::ip::tcp::socket> socket);
2015-04-04 22:44:29 +03:00
void Receive ();
2013-12-10 17:03:22 +04:00
private:
2014-07-16 20:41:40 +04:00
void HandleReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
void Terminate (const boost::system::error_code& ecode);
void RunRequest ();
2016-04-27 03:00:00 +03:00
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:
2014-07-16 20:41:40 +04:00
2015-11-23 01:01:37 +03:00
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;
2016-04-27 03:00:00 +03:00
bool needAuth;
std::string user;
std::string pass;
2016-12-15 21:10:12 +03:00
static std::map<uint32_t, uint32_t> m_Tokens; // token->timestamp in seconds
2014-07-16 20:41:40 +04:00
};
2013-12-10 17:03:22 +04:00
class HTTPServer
{
public:
2015-11-30 17:44:32 +03:00
HTTPServer (const std::string& address, int port);
~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 ();
2015-11-23 01:01:37 +03:00
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);
2014-08-05 00:30:37 +04:00
2013-12-10 17:03:22 +04:00
private:
bool m_IsRunning;
2015-11-23 01:01:37 +03:00
std::unique_ptr<std::thread> m_Thread;
2013-12-10 17:03:22 +04:00
boost::asio::io_service m_Service;
boost::asio::io_service::work m_Work;
boost::asio::ip::tcp::acceptor m_Acceptor;
2014-07-16 20:41:40 +04:00
};
2016-04-27 03:00:00 +03:00
} // http
} // i2p
2013-12-10 17:03:22 +04:00
#endif /* HTTP_SERVER_H__ */