DevTools: Enable network emulation on Service Workers

content/browser/devtools/devtools_agent_host_impl.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/devtools_agent_host_impl.h"
 
 #include <map>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/guid.h"
+#include "base/json/json_writer.h"
 #include "base/lazy_instance.h"
 #include "content/browser/devtools/devtools_manager.h"
 #include "content/browser/devtools/forwarding_agent_host.h"
+#include "content/browser/devtools/protocol/devtools_protocol_handler.h"
 #include "content/browser/devtools/render_frame_devtools_agent_host.h"
 #include "content/browser/devtools/service_worker_devtools_agent_host.h"
 #include "content/browser/devtools/service_worker_devtools_manager.h"
 #include "content/browser/devtools/shared_worker_devtools_agent_host.h"
 #include "content/browser/devtools/shared_worker_devtools_manager.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/devtools_manager_delegate.h"
 
 namespace content {
 
@@ skipping 33 matching lines @@
       SharedWorkerDevToolsManager::GetInstance()
           ->GetDevToolsAgentHostForWorker(worker_process_id,
                                           worker_route_id)) {
     return host;
   }
   return ServiceWorkerDevToolsManager::GetInstance()
       ->GetDevToolsAgentHostForWorker(worker_process_id, worker_route_id);
 }
 
 DevToolsAgentHostImpl::DevToolsAgentHostImpl()
-    : id_(base::GenerateGUID()),
+    : protocol_handler_(new DevToolsProtocolHandler(
+          base::Bind(&DevToolsAgentHostImpl::DispatchOnInspectorFrontend,
+                     base::Unretained(this)))),
+      id_(base::GenerateGUID()),
       client_(NULL) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   g_instances.Get()[id_] = this;
 }
 
 DevToolsAgentHostImpl::~DevToolsAgentHostImpl() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   g_instances.Get().erase(g_instances.Get().find(id_));
 }
 
@@ skipping 133 matching lines @@
   for (AgentStateCallbacks::iterator it = copy.begin(); it != copy.end(); ++it)
      (*it)->Run(agent_host, attached);
 }
 
 void DevToolsAgentHostImpl::Inspect(BrowserContext* browser_context) {
   DevToolsManager* manager = DevToolsManager::GetInstance();
   if (manager->delegate())
     manager->delegate()->Inspect(browser_context, this);
 }
 
+bool DevToolsAgentHostImpl::DispatchProtocolMessage(
+    const std::string& message) {
+  scoped_ptr<base::DictionaryValue> command =
+      protocol_handler_->ParseCommand(message);
+  if (!command)
+    return false;

[pfeldman, 2015/03/23 09:59:00]

See my comment below, this should be "true" since you consider malformed request
handled at this point.

[Kunihiko Sakamoto, 2015/03/23 10:23:58]

Done.

+
+  DevToolsManagerDelegate* delegate =
+      DevToolsManager::GetInstance()->delegate();
+  if (delegate) {
+    scoped_ptr<base::DictionaryValue> response(
+        delegate->HandleCommand(this, command.get()));
+    if (response) {
+      std::string json_response;
+      base::JSONWriter::Write(response.get(), &json_response);
+      DispatchOnInspectorFrontend(json_response);
+      return true;
+    }
+  }
+
+  if (protocol_handler_->HandleOptionalCommand(command.Pass()))
+    return true;
+
+  return false;
+}
+
+void DevToolsAgentHostImpl::DispatchOnInspectorFrontend(
+    const std::string& message) {
+  if (!IsAttached())

[pfeldman, 2015/03/23 09:59:00]

This check is already present in SendMessageToClient.

+    return;
+  SendMessageToClient(message);
+}
+
 }  // namespace content