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