Merge "Disallow restoring an offline license multiple times" into sc-dev

libwvdrmengine/cdm/core/include/cdm_session.h

@@ -312,6 +312,9 @@ class CdmSession {
   // license type release and offline related information
   CdmKeySetId key_set_id_;
 
+  bool has_license_been_loaded_ = false;
+  bool has_license_been_restored_ = false;
+
   bool mock_license_parser_in_use_;
   bool mock_policy_engine_in_use_;
 

libwvdrmengine/cdm/core/include/wv_cdm_types.h

@@ -414,7 +414,7 @@ enum CdmResponseType : int32_t {
   SHRINK_USAGE_TABLE_HEADER_ENTRY_IN_USE = 359,
   LICENSE_USAGE_ENTRY_MISSING = 360,
   LOAD_USAGE_ENTRY_INVALID_SESSION = 361,
-  // previously RESTORE_OFFLINE_LICENSE_ERROR_3 = 362,
+  RESTORE_OFFLINE_LICENSE_ERROR_3 = 362,
   NO_SRM_VERSION = 363,
   SESSION_NOT_FOUND_23 = 364,
   CERT_PROVISIONING_RESPONSE_ERROR_9 = 365,

libwvdrmengine/cdm/core/src/cdm_session.cpp

@@ -215,6 +215,12 @@ CdmResponseType CdmSession::RestoreOfflineSession(const CdmKeySetId& key_set_id,
   if (!key_set_id_.empty()) {
     file_handle_->UnreserveLicenseId(key_set_id_);
   }
+  if (has_license_been_loaded_ || has_license_been_restored_) {
+    LOGE(
+        "Disallow multiple offline license restores or restoring a license if "
+        "a license has already been loaded");
+    return RESTORE_OFFLINE_LICENSE_ERROR_3;
+  }
 
   key_set_id_ = key_set_id;
 
@@ -338,6 +344,7 @@ CdmResponseType CdmSession::RestoreOfflineSession(const CdmKeySetId& key_set_id,
   license_received_ = true;
   is_offline_ = true;
   is_release_ = license_type == kLicenseTypeRelease;
+  has_license_been_restored_ = true;
   return KEY_ADDED;
 }
 
@@ -582,6 +589,7 @@ CdmResponseType CdmSession::AddKeyInternal(const CdmKeyResponse& key_response) {
     if (sts != NO_ERROR) return sts;
   }
 
+  has_license_been_loaded_ = true;
   return KEY_ADDED;
 }