"""
Autotest profiler for blktrace
blktrace - generate traces of the i/o traffic on block devices
"""
import os
from autotest.client import profiler, utils
from autotest.client.shared import error
[docs]class blktrace(profiler.profiler):
version = 2
[docs] def initialize(self, **dargs):
self.mountpoint = '/sys/kernel/debug'
self.blktrace = os.path.join(self.srcdir, 'blktrace')
self.blkparse = os.path.join(self.srcdir, 'blkparse')
self.blktrace_job = None
self.job.require_gcc()
self.job.setup_dep(['libaio'])
ldflags = '-L ' + self.job.autodir + '/deps/libaio/lib'
cflags = '-I ' + self.job.autodir + '/deps/libaio/include'
self.gcc_flags = ldflags + ' ' + cflags
self.device = dargs.get('device', None)
[docs] def setup(self, tarball='blktrace.tar.bz2', **dargs):
# v1.0.1, pulled from git, 2009/06/10
# commit 1e09f6e9012826fca69fa07222b7bc53c3e629ee
self.tarball = utils.unmap_url(self.bindir, tarball, self.tmpdir)
utils.extract_tarball_to_dir(self.tarball, self.srcdir)
os.chdir(self.srcdir)
utils.make('"CFLAGS=' + self.gcc_flags + '"')
[docs] def get_device(self, test):
if getattr(test, 'device', None):
device = test.device
else:
if self.device:
device = self.device
else:
raise error.TestWarn('No device specified for blktrace')
return device
[docs] def start(self, test):
result = utils.system("mount | grep '%s'" % self.mountpoint,
ignore_status=True)
if result:
utils.system('mount -t debugfs debugfs /sys/kernel/debug')
device = self.get_device(test)
self.blktrace_job = utils.BgJob('%s /dev/%s' % (self.blktrace, device))
[docs] def stop(self, test):
if self.blktrace_job is not None:
utils.nuke_subprocess(self.blktrace_job.sp)
self.blktrace_job = None
[docs] def report(self, test):
output_file = os.path.join(test.profdir, 'blktrace')
if getattr(test, 'profile_tag', None):
output_file += '.' + test.profile_tag
device = self.get_device(test)
utils.system('%s %s > %s' % (self.blkparse, device, output_file))