| 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 |