Allow setting memory limits on media::DemuxerStream

media/blink/webmediaplayer_impl.cc

Index: media/blink/webmediaplayer_impl.cc
diff --git a/media/blink/webmediaplayer_impl.cc b/media/blink/webmediaplayer_impl.cc
index bc7a7a4d3b3b8a1d7749fabc72cc69ea86bbb827..f785d3fcf0b42fd5fe9f08ff02411c6ae0021b07 100644
--- a/media/blink/webmediaplayer_impl.cc
+++ b/media/blink/webmediaplayer_impl.cc
@@ -595,6 +595,32 @@ unsigned WebMediaPlayerImpl::videoDecodedByteCount() const {
   return stats.video_bytes_decoded;
 }
 
+size_t WebMediaPlayerImpl::getAudioBufferSize() const {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::AUDIO)) {
+    return demuxer_->GetStream(DemuxerStream::AUDIO)->GetMemoryLimit();
+  }
+  return 0;
+}
+
+void WebMediaPlayerImpl::setAudioBufferSize(size_t size) {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::AUDIO)) {
+    demuxer_->GetStream(DemuxerStream::AUDIO)->SetMemoryLimit(size);
+  }
+}
+
+size_t WebMediaPlayerImpl::getVideoBufferSize() const {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::VIDEO)) {
+    return demuxer_->GetStream(DemuxerStream::VIDEO)->GetMemoryLimit();
+  }
+  return 0;
+}
+
+void WebMediaPlayerImpl::setVideoBufferSize(size_t size) {
+  if (demuxer_ && demuxer_->GetStream(DemuxerStream::VIDEO)) {
+    demuxer_->GetStream(DemuxerStream::VIDEO)->SetMemoryLimit(size);
+  }
+}
+
 bool WebMediaPlayerImpl::copyVideoTextureToPlatformTexture(
     blink::WebGraphicsContext3D* web_graphics_context,
     unsigned int texture,