1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85


content / browser / devtools / protocol / devtools_network_resource_loader.h [blame]

// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_NETWORK_RESOURCE_LOADER_H_
#define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_NETWORK_RESOURCE_LOADER_H_

#include <memory>
#include <string_view>

#include "content/common/content_export.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/network/public/cpp/resource_request.h"
#include "services/network/public/cpp/simple_url_loader.h"
#include "services/network/public/cpp/simple_url_loader_stream_consumer.h"
#include "services/network/public/mojom/url_loader.mojom.h"
#include "services/network/public/mojom/url_loader_factory.mojom.h"

namespace content {
namespace protocol {

// The DevToolsNetworkResourceLoader loads a network resource for DevTools
// and passes it to the provided call-back once loading completed. Currently,
// the resource is provided as a string, but in the future this will use
// a DevToolsStreamPipe. This is why we don't just use DownloadToString.

class CONTENT_EXPORT DevToolsNetworkResourceLoader
    : public network::SimpleURLLoaderStreamConsumer {
 public:
  using CompletionCallback =
      base::OnceCallback<void(DevToolsNetworkResourceLoader*,
                              const net::HttpResponseHeaders* rh,
                              bool success,
                              int net_error,
                              std::string content)>;
  enum class Caching { kBypass, kDefault };
  enum class Credentials { kInclude, kSameSite };

  // The |origin| and |site_for_cookies| parameters are supplied by the caller,
  // and we trust the caller that these values are reasonable. They are usually
  // taken from a renderer host / worker host that was identified by the
  // DevTools front-end based on the inspected page.
  static std::unique_ptr<DevToolsNetworkResourceLoader> Create(
      mojo::Remote<network::mojom::URLLoaderFactory> url_loader_factory,
      GURL gurl,
      const url::Origin& origin,
      net::SiteForCookies site_for_cookies,
      Caching caching,
      Credentials include_credentials,
      CompletionCallback complete_callback);

  ~DevToolsNetworkResourceLoader() override;

  // Disallow copy and assignment.
  DevToolsNetworkResourceLoader(const DevToolsNetworkResourceLoader&) = delete;
  DevToolsNetworkResourceLoader& operator=(
      const DevToolsNetworkResourceLoader&) = delete;

 private:
  DevToolsNetworkResourceLoader(
      network::ResourceRequest resource_request,
      const net::NetworkTrafficAnnotationTag& traffic_annotation,
      mojo::Remote<network::mojom::URLLoaderFactory> url_loader_factory,
      CompletionCallback complete_callback);
  void DownloadAsStream();

  void OnDataReceived(std::string_view chunk,
                      base::OnceClosure resume) override;

  void OnComplete(bool success) override;

  void OnRetry(base::OnceClosure start_retry) override;

  const network::ResourceRequest resource_request_;
  const net::NetworkTrafficAnnotationTag traffic_annotation_;
  std::unique_ptr<network::SimpleURLLoader> loader_;
  mojo::Remote<network::mojom::URLLoaderFactory> url_loader_factory_;
  CompletionCallback completion_callback_;
  std::string content_;
};

}  // namespace protocol
}  // namespace content

#endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_NETWORK_RESOURCE_LOADER_H_