ChunkedDecoder Class Reference

#include <ChunkedDecoder.h>

Collaboration diagram for ChunkedDecoder:

Public Member Functions
void	feed (const std::string &data, size_t maxLength=MAX_HTTP_BODY_LENGTH)
bool	feedToFile (const std::string &data, std::function< bool(const char *data, size_t len)> writeFn, size_t maxLength=MAX_HTTP_BODY_LENGTH)
	Feed data to the decoder, writing it to a file using the provided write function.
std::string	getBuffer () const
	Get the current buffer content.
bool	wasTruncated () const
	Set if the content-length is longer than the maximum value allowed.
std::string	getDecoded ()
bool	isComplete () const

Private Member Functions
void	parseChunks (size_t maxLength)

Private Attributes
std::string	buffer
std::string	decoded
bool	complete = false
bool	truncated = false
size_t	totalDecoded = 0

Detailed Description

Definition at line 5 of file ChunkedDecoder.h.

Member Function Documentation

feed()

void ChunkedDecoder::feed	(	const std::string &	data,
		size_t	maxLength = MAX_HTTP_BODY_LENGTH
	)

Definition at line 4 of file ChunkedDecoder.cpp.

                                                                 {
    buffer += data;
    parseChunks(maxLength);
}

References buffer, and parseChunks().

Referenced by HttpParser::receiveChunkedBodyToString().

Here is the call graph for this function:

Here is the caller graph for this function:

feedToFile()

bool ChunkedDecoder::feedToFile	(	const std::string &	data,
		std::function< bool(const char *data, size_t len)>	writeFn,
		size_t	maxLength = MAX_HTTP_BODY_LENGTH
	)

This function will write the data to a file using the provided write function. If the data exceeds maxLength, it will be truncated and truncated flag will be set.

Parameters

data	The data to feed to the decoder.
writeFn	The function to write data to a file.
maxLength	The maximum length of data to process (default is MAX_HTTP_BODY_LENGTH).

Returns

true if all data was written successfully, false if it was truncated.

Note

The write function should return true if the data was written successfully, false otherwise. If the write function returns false, the decoder will stop processing further data. If the data exceeds maxLength, it will be truncated and the truncated flag will be set.

Definition at line 9 of file ChunkedDecoder.cpp.

{
    buffer += data;
    std::istringstream stream(buffer);
    std::string line;
    truncated = false;
 
    while (std::getline(stream, line)) {
        if (line.empty()) continue;
 
        int chunkSize = 0;
        std::istringstream chunkHeader(line);
        chunkHeader >> std::hex >> chunkSize;
 
        if (chunkSize == 0) {
            complete = true;
            buffer.clear();
            return true;
        }
 
        if ((int)stream.rdbuf()->in_avail() < chunkSize + 2) {
            return true; // wait for more input
        }
 
        char* chunkData = new char[chunkSize];
        stream.read(chunkData, chunkSize);
 
        size_t available = maxLength - totalDecoded;
        size_t toWrite = std::min((size_t)chunkSize, available);
 
        if (!writeFn(chunkData, toWrite)) {
            delete[] chunkData;
            return false;
        }
 
        delete[] chunkData;
        totalDecoded += toWrite;
 
        if (chunkSize > (int)available) {
            truncated = true;
            complete = true;
            return true;
        }
 
        std::getline(stream, line); // Consume CRLF
    }
 
    buffer.clear();
    return true;
}

References buffer, complete, totalDecoded, and truncated.

Referenced by HttpParser::receiveChunkedBodyToFile().

Here is the caller graph for this function:

getBuffer()

std::string ChunkedDecoder::getBuffer ( ) const

inline

This returns the raw buffer content that has been fed to the decoder.

Returns

The current buffer content as a string.

Definition at line 34 of file ChunkedDecoder.h.

{ return buffer; }

References buffer.

getDecoded()

std::string ChunkedDecoder::getDecoded

(

)

Definition at line 63 of file ChunkedDecoder.cpp.

                                     {
    return decoded;
}

References decoded.

Referenced by HttpParser::receiveChunkedBodyToString().

Here is the caller graph for this function:

isComplete()

bool ChunkedDecoder::isComplete

(

)

const

Definition at line 67 of file ChunkedDecoder.cpp.

                                      {
    return complete;
}

References complete.

Referenced by HttpParser::receiveChunkedBodyToFile(), and HttpParser::receiveChunkedBodyToString().

Here is the caller graph for this function:

parseChunks()

void ChunkedDecoder::parseChunks ( size_t  maxLength )

private

Definition at line 71 of file ChunkedDecoder.cpp.

                                                 {
    std::istringstream stream(buffer);
    std::string line;
    std::string tempDecoded;
    truncated = false;
 
    while (std::getline(stream, line)) {
        if (line.empty()) continue;
 
        int chunkSize = 0;
        std::istringstream chunkHeader(line);
        chunkHeader >> std::hex >> chunkSize;
 
        if (chunkSize == 0) {
            complete = true;
            break;
        }
 
        char* chunk = new char[chunkSize];
        stream.read(chunk, chunkSize);
 
        size_t remaining = maxLength - tempDecoded.size();
        if (chunkSize > remaining) {
            tempDecoded.append(chunk, remaining);
            truncated = true;
            delete[] chunk;
            break;
        }
 
        tempDecoded.append(chunk, chunkSize);
        delete[] chunk;
 
        std::getline(stream, line); // Consume CRLF
    }
 
    decoded = std::move(tempDecoded);
}

References buffer, complete, decoded, and truncated.

Referenced by feed().

Here is the caller graph for this function:

wasTruncated()

bool ChunkedDecoder::wasTruncated ( ) const

inline

This checks if the decoder has processed all chunks and is ready to return the decoded data.

Returns

true if decoding is complete, false otherwise.

Definition at line 42 of file ChunkedDecoder.h.

{ return truncated; }

References truncated.

Referenced by HttpParser::receiveChunkedBodyToFile(), and HttpParser::receiveChunkedBodyToString().

Here is the caller graph for this function:

Member Data Documentation

buffer

std::string ChunkedDecoder::buffer

private

Definition at line 47 of file ChunkedDecoder.h.

Referenced by feed(), feedToFile(), getBuffer(), and parseChunks().

complete

bool ChunkedDecoder::complete = false

private

Definition at line 49 of file ChunkedDecoder.h.

Referenced by feedToFile(), isComplete(), and parseChunks().

decoded

std::string ChunkedDecoder::decoded

private

Definition at line 48 of file ChunkedDecoder.h.

Referenced by getDecoded(), and parseChunks().

totalDecoded

size_t ChunkedDecoder::totalDecoded = 0

private

Definition at line 51 of file ChunkedDecoder.h.

Referenced by feedToFile().

truncated

bool ChunkedDecoder::truncated = false

private

Definition at line 50 of file ChunkedDecoder.h.

Referenced by feedToFile(), parseChunks(), and wasTruncated().

---

The documentation for this class was generated from the following files:

• ChunkedDecoder.h
• ChunkedDecoder.cpp