blob: e50f72bb2925267cd65202a9450ec16a54dd8465 [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
22#include "api/transport/field_trial_based_config.h"
23#include "modules/rtp_rtcp/source/rtcp_packet/dlrr.h"
24#include "modules/rtp_rtcp/source/rtp_rtcp_config.h"
25#include "rtc_base/checks.h"
26#include "rtc_base/logging.h"
27
28#ifdef _WIN32
29// Disable warning C4355: 'this' : used in base member initializer list.
30#pragma warning(disable : 4355)
31#endif
32
33namespace webrtc {
34namespace {
35const int64_t kRtpRtcpMaxIdleTimeProcessMs = 5;
36const int64_t kRtpRtcpRttProcessTimeMs = 1000;
37const int64_t kRtpRtcpBitrateProcessTimeMs = 10;
38const int64_t kDefaultExpectedRetransmissionTimeMs = 125;
39} // namespace
40
41ModuleRtpRtcpImpl2::RtpSenderContext::RtpSenderContext(
Tomas Gunnarssonf25761d2020-06-03 22:55:33 +020042 const RtpRtcpInterface::Configuration& config)
Tommi3a5742c2020-05-20 09:32:51 +020043 : packet_history(config.clock, config.enable_rtx_padding_prioritization),
44 packet_sender(config, &packet_history),
Erik Språng1b485322020-06-24 18:39:25 +000045 non_paced_sender(&packet_sender),
Tommi3a5742c2020-05-20 09:32:51 +020046 packet_generator(
47 config,
48 &packet_history,
49 config.paced_sender ? config.paced_sender : &non_paced_sender) {}
50
Tommi3a5742c2020-05-20 09:32:51 +020051ModuleRtpRtcpImpl2::ModuleRtpRtcpImpl2(const Configuration& configuration)
52 : rtcp_sender_(configuration),
53 rtcp_receiver_(configuration, this),
54 clock_(configuration.clock),
55 last_bitrate_process_time_(clock_->TimeInMilliseconds()),
56 last_rtt_process_time_(clock_->TimeInMilliseconds()),
57 next_process_time_(clock_->TimeInMilliseconds() +
58 kRtpRtcpMaxIdleTimeProcessMs),
59 packet_overhead_(28), // IPV4 UDP.
60 nack_last_time_sent_full_ms_(0),
61 nack_last_seq_number_sent_(0),
62 remote_bitrate_(configuration.remote_bitrate_estimator),
63 rtt_stats_(configuration.rtt_stats),
64 rtt_ms_(0) {
65 process_thread_checker_.Detach();
66 if (!configuration.receiver_only) {
67 rtp_sender_ = std::make_unique<RtpSenderContext>(configuration);
68 // Make sure rtcp sender use same timestamp offset as rtp sender.
69 rtcp_sender_.SetTimestampOffset(
70 rtp_sender_->packet_generator.TimestampOffset());
71 }
72
73 // Set default packet size limit.
74 // TODO(nisse): Kind-of duplicates
75 // webrtc::VideoSendStream::Config::Rtp::kDefaultMaxPacketSize.
76 const size_t kTcpOverIpv4HeaderSize = 40;
77 SetMaxRtpPacketSize(IP_PACKET_SIZE - kTcpOverIpv4HeaderSize);
78}
79
80ModuleRtpRtcpImpl2::~ModuleRtpRtcpImpl2() {
81 RTC_DCHECK_RUN_ON(&construction_thread_checker_);
82}
83
Tomas Gunnarssonfae05622020-06-03 08:54:39 +020084// static
Tomas Gunnarssonf25761d2020-06-03 22:55:33 +020085std::unique_ptr<ModuleRtpRtcpImpl2> ModuleRtpRtcpImpl2::Create(
Tomas Gunnarssonfae05622020-06-03 08:54:39 +020086 const Configuration& configuration) {
87 RTC_DCHECK(configuration.clock);
88 RTC_DCHECK(TaskQueueBase::Current());
89 return std::make_unique<ModuleRtpRtcpImpl2>(configuration);
90}
91
Tommi3a5742c2020-05-20 09:32:51 +020092// Returns the number of milliseconds until the module want a worker thread
93// to call Process.
94int64_t ModuleRtpRtcpImpl2::TimeUntilNextProcess() {
95 RTC_DCHECK_RUN_ON(&process_thread_checker_);
96 return std::max<int64_t>(0,
97 next_process_time_ - clock_->TimeInMilliseconds());
98}
99
100// Process any pending tasks such as timeouts (non time critical events).
101void ModuleRtpRtcpImpl2::Process() {
102 RTC_DCHECK_RUN_ON(&process_thread_checker_);
103 const int64_t now = clock_->TimeInMilliseconds();
104 // TODO(bugs.webrtc.org/11581): Figure out why we need to call Process() 200
105 // times a second.
106 next_process_time_ = now + kRtpRtcpMaxIdleTimeProcessMs;
107
108 if (rtp_sender_) {
109 if (now >= last_bitrate_process_time_ + kRtpRtcpBitrateProcessTimeMs) {
110 rtp_sender_->packet_sender.ProcessBitrateAndNotifyObservers();
111 last_bitrate_process_time_ = now;
112 // TODO(bugs.webrtc.org/11581): Is this a bug? At the top of the function,
113 // next_process_time_ is incremented by 5ms, here we effectively do a
114 // std::min() of (now + 5ms, now + 10ms). Seems like this is a no-op?
115 next_process_time_ =
116 std::min(next_process_time_, now + kRtpRtcpBitrateProcessTimeMs);
117 }
118 }
119
120 // TODO(bugs.webrtc.org/11581): We update the RTT once a second, whereas other
121 // things that run in this method are updated much more frequently. Move the
122 // RTT checking over to the worker thread, which matches better with where the
123 // stats are maintained.
124 bool process_rtt = now >= last_rtt_process_time_ + kRtpRtcpRttProcessTimeMs;
125 if (rtcp_sender_.Sending()) {
126 // Process RTT if we have received a report block and we haven't
127 // processed RTT for at least |kRtpRtcpRttProcessTimeMs| milliseconds.
128 // Note that LastReceivedReportBlockMs() grabs a lock, so check
129 // |process_rtt| first.
130 if (process_rtt &&
131 rtcp_receiver_.LastReceivedReportBlockMs() > last_rtt_process_time_) {
132 std::vector<RTCPReportBlock> receive_blocks;
133 rtcp_receiver_.StatisticsReceived(&receive_blocks);
134 int64_t max_rtt = 0;
135 for (std::vector<RTCPReportBlock>::iterator it = receive_blocks.begin();
136 it != receive_blocks.end(); ++it) {
137 int64_t rtt = 0;
138 rtcp_receiver_.RTT(it->sender_ssrc, &rtt, NULL, NULL, NULL);
139 max_rtt = (rtt > max_rtt) ? rtt : max_rtt;
140 }
141 // Report the rtt.
142 if (rtt_stats_ && max_rtt != 0)
143 rtt_stats_->OnRttUpdate(max_rtt);
144 }
145
146 // Verify receiver reports are delivered and the reported sequence number
147 // is increasing.
148 // TODO(bugs.webrtc.org/11581): The timeout value needs to be checked every
149 // few seconds (see internals of RtcpRrTimeout). Here, we may be polling it
150 // a couple of hundred times a second, which isn't great since it grabs a
151 // lock. Note also that LastReceivedReportBlockMs() (called above) and
152 // RtcpRrTimeout() both grab the same lock and check the same timer, so
153 // it should be possible to consolidate that work somehow.
154 if (rtcp_receiver_.RtcpRrTimeout()) {
155 RTC_LOG_F(LS_WARNING) << "Timeout: No RTCP RR received.";
156 } else if (rtcp_receiver_.RtcpRrSequenceNumberTimeout()) {
157 RTC_LOG_F(LS_WARNING) << "Timeout: No increase in RTCP RR extended "
158 "highest sequence number.";
159 }
160
161 if (remote_bitrate_ && rtcp_sender_.TMMBR()) {
162 unsigned int target_bitrate = 0;
163 std::vector<unsigned int> ssrcs;
164 if (remote_bitrate_->LatestEstimate(&ssrcs, &target_bitrate)) {
165 if (!ssrcs.empty()) {
166 target_bitrate = target_bitrate / ssrcs.size();
167 }
168 rtcp_sender_.SetTargetBitrate(target_bitrate);
169 }
170 }
171 } else {
172 // Report rtt from receiver.
173 if (process_rtt) {
174 int64_t rtt_ms;
175 if (rtt_stats_ && rtcp_receiver_.GetAndResetXrRrRtt(&rtt_ms)) {
176 rtt_stats_->OnRttUpdate(rtt_ms);
177 }
178 }
179 }
180
181 // Get processed rtt.
182 if (process_rtt) {
183 last_rtt_process_time_ = now;
184 // TODO(bugs.webrtc.org/11581): Is this a bug? At the top of the function,
185 // next_process_time_ is incremented by 5ms, here we effectively do a
186 // std::min() of (now + 5ms, now + 1000ms). Seems like this is a no-op?
187 next_process_time_ = std::min(
188 next_process_time_, last_rtt_process_time_ + kRtpRtcpRttProcessTimeMs);
189 if (rtt_stats_) {
190 // Make sure we have a valid RTT before setting.
191 int64_t last_rtt = rtt_stats_->LastProcessedRtt();
192 if (last_rtt >= 0)
193 set_rtt_ms(last_rtt);
194 }
195 }
196
197 if (rtcp_sender_.TimeToSendRTCPReport())
198 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
199
Tomas Gunnarsson64348642020-06-09 08:02:44 +0200200 if (rtcp_sender_.TMMBR() && rtcp_receiver_.UpdateTmmbrTimers()) {
Tommi3a5742c2020-05-20 09:32:51 +0200201 rtcp_receiver_.NotifyTmmbrUpdated();
202 }
203}
204
205void ModuleRtpRtcpImpl2::SetRtxSendStatus(int mode) {
206 rtp_sender_->packet_generator.SetRtxStatus(mode);
207}
208
209int ModuleRtpRtcpImpl2::RtxSendStatus() const {
210 return rtp_sender_ ? rtp_sender_->packet_generator.RtxStatus() : kRtxOff;
211}
212
213void ModuleRtpRtcpImpl2::SetRtxSendPayloadType(int payload_type,
214 int associated_payload_type) {
215 rtp_sender_->packet_generator.SetRtxPayloadType(payload_type,
216 associated_payload_type);
217}
218
219absl::optional<uint32_t> ModuleRtpRtcpImpl2::RtxSsrc() const {
220 return rtp_sender_ ? rtp_sender_->packet_generator.RtxSsrc() : absl::nullopt;
221}
222
223absl::optional<uint32_t> ModuleRtpRtcpImpl2::FlexfecSsrc() const {
224 if (rtp_sender_) {
225 return rtp_sender_->packet_generator.FlexfecSsrc();
226 }
227 return absl::nullopt;
228}
229
230void ModuleRtpRtcpImpl2::IncomingRtcpPacket(const uint8_t* rtcp_packet,
231 const size_t length) {
232 rtcp_receiver_.IncomingPacket(rtcp_packet, length);
233}
234
235void ModuleRtpRtcpImpl2::RegisterSendPayloadFrequency(int payload_type,
236 int payload_frequency) {
237 rtcp_sender_.SetRtpClockRate(payload_type, payload_frequency);
238}
239
240int32_t ModuleRtpRtcpImpl2::DeRegisterSendPayload(const int8_t payload_type) {
241 return 0;
242}
243
244uint32_t ModuleRtpRtcpImpl2::StartTimestamp() const {
245 return rtp_sender_->packet_generator.TimestampOffset();
246}
247
248// Configure start timestamp, default is a random number.
249void ModuleRtpRtcpImpl2::SetStartTimestamp(const uint32_t timestamp) {
250 rtcp_sender_.SetTimestampOffset(timestamp);
251 rtp_sender_->packet_generator.SetTimestampOffset(timestamp);
252 rtp_sender_->packet_sender.SetTimestampOffset(timestamp);
253}
254
255uint16_t ModuleRtpRtcpImpl2::SequenceNumber() const {
256 return rtp_sender_->packet_generator.SequenceNumber();
257}
258
259// Set SequenceNumber, default is a random number.
260void ModuleRtpRtcpImpl2::SetSequenceNumber(const uint16_t seq_num) {
261 rtp_sender_->packet_generator.SetSequenceNumber(seq_num);
262}
263
264void ModuleRtpRtcpImpl2::SetRtpState(const RtpState& rtp_state) {
265 rtp_sender_->packet_generator.SetRtpState(rtp_state);
266 rtp_sender_->packet_sender.SetMediaHasBeenSent(rtp_state.media_has_been_sent);
267 rtcp_sender_.SetTimestampOffset(rtp_state.start_timestamp);
268}
269
270void ModuleRtpRtcpImpl2::SetRtxState(const RtpState& rtp_state) {
271 rtp_sender_->packet_generator.SetRtxRtpState(rtp_state);
272}
273
274RtpState ModuleRtpRtcpImpl2::GetRtpState() const {
275 RtpState state = rtp_sender_->packet_generator.GetRtpState();
276 state.media_has_been_sent = rtp_sender_->packet_sender.MediaHasBeenSent();
277 return state;
278}
279
280RtpState ModuleRtpRtcpImpl2::GetRtxState() const {
281 return rtp_sender_->packet_generator.GetRtxRtpState();
282}
283
284void ModuleRtpRtcpImpl2::SetRid(const std::string& rid) {
285 if (rtp_sender_) {
286 rtp_sender_->packet_generator.SetRid(rid);
287 }
288}
289
290void ModuleRtpRtcpImpl2::SetMid(const std::string& mid) {
291 if (rtp_sender_) {
292 rtp_sender_->packet_generator.SetMid(mid);
293 }
294 // TODO(bugs.webrtc.org/4050): If we end up supporting the MID SDES item for
295 // RTCP, this will need to be passed down to the RTCPSender also.
296}
297
298void ModuleRtpRtcpImpl2::SetCsrcs(const std::vector<uint32_t>& csrcs) {
299 rtcp_sender_.SetCsrcs(csrcs);
300 rtp_sender_->packet_generator.SetCsrcs(csrcs);
301}
302
303// TODO(pbos): Handle media and RTX streams separately (separate RTCP
304// feedbacks).
305RTCPSender::FeedbackState ModuleRtpRtcpImpl2::GetFeedbackState() {
306 RTCPSender::FeedbackState state;
307 // This is called also when receiver_only is true. Hence below
308 // checks that rtp_sender_ exists.
309 if (rtp_sender_) {
310 StreamDataCounters rtp_stats;
311 StreamDataCounters rtx_stats;
312 rtp_sender_->packet_sender.GetDataCounters(&rtp_stats, &rtx_stats);
313 state.packets_sent =
314 rtp_stats.transmitted.packets + rtx_stats.transmitted.packets;
315 state.media_bytes_sent = rtp_stats.transmitted.payload_bytes +
316 rtx_stats.transmitted.payload_bytes;
317 state.send_bitrate =
318 rtp_sender_->packet_sender.GetSendRates().Sum().bps<uint32_t>();
319 }
320 state.receiver = &rtcp_receiver_;
321
322 LastReceivedNTP(&state.last_rr_ntp_secs, &state.last_rr_ntp_frac,
323 &state.remote_sr);
324
325 state.last_xr_rtis = rtcp_receiver_.ConsumeReceivedXrReferenceTimeInfo();
326
327 return state;
328}
329
330// TODO(nisse): This method shouldn't be called for a receive-only
331// stream. Delete rtp_sender_ check as soon as all applications are
332// updated.
333int32_t ModuleRtpRtcpImpl2::SetSendingStatus(const bool sending) {
334 if (rtcp_sender_.Sending() != sending) {
335 // Sends RTCP BYE when going from true to false
336 if (rtcp_sender_.SetSendingStatus(GetFeedbackState(), sending) != 0) {
337 RTC_LOG(LS_WARNING) << "Failed to send RTCP BYE";
338 }
339 }
340 return 0;
341}
342
343bool ModuleRtpRtcpImpl2::Sending() const {
344 return rtcp_sender_.Sending();
345}
346
347// TODO(nisse): This method shouldn't be called for a receive-only
348// stream. Delete rtp_sender_ check as soon as all applications are
349// updated.
350void ModuleRtpRtcpImpl2::SetSendingMediaStatus(const bool sending) {
351 if (rtp_sender_) {
352 rtp_sender_->packet_generator.SetSendingMediaStatus(sending);
353 } else {
354 RTC_DCHECK(!sending);
355 }
356}
357
358bool ModuleRtpRtcpImpl2::SendingMedia() const {
359 return rtp_sender_ ? rtp_sender_->packet_generator.SendingMedia() : false;
360}
361
362bool ModuleRtpRtcpImpl2::IsAudioConfigured() const {
363 return rtp_sender_ ? rtp_sender_->packet_generator.IsAudioConfigured()
364 : false;
365}
366
367void ModuleRtpRtcpImpl2::SetAsPartOfAllocation(bool part_of_allocation) {
368 RTC_CHECK(rtp_sender_);
369 rtp_sender_->packet_sender.ForceIncludeSendPacketsInAllocation(
370 part_of_allocation);
371}
372
373bool ModuleRtpRtcpImpl2::OnSendingRtpFrame(uint32_t timestamp,
374 int64_t capture_time_ms,
375 int payload_type,
376 bool force_sender_report) {
377 if (!Sending())
378 return false;
379
380 rtcp_sender_.SetLastRtpTime(timestamp, capture_time_ms, payload_type);
381 // Make sure an RTCP report isn't queued behind a key frame.
382 if (rtcp_sender_.TimeToSendRTCPReport(force_sender_report))
383 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
384
385 return true;
386}
387
388bool ModuleRtpRtcpImpl2::TrySendPacket(RtpPacketToSend* packet,
389 const PacedPacketInfo& pacing_info) {
390 RTC_DCHECK(rtp_sender_);
391 // TODO(sprang): Consider if we can remove this check.
392 if (!rtp_sender_->packet_generator.SendingMedia()) {
393 return false;
394 }
395 rtp_sender_->packet_sender.SendPacket(packet, pacing_info);
396 return true;
397}
398
399void ModuleRtpRtcpImpl2::OnPacketsAcknowledged(
400 rtc::ArrayView<const uint16_t> sequence_numbers) {
401 RTC_DCHECK(rtp_sender_);
402 rtp_sender_->packet_history.CullAcknowledgedPackets(sequence_numbers);
403}
404
405bool ModuleRtpRtcpImpl2::SupportsPadding() const {
406 RTC_DCHECK(rtp_sender_);
407 return rtp_sender_->packet_generator.SupportsPadding();
408}
409
410bool ModuleRtpRtcpImpl2::SupportsRtxPayloadPadding() const {
411 RTC_DCHECK(rtp_sender_);
412 return rtp_sender_->packet_generator.SupportsRtxPayloadPadding();
413}
414
415std::vector<std::unique_ptr<RtpPacketToSend>>
416ModuleRtpRtcpImpl2::GeneratePadding(size_t target_size_bytes) {
417 RTC_DCHECK(rtp_sender_);
418 return rtp_sender_->packet_generator.GeneratePadding(
419 target_size_bytes, rtp_sender_->packet_sender.MediaHasBeenSent());
420}
421
422std::vector<RtpSequenceNumberMap::Info>
423ModuleRtpRtcpImpl2::GetSentRtpPacketInfos(
424 rtc::ArrayView<const uint16_t> sequence_numbers) const {
425 RTC_DCHECK(rtp_sender_);
426 return rtp_sender_->packet_sender.GetSentRtpPacketInfos(sequence_numbers);
427}
428
429size_t ModuleRtpRtcpImpl2::ExpectedPerPacketOverhead() const {
430 if (!rtp_sender_) {
431 return 0;
432 }
433 return rtp_sender_->packet_generator.ExpectedPerPacketOverhead();
434}
435
436size_t ModuleRtpRtcpImpl2::MaxRtpPacketSize() const {
437 RTC_DCHECK(rtp_sender_);
438 return rtp_sender_->packet_generator.MaxRtpPacketSize();
439}
440
441void ModuleRtpRtcpImpl2::SetMaxRtpPacketSize(size_t rtp_packet_size) {
442 RTC_DCHECK_LE(rtp_packet_size, IP_PACKET_SIZE)
443 << "rtp packet size too large: " << rtp_packet_size;
444 RTC_DCHECK_GT(rtp_packet_size, packet_overhead_)
445 << "rtp packet size too small: " << rtp_packet_size;
446
447 rtcp_sender_.SetMaxRtpPacketSize(rtp_packet_size);
448 if (rtp_sender_) {
449 rtp_sender_->packet_generator.SetMaxRtpPacketSize(rtp_packet_size);
450 }
451}
452
453RtcpMode ModuleRtpRtcpImpl2::RTCP() const {
454 return rtcp_sender_.Status();
455}
456
457// Configure RTCP status i.e on/off.
458void ModuleRtpRtcpImpl2::SetRTCPStatus(const RtcpMode method) {
459 rtcp_sender_.SetRTCPStatus(method);
460}
461
462int32_t ModuleRtpRtcpImpl2::SetCNAME(const char* c_name) {
463 return rtcp_sender_.SetCNAME(c_name);
464}
465
Tommi3a5742c2020-05-20 09:32:51 +0200466int32_t ModuleRtpRtcpImpl2::RemoteNTP(uint32_t* received_ntpsecs,
467 uint32_t* received_ntpfrac,
468 uint32_t* rtcp_arrival_time_secs,
469 uint32_t* rtcp_arrival_time_frac,
470 uint32_t* rtcp_timestamp) const {
471 return rtcp_receiver_.NTP(received_ntpsecs, received_ntpfrac,
472 rtcp_arrival_time_secs, rtcp_arrival_time_frac,
473 rtcp_timestamp)
474 ? 0
475 : -1;
476}
477
478// Get RoundTripTime.
479int32_t ModuleRtpRtcpImpl2::RTT(const uint32_t remote_ssrc,
480 int64_t* rtt,
481 int64_t* avg_rtt,
482 int64_t* min_rtt,
483 int64_t* max_rtt) const {
484 int32_t ret = rtcp_receiver_.RTT(remote_ssrc, rtt, avg_rtt, min_rtt, max_rtt);
485 if (rtt && *rtt == 0) {
486 // Try to get RTT from RtcpRttStats class.
487 *rtt = rtt_ms();
488 }
489 return ret;
490}
491
492int64_t ModuleRtpRtcpImpl2::ExpectedRetransmissionTimeMs() const {
493 int64_t expected_retransmission_time_ms = rtt_ms();
494 if (expected_retransmission_time_ms > 0) {
495 return expected_retransmission_time_ms;
496 }
497 // No rtt available (|kRtpRtcpRttProcessTimeMs| not yet passed?), so try to
498 // poll avg_rtt_ms directly from rtcp receiver.
499 if (rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), nullptr,
500 &expected_retransmission_time_ms, nullptr,
501 nullptr) == 0) {
502 return expected_retransmission_time_ms;
503 }
504 return kDefaultExpectedRetransmissionTimeMs;
505}
506
507// Force a send of an RTCP packet.
508// Normal SR and RR are triggered via the process function.
509int32_t ModuleRtpRtcpImpl2::SendRTCP(RTCPPacketType packet_type) {
510 return rtcp_sender_.SendRTCP(GetFeedbackState(), packet_type);
511}
512
Tommi3a5742c2020-05-20 09:32:51 +0200513void ModuleRtpRtcpImpl2::SetRtcpXrRrtrStatus(bool enable) {
514 rtcp_receiver_.SetRtcpXrRrtrStatus(enable);
515 rtcp_sender_.SendRtcpXrReceiverReferenceTime(enable);
516}
517
518bool ModuleRtpRtcpImpl2::RtcpXrRrtrStatus() const {
519 return rtcp_sender_.RtcpXrReceiverReferenceTime();
520}
521
Tommi3a5742c2020-05-20 09:32:51 +0200522void ModuleRtpRtcpImpl2::GetSendStreamDataCounters(
523 StreamDataCounters* rtp_counters,
524 StreamDataCounters* rtx_counters) const {
525 rtp_sender_->packet_sender.GetDataCounters(rtp_counters, rtx_counters);
526}
527
528// Received RTCP report.
529int32_t ModuleRtpRtcpImpl2::RemoteRTCPStat(
530 std::vector<RTCPReportBlock>* receive_blocks) const {
531 return rtcp_receiver_.StatisticsReceived(receive_blocks);
532}
533
534std::vector<ReportBlockData> ModuleRtpRtcpImpl2::GetLatestReportBlockData()
535 const {
536 return rtcp_receiver_.GetLatestReportBlockData();
537}
538
539// (REMB) Receiver Estimated Max Bitrate.
540void ModuleRtpRtcpImpl2::SetRemb(int64_t bitrate_bps,
541 std::vector<uint32_t> ssrcs) {
542 rtcp_sender_.SetRemb(bitrate_bps, std::move(ssrcs));
543}
544
545void ModuleRtpRtcpImpl2::UnsetRemb() {
546 rtcp_sender_.UnsetRemb();
547}
548
549void ModuleRtpRtcpImpl2::SetExtmapAllowMixed(bool extmap_allow_mixed) {
550 rtp_sender_->packet_generator.SetExtmapAllowMixed(extmap_allow_mixed);
551}
552
Tommi3a5742c2020-05-20 09:32:51 +0200553void ModuleRtpRtcpImpl2::RegisterRtpHeaderExtension(absl::string_view uri,
554 int id) {
555 bool registered =
556 rtp_sender_->packet_generator.RegisterRtpHeaderExtension(uri, id);
557 RTC_CHECK(registered);
558}
559
560int32_t ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
561 const RTPExtensionType type) {
562 return rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(type);
563}
564void ModuleRtpRtcpImpl2::DeregisterSendRtpHeaderExtension(
565 absl::string_view uri) {
566 rtp_sender_->packet_generator.DeregisterRtpHeaderExtension(uri);
567}
568
Tommi3a5742c2020-05-20 09:32:51 +0200569void ModuleRtpRtcpImpl2::SetTmmbn(std::vector<rtcp::TmmbItem> bounding_set) {
570 rtcp_sender_.SetTmmbn(std::move(bounding_set));
571}
572
573// Send a Negative acknowledgment packet.
574int32_t ModuleRtpRtcpImpl2::SendNACK(const uint16_t* nack_list,
575 const uint16_t size) {
576 uint16_t nack_length = size;
577 uint16_t start_id = 0;
578 int64_t now_ms = clock_->TimeInMilliseconds();
579 if (TimeToSendFullNackList(now_ms)) {
580 nack_last_time_sent_full_ms_ = now_ms;
581 } else {
582 // Only send extended list.
583 if (nack_last_seq_number_sent_ == nack_list[size - 1]) {
584 // Last sequence number is the same, do not send list.
585 return 0;
586 }
587 // Send new sequence numbers.
588 for (int i = 0; i < size; ++i) {
589 if (nack_last_seq_number_sent_ == nack_list[i]) {
590 start_id = i + 1;
591 break;
592 }
593 }
594 nack_length = size - start_id;
595 }
596
597 // Our RTCP NACK implementation is limited to kRtcpMaxNackFields sequence
598 // numbers per RTCP packet.
599 if (nack_length > kRtcpMaxNackFields) {
600 nack_length = kRtcpMaxNackFields;
601 }
602 nack_last_seq_number_sent_ = nack_list[start_id + nack_length - 1];
603
604 return rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, nack_length,
605 &nack_list[start_id]);
606}
607
608void ModuleRtpRtcpImpl2::SendNack(
609 const std::vector<uint16_t>& sequence_numbers) {
610 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpNack, sequence_numbers.size(),
611 sequence_numbers.data());
612}
613
614bool ModuleRtpRtcpImpl2::TimeToSendFullNackList(int64_t now) const {
615 // Use RTT from RtcpRttStats class if provided.
616 int64_t rtt = rtt_ms();
617 if (rtt == 0) {
618 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
619 }
620
621 const int64_t kStartUpRttMs = 100;
622 int64_t wait_time = 5 + ((rtt * 3) >> 1); // 5 + RTT * 1.5.
623 if (rtt == 0) {
624 wait_time = kStartUpRttMs;
625 }
626
627 // Send a full NACK list once within every |wait_time|.
628 return now - nack_last_time_sent_full_ms_ > wait_time;
629}
630
631// Store the sent packets, needed to answer to Negative acknowledgment requests.
632void ModuleRtpRtcpImpl2::SetStorePacketsStatus(const bool enable,
633 const uint16_t number_to_store) {
634 rtp_sender_->packet_history.SetStorePacketsStatus(
635 enable ? RtpPacketHistory::StorageMode::kStoreAndCull
636 : RtpPacketHistory::StorageMode::kDisabled,
637 number_to_store);
638}
639
640bool ModuleRtpRtcpImpl2::StorePackets() const {
641 return rtp_sender_->packet_history.GetStorageMode() !=
642 RtpPacketHistory::StorageMode::kDisabled;
643}
644
645void ModuleRtpRtcpImpl2::SendCombinedRtcpPacket(
646 std::vector<std::unique_ptr<rtcp::RtcpPacket>> rtcp_packets) {
647 rtcp_sender_.SendCombinedRtcpPacket(std::move(rtcp_packets));
648}
649
650int32_t ModuleRtpRtcpImpl2::SendLossNotification(uint16_t last_decoded_seq_num,
651 uint16_t last_received_seq_num,
652 bool decodability_flag,
653 bool buffering_allowed) {
654 return rtcp_sender_.SendLossNotification(
655 GetFeedbackState(), last_decoded_seq_num, last_received_seq_num,
656 decodability_flag, buffering_allowed);
657}
658
659void ModuleRtpRtcpImpl2::SetRemoteSSRC(const uint32_t ssrc) {
660 // Inform about the incoming SSRC.
661 rtcp_sender_.SetRemoteSSRC(ssrc);
662 rtcp_receiver_.SetRemoteSSRC(ssrc);
663}
664
665// TODO(nisse): Delete video_rate amd fec_rate arguments.
666void ModuleRtpRtcpImpl2::BitrateSent(uint32_t* total_rate,
667 uint32_t* video_rate,
668 uint32_t* fec_rate,
669 uint32_t* nack_rate) const {
670 RtpSendRates send_rates = rtp_sender_->packet_sender.GetSendRates();
671 *total_rate = send_rates.Sum().bps<uint32_t>();
672 if (video_rate)
673 *video_rate = 0;
674 if (fec_rate)
675 *fec_rate = 0;
676 *nack_rate = send_rates[RtpPacketMediaType::kRetransmission].bps<uint32_t>();
677}
678
679RtpSendRates ModuleRtpRtcpImpl2::GetSendRates() const {
680 return rtp_sender_->packet_sender.GetSendRates();
681}
682
683void ModuleRtpRtcpImpl2::OnRequestSendReport() {
684 SendRTCP(kRtcpSr);
685}
686
687void ModuleRtpRtcpImpl2::OnReceivedNack(
688 const std::vector<uint16_t>& nack_sequence_numbers) {
689 if (!rtp_sender_)
690 return;
691
692 if (!StorePackets() || nack_sequence_numbers.empty()) {
693 return;
694 }
695 // Use RTT from RtcpRttStats class if provided.
696 int64_t rtt = rtt_ms();
697 if (rtt == 0) {
698 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
699 }
700 rtp_sender_->packet_generator.OnReceivedNack(nack_sequence_numbers, rtt);
701}
702
703void ModuleRtpRtcpImpl2::OnReceivedRtcpReportBlocks(
704 const ReportBlockList& report_blocks) {
705 if (rtp_sender_) {
706 uint32_t ssrc = SSRC();
707 absl::optional<uint32_t> rtx_ssrc;
708 if (rtp_sender_->packet_generator.RtxStatus() != kRtxOff) {
709 rtx_ssrc = rtp_sender_->packet_generator.RtxSsrc();
710 }
711
712 for (const RTCPReportBlock& report_block : report_blocks) {
713 if (ssrc == report_block.source_ssrc) {
714 rtp_sender_->packet_generator.OnReceivedAckOnSsrc(
715 report_block.extended_highest_sequence_number);
716 } else if (rtx_ssrc && *rtx_ssrc == report_block.source_ssrc) {
717 rtp_sender_->packet_generator.OnReceivedAckOnRtxSsrc(
718 report_block.extended_highest_sequence_number);
719 }
720 }
721 }
722}
723
724bool ModuleRtpRtcpImpl2::LastReceivedNTP(
725 uint32_t* rtcp_arrival_time_secs, // When we got the last report.
726 uint32_t* rtcp_arrival_time_frac,
727 uint32_t* remote_sr) const {
728 // Remote SR: NTP inside the last received (mid 16 bits from sec and frac).
729 uint32_t ntp_secs = 0;
730 uint32_t ntp_frac = 0;
731
732 if (!rtcp_receiver_.NTP(&ntp_secs, &ntp_frac, rtcp_arrival_time_secs,
733 rtcp_arrival_time_frac, NULL)) {
734 return false;
735 }
736 *remote_sr =
737 ((ntp_secs & 0x0000ffff) << 16) + ((ntp_frac & 0xffff0000) >> 16);
738 return true;
739}
740
741void ModuleRtpRtcpImpl2::set_rtt_ms(int64_t rtt_ms) {
742 {
743 rtc::CritScope cs(&critical_section_rtt_);
744 rtt_ms_ = rtt_ms;
745 }
746 if (rtp_sender_) {
747 rtp_sender_->packet_history.SetRtt(rtt_ms);
748 }
749}
750
751int64_t ModuleRtpRtcpImpl2::rtt_ms() const {
752 rtc::CritScope cs(&critical_section_rtt_);
753 return rtt_ms_;
754}
755
756void ModuleRtpRtcpImpl2::SetVideoBitrateAllocation(
757 const VideoBitrateAllocation& bitrate) {
758 rtcp_sender_.SetVideoBitrateAllocation(bitrate);
759}
760
761RTPSender* ModuleRtpRtcpImpl2::RtpSender() {
762 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
763}
764
765const RTPSender* ModuleRtpRtcpImpl2::RtpSender() const {
766 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
767}
768
Tommi3a5742c2020-05-20 09:32:51 +0200769} // namespace webrtc