| // 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, DefaultActionIsRead) { |
| { |
| InSequence dummy; |
| |
| // This intentionally does not set the action and relies on the parser using |
| // a default action value of kRead. |
| EXPECT_CALL(callback_, OnSegmentBegin(metadata_, NotNull())) |
| .WillOnce(Return(Status(Status::kOkCompleted))); |
| 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 |