[DevTools] Support multiple clients in DevToolsAgentHost

content/browser/devtools/render_frame_devtools_agent_host.cc

 // Copyright (c) 2012 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/devtools/render_frame_devtools_agent_host.h"
 
 #include <tuple>
 #include <utility>
 
 #include "base/containers/flat_map.h"
@@ skipping 164 matching lines @@
       host_->GetRoutingID(), agent_->GetId(), session->session_id()));
   GrantPolicy();
   SessionInfo& info = session_infos_[session->session_id()];
   info.chunk_processor.reset(new DevToolsMessageChunkProcessor(base::Bind(
       &RenderFrameDevToolsAgentHost::FrameHostHolder::SendChunkedMessage,
       base::Unretained(this))));
 }
 
 void RenderFrameDevToolsAgentHost::FrameHostHolder::Reattach(
     FrameHostHolder* old) {
-  int session_id = agent_->session()->session_id();
-  SessionInfo& info = session_infos_[session_id];
-  info.chunk_processor.reset(new DevToolsMessageChunkProcessor(base::Bind(
-      &RenderFrameDevToolsAgentHost::FrameHostHolder::SendChunkedMessage,
-      base::Unretained(this))));
-  SessionInfo* old_info = nullptr;
-  if (old) {
-    auto it = old->session_infos_.find(session_id);
-    old_info = it != old->session_infos_.end() ? &(it->second) : nullptr;
-  }
-  if (old_info)
-    info.chunk_processor->set_state_cookie(
-        old_info->chunk_processor->state_cookie());
-  host_->Send(new DevToolsAgentMsg_Reattach(
-      host_->GetRoutingID(), agent_->GetId(), session_id,
-      info.chunk_processor->state_cookie()));
-  if (old_info) {
-    if (IsBrowserSideNavigationEnabled()) {
-      for (const auto& pair :
-           old_info->sent_messages_whose_reply_came_while_suspended) {
+  for (auto& session_it : agent_->sessions()) {
+    int session_id = session_it.first;
+    SessionInfo& info = session_infos_[session_id];
+    info.chunk_processor.reset(new DevToolsMessageChunkProcessor(base::Bind(
+        &RenderFrameDevToolsAgentHost::FrameHostHolder::SendChunkedMessage,
+        base::Unretained(this))));
+    SessionInfo* old_info = nullptr;
+    if (old) {
+      auto it = old->session_infos_.find(session_id);
+      old_info = it != old->session_infos_.end() ? &(it->second) : nullptr;
+    }
+    if (old_info) {
+      info.chunk_processor->set_state_cookie(
+          old_info->chunk_processor->state_cookie());
+    }
+    host_->Send(new DevToolsAgentMsg_Reattach(
+        host_->GetRoutingID(), agent_->GetId(), session_id,
+        info.chunk_processor->state_cookie()));
+    if (old_info) {
+      if (IsBrowserSideNavigationEnabled()) {
+        for (const auto& pair :
+             old_info->sent_messages_whose_reply_came_while_suspended) {
+          DispatchProtocolMessage(session_id, pair.first, pair.second.method,
+                                  pair.second.message);
+        }
+      }
+      for (const auto& pair : old_info->sent_messages) {
         DispatchProtocolMessage(session_id, pair.first, pair.second.method,
                                 pair.second.message);
       }
     }
-    for (const auto& pair : old_info->sent_messages) {
-      DispatchProtocolMessage(session_id, pair.first, pair.second.method,
-                              pair.second.message);
-    }
   }
   GrantPolicy();
 }
 
 void RenderFrameDevToolsAgentHost::FrameHostHolder::Detach(int session_id) {
   host_->Send(new DevToolsAgentMsg_Detach(host_->GetRoutingID(), session_id));
   RevokePolicy();
   session_infos_.erase(session_id);
 }
 
@@ skipping 64 matching lines @@
   if (it == session_infos_.end())
     return;
   SessionInfo& info = it->second;
   int id = info.chunk_processor->last_call_id();
   MethodAndMessage sent_message = info.sent_messages[id];
   info.sent_messages.erase(id);
   if (suspended_) {
     info.sent_messages_whose_reply_came_while_suspended[id] = sent_message;
     info.pending_messages.push_back(message);
   } else {
-    // Filter any messages from previous sessions.
-    if (agent_->session() && agent_->session()->session_id() == session_id)
-      agent_->session()->SendMessageToClient(message);
+    auto session_it = agent_->sessions().find(session_id);
+    if (session_it != agent_->sessions().end())
+      session_it->second->SendMessageToClient(message);
     // |this| may be deleted at this point.
   }
 }
 
 void RenderFrameDevToolsAgentHost::FrameHostHolder::Suspend() {
   suspended_ = true;
 }
 
 void RenderFrameDevToolsAgentHost::FrameHostHolder::Resume() {
   suspended_ = false;
   for (auto& it : session_infos_) {
     SessionInfo& info = it.second;
-    if (agent_->session() && agent_->session()->session_id() == it.first) {
+    auto session_it = agent_->sessions().find(it.first);
+    if (session_it != agent_->sessions().end()) {
       for (const std::string& message : info.pending_messages)
-        agent_->session()->SendMessageToClient(message);
+        session_it->second->SendMessageToClient(message);
     }
     std::vector<std::string> empty;
     info.pending_messages.swap(empty);
     info.sent_messages_whose_reply_came_while_suspended.clear();
   }
 }
 
 // RenderFrameDevToolsAgentHost ------------------------------------------------
 
 // static
@@ skipping 263 matching lines @@
   session->AddHandler(base::WrapUnique(new protocol::TargetHandler()));
   session->AddHandler(base::WrapUnique(new protocol::TracingHandler(
       protocol::TracingHandler::Renderer,
       frame_tree_node_ ? frame_tree_node_->frame_tree_node_id() : 0,
       GetIOContext())));
 
   if (current_)
     current_->Attach(session);
   if (pending_)
     pending_->Attach(session);
-  OnClientAttached();
+  if (sessions().size() == 1)
+    OnClientsAttached();
 }
 
 void RenderFrameDevToolsAgentHost::DetachSession(int session_id) {
   if (current_)
     current_->Detach(session_id);
   if (pending_)
     pending_->Detach(session_id);
-  OnClientDetached();
+  if (sessions().empty())
+    OnClientsDetached();
 }
 
 bool RenderFrameDevToolsAgentHost::DispatchProtocolMessage(
     DevToolsSession* session,
     const std::string& message) {
   int call_id = 0;
   std::string method;
   if (session->Dispatch(message, &call_id, &method) !=
       protocol::Response::kFallThrough) {
     return true;
@@ skipping 20 matching lines @@
 void RenderFrameDevToolsAgentHost::InspectElement(
     DevToolsSession* session,
     int x,
     int y) {
   if (current_)
     current_->InspectElement(session->session_id(), x, y);
   if (pending_)
     pending_->InspectElement(session->session_id(), x, y);
 }
 
-void RenderFrameDevToolsAgentHost::OnClientAttached() {
+void RenderFrameDevToolsAgentHost::OnClientsAttached() {
   if (!web_contents())
     return;
 
   frame_trace_recorder_.reset(new DevToolsFrameTraceRecorder());
   CreatePowerSaveBlocker();
 }
 
-void RenderFrameDevToolsAgentHost::OnClientDetached() {
+void RenderFrameDevToolsAgentHost::OnClientsDetached() {
 #if defined(OS_ANDROID)
   power_save_blocker_.reset();
 #endif
   frame_trace_recorder_.reset();
   in_navigation_protocol_message_buffer_.clear();
 }
 
 RenderFrameDevToolsAgentHost::~RenderFrameDevToolsAgentHost() {
   Instances::iterator it = std::find(g_instances.Get().begin(),
                                      g_instances.Get().end(),
@@ skipping 174 matching lines @@
     FrameDeleted(rfh);
   else
     DCHECK(CheckConsistency());
 }
 
 void RenderFrameDevToolsAgentHost::DestroyOnRenderFrameGone() {
   DCHECK(current_);
   scoped_refptr<RenderFrameDevToolsAgentHost> protect(this);
   UpdateProtocolHandlers(nullptr);
   if (IsAttached())
-    OnClientDetached();
-  ForceDetach(false);
+    OnClientsDetached();
+  ForceDetachAllClients(false);
   pending_.reset();
   current_.reset();
   frame_tree_node_ = nullptr;
   pending_handle_ = nullptr;
   WebContentsObserver::Observe(nullptr);
   Release();
 }
 
 bool RenderFrameDevToolsAgentHost::CheckConsistency() {
   if (current_ && pending_ && current_->host() == pending_->host())
@@ skipping 138 matching lines @@
 #if DCHECK_IS_ON()
   // TODO(dgozman): fix this for browser side navigation.
   if (!IsBrowserSideNavigationEnabled()) {
     // Check that we don't have stale host object here by accessing some random
     // properties inside.
     if (handlers_frame_host_ && handlers_frame_host_->GetRenderWidgetHost())
       handlers_frame_host_->GetRenderWidgetHost()->GetRoutingID();
   }
 #endif
   handlers_frame_host_ = host;
-  if (session())
-    session()->SetRenderFrameHost(host);
+  for (auto& it : sessions())
+    it.second->SetRenderFrameHost(host);
 }
 
 void RenderFrameDevToolsAgentHost::DisconnectWebContents() {
   if (pending_)
     DiscardPending();
   UpdateProtocolHandlers(nullptr);
   disconnected_ = std::move(current_);
   disconnected_->Disconnect();
   frame_tree_node_ = nullptr;
   in_navigation_protocol_message_buffer_.clear();
@@ skipping 182 matching lines @@
     RenderFrameHost* host) {
   return (current_ && current_->host() == host) ||
       (pending_ && pending_->host() == host);
 }
 
 bool RenderFrameDevToolsAgentHost::IsChildFrame() {
   return current_ && current_->host()->GetParent();
 }
 
 }  // namespace content