chore: Add encrypted flag to init segments

shaka-project · Feb 27, 2025 · 0d5980d · 0d5980d
1 parent 6c01f18
commit 0d5980d
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 7 deletions.
diff --git a/lib/hls/hls_parser.js b/lib/hls/hls_parser.js
@@ -3604,19 +3604,23 @@ shaka.hls.HlsParser = class {
       /** @type {shaka.extern.aesKey|undefined} */
       let aesKey = undefined;
       let byteRangeTag = null;
+      let encrypted = false;
       for (const tag of tags) {
         if (tag.name == 'EXT-X-KEY') {
-          if (this.isAesMethod_(tag.getRequiredAttrValue('METHOD')) &&
-              tag.id < mapTag.id) {
+          const method = tag.getRequiredAttrValue('METHOD');
+          if (this.isAesMethod_(method) && tag.id < mapTag.id) {
+            encrypted = false;
             aesKey =
                 this.parseAESDrmTag_(tag, playlist, getUris, variables);
+          } else {
+            encrypted = method != 'NONE';
           }
         } else if (tag.name == 'EXT-X-BYTERANGE' && tag.id < mapTag.id) {
           byteRangeTag = tag;
         }
       }
       const initSegmentRef = this.createInitSegmentReference_(
-          absoluteInitSegmentUris, mapTag, byteRangeTag, aesKey);
+          absoluteInitSegmentUris, mapTag, byteRangeTag, aesKey, encrypted);
       this.mapTagToInitSegmentRefMap_.set(mapTagKey, initSegmentRef);
     }
     return this.mapTagToInitSegmentRefMap_.get(mapTagKey);
@@ -3629,11 +3633,12 @@ shaka.hls.HlsParser = class {
    * @param {!shaka.hls.Tag} mapTag EXT-X-MAP
    * @param {shaka.hls.Tag=} byteRangeTag EXT-X-BYTERANGE
    * @param {shaka.extern.aesKey=} aesKey
+   * @param {boolean=} encrypted
    * @return {!shaka.media.InitSegmentReference}
    * @private
    */
   createInitSegmentReference_(absoluteInitSegmentUris, mapTag, byteRangeTag,
-      aesKey) {
+      aesKey, encrypted) {
     let startByte = 0;
     let endByte = null;
     let byterange = mapTag.getAttributeValue('BYTERANGE');
@@ -3666,7 +3671,8 @@ shaka.hls.HlsParser = class {
         /* mediaQuality= */ null,
         /* timescale= */ null,
         /* segmentData= */ null,
-        aesKey);
+        aesKey,
+        encrypted);
     return initSegmentRef;
   }
 

diff --git a/lib/media/segment_reference.js b/lib/media/segment_reference.js
@@ -34,9 +34,10 @@ shaka.media.InitSegmentReference = class {
    * @param {(null|BufferSource)=} segmentData
    * @param {?shaka.extern.aesKey=} aesKey
    *  The segment's AES-128-CBC full segment encryption key and iv.
+   * @param {boolean=} encrypted
    */
   constructor(uris, startByte, endByte, mediaQuality = null, timescale = null,
-      segmentData = null, aesKey = null) {
+      segmentData = null, aesKey = null, encrypted = false) {
     /** @type {function(): !Array<string>} */
     this.getUris = uris;
 
@@ -63,6 +64,9 @@ shaka.media.InitSegmentReference = class {
 
     /** @type {?string} */
     this.mimeType = null;
+
+    /** @const {boolean} */
+    this.encrypted = encrypted;
   }
 
   /**

diff --git a/lib/mss/mss_parser.js b/lib/mss/mss_parser.js
@@ -731,7 +731,9 @@ shaka.mss.MssParser = class {
           /* endByte= */ null,
           qualityInfo,
           stream.mssPrivateData.timescale,
-          initSegmentData);
+          initSegmentData,
+          /* aesKey= */ null,
+          stream.encrypted);
 
       const segments = this.createSegments_(initSegmentRef,
           stream, streamIndex, timeline);