Telemetry / Devtools TraceHandler: exposes tracing via dev tools.

tools/telemetry/telemetry/page_runner.py

 # 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.
 import logging
 import os
 import time
 import traceback
 import urlparse
 import random
 
 from telemetry import browser_gone_exception
 from telemetry import page_test
 from telemetry import tab_crash_exception
 from telemetry import util
 from telemetry import wpr_modes
 
 class PageState(object):
   def __init__(self):
     self.did_login = False
 
 class _RunState(object):
   def __init__(self):
     self.first_browser = True
     self.browser = None
     self.tab = None
-    self.trace_tab = None
+    self.is_tracing = False
 
   def Close(self):
-    if self.trace_tab:
-      self.trace_tab.Disconnect()
-      self.trace_tab = None
-
     if self.tab:
       self.tab.Disconnect()
       self.tab = None
 
     if self.browser:
       self.browser.Close()
       self.browser = None
 
 def _ShufflePageSet(page_set, options):
   if options.test_shuffle_order_file and not options.test_shuffle:
@@ skipping 86 matching lines @@
             except tab_crash_exception.TabCrashException:
               if not options.show_stdout:
                 stdout = state.browser.GetStandardOutput()
                 stdout = (('\nStandard Output:\n') +
                           ('*' * 80) +
                           '\n\t' + stdout.replace('\n', '\n\t') + '\n' +
                           ('*' * 80))
               logging.warning('Tab crashed: %s%s', page.url, stdout)
               state.Close()
 
-            if options.trace_dir and state.trace_tab:
+            if options.trace_dir:
               self._EndTracing(state, options, page)
             break
           except browser_gone_exception.BrowserGoneException:
             logging.warning('Lost connection to browser. Retrying.')
             state.Close()
             tries -= 1
             if not tries:
               logging.error('Lost connection to browser 3 times. Failing.')
               raise
     finally:
@@ skipping 71 matching lines @@
     state.browser.credentials.credentials_path = credentials_path
     test.SetUpBrowser(state.browser)
 
     if state.first_browser:
       state.browser.credentials.WarnIfMissingCredentials(self.page_set)
       state.first_browser = False
 
     state.browser.SetReplayArchivePath(archive_path)
 
   def _SetupTracingTab(self, state):
-    if not state.trace_tab:
-      # Swap the two tabs because new tabs open to about:blank, and we
-      # can't navigate across protocols to chrome://tracing. The initial
-      # tab starts at chrome://newtab, so it works for that tab.
-      # TODO(dtu): If the trace_tab crashes, we're hosed.
-      state.trace_tab = state.tab
-      state.tab = state.browser.tabs.New()
-
-      state.trace_tab.page.Navigate('chrome://tracing')
-      state.trace_tab.WaitForDocumentReadyStateToBeInteractiveOrBetter()
-
-    # Start tracing.
-    state.trace_tab.runtime.Execute('tracingController.beginTracing('
-        'tracingController.supportsSystemTracing);')
+    if state.browser.supports_tracing:
+      state.is_tracing = True
+      state.browser.StartTracing()
 
   def _EndTracing(self, state, options, page):
-    def IsTracingRunning():
-      return state.trace_tab.runtime.Evaluate(
-          'tracingController.isTracingEnabled')
-    # Tracing might have ended already if the buffer filled up.
-    if IsTracingRunning():
-      state.trace_tab.runtime.Execute('tracingController.endTracing()')
-    util.WaitFor(lambda: not IsTracingRunning(), 10)
+    if state.is_tracing:
+      state.is_tracing = False
+      state.browser.StopTracing()
+      trace = state.browser.GetTrace()
+      logging.info('Processing trace...')
 
-    logging.info('Processing trace...')
+      trace_file_base = os.path.join(
+          options.trace_dir, page.url_as_file_safe_name)
 
-    trace_file_base = os.path.join(
-        options.trace_dir, page.url_as_file_safe_name)
+      if options.page_repeat != 1 or options.pageset_repeat != 1:
+        trace_file_index = 0
 
-    if options.page_repeat != 1 or options.pageset_repeat != 1:
-      trace_file_index = 0
-
-      while True:
-        trace_file = '%s_%03d.json' % (trace_file_base, trace_file_index)
-        if not os.path.exists(trace_file):
-          break
-        trace_file_index = trace_file_index + 1
-    else:
-      trace_file = '%s.json' % trace_file_base
-
-    with open(trace_file, 'w') as trace_file:
-      trace_file.write(state.trace_tab.runtime.Evaluate("""
-        JSON.stringify({
-          traceEvents: tracingController.traceEvents,
-          systemTraceEvents: tracingController.systemTraceEvents,
-          clientInfo: tracingController.clientInfo_,
-          gpuInfo: tracingController.gpuInfo_
-        });
-      """))
-    logging.info('Trace saved.')
+        while True:
+          trace_file = '%s_%03d.json' % (trace_file_base, trace_file_index)
+          if not os.path.exists(trace_file):
+            break
+          trace_file_index = trace_file_index + 1
+      else:
+        trace_file = '%s.json' % trace_file_base
+      with open(trace_file, 'w') as trace_file:
+        trace_file.write(trace)
+      logging.info('Trace saved.')
 
   def _PreparePage(self, page, tab, page_state, results):
     parsed_url = urlparse.urlparse(page.url)
     if parsed_url[0] == 'file':
       dirname, filename = page.url_base_dir_and_file
       tab.browser.SetHTTPServerDirectory(dirname)
       target_side_url = tab.browser.http_server.UrlOf(filename)
     else:
       target_side_url = page.url
 
@@ skipping 19 matching lines @@
     return True
 
   def _CleanUpPage(self, page, tab, page_state): # pylint: disable=R0201
     if page.credentials and page_state.did_login:
       tab.browser.credentials.LoginNoLongerNeeded(tab, page.credentials)
     try:
       tab.runtime.Evaluate("""window.chrome && chrome.benchmarking &&
                               chrome.benchmarking.closeConnections()""")
     except Exception:
       pass