aes_decryptor.cc revision ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16
1ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch// Copyright 2013 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch#include "media/cdm/aes_decryptor.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h" 11868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_number_conversions.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "crypto/encryptor.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "crypto/symmetric_key.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/audio_decoder_config.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/decoder_buffer.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/decrypt_config.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/video_decoder_config.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/video_frame.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace media { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)uint32 AesDecryptor::next_session_id_ = 1; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum ClearBytesBufferSel { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kSrcContainsClearBytes, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kDstContainsClearBytes 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static void CopySubsamples(const std::vector<SubsampleEntry>& subsamples, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const ClearBytesBufferSel sel, 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const uint8* src, 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint8* dst) { 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < subsamples.size(); i++) { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const SubsampleEntry& subsample = subsamples[i]; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (sel == kSrcContainsClearBytes) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) src += subsample.clear_bytes; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dst += subsample.clear_bytes; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memcpy(dst, src, subsample.cypher_bytes); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) src += subsample.cypher_bytes; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dst += subsample.cypher_bytes; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Decrypts |input| using |key|. Returns a DecoderBuffer with the decrypted 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// data if decryption succeeded or NULL if decryption failed. 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static scoped_refptr<DecoderBuffer> DecryptData(const DecoderBuffer& input, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crypto::SymmetricKey* key) { 50ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch CHECK(input.data_size()); 51ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch CHECK(input.decrypt_config()); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(key); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crypto::Encryptor encryptor; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!encryptor.Init(key, crypto::Encryptor::CTR, "")) { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not initialize decryptor."; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 60ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch DCHECK_EQ(input.decrypt_config()->iv().size(), 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<size_t>(DecryptConfig::kDecryptionKeySize)); 62ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch if (!encryptor.SetCounter(input.decrypt_config()->iv())) { 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not set counter block."; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 67ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch const int data_offset = input.decrypt_config()->data_offset(); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* sample = 69ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch reinterpret_cast<const char*>(input.data() + data_offset); 70ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch int sample_size = input.data_size() - data_offset; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 72a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) DCHECK_GT(sample_size, 0) << "No sample data to be decrypted."; 73a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (sample_size <= 0) 74a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) return NULL; 75a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 76ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch if (input.decrypt_config()->subsamples().empty()) { 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string decrypted_text; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringPiece encrypted_text(sample, sample_size); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!encryptor.Decrypt(encrypted_text, &decrypted_text)) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not decrypt data."; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(xhwang): Find a way to avoid this data copy. 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return DecoderBuffer::CopyFrom( 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<const uint8*>(decrypted_text.data()), 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) decrypted_text.size()); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SubsampleEntry>& subsamples = 91ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch input.decrypt_config()->subsamples(); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int total_clear_size = 0; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int total_encrypted_size = 0; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < subsamples.size(); i++) { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_clear_size += subsamples[i].clear_bytes; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_encrypted_size += subsamples[i].cypher_bytes; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (total_clear_size + total_encrypted_size != sample_size) { 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Subsample sizes do not equal input size"; 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 104a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // No need to decrypt if there is no encrypted data. 105a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (total_encrypted_size <= 0) { 106a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) return DecoderBuffer::CopyFrom(reinterpret_cast<const uint8*>(sample), 107a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) sample_size); 108a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) } 109a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The encrypted portions of all subsamples must form a contiguous block, 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // such that an encrypted subsample that ends away from a block boundary is 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // immediately followed by the start of the next encrypted subsample. We 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // copy all encrypted subsamples to a contiguous buffer, decrypt them, then 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // copy the decrypted bytes over the encrypted bytes in the output. 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(strobe): attempt to reduce number of memory copies 116c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<uint8[]> encrypted_bytes(new uint8[total_encrypted_size]); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CopySubsamples(subsamples, kSrcContainsClearBytes, 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<const uint8*>(sample), encrypted_bytes.get()); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringPiece encrypted_text( 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<const char*>(encrypted_bytes.get()), 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_encrypted_size); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string decrypted_text; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!encryptor.Decrypt(encrypted_text, &decrypted_text)) { 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not decrypt data."; 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 128a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) DCHECK_EQ(decrypted_text.size(), encrypted_text.size()); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DecoderBuffer> output = DecoderBuffer::CopyFrom( 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<const uint8*>(sample), sample_size); 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CopySubsamples(subsamples, kDstContainsClearBytes, 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<const uint8*>(decrypted_text.data()), 134ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch output->writable_data()); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return output; 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)AesDecryptor::AesDecryptor(const KeyAddedCB& key_added_cb, 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const KeyErrorCB& key_error_cb, 140eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch const KeyMessageCB& key_message_cb) 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : key_added_cb_(key_added_cb), 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key_error_cb_(key_error_cb), 143eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch key_message_cb_(key_message_cb) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AesDecryptor::~AesDecryptor() { 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteValues(&key_map_); 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)bool AesDecryptor::GenerateKeyRequest(const std::string& type, 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const uint8* init_data, 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int init_data_length) { 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string session_id_string(base::UintToString(next_session_id_++)); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1557d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // For now, the AesDecryptor does not care about |type|; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // just fire the event with the |init_data| as the request. 157eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch std::vector<uint8> message; 158eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch if (init_data && init_data_length) 159eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch message.assign(init_data, init_data + init_data_length); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1617d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) key_message_cb_.Run(session_id_string, message, std::string()); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1657d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)void AesDecryptor::AddKey(const uint8* key, 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int key_length, 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const uint8* init_data, 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int init_data_length, 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& session_id) { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(key); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK_GT(key_length, 0); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(xhwang): Add |session_id| check after we figure out how: 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // https://www.w3.org/Bugs/Public/show_bug.cgi?id=16550 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (key_length != DecryptConfig::kDecryptionKeySize) { 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Invalid key length: " << key_length; 1777d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) key_error_cb_.Run(session_id, MediaKeys::kUnknownError, 0); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(xhwang): Fix the decryptor to accept no |init_data|. See 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // http://crbug.com/123265. Until then, ensure a non-empty value is passed. 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const uint8 kDummyInitData[1] = { 0 }; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!init_data) { 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) init_data = kDummyInitData; 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) init_data_length = arraysize(kDummyInitData); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(xhwang): For now, use |init_data| for key ID. Make this more spec 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // compliant later (http://crbug.com/123262, http://crbug.com/123265). 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string key_id_string(reinterpret_cast<const char*>(init_data), 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) init_data_length); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string key_string(reinterpret_cast<const char*>(key) , key_length); 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<DecryptionKey> decryption_key(new DecryptionKey(key_string)); 195c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!decryption_key) { 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not create key."; 1977d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) key_error_cb_.Run(session_id, MediaKeys::kUnknownError, 0); 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!decryption_key->Init()) { 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Could not initialize decryption key."; 2037d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) key_error_cb_.Run(session_id, MediaKeys::kUnknownError, 0); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetKey(key_id_string, decryption_key.Pass()); 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!new_audio_key_cb_.is_null()) 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_audio_key_cb_.Run(); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!new_video_key_cb_.is_null()) 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_video_key_cb_.Run(); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2157d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) key_added_cb_.Run(session_id); 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)void AesDecryptor::CancelKeyRequest(const std::string& session_id) { 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 221eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochDecryptor* AesDecryptor::GetDecryptor() { 222868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return this; 223868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 224868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AesDecryptor::RegisterNewKeyCB(StreamType stream_type, 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const NewKeyCB& new_key_cb) { 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (stream_type) { 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case kAudio: 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_audio_key_cb_ = new_key_cb; 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case kVideo: 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_video_key_cb_ = new_key_cb; 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::Decrypt(StreamType stream_type, 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<DecoderBuffer>& encrypted, 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const DecryptCB& decrypt_cb) { 242ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch CHECK(encrypted->decrypt_config()); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DecoderBuffer> decrypted; 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // An empty iv string signals that the frame is unencrypted. 246ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch if (encrypted->decrypt_config()->iv().empty()) { 247ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch int data_offset = encrypted->decrypt_config()->data_offset(); 248ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch decrypted = DecoderBuffer::CopyFrom(encrypted->data() + data_offset, 249ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch encrypted->data_size() - data_offset); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 251ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch const std::string& key_id = encrypted->decrypt_config()->key_id(); 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DecryptionKey* key = GetKey(key_id); 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!key) { 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DVLOG(1) << "Could not find a matching key for the given key ID."; 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) decrypt_cb.Run(kNoKey, NULL); 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crypto::SymmetricKey* decryption_key = key->decryption_key(); 260868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) decrypted = DecryptData(*encrypted.get(), decryption_key); 261868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!decrypted.get()) { 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Decryption failed."; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) decrypt_cb.Run(kError, NULL); 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 268ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch decrypted->set_timestamp(encrypted->timestamp()); 269ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch decrypted->set_duration(encrypted->duration()); 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) decrypt_cb.Run(kSuccess, decrypted); 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::CancelDecrypt(StreamType stream_type) { 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Decrypt() calls the DecryptCB synchronously so there's nothing to cancel. 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AesDecryptor::InitializeAudioDecoder(const AudioDecoderConfig& config, 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const DecoderInitCB& init_cb) { 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AesDecryptor does not support audio decoding. 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) init_cb.Run(false); 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AesDecryptor::InitializeVideoDecoder(const VideoDecoderConfig& config, 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const DecoderInitCB& init_cb) { 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AesDecryptor does not support video decoding. 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) init_cb.Run(false); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::DecryptAndDecodeAudio( 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<DecoderBuffer>& encrypted, 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AudioDecodeCB& audio_decode_cb) { 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "AesDecryptor does not support audio decoding"; 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::DecryptAndDecodeVideo( 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const scoped_refptr<DecoderBuffer>& encrypted, 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const VideoDecodeCB& video_decode_cb) { 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "AesDecryptor does not support video decoding"; 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::ResetDecoder(StreamType stream_type) { 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "AesDecryptor does not support audio/video decoding"; 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::DeinitializeDecoder(StreamType stream_type) { 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "AesDecryptor does not support audio/video decoding"; 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AesDecryptor::SetKey(const std::string& key_id, 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<DecryptionKey> decryption_key) { 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::AutoLock auto_lock(key_map_lock_); 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) KeyMap::iterator found = key_map_.find(key_id); 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (found != key_map_.end()) { 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete found->second; 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_map_.erase(found); 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_map_[key_id] = decryption_key.release(); 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AesDecryptor::DecryptionKey* AesDecryptor::GetKey( 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& key_id) const { 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::AutoLock auto_lock(key_map_lock_); 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) KeyMap::const_iterator found = key_map_.find(key_id); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (found == key_map_.end()) 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return found->second; 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AesDecryptor::DecryptionKey::DecryptionKey(const std::string& secret) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : secret_(secret) { 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AesDecryptor::DecryptionKey::~DecryptionKey() {} 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AesDecryptor::DecryptionKey::Init() { 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(!secret_.empty()); 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) decryption_key_.reset(crypto::SymmetricKey::Import( 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crypto::SymmetricKey::AES, secret_)); 340c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!decryption_key_) 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace media 346