quota_permission_context_impl.cc 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  1. // Copyright (c) 2012 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 "components/permissions/quota_permission_context_impl.h"
  5. #include <stdint.h>
  6. #include <string>
  7. #include "base/bind.h"
  8. #include "base/compiler_specific.h"
  9. #include "build/build_config.h"
  10. #include "components/permissions/permission_request.h"
  11. #include "components/permissions/permission_request_manager.h"
  12. #include "components/permissions/request_type.h"
  13. #include "components/strings/grit/components_strings.h"
  14. #include "components/url_formatter/elide_url.h"
  15. #include "content/public/browser/browser_task_traits.h"
  16. #include "content/public/browser/browser_thread.h"
  17. #include "content/public/browser/navigation_details.h"
  18. #include "content/public/browser/render_frame_host.h"
  19. #include "content/public/browser/web_contents.h"
  20. #include "third_party/blink/public/mojom/quota/quota_types.mojom.h"
  21. #include "ui/base/l10n/l10n_util.h"
  22. #include "url/gurl.h"
  23. #if BUILDFLAG(IS_ANDROID)
  24. #include "components/resources/android/theme_resources.h"
  25. #else
  26. #include "components/vector_icons/vector_icons.h"
  27. #endif
  28. namespace permissions {
  29. namespace {
  30. // On Android, if the site requested larger quota than this threshold, show a
  31. // different message to the user.
  32. const int64_t kRequestLargeQuotaThreshold = 5 * 1024 * 1024;
  33. // QuotaPermissionRequest ---------------------------------------------
  34. class QuotaPermissionRequest : public PermissionRequest {
  35. public:
  36. QuotaPermissionRequest(
  37. QuotaPermissionContextImpl* context,
  38. const GURL& requesting_origin,
  39. bool is_large_quota_request,
  40. content::QuotaPermissionContext::PermissionCallback callback);
  41. QuotaPermissionRequest(const QuotaPermissionRequest&) = delete;
  42. QuotaPermissionRequest& operator=(const QuotaPermissionRequest&) = delete;
  43. ~QuotaPermissionRequest() override;
  44. // PermissionRequest:
  45. bool IsDuplicateOf(PermissionRequest* other_request) const override;
  46. #if BUILDFLAG(IS_ANDROID)
  47. std::u16string GetDialogMessageText() const override;
  48. #endif
  49. private:
  50. void PermissionDecided(ContentSetting result, bool is_one_time);
  51. void DeleteRequest();
  52. const scoped_refptr<QuotaPermissionContextImpl> context_;
  53. const bool is_large_quota_request_;
  54. content::QuotaPermissionContext::PermissionCallback callback_;
  55. };
  56. QuotaPermissionRequest::QuotaPermissionRequest(
  57. QuotaPermissionContextImpl* context,
  58. const GURL& requesting_origin,
  59. bool is_large_quota_request,
  60. content::QuotaPermissionContext::PermissionCallback callback)
  61. : PermissionRequest(
  62. requesting_origin,
  63. permissions::RequestType::kDiskQuota,
  64. /*has_gesture=*/false,
  65. base::BindOnce(&QuotaPermissionRequest::PermissionDecided,
  66. base::Unretained(this)),
  67. base::BindOnce(&QuotaPermissionRequest::DeleteRequest,
  68. base::Unretained(this))),
  69. context_(context),
  70. is_large_quota_request_(is_large_quota_request),
  71. callback_(std::move(callback)) {}
  72. QuotaPermissionRequest::~QuotaPermissionRequest() {}
  73. bool QuotaPermissionRequest::IsDuplicateOf(
  74. PermissionRequest* other_request) const {
  75. // The downcast here is safe because PermissionRequest::IsDuplicateOf ensures
  76. // that both requests are of type kDiskQuota.
  77. return permissions::PermissionRequest::IsDuplicateOf(other_request) &&
  78. is_large_quota_request_ ==
  79. static_cast<QuotaPermissionRequest*>(other_request)
  80. ->is_large_quota_request_;
  81. }
  82. #if BUILDFLAG(IS_ANDROID)
  83. std::u16string QuotaPermissionRequest::GetDialogMessageText() const {
  84. // If the site requested larger quota than this threshold, show a different
  85. // message to the user.
  86. return l10n_util::GetStringFUTF16(
  87. (is_large_quota_request_ ? IDS_REQUEST_LARGE_QUOTA_INFOBAR_TEXT
  88. : IDS_REQUEST_QUOTA_INFOBAR_TEXT),
  89. url_formatter::FormatUrlForSecurityDisplay(requesting_origin()));
  90. }
  91. #endif // BUILDFLAG(IS_ANDROID)
  92. void QuotaPermissionRequest::PermissionDecided(ContentSetting result,
  93. bool is_one_time) {
  94. DCHECK(!is_one_time);
  95. if (result == CONTENT_SETTING_DEFAULT) {
  96. // Handled by `DeleteRequest`.
  97. return;
  98. }
  99. auto response =
  100. result == ContentSetting::CONTENT_SETTING_ALLOW
  101. ? content::QuotaPermissionContext::QUOTA_PERMISSION_RESPONSE_ALLOW
  102. : content::QuotaPermissionContext::QUOTA_PERMISSION_RESPONSE_DISALLOW;
  103. context_->DispatchCallbackOnIOThread(std::move(callback_), response);
  104. }
  105. void QuotaPermissionRequest::DeleteRequest() {
  106. if (callback_) {
  107. context_->DispatchCallbackOnIOThread(
  108. std::move(callback_),
  109. content::QuotaPermissionContext::QUOTA_PERMISSION_RESPONSE_CANCELLED);
  110. }
  111. delete this;
  112. }
  113. } // namespace
  114. // QuotaPermissionContextImpl -----------------------------------------------
  115. QuotaPermissionContextImpl::QuotaPermissionContextImpl() {}
  116. void QuotaPermissionContextImpl::RequestQuotaPermission(
  117. const content::StorageQuotaParams& params,
  118. int render_process_id,
  119. PermissionCallback callback) {
  120. if (params.storage_type != blink::mojom::StorageType::kPersistent) {
  121. // For now we only support requesting quota with this interface
  122. // for Persistent storage type.
  123. std::move(callback).Run(QUOTA_PERMISSION_RESPONSE_DISALLOW);
  124. return;
  125. }
  126. if (!content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)) {
  127. content::GetUIThreadTaskRunner({})->PostTask(
  128. FROM_HERE,
  129. base::BindOnce(&QuotaPermissionContextImpl::RequestQuotaPermission,
  130. this, params, render_process_id, std::move(callback)));
  131. return;
  132. }
  133. content::RenderFrameHost* render_frame_host =
  134. content::RenderFrameHost::FromID(render_process_id,
  135. params.render_frame_id);
  136. if (!render_frame_host) {
  137. // The tab may have gone away or the request may not be from a tab.
  138. LOG(WARNING) << "Attempt to request quota tabless renderer: "
  139. << render_process_id << "," << params.render_frame_id;
  140. DispatchCallbackOnIOThread(std::move(callback),
  141. QUOTA_PERMISSION_RESPONSE_CANCELLED);
  142. return;
  143. }
  144. PermissionRequestManager* permission_request_manager =
  145. PermissionRequestManager::FromWebContents(
  146. content::WebContents::FromRenderFrameHost(render_frame_host));
  147. if (permission_request_manager) {
  148. bool is_large_quota_request =
  149. params.requested_size > kRequestLargeQuotaThreshold;
  150. permission_request_manager->AddRequest(
  151. render_frame_host, new QuotaPermissionRequest(this, params.origin_url,
  152. is_large_quota_request,
  153. std::move(callback)));
  154. return;
  155. }
  156. // The tab has no UI service for presenting the permissions request.
  157. LOG(WARNING) << "Attempt to request quota from a background page: "
  158. << render_process_id << "," << params.render_frame_id;
  159. DispatchCallbackOnIOThread(std::move(callback),
  160. QUOTA_PERMISSION_RESPONSE_CANCELLED);
  161. }
  162. void QuotaPermissionContextImpl::DispatchCallbackOnIOThread(
  163. PermissionCallback callback,
  164. QuotaPermissionResponse response) {
  165. DCHECK(callback);
  166. if (!content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)) {
  167. content::GetIOThreadTaskRunner({})->PostTask(
  168. FROM_HERE,
  169. base::BindOnce(&QuotaPermissionContextImpl::DispatchCallbackOnIOThread,
  170. this, std::move(callback), response));
  171. return;
  172. }
  173. std::move(callback).Run(response);
  174. }
  175. QuotaPermissionContextImpl::~QuotaPermissionContextImpl() {}
  176. } // namespace permissions