Add the ability for DevTools to wrap network transactions.

content/browser/devtools/renderer_overrides_handler.cc

 // Copyright (c) 2013 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/renderer_overrides_handler.h"
 
 #include <map>
 #include <string>
 
 #include "base/barrier_closure.h"
 #include "base/base64.h"
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/files/file_path.h"
 #include "base/strings/string16.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/values.h"
 #include "content/browser/child_process_security_policy_impl.h"
 #include "content/browser/devtools/devtools_protocol_constants.h"
 #include "content/browser/devtools/devtools_tracing_handler.h"
 #include "content/browser/renderer_host/dip_util.h"
 #include "content/browser/renderer_host/render_view_host_delegate.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
 #include "content/browser/renderer_host/render_widget_host_view_base.h"
 #include "content/common/view_messages.h"
+#include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/javascript_dialog_manager.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host_view.h"
+#include "content/public/browser/resource_context.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/page_transition_types.h"
 #include "content/public/common/referrer.h"
 #include "ipc/ipc_sender.h"
 #include "net/base/net_util.h"
 #include "third_party/WebKit/public/web/WebInputEvent.h"
 #include "ui/gfx/codec/jpeg_codec.h"
@@ skipping 51 matching lines @@
       devtools::Network::clearBrowserCache::kName,
       base::Bind(
           &RendererOverridesHandler::ClearBrowserCache,
           base::Unretained(this)));
   RegisterCommandHandler(
       devtools::Network::clearBrowserCookies::kName,
       base::Bind(
           &RendererOverridesHandler::ClearBrowserCookies,
           base::Unretained(this)));
   RegisterCommandHandler(
+      devtools::Network::emulateNetworkConditions::kName,
+      base::Bind(
+          &RendererOverridesHandler::EmulateNetworkConditions,
+          base::Unretained(this)));
+  RegisterCommandHandler(
       devtools::Page::disable::kName,
       base::Bind(
           &RendererOverridesHandler::PageDisable, base::Unretained(this)));
   RegisterCommandHandler(
       devtools::Page::handleJavaScriptDialog::kName,
       base::Bind(
           &RendererOverridesHandler::PageHandleJavaScriptDialog,
           base::Unretained(this)));
   RegisterCommandHandler(
       devtools::Page::navigate::kName,
@@ skipping 49 matching lines @@
       devtools::Input::dispatchGestureEvent::kName,
       base::Bind(
           &RendererOverridesHandler::InputDispatchGestureEvent,
           base::Unretained(this)));
 }
 
 RendererOverridesHandler::~RendererOverridesHandler() {}
 
 void RendererOverridesHandler::OnClientDetached() {
   screencast_command_ = NULL;
+  UpdateNetworkState(false);
 }
 
 void RendererOverridesHandler::OnSwapCompositorFrame(
     const cc::CompositorFrameMetadata& frame_metadata) {
   last_compositor_frame_metadata_ = frame_metadata;
 
   if (screencast_command_)
     InnerSwapCompositorFrame();
 }
 
@@ skipping 80 matching lines @@
       base::StringPiece(reinterpret_cast<const char*>(&png_data[0]),
                         png_data.size()),
       &base_64_data);
 
   base::DictionaryValue* response = new base::DictionaryValue();
   response->SetString(
       devtools::Page::captureScreenshot::kResponseData, base_64_data);
   return response;
 }
 
+void RendererOverridesHandler::UpdateNetworkState(bool offline) {
+  RenderViewHost* host = agent_->GetRenderViewHost();
+  if (!host)
+    return;
+  ResourceContext* context = host->GetSiteInstance()->GetProcess()->
+      GetBrowserContext()->GetResourceContext();
+  context->SetNetworkState(agent_->GetId(), offline);
+}
+
 // DOM agent handlers  --------------------------------------------------------
 
 scoped_refptr<DevToolsProtocol::Response>
 RendererOverridesHandler::GrantPermissionsForSetFileInputFiles(
     scoped_refptr<DevToolsProtocol::Command> command) {
   base::DictionaryValue* params = command->params();
   base::ListValue* file_list = NULL;
   const char* param =
       devtools::DOM::setFileInputFiles::kParamFiles;
   if (!params || !params->GetList(param, &file_list))
@@ skipping 22 matching lines @@
   return command->SuccessResponse(NULL);
 }
 
 scoped_refptr<DevToolsProtocol::Response>
 RendererOverridesHandler::ClearBrowserCookies(
     scoped_refptr<DevToolsProtocol::Command> command) {
   GetContentClient()->browser()->ClearCookies(agent_->GetRenderViewHost());
   return command->SuccessResponse(NULL);
 }
 
+scoped_refptr<DevToolsProtocol::Response>
+RendererOverridesHandler::EmulateNetworkConditions(
+    scoped_refptr<DevToolsProtocol::Command> command) {
+  base::DictionaryValue* params = command->params();
+  bool offline = false;
+  const char* offline_param =
+      devtools::Network::emulateNetworkConditions::kParamOffline;
+  if (!params || !params->GetBoolean(offline_param, &offline))
+    return command->InvalidParamResponse(offline_param);
+
+  UpdateNetworkState(offline);
+  return command->SuccessResponse(NULL);
+}
+
 
 // Page agent handlers  -------------------------------------------------------
 
 scoped_refptr<DevToolsProtocol::Response>
 RendererOverridesHandler::PageDisable(
     scoped_refptr<DevToolsProtocol::Command> command) {
   screencast_command_ = NULL;
   return NULL;
 }
 
@@ skipping 666 matching lines @@
       return NULL;
     }
     event.data.pinchUpdate.scale = static_cast<float>(scale);
   }
 
   host->ForwardGestureEvent(event);
   return command->SuccessResponse(NULL);
 }
 
 }  // namespace content