| # Copyright 2015 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 optparse |
| import os |
| import py_utils |
| import re |
| |
| from devil.android import flag_changer |
| from devil.android.perf import cache_control |
| from devil.android.sdk import intent |
| |
| from systrace import trace_result |
| from systrace import tracing_agents |
| |
| |
| class ChromeStartupTracingAgent(tracing_agents.TracingAgent): |
| def __init__(self, device, package_info, cold, url): |
| tracing_agents.TracingAgent.__init__(self) |
| self._device = device |
| self._package_info = package_info |
| self._cold = cold |
| self._logcat_monitor = self._device.GetLogcatMonitor() |
| self._url = url |
| self._trace_file = None |
| self._trace_finish_re = re.compile(r' Completed startup tracing to (.*)') |
| self._flag_changer = flag_changer.FlagChanger( |
| self._device, self._package_info.cmdline_file) |
| |
| def __repr__(self): |
| return 'Browser Startup Trace' |
| |
| def _SetupTracing(self): |
| # TODO(lizeb): Figure out how to clean up the command-line file when |
| # _TearDownTracing() is not executed in StopTracing(). |
| self._flag_changer.AddFlags(['--trace-startup']) |
| self._device.ForceStop(self._package_info.package) |
| if self._cold: |
| self._device.EnableRoot() |
| cache_control.CacheControl(self._device).DropRamCaches() |
| launch_intent = None |
| if self._url == '': |
| launch_intent = intent.Intent( |
| action='android.intent.action.MAIN', |
| package=self._package_info.package, |
| activity=self._package_info.activity) |
| else: |
| launch_intent = intent.Intent( |
| package=self._package_info.package, |
| activity=self._package_info.activity, |
| data=self._url, |
| extras={'create_new_tab': True}) |
| self._device.StartActivity(launch_intent, blocking=True) |
| |
| def _TearDownTracing(self): |
| self._flag_changer.Restore() |
| |
| @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT) |
| def StartAgentTracing(self, config, timeout=None): |
| self._SetupTracing() |
| self._logcat_monitor.Start() |
| return True |
| |
| @py_utils.Timeout(tracing_agents.START_STOP_TIMEOUT) |
| def StopAgentTracing(self, timeout=None): |
| try: |
| self._trace_file = self._logcat_monitor.WaitFor( |
| self._trace_finish_re).group(1) |
| finally: |
| self._TearDownTracing() |
| return True |
| |
| @py_utils.Timeout(tracing_agents.GET_RESULTS_TIMEOUT) |
| def GetResults(self, timeout=None): |
| with open(self._PullTrace(), 'r') as f: |
| trace_data = f.read() |
| return trace_result.TraceResult('traceEvents', trace_data) |
| |
| def _PullTrace(self): |
| trace_file = self._trace_file.replace('/storage/emulated/0/', '/sdcard/') |
| host_file = os.path.join(os.path.curdir, os.path.basename(trace_file)) |
| self._device.PullFile(trace_file, host_file) |
| return host_file |
| |
| def SupportsExplicitClockSync(self): |
| return False |
| |
| def RecordClockSyncMarker(self, sync_id, did_record_sync_marker_callback): |
| # pylint: disable=unused-argument |
| assert self.SupportsExplicitClockSync(), ('Clock sync marker cannot be ' |
| 'recorded since explicit clock sync is not supported.') |
| |
| |
| class ChromeStartupConfig(tracing_agents.TracingConfig): |
| def __init__(self, device, package_info, cold, url, chrome_categories): |
| tracing_agents.TracingConfig.__init__(self) |
| self.device = device |
| self.package_info = package_info |
| self.cold = cold |
| self.url = url |
| self.chrome_categories = chrome_categories |
| |
| |
| def try_create_agent(config): |
| return ChromeStartupTracingAgent(config.device, config.package_info, |
| config.cold, config.url) |
| |
| def add_options(parser): |
| options = optparse.OptionGroup(parser, 'Chrome startup tracing') |
| options.add_option('--url', help='URL to visit on startup. Default: ' |
| 'https://www.google.com. An empty URL launches Chrome ' |
| 'with a MAIN action instead of VIEW.', |
| default='https://www.google.com', metavar='URL') |
| options.add_option('--cold', help='Flush the OS page cache before starting ' |
| 'the browser. Note that this require a device with root ' |
| 'access.', default=False, action='store_true') |
| return options |
| |
| def get_config(options): |
| return ChromeStartupConfig(options.device, options.package_info, |
| options.cold, options.url, |
| options.chrome_categories) |
