safe_browsing_service.cc 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292
  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 "weblayer/browser/safe_browsing/safe_browsing_service.h"
  5. #include <memory>
  6. #include "base/bind.h"
  7. #include "base/path_service.h"
  8. #include "components/prefs/pref_service.h"
  9. #include "components/safe_browsing/android/remote_database_manager.h"
  10. #include "components/safe_browsing/android/safe_browsing_api_handler_bridge.h"
  11. #include "components/safe_browsing/content/browser/browser_url_loader_throttle.h"
  12. #include "components/safe_browsing/content/browser/mojo_safe_browsing_impl.h"
  13. #include "components/safe_browsing/content/browser/safe_browsing_navigation_throttle.h"
  14. #include "components/safe_browsing/content/browser/safe_browsing_network_context.h"
  15. #include "components/safe_browsing/content/browser/triggers/trigger_manager.h"
  16. #include "components/safe_browsing/core/browser/realtime/url_lookup_service.h"
  17. #include "components/safe_browsing/core/common/safe_browsing_prefs.h"
  18. #include "content/public/browser/browser_context.h"
  19. #include "content/public/browser/browser_task_traits.h"
  20. #include "content/public/browser/browser_thread.h"
  21. #include "content/public/browser/network_service_instance.h"
  22. #include "content/public/browser/render_process_host.h"
  23. #include "content/public/browser/resource_context.h"
  24. #include "services/cert_verifier/public/mojom/cert_verifier_service_factory.mojom.h"
  25. #include "services/network/public/mojom/network_context.mojom.h"
  26. #include "services/network/public/mojom/network_service.mojom.h"
  27. #include "third_party/blink/public/common/loader/url_loader_throttle.h"
  28. #include "weblayer/browser/browser_context_impl.h"
  29. #include "weblayer/browser/browser_process.h"
  30. #include "weblayer/browser/safe_browsing/url_checker_delegate_impl.h"
  31. #include "weblayer/browser/safe_browsing/weblayer_safe_browsing_blocking_page_factory.h"
  32. #include "weblayer/browser/safe_browsing/weblayer_ui_manager_delegate.h"
  33. #include "weblayer/common/features.h"
  34. namespace weblayer {
  35. namespace {
  36. network::mojom::NetworkContextParamsPtr CreateDefaultNetworkContextParams(
  37. const std::string& user_agent) {
  38. network::mojom::NetworkContextParamsPtr network_context_params =
  39. network::mojom::NetworkContextParams::New();
  40. network_context_params->cert_verifier_params = content::GetCertVerifierParams(
  41. cert_verifier::mojom::CertVerifierCreationParams::New());
  42. network_context_params->user_agent = user_agent;
  43. return network_context_params;
  44. }
  45. // Helper method that checks the RenderProcessHost is still alive and checks the
  46. // latest Safe Browsing pref value on the UI thread before hopping over to the
  47. // IO thread.
  48. void MaybeCreateSafeBrowsing(
  49. int rph_id,
  50. content::ResourceContext* resource_context,
  51. base::RepeatingCallback<scoped_refptr<safe_browsing::UrlCheckerDelegate>()>
  52. get_checker_delegate,
  53. mojo::PendingReceiver<safe_browsing::mojom::SafeBrowsing> receiver) {
  54. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  55. content::RenderProcessHost* render_process_host =
  56. content::RenderProcessHost::FromID(rph_id);
  57. if (!render_process_host)
  58. return;
  59. bool is_safe_browsing_enabled = safe_browsing::IsSafeBrowsingEnabled(
  60. *static_cast<BrowserContextImpl*>(
  61. render_process_host->GetBrowserContext())
  62. ->pref_service());
  63. if (!is_safe_browsing_enabled)
  64. return;
  65. content::GetIOThreadTaskRunner({})->PostTask(
  66. FROM_HERE,
  67. base::BindOnce(&safe_browsing::MojoSafeBrowsingImpl::MaybeCreate, rph_id,
  68. resource_context, std::move(get_checker_delegate),
  69. std::move(receiver)));
  70. }
  71. } // namespace
  72. SafeBrowsingService::SafeBrowsingService(const std::string& user_agent)
  73. : user_agent_(user_agent) {}
  74. SafeBrowsingService::~SafeBrowsingService() = default;
  75. void SafeBrowsingService::Initialize() {
  76. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  77. if (network_context_) {
  78. // already initialized
  79. return;
  80. }
  81. base::FilePath user_data_dir;
  82. bool result =
  83. base::PathService::Get(base::DIR_ANDROID_APP_DATA, &user_data_dir);
  84. DCHECK(result);
  85. // safebrowsing network context needs to be created on the UI thread.
  86. network_context_ =
  87. std::make_unique<safe_browsing::SafeBrowsingNetworkContext>(
  88. user_data_dir, /*trigger_migration=*/false,
  89. base::BindRepeating(CreateDefaultNetworkContextParams, user_agent_));
  90. CreateSafeBrowsingUIManager();
  91. // Needs to happen after |ui_manager_| is created.
  92. CreateTriggerManager();
  93. }
  94. std::unique_ptr<blink::URLLoaderThrottle>
  95. SafeBrowsingService::CreateURLLoaderThrottle(
  96. const base::RepeatingCallback<content::WebContents*()>& wc_getter,
  97. int frame_tree_node_id,
  98. safe_browsing::RealTimeUrlLookupServiceBase* url_lookup_service) {
  99. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  100. return safe_browsing::BrowserURLLoaderThrottle::Create(
  101. base::BindOnce(
  102. [](SafeBrowsingService* sb_service) {
  103. return sb_service->GetSafeBrowsingUrlCheckerDelegate();
  104. },
  105. base::Unretained(this)),
  106. wc_getter, frame_tree_node_id,
  107. url_lookup_service ? url_lookup_service->GetWeakPtr() : nullptr);
  108. }
  109. std::unique_ptr<content::NavigationThrottle>
  110. SafeBrowsingService::MaybeCreateSafeBrowsingNavigationThrottleFor(
  111. content::NavigationHandle* handle) {
  112. if (!base::FeatureList::IsEnabled(features::kWebLayerSafeBrowsing)) {
  113. return nullptr;
  114. }
  115. return safe_browsing::SafeBrowsingNavigationThrottle::MaybeCreateThrottleFor(
  116. handle, GetSafeBrowsingUIManager().get());
  117. }
  118. scoped_refptr<safe_browsing::UrlCheckerDelegate>
  119. SafeBrowsingService::GetSafeBrowsingUrlCheckerDelegate() {
  120. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  121. if (!safe_browsing_url_checker_delegate_) {
  122. safe_browsing_url_checker_delegate_ = new UrlCheckerDelegateImpl(
  123. GetSafeBrowsingDBManager(), GetSafeBrowsingUIManager());
  124. }
  125. return safe_browsing_url_checker_delegate_;
  126. }
  127. scoped_refptr<safe_browsing::RemoteSafeBrowsingDatabaseManager>
  128. SafeBrowsingService::GetSafeBrowsingDBManager() {
  129. if (!safe_browsing_db_manager_) {
  130. CreateAndStartSafeBrowsingDBManager();
  131. }
  132. return safe_browsing_db_manager_;
  133. }
  134. scoped_refptr<safe_browsing::SafeBrowsingUIManager>
  135. SafeBrowsingService::GetSafeBrowsingUIManager() {
  136. return ui_manager_;
  137. }
  138. safe_browsing::TriggerManager* SafeBrowsingService::GetTriggerManager() {
  139. return trigger_manager_.get();
  140. }
  141. void SafeBrowsingService::CreateSafeBrowsingUIManager() {
  142. DCHECK(!ui_manager_);
  143. ui_manager_ = new safe_browsing::SafeBrowsingUIManager(
  144. std::make_unique<WebLayerSafeBrowsingUIManagerDelegate>(),
  145. std::make_unique<WebLayerSafeBrowsingBlockingPageFactory>(),
  146. GURL(url::kAboutBlankURL));
  147. }
  148. void SafeBrowsingService::CreateTriggerManager() {
  149. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  150. trigger_manager_ = std::make_unique<safe_browsing::TriggerManager>(
  151. ui_manager_.get(), BrowserProcess::GetInstance()->GetLocalState());
  152. }
  153. void SafeBrowsingService::CreateAndStartSafeBrowsingDBManager() {
  154. DCHECK(!safe_browsing_db_manager_);
  155. safe_browsing_db_manager_ =
  156. new safe_browsing::RemoteSafeBrowsingDatabaseManager();
  157. if (!content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)) {
  158. // Posting a task to start the DB here ensures that it will be started by
  159. // the time that a consumer uses it on the IO thread, as such a consumer
  160. // would need to make it available for usage on the IO thread via a
  161. // PostTask() that will be ordered after this one.
  162. content::GetIOThreadTaskRunner({})->PostTask(
  163. FROM_HERE,
  164. base::BindOnce(
  165. &SafeBrowsingService::StartSafeBrowsingDBManagerOnIOThread,
  166. base::Unretained(this)));
  167. } else {
  168. StartSafeBrowsingDBManagerOnIOThread();
  169. }
  170. }
  171. void SafeBrowsingService::StartSafeBrowsingDBManagerOnIOThread() {
  172. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  173. DCHECK(safe_browsing_db_manager_);
  174. if (started_db_manager_)
  175. return;
  176. started_db_manager_ = true;
  177. // V4ProtocolConfig is not used. Just create one with empty values.
  178. safe_browsing::V4ProtocolConfig config("", false, "", "");
  179. safe_browsing_db_manager_->StartOnIOThread(GetURLLoaderFactoryOnIOThread(),
  180. config);
  181. }
  182. scoped_refptr<network::SharedURLLoaderFactory>
  183. SafeBrowsingService::GetURLLoaderFactoryOnIOThread() {
  184. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  185. if (!shared_url_loader_factory_on_io_) {
  186. content::GetUIThreadTaskRunner({})->PostTask(
  187. FROM_HERE,
  188. base::BindOnce(&SafeBrowsingService::CreateURLLoaderFactoryForIO,
  189. base::Unretained(this),
  190. url_loader_factory_on_io_.BindNewPipeAndPassReceiver()));
  191. shared_url_loader_factory_on_io_ =
  192. base::MakeRefCounted<network::WeakWrapperSharedURLLoaderFactory>(
  193. url_loader_factory_on_io_.get());
  194. }
  195. return shared_url_loader_factory_on_io_;
  196. }
  197. void SafeBrowsingService::CreateURLLoaderFactoryForIO(
  198. mojo::PendingReceiver<network::mojom::URLLoaderFactory> receiver) {
  199. DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  200. auto url_loader_factory_params =
  201. network::mojom::URLLoaderFactoryParams::New();
  202. url_loader_factory_params->process_id = network::mojom::kBrowserProcessId;
  203. url_loader_factory_params->is_corb_enabled = false;
  204. network_context_->GetNetworkContext()->CreateURLLoaderFactory(
  205. std::move(receiver), std::move(url_loader_factory_params));
  206. }
  207. void SafeBrowsingService::AddInterface(
  208. service_manager::BinderRegistry* registry,
  209. content::RenderProcessHost* render_process_host) {
  210. content::ResourceContext* resource_context =
  211. render_process_host->GetBrowserContext()->GetResourceContext();
  212. registry->AddInterface(
  213. base::BindRepeating(
  214. &MaybeCreateSafeBrowsing, render_process_host->GetID(),
  215. resource_context,
  216. base::BindRepeating(
  217. &SafeBrowsingService::GetSafeBrowsingUrlCheckerDelegate,
  218. base::Unretained(this))),
  219. content::GetUIThreadTaskRunner({}));
  220. }
  221. void SafeBrowsingService::StopDBManager() {
  222. content::GetIOThreadTaskRunner({})->PostTask(
  223. FROM_HERE, base::BindOnce(&SafeBrowsingService::StopDBManagerOnIOThread,
  224. base::Unretained(this)));
  225. }
  226. void SafeBrowsingService::StopDBManagerOnIOThread() {
  227. DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
  228. if (safe_browsing_db_manager_) {
  229. safe_browsing_db_manager_->StopOnIOThread(true /*shutdown*/);
  230. safe_browsing_db_manager_.reset();
  231. started_db_manager_ = false;
  232. }
  233. }
  234. network::mojom::NetworkContext* SafeBrowsingService::GetNetworkContext() {
  235. if (!network_context_)
  236. return nullptr;
  237. return network_context_->GetNetworkContext();
  238. }
  239. scoped_refptr<network::SharedURLLoaderFactory>
  240. SafeBrowsingService::GetURLLoaderFactory() {
  241. if (!network_context_)
  242. return nullptr;
  243. return network_context_->GetURLLoaderFactory();
  244. }
  245. } // namespace weblayer