Combine 'pssh' parsing routines.

media/cdm/cenc_utils_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/cdm/cenc_utils.h"
 
 #include "base/logging.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace media {
 
 const uint8_t kKey1Data[] = {
     0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03,
     0x7E, 0x57, 0x1D, 0x03, 0x7E, 0x57, 0x1D, 0x03
 };
 const uint8_t kKey2Data[] = {
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x04,
 };
 const uint8_t kKey3Data[] = {
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x05,
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x05,
 };
 const uint8_t kKey4Data[] = {
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x06,
     0x7E, 0x57, 0x1D, 0x04, 0x7E, 0x57, 0x1D, 0x06,
 };
+const uint8_t kClearKeyUuid[] = {

[ddorwin, 2015/06/04 18:08:45]

s/CK/CommonSystem/

[jrummell, 2015/06/04 21:47:10]

Done.

+    0x10, 0x77, 0xEF, 0xEC, 0xC0, 0xB2, 0x4D, 0x02,
+    0xAC, 0xE3, 0x3C, 0x1E, 0x52, 0xE2, 0xFB, 0x4B
+};
 
 class CencUtilsTest : public testing::Test {
  public:
   CencUtilsTest()
       : key1_(kKey1Data, kKey1Data + arraysize(kKey1Data)),
         key2_(kKey2Data, kKey2Data + arraysize(kKey2Data)),
         key3_(kKey3Data, kKey3Data + arraysize(kKey3Data)),
         key4_(kKey4Data, kKey4Data + arraysize(kKey4Data)) {}
 
  protected:
@@ skipping 111 matching lines @@
       box.push_back(key2[i]);
 
     // Add data_size (= 0).
     box.push_back(0);
     box.push_back(0);
     box.push_back(0);
     box.push_back(0);
     return box;
   }
 
+  void AppendData(std::vector<uint8_t>& pssh_box,
+                  const std::vector<uint8_t>& data) {
+    // This assumes that |pssh_box| has been created using the routines above,
+    // and simply appends the data to the end of it. It updates the box size
+    // and sets the data size.
+    DCHECK(data.size() < 100);
+    pssh_box[3] += data.size();
+    pssh_box.pop_back();
+    pssh_box.push_back(data.size());
+    pssh_box.insert(pssh_box.end(), data.begin(), data.end());
+  }
+
   const std::vector<uint8_t>& Key1() { return key1_; }
   const std::vector<uint8_t>& Key2() { return key2_; }
   const std::vector<uint8_t>& Key3() { return key3_; }
   const std::vector<uint8_t>& Key4() { return key4_; }
 
  private:
   std::vector<uint8_t> key1_;
   std::vector<uint8_t> key2_;
   std::vector<uint8_t> key3_;
   std::vector<uint8_t> key4_;
