TransportStrategy.h

Go to the documentation of this file.

#pragma once
 
#include <QtCore/QObject>
#include <QtCore/QTimer>
#include <QtNetwork/QSslCertificate>
#include <QtNetwork/QSslKey>
 
class TcpTransport;
class UdpTransport;
 
class TransportStrategy : public QObject
{
    Q_OBJECT
 
public:
    enum class Protocol
    {
        UDP,
        TCP,
        AutoFallback
    };
    Q_ENUM(Protocol)
 
    explicit TransportStrategy(QObject* parent = nullptr);
    ~TransportStrategy() override;
 
    [[nodiscard]] Protocol protocol() const { return _protocol; }
    [[nodiscard]] QString host() const { return _host; }
    [[nodiscard]] quint16 port() const { return _port; }
    [[nodiscard]] bool tlsEnabled() const { return _tlsEnabled; }
    [[nodiscard]] bool tlsVerifyPeer() const { return _tlsVerifyPeer; }
 
    void setProtocol(Protocol protocol);
    void setTarget(const QString& host, quint16 port);
 
    void setTlsEnabled(bool enabled);
    void setTlsVerifyPeer(bool verify);
    void setTlsCaCertificates(const QList<QSslCertificate>& certs);
    void setTlsClientCertificate(const QSslCertificate& cert, const QSslKey& key);
 
    quint64 send(const QByteArray& data);
 
    [[nodiscard]] bool isConnected() const;
    [[nodiscard]] bool tcpConnected() const;
 
    void reset();
 
signals:
    void connected();
    void disconnected();
    void errorOccurred(const QString& message);
 
private:
    void _ensureUdp();
    void _ensureTcp();
    void _configureTcp();
    bool _shouldUseTcp() const;
 
    void _onTcpConnected();
    void _onTcpDisconnected();
    void _onTransportError(const QString& message);
 
    Protocol _protocol = Protocol::UDP;
    QString _host;
    quint16 _port = 0;
 
    bool _tlsEnabled = false;
    bool _tlsVerifyPeer = true;
    QList<QSslCertificate> _tlsCaCerts;
    QSslCertificate _tlsClientCert;
    QSslKey _tlsClientKey;
 
    UdpTransport* _udpTransport = nullptr;
    TcpTransport* _tcpTransport = nullptr;
 
    int _tcpReconnectMs = kInitialReconnectMs;
 
    static constexpr int kUdpFailureThreshold = 10;
    static constexpr int kInitialReconnectMs = 5000;
    static constexpr int kMaxReconnectMs = 60000;
};