fix(HLS): Fix SAMPLE-AES playback

shaka-project · Apr 4, 2024 · 04b65cb · 04b65cb
1 parent 25427c7
commit 04b65cb
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/lib/media/drm_engine.js b/lib/media/drm_engine.js
@@ -2314,9 +2314,15 @@ shaka.media.DrmEngine = class {
          shaka.util.BufferUtils.equal(a.initData, b.initData);
     };
 
+    const clearkeyDataStart = 'data:application/json;base64,';
+    const clearKeyLicenseServers = [];
+
     for (const drmInfo of drmInfos) {
       // Build an array of unique license servers.
-      if (!licenseServers.includes(drmInfo.licenseServerUri)) {
+      if (drmInfo.keySystem == 'org.w3.clearkey' &&
+          drmInfo.licenseServerUri.startsWith(clearkeyDataStart)) {
+        clearKeyLicenseServers.push(drmInfo.licenseServerUri);
+      } else if (!licenseServers.includes(drmInfo.licenseServerUri)) {
         licenseServers.push(drmInfo.licenseServerUri);
       }
 
@@ -2353,6 +2359,21 @@ shaka.media.DrmEngine = class {
         }
       }
     }
+
+    if (clearKeyLicenseServers.length == 1) {
+      licenseServers.push(clearKeyLicenseServers[0]);
+    } else if (clearKeyLicenseServers.length > 0) {
+      const keys = [];
+      for (const clearKeyLicenseServer of clearKeyLicenseServers) {
+        const license = window.atob(
+            clearKeyLicenseServer.split(clearkeyDataStart).pop());
+        const jwkSet = /** @type {{keys: !Array}} */(JSON.parse(license));
+        keys.push(...jwkSet.keys);
+      }
+      const newJwkSet = {keys: keys};
+      const newLicense = JSON.stringify(newJwkSet);
+      licenseServers.push(clearkeyDataStart + window.btoa(newLicense));
+    }
   }
 
   /**