@@ skipping 38 matching lines @@
   EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
   EXPECT_EQ(2u, key_ids.size());
   EXPECT_EQ(key_ids[0], Key1());
   EXPECT_EQ(key_ids[1], Key2());
 }
 
 TEST_F(CencUtilsTest, PSSHVersion0Plus1) {
   std::vector<uint8_t> box0 = MakePSSHBox(0);
   std::vector<uint8_t> box1 = MakePSSHBox(1, Key1());
 
-  // Concatentate box1 into box0.
-  for (const auto& value : box1)
-    box0.push_back(value);
+  // Concatentate box1 onto end of box0.
+  box0.insert(box0.end(), box1.begin(), box1.end());
 
   KeyIdList key_ids;
   EXPECT_TRUE(ValidatePsshInput(box0));
   EXPECT_TRUE(GetKeyIdsForCommonSystemId(box0, &key_ids));
   EXPECT_EQ(1u, key_ids.size());
   EXPECT_EQ(key_ids[0], Key1());
 }
 
 TEST_F(CencUtilsTest, PSSHVersion1Plus0) {
   std::vector<uint8_t> box0 = MakePSSHBox(0);
   std::vector<uint8_t> box1 = MakePSSHBox(1, Key1());
 
-  // Concatentate box0 into box1.
-  for (const auto& value : box0)
-    box1.push_back(value);
+  // Concatentate box0 onto end of box1.
+  box1.insert(box1.end(), box0.begin(), box0.end());
 
   KeyIdList key_ids;
   EXPECT_TRUE(ValidatePsshInput(box1));
   EXPECT_TRUE(GetKeyIdsForCommonSystemId(box1, &key_ids));
   EXPECT_EQ(1u, key_ids.size());
   EXPECT_EQ(key_ids[0], Key1());
 }
 
 TEST_F(CencUtilsTest, MultiplePSSHVersion1) {
   std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
   std::vector<uint8_t> box1 = MakePSSHBox(1, Key3());
   std::vector<uint8_t> box2 = MakePSSHBox(1, Key4());
 
-  // Concatentate box1 into box.
-  for (const auto& value : box1)
-    box.push_back(value);
-  // Concatentate box2 into box.
-  for (const auto& value : box2)
-    box.push_back(value);
+  // Concatentate box1 and box2 onto end of box.
+  box.insert(box.end(), box1.begin(), box1.end());
+  box.insert(box.end(), box2.begin(), box2.end());
 
   KeyIdList key_ids;
   EXPECT_TRUE(ValidatePsshInput(box));
   EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
   EXPECT_EQ(4u, key_ids.size());
   EXPECT_EQ(key_ids[0], Key1());
   EXPECT_EQ(key_ids[1], Key2());
   EXPECT_EQ(key_ids[2], Key3());
   EXPECT_EQ(key_ids[3], Key4());
 }
 
 TEST_F(CencUtilsTest, InvalidPSSH) {
   std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
   KeyIdList key_ids;
   for (uint32 i = 1; i < box.size(); ++i) {
     // Modify size of data passed to be less than real size.
     std::vector<uint8_t> truncated(&box[0], &box[0] + i);
-    EXPECT_FALSE(ValidatePsshInput(truncated));
+    EXPECT_FALSE(ValidatePsshInput(truncated)) << "Failed for length " << i;
     EXPECT_FALSE(GetKeyIdsForCommonSystemId(truncated, &key_ids));
     // Modify starting point.
     std::vector<uint8_t> changed_offset(&box[i], &box[i] + box.size() - i);
-    EXPECT_FALSE(ValidatePsshInput(changed_offset));
+    EXPECT_FALSE(ValidatePsshInput(changed_offset)) << "Failed for offset "
+                                                    << i;
     EXPECT_FALSE(GetKeyIdsForCommonSystemId(changed_offset, &key_ids));
   }
 }
 
 TEST_F(CencUtilsTest, InvalidSystemID) {
   std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
 
   // Modify the System ID.
   ++box[20];
 
   KeyIdList key_ids;
   EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
   EXPECT_EQ(0u, key_ids.size());
 }
 
 TEST_F(CencUtilsTest, InvalidFlags) {
   std::vector<uint8_t> box = MakePSSHBox(1, Key1(), Key2());
 
   // Modify flags.
   box[10] = 3;
 
   KeyIdList key_ids;
-  // TODO(jrummell): This should fail as the 'pssh' box is skipped.
-  EXPECT_TRUE(GetKeyIdsForCommonSystemId(box, &key_ids));
-  EXPECT_EQ(0u, key_ids.size());
+  EXPECT_FALSE(GetKeyIdsForCommonSystemId(box, &key_ids));
 }
 
 TEST_F(CencUtilsTest, LongSize) {
   const uint8_t data[] = {
       0x00, 0x00, 0x00, 0x01,                          // size = 1
       0x70, 0x73, 0x73, 0x68,                          // 'pssh'
       0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x4c,  // longsize
       0x01,                                            // version
       0x00, 0x00, 0x00,                                // flags
       0x10, 0x77, 0xEF, 0xEC, 0xC0, 0xB2, 0x4D, 0x02,  // SystemID
@@ skipping 55 matching lines @@
       0x00, 0x00, 0x00, 0x00  // datasize
   };
 
   KeyIdList key_ids;
   EXPECT_FALSE(
       ValidatePsshInput(std::vector<uint8_t>(data, data + arraysize(data))));
   EXPECT_FALSE(GetKeyIdsForCommonSystemId(
       std::vector<uint8_t>(data, data + arraysize(data)), &key_ids));
 }
 
