srtsockettransport.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_SRT
#include <promeki/namespace.h>
#include <promeki/packettransport.h>
#include <promeki/socketaddress.h>
#include <promeki/string.h>
#include <promeki/srtsocket.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class SrtServer;
 
class SrtSocketTransport : public PacketTransport {
        public:
                using UPtr = UniquePtr<SrtSocketTransport>;
 
                enum Mode {
                        Caller,    
                        Listener,  
                        Rendezvous 
                };
 
                explicit SrtSocketTransport(Mode mode = Caller);
 
                ~SrtSocketTransport() override;
 
                Mode mode() const { return _mode; }
 
                void setLocalAddress(const SocketAddress &addr) { _localAddress = addr; }
 
                const SocketAddress &localAddress() const { return _localAddress; }
 
                void setPeerAddress(const SocketAddress &addr) { _peerAddress = addr; }
 
                const SocketAddress &peerAddress() const { return _peerAddress; }
 
                void setAcceptTimeoutMs(unsigned int ms) { _acceptTimeoutMs = ms; }
 
                unsigned int acceptTimeoutMs() const { return _acceptTimeoutMs; }
 
                Error setLatency(int ms) {
                        if (ms < 0 || ms > 60000) return Error::Invalid;
                        _latencyMs = ms;
                        return Error::Ok;
                }
                int latency() const { return _latencyMs; }
 
                Error setPassphrase(const String &passphrase) {
                        if (!passphrase.isEmpty() && (passphrase.byteCount() < 10 || passphrase.byteCount() > 79))
                                return Error::Invalid;
                        _passphrase = passphrase;
                        return Error::Ok;
                }
 
                Error setEncryptionKeyLength(int bytes) {
                        if (bytes != 0 && bytes != 16 && bytes != 24 && bytes != 32) return Error::Invalid;
                        _pbKeyLen = bytes;
                        return Error::Ok;
                }
 
                Error setStreamId(const String &id) {
                        if (id.byteCount() > 512) return Error::Invalid;
                        _streamId = id;
                        return Error::Ok;
                }
                const String &streamId() const { return _streamId; }
 
                Error setPayloadSize(int bytes) {
                        if (bytes < 0 || bytes > 1456) return Error::Invalid;
                        _payloadSize = bytes;
                        return Error::Ok;
                }
 
                void setMaxBandwidth(int64_t bytesPerSec) { _maxBw = bytesPerSec; }
 
                SrtSocket *socket() const { return _socket.get(); }
 
                SrtSocket::Stats stats(bool resetWindowed = false) const {
                        if (!_socket) return SrtSocket::Stats{};
                        return _socket->stats(resetWindowed);
                }
 
                Error open() override;
 
                void close() override;
 
                bool isOpen() const override;
 
                ssize_t sendPacket(const void *data, size_t size, const SocketAddress &dest) override;
 
                int sendPackets(const DatagramList &datagrams) override;
 
                ssize_t receivePacket(void *data, size_t maxSize, SocketAddress *sender = nullptr) override;
 
        private:
                Mode             _mode = Caller;
                SrtSocket::UPtr  _socket;
                UniquePtr<SrtServer> _listener;
                SocketAddress    _localAddress;
                SocketAddress    _peerAddress;
                String           _passphrase;
                String           _streamId;
                int              _latencyMs = 120;
                int              _payloadSize = 0;
                int              _pbKeyLen = 0;
                int64_t          _maxBw = 0;
                unsigned int     _acceptTimeoutMs = 0;
 
                Error applySocketOptions(SrtSocket &sock);
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_SRT