protocolfastcgi.h

/*
 * SPDX-FileCopyrightText: (C) 2017-2018 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#ifndef PROTOCOLFASTCGI_H
#define PROTOCOLFASTCGI_H
 
#include "protocol.h"
#include "socket.h"
 
#include <Cutelyst/Context>
 
#include <QObject>
 
namespace Cutelyst {
 
class Server;
class ProtoRequestFastCGI final
    : public ProtocolData
    , public Cutelyst::EngineRequest
{
    Q_GADGET
public:
    ProtoRequestFastCGI(Socket *sock, int bufferSize);
    ~ProtoRequestFastCGI() override;
 
    void setupNewConnection(Socket *sock) override;
 
    bool writeHeaders(quint16 status, const Cutelyst::Headers &headers) override final;
 
    qint64 doWrite(const char *data, qint64 len) override final;
 
    inline qint64 doWrite(const QByteArray &data) { return doWrite(data.constData(), data.size()); }
 
    void processingFinished() override final;
 
    inline void resetData() override final
    {
        ProtocolData::resetData();
 
        // EngineRequest
        // If we deleteLater the context, there might
        // be an event that tries to finalize the request
        // and it will encounter a null context pointer
        delete context;
        context = nullptr;
        body    = nullptr;
 
        startOfRequest = TimePointSteady{};
        status         = InitialState;
 
        stream_id = 0;
        pktsize   = 0;
    }
 
public:
    quint16 stream_id = 0;
    quint16 pktsize   = 0;
};
 
class ProtocolFastCGI final : public Protocol
{
public:
    ProtocolFastCGI(Server *wsgi);
    ~ProtocolFastCGI() override;
 
    Type type() const override;
 
    inline qint64 readBody(Socket *socket, QIODevice *io, qint64 bytesAvailable) const;
    void parse(Socket *sock, QIODevice *io) const override final;
 
    ProtocolData *createData(Socket *sock) const override final;
 
private:
    inline quint16 addHeader(ProtoRequestFastCGI *request,
                             const char *key,
                             quint16 keylen,
                             const char *val,
                             quint16 vallen) const;
    inline int parseHeaders(ProtoRequestFastCGI *request, const char *buf, quint16 len) const;
    inline int processPacket(ProtoRequestFastCGI *request) const;
    inline bool writeBody(ProtoRequestFastCGI *request, char *buf, qint64 len) const;
};
 
} // namespace Cutelyst
 
#endif // PROTOCOLFASTCGI_H