| Index: webrtc/test/channel_transport/udp_transport_impl.h
|
| diff --git a/webrtc/test/channel_transport/udp_transport_impl.h b/webrtc/test/channel_transport/udp_transport_impl.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f80ee02d710815d8a34779fdefea0e6d80725c6f
|
| --- /dev/null
|
| +++ b/webrtc/test/channel_transport/udp_transport_impl.h
|
| @@ -0,0 +1,259 @@
|
| +/*
|
| + * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#ifndef WEBRTC_TEST_CHANNEL_TRANSPORT_UDP_TRANSPORT_IMPL_H_
|
| +#define WEBRTC_TEST_CHANNEL_TRANSPORT_UDP_TRANSPORT_IMPL_H_
|
| +
|
| +#include "webrtc/test/channel_transport/udp_socket_wrapper.h"
|
| +#include "webrtc/test/channel_transport/udp_transport.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +class CriticalSectionWrapper;
|
| +class RWLockWrapper;
|
| +
|
| +namespace test {
|
| +
|
| +class UdpSocketManager;
|
| +
|
| +class UdpTransportImpl : public UdpTransport
|
| +{
|
| +public:
|
| + // A factory that returns a wrapped UDP socket or equivalent.
|
| + class SocketFactoryInterface {
|
| + public:
|
| + virtual ~SocketFactoryInterface() {}
|
| + virtual UdpSocketWrapper* CreateSocket(const int32_t id,
|
| + UdpSocketManager* mgr,
|
| + CallbackObj obj,
|
| + IncomingSocketCallback cb,
|
| + bool ipV6Enable,
|
| + bool disableGQOS) = 0;
|
| + };
|
| +
|
| + // Constructor, only called by UdpTransport::Create and tests.
|
| + // The constructor takes ownership of the "maker".
|
| + // The constructor does not take ownership of socket_manager.
|
| + UdpTransportImpl(const int32_t id,
|
| + SocketFactoryInterface* maker,
|
| + UdpSocketManager* socket_manager);
|
| + virtual ~UdpTransportImpl();
|
| +
|
| + // UdpTransport functions
|
| + int32_t InitializeSendSockets(const char* ipAddr,
|
| + const uint16_t rtpPort,
|
| + const uint16_t rtcpPort = 0) override;
|
| + int32_t InitializeReceiveSockets(UdpTransportData* const packetCallback,
|
| + const uint16_t rtpPort,
|
| + const char* ipAddr = NULL,
|
| + const char* multicastIpAddr = NULL,
|
| + const uint16_t rtcpPort = 0) override;
|
| + int32_t InitializeSourcePorts(const uint16_t rtpPort,
|
| + const uint16_t rtcpPort = 0) override;
|
| + int32_t SourcePorts(uint16_t& rtpPort, uint16_t& rtcpPort) const override;
|
| + int32_t ReceiveSocketInformation(
|
| + char ipAddr[kIpAddressVersion6Length],
|
| + uint16_t& rtpPort,
|
| + uint16_t& rtcpPort,
|
| + char multicastIpAddr[kIpAddressVersion6Length]) const override;
|
| + int32_t SendSocketInformation(char ipAddr[kIpAddressVersion6Length],
|
| + uint16_t& rtpPort,
|
| + uint16_t& rtcpPort) const override;
|
| + int32_t RemoteSocketInformation(char ipAddr[kIpAddressVersion6Length],
|
| + uint16_t& rtpPort,
|
| + uint16_t& rtcpPort) const override;
|
| + int32_t SetQoS(const bool QoS,
|
| + const int32_t serviceType,
|
| + const uint32_t maxBitrate = 0,
|
| + const int32_t overrideDSCP = 0,
|
| + const bool audio = false) override;
|
| + int32_t QoS(bool& QoS,
|
| + int32_t& serviceType,
|
| + int32_t& overrideDSCP) const override;
|
| + int32_t SetToS(const int32_t DSCP,
|
| + const bool useSetSockOpt = false) override;
|
| + int32_t ToS(int32_t& DSCP, bool& useSetSockOpt) const override;
|
| + int32_t SetPCP(const int32_t PCP) override;
|
| + int32_t PCP(int32_t& PCP) const override;
|
| + int32_t EnableIpV6() override;
|
| + bool IpV6Enabled() const override;
|
| + int32_t SetFilterIP(
|
| + const char filterIPAddress[kIpAddressVersion6Length]) override;
|
| + int32_t FilterIP(
|
| + char filterIPAddress[kIpAddressVersion6Length]) const override;
|
| + int32_t SetFilterPorts(const uint16_t rtpFilterPort,
|
| + const uint16_t rtcpFilterPort) override;
|
| + int32_t FilterPorts(uint16_t& rtpFilterPort,
|
| + uint16_t& rtcpFilterPort) const override;
|
| + int32_t StartReceiving(const uint32_t numberOfSocketBuffers) override;
|
| + int32_t StopReceiving() override;
|
| + bool Receiving() const override;
|
| + bool SendSocketsInitialized() const override;
|
| + bool SourcePortsInitialized() const override;
|
| + bool ReceiveSocketsInitialized() const override;
|
| + int32_t SendRaw(const int8_t* data,
|
| + size_t length,
|
| + int32_t isRTCP,
|
| + uint16_t portnr = 0,
|
| + const char* ip = NULL) override;
|
| + int32_t SendRTPPacketTo(const int8_t* data,
|
| + size_t length,
|
| + const SocketAddress& to) override;
|
| + int32_t SendRTCPPacketTo(const int8_t* data,
|
| + size_t length,
|
| + const SocketAddress& to) override;
|
| + int32_t SendRTPPacketTo(const int8_t* data,
|
| + size_t length,
|
| + uint16_t rtpPort) override;
|
| + int32_t SendRTCPPacketTo(const int8_t* data,
|
| + size_t length,
|
| + uint16_t rtcpPort) override;
|
| + // Transport functions
|
| + bool SendRtp(const uint8_t* data,
|
| + size_t length,
|
| + const PacketOptions& packet_options) override;
|
| + bool SendRtcp(const uint8_t* data, size_t length) override;
|
| +
|
| + // UdpTransport functions continue.
|
| + int32_t SetSendIP(const char* ipaddr) override;
|
| + int32_t SetSendPorts(const uint16_t rtpPort,
|
| + const uint16_t rtcpPort = 0) override;
|
| +
|
| + ErrorCode LastError() const override;
|
| +
|
| + int32_t IPAddressCached(const SocketAddress& address,
|
| + char* ip,
|
| + uint32_t& ipSize,
|
| + uint16_t& sourcePort) override;
|
| +
|
| + int32_t Id() const {return _id;}
|
| +protected:
|
| + // IncomingSocketCallback signature functions for receiving callbacks from
|
| + // UdpSocketWrapper.
|
| + static void IncomingRTPCallback(CallbackObj obj,
|
| + const int8_t* rtpPacket,
|
| + size_t rtpPacketLength,
|
| + const SocketAddress* from);
|
| + static void IncomingRTCPCallback(CallbackObj obj,
|
| + const int8_t* rtcpPacket,
|
| + size_t rtcpPacketLength,
|
| + const SocketAddress* from);
|
| +
|
| + void CloseSendSockets();
|
| + void CloseReceiveSockets();
|
| +
|
| + // Update _remoteRTPAddr according to _destPort and _destIP
|
| + void BuildRemoteRTPAddr();
|
| + // Update _remoteRTCPAddr according to _destPortRTCP and _destIP
|
| + void BuildRemoteRTCPAddr();
|
| +
|
| + void BuildSockaddrIn(uint16_t portnr, const char* ip,
|
| + SocketAddress& remoteAddr) const;
|
| +
|
| + ErrorCode BindLocalRTPSocket();
|
| + ErrorCode BindLocalRTCPSocket();
|
| +
|
| + ErrorCode BindRTPSendSocket();
|
| + ErrorCode BindRTCPSendSocket();
|
| +
|
| + void IncomingRTPFunction(const int8_t* rtpPacket,
|
| + size_t rtpPacketLength,
|
| + const SocketAddress* from);
|
| + void IncomingRTCPFunction(const int8_t* rtcpPacket,
|
| + size_t rtcpPacketLength,
|
| + const SocketAddress* from);
|
| +
|
| + bool FilterIPAddress(const SocketAddress* fromAddress);
|
| +
|
| + bool SetSockOptUsed();
|
| +
|
| + int32_t EnableQoS(int32_t serviceType, bool audio,
|
| + uint32_t maxBitrate, int32_t overrideDSCP);
|
| +
|
| + int32_t DisableQoS();
|
| +
|
| +private:
|
| + void GetCachedAddress(char* ip, uint32_t& ipSize,
|
| + uint16_t& sourcePort);
|
| +
|
| + int32_t _id;
|
| + SocketFactoryInterface* _socket_creator;
|
| + // Protects the sockets from being re-configured while receiving packets.
|
| + CriticalSectionWrapper* _crit;
|
| + CriticalSectionWrapper* _critFilter;
|
| + // _packetCallback's critical section.
|
| + CriticalSectionWrapper* _critPacketCallback;
|
| + UdpSocketManager* _mgr;
|
| + ErrorCode _lastError;
|
| +
|
| + // Remote RTP and RTCP ports.
|
| + uint16_t _destPort;
|
| + uint16_t _destPortRTCP;
|
| +
|
| + // Local RTP and RTCP ports.
|
| + uint16_t _localPort;
|
| + uint16_t _localPortRTCP;
|
| +
|
| + // Local port number when the local port for receiving and local port number
|
| + // for sending are not the same.
|
| + uint16_t _srcPort;
|
| + uint16_t _srcPortRTCP;
|
| +
|
| + // Remote port from which last received packet was sent.
|
| + uint16_t _fromPort;
|
| + uint16_t _fromPortRTCP;
|
| +
|
| + char _fromIP[kIpAddressVersion6Length];
|
| + char _destIP[kIpAddressVersion6Length];
|
| + char _localIP[kIpAddressVersion6Length];
|
| + char _localMulticastIP[kIpAddressVersion6Length];
|
| +
|
| + UdpSocketWrapper* _ptrRtpSocket;
|
| + UdpSocketWrapper* _ptrRtcpSocket;
|
| +
|
| + // Local port when the local port for receiving and local port for sending
|
| + // are not the same.
|
| + UdpSocketWrapper* _ptrSendRtpSocket;
|
| + UdpSocketWrapper* _ptrSendRtcpSocket;
|
| +
|
| + SocketAddress _remoteRTPAddr;
|
| + SocketAddress _remoteRTCPAddr;
|
| +
|
| + SocketAddress _localRTPAddr;
|
| + SocketAddress _localRTCPAddr;
|
| +
|
| + int32_t _tos;
|
| + bool _receiving;
|
| + bool _useSetSockOpt;
|
| + bool _qos;
|
| + int32_t _pcp;
|
| + bool _ipV6Enabled;
|
| + int32_t _serviceType;
|
| + int32_t _overrideDSCP;
|
| + uint32_t _maxBitrate;
|
| +
|
| + // Cache used by GetCachedAddress(..).
|
| + RWLockWrapper* _cachLock;
|
| + SocketAddress _previousAddress;
|
| + char _previousIP[kIpAddressVersion6Length];
|
| + uint32_t _previousIPSize;
|
| + uint16_t _previousSourcePort;
|
| +
|
| + SocketAddress _filterIPAddress;
|
| + uint16_t _rtpFilterPort;
|
| + uint16_t _rtcpFilterPort;
|
| +
|
| + UdpTransportData* _packetCallback;
|
| +};
|
| +
|
| +} // namespace test
|
| +} // namespace webrtc
|
| +
|
| +#endif // WEBRTC_TEST_CHANNEL_TRANSPORT_UDP_TRANSPORT_IMPL_H_
|
|
|