| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "webkit/browser/fileapi/quota/quota_backend_impl.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "base/bind.h" | |
| 10 #include "base/callback.h" | |
| 11 #include "base/logging.h" | |
| 12 #include "base/numerics/safe_conversions.h" | |
| 13 #include "base/sequenced_task_runner.h" | |
| 14 #include "webkit/browser/fileapi/file_system_usage_cache.h" | |
| 15 #include "webkit/browser/quota/quota_client.h" | |
| 16 #include "webkit/browser/quota/quota_manager_proxy.h" | |
| 17 #include "webkit/common/fileapi/file_system_util.h" | |
| 18 | |
| 19 namespace storage { | |
| 20 | |
| 21 QuotaBackendImpl::QuotaBackendImpl( | |
| 22 base::SequencedTaskRunner* file_task_runner, | |
| 23 ObfuscatedFileUtil* obfuscated_file_util, | |
| 24 FileSystemUsageCache* file_system_usage_cache, | |
| 25 storage::QuotaManagerProxy* quota_manager_proxy) | |
| 26 : file_task_runner_(file_task_runner), | |
| 27 obfuscated_file_util_(obfuscated_file_util), | |
| 28 file_system_usage_cache_(file_system_usage_cache), | |
| 29 quota_manager_proxy_(quota_manager_proxy), | |
| 30 weak_ptr_factory_(this) { | |
| 31 } | |
| 32 | |
| 33 QuotaBackendImpl::~QuotaBackendImpl() { | |
| 34 } | |
| 35 | |
| 36 void QuotaBackendImpl::ReserveQuota(const GURL& origin, | |
| 37 FileSystemType type, | |
| 38 int64 delta, | |
| 39 const ReserveQuotaCallback& callback) { | |
| 40 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 41 DCHECK(origin.is_valid()); | |
| 42 if (!delta) { | |
| 43 callback.Run(base::File::FILE_OK, 0); | |
| 44 return; | |
| 45 } | |
| 46 DCHECK(quota_manager_proxy_.get()); | |
| 47 quota_manager_proxy_->GetUsageAndQuota( | |
| 48 file_task_runner_.get(), | |
| 49 origin, | |
| 50 FileSystemTypeToQuotaStorageType(type), | |
| 51 base::Bind(&QuotaBackendImpl::DidGetUsageAndQuotaForReserveQuota, | |
| 52 weak_ptr_factory_.GetWeakPtr(), | |
| 53 QuotaReservationInfo(origin, type, delta), | |
| 54 callback)); | |
| 55 } | |
| 56 | |
| 57 void QuotaBackendImpl::ReleaseReservedQuota(const GURL& origin, | |
| 58 FileSystemType type, | |
| 59 int64 size) { | |
| 60 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 61 DCHECK(origin.is_valid()); | |
| 62 DCHECK_LE(0, size); | |
| 63 if (!size) | |
| 64 return; | |
| 65 ReserveQuotaInternal(QuotaReservationInfo(origin, type, -size)); | |
| 66 } | |
| 67 | |
| 68 void QuotaBackendImpl::CommitQuotaUsage(const GURL& origin, | |
| 69 FileSystemType type, | |
| 70 int64 delta) { | |
| 71 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 72 DCHECK(origin.is_valid()); | |
| 73 if (!delta) | |
| 74 return; | |
| 75 ReserveQuotaInternal(QuotaReservationInfo(origin, type, delta)); | |
| 76 base::FilePath path; | |
| 77 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
| 78 return; | |
| 79 bool result = file_system_usage_cache_->AtomicUpdateUsageByDelta(path, delta); | |
| 80 DCHECK(result); | |
| 81 } | |
| 82 | |
| 83 void QuotaBackendImpl::IncrementDirtyCount(const GURL& origin, | |
| 84 FileSystemType type) { | |
| 85 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 86 DCHECK(origin.is_valid()); | |
| 87 base::FilePath path; | |
| 88 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
| 89 return; | |
| 90 DCHECK(file_system_usage_cache_); | |
| 91 file_system_usage_cache_->IncrementDirty(path); | |
| 92 } | |
| 93 | |
| 94 void QuotaBackendImpl::DecrementDirtyCount(const GURL& origin, | |
| 95 FileSystemType type) { | |
| 96 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 97 DCHECK(origin.is_valid()); | |
| 98 base::FilePath path; | |
| 99 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
| 100 return; | |
| 101 DCHECK(file_system_usage_cache_); | |
| 102 file_system_usage_cache_->DecrementDirty(path); | |
| 103 } | |
| 104 | |
| 105 void QuotaBackendImpl::DidGetUsageAndQuotaForReserveQuota( | |
| 106 const QuotaReservationInfo& info, | |
| 107 const ReserveQuotaCallback& callback, | |
| 108 storage::QuotaStatusCode status, | |
| 109 int64 usage, | |
| 110 int64 quota) { | |
| 111 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 112 DCHECK(info.origin.is_valid()); | |
| 113 DCHECK_LE(0, usage); | |
| 114 DCHECK_LE(0, quota); | |
| 115 if (status != storage::kQuotaStatusOk) { | |
| 116 callback.Run(base::File::FILE_ERROR_FAILED, 0); | |
| 117 return; | |
| 118 } | |
| 119 | |
| 120 QuotaReservationInfo normalized_info = info; | |
| 121 if (info.delta > 0) { | |
| 122 int64 new_usage = | |
| 123 base::saturated_cast<int64>(usage + static_cast<uint64>(info.delta)); | |
| 124 if (quota < new_usage) | |
| 125 new_usage = quota; | |
| 126 normalized_info.delta = std::max(static_cast<int64>(0), new_usage - usage); | |
| 127 } | |
| 128 | |
| 129 ReserveQuotaInternal(normalized_info); | |
| 130 if (callback.Run(base::File::FILE_OK, normalized_info.delta)) | |
| 131 return; | |
| 132 // The requester could not accept the reserved quota. Revert it. | |
| 133 ReserveQuotaInternal( | |
| 134 QuotaReservationInfo(normalized_info.origin, | |
| 135 normalized_info.type, | |
| 136 -normalized_info.delta)); | |
| 137 } | |
| 138 | |
| 139 void QuotaBackendImpl::ReserveQuotaInternal(const QuotaReservationInfo& info) { | |
| 140 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 141 DCHECK(info.origin.is_valid()); | |
| 142 DCHECK(quota_manager_proxy_.get()); | |
| 143 quota_manager_proxy_->NotifyStorageModified( | |
| 144 storage::QuotaClient::kFileSystem, | |
| 145 info.origin, | |
| 146 FileSystemTypeToQuotaStorageType(info.type), | |
| 147 info.delta); | |
| 148 } | |
| 149 | |
| 150 base::File::Error QuotaBackendImpl::GetUsageCachePath( | |
| 151 const GURL& origin, | |
| 152 FileSystemType type, | |
| 153 base::FilePath* usage_file_path) { | |
| 154 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
| 155 DCHECK(origin.is_valid()); | |
| 156 DCHECK(usage_file_path); | |
| 157 base::File::Error error = base::File::FILE_OK; | |
| 158 *usage_file_path = | |
| 159 SandboxFileSystemBackendDelegate::GetUsageCachePathForOriginAndType( | |
| 160 obfuscated_file_util_, origin, type, &error); | |
| 161 return error; | |
| 162 } | |
| 163 | |
| 164 QuotaBackendImpl::QuotaReservationInfo::QuotaReservationInfo( | |
| 165 const GURL& origin, FileSystemType type, int64 delta) | |
| 166 : origin(origin), type(type), delta(delta) { | |
| 167 } | |
| 168 | |
| 169 QuotaBackendImpl::QuotaReservationInfo::~QuotaReservationInfo() { | |
| 170 } | |
| 171 | |
| 172 } // namespace storage | |
| OLD | NEW |