media/base/container_names_unittest.cc - chromium/src - Git at Google

 // Copyright (c) 2013 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 "base/files/file_util.h"
 #include "media/base/container_names.h"
 #include "media/base/test_data_util.h"
 #include "testing/gtest/include/gtest/gtest.h"

 namespace media {

 namespace container_names {

 // Using a macros to simplify tests. Since EXPECT_EQ outputs the second argument
 // as a string when it fails, this lets the output identify what item actually
 // failed.
 #define VERIFY(buffer, name)                                           \
   EXPECT_EQ(name,                                                      \
             DetermineContainer(reinterpret_cast<const uint8*>(buffer), \
                                sizeof(buffer)))

 // Test that small buffers are handled correctly.
 TEST(ContainerNamesTest, CheckSmallBuffer) {
   // Empty buffer.
   char buffer[1];  // ([0] not allowed on win)
   VERIFY(buffer, CONTAINER_UNKNOWN);

   // Try a simple SRT file.
   char buffer1[] =
       "1\n"
       "00:03:23,550 --> 00:03:24,375\n"
       "You always had a hard time finding your place in this world.\n"
       "\n"
       "2\n"
       "00:03:24,476 --> 00:03:25,175\n"
       "What are you talking about?\n";
   VERIFY(buffer1, CONTAINER_SRT);

   // HLS has it's own loop.
   char buffer2[] = "#EXTM3U"
                    "some other random stuff"
                    "#EXT-X-MEDIA-SEQUENCE:";
   VERIFY(buffer2, CONTAINER_HLS);

   // Try a large buffer all zeros.
   char buffer3[4096];
   memset(buffer3, 0, sizeof(buffer3));
   VERIFY(buffer3, CONTAINER_UNKNOWN);

   // Reuse buffer, but all \n this time.
   memset(buffer3, '\n', sizeof(buffer3));
   VERIFY(buffer3, CONTAINER_UNKNOWN);
 }

 #define BYTE_ORDER_MARK "\xef\xbb\xbf"

 // Note that the comparisons need at least 12 bytes, so make sure the buffer is
 // at least that size.
 const char kAmrBuffer[12] = "#!AMR";
 uint8 kAsfBuffer[] = { 0x30, 0x26, 0xb2, 0x75, 0x8e, 0x66, 0xcf, 0x11, 0xa6,
                        0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c };
 const char kAss1Buffer[] = "[Script Info]";
 const char kAss2Buffer[] = BYTE_ORDER_MARK "[Script Info]";
 uint8 kCafBuffer[] = { 'c', 'a', 'f', 'f', 0, 1, 0, 0, 'd', 'e', 's', 'c', 0, 0,
                        0, 0, 0, 0, 0, 32, 64, 229, 136, 128, 0, 0, 0, 0, 'a',
                        'a', 'c', ' ', 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0,
                        0, 2, 0, 0, 0, 0 };
 const char kDtshdBuffer[12] = "DTSHDHDR";
 const char kDxaBuffer[16] = "DEXA";
 const char kFlacBuffer[12] = "fLaC";
 uint8 kFlvBuffer[12] = { 'F', 'L', 'V', 0, 0, 0, 0, 1, 0, 0, 0, 0 };
 uint8 kIrcamBuffer[] = { 0x64, 0xa3, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1 };
 const char kRm1Buffer[12] = ".RMF\0\0";
 const char kRm2Buffer[12] = ".ra\xfd";
 uint8 kWtvBuffer[] = { 0xb7, 0xd8, 0x00, 0x20, 0x37, 0x49, 0xda, 0x11, 0xa6,
                        0x4e, 0x00, 0x07, 0xe9, 0x5e, 0xad, 0x8d };
 uint8 kBug263073Buffer[] = {
     0x00, 0x00, 0x00, 0x18, 0x66, 0x74, 0x79, 0x70, 0x6d, 0x70, 0x34, 0x32,
     0x00, 0x00, 0x00, 0x00, 0x69, 0x73, 0x6f, 0x6d, 0x6d, 0x70, 0x34, 0x32,
     0x00, 0x00, 0x00, 0x01, 0x6d, 0x64, 0x61, 0x74, 0x00, 0x00, 0x00, 0x00,
     0xaa, 0x2e, 0x22, 0xcf, 0x00, 0x00, 0x00, 0x37, 0x67, 0x64, 0x00, 0x28,
     0xac, 0x2c, 0xa4, 0x01, 0xe0, 0x08, 0x9f, 0x97, 0x01, 0x52, 0x02, 0x02,
     0x02, 0x80, 0x00, 0x01};

 // Test that containers that start with fixed strings are handled correctly.
 // This is to verify that the TAG matches the first 4 characters of the string.
 TEST(ContainerNamesTest, CheckFixedStrings) {
   VERIFY(kAmrBuffer, CONTAINER_AMR);
   VERIFY(kAsfBuffer, CONTAINER_ASF);
   VERIFY(kAss1Buffer, CONTAINER_ASS);
   VERIFY(kAss2Buffer, CONTAINER_ASS);
   VERIFY(kCafBuffer, CONTAINER_CAF);
   VERIFY(kDtshdBuffer, CONTAINER_DTSHD);
   VERIFY(kDxaBuffer, CONTAINER_DXA);
   VERIFY(kFlacBuffer, CONTAINER_FLAC);
   VERIFY(kFlvBuffer, CONTAINER_FLV);
   VERIFY(kIrcamBuffer, CONTAINER_IRCAM);
   VERIFY(kRm1Buffer, CONTAINER_RM);
   VERIFY(kRm2Buffer, CONTAINER_RM);
   VERIFY(kWtvBuffer, CONTAINER_WTV);
   VERIFY(kBug263073Buffer, CONTAINER_MOV);
 }

 // Determine the container type of a specified file.
 void TestFile(MediaContainerName expected, const base::FilePath& filename) {
   char buffer[8192];

   // Windows implementation of ReadFile fails if file smaller than desired size,
   // so use file length if file less than 8192 bytes (http://crbug.com/243885).
   int read_size = sizeof(buffer);
   int64 actual_size;
   if (base::GetFileSize(filename, &actual_size) && actual_size < read_size)
     read_size = actual_size;
   int read = base::ReadFile(filename, buffer, read_size);

   // Now verify the type.
   EXPECT_EQ(expected,
             DetermineContainer(reinterpret_cast<const uint8*>(buffer), read))
       << "Failure with file " << filename.value();
 }

 TEST(ContainerNamesTest, FileCheckOGG) {
   TestFile(CONTAINER_OGG, GetTestDataFilePath("bear.ogv"));
   TestFile(CONTAINER_OGG, GetTestDataFilePath("9ch.ogg"));
 }

 TEST(ContainerNamesTest, FileCheckWAV) {
   TestFile(CONTAINER_WAV, GetTestDataFilePath("4ch.wav"));
   TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_f32le.wav"));
   TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_s16le.wav"));
 }

 TEST(ContainerNamesTest, FileCheckMOV) {
   TestFile(CONTAINER_MOV, GetTestDataFilePath("bear-1280x720.mp4"));
   TestFile(CONTAINER_MOV, GetTestDataFilePath("sfx.m4a"));
 }

 TEST(ContainerNamesTest, FileCheckWEBM) {
   TestFile(CONTAINER_WEBM, GetTestDataFilePath("bear-320x240.webm"));
   TestFile(CONTAINER_WEBM, GetTestDataFilePath("no_streams.webm"));
   TestFile(CONTAINER_WEBM, GetTestDataFilePath("webm_ebml_element"));
 }

 TEST(ContainerNamesTest, FileCheckMP3) {
   TestFile(CONTAINER_MP3, GetTestDataFilePath("id3_test.mp3"));
   TestFile(CONTAINER_MP3, GetTestDataFilePath("sfx.mp3"));
 }

 TEST(ContainerNamesTest, FileCheckAC3) {
   TestFile(CONTAINER_AC3, GetTestDataFilePath("bear.ac3"));
 }

 TEST(ContainerNamesTest, FileCheckAAC) {
   TestFile(CONTAINER_AAC, GetTestDataFilePath("bear.adts"));
 }

 TEST(ContainerNamesTest, FileCheckAIFF) {
   TestFile(CONTAINER_AIFF, GetTestDataFilePath("bear.aiff"));
 }

 TEST(ContainerNamesTest, FileCheckASF) {
   TestFile(CONTAINER_ASF, GetTestDataFilePath("bear.asf"));
 }

 TEST(ContainerNamesTest, FileCheckAVI) {
   TestFile(CONTAINER_AVI, GetTestDataFilePath("bear.avi"));
 }

 TEST(ContainerNamesTest, FileCheckEAC3) {
   TestFile(CONTAINER_EAC3, GetTestDataFilePath("bear.eac3"));
 }

 TEST(ContainerNamesTest, FileCheckFLAC) {
   TestFile(CONTAINER_FLAC, GetTestDataFilePath("bear.flac"));
 }

 TEST(ContainerNamesTest, FileCheckFLV) {
   TestFile(CONTAINER_FLV, GetTestDataFilePath("bear.flv"));
 }

 TEST(ContainerNamesTest, FileCheckH261) {
   TestFile(CONTAINER_H261, GetTestDataFilePath("bear.h261"));
 }

 TEST(ContainerNamesTest, FileCheckH263) {
   TestFile(CONTAINER_H263, GetTestDataFilePath("bear.h263"));
 }

 TEST(ContainerNamesTest, FileCheckMJPEG) {
   TestFile(CONTAINER_MJPEG, GetTestDataFilePath("bear.mjpeg"));
 }

 TEST(ContainerNamesTest, FileCheckMPEG2PS) {
   TestFile(CONTAINER_MPEG2PS, GetTestDataFilePath("bear.mpeg"));
 }

 TEST(ContainerNamesTest, FileCheckMPEG2TS) {
   TestFile(CONTAINER_MPEG2TS, GetTestDataFilePath("bear.m2ts"));
 }

 TEST(ContainerNamesTest, FileCheckRM) {
   TestFile(CONTAINER_RM, GetTestDataFilePath("bear.rm"));
 }

 TEST(ContainerNamesTest, FileCheckSWF) {
   TestFile(CONTAINER_SWF, GetTestDataFilePath("bear.swf"));
 }

 // Try a few non containers.
 TEST(ContainerNamesTest, FileCheckUNKNOWN) {
   TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("ten_byte_file"));
   TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("README"));
   TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_P422.yuv"));
   TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_RGB24.rgb"));
   TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("webm_vp8_track_entry"));
 }

 }  // namespace container_names

 }  // namespace media
	// Copyright (c) 2013 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 "base/files/file_util.h"
	#include "media/base/container_names.h"
	#include "media/base/test_data_util.h"
	#include "testing/gtest/include/gtest/gtest.h"

	namespace media {

	namespace container_names {

	// Using a macros to simplify tests. Since EXPECT_EQ outputs the second argument
	// as a string when it fails, this lets the output identify what item actually
	// failed.
	#define VERIFY(buffer, name) \
	EXPECT_EQ(name, \
	DetermineContainer(reinterpret_cast<const uint8*>(buffer), \
	sizeof(buffer)))

	// Test that small buffers are handled correctly.
	TEST(ContainerNamesTest, CheckSmallBuffer) {
	// Empty buffer.
	char buffer[1]; // ([0] not allowed on win)
	VERIFY(buffer, CONTAINER_UNKNOWN);

	// Try a simple SRT file.
	char buffer1[] =
	"1\n"
	"00:03:23,550 --> 00:03:24,375\n"
	"You always had a hard time finding your place in this world.\n"
	"\n"
	"2\n"
	"00:03:24,476 --> 00:03:25,175\n"
	"What are you talking about?\n";
	VERIFY(buffer1, CONTAINER_SRT);

	// HLS has it's own loop.
	char buffer2[] = "#EXTM3U"
	"some other random stuff"
	"#EXT-X-MEDIA-SEQUENCE:";
	VERIFY(buffer2, CONTAINER_HLS);

	// Try a large buffer all zeros.
	char buffer3[4096];
	memset(buffer3, 0, sizeof(buffer3));
	VERIFY(buffer3, CONTAINER_UNKNOWN);

	// Reuse buffer, but all \n this time.
	memset(buffer3, '\n', sizeof(buffer3));
	VERIFY(buffer3, CONTAINER_UNKNOWN);
	}

	#define BYTE_ORDER_MARK "\xef\xbb\xbf"

	// Note that the comparisons need at least 12 bytes, so make sure the buffer is
	// at least that size.
	const char kAmrBuffer[12] = "#!AMR";
	uint8 kAsfBuffer[] = { 0x30, 0x26, 0xb2, 0x75, 0x8e, 0x66, 0xcf, 0x11, 0xa6,
	0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c };
	const char kAss1Buffer[] = "[Script Info]";
	const char kAss2Buffer[] = BYTE_ORDER_MARK "[Script Info]";
	uint8 kCafBuffer[] = { 'c', 'a', 'f', 'f', 0, 1, 0, 0, 'd', 'e', 's', 'c', 0, 0,
	0, 0, 0, 0, 0, 32, 64, 229, 136, 128, 0, 0, 0, 0, 'a',
	'a', 'c', ' ', 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0,
	0, 2, 0, 0, 0, 0 };
	const char kDtshdBuffer[12] = "DTSHDHDR";
	const char kDxaBuffer[16] = "DEXA";
	const char kFlacBuffer[12] = "fLaC";
	uint8 kFlvBuffer[12] = { 'F', 'L', 'V', 0, 0, 0, 0, 1, 0, 0, 0, 0 };
	uint8 kIrcamBuffer[] = { 0x64, 0xa3, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1 };
	const char kRm1Buffer[12] = ".RMF\0\0";
	const char kRm2Buffer[12] = ".ra\xfd";
	uint8 kWtvBuffer[] = { 0xb7, 0xd8, 0x00, 0x20, 0x37, 0x49, 0xda, 0x11, 0xa6,
	0x4e, 0x00, 0x07, 0xe9, 0x5e, 0xad, 0x8d };
	uint8 kBug263073Buffer[] = {
	0x00, 0x00, 0x00, 0x18, 0x66, 0x74, 0x79, 0x70, 0x6d, 0x70, 0x34, 0x32,
	0x00, 0x00, 0x00, 0x00, 0x69, 0x73, 0x6f, 0x6d, 0x6d, 0x70, 0x34, 0x32,
	0x00, 0x00, 0x00, 0x01, 0x6d, 0x64, 0x61, 0x74, 0x00, 0x00, 0x00, 0x00,
	0xaa, 0x2e, 0x22, 0xcf, 0x00, 0x00, 0x00, 0x37, 0x67, 0x64, 0x00, 0x28,
	0xac, 0x2c, 0xa4, 0x01, 0xe0, 0x08, 0x9f, 0x97, 0x01, 0x52, 0x02, 0x02,
	0x02, 0x80, 0x00, 0x01};

	// Test that containers that start with fixed strings are handled correctly.
	// This is to verify that the TAG matches the first 4 characters of the string.
	TEST(ContainerNamesTest, CheckFixedStrings) {
	VERIFY(kAmrBuffer, CONTAINER_AMR);
	VERIFY(kAsfBuffer, CONTAINER_ASF);
	VERIFY(kAss1Buffer, CONTAINER_ASS);
	VERIFY(kAss2Buffer, CONTAINER_ASS);
	VERIFY(kCafBuffer, CONTAINER_CAF);
	VERIFY(kDtshdBuffer, CONTAINER_DTSHD);
	VERIFY(kDxaBuffer, CONTAINER_DXA);
	VERIFY(kFlacBuffer, CONTAINER_FLAC);
	VERIFY(kFlvBuffer, CONTAINER_FLV);
	VERIFY(kIrcamBuffer, CONTAINER_IRCAM);
	VERIFY(kRm1Buffer, CONTAINER_RM);
	VERIFY(kRm2Buffer, CONTAINER_RM);
	VERIFY(kWtvBuffer, CONTAINER_WTV);
	VERIFY(kBug263073Buffer, CONTAINER_MOV);
	}

	// Determine the container type of a specified file.
	void TestFile(MediaContainerName expected, const base::FilePath& filename) {
	char buffer[8192];

	// Windows implementation of ReadFile fails if file smaller than desired size,
	// so use file length if file less than 8192 bytes (http://crbug.com/243885).
	int read_size = sizeof(buffer);
	int64 actual_size;
	if (base::GetFileSize(filename, &actual_size) && actual_size < read_size)
	read_size = actual_size;
	int read = base::ReadFile(filename, buffer, read_size);

	// Now verify the type.
	EXPECT_EQ(expected,
	DetermineContainer(reinterpret_cast<const uint8*>(buffer), read))
	<< "Failure with file " << filename.value();
	}

	TEST(ContainerNamesTest, FileCheckOGG) {
	TestFile(CONTAINER_OGG, GetTestDataFilePath("bear.ogv"));
	TestFile(CONTAINER_OGG, GetTestDataFilePath("9ch.ogg"));
	}

	TEST(ContainerNamesTest, FileCheckWAV) {
	TestFile(CONTAINER_WAV, GetTestDataFilePath("4ch.wav"));
	TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_f32le.wav"));
	TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_s16le.wav"));
	}

	TEST(ContainerNamesTest, FileCheckMOV) {
	TestFile(CONTAINER_MOV, GetTestDataFilePath("bear-1280x720.mp4"));
	TestFile(CONTAINER_MOV, GetTestDataFilePath("sfx.m4a"));
	}

	TEST(ContainerNamesTest, FileCheckWEBM) {
	TestFile(CONTAINER_WEBM, GetTestDataFilePath("bear-320x240.webm"));
	TestFile(CONTAINER_WEBM, GetTestDataFilePath("no_streams.webm"));
	TestFile(CONTAINER_WEBM, GetTestDataFilePath("webm_ebml_element"));
	}

	TEST(ContainerNamesTest, FileCheckMP3) {
	TestFile(CONTAINER_MP3, GetTestDataFilePath("id3_test.mp3"));
	TestFile(CONTAINER_MP3, GetTestDataFilePath("sfx.mp3"));
	}

	TEST(ContainerNamesTest, FileCheckAC3) {
	TestFile(CONTAINER_AC3, GetTestDataFilePath("bear.ac3"));
	}

	TEST(ContainerNamesTest, FileCheckAAC) {
	TestFile(CONTAINER_AAC, GetTestDataFilePath("bear.adts"));
	}

	TEST(ContainerNamesTest, FileCheckAIFF) {
	TestFile(CONTAINER_AIFF, GetTestDataFilePath("bear.aiff"));
	}

	TEST(ContainerNamesTest, FileCheckASF) {
	TestFile(CONTAINER_ASF, GetTestDataFilePath("bear.asf"));
	}

	TEST(ContainerNamesTest, FileCheckAVI) {
	TestFile(CONTAINER_AVI, GetTestDataFilePath("bear.avi"));
	}

	TEST(ContainerNamesTest, FileCheckEAC3) {
	TestFile(CONTAINER_EAC3, GetTestDataFilePath("bear.eac3"));
	}

	TEST(ContainerNamesTest, FileCheckFLAC) {
	TestFile(CONTAINER_FLAC, GetTestDataFilePath("bear.flac"));
	}

	TEST(ContainerNamesTest, FileCheckFLV) {
	TestFile(CONTAINER_FLV, GetTestDataFilePath("bear.flv"));
	}

	TEST(ContainerNamesTest, FileCheckH261) {
	TestFile(CONTAINER_H261, GetTestDataFilePath("bear.h261"));
	}

	TEST(ContainerNamesTest, FileCheckH263) {
	TestFile(CONTAINER_H263, GetTestDataFilePath("bear.h263"));
	}

	TEST(ContainerNamesTest, FileCheckMJPEG) {
	TestFile(CONTAINER_MJPEG, GetTestDataFilePath("bear.mjpeg"));
	}

	TEST(ContainerNamesTest, FileCheckMPEG2PS) {
	TestFile(CONTAINER_MPEG2PS, GetTestDataFilePath("bear.mpeg"));
	}

	TEST(ContainerNamesTest, FileCheckMPEG2TS) {
	TestFile(CONTAINER_MPEG2TS, GetTestDataFilePath("bear.m2ts"));
	}

	TEST(ContainerNamesTest, FileCheckRM) {
	TestFile(CONTAINER_RM, GetTestDataFilePath("bear.rm"));
	}

	TEST(ContainerNamesTest, FileCheckSWF) {
	TestFile(CONTAINER_SWF, GetTestDataFilePath("bear.swf"));
	}

	// Try a few non containers.
	TEST(ContainerNamesTest, FileCheckUNKNOWN) {
	TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("ten_byte_file"));
	TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("README"));
	TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_P422.yuv"));
	TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_RGB24.rgb"));
	TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("webm_vp8_track_entry"));
	}

	} // namespace container_names

	} // namespace media