OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "content/browser/renderer_host/frame_tree.h" | |
6 | |
7 #include <queue> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/callback.h" | |
11 #include "content/browser/renderer_host/frame_tree_node.h" | |
12 #include "content/browser/renderer_host/render_frame_host_impl.h" | |
13 | |
14 namespace content { | |
15 | |
16 namespace { | |
17 // Used with FrameTree::ForEach() to search for the FrameTreeNode | |
18 // corresponding to |frame_id|. | |
19 bool FrameTreeNodeForId(int64 frame_id, FrameTreeNode** out_node, | |
20 FrameTreeNode* node) { | |
21 if (node->frame_id() == frame_id) { | |
22 *out_node = node; | |
23 // Terminate iteration once the node has been found. | |
24 return false; | |
25 } | |
26 return true; | |
27 } | |
28 | |
29 } // namespace | |
30 | |
31 FrameTree::FrameTree() | |
32 : root_(new FrameTreeNode(FrameTreeNode::kInvalidFrameId, std::string(), | |
33 scoped_ptr<RenderFrameHostImpl>())) { | |
34 } | |
35 | |
36 FrameTree::~FrameTree() { | |
37 } | |
38 | |
39 FrameTreeNode* FrameTree::FindByID(int64 frame_id) { | |
40 FrameTreeNode* node = NULL; | |
41 ForEach(base::Bind(&FrameTreeNodeForId, frame_id, &node)); | |
42 return node; | |
43 } | |
44 | |
45 void FrameTree::ForEach( | |
46 const base::Callback<bool(FrameTreeNode*)>& on_node) const { | |
47 std::queue<FrameTreeNode*> queue; | |
48 queue.push(root_.get()); | |
49 | |
50 while (!queue.empty()) { | |
51 FrameTreeNode* node = queue.front(); | |
52 queue.pop(); | |
53 if (!on_node.Run(node)) | |
54 break; | |
55 | |
56 for (size_t i = 0; i < node->child_count(); ++i) | |
57 queue.push(node->child_at(i)); | |
58 } | |
59 } | |
60 | |
61 bool FrameTree::IsFirstNavigationAfterSwap() const { | |
62 return root_->frame_id() == FrameTreeNode::kInvalidFrameId; | |
63 } | |
64 | |
65 void FrameTree::OnFirstNavigationAfterSwap(int main_frame_id) { | |
66 root_->set_frame_id(main_frame_id); | |
67 } | |
68 | |
69 void FrameTree::AddFrame(int render_frame_host_id, int64 parent_frame_id, | |
70 int64 frame_id, const std::string& frame_name) { | |
71 FrameTreeNode* parent = FindByID(parent_frame_id); | |
72 // TODO(ajwong): Should the renderer be killed here? Would there be a race on | |
73 // shutdown that might make this case possible? | |
74 if (!parent) | |
75 return; | |
76 | |
77 parent->AddChild(CreateNode(frame_id, frame_name, render_frame_host_id, | |
78 parent->render_frame_host()->GetProcess())); | |
79 } | |
80 | |
81 void FrameTree::RemoveFrame(int64 parent_frame_id, int64 frame_id) { | |
82 // If switches::kSitePerProcess is not specified, then the FrameTree only | |
83 // contains a node for the root element. However, even in this case | |
84 // frame detachments need to be broadcast outwards. | |
85 // | |
86 // TODO(ajwong): Move this below the |parent| check after the FrameTree is | |
87 // guaranteed to be correctly populated even without the | |
88 // switches::kSitePerProcess flag. | |
89 FrameTreeNode* parent = FindByID(parent_frame_id); | |
90 if (!on_frame_removed_.is_null()) { | |
91 on_frame_removed_.Run( | |
92 root_->render_frame_host()->render_view_host(), frame_id); | |
93 } | |
94 | |
95 // TODO(ajwong): Should the renderer be killed here? Would there be a race on | |
96 // shutdown that might make this case possible? | |
97 if (!parent) | |
98 return; | |
99 | |
100 parent->RemoveChild(frame_id); | |
101 } | |
102 | |
103 void FrameTree::SetFrameUrl(int64 frame_id, const GURL& url) { | |
104 FrameTreeNode* node = FindByID(frame_id); | |
105 // TODO(ajwong): Should the renderer be killed here? Would there be a race on | |
106 // shutdown that might make this case possible? | |
107 if (!node) | |
108 return; | |
109 | |
110 if (node) | |
111 node->set_current_url(url); | |
112 } | |
113 | |
114 void FrameTree::SwapMainFrame(RenderFrameHostImpl* render_frame_host) { | |
115 return root_->ResetForMainFrame(render_frame_host); | |
116 } | |
117 | |
118 RenderFrameHostImpl* FrameTree::GetMainFrame() const { | |
119 return root_->render_frame_host(); | |
120 } | |
121 | |
122 void FrameTree::SetFrameRemoveListener( | |
123 const base::Callback<void(RenderViewHostImpl*, int64)>& on_frame_removed) { | |
124 on_frame_removed_ = on_frame_removed; | |
125 } | |
126 | |
127 scoped_ptr<FrameTreeNode> FrameTree::CreateNode( | |
128 int64 frame_id, const std::string& frame_name, int render_frame_host_id, | |
129 RenderProcessHost* render_process_host) { | |
130 scoped_ptr<RenderFrameHostImpl> render_frame_host( | |
131 new RenderFrameHostImpl(root_->render_frame_host()->render_view_host(), | |
132 this, render_frame_host_id, false)); | |
133 | |
134 return make_scoped_ptr(new FrameTreeNode(frame_id, frame_name, | |
135 render_frame_host.Pass())); | |
136 } | |
137 | |
138 } // namespace content | |
OLD | NEW |