Have GetKeyIds() return the key IDs from the first matching 'pssh' box

media/cdm/cenc_utils.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/stl_util.h"
 #include "media/formats/mp4/box_definitions.h"
 #include "media/formats/mp4/box_reader.h"
 
 namespace media {
 
 // The initialization data for encrypted media files using the ISO Common
 // Encryption ('cenc') protection scheme may contain one or more protection
 // system specific header ('pssh') boxes.
 // ref: https://w3c.github.io/encrypted-media/cenc-format.html
 
-// SystemID for the Common System.
+// CENC SystemID for the Common System.
 // https://w3c.github.io/encrypted-media/cenc-format.html#common-system
-const uint8_t kCommonSystemId[] = { 0x10, 0x77, 0xef, 0xec,
-                                    0xc0, 0xb2, 0x4d, 0x02,
-                                    0xac, 0xe3, 0x3c, 0x1e,
-                                    0x52, 0xe2, 0xfb, 0x4b };
+const uint8_t kCencCommonSystemId[] = {0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2,
+                                       0x4d, 0x02, 0xac, 0xe3, 0x3c, 0x1e,
+                                       0x52, 0xe2, 0xfb, 0x4b};
 
+// Returns true if |input| contains only 1 or more valid 'pssh' boxes, false
+// otherwise. |pssh_boxes| is updated as the set of parsed 'pssh' boxes.
+// Note: All boxes in |input| must be 'pssh' boxes. However, if they can't be
+//       properly parsed (e.g. unsupported version), then they will be skipped.
 static bool ReadAllPsshBoxes(
     const std::vector<uint8_t>& input,
     std::vector<mp4::FullProtectionSystemSpecificHeader>* pssh_boxes) {
   DCHECK(!input.empty());
 
   // Verify that |input| contains only 'pssh' boxes. ReadAllChildren() is
   // templated, so it checks that each box in |input| matches the box type of
   // the parameter (in this case mp4::ProtectionSystemSpecificHeader is a
   // 'pssh' box). mp4::ProtectionSystemSpecificHeader doesn't validate the
   // 'pssh' contents, so this simply verifies that |input| only contains
@@ skipping 28 matching lines @@
 
 bool ValidatePsshInput(const std::vector<uint8_t>& input) {
   // No 'pssh' boxes is considered valid.
   if (input.empty())
     return true;
 
   std::vector<mp4::FullProtectionSystemSpecificHeader> children;
   return ReadAllPsshBoxes(input, &children);
 }
 
-bool GetKeyIdsForCommonSystemId(const std::vector<uint8_t>& input,
+bool GetKeyIdsForCommonSystemId(const std::vector<uint8_t>& pssh_boxes,
                                 KeyIdList* key_ids) {
+  // If there are no 'pssh' boxes then no key IDs found.
+  if (pssh_boxes.empty())
+    return false;
+
+  std::vector<mp4::FullProtectionSystemSpecificHeader> children;
+  if (!ReadAllPsshBoxes(pssh_boxes, &children))
+    return false;
+
+  // Check all children for an appropriate 'pssh' box, returning the
+  // key IDs found.
   KeyIdList result;
   std::vector<uint8_t> common_system_id(
-      kCommonSystemId, kCommonSystemId + arraysize(kCommonSystemId));
-
-  if (!input.empty()) {
-    std::vector<mp4::FullProtectionSystemSpecificHeader> children;
-    if (!ReadAllPsshBoxes(input, &children))
-      return false;
-
-    // Check all children for an appropriate 'pssh' box, concatenating any
-    // key IDs found.
-    for (const auto& child : children) {
-      if (child.system_id == common_system_id && child.key_ids.size() > 0)
-        result.insert(result.end(), child.key_ids.begin(), child.key_ids.end());
+      kCencCommonSystemId,
+      kCencCommonSystemId + arraysize(kCencCommonSystemId));
+  for (const auto& child : children) {
+    if (child.system_id == common_system_id) {
+      key_ids->assign(child.key_ids.begin(), child.key_ids.end());
+      return key_ids->size() > 0;
     }
   }
 
-  key_ids->swap(result);
-  return key_ids->size() > 0;
+  // No matching 'pssh' box found.
+  return false;
 }
 
 bool GetPsshData(const std::vector<uint8_t>& input,
                  const std::vector<uint8_t>& system_id,
                  std::vector<uint8_t>* pssh_data) {
   if (input.empty())
     return false;
 
   std::vector<mp4::FullProtectionSystemSpecificHeader> children;
   if (!ReadAllPsshBoxes(input, &children))
     return false;
 
   // Check all children for an appropriate 'pssh' box, returning |data| from
   // the first one found.
   for (const auto& child : children) {
     if (child.system_id == system_id) {
       pssh_data->assign(child.data.begin(), child.data.end());
       return true;
     }
   }
 
   // No matching 'pssh' box found.
   return false;
 }
 
 }  // namespace media