Implement "Key Presence" step in "Encrypted Block Encounted" algorithm in EME.

media/crypto/aes_decryptor_unittest.cc

 // Copyright (c) 2012 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 <string>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/sys_byteorder.h"
 #include "media/base/decoder_buffer.h"
@@ skipping 222 matching lines @@
     EXPECT_CALL(*this, BufferDecrypted(AesDecryptor::kSuccess, NotNull()))
         .WillOnce(SaveArg<1>(&decrypted));
 
     decryptor_.Decrypt(encrypted_data, decrypt_cb_);
     ASSERT_TRUE(decrypted);
     ASSERT_EQ(plain_text_size, decrypted->GetDataSize());
     EXPECT_EQ(0, memcmp(plain_text, decrypted->GetData(), plain_text_size));
   }
 
   void DecryptAndExpectToFail(const uint8* data, int data_size,
-                              const uint8* plain_text, int plain_text_size,
                               const uint8* key_id, int key_id_size) {
     scoped_refptr<DecoderBuffer> encrypted_data =
         CreateWebMEncryptedBuffer(data, data_size, key_id, key_id_size);
     EXPECT_CALL(*this, BufferDecrypted(AesDecryptor::kError, IsNull()));
     decryptor_.Decrypt(encrypted_data, decrypt_cb_);
   }
 
   scoped_refptr<DecoderBuffer> encrypted_data_;
   MockDecryptorClient client_;
   AesDecryptor decryptor_;
@@ skipping 14 matching lines @@
                                                     frame.key_id_size));
 }
 
 TEST_F(AesDecryptorTest, WrongKey) {
   const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
   GenerateKeyRequest(frame.key_id, frame.key_id_size);
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            kWebmWrongKey, arraysize(kWebmWrongKey));
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToFail(frame.encrypted_data,
                                                  frame.encrypted_data_size,
-                                                 frame.plain_text,
-                                                 frame.plain_text_size,
                                                  frame.key_id,
                                                  frame.key_id_size));
 }
 
+TEST_F(AesDecryptorTest, NoKey) {
+  const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
+  GenerateKeyRequest(frame.key_id, frame.key_id_size);
+
+  scoped_refptr<DecoderBuffer> encrypted_data =
+      CreateWebMEncryptedBuffer(frame.encrypted_data, frame.encrypted_data_size,
+                                frame.key_id, frame.key_id_size);
+  EXPECT_CALL(*this, BufferDecrypted(AesDecryptor::kNoKey, IsNull()));
+  decryptor_.Decrypt(encrypted_data, decrypt_cb_);
+}
+
 TEST_F(AesDecryptorTest, KeyReplacement) {
   const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
   GenerateKeyRequest(frame.key_id, frame.key_id_size);
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            kWebmWrongKey, arraysize(kWebmWrongKey));
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToFail(frame.encrypted_data,
                                                  frame.encrypted_data_size,
-                                                 frame.plain_text,
-                                                 frame.plain_text_size,
                                                  frame.key_id,
                                                  frame.key_id_size));
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            frame.key, frame.key_size);
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToSucceed(frame.encrypted_data,
                                                     frame.encrypted_data_size,
                                                     frame.plain_text,
                                                     frame.plain_text_size,
                                                     frame.key_id,
                                                     frame.key_id_size));
@@ skipping 39 matching lines @@
                                                     frame2.key_id_size));
 }
 
 TEST_F(AesDecryptorTest, HmacCheckFailure) {
   const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
   GenerateKeyRequest(frame.key_id, frame.key_id_size);
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            frame.key, frame.key_size);
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToFail(kWebmFrame0HmacDataChanged,
                                                  frame.encrypted_data_size,
-                                                 frame.plain_text,
-                                                 frame.plain_text_size,
                                                  frame.key_id,
                                                  frame.key_id_size));
 }
 
 TEST_F(AesDecryptorTest, IvCheckFailure) {
   const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
   GenerateKeyRequest(frame.key_id, frame.key_id_size);
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            frame.key, frame.key_size);
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToFail(kWebmFrame0IvDataChanged,
                                                  frame.encrypted_data_size,
-                                                 frame.plain_text,
-                                                 frame.plain_text_size,
                                                  frame.key_id,
                                                  frame.key_id_size));
 }
 
 TEST_F(AesDecryptorTest, DataCheckFailure) {
   const WebmEncryptedData& frame = kWebmEncryptedFrames[0];
   GenerateKeyRequest(frame.key_id, frame.key_id_size);
   AddKeyAndExpectToSucceed(frame.key_id, frame.key_id_size,
                            frame.key, frame.key_size);
   ASSERT_NO_FATAL_FAILURE(DecryptAndExpectToFail(kWebmFrame0FrameDataChanged,
                                                  frame.encrypted_data_size,
-                                                 frame.plain_text,
-                                                 frame.plain_text_size,
                                                  frame.key_id,
                                                  frame.key_id_size));
 }
 
 }  // namespace media