unixfork.h

/*
 * SPDX-FileCopyrightText: (C) 2016-2017 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#ifndef UNIXFORK_H
#define UNIXFORK_H
 
#include "abstractfork.h"
 
#include <QHash>
#include <QObject>
#include <QVector>
 
struct Worker {
    bool null = true;
    int id;
    int restart = 0;
    int respawn = 0;
};
 
namespace Cutelyst {
class Server;
}
 
class QTimer;
class QSocketNotifier;
class UnixFork final : public AbstractFork
{
    Q_OBJECT
public:
    explicit UnixFork(int process, int threads, bool setupSignals, QObject *parent = nullptr);
    ~UnixFork() override;
 
    virtual bool continueMaster(int *exit = nullptr) override;
 
    virtual int exec(bool lazy, bool master) override;
 
    virtual void restart() override;
 
    int internalExec();
 
    bool createProcess(bool respawn);
    void decreaseWorkerRespawn();
 
    virtual void killChild() override;
    void killChild(qint64 pid);
 
    virtual void terminateChild() override;
    void terminateChild(qint64 pid);
 
    static void stopWSGI(const QString &pidfile);
 
    static bool setUmask(const QByteArray &valueStr);
    static bool setGidUid(const QString &gid, const QString &uid, bool noInitgroups);
    static void chownSocket(const QString &filename, const QString &uidGid);
 
    static int idealProcessCount();
    static int idealThreadCount();
 
    void handleSigHup();
    void handleSigTerm();
    void handleSigInt();
    void handleSigChld();
 
    static void setSched(Cutelyst::Server *wsgi, int workerId, int workerCore);
 
private:
    int setupUnixSignalHandlers();
    void setupSocketPair(bool closeSignalsFD, bool createPair);
    bool createChild(const Worker &worker, bool respawn);
    static void signalHandler(int signal);
    void setupCheckChildTimer();
    void postFork(int workerId);
 
    QHash<qint64, Worker> m_childs;
    QVector<Worker> m_recreateWorker;
    QSocketNotifier *m_signalNotifier = nullptr;
    QTimer *m_checkChildRestart       = nullptr;
    int m_threads;
    int m_processes;
    bool m_child       = false;
    bool m_terminating = false;
};
 
#endif // UNIXFORK_H