?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Add CHECK to diagnose the crash.
[chromium-blink-merge.git] / media / cdm / cenc_utils_unittest.cc
blob992361976d9be450e17bca96ff9fa07dfdcfb8a7
1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "media/cdm/cenc_utils.h"
6
7 #include "base/logging.h"
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 namespace media {
11
12 const uint8_t kKey1Data[] = {
13 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03,
14 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03
15 };
16 const uint8_t kKey2Data[] = {
17 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
18 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
19 };
20 const uint8_t kKey3Data[] = {
21 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x05,
22 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x05,
23 };
24 const uint8_t kKey4Data[] = {
25 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x06,
26 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x06,
27 };
28
29 class CencUtilsTest : public testing::Test {
30 public:
31 CencUtilsTest()
32 : key1_(kKey1Data, kKey1Data + arraysize(kKey1Data)),
33 key2_(kKey2Data, kKey2Data + arraysize(kKey2Data)),
34 key3_(kKey3Data, kKey3Data + arraysize(kKey3Data)),
35 key4_(kKey4Data, kKey4Data + arraysize(kKey4Data)) {}
36
37 protected:
38 // Initialize the start of the 'pssh' box (up to key_count)
39 void InitializePSSHBox(std::vector<uint8_t>* box,
40 uint8_t size,
41 uint8_t version) {
42 DCHECK(box->size() == 0);
43
44 box->reserve(size);
45 // Add size.
46 box->push_back(0);
47 box->push_back(0);
48 box->push_back(0);
49 box->push_back(size);
50 // Add 'pssh'.
51 box->push_back('p');
52 box->push_back('s');
53 box->push_back('s');
54 box->push_back('h');
55 // Add version.
56 box->push_back(version);
57 // Add flags.
58 box->push_back(0);
59 box->push_back(0);
60 box->push_back(0);
61 // Add Clear Key SystemID.
62 box->push_back(0x10);
63 box->push_back(0x77);
64 box->push_back(0xEF);
65 box->push_back(0xEC);
66 box->push_back(0xC0);
67 box->push_back(0xB2);
68 box->push_back(0x4D);
69 box->push_back(0x02);
70 box->push_back(0xAC);
71 box->push_back(0xE3);
72 box->push_back(0x3C);
73 box->push_back(0x1E);
74 box->push_back(0x52);
75 box->push_back(0xE2);
76 box->push_back(0xFB);
77 box->push_back(0x4B);
78 }
79
80 std::vector<uint8_t> MakePSSHBox(uint8_t version) {
81 std::vector<uint8_t> box;
82 uint8_t size = (version == 0) ? 32 : 36;
83 InitializePSSHBox(&box, size, version);
84 if (version > 0) {
85 // Add key_count (= 0).
86 box.push_back(0);
87 box.push_back(0);
88 box.push_back(0);
89 box.push_back(0);
90 }
91 // Add data_size (= 0).
92 box.push_back(0);
93 box.push_back(0);
94 box.push_back(0);
95 box.push_back(0);
96 return box;
97 }
98
99 std::vector<uint8_t> MakePSSHBox(uint8_t version,
100 const std::vector<uint8_t>& key1) {
101 DCHECK(version > 0);
102 DCHECK(key1.size() == 16);
103
104 std::vector<uint8_t> box;
105 uint8_t size = 52;
106 InitializePSSHBox(&box, size, version);
107
108 // Add key_count (= 1).
109 box.push_back(0);
110 box.push_back(0);
111 box.push_back(0);
112 box.push_back(1);
113
114 // Add key1.
115 for (int i = 0; i < 16; ++i)
116 box.push_back(key1[i]);
117
118 // Add data_size (= 0).
119 box.push_back(0);
120 box.push_back(0);
121 box.push_back(0);
122 box.push_back(0);
123 return box;
124 }
125
126 std::vector<uint8_t> MakePSSHBox(uint8_t version,
127 const std::vector<uint8_t>& key1,
128 const std::vector<uint8_t>& key2) {
129 DCHECK(version > 0);
130 DCHECK(key1.size() == 16);
131 DCHECK(key2.size() == 16);
132
133 std::vector<uint8_t> box;
134 uint8_t size = 68;
135 InitializePSSHBox(&box, size, version);
136
137 // Add key_count (= 2).
138 box.push_back(0);
139 box.push_back(0);
140 box.push_back(0);
141 box.push_back(2);
142
143 // Add key1.
144 for (int i = 0; i < 16; ++i)
145 box.push_back(key1[i]);
146
147 // Add key2.
148 for (int i = 0; i < 16; ++i)
149 box.push_back(key2[i]);
150
151 // Add data_size (= 0).
152 box.push_back(0);
153 box.push_back(0);
154 box.push_back(0);
155 box.push_back(0);
156 return box;
157 }
158
159 const std::vector<uint8_t>& Key1() { return key1_; }
160 const std::vector<uint8_t>& Key2() { return key2_; }
161 const std::vector<uint8_t>& Key3() { return key3_; }
162 const std::vector<uint8_t>& Key4() { return key4_; }
163
164 private:
165 std::vector<uint8_t> key1_;
166 std::vector<uint8_t> key2_;
167 std::vector<uint8_t> key3_;
168 std::vector<uint8_t> key4_;
169 };
170
171 TEST_F(CencUtilsTest, EmptyPSSH) {
172 KeyIdList key_ids;
173 EXPECT_TRUE(ValidatePsshInput(std::vector<uint8_t>()));
174 EXPECT_TRUE(GetKeyIdsForCommonSystemId(std::vector<uint8_t>(), &key_ids));
175 EXPECT_EQ(0u, key_ids.size());
176 }
177
178 TEST_F(CencUtilsTest, PSSHVersion0) {
179 std::vector<uint8_t> box = MakePSSHBox(0);
180 KeyIdList key_ids;
181 EXPECT_TRUE(ValidatePsshInput(box));
182 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
183 EXPECT_EQ(0u, key_ids.size());
184 }
185
186 TEST_F(CencUtilsTest, PSSHVersion1WithNoKeys) {
187 std::vector<uint8_t> box = MakePSSHBox(1);
188 KeyIdList key_ids;
189 EXPECT_TRUE(ValidatePsshInput(box));
190 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
191 EXPECT_EQ(0u, key_ids.size());
192 }
193
194 TEST_F(CencUtilsTest, PSSHVersion1WithOneKey) {
195 std::vector<uint8_t> box = MakePSSHBox(1, Key1());
196 KeyIdList key_ids;
197 EXPECT_TRUE(ValidatePsshInput(box));
198 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
199 EXPECT_EQ(1u, key_ids.size());
200 EXPECT_EQ(key_ids[0], Key1());
201 }
202
203 TEST_F(CencUtilsTest, PSSHVersion1WithTwoKeys) {
204 std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
205 KeyIdList key_ids;
206 EXPECT_TRUE(ValidatePsshInput(box));
207 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
208 EXPECT_EQ(2u, key_ids.size());
209 EXPECT_EQ(key_ids[0], Key1());
210 EXPECT_EQ(key_ids[1], Key2());
211 }
212
213 TEST_F(CencUtilsTest, PSSHVersion0Plus1) {
214 std::vector<uint8_t> box0 = MakePSSHBox(0);
215 std::vector<uint8_t> box1 = MakePSSHBox(1, Key1());
216
217 // Concatentate box1 into box0.
218 for (const auto& value : box1)
219 box0.push_back(value);
220
221 KeyIdList key_ids;
222 EXPECT_TRUE(ValidatePsshInput(box0));
223 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box0, &key_ids));
224 EXPECT_EQ(1u, key_ids.size());
225 EXPECT_EQ(key_ids[0], Key1());
226 }
227
228 TEST_F(CencUtilsTest, PSSHVersion1Plus0) {
229 std::vector<uint8_t> box0 = MakePSSHBox(0);
230 std::vector<uint8_t> box1 = MakePSSHBox(1, Key1());
231
232 // Concatentate box0 into box1.
233 for (const auto& value : box0)
234 box1.push_back(value);
235
236 KeyIdList key_ids;
237 EXPECT_TRUE(ValidatePsshInput(box1));
238 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box1, &key_ids));
239 EXPECT_EQ(1u, key_ids.size());
240 EXPECT_EQ(key_ids[0], Key1());
241 }
242
243 TEST_F(CencUtilsTest, MultiplePSSHVersion1) {
244 std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
245 std::vector<uint8_t> box1 = MakePSSHBox(1, Key3());
246 std::vector<uint8_t> box2 = MakePSSHBox(1, Key4());
247
248 // Concatentate box1 into box.
249 for (const auto& value : box1)
250 box.push_back(value);
251 // Concatentate box2 into box.
252 for (const auto& value : box2)
253 box.push_back(value);
254
255 KeyIdList key_ids;
256 EXPECT_TRUE(ValidatePsshInput(box));
257 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
258 EXPECT_EQ(4u, key_ids.size());
259 EXPECT_EQ(key_ids[0], Key1());
260 EXPECT_EQ(key_ids[1], Key2());
261 EXPECT_EQ(key_ids[2], Key3());
262 EXPECT_EQ(key_ids[3], Key4());
263 }
264
265 TEST_F(CencUtilsTest, InvalidPSSH) {
266 std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
267 KeyIdList key_ids;
268 for (uint32 i = 1; i < box.size(); ++i) {
269 // Modify size of data passed to be less than real size.
270 std::vector<uint8_t> truncated(&box[0], &box[0] + i);
271 EXPECT_FALSE(ValidatePsshInput(truncated));
272 EXPECT_FALSE(GetKeyIdsForCommonSystemId(truncated, &key_ids));
273 // Modify starting point.
274 std::vector<uint8_t> changed_offset(&box[i], &box[i] + box.size() - i);
275 EXPECT_FALSE(ValidatePsshInput(changed_offset));
276 EXPECT_FALSE(GetKeyIdsForCommonSystemId(changed_offset, &key_ids));
277 }
278 }
279
280 TEST_F(CencUtilsTest, InvalidSystemID) {
281 std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
282
283 // Modify the System ID.
284 ++box[20];
285
286 KeyIdList key_ids;
287 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
288 EXPECT_EQ(0u, key_ids.size());
289 }
290
291 TEST_F(CencUtilsTest, InvalidFlags) {
292 std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
293
294 // Modify flags.
295 box[10] = 3;
296
297 KeyIdList key_ids;
298 // TODO(jrummell): This should fail as the 'pssh' box is skipped.
299 EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
300 EXPECT_EQ(0u, key_ids.size());
301 }
302
303 TEST_F(CencUtilsTest, LongSize) {
304 const uint8_t data[] = {
305 0x00, 0x00, 0x00, 0x01, // size = 1
306 0x70, 0x73, 0x73, 0x68, // 'pssh'
307 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x4c, // longsize
308 0x01, // version
309 0x00, 0x00, 0x00, // flags
310 0x10, 0x77, 0xEF, 0xEC, 0xC0, 0xB2, 0x4D, 0x02, // SystemID
311 0xAC, 0xE3, 0x3C, 0x1E, 0x52, 0xE2, 0xFB, 0x4B,
312 0x00, 0x00, 0x00, 0x02, // key count
313 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03, // key1
314 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03,
315 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04, // key2
316 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
317 0x00, 0x00, 0x00, 0x00 // datasize
318 };
319
320 KeyIdList key_ids;
321 EXPECT_TRUE(
322 ValidatePsshInput(std::vector<uint8_t>(data, data + arraysize(data))));
323 EXPECT_TRUE(GetKeyIdsForCommonSystemId(
324 std::vector<uint8_t>(data, data + arraysize(data)), &key_ids));
325 EXPECT_EQ(2u, key_ids.size());
326 }
327
328 TEST_F(CencUtilsTest, NoSize) {
329 const uint8_t data[] = {
330 0x00, 0x00, 0x00, 0x00, // size = 0
331 0x70, 0x73, 0x73, 0x68, // 'pssh'
332 0x01, // version
333 0x00, 0x00, 0x00, // flags
334 0x10, 0x77, 0xEF, 0xEC, 0xC0, 0xB2, 0x4D, 0x02, // SystemID
335 0xAC, 0xE3, 0x3C, 0x1E, 0x52, 0xE2, 0xFB, 0x4B,
336 0x00, 0x00, 0x00, 0x02, // key count
337 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03, // key1
338 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03,
339 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04, // key2
340 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
341 0x00, 0x00, 0x00, 0x00 // datasize
342 };
343
344 KeyIdList key_ids;
345 EXPECT_TRUE(
346 ValidatePsshInput(std::vector<uint8_t>(data, data + arraysize(data))));
347 EXPECT_TRUE(GetKeyIdsForCommonSystemId(
348 std::vector<uint8_t>(data, data + arraysize(data)), &key_ids));
349 EXPECT_EQ(2u, key_ids.size());
350 }
351
352 TEST_F(CencUtilsTest, HugeSize) {
353 const uint8_t data[] = {
354 0x00, 0x00, 0x00, 0x01, // size = 1
355 0x70, 0x73, 0x73, 0x68, // 'pssh'
356 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, // longsize = big
357 0x01, // version
358 0x00, 0x00, 0x00, // flags
359 0x10, 0x77, 0xEF, 0xEC, 0xC0, 0xB2, 0x4D, 0x02, // SystemID
360 0xAC, 0xE3, 0x3C, 0x1E, 0x52, 0xE2, 0xFB, 0x4B,
361 0x00, 0x00, 0x00, 0x02, // key count
362 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03, // key1
363 0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03,
364 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04, // key2
365 0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
366 0x00, 0x00, 0x00, 0x00 // datasize
367 };
368
369 KeyIdList key_ids;
370 EXPECT_FALSE(
371 ValidatePsshInput(std::vector<uint8_t>(data, data + arraysize(data))));
372 EXPECT_FALSE(GetKeyIdsForCommonSystemId(
373 std::vector<uint8_t>(data, data + arraysize(data)), &key_ids));
374 }
375
376 } // namespace media