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 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 11 | #include "video/rtp_video_stream_receiver.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 12 | |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 13 | #include <algorithm> |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 14 | #include <utility> |
ilnik | 04f4d12 | 2017-06-19 07:18:55 -0700 | [diff] [blame] | 15 | #include <vector> |
mflodman@webrtc.org | 4fd5527 | 2013-02-06 17:46:39 +0000 | [diff] [blame] | 16 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 17 | #include "absl/memory/memory.h" |
| 18 | |
Mirko Bonadei | 7120742 | 2017-09-15 13:58:09 +0200 | [diff] [blame] | 19 | #include "common_types.h" // NOLINT(build/include) |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 20 | #include "media/base/mediaconstants.h" |
| 21 | #include "modules/pacing/packet_router.h" |
| 22 | #include "modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h" |
| 23 | #include "modules/rtp_rtcp/include/receive_statistics.h" |
| 24 | #include "modules/rtp_rtcp/include/rtp_cvo.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 25 | #include "modules/rtp_rtcp/include/rtp_rtcp.h" |
| 26 | #include "modules/rtp_rtcp/include/ulpfec_receiver.h" |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 27 | #include "modules/rtp_rtcp/source/rtp_format.h" |
philipel | b3e42a4 | 2018-09-13 10:57:14 +0200 | [diff] [blame] | 28 | #include "modules/rtp_rtcp/source/rtp_generic_frame_descriptor_extension.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 29 | #include "modules/rtp_rtcp/source/rtp_header_extensions.h" |
| 30 | #include "modules/rtp_rtcp/source/rtp_packet_received.h" |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 31 | #include "modules/rtp_rtcp/source/rtp_rtcp_config.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 32 | #include "modules/video_coding/frame_object.h" |
| 33 | #include "modules/video_coding/h264_sprop_parameter_sets.h" |
| 34 | #include "modules/video_coding/h264_sps_pps_tracker.h" |
Ilya Nikolaevskiy | 8643b78 | 2018-06-07 16:15:40 +0200 | [diff] [blame] | 35 | #include "modules/video_coding/nack_module.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 36 | #include "modules/video_coding/packet_buffer.h" |
| 37 | #include "modules/video_coding/video_coding_impl.h" |
| 38 | #include "rtc_base/checks.h" |
| 39 | #include "rtc_base/location.h" |
| 40 | #include "rtc_base/logging.h" |
Jonas Olsson | 366a50c | 2018-09-06 13:41:30 +0200 | [diff] [blame] | 41 | #include "rtc_base/strings/string_builder.h" |
Karl Wiberg | 80ba333 | 2018-02-05 10:33:35 +0100 | [diff] [blame] | 42 | #include "rtc_base/system/fallthrough.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 43 | #include "system_wrappers/include/field_trial.h" |
| 44 | #include "system_wrappers/include/metrics.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 45 | #include "video/receive_statistics_proxy.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 46 | |
| 47 | namespace webrtc { |
| 48 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 49 | namespace { |
philipel | 3bf97cf | 2017-08-10 18:10:59 +0200 | [diff] [blame] | 50 | // TODO(philipel): Change kPacketBufferStartSize back to 32 in M63 see: |
| 51 | // crbug.com/752886 |
| 52 | constexpr int kPacketBufferStartSize = 512; |
Johannes Kron | 201596f | 2018-10-22 14:33:39 +0200 | [diff] [blame^] | 53 | constexpr int kPacketBufferMaxSize = 2048; |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 54 | } // namespace |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 55 | |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 56 | std::unique_ptr<RtpRtcp> CreateRtpRtcpModule( |
| 57 | ReceiveStatistics* receive_statistics, |
| 58 | Transport* outgoing_transport, |
| 59 | RtcpRttStats* rtt_stats, |
| 60 | RtcpPacketTypeCounterObserver* rtcp_packet_type_counter_observer, |
nisse | 15389c0 | 2017-01-24 02:36:58 -0800 | [diff] [blame] | 61 | TransportSequenceNumberAllocator* transport_sequence_number_allocator) { |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 62 | RtpRtcp::Configuration configuration; |
| 63 | configuration.audio = false; |
| 64 | configuration.receiver_only = true; |
| 65 | configuration.receive_statistics = receive_statistics; |
| 66 | configuration.outgoing_transport = outgoing_transport; |
| 67 | configuration.intra_frame_callback = nullptr; |
| 68 | configuration.rtt_stats = rtt_stats; |
| 69 | configuration.rtcp_packet_type_counter_observer = |
| 70 | rtcp_packet_type_counter_observer; |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 71 | configuration.transport_sequence_number_allocator = |
| 72 | transport_sequence_number_allocator; |
| 73 | configuration.send_bitrate_observer = nullptr; |
| 74 | configuration.send_frame_count_observer = nullptr; |
| 75 | configuration.send_side_delay_observer = nullptr; |
asapersson | 35151f3 | 2016-05-02 23:44:01 -0700 | [diff] [blame] | 76 | configuration.send_packet_observer = nullptr; |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 77 | configuration.bandwidth_callback = nullptr; |
| 78 | configuration.transport_feedback_callback = nullptr; |
| 79 | |
| 80 | std::unique_ptr<RtpRtcp> rtp_rtcp(RtpRtcp::CreateRtpRtcp(configuration)); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 81 | rtp_rtcp->SetRTCPStatus(RtcpMode::kCompound); |
| 82 | |
| 83 | return rtp_rtcp; |
| 84 | } |
| 85 | |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 86 | static const int kPacketLogIntervalMs = 10000; |
| 87 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 88 | RtpVideoStreamReceiver::RtpVideoStreamReceiver( |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 89 | Transport* transport, |
| 90 | RtcpRttStats* rtt_stats, |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 91 | PacketRouter* packet_router, |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 92 | const VideoReceiveStream::Config* config, |
nisse | ca5706d | 2017-09-11 02:32:16 -0700 | [diff] [blame] | 93 | ReceiveStatistics* rtp_receive_statistics, |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 94 | ReceiveStatisticsProxy* receive_stats_proxy, |
Erik Språng | 737336d | 2016-07-29 12:59:36 +0200 | [diff] [blame] | 95 | ProcessThread* process_thread, |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 96 | NackSender* nack_sender, |
| 97 | KeyFrameRequestSender* keyframe_request_sender, |
Benjamin Wright | 192eeec | 2018-10-17 17:27:25 -0700 | [diff] [blame] | 98 | video_coding::OnCompleteFrameCallback* complete_frame_callback, |
| 99 | rtc::scoped_refptr<FrameDecryptorInterface> frame_decryptor) |
Tommi | 97888bd | 2016-01-21 23:24:59 +0100 | [diff] [blame] | 100 | : clock_(Clock::GetRealTimeClock()), |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 101 | config_(*config), |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 102 | packet_router_(packet_router), |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 103 | process_thread_(process_thread), |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 104 | ntp_estimator_(clock_), |
Niels Möller | b0573bc | 2017-09-25 10:47:00 +0200 | [diff] [blame] | 105 | rtp_header_extensions_(config_.rtp.extensions), |
nisse | ca5706d | 2017-09-11 02:32:16 -0700 | [diff] [blame] | 106 | rtp_receive_statistics_(rtp_receive_statistics), |
brandtr | d726a3f | 2017-06-29 02:45:35 -0700 | [diff] [blame] | 107 | ulpfec_receiver_(UlpfecReceiver::Create(config->rtp.remote_ssrc, this)), |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 108 | receiving_(false), |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 109 | last_packet_log_ms_(-1), |
nisse | ca5706d | 2017-09-11 02:32:16 -0700 | [diff] [blame] | 110 | rtp_rtcp_(CreateRtpRtcpModule(rtp_receive_statistics_, |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 111 | transport, |
| 112 | rtt_stats, |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 113 | receive_stats_proxy, |
nisse | 15389c0 | 2017-01-24 02:36:58 -0800 | [diff] [blame] | 114 | packet_router)), |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 115 | complete_frame_callback_(complete_frame_callback), |
| 116 | keyframe_request_sender_(keyframe_request_sender), |
Benjamin Wright | 192eeec | 2018-10-17 17:27:25 -0700 | [diff] [blame] | 117 | has_received_frame_(false), |
| 118 | frame_decryptor_(frame_decryptor) { |
eladalon | 822ff2b | 2017-08-01 06:30:28 -0700 | [diff] [blame] | 119 | constexpr bool remb_candidate = true; |
| 120 | packet_router_->AddReceiveRtpModule(rtp_rtcp_.get(), remb_candidate); |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 121 | rtp_receive_statistics_->RegisterRtpStatisticsCallback(receive_stats_proxy); |
| 122 | rtp_receive_statistics_->RegisterRtcpStatisticsCallback(receive_stats_proxy); |
| 123 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 124 | RTC_DCHECK(config_.rtp.rtcp_mode != RtcpMode::kOff) |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 125 | << "A stream should not be configured with RTCP disabled. This value is " |
| 126 | "reserved for internal usage."; |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 127 | RTC_DCHECK(config_.rtp.remote_ssrc != 0); |
| 128 | // TODO(pbos): What's an appropriate local_ssrc for receive-only streams? |
| 129 | RTC_DCHECK(config_.rtp.local_ssrc != 0); |
| 130 | RTC_DCHECK(config_.rtp.remote_ssrc != config_.rtp.local_ssrc); |
| 131 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 132 | rtp_rtcp_->SetRTCPStatus(config_.rtp.rtcp_mode); |
| 133 | rtp_rtcp_->SetSSRC(config_.rtp.local_ssrc); |
stefan | b4ab381 | 2017-06-09 06:12:11 -0700 | [diff] [blame] | 134 | rtp_rtcp_->SetRemoteSSRC(config_.rtp.remote_ssrc); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 135 | rtp_rtcp_->SetKeyFrameRequestMethod(kKeyFrameReqPliRtcp); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 136 | |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 137 | static const int kMaxPacketAgeToNack = 450; |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 138 | const int max_reordering_threshold = (config_.rtp.nack.rtp_history_ms > 0) |
| 139 | ? kMaxPacketAgeToNack |
| 140 | : kDefaultMaxReorderingThreshold; |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 141 | rtp_receive_statistics_->SetMaxReorderingThreshold(max_reordering_threshold); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 142 | |
Tommi | 733b547 | 2016-06-10 17:58:01 +0200 | [diff] [blame] | 143 | if (config_.rtp.rtcp_xr.receiver_reference_time_report) |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 144 | rtp_rtcp_->SetRtcpXrRrtrStatus(true); |
| 145 | |
| 146 | // Stats callback for CNAME changes. |
| 147 | rtp_rtcp_->RegisterRtcpStatisticsCallback(receive_stats_proxy); |
| 148 | |
tommi | dea489f | 2017-03-03 03:20:24 -0800 | [diff] [blame] | 149 | process_thread_->RegisterModule(rtp_rtcp_.get(), RTC_FROM_HERE); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 150 | |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 151 | if (config_.rtp.nack.rtp_history_ms != 0) { |
| 152 | nack_module_.reset( |
| 153 | new NackModule(clock_, nack_sender, keyframe_request_sender)); |
tommi | dea489f | 2017-03-03 03:20:24 -0800 | [diff] [blame] | 154 | process_thread_->RegisterModule(nack_module_.get(), RTC_FROM_HERE); |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 155 | } |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 156 | |
Johannes Kron | 201596f | 2018-10-22 14:33:39 +0200 | [diff] [blame^] | 157 | // The group here can be either a positive integer with an explicit size, in |
| 158 | // which case that is used as size. All other values shall result in the |
| 159 | // default value being used. |
| 160 | const std::string group_name = |
| 161 | webrtc::field_trial::FindFullName("WebRTC-PacketBufferMaxSize"); |
| 162 | int packet_buffer_max_size = kPacketBufferMaxSize; |
| 163 | if (!group_name.empty() && |
| 164 | (sscanf(group_name.c_str(), "%d", &packet_buffer_max_size) != 1 || |
| 165 | packet_buffer_max_size <= 0)) { |
| 166 | RTC_LOG(LS_WARNING) << "Invalid packet buffer max size: " << group_name; |
| 167 | packet_buffer_max_size = kPacketBufferMaxSize; |
| 168 | } |
| 169 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 170 | packet_buffer_ = video_coding::PacketBuffer::Create( |
Johannes Kron | 201596f | 2018-10-22 14:33:39 +0200 | [diff] [blame^] | 171 | clock_, kPacketBufferStartSize, packet_buffer_max_size, this); |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 172 | reference_finder_.reset(new video_coding::RtpFrameReferenceFinder(this)); |
mflodman | c0e58a3 | 2016-04-25 01:26:26 -0700 | [diff] [blame] | 173 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 174 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 175 | RtpVideoStreamReceiver::~RtpVideoStreamReceiver() { |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 176 | RTC_DCHECK(secondary_sinks_.empty()); |
| 177 | |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 178 | if (nack_module_) { |
| 179 | process_thread_->DeRegisterModule(nack_module_.get()); |
| 180 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 181 | |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 182 | process_thread_->DeRegisterModule(rtp_rtcp_.get()); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 183 | |
nisse | fdbfdc9 | 2017-03-31 05:44:52 -0700 | [diff] [blame] | 184 | packet_router_->RemoveReceiveRtpModule(rtp_rtcp_.get()); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 185 | UpdateHistograms(); |
asapersson@webrtc.org | 0800db7 | 2015-01-15 07:40:20 +0000 | [diff] [blame] | 186 | } |
| 187 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 188 | void RtpVideoStreamReceiver::AddReceiveCodec( |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 189 | const VideoCodec& video_codec, |
| 190 | const std::map<std::string, std::string>& codec_params) { |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 191 | pt_codec_type_.emplace(video_codec.plType, video_codec.codecType); |
| 192 | pt_codec_params_.emplace(video_codec.plType, codec_params); |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 193 | } |
| 194 | |
Niels Möller | df9e9ae | 2018-07-31 08:29:53 +0200 | [diff] [blame] | 195 | absl::optional<Syncable::Info> RtpVideoStreamReceiver::GetSyncInfo() const { |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 196 | Syncable::Info info; |
Niels Möller | df9e9ae | 2018-07-31 08:29:53 +0200 | [diff] [blame] | 197 | if (rtp_rtcp_->RemoteNTP(&info.capture_time_ntp_secs, |
| 198 | &info.capture_time_ntp_frac, nullptr, nullptr, |
| 199 | &info.capture_time_source_clock) != 0) { |
| 200 | return absl::nullopt; |
| 201 | } |
Niels Möller | b0d4b41 | 2018-08-28 13:58:15 +0200 | [diff] [blame] | 202 | { |
Jonas Oreland | 49ac595 | 2018-09-26 16:04:32 +0200 | [diff] [blame] | 203 | rtc::CritScope lock(&rtp_sources_lock_); |
Niels Möller | b0d4b41 | 2018-08-28 13:58:15 +0200 | [diff] [blame] | 204 | if (!last_received_rtp_timestamp_ || !last_received_rtp_system_time_ms_) { |
| 205 | return absl::nullopt; |
| 206 | } |
| 207 | info.latest_received_capture_timestamp = *last_received_rtp_timestamp_; |
| 208 | info.latest_receive_time_ms = *last_received_rtp_system_time_ms_; |
| 209 | } |
Niels Möller | df9e9ae | 2018-07-31 08:29:53 +0200 | [diff] [blame] | 210 | |
| 211 | // Leaves info.current_delay_ms uninitialized. |
| 212 | return info; |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 213 | } |
| 214 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 215 | int32_t RtpVideoStreamReceiver::OnReceivedPayloadData( |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 216 | const uint8_t* payload_data, |
Peter Boström | 0208322 | 2016-06-14 12:52:54 +0200 | [diff] [blame] | 217 | size_t payload_size, |
mflodman | fa66659 | 2016-04-28 23:15:33 -0700 | [diff] [blame] | 218 | const WebRtcRTPHeader* rtp_header) { |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 219 | return OnReceivedPayloadData(payload_data, payload_size, rtp_header, |
| 220 | absl::nullopt); |
| 221 | } |
| 222 | |
| 223 | int32_t RtpVideoStreamReceiver::OnReceivedPayloadData( |
| 224 | const uint8_t* payload_data, |
| 225 | size_t payload_size, |
| 226 | const WebRtcRTPHeader* rtp_header, |
| 227 | const absl::optional<RtpGenericFrameDescriptor>& generic_descriptor) { |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 228 | WebRtcRTPHeader rtp_header_with_ntp = *rtp_header; |
wu@webrtc.org | 88abf11 | 2014-05-14 16:53:51 +0000 | [diff] [blame] | 229 | rtp_header_with_ntp.ntp_time_ms = |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 230 | ntp_estimator_.Estimate(rtp_header->header.timestamp); |
Niels Möller | 8dad9b4 | 2018-08-22 10:36:35 +0200 | [diff] [blame] | 231 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 232 | VCMPacket packet(payload_data, payload_size, rtp_header_with_ntp); |
Niels Möller | 8dad9b4 | 2018-08-22 10:36:35 +0200 | [diff] [blame] | 233 | if (nack_module_) { |
| 234 | const bool is_keyframe = |
| 235 | rtp_header->video_header().is_first_packet_in_frame && |
| 236 | rtp_header->frameType == kVideoFrameKey; |
| 237 | |
| 238 | packet.timesNacked = nack_module_->OnReceivedPacket( |
| 239 | rtp_header->header.sequenceNumber, is_keyframe); |
| 240 | } else { |
| 241 | packet.timesNacked = -1; |
| 242 | } |
ilnik | 04f4d12 | 2017-06-19 07:18:55 -0700 | [diff] [blame] | 243 | packet.receive_time_ms = clock_->TimeInMilliseconds(); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 244 | |
philipel | 54ca919 | 2017-03-21 05:45:18 -0700 | [diff] [blame] | 245 | if (packet.sizeBytes == 0) { |
Niels Möller | bc01047 | 2018-03-23 13:22:29 +0100 | [diff] [blame] | 246 | NotifyReceiverOfEmptyPacket(packet.seqNum); |
philipel | 54ca919 | 2017-03-21 05:45:18 -0700 | [diff] [blame] | 247 | return 0; |
| 248 | } |
| 249 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 250 | if (packet.codec == kVideoCodecH264) { |
| 251 | // Only when we start to receive packets will we know what payload type |
| 252 | // that will be used. When we know the payload type insert the correct |
| 253 | // sps/pps into the tracker. |
| 254 | if (packet.payloadType != last_payload_type_) { |
| 255 | last_payload_type_ = packet.payloadType; |
| 256 | InsertSpsPpsIntoTracker(packet.payloadType); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 257 | } |
| 258 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 259 | switch (tracker_.CopyAndFixBitstream(&packet)) { |
| 260 | case video_coding::H264SpsPpsTracker::kRequestKeyframe: |
| 261 | keyframe_request_sender_->RequestKeyFrame(); |
Karl Wiberg | 80ba333 | 2018-02-05 10:33:35 +0100 | [diff] [blame] | 262 | RTC_FALLTHROUGH(); |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 263 | case video_coding::H264SpsPpsTracker::kDrop: |
| 264 | return 0; |
| 265 | case video_coding::H264SpsPpsTracker::kInsert: |
| 266 | break; |
| 267 | } |
| 268 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 269 | } else { |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 270 | uint8_t* data = new uint8_t[packet.sizeBytes]; |
| 271 | memcpy(data, packet.dataPtr, packet.sizeBytes); |
| 272 | packet.dataPtr = data; |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 273 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 274 | |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 275 | packet.generic_descriptor = generic_descriptor; |
| 276 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 277 | packet_buffer_->InsertPacket(&packet); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 278 | return 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 279 | } |
| 280 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 281 | void RtpVideoStreamReceiver::OnRecoveredPacket(const uint8_t* rtp_packet, |
| 282 | size_t rtp_packet_length) { |
Niels Möller | b0573bc | 2017-09-25 10:47:00 +0200 | [diff] [blame] | 283 | RtpPacketReceived packet; |
| 284 | if (!packet.Parse(rtp_packet, rtp_packet_length)) |
nisse | 30e8931 | 2017-05-29 08:16:37 -0700 | [diff] [blame] | 285 | return; |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 286 | if (packet.PayloadType() == config_.rtp.red_payload_type) { |
| 287 | RTC_LOG(LS_WARNING) << "Discarding recovered packet with RED encapsulation"; |
| 288 | return; |
| 289 | } |
| 290 | |
Niels Möller | b0573bc | 2017-09-25 10:47:00 +0200 | [diff] [blame] | 291 | packet.IdentifyExtensions(rtp_header_extensions_); |
| 292 | packet.set_payload_type_frequency(kVideoPayloadTypeFrequency); |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 293 | // TODO(nisse): UlpfecReceiverImpl::ProcessReceivedFec passes both |
| 294 | // original (decapsulated) media packets and recovered packets to |
| 295 | // this callback. We need a way to distinguish, for setting |
| 296 | // packet.recovered() correctly. Ideally, move RED decapsulation out |
| 297 | // of the Ulpfec implementation. |
Niels Möller | b0573bc | 2017-09-25 10:47:00 +0200 | [diff] [blame] | 298 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 299 | ReceivePacket(packet); |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 300 | } |
| 301 | |
nisse | d2ef314 | 2017-05-11 08:00:58 -0700 | [diff] [blame] | 302 | // This method handles both regular RTP packets and packets recovered |
| 303 | // via FlexFEC. |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 304 | void RtpVideoStreamReceiver::OnRtpPacket(const RtpPacketReceived& packet) { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 305 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 306 | |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 307 | if (!receiving_) { |
| 308 | return; |
| 309 | } |
solenberg@webrtc.org | fc32046 | 2014-02-11 15:27:49 +0000 | [diff] [blame] | 310 | |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 311 | if (!packet.recovered()) { |
Jonas Oreland | 49ac595 | 2018-09-26 16:04:32 +0200 | [diff] [blame] | 312 | // TODO(nisse): Exclude out-of-order packets? |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 313 | int64_t now_ms = clock_->TimeInMilliseconds(); |
Niels Möller | b0d4b41 | 2018-08-28 13:58:15 +0200 | [diff] [blame] | 314 | { |
Jonas Oreland | 49ac595 | 2018-09-26 16:04:32 +0200 | [diff] [blame] | 315 | rtc::CritScope cs(&rtp_sources_lock_); |
Niels Möller | b0d4b41 | 2018-08-28 13:58:15 +0200 | [diff] [blame] | 316 | last_received_rtp_timestamp_ = packet.Timestamp(); |
| 317 | last_received_rtp_system_time_ms_ = now_ms; |
Jonas Oreland | 49ac595 | 2018-09-26 16:04:32 +0200 | [diff] [blame] | 318 | |
| 319 | std::vector<uint32_t> csrcs = packet.Csrcs(); |
| 320 | contributing_sources_.Update(now_ms, csrcs); |
Niels Möller | b0d4b41 | 2018-08-28 13:58:15 +0200 | [diff] [blame] | 321 | } |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 322 | // Periodically log the RTP header of incoming packets. |
| 323 | if (now_ms - last_packet_log_ms_ > kPacketLogIntervalMs) { |
Jonas Olsson | 366a50c | 2018-09-06 13:41:30 +0200 | [diff] [blame] | 324 | rtc::StringBuilder ss; |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 325 | ss << "Packet received on SSRC: " << packet.Ssrc() |
| 326 | << " with payload type: " << static_cast<int>(packet.PayloadType()) |
| 327 | << ", timestamp: " << packet.Timestamp() |
| 328 | << ", sequence number: " << packet.SequenceNumber() |
| 329 | << ", arrival time: " << packet.arrival_time_ms(); |
| 330 | int32_t time_offset; |
| 331 | if (packet.GetExtension<TransmissionOffset>(&time_offset)) { |
| 332 | ss << ", toffset: " << time_offset; |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 333 | } |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 334 | uint32_t send_time; |
| 335 | if (packet.GetExtension<AbsoluteSendTime>(&send_time)) { |
| 336 | ss << ", abs send time: " << send_time; |
| 337 | } |
Mirko Bonadei | 675513b | 2017-11-09 11:09:25 +0100 | [diff] [blame] | 338 | RTC_LOG(LS_INFO) << ss.str(); |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 339 | last_packet_log_ms_ = now_ms; |
stefan@webrtc.org | eb24b04 | 2014-10-14 11:40:13 +0000 | [diff] [blame] | 340 | } |
| 341 | } |
wu@webrtc.org | a989080 | 2013-12-13 00:21:03 +0000 | [diff] [blame] | 342 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 343 | ReceivePacket(packet); |
nisse | 38cc1d6 | 2017-02-13 05:59:46 -0800 | [diff] [blame] | 344 | |
asapersson@webrtc.org | 1457b47 | 2014-05-26 13:06:04 +0000 | [diff] [blame] | 345 | // Update receive statistics after ReceivePacket. |
| 346 | // Receive statistics will be reset if the payload type changes (make sure |
| 347 | // that the first packet is included in the stats). |
nisse | d2ef314 | 2017-05-11 08:00:58 -0700 | [diff] [blame] | 348 | if (!packet.recovered()) { |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 349 | rtp_receive_statistics_->OnRtpPacket(packet); |
nisse | d2ef314 | 2017-05-11 08:00:58 -0700 | [diff] [blame] | 350 | } |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 351 | |
| 352 | for (RtpPacketSinkInterface* secondary_sink : secondary_sinks_) { |
| 353 | secondary_sink->OnRtpPacket(packet); |
| 354 | } |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 355 | } |
| 356 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 357 | int32_t RtpVideoStreamReceiver::RequestKeyFrame() { |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 358 | return rtp_rtcp_->RequestKeyFrame(); |
| 359 | } |
| 360 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 361 | bool RtpVideoStreamReceiver::IsUlpfecEnabled() const { |
nisse | 3b3622f | 2017-09-26 02:49:21 -0700 | [diff] [blame] | 362 | return config_.rtp.ulpfec_payload_type != -1; |
brandtr | e6f98c7 | 2016-11-11 03:28:30 -0800 | [diff] [blame] | 363 | } |
| 364 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 365 | bool RtpVideoStreamReceiver::IsRetransmissionsEnabled() const { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 366 | return config_.rtp.nack.rtp_history_ms > 0; |
| 367 | } |
| 368 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 369 | void RtpVideoStreamReceiver::RequestPacketRetransmit( |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 370 | const std::vector<uint16_t>& sequence_numbers) { |
| 371 | rtp_rtcp_->SendNack(sequence_numbers); |
| 372 | } |
| 373 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 374 | int32_t RtpVideoStreamReceiver::ResendPackets(const uint16_t* sequence_numbers, |
| 375 | uint16_t length) { |
mflodman | cfc8e3b | 2016-05-03 21:22:04 -0700 | [diff] [blame] | 376 | return rtp_rtcp_->SendNACK(sequence_numbers, length); |
| 377 | } |
| 378 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 379 | void RtpVideoStreamReceiver::OnReceivedFrame( |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 380 | std::unique_ptr<video_coding::RtpFrameObject> frame) { |
Benjamin Wright | 192eeec | 2018-10-17 17:27:25 -0700 | [diff] [blame] | 381 | // Optionally attempt to decrypt the raw video frame if it was provided. |
| 382 | if (frame_decryptor_ != nullptr) { |
| 383 | // When using encryption we expect the frame to have the generic descriptor. |
| 384 | absl::optional<RtpGenericFrameDescriptor> descriptor = |
| 385 | frame->GetGenericFrameDescriptor(); |
| 386 | if (!descriptor) { |
| 387 | RTC_LOG(LS_ERROR) << "No generic frame descriptor found dropping frame."; |
| 388 | return; |
| 389 | } |
| 390 | |
| 391 | // Retrieve the bitstream of the encrypted video frame. |
| 392 | rtc::ArrayView<const uint8_t> encrypted_frame_bitstream(frame->Buffer(), |
| 393 | frame->size()); |
| 394 | // Retrieve the maximum possible size of the decrypted payload. |
| 395 | const size_t max_plaintext_byte_size = |
| 396 | frame_decryptor_->GetMaxPlaintextByteSize(cricket::MEDIA_TYPE_VIDEO, |
| 397 | frame->size()); |
| 398 | RTC_CHECK(max_plaintext_byte_size <= frame->size()); |
| 399 | // Place the decrypted frame inline into the existing frame. |
| 400 | rtc::ArrayView<uint8_t> inline_decrypted_bitstream(frame->MutableBuffer(), |
| 401 | max_plaintext_byte_size); |
| 402 | |
| 403 | // Attempt to decrypt the video frame. |
| 404 | size_t bytes_written = 0; |
| 405 | if (frame_decryptor_->Decrypt( |
| 406 | cricket::MEDIA_TYPE_VIDEO, /*csrcs=*/{}, |
| 407 | /*additional_data=*/nullptr, encrypted_frame_bitstream, |
| 408 | inline_decrypted_bitstream, &bytes_written) != 0) { |
| 409 | return; |
| 410 | } |
| 411 | RTC_CHECK(bytes_written <= max_plaintext_byte_size); |
| 412 | // Update the frame to contain just the written bytes. |
| 413 | frame->SetLength(bytes_written); |
| 414 | } else if (config_.crypto_options.sframe.require_frame_encryption) { |
| 415 | RTC_LOG(LS_WARNING) << "Frame decryption required but not attached to this " |
| 416 | "stream. Dropping frame."; |
| 417 | return; |
| 418 | } |
| 419 | |
philipel | 2c53b13 | 2017-05-16 08:06:30 -0700 | [diff] [blame] | 420 | if (!has_received_frame_) { |
| 421 | has_received_frame_ = true; |
| 422 | if (frame->FrameType() != kVideoFrameKey) |
| 423 | keyframe_request_sender_->RequestKeyFrame(); |
| 424 | } |
| 425 | |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 426 | reference_finder_->ManageFrame(std::move(frame)); |
| 427 | } |
| 428 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 429 | void RtpVideoStreamReceiver::OnCompleteFrame( |
philipel | e7c891f | 2018-02-22 14:35:06 +0100 | [diff] [blame] | 430 | std::unique_ptr<video_coding::EncodedFrame> frame) { |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 431 | { |
| 432 | rtc::CritScope lock(&last_seq_num_cs_); |
| 433 | video_coding::RtpFrameObject* rtp_frame = |
| 434 | static_cast<video_coding::RtpFrameObject*>(frame.get()); |
philipel | 0fa82a6 | 2018-03-19 15:34:53 +0100 | [diff] [blame] | 435 | last_seq_num_for_pic_id_[rtp_frame->id.picture_id] = |
| 436 | rtp_frame->last_seq_num(); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 437 | } |
| 438 | complete_frame_callback_->OnCompleteFrame(std::move(frame)); |
| 439 | } |
| 440 | |
Tommi | 81de14f | 2018-03-25 22:19:25 +0200 | [diff] [blame] | 441 | void RtpVideoStreamReceiver::UpdateRtt(int64_t max_rtt_ms) { |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 442 | if (nack_module_) |
| 443 | nack_module_->UpdateRtt(max_rtt_ms); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 444 | } |
| 445 | |
Danil Chapovalov | b9b146c | 2018-06-15 12:28:07 +0200 | [diff] [blame] | 446 | absl::optional<int64_t> RtpVideoStreamReceiver::LastReceivedPacketMs() const { |
philipel | 3184f8e | 2017-05-18 08:08:53 -0700 | [diff] [blame] | 447 | return packet_buffer_->LastReceivedPacketMs(); |
| 448 | } |
| 449 | |
Danil Chapovalov | b9b146c | 2018-06-15 12:28:07 +0200 | [diff] [blame] | 450 | absl::optional<int64_t> RtpVideoStreamReceiver::LastReceivedKeyframePacketMs() |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 451 | const { |
philipel | 3184f8e | 2017-05-18 08:08:53 -0700 | [diff] [blame] | 452 | return packet_buffer_->LastReceivedKeyframePacketMs(); |
| 453 | } |
| 454 | |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 455 | void RtpVideoStreamReceiver::AddSecondarySink(RtpPacketSinkInterface* sink) { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 456 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 457 | RTC_DCHECK(std::find(secondary_sinks_.cbegin(), secondary_sinks_.cend(), |
| 458 | sink) == secondary_sinks_.cend()); |
| 459 | secondary_sinks_.push_back(sink); |
| 460 | } |
| 461 | |
| 462 | void RtpVideoStreamReceiver::RemoveSecondarySink( |
| 463 | const RtpPacketSinkInterface* sink) { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 464 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 465 | auto it = std::find(secondary_sinks_.begin(), secondary_sinks_.end(), sink); |
| 466 | if (it == secondary_sinks_.end()) { |
| 467 | // We might be rolling-back a call whose setup failed mid-way. In such a |
| 468 | // case, it's simpler to remove "everything" rather than remember what |
| 469 | // has already been added. |
Mirko Bonadei | 675513b | 2017-11-09 11:09:25 +0100 | [diff] [blame] | 470 | RTC_LOG(LS_WARNING) << "Removal of unknown sink."; |
eladalon | c0d481a | 2017-08-02 07:39:07 -0700 | [diff] [blame] | 471 | return; |
| 472 | } |
| 473 | secondary_sinks_.erase(it); |
| 474 | } |
| 475 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 476 | void RtpVideoStreamReceiver::ReceivePacket(const RtpPacketReceived& packet) { |
| 477 | if (packet.payload_size() == 0) { |
Niels Möller | 0b92678 | 2018-08-21 17:49:24 +0200 | [diff] [blame] | 478 | // Padding or keep-alive packet. |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 479 | // TODO(nisse): Could drop empty packets earlier, but need to figure out how |
| 480 | // they should be counted in stats. |
Niels Möller | 0b92678 | 2018-08-21 17:49:24 +0200 | [diff] [blame] | 481 | NotifyReceiverOfEmptyPacket(packet.SequenceNumber()); |
nisse | 30e8931 | 2017-05-29 08:16:37 -0700 | [diff] [blame] | 482 | return; |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 483 | } |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 484 | if (packet.PayloadType() == config_.rtp.red_payload_type) { |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 485 | ParseAndHandleEncapsulatingHeader(packet); |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 486 | return; |
wu@webrtc.org | 822fbd8 | 2013-08-15 23:38:54 +0000 | [diff] [blame] | 487 | } |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 488 | |
| 489 | const auto codec_type_it = pt_codec_type_.find(packet.PayloadType()); |
| 490 | if (codec_type_it == pt_codec_type_.end()) { |
| 491 | return; |
| 492 | } |
| 493 | auto depacketizer = |
| 494 | absl::WrapUnique(RtpDepacketizer::Create(codec_type_it->second)); |
| 495 | |
| 496 | if (!depacketizer) { |
| 497 | RTC_LOG(LS_ERROR) << "Failed to create depacketizer."; |
| 498 | return; |
| 499 | } |
| 500 | RtpDepacketizer::ParsedPayload parsed_payload; |
| 501 | if (!depacketizer->Parse(&parsed_payload, packet.payload().data(), |
| 502 | packet.payload().size())) { |
| 503 | RTC_LOG(LS_WARNING) << "Failed parsing payload."; |
| 504 | return; |
| 505 | } |
| 506 | |
| 507 | WebRtcRTPHeader webrtc_rtp_header = {}; |
| 508 | packet.GetHeader(&webrtc_rtp_header.header); |
| 509 | |
| 510 | webrtc_rtp_header.frameType = parsed_payload.frame_type; |
| 511 | webrtc_rtp_header.video_header() = parsed_payload.video_header(); |
| 512 | webrtc_rtp_header.video_header().rotation = kVideoRotation_0; |
| 513 | webrtc_rtp_header.video_header().content_type = VideoContentType::UNSPECIFIED; |
| 514 | webrtc_rtp_header.video_header().video_timing.flags = |
| 515 | VideoSendTiming::kInvalid; |
| 516 | webrtc_rtp_header.video_header().playout_delay.min_ms = -1; |
| 517 | webrtc_rtp_header.video_header().playout_delay.max_ms = -1; |
philipel | ef615ea | 2018-09-13 11:07:48 +0200 | [diff] [blame] | 518 | webrtc_rtp_header.video_header().is_last_packet_in_frame = |
| 519 | webrtc_rtp_header.header.markerBit; |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 520 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 521 | packet.GetExtension<VideoOrientation>( |
| 522 | &webrtc_rtp_header.video_header().rotation); |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 523 | packet.GetExtension<VideoContentTypeExtension>( |
| 524 | &webrtc_rtp_header.video_header().content_type); |
| 525 | packet.GetExtension<VideoTimingExtension>( |
| 526 | &webrtc_rtp_header.video_header().video_timing); |
| 527 | packet.GetExtension<PlayoutDelayLimits>( |
| 528 | &webrtc_rtp_header.video_header().playout_delay); |
| 529 | |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 530 | absl::optional<RtpGenericFrameDescriptor> generic_descriptor_wire; |
| 531 | generic_descriptor_wire.emplace(); |
philipel | b3e42a4 | 2018-09-13 10:57:14 +0200 | [diff] [blame] | 532 | if (packet.GetExtension<RtpGenericFrameDescriptorExtension>( |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 533 | &generic_descriptor_wire.value())) { |
| 534 | generic_descriptor_wire->SetByteRepresentation( |
| 535 | packet.GetRawExtension<RtpGenericFrameDescriptorExtension>()); |
philipel | b3e42a4 | 2018-09-13 10:57:14 +0200 | [diff] [blame] | 536 | webrtc_rtp_header.video_header().is_first_packet_in_frame = |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 537 | generic_descriptor_wire->FirstSubFrameInFrame() && |
| 538 | generic_descriptor_wire->FirstPacketInSubFrame(); |
philipel | ef615ea | 2018-09-13 11:07:48 +0200 | [diff] [blame] | 539 | webrtc_rtp_header.video_header().is_last_packet_in_frame = |
| 540 | webrtc_rtp_header.header.markerBit || |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 541 | (generic_descriptor_wire->LastSubFrameInFrame() && |
| 542 | generic_descriptor_wire->LastPacketInSubFrame()); |
philipel | fab9129 | 2018-10-17 14:36:08 +0200 | [diff] [blame] | 543 | |
| 544 | if (generic_descriptor_wire->FirstPacketInSubFrame()) { |
| 545 | webrtc_rtp_header.frameType = |
| 546 | generic_descriptor_wire->FrameDependenciesDiffs().empty() |
| 547 | ? kVideoFrameKey |
| 548 | : kVideoFrameDelta; |
| 549 | } |
| 550 | |
| 551 | webrtc_rtp_header.video_header().width = generic_descriptor_wire->Width(); |
| 552 | webrtc_rtp_header.video_header().height = generic_descriptor_wire->Height(); |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 553 | } else { |
| 554 | generic_descriptor_wire.reset(); |
philipel | b3e42a4 | 2018-09-13 10:57:14 +0200 | [diff] [blame] | 555 | } |
| 556 | |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 557 | OnReceivedPayloadData(parsed_payload.payload, parsed_payload.payload_length, |
philipel | 2837edc | 2018-10-02 13:55:47 +0200 | [diff] [blame] | 558 | &webrtc_rtp_header, generic_descriptor_wire); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 559 | } |
| 560 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 561 | void RtpVideoStreamReceiver::ParseAndHandleEncapsulatingHeader( |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 562 | const RtpPacketReceived& packet) { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 563 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 564 | if (packet.PayloadType() == config_.rtp.red_payload_type && |
| 565 | packet.payload_size() > 0) { |
| 566 | if (packet.payload()[0] == config_.rtp.ulpfec_payload_type) { |
| 567 | rtp_receive_statistics_->FecPacketReceived(packet); |
Peter Boström | 0b25072 | 2016-04-22 18:23:15 +0200 | [diff] [blame] | 568 | // Notify video_receiver about received FEC packets to avoid NACKing these |
| 569 | // packets. |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 570 | NotifyReceiverOfEmptyPacket(packet.SequenceNumber()); |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 571 | } |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 572 | RTPHeader header; |
| 573 | packet.GetHeader(&header); |
Niels Möller | 3f027b3 | 2018-03-14 08:04:58 +0100 | [diff] [blame] | 574 | if (ulpfec_receiver_->AddReceivedRedPacket( |
Niels Möller | 1f3206c | 2018-09-14 08:26:32 +0200 | [diff] [blame] | 575 | header, packet.data(), packet.size(), |
| 576 | config_.rtp.ulpfec_payload_type) != 0) { |
nisse | 30e8931 | 2017-05-29 08:16:37 -0700 | [diff] [blame] | 577 | return; |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 578 | } |
nisse | 30e8931 | 2017-05-29 08:16:37 -0700 | [diff] [blame] | 579 | ulpfec_receiver_->ProcessReceivedFec(); |
stefan@webrtc.org | 7bb8f02 | 2013-09-06 13:40:11 +0000 | [diff] [blame] | 580 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 581 | } |
| 582 | |
Niels Möller | bc01047 | 2018-03-23 13:22:29 +0100 | [diff] [blame] | 583 | // In the case of a video stream without picture ids and no rtx the |
| 584 | // RtpFrameReferenceFinder will need to know about padding to |
| 585 | // correctly calculate frame references. |
| 586 | void RtpVideoStreamReceiver::NotifyReceiverOfEmptyPacket(uint16_t seq_num) { |
| 587 | reference_finder_->PaddingReceived(seq_num); |
| 588 | packet_buffer_->PaddingReceived(seq_num); |
Niels Möller | bc01047 | 2018-03-23 13:22:29 +0100 | [diff] [blame] | 589 | if (nack_module_) { |
Niels Möller | 0b92678 | 2018-08-21 17:49:24 +0200 | [diff] [blame] | 590 | nack_module_->OnReceivedPacket(seq_num, /* is_keyframe = */ false); |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 591 | } |
asapersson@webrtc.org | 37c0559 | 2015-01-28 13:58:27 +0000 | [diff] [blame] | 592 | } |
| 593 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 594 | bool RtpVideoStreamReceiver::DeliverRtcp(const uint8_t* rtcp_packet, |
| 595 | size_t rtcp_packet_length) { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 596 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
| 597 | |
| 598 | if (!receiving_) { |
| 599 | return false; |
Peter Boström | 4fa7eca | 2016-03-02 15:05:53 +0100 | [diff] [blame] | 600 | } |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 601 | |
Per | 83d0910 | 2016-04-15 14:59:13 +0200 | [diff] [blame] | 602 | rtp_rtcp_->IncomingRtcpPacket(rtcp_packet, rtcp_packet_length); |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 603 | |
pkasting@chromium.org | 16825b1 | 2015-01-12 21:51:21 +0000 | [diff] [blame] | 604 | int64_t rtt = 0; |
Niels Möller | 2ff1f2a | 2018-08-09 16:16:34 +0200 | [diff] [blame] | 605 | rtp_rtcp_->RTT(config_.rtp.remote_ssrc, &rtt, nullptr, nullptr, nullptr); |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 606 | if (rtt == 0) { |
| 607 | // Waiting for valid rtt. |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 608 | return true; |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 609 | } |
| 610 | uint32_t ntp_secs = 0; |
| 611 | uint32_t ntp_frac = 0; |
| 612 | uint32_t rtp_timestamp = 0; |
Ilya Nikolaevskiy | 7172ea1 | 2017-10-30 11:17:34 +0100 | [diff] [blame] | 613 | uint32_t recieved_ntp_secs = 0; |
| 614 | uint32_t recieved_ntp_frac = 0; |
| 615 | if (rtp_rtcp_->RemoteNTP(&ntp_secs, &ntp_frac, &recieved_ntp_secs, |
| 616 | &recieved_ntp_frac, &rtp_timestamp) != 0) { |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 617 | // Waiting for RTCP. |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 618 | return true; |
minyue@webrtc.org | 2c0cdbc | 2014-10-09 10:52:43 +0000 | [diff] [blame] | 619 | } |
Ilya Nikolaevskiy | 7172ea1 | 2017-10-30 11:17:34 +0100 | [diff] [blame] | 620 | NtpTime recieved_ntp(recieved_ntp_secs, recieved_ntp_frac); |
| 621 | int64_t time_since_recieved = |
| 622 | clock_->CurrentNtpInMilliseconds() - recieved_ntp.ToMs(); |
| 623 | // Don't use old SRs to estimate time. |
| 624 | if (time_since_recieved <= 1) { |
| 625 | ntp_estimator_.UpdateRtcpTimestamp(rtt, ntp_secs, ntp_frac, rtp_timestamp); |
| 626 | } |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 627 | |
Peter Boström | d1d66ba | 2016-02-08 14:07:14 +0100 | [diff] [blame] | 628 | return true; |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 629 | } |
| 630 | |
philipel | d4fac69 | 2017-09-04 07:03:46 -0700 | [diff] [blame] | 631 | void RtpVideoStreamReceiver::FrameContinuous(int64_t picture_id) { |
tommi | f284b7f | 2017-02-27 01:59:36 -0800 | [diff] [blame] | 632 | if (!nack_module_) |
| 633 | return; |
| 634 | |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 635 | int seq_num = -1; |
| 636 | { |
| 637 | rtc::CritScope lock(&last_seq_num_cs_); |
| 638 | auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id); |
| 639 | if (seq_num_it != last_seq_num_for_pic_id_.end()) |
| 640 | seq_num = seq_num_it->second; |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 641 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 642 | if (seq_num != -1) |
| 643 | nack_module_->ClearUpTo(seq_num); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 644 | } |
| 645 | |
philipel | d4fac69 | 2017-09-04 07:03:46 -0700 | [diff] [blame] | 646 | void RtpVideoStreamReceiver::FrameDecoded(int64_t picture_id) { |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 647 | int seq_num = -1; |
| 648 | { |
| 649 | rtc::CritScope lock(&last_seq_num_cs_); |
| 650 | auto seq_num_it = last_seq_num_for_pic_id_.find(picture_id); |
| 651 | if (seq_num_it != last_seq_num_for_pic_id_.end()) { |
| 652 | seq_num = seq_num_it->second; |
| 653 | last_seq_num_for_pic_id_.erase(last_seq_num_for_pic_id_.begin(), |
| 654 | ++seq_num_it); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 655 | } |
philipel | a45102f | 2017-02-22 05:30:39 -0800 | [diff] [blame] | 656 | } |
| 657 | if (seq_num != -1) { |
| 658 | packet_buffer_->ClearTo(seq_num); |
| 659 | reference_finder_->ClearTo(seq_num); |
philipel | fd5a20f | 2016-11-15 00:57:57 -0800 | [diff] [blame] | 660 | } |
| 661 | } |
| 662 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 663 | void RtpVideoStreamReceiver::SignalNetworkState(NetworkState state) { |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 664 | rtp_rtcp_->SetRTCPStatus(state == kNetworkUp ? config_.rtp.rtcp_mode |
| 665 | : RtcpMode::kOff); |
| 666 | } |
| 667 | |
Ilya Nikolaevskiy | d397a0d | 2018-02-21 15:57:09 +0100 | [diff] [blame] | 668 | int RtpVideoStreamReceiver::GetUniqueFramesSeen() const { |
| 669 | return packet_buffer_->GetUniqueFramesSeen(); |
| 670 | } |
| 671 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 672 | void RtpVideoStreamReceiver::StartReceive() { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 673 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 674 | receiving_ = true; |
| 675 | } |
| 676 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 677 | void RtpVideoStreamReceiver::StopReceive() { |
eladalon | 8b07305 | 2017-08-25 00:49:08 -0700 | [diff] [blame] | 678 | RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 679 | receiving_ = false; |
| 680 | } |
| 681 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 682 | void RtpVideoStreamReceiver::UpdateHistograms() { |
brandtr | d55c3f6 | 2016-10-31 04:51:33 -0700 | [diff] [blame] | 683 | FecPacketCounter counter = ulpfec_receiver_->GetPacketCounter(); |
asapersson | 0c43f77 | 2016-11-30 01:42:26 -0800 | [diff] [blame] | 684 | if (counter.first_packet_time_ms == -1) |
| 685 | return; |
| 686 | |
| 687 | int64_t elapsed_sec = |
| 688 | (clock_->TimeInMilliseconds() - counter.first_packet_time_ms) / 1000; |
| 689 | if (elapsed_sec < metrics::kMinRunTimeInSeconds) |
| 690 | return; |
| 691 | |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 692 | if (counter.num_packets > 0) { |
asapersson | 1d02d3e | 2016-09-09 22:40:25 -0700 | [diff] [blame] | 693 | RTC_HISTOGRAM_PERCENTAGE( |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 694 | "WebRTC.Video.ReceivedFecPacketsInPercent", |
| 695 | static_cast<int>(counter.num_fec_packets * 100 / counter.num_packets)); |
| 696 | } |
| 697 | if (counter.num_fec_packets > 0) { |
asapersson | 1d02d3e | 2016-09-09 22:40:25 -0700 | [diff] [blame] | 698 | RTC_HISTOGRAM_PERCENTAGE("WebRTC.Video.RecoveredMediaPacketsInPercentOfFec", |
| 699 | static_cast<int>(counter.num_recovered_packets * |
| 700 | 100 / counter.num_fec_packets)); |
mflodman | dc7d0d2 | 2016-05-06 05:32:22 -0700 | [diff] [blame] | 701 | } |
| 702 | } |
| 703 | |
nisse | b1f2ff9 | 2017-06-09 04:01:55 -0700 | [diff] [blame] | 704 | void RtpVideoStreamReceiver::InsertSpsPpsIntoTracker(uint8_t payload_type) { |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 705 | auto codec_params_it = pt_codec_params_.find(payload_type); |
| 706 | if (codec_params_it == pt_codec_params_.end()) |
| 707 | return; |
| 708 | |
Mirko Bonadei | 675513b | 2017-11-09 11:09:25 +0100 | [diff] [blame] | 709 | RTC_LOG(LS_INFO) << "Found out of band supplied codec parameters for" |
| 710 | << " payload type: " << static_cast<int>(payload_type); |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 711 | |
| 712 | H264SpropParameterSets sprop_decoder; |
| 713 | auto sprop_base64_it = |
| 714 | codec_params_it->second.find(cricket::kH264FmtpSpropParameterSets); |
| 715 | |
| 716 | if (sprop_base64_it == codec_params_it->second.end()) |
| 717 | return; |
| 718 | |
johan | 62d02c3 | 2017-01-24 04:38:27 -0800 | [diff] [blame] | 719 | if (!sprop_decoder.DecodeSprop(sprop_base64_it->second.c_str())) |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 720 | return; |
| 721 | |
johan | d2b092f | 2017-01-24 02:38:17 -0800 | [diff] [blame] | 722 | tracker_.InsertSpsPpsNalus(sprop_decoder.sps_nalu(), |
| 723 | sprop_decoder.pps_nalu()); |
philipel | 022b54e | 2016-12-20 04:15:59 -0800 | [diff] [blame] | 724 | } |
| 725 | |
Jonas Oreland | 49ac595 | 2018-09-26 16:04:32 +0200 | [diff] [blame] | 726 | std::vector<webrtc::RtpSource> RtpVideoStreamReceiver::GetSources() const { |
| 727 | int64_t now_ms = rtc::TimeMillis(); |
| 728 | std::vector<RtpSource> sources; |
| 729 | { |
| 730 | rtc::CritScope cs(&rtp_sources_lock_); |
| 731 | sources = contributing_sources_.GetSources(now_ms); |
| 732 | if (last_received_rtp_system_time_ms_ >= |
| 733 | now_ms - ContributingSources::kHistoryMs) { |
| 734 | sources.emplace_back(*last_received_rtp_system_time_ms_, |
| 735 | config_.rtp.remote_ssrc, RtpSourceType::SSRC); |
| 736 | } |
| 737 | } |
| 738 | return sources; |
| 739 | } |
| 740 | |
mflodman@webrtc.org | ad4ee36 | 2011-11-28 22:39:24 +0000 | [diff] [blame] | 741 | } // namespace webrtc |