File: web_contents_task_provider.cc

package info (click to toggle)
chromium 138.0.7204.183-1~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-proposed-updates
  • size: 6,080,960 kB
  • sloc: cpp: 34,937,079; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,954; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,811; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (640 lines) | stat: -rw-r--r-- 24,868 bytes parent folder | download | duplicates (5)
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/task_manager/providers/web_contents/web_contents_task_provider.h"

#include <memory>

#include "base/check.h"
#include "base/containers/flat_map.h"
#include "base/containers/flat_set.h"
#include "base/functional/bind.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/notreached.h"
#include "chrome/browser/task_manager/providers/web_contents/back_forward_cache_task.h"
#include "chrome/browser/task_manager/providers/web_contents/fenced_frame_task.h"
#include "chrome/browser/task_manager/providers/web_contents/guest_task_mparch.h"
#include "chrome/browser/task_manager/providers/web_contents/prerender_task.h"
#include "chrome/browser/task_manager/providers/web_contents/subframe_task.h"
#include "chrome/browser/task_manager/providers/web_contents/web_contents_tags_manager.h"
#if !BUILDFLAG(IS_ANDROID)
#include "components/guest_view/browser/guest_view_base.h"
#endif  // !BUILDFLAG(IS_ANDROID)
#include "content/public/browser/navigation_entry.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/public/common/content_features.h"

using content::RenderFrameHost;
using content::RenderProcessHost;
using content::SiteInstance;
using content::WebContents;

