blob: 4e5138513964aeb501e1b1e3e6a8ccf10a449fc9 [file] [log] [blame]
niklase@google.com470e71d2011-07-07 08:21:25 +00001/*
leozwang@webrtc.org39e96592012-03-01 18:22:48 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
niklase@google.com470e71d2011-07-07 08:21:25 +00003 *
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
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#include "video/rtp_video_stream_receiver.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000012
philipelfd5a20f2016-11-15 00:57:57 -080013#include <utility>
ilnik04f4d122017-06-19 07:18:55 -070014#include <vector>
mflodman@webrtc.org4fd55272013-02-06 17:46:39 +000015
Steve Antonbd631a02019-03-28 10:51:27 -070016#include "absl/algorithm/container.h"
Niels Möller2ff1f2a2018-08-09 16:16:34 +020017#include "absl/memory/memory.h"
Steve Anton10542f22019-01-11 09:11:00 -080018#include "media/base/media_constants.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020019#include "modules/pacing/packet_router.h"
20#include "modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h"
21#include "modules/rtp_rtcp/include/receive_statistics.h"
22#include "modules/rtp_rtcp/include/rtp_cvo.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020023#include "modules/rtp_rtcp/include/rtp_rtcp.h"
24#include "modules/rtp_rtcp/include/ulpfec_receiver.h"
Niels Möller2ff1f2a2018-08-09 16:16:34 +020025#include "modules/rtp_rtcp/source/rtp_format.h"
philipelb3e42a42018-09-13 10:57:14 +020026#include "modules/rtp_rtcp/source/rtp_generic_frame_descriptor_extension.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020027#include "modules/rtp_rtcp/source/rtp_header_extensions.h"
28#include "modules/rtp_rtcp/source/rtp_packet_received.h"
Niels Möller2ff1f2a2018-08-09 16:16:34 +020029#include "modules/rtp_rtcp/source/rtp_rtcp_config.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020030#include "modules/video_coding/frame_object.h"
31#include "modules/video_coding/h264_sprop_parameter_sets.h"
32#include "modules/video_coding/h264_sps_pps_tracker.h"
Ilya Nikolaevskiy8643b782018-06-07 16:15:40 +020033#include "modules/video_coding/nack_module.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020034#include "modules/video_coding/packet_buffer.h"
35#include "modules/video_coding/video_coding_impl.h"
36#include "rtc_base/checks.h"
37#include "rtc_base/location.h"
38#include "rtc_base/logging.h"
Jonas Olsson366a50c2018-09-06 13:41:30 +020039#include "rtc_base/strings/string_builder.h"
Karl Wiberg80ba3332018-02-05 10:33:35 +010040#include "rtc_base/system/fallthrough.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020041#include "system_wrappers/include/field_trial.h"
42#include "system_wrappers/include/metrics.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020043#include "video/receive_statistics_proxy.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000044
45namespace webrtc {
46
philipelfd5a20f2016-11-15 00:57:57 -080047namespace {
philipel3bf97cf2017-08-10 18:10:59 +020048// TODO(philipel): Change kPacketBufferStartSize back to 32 in M63 see:
49// crbug.com/752886
50constexpr int kPacketBufferStartSize = 512;
Johannes Kron201596f2018-10-22 14:33:39 +020051constexpr int kPacketBufferMaxSize = 2048;
Yves Gerey665174f2018-06-19 15:03:05 +020052} // namespace
philipelfd5a20f2016-11-15 00:57:57 -080053
mflodmanc0e58a32016-04-25 01:26:26 -070054std::unique_ptr<RtpRtcp> CreateRtpRtcpModule(
Sebastian Jansson8026d602019-03-04 19:39:01 +010055 Clock* clock,
mflodmanc0e58a32016-04-25 01:26:26 -070056 ReceiveStatistics* receive_statistics,
57 Transport* outgoing_transport,
58 RtcpRttStats* rtt_stats,
Niels Möller60f4e292019-05-20 11:06:33 +020059 RtcpPacketTypeCounterObserver* rtcp_packet_type_counter_observer) {
mflodmanc0e58a32016-04-25 01:26:26 -070060 RtpRtcp::Configuration configuration;
Sebastian Jansson8026d602019-03-04 19:39:01 +010061 configuration.clock = clock;
mflodmanc0e58a32016-04-25 01:26:26 -070062 configuration.audio = false;
63 configuration.receiver_only = true;
64 configuration.receive_statistics = receive_statistics;
65 configuration.outgoing_transport = outgoing_transport;
mflodmanc0e58a32016-04-25 01:26:26 -070066 configuration.rtt_stats = rtt_stats;
67 configuration.rtcp_packet_type_counter_observer =
68 rtcp_packet_type_counter_observer;
mflodmanc0e58a32016-04-25 01:26:26 -070069
Danil Chapovalovc44f6cc2019-03-06 11:31:09 +010070 std::unique_ptr<RtpRtcp> rtp_rtcp = RtpRtcp::Create(configuration);
mflodmanc0e58a32016-04-25 01:26:26 -070071 rtp_rtcp->SetRTCPStatus(RtcpMode::kCompound);
72
73 return rtp_rtcp;
74}
75
stefan@webrtc.orgeb24b042014-10-14 11:40:13 +000076static const int kPacketLogIntervalMs = 10000;
77
nisseb1f2ff92017-06-09 04:01:55 -070078RtpVideoStreamReceiver::RtpVideoStreamReceiver(
Sebastian Jansson8026d602019-03-04 19:39:01 +010079 Clock* clock,
mflodmanfa666592016-04-28 23:15:33 -070080 Transport* transport,
81 RtcpRttStats* rtt_stats,
mflodmancfc8e3b2016-05-03 21:22:04 -070082 PacketRouter* packet_router,
Tommi733b5472016-06-10 17:58:01 +020083 const VideoReceiveStream::Config* config,
nisseca5706d2017-09-11 02:32:16 -070084 ReceiveStatistics* rtp_receive_statistics,
mflodmandc7d0d22016-05-06 05:32:22 -070085 ReceiveStatisticsProxy* receive_stats_proxy,
Erik Språng737336d2016-07-29 12:59:36 +020086 ProcessThread* process_thread,
philipelfd5a20f2016-11-15 00:57:57 -080087 NackSender* nack_sender,
Niels Möller2f5554d2019-05-29 13:35:14 +020088 KeyFrameRequestSender* keyframe_request_sender,
Benjamin Wright192eeec2018-10-17 17:27:25 -070089 video_coding::OnCompleteFrameCallback* complete_frame_callback,
90 rtc::scoped_refptr<FrameDecryptorInterface> frame_decryptor)
Sebastian Jansson8026d602019-03-04 19:39:01 +010091 : clock_(clock),
Tommi733b5472016-06-10 17:58:01 +020092 config_(*config),
mflodmanc0e58a32016-04-25 01:26:26 -070093 packet_router_(packet_router),
mflodmandc7d0d22016-05-06 05:32:22 -070094 process_thread_(process_thread),
Sebastian Jansson8026d602019-03-04 19:39:01 +010095 ntp_estimator_(clock),
Niels Möllerb0573bc2017-09-25 10:47:00 +020096 rtp_header_extensions_(config_.rtp.extensions),
nisseca5706d2017-09-11 02:32:16 -070097 rtp_receive_statistics_(rtp_receive_statistics),
brandtrd726a3f2017-06-29 02:45:35 -070098 ulpfec_receiver_(UlpfecReceiver::Create(config->rtp.remote_ssrc, this)),
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +000099 receiving_(false),
mflodmanc0e58a32016-04-25 01:26:26 -0700100 last_packet_log_ms_(-1),
Niels Möller2f5554d2019-05-29 13:35:14 +0200101 rtp_rtcp_(CreateRtpRtcpModule(clock,
102 rtp_receive_statistics_,
103 transport,
104 rtt_stats,
105 receive_stats_proxy)),
philipelfd5a20f2016-11-15 00:57:57 -0800106 complete_frame_callback_(complete_frame_callback),
Niels Möller2f5554d2019-05-29 13:35:14 +0200107 keyframe_request_sender_(keyframe_request_sender),
Benjamin Wright52426ed2019-03-01 11:01:59 -0800108 has_received_frame_(false),
109 frames_decryptable_(false) {
eladalon822ff2b2017-08-01 06:30:28 -0700110 constexpr bool remb_candidate = true;
Niels Möller60f4e292019-05-20 11:06:33 +0200111 if (packet_router_)
112 packet_router_->AddReceiveRtpModule(rtp_rtcp_.get(), remb_candidate);
mflodmancfc8e3b2016-05-03 21:22:04 -0700113
Tommi733b5472016-06-10 17:58:01 +0200114 RTC_DCHECK(config_.rtp.rtcp_mode != RtcpMode::kOff)
mflodmancfc8e3b2016-05-03 21:22:04 -0700115 << "A stream should not be configured with RTCP disabled. This value is "
116 "reserved for internal usage.";
mflodmandc7d0d22016-05-06 05:32:22 -0700117 RTC_DCHECK(config_.rtp.remote_ssrc != 0);
118 // TODO(pbos): What's an appropriate local_ssrc for receive-only streams?
119 RTC_DCHECK(config_.rtp.local_ssrc != 0);
120 RTC_DCHECK(config_.rtp.remote_ssrc != config_.rtp.local_ssrc);
121
Tommi733b5472016-06-10 17:58:01 +0200122 rtp_rtcp_->SetRTCPStatus(config_.rtp.rtcp_mode);
123 rtp_rtcp_->SetSSRC(config_.rtp.local_ssrc);
stefanb4ab3812017-06-09 06:12:11 -0700124 rtp_rtcp_->SetRemoteSSRC(config_.rtp.remote_ssrc);
mflodmanc0e58a32016-04-25 01:26:26 -0700125 rtp_rtcp_->SetKeyFrameRequestMethod(kKeyFrameReqPliRtcp);
mflodmandc7d0d22016-05-06 05:32:22 -0700126
mflodmancfc8e3b2016-05-03 21:22:04 -0700127 static const int kMaxPacketAgeToNack = 450;
Tommi733b5472016-06-10 17:58:01 +0200128 const int max_reordering_threshold = (config_.rtp.nack.rtp_history_ms > 0)
129 ? kMaxPacketAgeToNack
130 : kDefaultMaxReorderingThreshold;
Niels Möller87da1092019-05-24 14:04:28 +0200131 rtp_receive_statistics_->SetMaxReorderingThreshold(config_.rtp.remote_ssrc,
132 max_reordering_threshold);
133 // TODO(nisse): For historic reasons, we applied the above
134 // max_reordering_threshold also for RTX stats, which makes little sense since
135 // we don't NACK rtx packets. Consider deleting the below block, and rely on
136 // the default threshold.
137 if (config_.rtp.rtx_ssrc) {
138 rtp_receive_statistics_->SetMaxReorderingThreshold(
139 config_.rtp.rtx_ssrc, max_reordering_threshold);
140 }
Tommi733b5472016-06-10 17:58:01 +0200141 if (config_.rtp.rtcp_xr.receiver_reference_time_report)
mflodmandc7d0d22016-05-06 05:32:22 -0700142 rtp_rtcp_->SetRtcpXrRrtrStatus(true);
143
144 // Stats callback for CNAME changes.
145 rtp_rtcp_->RegisterRtcpStatisticsCallback(receive_stats_proxy);
146
tommidea489f2017-03-03 03:20:24 -0800147 process_thread_->RegisterModule(rtp_rtcp_.get(), RTC_FROM_HERE);
philipelfd5a20f2016-11-15 00:57:57 -0800148
Elad Alonfadb1812019-05-24 13:40:02 +0200149 if (config_.rtp.lntf.enabled) {
Elad Alon7d6a4c02019-02-25 13:00:51 +0100150 loss_notification_controller_ =
Niels Möller479c0552019-05-23 16:57:01 +0200151 absl::make_unique<LossNotificationController>(this, this);
Elad Alonca2c4302019-05-27 22:43:10 +0200152 }
153
154 if (config_.rtp.nack.rtp_history_ms != 0) {
Niels Möller479c0552019-05-23 16:57:01 +0200155 nack_module_ = absl::make_unique<NackModule>(clock_, nack_sender, this);
tommidea489f2017-03-03 03:20:24 -0800156 process_thread_->RegisterModule(nack_module_.get(), RTC_FROM_HERE);
tommif284b7f2017-02-27 01:59:36 -0800157 }
philipelfd5a20f2016-11-15 00:57:57 -0800158
Johannes Kron487e6942018-10-24 00:53:39 +0200159 // The group here must be a positive power of 2, in which case that is used as
160 // size. All other values shall result in the default value being used.
Johannes Kron201596f2018-10-22 14:33:39 +0200161 const std::string group_name =
162 webrtc::field_trial::FindFullName("WebRTC-PacketBufferMaxSize");
163 int packet_buffer_max_size = kPacketBufferMaxSize;
164 if (!group_name.empty() &&
165 (sscanf(group_name.c_str(), "%d", &packet_buffer_max_size) != 1 ||
Johannes Kron487e6942018-10-24 00:53:39 +0200166 packet_buffer_max_size <= 0 ||
167 // Verify that the number is a positive power of 2.
168 (packet_buffer_max_size & (packet_buffer_max_size - 1)) != 0)) {
Johannes Kron201596f2018-10-22 14:33:39 +0200169 RTC_LOG(LS_WARNING) << "Invalid packet buffer max size: " << group_name;
170 packet_buffer_max_size = kPacketBufferMaxSize;
171 }
172
philipela45102f2017-02-22 05:30:39 -0800173 packet_buffer_ = video_coding::PacketBuffer::Create(
Johannes Kron201596f2018-10-22 14:33:39 +0200174 clock_, kPacketBufferStartSize, packet_buffer_max_size, this);
Elad Alona8f54612018-11-06 11:21:25 +0100175 reference_finder_ =
176 absl::make_unique<video_coding::RtpFrameReferenceFinder>(this);
Benjamin Wrighta5564482019-04-03 10:44:18 -0700177
Benjamin Wright00765292018-11-30 16:18:26 -0800178 // Only construct the encrypted receiver if frame encryption is enabled.
Benjamin Wrighta5564482019-04-03 10:44:18 -0700179 if (config_.crypto_options.sframe.require_frame_encryption) {
Benjamin Wright00765292018-11-30 16:18:26 -0800180 buffered_frame_decryptor_ =
Benjamin Wrighta5564482019-04-03 10:44:18 -0700181 absl::make_unique<BufferedFrameDecryptor>(this, this);
182 if (frame_decryptor != nullptr) {
183 buffered_frame_decryptor_->SetFrameDecryptor(std::move(frame_decryptor));
184 }
Benjamin Wright00765292018-11-30 16:18:26 -0800185 }
mflodmanc0e58a32016-04-25 01:26:26 -0700186}
niklase@google.com470e71d2011-07-07 08:21:25 +0000187
nisseb1f2ff92017-06-09 04:01:55 -0700188RtpVideoStreamReceiver::~RtpVideoStreamReceiver() {
eladalonc0d481a2017-08-02 07:39:07 -0700189 RTC_DCHECK(secondary_sinks_.empty());
190
tommif284b7f2017-02-27 01:59:36 -0800191 if (nack_module_) {
192 process_thread_->DeRegisterModule(nack_module_.get());
193 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000194
tommif284b7f2017-02-27 01:59:36 -0800195 process_thread_->DeRegisterModule(rtp_rtcp_.get());
philipelfd5a20f2016-11-15 00:57:57 -0800196
Niels Möller60f4e292019-05-20 11:06:33 +0200197 if (packet_router_)
198 packet_router_->RemoveReceiveRtpModule(rtp_rtcp_.get());
mflodmandc7d0d22016-05-06 05:32:22 -0700199 UpdateHistograms();
asapersson@webrtc.org0800db72015-01-15 07:40:20 +0000200}
201
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200202void RtpVideoStreamReceiver::AddReceiveCodec(
philipel022b54e2016-12-20 04:15:59 -0800203 const VideoCodec& video_codec,
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +0200204 const std::map<std::string, std::string>& codec_params,
205 bool raw_payload) {
206 absl::optional<VideoCodecType> video_type;
207 if (!raw_payload) {
208 video_type = video_codec.codecType;
209 }
210 payload_type_map_.emplace(video_codec.plType, video_type);
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200211 pt_codec_params_.emplace(video_codec.plType, codec_params);
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000212}
213
Niels Möllerdf9e9ae2018-07-31 08:29:53 +0200214absl::optional<Syncable::Info> RtpVideoStreamReceiver::GetSyncInfo() const {
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200215 Syncable::Info info;
Niels Möllerdf9e9ae2018-07-31 08:29:53 +0200216 if (rtp_rtcp_->RemoteNTP(&info.capture_time_ntp_secs,
217 &info.capture_time_ntp_frac, nullptr, nullptr,
218 &info.capture_time_source_clock) != 0) {
219 return absl::nullopt;
220 }
Niels Möllerb0d4b412018-08-28 13:58:15 +0200221 {
Jonas Oreland49ac5952018-09-26 16:04:32 +0200222 rtc::CritScope lock(&rtp_sources_lock_);
Niels Möllerb0d4b412018-08-28 13:58:15 +0200223 if (!last_received_rtp_timestamp_ || !last_received_rtp_system_time_ms_) {
224 return absl::nullopt;
225 }
226 info.latest_received_capture_timestamp = *last_received_rtp_timestamp_;
227 info.latest_receive_time_ms = *last_received_rtp_system_time_ms_;
228 }
Niels Möllerdf9e9ae2018-07-31 08:29:53 +0200229
230 // Leaves info.current_delay_ms uninitialized.
231 return info;
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000232}
233
nisseb1f2ff92017-06-09 04:01:55 -0700234int32_t RtpVideoStreamReceiver::OnReceivedPayloadData(
mflodmanfa666592016-04-28 23:15:33 -0700235 const uint8_t* payload_data,
Peter Boström02083222016-06-14 12:52:54 +0200236 size_t payload_size,
Niels Möller125b5d62019-03-11 16:11:07 +0100237 const RTPHeader& rtp_header,
238 const RTPVideoHeader& video_header,
Ying Wangb32bb952018-10-31 10:12:27 +0100239 const absl::optional<RtpGenericFrameDescriptor>& generic_descriptor,
240 bool is_recovered) {
Niels Möller125b5d62019-03-11 16:11:07 +0100241 VCMPacket packet(payload_data, payload_size, rtp_header, video_header,
Niels Möllerabbc50e2019-04-24 09:41:16 +0200242 ntp_estimator_.Estimate(rtp_header.timestamp));
Elad Alon7d6a4c02019-02-25 13:00:51 +0100243 packet.generic_descriptor = generic_descriptor;
244
Elad Alonca2c4302019-05-27 22:43:10 +0200245 if (loss_notification_controller_) {
246 if (is_recovered) {
247 // TODO(bugs.webrtc.org/10336): Implement support for reordering.
248 RTC_LOG(LS_WARNING)
249 << "LossNotificationController does not support reordering.";
250 } else {
251 // TODO(bugs.webrtc.org/10336): Coordinate with |nack_module_| to make
252 // sure that only a single RTCP packet is produced if both a LNTF as
253 // well as a NACK (or key frame request) should be issued.
254 loss_notification_controller_->OnReceivedPacket(packet);
255 }
256 }
257
Niels Möller8dad9b42018-08-22 10:36:35 +0200258 if (nack_module_) {
Niels Möllerabbc50e2019-04-24 09:41:16 +0200259 const bool is_keyframe =
260 video_header.is_first_packet_in_frame &&
261 video_header.frame_type == VideoFrameType::kVideoFrameKey;
Niels Möller8dad9b42018-08-22 10:36:35 +0200262
263 packet.timesNacked = nack_module_->OnReceivedPacket(
Niels Möller125b5d62019-03-11 16:11:07 +0100264 rtp_header.sequenceNumber, is_keyframe, is_recovered);
Niels Möller8dad9b42018-08-22 10:36:35 +0200265 } else {
266 packet.timesNacked = -1;
267 }
ilnik04f4d122017-06-19 07:18:55 -0700268 packet.receive_time_ms = clock_->TimeInMilliseconds();
philipelfd5a20f2016-11-15 00:57:57 -0800269
philipel54ca9192017-03-21 05:45:18 -0700270 if (packet.sizeBytes == 0) {
Niels Möllerbc010472018-03-23 13:22:29 +0100271 NotifyReceiverOfEmptyPacket(packet.seqNum);
philipel54ca9192017-03-21 05:45:18 -0700272 return 0;
273 }
274
Niels Möllerd5e02f02019-02-20 13:12:21 +0100275 if (packet.codec() == kVideoCodecH264) {
philipela45102f2017-02-22 05:30:39 -0800276 // Only when we start to receive packets will we know what payload type
277 // that will be used. When we know the payload type insert the correct
278 // sps/pps into the tracker.
279 if (packet.payloadType != last_payload_type_) {
280 last_payload_type_ = packet.payloadType;
281 InsertSpsPpsIntoTracker(packet.payloadType);
philipelfd5a20f2016-11-15 00:57:57 -0800282 }
283
philipela45102f2017-02-22 05:30:39 -0800284 switch (tracker_.CopyAndFixBitstream(&packet)) {
285 case video_coding::H264SpsPpsTracker::kRequestKeyframe:
Niels Möller479c0552019-05-23 16:57:01 +0200286 RequestKeyFrame();
Karl Wiberg80ba3332018-02-05 10:33:35 +0100287 RTC_FALLTHROUGH();
philipela45102f2017-02-22 05:30:39 -0800288 case video_coding::H264SpsPpsTracker::kDrop:
289 return 0;
290 case video_coding::H264SpsPpsTracker::kInsert:
291 break;
292 }
293
philipelfd5a20f2016-11-15 00:57:57 -0800294 } else {
philipela45102f2017-02-22 05:30:39 -0800295 uint8_t* data = new uint8_t[packet.sizeBytes];
296 memcpy(data, packet.dataPtr, packet.sizeBytes);
297 packet.dataPtr = data;
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000298 }
philipela45102f2017-02-22 05:30:39 -0800299
300 packet_buffer_->InsertPacket(&packet);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000301 return 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000302}
303
nisseb1f2ff92017-06-09 04:01:55 -0700304void RtpVideoStreamReceiver::OnRecoveredPacket(const uint8_t* rtp_packet,
305 size_t rtp_packet_length) {
Niels Möllerb0573bc2017-09-25 10:47:00 +0200306 RtpPacketReceived packet;
307 if (!packet.Parse(rtp_packet, rtp_packet_length))
nisse30e89312017-05-29 08:16:37 -0700308 return;
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200309 if (packet.PayloadType() == config_.rtp.red_payload_type) {
310 RTC_LOG(LS_WARNING) << "Discarding recovered packet with RED encapsulation";
311 return;
312 }
313
Niels Möllerb0573bc2017-09-25 10:47:00 +0200314 packet.IdentifyExtensions(rtp_header_extensions_);
315 packet.set_payload_type_frequency(kVideoPayloadTypeFrequency);
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200316 // TODO(nisse): UlpfecReceiverImpl::ProcessReceivedFec passes both
317 // original (decapsulated) media packets and recovered packets to
318 // this callback. We need a way to distinguish, for setting
319 // packet.recovered() correctly. Ideally, move RED decapsulation out
320 // of the Ulpfec implementation.
Niels Möllerb0573bc2017-09-25 10:47:00 +0200321
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200322 ReceivePacket(packet);
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000323}
324
nissed2ef3142017-05-11 08:00:58 -0700325// This method handles both regular RTP packets and packets recovered
326// via FlexFEC.
nisseb1f2ff92017-06-09 04:01:55 -0700327void RtpVideoStreamReceiver::OnRtpPacket(const RtpPacketReceived& packet) {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200328 RTC_DCHECK_RUN_ON(&worker_task_checker_);
eladalonc0d481a2017-08-02 07:39:07 -0700329
eladalon8b073052017-08-25 00:49:08 -0700330 if (!receiving_) {
331 return;
332 }
solenberg@webrtc.orgfc320462014-02-11 15:27:49 +0000333
eladalon8b073052017-08-25 00:49:08 -0700334 if (!packet.recovered()) {
Jonas Oreland49ac5952018-09-26 16:04:32 +0200335 // TODO(nisse): Exclude out-of-order packets?
eladalon8b073052017-08-25 00:49:08 -0700336 int64_t now_ms = clock_->TimeInMilliseconds();
Niels Möllerb0d4b412018-08-28 13:58:15 +0200337 {
Jonas Oreland49ac5952018-09-26 16:04:32 +0200338 rtc::CritScope cs(&rtp_sources_lock_);
Niels Möllerb0d4b412018-08-28 13:58:15 +0200339 last_received_rtp_timestamp_ = packet.Timestamp();
340 last_received_rtp_system_time_ms_ = now_ms;
Jonas Oreland49ac5952018-09-26 16:04:32 +0200341
342 std::vector<uint32_t> csrcs = packet.Csrcs();
Jonas Oreland967f7d52018-11-06 07:35:06 +0100343 contributing_sources_.Update(now_ms, csrcs,
Johannes Kronb5d91832019-05-21 13:19:22 +0200344 /* audio level */ absl::nullopt,
345 packet.Timestamp());
Niels Möllerb0d4b412018-08-28 13:58:15 +0200346 }
eladalon8b073052017-08-25 00:49:08 -0700347 // Periodically log the RTP header of incoming packets.
348 if (now_ms - last_packet_log_ms_ > kPacketLogIntervalMs) {
Jonas Olsson366a50c2018-09-06 13:41:30 +0200349 rtc::StringBuilder ss;
eladalon8b073052017-08-25 00:49:08 -0700350 ss << "Packet received on SSRC: " << packet.Ssrc()
351 << " with payload type: " << static_cast<int>(packet.PayloadType())
352 << ", timestamp: " << packet.Timestamp()
353 << ", sequence number: " << packet.SequenceNumber()
354 << ", arrival time: " << packet.arrival_time_ms();
355 int32_t time_offset;
356 if (packet.GetExtension<TransmissionOffset>(&time_offset)) {
357 ss << ", toffset: " << time_offset;
nisse38cc1d62017-02-13 05:59:46 -0800358 }
eladalon8b073052017-08-25 00:49:08 -0700359 uint32_t send_time;
360 if (packet.GetExtension<AbsoluteSendTime>(&send_time)) {
361 ss << ", abs send time: " << send_time;
362 }
Mirko Bonadei675513b2017-11-09 11:09:25 +0100363 RTC_LOG(LS_INFO) << ss.str();
eladalon8b073052017-08-25 00:49:08 -0700364 last_packet_log_ms_ = now_ms;
stefan@webrtc.orgeb24b042014-10-14 11:40:13 +0000365 }
366 }
wu@webrtc.orga9890802013-12-13 00:21:03 +0000367
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200368 ReceivePacket(packet);
nisse38cc1d62017-02-13 05:59:46 -0800369
asapersson@webrtc.org1457b472014-05-26 13:06:04 +0000370 // Update receive statistics after ReceivePacket.
371 // Receive statistics will be reset if the payload type changes (make sure
372 // that the first packet is included in the stats).
nissed2ef3142017-05-11 08:00:58 -0700373 if (!packet.recovered()) {
Niels Möller1f3206c2018-09-14 08:26:32 +0200374 rtp_receive_statistics_->OnRtpPacket(packet);
nissed2ef3142017-05-11 08:00:58 -0700375 }
eladalonc0d481a2017-08-02 07:39:07 -0700376
377 for (RtpPacketSinkInterface* secondary_sink : secondary_sinks_) {
378 secondary_sink->OnRtpPacket(packet);
379 }
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000380}
381
Niels Möller41684372019-03-25 15:51:03 +0100382void RtpVideoStreamReceiver::RequestKeyFrame() {
Niels Möller2f5554d2019-05-29 13:35:14 +0200383 if (keyframe_request_sender_) {
384 keyframe_request_sender_->RequestKeyFrame();
385 } else {
386 rtp_rtcp_->RequestKeyFrame();
387 }
mflodmancfc8e3b2016-05-03 21:22:04 -0700388}
389
Elad Alon7d6a4c02019-02-25 13:00:51 +0100390void RtpVideoStreamReceiver::SendLossNotification(
391 uint16_t last_decoded_seq_num,
392 uint16_t last_received_seq_num,
393 bool decodability_flag) {
Elad Alonfadb1812019-05-24 13:40:02 +0200394 RTC_DCHECK(config_.rtp.lntf.enabled);
Elad Alon7d6a4c02019-02-25 13:00:51 +0100395 rtp_rtcp_->SendLossNotification(last_decoded_seq_num, last_received_seq_num,
396 decodability_flag);
397}
398
nisseb1f2ff92017-06-09 04:01:55 -0700399bool RtpVideoStreamReceiver::IsUlpfecEnabled() const {
nisse3b3622f2017-09-26 02:49:21 -0700400 return config_.rtp.ulpfec_payload_type != -1;
brandtre6f98c72016-11-11 03:28:30 -0800401}
402
nisseb1f2ff92017-06-09 04:01:55 -0700403bool RtpVideoStreamReceiver::IsRetransmissionsEnabled() const {
mflodmandc7d0d22016-05-06 05:32:22 -0700404 return config_.rtp.nack.rtp_history_ms > 0;
405}
406
nisseb1f2ff92017-06-09 04:01:55 -0700407void RtpVideoStreamReceiver::RequestPacketRetransmit(
mflodmandc7d0d22016-05-06 05:32:22 -0700408 const std::vector<uint16_t>& sequence_numbers) {
409 rtp_rtcp_->SendNack(sequence_numbers);
410}
411
Benjamin Wright52426ed2019-03-01 11:01:59 -0800412bool RtpVideoStreamReceiver::IsDecryptable() const {
413 return frames_decryptable_.load();
414}
415
Elad Alonb4643ad2019-02-22 11:19:50 +0100416void RtpVideoStreamReceiver::OnAssembledFrame(
philipelfd5a20f2016-11-15 00:57:57 -0800417 std::unique_ptr<video_coding::RtpFrameObject> frame) {
Benjamin Wright00765292018-11-30 16:18:26 -0800418 RTC_DCHECK_RUN_ON(&network_tc_);
Elad Alon7d6a4c02019-02-25 13:00:51 +0100419 RTC_DCHECK(frame);
420
421 absl::optional<RtpGenericFrameDescriptor> descriptor =
422 frame->GetGenericFrameDescriptor();
423
424 if (loss_notification_controller_ && descriptor) {
425 loss_notification_controller_->OnAssembledFrame(
426 frame->first_seq_num(), descriptor->FrameId(),
427 descriptor->Discardable().value_or(false),
428 descriptor->FrameDependenciesDiffs());
Elad Alonca2c4302019-05-27 22:43:10 +0200429 }
430
431 // Request a key frame as soon as possible.
432 if (!has_received_frame_) {
433 has_received_frame_ = true;
Niels Möller8f7ce222019-03-21 15:43:58 +0100434 if (frame->FrameType() != VideoFrameType::kVideoFrameKey) {
Elad Alonca2c4302019-05-27 22:43:10 +0200435 // TODO(bugs.webrtc.org/10336): Allow the sender to ignore these messages
436 // if it is relying on LNTF alone.
Niels Möller479c0552019-05-23 16:57:01 +0200437 RequestKeyFrame();
Benjamin Wright39feabe2018-10-22 13:33:09 -0700438 }
439 }
Elad Alon7d6a4c02019-02-25 13:00:51 +0100440
Benjamin Wright00765292018-11-30 16:18:26 -0800441 if (buffered_frame_decryptor_ == nullptr) {
442 reference_finder_->ManageFrame(std::move(frame));
443 } else {
444 buffered_frame_decryptor_->ManageEncryptedFrame(std::move(frame));
Benjamin Wright192eeec2018-10-17 17:27:25 -0700445 }
philipelfd5a20f2016-11-15 00:57:57 -0800446}
447
nisseb1f2ff92017-06-09 04:01:55 -0700448void RtpVideoStreamReceiver::OnCompleteFrame(
philipele7c891f2018-02-22 14:35:06 +0100449 std::unique_ptr<video_coding::EncodedFrame> frame) {
philipelfd5a20f2016-11-15 00:57:57 -0800450 {
451 rtc::CritScope lock(&last_seq_num_cs_);
452 video_coding::RtpFrameObject* rtp_frame =
453 static_cast<video_coding::RtpFrameObject*>(frame.get());
philipel0fa82a62018-03-19 15:34:53 +0100454 last_seq_num_for_pic_id_[rtp_frame->id.picture_id] =
455 rtp_frame->last_seq_num();
philipelfd5a20f2016-11-15 00:57:57 -0800456 }
457 complete_frame_callback_->OnCompleteFrame(std::move(frame));
458}
459
Benjamin Wright00765292018-11-30 16:18:26 -0800460void RtpVideoStreamReceiver::OnDecryptedFrame(
461 std::unique_ptr<video_coding::RtpFrameObject> frame) {
462 reference_finder_->ManageFrame(std::move(frame));
463}
464
Benjamin Wright2af5dcb2019-04-09 20:08:41 +0000465void RtpVideoStreamReceiver::OnDecryptionStatusChange(
466 FrameDecryptorInterface::Status status) {
467 frames_decryptable_.store(
468 (status == FrameDecryptorInterface::Status::kOk) ||
469 (status == FrameDecryptorInterface::Status::kRecoverable));
Benjamin Wright52426ed2019-03-01 11:01:59 -0800470}
471
Benjamin Wrighta5564482019-04-03 10:44:18 -0700472void RtpVideoStreamReceiver::SetFrameDecryptor(
473 rtc::scoped_refptr<FrameDecryptorInterface> frame_decryptor) {
474 RTC_DCHECK_RUN_ON(&network_tc_);
475 if (buffered_frame_decryptor_ == nullptr) {
476 buffered_frame_decryptor_ =
477 absl::make_unique<BufferedFrameDecryptor>(this, this);
478 }
479 buffered_frame_decryptor_->SetFrameDecryptor(std::move(frame_decryptor));
480}
481
Tommi81de14f2018-03-25 22:19:25 +0200482void RtpVideoStreamReceiver::UpdateRtt(int64_t max_rtt_ms) {
tommif284b7f2017-02-27 01:59:36 -0800483 if (nack_module_)
484 nack_module_->UpdateRtt(max_rtt_ms);
philipelfd5a20f2016-11-15 00:57:57 -0800485}
486
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200487absl::optional<int64_t> RtpVideoStreamReceiver::LastReceivedPacketMs() const {
philipel3184f8e2017-05-18 08:08:53 -0700488 return packet_buffer_->LastReceivedPacketMs();
489}
490
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200491absl::optional<int64_t> RtpVideoStreamReceiver::LastReceivedKeyframePacketMs()
nisseb1f2ff92017-06-09 04:01:55 -0700492 const {
philipel3184f8e2017-05-18 08:08:53 -0700493 return packet_buffer_->LastReceivedKeyframePacketMs();
494}
495
eladalonc0d481a2017-08-02 07:39:07 -0700496void RtpVideoStreamReceiver::AddSecondarySink(RtpPacketSinkInterface* sink) {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200497 RTC_DCHECK_RUN_ON(&worker_task_checker_);
Steve Antonbd631a02019-03-28 10:51:27 -0700498 RTC_DCHECK(!absl::c_linear_search(secondary_sinks_, sink));
eladalonc0d481a2017-08-02 07:39:07 -0700499 secondary_sinks_.push_back(sink);
500}
501
502void RtpVideoStreamReceiver::RemoveSecondarySink(
503 const RtpPacketSinkInterface* sink) {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200504 RTC_DCHECK_RUN_ON(&worker_task_checker_);
Steve Antonbd631a02019-03-28 10:51:27 -0700505 auto it = absl::c_find(secondary_sinks_, sink);
eladalonc0d481a2017-08-02 07:39:07 -0700506 if (it == secondary_sinks_.end()) {
507 // We might be rolling-back a call whose setup failed mid-way. In such a
508 // case, it's simpler to remove "everything" rather than remember what
509 // has already been added.
Mirko Bonadei675513b2017-11-09 11:09:25 +0100510 RTC_LOG(LS_WARNING) << "Removal of unknown sink.";
eladalonc0d481a2017-08-02 07:39:07 -0700511 return;
512 }
513 secondary_sinks_.erase(it);
514}
515
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200516void RtpVideoStreamReceiver::ReceivePacket(const RtpPacketReceived& packet) {
517 if (packet.payload_size() == 0) {
Niels Möller0b926782018-08-21 17:49:24 +0200518 // Padding or keep-alive packet.
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200519 // TODO(nisse): Could drop empty packets earlier, but need to figure out how
520 // they should be counted in stats.
Niels Möller0b926782018-08-21 17:49:24 +0200521 NotifyReceiverOfEmptyPacket(packet.SequenceNumber());
nisse30e89312017-05-29 08:16:37 -0700522 return;
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000523 }
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200524 if (packet.PayloadType() == config_.rtp.red_payload_type) {
Niels Möller1f3206c2018-09-14 08:26:32 +0200525 ParseAndHandleEncapsulatingHeader(packet);
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200526 return;
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000527 }
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200528
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +0200529 const auto type_it = payload_type_map_.find(packet.PayloadType());
530 if (type_it == payload_type_map_.end()) {
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200531 return;
532 }
533 auto depacketizer =
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +0200534 absl::WrapUnique(RtpDepacketizer::Create(type_it->second));
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200535
536 if (!depacketizer) {
537 RTC_LOG(LS_ERROR) << "Failed to create depacketizer.";
538 return;
539 }
540 RtpDepacketizer::ParsedPayload parsed_payload;
541 if (!depacketizer->Parse(&parsed_payload, packet.payload().data(),
542 packet.payload().size())) {
543 RTC_LOG(LS_WARNING) << "Failed parsing payload.";
544 return;
545 }
546
Niels Möller125b5d62019-03-11 16:11:07 +0100547 RTPHeader rtp_header;
548 packet.GetHeader(&rtp_header);
549 RTPVideoHeader video_header = parsed_payload.video_header();
550 video_header.rotation = kVideoRotation_0;
551 video_header.content_type = VideoContentType::UNSPECIFIED;
552 video_header.video_timing.flags = VideoSendTiming::kInvalid;
553 video_header.is_last_packet_in_frame = rtp_header.markerBit;
554 video_header.frame_marking.temporal_id = kNoTemporalIdx;
Johnny Lee1a1c52b2019-02-08 14:25:40 -0500555
Ilya Nikolaevskiy5546aef2018-12-04 15:54:52 +0100556 if (parsed_payload.video_header().codec == kVideoCodecVP9) {
557 const RTPVideoHeaderVP9& codec_header = absl::get<RTPVideoHeaderVP9>(
558 parsed_payload.video_header().video_type_header);
Niels Möller125b5d62019-03-11 16:11:07 +0100559 video_header.is_last_packet_in_frame |= codec_header.end_of_frame;
560 video_header.is_first_packet_in_frame |= codec_header.beginning_of_frame;
Ilya Nikolaevskiy5546aef2018-12-04 15:54:52 +0100561 }
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200562
Niels Möller125b5d62019-03-11 16:11:07 +0100563 packet.GetExtension<VideoOrientation>(&video_header.rotation);
564 packet.GetExtension<VideoContentTypeExtension>(&video_header.content_type);
565 packet.GetExtension<VideoTimingExtension>(&video_header.video_timing);
566 packet.GetExtension<PlayoutDelayLimits>(&video_header.playout_delay);
567 packet.GetExtension<FrameMarkingExtension>(&video_header.frame_marking);
Johnny Lee1a1c52b2019-02-08 14:25:40 -0500568
Johannes Krond51ec582019-04-15 13:32:41 +0200569 // Color space should only be transmitted in the last packet of a frame,
570 // therefore, neglect it otherwise so that last_color_space_ is not reset by
571 // mistake.
572 if (video_header.is_last_packet_in_frame) {
573 video_header.color_space = packet.GetExtension<ColorSpaceExtension>();
574 if (video_header.color_space ||
Niels Möllerabbc50e2019-04-24 09:41:16 +0200575 video_header.frame_type == VideoFrameType::kVideoFrameKey) {
Johannes Krond51ec582019-04-15 13:32:41 +0200576 // Store color space since it's only transmitted when changed or for key
577 // frames. Color space will be cleared if a key frame is transmitted
578 // without color space information.
579 last_color_space_ = video_header.color_space;
580 } else if (last_color_space_) {
581 video_header.color_space = last_color_space_;
582 }
Johannes Krond0b69a82018-12-03 14:18:53 +0100583 }
Johnny Lee1a1c52b2019-02-08 14:25:40 -0500584
philipel2837edc2018-10-02 13:55:47 +0200585 absl::optional<RtpGenericFrameDescriptor> generic_descriptor_wire;
586 generic_descriptor_wire.emplace();
Elad Alonccb9b752019-02-19 13:01:31 +0100587 const bool generic_descriptor_v00 =
588 packet.GetExtension<RtpGenericFrameDescriptorExtension00>(
589 &generic_descriptor_wire.value());
590 const bool generic_descriptor_v01 =
591 packet.GetExtension<RtpGenericFrameDescriptorExtension01>(
592 &generic_descriptor_wire.value());
593 if (generic_descriptor_v00 && generic_descriptor_v01) {
594 RTC_LOG(LS_WARNING) << "RTP packet had two different GFD versions.";
595 return;
596 }
597
598 if (generic_descriptor_v00 || generic_descriptor_v01) {
599 if (generic_descriptor_v00) {
600 generic_descriptor_wire->SetByteRepresentation(
601 packet.GetRawExtension<RtpGenericFrameDescriptorExtension00>());
602 } else {
603 generic_descriptor_wire->SetByteRepresentation(
604 packet.GetRawExtension<RtpGenericFrameDescriptorExtension01>());
605 }
606
Niels Möller125b5d62019-03-11 16:11:07 +0100607 video_header.is_first_packet_in_frame =
philipel2837edc2018-10-02 13:55:47 +0200608 generic_descriptor_wire->FirstPacketInSubFrame();
Niels Möller125b5d62019-03-11 16:11:07 +0100609 video_header.is_last_packet_in_frame =
610 rtp_header.markerBit || generic_descriptor_wire->LastPacketInSubFrame();
philipelfab91292018-10-17 14:36:08 +0200611
612 if (generic_descriptor_wire->FirstPacketInSubFrame()) {
Niels Möllerabbc50e2019-04-24 09:41:16 +0200613 video_header.frame_type =
philipelfab91292018-10-17 14:36:08 +0200614 generic_descriptor_wire->FrameDependenciesDiffs().empty()
Niels Möller8f7ce222019-03-21 15:43:58 +0100615 ? VideoFrameType::kVideoFrameKey
616 : VideoFrameType::kVideoFrameDelta;
philipelfab91292018-10-17 14:36:08 +0200617 }
618
Niels Möller125b5d62019-03-11 16:11:07 +0100619 video_header.width = generic_descriptor_wire->Width();
620 video_header.height = generic_descriptor_wire->Height();
philipel2837edc2018-10-02 13:55:47 +0200621 } else {
622 generic_descriptor_wire.reset();
philipelb3e42a42018-09-13 10:57:14 +0200623 }
624
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200625 OnReceivedPayloadData(parsed_payload.payload, parsed_payload.payload_length,
Niels Möllerabbc50e2019-04-24 09:41:16 +0200626 rtp_header, video_header, generic_descriptor_wire,
627 packet.recovered());
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000628}
629
nisseb1f2ff92017-06-09 04:01:55 -0700630void RtpVideoStreamReceiver::ParseAndHandleEncapsulatingHeader(
Niels Möller1f3206c2018-09-14 08:26:32 +0200631 const RtpPacketReceived& packet) {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200632 RTC_DCHECK_RUN_ON(&worker_task_checker_);
Niels Möller1f3206c2018-09-14 08:26:32 +0200633 if (packet.PayloadType() == config_.rtp.red_payload_type &&
634 packet.payload_size() > 0) {
635 if (packet.payload()[0] == config_.rtp.ulpfec_payload_type) {
636 rtp_receive_statistics_->FecPacketReceived(packet);
Peter Boström0b250722016-04-22 18:23:15 +0200637 // Notify video_receiver about received FEC packets to avoid NACKing these
638 // packets.
Niels Möller1f3206c2018-09-14 08:26:32 +0200639 NotifyReceiverOfEmptyPacket(packet.SequenceNumber());
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000640 }
Niels Möller1f3206c2018-09-14 08:26:32 +0200641 RTPHeader header;
642 packet.GetHeader(&header);
Niels Möller3f027b32018-03-14 08:04:58 +0100643 if (ulpfec_receiver_->AddReceivedRedPacket(
Niels Möller1f3206c2018-09-14 08:26:32 +0200644 header, packet.data(), packet.size(),
645 config_.rtp.ulpfec_payload_type) != 0) {
nisse30e89312017-05-29 08:16:37 -0700646 return;
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000647 }
nisse30e89312017-05-29 08:16:37 -0700648 ulpfec_receiver_->ProcessReceivedFec();
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000649 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000650}
651
Niels Möllerbc010472018-03-23 13:22:29 +0100652// In the case of a video stream without picture ids and no rtx the
653// RtpFrameReferenceFinder will need to know about padding to
654// correctly calculate frame references.
655void RtpVideoStreamReceiver::NotifyReceiverOfEmptyPacket(uint16_t seq_num) {
656 reference_finder_->PaddingReceived(seq_num);
657 packet_buffer_->PaddingReceived(seq_num);
Niels Möllerbc010472018-03-23 13:22:29 +0100658 if (nack_module_) {
Ying Wangb32bb952018-10-31 10:12:27 +0100659 nack_module_->OnReceivedPacket(seq_num, /* is_keyframe = */ false,
660 /* is _recovered = */ false);
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000661 }
Elad Alon7d6a4c02019-02-25 13:00:51 +0100662 if (loss_notification_controller_) {
Elad Alonca2c4302019-05-27 22:43:10 +0200663 // TODO(bugs.webrtc.org/10336): Handle empty packets.
Elad Alon7d6a4c02019-02-25 13:00:51 +0100664 RTC_LOG(LS_WARNING)
665 << "LossNotificationController does not expect empty packets.";
666 }
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000667}
668
nisseb1f2ff92017-06-09 04:01:55 -0700669bool RtpVideoStreamReceiver::DeliverRtcp(const uint8_t* rtcp_packet,
670 size_t rtcp_packet_length) {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200671 RTC_DCHECK_RUN_ON(&worker_task_checker_);
eladalon8b073052017-08-25 00:49:08 -0700672
673 if (!receiving_) {
674 return false;
Peter Boström4fa7eca2016-03-02 15:05:53 +0100675 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000676
Per83d09102016-04-15 14:59:13 +0200677 rtp_rtcp_->IncomingRtcpPacket(rtcp_packet, rtcp_packet_length);
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000678
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000679 int64_t rtt = 0;
Niels Möller2ff1f2a2018-08-09 16:16:34 +0200680 rtp_rtcp_->RTT(config_.rtp.remote_ssrc, &rtt, nullptr, nullptr, nullptr);
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000681 if (rtt == 0) {
682 // Waiting for valid rtt.
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100683 return true;
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000684 }
685 uint32_t ntp_secs = 0;
686 uint32_t ntp_frac = 0;
687 uint32_t rtp_timestamp = 0;
Ilya Nikolaevskiy7172ea12017-10-30 11:17:34 +0100688 uint32_t recieved_ntp_secs = 0;
689 uint32_t recieved_ntp_frac = 0;
690 if (rtp_rtcp_->RemoteNTP(&ntp_secs, &ntp_frac, &recieved_ntp_secs,
691 &recieved_ntp_frac, &rtp_timestamp) != 0) {
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000692 // Waiting for RTCP.
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100693 return true;
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000694 }
Ilya Nikolaevskiy7172ea12017-10-30 11:17:34 +0100695 NtpTime recieved_ntp(recieved_ntp_secs, recieved_ntp_frac);
696 int64_t time_since_recieved =
697 clock_->CurrentNtpInMilliseconds() - recieved_ntp.ToMs();
698 // Don't use old SRs to estimate time.
699 if (time_since_recieved <= 1) {
700 ntp_estimator_.UpdateRtcpTimestamp(rtt, ntp_secs, ntp_frac, rtp_timestamp);
701 }
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000702
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100703 return true;
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000704}
705
philipeld4fac692017-09-04 07:03:46 -0700706void RtpVideoStreamReceiver::FrameContinuous(int64_t picture_id) {
tommif284b7f2017-02-27 01:59:36 -0800707 if (!nack_module_)
708 return;
709
philipela45102f2017-02-22 05:30:39 -0800710 int seq_num = -1;
711 {
712 rtc::CritScope lock(&last_seq_num_cs_);
713 auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id);
714 if (seq_num_it != last_seq_num_for_pic_id_.end())
715 seq_num = seq_num_it->second;
philipelfd5a20f2016-11-15 00:57:57 -0800716 }
philipela45102f2017-02-22 05:30:39 -0800717 if (seq_num != -1)
718 nack_module_->ClearUpTo(seq_num);
philipelfd5a20f2016-11-15 00:57:57 -0800719}
720
philipeld4fac692017-09-04 07:03:46 -0700721void RtpVideoStreamReceiver::FrameDecoded(int64_t picture_id) {
philipela45102f2017-02-22 05:30:39 -0800722 int seq_num = -1;
723 {
724 rtc::CritScope lock(&last_seq_num_cs_);
725 auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id);
726 if (seq_num_it != last_seq_num_for_pic_id_.end()) {
727 seq_num = seq_num_it->second;
728 last_seq_num_for_pic_id_.erase(last_seq_num_for_pic_id_.begin(),
729 ++seq_num_it);
philipelfd5a20f2016-11-15 00:57:57 -0800730 }
philipela45102f2017-02-22 05:30:39 -0800731 }
732 if (seq_num != -1) {
733 packet_buffer_->ClearTo(seq_num);
734 reference_finder_->ClearTo(seq_num);
philipelfd5a20f2016-11-15 00:57:57 -0800735 }
736}
737
nisseb1f2ff92017-06-09 04:01:55 -0700738void RtpVideoStreamReceiver::SignalNetworkState(NetworkState state) {
mflodmandc7d0d22016-05-06 05:32:22 -0700739 rtp_rtcp_->SetRTCPStatus(state == kNetworkUp ? config_.rtp.rtcp_mode
740 : RtcpMode::kOff);
741}
742
Ilya Nikolaevskiyd397a0d2018-02-21 15:57:09 +0100743int RtpVideoStreamReceiver::GetUniqueFramesSeen() const {
744 return packet_buffer_->GetUniqueFramesSeen();
745}
746
nisseb1f2ff92017-06-09 04:01:55 -0700747void RtpVideoStreamReceiver::StartReceive() {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200748 RTC_DCHECK_RUN_ON(&worker_task_checker_);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000749 receiving_ = true;
750}
751
nisseb1f2ff92017-06-09 04:01:55 -0700752void RtpVideoStreamReceiver::StopReceive() {
Sebastian Janssonb55015e2019-04-09 13:44:04 +0200753 RTC_DCHECK_RUN_ON(&worker_task_checker_);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000754 receiving_ = false;
755}
756
nisseb1f2ff92017-06-09 04:01:55 -0700757void RtpVideoStreamReceiver::UpdateHistograms() {
brandtrd55c3f62016-10-31 04:51:33 -0700758 FecPacketCounter counter = ulpfec_receiver_->GetPacketCounter();
asapersson0c43f772016-11-30 01:42:26 -0800759 if (counter.first_packet_time_ms == -1)
760 return;
761
762 int64_t elapsed_sec =
763 (clock_->TimeInMilliseconds() - counter.first_packet_time_ms) / 1000;
764 if (elapsed_sec < metrics::kMinRunTimeInSeconds)
765 return;
766
mflodmandc7d0d22016-05-06 05:32:22 -0700767 if (counter.num_packets > 0) {
asapersson1d02d3e2016-09-09 22:40:25 -0700768 RTC_HISTOGRAM_PERCENTAGE(
mflodmandc7d0d22016-05-06 05:32:22 -0700769 "WebRTC.Video.ReceivedFecPacketsInPercent",
770 static_cast<int>(counter.num_fec_packets * 100 / counter.num_packets));
771 }
772 if (counter.num_fec_packets > 0) {
asapersson1d02d3e2016-09-09 22:40:25 -0700773 RTC_HISTOGRAM_PERCENTAGE("WebRTC.Video.RecoveredMediaPacketsInPercentOfFec",
774 static_cast<int>(counter.num_recovered_packets *
775 100 / counter.num_fec_packets));
mflodmandc7d0d22016-05-06 05:32:22 -0700776 }
777}
778
nisseb1f2ff92017-06-09 04:01:55 -0700779void RtpVideoStreamReceiver::InsertSpsPpsIntoTracker(uint8_t payload_type) {
philipel022b54e2016-12-20 04:15:59 -0800780 auto codec_params_it = pt_codec_params_.find(payload_type);
781 if (codec_params_it == pt_codec_params_.end())
782 return;
783
Mirko Bonadei675513b2017-11-09 11:09:25 +0100784 RTC_LOG(LS_INFO) << "Found out of band supplied codec parameters for"
785 << " payload type: " << static_cast<int>(payload_type);
philipel022b54e2016-12-20 04:15:59 -0800786
787 H264SpropParameterSets sprop_decoder;
788 auto sprop_base64_it =
789 codec_params_it->second.find(cricket::kH264FmtpSpropParameterSets);
790
791 if (sprop_base64_it == codec_params_it->second.end())
792 return;
793
johan62d02c32017-01-24 04:38:27 -0800794 if (!sprop_decoder.DecodeSprop(sprop_base64_it->second.c_str()))
philipel022b54e2016-12-20 04:15:59 -0800795 return;
796
johand2b092f2017-01-24 02:38:17 -0800797 tracker_.InsertSpsPpsNalus(sprop_decoder.sps_nalu(),
798 sprop_decoder.pps_nalu());
philipel022b54e2016-12-20 04:15:59 -0800799}
800
Jonas Oreland49ac5952018-09-26 16:04:32 +0200801std::vector<webrtc::RtpSource> RtpVideoStreamReceiver::GetSources() const {
802 int64_t now_ms = rtc::TimeMillis();
803 std::vector<RtpSource> sources;
804 {
805 rtc::CritScope cs(&rtp_sources_lock_);
806 sources = contributing_sources_.GetSources(now_ms);
807 if (last_received_rtp_system_time_ms_ >=
808 now_ms - ContributingSources::kHistoryMs) {
Johannes Kronb5d91832019-05-21 13:19:22 +0200809 RTC_DCHECK(last_received_rtp_timestamp_.has_value());
Jonas Oreland49ac5952018-09-26 16:04:32 +0200810 sources.emplace_back(*last_received_rtp_system_time_ms_,
Johannes Kronb5d91832019-05-21 13:19:22 +0200811 config_.rtp.remote_ssrc, RtpSourceType::SSRC,
812 /* audio_level */ absl::nullopt,
813 *last_received_rtp_timestamp_);
Jonas Oreland49ac5952018-09-26 16:04:32 +0200814 }
815 }
816 return sources;
817}
818
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000819} // namespace webrtc