zstein | d48dbda | 2017-04-04 19:45:57 -0700 | [diff] [blame] | 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 | |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 11 | #include "pc/rtp_transport.h" |
zstein | d48dbda | 2017-04-04 19:45:57 -0700 | [diff] [blame] | 12 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 13 | #include <errno.h> |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 14 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 15 | #include <string> |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 16 | #include <utility> |
| 17 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 18 | #include "api/rtp_headers.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 19 | #include "api/rtp_parameters.h" |
| 20 | #include "media/base/rtp_utils.h" |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 21 | #include "modules/rtp_rtcp/source/rtp_packet_received.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 22 | #include "rtc_base/checks.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 23 | #include "rtc_base/copy_on_write_buffer.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 24 | #include "rtc_base/logging.h" |
| 25 | #include "rtc_base/third_party/sigslot/sigslot.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 26 | #include "rtc_base/trace_event.h" |
zstein | d48dbda | 2017-04-04 19:45:57 -0700 | [diff] [blame] | 27 | |
| 28 | namespace webrtc { |
| 29 | |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 30 | void RtpTransport::SetRtcpMuxEnabled(bool enable) { |
| 31 | rtcp_mux_enabled_ = enable; |
| 32 | MaybeSignalReadyToSend(); |
zstein | d9ce764 | 2017-04-10 16:17:57 -0700 | [diff] [blame] | 33 | } |
| 34 | |
Bjorn A Mellem | 3a1b927 | 2019-05-24 16:13:08 -0700 | [diff] [blame] | 35 | const std::string& RtpTransport::transport_name() const { |
| 36 | return rtp_packet_transport_->transport_name(); |
| 37 | } |
| 38 | |
| 39 | int RtpTransport::SetRtpOption(rtc::Socket::Option opt, int value) { |
| 40 | return rtp_packet_transport_->SetOption(opt, value); |
| 41 | } |
| 42 | |
| 43 | int RtpTransport::SetRtcpOption(rtc::Socket::Option opt, int value) { |
| 44 | if (rtcp_packet_transport_) { |
| 45 | return rtcp_packet_transport_->SetOption(opt, value); |
| 46 | } |
| 47 | return -1; |
| 48 | } |
| 49 | |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 50 | void RtpTransport::SetRtpPacketTransport( |
| 51 | rtc::PacketTransportInternal* new_packet_transport) { |
| 52 | if (new_packet_transport == rtp_packet_transport_) { |
| 53 | return; |
| 54 | } |
| 55 | if (rtp_packet_transport_) { |
| 56 | rtp_packet_transport_->SignalReadyToSend.disconnect(this); |
zstein | 3dcf0e9 | 2017-06-01 13:22:42 -0700 | [diff] [blame] | 57 | rtp_packet_transport_->SignalReadPacket.disconnect(this); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 58 | rtp_packet_transport_->SignalNetworkRouteChanged.disconnect(this); |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 59 | rtp_packet_transport_->SignalWritableState.disconnect(this); |
| 60 | rtp_packet_transport_->SignalSentPacket.disconnect(this); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 61 | // Reset the network route of the old transport. |
Danil Chapovalov | 66cadcc | 2018-06-19 16:47:43 +0200 | [diff] [blame] | 62 | SignalNetworkRouteChanged(absl::optional<rtc::NetworkRoute>()); |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 63 | } |
| 64 | if (new_packet_transport) { |
| 65 | new_packet_transport->SignalReadyToSend.connect( |
| 66 | this, &RtpTransport::OnReadyToSend); |
zstein | 3dcf0e9 | 2017-06-01 13:22:42 -0700 | [diff] [blame] | 67 | new_packet_transport->SignalReadPacket.connect(this, |
| 68 | &RtpTransport::OnReadPacket); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 69 | new_packet_transport->SignalNetworkRouteChanged.connect( |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 70 | this, &RtpTransport::OnNetworkRouteChanged); |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 71 | new_packet_transport->SignalWritableState.connect( |
| 72 | this, &RtpTransport::OnWritableState); |
| 73 | new_packet_transport->SignalSentPacket.connect(this, |
| 74 | &RtpTransport::OnSentPacket); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 75 | // Set the network route for the new transport. |
| 76 | SignalNetworkRouteChanged(new_packet_transport->network_route()); |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 77 | } |
Zhi Huang | 8c316c1 | 2017-11-13 21:13:45 +0000 | [diff] [blame] | 78 | |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 79 | rtp_packet_transport_ = new_packet_transport; |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 80 | // Assumes the transport is ready to send if it is writable. If we are wrong, |
| 81 | // ready to send will be updated the next time we try to send. |
| 82 | SetReadyToSend(false, |
| 83 | rtp_packet_transport_ && rtp_packet_transport_->writable()); |
| 84 | } |
| 85 | |
| 86 | void RtpTransport::SetRtcpPacketTransport( |
| 87 | rtc::PacketTransportInternal* new_packet_transport) { |
| 88 | if (new_packet_transport == rtcp_packet_transport_) { |
| 89 | return; |
| 90 | } |
| 91 | if (rtcp_packet_transport_) { |
| 92 | rtcp_packet_transport_->SignalReadyToSend.disconnect(this); |
zstein | 3dcf0e9 | 2017-06-01 13:22:42 -0700 | [diff] [blame] | 93 | rtcp_packet_transport_->SignalReadPacket.disconnect(this); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 94 | rtcp_packet_transport_->SignalNetworkRouteChanged.disconnect(this); |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 95 | rtcp_packet_transport_->SignalWritableState.disconnect(this); |
| 96 | rtcp_packet_transport_->SignalSentPacket.disconnect(this); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 97 | // Reset the network route of the old transport. |
Danil Chapovalov | 66cadcc | 2018-06-19 16:47:43 +0200 | [diff] [blame] | 98 | SignalNetworkRouteChanged(absl::optional<rtc::NetworkRoute>()); |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 99 | } |
| 100 | if (new_packet_transport) { |
| 101 | new_packet_transport->SignalReadyToSend.connect( |
| 102 | this, &RtpTransport::OnReadyToSend); |
zstein | 3dcf0e9 | 2017-06-01 13:22:42 -0700 | [diff] [blame] | 103 | new_packet_transport->SignalReadPacket.connect(this, |
| 104 | &RtpTransport::OnReadPacket); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 105 | new_packet_transport->SignalNetworkRouteChanged.connect( |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 106 | this, &RtpTransport::OnNetworkRouteChanged); |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 107 | new_packet_transport->SignalWritableState.connect( |
| 108 | this, &RtpTransport::OnWritableState); |
| 109 | new_packet_transport->SignalSentPacket.connect(this, |
| 110 | &RtpTransport::OnSentPacket); |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 111 | // Set the network route for the new transport. |
| 112 | SignalNetworkRouteChanged(new_packet_transport->network_route()); |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 113 | } |
| 114 | rtcp_packet_transport_ = new_packet_transport; |
| 115 | |
| 116 | // Assumes the transport is ready to send if it is writable. If we are wrong, |
| 117 | // ready to send will be updated the next time we try to send. |
| 118 | SetReadyToSend(true, |
| 119 | rtcp_packet_transport_ && rtcp_packet_transport_->writable()); |
| 120 | } |
| 121 | |
| 122 | bool RtpTransport::IsWritable(bool rtcp) const { |
| 123 | rtc::PacketTransportInternal* transport = rtcp && !rtcp_mux_enabled_ |
| 124 | ? rtcp_packet_transport_ |
| 125 | : rtp_packet_transport_; |
| 126 | return transport && transport->writable(); |
| 127 | } |
| 128 | |
Zhi Huang | cf990f5 | 2017-09-22 12:12:30 -0700 | [diff] [blame] | 129 | bool RtpTransport::SendRtpPacket(rtc::CopyOnWriteBuffer* packet, |
| 130 | const rtc::PacketOptions& options, |
| 131 | int flags) { |
| 132 | return SendPacket(false, packet, options, flags); |
| 133 | } |
| 134 | |
| 135 | bool RtpTransport::SendRtcpPacket(rtc::CopyOnWriteBuffer* packet, |
| 136 | const rtc::PacketOptions& options, |
| 137 | int flags) { |
| 138 | return SendPacket(true, packet, options, flags); |
| 139 | } |
| 140 | |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 141 | bool RtpTransport::SendPacket(bool rtcp, |
zstein | 398c3fd | 2017-07-19 13:38:02 -0700 | [diff] [blame] | 142 | rtc::CopyOnWriteBuffer* packet, |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 143 | const rtc::PacketOptions& options, |
| 144 | int flags) { |
| 145 | rtc::PacketTransportInternal* transport = rtcp && !rtcp_mux_enabled_ |
| 146 | ? rtcp_packet_transport_ |
| 147 | : rtp_packet_transport_; |
Amit Hilbuch | e7a5f7b | 2019-03-12 11:10:27 -0700 | [diff] [blame] | 148 | int ret = transport->SendPacket(packet->cdata<char>(), packet->size(), |
| 149 | options, flags); |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 150 | if (ret != static_cast<int>(packet->size())) { |
| 151 | if (transport->GetError() == ENOTCONN) { |
Mirko Bonadei | 675513b | 2017-11-09 11:09:25 +0100 | [diff] [blame] | 152 | RTC_LOG(LS_WARNING) << "Got ENOTCONN from transport."; |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 153 | SetReadyToSend(rtcp, false); |
| 154 | } |
| 155 | return false; |
| 156 | } |
| 157 | return true; |
zstein | d48dbda | 2017-04-04 19:45:57 -0700 | [diff] [blame] | 158 | } |
| 159 | |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 160 | void RtpTransport::UpdateRtpHeaderExtensionMap( |
| 161 | const cricket::RtpHeaderExtensions& header_extensions) { |
| 162 | header_extension_map_ = RtpHeaderExtensionMap(header_extensions); |
| 163 | } |
| 164 | |
| 165 | bool RtpTransport::RegisterRtpDemuxerSink(const RtpDemuxerCriteria& criteria, |
| 166 | RtpPacketSinkInterface* sink) { |
| 167 | rtp_demuxer_.RemoveSink(sink); |
| 168 | if (!rtp_demuxer_.AddSink(criteria, sink)) { |
| 169 | RTC_LOG(LS_ERROR) << "Failed to register the sink for RTP demuxer."; |
| 170 | return false; |
| 171 | } |
| 172 | return true; |
| 173 | } |
| 174 | |
| 175 | bool RtpTransport::UnregisterRtpDemuxerSink(RtpPacketSinkInterface* sink) { |
| 176 | if (!rtp_demuxer_.RemoveSink(sink)) { |
| 177 | RTC_LOG(LS_ERROR) << "Failed to unregister the sink for RTP demuxer."; |
| 178 | return false; |
| 179 | } |
| 180 | return true; |
| 181 | } |
| 182 | |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 183 | void RtpTransport::DemuxPacket(rtc::CopyOnWriteBuffer packet, |
Niels Möller | e693381 | 2018-11-05 13:01:41 +0100 | [diff] [blame] | 184 | int64_t packet_time_us) { |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 185 | webrtc::RtpPacketReceived parsed_packet(&header_extension_map_); |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 186 | if (!parsed_packet.Parse(std::move(packet))) { |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 187 | RTC_LOG(LS_ERROR) |
| 188 | << "Failed to parse the incoming RTP packet before demuxing. Drop it."; |
| 189 | return; |
| 190 | } |
| 191 | |
Niels Möller | e693381 | 2018-11-05 13:01:41 +0100 | [diff] [blame] | 192 | if (packet_time_us != -1) { |
| 193 | parsed_packet.set_arrival_time_ms((packet_time_us + 500) / 1000); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 194 | } |
Steve Anton | 884adca | 2019-04-15 16:52:27 -0700 | [diff] [blame] | 195 | if (!rtp_demuxer_.OnRtpPacket(parsed_packet)) { |
| 196 | RTC_LOG(LS_WARNING) << "Failed to demux RTP packet: " |
| 197 | << RtpDemuxer::DescribePacket(parsed_packet); |
| 198 | } |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 199 | } |
| 200 | |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 201 | bool RtpTransport::IsTransportWritable() { |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 202 | auto rtcp_packet_transport = |
| 203 | rtcp_mux_enabled_ ? nullptr : rtcp_packet_transport_; |
| 204 | return rtp_packet_transport_ && rtp_packet_transport_->writable() && |
| 205 | (!rtcp_packet_transport || rtcp_packet_transport->writable()); |
| 206 | } |
| 207 | |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 208 | void RtpTransport::OnReadyToSend(rtc::PacketTransportInternal* transport) { |
| 209 | SetReadyToSend(transport == rtcp_packet_transport_, true); |
| 210 | } |
| 211 | |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 212 | void RtpTransport::OnNetworkRouteChanged( |
Danil Chapovalov | 66cadcc | 2018-06-19 16:47:43 +0200 | [diff] [blame] | 213 | absl::optional<rtc::NetworkRoute> network_route) { |
Zhi Huang | 942bc2e | 2017-11-13 13:26:07 -0800 | [diff] [blame] | 214 | SignalNetworkRouteChanged(network_route); |
| 215 | } |
| 216 | |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 217 | void RtpTransport::OnWritableState( |
| 218 | rtc::PacketTransportInternal* packet_transport) { |
| 219 | RTC_DCHECK(packet_transport == rtp_packet_transport_ || |
| 220 | packet_transport == rtcp_packet_transport_); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 221 | SignalWritableState(IsTransportWritable()); |
Zhi Huang | cd3fc5d | 2017-11-29 10:41:57 -0800 | [diff] [blame] | 222 | } |
| 223 | |
| 224 | void RtpTransport::OnSentPacket(rtc::PacketTransportInternal* packet_transport, |
| 225 | const rtc::SentPacket& sent_packet) { |
| 226 | RTC_DCHECK(packet_transport == rtp_packet_transport_ || |
| 227 | packet_transport == rtcp_packet_transport_); |
| 228 | SignalSentPacket(sent_packet); |
| 229 | } |
| 230 | |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 231 | void RtpTransport::OnRtpPacketReceived(rtc::CopyOnWriteBuffer packet, |
Niels Möller | e693381 | 2018-11-05 13:01:41 +0100 | [diff] [blame] | 232 | int64_t packet_time_us) { |
| 233 | DemuxPacket(packet, packet_time_us); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 234 | } |
| 235 | |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 236 | void RtpTransport::OnRtcpPacketReceived(rtc::CopyOnWriteBuffer packet, |
Niels Möller | e693381 | 2018-11-05 13:01:41 +0100 | [diff] [blame] | 237 | int64_t packet_time_us) { |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 238 | SignalRtcpPacketReceived(&packet, packet_time_us); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 239 | } |
| 240 | |
| 241 | void RtpTransport::OnReadPacket(rtc::PacketTransportInternal* transport, |
| 242 | const char* data, |
| 243 | size_t len, |
Niels Möller | e693381 | 2018-11-05 13:01:41 +0100 | [diff] [blame] | 244 | const int64_t& packet_time_us, |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 245 | int flags) { |
| 246 | TRACE_EVENT0("webrtc", "RtpTransport::OnReadPacket"); |
| 247 | |
| 248 | // When using RTCP multiplexing we might get RTCP packets on the RTP |
| 249 | // transport. We check the RTP payload type to determine if it is RTCP. |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 250 | auto array_view = rtc::MakeArrayView(data, len); |
| 251 | cricket::RtpPacketType packet_type = cricket::InferRtpPacketType(array_view); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 252 | // Filter out the packet that is neither RTP nor RTCP. |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 253 | if (packet_type == cricket::RtpPacketType::kUnknown) { |
| 254 | return; |
| 255 | } |
| 256 | |
| 257 | // Protect ourselves against crazy data. |
| 258 | if (!cricket::IsValidRtpPacketSize(packet_type, len)) { |
| 259 | RTC_LOG(LS_ERROR) << "Dropping incoming " |
| 260 | << cricket::RtpPacketTypeToString(packet_type) |
| 261 | << " packet: wrong size=" << len; |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 262 | return; |
| 263 | } |
| 264 | |
| 265 | rtc::CopyOnWriteBuffer packet(data, len); |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 266 | if (packet_type == cricket::RtpPacketType::kRtcp) { |
| 267 | OnRtcpPacketReceived(std::move(packet), packet_time_us); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 268 | } else { |
Amit Hilbuch | edd2054 | 2019-03-18 12:33:43 -0700 | [diff] [blame] | 269 | OnRtpPacketReceived(std::move(packet), packet_time_us); |
Zhi Huang | 365381f | 2018-04-13 16:44:34 -0700 | [diff] [blame] | 270 | } |
| 271 | } |
| 272 | |
zstein | 56162b9 | 2017-04-24 16:54:35 -0700 | [diff] [blame] | 273 | void RtpTransport::SetReadyToSend(bool rtcp, bool ready) { |
| 274 | if (rtcp) { |
| 275 | rtcp_ready_to_send_ = ready; |
| 276 | } else { |
| 277 | rtp_ready_to_send_ = ready; |
| 278 | } |
| 279 | |
| 280 | MaybeSignalReadyToSend(); |
| 281 | } |
| 282 | |
| 283 | void RtpTransport::MaybeSignalReadyToSend() { |
| 284 | bool ready_to_send = |
| 285 | rtp_ready_to_send_ && (rtcp_ready_to_send_ || rtcp_mux_enabled_); |
| 286 | if (ready_to_send != ready_to_send_) { |
| 287 | ready_to_send_ = ready_to_send; |
| 288 | SignalReadyToSend(ready_to_send); |
| 289 | } |
| 290 | } |
| 291 | |
zstein | d48dbda | 2017-04-04 19:45:57 -0700 | [diff] [blame] | 292 | } // namespace webrtc |