Sanitize data before providing it to the CDM

media/cdm/json_web_key_unittest.cc

 // Copyright 2013 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/json_web_key.h"
 
 #include "base/base64.h"
 #include "base/logging.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 53 matching lines @@
       VerifyKeyId(key, expected_key, expected_key_length);
   }
 
   void VerifyKeyId(std::vector<uint8> key,
                    const uint8* expected_key,
                    int expected_key_length) {
     std::vector<uint8> key_result(expected_key,
                                   expected_key + expected_key_length);
     EXPECT_EQ(key_result, key);
   }
+
+  KeyIdAndKeyPair MakeKeyIdAndKeyPair(const uint8* key,
+                                      int key_length,
+                                      const uint8* key_id,
+                                      int key_id_length) {
+    return std::make_pair(std::string(key_id, key_id + key_id_length),
+                          std::string(key, key + key_length));
+  }
 };
 
 TEST_F(JSONWebKeyTest, GenerateJWKSet) {
   const uint8 data1[] = { 0x01, 0x02 };
   const uint8 data2[] = { 0x01, 0x02, 0x03, 0x04 };
   const uint8 data3[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
                           0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10 };
 
   EXPECT_EQ("{\"keys\":[{\"k\":\"AQI\",\"kid\":\"AQI\",\"kty\":\"oct\"}]}",
             GenerateJWKSet(data1, arraysize(data1), data1, arraysize(data1)));
   EXPECT_EQ(
       "{\"keys\":[{\"k\":\"AQIDBA\",\"kid\":\"AQIDBA\",\"kty\":\"oct\"}]}",
       GenerateJWKSet(data2, arraysize(data2), data2, arraysize(data2)));
   EXPECT_EQ("{\"keys\":[{\"k\":\"AQI\",\"kid\":\"AQIDBA\",\"kty\":\"oct\"}]}",
             GenerateJWKSet(data1, arraysize(data1), data2, arraysize(data2)));
   EXPECT_EQ("{\"keys\":[{\"k\":\"AQIDBA\",\"kid\":\"AQI\",\"kty\":\"oct\"}]}",
             GenerateJWKSet(data2, arraysize(data2), data1, arraysize(data1)));
   EXPECT_EQ(
       "{\"keys\":[{\"k\":\"AQIDBAUGBwgJCgsMDQ4PEA\",\"kid\":"
       "\"AQIDBAUGBwgJCgsMDQ4PEA\",\"kty\":\"oct\"}]}",
       GenerateJWKSet(data3, arraysize(data3), data3, arraysize(data3)));
+
+  KeyIdAndKeyPairs keys;
+  keys.push_back(
+      MakeKeyIdAndKeyPair(data1, arraysize(data1), data1, arraysize(data1)));
+  EXPECT_EQ(
+      "{\"keys\":[{\"k\":\"AQI\",\"kid\":\"AQI\",\"kty\":\"oct\"}],\"type\":"
+      "\"temporary\"}",
+      GenerateJWKSet(keys, MediaKeys::TEMPORARY_SESSION));
+  keys.push_back(
+      MakeKeyIdAndKeyPair(data2, arraysize(data2), data2, arraysize(data2)));
+  EXPECT_EQ(
+      "{\"keys\":[{\"k\":\"AQI\",\"kid\":\"AQI\",\"kty\":\"oct\"},{\"k\":"
+      "\"AQIDBA\",\"kid\":\"AQIDBA\",\"kty\":\"oct\"}],\"type\":\"persistent-"
+      "license\"}",
+      GenerateJWKSet(keys, MediaKeys::PERSISTENT_LICENSE_SESSION));
+  keys.push_back(
+      MakeKeyIdAndKeyPair(data3, arraysize(data3), data3, arraysize(data3)));
+  EXPECT_EQ(
+      "{\"keys\":[{\"k\":\"AQI\",\"kid\":\"AQI\",\"kty\":\"oct\"},{\"k\":"
+      "\"AQIDBA\",\"kid\":\"AQIDBA\",\"kty\":\"oct\"},{\"k\":"
+      "\"AQIDBAUGBwgJCgsMDQ4PEA\",\"kid\":\"AQIDBAUGBwgJCgsMDQ4PEA\",\"kty\":"
+      "\"oct\"}],\"type\":\"persistent-release-message\"}",
+      GenerateJWKSet(keys, MediaKeys::PERSISTENT_RELEASE_MESSAGE_SESSION));
 }
 
 TEST_F(JSONWebKeyTest, ExtractValidJWKKeys) {
   // Try an empty 'keys' dictionary.
   ExtractJWKKeysAndExpect("{ \"keys\": [] }", true, 0);
 
   // Try a key list with one entry.
   const std::string kJwksOneEntry =
       "{"
       "  \"keys\": ["
@@ skipping 495 matching lines @@
   key_ids.push_back(std::vector<uint8>(data3, data3 + arraysize(data3)));
   std::vector<uint8> init_data3;
   CreateKeyIdsInitData(key_ids, &init_data3);
   std::string result3(init_data3.begin(), init_data3.end());
   EXPECT_EQ(result3,
             "{\"kids\":[\"AQI\",\"AQIDBA\",\"AQIDBAUGBwgJCgsMDQ4PEA\"]}");
 }
 
 }  // namespace media