| // Copyright 2018 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "content/test/storage_partition_test_utils.h" |
| |
| #include "base/bind.h" |
| #include "base/run_loop.h" |
| #include "base/task/post_task.h" |
| #include "content/browser/url_loader_factory_getter.h" |
| #include "content/public/browser/browser_task_traits.h" |
| #include "content/public/test/simple_url_loader_test_helper.h" |
| #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
| #include "services/network/public/cpp/simple_url_loader.h" |
| |
| namespace content { |
| namespace { |
| |
| using SharedURLLoaderFactoryGetterCallback = |
| base::OnceCallback<scoped_refptr<network::SharedURLLoaderFactory>()>; |
| |
| void InitializeSharedFactoryOnIOThread( |
| SharedURLLoaderFactoryGetterCallback shared_url_loader_factory_getter, |
| scoped_refptr<network::SharedURLLoaderFactory>* out_shared_factory) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::RunLoop run_loop; |
| base::PostTaskWithTraitsAndReply( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| [](SharedURLLoaderFactoryGetterCallback getter, |
| scoped_refptr<network::SharedURLLoaderFactory>* |
| shared_factory_ptr) { |
| DCHECK(!shared_factory_ptr->get()) |
| << "shared_url_loader_factory_ can only be initialized once."; |
| *shared_factory_ptr = std::move(getter).Run(); |
| }, |
| std::move(shared_url_loader_factory_getter), |
| base::Unretained(out_shared_factory)), |
| run_loop.QuitClosure()); |
| run_loop.Run(); |
| } |
| |
| network::SimpleURLLoader::BodyAsStringCallback RunOnUIThread( |
| network::SimpleURLLoader::BodyAsStringCallback ui_callback) { |
| return base::BindOnce( |
| [](network::SimpleURLLoader::BodyAsStringCallback callback, |
| std::unique_ptr<std::string> response_body) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), std::move(response_body))); |
| }, |
| std::move(ui_callback)); |
| } |
| |
| } // namespace |
| |
| // static |
| IOThreadSharedURLLoaderFactoryOwner::IOThreadSharedURLLoaderFactoryOwnerPtr |
| IOThreadSharedURLLoaderFactoryOwner::Create( |
| URLLoaderFactoryGetter* url_loader_factory_getter) { |
| return IOThreadSharedURLLoaderFactoryOwnerPtr( |
| new IOThreadSharedURLLoaderFactoryOwner(url_loader_factory_getter)); |
| } |
| |
| // static |
| IOThreadSharedURLLoaderFactoryOwner::IOThreadSharedURLLoaderFactoryOwnerPtr |
| IOThreadSharedURLLoaderFactoryOwner::Create( |
| std::unique_ptr<network::SharedURLLoaderFactoryInfo> info) { |
| return IOThreadSharedURLLoaderFactoryOwnerPtr( |
| new IOThreadSharedURLLoaderFactoryOwner(std::move(info))); |
| } |
| |
| IOThreadSharedURLLoaderFactoryOwner::IOThreadSharedURLLoaderFactoryOwner( |
| URLLoaderFactoryGetter* url_loader_factory_getter) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| InitializeSharedFactoryOnIOThread( |
| base::BindOnce(&URLLoaderFactoryGetter::GetNetworkFactory, |
| base::Unretained(url_loader_factory_getter)), |
| &shared_url_loader_factory_); |
| } |
| |
| IOThreadSharedURLLoaderFactoryOwner::IOThreadSharedURLLoaderFactoryOwner( |
| std::unique_ptr<network::SharedURLLoaderFactoryInfo> info) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| InitializeSharedFactoryOnIOThread( |
| base::BindOnce(&network::SharedURLLoaderFactory::Create, std::move(info)), |
| &shared_url_loader_factory_); |
| } |
| |
| IOThreadSharedURLLoaderFactoryOwner::~IOThreadSharedURLLoaderFactoryOwner() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| } |
| |
| int IOThreadSharedURLLoaderFactoryOwner::LoadBasicRequestOnIOThread( |
| const GURL& url) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| auto request = std::make_unique<network::ResourceRequest>(); |
| request->url = url; |
| |
| // |simple_loader_helper| lives on UI thread and shouldn't be accessed on |
| // other threads. |
| SimpleURLLoaderTestHelper simple_loader_helper; |
| |
| std::unique_ptr<network::SimpleURLLoader> simple_loader = |
| network::SimpleURLLoader::Create(std::move(request), |
| TRAFFIC_ANNOTATION_FOR_TESTS); |
| |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| [](network::SimpleURLLoader* loader, |
| network::mojom::URLLoaderFactory* factory, |
| network::SimpleURLLoader::BodyAsStringCallback |
| body_as_string_callback) { |
| loader->DownloadToStringOfUnboundedSizeUntilCrashAndDie( |
| factory, std::move(body_as_string_callback)); |
| }, |
| base::Unretained(simple_loader.get()), |
| base::Unretained(shared_url_loader_factory_.get()), |
| RunOnUIThread(simple_loader_helper.GetCallback()))); |
| |
| simple_loader_helper.WaitForCallback(); |
| return simple_loader->NetError(); |
| } |
| |
| } // namespace content |