blob: c6d19775f2f36ca18874a4b32bfbccf2ee0ed199 [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
56 // slept. Account for this, unless |duration| is exactly representable in
57 // 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),
66 packet_sender(config, &packet_history),
Erik Språng1d50cb62020-07-02 17:41:32 +020067 non_paced_sender(&packet_sender, this),
Tommi3a5742c2020-05-20 09:32:51 +020068 packet_generator(
69 config,
70 &packet_history,
71 config.paced_sender ? config.paced_sender : &non_paced_sender) {}
Erik Språng1d50cb62020-07-02 17:41:32 +020072void ModuleRtpRtcpImpl2::RtpSenderContext::AssignSequenceNumber(
73 RtpPacketToSend* packet) {
74 packet_generator.AssignSequenceNumber(packet);
75}
Tommi3a5742c2020-05-20 09:32:51 +020076
Tommi3a5742c2020-05-20 09:32:51 +020077ModuleRtpRtcpImpl2::ModuleRtpRtcpImpl2(const Configuration& configuration)
Niels Moller2accc7d2021-01-12 15:54:16 +000078 : worker_queue_(TaskQueueBase::Current()),
Markus Handell885d5382021-06-21 18:57:36 +020079 rtcp_sender_(AddRtcpSendEvaluationCallback(
80 RTCPSender::Configuration::FromRtpRtcpConfiguration(configuration),
81 [this](TimeDelta duration) {
82 ScheduleRtcpSendEvaluation(duration);
83 })),
Tommi3a5742c2020-05-20 09:32:51 +020084 rtcp_receiver_(configuration, this),
85 clock_(configuration.clock),
Tommi3a5742c2020-05-20 09:32:51 +020086 packet_overhead_(28), // IPV4 UDP.
87 nack_last_time_sent_full_ms_(0),
88 nack_last_seq_number_sent_(0),
89 remote_bitrate_(configuration.remote_bitrate_estimator),
90 rtt_stats_(configuration.rtt_stats),
91 rtt_ms_(0) {
Niels Moller2accc7d2021-01-12 15:54:16 +000092 RTC_DCHECK(worker_queue_);
Tommi08be9ba2021-06-15 23:01:57 +020093 packet_sequence_checker_.Detach();
Tommi3a5742c2020-05-20 09:32:51 +020094 if (!configuration.receiver_only) {
95 rtp_sender_ = std::make_unique<RtpSenderContext>(configuration);
96 // Make sure rtcp sender use same timestamp offset as rtp sender.
97 rtcp_sender_.SetTimestampOffset(
98 rtp_sender_->packet_generator.TimestampOffset());
99 }
100
101 // Set default packet size limit.
102 // TODO(nisse): Kind-of duplicates
103 // webrtc::VideoSendStream::Config::Rtp::kDefaultMaxPacketSize.
104 const size_t kTcpOverIpv4HeaderSize = 40;
105 SetMaxRtpPacketSize(IP_PACKET_SIZE - kTcpOverIpv4HeaderSize);
Ivo Creusen8c40d512021-07-13 12:53:22 +0000106 rtt_update_task_ = RepeatingTaskHandle::DelayedStart(
107 worker_queue_, kRttUpdateInterval, [this]() {
108 PeriodicUpdate();
109 return kRttUpdateInterval;
110 });
Tommi3a5742c2020-05-20 09:32:51 +0200111}
112
113ModuleRtpRtcpImpl2::~ModuleRtpRtcpImpl2() {
Niels Moller2accc7d2021-01-12 15:54:16 +0000114 RTC_DCHECK_RUN_ON(worker_queue_);
115 rtt_update_task_.Stop();
116}
117
118// static
119std::unique_ptr<ModuleRtpRtcpImpl2> ModuleRtpRtcpImpl2::Create(
120 const Configuration& configuration) {
121 RTC_DCHECK(configuration.clock);
122 RTC_DCHECK(TaskQueueBase::Current());
123 return std::make_unique<ModuleRtpRtcpImpl2>(configuration);
Tomas Gunnarssonfae05622020-06-03 08:54:39 +0200124}
125
Tommi3a5742c2020-05-20 09:32:51 +0200126void ModuleRtpRtcpImpl2::SetRtxSendStatus(int mode) {
127 rtp_sender_->packet_generator.SetRtxStatus(mode);
128}
129
130int ModuleRtpRtcpImpl2::RtxSendStatus() const {
131 return rtp_sender_ ? rtp_sender_->packet_generator.RtxStatus() : kRtxOff;
132}
133
134void ModuleRtpRtcpImpl2::SetRtxSendPayloadType(int payload_type,
135 int associated_payload_type) {
136 rtp_sender_->packet_generator.SetRtxPayloadType(payload_type,
137 associated_payload_type);
138}
139
140absl::optional<uint32_t> ModuleRtpRtcpImpl2::RtxSsrc() const {
141 return rtp_sender_ ? rtp_sender_->packet_generator.RtxSsrc() : absl::nullopt;
142}
143
144absl::optional<uint32_t> ModuleRtpRtcpImpl2::FlexfecSsrc() const {
145 if (rtp_sender_) {
146 return rtp_sender_->packet_generator.FlexfecSsrc();
147 }
148 return absl::nullopt;
149}
150
151void ModuleRtpRtcpImpl2::IncomingRtcpPacket(const uint8_t* rtcp_packet,
152 const size_t length) {
Tommi08be9ba2021-06-15 23:01:57 +0200153 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
Tommi3a5742c2020-05-20 09:32:51 +0200154 rtcp_receiver_.IncomingPacket(rtcp_packet, length);
155}
156
157void ModuleRtpRtcpImpl2::RegisterSendPayloadFrequency(int payload_type,
158 int payload_frequency) {
159 rtcp_sender_.SetRtpClockRate(payload_type, payload_frequency);
160}
161
162int32_t ModuleRtpRtcpImpl2::DeRegisterSendPayload(const int8_t payload_type) {
163 return 0;
164}
165
166uint32_t ModuleRtpRtcpImpl2::StartTimestamp() const {
167 return rtp_sender_->packet_generator.TimestampOffset();
168}
169
170// Configure start timestamp, default is a random number.
171void ModuleRtpRtcpImpl2::SetStartTimestamp(const uint32_t timestamp) {
172 rtcp_sender_.SetTimestampOffset(timestamp);
173 rtp_sender_->packet_generator.SetTimestampOffset(timestamp);
174 rtp_sender_->packet_sender.SetTimestampOffset(timestamp);
175}
176
177uint16_t ModuleRtpRtcpImpl2::SequenceNumber() const {
178 return rtp_sender_->packet_generator.SequenceNumber();
179}
180
181// Set SequenceNumber, default is a random number.
182void ModuleRtpRtcpImpl2::SetSequenceNumber(const uint16_t seq_num) {
183 rtp_sender_->packet_generator.SetSequenceNumber(seq_num);
184}
185
186void ModuleRtpRtcpImpl2::SetRtpState(const RtpState& rtp_state) {
187 rtp_sender_->packet_generator.SetRtpState(rtp_state);
Tommi3a5742c2020-05-20 09:32:51 +0200188 rtcp_sender_.SetTimestampOffset(rtp_state.start_timestamp);
189}
190
191void ModuleRtpRtcpImpl2::SetRtxState(const RtpState& rtp_state) {
192 rtp_sender_->packet_generator.SetRtxRtpState(rtp_state);
193}
194
195RtpState ModuleRtpRtcpImpl2::GetRtpState() const {
196 RtpState state = rtp_sender_->packet_generator.GetRtpState();
Tommi3a5742c2020-05-20 09:32:51 +0200197 return state;
198}
199
200RtpState ModuleRtpRtcpImpl2::GetRtxState() const {
201 return rtp_sender_->packet_generator.GetRtxRtpState();
202}
203
Ivo Creusen8c40d512021-07-13 12:53:22 +0000204void ModuleRtpRtcpImpl2::SetNonSenderRttMeasurement(bool enabled) {
205 rtcp_sender_.SetNonSenderRttMeasurement(enabled);
206 rtcp_receiver_.SetNonSenderRttMeasurement(enabled);
207}
208
Tommi08be9ba2021-06-15 23:01:57 +0200209uint32_t ModuleRtpRtcpImpl2::local_media_ssrc() const {
210 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
211 RTC_DCHECK_EQ(rtcp_receiver_.local_media_ssrc(), rtcp_sender_.SSRC());
212 return rtcp_receiver_.local_media_ssrc();
213}
214
Tommi3a5742c2020-05-20 09:32:51 +0200215void ModuleRtpRtcpImpl2::SetRid(const std::string& rid) {
216 if (rtp_sender_) {
217 rtp_sender_->packet_generator.SetRid(rid);
218 }
219}
220
221void ModuleRtpRtcpImpl2::SetMid(const std::string& mid) {
222 if (rtp_sender_) {
223 rtp_sender_->packet_generator.SetMid(mid);
224 }
225 // TODO(bugs.webrtc.org/4050): If we end up supporting the MID SDES item for
226 // RTCP, this will need to be passed down to the RTCPSender also.
227}
228
229void ModuleRtpRtcpImpl2::SetCsrcs(const std::vector<uint32_t>& csrcs) {
230 rtcp_sender_.SetCsrcs(csrcs);
231 rtp_sender_->packet_generator.SetCsrcs(csrcs);
232}
233
234// TODO(pbos): Handle media and RTX streams separately (separate RTCP
235// feedbacks).
236RTCPSender::FeedbackState ModuleRtpRtcpImpl2::GetFeedbackState() {
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200237 // TODO(bugs.webrtc.org/11581): Called by potentially multiple threads.
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200238 // Mostly "Send*" methods. Make sure it's only called on the
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200239 // construction thread.
240
Tommi3a5742c2020-05-20 09:32:51 +0200241 RTCPSender::FeedbackState state;
242 // This is called also when receiver_only is true. Hence below
243 // checks that rtp_sender_ exists.
244 if (rtp_sender_) {
245 StreamDataCounters rtp_stats;
246 StreamDataCounters rtx_stats;
247 rtp_sender_->packet_sender.GetDataCounters(&rtp_stats, &rtx_stats);
248 state.packets_sent =
249 rtp_stats.transmitted.packets + rtx_stats.transmitted.packets;
250 state.media_bytes_sent = rtp_stats.transmitted.payload_bytes +
251 rtx_stats.transmitted.payload_bytes;
252 state.send_bitrate =
253 rtp_sender_->packet_sender.GetSendRates().Sum().bps<uint32_t>();
254 }
255 state.receiver = &rtcp_receiver_;
256
Alessio Bazzica79011ef2021-03-10 14:52:35 +0100257 uint32_t received_ntp_secs = 0;
258 uint32_t received_ntp_frac = 0;
259 state.remote_sr = 0;
260 if (rtcp_receiver_.NTP(&received_ntp_secs, &received_ntp_frac,
261 /*rtcp_arrival_time_secs=*/&state.last_rr_ntp_secs,
262 /*rtcp_arrival_time_frac=*/&state.last_rr_ntp_frac,
Alessio Bazzica048adc72021-03-10 15:05:55 +0100263 /*rtcp_timestamp=*/nullptr,
264 /*remote_sender_packet_count=*/nullptr,
265 /*remote_sender_octet_count=*/nullptr,
266 /*remote_sender_reports_count=*/nullptr)) {
Alessio Bazzica79011ef2021-03-10 14:52:35 +0100267 state.remote_sr = ((received_ntp_secs & 0x0000ffff) << 16) +
268 ((received_ntp_frac & 0xffff0000) >> 16);
269 }
Tommi3a5742c2020-05-20 09:32:51 +0200270
271 state.last_xr_rtis = rtcp_receiver_.ConsumeReceivedXrReferenceTimeInfo();
272
273 return state;
274}
275
276// TODO(nisse): This method shouldn't be called for a receive-only
277// stream. Delete rtp_sender_ check as soon as all applications are
278// updated.
279int32_t ModuleRtpRtcpImpl2::SetSendingStatus(const bool sending) {
280 if (rtcp_sender_.Sending() != sending) {
281 // Sends RTCP BYE when going from true to false
Tomas Gunnarssondbcf5d32021-04-23 20:31:08 +0200282 rtcp_sender_.SetSendingStatus(GetFeedbackState(), sending);
Tommi3a5742c2020-05-20 09:32:51 +0200283 }
284 return 0;
285}
286
287bool ModuleRtpRtcpImpl2::Sending() const {
288 return rtcp_sender_.Sending();
289}
290
291// TODO(nisse): This method shouldn't be called for a receive-only
292// stream. Delete rtp_sender_ check as soon as all applications are
293// updated.
294void ModuleRtpRtcpImpl2::SetSendingMediaStatus(const bool sending) {
295 if (rtp_sender_) {
296 rtp_sender_->packet_generator.SetSendingMediaStatus(sending);
297 } else {
298 RTC_DCHECK(!sending);
299 }
300}
301
302bool ModuleRtpRtcpImpl2::SendingMedia() const {
303 return rtp_sender_ ? rtp_sender_->packet_generator.SendingMedia() : false;
304}
305
306bool ModuleRtpRtcpImpl2::IsAudioConfigured() const {
307 return rtp_sender_ ? rtp_sender_->packet_generator.IsAudioConfigured()
308 : false;
309}
310
311void ModuleRtpRtcpImpl2::SetAsPartOfAllocation(bool part_of_allocation) {
312 RTC_CHECK(rtp_sender_);
313 rtp_sender_->packet_sender.ForceIncludeSendPacketsInAllocation(
314 part_of_allocation);
315}
316
317bool ModuleRtpRtcpImpl2::OnSendingRtpFrame(uint32_t timestamp,
318 int64_t capture_time_ms,
319 int payload_type,
320 bool force_sender_report) {
321 if (!Sending())
322 return false;
323
Markus Handellc6b9ac72021-06-18 13:44:51 +0200324 // TODO(bugs.webrtc.org/12873): Migrate this method and it's users to use
325 // optional Timestamps.
326 absl::optional<Timestamp> capture_time;
327 if (capture_time_ms > 0) {
328 capture_time = Timestamp::Millis(capture_time_ms);
329 }
330 absl::optional<int> payload_type_optional;
331 if (payload_type >= 0)
332 payload_type_optional = payload_type;
333 rtcp_sender_.SetLastRtpTime(timestamp, capture_time, payload_type_optional);
Tommi3a5742c2020-05-20 09:32:51 +0200334 // Make sure an RTCP report isn't queued behind a key frame.
335 if (rtcp_sender_.TimeToSendRTCPReport(force_sender_report))
336 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
337
338 return true;
339}
340
341bool ModuleRtpRtcpImpl2::TrySendPacket(RtpPacketToSend* packet,
342 const PacedPacketInfo& pacing_info) {
343 RTC_DCHECK(rtp_sender_);
344 // TODO(sprang): Consider if we can remove this check.
345 if (!rtp_sender_->packet_generator.SendingMedia()) {
346 return false;
347 }
348 rtp_sender_->packet_sender.SendPacket(packet, pacing_info);
349 return true;
350}
351
Erik Språng1d50cb62020-07-02 17:41:32 +0200352void ModuleRtpRtcpImpl2::SetFecProtectionParams(
353 const FecProtectionParams& delta_params,
354 const FecProtectionParams& key_params) {
355 RTC_DCHECK(rtp_sender_);
356 rtp_sender_->packet_sender.SetFecProtectionParameters(delta_params,
357 key_params);
358}
359
360std::vector<std::unique_ptr<RtpPacketToSend>>
361ModuleRtpRtcpImpl2::FetchFecPackets() {
362 RTC_DCHECK(rtp_sender_);
363 auto fec_packets = rtp_sender_->packet_sender.FetchFecPackets();
364 if (!fec_packets.empty()) {
365 // Don't assign sequence numbers for FlexFEC packets.
366 const bool generate_sequence_numbers =
367 !rtp_sender_->packet_sender.FlexFecSsrc().has_value();
368 if (generate_sequence_numbers) {
369 for (auto& fec_packet : fec_packets) {
370 rtp_sender_->packet_generator.AssignSequenceNumber(fec_packet.get());
371 }
372 }
373 }
374 return fec_packets;
375}
376
Tommi3a5742c2020-05-20 09:32:51 +0200377void ModuleRtpRtcpImpl2::OnPacketsAcknowledged(
378 rtc::ArrayView<const uint16_t> sequence_numbers) {
379 RTC_DCHECK(rtp_sender_);
380 rtp_sender_->packet_history.CullAcknowledgedPackets(sequence_numbers);
381}
382
383bool ModuleRtpRtcpImpl2::SupportsPadding() const {
384 RTC_DCHECK(rtp_sender_);
385 return rtp_sender_->packet_generator.SupportsPadding();
386}
387
388bool ModuleRtpRtcpImpl2::SupportsRtxPayloadPadding() const {
389 RTC_DCHECK(rtp_sender_);
390 return rtp_sender_->packet_generator.SupportsRtxPayloadPadding();
391}
392
393std::vector<std::unique_ptr<RtpPacketToSend>>
394ModuleRtpRtcpImpl2::GeneratePadding(size_t target_size_bytes) {
395 RTC_DCHECK(rtp_sender_);
396 return rtp_sender_->packet_generator.GeneratePadding(
397 target_size_bytes, rtp_sender_->packet_sender.MediaHasBeenSent());
398}
399
400std::vector<RtpSequenceNumberMap::Info>
401ModuleRtpRtcpImpl2::GetSentRtpPacketInfos(
402 rtc::ArrayView<const uint16_t> sequence_numbers) const {
403 RTC_DCHECK(rtp_sender_);
404 return rtp_sender_->packet_sender.GetSentRtpPacketInfos(sequence_numbers);
405}
406
407size_t ModuleRtpRtcpImpl2::ExpectedPerPacketOverhead() const {
408 if (!rtp_sender_) {
409 return 0;
410 }
411 return rtp_sender_->packet_generator.ExpectedPerPacketOverhead();
412}
413
414size_t ModuleRtpRtcpImpl2::MaxRtpPacketSize() const {
415 RTC_DCHECK(rtp_sender_);
416 return rtp_sender_->packet_generator.MaxRtpPacketSize();
417}
418
419void ModuleRtpRtcpImpl2::SetMaxRtpPacketSize(size_t rtp_packet_size) {
420 RTC_DCHECK_LE(rtp_packet_size, IP_PACKET_SIZE)
421 << "rtp packet size too large: " << rtp_packet_size;
422 RTC_DCHECK_GT(rtp_packet_size, packet_overhead_)
423 << "rtp packet size too small: " << rtp_packet_size;
424
425 rtcp_sender_.SetMaxRtpPacketSize(rtp_packet_size);
426 if (rtp_sender_) {
427 rtp_sender_->packet_generator.SetMaxRtpPacketSize(rtp_packet_size);
428 }
429}
430
431RtcpMode ModuleRtpRtcpImpl2::RTCP() const {
432 return rtcp_sender_.Status();
433}
434
435// Configure RTCP status i.e on/off.
436void ModuleRtpRtcpImpl2::SetRTCPStatus(const RtcpMode method) {
437 rtcp_sender_.SetRTCPStatus(method);
438}
439
440int32_t ModuleRtpRtcpImpl2::SetCNAME(const char* c_name) {
441 return rtcp_sender_.SetCNAME(c_name);
442}
443
Tommi3a5742c2020-05-20 09:32:51 +0200444int32_t ModuleRtpRtcpImpl2::RemoteNTP(uint32_t* received_ntpsecs,
445 uint32_t* received_ntpfrac,
446 uint32_t* rtcp_arrival_time_secs,
447 uint32_t* rtcp_arrival_time_frac,
448 uint32_t* rtcp_timestamp) const {
449 return rtcp_receiver_.NTP(received_ntpsecs, received_ntpfrac,
450 rtcp_arrival_time_secs, rtcp_arrival_time_frac,
Alessio Bazzica048adc72021-03-10 15:05:55 +0100451 rtcp_timestamp,
452 /*remote_sender_packet_count=*/nullptr,
453 /*remote_sender_octet_count=*/nullptr,
454 /*remote_sender_reports_count=*/nullptr)
Tommi3a5742c2020-05-20 09:32:51 +0200455 ? 0
456 : -1;
457}
458
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200459// TODO(tommi): Check if |avg_rtt_ms|, |min_rtt_ms|, |max_rtt_ms| params are
460// actually used in practice (some callers ask for it but don't use it). It
461// could be that only |rtt| is needed and if so, then the fast path could be to
462// just call rtt_ms() and rely on the calculation being done periodically.
Tommi3a5742c2020-05-20 09:32:51 +0200463int32_t ModuleRtpRtcpImpl2::RTT(const uint32_t remote_ssrc,
464 int64_t* rtt,
465 int64_t* avg_rtt,
466 int64_t* min_rtt,
467 int64_t* max_rtt) const {
468 int32_t ret = rtcp_receiver_.RTT(remote_ssrc, rtt, avg_rtt, min_rtt, max_rtt);
469 if (rtt && *rtt == 0) {
470 // Try to get RTT from RtcpRttStats class.
471 *rtt = rtt_ms();
472 }
473 return ret;
474}
475
476int64_t ModuleRtpRtcpImpl2::ExpectedRetransmissionTimeMs() const {
477 int64_t expected_retransmission_time_ms = rtt_ms();
478 if (expected_retransmission_time_ms > 0) {
479 return expected_retransmission_time_ms;
480 }
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200481 // No rtt available (|kRttUpdateInterval| not yet passed?), so try to
Tommi3a5742c2020-05-20 09:32:51 +0200482 // poll avg_rtt_ms directly from rtcp receiver.
483 if (rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), nullptr,
484 &expected_retransmission_time_ms, nullptr,
485 nullptr) == 0) {
486 return expected_retransmission_time_ms;
487 }
488 return kDefaultExpectedRetransmissionTimeMs;
489}
490
491// Force a send of an RTCP packet.
492// Normal SR and RR are triggered via the process function.
493int32_t ModuleRtpRtcpImpl2::SendRTCP(RTCPPacketType packet_type) {
494 return rtcp_sender_.SendRTCP(GetFeedbackState(), packet_type);
495}
496
Tommi3a5742c2020-05-20 09:32:51 +0200497void ModuleRtpRtcpImpl2::GetSendStreamDataCounters(
498 StreamDataCounters* rtp_counters,
499 StreamDataCounters* rtx_counters) const {
500 rtp_sender_->packet_sender.GetDataCounters(rtp_counters, rtx_counters);
501}
502
503// Received RTCP report.
Tommi3a5742c2020-05-20 09:32:51 +0200504std::vector<ReportBlockData> ModuleRtpRtcpImpl2::GetLatestReportBlockData()
505 const {
506 return rtcp_receiver_.GetLatestReportBlockData();
507}
508
Alessio Bazzicabc1c93d2021-03-12 17:45:26 +0100509absl::optional<RtpRtcpInterface::SenderReportStats>
510ModuleRtpRtcpImpl2::GetSenderReportStats() const {
511 SenderReportStats stats;
512 uint32_t remote_timestamp_secs;
513 uint32_t remote_timestamp_frac;
514 uint32_t arrival_timestamp_secs;
515 uint32_t arrival_timestamp_frac;
516 if (rtcp_receiver_.NTP(&remote_timestamp_secs, &remote_timestamp_frac,
517 &arrival_timestamp_secs, &arrival_timestamp_frac,
518 /*rtcp_timestamp=*/nullptr, &stats.packets_sent,
519 &stats.bytes_sent, &stats.reports_count)) {
520 stats.last_remote_timestamp.Set(remote_timestamp_secs,
521 remote_timestamp_frac);
522 stats.last_arrival_timestamp.Set(arrival_timestamp_secs,
523 arrival_timestamp_frac);
524 return stats;
525 }
526 return absl::nullopt;
527}
528
Tommi3a5742c2020-05-20 09:32:51 +0200529// (REMB) Receiver Estimated Max Bitrate.
530void ModuleRtpRtcpImpl2::SetRemb(int64_t bitrate_bps,
531 std::vector<uint32_t> ssrcs) {
532 rtcp_sender_.SetRemb(bitrate_bps, std::move(ssrcs));
533}
534
535void ModuleRtpRtcpImpl2::UnsetRemb() {
536 rtcp_sender_.UnsetRemb();
537}
538
539void ModuleRtpRtcpImpl2::SetExtmapAllowMixed(bool extmap_allow_mixed) {
540 rtp_sender_->packet_generator.SetExtmapAllowMixed(extmap_allow_mixed);
541}
542
Tommi3a5742c2020-05-20 09:32:51 +0200543void ModuleRtpRtcpImpl2::RegisterRtpHeaderExtension(absl::string_view uri,
544 int id) {
545 bool registered =
546 rtp_sender_->packet_generator.RegisterRtpHeaderExtension(uri, id);
547 RTC_CHECK(registered);
548}
549
550int32_t ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
551 const RTPExtensionType type) {
552 return rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(type);
553}
554void ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
555 absl::string_view uri) {
556 rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(uri);
557}
558
Tommi3a5742c2020-05-20 09:32:51 +0200559void ModuleRtpRtcpImpl2::SetTmmbn(std::vector<rtcp::TmmbItem> bounding_set) {
560 rtcp_sender_.SetTmmbn(std::move(bounding_set));
561}
562
563// Send a Negative acknowledgment packet.
564int32_t ModuleRtpRtcpImpl2::SendNACK(const uint16_t* nack_list,
565 const uint16_t size) {
566 uint16_t nack_length = size;
567 uint16_t start_id = 0;
568 int64_t now_ms = clock_->TimeInMilliseconds();
569 if (TimeToSendFullNackList(now_ms)) {
570 nack_last_time_sent_full_ms_ = now_ms;
571 } else {
572 // Only send extended list.
573 if (nack_last_seq_number_sent_ == nack_list[size - 1]) {
574 // Last sequence number is the same, do not send list.
575 return 0;
576 }
577 // Send new sequence numbers.
578 for (int i = 0; i < size; ++i) {
579 if (nack_last_seq_number_sent_ == nack_list[i]) {
580 start_id = i + 1;
581 break;
582 }
583 }
584 nack_length = size - start_id;
585 }
586
587 // Our RTCP NACK implementation is limited to kRtcpMaxNackFields sequence
588 // numbers per RTCP packet.
589 if (nack_length > kRtcpMaxNackFields) {
590 nack_length = kRtcpMaxNackFields;
591 }
592 nack_last_seq_number_sent_ = nack_list[start_id + nack_length - 1];
593
594 return rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, nack_length,
595 &nack_list[start_id]);
596}
597
598void ModuleRtpRtcpImpl2::SendNack(
599 const std::vector<uint16_t>& sequence_numbers) {
600 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, sequence_numbers.size(),
601 sequence_numbers.data());
602}
603
604bool ModuleRtpRtcpImpl2::TimeToSendFullNackList(int64_t now) const {
605 // Use RTT from RtcpRttStats class if provided.
606 int64_t rtt = rtt_ms();
607 if (rtt == 0) {
608 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
609 }
610
611 const int64_t kStartUpRttMs = 100;
612 int64_t wait_time = 5 + ((rtt * 3) >> 1); // 5 + RTT * 1.5.
613 if (rtt == 0) {
614 wait_time = kStartUpRttMs;
615 }
616
617 // Send a full NACK list once within every |wait_time|.
618 return now - nack_last_time_sent_full_ms_ > wait_time;
619}
620
621// Store the sent packets, needed to answer to Negative acknowledgment requests.
622void ModuleRtpRtcpImpl2::SetStorePacketsStatus(const bool enable,
623 const uint16_t number_to_store) {
624 rtp_sender_->packet_history.SetStorePacketsStatus(
625 enable ? RtpPacketHistory::StorageMode::kStoreAndCull
626 : RtpPacketHistory::StorageMode::kDisabled,
627 number_to_store);
628}
629
630bool ModuleRtpRtcpImpl2::StorePackets() const {
631 return rtp_sender_->packet_history.GetStorageMode() !=
632 RtpPacketHistory::StorageMode::kDisabled;
633}
634
635void ModuleRtpRtcpImpl2::SendCombinedRtcpPacket(
636 std::vector<std::unique_ptr<rtcp::RtcpPacket>> rtcp_packets) {
637 rtcp_sender_.SendCombinedRtcpPacket(std::move(rtcp_packets));
638}
639
640int32_t ModuleRtpRtcpImpl2::SendLossNotification(uint16_t last_decoded_seq_num,
641 uint16_t last_received_seq_num,
642 bool decodability_flag,
643 bool buffering_allowed) {
644 return rtcp_sender_.SendLossNotification(
645 GetFeedbackState(), last_decoded_seq_num, last_received_seq_num,
646 decodability_flag, buffering_allowed);
647}
648
649void ModuleRtpRtcpImpl2::SetRemoteSSRC(const uint32_t ssrc) {
650 // Inform about the incoming SSRC.
651 rtcp_sender_.SetRemoteSSRC(ssrc);
652 rtcp_receiver_.SetRemoteSSRC(ssrc);
653}
654
Tommi08be9ba2021-06-15 23:01:57 +0200655void ModuleRtpRtcpImpl2::SetLocalSsrc(uint32_t local_ssrc) {
656 RTC_DCHECK_RUN_ON(&packet_sequence_checker_);
657 rtcp_receiver_.set_local_media_ssrc(local_ssrc);
658 rtcp_sender_.SetSsrc(local_ssrc);
659}
660
Tommi3a5742c2020-05-20 09:32:51 +0200661RtpSendRates ModuleRtpRtcpImpl2::GetSendRates() const {
Tommi1050fbc2021-06-03 17:58:28 +0200662 // Typically called on the `rtp_transport_queue_` owned by an
663 // RtpTransportControllerSendInterface instance.
Tommi3a5742c2020-05-20 09:32:51 +0200664 return rtp_sender_->packet_sender.GetSendRates();
665}
666
667void ModuleRtpRtcpImpl2::OnRequestSendReport() {
668 SendRTCP(kRtcpSr);
669}
670
671void ModuleRtpRtcpImpl2::OnReceivedNack(
672 const std::vector<uint16_t>& nack_sequence_numbers) {
673 if (!rtp_sender_)
674 return;
675
676 if (!StorePackets() || nack_sequence_numbers.empty()) {
677 return;
678 }
679 // Use RTT from RtcpRttStats class if provided.
680 int64_t rtt = rtt_ms();
681 if (rtt == 0) {
682 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
683 }
684 rtp_sender_->packet_generator.OnReceivedNack(nack_sequence_numbers, rtt);
685}
686
687void ModuleRtpRtcpImpl2::OnReceivedRtcpReportBlocks(
688 const ReportBlockList& report_blocks) {
689 if (rtp_sender_) {
690 uint32_t ssrc = SSRC();
691 absl::optional<uint32_t> rtx_ssrc;
692 if (rtp_sender_->packet_generator.RtxStatus() != kRtxOff) {
693 rtx_ssrc = rtp_sender_->packet_generator.RtxSsrc();
694 }
695
696 for (const RTCPReportBlock& report_block : report_blocks) {
697 if (ssrc == report_block.source_ssrc) {
698 rtp_sender_->packet_generator.OnReceivedAckOnSsrc(
699 report_block.extended_highest_sequence_number);
700 } else if (rtx_ssrc && *rtx_ssrc == report_block.source_ssrc) {
701 rtp_sender_->packet_generator.OnReceivedAckOnRtxSsrc(
702 report_block.extended_highest_sequence_number);
703 }
704 }
705 }
706}
707
Tommi3a5742c2020-05-20 09:32:51 +0200708void ModuleRtpRtcpImpl2::set_rtt_ms(int64_t rtt_ms) {
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200709 RTC_DCHECK_RUN_ON(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +0200710 {
Markus Handellf7303e62020-07-09 01:34:42 +0200711 MutexLock lock(&mutex_rtt_);
Tommi3a5742c2020-05-20 09:32:51 +0200712 rtt_ms_ = rtt_ms;
713 }
714 if (rtp_sender_) {
715 rtp_sender_->packet_history.SetRtt(rtt_ms);
716 }
717}
718
719int64_t ModuleRtpRtcpImpl2::rtt_ms() const {
Markus Handellf7303e62020-07-09 01:34:42 +0200720 MutexLock lock(&mutex_rtt_);
Tommi3a5742c2020-05-20 09:32:51 +0200721 return rtt_ms_;
722}
723
724void ModuleRtpRtcpImpl2::SetVideoBitrateAllocation(
725 const VideoBitrateAllocation& bitrate) {
726 rtcp_sender_.SetVideoBitrateAllocation(bitrate);
727}
728
729RTPSender* ModuleRtpRtcpImpl2::RtpSender() {
730 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
731}
732
733const RTPSender* ModuleRtpRtcpImpl2::RtpSender() const {
734 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
735}
736
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200737void ModuleRtpRtcpImpl2::PeriodicUpdate() {
738 RTC_DCHECK_RUN_ON(worker_queue_);
739
740 Timestamp check_since = clock_->CurrentTime() - kRttUpdateInterval;
741 absl::optional<TimeDelta> rtt =
742 rtcp_receiver_.OnPeriodicRttUpdate(check_since, rtcp_sender_.Sending());
743 if (rtt) {
Ivo Creusen8c40d512021-07-13 12:53:22 +0000744 if (rtt_stats_) {
745 rtt_stats_->OnRttUpdate(rtt->ms());
746 }
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200747 set_rtt_ms(rtt->ms());
748 }
Tomas Gunnarssonba0ba712020-07-01 08:53:21 +0200749}
750
Markus Handell885d5382021-06-21 18:57:36 +0200751// RTC_RUN_ON(worker_queue_);
752void ModuleRtpRtcpImpl2::MaybeSendRtcp() {
753 if (rtcp_sender_.TimeToSendRTCPReport())
754 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
755}
756
757// TODO(bugs.webrtc.org/12889): Consider removing this function when the issue
758// is resolved.
759// RTC_RUN_ON(worker_queue_);
760void ModuleRtpRtcpImpl2::MaybeSendRtcpAtOrAfterTimestamp(
761 Timestamp execution_time) {
762 Timestamp now = clock_->CurrentTime();
763 if (now >= execution_time) {
764 MaybeSendRtcp();
765 return;
766 }
767
768 RTC_DLOG(LS_WARNING)
769 << "BUGBUG: Task queue scheduled delayed call too early.";
770
771 ScheduleMaybeSendRtcpAtOrAfterTimestamp(execution_time, execution_time - now);
772}
773
774void ModuleRtpRtcpImpl2::ScheduleRtcpSendEvaluation(TimeDelta duration) {
775 // We end up here under various sequences including the worker queue, and
776 // the RTCPSender lock is held.
777 // We're assuming that the fact that RTCPSender executes under other sequences
778 // than the worker queue on which it's created on implies that external
779 // synchronization is present and removes this activity before destruction.
780 if (duration.IsZero()) {
781 worker_queue_->PostTask(ToQueuedTask(task_safety_, [this] {
782 RTC_DCHECK_RUN_ON(worker_queue_);
783 MaybeSendRtcp();
784 }));
785 } else {
786 Timestamp execution_time = clock_->CurrentTime() + duration;
787 ScheduleMaybeSendRtcpAtOrAfterTimestamp(execution_time, duration);
788 }
789}
790
791void ModuleRtpRtcpImpl2::ScheduleMaybeSendRtcpAtOrAfterTimestamp(
792 Timestamp execution_time,
793 TimeDelta duration) {
794 // We end up here under various sequences including the worker queue, and
795 // the RTCPSender lock is held.
796 // See note in ScheduleRtcpSendEvaluation about why |worker_queue_| can be
797 // accessed.
798 worker_queue_->PostDelayedTask(
799 ToQueuedTask(task_safety_,
800 [this, execution_time] {
801 RTC_DCHECK_RUN_ON(worker_queue_);
802 MaybeSendRtcpAtOrAfterTimestamp(execution_time);
803 }),
804 DelayMillisForDuration(duration));
805}
806
Tommi3a5742c2020-05-20 09:32:51 +0200807} // namespace webrtc