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