Moving src/webrtc into src/.
In order to eliminate the WebRTC Subtree mirror in Chromium, WebRTC is moving the content of the src/webrtc directory up to the src/ directory. NOPRESUBMIT=true NOTREECHECKS=true NOTRY=true TBR=tommi@webrtc.org Bug: chromium:611808 Change-Id: Iac59c5b51b950f174119565bac87955a7994bc38 Reviewed-on: https://webrtc-review.googlesource.com/1560 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19845}
This commit is contained in:
committed by
Commit Bot
parent
6674846b4a
commit
bb547203bf
571
modules/rtp_rtcp/source/flexfec_receiver_unittest.cc
Normal file
571
modules/rtp_rtcp/source/flexfec_receiver_unittest.cc
Normal file
@ -0,0 +1,571 @@
|
||||
/*
|
||||
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
||||
*
|
||||
* Use of this source code is governed by a BSD-style license
|
||||
* that can be found in the LICENSE file in the root of the source
|
||||
* tree. An additional intellectual property rights grant can be found
|
||||
* in the file PATENTS. All contributing project authors may
|
||||
* be found in the AUTHORS file in the root of the source tree.
|
||||
*/
|
||||
|
||||
#include <algorithm>
|
||||
#include <memory>
|
||||
|
||||
#include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h"
|
||||
#include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h"
|
||||
#include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h"
|
||||
#include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h"
|
||||
#include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h"
|
||||
#include "webrtc/rtc_base/basictypes.h"
|
||||
#include "webrtc/test/gmock.h"
|
||||
#include "webrtc/test/gtest.h"
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
namespace {
|
||||
|
||||
using ::testing::_;
|
||||
using ::testing::Args;
|
||||
using ::testing::ElementsAreArray;
|
||||
using ::testing::Return;
|
||||
|
||||
using test::fec::FlexfecPacketGenerator;
|
||||
using Packet = ForwardErrorCorrection::Packet;
|
||||
using PacketList = ForwardErrorCorrection::PacketList;
|
||||
|
||||
constexpr size_t kPayloadLength = 500;
|
||||
constexpr uint32_t kFlexfecSsrc = 42984;
|
||||
constexpr uint32_t kMediaSsrc = 8353;
|
||||
|
||||
RtpPacketReceived ParsePacket(const Packet& packet) {
|
||||
RtpPacketReceived parsed_packet;
|
||||
EXPECT_TRUE(parsed_packet.Parse(packet.data, packet.length));
|
||||
return parsed_packet;
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
class FlexfecReceiverForTest : public FlexfecReceiver {
|
||||
public:
|
||||
FlexfecReceiverForTest(uint32_t ssrc,
|
||||
uint32_t protected_media_ssrc,
|
||||
RecoveredPacketReceiver* recovered_packet_receiver)
|
||||
: FlexfecReceiver(ssrc, protected_media_ssrc, recovered_packet_receiver) {
|
||||
}
|
||||
// Expose methods for tests.
|
||||
using FlexfecReceiver::AddReceivedPacket;
|
||||
using FlexfecReceiver::ProcessReceivedPackets;
|
||||
};
|
||||
|
||||
class FlexfecReceiverTest : public ::testing::Test {
|
||||
protected:
|
||||
FlexfecReceiverTest()
|
||||
: receiver_(kFlexfecSsrc, kMediaSsrc, &recovered_packet_receiver_),
|
||||
erasure_code_(
|
||||
ForwardErrorCorrection::CreateFlexfec(kFlexfecSsrc, kMediaSsrc)),
|
||||
packet_generator_(kMediaSsrc, kFlexfecSsrc) {}
|
||||
|
||||
// Generates |num_media_packets| corresponding to a single frame.
|
||||
void PacketizeFrame(size_t num_media_packets,
|
||||
size_t frame_offset,
|
||||
PacketList* media_packets);
|
||||
|
||||
// Generates |num_fec_packets| FEC packets, given |media_packets|.
|
||||
std::list<Packet*> EncodeFec(const PacketList& media_packets,
|
||||
size_t num_fec_packets);
|
||||
|
||||
FlexfecReceiverForTest receiver_;
|
||||
std::unique_ptr<ForwardErrorCorrection> erasure_code_;
|
||||
|
||||
FlexfecPacketGenerator packet_generator_;
|
||||
testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver_;
|
||||
};
|
||||
|
||||
void FlexfecReceiverTest::PacketizeFrame(size_t num_media_packets,
|
||||
size_t frame_offset,
|
||||
PacketList* media_packets) {
|
||||
packet_generator_.NewFrame(num_media_packets);
|
||||
for (size_t i = 0; i < num_media_packets; ++i) {
|
||||
std::unique_ptr<Packet> next_packet(
|
||||
packet_generator_.NextPacket(frame_offset + i, kPayloadLength));
|
||||
media_packets->push_back(std::move(next_packet));
|
||||
}
|
||||
}
|
||||
|
||||
std::list<Packet*> FlexfecReceiverTest::EncodeFec(
|
||||
const PacketList& media_packets,
|
||||
size_t num_fec_packets) {
|
||||
const uint8_t protection_factor =
|
||||
num_fec_packets * 255 / media_packets.size();
|
||||
constexpr int kNumImportantPackets = 0;
|
||||
constexpr bool kUseUnequalProtection = false;
|
||||
constexpr FecMaskType kFecMaskType = kFecMaskRandom;
|
||||
std::list<Packet*> fec_packets;
|
||||
EXPECT_EQ(0, erasure_code_->EncodeFec(
|
||||
media_packets, protection_factor, kNumImportantPackets,
|
||||
kUseUnequalProtection, kFecMaskType, &fec_packets));
|
||||
EXPECT_EQ(num_fec_packets, fec_packets.size());
|
||||
return fec_packets;
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, ReceivesMediaPacket) {
|
||||
packet_generator_.NewFrame(1);
|
||||
std::unique_ptr<Packet> media_packet(
|
||||
packet_generator_.NextPacket(0, kPayloadLength));
|
||||
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, ReceivesMediaAndFecPackets) {
|
||||
const size_t kNumMediaPackets = 1;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
const auto& media_packet = media_packets.front();
|
||||
auto fec_packet = packet_generator_.BuildFlexfecPacket(*fec_packets.front());
|
||||
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, FailsOnTruncatedFecPacket) {
|
||||
const size_t kNumMediaPackets = 1;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
const auto& media_packet = media_packets.front();
|
||||
// Simulate truncated FlexFEC payload.
|
||||
fec_packets.front()->length = 1;
|
||||
auto fec_packet = packet_generator_.BuildFlexfecPacket(*fec_packets.front());
|
||||
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet)));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, FailsOnUnknownMediaSsrc) {
|
||||
const size_t kNumMediaPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
auto& media_packet = media_packets.front();
|
||||
// Corrupt the SSRC.
|
||||
media_packet->data[8] = 0;
|
||||
media_packet->data[9] = 1;
|
||||
media_packet->data[10] = 2;
|
||||
media_packet->data[11] = 3;
|
||||
|
||||
EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, FailsOnUnknownFecSsrc) {
|
||||
const size_t kNumMediaPackets = 1;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
const auto& media_packet = media_packets.front();
|
||||
auto fec_packet = packet_generator_.BuildFlexfecPacket(*fec_packets.front());
|
||||
// Corrupt the SSRC.
|
||||
fec_packet->data[8] = 4;
|
||||
fec_packet->data[9] = 5;
|
||||
fec_packet->data[10] = 6;
|
||||
fec_packet->data[11] = 7;
|
||||
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet)));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, ReceivesMultiplePackets) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive all media packets.
|
||||
for (const auto& media_packet : media_packets) {
|
||||
EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
}
|
||||
|
||||
// Receive FEC packet.
|
||||
auto fec_packet = fec_packets.front();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(*fec_packet);
|
||||
EXPECT_TRUE(
|
||||
receiver_.AddReceivedPacket(ParsePacket(*packet_with_rtp_header)));
|
||||
EXPECT_TRUE(receiver_.ProcessReceivedPackets());
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, RecoversFromSingleMediaLoss) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive first media packet but drop second.
|
||||
auto media_it = media_packets.begin();
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Receive FEC packet and ensure recovery of lost media packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
media_it++;
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, RecoversFromDoubleMediaLoss) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 2;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Drop both media packets.
|
||||
|
||||
// Receive first FEC packet and recover first lost media packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
auto media_it = media_packets.begin();
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
|
||||
// Receive second FEC packet and recover second lost media packet.
|
||||
fec_it++;
|
||||
packet_with_rtp_header = packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
media_it++;
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, DoesNotRecoverFromMediaAndFecLoss) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive first media packet.
|
||||
auto media_it = media_packets.begin();
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Drop second media packet and FEC packet. Do not expect call back.
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, DoesNotCallbackTwice) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive first media packet but drop second.
|
||||
auto media_it = media_packets.begin();
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Receive FEC packet and ensure recovery of lost media packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
media_it++;
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
|
||||
// Receive the FEC packet again, but do not call back.
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
|
||||
// Receive the first media packet again, but do not call back.
|
||||
media_it = media_packets.begin();
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Receive the second media packet again (the one recovered above),
|
||||
// but do not call back again.
|
||||
media_it++;
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
}
|
||||
|
||||
// Here we are implicitly assuming packet masks that are suitable for
|
||||
// this type of 50% correlated loss. If we are changing our precomputed
|
||||
// packet masks, this test might need to be updated.
|
||||
TEST_F(FlexfecReceiverTest, RecoversFrom50PercentLoss) {
|
||||
const size_t kNumFecPackets = 5;
|
||||
const size_t kNumFrames = 2 * kNumFecPackets;
|
||||
const size_t kNumMediaPacketsPerFrame = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
for (size_t i = 0; i < kNumFrames; ++i) {
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, i, &media_packets);
|
||||
}
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Drop every second media packet.
|
||||
auto media_it = media_packets.begin();
|
||||
while (media_it != media_packets.end()) {
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
++media_it;
|
||||
if (media_it == media_packets.end()) {
|
||||
break;
|
||||
}
|
||||
++media_it;
|
||||
}
|
||||
|
||||
// Receive all FEC packets.
|
||||
media_it = media_packets.begin();
|
||||
for (const auto& fec_packet : fec_packets) {
|
||||
std::unique_ptr<Packet> fec_packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(*fec_packet);
|
||||
++media_it;
|
||||
if (media_it == media_packets.end()) {
|
||||
break;
|
||||
}
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(Args<0, 1>(
|
||||
ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*fec_packet_with_rtp_header));
|
||||
++media_it;
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, DelayedFecPacketDoesHelp) {
|
||||
// These values need to be updated if the underlying erasure code
|
||||
// implementation changes.
|
||||
const size_t kNumFrames = 48;
|
||||
const size_t kNumMediaPacketsPerFrame = 1;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, 0, &media_packets);
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, 1, &media_packets);
|
||||
// Protect two first frames.
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
for (size_t i = 2; i < kNumFrames; ++i) {
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, i, &media_packets);
|
||||
}
|
||||
|
||||
// Drop first media packet and delay FEC packet.
|
||||
auto media_it = media_packets.begin();
|
||||
++media_it;
|
||||
|
||||
// Receive all other media packets.
|
||||
while (media_it != media_packets.end()) {
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
++media_it;
|
||||
}
|
||||
|
||||
// Receive FEC packet and recover first media packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
media_it = media_packets.begin();
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, TooDelayedFecPacketDoesNotHelp) {
|
||||
// These values need to be updated if the underlying erasure code
|
||||
// implementation changes.
|
||||
const size_t kNumFrames = 49;
|
||||
const size_t kNumMediaPacketsPerFrame = 1;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, 0, &media_packets);
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, 1, &media_packets);
|
||||
// Protect two first frames.
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
for (size_t i = 2; i < kNumFrames; ++i) {
|
||||
PacketizeFrame(kNumMediaPacketsPerFrame, i, &media_packets);
|
||||
}
|
||||
|
||||
// Drop first media packet and delay FEC packet.
|
||||
auto media_it = media_packets.begin();
|
||||
++media_it;
|
||||
|
||||
// Receive all other media packets.
|
||||
while (media_it != media_packets.end()) {
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
++media_it;
|
||||
}
|
||||
|
||||
// Receive FEC packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
|
||||
// Do not expect a call back.
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, RecoversWithMediaPacketsOutOfOrder) {
|
||||
const size_t kNumMediaPackets = 6;
|
||||
const size_t kNumFecPackets = 2;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Lose two media packets, and receive the others out of order.
|
||||
auto media_it = media_packets.begin();
|
||||
auto media_packet0 = media_it++;
|
||||
auto media_packet1 = media_it++;
|
||||
auto media_packet2 = media_it++;
|
||||
auto media_packet3 = media_it++;
|
||||
auto media_packet4 = media_it++;
|
||||
auto media_packet5 = media_it++;
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_packet5));
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_packet2));
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_packet3));
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_packet0));
|
||||
|
||||
// Expect to recover lost media packets.
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_packet1)->length))
|
||||
.With(Args<0, 1>(
|
||||
ElementsAreArray((*media_packet1)->data, (*media_packet1)->length)));
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_packet4)->length))
|
||||
.With(Args<0, 1>(
|
||||
ElementsAreArray((*media_packet4)->data, (*media_packet4)->length)));
|
||||
|
||||
// Add FEC packets.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header;
|
||||
while (fec_it != fec_packets.end()) {
|
||||
packet_with_rtp_header = packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
++fec_it;
|
||||
}
|
||||
}
|
||||
|
||||
// Recovered media packets may be fed back into the FlexfecReceiver by the
|
||||
// callback. This test ensures the idempotency of such a situation.
|
||||
TEST_F(FlexfecReceiverTest, RecoveryCallbackDoesNotLoopInfinitely) {
|
||||
class LoopbackRecoveredPacketReceiver : public RecoveredPacketReceiver {
|
||||
public:
|
||||
const int kMaxRecursionDepth = 10;
|
||||
|
||||
LoopbackRecoveredPacketReceiver()
|
||||
: receiver_(nullptr),
|
||||
did_receive_call_back_(false),
|
||||
recursion_depth_(0),
|
||||
deep_recursion_(false) {}
|
||||
|
||||
void SetReceiver(FlexfecReceiver* receiver) { receiver_ = receiver; }
|
||||
bool DidReceiveCallback() const { return did_receive_call_back_; }
|
||||
bool DeepRecursion() const { return deep_recursion_; }
|
||||
|
||||
// Implements RecoveredPacketReceiver.
|
||||
void OnRecoveredPacket(const uint8_t* packet, size_t length) {
|
||||
RtpPacketReceived parsed_packet;
|
||||
EXPECT_TRUE(parsed_packet.Parse(packet, length));
|
||||
|
||||
did_receive_call_back_ = true;
|
||||
|
||||
if (recursion_depth_ > kMaxRecursionDepth) {
|
||||
deep_recursion_ = true;
|
||||
return;
|
||||
}
|
||||
++recursion_depth_;
|
||||
RTC_DCHECK(receiver_);
|
||||
receiver_->OnRtpPacket(parsed_packet);
|
||||
--recursion_depth_;
|
||||
}
|
||||
|
||||
private:
|
||||
FlexfecReceiver* receiver_;
|
||||
bool did_receive_call_back_;
|
||||
int recursion_depth_;
|
||||
bool deep_recursion_;
|
||||
} loopback_recovered_packet_receiver;
|
||||
|
||||
// Feed recovered packets back into |receiver|.
|
||||
FlexfecReceiver receiver(kFlexfecSsrc, kMediaSsrc,
|
||||
&loopback_recovered_packet_receiver);
|
||||
loopback_recovered_packet_receiver.SetReceiver(&receiver);
|
||||
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive first media packet but drop second.
|
||||
auto media_it = media_packets.begin();
|
||||
receiver.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Receive FEC packet and verify that a packet was recovered.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
receiver.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
EXPECT_TRUE(loopback_recovered_packet_receiver.DidReceiveCallback());
|
||||
EXPECT_FALSE(loopback_recovered_packet_receiver.DeepRecursion());
|
||||
}
|
||||
|
||||
TEST_F(FlexfecReceiverTest, CalculatesNumberOfPackets) {
|
||||
const size_t kNumMediaPackets = 2;
|
||||
const size_t kNumFecPackets = 1;
|
||||
|
||||
PacketList media_packets;
|
||||
PacketizeFrame(kNumMediaPackets, 0, &media_packets);
|
||||
std::list<Packet*> fec_packets = EncodeFec(media_packets, kNumFecPackets);
|
||||
|
||||
// Receive first media packet but drop second.
|
||||
auto media_it = media_packets.begin();
|
||||
receiver_.OnRtpPacket(ParsePacket(**media_it));
|
||||
|
||||
// Receive FEC packet and ensure recovery of lost media packet.
|
||||
auto fec_it = fec_packets.begin();
|
||||
std::unique_ptr<Packet> packet_with_rtp_header =
|
||||
packet_generator_.BuildFlexfecPacket(**fec_it);
|
||||
media_it++;
|
||||
EXPECT_CALL(recovered_packet_receiver_,
|
||||
OnRecoveredPacket(_, (*media_it)->length))
|
||||
.With(
|
||||
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length)));
|
||||
receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header));
|
||||
|
||||
// Check stats calculations.
|
||||
FecPacketCounter packet_counter = receiver_.GetPacketCounter();
|
||||
EXPECT_EQ(2U, packet_counter.num_packets);
|
||||
EXPECT_EQ(1U, packet_counter.num_fec_packets);
|
||||
EXPECT_EQ(1U, packet_counter.num_recovered_packets);
|
||||
}
|
||||
|
||||
} // namespace webrtc
|
||||
Reference in New Issue
Block a user