| // 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. |
| |
| #ifndef CONTENT_BROWSER_LOADER_NAVIGATION_URL_LOADER_IMPL_H_ |
| #define CONTENT_BROWSER_LOADER_NAVIGATION_URL_LOADER_IMPL_H_ |
| |
| #include "base/macros.h" |
| #include "base/memory/weak_ptr.h" |
| #include "base/optional.h" |
| #include "base/time/time.h" |
| #include "content/browser/loader/navigation_url_loader.h" |
| #include "content/common/navigation_params.h" |
| #include "content/public/browser/content_browser_client.h" |
| #include "content/public/browser/ssl_status.h" |
| #include "content/public/common/previews_state.h" |
| #include "services/network/public/mojom/url_loader.mojom.h" |
| #include "services/network/public/mojom/url_loader_factory.mojom.h" |
| |
| namespace net { |
| struct RedirectInfo; |
| } |
| |
| namespace content { |
| |
| class NavigationData; |
| class NavigationLoaderInterceptor; |
| class ResourceContext; |
| class StoragePartition; |
| class StoragePartitionImpl; |
| struct GlobalRequestID; |
| |
| class CONTENT_EXPORT NavigationURLLoaderImpl : public NavigationURLLoader { |
| public: |
| // The caller is responsible for ensuring that |delegate| outlives the loader. |
| // Note |initial_interceptors| is there for test purposes only. |
| NavigationURLLoaderImpl( |
| ResourceContext* resource_context, |
| StoragePartition* storage_partition, |
| std::unique_ptr<NavigationRequestInfo> request_info, |
| std::unique_ptr<NavigationUIData> navigation_ui_data, |
| ServiceWorkerNavigationHandle* service_worker_handle, |
| AppCacheNavigationHandle* appcache_handle, |
| NavigationURLLoaderDelegate* delegate, |
| std::vector<std::unique_ptr<NavigationLoaderInterceptor>> |
| initial_interceptors); |
| ~NavigationURLLoaderImpl() override; |
| |
| // NavigationURLLoader implementation: |
| void FollowRedirect(const std::vector<std::string>& removed_headers, |
| const net::HttpRequestHeaders& modified_headers, |
| PreviewsState new_previews_state) override; |
| void ProceedWithResponse() override; |
| |
| void OnReceiveResponse( |
| scoped_refptr<network::ResourceResponse> response, |
| network::mojom::URLLoaderClientEndpointsPtr url_loader_client_endpoints, |
| std::unique_ptr<NavigationData> navigation_data, |
| const GlobalRequestID& global_request_id, |
| bool is_download, |
| bool is_stream, |
| base::TimeDelta total_ui_to_io_time, |
| base::Time io_post_time); |
| void OnReceiveRedirect(const net::RedirectInfo& redirect_info, |
| scoped_refptr<network::ResourceResponse> response, |
| base::Time io_post_time); |
| void OnComplete(const network::URLLoaderCompletionStatus& status); |
| |
| // Overrides loading of frame requests when the network service is disabled. |
| // If the callback returns true, the frame request was intercepted. Otherwise |
| // it should be loaded normally through ResourceDispatcherHost. Passing an |
| // empty callback will restore the default behavior. |
| // This method must be called either on the IO thread or before threads start. |
| // This callback is run on the IO thread. |
| using BeginNavigationInterceptor = base::RepeatingCallback<bool( |
| network::mojom::URLLoaderRequest* request, |
| int32_t routing_id, |
| int32_t request_id, |
| uint32_t options, |
| const network::ResourceRequest& url_request, |
| network::mojom::URLLoaderClientPtr* client, |
| const net::MutableNetworkTrafficAnnotationTag& traffic_annotation)>; |
| static void SetBeginNavigationInterceptorForTesting( |
| const BeginNavigationInterceptor& interceptor); |
| |
| // Intercepts loading of frame requests when network service is enabled and |
| // either a network::mojom::TrustedURLLoaderHeaderClient is being used or for |
| // schemes not handled by network service (e.g. files). This must be called on |
| // the UI thread or before threads start. |
| using URLLoaderFactoryInterceptor = base::RepeatingCallback<void( |
| network::mojom::URLLoaderFactoryRequest* request)>; |
| static void SetURLLoaderFactoryInterceptorForTesting( |
| const URLLoaderFactoryInterceptor& interceptor); |
| |
| // Creates a URLLoaderFactory for a navigation. The factory uses |
| // |header_client|. This should have the same settings as the factory from the |
| // URLLoaderFactoryGetter. Called on the UI thread. |
| static void CreateURLLoaderFactoryWithHeaderClient( |
| network::mojom::TrustedURLLoaderHeaderClientPtrInfo header_client, |
| network::mojom::URLLoaderFactoryRequest factory_request, |
| StoragePartitionImpl* partition); |
| |
| // Returns a Request ID for browser-initiated navigation requests. Called on |
| // the IO thread. |
| static GlobalRequestID MakeGlobalRequestID(); |
| |
| private: |
| class URLLoaderRequestController; |
| void OnRequestStarted(base::TimeTicks timestamp); |
| |
| void BindNonNetworkURLLoaderFactoryRequest( |
| int frame_tree_node_id, |
| const GURL& url, |
| network::mojom::URLLoaderFactoryRequest factory); |
| |
| NavigationURLLoaderDelegate* delegate_; |
| |
| // Lives on the IO thread. |
| std::unique_ptr<URLLoaderRequestController> request_controller_; |
| |
| NavigationDownloadPolicy download_policy_; |
| |
| // Factories to handle navigation requests for non-network resources. |
| ContentBrowserClient::NonNetworkURLLoaderFactoryMap |
| non_network_url_loader_factories_; |
| |
| // Counts the time overhead of all the hops from the IO to the UI threads. |
| base::TimeDelta io_to_ui_time_; |
| |
| base::WeakPtrFactory<NavigationURLLoaderImpl> weak_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(NavigationURLLoaderImpl); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_LOADER_NAVIGATION_URL_LOADER_IMPL_H_ |