Add Entitlement License to OEMCrypto

This CL adds entitlement license features and moves cipher mode from
LoadKeys to SelectKeys.

Merge from Widevine repo of http://go/wvgerrit/41660

bug: 70334840 Entitlement License - cdm layer
bug: 70334345 Entitlement License - reference code and unit tests

test: Entitlement license unit tests pass.
Change-Id: Ic7d7f42c15e6d83ef7fcfd8a866c778adc4c8095

libwvdrmengine/oemcrypto/mock/src/oemcrypto_session_key_table.cpp

@@ -43,4 +43,75 @@ void SessionKeyTable::UpdateDuration(const KeyControlBlock& 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] = 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 NULL;
+  }
+
+  if (keys_.find(it->second) == keys_.end()) {
+    return NULL;
+  }
+  return keys_[it->second];
+}
+
+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;
+}
+
+bool EntitlementKeyTable::GetEntitlementKey(
+    const KeyId& entitlement_id,
+    const std::vector<uint8_t>** entitlement_key) {
+  if (!entitlement_key) {
+    return false;
+  }
+  EntitlementKeyMap::iterator it = keys_.find(entitlement_id);
+  if (it == keys_.end()) {
+    return false;
+  }
+  *entitlement_key = &it->second->entitlement_key();
+  return true;
+}
+
 }  // namespace wvoec_mock