fix: Fix ClearKey license on old CDMs (#7816)

shaka-project · Dec 31, 2024 · fe1f35b · fe1f35b
1 parent 626591e
commit fe1f35b
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 0 deletions.
diff --git a/lib/drm/drm_engine.js b/lib/drm/drm_engine.js
@@ -1525,6 +1525,11 @@ shaka.drm.DrmEngine = class {
     }
     // NOTE: allowCrossSiteCredentials can be set in a request filter.
 
+    if (shaka.drm.DrmUtils.isClearKeySystem(
+        this.currentDrmInfo_.keySystem)) {
+      this.fixClearKeyRequest_(request, this.currentDrmInfo_);
+    }
+
     if (shaka.drm.DrmUtils.isPlayReadyKeySystem(
         this.currentDrmInfo_.keySystem)) {
       this.unpackPlayReadyRequest_(request);
@@ -1678,6 +1683,31 @@ shaka.drm.DrmEngine = class {
         /** @type{string} */(shaka.util.TXml.getTextContents(challenge)));
   }
 
+  /**
+   * Some old ClearKey CDMs don't include the type in the body request.
+   *
+   * @param {shaka.extern.Request} request
+   * @param {shaka.extern.DrmInfo} drmInfo
+   * @private
+   */
+  fixClearKeyRequest_(request, drmInfo) {
+    try {
+      const body = shaka.util.StringUtils.fromBytesAutoDetect(request.body);
+      if (body) {
+        const licenseBody =
+          /** @type {shaka.drm.DrmEngine.ClearKeyLicenceRequestFormat} */ (
+            JSON.parse(body));
+        if (!licenseBody.type) {
+          licenseBody.type = drmInfo.sessionType;
+          request.body =
+              shaka.util.StringUtils.toUTF8(JSON.stringify(licenseBody));
+        }
+      }
+    } catch (e) {
+      shaka.log.info('Error unpacking ClearKey license', e);
+    }
+  }
+
   /**
    * @param {!Event} event
    * @private
@@ -2729,6 +2759,21 @@ shaka.drm.DrmEngine.SessionMetaData;
  */
 shaka.drm.DrmEngine.PlayerInterface;
 
+/**
+ * @typedef {{
+ *   kids: !Array.<string>,
+ *   type: string
+ * }}
+ *
+ * @property {!Array.<string>} kids
+ *   An array of key IDs. Each element of the array is the base64url encoding of
+ *   the octet sequence containing the key ID value.
+ * @property {string} type
+ *   The requested MediaKeySessionType.
+ * @see https://www.w3.org/TR/encrypted-media/#clear-key-request-format
+ */
+shaka.drm.DrmEngine.ClearKeyLicenceRequestFormat;
+
 /**
  * The amount of time, in seconds, we wait to consider a session closed.
  * This allows us to work around Chrome bug https://crbug.com/1108158.

diff --git a/lib/drm/drm_utils.js b/lib/drm/drm_utils.js
@@ -113,6 +113,14 @@ shaka.drm.DrmUtils = class {
     return drmInfo ? drmInfo.keySystem : '';
   }
 
+  /**
+   * @param {?string} keySystem
+   * @return {boolean}
+   */
+  static isClearKeySystem(keySystem) {
+    return keySystem === 'org.w3.clearkey';
+  }
+
   /**
    * @param {?string} keySystem
    * @return {boolean}

diff --git a/test/drm/drm_utils_unit.js b/test/drm/drm_utils_unit.js
@@ -145,6 +145,22 @@ describe('DrmUtils', () => {
     });
   }); // describe('getCommonDrmInfos')
 
+  describe('isClearKeySystem', () => {
+    it('should return true for ClearKey', () => {
+      expect(shaka.drm.DrmUtils.isClearKeySystem(
+          'org.w3.clearkey')).toBe(true);
+    });
+
+    it('should return false for non-ClearKey key systems', () => {
+      expect(shaka.drm.DrmUtils.isClearKeySystem(
+          'com.widevine.alpha')).toBe(false);
+      expect(shaka.drm.DrmUtils.isClearKeySystem(
+          'com.microsoft.playready')).toBe(false);
+      expect(shaka.drm.DrmUtils.isClearKeySystem(
+          'com.apple.fps')).toBe(false);
+    });
+  });
+
   describe('isWidevineKeySystem', () => {
     it('should return true for Widevine', () => {
       expect(shaka.drm.DrmUtils.isWidevineKeySystem(