+TEST_F(CencUtilsTest, GetPsshDataVersion0) {

[ddorwin, 2015/06/04 18:08:45]

Add an underscore to all of these: GetPsshData_<test-details>

[jrummell, 2015/06/04 21:47:10]

Done.

+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(0);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(0u, pssh_data.size());
+
+  std::vector<uint8_t> data = {0x01, 0x02, 0x03, 0x04};
+  AppendData(box, data);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(data, pssh_data);
+}
+
+TEST_F(CencUtilsTest, GetPsshDataVersion1NoKeys) {
+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(1);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(0u, pssh_data.size());
+
+  std::vector<uint8_t> data = {0x05, 0x06, 0x07, 0x08};
+  AppendData(box, data);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(data, pssh_data);
+}
+
+TEST_F(CencUtilsTest, GetPsshDataVersion1WithKeys) {
+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(1, Key1());
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(0u, pssh_data.size());
+
+  std::vector<uint8_t> data = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08};
+  AppendData(box, data);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_EQ(data, pssh_data);
+}
+
+TEST_F(CencUtilsTest, GetPsshDataVersion2) {

[ddorwin, 2015/06/04 18:08:45]

Should test Version2 then 1 and vice versa.

[jrummell, 2015/06/04 21:47:10]

Done.

+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(1, Key1());
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+
+  // Change the version manually, since we don't know what v2 will contain.
+  box[8] = 2;
+  EXPECT_FALSE(GetPsshData(box, clear_key, &pssh_data));
+}
+
+TEST_F(CencUtilsTest, GetPsshDataDifferentSystemID) {
+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> unknown_system_id(kKey1Data,
+                                         kKey1Data + arraysize(kKey1Data));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(1, Key1());
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+  EXPECT_FALSE(GetPsshData(box, unknown_system_id, &pssh_data));
+}
+
+TEST_F(CencUtilsTest, GetPsshDataMissingData) {
+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box = MakePSSHBox(1, Key1());
+  std::vector<uint8_t> data = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08};
+  AppendData(box, data);
+  EXPECT_TRUE(GetPsshData(box, clear_key, &pssh_data));
+
+  // Remove some data from the end, so now the size is incorrect.
+  box.pop_back();
+  box.pop_back();
+  EXPECT_FALSE(GetPsshData(box, clear_key, &pssh_data));
+}
+
+TEST_F(CencUtilsTest, GetPsshDataMultiplePssh) {
+  std::vector<uint8_t> clear_key(kClearKeyUuid,
+                                 kClearKeyUuid + arraysize(kClearKeyUuid));
+  std::vector<uint8_t> pssh_data;
+
+  std::vector<uint8_t> box1 = MakePSSHBox(1, Key1());
+  std::vector<uint8_t> data1 = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08};
+  AppendData(box1, data1);
+
+  std::vector<uint8_t> box2 = MakePSSHBox(0);
+  std::vector<uint8_t> data2 = {0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8};
+  AppendData(box2, data2);
+
+  box1.insert(box1.end(), box2.begin(), box2.end());
+  EXPECT_TRUE(GetPsshData(box1, clear_key, &pssh_data));
+  EXPECT_EQ(data1, pssh_data);
+  EXPECT_NE(data2, pssh_data);
+}
 }  // namespace media