fix: Fix reset MSE to last independent segment (#7494)

shaka-project · Nov 12, 2024 · d666fbf · d666fbf
1 parent ad69e08
commit d666fbf
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/lib/media/segment_index.js b/lib/media/segment_index.js
@@ -679,6 +679,26 @@ shaka.media.SegmentIterator = class {
       'done': !res,
     };
   }
+
+  /**
+   * @export
+   */
+  resetToLastIndependent() {
+    const current = this.current();
+    if (current.isPartial() && !current.isIndependent()) {
+      const ref = this.segmentIndex_.get(this.currentPosition_);
+      if (ref && ref.hasPartialSegments()) {
+        let partial = ref.partialReferences[this.currentPartialPosition_];
+        while (partial.isIndependent()) {
+          if (this.currentPartialPosition_ <= 0) {
+            break;
+          }
+          this.currentPartialPosition_--;
+          partial = ref.partialReferences[this.currentPartialPosition_];
+        }
+      }
+    }
+  }
 };
 
 

diff --git a/lib/media/streaming_engine.js b/lib/media/streaming_engine.js
@@ -2871,12 +2871,18 @@ shaka.media.StreamingEngine = class {
       audioMediaState.lastInitSegmentReference = null;
       this.forceClearBuffer_(audioMediaState);
       this.abortOperations_(audioMediaState).catch(() => {});
+      if (audioMediaState.segmentIterator) {
+        audioMediaState.segmentIterator.resetToLastIndependent();
+      }
     }
     const videoMediaState = this.mediaStates_.get(ContentType.VIDEO);
     if (videoMediaState) {
       videoMediaState.lastInitSegmentReference = null;
       this.forceClearBuffer_(videoMediaState);
       this.abortOperations_(videoMediaState).catch(() => {});
+      if (videoMediaState.segmentIterator) {
+        videoMediaState.segmentIterator.resetToLastIndependent();
+      }
     }
     /**
      * @type {!Map.<shaka.util.ManifestParserUtils.ContentType,

diff --git a/test/test/util/simple_fakes.js b/test/test/util/simple_fakes.js
@@ -513,6 +513,8 @@ shaka.test.FakeSegmentIndex = class {
             },
 
             setReverse: () => {},
+
+            resetToLastIndependent: () => {},
           };
         });
   }