e9dfca1219
Merge from Widevine repo of http://go/wvgerrit/100385 The map now contains unique_ptr instead of raw pointers to ensure the memory is released. Bug: 156780432 OEMCrypto Fuzzing: Fix OEMCrypto Memory Leak. Test: oemcrypto reference code only Change-Id: I78054f9207399f052d6e4bfdfa96824f6e050bac
106 lines
3.1 KiB
C++
106 lines
3.1 KiB
C++
// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
|
|
// source code may only be used and distributed under the Widevine Master
|
|
// License Agreement.
|
|
//
|
|
// Reference implementation of OEMCrypto APIs
|
|
//
|
|
#include "oemcrypto_session_key_table.h"
|
|
|
|
#include "keys.h"
|
|
#include "log.h"
|
|
|
|
namespace wvoec_ref {
|
|
|
|
bool SessionKeyTable::Insert(const KeyId key_id, const Key& key_data) {
|
|
if (keys_.find(key_id) != keys_.end()) return false;
|
|
keys_[key_id] = std::unique_ptr<Key>(new Key(key_data));
|
|
return true;
|
|
}
|
|
|
|
Key* SessionKeyTable::Find(const KeyId key_id) {
|
|
if (keys_.find(key_id) == keys_.end()) {
|
|
return nullptr;
|
|
}
|
|
return keys_[key_id].get();
|
|
}
|
|
|
|
void SessionKeyTable::Remove(const KeyId key_id) {
|
|
if (keys_.find(key_id) != keys_.end()) {
|
|
keys_.erase(key_id);
|
|
}
|
|
}
|
|
|
|
void SessionKeyTable::UpdateDuration(const KeyControlBlock& control) {
|
|
for (KeyMap::iterator it = keys_.begin(); it != keys_.end(); ++it) {
|
|
it->second->UpdateDuration(control);
|
|
}
|
|
}
|
|
|
|
bool EntitlementKeyTable::Insert(const KeyId key_id, const Key& key_data) {
|
|
// |key_id| and |key_data| are for an entitlement key. Insert a new
|
|
// entitlement key entry.
|
|
if (keys_.find(key_id) != keys_.end()) return false;
|
|
keys_[key_id] = std::unique_ptr<EntitlementKey>(new EntitlementKey(key_data));
|
|
// If this is a new insertion, we don't have a content key assigned yet.
|
|
return true;
|
|
}
|
|
|
|
Key* EntitlementKeyTable::Find(const KeyId key_id) {
|
|
// |key_id| refers to a content key.
|
|
ContentIdToEntitlementIdMap::iterator it =
|
|
contentid_to_entitlementid_.find(key_id);
|
|
if (it == contentid_to_entitlementid_.end()) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (keys_.find(it->second) == keys_.end()) {
|
|
return nullptr;
|
|
}
|
|
return keys_[it->second].get();
|
|
}
|
|
|
|
void EntitlementKeyTable::Remove(const KeyId key_id) {
|
|
// |key_id| refers to a content key. No one currently calls Remove so this
|
|
// method is free to change if needed.
|
|
ContentIdToEntitlementIdMap::iterator it =
|
|
contentid_to_entitlementid_.find(key_id);
|
|
if (it == contentid_to_entitlementid_.end()) {
|
|
return;
|
|
}
|
|
keys_.erase(it->second);
|
|
contentid_to_entitlementid_.erase(key_id);
|
|
}
|
|
|
|
void EntitlementKeyTable::UpdateDuration(const KeyControlBlock& control) {
|
|
for (EntitlementKeyMap::iterator it = keys_.begin(); it != keys_.end();
|
|
++it) {
|
|
it->second->UpdateDuration(control);
|
|
}
|
|
}
|
|
|
|
bool EntitlementKeyTable::SetContentKey(
|
|
const KeyId& entitlement_id, const KeyId& content_key_id,
|
|
const std::vector<uint8_t> content_key) {
|
|
EntitlementKeyMap::iterator it = keys_.find(entitlement_id);
|
|
if (it == keys_.end()) {
|
|
return false;
|
|
}
|
|
contentid_to_entitlementid_.erase(it->second->content_key_id());
|
|
if (!it->second->SetContentKey(content_key_id, content_key)) {
|
|
return false;
|
|
}
|
|
contentid_to_entitlementid_[content_key_id] = entitlement_id;
|
|
return true;
|
|
}
|
|
|
|
EntitlementKey* EntitlementKeyTable::GetEntitlementKey(
|
|
const KeyId& entitlement_id) {
|
|
EntitlementKeyMap::iterator it = keys_.find(entitlement_id);
|
|
if (it == keys_.end()) {
|
|
return nullptr;
|
|
}
|
|
return it->second.get();
|
|
}
|
|
|
|
} // namespace wvoec_ref
|