aw_proxying_restricted_cookie_manager.cc 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223
  1. // Copyright 2019 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/network_service/aw_proxying_restricted_cookie_manager.h"
  5. #include <memory>
  6. #include <string>
  7. #include <utility>
  8. #include <vector>
  9. #include "android_webview/browser/aw_cookie_access_policy.h"
  10. #include "base/memory/ptr_util.h"
  11. #include "content/public/browser/browser_task_traits.h"
  12. #include "content/public/browser/browser_thread.h"
  13. #include "mojo/public/cpp/bindings/pending_remote.h"
  14. #include "mojo/public/cpp/bindings/remote.h"
  15. #include "mojo/public/cpp/bindings/self_owned_receiver.h"
  16. #include "url/gurl.h"
  17. namespace android_webview {
  18. class AwProxyingRestrictedCookieManagerListener
  19. : public network::mojom::CookieChangeListener {
  20. public:
  21. AwProxyingRestrictedCookieManagerListener(
  22. const GURL& url,
  23. const net::SiteForCookies& site_for_cookies,
  24. base::WeakPtr<AwProxyingRestrictedCookieManager>
  25. aw_restricted_cookie_manager,
  26. mojo::PendingRemote<network::mojom::CookieChangeListener> client_listener)
  27. : url_(url),
  28. site_for_cookies_(site_for_cookies),
  29. aw_restricted_cookie_manager_(aw_restricted_cookie_manager),
  30. client_listener_(std::move(client_listener)) {}
  31. void OnCookieChange(const net::CookieChangeInfo& change) override {
  32. if (aw_restricted_cookie_manager_ &&
  33. aw_restricted_cookie_manager_->AllowCookies(url_, site_for_cookies_))
  34. client_listener_->OnCookieChange(change);
  35. }
  36. private:
  37. const GURL url_;
  38. const net::SiteForCookies site_for_cookies_;
  39. base::WeakPtr<AwProxyingRestrictedCookieManager>
  40. aw_restricted_cookie_manager_;
  41. mojo::Remote<network::mojom::CookieChangeListener> client_listener_;
  42. };
  43. // static
  44. void AwProxyingRestrictedCookieManager::CreateAndBind(
  45. mojo::PendingRemote<network::mojom::RestrictedCookieManager> underlying_rcm,
  46. bool is_service_worker,
  47. int process_id,
  48. int frame_id,
  49. mojo::PendingReceiver<network::mojom::RestrictedCookieManager> receiver) {
  50. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  51. content::GetIOThreadTaskRunner({})->PostTask(
  52. FROM_HERE,
  53. base::BindOnce(
  54. &AwProxyingRestrictedCookieManager::CreateAndBindOnIoThread,
  55. std::move(underlying_rcm), is_service_worker, process_id, frame_id,
  56. std::move(receiver)));
  57. }
  58. AwProxyingRestrictedCookieManager::~AwProxyingRestrictedCookieManager() {
  59. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  60. }
  61. void AwProxyingRestrictedCookieManager::GetAllForUrl(
  62. const GURL& url,
  63. const net::SiteForCookies& site_for_cookies,
  64. const url::Origin& top_frame_origin,
  65. network::mojom::CookieManagerGetOptionsPtr options,
  66. bool partitioned_cookies_runtime_feature_enabled,
  67. GetAllForUrlCallback callback) {
  68. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  69. if (AllowCookies(url, site_for_cookies)) {
  70. underlying_restricted_cookie_manager_->GetAllForUrl(
  71. url, site_for_cookies, top_frame_origin, std::move(options),
  72. partitioned_cookies_runtime_feature_enabled, std::move(callback));
  73. } else {
  74. std::move(callback).Run(std::vector<net::CookieWithAccessResult>());
  75. }
  76. }
  77. void AwProxyingRestrictedCookieManager::SetCanonicalCookie(
  78. const net::CanonicalCookie& cookie,
  79. const GURL& url,
  80. const net::SiteForCookies& site_for_cookies,
  81. const url::Origin& top_frame_origin,
  82. net::CookieInclusionStatus status,
  83. SetCanonicalCookieCallback callback) {
  84. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  85. if (AllowCookies(url, site_for_cookies)) {
  86. underlying_restricted_cookie_manager_->SetCanonicalCookie(
  87. cookie, url, site_for_cookies, top_frame_origin, status,
  88. std::move(callback));
  89. } else {
  90. std::move(callback).Run(false);
  91. }
  92. }
  93. void AwProxyingRestrictedCookieManager::AddChangeListener(
  94. const GURL& url,
  95. const net::SiteForCookies& site_for_cookies,
  96. const url::Origin& top_frame_origin,
  97. mojo::PendingRemote<network::mojom::CookieChangeListener> listener,
  98. AddChangeListenerCallback callback) {
  99. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  100. mojo::PendingRemote<network::mojom::CookieChangeListener>
  101. proxy_listener_remote;
  102. auto proxy_listener =
  103. std::make_unique<AwProxyingRestrictedCookieManagerListener>(
  104. url, site_for_cookies, weak_factory_.GetWeakPtr(),
  105. std::move(listener));
  106. mojo::MakeSelfOwnedReceiver(
  107. std::move(proxy_listener),
  108. proxy_listener_remote.InitWithNewPipeAndPassReceiver());
  109. underlying_restricted_cookie_manager_->AddChangeListener(
  110. url, site_for_cookies, top_frame_origin, std::move(proxy_listener_remote),
  111. std::move(callback));
  112. }
  113. void AwProxyingRestrictedCookieManager::SetCookieFromString(
  114. const GURL& url,
  115. const net::SiteForCookies& site_for_cookies,
  116. const url::Origin& top_frame_origin,
  117. const std::string& cookie,
  118. bool partitioned_cookies_runtime_feature_enabled,
  119. SetCookieFromStringCallback callback) {
  120. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  121. if (AllowCookies(url, site_for_cookies)) {
  122. underlying_restricted_cookie_manager_->SetCookieFromString(
  123. url, site_for_cookies, top_frame_origin, cookie,
  124. partitioned_cookies_runtime_feature_enabled, std::move(callback));
  125. } else {
  126. std::move(callback).Run(/*site_for_cookies_ok=*/true,
  127. /*top_frame_origin_ok=*/true);
  128. }
  129. }
  130. void AwProxyingRestrictedCookieManager::GetCookiesString(
  131. const GURL& url,
  132. const net::SiteForCookies& site_for_cookies,
  133. const url::Origin& top_frame_origin,
  134. bool partitioned_cookies_runtime_feature_enabled,
  135. GetCookiesStringCallback callback) {
  136. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  137. if (AllowCookies(url, site_for_cookies)) {
  138. underlying_restricted_cookie_manager_->GetCookiesString(
  139. url, site_for_cookies, top_frame_origin,
  140. partitioned_cookies_runtime_feature_enabled, std::move(callback));
  141. } else {
  142. std::move(callback).Run("");
  143. }
  144. }
  145. void AwProxyingRestrictedCookieManager::CookiesEnabledFor(
  146. const GURL& url,
  147. const net::SiteForCookies& site_for_cookies,
  148. const url::Origin& top_frame_origin,
  149. CookiesEnabledForCallback callback) {
  150. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  151. std::move(callback).Run(AllowCookies(url, site_for_cookies));
  152. }
  153. AwProxyingRestrictedCookieManager::AwProxyingRestrictedCookieManager(
  154. mojo::PendingRemote<network::mojom::RestrictedCookieManager>
  155. underlying_restricted_cookie_manager,
  156. bool is_service_worker,
  157. int process_id,
  158. int frame_id)
  159. : underlying_restricted_cookie_manager_(
  160. std::move(underlying_restricted_cookie_manager)),
  161. is_service_worker_(is_service_worker),
  162. process_id_(process_id),
  163. frame_id_(frame_id) {
  164. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  165. }
  166. // static
  167. void AwProxyingRestrictedCookieManager::CreateAndBindOnIoThread(
  168. mojo::PendingRemote<network::mojom::RestrictedCookieManager> underlying_rcm,
  169. bool is_service_worker,
  170. int process_id,
  171. int frame_id,
  172. mojo::PendingReceiver<network::mojom::RestrictedCookieManager> receiver) {
  173. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  174. auto wrapper = base::WrapUnique(new AwProxyingRestrictedCookieManager(
  175. std::move(underlying_rcm), is_service_worker, process_id, frame_id));
  176. mojo::MakeSelfOwnedReceiver(std::move(wrapper), std::move(receiver));
  177. }
  178. bool AwProxyingRestrictedCookieManager::AllowCookies(
  179. const GURL& url,
  180. const net::SiteForCookies& site_for_cookies) const {
  181. if (is_service_worker_) {
  182. // Service worker cookies are always first-party, so only need to check
  183. // the global toggle.
  184. return AwCookieAccessPolicy::GetInstance()->GetShouldAcceptCookies();
  185. } else {
  186. return AwCookieAccessPolicy::GetInstance()->AllowCookies(
  187. url, site_for_cookies, process_id_, frame_id_);
  188. }
  189. }
  190. void AwProxyingRestrictedCookieManager::
  191. ConvertPartitionedCookiesToUnpartitioned(const GURL& url) {
  192. underlying_restricted_cookie_manager_
  193. ->ConvertPartitionedCookiesToUnpartitioned(url);
  194. }
  195. } // namespace android_webview