namespace task_manager {

namespace {

bool IsMPArchGuestMainFrame(RenderFrameHost* render_frame_host) {
  if (!base::FeatureList::IsEnabled(features::kGuestViewMPArch)) {
    return false;
  }
#if BUILDFLAG(IS_ANDROID)
  // Guest view is not enabled on Android.
  return false;
#else   // BUILDFLAG(IS_ANDROID)
  auto* guest =
      guest_view::GuestViewBase::FromRenderFrameHost(render_frame_host);
  return guest && guest->GetGuestMainFrame() == render_frame_host;
#endif  // BUILDFLAG(IS_ANDROID)
}

}  // namespace

// Defines an entry for each WebContents that will be tracked by the provider.
// The entry is used to observe certain events in its corresponding WebContents
// and then it notifies the provider or the render task (representing the
// WebContents) of these events.
// The entry owns the created tasks representing the WebContents, and it is
// itself owned by the provider.
class WebContentsTaskProvider::WebContentsEntry
    : public content::WebContentsObserver {
 public:
  WebContentsEntry(content::WebContents* web_contents,
                   WebContentsTaskProvider* provider);
  ~WebContentsEntry() override;
  WebContentsEntry(const WebContentsEntry&) = delete;
  WebContentsEntry& operator=(const WebContentsEntry&) = delete;

  // Creates all the tasks associated with each |RenderFrameHost| in this
  // entry's WebContents.
  void CreateAllTasks();

  // Clears all the tasks in this entry. The provider's observer will be
  // notified if |notify_observer| is true.
  void ClearAllTasks(bool notify_observer);

  // Returns the |RendererTask| that corresponds to the given
  // |render_frame_host| or |nullptr| if the given frame is not tracked by this
  // entry.
  base::WeakPtr<RendererTask> GetTaskForFrame(
      RenderFrameHost* render_frame_host) const;

  // content::WebContentsObserver:
  void RenderFrameDeleted(RenderFrameHost* render_frame_host) override;
  void RenderFrameHostChanged(RenderFrameHost* old_host,
                              RenderFrameHost* new_host) override;
  void RenderFrameHostStateChanged(
      RenderFrameHost* render_frame_host,
      RenderFrameHost::LifecycleState old_state,
      RenderFrameHost::LifecycleState new_state) override;
  void WebContentsDestroyed() override;
  void OnRendererUnresponsive(RenderProcessHost* render_process_host) override;
  void DidFinishNavigation(
      content::NavigationHandle* navigation_handle) override;
  void TitleWasSetForMainFrame(
      content::RenderFrameHost* render_frame_host) override;

  void RenderFrameReady(int process_routing_id, int frame_routing_id);

 private:
  // Defines a callback for WebContents::ForEachRenderFrameHost() to create a
  // corresponding task for the given |render_frame_host| and notifying the
  // provider's observer of the new task.
  void CreateTaskForFrame(RenderFrameHost* render_frame_host);

  // Clears the task that corresponds to the given |render_frame_host| and
  // notifies the provider's observer of the tasks removal.
  void ClearTaskForFrame(RenderFrameHost* render_frame_host);

  // Same as |ClearTaskForFrame|, but for every descendant of
  // |ancestor|.
  void ClearTasksForDescendantsOf(RenderFrameHost* ancestor);

  // Walks parents until hitting a process boundary. Returns the highest frame
  // in the same SiteInstance as |render_frame_host|.
  RenderFrameHost* FindLocalRoot(RenderFrameHost* render_frame_host) const;

  // The provider that owns this entry.
  raw_ptr<WebContentsTaskProvider> provider_;

  // For each SiteInstance, the frames within it that are tracked (local roots
  // only as per FindLocalRoot()), and its RendererTask. The number of tracked
  // items is small, thus flat_map and flat_set.
  struct SiteInstanceInfo {
    base::flat_set<raw_ptr<RenderFrameHost, CtnExperimental>> frames;
    std::unique_ptr<RendererTask> renderer_task;
  };
  base::flat_map<SiteInstance*, SiteInstanceInfo> site_instance_infos_;

  // States whether we did record a main frame for this entry.
  raw_ptr<SiteInstance> primary_main_frame_site_instance_ = nullptr;

  base::WeakPtrFactory<WebContentsEntry> weak_factory_{this};
};

////////////////////////////////////////////////////////////////////////////////

WebContentsTaskProvider::WebContentsEntry::WebContentsEntry(
    content::WebContents* web_contents,
    WebContentsTaskProvider* provider)
    : WebContentsObserver(web_contents), provider_(provider) {}

WebContentsTaskProvider::WebContentsEntry::~WebContentsEntry() {
  ClearAllTasks(false);
}

void WebContentsTaskProvider::WebContentsEntry::CreateAllTasks() {
  DCHECK(web_contents()->GetPrimaryMainFrame());
  web_contents()->ForEachRenderFrameHost(
      [this](content::RenderFrameHost* render_frame_host) {
        const auto state = render_frame_host->GetLifecycleState();
        // `WebContents::ForEachRenderFrameHost` does not iterate over
        // speculative or pending commit RFHs.
        //
        // TODO(crbug.com/40262518): Move this CHECK into
        // `WebContents::ForEachRenderFrameHost`.
        CHECK_NE(state, RenderFrameHost::LifecycleState::kPendingCommit);
        // TODO(crbug.com/40262518):
        // `WebContents::ForEachRenderFrameHost` should explicitly exclude
        // `kPendingDeletion`, just like `kSpeculative` and `kPendingCommit`.
        if (state == RenderFrameHost::LifecycleState::kPendingDeletion) {
          // A `kPendingDeletion` RFH will soon be destroyed. The task manager
          // does not need to create a task for such a RFH.
          return;
        }
        CreateTaskForFrame(render_frame_host);
      });
}

void WebContentsTaskProvider::WebContentsEntry::ClearAllTasks(
    bool notify_observer) {
  for (auto& it : site_instance_infos_) {
    RendererTask* task = it.second.renderer_task.get();
    task->set_termination_status(web_contents()->GetCrashedStatus());
    task->set_termination_error_code(web_contents()->GetCrashedErrorCode());

    if (notify_observer)
      provider_->NotifyObserverTaskRemoved(task);
  }

  site_instance_infos_.clear();
  primary_main_frame_site_instance_ = nullptr;
}

base::WeakPtr<RendererTask>
WebContentsTaskProvider::WebContentsEntry::GetTaskForFrame(
    RenderFrameHost* render_frame_host) const {
  SiteInstance* site_instance = render_frame_host->GetSiteInstance();
  auto itr = site_instance_infos_.find(site_instance);
  if (itr == site_instance_infos_.end())
    return nullptr;

  // Only local roots are in the frame list.
  if (!itr->second.frames.count(FindLocalRoot(render_frame_host)))
    return nullptr;

  return itr->second.renderer_task->AsWeakPtr();
}

RenderFrameHost* WebContentsTaskProvider::WebContentsEntry::FindLocalRoot(
    RenderFrameHost* render_frame_host) const {
  RenderFrameHost* candidate = render_frame_host;
  while (RenderFrameHost* parent = candidate->GetParent()) {
    if (parent->GetProcess() != candidate->GetProcess()) {
      break;
    }
    candidate = parent;
  }
  return candidate;
}

void WebContentsTaskProvider::WebContentsEntry::RenderFrameDeleted(
    RenderFrameHost* render_frame_host) {
  ClearTaskForFrame(render_frame_host);
}

void WebContentsTaskProvider::WebContentsEntry::RenderFrameHostChanged(
    RenderFrameHost* old_host,
    RenderFrameHost* new_host) {
  // The navigating frame and its subframes are now pending deletion. Stop
  // tracking them immediately rather than when they are destroyed. The order of
  // deletion is important. The children must be removed first.
  if (old_host) {
    ClearTasksForDescendantsOf(old_host);
    ClearTaskForFrame(old_host);
  }
  // Tasks creation for |new_host| is delayed to |DidFinishNavigation|.
}

// Handles creation and deletion of BFCache tasks for pages entering and leaving
// the BFCache, and deletion of prerender tasks after prerendering activation.
void WebContentsTaskProvider::WebContentsEntry::RenderFrameHostStateChanged(
    RenderFrameHost* render_frame_host,
    RenderFrameHost::LifecycleState old_state,
    RenderFrameHost::LifecycleState new_state) {
  if (old_state == RenderFrameHost::LifecycleState::kInBackForwardCache)
    ClearTaskForFrame(render_frame_host);

  if (new_state == RenderFrameHost::LifecycleState::kInBackForwardCache)
    CreateTaskForFrame(render_frame_host);

  // |RenderFrameDeleted| is not fired for prerender page activation so we need
  // to delete prerender task. |RenderFrameHostStateChanged| is the earliest
  // event of the prerendering activation flow.
  if (old_state == RenderFrameHost::LifecycleState::kPrerendering &&
      new_state == RenderFrameHost::LifecycleState::kActive) {
    ClearTaskForFrame(render_frame_host);
  }
}

void WebContentsTaskProvider::WebContentsEntry::RenderFrameReady(
    int render_process_id,
    int render_frame_id) {
  // We get here when a RenderProcessHost we are tracking transitions to the
  // IsReady state. This might mean we know its process ID.
  content::RenderFrameHost* render_frame_host =
      content::RenderFrameHost::FromID(render_process_id, render_frame_id);
  if (!render_frame_host)
    return;

  base::WeakPtr<Task> task = GetTaskForFrame(render_frame_host);
  if (!task) {
    return;
  }

  const base::ProcessId determine_pid_from_handle = base::kNullProcessId;
  provider_->UpdateTaskProcessInfoAndNotifyObserver(
      task.get(), render_frame_host->GetProcess()->GetProcess().Handle(),
      determine_pid_from_handle);
}

void WebContentsTaskProvider::WebContentsEntry::WebContentsDestroyed() {
  ClearAllTasks(true);
  provider_->DeleteEntry(web_contents());
}

void WebContentsTaskProvider::WebContentsEntry::OnRendererUnresponsive(
    RenderProcessHost* render_process_host) {
  for (const auto& pair : site_instance_infos_) {
    if (pair.first->GetProcess() == render_process_host) {
      provider_->NotifyObserverTaskUnresponsive(
          pair.second.renderer_task.get());
      return;
    }
  }
}

void WebContentsTaskProvider::WebContentsEntry::DidFinishNavigation(
    content::NavigationHandle* navigation_handle) {
  // No task creation/update for downloads and HTTP204/205.
  if (!navigation_handle->HasCommitted()) {
    return;
  }

  // Creation.
  //
  // Create a task if we have not encountered this site instance before, or it
  // is a new RenderFrameHost for this site instance. For same-document
  // navigation, since neither |RenderFrameDeleted| nor |RenderFrameHostChanged|
  // is fired to delete the existing task, we do not recreate them.
  //
  // TODO(crbug.com/40171294): DidFinishNavigation is not called when we
  // create initial empty documents, and as a result, we will not create new
  // tasks for these empty documents if they are in a different process from
  // their embedder/opener (eg: an empty fenced frame or a blank tab created by
  // window.open('', '_blank', 'noopener')). Ideally, we would call
  // CreateTaskForFrame inside RenderFrameCreated instead (which is called for
  // initial documents), but CreateTaskForFrame uses RFH::GetLifecycleState,
  // which cannot currently be called inside RenderFrameCreated (due to a DCHECK
  // which doesn't allow the method to be called when the state is
  // 'kSpeculative').
  {
    auto* rfh = navigation_handle->GetRenderFrameHost();
    auto* site_instance = rfh->GetSiteInstance();
    auto it = site_instance_infos_.find(site_instance);
    if (!navigation_handle->IsSameDocument() &&
        (it == site_instance_infos_.end() ||
         it->second.frames.find(rfh) == it->second.frames.end())) {
      CreateTaskForFrame(rfh);
    }
  }

  // Update.
  //
  // We only need to update tasks for primary main frame navigations except for
  // MPArch guest view.
  //
  // FencedFrame task gets the title from |SiteInstance::GetSiteURL()| which
  // does not change for the same site instance, thus no need to update;
  // prerender does not support multiple navigations thus no need to update its
  // title.
  if (!navigation_handle->IsInPrimaryMainFrame() &&
      !IsMPArchGuestMainFrame(navigation_handle->GetRenderFrameHost())) {
    return;
  }

  base::WeakPtr<RendererTask> main_frame_task =
      GetTaskForFrame(web_contents()->GetPrimaryMainFrame());
  if (!main_frame_task) {
    return;
  }

  for (auto& it : site_instance_infos_) {
    base::WeakPtr<RendererTask> task = it.second.renderer_task->AsWeakPtr();

    // Listening to WebContentsObserver::TitleWasSetForMainFrame() only is
    // not enough in some cases when the the web page doesn't have a title.
    // That's why we update the title here as well.
    task->UpdateTitle();

    // Call RendererTask::UpdateFavicon() to set the current favicon to the
    // default favicon. If the page has a non-default favicon,
    // RendererTask::OnFaviconUpdated() will update the current favicon once
    // FaviconDriver figures out the correct favicon for the page.
    task->UpdateFavicon();
  }
}

void WebContentsTaskProvider::WebContentsEntry::TitleWasSetForMainFrame(
    content::RenderFrameHost* render_frame_host) {
  content::SiteInstance* site_instance = render_frame_host->GetSiteInstance();
  if (auto it = site_instance_infos_.find(site_instance);
      it != site_instance_infos_.end()) {
    it->second.renderer_task->UpdateTitle();
    it->second.renderer_task->UpdateFavicon();
  }
}

void WebContentsTaskProvider::WebContentsEntry::CreateTaskForFrame(
    RenderFrameHost* render_frame_host) {
  DCHECK(render_frame_host);
  // Currently we do not track speculative RenderFrameHosts or RenderFrameHosts
  // which are pending deletion.
  const auto rfh_state = render_frame_host->GetLifecycleState();
  switch (rfh_state) {
    case RenderFrameHost::LifecycleState::kPrerendering:
    case RenderFrameHost::LifecycleState::kInBackForwardCache:
    case RenderFrameHost::LifecycleState::kActive:
      break;
    default:
      NOTREACHED() << "Illegal RFH state for TaskManager: "
                   << static_cast<int>(rfh_state);
  }

  // Exclude sad tabs, sad OOPIFs.
  if (!render_frame_host->IsRenderFrameLive()) {
    return;
  }

  // Another instance of this class will be created for inner WebContents. If we
  // iterate into an inner WebContents that is not associated with `this`, skip
  // it, so we don't create duplicated tasks. Task creation for RenderFrameHosts
  // not associated with a WebContents should be handled by a different type of
  // TaskProvider.
  if (content::WebContents::FromRenderFrameHost(render_frame_host) !=
      web_contents()) {
    return;
  }

  // Exclude frames in the same SiteInstance or same process as their parent;
  // |site_instance_infos_| only contains local roots.
  content::SiteInstance* site_instance = render_frame_host->GetSiteInstance();
  auto* parent = render_frame_host->GetParentOrOuterDocument();
  if (parent && (site_instance == parent->GetSiteInstance() ||
                 site_instance->GetProcess() ==
                     parent->GetSiteInstance()->GetProcess())) {
    return;
  }

  bool site_instance_exists = site_instance_infos_.count(site_instance) != 0;
  bool is_primary_main_frame = render_frame_host->IsInPrimaryMainFrame();
  bool site_instance_is_main =
      (site_instance == primary_main_frame_site_instance_);

  std::unique_ptr<RendererTask> new_task;

  // We need to create a task if one doesn't already exist for this
  // SiteInstance, or if the main frame navigates to a process that currently is
  // represented by a SubframeTask.
  if (!site_instance_exists ||
      (is_primary_main_frame && !site_instance_is_main)) {
    base::WeakPtr<RendererTask> primary_main_frame_task =
        GetTaskForFrame(web_contents()->GetPrimaryMainFrame());

    if (rfh_state == RenderFrameHost::LifecycleState::kInBackForwardCache) {
      // Use RFH::GetMainFrame instead web_contents()->GetPrimaryMainFrame()
      // because the BFCached frames are not the currently active main frame.
      RenderFrameHost* main_frame = render_frame_host->GetMainFrame();
      new_task = std::make_unique<BackForwardCacheTask>(
          render_frame_host, GetTaskForFrame(main_frame), provider_);
    } else if (rfh_state == RenderFrameHost::LifecycleState::kPrerendering) {
      new_task = std::make_unique<PrerenderTask>(render_frame_host, provider_);
    } else if (is_primary_main_frame) {
      const WebContentsTag* tag =
          WebContentsTag::FromWebContents(web_contents());
      new_task = tag->CreateTask(provider_);
      primary_main_frame_site_instance_ = site_instance;
    } else if (render_frame_host->IsFencedFrameRoot()) {
      new_task = std::make_unique<FencedFrameTask>(
          render_frame_host, std::move(primary_main_frame_task));
    } else if (IsMPArchGuestMainFrame(render_frame_host)) {
      new_task = std::make_unique<GuestTaskMPArch>(
          render_frame_host, std::move(primary_main_frame_task));
    } else {
      new_task = std::make_unique<SubframeTask>(
          render_frame_host, std::move(primary_main_frame_task));
    }
  }

  auto insert_result =
      site_instance_infos_[site_instance].frames.insert(render_frame_host);
  DCHECK(insert_result.second);

  if (new_task) {
    if (site_instance_exists) {
      provider_->NotifyObserverTaskRemoved(
          site_instance_infos_[site_instance].renderer_task.get());
    }

    RendererTask* new_task_ptr = new_task.get();
    site_instance_infos_[site_instance].renderer_task = std::move(new_task);
    provider_->NotifyObserverTaskAdded(new_task_ptr);

    // If we don't know the OS process handle yet (e.g., because this task is
    // still launching), update the task when it becomes available.
    if (new_task_ptr->process_id() == base::kNullProcessId) {
      render_frame_host->GetProcess()->PostTaskWhenProcessIsReady(
          base::BindOnce(&WebContentsEntry::RenderFrameReady,
                         weak_factory_.GetWeakPtr(),
                         render_frame_host->GetProcess()->GetDeprecatedID(),
                         render_frame_host->GetRoutingID()));
    }
  }
}

void WebContentsTaskProvider::WebContentsEntry::ClearTaskForFrame(
    RenderFrameHost* render_frame_host) {
  DCHECK(render_frame_host);
  content::SiteInstance* site_instance = render_frame_host->GetSiteInstance();
  auto itr = site_instance_infos_.find(site_instance);
  if (itr == site_instance_infos_.end())
    return;

  auto& frames = itr->second.frames;
  size_t count = frames.erase(render_frame_host);
  if (!count)
    return;

  if (frames.empty()) {
    std::unique_ptr<RendererTask> renderer_task =
        std::move(itr->second.renderer_task);
    site_instance_infos_.erase(itr);
    provider_->NotifyObserverTaskRemoved(renderer_task.get());

    if (site_instance == primary_main_frame_site_instance_)
      primary_main_frame_site_instance_ = nullptr;
  }

#if DCHECK_IS_ON()
  // Whenever we have a task, we should have a main frame site instance.
  // However, when a tab is destroyed and there was a BFCached Task or a
  // prerender task, the main task may be cleaned up before the
  // BFCached/prerender Task. BFCache or prerender tasks will be deleted
  // asynchronously after the main frame is deleted.

  bool only_bfcache_or_prerender_rfhs = true;
  for (auto& [ignore, site_instance_info] : site_instance_infos_) {
    for (RenderFrameHost* rfh : site_instance_info.frames) {
      const auto state = rfh->GetLifecycleState();
      if (state != RenderFrameHost::LifecycleState::kInBackForwardCache &&
          state != RenderFrameHost::LifecycleState::kPrerendering) {
        only_bfcache_or_prerender_rfhs = false;
      }
    }
  }
  DCHECK(only_bfcache_or_prerender_rfhs ||
         site_instance_infos_.empty() ==
             (primary_main_frame_site_instance_ == nullptr));
#endif
}

void WebContentsTaskProvider::WebContentsEntry::ClearTasksForDescendantsOf(
    RenderFrameHost* ancestor) {
  ancestor->ForEachRenderFrameHost(
      [this, ancestor](content::RenderFrameHost* render_frame_host) {
        if (render_frame_host == ancestor)
          return;
        ClearTaskForFrame(render_frame_host);
      });
}

////////////////////////////////////////////////////////////////////////////////

WebContentsTaskProvider::WebContentsTaskProvider() = default;

WebContentsTaskProvider::~WebContentsTaskProvider() {
  if (is_updating_) {
    StopUpdating();
  }
}

void WebContentsTaskProvider::OnWebContentsTagCreated(
    const WebContentsTag* tag) {
  DCHECK(tag);
  content::WebContents* web_contents = tag->web_contents();
  DCHECK(web_contents);

  // TODO(afakhry): Check if we need this check. It seems that we no longer
  // need it in the new implementation.
  std::unique_ptr<WebContentsEntry>& entry = entries_map_[web_contents];
  if (entry) {
    // This case may happen if we added a WebContents while collecting all the
    // pre-existing ones at the time |StartUpdating()| was called, but the
    // notification of its connection hasn't been fired yet. In this case we
    // ignore it since we're already tracking it.
    return;
  }

  entry = std::make_unique<WebContentsEntry>(web_contents, this);
  entry->CreateAllTasks();
}

void WebContentsTaskProvider::OnWebContentsTagRemoved(
    const WebContentsTag* tag) {
  DCHECK(tag);
  content::WebContents* web_contents = tag->web_contents();
  DCHECK(web_contents);

  auto itr = entries_map_.find(web_contents);
  CHECK(itr != entries_map_.end());

  // Must manually clear the tasks and notify the observer.
  itr->second->ClearAllTasks(true);
  entries_map_.erase(itr);  // Deletes the WebContentsEntry.
}

Task* WebContentsTaskProvider::GetTaskOfUrlRequest(int child_id, int route_id) {
  content::RenderFrameHost* rfh =
      content::RenderFrameHost::FromID(child_id, route_id);
  return GetTaskOfFrame(rfh).get();
}

bool WebContentsTaskProvider::HasWebContents(
    content::WebContents* web_contents) const {
  return entries_map_.count(web_contents) != 0;
}

base::WeakPtr<Task> WebContentsTaskProvider::GetTaskOfFrame(
    content::RenderFrameHost* rfh) {
  content::WebContents* web_contents =
      content::WebContents::FromRenderFrameHost(rfh);

  if (!web_contents)
    return nullptr;

  auto itr = entries_map_.find(web_contents);
  if (itr == entries_map_.end()) {
    // Can happen if the tab was closed while a network request was being
    // performed.
    return nullptr;
  }

  return itr->second->GetTaskForFrame(rfh);
}

void WebContentsTaskProvider::StartUpdating() {
  is_updating_ = true;

  // 1- Collect all pre-existing WebContents from the WebContentsTagsManager.
  WebContentsTagsManager* tags_manager = WebContentsTagsManager::GetInstance();
  for (const task_manager::WebContentsTag* tag : tags_manager->tracked_tags()) {
    OnWebContentsTagCreated(tag);
  }

  // 2- Start observing newly connected ones.
  tags_manager->SetProvider(this);
}

void WebContentsTaskProvider::StopUpdating() {
  is_updating_ = false;

  // 1- Stop observing.
  WebContentsTagsManager::GetInstance()->ClearProvider();

  // 2- Clear storage.
  entries_map_.clear();
}

void WebContentsTaskProvider::DeleteEntry(content::WebContents* web_contents) {
  // This erase() will delete the WebContentsEntry, which is actually our
  // caller, but it's expecting us to delete it.
  bool success = entries_map_.erase(web_contents) != 0;
  DCHECK(success);
}

}  // namespace task_manager