1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
# 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 logging
import optparse
import os
import re
import py_utils
from devil.android import flag_changer
from devil.android.constants import webapk
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, webapk_package, cold, url,
trace_time=None):
tracing_agents.TracingAgent.__init__(self)
self._device = device
self._package_info = package_info
self._webapk_package = webapk_package
self._cold = cold
self._logcat_monitor = self._device.GetLogcatMonitor()
self._url = url
self._trace_time = trace_time
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().
flags = ['--trace-startup', '--enable-perfetto']
if self._trace_time is not None:
flags.append('--trace-startup-duration={}'.format(self._trace_time))
self._flag_changer.AddFlags(flags)
self._device.ForceStop(self._package_info.package)
if self._webapk_package:
self._device.ForceStop(self._webapk_package)
logging.warning('Forces to stop the WebAPK and the browser provided by '
'--browser: %s. Please make sure that this browser '
'matches the host browser of the WebAPK %s. ',
self._package_info.package,
self._webapk_package)
if self._cold:
self._device.EnableRoot()
cache_control.CacheControl(self._device).DropRamCaches()
launch_intent = None
if self._webapk_package:
launch_intent = intent.Intent(
package=self._webapk_package,
activity=webapk.WEBAPK_MAIN_ACTIVITY,
data=self._url)
elif 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._logcat_monitor.Start()
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()
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, webapk_package, cold, url,
chrome_categories, trace_time):
tracing_agents.TracingConfig.__init__(self)
self.device = device
self.package_info = package_info
self.webapk_package = webapk_package
self.cold = cold
self.url = url
self.chrome_categories = chrome_categories
self.trace_time = trace_time
def try_create_agent(config):
return ChromeStartupTracingAgent(config.device, config.package_info,
config.webapk_package,
config.cold, config.url, config.trace_time)
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')
options.add_option('--webapk-package', help='Specify the package name '
'of the WebAPK to launch the given URL. An empty URL '
'laucnhes the host browser of the WebAPK with an new '
'tab.', default=None)
return options
def get_config(options):
return ChromeStartupConfig(options.device, options.package_info,
options.webapk_package, options.cold,
options.url, options.chrome_categories,
options.trace_time)
|
