OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <string.h> | 5 #include <string.h> |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "media/base/mock_media_log.h" |
10 #include "media/formats/mp4/box_reader.h" | 11 #include "media/formats/mp4/box_reader.h" |
11 #include "media/formats/mp4/rcheck.h" | 12 #include "media/formats/mp4/rcheck.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
13 | 15 |
| 16 using ::testing::HasSubstr; |
| 17 using ::testing::StrictMock; |
| 18 |
14 namespace media { | 19 namespace media { |
15 namespace mp4 { | 20 namespace mp4 { |
16 | 21 |
17 static const uint8 kSkipBox[] = { | 22 static const uint8 kSkipBox[] = { |
18 // Top-level test box containing three children | 23 // Top-level test box containing three children |
19 0x00, 0x00, 0x00, 0x40, 's', 'k', 'i', 'p', | 24 0x00, 0x00, 0x00, 0x40, 's', 'k', 'i', 'p', |
20 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, | 25 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, |
21 0xf9, 0x0a, 0x0b, 0x0c, 0xfd, 0x0e, 0x0f, 0x10, | 26 0xf9, 0x0a, 0x0b, 0x0c, 0xfd, 0x0e, 0x0f, 0x10, |
22 // Ordinary (8-byte header) child box | 27 // Ordinary (8-byte header) child box |
23 0x00, 0x00, 0x00, 0x0c, 'p', 's', 's', 'h', 0xde, 0xad, 0xbe, 0xef, | 28 0x00, 0x00, 0x00, 0x0c, 'p', 's', 's', 'h', 0xde, 0xad, 0xbe, 0xef, |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
69 FourCC BoxType() const override { return FOURCC_SKIP; } | 74 FourCC BoxType() const override { return FOURCC_SKIP; } |
70 | 75 |
71 SkipBox(); | 76 SkipBox(); |
72 ~SkipBox() override; | 77 ~SkipBox() override; |
73 }; | 78 }; |
74 | 79 |
75 SkipBox::SkipBox() {} | 80 SkipBox::SkipBox() {} |
76 SkipBox::~SkipBox() {} | 81 SkipBox::~SkipBox() {} |
77 | 82 |
78 class BoxReaderTest : public testing::Test { | 83 class BoxReaderTest : public testing::Test { |
| 84 public: |
| 85 BoxReaderTest() : media_log_(new StrictMock<MockMediaLog>()) {} |
| 86 |
79 protected: | 87 protected: |
80 std::vector<uint8> GetBuf() { | 88 std::vector<uint8> GetBuf() { |
81 return std::vector<uint8>(kSkipBox, kSkipBox + sizeof(kSkipBox)); | 89 return std::vector<uint8>(kSkipBox, kSkipBox + sizeof(kSkipBox)); |
82 } | 90 } |
| 91 |
| 92 void TestTopLevelBox(const uint8* data, int size, uint32 fourCC) { |
| 93 std::vector<uint8> buf(data, data + size); |
| 94 |
| 95 bool err; |
| 96 scoped_ptr<BoxReader> reader( |
| 97 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
| 98 |
| 99 EXPECT_FALSE(err); |
| 100 EXPECT_TRUE(reader); |
| 101 EXPECT_EQ(fourCC, reader->type()); |
| 102 EXPECT_EQ(reader->size(), size); |
| 103 } |
| 104 |
| 105 scoped_refptr<StrictMock<MockMediaLog>> media_log_; |
83 }; | 106 }; |
84 | 107 |
85 TEST_F(BoxReaderTest, ExpectedOperationTest) { | 108 TEST_F(BoxReaderTest, ExpectedOperationTest) { |
86 std::vector<uint8> buf = GetBuf(); | 109 std::vector<uint8> buf = GetBuf(); |
87 bool err; | 110 bool err; |
88 scoped_ptr<BoxReader> reader( | 111 scoped_ptr<BoxReader> reader( |
89 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | 112 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
90 EXPECT_FALSE(err); | 113 EXPECT_FALSE(err); |
91 EXPECT_TRUE(reader.get()); | 114 EXPECT_TRUE(reader.get()); |
92 | 115 |
93 SkipBox box; | 116 SkipBox box; |
94 EXPECT_TRUE(box.Parse(reader.get())); | 117 EXPECT_TRUE(box.Parse(reader.get())); |
95 EXPECT_EQ(0x01, reader->version()); | 118 EXPECT_EQ(0x01, reader->version()); |
96 EXPECT_EQ(0x020304u, reader->flags()); | 119 EXPECT_EQ(0x020304u, reader->flags()); |
97 EXPECT_EQ(0x05, box.a); | 120 EXPECT_EQ(0x05, box.a); |
98 EXPECT_EQ(0x06, box.b); | 121 EXPECT_EQ(0x06, box.b); |
99 EXPECT_EQ(0x0708, box.c); | 122 EXPECT_EQ(0x0708, box.c); |
100 EXPECT_EQ(static_cast<int32>(0xf90a0b0c), box.d); | 123 EXPECT_EQ(static_cast<int32>(0xf90a0b0c), box.d); |
101 EXPECT_EQ(static_cast<int32>(0xfd0e0f10), box.e); | 124 EXPECT_EQ(static_cast<int32>(0xfd0e0f10), box.e); |
102 | 125 |
103 EXPECT_EQ(2u, box.kids.size()); | 126 EXPECT_EQ(2u, box.kids.size()); |
104 EXPECT_EQ(0xdeadbeef, box.kids[0].val); | 127 EXPECT_EQ(0xdeadbeef, box.kids[0].val); |
105 EXPECT_EQ(0xfacecafe, box.kids[1].val); | 128 EXPECT_EQ(0xfacecafe, box.kids[1].val); |
106 | 129 |
107 // Accounting for the extra byte outside of the box above | 130 // Accounting for the extra byte outside of the box above |
108 EXPECT_EQ(buf.size(), static_cast<uint64>(reader->size() + 1)); | 131 EXPECT_EQ(buf.size(), static_cast<uint64>(reader->size() + 1)); |
109 } | 132 } |
110 | 133 |
111 TEST_F(BoxReaderTest, OuterTooShortTest) { | 134 TEST_F(BoxReaderTest, OuterTooShortTest) { |
112 std::vector<uint8> buf = GetBuf(); | 135 std::vector<uint8> buf = GetBuf(); |
113 bool err; | 136 bool err; |
114 | 137 |
115 // Create a soft failure by truncating the outer box. | 138 // Create a soft failure by truncating the outer box. |
116 scoped_ptr<BoxReader> r(BoxReader::ReadTopLevelBox(&buf[0], buf.size() - 2, | 139 scoped_ptr<BoxReader> r( |
117 new MediaLog(), &err)); | 140 BoxReader::ReadTopLevelBox(&buf[0], buf.size() - 2, media_log_, &err)); |
118 | 141 |
119 EXPECT_FALSE(err); | 142 EXPECT_FALSE(err); |
120 EXPECT_FALSE(r.get()); | 143 EXPECT_FALSE(r.get()); |
121 } | 144 } |
122 | 145 |
123 TEST_F(BoxReaderTest, InnerTooLongTest) { | 146 TEST_F(BoxReaderTest, InnerTooLongTest) { |
124 std::vector<uint8> buf = GetBuf(); | 147 std::vector<uint8> buf = GetBuf(); |
125 bool err; | 148 bool err; |
126 | 149 |
127 // Make an inner box too big for its outer box. | 150 // Make an inner box too big for its outer box. |
128 buf[25] = 1; | 151 buf[25] = 1; |
129 scoped_ptr<BoxReader> reader( | 152 scoped_ptr<BoxReader> reader( |
130 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | 153 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
131 | 154 |
132 SkipBox box; | 155 SkipBox box; |
133 EXPECT_FALSE(box.Parse(reader.get())); | 156 EXPECT_FALSE(box.Parse(reader.get())); |
134 } | 157 } |
135 | 158 |
136 TEST_F(BoxReaderTest, WrongFourCCTest) { | 159 TEST_F(BoxReaderTest, WrongFourCCTest) { |
137 std::vector<uint8> buf = GetBuf(); | 160 std::vector<uint8> buf = GetBuf(); |
138 bool err; | 161 bool err; |
139 | 162 |
140 // Set an unrecognized top-level FourCC. | 163 // Set an unrecognized top-level FourCC. |
141 buf[5] = 1; | 164 buf[5] = 1; |
| 165 |
| 166 EXPECT_MEDIA_LOG(HasSubstr("Unrecognized top-level box type s\\u0001ip")); |
| 167 |
142 scoped_ptr<BoxReader> reader( | 168 scoped_ptr<BoxReader> reader( |
143 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | 169 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
144 EXPECT_FALSE(reader.get()); | 170 EXPECT_FALSE(reader.get()); |
145 EXPECT_TRUE(err); | 171 EXPECT_TRUE(err); |
146 } | 172 } |
147 | 173 |
148 TEST_F(BoxReaderTest, ScanChildrenTest) { | 174 TEST_F(BoxReaderTest, ScanChildrenTest) { |
149 std::vector<uint8> buf = GetBuf(); | 175 std::vector<uint8> buf = GetBuf(); |
150 bool err; | 176 bool err; |
151 scoped_ptr<BoxReader> reader( | 177 scoped_ptr<BoxReader> reader( |
152 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | 178 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
153 | 179 |
154 EXPECT_TRUE(reader->SkipBytes(16) && reader->ScanChildren()); | 180 EXPECT_TRUE(reader->SkipBytes(16) && reader->ScanChildren()); |
155 | 181 |
156 FreeBox free; | 182 FreeBox free; |
157 EXPECT_TRUE(reader->ReadChild(&free)); | 183 EXPECT_TRUE(reader->ReadChild(&free)); |
158 EXPECT_FALSE(reader->ReadChild(&free)); | 184 EXPECT_FALSE(reader->ReadChild(&free)); |
159 EXPECT_TRUE(reader->MaybeReadChild(&free)); | 185 EXPECT_TRUE(reader->MaybeReadChild(&free)); |
160 | 186 |
161 std::vector<PsshBox> kids; | 187 std::vector<PsshBox> kids; |
162 | 188 |
163 EXPECT_TRUE(reader->ReadChildren(&kids)); | 189 EXPECT_TRUE(reader->ReadChildren(&kids)); |
164 EXPECT_EQ(2u, kids.size()); | 190 EXPECT_EQ(2u, kids.size()); |
165 kids.clear(); | 191 kids.clear(); |
166 EXPECT_FALSE(reader->ReadChildren(&kids)); | 192 EXPECT_FALSE(reader->ReadChildren(&kids)); |
167 EXPECT_TRUE(reader->MaybeReadChildren(&kids)); | 193 EXPECT_TRUE(reader->MaybeReadChildren(&kids)); |
168 } | 194 } |
169 | 195 |
170 TEST_F(BoxReaderTest, ReadAllChildrenTest) { | 196 TEST_F(BoxReaderTest, ReadAllChildrenTest) { |
171 std::vector<uint8> buf = GetBuf(); | 197 std::vector<uint8> buf = GetBuf(); |
172 // Modify buffer to exclude its last 'free' box | 198 // Modify buffer to exclude its last 'free' box |
173 buf[3] = 0x38; | 199 buf[3] = 0x38; |
174 bool err; | 200 bool err; |
175 scoped_ptr<BoxReader> reader( | 201 scoped_ptr<BoxReader> reader( |
176 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | 202 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), media_log_, &err)); |
177 | 203 |
178 std::vector<PsshBox> kids; | 204 std::vector<PsshBox> kids; |
179 EXPECT_TRUE(reader->SkipBytes(16) && reader->ReadAllChildren(&kids)); | 205 EXPECT_TRUE(reader->SkipBytes(16) && reader->ReadAllChildren(&kids)); |
180 EXPECT_EQ(2u, kids.size()); | 206 EXPECT_EQ(2u, kids.size()); |
181 EXPECT_EQ(kids[0].val, 0xdeadbeef); // Ensure order is preserved | 207 EXPECT_EQ(kids[0].val, 0xdeadbeef); // Ensure order is preserved |
182 } | 208 } |
183 | 209 |
184 static void TestTopLevelBox(const uint8* data, int size, uint32 fourCC) { | |
185 | |
186 std::vector<uint8> buf(data, data + size); | |
187 | |
188 bool err; | |
189 scoped_ptr<BoxReader> reader( | |
190 BoxReader::ReadTopLevelBox(&buf[0], buf.size(), new MediaLog(), &err)); | |
191 | |
192 EXPECT_FALSE(err); | |
193 EXPECT_TRUE(reader); | |
194 EXPECT_EQ(fourCC, reader->type()); | |
195 EXPECT_EQ(reader->size(), size); | |
196 } | |
197 | |
198 TEST_F(BoxReaderTest, SkippingBloc) { | 210 TEST_F(BoxReaderTest, SkippingBloc) { |
199 static const uint8 kData[] = { | 211 static const uint8 kData[] = { |
200 0x00, 0x00, 0x00, 0x09, 'b', 'l', 'o', 'c', 0x00 | 212 0x00, 0x00, 0x00, 0x09, 'b', 'l', 'o', 'c', 0x00 |
201 }; | 213 }; |
202 | 214 |
203 TestTopLevelBox(kData, sizeof(kData), FOURCC_BLOC); | 215 TestTopLevelBox(kData, sizeof(kData), FOURCC_BLOC); |
204 } | 216 } |
205 | 217 |
206 TEST_F(BoxReaderTest, SkippingEmsg) { | 218 TEST_F(BoxReaderTest, SkippingEmsg) { |
207 static const uint8 kData[] = { | 219 static const uint8 kData[] = { |
(...skipping 18 matching lines...) Expand all Loading... |
226 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, | 238 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, |
227 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, // usertype | 239 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, // usertype |
228 0x00, | 240 0x00, |
229 }; | 241 }; |
230 | 242 |
231 TestTopLevelBox(kData, sizeof(kData), FOURCC_UUID); | 243 TestTopLevelBox(kData, sizeof(kData), FOURCC_UUID); |
232 } | 244 } |
233 | 245 |
234 } // namespace mp4 | 246 } // namespace mp4 |
235 } // namespace media | 247 } // namespace media |
OLD | NEW |