httpmultipartreader.hpp

/* Copyright (c) 2014, The Nuria Project
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any damages
 * arising from the use of this software.
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *    1. The origin of this software must not be misrepresented; you must not
 *       claim that you wrote the original software. If you use this software
 *       in a product, an acknowledgment in the product documentation would be
 *       appreciated but is not required.
 *    2. Altered source versions must be plainly marked as such, and must not be
 *       misrepresented as being the original software.
 *    3. This notice may not be removed or altered from any source
 *       distribution.
 */

#ifndef NURIA_HTTPMULTIPARTREADER_HPP
#define NURIA_HTTPMULTIPARTREADER_HPP

#include "httppostbodyreader.hpp"

namespace Nuria {

class HttpMultiPartReaderPrivate;
class TemporaryBufferDevice;

class NURIA_NETWORK_EXPORT HttpMultiPartReader : public HttpPostBodyReader {
    Q_OBJECT
public:
    
    HttpMultiPartReader (QIODevice *device, const QByteArray &boundary, QObject *parent = nullptr);
    
    ~HttpMultiPartReader ();
    
    bool isComplete () const override;
    bool hasFailed () const override;
    bool hasField (const QString &field) const override;
    QStringList fieldNames () const override;
    QString fieldMimeType (const QString &field) const override;
    qint64 fieldLength (const QString &field) const override;
    qint64 fieldBytesTransferred (const QString &field) const override;
    bool isFieldComplete (const QString &field) const override;
    QIODevice *fieldStream (const QString &field) override;
    
private:
    
    void setState (int state);
    void processData ();
    void processChunk ();
    bool processChunkStep ();
    int processFirstLine ();
    int processAllHeaders ();
    int processHeaders (int offset);
    bool processHeaderLine (const QByteArray &line);
    bool parseHeaders ();
    qint64 parseContentLength (const QByteArray &data);
    QString parseContentDispositionValue (const QByteArray &data);
    QString parseContentDispositionFormDataValue (const QByteArray &data);
    int processContent ();
    int isBoundingLine (const QByteArray &buffer, bool &last, int offset);
    int boundingLineLength () const;
    void stopListening ();
    int appendToCurrentBuffer (const QByteArray &data, int length);
    int processContentBoundary (int idxOfNewline);
    void initCurrentBuffer (const QString &name, TemporaryBufferDevice *device);
    
    HttpMultiPartReaderPrivate *d_ptr;
    
};

}

#endif // NURIA_HTTPMULTIPARTREADER_HPP