1 /*
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
11 #include <stdint.h>
12
13 #include "webrtc/base/array_view.h"
14 #include "webrtc/call/flexfec_receive_stream_impl.h"
15 #include "webrtc/modules/pacing/packet_router.h"
16 #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h"
17 #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h"
18 #include "webrtc/modules/rtp_rtcp/mocks/mock_rtcp_rtt_stats.h"
19 #include "webrtc/modules/rtp_rtcp/source/byte_io.h"
20 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h"
21 #include "webrtc/modules/utility/include/mock/mock_process_thread.h"
22 #include "webrtc/test/gmock.h"
23 #include "webrtc/test/gtest.h"
24 #include "webrtc/test/mock_transport.h"
25
26 namespace webrtc {
27
28 namespace {
29
30 constexpr uint8_t kFlexfecPlType = 118;
31 constexpr uint8_t kFlexfecSsrc[] = {0x00, 0x00, 0x00, 0x01};
32 constexpr uint8_t kMediaSsrc[] = {0x00, 0x00, 0x00, 0x02};
33
CreateDefaultConfig(Transport * rtcp_send_transport)34 FlexfecReceiveStream::Config CreateDefaultConfig(
35 Transport* rtcp_send_transport) {
36 FlexfecReceiveStream::Config config(rtcp_send_transport);
37 config.payload_type = kFlexfecPlType;
38 config.remote_ssrc = ByteReader<uint32_t>::ReadBigEndian(kFlexfecSsrc);
39 config.protected_media_ssrcs = {
40 ByteReader<uint32_t>::ReadBigEndian(kMediaSsrc)};
41 EXPECT_TRUE(config.IsCompleteAndEnabled());
42 return config;
43 }
44
ParsePacket(rtc::ArrayView<const uint8_t> packet)45 RtpPacketReceived ParsePacket(rtc::ArrayView<const uint8_t> packet) {
46 RtpPacketReceived parsed_packet(nullptr);
47 EXPECT_TRUE(parsed_packet.Parse(packet));
48 return parsed_packet;
49 }
50
51 } // namespace
52
TEST(FlexfecReceiveStreamConfigTest,IsCompleteAndEnabled)53 TEST(FlexfecReceiveStreamConfigTest, IsCompleteAndEnabled) {
54 MockTransport rtcp_send_transport;
55 FlexfecReceiveStream::Config config(&rtcp_send_transport);
56
57 config.local_ssrc = 18374743;
58 config.rtcp_mode = RtcpMode::kCompound;
59 config.transport_cc = true;
60 config.rtp_header_extensions.emplace_back(TransportSequenceNumber::kUri, 7);
61 EXPECT_FALSE(config.IsCompleteAndEnabled());
62
63 config.payload_type = 123;
64 EXPECT_FALSE(config.IsCompleteAndEnabled());
65
66 config.remote_ssrc = 238423838;
67 EXPECT_FALSE(config.IsCompleteAndEnabled());
68
69 config.protected_media_ssrcs.push_back(138989393);
70 EXPECT_TRUE(config.IsCompleteAndEnabled());
71
72 config.protected_media_ssrcs.push_back(33423423);
73 EXPECT_FALSE(config.IsCompleteAndEnabled());
74 }
75
76 class FlexfecReceiveStreamTest : public ::testing::Test {
77 protected:
FlexfecReceiveStreamTest()78 FlexfecReceiveStreamTest()
79 : config_(CreateDefaultConfig(&rtcp_send_transport_)),
80 receive_stream_(config_,
81 &recovered_packet_receiver_,
82 &rtt_stats_,
83 &process_thread_) {}
84
85 MockTransport rtcp_send_transport_;
86 FlexfecReceiveStream::Config config_;
87 MockRecoveredPacketReceiver recovered_packet_receiver_;
88 MockRtcpRttStats rtt_stats_;
89 MockProcessThread process_thread_;
90
91 FlexfecReceiveStreamImpl receive_stream_;
92 };
93
TEST_F(FlexfecReceiveStreamTest,ConstructDestruct)94 TEST_F(FlexfecReceiveStreamTest, ConstructDestruct) {}
95
TEST_F(FlexfecReceiveStreamTest,StartStop)96 TEST_F(FlexfecReceiveStreamTest, StartStop) {
97 receive_stream_.Start();
98 receive_stream_.Stop();
99 }
100
101 // Create a FlexFEC packet that protects a single media packet and ensure
102 // that the callback is called. Correctness of recovery is checked in the
103 // FlexfecReceiver unit tests.
TEST_F(FlexfecReceiveStreamTest,RecoversPacketWhenStarted)104 TEST_F(FlexfecReceiveStreamTest, RecoversPacketWhenStarted) {
105 constexpr uint8_t kFlexfecSeqNum[] = {0x00, 0x01};
106 constexpr uint8_t kFlexfecTs[] = {0x00, 0x11, 0x22, 0x33};
107 constexpr uint8_t kMediaPlType = 107;
108 constexpr uint8_t kMediaSeqNum[] = {0x00, 0x02};
109 constexpr uint8_t kMediaTs[] = {0xaa, 0xbb, 0xcc, 0xdd};
110
111 // This packet mask protects a single media packet, i.e., the FlexFEC payload
112 // is a copy of that media packet. When inserted in the FlexFEC pipeline,
113 // it will thus trivially recover the lost media packet.
114 constexpr uint8_t kKBit0 = 1 << 7;
115 constexpr uint8_t kFlexfecPktMask[] = {kKBit0 | 0x00, 0x01};
116 constexpr uint8_t kPayloadLength[] = {0x00, 0x04};
117 constexpr uint8_t kSsrcCount = 1;
118 constexpr uint8_t kReservedBits = 0x00;
119 constexpr uint8_t kPayloadBits = 0x00;
120 // clang-format off
121 constexpr uint8_t kFlexfecPacket[] = {
122 // RTP header.
123 0x80, kFlexfecPlType, kFlexfecSeqNum[0], kFlexfecSeqNum[1],
124 kFlexfecTs[0], kFlexfecTs[1], kFlexfecTs[2], kFlexfecTs[3],
125 kFlexfecSsrc[0], kFlexfecSsrc[1], kFlexfecSsrc[2], kFlexfecSsrc[3],
126 // FlexFEC header.
127 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1],
128 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3],
129 kSsrcCount, kReservedBits, kReservedBits, kReservedBits,
130 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3],
131 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1],
132 // FEC payload.
133 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits};
134 // clang-format on
135
136 testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver;
137 FlexfecReceiveStreamImpl receive_stream(config_, &recovered_packet_receiver,
138 &rtt_stats_, &process_thread_);
139
140 // Do not call back before being started.
141 receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
142
143 // Call back after being started.
144 receive_stream.Start();
145 EXPECT_CALL(
146 recovered_packet_receiver,
147 OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1]));
148 receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
149 }
150
151 } // namespace webrtc
152