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


content / browser / preloading / prerender / prerender_url_loader_throttle.cc [blame]

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

#include "content/browser/preloading/prerender/prerender_url_loader_throttle.h"

#include "content/browser/preloading/prerender/prerender_features.h"
#include "content/browser/preloading/prerender/prerender_final_status.h"
#include "content/browser/preloading/prerender/prerender_host.h"
#include "content/browser/preloading/prerender/prerender_host_registry.h"
#include "content/browser/preloading/prerender/prerender_metrics.h"
#include "content/browser/renderer_host/frame_tree_node.h"
#include "content/browser/renderer_host/render_frame_host_delegate.h"

namespace content {

PrerenderURLLoaderThrottle::PrerenderURLLoaderThrottle(
    FrameTreeNodeId frame_tree_node_id)
    : frame_tree_node_id_(frame_tree_node_id) {}

// static
std::unique_ptr<PrerenderURLLoaderThrottle>
PrerenderURLLoaderThrottle::MaybeCreate(FrameTreeNodeId frame_tree_node_id) {
  if (!base::FeatureList::IsEnabled(
          features::kPrerender2FallbackPrefetchSpecRules)) {
    return nullptr;
  }

  auto* frame_tree_node = FrameTreeNode::GloballyFindByID(frame_tree_node_id);
  if (!frame_tree_node) {
    return nullptr;
  }

  PrerenderHostRegistry* prerender_host_registry =
      frame_tree_node->current_frame_host()
          ->delegate()
          ->GetPrerenderHostRegistry();
  if (!prerender_host_registry) {
    return nullptr;
  }

  PrerenderHost* prerender_host =
      prerender_host_registry->FindNonReservedHostById(frame_tree_node_id);
  if (!prerender_host) {
    return nullptr;
  }

  if (!prerender_host->ShouldAbortNavigationBecausePrefetchUnavailable()) {
    return nullptr;
  }

  // If the prefetch ahead of prerender "failed", `PrerenderURLLoaderThrottle`
  // is added to the `ThrottlingURLLoader` for the corresponding prerendering
  // navigation, and the `PrerenderURLLoaderThrottle` always cancels the network
  // request before it starts, which cancels the prerender.
  return std::make_unique<PrerenderURLLoaderThrottle>(frame_tree_node_id);
}

void PrerenderURLLoaderThrottle::WillStartRequest(
    network::ResourceRequest* request,
    bool* defer) {
  delegate_->CancelWithError(net::ERR_ABORTED);

  auto* frame_tree_node = FrameTreeNode::GloballyFindByID(frame_tree_node_id_);
  if (!frame_tree_node) {
    return;
  }

  PrerenderHostRegistry* prerender_host_registry =
      frame_tree_node->current_frame_host()
          ->delegate()
          ->GetPrerenderHostRegistry();
  if (!prerender_host_registry) {
    return;
  }

  prerender_host_registry->CancelHost(
      frame_tree_node_id_,
      PrerenderCancellationReason(
          PrerenderFinalStatus::kPrerenderFailedDuringPrefetch));
}

}  // namespace content