AppCache: Provide a way to override the default quota for an origin. The inte...

webkit/appcache/appcache_storage_impl.cc

 // Copyright (c) 2010 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 "webkit/appcache/appcache_storage_impl.h"
 
 #include "app/sql/connection.h"
 #include "app/sql/transaction.h"
 #include "base/file_util.h"
 #include "base/logging.h"
@@ skipping 384 matching lines @@
                          AppCache* newest_cache);
 
   virtual void Run();
   virtual void RunCompleted();
   virtual void CancelCompletion();
 
   scoped_refptr<AppCacheGroup> group_;
   scoped_refptr<AppCache> cache_;
   bool success_;
   bool would_exceed_quota_;
+  int64 quota_override_;
   std::vector<int64> newly_deletable_response_ids_;
 };
 
 AppCacheStorageImpl::StoreGroupAndCacheTask::StoreGroupAndCacheTask(
     AppCacheStorageImpl* storage, AppCacheGroup* group, AppCache* newest_cache)
     : StoreOrLoadTask(storage), group_(group), cache_(newest_cache),
-      success_(false), would_exceed_quota_(false) {
+      success_(false), would_exceed_quota_(false),
+      quota_override_(
+          storage->GetOriginQuotaInMemory(group->manifest_url().GetOrigin())) {
   group_record_.group_id = group->group_id();
   group_record_.manifest_url = group->manifest_url();
   group_record_.origin = group_record_.manifest_url.GetOrigin();
   newest_cache->ToDatabaseRecords(
       group,
       &cache_record_, &entry_records_, &fallback_namespace_records_,
       &online_whitelist_records_);
 }
 
 void AppCacheStorageImpl::StoreGroupAndCacheTask::Run() {
@@ skipping 56 matching lines @@
   success_ =
       success_ &&
       database_->InsertCache(&cache_record_) &&
       database_->InsertEntryRecords(entry_records_) &&
       database_->InsertFallbackNameSpaceRecords(fallback_namespace_records_)&&
       database_->InsertOnlineWhiteListRecords(online_whitelist_records_);
 
   if (!success_)
     return;
 
-  if (database_->GetOriginUsage(group_record_.origin) >
-      database_->GetOriginQuota(group_record_.origin)) {
+  int64 quota = (quota_override_ >= 0) ?
+      quota_override_ :
+      database_->GetOriginQuota(group_record_.origin);
+
+  if (database_->GetOriginUsage(group_record_.origin) > quota) {
     would_exceed_quota_ = true;
     success_ = false;
     return;
   }
 
   success_ = transaction.Commit();
 }
 
 void AppCacheStorageImpl::StoreGroupAndCacheTask::RunCompleted() {
   if (success_) {
@@ skipping 813 matching lines @@
     Disable();
     if (!is_incognito_) {
       LOG(INFO) << "Deleting existing appcache data and starting over.";
       AppCacheThread::PostTask(AppCacheThread::db(), FROM_HERE,
           NewRunnableFunction(DeleteDirectory, cache_directory_));
     }
   }
 }
 
 }  // namespace appcache