blob: 9a1aad90c15a64a8070a5ceb4387009dcbfd7608 [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;
Tommi3a5742c2020-05-20 09:32:51 +020037const int64_t kDefaultExpectedRetransmissionTimeMs = 125;
38} // namespace
39
40ModuleRtpRtcpImpl2::RtpSenderContext::RtpSenderContext(
Tomas Gunnarssonf25761d2020-06-03 22:55:33 +020041 const RtpRtcpInterface::Configuration& config)
Tommi3a5742c2020-05-20 09:32:51 +020042 : packet_history(config.clock, config.enable_rtx_padding_prioritization),
43 packet_sender(config, &packet_history),
Erik Språng1b485322020-06-24 18:39:25 +000044 non_paced_sender(&packet_sender),
Tommi3a5742c2020-05-20 09:32:51 +020045 packet_generator(
46 config,
47 &packet_history,
48 config.paced_sender ? config.paced_sender : &non_paced_sender) {}
49
Tommi3a5742c2020-05-20 09:32:51 +020050ModuleRtpRtcpImpl2::ModuleRtpRtcpImpl2(const Configuration& configuration)
Tomas Gunnarsson473bbd82020-06-27 17:44:55 +020051 : worker_queue_(TaskQueueBase::Current()),
52 rtcp_sender_(configuration),
Tommi3a5742c2020-05-20 09:32:51 +020053 rtcp_receiver_(configuration, this),
54 clock_(configuration.clock),
Tommi3a5742c2020-05-20 09:32:51 +020055 last_rtt_process_time_(clock_->TimeInMilliseconds()),
56 next_process_time_(clock_->TimeInMilliseconds() +
57 kRtpRtcpMaxIdleTimeProcessMs),
58 packet_overhead_(28), // IPV4 UDP.
59 nack_last_time_sent_full_ms_(0),
60 nack_last_seq_number_sent_(0),
61 remote_bitrate_(configuration.remote_bitrate_estimator),
62 rtt_stats_(configuration.rtt_stats),
63 rtt_ms_(0) {
Tomas Gunnarsson473bbd82020-06-27 17:44:55 +020064 RTC_DCHECK(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +020065 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() {
Tomas Gunnarsson473bbd82020-06-27 17:44:55 +020081 RTC_DCHECK_RUN_ON(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +020082}
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
Tommi3a5742c2020-05-20 09:32:51 +0200108 // TODO(bugs.webrtc.org/11581): We update the RTT once a second, whereas other
109 // things that run in this method are updated much more frequently. Move the
110 // RTT checking over to the worker thread, which matches better with where the
111 // stats are maintained.
112 bool process_rtt = now >= last_rtt_process_time_ + kRtpRtcpRttProcessTimeMs;
113 if (rtcp_sender_.Sending()) {
114 // Process RTT if we have received a report block and we haven't
115 // processed RTT for at least |kRtpRtcpRttProcessTimeMs| milliseconds.
116 // Note that LastReceivedReportBlockMs() grabs a lock, so check
117 // |process_rtt| first.
118 if (process_rtt &&
119 rtcp_receiver_.LastReceivedReportBlockMs() > last_rtt_process_time_) {
120 std::vector<RTCPReportBlock> receive_blocks;
121 rtcp_receiver_.StatisticsReceived(&receive_blocks);
122 int64_t max_rtt = 0;
123 for (std::vector<RTCPReportBlock>::iterator it = receive_blocks.begin();
124 it != receive_blocks.end(); ++it) {
125 int64_t rtt = 0;
126 rtcp_receiver_.RTT(it->sender_ssrc, &rtt, NULL, NULL, NULL);
127 max_rtt = (rtt > max_rtt) ? rtt : max_rtt;
128 }
129 // Report the rtt.
130 if (rtt_stats_ && max_rtt != 0)
131 rtt_stats_->OnRttUpdate(max_rtt);
132 }
133
134 // Verify receiver reports are delivered and the reported sequence number
135 // is increasing.
136 // TODO(bugs.webrtc.org/11581): The timeout value needs to be checked every
137 // few seconds (see internals of RtcpRrTimeout). Here, we may be polling it
138 // a couple of hundred times a second, which isn't great since it grabs a
139 // lock. Note also that LastReceivedReportBlockMs() (called above) and
140 // RtcpRrTimeout() both grab the same lock and check the same timer, so
141 // it should be possible to consolidate that work somehow.
142 if (rtcp_receiver_.RtcpRrTimeout()) {
143 RTC_LOG_F(LS_WARNING) << "Timeout: No RTCP RR received.";
144 } else if (rtcp_receiver_.RtcpRrSequenceNumberTimeout()) {
145 RTC_LOG_F(LS_WARNING) << "Timeout: No increase in RTCP RR extended "
146 "highest sequence number.";
147 }
148
149 if (remote_bitrate_ && rtcp_sender_.TMMBR()) {
150 unsigned int target_bitrate = 0;
151 std::vector<unsigned int> ssrcs;
152 if (remote_bitrate_->LatestEstimate(&ssrcs, &target_bitrate)) {
153 if (!ssrcs.empty()) {
154 target_bitrate = target_bitrate / ssrcs.size();
155 }
156 rtcp_sender_.SetTargetBitrate(target_bitrate);
157 }
158 }
159 } else {
160 // Report rtt from receiver.
161 if (process_rtt) {
162 int64_t rtt_ms;
163 if (rtt_stats_ && rtcp_receiver_.GetAndResetXrRrRtt(&rtt_ms)) {
164 rtt_stats_->OnRttUpdate(rtt_ms);
165 }
166 }
167 }
168
169 // Get processed rtt.
170 if (process_rtt) {
171 last_rtt_process_time_ = now;
172 // TODO(bugs.webrtc.org/11581): Is this a bug? At the top of the function,
173 // next_process_time_ is incremented by 5ms, here we effectively do a
174 // std::min() of (now + 5ms, now + 1000ms). Seems like this is a no-op?
175 next_process_time_ = std::min(
176 next_process_time_, last_rtt_process_time_ + kRtpRtcpRttProcessTimeMs);
177 if (rtt_stats_) {
178 // Make sure we have a valid RTT before setting.
179 int64_t last_rtt = rtt_stats_->LastProcessedRtt();
180 if (last_rtt >= 0)
181 set_rtt_ms(last_rtt);
182 }
183 }
184
185 if (rtcp_sender_.TimeToSendRTCPReport())
186 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
187
Tomas Gunnarsson64348642020-06-09 08:02:44 +0200188 if (rtcp_sender_.TMMBR() && rtcp_receiver_.UpdateTmmbrTimers()) {
Tommi3a5742c2020-05-20 09:32:51 +0200189 rtcp_receiver_.NotifyTmmbrUpdated();
190 }
191}
192
193void ModuleRtpRtcpImpl2::SetRtxSendStatus(int mode) {
194 rtp_sender_->packet_generator.SetRtxStatus(mode);
195}
196
197int ModuleRtpRtcpImpl2::RtxSendStatus() const {
198 return rtp_sender_ ? rtp_sender_->packet_generator.RtxStatus() : kRtxOff;
199}
200
201void ModuleRtpRtcpImpl2::SetRtxSendPayloadType(int payload_type,
202 int associated_payload_type) {
203 rtp_sender_->packet_generator.SetRtxPayloadType(payload_type,
204 associated_payload_type);
205}
206
207absl::optional<uint32_t> ModuleRtpRtcpImpl2::RtxSsrc() const {
208 return rtp_sender_ ? rtp_sender_->packet_generator.RtxSsrc() : absl::nullopt;
209}
210
211absl::optional<uint32_t> ModuleRtpRtcpImpl2::FlexfecSsrc() const {
212 if (rtp_sender_) {
213 return rtp_sender_->packet_generator.FlexfecSsrc();
214 }
215 return absl::nullopt;
216}
217
218void ModuleRtpRtcpImpl2::IncomingRtcpPacket(const uint8_t* rtcp_packet,
219 const size_t length) {
220 rtcp_receiver_.IncomingPacket(rtcp_packet, length);
221}
222
223void ModuleRtpRtcpImpl2::RegisterSendPayloadFrequency(int payload_type,
224 int payload_frequency) {
225 rtcp_sender_.SetRtpClockRate(payload_type, payload_frequency);
226}
227
228int32_t ModuleRtpRtcpImpl2::DeRegisterSendPayload(const int8_t payload_type) {
229 return 0;
230}
231
232uint32_t ModuleRtpRtcpImpl2::StartTimestamp() const {
233 return rtp_sender_->packet_generator.TimestampOffset();
234}
235
236// Configure start timestamp, default is a random number.
237void ModuleRtpRtcpImpl2::SetStartTimestamp(const uint32_t timestamp) {
238 rtcp_sender_.SetTimestampOffset(timestamp);
239 rtp_sender_->packet_generator.SetTimestampOffset(timestamp);
240 rtp_sender_->packet_sender.SetTimestampOffset(timestamp);
241}
242
243uint16_t ModuleRtpRtcpImpl2::SequenceNumber() const {
244 return rtp_sender_->packet_generator.SequenceNumber();
245}
246
247// Set SequenceNumber, default is a random number.
248void ModuleRtpRtcpImpl2::SetSequenceNumber(const uint16_t seq_num) {
249 rtp_sender_->packet_generator.SetSequenceNumber(seq_num);
250}
251
252void ModuleRtpRtcpImpl2::SetRtpState(const RtpState& rtp_state) {
253 rtp_sender_->packet_generator.SetRtpState(rtp_state);
Tommi3a5742c2020-05-20 09:32:51 +0200254 rtcp_sender_.SetTimestampOffset(rtp_state.start_timestamp);
255}
256
257void ModuleRtpRtcpImpl2::SetRtxState(const RtpState& rtp_state) {
258 rtp_sender_->packet_generator.SetRtxRtpState(rtp_state);
259}
260
261RtpState ModuleRtpRtcpImpl2::GetRtpState() const {
262 RtpState state = rtp_sender_->packet_generator.GetRtpState();
Tommi3a5742c2020-05-20 09:32:51 +0200263 return state;
264}
265
266RtpState ModuleRtpRtcpImpl2::GetRtxState() const {
267 return rtp_sender_->packet_generator.GetRtxRtpState();
268}
269
270void ModuleRtpRtcpImpl2::SetRid(const std::string& rid) {
271 if (rtp_sender_) {
272 rtp_sender_->packet_generator.SetRid(rid);
273 }
274}
275
276void ModuleRtpRtcpImpl2::SetMid(const std::string& mid) {
277 if (rtp_sender_) {
278 rtp_sender_->packet_generator.SetMid(mid);
279 }
280 // TODO(bugs.webrtc.org/4050): If we end up supporting the MID SDES item for
281 // RTCP, this will need to be passed down to the RTCPSender also.
282}
283
284void ModuleRtpRtcpImpl2::SetCsrcs(const std::vector<uint32_t>& csrcs) {
285 rtcp_sender_.SetCsrcs(csrcs);
286 rtp_sender_->packet_generator.SetCsrcs(csrcs);
287}
288
289// TODO(pbos): Handle media and RTX streams separately (separate RTCP
290// feedbacks).
291RTCPSender::FeedbackState ModuleRtpRtcpImpl2::GetFeedbackState() {
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200292 // TODO(bugs.webrtc.org/11581): Called by potentially multiple threads.
293 // "Send*" methods and on the ProcessThread. Make sure it's only called on the
294 // construction thread.
295
Tommi3a5742c2020-05-20 09:32:51 +0200296 RTCPSender::FeedbackState state;
297 // This is called also when receiver_only is true. Hence below
298 // checks that rtp_sender_ exists.
299 if (rtp_sender_) {
300 StreamDataCounters rtp_stats;
301 StreamDataCounters rtx_stats;
302 rtp_sender_->packet_sender.GetDataCounters(&rtp_stats, &rtx_stats);
303 state.packets_sent =
304 rtp_stats.transmitted.packets + rtx_stats.transmitted.packets;
305 state.media_bytes_sent = rtp_stats.transmitted.payload_bytes +
306 rtx_stats.transmitted.payload_bytes;
307 state.send_bitrate =
308 rtp_sender_->packet_sender.GetSendRates().Sum().bps<uint32_t>();
309 }
310 state.receiver = &rtcp_receiver_;
311
312 LastReceivedNTP(&state.last_rr_ntp_secs, &state.last_rr_ntp_frac,
313 &state.remote_sr);
314
315 state.last_xr_rtis = rtcp_receiver_.ConsumeReceivedXrReferenceTimeInfo();
316
317 return state;
318}
319
320// TODO(nisse): This method shouldn't be called for a receive-only
321// stream. Delete rtp_sender_ check as soon as all applications are
322// updated.
323int32_t ModuleRtpRtcpImpl2::SetSendingStatus(const bool sending) {
324 if (rtcp_sender_.Sending() != sending) {
325 // Sends RTCP BYE when going from true to false
326 if (rtcp_sender_.SetSendingStatus(GetFeedbackState(), sending) != 0) {
327 RTC_LOG(LS_WARNING) << "Failed to send RTCP BYE";
328 }
329 }
330 return 0;
331}
332
333bool ModuleRtpRtcpImpl2::Sending() const {
334 return rtcp_sender_.Sending();
335}
336
337// TODO(nisse): This method shouldn't be called for a receive-only
338// stream. Delete rtp_sender_ check as soon as all applications are
339// updated.
340void ModuleRtpRtcpImpl2::SetSendingMediaStatus(const bool sending) {
341 if (rtp_sender_) {
342 rtp_sender_->packet_generator.SetSendingMediaStatus(sending);
343 } else {
344 RTC_DCHECK(!sending);
345 }
346}
347
348bool ModuleRtpRtcpImpl2::SendingMedia() const {
349 return rtp_sender_ ? rtp_sender_->packet_generator.SendingMedia() : false;
350}
351
352bool ModuleRtpRtcpImpl2::IsAudioConfigured() const {
353 return rtp_sender_ ? rtp_sender_->packet_generator.IsAudioConfigured()
354 : false;
355}
356
357void ModuleRtpRtcpImpl2::SetAsPartOfAllocation(bool part_of_allocation) {
358 RTC_CHECK(rtp_sender_);
359 rtp_sender_->packet_sender.ForceIncludeSendPacketsInAllocation(
360 part_of_allocation);
361}
362
363bool ModuleRtpRtcpImpl2::OnSendingRtpFrame(uint32_t timestamp,
364 int64_t capture_time_ms,
365 int payload_type,
366 bool force_sender_report) {
367 if (!Sending())
368 return false;
369
370 rtcp_sender_.SetLastRtpTime(timestamp, capture_time_ms, payload_type);
371 // Make sure an RTCP report isn't queued behind a key frame.
372 if (rtcp_sender_.TimeToSendRTCPReport(force_sender_report))
373 rtcp_sender_.SendRTCP(GetFeedbackState(), kRtcpReport);
374
375 return true;
376}
377
378bool ModuleRtpRtcpImpl2::TrySendPacket(RtpPacketToSend* packet,
379 const PacedPacketInfo& pacing_info) {
380 RTC_DCHECK(rtp_sender_);
381 // TODO(sprang): Consider if we can remove this check.
382 if (!rtp_sender_->packet_generator.SendingMedia()) {
383 return false;
384 }
385 rtp_sender_->packet_sender.SendPacket(packet, pacing_info);
386 return true;
387}
388
389void ModuleRtpRtcpImpl2::OnPacketsAcknowledged(
390 rtc::ArrayView<const uint16_t> sequence_numbers) {
391 RTC_DCHECK(rtp_sender_);
392 rtp_sender_->packet_history.CullAcknowledgedPackets(sequence_numbers);
393}
394
395bool ModuleRtpRtcpImpl2::SupportsPadding() const {
396 RTC_DCHECK(rtp_sender_);
397 return rtp_sender_->packet_generator.SupportsPadding();
398}
399
400bool ModuleRtpRtcpImpl2::SupportsRtxPayloadPadding() const {
401 RTC_DCHECK(rtp_sender_);
402 return rtp_sender_->packet_generator.SupportsRtxPayloadPadding();
403}
404
405std::vector<std::unique_ptr<RtpPacketToSend>>
406ModuleRtpRtcpImpl2::GeneratePadding(size_t target_size_bytes) {
407 RTC_DCHECK(rtp_sender_);
408 return rtp_sender_->packet_generator.GeneratePadding(
409 target_size_bytes, rtp_sender_->packet_sender.MediaHasBeenSent());
410}
411
412std::vector<RtpSequenceNumberMap::Info>
413ModuleRtpRtcpImpl2::GetSentRtpPacketInfos(
414 rtc::ArrayView<const uint16_t> sequence_numbers) const {
415 RTC_DCHECK(rtp_sender_);
416 return rtp_sender_->packet_sender.GetSentRtpPacketInfos(sequence_numbers);
417}
418
419size_t ModuleRtpRtcpImpl2::ExpectedPerPacketOverhead() const {
420 if (!rtp_sender_) {
421 return 0;
422 }
423 return rtp_sender_->packet_generator.ExpectedPerPacketOverhead();
424}
425
426size_t ModuleRtpRtcpImpl2::MaxRtpPacketSize() const {
427 RTC_DCHECK(rtp_sender_);
428 return rtp_sender_->packet_generator.MaxRtpPacketSize();
429}
430
431void ModuleRtpRtcpImpl2::SetMaxRtpPacketSize(size_t rtp_packet_size) {
432 RTC_DCHECK_LE(rtp_packet_size, IP_PACKET_SIZE)
433 << "rtp packet size too large: " << rtp_packet_size;
434 RTC_DCHECK_GT(rtp_packet_size, packet_overhead_)
435 << "rtp packet size too small: " << rtp_packet_size;
436
437 rtcp_sender_.SetMaxRtpPacketSize(rtp_packet_size);
438 if (rtp_sender_) {
439 rtp_sender_->packet_generator.SetMaxRtpPacketSize(rtp_packet_size);
440 }
441}
442
443RtcpMode ModuleRtpRtcpImpl2::RTCP() const {
444 return rtcp_sender_.Status();
445}
446
447// Configure RTCP status i.e on/off.
448void ModuleRtpRtcpImpl2::SetRTCPStatus(const RtcpMode method) {
449 rtcp_sender_.SetRTCPStatus(method);
450}
451
452int32_t ModuleRtpRtcpImpl2::SetCNAME(const char* c_name) {
453 return rtcp_sender_.SetCNAME(c_name);
454}
455
Tommi3a5742c2020-05-20 09:32:51 +0200456int32_t ModuleRtpRtcpImpl2::RemoteNTP(uint32_t* received_ntpsecs,
457 uint32_t* received_ntpfrac,
458 uint32_t* rtcp_arrival_time_secs,
459 uint32_t* rtcp_arrival_time_frac,
460 uint32_t* rtcp_timestamp) const {
461 return rtcp_receiver_.NTP(received_ntpsecs, received_ntpfrac,
462 rtcp_arrival_time_secs, rtcp_arrival_time_frac,
463 rtcp_timestamp)
464 ? 0
465 : -1;
466}
467
468// Get RoundTripTime.
469int32_t ModuleRtpRtcpImpl2::RTT(const uint32_t remote_ssrc,
470 int64_t* rtt,
471 int64_t* avg_rtt,
472 int64_t* min_rtt,
473 int64_t* max_rtt) const {
474 int32_t ret = rtcp_receiver_.RTT(remote_ssrc, rtt, avg_rtt, min_rtt, max_rtt);
475 if (rtt && *rtt == 0) {
476 // Try to get RTT from RtcpRttStats class.
477 *rtt = rtt_ms();
478 }
479 return ret;
480}
481
482int64_t ModuleRtpRtcpImpl2::ExpectedRetransmissionTimeMs() const {
483 int64_t expected_retransmission_time_ms = rtt_ms();
484 if (expected_retransmission_time_ms > 0) {
485 return expected_retransmission_time_ms;
486 }
487 // No rtt available (|kRtpRtcpRttProcessTimeMs| not yet passed?), so try to
488 // poll avg_rtt_ms directly from rtcp receiver.
489 if (rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), nullptr,
490 &expected_retransmission_time_ms, nullptr,
491 nullptr) == 0) {
492 return expected_retransmission_time_ms;
493 }
494 return kDefaultExpectedRetransmissionTimeMs;
495}
496
497// Force a send of an RTCP packet.
498// Normal SR and RR are triggered via the process function.
499int32_t ModuleRtpRtcpImpl2::SendRTCP(RTCPPacketType packet_type) {
500 return rtcp_sender_.SendRTCP(GetFeedbackState(), packet_type);
501}
502
Tommi3a5742c2020-05-20 09:32:51 +0200503void ModuleRtpRtcpImpl2::SetRtcpXrRrtrStatus(bool enable) {
504 rtcp_receiver_.SetRtcpXrRrtrStatus(enable);
505 rtcp_sender_.SendRtcpXrReceiverReferenceTime(enable);
506}
507
508bool ModuleRtpRtcpImpl2::RtcpXrRrtrStatus() const {
509 return rtcp_sender_.RtcpXrReceiverReferenceTime();
510}
511
Tommi3a5742c2020-05-20 09:32:51 +0200512void ModuleRtpRtcpImpl2::GetSendStreamDataCounters(
513 StreamDataCounters* rtp_counters,
514 StreamDataCounters* rtx_counters) const {
515 rtp_sender_->packet_sender.GetDataCounters(rtp_counters, rtx_counters);
516}
517
518// Received RTCP report.
519int32_t ModuleRtpRtcpImpl2::RemoteRTCPStat(
520 std::vector<RTCPReportBlock>* receive_blocks) const {
521 return rtcp_receiver_.StatisticsReceived(receive_blocks);
522}
523
524std::vector<ReportBlockData> ModuleRtpRtcpImpl2::GetLatestReportBlockData()
525 const {
526 return rtcp_receiver_.GetLatestReportBlockData();
527}
528
529// (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
655// TODO(nisse): Delete video_rate amd fec_rate arguments.
656void ModuleRtpRtcpImpl2::BitrateSent(uint32_t* total_rate,
657 uint32_t* video_rate,
658 uint32_t* fec_rate,
659 uint32_t* nack_rate) const {
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200660 RTC_DCHECK_RUN_ON(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +0200661 RtpSendRates send_rates = rtp_sender_->packet_sender.GetSendRates();
662 *total_rate = send_rates.Sum().bps<uint32_t>();
663 if (video_rate)
664 *video_rate = 0;
665 if (fec_rate)
666 *fec_rate = 0;
667 *nack_rate = send_rates[RtpPacketMediaType::kRetransmission].bps<uint32_t>();
668}
669
670RtpSendRates ModuleRtpRtcpImpl2::GetSendRates() const {
Tomas Gunnarssona1163742020-06-29 17:41:22 +0200671 RTC_DCHECK_RUN_ON(worker_queue_);
Tommi3a5742c2020-05-20 09:32:51 +0200672 return rtp_sender_->packet_sender.GetSendRates();
673}
674
675void ModuleRtpRtcpImpl2::OnRequestSendReport() {
676 SendRTCP(kRtcpSr);
677}
678
679void ModuleRtpRtcpImpl2::OnReceivedNack(
680 const std::vector<uint16_t>& nack_sequence_numbers) {
681 if (!rtp_sender_)
682 return;
683
684 if (!StorePackets() || nack_sequence_numbers.empty()) {
685 return;
686 }
687 // Use RTT from RtcpRttStats class if provided.
688 int64_t rtt = rtt_ms();
689 if (rtt == 0) {
690 rtcp_receiver_.RTT(rtcp_receiver_.RemoteSSRC(), NULL, &rtt, NULL, NULL);
691 }
692 rtp_sender_->packet_generator.OnReceivedNack(nack_sequence_numbers, rtt);
693}
694
695void ModuleRtpRtcpImpl2::OnReceivedRtcpReportBlocks(
696 const ReportBlockList& report_blocks) {
697 if (rtp_sender_) {
698 uint32_t ssrc = SSRC();
699 absl::optional<uint32_t> rtx_ssrc;
700 if (rtp_sender_->packet_generator.RtxStatus() != kRtxOff) {
701 rtx_ssrc = rtp_sender_->packet_generator.RtxSsrc();
702 }
703
704 for (const RTCPReportBlock& report_block : report_blocks) {
705 if (ssrc == report_block.source_ssrc) {
706 rtp_sender_->packet_generator.OnReceivedAckOnSsrc(
707 report_block.extended_highest_sequence_number);
708 } else if (rtx_ssrc && *rtx_ssrc == report_block.source_ssrc) {
709 rtp_sender_->packet_generator.OnReceivedAckOnRtxSsrc(
710 report_block.extended_highest_sequence_number);
711 }
712 }
713 }
714}
715
716bool ModuleRtpRtcpImpl2::LastReceivedNTP(
717 uint32_t* rtcp_arrival_time_secs, // When we got the last report.
718 uint32_t* rtcp_arrival_time_frac,
719 uint32_t* remote_sr) const {
720 // Remote SR: NTP inside the last received (mid 16 bits from sec and frac).
721 uint32_t ntp_secs = 0;
722 uint32_t ntp_frac = 0;
723
724 if (!rtcp_receiver_.NTP(&ntp_secs, &ntp_frac, rtcp_arrival_time_secs,
725 rtcp_arrival_time_frac, NULL)) {
726 return false;
727 }
728 *remote_sr =
729 ((ntp_secs & 0x0000ffff) << 16) + ((ntp_frac & 0xffff0000) >> 16);
730 return true;
731}
732
733void ModuleRtpRtcpImpl2::set_rtt_ms(int64_t rtt_ms) {
734 {
735 rtc::CritScope cs(&critical_section_rtt_);
736 rtt_ms_ = rtt_ms;
737 }
738 if (rtp_sender_) {
739 rtp_sender_->packet_history.SetRtt(rtt_ms);
740 }
741}
742
743int64_t ModuleRtpRtcpImpl2::rtt_ms() const {
744 rtc::CritScope cs(&critical_section_rtt_);
745 return rtt_ms_;
746}
747
748void ModuleRtpRtcpImpl2::SetVideoBitrateAllocation(
749 const VideoBitrateAllocation& bitrate) {
750 rtcp_sender_.SetVideoBitrateAllocation(bitrate);
751}
752
753RTPSender* ModuleRtpRtcpImpl2::RtpSender() {
754 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
755}
756
757const RTPSender* ModuleRtpRtcpImpl2::RtpSender() const {
758 return rtp_sender_ ? &rtp_sender_->packet_generator : nullptr;
759}
760
Tommi3a5742c2020-05-20 09:32:51 +0200761} // namespace webrtc