You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.8 KiB
80 lines
2.8 KiB
7 months ago
|
"""
|
||
|
|
perf is a tool included in the linux kernel tree that
|
||
|
|
supports functionality similar to oprofile and more.
|
||
|
|
|
||
|
|
@see: http://lwn.net/Articles/310260/
|
||
|
|
"""
|
||
|
|
|
||
|
|
import time, os, stat, subprocess, signal
|
||
|
|
import logging
|
||
|
|
from autotest_lib.client.bin import profiler, os_dep, utils
|
||
|
|
|
||
|
|
|
||
|
|
class perf(profiler.profiler):
|
||
|
|
version = 1
|
||
|
|
|
||
|
|
def initialize(self, events=["cycles","instructions"], trace=False):
|
||
|
|
if type(events) == str:
|
||
|
|
self.events = [events]
|
||
|
|
else:
|
||
|
|
self.events = events
|
||
|
|
self.trace = trace
|
||
|
|
self.perf_bin = os_dep.command('perf')
|
||
|
|
perf_help = utils.run('%s report help' % self.perf_bin,
|
||
|
|
ignore_status=True).stderr
|
||
|
|
self.sort_keys = None
|
||
|
|
for line in perf_help.split('\n'):
|
||
|
|
a = "sort by key(s):"
|
||
|
|
if a in line:
|
||
|
|
line = line.replace(a, "")
|
||
|
|
self.sort_keys = [k.rstrip(",") for k in line.split() if
|
||
|
|
k.rstrip(",") != 'dso']
|
||
|
|
if not self.sort_keys:
|
||
|
|
self.sort_keys = ['comm', 'cpu']
|
||
|
|
|
||
|
|
|
||
|
|
def start(self, test):
|
||
|
|
self.logfile = os.path.join(test.profdir, "perf")
|
||
|
|
cmd = ("exec %s record -a -o %s" %
|
||
|
|
(self.perf_bin, self.logfile))
|
||
|
|
if "parent" in self.sort_keys:
|
||
|
|
cmd += " -g"
|
||
|
|
if self.trace:
|
||
|
|
cmd += " -R"
|
||
|
|
for event in self.events:
|
||
|
|
cmd += " -e %s" % event
|
||
|
|
self._process = subprocess.Popen(cmd, shell=True,
|
||
|
|
stderr=subprocess.STDOUT)
|
||
|
|
|
||
|
|
|
||
|
|
def stop(self, test):
|
||
|
|
os.kill(self._process.pid, signal.SIGINT)
|
||
|
|
self._process.wait()
|
||
|
|
|
||
|
|
|
||
|
|
def report(self, test):
|
||
|
|
for key in self.sort_keys:
|
||
|
|
reportfile = os.path.join(test.profdir, '%s.comm' % key)
|
||
|
|
cmd = ("%s report -i %s --sort %s,dso" % (self.perf_bin,
|
||
|
|
self.logfile,
|
||
|
|
key))
|
||
|
|
outfile = open(reportfile, 'w')
|
||
|
|
p = subprocess.Popen(cmd, shell=True, stdout=outfile,
|
||
|
|
stderr=subprocess.STDOUT)
|
||
|
|
p.wait()
|
||
|
|
|
||
|
|
if self.trace:
|
||
|
|
tracefile = os.path.join(test.profdir, 'trace')
|
||
|
|
cmd = ("%s script -i %s" % (self.perf_bin, self.logfile,))
|
||
|
|
|
||
|
|
outfile = open(tracefile, 'w')
|
||
|
|
p = subprocess.Popen(cmd, shell=True, stdout=outfile,
|
||
|
|
stderr=subprocess.STDOUT)
|
||
|
|
p.wait()
|
||
|
|
|
||
|
|
# The raw detailed perf output is HUGE. We cannot store it by default.
|
||
|
|
perf_log_size = os.stat(self.logfile)[stat.ST_SIZE]
|
||
|
|
logging.info('Removing %s after generating reports (saving %s bytes).',
|
||
|
|
self.logfile, perf_log_size)
|
||
|
|
os.unlink(self.logfile)
|