Allow setting memory limits on media::DemuxerStream

content/renderer/media/android/media_source_delegate.cc

Index: content/renderer/media/android/media_source_delegate.cc
diff --git a/content/renderer/media/android/media_source_delegate.cc b/content/renderer/media/android/media_source_delegate.cc
index 5d4aca1baba5b880f93c415fe3e81c09a08ebc32..b3a186e877a5491e235f3d462d8247c4d1e64733 100644
--- a/content/renderer/media/android/media_source_delegate.cc
+++ b/content/renderer/media/android/media_source_delegate.cc
@@ -204,6 +204,40 @@ size_t MediaSourceDelegate::VideoDecodedByteCount() const {
   return statistics_.video_bytes_decoded;
 }
 
+size_t MediaSourceDelegate::getAudioBufferSize() const {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::AUDIO);
+  if (stream)
+    return stream->GetMemoryLimit();
+  return 0;
+}
+
+void MediaSourceDelegate::setAudioBufferSize(size_t size) {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::AUDIO);
+  if (stream)
+    return stream->SetMemoryLimit(size);
+}
+
+size_t MediaSourceDelegate::getVideoBufferSize() const {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::VIDEO);
+  if (stream)
+    return stream->GetMemoryLimit();
+  return 0;
+}
+
+void MediaSourceDelegate::setVideoBufferSize(size_t size) {
+  media::DemuxerStream* stream = nullptr;
+  if (chunk_demuxer_)
+    stream = chunk_demuxer_->GetStream(DemuxerStream::VIDEO);
+  if (stream)
+    return stream->SetMemoryLimit(size);
+}
+
 void MediaSourceDelegate::CancelPendingSeek(const base::TimeDelta& seek_time) {
   DCHECK(main_task_runner_->BelongsToCurrentThread());
   DVLOG(1) << __FUNCTION__ << "(" << seek_time.InSecondsF() << ") : "