forked from cat/WebBase
74 lines
2.0 KiB
C++
74 lines
2.0 KiB
C++
#include <asio.hpp>
|
|
#include <asio/buffer.hpp>
|
|
#include <asio/error_code.hpp>
|
|
#include <asio/impl/read.hpp>
|
|
#include <asio/impl/read_until.hpp>
|
|
#include <asio/impl/write.hpp>
|
|
#include <asio/io_context.hpp>
|
|
#include <asio/ip/address.hpp>
|
|
#include <asio/ip/tcp.hpp>
|
|
#include <asio/placeholders.hpp>
|
|
#include <asio/streambuf.hpp>
|
|
|
|
#include <functional>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
#define IP "127.0.0.1"
|
|
#define PORT 8000
|
|
using asocket = asio::ip::tcp::socket;
|
|
using asocket_ptr = std::shared_ptr<asocket>;
|
|
|
|
// We handle HTTP requests with processHTTPHeader, writeData, and processRequest
|
|
class HTTPrequest : public std::enable_shared_from_this<HTTPrequest> {
|
|
public:
|
|
typedef std::shared_ptr<HTTPrequest> HTTPrequest_ptr;
|
|
|
|
asio::ip::tcp::socket &socket();
|
|
static HTTPrequest_ptr create(asio::io_context &context);
|
|
|
|
void start();
|
|
void sendResponse(std::string status, std::string mime, std::string data);
|
|
|
|
// Request itself
|
|
std::string requestType, requestPath, responseText;
|
|
std::unordered_map<std::string, std::string> headers, args;
|
|
std::string bodyContent;
|
|
|
|
private:
|
|
HTTPrequest(asio::io_context &context);
|
|
void processRequest();
|
|
void waitForClientClose();
|
|
|
|
// Breaking Header to in lines
|
|
void processHTTPHeader();
|
|
void inline processArgs();
|
|
void inline processHeaderValues(std::basic_istream<char> &stream,
|
|
std::size_t &octetCount,
|
|
std::size_t headerSize);
|
|
void inline processBody();
|
|
|
|
// Networking
|
|
asio::ip::tcp::socket sock;
|
|
asio::streambuf buffer;
|
|
};
|
|
|
|
// Server manages connections/requests
|
|
class Webserver : public std::enable_shared_from_this<Webserver> {
|
|
public:
|
|
Webserver(asio::io_context &context);
|
|
|
|
// Responses
|
|
static std::unordered_map<
|
|
std::string,
|
|
std::unordered_map<std::string, std::function<void(HTTPrequest *self)>>>
|
|
responseMethods;
|
|
|
|
private:
|
|
void begin();
|
|
static void initResponses();
|
|
|
|
asio::io_context &io;
|
|
asio::ip::tcp::acceptor accept;
|
|
};
|