1 /* 2 * Copyright 2017 The WebRTC project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef PC_RTP_TRANSPORT_H_ 12 #define PC_RTP_TRANSPORT_H_ 13 14 #include <stddef.h> 15 #include <stdint.h> 16 17 #include <string> 18 19 #include "absl/types/optional.h" 20 #include "call/rtp_demuxer.h" 21 #include "call/video_receive_stream.h" 22 #include "modules/rtp_rtcp/include/rtp_header_extension_map.h" 23 #include "p2p/base/packet_transport_internal.h" 24 #include "pc/rtp_transport_internal.h" 25 #include "pc/session_description.h" 26 #include "rtc_base/async_packet_socket.h" 27 #include "rtc_base/copy_on_write_buffer.h" 28 #include "rtc_base/network/sent_packet.h" 29 #include "rtc_base/network_route.h" 30 #include "rtc_base/socket.h" 31 32 namespace rtc { 33 34 class CopyOnWriteBuffer; 35 struct PacketOptions; 36 class PacketTransportInternal; 37 38 } // namespace rtc 39 40 namespace webrtc { 41 42 class RtpTransport : public RtpTransportInternal { 43 public: 44 RtpTransport(const RtpTransport&) = delete; 45 RtpTransport& operator=(const RtpTransport&) = delete; 46 RtpTransport(bool rtcp_mux_enabled)47 explicit RtpTransport(bool rtcp_mux_enabled) 48 : rtcp_mux_enabled_(rtcp_mux_enabled) {} 49 rtcp_mux_enabled()50 bool rtcp_mux_enabled() const override { return rtcp_mux_enabled_; } 51 void SetRtcpMuxEnabled(bool enable) override; 52 53 const std::string& transport_name() const override; 54 55 int SetRtpOption(rtc::Socket::Option opt, int value) override; 56 int SetRtcpOption(rtc::Socket::Option opt, int value) override; 57 rtp_packet_transport()58 rtc::PacketTransportInternal* rtp_packet_transport() const { 59 return rtp_packet_transport_; 60 } 61 void SetRtpPacketTransport(rtc::PacketTransportInternal* rtp); 62 rtcp_packet_transport()63 rtc::PacketTransportInternal* rtcp_packet_transport() const { 64 return rtcp_packet_transport_; 65 } 66 void SetRtcpPacketTransport(rtc::PacketTransportInternal* rtcp); 67 IsReadyToSend()68 bool IsReadyToSend() const override { return ready_to_send_; } 69 70 bool IsWritable(bool rtcp) const override; 71 72 bool SendRtpPacket(rtc::CopyOnWriteBuffer* packet, 73 const rtc::PacketOptions& options, 74 int flags) override; 75 76 bool SendRtcpPacket(rtc::CopyOnWriteBuffer* packet, 77 const rtc::PacketOptions& options, 78 int flags) override; 79 IsSrtpActive()80 bool IsSrtpActive() const override { return false; } 81 82 void UpdateRtpHeaderExtensionMap( 83 const cricket::RtpHeaderExtensions& header_extensions) override; 84 85 bool RegisterRtpDemuxerSink(const RtpDemuxerCriteria& criteria, 86 RtpPacketSinkInterface* sink) override; 87 88 bool UnregisterRtpDemuxerSink(RtpPacketSinkInterface* sink) override; 89 90 protected: 91 // These methods will be used in the subclasses. 92 void DemuxPacket(rtc::CopyOnWriteBuffer packet, int64_t packet_time_us); 93 94 bool SendPacket(bool rtcp, 95 rtc::CopyOnWriteBuffer* packet, 96 const rtc::PacketOptions& options, 97 int flags); 98 99 // Overridden by SrtpTransport. 100 virtual void OnNetworkRouteChanged( 101 absl::optional<rtc::NetworkRoute> network_route); 102 virtual void OnRtpPacketReceived(rtc::CopyOnWriteBuffer packet, 103 int64_t packet_time_us); 104 virtual void OnRtcpPacketReceived(rtc::CopyOnWriteBuffer packet, 105 int64_t packet_time_us); 106 // Overridden by SrtpTransport and DtlsSrtpTransport. 107 virtual void OnWritableState(rtc::PacketTransportInternal* packet_transport); 108 109 private: 110 void OnReadyToSend(rtc::PacketTransportInternal* transport); 111 void OnSentPacket(rtc::PacketTransportInternal* packet_transport, 112 const rtc::SentPacket& sent_packet); 113 void OnReadPacket(rtc::PacketTransportInternal* transport, 114 const char* data, 115 size_t len, 116 const int64_t& packet_time_us, 117 int flags); 118 119 // Updates "ready to send" for an individual channel and fires 120 // SignalReadyToSend. 121 void SetReadyToSend(bool rtcp, bool ready); 122 123 void MaybeSignalReadyToSend(); 124 125 bool IsTransportWritable(); 126 127 bool rtcp_mux_enabled_; 128 129 rtc::PacketTransportInternal* rtp_packet_transport_ = nullptr; 130 rtc::PacketTransportInternal* rtcp_packet_transport_ = nullptr; 131 132 bool ready_to_send_ = false; 133 bool rtp_ready_to_send_ = false; 134 bool rtcp_ready_to_send_ = false; 135 136 RtpDemuxer rtp_demuxer_; 137 138 // Used for identifying the MID for RtpDemuxer. 139 RtpHeaderExtensionMap header_extension_map_; 140 }; 141 142 } // namespace webrtc 143 144 #endif // PC_RTP_TRANSPORT_H_ 145