blob: 0c165212408ff3fd978e814ae7a44d36aca108b7 [file] [log] [blame]
brandtr76648da2016-10-20 04:54:48 -07001/*
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Jonas Olssona4d87372019-07-05 19:08:33 +020011#include "call/flexfec_receive_stream.h"
12
Yves Gerey3e707812018-11-28 16:47:49 +010013#include <cstdint>
eladalon7ab7fd62017-07-03 06:57:13 -070014#include <memory>
Yves Gerey3e707812018-11-28 16:47:49 +010015#include <vector>
pbosc7c26a02017-01-02 08:42:32 -080016
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020017#include "api/array_view.h"
Yves Gerey3e707812018-11-28 16:47:49 +010018#include "api/call/transport.h"
19#include "api/rtp_headers.h"
Steve Anton10542f22019-01-11 09:11:00 -080020#include "api/rtp_parameters.h"
Yves Gerey3e707812018-11-28 16:47:49 +010021#include "call/flexfec_receive_stream_impl.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020022#include "call/rtp_stream_receiver_controller.h"
Yves Gerey3e707812018-11-28 16:47:49 +010023#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020024#include "modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h"
25#include "modules/rtp_rtcp/mocks/mock_rtcp_rtt_stats.h"
26#include "modules/rtp_rtcp/source/byte_io.h"
27#include "modules/rtp_rtcp/source/rtp_header_extensions.h"
Yves Gerey3e707812018-11-28 16:47:49 +010028#include "modules/rtp_rtcp/source/rtp_packet_received.h"
Niels Möller83830f32022-05-20 09:12:57 +020029#include "rtc_base/thread.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020030#include "test/gmock.h"
31#include "test/gtest.h"
32#include "test/mock_transport.h"
brandtr76648da2016-10-20 04:54:48 -070033
34namespace webrtc {
35
brandtrb29e6522016-12-21 06:37:18 -080036namespace {
37
eladalon7ab7fd62017-07-03 06:57:13 -070038using ::testing::_;
Per K83c357f2023-01-05 12:01:52 +010039using ::testing::Eq;
40using ::testing::Property;
eladalon7ab7fd62017-07-03 06:57:13 -070041
brandtr8313a6f2017-01-13 07:41:19 -080042constexpr uint8_t kFlexfecPlType = 118;
43constexpr uint8_t kFlexfecSsrc[] = {0x00, 0x00, 0x00, 0x01};
44constexpr uint8_t kMediaSsrc[] = {0x00, 0x00, 0x00, 0x02};
45
46FlexfecReceiveStream::Config CreateDefaultConfig(
47 Transport* rtcp_send_transport) {
48 FlexfecReceiveStream::Config config(rtcp_send_transport);
49 config.payload_type = kFlexfecPlType;
Tommi1c1f5402021-06-14 10:54:20 +020050 config.rtp.remote_ssrc = ByteReader<uint32_t>::ReadBigEndian(kFlexfecSsrc);
brandtr8313a6f2017-01-13 07:41:19 -080051 config.protected_media_ssrcs = {
52 ByteReader<uint32_t>::ReadBigEndian(kMediaSsrc)};
53 EXPECT_TRUE(config.IsCompleteAndEnabled());
54 return config;
55}
56
brandtrb29e6522016-12-21 06:37:18 -080057RtpPacketReceived ParsePacket(rtc::ArrayView<const uint8_t> packet) {
58 RtpPacketReceived parsed_packet(nullptr);
59 EXPECT_TRUE(parsed_packet.Parse(packet));
60 return parsed_packet;
61}
62
63} // namespace
64
brandtr8313a6f2017-01-13 07:41:19 -080065TEST(FlexfecReceiveStreamConfigTest, IsCompleteAndEnabled) {
66 MockTransport rtcp_send_transport;
67 FlexfecReceiveStream::Config config(&rtcp_send_transport);
brandtr76648da2016-10-20 04:54:48 -070068
Tommi1c1f5402021-06-14 10:54:20 +020069 config.rtp.local_ssrc = 18374743;
brandtr8313a6f2017-01-13 07:41:19 -080070 config.rtcp_mode = RtcpMode::kCompound;
Danil Chapovalovd0321c52021-09-14 12:58:51 +020071 config.rtp.extensions.emplace_back(TransportSequenceNumber::Uri(), 7);
brandtr8313a6f2017-01-13 07:41:19 -080072 EXPECT_FALSE(config.IsCompleteAndEnabled());
73
74 config.payload_type = 123;
75 EXPECT_FALSE(config.IsCompleteAndEnabled());
76
Tommi1c1f5402021-06-14 10:54:20 +020077 config.rtp.remote_ssrc = 238423838;
brandtr8313a6f2017-01-13 07:41:19 -080078 EXPECT_FALSE(config.IsCompleteAndEnabled());
79
80 config.protected_media_ssrcs.push_back(138989393);
81 EXPECT_TRUE(config.IsCompleteAndEnabled());
82
83 config.protected_media_ssrcs.push_back(33423423);
84 EXPECT_FALSE(config.IsCompleteAndEnabled());
brandtr76648da2016-10-20 04:54:48 -070085}
86
brandtr8313a6f2017-01-13 07:41:19 -080087class FlexfecReceiveStreamTest : public ::testing::Test {
88 protected:
89 FlexfecReceiveStreamTest()
eladalon7ab7fd62017-07-03 06:57:13 -070090 : config_(CreateDefaultConfig(&rtcp_send_transport_)) {
Mirko Bonadei317a1f02019-09-17 17:06:18 +020091 receive_stream_ = std::make_unique<FlexfecReceiveStreamImpl>(
Tommi0377bab2021-05-31 14:26:05 +020092 Clock::GetRealTimeClock(), config_, &recovered_packet_receiver_,
Markus Handelleb61b7f2021-06-22 10:46:48 +020093 &rtt_stats_);
Tommi0377bab2021-05-31 14:26:05 +020094 receive_stream_->RegisterWithTransport(&rtp_stream_receiver_controller_);
eladalon7ab7fd62017-07-03 06:57:13 -070095 }
96
97 ~FlexfecReceiveStreamTest() {
Tommi0377bab2021-05-31 14:26:05 +020098 receive_stream_->UnregisterFromTransport();
eladalon7ab7fd62017-07-03 06:57:13 -070099 }
brandtr76648da2016-10-20 04:54:48 -0700100
Niels Möller83830f32022-05-20 09:12:57 +0200101 rtc::AutoThread main_thread_;
brandtrfa5a3682017-01-17 01:33:54 -0800102 MockTransport rtcp_send_transport_;
brandtr8313a6f2017-01-13 07:41:19 -0800103 FlexfecReceiveStream::Config config_;
104 MockRecoveredPacketReceiver recovered_packet_receiver_;
brandtrfa5a3682017-01-17 01:33:54 -0800105 MockRtcpRttStats rtt_stats_;
nisse0f15f922017-06-21 01:05:22 -0700106 RtpStreamReceiverController rtp_stream_receiver_controller_;
eladalon7ab7fd62017-07-03 06:57:13 -0700107 std::unique_ptr<FlexfecReceiveStreamImpl> receive_stream_;
brandtr8313a6f2017-01-13 07:41:19 -0800108};
109
110TEST_F(FlexfecReceiveStreamTest, ConstructDestruct) {}
111
brandtr76648da2016-10-20 04:54:48 -0700112// Create a FlexFEC packet that protects a single media packet and ensure
113// that the callback is called. Correctness of recovery is checked in the
114// FlexfecReceiver unit tests.
Niels Möller2bf9e732017-08-14 11:26:16 +0200115TEST_F(FlexfecReceiveStreamTest, RecoversPacket) {
brandtr76648da2016-10-20 04:54:48 -0700116 constexpr uint8_t kFlexfecSeqNum[] = {0x00, 0x01};
117 constexpr uint8_t kFlexfecTs[] = {0x00, 0x11, 0x22, 0x33};
brandtr76648da2016-10-20 04:54:48 -0700118 constexpr uint8_t kMediaPlType = 107;
119 constexpr uint8_t kMediaSeqNum[] = {0x00, 0x02};
120 constexpr uint8_t kMediaTs[] = {0xaa, 0xbb, 0xcc, 0xdd};
brandtr76648da2016-10-20 04:54:48 -0700121
122 // This packet mask protects a single media packet, i.e., the FlexFEC payload
123 // is a copy of that media packet. When inserted in the FlexFEC pipeline,
124 // it will thus trivially recover the lost media packet.
125 constexpr uint8_t kKBit0 = 1 << 7;
126 constexpr uint8_t kFlexfecPktMask[] = {kKBit0 | 0x00, 0x01};
127 constexpr uint8_t kPayloadLength[] = {0x00, 0x04};
128 constexpr uint8_t kSsrcCount = 1;
129 constexpr uint8_t kReservedBits = 0x00;
130 constexpr uint8_t kPayloadBits = 0x00;
131 // clang-format off
132 constexpr uint8_t kFlexfecPacket[] = {
133 // RTP header.
134 0x80, kFlexfecPlType, kFlexfecSeqNum[0], kFlexfecSeqNum[1],
135 kFlexfecTs[0], kFlexfecTs[1], kFlexfecTs[2], kFlexfecTs[3],
136 kFlexfecSsrc[0], kFlexfecSsrc[1], kFlexfecSsrc[2], kFlexfecSsrc[3],
137 // FlexFEC header.
138 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1],
139 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3],
140 kSsrcCount, kReservedBits, kReservedBits, kReservedBits,
141 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3],
142 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1],
143 // FEC payload.
144 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits};
145 // clang-format on
brandtr76648da2016-10-20 04:54:48 -0700146
Philipp Hancke0018def2022-07-06 11:29:09 +0200147 EXPECT_CALL(recovered_packet_receiver_,
Per K83c357f2023-01-05 12:01:52 +0100148 OnRecoveredPacket(Property(&RtpPacketReceived::payload_size,
149 Eq(kPayloadLength[1]))));
Niels Möller2bf9e732017-08-14 11:26:16 +0200150
Philipp Hancke0018def2022-07-06 11:29:09 +0200151 receive_stream_->OnRtpPacket(ParsePacket(kFlexfecPacket));
eladalon7ab7fd62017-07-03 06:57:13 -0700152
153 // Tear-down
Philipp Hancke0018def2022-07-06 11:29:09 +0200154 receive_stream_->UnregisterFromTransport();
brandtr76648da2016-10-20 04:54:48 -0700155}
156
157} // namespace webrtc