QXmppStun.h

/*
 * Copyright (C) 2008-2019 The QXmpp developers
 *
 * Author:
 *  Jeremy Lainé
 *
 * Source:
 *  https://github.com/qxmpp-project/qxmpp
 *
 * This file is a part of QXmpp library.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 */
 
#ifndef QXMPPSTUN_H
#define QXMPPSTUN_H
 
#include <QObject>
 
#include "QXmppLogger.h"
#include "QXmppJingleIq.h"
 
class CandidatePair;
class QDataStream;
class QUdpSocket;
class QTimer;
class QXmppIceComponentPrivate;
class QXmppIceConnectionPrivate;
class QXmppIcePrivate;
 
 
class QXMPP_EXPORT QXmppStunMessage
{
public:
    enum MethodType {
        Binding      = 0x1,
        SharedSecret = 0x2,
        Allocate     = 0x3,
        Refresh      = 0x4,
        Send         = 0x6,
        Data         = 0x7,
        CreatePermission = 0x8,
        ChannelBind  = 0x9
    };
 
    enum ClassType {
        Request    = 0x000,
        Indication = 0x010,
        Response   = 0x100,
        Error      = 0x110
    };
 
    QXmppStunMessage();
 
    quint32 cookie() const;
    void setCookie(quint32 cookie);
 
    QByteArray id() const;
    void setId(const QByteArray &id);
 
    quint16 messageClass() const;
    quint16 messageMethod() const;
 
    quint16 type() const;
    void setType(quint16 type);
 
    // attributes
 
    quint32 changeRequest() const;
    void setChangeRequest(quint32 changeRequest);
 
    quint16 channelNumber() const;
    void setChannelNumber(quint16 channelNumber);
 
    QByteArray data() const;
    void setData(const QByteArray &data);
 
    quint32 lifetime() const;
    void setLifetime(quint32 changeRequest);
 
    QByteArray nonce() const;
    void setNonce(const QByteArray &nonce);
 
    quint32 priority() const;
    void setPriority(quint32 priority);
 
    QString realm() const;
    void setRealm(const QString &realm);
 
    QByteArray reservationToken() const;
    void setReservationToken(const QByteArray &reservationToken);
 
    quint8 requestedTransport() const;
    void setRequestedTransport(quint8 requestedTransport);
 
    QString software() const;
    void setSoftware(const QString &software);
 
    QString username() const;
    void setUsername(const QString &username);
 
    QByteArray encode(const QByteArray &key = QByteArray(), bool addFingerprint = true) const;
    bool decode(const QByteArray &buffer, const QByteArray &key = QByteArray(), QStringList *errors = 0);
    QString toString() const;
    static quint16 peekType(const QByteArray &buffer, quint32 &cookie, QByteArray &id);
 
    // attributes
    int errorCode;
    QString errorPhrase;
    QByteArray iceControlling;
    QByteArray iceControlled;
    QHostAddress changedHost;
    quint16 changedPort;
    QHostAddress mappedHost;
    quint16 mappedPort;
    QHostAddress otherHost;
    quint16 otherPort;
    QHostAddress sourceHost;
    quint16 sourcePort;
    QHostAddress xorMappedHost;
    quint16 xorMappedPort;
    QHostAddress xorPeerHost;
    quint16 xorPeerPort;
    QHostAddress xorRelayedHost;
    quint16 xorRelayedPort;
    bool useCandidate;
 
private:
    quint32 m_cookie;
    QByteArray m_id;
    quint16 m_type;
 
    // attributes
    QSet<quint16> m_attributes;
    quint32 m_changeRequest;
    quint16 m_channelNumber;
    QByteArray m_data;
    quint32 m_lifetime;
    QByteArray m_nonce;
    quint32 m_priority;
    QString m_realm;
    quint8 m_requestedTransport;
    QByteArray m_reservationToken;
    QString m_software;
    QString m_username;
};
 
 
class QXMPP_EXPORT QXmppIceComponent : public QXmppLoggable
{
    Q_OBJECT
 
public:
    ~QXmppIceComponent();
 
    int component() const;
    bool isConnected() const;
    QList<QXmppJingleCandidate> localCandidates() const;
 
    static QList<QHostAddress> discoverAddresses();
    static QList<QUdpSocket*> reservePorts(const QList<QHostAddress> &addresses, int count, QObject *parent = 0);
 
public slots:
    void close();
    void connectToHost();
    qint64 sendDatagram(const QByteArray &datagram);
 
private slots:
    void checkCandidates();
    void handleDatagram(const QByteArray &datagram, const QHostAddress &host, quint16 port);
    void turnConnected();
    void transactionFinished();
    void updateGatheringState();
    void writeStun(const QXmppStunMessage &request);
 
signals:
    void connected();
 
    void datagramReceived(const QByteArray &datagram);
 
    void gatheringStateChanged();
 
    void localCandidatesChanged();
 
private:
    QXmppIceComponent(int component, QXmppIcePrivate *config, QObject *parent=0);
 
    QXmppIceComponentPrivate *d;
    friend class QXmppIceComponentPrivate;
    friend class QXmppIceConnection;
};
 
 
class QXMPP_EXPORT QXmppIceConnection : public QXmppLoggable
{
    Q_OBJECT
    Q_ENUMS(GatheringState)
    Q_PROPERTY(QXmppIceConnection::GatheringState gatheringState READ gatheringState NOTIFY gatheringStateChanged)
 
public:
    enum GatheringState
    {
        NewGatheringState,
        BusyGatheringState,
        CompleteGatheringState
    };
 
    QXmppIceConnection(QObject *parent = 0);
    ~QXmppIceConnection();
 
    QXmppIceComponent *component(int component);
    void addComponent(int component);
    void setIceControlling(bool controlling);
 
    QList<QXmppJingleCandidate> localCandidates() const;
    QString localUser() const;
    QString localPassword() const;
 
    void addRemoteCandidate(const QXmppJingleCandidate &candidate);
    void setRemoteUser(const QString &user);
    void setRemotePassword(const QString &password);
 
    void setStunServer(const QHostAddress &host, quint16 port = 3478);
    void setTurnServer(const QHostAddress &host, quint16 port = 3478);
    void setTurnUser(const QString &user);
    void setTurnPassword(const QString &password);
 
    bool bind(const QList<QHostAddress> &addresses);
    bool isConnected() const;
 
    GatheringState gatheringState() const;
 
signals:
    void connected();
 
    void disconnected();
 
    void gatheringStateChanged();
 
    void localCandidatesChanged();
 
public slots:
    void close();
    void connectToHost();
 
private slots:
    void slotConnected();
    void slotGatheringStateChanged();
    void slotTimeout();
 
private:
    QXmppIceConnectionPrivate *d;
};
 
#endif