2 files changed, 15 insertions, 2 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
diff --git a/content/renderer/media/webmediasourceclient_impl.h b/content/renderer/media/webmediasourceclient_impl.h
index 4607dd3..bebdd54 100644
--- a/content/renderer/media/webmediasourceclient_impl.h
+++ b/content/renderer/media/webmediasourceclient_impl.h
@@ -29,7 +29,10 @@ class WebMediaSourceClientImpl : public WebKit::WebMediaSourceClient {
       WebKit::WebSourceBuffer** source_buffer) OVERRIDE;
   virtual double duration() OVERRIDE;
   virtual void setDuration(double duration) OVERRIDE;
-  virtual void endOfStream(EndOfStreamStatus status) OVERRIDE;
+  // TODO(acolwell): Remove this once endOfStream() is removed from Blink.
+  virtual void endOfStream(EndOfStreamStatus status);
+  virtual void markEndOfStream(EndOfStreamStatus status) OVERRIDE;
+  virtual void unmarkEndOfStream() OVERRIDE;
 
  private:
   media::ChunkDemuxer* demuxer_;  // Owned by WebMediaPlayerImpl.