OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/browser/frame_host/frame_tree.h" | 5 #include "content/browser/frame_host/frame_tree.h" |
6 | 6 |
7 #include <queue> | 7 #include <queue> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
11 #include "base/containers/hash_tables.h" | 11 #include "base/containers/hash_tables.h" |
12 #include "base/lazy_instance.h" | 12 #include "base/lazy_instance.h" |
13 #include "content/browser/frame_host/frame_tree_node.h" | 13 #include "content/browser/frame_host/frame_tree_node.h" |
14 #include "content/browser/frame_host/navigator.h" | 14 #include "content/browser/frame_host/navigator.h" |
15 #include "content/browser/frame_host/render_frame_host_factory.h" | 15 #include "content/browser/frame_host/render_frame_host_factory.h" |
16 #include "content/browser/frame_host/render_frame_host_impl.h" | 16 #include "content/browser/frame_host/render_frame_host_impl.h" |
17 #include "content/browser/frame_host/render_frame_proxy_host.h" | |
18 #include "content/browser/renderer_host/render_view_host_factory.h" | 17 #include "content/browser/renderer_host/render_view_host_factory.h" |
19 #include "content/browser/renderer_host/render_view_host_impl.h" | 18 #include "content/browser/renderer_host/render_view_host_impl.h" |
20 #include "content/public/browser/browser_thread.h" | 19 #include "content/public/browser/browser_thread.h" |
21 | 20 |
22 namespace content { | 21 namespace content { |
23 | 22 |
24 namespace { | 23 namespace { |
25 | 24 |
26 // This is a global map between frame_tree_node_ids and pointer to | 25 // This is a global map between frame_tree_node_ids and pointer to |
27 // FrameTreeNodes. | 26 // FrameTreeNodes. |
28 typedef base::hash_map<int64, FrameTreeNode*> FrameTreeNodeIDMap; | 27 typedef base::hash_map<int64, FrameTreeNode*> FrameTreeNodeIDMap; |
29 | 28 |
30 base::LazyInstance<FrameTreeNodeIDMap> g_frame_tree_node_id_map = | 29 base::LazyInstance<FrameTreeNodeIDMap> g_frame_tree_node_id_map = |
31 LAZY_INSTANCE_INITIALIZER; | 30 LAZY_INSTANCE_INITIALIZER; |
32 | 31 |
33 // Used with FrameTree::ForEach() to search for the FrameTreeNode | 32 // Used with FrameTree::ForEach() to search for the FrameTreeNode |
34 // corresponding to |frame_tree_node_id| whithin a specific FrameTree. | 33 // corresponding to |frame_tree_node_id| whithin a specific FrameTree. |
35 bool FrameTreeNodeForId(int64 frame_tree_node_id, | 34 bool FrameTreeNodeForId(int64 frame_tree_node_id, |
36 FrameTreeNode** out_node, | 35 FrameTreeNode** out_node, |
37 FrameTreeNode* node) { | 36 FrameTreeNode* node) { |
38 if (node->frame_tree_node_id() == frame_tree_node_id) { | 37 if (node->frame_tree_node_id() == frame_tree_node_id) { |
39 *out_node = node; | 38 *out_node = node; |
40 // Terminate iteration once the node has been found. | 39 // Terminate iteration once the node has been found. |
41 return false; | 40 return false; |
| 41 } |
| 42 return true; |
| 43 } |
| 44 |
| 45 bool FrameTreeNodeForRoutingId(int routing_id, |
| 46 int process_id, |
| 47 FrameTreeNode** out_node, |
| 48 FrameTreeNode* node) { |
| 49 // TODO(creis): Look through the swapped out RFHs as well. |
| 50 if (node->current_frame_host()->GetProcess()->GetID() == process_id && |
| 51 node->current_frame_host()->GetRoutingID() == routing_id) { |
| 52 *out_node = node; |
| 53 // Terminate iteration once the node has been found. |
| 54 return false; |
42 } | 55 } |
43 return true; | 56 return true; |
44 } | 57 } |
45 | 58 |
46 // Iterate over the FrameTree to reset any node affected by the loss of the | 59 // Iterate over the FrameTree to reset any node affected by the loss of the |
47 // given RenderViewHost's process. | 60 // given RenderViewHost's process. |
48 bool ResetNodesForNewProcess(RenderViewHost* render_view_host, | 61 bool ResetNodesForNewProcess(RenderViewHost* render_view_host, |
49 FrameTreeNode* node) { | 62 FrameTreeNode* node) { |
50 if (render_view_host == node->current_frame_host()->render_view_host()) | 63 if (render_view_host == node->current_frame_host()->render_view_host()) |
51 node->ResetForNewProcess(); | 64 node->ResetForNewProcess(); |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
100 return it == nodes->end() ? NULL : it->second; | 113 return it == nodes->end() ? NULL : it->second; |
101 } | 114 } |
102 | 115 |
103 FrameTreeNode* FrameTree::FindByID(int64 frame_tree_node_id) { | 116 FrameTreeNode* FrameTree::FindByID(int64 frame_tree_node_id) { |
104 FrameTreeNode* node = NULL; | 117 FrameTreeNode* node = NULL; |
105 ForEach(base::Bind(&FrameTreeNodeForId, frame_tree_node_id, &node)); | 118 ForEach(base::Bind(&FrameTreeNodeForId, frame_tree_node_id, &node)); |
106 return node; | 119 return node; |
107 } | 120 } |
108 | 121 |
109 FrameTreeNode* FrameTree::FindByRoutingID(int routing_id, int process_id) { | 122 FrameTreeNode* FrameTree::FindByRoutingID(int routing_id, int process_id) { |
110 RenderFrameHostImpl* render_frame_host = | 123 FrameTreeNode* node = NULL; |
111 RenderFrameHostImpl::FromID(process_id, routing_id); | 124 ForEach( |
112 if (render_frame_host) { | 125 base::Bind(&FrameTreeNodeForRoutingId, routing_id, process_id, &node)); |
113 FrameTreeNode* result = render_frame_host->frame_tree_node(); | 126 return node; |
114 if (this == result->frame_tree()) | |
115 return result; | |
116 } | |
117 | |
118 RenderFrameProxyHost* render_frame_proxy_host = | |
119 RenderFrameProxyHost::FromID(process_id, routing_id); | |
120 if (render_frame_proxy_host) { | |
121 FrameTreeNode* result = render_frame_proxy_host->frame_tree_node(); | |
122 if (this == result->frame_tree()) | |
123 return result; | |
124 } | |
125 | |
126 return NULL; | |
127 } | 127 } |
128 | 128 |
129 void FrameTree::ForEach( | 129 void FrameTree::ForEach( |
130 const base::Callback<bool(FrameTreeNode*)>& on_node) const { | 130 const base::Callback<bool(FrameTreeNode*)>& on_node) const { |
131 std::queue<FrameTreeNode*> queue; | 131 std::queue<FrameTreeNode*> queue; |
132 queue.push(root_.get()); | 132 queue.push(root_.get()); |
133 | 133 |
134 while (!queue.empty()) { | 134 while (!queue.empty()) { |
135 FrameTreeNode* node = queue.front(); | 135 FrameTreeNode* node = queue.front(); |
136 queue.pop(); | 136 queue.pop(); |
(...skipping 185 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
322 rvh->Shutdown(); | 322 rvh->Shutdown(); |
323 render_view_host_pending_shutdown_map_.erase(multi_iter); | 323 render_view_host_pending_shutdown_map_.erase(multi_iter); |
324 } | 324 } |
325 break; | 325 break; |
326 } | 326 } |
327 CHECK(render_view_host_found); | 327 CHECK(render_view_host_found); |
328 } | 328 } |
329 } | 329 } |
330 | 330 |
331 } // namespace content | 331 } // namespace content |
OLD | NEW |