TeChunkedParser.h

Go to the documentation of this file.

/*
 * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */
 
#ifndef SQUID_SRC_HTTP_ONE_TECHUNKEDPARSER_H
#define SQUID_SRC_HTTP_ONE_TECHUNKEDPARSER_H
 
#include "http/one/Parser.h"
 
class MemBuf;
 
namespace Http
{
namespace One
{
 
using ::Parser::InsufficientInput;
 
// TODO: Move this class into http/one/ChunkExtensionValueParser.*
class ChunkExtensionValueParser
{
public:
    typedef ::Parser::Tokenizer Tokenizer;
 
    static void Ignore(Tokenizer &tok, const SBuf &extName);
 
    virtual void parse(Tokenizer &tok, const SBuf &extName) = 0;
};
 
class TeChunkedParser : public Http1::Parser
{
public:
    TeChunkedParser();
    ~TeChunkedParser() override { theOut=nullptr; /* we do not own this object */ }
 
    void setPayloadBuffer(MemBuf *parsedContent) {theOut = parsedContent;}
 
    void parseExtensionValuesWith(ChunkExtensionValueParser *parser) { customExtensionValueParser = parser; }
 
    bool needsMoreSpace() const;
 
    /* Http1::Parser API */
    void clear() override;
    bool parse(const SBuf &) override;
    Parser::size_type firstLineSize() const override {return 0;} // has no meaning with multiple chunks
 
private:
    bool parseChunkSize(Tokenizer &tok);
    bool parseChunkMetadataSuffix(Tokenizer &);
    void parseChunkExtensions(Tokenizer &);
    void parseOneChunkExtension(Tokenizer &);
    bool parseChunkBody(Tokenizer &tok);
    bool parseChunkEnd(Tokenizer &tok);
 
    MemBuf *theOut;
    uint64_t theChunkSize;
    uint64_t theLeftBodySize;
 
    ChunkExtensionValueParser *customExtensionValueParser;
};
 
} // namespace One
} // namespace Http
 
#endif /* SQUID_SRC_HTTP_ONE_TECHUNKEDPARSER_H */