amnezia-client/client/ui/models/protocols/awgConfigModel.h

#ifndef AWGCONFIGMODEL_H
#define AWGCONFIGMODEL_H

#include <QAbstractListModel>
#include <QJsonObject>

#include "containers/containers_defs.h"

namespace AwgConstant
{
    const int messageInitiationSize = 148;
    const int messageResponseSize = 92;
    const int messageCookieReplySize = 64;
    const int messageTransportSize = 32;
}

struct AwgConfig
{
    AwgConfig(const QJsonObject &serverProtocolConfig);

    QString subnetAddress;
    QString port;

    QString clientMtu;
    QString clientJunkPacketCount;
    QString clientJunkPacketMinSize;
    QString clientJunkPacketMaxSize;
    QString clientSpecialJunk1;
    QString clientSpecialJunk2;
    QString clientSpecialJunk3;
    QString clientSpecialJunk4;
    QString clientSpecialJunk5;

    QString serverJunkPacketCount;
    QString serverJunkPacketMinSize;
    QString serverJunkPacketMaxSize;
    QString serverInitPacketJunkSize;
    QString serverResponsePacketJunkSize;
    QString serverCookieReplyPacketJunkSize;
    QString serverTransportPacketJunkSize;
    QString serverInitPacketMagicHeader;
    QString serverResponsePacketMagicHeader;
    QString serverUnderloadPacketMagicHeader;
    QString serverTransportPacketMagicHeader;
    QString serverSpecialJunk1;
    QString serverSpecialJunk2;
    QString serverSpecialJunk3;
    QString serverSpecialJunk4;
    QString serverSpecialJunk5;

    bool hasEqualServerSettings(const AwgConfig &other) const;
    bool hasEqualClientSettings(const AwgConfig &other) const;

private:
    bool m_isProtocolV2;
};

class AwgConfigModel : public QAbstractListModel
{
    Q_OBJECT

public:
    enum Roles {
        SubnetAddressRole = Qt::UserRole + 1,
        PortRole,

        ClientMtuRole,
        ClientJunkPacketCountRole,
        ClientJunkPacketMinSizeRole,
        ClientJunkPacketMaxSizeRole,
        ClientSpecialJunk1Role,
        ClientSpecialJunk2Role,
        ClientSpecialJunk3Role,
        ClientSpecialJunk4Role,
        ClientSpecialJunk5Role,

        ServerJunkPacketCountRole,
        ServerJunkPacketMinSizeRole,
        ServerJunkPacketMaxSizeRole,
        ServerInitPacketJunkSizeRole,
        ServerResponsePacketJunkSizeRole,
        ServerCookieReplyPacketJunkSizeRole,
        ServerTransportPacketJunkSizeRole,

        ServerInitPacketMagicHeaderRole,
        ServerResponsePacketMagicHeaderRole,
        ServerUnderloadPacketMagicHeaderRole,
        ServerTransportPacketMagicHeaderRole,
        ServerSpecialJunk1Role,
        ServerSpecialJunk2Role,
        ServerSpecialJunk3Role,
        ServerSpecialJunk4Role,
        ServerSpecialJunk5Role,

        IsAwg2Role
    };

    explicit AwgConfigModel(QObject *parent = nullptr);

    int rowCount(const QModelIndex &parent = QModelIndex()) const override;

    bool setData(const QModelIndex &index, const QVariant &value, int role) override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

public slots:
    void updateModel(const QJsonObject &config);
    QJsonObject getConfig();

    bool isHeadersEqual(const QString &h1, const QString &h2, const QString &h3, const QString &h4);
    bool isPacketSizeEqual(const int s1, const int s2, const int s3, const int s4);

    bool isServerSettingsEqual();

protected:
    QHash<int, QByteArray> roleNames() const override;

private:
    DockerContainer m_container;
    QJsonObject m_serverProtocolConfig;
    QJsonObject m_clientProtocolConfig;
    QJsonObject m_fullConfig;
};

#endif // AWGCONFIGMODEL_H