// Copyright 2014 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include #include #include "media/ffmpeg/ffmpeg_common.h" #include "media/ffmpeg/scoped_av_packet.h" #include "media/filters/ffmpeg_aac_bitstream_converter.h" #include "testing/gtest/include/gtest/gtest.h" namespace media { namespace { const int kAacMainProfile = 0; const int kAacLowComplexityProfile = 1; } // namespace // Class for testing the FFmpegAACBitstreamConverter. class FFmpegAACBitstreamConverterTest : public testing::Test { public: FFmpegAACBitstreamConverterTest(const FFmpegAACBitstreamConverterTest&) = delete; FFmpegAACBitstreamConverterTest& operator=( const FFmpegAACBitstreamConverterTest&) = delete; protected: FFmpegAACBitstreamConverterTest() { // Minimal extra data header memset(extradata_header_, 0, sizeof(extradata_header_)); // Set up reasonable aac parameters memset(&test_parameters_, 0, sizeof(AVCodecParameters)); test_parameters_.codec_id = AV_CODEC_ID_AAC; test_parameters_.profile = FF_PROFILE_AAC_MAIN; test_parameters_.ch_layout.nb_channels = 2; test_parameters_.extradata = extradata_header_; test_parameters_.extradata_size = sizeof(extradata_header_); } void CreatePacket(AVPacket* packet, const uint8_t* data, uint32_t data_size) { // Create new packet sized of |data_size| from |data|. EXPECT_EQ(av_new_packet(packet, data_size), 0); memcpy(packet->data, data, data_size); } // Variable to hold valid dummy parameters for testing. AVCodecParameters test_parameters_; private: uint8_t extradata_header_[2]; }; TEST_F(FFmpegAACBitstreamConverterTest, Conversion_Success) { FFmpegAACBitstreamConverter converter(&test_parameters_); uint8_t dummy_packet[1000]; // Fill dummy packet with junk data. aac converter doesn't look into packet // data, just header, so can fill with whatever we want for test. for(size_t i = 0; i < sizeof(dummy_packet); i++) { dummy_packet[i] = i & 0xFF; // Repeated sequences of 0-255 } auto test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); // Try out the actual conversion (should be successful and allocate new // packet and destroy the old one). EXPECT_TRUE(converter.ConvertPacket(test_packet.get())); // Check that a header was added and that packet data was preserved EXPECT_EQ(static_cast(test_packet->size), static_cast(sizeof(dummy_packet) + FFmpegAACBitstreamConverter::kAdtsHeaderSize)); EXPECT_EQ(memcmp( reinterpret_cast(test_packet->data + FFmpegAACBitstreamConverter::kAdtsHeaderSize), reinterpret_cast(dummy_packet), sizeof(dummy_packet)), 0); } TEST_F(FFmpegAACBitstreamConverterTest, Conversion_FailureNullParams) { // Set up AVCConfigurationRecord to represent NULL data. AVCodecParameters dummy_parameters; dummy_parameters.extradata = nullptr; dummy_parameters.extradata_size = 0; FFmpegAACBitstreamConverter converter(&dummy_parameters); uint8_t dummy_packet[1000] = {0}; // Try out the actual conversion with NULL parameter. EXPECT_FALSE(converter.ConvertPacket(NULL)); // Create new packet to test actual conversion. auto test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); // Try out the actual conversion. This should fail due to missing extradata. EXPECT_FALSE(converter.ConvertPacket(test_packet.get())); } TEST_F(FFmpegAACBitstreamConverterTest, Conversion_AudioProfileType) { FFmpegAACBitstreamConverter converter(&test_parameters_); uint8_t dummy_packet[1000] = {0}; auto test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); EXPECT_TRUE(converter.ConvertPacket(test_packet.get())); // Check that the ADTS header profile matches the parameters int profile = ((test_packet->data[2] & 0xC0) >> 6); EXPECT_EQ(profile, kAacMainProfile); test_parameters_.profile = FF_PROFILE_AAC_HE; FFmpegAACBitstreamConverter converter_he(&test_parameters_); test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); EXPECT_TRUE(converter_he.ConvertPacket(test_packet.get())); profile = ((test_packet->data[2] & 0xC0) >> 6); EXPECT_EQ(profile, kAacLowComplexityProfile); test_parameters_.profile = FF_PROFILE_AAC_ELD; FFmpegAACBitstreamConverter converter_eld(&test_parameters_); test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); EXPECT_FALSE(converter_eld.ConvertPacket(test_packet.get())); } TEST_F(FFmpegAACBitstreamConverterTest, Conversion_MultipleLength) { FFmpegAACBitstreamConverter converter(&test_parameters_); uint8_t dummy_packet[1000]; auto test_packet = ScopedAVPacket::Allocate(); CreatePacket(test_packet.get(), dummy_packet, sizeof(dummy_packet)); // Try out the actual conversion (should be successful and allocate new // packet and destroy the old one). EXPECT_TRUE(converter.ConvertPacket(test_packet.get())); // Check that the ADTS header frame length matches the packet size int frame_length = ((test_packet->data[3] & 0x03) << 11) | ((test_packet->data[4] & 0xFF) << 3) | ((test_packet->data[5] & 0xE0) >> 5); EXPECT_EQ(frame_length, test_packet->size); // Create a second packet that is 1 byte smaller than the first one auto second_test_packet = ScopedAVPacket::Allocate(); CreatePacket(second_test_packet.get(), dummy_packet, sizeof(dummy_packet) - 1); // Try out the actual conversion (should be successful and allocate new // packet and destroy the old one). EXPECT_TRUE(converter.ConvertPacket(second_test_packet.get())); // Check that the ADTS header frame length matches the packet size frame_length = ((second_test_packet->data[3] & 0x03) << 11) | ((second_test_packet->data[4] & 0xFF) << 3) | ((second_test_packet->data[5] & 0xE0) >> 5); EXPECT_EQ(frame_length, second_test_packet->size); } } // namespace media