Add the ability for DevTools to wrap network transactions.

content/browser/devtools/render_view_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_view_devtools_agent_host.h"
 
 #include "base/basictypes.h"
 #include "base/lazy_instance.h"
 #include "content/browser/child_process_security_policy_impl.h"
 #include "content/browser/devtools/devtools_manager_impl.h"
@@ skipping 123 matching lines @@
     const IPC::Message& message) {
   RenderViewDevToolsAgentHost* agent_host = FindAgentHost(source);
   return agent_host && agent_host->DispatchIPCMessage(message);
 }
 
 RenderViewDevToolsAgentHost::RenderViewDevToolsAgentHost(
     RenderViewHost* rvh)
     : render_view_host_(NULL),
       overrides_handler_(new RendererOverridesHandler(this)),
       tracing_handler_(new DevToolsTracingHandler()),
-      power_handler_(new DevToolsPowerHandler())
+      power_handler_(new DevToolsPowerHandler()),
+      network_disabled_(false)
  {
   SetRenderViewHost(rvh);
   DevToolsProtocol::Notifier notifier(base::Bind(
       &RenderViewDevToolsAgentHost::OnDispatchOnInspectorFrontend,
       base::Unretained(this)));
   overrides_handler_->SetNotifier(notifier);
   tracing_handler_->SetNotifier(notifier);
   power_handler_->SetNotifier(notifier);
   g_instances.Get().push_back(this);
   AddRef();  // Balanced in RenderViewHostDestroyed.
@@ skipping 37 matching lines @@
   if (!render_view_host_)
     return;
 
   ChildProcessSecurityPolicyImpl::GetInstance()->GrantReadRawCookies(
       render_view_host_->GetProcess()->GetID());
 
   // TODO(kaznacheev): Move this call back to DevToolsManagerImpl when
   // extensions::ProcessManager no longer relies on this notification.
   DevToolsManagerImpl::GetInstance()->NotifyObservers(this, true);
 
+  UpdateNetworkState(network_disabled_);
+
 #if defined(OS_ANDROID)
   power_save_blocker_.reset(
       static_cast<PowerSaveBlockerImpl*>(
           PowerSaveBlocker::Create(
               PowerSaveBlocker::kPowerSaveBlockPreventDisplaySleep,
               "DevTools").release()));
   if (render_view_host_->GetView()) {
     power_save_blocker_.get()->
         InitDisplaySleepBlocker(render_view_host_->GetView()->GetNativeView());
   }
 #endif
 }
 
 void RenderViewDevToolsAgentHost::OnClientDetached() {
 #if defined(OS_ANDROID)
   power_save_blocker_.reset();
 #endif
+  UpdateNetworkState(false);
+
   overrides_handler_->OnClientDetached();
   ClientDetachedFromRenderer();
 }
 
 void RenderViewDevToolsAgentHost::ClientDetachedFromRenderer() {
   if (!render_view_host_)
     return;
 
   bool process_has_agents = false;
   RenderProcessHost* render_process_host = render_view_host_->GetProcess();
@@ skipping 145 matching lines @@
 
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(RenderViewDevToolsAgentHost, msg)
     IPC_MESSAGE_HANDLER(DevToolsClientMsg_DispatchOnInspectorFrontend,
                         OnDispatchOnInspectorFrontend)
     IPC_MESSAGE_HANDLER(DevToolsHostMsg_SaveAgentRuntimeState,
                         OnSaveAgentRuntimeState)
     IPC_MESSAGE_HANDLER(DevToolsHostMsg_ClearBrowserCache, OnClearBrowserCache)
     IPC_MESSAGE_HANDLER(DevToolsHostMsg_ClearBrowserCookies,
                         OnClearBrowserCookies)
+    IPC_MESSAGE_HANDLER(DevToolsHostMsg_DisableNetwork,
+                        OnDisableNetwork)
     IPC_MESSAGE_HANDLER_GENERIC(ViewHostMsg_SwapCompositorFrame,
                                 handled = false; OnSwapCompositorFrame(msg))
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void RenderViewDevToolsAgentHost::OnSwapCompositorFrame(
     const IPC::Message& message) {
   ViewHostMsg_SwapCompositorFrame::Param param;
@@ skipping 27 matching lines @@
 void RenderViewDevToolsAgentHost::OnClearBrowserCache() {
   if (render_view_host_)
     GetContentClient()->browser()->ClearCache(render_view_host_);
 }
 
 void RenderViewDevToolsAgentHost::OnClearBrowserCookies() {
   if (render_view_host_)
     GetContentClient()->browser()->ClearCookies(render_view_host_);
 }
 
+void RenderViewDevToolsAgentHost::UpdateNetworkState(bool disable_network) {
+  if (render_view_host_) {
+    GetContentClient()->browser()->DisableNetwork(
+        render_view_host_, GetId(), disable_network);
+  }
+}
+
+void RenderViewDevToolsAgentHost::OnDisableNetwork(
+  bool disable_network) {
+  network_disabled_ = disable_network;
+  UpdateNetworkState(disable_network);
+}
+
 }  // namespace content