summaryrefslogtreecommitdiffstats
path: root/content/browser/loader/render_view_host_tracker.cc
blob: 2f006e8e990aa47df9f45adaf42be82a61aea791 (plain)
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
// Copyright (c) 2013 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.

#include "content/browser/loader/render_view_host_tracker.h"

#include "base/bind_helpers.h"
#include "base/stl_util.h"
#include "content/browser/loader/resource_dispatcher_host_impl.h"
#include "content/browser/loader/resource_scheduler.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_view_host.h"

namespace content {

RenderViewHostTracker::RenderViewHostTracker()
    : rvh_created_callback_(
          base::Bind(&RenderViewHostTracker::RenderViewHostCreated,
                     base::Unretained(this))) {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  RenderViewHost::AddCreatedCallback(rvh_created_callback_);
}

RenderViewHostTracker::~RenderViewHostTracker() {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  DCHECK(observers_.empty());
  RenderViewHost::RemoveCreatedCallback(rvh_created_callback_);
}

void RenderViewHostTracker::RenderViewHostCreated(RenderViewHost* rvh) {
  Observer* observer = new Observer(rvh, this);
  observers_.insert(observer);

  int child_id = rvh->GetProcess()->GetID();
  int route_id = rvh->GetRoutingID();
  BrowserThread::PostTask(
      BrowserThread::IO, FROM_HERE,
      base::Bind(&ResourceDispatcherHostImpl::OnRenderViewHostCreated,
                 base::Unretained(ResourceDispatcherHostImpl::Get()),
                 child_id, route_id));
}

void RenderViewHostTracker::RemoveObserver(Observer* observer) {
  DCHECK(ContainsKey(observers_, observer));
  observers_.erase(observer);
  delete observer;
}

RenderViewHostTracker::Observer::Observer(RenderViewHost* rvh,
                                          RenderViewHostTracker* tracker)
    : RenderViewHostObserver(rvh),
      tracker_(tracker) {
}

RenderViewHostTracker::Observer::~Observer() {
}

void RenderViewHostTracker::Observer::RenderViewHostDestroyed(
    RenderViewHost* rvh) {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  int child_id = rvh->GetProcess()->GetID();
  int route_id = rvh->GetRoutingID();

  BrowserThread::PostTask(
      BrowserThread::IO, FROM_HERE,
      base::Bind(&ResourceDispatcherHostImpl::OnRenderViewHostDeleted,
                 base::Unretained(ResourceDispatcherHostImpl::Get()),
                 child_id, route_id));

  tracker_->RemoveObserver(this);
}

}  // namespace content