niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 1 | /* |
leozwang@webrtc.org | 39e9659 | 2012-03-01 18:22:48 +0000 | [diff] [blame] | 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 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 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 11 | #include "webrtc/video/rtp_stream_receiver.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 12 | |
mflodman@webrtc.org | 4fd5527 | 2013-02-06 17:46:39 +0000 | [diff] [blame] | 13 | #include <vector> |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 14 | #include <utility> |
mflodman@webrtc.org | 4fd5527 | 2013-02-06 17:46:39 +0000 | [diff] [blame] | 15 | |
mflodman | 4cd2790 | 2016-08-05 06:28:45 -0700 | [diff] [blame] | 16 | #include "webrtc/base/checks.h" |
tommi | dea489f | 2017-03-03 03:20:24 -0800 | [diff] [blame] | 17 | #include "webrtc/base/location.h" |
Peter Boström | 415d2cd | 2015-10-26 11:35:17 +0100 | [diff] [blame] | 18 | #include "webrtc/base/logging.h" |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 19 | #include "webrtc/common_types.h" |
Peter Boström | 9c01725 | 2016-02-26 16:26:20 +0100 | [diff] [blame] | 20 | #include "webrtc/config.h" |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 21 | #include "webrtc/media/base/mediaconstants.h" |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 22 | #include "webrtc/modules/pacing/packet_router.h" |
pbos@webrtc.org | f5d4cb1 | 2013-05-17 13:44:48 +0000 | [diff] [blame] | 23 | #include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h" |
Henrik Kjellander | ff761fb | 2015-11-04 08:31:52 +0100 | [diff] [blame] | 24 | #include "webrtc/modules/rtp_rtcp/include/receive_statistics.h" |
Henrik Kjellander | ff761fb | 2015-11-04 08:31:52 +0100 | [diff] [blame] | 25 | #include "webrtc/modules/rtp_rtcp/include/rtp_cvo.h" |
| 26 | #include "webrtc/modules/rtp_rtcp/include/rtp_header_parser.h" |
Henrik Kjellander | ff761fb | 2015-11-04 08:31:52 +0100 | [diff] [blame] | 27 | #include "webrtc/modules/rtp_rtcp/include/rtp_receiver.h" |
| 28 | #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 29 | #include "webrtc/modules/rtp_rtcp/include/ulpfec_receiver.h" |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 30 | #include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h" |
| 31 | #include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h" |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 32 | #include "webrtc/modules/video_coding/frame_object.h" |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 33 | #include "webrtc/modules/video_coding/h264_sprop_parameter_sets.h" |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 34 | #include "webrtc/modules/video_coding/h264_sps_pps_tracker.h" |
| 35 | #include "webrtc/modules/video_coding/packet_buffer.h" |
Peter Boström | 0b25072 | 2016-04-22 18:23:15 +0200 | [diff] [blame] | 36 | #include "webrtc/modules/video_coding/video_coding_impl.h" |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 37 | #include "webrtc/system_wrappers/include/field_trial.h" |
Henrik Kjellander | 98f5351 | 2015-10-28 18:17:40 +0100 | [diff] [blame] | 38 | #include "webrtc/system_wrappers/include/metrics.h" |
Henrik Kjellander | 98f5351 | 2015-10-28 18:17:40 +0100 | [diff] [blame] | 39 | #include "webrtc/system_wrappers/include/timestamp_extrapolator.h" |
| 40 | #include "webrtc/system_wrappers/include/trace.h" |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 41 | #include "webrtc/video/receive_statistics_proxy.h" |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 42 | #include "webrtc/video/vie_remb.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 43 | |
| 44 | namespace webrtc { |
| 45 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 46 | namespace { |
| 47 | constexpr int kPacketBufferStartSize = 32; |
| 48 | constexpr int kPacketBufferMaxSixe = 2048; |
| 49 | } |
| 50 | |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 51 | std::unique_ptr<RtpRtcp> CreateRtpRtcpModule( |
| 52 | ReceiveStatistics* receive_statistics, |
| 53 | Transport* outgoing_transport, |
| 54 | RtcpRttStats* rtt_stats, |
| 55 | RtcpPacketTypeCounterObserver* rtcp_packet_type_counter_observer, |
nisse | 15389c0 | 2017-01-24 02:36:58 -0800 | [diff] [blame] | 56 | TransportSequenceNumberAllocator* transport_sequence_number_allocator) { |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 57 | RtpRtcp::Configuration configuration; |
| 58 | configuration.audio = false; |
| 59 | configuration.receiver_only = true; |
| 60 | configuration.receive_statistics = receive_statistics; |
| 61 | configuration.outgoing_transport = outgoing_transport; |
| 62 | configuration.intra_frame_callback = nullptr; |
| 63 | configuration.rtt_stats = rtt_stats; |
| 64 | configuration.rtcp_packet_type_counter_observer = |
| 65 | rtcp_packet_type_counter_observer; |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 66 | configuration.transport_sequence_number_allocator = |
| 67 | transport_sequence_number_allocator; |
| 68 | configuration.send_bitrate_observer = nullptr; |
| 69 | configuration.send_frame_count_observer = nullptr; |
| 70 | configuration.send_side_delay_observer = nullptr; |
asapersson | 35151f3 | 2016-05-02 23:44:01 -0700 | [diff] [blame] | 71 | configuration.send_packet_observer = nullptr; |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 72 | configuration.bandwidth_callback = nullptr; |
| 73 | configuration.transport_feedback_callback = nullptr; |
| 74 | |
| 75 | std::unique_ptr<RtpRtcp> rtp_rtcp(RtpRtcp::CreateRtpRtcp(configuration)); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 76 | rtp_rtcp->SetRTCPStatus(RtcpMode::kCompound); |
| 77 | |
| 78 | return rtp_rtcp; |
| 79 | } |
| 80 | |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 81 | static const int kPacketLogIntervalMs = 10000; |
| 82 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 83 | RtpStreamReceiver::RtpStreamReceiver( |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 84 | Transport* transport, |
| 85 | RtcpRttStats* rtt_stats, |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 86 | PacketRouter* packet_router, |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 87 | VieRemb* remb, |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 88 | const VideoReceiveStream::Config* config, |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 89 | ReceiveStatisticsProxy* receive_stats_proxy, |
Erik Språng | 737336d | 2016-07-29 12:59:36 +0200 | [diff] [blame] | 90 | ProcessThread* process_thread, |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 91 | NackSender* nack_sender, |
| 92 | KeyFrameRequestSender* keyframe_request_sender, |
| 93 | video_coding::OnCompleteFrameCallback* complete_frame_callback, |
| 94 | VCMTiming* timing) |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 95 | : clock_(Clock::GetRealTimeClock()), |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 96 | config_(*config), |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 97 | packet_router_(packet_router), |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 98 | remb_(remb), |
| 99 | process_thread_(process_thread), |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 100 | ntp_estimator_(clock_), |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 101 | rtp_header_parser_(RtpHeaderParser::Create()), |
| 102 | rtp_receiver_(RtpReceiver::CreateVideoReceiver(clock_, |
| 103 | this, |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 104 | this, |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 105 | &rtp_payload_registry_)), |
| 106 | rtp_receive_statistics_(ReceiveStatistics::Create(clock_)), |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 107 | ulpfec_receiver_(UlpfecReceiver::Create(this)), |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 108 | receiving_(false), |
solenberg@webrtc.org | 3fb8f7b | 2014-03-24 20:28:11 +0000 | [diff] [blame] | 109 | restored_packet_in_use_(false), |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 110 | last_packet_log_ms_(-1), |
| 111 | rtp_rtcp_(CreateRtpRtcpModule(rtp_receive_statistics_.get(), |
| 112 | transport, |
| 113 | rtt_stats, |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 114 | receive_stats_proxy, |
nisse | 15389c0 | 2017-01-24 02:36:58 -0800 | [diff] [blame] | 115 | packet_router)), |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 116 | complete_frame_callback_(complete_frame_callback), |
| 117 | keyframe_request_sender_(keyframe_request_sender), |
| 118 | timing_(timing) { |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 119 | packet_router_->AddRtpModule(rtp_rtcp_.get()); |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 120 | rtp_receive_statistics_->RegisterRtpStatisticsCallback(receive_stats_proxy); |
| 121 | rtp_receive_statistics_->RegisterRtcpStatisticsCallback(receive_stats_proxy); |
| 122 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 123 | RTC_DCHECK(config_.rtp.rtcp_mode != RtcpMode::kOff) |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 124 | << "A stream should not be configured with RTCP disabled. This value is " |
| 125 | "reserved for internal usage."; |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 126 | RTC_DCHECK(config_.rtp.remote_ssrc != 0); |
| 127 | // TODO(pbos): What's an appropriate local_ssrc for receive-only streams? |
| 128 | RTC_DCHECK(config_.rtp.local_ssrc != 0); |
| 129 | RTC_DCHECK(config_.rtp.remote_ssrc != config_.rtp.local_ssrc); |
| 130 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 131 | rtp_rtcp_->SetRTCPStatus(config_.rtp.rtcp_mode); |
| 132 | rtp_rtcp_->SetSSRC(config_.rtp.local_ssrc); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 133 | rtp_rtcp_->SetKeyFrameRequestMethod(kKeyFrameReqPliRtcp); |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 134 | if (config_.rtp.remb) { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 135 | rtp_rtcp_->SetREMBStatus(true); |
| 136 | remb_->AddReceiveChannel(rtp_rtcp_.get()); |
| 137 | } |
| 138 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 139 | for (size_t i = 0; i < config_.rtp.extensions.size(); ++i) { |
| 140 | EnableReceiveRtpHeaderExtension(config_.rtp.extensions[i].uri, |
| 141 | config_.rtp.extensions[i].id); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 142 | } |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 143 | |
| 144 | static const int kMaxPacketAgeToNack = 450; |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 145 | const int max_reordering_threshold = (config_.rtp.nack.rtp_history_ms > 0) |
| 146 | ? kMaxPacketAgeToNack |
| 147 | : kDefaultMaxReorderingThreshold; |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 148 | rtp_receive_statistics_->SetMaxReorderingThreshold(max_reordering_threshold); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 149 | |
brandtr | 1474212 | 2017-01-27 04:53:07 -0800 | [diff] [blame] | 150 | if (config_.rtp.rtx_ssrc) { |
| 151 | rtp_payload_registry_.SetRtxSsrc(config_.rtp.rtx_ssrc); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 152 | |
brandtr | 1474212 | 2017-01-27 04:53:07 -0800 | [diff] [blame] | 153 | for (const auto& kv : config_.rtp.rtx_payload_types) { |
| 154 | RTC_DCHECK(kv.second != 0); |
| 155 | rtp_payload_registry_.SetRtxPayloadType(kv.second, kv.first); |
| 156 | } |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 157 | } |
| 158 | |
brandtr | f7c6d72 | 2016-12-08 08:25:47 -0800 | [diff] [blame] | 159 | if (IsUlpfecEnabled()) { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 160 | VideoCodec ulpfec_codec = {}; |
| 161 | ulpfec_codec.codecType = kVideoCodecULPFEC; |
| 162 | strncpy(ulpfec_codec.plName, "ulpfec", sizeof(ulpfec_codec.plName)); |
brandtr | b5f2c3f | 2016-10-04 23:28:39 -0700 | [diff] [blame] | 163 | ulpfec_codec.plType = config_.rtp.ulpfec.ulpfec_payload_type; |
johan | 07e276c | 2016-12-13 02:23:03 -0800 | [diff] [blame] | 164 | RTC_CHECK(AddReceiveCodec(ulpfec_codec)); |
brandtr | e6f98c7 | 2016-11-11 03:28:30 -0800 | [diff] [blame] | 165 | } |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 166 | |
brandtr | e6f98c7 | 2016-11-11 03:28:30 -0800 | [diff] [blame] | 167 | if (IsRedEnabled()) { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 168 | VideoCodec red_codec = {}; |
| 169 | red_codec.codecType = kVideoCodecRED; |
| 170 | strncpy(red_codec.plName, "red", sizeof(red_codec.plName)); |
brandtr | b5f2c3f | 2016-10-04 23:28:39 -0700 | [diff] [blame] | 171 | red_codec.plType = config_.rtp.ulpfec.red_payload_type; |
johan | 07e276c | 2016-12-13 02:23:03 -0800 | [diff] [blame] | 172 | RTC_CHECK(AddReceiveCodec(red_codec)); |
brandtr | b5f2c3f | 2016-10-04 23:28:39 -0700 | [diff] [blame] | 173 | if (config_.rtp.ulpfec.red_rtx_payload_type != -1) { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 174 | rtp_payload_registry_.SetRtxPayloadType( |
brandtr | b5f2c3f | 2016-10-04 23:28:39 -0700 | [diff] [blame] | 175 | config_.rtp.ulpfec.red_rtx_payload_type, |
| 176 | config_.rtp.ulpfec.red_payload_type); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 177 | } |
| 178 | } |
| 179 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 180 | if (config_.rtp.rtcp_xr.receiver_reference_time_report) |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 181 | rtp_rtcp_->SetRtcpXrRrtrStatus(true); |
| 182 | |
| 183 | // Stats callback for CNAME changes. |
| 184 | rtp_rtcp_->RegisterRtcpStatisticsCallback(receive_stats_proxy); |
| 185 | |
tommi | dea489f | 2017-03-03 03:20:24 -0800 | [diff] [blame] | 186 | process_thread_->RegisterModule(rtp_rtcp_.get(), RTC_FROM_HERE); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 187 | |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 188 | if (config_.rtp.nack.rtp_history_ms != 0) { |
| 189 | nack_module_.reset( |
| 190 | new NackModule(clock_, nack_sender, keyframe_request_sender)); |
tommi | dea489f | 2017-03-03 03:20:24 -0800 | [diff] [blame] | 191 | process_thread_->RegisterModule(nack_module_.get(), RTC_FROM_HERE); |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 192 | } |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 193 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 194 | packet_buffer_ = video_coding::PacketBuffer::Create( |
| 195 | clock_, kPacketBufferStartSize, kPacketBufferMaxSixe, this); |
| 196 | reference_finder_.reset(new video_coding::RtpFrameReferenceFinder(this)); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 197 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 198 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 199 | RtpStreamReceiver::~RtpStreamReceiver() { |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 200 | if (nack_module_) { |
| 201 | process_thread_->DeRegisterModule(nack_module_.get()); |
| 202 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 203 | |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 204 | process_thread_->DeRegisterModule(rtp_rtcp_.get()); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 205 | |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 206 | packet_router_->RemoveRtpModule(rtp_rtcp_.get()); |
| 207 | rtp_rtcp_->SetREMBStatus(false); |
johan | 62d02c3 | 2017-01-24 04:38:27 -0800 | [diff] [blame] | 208 | if (config_.rtp.remb) { |
| 209 | remb_->RemoveReceiveChannel(rtp_rtcp_.get()); |
| 210 | } |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 211 | UpdateHistograms(); |
asapersson@webrtc.org | 0800db7 | 2015-01-15 07:40:20 +0000 | [diff] [blame] | 212 | } |
| 213 | |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 214 | bool RtpStreamReceiver::AddReceiveCodec( |
| 215 | const VideoCodec& video_codec, |
| 216 | const std::map<std::string, std::string>& codec_params) { |
| 217 | pt_codec_params_.insert(make_pair(video_codec.plType, codec_params)); |
| 218 | return AddReceiveCodec(video_codec); |
| 219 | } |
| 220 | |
johan | 07e276c | 2016-12-13 02:23:03 -0800 | [diff] [blame] | 221 | bool RtpStreamReceiver::AddReceiveCodec(const VideoCodec& video_codec) { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 222 | int8_t old_pltype = -1; |
magjed | 56124bd | 2016-11-24 09:34:46 -0800 | [diff] [blame] | 223 | if (rtp_payload_registry_.ReceivePayloadType(video_codec, &old_pltype) != |
| 224 | -1) { |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 225 | rtp_payload_registry_.DeRegisterReceivePayload(old_pltype); |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 226 | } |
magjed | 56124bd | 2016-11-24 09:34:46 -0800 | [diff] [blame] | 227 | return rtp_payload_registry_.RegisterReceivePayload(video_codec) == 0; |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 228 | } |
| 229 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 230 | uint32_t RtpStreamReceiver::GetRemoteSsrc() const { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 231 | return rtp_receiver_->SSRC(); |
| 232 | } |
| 233 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 234 | int RtpStreamReceiver::GetCsrcs(uint32_t* csrcs) const { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 235 | return rtp_receiver_->CSRCs(csrcs); |
| 236 | } |
| 237 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 238 | RtpReceiver* RtpStreamReceiver::GetRtpReceiver() const { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 239 | return rtp_receiver_.get(); |
| 240 | } |
| 241 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 242 | int32_t RtpStreamReceiver::OnReceivedPayloadData( |
| 243 | const uint8_t* payload_data, |
Peter Boström | 0208322 | 2016-06-14 12:52:54 +0200 | [diff] [blame] | 244 | size_t payload_size, |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 245 | const WebRtcRTPHeader* rtp_header) { |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 246 | WebRtcRTPHeader rtp_header_with_ntp = *rtp_header; |
wu@webrtc.org | 88abf11 | 2014-05-14 16:53:51 +0000 | [diff] [blame] | 247 | rtp_header_with_ntp.ntp_time_ms = |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 248 | ntp_estimator_.Estimate(rtp_header->header.timestamp); |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 249 | VCMPacket packet(payload_data, payload_size, rtp_header_with_ntp); |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 250 | packet.timesNacked = |
| 251 | nack_module_ ? nack_module_->OnReceivedPacket(packet) : -1; |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 252 | |
philipel | 54ca919 | 2017-03-21 05:45:18 -0700 | [diff] [blame^] | 253 | // In the case of a video stream without picture ids and no rtx the |
| 254 | // RtpFrameReferenceFinder will need to know about padding to |
| 255 | // correctly calculate frame references. |
| 256 | if (packet.sizeBytes == 0) { |
| 257 | reference_finder_->PaddingReceived(packet.seqNum); |
| 258 | return 0; |
| 259 | } |
| 260 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 261 | if (packet.codec == kVideoCodecH264) { |
| 262 | // Only when we start to receive packets will we know what payload type |
| 263 | // that will be used. When we know the payload type insert the correct |
| 264 | // sps/pps into the tracker. |
| 265 | if (packet.payloadType != last_payload_type_) { |
| 266 | last_payload_type_ = packet.payloadType; |
| 267 | InsertSpsPpsIntoTracker(packet.payloadType); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 268 | } |
| 269 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 270 | switch (tracker_.CopyAndFixBitstream(&packet)) { |
| 271 | case video_coding::H264SpsPpsTracker::kRequestKeyframe: |
| 272 | keyframe_request_sender_->RequestKeyFrame(); |
| 273 | FALLTHROUGH(); |
| 274 | case video_coding::H264SpsPpsTracker::kDrop: |
| 275 | return 0; |
| 276 | case video_coding::H264SpsPpsTracker::kInsert: |
| 277 | break; |
| 278 | } |
| 279 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 280 | } else { |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 281 | uint8_t* data = new uint8_t[packet.sizeBytes]; |
| 282 | memcpy(data, packet.dataPtr, packet.sizeBytes); |
| 283 | packet.dataPtr = data; |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 284 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 285 | |
| 286 | packet_buffer_->InsertPacket(&packet); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 287 | return 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 288 | } |
| 289 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 290 | bool RtpStreamReceiver::OnRecoveredPacket(const uint8_t* rtp_packet, |
| 291 | size_t rtp_packet_length) { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 292 | RTPHeader header; |
| 293 | if (!rtp_header_parser_->Parse(rtp_packet, rtp_packet_length, &header)) { |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 294 | return false; |
| 295 | } |
| 296 | header.payload_type_frequency = kVideoPayloadTypeFrequency; |
stefan@webrtc.org | 01581da | 2014-09-04 06:48:14 +0000 | [diff] [blame] | 297 | bool in_order = IsPacketInOrder(header); |
| 298 | return ReceivePacket(rtp_packet, rtp_packet_length, header, in_order); |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 299 | } |
| 300 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 301 | // TODO(pbos): Remove as soon as audio can handle a changing payload type |
| 302 | // without this callback. |
| 303 | int32_t RtpStreamReceiver::OnInitializeDecoder( |
| 304 | const int8_t payload_type, |
| 305 | const char payload_name[RTP_PAYLOAD_NAME_SIZE], |
| 306 | const int frequency, |
| 307 | const size_t channels, |
| 308 | const uint32_t rate) { |
| 309 | RTC_NOTREACHED(); |
| 310 | return 0; |
| 311 | } |
| 312 | |
| 313 | void RtpStreamReceiver::OnIncomingSSRCChanged(const uint32_t ssrc) { |
| 314 | rtp_rtcp_->SetRemoteSSRC(ssrc); |
| 315 | } |
| 316 | |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 317 | void RtpStreamReceiver::OnRtpPacket(const RtpPacketReceived& packet) { |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 318 | { |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 319 | rtc::CritScope lock(&receive_cs_); |
braveyao@webrtc.org | b6433b7 | 2013-07-26 09:02:46 +0000 | [diff] [blame] | 320 | if (!receiving_) { |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 321 | return; |
braveyao@webrtc.org | b6433b7 | 2013-07-26 09:02:46 +0000 | [diff] [blame] | 322 | } |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 323 | } |
solenberg@webrtc.org | fc32046 | 2014-02-11 15:27:49 +0000 | [diff] [blame] | 324 | |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 325 | int64_t now_ms = clock_->TimeInMilliseconds(); |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 326 | |
| 327 | { |
| 328 | // Periodically log the RTP header of incoming packets. |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 329 | rtc::CritScope lock(&receive_cs_); |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 330 | if (now_ms - last_packet_log_ms_ > kPacketLogIntervalMs) { |
| 331 | std::stringstream ss; |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 332 | ss << "Packet received on SSRC: " << packet.Ssrc() |
| 333 | << " with payload type: " << static_cast<int>(packet.PayloadType()) |
| 334 | << ", timestamp: " << packet.Timestamp() |
| 335 | << ", sequence number: " << packet.SequenceNumber() |
| 336 | << ", arrival time: " << packet.arrival_time_ms(); |
| 337 | int32_t time_offset; |
| 338 | if (packet.GetExtension<TransmissionOffset>(&time_offset)) { |
| 339 | ss << ", toffset: " << time_offset; |
| 340 | } |
| 341 | uint32_t send_time; |
| 342 | if (packet.GetExtension<AbsoluteSendTime>(&send_time)) { |
| 343 | ss << ", abs send time: " << send_time; |
| 344 | } |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 345 | LOG(LS_INFO) << ss.str(); |
| 346 | last_packet_log_ms_ = now_ms; |
| 347 | } |
| 348 | } |
wu@webrtc.org | a989080 | 2013-12-13 00:21:03 +0000 | [diff] [blame] | 349 | |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 350 | // TODO(nisse): Delete use of GetHeader, but needs refactoring of |
| 351 | // ReceivePacket and IncomingPacket methods below. |
| 352 | RTPHeader header; |
| 353 | packet.GetHeader(&header); |
| 354 | |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 355 | header.payload_type_frequency = kVideoPayloadTypeFrequency; |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 356 | |
stefan@webrtc.org | 48df381 | 2013-11-08 15:18:52 +0000 | [diff] [blame] | 357 | bool in_order = IsPacketInOrder(header); |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 358 | rtp_payload_registry_.SetIncomingPayloadType(header); |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 359 | ReceivePacket(packet.data(), packet.size(), header, in_order); |
asapersson@webrtc.org | 1457b47 | 2014-05-26 13:06:04 +0000 | [diff] [blame] | 360 | // Update receive statistics after ReceivePacket. |
| 361 | // Receive statistics will be reset if the payload type changes (make sure |
| 362 | // that the first packet is included in the stats). |
| 363 | rtp_receive_statistics_->IncomingPacket( |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 364 | header, packet.size(), IsPacketRetransmitted(header, in_order)); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 365 | } |
| 366 | |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 367 | int32_t RtpStreamReceiver::RequestKeyFrame() { |
| 368 | return rtp_rtcp_->RequestKeyFrame(); |
| 369 | } |
| 370 | |
brandtr | f7c6d72 | 2016-12-08 08:25:47 -0800 | [diff] [blame] | 371 | bool RtpStreamReceiver::IsUlpfecEnabled() const { |
brandtr | e6f98c7 | 2016-11-11 03:28:30 -0800 | [diff] [blame] | 372 | return config_.rtp.ulpfec.ulpfec_payload_type != -1; |
| 373 | } |
| 374 | |
| 375 | bool RtpStreamReceiver::IsRedEnabled() const { |
| 376 | return config_.rtp.ulpfec.red_payload_type != -1; |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 377 | } |
| 378 | |
| 379 | bool RtpStreamReceiver::IsRetransmissionsEnabled() const { |
| 380 | return config_.rtp.nack.rtp_history_ms > 0; |
| 381 | } |
| 382 | |
| 383 | void RtpStreamReceiver::RequestPacketRetransmit( |
| 384 | const std::vector<uint16_t>& sequence_numbers) { |
| 385 | rtp_rtcp_->SendNack(sequence_numbers); |
| 386 | } |
| 387 | |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 388 | int32_t RtpStreamReceiver::ResendPackets(const uint16_t* sequence_numbers, |
| 389 | uint16_t length) { |
| 390 | return rtp_rtcp_->SendNACK(sequence_numbers, length); |
| 391 | } |
| 392 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 393 | void RtpStreamReceiver::OnReceivedFrame( |
| 394 | std::unique_ptr<video_coding::RtpFrameObject> frame) { |
philipel | bd26ba7 | 2017-01-29 04:04:47 -0800 | [diff] [blame] | 395 | if (!frame->delayed_by_retransmission()) |
| 396 | timing_->IncomingTimestamp(frame->timestamp, clock_->TimeInMilliseconds()); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 397 | reference_finder_->ManageFrame(std::move(frame)); |
| 398 | } |
| 399 | |
| 400 | void RtpStreamReceiver::OnCompleteFrame( |
| 401 | std::unique_ptr<video_coding::FrameObject> frame) { |
| 402 | { |
| 403 | rtc::CritScope lock(&last_seq_num_cs_); |
| 404 | video_coding::RtpFrameObject* rtp_frame = |
| 405 | static_cast<video_coding::RtpFrameObject*>(frame.get()); |
| 406 | last_seq_num_for_pic_id_[rtp_frame->picture_id] = rtp_frame->last_seq_num(); |
| 407 | } |
| 408 | complete_frame_callback_->OnCompleteFrame(std::move(frame)); |
| 409 | } |
| 410 | |
| 411 | void RtpStreamReceiver::OnRttUpdate(int64_t avg_rtt_ms, int64_t max_rtt_ms) { |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 412 | if (nack_module_) |
| 413 | nack_module_->UpdateRtt(max_rtt_ms); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 414 | } |
| 415 | |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 416 | // TODO(nisse): Drop return value. |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 417 | bool RtpStreamReceiver::ReceivePacket(const uint8_t* packet, |
| 418 | size_t packet_length, |
| 419 | const RTPHeader& header, |
| 420 | bool in_order) { |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 421 | if (rtp_payload_registry_.IsEncapsulated(header)) { |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 422 | return ParseAndHandleEncapsulatingHeader(packet, packet_length, header); |
| 423 | } |
| 424 | const uint8_t* payload = packet + header.headerLength; |
pkasting@chromium.org | 4591fbd | 2014-11-20 22:28:14 +0000 | [diff] [blame] | 425 | assert(packet_length >= header.headerLength); |
| 426 | size_t payload_length = packet_length - header.headerLength; |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 427 | PayloadUnion payload_specific; |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 428 | if (!rtp_payload_registry_.GetPayloadSpecifics(header.payloadType, |
| 429 | &payload_specific)) { |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 430 | return false; |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 431 | } |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 432 | return rtp_receiver_->IncomingRtpPacket(header, payload, payload_length, |
| 433 | payload_specific, in_order); |
| 434 | } |
| 435 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 436 | bool RtpStreamReceiver::ParseAndHandleEncapsulatingHeader( |
| 437 | const uint8_t* packet, size_t packet_length, const RTPHeader& header) { |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 438 | if (rtp_payload_registry_.IsRed(header)) { |
| 439 | int8_t ulpfec_pt = rtp_payload_registry_.ulpfec_payload_type(); |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 440 | if (packet[header.headerLength] == ulpfec_pt) { |
asapersson@webrtc.org | 273fbbb | 2015-01-27 12:17:29 +0000 | [diff] [blame] | 441 | rtp_receive_statistics_->FecPacketReceived(header, packet_length); |
Peter Boström | 0b25072 | 2016-04-22 18:23:15 +0200 | [diff] [blame] | 442 | // Notify video_receiver about received FEC packets to avoid NACKing these |
| 443 | // packets. |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 444 | NotifyReceiverOfFecPacket(header); |
| 445 | } |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 446 | if (ulpfec_receiver_->AddReceivedRedPacket(header, packet, packet_length, |
| 447 | ulpfec_pt) != 0) { |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 448 | return false; |
| 449 | } |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 450 | return ulpfec_receiver_->ProcessReceivedFec() == 0; |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 451 | } else if (rtp_payload_registry_.IsRtx(header)) { |
stefan@webrtc.org | 7c6ff2d | 2014-03-19 18:14:52 +0000 | [diff] [blame] | 452 | if (header.headerLength + header.paddingLength == packet_length) { |
| 453 | // This is an empty packet and should be silently dropped before trying to |
| 454 | // parse the RTX header. |
| 455 | return true; |
| 456 | } |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 457 | // Remove the RTX header and parse the original RTP header. |
| 458 | if (packet_length < header.headerLength) |
| 459 | return false; |
pkasting@chromium.org | 4591fbd | 2014-11-20 22:28:14 +0000 | [diff] [blame] | 460 | if (packet_length > sizeof(restored_packet_)) |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 461 | return false; |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 462 | rtc::CritScope lock(&receive_cs_); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 463 | if (restored_packet_in_use_) { |
pbos@webrtc.org | 4e2806d | 2014-05-14 08:02:22 +0000 | [diff] [blame] | 464 | LOG(LS_WARNING) << "Multiple RTX headers detected, dropping packet."; |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 465 | return false; |
| 466 | } |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 467 | if (!rtp_payload_registry_.RestoreOriginalPacket( |
noahric | 65220a7 | 2015-10-14 11:29:49 -0700 | [diff] [blame] | 468 | restored_packet_, packet, &packet_length, rtp_receiver_->SSRC(), |
| 469 | header)) { |
Stefan Holmer | 1088001 | 2016-02-03 13:29:59 +0100 | [diff] [blame] | 470 | LOG(LS_WARNING) << "Incoming RTX packet: Invalid RTP header ssrc: " |
| 471 | << header.ssrc << " payload type: " |
| 472 | << static_cast<int>(header.payloadType); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 473 | return false; |
| 474 | } |
| 475 | restored_packet_in_use_ = true; |
noahric | 65220a7 | 2015-10-14 11:29:49 -0700 | [diff] [blame] | 476 | bool ret = OnRecoveredPacket(restored_packet_, packet_length); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 477 | restored_packet_in_use_ = false; |
| 478 | return ret; |
| 479 | } |
| 480 | return false; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 481 | } |
| 482 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 483 | void RtpStreamReceiver::NotifyReceiverOfFecPacket(const RTPHeader& header) { |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 484 | int8_t last_media_payload_type = |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 485 | rtp_payload_registry_.last_received_media_payload_type(); |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 486 | if (last_media_payload_type < 0) { |
| 487 | LOG(LS_WARNING) << "Failed to get last media payload type."; |
| 488 | return; |
| 489 | } |
| 490 | // Fake an empty media packet. |
| 491 | WebRtcRTPHeader rtp_header = {}; |
| 492 | rtp_header.header = header; |
| 493 | rtp_header.header.payloadType = last_media_payload_type; |
| 494 | rtp_header.header.paddingLength = 0; |
| 495 | PayloadUnion payload_specific; |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 496 | if (!rtp_payload_registry_.GetPayloadSpecifics(last_media_payload_type, |
| 497 | &payload_specific)) { |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 498 | LOG(LS_WARNING) << "Failed to get payload specifics."; |
| 499 | return; |
| 500 | } |
| 501 | rtp_header.type.Video.codec = payload_specific.Video.videoCodecType; |
guoweis@webrtc.org | fdd1057 | 2015-03-12 20:50:57 +0000 | [diff] [blame] | 502 | rtp_header.type.Video.rotation = kVideoRotation_0; |
| 503 | if (header.extension.hasVideoRotation) { |
magjed | 71eb61c | 2016-09-08 03:24:58 -0700 | [diff] [blame] | 504 | rtp_header.type.Video.rotation = header.extension.videoRotation; |
guoweis@webrtc.org | fdd1057 | 2015-03-12 20:50:57 +0000 | [diff] [blame] | 505 | } |
isheriff | 6b4b5f3 | 2016-06-08 00:24:21 -0700 | [diff] [blame] | 506 | rtp_header.type.Video.playout_delay = header.extension.playout_delay; |
| 507 | |
Peter Boström | 74f6e9e | 2016-04-04 17:56:10 +0200 | [diff] [blame] | 508 | OnReceivedPayloadData(nullptr, 0, &rtp_header); |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 509 | } |
| 510 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 511 | bool RtpStreamReceiver::DeliverRtcp(const uint8_t* rtcp_packet, |
| 512 | size_t rtcp_packet_length) { |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 513 | { |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 514 | rtc::CritScope lock(&receive_cs_); |
braveyao@webrtc.org | b6433b7 | 2013-07-26 09:02:46 +0000 | [diff] [blame] | 515 | if (!receiving_) { |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 516 | return false; |
braveyao@webrtc.org | b6433b7 | 2013-07-26 09:02:46 +0000 | [diff] [blame] | 517 | } |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 518 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 519 | |
Per | 83d0910 | 2016-04-15 14:59:13 +0200 | [diff] [blame] | 520 | rtp_rtcp_->IncomingRtcpPacket(rtcp_packet, rtcp_packet_length); |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 521 | |
pkasting@chromium.org | 16825b1 | 2015-01-12 21:51:21 +0000 | [diff] [blame] | 522 | int64_t rtt = 0; |
Per | 83d0910 | 2016-04-15 14:59:13 +0200 | [diff] [blame] | 523 | rtp_rtcp_->RTT(rtp_receiver_->SSRC(), &rtt, nullptr, nullptr, nullptr); |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 524 | if (rtt == 0) { |
| 525 | // Waiting for valid rtt. |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 526 | return true; |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 527 | } |
| 528 | uint32_t ntp_secs = 0; |
| 529 | uint32_t ntp_frac = 0; |
| 530 | uint32_t rtp_timestamp = 0; |
Per | 83d0910 | 2016-04-15 14:59:13 +0200 | [diff] [blame] | 531 | if (rtp_rtcp_->RemoteNTP(&ntp_secs, &ntp_frac, nullptr, nullptr, |
| 532 | &rtp_timestamp) != 0) { |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 533 | // Waiting for RTCP. |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 534 | return true; |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 535 | } |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 536 | ntp_estimator_.UpdateRtcpTimestamp(rtt, ntp_secs, ntp_frac, rtp_timestamp); |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 537 | |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 538 | return true; |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 539 | } |
| 540 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 541 | void RtpStreamReceiver::FrameContinuous(uint16_t picture_id) { |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 542 | if (!nack_module_) |
| 543 | return; |
| 544 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 545 | int seq_num = -1; |
| 546 | { |
| 547 | rtc::CritScope lock(&last_seq_num_cs_); |
| 548 | auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id); |
| 549 | if (seq_num_it != last_seq_num_for_pic_id_.end()) |
| 550 | seq_num = seq_num_it->second; |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 551 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 552 | if (seq_num != -1) |
| 553 | nack_module_->ClearUpTo(seq_num); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 554 | } |
| 555 | |
| 556 | void RtpStreamReceiver::FrameDecoded(uint16_t picture_id) { |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 557 | int seq_num = -1; |
| 558 | { |
| 559 | rtc::CritScope lock(&last_seq_num_cs_); |
| 560 | auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id); |
| 561 | if (seq_num_it != last_seq_num_for_pic_id_.end()) { |
| 562 | seq_num = seq_num_it->second; |
| 563 | last_seq_num_for_pic_id_.erase(last_seq_num_for_pic_id_.begin(), |
| 564 | ++seq_num_it); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 565 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 566 | } |
| 567 | if (seq_num != -1) { |
| 568 | packet_buffer_->ClearTo(seq_num); |
| 569 | reference_finder_->ClearTo(seq_num); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 570 | } |
| 571 | } |
| 572 | |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 573 | void RtpStreamReceiver::SignalNetworkState(NetworkState state) { |
| 574 | rtp_rtcp_->SetRTCPStatus(state == kNetworkUp ? config_.rtp.rtcp_mode |
| 575 | : RtcpMode::kOff); |
| 576 | } |
| 577 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 578 | void RtpStreamReceiver::StartReceive() { |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 579 | rtc::CritScope lock(&receive_cs_); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 580 | receiving_ = true; |
| 581 | } |
| 582 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 583 | void RtpStreamReceiver::StopReceive() { |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 584 | rtc::CritScope lock(&receive_cs_); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 585 | receiving_ = false; |
| 586 | } |
| 587 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 588 | bool RtpStreamReceiver::IsPacketInOrder(const RTPHeader& header) const { |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 589 | StreamStatistician* statistician = |
| 590 | rtp_receive_statistics_->GetStatistician(header.ssrc); |
| 591 | if (!statistician) |
| 592 | return false; |
| 593 | return statistician->IsPacketInOrder(header.sequenceNumber); |
| 594 | } |
| 595 | |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 596 | bool RtpStreamReceiver::IsPacketRetransmitted(const RTPHeader& header, |
| 597 | bool in_order) const { |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 598 | // Retransmissions are handled separately if RTX is enabled. |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 599 | if (rtp_payload_registry_.RtxEnabled()) |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 600 | return false; |
| 601 | StreamStatistician* statistician = |
| 602 | rtp_receive_statistics_->GetStatistician(header.ssrc); |
| 603 | if (!statistician) |
| 604 | return false; |
| 605 | // Check if this is a retransmission. |
pkasting@chromium.org | 16825b1 | 2015-01-12 21:51:21 +0000 | [diff] [blame] | 606 | int64_t min_rtt = 0; |
Per | 83d0910 | 2016-04-15 14:59:13 +0200 | [diff] [blame] | 607 | rtp_rtcp_->RTT(rtp_receiver_->SSRC(), nullptr, nullptr, &min_rtt, nullptr); |
stefan@webrtc.org | 48df381 | 2013-11-08 15:18:52 +0000 | [diff] [blame] | 608 | return !in_order && |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 609 | statistician->IsRetransmitOfOldPacket(header, min_rtt); |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 610 | } |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 611 | |
| 612 | void RtpStreamReceiver::UpdateHistograms() { |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 613 | FecPacketCounter counter = ulpfec_receiver_->GetPacketCounter(); |
asapersson | 0c43f77 | 2016-11-30 01:42:26 -0800 | [diff] [blame] | 614 | if (counter.first_packet_time_ms == -1) |
| 615 | return; |
| 616 | |
| 617 | int64_t elapsed_sec = |
| 618 | (clock_->TimeInMilliseconds() - counter.first_packet_time_ms) / 1000; |
| 619 | if (elapsed_sec < metrics::kMinRunTimeInSeconds) |
| 620 | return; |
| 621 | |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 622 | if (counter.num_packets > 0) { |
asapersson | 1d02d3e | 2016-09-09 22:40:25 -0700 | [diff] [blame] | 623 | RTC_HISTOGRAM_PERCENTAGE( |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 624 | "WebRTC.Video.ReceivedFecPacketsInPercent", |
| 625 | static_cast<int>(counter.num_fec_packets * 100 / counter.num_packets)); |
| 626 | } |
| 627 | if (counter.num_fec_packets > 0) { |
asapersson | 1d02d3e | 2016-09-09 22:40:25 -0700 | [diff] [blame] | 628 | RTC_HISTOGRAM_PERCENTAGE("WebRTC.Video.RecoveredMediaPacketsInPercentOfFec", |
| 629 | static_cast<int>(counter.num_recovered_packets * |
| 630 | 100 / counter.num_fec_packets)); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 631 | } |
| 632 | } |
| 633 | |
| 634 | void RtpStreamReceiver::EnableReceiveRtpHeaderExtension( |
| 635 | const std::string& extension, int id) { |
| 636 | // One-byte-extension local identifiers are in the range 1-14 inclusive. |
| 637 | RTC_DCHECK_GE(id, 1); |
| 638 | RTC_DCHECK_LE(id, 14); |
| 639 | RTC_DCHECK(RtpExtension::IsSupportedForVideo(extension)); |
| 640 | RTC_CHECK(rtp_header_parser_->RegisterRtpHeaderExtension( |
| 641 | StringToRtpExtensionType(extension), id)); |
| 642 | } |
| 643 | |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 644 | void RtpStreamReceiver::InsertSpsPpsIntoTracker(uint8_t payload_type) { |
| 645 | auto codec_params_it = pt_codec_params_.find(payload_type); |
| 646 | if (codec_params_it == pt_codec_params_.end()) |
| 647 | return; |
| 648 | |
| 649 | LOG(LS_INFO) << "Found out of band supplied codec parameters for" |
johan | 62d02c3 | 2017-01-24 04:38:27 -0800 | [diff] [blame] | 650 | << " payload type: " << static_cast<int>(payload_type); |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 651 | |
| 652 | H264SpropParameterSets sprop_decoder; |
| 653 | auto sprop_base64_it = |
| 654 | codec_params_it->second.find(cricket::kH264FmtpSpropParameterSets); |
| 655 | |
| 656 | if (sprop_base64_it == codec_params_it->second.end()) |
| 657 | return; |
| 658 | |
johan | 62d02c3 | 2017-01-24 04:38:27 -0800 | [diff] [blame] | 659 | if (!sprop_decoder.DecodeSprop(sprop_base64_it->second.c_str())) |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 660 | return; |
| 661 | |
johan | d2b092f | 2017-01-24 02:38:17 -0800 | [diff] [blame] | 662 | tracker_.InsertSpsPpsNalus(sprop_decoder.sps_nalu(), |
| 663 | sprop_decoder.pps_nalu()); |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 664 | } |
| 665 | |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 666 | } // namespace webrtc |