blob: 53aa1ef3c10fb1305a32d12fc8cfe849529d330a [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
mflodmanfa666592016-04-28 23:15:33 -070011#include "webrtc/video/rtp_stream_receiver.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000012
mflodman@webrtc.org4fd55272013-02-06 17:46:39 +000013#include <vector>
philipelfd5a20f2016-11-15 00:57:57 -080014#include <utility>
mflodman@webrtc.org4fd55272013-02-06 17:46:39 +000015
mflodman4cd27902016-08-05 06:28:45 -070016#include "webrtc/base/checks.h"
tommidea489f2017-03-03 03:20:24 -080017#include "webrtc/base/location.h"
Peter Boström415d2cd2015-10-26 11:35:17 +010018#include "webrtc/base/logging.h"
mflodmancfc8e3b2016-05-03 21:22:04 -070019#include "webrtc/common_types.h"
Peter Boström9c017252016-02-26 16:26:20 +010020#include "webrtc/config.h"
philipel022b54e2016-12-20 04:15:59 -080021#include "webrtc/media/base/mediaconstants.h"
mflodmanc0e58a32016-04-25 01:26:26 -070022#include "webrtc/modules/pacing/packet_router.h"
pbos@webrtc.orgf5d4cb12013-05-17 13:44:48 +000023#include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h"
Henrik Kjellanderff761fb2015-11-04 08:31:52 +010024#include "webrtc/modules/rtp_rtcp/include/receive_statistics.h"
Henrik Kjellanderff761fb2015-11-04 08:31:52 +010025#include "webrtc/modules/rtp_rtcp/include/rtp_cvo.h"
26#include "webrtc/modules/rtp_rtcp/include/rtp_header_parser.h"
Henrik Kjellanderff761fb2015-11-04 08:31:52 +010027#include "webrtc/modules/rtp_rtcp/include/rtp_receiver.h"
28#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h"
brandtrd55c3f62016-10-31 04:51:33 -070029#include "webrtc/modules/rtp_rtcp/include/ulpfec_receiver.h"
nisse38cc1d62017-02-13 05:59:46 -080030#include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h"
31#include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h"
philipelfd5a20f2016-11-15 00:57:57 -080032#include "webrtc/modules/video_coding/frame_object.h"
philipel022b54e2016-12-20 04:15:59 -080033#include "webrtc/modules/video_coding/h264_sprop_parameter_sets.h"
philipelfd5a20f2016-11-15 00:57:57 -080034#include "webrtc/modules/video_coding/h264_sps_pps_tracker.h"
35#include "webrtc/modules/video_coding/packet_buffer.h"
Peter Boström0b250722016-04-22 18:23:15 +020036#include "webrtc/modules/video_coding/video_coding_impl.h"
philipelfd5a20f2016-11-15 00:57:57 -080037#include "webrtc/system_wrappers/include/field_trial.h"
Henrik Kjellander98f53512015-10-28 18:17:40 +010038#include "webrtc/system_wrappers/include/metrics.h"
Henrik Kjellander98f53512015-10-28 18:17:40 +010039#include "webrtc/system_wrappers/include/timestamp_extrapolator.h"
mflodmancfc8e3b2016-05-03 21:22:04 -070040#include "webrtc/video/receive_statistics_proxy.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000041
42namespace webrtc {
43
philipelfd5a20f2016-11-15 00:57:57 -080044namespace {
45constexpr int kPacketBufferStartSize = 32;
46constexpr int kPacketBufferMaxSixe = 2048;
47}
48
mflodmanc0e58a32016-04-25 01:26:26 -070049std::unique_ptr<RtpRtcp> CreateRtpRtcpModule(
50 ReceiveStatistics* receive_statistics,
51 Transport* outgoing_transport,
52 RtcpRttStats* rtt_stats,
53 RtcpPacketTypeCounterObserver* rtcp_packet_type_counter_observer,
nisse15389c02017-01-24 02:36:58 -080054 TransportSequenceNumberAllocator* transport_sequence_number_allocator) {
mflodmanc0e58a32016-04-25 01:26:26 -070055 RtpRtcp::Configuration configuration;
56 configuration.audio = false;
57 configuration.receiver_only = true;
58 configuration.receive_statistics = receive_statistics;
59 configuration.outgoing_transport = outgoing_transport;
60 configuration.intra_frame_callback = nullptr;
61 configuration.rtt_stats = rtt_stats;
62 configuration.rtcp_packet_type_counter_observer =
63 rtcp_packet_type_counter_observer;
mflodmanc0e58a32016-04-25 01:26:26 -070064 configuration.transport_sequence_number_allocator =
65 transport_sequence_number_allocator;
66 configuration.send_bitrate_observer = nullptr;
67 configuration.send_frame_count_observer = nullptr;
68 configuration.send_side_delay_observer = nullptr;
asapersson35151f32016-05-02 23:44:01 -070069 configuration.send_packet_observer = nullptr;
mflodmanc0e58a32016-04-25 01:26:26 -070070 configuration.bandwidth_callback = nullptr;
71 configuration.transport_feedback_callback = nullptr;
72
73 std::unique_ptr<RtpRtcp> rtp_rtcp(RtpRtcp::CreateRtpRtcp(configuration));
mflodmanc0e58a32016-04-25 01:26:26 -070074 rtp_rtcp->SetRTCPStatus(RtcpMode::kCompound);
75
76 return rtp_rtcp;
77}
78
stefan@webrtc.orgeb24b042014-10-14 11:40:13 +000079static const int kPacketLogIntervalMs = 10000;
80
mflodmanfa666592016-04-28 23:15:33 -070081RtpStreamReceiver::RtpStreamReceiver(
mflodmanfa666592016-04-28 23:15:33 -070082 Transport* transport,
83 RtcpRttStats* rtt_stats,
mflodmancfc8e3b2016-05-03 21:22:04 -070084 PacketRouter* packet_router,
Tommi733b5472016-06-10 17:58:01 +020085 const VideoReceiveStream::Config* config,
mflodmandc7d0d22016-05-06 05:32:22 -070086 ReceiveStatisticsProxy* receive_stats_proxy,
Erik Språng737336d2016-07-29 12:59:36 +020087 ProcessThread* process_thread,
philipelfd5a20f2016-11-15 00:57:57 -080088 NackSender* nack_sender,
89 KeyFrameRequestSender* keyframe_request_sender,
90 video_coding::OnCompleteFrameCallback* complete_frame_callback,
91 VCMTiming* timing)
Tommi97888bd2016-01-21 23:24:59 +010092 : clock_(Clock::GetRealTimeClock()),
Tommi733b5472016-06-10 17:58:01 +020093 config_(*config),
mflodmanc0e58a32016-04-25 01:26:26 -070094 packet_router_(packet_router),
mflodmandc7d0d22016-05-06 05:32:22 -070095 process_thread_(process_thread),
Peter Boström4fa7eca2016-03-02 15:05:53 +010096 ntp_estimator_(clock_),
Peter Boström4fa7eca2016-03-02 15:05:53 +010097 rtp_header_parser_(RtpHeaderParser::Create()),
98 rtp_receiver_(RtpReceiver::CreateVideoReceiver(clock_,
99 this,
mflodmanfa666592016-04-28 23:15:33 -0700100 this,
Peter Boström4fa7eca2016-03-02 15:05:53 +0100101 &rtp_payload_registry_)),
102 rtp_receive_statistics_(ReceiveStatistics::Create(clock_)),
brandtrd55c3f62016-10-31 04:51:33 -0700103 ulpfec_receiver_(UlpfecReceiver::Create(this)),
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000104 receiving_(false),
solenberg@webrtc.org3fb8f7b2014-03-24 20:28:11 +0000105 restored_packet_in_use_(false),
mflodmanc0e58a32016-04-25 01:26:26 -0700106 last_packet_log_ms_(-1),
107 rtp_rtcp_(CreateRtpRtcpModule(rtp_receive_statistics_.get(),
108 transport,
109 rtt_stats,
mflodmancfc8e3b2016-05-03 21:22:04 -0700110 receive_stats_proxy,
nisse15389c02017-01-24 02:36:58 -0800111 packet_router)),
philipelfd5a20f2016-11-15 00:57:57 -0800112 complete_frame_callback_(complete_frame_callback),
113 keyframe_request_sender_(keyframe_request_sender),
114 timing_(timing) {
nissefdbfdc92017-03-31 05:44:52 -0700115 packet_router_->AddReceiveRtpModule(rtp_rtcp_.get());
mflodmancfc8e3b2016-05-03 21:22:04 -0700116 rtp_receive_statistics_->RegisterRtpStatisticsCallback(receive_stats_proxy);
117 rtp_receive_statistics_->RegisterRtcpStatisticsCallback(receive_stats_proxy);
118
Tommi733b5472016-06-10 17:58:01 +0200119 RTC_DCHECK(config_.rtp.rtcp_mode != RtcpMode::kOff)
mflodmancfc8e3b2016-05-03 21:22:04 -0700120 << "A stream should not be configured with RTCP disabled. This value is "
121 "reserved for internal usage.";
mflodmandc7d0d22016-05-06 05:32:22 -0700122 RTC_DCHECK(config_.rtp.remote_ssrc != 0);
123 // TODO(pbos): What's an appropriate local_ssrc for receive-only streams?
124 RTC_DCHECK(config_.rtp.local_ssrc != 0);
125 RTC_DCHECK(config_.rtp.remote_ssrc != config_.rtp.local_ssrc);
126
Tommi733b5472016-06-10 17:58:01 +0200127 rtp_rtcp_->SetRTCPStatus(config_.rtp.rtcp_mode);
128 rtp_rtcp_->SetSSRC(config_.rtp.local_ssrc);
mflodmanc0e58a32016-04-25 01:26:26 -0700129 rtp_rtcp_->SetKeyFrameRequestMethod(kKeyFrameReqPliRtcp);
mflodmandc7d0d22016-05-06 05:32:22 -0700130
Tommi733b5472016-06-10 17:58:01 +0200131 for (size_t i = 0; i < config_.rtp.extensions.size(); ++i) {
132 EnableReceiveRtpHeaderExtension(config_.rtp.extensions[i].uri,
133 config_.rtp.extensions[i].id);
mflodmandc7d0d22016-05-06 05:32:22 -0700134 }
mflodmancfc8e3b2016-05-03 21:22:04 -0700135
136 static const int kMaxPacketAgeToNack = 450;
Tommi733b5472016-06-10 17:58:01 +0200137 const int max_reordering_threshold = (config_.rtp.nack.rtp_history_ms > 0)
138 ? kMaxPacketAgeToNack
139 : kDefaultMaxReorderingThreshold;
mflodmancfc8e3b2016-05-03 21:22:04 -0700140 rtp_receive_statistics_->SetMaxReorderingThreshold(max_reordering_threshold);
mflodmandc7d0d22016-05-06 05:32:22 -0700141
brandtr14742122017-01-27 04:53:07 -0800142 if (config_.rtp.rtx_ssrc) {
143 rtp_payload_registry_.SetRtxSsrc(config_.rtp.rtx_ssrc);
mflodmandc7d0d22016-05-06 05:32:22 -0700144
brandtr14742122017-01-27 04:53:07 -0800145 for (const auto& kv : config_.rtp.rtx_payload_types) {
146 RTC_DCHECK(kv.second != 0);
147 rtp_payload_registry_.SetRtxPayloadType(kv.second, kv.first);
148 }
mflodmandc7d0d22016-05-06 05:32:22 -0700149 }
150
brandtrf7c6d722016-12-08 08:25:47 -0800151 if (IsUlpfecEnabled()) {
mflodmandc7d0d22016-05-06 05:32:22 -0700152 VideoCodec ulpfec_codec = {};
153 ulpfec_codec.codecType = kVideoCodecULPFEC;
154 strncpy(ulpfec_codec.plName, "ulpfec", sizeof(ulpfec_codec.plName));
brandtrb5f2c3f2016-10-04 23:28:39 -0700155 ulpfec_codec.plType = config_.rtp.ulpfec.ulpfec_payload_type;
johan07e276c2016-12-13 02:23:03 -0800156 RTC_CHECK(AddReceiveCodec(ulpfec_codec));
brandtre6f98c72016-11-11 03:28:30 -0800157 }
mflodmandc7d0d22016-05-06 05:32:22 -0700158
brandtre6f98c72016-11-11 03:28:30 -0800159 if (IsRedEnabled()) {
mflodmandc7d0d22016-05-06 05:32:22 -0700160 VideoCodec red_codec = {};
161 red_codec.codecType = kVideoCodecRED;
162 strncpy(red_codec.plName, "red", sizeof(red_codec.plName));
brandtrb5f2c3f2016-10-04 23:28:39 -0700163 red_codec.plType = config_.rtp.ulpfec.red_payload_type;
johan07e276c2016-12-13 02:23:03 -0800164 RTC_CHECK(AddReceiveCodec(red_codec));
brandtrb5f2c3f2016-10-04 23:28:39 -0700165 if (config_.rtp.ulpfec.red_rtx_payload_type != -1) {
mflodmandc7d0d22016-05-06 05:32:22 -0700166 rtp_payload_registry_.SetRtxPayloadType(
brandtrb5f2c3f2016-10-04 23:28:39 -0700167 config_.rtp.ulpfec.red_rtx_payload_type,
168 config_.rtp.ulpfec.red_payload_type);
mflodmandc7d0d22016-05-06 05:32:22 -0700169 }
170 }
171
Tommi733b5472016-06-10 17:58:01 +0200172 if (config_.rtp.rtcp_xr.receiver_reference_time_report)
mflodmandc7d0d22016-05-06 05:32:22 -0700173 rtp_rtcp_->SetRtcpXrRrtrStatus(true);
174
175 // Stats callback for CNAME changes.
176 rtp_rtcp_->RegisterRtcpStatisticsCallback(receive_stats_proxy);
177
tommidea489f2017-03-03 03:20:24 -0800178 process_thread_->RegisterModule(rtp_rtcp_.get(), RTC_FROM_HERE);
philipelfd5a20f2016-11-15 00:57:57 -0800179
tommif284b7f2017-02-27 01:59:36 -0800180 if (config_.rtp.nack.rtp_history_ms != 0) {
181 nack_module_.reset(
182 new NackModule(clock_, nack_sender, keyframe_request_sender));
tommidea489f2017-03-03 03:20:24 -0800183 process_thread_->RegisterModule(nack_module_.get(), RTC_FROM_HERE);
tommif284b7f2017-02-27 01:59:36 -0800184 }
philipelfd5a20f2016-11-15 00:57:57 -0800185
philipela45102f2017-02-22 05:30:39 -0800186 packet_buffer_ = video_coding::PacketBuffer::Create(
187 clock_, kPacketBufferStartSize, kPacketBufferMaxSixe, this);
188 reference_finder_.reset(new video_coding::RtpFrameReferenceFinder(this));
mflodmanc0e58a32016-04-25 01:26:26 -0700189}
niklase@google.com470e71d2011-07-07 08:21:25 +0000190
mflodmanfa666592016-04-28 23:15:33 -0700191RtpStreamReceiver::~RtpStreamReceiver() {
tommif284b7f2017-02-27 01:59:36 -0800192 if (nack_module_) {
193 process_thread_->DeRegisterModule(nack_module_.get());
194 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000195
tommif284b7f2017-02-27 01:59:36 -0800196 process_thread_->DeRegisterModule(rtp_rtcp_.get());
philipelfd5a20f2016-11-15 00:57:57 -0800197
nissefdbfdc92017-03-31 05:44:52 -0700198 packet_router_->RemoveReceiveRtpModule(rtp_rtcp_.get());
mflodmandc7d0d22016-05-06 05:32:22 -0700199 UpdateHistograms();
asapersson@webrtc.org0800db72015-01-15 07:40:20 +0000200}
201
philipel022b54e2016-12-20 04:15:59 -0800202bool RtpStreamReceiver::AddReceiveCodec(
203 const VideoCodec& video_codec,
204 const std::map<std::string, std::string>& codec_params) {
205 pt_codec_params_.insert(make_pair(video_codec.plType, codec_params));
206 return AddReceiveCodec(video_codec);
207}
208
johan07e276c2016-12-13 02:23:03 -0800209bool RtpStreamReceiver::AddReceiveCodec(const VideoCodec& video_codec) {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000210 int8_t old_pltype = -1;
magjed56124bd2016-11-24 09:34:46 -0800211 if (rtp_payload_registry_.ReceivePayloadType(video_codec, &old_pltype) !=
212 -1) {
Peter Boström4fa7eca2016-03-02 15:05:53 +0100213 rtp_payload_registry_.DeRegisterReceivePayload(old_pltype);
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000214 }
magjed56124bd2016-11-24 09:34:46 -0800215 return rtp_payload_registry_.RegisterReceivePayload(video_codec) == 0;
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000216}
217
mflodmanfa666592016-04-28 23:15:33 -0700218uint32_t RtpStreamReceiver::GetRemoteSsrc() const {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000219 return rtp_receiver_->SSRC();
220}
221
mflodmanfa666592016-04-28 23:15:33 -0700222int RtpStreamReceiver::GetCsrcs(uint32_t* csrcs) const {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000223 return rtp_receiver_->CSRCs(csrcs);
224}
225
mflodmanfa666592016-04-28 23:15:33 -0700226RtpReceiver* RtpStreamReceiver::GetRtpReceiver() const {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000227 return rtp_receiver_.get();
228}
229
mflodmanfa666592016-04-28 23:15:33 -0700230int32_t RtpStreamReceiver::OnReceivedPayloadData(
231 const uint8_t* payload_data,
Peter Boström02083222016-06-14 12:52:54 +0200232 size_t payload_size,
mflodmanfa666592016-04-28 23:15:33 -0700233 const WebRtcRTPHeader* rtp_header) {
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000234 WebRtcRTPHeader rtp_header_with_ntp = *rtp_header;
wu@webrtc.org88abf112014-05-14 16:53:51 +0000235 rtp_header_with_ntp.ntp_time_ms =
Peter Boström4fa7eca2016-03-02 15:05:53 +0100236 ntp_estimator_.Estimate(rtp_header->header.timestamp);
philipela45102f2017-02-22 05:30:39 -0800237 VCMPacket packet(payload_data, payload_size, rtp_header_with_ntp);
tommif284b7f2017-02-27 01:59:36 -0800238 packet.timesNacked =
239 nack_module_ ? nack_module_->OnReceivedPacket(packet) : -1;
philipelfd5a20f2016-11-15 00:57:57 -0800240
philipel54ca9192017-03-21 05:45:18 -0700241 // In the case of a video stream without picture ids and no rtx the
242 // RtpFrameReferenceFinder will need to know about padding to
243 // correctly calculate frame references.
244 if (packet.sizeBytes == 0) {
245 reference_finder_->PaddingReceived(packet.seqNum);
246 return 0;
247 }
248
philipela45102f2017-02-22 05:30:39 -0800249 if (packet.codec == kVideoCodecH264) {
250 // Only when we start to receive packets will we know what payload type
251 // that will be used. When we know the payload type insert the correct
252 // sps/pps into the tracker.
253 if (packet.payloadType != last_payload_type_) {
254 last_payload_type_ = packet.payloadType;
255 InsertSpsPpsIntoTracker(packet.payloadType);
philipelfd5a20f2016-11-15 00:57:57 -0800256 }
257
philipela45102f2017-02-22 05:30:39 -0800258 switch (tracker_.CopyAndFixBitstream(&packet)) {
259 case video_coding::H264SpsPpsTracker::kRequestKeyframe:
260 keyframe_request_sender_->RequestKeyFrame();
261 FALLTHROUGH();
262 case video_coding::H264SpsPpsTracker::kDrop:
263 return 0;
264 case video_coding::H264SpsPpsTracker::kInsert:
265 break;
266 }
267
philipelfd5a20f2016-11-15 00:57:57 -0800268 } else {
philipela45102f2017-02-22 05:30:39 -0800269 uint8_t* data = new uint8_t[packet.sizeBytes];
270 memcpy(data, packet.dataPtr, packet.sizeBytes);
271 packet.dataPtr = data;
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000272 }
philipela45102f2017-02-22 05:30:39 -0800273
274 packet_buffer_->InsertPacket(&packet);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000275 return 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000276}
277
nissed2ef3142017-05-11 08:00:58 -0700278// TODO(nisse): Try to delete this method. Obstacles: It is used by
279// ParseAndHandleEncapsulatingHeader, for handling Rtx packets. And
280// it's part of the RtpData interface which we implement.
mflodmanfa666592016-04-28 23:15:33 -0700281bool RtpStreamReceiver::OnRecoveredPacket(const uint8_t* rtp_packet,
282 size_t rtp_packet_length) {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000283 RTPHeader header;
284 if (!rtp_header_parser_->Parse(rtp_packet, rtp_packet_length, &header)) {
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000285 return false;
286 }
287 header.payload_type_frequency = kVideoPayloadTypeFrequency;
stefan@webrtc.org01581da2014-09-04 06:48:14 +0000288 bool in_order = IsPacketInOrder(header);
289 return ReceivePacket(rtp_packet, rtp_packet_length, header, in_order);
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000290}
291
mflodmanfa666592016-04-28 23:15:33 -0700292// TODO(pbos): Remove as soon as audio can handle a changing payload type
293// without this callback.
294int32_t RtpStreamReceiver::OnInitializeDecoder(
295 const int8_t payload_type,
296 const char payload_name[RTP_PAYLOAD_NAME_SIZE],
297 const int frequency,
298 const size_t channels,
299 const uint32_t rate) {
300 RTC_NOTREACHED();
301 return 0;
302}
303
304void RtpStreamReceiver::OnIncomingSSRCChanged(const uint32_t ssrc) {
305 rtp_rtcp_->SetRemoteSSRC(ssrc);
306}
307
nissed2ef3142017-05-11 08:00:58 -0700308// This method handles both regular RTP packets and packets recovered
309// via FlexFEC.
nisse38cc1d62017-02-13 05:59:46 -0800310void RtpStreamReceiver::OnRtpPacket(const RtpPacketReceived& packet) {
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000311 {
Tommi97888bd2016-01-21 23:24:59 +0100312 rtc::CritScope lock(&receive_cs_);
braveyao@webrtc.orgb6433b72013-07-26 09:02:46 +0000313 if (!receiving_) {
nisse38cc1d62017-02-13 05:59:46 -0800314 return;
braveyao@webrtc.orgb6433b72013-07-26 09:02:46 +0000315 }
solenberg@webrtc.orgfc320462014-02-11 15:27:49 +0000316
nissed2ef3142017-05-11 08:00:58 -0700317 if (!packet.recovered()) {
318 int64_t now_ms = clock_->TimeInMilliseconds();
stefan@webrtc.orgeb24b042014-10-14 11:40:13 +0000319
nissed2ef3142017-05-11 08:00:58 -0700320 // Periodically log the RTP header of incoming packets.
321 if (now_ms - last_packet_log_ms_ > kPacketLogIntervalMs) {
322 std::stringstream ss;
323 ss << "Packet received on SSRC: " << packet.Ssrc()
324 << " with payload type: " << static_cast<int>(packet.PayloadType())
325 << ", timestamp: " << packet.Timestamp()
326 << ", sequence number: " << packet.SequenceNumber()
327 << ", arrival time: " << packet.arrival_time_ms();
328 int32_t time_offset;
329 if (packet.GetExtension<TransmissionOffset>(&time_offset)) {
330 ss << ", toffset: " << time_offset;
331 }
332 uint32_t send_time;
333 if (packet.GetExtension<AbsoluteSendTime>(&send_time)) {
334 ss << ", abs send time: " << send_time;
335 }
336 LOG(LS_INFO) << ss.str();
337 last_packet_log_ms_ = now_ms;
nisse38cc1d62017-02-13 05:59:46 -0800338 }
stefan@webrtc.orgeb24b042014-10-14 11:40:13 +0000339 }
340 }
wu@webrtc.orga9890802013-12-13 00:21:03 +0000341
nisse38cc1d62017-02-13 05:59:46 -0800342 // TODO(nisse): Delete use of GetHeader, but needs refactoring of
343 // ReceivePacket and IncomingPacket methods below.
344 RTPHeader header;
345 packet.GetHeader(&header);
346
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000347 header.payload_type_frequency = kVideoPayloadTypeFrequency;
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000348
stefan@webrtc.org48df3812013-11-08 15:18:52 +0000349 bool in_order = IsPacketInOrder(header);
nissed2ef3142017-05-11 08:00:58 -0700350 if (!packet.recovered()) {
351 // TODO(nisse): Why isn't this done for recovered packets?
352 rtp_payload_registry_.SetIncomingPayloadType(header);
353 }
nisse38cc1d62017-02-13 05:59:46 -0800354 ReceivePacket(packet.data(), packet.size(), header, in_order);
asapersson@webrtc.org1457b472014-05-26 13:06:04 +0000355 // Update receive statistics after ReceivePacket.
356 // Receive statistics will be reset if the payload type changes (make sure
357 // that the first packet is included in the stats).
nissed2ef3142017-05-11 08:00:58 -0700358 if (!packet.recovered()) {
359 // TODO(nisse): We should pass a recovered flag to stats, to aid
360 // fixing bug bugs.webrtc.org/6339.
361 rtp_receive_statistics_->IncomingPacket(
362 header, packet.size(), IsPacketRetransmitted(header, in_order));
363 }
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000364}
365
mflodmancfc8e3b2016-05-03 21:22:04 -0700366int32_t RtpStreamReceiver::RequestKeyFrame() {
367 return rtp_rtcp_->RequestKeyFrame();
368}
369
brandtrf7c6d722016-12-08 08:25:47 -0800370bool RtpStreamReceiver::IsUlpfecEnabled() const {
brandtre6f98c72016-11-11 03:28:30 -0800371 return config_.rtp.ulpfec.ulpfec_payload_type != -1;
372}
373
374bool RtpStreamReceiver::IsRedEnabled() const {
375 return config_.rtp.ulpfec.red_payload_type != -1;
mflodmandc7d0d22016-05-06 05:32:22 -0700376}
377
378bool RtpStreamReceiver::IsRetransmissionsEnabled() const {
379 return config_.rtp.nack.rtp_history_ms > 0;
380}
381
382void RtpStreamReceiver::RequestPacketRetransmit(
383 const std::vector<uint16_t>& sequence_numbers) {
384 rtp_rtcp_->SendNack(sequence_numbers);
385}
386
mflodmancfc8e3b2016-05-03 21:22:04 -0700387int32_t RtpStreamReceiver::ResendPackets(const uint16_t* sequence_numbers,
388 uint16_t length) {
389 return rtp_rtcp_->SendNACK(sequence_numbers, length);
390}
391
philipelfd5a20f2016-11-15 00:57:57 -0800392void RtpStreamReceiver::OnReceivedFrame(
393 std::unique_ptr<video_coding::RtpFrameObject> frame) {
philipelbd26ba72017-01-29 04:04:47 -0800394 if (!frame->delayed_by_retransmission())
395 timing_->IncomingTimestamp(frame->timestamp, clock_->TimeInMilliseconds());
philipelfd5a20f2016-11-15 00:57:57 -0800396 reference_finder_->ManageFrame(std::move(frame));
397}
398
399void RtpStreamReceiver::OnCompleteFrame(
400 std::unique_ptr<video_coding::FrameObject> frame) {
401 {
402 rtc::CritScope lock(&last_seq_num_cs_);
403 video_coding::RtpFrameObject* rtp_frame =
404 static_cast<video_coding::RtpFrameObject*>(frame.get());
405 last_seq_num_for_pic_id_[rtp_frame->picture_id] = rtp_frame->last_seq_num();
406 }
407 complete_frame_callback_->OnCompleteFrame(std::move(frame));
408}
409
410void RtpStreamReceiver::OnRttUpdate(int64_t avg_rtt_ms, int64_t max_rtt_ms) {
tommif284b7f2017-02-27 01:59:36 -0800411 if (nack_module_)
412 nack_module_->UpdateRtt(max_rtt_ms);
philipelfd5a20f2016-11-15 00:57:57 -0800413}
414
nisse38cc1d62017-02-13 05:59:46 -0800415// TODO(nisse): Drop return value.
mflodmanfa666592016-04-28 23:15:33 -0700416bool RtpStreamReceiver::ReceivePacket(const uint8_t* packet,
417 size_t packet_length,
418 const RTPHeader& header,
419 bool in_order) {
Peter Boström4fa7eca2016-03-02 15:05:53 +0100420 if (rtp_payload_registry_.IsEncapsulated(header)) {
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000421 return ParseAndHandleEncapsulatingHeader(packet, packet_length, header);
422 }
423 const uint8_t* payload = packet + header.headerLength;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000424 assert(packet_length >= header.headerLength);
425 size_t payload_length = packet_length - header.headerLength;
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000426 PayloadUnion payload_specific;
Peter Boström4fa7eca2016-03-02 15:05:53 +0100427 if (!rtp_payload_registry_.GetPayloadSpecifics(header.payloadType,
428 &payload_specific)) {
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000429 return false;
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000430 }
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000431 return rtp_receiver_->IncomingRtpPacket(header, payload, payload_length,
432 payload_specific, in_order);
433}
434
mflodmanfa666592016-04-28 23:15:33 -0700435bool RtpStreamReceiver::ParseAndHandleEncapsulatingHeader(
436 const uint8_t* packet, size_t packet_length, const RTPHeader& header) {
Peter Boström4fa7eca2016-03-02 15:05:53 +0100437 if (rtp_payload_registry_.IsRed(header)) {
438 int8_t ulpfec_pt = rtp_payload_registry_.ulpfec_payload_type();
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000439 if (packet[header.headerLength] == ulpfec_pt) {
asapersson@webrtc.org273fbbb2015-01-27 12:17:29 +0000440 rtp_receive_statistics_->FecPacketReceived(header, packet_length);
Peter Boström0b250722016-04-22 18:23:15 +0200441 // Notify video_receiver about received FEC packets to avoid NACKing these
442 // packets.
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000443 NotifyReceiverOfFecPacket(header);
444 }
brandtrd55c3f62016-10-31 04:51:33 -0700445 if (ulpfec_receiver_->AddReceivedRedPacket(header, packet, packet_length,
446 ulpfec_pt) != 0) {
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000447 return false;
448 }
brandtrd55c3f62016-10-31 04:51:33 -0700449 return ulpfec_receiver_->ProcessReceivedFec() == 0;
Peter Boström4fa7eca2016-03-02 15:05:53 +0100450 } else if (rtp_payload_registry_.IsRtx(header)) {
stefan@webrtc.org7c6ff2d2014-03-19 18:14:52 +0000451 if (header.headerLength + header.paddingLength == packet_length) {
452 // This is an empty packet and should be silently dropped before trying to
453 // parse the RTX header.
454 return true;
455 }
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000456 // Remove the RTX header and parse the original RTP header.
457 if (packet_length < header.headerLength)
458 return false;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000459 if (packet_length > sizeof(restored_packet_))
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000460 return false;
Tommi97888bd2016-01-21 23:24:59 +0100461 rtc::CritScope lock(&receive_cs_);
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000462 if (restored_packet_in_use_) {
pbos@webrtc.org4e2806d2014-05-14 08:02:22 +0000463 LOG(LS_WARNING) << "Multiple RTX headers detected, dropping packet.";
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000464 return false;
465 }
Peter Boström4fa7eca2016-03-02 15:05:53 +0100466 if (!rtp_payload_registry_.RestoreOriginalPacket(
noahric65220a72015-10-14 11:29:49 -0700467 restored_packet_, packet, &packet_length, rtp_receiver_->SSRC(),
468 header)) {
Stefan Holmer10880012016-02-03 13:29:59 +0100469 LOG(LS_WARNING) << "Incoming RTX packet: Invalid RTP header ssrc: "
470 << header.ssrc << " payload type: "
471 << static_cast<int>(header.payloadType);
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000472 return false;
473 }
474 restored_packet_in_use_ = true;
noahric65220a72015-10-14 11:29:49 -0700475 bool ret = OnRecoveredPacket(restored_packet_, packet_length);
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000476 restored_packet_in_use_ = false;
477 return ret;
478 }
479 return false;
niklase@google.com470e71d2011-07-07 08:21:25 +0000480}
481
mflodmanfa666592016-04-28 23:15:33 -0700482void RtpStreamReceiver::NotifyReceiverOfFecPacket(const RTPHeader& header) {
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000483 int8_t last_media_payload_type =
Peter Boström4fa7eca2016-03-02 15:05:53 +0100484 rtp_payload_registry_.last_received_media_payload_type();
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000485 if (last_media_payload_type < 0) {
486 LOG(LS_WARNING) << "Failed to get last media payload type.";
487 return;
488 }
489 // Fake an empty media packet.
490 WebRtcRTPHeader rtp_header = {};
491 rtp_header.header = header;
492 rtp_header.header.payloadType = last_media_payload_type;
493 rtp_header.header.paddingLength = 0;
494 PayloadUnion payload_specific;
Peter Boström4fa7eca2016-03-02 15:05:53 +0100495 if (!rtp_payload_registry_.GetPayloadSpecifics(last_media_payload_type,
496 &payload_specific)) {
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000497 LOG(LS_WARNING) << "Failed to get payload specifics.";
498 return;
499 }
500 rtp_header.type.Video.codec = payload_specific.Video.videoCodecType;
guoweis@webrtc.orgfdd10572015-03-12 20:50:57 +0000501 rtp_header.type.Video.rotation = kVideoRotation_0;
502 if (header.extension.hasVideoRotation) {
magjed71eb61c2016-09-08 03:24:58 -0700503 rtp_header.type.Video.rotation = header.extension.videoRotation;
guoweis@webrtc.orgfdd10572015-03-12 20:50:57 +0000504 }
ilnik00d802b2017-04-11 10:34:31 -0700505 rtp_header.type.Video.content_type = VideoContentType::UNSPECIFIED;
506 if (header.extension.hasVideoContentType) {
507 rtp_header.type.Video.content_type = header.extension.videoContentType;
508 }
isheriff6b4b5f32016-06-08 00:24:21 -0700509 rtp_header.type.Video.playout_delay = header.extension.playout_delay;
510
Peter Boström74f6e9e2016-04-04 17:56:10 +0200511 OnReceivedPayloadData(nullptr, 0, &rtp_header);
asapersson@webrtc.org37c05592015-01-28 13:58:27 +0000512}
513
mflodmanfa666592016-04-28 23:15:33 -0700514bool RtpStreamReceiver::DeliverRtcp(const uint8_t* rtcp_packet,
515 size_t rtcp_packet_length) {
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000516 {
Tommi97888bd2016-01-21 23:24:59 +0100517 rtc::CritScope lock(&receive_cs_);
braveyao@webrtc.orgb6433b72013-07-26 09:02:46 +0000518 if (!receiving_) {
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100519 return false;
braveyao@webrtc.orgb6433b72013-07-26 09:02:46 +0000520 }
Peter Boström4fa7eca2016-03-02 15:05:53 +0100521 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000522
Per83d09102016-04-15 14:59:13 +0200523 rtp_rtcp_->IncomingRtcpPacket(rtcp_packet, rtcp_packet_length);
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000524
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000525 int64_t rtt = 0;
Per83d09102016-04-15 14:59:13 +0200526 rtp_rtcp_->RTT(rtp_receiver_->SSRC(), &rtt, nullptr, nullptr, nullptr);
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000527 if (rtt == 0) {
528 // Waiting for valid rtt.
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100529 return true;
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000530 }
531 uint32_t ntp_secs = 0;
532 uint32_t ntp_frac = 0;
533 uint32_t rtp_timestamp = 0;
Per83d09102016-04-15 14:59:13 +0200534 if (rtp_rtcp_->RemoteNTP(&ntp_secs, &ntp_frac, nullptr, nullptr,
535 &rtp_timestamp) != 0) {
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000536 // Waiting for RTCP.
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100537 return true;
minyue@webrtc.org2c0cdbc2014-10-09 10:52:43 +0000538 }
Peter Boström4fa7eca2016-03-02 15:05:53 +0100539 ntp_estimator_.UpdateRtcpTimestamp(rtt, ntp_secs, ntp_frac, rtp_timestamp);
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000540
Peter Boströmd1d66ba2016-02-08 14:07:14 +0100541 return true;
wu@webrtc.orgcd701192014-04-24 22:10:24 +0000542}
543
philipelfd5a20f2016-11-15 00:57:57 -0800544void RtpStreamReceiver::FrameContinuous(uint16_t picture_id) {
tommif284b7f2017-02-27 01:59:36 -0800545 if (!nack_module_)
546 return;
547
philipela45102f2017-02-22 05:30:39 -0800548 int seq_num = -1;
549 {
550 rtc::CritScope lock(&last_seq_num_cs_);
551 auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id);
552 if (seq_num_it != last_seq_num_for_pic_id_.end())
553 seq_num = seq_num_it->second;
philipelfd5a20f2016-11-15 00:57:57 -0800554 }
philipela45102f2017-02-22 05:30:39 -0800555 if (seq_num != -1)
556 nack_module_->ClearUpTo(seq_num);
philipelfd5a20f2016-11-15 00:57:57 -0800557}
558
559void RtpStreamReceiver::FrameDecoded(uint16_t picture_id) {
philipela45102f2017-02-22 05:30:39 -0800560 int seq_num = -1;
561 {
562 rtc::CritScope lock(&last_seq_num_cs_);
563 auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id);
564 if (seq_num_it != last_seq_num_for_pic_id_.end()) {
565 seq_num = seq_num_it->second;
566 last_seq_num_for_pic_id_.erase(last_seq_num_for_pic_id_.begin(),
567 ++seq_num_it);
philipelfd5a20f2016-11-15 00:57:57 -0800568 }
philipela45102f2017-02-22 05:30:39 -0800569 }
570 if (seq_num != -1) {
571 packet_buffer_->ClearTo(seq_num);
572 reference_finder_->ClearTo(seq_num);
philipelfd5a20f2016-11-15 00:57:57 -0800573 }
574}
575
mflodmandc7d0d22016-05-06 05:32:22 -0700576void RtpStreamReceiver::SignalNetworkState(NetworkState state) {
577 rtp_rtcp_->SetRTCPStatus(state == kNetworkUp ? config_.rtp.rtcp_mode
578 : RtcpMode::kOff);
579}
580
mflodmanfa666592016-04-28 23:15:33 -0700581void RtpStreamReceiver::StartReceive() {
Tommi97888bd2016-01-21 23:24:59 +0100582 rtc::CritScope lock(&receive_cs_);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000583 receiving_ = true;
584}
585
mflodmanfa666592016-04-28 23:15:33 -0700586void RtpStreamReceiver::StopReceive() {
Tommi97888bd2016-01-21 23:24:59 +0100587 rtc::CritScope lock(&receive_cs_);
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000588 receiving_ = false;
589}
590
mflodmanfa666592016-04-28 23:15:33 -0700591bool RtpStreamReceiver::IsPacketInOrder(const RTPHeader& header) const {
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000592 StreamStatistician* statistician =
593 rtp_receive_statistics_->GetStatistician(header.ssrc);
594 if (!statistician)
595 return false;
596 return statistician->IsPacketInOrder(header.sequenceNumber);
597}
598
mflodmanfa666592016-04-28 23:15:33 -0700599bool RtpStreamReceiver::IsPacketRetransmitted(const RTPHeader& header,
600 bool in_order) const {
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000601 // Retransmissions are handled separately if RTX is enabled.
Peter Boström4fa7eca2016-03-02 15:05:53 +0100602 if (rtp_payload_registry_.RtxEnabled())
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000603 return false;
604 StreamStatistician* statistician =
605 rtp_receive_statistics_->GetStatistician(header.ssrc);
606 if (!statistician)
607 return false;
608 // Check if this is a retransmission.
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000609 int64_t min_rtt = 0;
Per83d09102016-04-15 14:59:13 +0200610 rtp_rtcp_->RTT(rtp_receiver_->SSRC(), nullptr, nullptr, &min_rtt, nullptr);
stefan@webrtc.org48df3812013-11-08 15:18:52 +0000611 return !in_order &&
stefan@webrtc.org7bb8f022013-09-06 13:40:11 +0000612 statistician->IsRetransmitOfOldPacket(header, min_rtt);
wu@webrtc.org822fbd82013-08-15 23:38:54 +0000613}
mflodmandc7d0d22016-05-06 05:32:22 -0700614
615void RtpStreamReceiver::UpdateHistograms() {
brandtrd55c3f62016-10-31 04:51:33 -0700616 FecPacketCounter counter = ulpfec_receiver_->GetPacketCounter();
asapersson0c43f772016-11-30 01:42:26 -0800617 if (counter.first_packet_time_ms == -1)
618 return;
619
620 int64_t elapsed_sec =
621 (clock_->TimeInMilliseconds() - counter.first_packet_time_ms) / 1000;
622 if (elapsed_sec < metrics::kMinRunTimeInSeconds)
623 return;
624
mflodmandc7d0d22016-05-06 05:32:22 -0700625 if (counter.num_packets > 0) {
asapersson1d02d3e2016-09-09 22:40:25 -0700626 RTC_HISTOGRAM_PERCENTAGE(
mflodmandc7d0d22016-05-06 05:32:22 -0700627 "WebRTC.Video.ReceivedFecPacketsInPercent",
628 static_cast<int>(counter.num_fec_packets * 100 / counter.num_packets));
629 }
630 if (counter.num_fec_packets > 0) {
asapersson1d02d3e2016-09-09 22:40:25 -0700631 RTC_HISTOGRAM_PERCENTAGE("WebRTC.Video.RecoveredMediaPacketsInPercentOfFec",
632 static_cast<int>(counter.num_recovered_packets *
633 100 / counter.num_fec_packets));
mflodmandc7d0d22016-05-06 05:32:22 -0700634 }
635}
636
637void RtpStreamReceiver::EnableReceiveRtpHeaderExtension(
638 const std::string& extension, int id) {
639 // One-byte-extension local identifiers are in the range 1-14 inclusive.
640 RTC_DCHECK_GE(id, 1);
641 RTC_DCHECK_LE(id, 14);
642 RTC_DCHECK(RtpExtension::IsSupportedForVideo(extension));
643 RTC_CHECK(rtp_header_parser_->RegisterRtpHeaderExtension(
644 StringToRtpExtensionType(extension), id));
645}
646
philipel022b54e2016-12-20 04:15:59 -0800647void RtpStreamReceiver::InsertSpsPpsIntoTracker(uint8_t payload_type) {
648 auto codec_params_it = pt_codec_params_.find(payload_type);
649 if (codec_params_it == pt_codec_params_.end())
650 return;
651
652 LOG(LS_INFO) << "Found out of band supplied codec parameters for"
johan62d02c32017-01-24 04:38:27 -0800653 << " payload type: " << static_cast<int>(payload_type);
philipel022b54e2016-12-20 04:15:59 -0800654
655 H264SpropParameterSets sprop_decoder;
656 auto sprop_base64_it =
657 codec_params_it->second.find(cricket::kH264FmtpSpropParameterSets);
658
659 if (sprop_base64_it == codec_params_it->second.end())
660 return;
661
johan62d02c32017-01-24 04:38:27 -0800662 if (!sprop_decoder.DecodeSprop(sprop_base64_it->second.c_str()))
philipel022b54e2016-12-20 04:15:59 -0800663 return;
664
johand2b092f2017-01-24 02:38:17 -0800665 tracker_.InsertSpsPpsNalus(sprop_decoder.sps_nalu(),
666 sprop_decoder.pps_nalu());
philipel022b54e2016-12-20 04:15:59 -0800667}
668
mflodman@webrtc.orgad4ee362011-11-28 22:39:24 +0000669} // namespace webrtc