25d26028c1
Remove webm_parser/clang-format-style. Run clang-format -style=file -i for each source file. Change-Id: Ieaf44bd323375cbcfec67014e94b7742d6bda14a
356 lines
10 KiB
C++
356 lines
10 KiB
C++
// Copyright (c) 2016 The WebM 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 "src/segment_parser.h"
|
|
|
|
#include <cstdint>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "test_utils/element_parser_test.h"
|
|
#include "webm/id.h"
|
|
#include "webm/status.h"
|
|
|
|
using testing::_;
|
|
using testing::DoAll;
|
|
using testing::InSequence;
|
|
using testing::NotNull;
|
|
using testing::Return;
|
|
using testing::SetArgPointee;
|
|
|
|
using webm::Action;
|
|
using webm::Ancestory;
|
|
using webm::ElementMetadata;
|
|
using webm::ElementParserTest;
|
|
using webm::Id;
|
|
using webm::SegmentParser;
|
|
using webm::Status;
|
|
|
|
namespace {
|
|
|
|
class SegmentParserTest
|
|
: public ElementParserTest<SegmentParser, Id::kSegment> {};
|
|
|
|
TEST_F(SegmentParserTest, EmptyParse) {
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull())).Times(1);
|
|
EXPECT_CALL(callback_, OnSegmentEnd(metadata_)).Times(1);
|
|
}
|
|
|
|
ParseAndVerify();
|
|
}
|
|
|
|
TEST_F(SegmentParserTest, EmptyParseWithDelayedStart) {
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull()))
|
|
.WillOnce(Return(Status(Status::kOkPartial)))
|
|
.WillOnce(DoAll(SetArgPointee<1>(Action::kRead),
|
|
Return(Status(Status::kOkCompleted))));
|
|
|
|
EXPECT_CALL(callback_, OnSegmentEnd(metadata_)).Times(1);
|
|
}
|
|
|
|
IncrementalParseAndVerify();
|
|
}
|
|
|
|
TEST_F(SegmentParserTest, DefaultValues) {
|
|
SetReaderData({
|
|
0x11, 0x4D, 0x9B, 0x74, // ID = 0x114D9B74 (SeekHead).
|
|
0x80, // Size = 0.
|
|
|
|
0x15, 0x49, 0xA9, 0x66, // ID = 0x1549A966 (Info).
|
|
0x80, // Size = 0.
|
|
|
|
0x1F, 0x43, 0xB6, 0x75, // ID = 0x1F43B675 (Cluster).
|
|
0x80, // Size = 0.
|
|
|
|
0x16, 0x54, 0xAE, 0x6B, // ID = 0x1654AE6B (Tracks).
|
|
0x80, // Size = 0.
|
|
|
|
0x1C, 0x53, 0xBB, 0x6B, // ID = 0x1C53BB6B (Cues).
|
|
0x80, // Size = 0.
|
|
|
|
0x10, 0x43, 0xA7, 0x70, // ID = 0x1043A770 (Chapters).
|
|
0x80, // Size = 0.
|
|
|
|
0x12, 0x54, 0xC3, 0x67, // ID = 0x1254C367 (Tags).
|
|
0x80, // Size = 0.
|
|
});
|
|
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull())).Times(1);
|
|
|
|
ElementMetadata metadata = {Id::kInfo, 5, 0, 5};
|
|
EXPECT_CALL(callback_, OnInfo(metadata, _)).Times(1);
|
|
|
|
metadata = {Id::kCluster, 5, 0, 10};
|
|
EXPECT_CALL(callback_, OnClusterBegin(metadata, _, NotNull())).Times(1);
|
|
EXPECT_CALL(callback_, OnClusterEnd(metadata, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnSegmentEnd(metadata_)).Times(1);
|
|
}
|
|
|
|
ParseAndVerify();
|
|
}
|
|
|
|
TEST_F(SegmentParserTest, RepeatedValues) {
|
|
SetReaderData({
|
|
// Mutliple SeekHead elements.
|
|
0x11, 0x4D, 0x9B, 0x74, // ID = 0x114D9B74 (SeekHead).
|
|
0x8D, // Size = 13.
|
|
|
|
0x4D, 0xBB, // ID = 0x4DBB (Seek).
|
|
0x10, 0x00, 0x00, 0x07, // Size = 7.
|
|
|
|
0x53, 0xAC, // ID = 0x53AC (SeekPosition).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x11, 0x4D, 0x9B, 0x74, // ID = 0x114D9B74 (SeekHead).
|
|
0x8D, // Size = 13.
|
|
|
|
0x4D, 0xBB, // ID = 0x4DBB (Seek).
|
|
0x10, 0x00, 0x00, 0x07, // Size = 7.
|
|
|
|
0x53, 0xAC, // ID = 0x53AC (SeekPosition).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
|
|
// Multiple Info elements.
|
|
0x15, 0x49, 0xA9, 0x66, // ID = 0x1549A966 (Info).
|
|
0x88, // Size = 8.
|
|
|
|
0x2A, 0xD7, 0xB1, // ID = 0x2AD7B1 (TimecodeScale).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x15, 0x49, 0xA9, 0x66, // ID = 0x1549A966 (Info).
|
|
0x88, // Size = 8.
|
|
|
|
0x2A, 0xD7, 0xB1, // ID = 0x2AD7B1 (TimecodeScale).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
|
|
// Multiple Cluster elements.
|
|
0x1F, 0x43, 0xB6, 0x75, // ID = 0x1F43B675 (Cluster).
|
|
0x86, // Size = 6.
|
|
|
|
0xE7, // ID = 0xE7 (Timecode).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x1F, 0x43, 0xB6, 0x75, // ID = 0x1F43B675 (Cluster).
|
|
0x86, // Size = 6.
|
|
|
|
0xE7, // ID = 0xE7 (Timecode).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
|
|
// Multiple Tracks elements.
|
|
0x16, 0x54, 0xAE, 0x6B, // ID = 0x1654AE6B (Tracks).
|
|
0x8B, // Size = 11.
|
|
|
|
0xAE, // ID = 0xAE (TrackEntry).
|
|
0x10, 0x00, 0x00, 0x06, // Size = 6.
|
|
|
|
0xD7, // ID = 0xD7 (TrackNumber).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x16, 0x54, 0xAE, 0x6B, // ID = 0x1654AE6B (Tracks).
|
|
0x8B, // Size = 11.
|
|
|
|
0xAE, // ID = 0xAE (TrackEntry).
|
|
0x10, 0x00, 0x00, 0x06, // Size = 6.
|
|
|
|
0xD7, // ID = 0xD7 (TrackNumber).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
|
|
// Single Cues element.
|
|
0x1C, 0x53, 0xBB, 0x6B, // ID = 0x1C53BB6B (Cues).
|
|
0x8B, // Size = 11.
|
|
|
|
0xBB, // ID = 0xBB (CuePoint).
|
|
0x10, 0x00, 0x00, 0x06, // Size = 6.
|
|
|
|
0xB3, // ID = 0xB3 (CueTime).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
// Single Chapters element.
|
|
0x10, 0x43, 0xA7, 0x70, // ID = 0x1043A770 (Chapters).
|
|
0x8D, // Size = 13.
|
|
|
|
0x45, 0xB9, // ID = 0x45B9 (EditionEntry).
|
|
0x10, 0x00, 0x00, 0x07, // Size = 7.
|
|
|
|
0x45, 0xBC, // ID = 0x45BC (EditionUID).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
// Multiple Tags elements.
|
|
0x12, 0x54, 0xC3, 0x67, // ID = 0x1254C367 (Tags).
|
|
0x93, // Size = 19.
|
|
|
|
0x73, 0x73, // ID = 0x7373 (Tag).
|
|
0x10, 0x00, 0x00, 0x0D, // Size = 13.
|
|
|
|
0x63, 0xC0, // ID = 0x63C0 (Targets).
|
|
0x10, 0x00, 0x00, 0x07, // Size = 7.
|
|
|
|
0x68, 0xCA, // ID = 0x68CA (TargetTypeValue).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x12, 0x54, 0xC3, 0x67, // ID = 0x1254C367 (Tags).
|
|
0x93, // Size = 19.
|
|
|
|
0x73, 0x73, // ID = 0x7373 (Tag).
|
|
0x10, 0x00, 0x00, 0x0D, // Size = 13.
|
|
|
|
0x63, 0xC0, // ID = 0x63C0 (Targets).
|
|
0x10, 0x00, 0x00, 0x07, // Size = 7.
|
|
|
|
0x68, 0xCA, // ID = 0x68CA (TargetTypeValue).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
});
|
|
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull())).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnSeek(_, _)).Times(2);
|
|
|
|
EXPECT_CALL(callback_, OnInfo(_, _)).Times(2);
|
|
|
|
EXPECT_CALL(callback_, OnClusterBegin(_, _, NotNull())).Times(1);
|
|
EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnClusterBegin(_, _, NotNull())).Times(1);
|
|
EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnTrackEntry(_, _)).Times(2);
|
|
|
|
EXPECT_CALL(callback_, OnCuePoint(_, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnEditionEntry(_, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnTag(_, _)).Times(2);
|
|
|
|
EXPECT_CALL(callback_, OnSegmentEnd(metadata_)).Times(1);
|
|
}
|
|
|
|
ParseAndVerify();
|
|
}
|
|
|
|
TEST_F(SegmentParserTest, Skip) {
|
|
SetReaderData({
|
|
0x1F, 0x43, 0xB6, 0x75, // ID = 0x1F43B675 (Cluster).
|
|
0x80, // Size = 0.
|
|
});
|
|
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull()))
|
|
.WillOnce(Return(Status(Status::kOkPartial)))
|
|
.WillOnce(DoAll(SetArgPointee<1>(Action::kSkip),
|
|
Return(Status(Status::kOkCompleted))));
|
|
|
|
EXPECT_CALL(callback_, OnClusterBegin(_, _, _)).Times(0);
|
|
EXPECT_CALL(callback_, OnClusterEnd(_, _)).Times(0);
|
|
|
|
EXPECT_CALL(callback_, OnSegmentEnd(_)).Times(0);
|
|
}
|
|
|
|
IncrementalParseAndVerify();
|
|
}
|
|
|
|
TEST_F(SegmentParserTest, Seek) {
|
|
SetReaderData({
|
|
// We start the reader at the beginning of a FlagInterlaced element
|
|
// (skipping its ID and size, as they're passed into InitAfterSeek).
|
|
// FlagInterlaced, StereoMode, and AlphaMode are all part of the Video
|
|
// element, with the ancestory: Segment -> Tracks -> TrackEntry -> Video.
|
|
0x01, // Body (value = 1).
|
|
|
|
0x53, 0xB8, // ID = 0x53B8 (StereoMode).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x02, // Body (value = 2).
|
|
|
|
0x53, 0xC0, // ID = 0x53C0 (AlphaMode).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x03, // Body (value = 3).
|
|
|
|
// Single Cues element.
|
|
0x1C, 0x53, 0xBB, 0x6B, // ID = 0x1C53BB6B (Cues).
|
|
0x8B, // Size = 11.
|
|
|
|
0xBB, // ID = 0xBB (CuePoint).
|
|
0x10, 0x00, 0x00, 0x06, // Size = 6.
|
|
|
|
0xB3, // ID = 0xB3 (CueTime).
|
|
0x10, 0x00, 0x00, 0x01, // Size = 1.
|
|
0x01, // Body (value = 1).
|
|
});
|
|
|
|
const ElementMetadata flag_interlaced_metadata = {Id::kFlagInterlaced, 0, 1,
|
|
0};
|
|
|
|
EXPECT_CALL(callback_, OnSegmentBegin(_, _)).Times(0);
|
|
{
|
|
InSequence dummy;
|
|
|
|
EXPECT_CALL(callback_, OnElementBegin(flag_interlaced_metadata, NotNull()))
|
|
.Times(1);
|
|
|
|
ElementMetadata metadata = {Id::kStereoMode, 6, 1, 1};
|
|
EXPECT_CALL(callback_, OnElementBegin(metadata, NotNull())).Times(1);
|
|
|
|
metadata = {Id::kAlphaMode, 6, 1, 8};
|
|
EXPECT_CALL(callback_, OnElementBegin(metadata, NotNull())).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnTrackEntry(_, _)).Times(1);
|
|
|
|
metadata = {Id::kCues, 5, 11, 15};
|
|
EXPECT_CALL(callback_, OnElementBegin(metadata, NotNull())).Times(1);
|
|
|
|
metadata = {Id::kCuePoint, 5, 6, 20};
|
|
EXPECT_CALL(callback_, OnElementBegin(metadata, NotNull())).Times(1);
|
|
|
|
metadata = {Id::kCueTime, 5, 1, 25};
|
|
EXPECT_CALL(callback_, OnElementBegin(metadata, NotNull())).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnCuePoint(_, _)).Times(1);
|
|
|
|
EXPECT_CALL(callback_, OnSegmentEnd(_)).Times(1);
|
|
}
|
|
|
|
Ancestory ancestory;
|
|
ASSERT_TRUE(Ancestory::ById(flag_interlaced_metadata.id, &ancestory));
|
|
ancestory = ancestory.next(); // Skip the Segment ancestor.
|
|
|
|
parser_.InitAfterSeek(ancestory, flag_interlaced_metadata);
|
|
|
|
std::uint64_t num_bytes_read = 0;
|
|
const Status status = parser_.Feed(&callback_, &reader_, &num_bytes_read);
|
|
EXPECT_EQ(Status::kOkCompleted, status.code);
|
|
EXPECT_EQ(reader_.size(), num_bytes_read);
|
|
}
|
|
|
|
} // namespace
|