File: no_state_prefetch_helper.cc

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (103 lines) | stat: -rw-r--r-- 3,509 bytes parent folder | download | duplicates (6)
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/no_state_prefetch/renderer/no_state_prefetch_helper.h"

#include "base/metrics/field_trial.h"
#include "base/metrics/histogram_macros.h"
#include "components/no_state_prefetch/common/no_state_prefetch_url_loader_throttle.h"
#include "content/public/renderer/render_frame.h"
#include "content/public/renderer/render_thread.h"
#include "third_party/blink/public/platform/browser_interface_broker_proxy.h"
#include "third_party/blink/public/web/web_local_frame.h"

namespace prerender {

NoStatePrefetchHelper::NoStatePrefetchHelper(
    content::RenderFrame* render_frame,
    const std::string& histogram_prefix)
    : content::RenderFrameObserver(render_frame),
      content::RenderFrameObserverTracker<NoStatePrefetchHelper>(render_frame),
      histogram_prefix_(histogram_prefix),
      start_time_(base::TimeTicks::Now()) {}

NoStatePrefetchHelper::~NoStatePrefetchHelper() = default;

// static
std::unique_ptr<blink::URLLoaderThrottle>
NoStatePrefetchHelper::MaybeCreateThrottle(
    const blink::LocalFrameToken& frame_token) {
  // Currently NoStatePrefetchHelper doesn't work on the background thread.
  if (!content::RenderThread::IsMainThread()) {
    return nullptr;
  }
  blink::WebLocalFrame* web_frame =
      blink::WebLocalFrame::FromFrameToken(frame_token);
  if (!web_frame) {
    return nullptr;
  }
  content::RenderFrame* render_frame =
      content::RenderFrame::FromWebFrame(web_frame);
  auto* helper =
      render_frame
          ? NoStatePrefetchHelper::Get(render_frame->GetMainRenderFrame())
          : nullptr;
  if (!helper) {
    return nullptr;
  }

  mojo::PendingRemote<mojom::NoStatePrefetchCanceler> canceler;
  render_frame->GetBrowserInterfaceBroker().GetInterface(
      canceler.InitWithNewPipeAndPassReceiver());

  auto throttle =
      std::make_unique<NoStatePrefetchURLLoaderThrottle>(std::move(canceler));
  helper->AddThrottle(*throttle);
  return throttle;
}

// static.
bool NoStatePrefetchHelper::IsPrefetching(
    const content::RenderFrame* render_frame) {
  return NoStatePrefetchHelper::Get(render_frame) != nullptr;
}

void NoStatePrefetchHelper::DidDispatchDOMContentLoadedEvent() {
  prefetch_finished_ = true;
  if (prefetch_count_ == 0) {
    SendPrefetchFinished();
  }
}

void NoStatePrefetchHelper::OnDestruct() {
  delete this;
}

void NoStatePrefetchHelper::AddThrottle(
    NoStatePrefetchURLLoaderThrottle& throttle) {
  // Keep track of how many pending throttles we have, as we want to defer
  // sending the "prefetch finished" signal until they are destroyed. This is
  // important since that signal tells the browser that it can tear down this
  // renderer which could interrupt subresource prefetching.
  prefetch_count_++;
  throttle.set_destruction_closure(base::BindOnce(
      &NoStatePrefetchHelper::OnThrottleDestroyed, weak_factory_.GetWeakPtr()));
}

void NoStatePrefetchHelper::OnThrottleDestroyed() {
  if (--prefetch_count_ == 0 && prefetch_finished_) {
    SendPrefetchFinished();
  }
}

void NoStatePrefetchHelper::SendPrefetchFinished() {
  DCHECK(prefetch_count_ == 0 && prefetch_finished_);

  mojo::Remote<mojom::NoStatePrefetchCanceler> canceler;
  render_frame()->GetBrowserInterfaceBroker().GetInterface(
      canceler.BindNewPipeAndPassReceiver());
  canceler->CancelNoStatePrefetchAfterSubresourcesDiscovered();
}

}  // namespace prerender