MultipartParser.h

Go to the documentation of this file.

 
#ifndef MULTIPART_PARSER_HPP
#define MULTIPART_PARSER_HPP
#pragma once
 
#include <string>
#include <unordered_map>
#include "http/HttpRequest.h"
#include "network/Tcp.h"
 
enum State
{
    SEARCHING_FOR_BOUNDARY,
    FOUND_BOUNDARY,
    FOUND_DATA_START,
    COMPLETE
};
 
class MultipartParser
{
public:
    MultipartParser();
 
    bool handleMultipart(HttpRequest& req, HttpResponse& res); // external interface for handling multipart uploads
 
    void setBoundaryFromContentType(const std::string& contentType);
 
private:
    Tcp* tcp; 
    std::string boundary;
    std::string filename;
    std::string leftoverData;
    std::string buffer;
 
 
    static State currentState;
 
    bool extractFilename(const std::string &contentDisposition);
 
    bool processFileData(const std::string &fileData);
 
    bool handleChunk(std::string &chunkData);
 
    int appendFileToSD(const char *filename, const char *data, size_t size);
 
    int file_exists(const std::string& filename);
 
    int findDataStart(std::string &currentData);
 
    bool handleFinalBoundary(std::string &fileData);
 
    void sendHttpResponse(int statusCode, const std::string &message);
};
 
#endif // MULTIPART_PARSER_HPP