| OLD | NEW |
| (Empty) |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import json | |
| 6 import os | |
| 7 import re | |
| 8 import time | |
| 9 | |
| 10 from profile_chrome import controllers | |
| 11 | |
| 12 from devil.android import device_errors | |
| 13 from devil.android.sdk import intent | |
| 14 | |
| 15 | |
| 16 _HEAP_PROFILE_MMAP_PROPERTY = 'heapprof.mmap' | |
| 17 | |
| 18 class ChromeTracingController(controllers.BaseController): | |
| 19 def __init__(self, device, package_info, | |
| 20 categories, ring_buffer, trace_memory=False): | |
| 21 controllers.BaseController.__init__(self) | |
| 22 self._device = device | |
| 23 self._package_info = package_info | |
| 24 self._categories = categories | |
| 25 self._ring_buffer = ring_buffer | |
| 26 self._logcat_monitor = self._device.GetLogcatMonitor() | |
| 27 self._trace_file = None | |
| 28 self._trace_interval = None | |
| 29 self._trace_memory = trace_memory | |
| 30 self._is_tracing = False | |
| 31 self._trace_start_re = \ | |
| 32 re.compile(r'Logging performance trace to file') | |
| 33 self._trace_finish_re = \ | |
| 34 re.compile(r'Profiler finished[.] Results are in (.*)[.]') | |
| 35 | |
| 36 def __repr__(self): | |
| 37 return 'chrome trace' | |
| 38 | |
| 39 @staticmethod | |
| 40 def GetCategories(device, package_info): | |
| 41 with device.GetLogcatMonitor() as logmon: | |
| 42 device.BroadcastIntent(intent.Intent( | |
| 43 action='%s.GPU_PROFILER_LIST_CATEGORIES' % package_info.package)) | |
| 44 try: | |
| 45 json_category_list = logmon.WaitFor( | |
| 46 re.compile(r'{"traceCategoriesList(.*)'), timeout=5).group(0) | |
| 47 except device_errors.CommandTimeoutError: | |
| 48 raise RuntimeError('Performance trace category list marker not found. ' | |
| 49 'Is the correct version of the browser running?') | |
| 50 | |
| 51 record_categories = set() | |
| 52 disabled_by_default_categories = set() | |
| 53 json_data = json.loads(json_category_list)['traceCategoriesList'] | |
| 54 for item in json_data: | |
| 55 for category in item.split(','): | |
| 56 if category.startswith('disabled-by-default'): | |
| 57 disabled_by_default_categories.add(category) | |
| 58 else: | |
| 59 record_categories.add(category) | |
| 60 | |
| 61 return list(record_categories), list(disabled_by_default_categories) | |
| 62 | |
| 63 def StartTracing(self, interval): | |
| 64 self._trace_interval = interval | |
| 65 self._logcat_monitor.Start() | |
| 66 start_extras = {'categories': ','.join(self._categories)} | |
| 67 if self._ring_buffer: | |
| 68 start_extras['continuous'] = None | |
| 69 self._device.BroadcastIntent(intent.Intent( | |
| 70 action='%s.GPU_PROFILER_START' % self._package_info.package, | |
| 71 extras=start_extras)) | |
| 72 | |
| 73 if self._trace_memory: | |
| 74 self._device.EnableRoot() | |
| 75 self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 1) | |
| 76 | |
| 77 # Chrome logs two different messages related to tracing: | |
| 78 # | |
| 79 # 1. "Logging performance trace to file" | |
| 80 # 2. "Profiler finished. Results are in [...]" | |
| 81 # | |
| 82 # The first one is printed when tracing starts and the second one indicates | |
| 83 # that the trace file is ready to be pulled. | |
| 84 try: | |
| 85 self._logcat_monitor.WaitFor(self._trace_start_re, timeout=5) | |
| 86 self._is_tracing = True | |
| 87 except device_errors.CommandTimeoutError: | |
| 88 raise RuntimeError( | |
| 89 'Trace start marker not found. Possible causes: 1) Is the correct ' | |
| 90 'version of the browser running? 2) Is the browser already launched?') | |
| 91 | |
| 92 def StopTracing(self): | |
| 93 if self._is_tracing: | |
| 94 self._device.BroadcastIntent(intent.Intent( | |
| 95 action='%s.GPU_PROFILER_STOP' % self._package_info.package)) | |
| 96 self._trace_file = self._logcat_monitor.WaitFor( | |
| 97 self._trace_finish_re, timeout=120).group(1) | |
| 98 self._is_tracing = False | |
| 99 if self._trace_memory: | |
| 100 self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 0) | |
| 101 | |
| 102 def PullTrace(self): | |
| 103 # Wait a bit for the browser to finish writing the trace file. | |
| 104 time.sleep(self._trace_interval / 4 + 1) | |
| 105 | |
| 106 trace_file = self._trace_file.replace('/storage/emulated/0/', '/sdcard/') | |
| 107 host_file = os.path.join(os.path.curdir, os.path.basename(trace_file)) | |
| 108 try: | |
| 109 self._device.PullFile(trace_file, host_file) | |
| 110 except device_errors.AdbCommandFailedError: | |
| 111 raise RuntimeError( | |
| 112 'Cannot pull the trace file. Have you granted Storage permission to ' | |
| 113 'the browser? (Android Settings -> Apps -> [the browser app] -> ' | |
| 114 'Permissions -> Storage)') | |
| 115 return host_file | |
| OLD | NEW |