Nagram/TMessagesProj/jni/webrtc/pc/composite_rtp_transport.h
2020-08-14 19:58:22 +03:00

120 lines
4.7 KiB
C++

/*
* Copyright 2019 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 PC_COMPOSITE_RTP_TRANSPORT_H_
#define PC_COMPOSITE_RTP_TRANSPORT_H_
#include <memory>
#include <set>
#include <string>
#include <vector>
#include "call/rtp_demuxer.h"
#include "call/rtp_packet_sink_interface.h"
#include "pc/rtp_transport_internal.h"
#include "pc/session_description.h"
#include "rtc_base/async_packet_socket.h"
#include "rtc_base/copy_on_write_buffer.h"
namespace webrtc {
// Composite RTP transport capable of receiving from multiple sub-transports.
//
// CompositeRtpTransport is receive-only until the caller explicitly chooses
// which transport will be used to send and calls |SetSendTransport|. This
// choice must be made as part of the SDP negotiation process, based on receipt
// of a provisional answer. |CompositeRtpTransport| does not become writable or
// ready to send until |SetSendTransport| is called.
//
// When a full answer is received, the user should replace the composite
// transport with the single, chosen RTP transport, then delete the composite
// and all non-chosen transports.
class CompositeRtpTransport : public RtpTransportInternal {
public:
// Constructs a composite out of the given |transports|. |transports| must
// not be empty. All |transports| must outlive the composite.
explicit CompositeRtpTransport(std::vector<RtpTransportInternal*> transports);
// Sets which transport will be used for sending packets. Once called,
// |IsReadyToSend|, |IsWritable|, and the associated signals will reflect the
// state of |send_tranpsort|.
void SetSendTransport(RtpTransportInternal* send_transport);
// Removes |transport| from the composite. No-op if |transport| is null or
// not found in the composite. Removing a transport disconnects all signals
// and RTP demux sinks from that transport. The send transport may not be
// removed.
void RemoveTransport(RtpTransportInternal* transport);
// All transports within a composite must have the same name.
const std::string& transport_name() const override;
int SetRtpOption(rtc::Socket::Option opt, int value) override;
int SetRtcpOption(rtc::Socket::Option opt, int value) override;
// All transports within a composite must either enable or disable RTCP mux.
bool rtcp_mux_enabled() const override;
// Enables or disables RTCP mux for all component transports.
void SetRtcpMuxEnabled(bool enabled) override;
// The composite is ready to send if |send_transport_| is set and ready to
// send.
bool IsReadyToSend() const override;
// The composite is writable if |send_transport_| is set and writable.
bool IsWritable(bool rtcp) const override;
// Sends an RTP packet. May only be called after |send_transport_| is set.
bool SendRtpPacket(rtc::CopyOnWriteBuffer* packet,
const rtc::PacketOptions& options,
int flags) override;
// Sends an RTCP packet. May only be called after |send_transport_| is set.
bool SendRtcpPacket(rtc::CopyOnWriteBuffer* packet,
const rtc::PacketOptions& options,
int flags) override;
// Updates the mapping of RTP header extensions for all component transports.
void UpdateRtpHeaderExtensionMap(
const cricket::RtpHeaderExtensions& header_extensions) override;
// SRTP is only active for a composite if it is active for all component
// transports.
bool IsSrtpActive() const override;
// Registers an RTP demux sink with all component transports.
bool RegisterRtpDemuxerSink(const RtpDemuxerCriteria& criteria,
RtpPacketSinkInterface* sink) override;
bool UnregisterRtpDemuxerSink(RtpPacketSinkInterface* sink) override;
private:
// Receive-side signals.
void OnNetworkRouteChanged(absl::optional<rtc::NetworkRoute> route);
void OnRtcpPacketReceived(rtc::CopyOnWriteBuffer* packet,
int64_t packet_time_us);
// Send-side signals.
void OnWritableState(bool writable);
void OnReadyToSend(bool ready_to_send);
void OnSentPacket(const rtc::SentPacket& packet);
std::vector<RtpTransportInternal*> transports_;
RtpTransportInternal* send_transport_ = nullptr;
// Record of registered RTP demuxer sinks. Used to unregister sinks when a
// transport is removed.
std::set<RtpPacketSinkInterface*> rtp_demuxer_sinks_;
};
} // namespace webrtc
#endif // PC_COMPOSITE_RTP_TRANSPORT_H_