blob: 136c11cb189dbf96bb9506aca85e82802715eb4b [file] [log] [blame]
Tommi3a5742c2020-05-20 09:32:51 +02001/*
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
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
11#include "modules/rtp_rtcp/source/rtp_rtcp_impl2.h"
12
13#include <string.h>
14
15#include <algorithm>
16#include <cstdint>
17#include <memory>
18#include <set>
19#include <string>
20#include <utility>
21
Markus Handellc6b9ac72021-06-18 13:44:51 +020022#include "absl/types/optional.h"
Markus Handell885d5382021-06-21 18:57:36 +020023#include "api/sequence_checker.h"
Tommi3a5742c2020-05-20 09:32:51 +020024#include "api/transport/field_trial_based_config.h"
Markus Handell885d5382021-06-21 18:57:36 +020025#include "api/units/time_delta.h"
Markus Handellc6b9ac72021-06-18 13:44:51 +020026#include "api/units/timestamp.h"
Tommi3a5742c2020-05-20 09:32:51 +020027#include "modules/rtp_rtcp/source/rtcp_packet/dlrr.h"
28#include "modules/rtp_rtcp/source/rtp_rtcp_config.h"
29#include "rtc_base/checks.h"
30#include "rtc_base/logging.h"
Markus Handell885d5382021-06-21 18:57:36 +020031#include "rtc_base/task_utils/to_queued_task.h"
32#include "rtc_base/time_utils.h"
Alessio Bazzicabc1c93d2021-03-12 17:45:26 +010033#include "system_wrappers/include/ntp_time.h"
Tommi3a5742c2020-05-20 09:32:51 +020034
35#ifdef _WIN32
36// Disable warning C4355: 'this' : used in base member initializer list.
37#pragma warning(disable : 4355)
38#endif
39
40namespace webrtc {
41namespace {
Tommi3a5742c2020-05-20 09:32:51 +020042const int64_t kDefaultExpectedRetransmissionTimeMs = 125;
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +020043
44constexpr TimeDelta kRttUpdateInterval = TimeDelta::Millis(1000);
Markus Handell885d5382021-06-21 18:57:36 +020045
46RTCPSender::Configuration AddRtcpSendEvaluationCallback(
47 RTCPSender::Configuration config,
48 std::function<void(TimeDelta)> send_evaluation_callback) {
49 config.schedule_next_rtcp_send_evaluation_function =
50 std::move(send_evaluation_callback);
51 return config;
52}
53
54int DelayMillisForDuration(TimeDelta duration) {
55 // TimeDelta::ms() rounds downwards sometimes which leads to too little time
Artem Titov913cfa72021-07-28 23:57:33 +020056 // slept. Account for this, unless `duration` is exactly representable in
Markus Handell885d5382021-06-21 18:57:36 +020057 // millisecs.
58 return (duration.us() + rtc::kNumMillisecsPerSec - 1) /
59 rtc::kNumMicrosecsPerMillisec;
60}
Tommi3a5742c2020-05-20 09:32:51 +020061} // namespace
62
63ModuleRtpRtcpImpl2::RtpSenderContext::RtpSenderContext(
Tomas Gunnarssonf25761d2020-06-03 22:55:33 +020064 const RtpRtcpInterface::Configuration& config)
Tommi3a5742c2020-05-20 09:32:51 +020065 : packet_history(config.clock, config.enable_rtx_padding_prioritization),
Erik Språngbb904972021-08-06 13:10:11 +020066 deferred_sequencing_(config.use_deferred_sequencing),
Erik Språngbfcfe032021-08-04 14:45:32 +020067 sequencer_(config.local_media_ssrc,
68 config.rtx_send_ssrc,
69 /*require_marker_before_media_padding=*/!config.audio,
70 config.clock),
Tommi3a5742c2020-05-20 09:32:51 +020071 packet_sender(config, &packet_history),
Erik Språngbb904972021-08-06 13:10:11 +020072 non_paced_sender(&packet_sender, this, config.use_deferred_sequencing),
Tommi3a5742c2020-05-20 09:32:51 +020073 packet_generator(
74 config,
75 &packet_history,
Erik Språngbfcfe032021-08-04 14:45:32 +020076 config.paced_sender ? config.paced_sender : &non_paced_sender,
Erik Språngbb904972021-08-06 13:10:11 +020077 config.use_deferred_sequencing ? nullptr : &sequencer_) {}
Erik Språng1d50cb62020-07-02 17:41:32 +020078void ModuleRtpRtcpImpl2::RtpSenderContext::AssignSequenceNumber(
79 RtpPacketToSend* packet) {
Erik Språngbb904972021-08-06 13:10:11 +020080 if (deferred_sequencing_) {
81 sequencer_.Sequence(*packet);
82 } else {
83 packet_generator.AssignSequenceNumber(packet);
84 }
Erik Språng1d50cb62020-07-02 17:41:32 +020085}
Tommi3a5742c2020-05-20 09:32:51 +020086
Tommi3a5742c2020-05-20 09:32:51 +020087ModuleRtpRtcpImpl2::ModuleRtpRtcpImpl2(const Configuration& configuration)
Niels Moller2accc7d2021-01-12 15:54:16 +000088 : worker_queue_(TaskQueueBase::Current()),
Markus Handell885d5382021-06-21 18:57:36 +020089 rtcp_sender_(AddRtcpSendEvaluationCallback(
90 RTCPSender::Configuration::FromRtpRtcpConfiguration(configuration),
91 [this](TimeDelta duration) {
92 ScheduleRtcpSendEvaluation(duration);
93 })),
Tommi3a5742c2020-05-20 09:32:51 +020094 rtcp_receiver_(configuration, this),
95 clock_(configuration.clock),
Tommi3a5742c2020-05-20 09:32:51 +020096 packet_overhead_(28), // IPV4 UDP.
97 nack_last_time_sent_full_ms_(0),
98 nack_last_seq_number_sent_(0),
99 remote_bitrate_(configuration.remote_bitrate_estimator),
100 rtt_stats_(configuration.rtt_stats),
101 rtt_ms_(0) {
Niels Moller2accc7d2021-01-12 15:54:16 +0000102 RTC_DCHECK(worker_queue_);
Tommi08be9ba2021-06-15 23:01:57 +0200103 packet_sequence_checker_.Detach();
Erik Språngbb904972021-08-06 13:10:11 +0200104 pacer_thread_checker_.Detach();
Tommi3a5742c2020-05-20 09:32:51 +0200105 if (!configuration.receiver_only) {
106 rtp_sender_ = std::make_unique<RtpSenderContext>(configuration);
107 // Make sure rtcp sender use same timestamp offset as rtp sender.
108 rtcp_sender_.SetTimestampOffset(
109 rtp_sender_->packet_generator.TimestampOffset());
110 }
111
112 // Set default packet size limit.
113 // TODO(nisse): Kind-of duplicates
114 // webrtc::VideoSendStream::Config::Rtp::kDefaultMaxPacketSize.
115 const size_t kTcpOverIpv4HeaderSize = 40;
116 SetMaxRtpPacketSize(IP_PACKET_SIZE - kTcpOverIpv4HeaderSize);
Ivo Creusen8c40d512021-07-13 12:53:22 +0000117 rtt_update_task_ = RepeatingTaskHandle::DelayedStart(
118 worker_queue_, kRttUpdateInterval, [this]() {
119 PeriodicUpdate();
120 return kRttUpdateInterval;
121 });
Tommi3a5742c2020-05-20 09:32:51 +0200122}
123
124ModuleRtpRtcpImpl2::~ModuleRtpRtcpImpl2() {
Niels Moller2accc7d2021-01-12 15:54:16 +0000125 RTC_DCHECK_RUN_ON(worker_queue_);
126 rtt_update_task_.Stop();
127}
128
129// static
130std::unique_ptr<ModuleRtpRtcpImpl2> ModuleRtpRtcpImpl2::Create(
131 const Configuration& configuration) {
132 RTC_DCHECK(configuration.clock);
133 RTC_DCHECK(TaskQueueBase::Current());
134 return std::make_unique<ModuleRtpRtcpImpl2>(configuration);
Tomas Gunnarssonfae05622020-06-03 08:54:39 +0200135}
136
Tommi3a5742c2020-05-20 09:32:51 +0200137void ModuleRtpRtcpImpl2::SetRtxSendStatus(int mode) {
138 rtp_sender_->packet_generator.SetRtxStatus(mode);
139}
140
141int ModuleRtpRtcpImpl2::RtxSendStatus() const {
142 return rtp_sender_ ? rtp_sender_->packet_generator.RtxStatus() : kRtxOff;
143}
144
145void ModuleRtpRtcpImpl2::SetRtxSendPayloadType(int payload_type,
146 int associated_payload_type) {
147 rtp_sender_->packet_generator.SetRtxPayloadType(payload_type,
148 associated_payload_type);
149}
150
151absl::optional<uint32_t> ModuleRtpRtcpImpl2::RtxSsrc() const {
152 return rtp_sender_ ? rtp_sender_->packet_generator.RtxSsrc() : absl::nullopt;
153}
154
155absl::optional<uint32_t> ModuleRtpRtcpImpl2::FlexfecSsrc() const {
156 if (rtp_sender_) {
157 return rtp_sender_->packet_generator.FlexfecSsrc();
158 }
159 return absl::nullopt;
160}
161
162void ModuleRtpRtcpImpl2::IncomingRtcpPacket(const uint8_t* rtcp_packet,
163 const size_t length) {
Tommi08be9ba2021-06-15 23:01:57 +0200164 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
Tommi3a5742c2020-05-20 09:32:51 +0200165 rtcp_receiver_.IncomingPacket(rtcp_packet, length);
166}
167
168void ModuleRtpRtcpImpl2::RegisterSendPayloadFrequency(int payload_type,
169 int payload_frequency) {
170 rtcp_sender_.SetRtpClockRate(payload_type, payload_frequency);
171}
172
173int32_t ModuleRtpRtcpImpl2::DeRegisterSendPayload(const int8_t payload_type) {
174 return 0;
175}
176
177uint32_t ModuleRtpRtcpImpl2::StartTimestamp() const {
178 return rtp_sender_->packet_generator.TimestampOffset();
179}
180
181// Configure start timestamp, default is a random number.
182void ModuleRtpRtcpImpl2::SetStartTimestamp(const uint32_t timestamp) {
183 rtcp_sender_.SetTimestampOffset(timestamp);
184 rtp_sender_->packet_generator.SetTimestampOffset(timestamp);
185 rtp_sender_->packet_sender.SetTimestampOffset(timestamp);
186}
187
188uint16_t ModuleRtpRtcpImpl2::SequenceNumber() const {
Erik Språngbb904972021-08-06 13:10:11 +0200189 if (rtp_sender_->deferred_sequencing_) {
190 return rtp_sender_->sequencer_.media_sequence_number();
191 }
Tommi3a5742c2020-05-20 09:32:51 +0200192 return rtp_sender_->packet_generator.SequenceNumber();
193}
194
195// Set SequenceNumber, default is a random number.
196void ModuleRtpRtcpImpl2::SetSequenceNumber(const uint16_t seq_num) {
Erik Språngbb904972021-08-06 13:10:11 +0200197 if (rtp_sender_->deferred_sequencing_) {
198 RTC_DCHECK_RUN_ON(&pacer_thread_checker_);
199 if (rtp_sender_->sequencer_.media_sequence_number() != seq_num) {
200 rtp_sender_->sequencer_.set_media_sequence_number(seq_num);
201 rtp_sender_->packet_history.Clear();
202 }
203 } else {
204 rtp_sender_->packet_generator.SetSequenceNumber(seq_num);
205 }
Tommi3a5742c2020-05-20 09:32:51 +0200206}
207
208void ModuleRtpRtcpImpl2::SetRtpState(const RtpState& rtp_state) {
209 rtp_sender_->packet_generator.SetRtpState(rtp_state);
Erik Språngbb904972021-08-06 13:10:11 +0200210 if (rtp_sender_->deferred_sequencing_) {
211 rtp_sender_->sequencer_.SetRtpState(rtp_state);
212 }
Tommi3a5742c2020-05-20 09:32:51 +0200213 rtcp_sender_.SetTimestampOffset(rtp_state.start_timestamp);
214}
215
216void ModuleRtpRtcpImpl2::SetRtxState(const RtpState& rtp_state) {
217 rtp_sender_->packet_generator.SetRtxRtpState(rtp_state);
Erik Språngbb904972021-08-06 13:10:11 +0200218 if (rtp_sender_->deferred_sequencing_) {
219 rtp_sender_->sequencer_.set_rtx_sequence_number(rtp_state.sequence_number);
220 }
Tommi3a5742c2020-05-20 09:32:51 +0200221}
222
223RtpState ModuleRtpRtcpImpl2::GetRtpState() const {
224 RtpState state = rtp_sender_->packet_generator.GetRtpState();
Erik Språngbb904972021-08-06 13:10:11 +0200225 if (rtp_sender_->deferred_sequencing_) {
226 rtp_sender_->sequencer_.PopulateRtpState(state);
227 }
Tommi3a5742c2020-05-20 09:32:51 +0200228 return state;
229}
230
231RtpState ModuleRtpRtcpImpl2::GetRtxState() const {
Erik Språngbb904972021-08-06 13:10:11 +0200232 RtpState state = rtp_sender_->packet_generator.GetRtxRtpState();
233 if (rtp_sender_->deferred_sequencing_) {
234 state.sequence_number = rtp_sender_->sequencer_.rtx_sequence_number();
235 }
236 return state;
Tommi3a5742c2020-05-20 09:32:51 +0200237}
238
Ivo Creusen8c40d512021-07-13 12:53:22 +0000239void ModuleRtpRtcpImpl2::SetNonSenderRttMeasurement(bool enabled) {
240 rtcp_sender_.SetNonSenderRttMeasurement(enabled);
241 rtcp_receiver_.SetNonSenderRttMeasurement(enabled);
242}
243
Tommi08be9ba2021-06-15 23:01:57 +0200244uint32_t ModuleRtpRtcpImpl2::local_media_ssrc() const {
245 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
246 RTC_DCHECK_EQ(rtcp_receiver_.local_media_ssrc(), rtcp_sender_.SSRC());
247 return rtcp_receiver_.local_media_ssrc();
248}
249
Tommi3a5742c2020-05-20 09:32:51 +0200250void ModuleRtpRtcpImpl2::SetRid(const std::string& rid) {
251 if (rtp_sender_) {
252 rtp_sender_->packet_generator.SetRid(rid);
253 }
254}
255
256void ModuleRtpRtcpImpl2::SetMid(const std::string& mid) {
257 if (rtp_sender_) {
258 rtp_sender_->packet_generator.SetMid(mid);
259 }
260 // TODO(bugs.webrtc.org/4050): If we end up supporting the MID SDES item for
261 // RTCP, this will need to be passed down to the RTCPSender also.
262}
263
264void ModuleRtpRtcpImpl2::SetCsrcs(const std::vector<uint32_t>& csrcs) {
265 rtcp_sender_.SetCsrcs(csrcs);
266 rtp_sender_->packet_generator.SetCsrcs(csrcs);
267}
268
269// TODO(pbos): Handle media and RTX streams separately (separate RTCP
270// feedbacks).
271RTCPSender::FeedbackState ModuleRtpRtcpImpl2::GetFeedbackState() {
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200272 // TODO(bugs.webrtc.org/11581): Called by potentially multiple threads.
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200273 // Mostly "Send*" methods. Make sure it's only called on the
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200274 // construction thread.
275
Tommi3a5742c2020-05-20 09:32:51 +0200276 RTCPSender::FeedbackState state;
277 // This is called also when receiver_only is true. Hence below
278 // checks that rtp_sender_ exists.
279 if (rtp_sender_) {
280 StreamDataCounters rtp_stats;
281 StreamDataCounters rtx_stats;
282 rtp_sender_->packet_sender.GetDataCounters(&rtp_stats, &rtx_stats);
283 state.packets_sent =
284 rtp_stats.transmitted.packets + rtx_stats.transmitted.packets;
285 state.media_bytes_sent = rtp_stats.transmitted.payload_bytes +
286 rtx_stats.transmitted.payload_bytes;
287 state.send_bitrate =
288 rtp_sender_->packet_sender.GetSendRates().Sum().bps<uint32_t>();
289 }
290 state.receiver = &rtcp_receiver_;
291
Alessio Bazzica79011ef2021-03-10 14:52:35 +0100292 uint32_t received_ntp_secs = 0;
293 uint32_t received_ntp_frac = 0;
294 state.remote_sr = 0;
295 if (rtcp_receiver_.NTP(&received_ntp_secs, &received_ntp_frac,
296 /*rtcp_arrival_time_secs=*/&state.last_rr_ntp_secs,
297 /*rtcp_arrival_time_frac=*/&state.last_rr_ntp_frac,
Alessio Bazzica048adc72021-03-10 15:05:55 +0100298 /*rtcp_timestamp=*/nullptr,
299 /*remote_sender_packet_count=*/nullptr,
300 /*remote_sender_octet_count=*/nullptr,
301 /*remote_sender_reports_count=*/nullptr)) {
Alessio Bazzica79011ef2021-03-10 14:52:35 +0100302 state.remote_sr = ((received_ntp_secs & 0x0000ffff) << 16) +
303 ((received_ntp_frac & 0xffff0000) >> 16);
304 }
Tommi3a5742c2020-05-20 09:32:51 +0200305
306 state.last_xr_rtis = rtcp_receiver_.ConsumeReceivedXrReferenceTimeInfo();
307
308 return state;
309}
310
311// TODO(nisse): This method shouldn't be called for a receive-only
312// stream. Delete rtp_sender_ check as soon as all applications are
313// updated.
314int32_t ModuleRtpRtcpImpl2::SetSendingStatus(const bool sending) {
315 if (rtcp_sender_.Sending() != sending) {
316 // Sends RTCP BYE when going from true to false
Tomas Gunnarssondbcf5d32021-04-23 20:31:08 +0200317 rtcp_sender_.SetSendingStatus(GetFeedbackState(), sending);
Tommi3a5742c2020-05-20 09:32:51 +0200318 }
319 return 0;
320}
321
322bool ModuleRtpRtcpImpl2::Sending() const {
323 return rtcp_sender_.Sending();
324}
325
326// TODO(nisse): This method shouldn't be called for a receive-only
327// stream. Delete rtp_sender_ check as soon as all applications are
328// updated.
329void ModuleRtpRtcpImpl2::SetSendingMediaStatus(const bool sending) {
330 if (rtp_sender_) {
Erik Språngbb904972021-08-06 13:10:11 +0200331 // Turning on or off sending status indicates module being set
332 // up or torn down, detach thread checker since subsequent calls
333 // may be from a different thread.
334 if (rtp_sender_->packet_generator.SendingMedia() != sending) {
335 pacer_thread_checker_.Detach();
336 }
Tommi3a5742c2020-05-20 09:32:51 +0200337 rtp_sender_->packet_generator.SetSendingMediaStatus(sending);
338 } else {
339 RTC_DCHECK(!sending);
340 }
341}
342
343bool ModuleRtpRtcpImpl2::SendingMedia() const {
344 return rtp_sender_ ? rtp_sender_->packet_generator.SendingMedia() : false;
345}
346
347bool ModuleRtpRtcpImpl2::IsAudioConfigured() const {
348 return rtp_sender_ ? rtp_sender_->packet_generator.IsAudioConfigured()
349 : false;
350}
351
352void ModuleRtpRtcpImpl2::SetAsPartOfAllocation(bool part_of_allocation) {
353 RTC_CHECK(rtp_sender_);
354 rtp_sender_->packet_sender.ForceIncludeSendPacketsInAllocation(
355 part_of_allocation);
356}
357
358bool ModuleRtpRtcpImpl2::OnSendingRtpFrame(uint32_t timestamp,
359 int64_t capture_time_ms,
360 int payload_type,
361 bool force_sender_report) {
362 if (!Sending())
363 return false;
364
Markus Handellc6b9ac72021-06-18 13:44:51 +0200365 // TODO(bugs.webrtc.org/12873): Migrate this method and it's users to use
366 // optional Timestamps.
367 absl::optional<Timestamp> capture_time;
368 if (capture_time_ms > 0) {
369 capture_time = Timestamp::Millis(capture_time_ms);
370 }
371 absl::optional<int> payload_type_optional;
372 if (payload_type >= 0)
373 payload_type_optional = payload_type;
374 rtcp_sender_.SetLastRtpTime(timestamp, capture_time, payload_type_optional);
Tommi3a5742c2020-05-20 09:32:51 +0200375 // Make sure an RTCP report isn't queued behind a key frame.
376 if (rtcp_sender_.TimeToSendRTCPReport(force_sender_report))
377 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
378
379 return true;
380}
381
382bool ModuleRtpRtcpImpl2::TrySendPacket(RtpPacketToSend* packet,
383 const PacedPacketInfo& pacing_info) {
384 RTC_DCHECK(rtp_sender_);
Erik Språngbb904972021-08-06 13:10:11 +0200385 RTC_DCHECK_RUN_ON(&pacer_thread_checker_);
386 if (rtp_sender_->deferred_sequencing_) {
Erik Språng2373bb92021-08-06 15:22:28 +0200387 if (!rtp_sender_->packet_generator.SendingMedia()) {
388 return false;
389 }
Erik Språngbb904972021-08-06 13:10:11 +0200390 if (packet->packet_type() == RtpPacketMediaType::kPadding &&
391 packet->Ssrc() == rtp_sender_->packet_generator.SSRC() &&
392 !rtp_sender_->sequencer_.CanSendPaddingOnMediaSsrc()) {
393 // New media packet preempted this generated padding packet, discard it.
394 return false;
395 }
396 bool is_flexfec =
397 packet->packet_type() == RtpPacketMediaType::kForwardErrorCorrection &&
398 packet->Ssrc() == rtp_sender_->packet_generator.FlexfecSsrc();
399 if (!is_flexfec) {
400 rtp_sender_->sequencer_.Sequence(*packet);
401 }
402 } else if (!rtp_sender_->packet_generator.SendingMedia()) {
Tommi3a5742c2020-05-20 09:32:51 +0200403 return false;
404 }
405 rtp_sender_->packet_sender.SendPacket(packet, pacing_info);
406 return true;
407}
408
Erik Språng1d50cb62020-07-02 17:41:32 +0200409void ModuleRtpRtcpImpl2::SetFecProtectionParams(
410 const FecProtectionParams& delta_params,
411 const FecProtectionParams& key_params) {
412 RTC_DCHECK(rtp_sender_);
413 rtp_sender_->packet_sender.SetFecProtectionParameters(delta_params,
414 key_params);
415}
416
417std::vector<std::unique_ptr<RtpPacketToSend>>
418ModuleRtpRtcpImpl2::FetchFecPackets() {
419 RTC_DCHECK(rtp_sender_);
Erik Språngbb904972021-08-06 13:10:11 +0200420 RTC_DCHECK_RUN_ON(&pacer_thread_checker_);
Erik Språng1d50cb62020-07-02 17:41:32 +0200421 auto fec_packets = rtp_sender_->packet_sender.FetchFecPackets();
Erik Språngbb904972021-08-06 13:10:11 +0200422 if (!fec_packets.empty() && !rtp_sender_->deferred_sequencing_) {
423 // Only assign sequence numbers for FEC packets in non-deferred mode, and
424 // never for FlexFEC which has as separate sequence number series.
Erik Språng1d50cb62020-07-02 17:41:32 +0200425 const bool generate_sequence_numbers =
426 !rtp_sender_->packet_sender.FlexFecSsrc().has_value();
427 if (generate_sequence_numbers) {
428 for (auto& fec_packet : fec_packets) {
429 rtp_sender_->packet_generator.AssignSequenceNumber(fec_packet.get());
430 }
431 }
432 }
433 return fec_packets;
434}
435
Tommi3a5742c2020-05-20 09:32:51 +0200436void ModuleRtpRtcpImpl2::OnPacketsAcknowledged(
437 rtc::ArrayView<const uint16_t> sequence_numbers) {
438 RTC_DCHECK(rtp_sender_);
439 rtp_sender_->packet_history.CullAcknowledgedPackets(sequence_numbers);
440}
441
442bool ModuleRtpRtcpImpl2::SupportsPadding() const {
443 RTC_DCHECK(rtp_sender_);
444 return rtp_sender_->packet_generator.SupportsPadding();
445}
446
447bool ModuleRtpRtcpImpl2::SupportsRtxPayloadPadding() const {
448 RTC_DCHECK(rtp_sender_);
449 return rtp_sender_->packet_generator.SupportsRtxPayloadPadding();
450}
451
452std::vector<std::unique_ptr<RtpPacketToSend>>
453ModuleRtpRtcpImpl2::GeneratePadding(size_t target_size_bytes) {
454 RTC_DCHECK(rtp_sender_);
Erik Språngbb904972021-08-06 13:10:11 +0200455
Erik Språng51e30832021-08-05 16:15:14 +0200456 // `can_send_padding_on_media_ssrc` set to false but is ignored at this
457 // point, RTPSender will internally query `sequencer_` while holding the
458 // send lock.
Erik Språngbb904972021-08-06 13:10:11 +0200459 RTC_DCHECK_RUN_ON(&pacer_thread_checker_);
460
461 // `can_send_padding_on_media_ssrc` set to false when deferred sequencing
462 // is off. It will be ignored in that case, RTPSender will internally query
463 // `sequencer_` while holding the send lock instead.
Tommi3a5742c2020-05-20 09:32:51 +0200464 return rtp_sender_->packet_generator.GeneratePadding(
Erik Språngbfcfe032021-08-04 14:45:32 +0200465 target_size_bytes, rtp_sender_->packet_sender.MediaHasBeenSent(),
Erik Språngbb904972021-08-06 13:10:11 +0200466
467 rtp_sender_->deferred_sequencing_
468 ? rtp_sender_->sequencer_.CanSendPaddingOnMediaSsrc()
469 : false);
Tommi3a5742c2020-05-20 09:32:51 +0200470}
471
472std::vector<RtpSequenceNumberMap::Info>
473ModuleRtpRtcpImpl2::GetSentRtpPacketInfos(
474 rtc::ArrayView<const uint16_t> sequence_numbers) const {
475 RTC_DCHECK(rtp_sender_);
476 return rtp_sender_->packet_sender.GetSentRtpPacketInfos(sequence_numbers);
477}
478
479size_t ModuleRtpRtcpImpl2::ExpectedPerPacketOverhead() const {
480 if (!rtp_sender_) {
481 return 0;
482 }
483 return rtp_sender_->packet_generator.ExpectedPerPacketOverhead();
484}
485
486size_t ModuleRtpRtcpImpl2::MaxRtpPacketSize() const {
487 RTC_DCHECK(rtp_sender_);
488 return rtp_sender_->packet_generator.MaxRtpPacketSize();
489}
490
491void ModuleRtpRtcpImpl2::SetMaxRtpPacketSize(size_t rtp_packet_size) {
492 RTC_DCHECK_LE(rtp_packet_size, IP_PACKET_SIZE)
493 << "rtp packet size too large: " << rtp_packet_size;
494 RTC_DCHECK_GT(rtp_packet_size, packet_overhead_)
495 << "rtp packet size too small: " << rtp_packet_size;
496
497 rtcp_sender_.SetMaxRtpPacketSize(rtp_packet_size);
498 if (rtp_sender_) {
499 rtp_sender_->packet_generator.SetMaxRtpPacketSize(rtp_packet_size);
500 }
501}
502
503RtcpMode ModuleRtpRtcpImpl2::RTCP() const {
504 return rtcp_sender_.Status();
505}
506
507// Configure RTCP status i.e on/off.
508void ModuleRtpRtcpImpl2::SetRTCPStatus(const RtcpMode method) {
509 rtcp_sender_.SetRTCPStatus(method);
510}
511
512int32_t ModuleRtpRtcpImpl2::SetCNAME(const char* c_name) {
513 return rtcp_sender_.SetCNAME(c_name);
514}
515
Tommi3a5742c2020-05-20 09:32:51 +0200516int32_t ModuleRtpRtcpImpl2::RemoteNTP(uint32_t* received_ntpsecs,
517 uint32_t* received_ntpfrac,
518 uint32_t* rtcp_arrival_time_secs,
519 uint32_t* rtcp_arrival_time_frac,
520 uint32_t* rtcp_timestamp) const {
521 return rtcp_receiver_.NTP(received_ntpsecs, received_ntpfrac,
522 rtcp_arrival_time_secs, rtcp_arrival_time_frac,
Alessio Bazzica048adc72021-03-10 15:05:55 +0100523 rtcp_timestamp,
524 /*remote_sender_packet_count=*/nullptr,
525 /*remote_sender_octet_count=*/nullptr,
526 /*remote_sender_reports_count=*/nullptr)
Tommi3a5742c2020-05-20 09:32:51 +0200527 ? 0
528 : -1;
529}
530
Artem Titov913cfa72021-07-28 23:57:33 +0200531// TODO(tommi): Check if `avg_rtt_ms`, `min_rtt_ms`, `max_rtt_ms` params are
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200532// actually used in practice (some callers ask for it but don't use it). It
Artem Titov913cfa72021-07-28 23:57:33 +0200533// could be that only `rtt` is needed and if so, then the fast path could be to
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200534// just call rtt_ms() and rely on the calculation being done periodically.
Tommi3a5742c2020-05-20 09:32:51 +0200535int32_t ModuleRtpRtcpImpl2::RTT(const uint32_t remote_ssrc,
536 int64_t* rtt,
537 int64_t* avg_rtt,
538 int64_t* min_rtt,
539 int64_t* max_rtt) const {
540 int32_t ret = rtcp_receiver_.RTT(remote_ssrc, rtt, avg_rtt, min_rtt, max_rtt);
541 if (rtt && *rtt == 0) {
542 // Try to get RTT from RtcpRttStats class.
543 *rtt = rtt_ms();
544 }
545 return ret;
546}
547
548int64_t ModuleRtpRtcpImpl2::ExpectedRetransmissionTimeMs() const {
549 int64_t expected_retransmission_time_ms = rtt_ms();
550 if (expected_retransmission_time_ms > 0) {
551 return expected_retransmission_time_ms;
552 }
Artem Titov913cfa72021-07-28 23:57:33 +0200553 // No rtt available (`kRttUpdateInterval` not yet passed?), so try to
Tommi3a5742c2020-05-20 09:32:51 +0200554 // poll avg_rtt_ms directly from rtcp receiver.
555 if (rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), nullptr,
556 &expected_retransmission_time_ms, nullptr,
557 nullptr) == 0) {
558 return expected_retransmission_time_ms;
559 }
560 return kDefaultExpectedRetransmissionTimeMs;
561}
562
563// Force a send of an RTCP packet.
564// Normal SR and RR are triggered via the process function.
565int32_t ModuleRtpRtcpImpl2::SendRTCP(RTCPPacketType packet_type) {
566 return rtcp_sender_.SendRTCP(GetFeedbackState(), packet_type);
567}
568
Tommi3a5742c2020-05-20 09:32:51 +0200569void ModuleRtpRtcpImpl2::GetSendStreamDataCounters(
570 StreamDataCounters* rtp_counters,
571 StreamDataCounters* rtx_counters) const {
572 rtp_sender_->packet_sender.GetDataCounters(rtp_counters, rtx_counters);
573}
574
575// Received RTCP report.
Tommi3a5742c2020-05-20 09:32:51 +0200576std::vector<ReportBlockData> ModuleRtpRtcpImpl2::GetLatestReportBlockData()
577 const {
578 return rtcp_receiver_.GetLatestReportBlockData();
579}
580
Alessio Bazzicabc1c93d2021-03-12 17:45:26 +0100581absl::optional<RtpRtcpInterface::SenderReportStats>
582ModuleRtpRtcpImpl2::GetSenderReportStats() const {
583 SenderReportStats stats;
584 uint32_t remote_timestamp_secs;
585 uint32_t remote_timestamp_frac;
586 uint32_t arrival_timestamp_secs;
587 uint32_t arrival_timestamp_frac;
588 if (rtcp_receiver_.NTP(&remote_timestamp_secs, &remote_timestamp_frac,
589 &arrival_timestamp_secs, &arrival_timestamp_frac,
590 /*rtcp_timestamp=*/nullptr, &stats.packets_sent,
591 &stats.bytes_sent, &stats.reports_count)) {
592 stats.last_remote_timestamp.Set(remote_timestamp_secs,
593 remote_timestamp_frac);
594 stats.last_arrival_timestamp.Set(arrival_timestamp_secs,
595 arrival_timestamp_frac);
596 return stats;
597 }
598 return absl::nullopt;
599}
600
Tommi3a5742c2020-05-20 09:32:51 +0200601// (REMB) Receiver Estimated Max Bitrate.
602void ModuleRtpRtcpImpl2::SetRemb(int64_t bitrate_bps,
603 std::vector<uint32_t> ssrcs) {
604 rtcp_sender_.SetRemb(bitrate_bps, std::move(ssrcs));
605}
606
607void ModuleRtpRtcpImpl2::UnsetRemb() {
608 rtcp_sender_.UnsetRemb();
609}
610
611void ModuleRtpRtcpImpl2::SetExtmapAllowMixed(bool extmap_allow_mixed) {
612 rtp_sender_->packet_generator.SetExtmapAllowMixed(extmap_allow_mixed);
613}
614
Tommi3a5742c2020-05-20 09:32:51 +0200615void ModuleRtpRtcpImpl2::RegisterRtpHeaderExtension(absl::string_view uri,
616 int id) {
617 bool registered =
618 rtp_sender_->packet_generator.RegisterRtpHeaderExtension(uri, id);
619 RTC_CHECK(registered);
620}
621
622int32_t ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
623 const RTPExtensionType type) {
624 return rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(type);
625}
626void ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
627 absl::string_view uri) {
628 rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(uri);
629}
630
Tommi3a5742c2020-05-20 09:32:51 +0200631void ModuleRtpRtcpImpl2::SetTmmbn(std::vector<rtcp::TmmbItem> bounding_set) {
632 rtcp_sender_.SetTmmbn(std::move(bounding_set));
633}
634
635// Send a Negative acknowledgment packet.
636int32_t ModuleRtpRtcpImpl2::SendNACK(const uint16_t* nack_list,
637 const uint16_t size) {
638 uint16_t nack_length = size;
639 uint16_t start_id = 0;
640 int64_t now_ms = clock_->TimeInMilliseconds();
641 if (TimeToSendFullNackList(now_ms)) {
642 nack_last_time_sent_full_ms_ = now_ms;
643 } else {
644 // Only send extended list.
645 if (nack_last_seq_number_sent_ == nack_list[size - 1]) {
646 // Last sequence number is the same, do not send list.
647 return 0;
648 }
649 // Send new sequence numbers.
650 for (int i = 0; i < size; ++i) {
651 if (nack_last_seq_number_sent_ == nack_list[i]) {
652 start_id = i + 1;
653 break;
654 }
655 }
656 nack_length = size - start_id;
657 }
658
659 // Our RTCP NACK implementation is limited to kRtcpMaxNackFields sequence
660 // numbers per RTCP packet.
661 if (nack_length > kRtcpMaxNackFields) {
662 nack_length = kRtcpMaxNackFields;
663 }
664 nack_last_seq_number_sent_ = nack_list[start_id + nack_length - 1];
665
666 return rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, nack_length,
667 &nack_list[start_id]);
668}
669
670void ModuleRtpRtcpImpl2::SendNack(
671 const std::vector<uint16_t>& sequence_numbers) {
672 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, sequence_numbers.size(),
673 sequence_numbers.data());
674}
675
676bool ModuleRtpRtcpImpl2::TimeToSendFullNackList(int64_t now) const {
677 // Use RTT from RtcpRttStats class if provided.
678 int64_t rtt = rtt_ms();
679 if (rtt == 0) {
680 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
681 }
682
683 const int64_t kStartUpRttMs = 100;
684 int64_t wait_time = 5 + ((rtt * 3) >> 1); // 5 + RTT * 1.5.
685 if (rtt == 0) {
686 wait_time = kStartUpRttMs;
687 }
688
Artem Titov913cfa72021-07-28 23:57:33 +0200689 // Send a full NACK list once within every `wait_time`.
Tommi3a5742c2020-05-20 09:32:51 +0200690 return now - nack_last_time_sent_full_ms_ > wait_time;
691}
692
693// Store the sent packets, needed to answer to Negative acknowledgment requests.
694void ModuleRtpRtcpImpl2::SetStorePacketsStatus(const bool enable,
695 const uint16_t number_to_store) {
696 rtp_sender_->packet_history.SetStorePacketsStatus(
697 enable ? RtpPacketHistory::StorageMode::kStoreAndCull
698 : RtpPacketHistory::StorageMode::kDisabled,
699 number_to_store);
700}
701
702bool ModuleRtpRtcpImpl2::StorePackets() const {
703 return rtp_sender_->packet_history.GetStorageMode() !=
704 RtpPacketHistory::StorageMode::kDisabled;
705}
706
707void ModuleRtpRtcpImpl2::SendCombinedRtcpPacket(
708 std::vector<std::unique_ptr<rtcp::RtcpPacket>> rtcp_packets) {
709 rtcp_sender_.SendCombinedRtcpPacket(std::move(rtcp_packets));
710}
711
712int32_t ModuleRtpRtcpImpl2::SendLossNotification(uint16_t last_decoded_seq_num,
713 uint16_t last_received_seq_num,
714 bool decodability_flag,
715 bool buffering_allowed) {
716 return rtcp_sender_.SendLossNotification(
717 GetFeedbackState(), last_decoded_seq_num, last_received_seq_num,
718 decodability_flag, buffering_allowed);
719}
720
721void ModuleRtpRtcpImpl2::SetRemoteSSRC(const uint32_t ssrc) {
722 // Inform about the incoming SSRC.
723 rtcp_sender_.SetRemoteSSRC(ssrc);
724 rtcp_receiver_.SetRemoteSSRC(ssrc);
725}
726
Tommi08be9ba2021-06-15 23:01:57 +0200727void ModuleRtpRtcpImpl2::SetLocalSsrc(uint32_t local_ssrc) {
728 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
729 rtcp_receiver_.set_local_media_ssrc(local_ssrc);
730 rtcp_sender_.SetSsrc(local_ssrc);
731}
732
Tommi3a5742c2020-05-20 09:32:51 +0200733RtpSendRates ModuleRtpRtcpImpl2::GetSendRates() const {
Tommi1050fbc2021-06-03 17:58:28 +0200734 // Typically called on the `rtp_transport_queue_` owned by an
735 // RtpTransportControllerSendInterface instance.
Tommi3a5742c2020-05-20 09:32:51 +0200736 return rtp_sender_->packet_sender.GetSendRates();
737}
738
739void ModuleRtpRtcpImpl2::OnRequestSendReport() {
740 SendRTCP(kRtcpSr);
741}
742
743void ModuleRtpRtcpImpl2::OnReceivedNack(
744 const std::vector<uint16_t>& nack_sequence_numbers) {
745 if (!rtp_sender_)
746 return;
747
748 if (!StorePackets() || nack_sequence_numbers.empty()) {
749 return;
750 }
751 // Use RTT from RtcpRttStats class if provided.
752 int64_t rtt = rtt_ms();
753 if (rtt == 0) {
754 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
755 }
756 rtp_sender_->packet_generator.OnReceivedNack(nack_sequence_numbers, rtt);
757}
758
759void ModuleRtpRtcpImpl2::OnReceivedRtcpReportBlocks(
760 const ReportBlockList& report_blocks) {
761 if (rtp_sender_) {
762 uint32_t ssrc = SSRC();
763 absl::optional<uint32_t> rtx_ssrc;
764 if (rtp_sender_->packet_generator.RtxStatus() != kRtxOff) {
765 rtx_ssrc = rtp_sender_->packet_generator.RtxSsrc();
766 }
767
768 for (const RTCPReportBlock& report_block : report_blocks) {
769 if (ssrc == report_block.source_ssrc) {
770 rtp_sender_->packet_generator.OnReceivedAckOnSsrc(
771 report_block.extended_highest_sequence_number);
772 } else if (rtx_ssrc && *rtx_ssrc == report_block.source_ssrc) {
773 rtp_sender_->packet_generator.OnReceivedAckOnRtxSsrc(
774 report_block.extended_highest_sequence_number);
775 }
776 }
777 }
778}
779
Tommi3a5742c2020-05-20 09:32:51 +0200780void ModuleRtpRtcpImpl2::set_rtt_ms(int64_t rtt_ms) {
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200781 RTC_DCHECK_RUN_ON(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +0200782 {
Markus Handellf7303e62020-07-09 01:34:42 +0200783 MutexLock lock(&mutex_rtt_);
Tommi3a5742c2020-05-20 09:32:51 +0200784 rtt_ms_ = rtt_ms;
785 }
786 if (rtp_sender_) {
787 rtp_sender_->packet_history.SetRtt(rtt_ms);
788 }
789}
790
791int64_t ModuleRtpRtcpImpl2::rtt_ms() const {
Markus Handellf7303e62020-07-09 01:34:42 +0200792 MutexLock lock(&mutex_rtt_);
Tommi3a5742c2020-05-20 09:32:51 +0200793 return rtt_ms_;
794}
795
796void ModuleRtpRtcpImpl2::SetVideoBitrateAllocation(
797 const VideoBitrateAllocation& bitrate) {
798 rtcp_sender_.SetVideoBitrateAllocation(bitrate);
799}
800
801RTPSender* ModuleRtpRtcpImpl2::RtpSender() {
802 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
803}
804
805const RTPSender* ModuleRtpRtcpImpl2::RtpSender() const {
806 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
807}
808
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200809void ModuleRtpRtcpImpl2::PeriodicUpdate() {
810 RTC_DCHECK_RUN_ON(worker_queue_);
811
812 Timestamp check_since = clock_->CurrentTime() - kRttUpdateInterval;
813 absl::optional<TimeDelta> rtt =
814 rtcp_receiver_.OnPeriodicRttUpdate(check_since, rtcp_sender_.Sending());
815 if (rtt) {
Ivo Creusen8c40d512021-07-13 12:53:22 +0000816 if (rtt_stats_) {
817 rtt_stats_->OnRttUpdate(rtt->ms());
818 }
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200819 set_rtt_ms(rtt->ms());
820 }
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200821}
822
Markus Handell885d5382021-06-21 18:57:36 +0200823// RTC_RUN_ON(worker_queue_);
824void ModuleRtpRtcpImpl2::MaybeSendRtcp() {
825 if (rtcp_sender_.TimeToSendRTCPReport())
826 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
827}
828
829// TODO(bugs.webrtc.org/12889): Consider removing this function when the issue
830// is resolved.
831// RTC_RUN_ON(worker_queue_);
832void ModuleRtpRtcpImpl2::MaybeSendRtcpAtOrAfterTimestamp(
833 Timestamp execution_time) {
834 Timestamp now = clock_->CurrentTime();
835 if (now >= execution_time) {
836 MaybeSendRtcp();
837 return;
838 }
839
840 RTC_DLOG(LS_WARNING)
841 << "BUGBUG: Task queue scheduled delayed call too early.";
842
843 ScheduleMaybeSendRtcpAtOrAfterTimestamp(execution_time, execution_time - now);
844}
845
846void ModuleRtpRtcpImpl2::ScheduleRtcpSendEvaluation(TimeDelta duration) {
847 // We end up here under various sequences including the worker queue, and
848 // the RTCPSender lock is held.
849 // We're assuming that the fact that RTCPSender executes under other sequences
850 // than the worker queue on which it's created on implies that external
851 // synchronization is present and removes this activity before destruction.
852 if (duration.IsZero()) {
853 worker_queue_->PostTask(ToQueuedTask(task_safety_, [this] {
854 RTC_DCHECK_RUN_ON(worker_queue_);
855 MaybeSendRtcp();
856 }));
857 } else {
858 Timestamp execution_time = clock_->CurrentTime() + duration;
859 ScheduleMaybeSendRtcpAtOrAfterTimestamp(execution_time, duration);
860 }
861}
862
863void ModuleRtpRtcpImpl2::ScheduleMaybeSendRtcpAtOrAfterTimestamp(
864 Timestamp execution_time,
865 TimeDelta duration) {
866 // We end up here under various sequences including the worker queue, and
867 // the RTCPSender lock is held.
Artem Titov913cfa72021-07-28 23:57:33 +0200868 // See note in ScheduleRtcpSendEvaluation about why `worker_queue_` can be
Markus Handell885d5382021-06-21 18:57:36 +0200869 // accessed.
870 worker_queue_->PostDelayedTask(
871 ToQueuedTask(task_safety_,
872 [this, execution_time] {
873 RTC_DCHECK_RUN_ON(worker_queue_);
874 MaybeSendRtcpAtOrAfterTimestamp(execution_time);
875 }),
876 DelayMillisForDuration(duration));
877}
878
Tommi3a5742c2020-05-20 09:32:51 +0200879} // namespace webrtc