#pragma once
#if USE(LIBWEBRTC)
#include <WebCore/LibWebRTCProvider.h>
#include <webrtc/rtc_base/asyncpacketsocket.h>
#include <wtf/Deque.h>
#include <wtf/Forward.h>
namespace IPC {
class Connection;
class DataReference;
class Decoder;
}
namespace WebCore {
class SharedBuffer;
}
namespace WebKit {
class LibWebRTCSocketFactory;
class LibWebRTCSocket final : public rtc::AsyncPacketSocket {
public:
enum class Type { UDP, ServerTCP, ClientTCP, ServerConnectionTCP };
LibWebRTCSocket(LibWebRTCSocketFactory&, uint64_t identifier, Type, const rtc::SocketAddress& localAddress, const rtc::SocketAddress& remoteAddress);
~LibWebRTCSocket();
uint64_t identifier() const { return m_identifier; }
const rtc::SocketAddress& localAddress() const { return m_localAddress; }
const rtc::SocketAddress& remoteAddress() const { return m_remoteAddress; }
void setError(int error) { m_error = error; }
void setState(State state) { m_state = state; }
private:
bool willSend(size_t);
friend class WebRTCSocket;
void signalReadPacket(const WebCore::SharedBuffer&, rtc::SocketAddress&&, int64_t);
void signalSentPacket(int, int64_t);
void signalAddressReady(const rtc::SocketAddress&);
void signalConnect();
void signalClose(int);
void signalNewConnection(rtc::AsyncPacketSocket*);
int GetError() const final { return m_error; }
void SetError(int error) final { setError(error); }
rtc::SocketAddress GetLocalAddress() const final;
rtc::SocketAddress GetRemoteAddress() const final;
int Send(const void *pv, size_t cb, const rtc::PacketOptions& options) final { return SendTo(pv, cb, m_remoteAddress, options); }
int SendTo(const void *, size_t, const rtc::SocketAddress&, const rtc::PacketOptions&) final;
int Close() final;
State GetState() const final { return m_state; }
int GetOption(rtc::Socket::Option, int*) final;
int SetOption(rtc::Socket::Option, int) final;
LibWebRTCSocketFactory& m_factory;
uint64_t m_identifier { 0 };
Type m_type;
rtc::SocketAddress m_localAddress;
rtc::SocketAddress m_remoteAddress;
int m_error { 0 };
State m_state { STATE_BINDING };
static const unsigned MAX_SOCKET_OPTION { rtc::Socket::OPT_RTP_SENDTIME_EXTN_ID + 1 };
std::optional<int> m_options[MAX_SOCKET_OPTION];
Deque<size_t> m_beingSentPacketSizes;
size_t m_availableSendingBytes { 65536 };
bool m_shouldSignalReadyToSend { false };
};
}
#endif // USE(LIBWEBRTC)