| // Copyright 2018 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "third_party/blink/public/platform/child_url_loader_factory_bundle.h" |
| |
| #include <memory> |
| #include <optional> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/check.h" |
| #include "mojo/public/cpp/bindings/self_owned_receiver.h" |
| #include "net/base/load_flags.h" |
| #include "services/network/public/cpp/record_ontransfersizeupdate_utils.h" |
| #include "services/network/public/cpp/resource_request.h" |
| #include "services/network/public/mojom/early_hints.mojom.h" |
| #include "services/network/public/mojom/url_loader.mojom.h" |
| #include "third_party/blink/public/common/features.h" |
| #include "third_party/blink/public/mojom/loader/resource_load_info.mojom-shared.h" |
| #include "url/gurl.h" |
| #include "url/origin.h" |
| #include "url/url_constants.h" |
| |
| namespace blink { |
| |
| namespace { |
| |
| class URLLoaderRelay : public network::mojom::URLLoaderClient, |
| public network::mojom::URLLoader { |
| public: |
| URLLoaderRelay( |
| mojo::PendingRemote<network::mojom::URLLoader> loader_sink, |
| mojo::PendingReceiver<network::mojom::URLLoaderClient> client_source, |
| mojo::Remote<network::mojom::URLLoaderClient> client_sink) |
| : loader_sink_(std::move(loader_sink)), |
| client_source_receiver_(this, std::move(client_source)), |
| client_sink_(std::move(client_sink)) {} |
| |
| // network::mojom::URLLoader implementation: |
| void FollowRedirect( |
| const std::vector<std::string>& removed_headers, |
| const net::HttpRequestHeaders& modified_request_headers, |
| const net::HttpRequestHeaders& modified_cors_exempt_request_headers, |
| const std::optional<GURL>& new_url) override { |
| DCHECK(removed_headers.empty() && modified_request_headers.IsEmpty() && |
| modified_cors_exempt_request_headers.IsEmpty()) |
| << "Redirect with removed or modified headers was not supported yet. " |
| "crbug.com/845683"; |
| DCHECK(!new_url.has_value()) |
| << "Redirect with modified URL was not supported yet. " |
| "crbug.com/845683"; |
| loader_sink_->FollowRedirect( |
| {} /* removed_headers */, {} /* modified_headers */, |
| {} /* modified_cors_exempt_headers */, std::nullopt /* new_url */); |
| } |
| |
| void SetPriority(net::RequestPriority priority, |
| int32_t intra_priority_value) override { |
| loader_sink_->SetPriority(priority, intra_priority_value); |
| } |
| |
| void PauseReadingBodyFromNet() override { |
| loader_sink_->PauseReadingBodyFromNet(); |
| } |
| |
| void ResumeReadingBodyFromNet() override { |
| loader_sink_->ResumeReadingBodyFromNet(); |
| } |
| |
| // network::mojom::URLLoaderClient implementation: |
| void OnReceiveEarlyHints(network::mojom::EarlyHintsPtr early_hints) override { |
| client_sink_->OnReceiveEarlyHints(std::move(early_hints)); |
| } |
| |
| void OnReceiveResponse( |
| network::mojom::URLResponseHeadPtr head, |
| mojo::ScopedDataPipeConsumerHandle body, |
| std::optional<mojo_base::BigBuffer> cached_metadata) override { |
| client_sink_->OnReceiveResponse(std::move(head), std::move(body), |
| std::move(cached_metadata)); |
| } |
| |
| void OnReceiveRedirect(const net::RedirectInfo& redirect_info, |
| network::mojom::URLResponseHeadPtr head) override { |
| client_sink_->OnReceiveRedirect(redirect_info, std::move(head)); |
| } |
| |
| void OnUploadProgress(int64_t current_position, |
| int64_t total_size, |
| OnUploadProgressCallback callback) override { |
| client_sink_->OnUploadProgress(current_position, total_size, |
| std::move(callback)); |
| } |
| |
| void OnTransferSizeUpdated(int32_t transfer_size_diff) override { |
| network::RecordOnTransferSizeUpdatedUMA( |
| network::OnTransferSizeUpdatedFrom::kURLLoaderRelay); |
| |
| client_sink_->OnTransferSizeUpdated(transfer_size_diff); |
| } |
| |
| void OnComplete(const network::URLLoaderCompletionStatus& status) override { |
| client_sink_->OnComplete(status); |
| } |
| |
| private: |
| mojo::Remote<network::mojom::URLLoader> loader_sink_; |
| mojo::Receiver<network::mojom::URLLoaderClient> client_source_receiver_; |
| mojo::Remote<network::mojom::URLLoaderClient> client_sink_; |
| }; |
| |
| template <typename TKey> |
| static std::map<TKey, mojo::PendingRemote<network::mojom::URLLoaderFactory>> |
| BoundRemoteMapToPendingRemoteMap( |
| std::map<TKey, mojo::Remote<network::mojom::URLLoaderFactory>> input) { |
| std::map<TKey, mojo::PendingRemote<network::mojom::URLLoaderFactory>> output; |
| for (auto& it : input) { |
| const TKey& key = it.first; |
| mojo::Remote<network::mojom::URLLoaderFactory>& factory = it.second; |
| if (factory.is_bound()) |
| output.emplace(key, factory.Unbind()); |
| } |
| return output; |
| } |
| |
| } // namespace |
| |
| ChildPendingURLLoaderFactoryBundle::ChildPendingURLLoaderFactoryBundle() = |
| default; |
| |
| ChildPendingURLLoaderFactoryBundle::ChildPendingURLLoaderFactoryBundle( |
| std::unique_ptr<PendingURLLoaderFactoryBundle> base_factories) |
| : PendingURLLoaderFactoryBundle( |
| std::move(base_factories->pending_default_factory()), |
| std::move(base_factories->pending_scheme_specific_factories()), |
| std::move(base_factories->pending_isolated_world_factories()), |
| base_factories->bypass_redirect_checks()) { |
| } |
| |
| ChildPendingURLLoaderFactoryBundle::ChildPendingURLLoaderFactoryBundle( |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_default_factory, |
| SchemeMap pending_scheme_specific_factories, |
| OriginMap pending_isolated_world_factories, |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_subresource_proxying_loader_factory, |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_keep_alive_loader_factory, |
| mojo::PendingAssociatedRemote<blink::mojom::FetchLaterLoaderFactory> |
| pending_fetch_later_loader_factory, |
| bool bypass_redirect_checks) |
| : PendingURLLoaderFactoryBundle( |
| std::move(pending_default_factory), |
| std::move(pending_scheme_specific_factories), |
| std::move(pending_isolated_world_factories), |
| bypass_redirect_checks), |
| pending_subresource_proxying_loader_factory_( |
| std::move(pending_subresource_proxying_loader_factory)), |
| pending_keep_alive_loader_factory_( |
| std::move(pending_keep_alive_loader_factory)), |
| pending_fetch_later_loader_factory_( |
| std::move(pending_fetch_later_loader_factory)) {} |
| |
| ChildPendingURLLoaderFactoryBundle::~ChildPendingURLLoaderFactoryBundle() = |
| default; |
| |
| scoped_refptr<network::SharedURLLoaderFactory> |
| ChildPendingURLLoaderFactoryBundle::CreateFactory() { |
| auto other = std::make_unique<ChildPendingURLLoaderFactoryBundle>(); |
| other->pending_default_factory_ = std::move(pending_default_factory_); |
| other->pending_scheme_specific_factories_ = |
| std::move(pending_scheme_specific_factories_); |
| other->pending_isolated_world_factories_ = |
| std::move(pending_isolated_world_factories_); |
| other->pending_subresource_proxying_loader_factory_ = |
| std::move(pending_subresource_proxying_loader_factory_); |
| other->pending_keep_alive_loader_factory_ = |
| std::move(pending_keep_alive_loader_factory_); |
| other->pending_fetch_later_loader_factory_ = |
| std::move(pending_fetch_later_loader_factory_); |
| other->bypass_redirect_checks_ = bypass_redirect_checks_; |
| |
| return base::MakeRefCounted<ChildURLLoaderFactoryBundle>(std::move(other)); |
| } |
| |
| // ----------------------------------------------------------------------------- |
| |
| ChildURLLoaderFactoryBundle::ChildURLLoaderFactoryBundle() = default; |
| |
| ChildURLLoaderFactoryBundle::ChildURLLoaderFactoryBundle( |
| std::unique_ptr<ChildPendingURLLoaderFactoryBundle> pending_factories) { |
| Update(std::move(pending_factories)); |
| } |
| |
| ChildURLLoaderFactoryBundle::~ChildURLLoaderFactoryBundle() = default; |
| |
| void ChildURLLoaderFactoryBundle::CreateLoaderAndStart( |
| mojo::PendingReceiver<network::mojom::URLLoader> loader, |
| int32_t request_id, |
| uint32_t options, |
| const network::ResourceRequest& request, |
| mojo::PendingRemote<network::mojom::URLLoaderClient> client, |
| const net::MutableNetworkTrafficAnnotationTag& traffic_annotation) { |
| auto override_iter = subresource_overrides_.find(request.url); |
| if (override_iter != subresource_overrides_.end()) { |
| blink::mojom::TransferrableURLLoaderPtr transferrable_loader = |
| std::move(override_iter->second); |
| subresource_overrides_.erase(override_iter); |
| |
| mojo::Remote<network::mojom::URLLoaderClient> client_remote( |
| std::move(client)); |
| client_remote->OnReceiveResponse(std::move(transferrable_loader->head), |
| std::move(transferrable_loader->body), |
| std::nullopt); |
| mojo::MakeSelfOwnedReceiver( |
| std::make_unique<URLLoaderRelay>( |
| std::move(transferrable_loader->url_loader), |
| std::move(transferrable_loader->url_loader_client), |
| std::move(client_remote)), |
| std::move(loader)); |
| |
| return; |
| } |
| |
| // Prefetch is disjoint with browsing_topics, ad_auction_headers, and |
| // keepalive. |
| // TODO(https://crbug.com/1441113): keepalive is disjoint with browsing_topics |
| // and ad_auction_headers in our implementation, but the fetch API does not |
| // enforce this, so `subresource_proxying_loader_factory_` (that handles |
| // browsing_topics and ad_auction_headers) wins and keepalive is ignored. |
| // Either allow them simultaneously or make them mutually exclusive in the |
| // fetch API. |
| const bool request_is_prefetch = request.load_flags & net::LOAD_PREFETCH; |
| CHECK(!(request_is_prefetch && request.browsing_topics)); |
| CHECK(!(request_is_prefetch && request.ad_auction_headers)); |
| CHECK(!(request_is_prefetch && request.keepalive)); |
| |
| // Use |subresource_proxying_loader_factory_| for prefetch, browsing_topics, |
| // and ad_auction_headers requests to send the requests to |
| // `SubresourceProxyingURLLoaderService` in the browser process and trigger |
| // the special handling. |
| // TODO(horo): Move this routing logic to network service, when we will have |
| // the special prefetch handling in network service. |
| if ((request_is_prefetch || request.browsing_topics || |
| request.ad_auction_headers) && |
| subresource_proxying_loader_factory_) { |
| // For prefetch, this is no-state prefetch (see |
| // WebURLRequest::GetLoadFlagsForWebUrlRequest). |
| subresource_proxying_loader_factory_->CreateLoaderAndStart( |
| std::move(loader), request_id, options, request, std::move(client), |
| traffic_annotation); |
| return; |
| } |
| |
| // Use |keep_alive_loader_factory_| to send the keepalive requests to the |
| // KeepAliveURLLoaderService in the browser process and trigger the special |
| // keepalive request handling. |
| // |keep_alive_loader_factory_| only presents when |
| // features::kKeepAliveInBrowserMigration is true. |
| if (request.keepalive && keep_alive_loader_factory_ && |
| base::FeatureList::IsEnabled(features::kKeepAliveInBrowserMigration) && |
| (request.attribution_reporting_eligibility == |
| network::mojom::AttributionReportingEligibility::kUnset || |
| base::FeatureList::IsEnabled( |
| features::kAttributionReportingInBrowserMigration))) { |
| keep_alive_loader_factory_->CreateLoaderAndStart( |
| std::move(loader), request_id, options, request, std::move(client), |
| traffic_annotation); |
| return; |
| } |
| |
| // Default request handling. |
| URLLoaderFactoryBundle::CreateLoaderAndStart( |
| std::move(loader), request_id, options, request, std::move(client), |
| traffic_annotation); |
| } |
| |
| std::unique_ptr<network::PendingSharedURLLoaderFactory> |
| ChildURLLoaderFactoryBundle::Clone() { |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| default_factory_pending_remote; |
| if (default_factory_) { |
| default_factory_->Clone( |
| default_factory_pending_remote.InitWithNewPipeAndPassReceiver()); |
| } |
| |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_subresource_proxying_loader_factory; |
| if (subresource_proxying_loader_factory_) { |
| subresource_proxying_loader_factory_->Clone( |
| pending_subresource_proxying_loader_factory |
| .InitWithNewPipeAndPassReceiver()); |
| } |
| |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_keep_alive_loader_factory; |
| if (keep_alive_loader_factory_) { |
| keep_alive_loader_factory_->Clone( |
| pending_keep_alive_loader_factory.InitWithNewPipeAndPassReceiver()); |
| } |
| mojo::PendingAssociatedRemote<blink::mojom::FetchLaterLoaderFactory> |
| pending_fetch_later_loader_factory; |
| if (fetch_later_loader_factory_) { |
| fetch_later_loader_factory_->Clone( |
| pending_fetch_later_loader_factory |
| .InitWithNewEndpointAndPassReceiver()); |
| } |
| |
| // Currently there is no need to override subresources from workers, |
| // therefore |subresource_overrides| are not shared with the clones. |
| |
| return std::make_unique<ChildPendingURLLoaderFactoryBundle>( |
| std::move(default_factory_pending_remote), |
| CloneRemoteMapToPendingRemoteMap(scheme_specific_factories_), |
| CloneRemoteMapToPendingRemoteMap(isolated_world_factories_), |
| std::move(pending_subresource_proxying_loader_factory), |
| std::move(pending_keep_alive_loader_factory), |
| std::move(pending_fetch_later_loader_factory), bypass_redirect_checks_); |
| } |
| |
| std::unique_ptr<ChildPendingURLLoaderFactoryBundle> |
| ChildURLLoaderFactoryBundle::PassInterface() { |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> pending_default_factory; |
| if (default_factory_) |
| pending_default_factory = default_factory_.Unbind(); |
| |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_subresource_proxying_loader_factory; |
| if (subresource_proxying_loader_factory_) { |
| pending_subresource_proxying_loader_factory = |
| subresource_proxying_loader_factory_.Unbind(); |
| } |
| |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| pending_keep_alive_loader_factory; |
| if (keep_alive_loader_factory_) { |
| pending_keep_alive_loader_factory = keep_alive_loader_factory_.Unbind(); |
| } |
| mojo::PendingAssociatedRemote<blink::mojom::FetchLaterLoaderFactory> |
| pending_fetch_later_loader_factory; |
| if (fetch_later_loader_factory_) { |
| pending_fetch_later_loader_factory = fetch_later_loader_factory_.Unbind(); |
| } |
| |
| return std::make_unique<ChildPendingURLLoaderFactoryBundle>( |
| std::move(pending_default_factory), |
| BoundRemoteMapToPendingRemoteMap(std::move(scheme_specific_factories_)), |
| BoundRemoteMapToPendingRemoteMap(std::move(isolated_world_factories_)), |
| std::move(pending_subresource_proxying_loader_factory), |
| std::move(pending_keep_alive_loader_factory), |
| std::move(pending_fetch_later_loader_factory), bypass_redirect_checks_); |
| } |
| |
| void ChildURLLoaderFactoryBundle::Update( |
| std::unique_ptr<ChildPendingURLLoaderFactoryBundle> pending_factories) { |
| if (pending_factories->pending_subresource_proxying_loader_factory()) { |
| subresource_proxying_loader_factory_.Bind(std::move( |
| pending_factories->pending_subresource_proxying_loader_factory())); |
| } |
| if (pending_factories->pending_keep_alive_loader_factory()) { |
| keep_alive_loader_factory_.Bind( |
| std::move(pending_factories->pending_keep_alive_loader_factory())); |
| } |
| if (pending_factories->pending_fetch_later_loader_factory()) { |
| fetch_later_loader_factory_.Bind( |
| std::move(pending_factories->pending_fetch_later_loader_factory())); |
| } |
| URLLoaderFactoryBundle::Update(std::move(pending_factories)); |
| } |
| |
| void ChildURLLoaderFactoryBundle::UpdateSubresourceOverrides( |
| std::vector<blink::mojom::TransferrableURLLoaderPtr>* |
| subresource_overrides) { |
| for (auto& element : *subresource_overrides) |
| subresource_overrides_[element->url] = std::move(element); |
| } |
| |
| void ChildURLLoaderFactoryBundle::SetSubresourceProxyingLoaderFactory( |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| subresource_proxying_loader_factory) { |
| subresource_proxying_loader_factory_.Bind( |
| std::move(subresource_proxying_loader_factory)); |
| } |
| |
| void ChildURLLoaderFactoryBundle::SetKeepAliveLoaderFactory( |
| mojo::PendingRemote<network::mojom::URLLoaderFactory> |
| keep_alive_loader_factory) { |
| keep_alive_loader_factory_.Bind(std::move(keep_alive_loader_factory)); |
| } |
| |
| void ChildURLLoaderFactoryBundle::SetFetchLaterLoaderFactory( |
| mojo::PendingAssociatedRemote<blink::mojom::FetchLaterLoaderFactory> |
| fetch_later_loader_factory) { |
| fetch_later_loader_factory_.Bind(std::move(fetch_later_loader_factory)); |
| } |
| |
| void ChildURLLoaderFactoryBundle::CreateFetchLaterLoader( |
| blink::CrossVariantMojoAssociatedReceiver< |
| mojom::FetchLaterLoaderInterfaceBase> loader, |
| int32_t request_id, |
| uint32_t options, |
| const network::ResourceRequest& request, |
| const net::MutableNetworkTrafficAnnotationTag& traffic_annotation) { |
| fetch_later_loader_factory_->CreateLoader( |
| std::move(loader), request_id, options, request, traffic_annotation); |
| } |
| |
| bool ChildURLLoaderFactoryBundle::IsHostChildURLLoaderFactoryBundle() const { |
| return false; |
| } |
| |
| } // namespace blink |