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 fileapi { | |
20 | |
21 QuotaBackendImpl::QuotaBackendImpl( | |
22 base::SequencedTaskRunner* file_task_runner, | |
23 ObfuscatedFileUtil* obfuscated_file_util, | |
24 FileSystemUsageCache* file_system_usage_cache, | |
25 quota::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_); | |
47 quota_manager_proxy_->GetUsageAndQuota( | |
48 file_task_runner_, origin, FileSystemTypeToQuotaStorageType(type), | |
49 base::Bind(&QuotaBackendImpl::DidGetUsageAndQuotaForReserveQuota, | |
50 weak_ptr_factory_.GetWeakPtr(), | |
51 QuotaReservationInfo(origin, type, delta), callback)); | |
52 } | |
53 | |
54 void QuotaBackendImpl::ReleaseReservedQuota(const GURL& origin, | |
55 FileSystemType type, | |
56 int64 size) { | |
57 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
58 DCHECK(origin.is_valid()); | |
59 DCHECK_LE(0, size); | |
60 if (!size) | |
61 return; | |
62 ReserveQuotaInternal(QuotaReservationInfo(origin, type, -size)); | |
63 } | |
64 | |
65 void QuotaBackendImpl::CommitQuotaUsage(const GURL& origin, | |
66 FileSystemType type, | |
67 int64 delta) { | |
68 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
69 DCHECK(origin.is_valid()); | |
70 if (!delta) | |
71 return; | |
72 ReserveQuotaInternal(QuotaReservationInfo(origin, type, delta)); | |
73 base::FilePath path; | |
74 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
75 return; | |
76 bool result = file_system_usage_cache_->AtomicUpdateUsageByDelta(path, delta); | |
77 DCHECK(result); | |
78 } | |
79 | |
80 void QuotaBackendImpl::IncrementDirtyCount(const GURL& origin, | |
81 FileSystemType type) { | |
82 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
83 DCHECK(origin.is_valid()); | |
84 base::FilePath path; | |
85 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
86 return; | |
87 DCHECK(file_system_usage_cache_); | |
88 file_system_usage_cache_->IncrementDirty(path); | |
89 } | |
90 | |
91 void QuotaBackendImpl::DecrementDirtyCount(const GURL& origin, | |
92 FileSystemType type) { | |
93 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
94 DCHECK(origin.is_valid()); | |
95 base::FilePath path; | |
96 if (GetUsageCachePath(origin, type, &path) != base::File::FILE_OK) | |
97 return; | |
98 DCHECK(file_system_usage_cache_); | |
99 file_system_usage_cache_->DecrementDirty(path); | |
100 } | |
101 | |
102 void QuotaBackendImpl::DidGetUsageAndQuotaForReserveQuota( | |
103 const QuotaReservationInfo& info, | |
104 const ReserveQuotaCallback& callback, | |
105 quota::QuotaStatusCode status, int64 usage, int64 quota) { | |
106 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
107 DCHECK(info.origin.is_valid()); | |
108 DCHECK_LE(0, usage); | |
109 DCHECK_LE(0, quota); | |
110 if (status != quota::kQuotaStatusOk) { | |
111 callback.Run(base::File::FILE_ERROR_FAILED, 0); | |
112 return; | |
113 } | |
114 | |
115 QuotaReservationInfo normalized_info = info; | |
116 if (info.delta > 0) { | |
117 int64 new_usage = | |
118 base::saturated_cast<int64>(usage + static_cast<uint64>(info.delta)); | |
119 if (quota < new_usage) | |
120 new_usage = quota; | |
121 normalized_info.delta = std::max(static_cast<int64>(0), new_usage - usage); | |
122 } | |
123 | |
124 ReserveQuotaInternal(normalized_info); | |
125 if (callback.Run(base::File::FILE_OK, normalized_info.delta)) | |
126 return; | |
127 // The requester could not accept the reserved quota. Revert it. | |
128 ReserveQuotaInternal( | |
129 QuotaReservationInfo(normalized_info.origin, | |
130 normalized_info.type, | |
131 -normalized_info.delta)); | |
132 } | |
133 | |
134 void QuotaBackendImpl::ReserveQuotaInternal(const QuotaReservationInfo& info) { | |
135 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
136 DCHECK(info.origin.is_valid()); | |
137 DCHECK(quota_manager_proxy_); | |
138 quota_manager_proxy_->NotifyStorageModified( | |
139 quota::QuotaClient::kFileSystem, info.origin, | |
140 FileSystemTypeToQuotaStorageType(info.type), info.delta); | |
141 } | |
142 | |
143 base::File::Error QuotaBackendImpl::GetUsageCachePath( | |
144 const GURL& origin, | |
145 FileSystemType type, | |
146 base::FilePath* usage_file_path) { | |
147 DCHECK(file_task_runner_->RunsTasksOnCurrentThread()); | |
148 DCHECK(origin.is_valid()); | |
149 DCHECK(usage_file_path); | |
150 base::File::Error error = base::File::FILE_OK; | |
151 *usage_file_path = | |
152 SandboxFileSystemBackendDelegate::GetUsageCachePathForOriginAndType( | |
153 obfuscated_file_util_, origin, type, &error); | |
154 return error; | |
155 } | |
156 | |
157 QuotaBackendImpl::QuotaReservationInfo::QuotaReservationInfo( | |
158 const GURL& origin, FileSystemType type, int64 delta) | |
159 : origin(origin), type(type), delta(delta) { | |
160 } | |
161 | |
162 QuotaBackendImpl::QuotaReservationInfo::~QuotaReservationInfo() { | |
163 } | |
164 | |
165 } // namespace fileapi | |
OLD | NEW |