NetworkSocket.h

//
//  NetworkSocket.h
//  libraries/networking/src/udt
//
//  Created by David Rowe on 21 Jun 2021.
//  Copyright 2021 Vircadia contributors.
//
 
#ifndef overte_NetworkSocket_h
#define overte_NetworkSocket_h
 
#include <QObject>
#include <QUdpSocket>
 
#include <shared/WebRTC.h>
 
#include "../SockAddr.h"
#include "../NodeType.h"
#include "../SocketType.h"
#if defined(WEBRTC_DATA_CHANNELS)
#include "../webrtc/WebRTCSocket.h"
#endif
 
 
 
class NetworkSocket : public QObject {
    Q_OBJECT
 
public:
 
    NetworkSocket(QObject* parent);
 
 
    void setSocketOption(SocketType socketType, QAbstractSocket::SocketOption option, const QVariant& value);
    
    QVariant socketOption(SocketType socketType, QAbstractSocket::SocketOption option);
 
 
    void bind(SocketType socketType, const QHostAddress& address, quint16 port = 0);
    
    void abort(SocketType socketType);
 
 
    quint16 localPort(SocketType socketType) const;
 
    qintptr socketDescriptor(SocketType socketType) const;
 
 
    qint64 writeDatagram(const QByteArray& datagram, const SockAddr& sockAddr);
 
    qint64 bytesToWrite(SocketType socketType, const SockAddr& address = SockAddr()) const;
 
 
    bool hasPendingDatagrams() const;
    
    qint64 pendingDatagramSize();
    
    qint64 readDatagram(char* data, qint64 maxSize, SockAddr* sockAddr = nullptr);
 
    
    QAbstractSocket::SocketState state(SocketType socketType) const;
 
 
    QAbstractSocket::SocketError error(SocketType socketType) const;
 
    QString errorString(SocketType socketType) const;
 
 
#if defined(WEBRTC_DATA_CHANNELS)
    const WebRTCSocket* getWebRTCSocket();
#endif
 
signals:
 
    void readyRead();
 
    void stateChanged(SocketType socketType, QAbstractSocket::SocketState socketState);
 
    void socketError(SocketType socketType, QAbstractSocket::SocketError socketError);
 
private slots:
 
    void onUDPStateChanged(QAbstractSocket::SocketState socketState);
    void onWebRTCStateChanged(QAbstractSocket::SocketState socketState);
 
    void onUDPSocketError(QAbstractSocket::SocketError socketError);
    void onWebRTCSocketError(QAbstractSocket::SocketError socketError);
 
private:
 
    QObject* _parent;
 
    QUdpSocket _udpSocket;
#if defined(WEBRTC_DATA_CHANNELS)
    WebRTCSocket _webrtcSocket;
#endif
 
#if defined(WEBRTC_DATA_CHANNELS)
    SocketType _pendingDatagramSizeSocketType { SocketType::Unknown };
    SocketType _lastSocketTypeRead { SocketType::Unknown };
#endif
};
 
 
 
#endif // overte_NetworkSocket_h