| 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 threading | |
| 6 import zlib | |
| 7 | |
| 8 from devil.utils import cmd_helper | |
| 9 | |
| 10 from profile_chrome import controllers | |
| 11 from profile_chrome import util | |
| 12 | |
| 13 | |
| 14 _SYSTRACE_OPTIONS = [ | |
| 15 # Compress the trace before sending it over USB. | |
| 16 '-z', | |
| 17 # Use a large trace buffer to increase the polling interval. | |
| 18 '-b', '16384' | |
| 19 ] | |
| 20 | |
| 21 # Interval in seconds for sampling systrace data. | |
| 22 _SYSTRACE_INTERVAL = 15 | |
| 23 | |
| 24 _TRACING_ON_PATH = '/sys/kernel/debug/tracing/tracing_on' | |
| 25 | |
| 26 | |
| 27 class SystraceController(controllers.BaseController): | |
| 28 def __init__(self, device, categories, ring_buffer): | |
| 29 controllers.BaseController.__init__(self) | |
| 30 self._device = device | |
| 31 self._categories = categories | |
| 32 self._ring_buffer = ring_buffer | |
| 33 self._done = threading.Event() | |
| 34 self._thread = None | |
| 35 self._trace_data = None | |
| 36 | |
| 37 def __repr__(self): | |
| 38 return 'systrace' | |
| 39 | |
| 40 @staticmethod | |
| 41 def GetCategories(device): | |
| 42 return device.RunShellCommand('atrace --list_categories') | |
| 43 | |
| 44 def StartTracing(self, _): | |
| 45 self._thread = threading.Thread(target=self._CollectData) | |
| 46 self._thread.start() | |
| 47 | |
| 48 def StopTracing(self): | |
| 49 self._done.set() | |
| 50 | |
| 51 def PullTrace(self): | |
| 52 self._thread.join() | |
| 53 self._thread = None | |
| 54 if self._trace_data: | |
| 55 output_name = 'systrace-%s' % util.GetTraceTimestamp() | |
| 56 with open(output_name, 'w') as out: | |
| 57 out.write(self._trace_data) | |
| 58 return output_name | |
| 59 | |
| 60 def IsTracingOn(self): | |
| 61 result = self._RunAdbShellCommand(['cat', _TRACING_ON_PATH]) | |
| 62 return result.strip() == '1' | |
| 63 | |
| 64 def _RunAdbShellCommand(self, command): | |
| 65 # We use a separate interface to adb because the one from AndroidCommands | |
| 66 # isn't re-entrant. | |
| 67 # TODO(jbudorick) Look at providing a way to unhandroll this once the | |
| 68 # adb rewrite has fully landed. | |
| 69 device_param = (['-s', str(self._device)] if str(self._device) else []) | |
| 70 cmd = ['adb'] + device_param + ['shell'] + command | |
| 71 return cmd_helper.GetCmdOutput(cmd) | |
| 72 | |
| 73 def _RunATraceCommand(self, command): | |
| 74 cmd = ['atrace', '--%s' % command] + _SYSTRACE_OPTIONS + self._categories | |
| 75 return self._RunAdbShellCommand(cmd) | |
| 76 | |
| 77 def _ForceStopAtrace(self): | |
| 78 # atrace on pre-M Android devices cannot be stopped asynchronously | |
| 79 # correctly. Use synchronous mode to force stop. | |
| 80 cmd = ['atrace', '-t', '0'] | |
| 81 return self._RunAdbShellCommand(cmd) | |
| 82 | |
| 83 def _CollectData(self): | |
| 84 trace_data = [] | |
| 85 self._RunATraceCommand('async_start') | |
| 86 try: | |
| 87 while not self._done.is_set(): | |
| 88 self._done.wait(_SYSTRACE_INTERVAL) | |
| 89 if not self._ring_buffer or self._done.is_set(): | |
| 90 trace_data.append( | |
| 91 self._DecodeTraceData(self._RunATraceCommand('async_dump'))) | |
| 92 finally: | |
| 93 trace_data.append( | |
| 94 self._DecodeTraceData(self._RunATraceCommand('async_stop'))) | |
| 95 if self.IsTracingOn(): | |
| 96 self._ForceStopAtrace() | |
| 97 self._trace_data = ''.join([zlib.decompress(d) for d in trace_data]) | |
| 98 | |
| 99 @staticmethod | |
| 100 def _DecodeTraceData(trace_data): | |
| 101 try: | |
| 102 trace_start = trace_data.index('TRACE:') | |
| 103 except ValueError: | |
| 104 raise RuntimeError('Systrace start marker not found') | |
| 105 trace_data = trace_data[trace_start + 6:] | |
| 106 | |
| 107 # Collapse CRLFs that are added by adb shell. | |
| 108 if trace_data.startswith('\r\n'): | |
| 109 trace_data = trace_data.replace('\r\n', '\n') | |
| 110 | |
| 111 # Skip the initial newline. | |
| 112 return trace_data[1:] | |
| OLD | NEW |