Pipe the devTools FrameId from blink into the browser for headless

headless/lib/browser/headless_browser_context_impl.cc

Index: headless/lib/browser/headless_browser_context_impl.cc
diff --git a/headless/lib/browser/headless_browser_context_impl.cc b/headless/lib/browser/headless_browser_context_impl.cc
index b691760e4a4206b8a7f344cdd6f24ca768237cfd..8bbfd77709cd1948f4d42f6f365a03fc295b962b 100644
--- a/headless/lib/browser/headless_browser_context_impl.cc
+++ b/headless/lib/browser/headless_browser_context_impl.cc
@@ -144,6 +144,31 @@ HeadlessBrowserContextImpl::GetAllWebContents() {
   return result;
 }
 
+void HeadlessBrowserContextImpl::SetFrameTreeNodeId(int render_process_id,
+                                                    int render_frame_id,
+                                                    int frame_tree_node_id) {
+  base::AutoLock lock(frame_tree_node_map_lock_);
+  frame_tree_node_map_[std::make_pair(render_process_id, render_frame_id)] =
+      frame_tree_node_id;
+}
+
+void HeadlessBrowserContextImpl::RemoveFrameTreeNode(int render_process_id,
+                                                     int render_frame_id) {
+  base::AutoLock lock(frame_tree_node_map_lock_);
+  frame_tree_node_map_.erase(
+      std::make_pair(render_process_id, render_frame_id));
+}
+
+int HeadlessBrowserContextImpl::GetFrameTreeNodeId(int render_process_id,
+                                                   int render_frame_id) const {
+  base::AutoLock lock(frame_tree_node_map_lock_);
+  const auto& find_it = frame_tree_node_map_.find(
+      std::make_pair(render_process_id, render_frame_id));
+  if (find_it == frame_tree_node_map_.end())
+    return -1;
+  return find_it->second;
+}
+
 void HeadlessBrowserContextImpl::Close() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   browser_->DestroyBrowserContext(this);