| Index: webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc b/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc
|
| index ea59dc1b4da2a3878028e758ca6754820a75c2d8..75ca2e5534a9ef5962d4075a509d8a022b55427e 100644
|
| --- a/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc
|
| +++ b/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc
|
| @@ -22,8 +22,8 @@ namespace webrtc {
|
|
|
| void VerifyHeader(uint16_t seq_num,
|
| uint32_t timestamp,
|
| - int red_pltype,
|
| - int fec_pltype,
|
| + int red_payload_type,
|
| + int fec_payload_type,
|
| RedPacket* packet,
|
| bool marker_bit) {
|
| EXPECT_GT(packet->length(), kRtpHeaderSize);
|
| @@ -31,30 +31,25 @@ void VerifyHeader(uint16_t seq_num,
|
| uint8_t* data = packet->data();
|
| // Marker bit not set.
|
| EXPECT_EQ(marker_bit ? 0x80 : 0, data[1] & 0x80);
|
| - EXPECT_EQ(red_pltype, data[1] & 0x7F);
|
| + EXPECT_EQ(red_payload_type, data[1] & 0x7F);
|
| EXPECT_EQ(seq_num, (data[2] << 8) + data[3]);
|
| uint32_t parsed_timestamp = (data[4] << 24) + (data[5] << 16) +
|
| (data[6] << 8) + data[7];
|
| EXPECT_EQ(timestamp, parsed_timestamp);
|
| - EXPECT_EQ(static_cast<uint8_t>(fec_pltype), data[kRtpHeaderSize]);
|
| + EXPECT_EQ(static_cast<uint8_t>(fec_payload_type), data[kRtpHeaderSize]);
|
| }
|
|
|
| class ProducerFecTest : public ::testing::Test {
|
| protected:
|
| virtual void SetUp() {
|
| - fec_ = new ForwardErrorCorrection();
|
| - producer_ = new ProducerFec(fec_);
|
| - generator_ = new FrameGenerator();
|
| + fec_.reset(new ForwardErrorCorrection());
|
| + producer_.reset(new ProducerFec(fec_.get()));
|
| + generator_.reset(new FrameGenerator());
|
| }
|
|
|
| - virtual void TearDown() {
|
| - delete producer_;
|
| - delete fec_;
|
| - delete generator_;
|
| - }
|
| - ForwardErrorCorrection* fec_;
|
| - ProducerFec* producer_;
|
| - FrameGenerator* generator_;
|
| + std::unique_ptr<ForwardErrorCorrection> fec_;
|
| + std::unique_ptr<ProducerFec> producer_;
|
| + std::unique_ptr<FrameGenerator> generator_;
|
| };
|
|
|
| // Verifies bug found via fuzzing, where a gap in the packet sequence caused us
|
| @@ -92,15 +87,14 @@ TEST_F(ProducerFecTest, NoEmptyFecWithSeqNumGaps) {
|
| producer_->AddRtpPacketAndGenerateFec(packet, p.payload_size,
|
| p.header_size);
|
| uint16_t num_fec_packets = producer_->NumAvailableFecPackets();
|
| - std::vector<RedPacket*> fec_packets;
|
| if (num_fec_packets > 0) {
|
| - fec_packets =
|
| - producer_->GetFecPackets(kRedPayloadType, 99, 100, p.header_size);
|
| + std::vector<std::unique_ptr<RedPacket>> fec_packets =
|
| + producer_->GetFecPacketsAsRed(kRedPayloadType,
|
| + kFecPayloadType,
|
| + 100,
|
| + p.header_size);
|
| EXPECT_EQ(num_fec_packets, fec_packets.size());
|
| }
|
| - for (RedPacket* fec_packet : fec_packets) {
|
| - delete fec_packet;
|
| - }
|
| }
|
| }
|
|
|
| @@ -127,19 +121,19 @@ TEST_F(ProducerFecTest, OneFrameFec) {
|
| }
|
| EXPECT_TRUE(producer_->FecAvailable());
|
| uint16_t seq_num = generator_->NextSeqNum();
|
| - std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType,
|
| - kFecPayloadType,
|
| - seq_num,
|
| - kRtpHeaderSize);
|
| + std::vector<std::unique_ptr<RedPacket>> packets =
|
| + producer_->GetFecPacketsAsRed(kRedPayloadType,
|
| + kFecPayloadType,
|
| + seq_num,
|
| + kRtpHeaderSize);
|
| EXPECT_FALSE(producer_->FecAvailable());
|
| ASSERT_EQ(1u, packets.size());
|
| VerifyHeader(seq_num, last_timestamp,
|
| - kRedPayloadType, kFecPayloadType, packets.front(), false);
|
| + kRedPayloadType, kFecPayloadType, packets.front().get(), false);
|
| while (!rtp_packets.empty()) {
|
| delete rtp_packets.front();
|
| rtp_packets.pop_front();
|
| }
|
| - delete packets.front();
|
| }
|
|
|
| TEST_F(ProducerFecTest, TwoFrameFec) {
|
| @@ -171,27 +165,27 @@ TEST_F(ProducerFecTest, TwoFrameFec) {
|
| }
|
| EXPECT_TRUE(producer_->FecAvailable());
|
| uint16_t seq_num = generator_->NextSeqNum();
|
| - std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType,
|
| - kFecPayloadType,
|
| - seq_num,
|
| - kRtpHeaderSize);
|
| + std::vector<std::unique_ptr<RedPacket>> packets =
|
| + producer_->GetFecPacketsAsRed(kRedPayloadType,
|
| + kFecPayloadType,
|
| + seq_num,
|
| + kRtpHeaderSize);
|
| EXPECT_FALSE(producer_->FecAvailable());
|
| ASSERT_EQ(1u, packets.size());
|
| VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType,
|
| - packets.front(), false);
|
| + packets.front().get(), false);
|
| while (!rtp_packets.empty()) {
|
| delete rtp_packets.front();
|
| rtp_packets.pop_front();
|
| }
|
| - delete packets.front();
|
| }
|
|
|
| TEST_F(ProducerFecTest, BuildRedPacket) {
|
| generator_->NewFrame(1);
|
| test::RawRtpPacket* packet = generator_->NextPacket(0, 10);
|
| - std::unique_ptr<RedPacket> red_packet(producer_->BuildRedPacket(
|
| + std::unique_ptr<RedPacket> red_packet = ProducerFec::BuildRedPacket(
|
| packet->data, packet->length - kRtpHeaderSize, kRtpHeaderSize,
|
| - kRedPayloadType));
|
| + kRedPayloadType);
|
| EXPECT_EQ(packet->length + 1, red_packet->length());
|
| VerifyHeader(packet->header.header.sequenceNumber,
|
| packet->header.header.timestamp,
|
|
|