123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132 |
- // Copyright 2020 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 COMPONENTS_WEB_PACKAGE_WEB_BUNDLE_URL_LOADER_FACTORY_H_
- #define COMPONENTS_WEB_PACKAGE_WEB_BUNDLE_URL_LOADER_FACTORY_H_
- #include <vector>
- #include "base/memory/raw_ptr.h"
- #include "base/memory/weak_ptr.h"
- #include "components/web_package/mojom/web_bundle_parser.mojom.h"
- #include "mojo/public/cpp/bindings/remote.h"
- #include "services/network/public/cpp/corb/corb_api.h"
- #include "services/network/public/cpp/cross_origin_embedder_policy.h"
- #include "services/network/public/cpp/resource_request.h"
- #include "services/network/public/mojom/devtools_observer.mojom.h"
- #include "services/network/public/mojom/network_context.mojom.h"
- #include "services/network/public/mojom/web_bundle_handle.mojom.h"
- #include "third_party/abseil-cpp/absl/types/optional.h"
- namespace web_package {
- class WebBundleMemoryQuotaConsumer;
- class WebBundleURLLoaderFactory {
- public:
- // Used for UMA. Append-only.
- enum class SubresourceWebBundleLoadResult {
- kSuccess = 0,
- kMetadataParseError = 1,
- kMemoryQuotaExceeded = 2,
- kServingConstraintsNotMet = 3,
- kWebBundleFetchFailed = 4,
- kWebBundleRedirected = 5,
- kMaxValue = kWebBundleRedirected,
- };
- WebBundleURLLoaderFactory(
- const GURL& bundle_url,
- const network::ResourceRequest::WebBundleTokenParams&
- web_bundle_token_params,
- mojo::Remote<network::mojom::WebBundleHandle> web_bundle_handle,
- std::unique_ptr<WebBundleMemoryQuotaConsumer>
- web_bundle_memory_quota_consumer,
- mojo::PendingRemote<network::mojom::DevToolsObserver> devtools_observer,
- absl::optional<std::string> devtools_request_id,
- const network::CrossOriginEmbedderPolicy& cross_origin_embedder_policy,
- network::mojom::CrossOriginEmbedderPolicyReporter* coep_reporter);
- ~WebBundleURLLoaderFactory();
- WebBundleURLLoaderFactory(const WebBundleURLLoaderFactory&) = delete;
- WebBundleURLLoaderFactory& operator=(const WebBundleURLLoaderFactory&) =
- delete;
- base::WeakPtr<WebBundleURLLoaderFactory> GetWeakPtr() const;
- void SetBundleStream(mojo::ScopedDataPipeConsumerHandle body);
- void ReportErrorAndCancelPendingLoaders(
- SubresourceWebBundleLoadResult result,
- network::mojom::WebBundleErrorType error,
- const std::string& message);
- mojo::PendingRemote<network::mojom::URLLoaderClient> MaybeWrapURLLoaderClient(
- mojo::PendingRemote<network::mojom::URLLoaderClient> wrapped);
- void StartSubresourceRequest(
- mojo::PendingReceiver<network::mojom::URLLoader> receiver,
- const network::ResourceRequest& url_request,
- mojo::PendingRemote<network::mojom::URLLoaderClient> client,
- mojo::Remote<network::mojom::TrustedHeaderClient> trusted_header_client,
- base::Time request_start_time,
- base::TimeTicks request_start_time_ticks);
- void OnWebBundleFetchFailed();
- private:
- class BundleDataSource;
- class URLLoader;
- bool HasError() const;
- void OnBeforeSendHeadersComplete(
- base::WeakPtr<URLLoader> loader,
- int result,
- const absl::optional<net::HttpRequestHeaders>& headers);
- void QueueOrStartLoader(base::WeakPtr<URLLoader> loader);
- void StartLoad(base::WeakPtr<URLLoader> loader);
- void OnMetadataParsed(web_package::mojom::BundleMetadataPtr metadata,
- web_package::mojom::BundleMetadataParseErrorPtr error);
- bool IsAllowedExchangeUrl(const GURL& relative_url);
- void OnResponseParsed(base::WeakPtr<URLLoader> loader,
- web_package::mojom::BundleResponsePtr response,
- web_package::mojom::BundleResponseParseErrorPtr error);
- void OnHeadersReceivedComplete(
- base::WeakPtr<URLLoader> loader,
- const std::string& original_header,
- uint64_t payload_offset,
- uint64_t payload_length,
- int result,
- const absl::optional<std::string>& headers,
- const absl::optional<GURL>& preserve_fragment_on_redirect_url);
- void SendResponseToLoader(base::WeakPtr<URLLoader> loader,
- const std::string& headers,
- uint64_t payload_offset,
- uint64_t payload_length);
- void OnMemoryQuotaExceeded();
- void OnDataCompleted();
- void MaybeReportLoadResult(SubresourceWebBundleLoadResult result);
- GURL bundle_url_;
- mojo::Remote<network::mojom::WebBundleHandle> web_bundle_handle_;
- const absl::optional<::url::Origin> request_initiator_origin_lock_;
- std::unique_ptr<WebBundleMemoryQuotaConsumer>
- web_bundle_memory_quota_consumer_;
- mojo::Remote<network::mojom::DevToolsObserver> devtools_observer_;
- absl::optional<std::string> devtools_request_id_;
- const network::CrossOriginEmbedderPolicy cross_origin_embedder_policy_;
- raw_ptr<network::mojom::CrossOriginEmbedderPolicyReporter> coep_reporter_;
- std::unique_ptr<BundleDataSource> source_;
- mojo::Remote<web_package::mojom::WebBundleParser> parser_;
- web_package::mojom::BundleMetadataPtr metadata_;
- absl::optional<SubresourceWebBundleLoadResult> load_result_;
- bool data_completed_ = false;
- std::vector<base::WeakPtr<URLLoader>> pending_loaders_;
- network::corb::PerFactoryState corb_state_;
- base::WeakPtrFactory<WebBundleURLLoaderFactory> weak_ptr_factory_{this};
- };
- } // namespace web_package
- #endif // COMPONENTS_WEB_PACKAGE_WEB_BUNDLE_URL_LOADER_FACTORY_H_
|