| // 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 |