1 files changed, 11 insertions, 1 deletions
diff --git a/content/renderer/media/webmediasourceclient_impl.cc b/content/renderer/media/webmediasourceclient_impl.cc
index 1aef773..66a3f02 100644
--- a/content/renderer/media/webmediasourceclient_impl.cc
+++ b/content/renderer/media/webmediasourceclient_impl.cc
@@ -60,8 +60,14 @@ void WebMediaSourceClientImpl::setDuration(double new_duration) {
   demuxer_->SetDuration(new_duration);
 }
 
+// TODO(acolwell): Remove this once endOfStream() is removed from Blink.
 void WebMediaSourceClientImpl::endOfStream(
     WebMediaSourceClient::EndOfStreamStatus status) {
+  markEndOfStream(status);
+}
+
+void WebMediaSourceClientImpl::markEndOfStream(
+    WebMediaSourceClient::EndOfStreamStatus status) {
   media::PipelineStatus pipeline_status = media::PIPELINE_OK;
 
   switch (status) {
@@ -77,7 +83,11 @@ void WebMediaSourceClientImpl::endOfStream(
       NOTIMPLEMENTED();
   }
 
-  demuxer_->EndOfStream(pipeline_status);
+  demuxer_->MarkEndOfStream(pipeline_status);
+}
+
+void WebMediaSourceClientImpl::unmarkEndOfStream() {
+  demuxer_->UnmarkEndOfStream();
 }
 
 }  // namespace content