blob: 0d5c941765112a308e6ddac48d5eab484f4074ec [file] [log] [blame]
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +00001/*
phoglund@webrtc.org78088c22012-02-07 14:56:45 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +00003 *
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
Jonas Olssona4d87372019-07-05 19:08:33 +020011#include "modules/rtp_rtcp/source/rtp_sender.h"
12
kwiberg84be5112016-04-27 01:19:58 -070013#include <memory>
danilchapb8b6fbb2015-12-10 05:05:27 -080014#include <vector>
15
Danil Chapovalov16cb1f62019-09-05 11:29:59 +020016#include "api/rtc_event_log/rtc_event.h"
Per Kjellandere11b7d22019-02-21 07:55:59 +010017#include "api/transport/field_trial_based_config.h"
Erik Språngf93eda12019-01-16 17:10:57 +010018#include "api/video/video_codec_constants.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020019#include "api/video/video_timing.h"
20#include "logging/rtc_event_log/mock/mock_rtc_event_log.h"
21#include "modules/rtp_rtcp/include/rtp_cvo.h"
22#include "modules/rtp_rtcp/include/rtp_header_extension_map.h"
Erik Språngaa59eca2019-07-24 14:52:55 +020023#include "modules/rtp_rtcp/include/rtp_packet_sender.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020024#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
25#include "modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h"
26#include "modules/rtp_rtcp/source/rtp_format_video_generic.h"
philipelb3e42a42018-09-13 10:57:14 +020027#include "modules/rtp_rtcp/source/rtp_generic_frame_descriptor.h"
28#include "modules/rtp_rtcp/source/rtp_generic_frame_descriptor_extension.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020029#include "modules/rtp_rtcp/source/rtp_header_extensions.h"
30#include "modules/rtp_rtcp/source/rtp_packet_received.h"
31#include "modules/rtp_rtcp/source/rtp_packet_to_send.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020032#include "modules/rtp_rtcp/source/rtp_sender_video.h"
33#include "modules/rtp_rtcp/source/rtp_utility.h"
34#include "rtc_base/arraysize.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020035#include "rtc_base/rate_limiter.h"
36#include "test/field_trial.h"
37#include "test/gmock.h"
38#include "test/gtest.h"
39#include "test/mock_transport.h"
Tommi25eb47c2019-08-29 16:39:05 +020040#include "test/rtp_header_parser.h"
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +000041
42namespace webrtc {
43
andrew@webrtc.org8a442592011-12-16 21:24:30 +000044namespace {
Elad Alond8d32482019-02-18 23:45:57 +010045enum : int { // The first valid value is 1.
46 kAbsoluteSendTimeExtensionId = 1,
47 kAudioLevelExtensionId,
Elad Alonccb9b752019-02-19 13:01:31 +010048 kGenericDescriptorId00,
49 kGenericDescriptorId01,
Elad Alond8d32482019-02-18 23:45:57 +010050 kMidExtensionId,
51 kRepairedRidExtensionId,
52 kRidExtensionId,
53 kTransmissionTimeOffsetExtensionId,
54 kTransportSequenceNumberExtensionId,
55 kVideoRotationExtensionId,
56 kVideoTimingExtensionId,
57};
58
andrew@webrtc.org8a442592011-12-16 21:24:30 +000059const int kPayload = 100;
Shao Changbine62202f2015-04-21 20:24:50 +080060const int kRtxPayload = 98;
andrew@webrtc.org8a442592011-12-16 21:24:30 +000061const uint32_t kTimestamp = 10;
62const uint16_t kSeqNum = 33;
brandtr9dfff292016-11-14 05:14:50 -080063const uint32_t kSsrc = 725242;
Erik Språng9c771c22019-06-17 16:31:53 +020064const uint32_t kRtxSsrc = 12345;
65const uint32_t kFlexFecSsrc = 45678;
Erik Språng1fbfecd2019-08-26 19:00:05 +020066const uint16_t kTransportSequenceNumber = 1;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +000067const uint64_t kStartTime = 123456789;
pbos@webrtc.org2f4b14e2014-07-15 15:25:39 +000068const size_t kMaxPaddingSize = 224u;
Stefan Holmera246cfb2016-08-23 17:51:42 +020069const uint8_t kPayloadData[] = {47, 11, 32, 93, 89};
spranga8ae6f22017-09-04 07:23:56 -070070const int64_t kDefaultExpectedRetransmissionTimeMs = 125;
Amit Hilbuch77938e62018-12-21 09:23:38 -080071const char kNoRid[] = "";
72const char kNoMid[] = "";
andrew@webrtc.org8a442592011-12-16 21:24:30 +000073
Danil Chapovalov5e57b172016-09-02 19:15:59 +020074using ::testing::_;
Erik Språng30a276b2019-04-23 12:00:11 +020075using ::testing::AllOf;
Danil Chapovalov5e57b172016-09-02 19:15:59 +020076using ::testing::ElementsAreArray;
Erik Språng30a276b2019-04-23 12:00:11 +020077using ::testing::Field;
Erik Språng9c771c22019-06-17 16:31:53 +020078using ::testing::NiceMock;
Erik Språng478cb462019-06-26 15:49:27 +020079using ::testing::Pointee;
80using ::testing::Property;
Erik Språng214f5432019-06-20 15:09:58 +020081using ::testing::Return;
Erik Språng9c771c22019-06-17 16:31:53 +020082using ::testing::StrictMock;
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +000083
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +000084uint64_t ConvertMsToAbsSendTime(int64_t time_ms) {
Stefan Holmer0a87ffc2015-10-21 13:41:48 +020085 return (((time_ms << 18) + 500) / 1000) & 0x00ffffff;
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +000086}
87
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +000088class LoopbackTransportTest : public webrtc::Transport {
89 public:
Petter Strandmark26bc6692018-05-29 08:43:35 +020090 LoopbackTransportTest() : total_bytes_sent_(0) {
Danil Chapovalova74e4772019-09-12 17:53:04 +020091 receivers_extensions_.Register<TransmissionOffset>(
92 kTransmissionTimeOffsetExtensionId);
93 receivers_extensions_.Register<AbsoluteSendTime>(
94 kAbsoluteSendTimeExtensionId);
95 receivers_extensions_.Register<TransportSequenceNumber>(
96 kTransportSequenceNumberExtensionId);
97 receivers_extensions_.Register<VideoOrientation>(kVideoRotationExtensionId);
98 receivers_extensions_.Register<AudioLevel>(kAudioLevelExtensionId);
99 receivers_extensions_.Register<VideoTimingExtension>(
100 kVideoTimingExtensionId);
101 receivers_extensions_.Register<RtpMid>(kMidExtensionId);
102 receivers_extensions_.Register<RtpGenericFrameDescriptorExtension00>(
103 kGenericDescriptorId00);
104 receivers_extensions_.Register<RtpGenericFrameDescriptorExtension01>(
105 kGenericDescriptorId01);
106 receivers_extensions_.Register<RtpStreamId>(kRidExtensionId);
107 receivers_extensions_.Register<RepairedRtpStreamId>(
108 kRepairedRidExtensionId);
guoweis@webrtc.org45362892015-03-04 22:55:15 +0000109 }
danilchap12ba1862016-10-26 02:41:55 -0700110
stefan1d8a5062015-10-02 03:39:33 -0700111 bool SendRtp(const uint8_t* data,
112 size_t len,
113 const PacketOptions& options) override {
Petter Strandmark26bc6692018-05-29 08:43:35 +0200114 last_options_ = options;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000115 total_bytes_sent_ += len;
danilchap12ba1862016-10-26 02:41:55 -0700116 sent_packets_.push_back(RtpPacketReceived(&receivers_extensions_));
117 EXPECT_TRUE(sent_packets_.back().Parse(data, len));
pbos2d566682015-09-28 09:59:31 -0700118 return true;
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000119 }
danilchap162abd32015-12-10 02:39:40 -0800120 bool SendRtcp(const uint8_t* data, size_t len) override { return false; }
danilchap12ba1862016-10-26 02:41:55 -0700121 const RtpPacketReceived& last_sent_packet() { return sent_packets_.back(); }
122 int packets_sent() { return sent_packets_.size(); }
123
pbos@webrtc.org72491b92014-07-10 16:24:54 +0000124 size_t total_bytes_sent_;
Petter Strandmark26bc6692018-05-29 08:43:35 +0200125 PacketOptions last_options_;
danilchap12ba1862016-10-26 02:41:55 -0700126 std::vector<RtpPacketReceived> sent_packets_;
127
128 private:
129 RtpHeaderExtensionMap receivers_extensions_;
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000130};
131
Elad Alon4a87e1c2017-10-03 16:11:34 +0200132MATCHER_P(SameRtcEventTypeAs, value, "") {
133 return value == arg->GetType();
134}
135
Erik Språngf6468d22019-07-05 16:53:43 +0200136struct TestConfig {
Erik Språngf5815fa2019-08-21 14:27:31 +0200137 explicit TestConfig(bool with_overhead) : with_overhead(with_overhead) {}
Erik Språngf6468d22019-07-05 16:53:43 +0200138 bool with_overhead = false;
Erik Språngf6468d22019-07-05 16:53:43 +0200139};
140
141std::string ToFieldTrialString(TestConfig config) {
142 std::string field_trials;
143 if (config.with_overhead) {
144 field_trials += "WebRTC-SendSideBwe-WithOverhead/Enabled/";
145 }
Erik Språngf6468d22019-07-05 16:53:43 +0200146 return field_trials;
147}
148
guoweis@webrtc.org45362892015-03-04 22:55:15 +0000149} // namespace
150
Erik Språngaa59eca2019-07-24 14:52:55 +0200151class MockRtpPacketPacer : public RtpPacketSender {
sprangebbf8a82015-09-21 15:11:14 -0700152 public:
Erik Språng59b86542019-06-23 18:24:46 +0200153 MockRtpPacketPacer() {}
154 virtual ~MockRtpPacketPacer() {}
155
156 MOCK_METHOD1(EnqueuePacket, void(std::unique_ptr<RtpPacketToSend>));
sprangebbf8a82015-09-21 15:11:14 -0700157
Erik Språngaa59eca2019-07-24 14:52:55 +0200158 MOCK_METHOD2(CreateProbeCluster, void(int bitrate_bps, int cluster_id));
159
160 MOCK_METHOD0(Pause, void());
161 MOCK_METHOD0(Resume, void());
162 MOCK_METHOD1(SetCongestionWindow,
163 void(absl::optional<int64_t> congestion_window_bytes));
164 MOCK_METHOD1(UpdateOutstandingData, void(int64_t outstanding_bytes));
165 MOCK_METHOD1(SetAccountForAudioPackets, void(bool account_for_audio));
sprangebbf8a82015-09-21 15:11:14 -0700166};
167
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200168class MockSendSideDelayObserver : public SendSideDelayObserver {
169 public:
Henrik Boström9fe18342019-05-16 18:38:20 +0200170 MOCK_METHOD4(SendSideDelayUpdated, void(int, int, uint64_t, uint32_t));
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200171};
172
asapersson35151f32016-05-02 23:44:01 -0700173class MockSendPacketObserver : public SendPacketObserver {
174 public:
175 MOCK_METHOD3(OnSendPacket, void(uint16_t, int64_t, uint32_t));
176};
177
Stefan Holmera246cfb2016-08-23 17:51:42 +0200178class MockTransportFeedbackObserver : public TransportFeedbackObserver {
179 public:
Erik Språng30a276b2019-04-23 12:00:11 +0200180 MOCK_METHOD1(OnAddPacket, void(const RtpPacketSendInfo&));
Stefan Holmera246cfb2016-08-23 17:51:42 +0200181 MOCK_METHOD1(OnTransportFeedback, void(const rtcp::TransportFeedback&));
elad.alonf9490002017-03-06 05:32:21 -0800182 MOCK_CONST_METHOD0(GetTransportFeedbackVector, std::vector<PacketFeedback>());
Stefan Holmera246cfb2016-08-23 17:51:42 +0200183};
184
minyue3a407ee2017-04-03 01:10:33 -0700185class MockOverheadObserver : public OverheadObserver {
186 public:
187 MOCK_METHOD1(OnOverheadChanged, void(size_t overhead_bytes_per_packet));
188};
189
Erik Språngf6468d22019-07-05 16:53:43 +0200190class RtpSenderTest : public ::testing::TestWithParam<TestConfig> {
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +0000191 protected:
192 RtpSenderTest()
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +0000193 : fake_clock_(kStartTime),
terelius429c3452016-01-21 05:42:04 -0800194 mock_rtc_event_log_(),
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +0000195 mock_paced_sender_(),
sprangcd349d92016-07-13 09:11:28 -0700196 retransmission_rate_limiter_(&fake_clock_, 1000),
Erik Språng4580ca22019-07-04 10:38:43 +0200197 flexfec_sender_(0,
198 kFlexFecSsrc,
199 kSsrc,
200 "",
201 std::vector<RtpExtension>(),
202 std::vector<RtpExtensionSize>(),
203 nullptr,
204 &fake_clock_),
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +0000205 rtp_sender_(),
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +0000206 transport_(),
minyue3a407ee2017-04-03 01:10:33 -0700207 kMarkerBit(true),
Erik Språngf6468d22019-07-05 16:53:43 +0200208 field_trials_(ToFieldTrialString(GetParam())) {}
solenberg@webrtc.orgc0352d52013-05-20 20:55:07 +0000209
Erik Språng7b52f102018-02-07 14:37:37 +0100210 void SetUp() override { SetUpRtpSender(true, false); }
Peter Boströme23e7372015-10-08 11:44:14 +0200211
Erik Språng7b52f102018-02-07 14:37:37 +0100212 void SetUpRtpSender(bool pacer, bool populate_network2) {
Erik Språng4580ca22019-07-04 10:38:43 +0200213 RtpRtcp::Configuration config;
214 config.clock = &fake_clock_;
215 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200216 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200217 config.rtx_send_ssrc = kRtxSsrc;
218 config.flexfec_sender = &flexfec_sender_;
Erik Språng4580ca22019-07-04 10:38:43 +0200219 config.event_log = &mock_rtc_event_log_;
220 config.send_packet_observer = &send_packet_observer_;
221 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
222 config.paced_sender = pacer ? &mock_paced_sender_ : nullptr;
223 config.populate_network2_timestamp = populate_network2;
224 rtp_sender_.reset(new RTPSender(config));
brandtr9dfff292016-11-14 05:14:50 -0800225 rtp_sender_->SetSequenceNumber(kSeqNum);
danilchap71fead22016-08-18 02:01:49 -0700226 rtp_sender_->SetTimestampOffset(0);
solenberg@webrtc.orgc0352d52013-05-20 20:55:07 +0000227 }
228
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000229 SimulatedClock fake_clock_;
Erik Språng9c771c22019-06-17 16:31:53 +0200230 NiceMock<MockRtcEventLog> mock_rtc_event_log_;
Erik Språng59b86542019-06-23 18:24:46 +0200231 MockRtpPacketPacer mock_paced_sender_;
Erik Språng9c771c22019-06-17 16:31:53 +0200232 StrictMock<MockSendPacketObserver> send_packet_observer_;
233 StrictMock<MockTransportFeedbackObserver> feedback_observer_;
sprangcd349d92016-07-13 09:11:28 -0700234 RateLimiter retransmission_rate_limiter_;
Erik Språng4580ca22019-07-04 10:38:43 +0200235 FlexfecSender flexfec_sender_;
kwiberg84be5112016-04-27 01:19:58 -0700236 std::unique_ptr<RTPSender> rtp_sender_;
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000237 LoopbackTransportTest transport_;
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +0000238 const bool kMarkerBit;
minyue3a407ee2017-04-03 01:10:33 -0700239 test::ScopedFieldTrials field_trials_;
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +0000240
danilchapb6f1fb52016-10-19 06:11:39 -0700241 std::unique_ptr<RtpPacketToSend> BuildRtpPacket(int payload_type,
242 bool marker_bit,
243 uint32_t timestamp,
244 int64_t capture_time_ms) {
245 auto packet = rtp_sender_->AllocatePacket();
246 packet->SetPayloadType(payload_type);
Erik Språng60ffc312019-07-30 22:03:49 +0200247 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
danilchapb6f1fb52016-10-19 06:11:39 -0700248 packet->SetMarker(marker_bit);
249 packet->SetTimestamp(timestamp);
250 packet->set_capture_time_ms(capture_time_ms);
251 EXPECT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
252 return packet;
253 }
254
Erik Språngf6468d22019-07-05 16:53:43 +0200255 std::unique_ptr<RtpPacketToSend> SendPacket(int64_t capture_time_ms,
256 int payload_length) {
stefan@webrtc.org7e9315b2013-12-04 10:24:26 +0000257 uint32_t timestamp = capture_time_ms * 90;
danilchapb6f1fb52016-10-19 06:11:39 -0700258 auto packet =
259 BuildRtpPacket(kPayload, kMarkerBit, timestamp, capture_time_ms);
260 packet->AllocatePayload(payload_length);
Erik Språng70768f42019-08-27 18:16:26 +0200261 packet->set_allow_retransmission(true);
stefan@webrtc.org7e9315b2013-12-04 10:24:26 +0000262
263 // Packet should be stored in a send bucket.
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200264 EXPECT_TRUE(
265 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +0200266 return packet;
stefan@webrtc.org7e9315b2013-12-04 10:24:26 +0000267 }
asapersson35151f32016-05-02 23:44:01 -0700268
Erik Språngf6468d22019-07-05 16:53:43 +0200269 std::unique_ptr<RtpPacketToSend> SendGenericPacket() {
asapersson35151f32016-05-02 23:44:01 -0700270 const int64_t kCaptureTimeMs = fake_clock_.TimeInMilliseconds();
Erik Språngf6468d22019-07-05 16:53:43 +0200271 return SendPacket(kCaptureTimeMs, sizeof(kPayloadData));
asapersson35151f32016-05-02 23:44:01 -0700272 }
Steve Anton2bac7da2019-07-21 15:04:21 -0400273
Erik Språng4208a132019-08-26 08:58:45 +0200274 size_t GenerateAndSendPadding(size_t target_size_bytes) {
275 size_t generated_bytes = 0;
276 for (auto& packet : rtp_sender_->GeneratePadding(target_size_bytes)) {
277 generated_bytes += packet->payload_size() + packet->padding_size();
278 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
279 }
280 return generated_bytes;
281 }
282
Steve Anton2bac7da2019-07-21 15:04:21 -0400283 // The following are helpers for configuring the RTPSender. They must be
284 // called before sending any packets.
285
286 // Enable the retransmission stream with sizable packet storage.
287 void EnableRtx() {
288 // RTX needs to be able to read the source packets from the packet store.
289 // Pick a number of packets to store big enough for any unit test.
290 constexpr uint16_t kNumberOfPacketsToStore = 100;
291 rtp_sender_->SetStorePacketsStatus(true, kNumberOfPacketsToStore);
292 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
293 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
294 }
295
296 // Enable sending of the MID header extension for both the primary SSRC and
297 // the RTX SSRC.
298 void EnableMidSending(const std::string& mid) {
299 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionMid, kMidExtensionId);
300 rtp_sender_->SetMid(mid);
301 }
302
303 // Enable sending of the RSID header extension for the primary SSRC and the
304 // RRSID header extension for the RTX SSRC.
305 void EnableRidSending(const std::string& rid) {
306 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionRtpStreamId,
307 kRidExtensionId);
308 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionRepairedRtpStreamId,
309 kRepairedRidExtensionId);
310 rtp_sender_->SetRid(rid);
311 }
asapersson@webrtc.org5249cc82011-12-16 14:31:37 +0000312};
313
Peter Boströme23e7372015-10-08 11:44:14 +0200314// TODO(pbos): Move tests over from WithoutPacer to RtpSenderTest as this is our
315// default code path.
316class RtpSenderTestWithoutPacer : public RtpSenderTest {
317 public:
Erik Språng7b52f102018-02-07 14:37:37 +0100318 void SetUp() override { SetUpRtpSender(false, false); }
Peter Boströme23e7372015-10-08 11:44:14 +0200319};
320
minyue3a407ee2017-04-03 01:10:33 -0700321TEST_P(RtpSenderTestWithoutPacer, AllocatePacketSetCsrc) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200322 // Configure rtp_sender with csrc.
323 std::vector<uint32_t> csrcs;
324 csrcs.push_back(0x23456789);
325 rtp_sender_->SetCsrcs(csrcs);
326
327 auto packet = rtp_sender_->AllocatePacket();
328
329 ASSERT_TRUE(packet);
330 EXPECT_EQ(rtp_sender_->SSRC(), packet->Ssrc());
331 EXPECT_EQ(csrcs, packet->Csrcs());
332}
333
minyue3a407ee2017-04-03 01:10:33 -0700334TEST_P(RtpSenderTestWithoutPacer, AllocatePacketReserveExtensions) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200335 // Configure rtp_sender with extensions.
336 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
337 kRtpExtensionTransmissionTimeOffset,
338 kTransmissionTimeOffsetExtensionId));
339 ASSERT_EQ(
340 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
341 kAbsoluteSendTimeExtensionId));
342 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAudioLevel,
343 kAudioLevelExtensionId));
344 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
345 kRtpExtensionTransportSequenceNumber,
346 kTransportSequenceNumberExtensionId));
347 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
348 kRtpExtensionVideoRotation, kVideoRotationExtensionId));
349
350 auto packet = rtp_sender_->AllocatePacket();
351
352 ASSERT_TRUE(packet);
353 // Preallocate BWE extensions RtpSender set itself.
354 EXPECT_TRUE(packet->HasExtension<TransmissionOffset>());
355 EXPECT_TRUE(packet->HasExtension<AbsoluteSendTime>());
356 EXPECT_TRUE(packet->HasExtension<TransportSequenceNumber>());
357 // Do not allocate media specific extensions.
358 EXPECT_FALSE(packet->HasExtension<AudioLevel>());
359 EXPECT_FALSE(packet->HasExtension<VideoOrientation>());
360}
361
minyue3a407ee2017-04-03 01:10:33 -0700362TEST_P(RtpSenderTestWithoutPacer, AssignSequenceNumberAdvanceSequenceNumber) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200363 auto packet = rtp_sender_->AllocatePacket();
364 ASSERT_TRUE(packet);
365 const uint16_t sequence_number = rtp_sender_->SequenceNumber();
366
367 EXPECT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
368
369 EXPECT_EQ(sequence_number, packet->SequenceNumber());
370 EXPECT_EQ(sequence_number + 1, rtp_sender_->SequenceNumber());
371}
372
minyue3a407ee2017-04-03 01:10:33 -0700373TEST_P(RtpSenderTestWithoutPacer, AssignSequenceNumberFailsOnNotSending) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200374 auto packet = rtp_sender_->AllocatePacket();
375 ASSERT_TRUE(packet);
376
377 rtp_sender_->SetSendingMediaStatus(false);
378 EXPECT_FALSE(rtp_sender_->AssignSequenceNumber(packet.get()));
379}
380
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100381TEST_P(RtpSenderTestWithoutPacer, AssignSequenceNumberMayAllowPaddingOnVideo) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200382 constexpr size_t kPaddingSize = 100;
383 auto packet = rtp_sender_->AllocatePacket();
384 ASSERT_TRUE(packet);
385
Erik Språng4208a132019-08-26 08:58:45 +0200386 ASSERT_TRUE(rtp_sender_->GeneratePadding(kPaddingSize).empty());
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200387 packet->SetMarker(false);
388 ASSERT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100389 // Packet without marker bit doesn't allow padding on video stream.
Erik Språng4208a132019-08-26 08:58:45 +0200390 ASSERT_TRUE(rtp_sender_->GeneratePadding(kPaddingSize).empty());
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200391
392 packet->SetMarker(true);
393 ASSERT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
394 // Packet with marker bit allows send padding.
Erik Språng4208a132019-08-26 08:58:45 +0200395 ASSERT_FALSE(rtp_sender_->GeneratePadding(kPaddingSize).empty());
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200396}
397
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100398TEST_P(RtpSenderTest, AssignSequenceNumberAllowsPaddingOnAudio) {
399 MockTransport transport;
Erik Språng4580ca22019-07-04 10:38:43 +0200400 RtpRtcp::Configuration config;
401 config.audio = true;
402 config.clock = &fake_clock_;
403 config.outgoing_transport = &transport;
404 config.paced_sender = &mock_paced_sender_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200405 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200406 config.event_log = &mock_rtc_event_log_;
407 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200408 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +0200409
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100410 rtp_sender_->SetTimestampOffset(0);
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100411
412 std::unique_ptr<RtpPacketToSend> audio_packet = rtp_sender_->AllocatePacket();
413 // Padding on audio stream allowed regardless of marker in the last packet.
414 audio_packet->SetMarker(false);
415 audio_packet->SetPayloadType(kPayload);
416 rtp_sender_->AssignSequenceNumber(audio_packet.get());
417
418 const size_t kPaddingSize = 59;
419 EXPECT_CALL(transport, SendRtp(_, kPaddingSize + kRtpHeaderSize, _))
Erik Språng214f5432019-06-20 15:09:58 +0200420 .WillOnce(Return(true));
Erik Språng4208a132019-08-26 08:58:45 +0200421 EXPECT_EQ(kPaddingSize, GenerateAndSendPadding(kPaddingSize));
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100422
423 // Requested padding size is too small, will send a larger one.
424 const size_t kMinPaddingSize = 50;
425 EXPECT_CALL(transport, SendRtp(_, kMinPaddingSize + kRtpHeaderSize, _))
Erik Språng214f5432019-06-20 15:09:58 +0200426 .WillOnce(Return(true));
Erik Språng4208a132019-08-26 08:58:45 +0200427 EXPECT_EQ(kMinPaddingSize, GenerateAndSendPadding(kMinPaddingSize - 5));
Danil Chapovalovb3179c72018-03-22 10:13:07 +0100428}
429
minyue3a407ee2017-04-03 01:10:33 -0700430TEST_P(RtpSenderTestWithoutPacer, AssignSequenceNumberSetPaddingTimestamps) {
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200431 constexpr size_t kPaddingSize = 100;
432 auto packet = rtp_sender_->AllocatePacket();
433 ASSERT_TRUE(packet);
434 packet->SetMarker(true);
435 packet->SetTimestamp(kTimestamp);
436
437 ASSERT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
Erik Språng4208a132019-08-26 08:58:45 +0200438 auto padding_packets = rtp_sender_->GeneratePadding(kPaddingSize);
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200439
Erik Språng4208a132019-08-26 08:58:45 +0200440 ASSERT_EQ(1u, padding_packets.size());
danilchap12ba1862016-10-26 02:41:55 -0700441 // Verify padding packet timestamp.
Erik Språng4208a132019-08-26 08:58:45 +0200442 EXPECT_EQ(kTimestamp, padding_packets[0]->Timestamp());
Danil Chapovalov5e57b172016-09-02 19:15:59 +0200443}
444
minyue3a407ee2017-04-03 01:10:33 -0700445TEST_P(RtpSenderTestWithoutPacer,
446 TransportFeedbackObserverGetsCorrectByteCount) {
447 constexpr int kRtpOverheadBytesPerPacket = 12 + 8;
Erik Språng9c771c22019-06-17 16:31:53 +0200448 NiceMock<MockOverheadObserver> mock_overhead_observer;
Erik Språng4580ca22019-07-04 10:38:43 +0200449
450 RtpRtcp::Configuration config;
451 config.clock = &fake_clock_;
452 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200453 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200454 config.transport_feedback_callback = &feedback_observer_;
455 config.event_log = &mock_rtc_event_log_;
456 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
457 config.overhead_observer = &mock_overhead_observer;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200458 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +0200459
minyue3a407ee2017-04-03 01:10:33 -0700460 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
461 kRtpExtensionTransportSequenceNumber,
462 kTransportSequenceNumberExtensionId));
minyue3a407ee2017-04-03 01:10:33 -0700463
464 const size_t expected_bytes =
Erik Språngf6468d22019-07-05 16:53:43 +0200465 GetParam().with_overhead
466 ? sizeof(kPayloadData) + kRtpOverheadBytesPerPacket
467 : sizeof(kPayloadData);
minyue3a407ee2017-04-03 01:10:33 -0700468
469 EXPECT_CALL(feedback_observer_,
Erik Språng30a276b2019-04-23 12:00:11 +0200470 OnAddPacket(AllOf(
471 Field(&RtpPacketSendInfo::ssrc, rtp_sender_->SSRC()),
472 Field(&RtpPacketSendInfo::transport_sequence_number,
473 kTransportSequenceNumber),
474 Field(&RtpPacketSendInfo::rtp_sequence_number,
475 rtp_sender_->SequenceNumber()),
476 Field(&RtpPacketSendInfo::length, expected_bytes),
477 Field(&RtpPacketSendInfo::pacing_info, PacedPacketInfo()))))
minyue3a407ee2017-04-03 01:10:33 -0700478 .Times(1);
479 EXPECT_CALL(mock_overhead_observer,
480 OnOverheadChanged(kRtpOverheadBytesPerPacket))
481 .Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100482 SendGenericPacket();
minyue3a407ee2017-04-03 01:10:33 -0700483}
484
485TEST_P(RtpSenderTestWithoutPacer, SendsPacketsWithTransportSequenceNumber) {
Erik Språng4580ca22019-07-04 10:38:43 +0200486 RtpRtcp::Configuration config;
487 config.clock = &fake_clock_;
488 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200489 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200490 config.transport_feedback_callback = &feedback_observer_;
491 config.event_log = &mock_rtc_event_log_;
492 config.send_packet_observer = &send_packet_observer_;
493 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200494 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +0200495
Stefan Holmerf5dca482016-01-27 12:58:51 +0100496 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
497 kRtpExtensionTransportSequenceNumber,
498 kTransportSequenceNumberExtensionId));
499
asapersson35151f32016-05-02 23:44:01 -0700500 EXPECT_CALL(send_packet_observer_,
501 OnSendPacket(kTransportSequenceNumber, _, _))
502 .Times(1);
minyue3a407ee2017-04-03 01:10:33 -0700503
504 EXPECT_CALL(feedback_observer_,
Erik Språng30a276b2019-04-23 12:00:11 +0200505 OnAddPacket(AllOf(
506 Field(&RtpPacketSendInfo::ssrc, rtp_sender_->SSRC()),
507 Field(&RtpPacketSendInfo::transport_sequence_number,
508 kTransportSequenceNumber),
509 Field(&RtpPacketSendInfo::rtp_sequence_number,
510 rtp_sender_->SequenceNumber()),
511 Field(&RtpPacketSendInfo::pacing_info, PacedPacketInfo()))))
Stefan Holmera246cfb2016-08-23 17:51:42 +0200512 .Times(1);
asapersson35151f32016-05-02 23:44:01 -0700513
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100514 SendGenericPacket();
Stefan Holmerf5dca482016-01-27 12:58:51 +0100515
danilchap12ba1862016-10-26 02:41:55 -0700516 const auto& packet = transport_.last_sent_packet();
517 uint16_t transport_seq_no;
518 ASSERT_TRUE(packet.GetExtension<TransportSequenceNumber>(&transport_seq_no));
519 EXPECT_EQ(kTransportSequenceNumber, transport_seq_no);
Petter Strandmark26bc6692018-05-29 08:43:35 +0200520 EXPECT_EQ(transport_.last_options_.packet_id, transport_seq_no);
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200521 EXPECT_TRUE(transport_.last_options_.included_in_allocation);
Petter Strandmark26bc6692018-05-29 08:43:35 +0200522}
523
524TEST_P(RtpSenderTestWithoutPacer, PacketOptionsNoRetransmission) {
Erik Språng4580ca22019-07-04 10:38:43 +0200525 RtpRtcp::Configuration config;
526 config.clock = &fake_clock_;
527 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200528 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200529 config.transport_feedback_callback = &feedback_observer_;
530 config.event_log = &mock_rtc_event_log_;
531 config.send_packet_observer = &send_packet_observer_;
532 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200533 rtp_sender_ = std::make_unique<RTPSender>(config);
Petter Strandmark26bc6692018-05-29 08:43:35 +0200534
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100535 SendGenericPacket();
Petter Strandmark26bc6692018-05-29 08:43:35 +0200536
537 EXPECT_FALSE(transport_.last_options_.is_retransmit);
Stefan Holmerf5dca482016-01-27 12:58:51 +0100538}
539
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200540TEST_P(RtpSenderTestWithoutPacer,
541 SetsIncludedInFeedbackWhenTransportSequenceNumberExtensionIsRegistered) {
542 SetUpRtpSender(false, false);
543 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionTransportSequenceNumber,
544 kTransportSequenceNumberExtensionId);
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200545 EXPECT_CALL(send_packet_observer_, OnSendPacket).Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100546 SendGenericPacket();
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200547 EXPECT_TRUE(transport_.last_options_.included_in_feedback);
548}
549
550TEST_P(
551 RtpSenderTestWithoutPacer,
552 SetsIncludedInAllocationWhenTransportSequenceNumberExtensionIsRegistered) {
553 SetUpRtpSender(false, false);
554 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionTransportSequenceNumber,
555 kTransportSequenceNumberExtensionId);
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200556 EXPECT_CALL(send_packet_observer_, OnSendPacket).Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100557 SendGenericPacket();
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200558 EXPECT_TRUE(transport_.last_options_.included_in_allocation);
559}
560
561TEST_P(RtpSenderTestWithoutPacer,
562 SetsIncludedInAllocationWhenForcedAsPartOfAllocation) {
563 SetUpRtpSender(false, false);
564 rtp_sender_->SetAsPartOfAllocation(true);
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100565 SendGenericPacket();
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200566 EXPECT_FALSE(transport_.last_options_.included_in_feedback);
567 EXPECT_TRUE(transport_.last_options_.included_in_allocation);
568}
569
570TEST_P(RtpSenderTestWithoutPacer, DoesnSetIncludedInAllocationByDefault) {
571 SetUpRtpSender(false, false);
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100572 SendGenericPacket();
Sebastian Jansson1bca65b2018-10-10 09:58:08 +0200573 EXPECT_FALSE(transport_.last_options_.included_in_feedback);
574 EXPECT_FALSE(transport_.last_options_.included_in_allocation);
stefana23fc622016-07-28 07:56:38 -0700575}
asapersson35151f32016-05-02 23:44:01 -0700576
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200577TEST_P(RtpSenderTestWithoutPacer, OnSendSideDelayUpdated) {
Erik Språng9c771c22019-06-17 16:31:53 +0200578 StrictMock<MockSendSideDelayObserver> send_side_delay_observer_;
Erik Språng4580ca22019-07-04 10:38:43 +0200579
580 RtpRtcp::Configuration config;
581 config.clock = &fake_clock_;
582 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200583 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200584 config.send_side_delay_observer = &send_side_delay_observer_;
585 config.event_log = &mock_rtc_event_log_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200586 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +0200587
Niels Möller5fe95102019-03-04 16:49:25 +0100588 PlayoutDelayOracle playout_delay_oracle;
Niels Möller59ab1cf2019-02-06 22:48:11 +0100589 RTPSenderVideo rtp_sender_video(&fake_clock_, rtp_sender_.get(), nullptr,
Elad Alona0e99432019-05-24 13:50:56 +0200590 &playout_delay_oracle, nullptr, false, false,
Andrei Dumitru09872732019-08-29 12:00:29 -0700591 false, FieldTrialBasedConfig());
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200592
593 const uint8_t kPayloadType = 127;
Niels Möller8a40edd2019-01-24 18:04:44 +0100594 const char payload_name[] = "GENERIC";
Niels Möller59ab1cf2019-02-06 22:48:11 +0100595
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +0200596 rtp_sender_video.RegisterPayloadType(kPayloadType, payload_name,
597 /*raw_payload=*/false);
Niels Möller59ab1cf2019-02-06 22:48:11 +0100598
599 const uint32_t kCaptureTimeMsToRtpTimestamp = 90; // 90 kHz clock
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200600 RTPVideoHeader video_header;
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200601
Henrik Boström9fe18342019-05-16 18:38:20 +0200602 // Send packet with 10 ms send-side delay. The average, max and total should
603 // be 10 ms.
604 EXPECT_CALL(send_side_delay_observer_,
605 SendSideDelayUpdated(10, 10, 10, kSsrc))
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200606 .Times(1);
607 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
608 fake_clock_.AdvanceTimeMilliseconds(10);
Niels Möller59ab1cf2019-02-06 22:48:11 +0100609 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +0100610 VideoFrameType::kVideoFrameKey, kPayloadType,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200611 capture_time_ms * kCaptureTimeMsToRtpTimestamp, capture_time_ms,
Niels Möller59ab1cf2019-02-06 22:48:11 +0100612 kPayloadData, sizeof(kPayloadData), nullptr, &video_header,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200613 kDefaultExpectedRetransmissionTimeMs));
614
Henrik Boström9fe18342019-05-16 18:38:20 +0200615 // Send another packet with 20 ms delay. The average, max and total should be
616 // 15, 20 and 30 ms respectively.
617 EXPECT_CALL(send_side_delay_observer_,
618 SendSideDelayUpdated(15, 20, 30, kSsrc))
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200619 .Times(1);
620 fake_clock_.AdvanceTimeMilliseconds(10);
Niels Möller59ab1cf2019-02-06 22:48:11 +0100621 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +0100622 VideoFrameType::kVideoFrameKey, kPayloadType,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200623 capture_time_ms * kCaptureTimeMsToRtpTimestamp, capture_time_ms,
Niels Möller59ab1cf2019-02-06 22:48:11 +0100624 kPayloadData, sizeof(kPayloadData), nullptr, &video_header,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200625 kDefaultExpectedRetransmissionTimeMs));
626
627 // Send another packet at the same time, which replaces the last packet.
628 // Since this packet has 0 ms delay, the average is now 5 ms and max is 10 ms.
Henrik Boström9fe18342019-05-16 18:38:20 +0200629 // The total counter stays the same though.
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200630 // TODO(terelius): Is is not clear that this is the right behavior.
Henrik Boström9fe18342019-05-16 18:38:20 +0200631 EXPECT_CALL(send_side_delay_observer_, SendSideDelayUpdated(5, 10, 30, kSsrc))
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200632 .Times(1);
633 capture_time_ms = fake_clock_.TimeInMilliseconds();
Niels Möller59ab1cf2019-02-06 22:48:11 +0100634 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +0100635 VideoFrameType::kVideoFrameKey, kPayloadType,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200636 capture_time_ms * kCaptureTimeMsToRtpTimestamp, capture_time_ms,
Niels Möller59ab1cf2019-02-06 22:48:11 +0100637 kPayloadData, sizeof(kPayloadData), nullptr, &video_header,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200638 kDefaultExpectedRetransmissionTimeMs));
639
640 // Send a packet 1 second later. The earlier packets should have timed
Henrik Boström9fe18342019-05-16 18:38:20 +0200641 // out, so both max and average should be the delay of this packet. The total
642 // keeps increasing.
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200643 fake_clock_.AdvanceTimeMilliseconds(1000);
644 capture_time_ms = fake_clock_.TimeInMilliseconds();
645 fake_clock_.AdvanceTimeMilliseconds(1);
Henrik Boström9fe18342019-05-16 18:38:20 +0200646 EXPECT_CALL(send_side_delay_observer_, SendSideDelayUpdated(1, 1, 31, kSsrc))
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200647 .Times(1);
Niels Möller59ab1cf2019-02-06 22:48:11 +0100648 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +0100649 VideoFrameType::kVideoFrameKey, kPayloadType,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200650 capture_time_ms * kCaptureTimeMsToRtpTimestamp, capture_time_ms,
Niels Möller59ab1cf2019-02-06 22:48:11 +0100651 kPayloadData, sizeof(kPayloadData), nullptr, &video_header,
Johannes Kron4a8a5e72018-09-26 09:57:48 +0200652 kDefaultExpectedRetransmissionTimeMs));
653}
654
minyue3a407ee2017-04-03 01:10:33 -0700655TEST_P(RtpSenderTestWithoutPacer, OnSendPacketUpdated) {
stefana23fc622016-07-28 07:56:38 -0700656 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
657 kRtpExtensionTransportSequenceNumber,
658 kTransportSequenceNumberExtensionId));
asapersson35151f32016-05-02 23:44:01 -0700659 EXPECT_CALL(send_packet_observer_,
660 OnSendPacket(kTransportSequenceNumber, _, _))
661 .Times(1);
662
Niels Möllere7b9e6b2019-02-06 18:23:44 +0100663 SendGenericPacket();
asapersson35151f32016-05-02 23:44:01 -0700664}
665
minyue3a407ee2017-04-03 01:10:33 -0700666TEST_P(RtpSenderTest, SendsPacketsWithTransportSequenceNumber) {
Erik Språng4580ca22019-07-04 10:38:43 +0200667 RtpRtcp::Configuration config;
668 config.clock = &fake_clock_;
669 config.outgoing_transport = &transport_;
670 config.paced_sender = &mock_paced_sender_;
Erik Språng54d5d2c2019-08-20 17:22:36 +0200671 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +0200672 config.transport_feedback_callback = &feedback_observer_;
673 config.event_log = &mock_rtc_event_log_;
674 config.send_packet_observer = &send_packet_observer_;
675 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200676 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +0200677
brandtr9dfff292016-11-14 05:14:50 -0800678 rtp_sender_->SetSequenceNumber(kSeqNum);
Stefan Holmera246cfb2016-08-23 17:51:42 +0200679 rtp_sender_->SetStorePacketsStatus(true, 10);
680 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
681 kRtpExtensionTransportSequenceNumber,
682 kTransportSequenceNumberExtensionId));
683
Stefan Holmera246cfb2016-08-23 17:51:42 +0200684 EXPECT_CALL(send_packet_observer_,
685 OnSendPacket(kTransportSequenceNumber, _, _))
686 .Times(1);
minyue3a407ee2017-04-03 01:10:33 -0700687 EXPECT_CALL(feedback_observer_,
Erik Språng30a276b2019-04-23 12:00:11 +0200688 OnAddPacket(AllOf(
689 Field(&RtpPacketSendInfo::ssrc, rtp_sender_->SSRC()),
690 Field(&RtpPacketSendInfo::transport_sequence_number,
691 kTransportSequenceNumber),
692 Field(&RtpPacketSendInfo::rtp_sequence_number,
693 rtp_sender_->SequenceNumber()),
694 Field(&RtpPacketSendInfo::pacing_info, PacedPacketInfo()))))
Stefan Holmera246cfb2016-08-23 17:51:42 +0200695 .Times(1);
696
Erik Språngf5815fa2019-08-21 14:27:31 +0200697 EXPECT_CALL(
698 mock_paced_sender_,
699 EnqueuePacket(
700 AllOf(Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
701 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
702 auto packet = SendGenericPacket();
703 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
704 // Transport sequence number is set by PacketRouter, before TrySendPacket().
705 packet->SetExtension<TransportSequenceNumber>(kTransportSequenceNumber);
706 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
Stefan Holmera246cfb2016-08-23 17:51:42 +0200707
danilchap12ba1862016-10-26 02:41:55 -0700708 uint16_t transport_seq_no;
Erik Språngf5815fa2019-08-21 14:27:31 +0200709 EXPECT_TRUE(
710 transport_.last_sent_packet().GetExtension<TransportSequenceNumber>(
711 &transport_seq_no));
danilchap12ba1862016-10-26 02:41:55 -0700712 EXPECT_EQ(kTransportSequenceNumber, transport_seq_no);
Petter Strandmark26bc6692018-05-29 08:43:35 +0200713 EXPECT_EQ(transport_.last_options_.packet_id, transport_seq_no);
Stefan Holmera246cfb2016-08-23 17:51:42 +0200714}
715
Erik Språng7b52f102018-02-07 14:37:37 +0100716TEST_P(RtpSenderTest, WritesPacerExitToTimingExtension) {
ilnik04f4d122017-06-19 07:18:55 -0700717 rtp_sender_->SetStorePacketsStatus(true, 10);
718 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
719 kRtpExtensionVideoTiming, kVideoTimingExtensionId));
720 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
721 auto packet = rtp_sender_->AllocatePacket();
722 packet->SetPayloadType(kPayload);
723 packet->SetMarker(true);
724 packet->SetTimestamp(kTimestamp);
725 packet->set_capture_time_ms(capture_time_ms);
ilnik2edc6842017-07-06 03:06:50 -0700726 const VideoSendTiming kVideoTiming = {0u, 0u, 0u, 0u, 0u, 0u, true};
ilnik04f4d122017-06-19 07:18:55 -0700727 packet->SetExtension<VideoTimingExtension>(kVideoTiming);
728 EXPECT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
729 size_t packet_size = packet->size();
ilnik04f4d122017-06-19 07:18:55 -0700730
731 const int kStoredTimeInMs = 100;
Erik Språngf5815fa2019-08-21 14:27:31 +0200732 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Erik Språng70768f42019-08-27 18:16:26 +0200733 packet->set_allow_retransmission(true);
Erik Språngf5815fa2019-08-21 14:27:31 +0200734 EXPECT_CALL(mock_paced_sender_,
735 EnqueuePacket(Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc))));
Erik Språng70768f42019-08-27 18:16:26 +0200736 EXPECT_TRUE(
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200737 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf5815fa2019-08-21 14:27:31 +0200738 fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs);
739 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
ilnik04f4d122017-06-19 07:18:55 -0700740 EXPECT_EQ(1, transport_.packets_sent());
741 EXPECT_EQ(packet_size, transport_.last_sent_packet().size());
742
danilchapce251812017-09-11 12:24:41 -0700743 VideoSendTiming video_timing;
744 EXPECT_TRUE(transport_.last_sent_packet().GetExtension<VideoTimingExtension>(
745 &video_timing));
746 EXPECT_EQ(kStoredTimeInMs, video_timing.pacer_exit_delta_ms);
Erik Språng7b52f102018-02-07 14:37:37 +0100747}
ilnik04f4d122017-06-19 07:18:55 -0700748
Danil Chapovalovaf52b682018-11-27 10:48:27 +0100749TEST_P(RtpSenderTest, WritesNetwork2ToTimingExtensionWithPacer) {
750 SetUpRtpSender(/*pacer=*/true, /*populate_network2=*/true);
Erik Språng7b52f102018-02-07 14:37:37 +0100751 rtp_sender_->SetStorePacketsStatus(true, 10);
752 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
753 kRtpExtensionVideoTiming, kVideoTimingExtensionId));
754 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
755 auto packet = rtp_sender_->AllocatePacket();
756 packet->SetPayloadType(kPayload);
757 packet->SetMarker(true);
758 packet->SetTimestamp(kTimestamp);
759 packet->set_capture_time_ms(capture_time_ms);
760 const uint16_t kPacerExitMs = 1234u;
761 const VideoSendTiming kVideoTiming = {0u, 0u, 0u, kPacerExitMs, 0u, 0u, true};
762 packet->SetExtension<VideoTimingExtension>(kVideoTiming);
763 EXPECT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
764 size_t packet_size = packet->size();
765
766 const int kStoredTimeInMs = 100;
Erik Språngf6468d22019-07-05 16:53:43 +0200767
Erik Språngf6468d22019-07-05 16:53:43 +0200768 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Erik Språng70768f42019-08-27 18:16:26 +0200769 packet->set_allow_retransmission(true);
Erik Språngf6468d22019-07-05 16:53:43 +0200770 EXPECT_CALL(
771 mock_paced_sender_,
772 EnqueuePacket(Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc))));
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200773 EXPECT_TRUE(
774 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +0200775 fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs);
776 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +0200777
Erik Språng7b52f102018-02-07 14:37:37 +0100778 EXPECT_EQ(1, transport_.packets_sent());
ilnik04f4d122017-06-19 07:18:55 -0700779 EXPECT_EQ(packet_size, transport_.last_sent_packet().size());
780
Erik Språng7b52f102018-02-07 14:37:37 +0100781 VideoSendTiming video_timing;
danilchapce251812017-09-11 12:24:41 -0700782 EXPECT_TRUE(transport_.last_sent_packet().GetExtension<VideoTimingExtension>(
783 &video_timing));
Erik Språng7b52f102018-02-07 14:37:37 +0100784 EXPECT_EQ(kStoredTimeInMs, video_timing.network2_timestamp_delta_ms);
785 EXPECT_EQ(kPacerExitMs, video_timing.pacer_exit_delta_ms);
ilnik04f4d122017-06-19 07:18:55 -0700786}
787
Danil Chapovalovaf52b682018-11-27 10:48:27 +0100788TEST_P(RtpSenderTest, WritesNetwork2ToTimingExtensionWithoutPacer) {
789 SetUpRtpSender(/*pacer=*/false, /*populate_network2=*/true);
790 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
791 kRtpExtensionVideoTiming, kVideoTimingExtensionId));
792 auto packet = rtp_sender_->AllocatePacket();
793 packet->SetMarker(true);
794 packet->set_capture_time_ms(fake_clock_.TimeInMilliseconds());
795 const VideoSendTiming kVideoTiming = {0u, 0u, 0u, 0u, 0u, 0u, true};
796 packet->SetExtension<VideoTimingExtension>(kVideoTiming);
Erik Språng70768f42019-08-27 18:16:26 +0200797 packet->set_allow_retransmission(true);
Danil Chapovalovaf52b682018-11-27 10:48:27 +0100798 EXPECT_TRUE(rtp_sender_->AssignSequenceNumber(packet.get()));
Erik Språng1fbfecd2019-08-26 19:00:05 +0200799 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Danil Chapovalovaf52b682018-11-27 10:48:27 +0100800
801 const int kPropagateTimeMs = 10;
802 fake_clock_.AdvanceTimeMilliseconds(kPropagateTimeMs);
803
Erik Språng70768f42019-08-27 18:16:26 +0200804 EXPECT_TRUE(rtp_sender_->SendToNetwork(std::move(packet)));
Danil Chapovalovaf52b682018-11-27 10:48:27 +0100805
806 EXPECT_EQ(1, transport_.packets_sent());
807 absl::optional<VideoSendTiming> video_timing =
808 transport_.last_sent_packet().GetExtension<VideoTimingExtension>();
809 ASSERT_TRUE(video_timing);
810 EXPECT_EQ(kPropagateTimeMs, video_timing->network2_timestamp_delta_ms);
811}
812
minyue3a407ee2017-04-03 01:10:33 -0700813TEST_P(RtpSenderTest, TrafficSmoothingWithExtensions) {
Elad Alon4a87e1c2017-10-03 16:11:34 +0200814 EXPECT_CALL(mock_rtc_event_log_,
815 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)));
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000816
pwestin@webrtc.orgc66e8b32012-11-07 17:01:04 +0000817 rtp_sender_->SetStorePacketsStatus(true, 10);
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +0000818 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
danilchap162abd32015-12-10 02:39:40 -0800819 kRtpExtensionTransmissionTimeOffset,
820 kTransmissionTimeOffsetExtensionId));
821 EXPECT_EQ(
822 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
823 kAbsoluteSendTimeExtensionId));
stefan@webrtc.org8ccb9f92013-06-19 14:13:42 +0000824 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
danilchapb6f1fb52016-10-19 06:11:39 -0700825 auto packet =
826 BuildRtpPacket(kPayload, kMarkerBit, kTimestamp, capture_time_ms);
827 size_t packet_size = packet->size();
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000828
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000829 const int kStoredTimeInMs = 100;
Erik Språngf6468d22019-07-05 16:53:43 +0200830 EXPECT_CALL(
831 mock_paced_sender_,
832 EnqueuePacket(AllOf(
833 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
834 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
835 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Erik Språng70768f42019-08-27 18:16:26 +0200836 packet->set_allow_retransmission(true);
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200837 EXPECT_TRUE(
838 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +0200839 EXPECT_EQ(0, transport_.packets_sent());
840 fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs);
841 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000842
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000843 // Process send bucket. Packet should now be sent.
danilchap12ba1862016-10-26 02:41:55 -0700844 EXPECT_EQ(1, transport_.packets_sent());
845 EXPECT_EQ(packet_size, transport_.last_sent_packet().size());
846
stefan@webrtc.orga5cb98c2013-05-29 12:12:51 +0000847 webrtc::RTPHeader rtp_header;
danilchap12ba1862016-10-26 02:41:55 -0700848 transport_.last_sent_packet().GetHeader(&rtp_header);
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000849
850 // Verify transmission time offset.
851 EXPECT_EQ(kStoredTimeInMs * 90, rtp_header.extension.transmissionTimeOffset);
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +0000852 uint64_t expected_send_time =
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000853 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds());
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +0000854 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime);
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000855}
856
minyue3a407ee2017-04-03 01:10:33 -0700857TEST_P(RtpSenderTest, TrafficSmoothingRetransmits) {
Elad Alon4a87e1c2017-10-03 16:11:34 +0200858 EXPECT_CALL(mock_rtc_event_log_,
859 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)));
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000860
861 rtp_sender_->SetStorePacketsStatus(true, 10);
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +0000862 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
danilchap162abd32015-12-10 02:39:40 -0800863 kRtpExtensionTransmissionTimeOffset,
864 kTransmissionTimeOffsetExtensionId));
865 EXPECT_EQ(
866 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
867 kAbsoluteSendTimeExtensionId));
stefan@webrtc.org8ccb9f92013-06-19 14:13:42 +0000868 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
danilchapb6f1fb52016-10-19 06:11:39 -0700869 auto packet =
870 BuildRtpPacket(kPayload, kMarkerBit, kTimestamp, capture_time_ms);
871 size_t packet_size = packet->size();
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000872
873 // Packet should be stored in a send bucket.
Erik Språngf6468d22019-07-05 16:53:43 +0200874 EXPECT_CALL(
875 mock_paced_sender_,
876 EnqueuePacket(AllOf(
877 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
878 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
879 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
880 packet->set_allow_retransmission(true);
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200881 EXPECT_TRUE(
882 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +0200883 // Immediately process send bucket and send packet.
884 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +0200885
Erik Språng0f4f0552019-05-08 10:15:05 -0700886 EXPECT_EQ(1, transport_.packets_sent());
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000887
Erik Språng0f4f0552019-05-08 10:15:05 -0700888 // Retransmit packet.
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000889 const int kStoredTimeInMs = 100;
890 fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs);
891
Erik Språng0f4f0552019-05-08 10:15:05 -0700892 EXPECT_CALL(mock_rtc_event_log_,
893 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)));
Erik Språngf6468d22019-07-05 16:53:43 +0200894 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
895 packet->set_retransmitted_sequence_number(kSeqNum);
896 EXPECT_CALL(
897 mock_paced_sender_,
898 EnqueuePacket(AllOf(
899 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
900 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
901 EXPECT_EQ(static_cast<int>(packet_size),
902 rtp_sender_->ReSendPacket(kSeqNum));
903 EXPECT_EQ(1, transport_.packets_sent());
904 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000905
906 // Process send bucket. Packet should now be sent.
Erik Språng0f4f0552019-05-08 10:15:05 -0700907 EXPECT_EQ(2, transport_.packets_sent());
danilchap12ba1862016-10-26 02:41:55 -0700908 EXPECT_EQ(packet_size, transport_.last_sent_packet().size());
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000909
stefan@webrtc.orga5cb98c2013-05-29 12:12:51 +0000910 webrtc::RTPHeader rtp_header;
danilchap12ba1862016-10-26 02:41:55 -0700911 transport_.last_sent_packet().GetHeader(&rtp_header);
pwestin@webrtc.orgb0061f92013-04-27 00:41:08 +0000912
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +0000913 // Verify transmission time offset.
914 EXPECT_EQ(kStoredTimeInMs * 90, rtp_header.extension.transmissionTimeOffset);
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +0000915 uint64_t expected_send_time =
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000916 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds());
917 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime);
918}
919
920// This test sends 1 regular video packet, then 4 padding packets, and then
921// 1 more regular packet.
minyue3a407ee2017-04-03 01:10:33 -0700922TEST_P(RtpSenderTest, SendPadding) {
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000923 // Make all (non-padding) packets go to send queue.
Elad Alon4a87e1c2017-10-03 16:11:34 +0200924 EXPECT_CALL(mock_rtc_event_log_,
925 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)))
926 .Times(1 + 4 + 1);
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000927
928 uint16_t seq_num = kSeqNum;
929 uint32_t timestamp = kTimestamp;
930 rtp_sender_->SetStorePacketsStatus(true, 10);
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000931 size_t rtp_header_len = kRtpHeaderSize;
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000932 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
danilchap162abd32015-12-10 02:39:40 -0800933 kRtpExtensionTransmissionTimeOffset,
934 kTransmissionTimeOffsetExtensionId));
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000935 rtp_header_len += 4; // 4 bytes extension.
danilchap162abd32015-12-10 02:39:40 -0800936 EXPECT_EQ(
937 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
938 kAbsoluteSendTimeExtensionId));
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000939 rtp_header_len += 4; // 4 bytes extension.
940 rtp_header_len += 4; // 4 extra bytes common to all extension headers.
941
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000942 webrtc::RTPHeader rtp_header;
943
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000944 int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
danilchapb6f1fb52016-10-19 06:11:39 -0700945 auto packet =
946 BuildRtpPacket(kPayload, kMarkerBit, timestamp, capture_time_ms);
Stefan Holmer586b19b2015-09-18 11:14:31 +0200947 const uint32_t media_packet_timestamp = timestamp;
danilchapb6f1fb52016-10-19 06:11:39 -0700948 size_t packet_size = packet->size();
Erik Språngf6468d22019-07-05 16:53:43 +0200949 int total_packets_sent = 0;
950 const int kStoredTimeInMs = 100;
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000951
952 // Packet should be stored in a send bucket.
Erik Språngf6468d22019-07-05 16:53:43 +0200953 EXPECT_CALL(
954 mock_paced_sender_,
955 EnqueuePacket(AllOf(
956 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
957 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
958 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
959 packet->set_allow_retransmission(true);
Mirko Bonadei317a1f02019-09-17 17:06:18 +0200960 EXPECT_TRUE(
961 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +0200962 EXPECT_EQ(total_packets_sent, transport_.packets_sent());
963 fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs);
964 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
965 ++seq_num;
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000966
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000967 // Packet should now be sent. This test doesn't verify the regular video
968 // packet, since it is tested in another test.
danilchap12ba1862016-10-26 02:41:55 -0700969 EXPECT_EQ(++total_packets_sent, transport_.packets_sent());
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000970 timestamp += 90 * kStoredTimeInMs;
971
972 // Send padding 4 times, waiting 50 ms between each.
973 for (int i = 0; i < 4; ++i) {
974 const int kPaddingPeriodMs = 50;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000975 const size_t kPaddingBytes = 100;
976 const size_t kMaxPaddingLength = 224; // Value taken from rtp_sender.cc.
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000977 // Padding will be forced to full packets.
Erik Språng4208a132019-08-26 08:58:45 +0200978 EXPECT_EQ(kMaxPaddingLength, GenerateAndSendPadding(kPaddingBytes));
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000979
980 // Process send bucket. Padding should now be sent.
danilchap12ba1862016-10-26 02:41:55 -0700981 EXPECT_EQ(++total_packets_sent, transport_.packets_sent());
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000982 EXPECT_EQ(kMaxPaddingLength + rtp_header_len,
danilchap12ba1862016-10-26 02:41:55 -0700983 transport_.last_sent_packet().size());
984
985 transport_.last_sent_packet().GetHeader(&rtp_header);
pbosbd2522a2015-07-01 05:35:53 -0700986 EXPECT_EQ(kMaxPaddingLength, rtp_header.paddingLength);
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000987
Stefan Holmer586b19b2015-09-18 11:14:31 +0200988 // Verify sequence number and timestamp. The timestamp should be the same
989 // as the last media packet.
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000990 EXPECT_EQ(seq_num++, rtp_header.sequenceNumber);
Stefan Holmer586b19b2015-09-18 11:14:31 +0200991 EXPECT_EQ(media_packet_timestamp, rtp_header.timestamp);
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000992 // Verify transmission time offset.
Stefan Holmer586b19b2015-09-18 11:14:31 +0200993 int offset = timestamp - media_packet_timestamp;
994 EXPECT_EQ(offset, rtp_header.extension.transmissionTimeOffset);
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +0000995 uint64_t expected_send_time =
996 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds());
997 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime);
998 fake_clock_.AdvanceTimeMilliseconds(kPaddingPeriodMs);
999 timestamp += 90 * kPaddingPeriodMs;
1000 }
1001
1002 // Send a regular video packet again.
1003 capture_time_ms = fake_clock_.TimeInMilliseconds();
danilchapb6f1fb52016-10-19 06:11:39 -07001004 packet = BuildRtpPacket(kPayload, kMarkerBit, timestamp, capture_time_ms);
1005 packet_size = packet->size();
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +00001006
Erik Språngf6468d22019-07-05 16:53:43 +02001007 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Erik Språng70768f42019-08-27 18:16:26 +02001008 packet->set_allow_retransmission(true);
Erik Språngf6468d22019-07-05 16:53:43 +02001009 EXPECT_CALL(
1010 mock_paced_sender_,
1011 EnqueuePacket(AllOf(
1012 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
1013 Pointee(Property(&RtpPacketToSend::SequenceNumber, seq_num)))));
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001014 EXPECT_TRUE(
1015 rtp_sender_->SendToNetwork(std::make_unique<RtpPacketToSend>(*packet)));
Erik Språngf6468d22019-07-05 16:53:43 +02001016 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
terelius5d332ac2016-01-14 14:37:39 -08001017
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +00001018 // Process send bucket.
danilchap12ba1862016-10-26 02:41:55 -07001019 EXPECT_EQ(++total_packets_sent, transport_.packets_sent());
1020 EXPECT_EQ(packet_size, transport_.last_sent_packet().size());
1021 transport_.last_sent_packet().GetHeader(&rtp_header);
henrik.lundin@webrtc.org6e95d7a2013-11-15 08:59:19 +00001022
1023 // Verify sequence number and timestamp.
1024 EXPECT_EQ(seq_num, rtp_header.sequenceNumber);
1025 EXPECT_EQ(timestamp, rtp_header.timestamp);
1026 // Verify transmission time offset. This packet is sent without delay.
1027 EXPECT_EQ(0, rtp_header.extension.transmissionTimeOffset);
1028 uint64_t expected_send_time =
1029 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds());
solenberg@webrtc.org7ebbea12013-05-16 11:10:31 +00001030 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime);
asapersson@webrtc.org0b3c35a2012-01-16 11:06:31 +00001031}
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001032
minyue3a407ee2017-04-03 01:10:33 -07001033TEST_P(RtpSenderTest, OnSendPacketUpdated) {
stefana23fc622016-07-28 07:56:38 -07001034 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
1035 kRtpExtensionTransportSequenceNumber,
1036 kTransportSequenceNumberExtensionId));
asapersson35151f32016-05-02 23:44:01 -07001037 rtp_sender_->SetStorePacketsStatus(true, 10);
1038
1039 EXPECT_CALL(send_packet_observer_,
1040 OnSendPacket(kTransportSequenceNumber, _, _))
1041 .Times(1);
asapersson35151f32016-05-02 23:44:01 -07001042
Erik Språngf6468d22019-07-05 16:53:43 +02001043 EXPECT_CALL(
1044 mock_paced_sender_,
1045 EnqueuePacket(AllOf(
1046 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
1047 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
1048 auto packet = SendGenericPacket();
1049 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
1050 packet->SetExtension<TransportSequenceNumber>(kTransportSequenceNumber);
1051 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +02001052
danilchap12ba1862016-10-26 02:41:55 -07001053 EXPECT_EQ(1, transport_.packets_sent());
asapersson35151f32016-05-02 23:44:01 -07001054}
1055
minyue3a407ee2017-04-03 01:10:33 -07001056TEST_P(RtpSenderTest, OnSendPacketNotUpdatedForRetransmits) {
stefana23fc622016-07-28 07:56:38 -07001057 EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
1058 kRtpExtensionTransportSequenceNumber,
1059 kTransportSequenceNumberExtensionId));
asapersson35151f32016-05-02 23:44:01 -07001060 rtp_sender_->SetStorePacketsStatus(true, 10);
1061
1062 EXPECT_CALL(send_packet_observer_, OnSendPacket(_, _, _)).Times(0);
asapersson35151f32016-05-02 23:44:01 -07001063
Erik Språngf6468d22019-07-05 16:53:43 +02001064 EXPECT_CALL(
1065 mock_paced_sender_,
1066 EnqueuePacket(AllOf(
1067 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
1068 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))));
1069 auto packet = SendGenericPacket();
1070 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
1071 packet->SetExtension<TransportSequenceNumber>(kTransportSequenceNumber);
1072 rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +02001073
danilchap12ba1862016-10-26 02:41:55 -07001074 EXPECT_EQ(1, transport_.packets_sent());
Petter Strandmark26bc6692018-05-29 08:43:35 +02001075 EXPECT_TRUE(transport_.last_options_.is_retransmit);
asapersson35151f32016-05-02 23:44:01 -07001076}
1077
minyue3a407ee2017-04-03 01:10:33 -07001078TEST_P(RtpSenderTestWithoutPacer, SendGenericVideo) {
Niels Möller8a40edd2019-01-24 18:04:44 +01001079 const char payload_name[] = "GENERIC";
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001080 const uint8_t payload_type = 127;
Niels Möller5fe95102019-03-04 16:49:25 +01001081 PlayoutDelayOracle playout_delay_oracle;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001082 RTPSenderVideo rtp_sender_video(&fake_clock_, rtp_sender_.get(), nullptr,
Elad Alona0e99432019-05-24 13:50:56 +02001083 &playout_delay_oracle, nullptr, false, false,
Andrei Dumitru09872732019-08-29 12:00:29 -07001084 false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001085 rtp_sender_video.RegisterPayloadType(payload_type, payload_name,
1086 /*raw_payload=*/false);
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001087 uint8_t payload[] = {47, 11, 32, 93, 89};
1088
1089 // Send keyframe
Sami Kalliomäki426a80c2018-08-08 11:37:59 +02001090 RTPVideoHeader video_header;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001091 ASSERT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001092 VideoFrameType::kVideoFrameKey, payload_type, 1234, 4321, payload,
1093 sizeof(payload), nullptr, &video_header,
1094 kDefaultExpectedRetransmissionTimeMs));
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001095
danilchap96c15872016-11-21 01:35:29 -08001096 auto sent_payload = transport_.last_sent_packet().payload();
1097 uint8_t generic_header = sent_payload[0];
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001098 EXPECT_TRUE(generic_header & RtpFormatVideoGeneric::kKeyFrameBit);
1099 EXPECT_TRUE(generic_header & RtpFormatVideoGeneric::kFirstPacketBit);
danilchap96c15872016-11-21 01:35:29 -08001100 EXPECT_THAT(sent_payload.subview(1), ElementsAreArray(payload));
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001101
1102 // Send delta frame
1103 payload[0] = 13;
1104 payload[1] = 42;
1105 payload[4] = 13;
1106
Niels Möller59ab1cf2019-02-06 22:48:11 +01001107 ASSERT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001108 VideoFrameType::kVideoFrameDelta, payload_type, 1234, 4321, payload,
1109 sizeof(payload), nullptr, &video_header,
1110 kDefaultExpectedRetransmissionTimeMs));
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001111
danilchap96c15872016-11-21 01:35:29 -08001112 sent_payload = transport_.last_sent_packet().payload();
1113 generic_header = sent_payload[0];
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001114 EXPECT_FALSE(generic_header & RtpFormatVideoGeneric::kKeyFrameBit);
1115 EXPECT_TRUE(generic_header & RtpFormatVideoGeneric::kFirstPacketBit);
danilchap96c15872016-11-21 01:35:29 -08001116 EXPECT_THAT(sent_payload.subview(1), ElementsAreArray(payload));
pbos@webrtc.org8911ce42013-03-18 16:39:03 +00001117}
1118
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001119TEST_P(RtpSenderTestWithoutPacer, SendRawVideo) {
1120 const char payload_name[] = "VP8";
1121 const uint8_t payload_type = 111;
1122 const uint8_t payload[] = {11, 22, 33, 44, 55};
1123
1124 PlayoutDelayOracle playout_delay_oracle;
1125 RTPSenderVideo rtp_sender_video(&fake_clock_, rtp_sender_.get(), nullptr,
Elad Alona0e99432019-05-24 13:50:56 +02001126 &playout_delay_oracle, nullptr, false, false,
Andrei Dumitru09872732019-08-29 12:00:29 -07001127 false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001128 rtp_sender_video.RegisterPayloadType(payload_type, payload_name,
1129 /*raw_payload=*/true);
1130
1131 // Send a frame.
1132 RTPVideoHeader video_header;
1133 ASSERT_TRUE(rtp_sender_video.SendVideo(
1134 VideoFrameType::kVideoFrameKey, payload_type, 1234, 4321, payload,
1135 sizeof(payload), nullptr, &video_header,
1136 kDefaultExpectedRetransmissionTimeMs));
1137
1138 auto sent_payload = transport_.last_sent_packet().payload();
1139 EXPECT_THAT(sent_payload, ElementsAreArray(payload));
1140}
1141
minyue3a407ee2017-04-03 01:10:33 -07001142TEST_P(RtpSenderTest, SendFlexfecPackets) {
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001143 constexpr uint32_t kTimestamp = 1234;
brandtrdbdb3f12016-11-10 05:04:48 -08001144 constexpr int kMediaPayloadType = 127;
1145 constexpr int kFlexfecPayloadType = 118;
brandtrdbdb3f12016-11-10 05:04:48 -08001146 const std::vector<RtpExtension> kNoRtpExtensions;
erikvarga27883732017-05-17 05:08:38 -07001147 const std::vector<RtpExtensionSize> kNoRtpExtensionSizes;
Erik Språng4580ca22019-07-04 10:38:43 +02001148 FlexfecSender flexfec_sender(kFlexfecPayloadType, kFlexFecSsrc, kSsrc, kNoMid,
1149 kNoRtpExtensions, kNoRtpExtensionSizes,
brandtr48d21a22017-05-30 02:32:12 -07001150 nullptr /* rtp_state */, &fake_clock_);
brandtrdbdb3f12016-11-10 05:04:48 -08001151
1152 // Reset |rtp_sender_| to use FlexFEC.
Erik Språng4580ca22019-07-04 10:38:43 +02001153 RtpRtcp::Configuration config;
1154 config.clock = &fake_clock_;
1155 config.outgoing_transport = &transport_;
1156 config.paced_sender = &mock_paced_sender_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001157 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001158 config.flexfec_sender = &flexfec_sender_;
Erik Språng4580ca22019-07-04 10:38:43 +02001159 config.event_log = &mock_rtc_event_log_;
1160 config.send_packet_observer = &send_packet_observer_;
1161 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001162 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +02001163
brandtrdbdb3f12016-11-10 05:04:48 -08001164 rtp_sender_->SetSequenceNumber(kSeqNum);
brandtrdbdb3f12016-11-10 05:04:48 -08001165 rtp_sender_->SetStorePacketsStatus(true, 10);
1166
Niels Möller5fe95102019-03-04 16:49:25 +01001167 PlayoutDelayOracle playout_delay_oracle;
Elad Alona0e99432019-05-24 13:50:56 +02001168 RTPSenderVideo rtp_sender_video(
1169 &fake_clock_, rtp_sender_.get(), &flexfec_sender, &playout_delay_oracle,
Andrei Dumitru09872732019-08-29 12:00:29 -07001170 nullptr, false, false, false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001171 rtp_sender_video.RegisterPayloadType(kMediaPayloadType, "GENERIC",
1172 /*raw_payload=*/false);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001173
brandtrdbdb3f12016-11-10 05:04:48 -08001174 // Parameters selected to generate a single FEC packet per media packet.
1175 FecProtectionParams params;
1176 params.fec_rate = 15;
1177 params.max_fec_frames = 1;
1178 params.fec_mask_type = kFecMaskRandom;
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001179 rtp_sender_video.SetFecParameters(params, params);
brandtrdbdb3f12016-11-10 05:04:48 -08001180
brandtr9dfff292016-11-14 05:14:50 -08001181 uint16_t flexfec_seq_num;
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001182 RTPVideoHeader video_header;
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001183
Erik Språngf6468d22019-07-05 16:53:43 +02001184 std::unique_ptr<RtpPacketToSend> media_packet;
1185 std::unique_ptr<RtpPacketToSend> fec_packet;
1186
1187 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
1188 .Times(2)
1189 .WillRepeatedly([&](std::unique_ptr<RtpPacketToSend> packet) {
1190 if (packet->packet_type() == RtpPacketToSend::Type::kVideo) {
1191 EXPECT_EQ(packet->Ssrc(), kSsrc);
1192 EXPECT_EQ(packet->SequenceNumber(), kSeqNum);
1193 media_packet = std::move(packet);
1194 } else {
1195 EXPECT_EQ(packet->packet_type(),
1196 RtpPacketToSend::Type::kForwardErrorCorrection);
1197 EXPECT_EQ(packet->Ssrc(), kFlexFecSsrc);
1198 fec_packet = std::move(packet);
1199 }
1200 });
1201
1202 EXPECT_TRUE(rtp_sender_video.SendVideo(
1203 VideoFrameType::kVideoFrameKey, kMediaPayloadType, kTimestamp,
1204 fake_clock_.TimeInMilliseconds(), kPayloadData, sizeof(kPayloadData),
1205 nullptr, &video_header, kDefaultExpectedRetransmissionTimeMs));
1206 ASSERT_TRUE(media_packet != nullptr);
1207 ASSERT_TRUE(fec_packet != nullptr);
1208
1209 flexfec_seq_num = fec_packet->SequenceNumber();
1210 rtp_sender_->TrySendPacket(media_packet.get(), PacedPacketInfo());
1211 rtp_sender_->TrySendPacket(fec_packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +02001212
brandtr9dfff292016-11-14 05:14:50 -08001213 ASSERT_EQ(2, transport_.packets_sent());
Erik Språngf5815fa2019-08-21 14:27:31 +02001214 const RtpPacketReceived& sent_media_packet = transport_.sent_packets_[0];
1215 EXPECT_EQ(kMediaPayloadType, sent_media_packet.PayloadType());
1216 EXPECT_EQ(kSeqNum, sent_media_packet.SequenceNumber());
1217 EXPECT_EQ(kSsrc, sent_media_packet.Ssrc());
1218 const RtpPacketReceived& sent_flexfec_packet = transport_.sent_packets_[1];
1219 EXPECT_EQ(kFlexfecPayloadType, sent_flexfec_packet.PayloadType());
1220 EXPECT_EQ(flexfec_seq_num, sent_flexfec_packet.SequenceNumber());
1221 EXPECT_EQ(kFlexFecSsrc, sent_flexfec_packet.Ssrc());
brandtrdbdb3f12016-11-10 05:04:48 -08001222}
1223
Erik Språngf6468d22019-07-05 16:53:43 +02001224// TODO(ilnik): because of webrtc:7859. Once FEC moved below pacer, this test
1225// should be removed.
1226TEST_P(RtpSenderTest, NoFlexfecForTimingFrames) {
1227 constexpr uint32_t kTimestamp = 1234;
1228 const int64_t kCaptureTimeMs = fake_clock_.TimeInMilliseconds();
1229 constexpr int kMediaPayloadType = 127;
1230 constexpr int kFlexfecPayloadType = 118;
1231 const std::vector<RtpExtension> kNoRtpExtensions;
1232 const std::vector<RtpExtensionSize> kNoRtpExtensionSizes;
1233
1234 FlexfecSender flexfec_sender(kFlexfecPayloadType, kFlexFecSsrc, kSsrc, kNoMid,
1235 kNoRtpExtensions, kNoRtpExtensionSizes,
1236 nullptr /* rtp_state */, &fake_clock_);
1237
1238 // Reset |rtp_sender_| to use FlexFEC.
Erik Språng630443a2019-07-26 16:44:55 +02001239 RtpRtcp::Configuration config;
1240 config.clock = &fake_clock_;
1241 config.outgoing_transport = &transport_;
1242 config.paced_sender = &mock_paced_sender_;
1243 config.flexfec_sender = &flexfec_sender;
Erik Språng630443a2019-07-26 16:44:55 +02001244 config.event_log = &mock_rtc_event_log_;
1245 config.send_packet_observer = &send_packet_observer_;
1246 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001247 config.local_media_ssrc = kSsrc;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001248 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språngf6468d22019-07-05 16:53:43 +02001249 rtp_sender_->SetSequenceNumber(kSeqNum);
1250 rtp_sender_->SetStorePacketsStatus(true, 10);
1251
1252 PlayoutDelayOracle playout_delay_oracle;
1253 RTPSenderVideo rtp_sender_video(
1254 &fake_clock_, rtp_sender_.get(), &flexfec_sender, &playout_delay_oracle,
Andrei Dumitru09872732019-08-29 12:00:29 -07001255 nullptr, false, false, false, FieldTrialBasedConfig());
Erik Språngf6468d22019-07-05 16:53:43 +02001256 rtp_sender_video.RegisterPayloadType(kMediaPayloadType, "GENERIC",
1257 /*raw_payload=*/false);
1258
1259 // Need extension to be registered for timing frames to be sent.
1260 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
1261 kRtpExtensionVideoTiming, kVideoTimingExtensionId));
1262
1263 // Parameters selected to generate a single FEC packet per media packet.
1264 FecProtectionParams params;
1265 params.fec_rate = 15;
1266 params.max_fec_frames = 1;
1267 params.fec_mask_type = kFecMaskRandom;
1268 rtp_sender_video.SetFecParameters(params, params);
1269
1270 RTPVideoHeader video_header;
1271 video_header.video_timing.flags = VideoSendTiming::kTriggeredByTimer;
1272
1273 EXPECT_CALL(mock_rtc_event_log_,
1274 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)))
1275 .Times(1);
Erik Språngf6468d22019-07-05 16:53:43 +02001276 std::unique_ptr<RtpPacketToSend> rtp_packet;
1277 EXPECT_CALL(
1278 mock_paced_sender_,
1279 EnqueuePacket(AllOf(
1280 Pointee(Property(&RtpPacketToSend::Ssrc, kSsrc)),
1281 Pointee(Property(&RtpPacketToSend::SequenceNumber, kSeqNum)))))
1282 .WillOnce([&rtp_packet](std::unique_ptr<RtpPacketToSend> packet) {
1283 rtp_packet = std::move(packet);
1284 });
1285
1286 EXPECT_CALL(
1287 mock_paced_sender_,
1288 EnqueuePacket(Pointee(Property(&RtpPacketToSend::Ssrc, kFlexFecSsrc))))
1289 .Times(0); // Not called because packet should not be protected.
1290
1291 EXPECT_TRUE(rtp_sender_video.SendVideo(
1292 VideoFrameType::kVideoFrameKey, kMediaPayloadType, kTimestamp,
1293 kCaptureTimeMs, kPayloadData, sizeof(kPayloadData), nullptr,
1294 &video_header, kDefaultExpectedRetransmissionTimeMs));
1295
1296 EXPECT_TRUE(
1297 rtp_sender_->TrySendPacket(rtp_packet.get(), PacedPacketInfo()));
Erik Språngf6468d22019-07-05 16:53:43 +02001298
1299 ASSERT_EQ(1, transport_.packets_sent());
Erik Språngf5815fa2019-08-21 14:27:31 +02001300 const RtpPacketReceived& sent_media_packet1 = transport_.sent_packets_[0];
1301 EXPECT_EQ(kMediaPayloadType, sent_media_packet1.PayloadType());
1302 EXPECT_EQ(kSeqNum, sent_media_packet1.SequenceNumber());
1303 EXPECT_EQ(kSsrc, sent_media_packet1.Ssrc());
Erik Språngf6468d22019-07-05 16:53:43 +02001304
1305 // Now try to send not a timing frame.
1306 uint16_t flexfec_seq_num;
1307
1308 EXPECT_CALL(mock_rtc_event_log_,
1309 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)))
1310 .Times(2);
Erik Språngf5815fa2019-08-21 14:27:31 +02001311 std::unique_ptr<RtpPacketToSend> media_packet2;
1312 std::unique_ptr<RtpPacketToSend> fec_packet;
Erik Språngf6468d22019-07-05 16:53:43 +02001313
Erik Språngf5815fa2019-08-21 14:27:31 +02001314 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
1315 .Times(2)
1316 .WillRepeatedly([&](std::unique_ptr<RtpPacketToSend> packet) {
1317 if (packet->packet_type() == RtpPacketToSend::Type::kVideo) {
1318 EXPECT_EQ(packet->Ssrc(), kSsrc);
1319 EXPECT_EQ(packet->SequenceNumber(), kSeqNum + 1);
1320 media_packet2 = std::move(packet);
1321 } else {
1322 EXPECT_EQ(packet->packet_type(),
1323 RtpPacketToSend::Type::kForwardErrorCorrection);
1324 EXPECT_EQ(packet->Ssrc(), kFlexFecSsrc);
1325 fec_packet = std::move(packet);
1326 }
1327 });
Erik Språngf6468d22019-07-05 16:53:43 +02001328
Erik Språngf5815fa2019-08-21 14:27:31 +02001329 video_header.video_timing.flags = VideoSendTiming::kInvalid;
1330 EXPECT_TRUE(rtp_sender_video.SendVideo(
1331 VideoFrameType::kVideoFrameKey, kMediaPayloadType, kTimestamp + 1,
1332 kCaptureTimeMs + 1, kPayloadData, sizeof(kPayloadData), nullptr,
1333 &video_header, kDefaultExpectedRetransmissionTimeMs));
Erik Språngf6468d22019-07-05 16:53:43 +02001334
Erik Språngf5815fa2019-08-21 14:27:31 +02001335 ASSERT_TRUE(media_packet2 != nullptr);
1336 ASSERT_TRUE(fec_packet != nullptr);
Erik Språngf6468d22019-07-05 16:53:43 +02001337
Erik Språngf5815fa2019-08-21 14:27:31 +02001338 flexfec_seq_num = fec_packet->SequenceNumber();
1339 rtp_sender_->TrySendPacket(media_packet2.get(), PacedPacketInfo());
1340 rtp_sender_->TrySendPacket(fec_packet.get(), PacedPacketInfo());
Erik Språngf6468d22019-07-05 16:53:43 +02001341
1342 ASSERT_EQ(3, transport_.packets_sent());
Erik Språngf5815fa2019-08-21 14:27:31 +02001343 const RtpPacketReceived& sent_media_packet2 = transport_.sent_packets_[1];
1344 EXPECT_EQ(kMediaPayloadType, sent_media_packet2.PayloadType());
1345 EXPECT_EQ(kSeqNum + 1, sent_media_packet2.SequenceNumber());
1346 EXPECT_EQ(kSsrc, sent_media_packet2.Ssrc());
Erik Språngf6468d22019-07-05 16:53:43 +02001347 const RtpPacketReceived& flexfec_packet = transport_.sent_packets_[2];
1348 EXPECT_EQ(kFlexfecPayloadType, flexfec_packet.PayloadType());
1349 EXPECT_EQ(flexfec_seq_num, flexfec_packet.SequenceNumber());
1350 EXPECT_EQ(kFlexFecSsrc, flexfec_packet.Ssrc());
1351}
1352
minyue3a407ee2017-04-03 01:10:33 -07001353TEST_P(RtpSenderTestWithoutPacer, SendFlexfecPackets) {
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001354 constexpr uint32_t kTimestamp = 1234;
brandtrdbdb3f12016-11-10 05:04:48 -08001355 constexpr int kMediaPayloadType = 127;
1356 constexpr int kFlexfecPayloadType = 118;
brandtrdbdb3f12016-11-10 05:04:48 -08001357 const std::vector<RtpExtension> kNoRtpExtensions;
erikvarga27883732017-05-17 05:08:38 -07001358 const std::vector<RtpExtensionSize> kNoRtpExtensionSizes;
Erik Språngf6468d22019-07-05 16:53:43 +02001359 FlexfecSender flexfec_sender(kFlexfecPayloadType, kFlexFecSsrc, kSsrc, kNoMid,
Erik Språng4580ca22019-07-04 10:38:43 +02001360 kNoRtpExtensions, kNoRtpExtensionSizes,
brandtr48d21a22017-05-30 02:32:12 -07001361 nullptr /* rtp_state */, &fake_clock_);
brandtrdbdb3f12016-11-10 05:04:48 -08001362
1363 // Reset |rtp_sender_| to use FlexFEC.
Erik Språng4580ca22019-07-04 10:38:43 +02001364 RtpRtcp::Configuration config;
1365 config.clock = &fake_clock_;
1366 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001367 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001368 config.flexfec_sender = &flexfec_sender;
Erik Språng4580ca22019-07-04 10:38:43 +02001369 config.event_log = &mock_rtc_event_log_;
1370 config.send_packet_observer = &send_packet_observer_;
1371 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001372 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +02001373
brandtrdbdb3f12016-11-10 05:04:48 -08001374 rtp_sender_->SetSequenceNumber(kSeqNum);
brandtrdbdb3f12016-11-10 05:04:48 -08001375
Niels Möller5fe95102019-03-04 16:49:25 +01001376 PlayoutDelayOracle playout_delay_oracle;
Elad Alona0e99432019-05-24 13:50:56 +02001377 RTPSenderVideo rtp_sender_video(
1378 &fake_clock_, rtp_sender_.get(), &flexfec_sender, &playout_delay_oracle,
Andrei Dumitru09872732019-08-29 12:00:29 -07001379 nullptr, false, false, false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001380 rtp_sender_video.RegisterPayloadType(kMediaPayloadType, "GENERIC",
1381 /*raw_payload=*/false);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001382
brandtrdbdb3f12016-11-10 05:04:48 -08001383 // Parameters selected to generate a single FEC packet per media packet.
1384 FecProtectionParams params;
1385 params.fec_rate = 15;
1386 params.max_fec_frames = 1;
1387 params.fec_mask_type = kFecMaskRandom;
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001388 rtp_sender_video.SetFecParameters(params, params);
brandtrdbdb3f12016-11-10 05:04:48 -08001389
Elad Alon4a87e1c2017-10-03 16:11:34 +02001390 EXPECT_CALL(mock_rtc_event_log_,
1391 LogProxy(SameRtcEventTypeAs(RtcEvent::Type::RtpPacketOutgoing)))
1392 .Times(2);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001393 RTPVideoHeader video_header;
1394 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001395 VideoFrameType::kVideoFrameKey, kMediaPayloadType, kTimestamp,
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001396 fake_clock_.TimeInMilliseconds(), kPayloadData, sizeof(kPayloadData),
1397 nullptr, &video_header, kDefaultExpectedRetransmissionTimeMs));
1398
brandtrdbdb3f12016-11-10 05:04:48 -08001399 ASSERT_EQ(2, transport_.packets_sent());
1400 const RtpPacketReceived& media_packet = transport_.sent_packets_[0];
1401 EXPECT_EQ(kMediaPayloadType, media_packet.PayloadType());
Erik Språng4580ca22019-07-04 10:38:43 +02001402 EXPECT_EQ(kSsrc, media_packet.Ssrc());
brandtrdbdb3f12016-11-10 05:04:48 -08001403 const RtpPacketReceived& flexfec_packet = transport_.sent_packets_[1];
1404 EXPECT_EQ(kFlexfecPayloadType, flexfec_packet.PayloadType());
Erik Språngf6468d22019-07-05 16:53:43 +02001405 EXPECT_EQ(kFlexFecSsrc, flexfec_packet.Ssrc());
brandtrdbdb3f12016-11-10 05:04:48 -08001406}
1407
Steve Anton296a0ce2018-03-22 15:17:27 -07001408// Test that the MID header extension is included on sent packets when
1409// configured.
1410TEST_P(RtpSenderTestWithoutPacer, MidIncludedOnSentPackets) {
1411 const char kMid[] = "mid";
1412
Steve Anton2bac7da2019-07-21 15:04:21 -04001413 EnableMidSending(kMid);
Steve Anton296a0ce2018-03-22 15:17:27 -07001414
1415 // Send a couple packets.
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001416 SendGenericPacket();
1417 SendGenericPacket();
Steve Anton296a0ce2018-03-22 15:17:27 -07001418
1419 // Expect both packets to have the MID set.
1420 ASSERT_EQ(2u, transport_.sent_packets_.size());
1421 for (const RtpPacketReceived& packet : transport_.sent_packets_) {
1422 std::string mid;
1423 ASSERT_TRUE(packet.GetExtension<RtpMid>(&mid));
1424 EXPECT_EQ(kMid, mid);
1425 }
1426}
1427
Amit Hilbuch77938e62018-12-21 09:23:38 -08001428TEST_P(RtpSenderTestWithoutPacer, RidIncludedOnSentPackets) {
1429 const char kRid[] = "f";
1430
Steve Anton2bac7da2019-07-21 15:04:21 -04001431 EnableRidSending(kRid);
Amit Hilbuch77938e62018-12-21 09:23:38 -08001432
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001433 SendGenericPacket();
Amit Hilbuch77938e62018-12-21 09:23:38 -08001434
1435 ASSERT_EQ(1u, transport_.sent_packets_.size());
1436 const RtpPacketReceived& packet = transport_.sent_packets_[0];
1437 std::string rid;
1438 ASSERT_TRUE(packet.GetExtension<RtpStreamId>(&rid));
1439 EXPECT_EQ(kRid, rid);
1440}
1441
1442TEST_P(RtpSenderTestWithoutPacer, RidIncludedOnRtxSentPackets) {
1443 const char kRid[] = "f";
Amit Hilbuch77938e62018-12-21 09:23:38 -08001444
Steve Anton2bac7da2019-07-21 15:04:21 -04001445 EnableRtx();
1446 EnableRidSending(kRid);
Amit Hilbuch77938e62018-12-21 09:23:38 -08001447
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001448 SendGenericPacket();
Amit Hilbuch77938e62018-12-21 09:23:38 -08001449 ASSERT_EQ(1u, transport_.sent_packets_.size());
1450 const RtpPacketReceived& packet = transport_.sent_packets_[0];
1451 std::string rid;
1452 ASSERT_TRUE(packet.GetExtension<RtpStreamId>(&rid));
1453 EXPECT_EQ(kRid, rid);
1454 rid = kNoRid;
Steve Anton2bac7da2019-07-21 15:04:21 -04001455 EXPECT_FALSE(packet.HasExtension<RepairedRtpStreamId>());
Amit Hilbuch77938e62018-12-21 09:23:38 -08001456
1457 uint16_t packet_id = packet.SequenceNumber();
1458 rtp_sender_->ReSendPacket(packet_id);
1459 ASSERT_EQ(2u, transport_.sent_packets_.size());
1460 const RtpPacketReceived& rtx_packet = transport_.sent_packets_[1];
1461 ASSERT_TRUE(rtx_packet.GetExtension<RepairedRtpStreamId>(&rid));
1462 EXPECT_EQ(kRid, rid);
1463 EXPECT_FALSE(rtx_packet.HasExtension<RtpStreamId>());
1464}
1465
Steve Anton2bac7da2019-07-21 15:04:21 -04001466TEST_P(RtpSenderTestWithoutPacer, MidAndRidNotIncludedOnSentPacketsAfterAck) {
1467 const char kMid[] = "mid";
1468 const char kRid[] = "f";
1469
1470 EnableMidSending(kMid);
1471 EnableRidSending(kRid);
1472
1473 // This first packet should include both MID and RID.
1474 auto first_built_packet = SendGenericPacket();
1475
1476 rtp_sender_->OnReceivedAckOnSsrc(first_built_packet->SequenceNumber());
1477
1478 // The second packet should include neither since an ack was received.
1479 SendGenericPacket();
1480
1481 ASSERT_EQ(2u, transport_.sent_packets_.size());
1482
1483 const RtpPacketReceived& first_packet = transport_.sent_packets_[0];
1484 std::string mid, rid;
1485 ASSERT_TRUE(first_packet.GetExtension<RtpMid>(&mid));
1486 EXPECT_EQ(kMid, mid);
1487 ASSERT_TRUE(first_packet.GetExtension<RtpStreamId>(&rid));
1488 EXPECT_EQ(kRid, rid);
1489
1490 const RtpPacketReceived& second_packet = transport_.sent_packets_[1];
1491 EXPECT_FALSE(second_packet.HasExtension<RtpMid>());
1492 EXPECT_FALSE(second_packet.HasExtension<RtpStreamId>());
1493}
1494
1495// Test that the first RTX packet includes both MID and RRID even if the packet
1496// being retransmitted did not have MID or RID. The MID and RID are needed on
1497// the first packets for a given SSRC, and RTX packets are sent on a separate
1498// SSRC.
1499TEST_P(RtpSenderTestWithoutPacer, MidAndRidIncludedOnFirstRtxPacket) {
1500 const char kMid[] = "mid";
1501 const char kRid[] = "f";
1502
1503 EnableRtx();
1504 EnableMidSending(kMid);
1505 EnableRidSending(kRid);
1506
1507 // This first packet will include both MID and RID.
1508 auto first_built_packet = SendGenericPacket();
1509 rtp_sender_->OnReceivedAckOnSsrc(first_built_packet->SequenceNumber());
1510
1511 // The second packet will include neither since an ack was received.
1512 auto second_built_packet = SendGenericPacket();
1513
1514 // The first RTX packet should include MID and RRID.
1515 ASSERT_LT(0,
1516 rtp_sender_->ReSendPacket(second_built_packet->SequenceNumber()));
1517
1518 ASSERT_EQ(3u, transport_.sent_packets_.size());
1519
1520 const RtpPacketReceived& rtx_packet = transport_.sent_packets_[2];
1521 std::string mid, rrid;
1522 ASSERT_TRUE(rtx_packet.GetExtension<RtpMid>(&mid));
1523 EXPECT_EQ(kMid, mid);
1524 ASSERT_TRUE(rtx_packet.GetExtension<RepairedRtpStreamId>(&rrid));
1525 EXPECT_EQ(kRid, rrid);
1526}
1527
1528// Test that the RTX packets sent after receving an ACK on the RTX SSRC does
1529// not include either MID or RRID even if the packet being retransmitted did
1530// had a MID or RID.
1531TEST_P(RtpSenderTestWithoutPacer, MidAndRidNotIncludedOnRtxPacketsAfterAck) {
1532 const char kMid[] = "mid";
1533 const char kRid[] = "f";
1534
1535 EnableRtx();
1536 EnableMidSending(kMid);
1537 EnableRidSending(kRid);
1538
1539 // This first packet will include both MID and RID.
1540 auto first_built_packet = SendGenericPacket();
1541 rtp_sender_->OnReceivedAckOnSsrc(first_built_packet->SequenceNumber());
1542
1543 // The second packet will include neither since an ack was received.
1544 auto second_built_packet = SendGenericPacket();
1545
1546 // The first RTX packet will include MID and RRID.
1547 ASSERT_LT(0,
1548 rtp_sender_->ReSendPacket(second_built_packet->SequenceNumber()));
1549
1550 ASSERT_EQ(3u, transport_.sent_packets_.size());
1551 const RtpPacketReceived& first_rtx_packet = transport_.sent_packets_[2];
1552
1553 rtp_sender_->OnReceivedAckOnRtxSsrc(first_rtx_packet.SequenceNumber());
1554
1555 // The second and third RTX packets should not include MID nor RRID.
1556 ASSERT_LT(0, rtp_sender_->ReSendPacket(first_built_packet->SequenceNumber()));
1557 ASSERT_LT(0,
1558 rtp_sender_->ReSendPacket(second_built_packet->SequenceNumber()));
1559
1560 ASSERT_EQ(5u, transport_.sent_packets_.size());
1561
1562 const RtpPacketReceived& second_rtx_packet = transport_.sent_packets_[3];
1563 EXPECT_FALSE(second_rtx_packet.HasExtension<RtpMid>());
1564 EXPECT_FALSE(second_rtx_packet.HasExtension<RepairedRtpStreamId>());
1565
1566 const RtpPacketReceived& third_rtx_packet = transport_.sent_packets_[4];
1567 EXPECT_FALSE(third_rtx_packet.HasExtension<RtpMid>());
1568 EXPECT_FALSE(third_rtx_packet.HasExtension<RepairedRtpStreamId>());
1569}
1570
1571// Test that if the RtpState indicates an ACK has been received on that SSRC
1572// then neither the MID nor RID header extensions will be sent.
1573TEST_P(RtpSenderTestWithoutPacer,
1574 MidAndRidNotIncludedOnSentPacketsAfterRtpStateRestored) {
1575 const char kMid[] = "mid";
1576 const char kRid[] = "f";
1577
1578 EnableMidSending(kMid);
1579 EnableRidSending(kRid);
1580
1581 RtpState state = rtp_sender_->GetRtpState();
1582 EXPECT_FALSE(state.ssrc_has_acked);
1583 state.ssrc_has_acked = true;
1584 rtp_sender_->SetRtpState(state);
1585
1586 SendGenericPacket();
1587
1588 ASSERT_EQ(1u, transport_.sent_packets_.size());
1589 const RtpPacketReceived& packet = transport_.sent_packets_[0];
1590 EXPECT_FALSE(packet.HasExtension<RtpMid>());
1591 EXPECT_FALSE(packet.HasExtension<RtpStreamId>());
1592}
1593
1594// Test that if the RTX RtpState indicates an ACK has been received on that
1595// RTX SSRC then neither the MID nor RRID header extensions will be sent on
1596// RTX packets.
1597TEST_P(RtpSenderTestWithoutPacer,
1598 MidAndRridNotIncludedOnRtxPacketsAfterRtpStateRestored) {
1599 const char kMid[] = "mid";
1600 const char kRid[] = "f";
1601
1602 EnableRtx();
1603 EnableMidSending(kMid);
1604 EnableRidSending(kRid);
1605
1606 RtpState rtx_state = rtp_sender_->GetRtxRtpState();
1607 EXPECT_FALSE(rtx_state.ssrc_has_acked);
1608 rtx_state.ssrc_has_acked = true;
1609 rtp_sender_->SetRtxRtpState(rtx_state);
1610
1611 auto built_packet = SendGenericPacket();
1612 ASSERT_LT(0, rtp_sender_->ReSendPacket(built_packet->SequenceNumber()));
1613
1614 ASSERT_EQ(2u, transport_.sent_packets_.size());
1615 const RtpPacketReceived& rtx_packet = transport_.sent_packets_[1];
1616 EXPECT_FALSE(rtx_packet.HasExtension<RtpMid>());
1617 EXPECT_FALSE(rtx_packet.HasExtension<RepairedRtpStreamId>());
1618}
1619
minyue3a407ee2017-04-03 01:10:33 -07001620TEST_P(RtpSenderTest, FecOverheadRate) {
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001621 constexpr uint32_t kTimestamp = 1234;
1622 constexpr int kMediaPayloadType = 127;
brandtr81eab612017-01-24 04:06:09 -08001623 constexpr int kFlexfecPayloadType = 118;
brandtr81eab612017-01-24 04:06:09 -08001624 const std::vector<RtpExtension> kNoRtpExtensions;
erikvarga27883732017-05-17 05:08:38 -07001625 const std::vector<RtpExtensionSize> kNoRtpExtensionSizes;
Erik Språng4580ca22019-07-04 10:38:43 +02001626 FlexfecSender flexfec_sender(kFlexfecPayloadType, kFlexFecSsrc, kSsrc, kNoMid,
1627 kNoRtpExtensions, kNoRtpExtensionSizes,
brandtr48d21a22017-05-30 02:32:12 -07001628 nullptr /* rtp_state */, &fake_clock_);
brandtr81eab612017-01-24 04:06:09 -08001629
1630 // Reset |rtp_sender_| to use FlexFEC.
Erik Språng4580ca22019-07-04 10:38:43 +02001631 RtpRtcp::Configuration config;
1632 config.clock = &fake_clock_;
1633 config.outgoing_transport = &transport_;
1634 config.paced_sender = &mock_paced_sender_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001635 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001636 config.flexfec_sender = &flexfec_sender;
Erik Språng4580ca22019-07-04 10:38:43 +02001637 config.event_log = &mock_rtc_event_log_;
1638 config.send_packet_observer = &send_packet_observer_;
1639 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001640 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng4580ca22019-07-04 10:38:43 +02001641
brandtr81eab612017-01-24 04:06:09 -08001642 rtp_sender_->SetSequenceNumber(kSeqNum);
brandtr81eab612017-01-24 04:06:09 -08001643
Niels Möller5fe95102019-03-04 16:49:25 +01001644 PlayoutDelayOracle playout_delay_oracle;
Elad Alona0e99432019-05-24 13:50:56 +02001645 RTPSenderVideo rtp_sender_video(
1646 &fake_clock_, rtp_sender_.get(), &flexfec_sender, &playout_delay_oracle,
Andrei Dumitru09872732019-08-29 12:00:29 -07001647 nullptr, false, false, false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001648 rtp_sender_video.RegisterPayloadType(kMediaPayloadType, "GENERIC",
1649 /*raw_payload=*/false);
brandtr81eab612017-01-24 04:06:09 -08001650 // Parameters selected to generate a single FEC packet per media packet.
1651 FecProtectionParams params;
1652 params.fec_rate = 15;
1653 params.max_fec_frames = 1;
1654 params.fec_mask_type = kFecMaskRandom;
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001655 rtp_sender_video.SetFecParameters(params, params);
brandtr81eab612017-01-24 04:06:09 -08001656
1657 constexpr size_t kNumMediaPackets = 10;
1658 constexpr size_t kNumFecPackets = kNumMediaPackets;
1659 constexpr int64_t kTimeBetweenPacketsMs = 10;
Erik Språngf6468d22019-07-05 16:53:43 +02001660 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
1661 .Times(kNumMediaPackets + kNumFecPackets);
brandtr81eab612017-01-24 04:06:09 -08001662 for (size_t i = 0; i < kNumMediaPackets; ++i) {
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001663 RTPVideoHeader video_header;
1664
1665 EXPECT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001666 VideoFrameType::kVideoFrameKey, kMediaPayloadType, kTimestamp,
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001667 fake_clock_.TimeInMilliseconds(), kPayloadData, sizeof(kPayloadData),
1668 nullptr, &video_header, kDefaultExpectedRetransmissionTimeMs));
1669
brandtr81eab612017-01-24 04:06:09 -08001670 fake_clock_.AdvanceTimeMilliseconds(kTimeBetweenPacketsMs);
1671 }
1672 constexpr size_t kRtpHeaderLength = 12;
1673 constexpr size_t kFlexfecHeaderLength = 20;
1674 constexpr size_t kGenericCodecHeaderLength = 1;
1675 constexpr size_t kPayloadLength = sizeof(kPayloadData);
1676 constexpr size_t kPacketLength = kRtpHeaderLength + kFlexfecHeaderLength +
1677 kGenericCodecHeaderLength + kPayloadLength;
1678 EXPECT_NEAR(kNumFecPackets * kPacketLength * 8 /
1679 (kNumFecPackets * kTimeBetweenPacketsMs / 1000.0f),
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001680 rtp_sender_video.FecOverheadRate(), 500);
brandtr81eab612017-01-24 04:06:09 -08001681}
1682
minyue3a407ee2017-04-03 01:10:33 -07001683TEST_P(RtpSenderTest, BitrateCallbacks) {
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001684 class TestCallback : public BitrateStatisticsObserver {
1685 public:
sprangcd349d92016-07-13 09:11:28 -07001686 TestCallback()
1687 : BitrateStatisticsObserver(),
1688 num_calls_(0),
1689 ssrc_(0),
1690 total_bitrate_(0),
1691 retransmit_bitrate_(0) {}
Danil Chapovalovdd7e2842018-03-09 15:37:03 +00001692 ~TestCallback() override = default;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001693
sprangcd349d92016-07-13 09:11:28 -07001694 void Notify(uint32_t total_bitrate,
1695 uint32_t retransmit_bitrate,
kjellander@webrtc.org14665ff2015-03-04 12:58:35 +00001696 uint32_t ssrc) override {
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001697 ++num_calls_;
1698 ssrc_ = ssrc;
sprangcd349d92016-07-13 09:11:28 -07001699 total_bitrate_ = total_bitrate;
1700 retransmit_bitrate_ = retransmit_bitrate;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001701 }
1702
1703 uint32_t num_calls_;
1704 uint32_t ssrc_;
sprangcd349d92016-07-13 09:11:28 -07001705 uint32_t total_bitrate_;
1706 uint32_t retransmit_bitrate_;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001707 } callback;
Erik Språng4580ca22019-07-04 10:38:43 +02001708
1709 RtpRtcp::Configuration config;
1710 config.clock = &fake_clock_;
1711 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001712 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001713 config.send_bitrate_observer = &callback;
1714 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001715 rtp_sender_ = std::make_unique<RTPSender>(config);
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001716
Niels Möller5fe95102019-03-04 16:49:25 +01001717 PlayoutDelayOracle playout_delay_oracle;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001718 RTPSenderVideo rtp_sender_video(&fake_clock_, rtp_sender_.get(), nullptr,
Elad Alona0e99432019-05-24 13:50:56 +02001719 &playout_delay_oracle, nullptr, false, false,
Andrei Dumitru09872732019-08-29 12:00:29 -07001720 false, FieldTrialBasedConfig());
Niels Möller59ab1cf2019-02-06 22:48:11 +01001721 const char payload_name[] = "GENERIC";
1722 const uint8_t payload_type = 127;
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001723 rtp_sender_video.RegisterPayloadType(payload_type, payload_name,
1724 /*raw_payload=*/false);
Niels Möller59ab1cf2019-02-06 22:48:11 +01001725
sprangcd349d92016-07-13 09:11:28 -07001726 // Simulate kNumPackets sent with kPacketInterval ms intervals, with the
1727 // number of packets selected so that we fill (but don't overflow) the one
1728 // second averaging window.
1729 const uint32_t kWindowSizeMs = 1000;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001730 const uint32_t kPacketInterval = 20;
sprangcd349d92016-07-13 09:11:28 -07001731 const uint32_t kNumPackets =
1732 (kWindowSizeMs - kPacketInterval) / kPacketInterval;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001733 // Overhead = 12 bytes RTP header + 1 byte generic header.
1734 const uint32_t kPacketOverhead = 13;
1735
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001736 uint8_t payload[] = {47, 11, 32, 93, 89};
1737 rtp_sender_->SetStorePacketsStatus(true, 1);
1738 uint32_t ssrc = rtp_sender_->SSRC();
1739
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001740 // Initial process call so we get a new time window.
1741 rtp_sender_->ProcessBitrate();
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001742
1743 // Send a few frames.
Sami Kalliomäki426a80c2018-08-08 11:37:59 +02001744 RTPVideoHeader video_header;
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001745 for (uint32_t i = 0; i < kNumPackets; ++i) {
Niels Möller59ab1cf2019-02-06 22:48:11 +01001746 ASSERT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001747 VideoFrameType::kVideoFrameKey, payload_type, 1234, 4321, payload,
1748 sizeof(payload), nullptr, &video_header,
1749 kDefaultExpectedRetransmissionTimeMs));
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001750 fake_clock_.AdvanceTimeMilliseconds(kPacketInterval);
1751 }
1752
1753 rtp_sender_->ProcessBitrate();
1754
stefan@webrtc.org0bae1fa2014-11-05 14:05:29 +00001755 // We get one call for every stats updated, thus two calls since both the
1756 // stream stats and the retransmit stats are updated once.
1757 EXPECT_EQ(2u, callback.num_calls_);
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001758 EXPECT_EQ(ssrc, callback.ssrc_);
sprangcd349d92016-07-13 09:11:28 -07001759 const uint32_t kTotalPacketSize = kPacketOverhead + sizeof(payload);
1760 // Bitrate measured over delta between last and first timestamp, plus one.
1761 const uint32_t kExpectedWindowMs = kNumPackets * kPacketInterval + 1;
1762 const uint32_t kExpectedBitsAccumulated = kTotalPacketSize * kNumPackets * 8;
1763 const uint32_t kExpectedRateBps =
1764 (kExpectedBitsAccumulated * 1000 + (kExpectedWindowMs / 2)) /
1765 kExpectedWindowMs;
1766 EXPECT_EQ(kExpectedRateBps, callback.total_bitrate_);
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001767
andresp@webrtc.orgd11bec42014-07-08 14:32:58 +00001768 rtp_sender_.reset();
sprang@webrtc.org6811b6e2013-12-13 09:46:59 +00001769}
1770
minyue3a407ee2017-04-03 01:10:33 -07001771TEST_P(RtpSenderTestWithoutPacer, StreamDataCountersCallbacks) {
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001772 class TestCallback : public StreamDataCountersCallback {
1773 public:
danilchap162abd32015-12-10 02:39:40 -08001774 TestCallback() : StreamDataCountersCallback(), ssrc_(0), counters_() {}
Danil Chapovalovdd7e2842018-03-09 15:37:03 +00001775 ~TestCallback() override = default;
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001776
kjellander@webrtc.org14665ff2015-03-04 12:58:35 +00001777 void DataCountersUpdated(const StreamDataCounters& counters,
1778 uint32_t ssrc) override {
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001779 ssrc_ = ssrc;
1780 counters_ = counters;
1781 }
1782
1783 uint32_t ssrc_;
1784 StreamDataCounters counters_;
asapersson@webrtc.org44149392015-02-04 08:34:47 +00001785
1786 void MatchPacketCounter(const RtpPacketCounter& expected,
1787 const RtpPacketCounter& actual) {
1788 EXPECT_EQ(expected.payload_bytes, actual.payload_bytes);
1789 EXPECT_EQ(expected.header_bytes, actual.header_bytes);
1790 EXPECT_EQ(expected.padding_bytes, actual.padding_bytes);
1791 EXPECT_EQ(expected.packets, actual.packets);
1792 }
1793
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001794 void Matches(uint32_t ssrc, const StreamDataCounters& counters) {
1795 EXPECT_EQ(ssrc, ssrc_);
asapersson@webrtc.org44149392015-02-04 08:34:47 +00001796 MatchPacketCounter(counters.transmitted, counters_.transmitted);
1797 MatchPacketCounter(counters.retransmitted, counters_.retransmitted);
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001798 EXPECT_EQ(counters.fec.packets, counters_.fec.packets);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001799 }
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001800 } callback;
1801
1802 const uint8_t kRedPayloadType = 96;
1803 const uint8_t kUlpfecPayloadType = 97;
Niels Möller8a40edd2019-01-24 18:04:44 +01001804 const char payload_name[] = "GENERIC";
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001805 const uint8_t payload_type = 127;
Niels Möller5fe95102019-03-04 16:49:25 +01001806 PlayoutDelayOracle playout_delay_oracle;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001807 RTPSenderVideo rtp_sender_video(&fake_clock_, rtp_sender_.get(), nullptr,
Elad Alona0e99432019-05-24 13:50:56 +02001808 &playout_delay_oracle, nullptr, false, false,
Andrei Dumitru09872732019-08-29 12:00:29 -07001809 false, FieldTrialBasedConfig());
Mirta Dvornicicfe68daa2019-05-23 13:21:12 +02001810 rtp_sender_video.RegisterPayloadType(payload_type, payload_name,
1811 /*raw_payload=*/false);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001812 uint8_t payload[] = {47, 11, 32, 93, 89};
1813 rtp_sender_->SetStorePacketsStatus(true, 1);
1814 uint32_t ssrc = rtp_sender_->SSRC();
1815
1816 rtp_sender_->RegisterRtpStatisticsCallback(&callback);
1817
1818 // Send a frame.
Sami Kalliomäki426a80c2018-08-08 11:37:59 +02001819 RTPVideoHeader video_header;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001820 ASSERT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001821 VideoFrameType::kVideoFrameKey, payload_type, 1234, 4321, payload,
1822 sizeof(payload), nullptr, &video_header,
1823 kDefaultExpectedRetransmissionTimeMs));
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001824 StreamDataCounters expected;
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001825 expected.transmitted.payload_bytes = 6;
1826 expected.transmitted.header_bytes = 12;
1827 expected.transmitted.padding_bytes = 0;
1828 expected.transmitted.packets = 1;
1829 expected.retransmitted.payload_bytes = 0;
1830 expected.retransmitted.header_bytes = 0;
1831 expected.retransmitted.padding_bytes = 0;
1832 expected.retransmitted.packets = 0;
1833 expected.fec.packets = 0;
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001834 callback.Matches(ssrc, expected);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001835
1836 // Retransmit a frame.
1837 uint16_t seqno = rtp_sender_->SequenceNumber() - 1;
Erik Språnga12b1d62018-03-14 12:39:24 +01001838 rtp_sender_->ReSendPacket(seqno);
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001839 expected.transmitted.payload_bytes = 12;
1840 expected.transmitted.header_bytes = 24;
1841 expected.transmitted.packets = 2;
1842 expected.retransmitted.payload_bytes = 6;
1843 expected.retransmitted.header_bytes = 12;
1844 expected.retransmitted.padding_bytes = 0;
1845 expected.retransmitted.packets = 1;
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001846 callback.Matches(ssrc, expected);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001847
1848 // Send padding.
Erik Språng4208a132019-08-26 08:58:45 +02001849 GenerateAndSendPadding(kMaxPaddingSize);
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001850 expected.transmitted.payload_bytes = 12;
1851 expected.transmitted.header_bytes = 36;
1852 expected.transmitted.padding_bytes = kMaxPaddingSize;
1853 expected.transmitted.packets = 3;
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001854 callback.Matches(ssrc, expected);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001855
brandtrf1bb4762016-11-07 03:05:06 -08001856 // Send ULPFEC.
Niels Möller59ab1cf2019-02-06 22:48:11 +01001857 rtp_sender_video.SetUlpfecConfig(kRedPayloadType, kUlpfecPayloadType);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001858 FecProtectionParams fec_params;
1859 fec_params.fec_mask_type = kFecMaskRandom;
1860 fec_params.fec_rate = 1;
1861 fec_params.max_fec_frames = 1;
Niels Möller59ab1cf2019-02-06 22:48:11 +01001862 rtp_sender_video.SetFecParameters(fec_params, fec_params);
1863 ASSERT_TRUE(rtp_sender_video.SendVideo(
Niels Möller8f7ce222019-03-21 15:43:58 +01001864 VideoFrameType::kVideoFrameDelta, payload_type, 1234, 4321, payload,
1865 sizeof(payload), nullptr, &video_header,
1866 kDefaultExpectedRetransmissionTimeMs));
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001867 expected.transmitted.payload_bytes = 40;
1868 expected.transmitted.header_bytes = 60;
1869 expected.transmitted.packets = 5;
1870 expected.fec.packets = 1;
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001871 callback.Matches(ssrc, expected);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001872
sprang867fb522015-08-03 04:38:41 -07001873 rtp_sender_->RegisterRtpStatisticsCallback(nullptr);
sprang@webrtc.orgebad7652013-12-05 14:29:02 +00001874}
1875
minyue3a407ee2017-04-03 01:10:33 -07001876TEST_P(RtpSenderTestWithoutPacer, BytesReportedCorrectly) {
Niels Möller59ab1cf2019-02-06 22:48:11 +01001877 // XXX const char* kPayloadName = "GENERIC";
pbos@webrtc.org72491b92014-07-10 16:24:54 +00001878 const uint8_t kPayloadType = 127;
Shao Changbine62202f2015-04-21 20:24:50 +08001879 rtp_sender_->SetRtxPayloadType(kPayloadType - 1, kPayloadType);
pbos@webrtc.org0b0c2412015-01-13 14:15:15 +00001880 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
pbos@webrtc.org72491b92014-07-10 16:24:54 +00001881
Niels Möller59ab1cf2019-02-06 22:48:11 +01001882 SendGenericPacket();
pbos@webrtc.org2f4b14e2014-07-15 15:25:39 +00001883 // Will send 2 full-size padding packets.
Erik Språng4208a132019-08-26 08:58:45 +02001884 GenerateAndSendPadding(1);
1885 GenerateAndSendPadding(1);
pbos@webrtc.org72491b92014-07-10 16:24:54 +00001886
pbos@webrtc.org2f4b14e2014-07-15 15:25:39 +00001887 StreamDataCounters rtp_stats;
1888 StreamDataCounters rtx_stats;
1889 rtp_sender_->GetDataCounters(&rtp_stats, &rtx_stats);
pbos@webrtc.org72491b92014-07-10 16:24:54 +00001890
Niels Möller59ab1cf2019-02-06 22:48:11 +01001891 // Payload
asapersson@webrtc.orgd08d3892014-12-16 12:03:11 +00001892 EXPECT_GT(rtp_stats.first_packet_time_ms, -1);
Niels Möller59ab1cf2019-02-06 22:48:11 +01001893 EXPECT_EQ(rtp_stats.transmitted.payload_bytes, sizeof(kPayloadData));
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001894 EXPECT_EQ(rtp_stats.transmitted.header_bytes, 12u);
1895 EXPECT_EQ(rtp_stats.transmitted.padding_bytes, 0u);
1896 EXPECT_EQ(rtx_stats.transmitted.payload_bytes, 0u);
1897 EXPECT_EQ(rtx_stats.transmitted.header_bytes, 24u);
1898 EXPECT_EQ(rtx_stats.transmitted.padding_bytes, 2 * kMaxPaddingSize);
pbos@webrtc.org2f4b14e2014-07-15 15:25:39 +00001899
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001900 EXPECT_EQ(rtp_stats.transmitted.TotalBytes(),
danilchap162abd32015-12-10 02:39:40 -08001901 rtp_stats.transmitted.payload_bytes +
1902 rtp_stats.transmitted.header_bytes +
1903 rtp_stats.transmitted.padding_bytes);
asapersson@webrtc.orgcfd82df2015-01-22 09:39:59 +00001904 EXPECT_EQ(rtx_stats.transmitted.TotalBytes(),
danilchap162abd32015-12-10 02:39:40 -08001905 rtx_stats.transmitted.payload_bytes +
1906 rtx_stats.transmitted.header_bytes +
1907 rtx_stats.transmitted.padding_bytes);
asapersson@webrtc.org97d04892014-12-09 09:47:53 +00001908
danilchap162abd32015-12-10 02:39:40 -08001909 EXPECT_EQ(
1910 transport_.total_bytes_sent_,
1911 rtp_stats.transmitted.TotalBytes() + rtx_stats.transmitted.TotalBytes());
pbos@webrtc.org72491b92014-07-10 16:24:54 +00001912}
guoweis@webrtc.org45362892015-03-04 22:55:15 +00001913
minyue3a407ee2017-04-03 01:10:33 -07001914TEST_P(RtpSenderTestWithoutPacer, RespectsNackBitrateLimit) {
sprang38778b02015-09-29 09:48:22 -07001915 const int32_t kPacketSize = 1400;
1916 const int32_t kNumPackets = 30;
1917
sprangcd349d92016-07-13 09:11:28 -07001918 retransmission_rate_limiter_.SetMaxRate(kPacketSize * kNumPackets * 8);
1919
sprang38778b02015-09-29 09:48:22 -07001920 rtp_sender_->SetStorePacketsStatus(true, kNumPackets);
sprang38778b02015-09-29 09:48:22 -07001921 const uint16_t kStartSequenceNumber = rtp_sender_->SequenceNumber();
Danil Chapovalov2800d742016-08-26 18:48:46 +02001922 std::vector<uint16_t> sequence_numbers;
sprang38778b02015-09-29 09:48:22 -07001923 for (int32_t i = 0; i < kNumPackets; ++i) {
1924 sequence_numbers.push_back(kStartSequenceNumber + i);
1925 fake_clock_.AdvanceTimeMilliseconds(1);
1926 SendPacket(fake_clock_.TimeInMilliseconds(), kPacketSize);
1927 }
danilchap12ba1862016-10-26 02:41:55 -07001928 EXPECT_EQ(kNumPackets, transport_.packets_sent());
sprang38778b02015-09-29 09:48:22 -07001929
1930 fake_clock_.AdvanceTimeMilliseconds(1000 - kNumPackets);
1931
1932 // Resending should work - brings the bandwidth up to the limit.
1933 // NACK bitrate is capped to the same bitrate as the encoder, since the max
1934 // protection overhead is 50% (see MediaOptimization::SetTargetRates).
Danil Chapovalov2800d742016-08-26 18:48:46 +02001935 rtp_sender_->OnReceivedNack(sequence_numbers, 0);
danilchap12ba1862016-10-26 02:41:55 -07001936 EXPECT_EQ(kNumPackets * 2, transport_.packets_sent());
sprang38778b02015-09-29 09:48:22 -07001937
sprangcd349d92016-07-13 09:11:28 -07001938 // Must be at least 5ms in between retransmission attempts.
1939 fake_clock_.AdvanceTimeMilliseconds(5);
1940
sprang38778b02015-09-29 09:48:22 -07001941 // Resending should not work, bandwidth exceeded.
Danil Chapovalov2800d742016-08-26 18:48:46 +02001942 rtp_sender_->OnReceivedNack(sequence_numbers, 0);
danilchap12ba1862016-10-26 02:41:55 -07001943 EXPECT_EQ(kNumPackets * 2, transport_.packets_sent());
sprang38778b02015-09-29 09:48:22 -07001944}
1945
minyue3a407ee2017-04-03 01:10:33 -07001946TEST_P(RtpSenderTest, OnOverheadChanged) {
michaelt4da30442016-11-17 01:38:43 -08001947 MockOverheadObserver mock_overhead_observer;
Erik Språng4580ca22019-07-04 10:38:43 +02001948 RtpRtcp::Configuration config;
1949 config.clock = &fake_clock_;
1950 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001951 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001952 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
1953 config.overhead_observer = &mock_overhead_observer;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001954 rtp_sender_ = std::make_unique<RTPSender>(config);
michaelt4da30442016-11-17 01:38:43 -08001955
michaelt4da30442016-11-17 01:38:43 -08001956 // RTP overhead is 12B.
nisse284542b2017-01-10 08:58:32 -08001957 EXPECT_CALL(mock_overhead_observer, OnOverheadChanged(12)).Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001958 SendGenericPacket();
michaelt4da30442016-11-17 01:38:43 -08001959
1960 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionTransmissionTimeOffset,
1961 kTransmissionTimeOffsetExtensionId);
1962
1963 // TransmissionTimeOffset extension has a size of 8B.
nisse284542b2017-01-10 08:58:32 -08001964 // 12B + 8B = 20B
1965 EXPECT_CALL(mock_overhead_observer, OnOverheadChanged(20)).Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001966 SendGenericPacket();
michaelt4da30442016-11-17 01:38:43 -08001967}
1968
minyue3a407ee2017-04-03 01:10:33 -07001969TEST_P(RtpSenderTest, DoesNotUpdateOverheadOnEqualSize) {
michaelt4da30442016-11-17 01:38:43 -08001970 MockOverheadObserver mock_overhead_observer;
Erik Språng4580ca22019-07-04 10:38:43 +02001971 RtpRtcp::Configuration config;
1972 config.clock = &fake_clock_;
1973 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02001974 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02001975 config.retransmission_rate_limiter = &retransmission_rate_limiter_;
1976 config.overhead_observer = &mock_overhead_observer;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02001977 rtp_sender_ = std::make_unique<RTPSender>(config);
michaelt4da30442016-11-17 01:38:43 -08001978
1979 EXPECT_CALL(mock_overhead_observer, OnOverheadChanged(_)).Times(1);
Niels Möllere7b9e6b2019-02-06 18:23:44 +01001980 SendGenericPacket();
1981 SendGenericPacket();
michaelt4da30442016-11-17 01:38:43 -08001982}
1983
Erik Språng9c771c22019-06-17 16:31:53 +02001984TEST_P(RtpSenderTest, TrySendPacketMatchesVideo) {
1985 std::unique_ptr<RtpPacketToSend> packet =
1986 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
1987 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
1988
1989 // Verify not sent with wrong SSRC.
1990 packet->SetSsrc(kSsrc + 1);
1991 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
1992
1993 // Verify sent with correct SSRC.
1994 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
1995 packet->SetSsrc(kSsrc);
1996 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
1997 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
1998}
1999
2000TEST_P(RtpSenderTest, TrySendPacketMatchesAudio) {
2001 std::unique_ptr<RtpPacketToSend> packet =
2002 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2003 packet->set_packet_type(RtpPacketToSend::Type::kAudio);
2004
2005 // Verify not sent with wrong SSRC.
2006 packet->SetSsrc(kSsrc + 1);
2007 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2008
2009 // Verify sent with correct SSRC.
2010 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2011 packet->SetSsrc(kSsrc);
2012 packet->set_packet_type(RtpPacketToSend::Type::kAudio);
2013 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2014}
2015
2016TEST_P(RtpSenderTest, TrySendPacketMatchesRetransmissions) {
2017 std::unique_ptr<RtpPacketToSend> packet =
2018 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2019 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2020
2021 // Verify not sent with wrong SSRC.
2022 packet->SetSsrc(kSsrc + 1);
2023 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2024
2025 // Verify sent with correct SSRC (non-RTX).
2026 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2027 packet->SetSsrc(kSsrc);
2028 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2029 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2030
2031 // RTX retransmission.
2032 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2033 packet->SetSsrc(kRtxSsrc);
2034 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2035 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2036}
2037
2038TEST_P(RtpSenderTest, TrySendPacketMatchesPadding) {
2039 std::unique_ptr<RtpPacketToSend> packet =
2040 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2041 packet->set_packet_type(RtpPacketToSend::Type::kPadding);
2042
2043 // Verify not sent with wrong SSRC.
2044 packet->SetSsrc(kSsrc + 1);
2045 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2046
2047 // Verify sent with correct SSRC (non-RTX).
2048 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2049 packet->SetSsrc(kSsrc);
2050 packet->set_packet_type(RtpPacketToSend::Type::kPadding);
2051 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2052
2053 // RTX padding.
2054 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2055 packet->SetSsrc(kRtxSsrc);
2056 packet->set_packet_type(RtpPacketToSend::Type::kPadding);
2057 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2058}
2059
2060TEST_P(RtpSenderTest, TrySendPacketMatchesFlexfec) {
2061 std::unique_ptr<RtpPacketToSend> packet =
2062 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2063 packet->set_packet_type(RtpPacketToSend::Type::kForwardErrorCorrection);
2064
2065 // Verify not sent with wrong SSRC.
2066 packet->SetSsrc(kSsrc + 1);
2067 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2068
2069 // Verify sent with correct SSRC.
2070 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2071 packet->SetSsrc(kFlexFecSsrc);
2072 packet->set_packet_type(RtpPacketToSend::Type::kForwardErrorCorrection);
2073 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2074}
2075
2076TEST_P(RtpSenderTest, TrySendPacketMatchesUlpfec) {
2077 std::unique_ptr<RtpPacketToSend> packet =
2078 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2079 packet->set_packet_type(RtpPacketToSend::Type::kForwardErrorCorrection);
2080
2081 // Verify not sent with wrong SSRC.
2082 packet->SetSsrc(kSsrc + 1);
2083 EXPECT_FALSE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2084
2085 // Verify sent with correct SSRC.
2086 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2087 packet->SetSsrc(kSsrc);
2088 packet->set_packet_type(RtpPacketToSend::Type::kForwardErrorCorrection);
2089 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2090}
2091
2092TEST_P(RtpSenderTest, TrySendPacketHandlesRetransmissionHistory) {
2093 rtp_sender_->SetStorePacketsStatus(true, 10);
2094
2095 // Build a media packet and send it.
2096 std::unique_ptr<RtpPacketToSend> packet =
2097 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2098 const uint16_t media_sequence_number = packet->SequenceNumber();
2099 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2100 packet->set_allow_retransmission(true);
2101 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2102
2103 // Simulate retransmission request.
2104 fake_clock_.AdvanceTimeMilliseconds(30);
2105 EXPECT_GT(rtp_sender_->ReSendPacket(media_sequence_number), 0);
2106
2107 // Packet already pending, retransmission not allowed.
2108 fake_clock_.AdvanceTimeMilliseconds(30);
2109 EXPECT_EQ(rtp_sender_->ReSendPacket(media_sequence_number), 0);
2110
2111 // Packet exiting pacer, mark as not longer pending.
2112 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2113 EXPECT_NE(packet->SequenceNumber(), media_sequence_number);
2114 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2115 packet->SetSsrc(kRtxSsrc);
2116 packet->set_retransmitted_sequence_number(media_sequence_number);
2117 packet->set_allow_retransmission(false);
2118 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2119
2120 // Retransmissions allowed again.
2121 fake_clock_.AdvanceTimeMilliseconds(30);
2122 EXPECT_GT(rtp_sender_->ReSendPacket(media_sequence_number), 0);
2123
2124 // Retransmission of RTX packet should not be allowed.
2125 EXPECT_EQ(rtp_sender_->ReSendPacket(packet->SequenceNumber()), 0);
2126}
2127
2128TEST_P(RtpSenderTest, TrySendPacketUpdatesExtensions) {
2129 ASSERT_EQ(rtp_sender_->RegisterRtpHeaderExtension(
2130 kRtpExtensionTransmissionTimeOffset,
2131 kTransmissionTimeOffsetExtensionId),
2132 0);
2133 ASSERT_EQ(rtp_sender_->RegisterRtpHeaderExtension(
2134 kRtpExtensionAbsoluteSendTime, kAbsoluteSendTimeExtensionId),
2135 0);
2136 ASSERT_EQ(rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionVideoTiming,
2137 kVideoTimingExtensionId),
2138 0);
2139
2140 std::unique_ptr<RtpPacketToSend> packet =
2141 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2142 packet->set_packetization_finish_time_ms(fake_clock_.TimeInMilliseconds());
2143
2144 const int32_t kDiffMs = 10;
2145 fake_clock_.AdvanceTimeMilliseconds(kDiffMs);
2146
2147 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2148 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2149
2150 const RtpPacketReceived& received_packet = transport_.last_sent_packet();
2151
2152 EXPECT_EQ(received_packet.GetExtension<TransmissionOffset>(), kDiffMs * 90);
2153
2154 EXPECT_EQ(received_packet.GetExtension<AbsoluteSendTime>(),
2155 AbsoluteSendTime::MsTo24Bits(fake_clock_.TimeInMilliseconds()));
2156
2157 VideoSendTiming timing;
2158 EXPECT_TRUE(received_packet.GetExtension<VideoTimingExtension>(&timing));
2159 EXPECT_EQ(timing.pacer_exit_delta_ms, kDiffMs);
2160}
2161
2162TEST_P(RtpSenderTest, TrySendPacketSetsPacketOptions) {
2163 const uint16_t kPacketId = 42;
2164 ASSERT_EQ(rtp_sender_->RegisterRtpHeaderExtension(
2165 kRtpExtensionTransportSequenceNumber,
2166 kTransportSequenceNumberExtensionId),
2167 0);
2168 std::unique_ptr<RtpPacketToSend> packet =
2169 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2170 packet->SetExtension<TransportSequenceNumber>(kPacketId);
2171
2172 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2173 EXPECT_CALL(send_packet_observer_, OnSendPacket);
2174 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2175
2176 EXPECT_EQ(transport_.last_options_.packet_id, kPacketId);
2177 EXPECT_TRUE(transport_.last_options_.included_in_allocation);
2178 EXPECT_TRUE(transport_.last_options_.included_in_feedback);
2179 EXPECT_FALSE(transport_.last_options_.is_retransmit);
2180
2181 // Send another packet as retransmission, verify options are populated.
2182 packet = BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2183 packet->SetExtension<TransportSequenceNumber>(kPacketId + 1);
2184 packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2185 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2186 EXPECT_TRUE(transport_.last_options_.is_retransmit);
2187}
2188
2189TEST_P(RtpSenderTest, TrySendPacketUpdatesStats) {
2190 const size_t kPayloadSize = 1000;
2191
2192 StrictMock<MockSendSideDelayObserver> send_side_delay_observer;
Erik Språng4580ca22019-07-04 10:38:43 +02002193
2194 RtpRtcp::Configuration config;
2195 config.clock = &fake_clock_;
2196 config.outgoing_transport = &transport_;
Erik Språng54d5d2c2019-08-20 17:22:36 +02002197 config.local_media_ssrc = kSsrc;
Erik Språng4580ca22019-07-04 10:38:43 +02002198 config.rtx_send_ssrc = kRtxSsrc;
2199 config.flexfec_sender = &flexfec_sender_;
2200 config.send_side_delay_observer = &send_side_delay_observer;
2201 config.event_log = &mock_rtc_event_log_;
2202 config.send_packet_observer = &send_packet_observer_;
Mirko Bonadei317a1f02019-09-17 17:06:18 +02002203 rtp_sender_ = std::make_unique<RTPSender>(config);
Erik Språng9c771c22019-06-17 16:31:53 +02002204 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
2205 kRtpExtensionTransportSequenceNumber,
2206 kTransportSequenceNumberExtensionId));
2207
2208 const int64_t capture_time_ms = fake_clock_.TimeInMilliseconds();
2209
2210 std::unique_ptr<RtpPacketToSend> video_packet =
2211 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2212 video_packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2213 video_packet->SetPayloadSize(kPayloadSize);
2214 video_packet->SetExtension<TransportSequenceNumber>(1);
2215
2216 std::unique_ptr<RtpPacketToSend> rtx_packet =
2217 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2218 rtx_packet->SetSsrc(kRtxSsrc);
2219 rtx_packet->set_packet_type(RtpPacketToSend::Type::kRetransmission);
2220 rtx_packet->SetPayloadSize(kPayloadSize);
2221 rtx_packet->SetExtension<TransportSequenceNumber>(2);
2222
2223 std::unique_ptr<RtpPacketToSend> fec_packet =
2224 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2225 fec_packet->SetSsrc(kFlexFecSsrc);
2226 fec_packet->set_packet_type(RtpPacketToSend::Type::kForwardErrorCorrection);
2227 fec_packet->SetPayloadSize(kPayloadSize);
2228 fec_packet->SetExtension<TransportSequenceNumber>(3);
2229
2230 const int64_t kDiffMs = 25;
2231 fake_clock_.AdvanceTimeMilliseconds(kDiffMs);
2232
2233 EXPECT_CALL(send_side_delay_observer,
2234 SendSideDelayUpdated(kDiffMs, kDiffMs, kDiffMs, kSsrc));
2235 EXPECT_CALL(
2236 send_side_delay_observer,
2237 SendSideDelayUpdated(kDiffMs, kDiffMs, 2 * kDiffMs, kFlexFecSsrc));
2238
2239 EXPECT_CALL(send_packet_observer_, OnSendPacket(1, capture_time_ms, kSsrc));
2240 EXPECT_TRUE(
2241 rtp_sender_->TrySendPacket(video_packet.get(), PacedPacketInfo()));
2242
2243 // Send packet observer not called for padding/retransmissions.
2244 EXPECT_CALL(send_packet_observer_, OnSendPacket(2, _, _)).Times(0);
2245 EXPECT_TRUE(rtp_sender_->TrySendPacket(rtx_packet.get(), PacedPacketInfo()));
2246
2247 EXPECT_CALL(send_packet_observer_,
2248 OnSendPacket(3, capture_time_ms, kFlexFecSsrc));
2249 EXPECT_TRUE(rtp_sender_->TrySendPacket(fec_packet.get(), PacedPacketInfo()));
2250
2251 StreamDataCounters rtp_stats;
2252 StreamDataCounters rtx_stats;
2253 rtp_sender_->GetDataCounters(&rtp_stats, &rtx_stats);
2254 EXPECT_EQ(rtp_stats.transmitted.packets, 2u);
2255 EXPECT_EQ(rtp_stats.fec.packets, 1u);
2256 EXPECT_EQ(rtx_stats.retransmitted.packets, 1u);
2257}
2258
Erik Språng478cb462019-06-26 15:49:27 +02002259TEST_P(RtpSenderTest, GeneratePaddingResendsOldPacketsWithRtx) {
Mirko Bonadeia7e3bce2019-07-12 17:35:56 +00002260 // Min requested size in order to use RTX payload.
2261 const size_t kMinPaddingSize = 50;
2262
Erik Språng478cb462019-06-26 15:49:27 +02002263 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
2264 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
2265 rtp_sender_->SetStorePacketsStatus(true, 1);
2266
Erik Språng0f6191d2019-07-15 20:33:40 +02002267 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
2268 kRtpExtensionTransmissionTimeOffset,
2269 kTransmissionTimeOffsetExtensionId));
2270 ASSERT_EQ(
2271 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
2272 kAbsoluteSendTimeExtensionId));
2273 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
2274 kRtpExtensionTransportSequenceNumber,
2275 kTransportSequenceNumberExtensionId));
2276
Erik Språng478cb462019-06-26 15:49:27 +02002277 const size_t kPayloadPacketSize = 1234;
2278 std::unique_ptr<RtpPacketToSend> packet =
2279 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2280 packet->set_allow_retransmission(true);
2281 packet->SetPayloadSize(kPayloadPacketSize);
2282 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2283
2284 // Send a dummy video packet so it ends up in the packet history.
Erik Språng0f6191d2019-07-15 20:33:40 +02002285 EXPECT_CALL(send_packet_observer_, OnSendPacket).Times(1);
Erik Språng478cb462019-06-26 15:49:27 +02002286 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2287
2288 // Generated padding has large enough budget that the video packet should be
2289 // retransmitted as padding.
Erik Språngf6468d22019-07-05 16:53:43 +02002290 std::vector<std::unique_ptr<RtpPacketToSend>> generated_packets =
Mirko Bonadeia7e3bce2019-07-12 17:35:56 +00002291 rtp_sender_->GeneratePadding(kMinPaddingSize);
Erik Språngf6468d22019-07-05 16:53:43 +02002292 ASSERT_EQ(generated_packets.size(), 1u);
2293 auto& padding_packet = generated_packets.front();
2294 EXPECT_EQ(padding_packet->packet_type(), RtpPacketToSend::Type::kPadding);
2295 EXPECT_EQ(padding_packet->Ssrc(), kRtxSsrc);
2296 EXPECT_EQ(padding_packet->payload_size(),
2297 kPayloadPacketSize + kRtxHeaderSize);
Erik Språng0f6191d2019-07-15 20:33:40 +02002298 EXPECT_TRUE(padding_packet->IsExtensionReserved<TransportSequenceNumber>());
2299 EXPECT_TRUE(padding_packet->IsExtensionReserved<AbsoluteSendTime>());
2300 EXPECT_TRUE(padding_packet->IsExtensionReserved<TransmissionOffset>());
2301
2302 // Verify all header extensions are received.
2303 EXPECT_TRUE(
2304 rtp_sender_->TrySendPacket(padding_packet.get(), PacedPacketInfo()));
2305 webrtc::RTPHeader rtp_header;
2306 transport_.last_sent_packet().GetHeader(&rtp_header);
2307 EXPECT_TRUE(rtp_header.extension.hasAbsoluteSendTime);
2308 EXPECT_TRUE(rtp_header.extension.hasTransmissionTimeOffset);
2309 EXPECT_TRUE(rtp_header.extension.hasTransportSequenceNumber);
Erik Språng478cb462019-06-26 15:49:27 +02002310
2311 // Not enough budged for payload padding, use plain padding instead.
Mirko Bonadeia7e3bce2019-07-12 17:35:56 +00002312 const size_t kPaddingBytesRequested = kMinPaddingSize - 1;
Erik Språngf6468d22019-07-05 16:53:43 +02002313
2314 size_t padding_bytes_generated = 0;
2315 generated_packets = rtp_sender_->GeneratePadding(kPaddingBytesRequested);
Mirko Bonadeia7e3bce2019-07-12 17:35:56 +00002316 EXPECT_EQ(generated_packets.size(), 1u);
Erik Språngf6468d22019-07-05 16:53:43 +02002317 for (auto& packet : generated_packets) {
2318 EXPECT_EQ(packet->packet_type(), RtpPacketToSend::Type::kPadding);
2319 EXPECT_EQ(packet->Ssrc(), kRtxSsrc);
2320 EXPECT_EQ(packet->payload_size(), 0u);
2321 EXPECT_GT(packet->padding_size(), 0u);
2322 padding_bytes_generated += packet->padding_size();
Erik Språng0f6191d2019-07-15 20:33:40 +02002323
2324 EXPECT_TRUE(packet->IsExtensionReserved<TransportSequenceNumber>());
2325 EXPECT_TRUE(packet->IsExtensionReserved<AbsoluteSendTime>());
2326 EXPECT_TRUE(packet->IsExtensionReserved<TransmissionOffset>());
2327
2328 // Verify all header extensions are received.
2329 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2330 webrtc::RTPHeader rtp_header;
2331 transport_.last_sent_packet().GetHeader(&rtp_header);
2332 EXPECT_TRUE(rtp_header.extension.hasAbsoluteSendTime);
2333 EXPECT_TRUE(rtp_header.extension.hasTransmissionTimeOffset);
2334 EXPECT_TRUE(rtp_header.extension.hasTransportSequenceNumber);
Erik Språngf6468d22019-07-05 16:53:43 +02002335 }
2336
Mirko Bonadeia7e3bce2019-07-12 17:35:56 +00002337 EXPECT_EQ(padding_bytes_generated, kMaxPaddingSize);
Erik Språng478cb462019-06-26 15:49:27 +02002338}
2339
2340TEST_P(RtpSenderTest, GeneratePaddingCreatesPurePaddingWithoutRtx) {
2341 rtp_sender_->SetStorePacketsStatus(true, 1);
Erik Språng0f6191d2019-07-15 20:33:40 +02002342 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
2343 kRtpExtensionTransmissionTimeOffset,
2344 kTransmissionTimeOffsetExtensionId));
2345 ASSERT_EQ(
2346 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
2347 kAbsoluteSendTimeExtensionId));
2348 ASSERT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension(
2349 kRtpExtensionTransportSequenceNumber,
2350 kTransportSequenceNumberExtensionId));
Erik Språng478cb462019-06-26 15:49:27 +02002351
2352 const size_t kPayloadPacketSize = 1234;
2353 // Send a dummy video packet so it ends up in the packet history. Since we
2354 // are not using RTX, it should never be used as padding.
2355 std::unique_ptr<RtpPacketToSend> packet =
2356 BuildRtpPacket(kPayload, true, 0, fake_clock_.TimeInMilliseconds());
2357 packet->set_allow_retransmission(true);
2358 packet->SetPayloadSize(kPayloadPacketSize);
2359 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
Erik Språng0f6191d2019-07-15 20:33:40 +02002360 EXPECT_CALL(send_packet_observer_, OnSendPacket).Times(1);
Erik Språng478cb462019-06-26 15:49:27 +02002361 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2362
2363 // Payload padding not available without RTX, only generate plain padding on
2364 // the media SSRC.
2365 // Number of padding packets is the requested padding size divided by max
2366 // padding packet size, rounded up. Pure padding packets are always of the
2367 // maximum size.
2368 const size_t kPaddingBytesRequested = kPayloadPacketSize + kRtxHeaderSize;
2369 const size_t kExpectedNumPaddingPackets =
2370 (kPaddingBytesRequested + kMaxPaddingSize - 1) / kMaxPaddingSize;
Erik Språngf6468d22019-07-05 16:53:43 +02002371 size_t padding_bytes_generated = 0;
2372 std::vector<std::unique_ptr<RtpPacketToSend>> padding_packets =
2373 rtp_sender_->GeneratePadding(kPaddingBytesRequested);
2374 EXPECT_EQ(padding_packets.size(), kExpectedNumPaddingPackets);
2375 for (auto& packet : padding_packets) {
2376 EXPECT_EQ(packet->packet_type(), RtpPacketToSend::Type::kPadding);
2377 EXPECT_EQ(packet->Ssrc(), kSsrc);
2378 EXPECT_EQ(packet->payload_size(), 0u);
2379 EXPECT_GT(packet->padding_size(), 0u);
2380 padding_bytes_generated += packet->padding_size();
Erik Språng0f6191d2019-07-15 20:33:40 +02002381 EXPECT_TRUE(packet->IsExtensionReserved<TransportSequenceNumber>());
2382 EXPECT_TRUE(packet->IsExtensionReserved<AbsoluteSendTime>());
2383 EXPECT_TRUE(packet->IsExtensionReserved<TransmissionOffset>());
2384
2385 // Verify all header extensions are received.
2386 EXPECT_TRUE(rtp_sender_->TrySendPacket(packet.get(), PacedPacketInfo()));
2387 webrtc::RTPHeader rtp_header;
2388 transport_.last_sent_packet().GetHeader(&rtp_header);
2389 EXPECT_TRUE(rtp_header.extension.hasAbsoluteSendTime);
2390 EXPECT_TRUE(rtp_header.extension.hasTransmissionTimeOffset);
2391 EXPECT_TRUE(rtp_header.extension.hasTransportSequenceNumber);
Erik Språngf6468d22019-07-05 16:53:43 +02002392 }
2393
2394 EXPECT_EQ(padding_bytes_generated,
2395 kExpectedNumPaddingPackets * kMaxPaddingSize);
Erik Språng478cb462019-06-26 15:49:27 +02002396}
2397
Mirko Bonadei999a72a2019-07-12 17:33:46 +00002398TEST_P(RtpSenderTest, SupportsPadding) {
2399 bool kSendingMediaStats[] = {true, false};
2400 bool kEnableRedundantPayloads[] = {true, false};
2401 RTPExtensionType kBweExtensionTypes[] = {
2402 kRtpExtensionTransportSequenceNumber,
2403 kRtpExtensionTransportSequenceNumber02, kRtpExtensionAbsoluteSendTime,
2404 kRtpExtensionTransmissionTimeOffset};
2405 const int kExtensionsId = 7;
2406
2407 for (bool sending_media : kSendingMediaStats) {
2408 rtp_sender_->SetSendingMediaStatus(sending_media);
2409 for (bool redundant_payloads : kEnableRedundantPayloads) {
2410 int rtx_mode = kRtxRetransmitted;
2411 if (redundant_payloads) {
2412 rtx_mode |= kRtxRedundantPayloads;
2413 }
2414 rtp_sender_->SetRtxStatus(rtx_mode);
2415
2416 for (auto extension_type : kBweExtensionTypes) {
2417 EXPECT_FALSE(rtp_sender_->SupportsPadding());
2418 rtp_sender_->RegisterRtpHeaderExtension(extension_type, kExtensionsId);
2419 if (!sending_media) {
2420 EXPECT_FALSE(rtp_sender_->SupportsPadding());
2421 } else {
2422 EXPECT_TRUE(rtp_sender_->SupportsPadding());
2423 if (redundant_payloads) {
2424 EXPECT_TRUE(rtp_sender_->SupportsRtxPayloadPadding());
2425 } else {
2426 EXPECT_FALSE(rtp_sender_->SupportsRtxPayloadPadding());
2427 }
2428 }
2429 rtp_sender_->DeregisterRtpHeaderExtension(extension_type);
2430 EXPECT_FALSE(rtp_sender_->SupportsPadding());
2431 }
2432 }
2433 }
2434}
2435
Erik Språnga57711c2019-07-24 10:47:20 +02002436TEST_P(RtpSenderTest, SetsCaptureTimeAndPopulatesTransmissionOffset) {
2437 rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionTransmissionTimeOffset,
2438 kTransmissionTimeOffsetExtensionId);
2439
2440 rtp_sender_->SetSendingMediaStatus(true);
2441 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
2442 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
2443 rtp_sender_->SetStorePacketsStatus(true, 10);
2444
2445 const int64_t kMissingCaptureTimeMs = 0;
2446 const uint32_t kTimestampTicksPerMs = 90;
2447 const int64_t kOffsetMs = 10;
2448
Erik Språnga57711c2019-07-24 10:47:20 +02002449 auto packet =
2450 BuildRtpPacket(kPayload, kMarkerBit, fake_clock_.TimeInMilliseconds(),
2451 kMissingCaptureTimeMs);
2452 packet->set_packet_type(RtpPacketToSend::Type::kVideo);
2453 packet->ReserveExtension<TransmissionOffset>();
2454 packet->AllocatePayload(sizeof(kPayloadData));
2455
2456 std::unique_ptr<RtpPacketToSend> packet_to_pace;
2457 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
2458 .WillOnce([&](std::unique_ptr<RtpPacketToSend> packet) {
2459 EXPECT_GT(packet->capture_time_ms(), 0);
2460 packet_to_pace = std::move(packet);
2461 });
2462
Erik Språng70768f42019-08-27 18:16:26 +02002463 packet->set_allow_retransmission(true);
2464 EXPECT_TRUE(rtp_sender_->SendToNetwork(std::move(packet)));
Erik Språnga57711c2019-07-24 10:47:20 +02002465
2466 fake_clock_.AdvanceTimeMilliseconds(kOffsetMs);
2467
2468 rtp_sender_->TrySendPacket(packet_to_pace.get(), PacedPacketInfo());
Erik Språnga57711c2019-07-24 10:47:20 +02002469
2470 EXPECT_EQ(1, transport_.packets_sent());
2471 absl::optional<int32_t> transmission_time_extension =
2472 transport_.sent_packets_.back().GetExtension<TransmissionOffset>();
2473 ASSERT_TRUE(transmission_time_extension.has_value());
2474 EXPECT_EQ(*transmission_time_extension, kOffsetMs * kTimestampTicksPerMs);
2475
2476 // Retransmit packet. The RTX packet should get the same capture time as the
2477 // original packet, so offset is delta from original packet to now.
2478 fake_clock_.AdvanceTimeMilliseconds(kOffsetMs);
2479
Erik Språnga57711c2019-07-24 10:47:20 +02002480 std::unique_ptr<RtpPacketToSend> rtx_packet_to_pace;
2481 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
2482 .WillOnce([&](std::unique_ptr<RtpPacketToSend> packet) {
2483 EXPECT_GT(packet->capture_time_ms(), 0);
2484 rtx_packet_to_pace = std::move(packet);
2485 });
2486
2487 EXPECT_GT(rtp_sender_->ReSendPacket(kSeqNum), 0);
2488 rtp_sender_->TrySendPacket(rtx_packet_to_pace.get(), PacedPacketInfo());
Erik Språnga57711c2019-07-24 10:47:20 +02002489
2490 EXPECT_EQ(2, transport_.packets_sent());
2491 transmission_time_extension =
2492 transport_.sent_packets_.back().GetExtension<TransmissionOffset>();
2493 ASSERT_TRUE(transmission_time_extension.has_value());
2494 EXPECT_EQ(*transmission_time_extension, 2 * kOffsetMs * kTimestampTicksPerMs);
2495}
2496
Erik Språng6cacef22019-07-24 14:15:51 +02002497TEST_P(RtpSenderTestWithoutPacer, ClearHistoryOnSsrcChange) {
2498 const int64_t kRtt = 10;
2499
2500 rtp_sender_->SetSendingMediaStatus(true);
2501 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
2502 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
2503 rtp_sender_->SetStorePacketsStatus(true, 10);
2504 rtp_sender_->SetRtt(kRtt);
Erik Språng6cacef22019-07-24 14:15:51 +02002505
2506 // Send a packet and record its sequence numbers.
2507 SendGenericPacket();
2508 ASSERT_EQ(1u, transport_.sent_packets_.size());
2509 const uint16_t packet_seqence_number =
2510 transport_.sent_packets_.back().SequenceNumber();
2511
2512 // Advance time and make sure it can be retransmitted, even if we try to set
2513 // the ssrc the what it already is.
2514 rtp_sender_->SetSSRC(kSsrc);
2515 fake_clock_.AdvanceTimeMilliseconds(kRtt);
2516 EXPECT_GT(rtp_sender_->ReSendPacket(packet_seqence_number), 0);
2517
2518 // Change the SSRC, then move the time and try to retransmit again. The old
2519 // packet should now be gone.
2520 rtp_sender_->SetSSRC(kSsrc + 1);
2521 fake_clock_.AdvanceTimeMilliseconds(kRtt);
2522 EXPECT_EQ(rtp_sender_->ReSendPacket(packet_seqence_number), 0);
2523}
2524
2525TEST_P(RtpSenderTestWithoutPacer, ClearHistoryOnSequenceNumberCange) {
2526 const int64_t kRtt = 10;
2527
2528 rtp_sender_->SetSendingMediaStatus(true);
2529 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
2530 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
2531 rtp_sender_->SetStorePacketsStatus(true, 10);
2532 rtp_sender_->SetRtt(kRtt);
2533
2534 // Send a packet and record its sequence numbers.
2535 SendGenericPacket();
2536 ASSERT_EQ(1u, transport_.sent_packets_.size());
2537 const uint16_t packet_seqence_number =
2538 transport_.sent_packets_.back().SequenceNumber();
2539
2540 // Advance time and make sure it can be retransmitted, even if we try to set
2541 // the ssrc the what it already is.
2542 rtp_sender_->SetSequenceNumber(rtp_sender_->SequenceNumber());
2543 fake_clock_.AdvanceTimeMilliseconds(kRtt);
2544 EXPECT_GT(rtp_sender_->ReSendPacket(packet_seqence_number), 0);
2545
2546 // Change the sequence number, then move the time and try to retransmit again.
2547 // The old packet should now be gone.
2548 rtp_sender_->SetSequenceNumber(rtp_sender_->SequenceNumber() - 1);
2549 fake_clock_.AdvanceTimeMilliseconds(kRtt);
2550 EXPECT_EQ(rtp_sender_->ReSendPacket(packet_seqence_number), 0);
2551}
2552
Erik Språng60ffc312019-07-30 22:03:49 +02002553TEST_P(RtpSenderTest, IgnoresNackAfterDisablingMedia) {
2554 const int64_t kRtt = 10;
2555
2556 rtp_sender_->SetSendingMediaStatus(true);
2557 rtp_sender_->SetRtxStatus(kRtxRetransmitted | kRtxRedundantPayloads);
2558 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
2559 rtp_sender_->SetStorePacketsStatus(true, 10);
2560 rtp_sender_->SetRtt(kRtt);
2561
2562 // Send a packet so it is in the packet history.
Erik Språng60ffc312019-07-30 22:03:49 +02002563 std::unique_ptr<RtpPacketToSend> packet_to_pace;
2564 EXPECT_CALL(mock_paced_sender_, EnqueuePacket)
2565 .WillOnce([&](std::unique_ptr<RtpPacketToSend> packet) {
2566 packet_to_pace = std::move(packet);
2567 });
2568
2569 SendGenericPacket();
2570 rtp_sender_->TrySendPacket(packet_to_pace.get(), PacedPacketInfo());
2571
2572 ASSERT_EQ(1u, transport_.sent_packets_.size());
2573
2574 // Disable media sending and try to retransmit the packet, it should fail.
2575 rtp_sender_->SetSendingMediaStatus(false);
2576 fake_clock_.AdvanceTimeMilliseconds(kRtt);
2577 EXPECT_LT(rtp_sender_->ReSendPacket(kSeqNum), 0);
Erik Språng60ffc312019-07-30 22:03:49 +02002578}
2579
Mirko Bonadeic84f6612019-01-31 12:20:57 +01002580INSTANTIATE_TEST_SUITE_P(WithAndWithoutOverhead,
2581 RtpSenderTest,
Erik Språngf5815fa2019-08-21 14:27:31 +02002582 ::testing::Values(TestConfig{false},
2583 TestConfig{true}));
Erik Språngf6468d22019-07-05 16:53:43 +02002584
Mirko Bonadeic84f6612019-01-31 12:20:57 +01002585INSTANTIATE_TEST_SUITE_P(WithAndWithoutOverhead,
2586 RtpSenderTestWithoutPacer,
Erik Språngf5815fa2019-08-21 14:27:31 +02002587 ::testing::Values(TestConfig{false},
2588 TestConfig{true}));
Niels Möllera34d7762019-02-01 14:13:29 +01002589
solenberg@webrtc.orgc0352d52013-05-20 20:55:07 +00002590} // namespace webrtc