| // Copyright 2017 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/common/url_loader_factory_bundle.h" |
| |
| #include <utility> |
| |
| #include "base/logging.h" |
| #include "services/network/public/cpp/resource_request.h" |
| #include "url/gurl.h" |
| |
| namespace content { |
| |
| namespace { |
| |
| template <typename TKey> |
| void BindPtrInfoMapToPtrMap( |
| std::map<TKey, network::mojom::URLLoaderFactoryPtr>* target, |
| std::map<TKey, network::mojom::URLLoaderFactoryPtrInfo> input) { |
| for (auto& it : input) { |
| const TKey& key = it.first; |
| network::mojom::URLLoaderFactoryPtrInfo& factory_info = it.second; |
| (*target)[key].Bind(std::move(factory_info)); |
| } |
| } |
| |
| } // namespace |
| |
| URLLoaderFactoryBundleInfo::URLLoaderFactoryBundleInfo() = default; |
| |
| URLLoaderFactoryBundleInfo::URLLoaderFactoryBundleInfo( |
| network::mojom::URLLoaderFactoryPtrInfo default_factory_info, |
| SchemeMap scheme_specific_factory_infos, |
| OriginMap initiator_specific_factory_infos, |
| bool bypass_redirect_checks) |
| : default_factory_info_(std::move(default_factory_info)), |
| scheme_specific_factory_infos_(std::move(scheme_specific_factory_infos)), |
| initiator_specific_factory_infos_( |
| std::move(initiator_specific_factory_infos)), |
| bypass_redirect_checks_(bypass_redirect_checks) {} |
| |
| URLLoaderFactoryBundleInfo::~URLLoaderFactoryBundleInfo() = default; |
| |
| scoped_refptr<network::SharedURLLoaderFactory> |
| URLLoaderFactoryBundleInfo::CreateFactory() { |
| auto other = std::make_unique<URLLoaderFactoryBundleInfo>(); |
| other->default_factory_info_ = std::move(default_factory_info_); |
| other->appcache_factory_info_ = std::move(appcache_factory_info_); |
| other->scheme_specific_factory_infos_ = |
| std::move(scheme_specific_factory_infos_); |
| other->initiator_specific_factory_infos_ = |
| std::move(initiator_specific_factory_infos_); |
| other->bypass_redirect_checks_ = bypass_redirect_checks_; |
| |
| return base::MakeRefCounted<URLLoaderFactoryBundle>(std::move(other)); |
| } |
| |
| // ----------------------------------------------------------------------------- |
| |
| URLLoaderFactoryBundle::URLLoaderFactoryBundle() = default; |
| |
| URLLoaderFactoryBundle::URLLoaderFactoryBundle( |
| std::unique_ptr<URLLoaderFactoryBundleInfo> info) { |
| Update(std::move(info)); |
| } |
| |
| URLLoaderFactoryBundle::~URLLoaderFactoryBundle() = default; |
| |
| void URLLoaderFactoryBundle::SetDefaultFactory( |
| network::mojom::URLLoaderFactoryPtr factory) { |
| default_factory_ = std::move(factory); |
| } |
| |
| network::mojom::URLLoaderFactory* URLLoaderFactoryBundle::GetFactory( |
| const network::ResourceRequest& request) { |
| auto it = scheme_specific_factories_.find(request.url.scheme()); |
| if (it != scheme_specific_factories_.end()) |
| return it->second.get(); |
| |
| if (request.request_initiator.has_value()) { |
| auto it2 = |
| initiator_specific_factories_.find(request.request_initiator.value()); |
| if (it2 != initiator_specific_factories_.end()) |
| return it2->second.get(); |
| } |
| |
| // AppCache factory must be used if it's given. |
| if (appcache_factory_) |
| return appcache_factory_.get(); |
| |
| return default_factory_.get(); |
| } |
| |
| void URLLoaderFactoryBundle::CreateLoaderAndStart( |
| network::mojom::URLLoaderRequest loader, |
| int32_t routing_id, |
| int32_t request_id, |
| uint32_t options, |
| const network::ResourceRequest& request, |
| network::mojom::URLLoaderClientPtr client, |
| const net::MutableNetworkTrafficAnnotationTag& traffic_annotation) { |
| network::mojom::URLLoaderFactory* factory_ptr = GetFactory(request); |
| factory_ptr->CreateLoaderAndStart(std::move(loader), routing_id, request_id, |
| options, request, std::move(client), |
| traffic_annotation); |
| } |
| |
| void URLLoaderFactoryBundle::Clone( |
| network::mojom::URLLoaderFactoryRequest request) { |
| NOTREACHED(); |
| } |
| |
| std::unique_ptr<network::SharedURLLoaderFactoryInfo> |
| URLLoaderFactoryBundle::Clone() { |
| network::mojom::URLLoaderFactoryPtrInfo default_factory_info; |
| if (default_factory_) |
| default_factory_->Clone(mojo::MakeRequest(&default_factory_info)); |
| |
| auto info = std::make_unique<URLLoaderFactoryBundleInfo>( |
| std::move(default_factory_info), |
| ClonePtrMapToPtrInfoMap(scheme_specific_factories_), |
| ClonePtrMapToPtrInfoMap(initiator_specific_factories_), |
| bypass_redirect_checks_); |
| |
| if (appcache_factory_) |
| appcache_factory_->Clone(mojo::MakeRequest(&info->appcache_factory_info())); |
| |
| return info; |
| } |
| |
| bool URLLoaderFactoryBundle::BypassRedirectChecks() const { |
| return bypass_redirect_checks_; |
| } |
| |
| void URLLoaderFactoryBundle::Update( |
| std::unique_ptr<URLLoaderFactoryBundleInfo> info) { |
| if (info->default_factory_info()) |
| default_factory_.Bind(std::move(info->default_factory_info())); |
| if (info->appcache_factory_info()) |
| appcache_factory_.Bind(std::move(info->appcache_factory_info())); |
| BindPtrInfoMapToPtrMap(&scheme_specific_factories_, |
| std::move(info->scheme_specific_factory_infos())); |
| BindPtrInfoMapToPtrMap(&initiator_specific_factories_, |
| std::move(info->initiator_specific_factory_infos())); |
| bypass_redirect_checks_ = info->bypass_redirect_checks(); |
| } |
| |
| } // namespace content |