RequestParser.hpp

Go to the documentation of this file.

#pragma once
#include "http/HttpStatus.hpp"
#include "http/Request.hpp"
#include "http/internal/ChunkedBodyParser.hpp"
#include <ostream>
#include <string>
#include <sys/types.h>
#include <unistd.h>
 
namespace http {
 
class RequestParser {
public:
    enum State {
        READING_HEADERS, // Waiting for '\r\n\r\n'
        HEADERS_READY,   // **PAUSE STATE**. Headers are parsed. Waiting for routing.
        READING_BODY,    // Consuming the request body.
        REQUEST_READY,   // Successfully parsed one full request.
        ERROR            // A protocol or policy error occurred.
    };

    RequestParser(Request &req, size_t maxHeaderSize);

    void reset();

    State feed(char const *chunk, size_t size);

    State proceedReadingBody();
 
    State state() const;
    HttpStatus errorStatus() const;
    bool hasLeftoverData() const { return !buffer_.empty(); }
 
private:
    State parseHeaders();

    State parseBody();
 
    void handleChunkedBody();
    void handleContentLengthBody();
    bool writeToBodyFile(const std::string &data);

    State setError(HttpStatus status);
 
    void setRequestReady();
 
    State state_;
    HttpStatus errorStatus_;
 
    std::string buffer_;
 
    size_t maxHeaderSize_;  // Hard-coded server limit
    size_t maxContentSize_; // Policy-driven limit, set in proceedReadingBody()
 
    size_t contentLength_;
    size_t bytesWrittenToBody_;
    bool isContentChunked_;
 
    Request &request_;
 
    ChunkedBodyParser chunkParser_;
};
 
std::ostream &operator<<(std::ostream &o, RequestParser::State st);
 
}; // namespace http