Add a command line switch to allow setting the max size...

chrome/browser/net/chrome_url_request_context.cc

 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/net/chrome_url_request_context.h"
 
 #include "base/command_line.h"
 #include "base/string_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/privacy_blacklist/blacklist.h"
@@ skipping 90 matching lines @@
   return net::ProxyService::Create(
       proxy_config.get(),
       use_v8,
       context,
       g_browser_process->io_thread()->message_loop());
 }
 
 // static
 ChromeURLRequestContext* ChromeURLRequestContext::CreateOriginal(
     Profile* profile, const FilePath& cookie_store_path,
-    const FilePath& disk_cache_path) {
+    const FilePath& disk_cache_path, int cache_size) {
   DCHECK(!profile->IsOffTheRecord());
   ChromeURLRequestContext* context = new ChromeURLRequestContext(profile);
 
   // Global host resolver for the context.
   context->host_resolver_ = chrome_browser_net::GetGlobalHostResolver();
 
   context->proxy_service_ = CreateProxyService(
       context, *CommandLine::ForCurrentProcess());
 
   net::HttpCache* cache =
       new net::HttpCache(context->host_resolver_,
                          context->proxy_service_,
-                         disk_cache_path.ToWStringHack(), 0);
+                         disk_cache_path.ToWStringHack(), cache_size);
 
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
   bool record_mode = chrome::kRecordModeEnabled &&
                      command_line.HasSwitch(switches::kRecordMode);
   bool playback_mode = command_line.HasSwitch(switches::kPlaybackMode);
 
   if (record_mode || playback_mode) {
     // Don't use existing cookies and use an in-memory store.
     context->cookie_store_ = new net::CookieMonster();
     cache->set_mode(
@@ skipping 19 matching lines @@
         cookie_store_path.ToWStringHack(),
         g_browser_process->db_thread()->message_loop()));
     context->cookie_store_ = new net::CookieMonster(context->cookie_db_.get());
   }
 
   return context;
 }
 
 // static
 ChromeURLRequestContext* ChromeURLRequestContext::CreateOriginalForMedia(
-    Profile* profile, const FilePath& disk_cache_path) {
+    Profile* profile, const FilePath& disk_cache_path, int cache_size) {
   DCHECK(!profile->IsOffTheRecord());
-  return CreateRequestContextForMedia(profile, disk_cache_path, false);
+  return CreateRequestContextForMedia(profile, disk_cache_path, cache_size,
+                                      false);
 }
 
 // static
 ChromeURLRequestContext* ChromeURLRequestContext::CreateOriginalForExtensions(
     Profile* profile, const FilePath& cookie_store_path) {
   DCHECK(!profile->IsOffTheRecord());
   ChromeURLRequestContext* context = new ChromeURLRequestContext(profile);
 
   // All we care about for extensions is the cookie store.
   DCHECK(!cookie_store_path.empty());
@@ skipping 39 matching lines @@
 
   // Enable cookies for extension URLs only.
   const char* schemes[] = {chrome::kExtensionScheme};
   context->cookie_store_->SetCookieableSchemes(schemes, 1);
 
   return context;
 }
 
 // static
 ChromeURLRequestContext* ChromeURLRequestContext::CreateRequestContextForMedia(
-    Profile* profile, const FilePath& disk_cache_path, bool off_the_record) {
+    Profile* profile, const FilePath& disk_cache_path, int cache_size,
+    bool off_the_record) {
   URLRequestContext* original_context =
       profile->GetOriginalProfile()->GetRequestContext();
   ChromeURLRequestContext* context = new ChromeURLRequestContext(profile);
   context->is_media_ = true;
 
   // Share the same proxy service of the common profile.
   context->proxy_service_ = original_context->proxy_service();
   // Also share the cookie store of the common profile.
   context->cookie_store_ = original_context->cookie_store();
 
   // Create a media cache with default size.
   // TODO(hclam): make the maximum size of media cache configurable.
   net::HttpCache* original_cache =
       original_context->http_transaction_factory()->GetCache();
   net::HttpCache* cache;
   if (original_cache) {
     // Try to reuse HttpNetworkSession in the original context, assuming that
     // HttpTransactionFactory (network_layer()) of HttpCache is implemented
     // by HttpNetworkLayer so we can reuse HttpNetworkSession within it. This
     // assumption will be invalid if the original HttpCache is constructed with
     // HttpCache(HttpTransactionFactory*, disk_cache::Backend*) constructor.
     net::HttpNetworkLayer* original_network_layer =
         static_cast<net::HttpNetworkLayer*>(original_cache->network_layer());
     cache = new net::HttpCache(original_network_layer->GetSession(),
-                               disk_cache_path.ToWStringHack(), 0);
+                               disk_cache_path.ToWStringHack(), cache_size);
   } else {
     // If original HttpCache doesn't exist, simply construct one with a whole
     // new set of network stack.
     cache = new net::HttpCache(original_context->host_resolver(),
                                original_context->proxy_service(),
-                               disk_cache_path.ToWStringHack(), 0);
+                               disk_cache_path.ToWStringHack(), cache_size);
   }
 
   cache->set_type(net::MEDIA_CACHE);
   context->http_transaction_factory_ = cache;
   return context;
 }
 
 ChromeURLRequestContext::ChromeURLRequestContext(Profile* profile)
     : prefs_(profile->GetPrefs()),
       is_media_(false),
@@ skipping 191 matching lines @@
     delete cookie_store_;
 
   // Do not delete the proxy service in the case of OTR or media contexts, as
   // it is owned by the original URLRequestContext.
   if (!is_off_the_record_ && !is_media_)
     delete proxy_service_;
 
   // Do not delete host_resolver_; it will be freed by FreeGlobalHostResolver()
   // during the teardown of DNS prefetching.
 }