Limit the total memory usage for Stream instances

content/browser/streams/stream_registry.h

Index: content/browser/streams/stream_registry.h
diff --git a/content/browser/streams/stream_registry.h b/content/browser/streams/stream_registry.h
index e75c97c1699ae1f625180b929bdc779c2696a619..7bdd47935940cc1ca18fdf951ea78ea6e4529045 100644
--- a/content/browser/streams/stream_registry.h
+++ b/content/browser/streams/stream_registry.h
@@ -32,20 +32,34 @@ class CONTENT_EXPORT StreamRegistry : public base::NonThreadSafe {
 
   void UnregisterStream(const GURL& url);
 
+  // Called by Stream instances to request increase of memory usage. If the
+  // total memory usage for this registry is going to exceed the limit,
+  // returns false. Otherwise, updates |total_memory_usage_| and returns true.
+  bool UpdateMemoryUsage(const GURL& url, size_t new_size);
+
   // Gets the stream associated with |url|.  Returns NULL if there is no such
   // stream.
   scoped_refptr<Stream> GetStream(const GURL& url);
 
+  // Only for testing.
+  void set_max_memory_usage(size_t size) {

[Charlie Reis, 2013/08/20 22:24:52]

We usually end such method names with _for_testing() to help prevent others from
calling them in normal code.

[tyoshino (SeeGerritForStatus), 2013/08/21 05:37:28]

Done.

+    max_memory_usage_ = size;
+  }
+
  private:
   typedef std::map<GURL, scoped_refptr<Stream> > StreamMap;
 
   StreamMap streams_;
 
+  size_t total_memory_usage_;
+
+  // Maximum amount of memory allowed to use for Stream instances registered
+  // with this 5Aregistry.

[Charlie Reis, 2013/08/20 22:24:52]

Is 5Aregistry a typo?

[tyoshino (SeeGerritForStatus), 2013/08/21 05:37:28]

Yes.. Fixed.

+  size_t max_memory_usage_;
+
   DISALLOW_COPY_AND_ASSIGN(StreamRegistry);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_STREAMS_STREAM_REGISTRY_H_
-
-