Index: webrtc/modules/rtp_rtcp/source/rtp_format_h264_unittest.cc |
diff --git a/webrtc/modules/rtp_rtcp/source/rtp_format_h264_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtp_format_h264_unittest.cc |
index d70ec5f5f57632aa1da024d2bb9d77f38b3cc0b4..894415da8a7960d70bad206f115182e67e35f235 100644 |
--- a/webrtc/modules/rtp_rtcp/source/rtp_format_h264_unittest.cc |
+++ b/webrtc/modules/rtp_rtcp/source/rtp_format_h264_unittest.cc |
@@ -42,14 +42,6 @@ |
// Bit masks for FU (A and B) headers. |
enum FuDefs { kSBit = 0x80, kEBit = 0x40, kRBit = 0x20 }; |
- |
-RtpPacketizer* CreateH264Packetizer(H264PacketizationMode mode, |
- size_t max_payload_size) { |
- RTPVideoTypeHeader type_header; |
- type_header.H264.packetization_mode = mode; |
- return RtpPacketizer::Create(kRtpVideoH264, max_payload_size, &type_header, |
- kEmptyFrame); |
-} |
void VerifyFua(size_t fua_index, |
const uint8_t* expected_payload, |
@@ -92,8 +84,8 @@ |
fragmentation.VerifyAndAllocateFragmentationHeader(1); |
fragmentation.fragmentationOffset[0] = 0; |
fragmentation.fragmentationLength[0] = frame_size; |
- std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode1, max_payload_size)); |
+ std::unique_ptr<RtpPacketizer> packetizer(RtpPacketizer::Create( |
+ kRtpVideoH264, max_payload_size, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame.get(), frame_size, &fragmentation); |
std::unique_ptr<uint8_t[]> packet(new uint8_t[max_payload_size]); |
@@ -160,19 +152,14 @@ |
} |
} // namespace |
-// Tests that should work with both packetization mode 0 and |
-// packetization mode 1. |
-class RtpPacketizerH264ModeTest |
- : public ::testing::TestWithParam<H264PacketizationMode> {}; |
- |
-TEST_P(RtpPacketizerH264ModeTest, TestSingleNalu) { |
+TEST(RtpPacketizerH264Test, TestSingleNalu) { |
const uint8_t frame[2] = {0x05, 0xFF}; // F=0, NRI=0, Type=5. |
RTPFragmentationHeader fragmentation; |
fragmentation.VerifyAndAllocateFragmentationHeader(1); |
fragmentation.fragmentationOffset[0] = 0; |
fragmentation.fragmentationLength[0] = sizeof(frame); |
std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(GetParam(), kMaxPayloadSize)); |
+ RtpPacketizer::Create(kRtpVideoH264, kMaxPayloadSize, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame, sizeof(frame), &fragmentation); |
uint8_t packet[kMaxPayloadSize] = {0}; |
size_t length = 0; |
@@ -180,12 +167,12 @@ |
ASSERT_TRUE(packetizer->NextPacket(packet, &length, &last)); |
EXPECT_EQ(2u, length); |
EXPECT_TRUE(last); |
- VerifySingleNaluPayload(fragmentation, 0, frame, sizeof(frame), packet, |
- length); |
+ VerifySingleNaluPayload( |
+ fragmentation, 0, frame, sizeof(frame), packet, length); |
EXPECT_FALSE(packetizer->NextPacket(packet, &length, &last)); |
} |
-TEST_P(RtpPacketizerH264ModeTest, TestSingleNaluTwoPackets) { |
+TEST(RtpPacketizerH264Test, TestSingleNaluTwoPackets) { |
const size_t kFrameSize = kMaxPayloadSize + 100; |
uint8_t frame[kFrameSize] = {0}; |
for (size_t i = 0; i < kFrameSize; ++i) |
@@ -201,7 +188,7 @@ |
frame[fragmentation.fragmentationOffset[1]] = 0x01; |
std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(GetParam(), kMaxPayloadSize)); |
+ RtpPacketizer::Create(kRtpVideoH264, kMaxPayloadSize, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame, kFrameSize, &fragmentation); |
uint8_t packet[kMaxPayloadSize] = {0}; |
@@ -218,11 +205,6 @@ |
EXPECT_FALSE(packetizer->NextPacket(packet, &length, &last)); |
} |
- |
-INSTANTIATE_TEST_CASE_P(PacketMode, |
- RtpPacketizerH264ModeTest, |
- ::testing::Values(kH264PacketizationMode0, |
- kH264PacketizationMode1)); |
TEST(RtpPacketizerH264Test, TestStapA) { |
const size_t kFrameSize = |
@@ -243,7 +225,7 @@ |
fragmentation.fragmentationLength[2] = |
kNalHeaderSize + kFrameSize - kPayloadOffset; |
std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode1, kMaxPayloadSize)); |
+ RtpPacketizer::Create(kRtpVideoH264, kMaxPayloadSize, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame, kFrameSize, &fragmentation); |
uint8_t packet[kMaxPayloadSize] = {0}; |
@@ -257,39 +239,6 @@ |
for (size_t i = 0; i < fragmentation.fragmentationVectorSize; ++i) |
VerifyStapAPayload(fragmentation, 0, i, frame, kFrameSize, packet, length); |
- EXPECT_FALSE(packetizer->NextPacket(packet, &length, &last)); |
-} |
- |
-TEST(RtpPacketizerH264Test, TestMode0HasNoStapA) { |
- // This is the same setup as for the TestStapA test. |
- const size_t kFrameSize = |
- kMaxPayloadSize - 3 * kLengthFieldLength - kNalHeaderSize; |
- uint8_t frame[kFrameSize] = {0x07, 0xFF, // F=0, NRI=0, Type=7 (SPS). |
- 0x08, 0xFF, // F=0, NRI=0, Type=8 (PPS). |
- 0x05}; // F=0, NRI=0, Type=5 (IDR). |
- const size_t kPayloadOffset = 5; |
- for (size_t i = 0; i < kFrameSize - kPayloadOffset; ++i) |
- frame[i + kPayloadOffset] = i; |
- RTPFragmentationHeader fragmentation; |
- fragmentation.VerifyAndAllocateFragmentationHeader(3); |
- fragmentation.fragmentationOffset[0] = 0; |
- fragmentation.fragmentationLength[0] = 2; |
- fragmentation.fragmentationOffset[1] = 2; |
- fragmentation.fragmentationLength[1] = 2; |
- fragmentation.fragmentationOffset[2] = 4; |
- fragmentation.fragmentationLength[2] = |
- kNalHeaderSize + kFrameSize - kPayloadOffset; |
- std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode0, kMaxPayloadSize)); |
- packetizer->SetPayloadData(frame, kFrameSize, &fragmentation); |
- |
- uint8_t packet[kMaxPayloadSize] = {0}; |
- size_t length = 0; |
- bool last = false; |
- // The three fragments should be returned as three packets. |
- ASSERT_TRUE(packetizer->NextPacket(packet, &length, &last)); |
- ASSERT_TRUE(packetizer->NextPacket(packet, &length, &last)); |
- ASSERT_TRUE(packetizer->NextPacket(packet, &length, &last)); |
EXPECT_FALSE(packetizer->NextPacket(packet, &length, &last)); |
} |
@@ -311,7 +260,7 @@ |
fragmentation.fragmentationLength[2] = |
kNalHeaderSize + kFrameSize - kPayloadOffset; |
std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode1, kMaxPayloadSize)); |
+ RtpPacketizer::Create(kRtpVideoH264, kMaxPayloadSize, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame, kFrameSize, &fragmentation); |
uint8_t packet[kMaxPayloadSize] = {0}; |
@@ -359,7 +308,7 @@ |
} |
} |
std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode1, kMaxPayloadSize)); |
+ RtpPacketizer::Create(kRtpVideoH264, kMaxPayloadSize, NULL, kEmptyFrame)); |
packetizer->SetPayloadData(frame, kFrameSize, &fragmentation); |
// First expecting two FU-A packets. |
@@ -432,28 +381,6 @@ |
sizeof(kExpectedPayloadSizes) / sizeof(size_t))); |
} |
-#if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
- |
-TEST(RtpPacketizerH264DeathTest, SendOverlongDataInPacketizationMode0) { |
- const size_t kFrameSize = kMaxPayloadSize + 100; |
- uint8_t frame[kFrameSize] = {0}; |
- for (size_t i = 0; i < kFrameSize; ++i) |
- frame[i] = i; |
- RTPFragmentationHeader fragmentation; |
- fragmentation.VerifyAndAllocateFragmentationHeader(1); |
- fragmentation.fragmentationOffset[0] = 0; |
- fragmentation.fragmentationLength[0] = kFrameSize; |
- // Set NAL headers. |
- frame[fragmentation.fragmentationOffset[0]] = 0x01; |
- |
- std::unique_ptr<RtpPacketizer> packetizer( |
- CreateH264Packetizer(kH264PacketizationMode0, kMaxPayloadSize)); |
- EXPECT_DEATH(packetizer->SetPayloadData(frame, kFrameSize, &fragmentation), |
- "payload_size"); |
-} |
- |
-#endif // RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
- |
namespace { |
const uint8_t kStartSequence[] = {0x00, 0x00, 0x00, 0x01}; |
const uint8_t kOriginalSps[] = {kSps, 0x00, 0x00, 0x03, 0x03, |
@@ -494,8 +421,9 @@ |
const size_t kHeaderOverhead = kFuAHeaderSize + 1; |
// Set size to fragment SPS into two FU-A packets. |
- packetizer_.reset(CreateH264Packetizer( |
- kH264PacketizationMode1, sizeof(kOriginalSps) - 2 + kHeaderOverhead)); |
+ packetizer_.reset(RtpPacketizer::Create( |
+ kRtpVideoH264, sizeof(kOriginalSps) - 2 + kHeaderOverhead, nullptr, |
+ kEmptyFrame)); |
packetizer_->SetPayloadData(in_buffer_.data(), in_buffer_.size(), |
&fragmentation_header_); |
@@ -531,8 +459,9 @@ |
sizeof(kIdrTwo) + (kLengthFieldLength * 3); |
// Set size to include SPS and the rest of the packets in a Stap-A package. |
- packetizer_.reset(CreateH264Packetizer(kH264PacketizationMode1, |
- kExpectedTotalSize + kHeaderOverhead)); |
+ packetizer_.reset(RtpPacketizer::Create(kRtpVideoH264, |
+ kExpectedTotalSize + kHeaderOverhead, |
+ nullptr, kEmptyFrame)); |
packetizer_->SetPayloadData(in_buffer_.data(), in_buffer_.size(), |
&fragmentation_header_); |