aw_quota_manager_bridge.cc 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284
  1. // Copyright (c) 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. #include "android_webview/browser/aw_quota_manager_bridge.h"
  5. #include <set>
  6. #include "android_webview/browser/aw_browser_context.h"
  7. #include "android_webview/browser/aw_content_browser_client.h"
  8. #include "android_webview/browser_jni_headers/AwQuotaManagerBridge_jni.h"
  9. #include "base/android/callback_android.h"
  10. #include "base/android/jni_array.h"
  11. #include "base/android/jni_string.h"
  12. #include "base/bind.h"
  13. #include "base/callback_helpers.h"
  14. #include "base/memory/scoped_refptr.h"
  15. #include "base/synchronization/waitable_event.h"
  16. #include "content/public/browser/browser_task_traits.h"
  17. #include "content/public/browser/browser_thread.h"
  18. #include "content/public/browser/storage_partition.h"
  19. #include "content/public/common/content_client.h"
  20. #include "storage/browser/quota/quota_manager.h"
  21. #include "third_party/blink/public/common/storage_key/storage_key.h"
  22. #include "third_party/blink/public/mojom/quota/quota_types.mojom.h"
  23. #include "url/gurl.h"
  24. #include "url/origin.h"
  25. using base::android::AttachCurrentThread;
  26. using base::android::JavaParamRef;
  27. using base::android::JavaRef;
  28. using base::android::ScopedJavaGlobalRef;
  29. using content::BrowserThread;
  30. using content::StoragePartition;
  31. using storage::QuotaManager;
  32. namespace android_webview {
  33. namespace {
  34. // This object lives on UI and IO threads. Care need to be taken to make sure
  35. // there are no concurrent accesses to instance variables. Also this object
  36. // is refcounted in the various callbacks, and is destroyed when all callbacks
  37. // are destroyed at the end of DoneOnUIThread.
  38. class GetStorageKeysTask
  39. : public base::RefCountedThreadSafe<GetStorageKeysTask> {
  40. public:
  41. GetStorageKeysTask(AwQuotaManagerBridge::GetOriginsCallback callback,
  42. QuotaManager* quota_manager);
  43. GetStorageKeysTask(const GetStorageKeysTask&) = delete;
  44. GetStorageKeysTask& operator=(const GetStorageKeysTask&) = delete;
  45. void Run();
  46. private:
  47. friend class base::RefCountedThreadSafe<GetStorageKeysTask>;
  48. ~GetStorageKeysTask();
  49. void OnStorageKeysObtained(blink::mojom::StorageType type,
  50. const std::set<blink::StorageKey>& storage_keys);
  51. void OnUsageAndQuotaObtained(const blink::StorageKey& storage_key,
  52. blink::mojom::QuotaStatusCode status_code,
  53. int64_t usage,
  54. int64_t quota);
  55. void CheckDone();
  56. void DoneOnUIThread();
  57. AwQuotaManagerBridge::GetOriginsCallback ui_callback_;
  58. scoped_refptr<QuotaManager> quota_manager_;
  59. std::vector<std::string> origin_;
  60. std::vector<int64_t> usage_;
  61. std::vector<int64_t> quota_;
  62. size_t num_callbacks_to_wait_;
  63. size_t num_callbacks_received_;
  64. };
  65. GetStorageKeysTask::GetStorageKeysTask(
  66. AwQuotaManagerBridge::GetOriginsCallback callback,
  67. QuotaManager* quota_manager)
  68. : ui_callback_(std::move(callback)), quota_manager_(quota_manager) {
  69. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  70. }
  71. GetStorageKeysTask::~GetStorageKeysTask() {}
  72. void GetStorageKeysTask::Run() {
  73. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  74. content::GetIOThreadTaskRunner({})->PostTask(
  75. FROM_HERE,
  76. base::BindOnce(
  77. &QuotaManager::GetStorageKeysForType, quota_manager_,
  78. blink::mojom::StorageType::kTemporary,
  79. base::BindOnce(&GetStorageKeysTask::OnStorageKeysObtained, this,
  80. blink::mojom::StorageType::kTemporary)));
  81. }
  82. void GetStorageKeysTask::OnStorageKeysObtained(
  83. blink::mojom::StorageType type,
  84. const std::set<blink::StorageKey>& storage_keys) {
  85. DCHECK_CURRENTLY_ON(BrowserThread::IO);
  86. num_callbacks_to_wait_ = storage_keys.size();
  87. num_callbacks_received_ = 0u;
  88. for (const blink::StorageKey& storage_key : storage_keys) {
  89. quota_manager_->GetUsageAndQuota(
  90. storage_key, type,
  91. base::BindOnce(&GetStorageKeysTask::OnUsageAndQuotaObtained, this,
  92. storage_key));
  93. }
  94. CheckDone();
  95. }
  96. void GetStorageKeysTask::OnUsageAndQuotaObtained(
  97. const blink::StorageKey& storage_key,
  98. blink::mojom::QuotaStatusCode status_code,
  99. int64_t usage,
  100. int64_t quota) {
  101. DCHECK_CURRENTLY_ON(BrowserThread::IO);
  102. if (status_code == blink::mojom::QuotaStatusCode::kOk) {
  103. origin_.push_back(storage_key.origin().GetURL().spec());
  104. usage_.push_back(usage);
  105. quota_.push_back(quota);
  106. }
  107. ++num_callbacks_received_;
  108. CheckDone();
  109. }
  110. void GetStorageKeysTask::CheckDone() {
  111. DCHECK_CURRENTLY_ON(BrowserThread::IO);
  112. if (num_callbacks_received_ == num_callbacks_to_wait_) {
  113. content::GetUIThreadTaskRunner({})->PostTask(
  114. FROM_HERE, base::BindOnce(&GetStorageKeysTask::DoneOnUIThread, this));
  115. } else if (num_callbacks_received_ > num_callbacks_to_wait_) {
  116. NOTREACHED();
  117. }
  118. }
  119. // This method is to avoid copying the 3 vector arguments into a bound callback.
  120. void GetStorageKeysTask::DoneOnUIThread() {
  121. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  122. std::move(ui_callback_).Run(origin_, usage_, quota_);
  123. }
  124. } // namespace
  125. // static
  126. scoped_refptr<AwQuotaManagerBridge> AwQuotaManagerBridge::Create(
  127. AwBrowserContext* browser_context) {
  128. return new AwQuotaManagerBridge(browser_context);
  129. }
  130. AwQuotaManagerBridge::AwQuotaManagerBridge(AwBrowserContext* browser_context)
  131. : browser_context_(browser_context) {}
  132. AwQuotaManagerBridge::~AwQuotaManagerBridge() {}
  133. void AwQuotaManagerBridge::Init(JNIEnv* env,
  134. const JavaParamRef<jobject>& object) {
  135. java_ref_ = JavaObjectWeakGlobalRef(env, object);
  136. }
  137. StoragePartition* AwQuotaManagerBridge::GetStoragePartition() const {
  138. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  139. // AndroidWebview does not use per-site storage partitions.
  140. StoragePartition* storage_partition =
  141. browser_context_->GetDefaultStoragePartition();
  142. DCHECK(storage_partition);
  143. return storage_partition;
  144. }
  145. QuotaManager* AwQuotaManagerBridge::GetQuotaManager() const {
  146. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  147. QuotaManager* quota_manager = GetStoragePartition()->GetQuotaManager();
  148. DCHECK(quota_manager);
  149. return quota_manager;
  150. }
  151. void AwQuotaManagerBridge::DeleteAllData(JNIEnv* env,
  152. const JavaParamRef<jobject>& object) {
  153. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  154. GetStoragePartition()->ClearData(
  155. // Clear all web storage data except cookies.
  156. StoragePartition::REMOVE_DATA_MASK_FILE_SYSTEMS |
  157. StoragePartition::REMOVE_DATA_MASK_INDEXEDDB |
  158. StoragePartition::REMOVE_DATA_MASK_LOCAL_STORAGE |
  159. StoragePartition::REMOVE_DATA_MASK_WEBSQL,
  160. StoragePartition::QUOTA_MANAGED_STORAGE_MASK_TEMPORARY,
  161. blink::StorageKey(), base::Time(), base::Time::Max(), base::DoNothing());
  162. }
  163. void AwQuotaManagerBridge::DeleteOrigin(JNIEnv* env,
  164. const JavaParamRef<jobject>& object,
  165. const JavaParamRef<jstring>& origin) {
  166. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  167. std::u16string origin_string(
  168. base::android::ConvertJavaStringToUTF16(env, origin));
  169. StoragePartition* storage_partition = GetStoragePartition();
  170. storage_partition->ClearDataForOrigin(
  171. // All (temporary) QuotaClient types.
  172. StoragePartition::REMOVE_DATA_MASK_FILE_SYSTEMS |
  173. StoragePartition::REMOVE_DATA_MASK_INDEXEDDB |
  174. StoragePartition::REMOVE_DATA_MASK_WEBSQL,
  175. StoragePartition::QUOTA_MANAGED_STORAGE_MASK_TEMPORARY,
  176. GURL(origin_string), base::DoNothing());
  177. }
  178. void AwQuotaManagerBridge::GetOrigins(JNIEnv* env,
  179. const JavaParamRef<jobject>& object,
  180. const JavaParamRef<jobject>& callback) {
  181. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  182. GetOriginsCallback ui_callback = base::BindOnce(
  183. [](const JavaRef<jobject>& obj, const JavaRef<jobject>& callback,
  184. const std::vector<std::string>& origin,
  185. const std::vector<int64_t>& usage, const std::vector<int64_t>& quota) {
  186. JNIEnv* env = AttachCurrentThread();
  187. Java_AwQuotaManagerBridge_onGetOriginsCallback(
  188. env, obj, callback,
  189. base::android::ToJavaArrayOfStrings(env, origin),
  190. base::android::ToJavaLongArray(env, usage),
  191. base::android::ToJavaLongArray(env, quota));
  192. },
  193. ScopedJavaGlobalRef<jobject>(env, object),
  194. ScopedJavaGlobalRef<jobject>(env, callback));
  195. base::MakeRefCounted<GetStorageKeysTask>(std::move(ui_callback),
  196. GetQuotaManager())
  197. ->Run();
  198. }
  199. namespace {
  200. void OnUsageAndQuotaObtained(
  201. AwQuotaManagerBridge::QuotaUsageCallback ui_callback,
  202. blink::mojom::QuotaStatusCode status_code,
  203. int64_t usage,
  204. int64_t quota) {
  205. DCHECK_CURRENTLY_ON(BrowserThread::IO);
  206. if (status_code != blink::mojom::QuotaStatusCode::kOk) {
  207. usage = 0;
  208. quota = 0;
  209. }
  210. content::GetUIThreadTaskRunner({})->PostTask(
  211. FROM_HERE, base::BindOnce(std::move(ui_callback), usage, quota));
  212. }
  213. } // namespace
  214. void AwQuotaManagerBridge::GetUsageAndQuotaForOrigin(
  215. JNIEnv* env,
  216. const JavaParamRef<jobject>& object,
  217. const JavaParamRef<jstring>& origin,
  218. const JavaParamRef<jobject>& callback,
  219. bool is_quota) {
  220. DCHECK_CURRENTLY_ON(BrowserThread::UI);
  221. std::u16string origin_string(
  222. base::android::ConvertJavaStringToUTF16(env, origin));
  223. QuotaUsageCallback ui_callback = base::BindOnce(
  224. [](const JavaRef<jobject>& callback, bool is_quota, int64_t usage,
  225. int64_t quota) {
  226. base::android::RunLongCallbackAndroid(callback,
  227. (is_quota ? quota : usage));
  228. },
  229. ScopedJavaGlobalRef<jobject>(env, callback), is_quota);
  230. // TODO(crbug.com/889590): Use helper for url::Origin creation from string.
  231. content::GetIOThreadTaskRunner({})->PostTask(
  232. FROM_HERE,
  233. base::BindOnce(
  234. &QuotaManager::GetUsageAndQuota, GetQuotaManager(),
  235. blink::StorageKey(url::Origin::Create(GURL(origin_string))),
  236. blink::mojom::StorageType::kTemporary,
  237. base::BindOnce(&OnUsageAndQuotaObtained, std::move(ui_callback))));
  238. }
  239. } // namespace android_webview