Telemetry / Devtools TraceHandler: exposes tracing via dev tools.

content/browser/debugger/devtools_http_handler_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/debugger/devtools_http_handler_impl.h"
 
 #include <algorithm>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/compiler_specific.h"
 #include "base/file_util.h"
 #include "base/json/json_writer.h"
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/message_loop_proxy.h"
 #include "base/string_number_conversions.h"
 #include "base/stringprintf.h"
 #include "base/threading/thread.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
+#include "content/browser/debugger/devtools_browser_target.h"
+#include "content/browser/debugger/devtools_trace_domain_handler.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/devtools_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/devtools_agent_host_registry.h"
 #include "content/public/browser/devtools_client_host.h"
 #include "content/public/browser/devtools_http_handler_delegate.h"
 #include "content/public/browser/devtools_manager.h"
 #include "content/public/browser/favicon_status.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
@@ skipping 269 matching lines @@
     BrowserThread::PostTask(
         BrowserThread::UI,
         FROM_HERE,
         base::Bind(&DevToolsHttpHandlerImpl::OnThumbnailRequestUI,
                    this,
                    connection_id,
                    info));
     return;
   }
 
+  if (info.path.find("/trace/") == 0) {

[pfeldman, 2012/12/12 20:17:12]

nuke

[bulach, 2012/12/13 17:36:23]

Done.

+    // Trace request.
+    BrowserThread::PostTask(
+        BrowserThread::UI,
+        FROM_HERE,
+        base::Bind(&DevToolsHttpHandlerImpl::OnTraceRequestUI,
+                   this,
+                   connection_id,
+                   info));
+    return;
+  }
+
   if (info.path == "" || info.path == "/") {
     // Discovery page request.
     BrowserThread::PostTask(
         BrowserThread::UI,
         FROM_HERE,
         base::Bind(&DevToolsHttpHandlerImpl::OnDiscoveryPageRequestUI,
                    this,
                    connection_id));
     return;
   }
@@ skipping 250 matching lines @@
   }
 
   std::string page_url = info.path.substr(prefix.length());
   std::string data = delegate_->GetPageThumbnailData(GURL(page_url));
   if (!data.empty())
     Send200(connection_id, data, "image/png");
   else
     Send404(connection_id);
 }
 
+void DevToolsHttpHandlerImpl::OnTraceRequestUI(
+    int connection_id,
+    const net::HttpServerRequestInfo& info) {
+/*
+  std::string prefix = "/trace/";
+  size_t pos = info.path.find(prefix);
+  if (pos != 0) {
+    Send404(connection_id);
+    return;
+  }
+
+  std::string command = info.path.substr(prefix.length());
+
+  std::string response("ok");
+  if (command == "start") {
+    // TODO(bulach): capture categories somehow.
+    std::string categories;
+    devtools_trace_handler_.BeginTracing(categories);
+  } else if (command == "stop")
+    devtools_trace_handler_.EndTracingAsync();
+  else if (command == "has_completed") {
+    response = devtools_trace_handler_.has_completed() ?
+        "ok" : "pending";
+  } else if (command == "get") {
+    if (devtools_trace_handler_.has_completed()) {
+      Send200(
+          connection_id, devtools_trace_handler_.GetAndResetTrace(),
+          "binary/octet-stream");
+      return;
+    }
+    Send404(connection_id);
+    return;
+  }
+  Send200(connection_id, response, "text/html; charset=UTF-8");
+*/
+}
+
 void DevToolsHttpHandlerImpl::OnDiscoveryPageRequestUI(int connection_id) {
   std::string response = delegate_->GetDiscoveryPageHTML();
   Send200(connection_id, response, "text/html; charset=UTF-8");
 }
 
 void DevToolsHttpHandlerImpl::OnWebSocketRequestUI(
     int connection_id,
     const net::HttpServerRequestInfo& request) {
   if (!thread_.get())
     return;
 
-  std::string prefix = "/devtools/page/";
-  size_t pos = request.path.find(prefix);
+  std::string browser_prefix = "/devtools/browser/";

[pfeldman, 2012/12/12 20:17:12]

No need for the trailing slash here

[bulach, 2012/12/13 17:36:23]

Done.

+  size_t browser_pos = request.path.find(browser_prefix);
+  if (browser_pos == 0) {
+    browser_target_.reset(new DevToolsBrowserTarget(connection_id));
+    browser_target_->RegisterHandler(new DevToolsTraceDomainHandler());
+    AcceptWebSocket(connection_id, request);
+    return;
+  }
+
+
+  std::string page_prefix = "/devtools/page/";
+  size_t pos = request.path.find(page_prefix);
   if (pos != 0) {
     Send404(connection_id);
     return;
   }
 
-  std::string page_id = request.path.substr(prefix.length());
+  std::string page_id = request.path.substr(page_prefix.length());
   RenderViewHost* rvh = binding_->ForIdentifier(page_id);
   if (!rvh) {
     Send500(connection_id, "No such target id: " + page_id);
     return;
   }
 
   DevToolsManager* manager = DevToolsManager::GetInstance();
   DevToolsAgentHost* agent = DevToolsAgentHostRegistry::GetDevToolsAgentHost(
       rvh);
   if (manager->GetDevToolsClientHostFor(agent)) {
@@ skipping 14 matching lines @@
 }
 
 void DevToolsHttpHandlerImpl::OnWebSocketMessageUI(
     int connection_id,
     const std::string& data) {
   ConnectionToClientHostMap::iterator it =
       connection_to_client_host_ui_.find(connection_id);
   if (it == connection_to_client_host_ui_.end())
     return;
 
+  if (browser_target_ && connection_id == browser_target_->connection_id()) {
+    browser_target_->OnWebSocketMessage(connection_id, data, this);
+    return;
+  }
+
   DevToolsManager* manager = DevToolsManager::GetInstance();
   manager->DispatchOnInspectorBackend(it->second, data);
 }
 
 void DevToolsHttpHandlerImpl::OnCloseUI(int connection_id) {
   ConnectionToClientHostMap::iterator it =
       connection_to_client_host_ui_.find(connection_id);
   if (it != connection_to_client_host_ui_.end()) {

[pfeldman, 2012/12/12 20:17:12]

You should delete browser_target_ here.

[bulach, 2012/12/13 17:36:23]

Done.

     DevToolsClientHostImpl* client_host =
         static_cast<DevToolsClientHostImpl*>(it->second);
     DevToolsManager::GetInstance()->ClientHostClosing(client_host);
     delete client_host;
     connection_to_client_host_ui_.erase(connection_id);
   }
 }
 
 DevToolsHttpHandlerImpl::DevToolsHttpHandlerImpl(
     const net::StreamListenSocketFactory* socket_factory,
@@ skipping 173 matching lines @@
                                              page_info.id.c_str()));
     std::string devtools_frontend_url = GetFrontendURLInternal(
         page_info.id.c_str(),
         host);
     dictionary->SetString("devtoolsFrontendUrl", devtools_frontend_url);
   }
   return dictionary;
 }
 
 }  // namespace content